mirror of
https://github.com/KusakabeShi/EtherGuard-VPN.git
synced 2024-11-22 23:33:09 +01:00
401 lines
9.3 KiB
Go
401 lines
9.3 KiB
Go
/* SPDX-License-Identifier: MIT
|
|
*
|
|
* Copyright (C) 2018-2019 WireGuard LLC. All Rights Reserved.
|
|
*/
|
|
|
|
package tun
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"sync"
|
|
"sync/atomic"
|
|
"time"
|
|
"unsafe"
|
|
|
|
"golang.org/x/sys/windows"
|
|
"golang.zx2c4.com/wireguard/tun/wintun"
|
|
)
|
|
|
|
const (
|
|
packetExchangeAlignment uint32 = 4 // Number of bytes packets are aligned to in exchange buffers
|
|
packetSizeMax uint32 = 0xf000 - packetExchangeAlignment // Maximum packet size
|
|
packetExchangeSize uint32 = 0x100000 // Exchange buffer size (defaults to 1MiB)
|
|
retryRate = 4 // Number of retries per second to reopen device pipe
|
|
retryTimeout = 30 // Number of seconds to tolerate adapter unavailable
|
|
)
|
|
|
|
type exchgBufRead struct {
|
|
data [packetExchangeSize]byte
|
|
offset uint32
|
|
avail uint32
|
|
}
|
|
|
|
type exchgBufWrite struct {
|
|
data [packetExchangeSize]byte
|
|
offset uint32
|
|
}
|
|
|
|
type NativeTun struct {
|
|
wt *wintun.Wintun
|
|
tunFileRead *os.File
|
|
tunFileWrite *os.File
|
|
haveRegisteredWriteBuffer int32
|
|
tunLock sync.Mutex
|
|
close bool
|
|
rdBuff *exchgBufRead
|
|
wrBuff *exchgBufWrite
|
|
events chan Event
|
|
errors chan error
|
|
forcedMTU int
|
|
}
|
|
|
|
func packetAlign(size uint32) uint32 {
|
|
return (size + (packetExchangeAlignment - 1)) &^ (packetExchangeAlignment - 1)
|
|
}
|
|
|
|
var shouldRetryOpen = windows.RtlGetVersion().MajorVersion < 10
|
|
|
|
func maybeRetry(x int) int {
|
|
if shouldRetryOpen {
|
|
return x
|
|
}
|
|
return 0
|
|
}
|
|
|
|
//
|
|
// CreateTUN creates a Wintun adapter with the given name. Should a Wintun
|
|
// adapter with the same name exist, it is reused.
|
|
//
|
|
func CreateTUN(ifname string) (Device, error) {
|
|
return CreateTUNWithRequestedGUID(ifname, nil)
|
|
}
|
|
|
|
//
|
|
// CreateTUNWithRequestedGUID creates a Wintun adapter with the given name and
|
|
// a requested GUID. Should a Wintun adapter with the same name exist, it is reused.
|
|
//
|
|
func CreateTUNWithRequestedGUID(ifname string, requestedGUID *windows.GUID) (Device, error) {
|
|
var err error
|
|
var wt *wintun.Wintun
|
|
|
|
// Does an interface with this name already exist?
|
|
wt, err = wintun.GetInterface(ifname)
|
|
if err == nil {
|
|
// If so, we delete it, in case it has weird residual configuration.
|
|
_, err = wt.DeleteInterface()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("Unable to delete already existing Wintun interface: %v", err)
|
|
}
|
|
} else if err == windows.ERROR_ALREADY_EXISTS {
|
|
return nil, fmt.Errorf("Foreign network interface with the same name exists")
|
|
}
|
|
wt, _, err = wintun.CreateInterface("WireGuard Tunnel Adapter", requestedGUID)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("Unable to create Wintun interface: %v", err)
|
|
}
|
|
|
|
err = wt.SetInterfaceName(ifname)
|
|
if err != nil {
|
|
wt.DeleteInterface()
|
|
return nil, fmt.Errorf("Unable to set name of Wintun interface: %v", err)
|
|
}
|
|
|
|
return &NativeTun{
|
|
wt: wt,
|
|
rdBuff: &exchgBufRead{},
|
|
wrBuff: &exchgBufWrite{},
|
|
events: make(chan Event, 10),
|
|
errors: make(chan error, 1),
|
|
forcedMTU: 1500,
|
|
}, nil
|
|
}
|
|
|
|
func (tun *NativeTun) openTUN() error {
|
|
retries := maybeRetry(retryTimeout * retryRate)
|
|
if tun.close {
|
|
return os.ErrClosed
|
|
}
|
|
|
|
var err error
|
|
name := tun.wt.DataFileName()
|
|
for tun.tunFileRead == nil {
|
|
tun.tunFileRead, err = os.OpenFile(name, os.O_RDONLY, 0)
|
|
if err != nil {
|
|
if retries > 0 && !tun.close {
|
|
time.Sleep(time.Second / retryRate)
|
|
retries--
|
|
continue
|
|
}
|
|
return err
|
|
}
|
|
}
|
|
for tun.tunFileWrite == nil {
|
|
tun.tunFileWrite, err = os.OpenFile(name, os.O_WRONLY, 0)
|
|
if err != nil {
|
|
if retries > 0 && !tun.close {
|
|
time.Sleep(time.Second / retryRate)
|
|
retries--
|
|
continue
|
|
}
|
|
return err
|
|
}
|
|
atomic.StoreInt32(&tun.haveRegisteredWriteBuffer, 0)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (tun *NativeTun) closeTUN() (err error) {
|
|
for tun.tunFileRead != nil {
|
|
tun.tunLock.Lock()
|
|
if tun.tunFileRead == nil {
|
|
tun.tunLock.Unlock()
|
|
break
|
|
}
|
|
t := tun.tunFileRead
|
|
tun.tunFileRead = nil
|
|
windows.CancelIoEx(windows.Handle(t.Fd()), nil)
|
|
err = t.Close()
|
|
tun.tunLock.Unlock()
|
|
break
|
|
}
|
|
for tun.tunFileWrite != nil {
|
|
tun.tunLock.Lock()
|
|
if tun.tunFileWrite == nil {
|
|
tun.tunLock.Unlock()
|
|
break
|
|
}
|
|
t := tun.tunFileWrite
|
|
tun.tunFileWrite = nil
|
|
windows.CancelIoEx(windows.Handle(t.Fd()), nil)
|
|
err2 := t.Close()
|
|
tun.tunLock.Unlock()
|
|
if err == nil {
|
|
err = err2
|
|
}
|
|
break
|
|
}
|
|
return
|
|
}
|
|
|
|
func (tun *NativeTun) getTUN() (read *os.File, write *os.File, err error) {
|
|
read, write = tun.tunFileRead, tun.tunFileWrite
|
|
if read == nil || write == nil {
|
|
read, write = nil, nil
|
|
tun.tunLock.Lock()
|
|
if tun.tunFileRead != nil && tun.tunFileWrite != nil {
|
|
read, write = tun.tunFileRead, tun.tunFileWrite
|
|
tun.tunLock.Unlock()
|
|
return
|
|
}
|
|
err = tun.closeTUN()
|
|
if err != nil {
|
|
tun.tunLock.Unlock()
|
|
return
|
|
}
|
|
err = tun.openTUN()
|
|
if err == nil {
|
|
read, write = tun.tunFileRead, tun.tunFileWrite
|
|
}
|
|
tun.tunLock.Unlock()
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
func (tun *NativeTun) Name() (string, error) {
|
|
return tun.wt.InterfaceName()
|
|
}
|
|
|
|
func (tun *NativeTun) File() *os.File {
|
|
return nil
|
|
}
|
|
|
|
func (tun *NativeTun) Events() chan Event {
|
|
return tun.events
|
|
}
|
|
|
|
func (tun *NativeTun) Close() error {
|
|
tun.close = true
|
|
err1 := tun.closeTUN()
|
|
|
|
if tun.events != nil {
|
|
close(tun.events)
|
|
}
|
|
|
|
_, err2 := tun.wt.DeleteInterface()
|
|
if err1 == nil {
|
|
err1 = err2
|
|
}
|
|
|
|
return err1
|
|
}
|
|
|
|
func (tun *NativeTun) MTU() (int, error) {
|
|
return tun.forcedMTU, nil
|
|
}
|
|
|
|
// TODO: This is a temporary hack. We really need to be monitoring the interface in real time and adapting to MTU changes.
|
|
func (tun *NativeTun) ForceMTU(mtu int) {
|
|
tun.forcedMTU = mtu
|
|
}
|
|
|
|
func (tun *NativeTun) Read(buff []byte, offset int) (int, error) {
|
|
select {
|
|
case err := <-tun.errors:
|
|
return 0, err
|
|
default:
|
|
}
|
|
|
|
retries := maybeRetry(1000)
|
|
for {
|
|
if tun.rdBuff.offset+packetExchangeAlignment <= tun.rdBuff.avail {
|
|
// Get packet from the exchange buffer.
|
|
packet := tun.rdBuff.data[tun.rdBuff.offset:]
|
|
size := *(*uint32)(unsafe.Pointer(&packet[0]))
|
|
pSize := packetAlign(size) + packetExchangeAlignment
|
|
if packetSizeMax < size || tun.rdBuff.avail < tun.rdBuff.offset+pSize {
|
|
// Invalid packet size.
|
|
tun.rdBuff.avail = 0
|
|
continue
|
|
}
|
|
packet = packet[packetExchangeAlignment : packetExchangeAlignment+size]
|
|
|
|
// Copy data.
|
|
copy(buff[offset:], packet)
|
|
tun.rdBuff.offset += pSize
|
|
return int(size), nil
|
|
}
|
|
|
|
// Get TUN data pipe.
|
|
file, _, err := tun.getTUN()
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
n, err := file.Read(tun.rdBuff.data[:])
|
|
if err != nil {
|
|
tun.rdBuff.offset = 0
|
|
tun.rdBuff.avail = 0
|
|
pe, ok := err.(*os.PathError)
|
|
if tun.close {
|
|
return 0, os.ErrClosed
|
|
}
|
|
if retries > 0 && ok && (pe.Err == windows.ERROR_HANDLE_EOF || pe.Err == windows.ERROR_OPERATION_ABORTED) {
|
|
retries--
|
|
tun.closeTUN()
|
|
time.Sleep(time.Millisecond * 2)
|
|
continue
|
|
}
|
|
return 0, err
|
|
}
|
|
if n == 0 {
|
|
if retries == 0 {
|
|
return 0, io.ErrShortBuffer
|
|
}
|
|
retries--
|
|
continue
|
|
}
|
|
tun.rdBuff.offset = 0
|
|
tun.rdBuff.avail = uint32(n)
|
|
}
|
|
}
|
|
|
|
// Note: flush() and putTunPacket() assume the caller comes only from a single thread; there's no locking.
|
|
|
|
func (tun *NativeTun) Flush() error {
|
|
if tun.wrBuff.offset == 0 {
|
|
return nil
|
|
}
|
|
defer func() {
|
|
tun.wrBuff.offset = 0
|
|
}()
|
|
retries := maybeRetry(1000)
|
|
|
|
for {
|
|
// Get TUN data pipe.
|
|
_, file, err := tun.getTUN()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if atomic.CompareAndSwapInt32(&tun.haveRegisteredWriteBuffer, 0, 1) {
|
|
firstSize := (*uint32)(unsafe.Pointer(&tun.wrBuff.data[0]))
|
|
saved := *firstSize
|
|
*firstSize = 0
|
|
// Set the maximum buffer length with an invalid write.
|
|
tun.tunFileWrite.Write(tun.wrBuff.data[:])
|
|
*firstSize = saved
|
|
}
|
|
|
|
for {
|
|
_, err = file.Write(tun.wrBuff.data[:tun.wrBuff.offset])
|
|
if err != nil {
|
|
pe, ok := err.(*os.PathError)
|
|
if tun.close {
|
|
return os.ErrClosed
|
|
}
|
|
if retries > 0 && ok && pe.Err == windows.ERROR_OPERATION_ABORTED { // Adapter is paused or in low-power state.
|
|
retries--
|
|
time.Sleep(time.Millisecond * 2)
|
|
continue
|
|
}
|
|
if retries > 0 && ok && pe.Err == windows.ERROR_HANDLE_EOF { // Adapter is going down.
|
|
retries--
|
|
tun.closeTUN()
|
|
time.Sleep(time.Millisecond * 2)
|
|
break
|
|
}
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
}
|
|
}
|
|
|
|
func (tun *NativeTun) putTunPacket(buff []byte) error {
|
|
size := uint32(len(buff))
|
|
if size == 0 {
|
|
return errors.New("Empty packet")
|
|
}
|
|
if size > packetSizeMax {
|
|
return errors.New("Packet too big")
|
|
}
|
|
pSize := packetAlign(size) + packetExchangeAlignment
|
|
|
|
if tun.wrBuff.offset+pSize >= packetExchangeSize {
|
|
// Exchange buffer is full -> flush first.
|
|
err := tun.Flush()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
// Write packet to the exchange buffer.
|
|
packet := tun.wrBuff.data[tun.wrBuff.offset : tun.wrBuff.offset+pSize]
|
|
*(*uint32)(unsafe.Pointer(&packet[0])) = size
|
|
packet = packet[packetExchangeAlignment : packetExchangeAlignment+size]
|
|
copy(packet, buff)
|
|
|
|
tun.wrBuff.offset += pSize
|
|
|
|
return nil
|
|
}
|
|
|
|
func (tun *NativeTun) Write(buff []byte, offset int) (int, error) {
|
|
err := tun.putTunPacket(buff[offset:])
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
return len(buff) - offset, nil
|
|
}
|
|
|
|
//
|
|
// LUID returns Windows adapter instance ID.
|
|
//
|
|
func (tun *NativeTun) LUID() uint64 {
|
|
return tun.wt.LUID()
|
|
}
|