mov.c 71.0 KB
Newer Older
1
/*
2
 * MOV demuxer
3
 * Copyright (c) 2001 Fabrice Bellard
4
 * Copyright (c) 2009 Baptiste Coudurier <baptiste dot coudurier at gmail dot com>
5
 *
6 7 8
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
F
Fabrice Bellard 已提交
9 10
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
11
 * version 2.1 of the License, or (at your option) any later version.
12
 *
13
 * FFmpeg is distributed in the hope that it will be useful,
14
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
F
Fabrice Bellard 已提交
15 16
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
17
 *
F
Fabrice Bellard 已提交
18
 * You should have received a copy of the GNU Lesser General Public
19
 * License along with FFmpeg; if not, write to the Free Software
20
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21
 */
22 23

#include <limits.h>
24

B
Baptiste Coudurier 已提交
25
//#define DEBUG
26 27
//#define DEBUG_METADATA
//#define MOV_EXPORT_ALL_METADATA
28

29
#include "libavutil/intreadwrite.h"
30
#include "libavutil/avstring.h"
31
#include "avformat.h"
32
#include "riff.h"
33
#include "isom.h"
34 35
#include "libavcodec/mpeg4audio.h"
#include "libavcodec/mpegaudiodata.h"
36

37
#if CONFIG_ZLIB
38 39 40
#include <zlib.h>
#endif

41 42
/*
 * First version by Francois Revol revol@free.fr
43
 * Seek function by Gael Chardon gael.dev@4now.net
44
 *
45
 * Features and limitations:
46
 * - reads most of the QT files I have (at least the structure),
47 48
 *   Sample QuickTime files with mp3 audio can be found at: http://www.3ivx.com/showcase.html
 * - the code is quite ugly... maybe I won't do it recursive next time :-)
49
 *
50 51
 * Funny I didn't know about http://sourceforge.net/projects/qt-ffmpeg/
 * when coding this :) (it's a writer anyway)
52
 *
53 54 55
 * Reference documents:
 * http://www.geocities.com/xhelmboyx/quicktime/formats/qtm-layout.txt
 * Apple:
G
Gael Chardon 已提交
56
 *  http://developer.apple.com/documentation/QuickTime/QTFF/
57
 *  http://developer.apple.com/documentation/QuickTime/QTFF/qtff.pdf
58 59 60
 * QuickTime is a trademark of Apple (AFAIK :))
 */

61 62
#include "qtpalette.h"

G
Gael Chardon 已提交
63

64 65 66
#undef NDEBUG
#include <assert.h>

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

/* those functions parse an atom */
/* return code:
B
Baptiste Coudurier 已提交
71
  0: continue to parse next atom
D
Diego Biurrun 已提交
72
 <0: error occurred, exit
B
Baptiste Coudurier 已提交
73
*/
74 75
/* links atom IDs to parse functions */
typedef struct MOVParseTableEntry {
76
    uint32_t type;
77
    int (*parse)(MOVContext *ctx, ByteIOContext *pb, MOVAtom atom);
78 79
} MOVParseTableEntry;

80 81
static const MOVParseTableEntry mov_default_parse_table[];

B
Baptiste Coudurier 已提交
82 83 84 85 86 87 88 89 90 91 92 93 94
static int mov_metadata_trkn(MOVContext *c, ByteIOContext *pb, unsigned len)
{
    char buf[16];

    get_be16(pb); // unknown
    snprintf(buf, sizeof(buf), "%d", get_be16(pb));
    av_metadata_set(&c->fc->metadata, "track", buf);

    get_be16(pb); // total tracks

    return 0;
}

95 96 97 98 99 100 101 102
static int mov_read_udta_string(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
{
#ifdef MOV_EXPORT_ALL_METADATA
    char tmp_key[5];
#endif
    char str[1024], key2[16], language[4] = {0};
    const char *key = NULL;
    uint16_t str_size;
B
Baptiste Coudurier 已提交
103
    int (*parse)(MOVContext*, ByteIOContext*, unsigned) = NULL;
104 105 106 107 108 109 110 111 112 113 114 115 116 117

    switch (atom.type) {
    case MKTAG(0xa9,'n','a','m'): key = "title";     break;
    case MKTAG(0xa9,'a','u','t'):
    case MKTAG(0xa9,'A','R','T'):
    case MKTAG(0xa9,'w','r','t'): key = "author";    break;
    case MKTAG(0xa9,'c','p','y'): key = "copyright"; break;
    case MKTAG(0xa9,'c','m','t'):
    case MKTAG(0xa9,'i','n','f'): key = "comment";   break;
    case MKTAG(0xa9,'a','l','b'): key = "album";     break;
    case MKTAG(0xa9,'d','a','y'): key = "year";      break;
    case MKTAG(0xa9,'g','e','n'): key = "genre";     break;
    case MKTAG(0xa9,'t','o','o'):
    case MKTAG(0xa9,'e','n','c'): key = "muxer";     break;
B
Baptiste Coudurier 已提交
118 119
    case MKTAG( 't','r','k','n'): key = "track";
        parse = mov_metadata_trkn; break;
120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150
    }

    if (c->itunes_metadata && atom.size > 8) {
        int data_size = get_be32(pb);
        int tag = get_le32(pb);
        if (tag == MKTAG('d','a','t','a')) {
            get_be32(pb); // type
            get_be32(pb); // unknown
            str_size = data_size - 16;
            atom.size -= 16;
        } else return 0;
    } else if (atom.size > 4 && key && !c->itunes_metadata) {
        str_size = get_be16(pb); // string length
        ff_mov_lang_to_iso639(get_be16(pb), language);
        atom.size -= 4;
    } else
        str_size = atom.size;

#ifdef MOV_EXPORT_ALL_METADATA
    if (!key) {
        snprintf(tmp_key, 5, "%.4s", (char*)&atom.type);
        key = tmp_key;
    }
#endif

    if (!key)
        return 0;
    if (atom.size < 0)
        return -1;

    str_size = FFMIN3(sizeof(str)-1, str_size, atom.size);
B
Baptiste Coudurier 已提交
151 152 153 154

    if (parse)
        parse(c, pb, str_size);
    else {
B
Baptiste Coudurier 已提交
155 156 157 158 159 160 161
        get_buffer(pb, str, str_size);
        str[str_size] = 0;
        av_metadata_set(&c->fc->metadata, key, str);
        if (*language && strcmp(language, "und")) {
            snprintf(key2, sizeof(key2), "%s-%s", key, language);
            av_metadata_set(&c->fc->metadata, key2, str);
        }
B
Baptiste Coudurier 已提交
162
    }
163 164 165 166 167 168 169 170
#ifdef DEBUG_METADATA
    av_log(c->fc, AV_LOG_DEBUG, "lang \"%3s\" ", language);
    av_log(c->fc, AV_LOG_DEBUG, "tag \"%s\" value \"%s\" atom \"%.4s\" %d %lld\n",
           key, str, (char*)&atom.type, str_size, atom.size);
#endif

    return 0;
}
171

172
static int mov_read_default(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
173
{
174
    int64_t total_size = 0;
175
    MOVAtom a;
176 177
    int i;
    int err = 0;
178 179

    a.offset = atom.offset;
180

181
    if (atom.size < 0)
B
Baptiste Coudurier 已提交
182
        atom.size = INT64_MAX;
183
    while(((total_size + 8) < atom.size) && !url_feof(pb) && !err) {
184
        int (*parse)(MOVContext*, ByteIOContext*, MOVAtom) = NULL;
185
        a.size = atom.size;
B
Baptiste Coudurier 已提交
186
        a.type=0;
187
        if(atom.size >= 8) {
188
            a.size = get_be32(pb);
189
            a.type = get_le32(pb);
190
        }
191
        total_size += 8;
192
        a.offset += 8;
193 194
        dprintf(c->fc, "type: %08x  %.4s  sz: %"PRIx64"  %"PRIx64"   %"PRIx64"\n",
                a.type, (char*)&a.type, a.size, atom.size, total_size);
195
        if (a.size == 1) { /* 64 bit extended size */
196
            a.size = get_be64(pb) - 8;
197 198
            a.offset += 8;
            total_size += 8;
199
        }
200 201 202
        if (a.size == 0) {
            a.size = atom.size - total_size;
            if (a.size <= 8)
203
                break;
204 205
        }
        a.size -= 8;
206
        if(a.size < 0)
207
            break;
B
Baptiste Coudurier 已提交
208
        a.size = FFMIN(a.size, atom.size - total_size);
209

210 211 212 213 214
        for (i = 0; mov_default_parse_table[i].type; i++)
            if (mov_default_parse_table[i].type == a.type) {
                parse = mov_default_parse_table[i].parse;
                break;
            }
215

216 217 218 219 220 221
        // container is user data
        if (!parse && (atom.type == MKTAG('u','d','t','a') ||
                       atom.type == MKTAG('i','l','s','t')))
            parse = mov_read_udta_string;

        if (!parse) { /* skip leaf atoms data */
222
            url_fskip(pb, a.size);
223
        } else {
224
            int64_t start_pos = url_ftell(pb);
225
            int64_t left;
226
            err = parse(c, pb, a);
227
            if (url_is_streamed(pb) && c->found_moov && c->found_mdat)
B
Baptiste Coudurier 已提交
228
                break;
229 230 231
            left = a.size - url_ftell(pb) + start_pos;
            if (left > 0) /* skip garbage at atom end */
                url_fskip(pb, left);
232
        }
233

234
        a.offset += a.size;
235
        total_size += a.size;
236 237
    }

B
Baptiste Coudurier 已提交
238
    if (!err && total_size < atom.size && atom.size < 0x7ffff)
239
        url_fskip(pb, atom.size - total_size);
240

241 242 243
    return err;
}

244
static int mov_read_dref(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
245 246 247 248 249 250 251 252 253 254
{
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];
    MOVStreamContext *sc = st->priv_data;
    int entries, i, j;

    get_be32(pb); // version + flags
    entries = get_be32(pb);
    if (entries >= UINT_MAX / sizeof(*sc->drefs))
        return -1;
    sc->drefs = av_mallocz(entries * sizeof(*sc->drefs));
255 256 257
    if (!sc->drefs)
        return AVERROR(ENOMEM);
    sc->drefs_count = entries;
258 259

    for (i = 0; i < sc->drefs_count; i++) {
260
        MOVDref *dref = &sc->drefs[i];
261
        uint32_t size = get_be32(pb);
262
        int64_t next = url_ftell(pb) + size - 4;
263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290

        dref->type = get_le32(pb);
        get_be32(pb); // version + flags
        dprintf(c->fc, "type %.4s size %d\n", (char*)&dref->type, size);

        if (dref->type == MKTAG('a','l','i','s') && size > 150) {
            /* macintosh alias record */
            uint16_t volume_len, len;
            char volume[28];
            int16_t type;

            url_fskip(pb, 10);

            volume_len = get_byte(pb);
            volume_len = FFMIN(volume_len, 27);
            get_buffer(pb, volume, 27);
            volume[volume_len] = 0;
            av_log(c->fc, AV_LOG_DEBUG, "volume %s, len %d\n", volume, volume_len);

            url_fskip(pb, 112);

            for (type = 0; type != -1 && url_ftell(pb) < next; ) {
                type = get_be16(pb);
                len = get_be16(pb);
                av_log(c->fc, AV_LOG_DEBUG, "type %d, len %d\n", type, len);
                if (len&1)
                    len += 1;
                if (type == 2) { // absolute path
291
                    av_free(dref->path);
292
                    dref->path = av_mallocz(len+1);
293 294
                    if (!dref->path)
                        return AVERROR(ENOMEM);
295
                    get_buffer(pb, dref->path, len);
296
                    if (len > volume_len && !strncmp(dref->path, volume, volume_len)) {
297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313
                        len -= volume_len;
                        memmove(dref->path, dref->path+volume_len, len);
                        dref->path[len] = 0;
                    }
                    for (j = 0; j < len; j++)
                        if (dref->path[j] == ':')
                            dref->path[j] = '/';
                    av_log(c->fc, AV_LOG_DEBUG, "path %s\n", dref->path);
                } else
                    url_fskip(pb, len);
            }
        }
        url_fseek(pb, next, SEEK_SET);
    }
    return 0;
}

314
static int mov_read_hdlr(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
315
{
316
    AVStream *st;
317 318
    uint32_t type;
    uint32_t ctype;
319

320 321 322 323 324
    if (c->fc->nb_streams < 1) // meta before first trak
        return 0;

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

325
    get_byte(pb); /* version */
326
    get_be24(pb); /* flags */
327 328 329 330 331

    /* component type */
    ctype = get_le32(pb);
    type = get_le32(pb); /* component subtype */

B
Baptiste Coudurier 已提交
332 333
    dprintf(c->fc, "ctype= %.4s (0x%08x)\n", (char*)&ctype, ctype);
    dprintf(c->fc, "stype= %.4s\n", (char*)&type);
334

335
    if     (type == MKTAG('v','i','d','e'))
336
        st->codec->codec_type = CODEC_TYPE_VIDEO;
337
    else if(type == MKTAG('s','o','u','n'))
338
        st->codec->codec_type = CODEC_TYPE_AUDIO;
339
    else if(type == MKTAG('m','1','a',' '))
340
        st->codec->codec_id = CODEC_ID_MP2;
341
    else if(type == MKTAG('s','u','b','p')) {
342 343
        st->codec->codec_type = CODEC_TYPE_SUBTITLE;
    }
344 345 346 347
    get_be32(pb); /* component  manufacture */
    get_be32(pb); /* component flags */
    get_be32(pb); /* component flags mask */

348
    if(atom.size <= 24)
349
        return 0; /* nothing left to read */
350

351
    url_fskip(pb, atom.size - (url_ftell(pb) - atom.offset));
352 353 354
    return 0;
}

B
Baptiste Coudurier 已提交
355
static int mp4_read_descr_len(ByteIOContext *pb)
356
{
357
    int len = 0;
358 359
    int count = 4;
    while (count--) {
360
        int c = get_byte(pb);
361 362 363
        len = (len << 7) | (c & 0x7f);
        if (!(c & 0x80))
            break;
364 365 366 367
    }
    return len;
}

B
Baptiste Coudurier 已提交
368
static int mp4_read_descr(MOVContext *c, ByteIOContext *pb, int *tag)
369 370 371
{
    int len;
    *tag = get_byte(pb);
B
Baptiste Coudurier 已提交
372
    len = mp4_read_descr_len(pb);
M
Michel Bardiaux 已提交
373
    dprintf(c->fc, "MPEG4 description: tag=0x%02x len=%d\n", *tag, len);
374 375 376
    return len;
}

377 378 379 380
#define MP4ESDescrTag                   0x03
#define MP4DecConfigDescrTag            0x04
#define MP4DecSpecificDescrTag          0x05

381 382 383 384 385 386 387 388
static const AVCodecTag mp4_audio_types[] = {
    { CODEC_ID_MP3ON4, 29 }, /* old mp3on4 draft */
    { CODEC_ID_MP3ON4, 32 }, /* layer 1 */
    { CODEC_ID_MP3ON4, 33 }, /* layer 2 */
    { CODEC_ID_MP3ON4, 34 }, /* layer 3 */
    { CODEC_ID_NONE,    0 },
};

389
static int mov_read_esds(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
390 391 392
{
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];
    int tag, len;
393

394
    get_be32(pb); /* version + flags */
B
Baptiste Coudurier 已提交
395
    len = mp4_read_descr(c, pb, &tag);
396
    if (tag == MP4ESDescrTag) {
397 398
        get_be16(pb); /* ID */
        get_byte(pb); /* priority */
399
    } else
400
        get_be16(pb); /* ID */
401

B
Baptiste Coudurier 已提交
402
    len = mp4_read_descr(c, pb, &tag);
403
    if (tag == MP4DecConfigDescrTag) {
404 405 406 407 408 409 410 411
        int object_type_id = get_byte(pb);
        get_byte(pb); /* stream type */
        get_be24(pb); /* buffer size db */
        get_be32(pb); /* max bitrate */
        get_be32(pb); /* avg bitrate */

        st->codec->codec_id= codec_get_id(ff_mp4_obj_type, object_type_id);
        dprintf(c->fc, "esds object type id %d\n", object_type_id);
B
Baptiste Coudurier 已提交
412
        len = mp4_read_descr(c, pb, &tag);
413
        if (tag == MP4DecSpecificDescrTag) {
M
Michel Bardiaux 已提交
414
            dprintf(c->fc, "Specific MPEG4 header len=%d\n", len);
B
Baptiste Coudurier 已提交
415 416
            if((uint64_t)len > (1<<30))
                return -1;
B
Baptiste Coudurier 已提交
417
            st->codec->extradata = av_mallocz(len + FF_INPUT_BUFFER_PADDING_SIZE);
418 419
            if (!st->codec->extradata)
                return AVERROR(ENOMEM);
B
Baptiste Coudurier 已提交
420 421
            get_buffer(pb, st->codec->extradata, len);
            st->codec->extradata_size = len;
422 423 424 425
            if (st->codec->codec_id == CODEC_ID_AAC) {
                MPEG4AudioConfig cfg;
                ff_mpeg4audio_get_config(&cfg, st->codec->extradata,
                                         st->codec->extradata_size);
426
                if (cfg.chan_config > 7)
427 428 429 430 431 432 433 434 435 436
                    return -1;
                st->codec->channels = ff_mpeg4audio_channels[cfg.chan_config];
                if (cfg.object_type == 29 && cfg.sampling_index < 3) // old mp3on4
                    st->codec->sample_rate = ff_mpa_freq_tab[cfg.sampling_index];
                else
                    st->codec->sample_rate = cfg.sample_rate; // ext sample rate ?
                dprintf(c->fc, "mp4a config channels %d obj %d ext obj %d "
                        "sample rate %d ext sample rate %d\n", st->codec->channels,
                        cfg.object_type, cfg.ext_object_type,
                        cfg.sample_rate, cfg.ext_sample_rate);
437 438 439
                if (!(st->codec->codec_id = codec_get_id(mp4_audio_types,
                                                         cfg.object_type)))
                    st->codec->codec_id = CODEC_ID_AAC;
B
Baptiste Coudurier 已提交
440
            }
441
        }
442 443 444 445
    }
    return 0;
}

446 447 448 449 450 451
static int mov_read_pasp(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
{
    const int num = get_be32(pb);
    const int den = get_be32(pb);
    AVStream * const st = c->fc->streams[c->fc->nb_streams-1];
    if (den != 0) {
452 453
        if ((st->sample_aspect_ratio.den != 1 || st->sample_aspect_ratio.num) && // default
            (den != st->sample_aspect_ratio.den || num != st->sample_aspect_ratio.num))
454
            av_log(c->fc, AV_LOG_WARNING,
455 456
                   "sample aspect ratio already set to %d:%d, overriding by 'pasp' atom\n",
                   st->sample_aspect_ratio.num, st->sample_aspect_ratio.den);
457 458 459 460 461 462
        st->sample_aspect_ratio.num = num;
        st->sample_aspect_ratio.den = den;
    }
    return 0;
}

463
/* this atom contains actual media data */
464
static int mov_read_mdat(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
465
{
466 467 468 469 470 471
    if(atom.size == 0) /* wrong one (MP4) */
        return 0;
    c->found_mdat=1;
    return 0; /* now go for moov */
}

472
static int mov_read_ftyp(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
473 474 475
{
    uint32_t type = get_le32(pb);

476 477
    if (type != MKTAG('q','t',' ',' '))
        c->isom = 1;
478
    av_log(c->fc, AV_LOG_DEBUG, "ISO: File Type Major Brand: %.4s\n",(char *)&type);
479 480 481 482 483
    get_be32(pb); /* minor version */
    url_fskip(pb, atom.size - 8);
    return 0;
}

484
/* this atom should contain all header atoms */
485
static int mov_read_moov(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
486
{
487 488
    if (mov_read_default(c, pb, atom) < 0)
        return -1;
489 490 491 492 493 494
    /* we parsed the 'moov' atom, we can terminate the parsing as soon as we find the 'mdat' */
    /* so we don't parse the whole file if over a network */
    c->found_moov=1;
    return 0; /* now go for mdat */
}

495
static int mov_read_moof(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
496 497 498 499 500
{
    c->fragment.moof_offset = url_ftell(pb) - 8;
    dprintf(c->fc, "moof offset %llx\n", c->fragment.moof_offset);
    return mov_read_default(c, pb, atom);
}
501

502
static int mov_read_mdhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
503
{
B
clean  
Baptiste Coudurier 已提交
504
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];
505
    MOVStreamContext *sc = st->priv_data;
B
clean  
Baptiste Coudurier 已提交
506
    int version = get_byte(pb);
507
    char language[4] = {0};
508
    unsigned lang;
509

510
    if (version > 1)
B
Baptiste Coudurier 已提交
511
        return -1; /* unsupported */
512

513
    get_be24(pb); /* flags */
B
clean  
Baptiste Coudurier 已提交
514 515 516 517 518 519 520
    if (version == 1) {
        get_be64(pb);
        get_be64(pb);
    } else {
        get_be32(pb); /* creation time */
        get_be32(pb); /* modification time */
    }
521

B
clean  
Baptiste Coudurier 已提交
522 523
    sc->time_scale = get_be32(pb);
    st->duration = (version == 1) ? get_be64(pb) : get_be32(pb); /* duration */
524

525
    lang = get_be16(pb); /* language */
526 527
    if (ff_mov_lang_to_iso639(lang, language))
        av_metadata_set(&st->metadata, "language", language);
528 529 530 531 532
    get_be16(pb); /* quality */

    return 0;
}

533
static int mov_read_mvhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
534
{
B
Baptiste Coudurier 已提交
535
    int version = get_byte(pb); /* version */
536
    get_be24(pb); /* flags */
537

B
Baptiste Coudurier 已提交
538 539 540 541 542 543 544
    if (version == 1) {
        get_be64(pb);
        get_be64(pb);
    } else {
        get_be32(pb); /* creation time */
        get_be32(pb); /* modification time */
    }
545
    c->time_scale = get_be32(pb); /* time scale */
546 547 548

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

B
Baptiste Coudurier 已提交
549
    c->duration = (version == 1) ? get_be64(pb) : get_be32(pb); /* duration */
550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568
    get_be32(pb); /* preferred scale */

    get_be16(pb); /* preferred volume */

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

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

    get_be32(pb); /* preview time */
    get_be32(pb); /* preview duration */
    get_be32(pb); /* poster time */
    get_be32(pb); /* selection time */
    get_be32(pb); /* selection duration */
    get_be32(pb); /* current time */
    get_be32(pb); /* next track ID */

    return 0;
}

569
static int mov_read_smi(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
570 571 572
{
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];

573 574
    if((uint64_t)atom.size > (1<<30))
        return -1;
575

576 577
    // currently SVQ3 decoder expect full STSD header - so let's fake it
    // this should be fixed and just SMI header should be passed
578
    av_free(st->codec->extradata);
579 580 581
    st->codec->extradata = av_mallocz(atom.size + 0x5a + FF_INPUT_BUFFER_PADDING_SIZE);
    if (!st->codec->extradata)
        return AVERROR(ENOMEM);
B
Baptiste Coudurier 已提交
582 583 584 585
    st->codec->extradata_size = 0x5a + atom.size;
    memcpy(st->codec->extradata, "SVQ3", 4); // fake
    get_buffer(pb, st->codec->extradata + 0x5a, atom.size);
    dprintf(c->fc, "Reading SMI %"PRId64"  %s\n", atom.size, st->codec->extradata + 0x5a);
586 587
    return 0;
}
588

589
static int mov_read_enda(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
590 591 592 593
{
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];
    int little_endian = get_be16(pb);

594 595
    dprintf(c->fc, "enda %d\n", little_endian);
    if (little_endian == 1) {
596 597 598 599 600 601 602
        switch (st->codec->codec_id) {
        case CODEC_ID_PCM_S24BE:
            st->codec->codec_id = CODEC_ID_PCM_S24LE;
            break;
        case CODEC_ID_PCM_S32BE:
            st->codec->codec_id = CODEC_ID_PCM_S32LE;
            break;
603 604 605 606 607 608
        case CODEC_ID_PCM_F32BE:
            st->codec->codec_id = CODEC_ID_PCM_F32LE;
            break;
        case CODEC_ID_PCM_F64BE:
            st->codec->codec_id = CODEC_ID_PCM_F64LE;
            break;
609 610 611 612 613 614 615
        default:
            break;
        }
    }
    return 0;
}

616
/* FIXME modify qdm2/svq3/h264 decoders to take full atom as extradata */
617
static int mov_read_extradata(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
618
{
619 620
    AVStream *st;
    uint64_t size;
621
    uint8_t *buf;
622 623 624 625 626

    if (c->fc->nb_streams < 1) // will happen with jp2 files
        return 0;
    st= c->fc->streams[c->fc->nb_streams-1];
    size= (uint64_t)st->codec->extradata_size + atom.size + 8 + FF_INPUT_BUFFER_PADDING_SIZE;
627
    if(size > INT_MAX || (uint64_t)atom.size > INT_MAX)
628
        return -1;
629 630 631 632 633 634 635 636 637
    buf= av_realloc(st->codec->extradata, size);
    if(!buf)
        return -1;
    st->codec->extradata= buf;
    buf+= st->codec->extradata_size;
    st->codec->extradata_size= size - FF_INPUT_BUFFER_PADDING_SIZE;
    AV_WB32(       buf    , atom.size + 8);
    AV_WL32(       buf + 4, atom.type);
    get_buffer(pb, buf + 8, atom.size);
638 639 640
    return 0;
}

641
static int mov_read_wave(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
R
Roberto Togni 已提交
642 643 644 645 646
{
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];

    if((uint64_t)atom.size > (1<<30))
        return -1;
647

648 649 650
    if (st->codec->codec_id == CODEC_ID_QDM2) {
        // pass all frma atom to codec, needed at least for QDM2
        av_free(st->codec->extradata);
651 652 653
        st->codec->extradata = av_mallocz(atom.size + FF_INPUT_BUFFER_PADDING_SIZE);
        if (!st->codec->extradata)
            return AVERROR(ENOMEM);
654
        st->codec->extradata_size = atom.size;
655
        get_buffer(pb, st->codec->extradata, atom.size);
656
    } else if (atom.size > 8) { /* to read frma, esds atoms */
657 658
        if (mov_read_default(c, pb, atom) < 0)
            return -1;
659
    } else
660
        url_fskip(pb, atom.size);
R
Roberto Togni 已提交
661 662 663
    return 0;
}

664 665 666 667
/**
 * This function reads atom content and puts data in extradata without tag
 * nor size unlike mov_read_extradata.
 */
668
static int mov_read_glbl(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
669 670 671
{
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];

672 673 674
    if((uint64_t)atom.size > (1<<30))
        return -1;

675
    av_free(st->codec->extradata);
676 677 678
    st->codec->extradata = av_mallocz(atom.size + FF_INPUT_BUFFER_PADDING_SIZE);
    if (!st->codec->extradata)
        return AVERROR(ENOMEM);
679
    st->codec->extradata_size = atom.size;
680
    get_buffer(pb, st->codec->extradata, atom.size);
681 682 683
    return 0;
}

684
static int mov_read_stco(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
685 686
{
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];
687
    MOVStreamContext *sc = st->priv_data;
688
    unsigned int i, entries;
689 690

    get_byte(pb); /* version */
691
    get_be24(pb); /* flags */
692 693

    entries = get_be32(pb);
694

695 696
    if(entries >= UINT_MAX/sizeof(int64_t))
        return -1;
697

B
Baptiste Coudurier 已提交
698
    sc->chunk_offsets = av_malloc(entries * sizeof(int64_t));
699
    if (!sc->chunk_offsets)
700 701 702
        return AVERROR(ENOMEM);
    sc->chunk_count = entries;

703 704
    if      (atom.type == MKTAG('s','t','c','o'))
        for(i=0; i<entries; i++)
705
            sc->chunk_offsets[i] = get_be32(pb);
706 707
    else if (atom.type == MKTAG('c','o','6','4'))
        for(i=0; i<entries; i++)
708
            sc->chunk_offsets[i] = get_be64(pb);
709
    else
710
        return -1;
711

712 713 714
    return 0;
}

715 716 717 718
/**
 * Compute codec id for 'lpcm' tag.
 * See CoreAudioTypes and AudioStreamBasicDescription at Apple.
 */
D
Diego Pettenò 已提交
719
static enum CodecID mov_get_lpcm_codec_id(int bps, int flags)
720 721 722 723
{
    if (flags & 1) { // floating point
        if (flags & 2) { // big endian
            if      (bps == 32) return CODEC_ID_PCM_F32BE;
724
            else if (bps == 64) return CODEC_ID_PCM_F64BE;
725
        } else {
726 727
            if      (bps == 32) return CODEC_ID_PCM_F32LE;
            else if (bps == 64) return CODEC_ID_PCM_F64LE;
728 729 730 731 732 733 734 735 736 737 738 739 740 741
        }
    } else {
        if (flags & 2) {
            if      (bps == 8)
                // signed integer
                if (flags & 4)  return CODEC_ID_PCM_S8;
                else            return CODEC_ID_PCM_U8;
            else if (bps == 16) return CODEC_ID_PCM_S16BE;
            else if (bps == 24) return CODEC_ID_PCM_S24BE;
            else if (bps == 32) return CODEC_ID_PCM_S32BE;
        } else {
            if      (bps == 8)
                if (flags & 4)  return CODEC_ID_PCM_S8;
                else            return CODEC_ID_PCM_U8;
B
Baptiste Coudurier 已提交
742
            else if (bps == 16) return CODEC_ID_PCM_S16LE;
743 744 745 746
            else if (bps == 24) return CODEC_ID_PCM_S24LE;
            else if (bps == 32) return CODEC_ID_PCM_S32LE;
        }
    }
D
Diego Pettenò 已提交
747
    return CODEC_ID_NONE;
748 749
}

750
static int mov_read_stsd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
751 752
{
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];
753
    MOVStreamContext *sc = st->priv_data;
754
    int j, entries, pseudo_stream_id;
755

756
    get_byte(pb); /* version */
757
    get_be24(pb); /* flags */
758 759 760

    entries = get_be32(pb);

761 762
    for(pseudo_stream_id=0; pseudo_stream_id<entries; pseudo_stream_id++) {
        //Parsing Sample description table
763
        enum CodecID id;
764
        int dref_id;
765
        MOVAtom a = { 0, 0, 0 };
766
        int64_t start_pos = url_ftell(pb);
767
        int size = get_be32(pb); /* size */
768
        uint32_t format = get_le32(pb); /* data format */
769

770 771
        get_be32(pb); /* reserved */
        get_be16(pb); /* reserved */
772
        dref_id = get_be16(pb);
773

774
        if (st->codec->codec_tag &&
775
            st->codec->codec_tag != format &&
776
            (c->fc->video_codec_id ? codec_get_id(codec_movvideo_tags, format) != c->fc->video_codec_id
777
                                   : st->codec->codec_tag != MKTAG('j','p','e','g'))
778
           ){
D
Diego Biurrun 已提交
779 780 781
            /* Multiple fourcc, we skip JPEG. This is not correct, we should
             * export it as a separate AVStream but this needs a few changes
             * in the MOV demuxer, patch welcome. */
B
Baptiste Coudurier 已提交
782
            av_log(c->fc, AV_LOG_WARNING, "multiple fourcc not supported\n");
783 784 785
            url_fskip(pb, size - (url_ftell(pb) - start_pos));
            continue;
        }
786
        sc->pseudo_stream_id = st->codec->codec_tag ? -1 : pseudo_stream_id;
787
        sc->dref_id= dref_id;
788

789
        st->codec->codec_tag = format;
790
        id = codec_get_id(codec_movaudio_tags, format);
791
        if (id<=0 && (format&0xFFFF) == 'm'+('s'<<8))
M
Michael Niedermayer 已提交
792 793
            id = codec_get_id(codec_wav_tags, bswap_32(format)&0xFFFF);

794
        if (st->codec->codec_type != CODEC_TYPE_VIDEO && id > 0) {
795
            st->codec->codec_type = CODEC_TYPE_AUDIO;
796
        } else if (st->codec->codec_type != CODEC_TYPE_AUDIO && /* do not overwrite codec type */
797
                   format && format != MKTAG('m','p','4','s')) { /* skip old asf mpeg4 tag */
798
            id = codec_get_id(codec_movvideo_tags, format);
799
            if (id <= 0)
800
                id = codec_get_id(codec_bmp_tags, format);
801 802
            if (id > 0)
                st->codec->codec_type = CODEC_TYPE_VIDEO;
803 804 805 806 807
            else if(st->codec->codec_type == CODEC_TYPE_DATA){
                id = codec_get_id(ff_codec_movsubtitle_tags, format);
                if(id > 0)
                    st->codec->codec_type = CODEC_TYPE_SUBTITLE;
            }
808 809
        }

810 811 812
        dprintf(c->fc, "size=%d 4CC= %c%c%c%c codec_type=%d\n", size,
                (format >> 0) & 0xff, (format >> 8) & 0xff, (format >> 16) & 0xff,
                (format >> 24) & 0xff, st->codec->codec_type);
813

814
        if(st->codec->codec_type==CODEC_TYPE_VIDEO) {
815 816 817 818
            uint8_t codec_name[32];
            unsigned int color_depth;
            int color_greyscale;

819
            st->codec->codec_id = id;
820 821 822 823
            get_be16(pb); /* version */
            get_be16(pb); /* revision level */
            get_be32(pb); /* vendor */
            get_be32(pb); /* temporal quality */
D
Diego Biurrun 已提交
824
            get_be32(pb); /* spatial quality */
825 826 827 828

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

829 830 831
            get_be32(pb); /* horiz resolution */
            get_be32(pb); /* vert resolution */
            get_be32(pb); /* data size, always 0 */
B
Baptiste Coudurier 已提交
832
            get_be16(pb); /* frames per samples */
833

B
Baptiste Coudurier 已提交
834
            get_buffer(pb, codec_name, 32); /* codec name, pascal string */
B
Baptiste Coudurier 已提交
835 836 837 838
            if (codec_name[0] <= 31) {
                memcpy(st->codec->codec_name, &codec_name[1],codec_name[0]);
                st->codec->codec_name[codec_name[0]] = 0;
            }
839

840
            st->codec->bits_per_coded_sample = get_be16(pb); /* depth */
841
            st->codec->color_table_id = get_be16(pb); /* colortable id */
842
            dprintf(c->fc, "depth %d, ctab id %d\n",
843
                   st->codec->bits_per_coded_sample, st->codec->color_table_id);
844
            /* figure out the palette situation */
845 846
            color_depth = st->codec->bits_per_coded_sample & 0x1F;
            color_greyscale = st->codec->bits_per_coded_sample & 0x20;
847 848

            /* if the depth is 2, 4, or 8 bpp, file is palettized */
849
            if ((color_depth == 2) || (color_depth == 4) ||
850
                (color_depth == 8)) {
851 852 853 854
                /* for palette traversal */
                unsigned int color_start, color_count, color_end;
                unsigned char r, g, b;

855
                st->codec->palctrl = av_malloc(sizeof(*st->codec->palctrl));
856
                if (color_greyscale) {
857
                    int color_index, color_dec;
858
                    /* compute the greyscale palette */
859
                    st->codec->bits_per_coded_sample = color_depth;
860 861 862 863 864
                    color_count = 1 << color_depth;
                    color_index = 255;
                    color_dec = 256 / (color_count - 1);
                    for (j = 0; j < color_count; j++) {
                        r = g = b = color_index;
865
                        st->codec->palctrl->palette[j] =
866 867 868 869 870
                            (r << 16) | (g << 8) | (b);
                        color_index -= color_dec;
                        if (color_index < 0)
                            color_index = 0;
                    }
871
                } else if (st->codec->color_table_id) {
872
                    const uint8_t *color_table;
873 874 875
                    /* if flag bit 3 is set, use the default palette */
                    color_count = 1 << color_depth;
                    if (color_depth == 2)
M
Michael Niedermayer 已提交
876
                        color_table = ff_qt_default_palette_4;
877
                    else if (color_depth == 4)
M
Michael Niedermayer 已提交
878
                        color_table = ff_qt_default_palette_16;
879
                    else
M
Michael Niedermayer 已提交
880
                        color_table = ff_qt_default_palette_256;
881 882 883 884 885

                    for (j = 0; j < color_count; j++) {
                        r = color_table[j * 4 + 0];
                        g = color_table[j * 4 + 1];
                        b = color_table[j * 4 + 2];
886
                        st->codec->palctrl->palette[j] =
887 888 889 890 891 892 893
                            (r << 16) | (g << 8) | (b);
                    }
                } else {
                    /* load the palette from the file */
                    color_start = get_be32(pb);
                    color_count = get_be16(pb);
                    color_end = get_be16(pb);
894 895
                    if ((color_start <= 255) &&
                        (color_end <= 255)) {
M
Mike Melanson 已提交
896 897 898 899 900 901 902 903 904 905 906 907
                        for (j = color_start; j <= color_end; j++) {
                            /* each R, G, or B component is 16 bits;
                             * only use the top 8 bits; skip alpha bytes
                             * up front */
                            get_byte(pb);
                            get_byte(pb);
                            r = get_byte(pb);
                            get_byte(pb);
                            g = get_byte(pb);
                            get_byte(pb);
                            b = get_byte(pb);
                            get_byte(pb);
908
                            st->codec->palctrl->palette[j] =
M
Mike Melanson 已提交
909
                                (r << 16) | (g << 8) | (b);
910
                        }
911 912
                    }
                }
913
                st->codec->palctrl->palette_changed = 1;
914
            }
915
        } else if(st->codec->codec_type==CODEC_TYPE_AUDIO) {
916
            int bits_per_sample, flags;
917
            uint16_t version = get_be16(pb);
918

919
            st->codec->codec_id = id;
920 921
            get_be16(pb); /* revision level */
            get_be32(pb); /* vendor */
922

923
            st->codec->channels = get_be16(pb);             /* channel count */
M
Michel Bardiaux 已提交
924
            dprintf(c->fc, "audio channels %d\n", st->codec->channels);
925
            st->codec->bits_per_coded_sample = get_be16(pb);      /* sample size */
926

927
            sc->audio_cid = get_be16(pb);
928 929 930 931
            get_be16(pb); /* packet size = 0 */

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

932 933 934 935 936 937 938 939 940 941 942 943 944
            //Read QT version 1 fields. In version 0 these do not exist.
            dprintf(c->fc, "version =%d, isom =%d\n",version,c->isom);
            if(!c->isom) {
                if(version==1) {
                    sc->samples_per_frame = get_be32(pb);
                    get_be32(pb); /* bytes per packet */
                    sc->bytes_per_frame = get_be32(pb);
                    get_be32(pb); /* bytes per sample */
                } else if(version==2) {
                    get_be32(pb); /* sizeof struct only */
                    st->codec->sample_rate = av_int2dbl(get_be64(pb)); /* float 64 */
                    st->codec->channels = get_be32(pb);
                    get_be32(pb); /* always 0x7F000000 */
945
                    st->codec->bits_per_coded_sample = get_be32(pb); /* bits per channel if sound is uncompressed */
946
                    flags = get_be32(pb); /* lcpm format specific flag */
B
Baptiste Coudurier 已提交
947 948
                    sc->bytes_per_frame = get_be32(pb); /* bytes per audio packet if constant */
                    sc->samples_per_frame = get_be32(pb); /* lpcm frames per audio packet if constant */
949
                    if (format == MKTAG('l','p','c','m'))
950
                        st->codec->codec_id = mov_get_lpcm_codec_id(st->codec->bits_per_coded_sample, flags);
951 952 953
                }
            }

954
            switch (st->codec->codec_id) {
955 956
            case CODEC_ID_PCM_S8:
            case CODEC_ID_PCM_U8:
957
                if (st->codec->bits_per_coded_sample == 16)
958 959
                    st->codec->codec_id = CODEC_ID_PCM_S16BE;
                break;
960
            case CODEC_ID_PCM_S16LE:
961
            case CODEC_ID_PCM_S16BE:
962
                if (st->codec->bits_per_coded_sample == 8)
963
                    st->codec->codec_id = CODEC_ID_PCM_S8;
964
                else if (st->codec->bits_per_coded_sample == 24)
965 966 967
                    st->codec->codec_id =
                        st->codec->codec_id == CODEC_ID_PCM_S16BE ?
                        CODEC_ID_PCM_S24BE : CODEC_ID_PCM_S24LE;
968
                break;
969 970 971 972 973 974 975 976 977 978 979 980 981
            /* set values for old format before stsd version 1 appeared */
            case CODEC_ID_MACE3:
                sc->samples_per_frame = 6;
                sc->bytes_per_frame = 2*st->codec->channels;
                break;
            case CODEC_ID_MACE6:
                sc->samples_per_frame = 6;
                sc->bytes_per_frame = 1*st->codec->channels;
                break;
            case CODEC_ID_ADPCM_IMA_QT:
                sc->samples_per_frame = 64;
                sc->bytes_per_frame = 34*st->codec->channels;
                break;
982 983 984 985
            case CODEC_ID_GSM:
                sc->samples_per_frame = 160;
                sc->bytes_per_frame = 33;
                break;
986 987
            default:
                break;
988
            }
989

990 991
            bits_per_sample = av_get_bits_per_sample(st->codec->codec_id);
            if (bits_per_sample) {
992
                st->codec->bits_per_coded_sample = bits_per_sample;
993 994
                sc->sample_size = (bits_per_sample >> 3) * st->codec->channels;
            }
995
        } else if(st->codec->codec_type==CODEC_TYPE_SUBTITLE){
996 997 998 999
            // ttxt stsd contains display flags, justification, background
            // color, fonts, and default styles, so fake an atom to read it
            MOVAtom fake_atom = { .size = size - (url_ftell(pb) - start_pos) };
            mov_read_glbl(c, pb, fake_atom);
1000
            st->codec->codec_id= id;
1001 1002
            st->codec->width = sc->width;
            st->codec->height = sc->height;
1003 1004 1005
        } else {
            /* other codec type, just skip (rtp, mp4s, tmcd ...) */
            url_fskip(pb, size - (url_ftell(pb) - start_pos));
1006
        }
1007 1008
        /* this will read extra atoms at the end (wave, alac, damr, avcC, SMI ...) */
        a.size = size - (url_ftell(pb) - start_pos);
1009 1010 1011 1012
        if (a.size > 8) {
            if (mov_read_default(c, pb, a) < 0)
                return -1;
        } else if (a.size > 0)
1013
            url_fskip(pb, a.size);
1014
    }
1015

B
Baptiste Coudurier 已提交
1016
    if(st->codec->codec_type==CODEC_TYPE_AUDIO && st->codec->sample_rate==0 && sc->time_scale>1)
1017
        st->codec->sample_rate= sc->time_scale;
1018

1019
    /* special codec parameters handling */
1020
    switch (st->codec->codec_id) {
1021
#if CONFIG_DV_DEMUXER
1022
    case CODEC_ID_DVAUDIO:
1023
        c->dv_fctx = avformat_alloc_context();
1024 1025 1026 1027 1028 1029 1030 1031
        c->dv_demux = dv_init_demux(c->dv_fctx);
        if (!c->dv_demux) {
            av_log(c->fc, AV_LOG_ERROR, "dv demux context init error\n");
            return -1;
        }
        sc->dv_audio_container = 1;
        st->codec->codec_id = CODEC_ID_PCM_S16LE;
        break;
1032
#endif
1033
    /* no ifdef since parameters are always those */
1034
    case CODEC_ID_QCELP:
1035
        st->codec->frame_size= 160;
1036
        st->codec->channels= 1; /* really needed */
1037
        break;
1038
    case CODEC_ID_AMR_NB:
1039
    case CODEC_ID_AMR_WB:
B
Baptiste Coudurier 已提交
1040
        st->codec->frame_size= sc->samples_per_frame;
1041
        st->codec->channels= 1; /* really needed */
1042
        /* force sample rate for amr, stsd in 3gp does not store sample rate */
B
Baptiste Coudurier 已提交
1043
        if (st->codec->codec_id == CODEC_ID_AMR_NB)
1044
            st->codec->sample_rate = 8000;
B
Baptiste Coudurier 已提交
1045 1046
        else if (st->codec->codec_id == CODEC_ID_AMR_WB)
            st->codec->sample_rate = 16000;
1047
        break;
1048
    case CODEC_ID_MP2:
1049
    case CODEC_ID_MP3:
1050
        st->codec->codec_type = CODEC_TYPE_AUDIO; /* force type after stsd for m1a hdlr */
A
Aurelien Jacobs 已提交
1051
        st->need_parsing = AVSTREAM_PARSE_FULL;
1052
        break;
1053
    case CODEC_ID_GSM:
1054 1055 1056 1057
    case CODEC_ID_ADPCM_MS:
    case CODEC_ID_ADPCM_IMA_WAV:
        st->codec->block_align = sc->bytes_per_frame;
        break;
1058
    case CODEC_ID_ALAC:
1059
        if (st->codec->extradata_size == 36) {
1060 1061
            st->codec->frame_size = AV_RB32(st->codec->extradata+12);
            st->codec->channels   = AV_RB8 (st->codec->extradata+21);
1062
        }
1063
        break;
1064 1065 1066
    default:
        break;
    }
1067

1068 1069 1070
    return 0;
}

1071
static int mov_read_stsc(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1072
{
1073
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1074
    MOVStreamContext *sc = st->priv_data;
1075
    unsigned int i, entries;
1076

1077
    get_byte(pb); /* version */
1078
    get_be24(pb); /* flags */
1079 1080

    entries = get_be32(pb);
1081

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

1084 1085
    if(entries >= UINT_MAX / sizeof(*sc->stsc_data))
        return -1;
1086 1087
    sc->stsc_data = av_malloc(entries * sizeof(*sc->stsc_data));
    if (!sc->stsc_data)
1088 1089 1090
        return AVERROR(ENOMEM);
    sc->stsc_count = entries;

1091
    for(i=0; i<entries; i++) {
1092 1093 1094
        sc->stsc_data[i].first = get_be32(pb);
        sc->stsc_data[i].count = get_be32(pb);
        sc->stsc_data[i].id = get_be32(pb);
1095 1096 1097 1098
    }
    return 0;
}

1099
static int mov_read_stss(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1100 1101
{
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1102
    MOVStreamContext *sc = st->priv_data;
1103
    unsigned int i, entries;
1104 1105

    get_byte(pb); /* version */
1106
    get_be24(pb); /* flags */
1107 1108

    entries = get_be32(pb);
1109

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

B
Baptiste Coudurier 已提交
1112
    if(entries >= UINT_MAX / sizeof(int))
1113
        return -1;
B
Baptiste Coudurier 已提交
1114
    sc->keyframes = av_malloc(entries * sizeof(int));
1115
    if (!sc->keyframes)
1116 1117 1118
        return AVERROR(ENOMEM);
    sc->keyframe_count = entries;

1119 1120
    for(i=0; i<entries; i++) {
        sc->keyframes[i] = get_be32(pb);
1121
        //dprintf(c->fc, "keyframes[]=%d\n", sc->keyframes[i]);
1122 1123 1124 1125
    }
    return 0;
}

1126
static int mov_read_stsz(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1127
{
1128
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1129
    MOVStreamContext *sc = st->priv_data;
1130
    unsigned int i, entries, sample_size;
1131

1132
    get_byte(pb); /* version */
1133
    get_be24(pb); /* flags */
1134

1135 1136 1137
    sample_size = get_be32(pb);
    if (!sc->sample_size) /* do not overwrite value computed in stsd */
        sc->sample_size = sample_size;
1138
    entries = get_be32(pb);
1139 1140

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

1142
    sc->sample_count = entries;
1143 1144 1145
    if (sample_size)
        return 0;

1146 1147
    if(entries >= UINT_MAX / sizeof(int))
        return -1;
B
Baptiste Coudurier 已提交
1148
    sc->sample_sizes = av_malloc(entries * sizeof(int));
1149
    if (!sc->sample_sizes)
1150 1151
        return AVERROR(ENOMEM);

1152
    for(i=0; i<entries; i++)
1153 1154 1155 1156
        sc->sample_sizes[i] = get_be32(pb);
    return 0;
}

1157
static int mov_read_stts(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1158
{
1159
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1160
    MOVStreamContext *sc = st->priv_data;
1161
    unsigned int i, entries;
1162 1163
    int64_t duration=0;
    int64_t total_sample_count=0;
1164

1165
    get_byte(pb); /* version */
1166
    get_be24(pb); /* flags */
1167
    entries = get_be32(pb);
1168 1169 1170

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

1171
    if(entries >= UINT_MAX / sizeof(*sc->stts_data))
1172
        return -1;
1173
    sc->stts_data = av_malloc(entries * sizeof(*sc->stts_data));
1174
    if (!sc->stts_data)
1175 1176
        return AVERROR(ENOMEM);
    sc->stts_count = entries;
1177

1178
    for(i=0; i<entries; i++) {
M
cleanup  
Michael Niedermayer 已提交
1179 1180
        int sample_duration;
        int sample_count;
1181

1182
        sample_count=get_be32(pb);
1183
        sample_duration = get_be32(pb);
1184 1185 1186
        sc->stts_data[i].count= sample_count;
        sc->stts_data[i].duration= sample_duration;

1187
        sc->time_rate= av_gcd(sc->time_rate, sample_duration);
1188

M
Michel Bardiaux 已提交
1189
        dprintf(c->fc, "sample_count=%d, sample_duration=%d\n",sample_count,sample_duration);
1190

B
Baptiste Coudurier 已提交
1191
        duration+=(int64_t)sample_duration*sample_count;
1192 1193 1194
        total_sample_count+=sample_count;
    }

1195 1196 1197
    st->nb_frames= total_sample_count;
    if(duration)
        st->duration= duration;
1198 1199 1200
    return 0;
}

1201
static int mov_read_ctts(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
M
Michael Niedermayer 已提交
1202
{
1203
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1204
    MOVStreamContext *sc = st->priv_data;
M
Michael Niedermayer 已提交
1205 1206 1207
    unsigned int i, entries;

    get_byte(pb); /* version */
1208
    get_be24(pb); /* flags */
M
Michael Niedermayer 已提交
1209
    entries = get_be32(pb);
1210 1211 1212

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

1213
    if(entries >= UINT_MAX / sizeof(*sc->ctts_data))
M
Michael Niedermayer 已提交
1214
        return -1;
1215
    sc->ctts_data = av_malloc(entries * sizeof(*sc->ctts_data));
1216
    if (!sc->ctts_data)
1217 1218
        return AVERROR(ENOMEM);
    sc->ctts_count = entries;
1219

M
Michael Niedermayer 已提交
1220
    for(i=0; i<entries; i++) {
1221 1222 1223
        int count    =get_be32(pb);
        int duration =get_be32(pb);

1224
        if (duration < 0) {
1225 1226
            sc->wrong_dts = 1;
            st->codec->has_b_frames = 1;
1227
        }
1228 1229 1230
        sc->ctts_data[i].count   = count;
        sc->ctts_data[i].duration= duration;

1231
        sc->time_rate= av_gcd(sc->time_rate, FFABS(duration));
M
Michael Niedermayer 已提交
1232 1233 1234 1235
    }
    return 0;
}

1236 1237 1238
static void mov_build_index(MOVContext *mov, AVStream *st)
{
    MOVStreamContext *sc = st->priv_data;
1239
    int64_t current_offset;
1240 1241 1242 1243 1244 1245
    int64_t current_dts = 0;
    unsigned int stts_index = 0;
    unsigned int stsc_index = 0;
    unsigned int stss_index = 0;
    unsigned int i, j;

1246 1247 1248 1249 1250 1251
    /* adjust first dts according to edit list */
    if (sc->time_offset) {
        assert(sc->time_offset % sc->time_rate == 0);
        current_dts = - (sc->time_offset / sc->time_rate);
    }

1252 1253 1254
    /* only use old uncompressed audio chunk demuxing when stts specifies it */
    if (!(st->codec->codec_type == CODEC_TYPE_AUDIO &&
          sc->stts_count == 1 && sc->stts_data[0].duration == 1)) {
1255 1256 1257 1258 1259 1260 1261 1262 1263
        unsigned int current_sample = 0;
        unsigned int stts_sample = 0;
        unsigned int keyframe, sample_size;
        unsigned int distance = 0;
        int key_off = sc->keyframes && sc->keyframes[0] == 1;

        st->nb_frames = sc->sample_count;
        for (i = 0; i < sc->chunk_count; i++) {
            current_offset = sc->chunk_offsets[i];
1264 1265
            if (stsc_index + 1 < sc->stsc_count &&
                i + 1 == sc->stsc_data[stsc_index + 1].first)
1266
                stsc_index++;
1267
            for (j = 0; j < sc->stsc_data[stsc_index].count; j++) {
1268 1269
                if (current_sample >= sc->sample_count) {
                    av_log(mov->fc, AV_LOG_ERROR, "wrong sample count\n");
1270
                    return;
1271 1272 1273 1274 1275 1276 1277 1278
                }
                keyframe = !sc->keyframe_count || current_sample+key_off == sc->keyframes[stss_index];
                if (keyframe) {
                    distance = 0;
                    if (stss_index + 1 < sc->keyframe_count)
                        stss_index++;
                }
                sample_size = sc->sample_size > 0 ? sc->sample_size : sc->sample_sizes[current_sample];
1279
                if(sc->pseudo_stream_id == -1 ||
1280
                   sc->stsc_data[stsc_index].id - 1 == sc->pseudo_stream_id) {
1281 1282
                    av_add_index_entry(st, current_offset, current_dts, sample_size, distance,
                                    keyframe ? AVINDEX_KEYFRAME : 0);
1283 1284 1285 1286
                    dprintf(mov->fc, "AVIndex stream %d, sample %d, offset %"PRIx64", dts %"PRId64", "
                            "size %d, distance %d, keyframe %d\n", st->index, current_sample,
                            current_offset, current_dts, sample_size, distance, keyframe);
                }
1287

1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304
                current_offset += sample_size;
                assert(sc->stts_data[stts_index].duration % sc->time_rate == 0);
                current_dts += sc->stts_data[stts_index].duration / sc->time_rate;
                distance++;
                stts_sample++;
                current_sample++;
                if (stts_index + 1 < sc->stts_count && stts_sample == sc->stts_data[stts_index].count) {
                    stts_sample = 0;
                    stts_index++;
                }
            }
        }
    } else { /* read whole chunk */
        unsigned int chunk_samples, chunk_size, chunk_duration;
        unsigned int frames = 1;
        for (i = 0; i < sc->chunk_count; i++) {
            current_offset = sc->chunk_offsets[i];
1305 1306
            if (stsc_index + 1 < sc->stsc_count &&
                i + 1 == sc->stsc_data[stsc_index + 1].first)
1307
                stsc_index++;
1308
            chunk_samples = sc->stsc_data[stsc_index].count;
1309 1310 1311
            /* get chunk size, beware of alaw/ulaw/mace */
            if (sc->samples_per_frame > 0 &&
                (chunk_samples * sc->bytes_per_frame % sc->samples_per_frame == 0)) {
1312
                if (sc->samples_per_frame < 160)
1313 1314 1315 1316 1317 1318
                    chunk_size = chunk_samples * sc->bytes_per_frame / sc->samples_per_frame;
                else {
                    chunk_size = sc->bytes_per_frame;
                    frames = chunk_samples / sc->samples_per_frame;
                    chunk_samples = sc->samples_per_frame;
                }
1319
            } else
1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337
                chunk_size = chunk_samples * sc->sample_size;
            for (j = 0; j < frames; j++) {
                av_add_index_entry(st, current_offset, current_dts, chunk_size, 0, AVINDEX_KEYFRAME);
                /* get chunk duration */
                chunk_duration = 0;
                while (chunk_samples > 0) {
                    if (chunk_samples < sc->stts_data[stts_index].count) {
                        chunk_duration += sc->stts_data[stts_index].duration * chunk_samples;
                        sc->stts_data[stts_index].count -= chunk_samples;
                        break;
                    } else {
                        chunk_duration += sc->stts_data[stts_index].duration * chunk_samples;
                        chunk_samples -= sc->stts_data[stts_index].count;
                        if (stts_index + 1 < sc->stts_count)
                            stts_index++;
                    }
                }
                current_offset += sc->bytes_per_frame;
1338 1339 1340
                dprintf(mov->fc, "AVIndex stream %d, chunk %d, offset %"PRIx64", dts %"PRId64", "
                        "size %d, duration %d\n", st->index, i, current_offset, current_dts,
                        chunk_size, chunk_duration);
1341 1342 1343 1344 1345 1346
                assert(chunk_duration % sc->time_rate == 0);
                current_dts += chunk_duration / sc->time_rate;
            }
        }
    }
}
1347

1348
static int mov_read_trak(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1349 1350 1351
{
    AVStream *st;
    MOVStreamContext *sc;
1352
    int ret;
1353 1354

    st = av_new_stream(c->fc, c->fc->nb_streams);
B
Baptiste Coudurier 已提交
1355
    if (!st) return AVERROR(ENOMEM);
B
Baptiste Coudurier 已提交
1356
    sc = av_mallocz(sizeof(MOVStreamContext));
1357
    if (!sc) return AVERROR(ENOMEM);
1358 1359

    st->priv_data = sc;
1360
    st->codec->codec_type = CODEC_TYPE_DATA;
1361
    sc->ffindex = st->index;
1362

1363 1364 1365 1366
    if ((ret = mov_read_default(c, pb, atom)) < 0)
        return ret;

    /* sanity checks */
1367 1368
    if (sc->chunk_count && (!sc->stts_count || !sc->stsc_count ||
                            (!sc->sample_size && !sc->sample_count))) {
1369 1370
        av_log(c->fc, AV_LOG_ERROR, "stream %d, missing mandatory atoms, broken header\n",
               st->index);
1371 1372
        return 0;
    }
1373 1374 1375 1376 1377 1378

    if (!sc->time_rate)
        sc->time_rate = 1;
    if (!sc->time_scale)
        sc->time_scale = c->time_scale;

1379 1380
    av_set_pts_info(st, 64, sc->time_rate, sc->time_scale);

1381
    if (st->codec->codec_type == CODEC_TYPE_AUDIO &&
1382 1383 1384 1385 1386
        !st->codec->frame_size && sc->stts_count == 1) {
        st->codec->frame_size = av_rescale(sc->stts_data[0].duration,
                                           st->codec->sample_rate, sc->time_scale);
        dprintf(c->fc, "frame size %d\n", st->codec->frame_size);
    }
1387

1388
    if (st->duration != AV_NOPTS_VALUE) {
1389 1390 1391
        assert(st->duration % sc->time_rate == 0);
        st->duration /= sc->time_rate;
    }
1392

1393 1394 1395 1396
    mov_build_index(c, st);

    if (sc->dref_id-1 < sc->drefs_count && sc->drefs[sc->dref_id-1].path) {
        if (url_fopen(&sc->pb, sc->drefs[sc->dref_id-1].path, URL_RDONLY) < 0)
1397 1398
            av_log(c->fc, AV_LOG_ERROR, "stream %d, error opening file %s: %s\n",
                   st->index, sc->drefs[sc->dref_id-1].path, strerror(errno));
1399 1400 1401 1402
    } else
        sc->pb = c->fc->pb;

    switch (st->codec->codec_id) {
1403
#if CONFIG_H261_DECODER
1404 1405
    case CODEC_ID_H261:
#endif
1406
#if CONFIG_H263_DECODER
1407 1408
    case CODEC_ID_H263:
#endif
1409
#if CONFIG_MPEG4_DECODER
1410 1411
    case CODEC_ID_MPEG4:
#endif
1412
        st->codec->width = 0; /* let decoder init width/height */
1413 1414 1415
        st->codec->height= 0;
        break;
    }
B
Baptiste Coudurier 已提交
1416 1417 1418

    /* Do not need those anymore. */
    av_freep(&sc->chunk_offsets);
1419
    av_freep(&sc->stsc_data);
B
Baptiste Coudurier 已提交
1420 1421 1422 1423
    av_freep(&sc->sample_sizes);
    av_freep(&sc->keyframes);
    av_freep(&sc->stts_data);

1424
    return 0;
1425 1426
}

1427
static int mov_read_ilst(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1428 1429 1430 1431 1432 1433 1434 1435
{
    int ret;
    c->itunes_metadata = 1;
    ret = mov_read_default(c, pb, atom);
    c->itunes_metadata = 0;
    return ret;
}

1436
static int mov_read_meta(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1437
{
1438 1439 1440 1441 1442 1443 1444 1445 1446 1447
    while (atom.size > 8) {
        uint32_t tag = get_le32(pb);
        atom.size -= 4;
        if (tag == MKTAG('h','d','l','r')) {
            url_fseek(pb, -8, SEEK_CUR);
            atom.size += 8;
            return mov_read_default(c, pb, atom);
        }
    }
    return 0;
1448 1449
}

1450
static int mov_read_tkhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1451
{
1452 1453 1454 1455 1456
    int i;
    int width;
    int height;
    int64_t disp_transform[2];
    int display_matrix[3][2];
B
clean  
Baptiste Coudurier 已提交
1457
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1458
    MOVStreamContext *sc = st->priv_data;
B
clean  
Baptiste Coudurier 已提交
1459
    int version = get_byte(pb);
1460

1461
    get_be24(pb); /* flags */
1462 1463 1464 1465 1466 1467 1468
    /*
    MOV_TRACK_ENABLED 0x0001
    MOV_TRACK_IN_MOVIE 0x0002
    MOV_TRACK_IN_PREVIEW 0x0004
    MOV_TRACK_IN_POSTER 0x0008
    */

B
Baptiste Coudurier 已提交
1469 1470 1471 1472 1473 1474 1475
    if (version == 1) {
        get_be64(pb);
        get_be64(pb);
    } else {
        get_be32(pb); /* creation time */
        get_be32(pb); /* modification time */
    }
1476 1477
    st->id = (int)get_be32(pb); /* track id (NOT 0 !)*/
    get_be32(pb); /* reserved */
1478

1479 1480
    /* highlevel (considering edits) duration in movie timebase */
    (version == 1) ? get_be64(pb) : get_be32(pb);
1481 1482 1483 1484 1485 1486 1487 1488
    get_be32(pb); /* reserved */
    get_be32(pb); /* reserved */

    get_be16(pb); /* layer */
    get_be16(pb); /* alternate group */
    get_be16(pb); /* volume */
    get_be16(pb); /* reserved */

1489 1490 1491 1492 1493 1494 1495 1496
    //read in the display matrix (outlined in ISO 14496-12, Section 6.2.2)
    // they're kept in fixed point format through all calculations
    // ignore u,v,z b/c we don't need the scale factor to calc aspect ratio
    for (i = 0; i < 3; i++) {
        display_matrix[i][0] = get_be32(pb);   // 16.16 fixed point
        display_matrix[i][1] = get_be32(pb);   // 16.16 fixed point
        get_be32(pb);           // 2.30 fixed point (not used)
    }
1497

1498 1499
    width = get_be32(pb);       // 16.16 fixed point track width
    height = get_be32(pb);      // 16.16 fixed point track height
1500 1501
    sc->width = width >> 16;
    sc->height = height >> 16;
1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516

    //transform the display width/height according to the matrix
    // skip this if the display matrix is the default identity matrix
    // to keep the same scale, use [width height 1<<16]
    if (width && height &&
        (display_matrix[0][0] != 65536 || display_matrix[0][1]           ||
        display_matrix[1][0]           || display_matrix[1][1] != 65536  ||
        display_matrix[2][0]           || display_matrix[2][1])) {
        for (i = 0; i < 2; i++)
            disp_transform[i] =
                (int64_t)  width  * display_matrix[0][i] +
                (int64_t)  height * display_matrix[1][i] +
                ((int64_t) display_matrix[2][i] << 16);

        //sample aspect ratio is new width/height divided by old width/height
1517
        st->sample_aspect_ratio = av_d2q(
1518 1519 1520
            ((double) disp_transform[0] * height) /
            ((double) disp_transform[1] * width), INT_MAX);
    }
1521 1522 1523
    return 0;
}

1524
static int mov_read_tfhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
1525 1526 1527 1528 1529 1530 1531 1532 1533
{
    MOVFragment *frag = &c->fragment;
    MOVTrackExt *trex = NULL;
    int flags, track_id, i;

    get_byte(pb); /* version */
    flags = get_be24(pb);

    track_id = get_be32(pb);
1534
    if (!track_id)
B
Baptiste Coudurier 已提交
1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558
        return -1;
    frag->track_id = track_id;
    for (i = 0; i < c->trex_count; i++)
        if (c->trex_data[i].track_id == frag->track_id) {
            trex = &c->trex_data[i];
            break;
        }
    if (!trex) {
        av_log(c->fc, AV_LOG_ERROR, "could not find corresponding trex\n");
        return -1;
    }

    if (flags & 0x01) frag->base_data_offset = get_be64(pb);
    else              frag->base_data_offset = frag->moof_offset;
    if (flags & 0x02) frag->stsd_id          = get_be32(pb);
    else              frag->stsd_id          = trex->stsd_id;

    frag->duration = flags & 0x08 ? get_be32(pb) : trex->duration;
    frag->size     = flags & 0x10 ? get_be32(pb) : trex->size;
    frag->flags    = flags & 0x20 ? get_be32(pb) : trex->flags;
    dprintf(c->fc, "frag flags 0x%x\n", frag->flags);
    return 0;
}

1559
static int mov_read_trex(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
1560 1561 1562 1563 1564
{
    MOVTrackExt *trex;

    if ((uint64_t)c->trex_count+1 >= UINT_MAX / sizeof(*c->trex_data))
        return -1;
1565 1566
    trex = av_realloc(c->trex_data, (c->trex_count+1)*sizeof(*c->trex_data));
    if (!trex)
B
Baptiste Coudurier 已提交
1567
        return AVERROR(ENOMEM);
1568
    c->trex_data = trex;
B
Baptiste Coudurier 已提交
1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579
    trex = &c->trex_data[c->trex_count++];
    get_byte(pb); /* version */
    get_be24(pb); /* flags */
    trex->track_id = get_be32(pb);
    trex->stsd_id  = get_be32(pb);
    trex->duration = get_be32(pb);
    trex->size     = get_be32(pb);
    trex->flags    = get_be32(pb);
    return 0;
}

1580
static int mov_read_trun(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
1581 1582
{
    MOVFragment *frag = &c->fragment;
1583
    AVStream *st = NULL;
1584
    MOVStreamContext *sc;
B
Baptiste Coudurier 已提交
1585 1586 1587 1588 1589 1590
    uint64_t offset;
    int64_t dts;
    int data_offset = 0;
    unsigned entries, first_sample_flags = frag->flags;
    int flags, distance, i;

1591 1592 1593 1594 1595 1596 1597 1598
    for (i = 0; i < c->fc->nb_streams; i++) {
        if (c->fc->streams[i]->id == frag->track_id) {
            st = c->fc->streams[i];
            break;
        }
    }
    if (!st) {
        av_log(c->fc, AV_LOG_ERROR, "could not find corresponding track id %d\n", frag->track_id);
1599
        return -1;
1600
    }
1601
    sc = st->priv_data;
B
Baptiste Coudurier 已提交
1602 1603 1604 1605 1606 1607 1608 1609 1610
    if (sc->pseudo_stream_id+1 != frag->stsd_id)
        return 0;
    get_byte(pb); /* version */
    flags = get_be24(pb);
    entries = get_be32(pb);
    dprintf(c->fc, "flags 0x%x entries %d\n", flags, entries);
    if (flags & 0x001) data_offset        = get_be32(pb);
    if (flags & 0x004) first_sample_flags = get_be32(pb);
    if (flags & 0x800) {
1611
        MOVStts *ctts_data;
B
Baptiste Coudurier 已提交
1612 1613
        if ((uint64_t)entries+sc->ctts_count >= UINT_MAX/sizeof(*sc->ctts_data))
            return -1;
1614 1615 1616
        ctts_data = av_realloc(sc->ctts_data,
                               (entries+sc->ctts_count)*sizeof(*sc->ctts_data));
        if (!ctts_data)
B
Baptiste Coudurier 已提交
1617
            return AVERROR(ENOMEM);
1618
        sc->ctts_data = ctts_data;
B
Baptiste Coudurier 已提交
1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655
    }
    dts = st->duration;
    offset = frag->base_data_offset + data_offset;
    distance = 0;
    dprintf(c->fc, "first sample flags 0x%x\n", first_sample_flags);
    for (i = 0; i < entries; i++) {
        unsigned sample_size = frag->size;
        int sample_flags = i ? frag->flags : first_sample_flags;
        unsigned sample_duration = frag->duration;
        int keyframe;

        if (flags & 0x100) sample_duration = get_be32(pb);
        if (flags & 0x200) sample_size     = get_be32(pb);
        if (flags & 0x400) sample_flags    = get_be32(pb);
        if (flags & 0x800) {
            sc->ctts_data[sc->ctts_count].count = 1;
            sc->ctts_data[sc->ctts_count].duration = get_be32(pb);
            sc->ctts_count++;
        }
        if ((keyframe = st->codec->codec_type == CODEC_TYPE_AUDIO ||
             (flags & 0x004 && !i && !sample_flags) || sample_flags & 0x2000000))
            distance = 0;
        av_add_index_entry(st, offset, dts, sample_size, distance,
                           keyframe ? AVINDEX_KEYFRAME : 0);
        dprintf(c->fc, "AVIndex stream %d, sample %d, offset %"PRIx64", dts %"PRId64", "
                "size %d, distance %d, keyframe %d\n", st->index, sc->sample_count+i,
                offset, dts, sample_size, distance, keyframe);
        distance++;
        assert(sample_duration % sc->time_rate == 0);
        dts += sample_duration / sc->time_rate;
        offset += sample_size;
    }
    frag->moof_offset = offset;
    st->duration = dts;
    return 0;
}

1656 1657 1658
/* this atom should be null (from specs), but some buggy files put the 'moov' atom inside it... */
/* like the files created with Adobe Premiere 5.0, for samples see */
/* http://graphics.tudelft.nl/~wouter/publications/soundtests/ */
1659
static int mov_read_wide(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671
{
    int err;

    if (atom.size < 8)
        return 0; /* continue */
    if (get_be32(pb) != 0) { /* 0 sized mdat atom... use the 'wide' atom size */
        url_fskip(pb, atom.size - 4);
        return 0;
    }
    atom.type = get_le32(pb);
    atom.offset += 8;
    atom.size -= 8;
1672
    if (atom.type != MKTAG('m','d','a','t')) {
1673 1674 1675 1676 1677 1678 1679
        url_fskip(pb, atom.size);
        return 0;
    }
    err = mov_read_mdat(c, pb, atom);
    return err;
}

1680
static int mov_read_cmov(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1681
{
1682
#if CONFIG_ZLIB
1683
    ByteIOContext ctx;
1684 1685
    uint8_t *cmov_data;
    uint8_t *moov_data; /* uncompressed data */
1686
    long cmov_len, moov_len;
1687
    int ret = -1;
1688

1689
    get_be32(pb); /* dcom atom */
1690
    if (get_le32(pb) != MKTAG('d','c','o','m'))
1691
        return -1;
1692
    if (get_le32(pb) != MKTAG('z','l','i','b')) {
B
Benoit Fouet 已提交
1693
        av_log(c->fc, AV_LOG_ERROR, "unknown compression for cmov atom !");
1694 1695 1696
        return -1;
    }
    get_be32(pb); /* cmvd atom */
1697
    if (get_le32(pb) != MKTAG('c','m','v','d'))
1698 1699
        return -1;
    moov_len = get_be32(pb); /* uncompressed size */
1700
    cmov_len = atom.size - 6 * 4;
1701

B
Baptiste Coudurier 已提交
1702
    cmov_data = av_malloc(cmov_len);
1703
    if (!cmov_data)
1704
        return AVERROR(ENOMEM);
B
Baptiste Coudurier 已提交
1705
    moov_data = av_malloc(moov_len);
1706 1707
    if (!moov_data) {
        av_free(cmov_data);
1708
        return AVERROR(ENOMEM);
1709 1710
    }
    get_buffer(pb, cmov_data, cmov_len);
1711
    if(uncompress (moov_data, (uLongf *) &moov_len, (const Bytef *)cmov_data, cmov_len) != Z_OK)
1712
        goto free_and_return;
1713
    if(init_put_byte(&ctx, moov_data, moov_len, 0, NULL, NULL, NULL, NULL) != 0)
1714
        goto free_and_return;
1715
    atom.type = MKTAG('m','o','o','v');
1716 1717
    atom.offset = 0;
    atom.size = moov_len;
1718
#ifdef DEBUG
M
Michael Niedermayer 已提交
1719
//    { int fd = open("/tmp/uncompheader.mov", O_WRONLY | O_CREAT); write(fd, moov_data, moov_len); close(fd); }
1720
#endif
1721
    ret = mov_read_default(c, &ctx, atom);
1722
free_and_return:
1723 1724 1725
    av_free(moov_data);
    av_free(cmov_data);
    return ret;
1726 1727 1728
#else
    av_log(c->fc, AV_LOG_ERROR, "this file requires zlib support compiled in\n");
    return -1;
1729
#endif
1730
}
1731

G
Gael Chardon 已提交
1732
/* edit list atom */
1733
static int mov_read_elst(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
G
Gael Chardon 已提交
1734
{
1735
    MOVStreamContext *sc = c->fc->streams[c->fc->nb_streams-1]->priv_data;
B
Baptiste Coudurier 已提交
1736 1737 1738
    int i, edit_count;

    get_byte(pb); /* version */
1739
    get_be24(pb); /* flags */
B
Baptiste Coudurier 已提交
1740
    edit_count = get_be32(pb); /* entries */
B
Baptiste Coudurier 已提交
1741 1742

    for(i=0; i<edit_count; i++){
1743
        int time;
B
Baptiste Coudurier 已提交
1744
        get_be32(pb); /* Track duration */
1745
        time = get_be32(pb); /* Media time */
B
Baptiste Coudurier 已提交
1746
        get_be32(pb); /* Media rate */
1747
        if (i == 0 && time != -1) {
1748
            sc->time_offset = time;
1749 1750
            sc->time_rate = av_gcd(sc->time_rate, time);
        }
B
Baptiste Coudurier 已提交
1751
    }
1752 1753 1754 1755 1756

    if(edit_count > 1)
        av_log(c->fc, AV_LOG_WARNING, "multiple edit list entries, "
               "a/v desync might occur, patch welcome\n");

1757
    dprintf(c->fc, "track[%i].edit_count = %i\n", c->fc->nb_streams-1, edit_count);
B
Baptiste Coudurier 已提交
1758
    return 0;
G
Gael Chardon 已提交
1759 1760
}

1761
static const MOVParseTableEntry mov_default_parse_table[] = {
1762
{ MKTAG('a','v','s','s'), mov_read_extradata },
1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773
{ MKTAG('c','o','6','4'), mov_read_stco },
{ MKTAG('c','t','t','s'), mov_read_ctts }, /* composition time to sample */
{ MKTAG('d','i','n','f'), mov_read_default },
{ MKTAG('d','r','e','f'), mov_read_dref },
{ MKTAG('e','d','t','s'), mov_read_default },
{ MKTAG('e','l','s','t'), mov_read_elst },
{ MKTAG('e','n','d','a'), mov_read_enda },
{ MKTAG('f','i','e','l'), mov_read_extradata },
{ MKTAG('f','t','y','p'), mov_read_ftyp },
{ MKTAG('g','l','b','l'), mov_read_glbl },
{ MKTAG('h','d','l','r'), mov_read_hdlr },
1774
{ MKTAG('i','l','s','t'), mov_read_ilst },
1775 1776 1777 1778
{ MKTAG('j','p','2','h'), mov_read_extradata },
{ MKTAG('m','d','a','t'), mov_read_mdat },
{ MKTAG('m','d','h','d'), mov_read_mdhd },
{ MKTAG('m','d','i','a'), mov_read_default },
1779
{ MKTAG('m','e','t','a'), mov_read_meta },
1780 1781 1782 1783 1784 1785 1786 1787
{ MKTAG('m','i','n','f'), mov_read_default },
{ MKTAG('m','o','o','f'), mov_read_moof },
{ MKTAG('m','o','o','v'), mov_read_moov },
{ MKTAG('m','v','e','x'), mov_read_default },
{ MKTAG('m','v','h','d'), mov_read_mvhd },
{ MKTAG('S','M','I',' '), mov_read_smi }, /* Sorenson extension ??? */
{ MKTAG('a','l','a','c'), mov_read_extradata }, /* alac specific atom */
{ MKTAG('a','v','c','C'), mov_read_glbl },
1788
{ MKTAG('p','a','s','p'), mov_read_pasp },
1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801
{ MKTAG('s','t','b','l'), mov_read_default },
{ MKTAG('s','t','c','o'), mov_read_stco },
{ MKTAG('s','t','s','c'), mov_read_stsc },
{ MKTAG('s','t','s','d'), mov_read_stsd }, /* sample description */
{ MKTAG('s','t','s','s'), mov_read_stss }, /* sync sample */
{ MKTAG('s','t','s','z'), mov_read_stsz }, /* sample size */
{ MKTAG('s','t','t','s'), mov_read_stts },
{ MKTAG('t','k','h','d'), mov_read_tkhd }, /* track header */
{ MKTAG('t','f','h','d'), mov_read_tfhd }, /* track fragment header */
{ MKTAG('t','r','a','k'), mov_read_trak },
{ MKTAG('t','r','a','f'), mov_read_default },
{ MKTAG('t','r','e','x'), mov_read_trex },
{ MKTAG('t','r','u','n'), mov_read_trun },
B
Baptiste Coudurier 已提交
1802
{ MKTAG('u','d','t','a'), mov_read_default },
1803 1804 1805 1806
{ MKTAG('w','a','v','e'), mov_read_wave },
{ MKTAG('e','s','d','s'), mov_read_esds },
{ MKTAG('w','i','d','e'), mov_read_wide }, /* place holder */
{ MKTAG('c','m','o','v'), mov_read_cmov },
B
Baptiste Coudurier 已提交
1807
{ 0, NULL }
1808 1809
};

F
Fabrice Bellard 已提交
1810 1811
static int mov_probe(AVProbeData *p)
{
1812 1813
    unsigned int offset;
    uint32_t tag;
1814
    int score = 0;
1815

F
Fabrice Bellard 已提交
1816
    /* check file header */
1817 1818 1819 1820
    offset = 0;
    for(;;) {
        /* ignore invalid offset */
        if ((offset + 8) > (unsigned int)p->buf_size)
1821
            return score;
1822
        tag = AV_RL32(p->buf + offset + 4);
1823
        switch(tag) {
1824
        /* check for obvious tags */
1825 1826 1827 1828 1829
        case MKTAG('j','P',' ',' '): /* jpeg 2000 signature */
        case MKTAG('m','o','o','v'):
        case MKTAG('m','d','a','t'):
        case MKTAG('p','n','o','t'): /* detect movs with preview pics like ew.mov and april.mov */
        case MKTAG('u','d','t','a'): /* Packet Video PVAuthor adds this and a lot of more junk */
1830
        case MKTAG('f','t','y','p'):
1831
            return AVPROBE_SCORE_MAX;
1832
        /* those are more common words, so rate then a bit less */
1833 1834 1835 1836 1837
        case MKTAG('e','d','i','w'): /* xdcam files have reverted first tags */
        case MKTAG('w','i','d','e'):
        case MKTAG('f','r','e','e'):
        case MKTAG('j','u','n','k'):
        case MKTAG('p','i','c','t'):
1838
            return AVPROBE_SCORE_MAX - 5;
B
Baptiste Coudurier 已提交
1839
        case MKTAG(0x82,0x82,0x7f,0x7d):
1840 1841 1842
        case MKTAG('s','k','i','p'):
        case MKTAG('u','u','i','d'):
        case MKTAG('p','r','f','l'):
1843
            offset = AV_RB32(p->buf+offset) + offset;
1844 1845
            /* if we only find those cause probedata is too small at least rate them */
            score = AVPROBE_SCORE_MAX - 50;
1846 1847 1848
            break;
        default:
            /* unrecognized tag */
1849
            return score;
1850
        }
1851
    }
1852
    return score;
F
Fabrice Bellard 已提交
1853 1854
}

Z
Zdenek Kabelac 已提交
1855
static int mov_read_header(AVFormatContext *s, AVFormatParameters *ap)
1856
{
1857
    MOVContext *mov = s->priv_data;
1858
    ByteIOContext *pb = s->pb;
B
Baptiste Coudurier 已提交
1859
    int err;
1860
    MOVAtom atom = { 0, 0, 0 };
1861 1862

    mov->fc = s;
1863 1864
    /* .mov and .mp4 aren't streamable anyway (only progressive download if moov is before mdat) */
    if(!url_is_streamed(pb))
1865
        atom.size = url_fsize(pb);
1866
    else
B
Baptiste Coudurier 已提交
1867
        atom.size = INT64_MAX;
1868 1869

    /* check MOV header */
B
Baptiste Coudurier 已提交
1870 1871 1872 1873 1874 1875
    if ((err = mov_read_default(mov, pb, atom)) < 0) {
        av_log(s, AV_LOG_ERROR, "error reading header: %d\n", err);
        return err;
    }
    if (!mov->found_moov) {
        av_log(s, AV_LOG_ERROR, "moov atom not found\n");
1876
        return -1;
1877
    }
B
Baptiste Coudurier 已提交
1878
    dprintf(mov->fc, "on_parse_exit_offset=%lld\n", url_ftell(pb));
1879

1880 1881 1882
    return 0;
}

Z
Zdenek Kabelac 已提交
1883
static int mov_read_packet(AVFormatContext *s, AVPacket *pkt)
1884
{
1885 1886 1887
    MOVContext *mov = s->priv_data;
    MOVStreamContext *sc = 0;
    AVIndexEntry *sample = 0;
1888
    int64_t best_dts = INT64_MAX;
1889
    int i, ret;
1890
 retry:
B
Baptiste Coudurier 已提交
1891
    for (i = 0; i < s->nb_streams; i++) {
1892 1893
        AVStream *st = s->streams[i];
        MOVStreamContext *msc = st->priv_data;
1894
        if (st->discard != AVDISCARD_ALL && msc->pb && msc->current_sample < st->nb_index_entries) {
1895
            AVIndexEntry *current_sample = &st->index_entries[msc->current_sample];
1896 1897
            int64_t dts = av_rescale(current_sample->timestamp * (int64_t)msc->time_rate,
                                     AV_TIME_BASE, msc->time_scale);
B
Baptiste Coudurier 已提交
1898
            dprintf(s, "stream %d, sample %d, dts %"PRId64"\n", i, msc->current_sample, dts);
B
Baptiste Coudurier 已提交
1899 1900
            if (!sample || (url_is_streamed(s->pb) && current_sample->pos < sample->pos) ||
                (!url_is_streamed(s->pb) &&
1901
                 ((msc->pb != s->pb && dts < best_dts) || (msc->pb == s->pb &&
B
Baptiste Coudurier 已提交
1902
                 ((FFABS(best_dts - dts) <= AV_TIME_BASE && current_sample->pos < sample->pos) ||
1903
                  (FFABS(best_dts - dts) > AV_TIME_BASE && dts < best_dts)))))) {
1904 1905 1906
                sample = current_sample;
                best_dts = dts;
                sc = msc;
1907
            }
1908 1909
        }
    }
1910 1911 1912
    if (!sample) {
        mov->found_mdat = 0;
        if (!url_is_streamed(s->pb) ||
1913
            mov_read_default(mov, s->pb, (MOVAtom){ 0, 0, INT64_MAX }) < 0 ||
1914 1915 1916 1917 1918
            url_feof(s->pb))
            return -1;
        dprintf(s, "read fragments, offset 0x%llx\n", url_ftell(s->pb));
        goto retry;
    }
1919 1920
    /* must be done just before reading, to avoid infinite loop on sample */
    sc->current_sample++;
1921
    if (url_fseek(sc->pb, sample->pos, SEEK_SET) != sample->pos) {
1922 1923
        av_log(mov->fc, AV_LOG_ERROR, "stream %d, offset 0x%"PRIx64": partial file\n",
               sc->ffindex, sample->pos);
1924
        return -1;
1925
    }
1926 1927 1928
    ret = av_get_packet(sc->pb, pkt, sample->size);
    if (ret < 0)
        return ret;
1929
#if CONFIG_DV_DEMUXER
B
Baptiste Coudurier 已提交
1930 1931 1932
    if (mov->dv_demux && sc->dv_audio_container) {
        dv_produce_packet(mov->dv_demux, pkt, pkt->data, pkt->size);
        av_free(pkt->data);
1933 1934 1935
        pkt->size = 0;
        if (dv_get_packet(mov->dv_demux, pkt) < 0)
            return -1;
B
Baptiste Coudurier 已提交
1936
    }
1937
#endif
1938 1939 1940
    pkt->stream_index = sc->ffindex;
    pkt->dts = sample->timestamp;
    if (sc->ctts_data) {
1941 1942
        assert(sc->ctts_data[sc->ctts_index].duration % sc->time_rate == 0);
        pkt->pts = pkt->dts + sc->ctts_data[sc->ctts_index].duration / sc->time_rate;
1943
        /* update ctts context */
1944 1945 1946 1947 1948
        sc->ctts_sample++;
        if (sc->ctts_index < sc->ctts_count &&
            sc->ctts_data[sc->ctts_index].count == sc->ctts_sample) {
            sc->ctts_index++;
            sc->ctts_sample = 0;
1949
        }
1950 1951
        if (sc->wrong_dts)
            pkt->dts = AV_NOPTS_VALUE;
1952
    } else {
1953
        AVStream *st = s->streams[sc->ffindex];
1954
        int64_t next_dts = (sc->current_sample < st->nb_index_entries) ?
1955 1956
            st->index_entries[sc->current_sample].timestamp : st->duration;
        pkt->duration = next_dts - pkt->dts;
1957
        pkt->pts = pkt->dts;
1958
    }
1959 1960
    pkt->flags |= sample->flags & AVINDEX_KEYFRAME ? PKT_FLAG_KEY : 0;
    pkt->pos = sample->pos;
1961 1962
    dprintf(s, "stream %d, pts %"PRId64", dts %"PRId64", pos 0x%"PRIx64", duration %d\n",
            pkt->stream_index, pkt->pts, pkt->dts, pkt->pos, pkt->duration);
1963 1964
    return 0;
}
1965

1966 1967 1968 1969 1970
static int mov_seek_stream(AVStream *st, int64_t timestamp, int flags)
{
    MOVStreamContext *sc = st->priv_data;
    int sample, time_sample;
    int i;
1971

1972
    sample = av_index_search_timestamp(st, timestamp, flags);
M
Michel Bardiaux 已提交
1973
    dprintf(st->codec, "stream %d, timestamp %"PRId64", sample %d\n", st->index, timestamp, sample);
1974 1975 1976
    if (sample < 0) /* not sure what to do */
        return -1;
    sc->current_sample = sample;
B
Baptiste Coudurier 已提交
1977
    dprintf(st->codec, "stream %d, found sample %d\n", st->index, sc->current_sample);
1978 1979 1980 1981
    /* adjust ctts index */
    if (sc->ctts_data) {
        time_sample = 0;
        for (i = 0; i < sc->ctts_count; i++) {
1982 1983
            int next = time_sample + sc->ctts_data[i].count;
            if (next > sc->current_sample) {
1984 1985
                sc->ctts_index = i;
                sc->ctts_sample = sc->current_sample - time_sample;
1986
                break;
1987
            }
1988
            time_sample = next;
1989 1990
        }
    }
1991
    return sample;
1992 1993
}

1994
static int mov_read_seek(AVFormatContext *s, int stream_index, int64_t sample_time, int flags)
G
Gael Chardon 已提交
1995
{
1996 1997 1998 1999
    AVStream *st;
    int64_t seek_timestamp, timestamp;
    int sample;
    int i;
G
Gael Chardon 已提交
2000

2001
    if (stream_index >= s->nb_streams)
G
Gael Chardon 已提交
2002
        return -1;
2003 2004
    if (sample_time < 0)
        sample_time = 0;
G
Gael Chardon 已提交
2005

2006 2007 2008
    st = s->streams[stream_index];
    sample = mov_seek_stream(st, sample_time, flags);
    if (sample < 0)
G
Gael Chardon 已提交
2009 2010
        return -1;

2011 2012
    /* adjust seek timestamp to found sample timestamp */
    seek_timestamp = st->index_entries[sample].timestamp;
G
Gael Chardon 已提交
2013

2014 2015 2016 2017
    for (i = 0; i < s->nb_streams; i++) {
        st = s->streams[i];
        if (stream_index == i || st->discard == AVDISCARD_ALL)
            continue;
G
Gael Chardon 已提交
2018

2019 2020
        timestamp = av_rescale_q(seek_timestamp, s->streams[stream_index]->time_base, st->time_base);
        mov_seek_stream(st, timestamp, flags);
2021
    }
G
Gael Chardon 已提交
2022 2023 2024
    return 0;
}

Z
Zdenek Kabelac 已提交
2025
static int mov_read_close(AVFormatContext *s)
2026
{
2027
    MOVContext *mov = s->priv_data;
2028 2029 2030
    int i, j;

    for (i = 0; i < s->nb_streams; i++) {
2031 2032
        AVStream *st = s->streams[i];
        MOVStreamContext *sc = st->priv_data;
2033

2034
        av_freep(&sc->ctts_data);
2035
        for (j = 0; j < sc->drefs_count; j++)
2036 2037 2038 2039
            av_freep(&sc->drefs[j].path);
        av_freep(&sc->drefs);
        if (sc->pb && sc->pb != s->pb)
            url_fclose(sc->pb);
2040 2041

        av_freep(&st->codec->palctrl);
2042
    }
2043 2044 2045

    if (mov->dv_demux) {
        for(i = 0; i < mov->dv_fctx->nb_streams; i++) {
2046 2047 2048 2049 2050 2051
            av_freep(&mov->dv_fctx->streams[i]->codec);
            av_freep(&mov->dv_fctx->streams[i]);
        }
        av_freep(&mov->dv_fctx);
        av_freep(&mov->dv_demux);
    }
2052

B
Baptiste Coudurier 已提交
2053
    av_freep(&mov->trex_data);
2054

2055 2056 2057
    return 0;
}

2058
AVInputFormat mov_demuxer = {
2059
    "mov,mp4,m4a,3gp,3g2,mj2",
2060
    NULL_IF_CONFIG_SMALL("QuickTime/MPEG-4/Motion JPEG 2000 format"),
F
Fabrice Bellard 已提交
2061 2062
    sizeof(MOVContext),
    mov_probe,
2063 2064 2065
    mov_read_header,
    mov_read_packet,
    mov_read_close,
G
Gael Chardon 已提交
2066
    mov_read_seek,
2067
};