DevFW-CICD/garm
15
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 2

Set read limit to 16 KB

Browse source 
Signed-off-by: Gabriel Adrian Samfira <gsamfira@cloudbasesolutions.com>
This commit is contained in:
Gabriel Adrian Samfira 2024-07-04 15:29:15 +00:00
parent a2c606b87d
commit 49f1b7aa14
2 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
websocket/client.go
View file

@ -29,7 +29,7 @@ const (
pingPeriod = (pongWait * 9) / 10
// Maximum message size allowed from peer.
maxMessageSize = 1024
maxMessageSize = 16384 // 16 KB
)
type HandleWebsocketMessage func([]byte) error

3
websocket/util.go
View file

@ -113,8 +113,11 @@ func (w *Reader) Start() error {
func (w *Reader) printWebsocketToConsole() {
defer w.Stop()
w.writeMux.Lock()
w.conn.SetReadLimit(maxMessageSize)
w.conn.SetReadDeadline(time.Now().Add(pongWait))
w.conn.SetPongHandler(func(string) error { w.conn.SetReadDeadline(time.Now().Add(pongWait)); return nil })
w.writeMux.Unlock()
for {
_, message, err := w.conn.ReadMessage()
if err != nil {