add proxyProtocol support for dialer

2025-08-03 15:36:29 +08:00
parent f71351f5ef
commit ad5cf6fd61
23 changed files with 260 additions and 48 deletions
@@ -11,6 +11,8 @@ import (
 	"github.com/go-gost/core/dialer"
 	"github.com/go-gost/core/logger"
 	md "github.com/go-gost/core/metadata"
+	ctxvalue "github.com/go-gost/x/ctx"
+	"github.com/go-gost/x/internal/net/proxyproto"
 	"github.com/go-gost/x/internal/util/mux"
 	"github.com/go-gost/x/registry"
 )
@@ -73,6 +75,12 @@ func (d *mtlsDialer) Dial(ctx context.Context, addr string, opts ...dialer.DialO
 			return
 		}

+		conn = proxyproto.WrapClientConn(
+			d.options.ProxyProtocol,
+			ctxvalue.SrcAddrFromContext(ctx),
+			ctxvalue.DstAddrFromContext(ctx),
+			conn)
+
 		session = &muxSession{conn: conn}
 		d.sessions[addr] = session
 	}