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

#include <limits.h>
23

B
Baptiste Coudurier 已提交
24
//#define DEBUG
25

26
#include "libavutil/intreadwrite.h"
27
#include "libavutil/avstring.h"
28
#include "avformat.h"
29
#include "riff.h"
30
#include "isom.h"
31
#include "dv.h"
32 33
#include "libavcodec/mpeg4audio.h"
#include "libavcodec/mpegaudiodata.h"
34

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

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

59 60
#include "qtpalette.h"

G
Gael Chardon 已提交
61

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

65 66 67 68 69
/* the QuickTime file format is quite convoluted...
 * it has lots of index tables, each indexing something in another one...
 * Here we just use what is needed to read the chunks
 */

70
typedef struct {
B
Baptiste Coudurier 已提交
71 72 73
    int first;
    int count;
    int id;
74
} MOVStsc;
75

76 77 78
typedef struct {
    uint32_t type;
    char *path;
79
} MOVDref;
80

81 82 83 84
typedef struct {
    uint32_t type;
    int64_t offset;
    int64_t size; /* total size (excluding the size and type fields) */
85
} MOVAtom;
86

87 88
struct MOVParseTableEntry;

B
Baptiste Coudurier 已提交
89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106
typedef struct {
    unsigned track_id;
    uint64_t base_data_offset;
    uint64_t moof_offset;
    unsigned stsd_id;
    unsigned duration;
    unsigned size;
    unsigned flags;
} MOVFragment;

typedef struct {
    unsigned track_id;
    unsigned stsd_id;
    unsigned duration;
    unsigned size;
    unsigned flags;
} MOVTrackExt;

107
typedef struct MOVStreamContext {
108
    ByteIOContext *pb;
109
    int ffindex; /* the ffmpeg stream id */
B
Baptiste Coudurier 已提交
110
    int next_chunk;
B
Baptiste Coudurier 已提交
111
    unsigned int chunk_count;
112
    int64_t *chunk_offsets;
B
Baptiste Coudurier 已提交
113
    unsigned int stts_count;
114
    MOVStts *stts_data;
B
Baptiste Coudurier 已提交
115
    unsigned int ctts_count;
116
    MOVStts *ctts_data;
117 118
    unsigned int stsc_count;
    MOVStsc *stsc_data;
119 120
    int ctts_index;
    int ctts_sample;
B
Baptiste Coudurier 已提交
121 122
    unsigned int sample_size;
    unsigned int sample_count;
B
Baptiste Coudurier 已提交
123
    int *sample_sizes;
B
Baptiste Coudurier 已提交
124
    unsigned int keyframe_count;
B
Baptiste Coudurier 已提交
125
    int *keyframes;
126
    int time_scale;
127
    int time_rate;
B
Baptiste Coudurier 已提交
128
    int time_offset; ///< time offset of the first edit list entry
B
Baptiste Coudurier 已提交
129
    int current_sample;
130 131
    unsigned int bytes_per_frame;
    unsigned int samples_per_frame;
132
    int dv_audio_container;
133
    int pseudo_stream_id; ///< -1 means demux all ids
134
    int16_t audio_cid; ///< stsd audio compression id
135
    unsigned drefs_count;
136
    MOVDref *drefs;
137
    int dref_id;
138
    int wrong_dts; ///< dts are wrong due to negative ctts
139 140
    int width;  ///< tkhd width
    int height; ///< tkhd height
141 142 143 144
} MOVStreamContext;

typedef struct MOVContext {
    AVFormatContext *fc;
145
    int time_scale;
B
Baptiste Coudurier 已提交
146
    int64_t duration; /* duration of the longest track */
147 148
    int found_moov; /* when both 'moov' and 'mdat' sections has been found */
    int found_mdat; /* we suppose we have enough data to read the file */
149
    AVPaletteControl palette_control;
150 151
    DVDemuxContext *dv_demux;
    AVFormatContext *dv_fctx;
152
    int isom; /* 1 if file is ISO Media (mp4/3gp) */
B
Baptiste Coudurier 已提交
153 154 155
    MOVFragment fragment; ///< current fragment in moof atom
    MOVTrackExt *trex_data;
    unsigned trex_count;
156
    int itunes_metadata; ///< metadata are itunes style
157 158 159 160 161 162 163
} MOVContext;


/* 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 已提交
164
  0: continue to parse next atom
D
Diego Biurrun 已提交
165
 <0: error occurred, exit
B
Baptiste Coudurier 已提交
166
*/
167 168
/* links atom IDs to parse functions */
typedef struct MOVParseTableEntry {
169
    uint32_t type;
170
    int (*parse)(MOVContext *ctx, ByteIOContext *pb, MOVAtom atom);
171 172
} MOVParseTableEntry;

173 174
static const MOVParseTableEntry mov_default_parse_table[];

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

    a.offset = atom.offset;
183

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

B
Baptiste Coudurier 已提交
212
        for (i = 0; mov_default_parse_table[i].type != 0
213
             && mov_default_parse_table[i].type != a.type; i++)
214 215
            /* empty */;

216
        if (mov_default_parse_table[i].type == 0) { /* skip leaf atoms data */
217
            url_fskip(pb, a.size);
218
        } else {
219
            int64_t start_pos = url_ftell(pb);
220
            int64_t left;
221
            err = mov_default_parse_table[i].parse(c, pb, a);
222
            if (url_is_streamed(pb) && c->found_moov && c->found_mdat)
B
Baptiste Coudurier 已提交
223
                break;
224 225 226
            left = a.size - url_ftell(pb) + start_pos;
            if (left > 0) /* skip garbage at atom end */
                url_fskip(pb, left);
227
        }
228

229
        a.offset += a.size;
230
        total_size += a.size;
231 232
    }

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

236 237 238
    return err;
}

239
static int mov_read_dref(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
240 241 242 243 244 245 246 247 248 249
{
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];
    MOVStreamContext *sc = st->priv_data;
    int entries, i, j;

    get_be32(pb); // version + flags
    entries = get_be32(pb);
    if (entries >= UINT_MAX / sizeof(*sc->drefs))
        return -1;
    sc->drefs = av_mallocz(entries * sizeof(*sc->drefs));
250 251 252
    if (!sc->drefs)
        return AVERROR(ENOMEM);
    sc->drefs_count = entries;
253 254

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

        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
286
                    av_free(dref->path);
287
                    dref->path = av_mallocz(len+1);
288 289
                    if (!dref->path)
                        return AVERROR(ENOMEM);
290
                    get_buffer(pb, dref->path, len);
291
                    if (len > volume_len && !strncmp(dref->path, volume, volume_len)) {
292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308
                        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;
}

309
static int mov_read_hdlr(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
310
{
311
    AVStream *st;
312 313
    uint32_t type;
    uint32_t ctype;
314

315 316 317 318 319
    if (c->fc->nb_streams < 1) // meta before first trak
        return 0;

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

320
    get_byte(pb); /* version */
321
    get_be24(pb); /* flags */
322 323 324 325 326

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

327 328 329 330
    dprintf(c->fc, "ctype= %c%c%c%c (0x%08x)\n", *((char *)&ctype), ((char *)&ctype)[1],
            ((char *)&ctype)[2], ((char *)&ctype)[3], (int) ctype);
    dprintf(c->fc, "stype= %c%c%c%c\n",
            *((char *)&type), ((char *)&type)[1], ((char *)&type)[2], ((char *)&type)[3]);
331 332
    if(!ctype)
        c->isom = 1;
333
    if     (type == MKTAG('v','i','d','e'))
334
        st->codec->codec_type = CODEC_TYPE_VIDEO;
335
    else if(type == MKTAG('s','o','u','n'))
336
        st->codec->codec_type = CODEC_TYPE_AUDIO;
337
    else if(type == MKTAG('m','1','a',' '))
338
        st->codec->codec_id = CODEC_ID_MP2;
339
    else if(type == MKTAG('s','u','b','p')) {
340 341
        st->codec->codec_type = CODEC_TYPE_SUBTITLE;
    }
342 343 344 345
    get_be32(pb); /* component  manufacture */
    get_be32(pb); /* component flags */
    get_be32(pb); /* component flags mask */

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

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

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

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

375 376 377 378
#define MP4ESDescrTag                   0x03
#define MP4DecConfigDescrTag            0x04
#define MP4DecSpecificDescrTag          0x05

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

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

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

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

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

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

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

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

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

482
/* this atom should contain all header atoms */
483
static int mov_read_moov(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
484
{
485 486
    if (mov_read_default(c, pb, atom) < 0)
        return -1;
487 488 489 490 491 492
    /* 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 */
}

493
static int mov_read_moof(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
494 495 496 497 498
{
    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);
}
499

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

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

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

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

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

    return 0;
}

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

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

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

B
Baptiste Coudurier 已提交
547
    c->duration = (version == 1) ? get_be64(pb) : get_be32(pb); /* duration */
548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566
    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;
}

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

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

574 575
    // currently SVQ3 decoder expect full STSD header - so let's fake it
    // this should be fixed and just SMI header should be passed
576
    av_free(st->codec->extradata);
577 578 579
    st->codec->extradata = av_mallocz(atom.size + 0x5a + FF_INPUT_BUFFER_PADDING_SIZE);
    if (!st->codec->extradata)
        return AVERROR(ENOMEM);
B
Baptiste Coudurier 已提交
580 581 582 583
    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);
584 585
    return 0;
}
586

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

592 593
    dprintf(c->fc, "enda %d\n", little_endian);
    if (little_endian == 1) {
594 595 596 597 598 599 600
        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;
601 602 603 604 605 606
        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;
607 608 609 610 611 612 613
        default:
            break;
        }
    }
    return 0;
}

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

    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;
625
    if(size > INT_MAX || (uint64_t)atom.size > INT_MAX)
626
        return -1;
627 628 629 630 631 632 633 634 635
    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);
636 637 638
    return 0;
}

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

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

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

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

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

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

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

    get_byte(pb); /* version */
689
    get_be24(pb); /* flags */
690 691

    entries = get_be32(pb);
692

693 694
    if(entries >= UINT_MAX/sizeof(int64_t))
        return -1;
695

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

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

710 711 712
    return 0;
}

713 714 715 716
/**
 * Compute codec id for 'lpcm' tag.
 * See CoreAudioTypes and AudioStreamBasicDescription at Apple.
 */
D
Diego Pettenò 已提交
717
static enum CodecID mov_get_lpcm_codec_id(int bps, int flags)
718 719 720 721
{
    if (flags & 1) { // floating point
        if (flags & 2) { // big endian
            if      (bps == 32) return CODEC_ID_PCM_F32BE;
722
            else if (bps == 64) return CODEC_ID_PCM_F64BE;
723
        } else {
724 725
            if      (bps == 32) return CODEC_ID_PCM_F32LE;
            else if (bps == 64) return CODEC_ID_PCM_F64LE;
726 727 728 729 730 731 732 733 734 735 736 737 738 739
        }
    } 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 已提交
740
            else if (bps == 16) return CODEC_ID_PCM_S16LE;
741 742 743 744
            else if (bps == 24) return CODEC_ID_PCM_S24LE;
            else if (bps == 32) return CODEC_ID_PCM_S32LE;
        }
    }
D
Diego Pettenò 已提交
745
    return CODEC_ID_NONE;
746 747
}

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

754
    get_byte(pb); /* version */
755
    get_be24(pb); /* flags */
756 757 758

    entries = get_be32(pb);

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

768 769
        get_be32(pb); /* reserved */
        get_be16(pb); /* reserved */
770
        dref_id = get_be16(pb);
771

772
        if (st->codec->codec_tag &&
773
            st->codec->codec_tag != format &&
774
            (c->fc->video_codec_id ? codec_get_id(codec_movvideo_tags, format) != c->fc->video_codec_id
775
                                   : st->codec->codec_tag != MKTAG('j','p','e','g'))
776
           ){
D
Diego Biurrun 已提交
777 778 779
            /* 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 已提交
780
            av_log(c->fc, AV_LOG_WARNING, "multiple fourcc not supported\n");
781 782 783
            url_fskip(pb, size - (url_ftell(pb) - start_pos));
            continue;
        }
784
        sc->pseudo_stream_id = st->codec->codec_tag ? -1 : pseudo_stream_id;
785
        sc->dref_id= dref_id;
786

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

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

808 809 810
        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);
811

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

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

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

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

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

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

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

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

                    for (j = 0; j < color_count; j++) {
                        r = color_table[j * 4 + 0];
                        g = color_table[j * 4 + 1];
                        b = color_table[j * 4 + 2];
                        c->palette_control.palette[j] =
                            (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);
891 892
                    if ((color_start <= 255) &&
                        (color_end <= 255)) {
M
Mike Melanson 已提交
893 894 895 896 897 898 899 900 901 902 903 904 905 906
                        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);
                            c->palette_control.palette[j] =
                                (r << 16) | (g << 8) | (b);
907
                        }
908 909
                    }
                }
910 911
                st->codec->palctrl = &c->palette_control;
                st->codec->palctrl->palette_changed = 1;
912
            } else
913
                st->codec->palctrl = NULL;
914
        } else if(st->codec->codec_type==CODEC_TYPE_AUDIO) {
915
            int bits_per_sample, flags;
916
            uint16_t version = get_be16(pb);
917

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

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

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

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

931 932 933 934 935 936 937 938 939 940 941 942 943
            //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 */
944
                    st->codec->bits_per_coded_sample = get_be32(pb); /* bits per channel if sound is uncompressed */
945
                    flags = get_be32(pb); /* lcpm format specific flag */
B
Baptiste Coudurier 已提交
946 947
                    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 */
948
                    if (format == MKTAG('l','p','c','m'))
949
                        st->codec->codec_id = mov_get_lpcm_codec_id(st->codec->bits_per_coded_sample, flags);
950 951 952
                }
            }

953
            switch (st->codec->codec_id) {
954 955
            case CODEC_ID_PCM_S8:
            case CODEC_ID_PCM_U8:
956
                if (st->codec->bits_per_coded_sample == 16)
957 958
                    st->codec->codec_id = CODEC_ID_PCM_S16BE;
                break;
959
            case CODEC_ID_PCM_S16LE:
960
            case CODEC_ID_PCM_S16BE:
961
                if (st->codec->bits_per_coded_sample == 8)
962
                    st->codec->codec_id = CODEC_ID_PCM_S8;
963
                else if (st->codec->bits_per_coded_sample == 24)
964 965 966
                    st->codec->codec_id =
                        st->codec->codec_id == CODEC_ID_PCM_S16BE ?
                        CODEC_ID_PCM_S24BE : CODEC_ID_PCM_S24LE;
967
                break;
968 969 970 971 972 973 974 975 976 977 978 979 980
            /* 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;
981 982 983 984
            case CODEC_ID_GSM:
                sc->samples_per_frame = 160;
                sc->bytes_per_frame = 33;
                break;
985 986
            default:
                break;
987
            }
988

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

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

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

1067 1068 1069
    return 0;
}

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

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

    entries = get_be32(pb);
1080

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

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

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

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

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

    entries = get_be32(pb);
1108

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

1251 1252 1253
    /* 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)) {
1254 1255 1256 1257 1258 1259 1260 1261 1262
        unsigned int current_sample = 0;
        unsigned int stts_sample = 0;
        unsigned int keyframe, sample_size;
        unsigned int distance = 0;
        int key_off = sc->keyframes && sc->keyframes[0] == 1;

        st->nb_frames = sc->sample_count;
        for (i = 0; i < sc->chunk_count; i++) {
            current_offset = sc->chunk_offsets[i];
1263 1264
            if (stsc_index + 1 < sc->stsc_count &&
                i + 1 == sc->stsc_data[stsc_index + 1].first)
1265
                stsc_index++;
1266
            for (j = 0; j < sc->stsc_data[stsc_index].count; j++) {
1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277
                if (current_sample >= sc->sample_count) {
                    av_log(mov->fc, AV_LOG_ERROR, "wrong sample count\n");
                    goto out;
                }
                keyframe = !sc->keyframe_count || current_sample+key_off == sc->keyframes[stss_index];
                if (keyframe) {
                    distance = 0;
                    if (stss_index + 1 < sc->keyframe_count)
                        stss_index++;
                }
                sample_size = sc->sample_size > 0 ? sc->sample_size : sc->sample_sizes[current_sample];
1278
                if(sc->pseudo_stream_id == -1 ||
1279
                   sc->stsc_data[stsc_index].id - 1 == sc->pseudo_stream_id) {
1280 1281
                    av_add_index_entry(st, current_offset, current_dts, sample_size, distance,
                                    keyframe ? AVINDEX_KEYFRAME : 0);
1282 1283 1284 1285
                    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);
                }
1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302
                current_offset += sample_size;
                assert(sc->stts_data[stts_index].duration % sc->time_rate == 0);
                current_dts += sc->stts_data[stts_index].duration / sc->time_rate;
                distance++;
                stts_sample++;
                current_sample++;
                if (stts_index + 1 < sc->stts_count && stts_sample == sc->stts_data[stts_index].count) {
                    stts_sample = 0;
                    stts_index++;
                }
            }
        }
    } else { /* read whole chunk */
        unsigned int chunk_samples, chunk_size, chunk_duration;
        unsigned int frames = 1;
        for (i = 0; i < sc->chunk_count; i++) {
            current_offset = sc->chunk_offsets[i];
1303 1304
            if (stsc_index + 1 < sc->stsc_count &&
                i + 1 == sc->stsc_data[stsc_index + 1].first)
1305
                stsc_index++;
1306
            chunk_samples = sc->stsc_data[stsc_index].count;
1307 1308 1309
            /* get chunk size, beware of alaw/ulaw/mace */
            if (sc->samples_per_frame > 0 &&
                (chunk_samples * sc->bytes_per_frame % sc->samples_per_frame == 0)) {
1310
                if (sc->samples_per_frame < 160)
1311 1312 1313 1314 1315 1316
                    chunk_size = chunk_samples * sc->bytes_per_frame / sc->samples_per_frame;
                else {
                    chunk_size = sc->bytes_per_frame;
                    frames = chunk_samples / sc->samples_per_frame;
                    chunk_samples = sc->samples_per_frame;
                }
1317
            } else
1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335
                chunk_size = chunk_samples * sc->sample_size;
            for (j = 0; j < frames; j++) {
                av_add_index_entry(st, current_offset, current_dts, chunk_size, 0, AVINDEX_KEYFRAME);
                /* get chunk duration */
                chunk_duration = 0;
                while (chunk_samples > 0) {
                    if (chunk_samples < sc->stts_data[stts_index].count) {
                        chunk_duration += sc->stts_data[stts_index].duration * chunk_samples;
                        sc->stts_data[stts_index].count -= chunk_samples;
                        break;
                    } else {
                        chunk_duration += sc->stts_data[stts_index].duration * chunk_samples;
                        chunk_samples -= sc->stts_data[stts_index].count;
                        if (stts_index + 1 < sc->stts_count)
                            stts_index++;
                    }
                }
                current_offset += sc->bytes_per_frame;
1336 1337 1338
                dprintf(mov->fc, "AVIndex stream %d, chunk %d, offset %"PRIx64", dts %"PRId64", "
                        "size %d, duration %d\n", st->index, i, current_offset, current_dts,
                        chunk_size, chunk_duration);
1339 1340 1341 1342 1343 1344 1345 1346 1347
                assert(chunk_duration % sc->time_rate == 0);
                current_dts += chunk_duration / sc->time_rate;
            }
        }
    }
 out:
    /* adjust sample count to avindex entries */
    sc->sample_count = st->nb_index_entries;
}
1348

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

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

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

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

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

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

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

1393 1394 1395 1396
    mov_build_index(c, st);

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

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

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

1424
    return 0;
1425 1426
}

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

1436
static int mov_read_meta(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1437
{
1438 1439
    url_fskip(pb, 4); // version + flags
    atom.size -= 4;
1440 1441 1442
    return mov_read_default(c, pb, atom);
}

1443
static int mov_read_trkn(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1444
{
1445
    char track[16];
1446 1447
    get_be32(pb); // type
    get_be32(pb); // unknown
1448 1449 1450
    snprintf(track, sizeof(track), "%d", get_be32(pb));
    av_metadata_set(&c->fc->metadata, "track", track);
    dprintf(c->fc, "%.4s %s\n", (char*)&atom.type, track);
1451 1452 1453
    return 0;
}

1454
static int mov_read_udta_string(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1455
{
1456 1457
    char str[1024], key2[16], language[4] = {0};
    const char *key = NULL;
1458 1459 1460 1461 1462 1463 1464 1465 1466
    uint16_t str_size;

    if (c->itunes_metadata) {
        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;
1467
            atom.size -= 16;
1468 1469 1470
        } else return 0;
    } else {
        str_size = get_be16(pb); // string length
1471
        ff_mov_lang_to_iso639(get_be16(pb), language);
1472
        atom.size -= 4;
1473
    }
B
Baptiste Coudurier 已提交
1474
    switch (atom.type) {
1475
    case MKTAG(0xa9,'n','a','m'): key = "title";     break;
1476
    case MKTAG(0xa9,'a','u','t'):
1477
    case MKTAG(0xa9,'A','R','T'):
1478 1479
    case MKTAG(0xa9,'w','r','t'): key = "author";    break;
    case MKTAG(0xa9,'c','p','y'): key = "copyright"; break;
1480
    case MKTAG(0xa9,'c','m','t'):
1481 1482
    case MKTAG(0xa9,'i','n','f'): key = "comment";   break;
    case MKTAG(0xa9,'a','l','b'): key = "album";     break;
1483 1484 1485 1486
    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 已提交
1487
    }
1488
    if (!key)
B
Baptiste Coudurier 已提交
1489
        return 0;
1490 1491 1492
    if (atom.size < 0)
        return -1;

1493 1494 1495 1496 1497 1498 1499 1500
    str_size = FFMIN3(sizeof(str)-1, str_size, atom.size);
    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);
    }
1501
    dprintf(c->fc, "%.4s %s %d %lld\n", (char*)&atom.type, str, str_size, atom.size);
1502 1503 1504
    return 0;
}

1505
static int mov_read_tkhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1506
{
1507 1508 1509 1510 1511
    int i;
    int width;
    int height;
    int64_t disp_transform[2];
    int display_matrix[3][2];
B
clean  
Baptiste Coudurier 已提交
1512
    AVStream *st = c->fc->streams[c->fc->nb_streams-1];
1513
    MOVStreamContext *sc = st->priv_data;
B
clean  
Baptiste Coudurier 已提交
1514
    int version = get_byte(pb);
1515

1516
    get_be24(pb); /* flags */
1517 1518 1519 1520 1521 1522 1523
    /*
    MOV_TRACK_ENABLED 0x0001
    MOV_TRACK_IN_MOVIE 0x0002
    MOV_TRACK_IN_PREVIEW 0x0004
    MOV_TRACK_IN_POSTER 0x0008
    */

B
Baptiste Coudurier 已提交
1524 1525 1526 1527 1528 1529 1530
    if (version == 1) {
        get_be64(pb);
        get_be64(pb);
    } else {
        get_be32(pb); /* creation time */
        get_be32(pb); /* modification time */
    }
1531 1532
    st->id = (int)get_be32(pb); /* track id (NOT 0 !)*/
    get_be32(pb); /* reserved */
1533

1534 1535
    /* highlevel (considering edits) duration in movie timebase */
    (version == 1) ? get_be64(pb) : get_be32(pb);
1536 1537 1538 1539 1540 1541 1542 1543
    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 */

1544 1545 1546 1547 1548 1549 1550 1551
    //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)
    }
1552

1553 1554
    width = get_be32(pb);       // 16.16 fixed point track width
    height = get_be32(pb);      // 16.16 fixed point track height
1555 1556
    sc->width = width >> 16;
    sc->height = height >> 16;
1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571

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

        //sample aspect ratio is new width/height divided by old width/height
1572
        st->sample_aspect_ratio = av_d2q(
1573 1574 1575
            ((double) disp_transform[0] * height) /
            ((double) disp_transform[1] * width), INT_MAX);
    }
1576 1577 1578
    return 0;
}

1579
static int mov_read_tfhd(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
1580 1581 1582 1583 1584 1585 1586 1587 1588
{
    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);
1589
    if (!track_id)
B
Baptiste Coudurier 已提交
1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613
        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;
}

1614
static int mov_read_trex(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
1615 1616 1617 1618 1619
{
    MOVTrackExt *trex;

    if ((uint64_t)c->trex_count+1 >= UINT_MAX / sizeof(*c->trex_data))
        return -1;
1620 1621
    trex = av_realloc(c->trex_data, (c->trex_count+1)*sizeof(*c->trex_data));
    if (!trex)
B
Baptiste Coudurier 已提交
1622
        return AVERROR(ENOMEM);
1623
    c->trex_data = trex;
B
Baptiste Coudurier 已提交
1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634
    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;
}

1635
static int mov_read_trun(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
1636 1637
{
    MOVFragment *frag = &c->fragment;
1638
    AVStream *st = NULL;
1639
    MOVStreamContext *sc;
B
Baptiste Coudurier 已提交
1640 1641 1642 1643 1644 1645
    uint64_t offset;
    int64_t dts;
    int data_offset = 0;
    unsigned entries, first_sample_flags = frag->flags;
    int flags, distance, i;

1646 1647 1648 1649 1650 1651 1652 1653
    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);
1654
        return -1;
1655
    }
1656
    sc = st->priv_data;
B
Baptiste Coudurier 已提交
1657 1658 1659 1660 1661 1662 1663 1664 1665
    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) {
1666
        MOVStts *ctts_data;
B
Baptiste Coudurier 已提交
1667 1668
        if ((uint64_t)entries+sc->ctts_count >= UINT_MAX/sizeof(*sc->ctts_data))
            return -1;
1669 1670 1671
        ctts_data = av_realloc(sc->ctts_data,
                               (entries+sc->ctts_count)*sizeof(*sc->ctts_data));
        if (!ctts_data)
B
Baptiste Coudurier 已提交
1672
            return AVERROR(ENOMEM);
1673
        sc->ctts_data = ctts_data;
B
Baptiste Coudurier 已提交
1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711
    }
    dts = st->duration;
    offset = frag->base_data_offset + data_offset;
    distance = 0;
    dprintf(c->fc, "first sample flags 0x%x\n", first_sample_flags);
    for (i = 0; i < entries; i++) {
        unsigned sample_size = frag->size;
        int sample_flags = i ? frag->flags : first_sample_flags;
        unsigned sample_duration = frag->duration;
        int keyframe;

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

1712 1713 1714
/* 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/ */
1715
static int mov_read_wide(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727
{
    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;
1728
    if (atom.type != MKTAG('m','d','a','t')) {
1729 1730 1731 1732 1733 1734 1735
        url_fskip(pb, atom.size);
        return 0;
    }
    err = mov_read_mdat(c, pb, atom);
    return err;
}

1736
static int mov_read_cmov(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
1737
{
1738
#if CONFIG_ZLIB
1739
    ByteIOContext ctx;
1740 1741
    uint8_t *cmov_data;
    uint8_t *moov_data; /* uncompressed data */
1742
    long cmov_len, moov_len;
1743
    int ret = -1;
1744

1745
    get_be32(pb); /* dcom atom */
1746
    if (get_le32(pb) != MKTAG('d','c','o','m'))
1747
        return -1;
1748
    if (get_le32(pb) != MKTAG('z','l','i','b')) {
B
Benoit Fouet 已提交
1749
        av_log(c->fc, AV_LOG_ERROR, "unknown compression for cmov atom !");
1750 1751 1752
        return -1;
    }
    get_be32(pb); /* cmvd atom */
1753
    if (get_le32(pb) != MKTAG('c','m','v','d'))
1754 1755
        return -1;
    moov_len = get_be32(pb); /* uncompressed size */
1756
    cmov_len = atom.size - 6 * 4;
1757

B
Baptiste Coudurier 已提交
1758
    cmov_data = av_malloc(cmov_len);
1759
    if (!cmov_data)
1760
        return AVERROR(ENOMEM);
B
Baptiste Coudurier 已提交
1761
    moov_data = av_malloc(moov_len);
1762 1763
    if (!moov_data) {
        av_free(cmov_data);
1764
        return AVERROR(ENOMEM);
1765 1766
    }
    get_buffer(pb, cmov_data, cmov_len);
1767
    if(uncompress (moov_data, (uLongf *) &moov_len, (const Bytef *)cmov_data, cmov_len) != Z_OK)
1768
        goto free_and_return;
1769
    if(init_put_byte(&ctx, moov_data, moov_len, 0, NULL, NULL, NULL, NULL) != 0)
1770
        goto free_and_return;
1771
    atom.type = MKTAG('m','o','o','v');
1772 1773
    atom.offset = 0;
    atom.size = moov_len;
1774
#ifdef DEBUG
M
Michael Niedermayer 已提交
1775
//    { int fd = open("/tmp/uncompheader.mov", O_WRONLY | O_CREAT); write(fd, moov_data, moov_len); close(fd); }
1776
#endif
1777
    ret = mov_read_default(c, &ctx, atom);
1778
free_and_return:
1779 1780 1781
    av_free(moov_data);
    av_free(cmov_data);
    return ret;
1782 1783 1784
#else
    av_log(c->fc, AV_LOG_ERROR, "this file requires zlib support compiled in\n");
    return -1;
1785
#endif
1786
}
1787

G
Gael Chardon 已提交
1788
/* edit list atom */
1789
static int mov_read_elst(MOVContext *c, ByteIOContext *pb, MOVAtom atom)
G
Gael Chardon 已提交
1790
{
1791
    MOVStreamContext *sc = c->fc->streams[c->fc->nb_streams-1]->priv_data;
B
Baptiste Coudurier 已提交
1792 1793 1794
    int i, edit_count;

    get_byte(pb); /* version */
1795
    get_be24(pb); /* flags */
B
Baptiste Coudurier 已提交
1796
    edit_count = get_be32(pb); /* entries */
B
Baptiste Coudurier 已提交
1797 1798

    for(i=0; i<edit_count; i++){
1799
        int time;
B
Baptiste Coudurier 已提交
1800
        get_be32(pb); /* Track duration */
1801
        time = get_be32(pb); /* Media time */
B
Baptiste Coudurier 已提交
1802
        get_be32(pb); /* Media rate */
1803
        if (i == 0 && time != -1) {
1804
            sc->time_offset = time;
1805 1806
            sc->time_rate = av_gcd(sc->time_rate, time);
        }
B
Baptiste Coudurier 已提交
1807
    }
1808 1809 1810 1811 1812

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

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

1817
static const MOVParseTableEntry mov_default_parse_table[] = {
1818
{ MKTAG('a','v','s','s'), mov_read_extradata },
1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829
{ MKTAG('c','o','6','4'), mov_read_stco },
{ MKTAG('c','t','t','s'), mov_read_ctts }, /* composition time to sample */
{ MKTAG('d','i','n','f'), mov_read_default },
{ MKTAG('d','r','e','f'), mov_read_dref },
{ MKTAG('e','d','t','s'), mov_read_default },
{ MKTAG('e','l','s','t'), mov_read_elst },
{ MKTAG('e','n','d','a'), mov_read_enda },
{ MKTAG('f','i','e','l'), mov_read_extradata },
{ MKTAG('f','t','y','p'), mov_read_ftyp },
{ MKTAG('g','l','b','l'), mov_read_glbl },
{ MKTAG('h','d','l','r'), mov_read_hdlr },
1830
{ MKTAG('i','l','s','t'), mov_read_ilst },
1831 1832 1833 1834
{ 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 },
1835
{ MKTAG('m','e','t','a'), mov_read_meta },
1836 1837 1838 1839 1840 1841 1842 1843
{ 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 },
1844
{ MKTAG('p','a','s','p'), mov_read_pasp },
1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856
{ MKTAG('s','t','b','l'), mov_read_default },
{ MKTAG('s','t','c','o'), mov_read_stco },
{ MKTAG('s','t','s','c'), mov_read_stsc },
{ MKTAG('s','t','s','d'), mov_read_stsd }, /* sample description */
{ MKTAG('s','t','s','s'), mov_read_stss }, /* sync sample */
{ MKTAG('s','t','s','z'), mov_read_stsz }, /* sample size */
{ MKTAG('s','t','t','s'), mov_read_stts },
{ MKTAG('t','k','h','d'), mov_read_tkhd }, /* track header */
{ MKTAG('t','f','h','d'), mov_read_tfhd }, /* track fragment header */
{ MKTAG('t','r','a','k'), mov_read_trak },
{ MKTAG('t','r','a','f'), mov_read_default },
{ MKTAG('t','r','e','x'), mov_read_trex },
1857
{ MKTAG('t','r','k','n'), mov_read_trkn },
1858
{ MKTAG('t','r','u','n'), mov_read_trun },
B
Baptiste Coudurier 已提交
1859
{ MKTAG('u','d','t','a'), mov_read_default },
1860 1861 1862 1863
{ 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 已提交
1864 1865 1866 1867
{ MKTAG(0xa9,'n','a','m'), mov_read_udta_string },
{ MKTAG(0xa9,'w','r','t'), mov_read_udta_string },
{ MKTAG(0xa9,'c','p','y'), mov_read_udta_string },
{ MKTAG(0xa9,'i','n','f'), mov_read_udta_string },
1868 1869 1870 1871
{ MKTAG(0xa9,'i','n','f'), mov_read_udta_string },
{ MKTAG(0xa9,'A','R','T'), mov_read_udta_string },
{ MKTAG(0xa9,'a','l','b'), mov_read_udta_string },
{ MKTAG(0xa9,'c','m','t'), mov_read_udta_string },
1872 1873 1874 1875 1876
{ MKTAG(0xa9,'a','u','t'), mov_read_udta_string },
{ MKTAG(0xa9,'d','a','y'), mov_read_udta_string },
{ MKTAG(0xa9,'g','e','n'), mov_read_udta_string },
{ MKTAG(0xa9,'e','n','c'), mov_read_udta_string },
{ MKTAG(0xa9,'t','o','o'), mov_read_udta_string },
B
Baptiste Coudurier 已提交
1877
{ 0, NULL }
1878 1879
};

F
Fabrice Bellard 已提交
1880 1881
static int mov_probe(AVProbeData *p)
{
1882 1883
    unsigned int offset;
    uint32_t tag;
1884
    int score = 0;
1885

F
Fabrice Bellard 已提交
1886
    /* check file header */
1887 1888 1889 1890
    offset = 0;
    for(;;) {
        /* ignore invalid offset */
        if ((offset + 8) > (unsigned int)p->buf_size)
1891
            return score;
1892
        tag = AV_RL32(p->buf + offset + 4);
1893
        switch(tag) {
1894
        /* check for obvious tags */
1895 1896 1897 1898 1899
        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 */
1900
        case MKTAG('f','t','y','p'):
1901
            return AVPROBE_SCORE_MAX;
1902
        /* those are more common words, so rate then a bit less */
1903 1904 1905 1906 1907
        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'):
1908
            return AVPROBE_SCORE_MAX - 5;
B
Baptiste Coudurier 已提交
1909
        case MKTAG(0x82,0x82,0x7f,0x7d):
1910 1911 1912
        case MKTAG('s','k','i','p'):
        case MKTAG('u','u','i','d'):
        case MKTAG('p','r','f','l'):
1913
            offset = AV_RB32(p->buf+offset) + offset;
1914 1915
            /* if we only find those cause probedata is too small at least rate them */
            score = AVPROBE_SCORE_MAX - 50;
1916 1917 1918
            break;
        default:
            /* unrecognized tag */
1919
            return score;
1920
        }
1921
    }
1922
    return score;
F
Fabrice Bellard 已提交
1923 1924
}

Z
Zdenek Kabelac 已提交
1925
static int mov_read_header(AVFormatContext *s, AVFormatParameters *ap)
1926
{
1927
    MOVContext *mov = s->priv_data;
1928
    ByteIOContext *pb = s->pb;
B
Baptiste Coudurier 已提交
1929
    int err;
1930
    MOVAtom atom = { 0, 0, 0 };
1931 1932

    mov->fc = s;
1933 1934
    /* .mov and .mp4 aren't streamable anyway (only progressive download if moov is before mdat) */
    if(!url_is_streamed(pb))
1935
        atom.size = url_fsize(pb);
1936
    else
B
Baptiste Coudurier 已提交
1937
        atom.size = INT64_MAX;
1938 1939

    /* check MOV header */
B
Baptiste Coudurier 已提交
1940 1941 1942 1943 1944 1945
    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");
1946
        return -1;
1947
    }
B
Baptiste Coudurier 已提交
1948
    dprintf(mov->fc, "on_parse_exit_offset=%lld\n", url_ftell(pb));
1949

1950 1951 1952
    return 0;
}

Z
Zdenek Kabelac 已提交
1953
static int mov_read_packet(AVFormatContext *s, AVPacket *pkt)
1954
{
1955 1956 1957
    MOVContext *mov = s->priv_data;
    MOVStreamContext *sc = 0;
    AVIndexEntry *sample = 0;
1958
    int64_t best_dts = INT64_MAX;
1959
    int i, ret;
1960
 retry:
B
Baptiste Coudurier 已提交
1961
    for (i = 0; i < s->nb_streams; i++) {
1962 1963
        AVStream *st = s->streams[i];
        MOVStreamContext *msc = st->priv_data;
1964
        if (st->discard != AVDISCARD_ALL && msc->pb && msc->current_sample < msc->sample_count) {
1965
            AVIndexEntry *current_sample = &st->index_entries[msc->current_sample];
1966 1967
            int64_t dts = av_rescale(current_sample->timestamp * (int64_t)msc->time_rate,
                                     AV_TIME_BASE, msc->time_scale);
B
Baptiste Coudurier 已提交
1968
            dprintf(s, "stream %d, sample %d, dts %"PRId64"\n", i, msc->current_sample, dts);
B
Baptiste Coudurier 已提交
1969 1970
            if (!sample || (url_is_streamed(s->pb) && current_sample->pos < sample->pos) ||
                (!url_is_streamed(s->pb) &&
1971
                 ((msc->pb != s->pb && dts < best_dts) || (msc->pb == s->pb &&
B
Baptiste Coudurier 已提交
1972
                 ((FFABS(best_dts - dts) <= AV_TIME_BASE && current_sample->pos < sample->pos) ||
1973
                  (FFABS(best_dts - dts) > AV_TIME_BASE && dts < best_dts)))))) {
1974 1975 1976
                sample = current_sample;
                best_dts = dts;
                sc = msc;
1977
            }
1978 1979
        }
    }
1980 1981 1982
    if (!sample) {
        mov->found_mdat = 0;
        if (!url_is_streamed(s->pb) ||
1983
            mov_read_default(mov, s->pb, (MOVAtom){ 0, 0, INT64_MAX }) < 0 ||
1984 1985 1986 1987 1988
            url_feof(s->pb))
            return -1;
        dprintf(s, "read fragments, offset 0x%llx\n", url_ftell(s->pb));
        goto retry;
    }
1989 1990
    /* must be done just before reading, to avoid infinite loop on sample */
    sc->current_sample++;
1991
    if (url_fseek(sc->pb, sample->pos, SEEK_SET) != sample->pos) {
1992 1993
        av_log(mov->fc, AV_LOG_ERROR, "stream %d, offset 0x%"PRIx64": partial file\n",
               sc->ffindex, sample->pos);
1994
        return -1;
1995
    }
1996 1997 1998
    ret = av_get_packet(sc->pb, pkt, sample->size);
    if (ret < 0)
        return ret;
1999
#if CONFIG_DV_DEMUXER
B
Baptiste Coudurier 已提交
2000 2001 2002
    if (mov->dv_demux && sc->dv_audio_container) {
        dv_produce_packet(mov->dv_demux, pkt, pkt->data, pkt->size);
        av_free(pkt->data);
2003 2004 2005
        pkt->size = 0;
        if (dv_get_packet(mov->dv_demux, pkt) < 0)
            return -1;
B
Baptiste Coudurier 已提交
2006
    }
2007
#endif
2008 2009 2010
    pkt->stream_index = sc->ffindex;
    pkt->dts = sample->timestamp;
    if (sc->ctts_data) {
2011 2012
        assert(sc->ctts_data[sc->ctts_index].duration % sc->time_rate == 0);
        pkt->pts = pkt->dts + sc->ctts_data[sc->ctts_index].duration / sc->time_rate;
2013
        /* update ctts context */
2014 2015 2016 2017 2018
        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;
2019
        }
2020 2021
        if (sc->wrong_dts)
            pkt->dts = AV_NOPTS_VALUE;
2022
    } else {
2023 2024 2025 2026
        AVStream *st = s->streams[sc->ffindex];
        int64_t next_dts = (sc->current_sample < sc->sample_count) ?
            st->index_entries[sc->current_sample].timestamp : st->duration;
        pkt->duration = next_dts - pkt->dts;
2027
        pkt->pts = pkt->dts;
2028
    }
2029 2030
    pkt->flags |= sample->flags & AVINDEX_KEYFRAME ? PKT_FLAG_KEY : 0;
    pkt->pos = sample->pos;
2031 2032
    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);
2033 2034
    return 0;
}
2035

2036 2037 2038 2039 2040
static int mov_seek_stream(AVStream *st, int64_t timestamp, int flags)
{
    MOVStreamContext *sc = st->priv_data;
    int sample, time_sample;
    int i;
2041

2042
    sample = av_index_search_timestamp(st, timestamp, flags);
M
Michel Bardiaux 已提交
2043
    dprintf(st->codec, "stream %d, timestamp %"PRId64", sample %d\n", st->index, timestamp, sample);
2044 2045 2046
    if (sample < 0) /* not sure what to do */
        return -1;
    sc->current_sample = sample;
B
Baptiste Coudurier 已提交
2047
    dprintf(st->codec, "stream %d, found sample %d\n", st->index, sc->current_sample);
2048 2049 2050 2051
    /* adjust ctts index */
    if (sc->ctts_data) {
        time_sample = 0;
        for (i = 0; i < sc->ctts_count; i++) {
2052 2053
            int next = time_sample + sc->ctts_data[i].count;
            if (next > sc->current_sample) {
2054 2055
                sc->ctts_index = i;
                sc->ctts_sample = sc->current_sample - time_sample;
2056
                break;
2057
            }
2058
            time_sample = next;
2059 2060
        }
    }
2061
    return sample;
2062 2063
}

2064
static int mov_read_seek(AVFormatContext *s, int stream_index, int64_t sample_time, int flags)
G
Gael Chardon 已提交
2065
{
2066 2067 2068 2069
    AVStream *st;
    int64_t seek_timestamp, timestamp;
    int sample;
    int i;
G
Gael Chardon 已提交
2070

2071
    if (stream_index >= s->nb_streams)
G
Gael Chardon 已提交
2072
        return -1;
2073 2074
    if (sample_time < 0)
        sample_time = 0;
G
Gael Chardon 已提交
2075

2076 2077 2078
    st = s->streams[stream_index];
    sample = mov_seek_stream(st, sample_time, flags);
    if (sample < 0)
G
Gael Chardon 已提交
2079 2080
        return -1;

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

2084 2085 2086 2087
    for (i = 0; i < s->nb_streams; i++) {
        st = s->streams[i];
        if (stream_index == i || st->discard == AVDISCARD_ALL)
            continue;
G
Gael Chardon 已提交
2088

2089 2090
        timestamp = av_rescale_q(seek_timestamp, s->streams[stream_index]->time_base, st->time_base);
        mov_seek_stream(st, timestamp, flags);
2091
    }
G
Gael Chardon 已提交
2092 2093 2094
    return 0;
}

Z
Zdenek Kabelac 已提交
2095
static int mov_read_close(AVFormatContext *s)
2096
{
2097
    int i, j;
2098
    MOVContext *mov = s->priv_data;
B
Baptiste Coudurier 已提交
2099
    for(i=0; i<s->nb_streams; i++) {
2100 2101
        MOVStreamContext *sc = s->streams[i]->priv_data;
        av_freep(&sc->ctts_data);
2102 2103 2104 2105 2106
        for (j=0; j<sc->drefs_count; j++)
            av_freep(&sc->drefs[j].path);
        av_freep(&sc->drefs);
        if (sc->pb && sc->pb != s->pb)
            url_fclose(sc->pb);
2107
    }
2108 2109 2110 2111 2112 2113 2114 2115
    if(mov->dv_demux){
        for(i=0; i<mov->dv_fctx->nb_streams; i++){
            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);
    }
B
Baptiste Coudurier 已提交
2116
    av_freep(&mov->trex_data);
2117 2118 2119
    return 0;
}

2120
AVInputFormat mov_demuxer = {
2121
    "mov,mp4,m4a,3gp,3g2,mj2",
2122
    NULL_IF_CONFIG_SMALL("QuickTime/MPEG-4/Motion JPEG 2000 format"),
F
Fabrice Bellard 已提交
2123 2124
    sizeof(MOVContext),
    mov_probe,
2125 2126 2127
    mov_read_header,
    mov_read_packet,
    mov_read_close,
G
Gael Chardon 已提交
2128
    mov_read_seek,
2129
};