'''
Created on 2019年10月15日
@author: lg
'''
from Crypto.Cipher import AES
from binascii import b2a_hex, a2b_hex
# 强制输入明文增加到16字节的倍数,在末尾补'\0'
def add_to_16(text):
text = str(text, 'ISO-8859-1')
if len(text.encode('ISO-8859-1')) % 16:
add = 16 - (len(text.encode('ISO-8859-1')) % 16)
else:
add = 0
text = text + ('\0' * add)
return text.encode('ISO-8859-1')
# 加密函数,传入明文和秘钥,返回密文,这三个都是bytes格式的
def encrypt(text, key):
mode = AES.MODE_ECB
text = add_to_16(text)
cryptos = AES.new(key, mode)
cipher_text = cryptos.encrypt(text)
return cipher_text
# 解密函数,传入密文和秘钥,返回明文,这三个都是bytes格式的
def decrypt(text, key):
mode = AES.MODE_ECB
cryptor = AES.new(key, mode)
plain_text = cryptor.decrypt(text)
return plain_text
if __name__ == '__main__':
# 需要加密的明文。下面打印的返回值第一个为实际的bytes明文数据,第二个其实实际将bytes转成了16进制字符串以便于查看分析,但实际已经不是原来的数了,勿用b2a_hex的第二个数据给小程序,小程序接收的应该是第一个!
# 初始值
encData = []
encSum = 0x00
# 流水号
encData.append(0x00)
encSum += 0x00
encData.append(0x01)
encSum += 0x01
# 开锁
encData.append(0x4F)
encSum += 0x4F
# 关锁
# encData.append(0x43)
# encSum += 0x43
# 鸣笛
# encData.append(0x55)
# encSum += 0x55
# 数据段
for i in range(11):
encData.append(0xEE)
encSum += 0xEE
# 校验和
encData.append(int(encSum / 256))
encData.append(encSum % 256)
print(encData)
# 明文list转为bytes类型
encBytes = bytes(encData)
# encBytes = b'\x00\x01\x4F\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x11\x01\x0B'
print("明文为:", encBytes)
print("明文为:", b2a_hex(encBytes))
# 秘钥,字符串格式的转为bytes格式,实际需要传入的应是bytes格式
key = 'I637S01KYHH00000'.encode('ISO-8859-1')
# 将明文的bytes类型转为str类型传入,加密。返回值同上,第一个为实际的bytes格式密文 str(encBytes,'utf-8')
enc = encrypt(encBytes, key)
print("加密后:", enc)
print("加密后:", b2a_hex(enc))
# 解密,将bytes类型密文传入,返回值即为最后的bytes类型明文,同上
dec = decrypt(enc, key)
print("解密后:", dec)
print("解密后:", b2a_hex(dec))