diff options
author | Michael Niedermayer <michaelni@gmx.at> | 2015-02-22 21:23:39 +0100 |
---|---|---|
committer | Michael Niedermayer <michaelni@gmx.at> | 2015-02-22 21:23:39 +0100 |
commit | 8559a714155a7bc79d4802558f509e211379a8a0 (patch) | |
tree | eb28a297ab4db17e0c52976ec8d0f210aa9fc0ca | |
parent | e0c7ba40cf0bb521430f4a808ed1f1a408093f9c (diff) | |
download | ffmpeg-8559a714155a7bc79d4802558f509e211379a8a0.tar.gz |
avcodec/wmaenc: Check extradata allocations
Fixes CID1257842
Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
-rw-r--r-- | libavcodec/wmaenc.c | 4 |
1 files changed, 4 insertions, 0 deletions
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); |