set-dst:// dialer

Author: Snawoot

dialer/dialer.go

@@ -21,6 +21,7 @@ func init() {
 	xproxy.RegisterDialerType("h2", H2ProxyDialerFromURL)
 	xproxy.RegisterDialerType("h2c", H2ProxyDialerFromURL)
 	xproxy.RegisterDialerType("set-src-hints", NewHintsSettingDialerFromURL)
+	xproxy.RegisterDialerType("set-dst", NewFixedDstDialerFromURL)
 	xproxy.RegisterDialerType("cached", GetCachedDialer)
 	xproxy.RegisterDialerType("socks5s", SOCKS5SDialerFromURL)
 	xproxy.RegisterDialerType("socks5hs", SOCKS5SDialerFromURL)

dialer/fixed.go

@@ -0,0 +1,45 @@
+package dialer
+
+import (
+	"context"
+	"errors"
+	"net"
+	"net/url"
+
+	xproxy "golang.org/x/net/proxy"
+)
+
+type FixedDstDialer struct {
+	addr string
+	next Dialer
+}
+
+func NewFixedDstDialerFromURL(u *url.URL, next xproxy.Dialer) (xproxy.Dialer, error) {
+	host, port := u.Hostname(), u.Port()
+	if host == "" {
+		return nil, errors.New("missing hostname")
+	}
+	if port == "" {
+		return nil, errors.New("missing port")
+	}
+	return &FixedDstDialer{
+		addr: net.JoinHostPort(host, port),
+		next: MaybeWrapWithContextDialer(next),
+	}, nil
+}
+
+func (d *FixedDstDialer) Dial(network, address string) (net.Conn, error) {
+	return d.DialContext(context.Background(), network, address)
+}
+
+func (d *FixedDstDialer) DialContext(ctx context.Context, network, address string) (net.Conn, error) {
+	return d.next.DialContext(ctx, network, d.addr)
+}
+
+func (d *FixedDstDialer) WantsHostname(_ context.Context, _, _ string) bool {
+	// there is no point resolving address which we will discard anyway
+	return true
+}
+
+var _ Dialer = new(FixedDstDialer)
+var _ HostnameWanter = new(FixedDstDialer)