comparison vp8.c @ 12169:7501f327cfd1 libavcodec

vp8: Check for malloc failure
author conrad
date Fri, 16 Jul 2010 07:20:31 +0000
parents e97aba4d16ea
children 6f0db2eeaf70
comparison
equal deleted inserted replaced
12168:b246b214c2e9 12169:7501f327cfd1
222 s->b4_stride = 4*s->mb_stride; 222 s->b4_stride = 4*s->mb_stride;
223 223
224 s->macroblocks_base = av_mallocz(s->mb_stride*(s->mb_height+1)*sizeof(*s->macroblocks)); 224 s->macroblocks_base = av_mallocz(s->mb_stride*(s->mb_height+1)*sizeof(*s->macroblocks));
225 s->intra4x4_pred_mode_base = av_mallocz(s->b4_stride*(4*s->mb_height+1)); 225 s->intra4x4_pred_mode_base = av_mallocz(s->b4_stride*(4*s->mb_height+1));
226 s->top_nnz = av_mallocz(s->mb_width*sizeof(*s->top_nnz)); 226 s->top_nnz = av_mallocz(s->mb_width*sizeof(*s->top_nnz));
227
228 if (!s->macroblocks_base || !s->intra4x4_pred_mode_base || !s->top_nnz)
229 return AVERROR(ENOMEM);
227 230
228 s->macroblocks = s->macroblocks_base + 1 + s->mb_stride; 231 s->macroblocks = s->macroblocks_base + 1 + s->mb_stride;
229 s->intra4x4_pred_mode = s->intra4x4_pred_mode_base + 4 + s->b4_stride; 232 s->intra4x4_pred_mode = s->intra4x4_pred_mode_base + 4 + s->b4_stride;
230 233
231 memset(s->intra4x4_pred_mode_base, DC_PRED, s->b4_stride); 234 memset(s->intra4x4_pred_mode_base, DC_PRED, s->b4_stride);