Auto Restart Processes

Author: CrawLeyYou

src/scanner.mjs

@@ -46,7 +46,7 @@ const getColor = (data, mode) => {
             }
         case "status":
             if (data.extra > 5) {
-                return colors[0] + "Error"
+                return colors[0] + "Down"
             } else {
                 if (data.data === 0) {
                     return colors[2] + "Idle"
@@ -71,7 +71,7 @@ const Main = async () => {
         for (var i = 0; i < os.availableParallelism(); i++) {
             processList.push([cprocess.fork("./libs/thread.mjs", [i]), []])
             logger.debug("Process: " + processList[i][0].pid + " started")
-            assignedList.push([i, { currentIp: "", lastresponsetext: 0, status: "Idle", assigned: 0, finds: 0, total: 0, currentPings: 0 }])
+            assignedList.push([i, { currentIp: "", lastresponsetext: 0, status: "Idle", assigned: 0, finds: 0, total: 0, currentPings: 0, restarts: 0 }])
         }
         for (var i = 0; i < processList.length; i++) {
             processList[i][0].on("message", (data) => {
@@ -86,29 +86,39 @@ const Main = async () => {
                     assignedList[x][1].assigned--
                     assignedList[x][1].lastresponsetext = parseInt(Date.now()) - data.time
                     assignedList[x][1].currentIp = data.ip
+                    var y = processList[assignedList[x][0]][1].findIndex((sarray) => {
+                        const [number] = sarray
+                        return number === data.ip
+                    })
+                    processList[assignedList[x][0]][1].splice(y, 1)
                     logger.debug(processList[assignedList[x][0]][0].pid + " processed data from ip: " + assignedList[x][1].currentIp);
                 }
                 else if (data.status === "error") {
                     currData.assigned--
                     assignedList[x][1].assigned--
                     assignedList[x][1].lastresponsetext = parseInt(Date.now()) - data.time
                     assignedList[x][1].currentIp = data.ip
+                    var y = processList[assignedList[x][0]][1].findIndex((sarray) => {
+                        const [number] = sarray
+                        return number === data.ip
+                    })
+                    processList[assignedList[x][0]][1].splice(y, 1)
                     logger.debug(processList[assignedList[x][0]][0].pid + " failed to processing ip: " + assignedList[x][1].currentIp);
                 }
                 else if (data.status === "ping") {
                     assignedList[x][1].currentPings = 0
-                    logger.info(processList[assignedList[x][0]][0].pid + " responded to ping request.")
+                    logger.debug(processList[assignedList[x][0]][0].pid + " responded to ping request.")
                 }
             })
         }
         console.log("Parser will be started in 3 seconds...")
         setTimeout(() => {
-            //startParser('java -Xmx5G -jar ' + process.cwd() + '/libs/copenJSParser-1.0-SNAPSHOT-5ms.jar')
+            startParser('java -Xmx5G -jar ' + process.cwd() + '/libs/copenJSParser-1.0-SNAPSHOT-5ms.jar')
             setInterval(async () => {
                 assignedList.sort((a, b) => { return a[0] - b[0] })
                 let defaultText = `Main Process: \n   Current IP/s: ${currData.total - currData.totalLast} IP/s \n   Total Assigned: ${currData.assigned} \n   Total TCP Restarts: ${currData.tcpRestarts} \n   Total Finds & Total Try: ${currData.finds} & ${currData.total} \n   Current Rate: ${((currData.finds / currData.total) * 100).toFixed(2)}% \n   Total Sub-Processes: ${currData.subprocesses} \n   Current IP: ${currData.lastIp} \nSub-Processes: `
                 assignedList.forEach(async (data, n) => {
-                    defaultText += "\n   Process processNumber: Current IP: currentIP Last Response: lastResponseText Status: statusText Total Assigned: assignedInt Total Finds: findInt ".replace("processNumber", n.toString()).replace("findInt", data[1].finds).replace("assignedInt", data[1].assigned).replace("currentIP", data[1].currentIp).replace("lastResponseText", `${getColor(data[1].lastresponsetext, "ms")}${data[1].lastresponsetext}ms\x1b[0m`).replace("statusText", `${getColor({ data: data[1].assigned, extra: data[1].currentPings }, "status")}\x1b[0m`)
+                    defaultText += "\n   Process processNumber: Current IP: currentIP Last Response: lastResponseText Status: statusText Total Assigned: assignedInt Total Finds: findInt Restarts: restartInt Responseless Pings: pingInt ".replace("processNumber", n.toString()).replace("findInt", data[1].finds).replace("assignedInt", data[1].assigned).replace("currentIP", data[1].currentIp).replace("lastResponseText", `${getColor(data[1].lastresponsetext, "ms")}${data[1].lastresponsetext}ms\x1b[0m`).replace("statusText", `${getColor({ data: data[1].assigned, extra: data[1].currentPings }, "status")}\x1b[0m`).replace("restartInt", data[1].restarts).replace("pingInt", data[1].currentPings)
                 })
                 text.text = defaultText
                 currData.totalLast = currData.total
@@ -127,13 +137,32 @@ const Main = async () => {
                     }
                 }
             }, 10000)
+            setInterval(async () => {
+                for (var i = 0; i < processList.length; i++) {
+                    if (assignedList[i][1].currentPings > 5) {
+                        logger.info("Restarting Thread: " + assignedList[i][0]),
+                        processList[assignedList[i][0]][0].kill("SIGINT")
+                        processList[assignedList[i][0]][0] = cprocess.fork("./libs/thread.mjs", [assignedList[i][0]])
+                        assignedList[i][1].restarts++
+                        processList[assignedList[i][0]][1].shift()
+                        assignedList[i][1].assigned--
+                        processList[assignedList[i][0]][1].forEach((data) => {
+                            processList[assignedList[i][0]][0].send({ mode: "search", ip: data, time: Date.now() })
+                            processList[assignedList[i][0]][1].shift()
+                            assignedList[i][1].assigned--
+                        });
+                        assignedList[i][1].currentPings = 0
+                    }
+                }
+            }, 30000)
         }, 3000)
     })
     server.on("connection", function (socket) {
         socket.on('error', function (err) {
             console.log("Client lost connection")
             logger.warn("The client lost the connection " + err)
             startParser('java -Xmx5G -jar ' + process.cwd() + '/libs/copenJSParser-1.0-SNAPSHOT-5ms.jar "' + currData.lastIp + '"')
+            currData.tcpRestarts++
         })
         socket.on('end', function () {
             console.log("Client disconnected")
@@ -150,6 +179,7 @@ const Main = async () => {
             }
             else {
                 processList[assignedList[x][0]][0].send({ mode: "search", ip: ip.toString(), time: Date.now() })
+                processList[assignedList[x][0]][1].push(ip.toString())
                 logger.debug(processList[assignedList[x][0]][0].pid + " assigned to " + ip.toString());
                 assignedList[x][1].assigned++
                 assignedList[x][1].total++