70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
import threading
|
|
from typing import Callable, Optional
|
|
|
|
from core.Function.common import popen_hidden, validate_host
|
|
|
|
|
|
OutputCallback = Callable[[str, Optional[str]], None]
|
|
DoneCallback = Callable[[], None]
|
|
|
|
|
|
class TracertFun:
|
|
def __init__(self, output: OutputCallback, done: Optional[DoneCallback] = None):
|
|
self.output = output
|
|
self.done = done or (lambda: None)
|
|
self.process = None
|
|
self.stop_event = threading.Event()
|
|
self.worker = None
|
|
|
|
def start_tracert(self, target: str, max_hops: int = 20, timeout_ms: int = 800) -> None:
|
|
target = validate_host(target)
|
|
max_hops = max(1, min(int(max_hops), 64))
|
|
timeout_ms = max(100, min(int(timeout_ms), 10000))
|
|
|
|
if self.is_running():
|
|
raise RuntimeError("路由追踪正在运行，请先停止当前任务")
|
|
|
|
self.stop_event.clear()
|
|
command = ["tracert", "-d", "-w", str(timeout_ms), "-h", str(max_hops), target]
|
|
self.output(f"开始路由追踪: {target}，最大 {max_hops} 跳，超时 {timeout_ms}ms\n\n", "muted")
|
|
self.worker = threading.Thread(target=self._run, args=(command,), daemon=True)
|
|
self.worker.start()
|
|
|
|
def _run(self, command) -> None:
|
|
try:
|
|
self.process = popen_hidden(command)
|
|
if not self.process.stdout:
|
|
return
|
|
for line in self.process.stdout:
|
|
if self.stop_event.is_set():
|
|
break
|
|
self.output(line, None)
|
|
except Exception as exc:
|
|
self.output(f"\n路由追踪失败: {exc}\n", "error")
|
|
finally:
|
|
if self.process:
|
|
try:
|
|
self.process.terminate()
|
|
except Exception:
|
|
pass
|
|
self.process = None
|
|
if self.stop_event.is_set():
|
|
self.output("\n路由追踪已停止\n", "warning")
|
|
else:
|
|
self.output("\n路由追踪完成\n", "success")
|
|
self.done()
|
|
|
|
def stop_tracert(self) -> None:
|
|
if not self.is_running():
|
|
raise RuntimeError("当前没有正在运行的路由追踪")
|
|
self.stop_event.set()
|
|
if self.process:
|
|
try:
|
|
self.process.terminate()
|
|
except Exception:
|
|
pass
|
|
self.output("\n正在停止路由追踪...\n", "warning")
|
|
|
|
def is_running(self) -> bool:
|
|
return bool(self.worker and self.worker.is_alive())
|