Add an index check to prevent out of index

pion · May 2, 2024 · 2528060 · 2528060
1 parent bc5124c
commit 2528060
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/codecs/h264_packet.go b/codecs/h264_packet.go
@@ -231,7 +231,11 @@ func (p *H264Packet) parseBody(payload []byte) ([]byte, error) {
 		currOffset := int(stapaHeaderSize)
 		result := []byte{}
 		for currOffset < len(payload) {
-			naluSize := int(binary.BigEndian.Uint16(payload[currOffset:]))
+			naluSizeBytes := payload[currOffset:]
+			if len(naluSizeBytes) < stapaNALULengthSize {
+				break
+			}
+			naluSize := int(binary.BigEndian.Uint16(naluSizeBytes))
 			currOffset += stapaNALULengthSize
 
 			if len(payload) < currOffset+naluSize {