diff options
| -rw-r--r-- | tun/tun_windows.go | 23 |
1 files changed, 19 insertions, 4 deletions
diff --git a/tun/tun_windows.go b/tun/tun_windows.go index 49cbdad..d22e130 100644 --- a/tun/tun_windows.go +++ b/tun/tun_windows.go @@ -267,6 +267,9 @@ func (tun *NativeTun) ForceMTU(mtu int) { tun.forcedMTU = mtu } +//go:linkname procyield runtime.procyield +func procyield(cycles uint32) + // Note: Read() and Write() assume the caller comes only from a single thread; there's no locking. func (tun *NativeTun) Read(buff []byte, offset int) (int, error) { @@ -277,6 +280,7 @@ func (tun *NativeTun) Read(buff []byte, offset int) (int, error) { } retries := maybeRetry(1000) +top: for !tun.close { _, err := tun.getTUN() if err != nil { @@ -288,10 +292,21 @@ func (tun *NativeTun) Read(buff []byte, offset int) (int, error) { return 0, errors.New("send ring head out of bounds") } - buffTail := atomic.LoadUint32(&tun.rings.send.ring.tail) - if buffHead == buffTail { - windows.WaitForSingleObject(tun.rings.send.tailMoved, windows.INFINITE) - continue + start := time.Now() + var buffTail uint32 + for { + buffTail = atomic.LoadUint32(&tun.rings.send.ring.tail) + if buffHead != buffTail { + break + } + if tun.close { + return 0, os.ErrClosed + } + if time.Since(start) >= time.Millisecond*50 { + windows.WaitForSingleObject(tun.rings.send.tailMoved, windows.INFINITE) + continue top + } + procyield(1) } if buffTail >= packetCapacity { if retries > 0 { |