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