tun: windows: Make writing persistent too

Signed-off-by: Simon Rozman <simon@rozman.si>

diff --git a/tun/tun_windows.go b/tun/tun_windows.go
index ce6ec8c9710206572862839fdc55b87f1b3ab3ee..9888e86975c111bbbcd3c98fe52a15b9ba6f6af6 100644 (file)
--- a/tun/tun_windows.go
+++ b/tun/tun_windows.go
@@ -284,24 +284,26 @@ func (tun *NativeTun) Flush() error {
                return nil
        }
 
-       // Get TUN data pipe.
-       _, file, err := tun.getTUN()
-       if err != nil {
-               return err
-       }
+       for {
+               // Get TUN data pipe.
+               _, file, err := tun.getTUN()
+               if err != nil {
+                       return err
+               }
 
-       // Flush write buffer.
-       _, err = file.Write(tun.wrBuff.data[:tun.wrBuff.offset])
-       tun.wrBuff.packetNum = 0
-       tun.wrBuff.offset = 0
-       if err != nil {
-               if tun.shouldReopenHandle(err) {
-                       tun.closeTUN()
-                       return nil
+               // Flush write buffer.
+               _, err = file.Write(tun.wrBuff.data[:tun.wrBuff.offset])
+               tun.wrBuff.packetNum = 0
+               tun.wrBuff.offset = 0
+               if err != nil {
+                       if tun.shouldReopenHandle(err) {
+                               tun.closeTUN()
+                               continue
+                       }
+                       return err
                }
-               return err
+               return nil
        }
-       return nil
 }
 
 func (tun *NativeTun) putTunPacket(buff []byte) error {