mirror of
https://github.com/hyb-oyqq/FRAISEMOE-Addons-Installer-NEXT.git
synced 2025-12-16 20:10:28 +00:00
96d20c6a5b
- 在主窗口和UI管理器中集成IPv6Manager，增强IPv6支持功能。 - 更新UI管理器以处理IPv6连接状态和切换事件，提供更好的用户反馈。 - 优化代码结构，简化IPv6相关功能的实现，提升可维护性。 - 添加打开hosts文件的功能，增强用户操作体验。
337 lines
13 KiB
Python
337 lines
13 KiB
Python
import os
|
||
import re
|
||
import subprocess
|
||
import sys
|
||
import time
|
||
from urllib.parse import urlparse
|
||
|
||
from PySide6.QtCore import QThread, Signal
|
||
from utils import resource_path
|
||
|
||
class IpOptimizer:
|
||
def __init__(self):
|
||
self.process = None
|
||
|
||
def get_optimal_ip(self, url: str) -> str | None:
|
||
"""
|
||
使用 CloudflareSpeedTest 工具获取给定 URL 的最优 Cloudflare IP。
|
||
|
||
Args:
|
||
url: 需要进行优选的下载链接。
|
||
|
||
Returns:
|
||
最优的 IP 地址字符串，如果找不到则返回 None。
|
||
"""
|
||
try:
|
||
cst_path = resource_path("cfst.exe")
|
||
if not os.path.exists(cst_path):
|
||
print(f"错误: cfst.exe 未在资源路径中找到。")
|
||
return None
|
||
|
||
ip_txt_path = resource_path("ip.txt")
|
||
|
||
# 正确的参数设置，根据cfst帮助文档
|
||
command = [
|
||
cst_path,
|
||
"-n", "1000", # 延迟测速线程数 (默认200)
|
||
"-p", "1", # 显示结果数量 (默认10个)
|
||
"-url", url, # 指定测速地址
|
||
"-f", ip_txt_path, # IP文件
|
||
"-dd", # 禁用下载测速，按延迟排序
|
||
"-o"," " # 不写入结果文件
|
||
]
|
||
|
||
creation_flags = subprocess.CREATE_NO_WINDOW if sys.platform == 'win32' else 0
|
||
|
||
print("--- CloudflareSpeedTest 开始执行 ---")
|
||
|
||
self.process = subprocess.Popen(
|
||
command,
|
||
stdin=subprocess.PIPE,
|
||
stdout=subprocess.PIPE,
|
||
stderr=subprocess.STDOUT,
|
||
text=True,
|
||
encoding='utf-8',
|
||
errors='replace',
|
||
creationflags=creation_flags,
|
||
bufsize=0
|
||
)
|
||
|
||
# 更新正则表达式以匹配cfst输出中的IP格式
|
||
# 匹配格式: IP地址在行首，后面跟着一些数字和文本
|
||
ip_pattern = re.compile(r'^(\d+\.\d+\.\d+\.\d+)\s+.*')
|
||
|
||
# 标记是否已经找到结果表头和完成标记
|
||
found_header = False
|
||
found_completion = False
|
||
|
||
stdout = self.process.stdout
|
||
if not stdout:
|
||
print("错误: 无法获取子进程的输出流。")
|
||
return None
|
||
|
||
optimal_ip = None
|
||
timeout_counter = 0
|
||
max_timeout = 300 # 增加超时时间到5分钟
|
||
|
||
while True:
|
||
if self.process.poll() is not None:
|
||
break
|
||
try:
|
||
ready = True
|
||
try:
|
||
line = stdout.readline()
|
||
except:
|
||
ready = False
|
||
|
||
if not ready or not line:
|
||
timeout_counter += 1
|
||
if timeout_counter > max_timeout:
|
||
print("超时: CloudflareSpeedTest 响应超时")
|
||
break
|
||
time.sleep(1)
|
||
continue
|
||
|
||
timeout_counter = 0
|
||
|
||
cleaned_line = line.strip()
|
||
if cleaned_line:
|
||
print(cleaned_line)
|
||
|
||
# 检测结果表头
|
||
if "IP 地址" in cleaned_line and "平均延迟" in cleaned_line:
|
||
print("检测到IP结果表头，准备获取IP地址...")
|
||
found_header = True
|
||
continue
|
||
|
||
# 检测完成标记
|
||
if "完整测速结果已写入" in cleaned_line or "按下 回车键 或 Ctrl+C 退出" in cleaned_line:
|
||
print("检测到测速完成信息")
|
||
found_completion = True
|
||
|
||
# 如果已经找到了IP，可以退出了
|
||
if optimal_ip:
|
||
break
|
||
|
||
# 已找到表头后，尝试匹配IP地址行
|
||
if found_header:
|
||
match = ip_pattern.search(cleaned_line)
|
||
if match and not optimal_ip: # 只保存第一个匹配的IP（最优IP）
|
||
optimal_ip = match.group(1)
|
||
print(f"找到最优 IP: {optimal_ip}")
|
||
# 找到最优IP后立即退出循环，不等待完成标记
|
||
break
|
||
|
||
except Exception as e:
|
||
print(f"读取输出时发生错误: {e}")
|
||
break
|
||
|
||
# 确保完全读取输出后再发送退出信号
|
||
if self.process and self.process.poll() is None:
|
||
try:
|
||
if self.process.stdin and not self.process.stdin.closed:
|
||
print("发送退出信号...")
|
||
self.process.stdin.write('\n')
|
||
self.process.stdin.flush()
|
||
except:
|
||
pass
|
||
|
||
self.stop()
|
||
|
||
print("--- CloudflareSpeedTest 执行结束 ---")
|
||
return optimal_ip
|
||
|
||
except Exception as e:
|
||
print(f"执行 CloudflareSpeedTest 时发生错误: {e}")
|
||
return None
|
||
|
||
def get_optimal_ipv6(self, url: str) -> str | None:
|
||
"""
|
||
使用 CloudflareSpeedTest 工具获取给定 URL 的最优 Cloudflare IPv6 地址。
|
||
|
||
Args:
|
||
url: 需要进行优选的下载链接。
|
||
|
||
Returns:
|
||
最优的 IPv6 地址字符串，如果找不到则返回 None。
|
||
"""
|
||
try:
|
||
cst_path = resource_path("cfst.exe")
|
||
if not os.path.exists(cst_path):
|
||
print(f"错误: cfst.exe 未在资源路径中找到。")
|
||
return None
|
||
|
||
ipv6_txt_path = resource_path("data/ipv6.txt")
|
||
if not os.path.exists(ipv6_txt_path):
|
||
print(f"错误: ipv6.txt 未在资源路径中找到。")
|
||
return None
|
||
|
||
# 正确的参数设置，根据cfst帮助文档
|
||
command = [
|
||
cst_path,
|
||
"-n", "1000", # 延迟测速线程数，IPv6测试线程稍少
|
||
"-p", "1", # 显示结果数量 (默认10个)
|
||
"-url", url, # 指定测速地址
|
||
"-f", ipv6_txt_path, # IPv6文件
|
||
"-dd", # 禁用下载测速，按延迟排序
|
||
"-o", " " # 不写入结果文件
|
||
]
|
||
|
||
creation_flags = subprocess.CREATE_NO_WINDOW if sys.platform == 'win32' else 0
|
||
|
||
print("--- CloudflareSpeedTest IPv6 开始执行 ---")
|
||
|
||
self.process = subprocess.Popen(
|
||
command,
|
||
stdin=subprocess.PIPE,
|
||
stdout=subprocess.PIPE,
|
||
stderr=subprocess.STDOUT,
|
||
text=True,
|
||
encoding='utf-8',
|
||
errors='replace',
|
||
creationflags=creation_flags,
|
||
bufsize=0
|
||
)
|
||
|
||
# 更新正则表达式以匹配cfst输出中的IPv6格式
|
||
# IPv6格式更加复杂，可能有多种表示形式
|
||
ipv6_pattern = re.compile(r'^([0-9a-fA-F:]+)\s+.*')
|
||
|
||
# 标记是否已经找到结果表头和完成标记
|
||
found_header = False
|
||
found_completion = False
|
||
|
||
stdout = self.process.stdout
|
||
if not stdout:
|
||
print("错误: 无法获取子进程的输出流。")
|
||
return None
|
||
|
||
optimal_ipv6 = None
|
||
timeout_counter = 0
|
||
max_timeout = 300 # 增加超时时间到5分钟
|
||
|
||
while True:
|
||
if self.process.poll() is not None:
|
||
break
|
||
try:
|
||
ready = True
|
||
try:
|
||
line = stdout.readline()
|
||
except:
|
||
ready = False
|
||
|
||
if not ready or not line:
|
||
timeout_counter += 1
|
||
if timeout_counter > max_timeout:
|
||
print("超时: CloudflareSpeedTest IPv6 响应超时")
|
||
break
|
||
time.sleep(1)
|
||
continue
|
||
|
||
timeout_counter = 0
|
||
|
||
cleaned_line = line.strip()
|
||
if cleaned_line:
|
||
print(cleaned_line)
|
||
|
||
# 检测结果表头
|
||
if "IP 地址" in cleaned_line and "平均延迟" in cleaned_line:
|
||
print("检测到IPv6结果表头，准备获取IPv6地址...")
|
||
found_header = True
|
||
continue
|
||
|
||
# 检测完成标记
|
||
if "完整测速结果已写入" in cleaned_line or "按下 回车键 或 Ctrl+C 退出" in cleaned_line:
|
||
print("检测到IPv6测速完成信息")
|
||
found_completion = True
|
||
|
||
# 如果已经找到了IPv6，可以退出了
|
||
if optimal_ipv6:
|
||
break
|
||
|
||
# 已找到表头后，尝试匹配IPv6地址行
|
||
if found_header:
|
||
match = ipv6_pattern.search(cleaned_line)
|
||
if match and not optimal_ipv6: # 只保存第一个匹配的IPv6（最优IPv6）
|
||
optimal_ipv6 = match.group(1)
|
||
print(f"找到最优 IPv6: {optimal_ipv6}")
|
||
# 找到最优IPv6后立即退出循环，不等待完成标记
|
||
break
|
||
|
||
except Exception as e:
|
||
print(f"读取输出时发生错误: {e}")
|
||
break
|
||
|
||
# 确保完全读取输出后再发送退出信号
|
||
if self.process and self.process.poll() is None:
|
||
try:
|
||
if self.process.stdin and not self.process.stdin.closed:
|
||
print("发送退出信号...")
|
||
self.process.stdin.write('\n')
|
||
self.process.stdin.flush()
|
||
except:
|
||
pass
|
||
|
||
self.stop()
|
||
|
||
print("--- CloudflareSpeedTest IPv6 执行结束 ---")
|
||
return optimal_ipv6
|
||
|
||
except Exception as e:
|
||
print(f"执行 CloudflareSpeedTest IPv6 时发生错误: {e}")
|
||
return None
|
||
|
||
def stop(self):
|
||
if self.process and self.process.poll() is None:
|
||
print("正在终止 CloudflareSpeedTest 进程...")
|
||
try:
|
||
if self.process.stdin and not self.process.stdin.closed:
|
||
self.process.stdin.write('\n')
|
||
self.process.stdin.flush()
|
||
self.process.stdin.close()
|
||
except:
|
||
pass
|
||
|
||
try:
|
||
self.process.terminate()
|
||
self.process.wait(timeout=5)
|
||
except subprocess.TimeoutExpired:
|
||
self.process.kill()
|
||
self.process.wait()
|
||
print("CloudflareSpeedTest 进程已终止。")
|
||
|
||
|
||
class IpOptimizerThread(QThread):
|
||
"""用于在后台线程中运行IP优化的类
|
||
|
||
注意：IPv6连接测试功能已迁移至IPv6Manager类，
|
||
本类仅负责IP优化相关功能
|
||
"""
|
||
finished = Signal(str)
|
||
|
||
def __init__(self, url, parent=None, use_ipv6=False):
|
||
super().__init__(parent)
|
||
self.url = url
|
||
self.optimizer = IpOptimizer()
|
||
self.use_ipv6 = use_ipv6
|
||
|
||
def run(self):
|
||
if self.use_ipv6:
|
||
optimal_ip = self.optimizer.get_optimal_ipv6(self.url)
|
||
else:
|
||
optimal_ip = self.optimizer.get_optimal_ip(self.url)
|
||
self.finished.emit(optimal_ip if optimal_ip else "")
|
||
|
||
def stop(self):
|
||
self.optimizer.stop()
|
||
|
||
|
||
if __name__ == '__main__':
|
||
# 用于直接测试此模块
|
||
test_url = "https://speed.cloudflare.com/__down?during=download&bytes=104857600"
|
||
optimizer = IpOptimizer()
|
||
ip = optimizer.get_optimal_ip(test_url)
|
||
if ip:
|
||
print(f"为 {test_url} 找到的最优 IP 是: {ip}")
|
||
else:
|
||
print(f"未能为 {test_url} 找到最优 IP。") |