code.ivysaur.me/libnmdc
1
0
Fork 0
Code Issues 2 Pull Requests Releases 17 Activity

libnmdc: patch panic() on nil connection

Browse Source
This commit is contained in:
mappu 2016-04-02 13:49:38 +13:00
parent 83792a4241
commit 433c1ddac9
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/libnmdc/libnmdc.go
View File

@ -314,6 +314,7 @@ func (this *HubConnection) processProtocolMessage(message string) {
func (this *HubConnection) worker() { func (this *HubConnection) worker() {
var fullBuffer string var fullBuffer string
var err error = nil var err error = nil
var nbytes int = 0
for { for {
@ -332,11 +333,13 @@ func (this *HubConnection) worker() {
} }
// Read from socket into our local buffer (blocking) // Read from socket into our local buffer (blocking)
if this.conn != nil {
readBuff := make([]byte, 4096) readBuff := make([]byte, 4096)
nbytes, err := this.conn.Read(readBuff) nbytes, err = this.conn.Read(readBuff)
if nbytes > 0 { if nbytes > 0 {
fullBuffer += string(readBuff[0:nbytes]) fullBuffer += string(readBuff[0:nbytes])
} }
}
// Maybe we disconnected // Maybe we disconnected
if err != nil { if err != nil {