feat: dispatch proxy error

client.go

@@ -117,7 +117,9 @@ func (c *Client) handleProxyRequest(ctx context.Context, m *control.Message) (*c
 		return nil, errors.WithStack(err)
 	}
 
-	pipe(stream, net)
+	if err := pipe(stream, net); err != nil {
+		return nil, errors.WithStack(err)
+	}
 
 	return nil, nil
 }

helper.go

@@ -5,22 +5,29 @@ import (
 	"net"
 )
 
-func pipe(client net.Conn, server net.Conn) {
+func pipe(client net.Conn, server net.Conn) (err error) {
 	stop := make(chan bool)
 
-	go relay(client, server, stop)
-	go relay(server, client, stop)
+	go func() {
+		err = relay(client, server, stop)
+	}()
+	go func() {
+		err = relay(server, client, stop)
+	}()
 
 	select {
 	case <-stop:
-		return
+		return err
 	}
 }
 
-func relay(src net.Conn, dst net.Conn, stop chan bool) {
-	io.Copy(dst, src)
+func relay(src io.ReadCloser, dst io.WriteCloser, stop chan bool) (err error) {
+	_, err = io.Copy(dst, src)
+
 	dst.Close()
 	src.Close()
+
 	stop <- true
+
 	return
 }

remote_client.go

@@ -125,6 +125,12 @@ func (c *RemoteClient) Proxy(ctx context.Context, network, address string) (net.
 	return c.control.Proxy(ctx, network, address)
 }
 
+func (c *RemoteClient) Close() {
+	if c.sess != nil && !c.sess.IsClosed() {
+		c.sess.Close()
+	}
+}
+
 func NewRemoteClient() *RemoteClient {
 	return &RemoteClient{}
 }

server.go

@@ -37,6 +37,7 @@ func (s *Server) handleNewConn(ctx context.Context, conn *kcp.UDPSession) {
 	ctx = logger.With(ctx, logger.F("remoteAddr", conn.RemoteAddr().String()))
 
 	remoteClient := NewRemoteClient()
+	defer remoteClient.Close()
 
 	remoteClient.ConfigureHooks(s.conf.Hooks)