falkTX
/
FFmpeg
mirror of https://github.com/falkTX/FFmpeg.git
1
0
Fork 0
Code Issues 0 Releases 338 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
18662 Commits
32 Branches
498MB
Tree: 7943a90a0d
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '7943a90a0d'
${ noResults }
FFmpeg/libavformat/mov.c

2034 lines
71KB
Raw Blame History 

  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 "libavutil/intreadwrite.h"

  27. #include "libavutil/avstring.h"

  28. #include "avformat.h"

  29. #include "riff.h"

  30. #include "isom.h"

  31. #include "libavcodec/mpeg4audio.h"

  32. #include "libavcodec/mpegaudiodata.h"

  33. 

  34. #if CONFIG_ZLIB

  35. #include <zlib.h>

  36. #endif

  37. 

  38. /*

  39.  * First version by Francois Revol revol@free.fr

  40.  * Seek function by Gael Chardon gael.dev@4now.net

  41.  *

  42.  * Features and limitations:

  43.  * - reads most of the QT files I have (at least the structure),

  44.  *   Sample QuickTime files with mp3 audio can be found at: http://www.3ivx.com/showcase.html

  45.  * - the code is quite ugly... maybe I won't do it recursive next time :-)

  46.  *

  47.  * Funny I didn't know about http://sourceforge.net/projects/qt-ffmpeg/

  48.  * when coding this :) (it's a writer anyway)

  49.  *

  50.  * Reference documents:

  51.  * http://www.geocities.com/xhelmboyx/quicktime/formats/qtm-layout.txt

  52.  * Apple:

  53.  *  http://developer.apple.com/documentation/QuickTime/QTFF/

  54.  *  http://developer.apple.com/documentation/QuickTime/QTFF/qtff.pdf

  55.  * QuickTime is a trademark of Apple (AFAIK :))

  56.  */

  57. 

  58. #include "qtpalette.h"

  59. 

  60. 

  61. #undef NDEBUG

  62. #include <assert.h>

  63. 

  64. /* XXX: it's the first time I make a recursive parser I think... sorry if it's ugly :P */

  65. 

  66. /* those functions parse an atom */

  67. /* return code:

  68.   0: continue to parse next atom

  69.  <0: error occurred, exit

  70. */

  71. /* links atom IDs to parse functions */

  72. typedef struct MOVParseTableEntry {

  73.     uint32_t type;

  74.     int (*parse)(MOVContext *ctx, ByteIOContext *pb, MOVAtom atom);

  75. } MOVParseTableEntry;

  76. 

  77. static const MOVParseTableEntry mov_default_parse_table[];

  78. 

  79. static int mov_read_default(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  80. {

  81.     int64_t total_size = 0;

  82.     MOVAtom a;

  83.     int i;

  84.     int err = 0;

  85. 

  86.     a.offset = atom.offset;

  87. 

  88.     if (atom.size < 0)

  89.         atom.size = INT64_MAX;

  90.     while(((total_size + 8) < atom.size) && !url_feof(pb) && !err) {

  91.         a.size = atom.size;

  92.         a.type=0;

  93.         if(atom.size >= 8) {

  94.             a.size = get_be32(pb);

  95.             a.type = get_le32(pb);

  96.         }

  97.         total_size += 8;

  98.         a.offset += 8;

  99.         dprintf(c->fc, "type: %08x  %.4s  sz: %"PRIx64"  %"PRIx64"   %"PRIx64"\n",

  100.                 a.type, (char*)&a.type, a.size, atom.size, total_size);

  101.         if (a.size == 1) { /* 64 bit extended size */

  102.             a.size = get_be64(pb) - 8;

  103.             a.offset += 8;

  104.             total_size += 8;

  105.         }

  106.         if (a.size == 0) {

  107.             a.size = atom.size - total_size;

  108.             if (a.size <= 8)

  109.                 break;

  110.         }

  111.         a.size -= 8;

  112.         if(a.size < 0)

  113.             break;

  114.         a.size = FFMIN(a.size, atom.size - total_size);

  115. 

  116.         for (i = 0; mov_default_parse_table[i].type != 0

  117.              && mov_default_parse_table[i].type != a.type; i++)

  118.             /* empty */;

  119. 

  120.         if (mov_default_parse_table[i].type == 0) { /* skip leaf atoms data */

  121.             url_fskip(pb, a.size);

  122.         } else {

  123.             int64_t start_pos = url_ftell(pb);

  124.             int64_t left;

  125.             err = mov_default_parse_table[i].parse(c, pb, a);

  126.             if (url_is_streamed(pb) && c->found_moov && c->found_mdat)

  127.                 break;

  128.             left = a.size - url_ftell(pb) + start_pos;

  129.             if (left > 0) /* skip garbage at atom end */

  130.                 url_fskip(pb, left);

  131.         }

  132. 

  133.         a.offset += a.size;

  134.         total_size += a.size;

  135.     }

  136. 

  137.     if (!err && total_size < atom.size && atom.size < 0x7ffff)

  138.         url_fskip(pb, atom.size - total_size);

  139. 

  140.     return err;

  141. }

  142. 

  143. static int mov_read_dref(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  144. {

  145.     AVStream *st = c->fc->streams[c->fc->nb_streams-1];

  146.     MOVStreamContext *sc = st->priv_data;

  147.     int entries, i, j;

  148. 

  149.     get_be32(pb); // version + flags

  150.     entries = get_be32(pb);

  151.     if (entries >= UINT_MAX / sizeof(*sc->drefs))

  152.         return -1;

  153.     sc->drefs = av_mallocz(entries * sizeof(*sc->drefs));

  154.     if (!sc->drefs)

  155.         return AVERROR(ENOMEM);

  156.     sc->drefs_count = entries;

  157. 

  158.     for (i = 0; i < sc->drefs_count; i++) {

  159.         MOVDref *dref = &sc->drefs[i];

  160.         uint32_t size = get_be32(pb);

  161.         int64_t next = url_ftell(pb) + size - 4;

  162. 

  163.         dref->type = get_le32(pb);

  164.         get_be32(pb); // version + flags

  165.         dprintf(c->fc, "type %.4s size %d\n", (char*)&dref->type, size);

  166. 

  167.         if (dref->type == MKTAG('a','l','i','s') && size > 150) {

  168.             /* macintosh alias record */

  169.             uint16_t volume_len, len;

  170.             char volume[28];

  171.             int16_t type;

  172. 

  173.             url_fskip(pb, 10);

  174. 

  175.             volume_len = get_byte(pb);

  176.             volume_len = FFMIN(volume_len, 27);

  177.             get_buffer(pb, volume, 27);

  178.             volume[volume_len] = 0;

  179.             av_log(c->fc, AV_LOG_DEBUG, "volume %s, len %d\n", volume, volume_len);

  180. 

  181.             url_fskip(pb, 112);

  182. 

  183.             for (type = 0; type != -1 && url_ftell(pb) < next; ) {

  184.                 type = get_be16(pb);

  185.                 len = get_be16(pb);

  186.                 av_log(c->fc, AV_LOG_DEBUG, "type %d, len %d\n", type, len);

  187.                 if (len&1)

  188.                     len += 1;

  189.                 if (type == 2) { // absolute path

  190.                     av_free(dref->path);

  191.                     dref->path = av_mallocz(len+1);

  192.                     if (!dref->path)

  193.                         return AVERROR(ENOMEM);

  194.                     get_buffer(pb, dref->path, len);

  195.                     if (len > volume_len && !strncmp(dref->path, volume, volume_len)) {

  196.                         len -= volume_len;

  197.                         memmove(dref->path, dref->path+volume_len, len);

  198.                         dref->path[len] = 0;

  199.                     }

  200.                     for (j = 0; j < len; j++)

  201.                         if (dref->path[j] == ':')

  202.                             dref->path[j] = '/';

  203.                     av_log(c->fc, AV_LOG_DEBUG, "path %s\n", dref->path);

  204.                 } else

  205.                     url_fskip(pb, len);

  206.             }

  207.         }

  208.         url_fseek(pb, next, SEEK_SET);

  209.     }

  210.     return 0;

  211. }

  212. 

  213. static int mov_read_hdlr(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  214. {

  215.     AVStream *st;

  216.     uint32_t type;

  217.     uint32_t ctype;

  218. 

  219.     if (c->fc->nb_streams < 1) // meta before first trak

  220.         return 0;

  221. 

  222.     st = c->fc->streams[c->fc->nb_streams-1];

  223. 

  224.     get_byte(pb); /* version */

  225.     get_be24(pb); /* flags */

  226. 

  227.     /* component type */

  228.     ctype = get_le32(pb);

  229.     type = get_le32(pb); /* component subtype */

  230. 

  231.     dprintf(c->fc, "ctype= %c%c%c%c (0x%08x)\n", *((char *)&ctype), ((char *)&ctype)[1],

  232.             ((char *)&ctype)[2], ((char *)&ctype)[3], (int) ctype);

  233.     dprintf(c->fc, "stype= %c%c%c%c\n",

  234.             *((char *)&type), ((char *)&type)[1], ((char *)&type)[2], ((char *)&type)[3]);

  235.     if(!ctype)

  236.         c->isom = 1;

  237.     if     (type == MKTAG('v','i','d','e'))

  238.         st->codec->codec_type = CODEC_TYPE_VIDEO;

  239.     else if(type == MKTAG('s','o','u','n'))

  240.         st->codec->codec_type = CODEC_TYPE_AUDIO;

  241.     else if(type == MKTAG('m','1','a',' '))

  242.         st->codec->codec_id = CODEC_ID_MP2;

  243.     else if(type == MKTAG('s','u','b','p')) {

  244.         st->codec->codec_type = CODEC_TYPE_SUBTITLE;

  245.     }

  246.     get_be32(pb); /* component  manufacture */

  247.     get_be32(pb); /* component flags */

  248.     get_be32(pb); /* component flags mask */

  249. 

  250.     if(atom.size <= 24)

  251.         return 0; /* nothing left to read */

  252. 

  253.     url_fskip(pb, atom.size - (url_ftell(pb) - atom.offset));

  254.     return 0;

  255. }

  256. 

  257. static int mp4_read_descr_len(ByteIOContext *pb)

  258. {

  259.     int len = 0;

  260.     int count = 4;

  261.     while (count--) {

  262.         int c = get_byte(pb);

  263.         len = (len << 7) | (c & 0x7f);

  264.         if (!(c & 0x80))

  265.             break;

  266.     }

  267.     return len;

  268. }

  269. 

  270. static int mp4_read_descr(MOVContext *c, ByteIOContext *pb, int *tag)

  271. {

  272.     int len;

  273.     *tag = get_byte(pb);

  274.     len = mp4_read_descr_len(pb);

  275.     dprintf(c->fc, "MPEG4 description: tag=0x%02x len=%d\n", *tag, len);

  276.     return len;

  277. }

  278. 

  279. #define MP4ESDescrTag                   0x03

  280. #define MP4DecConfigDescrTag            0x04

  281. #define MP4DecSpecificDescrTag          0x05

  282. 

  283. static const AVCodecTag mp4_audio_types[] = {

  284.     { CODEC_ID_MP3ON4, 29 }, /* old mp3on4 draft */

  285.     { CODEC_ID_MP3ON4, 32 }, /* layer 1 */

  286.     { CODEC_ID_MP3ON4, 33 }, /* layer 2 */

  287.     { CODEC_ID_MP3ON4, 34 }, /* layer 3 */

  288.     { CODEC_ID_NONE,    0 },

  289. };

  290. 

  291. static int mov_read_esds(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  292. {

  293.     AVStream *st = c->fc->streams[c->fc->nb_streams-1];

  294.     int tag, len;

  295. 

  296.     get_be32(pb); /* version + flags */

  297.     len = mp4_read_descr(c, pb, &tag);

  298.     if (tag == MP4ESDescrTag) {

  299.         get_be16(pb); /* ID */

  300.         get_byte(pb); /* priority */

  301.     } else

  302.         get_be16(pb); /* ID */

  303. 

  304.     len = mp4_read_descr(c, pb, &tag);

  305.     if (tag == MP4DecConfigDescrTag) {

  306.         int object_type_id = get_byte(pb);

  307.         get_byte(pb); /* stream type */

  308.         get_be24(pb); /* buffer size db */

  309.         get_be32(pb); /* max bitrate */

  310.         get_be32(pb); /* avg bitrate */

  311. 

  312.         st->codec->codec_id= codec_get_id(ff_mp4_obj_type, object_type_id);

  313.         dprintf(c->fc, "esds object type id %d\n", object_type_id);

  314.         len = mp4_read_descr(c, pb, &tag);

  315.         if (tag == MP4DecSpecificDescrTag) {

  316.             dprintf(c->fc, "Specific MPEG4 header len=%d\n", len);

  317.             if((uint64_t)len > (1<<30))

  318.                 return -1;

  319.             st->codec->extradata = av_mallocz(len + FF_INPUT_BUFFER_PADDING_SIZE);

  320.             if (!st->codec->extradata)

  321.                 return AVERROR(ENOMEM);

  322.             get_buffer(pb, st->codec->extradata, len);

  323.             st->codec->extradata_size = len;

  324.             if (st->codec->codec_id == CODEC_ID_AAC) {

  325.                 MPEG4AudioConfig cfg;

  326.                 ff_mpeg4audio_get_config(&cfg, st->codec->extradata,

  327.                                          st->codec->extradata_size);

  328.                 if (cfg.chan_config > 7)

  329.                     return -1;

  330.                 st->codec->channels = ff_mpeg4audio_channels[cfg.chan_config];

  331.                 if (cfg.object_type == 29 && cfg.sampling_index < 3) // old mp3on4

  332.                     st->codec->sample_rate = ff_mpa_freq_tab[cfg.sampling_index];

  333.                 else

  334.                     st->codec->sample_rate = cfg.sample_rate; // ext sample rate ?

  335.                 dprintf(c->fc, "mp4a config channels %d obj %d ext obj %d "

  336.                         "sample rate %d ext sample rate %d\n", st->codec->channels,

  337.                         cfg.object_type, cfg.ext_object_type,

  338.                         cfg.sample_rate, cfg.ext_sample_rate);

  339.                 if (!(st->codec->codec_id = codec_get_id(mp4_audio_types,

  340.                                                          cfg.object_type)))

  341.                     st->codec->codec_id = CODEC_ID_AAC;

  342.             }

  343.         }

  344.     }

  345.     return 0;

  346. }

  347. 

  348. static int mov_read_pasp(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  349. {

  350.     const int num = get_be32(pb);

  351.     const int den = get_be32(pb);

  352.     AVStream * const st = c->fc->streams[c->fc->nb_streams-1];

  353.     if (den != 0) {

  354.         if ((st->sample_aspect_ratio.den != 1 || st->sample_aspect_ratio.num) && // default

  355.             (den != st->sample_aspect_ratio.den || num != st->sample_aspect_ratio.num))

  356.             av_log(c->fc, AV_LOG_WARNING,

  357.                    "sample aspect ratio already set to %d:%d, overriding by 'pasp' atom\n",

  358.                    st->sample_aspect_ratio.num, st->sample_aspect_ratio.den);

  359.         st->sample_aspect_ratio.num = num;

  360.         st->sample_aspect_ratio.den = den;

  361.     }

  362.     return 0;

  363. }

  364. 

  365. /* this atom contains actual media data */

  366. static int mov_read_mdat(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  367. {

  368.     if(atom.size == 0) /* wrong one (MP4) */

  369.         return 0;

  370.     c->found_mdat=1;

  371.     return 0; /* now go for moov */

  372. }

  373. 

  374. static int mov_read_ftyp(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  375. {

  376.     uint32_t type = get_le32(pb);

  377. 

  378.     if (type != MKTAG('q','t',' ',' '))

  379.         c->isom = 1;

  380.     av_log(c->fc, AV_LOG_DEBUG, "ISO: File Type Major Brand: %.4s\n",(char *)&type);

  381.     get_be32(pb); /* minor version */

  382.     url_fskip(pb, atom.size - 8);

  383.     return 0;

  384. }

  385. 

  386. /* this atom should contain all header atoms */

  387. static int mov_read_moov(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  388. {

  389.     if (mov_read_default(c, pb, atom) < 0)

  390.         return -1;

  391.     /* we parsed the 'moov' atom, we can terminate the parsing as soon as we find the 'mdat' */

  392.     /* so we don't parse the whole file if over a network */

  393.     c->found_moov=1;

  394.     return 0; /* now go for mdat */

  395. }

  396. 

  397. static int mov_read_moof(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  398. {

  399.     c->fragment.moof_offset = url_ftell(pb) - 8;

  400.     dprintf(c->fc, "moof offset %llx\n", c->fragment.moof_offset);

  401.     return mov_read_default(c, pb, atom);

  402. }

  403. 

  404. static int mov_read_mdhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  405. {

  406.     AVStream *st = c->fc->streams[c->fc->nb_streams-1];

  407.     MOVStreamContext *sc = st->priv_data;

  408.     int version = get_byte(pb);

  409.     char language[4] = {0};

  410.     unsigned lang;

  411. 

  412.     if (version > 1)

  413.         return -1; /* unsupported */

  414. 

  415.     get_be24(pb); /* flags */

  416.     if (version == 1) {

  417.         get_be64(pb);

  418.         get_be64(pb);

  419.     } else {

  420.         get_be32(pb); /* creation time */

  421.         get_be32(pb); /* modification time */

  422.     }

  423. 

  424.     sc->time_scale = get_be32(pb);

  425.     st->duration = (version == 1) ? get_be64(pb) : get_be32(pb); /* duration */

  426. 

  427.     lang = get_be16(pb); /* language */

  428.     if (ff_mov_lang_to_iso639(lang, language))

  429.         av_metadata_set(&st->metadata, "language", language);

  430.     get_be16(pb); /* quality */

  431. 

  432.     return 0;

  433. }

  434. 

  435. static int mov_read_mvhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  436. {

  437.     int version = get_byte(pb); /* version */

  438.     get_be24(pb); /* flags */

  439. 

  440.     if (version == 1) {

  441.         get_be64(pb);

  442.         get_be64(pb);

  443.     } else {

  444.         get_be32(pb); /* creation time */

  445.         get_be32(pb); /* modification time */

  446.     }

  447.     c->time_scale = get_be32(pb); /* time scale */

  448. 

  449.     dprintf(c->fc, "time scale = %i\n", c->time_scale);

  450. 

  451.     c->duration = (version == 1) ? get_be64(pb) : get_be32(pb); /* duration */

  452.     get_be32(pb); /* preferred scale */

  453. 

  454.     get_be16(pb); /* preferred volume */

  455. 

  456.     url_fskip(pb, 10); /* reserved */

  457. 

  458.     url_fskip(pb, 36); /* display matrix */

  459. 

  460.     get_be32(pb); /* preview time */

  461.     get_be32(pb); /* preview duration */

  462.     get_be32(pb); /* poster time */

  463.     get_be32(pb); /* selection time */

  464.     get_be32(pb); /* selection duration */

  465.     get_be32(pb); /* current time */

  466.     get_be32(pb); /* next track ID */

  467. 

  468.     return 0;

  469. }

  470. 

  471. static int mov_read_smi(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  472. {

  473.     AVStream *st = c->fc->streams[c->fc->nb_streams-1];

  474. 

  475.     if((uint64_t)atom.size > (1<<30))

  476.         return -1;

  477. 

  478.     // currently SVQ3 decoder expect full STSD header - so let's fake it

  479.     // this should be fixed and just SMI header should be passed

  480.     av_free(st->codec->extradata);

  481.     st->codec->extradata = av_mallocz(atom.size + 0x5a + FF_INPUT_BUFFER_PADDING_SIZE);

  482.     if (!st->codec->extradata)

  483.         return AVERROR(ENOMEM);

  484.     st->codec->extradata_size = 0x5a + atom.size;

  485.     memcpy(st->codec->extradata, "SVQ3", 4); // fake

  486.     get_buffer(pb, st->codec->extradata + 0x5a, atom.size);

  487.     dprintf(c->fc, "Reading SMI %"PRId64"  %s\n", atom.size, st->codec->extradata + 0x5a);

  488.     return 0;

  489. }

  490. 

  491. static int mov_read_enda(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  492. {

  493.     AVStream *st = c->fc->streams[c->fc->nb_streams-1];

  494.     int little_endian = get_be16(pb);

  495. 

  496.     dprintf(c->fc, "enda %d\n", little_endian);

  497.     if (little_endian == 1) {

  498.         switch (st->codec->codec_id) {

  499.         case CODEC_ID_PCM_S24BE:

  500.             st->codec->codec_id = CODEC_ID_PCM_S24LE;

  501.             break;

  502.         case CODEC_ID_PCM_S32BE:

  503.             st->codec->codec_id = CODEC_ID_PCM_S32LE;

  504.             break;

  505.         case CODEC_ID_PCM_F32BE:

  506.             st->codec->codec_id = CODEC_ID_PCM_F32LE;

  507.             break;

  508.         case CODEC_ID_PCM_F64BE:

  509.             st->codec->codec_id = CODEC_ID_PCM_F64LE;

  510.             break;

  511.         default:

  512.             break;

  513.         }

  514.     }

  515.     return 0;

  516. }

  517. 

  518. /* FIXME modify qdm2/svq3/h264 decoders to take full atom as extradata */

  519. static int mov_read_extradata(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  520. {

  521.     AVStream *st;

  522.     uint64_t size;

  523.     uint8_t *buf;

  524. 

  525.     if (c->fc->nb_streams < 1) // will happen with jp2 files

  526.         return 0;

  527.     st= c->fc->streams[c->fc->nb_streams-1];

  528.     size= (uint64_t)st->codec->extradata_size + atom.size + 8 + FF_INPUT_BUFFER_PADDING_SIZE;

  529.     if(size > INT_MAX || (uint64_t)atom.size > INT_MAX)

  530.         return -1;

  531.     buf= av_realloc(st->codec->extradata, size);

  532.     if(!buf)

  533.         return -1;

  534.     st->codec->extradata= buf;

  535.     buf+= st->codec->extradata_size;

  536.     st->codec->extradata_size= size - FF_INPUT_BUFFER_PADDING_SIZE;

  537.     AV_WB32(       buf    , atom.size + 8);

  538.     AV_WL32(       buf + 4, atom.type);

  539.     get_buffer(pb, buf + 8, atom.size);

  540.     return 0;

  541. }

  542. 

  543. static int mov_read_wave(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  544. {

  545.     AVStream *st = c->fc->streams[c->fc->nb_streams-1];

  546. 

  547.     if((uint64_t)atom.size > (1<<30))

  548.         return -1;

  549. 

  550.     if (st->codec->codec_id == CODEC_ID_QDM2) {

  551.         // pass all frma atom to codec, needed at least for QDM2

  552.         av_free(st->codec->extradata);

  553.         st->codec->extradata = av_mallocz(atom.size + FF_INPUT_BUFFER_PADDING_SIZE);

  554.         if (!st->codec->extradata)

  555.             return AVERROR(ENOMEM);

  556.         st->codec->extradata_size = atom.size;

  557.         get_buffer(pb, st->codec->extradata, atom.size);

  558.     } else if (atom.size > 8) { /* to read frma, esds atoms */

  559.         if (mov_read_default(c, pb, atom) < 0)

  560.             return -1;

  561.     } else

  562.         url_fskip(pb, atom.size);

  563.     return 0;

  564. }

  565. 

  566. /**

  567.  * This function reads atom content and puts data in extradata without tag

  568.  * nor size unlike mov_read_extradata.

  569.  */

  570. static int mov_read_glbl(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  571. {

  572.     AVStream *st = c->fc->streams[c->fc->nb_streams-1];

  573. 

  574.     if((uint64_t)atom.size > (1<<30))

  575.         return -1;

  576. 

  577.     av_free(st->codec->extradata);

  578.     st->codec->extradata = av_mallocz(atom.size + FF_INPUT_BUFFER_PADDING_SIZE);

  579.     if (!st->codec->extradata)

  580.         return AVERROR(ENOMEM);

  581.     st->codec->extradata_size = atom.size;

  582.     get_buffer(pb, st->codec->extradata, atom.size);

  583.     return 0;

  584. }

  585. 

  586. static int mov_read_stco(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  587. {

  588.     AVStream *st = c->fc->streams[c->fc->nb_streams-1];

  589.     MOVStreamContext *sc = st->priv_data;

  590.     unsigned int i, entries;

  591. 

  592.     get_byte(pb); /* version */

  593.     get_be24(pb); /* flags */

  594. 

  595.     entries = get_be32(pb);

  596. 

  597.     if(entries >= UINT_MAX/sizeof(int64_t))

  598.         return -1;

  599. 

  600.     sc->chunk_offsets = av_malloc(entries * sizeof(int64_t));

  601.     if (!sc->chunk_offsets)

  602.         return AVERROR(ENOMEM);

  603.     sc->chunk_count = entries;

  604. 

  605.     if      (atom.type == MKTAG('s','t','c','o'))

  606.         for(i=0; i<entries; i++)

  607.             sc->chunk_offsets[i] = get_be32(pb);

  608.     else if (atom.type == MKTAG('c','o','6','4'))

  609.         for(i=0; i<entries; i++)

  610.             sc->chunk_offsets[i] = get_be64(pb);

  611.     else

  612.         return -1;

  613. 

  614.     return 0;

  615. }

  616. 

  617. /**

  618.  * Compute codec id for 'lpcm' tag.

  619.  * See CoreAudioTypes and AudioStreamBasicDescription at Apple.

  620.  */

  621. static enum CodecID mov_get_lpcm_codec_id(int bps, int flags)

  622. {

  623.     if (flags & 1) { // floating point

  624.         if (flags & 2) { // big endian

  625.             if      (bps == 32) return CODEC_ID_PCM_F32BE;

  626.             else if (bps == 64) return CODEC_ID_PCM_F64BE;

  627.         } else {

  628.             if      (bps == 32) return CODEC_ID_PCM_F32LE;

  629.             else if (bps == 64) return CODEC_ID_PCM_F64LE;

  630.         }

  631.     } else {

  632.         if (flags & 2) {

  633.             if      (bps == 8)

  634.                 // signed integer

  635.                 if (flags & 4)  return CODEC_ID_PCM_S8;

  636.                 else            return CODEC_ID_PCM_U8;

  637.             else if (bps == 16) return CODEC_ID_PCM_S16BE;

  638.             else if (bps == 24) return CODEC_ID_PCM_S24BE;

  639.             else if (bps == 32) return CODEC_ID_PCM_S32BE;

  640.         } else {

  641.             if      (bps == 8)

  642.                 if (flags & 4)  return CODEC_ID_PCM_S8;

  643.                 else            return CODEC_ID_PCM_U8;

  644.             else if (bps == 16) return CODEC_ID_PCM_S16LE;

  645.             else if (bps == 24) return CODEC_ID_PCM_S24LE;

  646.             else if (bps == 32) return CODEC_ID_PCM_S32LE;

  647.         }

  648.     }

  649.     return CODEC_ID_NONE;

  650. }

  651. 

  652. static int mov_read_stsd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  653. {

  654.     AVStream *st = c->fc->streams[c->fc->nb_streams-1];

  655.     MOVStreamContext *sc = st->priv_data;

  656.     int j, entries, pseudo_stream_id;

  657. 

  658.     get_byte(pb); /* version */

  659.     get_be24(pb); /* flags */

  660. 

  661.     entries = get_be32(pb);

  662. 

  663.     for(pseudo_stream_id=0; pseudo_stream_id<entries; pseudo_stream_id++) {

  664.         //Parsing Sample description table

  665.         enum CodecID id;

  666.         int dref_id;

  667.         MOVAtom a = { 0, 0, 0 };

  668.         int64_t start_pos = url_ftell(pb);

  669.         int size = get_be32(pb); /* size */

  670.         uint32_t format = get_le32(pb); /* data format */

  671. 

  672.         get_be32(pb); /* reserved */

  673.         get_be16(pb); /* reserved */

  674.         dref_id = get_be16(pb);

  675. 

  676.         if (st->codec->codec_tag &&

  677.             st->codec->codec_tag != format &&

  678.             (c->fc->video_codec_id ? codec_get_id(codec_movvideo_tags, format) != c->fc->video_codec_id

  679.                                    : st->codec->codec_tag != MKTAG('j','p','e','g'))

  680.            ){

  681.             /* Multiple fourcc, we skip JPEG. This is not correct, we should

  682.              * export it as a separate AVStream but this needs a few changes

  683.              * in the MOV demuxer, patch welcome. */

  684.             av_log(c->fc, AV_LOG_WARNING, "multiple fourcc not supported\n");

  685.             url_fskip(pb, size - (url_ftell(pb) - start_pos));

  686.             continue;

  687.         }

  688.         sc->pseudo_stream_id = st->codec->codec_tag ? -1 : pseudo_stream_id;

  689.         sc->dref_id= dref_id;

  690. 

  691.         st->codec->codec_tag = format;

  692.         id = codec_get_id(codec_movaudio_tags, format);

  693.         if (id<=0 && (format&0xFFFF) == 'm'+('s'<<8))

  694.             id = codec_get_id(codec_wav_tags, bswap_32(format)&0xFFFF);

  695. 

  696.         if (st->codec->codec_type != CODEC_TYPE_VIDEO && id > 0) {

  697.             st->codec->codec_type = CODEC_TYPE_AUDIO;

  698.         } else if (st->codec->codec_type != CODEC_TYPE_AUDIO && /* do not overwrite codec type */

  699.                    format && format != MKTAG('m','p','4','s')) { /* skip old asf mpeg4 tag */

  700.             id = codec_get_id(codec_movvideo_tags, format);

  701.             if (id <= 0)

  702.                 id = codec_get_id(codec_bmp_tags, format);

  703.             if (id > 0)

  704.                 st->codec->codec_type = CODEC_TYPE_VIDEO;

  705.             else if(st->codec->codec_type == CODEC_TYPE_DATA){

  706.                 id = codec_get_id(ff_codec_movsubtitle_tags, format);

  707.                 if(id > 0)

  708.                     st->codec->codec_type = CODEC_TYPE_SUBTITLE;

  709.             }

  710.         }

  711. 

  712.         dprintf(c->fc, "size=%d 4CC= %c%c%c%c codec_type=%d\n", size,

  713.                 (format >> 0) & 0xff, (format >> 8) & 0xff, (format >> 16) & 0xff,

  714.                 (format >> 24) & 0xff, st->codec->codec_type);

  715. 

  716.         if(st->codec->codec_type==CODEC_TYPE_VIDEO) {

  717.             uint8_t codec_name[32];

  718.             unsigned int color_depth;

  719.             int color_greyscale;

  720. 

  721.             st->codec->codec_id = id;

  722.             get_be16(pb); /* version */

  723.             get_be16(pb); /* revision level */

  724.             get_be32(pb); /* vendor */

  725.             get_be32(pb); /* temporal quality */

  726.             get_be32(pb); /* spatial quality */

  727. 

  728.             st->codec->width = get_be16(pb); /* width */

  729.             st->codec->height = get_be16(pb); /* height */

  730. 

  731.             get_be32(pb); /* horiz resolution */

  732.             get_be32(pb); /* vert resolution */

  733.             get_be32(pb); /* data size, always 0 */

  734.             get_be16(pb); /* frames per samples */

  735. 

  736.             get_buffer(pb, codec_name, 32); /* codec name, pascal string */

  737.             if (codec_name[0] <= 31) {

  738.                 memcpy(st->codec->codec_name, &codec_name[1],codec_name[0]);

  739.                 st->codec->codec_name[codec_name[0]] = 0;

  740.             }

  741. 

  742.             st->codec->bits_per_coded_sample = get_be16(pb); /* depth */

  743.             st->codec->color_table_id = get_be16(pb); /* colortable id */

  744.             dprintf(c->fc, "depth %d, ctab id %d\n",

  745.                    st->codec->bits_per_coded_sample, st->codec->color_table_id);

  746.             /* figure out the palette situation */

  747.             color_depth = st->codec->bits_per_coded_sample & 0x1F;

  748.             color_greyscale = st->codec->bits_per_coded_sample & 0x20;

  749. 

  750.             /* if the depth is 2, 4, or 8 bpp, file is palettized */

  751.             if ((color_depth == 2) || (color_depth == 4) ||

  752.                 (color_depth == 8)) {

  753.                 /* for palette traversal */

  754.                 unsigned int color_start, color_count, color_end;

  755.                 unsigned char r, g, b;

  756. 

  757.                 if (color_greyscale) {

  758.                     int color_index, color_dec;

  759.                     /* compute the greyscale palette */

  760.                     st->codec->bits_per_coded_sample = color_depth;

  761.                     color_count = 1 << color_depth;

  762.                     color_index = 255;

  763.                     color_dec = 256 / (color_count - 1);

  764.                     for (j = 0; j < color_count; j++) {

  765.                         r = g = b = color_index;

  766.                         c->palette_control.palette[j] =

  767.                             (r << 16) | (g << 8) | (b);

  768.                         color_index -= color_dec;

  769.                         if (color_index < 0)

  770.                             color_index = 0;

  771.                     }

  772.                 } else if (st->codec->color_table_id) {

  773.                     const uint8_t *color_table;

  774.                     /* if flag bit 3 is set, use the default palette */

  775.                     color_count = 1 << color_depth;

  776.                     if (color_depth == 2)

  777.                         color_table = ff_qt_default_palette_4;

  778.                     else if (color_depth == 4)

  779.                         color_table = ff_qt_default_palette_16;

  780.                     else

  781.                         color_table = ff_qt_default_palette_256;

  782. 

  783.                     for (j = 0; j < color_count; j++) {

  784.                         r = color_table[j * 4 + 0];

  785.                         g = color_table[j * 4 + 1];

  786.                         b = color_table[j * 4 + 2];

  787.                         c->palette_control.palette[j] =

  788.                             (r << 16) | (g << 8) | (b);

  789.                     }

  790.                 } else {

  791.                     /* load the palette from the file */

  792.                     color_start = get_be32(pb);

  793.                     color_count = get_be16(pb);

  794.                     color_end = get_be16(pb);

  795.                     if ((color_start <= 255) &&

  796.                         (color_end <= 255)) {

  797.                         for (j = color_start; j <= color_end; j++) {

  798.                             /* each R, G, or B component is 16 bits;

  799.                              * only use the top 8 bits; skip alpha bytes

  800.                              * up front */

  801.                             get_byte(pb);

  802.                             get_byte(pb);

  803.                             r = get_byte(pb);

  804.                             get_byte(pb);

  805.                             g = get_byte(pb);

  806.                             get_byte(pb);

  807.                             b = get_byte(pb);

  808.                             get_byte(pb);

  809.                             c->palette_control.palette[j] =

  810.                                 (r << 16) | (g << 8) | (b);

  811.                         }

  812.                     }

  813.                 }

  814.                 st->codec->palctrl = &c->palette_control;

  815.                 st->codec->palctrl->palette_changed = 1;

  816.             } else

  817.                 st->codec->palctrl = NULL;

  818.         } else if(st->codec->codec_type==CODEC_TYPE_AUDIO) {

  819.             int bits_per_sample, flags;

  820.             uint16_t version = get_be16(pb);

  821. 

  822.             st->codec->codec_id = id;

  823.             get_be16(pb); /* revision level */

  824.             get_be32(pb); /* vendor */

  825. 

  826.             st->codec->channels = get_be16(pb);             /* channel count */

  827.             dprintf(c->fc, "audio channels %d\n", st->codec->channels);

  828.             st->codec->bits_per_coded_sample = get_be16(pb);      /* sample size */

  829. 

  830.             sc->audio_cid = get_be16(pb);

  831.             get_be16(pb); /* packet size = 0 */

  832. 

  833.             st->codec->sample_rate = ((get_be32(pb) >> 16));

  834. 

  835.             //Read QT version 1 fields. In version 0 these do not exist.

  836.             dprintf(c->fc, "version =%d, isom =%d\n",version,c->isom);

  837.             if(!c->isom) {

  838.                 if(version==1) {

  839.                     sc->samples_per_frame = get_be32(pb);

  840.                     get_be32(pb); /* bytes per packet */

  841.                     sc->bytes_per_frame = get_be32(pb);

  842.                     get_be32(pb); /* bytes per sample */

  843.                 } else if(version==2) {

  844.                     get_be32(pb); /* sizeof struct only */

  845.                     st->codec->sample_rate = av_int2dbl(get_be64(pb)); /* float 64 */

  846.                     st->codec->channels = get_be32(pb);

  847.                     get_be32(pb); /* always 0x7F000000 */

  848.                     st->codec->bits_per_coded_sample = get_be32(pb); /* bits per channel if sound is uncompressed */

  849.                     flags = get_be32(pb); /* lcpm format specific flag */

  850.                     sc->bytes_per_frame = get_be32(pb); /* bytes per audio packet if constant */

  851.                     sc->samples_per_frame = get_be32(pb); /* lpcm frames per audio packet if constant */

  852.                     if (format == MKTAG('l','p','c','m'))

  853.                         st->codec->codec_id = mov_get_lpcm_codec_id(st->codec->bits_per_coded_sample, flags);

  854.                 }

  855.             }

  856. 

  857.             switch (st->codec->codec_id) {

  858.             case CODEC_ID_PCM_S8:

  859.             case CODEC_ID_PCM_U8:

  860.                 if (st->codec->bits_per_coded_sample == 16)

  861.                     st->codec->codec_id = CODEC_ID_PCM_S16BE;

  862.                 break;

  863.             case CODEC_ID_PCM_S16LE:

  864.             case CODEC_ID_PCM_S16BE:

  865.                 if (st->codec->bits_per_coded_sample == 8)

  866.                     st->codec->codec_id = CODEC_ID_PCM_S8;

  867.                 else if (st->codec->bits_per_coded_sample == 24)

  868.                     st->codec->codec_id =

  869.                         st->codec->codec_id == CODEC_ID_PCM_S16BE ?

  870.                         CODEC_ID_PCM_S24BE : CODEC_ID_PCM_S24LE;

  871.                 break;

  872.             /* set values for old format before stsd version 1 appeared */

  873.             case CODEC_ID_MACE3:

  874.                 sc->samples_per_frame = 6;

  875.                 sc->bytes_per_frame = 2*st->codec->channels;

  876.                 break;

  877.             case CODEC_ID_MACE6:

  878.                 sc->samples_per_frame = 6;

  879.                 sc->bytes_per_frame = 1*st->codec->channels;

  880.                 break;

  881.             case CODEC_ID_ADPCM_IMA_QT:

  882.                 sc->samples_per_frame = 64;

  883.                 sc->bytes_per_frame = 34*st->codec->channels;

  884.                 break;

  885.             case CODEC_ID_GSM:

  886.                 sc->samples_per_frame = 160;

  887.                 sc->bytes_per_frame = 33;

  888.                 break;

  889.             default:

  890.                 break;

  891.             }

  892. 

  893.             bits_per_sample = av_get_bits_per_sample(st->codec->codec_id);

  894.             if (bits_per_sample) {

  895.                 st->codec->bits_per_coded_sample = bits_per_sample;

  896.                 sc->sample_size = (bits_per_sample >> 3) * st->codec->channels;

  897.             }

  898.         } else if(st->codec->codec_type==CODEC_TYPE_SUBTITLE){

  899.             // ttxt stsd contains display flags, justification, background

  900.             // color, fonts, and default styles, so fake an atom to read it

  901.             MOVAtom fake_atom = { .size = size - (url_ftell(pb) - start_pos) };

  902.             mov_read_glbl(c, pb, fake_atom);

  903.             st->codec->codec_id= id;

  904.             st->codec->width = sc->width;

  905.             st->codec->height = sc->height;

  906.         } else {

  907.             /* other codec type, just skip (rtp, mp4s, tmcd ...) */

  908.             url_fskip(pb, size - (url_ftell(pb) - start_pos));

  909.         }

  910.         /* this will read extra atoms at the end (wave, alac, damr, avcC, SMI ...) */

  911.         a.size = size - (url_ftell(pb) - start_pos);

  912.         if (a.size > 8) {

  913.             if (mov_read_default(c, pb, a) < 0)

  914.                 return -1;

  915.         } else if (a.size > 0)

  916.             url_fskip(pb, a.size);

  917.     }

  918. 

  919.     if(st->codec->codec_type==CODEC_TYPE_AUDIO && st->codec->sample_rate==0 && sc->time_scale>1)

  920.         st->codec->sample_rate= sc->time_scale;

  921. 

  922.     /* special codec parameters handling */

  923.     switch (st->codec->codec_id) {

  924. #if CONFIG_DV_DEMUXER

  925.     case CODEC_ID_DVAUDIO:

  926.         c->dv_fctx = avformat_alloc_context();

  927.         c->dv_demux = dv_init_demux(c->dv_fctx);

  928.         if (!c->dv_demux) {

  929.             av_log(c->fc, AV_LOG_ERROR, "dv demux context init error\n");

  930.             return -1;

  931.         }

  932.         sc->dv_audio_container = 1;

  933.         st->codec->codec_id = CODEC_ID_PCM_S16LE;

  934.         break;

  935. #endif

  936.     /* no ifdef since parameters are always those */

  937.     case CODEC_ID_QCELP:

  938.         st->codec->frame_size= 160;

  939.         st->codec->channels= 1; /* really needed */

  940.         break;

  941.     case CODEC_ID_AMR_NB:

  942.     case CODEC_ID_AMR_WB:

  943.         st->codec->frame_size= sc->samples_per_frame;

  944.         st->codec->channels= 1; /* really needed */

  945.         /* force sample rate for amr, stsd in 3gp does not store sample rate */

  946.         if (st->codec->codec_id == CODEC_ID_AMR_NB)

  947.             st->codec->sample_rate = 8000;

  948.         else if (st->codec->codec_id == CODEC_ID_AMR_WB)

  949.             st->codec->sample_rate = 16000;

  950.         break;

  951.     case CODEC_ID_MP2:

  952.     case CODEC_ID_MP3:

  953.         st->codec->codec_type = CODEC_TYPE_AUDIO; /* force type after stsd for m1a hdlr */

  954.         st->need_parsing = AVSTREAM_PARSE_FULL;

  955.         break;

  956.     case CODEC_ID_GSM:

  957.     case CODEC_ID_ADPCM_MS:

  958.     case CODEC_ID_ADPCM_IMA_WAV:

  959.         st->codec->block_align = sc->bytes_per_frame;

  960.         break;

  961.     case CODEC_ID_ALAC:

  962.         if (st->codec->extradata_size == 36) {

  963.             st->codec->frame_size = AV_RB32(st->codec->extradata+12);

  964.             st->codec->channels   = AV_RB8 (st->codec->extradata+21);

  965.         }

  966.         break;

  967.     default:

  968.         break;

  969.     }

  970. 

  971.     return 0;

  972. }

  973. 

  974. static int mov_read_stsc(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  975. {

  976.     AVStream *st = c->fc->streams[c->fc->nb_streams-1];

  977.     MOVStreamContext *sc = st->priv_data;

  978.     unsigned int i, entries;

  979. 

  980.     get_byte(pb); /* version */

  981.     get_be24(pb); /* flags */

  982. 

  983.     entries = get_be32(pb);

  984. 

  985.     dprintf(c->fc, "track[%i].stsc.entries = %i\n", c->fc->nb_streams-1, entries);

  986. 

  987.     if(entries >= UINT_MAX / sizeof(*sc->stsc_data))

  988.         return -1;

  989.     sc->stsc_data = av_malloc(entries * sizeof(*sc->stsc_data));

  990.     if (!sc->stsc_data)

  991.         return AVERROR(ENOMEM);

  992.     sc->stsc_count = entries;

  993. 

  994.     for(i=0; i<entries; i++) {

  995.         sc->stsc_data[i].first = get_be32(pb);

  996.         sc->stsc_data[i].count = get_be32(pb);

  997.         sc->stsc_data[i].id = get_be32(pb);

  998.     }

  999.     return 0;

  1000. }

  1001. 

  1002. static int mov_read_stss(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  1003. {

  1004.     AVStream *st = c->fc->streams[c->fc->nb_streams-1];

  1005.     MOVStreamContext *sc = st->priv_data;

  1006.     unsigned int i, entries;

  1007. 

  1008.     get_byte(pb); /* version */

  1009.     get_be24(pb); /* flags */

  1010. 

  1011.     entries = get_be32(pb);

  1012. 

  1013.     dprintf(c->fc, "keyframe_count = %d\n", entries);

  1014. 

  1015.     if(entries >= UINT_MAX / sizeof(int))

  1016.         return -1;

  1017.     sc->keyframes = av_malloc(entries * sizeof(int));

  1018.     if (!sc->keyframes)

  1019.         return AVERROR(ENOMEM);

  1020.     sc->keyframe_count = entries;

  1021. 

  1022.     for(i=0; i<entries; i++) {

  1023.         sc->keyframes[i] = get_be32(pb);

  1024.         //dprintf(c->fc, "keyframes[]=%d\n", sc->keyframes[i]);

  1025.     }

  1026.     return 0;

  1027. }

  1028. 

  1029. static int mov_read_stsz(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  1030. {

  1031.     AVStream *st = c->fc->streams[c->fc->nb_streams-1];

  1032.     MOVStreamContext *sc = st->priv_data;

  1033.     unsigned int i, entries, sample_size;

  1034. 

  1035.     get_byte(pb); /* version */

  1036.     get_be24(pb); /* flags */

  1037. 

  1038.     sample_size = get_be32(pb);

  1039.     if (!sc->sample_size) /* do not overwrite value computed in stsd */

  1040.         sc->sample_size = sample_size;

  1041.     entries = get_be32(pb);

  1042. 

  1043.     dprintf(c->fc, "sample_size = %d sample_count = %d\n", sc->sample_size, entries);

  1044. 

  1045.     sc->sample_count = entries;

  1046.     if (sample_size)

  1047.         return 0;

  1048. 

  1049.     if(entries >= UINT_MAX / sizeof(int))

  1050.         return -1;

  1051.     sc->sample_sizes = av_malloc(entries * sizeof(int));

  1052.     if (!sc->sample_sizes)

  1053.         return AVERROR(ENOMEM);

  1054. 

  1055.     for(i=0; i<entries; i++)

  1056.         sc->sample_sizes[i] = get_be32(pb);

  1057.     return 0;

  1058. }

  1059. 

  1060. static int mov_read_stts(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  1061. {

  1062.     AVStream *st = c->fc->streams[c->fc->nb_streams-1];

  1063.     MOVStreamContext *sc = st->priv_data;

  1064.     unsigned int i, entries;

  1065.     int64_t duration=0;

  1066.     int64_t total_sample_count=0;

  1067. 

  1068.     get_byte(pb); /* version */

  1069.     get_be24(pb); /* flags */

  1070.     entries = get_be32(pb);

  1071. 

  1072.     dprintf(c->fc, "track[%i].stts.entries = %i\n", c->fc->nb_streams-1, entries);

  1073. 

  1074.     if(entries >= UINT_MAX / sizeof(*sc->stts_data))

  1075.         return -1;

  1076.     sc->stts_data = av_malloc(entries * sizeof(*sc->stts_data));

  1077.     if (!sc->stts_data)

  1078.         return AVERROR(ENOMEM);

  1079.     sc->stts_count = entries;

  1080. 

  1081.     for(i=0; i<entries; i++) {

  1082.         int sample_duration;

  1083.         int sample_count;

  1084. 

  1085.         sample_count=get_be32(pb);

  1086.         sample_duration = get_be32(pb);

  1087.         sc->stts_data[i].count= sample_count;

  1088.         sc->stts_data[i].duration= sample_duration;

  1089. 

  1090.         sc->time_rate= av_gcd(sc->time_rate, sample_duration);

  1091. 

  1092.         dprintf(c->fc, "sample_count=%d, sample_duration=%d\n",sample_count,sample_duration);

  1093. 

  1094.         duration+=(int64_t)sample_duration*sample_count;

  1095.         total_sample_count+=sample_count;

  1096.     }

  1097. 

  1098.     st->nb_frames= total_sample_count;

  1099.     if(duration)

  1100.         st->duration= duration;

  1101.     return 0;

  1102. }

  1103. 

  1104. static int mov_read_ctts(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  1105. {

  1106.     AVStream *st = c->fc->streams[c->fc->nb_streams-1];

  1107.     MOVStreamContext *sc = st->priv_data;

  1108.     unsigned int i, entries;

  1109. 

  1110.     get_byte(pb); /* version */

  1111.     get_be24(pb); /* flags */

  1112.     entries = get_be32(pb);

  1113. 

  1114.     dprintf(c->fc, "track[%i].ctts.entries = %i\n", c->fc->nb_streams-1, entries);

  1115. 

  1116.     if(entries >= UINT_MAX / sizeof(*sc->ctts_data))

  1117.         return -1;

  1118.     sc->ctts_data = av_malloc(entries * sizeof(*sc->ctts_data));

  1119.     if (!sc->ctts_data)

  1120.         return AVERROR(ENOMEM);

  1121.     sc->ctts_count = entries;

  1122. 

  1123.     for(i=0; i<entries; i++) {

  1124.         int count    =get_be32(pb);

  1125.         int duration =get_be32(pb);

  1126. 

  1127.         if (duration < 0) {

  1128.             sc->wrong_dts = 1;

  1129.             st->codec->has_b_frames = 1;

  1130.         }

  1131.         sc->ctts_data[i].count   = count;

  1132.         sc->ctts_data[i].duration= duration;

  1133. 

  1134.         sc->time_rate= av_gcd(sc->time_rate, FFABS(duration));

  1135.     }

  1136.     return 0;

  1137. }

  1138. 

  1139. static void mov_build_index(MOVContext *mov, AVStream *st)

  1140. {

  1141.     MOVStreamContext *sc = st->priv_data;

  1142.     int64_t current_offset;

  1143.     int64_t current_dts = 0;

  1144.     unsigned int stts_index = 0;

  1145.     unsigned int stsc_index = 0;

  1146.     unsigned int stss_index = 0;

  1147.     unsigned int i, j;

  1148. 

  1149.     /* adjust first dts according to edit list */

  1150.     if (sc->time_offset) {

  1151.         assert(sc->time_offset % sc->time_rate == 0);

  1152.         current_dts = - (sc->time_offset / sc->time_rate);

  1153.     }

  1154. 

  1155.     /* only use old uncompressed audio chunk demuxing when stts specifies it */

  1156.     if (!(st->codec->codec_type == CODEC_TYPE_AUDIO &&

  1157.           sc->stts_count == 1 && sc->stts_data[0].duration == 1)) {

  1158.         unsigned int current_sample = 0;

  1159.         unsigned int stts_sample = 0;

  1160.         unsigned int keyframe, sample_size;

  1161.         unsigned int distance = 0;

  1162.         int key_off = sc->keyframes && sc->keyframes[0] == 1;

  1163. 

  1164.         st->nb_frames = sc->sample_count;

  1165.         for (i = 0; i < sc->chunk_count; i++) {

  1166.             current_offset = sc->chunk_offsets[i];

  1167.             if (stsc_index + 1 < sc->stsc_count &&

  1168.                 i + 1 == sc->stsc_data[stsc_index + 1].first)

  1169.                 stsc_index++;

  1170.             for (j = 0; j < sc->stsc_data[stsc_index].count; j++) {

  1171.                 if (current_sample >= sc->sample_count) {

  1172.                     av_log(mov->fc, AV_LOG_ERROR, "wrong sample count\n");

  1173.                     goto out;

  1174.                 }

  1175.                 keyframe = !sc->keyframe_count || current_sample+key_off == sc->keyframes[stss_index];

  1176.                 if (keyframe) {

  1177.                     distance = 0;

  1178.                     if (stss_index + 1 < sc->keyframe_count)

  1179.                         stss_index++;

  1180.                 }

  1181.                 sample_size = sc->sample_size > 0 ? sc->sample_size : sc->sample_sizes[current_sample];

  1182.                 if(sc->pseudo_stream_id == -1 ||

  1183.                    sc->stsc_data[stsc_index].id - 1 == sc->pseudo_stream_id) {

  1184.                     av_add_index_entry(st, current_offset, current_dts, sample_size, distance,

  1185.                                     keyframe ? AVINDEX_KEYFRAME : 0);

  1186.                     dprintf(mov->fc, "AVIndex stream %d, sample %d, offset %"PRIx64", dts %"PRId64", "

  1187.                             "size %d, distance %d, keyframe %d\n", st->index, current_sample,

  1188.                             current_offset, current_dts, sample_size, distance, keyframe);

  1189.                 }

  1190.                 current_offset += sample_size;

  1191.                 assert(sc->stts_data[stts_index].duration % sc->time_rate == 0);

  1192.                 current_dts += sc->stts_data[stts_index].duration / sc->time_rate;

  1193.                 distance++;

  1194.                 stts_sample++;

  1195.                 current_sample++;

  1196.                 if (stts_index + 1 < sc->stts_count && stts_sample == sc->stts_data[stts_index].count) {

  1197.                     stts_sample = 0;

  1198.                     stts_index++;

  1199.                 }

  1200.             }

  1201.         }

  1202.     } else { /* read whole chunk */

  1203.         unsigned int chunk_samples, chunk_size, chunk_duration;

  1204.         unsigned int frames = 1;

  1205.         for (i = 0; i < sc->chunk_count; i++) {

  1206.             current_offset = sc->chunk_offsets[i];

  1207.             if (stsc_index + 1 < sc->stsc_count &&

  1208.                 i + 1 == sc->stsc_data[stsc_index + 1].first)

  1209.                 stsc_index++;

  1210.             chunk_samples = sc->stsc_data[stsc_index].count;

  1211.             /* get chunk size, beware of alaw/ulaw/mace */

  1212.             if (sc->samples_per_frame > 0 &&

  1213.                 (chunk_samples * sc->bytes_per_frame % sc->samples_per_frame == 0)) {

  1214.                 if (sc->samples_per_frame < 160)

  1215.                     chunk_size = chunk_samples * sc->bytes_per_frame / sc->samples_per_frame;

  1216.                 else {

  1217.                     chunk_size = sc->bytes_per_frame;

  1218.                     frames = chunk_samples / sc->samples_per_frame;

  1219.                     chunk_samples = sc->samples_per_frame;

  1220.                 }

  1221.             } else

  1222.                 chunk_size = chunk_samples * sc->sample_size;

  1223.             for (j = 0; j < frames; j++) {

  1224.                 av_add_index_entry(st, current_offset, current_dts, chunk_size, 0, AVINDEX_KEYFRAME);

  1225.                 /* get chunk duration */

  1226.                 chunk_duration = 0;

  1227.                 while (chunk_samples > 0) {

  1228.                     if (chunk_samples < sc->stts_data[stts_index].count) {

  1229.                         chunk_duration += sc->stts_data[stts_index].duration * chunk_samples;

  1230.                         sc->stts_data[stts_index].count -= chunk_samples;

  1231.                         break;

  1232.                     } else {

  1233.                         chunk_duration += sc->stts_data[stts_index].duration * chunk_samples;

  1234.                         chunk_samples -= sc->stts_data[stts_index].count;

  1235.                         if (stts_index + 1 < sc->stts_count)

  1236.                             stts_index++;

  1237.                     }

  1238.                 }

  1239.                 current_offset += sc->bytes_per_frame;

  1240.                 dprintf(mov->fc, "AVIndex stream %d, chunk %d, offset %"PRIx64", dts %"PRId64", "

  1241.                         "size %d, duration %d\n", st->index, i, current_offset, current_dts,

  1242.                         chunk_size, chunk_duration);

  1243.                 assert(chunk_duration % sc->time_rate == 0);

  1244.                 current_dts += chunk_duration / sc->time_rate;

  1245.             }

  1246.         }

  1247.     }

  1248.  out:

  1249.     /* adjust sample count to avindex entries */

  1250.     sc->sample_count = st->nb_index_entries;

  1251. }

  1252. 

  1253. static int mov_read_trak(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  1254. {

  1255.     AVStream *st;

  1256.     MOVStreamContext *sc;

  1257.     int ret;

  1258. 

  1259.     st = av_new_stream(c->fc, c->fc->nb_streams);

  1260.     if (!st) return AVERROR(ENOMEM);

  1261.     sc = av_mallocz(sizeof(MOVStreamContext));

  1262.     if (!sc) return AVERROR(ENOMEM);

  1263. 

  1264.     st->priv_data = sc;

  1265.     st->codec->codec_type = CODEC_TYPE_DATA;

  1266.     sc->ffindex = st->index;

  1267. 

  1268.     if ((ret = mov_read_default(c, pb, atom)) < 0)

  1269.         return ret;

  1270. 

  1271.     /* sanity checks */

  1272.     if(sc->chunk_count && (!sc->stts_count || !sc->stsc_count ||

  1273.                            (!sc->sample_size && !sc->sample_count))){

  1274.         av_log(c->fc, AV_LOG_ERROR, "stream %d, missing mandatory atoms, broken header\n",

  1275.                st->index);

  1276.         sc->sample_count = 0; //ignore track

  1277.         return 0;

  1278.     }

  1279.     if(!sc->time_rate)

  1280.         sc->time_rate=1;

  1281.     if(!sc->time_scale)

  1282.         sc->time_scale= c->time_scale;

  1283.     av_set_pts_info(st, 64, sc->time_rate, sc->time_scale);

  1284. 

  1285.     if (st->codec->codec_type == CODEC_TYPE_AUDIO &&

  1286.         !st->codec->frame_size && sc->stts_count == 1) {

  1287.         st->codec->frame_size = av_rescale(sc->stts_data[0].duration,

  1288.                                            st->codec->sample_rate, sc->time_scale);

  1289.         dprintf(c->fc, "frame size %d\n", st->codec->frame_size);

  1290.     }

  1291. 

  1292.     if(st->duration != AV_NOPTS_VALUE){

  1293.         assert(st->duration % sc->time_rate == 0);

  1294.         st->duration /= sc->time_rate;

  1295.     }

  1296. 

  1297.     mov_build_index(c, st);

  1298. 

  1299.     if (sc->dref_id-1 < sc->drefs_count && sc->drefs[sc->dref_id-1].path) {

  1300.         if (url_fopen(&sc->pb, sc->drefs[sc->dref_id-1].path, URL_RDONLY) < 0)

  1301.             av_log(c->fc, AV_LOG_ERROR, "stream %d, error opening file %s: %s\n",

  1302.                    st->index, sc->drefs[sc->dref_id-1].path, strerror(errno));

  1303.     } else

  1304.         sc->pb = c->fc->pb;

  1305. 

  1306.     switch (st->codec->codec_id) {

  1307. #if CONFIG_H261_DECODER

  1308.     case CODEC_ID_H261:

  1309. #endif

  1310. #if CONFIG_H263_DECODER

  1311.     case CODEC_ID_H263:

  1312. #endif

  1313. #if CONFIG_MPEG4_DECODER

  1314.     case CODEC_ID_MPEG4:

  1315. #endif

  1316.         st->codec->width= 0; /* let decoder init width/height */

  1317.         st->codec->height= 0;

  1318.         break;

  1319.     }

  1320. 

  1321.     /* Do not need those anymore. */

  1322.     av_freep(&sc->chunk_offsets);

  1323.     av_freep(&sc->stsc_data);

  1324.     av_freep(&sc->sample_sizes);

  1325.     av_freep(&sc->keyframes);

  1326.     av_freep(&sc->stts_data);

  1327. 

  1328.     return 0;

  1329. }

  1330. 

  1331. static int mov_read_ilst(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  1332. {

  1333.     int ret;

  1334.     c->itunes_metadata = 1;

  1335.     ret = mov_read_default(c, pb, atom);

  1336.     c->itunes_metadata = 0;

  1337.     return ret;

  1338. }

  1339. 

  1340. static int mov_read_meta(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  1341. {

  1342.     url_fskip(pb, 4); // version + flags

  1343.     atom.size -= 4;

  1344.     return mov_read_default(c, pb, atom);

  1345. }

  1346. 

  1347. static int mov_read_trkn(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  1348. {

  1349.     char track[16];

  1350.     get_be32(pb); // type

  1351.     get_be32(pb); // unknown

  1352.     snprintf(track, sizeof(track), "%d", get_be32(pb));

  1353.     av_metadata_set(&c->fc->metadata, "track", track);

  1354.     dprintf(c->fc, "%.4s %s\n", (char*)&atom.type, track);

  1355.     return 0;

  1356. }

  1357. 

  1358. static int mov_read_udta_string(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  1359. {

  1360.     char str[1024], key2[16], language[4] = {0};

  1361.     const char *key = NULL;

  1362.     uint16_t str_size;

  1363. 

  1364.     if (c->itunes_metadata) {

  1365.         int data_size = get_be32(pb);

  1366.         int tag = get_le32(pb);

  1367.         if (tag == MKTAG('d','a','t','a')) {

  1368.             get_be32(pb); // type

  1369.             get_be32(pb); // unknown

  1370.             str_size = data_size - 16;

  1371.             atom.size -= 16;

  1372.         } else return 0;

  1373.     } else {

  1374.         str_size = get_be16(pb); // string length

  1375.         ff_mov_lang_to_iso639(get_be16(pb), language);

  1376.         atom.size -= 4;

  1377.     }

  1378.     switch (atom.type) {

  1379.     case MKTAG(0xa9,'n','a','m'): key = "title";     break;

  1380.     case MKTAG(0xa9,'a','u','t'):

  1381.     case MKTAG(0xa9,'A','R','T'):

  1382.     case MKTAG(0xa9,'w','r','t'): key = "author";    break;

  1383.     case MKTAG(0xa9,'c','p','y'): key = "copyright"; break;

  1384.     case MKTAG(0xa9,'c','m','t'):

  1385.     case MKTAG(0xa9,'i','n','f'): key = "comment";   break;

  1386.     case MKTAG(0xa9,'a','l','b'): key = "album";     break;

  1387.     case MKTAG(0xa9,'d','a','y'): key = "year";      break;

  1388.     case MKTAG(0xa9,'g','e','n'): key = "genre";     break;

  1389.     case MKTAG(0xa9,'t','o','o'):

  1390.     case MKTAG(0xa9,'e','n','c'): key = "muxer";     break;

  1391.     }

  1392.     if (!key)

  1393.         return 0;

  1394.     if (atom.size < 0)

  1395.         return -1;

  1396. 

  1397.     str_size = FFMIN3(sizeof(str)-1, str_size, atom.size);

  1398.     get_buffer(pb, str, str_size);

  1399.     str[str_size] = 0;

  1400.     av_metadata_set(&c->fc->metadata, key, str);

  1401.     if (*language && strcmp(language, "und")) {

  1402.         snprintf(key2, sizeof(key2), "%s-%s", key, language);

  1403.         av_metadata_set(&c->fc->metadata, key2, str);

  1404.     }

  1405.     dprintf(c->fc, "%.4s %s %d %lld\n", (char*)&atom.type, str, str_size, atom.size);

  1406.     return 0;

  1407. }

  1408. 

  1409. static int mov_read_tkhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  1410. {

  1411.     int i;

  1412.     int width;

  1413.     int height;

  1414.     int64_t disp_transform[2];

  1415.     int display_matrix[3][2];

  1416.     AVStream *st = c->fc->streams[c->fc->nb_streams-1];

  1417.     MOVStreamContext *sc = st->priv_data;

  1418.     int version = get_byte(pb);

  1419. 

  1420.     get_be24(pb); /* flags */

  1421.     /*

  1422.     MOV_TRACK_ENABLED 0x0001

  1423.     MOV_TRACK_IN_MOVIE 0x0002

  1424.     MOV_TRACK_IN_PREVIEW 0x0004

  1425.     MOV_TRACK_IN_POSTER 0x0008

  1426.     */

  1427. 

  1428.     if (version == 1) {

  1429.         get_be64(pb);

  1430.         get_be64(pb);

  1431.     } else {

  1432.         get_be32(pb); /* creation time */

  1433.         get_be32(pb); /* modification time */

  1434.     }

  1435.     st->id = (int)get_be32(pb); /* track id (NOT 0 !)*/

  1436.     get_be32(pb); /* reserved */

  1437. 

  1438.     /* highlevel (considering edits) duration in movie timebase */

  1439.     (version == 1) ? get_be64(pb) : get_be32(pb);

  1440.     get_be32(pb); /* reserved */

  1441.     get_be32(pb); /* reserved */

  1442. 

  1443.     get_be16(pb); /* layer */

  1444.     get_be16(pb); /* alternate group */

  1445.     get_be16(pb); /* volume */

  1446.     get_be16(pb); /* reserved */

  1447. 

  1448.     //read in the display matrix (outlined in ISO 14496-12, Section 6.2.2)

  1449.     // they're kept in fixed point format through all calculations

  1450.     // ignore u,v,z b/c we don't need the scale factor to calc aspect ratio

  1451.     for (i = 0; i < 3; i++) {

  1452.         display_matrix[i][0] = get_be32(pb);   // 16.16 fixed point

  1453.         display_matrix[i][1] = get_be32(pb);   // 16.16 fixed point

  1454.         get_be32(pb);           // 2.30 fixed point (not used)

  1455.     }

  1456. 

  1457.     width = get_be32(pb);       // 16.16 fixed point track width

  1458.     height = get_be32(pb);      // 16.16 fixed point track height

  1459.     sc->width = width >> 16;

  1460.     sc->height = height >> 16;

  1461. 

  1462.     //transform the display width/height according to the matrix

  1463.     // skip this if the display matrix is the default identity matrix

  1464.     // to keep the same scale, use [width height 1<<16]

  1465.     if (width && height &&

  1466.         (display_matrix[0][0] != 65536 || display_matrix[0][1]           ||

  1467.         display_matrix[1][0]           || display_matrix[1][1] != 65536  ||

  1468.         display_matrix[2][0]           || display_matrix[2][1])) {

  1469.         for (i = 0; i < 2; i++)

  1470.             disp_transform[i] =

  1471.                 (int64_t)  width  * display_matrix[0][i] +

  1472.                 (int64_t)  height * display_matrix[1][i] +

  1473.                 ((int64_t) display_matrix[2][i] << 16);

  1474. 

  1475.         //sample aspect ratio is new width/height divided by old width/height

  1476.         st->sample_aspect_ratio = av_d2q(

  1477.             ((double) disp_transform[0] * height) /

  1478.             ((double) disp_transform[1] * width), INT_MAX);

  1479.     }

  1480.     return 0;

  1481. }

  1482. 

  1483. static int mov_read_tfhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  1484. {

  1485.     MOVFragment *frag = &c->fragment;

  1486.     MOVTrackExt *trex = NULL;

  1487.     int flags, track_id, i;

  1488. 

  1489.     get_byte(pb); /* version */

  1490.     flags = get_be24(pb);

  1491. 

  1492.     track_id = get_be32(pb);

  1493.     if (!track_id)

  1494.         return -1;

  1495.     frag->track_id = track_id;

  1496.     for (i = 0; i < c->trex_count; i++)

  1497.         if (c->trex_data[i].track_id == frag->track_id) {

  1498.             trex = &c->trex_data[i];

  1499.             break;

  1500.         }

  1501.     if (!trex) {

  1502.         av_log(c->fc, AV_LOG_ERROR, "could not find corresponding trex\n");

  1503.         return -1;

  1504.     }

  1505. 

  1506.     if (flags & 0x01) frag->base_data_offset = get_be64(pb);

  1507.     else              frag->base_data_offset = frag->moof_offset;

  1508.     if (flags & 0x02) frag->stsd_id          = get_be32(pb);

  1509.     else              frag->stsd_id          = trex->stsd_id;

  1510. 

  1511.     frag->duration = flags & 0x08 ? get_be32(pb) : trex->duration;

  1512.     frag->size     = flags & 0x10 ? get_be32(pb) : trex->size;

  1513.     frag->flags    = flags & 0x20 ? get_be32(pb) : trex->flags;

  1514.     dprintf(c->fc, "frag flags 0x%x\n", frag->flags);

  1515.     return 0;

  1516. }

  1517. 

  1518. static int mov_read_trex(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  1519. {

  1520.     MOVTrackExt *trex;

  1521. 

  1522.     if ((uint64_t)c->trex_count+1 >= UINT_MAX / sizeof(*c->trex_data))

  1523.         return -1;

  1524.     trex = av_realloc(c->trex_data, (c->trex_count+1)*sizeof(*c->trex_data));

  1525.     if (!trex)

  1526.         return AVERROR(ENOMEM);

  1527.     c->trex_data = trex;

  1528.     trex = &c->trex_data[c->trex_count++];

  1529.     get_byte(pb); /* version */

  1530.     get_be24(pb); /* flags */

  1531.     trex->track_id = get_be32(pb);

  1532.     trex->stsd_id  = get_be32(pb);

  1533.     trex->duration = get_be32(pb);

  1534.     trex->size     = get_be32(pb);

  1535.     trex->flags    = get_be32(pb);

  1536.     return 0;

  1537. }

  1538. 

  1539. static int mov_read_trun(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  1540. {

  1541.     MOVFragment *frag = &c->fragment;

  1542.     AVStream *st = NULL;

  1543.     MOVStreamContext *sc;

  1544.     uint64_t offset;

  1545.     int64_t dts;

  1546.     int data_offset = 0;

  1547.     unsigned entries, first_sample_flags = frag->flags;

  1548.     int flags, distance, i;

  1549. 

  1550.     for (i = 0; i < c->fc->nb_streams; i++) {

  1551.         if (c->fc->streams[i]->id == frag->track_id) {

  1552.             st = c->fc->streams[i];

  1553.             break;

  1554.         }

  1555.     }

  1556.     if (!st) {

  1557.         av_log(c->fc, AV_LOG_ERROR, "could not find corresponding track id %d\n", frag->track_id);

  1558.         return -1;

  1559.     }

  1560.     sc = st->priv_data;

  1561.     if (sc->pseudo_stream_id+1 != frag->stsd_id)

  1562.         return 0;

  1563.     get_byte(pb); /* version */

  1564.     flags = get_be24(pb);

  1565.     entries = get_be32(pb);

  1566.     dprintf(c->fc, "flags 0x%x entries %d\n", flags, entries);

  1567.     if (flags & 0x001) data_offset        = get_be32(pb);

  1568.     if (flags & 0x004) first_sample_flags = get_be32(pb);

  1569.     if (flags & 0x800) {

  1570.         MOVStts *ctts_data;

  1571.         if ((uint64_t)entries+sc->ctts_count >= UINT_MAX/sizeof(*sc->ctts_data))

  1572.             return -1;

  1573.         ctts_data = av_realloc(sc->ctts_data,

  1574.                                (entries+sc->ctts_count)*sizeof(*sc->ctts_data));

  1575.         if (!ctts_data)

  1576.             return AVERROR(ENOMEM);

  1577.         sc->ctts_data = ctts_data;

  1578.     }

  1579.     dts = st->duration;

  1580.     offset = frag->base_data_offset + data_offset;

  1581.     distance = 0;

  1582.     dprintf(c->fc, "first sample flags 0x%x\n", first_sample_flags);

  1583.     for (i = 0; i < entries; i++) {

  1584.         unsigned sample_size = frag->size;

  1585.         int sample_flags = i ? frag->flags : first_sample_flags;

  1586.         unsigned sample_duration = frag->duration;

  1587.         int keyframe;

  1588. 

  1589.         if (flags & 0x100) sample_duration = get_be32(pb);

  1590.         if (flags & 0x200) sample_size     = get_be32(pb);

  1591.         if (flags & 0x400) sample_flags    = get_be32(pb);

  1592.         if (flags & 0x800) {

  1593.             sc->ctts_data[sc->ctts_count].count = 1;

  1594.             sc->ctts_data[sc->ctts_count].duration = get_be32(pb);

  1595.             sc->ctts_count++;

  1596.         }

  1597.         if ((keyframe = st->codec->codec_type == CODEC_TYPE_AUDIO ||

  1598.              (flags & 0x004 && !i && !sample_flags) || sample_flags & 0x2000000))

  1599.             distance = 0;

  1600.         av_add_index_entry(st, offset, dts, sample_size, distance,

  1601.                            keyframe ? AVINDEX_KEYFRAME : 0);

  1602.         dprintf(c->fc, "AVIndex stream %d, sample %d, offset %"PRIx64", dts %"PRId64", "

  1603.                 "size %d, distance %d, keyframe %d\n", st->index, sc->sample_count+i,

  1604.                 offset, dts, sample_size, distance, keyframe);

  1605.         distance++;

  1606.         assert(sample_duration % sc->time_rate == 0);

  1607.         dts += sample_duration / sc->time_rate;

  1608.         offset += sample_size;

  1609.     }

  1610.     frag->moof_offset = offset;

  1611.     sc->sample_count = st->nb_index_entries;

  1612.     st->duration = dts;

  1613.     return 0;

  1614. }

  1615. 

  1616. /* this atom should be null (from specs), but some buggy files put the 'moov' atom inside it... */

  1617. /* like the files created with Adobe Premiere 5.0, for samples see */

  1618. /* http://graphics.tudelft.nl/~wouter/publications/soundtests/ */

  1619. static int mov_read_wide(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  1620. {

  1621.     int err;

  1622. 

  1623.     if (atom.size < 8)

  1624.         return 0; /* continue */

  1625.     if (get_be32(pb) != 0) { /* 0 sized mdat atom... use the 'wide' atom size */

  1626.         url_fskip(pb, atom.size - 4);

  1627.         return 0;

  1628.     }

  1629.     atom.type = get_le32(pb);

  1630.     atom.offset += 8;

  1631.     atom.size -= 8;

  1632.     if (atom.type != MKTAG('m','d','a','t')) {

  1633.         url_fskip(pb, atom.size);

  1634.         return 0;

  1635.     }

  1636.     err = mov_read_mdat(c, pb, atom);

  1637.     return err;

  1638. }

  1639. 

  1640. static int mov_read_cmov(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  1641. {

  1642. #if CONFIG_ZLIB

  1643.     ByteIOContext ctx;

  1644.     uint8_t *cmov_data;

  1645.     uint8_t *moov_data; /* uncompressed data */

  1646.     long cmov_len, moov_len;

  1647.     int ret = -1;

  1648. 

  1649.     get_be32(pb); /* dcom atom */

  1650.     if (get_le32(pb) != MKTAG('d','c','o','m'))

  1651.         return -1;

  1652.     if (get_le32(pb) != MKTAG('z','l','i','b')) {

  1653.         av_log(c->fc, AV_LOG_ERROR, "unknown compression for cmov atom !");

  1654.         return -1;

  1655.     }

  1656.     get_be32(pb); /* cmvd atom */

  1657.     if (get_le32(pb) != MKTAG('c','m','v','d'))

  1658.         return -1;

  1659.     moov_len = get_be32(pb); /* uncompressed size */

  1660.     cmov_len = atom.size - 6 * 4;

  1661. 

  1662.     cmov_data = av_malloc(cmov_len);

  1663.     if (!cmov_data)

  1664.         return AVERROR(ENOMEM);

  1665.     moov_data = av_malloc(moov_len);

  1666.     if (!moov_data) {

  1667.         av_free(cmov_data);

  1668.         return AVERROR(ENOMEM);

  1669.     }

  1670.     get_buffer(pb, cmov_data, cmov_len);

  1671.     if(uncompress (moov_data, (uLongf *) &moov_len, (const Bytef *)cmov_data, cmov_len) != Z_OK)

  1672.         goto free_and_return;

  1673.     if(init_put_byte(&ctx, moov_data, moov_len, 0, NULL, NULL, NULL, NULL) != 0)

  1674.         goto free_and_return;

  1675.     atom.type = MKTAG('m','o','o','v');

  1676.     atom.offset = 0;

  1677.     atom.size = moov_len;

  1678. #ifdef DEBUG

  1679. //    { int fd = open("/tmp/uncompheader.mov", O_WRONLY | O_CREAT); write(fd, moov_data, moov_len); close(fd); }

  1680. #endif

  1681.     ret = mov_read_default(c, &ctx, atom);

  1682. free_and_return:

  1683.     av_free(moov_data);

  1684.     av_free(cmov_data);

  1685.     return ret;

  1686. #else

  1687.     av_log(c->fc, AV_LOG_ERROR, "this file requires zlib support compiled in\n");

  1688.     return -1;

  1689. #endif

  1690. }

  1691. 

  1692. /* edit list atom */

  1693. static int mov_read_elst(MOVContext *c, ByteIOContext *pb, MOVAtom atom)

  1694. {

  1695.     MOVStreamContext *sc = c->fc->streams[c->fc->nb_streams-1]->priv_data;

  1696.     int i, edit_count;

  1697. 

  1698.     get_byte(pb); /* version */

  1699.     get_be24(pb); /* flags */

  1700.     edit_count = get_be32(pb); /* entries */

  1701. 

  1702.     for(i=0; i<edit_count; i++){

  1703.         int time;

  1704.         get_be32(pb); /* Track duration */

  1705.         time = get_be32(pb); /* Media time */

  1706.         get_be32(pb); /* Media rate */

  1707.         if (i == 0 && time != -1) {

  1708.             sc->time_offset = time;

  1709.             sc->time_rate = av_gcd(sc->time_rate, time);

  1710.         }

  1711.     }

  1712. 

  1713.     if(edit_count > 1)

  1714.         av_log(c->fc, AV_LOG_WARNING, "multiple edit list entries, "

  1715.                "a/v desync might occur, patch welcome\n");

  1716. 

  1717.     dprintf(c->fc, "track[%i].edit_count = %i\n", c->fc->nb_streams-1, edit_count);

  1718.     return 0;

  1719. }

  1720. 

  1721. static const MOVParseTableEntry mov_default_parse_table[] = {

  1722. { MKTAG('a','v','s','s'), mov_read_extradata },

  1723. { MKTAG('c','o','6','4'), mov_read_stco },

  1724. { MKTAG('c','t','t','s'), mov_read_ctts }, /* composition time to sample */

  1725. { MKTAG('d','i','n','f'), mov_read_default },

  1726. { MKTAG('d','r','e','f'), mov_read_dref },

  1727. { MKTAG('e','d','t','s'), mov_read_default },

  1728. { MKTAG('e','l','s','t'), mov_read_elst },

  1729. { MKTAG('e','n','d','a'), mov_read_enda },

  1730. { MKTAG('f','i','e','l'), mov_read_extradata },

  1731. { MKTAG('f','t','y','p'), mov_read_ftyp },

  1732. { MKTAG('g','l','b','l'), mov_read_glbl },

  1733. { MKTAG('h','d','l','r'), mov_read_hdlr },

  1734. { MKTAG('i','l','s','t'), mov_read_ilst },

  1735. { MKTAG('j','p','2','h'), mov_read_extradata },

  1736. { MKTAG('m','d','a','t'), mov_read_mdat },

  1737. { MKTAG('m','d','h','d'), mov_read_mdhd },

  1738. { MKTAG('m','d','i','a'), mov_read_default },

  1739. { MKTAG('m','e','t','a'), mov_read_meta },

  1740. { MKTAG('m','i','n','f'), mov_read_default },

  1741. { MKTAG('m','o','o','f'), mov_read_moof },

  1742. { MKTAG('m','o','o','v'), mov_read_moov },

  1743. { MKTAG('m','v','e','x'), mov_read_default },

  1744. { MKTAG('m','v','h','d'), mov_read_mvhd },

  1745. { MKTAG('S','M','I',' '), mov_read_smi }, /* Sorenson extension ??? */

  1746. { MKTAG('a','l','a','c'), mov_read_extradata }, /* alac specific atom */

  1747. { MKTAG('a','v','c','C'), mov_read_glbl },

  1748. { MKTAG('p','a','s','p'), mov_read_pasp },

  1749. { MKTAG('s','t','b','l'), mov_read_default },

  1750. { MKTAG('s','t','c','o'), mov_read_stco },

  1751. { MKTAG('s','t','s','c'), mov_read_stsc },

  1752. { MKTAG('s','t','s','d'), mov_read_stsd }, /* sample description */

  1753. { MKTAG('s','t','s','s'), mov_read_stss }, /* sync sample */

  1754. { MKTAG('s','t','s','z'), mov_read_stsz }, /* sample size */

  1755. { MKTAG('s','t','t','s'), mov_read_stts },

  1756. { MKTAG('t','k','h','d'), mov_read_tkhd }, /* track header */

  1757. { MKTAG('t','f','h','d'), mov_read_tfhd }, /* track fragment header */

  1758. { MKTAG('t','r','a','k'), mov_read_trak },

  1759. { MKTAG('t','r','a','f'), mov_read_default },

  1760. { MKTAG('t','r','e','x'), mov_read_trex },

  1761. { MKTAG('t','r','k','n'), mov_read_trkn },

  1762. { MKTAG('t','r','u','n'), mov_read_trun },

  1763. { MKTAG('u','d','t','a'), mov_read_default },

  1764. { MKTAG('w','a','v','e'), mov_read_wave },

  1765. { MKTAG('e','s','d','s'), mov_read_esds },

  1766. { MKTAG('w','i','d','e'), mov_read_wide }, /* place holder */

  1767. { MKTAG('c','m','o','v'), mov_read_cmov },

  1768. { MKTAG(0xa9,'n','a','m'), mov_read_udta_string },

  1769. { MKTAG(0xa9,'w','r','t'), mov_read_udta_string },

  1770. { MKTAG(0xa9,'c','p','y'), mov_read_udta_string },

  1771. { MKTAG(0xa9,'i','n','f'), mov_read_udta_string },

  1772. { MKTAG(0xa9,'i','n','f'), mov_read_udta_string },

  1773. { MKTAG(0xa9,'A','R','T'), mov_read_udta_string },

  1774. { MKTAG(0xa9,'a','l','b'), mov_read_udta_string },

  1775. { MKTAG(0xa9,'c','m','t'), mov_read_udta_string },

  1776. { MKTAG(0xa9,'a','u','t'), mov_read_udta_string },

  1777. { MKTAG(0xa9,'d','a','y'), mov_read_udta_string },

  1778. { MKTAG(0xa9,'g','e','n'), mov_read_udta_string },

  1779. { MKTAG(0xa9,'e','n','c'), mov_read_udta_string },

  1780. { MKTAG(0xa9,'t','o','o'), mov_read_udta_string },

  1781. { 0, NULL }

  1782. };

  1783. 

  1784. static int mov_probe(AVProbeData *p)

  1785. {

  1786.     unsigned int offset;

  1787.     uint32_t tag;

  1788.     int score = 0;

  1789. 

  1790.     /* check file header */

  1791.     offset = 0;

  1792.     for(;;) {

  1793.         /* ignore invalid offset */

  1794.         if ((offset + 8) > (unsigned int)p->buf_size)

  1795.             return score;

  1796.         tag = AV_RL32(p->buf + offset + 4);

  1797.         switch(tag) {

  1798.         /* check for obvious tags */

  1799.         case MKTAG('j','P',' ',' '): /* jpeg 2000 signature */

  1800.         case MKTAG('m','o','o','v'):

  1801.         case MKTAG('m','d','a','t'):

  1802.         case MKTAG('p','n','o','t'): /* detect movs with preview pics like ew.mov and april.mov */

  1803.         case MKTAG('u','d','t','a'): /* Packet Video PVAuthor adds this and a lot of more junk */

  1804.         case MKTAG('f','t','y','p'):

  1805.             return AVPROBE_SCORE_MAX;

  1806.         /* those are more common words, so rate then a bit less */

  1807.         case MKTAG('e','d','i','w'): /* xdcam files have reverted first tags */

  1808.         case MKTAG('w','i','d','e'):

  1809.         case MKTAG('f','r','e','e'):

  1810.         case MKTAG('j','u','n','k'):

  1811.         case MKTAG('p','i','c','t'):

  1812.             return AVPROBE_SCORE_MAX - 5;

  1813.         case MKTAG(0x82,0x82,0x7f,0x7d):

  1814.         case MKTAG('s','k','i','p'):

  1815.         case MKTAG('u','u','i','d'):

  1816.         case MKTAG('p','r','f','l'):

  1817.             offset = AV_RB32(p->buf+offset) + offset;

  1818.             /* if we only find those cause probedata is too small at least rate them */

  1819.             score = AVPROBE_SCORE_MAX - 50;

  1820.             break;

  1821.         default:

  1822.             /* unrecognized tag */

  1823.             return score;

  1824.         }

  1825.     }

  1826.     return score;

  1827. }

  1828. 

  1829. static int mov_read_header(AVFormatContext *s, AVFormatParameters *ap)

  1830. {

  1831.     MOVContext *mov = s->priv_data;

  1832.     ByteIOContext *pb = s->pb;

  1833.     int err;

  1834.     MOVAtom atom = { 0, 0, 0 };

  1835. 

  1836.     mov->fc = s;

  1837.     /* .mov and .mp4 aren't streamable anyway (only progressive download if moov is before mdat) */

  1838.     if(!url_is_streamed(pb))

  1839.         atom.size = url_fsize(pb);

  1840.     else

  1841.         atom.size = INT64_MAX;

  1842. 

  1843.     /* check MOV header */

  1844.     if ((err = mov_read_default(mov, pb, atom)) < 0) {

  1845.         av_log(s, AV_LOG_ERROR, "error reading header: %d\n", err);

  1846.         return err;

  1847.     }

  1848.     if (!mov->found_moov) {

  1849.         av_log(s, AV_LOG_ERROR, "moov atom not found\n");

  1850.         return -1;

  1851.     }

  1852.     dprintf(mov->fc, "on_parse_exit_offset=%lld\n", url_ftell(pb));

  1853. 

  1854.     return 0;

  1855. }

  1856. 

  1857. static int mov_read_packet(AVFormatContext *s, AVPacket *pkt)

  1858. {

  1859.     MOVContext *mov = s->priv_data;

  1860.     MOVStreamContext *sc = 0;

  1861.     AVIndexEntry *sample = 0;

  1862.     int64_t best_dts = INT64_MAX;

  1863.     int i, ret;

  1864.  retry:

  1865.     for (i = 0; i < s->nb_streams; i++) {

  1866.         AVStream *st = s->streams[i];

  1867.         MOVStreamContext *msc = st->priv_data;

  1868.         if (st->discard != AVDISCARD_ALL && msc->pb && msc->current_sample < msc->sample_count) {

  1869.             AVIndexEntry *current_sample = &st->index_entries[msc->current_sample];

  1870.             int64_t dts = av_rescale(current_sample->timestamp * (int64_t)msc->time_rate,

  1871.                                      AV_TIME_BASE, msc->time_scale);

  1872.             dprintf(s, "stream %d, sample %d, dts %"PRId64"\n", i, msc->current_sample, dts);

  1873.             if (!sample || (url_is_streamed(s->pb) && current_sample->pos < sample->pos) ||

  1874.                 (!url_is_streamed(s->pb) &&

  1875.                  ((msc->pb != s->pb && dts < best_dts) || (msc->pb == s->pb &&

  1876.                  ((FFABS(best_dts - dts) <= AV_TIME_BASE && current_sample->pos < sample->pos) ||

  1877.                   (FFABS(best_dts - dts) > AV_TIME_BASE && dts < best_dts)))))) {

  1878.                 sample = current_sample;

  1879.                 best_dts = dts;

  1880.                 sc = msc;

  1881.             }

  1882.         }

  1883.     }

  1884.     if (!sample) {

  1885.         mov->found_mdat = 0;

  1886.         if (!url_is_streamed(s->pb) ||

  1887.             mov_read_default(mov, s->pb, (MOVAtom){ 0, 0, INT64_MAX }) < 0 ||

  1888.             url_feof(s->pb))

  1889.             return -1;

  1890.         dprintf(s, "read fragments, offset 0x%llx\n", url_ftell(s->pb));

  1891.         goto retry;

  1892.     }

  1893.     /* must be done just before reading, to avoid infinite loop on sample */

  1894.     sc->current_sample++;

  1895.     if (url_fseek(sc->pb, sample->pos, SEEK_SET) != sample->pos) {

  1896.         av_log(mov->fc, AV_LOG_ERROR, "stream %d, offset 0x%"PRIx64": partial file\n",

  1897.                sc->ffindex, sample->pos);

  1898.         return -1;

  1899.     }

  1900.     ret = av_get_packet(sc->pb, pkt, sample->size);

  1901.     if (ret < 0)

  1902.         return ret;

  1903. #if CONFIG_DV_DEMUXER

  1904.     if (mov->dv_demux && sc->dv_audio_container) {

  1905.         dv_produce_packet(mov->dv_demux, pkt, pkt->data, pkt->size);

  1906.         av_free(pkt->data);

  1907.         pkt->size = 0;

  1908.         if (dv_get_packet(mov->dv_demux, pkt) < 0)

  1909.             return -1;

  1910.     }

  1911. #endif

  1912.     pkt->stream_index = sc->ffindex;

  1913.     pkt->dts = sample->timestamp;

  1914.     if (sc->ctts_data) {

  1915.         assert(sc->ctts_data[sc->ctts_index].duration % sc->time_rate == 0);

  1916.         pkt->pts = pkt->dts + sc->ctts_data[sc->ctts_index].duration / sc->time_rate;

  1917.         /* update ctts context */

  1918.         sc->ctts_sample++;

  1919.         if (sc->ctts_index < sc->ctts_count &&

  1920.             sc->ctts_data[sc->ctts_index].count == sc->ctts_sample) {

  1921.             sc->ctts_index++;

  1922.             sc->ctts_sample = 0;

  1923.         }

  1924.         if (sc->wrong_dts)

  1925.             pkt->dts = AV_NOPTS_VALUE;

  1926.     } else {

  1927.         AVStream *st = s->streams[sc->ffindex];

  1928.         int64_t next_dts = (sc->current_sample < sc->sample_count) ?

  1929.             st->index_entries[sc->current_sample].timestamp : st->duration;

  1930.         pkt->duration = next_dts - pkt->dts;

  1931.         pkt->pts = pkt->dts;

  1932.     }

  1933.     pkt->flags |= sample->flags & AVINDEX_KEYFRAME ? PKT_FLAG_KEY : 0;

  1934.     pkt->pos = sample->pos;

  1935.     dprintf(s, "stream %d, pts %"PRId64", dts %"PRId64", pos 0x%"PRIx64", duration %d\n",

  1936.             pkt->stream_index, pkt->pts, pkt->dts, pkt->pos, pkt->duration);

  1937.     return 0;

  1938. }

  1939. 

  1940. static int mov_seek_stream(AVStream *st, int64_t timestamp, int flags)

  1941. {

  1942.     MOVStreamContext *sc = st->priv_data;

  1943.     int sample, time_sample;

  1944.     int i;

  1945. 

  1946.     sample = av_index_search_timestamp(st, timestamp, flags);

  1947.     dprintf(st->codec, "stream %d, timestamp %"PRId64", sample %d\n", st->index, timestamp, sample);

  1948.     if (sample < 0) /* not sure what to do */

  1949.         return -1;

  1950.     sc->current_sample = sample;

  1951.     dprintf(st->codec, "stream %d, found sample %d\n", st->index, sc->current_sample);

  1952.     /* adjust ctts index */

  1953.     if (sc->ctts_data) {

  1954.         time_sample = 0;

  1955.         for (i = 0; i < sc->ctts_count; i++) {

  1956.             int next = time_sample + sc->ctts_data[i].count;

  1957.             if (next > sc->current_sample) {

  1958.                 sc->ctts_index = i;

  1959.                 sc->ctts_sample = sc->current_sample - time_sample;

  1960.                 break;

  1961.             }

  1962.             time_sample = next;

  1963.         }

  1964.     }

  1965.     return sample;

  1966. }

  1967. 

  1968. static int mov_read_seek(AVFormatContext *s, int stream_index, int64_t sample_time, int flags)

  1969. {

  1970.     AVStream *st;

  1971.     int64_t seek_timestamp, timestamp;

  1972.     int sample;

  1973.     int i;

  1974. 

  1975.     if (stream_index >= s->nb_streams)

  1976.         return -1;

  1977.     if (sample_time < 0)

  1978.         sample_time = 0;

  1979. 

  1980.     st = s->streams[stream_index];

  1981.     sample = mov_seek_stream(st, sample_time, flags);

  1982.     if (sample < 0)

  1983.         return -1;

  1984. 

  1985.     /* adjust seek timestamp to found sample timestamp */

  1986.     seek_timestamp = st->index_entries[sample].timestamp;

  1987. 

  1988.     for (i = 0; i < s->nb_streams; i++) {

  1989.         st = s->streams[i];

  1990.         if (stream_index == i || st->discard == AVDISCARD_ALL)

  1991.             continue;

  1992. 

  1993.         timestamp = av_rescale_q(seek_timestamp, s->streams[stream_index]->time_base, st->time_base);

  1994.         mov_seek_stream(st, timestamp, flags);

  1995.     }

  1996.     return 0;

  1997. }

  1998. 

  1999. static int mov_read_close(AVFormatContext *s)

  2000. {

  2001.     int i, j;

  2002.     MOVContext *mov = s->priv_data;

  2003.     for(i=0; i<s->nb_streams; i++) {

  2004.         MOVStreamContext *sc = s->streams[i]->priv_data;

  2005.         av_freep(&sc->ctts_data);

  2006.         for (j=0; j<sc->drefs_count; j++)

  2007.             av_freep(&sc->drefs[j].path);

  2008.         av_freep(&sc->drefs);

  2009.         if (sc->pb && sc->pb != s->pb)

  2010.             url_fclose(sc->pb);

  2011.     }

  2012.     if(mov->dv_demux){

  2013.         for(i=0; i<mov->dv_fctx->nb_streams; i++){

  2014.             av_freep(&mov->dv_fctx->streams[i]->codec);

  2015.             av_freep(&mov->dv_fctx->streams[i]);

  2016.         }

  2017.         av_freep(&mov->dv_fctx);

  2018.         av_freep(&mov->dv_demux);

  2019.     }

  2020.     av_freep(&mov->trex_data);

  2021.     return 0;

  2022. }

  2023. 

  2024. AVInputFormat mov_demuxer = {

  2025.     "mov,mp4,m4a,3gp,3g2,mj2",

  2026.     NULL_IF_CONFIG_SMALL("QuickTime/MPEG-4/Motion JPEG 2000 format"),

  2027.     sizeof(MOVContext),

  2028.     mov_probe,

  2029.     mov_read_header,

  2030.     mov_read_packet,

  2031.     mov_read_close,

  2032.     mov_read_seek,

  2033. };