Add transport wrapper

1 files changed, 17 insertions, 0 deletions

diff --git a/common/proxy/client.go b/common/proxy/client.go
index cd1bd98..eb062f8 100644
--- a/common/proxy/client.go
+++ b/common/proxy/client.go
@@ -251,3 +251,20 @@ func (r *dnsResolver) lookupIPAddr(ctx context.Context, host string, ipv6 bool)
 	}
 	return returnedIPs, nil
 }
+
+func NewTransportWrapper(sc *SocksClient, innerNet transport.Net) transport.Net {
+	return &transportWrapper{sc: sc, Net: innerNet}
+}
+
+type transportWrapper struct {
+	transport.Net
+	sc *SocksClient
+}
+
+func (t *transportWrapper) ListenPacket(network string, address string) (net.PacketConn, error) {
+	return t.sc.ListenPacket(network, nil)
+}
+
+func (t *transportWrapper) ResolveUDPAddr(network string, address string) (*net.UDPAddr, error) {
+	return t.sc.ResolveUDPAddr(network, address)
+}