Optimization code

Author: WillowSauceR

scan.py

@@ -113,6 +113,7 @@ def startThreads():
     with open(fileName, "w") as file:
         file.writelines(scanResultList)
     quietMode = False
+    print()
     log(f"BE Server Count: {scanResult['serverCount']}", info="I", quiet=quietMode)
     log(f"BDS Count: {scanResult['bdsCount']}", info="I", quiet=quietMode)
     log(f"NK Count: {scanResult['nkCount']}", info="I", quiet=quietMode)
@@ -200,7 +201,12 @@ def recvPackets(socketSendRecv: socket.socket, verboseMode: str, fileName: list,
             log(f"{errorInfo}, skipped.", info="R", quiet=quietMode)
             scanResult['error'] += 1
         finally:
-            pipe.send((scanResult, scanResultList))
+            try:
+                pipe.send((scanResult, scanResultList))
+            except OSError:
+                pass
+            except Exception as errorInfo:
+                log(f"Pipe error: {errorInfo}.", info="R", quiet=quietMode)
 
 def saveResults(fileName, scanResult, addr, date, infos, scanResultList):
     formatedScanResult = f"{date} | {scanResult['serverCount']} | {addr[0]} | {addr[1]} | {infos[1]} | {infos[3]} | {infos[4]} | {infos[5]}"

send.py

@@ -1,6 +1,7 @@
 import marshal
 import random
 import socket
+import subprocess
 import sys
 import time
 import os
@@ -99,6 +100,8 @@ def sendPacket(target: str, port, payloadFile: str, loops, interval):
     targetAddr = target
     autoScan = port
     targetFile, targetInfo = target.split(":")
+    targetFileToScan = targetFile
+    scanProcess = False
     for i in range(loops):
         if ":" in target:
             with open(targetFile, "r") as file: #, encoding="utf-8"
@@ -127,11 +130,19 @@ def sendPacket(target: str, port, payloadFile: str, loops, interval):
         try:
             if isDisplayMotd:
                 sendMotdPacket(targetAddr, port)
+                if scanProcess:
+                    scanProcess.kill()
+                    scanProcess = False
         except:
             if isDisplayMotd:
-                log(f"Target server offline.")
                 if ":" in target and int(autoScan) == 8:
-                    os.system(f"python scan.py {targetFile} 10000-21000 nn 0 {targetFile}")
+                    if not scanProcess:
+                        log(f"Target server offline.")
+                        log(f"Starting refresh ip list...")
+                        scanProcess = subprocess.Popen(["python", "scan.py", targetFileToScan, "10000-21000", "nn", "0", targetFileToScan])
+                        targetFile = "updated.txt"
+                else:
+                    log(f"Target server offline.")
                 #log(f"Sending packet...")
         try:
             if port == "*":