From b9987b4fdc92d3d19eaa9fec83a88681b2ef359b Mon Sep 17 00:00:00 2001
From: Arceliar <Arceliar@users.noreply.github.com>
Date: Sat, 3 Aug 2019 22:47:10 -0500
Subject: [PATCH] reduce time spent with a mutex held in sessionInfo.recvWorker

---
 src/yggdrasil/session.go | 42 +++++++++++++++++++++++-----------------
 1 file changed, 24 insertions(+), 18 deletions(-)

diff --git a/src/yggdrasil/session.go b/src/yggdrasil/session.go
index 993dc52..a48f44b 100644
--- a/src/yggdrasil/session.go
+++ b/src/yggdrasil/session.go
@@ -449,36 +449,42 @@ func (sinfo *sessionInfo) recvWorker() {
 		case p := <-sinfo.fromRouter:
 			var bs []byte
 			var err error
+			var k crypto.BoxSharedKey
 			sessionFunc := func() {
-				defer util.PutBytes(p.Payload)
-				// If the nonce is bad then drop the packet and return an error
 				if !sinfo.nonceIsOK(&p.Nonce) {
 					err = ConnError{errors.New("packet dropped due to invalid nonce"), false, true, false, 0}
 					return
 				}
-				// Decrypt the packet
-				var isOK bool
-				bs, isOK = crypto.BoxOpen(&sinfo.sharedSesKey, p.Payload, &p.Nonce)
-				// Check if we were unable to decrypt the packet for some reason and
-				// return an error if we couldn't
-				if !isOK {
-					err = ConnError{errors.New("packet dropped due to decryption failure"), false, true, false, 0}
+				k = sinfo.sharedSesKey
+			}
+			sinfo.doFunc(sessionFunc)
+			if err != nil {
+				util.PutBytes(p.Payload)
+				continue
+			}
+			var isOK bool
+			bs, isOK = crypto.BoxOpen(&k, p.Payload, &p.Nonce)
+			if !isOK {
+				util.PutBytes(bs)
+				continue
+			}
+			sessionFunc = func() {
+				if k != sinfo.sharedSesKey || !sinfo.nonceIsOK(&p.Nonce) {
+					// The session updated in the mean time, so return an error
+					err = ConnError{errors.New("session updated during crypto operation"), false, true, false, 0}
 					return
 				}
-				// Update the session
 				sinfo.updateNonce(&p.Nonce)
 				sinfo.time = time.Now()
 				sinfo.bytesRecvd += uint64(len(bs))
 			}
 			sinfo.doFunc(sessionFunc)
-			if len(bs) > 0 {
-				if err != nil {
-					// Bad packet, drop it
-					util.PutBytes(bs)
-				} else {
-					// Pass the packet to the buffer for Conn.Read
-					sinfo.recv <- bs
-				}
+			if err != nil {
+				// Not sure what else to do with this packet, I guess just drop it
+				util.PutBytes(bs)
+			} else {
+				// Pass the packet to the buffer for Conn.Read
+				sinfo.recv <- bs
 			}
 		}
 	}