Merge pull request #507 from osxtest/main

Fix onetimepool goroutine leak
projectdiscovery · Sep 2, 2024 · 48931aa · 48931aa
2 parents 40ca4ed + 75b3c49
commit 48931aa
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:
+ }
  }
  }
  }