Fix memleak in mjpeg decoder.

Originally committed as revision 25549 to svn://svn.ffmpeg.org/ffmpeg/trunk
This commit is contained in:
Michael Niedermayer 2010-10-22 02:25:26 +00:00
parent a2d4dfd493
commit 4c3267aaad
1 changed files with 6 additions and 2 deletions

View File

@ -194,8 +194,12 @@ int ff_mjpeg_decode_dht(MJpegDecodeContext *s)
if(build_vlc(&s->vlcs[class][index], bits_table, val_table, code_max + 1, 0, class > 0) < 0){ if(build_vlc(&s->vlcs[class][index], bits_table, val_table, code_max + 1, 0, class > 0) < 0){
return -1; return -1;
} }
if(class>0 && build_vlc(&s->vlcs[2][index], bits_table, val_table, code_max + 1, 0, 0) < 0){
if(class>0){
free_vlc(&s->vlcs[2][index]);
if(build_vlc(&s->vlcs[2][index], bits_table, val_table, code_max + 1, 0, 0) < 0){
return -1; return -1;
}
} }
} }
return 0; return 0;
@ -1521,7 +1525,7 @@ av_cold int ff_mjpeg_decode_end(AVCodecContext *avctx)
av_freep(&s->ljpeg_buffer); av_freep(&s->ljpeg_buffer);
s->ljpeg_buffer_size=0; s->ljpeg_buffer_size=0;
for(i=0;i<2;i++) { for(i=0;i<3;i++) {
for(j=0;j<4;j++) for(j=0;j<4;j++)
free_vlc(&s->vlcs[i][j]); free_vlc(&s->vlcs[i][j]);
} }