From b53ac2a5283da595b5f5c7f685c5102dc3ed50ca Mon Sep 17 00:00:00 2001 From: James Almer Date: Fri, 24 Mar 2017 18:27:37 -0300 Subject: avcodec/extract_extradata_bsf: use the parsing code from mpeg4video_split() It's a simplifaction of the same code, originally commited as 3b5ad8fbf77. Reviewed-by: Michael Niedermayer Signed-off-by: James Almer --- libavcodec/extract_extradata_bsf.c | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) (limited to 'libavcodec/extract_extradata_bsf.c') diff --git a/libavcodec/extract_extradata_bsf.c b/libavcodec/extract_extradata_bsf.c index 447afb9f9a..e9b3791dc0 100644 --- a/libavcodec/extract_extradata_bsf.c +++ b/libavcodec/extract_extradata_bsf.c @@ -201,14 +201,14 @@ static int extract_extradata_mpeg4(AVBSFContext *ctx, AVPacket *pkt, uint8_t **data, int *size) { ExtractExtradataContext *s = ctx->priv_data; + const uint8_t *ptr = pkt->data, *end = pkt->data + pkt->size; uint32_t state = UINT32_MAX; - int i; - for (i = 0; i < pkt->size; i++) { - state = (state << 8) | pkt->data[i]; - if ((state == 0x1B3 || state == 0x1B6)) { - if (i > 3) { - *size = i - 3; + while (ptr < end) { + ptr = avpriv_find_start_code(ptr, end, &state); + if (state == 0x1B3 || state == 0x1B6) { + if (ptr - pkt->data > 4) { + *size = ptr - 4 - pkt->data; *data = av_malloc(*size + AV_INPUT_BUFFER_PADDING_SIZE); if (!*data) return AVERROR(ENOMEM); -- cgit v1.2.3