Fix onetimepool goroutine leak

projectdiscovery · Aug 31, 2024 · 75b3c49 · 75b3c49
1 parent 40ca4ed
commit 75b3c49
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/conn/connpool/onetimepool.go b/conn/connpool/onetimepool.go
@@ -75,7 +75,11 @@ func (p *OneTimePool) Run() error {
 			}
 			if err == nil {
 				p.InFlightConns.Add(conn)
-				p.idleConnections <- conn
+				select {
+				case <-p.ctx.Done():
+					return p.ctx.Err()
+				case p.idleConnections <- conn:
+				}
 			}
 		}
 	}