From 83797da6e36c1aadd85f41ca237dce823fc7bfa1 Mon Sep 17 00:00:00 2001 From: Vittorio Giovara Date: Sun, 31 May 2015 14:55:35 +0200 Subject: wma: Check memory allocation --- libavcodec/wmaenc.c | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'libavcodec/wmaenc.c') diff --git a/libavcodec/wmaenc.c b/libavcodec/wmaenc.c index e801663d66..c176daa7c9 100644 --- a/libavcodec/wmaenc.c +++ b/libavcodec/wmaenc.c @@ -62,11 +62,15 @@ static av_cold int encode_init(AVCodecContext *avctx) flags2 = 1; if (avctx->codec->id == AV_CODEC_ID_WMAV1) { extradata = av_malloc(4); + if (!extradata) + return AVERROR(ENOMEM); avctx->extradata_size = 4; AV_WL16(extradata, flags1); AV_WL16(extradata + 2, flags2); } else if (avctx->codec->id == AV_CODEC_ID_WMAV2) { extradata = av_mallocz(10); + if (!extradata) + return AVERROR(ENOMEM); avctx->extradata_size = 10; AV_WL32(extradata, flags1); AV_WL16(extradata + 4, flags2); -- cgit v1.2.3