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)