feat(scan): support set threads used to scan

Author: WillowSauceR

README.md

@@ -11,7 +11,7 @@
 
 ## scan.py
 
-### 使用方法: ``python3 scan.py [目标地址] -i [可选：发包间隔] -p [可选：本地端口] -do [可选：只显示玩家在线数量大于或等于此数值的服务器] -e [在扫描到服务器后立刻执行的cmd命令，需要用双引号包含，具体使用请使用 -h 参数获取帮助] -v6 [使用IPv6协议]``
+### 使用方法: ``python3 scan.py [目标地址] -i [可选：发包间隔] -p [可选：本地端口] -t [可选：使用的线程数，默认为CPU核心数减一] -do [可选：只显示玩家在线数量大于或等于此数值的服务器] -e [在扫描到服务器后立刻执行的cmd命令，需要用双引号包含，具体使用请使用 -h 参数获取帮助] -v6 [使用IPv6协议]``
 
 #### 描述: 扫描IP上的所有BE协议服务器
 

scan.py

@@ -52,7 +52,7 @@ def progerss_monitor(conn):
 
 
 is_recving = False
-def scanner(udp_skt: socket.socket, addr: str, interval: float):
+def scanner(udp_skt: socket.socket, addr: str, interval: float, threads: int):
     global is_recving, prog_mon
     pbar = tqdm(iterable=range(65536), desc="Scaning progress",
                 leave=False, unit="Port", unit_scale=False)
@@ -62,7 +62,7 @@ def scanner(udp_skt: socket.socket, addr: str, interval: float):
         threading.Thread(target=recv_packets, args=(
             udp_skt, pbar), daemon=True).start()
 
-    port_ranges = split_list(65536, mp.cpu_count() - 1)
+    port_ranges = split_list(65536, threads)
 
     parent_conn, child_conn = mp.Pipe()
 
@@ -131,6 +131,8 @@ def exec_cmd_async(cmd: str, variables: dict = [None]) -> None:
                         help="send packet interval. recommand 0.01~0.0001")
     parser.add_argument("-p", "--port", default=random.randint(1024, 65535), type=int,
                         help="local port for send packet")
+    parser.add_argument("-t", "--threads", default=(mp.cpu_count() - 1), type=int,
+                        help="number of threads used to scan")
     parser.add_argument("-do", "--display-online", default=0, type=int,
                         help="only displayed when the number of online players is greater than or equal to this value")
     parser.add_argument("-e", "--exec-cmd", default="", type=str,
@@ -148,8 +150,9 @@ def exec_cmd_async(cmd: str, variables: dict = [None]) -> None:
     display_online = args.display_online
     exec_cmd = args.exec_cmd
     use_ipv6 = args.use_ipv6
+    threads = args.threads
 
     udp_skt = get_udp_socket(local_port, use_ipv6=use_ipv6)
 
     for addr in get_ip_list(addr):
-        scanner(udp_skt, addr, interval)
+        scanner(udp_skt, addr, interval, threads)