truemotion1: check av_fast_malloc() return value
authorPaul B Mahol <onemda@gmail.com>
Fri, 23 Aug 2013 23:16:13 +0000 (23:16 +0000)
committerPaul B Mahol <onemda@gmail.com>
Fri, 23 Aug 2013 23:22:39 +0000 (23:22 +0000)
Signed-off-by: Paul B Mahol <onemda@gmail.com>
libavcodec/truemotion1.c

index a1f4976..7e94deb 100644 (file)
@@ -407,6 +407,8 @@ static int truemotion1_decode_header(TrueMotion1Context *s)
         s->avctx->pix_fmt = new_pix_fmt;
         avcodec_set_dimensions(s->avctx, s->w, s->h);
         av_fast_malloc(&s->vert_pred, &s->vert_pred_size, s->avctx->width * sizeof(unsigned int));
+        if (!s->vert_pred)
+            return AVERROR(ENOMEM);
     }
 
     /* There is 1 change bit per 4 pixels, so each change byte represents
@@ -473,6 +475,8 @@ static av_cold int truemotion1_decode_init(AVCodecContext *avctx)
     /* there is a vertical predictor for each pixel in a line; each vertical
      * predictor is 0 to start with */
     av_fast_malloc(&s->vert_pred, &s->vert_pred_size, s->avctx->width * sizeof(unsigned int));
+    if (!s->vert_pred)
+        return AVERROR(ENOMEM);
 
     return 0;
 }