mirror of
https://github.com/hyb-oyqq/FRAISEMOE-Addons-Installer-NEXT.git
synced 2025-12-16 20:10:28 +00:00
ba5e3cdbc1
- 在下载管理器和离线模式管理器中集成哈希验证功能，确保补丁文件的完整性。 - 添加进度对话框以显示哈希验证过程，提升用户体验。 - 优化哈希验证线程，支持进度更新和错误处理，确保在验证失败时提供清晰反馈。 - 更新相关逻辑以支持新功能，提升代码可维护性和可读性。
313 lines
15 KiB
Python
313 lines
15 KiB
Python
import os
|
||
import hashlib
|
||
import py7zr
|
||
import tempfile
|
||
import traceback
|
||
from PySide6.QtCore import QThread, Signal
|
||
from utils.logger import setup_logger
|
||
|
||
# 初始化logger
|
||
logger = setup_logger("hash_thread")
|
||
|
||
class HashThread(QThread):
|
||
pre_finished = Signal(dict)
|
||
after_finished = Signal(dict)
|
||
|
||
def __init__(self, mode, install_paths, plugin_hash, installed_status, main_window=None):
|
||
"""初始化哈希检查线程
|
||
|
||
Args:
|
||
mode: 检查模式，"pre"或"after"
|
||
install_paths: 安装路径字典
|
||
plugin_hash: 插件哈希值字典
|
||
installed_status: 安装状态字典
|
||
main_window: 主窗口实例，用于访问UI和状态
|
||
"""
|
||
super().__init__()
|
||
self.mode = mode
|
||
self.install_paths = install_paths
|
||
self.plugin_hash = plugin_hash
|
||
self.installed_status = installed_status.copy()
|
||
self.main_window = main_window
|
||
|
||
def run(self):
|
||
"""运行线程"""
|
||
debug_mode = False
|
||
|
||
# 尝试检测是否处于调试模式
|
||
if self.main_window and hasattr(self.main_window, 'debug_manager'):
|
||
debug_mode = self.main_window.debug_manager._is_debug_mode()
|
||
|
||
if self.mode == "pre":
|
||
status_copy = self.installed_status.copy()
|
||
|
||
for game_version, install_path in self.install_paths.items():
|
||
if not os.path.exists(install_path):
|
||
status_copy[game_version] = False
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 哈希预检查 - {game_version} 补丁文件不存在: {install_path}")
|
||
continue
|
||
|
||
try:
|
||
expected_hash = self.plugin_hash.get(game_version, "")
|
||
if not expected_hash:
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 哈希预检查 - {game_version} 没有预期哈希值，跳过哈希检查")
|
||
# 当没有预期哈希值时，保持当前状态不变
|
||
continue
|
||
|
||
with open(install_path, "rb") as f:
|
||
file_hash = hashlib.sha256(f.read()).hexdigest()
|
||
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 哈希预检查 - {game_version}")
|
||
logger.debug(f"DEBUG: 文件路径: {install_path}")
|
||
logger.debug(f"DEBUG: 预期哈希值: {expected_hash}")
|
||
logger.debug(f"DEBUG: 实际哈希值: {file_hash}")
|
||
logger.debug(f"DEBUG: 哈希匹配: {file_hash == expected_hash}")
|
||
|
||
if file_hash == expected_hash:
|
||
status_copy[game_version] = True
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 哈希预检查 - {game_version} 哈希匹配成功")
|
||
else:
|
||
status_copy[game_version] = False
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 哈希预检查 - {game_version} 哈希不匹配")
|
||
except Exception as e:
|
||
status_copy[game_version] = False
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 哈希预检查异常 - {game_version}: {str(e)}")
|
||
|
||
self.pre_finished.emit(status_copy)
|
||
|
||
elif self.mode == "after":
|
||
result = {"passed": True, "game": "", "message": ""}
|
||
|
||
for game_version, install_path in self.install_paths.items():
|
||
if not os.path.exists(install_path):
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 哈希后检查 - {game_version} 补丁文件不存在: {install_path}")
|
||
continue
|
||
|
||
try:
|
||
expected_hash = self.plugin_hash.get(game_version, "")
|
||
if not expected_hash:
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 哈希后检查 - {game_version} 没有预期哈希值，跳过哈希检查")
|
||
# 当没有预期哈希值时，跳过检查
|
||
continue
|
||
|
||
with open(install_path, "rb") as f:
|
||
file_hash = hashlib.sha256(f.read()).hexdigest()
|
||
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 哈希后检查 - {game_version}")
|
||
logger.debug(f"DEBUG: 文件路径: {install_path}")
|
||
logger.debug(f"DEBUG: 预期哈希值: {expected_hash}")
|
||
logger.debug(f"DEBUG: 实际哈希值: {file_hash}")
|
||
logger.debug(f"DEBUG: 哈希匹配: {file_hash == expected_hash}")
|
||
|
||
if file_hash != expected_hash:
|
||
result["passed"] = False
|
||
result["game"] = game_version
|
||
result["message"] = f"\n{game_version} 安装后的文件校验失败。\n\n文件可能已损坏或被篡改，请重新安装。\n"
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 哈希后检查 - {game_version} 哈希不匹配")
|
||
break
|
||
elif debug_mode:
|
||
logger.debug(f"DEBUG: 哈希后检查 - {game_version} 哈希匹配成功")
|
||
except Exception as e:
|
||
result["passed"] = False
|
||
result["game"] = game_version
|
||
result["message"] = f"\n{game_version} 安装后的文件校验过程中发生错误。\n\n错误信息: {str(e)}\n"
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 哈希后检查异常 - {game_version}: {str(e)}")
|
||
break
|
||
|
||
self.after_finished.emit(result)
|
||
|
||
|
||
class OfflineHashVerifyThread(QThread):
|
||
"""离线模式下验证补丁文件哈希的线程，支持进度更新"""
|
||
|
||
progress = Signal(int) # 进度信号，0-100
|
||
finished = Signal(bool, str) # 完成信号，(成功/失败, 错误信息)
|
||
|
||
def __init__(self, game_version, file_path, plugin_hash, main_window=None):
|
||
"""初始化离线哈希验证线程
|
||
|
||
Args:
|
||
game_version: 游戏版本名称
|
||
file_path: 补丁压缩包文件路径
|
||
plugin_hash: 插件哈希值字典
|
||
main_window: 主窗口实例，用于访问UI和状态
|
||
"""
|
||
super().__init__()
|
||
self.game_version = game_version
|
||
self.file_path = file_path
|
||
self.plugin_hash = plugin_hash
|
||
self.main_window = main_window
|
||
|
||
def run(self):
|
||
"""运行线程"""
|
||
debug_mode = False
|
||
|
||
# 尝试检测是否处于调试模式
|
||
if self.main_window and hasattr(self.main_window, 'debug_manager'):
|
||
debug_mode = self.main_window.debug_manager._is_debug_mode()
|
||
|
||
# 获取预期的哈希值
|
||
expected_hash = self.plugin_hash.get(self.game_version, "")
|
||
|
||
if not expected_hash:
|
||
logger.warning(f"DEBUG: 未找到 {self.game_version} 的预期哈希值")
|
||
self.finished.emit(False, f"未找到 {self.game_version} 的预期哈希值")
|
||
return
|
||
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 开始验证补丁文件: {self.file_path}")
|
||
logger.debug(f"DEBUG: 游戏版本: {self.game_version}")
|
||
logger.debug(f"DEBUG: 预期哈希值: {expected_hash}")
|
||
|
||
try:
|
||
# 检查文件是否存在
|
||
if not os.path.exists(self.file_path):
|
||
if debug_mode:
|
||
logger.warning(f"DEBUG: 补丁文件不存在: {self.file_path}")
|
||
self.finished.emit(False, f"补丁文件不存在: {self.file_path}")
|
||
return
|
||
|
||
# 检查文件大小
|
||
file_size = os.path.getsize(self.file_path)
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 补丁文件大小: {file_size} 字节")
|
||
|
||
if file_size == 0:
|
||
if debug_mode:
|
||
logger.warning(f"DEBUG: 补丁文件大小为0，无效文件")
|
||
self.finished.emit(False, "补丁文件大小为0，无效文件")
|
||
return
|
||
|
||
# 创建临时目录用于解压文件
|
||
with tempfile.TemporaryDirectory() as temp_dir:
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 创建临时目录: {temp_dir}")
|
||
|
||
# 发送进度信号 - 10%
|
||
self.progress.emit(10)
|
||
|
||
# 解压补丁文件
|
||
try:
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 开始解压文件: {self.file_path}")
|
||
|
||
with py7zr.SevenZipFile(self.file_path, mode="r") as archive:
|
||
# 获取压缩包内文件列表
|
||
file_list = archive.getnames()
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 压缩包内文件列表: {file_list}")
|
||
|
||
# 解压所有文件
|
||
archive.extractall(path=temp_dir)
|
||
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 解压完成")
|
||
# 列出解压后的文件
|
||
extracted_files = []
|
||
for root, dirs, files in os.walk(temp_dir):
|
||
for file in files:
|
||
extracted_files.append(os.path.join(root, file))
|
||
logger.debug(f"DEBUG: 解压后的文件列表: {extracted_files}")
|
||
except Exception as e:
|
||
if debug_mode:
|
||
logger.error(f"DEBUG: 解压补丁文件失败: {e}")
|
||
logger.error(f"DEBUG: 错误类型: {type(e).__name__}")
|
||
logger.error(f"DEBUG: 错误堆栈: {traceback.format_exc()}")
|
||
self.finished.emit(False, f"解压补丁文件失败: {str(e)}")
|
||
return
|
||
|
||
# 发送进度信号 - 50%
|
||
self.progress.emit(50)
|
||
|
||
# 获取补丁文件路径
|
||
patch_file = None
|
||
if "Vol.1" in self.game_version:
|
||
patch_file = os.path.join(temp_dir, "vol.1", "adultsonly.xp3")
|
||
elif "Vol.2" in self.game_version:
|
||
patch_file = os.path.join(temp_dir, "vol.2", "adultsonly.xp3")
|
||
elif "Vol.3" in self.game_version:
|
||
patch_file = os.path.join(temp_dir, "vol.3", "update00.int")
|
||
elif "Vol.4" in self.game_version:
|
||
patch_file = os.path.join(temp_dir, "vol.4", "vol4adult.xp3")
|
||
elif "After" in self.game_version:
|
||
patch_file = os.path.join(temp_dir, "after", "afteradult.xp3")
|
||
|
||
if not patch_file or not os.path.exists(patch_file):
|
||
if debug_mode:
|
||
logger.warning(f"DEBUG: 未找到解压后的补丁文件: {patch_file}")
|
||
# 尝试查找可能的替代文件
|
||
alternative_files = []
|
||
for root, dirs, files in os.walk(temp_dir):
|
||
for file in files:
|
||
if file.endswith('.xp3') or file.endswith('.int'):
|
||
alternative_files.append(os.path.join(root, file))
|
||
if alternative_files:
|
||
logger.debug(f"DEBUG: 找到可能的替代文件: {alternative_files}")
|
||
|
||
# 检查解压目录结构
|
||
logger.debug(f"DEBUG: 检查解压目录结构:")
|
||
for root, dirs, files in os.walk(temp_dir):
|
||
logger.debug(f"DEBUG: 目录: {root}")
|
||
logger.debug(f"DEBUG: 子目录: {dirs}")
|
||
logger.debug(f"DEBUG: 文件: {files}")
|
||
self.finished.emit(False, f"未找到解压后的补丁文件")
|
||
return
|
||
|
||
# 发送进度信号 - 70%
|
||
self.progress.emit(70)
|
||
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 找到解压后的补丁文件: {patch_file}")
|
||
|
||
# 计算补丁文件哈希值
|
||
try:
|
||
# 读取文件内容并计算哈希值，同时更新进度
|
||
file_size = os.path.getsize(patch_file)
|
||
chunk_size = 1024 * 1024 # 1MB
|
||
hash_obj = hashlib.sha256()
|
||
|
||
with open(patch_file, "rb") as f:
|
||
bytes_read = 0
|
||
while chunk := f.read(chunk_size):
|
||
hash_obj.update(chunk)
|
||
bytes_read += len(chunk)
|
||
# 计算进度 (70-95%)
|
||
progress = 70 + int(25 * bytes_read / file_size)
|
||
self.progress.emit(min(95, progress))
|
||
|
||
file_hash = hash_obj.hexdigest()
|
||
|
||
# 比较哈希值
|
||
result = file_hash.lower() == expected_hash.lower()
|
||
|
||
# 发送进度信号 - 100%
|
||
self.progress.emit(100)
|
||
|
||
if debug_mode:
|
||
logger.debug(f"DEBUG: 补丁文件 {patch_file} 哈希值验证: {'成功' if result else '失败'}")
|
||
logger.debug(f"DEBUG: 预期哈希值: {expected_hash}")
|
||
logger.debug(f"DEBUG: 实际哈希值: {file_hash}")
|
||
|
||
self.finished.emit(result, "" if result else "补丁文件哈希验证失败，文件可能已损坏或被篡改")
|
||
except Exception as e:
|
||
if debug_mode:
|
||
logger.error(f"DEBUG: 计算补丁文件哈希值失败: {e}")
|
||
logger.error(f"DEBUG: 错误类型: {type(e).__name__}")
|
||
self.finished.emit(False, f"计算补丁文件哈希值失败: {str(e)}")
|
||
except Exception as e:
|
||
if debug_mode:
|
||
logger.error(f"DEBUG: 验证补丁哈希值失败: {e}")
|
||
logger.error(f"DEBUG: 错误类型: {type(e).__name__}")
|
||
logger.error(f"DEBUG: 错误堆栈: {traceback.format_exc()}")
|
||
self.finished.emit(False, f"验证补丁哈希值失败: {str(e)}") |