mov.c 85.8 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
#include "libavcodec/get_bits.h"
35

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

40 41
/*
 * First version by Francois Revol revol@free.fr
42
 * Seek function by Gael Chardon gael.dev@4now.net
43
 *
44
 * Features and limitations:
45
 * - reads most of the QT files I have (at least the structure),
46 47
 *   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 :-)
48
 *
49 50
 * Funny I didn't know about http://sourceforge.net/projects/qt-ffmpeg/
 * when coding this :) (it's a writer anyway)
51
 *
52 53 54
 * Reference documents:
 * http://www.geocities.com/xhelmboyx/quicktime/formats/qtm-layout.txt
 * Apple:
G
Gael Chardon 已提交
55
 *  http://developer.apple.com/documentation/QuickTime/QTFF/
56
 *  http://developer.apple.com/documentation/QuickTime/QTFF/qtff.pdf
57 58 59
 * QuickTime is a trademark of Apple (AFAIK :))
 */

60 61
#include "qtpalette.h"

G
Gael Chardon 已提交
62

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

66 67 68 69
/* 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 已提交
70
  0: continue to parse next atom
D
Diego Biurrun 已提交
71
 <0: error occurred, exit
B
Baptiste Coudurier 已提交
72
*/
73 74
/* links atom IDs to parse functions */
typedef struct MOVParseTableEntry {
75
    uint32_t type;
76
    int (*parse)(MOVContext *ctx, ByteIOContext *pb, MOVAtom atom);
77 78
} MOVParseTableEntry;

79 80
static const MOVParseTableEntry mov_default_parse_table[];

B
Baptiste Coudurier 已提交
81 82 83 84 85 86
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));
87
    av_metadata_set2(&c->fc->metadata, "track", buf, 0);
B
Baptiste Coudurier 已提交
88 89 90 91 92 93

    get_be16(pb); // total tracks

    return 0;
}

94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130
static const uint32_t mac_to_unicode[128] = {
    0x00C4,0x00C5,0x00C7,0x00C9,0x00D1,0x00D6,0x00DC,0x00E1,
    0x00E0,0x00E2,0x00E4,0x00E3,0x00E5,0x00E7,0x00E9,0x00E8,
    0x00EA,0x00EB,0x00ED,0x00EC,0x00EE,0x00EF,0x00F1,0x00F3,
    0x00F2,0x00F4,0x00F6,0x00F5,0x00FA,0x00F9,0x00FB,0x00FC,
    0x2020,0x00B0,0x00A2,0x00A3,0x00A7,0x2022,0x00B6,0x00DF,
    0x00AE,0x00A9,0x2122,0x00B4,0x00A8,0x2260,0x00C6,0x00D8,
    0x221E,0x00B1,0x2264,0x2265,0x00A5,0x00B5,0x2202,0x2211,
    0x220F,0x03C0,0x222B,0x00AA,0x00BA,0x03A9,0x00E6,0x00F8,
    0x00BF,0x00A1,0x00AC,0x221A,0x0192,0x2248,0x2206,0x00AB,
    0x00BB,0x2026,0x00A0,0x00C0,0x00C3,0x00D5,0x0152,0x0153,
    0x2013,0x2014,0x201C,0x201D,0x2018,0x2019,0x00F7,0x25CA,
    0x00FF,0x0178,0x2044,0x20AC,0x2039,0x203A,0xFB01,0xFB02,
    0x2021,0x00B7,0x201A,0x201E,0x2030,0x00C2,0x00CA,0x00C1,
    0x00CB,0x00C8,0x00CD,0x00CE,0x00CF,0x00CC,0x00D3,0x00D4,
    0xF8FF,0x00D2,0x00DA,0x00DB,0x00D9,0x0131,0x02C6,0x02DC,
    0x00AF,0x02D8,0x02D9,0x02DA,0x00B8,0x02DD,0x02DB,0x02C7,
};

static int mov_read_mac_string(MOVContext *c, ByteIOContext *pb, int len,
                               char *dst, int dstlen)
{
    char *p = dst;
    char *end = dst+dstlen-1;
    int i;

    for (i = 0; i < len; i++) {
        uint8_t t, c = get_byte(pb);
        if (c < 0x80 && p < end)
            *p++ = c;
        else
            PUT_UTF8(mac_to_unicode[c-0x80], t, if (p < end) *p++ = t;);
    }
    *p = 0;
    return p - dst;
}

131 132 133 134 135 136 137
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;
138 139
    uint16_t str_size, langcode = 0;
    uint32_t data_type = 0;
B
Baptiste Coudurier 已提交
140
    int (*parse)(MOVContext*, ByteIOContext*, unsigned) = NULL;
141 142 143 144

    switch (atom.type) {
    case MKTAG(0xa9,'n','a','m'): key = "title";     break;
    case MKTAG(0xa9,'a','u','t'):
145
    case MKTAG(0xa9,'A','R','T'): key = "artist";    break;
146
    case MKTAG(0xa9,'w','r','t'): key = "composer";  break;
147
    case MKTAG( 'c','p','r','t'):
148 149 150 151
    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;
152
    case MKTAG(0xa9,'d','a','y'): key = "date";      break;
153 154
    case MKTAG(0xa9,'g','e','n'): key = "genre";     break;
    case MKTAG(0xa9,'t','o','o'):
155
    case MKTAG(0xa9,'s','w','r'): key = "encoder";   break;
156
    case MKTAG(0xa9,'e','n','c'): key = "encoder";   break;
157 158 159 160 161
    case MKTAG( 'd','e','s','c'): key = "description";break;
    case MKTAG( 'l','d','e','s'): key = "synopsis";  break;
    case MKTAG( 't','v','s','h'): key = "show";      break;
    case MKTAG( 't','v','e','n'): key = "episode_id";break;
    case MKTAG( 't','v','n','n'): key = "network";   break;
B
Baptiste Coudurier 已提交
162 163
    case MKTAG( 't','r','k','n'): key = "track";
        parse = mov_metadata_trkn; break;
164 165 166 167 168 169
    }

    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')) {
170
            data_type = get_be32(pb); // type
171 172 173 174 175 176
            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
177 178
        langcode = get_be16(pb);
        ff_mov_lang_to_iso639(langcode, language);
179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195
        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 已提交
196 197 198 199

    if (parse)
        parse(c, pb, str_size);
    else {
200 201 202 203 204 205
        if (data_type == 3 || (data_type == 0 && langcode < 0x800)) { // MAC Encoded
            mov_read_mac_string(c, pb, str_size, str, sizeof(str));
        } else {
            get_buffer(pb, str, str_size);
            str[str_size] = 0;
        }
206
        av_metadata_set2(&c->fc->metadata, key, str, 0);
B
Baptiste Coudurier 已提交
207 208
        if (*language && strcmp(language, "und")) {
            snprintf(key2, sizeof(key2), "%s-%s", key, language);
209
            av_metadata_set2(&c->fc->metadata, key2, str, 0);
B
Baptiste Coudurier 已提交
210
        }
B
Baptiste Coudurier 已提交
211
    }
212 213 214 215 216 217 218 219
#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;
}
220

D
David Conrad 已提交
221 222 223
static int mov_read_chpl(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
{
    int64_t start;
D
David Conrad 已提交
224
    int i, nb_chapters, str_len, version;
D
David Conrad 已提交
225 226 227 228 229
    char str[256+1];

    if ((atom.size -= 5) < 0)
        return 0;

D
David Conrad 已提交
230 231 232 233
    version = get_byte(pb);
    get_be24(pb);
    if (version)
        get_be32(pb); // ???
D
David Conrad 已提交
234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252
    nb_chapters = get_byte(pb);

    for (i = 0; i < nb_chapters; i++) {
        if (atom.size < 9)
            return 0;

        start = get_be64(pb);
        str_len = get_byte(pb);

        if ((atom.size -= 9+str_len) < 0)
            return 0;

        get_buffer(pb, str, str_len);
        str[str_len] = 0;
        ff_new_chapter(c->fc, i, (AVRational){1,10000000}, start, AV_NOPTS_VALUE, str);
    }
    return 0;
}

253
static int mov_read_default(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
254
{
255
    int64_t total_size = 0;
256
    MOVAtom a;
257
    int i;
258

259
    if (atom.size < 0)
B
Baptiste Coudurier 已提交
260
        atom.size = INT64_MAX;
261
    while (total_size + 8 < atom.size && !url_feof(pb)) {
262
        int (*parse)(MOVContext*, ByteIOContext*, MOVAtom) = NULL;
263
        a.size = atom.size;
B
Baptiste Coudurier 已提交
264
        a.type=0;
265
        if(atom.size >= 8) {
266
            a.size = get_be32(pb);
267
            a.type = get_le32(pb);
268
        }
269 270
        dprintf(c->fc, "type: %08x '%.4s' parent:'%.4s' sz: %"PRId64" %"PRId64" %"PRId64"\n",
                a.type, (char*)&a.type, (char*)&atom.type, a.size, total_size, atom.size);
271
        total_size += 8;
272
        if (a.size == 1) { /* 64 bit extended size */
273
            a.size = get_be64(pb) - 8;
274
            total_size += 8;
275
        }
276 277 278
        if (a.size == 0) {
            a.size = atom.size - total_size;
            if (a.size <= 8)
279
                break;
280 281
        }
        a.size -= 8;
282
        if(a.size < 0)
283
            break;
B
Baptiste Coudurier 已提交
284
        a.size = FFMIN(a.size, atom.size - total_size);
285

286 287 288 289 290
        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;
            }
291

292 293 294 295 296 297
        // 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 */
298
            url_fskip(pb, a.size);
299
        } else {
300
            int64_t start_pos = url_ftell(pb);
301
            int64_t left;
302 303 304
            int err = parse(c, pb, a);
            if (err < 0)
                return err;
305 306 307
            if (c->found_moov && c->found_mdat &&
                (url_is_streamed(pb) || start_pos + a.size == url_fsize(pb)))
                return 0;
308 309 310
            left = a.size - url_ftell(pb) + start_pos;
            if (left > 0) /* skip garbage at atom end */
                url_fskip(pb, left);
311
        }
312

313
        total_size += a.size;
314 315
    }

316
    if (total_size < atom.size && atom.size < 0x7ffff)
317
        url_fskip(pb, atom.size - total_size);
318

319
    return 0;
320 321
}

322
static int mov_read_dref(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
323
{
324 325
    AVStream *st;
    MOVStreamContext *sc;
326 327
    int entries, i, j;

328 329 330 331 332
    if (c->fc->nb_streams < 1)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
    sc = st->priv_data;

333 334 335 336 337
    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));
338 339 340
    if (!sc->drefs)
        return AVERROR(ENOMEM);
    sc->drefs_count = entries;
341 342

    for (i = 0; i < sc->drefs_count; i++) {
343
        MOVDref *dref = &sc->drefs[i];
344
        uint32_t size = get_be32(pb);
345
        int64_t next = url_ftell(pb) + size - 4;
346

347 348 349
        if (size < 12)
            return -1;

350 351 352 353 354 355 356 357 358 359 360 361 362
        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;
            int16_t type;

            url_fskip(pb, 10);

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

367 368 369 370 371 372 373 374 375 376 377 378 379 380 381 382 383
            url_fskip(pb, 12);

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

            url_fskip(pb, 16);

            /* read next level up_from_alias/down_to_target */
            dref->nlvl_from = get_be16(pb);
            dref->nlvl_to   = get_be16(pb);
            av_log(c->fc, AV_LOG_DEBUG, "nlvl from %d, nlvl to %d\n",
                   dref->nlvl_from, dref->nlvl_to);

            url_fskip(pb, 16);
384 385 386 387 388 389 390 391

            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
392
                    av_free(dref->path);
393
                    dref->path = av_mallocz(len+1);
394 395
                    if (!dref->path)
                        return AVERROR(ENOMEM);
396
                    get_buffer(pb, dref->path, len);
397
                    if (len > volume_len && !strncmp(dref->path, dref->volume, volume_len)) {
398 399 400 401 402 403 404 405
                        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);
406 407 408 409 410 411 412 413 414 415 416
                } else if (type == 0) { // directory name
                    av_free(dref->dir);
                    dref->dir = av_malloc(len+1);
                    if (!dref->dir)
                        return AVERROR(ENOMEM);
                    get_buffer(pb, dref->dir, len);
                    dref->dir[len] = 0;
                    for (j = 0; j < len; j++)
                        if (dref->dir[j] == ':')
                            dref->dir[j] = '/';
                    av_log(c->fc, AV_LOG_DEBUG, "dir %s\n", dref->dir);
417 418 419 420 421 422 423 424 425
                } else
                    url_fskip(pb, len);
            }
        }
        url_fseek(pb, next, SEEK_SET);
    }
    return 0;
}

426
static int mov_read_hdlr(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
427
{
428
    AVStream *st;
429 430
    uint32_t type;
    uint32_t ctype;
431

432 433 434 435 436
    if (c->fc->nb_streams < 1) // meta before first trak
        return 0;

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

437
    get_byte(pb); /* version */
438
    get_be24(pb); /* flags */
439 440 441 442 443

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

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

447
    if     (type == MKTAG('v','i','d','e'))
448
        st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
449
    else if(type == MKTAG('s','o','u','n'))
450
        st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
451
    else if(type == MKTAG('m','1','a',' '))
452
        st->codec->codec_id = CODEC_ID_MP2;
453
    else if(type == MKTAG('s','u','b','p'))
454
        st->codec->codec_type = AVMEDIA_TYPE_SUBTITLE;
455

456 457 458 459 460 461 462
    get_be32(pb); /* component  manufacture */
    get_be32(pb); /* component flags */
    get_be32(pb); /* component flags mask */

    return 0;
}

463
int ff_mov_read_esds(AVFormatContext *fc, ByteIOContext *pb, MOVAtom atom)
464
{
465
    AVStream *st;
466
    int tag, len;
467

468
    if (fc->nb_streams < 1)
469
        return 0;
470
    st = fc->streams[fc->nb_streams-1];
471

472
    get_be32(pb); /* version + flags */
473
    len = ff_mp4_read_descr(fc, pb, &tag);
474
    if (tag == MP4ESDescrTag) {
475 476
        get_be16(pb); /* ID */
        get_byte(pb); /* priority */
477
    } else
478
        get_be16(pb); /* ID */
479

480 481 482
    len = ff_mp4_read_descr(fc, pb, &tag);
    if (tag == MP4DecConfigDescrTag)
        ff_mp4_read_dec_config_descr(fc, st, pb);
483 484 485
    return 0;
}

486 487 488 489 490
static int mov_read_esds(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
{
    return ff_mov_read_esds(c->fc, pb, atom);
}

491 492 493 494 495
static int mov_read_dac3(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
{
    AVStream *st;
    int ac3info, acmod, lfeon;

496 497
    if (c->fc->nb_streams < 1)
        return 0;
498 499 500 501 502 503 504 505 506 507
    st = c->fc->streams[c->fc->nb_streams-1];

    ac3info = get_be24(pb);
    acmod = (ac3info >> 11) & 0x7;
    lfeon = (ac3info >> 10) & 0x1;
    st->codec->channels = ((int[]){2,1,2,3,3,4,4,5})[acmod] + lfeon;

    return 0;
}

508 509 510 511
static int mov_read_pasp(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
{
    const int num = get_be32(pb);
    const int den = get_be32(pb);
512 513 514 515 516 517
    AVStream *st;

    if (c->fc->nb_streams < 1)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];

518 519 520 521 522 523 524
    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)) {
        av_log(c->fc, AV_LOG_WARNING,
               "sample aspect ratio already set to %d:%d, ignoring 'pasp' atom (%d:%d)\n",
               st->sample_aspect_ratio.num, st->sample_aspect_ratio.den,
               num, den);
    } else if (den != 0) {
525 526 527 528 529 530
        st->sample_aspect_ratio.num = num;
        st->sample_aspect_ratio.den = den;
    }
    return 0;
}

531
/* this atom contains actual media data */
532
static int mov_read_mdat(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
533
{
534 535 536 537 538 539
    if(atom.size == 0) /* wrong one (MP4) */
        return 0;
    c->found_mdat=1;
    return 0; /* now go for moov */
}

540
/* read major brand, minor version and compatible brands and store them as metadata */
541
static int mov_read_ftyp(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
542
{
543 544 545 546 547 548 549
    uint32_t minor_ver;
    int comp_brand_size;
    char minor_ver_str[11]; /* 32 bit integer -> 10 digits + null */
    char* comp_brands_str;
    uint8_t type[5] = {0};

    get_buffer(pb, type, 4);
B
Baptiste Coudurier 已提交
550
    if (strcmp(type, "qt  "))
551
        c->isom = 1;
552
    av_log(c->fc, AV_LOG_DEBUG, "ISO: File Type Major Brand: %.4s\n",(char *)&type);
553
    av_metadata_set2(&c->fc->metadata, "major_brand", type, 0);
554 555
    minor_ver = get_be32(pb); /* minor version */
    snprintf(minor_ver_str, sizeof(minor_ver_str), "%d", minor_ver);
556
    av_metadata_set2(&c->fc->metadata, "minor_version", minor_ver_str, 0);
557 558 559 560 561 562 563 564 565

    comp_brand_size = atom.size - 8;
    if (comp_brand_size < 0)
        return -1;
    comp_brands_str = av_malloc(comp_brand_size + 1); /* Add null terminator */
    if (!comp_brands_str)
        return AVERROR(ENOMEM);
    get_buffer(pb, comp_brands_str, comp_brand_size);
    comp_brands_str[comp_brand_size] = 0;
566
    av_metadata_set2(&c->fc->metadata, "compatible_brands", comp_brands_str, 0);
567 568
    av_freep(&comp_brands_str);

569 570 571
    return 0;
}

572
/* this atom should contain all header atoms */
573
static int mov_read_moov(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
574
{
575 576
    if (mov_read_default(c, pb, atom) < 0)
        return -1;
577 578 579 580 581 582
    /* 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 */
}

583
static int mov_read_moof(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
584 585 586 587 588
{
    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);
}
589

590 591 592 593
static void mov_metadata_creation_time(AVMetadata **metadata, time_t time)
{
    char buffer[32];
    if (time) {
594
        struct tm *ptm;
595
        time -= 2082844800;  /* seconds between 1904-01-01 and Epoch */
596 597 598
        ptm = gmtime(&time);
        if (!ptm) return;
        strftime(buffer, sizeof(buffer), "%Y-%m-%d %H:%M:%S", ptm);
599 600 601 602
        av_metadata_set2(metadata, "creation_time", buffer, 0);
    }
}

603
static int mov_read_mdhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
604
{
605 606 607
    AVStream *st;
    MOVStreamContext *sc;
    int version;
608
    char language[4] = {0};
609
    unsigned lang;
610
    time_t creation_time;
611

612 613 614 615 616 617
    if (c->fc->nb_streams < 1)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
    sc = st->priv_data;

    version = get_byte(pb);
618
    if (version > 1)
B
Baptiste Coudurier 已提交
619
        return -1; /* unsupported */
620

621
    get_be24(pb); /* flags */
B
clean  
Baptiste Coudurier 已提交
622
    if (version == 1) {
623
        creation_time = get_be64(pb);
B
clean  
Baptiste Coudurier 已提交
624 625
        get_be64(pb);
    } else {
626
        creation_time = get_be32(pb);
B
clean  
Baptiste Coudurier 已提交
627 628
        get_be32(pb); /* modification time */
    }
629
    mov_metadata_creation_time(&st->metadata, creation_time);
630

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

634
    lang = get_be16(pb); /* language */
635
    if (ff_mov_lang_to_iso639(lang, language))
636
        av_metadata_set2(&st->metadata, "language", language, 0);
637 638 639 640 641
    get_be16(pb); /* quality */

    return 0;
}

642
static int mov_read_mvhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
643
{
644
    time_t creation_time;
B
Baptiste Coudurier 已提交
645
    int version = get_byte(pb); /* version */
646
    get_be24(pb); /* flags */
647

B
Baptiste Coudurier 已提交
648
    if (version == 1) {
649
        creation_time = get_be64(pb);
B
Baptiste Coudurier 已提交
650 651
        get_be64(pb);
    } else {
652
        creation_time = get_be32(pb);
B
Baptiste Coudurier 已提交
653 654
        get_be32(pb); /* modification time */
    }
655
    mov_metadata_creation_time(&c->fc->metadata, creation_time);
656
    c->time_scale = get_be32(pb); /* time scale */
657 658 659

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

B
Baptiste Coudurier 已提交
660
    c->duration = (version == 1) ? get_be64(pb) : get_be32(pb); /* duration */
661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679
    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;
}

680
static int mov_read_smi(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
681
{
682 683 684 685 686
    AVStream *st;

    if (c->fc->nb_streams < 1)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
687

688 689
    if((uint64_t)atom.size > (1<<30))
        return -1;
690

691 692
    // currently SVQ3 decoder expect full STSD header - so let's fake it
    // this should be fixed and just SMI header should be passed
693
    av_free(st->codec->extradata);
694 695 696
    st->codec->extradata = av_mallocz(atom.size + 0x5a + FF_INPUT_BUFFER_PADDING_SIZE);
    if (!st->codec->extradata)
        return AVERROR(ENOMEM);
B
Baptiste Coudurier 已提交
697 698 699 700
    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);
701 702
    return 0;
}
703

704
static int mov_read_enda(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
705
{
706 707 708 709 710 711
    AVStream *st;
    int little_endian;

    if (c->fc->nb_streams < 1)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
712

713
    little_endian = get_be16(pb);
714 715
    dprintf(c->fc, "enda %d\n", little_endian);
    if (little_endian == 1) {
716 717 718 719 720 721 722
        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;
723 724 725 726 727 728
        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;
729 730 731 732 733 734 735
        default:
            break;
        }
    }
    return 0;
}

736
/* FIXME modify qdm2/svq3/h264 decoders to take full atom as extradata */
737
static int mov_read_extradata(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
738
{
739 740
    AVStream *st;
    uint64_t size;
741
    uint8_t *buf;
742 743 744 745 746

    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;
747
    if(size > INT_MAX || (uint64_t)atom.size > INT_MAX)
748
        return -1;
749 750 751 752 753 754 755 756 757
    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);
758 759 760
    return 0;
}

761
static int mov_read_wave(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
R
Roberto Togni 已提交
762
{
763 764 765 766 767
    AVStream *st;

    if (c->fc->nb_streams < 1)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
R
Roberto Togni 已提交
768 769 770

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

772 773 774
    if (st->codec->codec_id == CODEC_ID_QDM2) {
        // pass all frma atom to codec, needed at least for QDM2
        av_free(st->codec->extradata);
775 776 777
        st->codec->extradata = av_mallocz(atom.size + FF_INPUT_BUFFER_PADDING_SIZE);
        if (!st->codec->extradata)
            return AVERROR(ENOMEM);
778
        st->codec->extradata_size = atom.size;
779
        get_buffer(pb, st->codec->extradata, atom.size);
780
    } else if (atom.size > 8) { /* to read frma, esds atoms */
781 782
        if (mov_read_default(c, pb, atom) < 0)
            return -1;
783
    } else
784
        url_fskip(pb, atom.size);
R
Roberto Togni 已提交
785 786 787
    return 0;
}

788 789 790 791
/**
 * This function reads atom content and puts data in extradata without tag
 * nor size unlike mov_read_extradata.
 */
792
static int mov_read_glbl(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
793
{
794 795 796 797 798
    AVStream *st;

    if (c->fc->nb_streams < 1)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
799

800 801 802
    if((uint64_t)atom.size > (1<<30))
        return -1;

803
    av_free(st->codec->extradata);
804 805 806
    st->codec->extradata = av_mallocz(atom.size + FF_INPUT_BUFFER_PADDING_SIZE);
    if (!st->codec->extradata)
        return AVERROR(ENOMEM);
807
    st->codec->extradata_size = atom.size;
808
    get_buffer(pb, st->codec->extradata, atom.size);
809 810 811
    return 0;
}

M
Martin Storsjö 已提交
812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839
/**
 * An strf atom is a BITMAPINFOHEADER struct. This struct is 40 bytes itself,
 * but can have extradata appended at the end after the 40 bytes belonging
 * to the struct.
 */
static int mov_read_strf(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
{
    AVStream *st;

    if (c->fc->nb_streams < 1)
        return 0;
    if (atom.size <= 40)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];

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

    av_free(st->codec->extradata);
    st->codec->extradata = av_mallocz(atom.size - 40 + FF_INPUT_BUFFER_PADDING_SIZE);
    if (!st->codec->extradata)
        return AVERROR(ENOMEM);
    st->codec->extradata_size = atom.size - 40;
    url_fskip(pb, 40);
    get_buffer(pb, st->codec->extradata, atom.size - 40);
    return 0;
}

840
static int mov_read_stco(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
841
{
842 843
    AVStream *st;
    MOVStreamContext *sc;
844
    unsigned int i, entries;
845

846 847 848 849 850
    if (c->fc->nb_streams < 1)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
    sc = st->priv_data;

851
    get_byte(pb); /* version */
852
    get_be24(pb); /* flags */
853 854

    entries = get_be32(pb);
855

856 857
    if(entries >= UINT_MAX/sizeof(int64_t))
        return -1;
858

B
Baptiste Coudurier 已提交
859
    sc->chunk_offsets = av_malloc(entries * sizeof(int64_t));
860
    if (!sc->chunk_offsets)
861 862 863
        return AVERROR(ENOMEM);
    sc->chunk_count = entries;

864 865
    if      (atom.type == MKTAG('s','t','c','o'))
        for(i=0; i<entries; i++)
866
            sc->chunk_offsets[i] = get_be32(pb);
867 868
    else if (atom.type == MKTAG('c','o','6','4'))
        for(i=0; i<entries; i++)
869
            sc->chunk_offsets[i] = get_be64(pb);
870
    else
871
        return -1;
872

873 874 875
    return 0;
}

876 877 878 879
/**
 * Compute codec id for 'lpcm' tag.
 * See CoreAudioTypes and AudioStreamBasicDescription at Apple.
 */
880
enum CodecID ff_mov_get_lpcm_codec_id(int bps, int flags)
881 882 883 884
{
    if (flags & 1) { // floating point
        if (flags & 2) { // big endian
            if      (bps == 32) return CODEC_ID_PCM_F32BE;
885
            else if (bps == 64) return CODEC_ID_PCM_F64BE;
886
        } else {
887 888
            if      (bps == 32) return CODEC_ID_PCM_F32LE;
            else if (bps == 64) return CODEC_ID_PCM_F64LE;
889 890 891 892 893 894 895 896 897 898 899 900 901 902
        }
    } 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 已提交
903
            else if (bps == 16) return CODEC_ID_PCM_S16LE;
904 905 906 907
            else if (bps == 24) return CODEC_ID_PCM_S24LE;
            else if (bps == 32) return CODEC_ID_PCM_S32LE;
        }
    }
D
Diego Pettenò 已提交
908
    return CODEC_ID_NONE;
909 910
}

911
int ff_mov_read_stsd_entries(MOVContext *c, ByteIOContext *pb, int entries)
912
{
913 914
    AVStream *st;
    MOVStreamContext *sc;
915
    int j, pseudo_stream_id;
916

917 918 919 920 921
    if (c->fc->nb_streams < 1)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
    sc = st->priv_data;

922 923
    for(pseudo_stream_id=0; pseudo_stream_id<entries; pseudo_stream_id++) {
        //Parsing Sample description table
924
        enum CodecID id;
925
        int dref_id = 1;
926
        MOVAtom a = { AV_RL32("stsd") };
927
        int64_t start_pos = url_ftell(pb);
928
        int size = get_be32(pb); /* size */
929
        uint32_t format = get_le32(pb); /* data format */
930

931 932 933 934 935
        if (size >= 16) {
            get_be32(pb); /* reserved */
            get_be16(pb); /* reserved */
            dref_id = get_be16(pb);
        }
936

937
        if (st->codec->codec_tag &&
938
            st->codec->codec_tag != format &&
939
            (c->fc->video_codec_id ? ff_codec_get_id(codec_movvideo_tags, format) != c->fc->video_codec_id
940
                                   : st->codec->codec_tag != MKTAG('j','p','e','g'))
941
           ){
D
Diego Biurrun 已提交
942 943 944
            /* 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. */
945
        multiple_stsd:
B
Baptiste Coudurier 已提交
946
            av_log(c->fc, AV_LOG_WARNING, "multiple fourcc not supported\n");
947 948 949
            url_fskip(pb, size - (url_ftell(pb) - start_pos));
            continue;
        }
950 951 952
        /* we cannot demux concatenated h264 streams because of different extradata */
        if (st->codec->codec_tag && st->codec->codec_tag == AV_RL32("avc1"))
            goto multiple_stsd;
953
        sc->pseudo_stream_id = st->codec->codec_tag ? -1 : pseudo_stream_id;
954
        sc->dref_id= dref_id;
955

956
        st->codec->codec_tag = format;
957
        id = ff_codec_get_id(codec_movaudio_tags, format);
958
        if (id<=0 && ((format&0xFFFF) == 'm'+('s'<<8) || (format&0xFFFF) == 'T'+('S'<<8)))
M
Måns Rullgård 已提交
959
            id = ff_codec_get_id(ff_codec_wav_tags, av_bswap32(format)&0xFFFF);
M
Michael Niedermayer 已提交
960

961 962 963
        if (st->codec->codec_type != AVMEDIA_TYPE_VIDEO && id > 0) {
            st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
        } else if (st->codec->codec_type != AVMEDIA_TYPE_AUDIO && /* do not overwrite codec type */
964
                   format && format != MKTAG('m','p','4','s')) { /* skip old asf mpeg4 tag */
965
            id = ff_codec_get_id(codec_movvideo_tags, format);
966
            if (id <= 0)
967
                id = ff_codec_get_id(ff_codec_bmp_tags, format);
968
            if (id > 0)
969 970
                st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
            else if(st->codec->codec_type == AVMEDIA_TYPE_DATA){
971
                id = ff_codec_get_id(ff_codec_movsubtitle_tags, format);
972
                if(id > 0)
973
                    st->codec->codec_type = AVMEDIA_TYPE_SUBTITLE;
974
            }
975 976
        }

977 978 979
        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);
980

981
        if(st->codec->codec_type==AVMEDIA_TYPE_VIDEO) {
982
            unsigned int color_depth, len;
983 984
            int color_greyscale;

985
            st->codec->codec_id = id;
986 987 988 989
            get_be16(pb); /* version */
            get_be16(pb); /* revision level */
            get_be32(pb); /* vendor */
            get_be32(pb); /* temporal quality */
D
Diego Biurrun 已提交
990
            get_be32(pb); /* spatial quality */
991 992 993 994

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

995 996 997
            get_be32(pb); /* horiz resolution */
            get_be32(pb); /* vert resolution */
            get_be32(pb); /* data size, always 0 */
B
Baptiste Coudurier 已提交
998
            get_be16(pb); /* frames per samples */
999

1000 1001 1002 1003 1004 1005 1006 1007 1008
            len = get_byte(pb); /* codec name, pascal string */
            if (len > 31)
                len = 31;
            mov_read_mac_string(c, pb, len, st->codec->codec_name, 32);
            if (len < 31)
                url_fskip(pb, 31 - len);
            /* codec_tag YV12 triggers an UV swap in rawdec.c */
            if (!memcmp(st->codec->codec_name, "Planar Y'CbCr 8-bit 4:2:0", 25))
                st->codec->codec_tag=MKTAG('I', '4', '2', '0');
1009

1010
            st->codec->bits_per_coded_sample = get_be16(pb); /* depth */
1011
            st->codec->color_table_id = get_be16(pb); /* colortable id */
1012
            dprintf(c->fc, "depth %d, ctab id %d\n",
1013
                   st->codec->bits_per_coded_sample, st->codec->color_table_id);
1014
            /* figure out the palette situation */
1015 1016
            color_depth = st->codec->bits_per_coded_sample & 0x1F;
            color_greyscale = st->codec->bits_per_coded_sample & 0x20;
1017 1018

            /* if the depth is 2, 4, or 8 bpp, file is palettized */
1019
            if ((color_depth == 2) || (color_depth == 4) ||
1020
                (color_depth == 8)) {
1021 1022 1023 1024
                /* for palette traversal */
                unsigned int color_start, color_count, color_end;
                unsigned char r, g, b;

1025
                st->codec->palctrl = av_malloc(sizeof(*st->codec->palctrl));
1026
                if (color_greyscale) {
1027
                    int color_index, color_dec;
1028
                    /* compute the greyscale palette */
1029
                    st->codec->bits_per_coded_sample = color_depth;
1030 1031 1032 1033 1034
                    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;
1035
                        st->codec->palctrl->palette[j] =
1036 1037 1038 1039 1040
                            (r << 16) | (g << 8) | (b);
                        color_index -= color_dec;
                        if (color_index < 0)
                            color_index = 0;
                    }
1041
                } else if (st->codec->color_table_id) {
1042
                    const uint8_t *color_table;
1043 1044 1045
                    /* if flag bit 3 is set, use the default palette */
                    color_count = 1 << color_depth;
                    if (color_depth == 2)
M
Michael Niedermayer 已提交
1046
                        color_table = ff_qt_default_palette_4;
1047
                    else if (color_depth == 4)
M
Michael Niedermayer 已提交
1048
                        color_table = ff_qt_default_palette_16;
1049
                    else
M
Michael Niedermayer 已提交
1050
                        color_table = ff_qt_default_palette_256;
1051 1052

                    for (j = 0; j < color_count; j++) {
1053 1054 1055
                        r = color_table[j * 3 + 0];
                        g = color_table[j * 3 + 1];
                        b = color_table[j * 3 + 2];
1056
                        st->codec->palctrl->palette[j] =
1057 1058 1059 1060 1061 1062 1063
                            (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);
1064 1065
                    if ((color_start <= 255) &&
                        (color_end <= 255)) {
M
Mike Melanson 已提交
1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077
                        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);
1078
                            st->codec->palctrl->palette[j] =
M
Mike Melanson 已提交
1079
                                (r << 16) | (g << 8) | (b);
1080
                        }
1081 1082
                    }
                }
1083
                st->codec->palctrl->palette_changed = 1;
1084
            }
1085
        } else if(st->codec->codec_type==AVMEDIA_TYPE_AUDIO) {
1086
            int bits_per_sample, flags;
1087
            uint16_t version = get_be16(pb);
1088

1089
            st->codec->codec_id = id;
1090 1091
            get_be16(pb); /* revision level */
            get_be32(pb); /* vendor */
1092

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

1097
            sc->audio_cid = get_be16(pb);
1098 1099 1100 1101
            get_be16(pb); /* packet size = 0 */

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

1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114
            //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 */
1115
                    st->codec->bits_per_coded_sample = get_be32(pb); /* bits per channel if sound is uncompressed */
J
Jai Menon 已提交
1116
                    flags = get_be32(pb); /* lpcm format specific flag */
B
Baptiste Coudurier 已提交
1117 1118
                    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 */
1119
                    if (format == MKTAG('l','p','c','m'))
1120
                        st->codec->codec_id = ff_mov_get_lpcm_codec_id(st->codec->bits_per_coded_sample, flags);
1121 1122 1123
                }
            }

1124
            switch (st->codec->codec_id) {
1125 1126
            case CODEC_ID_PCM_S8:
            case CODEC_ID_PCM_U8:
1127
                if (st->codec->bits_per_coded_sample == 16)
1128 1129
                    st->codec->codec_id = CODEC_ID_PCM_S16BE;
                break;
1130
            case CODEC_ID_PCM_S16LE:
1131
            case CODEC_ID_PCM_S16BE:
1132
                if (st->codec->bits_per_coded_sample == 8)
1133
                    st->codec->codec_id = CODEC_ID_PCM_S8;
1134
                else if (st->codec->bits_per_coded_sample == 24)
1135 1136 1137
                    st->codec->codec_id =
                        st->codec->codec_id == CODEC_ID_PCM_S16BE ?
                        CODEC_ID_PCM_S24BE : CODEC_ID_PCM_S24LE;
1138
                break;
1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151
            /* 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;
1152 1153 1154 1155
            case CODEC_ID_GSM:
                sc->samples_per_frame = 160;
                sc->bytes_per_frame = 33;
                break;
1156 1157
            default:
                break;
1158
            }
1159

1160 1161
            bits_per_sample = av_get_bits_per_sample(st->codec->codec_id);
            if (bits_per_sample) {
1162
                st->codec->bits_per_coded_sample = bits_per_sample;
1163 1164
                sc->sample_size = (bits_per_sample >> 3) * st->codec->channels;
            }
1165
        } else if(st->codec->codec_type==AVMEDIA_TYPE_SUBTITLE){
1166 1167 1168
            // 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) };
1169
            if (format != AV_RL32("mp4s")) // mp4s contains a regular esds atom
R
Reimar Döffinger 已提交
1170
                mov_read_glbl(c, pb, fake_atom);
1171
            st->codec->codec_id= id;
1172 1173
            st->codec->width = sc->width;
            st->codec->height = sc->height;
1174 1175 1176
        } else {
            /* other codec type, just skip (rtp, mp4s, tmcd ...) */
            url_fskip(pb, size - (url_ftell(pb) - start_pos));
1177
        }
1178 1179
        /* this will read extra atoms at the end (wave, alac, damr, avcC, SMI ...) */
        a.size = size - (url_ftell(pb) - start_pos);
1180 1181 1182 1183
        if (a.size > 8) {
            if (mov_read_default(c, pb, a) < 0)
                return -1;
        } else if (a.size > 0)
1184
            url_fskip(pb, a.size);
1185
    }
1186

1187
    if(st->codec->codec_type==AVMEDIA_TYPE_AUDIO && st->codec->sample_rate==0 && sc->time_scale>1)
1188
        st->codec->sample_rate= sc->time_scale;
1189

1190
    /* special codec parameters handling */
1191
    switch (st->codec->codec_id) {
1192
#if CONFIG_DV_DEMUXER
1193
    case CODEC_ID_DVAUDIO:
1194
        c->dv_fctx = avformat_alloc_context();
1195 1196 1197 1198 1199 1200 1201 1202
        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;
1203
#endif
1204
    /* no ifdef since parameters are always those */
1205
    case CODEC_ID_QCELP:
1206 1207 1208
        // force sample rate for qcelp when not stored in mov
        if (st->codec->codec_tag != MKTAG('Q','c','l','p'))
            st->codec->sample_rate = 8000;
1209
        st->codec->frame_size= 160;
1210
        st->codec->channels= 1; /* really needed */
1211
        break;
1212
    case CODEC_ID_AMR_NB:
1213
    case CODEC_ID_AMR_WB:
B
Baptiste Coudurier 已提交
1214
        st->codec->frame_size= sc->samples_per_frame;
1215
        st->codec->channels= 1; /* really needed */
1216
        /* force sample rate for amr, stsd in 3gp does not store sample rate */
B
Baptiste Coudurier 已提交
1217
        if (st->codec->codec_id == CODEC_ID_AMR_NB)
1218
            st->codec->sample_rate = 8000;
B
Baptiste Coudurier 已提交
1219 1220
        else if (st->codec->codec_id == CODEC_ID_AMR_WB)
            st->codec->sample_rate = 16000;
1221
        break;
1222
    case CODEC_ID_MP2:
1223
    case CODEC_ID_MP3:
1224
        st->codec->codec_type = AVMEDIA_TYPE_AUDIO; /* force type after stsd for m1a hdlr */
A
Aurelien Jacobs 已提交
1225
        st->need_parsing = AVSTREAM_PARSE_FULL;
1226
        break;
1227
    case CODEC_ID_GSM:
1228 1229 1230 1231
    case CODEC_ID_ADPCM_MS:
    case CODEC_ID_ADPCM_IMA_WAV:
        st->codec->block_align = sc->bytes_per_frame;
        break;
1232
    case CODEC_ID_ALAC:
1233
        if (st->codec->extradata_size == 36) {
1234 1235
            st->codec->frame_size = AV_RB32(st->codec->extradata+12);
            st->codec->channels   = AV_RB8 (st->codec->extradata+21);
1236
            st->codec->sample_rate = AV_RB32(st->codec->extradata+32);
1237
        }
1238
        break;
1239 1240 1241
    default:
        break;
    }
1242

1243 1244 1245
    return 0;
}

1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256
static int mov_read_stsd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
{
    int entries;

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

    return ff_mov_read_stsd_entries(c, pb, entries);
}

1257
static int mov_read_stsc(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1258
{
1259 1260
    AVStream *st;
    MOVStreamContext *sc;
1261
    unsigned int i, entries;
1262

1263 1264 1265 1266 1267
    if (c->fc->nb_streams < 1)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
    sc = st->priv_data;

1268
    get_byte(pb); /* version */
1269
    get_be24(pb); /* flags */
1270 1271

    entries = get_be32(pb);
1272

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

1275 1276
    if(entries >= UINT_MAX / sizeof(*sc->stsc_data))
        return -1;
1277 1278
    sc->stsc_data = av_malloc(entries * sizeof(*sc->stsc_data));
    if (!sc->stsc_data)
1279 1280 1281
        return AVERROR(ENOMEM);
    sc->stsc_count = entries;

1282
    for(i=0; i<entries; i++) {
1283 1284 1285
        sc->stsc_data[i].first = get_be32(pb);
        sc->stsc_data[i].count = get_be32(pb);
        sc->stsc_data[i].id = get_be32(pb);
1286 1287 1288 1289
    }
    return 0;
}

1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318
static int mov_read_stps(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
{
    AVStream *st;
    MOVStreamContext *sc;
    unsigned i, entries;

    if (c->fc->nb_streams < 1)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
    sc = st->priv_data;

    get_be32(pb); // version + flags

    entries = get_be32(pb);
    if (entries >= UINT_MAX / sizeof(*sc->stps_data))
        return -1;
    sc->stps_data = av_malloc(entries * sizeof(*sc->stps_data));
    if (!sc->stps_data)
        return AVERROR(ENOMEM);
    sc->stps_count = entries;

    for (i = 0; i < entries; i++) {
        sc->stps_data[i] = get_be32(pb);
        //dprintf(c->fc, "stps %d\n", sc->stps_data[i]);
    }

    return 0;
}

1319
static int mov_read_stss(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1320
{
1321 1322
    AVStream *st;
    MOVStreamContext *sc;
1323
    unsigned int i, entries;
1324

1325 1326 1327 1328 1329
    if (c->fc->nb_streams < 1)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
    sc = st->priv_data;

1330
    get_byte(pb); /* version */
1331
    get_be24(pb); /* flags */
1332 1333

    entries = get_be32(pb);
1334

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

B
Baptiste Coudurier 已提交
1337
    if(entries >= UINT_MAX / sizeof(int))
1338
        return -1;
B
Baptiste Coudurier 已提交
1339
    sc->keyframes = av_malloc(entries * sizeof(int));
1340
    if (!sc->keyframes)
1341 1342 1343
        return AVERROR(ENOMEM);
    sc->keyframe_count = entries;

1344 1345
    for(i=0; i<entries; i++) {
        sc->keyframes[i] = get_be32(pb);
1346
        //dprintf(c->fc, "keyframes[]=%d\n", sc->keyframes[i]);
1347 1348 1349 1350
    }
    return 0;
}

1351
static int mov_read_stsz(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1352
{
1353 1354
    AVStream *st;
    MOVStreamContext *sc;
1355 1356 1357
    unsigned int i, entries, sample_size, field_size, num_bytes;
    GetBitContext gb;
    unsigned char* buf;
1358

1359 1360 1361 1362 1363
    if (c->fc->nb_streams < 1)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
    sc = st->priv_data;

1364
    get_byte(pb); /* version */
1365
    get_be24(pb); /* flags */
1366

1367
    if (atom.type == MKTAG('s','t','s','z')) {
1368 1369 1370 1371
        sample_size = get_be32(pb);
        if (!sc->sample_size) /* do not overwrite value computed in stsd */
            sc->sample_size = sample_size;
        field_size = 32;
1372 1373 1374 1375 1376
    } else {
        sample_size = 0;
        get_be24(pb); /* reserved */
        field_size = get_byte(pb);
    }
1377
    entries = get_be32(pb);
1378 1379

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

1381
    sc->sample_count = entries;
1382 1383 1384
    if (sample_size)
        return 0;

1385 1386 1387 1388 1389
    if (field_size != 4 && field_size != 8 && field_size != 16 && field_size != 32) {
        av_log(c->fc, AV_LOG_ERROR, "Invalid sample field size %d\n", field_size);
        return -1;
    }

1390
    if (entries >= UINT_MAX / sizeof(int) || entries >= (UINT_MAX - 4) / field_size)
1391
        return -1;
B
Baptiste Coudurier 已提交
1392
    sc->sample_sizes = av_malloc(entries * sizeof(int));
1393
    if (!sc->sample_sizes)
1394 1395
        return AVERROR(ENOMEM);

1396 1397
    num_bytes = (entries*field_size+4)>>3;

1398
    buf = av_malloc(num_bytes+FF_INPUT_BUFFER_PADDING_SIZE);
1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411
    if (!buf) {
        av_freep(&sc->sample_sizes);
        return AVERROR(ENOMEM);
    }

    if (get_buffer(pb, buf, num_bytes) < num_bytes) {
        av_freep(&sc->sample_sizes);
        av_free(buf);
        return -1;
    }

    init_get_bits(&gb, buf, 8*num_bytes);

1412
    for(i=0; i<entries; i++)
1413 1414 1415
        sc->sample_sizes[i] = get_bits_long(&gb, field_size);

    av_free(buf);
1416 1417 1418
    return 0;
}

1419
static int mov_read_stts(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1420
{
1421 1422
    AVStream *st;
    MOVStreamContext *sc;
1423
    unsigned int i, entries;
1424 1425
    int64_t duration=0;
    int64_t total_sample_count=0;
1426

1427 1428 1429 1430 1431
    if (c->fc->nb_streams < 1)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
    sc = st->priv_data;

1432
    get_byte(pb); /* version */
1433
    get_be24(pb); /* flags */
1434
    entries = get_be32(pb);
1435 1436 1437

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

1438
    if(entries >= UINT_MAX / sizeof(*sc->stts_data))
1439
        return -1;
1440
    sc->stts_data = av_malloc(entries * sizeof(*sc->stts_data));
1441
    if (!sc->stts_data)
1442 1443
        return AVERROR(ENOMEM);
    sc->stts_count = entries;
1444

1445
    for(i=0; i<entries; i++) {
M
cleanup  
Michael Niedermayer 已提交
1446 1447
        int sample_duration;
        int sample_count;
1448

1449
        sample_count=get_be32(pb);
1450
        sample_duration = get_be32(pb);
1451 1452 1453
        sc->stts_data[i].count= sample_count;
        sc->stts_data[i].duration= sample_duration;

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

B
Baptiste Coudurier 已提交
1456
        duration+=(int64_t)sample_duration*sample_count;
1457 1458 1459
        total_sample_count+=sample_count;
    }

1460 1461 1462
    st->nb_frames= total_sample_count;
    if(duration)
        st->duration= duration;
1463 1464 1465
    return 0;
}

1466
static int mov_read_ctts(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
M
Michael Niedermayer 已提交
1467
{
1468 1469
    AVStream *st;
    MOVStreamContext *sc;
M
Michael Niedermayer 已提交
1470 1471
    unsigned int i, entries;

1472 1473 1474 1475 1476
    if (c->fc->nb_streams < 1)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
    sc = st->priv_data;

M
Michael Niedermayer 已提交
1477
    get_byte(pb); /* version */
1478
    get_be24(pb); /* flags */
M
Michael Niedermayer 已提交
1479
    entries = get_be32(pb);
1480 1481 1482

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

1483
    if(entries >= UINT_MAX / sizeof(*sc->ctts_data))
M
Michael Niedermayer 已提交
1484
        return -1;
1485
    sc->ctts_data = av_malloc(entries * sizeof(*sc->ctts_data));
1486
    if (!sc->ctts_data)
1487 1488
        return AVERROR(ENOMEM);
    sc->ctts_count = entries;
1489

M
Michael Niedermayer 已提交
1490
    for(i=0; i<entries; i++) {
1491 1492 1493 1494 1495
        int count    =get_be32(pb);
        int duration =get_be32(pb);

        sc->ctts_data[i].count   = count;
        sc->ctts_data[i].duration= duration;
1496 1497
        if (duration < 0)
            sc->dts_shift = FFMAX(sc->dts_shift, -duration);
M
Michael Niedermayer 已提交
1498
    }
1499

B
Baptiste Coudurier 已提交
1500
    dprintf(c->fc, "dts shift %d\n", sc->dts_shift);
1501

M
Michael Niedermayer 已提交
1502 1503 1504
    return 0;
}

1505 1506 1507
static void mov_build_index(MOVContext *mov, AVStream *st)
{
    MOVStreamContext *sc = st->priv_data;
1508
    int64_t current_offset;
1509 1510 1511 1512
    int64_t current_dts = 0;
    unsigned int stts_index = 0;
    unsigned int stsc_index = 0;
    unsigned int stss_index = 0;
1513
    unsigned int stps_index = 0;
1514
    unsigned int i, j;
1515
    uint64_t stream_size = 0;
1516

1517
    /* adjust first dts according to edit list */
1518
    if (sc->time_offset && mov->time_scale > 0) {
1519
        int rescaled = sc->time_offset < 0 ? av_rescale(sc->time_offset, sc->time_scale, mov->time_scale) : sc->time_offset;
1520
        current_dts = -rescaled;
1521 1522
        if (sc->ctts_data && sc->stts_data &&
            sc->ctts_data[0].duration / sc->stts_data[0].duration > 16) {
1523 1524 1525 1526 1527
            /* more than 16 frames delay, dts are likely wrong
               this happens with files created by iMovie */
            sc->wrong_dts = 1;
            st->codec->has_b_frames = 1;
        }
1528 1529
    }

1530
    /* only use old uncompressed audio chunk demuxing when stts specifies it */
1531
    if (!(st->codec->codec_type == AVMEDIA_TYPE_AUDIO &&
1532
          sc->stts_count == 1 && sc->stts_data[0].duration == 1)) {
1533 1534
        unsigned int current_sample = 0;
        unsigned int stts_sample = 0;
1535
        unsigned int sample_size;
1536 1537 1538
        unsigned int distance = 0;
        int key_off = sc->keyframes && sc->keyframes[0] == 1;

1539 1540
        current_dts -= sc->dts_shift;

1541 1542 1543 1544 1545 1546 1547
        if (sc->sample_count >= UINT_MAX / sizeof(*st->index_entries))
            return;
        st->index_entries = av_malloc(sc->sample_count*sizeof(*st->index_entries));
        if (!st->index_entries)
            return;
        st->index_entries_allocated_size = sc->sample_count*sizeof(*st->index_entries);

1548 1549
        for (i = 0; i < sc->chunk_count; i++) {
            current_offset = sc->chunk_offsets[i];
1550 1551
            if (stsc_index + 1 < sc->stsc_count &&
                i + 1 == sc->stsc_data[stsc_index + 1].first)
1552
                stsc_index++;
1553
            for (j = 0; j < sc->stsc_data[stsc_index].count; j++) {
1554
                int keyframe = 0;
1555 1556
                if (current_sample >= sc->sample_count) {
                    av_log(mov->fc, AV_LOG_ERROR, "wrong sample count\n");
1557
                    return;
1558
                }
1559 1560 1561

                if (!sc->keyframe_count || current_sample+key_off == sc->keyframes[stss_index]) {
                    keyframe = 1;
1562 1563
                    if (stss_index + 1 < sc->keyframe_count)
                        stss_index++;
1564 1565 1566 1567
                } else if (sc->stps_count && current_sample+key_off == sc->stps_data[stps_index]) {
                    keyframe = 1;
                    if (stps_index + 1 < sc->stps_count)
                        stps_index++;
1568
                }
1569 1570
                if (keyframe)
                    distance = 0;
1571
                sample_size = sc->sample_size > 0 ? sc->sample_size : sc->sample_sizes[current_sample];
1572
                if(sc->pseudo_stream_id == -1 ||
1573
                   sc->stsc_data[stsc_index].id - 1 == sc->pseudo_stream_id) {
1574 1575 1576 1577 1578 1579
                    AVIndexEntry *e = &st->index_entries[st->nb_index_entries++];
                    e->pos = current_offset;
                    e->timestamp = current_dts;
                    e->size = sample_size;
                    e->min_distance = distance;
                    e->flags = keyframe ? AVINDEX_KEYFRAME : 0;
1580 1581 1582 1583
                    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);
                }
1584

1585
                current_offset += sample_size;
1586
                stream_size += sample_size;
1587
                current_dts += sc->stts_data[stts_index].duration;
1588 1589 1590 1591 1592 1593 1594 1595 1596
                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++;
                }
            }
        }
1597 1598
        if (st->duration > 0)
            st->codec->bit_rate = stream_size*8*sc->time_scale/st->duration;
1599
    } else {
1600
        unsigned chunk_samples, total = 0;
1601

1602 1603 1604
        // compute total chunk count
        for (i = 0; i < sc->stsc_count; i++) {
            unsigned count, chunk_count;
1605

1606
            chunk_samples = sc->stsc_data[i].count;
1607 1608 1609 1610 1611
            if (sc->samples_per_frame && chunk_samples % sc->samples_per_frame) {
                av_log(mov->fc, AV_LOG_ERROR, "error unaligned chunk\n");
                return;
            }

1612 1613 1614 1615 1616 1617 1618 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
            if (sc->samples_per_frame >= 160) { // gsm
                count = chunk_samples / sc->samples_per_frame;
            } else if (sc->samples_per_frame > 1) {
                unsigned samples = (1024/sc->samples_per_frame)*sc->samples_per_frame;
                count = (chunk_samples+samples-1) / samples;
            } else {
                count = (chunk_samples+1023) / 1024;
            }

            if (i < sc->stsc_count - 1)
                chunk_count = sc->stsc_data[i+1].first - sc->stsc_data[i].first;
            else
                chunk_count = sc->chunk_count - (sc->stsc_data[i].first - 1);
            total += chunk_count * count;
        }

        dprintf(mov->fc, "chunk count %d\n", total);
        if (total >= UINT_MAX / sizeof(*st->index_entries))
            return;
        st->index_entries = av_malloc(total*sizeof(*st->index_entries));
        if (!st->index_entries)
            return;
        st->index_entries_allocated_size = total*sizeof(*st->index_entries);

        // populate index
        for (i = 0; i < sc->chunk_count; i++) {
            current_offset = sc->chunk_offsets[i];
            if (stsc_index + 1 < sc->stsc_count &&
                i + 1 == sc->stsc_data[stsc_index + 1].first)
                stsc_index++;
            chunk_samples = sc->stsc_data[stsc_index].count;

1644
            while (chunk_samples > 0) {
1645
                AVIndexEntry *e;
1646 1647 1648 1649 1650 1651 1652 1653 1654 1655
                unsigned size, samples;

                if (sc->samples_per_frame >= 160) { // gsm
                    samples = sc->samples_per_frame;
                    size = sc->bytes_per_frame;
                } else {
                    if (sc->samples_per_frame > 1) {
                        samples = FFMIN((1024 / sc->samples_per_frame)*
                                        sc->samples_per_frame, chunk_samples);
                        size = (samples / sc->samples_per_frame) * sc->bytes_per_frame;
1656
                    } else {
1657 1658
                        samples = FFMIN(1024, chunk_samples);
                        size = samples * sc->sample_size;
1659 1660
                    }
                }
1661

1662 1663 1664 1665 1666 1667 1668 1669 1670 1671
                if (st->nb_index_entries >= total) {
                    av_log(mov->fc, AV_LOG_ERROR, "wrong chunk count %d\n", total);
                    return;
                }
                e = &st->index_entries[st->nb_index_entries++];
                e->pos = current_offset;
                e->timestamp = current_dts;
                e->size = size;
                e->min_distance = 0;
                e->flags = AVINDEX_KEYFRAME;
1672 1673
                dprintf(mov->fc, "AVIndex stream %d, chunk %d, offset %"PRIx64", dts %"PRId64", "
                        "size %d, duration %d\n", st->index, i, current_offset, current_dts,
1674 1675 1676
                        size, samples);

                current_offset += size;
1677
                current_dts += samples;
1678
                chunk_samples -= samples;
1679 1680 1681 1682
            }
        }
    }
}
1683

1684 1685
static int mov_open_dref(ByteIOContext **pb, char *src, MOVDref *ref)
{
1686 1687
    /* try relative path, we do not try the absolute because it can leak information about our
       system to an attacker */
1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709
    if (ref->nlvl_to > 0 && ref->nlvl_from > 0) {
        char filename[1024];
        char *src_path;
        int i, l;

        /* find a source dir */
        src_path = strrchr(src, '/');
        if (src_path)
            src_path++;
        else
            src_path = src;

        /* find a next level down to target */
        for (i = 0, l = strlen(ref->path) - 1; l >= 0; l--)
            if (ref->path[l] == '/') {
                if (i == ref->nlvl_to - 1)
                    break;
                else
                    i++;
            }

        /* compose filename if next level down to target was found */
1710
        if (i == ref->nlvl_to - 1 && src_path - src  < sizeof(filename)) {
1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726
            memcpy(filename, src, src_path - src);
            filename[src_path - src] = 0;

            for (i = 1; i < ref->nlvl_from; i++)
                av_strlcat(filename, "../", 1024);

            av_strlcat(filename, ref->path + l + 1, 1024);

            if (!url_fopen(pb, filename, URL_RDONLY))
                return 0;
        }
    }

    return AVERROR(ENOENT);
};

1727
static int mov_read_trak(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1728 1729 1730
{
    AVStream *st;
    MOVStreamContext *sc;
1731
    int ret;
1732 1733

    st = av_new_stream(c->fc, c->fc->nb_streams);
B
Baptiste Coudurier 已提交
1734
    if (!st) return AVERROR(ENOMEM);
B
Baptiste Coudurier 已提交
1735
    sc = av_mallocz(sizeof(MOVStreamContext));
1736
    if (!sc) return AVERROR(ENOMEM);
1737 1738

    st->priv_data = sc;
1739
    st->codec->codec_type = AVMEDIA_TYPE_DATA;
1740
    sc->ffindex = st->index;
1741

1742 1743 1744 1745
    if ((ret = mov_read_default(c, pb, atom)) < 0)
        return ret;

    /* sanity checks */
1746 1747
    if (sc->chunk_count && (!sc->stts_count || !sc->stsc_count ||
                            (!sc->sample_size && !sc->sample_count))) {
1748 1749
        av_log(c->fc, AV_LOG_ERROR, "stream %d, missing mandatory atoms, broken header\n",
               st->index);
1750 1751
        return 0;
    }
1752

1753
    if (sc->time_scale <= 0) {
1754
        av_log(c->fc, AV_LOG_WARNING, "stream %d, timescale not set\n", st->index);
1755
        sc->time_scale = c->time_scale;
1756
        if (sc->time_scale <= 0)
1757 1758
            sc->time_scale = 1;
    }
1759

1760
    av_set_pts_info(st, 64, 1, sc->time_scale);
1761

1762
    if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO &&
1763 1764 1765 1766 1767
        !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);
    }
1768 1769 1770 1771

    mov_build_index(c, st);

    if (sc->dref_id-1 < sc->drefs_count && sc->drefs[sc->dref_id-1].path) {
1772 1773 1774 1775 1776 1777 1778
        MOVDref *dref = &sc->drefs[sc->dref_id - 1];
        if (mov_open_dref(&sc->pb, c->fc->filename, dref) < 0)
            av_log(c->fc, AV_LOG_ERROR,
                   "stream %d, error opening alias: path='%s', dir='%s', "
                   "filename='%s', volume='%s', nlvl_from=%d, nlvl_to=%d\n",
                   st->index, dref->path, dref->dir, dref->filename,
                   dref->volume, dref->nlvl_from, dref->nlvl_to);
1779 1780 1781
    } else
        sc->pb = c->fc->pb;

1782
    if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
1783 1784 1785 1786
        if (!st->sample_aspect_ratio.num &&
            (st->codec->width != sc->width || st->codec->height != sc->height)) {
            st->sample_aspect_ratio = av_d2q(((double)st->codec->height * sc->width) /
                                             ((double)st->codec->width * sc->height), INT_MAX);
1787 1788 1789 1790
        }

        av_reduce(&st->avg_frame_rate.num, &st->avg_frame_rate.den,
                  sc->time_scale*st->nb_frames, st->duration, INT_MAX);
1791 1792 1793 1794

        if (sc->stts_count == 1 || (sc->stts_count == 2 && sc->stts_data[1].count == 1))
            av_reduce(&st->r_frame_rate.num, &st->r_frame_rate.den,
                      sc->time_scale, sc->stts_data[0].duration, INT_MAX);
1795 1796
    }

1797
    switch (st->codec->codec_id) {
1798
#if CONFIG_H261_DECODER
1799 1800
    case CODEC_ID_H261:
#endif
1801
#if CONFIG_H263_DECODER
1802 1803
    case CODEC_ID_H263:
#endif
1804 1805 1806
#if CONFIG_H264_DECODER
    case CODEC_ID_H264:
#endif
1807
#if CONFIG_MPEG4_DECODER
1808 1809
    case CODEC_ID_MPEG4:
#endif
1810
        st->codec->width = 0; /* let decoder init width/height */
1811 1812 1813
        st->codec->height= 0;
        break;
    }
B
Baptiste Coudurier 已提交
1814 1815 1816

    /* Do not need those anymore. */
    av_freep(&sc->chunk_offsets);
1817
    av_freep(&sc->stsc_data);
B
Baptiste Coudurier 已提交
1818 1819 1820
    av_freep(&sc->sample_sizes);
    av_freep(&sc->keyframes);
    av_freep(&sc->stts_data);
1821
    av_freep(&sc->stps_data);
B
Baptiste Coudurier 已提交
1822

1823
    return 0;
1824 1825
}

1826
static int mov_read_ilst(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1827 1828 1829 1830 1831 1832 1833 1834
{
    int ret;
    c->itunes_metadata = 1;
    ret = mov_read_default(c, pb, atom);
    c->itunes_metadata = 0;
    return ret;
}

1835
static int mov_read_meta(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1836
{
1837 1838 1839 1840 1841 1842 1843 1844 1845 1846
    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;
1847 1848
}

1849
static int mov_read_tkhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1850
{
1851 1852 1853 1854 1855
    int i;
    int width;
    int height;
    int64_t disp_transform[2];
    int display_matrix[3][2];
1856 1857 1858 1859 1860 1861 1862 1863
    AVStream *st;
    MOVStreamContext *sc;
    int version;

    if (c->fc->nb_streams < 1)
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
    sc = st->priv_data;
1864

1865
    version = get_byte(pb);
1866
    get_be24(pb); /* flags */
1867 1868 1869 1870 1871 1872 1873
    /*
    MOV_TRACK_ENABLED 0x0001
    MOV_TRACK_IN_MOVIE 0x0002
    MOV_TRACK_IN_PREVIEW 0x0004
    MOV_TRACK_IN_POSTER 0x0008
    */

B
Baptiste Coudurier 已提交
1874 1875 1876 1877 1878 1879 1880
    if (version == 1) {
        get_be64(pb);
        get_be64(pb);
    } else {
        get_be32(pb); /* creation time */
        get_be32(pb); /* modification time */
    }
1881 1882
    st->id = (int)get_be32(pb); /* track id (NOT 0 !)*/
    get_be32(pb); /* reserved */
1883

1884 1885
    /* highlevel (considering edits) duration in movie timebase */
    (version == 1) ? get_be64(pb) : get_be32(pb);
1886 1887 1888 1889 1890 1891 1892 1893
    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 */

1894 1895 1896 1897 1898 1899 1900 1901
    //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)
    }
1902

1903 1904
    width = get_be32(pb);       // 16.16 fixed point track width
    height = get_be32(pb);      // 16.16 fixed point track height
1905 1906
    sc->width = width >> 16;
    sc->height = height >> 16;
1907

1908
    // transform the display width/height according to the matrix
1909
    // skip this if the display matrix is the default identity matrix
1910
    // or if it is rotating the picture, ex iPhone 3GS
1911 1912
    // to keep the same scale, use [width height 1<<16]
    if (width && height &&
1913 1914 1915 1916 1917
        ((display_matrix[0][0] != 65536  ||
          display_matrix[1][1] != 65536) &&
         !display_matrix[0][1] &&
         !display_matrix[1][0] &&
         !display_matrix[2][0] && !display_matrix[2][1])) {
1918 1919 1920 1921 1922 1923 1924
        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
1925
        st->sample_aspect_ratio = av_d2q(
1926 1927 1928
            ((double) disp_transform[0] * height) /
            ((double) disp_transform[1] * width), INT_MAX);
    }
1929 1930 1931
    return 0;
}

1932
static int mov_read_tfhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
1933 1934 1935 1936 1937 1938 1939 1940 1941
{
    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);
1942
    if (!track_id)
B
Baptiste Coudurier 已提交
1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963 1964 1965 1966
        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;
}

D
David Conrad 已提交
1967 1968 1969 1970 1971 1972
static int mov_read_chap(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
{
    c->chapter_track = get_be32(pb);
    return 0;
}

1973
static int mov_read_trex(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
1974 1975 1976 1977 1978
{
    MOVTrackExt *trex;

    if ((uint64_t)c->trex_count+1 >= UINT_MAX / sizeof(*c->trex_data))
        return -1;
1979 1980
    trex = av_realloc(c->trex_data, (c->trex_count+1)*sizeof(*c->trex_data));
    if (!trex)
B
Baptiste Coudurier 已提交
1981
        return AVERROR(ENOMEM);
1982
    c->trex_data = trex;
B
Baptiste Coudurier 已提交
1983 1984 1985 1986 1987 1988 1989 1990 1991 1992 1993
    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;
}

1994
static int mov_read_trun(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
1995 1996
{
    MOVFragment *frag = &c->fragment;
1997
    AVStream *st = NULL;
1998
    MOVStreamContext *sc;
B
Baptiste Coudurier 已提交
1999 2000 2001 2002 2003 2004
    uint64_t offset;
    int64_t dts;
    int data_offset = 0;
    unsigned entries, first_sample_flags = frag->flags;
    int flags, distance, i;

2005 2006 2007 2008 2009 2010 2011 2012
    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);
2013
        return -1;
2014
    }
2015
    sc = st->priv_data;
B
Baptiste Coudurier 已提交
2016 2017 2018 2019 2020 2021 2022 2023 2024
    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) {
2025
        MOVStts *ctts_data;
B
Baptiste Coudurier 已提交
2026 2027
        if ((uint64_t)entries+sc->ctts_count >= UINT_MAX/sizeof(*sc->ctts_data))
            return -1;
2028 2029 2030
        ctts_data = av_realloc(sc->ctts_data,
                               (entries+sc->ctts_count)*sizeof(*sc->ctts_data));
        if (!ctts_data)
B
Baptiste Coudurier 已提交
2031
            return AVERROR(ENOMEM);
2032
        sc->ctts_data = ctts_data;
B
Baptiste Coudurier 已提交
2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051
    }
    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++;
        }
2052
        if ((keyframe = st->codec->codec_type == AVMEDIA_TYPE_AUDIO ||
B
Baptiste Coudurier 已提交
2053 2054 2055 2056 2057 2058 2059 2060
             (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++;
2061
        dts += sample_duration;
B
Baptiste Coudurier 已提交
2062 2063 2064 2065 2066 2067 2068
        offset += sample_size;
    }
    frag->moof_offset = offset;
    st->duration = dts;
    return 0;
}

2069 2070 2071
/* 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/ */
2072
static int mov_read_wide(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
2073 2074 2075 2076 2077 2078 2079 2080 2081 2082 2083
{
    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.size -= 8;
2084
    if (atom.type != MKTAG('m','d','a','t')) {
2085 2086 2087 2088 2089 2090 2091
        url_fskip(pb, atom.size);
        return 0;
    }
    err = mov_read_mdat(c, pb, atom);
    return err;
}

2092
static int mov_read_cmov(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
2093
{
2094
#if CONFIG_ZLIB
2095
    ByteIOContext ctx;
2096 2097
    uint8_t *cmov_data;
    uint8_t *moov_data; /* uncompressed data */
2098
    long cmov_len, moov_len;
2099
    int ret = -1;
2100

2101
    get_be32(pb); /* dcom atom */
2102
    if (get_le32(pb) != MKTAG('d','c','o','m'))
2103
        return -1;
2104
    if (get_le32(pb) != MKTAG('z','l','i','b')) {
B
Benoit Fouet 已提交
2105
        av_log(c->fc, AV_LOG_ERROR, "unknown compression for cmov atom !");
2106 2107 2108
        return -1;
    }
    get_be32(pb); /* cmvd atom */
2109
    if (get_le32(pb) != MKTAG('c','m','v','d'))
2110 2111
        return -1;
    moov_len = get_be32(pb); /* uncompressed size */
2112
    cmov_len = atom.size - 6 * 4;
2113

B
Baptiste Coudurier 已提交
2114
    cmov_data = av_malloc(cmov_len);
2115
    if (!cmov_data)
2116
        return AVERROR(ENOMEM);
B
Baptiste Coudurier 已提交
2117
    moov_data = av_malloc(moov_len);
2118 2119
    if (!moov_data) {
        av_free(cmov_data);
2120
        return AVERROR(ENOMEM);
2121 2122
    }
    get_buffer(pb, cmov_data, cmov_len);
2123
    if(uncompress (moov_data, (uLongf *) &moov_len, (const Bytef *)cmov_data, cmov_len) != Z_OK)
2124
        goto free_and_return;
2125
    if(init_put_byte(&ctx, moov_data, moov_len, 0, NULL, NULL, NULL, NULL) != 0)
2126
        goto free_and_return;
2127
    atom.type = MKTAG('m','o','o','v');
2128
    atom.size = moov_len;
2129
#ifdef DEBUG
M
Michael Niedermayer 已提交
2130
//    { int fd = open("/tmp/uncompheader.mov", O_WRONLY | O_CREAT); write(fd, moov_data, moov_len); close(fd); }
2131
#endif
2132
    ret = mov_read_default(c, &ctx, atom);
2133
free_and_return:
2134 2135 2136
    av_free(moov_data);
    av_free(cmov_data);
    return ret;
2137 2138 2139
#else
    av_log(c->fc, AV_LOG_ERROR, "this file requires zlib support compiled in\n");
    return -1;
2140
#endif
2141
}
2142

G
Gael Chardon 已提交
2143
/* edit list atom */
2144
static int mov_read_elst(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
G
Gael Chardon 已提交
2145
{
2146
    MOVStreamContext *sc;
B
Baptiste Coudurier 已提交
2147 2148
    int i, edit_count;

2149 2150 2151 2152
    if (c->fc->nb_streams < 1)
        return 0;
    sc = c->fc->streams[c->fc->nb_streams-1]->priv_data;

B
Baptiste Coudurier 已提交
2153
    get_byte(pb); /* version */
2154
    get_be24(pb); /* flags */
B
Baptiste Coudurier 已提交
2155
    edit_count = get_be32(pb); /* entries */
B
Baptiste Coudurier 已提交
2156

2157 2158 2159
    if((uint64_t)edit_count*12+8 > atom.size)
        return -1;

B
Baptiste Coudurier 已提交
2160
    for(i=0; i<edit_count; i++){
2161
        int time;
2162
        int duration = get_be32(pb); /* Track duration */
2163
        time = get_be32(pb); /* Media time */
B
Baptiste Coudurier 已提交
2164
        get_be32(pb); /* Media rate */
2165 2166
        if (i == 0 && time >= -1) {
            sc->time_offset = time != -1 ? time : -duration;
2167
        }
B
Baptiste Coudurier 已提交
2168
    }
2169 2170 2171 2172 2173

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

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

2178
static const MOVParseTableEntry mov_default_parse_table[] = {
2179
{ MKTAG('a','v','s','s'), mov_read_extradata },
D
David Conrad 已提交
2180
{ MKTAG('c','h','p','l'), mov_read_chpl },
2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191
{ 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 },
2192
{ MKTAG('i','l','s','t'), mov_read_ilst },
2193 2194 2195 2196
{ 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 },
2197
{ MKTAG('m','e','t','a'), mov_read_meta },
2198 2199 2200 2201 2202 2203 2204 2205
{ 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 },
2206
{ MKTAG('p','a','s','p'), mov_read_pasp },
2207 2208
{ MKTAG('s','t','b','l'), mov_read_default },
{ MKTAG('s','t','c','o'), mov_read_stco },
2209
{ MKTAG('s','t','p','s'), mov_read_stps },
M
Martin Storsjö 已提交
2210
{ MKTAG('s','t','r','f'), mov_read_strf },
2211 2212 2213 2214 2215
{ 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 },
2216
{ MKTAG('s','t','z','2'), mov_read_stsz }, /* compact sample size */
2217 2218 2219 2220
{ 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 },
D
David Conrad 已提交
2221 2222
{ MKTAG('t','r','e','f'), mov_read_default },
{ MKTAG('c','h','a','p'), mov_read_chap },
2223 2224
{ MKTAG('t','r','e','x'), mov_read_trex },
{ MKTAG('t','r','u','n'), mov_read_trun },
B
Baptiste Coudurier 已提交
2225
{ MKTAG('u','d','t','a'), mov_read_default },
2226 2227
{ MKTAG('w','a','v','e'), mov_read_wave },
{ MKTAG('e','s','d','s'), mov_read_esds },
2228
{ MKTAG('d','a','c','3'), mov_read_dac3 }, /* AC-3 info */
2229 2230
{ MKTAG('w','i','d','e'), mov_read_wide }, /* place holder */
{ MKTAG('c','m','o','v'), mov_read_cmov },
B
Baptiste Coudurier 已提交
2231
{ 0, NULL }
2232 2233
};

F
Fabrice Bellard 已提交
2234 2235
static int mov_probe(AVProbeData *p)
{
2236 2237
    unsigned int offset;
    uint32_t tag;
2238
    int score = 0;
2239

F
Fabrice Bellard 已提交
2240
    /* check file header */
2241 2242 2243 2244
    offset = 0;
    for(;;) {
        /* ignore invalid offset */
        if ((offset + 8) > (unsigned int)p->buf_size)
2245
            return score;
2246
        tag = AV_RL32(p->buf + offset + 4);
2247
        switch(tag) {
2248
        /* check for obvious tags */
2249 2250 2251 2252 2253
        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 */
2254
        case MKTAG('f','t','y','p'):
2255
            return AVPROBE_SCORE_MAX;
2256
        /* those are more common words, so rate then a bit less */
2257 2258 2259 2260 2261
        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'):
2262
            return AVPROBE_SCORE_MAX - 5;
B
Baptiste Coudurier 已提交
2263
        case MKTAG(0x82,0x82,0x7f,0x7d):
2264 2265 2266
        case MKTAG('s','k','i','p'):
        case MKTAG('u','u','i','d'):
        case MKTAG('p','r','f','l'):
2267
            offset = AV_RB32(p->buf+offset) + offset;
2268 2269
            /* if we only find those cause probedata is too small at least rate them */
            score = AVPROBE_SCORE_MAX - 50;
2270 2271 2272
            break;
        default:
            /* unrecognized tag */
2273
            return score;
2274
        }
2275
    }
2276
    return score;
F
Fabrice Bellard 已提交
2277 2278
}

D
David Conrad 已提交
2279 2280 2281 2282 2283 2284 2285
// must be done after parsing all trak because there's no order requirement
static void mov_read_chapters(AVFormatContext *s)
{
    MOVContext *mov = s->priv_data;
    AVStream *st = NULL;
    MOVStreamContext *sc;
    int64_t cur_pos;
2286 2287
    uint8_t *title = NULL;
    int i, len, i8, i16;
D
David Conrad 已提交
2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311

    for (i = 0; i < s->nb_streams; i++)
        if (s->streams[i]->id == mov->chapter_track) {
            st = s->streams[i];
            break;
        }
    if (!st) {
        av_log(s, AV_LOG_ERROR, "Referenced QT chapter track not found\n");
        return;
    }

    st->discard = AVDISCARD_ALL;
    sc = st->priv_data;
    cur_pos = url_ftell(sc->pb);

    for (i = 0; i < st->nb_index_entries; i++) {
        AVIndexEntry *sample = &st->index_entries[i];
        int64_t end = i+1 < st->nb_index_entries ? st->index_entries[i+1].timestamp : st->duration;

        if (url_fseek(sc->pb, sample->pos, SEEK_SET) != sample->pos) {
            av_log(s, AV_LOG_ERROR, "Chapter %d not found in file\n", i);
            goto finish;
        }

2312 2313 2314
        title = av_malloc(sample->size+2);
        get_buffer(sc->pb, title, sample->size);

D
David Conrad 已提交
2315
        // the first two bytes are the length of the title
2316
        len = AV_RB16(title);
D
David Conrad 已提交
2317 2318 2319 2320 2321 2322
        if (len > sample->size-2)
            continue;

        // The samples could theoretically be in any encoding if there's an encd
        // atom following, but in practice are only utf-8 or utf-16, distinguished
        // instead by the presence of a BOM
2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335
        if (AV_RB16(title+2) == 0xfeff) {
            uint8_t *utf8 = av_malloc(2*len+3);

            i8 = i16 = 0;
            while (i16 < len) {
                uint32_t ch;
                uint8_t tmp;
                GET_UTF16(ch, i16 < len ? AV_RB16(title + (i16+=2)) : 0, break;)
                PUT_UTF8(ch, tmp, if (i8 < 2*len) utf8[2+i8++] = tmp;)
            }
            utf8[2+i8] = 0;
            av_freep(&title);
            title = utf8;
D
David Conrad 已提交
2336 2337
        }

2338 2339
        ff_new_chapter(s, i, st->time_base, sample->timestamp, end, title+2);
        av_freep(&title);
D
David Conrad 已提交
2340 2341
    }
finish:
2342
    av_free(title);
D
David Conrad 已提交
2343 2344 2345
    url_fseek(sc->pb, cur_pos, SEEK_SET);
}

Z
Zdenek Kabelac 已提交
2346
static int mov_read_header(AVFormatContext *s, AVFormatParameters *ap)
2347
{
2348
    MOVContext *mov = s->priv_data;
2349
    ByteIOContext *pb = s->pb;
B
Baptiste Coudurier 已提交
2350
    int err;
2351
    MOVAtom atom = { AV_RL32("root") };
2352 2353

    mov->fc = s;
2354 2355
    /* .mov and .mp4 aren't streamable anyway (only progressive download if moov is before mdat) */
    if(!url_is_streamed(pb))
2356
        atom.size = url_fsize(pb);
2357
    else
B
Baptiste Coudurier 已提交
2358
        atom.size = INT64_MAX;
2359 2360

    /* check MOV header */
B
Baptiste Coudurier 已提交
2361 2362 2363 2364 2365 2366
    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");
2367
        return -1;
2368
    }
B
Baptiste Coudurier 已提交
2369
    dprintf(mov->fc, "on_parse_exit_offset=%lld\n", url_ftell(pb));
2370

D
David Conrad 已提交
2371 2372 2373
    if (!url_is_streamed(pb) && mov->chapter_track > 0)
        mov_read_chapters(s);

2374 2375 2376
    return 0;
}

2377
static AVIndexEntry *mov_find_next_sample(AVFormatContext *s, AVStream **st)
2378
{
2379
    AVIndexEntry *sample = NULL;
2380
    int64_t best_dts = INT64_MAX;
2381
    int i;
B
Baptiste Coudurier 已提交
2382
    for (i = 0; i < s->nb_streams; i++) {
2383 2384
        AVStream *avst = s->streams[i];
        MOVStreamContext *msc = avst->priv_data;
2385
        if (msc->pb && msc->current_sample < avst->nb_index_entries) {
2386
            AVIndexEntry *current_sample = &avst->index_entries[msc->current_sample];
2387
            int64_t dts = av_rescale(current_sample->timestamp, AV_TIME_BASE, msc->time_scale);
B
Baptiste Coudurier 已提交
2388
            dprintf(s, "stream %d, sample %d, dts %"PRId64"\n", i, msc->current_sample, dts);
B
Baptiste Coudurier 已提交
2389 2390
            if (!sample || (url_is_streamed(s->pb) && current_sample->pos < sample->pos) ||
                (!url_is_streamed(s->pb) &&
2391
                 ((msc->pb != s->pb && dts < best_dts) || (msc->pb == s->pb &&
B
Baptiste Coudurier 已提交
2392
                 ((FFABS(best_dts - dts) <= AV_TIME_BASE && current_sample->pos < sample->pos) ||
2393
                  (FFABS(best_dts - dts) > AV_TIME_BASE && dts < best_dts)))))) {
2394 2395
                sample = current_sample;
                best_dts = dts;
2396
                *st = avst;
2397
            }
2398 2399
        }
    }
2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411
    return sample;
}

static int mov_read_packet(AVFormatContext *s, AVPacket *pkt)
{
    MOVContext *mov = s->priv_data;
    MOVStreamContext *sc;
    AVIndexEntry *sample;
    AVStream *st = NULL;
    int ret;
 retry:
    sample = mov_find_next_sample(s, &st);
2412 2413 2414
    if (!sample) {
        mov->found_mdat = 0;
        if (!url_is_streamed(s->pb) ||
2415
            mov_read_default(mov, s->pb, (MOVAtom){ AV_RL32("root"), INT64_MAX }) < 0 ||
2416
            url_feof(s->pb))
B
Baptiste Coudurier 已提交
2417
            return AVERROR_EOF;
2418 2419 2420
        dprintf(s, "read fragments, offset 0x%llx\n", url_ftell(s->pb));
        goto retry;
    }
2421
    sc = st->priv_data;
2422 2423
    /* must be done just before reading, to avoid infinite loop on sample */
    sc->current_sample++;
2424 2425

    if (st->discard != AVDISCARD_ALL) {
R
Reimar Döffinger 已提交
2426 2427 2428 2429 2430 2431
        if (url_fseek(sc->pb, sample->pos, SEEK_SET) != sample->pos) {
            av_log(mov->fc, AV_LOG_ERROR, "stream %d, offset 0x%"PRIx64": partial file\n",
                   sc->ffindex, sample->pos);
            return -1;
        }
        ret = av_get_packet(sc->pb, pkt, sample->size);
2432 2433
        if (ret < 0)
            return ret;
R
Reimar Döffinger 已提交
2434 2435 2436 2437 2438 2439 2440 2441 2442
#if CONFIG_DV_DEMUXER
        if (mov->dv_demux && sc->dv_audio_container) {
            dv_produce_packet(mov->dv_demux, pkt, pkt->data, pkt->size);
            av_free(pkt->data);
            pkt->size = 0;
            ret = dv_get_packet(mov->dv_demux, pkt);
            if (ret < 0)
                return ret;
        }
2443
#endif
2444 2445
    }

2446 2447 2448
    pkt->stream_index = sc->ffindex;
    pkt->dts = sample->timestamp;
    if (sc->ctts_data) {
2449
        pkt->pts = pkt->dts + sc->dts_shift + sc->ctts_data[sc->ctts_index].duration;
2450
        /* update ctts context */
2451 2452 2453 2454 2455
        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;
2456
        }
2457 2458
        if (sc->wrong_dts)
            pkt->dts = AV_NOPTS_VALUE;
2459
    } else {
2460
        int64_t next_dts = (sc->current_sample < st->nb_index_entries) ?
2461 2462
            st->index_entries[sc->current_sample].timestamp : st->duration;
        pkt->duration = next_dts - pkt->dts;
2463
        pkt->pts = pkt->dts;
2464
    }
2465 2466
    if (st->discard == AVDISCARD_ALL)
        goto retry;
2467
    pkt->flags |= sample->flags & AVINDEX_KEYFRAME ? AV_PKT_FLAG_KEY : 0;
2468
    pkt->pos = sample->pos;
2469 2470
    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);
2471 2472
    return 0;
}
2473

2474
static int mov_seek_stream(AVFormatContext *s, AVStream *st, int64_t timestamp, int flags)
2475 2476 2477 2478
{
    MOVStreamContext *sc = st->priv_data;
    int sample, time_sample;
    int i;
2479

2480
    sample = av_index_search_timestamp(st, timestamp, flags);
2481
    dprintf(s, "stream %d, timestamp %"PRId64", sample %d\n", st->index, timestamp, sample);
2482 2483
    if (sample < 0 && st->nb_index_entries && timestamp < st->index_entries[0].timestamp)
        sample = 0;
2484 2485 2486
    if (sample < 0) /* not sure what to do */
        return -1;
    sc->current_sample = sample;
2487
    dprintf(s, "stream %d, found sample %d\n", st->index, sc->current_sample);
2488 2489 2490 2491
    /* adjust ctts index */
    if (sc->ctts_data) {
        time_sample = 0;
        for (i = 0; i < sc->ctts_count; i++) {
2492 2493
            int next = time_sample + sc->ctts_data[i].count;
            if (next > sc->current_sample) {
2494 2495
                sc->ctts_index = i;
                sc->ctts_sample = sc->current_sample - time_sample;
2496
                break;
2497
            }
2498
            time_sample = next;
2499 2500
        }
    }
2501
    return sample;
2502 2503
}

2504
static int mov_read_seek(AVFormatContext *s, int stream_index, int64_t sample_time, int flags)
G
Gael Chardon 已提交
2505
{
2506 2507 2508 2509
    AVStream *st;
    int64_t seek_timestamp, timestamp;
    int sample;
    int i;
G
Gael Chardon 已提交
2510

2511
    if (stream_index >= s->nb_streams)
G
Gael Chardon 已提交
2512
        return -1;
2513 2514
    if (sample_time < 0)
        sample_time = 0;
G
Gael Chardon 已提交
2515

2516
    st = s->streams[stream_index];
2517
    sample = mov_seek_stream(s, st, sample_time, flags);
2518
    if (sample < 0)
G
Gael Chardon 已提交
2519 2520
        return -1;

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

2524 2525
    for (i = 0; i < s->nb_streams; i++) {
        st = s->streams[i];
2526
        if (stream_index == i)
2527
            continue;
G
Gael Chardon 已提交
2528

2529
        timestamp = av_rescale_q(seek_timestamp, s->streams[stream_index]->time_base, st->time_base);
2530
        mov_seek_stream(s, st, timestamp, flags);
2531
    }
G
Gael Chardon 已提交
2532 2533 2534
    return 0;
}

Z
Zdenek Kabelac 已提交
2535
static int mov_read_close(AVFormatContext *s)
2536
{
2537
    MOVContext *mov = s->priv_data;
2538 2539 2540
    int i, j;

    for (i = 0; i < s->nb_streams; i++) {
2541 2542
        AVStream *st = s->streams[i];
        MOVStreamContext *sc = st->priv_data;
2543

2544
        av_freep(&sc->ctts_data);
2545
        for (j = 0; j < sc->drefs_count; j++) {
2546
            av_freep(&sc->drefs[j].path);
2547 2548
            av_freep(&sc->drefs[j].dir);
        }
2549 2550 2551
        av_freep(&sc->drefs);
        if (sc->pb && sc->pb != s->pb)
            url_fclose(sc->pb);
2552 2553

        av_freep(&st->codec->palctrl);
2554
    }
2555 2556 2557

    if (mov->dv_demux) {
        for(i = 0; i < mov->dv_fctx->nb_streams; i++) {
2558 2559 2560 2561 2562 2563
            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);
    }
2564

B
Baptiste Coudurier 已提交
2565
    av_freep(&mov->trex_data);
2566

2567 2568 2569
    return 0;
}

2570
AVInputFormat mov_demuxer = {
2571
    "mov,mp4,m4a,3gp,3g2,mj2",
2572
    NULL_IF_CONFIG_SMALL("QuickTime/MPEG-4/Motion JPEG 2000 format"),
F
Fabrice Bellard 已提交
2573 2574
    sizeof(MOVContext),
    mov_probe,
2575 2576 2577
    mov_read_header,
    mov_read_packet,
    mov_read_close,
G
Gael Chardon 已提交
2578
    mov_read_seek,
2579
};