OK
This commit is contained in:
@@ -0,0 +1,224 @@
|
||||
import threading
|
||||
import tkinter as tk
|
||||
from tkinter import filedialog, messagebox, ttk
|
||||
|
||||
from core.Function.ip_conflict_fun import ALL_ADAPTERS, MODE_BOTH, MODE_LOCAL, MODE_SUBNET, IpConflictDetector
|
||||
from core.ui.components import Console, Page, action_bar, button, combo, field
|
||||
|
||||
|
||||
class IpConflictTab(Page):
|
||||
def __init__(self, parent):
|
||||
super().__init__(parent, "IP 冲突", "检测本机 IP 是否被占用，并安全扫描网段内 IP/MAC 异常。")
|
||||
self.body.rowconfigure(3, weight=1)
|
||||
|
||||
status = self.section("实时状态", 0, columns=6)
|
||||
self.state = field(status, "状态", 1, 0, "等待", 10)
|
||||
self.adapter_total = field(status, "网卡数", 1, 1, "0", 8)
|
||||
self.current = field(status, "当前对象", 1, 2, "-", 18)
|
||||
self.max_risk = field(status, "最高风险", 1, 3, "正常", 10)
|
||||
self.progress = field(status, "进度", 1, 4, "0/0", 10)
|
||||
self.elapsed = field(status, "耗时", 1, 5, "0.0s", 10)
|
||||
for item in (self.state, self.adapter_total, self.current, self.max_risk, self.progress, self.elapsed):
|
||||
item["entry"].configure(state="disabled")
|
||||
|
||||
params = self.section("检测参数", 1, columns=6)
|
||||
self.adapter = combo(params, "检测网卡", 1, 0, [ALL_ADAPTERS], ALL_ADAPTERS, 22)
|
||||
self.mode = combo(params, "检测模式", 1, 1, [MODE_BOTH, MODE_LOCAL, MODE_SUBNET], MODE_BOTH, 14)
|
||||
self.scan_range = field(params, "扫描范围", 1, 2, "", 30, colspan=2)
|
||||
self.workers = field(params, "并发数", 1, 4, "64", 10)
|
||||
self.timeout = field(params, "超时 ms", 1, 5, "500", 10)
|
||||
self.max_hosts = field(params, "最大地址数", 2, 0, "254", 10)
|
||||
actions = action_bar(params, 3, 6)
|
||||
self.start_btn = button(actions, "开始检测", self.start_detection, "Primary.TButton")
|
||||
self.stop_btn = button(actions, "停止", self.stop_detection, "Danger.TButton")
|
||||
self.refresh_btn = button(actions, "刷新网卡", self.load_adapters, "Secondary.TButton")
|
||||
self.auto_range_btn = button(actions, "自动范围", self.fill_default_range, "Secondary.TButton")
|
||||
self.copy_btn = button(actions, "复制摘要", self.copy_summary, "Secondary.TButton")
|
||||
self.export_btn = button(actions, "导出 CSV", self.export_results, "Secondary.TButton")
|
||||
|
||||
results = self.section("检测结果", 2, columns=1)
|
||||
results.rowconfigure(1, weight=1)
|
||||
results.columnconfigure(0, weight=1)
|
||||
self.results_tree = ttk.Treeview(
|
||||
results,
|
||||
columns=("adapter", "ip", "mac", "conflict", "type", "level", "verdict"),
|
||||
show="headings",
|
||||
height=9,
|
||||
)
|
||||
headings = {
|
||||
"adapter": "网卡",
|
||||
"ip": "IP",
|
||||
"mac": "本机/观察 MAC",
|
||||
"conflict": "冲突 MAC",
|
||||
"type": "证据类型",
|
||||
"level": "风险",
|
||||
"verdict": "判断",
|
||||
}
|
||||
widths = {
|
||||
"adapter": 130,
|
||||
"ip": 150,
|
||||
"mac": 150,
|
||||
"conflict": 180,
|
||||
"type": 110,
|
||||
"level": 90,
|
||||
"verdict": 420,
|
||||
}
|
||||
for column, title in headings.items():
|
||||
self.results_tree.heading(column, text=title)
|
||||
self.results_tree.column(column, width=widths[column], anchor="w")
|
||||
self.results_tree.tag_configure("正常", foreground="#15803d")
|
||||
self.results_tree.tag_configure("可疑", foreground="#b7791f")
|
||||
self.results_tree.tag_configure("冲突风险", foreground="#dc2626")
|
||||
self.results_tree.grid(row=1, column=0, sticky="nsew", padx=18, pady=(0, 18))
|
||||
result_scroll = ttk.Scrollbar(results, orient="vertical", command=self.results_tree.yview)
|
||||
result_scroll.grid(row=1, column=1, sticky="ns", pady=(0, 18))
|
||||
self.results_tree.configure(yscrollcommand=result_scroll.set)
|
||||
|
||||
output = self.section("诊断控制台", 3, columns=1)
|
||||
output.rowconfigure(1, weight=1)
|
||||
output.columnconfigure(0, weight=1)
|
||||
self.console = Console(output, height=12)
|
||||
self.console.grid(row=1, column=0, sticky="nsew", padx=18, pady=(0, 18))
|
||||
|
||||
self.detector = IpConflictDetector(self.write, self.on_task_done, self.update_status, self.add_result)
|
||||
self.after(350, self.load_adapters)
|
||||
|
||||
def write(self, text, tag=None):
|
||||
self.console.write(text, tag)
|
||||
|
||||
def clear(self):
|
||||
self.console.clear()
|
||||
for item in self.results_tree.get_children():
|
||||
self.results_tree.delete(item)
|
||||
self.update_status(
|
||||
{
|
||||
"state": "等待",
|
||||
"total_adapters": 0,
|
||||
"current": "-",
|
||||
"max_risk": "正常",
|
||||
"total_targets": 0,
|
||||
"completed": 0,
|
||||
"elapsed": 0,
|
||||
}
|
||||
)
|
||||
|
||||
def load_adapters(self):
|
||||
def worker():
|
||||
try:
|
||||
values = self.detector.get_adapter_choices()
|
||||
default_range = self.detector.default_scan_range(values[0] if len(values) == 1 else ALL_ADAPTERS)
|
||||
except Exception as exc:
|
||||
values = [ALL_ADAPTERS]
|
||||
default_range = ""
|
||||
self.write(f"读取检测网卡失败: {exc}\n", "warning")
|
||||
self.after(0, lambda: self.apply_adapters(values, default_range))
|
||||
|
||||
threading.Thread(target=worker, daemon=True).start()
|
||||
|
||||
def apply_adapters(self, values, default_range):
|
||||
values = values or [ALL_ADAPTERS]
|
||||
values = list(dict.fromkeys(values))
|
||||
self.adapter["combobox"]["values"] = values
|
||||
if self.adapter["var"].get() not in values:
|
||||
self.adapter["var"].set(values[0])
|
||||
if not self.scan_range["var"].get() and default_range:
|
||||
self.scan_range["var"].set(default_range)
|
||||
|
||||
def fill_default_range(self):
|
||||
try:
|
||||
value = self.detector.default_scan_range(self.adapter["var"].get())
|
||||
if not value:
|
||||
messagebox.showinfo("提示", "未能根据当前网卡生成扫描范围")
|
||||
return
|
||||
self.scan_range["var"].set(value)
|
||||
self.write(f"已生成安全扫描范围: {value}\n", "success")
|
||||
except Exception as exc:
|
||||
messagebox.showwarning("生成失败", str(exc))
|
||||
|
||||
def start_detection(self):
|
||||
try:
|
||||
self.clear()
|
||||
self.start_btn.configure(state="disabled")
|
||||
self.detector.start_detection(self.adapter["var"].get(), self.options())
|
||||
except Exception as exc:
|
||||
self.start_btn.configure(state="normal")
|
||||
messagebox.showwarning("无法开始 IP 冲突检测", str(exc))
|
||||
|
||||
def stop_detection(self):
|
||||
try:
|
||||
self.detector.stop_detection()
|
||||
except Exception as exc:
|
||||
messagebox.showinfo("提示", str(exc))
|
||||
|
||||
def options(self):
|
||||
return {
|
||||
"mode": self.mode["var"].get(),
|
||||
"scan_range": self.scan_range["var"].get(),
|
||||
"workers": self.workers["var"].get(),
|
||||
"timeout_ms": self.timeout["var"].get(),
|
||||
"max_hosts": self.max_hosts["var"].get(),
|
||||
}
|
||||
|
||||
def export_results(self):
|
||||
path = filedialog.asksaveasfilename(
|
||||
title="导出 IP 冲突检测结果",
|
||||
defaultextension=".csv",
|
||||
filetypes=[("CSV 文件", "*.csv")],
|
||||
)
|
||||
if not path:
|
||||
return
|
||||
try:
|
||||
self.detector.export_results(path)
|
||||
self.write(f"已导出结果: {path}\n", "success")
|
||||
except Exception as exc:
|
||||
messagebox.showwarning("导出失败", str(exc))
|
||||
|
||||
def copy_summary(self):
|
||||
text = self.detector.copy_summary()
|
||||
if not text:
|
||||
messagebox.showinfo("提示", "还没有可复制的诊断摘要")
|
||||
return
|
||||
self.clipboard_clear()
|
||||
self.clipboard_append(text)
|
||||
self.write("已复制诊断摘要到剪贴板\n", "success")
|
||||
|
||||
def update_status(self, stats):
|
||||
def apply():
|
||||
total = stats.get("total_targets", 0)
|
||||
done = stats.get("completed", 0)
|
||||
values = [
|
||||
(self.state, stats.get("state", "等待")),
|
||||
(self.adapter_total, str(stats.get("total_adapters", 0))),
|
||||
(self.current, stats.get("current") or "-"),
|
||||
(self.max_risk, stats.get("max_risk", "正常")),
|
||||
(self.progress, f"{done}/{total}" if total else str(done)),
|
||||
(self.elapsed, f"{stats.get('elapsed', 0):.1f}s"),
|
||||
]
|
||||
for item, value in values:
|
||||
item["entry"].configure(state="normal")
|
||||
item["var"].set(value)
|
||||
item["entry"].configure(state="disabled")
|
||||
|
||||
self.after(0, apply)
|
||||
|
||||
def add_result(self, row):
|
||||
def apply():
|
||||
self.results_tree.insert(
|
||||
"",
|
||||
"end",
|
||||
values=(
|
||||
row.get("adapter", ""),
|
||||
row.get("ip", ""),
|
||||
row.get("mac", ""),
|
||||
row.get("conflict_macs", ""),
|
||||
row.get("evidence_type", ""),
|
||||
row.get("risk_level", ""),
|
||||
row.get("verdict", ""),
|
||||
),
|
||||
tags=(row.get("risk_level", ""),),
|
||||
)
|
||||
|
||||
self.after(0, apply)
|
||||
|
||||
def on_task_done(self):
|
||||
self.after(0, lambda: self.start_btn.configure(state="normal"))
|
||||
Reference in New Issue
Block a user