lavf/movenc: fix invalid free with timecode meta and tmcd data copy.
[ffmpeg.git] / libavformat / movenc.h
1 /*
2  * MOV, 3GP, MP4 muxer
3  * Copyright (c) 2003 Thomas Raivio
4  * Copyright (c) 2004 Gildas Bazin <gbazin at videolan dot org>
5  * Copyright (c) 2009 Baptiste Coudurier <baptiste dot coudurier at gmail dot com>
6  *
7  * This file is part of FFmpeg.
8  *
9  * FFmpeg is free software; you can redistribute it and/or
10  * modify it under the terms of the GNU Lesser General Public
11  * License as published by the Free Software Foundation; either
12  * version 2.1 of the License, or (at your option) any later version.
13  *
14  * FFmpeg is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
17  * Lesser General Public License for more details.
18  *
19  * You should have received a copy of the GNU Lesser General Public
20  * License along with FFmpeg; if not, write to the Free Software
21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
22  */
23
24 #ifndef AVFORMAT_MOVENC_H
25 #define AVFORMAT_MOVENC_H
26
27 #include "avformat.h"
28
29 #define MOV_INDEX_CLUSTER_SIZE 16384
30 #define MOV_TIMESCALE 1000
31
32 #define RTP_MAX_PACKET_SIZE 1450
33
34 #define MODE_MP4  0x01
35 #define MODE_MOV  0x02
36 #define MODE_3GP  0x04
37 #define MODE_PSP  0x08 // example working PSP command line:
38 // ffmpeg -i testinput.avi  -f psp -r 14.985 -s 320x240 -b 768 -ar 24000 -ab 32 M4V00001.MP4
39 #define MODE_3G2  0x10
40 #define MODE_IPOD 0x20
41 #define MODE_ISM  0x40
42
43 typedef struct MOVIentry {
44     uint64_t     pos;
45     int64_t      dts;
46     unsigned int size;
47     unsigned int samples_in_chunk;
48     unsigned int chunkNum;              ///< Chunk number if the current entry is a chunk start otherwise 0
49     unsigned int entries;
50     int          cts;
51 #define MOV_SYNC_SAMPLE         0x0001
52 #define MOV_PARTIAL_SYNC_SAMPLE 0x0002
53     uint32_t     flags;
54 } MOVIentry;
55
56 typedef struct HintSample {
57     uint8_t *data;
58     int size;
59     int sample_number;
60     int offset;
61     int own_data;
62 } HintSample;
63
64 typedef struct {
65     int size;
66     int len;
67     HintSample *samples;
68 } HintSampleQueue;
69
70 typedef struct {
71     int64_t offset;
72     int64_t time;
73     int64_t duration;
74     int64_t tfrf_offset;
75 } MOVFragmentInfo;
76
77 typedef struct MOVIndex {
78     int         mode;
79     int         entry;
80     unsigned    timescale;
81     uint64_t    time;
82     int64_t     track_duration;
83     long        sample_count;
84     long        sample_size;
85     long        chunkCount;
86     int         has_keyframes;
87 #define MOV_TRACK_CTTS         0x0001
88 #define MOV_TRACK_STPS         0x0002
89     uint32_t    flags;
90 #define MOV_TIMECODE_FLAG_DROPFRAME     0x0001
91 #define MOV_TIMECODE_FLAG_24HOURSMAX    0x0002
92 #define MOV_TIMECODE_FLAG_ALLOWNEGATIVE 0x0004
93     uint32_t    timecode_flags;
94     int         language;
95     int         track_id;
96     int         tag; ///< stsd fourcc
97     AVCodecContext *enc;
98
99     int         vos_len;
100     uint8_t     *vos_data;
101     MOVIentry   *cluster;
102     int         audio_vbr;
103     int         height; ///< active picture (w/o VBI) height for D-10/IMX
104     uint32_t    tref_tag;
105     int         tref_id; ///< trackID of the referenced track
106     int64_t     start_dts;
107
108     int         hint_track;   ///< the track that hints this track, -1 if no hint track is set
109     int         src_track;    ///< the track that this hint (or tmcd) track describes
110     AVFormatContext *rtp_ctx; ///< the format context for the hinting rtp muxer
111     uint32_t    prev_rtp_ts;
112     int64_t     cur_rtp_ts_unwrapped;
113     uint32_t    max_packet_size;
114
115     int64_t     default_duration;
116     uint32_t    default_sample_flags;
117     uint32_t    default_size;
118
119     HintSampleQueue sample_queue;
120
121     AVIOContext *mdat_buf;
122     int64_t     moof_size_offset;
123     int64_t     data_offset;
124     int64_t     frag_start;
125     int64_t     tfrf_offset;
126
127     int         nb_frag_info;
128     MOVFragmentInfo *frag_info;
129
130     struct {
131         int64_t struct_offset;
132         int     first_packet_seq;
133         int     first_packet_entry;
134         int     packet_seq;
135         int     packet_entry;
136         int     slices;
137     } vc1_info;
138 } MOVTrack;
139
140 typedef struct MOVMuxContext {
141     const AVClass *av_class;
142     int     mode;
143     int64_t time;
144     int     nb_streams;
145     int     nb_meta_tmcd;  ///< number of new created tmcd track based on metadata (aka not data copy)
146     int     chapter_track; ///< qt chapter track number
147     int64_t mdat_pos;
148     uint64_t mdat_size;
149     MOVTrack *tracks;
150
151     int flags;
152     int rtp_flags;
153     int reserved_moov_size;
154     int64_t reserved_moov_pos;
155
156     int iods_skip;
157     int iods_video_profile;
158     int iods_audio_profile;
159
160     int fragments;
161     int max_fragment_duration;
162     int min_fragment_duration;
163     int max_fragment_size;
164     int ism_lookahead;
165     AVIOContext *mdat_buf;
166 } MOVMuxContext;
167
168 #define FF_MOV_FLAG_RTP_HINT 1
169 #define FF_MOV_FLAG_FRAGMENT 2
170 #define FF_MOV_FLAG_EMPTY_MOOV 4
171 #define FF_MOV_FLAG_FRAG_KEYFRAME 8
172 #define FF_MOV_FLAG_SEPARATE_MOOF 16
173 #define FF_MOV_FLAG_FRAG_CUSTOM 32
174 #define FF_MOV_FLAG_ISML 64
175
176 int ff_mov_write_packet(AVFormatContext *s, AVPacket *pkt);
177
178 int ff_mov_init_hinting(AVFormatContext *s, int index, int src_index);
179 int ff_mov_add_hinted_packet(AVFormatContext *s, AVPacket *pkt,
180                              int track_index, int sample,
181                              uint8_t *sample_data, int sample_size);
182 void ff_mov_close_hinting(MOVTrack *track);
183
184 #endif /* AVFORMAT_MOVENC_H */