fix the panic issue in link.Rx() (#664)

ontio · Sep 27, 2018 · a814f51 · a814f51
1 parent f5219c7
commit a814f51
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/p2pserver/link/link.go b/p2pserver/link/link.go
@@ -110,7 +110,12 @@ func (this *Link) GetRXTime() time.Time {
 }
 
 func (this *Link) Rx() {
-	reader := bufio.NewReaderSize(this.conn, common.MAX_BUF_LEN)
+	conn := this.conn
+	if conn == nil {
+		return
+	}
+
+	reader := bufio.NewReaderSize(conn, common.MAX_BUF_LEN)
 
 	for {
 		msg, payloadSize, err := types.ReadMessage(reader)