fix: new ipPortNumMap after auto release

XinRoom · Feb 6, 2023 · 98ab122 · 98ab122
1 parent d1977c4
commit 98ab122
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/cmd/go-portScan.go b/cmd/go-portScan.go
@@ -196,7 +196,9 @@ func run(c *cli.Context) error {
 				}
 				if maxOpenPort > 0 {
 					ipPortNumRW.Lock()
-					ipPortNumMap[ret.Ip.String()] += 1
+					if _, ok := ipPortNumMap[ret.Ip.String()]; ok {
+						ipPortNumMap[ret.Ip.String()] += 1
+					}
 					ipPortNumRW.Unlock()
 				}
 				if sV || httpx {
@@ -266,6 +268,11 @@ func run(c *cli.Context) error {
 	portScan := func(ip net.IP) {
 		var ipPortNum int
 		var ipPortNumOk bool
+		if maxOpenPort > 0 {
+			ipPortNumRW.Lock()
+			ipPortNumMap[ip.String()] = 0
+			ipPortNumRW.Unlock()
+		}
 		for _, _port := range ports { // port
 			s.WaitLimiter() // limit rate