AES-128加密解密方式(逐步更新)
@AES-128加密解密方式
慢慢钻研也在慢慢更新
前言说一下，本人小白起步，断断续续的学习Python，慢慢的垒代码，持续学习中。部分内容，可能是直接找到网上内容来使用的，请大家勿喷，感谢。
由于最近项目接入一些东西呢，从产品运营的角度写一个简单的代码，去验证一些内容，所以写了这个代码。下面直接进入主题：
引入的库
from Crypto.Cipher import AES
from binascii import b2a_hex,a2b_hex
from Crypto import Random
import base64
检测key的长度
def check_key(key):
# 检测key的长度是否为16,24或者32bytes的长度
try:
if isinstance(key, bytes):
assert len(key) in [16, 24, 32]
return key
elif isinstance(key, str):
assert len(key.encode()) in [16, 24, 32]
return key.encode()
else:
raise Exception(f'密钥必须为str或bytes,不能为{type(key)}')
except AssertionError:
print('输入的长度不正确')
CFB/GCM模式
class AesEncryption(object):
# CFB模式
# def __init__(self, key, mode=AES.MODE_CFB):
# self.key = check_key(key)
# # 密钥key长度必须为16,24或者32bytes的长度
# self.mode = mode
# self.iv = Random.new().read(AES.block_size)
# GCM模式
def __init__(self, key, mode=AES.MODE_GCM):
self.key = check_key(key)
# 密钥key长度必须为16,24或者32bytes的长度
self.mode = mode
self.iv = Random.new().read(AES.block_size)
完整代码
from Crypto.Cipher import AES
from binascii import b2a_hex,a2b_hex
from Crypto import Random
import base64
def check_key(key):
# 检测key的长度是否为16,24或者32bytes的长度
try:
if isinstance(key, bytes):
assert len(key) in [16, 24, 32]
return key
elif isinstance(key, str):
assert len(key.encode()) in [16, 24, 32]
return key.encode()
else:
raise Exception(f'密钥必须为str或bytes,不能为{type(key)}')
except AssertionError:
print('输入的长度不正确')
class AesEncryption(object):
# CFB模式
# def __init__(self, key, mode=AES.MODE_CFB):
# self.key = check_key(key)
# # 密钥key长度必须为16,24或者32bytes的长度
# self.mode = mode
# self.iv = Random.new().read(AES.block_size)
# GCM模式
def __init__(self, key, mode=AES.MODE_GCM):
self.key = check_key(key)
# 密钥key长度必须为16,24或者32bytes的长度
self.mode = mode
self.iv = Random.new().read(AES.block_size)
def check_data(self,data):
# '检测加密的数据类型'
if isinstance(data, str):
data = data.encode()
elif isinstance(data, bytes):
pass
else:
raise Exception(f'加密的数据必须为str或bytes,不能为{type(data)}')
return data
def encrypt(self, data):
# 加密函数
data = self.check_data(data)
cryptor = AES.new(self.key, self.mode,self.iv)
return b2a_hex(cryptor.encrypt(data)).decode()
def decrypt(self,data):
# 解密函数
data = self.check_data(data)
cryptor = AES.new(self.key, self.mode,self.iv)
return cryptor.decrypt(a2b_hex(data)).decode()
if __name__ == '__main__':
key = input('请输入key:')
data = '{"ai":"test-accountId","name":"用户姓名","idNum":"371321199012310912"}'
aes = AesEncryption(key)
# 调用加密函数
e = aes.encrypt(data)
# 调用解密函数
d = aes.decrypt(e)
# 打印加密
print('加密内容: ', e)
# 打印解密
print('解密内容: ', d)
# 返回base64位的加密
en_text = base64.b64encode(bytes(e, 'utf-8'))
# 打听base64加密
print('获得的base64加密: ', en_text)