fix: memory leak

pool/pool.go

@@ -1,10 +1,13 @@
 package pool
 
-import "log"
+import (
+	"context"
+	"log"
+)
 
 type Worker struct {
-	Func chan func()
-	Stop chan struct{}
+	Func   chan func()
+	Cancel context.CancelFunc
 }
 
 type Pool struct {
@@ -41,12 +44,20 @@ func (w *Worker) Run() {
 			log.Println("Recovered ", r)
 		}
 	}()
+	ctx, cancel := context.WithCancel(context.Background())
+	w.Cancel = cancel
 	for {
 		select {
 		case f := <-w.Func:
 			f()
-		case <-w.Stop:
+		case <-ctx.Done():
 			return
 		}
 	}
 }
+
+func (p *Pool) Close() {
+	for _, w := range p.workers {
+		w.Cancel()
+	}
+}

scan/scan.go

@@ -88,6 +88,7 @@ func (s *Scanner) ScanAll(prefixs []netip.Prefix, port []int) []netip.AddrPort {
 	} else {
 		p := pool.Pool{Size: s.PortScanRate, Buffer: s.PortScanRate}
 		p.Init()
+		defer p.Close()
 		var wg sync.WaitGroup
 		wg.Add(addrCount * len(port))
 		ipGenerator(prefixs)(func(addr netip.Addr) {
@@ -111,6 +112,7 @@ func (s *Scanner) ScanAll(prefixs []netip.Prefix, port []int) []netip.AddrPort {
 	log.Println("start socks5 scan with 128 threads.")
 	p := pool.Pool{Size: 128, Buffer: 128}
 	p.Init()
+	defer p.Close()
 	c = utils.NewCollector[netip.AddrPort]()
 	var wg sync.WaitGroup
 	wg.Add(len(aliveTCPAddrs))