Fix a regression in #223: sanitise selected closures (#232)

mholt · Aug 12, 2024 · 94cd399 · 94cd399
1 parent 2e6e133
commit 94cd399
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/layer4/app.go b/layer4/app.go
@@ -76,11 +76,15 @@ func (a *App) Start() error {
 				case net.Listener:
 					a.listeners = append(a.listeners, ln)
 					lnAddr = caddy.JoinNetworkAddress(ln.Addr().Network(), ln.Addr().String(), "")
-					go func() { _ = s.serve(ln) }()
+					go func(s *Server, ln net.Listener) {
+						_ = s.serve(ln)
+					}(s, ln)
 				case net.PacketConn:
 					a.packetConns = append(a.packetConns, ln)
 					lnAddr = caddy.JoinNetworkAddress(ln.LocalAddr().Network(), ln.LocalAddr().String(), "")
-					go func() { _ = s.servePacket(ln) }()
+					go func(s *Server, pc net.PacketConn) {
+						_ = s.servePacket(pc)
+					}(s, ln)
 				}
 				s.logger.Debug("listening", zap.String("address", lnAddr))
 			}