first commit

test/port.py

@@ -0,0 +1,76 @@
+import socket
+import threading
+from queue import Queue
+import time
+
+# 定义要检查的IP范围和端口
+IP_BASE = "192.168.10."
+START_IP = 1
+END_IP = 255
+TARGET_PORT = 5555
+
+# 线程数量，可根据需要调整
+THREAD_COUNT = 30
+
+
+def check_port(ip, port, result_queue):
+    """检查指定IP的端口是否开放"""
+    try:
+        # 创建socket对象
+        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        # 设置超时时间为1秒
+        sock.settimeout(1)
+        # 尝试连接
+        result = sock.connect_ex((ip, port))
+        if result == 0:
+            result_queue.put(ip)
+        sock.close()
+    except Exception:
+        pass
+
+
+def worker(ip_queue, result_queue):
+    """工作线程，从队列中获取IP并检查端口"""
+    while not ip_queue.empty():
+        ip = ip_queue.get()
+        check_port(ip, TARGET_PORT, result_queue)
+        ip_queue.task_done()
+
+
+if __name__ == "__main__":
+    start_time = time.time()
+    print(f"开始扫描 {IP_BASE}{START_IP} 到 {IP_BASE}{END_IP} 的 {TARGET_PORT} 端口...")
+
+    # 创建IP队列
+    ip_queue = Queue()
+    for i in range(START_IP, END_IP + 1):
+        ip = f"{IP_BASE}{i}"
+        ip_queue.put(ip)
+
+    # 创建存储开放端口IP的队列
+    result_queue = Queue()
+
+    # 创建并启动线程
+    threads = []
+    for _ in range(THREAD_COUNT):
+        thread = threading.Thread(target=worker, args=(ip_queue, result_queue))
+        thread.start()
+        threads.append(thread)
+
+    # 等待所有IP都被处理
+    ip_queue.join()
+
+    # 收集结果
+    open_ips = []
+    while not result_queue.empty():
+        open_ips.append(result_queue.get())
+
+    # 排序结果
+    open_ips.sort(key=lambda x: int(x.split(".")[-1]))
+
+    # 输出结果
+    print("\n扫描完成！")
+    print(f"耗时: {time.time() - start_time:.2f} 秒")
+    print(f"发现 {len(open_ips)} 个设备的 {TARGET_PORT} 端口开放：")
+    for ip in open_ips:
+        print(f"  {ip}:{TARGET_PORT}")