From 69277069dd225f5ae02f0215e877a06522d834d3 Mon Sep 17 00:00:00 2001
From: Vittorio Giovara <vittorio.giovara@gmail.com>
Date: Sun, 31 May 2015 14:53:31 +0200
Subject: [PATCH] dct: Check memory allocation

---
 libavcodec/dct.c | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/libavcodec/dct.c b/libavcodec/dct.c
index 4dbbff867b..180477e621 100644
--- a/libavcodec/dct.c
+++ b/libavcodec/dct.c
@@ -191,6 +191,8 @@ av_cold int ff_dct_init(DCTContext *s, int nbits, enum DCTTransformType inverse)
 
         s->costab = ff_cos_tabs[nbits + 2];
         s->csc2   = av_malloc(n / 2 * sizeof(FFTSample));
+        if (!s->csc2)
+            return AVERROR(ENOMEM);
 
         if (ff_rdft_init(&s->rdft, nbits, inverse == DCT_III) < 0) {
             av_free(s->csc2);