[FFmpeg-cvslog] avformat/movenc: allow rewriting extradata

[Umair Khan]

ffmpeg | branch: master | Umair Khan <omerjerk at gmail.com> | Thu Aug 18 00:45:42 2016 +0530| [4f6f56114e5618b601bd01a34d95cf750c7baa9b] | committer: Michael Niedermayer

avformat/movenc: allow rewriting extradata

Signed-off-by: Umair Khan <omerjerk at gmail.com>
Signed-off-by: Michael Niedermayer <michael at niedermayer.cc>

> http://git.videolan.org/gitweb.cgi/ffmpeg.git/?a=commit;h=4f6f56114e5618b601bd01a34d95cf750c7baa9b
---

 libavformat/movenc.c | 31 +++++++++++++++++++++++++++++++
 libavformat/movenc.h |  2 ++
 2 files changed, 33 insertions(+)

diff --git a/libavformat/movenc.c b/libavformat/movenc.c
index 9064cec..282a3a8 100644
--- a/libavformat/movenc.c
+++ b/libavformat/movenc.c
@@ -4919,6 +4919,21 @@ static int mov_write_single_packet(AVFormatContext *s, AVPacket *pkt)
                     trk->start_cts = 0;
             }
 
+            if (trk->par->codec_id == AV_CODEC_ID_MP4ALS) {
+                int side_size = 0;
+                uint8_t *side = av_packet_get_side_data(pkt, AV_PKT_DATA_NEW_EXTRADATA, &side_size);
+                if (side && side_size > 0 && (side_size != par->extradata_size || memcmp(side, par->extradata, side_size))) {
+                    av_free(par->extradata);
+                    par->extradata = av_mallocz(side_size + AV_INPUT_BUFFER_PADDING_SIZE);
+                    if (!par->extradata) {
+                        return AVERROR(ENOMEM);
+                    }
+                    memcpy(par->extradata, side, side_size);
+                    par->extradata_size = side_size;
+                    mov->need_rewrite_extradata = 1;
+                }
+            }
+
             return 0;             /* Discard 0 sized packets */
         }
 
@@ -5921,6 +5936,22 @@ static int mov_write_trailer(AVFormatContext *s)
     int i;
     int64_t moov_pos;
 
+    if (mov->need_rewrite_extradata) {
+        for (i = 0; i < s->nb_streams; i++) {
+            MOVTrack *track = &mov->tracks[i];
+            AVCodecParameters *par = track->par;
+
+            track->vos_len  = par->extradata_size;
+            track->vos_data = av_malloc(track->vos_len);
+            if (!track->vos_data) {
+                AVERROR(ENOMEM);
+                goto error;
+            }
+            memcpy(track->vos_data, par->extradata, track->vos_len);
+        }
+        mov->need_rewrite_extradata = 0;
+    }
+
     /*
      * Before actually writing the trailer, make sure that there are no
      * dangling subtitles, that need a terminating sample.
diff --git a/libavformat/movenc.h b/libavformat/movenc.h
index 6e9f5ac..894a1b0 100644
--- a/libavformat/movenc.h
+++ b/libavformat/movenc.h
@@ -215,6 +215,8 @@ typedef struct MOVMuxContext {
     uint8_t *encryption_kid;
     int encryption_kid_len;
 
+    int need_rewrite_extradata;
+
 } MOVMuxContext;
 
 #define FF_MOV_FLAG_RTP_HINT              (1 <<  0)