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

11
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,10 +333,12 @@ func (this *HubConnection) worker() {
} }
// Read from socket into our local buffer (blocking) // Read from socket into our local buffer (blocking)
readBuff := make([]byte, 4096) if this.conn != nil {
nbytes, err := this.conn.Read(readBuff) readBuff := make([]byte, 4096)
if nbytes > 0 { nbytes, err = this.conn.Read(readBuff)
fullBuffer += string(readBuff[0:nbytes]) if nbytes > 0 {
fullBuffer += string(readBuff[0:nbytes])
}
} }
// Maybe we disconnected // Maybe we disconnected