wmv2enc: Check memory allocation

03eb5574 · Vittorio Giovara · a67b6794 · 03eb5574
Commit 03eb5574 authored Jul 24, 2015 by Vittorio Giovara
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

wmv2enc.c libavcodec/wmv2enc.c +3 -0

No files found.
--- a/libavcodec/wmv2enc.c
+++ b/libavcodec/wmv2enc.c
@@ -63,6 +63,9 @@ static av_cold int wmv2_encode_init(AVCodecContext *avctx)

    avctx->extradata_size = 4;
    avctx->extradata      = av_mallocz(avctx->extradata_size + 10);
+    if (!avctx->extradata)
+        return AVERROR(ENOMEM);
+
    encode_ext_header(w);

    return 0;