sangge/cryptopal
1
0
Fork 0
Code

chore: sync some code

Browse Source
This commit is contained in:
Smart-SangGe 2023-12-23 05:08:21 +08:00
parent 31c669b667
commit b892176e2d
3 changed files with 267 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

165
set1/challenge7.py Normal file
View File

@ -0,0 +1,165 @@
s_box = [
0x63, 0x7c, 0x77, 0x7b, 0xf2, 0x6b, 0x6f, 0xc5, 0x30, 0x01, 0x67, 0x2b, 0xfe, 0xd7, 0xab, 0x76,
0xca, 0x82, 0xc9, 0x7d, 0xfa, 0x59, 0x47, 0xf0, 0xad, 0xd4, 0xa2, 0xaf, 0x9c, 0xa4, 0x72, 0xc0,
0xb7, 0xfd, 0x93, 0x26, 0x36, 0x3f, 0xf7, 0xcc, 0x34, 0xa5, 0xe5, 0xf1, 0x71, 0xd8, 0x31, 0x15,
0x04, 0xc7, 0x23, 0xc3, 0x18, 0x96, 0x05, 0x9a, 0x07, 0x12, 0x80, 0xe2, 0xeb, 0x27, 0xb2, 0x75,
0x09, 0x83, 0x2c, 0x1a, 0x1b, 0x6e, 0x5a, 0xa0, 0x52, 0x3b, 0xd6, 0xb3, 0x29, 0xe3, 0x2f, 0x84,
0x53, 0xd1, 0x00, 0xed, 0x20, 0xfc, 0xb1, 0x5b, 0x6a, 0xcb, 0xbe, 0x39, 0x4a, 0x4c, 0x58, 0xcf,
0xd0, 0xef, 0xaa, 0xfb, 0x43, 0x4d, 0x33, 0x85, 0x45, 0xf9, 0x02, 0x7f, 0x50, 0x3c, 0x9f, 0xa8,
0x51, 0xa3, 0x40, 0x8f, 0x92, 0x9d, 0x38, 0xf5, 0xbc, 0xb6, 0xda, 0x21, 0x10, 0xff, 0xf3, 0xd2,
0xcd, 0x0c, 0x13, 0xec, 0x5f, 0x97, 0x44, 0x17, 0xc4, 0xa7, 0x7e, 0x3d, 0x64, 0x5d, 0x19, 0x73,
0x60, 0x81, 0x4f, 0xdc, 0x22, 0x2a, 0x90, 0x88, 0x46, 0xee, 0xb8, 0x14, 0xde, 0x5e, 0x0b, 0xdb,
0xe0, 0x32, 0x3a, 0x0a, 0x49, 0x06, 0x24, 0x5c, 0xc2, 0xd3, 0xac, 0x62, 0x91, 0x95, 0xe4, 0x79,
0xe7, 0xc8, 0x37, 0x6d, 0x8d, 0xd5, 0x4e, 0xa9, 0x6c, 0x56, 0xf4, 0xea, 0x65, 0x7a, 0xae, 0x08,
0xba, 0x78, 0x25, 0x2e, 0x1c, 0xa6, 0xb4, 0xc6, 0xe8, 0xdd, 0x74, 0x1f, 0x4b, 0xbd, 0x8b, 0x8a,
0x70, 0x3e, 0xb5, 0x66, 0x48, 0x03, 0xf6, 0x0e, 0x61, 0x35, 0x57, 0xb9, 0x86, 0xc1, 0x1d, 0x9e,
0xe1, 0xf8, 0x98, 0x11, 0x69, 0xd9, 0x8e, 0x94, 0x9b, 0x1e, 0x87, 0xe9, 0xce, 0x55, 0x28, 0xdf,
0x8c, 0xa1, 0x89, 0x0d, 0xbf, 0xe6, 0x42, 0x68, 0x41, 0x99, 0x2d, 0x0f, 0xb0, 0x54, 0xbb, 0x16
]
inv_s_box = [
0x52, 0x09, 0x6A, 0xD5, 0x30, 0x36, 0xA5, 0x38, 0xBF, 0x40, 0xA3, 0x9E, 0x81, 0xF3, 0xD7, 0xFB,
0x7C, 0xE3, 0x39, 0x82, 0x9B, 0x2F, 0xFF, 0x87, 0x34, 0x8E, 0x43, 0x44, 0xC4, 0xDE, 0xE9, 0xCB,
0x54, 0x7B, 0x94, 0x32, 0xA6, 0xC2, 0x23, 0x3D, 0xEE, 0x4C, 0x95, 0x0B, 0x42, 0xFA, 0xC3, 0x4E,
0x08, 0x2E, 0xA1, 0x66, 0x28, 0xD9, 0x24, 0xB2, 0x76, 0x5B, 0xA2, 0x49, 0x6D, 0x8B, 0xD1, 0x25,
0x72, 0xF8, 0xF6, 0x64, 0x86, 0x68, 0x98, 0x16, 0xD4, 0xA4, 0x5C, 0xCC, 0x5D, 0x65, 0xB6, 0x92,
0x6C, 0x70, 0x48, 0x50, 0xFD, 0xED, 0xB9, 0xDA, 0x5E, 0x15, 0x46, 0x57, 0xA7, 0x8D, 0x9D, 0x84,
0x90, 0xD8, 0xAB, 0x00, 0x8C, 0xBC, 0xD3, 0x0A, 0xF7, 0xE4, 0x58, 0x05, 0xB8, 0xB3, 0x45, 0x06,
0xD0, 0x2C, 0x1E, 0x8F, 0xCA, 0x3F, 0x0F, 0x02, 0xC1, 0xAF, 0xBD, 0x03, 0x01, 0x13, 0x8A, 0x6B,
0x3A, 0x91, 0x11, 0x41, 0x4F, 0x67, 0xDC, 0xEA, 0x97, 0xF2, 0xCF, 0xCE, 0xF0, 0xB4, 0xE6, 0x73,
0x96, 0xAC, 0x74, 0x22, 0xE7, 0xAD, 0x35, 0x85, 0xE2, 0xF9, 0x37, 0xE8, 0x1C, 0x75, 0xDF, 0x6E,
0x47, 0xF1, 0x1A, 0x71, 0x1D, 0x29, 0xC5, 0x89, 0x6F, 0xB7, 0x62, 0x0E, 0xAA, 0x18, 0xBE, 0x1B,
0xFC, 0x56, 0x3E, 0x4B, 0xC6, 0xD2, 0x79, 0x20, 0x9A, 0xDB, 0xC0, 0xFE, 0x78, 0xCD, 0x5A, 0xF4,
0x1F, 0xDD, 0xA8, 0x33, 0x88, 0x07, 0xC7, 0x31, 0xB1, 0x12, 0x10, 0x59, 0x27, 0x80, 0xEC, 0x5F,
0x60, 0x51, 0x7F, 0xA9, 0x19, 0xB5, 0x4A, 0x0D, 0x2D, 0xE5, 0x7A, 0x9F, 0x93, 0xC9, 0x9C, 0xEF,
0xA0, 0xE0, 0x3B, 0x4D, 0xAE, 0x2A, 0xF5, 0xB0, 0xC8, 0xEB, 0xBB, 0x3C, 0x83, 0x53, 0x99, 0x61,
0x17, 0x2B, 0x04, 0x7E, 0xBA, 0x77, 0xD6, 0x26, 0xE1, 0x69, 0x14, 0x63, 0x55, 0x21, 0x0C, 0x7D
]
rcons = [
0x01, 0x02, 0x04, 0x08, 0x10,
0x20, 0x40, 0x80, 0x1B, 0x36,
0x6C, 0xD8, 0xAB, 0x4D, 0x9A,
0x2F, 0x5E, 0xBC, 0x63, 0xC6,
0x97, 0x35, 0x6A, 0xD4, 0xB3,
0x7D, 0xFA, 0xEF, 0xC5, 0x91
]
def EBC_encrypt(plaintext: bytes, key: bytes) -> bytes:
# check padding
if len(plaintext) % 16 != 0:
raise ValueError("Invalid plaintext lenth. Try to pad.")
blocks = []
for i in range(0, len(plaintext), 16):
block = plaintext[i:i + 16]
blocks.append(block)
key = KeyExpend(key)
ciphertext = bytes()
for i in blocks:
i = AddRoundKey(i)
for j in range(10):
i = EncSubBytes(i)
i = EncShiftRows(i)
i = EncMixColumns(i)
i = AddRoundKey(i,roundkey)
ciphertext += i
return ciphertext
def EBC_decrypt(ciphertext: bytes, key: bytes) -> bytes:
plaintext = 1
return plaintext
def AddRoundKey(input: bytes, roundkey: bytes) -> bytes:
output = bytes()
return output
def EncSubBytes(input: bytes) -> bytes:
"""
AES enc SubBytes
Args:
input (bytes): input state
Returns:
bytes: Bytes after subsitute
"""
index = int.from_bytes(input)
output = s_box[index].to_bytes()
return output
def DecSubBytes(input: bytes) -> bytes:
"""
AES dec SubBytes
Args:
input (bytes): input bytes
Returns:
bytes: Bytes after subsitute
"""
index = int.from_bytes(input)
output = inv_s_box[index].to_bytes()
return output
def EncShiftRows(input: bytes) -> bytes:
output = bytes()
return output
def EncMixColumns(input: bytes) -> bytes:
output = bytes()
return output
def DecShiftRows(input: bytes) -> bytes:
output = bytes()
return output
def DecMixColumns(input: bytes) -> bytes:
output = bytes()
return output
def KeyExpend(key: bytes) -> bytes:
words = [bytes()] * 44
for i in range(44):
if i <= 3:
for j in range(4):
words[i] += key[i + 4*j].to_bytes()
continue
else:
if i % 4 == 0:
for j in range(4):
words[i] += XorBytes(words[i-4],words[i-4])
Expended_key = words
return Expended_key
def XorBytes(bytes1: bytes, bytes2: bytes) -> bytes:
if len(bytes1) != 1 or len(bytes2) != 1:
raise ValueError("Byte length must be 1")
# Convert bytes to int, XOR the ints, then convert back to bytes
int1 = int.from_bytes(bytes1, byteorder='big')
int2 = int.from_bytes(bytes2, byteorder='big')
xor_result_int = int1 ^ int2
xor_result_bytes = xor_result_int.to_bytes(max(len(bytes1), len(bytes2)), byteorder='big')
return xor_result_bytes
def GFunc(word: bytes, rcon: bytes) -> bytes:
high_byte = word[0:1]
low_bytes = word[1:]
shift_bytes = low_bytes+high_byte
rcon_list = [rcon, b'\x00', b'\x00', b'\x00']
output = bytes()
for i in range(4):
output += XorBytes(EncSubBytes(shift_bytes[i].to_bytes()), rcon_list[i])
return output

26
set2/challenge10.py Normal file
View File

@ -0,0 +1,26 @@
import myaes
def encrypt_ecb(plaintext: bytes, key: bytes) -> bytes:
cipher = myaes.new(key, myaes.MODE_ECB)
padded_plaintext = cipher.pad(plaintext, myaes.block_size)
ciphertext = cipher.encrypt(padded_plaintext)
return ciphertext
def decrypt_ecb(ciphertext: bytes, key: bytes) -> bytes:
cipher = myaes.new(key, myaes.MODE_ECB)
padded_plaintext = cipher.decrypt(ciphertext)
plaintext = cipher.unpad(padded_plaintext, myaes.block_size)
return plaintext
# Example Usage:
key = b'YELLOW SUBMARINE'
plaintext = b'Hello, World!'
# Encrypt the plaintext
ciphertext = encrypt_ecb(plaintext, key)
print(ciphertext)
# Decrypt the ciphertext
decrypted_plaintext = decrypt_ecb(ciphertext, key)
print(decrypted_plaintext)

76
set2/myaes.py Normal file
View File

@ -0,0 +1,76 @@
@staticmethod
def new(key, mode, *args, **kwargs):
return AESCipher(key, mode, *args, **kwargs)
MODE_ECB = 1 #: Electronic Code Book (:ref:`ecb_mode`)
MODE_CBC = 2 #: Cipher-Block Chaining (:ref:`cbc_mode`)
MODE_CFB = 3 #: Cipher Feedback (:ref:`cfb_mode`)
MODE_OFB = 5 #: Output Feedback (:ref:`ofb_mode`)
MODE_CTR = 6 #: Counter mode (:ref:`ctr_mode`)
MODE_OPENPGP = 7 #: OpenPGP mode (:ref:`openpgp_mode`)
MODE_CCM = 8 #: Counter with CBC-MAC (:ref:`ccm_mode`)
MODE_EAX = 9 #: :ref:`eax_mode`
MODE_SIV = 10 #: Synthetic Initialization Vector (:ref:`siv_mode`)
MODE_GCM = 11 #: Galois Counter Mode (:ref:`gcm_mode`)
MODE_OCB = 12 #: Offset Code Book (:ref:`ocb_mode`)
# Size of a data block (in bytes)
block_size = 16
class AESCipher:
def __init__(self, key, mode, *args, **kwargs):
self.key = key
self.mode = mode
self.key_size = (16, 24, 32)
self.iv = kwargs.get('iv', None)
self.nonce = kwargs.get('nonce', None)
def pad(self, data: bytes, block_size: int = block_size) -> bytes:
"""Apply PKCS#7 padding to the given data."""
padding_size = block_size - (len(data) % block_size)
padding = bytes([padding_size] * padding_size)
return data + padding
def unpad(self, data: bytes, block_size: int = block_size) -> bytes:
"""Remove PKCS#7 padding from the given data."""
padding_size = data[-1]
if padding_size > block_size or padding_size == 0:
raise ValueError('Invalid padding')
padding = bytes([padding_size] * padding_size)
if data[-padding_size:] != padding:
raise ValueError('Invalid padding')
return data[:-padding_size]
def encrypt(self, plaintext: bytes) -> bytes:
if self.mode == 1:
return self.EBC_encrypt(plaintext)
elif self.mode == 2:
return self.CBC_encrypt(plaintext, self.iv)
else:
raise ValueError("Incorrect mode")
def decrypt(self, ciphertext: bytes) -> bytes:
if self.mode == 1:
return self.EBC_decrypt(ciphertext)
elif self.mode == 2:
return self.CBC_decrypt(ciphertext, self.iv)
else:
raise ValueError("Incorrect mode")
def EBC_encrypt(self, plaintext: bytes) -> bytes:
ciphertext = 1
return ciphertext
def EBC_decrypt(self, ciphertext: bytes) -> bytes:
plaintext = 1
return plaintext
def CBC_encrypt(self, plaintext: bytes, iv: bytes) -> bytes:
ciphertext = 1
return ciphertext
def CBC_decrypt(self, ciphertext: bytes, iv: bytes) -> bytes:
plaintext = 1
return plaintext