[FFmpeg-devel] [PATCH 5/5] avcodec/apedec: Check max_samples

[Michael Niedermayer]

Fixes: OOM
Fixes: 16627/clusterfuzz-testcase-minimized-ffmpeg_AV_CODEC_ID_APE_fuzzer-5638059583864832

Found-by: continuous fuzzing process https://github.com/google/oss-fuzz/tree/master/projects/ffmpeg
Signed-off-by: Michael Niedermayer <michael at niedermayer.cc>
---
 libavcodec/apedec.c | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/libavcodec/apedec.c b/libavcodec/apedec.c
index e218f7c043..774ce18531 100644
--- a/libavcodec/apedec.c
+++ b/libavcodec/apedec.c
@@ -1475,6 +1475,9 @@ static int ape_decode_frame(AVCodecContext *avctx, void *data,
             return AVERROR_INVALIDDATA;
         }
 
+        if (nblocks * (int64_t)avctx->channels > avctx->max_samples)
+            return AVERROR(EINVAL);
+
         /* Initialize the frame decoder */
         if (init_frame_decoder(s) < 0) {
             av_log(avctx, AV_LOG_ERROR, "Error reading frame header\n");
-- 
2.23.0