/gdata/tlslite/utils/PyCrypto_AES.py

 1"""PyCrypto AES implementation."""
 2
 3from cryptomath import *
 4from AES import *
 5
 6if pycryptoLoaded:
 7    import Crypto.Cipher.AES
 8
 9    def new(key, mode, IV):
10        return PyCrypto_AES(key, mode, IV)
11
12    class PyCrypto_AES(AES):
13
14        def __init__(self, key, mode, IV):
15            AES.__init__(self, key, mode, IV, "pycrypto")
16            self.context = Crypto.Cipher.AES.new(key, mode, IV)
17
18        def encrypt(self, plaintext):
19            return self.context.encrypt(plaintext)
20
21        def decrypt(self, ciphertext):
22            return self.context.decrypt(ciphertext)