fix test status printing

Author: lanrat

README.md

@@ -24,7 +24,7 @@ OPTIONS:
      print version and exit
 ```
 
-Stargate now operates as a single SOCKS5 proxy server that randomly selects egress IP addresses from your specified CIDR range. This approach is much more memory-efficient and suitable for large IPv6 ranges.
+Stargate operates as a single SOCKS5 proxy server that randomly selects egress IP addresses from your specified CIDR range. This approach is much more memory-efficient and suitable for large IPv6 ranges.
 
 ## Test Flag - Preventing IP Address Leakage
 

test.go

@@ -67,20 +67,27 @@ func test(ctx context.Context, parsedNetwork netip.Prefix, cidrSize uint) error
 	})
 
 	// print testing status
-	statusStop := make(chan bool)
+	statusStop := make(chan bool, 1)
+	statusDone := make(chan bool, 1)
 	if !*verbose {
-		defer func() { statusStop <- true }()
 		go func() {
+			ticker := time.NewTicker(time.Second / 4)
+			defer ticker.Stop()
+			done := false
 			for {
 				select {
 				case <-statusStop:
+					done = true
+				case <-ticker.C:
+				}
+				testedCount := tested.Load()
+				totalHosts := ipItr.Size()
+				progress := float64(testedCount) / float64(totalHosts) * 100
+				fmt.Printf("\r Testing %d/%d (%.1f%%) failures: %d", testedCount, totalHosts, progress, failed.Load())
+				if done {
 					fmt.Printf("\n") // Clear the status
+					close(statusDone)
 					return
-				default:
-					testedCount := tested.Load()
-					totalHosts := ipItr.Size()
-					progress := float64(testedCount) / float64(totalHosts) * 100
-					fmt.Printf("\r Testing %d/%d (%.1f%%) failures: %d", testedCount, totalHosts, progress, failed.Load())
 				}
 			}
 		}()
@@ -115,9 +122,17 @@ func test(ctx context.Context, parsedNetwork netip.Prefix, cidrSize uint) error
 
 	// Wait for all goroutines to complete
 	if err := group.Wait(); err != nil {
+		statusStop <- true
 		return err
 	}
 
+	// stop status printing
+	if !*verbose {
+		statusStop <- true
+		// wait for last status print
+		<-statusDone
+	}
+
 	if failedCount := failed.Load(); failedCount > 0 {
 		return fmt.Errorf("test finished with %d/%d failures", failedCount, ipItr.Size())
 	}