BooleanAES/libshiftrows.py at master · archoad/BooleanAES · GitHub

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


from libmain import *


def shiftRows():
	Nb = 4
	result = []
	SR = []
	equa = []
	state = [[(i*Nb)+j for i in range(Nb)] for j in range(Nb)]
	tmp = [[0 for i in range(Nb)] for i in range(Nb)]
	for i in range(Nb):
		for j in range(Nb):
			tmp[i][j] = state[i][(j + i) % Nb]
	for row in range(Nb):
		for byte in range(Nb):
			result.append(tmp[byte][row])
	for numByte in result:
		for i in range(octetSize):
			SR.append(generateBitsBlock('x_' + str((numByte*octetSize)+i)))
	for i in range(blockSize):
		for bit in range(blockSize):
			if SR[i][bit] == '1':
				equa.append('x_%s' % (bit))
	return equa


def invShiftRows():
	Nb = 4
	result = []
	SR = []
	equa = []
	state = [[(i*Nb)+j for i in range(Nb)] for j in range(Nb)]
	tmp = [[0 for i in range(Nb)] for i in range(Nb)]
	for i in range(Nb):
		for j in range(Nb):
			tmp[i][(j + i) % Nb] = state[i][j]
	for row in range(Nb):
		for byte in range(Nb):
			result.append(tmp[byte][row])
	for numByte in result:
		for i in range(octetSize):
			SR.append(generateBitsBlock('x_' + str((numByte*octetSize)+i)))
	for i in range(blockSize):
		for bit in range(blockSize):
			if SR[i][bit] == '1':
				equa.append('x_%s' % (bit))
	return equa