avformat/sierravmd: check for memory allocation failures

Signed-off-by: Paul B Mahol <onemda@gmail.com>

diff --git a/libavformat/sierravmd.c b/libavformat/sierravmd.c
index b2d1e5dd5906a94f7f0ac384325bde2b0793c2bd..059945dc0f69c0e92e6edcb5e43643caec92e3bb 100644 (file)
--- a/libavformat/sierravmd.c
+++ b/libavformat/sierravmd.c
@@ -126,8 +126,10 @@ static int vmd_read_header(AVFormatContext *s)
             vst->codec->width >>= 1;
             vst->codec->height >>= 1;
         }
-        vst->codec->extradata_size = VMD_HEADER_SIZE;
         vst->codec->extradata = av_mallocz(VMD_HEADER_SIZE + FF_INPUT_BUFFER_PADDING_SIZE);
+        if (!vst->codec->extradata)
+            return AVERROR(ENOMEM);
+        vst->codec->extradata_size = VMD_HEADER_SIZE;
         memcpy(vst->codec->extradata, vmd->vmd_header, VMD_HEADER_SIZE);
     }