281 lines
9.1 KiB
Python
281 lines
9.1 KiB
Python
|
"""
|
|||
|
|
作者：太一
|
|||
|
|
微信：taiyi1224
|
|||
|
|
邮箱：shoubo1224@qq.com
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import shutil
|
|||
|
|
import subprocess
|
|||
|
|
import tempfile
|
|||
|
|
import json
|
|||
|
|
import struct
|
|||
|
|
import hashlib
|
|||
|
|
import zlib
|
|||
|
|
from typing import Tuple, Optional
|
|||
|
|
import time
|
|||
|
|
|
|||
|
|
class EXEEncryptor:
|
|||
|
|
"""EXE文件加密器类 - 提供EXE文件加密、编译和验证功能"""
|
|||
|
|
|
|||
|
|
def __init__(self):
|
|||
|
|
"""初始化加密器"""
|
|||
|
|
pass
|
|||
|
|
|
|||
|
|
# encryptor.py
|
|||
|
|
|
|||
|
|
# 新增依赖
|
|||
|
|
from tqdm import tqdm
|
|||
|
|
|
|||
|
|
def encrypt_file(self, source_path: str, output_path: str, validator_path: str,
|
|||
|
|
db_config: dict, progress_callback=None) -> Tuple[bool, str]:
|
|||
|
|
try:
|
|||
|
|
# 1. 读取原始 EXE
|
|||
|
|
with open(source_path, 'rb') as f:
|
|||
|
|
original_content = f.read()
|
|||
|
|
file_hash = hashlib.sha256(original_content).hexdigest()
|
|||
|
|
|
|||
|
|
# 2. 构造资源
|
|||
|
|
resource_data = {
|
|||
|
|
'original_exe': original_content.hex(),
|
|||
|
|
'original_size': len(original_content),
|
|||
|
|
'file_hash': file_hash,
|
|||
|
|
'db_config': db_config
|
|||
|
|
}
|
|||
|
|
resource_code = json.dumps(resource_data, separators=(',', ':'))
|
|||
|
|
|
|||
|
|
# 3. 读取模板并替换
|
|||
|
|
with open(validator_path, 'r', encoding='utf-8', errors='ignore') as f:
|
|||
|
|
tpl = f.read()
|
|||
|
|
code = tpl.replace('# RESOURCE_DATA_PLACEHOLDER\nRESOURCE_DATA = None',
|
|||
|
|
f'RESOURCE_DATA = {resource_code}')
|
|||
|
|
|
|||
|
|
# 4. 写入临时文件
|
|||
|
|
tmp_py = tempfile.mktemp(suffix='.py')
|
|||
|
|
with open(tmp_py, 'w', encoding='utf-8') as f:
|
|||
|
|
f.write(code)
|
|||
|
|
|
|||
|
|
# 5. 编译（带进度）
|
|||
|
|
if progress_callback:
|
|||
|
|
progress_callback(10)
|
|||
|
|
ok, msg = self.compile_to_exe(tmp_py, output_path)
|
|||
|
|
if progress_callback:
|
|||
|
|
progress_callback(100)
|
|||
|
|
return ok, msg
|
|||
|
|
except Exception as e:
|
|||
|
|
return False, str(e)
|
|||
|
|
|
|||
|
|
def _create_encryption_header(self, original_size: int, db_config: dict, file_hash: str) -> bytes:
|
|||
|
|
"""创建加密文件头信息"""
|
|||
|
|
header_data = {
|
|||
|
|
'version': '2.0',
|
|||
|
|
'original_size': original_size,
|
|||
|
|
'encryption_time': int(time.time()),
|
|||
|
|
'compression_level': 6,
|
|||
|
|
'file_hash': file_hash,
|
|||
|
|
'db_config': db_config
|
|||
|
|
}
|
|||
|
|
return json.dumps(header_data).encode('utf-8').ljust(256, b'\x00')
|
|||
|
|
|
|||
|
|
def _simple_encrypt(self, data: bytes) -> bytes:
|
|||
|
|
"""简化版XOR加密"""
|
|||
|
|
key = b'EXEProtector#2024'
|
|||
|
|
encrypted = bytearray(data)
|
|||
|
|
key_len = len(key)
|
|||
|
|
|
|||
|
|
for i in range(len(encrypted)):
|
|||
|
|
encrypted[i] ^= key[i % key_len]
|
|||
|
|
|
|||
|
|
return bytes(encrypted)
|
|||
|
|
|
|||
|
|
def compile_to_exe(self, python_file: str, output_path: str = None) -> Tuple[bool, str]:
|
|||
|
|
"""尝试将Python文件编译为EXE"""
|
|||
|
|
try:
|
|||
|
|
# 确保Python文件存在
|
|||
|
|
py_file = python_file.replace('.exe', '.py')
|
|||
|
|
if not os.path.exists(py_file):
|
|||
|
|
return False, f"Python源文件不存在: {py_file}"
|
|||
|
|
|
|||
|
|
print(f"Debug: 开始编译 {py_file} 到 {python_file}")
|
|||
|
|
|
|||
|
|
# 检查PyInstaller
|
|||
|
|
try:
|
|||
|
|
result = subprocess.run(['pyinstaller', '--version'],
|
|||
|
|
capture_output=True, text=True, timeout=10)
|
|||
|
|
if result.returncode != 0:
|
|||
|
|
return self._fallback_to_python_file(py_file, python_file)
|
|||
|
|
except (subprocess.TimeoutExpired, FileNotFoundError):
|
|||
|
|
return self._fallback_to_python_file(py_file, python_file)
|
|||
|
|
|
|||
|
|
# 获取文件信息
|
|||
|
|
file_dir = os.path.dirname(os.path.abspath(py_file))
|
|||
|
|
exe_name = os.path.basename(python_file).replace('.exe', '')
|
|||
|
|
|
|||
|
|
# 切换到文件目录
|
|||
|
|
original_cwd = os.getcwd()
|
|||
|
|
os.chdir(file_dir)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 创建规范文件内容
|
|||
|
|
spec_content = self._create_spec_content(py_file, exe_name)
|
|||
|
|
spec_file = f'{exe_name}.spec'
|
|||
|
|
|
|||
|
|
with open(spec_file, 'w', encoding='utf-8') as f:
|
|||
|
|
f.write(spec_content)
|
|||
|
|
|
|||
|
|
print(f"Debug: 创建规范文件: {spec_file}")
|
|||
|
|
|
|||
|
|
# 使用规范文件编译
|
|||
|
|
cmd = ['pyinstaller', '--clean', '--noconfirm', spec_file]
|
|||
|
|
|
|||
|
|
print(f"Debug: 执行编译命令: {' '.join(cmd)}")
|
|||
|
|
|
|||
|
|
# 执行编译
|
|||
|
|
result = subprocess.run(cmd, capture_output=True, text=True, timeout=300)
|
|||
|
|
|
|||
|
|
if result.returncode == 0:
|
|||
|
|
# 检查输出文件
|
|||
|
|
output_exe = os.path.join('dist', f'{exe_name}.exe')
|
|||
|
|
if os.path.exists(output_exe):
|
|||
|
|
# 移动到目标位置
|
|||
|
|
target_path = os.path.basename(python_file)
|
|||
|
|
final_output = output_path or os.path.basename(python_file).replace('.py', '.exe')
|
|||
|
|
os.makedirs(os.path.dirname(final_output), exist_ok=True)
|
|||
|
|
if os.path.exists(final_output):
|
|||
|
|
os.remove(final_output)
|
|||
|
|
shutil.move(output_exe, final_output)
|
|||
|
|
print(f"Debug: 编译成功，输出文件: {final_output}")
|
|||
|
|
return True, "EXE编译成功"
|
|||
|
|
else:
|
|||
|
|
print(f"Debug: 找不到输出文件: {output_exe}")
|
|||
|
|
return False, "编译完成但找不到输出文件"
|
|||
|
|
else:
|
|||
|
|
# 编译失败，保留Python文件
|
|||
|
|
error_info = result.stderr[:500] if result.stderr else result.stdout[:500]
|
|||
|
|
print(f"Debug: 编译失败: {error_info}")
|
|||
|
|
return self._fallback_to_python_file(py_file, python_file, f"编译失败: {error_info}")
|
|||
|
|
|
|||
|
|
finally:
|
|||
|
|
os.chdir(original_cwd)
|
|||
|
|
|
|||
|
|
except subprocess.TimeoutExpired:
|
|||
|
|
print("Debug: 编译超时")
|
|||
|
|
return self._fallback_to_python_file(py_file, python_file, "编译超时")
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"Debug: 编译过程异常: {e}")
|
|||
|
|
return False, f"编译过程异常: {str(e)}"
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _create_spec_content(self, py_file: str, exe_name: str) -> str:
|
|||
|
|
"""创建PyInstaller规范文件内容"""
|
|||
|
|
return f'''# -*- mode: python ; coding: utf-8 -*-
|
|||
|
|
|
|||
|
|
block_cipher = None
|
|||
|
|
|
|||
|
|
a = Analysis(
|
|||
|
|
['{os.path.basename(py_file)}'],
|
|||
|
|
pathex=[],
|
|||
|
|
binaries=[],
|
|||
|
|
datas=[],
|
|||
|
|
hiddenimports=[
|
|||
|
|
'mysql.connector',
|
|||
|
|
'cryptography.fernet',
|
|||
|
|
'tkinter',
|
|||
|
|
'tkinter.messagebox',
|
|||
|
|
'tkinter.scrolledtext',
|
|||
|
|
'hashlib',
|
|||
|
|
'subprocess',
|
|||
|
|
'tempfile',
|
|||
|
|
'threading',
|
|||
|
|
'atexit',
|
|||
|
|
'psutil',
|
|||
|
|
'ctypes',
|
|||
|
|
'win32api',
|
|||
|
|
'win32con',
|
|||
|
|
'win32security',
|
|||
|
|
'win32process',
|
|||
|
|
],
|
|||
|
|
hookspath=[],
|
|||
|
|
hooksconfig={{}},
|
|||
|
|
runtime_hooks=[],
|
|||
|
|
excludes=[],
|
|||
|
|
win_no_prefer_redirects=False,
|
|||
|
|
win_private_assemblies=False,
|
|||
|
|
cipher=block_cipher,
|
|||
|
|
noarchive=False,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
pyz = PYZ(a.pure, a.zipped_data, cipher=block_cipher)
|
|||
|
|
|
|||
|
|
exe = EXE(
|
|||
|
|
pyz,
|
|||
|
|
a.scripts,
|
|||
|
|
a.binaries,
|
|||
|
|
a.zipfiles,
|
|||
|
|
a.datas,
|
|||
|
|
[],
|
|||
|
|
name='{exe_name}',
|
|||
|
|
debug=False,
|
|||
|
|
bootloader_ignore_signals=False,
|
|||
|
|
strip=False,
|
|||
|
|
upx=False,
|
|||
|
|
upx_exclude=[],
|
|||
|
|
runtime_tmpdir=None,
|
|||
|
|
console=False,
|
|||
|
|
disable_windowed_traceback=False,
|
|||
|
|
argv_emulation=False,
|
|||
|
|
target_arch=None,
|
|||
|
|
codesign_identity=None,
|
|||
|
|
entitlements_file=None,
|
|||
|
|
uac_admin=False,
|
|||
|
|
uac_uiaccess=False,
|
|||
|
|
)
|
|||
|
|
'''
|
|||
|
|
|
|||
|
|
def _fallback_to_python_file(self, py_file: str, target_file: str,
|
|||
|
|
reason: str = "PyInstaller不可用") -> Tuple[bool, str]:
|
|||
|
|
"""回退到Python文件"""
|
|||
|
|
try:
|
|||
|
|
# 将.exe改为.py
|
|||
|
|
python_target = target_file.replace('.exe', '.py')
|
|||
|
|
|
|||
|
|
if py_file != python_target:
|
|||
|
|
if os.path.exists(python_target):
|
|||
|
|
os.remove(python_target)
|
|||
|
|
shutil.copy2(py_file, python_target)
|
|||
|
|
|
|||
|
|
message = f"{reason}，使用Python文件: {python_target}"
|
|||
|
|
print(f"Debug: {message}")
|
|||
|
|
return True, message
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
error_msg = f"回退失败: {str(e)}"
|
|||
|
|
print(f"Debug: {error_msg}")
|
|||
|
|
return False, error_msg
|
|||
|
|
|
|||
|
|
def _cleanup_build_files(self, exe_name: str):
|
|||
|
|
"""清理编译产生的临时文件"""
|
|||
|
|
try:
|
|||
|
|
# 删除build目录
|
|||
|
|
if os.path.exists('build'):
|
|||
|
|
shutil.rmtree('build', ignore_errors=True)
|
|||
|
|
|
|||
|
|
# 删除dist目录
|
|||
|
|
if os.path.exists('dist'):
|
|||
|
|
shutil.rmtree('dist', ignore_errors=True)
|
|||
|
|
|
|||
|
|
# 删除spec文件
|
|||
|
|
spec_file = f'{exe_name}.spec'
|
|||
|
|
if os.path.exists(spec_file):
|
|||
|
|
os.remove(spec_file)
|
|||
|
|
|
|||
|
|
print("Debug: 清理临时文件完成")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"Debug: 清理临时文件时出错: {e}")
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
|