bfi: check return value of av_malloc()

Signed-off-by: Paul B Mahol <onemda@gmail.com>
12 years ago · 51f1bf3309
parent b791a0831b
commit 51f1bf3309
1 changed files with 2 additions and 0 deletions
--- a/libavformat/bfi.c
+++ b/libavformat/bfi.c
@ -81,6 +81,8 @@ static int bfi_read_header(AVFormatContext * s)
    /*Load the palette to extradata */
    avio_skip(pb, 8);
    vstream->codec->extradata      = av_malloc(768);
+    if (!vstream->codec->extradata)
+        return AVERROR(ENOMEM);
    vstream->codec->extradata_size = 768;
    avio_read(pb, vstream->codec->extradata,
               vstream->codec->extradata_size);