count any contentConn disconnect that wasn't initiated by us as an error

This fixes connection failures not denying a client access to a media pool (#141).

1 files changed, 4 insertions, 3 deletions

diff --git a/content.go b/content.go
index f24972f..65282ba 100644
--- a/content.go
+++ b/content.go
@@ -32,7 +32,7 @@ type mediaFile struct {
 
 type contentConn struct {
 	mt.Peer
-	denied bool
+	success bool
 
 	logger *log.Logger
 
@@ -229,7 +229,6 @@ func handleContent(cc *contentConn) {
 				M: M,
 			})
 		case *mt.ToCltKick:
-			cc.denied = true
 			cc.log("<-", "deny access", cmd)
 		case *mt.ToCltAcceptAuth:
 			cc.auth.method = 0
@@ -286,6 +285,8 @@ func handleContent(cc *contentConn) {
 			if cmd.I == cmd.N-1 {
 				cc.updateCache()
 				cc.Close()
+
+				cc.success = true
 			}
 		}
 	}
@@ -356,7 +357,7 @@ func muxErrors(conns []*contentConn) map[string]struct{} {
 	for _, cc := range conns {
 		<-cc.done()
 
-		if cc.denied {
+		if !cc.success {
 			denyPools[cc.mediaPool] = struct{}{}
 		}
 	}