fix error when write file

Author: WillowSauceR

scan.py

@@ -110,8 +110,9 @@ def startThreads():
     mp.active_children()[0].terminate()
     # time.sleep(1)
     # print(scanResultList)
-    with open(fileName, "w") as file:
-        file.writelines(scanResultList)
+    if os.path.exists(fileName):
+        with open(fileName, "w") as file:
+            file.writelines(scanResultList)
     quietMode = False
     print()
     log(f"BE Server Count: {scanResult['serverCount']}", info="I", quiet=quietMode)
@@ -129,8 +130,11 @@ def recvPackets(socketSendRecv: socket.socket, verboseMode: str, fileName: list,
         quietMode = True
     else:
         quietMode = False
-    with open(fileName[1], "r") as file:
-        scanResultList = file.readlines()
+    if os.path.exists(fileName[1]):
+        with open(fileName[1], "r") as file:
+            scanResultList = file.readlines()
+    else:
+        scanResultList = []
     while True:
         try:
             data, addr = socketSendRecv.recvfrom(10240)
@@ -226,8 +230,7 @@ def saveResults(fileName, scanResult, addr, date, infos, scanResultList):
                     infos[1][3]
                 except:
                     return scanResultList
-                # if addr[0] not in scanResultList[index]:
-                if fileName[0] == fileName[1]:
+                if fileName[0] == fileName[1] and addr[0] not in scanResultList[index]:
                     with open("updated.txt", "a") as file:
                         pervInfo = scanResultList[index].split(" | ")
                         file.write(f"{formatedScanResult}  Pervious: {pervInfo[2]}:{pervInfo[3]}\n")
@@ -280,4 +283,7 @@ def saveResults(fileName, scanResult, addr, date, infos, scanResultList):
     t = threading.Thread(target=startThreads, daemon=True)
     t.start()
     while mp.active_children():
-        scanResult, scanResultList = pipe1.recv()
+        try:
+            scanResult, scanResultList = pipe1.recv()
+        except:
+            pass

send.py

@@ -112,8 +112,9 @@ def sendPacket(target: str, port, payloadFile: str, loops, interval):
                     # print(index)
                     info = fileContent[contentCount-index-1]
                     if targetInfo in info:
-                        log("Found target info:")
-                        log(info)
+                        if not scanProcess:
+                            log("Found target info:")
+                            log(info)
                         targetAddr = info.split(" | ")[2]
                         port = info.split(" | ")[3]
                         break
@@ -135,11 +136,14 @@ def sendPacket(target: str, port, payloadFile: str, loops, interval):
                     scanProcess = False
         except:
             if isDisplayMotd:
+                if scanProcess:
+                    if scanProcess.poll() != None:
+                        scanProcess = False
                 if ":" in target and int(autoScan) == 8:
                     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])
+                        scanProcess = subprocess.Popen(["python", "scan.py", targetFileToScan, "10000-20000", "nn", "0", targetFileToScan])
                         targetFile = "updated.txt"
                 else:
                     log(f"Target server offline.")
@@ -152,10 +156,11 @@ def sendPacket(target: str, port, payloadFile: str, loops, interval):
             else:
                 for line in payloads:
                     socketSend.sendto(line, (targetAddr, int(port)))
-            log(f"Loop ", str(i),
-                  " done, used local port: ", str(localPort), "\n")
+            if not scanProcess:
+                # if scanProcess.poll() != None:
+                log(f"Loop {i} done, used local port: {localPort} \n")
         except:
-            log(f"Loop ", str(i), " error! Skip...", "\n")
+            log(f"Loop {i} error! Skip...\n")
             pass
         if i+1 < loops:
             time.sleep(interval)