wmv2enc: Check memory allocation

diff --git a/libavcodec/wmv2enc.c b/libavcodec/wmv2enc.c
index e4e51d32f933f2d5f13f4e8e6efe1501c057082a..e4d4159e373e70a4cdca992d5a934ffd5fa1683f 100644 (file)
--- 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;