From cb8f01951b3e594f754d4d240f66cc01d6881e4d Mon Sep 17 00:00:00 2001 From: Andreas Rheinhardt Date: Thu, 6 May 2021 17:08:44 +0200 Subject: [PATCH] avcodec/wma: Check initializing VLC Initializing a VLC entails implicit allocations which may fail. Signed-off-by: Andreas Rheinhardt --- libavcodec/wma.c | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/libavcodec/wma.c b/libavcodec/wma.c index ee04ef725f..004ffb5515 100644 --- a/libavcodec/wma.c +++ b/libavcodec/wma.c @@ -41,9 +41,11 @@ static av_cold int init_coef_vlc(VLC *vlc, uint16_t **prun_table, const uint16_t *levels_table = vlc_table->levels; uint16_t *run_table, *int_table; float *flevel_table; - int i, l, j, k, level; + int i, l, j, k, level, ret; - init_vlc(vlc, VLCBITS, n, table_bits, 1, 1, table_codes, 4, 4, 0); + ret = init_vlc(vlc, VLCBITS, n, table_bits, 1, 1, table_codes, 4, 4, 0); + if (ret < 0) + return ret; run_table = av_malloc_array(n, sizeof(uint16_t)); flevel_table = av_malloc_array(n, sizeof(*flevel_table));