def test_set_4_problem_29(self):
cry = Crypto_Kit()
key = cry.generate_random_bytes(randint(5, 20))
original_message = b"comment1=cooking%20MCs;userdata=foo;comment2=%20like%20a%20pound%20of%20bacon"
hash = cry.sha1(key + original_message)
registers = [int(r, 16) for r in list(cry.chunked(8, hash))]
def get_padding(message):
padding = b''
og_bit_length = len(message) * 8
message += bytes([128])
padding += bytes([128])
while len(message) % 64 != 56:
message += bytes([0])
padding += bytes([0])
message += struct.pack(">Q", og_bit_length)
padding += struct.pack(">Q", og_bit_length)
return padding
for key_length in range(21):
padding = get_padding(b'a' * key_length + original_message)
new_message = b"comment2=%20like%20a%20pound%20of%20bacon;admin=true;000"
hash = cry.sha1(new_message, *registers,
(len(original_message) + len(padding) +
len(new_message) + key_length) * 8)
if cry.validate_sha1_hash(
key, hash, original_message + padding + new_message):
break
self.assertEqual(
cry.validate_sha1_hash(
key, hash, original_message + padding + new_message), True)
def setUp(self):
self.crypto_kit = Crypto_Kit()
class Set2(unittest.TestCase):
def setUp(self):
self.crypto_kit = Crypto_Kit()
def test_set_2_problem_9(self):
c = self.crypto_kit
pad_result = c.pad_to_length(b"YELLOW SUBMARINE", 20)
self.assertEqual(pad_result, b"YELLOW SUBMARINE\x04\x04\x04\x04")
def test_set_2_problem_10(self):
with open('./tests/10.txt', 'r') as myfile:
encrypted_base_64 = myfile.read()
c = self.crypto_kit
encrypted_bytes = c.decode_base64(encrypted_base_64)
iv = bytes([0] * 16)
key = b"YELLOW SUBMARINE"
self.assertEqual(
c.cbc_decrypt(key, iv, encrypted_bytes),
b"""I'm back and I'm ringin' the bell \nA rockin' on the mike while the fly girls yell \nIn ecstasy in the back of me \nWell that's my DJ Deshay cuttin' all them Z's \nHittin' hard and the girlies goin' crazy \nVanilla's on the mike, man I'm not lazy. \n\nI'm lettin' my drug kick in \nIt controls my mouth and I begin \nTo just let it flow, let my concepts go \nMy posse's to the side yellin', Go Vanilla Go! \n\nSmooth 'cause that's the way I will be \nAnd if you don't give a damn, then \nWhy you starin' at me \nSo get off 'cause I control the stage \nThere's no dissin' allowed \nI'm in my own phase \nThe girlies sa y they love me and that is ok \nAnd I can dance better than any kid n' play \n\nStage 2 -- Yea the one ya' wanna listen to \nIt's off my head so let the beat play through \nSo I can funk it up and make it sound good \n1-2-3 Yo -- Knock on some wood \nFor good luck, I like my rhymes atrocious \nSupercalafragilisticexpialidocious \nI'm an effect and that you can bet \nI can take a fly girl and make her wet. \n\nI'm like Samson -- Samson to Delilah \nThere's no denyin', You can try to hang \nBut you'll keep tryin' to get my style \nOver and over, practice makes perfect \nBut not if you're a loafer. \n\nYou'll get nowhere, no place, no time, no girls \nSoon -- Oh my God, homebody, you probably eat \nSpaghetti with a spoon! Come on and say it! \n\nVIP. Vanilla Ice yep, yep, I'm comin' hard like a rhino \nIntoxicating so you stagger like a wino \nSo punks stop trying and girl stop cryin' \nVanilla Ice is sellin' and you people are buyin' \n'Cause why the freaks are jockin' like Crazy Glue \nMovin' and groovin' trying to sing along \nAll through the ghetto groovin' this here song \nNow you're amazed by the VIP posse. \n\nSteppin' so hard like a German Nazi \nStartled by the bases hittin' ground \nThere's no trippin' on mine, I'm just gettin' down \nSparkamatic, I'm hangin' tight like a fanatic \nYou trapped me once and I thought that \nYou might have it \nSo step down and lend me your ear \n'89 in my time! You, '90 is my year. \n\nYou're weakenin' fast, YO! and I can tell it \nYour body's gettin' hot, so, so I can smell it \nSo don't be mad and don't be sad \n'Cause the lyrics belong to ICE, You can call me Dad \nYou're pitchin' a fit, so step back and endure \nLet the witch doctor, Ice, do the dance to cure \nSo come up close and don't be square \nYou wanna battle me -- Anytime, anywhere \n\nYou thought that I was weak, Boy, you're dead wrong \nSo come on, everybody and sing this song \n\nSay -- Play that funky music Say, go white boy, go white boy go \nplay that funky music Go white boy, go white boy, go \nLay down and boogie and play that funky music till you die. \n\nPlay that funky music Come on, Come on, let me hear \nPlay that funky music white boy you say it, say it \nPlay that funky music A little louder now \nPlay that funky music, white boy Come on, Come on, Come on \nPlay that funky music \n\x04\x04\x04\x04"""
)
def test_set_2_problem_11(self):
with open('./tests/10.txt', 'r') as myfile:
encrypted_base_64 = myfile.read()
c = self.crypto_kit
encrypted_bytes = c.decode_base64(encrypted_base_64)
self.assertEqual(self.crypto_kit.detect_cipher_mode(encrypted_bytes),
'cbc')
def test_set_2_problem_12(self):
c = self.crypto_kit
unknown_string = """
Um9sbGluJyBpbiBteSA1LjAKV2l0aCBteSByYWctdG9wIGRvd24gc28gbXkg
aGFpciBjYW4gYmxvdwpUaGUgZ2lybGllcyBvbiBzdGFuZGJ5IHdhdmluZyBq
dXN0IHRvIHNheSBoaQpEaWQgeW91IHN0b3A/IE5vLCBJIGp1c3QgZHJvdmUg
YnkK
"""
unencrypted_string = c.decode_base64(unknown_string)
cipher_length = c.get_ecb_cipher_length(unencrypted_string)
mode = c.detect_cipher_mode(
c.ecb_encryption_oracle(bytes([0] * cipher_length * 2),
unencrypted_string))
known_bytes = bytearray(bytes([0] * 15))
chunks = list(c.chunked(cipher_length, unencrypted_string))
decrypted_bytes = b''
for chunk in range(len(chunks)):
block = chunk * cipher_length
for i in range(cipher_length):
dictionary = {}
offset = cipher_length - 1 - i
bytes_we_know = bytes(known_bytes[-(cipher_length - 1):])
for j in range(256):
guess = bytes(bytes_we_know + bytes([j]))
key = c.ecb_encryption_oracle(
bytes(guess), unencrypted_string)[0:cipher_length]
dictionary[key] = guess
saved_byte = c.ecb_encryption_oracle(
known_bytes[0:offset],
unencrypted_string)[block:block + cipher_length]
if saved_byte in dictionary:
correct_byte = dictionary[saved_byte][-1]
known_bytes.append(correct_byte)
decrypted_bytes += bytes([correct_byte])
self.assertEqual(
decrypted_bytes,
b"""Rollin' in my 5.0\nWith my rag-top down so my hair can blow\nThe girlies on standby waving just to say hi\nDid you stop? No, I just drove by\n\x01"""
)
def test_set_2_problem_13(self):
c = self.crypto_kit
def get_cipher_length():
profile = c.profile_for("a")
length = len(profile)
for i in range(100):
result = c.profile_for("a" * i)
if len(result) != length:
return len(result) - length
cipher_length = get_cipher_length()
need_bytes = b'email=&uid=10&role='
email_length = math.ceil(
len(need_bytes) / cipher_length) * cipher_length - len(need_bytes)
email = "a" * email_length
first_half_encrypted = c.profile_for(email)[0:len(need_bytes) +
email_length]
admin_padding = cipher_length - len(b'admin')
second_half_decrypted = (
b'admin' + bytes([admin_padding] * admin_padding)).decode('utf-8')
def isolate_block():
for i in range(100):
if c.dup_block_counts(c.profile_for("a" * i)) > 0:
return i
second_half_encrypted = c.profile_for(
"a" * (isolate_block() - 16) +
second_half_decrypted)[32:32 + cipher_length]
self.assertEqual(
c.decrypt_profile(first_half_encrypted + second_half_encrypted),
b'email=aaaaaaaaaaaaa&uid=10&role=admin\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b\x0b'
)
def test_set_2_problem_14(self):
c = self.crypto_kit
unknown_string = """
Um9sbGluJyBpbiBteSA1LjAKV2l0aCBteSByYWctdG9wIGRvd24gc28gbXkg
aGFpciBjYW4gYmxvdwpUaGUgZ2lybGllcyBvbiBzdGFuZGJ5IHdhdmluZyBq
dXN0IHRvIHNheSBoaQpEaWQgeW91IHN0b3A/IE5vLCBJIGp1c3QgZHJvdmUg
YnkK
"""
unencrypted_string = c.decode_base64(unknown_string)
def get_unknown_bytes():
for i in range(100):
result = c.ecb_encryption_oracle_random_prefix(
b'a' * i, unencrypted_string)
byte_set = set()
prefix_bytes = b''
split_bytes = list(c.chunked(16, result))
for byte in split_bytes:
if byte in byte_set:
return result, i, len(prefix_bytes)
prefix_bytes += byte
byte_set.add(byte)
cipher_length = 16
result, bytes_amount, ignore_bytes = get_unknown_bytes()
bytes_amount -= 16
known_bytes = bytearray(bytes([0] * 15))
chunks = list(c.chunked(cipher_length, result[ignore_bytes:]))
decrypted_bytes = b''
for chunk in range(len(chunks)):
block = chunk * cipher_length
for i in range(cipher_length):
dictionary = {}
offset = cipher_length - 1 - i
bytes_we_know = bytes(known_bytes[-(cipher_length - 1):])
for j in range(256):
guess = bytes(bytes_we_know + bytes([j]))
key = c.ecb_encryption_oracle_random_prefix(
b'a' * bytes_amount + bytes(guess),
unencrypted_string)[ignore_bytes:ignore_bytes +
cipher_length]
dictionary[key] = guess
saved_byte = c.ecb_encryption_oracle_random_prefix(
b'a' * bytes_amount + known_bytes[0:offset],
unencrypted_string)[ignore_bytes + block:ignore_bytes +
block + cipher_length]
if saved_byte in dictionary:
correct_byte = dictionary[saved_byte][-1]
known_bytes.append(correct_byte)
decrypted_bytes += bytes([correct_byte])
print(known_bytes)
self.assertEqual(
decrypted_bytes,
b"Rollin' in my 5.0\nWith my rag-top down so my hair can blow\nThe girlies on standby waving just to say hi\nDid you stop? No, I just drove by\n\x01"
)
def test_set_2_problem_15(self):
c = self.crypto_kit
self.assertEqual(c.validate_pkcs_7(b"ICE ICE BABY\x04\x04\x04\x04"),
b"ICE ICE BABY")
self.assertEqual(c.validate_pkcs_7(b"ICE ICE BABY\x05\x05\x05\x05"),
False)
self.assertEqual(c.validate_pkcs_7(b"ICE ICE BABY\x05\x05\x05\x05"),
False)
def test_set_2_problem_16(self):
c = self.crypto_kit
attack_string = 'AAA admin true'
result = c.pend_attack(attack_string)
byte_array = bytearray(result)
byte_array[19] = bytes([byte_array[19] ^ 59 ^ 32])[0]
byte_array[25] = bytes([byte_array[25] ^ 61 ^ 32])[0]
new_bytes = bytes(byte_array)
self.assertEqual(c.is_admin(new_bytes), True)
import sys
import array
import math
import datetime
import time
import struct
from crypto.crypto_tools import Crypto_Kit
from crypto.mt19937 import MT19937
from Crypto.Cipher import AES
from random import randint
cry = Crypto_Kit()
key = cry.generate_random_bytes(randint(5, 20))
original_message = b"comment1=cooking%20MCs;userdata=foo;comment2=%20like%20a%20pound%20of%20bacon"
hash = cry.sha1(key + original_message)
registers = [int(r, 16) for r in list(cry.chunked(8, hash))]
def get_padding(message):
padding = b''
og_bit_length = len(message) * 8
message += bytes([128])
padding += bytes([128])
while len(message) % 64 != 56:
message += bytes([0])
padding += bytes([0])
message += struct.pack(">Q", og_bit_length)
padding += struct.pack(">Q", og_bit_length)
return padding
class Set1(unittest.TestCase):
def setUp(self):
self.crypto_kit = Crypto_Kit()
def test_set_1_problem_1(self):
self.assertEqual(
self.crypto_kit.hex_to_base64(
"49276d206b696c6c696e6720796f757220627261696e206c696b65206120706f69736f6e6f7573206d757368726f6f6d"
),
"SSdtIGtpbGxpbmcgeW91ciBicmFpbiBsaWtlIGEgcG9pc29ub3VzIG11c2hyb29t")
def test_set_1_problem_2(self):
c = self.crypto_kit
self.assertEqual(
c.fixed_xor(c.decode_hex("1c0111001f010100061a024b53535009181c"),
c.decode_hex("686974207468652062756c6c277320657965")),
c.decode_hex("746865206b696420646f6e277420706c6179"))
def test_set_1_problem_3(self):
c = self.crypto_kit
encrypted_result = "1b37373331363f78151b7f2b783431333d78397828372d363c78373e783a393b3736"
self.assertEqual(
c.most_likely_stanza(c.decode_hex(encrypted_result))[1],
b"Cooking MC's like a pound of bacon")
def test_set_1_problem_4(self):
c = self.crypto_kit
with open('./tests/4.txt', 'r') as myfile:
encrypted_result = myfile.read().splitlines()
best_score = 0
best_stanza = ""
for line in encrypted_result:
score, stanza, _ = c.most_likely_stanza(c.decode_hex(line))
if score > best_score:
best_score = score
best_stanza = stanza
self.assertEqual(best_stanza, b'Now that the party is jumping\n')
def test_set_1_problem_5(self):
c = self.crypto_kit
result = c.repeating_xor(
bytes(
"Burning 'em, if you ain't quick and nimble\nI go crazy when I hear a cymbal",
'utf-8'), bytes("ICE", 'utf-8'))
self.assertEqual(
c.encode_hex(result),
b'0b3637272a2b2e63622c2e69692a23693a2a3c6324202d623d63343c2a26226324272765272a282b2f20430a652e2c652a3124333a653e2b2027630c692b20283165286326302e27282f'
)
def test_set_1_problem_6(self):
c = self.crypto_kit
with open('./tests/6.txt', 'r') as myfile:
encrypted_result = myfile.read()
encrypted_bytes = c.decode_base64(encrypted_result)
key_length = c.find_key_length(encrypted_bytes)
bytes_sections = list(c.chunked(key_length, encrypted_bytes))
transposed = [*zip(*bytes_sections[0:-1])]
keys = []
for section in transposed:
bytes_string = array.array('B', section).tobytes()
_, _, single_key = c.most_likely_stanza(bytes_string)
keys.append(single_key[0])
decrypt_key = bytes(keys)
decrypted_bytes = c.repeating_xor(encrypted_bytes, decrypt_key)
self.assertEqual(
decrypted_bytes,
b"""I'm back and I'm ringin' the bell \nA rockin' on the mike while the fly girls yell \nIn ecstasy in the back of me \nWell that's my DJ Deshay cuttin' all them Z's \nHittin' hard and the girlies goin' crazy \nVanilla's on the mike, man I'm not lazy. \n\nI'm lettin' my drug kick in \nIt controls my mouth and I begin \nTo just let it flow, let my concepts go \nMy posse's to the side yellin', Go Vanilla Go! \n\nSmooth 'cause that's the way I will be \nAnd if you don't give a damn, then \nWhy you starin' at me \nSo get off 'cause I control the stage \nThere's no dissin' allowed \nI'm in my own phase \nThe girlies sa y they love me and that is ok \nAnd I can dance better than any kid n' play \n\nStage 2 -- Yea the one ya' wanna listen to \nIt's off my head so let the beat play through \nSo I can funk it up and make it sound good \n1-2-3 Yo -- Knock on some wood \nFor good luck, I like my rhymes atrocious \nSupercalafragilisticexpialidocious \nI'm an effect and that you can bet \nI can take a fly girl and make her wet. \n\nI'm like Samson -- Samson to Delilah \nThere's no denyin', You can try to hang \nBut you'll keep tryin' to get my style \nOver and over, practice makes perfect \nBut not if you're a loafer. \n\nYou'll get nowhere, no place, no time, no girls \nSoon -- Oh my God, homebody, you probably eat \nSpaghetti with a spoon! Come on and say it! \n\nVIP. Vanilla Ice yep, yep, I'm comin' hard like a rhino \nIntoxicating so you stagger like a wino \nSo punks stop trying and girl stop cryin' \nVanilla Ice is sellin' and you people are buyin' \n'Cause why the freaks are jockin' like Crazy Glue \nMovin' and groovin' trying to sing along \nAll through the ghetto groovin' this here song \nNow you're amazed by the VIP posse. \n\nSteppin' so hard like a German Nazi \nStartled by the bases hittin' ground \nThere's no trippin' on mine, I'm just gettin' down \nSparkamatic, I'm hangin' tight like a fanatic \nYou trapped me once and I thought that \nYou might have it \nSo step down and lend me your ear \n'89 in my time! You, '90 is my year. \n\nYou're weakenin' fast, YO! and I can tell it \nYour body's gettin' hot, so, so I can smell it \nSo don't be mad and don't be sad \n'Cause the lyrics belong to ICE, You can call me Dad \nYou're pitchin' a fit, so step back and endure \nLet the witch doctor, Ice, do the dance to cure \nSo come up close and don't be square \nYou wanna battle me -- Anytime, anywhere \n\nYou thought that I was weak, Boy, you're dead wrong \nSo come on, everybody and sing this song \n\nSay -- Play that funky music Say, go white boy, go white boy go \nplay that funky music Go white boy, go white boy, go \nLay down and boogie and play that funky music till you die. \n\nPlay that funky music Come on, Come on, let me hear \nPlay that funky music white boy you say it, say it \nPlay that funky music A little louder now \nPlay that funky music, white boy Come on, Come on, Come on \nPlay that funky music \n"""
)
def test_set_1_problem_7(self):
c = self.crypto_kit
key = b"YELLOW SUBMARINE"
cipher = AES.new(key, AES.MODE_ECB)
with open('./tests/7.txt', 'r') as myfile:
base_64_file = myfile.read()
encrypted_result = c.decode_base64(base_64_file)
self.assertEqual(
cipher.decrypt(encrypted_result),
b"""I'm back and I'm ringin' the bell \nA rockin' on the mike while the fly girls yell \nIn ecstasy in the back of me \nWell that's my DJ Deshay cuttin' all them Z's \nHittin' hard and the girlies goin' crazy \nVanilla's on the mike, man I'm not lazy. \n\nI'm lettin' my drug kick in \nIt controls my mouth and I begin \nTo just let it flow, let my concepts go \nMy posse's to the side yellin', Go Vanilla Go! \n\nSmooth 'cause that's the way I will be \nAnd if you don't give a damn, then \nWhy you starin' at me \nSo get off 'cause I control the stage \nThere's no dissin' allowed \nI'm in my own phase \nThe girlies sa y they love me and that is ok \nAnd I can dance better than any kid n' play \n\nStage 2 -- Yea the one ya' wanna listen to \nIt's off my head so let the beat play through \nSo I can funk it up and make it sound good \n1-2-3 Yo -- Knock on some wood \nFor good luck, I like my rhymes atrocious \nSupercalafragilisticexpialidocious \nI'm an effect and that you can bet \nI can take a fly girl and make her wet. \n\nI'm like Samson -- Samson to Delilah \nThere's no denyin', You can try to hang \nBut you'll keep tryin' to get my style \nOver and over, practice makes perfect \nBut not if you're a loafer. \n\nYou'll get nowhere, no place, no time, no girls \nSoon -- Oh my God, homebody, you probably eat \nSpaghetti with a spoon! Come on and say it! \n\nVIP. Vanilla Ice yep, yep, I'm comin' hard like a rhino \nIntoxicating so you stagger like a wino \nSo punks stop trying and girl stop cryin' \nVanilla Ice is sellin' and you people are buyin' \n'Cause why the freaks are jockin' like Crazy Glue \nMovin' and groovin' trying to sing along \nAll through the ghetto groovin' this here song \nNow you're amazed by the VIP posse. \n\nSteppin' so hard like a German Nazi \nStartled by the bases hittin' ground \nThere's no trippin' on mine, I'm just gettin' down \nSparkamatic, I'm hangin' tight like a fanatic \nYou trapped me once and I thought that \nYou might have it \nSo step down and lend me your ear \n'89 in my time! You, '90 is my year. \n\nYou're weakenin' fast, YO! and I can tell it \nYour body's gettin' hot, so, so I can smell it \nSo don't be mad and don't be sad \n'Cause the lyrics belong to ICE, You can call me Dad \nYou're pitchin' a fit, so step back and endure \nLet the witch doctor, Ice, do the dance to cure \nSo come up close and don't be square \nYou wanna battle me -- Anytime, anywhere \n\nYou thought that I was weak, Boy, you're dead wrong \nSo come on, everybody and sing this song \n\nSay -- Play that funky music Say, go white boy, go white boy go \nplay that funky music Go white boy, go white boy, go \nLay down and boogie and play that funky music till you die. \n\nPlay that funky music Come on, Come on, let me hear \nPlay that funky music white boy you say it, say it \nPlay that funky music A little louder now \nPlay that funky music, white boy Come on, Come on, Come on \nPlay that funky music \n\x04\x04\x04\x04"""
)
def test_set_1_problem_8(self):
c = self.crypto_kit
encrypted_with_ebc = None
rows_with_dupes = 0
with open('./tests/8.txt', 'r') as myfile:
encrypted_result = myfile.read().splitlines()
for line in encrypted_result:
dupes = 0
byte_set = set()
encrypted_bytes = c.decode_hex(line)
split_bytes = list(c.chunked(16, encrypted_bytes))
for byte in split_bytes:
if byte in byte_set:
dupes += 1
byte_set.add(byte)
if dupes > 0:
encrypted_with_ebc = line
rows_with_dupes += 1
self.assertEqual(
encrypted_with_ebc,
"d880619740a8a19b7840a8a31c810a3d08649af70dc06f4fd5d2d69c744cd283e2dd052f6b641dbf9d11b0348542bb5708649af70dc06f4fd5d2d69c744cd2839475c9dfdbc1d46597949d9c7e82bf5a08649af70dc06f4fd5d2d69c744cd28397a93eab8d6aecd566489154789a6b0308649af70dc06f4fd5d2d69c744cd283d403180c98c8f6db1f2a3f9c4040deb0ab51b29933f2c123c58386b06fba186a"
)
self.assertEqual(rows_with_dupes, 1)