blob: d3d38a97379bacf7eb8a94dfedbe24c7f9403ce0 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
#!/usr/bin/python
matrix_size = 4
round_constants = [0x00, 0x01, 0x02, 0x04, 0x08, 0x10, 0x20, 0x40, 0x80, 0x1B, 0x36]
def hex_to_matrix(hex_array):
matrix = []
for i in range(16):
if i % matrix_size == 0:
matrix.append([hex_array[i]])
else:
matrix[int(i / 4)].append(hex_array[i])
return matrix
def text_to_hex(text):
hex_array = []
for char in text:
hex_array.append(ord(char))
return hex_array
def shift_matrix(matrix):
for i in range(matrix_size):
matrix[i] = matrix[i][-i:] + matrix[i][:-i]
return matrix
def key_expansion(round_key):
round_keys = [round_key]
for i in range(1, 11):
# Move top byte to bottom
top = round_key[0][3]
for row in range(matrix_size - 1):
round_key[row][3] = round_key[row + 1][3]
# TODO: sBox
round_key[3][3] = top ^ round_constants[i]
# Modify first column by XORing it with the previous round key
for column in range(matrix_size - 1):
for row in range(matrix_size):
round_key[row][column] = round_key[row][column] ^ round_keys[i - 1][row][column]
round_keys.append(round_key)
return round_keys
test_key = text_to_hex("Thats my Kung Fu")
test_text = text_to_hex("Two One Nine Two")
print(hex_to_matrix(test_key))
print(key_expansion(hex_to_matrix(test_key)))
print(shift_matrix(hex_to_matrix(test_key)))
|
