wmv2enc: Check memory allocation

10 years ago · 03eb557414
parent a67b67944a
commit 03eb557414
1 changed files with 3 additions and 0 deletions
--- 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;