ReHiFace-S / face_feature /utils /data_encrypt.py

Upload 117 files

89cf463 verified about 2 years ago

3.14 kB

# -- 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 加解密失败，解密数据与元数据不相等')

	# -- 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 加解密失败，解密数据与元数据不相等')