Python中使用PBKDF2对文件进行加密和解密

Python中可以使用pbkdf2模块对文件进行加密和解密。pbkdf2是一个使用密码作为输入生成密钥的函数，它可以增加破解密码的难度，提高密码安全性。

使用pbkdf2进行文件加密和解密，需要以下步骤：

1. 导入pbkdf2、hmac和hashlib模块。

2. 定义一个加密函数，接收文件路径、密码和加密后的文件路径作为参数。

- 打开原文件和加密后的文件，以二进制方式读取和写入。

- 生成salt，可以用os.urandom()生成随机salt。

- 使用pbkdf2函数生成密钥，传入密码、salt和迭代次数。

- 生成一个hmac对象，传入密钥和原文件内容。

- 使用hmac对象的digest()方法生成加密后的内容，并写入加密后的文件。

3. 定义一个解密函数，接收加密后的文件路径、密码和解密后的文件路径作为参数。

- 打开加密后的文件和解密后的文件，以二进制方式读取和写入。

- 读取加密后的文件内容。

- 生成salt，可以用加密后的文件内容的前16字节作为salt。

- 使用pbkdf2函数生成密钥，传入密码、salt和迭代次数。

- 生成一个hmac对象，传入密钥和加密后的文件内容。

- 使用hmac对象的digest()方法生成解密后的内容，并写入解密后的文件。

下面是一个使用pbkdf2对文件进行加密和解密的例子：

import pbkdf2
import hmac
import hashlib
import os


def encrypt_file(file_path, password, encrypted_file_path):
    with open(file_path, 'rb') as file:
        with open(encrypted_file_path, 'wb') as encrypted_file:
            # 生成随机salt
            salt = os.urandom(16)
            # 生成密钥
            key = pbkdf2.pbkdf2_hmac('sha256', password.encode(), salt, 100000)
            # 生成hmac对象
            hmac_obj = hmac.new(key, digestmod=hashlib.sha256)
            while True:
                chunk = file.read(1024)
                if not chunk:
                    break
                # 更新hmac对象
                hmac_obj.update(chunk)
                # 写入加密后的内容
                encrypted_file.write(chunk)
            # 写入salt和hmac摘要
            encrypted_file.write(salt + hmac_obj.digest())


def decrypt_file(encrypted_file_path, password, decrypted_file_path):
    with open(encrypted_file_path, 'rb') as encrypted_file:
        with open(decrypted_file_path, 'wb') as decrypted_file:
            # 读取加密后的内容
            content = encrypted_file.read()
            # 获取salt
            salt = content[:16]
            # 获取hmac摘要
            hmac_digest = content[16:]
            # 生成密钥
            key = pbkdf2.pbkdf2_hmac('sha256', password.encode(), salt, 100000)
            # 生成hmac对象
            hmac_obj = hmac.new(key, digestmod=hashlib.sha256)
            while True:
                chunk = content[:1024]
                if not chunk:
                    break
                # 更新hmac对象
                hmac_obj.update(chunk)
                # 写入解密后的内容
                decrypted_file.write(chunk)
            # 验证hmac摘要
            if hmac_obj.digest() == hmac_digest:
                print('解密成功')
            else:
                print('解密失败')


# 加密文件
encrypt_file('plain.txt', 'password', 'encrypted.txt')

# 解密文件
decrypt_file('encrypted.txt', 'password', 'decrypted.txt')

这个例子中，我们使用'password'这个密码对'plain.txt'进行加密，得到'encrypted.txt'文件。然后再使用同样的密码对'encrypted.txt'进行解密，得到'decrypted.txt'文件。解密过程中会进行hmac摘要的验证，如果验证失败则表示解密失败。

需要注意的是，由于pbkdf2和hmac都是单向函数，所以加密后的文件无法还原成原始文件。解密过程中也无法得到密码，只能通过正确的密码才能进行解密。