From 8317f447d15ed08589e9b4f358594368bd2e368b Mon Sep 17 00:00:00 2001
From: Qian Wang <cbeuw.andy@gmail.com>
Date: Tue, 18 Jun 2019 22:21:13 +1000
Subject: [PATCH] Detect TCP connection issues immediately rather than waiting
 for tokens

---
 internal/multiplex/switchboard.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/internal/multiplex/switchboard.go b/internal/multiplex/switchboard.go
index d630b67..3c11fdf 100644
--- a/internal/multiplex/switchboard.go
+++ b/internal/multiplex/switchboard.go
@@ -134,13 +134,13 @@ func (sb *switchboard) deplex(ce *connEnclave) {
 	buf := make([]byte, 20480)
 	for {
 		n, err := sb.session.obfsedRead(ce.remoteConn, buf)
-		sb.rxWait(n)
 		if err != nil {
 			//log.Println(err)
 			go ce.remoteConn.Close()
 			sb.removeConn(ce)
 			return
 		}
+		sb.rxWait(n)
 		if sb.AddRxCredit(-int64(n)) < 0 {
 			log.Println(ErrNoRxCredit)
 			sb.session.Close()