wireguard-go/tun/tun_darwin.go

321 lines
5.7 KiB
Go
Raw Normal View History

2019-01-02 00:55:51 +00:00
/* SPDX-License-Identifier: MIT
*
* Copyright (C) 2017-2020 WireGuard LLC. All Rights Reserved.
*/
2018-05-23 00:10:54 +00:00
package tun
import (
"fmt"
2018-05-14 17:23:44 +00:00
"io/ioutil"
"net"
"os"
2018-05-24 13:29:16 +00:00
"syscall"
"time"
"unsafe"
2019-05-14 07:09:52 +00:00
"golang.org/x/net/ipv6"
"golang.org/x/sys/unix"
)
const utunControlName = "com.apple.net.utun_control"
type NativeTun struct {
2018-05-20 03:43:22 +00:00
name string
tunFile *os.File
events chan Event
2018-05-20 03:43:22 +00:00
errors chan error
routeSocket int
}
func retryInterfaceByIndex(index int) (iface *net.Interface, err error) {
for i := 0; i < 20; i++ {
iface, err = net.InterfaceByIndex(index)
if err != nil {
if opErr, ok := err.(*net.OpError); ok {
if syscallErr, ok := opErr.Err.(*os.SyscallError); ok && syscallErr.Err == syscall.ENOMEM {
time.Sleep(time.Duration(i) * time.Second / 3)
continue
}
}
}
return iface, err
}
return nil, err
}
func (tun *NativeTun) routineRouteListener(tunIfindex int) {
2018-05-20 03:43:22 +00:00
var (
statusUp bool
statusMTU int
)
defer close(tun.events)
2018-05-20 03:43:22 +00:00
data := make([]byte, os.Getpagesize())
for {
2018-05-24 13:29:16 +00:00
retry:
2018-05-20 03:43:22 +00:00
n, err := unix.Read(tun.routeSocket, data)
if err != nil {
2018-05-24 13:29:16 +00:00
if errno, ok := err.(syscall.Errno); ok && errno == syscall.EINTR {
goto retry
}
2018-05-20 03:43:22 +00:00
tun.errors <- err
return
}
if n < 14 {
continue
}
2018-05-21 15:27:18 +00:00
if data[3 /* type */] != unix.RTM_IFINFO {
2018-05-20 03:43:22 +00:00
continue
}
ifindex := int(*(*uint16)(unsafe.Pointer(&data[12 /* ifindex */])))
if ifindex != tunIfindex {
continue
}
iface, err := retryInterfaceByIndex(ifindex)
2018-05-20 03:43:22 +00:00
if err != nil {
tun.errors <- err
return
}
// Up / Down event
up := (iface.Flags & net.FlagUp) != 0
if up != statusUp && up {
tun.events <- EventUp
2018-05-20 03:43:22 +00:00
}
if up != statusUp && !up {
tun.events <- EventDown
2018-05-20 03:43:22 +00:00
}
statusUp = up
// MTU changes
if iface.MTU != statusMTU {
tun.events <- EventMTUUpdate
2018-05-20 03:43:22 +00:00
}
statusMTU = iface.MTU
}
}
func CreateTUN(name string, mtu int) (Device, error) {
ifIndex := -1
2018-05-05 00:48:21 +00:00
if name != "utun" {
_, err := fmt.Sscanf(name, "utun%d", &ifIndex)
if err != nil || ifIndex < 0 {
2018-05-04 19:11:38 +00:00
return nil, fmt.Errorf("Interface name must be utun[0-9]*")
}
}
fd, err := unix.Socket(unix.AF_SYSTEM, unix.SOCK_DGRAM, 2)
if err != nil {
2018-05-03 02:49:35 +00:00
return nil, err
}
ctlInfo := &unix.CtlInfo{}
copy(ctlInfo.Name[:], []byte(utunControlName))
err = unix.IoctlCtlInfo(fd, ctlInfo)
if err != nil {
return nil, fmt.Errorf("IoctlGetCtlInfo: %w", err)
}
sc := &unix.SockaddrCtl{
ID: ctlInfo.Id,
Unit: uint32(ifIndex) + 1,
}
err = unix.Connect(fd, sc)
if err != nil {
return nil, err
}
err = syscall.SetNonblock(fd, true)
if err != nil {
return nil, err
}
2018-05-23 00:10:54 +00:00
tun, err := CreateTUNFromFile(os.NewFile(uintptr(fd), ""), mtu)
if err == nil && name == "utun" {
2018-05-22 13:17:35 +00:00
fname := os.Getenv("WG_TUN_NAME_FILE")
2018-05-14 17:23:44 +00:00
if fname != "" {
ioutil.WriteFile(fname, []byte(tun.(*NativeTun).name+"\n"), 0400)
2018-05-14 17:23:44 +00:00
}
}
return tun, err
2018-05-03 02:49:35 +00:00
}
func CreateTUNFromFile(file *os.File, mtu int) (Device, error) {
tun := &NativeTun{
tunFile: file,
events: make(chan Event, 10),
2019-03-03 04:20:13 +00:00
errors: make(chan error, 5),
}
2018-05-20 03:43:22 +00:00
name, err := tun.Name()
if err != nil {
tun.tunFile.Close()
2018-05-20 03:43:22 +00:00
return nil, err
}
tunIfindex, err := func() (int, error) {
iface, err := net.InterfaceByName(name)
if err != nil {
return -1, err
}
return iface.Index, nil
}()
2018-05-03 02:49:35 +00:00
if err != nil {
tun.tunFile.Close()
2018-05-03 02:49:35 +00:00
return nil, err
}
2018-05-20 03:43:22 +00:00
tun.routeSocket, err = unix.Socket(unix.AF_ROUTE, unix.SOCK_RAW, unix.AF_UNSPEC)
if err != nil {
tun.tunFile.Close()
2018-05-20 03:43:22 +00:00
return nil, err
}
2018-05-23 00:10:54 +00:00
go tun.routineRouteListener(tunIfindex)
2019-03-03 04:20:13 +00:00
if mtu > 0 {
err = tun.setMTU(mtu)
if err != nil {
tun.Close()
return nil, err
}
2018-05-14 10:27:29 +00:00
}
2018-05-14 01:55:46 +00:00
return tun, nil
}
func (tun *NativeTun) Name() (string, error) {
var err error
tun.operateOnFd(func(fd uintptr) {
tun.name, err = unix.GetsockoptString(
int(fd),
2, /* #define SYSPROTO_CONTROL 2 */
2, /* #define UTUN_OPT_IFNAME 2 */
)
})
if err != nil {
return "", fmt.Errorf("GetSockoptString: %w", err)
}
2018-05-03 02:49:35 +00:00
return tun.name, nil
}
func (tun *NativeTun) File() *os.File {
return tun.tunFile
2018-05-03 02:49:35 +00:00
}
func (tun *NativeTun) Events() chan Event {
2018-05-03 02:49:35 +00:00
return tun.events
}
func (tun *NativeTun) Read(buff []byte, offset int) (int, error) {
2018-05-14 01:55:46 +00:00
select {
case err := <-tun.errors:
2018-05-03 02:49:35 +00:00
return 0, err
2018-05-14 01:55:46 +00:00
default:
buff := buff[offset-4:]
n, err := tun.tunFile.Read(buff[:])
2018-05-14 01:55:46 +00:00
if n < 4 {
return 0, err
}
return n - 4, err
}
}
func (tun *NativeTun) Write(buff []byte, offset int) (int, error) {
2018-05-03 02:49:35 +00:00
// reserve space for header
2018-05-03 02:49:35 +00:00
buff = buff[offset-4:]
2018-05-03 02:49:35 +00:00
// add packet information header
2018-05-03 02:49:35 +00:00
buff[0] = 0x00
buff[1] = 0x00
buff[2] = 0x00
2018-05-03 02:49:35 +00:00
if buff[4]>>4 == ipv6.Version {
buff[3] = unix.AF_INET6
} else {
buff[3] = unix.AF_INET
}
2018-05-03 02:49:35 +00:00
// write
return tun.tunFile.Write(buff)
}
2019-03-21 20:43:04 +00:00
func (tun *NativeTun) Flush() error {
2019-06-06 21:00:15 +00:00
// TODO: can flushing be implemented by buffering and using sendmmsg?
2019-03-21 20:43:04 +00:00
return nil
}
func (tun *NativeTun) Close() error {
var err2 error
err1 := tun.tunFile.Close()
2018-05-20 03:43:22 +00:00
if tun.routeSocket != -1 {
unix.Shutdown(tun.routeSocket, unix.SHUT_RDWR)
err2 = unix.Close(tun.routeSocket)
} else if tun.events != nil {
close(tun.events)
2018-05-20 03:43:22 +00:00
}
2018-05-14 01:55:46 +00:00
if err1 != nil {
return err1
}
return err2
}
func (tun *NativeTun) setMTU(n int) error {
fd, err := unix.Socket(
unix.AF_INET,
unix.SOCK_DGRAM,
0,
)
if err != nil {
return err
}
defer unix.Close(fd)
var ifr unix.IfreqMTU
copy(ifr.Name[:], tun.name)
ifr.MTU = int32(n)
err = unix.IoctlSetIfreqMTU(fd, &ifr)
if err != nil {
return fmt.Errorf("failed to set MTU on %s: %w", tun.name, err)
}
return nil
}
func (tun *NativeTun) MTU() (int, error) {
fd, err := unix.Socket(
unix.AF_INET,
unix.SOCK_DGRAM,
0,
)
if err != nil {
return 0, err
}
defer unix.Close(fd)
ifr, err := unix.IoctlGetIfreqMTU(fd, tun.name)
if err != nil {
return 0, fmt.Errorf("failed to get MTU on %s: %w", tun.name, err)
}
return int(ifr.MTU), nil
}