avcodec/diracdec: Check ff_set_dimensions() for failure

Signed-off-by: Michael Niedermayer <michael@niedermayer.cc>

diff --git a/libavcodec/diracdec.c b/libavcodec/diracdec.c
index f35fe453eed427288670c9cddb5feeb929a2a465..486c1b3ad7e2fa0fa0c2306c187be12ad8f5947d 100644 (file)
--- a/libavcodec/diracdec.c
+++ b/libavcodec/diracdec.c
@@ -1989,7 +1989,12 @@ static int dirac_decode_data_unit(AVCodecContext *avctx, const uint8_t *buf, int
             return ret;
         }
 
-        ff_set_dimensions(avctx, dsh->width, dsh->height);
+        ret = ff_set_dimensions(avctx, dsh->width, dsh->height);
+        if (ret < 0) {
+            av_freep(&dsh);
+            return ret;
+        }
+
         ff_set_sar(avctx, dsh->sample_aspect_ratio);
         avctx->pix_fmt         = dsh->pix_fmt;
         avctx->color_range     = dsh->color_range;