~pablo-elenya/exaile/diacritica

"""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)