opt: ipPortNumMap auto release

XinRoom · Feb 6, 2023 · d1977c4 · d1977c4
1 parent 7bc50a7
commit d1977c4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/cmd/go-portScan.go b/cmd/go-portScan.go
@@ -274,7 +274,7 @@ func run(c *cli.Context) error {
 				ipPortNum, ipPortNumOk = ipPortNumMap[ip.String()]
 				ipPortNumRW.RUnlock()
 				if ipPortNumOk && ipPortNum >= maxOpenPort {
-					return
+					break
 				}
 			}
 
@@ -284,6 +284,11 @@ func run(c *cli.Context) error {
 				Port: _port,
 			})
 		}
+		if maxOpenPort > 0 {
+			ipPortNumRW.Lock()
+			delete(ipPortNumMap, ip.String())
+			ipPortNumRW.Unlock()
+		}
 	}
 
 	// Pool - ping and port scan