diff options
author | Tobias Bindhammer <tobias.bindhammer@uni-ulm.de> | 2010-08-31 11:15:11 +0400 |
---|---|---|
committer | Tobias Bindhammer <tobias.bindhammer@uni-ulm.de> | 2010-08-31 11:15:11 +0400 |
commit | 8731c86d03d062ad19f098b77ab1f1bc4ad7c406 (patch) | |
tree | 11fcb73e613fee5554cc1ccbed60ba9834f5e57c | |
parent | d1cacdb8dda4eb2a5532267b0aeb1d2afdf95f05 (diff) |
Solving memory leak and initialization problem with prev_pkt / pkt.
Originally committed as revision 25004 to svn://svn.ffmpeg.org/ffmpeg/trunk
-rw-r--r-- | libavformat/a64.c | 16 |
1 files changed, 13 insertions, 3 deletions
diff --git a/libavformat/a64.c b/libavformat/a64.c index 3bc03a6794..66fbcaee40 100644 --- a/libavformat/a64.c +++ b/libavformat/a64.c @@ -125,8 +125,16 @@ static int a64_write_packet(struct AVFormatContext *s, AVPacket *pkt) } /* backup current packet for next turn */ if(pkt->data) { - av_new_packet(&c->prev_pkt, pkt->size); - memcpy(c->prev_pkt.data, pkt->data, pkt->size); + /* no backup packet yet? create one! */ + if(!c->prev_pkt.data) av_new_packet(&c->prev_pkt, pkt->size); + /* we have a packet and data is big enough, reuse it */ + if(c->prev_pkt.data && c->prev_pkt.size >= pkt->size) { + memcpy(c->prev_pkt.data, pkt->data, pkt->size); + c->prev_pkt.size = pkt->size; + } else { + av_log(avctx, AV_LOG_ERROR, "Too less memory for prev_pkt.\n"); + return AVERROR(ENOMEM); + } } c->prev_frame_count = frame_count; break; @@ -145,9 +153,11 @@ static int a64_write_packet(struct AVFormatContext *s, AVPacket *pkt) static int a64_write_trailer(struct AVFormatContext *s) { A64MuxerContext *c = s->priv_data; - AVPacket pkt; + AVPacket pkt = {0}; /* need to flush last packet? */ if(c->interleaved) a64_write_packet(s, &pkt); + /* discard backed up packet */ + if(c->prev_pkt.data) av_destruct_packet(&c->prev_pkt); return 0; } |