2019-01-02 00:55:51 +00:00
|
|
|
/* SPDX-License-Identifier: MIT
|
2018-05-22 13:17:35 +00:00
|
|
|
*
|
2019-01-02 00:55:51 +00:00
|
|
|
* Copyright (C) 2017-2019 WireGuard LLC. All Rights Reserved.
|
2018-05-22 13:17:35 +00:00
|
|
|
*/
|
|
|
|
|
2018-05-23 00:10:54 +00:00
|
|
|
package tun
|
2018-05-22 13:17:35 +00:00
|
|
|
|
|
|
|
import (
|
|
|
|
"fmt"
|
|
|
|
"io/ioutil"
|
|
|
|
"net"
|
|
|
|
"os"
|
|
|
|
"syscall"
|
|
|
|
"unsafe"
|
2019-05-14 07:09:52 +00:00
|
|
|
|
|
|
|
"golang.org/x/net/ipv6"
|
|
|
|
"golang.org/x/sys/unix"
|
2018-05-22 13:17:35 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
// Structure for iface mtu get/set ioctls
|
|
|
|
type ifreq_mtu struct {
|
|
|
|
Name [unix.IFNAMSIZ]byte
|
|
|
|
MTU uint32
|
|
|
|
Pad0 [12]byte
|
|
|
|
}
|
|
|
|
|
|
|
|
const _TUNSIFMODE = 0x8004745d
|
|
|
|
|
2019-02-28 23:05:57 +00:00
|
|
|
type NativeTun struct {
|
2018-05-22 13:17:35 +00:00
|
|
|
name string
|
2018-10-17 19:26:53 +00:00
|
|
|
tunFile *os.File
|
2018-05-22 13:17:35 +00:00
|
|
|
events chan TUNEvent
|
|
|
|
errors chan error
|
|
|
|
routeSocket int
|
|
|
|
}
|
|
|
|
|
2019-02-28 23:05:57 +00:00
|
|
|
func (tun *NativeTun) routineRouteListener(tunIfindex int) {
|
2018-05-22 13:17:35 +00:00
|
|
|
var (
|
|
|
|
statusUp bool
|
|
|
|
statusMTU int
|
|
|
|
)
|
|
|
|
|
|
|
|
defer close(tun.events)
|
|
|
|
|
|
|
|
data := make([]byte, os.Getpagesize())
|
|
|
|
for {
|
2018-05-24 13:29:16 +00:00
|
|
|
retry:
|
2018-05-22 13:17:35 +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-22 13:17:35 +00:00
|
|
|
tun.errors <- err
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
if n < 8 {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
if data[3 /* type */] != unix.RTM_IFINFO {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
ifindex := int(*(*uint16)(unsafe.Pointer(&data[6 /* ifindex */])))
|
|
|
|
if ifindex != tunIfindex {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
|
|
|
|
iface, err := net.InterfaceByIndex(ifindex)
|
|
|
|
if err != nil {
|
|
|
|
tun.errors <- err
|
|
|
|
return
|
|
|
|
}
|
|
|
|
|
|
|
|
// Up / Down event
|
|
|
|
up := (iface.Flags & net.FlagUp) != 0
|
|
|
|
if up != statusUp && up {
|
|
|
|
tun.events <- TUNEventUp
|
|
|
|
}
|
|
|
|
if up != statusUp && !up {
|
|
|
|
tun.events <- TUNEventDown
|
|
|
|
}
|
|
|
|
statusUp = up
|
|
|
|
|
|
|
|
// MTU changes
|
|
|
|
if iface.MTU != statusMTU {
|
|
|
|
tun.events <- TUNEventMTUUpdate
|
|
|
|
}
|
|
|
|
statusMTU = iface.MTU
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func errorIsEBUSY(err error) bool {
|
|
|
|
if pe, ok := err.(*os.PathError); ok {
|
2018-05-24 13:29:16 +00:00
|
|
|
err = pe.Err
|
2018-05-22 13:17:35 +00:00
|
|
|
}
|
|
|
|
if errno, ok := err.(syscall.Errno); ok && errno == syscall.EBUSY {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
2018-05-23 00:10:54 +00:00
|
|
|
func CreateTUN(name string, mtu int) (TUNDevice, error) {
|
2018-05-22 13:17:35 +00:00
|
|
|
ifIndex := -1
|
|
|
|
if name != "tun" {
|
|
|
|
_, err := fmt.Sscanf(name, "tun%d", &ifIndex)
|
|
|
|
if err != nil || ifIndex < 0 {
|
|
|
|
return nil, fmt.Errorf("Interface name must be tun[0-9]*")
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
var tunfile *os.File
|
|
|
|
var err error
|
|
|
|
|
|
|
|
if ifIndex != -1 {
|
|
|
|
tunfile, err = os.OpenFile(fmt.Sprintf("/dev/tun%d", ifIndex), unix.O_RDWR, 0)
|
|
|
|
} else {
|
|
|
|
for ifIndex = 0; ifIndex < 256; ifIndex += 1 {
|
|
|
|
tunfile, err = os.OpenFile(fmt.Sprintf("/dev/tun%d", ifIndex), unix.O_RDWR, 0)
|
|
|
|
if err == nil || !errorIsEBUSY(err) {
|
|
|
|
break
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2018-05-23 00:10:54 +00:00
|
|
|
tun, err := CreateTUNFromFile(tunfile, mtu)
|
2018-05-22 13:17:35 +00:00
|
|
|
|
|
|
|
if err == nil && name == "tun" {
|
|
|
|
fname := os.Getenv("WG_TUN_NAME_FILE")
|
|
|
|
if fname != "" {
|
2019-02-28 23:05:57 +00:00
|
|
|
ioutil.WriteFile(fname, []byte(tun.(*NativeTun).name+"\n"), 0400)
|
2018-05-22 13:17:35 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return tun, err
|
|
|
|
}
|
|
|
|
|
2018-05-23 00:10:54 +00:00
|
|
|
func CreateTUNFromFile(file *os.File, mtu int) (TUNDevice, error) {
|
2018-05-22 13:17:35 +00:00
|
|
|
|
2019-02-28 23:05:57 +00:00
|
|
|
tun := &NativeTun{
|
2018-10-17 19:26:53 +00:00
|
|
|
tunFile: file,
|
|
|
|
events: make(chan TUNEvent, 10),
|
|
|
|
errors: make(chan error, 1),
|
2018-05-22 13:17:35 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
name, err := tun.Name()
|
|
|
|
if err != nil {
|
2018-10-17 19:26:53 +00:00
|
|
|
tun.tunFile.Close()
|
2018-05-22 13:17:35 +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
|
|
|
|
}()
|
|
|
|
if err != nil {
|
2018-10-17 19:26:53 +00:00
|
|
|
tun.tunFile.Close()
|
2018-05-22 13:17:35 +00:00
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
tun.routeSocket, err = unix.Socket(unix.AF_ROUTE, unix.SOCK_RAW, unix.AF_UNSPEC)
|
|
|
|
if err != nil {
|
2018-10-17 19:26:53 +00:00
|
|
|
tun.tunFile.Close()
|
2018-05-22 13:17:35 +00:00
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
2018-05-23 00:10:54 +00:00
|
|
|
go tun.routineRouteListener(tunIfindex)
|
2018-05-22 13:17:35 +00:00
|
|
|
|
2018-05-23 00:10:54 +00:00
|
|
|
err = tun.setMTU(mtu)
|
2018-05-22 13:17:35 +00:00
|
|
|
if err != nil {
|
|
|
|
tun.Close()
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
return tun, nil
|
|
|
|
}
|
|
|
|
|
2019-02-28 23:05:57 +00:00
|
|
|
func (tun *NativeTun) Name() (string, error) {
|
2018-10-17 19:26:53 +00:00
|
|
|
gostat, err := tun.tunFile.Stat()
|
2018-05-22 13:17:35 +00:00
|
|
|
if err != nil {
|
|
|
|
tun.name = ""
|
|
|
|
return "", err
|
|
|
|
}
|
|
|
|
stat := gostat.Sys().(*syscall.Stat_t)
|
|
|
|
tun.name = fmt.Sprintf("tun%d", stat.Rdev%256)
|
|
|
|
return tun.name, nil
|
|
|
|
}
|
|
|
|
|
2019-02-28 23:05:57 +00:00
|
|
|
func (tun *NativeTun) File() *os.File {
|
2018-10-17 19:26:53 +00:00
|
|
|
return tun.tunFile
|
2018-05-22 13:17:35 +00:00
|
|
|
}
|
|
|
|
|
2019-02-28 23:05:57 +00:00
|
|
|
func (tun *NativeTun) Events() chan TUNEvent {
|
2018-05-22 13:17:35 +00:00
|
|
|
return tun.events
|
|
|
|
}
|
|
|
|
|
2019-02-28 23:05:57 +00:00
|
|
|
func (tun *NativeTun) Read(buff []byte, offset int) (int, error) {
|
2018-05-22 13:17:35 +00:00
|
|
|
select {
|
|
|
|
case err := <-tun.errors:
|
|
|
|
return 0, err
|
|
|
|
default:
|
|
|
|
buff := buff[offset-4:]
|
2018-10-17 19:26:53 +00:00
|
|
|
n, err := tun.tunFile.Read(buff[:])
|
2018-05-22 13:17:35 +00:00
|
|
|
if n < 4 {
|
|
|
|
return 0, err
|
|
|
|
}
|
|
|
|
return n - 4, err
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2019-02-28 23:05:57 +00:00
|
|
|
func (tun *NativeTun) Write(buff []byte, offset int) (int, error) {
|
2018-05-22 13:17:35 +00:00
|
|
|
|
|
|
|
// reserve space for header
|
|
|
|
|
|
|
|
buff = buff[offset-4:]
|
|
|
|
|
|
|
|
// add packet information header
|
|
|
|
|
|
|
|
buff[0] = 0x00
|
|
|
|
buff[1] = 0x00
|
|
|
|
buff[2] = 0x00
|
|
|
|
|
|
|
|
if buff[4]>>4 == ipv6.Version {
|
|
|
|
buff[3] = unix.AF_INET6
|
|
|
|
} else {
|
|
|
|
buff[3] = unix.AF_INET
|
|
|
|
}
|
|
|
|
|
|
|
|
// write
|
|
|
|
|
2018-10-17 19:26:53 +00:00
|
|
|
return tun.tunFile.Write(buff)
|
2018-05-22 13:17:35 +00:00
|
|
|
}
|
|
|
|
|
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
|
|
|
|
}
|
|
|
|
|
2019-02-28 23:05:57 +00:00
|
|
|
func (tun *NativeTun) Close() error {
|
tun: use netpoll instead of rwcancel
The new sysconn function of Go 1.12 makes this possible:
package main
import "log"
import "os"
import "unsafe"
import "time"
import "syscall"
import "sync"
import "golang.org/x/sys/unix"
func main() {
fd, err := os.OpenFile("/dev/net/tun", os.O_RDWR, 0)
if err != nil {
log.Fatal(err)
}
var ifr [unix.IFNAMSIZ + 64]byte
copy(ifr[:], []byte("cheese"))
*(*uint16)(unsafe.Pointer(&ifr[unix.IFNAMSIZ])) = unix.IFF_TUN
var errno syscall.Errno
s, _ := fd.SyscallConn()
s.Control(func(fd uintptr) {
_, _, errno = unix.Syscall(
unix.SYS_IOCTL,
fd,
uintptr(unix.TUNSETIFF),
uintptr(unsafe.Pointer(&ifr[0])),
)
})
if errno != 0 {
log.Fatal(errno)
}
b := [4]byte{}
wait := sync.WaitGroup{}
wait.Add(1)
go func() {
_, err := fd.Read(b[:])
log.Print("Read errored: ", err)
wait.Done()
}()
time.Sleep(time.Second)
log.Print("Closing")
err = fd.Close()
if err != nil {
log.Print("Close errored: " , err)
}
wait.Wait()
log.Print("Exiting")
}
2019-02-27 00:48:58 +00:00
|
|
|
var err2 error
|
|
|
|
err1 := tun.tunFile.Close()
|
2018-05-22 13:17:35 +00:00
|
|
|
if tun.routeSocket != -1 {
|
|
|
|
unix.Shutdown(tun.routeSocket, unix.SHUT_RDWR)
|
tun: use netpoll instead of rwcancel
The new sysconn function of Go 1.12 makes this possible:
package main
import "log"
import "os"
import "unsafe"
import "time"
import "syscall"
import "sync"
import "golang.org/x/sys/unix"
func main() {
fd, err := os.OpenFile("/dev/net/tun", os.O_RDWR, 0)
if err != nil {
log.Fatal(err)
}
var ifr [unix.IFNAMSIZ + 64]byte
copy(ifr[:], []byte("cheese"))
*(*uint16)(unsafe.Pointer(&ifr[unix.IFNAMSIZ])) = unix.IFF_TUN
var errno syscall.Errno
s, _ := fd.SyscallConn()
s.Control(func(fd uintptr) {
_, _, errno = unix.Syscall(
unix.SYS_IOCTL,
fd,
uintptr(unix.TUNSETIFF),
uintptr(unsafe.Pointer(&ifr[0])),
)
})
if errno != 0 {
log.Fatal(errno)
}
b := [4]byte{}
wait := sync.WaitGroup{}
wait.Add(1)
go func() {
_, err := fd.Read(b[:])
log.Print("Read errored: ", err)
wait.Done()
}()
time.Sleep(time.Second)
log.Print("Closing")
err = fd.Close()
if err != nil {
log.Print("Close errored: " , err)
}
wait.Wait()
log.Print("Exiting")
}
2019-02-27 00:48:58 +00:00
|
|
|
err2 = unix.Close(tun.routeSocket)
|
2018-05-22 13:17:35 +00:00
|
|
|
tun.routeSocket = -1
|
|
|
|
} else if tun.events != nil {
|
|
|
|
close(tun.events)
|
|
|
|
}
|
|
|
|
if err1 != nil {
|
|
|
|
return err1
|
|
|
|
}
|
tun: use netpoll instead of rwcancel
The new sysconn function of Go 1.12 makes this possible:
package main
import "log"
import "os"
import "unsafe"
import "time"
import "syscall"
import "sync"
import "golang.org/x/sys/unix"
func main() {
fd, err := os.OpenFile("/dev/net/tun", os.O_RDWR, 0)
if err != nil {
log.Fatal(err)
}
var ifr [unix.IFNAMSIZ + 64]byte
copy(ifr[:], []byte("cheese"))
*(*uint16)(unsafe.Pointer(&ifr[unix.IFNAMSIZ])) = unix.IFF_TUN
var errno syscall.Errno
s, _ := fd.SyscallConn()
s.Control(func(fd uintptr) {
_, _, errno = unix.Syscall(
unix.SYS_IOCTL,
fd,
uintptr(unix.TUNSETIFF),
uintptr(unsafe.Pointer(&ifr[0])),
)
})
if errno != 0 {
log.Fatal(errno)
}
b := [4]byte{}
wait := sync.WaitGroup{}
wait.Add(1)
go func() {
_, err := fd.Read(b[:])
log.Print("Read errored: ", err)
wait.Done()
}()
time.Sleep(time.Second)
log.Print("Closing")
err = fd.Close()
if err != nil {
log.Print("Close errored: " , err)
}
wait.Wait()
log.Print("Exiting")
}
2019-02-27 00:48:58 +00:00
|
|
|
return err2
|
2018-05-22 13:17:35 +00:00
|
|
|
}
|
|
|
|
|
2019-02-28 23:05:57 +00:00
|
|
|
func (tun *NativeTun) setMTU(n int) error {
|
2018-05-22 13:17:35 +00:00
|
|
|
// open datagram socket
|
|
|
|
|
|
|
|
var fd int
|
|
|
|
|
|
|
|
fd, err := unix.Socket(
|
|
|
|
unix.AF_INET,
|
|
|
|
unix.SOCK_DGRAM,
|
|
|
|
0,
|
|
|
|
)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
defer unix.Close(fd)
|
|
|
|
|
|
|
|
// do ioctl call
|
|
|
|
|
|
|
|
var ifr ifreq_mtu
|
|
|
|
copy(ifr.Name[:], tun.name)
|
|
|
|
ifr.MTU = uint32(n)
|
|
|
|
|
|
|
|
_, _, errno := unix.Syscall(
|
|
|
|
unix.SYS_IOCTL,
|
|
|
|
uintptr(fd),
|
|
|
|
uintptr(unix.SIOCSIFMTU),
|
|
|
|
uintptr(unsafe.Pointer(&ifr)),
|
|
|
|
)
|
|
|
|
|
|
|
|
if errno != 0 {
|
|
|
|
return fmt.Errorf("failed to set MTU on %s", tun.name)
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2019-02-28 23:05:57 +00:00
|
|
|
func (tun *NativeTun) MTU() (int, error) {
|
2018-05-22 13:17:35 +00:00
|
|
|
// open datagram socket
|
|
|
|
|
|
|
|
fd, err := unix.Socket(
|
|
|
|
unix.AF_INET,
|
|
|
|
unix.SOCK_DGRAM,
|
|
|
|
0,
|
|
|
|
)
|
|
|
|
|
|
|
|
if err != nil {
|
|
|
|
return 0, err
|
|
|
|
}
|
|
|
|
|
|
|
|
defer unix.Close(fd)
|
|
|
|
|
|
|
|
// do ioctl call
|
|
|
|
var ifr ifreq_mtu
|
|
|
|
copy(ifr.Name[:], tun.name)
|
|
|
|
|
|
|
|
_, _, errno := unix.Syscall(
|
|
|
|
unix.SYS_IOCTL,
|
|
|
|
uintptr(fd),
|
|
|
|
uintptr(unix.SIOCGIFMTU),
|
|
|
|
uintptr(unsafe.Pointer(&ifr)),
|
|
|
|
)
|
|
|
|
if errno != 0 {
|
|
|
|
return 0, fmt.Errorf("failed to get MTU on %s", tun.name)
|
|
|
|
}
|
|
|
|
|
2018-05-22 16:33:50 +00:00
|
|
|
return int(*(*int32)(unsafe.Pointer(&ifr.MTU))), nil
|
2018-05-22 13:17:35 +00:00
|
|
|
}
|