1 files changed, 34 insertions, 0 deletions
diff --git a/src/lib/tlslite/utils/Cryptlib_AES.py b/src/lib/tlslite/utils/Cryptlib_AES.py
new file mode 100755
index 000000000..9e101fc62
--- /dev/null
+++ b/src/lib/tlslite/utils/Cryptlib_AES.py
@@ -0,0 +1,34 @@
+"""Cryptlib AES implementation."""
+
+from cryptomath import *
+from AES import *
+
+if cryptlibpyLoaded:
+
+    def new(key, mode, IV):
+        return Cryptlib_AES(key, mode, IV)
+
+    class Cryptlib_AES(AES):
+
+        def __init__(self, key, mode, IV):
+            AES.__init__(self, key, mode, IV, "cryptlib")
+            self.context = cryptlib_py.cryptCreateContext(cryptlib_py.CRYPT_UNUSED, cryptlib_py.CRYPT_ALGO_AES)
+            cryptlib_py.cryptSetAttribute(self.context, cryptlib_py.CRYPT_CTXINFO_MODE, cryptlib_py.CRYPT_MODE_CBC)
+            cryptlib_py.cryptSetAttribute(self.context, cryptlib_py.CRYPT_CTXINFO_KEYSIZE, len(key))
+            cryptlib_py.cryptSetAttributeString(self.context, cryptlib_py.CRYPT_CTXINFO_KEY, key)
+            cryptlib_py.cryptSetAttributeString(self.context, cryptlib_py.CRYPT_CTXINFO_IV, IV)
+
+        def __del__(self):
+             cryptlib_py.cryptDestroyContext(self.context)
+
+        def encrypt(self, plaintext):
+            AES.encrypt(self, plaintext)
+            bytes = stringToBytes(plaintext)
+            cryptlib_py.cryptEncrypt(self.context, bytes)
+            return bytesToString(bytes)
+
+        def decrypt(self, ciphertext):
+            AES.decrypt(self, ciphertext)
+            bytes = stringToBytes(ciphertext)
+            cryptlib_py.cryptDecrypt(self.context, bytes)
+            return bytesToString(bytes)