mirror
/
galene
mirror of https://github.com/jech/galene.git
1
Fork 0
Code

Add timeout for initial handshake.

This commit is contained in:
Juliusz Chroboczek 2020-04-25 22:49:07 +02:00
parent b201c3d93c
commit c86b1c87d5
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
client.go
View File

@ -108,11 +108,23 @@ type closeMessage struct {
func startClient(conn *websocket.Conn) (err error) {
var m clientMessage
err = conn.SetReadDeadline(time.Now().Add(15 * time.Second))
if err != nil {
conn.Close()
return
}
err = conn.ReadJSON(&m)
if err != nil {
conn.Close()
return
}
err = conn.SetReadDeadline(time.Time{})
if err != nil {
conn.Close()
return
}
if m.Type != "handshake" {
conn.Close()
return