Fixes a regression in ca2f19b9cc3 with some mov/mp4 files. The files have
several NAL units in the supposed single NAL unit after the size field.
Annex B start code prefixes are used to separate them. The first NAL unit
is correctly parsed but the buffer does not point to the next size field.
Instead semi random data (it seems to be the rbsp_stop_one_bit and the
start code prefix) is then parsed as length and will exceed the
remaining length of the buffer.
Patch based on the code in h264's decode_nal_units() and a similar
patch by Hendrik Leppkes in FFmpeg (a9bb4cf87d1).
Bug-Id: ffmpeg/trac5529
Reported-By: Vittorio Giovara
---
libavcodec/h2645_parse.c | 46 ++++++++++++++++++++++++++++++++++------------
1 file changed, 34 insertions(+), 12 deletions(-)
diff --git a/libavcodec/h2645_parse.c b/libavcodec/h2645_parse.c
index defe001..a9504bb 100644
--- a/libavcodec/h2645_parse.c
+++ b/libavcodec/h2645_parse.c
@@ -195,11 +195,27 @@ static int h264_parse_nal_header(H2645NAL *nal, void
*logctx)
return 1;
}
+static int find_next_start_code(const uint8_t *buf, const uint8_t *next_avc)
+{
+ int i = 0;
+
+ if (buf + 3 >= next_avc)
+ return next_avc - buf;
+
+ while (buf + i + 3 < next_avc) {
+ if (buf[i] == 0 && buf[i + 1] == 0 && buf[i + 2] == 1)
+ break;
+ i++;
+ }
+ return i + 3;
+}
+
int ff_h2645_packet_split(H2645Packet *pkt, const uint8_t *buf, int length,
void *logctx, int is_nalff, int nal_length_size,
enum AVCodecID codec_id)
{
int consumed, ret = 0;
+ const uint8_t *next_avc = buf + (is_nalff ? 0 : length);
pkt->nb_nals = 0;
while (length >= 4) {
@@ -207,29 +223,35 @@ int ff_h2645_packet_split(H2645Packet *pkt, const uint8_t
*buf, int length,
int extract_length = 0;
int skip_trailing_zeros = 1;
- if (is_nalff) {
+ if (buf == next_avc) {
int i;
for (i = 0; i < nal_length_size; i++)
extract_length = (extract_length << 8) | buf[i];
- buf += nal_length_size;
- length -= nal_length_size;
if (extract_length > length) {
av_log(logctx, AV_LOG_ERROR, "Invalid NAL unit size.\n");
return AVERROR_INVALIDDATA;
}
+ buf += nal_length_size;
+ length -= nal_length_size;
+ next_avc = buf + extract_length;
} else {
- if (buf[2] == 0) {
- length--;
- buf++;
+ int buf_index = find_next_start_code(buf, next_avc);
+
+ buf += buf_index;
+ length -= buf_index;
+
+ if (buf == next_avc)
continue;
- }
- if (buf[0] != 0 || buf[1] != 0 || buf[2] != 1)
- return AVERROR_INVALIDDATA;
- buf += 3;
- length -= 3;
- extract_length = length;
+ if (length > 0) {
+ extract_length = length;
+ } else if (pkt->nb_nals == 0) {
+ av_log(logctx, AV_LOG_ERROR, "No NAL unit found\n");
+ return AVERROR_INVALIDDATA;
+ } else {
+ break;
+ }
}
if (pkt->nals_allocated < pkt->nb_nals + 1) {
--
2.9.0
_______________________________________________
libav-devel mailing list
[email protected]
https://lists.libav.org/mailman/listinfo/libav-devel
