This commit is contained in:
qsc
2026-06-06 01:57:43 +08:00
parent 2a97e1df17
commit 129dc41208
17 changed files with 2582 additions and 1048 deletions
+303 -110
View File
@@ -1,135 +1,328 @@
import tkinter as tk
from tkinter import ttk, filedialog, messagebox, scrolledtext
from tkinter import filedialog, messagebox, ttk
from core.ui.basic_ui import BasicUI
from core.Function.telnet_fun import PortScanner
from core.ui.components import Console, Page, action_bar, button, combo, field
import logging
logger = logging.getLogger(__name__)
class TelnetTab(ttk.Frame, BasicUI):
PORT_PRESETS = {
"常用端口": "21,22,23,25,53,80,110,139,143,443,445,1433,3306,3389,5432,6379,8080,8443,9200",
"Web 服务": "80,443,8000-8010,8080,8081,8443,8888,9000",
"数据库": "1433,1521,3306,5432,6379,9200,27017",
"远程管理": "22,23,3389,5900,5985,5986",
"邮件服务": "25,110,143,465,587,993,995",
"全端口": "1-65535",
}
class TelnetTab(Page):
def __init__(self, parent):
super().__init__(parent)
self.telnet_ui()
self.telnet_fun = PortScanner(self.result_box)
super().__init__(parent, "端口扫描", "单端口测试、端口扫描、批量主机巡检、服务识别与结果导出。")
self.body.rowconfigure(3, weight=1)
def telnet_ui(self):
"""端口扫描界面布局"""
self.create_assigntelnet_section()
self.create_listtelnet_section()
self.create_batchtelnet_section()
self.create_outputping_section()
# --------------------------------------UI界面布局函数--------------------------------------
def create_assigntelnet_section(self):
# 区域标签
frame = ttk.LabelFrame(self, text="指定端口连接测试")
frame.pack(side='top', fill='x', padx=10, pady=5)
status = self.section("实时状态", 0, columns=7)
self.state = field(status, "状态", 1, 0, "等待", 10)
self.total = field(status, "总数", 1, 1, "0", 8)
self.scanned = field(status, "已扫", 1, 2, "0", 8)
self.open_count = field(status, "开放", 1, 3, "0", 8)
self.closed_count = field(status, "关闭", 1, 4, "0", 8)
self.timeout_count = field(status, "超时", 1, 5, "0", 8)
self.progress = field(status, "进度 / 耗时", 1, 6, "0.0% / 0.0s", 16)
for item in (self.state, self.total, self.scanned, self.open_count, self.closed_count, self.timeout_count, self.progress):
item["entry"].configure(state="disabled")
#self.entry_assignIP_A = self.add_input(frame, "本地IP", row=0, col=0)
self.assignTelnet_IP = self.add_input(frame, "目标IP", row=0, col=0, inivar="202.89.233.100")
self.assignTelnet_port = self.add_input(frame, "目标端口", row=0, col=1, inivar="443", entry_width=10)
self.assignTelnet_test = self.add_button(frame, "测试", row=0, col=2, command=self.btn_assignTelnet_test)
modes = self.section("扫描模式", 1, columns=1)
self.tabs = ttk.Notebook(modes)
self.tabs.grid(row=1, column=0, sticky="ew", padx=18, pady=(0, 18))
self.single_tab = ttk.Frame(self.tabs, style="Panel.TFrame")
self.scan_tab = ttk.Frame(self.tabs, style="Panel.TFrame")
self.batch_tab = ttk.Frame(self.tabs, style="Panel.TFrame")
self.tabs.add(self.single_tab, text="单端口测试")
self.tabs.add(self.scan_tab, text="端口扫描")
self.tabs.add(self.batch_tab, text="批量主机扫描")
self.build_single_tab()
self.build_scan_tab()
self.build_batch_tab()
def create_listtelnet_section(self):
# 区域标签
frame = ttk.LabelFrame(self, text="批量端口连接测试")
frame.pack(side='top', fill='x', padx=10, pady=5)
results = self.section("结果列表", 2, columns=1)
results.rowconfigure(1, weight=1)
results.columnconfigure(0, weight=1)
self.results_tree = ttk.Treeview(
results,
columns=("host", "resolved_ip", "port", "service", "status", "latency", "banner"),
show="headings",
height=8,
)
headings = {
"host": "目标",
"resolved_ip": "解析 IP",
"port": "端口",
"service": "服务",
"status": "状态",
"latency": "延迟",
"banner": "Banner / 错误",
}
widths = {
"host": 150,
"resolved_ip": 130,
"port": 70,
"service": 100,
"status": 80,
"latency": 80,
"banner": 320,
}
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("open", foreground="#15803d")
self.results_tree.tag_configure("timeout", foreground="#b7791f")
self.results_tree.tag_configure("unreachable", foreground="#b7791f")
self.results_tree.tag_configure("dns_error", foreground="#dc2626")
self.results_tree.tag_configure("error", 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)
self.listtelnet_IP = self.add_input(frame, "目标IP", row=0, col=0, inivar="202.89.233.100")
self.listtelnet_ports = self.add_input(frame, "端口列表", row=0, col=1, inivar="21,22,23,25,80,110,143,443,1433,3306,3389", entry_width=40)
self.listtelnet_start = self.add_button(frame, "开始", row=0, col=2, command=self.btn_listTelnet_start)
self.listtelnet_stop = self.add_button(frame, "停止", row=0, col=3, command=self.btn_listTelnet_stop)
def create_batchtelnet_section(self):
# 区域标签
frame = ttk.LabelFrame(self, text="批量端口连接测试")
frame.pack(side='top', fill='x', padx=10, pady=5)
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.batchTelnet_IP = self.add_input(frame, "目标IP", row=0, col=0, inivar="202.89.233.100")
self.batchTelnet_port_begin = self.add_input(frame, "起始端口", row=0, col=1, inivar="100", entry_width=10)
self.batchTelnet_port_end = self.add_input(frame, "结束端口", row=0, col=2, inivar="1000", entry_width=10)
self.batchTelnet_start = self.add_button(frame, "开始", row=0, col=3, command=self.btn_batchTelnet_start)
self.batchTelnet_stop = self.add_button(frame, "停止", row=0, col=4, command=self.btn_batchTelnet_stop)
self.scanner = PortScanner(self.write, self.on_task_done, self.update_status, self.add_result)
def create_outputping_section(self):
# 区域标签
frame = ttk.LabelFrame(self, text="端口扫描结果输出")
frame.pack(side='top', fill='x', padx=10, pady=5)
def build_single_tab(self):
for col in range(5):
self.single_tab.columnconfigure(col, weight=1)
self.result_box = scrolledtext.ScrolledText(frame, width=100, height=20)
self.result_box.pack(pady=10)
# --------------------------------------按钮回调函数--------------------------------------
def btn_assignTelnet_test(self):
self.IP = self.assignTelnet_IP['var'].get()
self.port = self.assignTelnet_port['var'].get()
if not self.IP:
messagebox.showwarning("输入错误", "请输入目标IP或域名!")
return
if not self.port:
messagebox.showwarning("输入错误", "请输入目标端口!")
self.single_host = field(self.single_tab, "目标 IP / 域名", 0, 0, "127.0.0.1", 26)
self.single_port = field(self.single_tab, "目标端口", 0, 1, "443", 12)
self.single_timeout = field(self.single_tab, "超时 ms", 0, 2, "1000", 10)
self.single_banner_var = tk.BooleanVar(value=False)
self._check(self.single_tab, "Banner 探测", self.single_banner_var, 0, 3)
actions = action_bar(self.single_tab, 1, 5)
self.single_btn = button(actions, "测试连接", self.test_single, "Primary.TButton")
self.single_export_btn = button(actions, "导出 CSV", self.export_results, "Secondary.TButton")
def build_scan_tab(self):
for col in range(6):
self.scan_tab.columnconfigure(col, weight=1)
self.scan_host = field(self.scan_tab, "目标 IP / 域名", 0, 0, "127.0.0.1", 26)
self.scan_ports = field(self.scan_tab, "端口列表 / 范围", 0, 1, PORT_PRESETS["常用端口"], 44, colspan=2)
self.scan_preset = combo(self.scan_tab, "预设", 0, 3, list(PORT_PRESETS), "常用端口", 16)
self.scan_timeout = field(self.scan_tab, "超时 ms", 0, 4, "800", 10)
self.scan_workers = field(self.scan_tab, "并发数", 0, 5, "128", 10)
self.scan_show_closed_var = tk.BooleanVar(value=False)
self.scan_banner_var = tk.BooleanVar(value=False)
self._check(self.scan_tab, "显示关闭端口", self.scan_show_closed_var, 1, 0)
self._check(self.scan_tab, "Banner 探测", self.scan_banner_var, 1, 1)
self.scan_preset["combobox"].bind("<<ComboboxSelected>>", lambda _event: self.apply_preset(self.scan_preset, self.scan_ports))
actions = action_bar(self.scan_tab, 2, 6)
self.scan_start_btn = button(actions, "开始扫描", self.scan_ports_action, "Primary.TButton")
self.scan_stop_btn = button(actions, "停止", self.stop_scan, "Danger.TButton")
self.scan_copy_btn = button(actions, "复制开放端口", self.copy_open_ports, "Secondary.TButton")
self.scan_export_btn = button(actions, "导出 CSV", self.export_results, "Secondary.TButton")
def build_batch_tab(self):
for col in range(6):
self.batch_tab.columnconfigure(col, weight=1)
self.batch_hosts = field(self.batch_tab, "目标列表 / CIDR / IP 段", 0, 0, "192.168.1.1-254", 38, colspan=2)
self.batch_ports = field(self.batch_tab, "端口列表 / 范围", 0, 2, "22,80,443,3389", 32, colspan=2)
self.batch_preset = combo(self.batch_tab, "预设", 0, 4, list(PORT_PRESETS), "远程管理", 16)
self.batch_timeout = field(self.batch_tab, "超时 ms", 0, 5, "800", 10)
self.batch_workers = field(self.batch_tab, "并发数", 1, 0, "128", 10)
self.batch_show_closed_var = tk.BooleanVar(value=False)
self.batch_banner_var = tk.BooleanVar(value=False)
self._check(self.batch_tab, "显示关闭端口", self.batch_show_closed_var, 1, 1)
self._check(self.batch_tab, "Banner 探测", self.batch_banner_var, 1, 2)
self.batch_preset["combobox"].bind("<<ComboboxSelected>>", lambda _event: self.apply_preset(self.batch_preset, self.batch_ports))
actions = action_bar(self.batch_tab, 2, 6)
self.batch_start_btn = button(actions, "批量扫描", self.batch_scan_action, "Primary.TButton")
self.batch_stop_btn = button(actions, "停止", self.stop_scan, "Danger.TButton")
self.import_hosts_btn = button(actions, "导入目标", self.import_hosts, "Secondary.TButton")
self.batch_copy_btn = button(actions, "复制开放端口", self.copy_open_ports, "Secondary.TButton")
self.batch_export_btn = button(actions, "导出 CSV", self.export_results, "Secondary.TButton")
def _check(self, parent, text, variable, row, column):
frame = ttk.Frame(parent, style="Panel.TFrame")
frame.grid(row=row, column=column, sticky="ew", padx=18, pady=(22, 14))
ttk.Checkbutton(frame, text=text, variable=variable).pack(anchor="w")
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": 0,
"scanned": 0,
"open": 0,
"closed": 0,
"timeout": 0,
"progress": 0,
"elapsed": 0,
}
)
def test_single(self):
try:
self.clear()
self.single_btn.configure(state="disabled")
self.scanner.test_connect(
self.single_host["var"].get(),
int(self.single_port["var"].get()),
options={
"timeout_ms": self.single_timeout["var"].get(),
"workers": 1,
"show_closed": True,
"banner_probe": self.single_banner_var.get(),
},
)
except Exception as exc:
self.single_btn.configure(state="normal")
messagebox.showwarning("无法测试端口", str(exc))
def scan_ports_action(self):
try:
self.clear()
self._set_scan_buttons("disabled")
self.scanner.start_scan(self.scan_host["var"].get(), self.scan_ports["var"].get(), self.scan_options())
except Exception as exc:
self._set_scan_buttons("normal")
messagebox.showwarning("无法扫描端口", str(exc))
def batch_scan_action(self):
try:
self.clear()
self._set_scan_buttons("disabled")
self.scanner.start_batch_scan(self.batch_hosts["var"].get(), self.batch_ports["var"].get(), self.batch_options())
except Exception as exc:
self._set_scan_buttons("normal")
messagebox.showwarning("无法批量扫描", str(exc))
def stop_scan(self):
try:
self.scanner.stop_scan()
except Exception as exc:
messagebox.showinfo("提示", str(exc))
def scan_options(self):
return {
"timeout_ms": self.scan_timeout["var"].get(),
"workers": self.scan_workers["var"].get(),
"show_closed": self.scan_show_closed_var.get(),
"banner_probe": self.scan_banner_var.get(),
}
def batch_options(self):
return {
"timeout_ms": self.batch_timeout["var"].get(),
"workers": self.batch_workers["var"].get(),
"show_closed": self.batch_show_closed_var.get(),
"banner_probe": self.batch_banner_var.get(),
}
def apply_preset(self, preset_item, ports_item):
ports_item["var"].set(PORT_PRESETS.get(preset_item["var"].get(), PORT_PRESETS["常用端口"]))
def import_hosts(self):
path = filedialog.askopenfilename(
title="导入扫描目标",
filetypes=[("文本文件", "*.txt *.csv"), ("所有文件", "*.*")],
)
if not path:
return
try:
port = int(self.port)
except ValueError:
messagebox.showwarning("输入错误", "端口必须是整数!")
return
if port < 1 or port > 65535:
messagebox.showwarning("输入错误", "起始端口最小为1,结束端口最大为65535!")
return
logger.info(f"测试{self.IP}{port} 端口连接情况")
self.telnet_fun.test_connect(self.IP, port)
with open(path, "r", encoding="utf-8-sig") as file:
items = []
for line in file:
items.extend(part.strip() for part in line.replace("", ",").split(",") if part.strip())
self.batch_hosts["var"].set(",".join(items))
self.write(f"已导入 {len(items)} 个目标/表达式\n", "success")
except Exception as exc:
messagebox.showwarning("导入失败", str(exc))
def btn_batchTelnet_start(self):
self.IP = self.batchTelnet_IP['var'].get()
self.port_begin = self.batchTelnet_port_begin['var'].get()
self.port_end = self.batchTelnet_port_end['var'].get()
if not self.IP:
messagebox.showwarning("输入错误", "请输入目标IP或域名!")
return
if not self.port_begin or not self.port_end:
messagebox.showwarning("输入错误", "请输入目标端口!")
def export_results(self):
path = filedialog.asksaveasfilename(
title="导出端口扫描结果",
defaultextension=".csv",
filetypes=[("CSV 文件", "*.csv")],
)
if not path:
return
try:
port_begin = int(self.port_begin)
port_end = int(self.port_end)
except ValueError:
messagebox.showwarning("输入错误", "端口必须是整数!")
self.scanner.export_results(path)
self.write(f"已导出结果: {path}\n", "success")
except Exception as exc:
messagebox.showwarning("导出失败", str(exc))
def copy_open_ports(self):
text = self.scanner.open_ports_summary()
if not text:
messagebox.showinfo("提示", "没有可复制的开放端口")
return
if port_begin < 1 or port_end > 65535:
messagebox.showwarning("输入错误", "起始端口最小为1,结束端口最大为65535!")
return
logger.info(f"开始测试{self.IP}{port_begin}{port_end} 端口连接情况")
self.telnet_fun.start_range_scan(self.IP, port_begin, port_end)
def btn_batchTelnet_stop(self):
logger.info(f"停止测试{self.IP} 的端口连接情况")
self.telnet_fun.stop_scan()
self.clipboard_clear()
self.clipboard_append(text)
self.write("已复制开放端口汇总到剪贴板\n", "success")
def btn_listTelnet_start(self):
self.IP = self.listtelnet_IP['var'].get()
self.ports = self.listtelnet_ports['var'].get()
if not self.ports:
messagebox.showwarning("输入错误", "请输入端口号或范围!")
return
def update_status(self, stats):
def apply():
values = [
(self.state, stats.get("state", "等待")),
(self.total, str(stats.get("total", 0))),
(self.scanned, str(stats.get("scanned", 0))),
(self.open_count, str(stats.get("open", 0))),
(self.closed_count, str(stats.get("closed", 0))),
(self.timeout_count, str(stats.get("timeout", 0))),
(self.progress, f"{stats.get('progress', 0):.1f}% / {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")
# 支持格式:例如 "22,80,443,8080"
try:
ports = [int(p.strip()) for p in self.ports.split(',') if p.strip()]
except ValueError:
messagebox.showwarning("输入错误", "端口必须为整数,用逗号分隔")
return
self.after(0, apply)
logger.info(f"开始测试{self.IP}{ports} 端口连接情况")
self.telnet_fun.start_list_scan(self.IP, ports)
def add_result(self, row):
def apply():
port = row.get("port") or "-"
latency = f"{row.get('latency_ms', 0):.1f} ms" if row.get("latency_ms") else "-"
detail = row.get("banner") or row.get("error") or ""
self.results_tree.insert(
"",
"end",
values=(
row.get("host", ""),
row.get("resolved_ip", ""),
port,
row.get("service", ""),
row.get("status_text", ""),
latency,
detail,
),
tags=(row.get("status", ""),),
)
def btn_listTelnet_stop(self):
logger.info(f"停止测试{self.IP} 的端口连接情况")
self.telnet_fun.stop_scan()
self.after(0, apply)
def _set_scan_buttons(self, state):
self.scan_start_btn.configure(state=state)
self.batch_start_btn.configure(state=state)
def on_task_done(self):
self.after(
0,
lambda: (
self.single_btn.configure(state="normal"),
self._set_scan_buttons("normal"),
),
)