avcodec/atrac1: Check allocation of AVFloatDSPContext

Reviewed-by: Paul B Mahol <onemda@gmail.com>
Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@gmail.com>

diff --git a/libavcodec/atrac1.c b/libavcodec/atrac1.c
index a8c8c91bccbfc88db4945b4e5b0a7d9ba52c6230..9ecd49273d9fcad91974cd3521d3d187b28e5e90 100644 (file)
--- a/libavcodec/atrac1.c
+++ b/libavcodec/atrac1.c
@@ -362,6 +362,10 @@ static av_cold int atrac1_decode_init(AVCodecContext *avctx)
     ff_atrac_generate_tables();
 
     q->fdsp = avpriv_float_dsp_alloc(avctx->flags & AV_CODEC_FLAG_BITEXACT);
+    if (!q->fdsp) {
+        atrac1_decode_end(avctx);
+        return AVERROR(ENOMEM);
+    }
 
     q->bands[0] = q->low;
     q->bands[1] = q->mid;