From e8ac7ca81cb0301ed05628fa13b744bbdb051f44 Mon Sep 17 00:00:00 2001 From: wdvxdr Date: Sat, 13 Nov 2021 16:42:06 +0800 Subject: [PATCH] fix(server): fix read limit in ws-server --- server/websocket.go | 1 + 1 file changed, 1 insertion(+) diff --git a/server/websocket.go b/server/websocket.go index e415ff2..4e9fe04 100644 --- a/server/websocket.go +++ b/server/websocket.go @@ -313,6 +313,7 @@ func (s *webSocketServer) any(w http.ResponseWriter, r *http.Request) { func (s *webSocketServer) listenAPI(c *wsConn) { defer func() { _ = c.Close(websocket.StatusNormalClosure, "") }() + c.Conn.SetReadLimit(1024 * 1024 * 128) for { buffer := global.NewBuffer() t, reader, err := c.Reader(context.Background())