48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package dialer
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"net/netip"
|
|
"syscall"
|
|
)
|
|
|
|
type TunnelDialer func(context context.Context, network, address string) (net.Conn, error)
|
|
type SocketControl func(network, address string, conn syscall.RawConn) error
|
|
|
|
var DefaultTunnelDialer TunnelDialer
|
|
var DefaultSocketHook SocketControl
|
|
|
|
func DialTunnelContext(ctx context.Context, network, address string) (net.Conn, error) {
|
|
if dialer := DefaultTunnelDialer; dialer != nil {
|
|
return dialer(ctx, network, address)
|
|
}
|
|
|
|
return DialContext(ctx, network, address)
|
|
}
|
|
|
|
func dialContextHooked(ctx context.Context, network string, destination netip.Addr, port string) (net.Conn, error) {
|
|
dialer := &net.Dialer{
|
|
Control: DefaultSocketHook,
|
|
}
|
|
|
|
conn, err := dialer.DialContext(ctx, network, net.JoinHostPort(destination.String(), port))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if t, ok := conn.(*net.TCPConn); ok {
|
|
t.SetKeepAlive(false)
|
|
}
|
|
|
|
return conn, nil
|
|
}
|
|
|
|
func listenPacketHooked(ctx context.Context, network, address string) (net.PacketConn, error) {
|
|
lc := &net.ListenConfig{
|
|
Control: DefaultSocketHook,
|
|
}
|
|
|
|
return lc.ListenPacket(ctx, network, address)
|
|
}
|