File size: 3,141 Bytes
89cf463 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 |
# -- coding: utf-8 --
# @Time : 2021/11/26
# @Author : ykk648
# @Project : https://github.com/ykk648/AI_power
# pip install pycryptodome
from Crypto.Cipher import AES
import operator
import time
AES_BLOCK_SIZE = AES.block_size # AES 加密数据块大小, 只能是16
AES_KEY_SIZE = 16 # AES 密钥长度（单位字节），可选 16、24、32，对应 128、192、256 位密钥
# 待加密文本补齐到 block size 的整数倍
def PadTest(bytes):
while len(bytes) % AES_BLOCK_SIZE != 0: # 循环直到补齐 AES_BLOCK_SIZE 的倍数
bytes += ' '.encode() # 通过补空格（不影响源文件的可读）来补齐
return bytes # 返回补齐后的字节列表
# 待加密的密钥补齐到对应的位数
def PadKey(key):
if len(key) > AES_KEY_SIZE: # 如果密钥长度超过 AES_KEY_SIZE
return key[:AES_KEY_SIZE] # 截取前面部分作为密钥并返回
while len(key) % AES_KEY_SIZE != 0: # 不到 AES_KEY_SIZE 长度则补齐
key += ' '.encode() # 补齐的字符可用任意字符代替
return key # 返回补齐后的密钥
# AES 加密
def EnCrypt(key, bytes):
myCipher = AES.new(key, AES.MODE_ECB) # 新建一个 AES 算法实例，使用 ECB（电子密码本）模式
encryptData = myCipher.encrypt(bytes) # 调用加密方法，得到加密后的数据
return encryptData # 返回加密数据
# AES 解密
def DeCrypt(key, encryptData):
myCipher = AES.new(key, AES.MODE_ECB) # 新建一个 AES 算法实例，使用 ECB（电子密码本）模式
bytes = myCipher.decrypt(encryptData) # 调用解密方法，得到解密后的数据
return bytes # 返回解密数据
def load_encrypt_model(model_path, key='test'):
with open(model_path, 'rb') as f_:
bytes_aes = f_.read()
bytes = DeCrypt(PadKey(key.encode()), bytes_aes)
pad_len = int(model_path.split('_')[-1].split('.')[0])
return bytes[:-pad_len]
if __name__ == '__main__':
key = "test"
file_path = ''
output_path = ''
with open(file_path, 'rb') as f: # 以二进制模式打开文件
bytes_ori = f.read() # 将文件内容读取出来到字节列表中
print('源文件长度：{}'.format(len(bytes_ori)))
key = PadKey(key.encode()) # 将密钥转换位字节列表并补齐密钥
bytes = PadTest(bytes_ori) # 补齐原始数据
print('补齐后的源文件长度：{}'.format(len(bytes)))
encryptTest = EnCrypt(key, bytes) # 利用密钥对原始数据进行加密
start_time = time.time()
decryptTest = DeCrypt(key, encryptTest) # 利用密钥对加密的数据进行解密
dt = time.time() - start_time
print("解密时间： ", dt)
pad_len = len(bytes) - len(bytes_ori)
print(pad_len)
decryptTest = decryptTest[:-pad_len]
print('补齐长度：{} '.format(len(decryptTest)))
if operator.eq(bytes_ori, decryptTest): # 检查加解密是否成功
print('AES 加解密成功！')
with open(output_path, "wb") as fo:
fo.write(encryptTest)
else:
print('AES 加解密失败，解密数据与元数据不相等')
|