wma: Check memory allocation

83797da6 · Vittorio Giovara · Luca Barbato · e1ea365f · 83797da6
Commit 83797da6 authored May 31, 2015 by Vittorio Giovara Committed by Luca Barbato May 31, 2015
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

wmaenc.c libavcodec/wmaenc.c +4 -0

No files found.
--- 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);