2017-08-25 14:53:23 +02:00
|
|
|
// +build !linux
|
|
|
|
|
2018-05-03 15:04:00 +02:00
|
|
|
/* SPDX-License-Identifier: GPL-2.0
|
|
|
|
*
|
|
|
|
* Copyright (C) 2017-2018 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved.
|
2018-05-19 00:34:56 +02:00
|
|
|
* Copyright (C) 2017-2018 Mathias N. Hall-Andersen <mathias@hall-andersen.dk>.
|
2018-05-03 15:04:00 +02:00
|
|
|
*/
|
|
|
|
|
2017-08-25 14:53:23 +02:00
|
|
|
package main
|
|
|
|
|
|
|
|
import (
|
2018-05-21 17:27:18 +02:00
|
|
|
"golang.org/x/sys/unix"
|
2017-08-25 14:53:23 +02:00
|
|
|
"net"
|
2018-05-21 17:27:18 +02:00
|
|
|
"runtime"
|
2017-08-25 14:53:23 +02:00
|
|
|
)
|
|
|
|
|
2017-11-17 17:25:45 +01:00
|
|
|
/* This code is meant to be a temporary solution
|
|
|
|
* on platforms for which the sticky socket / source caching behavior
|
|
|
|
* has not yet been implemented.
|
|
|
|
*
|
|
|
|
* See conn_linux.go for an implementation on the linux platform.
|
|
|
|
*/
|
|
|
|
|
2017-11-19 00:21:58 +01:00
|
|
|
type NativeBind struct {
|
|
|
|
ipv4 *net.UDPConn
|
|
|
|
ipv6 *net.UDPConn
|
|
|
|
}
|
|
|
|
|
|
|
|
type NativeEndpoint net.UDPAddr
|
|
|
|
|
|
|
|
var _ Bind = (*NativeBind)(nil)
|
|
|
|
var _ Endpoint = (*NativeEndpoint)(nil)
|
|
|
|
|
|
|
|
func CreateEndpoint(s string) (Endpoint, error) {
|
|
|
|
addr, err := parseEndpoint(s)
|
2017-11-19 13:14:15 +01:00
|
|
|
return (*NativeEndpoint)(addr), err
|
2017-11-19 00:21:58 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
func (_ *NativeEndpoint) ClearSrc() {}
|
|
|
|
|
|
|
|
func (e *NativeEndpoint) DstIP() net.IP {
|
|
|
|
return (*net.UDPAddr)(e).IP
|
|
|
|
}
|
|
|
|
|
|
|
|
func (e *NativeEndpoint) SrcIP() net.IP {
|
|
|
|
return nil // not supported
|
|
|
|
}
|
2017-11-17 17:25:45 +01:00
|
|
|
|
2017-11-19 00:21:58 +01:00
|
|
|
func (e *NativeEndpoint) DstToBytes() []byte {
|
|
|
|
addr := (*net.UDPAddr)(e)
|
2018-05-18 05:02:35 +02:00
|
|
|
out := addr.IP.To4()
|
|
|
|
if out == nil {
|
|
|
|
out = addr.IP
|
|
|
|
}
|
2017-11-19 00:21:58 +01:00
|
|
|
out = append(out, byte(addr.Port&0xff))
|
|
|
|
out = append(out, byte((addr.Port>>8)&0xff))
|
|
|
|
return out
|
|
|
|
}
|
2017-11-17 17:25:45 +01:00
|
|
|
|
2017-11-19 00:21:58 +01:00
|
|
|
func (e *NativeEndpoint) DstToString() string {
|
|
|
|
return (*net.UDPAddr)(e).String()
|
|
|
|
}
|
|
|
|
|
|
|
|
func (e *NativeEndpoint) SrcToString() string {
|
|
|
|
return ""
|
|
|
|
}
|
|
|
|
|
2017-11-19 13:14:15 +01:00
|
|
|
func listenNet(network string, port int) (*net.UDPConn, int, error) {
|
2017-11-19 00:21:58 +01:00
|
|
|
|
|
|
|
// listen
|
|
|
|
|
2017-11-19 13:14:15 +01:00
|
|
|
conn, err := net.ListenUDP(network, &net.UDPAddr{Port: port})
|
2017-11-19 00:21:58 +01:00
|
|
|
if err != nil {
|
|
|
|
return nil, 0, err
|
|
|
|
}
|
|
|
|
|
|
|
|
// retrieve port
|
|
|
|
|
|
|
|
laddr := conn.LocalAddr()
|
2017-11-19 13:14:15 +01:00
|
|
|
uaddr, err := net.ResolveUDPAddr(
|
2017-11-19 00:21:58 +01:00
|
|
|
laddr.Network(),
|
|
|
|
laddr.String(),
|
|
|
|
)
|
2017-11-19 13:14:15 +01:00
|
|
|
if err != nil {
|
|
|
|
return nil, 0, err
|
|
|
|
}
|
2017-11-19 00:21:58 +01:00
|
|
|
return conn, uaddr.Port, nil
|
|
|
|
}
|
|
|
|
|
2018-05-14 03:00:40 +02:00
|
|
|
func CreateBind(uport uint16, device *Device) (Bind, uint16, error) {
|
2017-11-19 13:14:15 +01:00
|
|
|
var err error
|
|
|
|
var bind NativeBind
|
2017-11-17 17:25:45 +01:00
|
|
|
|
2017-11-19 13:14:15 +01:00
|
|
|
port := int(uport)
|
2017-11-17 17:25:45 +01:00
|
|
|
|
2017-11-19 13:14:15 +01:00
|
|
|
bind.ipv4, port, err = listenNet("udp4", port)
|
|
|
|
if err != nil {
|
|
|
|
return nil, 0, err
|
2017-11-17 17:25:45 +01:00
|
|
|
}
|
2017-11-19 13:14:15 +01:00
|
|
|
|
|
|
|
bind.ipv6, port, err = listenNet("udp6", port)
|
2017-11-17 17:25:45 +01:00
|
|
|
if err != nil {
|
2017-11-19 13:14:15 +01:00
|
|
|
bind.ipv4.Close()
|
2017-11-17 17:25:45 +01:00
|
|
|
return nil, 0, err
|
|
|
|
}
|
|
|
|
|
2017-11-19 13:14:15 +01:00
|
|
|
return &bind, uint16(port), nil
|
|
|
|
}
|
2017-11-17 17:25:45 +01:00
|
|
|
|
2017-11-19 13:14:15 +01:00
|
|
|
func (bind *NativeBind) Close() error {
|
|
|
|
err1 := bind.ipv4.Close()
|
|
|
|
err2 := bind.ipv6.Close()
|
|
|
|
if err1 != nil {
|
|
|
|
return err1
|
|
|
|
}
|
|
|
|
return err2
|
|
|
|
}
|
|
|
|
|
|
|
|
func (bind *NativeBind) ReceiveIPv4(buff []byte) (int, Endpoint, error) {
|
|
|
|
n, endpoint, err := bind.ipv4.ReadFromUDP(buff)
|
2018-05-18 05:02:35 +02:00
|
|
|
if endpoint != nil {
|
|
|
|
endpoint.IP = endpoint.IP.To4()
|
|
|
|
}
|
2017-11-19 13:14:15 +01:00
|
|
|
return n, (*NativeEndpoint)(endpoint), err
|
|
|
|
}
|
|
|
|
|
|
|
|
func (bind *NativeBind) ReceiveIPv6(buff []byte) (int, Endpoint, error) {
|
|
|
|
n, endpoint, err := bind.ipv6.ReadFromUDP(buff)
|
|
|
|
return n, (*NativeEndpoint)(endpoint), err
|
|
|
|
}
|
|
|
|
|
|
|
|
func (bind *NativeBind) Send(buff []byte, endpoint Endpoint) error {
|
|
|
|
var err error
|
|
|
|
nend := endpoint.(*NativeEndpoint)
|
2018-05-18 05:02:35 +02:00
|
|
|
if nend.IP.To4() != nil {
|
2017-11-19 13:14:15 +01:00
|
|
|
_, err = bind.ipv4.WriteToUDP(buff, (*net.UDPAddr)(nend))
|
2018-05-18 05:02:35 +02:00
|
|
|
} else {
|
|
|
|
_, err = bind.ipv6.WriteToUDP(buff, (*net.UDPAddr)(nend))
|
2017-11-19 13:14:15 +01:00
|
|
|
}
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2018-05-22 15:17:35 +02:00
|
|
|
var fwmarkIoctl int
|
|
|
|
|
|
|
|
func init() {
|
2018-05-21 17:27:18 +02:00
|
|
|
if runtime.GOOS == "freebsd" {
|
2018-05-22 15:17:35 +02:00
|
|
|
fwmarkIoctl = 0x1015 /* unix.SO_USER_COOKIE */
|
|
|
|
} else if runtime.GOOS == "openbsd" {
|
|
|
|
fwmarkIoctl = 0x1021 /* unix.SO_RTABLE */
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (bind *NativeBind) SetMark(mark uint32) error {
|
|
|
|
if fwmarkIoctl == 0 {
|
2018-05-21 17:27:18 +02:00
|
|
|
return nil
|
|
|
|
}
|
2018-05-22 15:17:35 +02:00
|
|
|
fd4, err1 := bind.ipv4.SyscallConn()
|
|
|
|
fd6, err2 := bind.ipv6.SyscallConn()
|
|
|
|
if err1 != nil {
|
|
|
|
return err1
|
|
|
|
}
|
|
|
|
if err2 != nil {
|
|
|
|
return err2
|
|
|
|
}
|
|
|
|
err3 := fd4.Control(func(fd uintptr) {
|
|
|
|
err1 = unix.SetsockoptInt(int(fd), unix.SOL_SOCKET, fwmarkIoctl, int(mark))
|
|
|
|
})
|
|
|
|
err4 := fd6.Control(func(fd uintptr) {
|
|
|
|
err2 = unix.SetsockoptInt(int(fd), unix.SOL_SOCKET, fwmarkIoctl, int(mark))
|
|
|
|
})
|
|
|
|
if err1 != nil {
|
|
|
|
return err1
|
|
|
|
}
|
|
|
|
if err2 != nil {
|
|
|
|
return err2
|
|
|
|
}
|
|
|
|
if err3 != nil {
|
|
|
|
return err3
|
|
|
|
}
|
|
|
|
if err4 != nil {
|
|
|
|
return err4
|
|
|
|
}
|
2017-11-19 13:14:15 +01:00
|
|
|
return nil
|
2017-11-17 17:25:45 +01:00
|
|
|
}
|