tta: check for allocation failure of decode_buffer

muzee-git · Oct 25, 2011 · 2f1d212 · 2f1d212
1 parent b505053
commit 2f1d212
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/libavcodec/tta.c b/libavcodec/tta.c
@@ -263,10 +263,14 @@ static av_cold int tta_decode_init(AVCodecContext * avctx)
 
         if (s->bps == 2) {
             s->decode_buffer = av_mallocz(sizeof(int32_t)*s->frame_length*s->channels);
+            if (!s->decode_buffer)
+                return AVERROR(ENOMEM);
         }
         s->ch_ctx = av_malloc(avctx->channels * sizeof(*s->ch_ctx));
-        if (!s->ch_ctx)
+        if (!s->ch_ctx) {
+            av_freep(&s->decode_buffer);
             return AVERROR(ENOMEM);
+        }
     } else {
         av_log(avctx, AV_LOG_ERROR, "Wrong extradata present\n");
         return -1;