[FFmpeg-cvslog] r22804 - in trunk/libavformat: rtpdec.c rtpdec.h

[mstorsjo]

Author: mstorsjo
Date: Mon Apr  5 19:25:39 2010
New Revision: 22804

Log:
Fix leaks in the AAC RTP depacketizer

Modified:
   trunk/libavformat/rtpdec.c
   trunk/libavformat/rtpdec.h

Modified: trunk/libavformat/rtpdec.c
==============================================================================
--- trunk/libavformat/rtpdec.c	Mon Apr  5 18:13:51 2010	(r22803)
+++ trunk/libavformat/rtpdec.c	Mon Apr  5 19:25:39 2010	(r22804)
@@ -400,7 +400,11 @@ static int rtp_parse_mp4_au(RTPDemuxCont
         return -1;
 
     infos->nb_au_headers = au_headers_length / au_header_size;
+    if (!infos->au_headers || infos->au_headers_allocated < infos->nb_au_headers) {
+        av_free(infos->au_headers);
     infos->au_headers = av_malloc(sizeof(struct AUHeaders) * infos->nb_au_headers);
+        infos->au_headers_allocated = infos->nb_au_headers;
+    }
 
     /* XXX: We handle multiple AU Section as only one (need to fix this for interleaving)
        In my test, the FAAD decoder does not behave correctly when sending each AU one by one
@@ -599,6 +603,8 @@ int rtp_parse_packet(RTPDemuxContext *s,
 void rtp_parse_close(RTPDemuxContext *s)
 {
     // TODO: fold this into the protocol specific data fields.
+    av_free(s->rtp_payload_data->mode);
+    av_free(s->rtp_payload_data->au_headers);
     if (!strcmp(ff_rtp_enc_name(s->payload_type), "MP2T")) {
         ff_mpegts_parse_close(s->ts);
     }

Modified: trunk/libavformat/rtpdec.h
==============================================================================
--- trunk/libavformat/rtpdec.h	Mon Apr  5 18:13:51 2010	(r22803)
+++ trunk/libavformat/rtpdec.h	Mon Apr  5 19:25:39 2010	(r22804)
@@ -48,6 +48,7 @@ typedef struct rtp_payload_data
         int rap_flag;
         int streamstate;
     } *au_headers;
+    int au_headers_allocated;
     int nb_au_headers;
     int au_headers_length_bytes;
     int cur_au_index;