Additional ws session check before writing

Avoiding reported "invalid memory address or nil pointer dereference"
2019-07-16 09:01:01 +02:00
parent 12c7ae682d
commit b006cb8784
1 changed files with 6 additions and 2 deletions
@@ -219,7 +219,7 @@ func (sess *Session) writeLoop() error {
 			return
 		}

-		if msg != nil {
+		if msg != nil && sess.conn != nil {
 			return sess.conn.WriteMessage(websocket.TextMessage, msg)
 		}

@@ -236,7 +236,11 @@ func (sess *Session) writeLoop() error {
 			return
 		}

-		return sess.conn.WriteMessage(websocket.PingMessage, nil)
+		if sess.conn != nil {
+			return sess.conn.WriteMessage(websocket.PingMessage, nil)
+		}
+
+		return
 	}

 	for {