Potential fix to an edge case nil pointer

This commit is contained in:
Qian Wang 2019-08-09 10:36:27 +01:00
parent af0a356000
commit abc83a9190
1 changed files with 6 additions and 2 deletions

View File

@ -76,8 +76,12 @@ func (sb *switchboard) send(data []byte, connId *uint32) (int, error) {
return 0, errBrokenSwitchboard
}
newConnId := rand.Intn(len(sb.conns))
conn = sb.conns[uint32(newConnId)]
return conn.Write(data)
conn, ok = sb.conns[uint32(newConnId)]
if !ok {
return 0, errBrokenSwitchboard
} else {
return conn.Write(data)
}
}
}