glider/proxy/direct.go

package proxy

import (
	"net"

	"github.com/nadoo/glider/common/log"
)

// Direct proxy
type Direct struct {
	*net.Dialer
}

// Default dialer
var Default = &Direct{Dialer: &net.Dialer{}}

// NewDirect returns a Direct dialer
func NewDirect(localip string) *Direct {
	d := &net.Dialer{LocalAddr: &net.TCPAddr{
		IP:   net.ParseIP(localip),
		Port: 0,
	}}
	return &Direct{Dialer: d}
}

// Addr returns forwarder's address
func (d *Direct) Addr() string { return "DIRECT" }

// Dial connects to the address addr on the network net
func (d *Direct) Dial(network, addr string) (net.Conn, error) {
	if network == "uot" {
		network = "udp"
	}

	c, err := d.Dialer.Dial(network, addr)
	if err != nil {
		return nil, err
	}

	if c, ok := c.(*net.TCPConn); ok {
		c.SetKeepAlive(true)
	}

	return c, err
}

// DialUDP connects to the given address
func (d *Direct) DialUDP(network, addr string) (net.PacketConn, net.Addr, error) {
	pc, err := net.ListenPacket(network, d.Dialer.LocalAddr.String())
	if err != nil {
		log.F("ListenPacket error: %s", err)
		return nil, nil, err
	}

	uAddr, err := net.ResolveUDPAddr("udp", addr)
	return pc, uAddr, err
}

// NextDialer returns the next dialer
func (d *Direct) NextDialer(dstAddr string) Dialer { return d }