Blowfish 加密解密 Python 代码 - 文件加密解密工具
import os
import random
import string
import tkinter as tk
from tkinter import filedialog, messagebox
from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
import pyperclip
# 填充函数
def pad(data, block_size):
padding_length = block_size - len(data) % block_size
padding = bytes([padding_length]) * padding_length
return data + padding
# 去除填充函数
def unpad(data, block_size):
padding_length = data[-1]
return data[:-padding_length]
class Blowfish:
block_size = 8
MODE_CBC = modes.CBC
# Blowfish 加密解密函数
def Blowfish_encrypt_decrypt(key, data, mode):
# 获取块大小，并生成随机的初始向量
bs = Blowfish.block_size
iv = os.urandom(bs)
# 创建 Blowfish 密码器并进行加密/解密操作
cipher = Cipher(algorithms.Blowfish(key), mode, backend=default_backend()).create_encryptor()
data = pad(data, bs) # 对数据进行填充
encrypted_data = iv + cipher.update(data) + cipher.finalize() # 添加初始向量，并加密数据
return encrypted_data
def Blowfish_encrypt(key, data):
# 使用 CBC 模式进行加密
return Blowfish_encrypt_decrypt(key, data, Blowfish.MODE_CBC)
def Blowfish_decrypt(key, data):
bs = Blowfish.block_size
# 判断是否需要填充
if len(data) % bs != 0:
raise ValueError('Invalid data length. Data length must be a multiple of {} bytes.'.format(bs))
# 获取初始向量和加密后的数据
iv = data[:bs]
encrypted_data = data[bs:]
# 创建 Blowfish 密码器并进行解密操作
cipher = Cipher(algorithms.Blowfish(key), Blowfish.MODE_CBC, backend=default_backend()).create_decryptor()
decrypted_data = cipher.update(encrypted_data) + cipher.finalize()
# 去除填充
unpadded_data = unpad(decrypted_data, bs)
return unpadded_data
# 验证密钥长度
def validate_key(key):
# 验证密钥长度是否符合要求
key_length = len(key.encode())
if key_length < 8 or key_length > 56 or key_length % 8 != 0:
raise ValueError('Invalid key length. Key must be between 8 and 56 bytes, and a multiple of 8 bytes.')
return key.encode()
# 创建主窗口
def create_new_window3():
window = tk.Tk()
window.title('Blowfish 加密解密工具')
# 输入文件路径
tk.Label(window, text='输入文件路径:').grid(row=0, column=0, padx=5, pady=5)
e_input = tk.Entry(window, width=50)
e_input.grid(row=0, column=1, padx=5, pady=5)
tk.Button(window, text='选择文件', command=on_select_input).grid(row=0, column=2, padx=5, pady=5)
# 输出文件路径
tk.Label(window, text='输出文件路径:').grid(row=1, column=0, padx=5, pady=5)
e_output = tk.Entry(window, width=50)
e_output.grid(row=1, column=1, padx=5, pady=5)
tk.Button(window, text='选择文件', command=on_select_output).grid(row=1, column=2, padx=5, pady=5)
# 密钥
tk.Label(window, text='密钥:').grid(row=2, column=0, padx=5, pady=5)
e_key = tk.Entry(window, width=50)
e_key.grid(row=2, column=1, padx=5, pady=5)
tk.Button(window, text='生成密钥', command=on_generate_key).grid(row=2, column=2, padx=5, pady=5)
tk.Button(window, text='复制密钥', command=on_copy_key).grid(row=2, column=3, padx=5, pady=5)
# 加密按钮
tk.Button(window, text='加密', command=on_encrypt, width=10, height=2).grid(row=3, column=0, columnspan=2, padx=5, pady=10)
# 解密按钮
tk.Button(window, text='解密', command=on_decrypt, width=10, height=2).grid(row=3, column=2, columnspan=2, padx=5, pady=10)
# 文件加密函数
def encrypt_file(input_file_path, output_file_path, key):
# 验证密钥长度是否符合要求
key = validate_key(key)
# 读取输入文件并加密文件内容
with open(input_file_path, 'rb') as f:
data = f.read()
encrypted_data = Blowfish_encrypt(key, data)
# 将加密后的内容写入输出文件
with open(output_file_path, 'wb') as f:
f.write(encrypted_data)
return True
# 文件解密函数
def decrypt_file(input_file_path, output_file_path, key):
# 验证密钥长度是否符合要求
key = validate_key(key)
# 读取输入文件并解密文件内容
with open(input_file_path, 'rb') as f:
encrypted_data = f.read()
decrypted_data = Blowfish_decrypt(key, encrypted_data)
# 将解密后的内容写入输出文件
with open(output_file_path, 'wb') as f:
f.write(decrypted_data)
return True
# 加密按钮事件处理函数
def on_encrypt():
# 获取输入文件路径、输出文件路径和密钥
input_file_path = e_input.get()
output_file_path = e_output.get()
key = e_key.get()
# 检查是否填写完整
if not input_file_path or not output_file_path or not key:
messagebox.showwarning('Warning', '请填写完整的输入、输出路径和密钥！')
return
# 调用文件加密函数进行处理
if encrypt_file(input_file_path, output_file_path, key):
messagebox.showinfo('Success', '文件加密成功！')
# 解密按钮事件处理函数
def on_decrypt():
# 获取输入文件路径、输出文件路径和密钥
input_file_path = e_input.get()
output_file_path = e_output.get()
key = e_key.get()
# 检查是否填写完整
if not input_file_path or not output_file_path or not key:
messagebox.showwarning('Warning', '请填写完整的输入、输出路径和密钥！')
return
# 调用文件解密函数进行处理
if decrypt_file(input_file_path, output_file_path, key):
messagebox.showinfo('Success', '文件解密成功！')
# 生成随机密钥函数
def generate_random_key():
# 生成随机密钥，包括字母和数字，长度为8
key = ''.join(random.choices(string.ascii_letters + string.digits, k=8))
return key
# 生成密钥按钮的事件处理函数
def on_generate_key():
# 调用生成随机密钥函数生成密钥
key = generate_random_key()
# 清空密钥输入框并将生成的密钥赋值给输入框
e_key.delete(0, tk.END)
e_key.insert(0, key)
# 复制按钮函数
def on_copy_key():
# 获取密钥
key = e_key.get()
if key:
# 将密钥复制到剪贴板
pyperclip.copy(key)
messagebox.showinfo('Copied', '密钥已复制到剪贴板！')
# 选择输入文件函数
def on_select_input():
# 打开文件选择对话框，获取选择的文件路径
file_path = filedialog.askopenfilename()
if file_path:
e_input.delete(0, tk.END)
e_input.insert(0, file_path)
# 选择输出文件函数
def on_select_output():
# 打开文件保存对话框，并设置默认文件扩展名为.docx
file_path = filedialog.asksaveasfilename(defaultextension='.docx')
if file_path:
e_output.delete(0, tk.END)
e_output.insert(0, file_path)
window.mainloop()
使用方法:
- 运行代码，会弹出一个窗口。
- 在“输入文件路径”框中选择要加密或解密的文件。
- 在“输出文件路径”框中选择保存加密或解密后的文件位置。
- 在“密钥”框中可以输入密钥，也可以点击“生成密钥”按钮生成随机密钥。
- 点击“加密”或“解密”按钮进行操作。
注意:
- 密钥长度必须在 8 到 56 字节之间，且必须是 8 的倍数。
- 密钥需要保密，否则无法解密数据。
- 为了安全起见，建议使用较长的密钥，并定期更换密钥。
- 本代码仅供学习参考，实际应用中可能需要进行更多安全处理。
代码功能:
- 使用 Blowfish 算法进行文件加密解密。
- 提供生成随机密钥的功能。
- 提供复制密钥的功能。
- 支持选择输入输出文件路径。
- 提供简单的错误处理。
原文地址: https://www.cveoy.top/t/topic/oRTy 著作权归作者所有。请勿转载和采集!