add bind for relay

2021-11-25 17:29:54 +08:00
parent 98ef6c7492
commit 6daf0a4d0f
29 changed files with 600 additions and 352 deletions
--- a/pkg/chain/transport.go
+++ b/pkg/chain/transport.go
@ -49,8 +49,12 @@ func (tr *Transport) dialOptions() []dialer.DialOption {
 }

 func (tr *Transport) Handshake(ctx context.Context, conn net.Conn) (net.Conn, error) {
+	var err error
 	if hs, ok := tr.dialer.(dialer.Handshaker); ok {
-		return hs.Handshake(ctx, conn)
+		conn, err = hs.Handshake(ctx, conn)
+		if err != nil {
+			return nil, err
+		}
 	}
 	if hs, ok := tr.connector.(connector.Handshaker); ok {
 		return hs.Handshake(ctx, conn)