From c5980c419a0aee3306f364073dac3be7bc823ab6 Mon Sep 17 00:00:00 2001 From: tongtongstudio Date: Fri, 6 Feb 2026 18:32:02 +0800 Subject: [PATCH] =?UTF-8?q?version:=20bugfixes:=20update:=E5=A2=9E?= =?UTF-8?q?=E5=8A=A0python=E5=8A=A0=E5=AF=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- py/encryption_video.py | 182 +++++++++++++++++++++++++++++++++++++++++ 1 file changed, 182 insertions(+) create mode 100644 py/encryption_video.py diff --git a/py/encryption_video.py b/py/encryption_video.py new file mode 100644 index 0000000..de1da7d --- /dev/null +++ b/py/encryption_video.py @@ -0,0 +1,182 @@ +import tkinter as tk +from tkinter import filedialog, messagebox, ttk +from Crypto.Cipher import AES +from Crypto.Util.Padding import pad +from Crypto.Random import get_random_bytes +import os +import base64 + +class VideoEncryptorGUI: + def __init__(self, master): + self.master = master + master.title("视频加密工具") + master.geometry("500x400") + + self.input_file_path = None + + # 创建GUI组件 + self.label_title = tk.Label(master, text="视频文件加密工具", font=("Arial", 14, "bold")) + self.label_title.pack(pady=10) + + self.btn_select = tk.Button(master, text="选择视频文件", command=self.select_file, width=20, height=2) + self.btn_select.pack(pady=5) + + self.label_file = tk.Label(master, text="未选择文件", wraplength=400) + self.label_file.pack(pady=5) + + # Key输入 + self.label_key = tk.Label(master, text="AES密钥 (Base64编码或16/24/32字节字符串):") + self.label_key.pack(pady=(10,0)) + self.entry_key = tk.Entry(master, width=50, show="*") + self.entry_key.pack(pady=5) + + # IV输入 + self.label_iv = tk.Label(master, text="初始化向量IV (Base64编码或16字节字符串):") + self.label_iv.pack(pady=(10,0)) + self.entry_iv = tk.Entry(master, width=50) + self.entry_iv.pack(pady=5) + + # 生成随机Key和IV按钮 + self.btn_generate = tk.Button(master, text="生成随机密钥和IV", command=self.generate_key_iv, width=20) + self.btn_generate.pack(pady=5) + + # 加密按钮 + self.btn_encrypt = tk.Button(master, text="加密文件", command=self.encrypt_video, width=20, height=2, bg="#4CAF50", fg="white") + self.btn_encrypt.pack(pady=10) + + self.status_label = tk.Label(master, text="", fg="blue") + self.status_label.pack(pady=5) + + # 说明文本 + help_text = "说明：输出文件将与输入文件同目录，扩展名为.hnv" + self.label_help = tk.Label(master, text=help_text, fg="gray", font=("Arial", 9)) + self.label_help.pack(pady=10) + + def select_file(self): + """选择要加密的文件""" + file_path = filedialog.askopenfilename( + title="选择视频文件", + filetypes=[("视频文件", "*.mp4 *.avi *.mov *.mkv"), ("所有文件", "*.*")] + ) + if file_path: + self.input_file_path = file_path + self.label_file.config(text=file_path) + self.status_label.config(text="文件已选择，请设置密钥") + + def generate_key_iv(self): + """生成随机的密钥和IV""" + # 生成随机密钥（32字节，AES-256）和IV（16字节） + key = get_random_bytes(32) + iv = get_random_bytes(16) + + # 转换为Base64编码字符串，显示在输入框中 + key_b64 = base64.b64encode(key).decode('utf-8') + iv_b64 = base64.b64encode(iv).decode('utf-8') + + self.entry_key.delete(0, tk.END) + self.entry_key.insert(0, key_b64) + self.entry_iv.delete(0, tk.END) + self.entry_iv.insert(0, iv_b64) + + self.status_label.config(text="已生成随机密钥和IV，请妥善保存！") + + def encrypt_video(self): + """加密视频文件""" + if not self.input_file_path: + messagebox.showerror("错误", "请先选择要加密的文件") + return + + key_str = self.entry_key.get().strip() + iv_str = self.entry_iv.get().strip() + + if not key_str or not iv_str: + messagebox.showerror("错误", "请输入密钥和IV") + return + + try: + # 处理密钥：尝试Base64解码，否则使用字符串编码 + try: + key = base64.b64decode(key_str) + except: + key = key_str.encode('utf-8') + + # 处理IV：尝试Base64解码，否则使用字符串编码 + try: + iv = base64.b64decode(iv_str) + except: + iv = iv_str.encode('utf-8') + + # 确保密钥长度为16、24或32字节 + if len(key) not in [16, 24, 32]: + if len(key) < 16: + key = key.ljust(16, b'\0') + elif len(key) < 24: + key = key.ljust(24, b'\0') + elif len(key) < 32: + key = key.ljust(32, b'\0') + else: + key = key[:32] + + # 确保IV长度为16字节 + if len(iv) != 16: + if len(iv) < 16: + iv = iv.ljust(16, b'\0') + else: + iv = iv[:16] + + # 生成输出文件路径 + file_dir = os.path.dirname(self.input_file_path) + file_name = os.path.basename(self.input_file_path) + file_base = os.path.splitext(file_name)[0] + output_file = os.path.join(file_dir, f"{file_base}.hnv") + + # 执行加密 + self.do_encryption(self.input_file_path, output_file, key, iv) + + self.status_label.config(text=f"加密成功！输出文件: {output_file}") + messagebox.showinfo("成功", f"文件加密完成！\n输出文件: {output_file}") + + except Exception as e: + messagebox.showerror("加密错误", f"加密过程中发生错误: {str(e)}") + + def do_encryption(self, input_path, output_path, key, iv): + """ + 执行加密操作 + 使用AES/CBC/PKCS5Padding模式，与Java代码保持一致 + """ + # 读取输入文件 + with open(input_path, 'rb') as f_in: + plaintext = f_in.read() + + # 创建密码器 + cipher = AES.new(key, AES.MODE_CBC, iv) + + # 应用PKCS5填充并加密 + padded_data = pad(plaintext, AES.block_size) + ciphertext = cipher.encrypt(padded_data) + + # 写入输出文件 + with open(output_path, 'wb') as f_out: + f_out.write(ciphertext) + +def main(): + # 检查所需库是否已安装 + try: + from Crypto.Cipher import AES + from Crypto.Util.Padding import pad + from Crypto.Random import get_random_bytes + except ImportError: + print("正在安装所需库...") + import subprocess + import sys + subprocess.check_call([sys.executable, "-m", "pip", "install", "pycryptodome"]) + print("库安装完成，请重新运行程序") + return + + # 创建主窗口 + root = tk.Tk() + app = VideoEncryptorGUI(root) + root.mainloop() + +if __name__ == "__main__": + main() \ No newline at end of file