11

core/Function/ping_fun.py

@@ -0,0 +1,81 @@
+import tkinter as tk
+from tkinter import scrolledtext, messagebox
+import subprocess
+import platform
+import threading
+import re
+
+class PingFun:
+    def __init__(self, result_box: scrolledtext.ScrolledText):
+        self.result_box = result_box
+
+        # Ping 状态和统计
+        self.process = None # 子进程对象
+        self.stop_flag = False # 停止标志
+        self.sent = 0   # 发送的包数
+        self.received = 0   # 接收的包数
+        self.rtts = []  # 存储延迟值
+
+    def strat_ping(self, host, local_ip=None, callback=None):
+        """开始 ping"""
+        self.callback = callback
+
+        # 清空显示框和统计数据
+        self.result_box.delete('1.0', tk.END)
+        self.sent = 0
+        self.received = 0
+        self.rtts.clear()
+        self.stop_flag = False
+
+        command = ['ping', host, '-t']
+        if local_ip:
+            command += ['-S', local_ip]
+
+        threading.Thread(target=self.ping, args=(command,), daemon=True).start()
+
+    def stop_ping(self):
+        """手动停止 ping"""
+        if self.process:
+            self.process.terminate()  # 立即终止子进程
+            self.result_box.insert(tk.END, "\nPing 已手动停止。\n")
+            self.result_box.see(tk.END)
+            self.process = None
+            self.show_statistics()
+        
+    def ping(self, command):
+        """执行 ping 命令并处理输出"""
+        rtt_pattern = re.compile(r'时间[=<](\d+)ms', re.IGNORECASE)
+        try:
+            self.process = subprocess.Popen(command, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, universal_newlines=True)
+            for line in iter(self.process.stdout.readline, ''):
+                if self.stop_flag:
+                    break
+                if line:
+                    self.result_box.insert(tk.END, line)
+                    self.result_box.see(tk.END)
+                    self.sent += 1
+                    # 解析延迟
+                    match = rtt_pattern.search(line)
+                    if match:
+                        self.received += 1
+                        self.rtts.append(float(match.group(1)))
+        except Exception as e:
+            self.result_box.insert(tk.END, f"Ping 失败: {e}\n")
+        finally:
+            # 清理进程对象
+            self.process = None
+            if self.callback:
+                self.callback()
+
+    def show_statistics(self):
+        """显示统计信息"""
+        if self.sent == 0:
+            return
+        loss = (self.sent - self.received) / self.sent * 100
+        stats = f"\n==== Ping 统计 ====\n" \
+                f"发送: {self.sent}，接收: {self.received}，丢包率: {loss:.2f}%\n"
+        if self.rtts:
+            stats += f"最小延迟: {min(self.rtts)} ms，最大延迟: {max(self.rtts)} ms，平均延迟: {sum(self.rtts)/len(self.rtts):.2f} ms\n"
+        self.result_box.insert(tk.END, stats)
+        self.result_box.see(tk.END)
+