class Authenticated_Encryption(object): def __init__(self, keys): self.mac = CBC_MAC((keys[0], keys[1])) self.aes_cbc = AES_CBC(keys[2]) def encrypt_and_sign(self, message): cipher = self.aes_cbc.encrypt(message) tag = self.mac.sign(cipher) return cipher, tag def authenticate_and_decrypt(self, cipher, tag): authenticated = self.mac.verify(cipher, tag) message = "" if not authenticated else self.aes_cbc.decrypt(cipher) return authenticated, message
def test_AES_CBC(self): key = os.urandom(16).encode('hex') decryptor = AES_CBC(key) for m in self.msgs: self.assertEqual(decryptor.decrypt(decryptor.encrypt(m)), m)