Don't attempt to read from conn if closed (#3913)

Signed-off-by: Nikki Attea <nikki@sensu.io>
sensu · Aug 5, 2020 · 4947298 · 4947298
1 parent 0c01649
commit 4947298
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/backend/agentd/agentd.go b/backend/agentd/agentd.go
@@ -120,9 +120,11 @@ func New(c Config, opts ...Option) (*Agentd, error) {
 		// Capture the log entries from agentd's HTTP server
 		ErrorLog: log.New(&logrusIOWriter{entry: logger}, "", 0),
 		ConnState: func(c net.Conn, cs http.ConnState) {
-			var msg []byte
-			if _, err := c.Read(msg); err != nil {
-				logger.WithError(err).Error("websocket connection error")
+			if cs != http.StateClosed {
+				var msg []byte
+				if _, err := c.Read(msg); err != nil {
+					logger.WithError(err).Error("websocket connection error")
+				}
 			}
 		},
 	}