欢迎访问宙启技术站
智能推送

在Python中实现一个基本的文件加密解密程序

发布时间:2023-12-04 15:39:46

在Python中实现文件加密解密程序可以使用加密算法和解密算法来对文件内容进行加密和解密的操作。常见的加密算法有对称加密算法和非对称加密算法,其中对称加密算法中最常用的是AES算法。

下面我们以AES算法为例来实现一个基本的文件加密解密程序:

1. 安装必要的Python库

首先,我们需要安装pycryptodome库,它是一个支持多种加密算法的Python库。可以使用以下命令来安装:

   pip install pycryptodome
   

2. 导入必要的库和模块

在Python程序中,我们需要导入一些库和模块来实现文件加密解密功能。在这里,我们需要导入Crypto.Cipher模块中的AES类和Padding模块。

   from Crypto.Cipher import AES
   from Crypto.Util.Padding import pad, unpad
   

3. 加密文件函数

下面是一个加密文件的函数,它将接收一个输入文件名、输出文件名和密钥作为参数,然后使用AES算法将输入文件加密并存储到输出文件中。

   def encrypt_file(input_file, output_file, key):
       cipher = AES.new(key, AES.MODE_CBC)  # 创建AES加密对象
       with open(input_file, 'rb') as file:
           data = file.read()  # 读取输入文件的内容
           padded_data = pad(data, AES.block_size)  # 填充数据至AES块大小的倍数
           encrypted_data = cipher.encrypt(padded_data)  # 加密数据
       with open(output_file, 'wb') as file:
           file.write(cipher.iv + encrypted_data)  # 合并iv和加密数据并将结果写入输出文件
   

4. 解密文件函数

下面是一个解密文件的函数,它将接收一个输入文件名、输出文件名和密钥作为参数,然后使用AES算法将输入文件解密并存储到输出文件中。

   def decrypt_file(input_file, output_file, key):
       with open(input_file, 'rb') as file:
           cipher_iv = file.read(AES.block_size)  # 读取文件中的iv
           encrypted_data = file.read()  # 读取文件中的加密数据
           cipher = AES.new(key, AES.MODE_CBC, iv=cipher_iv)  # 创建AES解密对象
           decrypted_data = unpad(cipher.decrypt(encrypted_data), AES.block_size)  # 解密数据并去除填充
       with open(output_file, 'wb') as file:
           file.write(decrypted_data)  # 将解密数据写入输出文件
   

5. 使用例子

在主程序中,我们可以使用上述的加密函数和解密函数来加密和解密文件。下面是一个简单的使用例子:

   # 定义输入文件、输出文件和密钥
   input_file = 'input.txt'
   encrypted_file = 'encrypted.txt'
   decrypted_file = 'decrypted.txt'
   key = b"ThisIsASecretKey"

   # 加密文件
   encrypt_file(input_file, encrypted_file, key)

   # 解密文件
   decrypt_file(encrypted_file, decrypted_file, key)
   

在上述例子中,我们首先定义了一个输入文件(input.txt)、加密后的输出文件(encrypted.txt),和解密后的输出文件(decrypted.txt),以及一个密钥(ThisIsASecretKey)。然后调用加密函数将输入文件加密为加密文件,并调用解密函数将加密文件解密为解密文件。

在这个基本的文件加密解密程序中,我们使用了AES算法来加密和解密文件内容。为了保证加密的安全性,我们使用了随机生成的iv(初始化向量)并将其与加密的数据一起存储到输出文件中,以用于解密时的IV匹配。这样即使使用相同的密钥,不同的文件加密后也会得到不同的加密结果,增加了破解的难度。不过在实际使用中,还需要考虑更多的安全性和可靠性的因素,例如密钥的管理和存储等。