191 lines
7.1 KiB
Python
191 lines
7.1 KiB
Python
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)
|
||
self.entry_key.pack(pady=5)
|
||
|
||
# IV输入
|
||
self.label_iv = tk.Label(master, text="初始化向量IV (Base64编码或8字节字符串):")
|
||
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):
|
||
"""生成随机的密钥和nonce"""
|
||
# 生成随机密钥（32字节，AES-256）和nonce（8字节用于CTR模式）
|
||
key = get_random_bytes(32)
|
||
nonce = get_random_bytes(8) # AES CTR模式使用的nonce应该是8字节
|
||
|
||
# 转换为Base64编码字符串，显示在输入框中
|
||
key_b64 = base64.b64encode(key).decode('utf-8')
|
||
nonce_b64 = base64.b64encode(nonce).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, nonce_b64)
|
||
|
||
self.status_label.config(text="已生成随机密钥和nonce，请妥善保存！")
|
||
|
||
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):
|
||
print(f"iv len: {len(iv)}")
|
||
|
||
"""
|
||
执行加密操作
|
||
使用AES/CTR/NoPadding模式
|
||
"""
|
||
try:
|
||
# 创建CTR模式加密器
|
||
cipher = AES.new(key, AES.MODE_CTR, nonce=iv)
|
||
with open(input_path, 'rb') as fin:
|
||
with open(output_path, 'wb') as fout:
|
||
# 将IV写入文件开头，解密时需要
|
||
fout.write(iv)
|
||
|
||
# 逐块加密文件
|
||
while True:
|
||
chunk = fin.read(4096) # 每次读取4KB
|
||
if len(chunk) == 0:
|
||
break
|
||
encrypted_chunk = cipher.encrypt(chunk)
|
||
fout.write(encrypted_chunk)
|
||
|
||
except Exception as e:
|
||
raise Exception(f"加密失败: {str(e)}")
|
||
|
||
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() |