From e75ccc81234a22eee90e2a31808639e754b97cd1 Mon Sep 17 00:00:00 2001
From: Andreas Rheinhardt <andreas.rheinhardt@gmail.com>
Date: Sun, 13 Sep 2020 20:53:46 +0200
Subject: [PATCH] avcodec/atrac1: Check allocation of AVFloatDSPContext

Reviewed-by: Paul B Mahol <onemda@gmail.com>
Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@gmail.com>
---
 libavcodec/atrac1.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/libavcodec/atrac1.c b/libavcodec/atrac1.c
index a8c8c91bcc..9ecd49273d 100644
--- 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;