Skip unimplemented handshake extension types (#68)

datarhei · Aug 28, 2024 · 35f558f · 35f558f
1 parent 23a403e
commit 35f558f
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/packet/packet.go b/packet/packet.go
@@ -681,8 +681,12 @@ func (c *CIFHandshake) Unmarshal(data []byte) error {
 			}
 
 			c.CongestionCtl = strings.TrimRight(b.String(), "\x00")
+		} else if extensionType == EXTTYPE_FILTER || extensionType == EXTTYPE_GROUP {
+			if len(pivot) < extensionLength {
+				return fmt.Errorf("invalid extension length of %d bytes (%s)", extensionLength, extensionType.String())
+			}
 		} else {
-			return fmt.Errorf("unimplemented extension (%d)", extensionType)
+			return fmt.Errorf("unknown extension (%d)", extensionType)
 		}
 
 		if len(pivot) > extensionLength {