dae/component/outbound/dialer/direct.go

package dialer

import (
	"golang.org/x/net/proxy"
	"net"
	"runtime"
	"syscall"
)

var SymmetricDirect = newDirect(false)
var FullconeDirect = newDirect(true)

func NewDirectDialer(option *GlobalOption, fullcone bool) *Dialer {
	if fullcone {
		return NewDialer(FullconeDirect, option, InstanceOption{Check: false}, true, "direct", "direct", "")
	} else {
		return NewDialer(SymmetricDirect, option, InstanceOption{Check: false}, true, "direct", "direct", "")
	}
}

type direct struct {
	proxy.Dialer
	netDialer *net.Dialer
	fullCone  bool
}

func newDirect(fullCone bool) proxy.Dialer {
	return &direct{
		netDialer: &net.Dialer{Control: func(network, address string, c syscall.RawConn) error {
			return SoMarkControl(c)
		}},
		fullCone: fullCone,
	}
}

func (d *direct) Dial(network, addr string) (c net.Conn, err error) {
	switch network {
	case "tcp":
		conn, err := d.netDialer.Dial(network, addr)
		return conn, err
	case "udp":
		if d.fullCone {
			conn, err := net.ListenUDP(network, nil)
			if err != nil {
				return nil, err
			}
			raw, e := conn.SyscallConn()
			if e == nil {
				_ = SoMarkControl(raw)
			}
			return &directUDPConn{UDPConn: conn, FullCone: true}, nil
		} else {
			conn, err := d.netDialer.Dial(network, addr)
			if err != nil {
				return nil, err
			}
			return &directUDPConn{UDPConn: conn.(*net.UDPConn), FullCone: false}, nil
		}
	default:
		return nil, net.UnknownNetworkError(network)
	}
}

type directUDPConn struct {
	*net.UDPConn
	FullCone bool
}

func (c *directUDPConn) WriteTo(b []byte, addr net.Addr) (int, error) {
	if !c.FullCone {
		// FIXME: check the addr
		return c.Write(b)
	}
	return c.UDPConn.WriteTo(b, addr)
}

func (c *directUDPConn) WriteMsgUDP(b, oob []byte, addr *net.UDPAddr) (n, oobn int, err error) {
	if !c.FullCone {
		n, err = c.Write(b)
		return n, 0, err
	}
	return c.UDPConn.WriteMsgUDP(b, oob, addr)
}

func (c *directUDPConn) WriteToUDP(b []byte, addr *net.UDPAddr) (int, error) {
	if !c.FullCone {
		return c.Write(b)
	}
	return c.UDPConn.WriteToUDP(b, addr)
}

var fwmarkIoctl int

func init() {
	switch runtime.GOOS {
	case "linux", "android":
		fwmarkIoctl = 36 /* unix.SO_MARK */
	case "freebsd":
		fwmarkIoctl = 0x1015 /* unix.SO_USER_COOKIE */
	case "openbsd":
		fwmarkIoctl = 0x1021 /* unix.SO_RTABLE */
	}
}

func SoMarkControl(c syscall.RawConn) error {
	return c.Control(func(fd uintptr) {
		//TODO: force to set 0xff. any chances to customize this value?
		err := syscall.SetsockoptInt(int(fd), syscall.SOL_SOCKET, fwmarkIoctl, 0x100)
		if err != nil {
			return
		}
	})
}