Commit b8ff4f5e authored by Paul B Mahol's avatar Paul B Mahol

truemotion1: check av_fast_malloc() return value

Signed-off-by: 's avatarPaul B Mahol <onemda@gmail.com>
parent 83b915d4
...@@ -407,6 +407,8 @@ static int truemotion1_decode_header(TrueMotion1Context *s) ...@@ -407,6 +407,8 @@ static int truemotion1_decode_header(TrueMotion1Context *s)
s->avctx->pix_fmt = new_pix_fmt; s->avctx->pix_fmt = new_pix_fmt;
avcodec_set_dimensions(s->avctx, s->w, s->h); 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)); 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 /* 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) ...@@ -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 /* there is a vertical predictor for each pixel in a line; each vertical
* predictor is 0 to start with */ * predictor is 0 to start with */
av_fast_malloc(&s->vert_pred, &s->vert_pred_size, s->avctx->width * sizeof(unsigned int)); 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; return 0;
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment