bitstream: Check the result of av_malloc()

29c455ce · Diego Biurrun · 3b4fa548 · 29c455ce
Commit 29c455ce authored Oct 15, 2013 by Diego Biurrun
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

bitstream.c libavcodec/bitstream.c +2 -0

No files found.
--- a/libavcodec/bitstream.c
+++ b/libavcodec/bitstream.c
@@ -283,6 +283,8 @@ int ff_init_vlc_sparse(VLC *vlc, int nb_bits, int nb_codes,
    av_dlog(NULL, "build table nb_codes=%d\n", nb_codes);

    buf = av_malloc((nb_codes + 1) * sizeof(VLCcode));
+    if (!buf)
+        return AVERROR(ENOMEM);

    assert(symbols_size <= 2 || !symbols);
    j = 0;