Fix singleConnectionListener to prevent immediate EOF

- Use channel-based approach to control listener lifecycle
- Prevent http.Serve from shutting down immediately after first Accept()
- Add proper Close() method with channel signaling
- Use constructor function for proper initialization
- Should fix the premature EOF issue

Co-authored-by: f0ssel <[email protected]>

Author: blink-so[bot]

proxy/proxy.go

@@ -537,12 +537,17 @@ func (p *Server) handleConnectionWithTLSDetection(conn net.Conn) {
 		}
 		p.logger.Debug("TLS handshake successful")
 		// Use HTTP server with TLS connection
-		http.Serve(&singleConnectionListener{conn: tlsConn}, http.HandlerFunc(p.handleHTTPS))
+		listener := newSingleConnectionListener(tlsConn)
+		defer listener.Close()
+		err = http.Serve(listener, http.HandlerFunc(p.handleHTTPS))
+		p.logger.Debug("http.Serve completed for HTTPS", "error", err)
 	} else {
 		p.logger.Debug("Detected HTTP request, handling normally")
 		// Use HTTP server with regular connection
 		p.logger.Debug("About to call http.Serve for HTTP connection")
-		err = http.Serve(&singleConnectionListener{conn: connWrapper}, http.HandlerFunc(p.handleHTTP))
+		listener := newSingleConnectionListener(connWrapper)
+		defer listener.Close()
+		err = http.Serve(listener, http.HandlerFunc(p.handleHTTP))
 		p.logger.Debug("http.Serve completed", "error", err)
 	}
 }
@@ -841,23 +846,54 @@ func (c *connectionWrapper) Read(p []byte) (int, error) {
 
 // singleConnectionListener wraps a single connection into a net.Listener
 type singleConnectionListener struct {
-	conn net.Conn
-	used bool
+	conn   net.Conn
+	used   bool
+	closed chan struct{}
+	mu     sync.Mutex
+}
+
+func newSingleConnectionListener(conn net.Conn) *singleConnectionListener {
+	return &singleConnectionListener{
+		conn:   conn,
+		closed: make(chan struct{}),
+	}
 }
 
 func (sl *singleConnectionListener) Accept() (net.Conn, error) {
+	sl.mu.Lock()
+	defer sl.mu.Unlock()
+	
 	if sl.used || sl.conn == nil {
+		// Wait for close signal
+		<-sl.closed
 		return nil, io.EOF
 	}
 	sl.used = true
 	return sl.conn, nil
 }
 
 func (sl *singleConnectionListener) Close() error {
+	sl.mu.Lock()
+	defer sl.mu.Unlock()
+	
+	select {
+	case <-sl.closed:
+		// Already closed
+	default:
+		close(sl.closed)
+	}
+	
+	if sl.conn != nil {
+		sl.conn.Close()
+		sl.conn = nil
+	}
 	return nil
 }
 
 func (sl *singleConnectionListener) Addr() net.Addr {
+	if sl.conn == nil {
+		return nil
+	}
 	return sl.conn.LocalAddr()
 }