From 0273ceebbd01f9fd5238558e6151e0b9aa3305ab Mon Sep 17 00:00:00 2001 From: Michael Niedermayer Date: Tue, 8 Jul 2003 22:12:17 +0000 Subject: few more error checks Originally committed as revision 2021 to svn://svn.ffmpeg.org/ffmpeg/trunk --- libavcodec/mjpeg.c | 13 +++++++++---- 1 file changed, 9 insertions(+), 4 deletions(-) (limited to 'libavcodec/mjpeg.c') diff --git a/libavcodec/mjpeg.c b/libavcodec/mjpeg.c index f3f83a94d7..ee319f5bdb 100644 --- a/libavcodec/mjpeg.c +++ b/libavcodec/mjpeg.c @@ -842,7 +842,7 @@ typedef struct MJpegDecodeContext { static int mjpeg_decode_dht(MJpegDecodeContext *s); -static void build_vlc(VLC *vlc, const uint8_t *bits_table, const uint8_t *val_table, +static int build_vlc(VLC *vlc, const uint8_t *bits_table, const uint8_t *val_table, int nb_codes) { uint8_t huff_size[256]; @@ -851,7 +851,7 @@ static void build_vlc(VLC *vlc, const uint8_t *bits_table, const uint8_t *val_ta memset(huff_size, 0, sizeof(huff_size)); build_huffman_codes(huff_size, huff_code, bits_table, val_table); - init_vlc(vlc, 9, nb_codes, huff_size, 1, 1, huff_code, 2, 2); + return init_vlc(vlc, 9, nb_codes, huff_size, 1, 1, huff_code, 2, 2); } static int mjpeg_decode_init(AVCodecContext *avctx) @@ -970,7 +970,9 @@ static int mjpeg_decode_dht(MJpegDecodeContext *s) free_vlc(&s->vlcs[class][index]); dprintf("class=%d index=%d nb_codes=%d\n", class, index, code_max + 1); - build_vlc(&s->vlcs[class][index], bits_table, val_table, code_max + 1); + if(build_vlc(&s->vlcs[class][index], bits_table, val_table, code_max + 1) < 0){ + return -1; + } } return 0; } @@ -1778,7 +1780,10 @@ static int mjpeg_decode_frame(AVCodecContext *avctx, mjpeg_decode_dqt(s); break; case DHT: - mjpeg_decode_dht(s); + if(mjpeg_decode_dht(s) < 0){ + fprintf(stderr, "huffman table decode error\n"); + return -1; + } break; case SOF0: s->lossless=0; -- cgit v1.2.3