fix: End without finishing output

XinRoom · Aug 25, 2022 · 9f2c240 · 9f2c240
1 parent 3cde1e7
commit 9f2c240
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/cmd/go-portScan.go b/cmd/go-portScan.go
@@ -148,11 +148,11 @@ func run(c *cli.Context) error {
 	})
 	defer poolPortIdentify.Release()
 	go func() {
+		sendOverFlag := false
 		for {
 			select {
 			case <-single1:
-				single2 <- struct{}{}
-				return
+				sendOverFlag = true
 			case ret := <-retChan:
 				if sV || httpx {
 					// port fingerprint
@@ -163,6 +163,10 @@ func run(c *cli.Context) error {
 					fmt.Printf("%v:%d\n", ret.Ip, ret.Port)
 				}
 			default:
+				if sendOverFlag {
+					single2 <- struct{}{}
+					return
+				}
 				time.Sleep(time.Millisecond * 10)
 			}
 		}