AES加解密算法簡單代碼

 最近在修改AES算法的CUDA程式設計代碼，但是不知道是否修改成功，對該算法的網上很多快速驗證網址表示懷疑，是以找到了下面這段代碼，借以檢驗AES算法結果是否正确

# coding: utf8
import sys
from Crypto.Cipher import AES
from binascii import b2a_hex, a2b_hex


class prpcrypt():
    def __init__(self, key):
        self.key = key
        self.mode = AES.MODE_CFB

    # 加密函數，如果text不是16的倍數【加密文本text必須為16的倍數！】，那就補足為16的倍數
    def encrypt(self, text):
        cryptor = AES.new(self.key, self.mode, self.key)
        # 這裡密鑰key 長度必須為16（AES-128）、24（AES-192）、或32（AES-256）Bytes 長度.目前AES-128足夠用
        length = 16
        count = len(text)
        if (count % length != 0):
            add = length - (count % length)
        else:
            add = 0
        text = text + ('\0' * add)
        self.ciphertext = cryptor.encrypt(text)
        # 因為AES加密時候得到的字元串不一定是ascii字元集的，輸出到終端或者儲存時候可能存在問題
        # 是以這裡統一把加密後的字元串轉化為16進制字元串
        return b2a_hex(self.ciphertext)

    # 解密後，去掉補足的空格用strip() 去掉
    def decrypt(self, text):
        cryptor = AES.new(self.key, self.mode, self.key)
        plain_text = cryptor.decrypt(a2b_hex(text))
        return plain_text.rstrip('\0')


if __name__ == '__main__':
    pc = prpcrypt('ABCDEFGHIJKLMNOP')  # 初始化密鑰
    e = pc.encrypt("ABCDEFGHIJKLMNOP")
    # d = pc.decrypt(e)
    print(e)
    e = pc.encrypt("00000000000000000000000000")
    # d = pc.decrypt(e)
    print(e)