generic-library/ffmpeg
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

wma: Check memory allocation

Browse Source
This commit is contained in:
Vittorio Giovara 2015-05-31 14:55:35 +02:00 committed by Luca Barbato
parent e1ea365f7e
commit 83797da6e3
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
libavcodec/wmaenc.c
View File

@ -62,11 +62,15 @@ static av_cold int encode_init(AVCodecContext *avctx)
flags2 = 1; flags2 = 1;
if (avctx->codec->id == AV_CODEC_ID_WMAV1) { if (avctx->codec->id == AV_CODEC_ID_WMAV1) {
extradata = av_malloc(4); extradata = av_malloc(4);
if (!extradata)
return AVERROR(ENOMEM);
avctx->extradata_size = 4; avctx->extradata_size = 4;
AV_WL16(extradata, flags1); AV_WL16(extradata, flags1);
AV_WL16(extradata + 2, flags2); AV_WL16(extradata + 2, flags2);
} else if (avctx->codec->id == AV_CODEC_ID_WMAV2) { } else if (avctx->codec->id == AV_CODEC_ID_WMAV2) {
extradata = av_mallocz(10); extradata = av_mallocz(10);
if (!extradata)
return AVERROR(ENOMEM);
avctx->extradata_size = 10; avctx->extradata_size = 10;
AV_WL32(extradata, flags1); AV_WL32(extradata, flags1);
AV_WL16(extradata + 4, flags2); AV_WL16(extradata + 4, flags2);