wireguard-go/conn.go

129 lines
2.6 KiB
Go
Raw Normal View History

package main
import (
"errors"
2017-11-14 15:27:53 +00:00
"golang.org/x/net/ipv4"
"golang.org/x/net/ipv6"
"net"
)
/* A Bind handles listening on a port for both IPv6 and IPv4 UDP traffic
*/
type Bind interface {
2017-10-08 20:03:32 +00:00
SetMark(value uint32) error
ReceiveIPv6(buff []byte) (int, Endpoint, error)
ReceiveIPv4(buff []byte) (int, Endpoint, error)
Send(buff []byte, end Endpoint) error
2017-10-08 20:03:32 +00:00
Close() error
}
/* An Endpoint maintains the source/destination caching for a peer
*
* dst : the remote address of a peer ("endpoint" in uapi terminology)
* src : the local address from which datagrams originate going to the peer
*/
type Endpoint interface {
ClearSrc() // clears the source address
SrcToString() string // returns the local source address (ip:port)
DstToString() string // returns the destination address (ip:port)
DstToBytes() []byte // used for mac2 cookie calculations
DstIP() net.IP
SrcIP() net.IP
}
func parseEndpoint(s string) (*net.UDPAddr, error) {
// ensure that the host is an IP address
host, _, err := net.SplitHostPort(s)
if err != nil {
return nil, err
}
if ip := net.ParseIP(host); ip == nil {
return nil, errors.New("Failed to parse IP address: " + host)
}
// parse address and port
addr, err := net.ResolveUDPAddr("udp", s)
if err != nil {
return nil, err
}
return addr, err
}
2017-11-11 14:43:55 +00:00
/* Must hold device and net lock
*/
2017-11-19 12:35:17 +00:00
func unsafeCloseBind(device *Device) error {
2017-11-11 22:26:44 +00:00
var err error
2017-11-11 14:43:55 +00:00
netc := &device.net
if netc.bind != nil {
2017-11-11 22:26:44 +00:00
err = netc.bind.Close()
2017-11-11 14:43:55 +00:00
netc.bind = nil
}
2017-11-11 22:26:44 +00:00
return err
2017-11-11 14:43:55 +00:00
}
func (device *Device) BindUpdate() error {
device.net.mutex.Lock()
defer device.net.mutex.Unlock()
device.peers.mutex.Lock()
defer device.peers.mutex.Unlock()
// close existing sockets
2017-11-19 12:35:17 +00:00
if err := unsafeCloseBind(device); err != nil {
2017-11-11 14:43:55 +00:00
return err
}
// open new sockets
2017-12-29 16:42:09 +00:00
if device.isUp.Get() {
2017-08-17 10:58:18 +00:00
2017-10-08 20:03:32 +00:00
// bind to new port
var err error
netc := &device.net
2017-11-18 23:21:58 +00:00
netc.bind, netc.port, err = CreateBind(netc.port)
2017-10-08 20:03:32 +00:00
if err != nil {
2017-11-11 14:43:55 +00:00
netc.bind = nil
2017-10-08 20:03:32 +00:00
return err
}
2017-08-17 10:58:18 +00:00
2017-10-08 20:03:32 +00:00
// set mark
err = netc.bind.SetMark(netc.fwmark)
if err != nil {
return err
2017-08-22 15:22:45 +00:00
}
// clear cached source addresses
for _, peer := range device.peers.keyMap {
peer.mutex.Lock()
defer peer.mutex.Unlock()
if peer.endpoint != nil {
peer.endpoint.ClearSrc()
}
}
2017-11-11 14:43:55 +00:00
// start receiving routines
2017-11-11 14:43:55 +00:00
go device.RoutineReceiveIncoming(ipv4.Version, netc.bind)
go device.RoutineReceiveIncoming(ipv6.Version, netc.bind)
2017-11-14 15:27:53 +00:00
2017-11-11 22:26:44 +00:00
device.log.Debug.Println("UDP bind has been updated")
}
2017-08-17 10:58:18 +00:00
return nil
}
func (device *Device) BindClose() error {
2017-11-11 14:43:55 +00:00
device.net.mutex.Lock()
2017-11-19 12:35:17 +00:00
err := unsafeCloseBind(device)
2017-11-11 14:43:55 +00:00
device.net.mutex.Unlock()
return err
}