From 94372592767fb551060217df37f5aa3130ba1ca8 Mon Sep 17 00:00:00 2001 From: Paul B Mahol Date: Fri, 26 Jul 2013 13:49:36 +0000 Subject: vmnc: check return values of reallocation Also avoid memory leaks. Signed-off-by: Paul B Mahol --- libavcodec/vmnc.c | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) (limited to 'libavcodec/vmnc.c') diff --git a/libavcodec/vmnc.c b/libavcodec/vmnc.c index 7ca9aefb8a..9fe99408a4 100644 --- a/libavcodec/vmnc.c +++ b/libavcodec/vmnc.c @@ -361,9 +361,11 @@ static int decode_frame(AVCodecContext *avctx, void *data, int *got_frame, av_log(avctx, AV_LOG_ERROR, "Cursor hot spot is not in image: %ix%i of %ix%i cursor size\n", c->cur_hx, c->cur_hy, c->cur_w, c->cur_h); c->cur_hx = c->cur_hy = 0; } - c->curbits = av_realloc(c->curbits, c->cur_w * c->cur_h * c->bpp2); - c->curmask = av_realloc(c->curmask, c->cur_w * c->cur_h * c->bpp2); - c->screendta = av_realloc(c->screendta, c->cur_w * c->cur_h * c->bpp2); + c->curbits = av_realloc_f(c->curbits, c->cur_w * c->cur_h, c->bpp2); + c->curmask = av_realloc_f(c->curmask, c->cur_w * c->cur_h, c->bpp2); + c->screendta = av_realloc_f(c->screendta, c->cur_w * c->cur_h, c->bpp2); + if (!c->curbits || !c->curmask || !c->screendta) + return AVERROR(ENOMEM); load_cursor(c, src); src += w * h * c->bpp2 * 2; break; -- cgit v1.2.3