From fef2f4722bcbfe1c3802898b0f5deb6356132c91 Mon Sep 17 00:00:00 2001 From: Vittorio Giovara Date: Sun, 31 May 2015 14:54:48 +0200 Subject: [PATCH] xvid: Check memory allocation --- libavcodec/libxvid.c | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/libavcodec/libxvid.c b/libavcodec/libxvid.c index 46da773052..97ff95b843 100644 --- a/libavcodec/libxvid.c +++ b/libavcodec/libxvid.c @@ -276,6 +276,8 @@ static int xvid_strip_vol_header(AVCodecContext *avctx, AVPacket *pkt, /* We need to store the header, so extract it */ if (!avctx->extradata) { avctx->extradata = av_malloc(vo_len); + if (!avctx->extradata) + return AVERROR(ENOMEM); memcpy(avctx->extradata, pkt->data, vo_len); avctx->extradata_size = vo_len; } @@ -594,11 +596,15 @@ static av_cold int xvid_encode_init(AVCodecContext *avctx) if (avctx->intra_matrix) { intra = avctx->intra_matrix; x->intra_matrix = av_malloc(sizeof(unsigned char) * 64); + if (!x->intra_matrix) + return AVERROR(ENOMEM); } else intra = NULL; if (avctx->inter_matrix) { inter = avctx->inter_matrix; x->inter_matrix = av_malloc(sizeof(unsigned char) * 64); + if (!x->inter_matrix) + return AVERROR(ENOMEM); } else inter = NULL;