wtv.c 41.4 KB
Newer Older
Peter Ross's avatar
Peter Ross committed
1 2
/*
 * Windows Television (WTV) demuxer
Peter Ross's avatar
Peter Ross committed
3
 * Copyright (c) 2010-2011 Peter Ross <pross@xvid.org>
Peter Ross's avatar
Peter Ross committed
4
 *
5
 * This file is part of Libav.
Peter Ross's avatar
Peter Ross committed
6
 *
7
 * Libav is free software; you can redistribute it and/or
Peter Ross's avatar
Peter Ross committed
8 9 10 11
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
12
 * Libav is distributed in the hope that it will be useful,
Peter Ross's avatar
Peter Ross committed
13 14 15 16 17
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
18
 * License along with Libav; if not, write to the Free Software
Peter Ross's avatar
Peter Ross committed
19 20 21 22 23 24 25 26 27 28
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

/**
 * @file
 * Windows Television (WTV) demuxer
 * @author Peter Ross <pross@xvid.org>
 */

#include "libavutil/intreadwrite.h"
29
#include "libavutil/intfloat.h"
30
#include "libavutil/dict.h"
Peter Ross's avatar
Peter Ross committed
31
#include "avformat.h"
32
#include "internal.h"
Peter Ross's avatar
Peter Ross committed
33 34
#include "riff.h"
#include "asf.h"
35
#include "mpegts.h"
Peter Ross's avatar
Peter Ross committed
36 37 38 39 40 41 42

/* Macros for formating GUIDs */
#define PRI_GUID \
    "%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x%02x"
#define ARG_GUID(g) \
    g[0],g[1],g[2],g[3],g[4],g[5],g[6],g[7],g[8],g[9],g[10],g[11],g[12],g[13],g[14],g[15]

Peter Ross's avatar
Peter Ross committed
43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58
#define PRI_PRETTY_GUID \
    "%08x-%04x-%04x-%02x%02x%02x%02x%02x%02x%02x%02x"
#define ARG_PRETTY_GUID(g) \
    AV_RL32(g),AV_RL16(g+4),AV_RL16(g+6),g[8],g[9],g[10],g[11],g[12],g[13],g[14],g[15]
#define LEN_PRETTY_GUID 34

/*
 *
 * File system routines
 *
 */

#define WTV_SECTOR_BITS    12
#define WTV_SECTOR_SIZE    (1 << WTV_SECTOR_BITS)
#define WTV_BIGSECTOR_BITS 18

59
typedef struct {
60
    AVIOContext *pb_filesystem;  /** file system (AVFormatContext->pb) */
61

Peter Ross's avatar
Peter Ross committed
62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
    int sector_bits;     /** sector shift bits; used to convert sector number into pb_filesystem offset */
    uint32_t *sectors;   /** file allocation table */
    int nb_sectors;      /** number of sectors */

    int error;
    int64_t position;
    int64_t length;
} WtvFile;

/**
 * @return bytes read, 0 on end of file, or <0 on error
 */
static int wtvfile_read_packet(void *opaque, uint8_t *buf, int buf_size)
{
    WtvFile *wf = opaque;
77
    AVIOContext *pb = wf->pb_filesystem;
Peter Ross's avatar
Peter Ross committed
78 79
    int nread = 0;

Anton Khirnov's avatar
Anton Khirnov committed
80
    if (wf->error || pb->error)
Peter Ross's avatar
Peter Ross committed
81
        return -1;
Anton Khirnov's avatar
Anton Khirnov committed
82
    if (wf->position >= wf->length || pb->eof_reached)
Peter Ross's avatar
Peter Ross committed
83 84 85 86 87 88 89 90
        return 0;

    buf_size = FFMIN(buf_size, wf->length - wf->position);
    while(nread < buf_size) {
        int n;
        int remaining_in_sector = (1 << wf->sector_bits) - (wf->position & ((1 << wf->sector_bits) - 1));
        int read_request        = FFMIN(buf_size - nread, remaining_in_sector);

91
        n = avio_read(pb, buf, read_request);
Peter Ross's avatar
Peter Ross committed
92 93 94 95 96 97 98 99 100
        if (n <= 0)
            break;
        nread += n;
        buf += n;
        wf->position += n;
        if (n == remaining_in_sector) {
            int i = wf->position >> wf->sector_bits;
            if (i >= wf->nb_sectors ||
                (wf->sectors[i] != wf->sectors[i - 1] + (1 << (wf->sector_bits - WTV_SECTOR_BITS)) &&
101
                avio_seek(pb, (int64_t)wf->sectors[i] << WTV_SECTOR_BITS, SEEK_SET) < 0)) {
Peter Ross's avatar
Peter Ross committed
102 103 104 105 106 107 108 109 110 111 112 113 114 115
                wf->error = 1;
                break;
            }
        }
    }
    return nread;
}

/**
 * @return position (or file length)
 */
static int64_t wtvfile_seek(void *opaque, int64_t offset, int whence)
{
    WtvFile *wf = opaque;
116
    AVIOContext *pb = wf->pb_filesystem;
Peter Ross's avatar
Peter Ross committed
117 118 119 120 121 122 123 124 125

    if (whence == AVSEEK_SIZE)
        return wf->length;
    else if (whence == SEEK_CUR)
        offset = wf->position + offset;
    else if (whence == SEEK_END)
        offset = wf->length;

    wf->error = offset < 0 || offset >= wf->length ||
126
                avio_seek(pb, ((int64_t)wf->sectors[offset >> wf->sector_bits] << WTV_SECTOR_BITS)
Peter Ross's avatar
Peter Ross committed
127 128 129 130 131 132 133 134 135 136 137 138
                              + (offset & ((1 << wf->sector_bits) - 1)), SEEK_SET) < 0;
    wf->position = offset;
    return offset;
}

/**
 * read non-zero integers (le32) from input stream
 * @param pb
 * @param[out] data destination
 * @param     count maximum number of integers to read
 * @return    total number of integers read
 */
139
static int read_ints(AVIOContext *pb, uint32_t *data, int count)
Peter Ross's avatar
Peter Ross committed
140 141 142
{
    int i, total = 0;
    for (i = 0; i < count; i++) {
143
        if ((data[total] = avio_rl32(pb)))
Peter Ross's avatar
Peter Ross committed
144 145 146 147 148 149 150 151 152 153 154 155
           total++;
    }
    return total;
}

/**
 * Open file
 * @param first_sector  First sector
 * @param length        Length of file (bytes)
 * @param depth         File allocation table depth
 * @return NULL on error
 */
156
static AVIOContext * wtvfile_open_sector(int first_sector, uint64_t length, int depth, AVFormatContext *s)
Peter Ross's avatar
Peter Ross committed
157
{
158
    AVIOContext *pb;
Peter Ross's avatar
Peter Ross committed
159 160 161
    WtvFile *wf;
    uint8_t *buffer;

162
    if (avio_seek(s->pb, first_sector << WTV_SECTOR_BITS, SEEK_SET) < 0)
Peter Ross's avatar
Peter Ross committed
163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197
        return NULL;

    wf = av_mallocz(sizeof(WtvFile));
    if (!wf)
        return NULL;

    if (depth == 0) {
        wf->sectors = av_malloc(sizeof(uint32_t));
        if (!wf->sectors) {
            av_free(wf);
            return NULL;
        }
        wf->sectors[0]  = first_sector;
        wf->nb_sectors  = 1;
        wf->sector_bits = WTV_SECTOR_BITS;
    } else if (depth == 1) {
        wf->sectors = av_malloc(WTV_SECTOR_SIZE);
        if (!wf->sectors) {
            av_free(wf);
            return NULL;
        }
        wf->nb_sectors  = read_ints(s->pb, wf->sectors, WTV_SECTOR_SIZE / 4);
        wf->sector_bits = length & (1ULL<<63) ? WTV_SECTOR_BITS : WTV_BIGSECTOR_BITS;
    } else if (depth == 2) {
        uint32_t sectors1[WTV_SECTOR_SIZE / 4];
        int nb_sectors1 = read_ints(s->pb, sectors1, WTV_SECTOR_SIZE / 4);
        int i;

        wf->sectors = av_malloc(nb_sectors1 << WTV_SECTOR_BITS);
        if (!wf->sectors) {
            av_free(wf);
            return NULL;
        }
        wf->nb_sectors = 0;
        for (i = 0; i < nb_sectors1; i++) {
198
            if (avio_seek(s->pb, (int64_t)sectors1[i] << WTV_SECTOR_BITS, SEEK_SET) < 0)
Peter Ross's avatar
Peter Ross committed
199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224
                break;
            wf->nb_sectors += read_ints(s->pb, wf->sectors + i * WTV_SECTOR_SIZE / 4, WTV_SECTOR_SIZE / 4);
        }
        wf->sector_bits = length & (1ULL<<63) ? WTV_SECTOR_BITS : WTV_BIGSECTOR_BITS;
    } else {
        av_log(s, AV_LOG_ERROR, "unsupported file allocation table depth (0x%x)\n", depth);
        av_free(wf);
        return NULL;
    }

    if (!wf->nb_sectors) {
        av_free(wf->sectors);
        av_free(wf);
        return NULL;
    }

    /* check length */
    length &= 0xFFFFFFFFFFFF;
    if (length > ((int64_t)wf->nb_sectors << wf->sector_bits)) {
        av_log(s, AV_LOG_WARNING, "reported file length (0x%"PRIx64") exceeds number of available sectors (0x%"PRIx64")\n", length, (int64_t)wf->nb_sectors << wf->sector_bits);
        length = (int64_t)wf->nb_sectors <<  wf->sector_bits;
    }
    wf->length = length;

    /* seek to intial sector */
    wf->position = 0;
225
    if (avio_seek(s->pb, (int64_t)wf->sectors[0] << WTV_SECTOR_BITS, SEEK_SET) < 0) {
Peter Ross's avatar
Peter Ross committed
226 227 228 229 230 231 232 233 234 235 236 237
        av_free(wf->sectors);
        av_free(wf);
        return NULL;
    }

    wf->pb_filesystem = s->pb;
    buffer = av_malloc(1 << wf->sector_bits);
    if (!buffer) {
        av_free(wf->sectors);
        av_free(wf);
        return NULL;
    }
Peter Ross's avatar
Peter Ross committed
238

239
    pb = avio_alloc_context(buffer, 1 << wf->sector_bits, 0, wf,
Peter Ross's avatar
Peter Ross committed
240 241 242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258 259
                           wtvfile_read_packet, NULL, wtvfile_seek);
    if (!pb) {
        av_free(buffer);
        av_free(wf->sectors);
        av_free(wf);
    }
    return pb;
}

static const ff_asf_guid dir_entry_guid =
    {0x92,0xB7,0x74,0x91,0x59,0x70,0x70,0x44,0x88,0xDF,0x06,0x3B,0x82,0xCC,0x21,0x3D};

/**
 * Open file using filename
 * @param[in]  buf       directory buffer
 * @param      buf_size  directory buffer size
 * @param[in]  filename
 * @param      filename_size size of filename
 * @return NULL on error
 */
260
static AVIOContext * wtvfile_open2(AVFormatContext *s, const uint8_t *buf, int buf_size, const uint8_t *filename, int filename_size)
Peter Ross's avatar
Peter Ross committed
261
{
Peter Ross's avatar
Peter Ross committed
262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292
    const uint8_t *buf_end = buf + buf_size;

    while(buf + 48 <= buf_end) {
        int dir_length, name_size, first_sector, depth;
        uint64_t file_length;
        const uint8_t *name;
        if (ff_guidcmp(buf, dir_entry_guid)) {
            av_log(s, AV_LOG_ERROR, "unknown guid "PRI_GUID", expected dir_entry_guid; "
                   "remaining directory entries ignored\n", ARG_GUID(buf));
            break;
        }
        dir_length  = AV_RL16(buf + 16);
        file_length = AV_RL64(buf + 24);
        name_size   = 2 * AV_RL32(buf + 32);
        if (buf + 48 + name_size > buf_end) {
            av_log(s, AV_LOG_ERROR, "filename exceeds buffer size; remaining directory entries ignored\n");
            break;
        }
        first_sector = AV_RL32(buf + 40 + name_size);
        depth        = AV_RL32(buf + 44 + name_size);

        /* compare file name; test optional null terminator */
        name = buf + 40;
        if (name_size >= filename_size &&
            !memcmp(name, filename, filename_size) &&
            (name_size < filename_size + 2 || !AV_RN16(name + filename_size)))
            return wtvfile_open_sector(first_sector, file_length, depth, s);

        buf += dir_length;
    }
    return 0;
Peter Ross's avatar
Peter Ross committed
293 294
}

Peter Ross's avatar
Peter Ross committed
295 296 297 298 299 300
#define wtvfile_open(s, buf, buf_size, filename) \
    wtvfile_open2(s, buf, buf_size, filename, sizeof(filename))

/**
 * Close file opened with wtvfile_open_sector(), or wtv_open()
 */
301
static void wtvfile_close(AVIOContext *pb)
Peter Ross's avatar
Peter Ross committed
302 303 304
{
    WtvFile *wf = pb->opaque;
    av_free(wf->sectors);
305 306
    av_free(wf);
    av_free(pb->buffer);
Peter Ross's avatar
Peter Ross committed
307 308 309 310 311 312 313 314 315 316 317 318 319 320
    av_free(pb);
}

/*
 *
 * Main demuxer
 *
 */

typedef struct {
    int seen_data;
} WtvStream;

typedef struct {
321
    AVIOContext *pb;       /** timeline file */
Peter Ross's avatar
Peter Ross committed
322 323 324 325 326 327 328 329 330 331 332
    int64_t epoch;
    int64_t pts;             /** pts for next data chunk */
    int64_t last_valid_pts;  /** latest valid pts, used for interative seeking */

    /* maintain private seek index, as the AVIndexEntry->pos is relative to the
       start of the 'timeline' file, not the file system (AVFormatContext->pb) */
    AVIndexEntry *index_entries;
    int nb_index_entries;
    unsigned int index_entries_allocated_size;
} WtvContext;

Peter Ross's avatar
Peter Ross committed
333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350
typedef struct {
    enum CodecID id;
    ff_asf_guid guid;
} AVCodecGuid;

static enum CodecID ff_codec_guid_get_id(const AVCodecGuid *guids, ff_asf_guid guid)
{
    int i;
    for (i = 0; guids[i].id != CODEC_ID_NONE; i++) {
        if (!ff_guidcmp(guids[i].guid, guid))
            return guids[i].id;
    }
    return CODEC_ID_NONE;
}

/* WTV GUIDs */
static const ff_asf_guid wtv_guid =
    {0xB7,0xD8,0x00,0x20,0x37,0x49,0xDA,0x11,0xA6,0x4E,0x00,0x07,0xE9,0x5E,0xAD,0x8D};
Peter Ross's avatar
Peter Ross committed
351
static const ff_asf_guid metadata_guid =
Peter Ross's avatar
Peter Ross committed
352 353 354 355 356 357 358
    {0x5A,0xFE,0xD7,0x6D,0xC8,0x1D,0x8F,0x4A,0x99,0x22,0xFA,0xB1,0x1C,0x38,0x14,0x53};
static const ff_asf_guid timestamp_guid =
    {0x5B,0x05,0xE6,0x1B,0x97,0xA9,0x49,0x43,0x88,0x17,0x1A,0x65,0x5A,0x29,0x8A,0x97};
static const ff_asf_guid data_guid =
    {0x95,0xC3,0xD2,0xC2,0x7E,0x9A,0xDA,0x11,0x8B,0xF7,0x00,0x07,0xE9,0x5E,0xAD,0x8D};
static const ff_asf_guid stream_guid =
    {0xED,0xA4,0x13,0x23,0x2D,0xBF,0x4F,0x45,0xAD,0x8A,0xD9,0x5B,0xA7,0xF9,0x1F,0xEE};
359 360
static const ff_asf_guid stream2_guid =
    {0xA2,0xC3,0xD2,0xC2,0x7E,0x9A,0xDA,0x11,0x8B,0xF7,0x00,0x07,0xE9,0x5E,0xAD,0x8D};
361
static const ff_asf_guid EVENTID_SubtitleSpanningEvent =
Peter Ross's avatar
Peter Ross committed
362
    {0x48,0xC0,0xCE,0x5D,0xB9,0xD0,0x63,0x41,0x87,0x2C,0x4F,0x32,0x22,0x3B,0xE8,0x8A};
363
static const ff_asf_guid EVENTID_LanguageSpanningEvent =
Peter Ross's avatar
Peter Ross committed
364
    {0x6D,0x66,0x92,0xE2,0x02,0x9C,0x8D,0x44,0xAA,0x8D,0x78,0x1A,0x93,0xFD,0xC3,0x95};
365 366 367 368 369 370
static const ff_asf_guid EVENTID_AudioDescriptorSpanningEvent =
    {0x1C,0xD4,0x7B,0x10,0xDA,0xA6,0x91,0x46,0x83,0x69,0x11,0xB2,0xCD,0xAA,0x28,0x8E};
static const ff_asf_guid EVENTID_CtxADescriptorSpanningEvent =
    {0xE6,0xA2,0xB4,0x3A,0x47,0x42,0x34,0x4B,0x89,0x6C,0x30,0xAF,0xA5,0xD2,0x1C,0x24};
static const ff_asf_guid EVENTID_CSDescriptorSpanningEvent =
    {0xD9,0x79,0xE7,0xEf,0xF0,0x97,0x86,0x47,0x80,0x0D,0x95,0xCF,0x50,0x5D,0xDC,0x66};
371 372
static const ff_asf_guid EVENTID_DVBScramblingControlSpanningEvent =
    {0xC4,0xE1,0xD4,0x4B,0xA1,0x90,0x09,0x41,0x82,0x36,0x27,0xF0,0x0E,0x7D,0xCC,0x5B};
373 374 375 376
static const ff_asf_guid EVENTID_StreamIDSpanningEvent =
    {0x68,0xAB,0xF1,0xCA,0x53,0xE1,0x41,0x4D,0xA6,0xB3,0xA7,0xC9,0x98,0xDB,0x75,0xEE};
static const ff_asf_guid EVENTID_TeletextSpanningEvent =
    {0x50,0xD9,0x99,0x95,0x33,0x5F,0x17,0x46,0xAF,0x7C,0x1E,0x54,0xB5,0x10,0xDA,0xA3};
377 378
static const ff_asf_guid EVENTID_AudioTypeSpanningEvent =
    {0xBE,0xBF,0x1C,0x50,0x49,0xB8,0xCE,0x42,0x9B,0xE9,0x3D,0xB8,0x69,0xFB,0x82,0xB3};
Peter Ross's avatar
Peter Ross committed
379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439

/* Windows media GUIDs */

#define MEDIASUBTYPE_BASE_GUID \
    0x00,0x00,0x10,0x00,0x80,0x00,0x00,0xAA,0x00,0x38,0x9B,0x71

/* Media types */
static const ff_asf_guid mediatype_audio =
    {'a','u','d','s',MEDIASUBTYPE_BASE_GUID};
static const ff_asf_guid mediatype_video =
    {'v','i','d','s',MEDIASUBTYPE_BASE_GUID};
static const ff_asf_guid mediasubtype_mpeg1payload =
    {0x81,0xEB,0x36,0xE4,0x4F,0x52,0xCE,0x11,0x9F,0x53,0x00,0x20,0xAF,0x0B,0xA7,0x70};
static const ff_asf_guid mediatype_mpeg2_sections =
    {0x6C,0x17,0x5F,0x45,0x06,0x4B,0xCE,0x47,0x9A,0xEF,0x8C,0xAE,0xF7,0x3D,0xF7,0xB5};
static const ff_asf_guid mediatype_mpeg2_pes =
    {0x20,0x80,0x6D,0xE0,0x46,0xDB,0xCF,0x11,0xB4,0xD1,0x00,0x80,0x5F,0x6C,0xBB,0xEA};
static const ff_asf_guid mediatype_mstvcaption =
    {0x89,0x8A,0x8B,0xB8,0x49,0xB0,0x80,0x4C,0xAD,0xCF,0x58,0x98,0x98,0x5E,0x22,0xC1};

/* Media subtypes */
static const ff_asf_guid mediasubtype_cpfilters_processed =
    {0x28,0xBD,0xAD,0x46,0xD0,0x6F,0x96,0x47,0x93,0xB2,0x15,0x5C,0x51,0xDC,0x04,0x8D};
static const ff_asf_guid mediasubtype_dvb_subtitle =
    {0xC3,0xCB,0xFF,0x34,0xB3,0xD5,0x71,0x41,0x90,0x02,0xD4,0xC6,0x03,0x01,0x69,0x7F};
static const ff_asf_guid mediasubtype_teletext =
    {0xE3,0x76,0x2A,0xF7,0x0A,0xEB,0xD0,0x11,0xAC,0xE4,0x00,0x00,0xC0,0xCC,0x16,0xBA};
static const ff_asf_guid mediasubtype_dtvccdata =
    {0xAA,0xDD,0x2A,0xF5,0xF0,0x36,0xF5,0x43,0x95,0xEA,0x6D,0x86,0x64,0x84,0x26,0x2A};
static const ff_asf_guid mediasubtype_mpeg2_sections =
    {0x79,0x85,0x9F,0x4A,0xF8,0x6B,0x92,0x43,0x8A,0x6D,0xD2,0xDD,0x09,0xFA,0x78,0x61};

/* Formats */
static const ff_asf_guid format_cpfilters_processed =
    {0x6F,0xB3,0x39,0x67,0x5F,0x1D,0xC2,0x4A,0x81,0x92,0x28,0xBB,0x0E,0x73,0xD1,0x6A};
static const ff_asf_guid format_waveformatex =
    {0x81,0x9F,0x58,0x05,0x56,0xC3,0xCE,0x11,0xBF,0x01,0x00,0xAA,0x00,0x55,0x59,0x5A};
static const ff_asf_guid format_videoinfo2 =
    {0xA0,0x76,0x2A,0xF7,0x0A,0xEB,0xD0,0x11,0xAC,0xE4,0x00,0x00,0xC0,0xCC,0x16,0xBA};
static const ff_asf_guid format_mpeg2_video =
    {0xE3,0x80,0x6D,0xE0,0x46,0xDB,0xCF,0x11,0xB4,0xD1,0x00,0x80,0x5F,0x6C,0xBB,0xEA};
static const ff_asf_guid format_none =
    {0xD6,0x17,0x64,0x0F,0x18,0xC3,0xD0,0x11,0xA4,0x3F,0x00,0xA0,0xC9,0x22,0x31,0x96};

static const AVCodecGuid video_guids[] = {
    {CODEC_ID_MPEG2VIDEO, {0x26,0x80,0x6D,0xE0,0x46,0xDB,0xCF,0x11,0xB4,0xD1,0x00,0x80,0x5F,0x6C,0xBB,0xEA}},
    {CODEC_ID_NONE}
};

static const AVCodecGuid audio_guids[] = {
    {CODEC_ID_AC3,        {0x2C,0x80,0x6D,0xE0,0x46,0xDB,0xCF,0x11,0xB4,0xD1,0x00,0x80,0x5F,0x6C,0xBB,0xEA}},
    {CODEC_ID_EAC3,       {0xAF,0x87,0xFB,0xA7,0x02,0x2D,0xFB,0x42,0xA4,0xD4,0x05,0xCD,0x93,0x84,0x3B,0xDD}},
    {CODEC_ID_MP2,        {0x2B,0x80,0x6D,0xE0,0x46,0xDB,0xCF,0x11,0xB4,0xD1,0x00,0x80,0x5F,0x6C,0xBB,0xEA}},
    {CODEC_ID_NONE}
};

static int read_probe(AVProbeData *p)
{
    return ff_guidcmp(p->buf, wtv_guid) ? 0 : AVPROBE_SCORE_MAX;
}

Peter Ross's avatar
Peter Ross committed
440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460 461 462
/**
 * Convert win32 FILETIME to ISO-8601 string
 */
static void filetime_to_iso8601(char *buf, int buf_size, int64_t value)
{
    time_t t = (value / 10000000LL) - 11644473600LL;
    strftime(buf, buf_size, "%Y-%m-%d %H:%M:%S", gmtime(&t));
}

/**
 * Convert crazy time (100ns since 1 Jan 0001) to ISO-8601 string
 */
static void crazytime_to_iso8601(char *buf, int buf_size, int64_t value)
{
    time_t t = (value / 10000000LL) - 719162LL*86400LL;
    strftime(buf, buf_size, "%Y-%m-%d %H:%M:%S", gmtime(&t));
}

/**
 * Convert OLE DATE to ISO-8601 string
 */
static void oledate_to_iso8601(char *buf, int buf_size, int64_t value)
{
463
    time_t t = 631112400LL + 86400*av_int2double(value);
Peter Ross's avatar
Peter Ross committed
464 465 466
    strftime(buf, buf_size, "%Y-%m-%d %H:%M:%S", gmtime(&t));
}

467
static void get_attachment(AVFormatContext *s, AVIOContext *pb, int length)
Peter Ross's avatar
Peter Ross committed
468 469 470 471 472
{
    char mime[1024];
    char description[1024];
    unsigned int filesize;
    AVStream *st;
473
    int64_t pos = avio_tell(pb);
Peter Ross's avatar
Peter Ross committed
474 475 476 477 478

    avio_get_str16le(pb, INT_MAX, mime, sizeof(mime));
    if (strcmp(mime, "image/jpeg"))
        goto done;

479
    avio_r8(pb);
Peter Ross's avatar
Peter Ross committed
480
    avio_get_str16le(pb, INT_MAX, description, sizeof(description));
481
    filesize = avio_rl32(pb);
Peter Ross's avatar
Peter Ross committed
482 483 484
    if (!filesize)
        goto done;

485
    st = avformat_new_stream(s, NULL);
Peter Ross's avatar
Peter Ross committed
486 487
    if (!st)
        goto done;
488
    av_dict_set(&st->metadata, "title", description, 0);
Peter Ross's avatar
Peter Ross committed
489 490 491 492 493 494
    st->codec->codec_id   = CODEC_ID_MJPEG;
    st->codec->codec_type = AVMEDIA_TYPE_ATTACHMENT;
    st->codec->extradata  = av_mallocz(filesize);
    if (!st->codec->extradata)
        goto done;
    st->codec->extradata_size = filesize;
495
    avio_read(pb, st->codec->extradata, filesize);
Peter Ross's avatar
Peter Ross committed
496
done:
497
    avio_seek(pb, pos + length, SEEK_SET);
Peter Ross's avatar
Peter Ross committed
498 499
}

500
static void get_tag(AVFormatContext *s, AVIOContext *pb, const char *key, int type, int length)
Peter Ross's avatar
Peter Ross committed
501 502 503 504 505 506 507
{
    int buf_size = FFMAX(2*length, LEN_PRETTY_GUID) + 1;
    char *buf = av_malloc(buf_size);
    if (!buf)
        return;

    if (type == 0 && length == 4) {
508
        snprintf(buf, buf_size, "%"PRIi32, avio_rl32(pb));
Peter Ross's avatar
Peter Ross committed
509 510 511 512 513 514 515
    } else if (type == 1) {
        avio_get_str16le(pb, length, buf, buf_size);
        if (!strlen(buf)) {
           av_free(buf);
           return;
        }
    } else if (type == 3 && length == 4) {
516
        strcpy(buf, avio_rl32(pb) ? "true" : "false");
Peter Ross's avatar
Peter Ross committed
517
    } else if (type == 4 && length == 8) {
518
        int64_t num = avio_rl64(pb);
Peter Ross's avatar
Peter Ross committed
519 520 521 522 523 524 525 526 527
        if (!strcmp(key, "WM/EncodingTime") ||
            !strcmp(key, "WM/MediaOriginalBroadcastDateTime"))
            filetime_to_iso8601(buf, buf_size, num);
        else if (!strcmp(key, "WM/WMRVEncodeTime") ||
                 !strcmp(key, "WM/WMRVEndTime"))
            crazytime_to_iso8601(buf, buf_size, num);
        else if (!strcmp(key, "WM/WMRVExpirationDate"))
            oledate_to_iso8601(buf, buf_size, num);
        else if (!strcmp(key, "WM/WMRVBitrate"))
528
            snprintf(buf, buf_size, "%f", av_int2double(num));
Peter Ross's avatar
Peter Ross committed
529 530 531
        else
            snprintf(buf, buf_size, "%"PRIi64, num);
    } else if (type == 5 && length == 2) {
532
        snprintf(buf, buf_size, "%"PRIi16, avio_rl16(pb));
Peter Ross's avatar
Peter Ross committed
533 534
    } else if (type == 6 && length == 16) {
        ff_asf_guid guid;
535
        avio_read(pb, guid, 16);
Peter Ross's avatar
Peter Ross committed
536 537 538 539 540 541 542 543
        snprintf(buf, buf_size, PRI_PRETTY_GUID, ARG_PRETTY_GUID(guid));
    } else if (type == 2 && !strcmp(key, "WM/Picture")) {
        get_attachment(s, pb, length);
        av_freep(&buf);
        return;
    } else {
        av_freep(&buf);
        av_log(s, AV_LOG_WARNING, "unsupported metadata entry; key:%s, type:%d, length:0x%x\n", key, type, length);
544
        avio_skip(pb, length);
Peter Ross's avatar
Peter Ross committed
545 546 547
        return;
    }

548
    av_dict_set(&s->metadata, key, buf, 0);
Peter Ross's avatar
Peter Ross committed
549 550 551 552 553 554
    av_freep(&buf);
}

/**
 * Parse metadata entries
 */
555
static void parse_legacy_attrib(AVFormatContext *s, AVIOContext *pb)
Peter Ross's avatar
Peter Ross committed
556 557 558
{
    ff_asf_guid guid;
    int length, type;
Anton Khirnov's avatar
Anton Khirnov committed
559
    while(!pb->eof_reached) {
Peter Ross's avatar
Peter Ross committed
560 561
        char key[1024];
        ff_get_guid(pb, &guid);
562 563
        type   = avio_rl32(pb);
        length = avio_rl32(pb);
Peter Ross's avatar
Peter Ross committed
564 565 566 567 568 569 570 571 572 573 574 575 576 577
        if (!length)
            break;
        if (ff_guidcmp(&guid, metadata_guid)) {
            av_log(s, AV_LOG_WARNING, "unknown guid "PRI_GUID", expected metadata_guid; "
                   "remaining metadata entries ignored\n", ARG_GUID(guid));
            break;
        }
        avio_get_str16le(pb, INT_MAX, key, sizeof(key));
        get_tag(s, pb, key, type, length);
    }

    ff_metadata_conv(&s->metadata, NULL, ff_asf_metadata_conv);
}

Peter Ross's avatar
Peter Ross committed
578 579 580 581 582 583
/**
 * parse VIDEOINFOHEADER2 structure
 * @return bytes consumed
 */
static int parse_videoinfoheader2(AVFormatContext *s, AVStream *st)
{
Peter Ross's avatar
Peter Ross committed
584
    WtvContext *wtv = s->priv_data;
585
    AVIOContext *pb = wtv->pb;
Peter Ross's avatar
Peter Ross committed
586

587
    avio_skip(pb, 72);  // picture aspect ratio is unreliable
Peter Ross's avatar
Peter Ross committed
588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609 610 611 612 613
    ff_get_bmp_header(pb, st);

    return 72 + 40;
}

/**
 * Parse MPEG1WAVEFORMATEX extradata structure
 */
static void parse_mpeg1waveformatex(AVStream *st)
{
    /* fwHeadLayer */
    switch (AV_RL16(st->codec->extradata)) {
    case 0x0001 : st->codec->codec_id = CODEC_ID_MP1; break;
    case 0x0002 : st->codec->codec_id = CODEC_ID_MP2; break;
    case 0x0004 : st->codec->codec_id = CODEC_ID_MP3; break;
    }

    st->codec->bit_rate = AV_RL32(st->codec->extradata + 2); /* dwHeadBitrate */

    /* dwHeadMode */
    switch (AV_RL16(st->codec->extradata + 6)) {
    case 1 : case 2 : case 4 : st->codec->channels = 2; break;
    case 8 :                   st->codec->channels = 1; break;
    }
}

614 615 616 617 618 619
/**
 * Initialise stream
 * @param st Stream to initialise, or NULL to create and initialise new stream
 * @return NULL on error
 */
static AVStream * new_stream(AVFormatContext *s, AVStream *st, int sid, int codec_type)
Peter Ross's avatar
Peter Ross committed
620
{
621 622 623 624 625 626
    if (st) {
        if (st->codec->extradata) {
            av_freep(&st->codec->extradata);
            st->codec->extradata_size = 0;
        }
    } else {
627 628 629
        WtvStream *wst = av_mallocz(sizeof(WtvStream));
        if (!wst)
            return NULL;
630
        st = avformat_new_stream(s, NULL);
631 632
        if (!st)
            return NULL;
633
        st->id = sid;
634
        st->priv_data = wst;
635
    }
Peter Ross's avatar
Peter Ross committed
636 637
    st->codec->codec_type = codec_type;
    st->need_parsing      = AVSTREAM_PARSE_FULL;
638
    avpriv_set_pts_info(st, 64, 1, 10000000);
Peter Ross's avatar
Peter Ross committed
639 640 641 642
    return st;
}

/**
643 644
 * parse Media Type structure and populate stream
 * @param st         Stream, or NULL to create new stream
Peter Ross's avatar
Peter Ross committed
645 646 647 648 649 650
 * @param mediatype  Mediatype GUID
 * @param subtype    Subtype GUID
 * @param formattype Format GUID
 * @param size       Size of format buffer
 * @return NULL on error
 */
651
static AVStream * parse_media_type(AVFormatContext *s, AVStream *st, int sid,
Peter Ross's avatar
Peter Ross committed
652 653 654
                                   ff_asf_guid mediatype, ff_asf_guid subtype,
                                   ff_asf_guid formattype, int size)
{
Peter Ross's avatar
Peter Ross committed
655
    WtvContext *wtv = s->priv_data;
656
    AVIOContext *pb = wtv->pb;
Peter Ross's avatar
Peter Ross committed
657 658 659 660 661 662 663
    if (!ff_guidcmp(subtype, mediasubtype_cpfilters_processed) &&
        !ff_guidcmp(formattype, format_cpfilters_processed)) {
        ff_asf_guid actual_subtype;
        ff_asf_guid actual_formattype;

        if (size < 32) {
            av_log(s, AV_LOG_WARNING, "format buffer size underflow\n");
664
            avio_skip(pb, size);
Peter Ross's avatar
Peter Ross committed
665 666 667
            return NULL;
        }

668
        avio_skip(pb, size - 32);
Peter Ross's avatar
Peter Ross committed
669 670
        ff_get_guid(pb, &actual_subtype);
        ff_get_guid(pb, &actual_formattype);
671
        avio_seek(pb, -size, SEEK_CUR);
Peter Ross's avatar
Peter Ross committed
672

673
        st = parse_media_type(s, st, sid, mediatype, actual_subtype, actual_formattype, size - 32);
674
        avio_skip(pb, 32);
Peter Ross's avatar
Peter Ross committed
675 676
        return st;
    } else if (!ff_guidcmp(mediatype, mediatype_audio)) {
677
        st = new_stream(s, st, sid, AVMEDIA_TYPE_AUDIO);
Peter Ross's avatar
Peter Ross committed
678 679 680
        if (!st)
            return NULL;
        if (!ff_guidcmp(formattype, format_waveformatex)) {
681 682 683
            int ret = ff_get_wav_header(pb, st->codec, size);
            if (ret < 0)
                return NULL;
Peter Ross's avatar
Peter Ross committed
684 685 686
        } else {
            if (ff_guidcmp(formattype, format_none))
                av_log(s, AV_LOG_WARNING, "unknown formattype:"PRI_GUID"\n", ARG_GUID(formattype));
687
            avio_skip(pb, size);
Peter Ross's avatar
Peter Ross committed
688 689 690 691 692 693 694 695 696 697 698
        }

        if (!memcmp(subtype + 4, (const uint8_t[]){MEDIASUBTYPE_BASE_GUID}, 12)) {
            st->codec->codec_id = ff_wav_codec_get_id(AV_RL32(subtype), st->codec->bits_per_coded_sample);
        } else if (!ff_guidcmp(subtype, mediasubtype_mpeg1payload)) {
            if (st->codec->extradata && st->codec->extradata_size >= 22)
                parse_mpeg1waveformatex(st);
            else
                av_log(s, AV_LOG_WARNING, "MPEG1WAVEFORMATEX underflow\n");
        } else {
            st->codec->codec_id = ff_codec_guid_get_id(audio_guids, subtype);
699 700
            if (st->codec->codec_id == CODEC_ID_NONE)
                av_log(s, AV_LOG_WARNING, "unknown subtype:"PRI_GUID"\n", ARG_GUID(subtype));
Peter Ross's avatar
Peter Ross committed
701 702 703
        }
        return st;
    } else if (!ff_guidcmp(mediatype, mediatype_video)) {
704
        st = new_stream(s, st, sid, AVMEDIA_TYPE_VIDEO);
Peter Ross's avatar
Peter Ross committed
705 706 707 708
        if (!st)
            return NULL;
        if (!ff_guidcmp(formattype, format_videoinfo2)) {
            int consumed = parse_videoinfoheader2(s, st);
709
            avio_skip(pb, FFMAX(size - consumed, 0));
Peter Ross's avatar
Peter Ross committed
710 711
        } else if (!ff_guidcmp(formattype, format_mpeg2_video)) {
            int consumed = parse_videoinfoheader2(s, st);
712
            avio_skip(pb, FFMAX(size - consumed, 0));
Peter Ross's avatar
Peter Ross committed
713 714 715
        } else {
            if (ff_guidcmp(formattype, format_none))
                av_log(s, AV_LOG_WARNING, "unknown formattype:"PRI_GUID"\n", ARG_GUID(formattype));
716
            avio_skip(pb, size);
Peter Ross's avatar
Peter Ross committed
717 718 719 720 721 722 723 724 725 726 727 728
        }

        if (!memcmp(subtype + 4, (const uint8_t[]){MEDIASUBTYPE_BASE_GUID}, 12)) {
            st->codec->codec_id = ff_codec_get_id(ff_codec_bmp_tags, AV_RL32(subtype));
        } else {
            st->codec->codec_id = ff_codec_guid_get_id(video_guids, subtype);
        }
        if (st->codec->codec_id == CODEC_ID_NONE)
            av_log(s, AV_LOG_WARNING, "unknown subtype:"PRI_GUID"\n", ARG_GUID(subtype));
        return st;
    } else if (!ff_guidcmp(mediatype, mediatype_mpeg2_pes) &&
               !ff_guidcmp(subtype, mediasubtype_dvb_subtitle)) {
729
        st = new_stream(s, st, sid, AVMEDIA_TYPE_SUBTITLE);
Peter Ross's avatar
Peter Ross committed
730 731 732 733
        if (!st)
            return NULL;
        if (ff_guidcmp(formattype, format_none))
            av_log(s, AV_LOG_WARNING, "unknown formattype:"PRI_GUID"\n", ARG_GUID(formattype));
734
        avio_skip(pb, size);
Peter Ross's avatar
Peter Ross committed
735 736 737 738
        st->codec->codec_id = CODEC_ID_DVB_SUBTITLE;
        return st;
    } else if (!ff_guidcmp(mediatype, mediatype_mstvcaption) &&
               (!ff_guidcmp(subtype, mediasubtype_teletext) || !ff_guidcmp(subtype, mediasubtype_dtvccdata))) {
739
        st = new_stream(s, st, sid, AVMEDIA_TYPE_SUBTITLE);
Peter Ross's avatar
Peter Ross committed
740 741 742 743
        if (!st)
            return NULL;
        if (ff_guidcmp(formattype, format_none))
            av_log(s, AV_LOG_WARNING, "unknown formattype:"PRI_GUID"\n", ARG_GUID(formattype));
744
        avio_skip(pb, size);
Peter Ross's avatar
Peter Ross committed
745 746 747 748 749 750
        st->codec->codec_id   = CODEC_ID_DVB_TELETEXT;
        return st;
    } else if (!ff_guidcmp(mediatype, mediatype_mpeg2_sections) &&
               !ff_guidcmp(subtype, mediasubtype_mpeg2_sections)) {
        if (ff_guidcmp(formattype, format_none))
            av_log(s, AV_LOG_WARNING, "unknown formattype:"PRI_GUID"\n", ARG_GUID(formattype));
751
        avio_skip(pb, size);
Peter Ross's avatar
Peter Ross committed
752 753 754 755 756 757
        return NULL;
    }

    av_log(s, AV_LOG_WARNING, "unknown media type, mediatype:"PRI_GUID
                              ", subtype:"PRI_GUID", formattype:"PRI_GUID"\n",
                              ARG_GUID(mediatype), ARG_GUID(subtype), ARG_GUID(formattype));
758
    avio_skip(pb, size);
Peter Ross's avatar
Peter Ross committed
759 760 761 762 763 764 765 766 767 768
    return NULL;
}

enum {
    SEEK_TO_DATA = 0,
    SEEK_TO_PTS,
};

/**
 * Parse WTV chunks
Peter Ross's avatar
Peter Ross committed
769 770
 * @param mode SEEK_TO_DATA or SEEK_TO_PTS
 * @param seekts timestamp
771
 * @param[out] len_ptr Length of data chunk
Peter Ross's avatar
Peter Ross committed
772 773 774 775 776
 * @return stream index of data chunk, or <0 on error
 */
static int parse_chunks(AVFormatContext *s, int mode, int64_t seekts, int *len_ptr)
{
    WtvContext *wtv = s->priv_data;
777
    AVIOContext *pb = wtv->pb;
Anton Khirnov's avatar
Anton Khirnov committed
778
    while (!pb->eof_reached) {
Peter Ross's avatar
Peter Ross committed
779 780 781 782
        ff_asf_guid g;
        int len, sid, consumed;

        ff_get_guid(pb, &g);
783
        len = avio_rl32(pb);
784 785
        if (len < 32)
            break;
786
        sid = avio_rl32(pb) & 0x7FFF;
787
        avio_skip(pb, 8);
Peter Ross's avatar
Peter Ross committed
788 789 790 791 792 793
        consumed = 32;

        if (!ff_guidcmp(g, stream_guid)) {
            if (ff_find_stream_index(s, sid) < 0) {
                ff_asf_guid mediatype, subtype, formattype;
                int size;
794
                avio_skip(pb, 28);
Peter Ross's avatar
Peter Ross committed
795 796
                ff_get_guid(pb, &mediatype);
                ff_get_guid(pb, &subtype);
797
                avio_skip(pb, 12);
Peter Ross's avatar
Peter Ross committed
798
                ff_get_guid(pb, &formattype);
799
                size = avio_rl32(pb);
Peter Ross's avatar
Peter Ross committed
800 801
                parse_media_type(s, 0, sid, mediatype, subtype, formattype, size);
                consumed += 92 + size;
Peter Ross's avatar
Peter Ross committed
802
            }
803 804
        } else if (!ff_guidcmp(g, stream2_guid)) {
            int stream_index = ff_find_stream_index(s, sid);
805
            if (stream_index >= 0 && !((WtvStream*)s->streams[stream_index]->priv_data)->seen_data) {
806 807
                ff_asf_guid mediatype, subtype, formattype;
                int size;
808
                avio_skip(pb, 12);
809 810
                ff_get_guid(pb, &mediatype);
                ff_get_guid(pb, &subtype);
811
                avio_skip(pb, 12);
812
                ff_get_guid(pb, &formattype);
813
                size = avio_rl32(pb);
814 815 816
                parse_media_type(s, s->streams[stream_index], sid, mediatype, subtype, formattype, size);
                consumed += 76 + size;
            }
817 818 819 820 821 822
        } else if (!ff_guidcmp(g, EVENTID_AudioDescriptorSpanningEvent) ||
                   !ff_guidcmp(g, EVENTID_CtxADescriptorSpanningEvent) ||
                   !ff_guidcmp(g, EVENTID_CSDescriptorSpanningEvent) ||
                   !ff_guidcmp(g, EVENTID_StreamIDSpanningEvent) ||
                   !ff_guidcmp(g, EVENTID_SubtitleSpanningEvent) ||
                   !ff_guidcmp(g, EVENTID_TeletextSpanningEvent)) {
Peter Ross's avatar
Peter Ross committed
823 824 825
            int stream_index = ff_find_stream_index(s, sid);
            if (stream_index >= 0) {
                AVStream *st = s->streams[stream_index];
826 827 828 829
                uint8_t buf[258];
                const uint8_t *pbuf = buf;
                int buf_size;

830
                avio_skip(pb, 8);
831 832 833
                consumed += 8;
                if (!ff_guidcmp(g, EVENTID_CtxADescriptorSpanningEvent) ||
                    !ff_guidcmp(g, EVENTID_CSDescriptorSpanningEvent)) {
834
                    avio_skip(pb, 6);
835 836
                    consumed += 6;
                }
Peter Ross's avatar
Peter Ross committed
837

838
                buf_size = FFMIN(len - consumed, sizeof(buf));
839
                avio_read(pb, buf, buf_size);
840
                consumed += buf_size;
841
                ff_parse_mpeg2_descriptor(s, st, 0, &pbuf, buf + buf_size, NULL, 0, 0, NULL);
Peter Ross's avatar
Peter Ross committed
842
            }
843 844 845 846 847
        } else if (!ff_guidcmp(g, EVENTID_AudioTypeSpanningEvent)) {
            int stream_index = ff_find_stream_index(s, sid);
            if (stream_index >= 0) {
                AVStream *st = s->streams[stream_index];
                int audio_type;
848
                avio_skip(pb, 8);
849
                audio_type = avio_r8(pb);
850 851 852 853 854 855
                if (audio_type == 2)
                    st->disposition |= AV_DISPOSITION_HEARING_IMPAIRED;
                else if (audio_type == 3)
                    st->disposition |= AV_DISPOSITION_VISUAL_IMPAIRED;
                consumed += 9;
            }
856 857 858
        } else if (!ff_guidcmp(g, EVENTID_DVBScramblingControlSpanningEvent)) {
            int stream_index = ff_find_stream_index(s, sid);
            if (stream_index >= 0) {
859
                avio_skip(pb, 12);
860
                if (avio_rl32(pb))
861 862 863
                    av_log(s, AV_LOG_WARNING, "DVB scrambled stream detected (st:%d), decoding will likely fail\n", stream_index);
                consumed += 16;
            }
864
        } else if (!ff_guidcmp(g, EVENTID_LanguageSpanningEvent)) {
Peter Ross's avatar
Peter Ross committed
865 866 867
            int stream_index = ff_find_stream_index(s, sid);
            if (stream_index >= 0) {
                AVStream *st = s->streams[stream_index];
868
                uint8_t language[4];
869
                avio_skip(pb, 12);
870
                avio_read(pb, language, 3);
871 872
                if (language[0]) {
                    language[3] = 0;
873
                    av_dict_set(&st->metadata, "language", language, 0);
874 875
                    if (!strcmp(language, "nar") || !strcmp(language, "NAR"))
                        st->disposition |= AV_DISPOSITION_VISUAL_IMPAIRED;
876
                }
Peter Ross's avatar
Peter Ross committed
877 878 879
                consumed += 15;
            }
        } else if (!ff_guidcmp(g, timestamp_guid)) {
880 881
            int stream_index = ff_find_stream_index(s, sid);
            if (stream_index >= 0) {
882
                avio_skip(pb, 8);
883
                wtv->pts = avio_rl64(pb);
Peter Ross's avatar
Peter Ross committed
884 885 886
                consumed += 16;
                if (wtv->pts == -1)
                    wtv->pts = AV_NOPTS_VALUE;
Peter Ross's avatar
Peter Ross committed
887 888 889 890
                else {
                    wtv->last_valid_pts = wtv->pts;
                    if (wtv->epoch == AV_NOPTS_VALUE || wtv->pts < wtv->epoch)
                        wtv->epoch = wtv->pts;
Peter Ross's avatar
Peter Ross committed
891
                if (mode == SEEK_TO_PTS && wtv->pts >= seekts) {
Peter Ross's avatar
Peter Ross committed
892
#define WTV_PAD8(x) (((x) + 7) & ~7)
893
                    avio_skip(pb, WTV_PAD8(len) - consumed);
Peter Ross's avatar
Peter Ross committed
894 895
                    return 0;
                }
Peter Ross's avatar
Peter Ross committed
896
                }
897
            }
Peter Ross's avatar
Peter Ross committed
898 899
        } else if (!ff_guidcmp(g, data_guid)) {
            int stream_index = ff_find_stream_index(s, sid);
900
            if (mode == SEEK_TO_DATA && stream_index >= 0 && len > 32) {
901 902
                WtvStream *wst = s->streams[stream_index]->priv_data;
                wst->seen_data = 1;
Peter Ross's avatar
Peter Ross committed
903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925
                if (len_ptr) {
                    *len_ptr = len;
                }
                return stream_index;
            }
        } else if (
            !ff_guidcmp(g, /* DSATTRIB_CAPTURE_STREAMTIME */ (const ff_asf_guid){0x14,0x56,0x1A,0x0C,0xCD,0x30,0x40,0x4F,0xBC,0xBF,0xD0,0x3E,0x52,0x30,0x62,0x07}) ||
            !ff_guidcmp(g, /* DSATTRIB_PicSampleSeq */ (const ff_asf_guid){0x02,0xAE,0x5B,0x2F,0x8F,0x7B,0x60,0x4F,0x82,0xD6,0xE4,0xEA,0x2F,0x1F,0x4C,0x99}) ||
            !ff_guidcmp(g, /* DSATTRIB_TRANSPORT_PROPERTIES */ (const ff_asf_guid){0x12,0xF6,0x22,0xB6,0xAD,0x47,0x71,0x46,0xAD,0x6C,0x05,0xA9,0x8E,0x65,0xDE,0x3A}) ||
            !ff_guidcmp(g, /* dvr_ms_vid_frame_rep_data */ (const ff_asf_guid){0xCC,0x32,0x64,0xDD,0x29,0xE2,0xDB,0x40,0x80,0xF6,0xD2,0x63,0x28,0xD2,0x76,0x1F}) ||
            !ff_guidcmp(g, /* EVENTID_ChannelChangeSpanningEvent */ (const ff_asf_guid){0xE5,0xC5,0x67,0x90,0x5C,0x4C,0x05,0x42,0x86,0xC8,0x7A,0xFE,0x20,0xFE,0x1E,0xFA}) ||
            !ff_guidcmp(g, /* EVENTID_ChannelInfoSpanningEvent */ (const ff_asf_guid){0x80,0x6D,0xF3,0x41,0x32,0x41,0xC2,0x4C,0xB1,0x21,0x01,0xA4,0x32,0x19,0xD8,0x1B}) ||
            !ff_guidcmp(g, /* EVENTID_ChannelTypeSpanningEvent */ (const ff_asf_guid){0x51,0x1D,0xAB,0x72,0xD2,0x87,0x9B,0x48,0xBA,0x11,0x0E,0x08,0xDC,0x21,0x02,0x43}) ||
            !ff_guidcmp(g, /* EVENTID_PIDListSpanningEvent */ (const ff_asf_guid){0x65,0x8F,0xFC,0x47,0xBB,0xE2,0x34,0x46,0x9C,0xEF,0xFD,0xBF,0xE6,0x26,0x1D,0x5C}) ||
            !ff_guidcmp(g, /* EVENTID_SignalAndServiceStatusSpanningEvent */ (const ff_asf_guid){0xCB,0xC5,0x68,0x80,0x04,0x3C,0x2B,0x49,0xB4,0x7D,0x03,0x08,0x82,0x0D,0xCE,0x51}) ||
            !ff_guidcmp(g, /* EVENTID_StreamTypeSpanningEvent */ (const ff_asf_guid){0xBC,0x2E,0xAF,0x82,0xA6,0x30,0x64,0x42,0xA8,0x0B,0xAD,0x2E,0x13,0x72,0xAC,0x60}) ||
            !ff_guidcmp(g, (const ff_asf_guid){0x1E,0xBE,0xC3,0xC5,0x43,0x92,0xDC,0x11,0x85,0xE5,0x00,0x12,0x3F,0x6F,0x73,0xB9}) ||
            !ff_guidcmp(g, (const ff_asf_guid){0x3B,0x86,0xA2,0xB1,0xEB,0x1E,0xC3,0x44,0x8C,0x88,0x1C,0xA3,0xFF,0xE3,0xE7,0x6A}) ||
            !ff_guidcmp(g, (const ff_asf_guid){0x4E,0x7F,0x4C,0x5B,0xC4,0xD0,0x38,0x4B,0xA8,0x3E,0x21,0x7F,0x7B,0xBF,0x52,0xE7}) ||
            !ff_guidcmp(g, (const ff_asf_guid){0x63,0x36,0xEB,0xFE,0xA1,0x7E,0xD9,0x11,0x83,0x08,0x00,0x07,0xE9,0x5E,0xAD,0x8D}) ||
            !ff_guidcmp(g, (const ff_asf_guid){0x70,0xE9,0xF1,0xF8,0x89,0xA4,0x4C,0x4D,0x83,0x73,0xB8,0x12,0xE0,0xD5,0xF8,0x1E}) ||
            !ff_guidcmp(g, (const ff_asf_guid){0x96,0xC3,0xD2,0xC2,0x7E,0x9A,0xDA,0x11,0x8B,0xF7,0x00,0x07,0xE9,0x5E,0xAD,0x8D}) ||
            !ff_guidcmp(g, (const ff_asf_guid){0x97,0xC3,0xD2,0xC2,0x7E,0x9A,0xDA,0x11,0x8B,0xF7,0x00,0x07,0xE9,0x5E,0xAD,0x8D}) ||
926
            !ff_guidcmp(g, (const ff_asf_guid){0xA1,0xC3,0xD2,0xC2,0x7E,0x9A,0xDA,0x11,0x8B,0xF7,0x00,0x07,0xE9,0x5E,0xAD,0x8D})) {
Peter Ross's avatar
Peter Ross committed
927 928
            //ignore known guids
        } else
Peter Ross's avatar
Peter Ross committed
929
            av_log(s, AV_LOG_WARNING, "unsupported chunk:"PRI_GUID"\n", ARG_GUID(g));
Peter Ross's avatar
Peter Ross committed
930

931
        avio_skip(pb, WTV_PAD8(len) - consumed);
Peter Ross's avatar
Peter Ross committed
932 933 934 935
    }
    return AVERROR_EOF;
}

Peter Ross's avatar
Peter Ross committed
936 937 938 939 940 941 942 943 944 945 946
/* declare utf16le strings */
#define _ , 0,
static const uint8_t timeline_le16[] =
    {'t'_'i'_'m'_'e'_'l'_'i'_'n'_'e', 0};
static const uint8_t table_0_entries_legacy_attrib_le16[] =
    {'t'_'a'_'b'_'l'_'e'_'.'_'0'_'.'_'e'_'n'_'t'_'r'_'i'_'e'_'s'_'.'_'l'_'e'_'g'_'a'_'c'_'y'_'_'_'a'_'t'_'t'_'r'_'i'_'b', 0};
static const uint8_t table_0_entries_time_le16[] =
    {'t'_'a'_'b'_'l'_'e'_'.'_'0'_'.'_'e'_'n'_'t'_'r'_'i'_'e'_'s'_'.'_'t'_'i'_'m'_'e', 0};
static const uint8_t timeline_table_0_entries_Events_le16[] =
    {'t'_'i'_'m'_'e'_'l'_'i'_'n'_'e'_'.'_'t'_'a'_'b'_'l'_'e'_'.'_'0'_'.'_'e'_'n'_'t'_'r'_'i'_'e'_'s'_'.'_'E'_'v'_'e'_'n'_'t'_'s', 0};
#undef _
Peter Ross's avatar
Peter Ross committed
947 948 949

static int read_header(AVFormatContext *s, AVFormatParameters *ap)
{
Peter Ross's avatar
Peter Ross committed
950 951 952
    WtvContext *wtv = s->priv_data;
    int root_sector, root_size;
    uint8_t root[WTV_SECTOR_SIZE];
953
    AVIOContext *pb;
Peter Ross's avatar
Peter Ross committed
954
    int64_t timeline_pos;
Peter Ross's avatar
Peter Ross committed
955 956
    int ret;

Peter Ross's avatar
Peter Ross committed
957 958 959 960 961
    wtv->epoch          =
    wtv->pts            =
    wtv->last_valid_pts = AV_NOPTS_VALUE;

    /* read root directory sector */
962
    avio_skip(s->pb, 0x30);
963
    root_size = avio_rl32(s->pb);
Peter Ross's avatar
Peter Ross committed
964 965 966 967
    if (root_size > sizeof(root)) {
        av_log(s, AV_LOG_ERROR, "root directory size exceeds sector size\n");
        return AVERROR_INVALIDDATA;
    }
968
    avio_skip(s->pb, 4);
969
    root_sector = avio_rl32(s->pb);
Peter Ross's avatar
Peter Ross committed
970

971
    avio_seek(s->pb, root_sector << WTV_SECTOR_BITS, SEEK_SET);
972
    root_size = avio_read(s->pb, root, root_size);
Peter Ross's avatar
Peter Ross committed
973 974 975 976 977 978 979 980 981 982
    if (root_size < 0)
        return AVERROR_INVALIDDATA;

    /* parse chunks up until first data chunk */
    wtv->pb = wtvfile_open(s, root, root_size, timeline_le16);
    if (!wtv->pb) {
        av_log(s, AV_LOG_ERROR, "timeline data missing\n");
        return AVERROR_INVALIDDATA;
    }

Peter Ross's avatar
Peter Ross committed
983 984 985
    ret = parse_chunks(s, SEEK_TO_DATA, 0, 0);
    if (ret < 0)
        return ret;
986
    avio_seek(wtv->pb, -32, SEEK_CUR);
Peter Ross's avatar
Peter Ross committed
987

988
    timeline_pos = avio_tell(s->pb); // save before opening another file
Peter Ross's avatar
Peter Ross committed
989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002

    /* read metadata */
    pb = wtvfile_open(s, root, root_size, table_0_entries_legacy_attrib_le16);
    if (pb) {
        parse_legacy_attrib(s, pb);
        wtvfile_close(pb);
    }

    /* read seek index */
    if (s->nb_streams) {
        AVStream *st = s->streams[0];
        pb = wtvfile_open(s, root, root_size, table_0_entries_time_le16);
        if (pb) {
            while(1) {
1003 1004
                uint64_t timestamp = avio_rl64(pb);
                uint64_t frame_nb  = avio_rl64(pb);
Anton Khirnov's avatar
Anton Khirnov committed
1005
                if (pb->eof_reached)
Peter Ross's avatar
Peter Ross committed
1006 1007 1008 1009 1010
                    break;
                ff_add_index_entry(&wtv->index_entries, &wtv->nb_index_entries, &wtv->index_entries_allocated_size,
                                   0, timestamp, frame_nb, 0, AVINDEX_KEYFRAME);
            }
            wtvfile_close(pb);
Peter Ross's avatar
Peter Ross committed
1011

Peter Ross's avatar
Peter Ross committed
1012 1013 1014 1015 1016
            if (wtv->nb_index_entries) {
                pb = wtvfile_open(s, root, root_size, timeline_table_0_entries_Events_le16);
                if (pb) {
                    int i;
                    while (1) {
1017 1018
                        uint64_t frame_nb = avio_rl64(pb);
                        uint64_t position = avio_rl64(pb);
Anton Khirnov's avatar
Anton Khirnov committed
1019
                        if (pb->eof_reached)