bfi: check return value of av_malloc()

[ffmpeg] / libavformat / bfi.c

diff --git a/libavformat/bfi.c b/libavformat/bfi.c
index a28e09a4f1cf19099f453535a8892389cd176845..7a4f436bf37b70980489fcb54347c8bacbfa3804 100644 (file)
--- 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);