From 98ab122fbd3cde74c146eaa86bada315cb800aa5 Mon Sep 17 00:00:00 2001
From: XinRoom <32238570+XinRoom@users.noreply.github.com>
Date: Mon, 6 Feb 2023 11:55:58 +0800
Subject: [PATCH] fix: new ipPortNumMap after auto release

---
 cmd/go-portScan.go | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/cmd/go-portScan.go b/cmd/go-portScan.go
index cc09893..0540483 100644
--- 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