From a0349d64fdd12295b826cd8db061922bdd050175 Mon Sep 17 00:00:00 2001 From: James Almer Date: Thu, 1 May 2025 14:45:11 -0300 Subject: [PATCH] avformat/apvdec: also look for an au_info PBU during probing If present, an Access unit information PBU must be the first in an AU. Signed-off-by: James Almer --- libavformat/apvdec.c | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) diff --git a/libavformat/apvdec.c b/libavformat/apvdec.c index 5d0b11b654..e53c9ed085 100644 --- a/libavformat/apvdec.c +++ b/libavformat/apvdec.c @@ -51,7 +51,24 @@ static int apv_extract_header_info(GetByteContext *gbc) if (zero != 0) return AVERROR_INVALIDDATA; - if (info->pbu_type != APV_PBU_PRIMARY_FRAME) + if (info->pbu_type == APV_PBU_ACCESS_UNIT_INFORMATION) { + unsigned int num_frames = bytestream2_get_be16(gbc); + int pbu_type; + + if (num_frames < 1) + return AVERROR_INVALIDDATA; + + info->pbu_type = bytestream2_get_byte(gbc); + if (info->pbu_type != APV_PBU_PRIMARY_FRAME && + info->pbu_type != APV_PBU_NON_PRIMARY_FRAME && + (info->pbu_type < APV_PBU_PREVIEW_FRAME || info->pbu_type > APV_PBU_ALPHA_FRAME)) + return AVERROR_INVALIDDATA; + + bytestream2_skip(gbc, 2); // group_id + zero = bytestream2_get_byte(gbc); + if (zero != 0) + return AVERROR_INVALIDDATA; + } else if (info->pbu_type != APV_PBU_PRIMARY_FRAME) return AVERROR_INVALIDDATA; info->profile_idc = bytestream2_get_byte(gbc);