以下是一个简单的 Python 示例代码，用于实现文件加密与访问控制系统的基本功能。此代码使用 AES 算法进行文件加密和解密，同时实现了简单的用户认证来控制文件访问。 ```python from Crypto.Cipher import AES from Crypto.Util.Padding import pad, unpad import os import json # 用户数据库，存储用户名和密码 users = { "user1": "password1", "user2": "password2" } # 加密函数 def encrypt_file(file_path, key): cipher = AES.new(key, AES.MODE_CBC) with open(file_path, 'rb') as f: plaintext = f.read() padded_plaintext = pad(plaintext, AES.block_size) ciphertext = cipher.encrypt(padded_plaintext) iv = cipher.iv encrypted_file_path = file_path + ".enc" with open(encrypted_file_path, 'wb') as f: f.write(iv) f.write(ciphertext) return encrypted_file_path # 解密函数 def decrypt_file(encrypted_file_path, key): with open(encrypted_file_path, 'rb') as f: iv = f.read(16) ciphertext = f.read() cipher = AES.new(key, AES.MODE_CBC, iv) decrypted_data = unpad(cipher.decrypt(ciphertext), AES.block_size) decrypted_file_path = encrypted_file_path.replace(".enc", "") with open(decrypted_file_path, 'wb') as f: f.write(decrypted_data) return decrypted_file_path # 用户认证函数 def authenticate_user(username, password): return users.get(username) == password # 主函数，模拟文件加密与访问控制 def main(): username = input("请输入用户名: ") password = input("请输入密码: ") if authenticate_user(username, password): key = os.urandom(16) # 生成 16 字节的 AES 密钥 file_path = input("请输入要加密的文件路径: ") encrypted_file = encrypt_file(file_path, key) print(f"文件已加密，加密文件路径: {encrypted_file}") decrypt_choice = input("是否要解密文件？(y/n): ") if decrypt_choice.lower() == 'y': decrypt_file(encrypted_file, key) print("文件已解密") else: print("认证失败，无法访问文件。") if __name__ == "__main__": main() ``` ### 代码说明 1. **用户认证**：使用 `authenticate_user` 函数验证用户输入的用户名和密码是否匹配存储在 `users` 字典中的信息。 2. **文件加密**：`encrypt_file` 函数使用 AES 算法的 CBC 模式对文件进行加密，并将加密后的文件保存为 `.enc` 扩展名。 3. **文件解密**：`decrypt_file` 函数对加密文件进行解密，并将解密后的文件保存为原始文件名。 4. **主函数**：`main` 函数负责用户交互，包括用户认证、文件加密和解密操作。