TCP remote forwarding with chain

gost/examples/bench/srv.go

@@ -34,8 +34,8 @@ func main() {
 	go wsServer()
 	go wssServer()
 	go kcpServer()
-	// go tcpForwardServer()
+	go tcpForwardServer()
-	// go rtcpForwardServer()
+	go tcpRemoteForwardServer()
 	// go rudpForwardServer()
 	// go tcpRedirectServer()
 	go sshTunnelServer()
@@ -138,10 +138,11 @@ func tcpForwardServer() {
 	log.Fatal(s.Serve(ln, h))
 }
 
-func rtcpForwardServer() {
+func tcpRemoteForwardServer() {
 	s := &gost.Server{}
 	ln, err := gost.TCPRemoteForwardListener(
 		":1222",
+		/*
 			gost.NewChain(
 				gost.Node{
 					Protocol:  "socks5",
@@ -154,13 +155,15 @@ func rtcpForwardServer() {
 					},
 				},
 			),
+		*/
+		nil,
 	)
 	if err != nil {
 		log.Fatal()
 	}
 	h := gost.TCPRemoteForwardHandler(
-		":1222",
+		":22",
-		gost.AddrHandlerOption("127.0.0.1:22"),
+		//gost.AddrHandlerOption("127.0.0.1:22"),
 	)
 	log.Fatal(s.Serve(ln, h))
 }

gost/forward.go

@@ -405,6 +405,7 @@ func (c *udpServerConn) SetWriteDeadline(t time.Time) error {
 type tcpRemoteForwardListener struct {
 	addr   net.Addr
 	chain  *Chain
+	ln     net.Listener
 	closed chan struct{}
 }
 
@@ -463,7 +464,13 @@ func (l *tcpRemoteForwardListener) accept() (conn net.Conn, err error) {
 			cc.Close()
 		}
 	} else {
-		err = errors.New("invalid chain")
+		if l.ln == nil {
+			l.ln, err = net.Listen("tcp", l.addr.String())
+			if err != nil {
+				return
+			}
+		}
+		conn, err = l.ln.Accept()
 	}
 	return
 }