[FFmpeg-cvslog] avformat/apvdec: also look for an au_info PBU during probing

[James Almer]

ffmpeg | branch: master | James Almer <jamrial at gmail.com> | Thu May  1 14:45:11 2025 -0300| [a0349d64fdd12295b826cd8db061922bdd050175] | committer: James Almer

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 <jamrial at gmail.com>

> http://git.videolan.org/gitweb.cgi/ffmpeg.git/?a=commit;h=a0349d64fdd12295b826cd8db061922bdd050175
---

 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);