avformat/mlvdec: read_string() received unsigned size, make the argument unsigned
[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 <stddef.h>
28 #include <stdint.h>
29
30 #include "libavutil/encryption_info.h"
31 #include "libavutil/mastering_display_metadata.h"
32 #include "libavutil/spherical.h"
33 #include "libavutil/stereo3d.h"
34
35 #include "avio.h"
36 #include "internal.h"
37 #include "dv.h"
38
39 /* isom.c */
40 extern const AVCodecTag ff_mp4_obj_type[];
41 extern const AVCodecTag ff_codec_movvideo_tags[];
42 extern const AVCodecTag ff_codec_movaudio_tags[];
43 extern const AVCodecTag ff_codec_movsubtitle_tags[];
44
45 int ff_mov_iso639_to_lang(const char lang[4], int mp4);
46 int ff_mov_lang_to_iso639(unsigned code, char to[4]);
47
48 struct AVAESCTR;
49
50 /* the QuickTime file format is quite convoluted...
51  * it has lots of index tables, each indexing something in another one...
52  * Here we just use what is needed to read the chunks
53  */
54
55 typedef struct MOVStts {
56     unsigned int count;
57     int duration;
58 } MOVStts;
59
60 typedef struct MOVStsc {
61     int first;
62     int count;
63     int id;
64 } MOVStsc;
65
66 typedef struct MOVElst {
67     int64_t duration;
68     int64_t time;
69     float rate;
70 } MOVElst;
71
72 typedef struct MOVDref {
73     uint32_t type;
74     char *path;
75     char *dir;
76     char volume[28];
77     char filename[64];
78     int16_t nlvl_to, nlvl_from;
79 } MOVDref;
80
81 typedef struct MOVAtom {
82     uint32_t type;
83     int64_t size; /* total size (excluding the size and type fields) */
84 } MOVAtom;
85
86 struct MOVParseTableEntry;
87
88 typedef struct MOVFragment {
89     unsigned track_id;
90     uint64_t base_data_offset;
91     uint64_t moof_offset;
92     uint64_t implicit_offset;
93     unsigned stsd_id;
94     unsigned duration;
95     unsigned size;
96     unsigned flags;
97 } MOVFragment;
98
99 typedef struct MOVTrackExt {
100     unsigned track_id;
101     unsigned stsd_id;
102     unsigned duration;
103     unsigned size;
104     unsigned flags;
105 } MOVTrackExt;
106
107 typedef struct MOVSbgp {
108     unsigned int count;
109     unsigned int index;
110 } MOVSbgp;
111
112 typedef struct MOVEncryptionIndex {
113     // Individual encrypted samples.  If there are no elements, then the default
114     // settings will be used.
115     unsigned int nb_encrypted_samples;
116     AVEncryptionInfo **encrypted_samples;
117
118     uint8_t* auxiliary_info_sizes;
119     size_t auxiliary_info_sample_count;
120     uint8_t auxiliary_info_default_size;
121     uint64_t* auxiliary_offsets;  ///< Absolute seek position
122     size_t auxiliary_offsets_count;
123 } MOVEncryptionIndex;
124
125 typedef struct MOVFragmentStreamInfo {
126     int id;
127     int64_t sidx_pts;
128     int64_t first_tfra_pts;
129     int64_t tfdt_dts;
130     int index_entry;
131     MOVEncryptionIndex *encryption_index;
132 } MOVFragmentStreamInfo;
133
134 typedef struct MOVFragmentIndexItem {
135     int64_t moof_offset;
136     int headers_read;
137     int current;
138     int nb_stream_info;
139     MOVFragmentStreamInfo * stream_info;
140 } MOVFragmentIndexItem;
141
142 typedef struct MOVFragmentIndex {
143     int allocated_size;
144     int complete;
145     int current;
146     int nb_items;
147     MOVFragmentIndexItem * item;
148 } MOVFragmentIndex;
149
150 typedef struct MOVIndexRange {
151     int64_t start;
152     int64_t end;
153 } MOVIndexRange;
154
155 typedef struct MOVStreamContext {
156     AVIOContext *pb;
157     int pb_is_copied;
158     int ffindex;          ///< AVStream index
159     int next_chunk;
160     unsigned int chunk_count;
161     int64_t *chunk_offsets;
162     unsigned int stts_count;
163     MOVStts *stts_data;
164     unsigned int ctts_count;
165     unsigned int ctts_allocated_size;
166     MOVStts *ctts_data;
167     unsigned int stsc_count;
168     MOVStsc *stsc_data;
169     unsigned int stsc_index;
170     int stsc_sample;
171     unsigned int stps_count;
172     unsigned *stps_data;  ///< partial sync sample for mpeg-2 open gop
173     MOVElst *elst_data;
174     unsigned int elst_count;
175     int ctts_index;
176     int ctts_sample;
177     unsigned int sample_size; ///< may contain value calculated from stsd or value from stsz atom
178     unsigned int stsz_sample_size; ///< always contains sample size from stsz atom
179     unsigned int sample_count;
180     int *sample_sizes;
181     int keyframe_absent;
182     unsigned int keyframe_count;
183     int *keyframes;
184     int time_scale;
185     int64_t time_offset;  ///< time offset of the edit list entries
186     int64_t min_corrected_pts;  ///< minimum Composition time shown by the edits excluding empty edits.
187     int current_sample;
188     int64_t current_index;
189     MOVIndexRange* index_ranges;
190     MOVIndexRange* current_index_range;
191     unsigned int bytes_per_frame;
192     unsigned int samples_per_frame;
193     int dv_audio_container;
194     int pseudo_stream_id; ///< -1 means demux all ids
195     int16_t audio_cid;    ///< stsd audio compression id
196     unsigned drefs_count;
197     MOVDref *drefs;
198     int dref_id;
199     int timecode_track;
200     int width;            ///< tkhd width
201     int height;           ///< tkhd height
202     int dts_shift;        ///< dts shift when ctts is negative
203     uint32_t palette[256];
204     int has_palette;
205     int64_t data_size;
206     uint32_t tmcd_flags;  ///< tmcd track flags
207     int64_t track_end;    ///< used for dts generation in fragmented movie files
208     int start_pad;        ///< amount of samples to skip due to enc-dec delay
209     unsigned int rap_group_count;
210     MOVSbgp *rap_group;
211
212     int nb_frames_for_fps;
213     int64_t duration_for_fps;
214
215     /** extradata array (and size) for multiple stsd */
216     uint8_t **extradata;
217     int *extradata_size;
218     int last_stsd_index;
219     int stsd_count;
220
221     int32_t *display_matrix;
222     AVStereo3D *stereo3d;
223     AVSphericalMapping *spherical;
224     size_t spherical_size;
225     AVMasteringDisplayMetadata *mastering;
226     AVContentLightMetadata *coll;
227     size_t coll_size;
228
229     uint32_t format;
230
231     int has_sidx;  // If there is an sidx entry for this stream.
232     struct {
233         struct AVAESCTR* aes_ctr;
234         unsigned int per_sample_iv_size;  // Either 0, 8, or 16.
235         AVEncryptionInfo *default_encrypted_sample;
236         MOVEncryptionIndex *encryption_index;
237     } cenc;
238 } MOVStreamContext;
239
240 typedef struct MOVContext {
241     const AVClass *class; ///< class for private options
242     AVFormatContext *fc;
243     int time_scale;
244     int64_t duration;     ///< duration of the longest track
245     int found_moov;       ///< 'moov' atom has been found
246     int found_mdat;       ///< 'mdat' atom has been found
247     int found_hdlr_mdta;  ///< 'hdlr' atom with type 'mdta' has been found
248     int trak_index;       ///< Index of the current 'trak'
249     char **meta_keys;
250     unsigned meta_keys_count;
251     DVDemuxContext *dv_demux;
252     AVFormatContext *dv_fctx;
253     int isom;             ///< 1 if file is ISO Media (mp4/3gp)
254     MOVFragment fragment; ///< current fragment in moof atom
255     MOVTrackExt *trex_data;
256     unsigned trex_count;
257     int itunes_metadata;  ///< metadata are itunes style
258     int handbrake_version;
259     int *chapter_tracks;
260     unsigned int nb_chapter_tracks;
261     int use_absolute_path;
262     int ignore_editlist;
263     int advanced_editlist;
264     int ignore_chapters;
265     int seek_individually;
266     int64_t next_root_atom; ///< offset of the next root atom
267     int export_all;
268     int export_xmp;
269     int *bitrates;          ///< bitrates read before streams creation
270     int bitrates_count;
271     int moov_retry;
272     int use_mfra_for;
273     int has_looked_for_mfra;
274     MOVFragmentIndex frag_index;
275     int atom_depth;
276     unsigned int aax_mode;  ///< 'aax' file has been detected
277     uint8_t file_key[20];
278     uint8_t file_iv[20];
279     void *activation_bytes;
280     int activation_bytes_size;
281     void *audible_fixed_key;
282     int audible_fixed_key_size;
283     struct AVAES *aes_decrypt;
284     uint8_t *decryption_key;
285     int decryption_key_len;
286     int enable_drefs;
287     int32_t movie_display_matrix[3][3]; ///< display matrix from mvhd
288 } MOVContext;
289
290 int ff_mp4_read_descr_len(AVIOContext *pb);
291 int ff_mp4_read_descr(AVFormatContext *fc, AVIOContext *pb, int *tag);
292 int ff_mp4_read_dec_config_descr(AVFormatContext *fc, AVStream *st, AVIOContext *pb);
293 void ff_mp4_parse_es_descr(AVIOContext *pb, int *es_id);
294
295 #define MP4ODescrTag                    0x01
296 #define MP4IODescrTag                   0x02
297 #define MP4ESDescrTag                   0x03
298 #define MP4DecConfigDescrTag            0x04
299 #define MP4DecSpecificDescrTag          0x05
300 #define MP4SLDescrTag                   0x06
301
302 #define MOV_TFHD_BASE_DATA_OFFSET       0x01
303 #define MOV_TFHD_STSD_ID                0x02
304 #define MOV_TFHD_DEFAULT_DURATION       0x08
305 #define MOV_TFHD_DEFAULT_SIZE           0x10
306 #define MOV_TFHD_DEFAULT_FLAGS          0x20
307 #define MOV_TFHD_DURATION_IS_EMPTY  0x010000
308 #define MOV_TFHD_DEFAULT_BASE_IS_MOOF 0x020000
309
310 #define MOV_TRUN_DATA_OFFSET            0x01
311 #define MOV_TRUN_FIRST_SAMPLE_FLAGS     0x04
312 #define MOV_TRUN_SAMPLE_DURATION       0x100
313 #define MOV_TRUN_SAMPLE_SIZE           0x200
314 #define MOV_TRUN_SAMPLE_FLAGS          0x400
315 #define MOV_TRUN_SAMPLE_CTS            0x800
316
317 #define MOV_FRAG_SAMPLE_FLAG_DEGRADATION_PRIORITY_MASK 0x0000ffff
318 #define MOV_FRAG_SAMPLE_FLAG_IS_NON_SYNC               0x00010000
319 #define MOV_FRAG_SAMPLE_FLAG_PADDING_MASK              0x000e0000
320 #define MOV_FRAG_SAMPLE_FLAG_REDUNDANCY_MASK           0x00300000
321 #define MOV_FRAG_SAMPLE_FLAG_DEPENDED_MASK             0x00c00000
322 #define MOV_FRAG_SAMPLE_FLAG_DEPENDS_MASK              0x03000000
323
324 #define MOV_FRAG_SAMPLE_FLAG_DEPENDS_NO                0x02000000
325 #define MOV_FRAG_SAMPLE_FLAG_DEPENDS_YES               0x01000000
326
327 #define MOV_TKHD_FLAG_ENABLED       0x0001
328 #define MOV_TKHD_FLAG_IN_MOVIE      0x0002
329 #define MOV_TKHD_FLAG_IN_PREVIEW    0x0004
330 #define MOV_TKHD_FLAG_IN_POSTER     0x0008
331
332 #define MOV_SAMPLE_DEPENDENCY_UNKNOWN 0x0
333 #define MOV_SAMPLE_DEPENDENCY_YES     0x1
334 #define MOV_SAMPLE_DEPENDENCY_NO      0x2
335
336
337 #define TAG_IS_AVCI(tag)                    \
338     ((tag) == MKTAG('a', 'i', '5', 'p') ||  \
339      (tag) == MKTAG('a', 'i', '5', 'q') ||  \
340      (tag) == MKTAG('a', 'i', '5', '2') ||  \
341      (tag) == MKTAG('a', 'i', '5', '3') ||  \
342      (tag) == MKTAG('a', 'i', '5', '5') ||  \
343      (tag) == MKTAG('a', 'i', '5', '6') ||  \
344      (tag) == MKTAG('a', 'i', '1', 'p') ||  \
345      (tag) == MKTAG('a', 'i', '1', 'q') ||  \
346      (tag) == MKTAG('a', 'i', '1', '2') ||  \
347      (tag) == MKTAG('a', 'i', '1', '3') ||  \
348      (tag) == MKTAG('a', 'i', '1', '5') ||  \
349      (tag) == MKTAG('a', 'i', '1', '6') ||  \
350      (tag) == MKTAG('a', 'i', 'v', 'x') ||  \
351      (tag) == MKTAG('A', 'V', 'i', 'n'))
352
353
354 int ff_mov_read_esds(AVFormatContext *fc, AVIOContext *pb);
355
356 int ff_mov_read_stsd_entries(MOVContext *c, AVIOContext *pb, int entries);
357 void ff_mov_write_chan(AVIOContext *pb, int64_t channel_layout);
358
359 #define FF_MOV_FLAG_MFRA_AUTO -1
360 #define FF_MOV_FLAG_MFRA_DTS 1
361 #define FF_MOV_FLAG_MFRA_PTS 2
362
363 /**
364  * Compute codec id for 'lpcm' tag.
365  * See CoreAudioTypes and AudioStreamBasicDescription at Apple.
366  */
367 static inline enum AVCodecID ff_mov_get_lpcm_codec_id(int bps, int flags)
368 {
369     /* lpcm flags:
370      * 0x1 = float
371      * 0x2 = big-endian
372      * 0x4 = signed
373      */
374     return ff_get_pcm_codec_id(bps, flags & 1, flags & 2, flags & 4 ? -1 : 0);
375 }
376
377 #endif /* AVFORMAT_ISOM_H */