vp8: Check for malloc failure

Originally committed as revision 24251 to svn://svn.ffmpeg.org/ffmpeg/trunk

1 files changed, 3 insertions, 0 deletions

diff --git a/libavcodec/vp8.c b/libavcodec/vp8.c
index 74a05f3310..85f1723ae9 100644
--- a/libavcodec/vp8.c
+++ b/libavcodec/vp8.c
@@ -225,6 +225,9 @@ static int update_dimensions(VP8Context *s, int width, int height)
     s->intra4x4_pred_mode_base = av_mallocz(s->b4_stride*(4*s->mb_height+1));
     s->top_nnz                 = av_mallocz(s->mb_width*sizeof(*s->top_nnz));
 
+    if (!s->macroblocks_base || !s->intra4x4_pred_mode_base || !s->top_nnz)
+        return AVERROR(ENOMEM);
+
     s->macroblocks        = s->macroblocks_base        + 1 + s->mb_stride;
     s->intra4x4_pred_mode = s->intra4x4_pred_mode_base + 4 + s->b4_stride;