1 files changed, 22 insertions, 0 deletions
diff --git a/src/lib/tlslite/utils/PyCrypto_AES.py b/src/lib/tlslite/utils/PyCrypto_AES.py
new file mode 100755
index 000000000..e38b19d6f
--- /dev/null
+++ b/src/lib/tlslite/utils/PyCrypto_AES.py
@@ -0,0 +1,22 @@
+"""PyCrypto AES implementation."""
+
+from cryptomath import *
+from AES import *
+
+if pycryptoLoaded:
+    import Crypto.Cipher.AES
+
+    def new(key, mode, IV):
+        return PyCrypto_AES(key, mode, IV)
+
+    class PyCrypto_AES(AES):
+
+        def __init__(self, key, mode, IV):
+            AES.__init__(self, key, mode, IV, "pycrypto")
+            self.context = Crypto.Cipher.AES.new(key, mode, IV)
+
+        def encrypt(self, plaintext):
+            return self.context.encrypt(plaintext)
+
+        def decrypt(self, ciphertext):
+            return self.context.decrypt(ciphertext)
+\ No newline at end of file