]> git.sesse.net Git - ffmpeg/blob - libavformat/mov.c
Segmentation fault fix when no video device is specified.
[ffmpeg] / libavformat / mov.c
1 /*
2  * MOV demuxer
3  * Copyright (c) 2001 Fabrice Bellard.
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21
22 #include <limits.h>
23
24 //#define DEBUG
25
26 #include "avformat.h"
27 #include "riff.h"
28 #include "isom.h"
29 #include "dv.h"
30
31 #ifdef CONFIG_ZLIB
32 #include <zlib.h>
33 #endif
34
35 /*
36  * First version by Francois Revol revol@free.fr
37  * Seek function by Gael Chardon gael.dev@4now.net
38  *
39  * Features and limitations:
40  * - reads most of the QT files I have (at least the structure),
41  *   the exceptions are .mov with zlib compressed headers ('cmov' section). It shouldn't be hard to implement.
42  *   FIXED, Francois Revol, 07/17/2002
43  * - ffmpeg has nearly none of the usual QuickTime codecs,
44  *   although I succesfully dumped raw and mp3 audio tracks off .mov files.
45  *   Sample QuickTime files with mp3 audio can be found at: http://www.3ivx.com/showcase.html
46  * - .mp4 parsing is still hazardous, although the format really is QuickTime with some minor changes
47  *   (to make .mov parser crash maybe ?), despite what they say in the MPEG FAQ at
48  *   http://mpeg.telecomitalialab.com/faq.htm
49  * - the code is quite ugly... maybe I won't do it recursive next time :-)
50  * - seek is not supported with files that contain edit list
51  *
52  * Funny I didn't know about http://sourceforge.net/projects/qt-ffmpeg/
53  * when coding this :) (it's a writer anyway)
54  *
55  * Reference documents:
56  * http://www.geocities.com/xhelmboyx/quicktime/formats/qtm-layout.txt
57  * Apple:
58  *  http://developer.apple.com/documentation/QuickTime/QTFF/
59  *  http://developer.apple.com/documentation/QuickTime/QTFF/qtff.pdf
60  * QuickTime is a trademark of Apple (AFAIK :))
61  */
62
63 #include "qtpalette.h"
64
65
66 #undef NDEBUG
67 #include <assert.h>
68
69 static const CodecTag mov_video_tags[] = {
70 /*  { CODEC_ID_, MKTAG('c', 'v', 'i', 'd') }, *//* Cinepak */
71 /*  { CODEC_ID_H263, MKTAG('r', 'a', 'w', ' ') }, *//* Uncompressed RGB */
72 /*  { CODEC_ID_H263, MKTAG('Y', 'u', 'v', '2') }, *//* Uncompressed YUV422 */
73 /*    { CODEC_ID_RAWVIDEO, MKTAG('A', 'V', 'U', 'I') }, *//* YUV with alpha-channel (AVID Uncompressed) */
74 /* Graphics */
75 /* Animation */
76 /* Apple video */
77 /* Kodak Photo CD */
78     { CODEC_ID_MJPEG, MKTAG('j', 'p', 'e', 'g') }, /* PhotoJPEG */
79     { CODEC_ID_MPEG1VIDEO, MKTAG('m', 'p', 'e', 'g') }, /* MPEG */
80     { CODEC_ID_MJPEG, MKTAG('m', 'j', 'p', 'a') }, /* Motion-JPEG (format A) */
81     { CODEC_ID_MJPEGB, MKTAG('m', 'j', 'p', 'b') }, /* Motion-JPEG (format B) */
82     { CODEC_ID_MJPEG, MKTAG('A', 'V', 'D', 'J') }, /* MJPEG with alpha-channel (AVID JFIF meridien compressed) */
83 /*    { CODEC_ID_MJPEG, MKTAG('A', 'V', 'R', 'n') }, *//* MJPEG with alpha-channel (AVID ABVB/Truevision NuVista) */
84     { CODEC_ID_GIF, MKTAG('g', 'i', 'f', ' ') }, /* embedded gif files as frames (usually one "click to play movie" frame) */
85 /* Sorenson video */
86     { CODEC_ID_SVQ1, MKTAG('S', 'V', 'Q', '1') }, /* Sorenson Video v1 */
87     { CODEC_ID_SVQ1, MKTAG('s', 'v', 'q', '1') }, /* Sorenson Video v1 */
88     { CODEC_ID_SVQ1, MKTAG('s', 'v', 'q', 'i') }, /* Sorenson Video v1 (from QT specs)*/
89     { CODEC_ID_SVQ3, MKTAG('S', 'V', 'Q', '3') }, /* Sorenson Video v3 */
90     { CODEC_ID_MPEG4, MKTAG('m', 'p', '4', 'v') },
91     { CODEC_ID_MPEG4, MKTAG('D', 'I', 'V', 'X') }, /* OpenDiVX *//* sample files at http://heroinewarrior.com/xmovie.php3 use this tag */
92     { CODEC_ID_MPEG4, MKTAG('X', 'V', 'I', 'D') },
93     { CODEC_ID_MPEG4, MKTAG('3', 'I', 'V', '2') }, /* experimental: 3IVX files before ivx D4 4.5.1 */
94 /*    { CODEC_ID_, MKTAG('I', 'V', '5', '0') }, *//* Indeo 5.0 */
95     { CODEC_ID_H263, MKTAG('h', '2', '6', '3') }, /* H263 */
96     { CODEC_ID_H263, MKTAG('s', '2', '6', '3') }, /* H263 ?? works */
97     { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 'c', ' ') }, /* DV NTSC */
98     { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 'c', 'p') }, /* DV PAL */
99     { CODEC_ID_VP3, MKTAG('V', 'P', '3', '1') }, /* On2 VP3 */
100     { CODEC_ID_RPZA, MKTAG('r', 'p', 'z', 'a') }, /* Apple Video (RPZA) */
101     { CODEC_ID_CINEPAK, MKTAG('c', 'v', 'i', 'd') }, /* Cinepak */
102     { CODEC_ID_8BPS, MKTAG('8', 'B', 'P', 'S') }, /* Planar RGB (8BPS) */
103     { CODEC_ID_SMC, MKTAG('s', 'm', 'c', ' ') }, /* Apple Graphics (SMC) */
104     { CODEC_ID_QTRLE, MKTAG('r', 'l', 'e', ' ') }, /* Apple Animation (RLE) */
105     { CODEC_ID_QDRAW, MKTAG('q', 'd', 'r', 'w') }, /* QuickDraw */
106     { CODEC_ID_H264, MKTAG('a', 'v', 'c', '1') }, /* AVC-1/H.264 */
107     { CODEC_ID_MPEG2VIDEO, MKTAG('h', 'd', 'v', '2') }, /* MPEG2 produced by Sony HD camera */
108     { CODEC_ID_MPEG2VIDEO, MKTAG('h', 'd', 'v', '3') }, /* HDV produced by FCP */
109     { CODEC_ID_MPEG2VIDEO, MKTAG('m', 'x', '5', 'n') }, /* MPEG2 IMX NTSC 525/60 50mb/s produced by FCP */
110     { CODEC_ID_MPEG2VIDEO, MKTAG('m', 'x', '5', 'p') }, /* MPEG2 IMX PAL 625/50 50mb/s produced by FCP */
111     { CODEC_ID_MPEG2VIDEO, MKTAG('m', 'x', '3', 'n') }, /* MPEG2 IMX NTSC 525/60 30mb/s produced by FCP */
112     { CODEC_ID_MPEG2VIDEO, MKTAG('m', 'x', '3', 'p') }, /* MPEG2 IMX PAL 625/50 30mb/s produced by FCP */
113     { CODEC_ID_DVVIDEO, MKTAG('d', 'v', 'p', 'p') }, /* DVCPRO PAL produced by FCP */
114     //{ CODEC_ID_DVVIDEO, MKTAG('d', 'v', 'h', '5') }, /* DVCPRO HD 50i produced by FCP */
115     //{ CODEC_ID_DVVIDEO, MKTAG('d', 'v', 'h', '6') }, /* DVCPRO HD 60i produced by FCP */
116     { CODEC_ID_DVVIDEO, MKTAG('d', 'v', '5', 'p') }, /* DVCPRO50 PAL produced by FCP */
117     { CODEC_ID_DVVIDEO, MKTAG('d', 'v', '5', 'n') }, /* DVCPRO50 NTSC produced by FCP */
118     { CODEC_ID_DVVIDEO, MKTAG('A', 'V', 'd', 'v') }, /* AVID DV */
119     //{ CODEC_ID_JPEG2000, MKTAG('m', 'j', 'p', '2') }, /* JPEG 2000 produced by FCP */
120     { CODEC_ID_TARGA, MKTAG('t', 'g', 'a', ' ') }, /* Truevision Targa */
121     { CODEC_ID_TIFF, MKTAG('t', 'i', 'f', 'f') }, /* TIFF embedded in MOV */
122     { CODEC_ID_RAWVIDEO, MKTAG('2', 'v', 'u', 'y') }, /* UNCOMPRESSED 8BIT 4:2:2 */
123     { CODEC_ID_VC1,  MKTAG('v', 'c', '-', '1') }, /* SMPTE RP 2025 */
124     { CODEC_ID_WMV3, MKTAG('v', 'c', '-', '1') }, /* SMPTE RP 2025 */
125     { CODEC_ID_NONE, 0 },
126 };
127
128 static const CodecTag mov_audio_tags[] = {
129     { CODEC_ID_PCM_S32BE, MKTAG('i', 'n', '3', '2') },
130     { CODEC_ID_PCM_S24BE, MKTAG('i', 'n', '2', '4') },
131     { CODEC_ID_PCM_S16BE, MKTAG('N', 'O', 'N', 'E') }, /* uncompressed */
132     { CODEC_ID_PCM_S16BE, MKTAG('t', 'w', 'o', 's') }, /* 16 bits */
133     { CODEC_ID_PCM_U8, MKTAG('r', 'a', 'w', ' ') }, /* 8 bits unsigned */
134     { CODEC_ID_PCM_S16LE, MKTAG('s', 'o', 'w', 't') }, /*  */
135     { CODEC_ID_PCM_MULAW, MKTAG('u', 'l', 'a', 'w') }, /*  */
136     { CODEC_ID_PCM_ALAW, MKTAG('a', 'l', 'a', 'w') }, /*  */
137     { CODEC_ID_ADPCM_IMA_QT, MKTAG('i', 'm', 'a', '4') }, /* IMA-4 ADPCM */
138     { CODEC_ID_ADPCM_MS, MKTAG('m', 's', 0x00, 0x02) }, /* MS ADPCM */
139     { CODEC_ID_MACE3, MKTAG('M', 'A', 'C', '3') }, /* Macintosh Audio Compression and Expansion 3:1 */
140     { CODEC_ID_MACE6, MKTAG('M', 'A', 'C', '6') }, /* Macintosh Audio Compression and Expansion 6:1 */
141
142     { CODEC_ID_MP3, MKTAG('.', 'm', 'p', '3') }, /* MPEG layer 3 */ /* sample files at http://www.3ivx.com/showcase.html use this tag */
143     { CODEC_ID_MP2, 0x6D730055 }, /* MPEG layer 3 */
144     { CODEC_ID_MP2, 0x5500736D }, /* MPEG layer 3 *//* XXX: check endianness */
145 /*    { CODEC_ID_OGG_VORBIS, MKTAG('O', 'g', 'g', 'S') }, *//* sample files at http://heroinewarrior.com/xmovie.php3 use this tag */
146 /* MP4 tags */
147     { CODEC_ID_AAC, MKTAG('m', 'p', '4', 'a') }, /* MPEG-4 AAC */
148     /* The standard for mpeg4 audio is still not normalised AFAIK anyway */
149     { CODEC_ID_AMR_NB, MKTAG('s', 'a', 'm', 'r') }, /* AMR-NB 3gp */
150     { CODEC_ID_AMR_WB, MKTAG('s', 'a', 'w', 'b') }, /* AMR-WB 3gp */
151     { CODEC_ID_AC3, MKTAG('m', 's', 0x20, 0x00) }, /* Dolby AC-3 */
152     { CODEC_ID_ALAC,MKTAG('a', 'l', 'a', 'c') }, /* Apple Lossless */
153     { CODEC_ID_QDM2,MKTAG('Q', 'D', 'M', '2') }, /* QDM2 */
154     { CODEC_ID_DVAUDIO, MKTAG('v', 'd', 'v', 'a') },
155     { CODEC_ID_DVAUDIO, MKTAG('d', 'v', 'c', 'a') },
156     { CODEC_ID_PCM_S16LE, MKTAG('l', 'p', 'c', 'm') },
157     { CODEC_ID_NONE, 0 },
158 };
159
160 /* the QuickTime file format is quite convoluted...
161  * it has lots of index tables, each indexing something in another one...
162  * Here we just use what is needed to read the chunks
163  */
164
165 typedef struct MOV_sample_to_chunk_tbl {
166     long first;
167     long count;
168     long id;
169 } MOV_sample_to_chunk_tbl;
170
171 typedef struct {
172     uint32_t type;
173     int64_t offset;
174     int64_t size; /* total size (excluding the size and type fields) */
175 } MOV_atom_t;
176
177 typedef struct {
178     int seed;
179     int flags;
180     int size;
181     void* clrs;
182 } MOV_ctab_t;
183
184 typedef struct MOV_mdat_atom_s {
185     offset_t offset;
186     int64_t size;
187 } MOV_mdat_atom_t;
188
189 typedef struct {
190     uint8_t  version;
191     uint32_t flags; // 24bit
192
193     /* 0x03 ESDescrTag */
194     uint16_t es_id;
195 #define MP4ODescrTag                    0x01
196 #define MP4IODescrTag                   0x02
197 #define MP4ESDescrTag                   0x03
198 #define MP4DecConfigDescrTag            0x04
199 #define MP4DecSpecificDescrTag          0x05
200 #define MP4SLConfigDescrTag             0x06
201 #define MP4ContentIdDescrTag            0x07
202 #define MP4SupplContentIdDescrTag       0x08
203 #define MP4IPIPtrDescrTag               0x09
204 #define MP4IPMPPtrDescrTag              0x0A
205 #define MP4IPMPDescrTag                 0x0B
206 #define MP4RegistrationDescrTag         0x0D
207 #define MP4ESIDIncDescrTag              0x0E
208 #define MP4ESIDRefDescrTag              0x0F
209 #define MP4FileIODescrTag               0x10
210 #define MP4FileODescrTag                0x11
211 #define MP4ExtProfileLevelDescrTag      0x13
212 #define MP4ExtDescrTagsStart            0x80
213 #define MP4ExtDescrTagsEnd              0xFE
214     uint8_t  stream_priority;
215
216     /* 0x04 DecConfigDescrTag */
217     uint8_t  object_type_id;
218     uint8_t  stream_type;
219     /* XXX: really streamType is
220      * only 6bit, followed by:
221      * 1bit  upStream
222      * 1bit  reserved
223      */
224     uint32_t buffer_size_db; // 24
225     uint32_t max_bitrate;
226     uint32_t avg_bitrate;
227
228     /* 0x05 DecSpecificDescrTag */
229     uint8_t  decoder_cfg_len;
230     uint8_t *decoder_cfg;
231
232     /* 0x06 SLConfigDescrTag */
233     uint8_t  sl_config_len;
234     uint8_t *sl_config;
235 } MOV_esds_t;
236
237 struct MOVParseTableEntry;
238
239 typedef struct MOVStreamContext {
240     int ffindex; /* the ffmpeg stream id */
241     long next_chunk;
242     long chunk_count;
243     int64_t *chunk_offsets;
244     int stts_count;
245     Time2Sample *stts_data;
246     int ctts_count;
247     Time2Sample *ctts_data;
248     int edit_count;             /* number of 'edit' (elst atom) */
249     long sample_to_chunk_sz;
250     MOV_sample_to_chunk_tbl *sample_to_chunk;
251     int sample_to_ctime_index;
252     int sample_to_ctime_sample;
253     long sample_size;
254     long sample_count;
255     long *sample_sizes;
256     long keyframe_count;
257     long *keyframes;
258     int time_scale;
259     int time_rate;
260     long current_sample;
261     MOV_esds_t esds;
262     AVRational sample_size_v1;
263     int dv_audio_container;
264 } MOVStreamContext;
265
266 typedef struct MOVContext {
267     AVFormatContext *fc;
268     int time_scale;
269     int64_t duration; /* duration of the longest track */
270     int found_moov; /* when both 'moov' and 'mdat' sections has been found */
271     int found_mdat; /* we suppose we have enough data to read the file */
272     int64_t mdat_size;
273     int64_t mdat_offset;
274     int total_streams;
275     /* some streams listed here aren't presented to the ffmpeg API, since they aren't either video nor audio
276      * but we need the info to be able to skip data from those streams in the 'mdat' section
277      */
278     MOVStreamContext *streams[MAX_STREAMS];
279
280     int ctab_size;
281     MOV_ctab_t **ctab;           /* color tables */
282     const struct MOVParseTableEntry *parse_table; /* could be eventually used to change the table */
283     /* NOTE: for recursion save to/ restore from local variable! */
284
285     AVPaletteControl palette_control;
286     MOV_mdat_atom_t *mdat_list;
287     int mdat_count;
288     DVDemuxContext *dv_demux;
289     AVFormatContext *dv_fctx;
290     int isom; /* 1 if file is ISO Media (mp4/3gp) */
291 } MOVContext;
292
293
294 /* XXX: it's the first time I make a recursive parser I think... sorry if it's ugly :P */
295
296 /* those functions parse an atom */
297 /* return code:
298  1: found what I wanted, exit
299  0: continue to parse next atom
300  -1: error occured, exit
301  */
302 typedef int (*mov_parse_function)(MOVContext *ctx, ByteIOContext *pb, MOV_atom_t atom);
303
304 /* links atom IDs to parse functions */
305 typedef struct MOVParseTableEntry {
306     uint32_t type;
307     mov_parse_function func;
308 } MOVParseTableEntry;
309
310 static int mov_read_default(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
311 {
312     int64_t total_size = 0;
313     MOV_atom_t a;
314     int i;
315     int err = 0;
316
317     a.offset = atom.offset;
318
319     if (atom.size < 0)
320         atom.size = 0x7fffffffffffffffLL;
321     while(((total_size + 8) < atom.size) && !url_feof(pb) && !err) {
322         a.size = atom.size;
323         a.type=0L;
324         if(atom.size >= 8) {
325             a.size = get_be32(pb);
326             a.type = get_le32(pb);
327         }
328         total_size += 8;
329         a.offset += 8;
330         dprintf("type: %08x  %.4s  sz: %"PRIx64"  %"PRIx64"   %"PRIx64"\n", a.type, (char*)&a.type, a.size, atom.size, total_size);
331         if (a.size == 1) { /* 64 bit extended size */
332             a.size = get_be64(pb) - 8;
333             a.offset += 8;
334             total_size += 8;
335         }
336         if (a.size == 0) {
337             a.size = atom.size - total_size;
338             if (a.size <= 8)
339                 break;
340         }
341         for (i = 0; c->parse_table[i].type != 0L
342              && c->parse_table[i].type != a.type; i++)
343             /* empty */;
344
345         a.size -= 8;
346
347         if(a.size < 0)
348             break;
349
350         if (c->parse_table[i].type == 0) { /* skip leaf atoms data */
351             url_fskip(pb, a.size);
352         } else {
353             offset_t start_pos = url_ftell(pb);
354             int64_t left;
355             err = (c->parse_table[i].func)(c, pb, a);
356             left = a.size - url_ftell(pb) + start_pos;
357             if (left > 0) /* skip garbage at atom end */
358                 url_fskip(pb, left);
359         }
360
361         a.offset += a.size;
362         total_size += a.size;
363     }
364
365     if (!err && total_size < atom.size && atom.size < 0x7ffff) {
366         url_fskip(pb, atom.size - total_size);
367     }
368
369     return err;
370 }
371
372 static int mov_read_ctab(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
373 {
374 #if 1
375     url_fskip(pb, atom.size); // for now
376 #else
377     VERY VERY BROKEN, NEVER execute this, needs rewrite
378     unsigned int len;
379     MOV_ctab_t *t;
380     c->ctab = av_realloc(c->ctab, ++c->ctab_size);
381     t = c->ctab[c->ctab_size];
382     t->seed = get_be32(pb);
383     t->flags = get_be16(pb);
384     t->size = get_be16(pb) + 1;
385     len = 2 * t->size * 4;
386     if (len > 0) {
387         t->clrs = av_malloc(len); // 16bit A R G B
388         if (t->clrs)
389             get_buffer(pb, t->clrs, len);
390     }
391 #endif
392
393     return 0;
394 }
395
396 static int mov_read_hdlr(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
397 {
398     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
399     uint32_t type;
400     uint32_t ctype;
401
402     get_byte(pb); /* version */
403     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
404
405     /* component type */
406     ctype = get_le32(pb);
407     type = get_le32(pb); /* component subtype */
408
409     dprintf("ctype= %c%c%c%c (0x%08lx)\n", *((char *)&ctype), ((char *)&ctype)[1], ((char *)&ctype)[2], ((char *)&ctype)[3], (long) ctype);
410     dprintf("stype= %c%c%c%c\n", *((char *)&type), ((char *)&type)[1], ((char *)&type)[2], ((char *)&type)[3]);
411     if(!ctype)
412         c->isom = 1;
413     if(type == MKTAG('v', 'i', 'd', 'e'))
414         st->codec->codec_type = CODEC_TYPE_VIDEO;
415     else if(type == MKTAG('s', 'o', 'u', 'n'))
416         st->codec->codec_type = CODEC_TYPE_AUDIO;
417     else if(type == MKTAG('m', '1', 'a', ' '))
418         st->codec->codec_id = CODEC_ID_MP2;
419     else if(type == MKTAG('s', 'u', 'b', 'p')) {
420         st->codec->codec_type = CODEC_TYPE_SUBTITLE;
421         st->codec->codec_id = CODEC_ID_DVD_SUBTITLE;
422     }
423     get_be32(pb); /* component  manufacture */
424     get_be32(pb); /* component flags */
425     get_be32(pb); /* component flags mask */
426
427     if(atom.size <= 24)
428         return 0; /* nothing left to read */
429
430     url_fskip(pb, atom.size - (url_ftell(pb) - atom.offset));
431     return 0;
432 }
433
434 static int mov_mp4_read_descr_len(ByteIOContext *pb)
435 {
436     int len = 0;
437     int count = 4;
438     while (count--) {
439         int c = get_byte(pb);
440         len = (len << 7) | (c & 0x7f);
441         if (!(c & 0x80))
442             break;
443     }
444     return len;
445 }
446
447 static int mov_mp4_read_descr(ByteIOContext *pb, int *tag)
448 {
449     int len;
450     *tag = get_byte(pb);
451     len = mov_mp4_read_descr_len(pb);
452     dprintf("MPEG4 description: tag=0x%02x len=%d\n", *tag, len);
453     return len;
454 }
455
456 static int mov_read_esds(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
457 {
458     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
459     MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
460     int tag, len;
461
462     /* Well, broken but suffisant for some MP4 streams */
463     get_be32(pb); /* version + flags */
464     len = mov_mp4_read_descr(pb, &tag);
465     if (tag == MP4ESDescrTag) {
466         get_be16(pb); /* ID */
467         get_byte(pb); /* priority */
468     } else
469         get_be16(pb); /* ID */
470
471     len = mov_mp4_read_descr(pb, &tag);
472     if (tag == MP4DecConfigDescrTag) {
473         sc->esds.object_type_id = get_byte(pb);
474         sc->esds.stream_type = get_byte(pb);
475         sc->esds.buffer_size_db = get_be24(pb);
476         sc->esds.max_bitrate = get_be32(pb);
477         sc->esds.avg_bitrate = get_be32(pb);
478
479         st->codec->codec_id= codec_get_id(ff_mov_obj_type, sc->esds.object_type_id);
480         dprintf("esds object type id %d\n", sc->esds.object_type_id);
481         len = mov_mp4_read_descr(pb, &tag);
482         if (tag == MP4DecSpecificDescrTag) {
483             dprintf("Specific MPEG4 header len=%d\n", len);
484             st->codec->extradata = av_mallocz(len + FF_INPUT_BUFFER_PADDING_SIZE);
485             if (st->codec->extradata) {
486                 get_buffer(pb, st->codec->extradata, len);
487                 st->codec->extradata_size = len;
488                 /* from mplayer */
489                 if ((*st->codec->extradata >> 3) == 29) {
490                     st->codec->codec_id = CODEC_ID_MP3ON4;
491                 }
492             }
493         }
494     }
495     return 0;
496 }
497
498 /* this atom contains actual media data */
499 static int mov_read_mdat(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
500 {
501     if(atom.size == 0) /* wrong one (MP4) */
502         return 0;
503     c->mdat_list = av_realloc(c->mdat_list, (c->mdat_count + 1) * sizeof(*c->mdat_list));
504     c->mdat_list[c->mdat_count].offset = atom.offset;
505     c->mdat_list[c->mdat_count].size = atom.size;
506     c->mdat_count++;
507     c->found_mdat=1;
508     c->mdat_offset = atom.offset;
509     c->mdat_size = atom.size;
510     if(c->found_moov)
511         return 1; /* found both, just go */
512     url_fskip(pb, atom.size);
513     return 0; /* now go for moov */
514 }
515
516 static int mov_read_ftyp(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
517 {
518     uint32_t type = get_le32(pb);
519
520     if (type != MKTAG('q','t',' ',' '))
521         c->isom = 1;
522     av_log(c->fc, AV_LOG_DEBUG, "ISO: File Type Major Brand: %.4s\n",(char *)&type);
523     get_be32(pb); /* minor version */
524     url_fskip(pb, atom.size - 8);
525     return 0;
526 }
527
528 /* this atom should contain all header atoms */
529 static int mov_read_moov(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
530 {
531     int err;
532
533     err = mov_read_default(c, pb, atom);
534     /* we parsed the 'moov' atom, we can terminate the parsing as soon as we find the 'mdat' */
535     /* so we don't parse the whole file if over a network */
536     c->found_moov=1;
537     if(c->found_mdat)
538         return 1; /* found both, just go */
539     return 0; /* now go for mdat */
540 }
541
542
543 static int mov_read_mdhd(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
544 {
545     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
546     MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
547     int version = get_byte(pb);
548     int lang;
549
550     if (version > 1)
551         return 1; /* unsupported */
552
553     get_byte(pb); get_byte(pb);
554     get_byte(pb); /* flags */
555
556     if (version == 1) {
557         get_be64(pb);
558         get_be64(pb);
559     } else {
560         get_be32(pb); /* creation time */
561         get_be32(pb); /* modification time */
562     }
563
564     sc->time_scale = get_be32(pb);
565     st->duration = (version == 1) ? get_be64(pb) : get_be32(pb); /* duration */
566
567     lang = get_be16(pb); /* language */
568     ff_mov_lang_to_iso639(lang, st->language);
569     get_be16(pb); /* quality */
570
571     return 0;
572 }
573
574 static int mov_read_mvhd(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
575 {
576     int version = get_byte(pb); /* version */
577     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
578
579     if (version == 1) {
580         get_be64(pb);
581         get_be64(pb);
582     } else {
583         get_be32(pb); /* creation time */
584         get_be32(pb); /* modification time */
585     }
586     c->time_scale = get_be32(pb); /* time scale */
587 #ifdef DEBUG
588     av_log(NULL, AV_LOG_DEBUG, "time scale = %i\n", c->time_scale);
589 #endif
590     c->duration = (version == 1) ? get_be64(pb) : get_be32(pb); /* duration */
591     get_be32(pb); /* preferred scale */
592
593     get_be16(pb); /* preferred volume */
594
595     url_fskip(pb, 10); /* reserved */
596
597     url_fskip(pb, 36); /* display matrix */
598
599     get_be32(pb); /* preview time */
600     get_be32(pb); /* preview duration */
601     get_be32(pb); /* poster time */
602     get_be32(pb); /* selection time */
603     get_be32(pb); /* selection duration */
604     get_be32(pb); /* current time */
605     get_be32(pb); /* next track ID */
606
607     return 0;
608 }
609
610 static int mov_read_smi(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
611 {
612     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
613
614     if((uint64_t)atom.size > (1<<30))
615         return -1;
616
617     // currently SVQ3 decoder expect full STSD header - so let's fake it
618     // this should be fixed and just SMI header should be passed
619     av_free(st->codec->extradata);
620     st->codec->extradata_size = 0x5a + atom.size;
621     st->codec->extradata = av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
622
623     if (st->codec->extradata) {
624         strcpy(st->codec->extradata, "SVQ3"); // fake
625         get_buffer(pb, st->codec->extradata + 0x5a, atom.size);
626         dprintf("Reading SMI %"PRId64"  %s\n", atom.size, st->codec->extradata + 0x5a);
627     } else
628         url_fskip(pb, atom.size);
629
630     return 0;
631 }
632
633 static int mov_read_enda(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
634 {
635     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
636     int little_endian = get_be16(pb);
637
638     if (little_endian) {
639         switch (st->codec->codec_id) {
640         case CODEC_ID_PCM_S24BE:
641             st->codec->codec_id = CODEC_ID_PCM_S24LE;
642             break;
643         case CODEC_ID_PCM_S32BE:
644             st->codec->codec_id = CODEC_ID_PCM_S32LE;
645             break;
646         default:
647             break;
648         }
649     }
650     return 0;
651 }
652
653 static int mov_read_alac(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
654 {
655     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
656
657     // currently ALAC decoder expect full atom header - so let's fake it
658     // this should be fixed and just ALAC header should be passed
659
660     av_free(st->codec->extradata);
661     st->codec->extradata_size = 36;
662     st->codec->extradata = av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
663
664     if (st->codec->extradata) {
665         strcpy(st->codec->extradata + 4, "alac"); // fake
666         get_buffer(pb, st->codec->extradata + 8, 36 - 8);
667         dprintf("Reading alac %d  %s\n", st->codec->extradata_size, st->codec->extradata);
668     } else
669         url_fskip(pb, atom.size);
670     return 0;
671 }
672
673 static int mov_read_wave(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
674 {
675     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
676
677     if((uint64_t)atom.size > (1<<30))
678         return -1;
679
680     if (st->codec->codec_id == CODEC_ID_QDM2) {
681         // pass all frma atom to codec, needed at least for QDM2
682         av_free(st->codec->extradata);
683         st->codec->extradata_size = atom.size;
684         st->codec->extradata = av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
685
686         if (st->codec->extradata) {
687             get_buffer(pb, st->codec->extradata, atom.size);
688         } else
689             url_fskip(pb, atom.size);
690     } else if (atom.size > 8) { /* to read frma, esds atoms */
691         mov_read_default(c, pb, atom);
692     } else
693         url_fskip(pb, atom.size);
694     return 0;
695 }
696
697 static int mov_read_jp2h(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
698 {
699     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
700
701     if((uint64_t)atom.size > (1<<30))
702         return -1;
703
704     av_free(st->codec->extradata);
705
706     st->codec->extradata_size = atom.size + 8;
707     st->codec->extradata = av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
708
709     /* pass all jp2h atom to codec */
710     if (st->codec->extradata) {
711         strcpy(st->codec->extradata + 4, "jp2h");
712         get_buffer(pb, st->codec->extradata + 8, atom.size);
713     } else
714         url_fskip(pb, atom.size);
715     return 0;
716 }
717
718 static int mov_read_avcC(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
719 {
720     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
721
722     if((uint64_t)atom.size > (1<<30))
723         return -1;
724
725     av_free(st->codec->extradata);
726
727     st->codec->extradata_size = atom.size;
728     st->codec->extradata = av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
729
730     if (st->codec->extradata) {
731         get_buffer(pb, st->codec->extradata, atom.size);
732     } else
733         url_fskip(pb, atom.size);
734
735     return 0;
736 }
737
738 static int mov_read_stco(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
739 {
740     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
741     MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
742     unsigned int i, entries;
743
744     get_byte(pb); /* version */
745     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
746
747     entries = get_be32(pb);
748
749     if(entries >= UINT_MAX/sizeof(int64_t))
750         return -1;
751
752     sc->chunk_count = entries;
753     sc->chunk_offsets = av_malloc(entries * sizeof(int64_t));
754     if (!sc->chunk_offsets)
755         return -1;
756     if (atom.type == MKTAG('s', 't', 'c', 'o')) {
757         for(i=0; i<entries; i++) {
758             sc->chunk_offsets[i] = get_be32(pb);
759         }
760     } else if (atom.type == MKTAG('c', 'o', '6', '4')) {
761         for(i=0; i<entries; i++) {
762             sc->chunk_offsets[i] = get_be64(pb);
763         }
764     } else
765         return -1;
766
767     return 0;
768 }
769
770 static int mov_read_stsd(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
771 {
772     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
773     MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
774     int entries, frames_per_sample;
775     uint32_t format;
776     uint8_t codec_name[32];
777
778     /* for palette traversal */
779     int color_depth;
780     int color_start;
781     int color_count;
782     int color_end;
783     int color_index;
784     int color_dec;
785     int color_greyscale;
786     unsigned char *color_table;
787     int j;
788     unsigned char r, g, b;
789
790     get_byte(pb); /* version */
791     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
792
793     entries = get_be32(pb);
794
795     while(entries--) { //Parsing Sample description table
796         enum CodecID id;
797         MOV_atom_t a = { 0, 0, 0 };
798         offset_t start_pos = url_ftell(pb);
799         int size = get_be32(pb); /* size */
800         format = get_le32(pb); /* data format */
801
802         get_be32(pb); /* reserved */
803         get_be16(pb); /* reserved */
804         get_be16(pb); /* index */
805
806         if (st->codec->codec_tag) {
807             /* multiple fourcc, just skip for now */
808             url_fskip(pb, size - (url_ftell(pb) - start_pos));
809             continue;
810         }
811
812         st->codec->codec_tag = format;
813         id = codec_get_id(mov_audio_tags, format);
814         if (st->codec->codec_type != CODEC_TYPE_VIDEO && id > 0) {
815             st->codec->codec_type = CODEC_TYPE_AUDIO;
816         } else if (st->codec->codec_type != CODEC_TYPE_AUDIO && /* do not overwrite codec type */
817                    format && format != MKTAG('m', 'p', '4', 's')) { /* skip old asf mpeg4 tag */
818             id = codec_get_id(mov_video_tags, format);
819             if (id <= 0)
820                 id = codec_get_id(codec_bmp_tags, format);
821             if (id > 0)
822                 st->codec->codec_type = CODEC_TYPE_VIDEO;
823         }
824
825         dprintf("size=%d 4CC= %c%c%c%c codec_type=%d\n",
826                 size,
827                 (format >> 0) & 0xff, (format >> 8) & 0xff, (format >> 16) & 0xff, (format >> 24) & 0xff,
828                 st->codec->codec_type);
829
830         if(st->codec->codec_type==CODEC_TYPE_VIDEO) {
831             st->codec->codec_id = id;
832             get_be16(pb); /* version */
833             get_be16(pb); /* revision level */
834             get_be32(pb); /* vendor */
835             get_be32(pb); /* temporal quality */
836             get_be32(pb); /* spacial quality */
837
838             st->codec->width = get_be16(pb); /* width */
839             st->codec->height = get_be16(pb); /* height */
840
841             get_be32(pb); /* horiz resolution */
842             get_be32(pb); /* vert resolution */
843             get_be32(pb); /* data size, always 0 */
844             frames_per_sample = get_be16(pb); /* frames per samples */
845 #ifdef DEBUG
846             av_log(NULL, AV_LOG_DEBUG, "frames/samples = %d\n", frames_per_sample);
847 #endif
848             get_buffer(pb, codec_name, 32); /* codec name, pascal string (FIXME: true for mp4?) */
849             if (codec_name[0] <= 31) {
850                 memcpy(st->codec->codec_name, &codec_name[1],codec_name[0]);
851                 st->codec->codec_name[codec_name[0]] = 0;
852             }
853
854             st->codec->bits_per_sample = get_be16(pb); /* depth */
855             st->codec->color_table_id = get_be16(pb); /* colortable id */
856
857             /* figure out the palette situation */
858             color_depth = st->codec->bits_per_sample & 0x1F;
859             color_greyscale = st->codec->bits_per_sample & 0x20;
860
861             /* if the depth is 2, 4, or 8 bpp, file is palettized */
862             if ((color_depth == 2) || (color_depth == 4) ||
863                 (color_depth == 8)) {
864
865                 if (color_greyscale) {
866
867                     /* compute the greyscale palette */
868                     color_count = 1 << color_depth;
869                     color_index = 255;
870                     color_dec = 256 / (color_count - 1);
871                     for (j = 0; j < color_count; j++) {
872                         r = g = b = color_index;
873                         c->palette_control.palette[j] =
874                             (r << 16) | (g << 8) | (b);
875                         color_index -= color_dec;
876                         if (color_index < 0)
877                             color_index = 0;
878                     }
879
880                 } else if (st->codec->color_table_id & 0x08) {
881
882                     /* if flag bit 3 is set, use the default palette */
883                     color_count = 1 << color_depth;
884                     if (color_depth == 2)
885                         color_table = ff_qt_default_palette_4;
886                     else if (color_depth == 4)
887                         color_table = ff_qt_default_palette_16;
888                     else
889                         color_table = ff_qt_default_palette_256;
890
891                     for (j = 0; j < color_count; j++) {
892                         r = color_table[j * 4 + 0];
893                         g = color_table[j * 4 + 1];
894                         b = color_table[j * 4 + 2];
895                         c->palette_control.palette[j] =
896                             (r << 16) | (g << 8) | (b);
897                     }
898
899                 } else {
900
901                     /* load the palette from the file */
902                     color_start = get_be32(pb);
903                     color_count = get_be16(pb);
904                     color_end = get_be16(pb);
905                     for (j = color_start; j <= color_end; j++) {
906                         /* each R, G, or B component is 16 bits;
907                          * only use the top 8 bits; skip alpha bytes
908                          * up front */
909                         get_byte(pb);
910                         get_byte(pb);
911                         r = get_byte(pb);
912                         get_byte(pb);
913                         g = get_byte(pb);
914                         get_byte(pb);
915                         b = get_byte(pb);
916                         get_byte(pb);
917                         c->palette_control.palette[j] =
918                             (r << 16) | (g << 8) | (b);
919                     }
920                 }
921
922                 st->codec->palctrl = &c->palette_control;
923                 st->codec->palctrl->palette_changed = 1;
924             } else
925                 st->codec->palctrl = NULL;
926         } else if(st->codec->codec_type==CODEC_TYPE_AUDIO) {
927             int bits_per_sample;
928             uint16_t version = get_be16(pb);
929
930             st->codec->codec_id = id;
931             get_be16(pb); /* revision level */
932             get_be32(pb); /* vendor */
933
934             st->codec->channels = get_be16(pb);             /* channel count */
935             dprintf("audio channels %d\n", st->codec->channels);
936             st->codec->bits_per_sample = get_be16(pb);      /* sample size */
937             /* do we need to force to 16 for AMR ? */
938
939             /* handle specific s8 codec */
940             get_be16(pb); /* compression id = 0*/
941             get_be16(pb); /* packet size = 0 */
942
943             st->codec->sample_rate = ((get_be32(pb) >> 16));
944
945             switch (st->codec->codec_id) {
946             case CODEC_ID_PCM_S8:
947             case CODEC_ID_PCM_U8:
948                 if (st->codec->bits_per_sample == 16)
949                     st->codec->codec_id = CODEC_ID_PCM_S16BE;
950                 break;
951             case CODEC_ID_PCM_S16LE:
952             case CODEC_ID_PCM_S16BE:
953                 if (st->codec->bits_per_sample == 8)
954                     st->codec->codec_id = CODEC_ID_PCM_S8;
955                 else if (st->codec->bits_per_sample == 24)
956                     st->codec->codec_id = CODEC_ID_PCM_S24BE;
957                 break;
958             default:
959                 break;
960             }
961
962             //Read QT version 1 fields. In version 0 theese dont exist
963             dprintf("version =%d, isom =%d\n",version,c->isom);
964             if(!c->isom) {
965                 if(version==1) {
966                     sc->sample_size_v1.den = get_be32(pb); /* samples per packet */
967                     get_be32(pb); /* bytes per packet */
968                     sc->sample_size_v1.num = get_be32(pb); /* bytes per frame */
969                     get_be32(pb); /* bytes per sample */
970                 } else if(version==2) {
971                     get_be32(pb); /* sizeof struct only */
972                     st->codec->sample_rate = av_int2dbl(get_be64(pb)); /* float 64 */
973                     st->codec->channels = get_be32(pb);
974                     get_be32(pb); /* always 0x7F000000 */
975                     get_be32(pb); /* bits per channel if sound is uncompressed */
976                     get_be32(pb); /* lcpm format specific flag */
977                     get_be32(pb); /* bytes per audio packet if constant */
978                     get_be32(pb); /* lpcm frames per audio packet if constant */
979                 }
980             }
981
982             bits_per_sample = av_get_bits_per_sample(st->codec->codec_id);
983             if (bits_per_sample) {
984                 st->codec->bits_per_sample = bits_per_sample;
985                 sc->sample_size = (bits_per_sample >> 3) * st->codec->channels;
986             }
987         } else {
988             /* other codec type, just skip (rtp, mp4s, tmcd ...) */
989             url_fskip(pb, size - (url_ftell(pb) - start_pos));
990         }
991         /* this will read extra atoms at the end (wave, alac, damr, avcC, SMI ...) */
992         a.size = size - (url_ftell(pb) - start_pos);
993         if (a.size > 8)
994             mov_read_default(c, pb, a);
995         else if (a.size > 0)
996             url_fskip(pb, a.size);
997     }
998
999     if(st->codec->codec_type==CODEC_TYPE_AUDIO && st->codec->sample_rate==0 && sc->time_scale>1) {
1000         st->codec->sample_rate= sc->time_scale;
1001     }
1002
1003     /* special codec parameters handling */
1004     switch (st->codec->codec_id) {
1005 #ifdef CONFIG_H261_DECODER
1006     case CODEC_ID_H261:
1007 #endif
1008 #ifdef CONFIG_H263_DECODER
1009     case CODEC_ID_H263:
1010 #endif
1011 #ifdef CONFIG_MPEG4_DECODER
1012     case CODEC_ID_MPEG4:
1013 #endif
1014         st->codec->width= 0; /* let decoder init width/height */
1015         st->codec->height= 0;
1016         break;
1017 #ifdef CONFIG_LIBFAAD
1018     case CODEC_ID_AAC:
1019 #endif
1020 #ifdef CONFIG_VORBIS_DECODER
1021     case CODEC_ID_VORBIS:
1022 #endif
1023     case CODEC_ID_MP3ON4:
1024         st->codec->sample_rate= 0; /* let decoder init parameters properly */
1025         break;
1026 #ifdef CONFIG_DV_DEMUXER
1027     case CODEC_ID_DVAUDIO:
1028         c->dv_fctx = av_alloc_format_context();
1029         c->dv_demux = dv_init_demux(c->dv_fctx);
1030         if (!c->dv_demux) {
1031             av_log(c->fc, AV_LOG_ERROR, "dv demux context init error\n");
1032             return -1;
1033         }
1034         sc->dv_audio_container = 1;
1035         st->codec->codec_id = CODEC_ID_PCM_S16LE;
1036         break;
1037 #endif
1038     /* no ifdef since parameters are always those */
1039     case CODEC_ID_AMR_WB:
1040         st->codec->sample_rate= 16000;
1041         st->codec->channels= 1; /* really needed */
1042         break;
1043     case CODEC_ID_AMR_NB:
1044         st->codec->sample_rate= 8000;
1045         st->codec->channels= 1; /* really needed */
1046         break;
1047     case CODEC_ID_MP2:
1048         st->codec->codec_type = CODEC_TYPE_AUDIO; /* force type after stsd for m1a hdlr */
1049         st->need_parsing = 1;
1050         break;
1051     default:
1052         break;
1053     }
1054
1055     return 0;
1056 }
1057
1058 static int mov_read_stsc(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
1059 {
1060     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1061     MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
1062     unsigned int i, entries;
1063
1064     get_byte(pb); /* version */
1065     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
1066
1067     entries = get_be32(pb);
1068
1069     if(entries >= UINT_MAX / sizeof(MOV_sample_to_chunk_tbl))
1070         return -1;
1071
1072 #ifdef DEBUG
1073 av_log(NULL, AV_LOG_DEBUG, "track[%i].stsc.entries = %i\n", c->fc->nb_streams-1, entries);
1074 #endif
1075     sc->sample_to_chunk_sz = entries;
1076     sc->sample_to_chunk = av_malloc(entries * sizeof(MOV_sample_to_chunk_tbl));
1077     if (!sc->sample_to_chunk)
1078         return -1;
1079     for(i=0; i<entries; i++) {
1080         sc->sample_to_chunk[i].first = get_be32(pb);
1081         sc->sample_to_chunk[i].count = get_be32(pb);
1082         sc->sample_to_chunk[i].id = get_be32(pb);
1083     }
1084     return 0;
1085 }
1086
1087 static int mov_read_stss(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
1088 {
1089     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1090     MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
1091     unsigned int i, entries;
1092
1093     get_byte(pb); /* version */
1094     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
1095
1096     entries = get_be32(pb);
1097
1098     if(entries >= UINT_MAX / sizeof(long))
1099         return -1;
1100
1101     sc->keyframe_count = entries;
1102 #ifdef DEBUG
1103     av_log(NULL, AV_LOG_DEBUG, "keyframe_count = %ld\n", sc->keyframe_count);
1104 #endif
1105     sc->keyframes = av_malloc(entries * sizeof(long));
1106     if (!sc->keyframes)
1107         return -1;
1108     for(i=0; i<entries; i++) {
1109         sc->keyframes[i] = get_be32(pb);
1110 #ifdef DEBUG
1111 /*        av_log(NULL, AV_LOG_DEBUG, "keyframes[]=%ld\n", sc->keyframes[i]); */
1112 #endif
1113     }
1114     return 0;
1115 }
1116
1117 static int mov_read_stsz(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
1118 {
1119     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1120     MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
1121     unsigned int i, entries, sample_size;
1122
1123     get_byte(pb); /* version */
1124     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
1125
1126     sample_size = get_be32(pb);
1127     if (!sc->sample_size) /* do not overwrite value computed in stsd */
1128         sc->sample_size = sample_size;
1129     entries = get_be32(pb);
1130     if(entries >= UINT_MAX / sizeof(long))
1131         return -1;
1132
1133     sc->sample_count = entries;
1134     if (sample_size)
1135         return 0;
1136
1137 #ifdef DEBUG
1138     av_log(NULL, AV_LOG_DEBUG, "sample_size = %ld sample_count = %ld\n", sc->sample_size, sc->sample_count);
1139 #endif
1140     sc->sample_sizes = av_malloc(entries * sizeof(long));
1141     if (!sc->sample_sizes)
1142         return -1;
1143     for(i=0; i<entries; i++) {
1144         sc->sample_sizes[i] = get_be32(pb);
1145 #ifdef DEBUG
1146         av_log(NULL, AV_LOG_DEBUG, "sample_sizes[]=%ld\n", sc->sample_sizes[i]);
1147 #endif
1148     }
1149     return 0;
1150 }
1151
1152 static int mov_read_stts(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
1153 {
1154     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1155     MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
1156     unsigned int i, entries;
1157     int64_t duration=0;
1158     int64_t total_sample_count=0;
1159
1160     get_byte(pb); /* version */
1161     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
1162     entries = get_be32(pb);
1163     if(entries >= UINT_MAX / sizeof(Time2Sample))
1164         return -1;
1165
1166     sc->stts_count = entries;
1167     sc->stts_data = av_malloc(entries * sizeof(Time2Sample));
1168
1169 #ifdef DEBUG
1170 av_log(NULL, AV_LOG_DEBUG, "track[%i].stts.entries = %i\n", c->fc->nb_streams-1, entries);
1171 #endif
1172
1173     sc->time_rate=0;
1174
1175     for(i=0; i<entries; i++) {
1176         int sample_duration;
1177         int sample_count;
1178
1179         sample_count=get_be32(pb);
1180         sample_duration = get_be32(pb);
1181         sc->stts_data[i].count= sample_count;
1182         sc->stts_data[i].duration= sample_duration;
1183
1184         sc->time_rate= ff_gcd(sc->time_rate, sample_duration);
1185
1186         dprintf("sample_count=%d, sample_duration=%d\n",sample_count,sample_duration);
1187
1188         duration+=(int64_t)sample_duration*sample_count;
1189         total_sample_count+=sample_count;
1190     }
1191
1192     st->nb_frames= total_sample_count;
1193     if(duration)
1194         st->duration= duration;
1195     return 0;
1196 }
1197
1198 static int mov_read_ctts(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
1199 {
1200     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1201     MOVStreamContext *sc = (MOVStreamContext *)st->priv_data;
1202     unsigned int i, entries;
1203
1204     get_byte(pb); /* version */
1205     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
1206     entries = get_be32(pb);
1207     if(entries >= UINT_MAX / sizeof(Time2Sample))
1208         return -1;
1209
1210     sc->ctts_count = entries;
1211     sc->ctts_data = av_malloc(entries * sizeof(Time2Sample));
1212
1213     dprintf("track[%i].ctts.entries = %i\n", c->fc->nb_streams-1, entries);
1214
1215     for(i=0; i<entries; i++) {
1216         int count    =get_be32(pb);
1217         int duration =get_be32(pb);
1218
1219         if (duration < 0) {
1220             av_log(c->fc, AV_LOG_ERROR, "negative ctts, ignoring\n");
1221             sc->ctts_count = 0;
1222             url_fskip(pb, 8 * (entries - i - 1));
1223             break;
1224         }
1225         sc->ctts_data[i].count   = count;
1226         sc->ctts_data[i].duration= duration;
1227
1228         sc->time_rate= ff_gcd(sc->time_rate, duration);
1229     }
1230     return 0;
1231 }
1232
1233 static int mov_read_trak(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
1234 {
1235     AVStream *st;
1236     MOVStreamContext *sc;
1237
1238     st = av_new_stream(c->fc, c->fc->nb_streams);
1239     if (!st) return -2;
1240     sc = av_mallocz(sizeof(MOVStreamContext));
1241     if (!sc) {
1242         av_free(st);
1243         return -1;
1244     }
1245
1246     st->priv_data = sc;
1247     st->codec->codec_type = CODEC_TYPE_DATA;
1248     st->start_time = 0; /* XXX: check */
1249     c->streams[c->fc->nb_streams-1] = sc;
1250
1251     return mov_read_default(c, pb, atom);
1252 }
1253
1254 static int mov_read_tkhd(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
1255 {
1256     AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1257     int version = get_byte(pb);
1258
1259     get_byte(pb); get_byte(pb);
1260     get_byte(pb); /* flags */
1261     /*
1262     MOV_TRACK_ENABLED 0x0001
1263     MOV_TRACK_IN_MOVIE 0x0002
1264     MOV_TRACK_IN_PREVIEW 0x0004
1265     MOV_TRACK_IN_POSTER 0x0008
1266     */
1267
1268     if (version == 1) {
1269         get_be64(pb);
1270         get_be64(pb);
1271     } else {
1272         get_be32(pb); /* creation time */
1273         get_be32(pb); /* modification time */
1274     }
1275     st->id = (int)get_be32(pb); /* track id (NOT 0 !)*/
1276     get_be32(pb); /* reserved */
1277     st->start_time = 0; /* check */
1278     (version == 1) ? get_be64(pb) : get_be32(pb); /* highlevel (considering edits) duration in movie timebase */
1279     get_be32(pb); /* reserved */
1280     get_be32(pb); /* reserved */
1281
1282     get_be16(pb); /* layer */
1283     get_be16(pb); /* alternate group */
1284     get_be16(pb); /* volume */
1285     get_be16(pb); /* reserved */
1286
1287     url_fskip(pb, 36); /* display matrix */
1288
1289     /* those are fixed-point */
1290     get_be32(pb); /* track width */
1291     get_be32(pb); /* track height */
1292
1293     return 0;
1294 }
1295
1296 /* this atom should be null (from specs), but some buggy files put the 'moov' atom inside it... */
1297 /* like the files created with Adobe Premiere 5.0, for samples see */
1298 /* http://graphics.tudelft.nl/~wouter/publications/soundtests/ */
1299 static int mov_read_wide(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
1300 {
1301     int err;
1302
1303     if (atom.size < 8)
1304         return 0; /* continue */
1305     if (get_be32(pb) != 0) { /* 0 sized mdat atom... use the 'wide' atom size */
1306         url_fskip(pb, atom.size - 4);
1307         return 0;
1308     }
1309     atom.type = get_le32(pb);
1310     atom.offset += 8;
1311     atom.size -= 8;
1312     if (atom.type != MKTAG('m', 'd', 'a', 't')) {
1313         url_fskip(pb, atom.size);
1314         return 0;
1315     }
1316     err = mov_read_mdat(c, pb, atom);
1317     return err;
1318 }
1319
1320 static int mov_read_cmov(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
1321 {
1322 #ifdef CONFIG_ZLIB
1323     ByteIOContext ctx;
1324     uint8_t *cmov_data;
1325     uint8_t *moov_data; /* uncompressed data */
1326     long cmov_len, moov_len;
1327     int ret;
1328
1329     get_be32(pb); /* dcom atom */
1330     if (get_le32(pb) != MKTAG( 'd', 'c', 'o', 'm' ))
1331         return -1;
1332     if (get_le32(pb) != MKTAG( 'z', 'l', 'i', 'b' )) {
1333         av_log(NULL, AV_LOG_ERROR, "unknown compression for cmov atom !");
1334         return -1;
1335     }
1336     get_be32(pb); /* cmvd atom */
1337     if (get_le32(pb) != MKTAG( 'c', 'm', 'v', 'd' ))
1338         return -1;
1339     moov_len = get_be32(pb); /* uncompressed size */
1340     cmov_len = atom.size - 6 * 4;
1341
1342     cmov_data = av_malloc(cmov_len);
1343     if (!cmov_data)
1344         return -1;
1345     moov_data = av_malloc(moov_len);
1346     if (!moov_data) {
1347         av_free(cmov_data);
1348         return -1;
1349     }
1350     get_buffer(pb, cmov_data, cmov_len);
1351     if(uncompress (moov_data, (uLongf *) &moov_len, (const Bytef *)cmov_data, cmov_len) != Z_OK)
1352         return -1;
1353     if(init_put_byte(&ctx, moov_data, moov_len, 0, NULL, NULL, NULL, NULL) != 0)
1354         return -1;
1355     atom.type = MKTAG( 'm', 'o', 'o', 'v' );
1356     atom.offset = 0;
1357     atom.size = moov_len;
1358 #ifdef DEBUG
1359 //    { int fd = open("/tmp/uncompheader.mov", O_WRONLY | O_CREAT); write(fd, moov_data, moov_len); close(fd); }
1360 #endif
1361     ret = mov_read_default(c, &ctx, atom);
1362     av_free(moov_data);
1363     av_free(cmov_data);
1364     return ret;
1365 #else
1366     av_log(c->fc, AV_LOG_ERROR, "this file requires zlib support compiled in\n");
1367     return -1;
1368 #endif
1369 }
1370
1371 /* edit list atom */
1372 static int mov_read_elst(MOVContext *c, ByteIOContext *pb, MOV_atom_t atom)
1373 {
1374     int i, edit_count;
1375
1376     get_byte(pb); /* version */
1377     get_byte(pb); get_byte(pb); get_byte(pb); /* flags */
1378     edit_count= c->streams[c->fc->nb_streams-1]->edit_count = get_be32(pb);     /* entries */
1379
1380     for(i=0; i<edit_count; i++){
1381         get_be32(pb); /* Track duration */
1382         get_be32(pb); /* Media time */
1383         get_be32(pb); /* Media rate */
1384     }
1385     dprintf("track[%i].edit_count = %i\n", c->fc->nb_streams-1, c->streams[c->fc->nb_streams-1]->edit_count);
1386     return 0;
1387 }
1388
1389 static const MOVParseTableEntry mov_default_parse_table[] = {
1390 /* mp4 atoms */
1391 { MKTAG( 'c', 'o', '6', '4' ), mov_read_stco },
1392 { MKTAG( 'c', 't', 't', 's' ), mov_read_ctts }, /* composition time to sample */
1393 { MKTAG( 'e', 'd', 't', 's' ), mov_read_default },
1394 { MKTAG( 'e', 'l', 's', 't' ), mov_read_elst },
1395 { MKTAG( 'e', 'n', 'd', 'a' ), mov_read_enda },
1396 { MKTAG( 'f', 't', 'y', 'p' ), mov_read_ftyp },
1397 { MKTAG( 'h', 'd', 'l', 'r' ), mov_read_hdlr },
1398 { MKTAG( 'j', 'p', '2', 'h' ), mov_read_jp2h },
1399 { MKTAG( 'm', 'd', 'a', 't' ), mov_read_mdat },
1400 { MKTAG( 'm', 'd', 'h', 'd' ), mov_read_mdhd },
1401 { MKTAG( 'm', 'd', 'i', 'a' ), mov_read_default },
1402 { MKTAG( 'm', 'i', 'n', 'f' ), mov_read_default },
1403 { MKTAG( 'm', 'o', 'o', 'v' ), mov_read_moov },
1404 { MKTAG( 'm', 'v', 'h', 'd' ), mov_read_mvhd },
1405 { MKTAG( 'S', 'M', 'I', ' ' ), mov_read_smi }, /* Sorenson extension ??? */
1406 { MKTAG( 'a', 'l', 'a', 'c' ), mov_read_alac }, /* alac specific atom */
1407 { MKTAG( 'a', 'v', 'c', 'C' ), mov_read_avcC },
1408 { MKTAG( 's', 't', 'b', 'l' ), mov_read_default },
1409 { MKTAG( 's', 't', 'c', 'o' ), mov_read_stco },
1410 { MKTAG( 's', 't', 's', 'c' ), mov_read_stsc },
1411 { MKTAG( 's', 't', 's', 'd' ), mov_read_stsd }, /* sample description */
1412 { MKTAG( 's', 't', 's', 's' ), mov_read_stss }, /* sync sample */
1413 { MKTAG( 's', 't', 's', 'z' ), mov_read_stsz }, /* sample size */
1414 { MKTAG( 's', 't', 't', 's' ), mov_read_stts },
1415 { MKTAG( 't', 'k', 'h', 'd' ), mov_read_tkhd }, /* track header */
1416 { MKTAG( 't', 'r', 'a', 'k' ), mov_read_trak },
1417 { MKTAG( 'w', 'a', 'v', 'e' ), mov_read_wave },
1418 { MKTAG( 'c', 't', 'a', 'b' ), mov_read_ctab },
1419 { MKTAG( 'e', 's', 'd', 's' ), mov_read_esds },
1420 { MKTAG( 'w', 'i', 'd', 'e' ), mov_read_wide }, /* place holder */
1421 { MKTAG( 'c', 'm', 'o', 'v' ), mov_read_cmov },
1422 { 0L, NULL }
1423 };
1424
1425 static void mov_free_stream_context(MOVStreamContext *sc)
1426 {
1427     if(sc) {
1428         av_freep(&sc->ctts_data);
1429         av_freep(&sc);
1430     }
1431 }
1432
1433 /* XXX: is it sufficient ? */
1434 static int mov_probe(AVProbeData *p)
1435 {
1436     unsigned int offset;
1437     uint32_t tag;
1438     int score = 0;
1439
1440     /* check file header */
1441     if (p->buf_size <= 12)
1442         return 0;
1443     offset = 0;
1444     for(;;) {
1445         /* ignore invalid offset */
1446         if ((offset + 8) > (unsigned int)p->buf_size)
1447             return score;
1448         tag = LE_32(p->buf + offset + 4);
1449         switch(tag) {
1450         /* check for obvious tags */
1451         case MKTAG( 'j', 'P', ' ', ' ' ): /* jpeg 2000 signature */
1452         case MKTAG( 'm', 'o', 'o', 'v' ):
1453         case MKTAG( 'm', 'd', 'a', 't' ):
1454         case MKTAG( 'p', 'n', 'o', 't' ): /* detect movs with preview pics like ew.mov and april.mov */
1455         case MKTAG( 'u', 'd', 't', 'a' ): /* Packet Video PVAuthor adds this and a lot of more junk */
1456             return AVPROBE_SCORE_MAX;
1457         /* those are more common words, so rate then a bit less */
1458         case MKTAG( 'w', 'i', 'd', 'e' ):
1459         case MKTAG( 'f', 'r', 'e', 'e' ):
1460         case MKTAG( 'j', 'u', 'n', 'k' ):
1461         case MKTAG( 'p', 'i', 'c', 't' ):
1462             return AVPROBE_SCORE_MAX - 5;
1463         case MKTAG( 'f', 't', 'y', 'p' ):
1464         case MKTAG( 's', 'k', 'i', 'p' ):
1465         case MKTAG( 'u', 'u', 'i', 'd' ):
1466             offset = BE_32(p->buf+offset) + offset;
1467             /* if we only find those cause probedata is too small at least rate them */
1468             score = AVPROBE_SCORE_MAX - 50;
1469             break;
1470         default:
1471             /* unrecognized tag */
1472             return score;
1473         }
1474     }
1475     return score;
1476 }
1477
1478 static void mov_build_index(MOVContext *mov, AVStream *st)
1479 {
1480     MOVStreamContext *sc = st->priv_data;
1481     offset_t current_offset;
1482     int64_t current_dts = 0;
1483     int stts_index = 0;
1484     int stsc_index = 0;
1485     int stss_index = 0;
1486     int i, j, k;
1487
1488     if (sc->sample_sizes || st->codec->codec_type == CODEC_TYPE_VIDEO || sc->dv_audio_container) {
1489         int keyframe, sample_size;
1490         int current_sample = 0;
1491         int stts_sample = 0;
1492         int distance = 0;
1493
1494         st->nb_frames = sc->sample_count;
1495         for (i = 0; i < sc->chunk_count; i++) {
1496             current_offset = sc->chunk_offsets[i];
1497             if (stsc_index + 1 < sc->sample_to_chunk_sz && i + 1 == sc->sample_to_chunk[stsc_index + 1].first)
1498                 stsc_index++;
1499             for (j = 0; j < sc->sample_to_chunk[stsc_index].count; j++) {
1500                 keyframe = !sc->keyframe_count || current_sample + 1 == sc->keyframes[stss_index];
1501                 if (keyframe) {
1502                     distance = 0;
1503                     if (stss_index + 1 < sc->keyframe_count)
1504                         stss_index++;
1505                 }
1506                 sample_size = sc->sample_size > 0 ? sc->sample_size : sc->sample_sizes[current_sample];
1507                 dprintf("AVIndex stream %d, sample %d, offset %"PRIx64", dts %"PRId64", size %d, distance %d, keyframe %d\n",
1508                         st->index, current_sample, current_offset, current_dts, sample_size, distance, keyframe);
1509                 av_add_index_entry(st, current_offset, current_dts, sample_size, distance, keyframe ? AVINDEX_KEYFRAME : 0);
1510                 current_offset += sample_size;
1511                 assert(sc->stts_data[stts_index].duration % sc->time_rate == 0);
1512                 current_dts += sc->stts_data[stts_index].duration / sc->time_rate;
1513                 distance++;
1514                 stts_sample++;
1515                 if (current_sample + 1 < sc->sample_count)
1516                     current_sample++;
1517                 if (stts_index + 1 < sc->stts_count && stts_sample == sc->stts_data[stts_index].count) {
1518                     stts_sample = 0;
1519                     stts_index++;
1520                 }
1521             }
1522         }
1523     } else { /* read whole chunk */
1524         int chunk_samples, chunk_size, chunk_duration;
1525
1526         for (i = 0; i < sc->chunk_count; i++) {
1527             current_offset = sc->chunk_offsets[i];
1528             if (stsc_index + 1 < sc->sample_to_chunk_sz && i + 1 == sc->sample_to_chunk[stsc_index + 1].first)
1529                 stsc_index++;
1530             chunk_samples = sc->sample_to_chunk[stsc_index].count;
1531             /* get chunk size */
1532             if (sc->sample_size > 1 || st->codec->codec_id == CODEC_ID_PCM_U8 || st->codec->codec_id == CODEC_ID_PCM_S8)
1533                 chunk_size = chunk_samples * sc->sample_size;
1534             else if (sc->sample_size_v1.den > 0 && (chunk_samples * sc->sample_size_v1.num % sc->sample_size_v1.den == 0))
1535                 chunk_size = chunk_samples * sc->sample_size_v1.num / sc->sample_size_v1.den;
1536             else { /* workaround to find nearest next chunk offset */
1537                 chunk_size = INT_MAX;
1538                 for (j = 0; j < mov->total_streams; j++) {
1539                     MOVStreamContext *msc = mov->streams[j];
1540
1541                     for (k = msc->next_chunk; k < msc->chunk_count; k++) {
1542                         if (msc->chunk_offsets[k] > current_offset && msc->chunk_offsets[k] - current_offset < chunk_size) {
1543                             chunk_size = msc->chunk_offsets[k] - current_offset;
1544                             msc->next_chunk = k;
1545                             break;
1546                         }
1547                     }
1548                 }
1549                 /* check for last chunk */
1550                 if (chunk_size == INT_MAX)
1551                     for (j = 0; j < mov->mdat_count; j++) {
1552                         dprintf("mdat %d, offset %"PRIx64", size %"PRId64", current offset %"PRIx64"\n",
1553                                 j, mov->mdat_list[j].offset, mov->mdat_list[j].size, current_offset);
1554                         if (mov->mdat_list[j].offset <= current_offset && mov->mdat_list[j].offset + mov->mdat_list[j].size > current_offset)
1555                             chunk_size = mov->mdat_list[j].offset + mov->mdat_list[j].size - current_offset;
1556                     }
1557                 assert(chunk_size != INT_MAX);
1558                 for (j = 0; j < mov->total_streams; j++) {
1559                     mov->streams[j]->next_chunk = 0;
1560                 }
1561             }
1562             av_add_index_entry(st, current_offset, current_dts, chunk_size, 0, AVINDEX_KEYFRAME);
1563             /* get chunk duration */
1564             chunk_duration = 0;
1565             while (chunk_samples > 0) {
1566                 if (chunk_samples < sc->stts_data[stts_index].count) {
1567                     chunk_duration += sc->stts_data[stts_index].duration * chunk_samples;
1568                     sc->stts_data[stts_index].count -= chunk_samples;
1569                     break;
1570                 } else {
1571                     chunk_duration += sc->stts_data[stts_index].duration * chunk_samples;
1572                     chunk_samples -= sc->stts_data[stts_index].count;
1573                     if (stts_index + 1 < sc->stts_count) {
1574                         stts_index++;
1575                     }
1576                 }
1577             }
1578             dprintf("AVIndex stream %d, chunk %d, offset %"PRIx64", dts %"PRId64", size %d, duration %d\n",
1579                     st->index, i, current_offset, current_dts, chunk_size, chunk_duration);
1580             assert(chunk_duration % sc->time_rate == 0);
1581             current_dts += chunk_duration / sc->time_rate;
1582         }
1583     }
1584     /* adjust sample count to avindex entries */
1585     sc->sample_count = st->nb_index_entries;
1586 }
1587
1588 static int mov_read_header(AVFormatContext *s, AVFormatParameters *ap)
1589 {
1590     MOVContext *mov = (MOVContext *) s->priv_data;
1591     ByteIOContext *pb = &s->pb;
1592     int i, err;
1593     MOV_atom_t atom = { 0, 0, 0 };
1594
1595     mov->fc = s;
1596     mov->parse_table = mov_default_parse_table;
1597
1598     if(!url_is_streamed(pb)) /* .mov and .mp4 aren't streamable anyway (only progressive download if moov is before mdat) */
1599         atom.size = url_fsize(pb);
1600     else
1601         atom.size = 0x7FFFFFFFFFFFFFFFLL;
1602
1603     /* check MOV header */
1604     err = mov_read_default(mov, pb, atom);
1605     if (err<0 || (!mov->found_moov && !mov->found_mdat)) {
1606         av_log(s, AV_LOG_ERROR, "mov: header not found !!! (err:%d, moov:%d, mdat:%d) pos:%"PRId64"\n",
1607                 err, mov->found_moov, mov->found_mdat, url_ftell(pb));
1608         return -1;
1609     }
1610     dprintf("on_parse_exit_offset=%d\n", (int) url_ftell(pb));
1611
1612     /* some cleanup : make sure we are on the mdat atom */
1613     if(!url_is_streamed(pb) && (url_ftell(pb) != mov->mdat_offset))
1614         url_fseek(pb, mov->mdat_offset, SEEK_SET);
1615
1616     mov->total_streams = s->nb_streams;
1617
1618     for(i=0; i<mov->total_streams; i++) {
1619         MOVStreamContext *sc = mov->streams[i];
1620
1621         if(!sc->time_rate)
1622             sc->time_rate=1;
1623         if(!sc->time_scale)
1624             sc->time_scale= mov->time_scale;
1625         av_set_pts_info(s->streams[i], 64, sc->time_rate, sc->time_scale);
1626
1627         if(s->streams[i]->duration != AV_NOPTS_VALUE){
1628             assert(s->streams[i]->duration % sc->time_rate == 0);
1629             s->streams[i]->duration /= sc->time_rate;
1630         }
1631         sc->ffindex = i;
1632         mov_build_index(mov, s->streams[i]);
1633     }
1634
1635     for(i=0; i<mov->total_streams; i++) {
1636         /* dont need those anymore */
1637         av_freep(&mov->streams[i]->chunk_offsets);
1638         av_freep(&mov->streams[i]->sample_to_chunk);
1639         av_freep(&mov->streams[i]->sample_sizes);
1640         av_freep(&mov->streams[i]->keyframes);
1641         av_freep(&mov->streams[i]->stts_data);
1642     }
1643     av_freep(&mov->mdat_list);
1644     return 0;
1645 }
1646
1647 static int mov_read_packet(AVFormatContext *s, AVPacket *pkt)
1648 {
1649     MOVContext *mov = s->priv_data;
1650     MOVStreamContext *sc = 0;
1651     AVIndexEntry *sample = 0;
1652     int64_t best_dts = INT64_MAX;
1653     int i;
1654
1655     for (i = 0; i < mov->total_streams; i++) {
1656         MOVStreamContext *msc = mov->streams[i];
1657
1658         if (s->streams[i]->discard != AVDISCARD_ALL && msc->current_sample < msc->sample_count) {
1659             AVIndexEntry *current_sample = &s->streams[i]->index_entries[msc->current_sample];
1660             int64_t dts = av_rescale(current_sample->timestamp * (int64_t)msc->time_rate, AV_TIME_BASE, msc->time_scale);
1661
1662             dprintf("stream %d, sample %ld, dts %"PRId64"\n", i, msc->current_sample, dts);
1663             if (dts < best_dts) {
1664                 sample = current_sample;
1665                 best_dts = dts;
1666                 sc = msc;
1667             }
1668         }
1669     }
1670     if (!sample)
1671         return -1;
1672     /* must be done just before reading, to avoid infinite loop on sample */
1673     sc->current_sample++;
1674     if (sample->pos >= url_fsize(&s->pb)) {
1675         av_log(mov->fc, AV_LOG_ERROR, "stream %d, offset 0x%"PRIx64": partial file\n", sc->ffindex, sample->pos);
1676         return -1;
1677     }
1678 #ifdef CONFIG_DV_DEMUXER
1679     if (sc->dv_audio_container) {
1680         dv_get_packet(mov->dv_demux, pkt);
1681         dprintf("dv audio pkt size %d\n", pkt->size);
1682     } else {
1683 #endif
1684         url_fseek(&s->pb, sample->pos, SEEK_SET);
1685         av_get_packet(&s->pb, pkt, sample->size);
1686 #ifdef CONFIG_DV_DEMUXER
1687         if (mov->dv_demux) {
1688             void *pkt_destruct_func = pkt->destruct;
1689             dv_produce_packet(mov->dv_demux, pkt, pkt->data, pkt->size);
1690             pkt->destruct = pkt_destruct_func;
1691         }
1692     }
1693 #endif
1694     pkt->stream_index = sc->ffindex;
1695     pkt->dts = sample->timestamp;
1696     if (sc->ctts_data) {
1697         assert(sc->ctts_data[sc->sample_to_ctime_index].duration % sc->time_rate == 0);
1698         pkt->pts = pkt->dts + sc->ctts_data[sc->sample_to_ctime_index].duration / sc->time_rate;
1699         /* update ctts context */
1700         sc->sample_to_ctime_sample++;
1701         if (sc->sample_to_ctime_index < sc->ctts_count && sc->ctts_data[sc->sample_to_ctime_index].count == sc->sample_to_ctime_sample) {
1702             sc->sample_to_ctime_index++;
1703             sc->sample_to_ctime_sample = 0;
1704         }
1705     } else {
1706         pkt->pts = pkt->dts;
1707     }
1708     pkt->flags |= sample->flags & AVINDEX_KEYFRAME ? PKT_FLAG_KEY : 0;
1709     pkt->pos = sample->pos;
1710     dprintf("stream %d, pts %"PRId64", dts %"PRId64", pos 0x%"PRIx64", duration %d\n", pkt->stream_index, pkt->pts, pkt->dts, pkt->pos, pkt->duration);
1711     return 0;
1712 }
1713
1714 static int mov_seek_stream(AVStream *st, int64_t timestamp, int flags)
1715 {
1716     MOVStreamContext *sc = st->priv_data;
1717     int sample, time_sample;
1718     int i;
1719
1720     sample = av_index_search_timestamp(st, timestamp, flags);
1721     dprintf("stream %d, timestamp %"PRId64", sample %d\n", st->index, timestamp, sample);
1722     if (sample < 0) /* not sure what to do */
1723         return -1;
1724     sc->current_sample = sample;
1725     dprintf("stream %d, found sample %ld\n", st->index, sc->current_sample);
1726     /* adjust ctts index */
1727     if (sc->ctts_data) {
1728         time_sample = 0;
1729         for (i = 0; i < sc->ctts_count; i++) {
1730             time_sample += sc->ctts_data[i].count;
1731             if (time_sample >= sc->current_sample) {
1732                 sc->sample_to_ctime_index = i;
1733                 sc->sample_to_ctime_sample = time_sample - sc->current_sample;
1734                 break;
1735             }
1736         }
1737     }
1738     return sample;
1739 }
1740
1741 static int mov_read_seek(AVFormatContext *s, int stream_index, int64_t sample_time, int flags)
1742 {
1743     AVStream *st;
1744     int64_t seek_timestamp, timestamp;
1745     int sample;
1746     int i;
1747
1748     if (stream_index >= s->nb_streams)
1749         return -1;
1750
1751     st = s->streams[stream_index];
1752     sample = mov_seek_stream(st, sample_time, flags);
1753     if (sample < 0)
1754         return -1;
1755
1756     /* adjust seek timestamp to found sample timestamp */
1757     seek_timestamp = st->index_entries[sample].timestamp;
1758
1759     for (i = 0; i < s->nb_streams; i++) {
1760         st = s->streams[i];
1761         if (stream_index == i || st->discard == AVDISCARD_ALL)
1762             continue;
1763
1764         timestamp = av_rescale_q(seek_timestamp, s->streams[stream_index]->time_base, st->time_base);
1765         mov_seek_stream(st, timestamp, flags);
1766     }
1767     return 0;
1768 }
1769
1770 static int mov_read_close(AVFormatContext *s)
1771 {
1772     int i;
1773     MOVContext *mov = (MOVContext *) s->priv_data;
1774     for(i=0; i<mov->total_streams; i++)
1775         mov_free_stream_context(mov->streams[i]);
1776     /* free color tabs */
1777     for(i=0; i<mov->ctab_size; i++)
1778         av_freep(&mov->ctab[i]);
1779     if(mov->dv_demux){
1780         for(i=0; i<mov->dv_fctx->nb_streams; i++){
1781             av_freep(&mov->dv_fctx->streams[i]->codec);
1782             av_freep(&mov->dv_fctx->streams[i]);
1783         }
1784         av_freep(&mov->dv_fctx);
1785         av_freep(&mov->dv_demux);
1786     }
1787     av_freep(&mov->ctab);
1788     return 0;
1789 }
1790
1791 AVInputFormat mov_demuxer = {
1792     "mov,mp4,m4a,3gp,3g2,mj2",
1793     "QuickTime/MPEG4/Motion JPEG 2000 format",
1794     sizeof(MOVContext),
1795     mov_probe,
1796     mov_read_header,
1797     mov_read_packet,
1798     mov_read_close,
1799     mov_read_seek,
1800 };