diff mbox

[FFmpeg-devel,2/3] avcodec/jpeg2000: Check sum of sizes of band->prec before allocating

Message ID 20180113233941.9407-2-michael@niedermayer.cc
State Accepted
Commit 6887e412434776eb260ad3904f565be491dd5726
Headers show

Commit Message

Michael Niedermayer Jan. 13, 2018, 11:39 p.m. UTC 
Fixes: OOM
Fixes: 4810/clusterfuzz-testcase-minimized-6034253235093504

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

Patch

diff --git a/libavcodec/jpeg2000.c b/libavcodec/jpeg2000.c
index e7f03bd0df..8e90980976 100644
--- a/libavcodec/jpeg2000.c
+++ b/libavcodec/jpeg2000.c
@@ -541,6 +541,9 @@  int ff_jpeg2000_init_component(Jpeg2000Component *comp,
         if (!reslevel->band)
             return AVERROR(ENOMEM);
 
+        if (reslevel->num_precincts_x * (uint64_t)reslevel->num_precincts_y * reslevel->nbands > avctx->max_pixels / sizeof(*reslevel->band->prec))
+            return AVERROR(ENOMEM);
+
         for (bandno = 0; bandno < reslevel->nbands; bandno++, gbandno++) {
             ret = init_band(avctx, reslevel,
                             comp, codsty, qntsty,