feat(dpos2.0):check EOF error of FilterLoad.Deserialize

elastos · Mar 29, 2023 · 582f585 · 582f585
1 parent d6773c1
commit 582f585
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/p2p/msg/filterload.go b/p2p/msg/filterload.go
@@ -106,9 +106,15 @@ func (msg *FilterLoad) Deserialize(r io.Reader) error {
 		return err
 	}
 
-	// deserialize TxTypes ignore the result
-	count, _ := common.ReadVarUint(r, 0)
-	msg.TxTypes = make([]common2.TxType, 0, count)
+	msg.TxTypes = make([]common2.TxType, 0)
+	count, err := common.ReadVarUint(r, 0)
+	if err != nil {
+		if err == io.EOF {
+			return nil
+		}
+		return err
+	}
+
 	for i := uint64(0); i < count; i++ {
 		var txType byte
 		err = common.ReadElement(r, &txType)