python实现TGZ文件的AES加密

下面是一个实现TGZ文件的AES加密的Python代码：

import os
import tarfile
from Crypto.Cipher import AES

# 定义加密函数
def encrypt_file(input_file, output_file, key, iv):
    chunk_size = 64 * 1024  # 64 KB
    file_size = os.path.getsize(input_file)

    # 创建 AES 加密器
    cipher = AES.new(key, AES.MODE_CBC, iv)

    with open(input_file, 'rb') as in_file, open(output_file, 'wb') as out_file:
        # 写入文件大小
        out_file.write(file_size.to_bytes(8, byteorder='big'))

        # 加密文件数据
        while True:
            chunk = in_file.read(chunk_size)
            if len(chunk) == 0:
                break
            elif len(chunk) % 16 != 0:
                # 填充数据
                chunk += b' ' * (16 - len(chunk) % 16)
            out_file.write(cipher.encrypt(chunk))

# 定义解密函数
def decrypt_file(input_file, output_file, key, iv):
    chunk_size = 64 * 1024  # 64 KB

    # 创建 AES 解密器
    cipher = AES.new(key, AES.MODE_CBC, iv)

    with open(input_file, 'rb') as in_file, open(output_file, 'wb') as out_file:
        # 读取文件大小
        file_size_bytes = in_file.read(8)
        file_size = int.from_bytes(file_size_bytes, byteorder='big')

        # 解密文件数据
        while True:
            chunk = in_file.read(chunk_size)
            if len(chunk) == 0:
                break
            out_file.write(cipher.decrypt(chunk))

        # 删除填充数据
        out_file.truncate(file_size)

# 定义 TGZ 文件加密函数
def encrypt_tgz(input_file, output_file, key, iv):
    # 创建压缩文件对象
    tar = tarfile.open(input_file, 'r:gz')

    # 创建加密文件对象
    out_file = tarfile.open(output_file, 'w:gz')

    # 遍历压缩文件中的所有文件，并加密
    for member in tar.getmembers():
        # 创建临时文件
        temp_file = member.name + '.temp'

        # 解压文件到临时文件
        tar.extract(member, path=temp_file)

        # 加密文件
        encrypt_file(temp_file, temp_file + '.enc', key, iv)

        # 将加密后的文件添加到加密文件对象中
        out_file.add(temp_file + '.enc', arcname=member.name)

        # 删除临时文件和加密后的文件
        os.remove(temp_file)
        os.remove(temp_file + '.enc')

    # 关闭文件对象
    tar.close()
    out_file.close()

# 定义 TGZ 文件解密函数
def decrypt_tgz(input_file, output_file, key, iv):
    # 创建解密文件对象
    out_file = tarfile.open(output_file, 'w:gz')

    # 创建加密文件对象
    tar = tarfile.open(input_file, 'r:gz')

    # 遍历加密文件中的所有文件，并解密
    for member in tar.getmembers():
        # 创建临时文件
        temp_file = member.name + '.temp'

        # 解密文件到临时文件
        decrypt_file(member.name, temp_file, key, iv)

        # 将解密后的文件添加到解密文件对象中
        out_file.add(temp_file, arcname=member.name)

        # 删除临时文件
        os.remove(temp_file)

    # 关闭文件对象
    tar.close()
    out_file.close()

# 使用示例
if __name__ == '__main__':
    input_file = 'test.tgz'
    output_file = 'test_encrypted.tgz'
    key = b'Sixteen byte key'
    iv = b'Sixteen byte iv'

    # 加密 TGZ 文件
    encrypt_tgz(input_file, output_file, key, iv)

    # 解密 TGZ 文件
    decrypt_tgz(output_file, 'test_decrypted.tgz', key, iv)

在上述代码中，我们使用了PyCrypto模块中的`Crypto.Cipher.AES`类来实现AES加密和解密。加密函数`encrypt_file`和解密函数`decrypt_file`实现了对单个文件的加密和解密。`encrypt_tgz`和`decrypt_tgz`函数则实现了对TGZ文件的加密和解密，这两个函数遍历TGZ文件中的所有文件，并对每个文件进行加密或解密，最后生成加密或解密后的TGZ文件。

需要注意的是，加密和解密过程中需要使用相同的密钥和初始向量（IV），否则解密将失败。因此，在使用上述代码时，请确保使用相同的密钥和IV。