From d34600b5f92c6a243ffd3427ebd89118cf918ce1 Mon Sep 17 00:00:00 2001
From: Neil Alexander <neilalexander@users.noreply.github.com>
Date: Wed, 17 Jul 2019 10:12:10 +0100
Subject: [PATCH] Try to fix TUN/TAP conn reader leakage

---
 src/tuntap/conn.go | 39 +++++++++++++++++++++++++++------------
 1 file changed, 27 insertions(+), 12 deletions(-)

diff --git a/src/tuntap/conn.go b/src/tuntap/conn.go
index d6bb7a7..f45e4a1 100644
--- a/src/tuntap/conn.go
+++ b/src/tuntap/conn.go
@@ -11,6 +11,8 @@ import (
 	"golang.org/x/net/ipv6"
 )
 
+const tunConnTimeout = 2 * time.Minute
+
 type tunConn struct {
 	tun   *TunAdapter
 	conn  *yggdrasil.Conn
@@ -49,24 +51,37 @@ func (s *tunConn) reader() error {
 		}
 	default:
 	}
+	s.tun.log.Debugln("Starting conn reader for", s)
 	var n int
 	var err error
 	read := make(chan bool)
 	b := make([]byte, 65535)
-	for {
-		go func() {
-			// TODO don't start a new goroutine for every packet read, this is probably a big part of the slowdowns we saw when refactoring
+	go func() {
+		s.tun.log.Debugln("Starting conn reader helper for", s)
+		for {
+			s.conn.SetReadDeadline(time.Now().Add(tunConnTimeout))
 			if n, err = s.conn.Read(b); err != nil {
 				s.tun.log.Errorln(s.conn.String(), "TUN/TAP conn read error:", err)
-				if e, eok := err.(yggdrasil.ConnError); eok && !e.Temporary() {
-					close(s.stop)
-				} else {
-					read <- false
+				if e, eok := err.(yggdrasil.ConnError); eok {
+					switch {
+					case e.Temporary():
+						read <- false
+						continue
+					case e.Timeout():
+						s.tun.log.Debugln("Conn reader for helper", s, "timed out")
+						fallthrough
+					default:
+						s.tun.log.Debugln("Stopping conn reader helper for", s)
+						s.close()
+						return
+					}
 				}
-				return
+				read <- false
 			}
 			read <- true
-		}()
+		}
+	}()
+	for {
 		select {
 		case r := <-read:
 			if r && n > 0 {
@@ -93,6 +108,7 @@ func (s *tunConn) writer() error {
 		}
 	default:
 	}
+	s.tun.log.Debugln("Starting conn writer for", s)
 	for {
 		select {
 		case <-s.stop:
@@ -134,8 +150,7 @@ func (s *tunConn) stillAlive() {
 }
 
 func (s *tunConn) checkForTimeouts() error {
-	const timeout = 2 * time.Minute
-	timer := time.NewTimer(timeout)
+	timer := time.NewTimer(tunConnTimeout)
 	defer util.TimerStop(timer)
 	defer s.close()
 	for {
@@ -145,7 +160,7 @@ func (s *tunConn) checkForTimeouts() error {
 				return errors.New("connection closed")
 			}
 			util.TimerStop(timer)
-			timer.Reset(timeout)
+			timer.Reset(tunConnTimeout)
 		case <-timer.C:
 			return errors.New("timed out")
 		}