avformat/mov: support cenc (common encryption)
[ffmpeg.git] / libavformat / isom.h
1 /*
2  * ISO Media common code
3  * copyright (c) 2001 Fabrice Bellard
4  * copyright (c) 2002 Francois Revol <revol@free.fr>
5  * copyright (c) 2006 Baptiste Coudurier <baptiste.coudurier@free.fr>
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_ISOM_H
25 #define AVFORMAT_ISOM_H
26
27 #include "avio.h"
28 #include "internal.h"
29 #include "dv.h"
30
31 /* isom.c */
32 extern const AVCodecTag ff_mp4_obj_type[];
33 extern const AVCodecTag ff_codec_movvideo_tags[];
34 extern const AVCodecTag ff_codec_movaudio_tags[];
35 extern const AVCodecTag ff_codec_movsubtitle_tags[];
36
37 int ff_mov_iso639_to_lang(const char lang[4], int mp4);
38 int ff_mov_lang_to_iso639(unsigned code, char to[4]);
39
40 struct AVAESCTR;
41
42 /* the QuickTime file format is quite convoluted...
43  * it has lots of index tables, each indexing something in another one...
44  * Here we just use what is needed to read the chunks
45  */
46
47 typedef struct MOVStts {
48     int count;
49     int duration;
50 } MOVStts;
51
52 typedef struct MOVStsc {
53     int first;
54     int count;
55     int id;
56 } MOVStsc;
57
58 typedef struct MOVElst {
59     int64_t duration;
60     int64_t time;
61     float rate;
62 } MOVElst;
63
64 typedef struct MOVDref {
65     uint32_t type;
66     char *path;
67     char *dir;
68     char volume[28];
69     char filename[64];
70     int16_t nlvl_to, nlvl_from;
71 } MOVDref;
72
73 typedef struct MOVAtom {
74     uint32_t type;
75     int64_t size; /* total size (excluding the size and type fields) */
76 } MOVAtom;
77
78 struct MOVParseTableEntry;
79
80 typedef struct MOVFragment {
81     unsigned track_id;
82     uint64_t base_data_offset;
83     uint64_t moof_offset;
84     uint64_t implicit_offset;
85     unsigned stsd_id;
86     unsigned duration;
87     unsigned size;
88     unsigned flags;
89     int64_t time;
90 } MOVFragment;
91
92 typedef struct MOVTrackExt {
93     unsigned track_id;
94     unsigned stsd_id;
95     unsigned duration;
96     unsigned size;
97     unsigned flags;
98 } MOVTrackExt;
99
100 typedef struct MOVSbgp {
101     unsigned int count;
102     unsigned int index;
103 } MOVSbgp;
104
105 typedef struct MOVFragmentIndexItem {
106     int64_t moof_offset;
107     int64_t time;
108     int headers_read;
109 } MOVFragmentIndexItem;
110
111 typedef struct MOVFragmentIndex {
112     unsigned track_id;
113     unsigned item_count;
114     unsigned current_item;
115     MOVFragmentIndexItem *items;
116 } MOVFragmentIndex;
117
118 typedef struct MOVStreamContext {
119     AVIOContext *pb;
120     int pb_is_copied;
121     int ffindex;          ///< AVStream index
122     int next_chunk;
123     unsigned int chunk_count;
124     int64_t *chunk_offsets;
125     unsigned int stts_count;
126     MOVStts *stts_data;
127     unsigned int ctts_count;
128     MOVStts *ctts_data;
129     unsigned int stsc_count;
130     MOVStsc *stsc_data;
131     unsigned int stps_count;
132     unsigned *stps_data;  ///< partial sync sample for mpeg-2 open gop
133     MOVElst *elst_data;
134     unsigned int elst_count;
135     int ctts_index;
136     int ctts_sample;
137     unsigned int sample_size; ///< may contain value calculated from stsd or value from stsz atom
138     unsigned int stsz_sample_size; ///< always contains sample size from stsz atom
139     unsigned int sample_count;
140     int *sample_sizes;
141     int keyframe_absent;
142     unsigned int keyframe_count;
143     int *keyframes;
144     int time_scale;
145     int64_t time_offset;  ///< time offset of the edit list entries
146     int current_sample;
147     unsigned int bytes_per_frame;
148     unsigned int samples_per_frame;
149     int dv_audio_container;
150     int pseudo_stream_id; ///< -1 means demux all ids
151     int16_t audio_cid;    ///< stsd audio compression id
152     unsigned drefs_count;
153     MOVDref *drefs;
154     int dref_id;
155     int timecode_track;
156     int wrong_dts;        ///< dts are wrong due to huge ctts offset (iMovie files)
157     int width;            ///< tkhd width
158     int height;           ///< tkhd height
159     int dts_shift;        ///< dts shift when ctts is negative
160     uint32_t palette[256];
161     int has_palette;
162     int64_t data_size;
163     uint32_t tmcd_flags;  ///< tmcd track flags
164     int64_t track_end;    ///< used for dts generation in fragmented movie files
165     int start_pad;        ///< amount of samples to skip due to enc-dec delay
166     unsigned int rap_group_count;
167     MOVSbgp *rap_group;
168
169     int nb_frames_for_fps;
170     int64_t duration_for_fps;
171
172     int32_t *display_matrix;
173     uint32_t format;
174
175     struct {
176         int use_subsamples;
177         uint8_t* auxiliary_info;
178         uint8_t* auxiliary_info_end;
179         uint8_t* auxiliary_info_pos;
180         struct AVAESCTR* aes_ctr;
181     } cenc;
182 } MOVStreamContext;
183
184 typedef struct MOVContext {
185     const AVClass *class; ///< class for private options
186     AVFormatContext *fc;
187     int time_scale;
188     int64_t duration;     ///< duration of the longest track
189     int found_moov;       ///< 'moov' atom has been found
190     int found_mdat;       ///< 'mdat' atom has been found
191     int found_hdlr_mdta;  ///< 'hdlr' atom with type 'mdta' has been found
192     int trak_index;       ///< Index of the current 'trak'
193     char **meta_keys;
194     unsigned meta_keys_count;
195     DVDemuxContext *dv_demux;
196     AVFormatContext *dv_fctx;
197     int isom;             ///< 1 if file is ISO Media (mp4/3gp)
198     MOVFragment fragment; ///< current fragment in moof atom
199     MOVTrackExt *trex_data;
200     unsigned trex_count;
201     int itunes_metadata;  ///< metadata are itunes style
202     int handbrake_version;
203     int chapter_track;
204     int use_absolute_path;
205     int ignore_editlist;
206     int ignore_chapters;
207     int seek_individually;
208     int64_t next_root_atom; ///< offset of the next root atom
209     int export_all;
210     int export_xmp;
211     int *bitrates;          ///< bitrates read before streams creation
212     int bitrates_count;
213     int moov_retry;
214     int use_mfra_for;
215     int has_looked_for_mfra;
216     MOVFragmentIndex** fragment_index_data;
217     unsigned fragment_index_count;
218     int fragment_index_complete;
219     int atom_depth;
220     unsigned int aax_mode;  ///< 'aax' file has been detected
221     uint8_t file_key[20];
222     uint8_t file_iv[20];
223     void *activation_bytes;
224     int activation_bytes_size;
225     void *audible_fixed_key;
226     int audible_fixed_key_size;
227     struct AVAES *aes_decrypt;
228     uint8_t *decryption_key;
229     int decryption_key_len;
230 } MOVContext;
231
232 int ff_mp4_read_descr_len(AVIOContext *pb);
233 int ff_mp4_read_descr(AVFormatContext *fc, AVIOContext *pb, int *tag);
234 int ff_mp4_read_dec_config_descr(AVFormatContext *fc, AVStream *st, AVIOContext *pb);
235 void ff_mp4_parse_es_descr(AVIOContext *pb, int *es_id);
236
237 #define MP4ODescrTag                    0x01
238 #define MP4IODescrTag                   0x02
239 #define MP4ESDescrTag                   0x03
240 #define MP4DecConfigDescrTag            0x04
241 #define MP4DecSpecificDescrTag          0x05
242 #define MP4SLDescrTag                   0x06
243
244 #define MOV_TFHD_BASE_DATA_OFFSET       0x01
245 #define MOV_TFHD_STSD_ID                0x02
246 #define MOV_TFHD_DEFAULT_DURATION       0x08
247 #define MOV_TFHD_DEFAULT_SIZE           0x10
248 #define MOV_TFHD_DEFAULT_FLAGS          0x20
249 #define MOV_TFHD_DURATION_IS_EMPTY  0x010000
250 #define MOV_TFHD_DEFAULT_BASE_IS_MOOF 0x020000
251
252 #define MOV_TRUN_DATA_OFFSET            0x01
253 #define MOV_TRUN_FIRST_SAMPLE_FLAGS     0x04
254 #define MOV_TRUN_SAMPLE_DURATION       0x100
255 #define MOV_TRUN_SAMPLE_SIZE           0x200
256 #define MOV_TRUN_SAMPLE_FLAGS          0x400
257 #define MOV_TRUN_SAMPLE_CTS            0x800
258
259 #define MOV_FRAG_SAMPLE_FLAG_DEGRADATION_PRIORITY_MASK 0x0000ffff
260 #define MOV_FRAG_SAMPLE_FLAG_IS_NON_SYNC               0x00010000
261 #define MOV_FRAG_SAMPLE_FLAG_PADDING_MASK              0x000e0000
262 #define MOV_FRAG_SAMPLE_FLAG_REDUNDANCY_MASK           0x00300000
263 #define MOV_FRAG_SAMPLE_FLAG_DEPENDED_MASK             0x00c00000
264 #define MOV_FRAG_SAMPLE_FLAG_DEPENDS_MASK              0x03000000
265
266 #define MOV_FRAG_SAMPLE_FLAG_DEPENDS_NO                0x02000000
267 #define MOV_FRAG_SAMPLE_FLAG_DEPENDS_YES               0x01000000
268
269 #define MOV_TKHD_FLAG_ENABLED       0x0001
270 #define MOV_TKHD_FLAG_IN_MOVIE      0x0002
271 #define MOV_TKHD_FLAG_IN_PREVIEW    0x0004
272 #define MOV_TKHD_FLAG_IN_POSTER     0x0008
273
274 #define TAG_IS_AVCI(tag)                    \
275     ((tag) == MKTAG('a', 'i', '5', 'p') ||  \
276      (tag) == MKTAG('a', 'i', '5', 'q') ||  \
277      (tag) == MKTAG('a', 'i', '5', '2') ||  \
278      (tag) == MKTAG('a', 'i', '5', '3') ||  \
279      (tag) == MKTAG('a', 'i', '5', '5') ||  \
280      (tag) == MKTAG('a', 'i', '5', '6') ||  \
281      (tag) == MKTAG('a', 'i', '1', 'p') ||  \
282      (tag) == MKTAG('a', 'i', '1', 'q') ||  \
283      (tag) == MKTAG('a', 'i', '1', '2') ||  \
284      (tag) == MKTAG('a', 'i', '1', '3') ||  \
285      (tag) == MKTAG('a', 'i', '1', '5') ||  \
286      (tag) == MKTAG('a', 'i', '1', '6') ||  \
287      (tag) == MKTAG('a', 'i', 'v', 'x') ||  \
288      (tag) == MKTAG('A', 'V', 'i', 'n'))
289
290
291 int ff_mov_read_esds(AVFormatContext *fc, AVIOContext *pb);
292 enum AVCodecID ff_mov_get_lpcm_codec_id(int bps, int flags);
293
294 int ff_mov_read_stsd_entries(MOVContext *c, AVIOContext *pb, int entries);
295 void ff_mov_write_chan(AVIOContext *pb, int64_t channel_layout);
296
297 #define FF_MOV_FLAG_MFRA_AUTO -1
298 #define FF_MOV_FLAG_MFRA_DTS 1
299 #define FF_MOV_FLAG_MFRA_PTS 2
300
301 #endif /* AVFORMAT_ISOM_H */