From 366cbd11a40a6a3fde5d0dc803ca6895859ed188 Mon Sep 17 00:00:00 2001 From: "Jason A. Donenfeld" Date: Wed, 27 Feb 2019 01:48:58 +0100 Subject: 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") } --- tun/tun_openbsd.go | 37 +++++-------------------------------- 1 file changed, 5 insertions(+), 32 deletions(-) (limited to 'tun/tun_openbsd.go') diff --git a/tun/tun_openbsd.go b/tun/tun_openbsd.go index b10c460..4e740c5 100644 --- a/tun/tun_openbsd.go +++ b/tun/tun_openbsd.go @@ -6,11 +6,9 @@ package tun import ( - "errors" "fmt" "golang.org/x/net/ipv6" "golang.org/x/sys/unix" - "golang.zx2c4.com/wireguard/rwcancel" "io/ioutil" "net" "os" @@ -30,7 +28,6 @@ const _TUNSIFMODE = 0x8004745d type nativeTun struct { name string tunFile *os.File - rwcancel *rwcancel.RWCancel events chan TUNEvent errors chan error routeSocket int @@ -167,14 +164,6 @@ func CreateTUNFromFile(file *os.File, mtu int) (TUNDevice, error) { return nil, err } - tun.operateOnFd(func(fd uintptr) { - tun.rwcancel, err = rwcancel.NewRWCancel(int(fd)) - }) - if err != nil { - tun.tunFile.Close() - return nil, err - } - tun.routeSocket, err = unix.Socket(unix.AF_ROUTE, unix.SOCK_RAW, unix.AF_UNSPEC) if err != nil { tun.tunFile.Close() @@ -211,7 +200,7 @@ func (tun *nativeTun) Events() chan TUNEvent { return tun.events } -func (tun *nativeTun) doRead(buff []byte, offset int) (int, error) { +func (tun *nativeTun) Read(buff []byte, offset int) (int, error) { select { case err := <-tun.errors: return 0, err @@ -225,18 +214,6 @@ func (tun *nativeTun) doRead(buff []byte, offset int) (int, error) { } } -func (tun *nativeTun) Read(buff []byte, offset int) (int, error) { - for { - n, err := tun.doRead(buff, offset) - if err == nil || !rwcancel.RetryAfterError(err) { - return n, err - } - if !tun.rwcancel.ReadyRead() { - return 0, errors.New("tun device closed") - } - } -} - func (tun *nativeTun) Write(buff []byte, offset int) (int, error) { // reserve space for header @@ -261,12 +238,11 @@ func (tun *nativeTun) Write(buff []byte, offset int) (int, error) { } func (tun *nativeTun) Close() error { - var err3 error - err1 := tun.rwcancel.Cancel() - err2 := tun.tunFile.Close() + var err2 error + err1 := tun.tunFile.Close() if tun.routeSocket != -1 { unix.Shutdown(tun.routeSocket, unix.SHUT_RDWR) - err3 = unix.Close(tun.routeSocket) + err2 = unix.Close(tun.routeSocket) tun.routeSocket = -1 } else if tun.events != nil { close(tun.events) @@ -274,10 +250,7 @@ func (tun *nativeTun) Close() error { if err1 != nil { return err1 } - if err2 != nil { - return err2 - } - return err3 + return err2 } func (tun *nativeTun) setMTU(n int) error { -- cgit v1.2.3