ctp-aes.py

import os
import binascii
from cryptography.hazmat.primitives import padding
from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
from cryptography.hazmat.backends import default_backend


class AESCrypto(object):

    # AES_CBC_KEY = os.urandom(32)
    # AES_CBC_IV = os.urandom(16)
    AES_CBC_KEY = binascii.a2b_hex('f8c7fd111dedb4c9cdfb99edf35e3db9')
    AES_CBC_IV = binascii.a2b_hex('0a379ab1ee9cb40a40398a6d36b7a002')
    AES_CBC_KEY = b'f8c7fd111dedb4c9cdfb99edf35e3db9'
    AES_CBC_IV = b'0a379ab1ee9cb40a40398a6d36b7a002'
    algorithms.AES.block_size = 256

    @classmethod
    def encrypt(cls, data, mode='cbc'):
        func_name = '{}_encrypt'.format(mode)
        func = getattr(cls, func_name)
        return func(data)

    @classmethod
    def decrypt(cls, data, mode='cbc'):
        func_name = '{}_decrypt'.format(mode)
        func = getattr(cls, func_name)
        return func(data)

    @staticmethod
    def pkcs7_padding(data):
        if not isinstance(data, bytes):
            data = data.encode()

        padder = padding.PKCS7(algorithms.AES.block_size).padder()

        padded_data = padder.update(data) + padder.finalize()

        return padded_data

    @classmethod
    def cbc_encrypt(cls, data):
        if not isinstance(data, bytes):
            data = data.encode()

        cipher = Cipher(algorithms.AES(cls.AES_CBC_KEY),
                        modes.CBC(cls.AES_CBC_IV),
                        backend=default_backend())
        encryptor = cipher.encryptor()

        padded_data = encryptor.update(cls.pkcs7_padding(data))

        return padded_data

    @classmethod
    def cbc_decrypt(cls, data):
        if not isinstance(data, bytes):
            data = data.encode()

        cipher = Cipher(algorithms.AES(cls.AES_CBC_KEY),
                        modes.CBC(cls.AES_CBC_IV),
                        backend=default_backend())
        decryptor = cipher.decryptor()

        uppaded_data = cls.pkcs7_unpadding(decryptor.update(data))

        uppaded_data = uppaded_data.decode()
        return uppaded_data

    @staticmethod
    def pkcs7_unpadding(padded_data):
        unpadder = padding.PKCS7(algorithms.AES.block_size).unpadder()
        data = unpadder.update(padded_data)

        try:
            uppadded_data = data + unpadder.finalize()
        except ValueError:
            raise Exception('无效的加密信息!')
        else:
            return uppadded_data


# token = 'qU136eVGd4j6wTjEd402WiT2vtRUgwZId+rkQ7J6xiaDgEW667EXFU1jlYtl+8IjjIaTfIJkZIeeMsM4+qJm2jq0d2S1vFLC5sT3Ts1N0GeNwHinmJZNO7j0kQGRbaFoy/UmBD5u5O4Dg6zn74Lt+MhrPjKjnqfRm8rSiSM/dlriGotge8iU9Etrck4SNnZD'
token = 'sdad#a651212#sdasda#tytyty#llomm'
print(AESCrypto.cbc_encrypt(token).decode())