proxy: start ctx

Author: Mizzick

proxy/proxy.go

@@ -346,7 +346,9 @@ func (p *Proxy) Start(ctx context.Context) (err error) {
 		return fmt.Errorf("configuring listeners: %w", errors.WithDeferred(err, closeErr))
 	}
 
-	p.serveListeners(ctx)
+	// Use context without cancel to prevent listeners' context from being
+	// canceled.
+	p.serveListeners(context.WithoutCancel(ctx))
 
 	p.started = true
 

proxy/serverquic.go

@@ -185,7 +185,7 @@ func (p *Proxy) handleQUICConnection(
 		// design specifies that for each subsequent query on a QUIC connection
 		// the client MUST select the next available client-initiated
 		// bidirectional stream.
-		stream, err := conn.AcceptStream(context.WithoutCancel(ctx))
+		stream, err := conn.AcceptStream(ctx)
 		if err != nil {
 			logQUICError(ctx, "accepting quic stream", err, p.logger)
 

proxy/serverudp.go

@@ -96,7 +96,7 @@ func (p *Proxy) udpPacketLoop(ctx context.Context, conn *net.UDPConn, reqSema sy
 			packet := make([]byte, n)
 			copy(packet, b)
 
-			sErr := reqSema.Acquire(context.WithoutCancel(ctx))
+			sErr := reqSema.Acquire(ctx)
 			if sErr != nil {
 				p.logger.ErrorContext(
 					ctx,