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

#include <limits.h>
24

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

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

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

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

62 63
#include "qtpalette.h"

G
Gael Chardon 已提交
64

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

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

81 82
static const MOVParseTableEntry mov_default_parse_table[];

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

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

    get_be16(pb); // total tracks

    return 0;
}

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

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

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

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

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

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

    if (parse)
        parse(c, pb, str_size);
    else {
B
Baptiste Coudurier 已提交
156 157 158 159 160 161 162
        get_buffer(pb, str, str_size);
        str[str_size] = 0;
        av_metadata_set(&c->fc->metadata, key, str);
        if (*language && strcmp(language, "und")) {
            snprintf(key2, sizeof(key2), "%s-%s", key, language);
            av_metadata_set(&c->fc->metadata, key2, str);
        }
B
Baptiste Coudurier 已提交
163
    }
164 165 166 167 168 169 170 171
#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;
}
172

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

    a.offset = atom.offset;
181

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

211 212 213 214 215
        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;
            }
216

217 218 219 220 221 222
        // 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 */
223
            url_fskip(pb, a.size);
224
        } else {
225
            int64_t start_pos = url_ftell(pb);
226
            int64_t left;
227
            err = parse(c, pb, a);
228
            if (url_is_streamed(pb) && c->found_moov && c->found_mdat)
B
Baptiste Coudurier 已提交
229
                break;
230 231 232
            left = a.size - url_ftell(pb) + start_pos;
            if (left > 0) /* skip garbage at atom end */
                url_fskip(pb, left);
233
        }
234

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

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

242 243 244
    return err;
}

245
static int mov_read_dref(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
246
{
247 248
    AVStream *st;
    MOVStreamContext *sc;
249 250
    int entries, i, j;

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

256 257 258 259 260
    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));
261 262 263
    if (!sc->drefs)
        return AVERROR(ENOMEM);
    sc->drefs_count = entries;
264 265

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

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

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

            url_fskip(pb, 10);

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

            url_fskip(pb, 112);

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

320
static int mov_read_hdlr(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
321
{
322
    AVStream *st;
323 324
    uint32_t type;
    uint32_t ctype;
325

326 327 328 329 330
    if (c->fc->nb_streams < 1) // meta before first trak
        return 0;

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

331
    get_byte(pb); /* version */
332
    get_be24(pb); /* flags */
333 334 335 336 337

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

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

341
    if     (type == MKTAG('v','i','d','e'))
342
        st->codec->codec_type = CODEC_TYPE_VIDEO;
343
    else if(type == MKTAG('s','o','u','n'))
344
        st->codec->codec_type = CODEC_TYPE_AUDIO;
345
    else if(type == MKTAG('m','1','a',' '))
346
        st->codec->codec_id = CODEC_ID_MP2;
347
    else if(type == MKTAG('s','u','b','p'))
348
        st->codec->codec_type = CODEC_TYPE_SUBTITLE;
349

350 351 352 353
    get_be32(pb); /* component  manufacture */
    get_be32(pb); /* component flags */
    get_be32(pb); /* component flags mask */

354
    if(atom.size <= 24)
355
        return 0; /* nothing left to read */
356

357
    url_fskip(pb, atom.size - (url_ftell(pb) - atom.offset));
358 359 360
    return 0;
}

361
int ff_mp4_read_descr_len(ByteIOContext *pb)
362
{
363
    int len = 0;
364 365
    int count = 4;
    while (count--) {
366
        int c = get_byte(pb);
367 368 369
        len = (len << 7) | (c & 0x7f);
        if (!(c & 0x80))
            break;
370 371 372 373
    }
    return len;
}

374
int mp4_read_descr(AVFormatContext *fc, ByteIOContext *pb, int *tag)
375 376 377
{
    int len;
    *tag = get_byte(pb);
378 379
    len = ff_mp4_read_descr_len(pb);
    dprintf(fc, "MPEG4 description: tag=0x%02x len=%d\n", *tag, len);
380 381 382
    return len;
}

383 384 385 386
#define MP4ESDescrTag                   0x03
#define MP4DecConfigDescrTag            0x04
#define MP4DecSpecificDescrTag          0x05

387
static const AVCodecTag mp4_audio_types[] = {
388 389 390 391
    { CODEC_ID_MP3ON4, AOT_PS   }, /* old mp3on4 draft */
    { CODEC_ID_MP3ON4, AOT_L1   }, /* layer 1 */
    { CODEC_ID_MP3ON4, AOT_L2   }, /* layer 2 */
    { CODEC_ID_MP3ON4, AOT_L3   }, /* layer 3 */
392
    { CODEC_ID_MP4ALS, AOT_ALS  }, /* MPEG-4 ALS */
393
    { CODEC_ID_NONE,   AOT_NULL },
394 395
};

396
int ff_mov_read_esds(AVFormatContext *fc, ByteIOContext *pb, MOVAtom atom)
397
{
398
    AVStream *st;
399
    int tag, len;
400

401
    if (fc->nb_streams < 1)
402
        return 0;
403
    st = fc->streams[fc->nb_streams-1];
404

405
    get_be32(pb); /* version + flags */
406
    len = mp4_read_descr(fc, pb, &tag);
407
    if (tag == MP4ESDescrTag) {
408 409
        get_be16(pb); /* ID */
        get_byte(pb); /* priority */
410
    } else
411
        get_be16(pb); /* ID */
412

413
    len = mp4_read_descr(fc, pb, &tag);
414
    if (tag == MP4DecConfigDescrTag) {
415 416 417 418 419 420
        int object_type_id = get_byte(pb);
        get_byte(pb); /* stream type */
        get_be24(pb); /* buffer size db */
        get_be32(pb); /* max bitrate */
        get_be32(pb); /* avg bitrate */

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

457 458 459 460 461
static int mov_read_esds(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
{
    return ff_mov_read_esds(c->fc, pb, atom);
}

462 463 464 465
static int mov_read_pasp(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
{
    const int num = get_be32(pb);
    const int den = get_be32(pb);
466 467 468 469 470 471
    AVStream *st;

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

472
    if (den != 0) {
473 474
        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))
475
            av_log(c->fc, AV_LOG_WARNING,
476 477
                   "sample aspect ratio already set to %d:%d, overriding by 'pasp' atom\n",
                   st->sample_aspect_ratio.num, st->sample_aspect_ratio.den);
478 479 480 481 482 483
        st->sample_aspect_ratio.num = num;
        st->sample_aspect_ratio.den = den;
    }
    return 0;
}

484
/* this atom contains actual media data */
485
static int mov_read_mdat(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
486
{
487 488 489 490 491 492
    if(atom.size == 0) /* wrong one (MP4) */
        return 0;
    c->found_mdat=1;
    return 0; /* now go for moov */
}

493
/* read major brand, minor version and compatible brands and store them as metadata */
494
static int mov_read_ftyp(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
495
{
496 497 498 499 500 501 502
    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 已提交
503
    if (strcmp(type, "qt  "))
504
        c->isom = 1;
505
    av_log(c->fc, AV_LOG_DEBUG, "ISO: File Type Major Brand: %.4s\n",(char *)&type);
506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521
    av_metadata_set(&c->fc->metadata, "major_brand", type);
    minor_ver = get_be32(pb); /* minor version */
    snprintf(minor_ver_str, sizeof(minor_ver_str), "%d", minor_ver);
    av_metadata_set(&c->fc->metadata, "minor_version", minor_ver_str);

    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;
    av_metadata_set(&c->fc->metadata, "compatible_brands", comp_brands_str);
    av_freep(&comp_brands_str);

522 523 524
    return 0;
}

525
/* this atom should contain all header atoms */
526
static int mov_read_moov(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
527
{
528 529
    if (mov_read_default(c, pb, atom) < 0)
        return -1;
530 531 532 533 534 535
    /* 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 */
}

536
static int mov_read_moof(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
537 538 539 540 541
{
    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);
}
542

543
static int mov_read_mdhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
544
{
545 546 547
    AVStream *st;
    MOVStreamContext *sc;
    int version;
548
    char language[4] = {0};
549
    unsigned lang;
550

551 552 553 554 555 556
    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);
557
    if (version > 1)
B
Baptiste Coudurier 已提交
558
        return -1; /* unsupported */
559

560
    get_be24(pb); /* flags */
B
clean  
Baptiste Coudurier 已提交
561 562 563 564 565 566 567
    if (version == 1) {
        get_be64(pb);
        get_be64(pb);
    } else {
        get_be32(pb); /* creation time */
        get_be32(pb); /* modification time */
    }
568

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

572
    lang = get_be16(pb); /* language */
573 574
    if (ff_mov_lang_to_iso639(lang, language))
        av_metadata_set(&st->metadata, "language", language);
575 576 577 578 579
    get_be16(pb); /* quality */

    return 0;
}

580
static int mov_read_mvhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
581
{
B
Baptiste Coudurier 已提交
582
    int version = get_byte(pb); /* version */
583
    get_be24(pb); /* flags */
584

B
Baptiste Coudurier 已提交
585 586 587 588 589 590 591
    if (version == 1) {
        get_be64(pb);
        get_be64(pb);
    } else {
        get_be32(pb); /* creation time */
        get_be32(pb); /* modification time */
    }
592
    c->time_scale = get_be32(pb); /* time scale */
593 594 595

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

B
Baptiste Coudurier 已提交
596
    c->duration = (version == 1) ? get_be64(pb) : get_be32(pb); /* duration */
597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615
    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;
}

616
static int mov_read_smi(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
617
{
618 619 620 621 622
    AVStream *st;

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

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

627 628
    // currently SVQ3 decoder expect full STSD header - so let's fake it
    // this should be fixed and just SMI header should be passed
629
    av_free(st->codec->extradata);
630 631 632
    st->codec->extradata = av_mallocz(atom.size + 0x5a + FF_INPUT_BUFFER_PADDING_SIZE);
    if (!st->codec->extradata)
        return AVERROR(ENOMEM);
B
Baptiste Coudurier 已提交
633 634 635 636
    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);
637 638
    return 0;
}
639

640
static int mov_read_enda(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
641
{
642 643 644 645 646 647
    AVStream *st;
    int little_endian;

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

649
    little_endian = get_be16(pb);
650 651
    dprintf(c->fc, "enda %d\n", little_endian);
    if (little_endian == 1) {
652 653 654 655 656 657 658
        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;
659 660 661 662 663 664
        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;
665 666 667 668 669 670 671
        default:
            break;
        }
    }
    return 0;
}

672
/* FIXME modify qdm2/svq3/h264 decoders to take full atom as extradata */
673
static int mov_read_extradata(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
674
{
675 676
    AVStream *st;
    uint64_t size;
677
    uint8_t *buf;
678 679 680 681 682

    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;
683
    if(size > INT_MAX || (uint64_t)atom.size > INT_MAX)
684
        return -1;
685 686 687 688 689 690 691 692 693
    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);
694 695 696
    return 0;
}

697
static int mov_read_wave(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
R
Roberto Togni 已提交
698
{
699 700 701 702 703
    AVStream *st;

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

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

708 709 710
    if (st->codec->codec_id == CODEC_ID_QDM2) {
        // pass all frma atom to codec, needed at least for QDM2
        av_free(st->codec->extradata);
711 712 713
        st->codec->extradata = av_mallocz(atom.size + FF_INPUT_BUFFER_PADDING_SIZE);
        if (!st->codec->extradata)
            return AVERROR(ENOMEM);
714
        st->codec->extradata_size = atom.size;
715
        get_buffer(pb, st->codec->extradata, atom.size);
716
    } else if (atom.size > 8) { /* to read frma, esds atoms */
717 718
        if (mov_read_default(c, pb, atom) < 0)
            return -1;
719
    } else
720
        url_fskip(pb, atom.size);
R
Roberto Togni 已提交
721 722 723
    return 0;
}

724 725 726 727
/**
 * This function reads atom content and puts data in extradata without tag
 * nor size unlike mov_read_extradata.
 */
728
static int mov_read_glbl(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
729
{
730 731 732 733 734
    AVStream *st;

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

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

739
    av_free(st->codec->extradata);
740 741 742
    st->codec->extradata = av_mallocz(atom.size + FF_INPUT_BUFFER_PADDING_SIZE);
    if (!st->codec->extradata)
        return AVERROR(ENOMEM);
743
    st->codec->extradata_size = atom.size;
744
    get_buffer(pb, st->codec->extradata, atom.size);
745 746 747
    return 0;
}

748
static int mov_read_stco(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
749
{
750 751
    AVStream *st;
    MOVStreamContext *sc;
752
    unsigned int i, entries;
753

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

759
    get_byte(pb); /* version */
760
    get_be24(pb); /* flags */
761 762

    entries = get_be32(pb);
763

764 765
    if(entries >= UINT_MAX/sizeof(int64_t))
        return -1;
766

B
Baptiste Coudurier 已提交
767
    sc->chunk_offsets = av_malloc(entries * sizeof(int64_t));
768
    if (!sc->chunk_offsets)
769 770 771
        return AVERROR(ENOMEM);
    sc->chunk_count = entries;

772 773
    if      (atom.type == MKTAG('s','t','c','o'))
        for(i=0; i<entries; i++)
774
            sc->chunk_offsets[i] = get_be32(pb);
775 776
    else if (atom.type == MKTAG('c','o','6','4'))
        for(i=0; i<entries; i++)
777
            sc->chunk_offsets[i] = get_be64(pb);
778
    else
779
        return -1;
780

781 782 783
    return 0;
}

784 785 786 787
/**
 * Compute codec id for 'lpcm' tag.
 * See CoreAudioTypes and AudioStreamBasicDescription at Apple.
 */
788
enum CodecID ff_mov_get_lpcm_codec_id(int bps, int flags)
789 790 791 792
{
    if (flags & 1) { // floating point
        if (flags & 2) { // big endian
            if      (bps == 32) return CODEC_ID_PCM_F32BE;
793
            else if (bps == 64) return CODEC_ID_PCM_F64BE;
794
        } else {
795 796
            if      (bps == 32) return CODEC_ID_PCM_F32LE;
            else if (bps == 64) return CODEC_ID_PCM_F64LE;
797 798 799 800 801 802 803 804 805 806 807 808 809 810
        }
    } 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 已提交
811
            else if (bps == 16) return CODEC_ID_PCM_S16LE;
812 813 814 815
            else if (bps == 24) return CODEC_ID_PCM_S24LE;
            else if (bps == 32) return CODEC_ID_PCM_S32LE;
        }
    }
D
Diego Pettenò 已提交
816
    return CODEC_ID_NONE;
817 818
}

819
static int mov_read_stsd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
820
{
821 822
    AVStream *st;
    MOVStreamContext *sc;
823
    int j, entries, pseudo_stream_id;
824

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

830
    get_byte(pb); /* version */
831
    get_be24(pb); /* flags */
832 833 834

    entries = get_be32(pb);

835 836
    for(pseudo_stream_id=0; pseudo_stream_id<entries; pseudo_stream_id++) {
        //Parsing Sample description table
837
        enum CodecID id;
838
        int dref_id = 1;
839
        MOVAtom a = { 0, 0, 0 };
840
        int64_t start_pos = url_ftell(pb);
841
        int size = get_be32(pb); /* size */
842
        uint32_t format = get_le32(pb); /* data format */
843

844 845 846 847 848
        if (size >= 16) {
            get_be32(pb); /* reserved */
            get_be16(pb); /* reserved */
            dref_id = get_be16(pb);
        }
849

850
        if (st->codec->codec_tag &&
851
            st->codec->codec_tag != format &&
852
            (c->fc->video_codec_id ? ff_codec_get_id(codec_movvideo_tags, format) != c->fc->video_codec_id
853
                                   : st->codec->codec_tag != MKTAG('j','p','e','g'))
854
           ){
D
Diego Biurrun 已提交
855 856 857
            /* Multiple fourcc, we skip JPEG. This is not correct, we should
             * export it as a separate AVStream but this needs a few changes
             * in the MOV demuxer, patch welcome. */
B
Baptiste Coudurier 已提交
858
            av_log(c->fc, AV_LOG_WARNING, "multiple fourcc not supported\n");
859 860 861
            url_fskip(pb, size - (url_ftell(pb) - start_pos));
            continue;
        }
862
        sc->pseudo_stream_id = st->codec->codec_tag ? -1 : pseudo_stream_id;
863
        sc->dref_id= dref_id;
864

865
        st->codec->codec_tag = format;
866
        id = ff_codec_get_id(codec_movaudio_tags, format);
867
        if (id<=0 && ((format&0xFFFF) == 'm'+('s'<<8) || (format&0xFFFF) == 'T'+('S'<<8)))
868
            id = ff_codec_get_id(ff_codec_wav_tags, bswap_32(format)&0xFFFF);
M
Michael Niedermayer 已提交
869

870
        if (st->codec->codec_type != CODEC_TYPE_VIDEO && id > 0) {
871
            st->codec->codec_type = CODEC_TYPE_AUDIO;
872
        } else if (st->codec->codec_type != CODEC_TYPE_AUDIO && /* do not overwrite codec type */
873
                   format && format != MKTAG('m','p','4','s')) { /* skip old asf mpeg4 tag */
874
            id = ff_codec_get_id(codec_movvideo_tags, format);
875
            if (id <= 0)
876
                id = ff_codec_get_id(ff_codec_bmp_tags, format);
877 878
            if (id > 0)
                st->codec->codec_type = CODEC_TYPE_VIDEO;
879
            else if(st->codec->codec_type == CODEC_TYPE_DATA){
880
                id = ff_codec_get_id(ff_codec_movsubtitle_tags, format);
881 882 883
                if(id > 0)
                    st->codec->codec_type = CODEC_TYPE_SUBTITLE;
            }
884 885
        }

886 887 888
        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);
889

890
        if(st->codec->codec_type==CODEC_TYPE_VIDEO) {
891 892 893 894
            uint8_t codec_name[32];
            unsigned int color_depth;
            int color_greyscale;

895
            st->codec->codec_id = id;
896 897 898 899
            get_be16(pb); /* version */
            get_be16(pb); /* revision level */
            get_be32(pb); /* vendor */
            get_be32(pb); /* temporal quality */
D
Diego Biurrun 已提交
900
            get_be32(pb); /* spatial quality */
901 902 903 904

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

905 906 907
            get_be32(pb); /* horiz resolution */
            get_be32(pb); /* vert resolution */
            get_be32(pb); /* data size, always 0 */
B
Baptiste Coudurier 已提交
908
            get_be16(pb); /* frames per samples */
909

B
Baptiste Coudurier 已提交
910
            get_buffer(pb, codec_name, 32); /* codec name, pascal string */
B
Baptiste Coudurier 已提交
911
            if (codec_name[0] <= 31) {
912 913 914 915
                int i;
                int pos = 0;
                for (i = 0; i < codec_name[0] && pos < sizeof(st->codec->codec_name) - 3; i++) {
                    uint8_t tmp;
916
                    PUT_UTF8(codec_name[i+1], tmp, st->codec->codec_name[pos++] = tmp;)
917 918
                }
                st->codec->codec_name[pos] = 0;
B
Baptiste Coudurier 已提交
919
            }
920

921
            st->codec->bits_per_coded_sample = get_be16(pb); /* depth */
922
            st->codec->color_table_id = get_be16(pb); /* colortable id */
923
            dprintf(c->fc, "depth %d, ctab id %d\n",
924
                   st->codec->bits_per_coded_sample, st->codec->color_table_id);
925
            /* figure out the palette situation */
926 927
            color_depth = st->codec->bits_per_coded_sample & 0x1F;
            color_greyscale = st->codec->bits_per_coded_sample & 0x20;
928 929

            /* if the depth is 2, 4, or 8 bpp, file is palettized */
930
            if ((color_depth == 2) || (color_depth == 4) ||
931
                (color_depth == 8)) {
932 933 934 935
                /* for palette traversal */
                unsigned int color_start, color_count, color_end;
                unsigned char r, g, b;

936
                st->codec->palctrl = av_malloc(sizeof(*st->codec->palctrl));
937
                if (color_greyscale) {
938
                    int color_index, color_dec;
939
                    /* compute the greyscale palette */
940
                    st->codec->bits_per_coded_sample = color_depth;
941 942 943 944 945
                    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;
946
                        st->codec->palctrl->palette[j] =
947 948 949 950 951
                            (r << 16) | (g << 8) | (b);
                        color_index -= color_dec;
                        if (color_index < 0)
                            color_index = 0;
                    }
952
                } else if (st->codec->color_table_id) {
953
                    const uint8_t *color_table;
954 955 956
                    /* if flag bit 3 is set, use the default palette */
                    color_count = 1 << color_depth;
                    if (color_depth == 2)
M
Michael Niedermayer 已提交
957
                        color_table = ff_qt_default_palette_4;
958
                    else if (color_depth == 4)
M
Michael Niedermayer 已提交
959
                        color_table = ff_qt_default_palette_16;
960
                    else
M
Michael Niedermayer 已提交
961
                        color_table = ff_qt_default_palette_256;
962 963

                    for (j = 0; j < color_count; j++) {
964 965 966
                        r = color_table[j * 3 + 0];
                        g = color_table[j * 3 + 1];
                        b = color_table[j * 3 + 2];
967
                        st->codec->palctrl->palette[j] =
968 969 970 971 972 973 974
                            (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);
975 976
                    if ((color_start <= 255) &&
                        (color_end <= 255)) {
M
Mike Melanson 已提交
977 978 979 980 981 982 983 984 985 986 987 988
                        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);
989
                            st->codec->palctrl->palette[j] =
M
Mike Melanson 已提交
990
                                (r << 16) | (g << 8) | (b);
991
                        }
992 993
                    }
                }
994
                st->codec->palctrl->palette_changed = 1;
995
            }
996
        } else if(st->codec->codec_type==CODEC_TYPE_AUDIO) {
997
            int bits_per_sample, flags;
998
            uint16_t version = get_be16(pb);
999

1000
            st->codec->codec_id = id;
1001 1002
            get_be16(pb); /* revision level */
            get_be32(pb); /* vendor */
1003

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

1008
            sc->audio_cid = get_be16(pb);
1009 1010 1011 1012
            get_be16(pb); /* packet size = 0 */

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

1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025
            //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 */
1026
                    st->codec->bits_per_coded_sample = get_be32(pb); /* bits per channel if sound is uncompressed */
1027
                    flags = get_be32(pb); /* lcpm format specific flag */
B
Baptiste Coudurier 已提交
1028 1029
                    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 */
1030
                    if (format == MKTAG('l','p','c','m'))
1031
                        st->codec->codec_id = ff_mov_get_lpcm_codec_id(st->codec->bits_per_coded_sample, flags);
1032 1033 1034
                }
            }

1035
            switch (st->codec->codec_id) {
1036 1037
            case CODEC_ID_PCM_S8:
            case CODEC_ID_PCM_U8:
1038
                if (st->codec->bits_per_coded_sample == 16)
1039 1040
                    st->codec->codec_id = CODEC_ID_PCM_S16BE;
                break;
1041
            case CODEC_ID_PCM_S16LE:
1042
            case CODEC_ID_PCM_S16BE:
1043
                if (st->codec->bits_per_coded_sample == 8)
1044
                    st->codec->codec_id = CODEC_ID_PCM_S8;
1045
                else if (st->codec->bits_per_coded_sample == 24)
1046 1047 1048
                    st->codec->codec_id =
                        st->codec->codec_id == CODEC_ID_PCM_S16BE ?
                        CODEC_ID_PCM_S24BE : CODEC_ID_PCM_S24LE;
1049
                break;
1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062
            /* 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;
1063 1064 1065 1066
            case CODEC_ID_GSM:
                sc->samples_per_frame = 160;
                sc->bytes_per_frame = 33;
                break;
1067 1068
            default:
                break;
1069
            }
1070

1071 1072
            bits_per_sample = av_get_bits_per_sample(st->codec->codec_id);
            if (bits_per_sample) {
1073
                st->codec->bits_per_coded_sample = bits_per_sample;
1074 1075
                sc->sample_size = (bits_per_sample >> 3) * st->codec->channels;
            }
1076
        } else if(st->codec->codec_type==CODEC_TYPE_SUBTITLE){
1077 1078 1079
            // 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) };
1080
            if (format != AV_RL32("mp4s")) // mp4s contains a regular esds atom
R
Reimar Döffinger 已提交
1081
                mov_read_glbl(c, pb, fake_atom);
1082
            st->codec->codec_id= id;
1083 1084
            st->codec->width = sc->width;
            st->codec->height = sc->height;
1085 1086 1087
        } else {
            /* other codec type, just skip (rtp, mp4s, tmcd ...) */
            url_fskip(pb, size - (url_ftell(pb) - start_pos));
1088
        }
1089 1090
        /* this will read extra atoms at the end (wave, alac, damr, avcC, SMI ...) */
        a.size = size - (url_ftell(pb) - start_pos);
1091 1092 1093 1094
        if (a.size > 8) {
            if (mov_read_default(c, pb, a) < 0)
                return -1;
        } else if (a.size > 0)
1095
            url_fskip(pb, a.size);
1096
    }
1097

B
Baptiste Coudurier 已提交
1098
    if(st->codec->codec_type==CODEC_TYPE_AUDIO && st->codec->sample_rate==0 && sc->time_scale>1)
1099
        st->codec->sample_rate= sc->time_scale;
1100

1101
    /* special codec parameters handling */
1102
    switch (st->codec->codec_id) {
1103
#if CONFIG_DV_DEMUXER
1104
    case CODEC_ID_DVAUDIO:
1105
        c->dv_fctx = avformat_alloc_context();
1106 1107 1108 1109 1110 1111 1112 1113
        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;
1114
#endif
1115
    /* no ifdef since parameters are always those */
1116
    case CODEC_ID_QCELP:
1117 1118 1119
        // 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;
1120
        st->codec->frame_size= 160;
1121
        st->codec->channels= 1; /* really needed */
1122
        break;
1123
    case CODEC_ID_AMR_NB:
1124
    case CODEC_ID_AMR_WB:
B
Baptiste Coudurier 已提交
1125
        st->codec->frame_size= sc->samples_per_frame;
1126
        st->codec->channels= 1; /* really needed */
1127
        /* force sample rate for amr, stsd in 3gp does not store sample rate */
B
Baptiste Coudurier 已提交
1128
        if (st->codec->codec_id == CODEC_ID_AMR_NB)
1129
            st->codec->sample_rate = 8000;
B
Baptiste Coudurier 已提交
1130 1131
        else if (st->codec->codec_id == CODEC_ID_AMR_WB)
            st->codec->sample_rate = 16000;
1132
        break;
1133
    case CODEC_ID_MP2:
1134
    case CODEC_ID_MP3:
1135
        st->codec->codec_type = CODEC_TYPE_AUDIO; /* force type after stsd for m1a hdlr */
A
Aurelien Jacobs 已提交
1136
        st->need_parsing = AVSTREAM_PARSE_FULL;
1137
        break;
1138
    case CODEC_ID_GSM:
1139 1140 1141 1142
    case CODEC_ID_ADPCM_MS:
    case CODEC_ID_ADPCM_IMA_WAV:
        st->codec->block_align = sc->bytes_per_frame;
        break;
1143
    case CODEC_ID_ALAC:
1144
        if (st->codec->extradata_size == 36) {
1145 1146
            st->codec->frame_size = AV_RB32(st->codec->extradata+12);
            st->codec->channels   = AV_RB8 (st->codec->extradata+21);
1147
        }
1148
        break;
1149 1150 1151
    default:
        break;
    }
1152

1153 1154 1155
    return 0;
}

1156
static int mov_read_stsc(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1157
{
1158 1159
    AVStream *st;
    MOVStreamContext *sc;
1160
    unsigned int i, entries;
1161

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

1167
    get_byte(pb); /* version */
1168
    get_be24(pb); /* flags */
1169 1170

    entries = get_be32(pb);
1171

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

1174 1175
    if(entries >= UINT_MAX / sizeof(*sc->stsc_data))
        return -1;
1176 1177
    sc->stsc_data = av_malloc(entries * sizeof(*sc->stsc_data));
    if (!sc->stsc_data)
1178 1179 1180
        return AVERROR(ENOMEM);
    sc->stsc_count = entries;

1181
    for(i=0; i<entries; i++) {
1182 1183 1184
        sc->stsc_data[i].first = get_be32(pb);
        sc->stsc_data[i].count = get_be32(pb);
        sc->stsc_data[i].id = get_be32(pb);
1185 1186 1187 1188
    }
    return 0;
}

1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217
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;
}

1218
static int mov_read_stss(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1219
{
1220 1221
    AVStream *st;
    MOVStreamContext *sc;
1222
    unsigned int i, entries;
1223

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

1229
    get_byte(pb); /* version */
1230
    get_be24(pb); /* flags */
1231 1232

    entries = get_be32(pb);
1233

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

B
Baptiste Coudurier 已提交
1236
    if(entries >= UINT_MAX / sizeof(int))
1237
        return -1;
B
Baptiste Coudurier 已提交
1238
    sc->keyframes = av_malloc(entries * sizeof(int));
1239
    if (!sc->keyframes)
1240 1241 1242
        return AVERROR(ENOMEM);
    sc->keyframe_count = entries;

1243 1244
    for(i=0; i<entries; i++) {
        sc->keyframes[i] = get_be32(pb);
1245
        //dprintf(c->fc, "keyframes[]=%d\n", sc->keyframes[i]);
1246 1247 1248 1249
    }
    return 0;
}

1250
static int mov_read_stsz(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1251
{
1252 1253
    AVStream *st;
    MOVStreamContext *sc;
1254 1255 1256
    unsigned int i, entries, sample_size, field_size, num_bytes;
    GetBitContext gb;
    unsigned char* buf;
1257

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

1263
    get_byte(pb); /* version */
1264
    get_be24(pb); /* flags */
1265

1266
    if (atom.type == MKTAG('s','t','s','z')) {
1267 1268 1269 1270
        sample_size = get_be32(pb);
        if (!sc->sample_size) /* do not overwrite value computed in stsd */
            sc->sample_size = sample_size;
        field_size = 32;
1271 1272 1273 1274 1275
    } else {
        sample_size = 0;
        get_be24(pb); /* reserved */
        field_size = get_byte(pb);
    }
1276
    entries = get_be32(pb);
1277 1278

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

1280
    sc->sample_count = entries;
1281 1282 1283
    if (sample_size)
        return 0;

1284 1285 1286 1287 1288
    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;
    }

1289
    if (entries >= UINT_MAX / sizeof(int) || entries >= (UINT_MAX - 4) / field_size)
1290
        return -1;
B
Baptiste Coudurier 已提交
1291
    sc->sample_sizes = av_malloc(entries * sizeof(int));
1292
    if (!sc->sample_sizes)
1293 1294
        return AVERROR(ENOMEM);

1295 1296
    num_bytes = (entries*field_size+4)>>3;

1297
    buf = av_malloc(num_bytes+FF_INPUT_BUFFER_PADDING_SIZE);
1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310
    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);

1311
    for(i=0; i<entries; i++)
1312 1313 1314
        sc->sample_sizes[i] = get_bits_long(&gb, field_size);

    av_free(buf);
1315 1316 1317
    return 0;
}

1318
static int mov_read_stts(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1319
{
1320 1321
    AVStream *st;
    MOVStreamContext *sc;
1322
    unsigned int i, entries;
1323 1324
    int64_t duration=0;
    int64_t total_sample_count=0;
1325

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

1331
    get_byte(pb); /* version */
1332
    get_be24(pb); /* flags */
1333
    entries = get_be32(pb);
1334 1335 1336

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

1337
    if(entries >= UINT_MAX / sizeof(*sc->stts_data))
1338
        return -1;
1339
    sc->stts_data = av_malloc(entries * sizeof(*sc->stts_data));
1340
    if (!sc->stts_data)
1341 1342
        return AVERROR(ENOMEM);
    sc->stts_count = entries;
1343

1344
    for(i=0; i<entries; i++) {
M
cleanup  
Michael Niedermayer 已提交
1345 1346
        int sample_duration;
        int sample_count;
1347

1348
        sample_count=get_be32(pb);
1349
        sample_duration = get_be32(pb);
1350 1351 1352
        sc->stts_data[i].count= sample_count;
        sc->stts_data[i].duration= sample_duration;

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

B
Baptiste Coudurier 已提交
1355
        duration+=(int64_t)sample_duration*sample_count;
1356 1357 1358
        total_sample_count+=sample_count;
    }

1359 1360 1361
    st->nb_frames= total_sample_count;
    if(duration)
        st->duration= duration;
1362 1363 1364
    return 0;
}

1365 1366 1367 1368 1369
static int mov_read_cslg(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
{
    AVStream *st;
    MOVStreamContext *sc;

B
Baptiste Coudurier 已提交
1370
    if (c->fc->nb_streams < 1)
1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
    sc = st->priv_data;

    get_be32(pb); // version + flags

    sc->dts_shift = get_be32(pb);
    dprintf(c->fc, "dts shift %d\n", sc->dts_shift);

    get_be32(pb); // least dts to pts delta
    get_be32(pb); // greatest dts to pts delta
    get_be32(pb); // pts start
    get_be32(pb); // pts end

    return 0;
}

1388
static int mov_read_ctts(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
M
Michael Niedermayer 已提交
1389
{
1390 1391
    AVStream *st;
    MOVStreamContext *sc;
M
Michael Niedermayer 已提交
1392 1393
    unsigned int i, entries;

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

M
Michael Niedermayer 已提交
1399
    get_byte(pb); /* version */
1400
    get_be24(pb); /* flags */
M
Michael Niedermayer 已提交
1401
    entries = get_be32(pb);
1402 1403 1404

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

1405
    if(entries >= UINT_MAX / sizeof(*sc->ctts_data))
M
Michael Niedermayer 已提交
1406
        return -1;
1407
    sc->ctts_data = av_malloc(entries * sizeof(*sc->ctts_data));
1408
    if (!sc->ctts_data)
1409 1410
        return AVERROR(ENOMEM);
    sc->ctts_count = entries;
1411

M
Michael Niedermayer 已提交
1412
    for(i=0; i<entries; i++) {
1413 1414 1415 1416 1417
        int count    =get_be32(pb);
        int duration =get_be32(pb);

        sc->ctts_data[i].count   = count;
        sc->ctts_data[i].duration= duration;
M
Michael Niedermayer 已提交
1418 1419 1420 1421
    }
    return 0;
}

1422 1423 1424
static void mov_build_index(MOVContext *mov, AVStream *st)
{
    MOVStreamContext *sc = st->priv_data;
1425
    int64_t current_offset;
1426 1427 1428 1429
    int64_t current_dts = 0;
    unsigned int stts_index = 0;
    unsigned int stsc_index = 0;
    unsigned int stss_index = 0;
1430
    unsigned int stps_index = 0;
1431
    unsigned int i, j;
1432
    uint64_t stream_size = 0;
1433

1434 1435
    /* adjust first dts according to edit list */
    if (sc->time_offset) {
1436
        int rescaled = sc->time_offset < 0 ? av_rescale(sc->time_offset, sc->time_scale, mov->time_scale) : sc->time_offset;
1437
        current_dts = -rescaled;
1438 1439 1440 1441 1442 1443
        if (sc->ctts_data && sc->ctts_data[0].duration / sc->stts_data[0].duration > 16) {
            /* 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;
        }
1444 1445
    }

1446 1447 1448
    /* only use old uncompressed audio chunk demuxing when stts specifies it */
    if (!(st->codec->codec_type == CODEC_TYPE_AUDIO &&
          sc->stts_count == 1 && sc->stts_data[0].duration == 1)) {
1449 1450
        unsigned int current_sample = 0;
        unsigned int stts_sample = 0;
1451
        unsigned int sample_size;
1452 1453 1454
        unsigned int distance = 0;
        int key_off = sc->keyframes && sc->keyframes[0] == 1;

1455 1456
        current_dts -= sc->dts_shift;

1457 1458 1459
        st->nb_frames = sc->sample_count;
        for (i = 0; i < sc->chunk_count; i++) {
            current_offset = sc->chunk_offsets[i];
1460 1461
            if (stsc_index + 1 < sc->stsc_count &&
                i + 1 == sc->stsc_data[stsc_index + 1].first)
1462
                stsc_index++;
1463
            for (j = 0; j < sc->stsc_data[stsc_index].count; j++) {
1464
                int keyframe = 0;
1465 1466
                if (current_sample >= sc->sample_count) {
                    av_log(mov->fc, AV_LOG_ERROR, "wrong sample count\n");
1467
                    return;
1468
                }
1469 1470 1471

                if (!sc->keyframe_count || current_sample+key_off == sc->keyframes[stss_index]) {
                    keyframe = 1;
1472 1473
                    if (stss_index + 1 < sc->keyframe_count)
                        stss_index++;
1474 1475 1476 1477
                } 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++;
1478
                }
1479 1480
                if (keyframe)
                    distance = 0;
1481
                sample_size = sc->sample_size > 0 ? sc->sample_size : sc->sample_sizes[current_sample];
1482
                if(sc->pseudo_stream_id == -1 ||
1483
                   sc->stsc_data[stsc_index].id - 1 == sc->pseudo_stream_id) {
1484 1485
                    av_add_index_entry(st, current_offset, current_dts, sample_size, distance,
                                    keyframe ? AVINDEX_KEYFRAME : 0);
1486 1487 1488 1489
                    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);
                }
1490

1491
                current_offset += sample_size;
1492
                stream_size += sample_size;
1493
                current_dts += sc->stts_data[stts_index].duration;
1494 1495 1496 1497 1498 1499 1500 1501 1502
                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++;
                }
            }
        }
1503
        st->codec->bit_rate = stream_size*8*sc->time_scale/st->duration;
1504
    } else {
1505
        for (i = 0; i < sc->chunk_count; i++) {
1506 1507
            unsigned chunk_samples;

1508
            current_offset = sc->chunk_offsets[i];
1509 1510
            if (stsc_index + 1 < sc->stsc_count &&
                i + 1 == sc->stsc_data[stsc_index + 1].first)
1511
                stsc_index++;
1512
            chunk_samples = sc->stsc_data[stsc_index].count;
1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529

            if (sc->samples_per_frame && chunk_samples % sc->samples_per_frame) {
                av_log(mov->fc, AV_LOG_ERROR, "error unaligned chunk\n");
                return;
            }

            while (chunk_samples > 0) {
                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;
1530
                    } else {
1531 1532
                        samples = FFMIN(1024, chunk_samples);
                        size = samples * sc->sample_size;
1533 1534
                    }
                }
1535 1536

                av_add_index_entry(st, current_offset, current_dts, size, 0, AVINDEX_KEYFRAME);
1537 1538
                dprintf(mov->fc, "AVIndex stream %d, chunk %d, offset %"PRIx64", dts %"PRId64", "
                        "size %d, duration %d\n", st->index, i, current_offset, current_dts,
1539 1540 1541
                        size, samples);

                current_offset += size;
1542
                current_dts += samples;
1543
                chunk_samples -= samples;
1544 1545 1546 1547
            }
        }
    }
}
1548

1549
static int mov_read_trak(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1550 1551 1552
{
    AVStream *st;
    MOVStreamContext *sc;
1553
    int ret;
1554 1555

    st = av_new_stream(c->fc, c->fc->nb_streams);
B
Baptiste Coudurier 已提交
1556
    if (!st) return AVERROR(ENOMEM);
B
Baptiste Coudurier 已提交
1557
    sc = av_mallocz(sizeof(MOVStreamContext));
1558
    if (!sc) return AVERROR(ENOMEM);
1559 1560

    st->priv_data = sc;
1561
    st->codec->codec_type = CODEC_TYPE_DATA;
1562
    sc->ffindex = st->index;
1563

1564 1565 1566 1567
    if ((ret = mov_read_default(c, pb, atom)) < 0)
        return ret;

    /* sanity checks */
1568 1569
    if (sc->chunk_count && (!sc->stts_count || !sc->stsc_count ||
                            (!sc->sample_size && !sc->sample_count))) {
1570 1571
        av_log(c->fc, AV_LOG_ERROR, "stream %d, missing mandatory atoms, broken header\n",
               st->index);
1572 1573
        return 0;
    }
1574 1575 1576 1577

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

1578
    av_set_pts_info(st, 64, 1, sc->time_scale);
1579

1580
    if (st->codec->codec_type == CODEC_TYPE_AUDIO &&
1581 1582 1583 1584 1585
        !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);
    }
1586 1587 1588 1589 1590

    mov_build_index(c, st);

    if (sc->dref_id-1 < sc->drefs_count && sc->drefs[sc->dref_id-1].path) {
        if (url_fopen(&sc->pb, sc->drefs[sc->dref_id-1].path, URL_RDONLY) < 0)
1591 1592
            av_log(c->fc, AV_LOG_ERROR, "stream %d, error opening file %s: %s\n",
                   st->index, sc->drefs[sc->dref_id-1].path, strerror(errno));
1593 1594 1595 1596
    } else
        sc->pb = c->fc->pb;

    switch (st->codec->codec_id) {
1597
#if CONFIG_H261_DECODER
1598 1599
    case CODEC_ID_H261:
#endif
1600
#if CONFIG_H263_DECODER
1601 1602
    case CODEC_ID_H263:
#endif
1603 1604 1605
#if CONFIG_H264_DECODER
    case CODEC_ID_H264:
#endif
1606
#if CONFIG_MPEG4_DECODER
1607 1608
    case CODEC_ID_MPEG4:
#endif
1609
        st->codec->width = 0; /* let decoder init width/height */
1610 1611 1612
        st->codec->height= 0;
        break;
    }
B
Baptiste Coudurier 已提交
1613 1614 1615

    /* Do not need those anymore. */
    av_freep(&sc->chunk_offsets);
1616
    av_freep(&sc->stsc_data);
B
Baptiste Coudurier 已提交
1617 1618 1619
    av_freep(&sc->sample_sizes);
    av_freep(&sc->keyframes);
    av_freep(&sc->stts_data);
1620
    av_freep(&sc->stps_data);
B
Baptiste Coudurier 已提交
1621

1622
    return 0;
1623 1624
}

1625
static int mov_read_ilst(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1626 1627 1628 1629 1630 1631 1632 1633
{
    int ret;
    c->itunes_metadata = 1;
    ret = mov_read_default(c, pb, atom);
    c->itunes_metadata = 0;
    return ret;
}

1634
static int mov_read_meta(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1635
{
1636 1637 1638 1639 1640 1641 1642 1643 1644 1645
    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;
1646 1647
}

1648
static int mov_read_tkhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1649
{
1650 1651 1652 1653 1654
    int i;
    int width;
    int height;
    int64_t disp_transform[2];
    int display_matrix[3][2];
1655 1656 1657 1658 1659 1660 1661 1662
    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;
1663

1664
    version = get_byte(pb);
1665
    get_be24(pb); /* flags */
1666 1667 1668 1669 1670 1671 1672
    /*
    MOV_TRACK_ENABLED 0x0001
    MOV_TRACK_IN_MOVIE 0x0002
    MOV_TRACK_IN_PREVIEW 0x0004
    MOV_TRACK_IN_POSTER 0x0008
    */

B
Baptiste Coudurier 已提交
1673 1674 1675 1676 1677 1678 1679
    if (version == 1) {
        get_be64(pb);
        get_be64(pb);
    } else {
        get_be32(pb); /* creation time */
        get_be32(pb); /* modification time */
    }
1680 1681
    st->id = (int)get_be32(pb); /* track id (NOT 0 !)*/
    get_be32(pb); /* reserved */
1682

1683 1684
    /* highlevel (considering edits) duration in movie timebase */
    (version == 1) ? get_be64(pb) : get_be32(pb);
1685 1686 1687 1688 1689 1690 1691 1692
    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 */

1693 1694 1695 1696 1697 1698 1699 1700
    //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)
    }
1701

1702 1703
    width = get_be32(pb);       // 16.16 fixed point track width
    height = get_be32(pb);      // 16.16 fixed point track height
1704 1705
    sc->width = width >> 16;
    sc->height = height >> 16;
1706

1707
    // transform the display width/height according to the matrix
1708
    // skip this if the display matrix is the default identity matrix
1709
    // or if it is rotating the picture, ex iPhone 3GS
1710 1711
    // to keep the same scale, use [width height 1<<16]
    if (width && height &&
1712 1713 1714 1715 1716
        ((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])) {
1717 1718 1719 1720 1721 1722 1723
        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
1724
        st->sample_aspect_ratio = av_d2q(
1725 1726 1727
            ((double) disp_transform[0] * height) /
            ((double) disp_transform[1] * width), INT_MAX);
    }
1728 1729 1730
    return 0;
}

1731
static int mov_read_tfhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
1732 1733 1734 1735 1736 1737 1738 1739 1740
{
    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);
1741
    if (!track_id)
B
Baptiste Coudurier 已提交
1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765
        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;
}

1766
static int mov_read_trex(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
1767 1768 1769 1770 1771
{
    MOVTrackExt *trex;

    if ((uint64_t)c->trex_count+1 >= UINT_MAX / sizeof(*c->trex_data))
        return -1;
1772 1773
    trex = av_realloc(c->trex_data, (c->trex_count+1)*sizeof(*c->trex_data));
    if (!trex)
B
Baptiste Coudurier 已提交
1774
        return AVERROR(ENOMEM);
1775
    c->trex_data = trex;
B
Baptiste Coudurier 已提交
1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786
    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;
}

1787
static int mov_read_trun(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
1788 1789
{
    MOVFragment *frag = &c->fragment;
1790
    AVStream *st = NULL;
1791
    MOVStreamContext *sc;
B
Baptiste Coudurier 已提交
1792 1793 1794 1795 1796 1797
    uint64_t offset;
    int64_t dts;
    int data_offset = 0;
    unsigned entries, first_sample_flags = frag->flags;
    int flags, distance, i;

1798 1799 1800 1801 1802 1803 1804 1805
    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);
1806
        return -1;
1807
    }
1808
    sc = st->priv_data;
B
Baptiste Coudurier 已提交
1809 1810 1811 1812 1813 1814 1815 1816 1817
    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) {
1818
        MOVStts *ctts_data;
B
Baptiste Coudurier 已提交
1819 1820
        if ((uint64_t)entries+sc->ctts_count >= UINT_MAX/sizeof(*sc->ctts_data))
            return -1;
1821 1822 1823
        ctts_data = av_realloc(sc->ctts_data,
                               (entries+sc->ctts_count)*sizeof(*sc->ctts_data));
        if (!ctts_data)
B
Baptiste Coudurier 已提交
1824
            return AVERROR(ENOMEM);
1825
        sc->ctts_data = ctts_data;
B
Baptiste Coudurier 已提交
1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853
    }
    dts = st->duration;
    offset = frag->base_data_offset + data_offset;
    distance = 0;
    dprintf(c->fc, "first sample flags 0x%x\n", first_sample_flags);
    for (i = 0; i < entries; i++) {
        unsigned sample_size = frag->size;
        int sample_flags = i ? frag->flags : first_sample_flags;
        unsigned sample_duration = frag->duration;
        int keyframe;

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

1862 1863 1864
/* 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/ */
1865
static int mov_read_wide(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1866 1867 1868 1869 1870 1871 1872 1873 1874 1875 1876 1877
{
    int err;

    if (atom.size < 8)
        return 0; /* continue */
    if (get_be32(pb) != 0) { /* 0 sized mdat atom... use the 'wide' atom size */
        url_fskip(pb, atom.size - 4);
        return 0;
    }
    atom.type = get_le32(pb);
    atom.offset += 8;
    atom.size -= 8;
1878
    if (atom.type != MKTAG('m','d','a','t')) {
1879 1880 1881 1882 1883 1884 1885
        url_fskip(pb, atom.size);
        return 0;
    }
    err = mov_read_mdat(c, pb, atom);
    return err;
}

1886
static int mov_read_cmov(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1887
{
1888
#if CONFIG_ZLIB
1889
    ByteIOContext ctx;
1890 1891
    uint8_t *cmov_data;
    uint8_t *moov_data; /* uncompressed data */
1892
    long cmov_len, moov_len;
1893
    int ret = -1;
1894

1895
    get_be32(pb); /* dcom atom */
1896
    if (get_le32(pb) != MKTAG('d','c','o','m'))
1897
        return -1;
1898
    if (get_le32(pb) != MKTAG('z','l','i','b')) {
B
Benoit Fouet 已提交
1899
        av_log(c->fc, AV_LOG_ERROR, "unknown compression for cmov atom !");
1900 1901 1902
        return -1;
    }
    get_be32(pb); /* cmvd atom */
1903
    if (get_le32(pb) != MKTAG('c','m','v','d'))
1904 1905
        return -1;
    moov_len = get_be32(pb); /* uncompressed size */
1906
    cmov_len = atom.size - 6 * 4;
1907

B
Baptiste Coudurier 已提交
1908
    cmov_data = av_malloc(cmov_len);
1909
    if (!cmov_data)
1910
        return AVERROR(ENOMEM);
B
Baptiste Coudurier 已提交
1911
    moov_data = av_malloc(moov_len);
1912 1913
    if (!moov_data) {
        av_free(cmov_data);
1914
        return AVERROR(ENOMEM);
1915 1916
    }
    get_buffer(pb, cmov_data, cmov_len);
1917
    if(uncompress (moov_data, (uLongf *) &moov_len, (const Bytef *)cmov_data, cmov_len) != Z_OK)
1918
        goto free_and_return;
1919
    if(init_put_byte(&ctx, moov_data, moov_len, 0, NULL, NULL, NULL, NULL) != 0)
1920
        goto free_and_return;
1921
    atom.type = MKTAG('m','o','o','v');
1922 1923
    atom.offset = 0;
    atom.size = moov_len;
1924
#ifdef DEBUG
M
Michael Niedermayer 已提交
1925
//    { int fd = open("/tmp/uncompheader.mov", O_WRONLY | O_CREAT); write(fd, moov_data, moov_len); close(fd); }
1926
#endif
1927
    ret = mov_read_default(c, &ctx, atom);
1928
free_and_return:
1929 1930 1931
    av_free(moov_data);
    av_free(cmov_data);
    return ret;
1932 1933 1934
#else
    av_log(c->fc, AV_LOG_ERROR, "this file requires zlib support compiled in\n");
    return -1;
1935
#endif
1936
}
1937

G
Gael Chardon 已提交
1938
/* edit list atom */
1939
static int mov_read_elst(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
G
Gael Chardon 已提交
1940
{
1941
    MOVStreamContext *sc;
B
Baptiste Coudurier 已提交
1942 1943
    int i, edit_count;

1944 1945 1946 1947
    if (c->fc->nb_streams < 1)
        return 0;
    sc = c->fc->streams[c->fc->nb_streams-1]->priv_data;

B
Baptiste Coudurier 已提交
1948
    get_byte(pb); /* version */
1949
    get_be24(pb); /* flags */
B
Baptiste Coudurier 已提交
1950
    edit_count = get_be32(pb); /* entries */
B
Baptiste Coudurier 已提交
1951

1952 1953 1954
    if((uint64_t)edit_count*12+8 > atom.size)
        return -1;

B
Baptiste Coudurier 已提交
1955
    for(i=0; i<edit_count; i++){
1956
        int time;
1957
        int duration = get_be32(pb); /* Track duration */
1958
        time = get_be32(pb); /* Media time */
B
Baptiste Coudurier 已提交
1959
        get_be32(pb); /* Media rate */
1960 1961
        if (i == 0 && time >= -1) {
            sc->time_offset = time != -1 ? time : -duration;
1962
        }
B
Baptiste Coudurier 已提交
1963
    }
1964 1965 1966 1967 1968

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

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

1973
static const MOVParseTableEntry mov_default_parse_table[] = {
1974
{ MKTAG('a','v','s','s'), mov_read_extradata },
1975
{ MKTAG('c','o','6','4'), mov_read_stco },
1976
{ MKTAG('c','s','l','g'), mov_read_cslg },
1977 1978 1979 1980 1981 1982 1983 1984 1985 1986
{ 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 },
1987
{ MKTAG('i','l','s','t'), mov_read_ilst },
1988 1989 1990 1991
{ 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 },
1992
{ MKTAG('m','e','t','a'), mov_read_meta },
1993 1994 1995 1996 1997 1998 1999 2000
{ 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 },
2001
{ MKTAG('p','a','s','p'), mov_read_pasp },
2002 2003
{ MKTAG('s','t','b','l'), mov_read_default },
{ MKTAG('s','t','c','o'), mov_read_stco },
2004
{ MKTAG('s','t','p','s'), mov_read_stps },
2005 2006 2007 2008 2009
{ 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 },
2010
{ MKTAG('s','t','z','2'), mov_read_stsz }, /* compact sample size */
2011 2012 2013 2014 2015 2016
{ MKTAG('t','k','h','d'), mov_read_tkhd }, /* track header */
{ MKTAG('t','f','h','d'), mov_read_tfhd }, /* track fragment header */
{ MKTAG('t','r','a','k'), mov_read_trak },
{ MKTAG('t','r','a','f'), mov_read_default },
{ MKTAG('t','r','e','x'), mov_read_trex },
{ MKTAG('t','r','u','n'), mov_read_trun },
B
Baptiste Coudurier 已提交
2017
{ MKTAG('u','d','t','a'), mov_read_default },
2018 2019 2020 2021
{ MKTAG('w','a','v','e'), mov_read_wave },
{ MKTAG('e','s','d','s'), mov_read_esds },
{ MKTAG('w','i','d','e'), mov_read_wide }, /* place holder */
{ MKTAG('c','m','o','v'), mov_read_cmov },
B
Baptiste Coudurier 已提交
2022
{ 0, NULL }
2023 2024
};

F
Fabrice Bellard 已提交
2025 2026
static int mov_probe(AVProbeData *p)
{
2027 2028
    unsigned int offset;
    uint32_t tag;
2029
    int score = 0;
2030

F
Fabrice Bellard 已提交
2031
    /* check file header */
2032 2033 2034 2035
    offset = 0;
    for(;;) {
        /* ignore invalid offset */
        if ((offset + 8) > (unsigned int)p->buf_size)
2036
            return score;
2037
        tag = AV_RL32(p->buf + offset + 4);
2038
        switch(tag) {
2039
        /* check for obvious tags */
2040 2041 2042 2043 2044
        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 */
2045
        case MKTAG('f','t','y','p'):
2046
            return AVPROBE_SCORE_MAX;
2047
        /* those are more common words, so rate then a bit less */
2048 2049 2050 2051 2052
        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'):
2053
            return AVPROBE_SCORE_MAX - 5;
B
Baptiste Coudurier 已提交
2054
        case MKTAG(0x82,0x82,0x7f,0x7d):
2055 2056 2057
        case MKTAG('s','k','i','p'):
        case MKTAG('u','u','i','d'):
        case MKTAG('p','r','f','l'):
2058
            offset = AV_RB32(p->buf+offset) + offset;
2059 2060
            /* if we only find those cause probedata is too small at least rate them */
            score = AVPROBE_SCORE_MAX - 50;
2061 2062 2063
            break;
        default:
            /* unrecognized tag */
2064
            return score;
2065
        }
2066
    }
2067
    return score;
F
Fabrice Bellard 已提交
2068 2069
}

Z
Zdenek Kabelac 已提交
2070
static int mov_read_header(AVFormatContext *s, AVFormatParameters *ap)
2071
{
2072
    MOVContext *mov = s->priv_data;
2073
    ByteIOContext *pb = s->pb;
B
Baptiste Coudurier 已提交
2074
    int err;
2075
    MOVAtom atom = { 0, 0, 0 };
2076 2077

    mov->fc = s;
2078 2079
    /* .mov and .mp4 aren't streamable anyway (only progressive download if moov is before mdat) */
    if(!url_is_streamed(pb))
2080
        atom.size = url_fsize(pb);
2081
    else
B
Baptiste Coudurier 已提交
2082
        atom.size = INT64_MAX;
2083 2084

    /* check MOV header */
B
Baptiste Coudurier 已提交
2085 2086 2087 2088 2089 2090
    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");
2091
        return -1;
2092
    }
B
Baptiste Coudurier 已提交
2093
    dprintf(mov->fc, "on_parse_exit_offset=%lld\n", url_ftell(pb));
2094

2095 2096 2097
    return 0;
}

2098
static AVIndexEntry *mov_find_next_sample(AVFormatContext *s, AVStream **st)
2099
{
2100
    AVIndexEntry *sample = NULL;
2101
    int64_t best_dts = INT64_MAX;
2102
    int i;
B
Baptiste Coudurier 已提交
2103
    for (i = 0; i < s->nb_streams; i++) {
2104 2105
        AVStream *avst = s->streams[i];
        MOVStreamContext *msc = avst->priv_data;
2106
        if (msc->pb && msc->current_sample < avst->nb_index_entries) {
2107
            AVIndexEntry *current_sample = &avst->index_entries[msc->current_sample];
2108
            int64_t dts = av_rescale(current_sample->timestamp, AV_TIME_BASE, msc->time_scale);
B
Baptiste Coudurier 已提交
2109
            dprintf(s, "stream %d, sample %d, dts %"PRId64"\n", i, msc->current_sample, dts);
B
Baptiste Coudurier 已提交
2110 2111
            if (!sample || (url_is_streamed(s->pb) && current_sample->pos < sample->pos) ||
                (!url_is_streamed(s->pb) &&
2112
                 ((msc->pb != s->pb && dts < best_dts) || (msc->pb == s->pb &&
B
Baptiste Coudurier 已提交
2113
                 ((FFABS(best_dts - dts) <= AV_TIME_BASE && current_sample->pos < sample->pos) ||
2114
                  (FFABS(best_dts - dts) > AV_TIME_BASE && dts < best_dts)))))) {
2115 2116
                sample = current_sample;
                best_dts = dts;
2117
                *st = avst;
2118
            }
2119 2120
        }
    }
2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132
    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);
2133 2134 2135
    if (!sample) {
        mov->found_mdat = 0;
        if (!url_is_streamed(s->pb) ||
2136
            mov_read_default(mov, s->pb, (MOVAtom){ 0, 0, INT64_MAX }) < 0 ||
2137
            url_feof(s->pb))
B
Baptiste Coudurier 已提交
2138
            return AVERROR_EOF;
2139 2140 2141
        dprintf(s, "read fragments, offset 0x%llx\n", url_ftell(s->pb));
        goto retry;
    }
2142
    sc = st->priv_data;
2143 2144
    /* must be done just before reading, to avoid infinite loop on sample */
    sc->current_sample++;
2145 2146

    if (st->discard != AVDISCARD_ALL) {
R
Reimar Döffinger 已提交
2147 2148 2149 2150 2151 2152
        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);
2153 2154
        if (ret < 0)
            return ret;
R
Reimar Döffinger 已提交
2155 2156 2157 2158 2159 2160 2161 2162 2163
#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;
        }
2164
#endif
2165 2166
    }

2167 2168 2169
    pkt->stream_index = sc->ffindex;
    pkt->dts = sample->timestamp;
    if (sc->ctts_data) {
2170
        pkt->pts = pkt->dts + sc->dts_shift + sc->ctts_data[sc->ctts_index].duration;
2171
        /* update ctts context */
2172 2173 2174 2175 2176
        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;
2177
        }
2178 2179
        if (sc->wrong_dts)
            pkt->dts = AV_NOPTS_VALUE;
2180
    } else {
2181
        int64_t next_dts = (sc->current_sample < st->nb_index_entries) ?
2182 2183
            st->index_entries[sc->current_sample].timestamp : st->duration;
        pkt->duration = next_dts - pkt->dts;
2184
        pkt->pts = pkt->dts;
2185
    }
2186 2187
    if (st->discard == AVDISCARD_ALL)
        goto retry;
2188 2189
    pkt->flags |= sample->flags & AVINDEX_KEYFRAME ? PKT_FLAG_KEY : 0;
    pkt->pos = sample->pos;
2190 2191
    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);
2192 2193
    return 0;
}
2194

2195 2196 2197 2198 2199
static int mov_seek_stream(AVStream *st, int64_t timestamp, int flags)
{
    MOVStreamContext *sc = st->priv_data;
    int sample, time_sample;
    int i;
2200

2201
    sample = av_index_search_timestamp(st, timestamp, flags);
M
Michel Bardiaux 已提交
2202
    dprintf(st->codec, "stream %d, timestamp %"PRId64", sample %d\n", st->index, timestamp, sample);
2203 2204 2205
    if (sample < 0) /* not sure what to do */
        return -1;
    sc->current_sample = sample;
B
Baptiste Coudurier 已提交
2206
    dprintf(st->codec, "stream %d, found sample %d\n", st->index, sc->current_sample);
2207 2208 2209 2210
    /* adjust ctts index */
    if (sc->ctts_data) {
        time_sample = 0;
        for (i = 0; i < sc->ctts_count; i++) {
2211 2212
            int next = time_sample + sc->ctts_data[i].count;
            if (next > sc->current_sample) {
2213 2214
                sc->ctts_index = i;
                sc->ctts_sample = sc->current_sample - time_sample;
2215
                break;
2216
            }
2217
            time_sample = next;
2218 2219
        }
    }
2220
    return sample;
2221 2222
}

2223
static int mov_read_seek(AVFormatContext *s, int stream_index, int64_t sample_time, int flags)
G
Gael Chardon 已提交
2224
{
2225 2226 2227 2228
    AVStream *st;
    int64_t seek_timestamp, timestamp;
    int sample;
    int i;
G
Gael Chardon 已提交
2229

2230
    if (stream_index >= s->nb_streams)
G
Gael Chardon 已提交
2231
        return -1;
2232 2233
    if (sample_time < 0)
        sample_time = 0;
G
Gael Chardon 已提交
2234

2235 2236 2237
    st = s->streams[stream_index];
    sample = mov_seek_stream(st, sample_time, flags);
    if (sample < 0)
G
Gael Chardon 已提交
2238 2239
        return -1;

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

2243 2244
    for (i = 0; i < s->nb_streams; i++) {
        st = s->streams[i];
2245
        if (stream_index == i)
2246
            continue;
G
Gael Chardon 已提交
2247

2248 2249
        timestamp = av_rescale_q(seek_timestamp, s->streams[stream_index]->time_base, st->time_base);
        mov_seek_stream(st, timestamp, flags);
2250
    }
G
Gael Chardon 已提交
2251 2252 2253
    return 0;
}

Z
Zdenek Kabelac 已提交
2254
static int mov_read_close(AVFormatContext *s)
2255
{
2256
    MOVContext *mov = s->priv_data;
2257 2258 2259
    int i, j;

    for (i = 0; i < s->nb_streams; i++) {
2260 2261
        AVStream *st = s->streams[i];
        MOVStreamContext *sc = st->priv_data;
2262

2263
        av_freep(&sc->ctts_data);
2264
        for (j = 0; j < sc->drefs_count; j++)
2265 2266 2267 2268
            av_freep(&sc->drefs[j].path);
        av_freep(&sc->drefs);
        if (sc->pb && sc->pb != s->pb)
            url_fclose(sc->pb);
2269 2270

        av_freep(&st->codec->palctrl);
2271
    }
2272 2273 2274

    if (mov->dv_demux) {
        for(i = 0; i < mov->dv_fctx->nb_streams; i++) {
2275 2276 2277 2278 2279 2280
            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);
    }
2281

B
Baptiste Coudurier 已提交
2282
    av_freep(&mov->trex_data);
2283

2284 2285 2286
    return 0;
}

2287
AVInputFormat mov_demuxer = {
2288
    "mov,mp4,m4a,3gp,3g2,mj2",
2289
    NULL_IF_CONFIG_SMALL("QuickTime/MPEG-4/Motion JPEG 2000 format"),
F
Fabrice Bellard 已提交
2290 2291
    sizeof(MOVContext),
    mov_probe,
2292 2293 2294
    mov_read_header,
    mov_read_packet,
    mov_read_close,
G
Gael Chardon 已提交
2295
    mov_read_seek,
2296
};