Add transport wrapper

This commit is contained in:
Shelikhoo 2023-10-18 16:48:59 +01:00
parent 8b46e60553
commit f43da1d2d2
No known key found for this signature in database
GPG key ID: C4D5E79D22B25316

View file

@ -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)
}