任意文件嵌入

Author: 邹上豪

Main.py

@@ -22,90 +22,88 @@ def generate_empty_audio(output_file, file_name):
         messagebox.showerror("错误", f"生成空音频失败：{str(e)}")
         return False
 
-def embed_image_in_audio(image_file):
+def embed_file_in_audio(file_path):
     """
-    将图片嵌入到生成的无声音频文件中
+    将任意文件嵌入到生成的无声音频文件中
     """
-    file_name = os.path.splitext(os.path.basename(image_file))[0]
-    output_audio_file = os.path.join(os.path.dirname(image_file), "temp_silent.mp3")
-    if not generate_empty_audio(output_audio_file, file_name):
+    file_name_with_extension = os.path.basename(file_path)  # 完整文件名
+    output_audio_file = os.path.join(os.path.dirname(file_path), "temp_silent.mp3")
+    if not generate_empty_audio(output_audio_file, file_name_with_extension):
         return
 
-    output_file = os.path.join(os.path.dirname(image_file), file_name + "(embedded).mp3")
+    output_file = os.path.join(os.path.dirname(file_path), os.path.splitext(file_name_with_extension)[0] + " (embedded).mp3")
     try:
-        with open(output_audio_file, "rb") as audio, open(image_file, "rb") as image:
+        with open(output_audio_file, "rb") as audio, open(file_path, "rb") as file:
             with open(output_file, "wb") as output:
                 output.write(audio.read())  # 写入音频数据
-                output.write(image.read())  # 写入图片数据
+                output.write(b"<FILE>" + file_name_with_extension.encode('utf-8') + b"</FILE>")  # 写入文件名，使用标签分隔
+                output.write(file.read())  # 写入文件数据
         os.remove(output_audio_file)  # 删除临时音频文件
-        messagebox.showinfo("完成", f"图片已嵌入！文件保存为：{output_file}")
+        messagebox.showinfo("完成", f"文件已嵌入！文件保存为：{output_file}")
     except Exception as e:
         messagebox.showerror("错误", f"嵌入失败：{str(e)}")
 
-def extract_image_from_file(input_file):
+def extract_file_from_audio(input_file):
     """
-    从伪装文件中提取图片，并生成图片文件
+    从伪装文件中提取嵌入的文件，并恢复其原始文件名
     """
     try:
         with open(input_file, "rb") as file:
             data = file.read()
 
-        # 检测图片格式
-        formats = {
-            b'\xFF\xD8\xFF': ".jpg",  # JPG 文件头
-            b'\x89PNG': ".png"          # PNG 文件头
-        }
-
-        start_index = -1
-        file_extension = None
-        for header, ext in formats.items():
-            index = data.find(header)
-            if index != -1:
-                start_index = index
-                file_extension = ext
-                break
-
-        if start_index == -1:
-            messagebox.showerror("错误", "未找到支持的图片文件头，文件可能不包含图片。")
+        # 查找文件名标签
+        start_tag = b"<FILE>"
+        end_tag = b"</FILE>"
+        start_index = data.find(start_tag) + len(start_tag)
+        end_index = data.find(end_tag)
+
+        if start_index == -1 or end_index == -1:
+            messagebox.showerror("错误", "未找到嵌入文件的标识符。")
             return
 
-        # 提取图片数据
-        embedded_file_name = os.path.splitext(os.path.basename(input_file))[0].replace("(embedded)", "")
-        output_image_file = os.path.join(os.path.dirname(input_file), f"{embedded_file_name}{file_extension}")
-        with open(output_image_file, "wb") as image_file:
-            image_file.write(data[start_index:])  # 保存从图片头开始的所有数据
+        # 提取文件名
+        file_name = data[start_index:end_index].decode('utf-8', errors='ignore').strip()
+
+        # 如果文件名为空，提供默认名称
+        if not file_name:
+            file_name = "extracted_file"
+
+        # 提取嵌入文件数据
+        output_file = os.path.join(os.path.dirname(input_file), file_name)
+        with open(output_file, "wb") as extracted_file:
+            extracted_file.write(data[end_index + len(end_tag):])  # 保存嵌入的文件数据
 
-        messagebox.showinfo("完成", f"图片已提取！文件保存为：{output_image_file}")
+        messagebox.showinfo("完成", f"文件已提取！文件保存为：{output_file}")
     except Exception as e:
         messagebox.showerror("错误", f"提取失败：{str(e)}")
 
 def embed_action():
-    """选择图片文件并嵌入到生成的无声音频中"""
-    image_file = filedialog.askopenfilename(title="选择图片文件", filetypes=[("图片文件", "*.jpg;*.jpeg;*.png")])
-    if not image_file:
+    """选择文件并嵌入到生成的无声音频中"""
+    file_path = filedialog.askopenfilename(title="选择文件", filetypes=[("所有文件", "*.*")])
+    if not file_path:
         return
 
-    embed_image_in_audio(image_file)
+    embed_file_in_audio(file_path)
 
 def extract_action():
-    """选择伪装文件并提取图片"""
+    """选择伪装文件并提取嵌入的文件"""
     input_file = filedialog.askopenfilename(title="选择伪装文件", filetypes=[("音频文件", "*.mp3")])
     if not input_file:
         return
 
-    extract_image_from_file(input_file)
+    extract_file_from_audio(input_file)
 
 # 创建主界面
 root = tk.Tk()
-root.title("音频图片嵌入与提取工具")
+root.title("文件嵌入与提取工具")
 root.geometry("400x200")
 
 # 嵌入按钮
-embed_button = tk.Button(root, text="嵌入图片到音频", command=embed_action, height=2, width=20)
+embed_button = tk.Button(root, text="嵌入文件到音频", command=embed_action, height=2, width=20)
 embed_button.pack(pady=20)
 
 # 提取按钮
-extract_button = tk.Button(root, text="提取图片从音频", command=extract_action, height=2, width=20)
+extract_button = tk.Button(root, text="提取文件从音频", command=extract_action, height=2, width=20)
 extract_button.pack(pady=20)
 
 # 运行主循环