add options, scan: --exec-cmd, send: --quiet

Author: WillowSauceR

README.md

@@ -2,19 +2,21 @@
 
 ## 直接在 [Releases](https://github.com/WillowSauceR/BedrockProtocolTools/releases/latest) 页面下载最新的版本，然后解压打开其中的exe文件进行使用
 
-## 或者使用 `git clone https://github.com/WillowSauceR/BedrockProtocolTools`将本项目克隆到本地或直接下载本项目，然后使用python3来运行
+## 或者使用 ``git clone https://github.com/WillowSauceR/BedrockProtocolTools``将本项目克隆到本地或直接下载本项目，然后使用python3来运
+
+## 行每个工具都可以单独使用参数 ``-h``来获取英文帮助，如 ``python3 scan.py -h``
 
 ## scan.py
 
-### 使用方法: `python3 scan.py [目标地址] -i [可选：发包间隔] -p [可选：本地端口] -do [可选：只显示玩家在线数量大于或等于此数值的服务器]`
+### 使用方法: ``python3 scan.py [目标地址] -i [可选：发包间隔] -p [可选：本地端口] -do [可选：只显示玩家在线数量大于或等于此数值的服务器] -e [在扫描到服务器后立刻执行的cmd命令，需要用双引号包含，具体使用请使用 -h 参数获取帮助]``
 
 #### 描述: 扫描IP上的所有BE协议服务器
 
 #### 注意：目标地址可以填域名，IP或者IP段。如 mc.163.com，11.4.5.14，191.191.81-255.0-255
 
 ## send.py
 
-### 使用方法: `python3 send.py [目标地址] [载体包文件] -p [可选：端口：默认19132] -l [可选：次数，默认1] -i [可选：间隔:秒，默认10] -d(自动显示MOTD) -pu(使用代理) -pc [代理的国家，如cn, ru, us]`
+### 使用方法: ``python3 send.py [目标地址] [载体包文件] -p [可选：端口：默认19132] -l [可选：次数，默认1] -i [可选：间隔:秒，默认10] -d(自动显示MOTD) -pu(使用代理) -pc [代理的国家，如cn, ru, us] -q [安静模式，不输出任何日志]``
 
 #### 描述: 发包复现工具, 需要一个内含byte数据的文件，一般为.dmp后缀名
 
@@ -24,6 +26,6 @@
 
 ## motd.py
 
-### 使用方法: `python3 motd.py [目标地址] [端口] -t [可选：超时时间]`
+### 使用方法: ``python3 motd.py [目标地址] [端口] -t [可选：超时时间]``
 
 #### 描述: motd一个BE服务器，支持自动解析返回的数据

api.py

@@ -6,18 +6,18 @@
 MOTD_PKT = b'\x01\x00\x00\x00\x00$\r\x12\xd3\x00\xff\xff\x00\xfe\xfe\xfe\xfe\xfd\xfd\xfd\xfd\x124Vx\n'
 
 
-def getLocalHostIP():
-    localHostIP = socket.gethostbyname(socket.gethostname())
+def get_local_host_ip():
+    local_host_ip = socket.gethostbyname(socket.gethostname())
     try:
         s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
         s.connect(('8.8.8.8', 80))
-        localHostIP = s.getsockname()[0]
+        local_host_ip = s.getsockname()[0]
     finally:
         s.close()
-    return localHostIP
+    return local_host_ip
 
 
-def getTime():
+def get_time():
     return time.strftime('%H:%M:%S')
 
 
@@ -112,7 +112,7 @@ def parse_raw_pkt(pkt):
     try:
         infos = {"motd": decode_unicode(infos[1]), "version_id": infos[2], "version": infos[3], "online": infos[4],
                  "max_player": infos[5], "unique_id": infos[6], "map": decode_unicode(infos[7]), "gamemode": infos[8],
-                 "source_port_v4": infos[10], "source_port_v6": infos[11], "addr": f"{addr[0]}:{addr[1]}"}
+                 "source_port_v4": infos[10], "source_port_v6": infos[11], "ip": addr[0], "addr": f"{addr[0]}:{addr[1]}"}
     except IndexError:
         return None, addr
     if infos["source_port_v4"] != str(addr[1]):

scan.py

@@ -4,13 +4,12 @@
 import time
 import traceback
 import argparse
+import os
 import multiprocessing as mp
 
 from tqdm import tqdm
 
-from api import MOTD_PKT, getTime, parse_raw_pkt, get_ip_list, get_udp_socket
-
-scan_res = []
+from api import MOTD_PKT, get_time, parse_raw_pkt, get_ip_list, get_udp_socket
 
 
 def split_list(total: int, num_splits: int) -> list:
@@ -91,7 +90,7 @@ def recv_packets(udp_skt, pbar):
 
             server_count += 1
 
-            values = [f"[Time   ] {getTime()}",
+            values = [f"[Time   ] {get_time()}",
                       f"[Address] {infos['addr']}",
                       f"[MotdInf] {infos['motd']}",
                       f"[Version] {infos['version']}/{infos['version_id']}",
@@ -101,7 +100,8 @@ def recv_packets(udp_skt, pbar):
                       ""]
             if int(infos['online']) >= display_online:
                 pbar.write("\n".join(values))
-            scan_res.append(values)
+            if exec_cmd:
+                threading.Thread(target=exec_cmd_async, args=(exec_cmd, infos), daemon=True).start()
         except socket.timeout:
             continue
         except ConnectionResetError:
@@ -116,6 +116,12 @@ def recv_packets(udp_skt, pbar):
             continue
 
 
+def exec_cmd_async(cmd: str, variables: dict = [None]) -> None:
+    for key in variables:
+        cmd = cmd.replace("{"+key+"}", variables[key])
+    os.system(cmd)
+
+
 if __name__ == "__main__":
     mp.freeze_support()
     parser = argparse.ArgumentParser()
@@ -126,13 +132,18 @@ def recv_packets(udp_skt, pbar):
                         help="local port for send packet")
     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,
+                        help="the cmd command that is executed immediately after scanning the server can use {var} as a variable, "
+                        + "the command needs to be enclosed in double quotes, the available variables are: motd, version_id, "
+                        + "version, online, max_player, unique_id, map, gamemode, source_port_v4, source_port_v6, addr and ip")
 
     args, unparsed = parser.parse_known_args()
 
     addr = args.addr
     interval = args.interval
     local_port = args.port
     display_online = args.display_online
+    exec_cmd = args.exec_cmd
 
     udp_skt = get_udp_socket(local_port)
 

send.py

@@ -9,21 +9,21 @@
 from api import get_udp_socket, log
 
 
-def getProxy() -> list:
+def get_proxy() -> list:
     try:
-        log(f"Trying to get proxy data from api...")
+        log(f"Trying to get proxy data from api...", quiet=quiet_mode)
         proxy_info = requests.get(
             f"https://www.proxyscan.io/api/proxy?last_check=3600&uptime=&ping=&limit=1&type=socks5&format=json&country={proxyCountry}").json()
         proxy_addr = proxy_info[0]['Ip']
         proxy_port = proxy_info[0]['Port']
         return proxy_addr, int(proxy_port)
     except:
         if not os.path.exists(r"socks5.txt"):
-            log(f"Request api failed. Downloading proxy list...")
+            log(f"Request api failed. Downloading proxy list...", quiet=quiet_mode)
             wget.download(
                 "https://raw.githubusercontent.com/ShiftyTR/Proxy-List/master/socks5.txt")
             print()
-            log(f"Proxy list downloaded!")
+            log(f"Proxy list downloaded!", quiet=quiet_mode)
         with open("socks5.txt", "r") as file:
             socks_list = file.readlines()
             random_index = random.randint(0, len(socks_list) - 1)
@@ -32,12 +32,12 @@ def getProxy() -> list:
             return proxy_addr, int(proxy_port)
 
 
-def createSocket():
+def create_socket():
     local_port = random.randint(1024, 65535)
     proxy_addr, proxy_port = None, None
     if proxy_used:
         udp_skt = socks.socksocket(socket.AF_INET, socket.SOCK_DGRAM)
-        proxy_addr, proxy_port = getProxy()
+        proxy_addr, proxy_port = get_proxy()
         udp_skt.set_proxy(socks.SOCKS5, proxy_addr, proxy_port)
         if proxy_used:
             log(f"Used proxy: {proxy_addr}:{proxy_port}")
@@ -49,21 +49,21 @@ def createSocket():
 def send_pkt(target_addr: str, port: int, payload_file: str, loops: int, interval: float):
     for i in range(loops):
         try:
-            local_port, udp_skt = createSocket()
+            local_port, udp_skt = create_socket()
         except:
             continue
         payloads = None
         try:
             payloads = marshal.load(open(payload_file, "rb"))
         except:
-            log(f"Payload file {payload_file} not found!")
+            log(f"Payload file {payload_file} not found!", quiet=quiet_mode)
             os._exit(114514)
         try:
             if is_display_motd:
                 send_motd_pkt(target_addr, port)
         except:
             if is_display_motd:
-                log(f"Target server offline.")
+                log(f"Target server offline.", quiet=quiet_mode)
         try:
             if port == "*":
                 for port in range(65535):
@@ -72,9 +72,9 @@ def send_pkt(target_addr: str, port: int, payload_file: str, loops: int, interva
             else:
                 for line in payloads:
                     udp_skt.sendto(line, (target_addr, int(port)))
-            log(f"Loop {i} done, used local port: {local_port} \n")
+            log(f"Loop {i} done, used local port: {local_port} \n", quiet=quiet_mode)
         except:
-            log(f"Loop {i} error! Skip...\n")
+            log(f"Loop {i} error! Skip...\n", quiet=quiet_mode)
             pass
         if i+1 < loops:
             time.sleep(interval)
@@ -87,6 +87,7 @@ def send_pkt(target_addr: str, port: int, payload_file: str, loops: int, interva
     parser.add_argument("-p", "--port", type=int, default=19132, help="Port (Number)")
     parser.add_argument("-l", "--loops", type=int, default=1, help="Loops (Number)")
     parser.add_argument("-i", "--interval", type=float, default=10, help="Interval (sec)")
+    parser.add_argument("-q", "--quiet", action="store_true", default=False, help="Quiet mode. no output")
     parser.add_argument("-d", "--display_motd", action="store_true", default=False, help="Display Motd")
     parser.add_argument("-pu", "--proxy_used", action="store_true", default=False, help="Use Proxy")
     parser.add_argument("-pc", "--proxy_country", type=str, default="us", help="ProxyCountry (like cn, ru, us or enter to use all)")
@@ -98,6 +99,7 @@ def send_pkt(target_addr: str, port: int, payload_file: str, loops: int, interva
     loops = args.loops
     interval = args.interval
     is_display_motd = args.display_motd
+    quiet_mode = args.quiet
     proxy_used = args.proxy_used
     proxyCountry = args.proxy_country