fix proxy protocol

handler/socks/v4/handler.go

@@ -12,6 +12,7 @@ import (
 	md "github.com/go-gost/core/metadata"
 	"github.com/go-gost/gosocks4"
 	netpkg "github.com/go-gost/x/internal/net"
+	sx "github.com/go-gost/x/internal/util/selector"
 	"github.com/go-gost/x/registry"
 )
 
@@ -123,6 +124,11 @@ func (h *socks4Handler) handleConnect(ctx context.Context, conn net.Conn, req *g
 		return resp.Write(conn)
 	}
 
+	switch h.md.hash {
+	case "host":
+		ctx = sx.ContextWithHash(ctx, &sx.Hash{Source: addr})
+	}
+
 	cc, err := h.router.Dial(ctx, "tcp", addr)
 	if err != nil {
 		resp := gosocks4.NewReply(gosocks4.Failed, nil)

handler/socks/v4/metadata.go

@@ -9,13 +9,16 @@ import (
 
 type metadata struct {
 	readTimeout time.Duration
+	hash        string
 }
 
 func (h *socks4Handler) parseMetadata(md mdata.Metadata) (err error) {
 	const (
 		readTimeout = "readTimeout"
+		hash        = "hash"
 	)
 
 	h.md.readTimeout = mdutil.GetDuration(md, readTimeout)
+	h.md.hash = mdutil.GetString(md, hash)
 	return
 }

handler/socks/v5/connect.go

@@ -9,6 +9,7 @@ import (
 	"github.com/go-gost/core/logger"
 	"github.com/go-gost/gosocks5"
 	netpkg "github.com/go-gost/x/internal/net"
+	sx "github.com/go-gost/x/internal/util/selector"
 )
 
 func (h *socks5Handler) handleConnect(ctx context.Context, conn net.Conn, network, address string, log logger.Logger) error {
@@ -25,6 +26,11 @@ func (h *socks5Handler) handleConnect(ctx context.Context, conn net.Conn, networ
 		return resp.Write(conn)
 	}
 
+	switch h.md.hash {
+	case "host":
+		ctx = sx.ContextWithHash(ctx, &sx.Hash{Source: address})
+	}
+
 	cc, err := h.router.Dial(ctx, network, address)
 	if err != nil {
 		resp := gosocks5.NewReply(gosocks5.NetUnreachable, nil)

handler/socks/v5/metadata.go

@@ -15,6 +15,7 @@ type metadata struct {
 	enableUDP         bool
 	udpBufferSize     int
 	compatibilityMode bool
+	hash              string
 }
 
 func (h *socks5Handler) parseMetadata(md mdata.Metadata) (err error) {
@@ -25,6 +26,7 @@ func (h *socks5Handler) parseMetadata(md mdata.Metadata) (err error) {
 		enableUDP         = "udp"
 		udpBufferSize     = "udpBufferSize"
 		compatibilityMode = "comp"
+		hash              = "hash"
 	)
 
 	h.md.readTimeout = mdutil.GetDuration(md, readTimeout)
@@ -39,6 +41,7 @@ func (h *socks5Handler) parseMetadata(md mdata.Metadata) (err error) {
 	}
 
 	h.md.compatibilityMode = mdutil.GetBool(md, compatibilityMode)
+	h.md.hash = mdutil.GetString(md, hash)
 
 	return nil
 }