From 9d5894b2399e3b2ce9fe8ddb5a28db66affefe25 Mon Sep 17 00:00:00 2001 From: Michael Niedermayer Date: Sun, 22 May 2016 01:54:50 +0200 Subject: avformat/avienc: Fix memleaks on errors Fixes CID1361951 Signed-off-by: Michael Niedermayer --- libavformat/avienc.c | 15 +++++++++------ 1 file changed, 9 insertions(+), 6 deletions(-) (limited to 'libavformat/avienc.c') diff --git a/libavformat/avienc.c b/libavformat/avienc.c index 2c748efb2d..9504a022f0 100644 --- a/libavformat/avienc.c +++ b/libavformat/avienc.c @@ -728,10 +728,10 @@ static int avi_write_packet(AVFormatContext *s, AVPacket *pkt) } else reshuffle_ret = 0; if (par->format == AV_PIX_FMT_PAL8) { - int ret2 = ff_get_packet_palette(s, opkt, reshuffle_ret, avist->palette); - if (ret2 < 0) - return ret2; - if (ret2) { + ret = ff_get_packet_palette(s, opkt, reshuffle_ret, avist->palette); + if (ret < 0) + goto fail; + if (ret) { int pal_size = 1 << par->bits_per_coded_sample; int pc_tag, i; @@ -763,7 +763,7 @@ static int avi_write_packet(AVFormatContext *s, AVPacket *pkt) ret = avi_add_ientry(s, stream_index, tag, AVIIF_NO_TIME, pal_size * 4 + 4); if (ret < 0) - return ret; + goto fail; } pc_tag = ff_start_tag(pb, tag); avio_w8(pb, 0); @@ -780,7 +780,10 @@ static int avi_write_packet(AVFormatContext *s, AVPacket *pkt) } if (reshuffle_ret) { ret = avi_write_packet_internal(s, pkt); - av_packet_free(&pkt); + +fail: + if (reshuffle_ret) + av_packet_free(&pkt); return ret; } } -- cgit v1.2.3