mov.c 75.6 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 503
    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);
    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 912 913 914
            if (codec_name[0] <= 31) {
                memcpy(st->codec->codec_name, &codec_name[1],codec_name[0]);
                st->codec->codec_name[codec_name[0]] = 0;
            }
915

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

            /* if the depth is 2, 4, or 8 bpp, file is palettized */
925
            if ((color_depth == 2) || (color_depth == 4) ||
926
                (color_depth == 8)) {
927 928 929 930
                /* for palette traversal */
                unsigned int color_start, color_count, color_end;
                unsigned char r, g, b;

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

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

995
            st->codec->codec_id = id;
996 997
            get_be16(pb); /* revision level */
            get_be32(pb); /* vendor */
998

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

1003
            sc->audio_cid = get_be16(pb);
1004 1005 1006 1007
            get_be16(pb); /* packet size = 0 */

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

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

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

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

B
Baptiste Coudurier 已提交
1093
    if(st->codec->codec_type==CODEC_TYPE_AUDIO && st->codec->sample_rate==0 && sc->time_scale>1)
1094
        st->codec->sample_rate= sc->time_scale;
1095

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

1148 1149 1150
    return 0;
}

1151
static int mov_read_stsc(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1152
{
1153 1154
    AVStream *st;
    MOVStreamContext *sc;
1155
    unsigned int i, entries;
1156

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

1162
    get_byte(pb); /* version */
1163
    get_be24(pb); /* flags */
1164 1165

    entries = get_be32(pb);
1166

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

1169 1170
    if(entries >= UINT_MAX / sizeof(*sc->stsc_data))
        return -1;
1171 1172
    sc->stsc_data = av_malloc(entries * sizeof(*sc->stsc_data));
    if (!sc->stsc_data)
1173 1174 1175
        return AVERROR(ENOMEM);
    sc->stsc_count = entries;

1176
    for(i=0; i<entries; i++) {
1177 1178 1179
        sc->stsc_data[i].first = get_be32(pb);
        sc->stsc_data[i].count = get_be32(pb);
        sc->stsc_data[i].id = get_be32(pb);
1180 1181 1182 1183
    }
    return 0;
}

1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212
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;
}

1213
static int mov_read_stss(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1214
{
1215 1216
    AVStream *st;
    MOVStreamContext *sc;
1217
    unsigned int i, entries;
1218

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

1224
    get_byte(pb); /* version */
1225
    get_be24(pb); /* flags */
1226 1227

    entries = get_be32(pb);
1228

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

B
Baptiste Coudurier 已提交
1231
    if(entries >= UINT_MAX / sizeof(int))
1232
        return -1;
B
Baptiste Coudurier 已提交
1233
    sc->keyframes = av_malloc(entries * sizeof(int));
1234
    if (!sc->keyframes)
1235 1236 1237
        return AVERROR(ENOMEM);
    sc->keyframe_count = entries;

1238 1239
    for(i=0; i<entries; i++) {
        sc->keyframes[i] = get_be32(pb);
1240
        //dprintf(c->fc, "keyframes[]=%d\n", sc->keyframes[i]);
1241 1242 1243 1244
    }
    return 0;
}

1245
static int mov_read_stsz(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1246
{
1247 1248
    AVStream *st;
    MOVStreamContext *sc;
1249 1250 1251
    unsigned int i, entries, sample_size, field_size, num_bytes;
    GetBitContext gb;
    unsigned char* buf;
1252

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

1258
    get_byte(pb); /* version */
1259
    get_be24(pb); /* flags */
1260

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

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

1275
    sc->sample_count = entries;
1276 1277 1278
    if (sample_size)
        return 0;

1279 1280 1281 1282 1283
    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;
    }

1284
    if (entries >= UINT_MAX / sizeof(int) || entries >= (UINT_MAX - 4) / field_size)
1285
        return -1;
B
Baptiste Coudurier 已提交
1286
    sc->sample_sizes = av_malloc(entries * sizeof(int));
1287
    if (!sc->sample_sizes)
1288 1289
        return AVERROR(ENOMEM);

1290 1291
    num_bytes = (entries*field_size+4)>>3;

1292
    buf = av_malloc(num_bytes+FF_INPUT_BUFFER_PADDING_SIZE);
1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305
    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);

1306
    for(i=0; i<entries; i++)
1307 1308 1309
        sc->sample_sizes[i] = get_bits_long(&gb, field_size);

    av_free(buf);
1310 1311 1312
    return 0;
}

1313
static int mov_read_stts(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1314
{
1315 1316
    AVStream *st;
    MOVStreamContext *sc;
1317
    unsigned int i, entries;
1318 1319
    int64_t duration=0;
    int64_t total_sample_count=0;
1320

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

1326
    get_byte(pb); /* version */
1327
    get_be24(pb); /* flags */
1328
    entries = get_be32(pb);
1329 1330 1331

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

1332
    if(entries >= UINT_MAX / sizeof(*sc->stts_data))
1333
        return -1;
1334
    sc->stts_data = av_malloc(entries * sizeof(*sc->stts_data));
1335
    if (!sc->stts_data)
1336 1337
        return AVERROR(ENOMEM);
    sc->stts_count = entries;
1338

1339
    for(i=0; i<entries; i++) {
M
cleanup  
Michael Niedermayer 已提交
1340 1341
        int sample_duration;
        int sample_count;
1342

1343
        sample_count=get_be32(pb);
1344
        sample_duration = get_be32(pb);
1345 1346 1347
        sc->stts_data[i].count= sample_count;
        sc->stts_data[i].duration= sample_duration;

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

B
Baptiste Coudurier 已提交
1350
        duration+=(int64_t)sample_duration*sample_count;
1351 1352 1353
        total_sample_count+=sample_count;
    }

1354 1355 1356
    st->nb_frames= total_sample_count;
    if(duration)
        st->duration= duration;
1357 1358 1359
    return 0;
}

1360 1361 1362 1363 1364
static int mov_read_cslg(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
{
    AVStream *st;
    MOVStreamContext *sc;

B
Baptiste Coudurier 已提交
1365
    if (c->fc->nb_streams < 1)
1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382
        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;
}

1383
static int mov_read_ctts(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
M
Michael Niedermayer 已提交
1384
{
1385 1386
    AVStream *st;
    MOVStreamContext *sc;
M
Michael Niedermayer 已提交
1387 1388
    unsigned int i, entries;

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

M
Michael Niedermayer 已提交
1394
    get_byte(pb); /* version */
1395
    get_be24(pb); /* flags */
M
Michael Niedermayer 已提交
1396
    entries = get_be32(pb);
1397 1398 1399

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

1400
    if(entries >= UINT_MAX / sizeof(*sc->ctts_data))
M
Michael Niedermayer 已提交
1401
        return -1;
1402
    sc->ctts_data = av_malloc(entries * sizeof(*sc->ctts_data));
1403
    if (!sc->ctts_data)
1404 1405
        return AVERROR(ENOMEM);
    sc->ctts_count = entries;
1406

M
Michael Niedermayer 已提交
1407
    for(i=0; i<entries; i++) {
1408 1409 1410 1411 1412
        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 已提交
1413 1414 1415 1416
    }
    return 0;
}

1417 1418 1419
static void mov_build_index(MOVContext *mov, AVStream *st)
{
    MOVStreamContext *sc = st->priv_data;
1420
    int64_t current_offset;
1421 1422 1423 1424
    int64_t current_dts = 0;
    unsigned int stts_index = 0;
    unsigned int stsc_index = 0;
    unsigned int stss_index = 0;
1425
    unsigned int stps_index = 0;
1426 1427
    unsigned int i, j;

1428 1429
    /* adjust first dts according to edit list */
    if (sc->time_offset) {
1430
        int rescaled = sc->time_offset < 0 ? av_rescale(sc->time_offset, sc->time_scale, mov->time_scale) : sc->time_offset;
1431
        current_dts = -rescaled;
1432 1433 1434 1435 1436 1437
        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;
        }
1438 1439
    }

1440 1441 1442
    /* 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)) {
1443 1444
        unsigned int current_sample = 0;
        unsigned int stts_sample = 0;
1445
        unsigned int sample_size;
1446 1447 1448
        unsigned int distance = 0;
        int key_off = sc->keyframes && sc->keyframes[0] == 1;

1449 1450
        current_dts -= sc->dts_shift;

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

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

1485
                current_offset += sample_size;
1486
                current_dts += sc->stts_data[stts_index].duration;
1487 1488 1489 1490 1491 1492 1493 1494 1495
                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++;
                }
            }
        }
1496
    } else {
1497
        for (i = 0; i < sc->chunk_count; i++) {
1498 1499
            unsigned chunk_samples;

1500
            current_offset = sc->chunk_offsets[i];
1501 1502
            if (stsc_index + 1 < sc->stsc_count &&
                i + 1 == sc->stsc_data[stsc_index + 1].first)
1503
                stsc_index++;
1504
            chunk_samples = sc->stsc_data[stsc_index].count;
1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521

            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;
1522
                    } else {
1523 1524
                        samples = FFMIN(1024, chunk_samples);
                        size = samples * sc->sample_size;
1525 1526
                    }
                }
1527 1528

                av_add_index_entry(st, current_offset, current_dts, size, 0, AVINDEX_KEYFRAME);
1529 1530
                dprintf(mov->fc, "AVIndex stream %d, chunk %d, offset %"PRIx64", dts %"PRId64", "
                        "size %d, duration %d\n", st->index, i, current_offset, current_dts,
1531 1532 1533
                        size, samples);

                current_offset += size;
1534
                current_dts += samples;
1535
                chunk_samples -= samples;
1536 1537 1538 1539
            }
        }
    }
}
1540

1541
static int mov_read_trak(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1542 1543 1544
{
    AVStream *st;
    MOVStreamContext *sc;
1545
    int ret;
1546 1547

    st = av_new_stream(c->fc, c->fc->nb_streams);
B
Baptiste Coudurier 已提交
1548
    if (!st) return AVERROR(ENOMEM);
B
Baptiste Coudurier 已提交
1549
    sc = av_mallocz(sizeof(MOVStreamContext));
1550
    if (!sc) return AVERROR(ENOMEM);
1551 1552

    st->priv_data = sc;
1553
    st->codec->codec_type = CODEC_TYPE_DATA;
1554
    sc->ffindex = st->index;
1555

1556 1557 1558 1559
    if ((ret = mov_read_default(c, pb, atom)) < 0)
        return ret;

    /* sanity checks */
1560 1561
    if (sc->chunk_count && (!sc->stts_count || !sc->stsc_count ||
                            (!sc->sample_size && !sc->sample_count))) {
1562 1563
        av_log(c->fc, AV_LOG_ERROR, "stream %d, missing mandatory atoms, broken header\n",
               st->index);
1564 1565
        return 0;
    }
1566 1567 1568 1569

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

1570
    av_set_pts_info(st, 64, 1, sc->time_scale);
1571

1572
    if (st->codec->codec_type == CODEC_TYPE_AUDIO &&
1573 1574 1575 1576 1577
        !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);
    }
1578 1579 1580 1581 1582

    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)
1583 1584
            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));
1585 1586 1587 1588
    } else
        sc->pb = c->fc->pb;

    switch (st->codec->codec_id) {
1589
#if CONFIG_H261_DECODER
1590 1591
    case CODEC_ID_H261:
#endif
1592
#if CONFIG_H263_DECODER
1593 1594
    case CODEC_ID_H263:
#endif
1595 1596 1597
#if CONFIG_H264_DECODER
    case CODEC_ID_H264:
#endif
1598
#if CONFIG_MPEG4_DECODER
1599 1600
    case CODEC_ID_MPEG4:
#endif
1601
        st->codec->width = 0; /* let decoder init width/height */
1602 1603 1604
        st->codec->height= 0;
        break;
    }
B
Baptiste Coudurier 已提交
1605 1606 1607

    /* Do not need those anymore. */
    av_freep(&sc->chunk_offsets);
1608
    av_freep(&sc->stsc_data);
B
Baptiste Coudurier 已提交
1609 1610 1611
    av_freep(&sc->sample_sizes);
    av_freep(&sc->keyframes);
    av_freep(&sc->stts_data);
1612
    av_freep(&sc->stps_data);
B
Baptiste Coudurier 已提交
1613

1614
    return 0;
1615 1616
}

1617
static int mov_read_ilst(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1618 1619 1620 1621 1622 1623 1624 1625
{
    int ret;
    c->itunes_metadata = 1;
    ret = mov_read_default(c, pb, atom);
    c->itunes_metadata = 0;
    return ret;
}

1626
static int mov_read_meta(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1627
{
1628 1629 1630 1631 1632 1633 1634 1635 1636 1637
    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;
1638 1639
}

1640
static int mov_read_tkhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1641
{
1642 1643 1644 1645 1646
    int i;
    int width;
    int height;
    int64_t disp_transform[2];
    int display_matrix[3][2];
1647 1648 1649 1650 1651 1652 1653 1654
    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;
1655

1656
    version = get_byte(pb);
1657
    get_be24(pb); /* flags */
1658 1659 1660 1661 1662 1663 1664
    /*
    MOV_TRACK_ENABLED 0x0001
    MOV_TRACK_IN_MOVIE 0x0002
    MOV_TRACK_IN_PREVIEW 0x0004
    MOV_TRACK_IN_POSTER 0x0008
    */

B
Baptiste Coudurier 已提交
1665 1666 1667 1668 1669 1670 1671
    if (version == 1) {
        get_be64(pb);
        get_be64(pb);
    } else {
        get_be32(pb); /* creation time */
        get_be32(pb); /* modification time */
    }
1672 1673
    st->id = (int)get_be32(pb); /* track id (NOT 0 !)*/
    get_be32(pb); /* reserved */
1674

1675 1676
    /* highlevel (considering edits) duration in movie timebase */
    (version == 1) ? get_be64(pb) : get_be32(pb);
1677 1678 1679 1680 1681 1682 1683 1684
    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 */

1685 1686 1687 1688 1689 1690 1691 1692
    //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)
    }
1693

1694 1695
    width = get_be32(pb);       // 16.16 fixed point track width
    height = get_be32(pb);      // 16.16 fixed point track height
1696 1697
    sc->width = width >> 16;
    sc->height = height >> 16;
1698

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

1723
static int mov_read_tfhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
1724 1725 1726 1727 1728 1729 1730 1731 1732
{
    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);
1733
    if (!track_id)
B
Baptiste Coudurier 已提交
1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757
        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;
}

1758
static int mov_read_trex(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
1759 1760 1761 1762 1763
{
    MOVTrackExt *trex;

    if ((uint64_t)c->trex_count+1 >= UINT_MAX / sizeof(*c->trex_data))
        return -1;
1764 1765
    trex = av_realloc(c->trex_data, (c->trex_count+1)*sizeof(*c->trex_data));
    if (!trex)
B
Baptiste Coudurier 已提交
1766
        return AVERROR(ENOMEM);
1767
    c->trex_data = trex;
B
Baptiste Coudurier 已提交
1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778
    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;
}

1779
static int mov_read_trun(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
1780 1781
{
    MOVFragment *frag = &c->fragment;
1782
    AVStream *st = NULL;
1783
    MOVStreamContext *sc;
B
Baptiste Coudurier 已提交
1784 1785 1786 1787 1788 1789
    uint64_t offset;
    int64_t dts;
    int data_offset = 0;
    unsigned entries, first_sample_flags = frag->flags;
    int flags, distance, i;

1790 1791 1792 1793 1794 1795 1796 1797
    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);
1798
        return -1;
1799
    }
1800
    sc = st->priv_data;
B
Baptiste Coudurier 已提交
1801 1802 1803 1804 1805 1806 1807 1808 1809
    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) {
1810
        MOVStts *ctts_data;
B
Baptiste Coudurier 已提交
1811 1812
        if ((uint64_t)entries+sc->ctts_count >= UINT_MAX/sizeof(*sc->ctts_data))
            return -1;
1813 1814 1815
        ctts_data = av_realloc(sc->ctts_data,
                               (entries+sc->ctts_count)*sizeof(*sc->ctts_data));
        if (!ctts_data)
B
Baptiste Coudurier 已提交
1816
            return AVERROR(ENOMEM);
1817
        sc->ctts_data = ctts_data;
B
Baptiste Coudurier 已提交
1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845
    }
    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++;
1846
        dts += sample_duration;
B
Baptiste Coudurier 已提交
1847 1848 1849 1850 1851 1852 1853
        offset += sample_size;
    }
    frag->moof_offset = offset;
    st->duration = dts;
    return 0;
}

1854 1855 1856
/* 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/ */
1857
static int mov_read_wide(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1858 1859 1860 1861 1862 1863 1864 1865 1866 1867 1868 1869
{
    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;
1870
    if (atom.type != MKTAG('m','d','a','t')) {
1871 1872 1873 1874 1875 1876 1877
        url_fskip(pb, atom.size);
        return 0;
    }
    err = mov_read_mdat(c, pb, atom);
    return err;
}

1878
static int mov_read_cmov(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1879
{
1880
#if CONFIG_ZLIB
1881
    ByteIOContext ctx;
1882 1883
    uint8_t *cmov_data;
    uint8_t *moov_data; /* uncompressed data */
1884
    long cmov_len, moov_len;
1885
    int ret = -1;
1886

1887
    get_be32(pb); /* dcom atom */
1888
    if (get_le32(pb) != MKTAG('d','c','o','m'))
1889
        return -1;
1890
    if (get_le32(pb) != MKTAG('z','l','i','b')) {
B
Benoit Fouet 已提交
1891
        av_log(c->fc, AV_LOG_ERROR, "unknown compression for cmov atom !");
1892 1893 1894
        return -1;
    }
    get_be32(pb); /* cmvd atom */
1895
    if (get_le32(pb) != MKTAG('c','m','v','d'))
1896 1897
        return -1;
    moov_len = get_be32(pb); /* uncompressed size */
1898
    cmov_len = atom.size - 6 * 4;
1899

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

G
Gael Chardon 已提交
1930
/* edit list atom */
1931
static int mov_read_elst(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
G
Gael Chardon 已提交
1932
{
1933
    MOVStreamContext *sc;
B
Baptiste Coudurier 已提交
1934 1935
    int i, edit_count;

1936 1937 1938 1939
    if (c->fc->nb_streams < 1)
        return 0;
    sc = c->fc->streams[c->fc->nb_streams-1]->priv_data;

B
Baptiste Coudurier 已提交
1940
    get_byte(pb); /* version */
1941
    get_be24(pb); /* flags */
B
Baptiste Coudurier 已提交
1942
    edit_count = get_be32(pb); /* entries */
B
Baptiste Coudurier 已提交
1943

1944 1945 1946
    if((uint64_t)edit_count*12+8 > atom.size)
        return -1;

B
Baptiste Coudurier 已提交
1947
    for(i=0; i<edit_count; i++){
1948
        int time;
1949
        int duration = get_be32(pb); /* Track duration */
1950
        time = get_be32(pb); /* Media time */
B
Baptiste Coudurier 已提交
1951
        get_be32(pb); /* Media rate */
1952 1953
        if (i == 0 && time >= -1) {
            sc->time_offset = time != -1 ? time : -duration;
1954
        }
B
Baptiste Coudurier 已提交
1955
    }
1956 1957 1958 1959 1960

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

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

1965
static const MOVParseTableEntry mov_default_parse_table[] = {
1966
{ MKTAG('a','v','s','s'), mov_read_extradata },
1967
{ MKTAG('c','o','6','4'), mov_read_stco },
1968
{ MKTAG('c','s','l','g'), mov_read_cslg },
1969 1970 1971 1972 1973 1974 1975 1976 1977 1978
{ 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 },
1979
{ MKTAG('i','l','s','t'), mov_read_ilst },
1980 1981 1982 1983
{ 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 },
1984
{ MKTAG('m','e','t','a'), mov_read_meta },
1985 1986 1987 1988 1989 1990 1991 1992
{ 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 },
1993
{ MKTAG('p','a','s','p'), mov_read_pasp },
1994 1995
{ MKTAG('s','t','b','l'), mov_read_default },
{ MKTAG('s','t','c','o'), mov_read_stco },
1996
{ MKTAG('s','t','p','s'), mov_read_stps },
1997 1998 1999 2000 2001
{ 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 },
2002
{ MKTAG('s','t','z','2'), mov_read_stsz }, /* compact sample size */
2003 2004 2005 2006 2007 2008
{ 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 已提交
2009
{ MKTAG('u','d','t','a'), mov_read_default },
2010 2011 2012 2013
{ 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 已提交
2014
{ 0, NULL }
2015 2016
};

F
Fabrice Bellard 已提交
2017 2018
static int mov_probe(AVProbeData *p)
{
2019 2020
    unsigned int offset;
    uint32_t tag;
2021
    int score = 0;
2022

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

Z
Zdenek Kabelac 已提交
2062
static int mov_read_header(AVFormatContext *s, AVFormatParameters *ap)
2063
{
2064
    MOVContext *mov = s->priv_data;
2065
    ByteIOContext *pb = s->pb;
B
Baptiste Coudurier 已提交
2066
    int err;
2067
    MOVAtom atom = { 0, 0, 0 };
2068 2069

    mov->fc = s;
2070 2071
    /* .mov and .mp4 aren't streamable anyway (only progressive download if moov is before mdat) */
    if(!url_is_streamed(pb))
2072
        atom.size = url_fsize(pb);
2073
    else
B
Baptiste Coudurier 已提交
2074
        atom.size = INT64_MAX;
2075 2076

    /* check MOV header */
B
Baptiste Coudurier 已提交
2077 2078 2079 2080 2081 2082
    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");
2083
        return -1;
2084
    }
B
Baptiste Coudurier 已提交
2085
    dprintf(mov->fc, "on_parse_exit_offset=%lld\n", url_ftell(pb));
2086

2087 2088 2089
    return 0;
}

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

    if (st->discard != AVDISCARD_ALL) {
R
Reimar Döffinger 已提交
2139 2140 2141 2142 2143 2144
        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);
2145 2146
        if (ret < 0)
            return ret;
R
Reimar Döffinger 已提交
2147 2148 2149 2150 2151 2152 2153 2154 2155
#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;
        }
2156
#endif
2157 2158
    }

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

2187 2188 2189 2190 2191
static int mov_seek_stream(AVStream *st, int64_t timestamp, int flags)
{
    MOVStreamContext *sc = st->priv_data;
    int sample, time_sample;
    int i;
2192

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

2215
static int mov_read_seek(AVFormatContext *s, int stream_index, int64_t sample_time, int flags)
G
Gael Chardon 已提交
2216
{
2217 2218 2219 2220
    AVStream *st;
    int64_t seek_timestamp, timestamp;
    int sample;
    int i;
G
Gael Chardon 已提交
2221

2222
    if (stream_index >= s->nb_streams)
G
Gael Chardon 已提交
2223
        return -1;
2224 2225
    if (sample_time < 0)
        sample_time = 0;
G
Gael Chardon 已提交
2226

2227 2228 2229
    st = s->streams[stream_index];
    sample = mov_seek_stream(st, sample_time, flags);
    if (sample < 0)
G
Gael Chardon 已提交
2230 2231
        return -1;

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

2235 2236
    for (i = 0; i < s->nb_streams; i++) {
        st = s->streams[i];
2237
        if (stream_index == i)
2238
            continue;
G
Gael Chardon 已提交
2239

2240 2241
        timestamp = av_rescale_q(seek_timestamp, s->streams[stream_index]->time_base, st->time_base);
        mov_seek_stream(st, timestamp, flags);
2242
    }
G
Gael Chardon 已提交
2243 2244 2245
    return 0;
}

Z
Zdenek Kabelac 已提交
2246
static int mov_read_close(AVFormatContext *s)
2247
{
2248
    MOVContext *mov = s->priv_data;
2249 2250 2251
    int i, j;

    for (i = 0; i < s->nb_streams; i++) {
2252 2253
        AVStream *st = s->streams[i];
        MOVStreamContext *sc = st->priv_data;
2254

2255
        av_freep(&sc->ctts_data);
2256
        for (j = 0; j < sc->drefs_count; j++)
2257 2258 2259 2260
            av_freep(&sc->drefs[j].path);
        av_freep(&sc->drefs);
        if (sc->pb && sc->pb != s->pb)
            url_fclose(sc->pb);
2261 2262

        av_freep(&st->codec->palctrl);
2263
    }
2264 2265 2266

    if (mov->dv_demux) {
        for(i = 0; i < mov->dv_fctx->nb_streams; i++) {
2267 2268 2269 2270 2271 2272
            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);
    }
2273

B
Baptiste Coudurier 已提交
2274
    av_freep(&mov->trex_data);
2275

2276 2277 2278
    return 0;
}

2279
AVInputFormat mov_demuxer = {
2280
    "mov,mp4,m4a,3gp,3g2,mj2",
2281
    NULL_IF_CONFIG_SMALL("QuickTime/MPEG-4/Motion JPEG 2000 format"),
F
Fabrice Bellard 已提交
2282 2283
    sizeof(MOVContext),
    mov_probe,
2284 2285 2286
    mov_read_header,
    mov_read_packet,
    mov_read_close,
G
Gael Chardon 已提交
2287
    mov_read_seek,
2288
};