mirror/galene
1
Fork 0
mirror of https://github.com/jech/galene.git synced 2024-11-22 16:45:58 +01:00
Code Activity

Add timeout for initial handshake.

Browse source
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) { func startClient(conn *websocket.Conn) (err error) {
var m clientMessage var m clientMessage
err = conn.SetReadDeadline(time.Now().Add(15 * time.Second))
if err != nil {
conn.Close()
return
}
err = conn.ReadJSON(&m) err = conn.ReadJSON(&m)
if err != nil { if err != nil {
conn.Close() conn.Close()
return return
} }
err = conn.SetReadDeadline(time.Time{})
if err != nil {
conn.Close()
return
}
if m.Type != "handshake" { if m.Type != "handshake" {
conn.Close() conn.Close()
return return