From 50cbe09d8ced75422571d29bbec1f35a33a0d3ed Mon Sep 17 00:00:00 2001 From: Michael Niedermayer Date: Fri, 30 Nov 2012 15:58:19 +0100 Subject: smacker: check smacker_decode_tree() return value Signed-off-by: Michael Niedermayer --- libavcodec/smacker.c | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) (limited to 'libavcodec/smacker.c') diff --git a/libavcodec/smacker.c b/libavcodec/smacker.c index ce1262d90b..095f0d0e52 100644 --- a/libavcodec/smacker.c +++ b/libavcodec/smacker.c @@ -205,7 +205,9 @@ static int smacker_decode_header_tree(SmackVContext *smk, GetBitContext *gb, int tmp2.values = av_mallocz(256 * sizeof(int)); if(get_bits1(gb)) { - smacker_decode_tree(gb, &tmp1, 0, 0); + res = smacker_decode_tree(gb, &tmp1, 0, 0); + if (res < 0) + return res; skip_bits1(gb); if(tmp1.current > 1) { res = init_vlc(&vlc[0], SMKTREE_BITS, tmp1.length, @@ -221,7 +223,9 @@ static int smacker_decode_header_tree(SmackVContext *smk, GetBitContext *gb, int av_log(smk->avctx, AV_LOG_ERROR, "Skipping low bytes tree\n"); } if(get_bits1(gb)){ - smacker_decode_tree(gb, &tmp2, 0, 0); + res = smacker_decode_tree(gb, &tmp2, 0, 0); + if (res < 0) + return res; skip_bits1(gb); if(tmp2.current > 1) { res = init_vlc(&vlc[1], SMKTREE_BITS, tmp2.length, @@ -655,7 +659,9 @@ static int smka_decode_frame(AVCodecContext *avctx, void *data, h[i].lengths = av_mallocz(256 * sizeof(int)); h[i].values = av_mallocz(256 * sizeof(int)); skip_bits1(&gb); - smacker_decode_tree(&gb, &h[i], 0, 0); + res = smacker_decode_tree(&gb, &h[i], 0, 0); + if (res < 0) + return res; skip_bits1(&gb); if(h[i].current > 1) { res = init_vlc(&vlc[i], SMKTREE_BITS, h[i].length, -- cgit v1.2.3