1
0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-11-26 19:01:44 +02:00

smvjpeg: use refcounted frames to avoid mem leaks

Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
Hendrik Leppkes 2013-06-21 21:56:18 +02:00 committed by Michael Niedermayer
parent 1a405c683e
commit d76fff7df7

View File

@ -111,6 +111,7 @@ static av_cold int smvjpeg_decode_init(AVCodecContext *avctx)
s->avctx = avcodec_alloc_context3(codec); s->avctx = avcodec_alloc_context3(codec);
av_dict_set(&thread_opt, "threads", "1", 0); av_dict_set(&thread_opt, "threads", "1", 0);
s->avctx->refcounted_frames = 1;
s->avctx->flags = avctx->flags; s->avctx->flags = avctx->flags;
s->avctx->idct_algo = avctx->idct_algo; s->avctx->idct_algo = avctx->idct_algo;
if (ff_codec_open2_recursive(s->avctx, codec, &thread_opt) < 0) { if (ff_codec_open2_recursive(s->avctx, codec, &thread_opt) < 0) {
@ -133,9 +134,10 @@ static int smvjpeg_decode_frame(AVCodecContext *avctx, void *data, int *data_siz
cur_frame = avpkt->pts % s->frames_per_jpeg; cur_frame = avpkt->pts % s->frames_per_jpeg;
/* Are we at the start of a block? */ /* Are we at the start of a block? */
if (!cur_frame) if (!cur_frame) {
av_frame_unref(mjpeg_data);
ret = avcodec_decode_video2(s->avctx, mjpeg_data, &s->mjpeg_data_size, avpkt); ret = avcodec_decode_video2(s->avctx, mjpeg_data, &s->mjpeg_data_size, avpkt);
else if (!s->mjpeg_data_size) } else if (!s->mjpeg_data_size)
return AVERROR(EINVAL); return AVERROR(EINVAL);
desc = av_pix_fmt_desc_get(s->avctx->pix_fmt); desc = av_pix_fmt_desc_get(s->avctx->pix_fmt);
@ -177,6 +179,7 @@ static av_cold int smvjpeg_decode_end(AVCodecContext *avctx)
MJpegDecodeContext *jpg = &s->jpg; MJpegDecodeContext *jpg = &s->jpg;
jpg->picture_ptr = NULL; jpg->picture_ptr = NULL;
av_frame_free(&s->picture[0]);
av_frame_free(&s->picture[1]); av_frame_free(&s->picture[1]);
ff_codec_close_recursive(s->avctx); ff_codec_close_recursive(s->avctx);
av_freep(&s->avctx); av_freep(&s->avctx);