tun: windows: wintun does iocp

This commit is contained in:
Jason A. Donenfeld 2019-03-18 02:42:00 -06:00
parent b0e0ab308d
commit 03f2e2614a
2 changed files with 49 additions and 142 deletions

View File

@ -37,13 +37,10 @@ type exchgBufWrite struct {
type NativeTun struct { type NativeTun struct {
wt *wintun.Wintun wt *wintun.Wintun
tunName string tunName string
signalName *uint16
tunFile *os.File tunFile *os.File
tunLock sync.Mutex tunLock sync.Mutex
rdBuff *exchgBufRead rdBuff *exchgBufRead
wrBuff *exchgBufWrite wrBuff *exchgBufWrite
tunDataAvail windows.Handle
userClose windows.Handle
events chan TUNEvent events chan TUNEvent
errors chan error errors chan error
forcedMtu int forcedMtu int
@ -86,106 +83,51 @@ func CreateTUN(ifname string) (TUNDevice, error) {
return nil, errors.New("Flushing interface failed: " + err.Error()) return nil, errors.New("Flushing interface failed: " + err.Error())
} }
signalNameUTF16, err := windows.UTF16PtrFromString(wt.SignalEventName()) return &NativeTun{
if err != nil {
wt.DeleteInterface(0)
return nil, err
}
// Create instance.
tun := &NativeTun{
wt: wt, wt: wt,
tunName: wt.DataFileName(), tunName: wt.DataFileName(),
signalName: signalNameUTF16,
rdBuff: &exchgBufRead{}, rdBuff: &exchgBufRead{},
wrBuff: &exchgBufWrite{}, wrBuff: &exchgBufWrite{},
events: make(chan TUNEvent, 10), events: make(chan TUNEvent, 10),
errors: make(chan error, 1), errors: make(chan error, 1),
forcedMtu: 1500, forcedMtu: 1500,
} }, nil
// Create close event.
tun.userClose, err = windows.CreateEvent(nil, 1 /*TRUE*/, 0 /*FALSE*/, nil)
if err != nil {
wt.DeleteInterface(0)
return nil, err
}
return tun, nil
} }
func (tun *NativeTun) openTUN() error { func (tun *NativeTun) openTUN() {
for { for {
// Open interface data pipe. file, err := os.OpenFile(tun.tunName, os.O_RDWR, 0)
// Data pipe must be opened first, as the interface data available event is created when somebody actually connects to the data pipe.
file, err := os.OpenFile(tun.tunName, os.O_RDWR|os.O_SYNC, 0600)
if err != nil { if err != nil {
// After examining possible error conditions, many arose that were only temporary: windows.ERROR_FILE_NOT_FOUND, "read <filename> closed", etc.
// To simplify, we will enter a retry-loop on _any_ error until session is closed by user.
switch evt, e := windows.WaitForSingleObject(tun.userClose, 1000); evt {
case windows.WAIT_OBJECT_0, windows.WAIT_ABANDONED:
return errors.New("TUN closed")
case windows.WAIT_TIMEOUT:
continue continue
default:
return errors.New("Unexpected result from WaitForSingleObject: " + e.Error())
} }
}
// Open interface data available event.
event, err := windows.OpenEvent(windows.SYNCHRONIZE, false, tun.signalName)
if err != nil {
file.Close()
return errors.New("Opening interface data ready event failed: " + err.Error())
}
tun.tunFile = file tun.tunFile = file
tun.tunDataAvail = event
return nil
} }
} }
func (tun *NativeTun) closeTUN() (err error) { func (tun *NativeTun) closeTUN() (err error) {
if tun.tunFile != nil {
tun.tunLock.Lock() tun.tunLock.Lock()
defer tun.tunLock.Unlock() defer tun.tunLock.Unlock()
if tun.tunFile == nil {
if tun.tunDataAvail != 0 { return
// Close interface data ready event.
e := windows.CloseHandle(tun.tunDataAvail)
if err != nil {
err = e
} }
t := tun.tunFile
tun.tunDataAvail = 0
}
if tun.tunFile != nil {
// Close interface data pipe.
e := tun.tunFile.Close()
if err != nil {
err = e
}
tun.tunFile = nil tun.tunFile = nil
err = t.Close()
} }
return return
} }
func (tun *NativeTun) getTUN() (*os.File, windows.Handle, error) { func (tun *NativeTun) getTUN() (*os.File, error) {
if tun.tunFile == nil {
tun.tunLock.Lock() tun.tunLock.Lock()
defer tun.tunLock.Unlock() defer tun.tunLock.Unlock()
if tun.tunFile != nil {
if tun.tunFile == nil || tun.tunDataAvail == 0 { return tun.tunFile, nil
// TUN device is not open (yet).
err := tun.openTUN()
if err != nil {
return nil, 0, err
} }
tun.openTUN()
} }
return tun.tunFile, nil
return tun.tunFile, tun.tunDataAvail, nil
} }
func (tun *NativeTun) Name() (string, error) { func (tun *NativeTun) Name() (string, error) {
@ -201,24 +143,18 @@ func (tun *NativeTun) Events() chan TUNEvent {
} }
func (tun *NativeTun) Close() error { func (tun *NativeTun) Close() error {
windows.SetEvent(tun.userClose) err1 := tun.closeTUN()
err := windows.CloseHandle(tun.userClose)
e := tun.closeTUN()
if err == nil {
err = e
}
if tun.events != nil { if tun.events != nil {
close(tun.events) close(tun.events)
} }
_, _, e = tun.wt.DeleteInterface(0) _, _, err2 := tun.wt.DeleteInterface(0)
if err == nil { if err1 == nil {
err = e err1 = err2
} }
return err return err1
} }
func (tun *NativeTun) MTU() (int, error) { func (tun *NativeTun) MTU() (int, error) {
@ -248,42 +184,16 @@ func (tun *NativeTun) Read(buff []byte, offset int) (int, error) {
tun.rdBuff.avail = 0 tun.rdBuff.avail = 0
continue continue
} }
packet = packet[:pSize] packet = packet[packetExchangeAlignment : packetExchangeAlignment+size]
// Copy data. // Copy data.
copy(buff[offset:], packet[packetExchangeAlignment:packetExchangeAlignment+size]) copy(buff[offset:], packet)
tun.rdBuff.offset += pSize tun.rdBuff.offset += pSize
return int(size), nil return int(size), nil
} }
// Get TUN data ready event.
_, tunDataAvail, err := tun.getTUN()
if err != nil {
return 0, err
}
// Wait for user close or interface data.
r, err := windows.WaitForMultipleObjects([]windows.Handle{tun.userClose, tunDataAvail}, false, windows.INFINITE)
if err != nil {
return 0, errors.New("Waiting for data failed: " + err.Error())
}
switch r {
case windows.WAIT_OBJECT_0 + 0, windows.WAIT_ABANDONED + 0:
return 0, errors.New("TUN closed")
case windows.WAIT_OBJECT_0 + 1:
// Data is available.
case windows.WAIT_ABANDONED + 1:
// TUN stopped.
tun.closeTUN()
case windows.WAIT_TIMEOUT:
// Congratulations, we reached infinity. Let's do it again! :)
continue
default:
return 0, errors.New("unexpected result from WaitForMultipleObjects")
}
// Get TUN data pipe. // Get TUN data pipe.
file, _, err := tun.getTUN() file, err := tun.getTUN()
if err != nil { if err != nil {
return 0, err return 0, err
} }
@ -291,6 +201,9 @@ func (tun *NativeTun) Read(buff []byte, offset int) (int, error) {
// Fill queue. // Fill queue.
n, err := file.Read(tun.rdBuff.data[:]) n, err := file.Read(tun.rdBuff.data[:])
if err != nil { if err != nil {
if pe, ok := err.(*os.PathError); ok && pe.Err == os.ErrClosed {
return 0, err
}
// TUN interface stopped, failed, etc. Retry. // TUN interface stopped, failed, etc. Retry.
tun.rdBuff.avail = 0 tun.rdBuff.avail = 0
tun.closeTUN() tun.closeTUN()
@ -305,7 +218,7 @@ func (tun *NativeTun) Read(buff []byte, offset int) (int, error) {
func (tun *NativeTun) flush() error { func (tun *NativeTun) flush() error {
// Get TUN data pipe. // Get TUN data pipe.
file, _, err := tun.getTUN() file, err := tun.getTUN()
if err != nil { if err != nil {
return err return err
} }
@ -344,7 +257,8 @@ func (tun *NativeTun) putTunPacket(buff []byte) error {
// Write packet to the exchange buffer. // Write packet to the exchange buffer.
packet := tun.wrBuff.data[tun.wrBuff.offset : tun.wrBuff.offset+pSize] packet := tun.wrBuff.data[tun.wrBuff.offset : tun.wrBuff.offset+pSize]
*(*uint32)(unsafe.Pointer(&packet[0])) = size *(*uint32)(unsafe.Pointer(&packet[0])) = size
copy(packet[packetExchangeAlignment:packetExchangeAlignment+size], buff) packet = packet[packetExchangeAlignment : packetExchangeAlignment+size]
copy(packet, buff)
tun.wrBuff.packetNum++ tun.wrBuff.packetNum++
tun.wrBuff.offset += pSize tun.wrBuff.offset += pSize

View File

@ -501,13 +501,6 @@ func getRegStringValue(key registry.Key, name string) (string, error) {
return valueExp, nil return valueExp, nil
} }
//
// SignalEventName returns Wintun device data-ready event name.
//
func (wintun *Wintun) SignalEventName() string {
return fmt.Sprintf("Global\\WINTUN_EVENT_%s", guid.ToString((*windows.GUID)(wintun)))
}
// //
// DataFileName returns Wintun device data pipe name. // DataFileName returns Wintun device data pipe name.
// //