上传文件至 /

wrapper.py

@@ -0,0 +1,247 @@
+# wrapper_embed.py – 由 PyInstaller 打包，运行时提取并解密原始 EXE
+import os, sys, json, hashlib, tempfile, subprocess, struct
+from datetime import datetime
+import mysql.connector
+import tkinter as tk
+from tkinter import messagebox
+import platform
+
+MAGIC = b'ENC_MAGIC'
+HEAD = 512
+KEY = b'EXEWrapper#2024'
+
+
+def xor(data):
+    return bytes(d ^ KEY[i % len(KEY)] for i, d in enumerate(data))
+
+
+def get_machine_code():
+    import uuid
+    return hashlib.md5(str(uuid.getnode()).encode()).hexdigest()[:16].upper()
+
+
+def validate(key, machine):
+    try:
+        conn = mysql.connector.connect(
+            host='taiyiagi.xyz', user='taiyi', password='taiyi1224',
+            database='filesend', connection_timeout=10, autocommit=True
+        )
+        cur = conn.cursor(dictionary=True)
+        cur.execute('SELECT * FROM license_keys WHERE key_code=%s', (key,))
+        print("验证激活码")
+        rec = cur.fetchone()
+        if not rec: return False, '激活码不存在'
+        if rec['status'] == 'banned': return False, '激活码被封禁'
+        if rec['end_time'] < datetime.now(): return False, '激活码已过期'
+        if rec['status'] == 'active':
+            return (True, '验证成功') if rec['machine_code'] == machine else (False, '激活码已绑定其他设备')
+        if rec['status'] == 'unused':
+            cur.execute('UPDATE license_keys SET status=%s, machine_code=%s, start_time=NOW() WHERE key_code=%s',
+                        ('active', machine, key))
+            return cur.rowcount > 0, '激活成功' if cur.rowcount else '激活失败'
+        return False, '激活码状态异常'
+    except Exception as e:
+        return False, str(e)
+
+
+def get_bundled_data_path():
+    """获取打包数据文件路径"""
+    if getattr(sys, 'frozen', False):
+        # 如果是打包后的exe，数据文件在临时目录中
+        bundle_dir = sys._MEIPASS
+    else:
+        # 如果是脚本运行，数据文件在当前目录
+        bundle_dir = os.path.dirname(os.path.abspath(__file__))
+
+    # 查找 .dat 文件
+    for file in os.listdir(bundle_dir):
+        if file.endswith('.dat'):
+            return os.path.join(bundle_dir, file)
+    return None
+
+
+def extract_exe():
+    """从打包的数据文件中提取并解密原始EXE"""
+    try:
+        # 首先尝试从打包的数据文件中读取
+        data_file = get_bundled_data_path()
+        if data_file and os.path.exists(data_file):
+            with open(data_file, 'rb') as f:
+                data = f.read()
+        else:
+            # 如果没有找到数据文件，尝试从当前exe文件末尾读取
+            if getattr(sys, 'frozen', False):
+                exe = sys.executable
+                with open(exe, 'rb') as f:
+                    data = f.read()
+            else:
+                print("未找到加密数据文件")
+                return None
+
+        # 查找魔数位置
+        pos = data.find(MAGIC)
+        if pos == -1:
+            print("未找到加密数据标识")
+            return None
+
+        pos += len(MAGIC)
+
+        # 读取头部信息
+        if pos + HEAD > len(data):
+            print("数据文件损坏：头部信息不完整")
+            return None
+
+        header_data = data[pos:pos + HEAD].rstrip(b'\0')
+        try:
+            hdr = json.loads(header_data.decode())
+        except json.JSONDecodeError as e:
+            print(f"头部信息解析失败: {e}")
+            return None
+
+        pos += HEAD
+
+        # 解密数据
+        if pos >= len(data):
+            print("数据文件损坏：没有加密数据")
+            return None
+
+        encrypted_data = data[pos:]
+        if len(encrypted_data) == 0:
+            print("加密数据为空")
+            return None
+
+        dec = xor(encrypted_data)
+
+        # 验证文件完整性
+        if hashlib.sha256(dec).hexdigest() != hdr['hash']:
+            print("文件完整性校验失败")
+            return None
+
+        if len(dec) != hdr['size']:
+            print(f"文件大小不匹配，期望: {hdr['size']}, 实际: {len(dec)}")
+            return None
+
+        # 写入临时文件
+        temp_dir = tempfile.gettempdir()
+        filename = hdr['filename']
+        tmp = os.path.join(temp_dir, f"temp_{os.getpid()}_{filename}")
+
+        try:
+            with open(tmp, 'wb') as f:
+                f.write(dec)
+
+            # 设置可执行权限
+            if hasattr(os, 'chmod'):
+                os.chmod(tmp, 0o755)
+
+            print(f"成功提取文件到: {tmp}")
+            return tmp
+
+        except Exception as e:
+            print(f"写入临时文件失败: {e}")
+            return None
+
+    except Exception as e:
+        print(f"提取EXE文件时发生错误: {e}")
+        return None
+
+
+def run(path):
+    """运行解密后的程序"""
+    try:
+        if not os.path.exists(path):
+            print(f"文件不存在: {path}")
+            return False
+
+        print(f"正在启动程序: {path}")
+
+        if os.name == 'nt':  # Windows
+            # 使用 CREATE_NEW_CONSOLE 标志在新窗口中启动
+            import subprocess
+            subprocess.Popen([path],
+                             creationflags=subprocess.CREATE_NEW_CONSOLE,
+                             close_fds=False)
+        else:  # Linux/macOS
+            subprocess.Popen([path], close_fds=True)
+
+        # 给程序一些启动时间
+        import time
+        time.sleep(1)
+
+        return True
+
+    except Exception as e:
+        print(f"启动程序失败: {e}")
+        return False
+
+
+def gui():
+    def ok():
+        k = e.get().strip()
+        if not k:
+            return messagebox.showerror('错误', '请输入激活码')
+
+        try:
+            m = get_machine_code()
+            success, msg = validate(k, m)
+
+            if success:
+                messagebox.showinfo('成功', '激活成功！正在启动程序...')
+                root.destroy()
+
+                # 提取并运行程序
+                exe = extract_exe()
+                if exe:
+                    if run(exe):
+                        print("程序启动成功")
+                        # 延迟退出，给目标程序启动时间
+                        import time
+                        time.sleep(2)
+                        sys.exit(0)
+                    else:
+                        messagebox.showerror('错误', '程序启动失败')
+                else:
+                    messagebox.showerror('错误', '无法提取程序文件，请检查文件完整性')
+            else:
+                messagebox.showerror('失败', msg)
+
+        except Exception as e:
+            messagebox.showerror('错误', f'验证过程出错: {str(e)}')
+
+    root = tk.Tk()
+    root.title('软件激活')
+    root.geometry('400x200')
+    root.resizable(False, False)
+
+    # 居中显示窗口
+    root.eval('tk::PlaceWindow . center')
+
+    # 界面元素
+    tk.Label(root, text='请输入激活码', font=('', 13)).pack(pady=15)
+    e = tk.Entry(root, width=25)
+    e.pack()
+    e.focus()
+
+    tk.Button(root, text='验证并启动', command=ok).pack(pady=15)
+
+    # 显示机器码（用于调试）
+    machine_code = get_machine_code()
+    tk.Label(root, text=f'机器码: {machine_code}', font=('', 8), fg='gray').pack(pady=(10, 0))
+
+    root.bind('<Return>', lambda x: ok())
+    root.protocol('WM_DELETE_WINDOW', sys.exit)
+    root.mainloop()
+
+
+if __name__ == '__main__':
+    # 添加调试信息
+    print("程序启动...")
+    print(f"Python版本: {sys.version}")
+    print(f"当前工作目录: {os.getcwd()}")
+    print(f"是否为打包环境: {getattr(sys, 'frozen', False)}")
+
+    if getattr(sys, 'frozen', False):
+        print(f"打包临时目录: {sys._MEIPASS}")
+        print(f"打包临时目录内容: {os.listdir(sys._MEIPASS)}")
+
+    gui()