Commit f80ccf07 authored by Michael Niedermayer's avatar Michael Niedermayer

avcodec/ff_init_vlc_sparse: use a local array for static vlcs avoiding malloc()

Signed-off-by: 's avatarMichael Niedermayer <michaelni@gmx.at>
parent 6af83263
...@@ -270,6 +270,7 @@ int ff_init_vlc_sparse(VLC *vlc, int nb_bits, int nb_codes, ...@@ -270,6 +270,7 @@ int ff_init_vlc_sparse(VLC *vlc, int nb_bits, int nb_codes,
{ {
VLCcode *buf; VLCcode *buf;
int i, j, ret; int i, j, ret;
VLCcode localbuf[1500]; // the maximum currently needed is 1296 by rv34
void *state; void *state;
vlc->bits = nb_bits; vlc->bits = nb_bits;
...@@ -281,17 +282,18 @@ int ff_init_vlc_sparse(VLC *vlc, int nb_bits, int nb_codes, ...@@ -281,17 +282,18 @@ int ff_init_vlc_sparse(VLC *vlc, int nb_bits, int nb_codes,
} }
} }
av_assert0(!vlc->table_size); av_assert0(!vlc->table_size);
av_assert0(nb_codes + 1 <= FF_ARRAY_ELEMS(localbuf));
buf = localbuf;
} else { } else {
vlc->table = NULL; vlc->table = NULL;
vlc->table_allocated = 0; vlc->table_allocated = 0;
vlc->table_size = 0; vlc->table_size = 0;
}
av_dlog(NULL, "build table nb_codes=%d\n", nb_codes);
buf = av_malloc((nb_codes + 1) * sizeof(VLCcode)); buf = av_malloc((nb_codes + 1) * sizeof(VLCcode));
if (!buf) if (!buf)
return AVERROR(ENOMEM); return AVERROR(ENOMEM);
}
av_assert0(symbols_size <= 2 || !symbols); av_assert0(symbols_size <= 2 || !symbols);
j = 0; j = 0;
...@@ -329,7 +331,6 @@ int ff_init_vlc_sparse(VLC *vlc, int nb_bits, int nb_codes, ...@@ -329,7 +331,6 @@ int ff_init_vlc_sparse(VLC *vlc, int nb_bits, int nb_codes,
ret = build_table(vlc, nb_bits, nb_codes, buf, flags); ret = build_table(vlc, nb_bits, nb_codes, buf, flags);
av_free(buf);
if (flags & INIT_VLC_USE_NEW_STATIC) { if (flags & INIT_VLC_USE_NEW_STATIC) {
if(vlc->table_size != vlc->table_allocated) if(vlc->table_size != vlc->table_allocated)
av_log(NULL, AV_LOG_ERROR, "needed %d had %d\n", vlc->table_size, vlc->table_allocated); av_log(NULL, AV_LOG_ERROR, "needed %d had %d\n", vlc->table_size, vlc->table_allocated);
...@@ -337,6 +338,7 @@ int ff_init_vlc_sparse(VLC *vlc, int nb_bits, int nb_codes, ...@@ -337,6 +338,7 @@ int ff_init_vlc_sparse(VLC *vlc, int nb_bits, int nb_codes,
av_assert0(state == vlc); av_assert0(state == vlc);
av_assert0(ret >= 0); av_assert0(ret >= 0);
} else { } else {
av_free(buf);
if (ret < 0) { if (ret < 0) {
av_freep(&vlc->table); av_freep(&vlc->table);
return ret; return ret;
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment