dct: Check memory allocation

69277069 · Vittorio Giovara · Luca Barbato · e524f373 · 69277069
Commit 69277069 authored May 31, 2015 by Vittorio Giovara Committed by Luca Barbato May 31, 2015
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

dct.c libavcodec/dct.c +2 -0

No files found.
--- 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);