h265: fix panic when marshaling SPS

bluenviron · Aug 1, 2024 · 0f439fd · 0f439fd
1 parent a94f00b
commit 0f439fd
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pkg/codecs/h265/sps.go b/pkg/codecs/h265/sps.go
@@ -741,6 +741,11 @@ func (s *SPS) Unmarshal(buf []byte) error {
 		return err
 	}
 
+	// this prevents a panic in Marshal()
+	if s.ChromaFormatIdc > 3 {
+		return fmt.Errorf("invalid chroma_format_idc")
+	}
+
 	if s.ChromaFormatIdc == 3 {
 		s.SeparateColourPlaneFlag, err = bits.ReadFlag(buf, &pos)
 		if err != nil {