tests.py

from aes.transformations import *
from aes.key_expansion import *

TESTKEY = bytearray(range(16))


def test_sboxes():
	"""
		>>> test_sboxes()
		63 7c 77 7b f2 6b 6f c5 30 01 67 2b fe d7 ab 76
		ca 82 c9 7d fa 59 47 f0 ad d4 a2 af 9c a4 72 c0
		b7 fd 93 26 36 3f f7 cc 34 a5 e5 f1 71 d8 31 15
		04 c7 23 c3 18 96 05 9a 07 12 80 e2 eb 27 b2 75
		09 83 2c 1a 1b 6e 5a a0 52 3b d6 b3 29 e3 2f 84
		53 d1 00 ed 20 fc b1 5b 6a cb be 39 4a 4c 58 cf
		d0 ef aa fb 43 4d 33 85 45 f9 02 7f 50 3c 9f a8
		51 a3 40 8f 92 9d 38 f5 bc b6 da 21 10 ff f3 d2
		cd 0c 13 ec 5f 97 44 17 c4 a7 7e 3d 64 5d 19 73
		60 81 4f dc 22 2a 90 88 46 ee b8 14 de 5e 0b db
		e0 32 3a 0a 49 06 24 5c c2 d3 ac 62 91 95 e4 79
		e7 c8 37 6d 8d d5 4e a9 6c 56 f4 ea 65 7a ae 08
		ba 78 25 2e 1c a6 b4 c6 e8 dd 74 1f 4b bd 8b 8a
		70 3e b5 66 48 03 f6 0e 61 35 57 b9 86 c1 1d 9e
		e1 f8 98 11 69 d9 8e 94 9b 1e 87 e9 ce 55 28 df
		8c a1 89 0d bf e6 42 68 41 99 2d 0f b0 54 bb 16
		<BLANKLINE>
		52 09 6a d5 30 36 a5 38 bf 40 a3 9e 81 f3 d7 fb
		7c e3 39 82 9b 2f ff 87 34 8e 43 44 c4 de e9 cb
		54 7b 94 32 a6 c2 23 3d ee 4c 95 0b 42 fa c3 4e
		08 2e a1 66 28 d9 24 b2 76 5b a2 49 6d 8b d1 25
		72 f8 f6 64 86 68 98 16 d4 a4 5c cc 5d 65 b6 92
		6c 70 48 50 fd ed b9 da 5e 15 46 57 a7 8d 9d 84
		90 d8 ab 00 8c bc d3 0a f7 e4 58 05 b8 b3 45 06
		d0 2c 1e 8f ca 3f 0f 02 c1 af bd 03 01 13 8a 6b
		3a 91 11 41 4f 67 dc ea 97 f2 cf ce f0 b4 e6 73
		96 ac 74 22 e7 ad 35 85 e2 f9 37 e8 1c 75 df 6e
		47 f1 1a 71 1d 29 c5 89 6f b7 62 0e aa 18 be 1b
		fc 56 3e 4b c6 d2 79 20 9a db c0 fe 78 cd 5a f4
		1f dd a8 33 88 07 c7 31 b1 12 10 59 27 80 ec 5f
		60 51 7f a9 19 b5 4a 0d 2d e5 7a 9f 93 c9 9c ef
		a0 e0 3b 4d ae 2a f5 b0 c8 eb bb 3c 83 53 99 61
		17 2b 04 7e ba 77 d6 26 e1 69 14 63 55 21 0c 7d
		
	"""

	for a in range(16):
		row = ""
		for b in range(16):
			x = a*16+b
			row += "{:02x} ".format(sub_byte(x))
		print(row.strip())

	print()
	for a in range(16):
		row = ""
		for b in range(16):
			x = a*16+b
			row += "{:02x} ".format(inv_sub_byte(x))
		print(row.strip())


def test_key_expansion():
	"""
		>>> test_key_expansion()
		00: 000102030405060708090a0b0c0d0e0f
		01: d6aa74fdd2af72fadaa678f1d6ab76fe
		02: b692cf0b643dbdf1be9bc5006830b3fe
		03: b6ff744ed2c2c9bf6c590cbf0469bf41
		04: 47f7f7bc95353e03f96c32bcfd058dfd
		05: 3caaa3e8a99f9deb50f3af57adf622aa
		06: 5e390f7df7a69296a7553dc10aa31f6b
		07: 14f9701ae35fe28c440adf4d4ea9c026
		08: 47438735a41c65b9e016baf4aebf7ad2
		09: 549932d1f08557681093ed9cbe2c974e
		10: 13111d7fe3944a17f307a78b4d2b30c5
		
	"""

	ekey = key_expansion(TESTKEY)
	for i in range(len(ekey)):
		print("{:02d}: {}".format(i, ekey[i].hex()))

if __name__ == "__main__":
	import doctest
	import aes.transformations as xforms
	import aes.finite_field as ff
	import tests

	doctest.testmod(xforms)
	doctest.testmod(ff)
	doctest.testmod(tests)  # yup, we can test ourself...