13

core/ui/tab_telnet.py

@@ -0,0 +1,106 @@
+import tkinter as tk
+from tkinter import ttk, filedialog, messagebox, scrolledtext
+
+from core.ui.basic_ui import BasicUI
+from core.Function.telnet_fun import PortScanner
+
+import logging
+logger = logging.getLogger(__name__)
+
+class TelnetTab(ttk.Frame, BasicUI):
+    def __init__(self, parent):
+        super().__init__(parent)   
+        self.telnet_ui()
+        self.telnet_fun = PortScanner(self.result_box)
+
+    def telnet_ui(self):
+        """端口扫描界面布局"""
+        self.create_assigntelnet_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)
+
+        #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)
+    
+    def create_batchtelnet_section(self):
+        # 区域标签
+        frame = ttk.LabelFrame(self, text="批量端口连接测试")
+        frame.pack(side='top', fill='x', padx=10, pady=5)
+
+        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)
+
+    def create_outputping_section(self):
+        # 区域标签
+        frame = ttk.LabelFrame(self, text="端口扫描结果输出")
+        frame.pack(side='top', fill='x', padx=10, pady=5)
+
+        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("输入错误", "请输入目标端口！")
+            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)
+
+    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("输入错误", "请输入目标端口！")
+            return
+        try:
+            port_begin = int(self.port_begin)
+            port_end = int(self.port_end)
+        except ValueError:
+            messagebox.showwarning("输入错误", "端口必须是整数！")
+            return
+        if port_begin < 1 or port_end > 65535:
+            messagebox.showwarning("输入错误", "起始端口最小为1，结束端口最大为65535！")
+            return
+        logger.info(f"开始测试{self.IP} 的 {port_begin} 到 {port_end} 端口连接情况")
+        self.batchTelnet_start['btn'].config(state='disabled')
+        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.batchTelnet_start['btn'].config(state='normal')
+
+    def batchTelnet_callback(self):
+        self.batchTelnet_start['btn'].config(state='normal')
+
+
+
+
+   
+        

core/ui/ui_main.py

@@ -3,6 +3,7 @@ from tkinter import ttk, filedialog, messagebox
 
 from core.ui.basic_ui import BasicUI
 from core.ui.tab_ping import PingTab
+from core.ui.tab_telnet import TelnetTab
 
 import logging
 logger = logging.getLogger(__name__)
@@ -19,6 +20,7 @@ class MainUI(BasicUI):
         # 实例化各个功能页
         self.tabs = {
             "Ping测试": PingTab(self.tab_control),
+            "端口扫描服务端": TelnetTab(self.tab_control),
         }
         # 添加到 Notebook
         for name, tab in self.tabs.items():