mov.c 113.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
 * first version by Francois Revol <revol@free.fr>
 * seek function by Gael Chardon <gael.dev@4now.net>
 *
9
 * This file is part of Libav.
10
 *
11
 * Libav is free software; you can redistribute it and/or
F
Fabrice Bellard 已提交
12 13
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
14
 * version 2.1 of the License, or (at your option) any later version.
15
 *
16
 * Libav is distributed in the hope that it will be useful,
17
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
F
Fabrice Bellard 已提交
18 19
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
20
 *
F
Fabrice Bellard 已提交
21
 * You should have received a copy of the GNU Lesser General Public
22
 * License along with Libav; if not, write to the Free Software
23
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
24
 */
25

26
#include <inttypes.h>
27
#include <limits.h>
28
#include <stdint.h>
29

30
#include "libavutil/attributes.h"
31
#include "libavutil/channel_layout.h"
32
#include "libavutil/intreadwrite.h"
33
#include "libavutil/intfloat.h"
34
#include "libavutil/mathematics.h"
35
#include "libavutil/time_internal.h"
36
#include "libavutil/avstring.h"
37
#include "libavutil/dict.h"
38
#include "libavutil/opt.h"
39
#include "libavcodec/ac3tab.h"
40
#include "avformat.h"
41
#include "internal.h"
42
#include "avio_internal.h"
43
#include "riff.h"
44
#include "isom.h"
45
#include "libavcodec/get_bits.h"
R
Raivo Hool 已提交
46
#include "id3v1.h"
47
#include "mov_chan.h"
48
#include "replaygain.h"
49

50
#if CONFIG_ZLIB
51 52 53
#include <zlib.h>
#endif

54 55
#include "qtpalette.h"

G
Gael Chardon 已提交
56

57 58 59
#undef NDEBUG
#include <assert.h>

60 61 62
/* those functions parse an atom */
/* links atom IDs to parse functions */
typedef struct MOVParseTableEntry {
63
    uint32_t type;
64
    int (*parse)(MOVContext *ctx, AVIOContext *pb, MOVAtom atom);
65 66
} MOVParseTableEntry;

67
static int mov_read_default(MOVContext *c, AVIOContext *pb, MOVAtom atom);
68

69 70
static int mov_metadata_track_or_disc_number(MOVContext *c, AVIOContext *pb,
                                             unsigned len, const char *key)
B
Baptiste Coudurier 已提交
71 72 73
{
    char buf[16];

74
    short current, total = 0;
75
    avio_rb16(pb); // unknown
76
    current = avio_rb16(pb);
77 78
    if (len >= 6)
        total = avio_rb16(pb);
79 80 81 82
    if (!total)
        snprintf(buf, sizeof(buf), "%d", current);
    else
        snprintf(buf, sizeof(buf), "%d/%d", current, total);
83
    c->fc->event_flags |= AVFMT_EVENT_FLAG_METADATA_UPDATED;
84
    av_dict_set(&c->fc->metadata, key, buf, 0);
B
Baptiste Coudurier 已提交
85 86 87 88

    return 0;
}

89 90
static int mov_metadata_int8_bypass_padding(MOVContext *c, AVIOContext *pb,
                                            unsigned len, const char *key)
91
{
92
    char buf[16];
93

94 95 96 97
    /* bypass padding bytes */
    avio_r8(pb);
    avio_r8(pb);
    avio_r8(pb);
98

99
    snprintf(buf, sizeof(buf), "%d", avio_r8(pb));
100
    c->fc->event_flags |= AVFMT_EVENT_FLAG_METADATA_UPDATED;
101
    av_dict_set(&c->fc->metadata, key, buf, 0);
102

103
    return 0;
104 105
}

106 107
static int mov_metadata_int8_no_padding(MOVContext *c, AVIOContext *pb,
                                        unsigned len, const char *key)
108
{
109
    char buf[16];
110

111
    snprintf(buf, sizeof(buf), "%d", avio_r8(pb));
112
    c->fc->event_flags |= AVFMT_EVENT_FLAG_METADATA_UPDATED;
113
    av_dict_set(&c->fc->metadata, key, buf, 0);
114

115
    return 0;
116 117
}

R
Raivo Hool 已提交
118 119 120 121 122 123 124 125 126 127 128 129
static int mov_metadata_gnre(MOVContext *c, AVIOContext *pb,
                             unsigned len, const char *key)
{
    short genre;
    char buf[20];

    avio_r8(pb); // unknown

    genre = avio_r8(pb);
    if (genre < 1 || genre > ID3v1_GENRE_MAX)
        return 0;
    snprintf(buf, sizeof(buf), "%s", ff_id3v1_genre_str[genre-1]);
130
    c->fc->event_flags |= AVFMT_EVENT_FLAG_METADATA_UPDATED;
R
Raivo Hool 已提交
131 132 133 134 135
    av_dict_set(&c->fc->metadata, key, buf, 0);

    return 0;
}

136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154
static const uint32_t mac_to_unicode[128] = {
    0x00C4,0x00C5,0x00C7,0x00C9,0x00D1,0x00D6,0x00DC,0x00E1,
    0x00E0,0x00E2,0x00E4,0x00E3,0x00E5,0x00E7,0x00E9,0x00E8,
    0x00EA,0x00EB,0x00ED,0x00EC,0x00EE,0x00EF,0x00F1,0x00F3,
    0x00F2,0x00F4,0x00F6,0x00F5,0x00FA,0x00F9,0x00FB,0x00FC,
    0x2020,0x00B0,0x00A2,0x00A3,0x00A7,0x2022,0x00B6,0x00DF,
    0x00AE,0x00A9,0x2122,0x00B4,0x00A8,0x2260,0x00C6,0x00D8,
    0x221E,0x00B1,0x2264,0x2265,0x00A5,0x00B5,0x2202,0x2211,
    0x220F,0x03C0,0x222B,0x00AA,0x00BA,0x03A9,0x00E6,0x00F8,
    0x00BF,0x00A1,0x00AC,0x221A,0x0192,0x2248,0x2206,0x00AB,
    0x00BB,0x2026,0x00A0,0x00C0,0x00C3,0x00D5,0x0152,0x0153,
    0x2013,0x2014,0x201C,0x201D,0x2018,0x2019,0x00F7,0x25CA,
    0x00FF,0x0178,0x2044,0x20AC,0x2039,0x203A,0xFB01,0xFB02,
    0x2021,0x00B7,0x201A,0x201E,0x2030,0x00C2,0x00CA,0x00C1,
    0x00CB,0x00C8,0x00CD,0x00CE,0x00CF,0x00CC,0x00D3,0x00D4,
    0xF8FF,0x00D2,0x00DA,0x00DB,0x00D9,0x0131,0x02C6,0x02DC,
    0x00AF,0x02D8,0x02D9,0x02DA,0x00B8,0x02DD,0x02DB,0x02C7,
};

155
static int mov_read_mac_string(MOVContext *c, AVIOContext *pb, int len,
156 157 158 159 160 161 162
                               char *dst, int dstlen)
{
    char *p = dst;
    char *end = dst+dstlen-1;
    int i;

    for (i = 0; i < len; i++) {
163
        uint8_t t, c = avio_r8(pb);
164 165 166 167 168 169 170 171 172
        if (c < 0x80 && p < end)
            *p++ = c;
        else
            PUT_UTF8(mac_to_unicode[c-0x80], t, if (p < end) *p++ = t;);
    }
    *p = 0;
    return p - dst;
}

A
Anton Khirnov 已提交
173 174 175 176 177
static int mov_read_covr(MOVContext *c, AVIOContext *pb, int type, int len)
{
    AVPacket pkt;
    AVStream *st;
    MOVStreamContext *sc;
178
    enum AVCodecID id;
A
Anton Khirnov 已提交
179 180 181
    int ret;

    switch (type) {
182 183 184
    case 0xd:  id = AV_CODEC_ID_MJPEG; break;
    case 0xe:  id = AV_CODEC_ID_PNG;   break;
    case 0x1b: id = AV_CODEC_ID_BMP;   break;
A
Anton Khirnov 已提交
185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214
    default:
        av_log(c->fc, AV_LOG_WARNING, "Unknown cover type: 0x%x.\n", type);
        avio_skip(pb, len);
        return 0;
    }

    st = avformat_new_stream(c->fc, NULL);
    if (!st)
        return AVERROR(ENOMEM);
    sc = av_mallocz(sizeof(*sc));
    if (!sc)
        return AVERROR(ENOMEM);
    st->priv_data = sc;

    ret = av_get_packet(pb, &pkt, len);
    if (ret < 0)
        return ret;

    st->disposition              |= AV_DISPOSITION_ATTACHED_PIC;

    st->attached_pic              = pkt;
    st->attached_pic.stream_index = st->index;
    st->attached_pic.flags       |= AV_PKT_FLAG_KEY;

    st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
    st->codec->codec_id   = id;

    return 0;
}

215 216 217 218 219
static int mov_metadata_loci(MOVContext *c, AVIOContext *pb, unsigned len)
{
    char language[4] = { 0 };
    char buf[100];
    uint16_t langcode = 0;
220
    double longitude, latitude;
221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248
    const char *key = "location";

    if (len < 4 + 2 + 1 + 1 + 4 + 4 + 4)
        return AVERROR_INVALIDDATA;

    avio_skip(pb, 4); // version+flags
    langcode = avio_rb16(pb);
    ff_mov_lang_to_iso639(langcode, language);
    len -= 6;

    len -= avio_get_str(pb, len, buf, sizeof(buf)); // place name
    if (len < 1)
        return AVERROR_INVALIDDATA;
    avio_skip(pb, 1); // role
    len -= 1;

    if (len < 14)
        return AVERROR_INVALIDDATA;
    longitude = ((int32_t) avio_rb32(pb)) / (float) (1 << 16);
    latitude  = ((int32_t) avio_rb32(pb)) / (float) (1 << 16);

    // Try to output in the same format as the ?xyz field
    snprintf(buf, sizeof(buf), "%+08.4f%+09.4f/", latitude, longitude);
    if (*language && strcmp(language, "und")) {
        char key2[16];
        snprintf(key2, sizeof(key2), "%s-%s", key, language);
        av_dict_set(&c->fc->metadata, key2, buf, 0);
    }
249
    c->fc->event_flags |= AVFMT_EVENT_FLAG_METADATA_UPDATED;
250 251 252
    return av_dict_set(&c->fc->metadata, key, buf, 0);
}

253
static int mov_read_udta_string(MOVContext *c, AVIOContext *pb, MOVAtom atom)
254 255
{
    char tmp_key[5];
256
    char *str, key2[32], language[4] = {0};
257
    const char *key = NULL;
258
    uint16_t langcode = 0;
259
    uint32_t data_type = 0, str_size, str_size_alloc;
260
    int (*parse)(MOVContext*, AVIOContext*, unsigned, const char*) = NULL;
261
    int raw = 0;
262 263

    switch (atom.type) {
264 265
    case MKTAG( '@','P','R','M'): key = "premiere_version"; raw = 1; break;
    case MKTAG( '@','P','R','Q'): key = "quicktime_version"; raw = 1; break;
266 267
    case MKTAG( 'X','M','P','_'):
        if (c->export_xmp) { key = "xmp"; raw = 1; } break;
R
Raivo Hool 已提交
268
    case MKTAG( 'a','A','R','T'): key = "album_artist";    break;
269 270 271 272 273 274
    case MKTAG( 'a','k','I','D'): key = "account_type";
        parse = mov_metadata_int8_no_padding; break;
    case MKTAG( 'a','p','I','D'): key = "account_id"; break;
    case MKTAG( 'c','a','t','g'): key = "category"; break;
    case MKTAG( 'c','p','i','l'): key = "compilation";
        parse = mov_metadata_int8_no_padding; break;
275 276 277 278
    case MKTAG( 'c','p','r','t'): key = "copyright"; break;
    case MKTAG( 'd','e','s','c'): key = "description"; break;
    case MKTAG( 'd','i','s','k'): key = "disc";
        parse = mov_metadata_track_or_disc_number; break;
279 280
    case MKTAG( 'e','g','i','d'): key = "episode_uid";
        parse = mov_metadata_int8_no_padding; break;
R
Raivo Hool 已提交
281 282
    case MKTAG( 'g','n','r','e'): key = "genre";
        parse = mov_metadata_gnre; break;
283 284
    case MKTAG( 'h','d','v','d'): key = "hd_video";
        parse = mov_metadata_int8_no_padding; break;
285
    case MKTAG( 'k','e','y','w'): key = "keywords";  break;
286
    case MKTAG( 'l','d','e','s'): key = "synopsis";  break;
287 288
    case MKTAG( 'l','o','c','i'):
        return mov_metadata_loci(c, pb, atom.size);
289 290
    case MKTAG( 'p','c','s','t'): key = "podcast";
        parse = mov_metadata_int8_no_padding; break;
291 292
    case MKTAG( 'p','g','a','p'): key = "gapless_playback";
        parse = mov_metadata_int8_no_padding; break;
293 294 295 296 297 298 299 300 301
    case MKTAG( 'p','u','r','d'): key = "purchase_date"; break;
    case MKTAG( 'r','t','n','g'): key = "rating";
        parse = mov_metadata_int8_no_padding; break;
    case MKTAG( 's','o','a','a'): key = "sort_album_artist"; break;
    case MKTAG( 's','o','a','l'): key = "sort_album";   break;
    case MKTAG( 's','o','a','r'): key = "sort_artist";  break;
    case MKTAG( 's','o','c','o'): key = "sort_composer"; break;
    case MKTAG( 's','o','n','m'): key = "sort_name";    break;
    case MKTAG( 's','o','s','n'): key = "sort_show";    break;
302 303
    case MKTAG( 's','t','i','k'): key = "media_type";
        parse = mov_metadata_int8_no_padding; break;
B
Baptiste Coudurier 已提交
304
    case MKTAG( 't','r','k','n'): key = "track";
305
        parse = mov_metadata_track_or_disc_number; break;
306
    case MKTAG( 't','v','e','n'): key = "episode_id"; break;
307
    case MKTAG( 't','v','e','s'): key = "episode_sort";
308
        parse = mov_metadata_int8_bypass_padding; break;
309 310
    case MKTAG( 't','v','n','n'): key = "network";   break;
    case MKTAG( 't','v','s','h'): key = "show";      break;
311
    case MKTAG( 't','v','s','n'): key = "season_number";
312
        parse = mov_metadata_int8_bypass_padding; break;
313 314 315 316 317 318 319 320
    case MKTAG(0xa9,'A','R','T'): key = "artist";    break;
    case MKTAG(0xa9,'a','l','b'): key = "album";     break;
    case MKTAG(0xa9,'a','u','t'): key = "artist";    break;
    case MKTAG(0xa9,'c','m','t'): key = "comment";   break;
    case MKTAG(0xa9,'c','p','y'): key = "copyright"; break;
    case MKTAG(0xa9,'d','a','y'): key = "date";      break;
    case MKTAG(0xa9,'e','n','c'): key = "encoder";   break;
    case MKTAG(0xa9,'g','e','n'): key = "genre";     break;
321
    case MKTAG(0xa9,'g','r','p'): key = "grouping";  break;
322
    case MKTAG(0xa9,'i','n','f'): key = "comment";   break;
323
    case MKTAG(0xa9,'l','y','r'): key = "lyrics";    break;
324
    case MKTAG(0xa9,'n','a','m'): key = "title";     break;
325
    case MKTAG(0xa9,'s','t','3'): key = "subtitle";  break;
326 327 328 329
    case MKTAG(0xa9,'s','w','r'): key = "encoder";   break;
    case MKTAG(0xa9,'t','o','o'): key = "encoder";   break;
    case MKTAG(0xa9,'w','r','t'): key = "composer";  break;
    case MKTAG(0xa9,'x','y','z'): key = "location";  break;
330 331 332
    }

    if (c->itunes_metadata && atom.size > 8) {
333 334
        int data_size = avio_rb32(pb);
        int tag = avio_rl32(pb);
335
        if (tag == MKTAG('d','a','t','a')) {
336 337
            data_type = avio_rb32(pb); // type
            avio_rb32(pb); // unknown
338 339
            str_size = data_size - 16;
            atom.size -= 16;
A
Anton Khirnov 已提交
340 341 342 343 344 345 346 347

            if (atom.type == MKTAG('c', 'o', 'v', 'r')) {
                int ret = mov_read_covr(c, pb, data_type, str_size);
                if (ret < 0) {
                    av_log(c->fc, AV_LOG_ERROR, "Error parsing cover art.\n");
                    return ret;
                }
            }
348
        } else return 0;
349
    } else if (atom.size > 4 && key && !c->itunes_metadata && !raw) {
350 351
        str_size = avio_rb16(pb); // string length
        langcode = avio_rb16(pb);
352
        ff_mov_lang_to_iso639(langcode, language);
353 354 355 356
        atom.size -= 4;
    } else
        str_size = atom.size;

357
    if (c->export_all && !key) {
358 359 360 361 362 363 364
        snprintf(tmp_key, 5, "%.4s", (char*)&atom.type);
        key = tmp_key;
    }

    if (!key)
        return 0;
    if (atom.size < 0)
365
        return AVERROR_INVALIDDATA;
366

367
    // allocate twice as much as worst-case
368
    str_size_alloc = raw ? str_size + 1 : str_size * 2;
369 370 371
    str = av_malloc(str_size_alloc);
    if (!str)
        return AVERROR(ENOMEM);
B
Baptiste Coudurier 已提交
372 373

    if (parse)
374
        parse(c, pb, str_size, key);
B
Baptiste Coudurier 已提交
375
    else {
376
        if (!raw && (data_type == 3 || (data_type == 0 && (langcode < 0x400 || langcode == 0x7fff)))) { // MAC Encoded
377
            mov_read_mac_string(c, pb, str_size, str, str_size_alloc);
378
        } else {
379
            avio_read(pb, str, str_size);
380 381
            str[str_size] = 0;
        }
382
        c->fc->event_flags |= AVFMT_EVENT_FLAG_METADATA_UPDATED;
383
        av_dict_set(&c->fc->metadata, key, str, 0);
B
Baptiste Coudurier 已提交
384 385
        if (*language && strcmp(language, "und")) {
            snprintf(key2, sizeof(key2), "%s-%s", key, language);
386
            av_dict_set(&c->fc->metadata, key2, str, 0);
B
Baptiste Coudurier 已提交
387
        }
B
Baptiste Coudurier 已提交
388
    }
389 390
    av_dlog(c->fc, "lang \"%3s\" ", language);
    av_dlog(c->fc, "tag \"%s\" value \"%s\" atom \"%.4s\" %d %"PRId64"\n",
391
            key, str, (char*)&atom.type, str_size_alloc, atom.size);
392

393
    av_freep(&str);
394 395
    return 0;
}
396

397
static int mov_read_chpl(MOVContext *c, AVIOContext *pb, MOVAtom atom)
D
David Conrad 已提交
398 399
{
    int64_t start;
D
David Conrad 已提交
400
    int i, nb_chapters, str_len, version;
D
David Conrad 已提交
401 402 403 404 405
    char str[256+1];

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

406 407
    version = avio_r8(pb);
    avio_rb24(pb);
D
David Conrad 已提交
408
    if (version)
409 410
        avio_rb32(pb); // ???
    nb_chapters = avio_r8(pb);
D
David Conrad 已提交
411 412 413 414 415

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

416 417
        start = avio_rb64(pb);
        str_len = avio_r8(pb);
D
David Conrad 已提交
418 419 420 421

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

422
        avio_read(pb, str, str_len);
D
David Conrad 已提交
423
        str[str_len] = 0;
424
        avpriv_new_chapter(c->fc, i, (AVRational){1,10000000}, start, AV_NOPTS_VALUE, str);
D
David Conrad 已提交
425 426 427 428
    }
    return 0;
}

429
#define MIN_DATA_ENTRY_BOX_SIZE 12
430
static int mov_read_dref(MOVContext *c, AVIOContext *pb, MOVAtom atom)
431
{
432 433
    AVStream *st;
    MOVStreamContext *sc;
434 435
    int entries, i, j;

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

441 442
    avio_rb32(pb); // version + flags
    entries = avio_rb32(pb);
443 444
    if (entries >  (atom.size - 1) / MIN_DATA_ENTRY_BOX_SIZE + 1 ||
        entries >= UINT_MAX / sizeof(*sc->drefs))
445
        return AVERROR_INVALIDDATA;
446
    av_free(sc->drefs);
447
    sc->drefs = av_mallocz(entries * sizeof(*sc->drefs));
448 449 450
    if (!sc->drefs)
        return AVERROR(ENOMEM);
    sc->drefs_count = entries;
451 452

    for (i = 0; i < sc->drefs_count; i++) {
453
        MOVDref *dref = &sc->drefs[i];
454
        uint32_t size = avio_rb32(pb);
455
        int64_t next = avio_tell(pb) + size - 4;
456

457
        if (size < 12)
458
            return AVERROR_INVALIDDATA;
459

460 461
        dref->type = avio_rl32(pb);
        avio_rb32(pb); // version + flags
L
Luca Barbato 已提交
462
        av_dlog(c->fc, "type %.4s size %d\n", (char*)&dref->type, size);
463 464 465 466 467 468

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

469
            avio_skip(pb, 10);
470

471
            volume_len = avio_r8(pb);
472
            volume_len = FFMIN(volume_len, 27);
473
            avio_read(pb, dref->volume, 27);
474 475
            dref->volume[volume_len] = 0;
            av_log(c->fc, AV_LOG_DEBUG, "volume %s, len %d\n", dref->volume, volume_len);
476

477
            avio_skip(pb, 12);
478

479
            len = avio_r8(pb);
480
            len = FFMIN(len, 63);
481
            avio_read(pb, dref->filename, 63);
482 483 484
            dref->filename[len] = 0;
            av_log(c->fc, AV_LOG_DEBUG, "filename %s, len %d\n", dref->filename, len);

485
            avio_skip(pb, 16);
486 487

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

493
            avio_skip(pb, 16);
494

495
            for (type = 0; type != -1 && avio_tell(pb) < next; ) {
496 497
                if (pb->eof_reached)
                    return AVERROR_EOF;
498 499
                type = avio_rb16(pb);
                len = avio_rb16(pb);
500 501 502 503
                av_log(c->fc, AV_LOG_DEBUG, "type %d, len %d\n", type, len);
                if (len&1)
                    len += 1;
                if (type == 2) { // absolute path
504
                    av_free(dref->path);
505
                    dref->path = av_mallocz(len+1);
506 507
                    if (!dref->path)
                        return AVERROR(ENOMEM);
508
                    avio_read(pb, dref->path, len);
509
                    if (len > volume_len && !strncmp(dref->path, dref->volume, volume_len)) {
510 511 512 513 514 515 516 517
                        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);
518 519 520 521 522
                } else if (type == 0) { // directory name
                    av_free(dref->dir);
                    dref->dir = av_malloc(len+1);
                    if (!dref->dir)
                        return AVERROR(ENOMEM);
523
                    avio_read(pb, dref->dir, len);
524 525 526 527 528
                    dref->dir[len] = 0;
                    for (j = 0; j < len; j++)
                        if (dref->dir[j] == ':')
                            dref->dir[j] = '/';
                    av_log(c->fc, AV_LOG_DEBUG, "dir %s\n", dref->dir);
529
                } else
530
                    avio_skip(pb, len);
531 532
            }
        }
A
Anton Khirnov 已提交
533
        avio_seek(pb, next, SEEK_SET);
534 535 536 537
    }
    return 0;
}

538
static int mov_read_hdlr(MOVContext *c, AVIOContext *pb, MOVAtom atom)
539
{
540
    AVStream *st;
541
    uint32_t type;
542
    uint32_t av_unused ctype;
543

544 545 546 547 548
    if (c->fc->nb_streams < 1) // meta before first trak
        return 0;

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

549 550
    avio_r8(pb); /* version */
    avio_rb24(pb); /* flags */
551 552

    /* component type */
553 554
    ctype = avio_rl32(pb);
    type = avio_rl32(pb); /* component subtype */
555

L
Luca Barbato 已提交
556 557
    av_dlog(c->fc, "ctype= %.4s (0x%08x)\n", (char*)&ctype, ctype);
    av_dlog(c->fc, "stype= %.4s\n", (char*)&type);
558

559
    if     (type == MKTAG('v','i','d','e'))
560
        st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
561
    else if (type == MKTAG('s','o','u','n'))
562
        st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
563
    else if (type == MKTAG('m','1','a',' '))
564
        st->codec->codec_id = AV_CODEC_ID_MP2;
565
    else if ((type == MKTAG('s','u','b','p')) || (type == MKTAG('c','l','c','p')))
566
        st->codec->codec_type = AVMEDIA_TYPE_SUBTITLE;
567

568 569 570
    avio_rb32(pb); /* component  manufacture */
    avio_rb32(pb); /* component flags */
    avio_rb32(pb); /* component flags mask */
571 572 573 574

    return 0;
}

575
int ff_mov_read_esds(AVFormatContext *fc, AVIOContext *pb)
576
{
577
    AVStream *st;
M
Mans Rullgard 已提交
578
    int tag;
579

580
    if (fc->nb_streams < 1)
581
        return 0;
582
    st = fc->streams[fc->nb_streams-1];
583

584
    avio_rb32(pb); /* version + flags */
M
Mans Rullgard 已提交
585
    ff_mp4_read_descr(fc, pb, &tag);
586
    if (tag == MP4ESDescrTag) {
587
        ff_mp4_parse_es_descr(pb, NULL);
588
    } else
589
        avio_rb16(pb); /* ID */
590

M
Mans Rullgard 已提交
591
    ff_mp4_read_descr(fc, pb, &tag);
592 593
    if (tag == MP4DecConfigDescrTag)
        ff_mp4_read_dec_config_descr(fc, st, pb);
594 595 596
    return 0;
}

597
static int mov_read_esds(MOVContext *c, AVIOContext *pb, MOVAtom atom)
598
{
599
    return ff_mov_read_esds(c->fc, pb);
600 601
}

602
static int mov_read_dac3(MOVContext *c, AVIOContext *pb, MOVAtom atom)
603 604
{
    AVStream *st;
605
    int ac3info, acmod, lfeon, bsmod;
606

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

611
    ac3info = avio_rb24(pb);
612
    bsmod = (ac3info >> 14) & 0x7;
613 614 615
    acmod = (ac3info >> 11) & 0x7;
    lfeon = (ac3info >> 10) & 0x1;
    st->codec->channels = ((int[]){2,1,2,3,3,4,4,5})[acmod] + lfeon;
616 617 618
    st->codec->channel_layout = avpriv_ac3_channel_layout_tab[acmod];
    if (lfeon)
        st->codec->channel_layout |= AV_CH_LOW_FREQUENCY;
619 620 621
    st->codec->audio_service_type = bsmod;
    if (st->codec->channels > 1 && bsmod == 0x7)
        st->codec->audio_service_type = AV_AUDIO_SERVICE_TYPE_KARAOKE;
622 623 624 625

    return 0;
}

626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653
static int mov_read_dec3(MOVContext *c, AVIOContext *pb, MOVAtom atom)
{
    AVStream *st;
    int eac3info, acmod, lfeon, bsmod;

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

    /* No need to parse fields for additional independent substreams and its
     * associated dependent substreams since libavcodec's E-AC-3 decoder
     * does not support them yet. */
    avio_rb16(pb); /* data_rate and num_ind_sub */
    eac3info = avio_rb24(pb);
    bsmod = (eac3info >> 12) & 0x1f;
    acmod = (eac3info >>  9) & 0x7;
    lfeon = (eac3info >>  8) & 0x1;
    st->codec->channel_layout = avpriv_ac3_channel_layout_tab[acmod];
    if (lfeon)
        st->codec->channel_layout |= AV_CH_LOW_FREQUENCY;
    st->codec->channels = av_get_channel_layout_nb_channels(st->codec->channel_layout);
    st->codec->audio_service_type = bsmod;
    if (st->codec->channels > 1 && bsmod == 0x7)
        st->codec->audio_service_type = AV_AUDIO_SERVICE_TYPE_KARAOKE;

    return 0;
}

654 655 656 657 658 659 660 661 662 663 664
static int mov_read_chan(MOVContext *c, AVIOContext *pb, MOVAtom atom)
{
    AVStream *st;

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

    if (atom.size < 16)
        return 0;

665 666 667
    /* skip version and flags */
    avio_skip(pb, 4);

668
    ff_mov_read_chan(c->fc, pb, st, atom.size - 4);
669 670 671 672

    return 0;
}

673 674 675 676 677 678 679 680
static int mov_read_wfex(MOVContext *c, AVIOContext *pb, MOVAtom atom)
{
    AVStream *st;

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

681
    return ff_get_wav_header(pb, st->codec, atom.size);
682 683
}

684
static int mov_read_pasp(MOVContext *c, AVIOContext *pb, MOVAtom atom)
685
{
686 687
    const int num = avio_rb32(pb);
    const int den = avio_rb32(pb);
688 689 690 691 692 693
    AVStream *st;

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

694 695 696 697 698 699 700
    if ((st->sample_aspect_ratio.den != 1 || st->sample_aspect_ratio.num) && // default
        (den != st->sample_aspect_ratio.den || num != st->sample_aspect_ratio.num)) {
        av_log(c->fc, AV_LOG_WARNING,
               "sample aspect ratio already set to %d:%d, ignoring 'pasp' atom (%d:%d)\n",
               st->sample_aspect_ratio.num, st->sample_aspect_ratio.den,
               num, den);
    } else if (den != 0) {
701 702 703 704 705 706
        st->sample_aspect_ratio.num = num;
        st->sample_aspect_ratio.den = den;
    }
    return 0;
}

707
/* this atom contains actual media data */
708
static int mov_read_mdat(MOVContext *c, AVIOContext *pb, MOVAtom atom)
709
{
710
    if (atom.size == 0) /* wrong one (MP4) */
711 712 713 714 715
        return 0;
    c->found_mdat=1;
    return 0; /* now go for moov */
}

716
/* read major brand, minor version and compatible brands and store them as metadata */
717
static int mov_read_ftyp(MOVContext *c, AVIOContext *pb, MOVAtom atom)
718
{
719 720 721 722 723 724
    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};

725
    avio_read(pb, type, 4);
B
Baptiste Coudurier 已提交
726
    if (strcmp(type, "qt  "))
727
        c->isom = 1;
728
    av_log(c->fc, AV_LOG_DEBUG, "ISO: File Type Major Brand: %.4s\n",(char *)&type);
729
    av_dict_set(&c->fc->metadata, "major_brand", type, 0);
730
    minor_ver = avio_rb32(pb); /* minor version */
731
    snprintf(minor_ver_str, sizeof(minor_ver_str), "%"PRIu32"", minor_ver);
732
    av_dict_set(&c->fc->metadata, "minor_version", minor_ver_str, 0);
733 734 735

    comp_brand_size = atom.size - 8;
    if (comp_brand_size < 0)
736
        return AVERROR_INVALIDDATA;
737 738 739
    comp_brands_str = av_malloc(comp_brand_size + 1); /* Add null terminator */
    if (!comp_brands_str)
        return AVERROR(ENOMEM);
740
    avio_read(pb, comp_brands_str, comp_brand_size);
741
    comp_brands_str[comp_brand_size] = 0;
742
    av_dict_set(&c->fc->metadata, "compatible_brands", comp_brands_str, 0);
743 744
    av_freep(&comp_brands_str);

745 746 747
    return 0;
}

748
/* this atom should contain all header atoms */
749
static int mov_read_moov(MOVContext *c, AVIOContext *pb, MOVAtom atom)
750
{
751 752 753 754
    int ret;

    if ((ret = mov_read_default(c, pb, atom)) < 0)
        return ret;
755 756 757 758 759 760
    /* 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 */
}

761
static int mov_read_moof(MOVContext *c, AVIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
762
{
763
    c->fragment.moof_offset = c->fragment.implicit_offset = avio_tell(pb) - 8;
764
    av_dlog(c->fc, "moof offset %"PRIx64"\n", c->fragment.moof_offset);
B
Baptiste Coudurier 已提交
765 766
    return mov_read_default(c, pb, atom);
}
767

768
static void mov_metadata_creation_time(AVDictionary **metadata, time_t time)
769 770 771
{
    char buffer[32];
    if (time) {
772
        struct tm *ptm, tmbuf;
773
        time -= 2082844800;  /* seconds between 1904-01-01 and Epoch */
774
        ptm = gmtime_r(&time, &tmbuf);
775
        if (!ptm) return;
776 777
        if (strftime(buffer, sizeof(buffer), "%Y-%m-%d %H:%M:%S", ptm))
            av_dict_set(metadata, "creation_time", buffer, 0);
778 779 780
    }
}

781
static int mov_read_mdhd(MOVContext *c, AVIOContext *pb, MOVAtom atom)
782
{
783 784 785
    AVStream *st;
    MOVStreamContext *sc;
    int version;
786
    char language[4] = {0};
787
    unsigned lang;
788
    time_t creation_time;
789

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

795 796 797 798 799
    if (sc->time_scale) {
        av_log(c->fc, AV_LOG_ERROR, "Multiple mdhd?\n");
        return AVERROR_INVALIDDATA;
    }

800
    version = avio_r8(pb);
801
    if (version > 1) {
802
        avpriv_request_sample(c->fc, "Version %d", version);
803 804
        return AVERROR_PATCHWELCOME;
    }
805
    avio_rb24(pb); /* flags */
B
clean  
Baptiste Coudurier 已提交
806
    if (version == 1) {
807 808
        creation_time = avio_rb64(pb);
        avio_rb64(pb);
B
clean  
Baptiste Coudurier 已提交
809
    } else {
810 811
        creation_time = avio_rb32(pb);
        avio_rb32(pb); /* modification time */
B
clean  
Baptiste Coudurier 已提交
812
    }
813
    mov_metadata_creation_time(&st->metadata, creation_time);
814

815 816
    sc->time_scale = avio_rb32(pb);
    st->duration = (version == 1) ? avio_rb64(pb) : avio_rb32(pb); /* duration */
817

818
    lang = avio_rb16(pb); /* language */
819
    if (ff_mov_lang_to_iso639(lang, language))
820
        av_dict_set(&st->metadata, "language", language, 0);
821
    avio_rb16(pb); /* quality */
822 823 824 825

    return 0;
}

826
static int mov_read_mvhd(MOVContext *c, AVIOContext *pb, MOVAtom atom)
827
{
828
    time_t creation_time;
829 830
    int version = avio_r8(pb); /* version */
    avio_rb24(pb); /* flags */
831

B
Baptiste Coudurier 已提交
832
    if (version == 1) {
833 834
        creation_time = avio_rb64(pb);
        avio_rb64(pb);
B
Baptiste Coudurier 已提交
835
    } else {
836 837
        creation_time = avio_rb32(pb);
        avio_rb32(pb); /* modification time */
B
Baptiste Coudurier 已提交
838
    }
839
    mov_metadata_creation_time(&c->fc->metadata, creation_time);
840
    c->time_scale = avio_rb32(pb); /* time scale */
841

L
Luca Barbato 已提交
842
    av_dlog(c->fc, "time scale = %i\n", c->time_scale);
843

844 845
    c->duration = (version == 1) ? avio_rb64(pb) : avio_rb32(pb); /* duration */
    avio_rb32(pb); /* preferred scale */
846

847
    avio_rb16(pb); /* preferred volume */
848

849
    avio_skip(pb, 10); /* reserved */
850

851
    avio_skip(pb, 36); /* display matrix */
852

853 854 855 856 857 858 859
    avio_rb32(pb); /* preview time */
    avio_rb32(pb); /* preview duration */
    avio_rb32(pb); /* poster time */
    avio_rb32(pb); /* selection time */
    avio_rb32(pb); /* selection duration */
    avio_rb32(pb); /* current time */
    avio_rb32(pb); /* next track ID */
860 861 862 863

    return 0;
}

864
static int mov_read_smi(MOVContext *c, AVIOContext *pb, MOVAtom atom)
865
{
866 867 868 869 870
    AVStream *st;

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

872
    if ((uint64_t)atom.size > (1<<30))
873
        return AVERROR_INVALIDDATA;
874

875 876
    // currently SVQ3 decoder expect full STSD header - so let's fake it
    // this should be fixed and just SMI header should be passed
877
    av_free(st->codec->extradata);
878 879 880
    st->codec->extradata = av_mallocz(atom.size + 0x5a + FF_INPUT_BUFFER_PADDING_SIZE);
    if (!st->codec->extradata)
        return AVERROR(ENOMEM);
B
Baptiste Coudurier 已提交
881 882
    st->codec->extradata_size = 0x5a + atom.size;
    memcpy(st->codec->extradata, "SVQ3", 4); // fake
883
    avio_read(pb, st->codec->extradata + 0x5a, atom.size);
L
Luca Barbato 已提交
884
    av_dlog(c->fc, "Reading SMI %"PRId64"  %s\n", atom.size, st->codec->extradata + 0x5a);
885 886
    return 0;
}
887

888
static int mov_read_enda(MOVContext *c, AVIOContext *pb, MOVAtom atom)
889
{
890 891 892 893 894 895
    AVStream *st;
    int little_endian;

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

897
    little_endian = avio_rb16(pb);
L
Luca Barbato 已提交
898
    av_dlog(c->fc, "enda %d\n", little_endian);
899
    if (little_endian == 1) {
900
        switch (st->codec->codec_id) {
901 902
        case AV_CODEC_ID_PCM_S24BE:
            st->codec->codec_id = AV_CODEC_ID_PCM_S24LE;
903
            break;
904 905
        case AV_CODEC_ID_PCM_S32BE:
            st->codec->codec_id = AV_CODEC_ID_PCM_S32LE;
906
            break;
907 908
        case AV_CODEC_ID_PCM_F32BE:
            st->codec->codec_id = AV_CODEC_ID_PCM_F32LE;
909
            break;
910 911
        case AV_CODEC_ID_PCM_F64BE:
            st->codec->codec_id = AV_CODEC_ID_PCM_F64LE;
912
            break;
913 914 915 916 917 918 919
        default:
            break;
        }
    }
    return 0;
}

920 921 922 923 924 925 926 927 928 929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955
static int mov_read_colr(MOVContext *c, AVIOContext *pb, MOVAtom atom)
{
    AVStream *st;
    char color_parameter_type[5] = { 0 };
    int color_primaries, color_trc, color_matrix;

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

    avio_read(pb, color_parameter_type, 4);
    if (strncmp(color_parameter_type, "nclx", 4) &&
        strncmp(color_parameter_type, "nclc", 4)) {
        av_log(c->fc, AV_LOG_WARNING, "unsupported color_parameter_type %s\n",
               color_parameter_type);
        return 0;
    }

    color_primaries = avio_rb16(pb);
    color_trc = avio_rb16(pb);
    color_matrix = avio_rb16(pb);

    av_dlog(c->fc, "%s: pri %"PRIu16" trc %"PRIu16" matrix %"PRIu16"",
            color_parameter_type, color_primaries, color_trc, color_matrix);

    if (c->isom) {
        uint8_t color_range = avio_r8(pb) >> 7;
        av_dlog(c->fc, " full %"PRIu8"", color_range)
        if (color_range)
            st->codec->color_range = AVCOL_RANGE_JPEG;
        else
            st->codec->color_range = AVCOL_RANGE_MPEG;
        /* 14496-12 references JPEG XR specs (rather than the more complete
         * 23001-8) so some adjusting is required */
        if (color_primaries >= AVCOL_PRI_FILM)
            color_primaries = AVCOL_PRI_UNSPECIFIED;
956 957
        if ((color_trc >= AVCOL_TRC_LINEAR &&
             color_trc <= AVCOL_TRC_LOG_SQRT) ||
958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988
            color_trc >= AVCOL_TRC_BT2020_10)
            color_trc = AVCOL_TRC_UNSPECIFIED;
        if (color_matrix >= AVCOL_SPC_BT2020_NCL)
            color_matrix = AVCOL_SPC_UNSPECIFIED;
        st->codec->color_primaries = color_primaries;
        st->codec->color_trc = color_trc;
        st->codec->colorspace = color_matrix;
    } else {
        /* color primaries, Table 4-4 */
        switch (color_primaries) {
        case 1: st->codec->color_primaries = AVCOL_PRI_BT709; break;
        case 5: st->codec->color_primaries = AVCOL_PRI_SMPTE170M; break;
        case 6: st->codec->color_primaries = AVCOL_PRI_SMPTE240M; break;
        }
        /* color transfer, Table 4-5 */
        switch (color_trc) {
        case 1: st->codec->color_trc = AVCOL_TRC_BT709; break;
        case 7: st->codec->color_trc = AVCOL_TRC_SMPTE240M; break;
        }
        /* color matrix, Table 4-6 */
        switch (color_matrix) {
        case 1: st->codec->colorspace = AVCOL_SPC_BT709; break;
        case 6: st->codec->colorspace = AVCOL_SPC_BT470BG; break;
        case 7: st->codec->colorspace = AVCOL_SPC_SMPTE240M; break;
        }
    }
    av_dlog(c->fc, "\n")

    return 0;
}

989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022
static int mov_read_fiel(MOVContext *c, AVIOContext *pb, MOVAtom atom)
{
    AVStream *st;
    unsigned mov_field_order;
    enum AVFieldOrder decoded_field_order = AV_FIELD_UNKNOWN;

    if (c->fc->nb_streams < 1) // will happen with jp2 files
        return 0;
    st = c->fc->streams[c->fc->nb_streams-1];
    if (atom.size < 2)
        return AVERROR_INVALIDDATA;
    mov_field_order = avio_rb16(pb);
    if ((mov_field_order & 0xFF00) == 0x0100)
        decoded_field_order = AV_FIELD_PROGRESSIVE;
    else if ((mov_field_order & 0xFF00) == 0x0200) {
        switch (mov_field_order & 0xFF) {
        case 0x01: decoded_field_order = AV_FIELD_TT;
                   break;
        case 0x06: decoded_field_order = AV_FIELD_BB;
                   break;
        case 0x09: decoded_field_order = AV_FIELD_TB;
                   break;
        case 0x0E: decoded_field_order = AV_FIELD_BT;
                   break;
        }
    }
    if (decoded_field_order == AV_FIELD_UNKNOWN && mov_field_order) {
        av_log(NULL, AV_LOG_ERROR, "Unknown MOV field order 0x%04x\n", mov_field_order);
    }
    st->codec->field_order = decoded_field_order;

    return 0;
}

1023
/* FIXME modify qdm2/svq3/h264 decoders to take full atom as extradata */
1024
static int mov_read_extradata(MOVContext *c, AVIOContext *pb, MOVAtom atom)
1025
{
1026 1027
    AVStream *st;
    uint64_t size;
1028
    uint8_t *buf;
1029
    int err;
1030 1031 1032 1033 1034

    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;
1035
    if (size > INT_MAX || (uint64_t)atom.size > INT_MAX)
1036
        return AVERROR_INVALIDDATA;
1037 1038
    if ((err = av_reallocp(&st->codec->extradata, size)) < 0) {
        st->codec->extradata_size = 0;
1039
        return err;
1040
    }
1041
    buf = st->codec->extradata + st->codec->extradata_size;
1042 1043 1044
    st->codec->extradata_size= size - FF_INPUT_BUFFER_PADDING_SIZE;
    AV_WB32(       buf    , atom.size + 8);
    AV_WL32(       buf + 4, atom.type);
1045
    avio_read(pb, buf + 8, atom.size);
1046 1047 1048
    return 0;
}

1049
static int mov_read_wave(MOVContext *c, AVIOContext *pb, MOVAtom atom)
R
Roberto Togni 已提交
1050
{
1051 1052 1053 1054 1055
    AVStream *st;

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

1057
    if ((uint64_t)atom.size > (1<<30))
1058
        return AVERROR_INVALIDDATA;
1059

1060
    if (st->codec->codec_id == AV_CODEC_ID_QDM2 || st->codec->codec_id == AV_CODEC_ID_QDMC) {
1061
        // pass all frma atom to codec, needed at least for QDMC and QDM2
1062
        av_free(st->codec->extradata);
1063 1064 1065
        st->codec->extradata = av_mallocz(atom.size + FF_INPUT_BUFFER_PADDING_SIZE);
        if (!st->codec->extradata)
            return AVERROR(ENOMEM);
1066
        st->codec->extradata_size = atom.size;
1067
        avio_read(pb, st->codec->extradata, atom.size);
1068
    } else if (atom.size > 8) { /* to read frma, esds atoms */
1069 1070 1071
        int ret;
        if ((ret = mov_read_default(c, pb, atom)) < 0)
            return ret;
1072
    } else
1073
        avio_skip(pb, atom.size);
R
Roberto Togni 已提交
1074 1075 1076
    return 0;
}

1077 1078 1079 1080
/**
 * This function reads atom content and puts data in extradata without tag
 * nor size unlike mov_read_extradata.
 */
1081
static int mov_read_glbl(MOVContext *c, AVIOContext *pb, MOVAtom atom)
1082
{
1083 1084 1085 1086 1087
    AVStream *st;

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

1089
    if ((uint64_t)atom.size > (1<<30))
1090
        return AVERROR_INVALIDDATA;
1091

1092
    if (atom.size >= 10) {
1093
        // Broken files created by legacy versions of libavformat will
1094 1095 1096 1097 1098 1099 1100
        // wrap a whole fiel atom inside of a glbl atom.
        unsigned size = avio_rb32(pb);
        unsigned type = avio_rl32(pb);
        avio_seek(pb, -8, SEEK_CUR);
        if (type == MKTAG('f','i','e','l') && size == atom.size)
            return mov_read_default(c, pb, atom);
    }
1101
    av_free(st->codec->extradata);
1102 1103 1104
    st->codec->extradata = av_mallocz(atom.size + FF_INPUT_BUFFER_PADDING_SIZE);
    if (!st->codec->extradata)
        return AVERROR(ENOMEM);
1105
    st->codec->extradata_size = atom.size;
1106
    avio_read(pb, st->codec->extradata, atom.size);
1107 1108 1109
    return 0;
}

M
Martin Storsjö 已提交
1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122
static int mov_read_dvc1(MOVContext *c, AVIOContext *pb, MOVAtom atom)
{
    AVStream *st;
    uint8_t profile_level;

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

    if (atom.size >= (1<<28) || atom.size < 7)
        return AVERROR_INVALIDDATA;

    profile_level = avio_r8(pb);
1123
    if ((profile_level & 0xf0) != 0xc0)
M
Martin Storsjö 已提交
1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135
        return 0;

    av_free(st->codec->extradata);
    st->codec->extradata = av_mallocz(atom.size - 7 + FF_INPUT_BUFFER_PADDING_SIZE);
    if (!st->codec->extradata)
        return AVERROR(ENOMEM);
    st->codec->extradata_size = atom.size - 7;
    avio_seek(pb, 6, SEEK_CUR);
    avio_read(pb, st->codec->extradata, st->codec->extradata_size);
    return 0;
}

M
Martin Storsjö 已提交
1136 1137 1138 1139 1140
/**
 * An strf atom is a BITMAPINFOHEADER struct. This struct is 40 bytes itself,
 * but can have extradata appended at the end after the 40 bytes belonging
 * to the struct.
 */
1141
static int mov_read_strf(MOVContext *c, AVIOContext *pb, MOVAtom atom)
M
Martin Storsjö 已提交
1142 1143 1144 1145 1146 1147 1148 1149 1150
{
    AVStream *st;

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

1151
    if ((uint64_t)atom.size > (1<<30))
1152
        return AVERROR_INVALIDDATA;
M
Martin Storsjö 已提交
1153 1154 1155 1156 1157 1158

    av_free(st->codec->extradata);
    st->codec->extradata = av_mallocz(atom.size - 40 + FF_INPUT_BUFFER_PADDING_SIZE);
    if (!st->codec->extradata)
        return AVERROR(ENOMEM);
    st->codec->extradata_size = atom.size - 40;
1159
    avio_skip(pb, 40);
1160
    avio_read(pb, st->codec->extradata, atom.size - 40);
M
Martin Storsjö 已提交
1161 1162 1163
    return 0;
}

1164
static int mov_read_stco(MOVContext *c, AVIOContext *pb, MOVAtom atom)
1165
{
1166 1167
    AVStream *st;
    MOVStreamContext *sc;
1168
    unsigned int i, entries;
1169

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

1175 1176
    avio_r8(pb); /* version */
    avio_rb24(pb); /* flags */
1177

1178
    entries = avio_rb32(pb);
1179

A
Alex Converse 已提交
1180 1181
    if (!entries)
        return 0;
1182
    if (entries >= UINT_MAX/sizeof(int64_t))
1183
        return AVERROR_INVALIDDATA;
1184

B
Baptiste Coudurier 已提交
1185
    sc->chunk_offsets = av_malloc(entries * sizeof(int64_t));
1186
    if (!sc->chunk_offsets)
1187 1188 1189
        return AVERROR(ENOMEM);
    sc->chunk_count = entries;

1190
    if      (atom.type == MKTAG('s','t','c','o'))
1191
        for (i = 0; i < entries && !pb->eof_reached; i++)
1192
            sc->chunk_offsets[i] = avio_rb32(pb);
1193
    else if (atom.type == MKTAG('c','o','6','4'))
1194
        for (i = 0; i < entries && !pb->eof_reached; i++)
1195
            sc->chunk_offsets[i] = avio_rb64(pb);
1196
    else
1197
        return AVERROR_INVALIDDATA;
1198

1199 1200 1201 1202 1203
    sc->chunk_count = i;

    if (pb->eof_reached)
        return AVERROR_EOF;

1204 1205 1206
    return 0;
}

1207 1208 1209 1210
/**
 * Compute codec id for 'lpcm' tag.
 * See CoreAudioTypes and AudioStreamBasicDescription at Apple.
 */
1211
enum AVCodecID ff_mov_get_lpcm_codec_id(int bps, int flags)
1212
{
1213 1214 1215 1216 1217 1218
    /* lpcm flags:
     * 0x1 = float
     * 0x2 = big-endian
     * 0x4 = signed
     */
    return ff_get_pcm_codec_id(bps, flags & 1, flags & 2, flags & 4 ? -1 : 0);
1219 1220
}

1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251
static int mov_codec_id(AVStream *st, uint32_t format)
{
    int id = ff_codec_get_id(ff_codec_movaudio_tags, format);

    if (id <= 0 &&
        ((format & 0xFFFF) == 'm' + ('s' << 8) ||
         (format & 0xFFFF) == 'T' + ('S' << 8)))
        id = ff_codec_get_id(ff_codec_wav_tags, av_bswap32(format) & 0xFFFF);

    if (st->codec->codec_type != AVMEDIA_TYPE_VIDEO && id > 0) {
        st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
    } else if (st->codec->codec_type != AVMEDIA_TYPE_AUDIO &&
               /* skip old asf mpeg4 tag */
               format && format != MKTAG('m','p','4','s')) {
        id = ff_codec_get_id(ff_codec_movvideo_tags, format);
        if (id <= 0)
            id = ff_codec_get_id(ff_codec_bmp_tags, format);
        if (id > 0)
            st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
        else if (st->codec->codec_type == AVMEDIA_TYPE_DATA) {
            id = ff_codec_get_id(ff_codec_movsubtitle_tags, format);
            if (id > 0)
                st->codec->codec_type = AVMEDIA_TYPE_SUBTITLE;
        }
    }

    st->codec->codec_tag = format;

    return id;
}

1252 1253 1254
static void mov_parse_stsd_video(MOVContext *c, AVIOContext *pb,
                                 AVStream *st, MOVStreamContext *sc)
{
1255
    uint8_t codec_name[32];
1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276
    unsigned int color_depth, len, j;
    int color_greyscale;
    int color_table_id;

    avio_rb16(pb); /* version */
    avio_rb16(pb); /* revision level */
    avio_rb32(pb); /* vendor */
    avio_rb32(pb); /* temporal quality */
    avio_rb32(pb); /* spatial quality */

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

    avio_rb32(pb); /* horiz resolution */
    avio_rb32(pb); /* vert resolution */
    avio_rb32(pb); /* data size, always 0 */
    avio_rb16(pb); /* frames per samples */

    len = avio_r8(pb); /* codec name, pascal string */
    if (len > 31)
        len = 31;
1277
    mov_read_mac_string(c, pb, len, codec_name, sizeof(codec_name));
1278 1279
    if (len < 31)
        avio_skip(pb, 31 - len);
1280 1281 1282 1283

    if (codec_name[0])
        av_dict_set(&st->metadata, "encoder", codec_name, 0);

1284
    /* codec_tag YV12 triggers an UV swap in rawdec.c */
1285
    if (!memcmp(codec_name, "Planar Y'CbCr 8-bit 4:2:0", 25))
1286 1287 1288
        st->codec->codec_tag = MKTAG('I', '4', '2', '0');
    /* Flash Media Server uses tag H263 with Sorenson Spark */
    if (st->codec->codec_tag == MKTAG('H','2','6','3') &&
1289
        !memcmp(codec_name, "Sorenson H263", 13))
1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362
        st->codec->codec_id = AV_CODEC_ID_FLV1;

    st->codec->bits_per_coded_sample = avio_rb16(pb); /* depth */
    color_table_id = avio_rb16(pb); /* colortable id */
    av_dlog(c->fc, "depth %d, ctab id %d\n",
            st->codec->bits_per_coded_sample, color_table_id);
    /* figure out the palette situation */
    color_depth     = st->codec->bits_per_coded_sample & 0x1F;
    color_greyscale = st->codec->bits_per_coded_sample & 0x20;

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

        if (color_greyscale) {
            int color_index, color_dec;
            /* compute the greyscale palette */
            st->codec->bits_per_coded_sample = color_depth;
            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;
                sc->palette[j] = (r << 16) | (g << 8) | (b);
                color_index -= color_dec;
                if (color_index < 0)
                    color_index = 0;
            }
        } else if (color_table_id) {
            const uint8_t *color_table;
            /* if flag bit 3 is set, use the default palette */
            color_count = 1 << color_depth;
            if (color_depth == 2)
                color_table = ff_qt_default_palette_4;
            else if (color_depth == 4)
                color_table = ff_qt_default_palette_16;
            else
                color_table = ff_qt_default_palette_256;

            for (j = 0; j < color_count; j++) {
                r = color_table[j * 3 + 0];
                g = color_table[j * 3 + 1];
                b = color_table[j * 3 + 2];
                sc->palette[j] = (r << 16) | (g << 8) | (b);
            }
        } else {
            /* load the palette from the file */
            color_start = avio_rb32(pb);
            color_count = avio_rb16(pb);
            color_end   = avio_rb16(pb);
            if ((color_start <= 255) && (color_end <= 255)) {
                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 */
                    avio_r8(pb);
                    avio_r8(pb);
                    r = avio_r8(pb);
                    avio_r8(pb);
                    g = avio_r8(pb);
                    avio_r8(pb);
                    b = avio_r8(pb);
                    avio_r8(pb);
                    sc->palette[j] = (r << 16) | (g << 8) | (b);
                }
            }
        }
        sc->has_palette = 1;
    }
}

1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448
static void mov_parse_stsd_audio(MOVContext *c, AVIOContext *pb,
                                 AVStream *st, MOVStreamContext *sc)
{
    int bits_per_sample, flags;
    uint16_t version = avio_rb16(pb);

    avio_rb16(pb); /* revision level */
    avio_rb32(pb); /* vendor */

    st->codec->channels              = avio_rb16(pb); /* channel count */
    st->codec->bits_per_coded_sample = avio_rb16(pb); /* sample size */
    av_dlog(c->fc, "audio channels %d\n", st->codec->channels);

    sc->audio_cid = avio_rb16(pb);
    avio_rb16(pb); /* packet size = 0 */

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

    // Read QT version 1 fields. In version 0 these do not exist.
    av_dlog(c->fc, "version =%d, isom =%d\n", version, c->isom);
    if (!c->isom) {
        if (version == 1) {
            sc->samples_per_frame = avio_rb32(pb);
            avio_rb32(pb); /* bytes per packet */
            sc->bytes_per_frame = avio_rb32(pb);
            avio_rb32(pb); /* bytes per sample */
        } else if (version == 2) {
            avio_rb32(pb); /* sizeof struct only */
            st->codec->sample_rate = av_int2double(avio_rb64(pb));
            st->codec->channels    = avio_rb32(pb);
            avio_rb32(pb); /* always 0x7F000000 */
            st->codec->bits_per_coded_sample = avio_rb32(pb);

            flags = avio_rb32(pb); /* lpcm format specific flag */
            sc->bytes_per_frame   = avio_rb32(pb);
            sc->samples_per_frame = avio_rb32(pb);
            if (st->codec->codec_tag == MKTAG('l','p','c','m'))
                st->codec->codec_id =
                    ff_mov_get_lpcm_codec_id(st->codec->bits_per_coded_sample,
                                             flags);
        }
    }

    switch (st->codec->codec_id) {
    case AV_CODEC_ID_PCM_S8:
    case AV_CODEC_ID_PCM_U8:
        if (st->codec->bits_per_coded_sample == 16)
            st->codec->codec_id = AV_CODEC_ID_PCM_S16BE;
        break;
    case AV_CODEC_ID_PCM_S16LE:
    case AV_CODEC_ID_PCM_S16BE:
        if (st->codec->bits_per_coded_sample == 8)
            st->codec->codec_id = AV_CODEC_ID_PCM_S8;
        else if (st->codec->bits_per_coded_sample == 24)
            st->codec->codec_id =
                st->codec->codec_id == AV_CODEC_ID_PCM_S16BE ?
                AV_CODEC_ID_PCM_S24BE : AV_CODEC_ID_PCM_S24LE;
        break;
    /* set values for old format before stsd version 1 appeared */
    case AV_CODEC_ID_MACE3:
        sc->samples_per_frame = 6;
        sc->bytes_per_frame   = 2 * st->codec->channels;
        break;
    case AV_CODEC_ID_MACE6:
        sc->samples_per_frame = 6;
        sc->bytes_per_frame   = 1 * st->codec->channels;
        break;
    case AV_CODEC_ID_ADPCM_IMA_QT:
        sc->samples_per_frame = 64;
        sc->bytes_per_frame   = 34 * st->codec->channels;
        break;
    case AV_CODEC_ID_GSM:
        sc->samples_per_frame = 160;
        sc->bytes_per_frame   = 33;
        break;
    default:
        break;
    }

    bits_per_sample = av_get_bits_per_sample(st->codec->codec_id);
    if (bits_per_sample) {
        st->codec->bits_per_coded_sample = bits_per_sample;
        sc->sample_size = (bits_per_sample >> 3) * st->codec->channels;
    }
}

1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462
static void mov_parse_stsd_subtitle(MOVContext *c, AVIOContext *pb,
                                    AVStream *st, MOVStreamContext *sc,
                                    int size)
{
    // ttxt stsd contains display flags, justification, background
    // color, fonts, and default styles, so fake an atom to read it
    MOVAtom fake_atom = { .size = size };
    // mp4s contains a regular esds atom
    if (st->codec->codec_tag != AV_RL32("mp4s"))
        mov_read_glbl(c, pb, fake_atom);
    st->codec->width  = sc->width;
    st->codec->height = sc->height;
}

1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496
static uint32_t yuv_to_rgba(uint32_t ycbcr)
{
    uint8_t r, g, b;
    int y, cb, cr;

    y  = (ycbcr >> 16) & 0xFF;
    cr = (ycbcr >> 8)  & 0xFF;
    cb =  ycbcr        & 0xFF;

    b = av_clip_uint8(1.164 * (y - 16)                      + 2.018 * (cb - 128));
    g = av_clip_uint8(1.164 * (y - 16) - 0.813 * (cr - 128) - 0.391 * (cb - 128));
    r = av_clip_uint8(1.164 * (y - 16) + 1.596 * (cr - 128));

    return (r << 16) | (g << 8) | b;
}

static int mov_rewrite_dvd_sub_extradata(AVStream *st)
{
    char buf[256] = {0};
    uint8_t *src = st->codec->extradata;
    int i;

    if (st->codec->extradata_size != 64)
        return 0;

    if (st->codec->width > 0 &&  st->codec->height > 0)
        snprintf(buf, sizeof(buf), "size: %dx%d\n",
                 st->codec->width, st->codec->height);
    av_strlcat(buf, "palette: ", sizeof(buf));

    for (i = 0; i < 16; i++) {
        uint32_t yuv = AV_RB32(src + i * 4);
        uint32_t rgba = yuv_to_rgba(yuv);

1497
        av_strlcatf(buf, sizeof(buf), "%06"PRIx32"%s", rgba, i != 15 ? ", " : "");
1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513
    }

    if (av_strlcat(buf, "\n", sizeof(buf)) >= sizeof(buf))
        return 0;

    av_freep(&st->codec->extradata);
    st->codec->extradata_size = 0;
    st->codec->extradata = av_mallocz(strlen(buf) + FF_INPUT_BUFFER_PADDING_SIZE);
    if (!st->codec->extradata)
        return AVERROR(ENOMEM);
    st->codec->extradata_size = strlen(buf);
    memcpy(st->codec->extradata, buf, st->codec->extradata_size);

    return 0;
}

L
Luca Barbato 已提交
1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530
static int mov_parse_stsd_data(MOVContext *c, AVIOContext *pb,
                                AVStream *st, MOVStreamContext *sc,
                                int size)
{
    if (st->codec->codec_tag == MKTAG('t','m','c','d')) {
        st->codec->extradata_size = size;
        st->codec->extradata = av_malloc(size + FF_INPUT_BUFFER_PADDING_SIZE);
        if (!st->codec->extradata)
            return AVERROR(ENOMEM);
        avio_read(pb, st->codec->extradata, size);
    } else {
        /* other codec type, just skip (rtp, mp4s ...) */
        avio_skip(pb, size);
    }
    return 0;
}

1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594
static int mov_finalize_stsd_codec(MOVContext *c, AVIOContext *pb,
                                   AVStream *st, MOVStreamContext *sc)
{
    if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO &&
        !st->codec->sample_rate && sc->time_scale > 1)
        st->codec->sample_rate = sc->time_scale;

    /* special codec parameters handling */
    switch (st->codec->codec_id) {
#if CONFIG_DV_DEMUXER
    case AV_CODEC_ID_DVAUDIO:
        c->dv_fctx  = avformat_alloc_context();
        c->dv_demux = avpriv_dv_init_demux(c->dv_fctx);
        if (!c->dv_demux) {
            av_log(c->fc, AV_LOG_ERROR, "dv demux context init error\n");
            return AVERROR(ENOMEM);
        }
        sc->dv_audio_container = 1;
        st->codec->codec_id    = AV_CODEC_ID_PCM_S16LE;
        break;
#endif
    /* no ifdef since parameters are always those */
    case AV_CODEC_ID_QCELP:
        st->codec->channels = 1;
        // 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;
        break;
    case AV_CODEC_ID_AMR_NB:
        st->codec->channels    = 1;
        /* force sample rate for amr, stsd in 3gp does not store sample rate */
        st->codec->sample_rate = 8000;
        break;
    case AV_CODEC_ID_AMR_WB:
        st->codec->channels    = 1;
        st->codec->sample_rate = 16000;
        break;
    case AV_CODEC_ID_MP2:
    case AV_CODEC_ID_MP3:
        /* force type after stsd for m1a hdlr */
        st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
        st->need_parsing      = AVSTREAM_PARSE_FULL;
        break;
    case AV_CODEC_ID_GSM:
    case AV_CODEC_ID_ADPCM_MS:
    case AV_CODEC_ID_ADPCM_IMA_WAV:
    case AV_CODEC_ID_ILBC:
        st->codec->block_align = sc->bytes_per_frame;
        break;
    case AV_CODEC_ID_ALAC:
        if (st->codec->extradata_size == 36) {
            st->codec->channels    = AV_RB8 (st->codec->extradata + 21);
            st->codec->sample_rate = AV_RB32(st->codec->extradata + 32);
        }
        break;
    case AV_CODEC_ID_VC1:
        st->need_parsing = AVSTREAM_PARSE_FULL;
        break;
    default:
        break;
    }
    return 0;
}

1595 1596 1597 1598 1599 1600 1601 1602
static int mov_skip_multiple_stsd(MOVContext *c, AVIOContext *pb,
                                  int codec_tag, int format,
                                  int size)
{
    int video_codec_id = ff_codec_get_id(ff_codec_movvideo_tags, format);

    if (codec_tag &&
        (codec_tag == AV_RL32("avc1") ||
Y
Yusuke Nakamura 已提交
1603 1604
         codec_tag == AV_RL32("hvc1") ||
         codec_tag == AV_RL32("hev1") ||
1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619
         (codec_tag != format &&
          (c->fc->video_codec_id ? video_codec_id != c->fc->video_codec_id
                                 : codec_tag != MKTAG('j','p','e','g'))))) {
        /* 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. */

        av_log(c->fc, AV_LOG_WARNING, "multiple fourcc not supported\n");
        avio_skip(pb, size);
        return 1;
    }

    return 0;
}

1620
int ff_mov_read_stsd_entries(MOVContext *c, AVIOContext *pb, int entries)
1621
{
1622 1623
    AVStream *st;
    MOVStreamContext *sc;
1624
    int pseudo_stream_id;
1625

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

1631 1632 1633
    for (pseudo_stream_id = 0;
         pseudo_stream_id < entries && !pb->eof_reached;
         pseudo_stream_id++) {
1634
        //Parsing Sample description table
1635
        enum AVCodecID id;
L
Luca Barbato 已提交
1636
        int ret, dref_id = 1;
1637
        MOVAtom a = { AV_RL32("stsd") };
1638
        int64_t start_pos = avio_tell(pb);
1639
        uint32_t size = avio_rb32(pb); /* size */
1640
        uint32_t format = avio_rl32(pb); /* data format */
1641

1642
        if (size >= 16) {
1643 1644 1645
            avio_rb32(pb); /* reserved */
            avio_rb16(pb); /* reserved */
            dref_id = avio_rb16(pb);
1646
        } else {
1647
            av_log(c->fc, AV_LOG_ERROR, "invalid size %"PRIu32" in stsd\n", size);
1648
            return AVERROR_INVALIDDATA;
1649
        }
1650

1651 1652
        if (mov_skip_multiple_stsd(c, pb, st->codec->codec_tag, format,
                                   size - (avio_tell(pb) - start_pos)))
1653
            continue;
1654

1655
        sc->pseudo_stream_id = st->codec->codec_tag ? -1 : pseudo_stream_id;
1656
        sc->dref_id= dref_id;
1657

1658
        id = mov_codec_id(st, format);
1659

1660
        av_dlog(c->fc, "size=%"PRIu32" 4CC= %"PRIu8"%"PRIu8"%"PRIu8"%"PRIu8" codec_type=%d\n", size,
1661 1662
                (format >> 0) & 0xff, (format >> 8) & 0xff, (format >> 16) & 0xff,
                (format >> 24) & 0xff, st->codec->codec_type);
1663

1664
        if (st->codec->codec_type==AVMEDIA_TYPE_VIDEO) {
1665
            st->codec->codec_id = id;
1666
            mov_parse_stsd_video(c, pb, st, sc);
1667
        } else if (st->codec->codec_type==AVMEDIA_TYPE_AUDIO) {
1668
            st->codec->codec_id = id;
1669
            mov_parse_stsd_audio(c, pb, st, sc);
1670
        } else if (st->codec->codec_type==AVMEDIA_TYPE_SUBTITLE){
1671 1672 1673
            st->codec->codec_id = id;
            mov_parse_stsd_subtitle(c, pb, st, sc,
                                    size - (avio_tell(pb) - start_pos));
1674
        } else {
L
Luca Barbato 已提交
1675 1676 1677 1678
            ret = mov_parse_stsd_data(c, pb, st, sc,
                                      size - (avio_tell(pb) - start_pos));
            if (ret < 0)
                return ret;
1679
        }
Y
Yusuke Nakamura 已提交
1680
        /* this will read extra atoms at the end (wave, alac, damr, avcC, hvcC, SMI ...) */
1681
        a.size = size - (avio_tell(pb) - start_pos);
1682
        if (a.size > 8) {
1683 1684
            if ((ret = mov_read_default(c, pb, a)) < 0)
                return ret;
1685
        } else if (a.size > 0)
1686
            avio_skip(pb, a.size);
1687
    }
1688

1689 1690 1691
    if (pb->eof_reached)
        return AVERROR_EOF;

1692
    return mov_finalize_stsd_codec(c, pb, st, sc);
1693 1694
}

1695
static int mov_read_stsd(MOVContext *c, AVIOContext *pb, MOVAtom atom)
1696 1697 1698
{
    int entries;

1699 1700 1701
    avio_r8(pb); /* version */
    avio_rb24(pb); /* flags */
    entries = avio_rb32(pb);
1702 1703 1704 1705

    return ff_mov_read_stsd_entries(c, pb, entries);
}

1706
static int mov_read_stsc(MOVContext *c, AVIOContext *pb, MOVAtom atom)
1707
{
1708 1709
    AVStream *st;
    MOVStreamContext *sc;
1710
    unsigned int i, entries;
1711

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

1717 1718
    avio_r8(pb); /* version */
    avio_rb24(pb); /* flags */
1719

1720
    entries = avio_rb32(pb);
1721

L
Luca Barbato 已提交
1722
    av_dlog(c->fc, "track[%i].stsc.entries = %i\n", c->fc->nb_streams-1, entries);
1723

A
Alex Converse 已提交
1724 1725
    if (!entries)
        return 0;
1726
    if (entries >= UINT_MAX / sizeof(*sc->stsc_data))
1727
        return AVERROR_INVALIDDATA;
1728 1729
    sc->stsc_data = av_malloc(entries * sizeof(*sc->stsc_data));
    if (!sc->stsc_data)
1730 1731
        return AVERROR(ENOMEM);

1732
    for (i = 0; i < entries && !pb->eof_reached; i++) {
1733 1734 1735
        sc->stsc_data[i].first = avio_rb32(pb);
        sc->stsc_data[i].count = avio_rb32(pb);
        sc->stsc_data[i].id = avio_rb32(pb);
1736
    }
1737 1738 1739 1740 1741 1742

    sc->stsc_count = i;

    if (pb->eof_reached)
        return AVERROR_EOF;

1743 1744 1745
    return 0;
}

1746
static int mov_read_stps(MOVContext *c, AVIOContext *pb, MOVAtom atom)
1747 1748 1749 1750 1751 1752 1753 1754 1755 1756
{
    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;

1757
    avio_rb32(pb); // version + flags
1758

1759
    entries = avio_rb32(pb);
1760
    if (entries >= UINT_MAX / sizeof(*sc->stps_data))
1761
        return AVERROR_INVALIDDATA;
1762 1763 1764 1765
    sc->stps_data = av_malloc(entries * sizeof(*sc->stps_data));
    if (!sc->stps_data)
        return AVERROR(ENOMEM);

1766
    for (i = 0; i < entries && !pb->eof_reached; i++) {
1767
        sc->stps_data[i] = avio_rb32(pb);
L
Luca Barbato 已提交
1768
        //av_dlog(c->fc, "stps %d\n", sc->stps_data[i]);
1769 1770
    }

1771 1772 1773 1774 1775
    sc->stps_count = i;

    if (pb->eof_reached)
        return AVERROR_EOF;

1776 1777 1778
    return 0;
}

1779
static int mov_read_stss(MOVContext *c, AVIOContext *pb, MOVAtom atom)
1780
{
1781 1782
    AVStream *st;
    MOVStreamContext *sc;
1783
    unsigned int i, entries;
1784

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

1790 1791
    avio_r8(pb); /* version */
    avio_rb24(pb); /* flags */
1792

1793
    entries = avio_rb32(pb);
1794

L
Luca Barbato 已提交
1795
    av_dlog(c->fc, "keyframe_count = %d\n", entries);
1796

1797
    if (!entries)
1798 1799
    {
        sc->keyframe_absent = 1;
1800
        return 0;
1801
    }
1802
    if (entries >= UINT_MAX / sizeof(int))
1803
        return AVERROR_INVALIDDATA;
B
Baptiste Coudurier 已提交
1804
    sc->keyframes = av_malloc(entries * sizeof(int));
1805
    if (!sc->keyframes)
1806 1807
        return AVERROR(ENOMEM);

1808
    for (i = 0; i < entries && !pb->eof_reached; i++) {
1809
        sc->keyframes[i] = avio_rb32(pb);
L
Luca Barbato 已提交
1810
        //av_dlog(c->fc, "keyframes[]=%d\n", sc->keyframes[i]);
1811
    }
1812 1813 1814 1815 1816 1817

    sc->keyframe_count = i;

    if (pb->eof_reached)
        return AVERROR_EOF;

1818 1819 1820
    return 0;
}

1821
static int mov_read_stsz(MOVContext *c, AVIOContext *pb, MOVAtom atom)
1822
{
1823 1824
    AVStream *st;
    MOVStreamContext *sc;
1825 1826 1827
    unsigned int i, entries, sample_size, field_size, num_bytes;
    GetBitContext gb;
    unsigned char* buf;
1828

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

1834 1835
    avio_r8(pb); /* version */
    avio_rb24(pb); /* flags */
1836

1837
    if (atom.type == MKTAG('s','t','s','z')) {
1838
        sample_size = avio_rb32(pb);
1839 1840 1841
        if (!sc->sample_size) /* do not overwrite value computed in stsd */
            sc->sample_size = sample_size;
        field_size = 32;
1842 1843
    } else {
        sample_size = 0;
1844 1845
        avio_rb24(pb); /* reserved */
        field_size = avio_r8(pb);
1846
    }
1847
    entries = avio_rb32(pb);
1848

L
Luca Barbato 已提交
1849
    av_dlog(c->fc, "sample_size = %d sample_count = %d\n", sc->sample_size, entries);
1850

1851
    sc->sample_count = entries;
1852 1853 1854
    if (sample_size)
        return 0;

1855 1856
    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);
1857
        return AVERROR_INVALIDDATA;
1858 1859
    }

A
Alex Converse 已提交
1860 1861
    if (!entries)
        return 0;
1862
    if (entries >= UINT_MAX / sizeof(int) || entries >= (UINT_MAX - 4) / field_size)
1863
        return AVERROR_INVALIDDATA;
B
Baptiste Coudurier 已提交
1864
    sc->sample_sizes = av_malloc(entries * sizeof(int));
1865
    if (!sc->sample_sizes)
1866 1867
        return AVERROR(ENOMEM);

1868 1869
    num_bytes = (entries*field_size+4)>>3;

1870
    buf = av_malloc(num_bytes+FF_INPUT_BUFFER_PADDING_SIZE);
1871 1872 1873 1874 1875
    if (!buf) {
        av_freep(&sc->sample_sizes);
        return AVERROR(ENOMEM);
    }

1876
    if (avio_read(pb, buf, num_bytes) < num_bytes) {
1877 1878
        av_freep(&sc->sample_sizes);
        av_free(buf);
1879
        return AVERROR_INVALIDDATA;
1880 1881 1882 1883
    }

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

1884
    for (i = 0; i < entries && !pb->eof_reached; i++) {
1885
        sc->sample_sizes[i] = get_bits_long(&gb, field_size);
1886 1887
        sc->data_size += sc->sample_sizes[i];
    }
1888

1889 1890 1891 1892 1893
    sc->sample_count = i;

    if (pb->eof_reached)
        return AVERROR_EOF;

1894
    av_free(buf);
1895 1896 1897
    return 0;
}

1898
static int mov_read_stts(MOVContext *c, AVIOContext *pb, MOVAtom atom)
1899
{
1900 1901
    AVStream *st;
    MOVStreamContext *sc;
1902
    unsigned int i, entries;
1903 1904
    int64_t duration=0;
    int64_t total_sample_count=0;
1905

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

1911 1912 1913
    avio_r8(pb); /* version */
    avio_rb24(pb); /* flags */
    entries = avio_rb32(pb);
1914

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

A
Alex Converse 已提交
1918 1919 1920
    if (!entries)
        return 0;
    if (entries >= UINT_MAX / sizeof(*sc->stts_data))
L
Luca Barbato 已提交
1921
        return AVERROR(EINVAL);
1922

1923
    av_free(sc->stts_data);
1924
    sc->stts_data = av_malloc(entries * sizeof(*sc->stts_data));
1925
    if (!sc->stts_data)
1926
        return AVERROR(ENOMEM);
1927

1928
    for (i = 0; i < entries && !pb->eof_reached; i++) {
M
cleanup  
Michael Niedermayer 已提交
1929 1930
        int sample_duration;
        int sample_count;
1931

1932 1933
        sample_count=avio_rb32(pb);
        sample_duration = avio_rb32(pb);
1934 1935 1936 1937
        if (sample_count < 0) {
            av_log(c->fc, AV_LOG_ERROR, "Invalid sample_count=%d\n", sample_count);
            return AVERROR_INVALIDDATA;
        }
1938 1939 1940
        sc->stts_data[i].count= sample_count;
        sc->stts_data[i].duration= sample_duration;

1941 1942
        av_dlog(c->fc, "sample_count=%d, sample_duration=%d\n",
                sample_count, sample_duration);
1943

B
Baptiste Coudurier 已提交
1944
        duration+=(int64_t)sample_duration*sample_count;
1945 1946 1947
        total_sample_count+=sample_count;
    }

1948 1949 1950 1951 1952
    sc->stts_count = i;

    if (pb->eof_reached)
        return AVERROR_EOF;

1953
    st->nb_frames= total_sample_count;
1954
    if (duration)
1955
        st->duration= duration;
1956
    sc->track_end = duration;
1957 1958 1959
    return 0;
}

1960
static int mov_read_ctts(MOVContext *c, AVIOContext *pb, MOVAtom atom)
M
Michael Niedermayer 已提交
1961
{
1962 1963
    AVStream *st;
    MOVStreamContext *sc;
M
Michael Niedermayer 已提交
1964 1965
    unsigned int i, entries;

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

1971 1972 1973
    avio_r8(pb); /* version */
    avio_rb24(pb); /* flags */
    entries = avio_rb32(pb);
1974

L
Luca Barbato 已提交
1975
    av_dlog(c->fc, "track[%i].ctts.entries = %i\n", c->fc->nb_streams-1, entries);
1976

A
Alex Converse 已提交
1977 1978
    if (!entries)
        return 0;
1979
    if (entries >= UINT_MAX / sizeof(*sc->ctts_data))
1980
        return AVERROR_INVALIDDATA;
1981
    sc->ctts_data = av_malloc(entries * sizeof(*sc->ctts_data));
1982
    if (!sc->ctts_data)
1983
        return AVERROR(ENOMEM);
1984

1985
    for (i = 0; i < entries && !pb->eof_reached; i++) {
1986 1987
        int count    =avio_rb32(pb);
        int duration =avio_rb32(pb);
1988 1989 1990

        sc->ctts_data[i].count   = count;
        sc->ctts_data[i].duration= duration;
1991 1992
        if (duration < 0)
            sc->dts_shift = FFMAX(sc->dts_shift, -duration);
M
Michael Niedermayer 已提交
1993
    }
1994

1995 1996 1997 1998 1999
    sc->ctts_count = i;

    if (pb->eof_reached)
        return AVERROR_EOF;

L
Luca Barbato 已提交
2000
    av_dlog(c->fc, "dts shift %d\n", sc->dts_shift);
2001

M
Michael Niedermayer 已提交
2002 2003 2004
    return 0;
}

2005 2006 2007 2008 2009 2010 2011 2012 2013 2014 2015 2016 2017 2018 2019 2020 2021 2022 2023 2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044
static int mov_read_sbgp(MOVContext *c, AVIOContext *pb, MOVAtom atom)
{
    AVStream *st;
    MOVStreamContext *sc;
    unsigned int i, entries;
    uint8_t version;
    uint32_t grouping_type;

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

    version = avio_r8(pb); /* version */
    avio_rb24(pb); /* flags */
    grouping_type = avio_rl32(pb);
    if (grouping_type != MKTAG( 'r','a','p',' '))
        return 0; /* only support 'rap ' grouping */
    if (version == 1)
        avio_rb32(pb); /* grouping_type_parameter */

    entries = avio_rb32(pb);
    if (!entries)
        return 0;
    if (entries >= UINT_MAX / sizeof(*sc->rap_group))
        return AVERROR_INVALIDDATA;
    sc->rap_group = av_malloc(entries * sizeof(*sc->rap_group));
    if (!sc->rap_group)
        return AVERROR(ENOMEM);

    for (i = 0; i < entries && !pb->eof_reached; i++) {
        sc->rap_group[i].count = avio_rb32(pb); /* sample_count */
        sc->rap_group[i].index = avio_rb32(pb); /* group_description_index */
    }

    sc->rap_group_count = i;

    return pb->eof_reached ? AVERROR_EOF : 0;
}

2045 2046 2047
static void mov_build_index(MOVContext *mov, AVStream *st)
{
    MOVStreamContext *sc = st->priv_data;
2048
    int64_t current_offset;
2049 2050 2051 2052
    int64_t current_dts = 0;
    unsigned int stts_index = 0;
    unsigned int stsc_index = 0;
    unsigned int stss_index = 0;
2053
    unsigned int stps_index = 0;
2054
    unsigned int i, j;
2055
    uint64_t stream_size = 0;
2056

2057
    /* adjust first dts according to edit list */
2058
    if (sc->time_offset && mov->time_scale > 0) {
2059 2060 2061
        if (sc->time_offset < 0)
            sc->time_offset = av_rescale(sc->time_offset, sc->time_scale, mov->time_scale);
        current_dts = -sc->time_offset;
2062
        if (sc->ctts_data && sc->stts_data && sc->stts_data[0].duration &&
2063
            sc->ctts_data[0].duration / sc->stts_data[0].duration > 16) {
2064 2065 2066 2067 2068
            /* 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;
        }
2069 2070
    }

2071
    /* only use old uncompressed audio chunk demuxing when stts specifies it */
2072
    if (!(st->codec->codec_type == AVMEDIA_TYPE_AUDIO &&
2073
          sc->stts_count == 1 && sc->stts_data[0].duration == 1)) {
2074 2075
        unsigned int current_sample = 0;
        unsigned int stts_sample = 0;
2076
        unsigned int sample_size;
2077
        unsigned int distance = 0;
2078 2079 2080
        unsigned int rap_group_index = 0;
        unsigned int rap_group_sample = 0;
        int rap_group_present = sc->rap_group_count && sc->rap_group;
2081
        int key_off = (sc->keyframes && sc->keyframes[0] > 0) || (sc->stps_data && sc->stps_data[0] > 0);
2082

2083 2084
        current_dts -= sc->dts_shift;

A
Alex Converse 已提交
2085 2086
        if (!sc->sample_count)
            return;
2087
        if (sc->sample_count >= UINT_MAX / sizeof(*st->index_entries) - st->nb_index_entries)
2088
            return;
2089 2090 2091 2092
        if (av_reallocp_array(&st->index_entries,
                              st->nb_index_entries + sc->sample_count,
                              sizeof(*st->index_entries)) < 0) {
            st->nb_index_entries = 0;
2093
            return;
2094
        }
2095
        st->index_entries_allocated_size = (st->nb_index_entries + sc->sample_count) * sizeof(*st->index_entries);
2096

2097 2098
        for (i = 0; i < sc->chunk_count; i++) {
            current_offset = sc->chunk_offsets[i];
2099
            while (stsc_index + 1 < sc->stsc_count &&
2100
                i + 1 == sc->stsc_data[stsc_index + 1].first)
2101
                stsc_index++;
2102
            for (j = 0; j < sc->stsc_data[stsc_index].count; j++) {
2103
                int keyframe = 0;
2104 2105
                if (current_sample >= sc->sample_count) {
                    av_log(mov->fc, AV_LOG_ERROR, "wrong sample count\n");
2106
                    return;
2107
                }
2108

2109
                if (!sc->keyframe_absent && (!sc->keyframe_count || current_sample+key_off == sc->keyframes[stss_index])) {
2110
                    keyframe = 1;
2111 2112
                    if (stss_index + 1 < sc->keyframe_count)
                        stss_index++;
2113 2114 2115 2116
                } 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++;
2117
                }
2118 2119 2120 2121 2122 2123 2124 2125
                if (rap_group_present && rap_group_index < sc->rap_group_count) {
                    if (sc->rap_group[rap_group_index].index > 0)
                        keyframe = 1;
                    if (++rap_group_sample == sc->rap_group[rap_group_index].count) {
                        rap_group_sample = 0;
                        rap_group_index++;
                    }
                }
2126 2127
                if (keyframe)
                    distance = 0;
2128
                sample_size = sc->sample_size > 0 ? sc->sample_size : sc->sample_sizes[current_sample];
2129
                if (sc->pseudo_stream_id == -1 ||
2130
                   sc->stsc_data[stsc_index].id - 1 == sc->pseudo_stream_id) {
2131 2132 2133 2134 2135 2136
                    AVIndexEntry *e = &st->index_entries[st->nb_index_entries++];
                    e->pos = current_offset;
                    e->timestamp = current_dts;
                    e->size = sample_size;
                    e->min_distance = distance;
                    e->flags = keyframe ? AVINDEX_KEYFRAME : 0;
L
Luca Barbato 已提交
2137
                    av_dlog(mov->fc, "AVIndex stream %d, sample %d, offset %"PRIx64", dts %"PRId64", "
2138 2139 2140
                            "size %d, distance %d, keyframe %d\n", st->index, current_sample,
                            current_offset, current_dts, sample_size, distance, keyframe);
                }
2141

2142
                current_offset += sample_size;
2143
                stream_size += sample_size;
2144
                current_dts += sc->stts_data[stts_index].duration;
2145 2146 2147 2148 2149 2150 2151 2152 2153
                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++;
                }
            }
        }
2154 2155
        if (st->duration > 0)
            st->codec->bit_rate = stream_size*8*sc->time_scale/st->duration;
2156
    } else {
2157
        unsigned chunk_samples, total = 0;
2158

2159 2160 2161
        // compute total chunk count
        for (i = 0; i < sc->stsc_count; i++) {
            unsigned count, chunk_count;
2162

2163
            chunk_samples = sc->stsc_data[i].count;
2164 2165
            if (i != sc->stsc_count - 1 &&
                sc->samples_per_frame && chunk_samples % sc->samples_per_frame) {
2166 2167 2168 2169
                av_log(mov->fc, AV_LOG_ERROR, "error unaligned chunk\n");
                return;
            }

2170 2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185
            if (sc->samples_per_frame >= 160) { // gsm
                count = chunk_samples / sc->samples_per_frame;
            } else if (sc->samples_per_frame > 1) {
                unsigned samples = (1024/sc->samples_per_frame)*sc->samples_per_frame;
                count = (chunk_samples+samples-1) / samples;
            } else {
                count = (chunk_samples+1023) / 1024;
            }

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

L
Luca Barbato 已提交
2186
        av_dlog(mov->fc, "chunk count %d\n", total);
2187
        if (total >= UINT_MAX / sizeof(*st->index_entries) - st->nb_index_entries)
2188
            return;
2189 2190 2191 2192
        if (av_reallocp_array(&st->index_entries,
                              st->nb_index_entries + total,
                              sizeof(*st->index_entries)) < 0) {
            st->nb_index_entries = 0;
2193
            return;
2194
        }
2195
        st->index_entries_allocated_size = (st->nb_index_entries + total) * sizeof(*st->index_entries);
2196 2197 2198 2199 2200 2201 2202 2203 2204

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

2205
            while (chunk_samples > 0) {
2206
                AVIndexEntry *e;
2207 2208 2209 2210 2211 2212 2213 2214 2215 2216
                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;
2217
                    } else {
2218 2219
                        samples = FFMIN(1024, chunk_samples);
                        size = samples * sc->sample_size;
2220 2221
                    }
                }
2222

2223 2224 2225 2226 2227 2228 2229 2230 2231 2232
                if (st->nb_index_entries >= total) {
                    av_log(mov->fc, AV_LOG_ERROR, "wrong chunk count %d\n", total);
                    return;
                }
                e = &st->index_entries[st->nb_index_entries++];
                e->pos = current_offset;
                e->timestamp = current_dts;
                e->size = size;
                e->min_distance = 0;
                e->flags = AVINDEX_KEYFRAME;
L
Luca Barbato 已提交
2233
                av_dlog(mov->fc, "AVIndex stream %d, chunk %d, offset %"PRIx64", dts %"PRId64", "
2234
                        "size %d, duration %d\n", st->index, i, current_offset, current_dts,
2235 2236 2237
                        size, samples);

                current_offset += size;
2238
                current_dts += samples;
2239
                chunk_samples -= samples;
2240 2241 2242 2243
            }
        }
    }
}
2244

2245 2246
static int mov_open_dref(AVIOContext **pb, char *src, MOVDref *ref,
                         AVIOInterruptCB *int_cb)
2247
{
2248 2249
    /* try relative path, we do not try the absolute because it can leak information about our
       system to an attacker */
2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271
    if (ref->nlvl_to > 0 && ref->nlvl_from > 0) {
        char filename[1024];
        char *src_path;
        int i, l;

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

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

        /* compose filename if next level down to target was found */
2272
        if (i == ref->nlvl_to - 1 && src_path - src  < sizeof(filename)) {
2273 2274 2275 2276 2277 2278 2279 2280
            memcpy(filename, src, src_path - src);
            filename[src_path - src] = 0;

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

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

2281
            if (!avio_open2(pb, filename, AVIO_FLAG_READ, int_cb, NULL))
2282 2283 2284 2285 2286
                return 0;
        }
    }

    return AVERROR(ENOENT);
M
Mans Rullgard 已提交
2287
}
2288

2289
static int mov_read_trak(MOVContext *c, AVIOContext *pb, MOVAtom atom)
2290 2291 2292
{
    AVStream *st;
    MOVStreamContext *sc;
2293
    int ret;
2294

2295
    st = avformat_new_stream(c->fc, NULL);
B
Baptiste Coudurier 已提交
2296
    if (!st) return AVERROR(ENOMEM);
2297
    st->id = c->fc->nb_streams;
B
Baptiste Coudurier 已提交
2298
    sc = av_mallocz(sizeof(MOVStreamContext));
2299
    if (!sc) return AVERROR(ENOMEM);
2300 2301

    st->priv_data = sc;
2302
    st->codec->codec_type = AVMEDIA_TYPE_DATA;
2303
    sc->ffindex = st->index;
2304

2305 2306 2307 2308
    if ((ret = mov_read_default(c, pb, atom)) < 0)
        return ret;

    /* sanity checks */
2309 2310
    if (sc->chunk_count && (!sc->stts_count || !sc->stsc_count ||
                            (!sc->sample_size && !sc->sample_count))) {
2311 2312
        av_log(c->fc, AV_LOG_ERROR, "stream %d, missing mandatory atoms, broken header\n",
               st->index);
2313 2314
        return 0;
    }
2315

2316
    if (sc->time_scale <= 0) {
2317
        av_log(c->fc, AV_LOG_WARNING, "stream %d, timescale not set\n", st->index);
2318
        sc->time_scale = c->time_scale;
2319
        if (sc->time_scale <= 0)
2320 2321
            sc->time_scale = 1;
    }
2322

2323
    avpriv_set_pts_info(st, 64, 1, sc->time_scale);
2324 2325 2326 2327

    mov_build_index(c, st);

    if (sc->dref_id-1 < sc->drefs_count && sc->drefs[sc->dref_id-1].path) {
2328
        MOVDref *dref = &sc->drefs[sc->dref_id - 1];
2329
        if (mov_open_dref(&sc->pb, c->fc->filename, dref, &c->fc->interrupt_callback) < 0)
2330 2331 2332 2333 2334
            av_log(c->fc, AV_LOG_ERROR,
                   "stream %d, error opening alias: path='%s', dir='%s', "
                   "filename='%s', volume='%s', nlvl_from=%d, nlvl_to=%d\n",
                   st->index, dref->path, dref->dir, dref->filename,
                   dref->volume, dref->nlvl_from, dref->nlvl_to);
2335 2336 2337
    } else
        sc->pb = c->fc->pb;

2338
    if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
2339 2340 2341 2342
        if (!st->sample_aspect_ratio.num &&
            (st->codec->width != sc->width || st->codec->height != sc->height)) {
            st->sample_aspect_ratio = av_d2q(((double)st->codec->height * sc->width) /
                                             ((double)st->codec->width * sc->height), INT_MAX);
2343
        }
2344 2345
    }

2346 2347 2348 2349 2350 2351 2352 2353
    // done for ai5q, ai52, ai55, ai1q, ai12 and ai15.
    if (!st->codec->extradata_size && st->codec->codec_id == AV_CODEC_ID_H264 &&
        TAG_IS_AVCI(st->codec->codec_tag)) {
        ret = ff_generate_avci_extradata(st);
        if (ret < 0)
            return ret;
    }

2354
    switch (st->codec->codec_id) {
2355
#if CONFIG_H261_DECODER
2356
    case AV_CODEC_ID_H261:
2357
#endif
2358
#if CONFIG_H263_DECODER
2359
    case AV_CODEC_ID_H263:
2360
#endif
2361
#if CONFIG_MPEG4_DECODER
2362
    case AV_CODEC_ID_MPEG4:
2363
#endif
2364
        st->codec->width = 0; /* let decoder init width/height */
2365 2366 2367
        st->codec->height= 0;
        break;
    }
B
Baptiste Coudurier 已提交
2368 2369 2370

    /* Do not need those anymore. */
    av_freep(&sc->chunk_offsets);
2371
    av_freep(&sc->stsc_data);
B
Baptiste Coudurier 已提交
2372 2373 2374
    av_freep(&sc->sample_sizes);
    av_freep(&sc->keyframes);
    av_freep(&sc->stts_data);
2375
    av_freep(&sc->stps_data);
2376
    av_freep(&sc->rap_group);
B
Baptiste Coudurier 已提交
2377

2378
    return 0;
2379 2380
}

2381
static int mov_read_ilst(MOVContext *c, AVIOContext *pb, MOVAtom atom)
2382 2383 2384 2385 2386 2387 2388 2389
{
    int ret;
    c->itunes_metadata = 1;
    ret = mov_read_default(c, pb, atom);
    c->itunes_metadata = 0;
    return ret;
}

2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422 2423 2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471
static int mov_read_replaygain(MOVContext *c, AVIOContext *pb, int size)
{
    int64_t end = avio_tell(pb) + size;
    uint8_t *key = NULL, *val = NULL;
    int i;

    for (i = 0; i < 2; i++) {
        uint8_t **p;
        uint32_t len, tag;

        if (end - avio_tell(pb) <= 12)
            break;

        len = avio_rb32(pb);
        tag = avio_rl32(pb);
        avio_skip(pb, 4); // flags

        if (len < 12 || len - 12 > end - avio_tell(pb))
            break;
        len -= 12;

        if (tag == MKTAG('n', 'a', 'm', 'e'))
            p = &key;
        else if (tag == MKTAG('d', 'a', 't', 'a') && len > 4) {
            avio_skip(pb, 4);
            len -= 4;
            p = &val;
        } else
            break;

        *p = av_malloc(len + 1);
        if (!*p)
            break;
        avio_read(pb, *p, len);
        (*p)[len] = 0;
    }

    if (key && val) {
        av_dict_set(&c->fc->metadata, key, val,
                    AV_DICT_DONT_STRDUP_KEY | AV_DICT_DONT_STRDUP_VAL);
        key = val = NULL;
    }

    avio_seek(pb, end, SEEK_SET);
    av_freep(&key);
    av_freep(&val);
    return 0;
}

static int mov_read_custom(MOVContext *c, AVIOContext *pb, MOVAtom atom)
{
    int64_t end = avio_tell(pb) + atom.size;
    uint32_t tag, len;

    if (atom.size < 8)
        goto fail;

    len = avio_rb32(pb);
    tag = avio_rl32(pb);

    if (len > atom.size)
        goto fail;

    if (tag == MKTAG('m', 'e', 'a', 'n') && len > 12) {
        uint8_t domain[128];
        int domain_len;

        avio_skip(pb, 4); // flags
        len -= 12;

        domain_len = avio_get_str(pb, len, domain, sizeof(domain));
        avio_skip(pb, len - domain_len);
        if (!strcmp(domain, "org.hydrogenaudio.replaygain"))
            return mov_read_replaygain(c, pb, end - avio_tell(pb));
    }

fail:
    av_log(c->fc, AV_LOG_VERBOSE,
           "Unhandled or malformed custom metadata of size %"PRId64"\n", atom.size);
    return 0;
}

2472
static int mov_read_meta(MOVContext *c, AVIOContext *pb, MOVAtom atom)
2473
{
2474
    while (atom.size > 8) {
2475
        uint32_t tag = avio_rl32(pb);
2476 2477
        atom.size -= 4;
        if (tag == MKTAG('h','d','l','r')) {
A
Anton Khirnov 已提交
2478
            avio_seek(pb, -8, SEEK_CUR);
2479 2480 2481 2482 2483
            atom.size += 8;
            return mov_read_default(c, pb, atom);
        }
    }
    return 0;
2484 2485
}

2486
static int mov_read_tkhd(MOVContext *c, AVIOContext *pb, MOVAtom atom)
2487
{
2488 2489 2490 2491
    int i;
    int width;
    int height;
    int64_t disp_transform[2];
2492
    int display_matrix[3][3];
2493 2494 2495
    AVStream *st;
    MOVStreamContext *sc;
    int version;
2496
    int flags;
2497 2498 2499 2500 2501

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

2503
    version = avio_r8(pb);
2504 2505
    flags = avio_rb24(pb);
    st->disposition |= (flags & MOV_TKHD_FLAG_ENABLED) ? AV_DISPOSITION_DEFAULT : 0;
2506

B
Baptiste Coudurier 已提交
2507
    if (version == 1) {
2508 2509
        avio_rb64(pb);
        avio_rb64(pb);
B
Baptiste Coudurier 已提交
2510
    } else {
2511 2512
        avio_rb32(pb); /* creation time */
        avio_rb32(pb); /* modification time */
B
Baptiste Coudurier 已提交
2513
    }
2514 2515
    st->id = (int)avio_rb32(pb); /* track id (NOT 0 !)*/
    avio_rb32(pb); /* reserved */
2516

2517
    /* highlevel (considering edits) duration in movie timebase */
2518 2519 2520
    (version == 1) ? avio_rb64(pb) : avio_rb32(pb);
    avio_rb32(pb); /* reserved */
    avio_rb32(pb); /* reserved */
2521

2522 2523 2524 2525
    avio_rb16(pb); /* layer */
    avio_rb16(pb); /* alternate group */
    avio_rb16(pb); /* volume */
    avio_rb16(pb); /* reserved */
2526

2527 2528
    //read in the display matrix (outlined in ISO 14496-12, Section 6.2.2)
    // they're kept in fixed point format through all calculations
2529 2530
    // save u,v,z to store the whole matrix in the AV_PKT_DATA_DISPLAYMATRIX
    // side data, but the scale factor is not needed to calculate aspect ratio
2531
    for (i = 0; i < 3; i++) {
2532 2533
        display_matrix[i][0] = avio_rb32(pb);   // 16.16 fixed point
        display_matrix[i][1] = avio_rb32(pb);   // 16.16 fixed point
2534
        display_matrix[i][2] = avio_rb32(pb);   //  2.30 fixed point
2535
    }
2536

2537 2538
    width = avio_rb32(pb);       // 16.16 fixed point track width
    height = avio_rb32(pb);      // 16.16 fixed point track height
2539 2540
    sc->width = width >> 16;
    sc->height = height >> 16;
2541

2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560
    // save the matrix when it is not the default identity
    if (display_matrix[0][0] != (1 << 16) ||
        display_matrix[1][1] != (1 << 16) ||
        display_matrix[2][2] != (1 << 30) ||
        display_matrix[0][1] || display_matrix[0][2] ||
        display_matrix[1][0] || display_matrix[1][2] ||
        display_matrix[2][0] || display_matrix[2][1]) {
        int i, j;

        av_freep(&sc->display_matrix);
        sc->display_matrix = av_malloc(sizeof(int32_t) * 9);
        if (!sc->display_matrix)
            return AVERROR(ENOMEM);

        for (i = 0; i < 3; i++)
            for (j = 0; j < 3; j++)
                sc->display_matrix[i * 3 + j] = display_matrix[j][i];
    }

2561
    // transform the display width/height according to the matrix
2562
    // skip this if the display matrix is the default identity matrix
2563
    // or if it is rotating the picture, ex iPhone 3GS
2564 2565
    // to keep the same scale, use [width height 1<<16]
    if (width && height &&
2566 2567 2568 2569 2570
        ((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])) {
2571 2572 2573 2574 2575 2576 2577
        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
2578
        st->sample_aspect_ratio = av_d2q(
2579 2580 2581
            ((double) disp_transform[0] * height) /
            ((double) disp_transform[1] * width), INT_MAX);
    }
2582 2583 2584
    return 0;
}

2585
static int mov_read_tfhd(MOVContext *c, AVIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
2586 2587 2588 2589 2590
{
    MOVFragment *frag = &c->fragment;
    MOVTrackExt *trex = NULL;
    int flags, track_id, i;

2591 2592
    avio_r8(pb); /* version */
    flags = avio_rb24(pb);
B
Baptiste Coudurier 已提交
2593

2594
    track_id = avio_rb32(pb);
2595
    if (!track_id)
2596
        return AVERROR_INVALIDDATA;
B
Baptiste Coudurier 已提交
2597 2598 2599 2600 2601 2602 2603 2604
    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");
2605
        return AVERROR_INVALIDDATA;
B
Baptiste Coudurier 已提交
2606 2607
    }

2608
    frag->base_data_offset = flags & MOV_TFHD_BASE_DATA_OFFSET ?
2609 2610
                             avio_rb64(pb) : flags & MOV_TFHD_DEFAULT_BASE_IS_MOOF ?
                             frag->moof_offset : frag->implicit_offset;
2611 2612 2613 2614 2615 2616 2617 2618
    frag->stsd_id  = flags & MOV_TFHD_STSD_ID ? avio_rb32(pb) : trex->stsd_id;

    frag->duration = flags & MOV_TFHD_DEFAULT_DURATION ?
                     avio_rb32(pb) : trex->duration;
    frag->size     = flags & MOV_TFHD_DEFAULT_SIZE ?
                     avio_rb32(pb) : trex->size;
    frag->flags    = flags & MOV_TFHD_DEFAULT_FLAGS ?
                     avio_rb32(pb) : trex->flags;
L
Luca Barbato 已提交
2619
    av_dlog(c->fc, "frag flags 0x%x\n", frag->flags);
B
Baptiste Coudurier 已提交
2620 2621 2622
    return 0;
}

2623
static int mov_read_chap(MOVContext *c, AVIOContext *pb, MOVAtom atom)
D
David Conrad 已提交
2624
{
2625
    c->chapter_track = avio_rb32(pb);
D
David Conrad 已提交
2626 2627 2628
    return 0;
}

2629
static int mov_read_trex(MOVContext *c, AVIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
2630 2631
{
    MOVTrackExt *trex;
2632
    int err;
B
Baptiste Coudurier 已提交
2633 2634

    if ((uint64_t)c->trex_count+1 >= UINT_MAX / sizeof(*c->trex_data))
2635
        return AVERROR_INVALIDDATA;
2636 2637 2638 2639 2640
    if ((err = av_reallocp_array(&c->trex_data, c->trex_count + 1,
                                 sizeof(*c->trex_data))) < 0) {
        c->trex_count = 0;
        return err;
    }
B
Baptiste Coudurier 已提交
2641
    trex = &c->trex_data[c->trex_count++];
2642 2643 2644 2645 2646 2647 2648
    avio_r8(pb); /* version */
    avio_rb24(pb); /* flags */
    trex->track_id = avio_rb32(pb);
    trex->stsd_id  = avio_rb32(pb);
    trex->duration = avio_rb32(pb);
    trex->size     = avio_rb32(pb);
    trex->flags    = avio_rb32(pb);
B
Baptiste Coudurier 已提交
2649 2650 2651
    return 0;
}

M
Martin Storsjö 已提交
2652 2653 2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669 2670 2671 2672 2673 2674 2675 2676 2677 2678 2679 2680 2681
static int mov_read_tfdt(MOVContext *c, AVIOContext *pb, MOVAtom atom)
{
    MOVFragment *frag = &c->fragment;
    AVStream *st = NULL;
    MOVStreamContext *sc;
    int version, i;

    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);
        return AVERROR_INVALIDDATA;
    }
    sc = st->priv_data;
    if (sc->pseudo_stream_id + 1 != frag->stsd_id)
        return 0;
    version = avio_r8(pb);
    avio_rb24(pb); /* flags */
    if (version) {
        sc->track_end = avio_rb64(pb);
    } else {
        sc->track_end = avio_rb32(pb);
    }
    return 0;
}

2682
static int mov_read_trun(MOVContext *c, AVIOContext *pb, MOVAtom atom)
B
Baptiste Coudurier 已提交
2683 2684
{
    MOVFragment *frag = &c->fragment;
2685
    AVStream *st = NULL;
2686
    MOVStreamContext *sc;
2687
    MOVStts *ctts_data;
B
Baptiste Coudurier 已提交
2688 2689 2690 2691
    uint64_t offset;
    int64_t dts;
    int data_offset = 0;
    unsigned entries, first_sample_flags = frag->flags;
2692
    int flags, distance, i, found_keyframe = 0, err;
B
Baptiste Coudurier 已提交
2693

2694 2695 2696 2697 2698 2699 2700 2701
    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);
2702
        return AVERROR_INVALIDDATA;
2703
    }
2704
    sc = st->priv_data;
B
Baptiste Coudurier 已提交
2705 2706
    if (sc->pseudo_stream_id+1 != frag->stsd_id)
        return 0;
2707 2708 2709
    avio_r8(pb); /* version */
    flags = avio_rb24(pb);
    entries = avio_rb32(pb);
L
Luca Barbato 已提交
2710
    av_dlog(c->fc, "flags 0x%x entries %d\n", flags, entries);
2711 2712 2713 2714 2715 2716 2717 2718 2719

    /* Always assume the presence of composition time offsets.
     * Without this assumption, for instance, we cannot deal with a track in fragmented movies that meet the following.
     *  1) in the initial movie, there are no samples.
     *  2) in the first movie fragment, there is only one sample without composition time offset.
     *  3) in the subsequent movie fragments, there are samples with composition time offset. */
    if (!sc->ctts_count && sc->sample_count)
    {
        /* Complement ctts table if moov atom doesn't have ctts atom. */
2720
        ctts_data = av_realloc(NULL, sizeof(*sc->ctts_data));
2721
        if (!ctts_data)
B
Baptiste Coudurier 已提交
2722
            return AVERROR(ENOMEM);
2723
        sc->ctts_data = ctts_data;
2724 2725 2726
        sc->ctts_data[sc->ctts_count].count = sc->sample_count;
        sc->ctts_data[sc->ctts_count].duration = 0;
        sc->ctts_count++;
B
Baptiste Coudurier 已提交
2727
    }
2728
    if ((uint64_t)entries+sc->ctts_count >= UINT_MAX/sizeof(*sc->ctts_data))
2729
        return AVERROR_INVALIDDATA;
2730 2731 2732 2733 2734
    if ((err = av_reallocp_array(&sc->ctts_data, entries + sc->ctts_count,
                                 sizeof(*sc->ctts_data))) < 0) {
        sc->ctts_count = 0;
        return err;
    }
2735 2736
    if (flags & MOV_TRUN_DATA_OFFSET)        data_offset        = avio_rb32(pb);
    if (flags & MOV_TRUN_FIRST_SAMPLE_FLAGS) first_sample_flags = avio_rb32(pb);
2737
    dts    = sc->track_end - sc->time_offset;
B
Baptiste Coudurier 已提交
2738 2739
    offset = frag->base_data_offset + data_offset;
    distance = 0;
L
Luca Barbato 已提交
2740
    av_dlog(c->fc, "first sample flags 0x%x\n", first_sample_flags);
2741
    for (i = 0; i < entries && !pb->eof_reached; i++) {
B
Baptiste Coudurier 已提交
2742 2743 2744
        unsigned sample_size = frag->size;
        int sample_flags = i ? frag->flags : first_sample_flags;
        unsigned sample_duration = frag->duration;
2745
        int keyframe = 0;
B
Baptiste Coudurier 已提交
2746

2747 2748 2749
        if (flags & MOV_TRUN_SAMPLE_DURATION) sample_duration = avio_rb32(pb);
        if (flags & MOV_TRUN_SAMPLE_SIZE)     sample_size     = avio_rb32(pb);
        if (flags & MOV_TRUN_SAMPLE_FLAGS)    sample_flags    = avio_rb32(pb);
2750
        sc->ctts_data[sc->ctts_count].count = 1;
2751 2752
        sc->ctts_data[sc->ctts_count].duration = (flags & MOV_TRUN_SAMPLE_CTS) ?
                                                  avio_rb32(pb) : 0;
2753
        sc->ctts_count++;
2754 2755 2756 2757 2758 2759 2760
        if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO)
            keyframe = 1;
        else if (!found_keyframe)
            keyframe = found_keyframe =
                !(sample_flags & (MOV_FRAG_SAMPLE_FLAG_IS_NON_SYNC |
                                  MOV_FRAG_SAMPLE_FLAG_DEPENDS_YES));
        if (keyframe)
B
Baptiste Coudurier 已提交
2761 2762 2763
            distance = 0;
        av_add_index_entry(st, offset, dts, sample_size, distance,
                           keyframe ? AVINDEX_KEYFRAME : 0);
L
Luca Barbato 已提交
2764
        av_dlog(c->fc, "AVIndex stream %d, sample %d, offset %"PRIx64", dts %"PRId64", "
B
Baptiste Coudurier 已提交
2765 2766 2767
                "size %d, distance %d, keyframe %d\n", st->index, sc->sample_count+i,
                offset, dts, sample_size, distance, keyframe);
        distance++;
2768
        dts += sample_duration;
B
Baptiste Coudurier 已提交
2769
        offset += sample_size;
2770
        sc->data_size += sample_size;
B
Baptiste Coudurier 已提交
2771
    }
2772 2773 2774 2775

    if (pb->eof_reached)
        return AVERROR_EOF;

2776
    frag->implicit_offset = offset;
2777
    st->duration = sc->track_end = dts + sc->time_offset;
B
Baptiste Coudurier 已提交
2778 2779 2780
    return 0;
}

2781 2782 2783
/* 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/ */
2784
static int mov_read_wide(MOVContext *c, AVIOContext *pb, MOVAtom atom)
2785 2786 2787 2788 2789
{
    int err;

    if (atom.size < 8)
        return 0; /* continue */
2790
    if (avio_rb32(pb) != 0) { /* 0 sized mdat atom... use the 'wide' atom size */
2791
        avio_skip(pb, atom.size - 4);
2792 2793
        return 0;
    }
2794
    atom.type = avio_rl32(pb);
2795
    atom.size -= 8;
2796
    if (atom.type != MKTAG('m','d','a','t')) {
2797
        avio_skip(pb, atom.size);
2798 2799 2800 2801 2802 2803
        return 0;
    }
    err = mov_read_mdat(c, pb, atom);
    return err;
}

2804
static int mov_read_cmov(MOVContext *c, AVIOContext *pb, MOVAtom atom)
2805
{
2806
#if CONFIG_ZLIB
2807
    AVIOContext ctx;
2808 2809
    uint8_t *cmov_data;
    uint8_t *moov_data; /* uncompressed data */
2810
    long cmov_len, moov_len;
2811
    int ret = -1;
2812

2813 2814
    avio_rb32(pb); /* dcom atom */
    if (avio_rl32(pb) != MKTAG('d','c','o','m'))
2815
        return AVERROR_INVALIDDATA;
2816
    if (avio_rl32(pb) != MKTAG('z','l','i','b')) {
B
Benoit Fouet 已提交
2817
        av_log(c->fc, AV_LOG_ERROR, "unknown compression for cmov atom !");
2818
        return AVERROR_INVALIDDATA;
2819
    }
2820 2821
    avio_rb32(pb); /* cmvd atom */
    if (avio_rl32(pb) != MKTAG('c','m','v','d'))
2822
        return AVERROR_INVALIDDATA;
2823
    moov_len = avio_rb32(pb); /* uncompressed size */
2824
    cmov_len = atom.size - 6 * 4;
2825

B
Baptiste Coudurier 已提交
2826
    cmov_data = av_malloc(cmov_len);
2827
    if (!cmov_data)
2828
        return AVERROR(ENOMEM);
B
Baptiste Coudurier 已提交
2829
    moov_data = av_malloc(moov_len);
2830 2831
    if (!moov_data) {
        av_free(cmov_data);
2832
        return AVERROR(ENOMEM);
2833
    }
2834
    avio_read(pb, cmov_data, cmov_len);
2835
    if (uncompress (moov_data, (uLongf *) &moov_len, (const Bytef *)cmov_data, cmov_len) != Z_OK)
2836
        goto free_and_return;
2837
    if (ffio_init_context(&ctx, moov_data, moov_len, 0, NULL, NULL, NULL, NULL) != 0)
2838
        goto free_and_return;
2839
    atom.type = MKTAG('m','o','o','v');
2840 2841
    atom.size = moov_len;
    ret = mov_read_default(c, &ctx, atom);
2842
free_and_return:
2843 2844 2845
    av_free(moov_data);
    av_free(cmov_data);
    return ret;
2846 2847
#else
    av_log(c->fc, AV_LOG_ERROR, "this file requires zlib support compiled in\n");
2848
    return AVERROR(ENOSYS);
2849
#endif
2850
}
2851

G
Gael Chardon 已提交
2852
/* edit list atom */
2853
static int mov_read_elst(MOVContext *c, AVIOContext *pb, MOVAtom atom)
G
Gael Chardon 已提交
2854
{
2855
    MOVStreamContext *sc;
2856
    int i, edit_count, version;
B
Baptiste Coudurier 已提交
2857

2858 2859 2860 2861
    if (c->fc->nb_streams < 1)
        return 0;
    sc = c->fc->streams[c->fc->nb_streams-1]->priv_data;

2862
    version = avio_r8(pb); /* version */
2863 2864
    avio_rb24(pb); /* flags */
    edit_count = avio_rb32(pb); /* entries */
B
Baptiste Coudurier 已提交
2865

2866
    if ((uint64_t)edit_count*12+8 > atom.size)
2867
        return AVERROR_INVALIDDATA;
2868

2869
    for (i=0; i<edit_count; i++){
2870 2871 2872 2873 2874 2875 2876
        int64_t time;
        int64_t duration;
        if (version == 1) {
            duration = avio_rb64(pb);
            time     = avio_rb64(pb);
        } else {
            duration = avio_rb32(pb); /* segment duration */
Y
Yusuke Nakamura 已提交
2877
            time     = (int32_t)avio_rb32(pb); /* media time */
2878
        }
2879
        avio_rb32(pb); /* Media rate */
2880 2881
        if (i == 0 && time >= -1) {
            sc->time_offset = time != -1 ? time : -duration;
2882
        }
B
Baptiste Coudurier 已提交
2883
    }
2884

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

L
Luca Barbato 已提交
2889
    av_dlog(c->fc, "track[%i].edit_count = %i\n", c->fc->nb_streams-1, edit_count);
B
Baptiste Coudurier 已提交
2890
    return 0;
G
Gael Chardon 已提交
2891 2892
}

2893
static const MOVParseTableEntry mov_default_parse_table[] = {
2894
{ MKTAG('a','v','s','s'), mov_read_extradata },
D
David Conrad 已提交
2895
{ MKTAG('c','h','p','l'), mov_read_chpl },
2896
{ MKTAG('c','o','6','4'), mov_read_stco },
2897
{ MKTAG('c','o','l','r'), mov_read_colr },
2898 2899 2900 2901 2902 2903
{ 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 },
2904
{ MKTAG('f','i','e','l'), mov_read_fiel },
2905 2906 2907
{ MKTAG('f','t','y','p'), mov_read_ftyp },
{ MKTAG('g','l','b','l'), mov_read_glbl },
{ MKTAG('h','d','l','r'), mov_read_hdlr },
2908
{ MKTAG('i','l','s','t'), mov_read_ilst },
2909 2910 2911 2912
{ 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 },
2913
{ MKTAG('m','e','t','a'), mov_read_meta },
2914 2915 2916 2917 2918 2919 2920 2921
{ 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 },
2922
{ MKTAG('p','a','s','p'), mov_read_pasp },
2923 2924
{ MKTAG('s','t','b','l'), mov_read_default },
{ MKTAG('s','t','c','o'), mov_read_stco },
2925
{ MKTAG('s','t','p','s'), mov_read_stps },
M
Martin Storsjö 已提交
2926
{ MKTAG('s','t','r','f'), mov_read_strf },
2927 2928 2929 2930 2931
{ 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 },
2932
{ MKTAG('s','t','z','2'), mov_read_stsz }, /* compact sample size */
2933
{ MKTAG('t','k','h','d'), mov_read_tkhd }, /* track header */
M
Martin Storsjö 已提交
2934
{ MKTAG('t','f','d','t'), mov_read_tfdt },
2935 2936 2937
{ MKTAG('t','f','h','d'), mov_read_tfhd }, /* track fragment header */
{ MKTAG('t','r','a','k'), mov_read_trak },
{ MKTAG('t','r','a','f'), mov_read_default },
D
David Conrad 已提交
2938 2939
{ MKTAG('t','r','e','f'), mov_read_default },
{ MKTAG('c','h','a','p'), mov_read_chap },
2940 2941
{ MKTAG('t','r','e','x'), mov_read_trex },
{ MKTAG('t','r','u','n'), mov_read_trun },
B
Baptiste Coudurier 已提交
2942
{ MKTAG('u','d','t','a'), mov_read_default },
2943 2944
{ MKTAG('w','a','v','e'), mov_read_wave },
{ MKTAG('e','s','d','s'), mov_read_esds },
2945
{ MKTAG('d','a','c','3'), mov_read_dac3 }, /* AC-3 info */
2946
{ MKTAG('d','e','c','3'), mov_read_dec3 }, /* EAC-3 info */
2947
{ MKTAG('w','i','d','e'), mov_read_wide }, /* place holder */
2948
{ MKTAG('w','f','e','x'), mov_read_wfex },
2949
{ MKTAG('c','m','o','v'), mov_read_cmov },
2950
{ MKTAG('c','h','a','n'), mov_read_chan }, /* channel layout */
M
Martin Storsjö 已提交
2951
{ MKTAG('d','v','c','1'), mov_read_dvc1 },
2952
{ MKTAG('s','b','g','p'), mov_read_sbgp },
Y
Yusuke Nakamura 已提交
2953
{ MKTAG('h','v','c','C'), mov_read_glbl },
2954
{ MKTAG('-','-','-','-'), mov_read_custom },
B
Baptiste Coudurier 已提交
2955
{ 0, NULL }
2956 2957
};

2958 2959 2960 2961 2962 2963 2964 2965 2966 2967 2968 2969 2970 2971 2972 2973 2974 2975 2976 2977 2978 2979 2980 2981 2982 2983 2984 2985 2986 2987 2988 2989 2990 2991 2992 2993 2994 2995 2996 2997 2998 2999 3000 3001 3002 3003 3004 3005 3006 3007 3008 3009 3010 3011 3012 3013 3014 3015 3016 3017 3018 3019
static int mov_read_default(MOVContext *c, AVIOContext *pb, MOVAtom atom)
{
    int64_t total_size = 0;
    MOVAtom a;
    int i;

    if (atom.size < 0)
        atom.size = INT64_MAX;
    while (total_size + 8 < atom.size && !pb->eof_reached) {
        int (*parse)(MOVContext*, AVIOContext*, MOVAtom) = NULL;
        a.size = atom.size;
        a.type=0;
        if (atom.size >= 8) {
            a.size = avio_rb32(pb);
            a.type = avio_rl32(pb);
        }
        av_dlog(c->fc, "type: %08x '%.4s' parent:'%.4s' sz: %"PRId64" %"PRId64" %"PRId64"\n",
                a.type, (char*)&a.type, (char*)&atom.type, a.size, total_size, atom.size);
        total_size += 8;
        if (a.size == 1) { /* 64 bit extended size */
            a.size = avio_rb64(pb) - 8;
            total_size += 8;
        }
        if (a.size == 0) {
            a.size = atom.size - total_size;
            if (a.size <= 8)
                break;
        }
        a.size -= 8;
        if (a.size < 0)
            break;
        a.size = FFMIN(a.size, atom.size - total_size);

        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;
            }

        // 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 */
            avio_skip(pb, a.size);
        } else {
            int64_t start_pos = avio_tell(pb);
            int64_t left;
            int err = parse(c, pb, a);
            if (err < 0)
                return err;
            if (c->found_moov && c->found_mdat &&
                ((!pb->seekable || c->fc->flags & AVFMT_FLAG_IGNIDX) ||
                 start_pos + a.size == avio_size(pb))) {
                if (!pb->seekable || c->fc->flags & AVFMT_FLAG_IGNIDX)
                    c->next_root_atom = start_pos + a.size;
                return 0;
            }
            left = a.size - avio_tell(pb) + start_pos;
            if (left > 0) /* skip garbage at atom end */
                avio_skip(pb, left);
3020 3021 3022 3023 3024 3025
            else if (left < 0) {
                av_log(c->fc, AV_LOG_WARNING,
                       "overread end of atom '%.4s' by %"PRId64" bytes\n",
                       (char*)&a.type, -left);
                avio_seek(pb, left, SEEK_CUR);
            }
3026 3027 3028 3029 3030 3031 3032 3033 3034 3035 3036
        }

        total_size += a.size;
    }

    if (total_size < atom.size && atom.size < 0x7ffff)
        avio_skip(pb, atom.size - total_size);

    return 0;
}

F
Fabrice Bellard 已提交
3037 3038
static int mov_probe(AVProbeData *p)
{
3039 3040
    unsigned int offset;
    uint32_t tag;
3041
    int score = 0;
3042

F
Fabrice Bellard 已提交
3043
    /* check file header */
3044
    offset = 0;
3045
    for (;;) {
3046 3047
        /* ignore invalid offset */
        if ((offset + 8) > (unsigned int)p->buf_size)
3048
            return score;
3049
        tag = AV_RL32(p->buf + offset + 4);
3050
        switch(tag) {
3051
        /* check for obvious tags */
3052 3053 3054 3055 3056
        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 */
3057
        case MKTAG('f','t','y','p'):
3058
            return AVPROBE_SCORE_MAX;
3059
        /* those are more common words, so rate then a bit less */
3060 3061 3062 3063 3064
        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'):
3065
            return AVPROBE_SCORE_MAX - 5;
B
Baptiste Coudurier 已提交
3066
        case MKTAG(0x82,0x82,0x7f,0x7d):
3067 3068 3069
        case MKTAG('s','k','i','p'):
        case MKTAG('u','u','i','d'):
        case MKTAG('p','r','f','l'):
3070
            offset = AV_RB32(p->buf+offset) + offset;
3071
            /* if we only find those cause probedata is too small at least rate them */
3072
            score = AVPROBE_SCORE_EXTENSION;
3073 3074 3075
            break;
        default:
            /* unrecognized tag */
3076
            return score;
3077
        }
3078
    }
F
Fabrice Bellard 已提交
3079 3080
}

D
David Conrad 已提交
3081 3082 3083 3084 3085 3086 3087
// must be done after parsing all trak because there's no order requirement
static void mov_read_chapters(AVFormatContext *s)
{
    MOVContext *mov = s->priv_data;
    AVStream *st = NULL;
    MOVStreamContext *sc;
    int64_t cur_pos;
3088
    int i;
D
David Conrad 已提交
3089 3090 3091 3092 3093 3094 3095 3096 3097 3098 3099 3100 3101

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

    st->discard = AVDISCARD_ALL;
    sc = st->priv_data;
3102
    cur_pos = avio_tell(sc->pb);
D
David Conrad 已提交
3103 3104 3105 3106

    for (i = 0; i < st->nb_index_entries; i++) {
        AVIndexEntry *sample = &st->index_entries[i];
        int64_t end = i+1 < st->nb_index_entries ? st->index_entries[i+1].timestamp : st->duration;
3107 3108 3109
        uint8_t *title;
        uint16_t ch;
        int len, title_len;
D
David Conrad 已提交
3110

A
Anton Khirnov 已提交
3111
        if (avio_seek(sc->pb, sample->pos, SEEK_SET) != sample->pos) {
D
David Conrad 已提交
3112 3113 3114 3115 3116
            av_log(s, AV_LOG_ERROR, "Chapter %d not found in file\n", i);
            goto finish;
        }

        // the first two bytes are the length of the title
3117
        len = avio_rb16(sc->pb);
D
David Conrad 已提交
3118 3119
        if (len > sample->size-2)
            continue;
3120 3121 3122
        title_len = 2*len + 1;
        if (!(title = av_mallocz(title_len)))
            goto finish;
D
David Conrad 已提交
3123 3124 3125 3126

        // The samples could theoretically be in any encoding if there's an encd
        // atom following, but in practice are only utf-8 or utf-16, distinguished
        // instead by the presence of a BOM
3127 3128 3129 3130 3131 3132 3133 3134 3135 3136 3137
        if (!len) {
            title[0] = 0;
        } else {
            ch = avio_rb16(sc->pb);
            if (ch == 0xfeff)
                avio_get_str16be(sc->pb, len, title, title_len);
            else if (ch == 0xfffe)
                avio_get_str16le(sc->pb, len, title, title_len);
            else {
                AV_WB16(title, ch);
                if (len == 1 || len == 2)
3138
                    title[len] = 0;
3139 3140 3141
                else
                    avio_get_str(sc->pb, len - 2, title + 2, title_len - 2);
            }
D
David Conrad 已提交
3142 3143
        }

3144
        avpriv_new_chapter(s, i, st->time_base, sample->timestamp, end, title);
3145
        av_freep(&title);
D
David Conrad 已提交
3146 3147
    }
finish:
A
Anton Khirnov 已提交
3148
    avio_seek(sc->pb, cur_pos, SEEK_SET);
D
David Conrad 已提交
3149 3150
}

3151 3152 3153 3154 3155 3156 3157 3158 3159 3160 3161 3162 3163 3164 3165 3166 3167
static int mov_read_close(AVFormatContext *s)
{
    MOVContext *mov = s->priv_data;
    int i, j;

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

        av_freep(&sc->ctts_data);
        for (j = 0; j < sc->drefs_count; j++) {
            av_freep(&sc->drefs[j].path);
            av_freep(&sc->drefs[j].dir);
        }
        av_freep(&sc->drefs);
        if (sc->pb && sc->pb != s->pb)
            avio_close(sc->pb);
3168 3169 3170 3171 3172 3173 3174 3175

        av_freep(&sc->chunk_offsets);
        av_freep(&sc->stsc_data);
        av_freep(&sc->sample_sizes);
        av_freep(&sc->keyframes);
        av_freep(&sc->stts_data);
        av_freep(&sc->stps_data);
        av_freep(&sc->rap_group);
3176
        av_freep(&sc->display_matrix);
3177 3178 3179
    }

    if (mov->dv_demux) {
3180 3181
        avformat_free_context(mov->dv_fctx);
        mov->dv_fctx = NULL;
3182 3183 3184 3185 3186 3187 3188
    }

    av_freep(&mov->trex_data);

    return 0;
}

3189
static int mov_read_header(AVFormatContext *s)
3190
{
3191
    MOVContext *mov = s->priv_data;
3192
    AVIOContext *pb = s->pb;
B
Baptiste Coudurier 已提交
3193
    int err;
3194
    MOVAtom atom = { AV_RL32("root") };
3195
    int i;
3196 3197

    mov->fc = s;
3198
    /* .mov and .mp4 aren't streamable anyway (only progressive download if moov is before mdat) */
3199
    if (pb->seekable)
A
Anton Khirnov 已提交
3200
        atom.size = avio_size(pb);
3201
    else
B
Baptiste Coudurier 已提交
3202
        atom.size = INT64_MAX;
3203 3204

    /* check MOV header */
B
Baptiste Coudurier 已提交
3205 3206
    if ((err = mov_read_default(mov, pb, atom)) < 0) {
        av_log(s, AV_LOG_ERROR, "error reading header: %d\n", err);
3207
        mov_read_close(s);
B
Baptiste Coudurier 已提交
3208 3209 3210 3211
        return err;
    }
    if (!mov->found_moov) {
        av_log(s, AV_LOG_ERROR, "moov atom not found\n");
3212
        mov_read_close(s);
3213
        return AVERROR_INVALIDDATA;
3214
    }
3215
    av_dlog(mov->fc, "on_parse_exit_offset=%"PRId64"\n", avio_tell(pb));
3216

3217
    if (pb->seekable && mov->chapter_track > 0)
D
David Conrad 已提交
3218 3219
        mov_read_chapters(s);

3220 3221 3222 3223 3224
    for (i = 0; i < s->nb_streams; i++) {
        AVStream *st = s->streams[i];
        MOVStreamContext *sc = st->priv_data;

        if (st->codec->codec_type == AVMEDIA_TYPE_SUBTITLE) {
V
Vittorio Giovara 已提交
3225
            if (st->codec->width <= 0 || st->codec->height <= 0) {
3226 3227 3228
                st->codec->width  = sc->width;
                st->codec->height = sc->height;
            }
3229 3230 3231 3232
            if (st->codec->codec_id == AV_CODEC_ID_DVD_SUBTITLE) {
                if ((err = mov_rewrite_dvd_sub_extradata(st)) < 0)
                    return err;
            }
3233 3234 3235
        }
    }

3236 3237 3238 3239
    if (mov->trex_data) {
        for (i = 0; i < s->nb_streams; i++) {
            AVStream *st = s->streams[i];
            MOVStreamContext *sc = st->priv_data;
3240
            if (st->duration > 0)
3241 3242 3243 3244
                st->codec->bit_rate = sc->data_size * 8 * sc->time_scale / st->duration;
        }
    }

3245 3246
    for (i = 0; i < s->nb_streams; i++) {
        AVStream *st = s->streams[i];
3247
        MOVStreamContext *sc = st->priv_data;
3248

3249 3250 3251 3252 3253 3254 3255 3256 3257 3258 3259 3260 3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271 3272 3273 3274 3275
        switch (st->codec->codec_type) {
        case AVMEDIA_TYPE_AUDIO:
            err = ff_replaygain_export(st, s->metadata);
            if (err < 0) {
                mov_read_close(s);
                return err;
            }
            break;
        case AVMEDIA_TYPE_VIDEO:
            if (sc->display_matrix) {
                AVPacketSideData *sd, *tmp;

                tmp = av_realloc_array(st->side_data,
                                       st->nb_side_data + 1, sizeof(*tmp));
                if (!tmp)
                    return AVERROR(ENOMEM);

                st->side_data = tmp;
                st->nb_side_data++;

                sd = &st->side_data[st->nb_side_data - 1];
                sd->type = AV_PKT_DATA_DISPLAYMATRIX;
                sd->size = sizeof(int32_t) * 9;
                sd->data = (uint8_t*)sc->display_matrix;
                sc->display_matrix = NULL;
            }
            break;
3276 3277 3278
        }
    }

3279 3280 3281
    return 0;
}

3282
static AVIndexEntry *mov_find_next_sample(AVFormatContext *s, AVStream **st)
3283
{
3284
    AVIndexEntry *sample = NULL;
3285
    int64_t best_dts = INT64_MAX;
3286
    int i;
B
Baptiste Coudurier 已提交
3287
    for (i = 0; i < s->nb_streams; i++) {
3288 3289
        AVStream *avst = s->streams[i];
        MOVStreamContext *msc = avst->priv_data;
3290
        if (msc->pb && msc->current_sample < avst->nb_index_entries) {
3291
            AVIndexEntry *current_sample = &avst->index_entries[msc->current_sample];
3292
            int64_t dts = av_rescale(current_sample->timestamp, AV_TIME_BASE, msc->time_scale);
L
Luca Barbato 已提交
3293
            av_dlog(s, "stream %d, sample %d, dts %"PRId64"\n", i, msc->current_sample, dts);
3294 3295
            if (!sample || (!s->pb->seekable && current_sample->pos < sample->pos) ||
                (s->pb->seekable &&
3296
                 ((msc->pb != s->pb && dts < best_dts) || (msc->pb == s->pb &&
B
Baptiste Coudurier 已提交
3297
                 ((FFABS(best_dts - dts) <= AV_TIME_BASE && current_sample->pos < sample->pos) ||
3298
                  (FFABS(best_dts - dts) > AV_TIME_BASE && dts < best_dts)))))) {
3299 3300
                sample = current_sample;
                best_dts = dts;
3301
                *st = avst;
3302
            }
3303 3304
        }
    }
3305 3306 3307 3308 3309 3310 3311 3312 3313 3314 3315 3316
    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);
3317 3318
    if (!sample) {
        mov->found_mdat = 0;
3319 3320 3321 3322 3323
        if (!mov->next_root_atom)
            return AVERROR_EOF;
        avio_seek(s->pb, mov->next_root_atom, SEEK_SET);
        mov->next_root_atom = 0;
        if (mov_read_default(mov, s->pb, (MOVAtom){ AV_RL32("root"), INT64_MAX }) < 0 ||
A
Anton Khirnov 已提交
3324
            s->pb->eof_reached)
B
Baptiste Coudurier 已提交
3325
            return AVERROR_EOF;
3326
        av_dlog(s, "read fragments, offset 0x%"PRIx64"\n", avio_tell(s->pb));
3327 3328
        goto retry;
    }
3329
    sc = st->priv_data;
3330 3331
    /* must be done just before reading, to avoid infinite loop on sample */
    sc->current_sample++;
3332 3333

    if (st->discard != AVDISCARD_ALL) {
A
Anton Khirnov 已提交
3334
        if (avio_seek(sc->pb, sample->pos, SEEK_SET) != sample->pos) {
R
Reimar Döffinger 已提交
3335 3336
            av_log(mov->fc, AV_LOG_ERROR, "stream %d, offset 0x%"PRIx64": partial file\n",
                   sc->ffindex, sample->pos);
3337
            return AVERROR_INVALIDDATA;
R
Reimar Döffinger 已提交
3338 3339
        }
        ret = av_get_packet(sc->pb, pkt, sample->size);
3340 3341
        if (ret < 0)
            return ret;
3342 3343 3344 3345 3346 3347 3348 3349 3350 3351 3352
        if (sc->has_palette) {
            uint8_t *pal;

            pal = av_packet_new_side_data(pkt, AV_PKT_DATA_PALETTE, AVPALETTE_SIZE);
            if (!pal) {
                av_log(mov->fc, AV_LOG_ERROR, "Cannot append palette to packet\n");
            } else {
                memcpy(pal, sc->palette, AVPALETTE_SIZE);
                sc->has_palette = 0;
            }
        }
R
Reimar Döffinger 已提交
3353 3354
#if CONFIG_DV_DEMUXER
        if (mov->dv_demux && sc->dv_audio_container) {
3355
            avpriv_dv_produce_packet(mov->dv_demux, pkt, pkt->data, pkt->size);
R
Reimar Döffinger 已提交
3356 3357
            av_free(pkt->data);
            pkt->size = 0;
3358
            ret = avpriv_dv_get_packet(mov->dv_demux, pkt);
R
Reimar Döffinger 已提交
3359 3360 3361
            if (ret < 0)
                return ret;
        }
3362
#endif
3363 3364
    }

3365 3366
    pkt->stream_index = sc->ffindex;
    pkt->dts = sample->timestamp;
3367
    if (sc->ctts_data && sc->ctts_index < sc->ctts_count) {
3368
        pkt->pts = pkt->dts + sc->dts_shift + sc->ctts_data[sc->ctts_index].duration;
3369
        /* update ctts context */
3370 3371 3372 3373 3374
        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;
3375
        }
3376 3377
        if (sc->wrong_dts)
            pkt->dts = AV_NOPTS_VALUE;
3378
    } else {
3379
        int64_t next_dts = (sc->current_sample < st->nb_index_entries) ?
3380 3381
            st->index_entries[sc->current_sample].timestamp : st->duration;
        pkt->duration = next_dts - pkt->dts;
3382
        pkt->pts = pkt->dts;
3383
    }
3384 3385
    if (st->discard == AVDISCARD_ALL)
        goto retry;
3386
    pkt->flags |= sample->flags & AVINDEX_KEYFRAME ? AV_PKT_FLAG_KEY : 0;
3387
    pkt->pos = sample->pos;
L
Luca Barbato 已提交
3388
    av_dlog(s, "stream %d, pts %"PRId64", dts %"PRId64", pos 0x%"PRIx64", duration %d\n",
3389
            pkt->stream_index, pkt->pts, pkt->dts, pkt->pos, pkt->duration);
3390 3391
    return 0;
}
3392

3393
static int mov_seek_stream(AVFormatContext *s, AVStream *st, int64_t timestamp, int flags)
3394 3395 3396 3397
{
    MOVStreamContext *sc = st->priv_data;
    int sample, time_sample;
    int i;
3398

3399
    sample = av_index_search_timestamp(st, timestamp, flags);
L
Luca Barbato 已提交
3400
    av_dlog(s, "stream %d, timestamp %"PRId64", sample %d\n", st->index, timestamp, sample);
3401 3402
    if (sample < 0 && st->nb_index_entries && timestamp < st->index_entries[0].timestamp)
        sample = 0;
3403
    if (sample < 0) /* not sure what to do */
3404
        return AVERROR_INVALIDDATA;
3405
    sc->current_sample = sample;
L
Luca Barbato 已提交
3406
    av_dlog(s, "stream %d, found sample %d\n", st->index, sc->current_sample);
3407 3408 3409 3410
    /* adjust ctts index */
    if (sc->ctts_data) {
        time_sample = 0;
        for (i = 0; i < sc->ctts_count; i++) {
3411 3412
            int next = time_sample + sc->ctts_data[i].count;
            if (next > sc->current_sample) {
3413 3414
                sc->ctts_index = i;
                sc->ctts_sample = sc->current_sample - time_sample;
3415
                break;
3416
            }
3417
            time_sample = next;
3418 3419
        }
    }
3420
    return sample;
3421 3422
}

3423
static int mov_read_seek(AVFormatContext *s, int stream_index, int64_t sample_time, int flags)
G
Gael Chardon 已提交
3424
{
3425 3426 3427 3428
    AVStream *st;
    int64_t seek_timestamp, timestamp;
    int sample;
    int i;
G
Gael Chardon 已提交
3429

3430
    if (stream_index >= s->nb_streams)
3431
        return AVERROR_INVALIDDATA;
3432 3433
    if (sample_time < 0)
        sample_time = 0;
G
Gael Chardon 已提交
3434

3435
    st = s->streams[stream_index];
3436
    sample = mov_seek_stream(s, st, sample_time, flags);
3437
    if (sample < 0)
3438
        return sample;
G
Gael Chardon 已提交
3439

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

3443 3444
    for (i = 0; i < s->nb_streams; i++) {
        st = s->streams[i];
3445
        if (stream_index == i)
3446
            continue;
G
Gael Chardon 已提交
3447

3448
        timestamp = av_rescale_q(seek_timestamp, s->streams[stream_index]->time_base, st->time_base);
3449
        mov_seek_stream(s, st, timestamp, flags);
3450
    }
G
Gael Chardon 已提交
3451 3452 3453
    return 0;
}

3454 3455 3456 3457 3458
#define OFFSET(x) offsetof(MOVContext, x)
#define FLAGS AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_DECODING_PARAM
static const AVOption mov_options[] = {
    { "export_all", "Export unrecognized metadata entries", OFFSET(export_all),
        AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, .flags = FLAGS },
3459 3460
    { "export_xmp", "Export full XMP metadata", OFFSET(export_xmp),
        AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 1, .flags = FLAGS },
3461 3462 3463 3464 3465 3466 3467 3468 3469 3470
    { NULL },
};

static const AVClass mov_class = {
    .class_name = "mov,mp4,m4a,3gp,3g2,mj2",
    .item_name  = av_default_item_name,
    .option     = mov_options,
    .version    = LIBAVUTIL_VERSION_INT,
};

3471
AVInputFormat ff_mov_demuxer = {
3472
    .name           = "mov,mp4,m4a,3gp,3g2,mj2",
3473
    .long_name      = NULL_IF_CONFIG_SMALL("QuickTime / MOV"),
3474
    .priv_class     = &mov_class,
3475
    .priv_data_size = sizeof(MOVContext),
3476
    .extensions     = "mov,mp4,m4a,3gp,3g2,mj2",
3477 3478 3479 3480 3481
    .read_probe     = mov_probe,
    .read_header    = mov_read_header,
    .read_packet    = mov_read_packet,
    .read_close     = mov_read_close,
    .read_seek      = mov_read_seek,
3482
};