accept > 2 items on EOSE and NOTICE envelopes, but ignore them.

fixes #110
nbd-wtf · Nov 14, 2023 · 9287b9f · 9287b9f
1 parent 93b43ed
commit 9287b9f
Showing 1 changed file with 6 additions and 10 deletions.
diff --git a/envelopes.go b/envelopes.go
@@ -200,13 +200,11 @@ func (_ NoticeEnvelope) Label() string { return "NOTICE" }
 func (v *NoticeEnvelope) UnmarshalJSON(data []byte) error {
 	r := gjson.ParseBytes(data)
 	arr := r.Array()
-	switch len(arr) {
-	case 2:
-		*v = NoticeEnvelope(arr[1].Str)
-		return nil
-	default:
+	if len(arr) < 2 {
 		return fmt.Errorf("failed to decode NOTICE envelope")
 	}
+	*v = NoticeEnvelope(arr[1].Str)
+	return nil
 }
 
 func (v NoticeEnvelope) MarshalJSON() ([]byte, error) {
@@ -224,13 +222,11 @@ func (_ EOSEEnvelope) Label() string { return "EOSE" }
 func (v *EOSEEnvelope) UnmarshalJSON(data []byte) error {
 	r := gjson.ParseBytes(data)
 	arr := r.Array()
-	switch len(arr) {
-	case 2:
-		*v = EOSEEnvelope(arr[1].Str)
-		return nil
-	default:
+	if len(arr) < 2 {
 		return fmt.Errorf("failed to decode EOSE envelope")
 	}
+	*v = EOSEEnvelope(arr[1].Str)
+	return nil
 }
 
 func (v EOSEEnvelope) MarshalJSON() ([]byte, error) {