apedec: check for data buffer realloc failure

diff --git a/libavcodec/apedec.c b/libavcodec/apedec.c
index a754fd75b0992a99499e61b9dea9294952c7d9fe..2041e2b3c528b448d5bf903842887a5af0e42386 100644 (file)
--- a/libavcodec/apedec.c
+++ b/libavcodec/apedec.c
@@ -824,7 +824,10 @@ static int ape_decode_frame(AVCodecContext *avctx,
     }
 
     if(!s->samples){
-        s->data = av_realloc(s->data, (buf_size + 3) & ~3);
+        void *tmp_data = av_realloc(s->data, (buf_size + 3) & ~3);
+        if (!tmp_data)
+            return AVERROR(ENOMEM);
+        s->data = tmp_data;
         s->dsp.bswap_buf((uint32_t*)s->data, (const uint32_t*)buf, buf_size >> 2);
         s->ptr = s->last_ptr = s->data;
         s->data_end = s->data + buf_size;