maride/gotty
1
0
Fork 0
mirror of https://github.com/sorenisanerd/gotty.git synced 2024-11-22 12:24:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make sure we read the full message

Browse Source
This commit is contained in:
Søren L. Hansen 2022-03-31 10:42:51 -07:00
parent 82c3acf3b1
commit 1eed97f0f8
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
server/ws_wrapper.go
View File

@ -1,7 +1,10 @@
package server package server
import ( import (
"io/ioutil"
"github.com/gorilla/websocket" "github.com/gorilla/websocket"
"github.com/pkg/errors"
) )
type wsWrapper struct { type wsWrapper struct {
@ -28,6 +31,11 @@ func (wsw *wsWrapper) Read(p []byte) (n int, err error) {
continue continue
} }
return reader.Read(p) b, err := ioutil.ReadAll(reader)
if len(b) > len(p) {
return 0, errors.Wrapf(err, "Client message exceeded buffer size")
}
n = copy(p, b)
return n, err
} }
} }