From 8559a714155a7bc79d4802558f509e211379a8a0 Mon Sep 17 00:00:00 2001 From: Michael Niedermayer Date: Sun, 22 Feb 2015 21:23:39 +0100 Subject: avcodec/wmaenc: Check extradata allocations Fixes CID1257842 Signed-off-by: Michael Niedermayer --- 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 08d45e9273..b922acdee8 100644 --- a/libavcodec/wmaenc.c +++ b/libavcodec/wmaenc.c @@ -60,11 +60,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