virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2025-06-04 05:57:49 +02:00
Code Issues Releases Activity

zmbv: check av_realloc() return values and avoid memleaks on ENOMEM

Browse Source
This commit is contained in:
Janne Grunau 2012-02-01 01:00:17 +01:00
parent d4eeadcbbf
commit cb0b284381
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
libavcodec/zmbv.c
View File

@ -404,6 +404,7 @@ static int decode_frame(AVCodecContext *avctx, void *data, int *data_size, AVPac
int zret = Z_OK; // Zlib return code int zret = Z_OK; // Zlib return code
int len = buf_size; int len = buf_size;
int hi_ver, lo_ver; int hi_ver, lo_ver;
uint8_t *tmp;
if (c->pic.data[0]) if (c->pic.data[0])
avctx->release_buffer(avctx, &c->pic); avctx->release_buffer(avctx, &c->pic);
@ -485,8 +486,14 @@ static int decode_frame(AVCodecContext *avctx, void *data, int *data_size, AVPac
return -1; return -1;
} }
c->cur = av_realloc(c->cur, avctx->width * avctx->height * (c->bpp / 8)); tmp = av_realloc(c->cur, avctx->width * avctx->height * (c->bpp / 8));
c->prev = av_realloc(c->prev, avctx->width * avctx->height * (c->bpp / 8)); if (!tmp)
return AVERROR(ENOMEM);
c->cur = tmp;
tmp = av_realloc(c->prev, avctx->width * avctx->height * (c->bpp / 8));
if (!tmp)
return AVERROR(ENOMEM);
c->prev = tmp;
c->bx = (c->width + c->bw - 1) / c->bw; c->bx = (c->width + c->bw - 1) / c->bw;
c->by = (c->height+ c->bh - 1) / c->bh; c->by = (c->height+ c->bh - 1) / c->bh;
} }