def encrypt_file(key, in_path, out_path):
    """使用AES-256 CBC模式加密文件"""
    iv = get_random_bytes(16)  # 生成隨機(jī)初始化向量
    cipher = AES.new(key, AES.MODE_CBC, iv)  # 創(chuàng)建加密器
    
    with open(in_path, 'rb') as f_in:
        data = f_in.read()  # 讀取原始文件內(nèi)容
    
    # 加密并填充數(shù)據(jù)
    ct_bytes = cipher.encrypt(pad(data, AES.block_size))
    encrypted = iv + ct_bytes  # 組合IV和密文
    
    with open(out_path, 'wb') as f_out:
        f_out.write(encrypted)  # 寫(xiě)入加密文件

def generate_stub_code(password_hash, encrypted_data_base64):
    """生成包含密碼驗(yàn)證界面的Python代碼"""
    return f'''#!/usr/bin/env python
# Tkinter運(yùn)行時(shí)修復(fù) - 解決打包環(huán)境下的關(guān)鍵問(wèn)題
def fix_tkinter_runtime():
    if getattr(sys, 'frozen', False):
        base_path = sys._MEIPASS
        tk_data_dir = os.path.join(base_path, '_tk_data')
        if not os.path.exists(tk_data_dir):
            tk_data_dir = os.path.join(base_path, 'tk', 'data')
        os.environ['TKDATA'] = tk_data_dir  # 設(shè)置環(huán)境變量

# 在創(chuàng)建Tkinter界面之前調(diào)用修復(fù)函數(shù)
fix_tkinter_runtime()

# 密碼驗(yàn)證邏輯
def check_password():
    password = password_entry.get()
    key = hashlib.sha256(password.encode()).digest()  # 從密碼生成密鑰
    decrypted_data = decrypt_data(key)  # 嘗試解密
    if decrypted_data:
        root.destroy()
        execute_decrypted(decrypted_data)  # 執(zhí)行解密后的程序

# 創(chuàng)建美觀的解鎖界面
root = tk.Tk()
root.title("程序解鎖")
root.geometry("260x230")
root.resizable(False, False)
root.configure(bg="#f0f0f0")

# 界面組件：圖標(biāo)、輸入框、按鈕等
icon_label = ttk.Label(main_frame, text="??", font=("Arial", 24))
password_entry = ttk.Entry(main_frame, show="*", width=25)
submit_btn = ttk.Button(main_frame, text="解鎖程序", command=check_password, width=15)
'''

def main():
    """主GUI界面和處理流程"""
    # 文件選擇函數(shù)
    def select_exe():
        # 自動(dòng)設(shè)置輸出路徑
        output_path = os.path.join(dir_name, f"{name_without_ext}_protected.exe")
    
    # 加密處理核心邏輯
    def encrypt():
        # 驗(yàn)證輸入
        if not all([exe_path, output_path, password, confirm]):
            messagebox.showerror("錯(cuò)誤", "所有字段都必須填寫(xiě)")
            return
        
        # 生成密鑰和哈希
        password_hash = hashlib.sha256(password.encode()).hexdigest()
        key = hashlib.sha256(password.encode()).digest()
        
        # 加密文件
        encrypt_file(key, exe_path, encrypted_temp_path)
        
        # 生成stub代碼
        stub_code = generate_stub_code(password_hash, encrypted_data_base64)
        
        # PyInstaller打包處理
        pyinstaller_cmd = [
            *find_pyinstaller(),  # 自動(dòng)檢測(cè)PyInstaller
            '--onefile', '--noconsole',
            '--name', output_basename,
            *icon_cmd,  # 圖標(biāo)參數(shù)
            stub_path
        ]
        
        # 執(zhí)行打包
        subprocess.run(pyinstaller_cmd, shell=sys.platform.startswith('win'))
        
        # 處理結(jié)果
        if os.path.exists(generated_exe):
            shutil.copy(generated_exe, output_path)
            status_label.config(text=f"加密成功! 文件已保存到:\n{output_path}", fg="green")

if __name__ == "__main__":
    # 自動(dòng)安裝缺失依賴(lài)
    try:
        from Crypto.Cipher import AES
    except ImportError:
        print("正在安裝依賴(lài)庫(kù)pycryptodome...")
        subprocess.check_call([sys.executable, "-m", "pip", "install", "pycryptodome"])
        os.execl(sys.executable, sys.executable, *sys.argv)  # 重啟腳本
    
    # 檢查PyInstaller
    try:
        import PyInstaller
    except ImportError:
        subprocess.check_call([sys.executable, "-m", "pip", "install", "pyinstaller"])
    
    main()

def fix_tkinter_runtime():
    """解決PyInstaller打包環(huán)境下Tkinter的資源路徑問(wèn)題"""
    if getattr(sys, 'frozen', False):
        base_path = sys._MEIPASS
        tk_data_dir = os.path.join(base_path, '_tk_data')
        if not os.path.exists(tk_data_dir):
            tk_data_dir = os.path.join(base_path, 'tk', 'data')
        os.environ['TKDATA'] = tk_data_dir

def find_pyinstaller():
    """自動(dòng)檢測(cè)系統(tǒng)中可用的PyInstaller安裝方式"""
    if shutil.which('pyinstaller'):
        return ['pyinstaller']
    
    # 嘗試通過(guò)Python解釋器調(diào)用
    for py_exe in ['python', 'python3']:
        if shutil.which(py_exe):
            try:
                subprocess.run([py_exe, '-c', 'import PyInstaller'], check=True)
                return [py_exe, '-m', 'PyInstaller']
            except:
                continue
    return None

try:
    # 加密和打包過(guò)程
except Exception as e:
    error_details = traceback.format_exc()
    messagebox.showerror(
        "錯(cuò)誤", 
        f"加密過(guò)程中出錯(cuò):\n\n{str(e)}\n\n"
        f"詳細(xì)錯(cuò)誤信息:\n{error_details[:1000]}"
    )
finally:
    # 確保清理臨時(shí)文件
    shutil.rmtree(temp_dir, ignore_errors=True)