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

Fix error handling at websocket close.

Browse source
This commit is contained in:
Juliusz Chroboczek 2020-12-05 01:18:27 +01:00
parent 26e6ecc3bc
commit e8c732c54c
1 changed files with 13 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
rtpconn/webclient.go
View file

@ -657,13 +657,13 @@ func readMessage(conn *websocket.Conn, m *clientMessage) error {
return conn.ReadJSON(&m)
}
func StartClient(conn *websocket.Conn) error {
func StartClient(conn *websocket.Conn) (err error) {
var m clientMessage
err := readMessage(conn, &m)
err = readMessage(conn, &m)
if err != nil {
conn.Close()
return err
return
}
if m.Type != "handshake" {
@ -674,7 +674,8 @@ func StartClient(conn *websocket.Conn) error {
),
)
conn.Close()
return group.ProtocolError("client didn't handshake")
err = group.ProtocolError("client didn't handshake")
return
}
c := &webClient{
@ -689,13 +690,14 @@ func StartClient(conn *websocket.Conn) error {
c.writerDone = make(chan struct{})
go clientWriter(conn, c.writeCh, c.writerDone)
defer func() {
var e []byte
if !isWSNormalError(err) {
var m *clientMessage
m, e = errorToWSCloseMessage(c.id, err)
if m != nil {
c.write(*m)
}
m, e := errorToWSCloseMessage(c.id, err)
if isWSNormalError(err) {
err = nil
} else if _, ok := err.(group.KickError); ok {
err = nil
}
if m != nil {
c.write(*m)
}
c.close(e)
}()