python 实现AES加解密相关的知识,可以参考以下文章:
python实现AES加密解密
但该文章中,对于加密前数据的补全、及解密后去掉多余数据,由作者自己进行了封装。导致代码较为复杂。
实际可以使用库中pad和unpad来解决该问题,而使代码非常简洁。
代码如下:
import string
import base64
import random
from Crypto.Cipher import AES
from Crypto.Util.Padding import pad, unpad
def get_random_digits(size=16):
"""Generate random number by size"""
return "".join(random.choice(string.digits) for i in range(size))
def get_aes_key_iv():
"""Generate AES cryptor key and iv"""
return get_random_digits(16).encode(), get_random_digits(16).encode()
def aes_encrypt(plain_text, aes_key, aes_iv):
"""
AES encrypt
:param plain_text: bytes
:param aes_key: bytes
:param aes_iv: bytes
:return: bytes
"""
return AES.new(aes_key, AES.MODE_CBC, aes_iv).encrypt(pad(plain_text, AES.block_size))
def aes_decrypt(plain_text, aes_key, aes_iv):
"""
AES decrypt
:param plain_text: bytes
:param aes_key: bytes, aes_key
:param aes_iv: bytes, aes_iv
:return: bytes
"""
return unpad(AES.new(aes_key, AES.MODE_CBC, aes_iv).decrypt(plain_text), AES.block_size)
def test_aes_cryptor(data, my_key, my_iv):
rData = aes_encrypt(data, my_key, my_iv)
print("密文:", base64.b64encode(rData)) # 转换成base64
rData = aes_decrypt(rData, my_key, my_iv)
print("明文:", rData)
if __name__ == "__main__":
my_key1, my_iv1 = get_aes_key_iv()
data1 = "hello word"
test_aes_cryptor(data1.encode(), my_key1, my_iv1)
运行结果如下:
密文: b'mBtSmX/dSUa3nKTXvP1Prg=='
明文: b'hello word'