From b27c9fe7bc42794167f7a11c0e1a213cc56df56d Mon Sep 17 00:00:00 2001
From: Anton Kovalenko <anton@sw4me.com>
Date: Mon, 13 May 2024 23:38:48 +0300
Subject: [PATCH] fix: avoid setting read limit on nil conn

---
 main.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.go b/main.go
index 6e653fe..a6cc153 100644
--- a/main.go
+++ b/main.go
@@ -173,11 +173,11 @@ func (h *Handler) LoadPeers(filename string) error {
 
 func (h *Handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
 	wsConn, err := websocket.Accept(w, r, nil)
-	wsConn.SetReadLimit(-1)
 	if err != nil {
 		log.WithField("client", r.RemoteAddr).Error(err)
 		return
 	}
+	wsConn.SetReadLimit(-1)
 	go h.ServeWebSocket(wsConn, r.RemoteAddr)
 }