def test_key_expansion(self, runtime): aes = AES(runtime, 256, quiet=True) key = [] ascii_key = [] for i in xrange(8): key.append([]) for j in xrange(4): b = 15 * i + j key[i].append(Share(runtime, GF256, GF256(b))) ascii_key.append(chr(b)) result = aes.key_expansion(key) r = rijndael(ascii_key) expected_result = [] for round_key in r.Ke: for word in round_key: split_word = [] expected_result.append(split_word) for j in xrange(4): split_word.insert(0, word % 256) word /= 256 self.verify(runtime, result, expected_result)
def test_encrypt(self, runtime): cleartext = "Encrypt this!!!!" key = "Supposed to be secret!?!" aes = AES(runtime, 192, quiet=True) r = rijndael(key) result = aes.encrypt(cleartext, key) expected = [ord(c) for c in r.encrypt(cleartext)] return self.verify(runtime, [result], [expected])