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

Michael Niedermayer's avatar
Michael Niedermayer committed
27 28 29
#undef NDEBUG
#include <assert.h>

30
#define FRAME_HEADER_SIZE 17
31
// Fix Me! FRAME_HEADER_SIZE may be different.
32

Fabrice Bellard's avatar
Fabrice Bellard committed
33 34 35 36 37 38 39 40 41 42
static const GUID index_guid = {
    0x33000890, 0xe5b1, 0x11cf, { 0x89, 0xf4, 0x00, 0xa0, 0xc9, 0x03, 0x49, 0xcb },
};

/**********************************/
/* decoding */

//#define DEBUG

#ifdef DEBUG
François Revol's avatar
François Revol committed
43 44 45 46
#define PRINT_IF_GUID(g,cmp) \
if (!memcmp(g, &cmp, sizeof(GUID))) \
    printf("(GUID: %s) ", #cmp)

Fabrice Bellard's avatar
Fabrice Bellard committed
47 48 49
static void print_guid(const GUID *g)
{
    int i;
François Revol's avatar
François Revol committed
50 51 52 53 54 55 56
    PRINT_IF_GUID(g, asf_header);
    else PRINT_IF_GUID(g, file_header);
    else PRINT_IF_GUID(g, stream_header);
    else PRINT_IF_GUID(g, audio_stream);
    else PRINT_IF_GUID(g, audio_conceal_none);
    else PRINT_IF_GUID(g, video_stream);
    else PRINT_IF_GUID(g, video_conceal_none);
57
    else PRINT_IF_GUID(g, command_stream);
François Revol's avatar
François Revol committed
58 59 60 61 62 63 64 65
    else PRINT_IF_GUID(g, comment_header);
    else PRINT_IF_GUID(g, codec_comment_header);
    else PRINT_IF_GUID(g, codec_comment1_header);
    else PRINT_IF_GUID(g, data_header);
    else PRINT_IF_GUID(g, index_guid);
    else PRINT_IF_GUID(g, head1_guid);
    else PRINT_IF_GUID(g, head2_guid);
    else PRINT_IF_GUID(g, my_guid);
66 67 68 69
    else PRINT_IF_GUID(g, ext_stream_header);
    else PRINT_IF_GUID(g, extended_content_header);
    else PRINT_IF_GUID(g, ext_stream_embed_stream_header);
    else PRINT_IF_GUID(g, ext_stream_audio_stream);
François Revol's avatar
François Revol committed
70 71
    else
        printf("(GUID: unknown) ");
Fabrice Bellard's avatar
Fabrice Bellard committed
72
    printf("0x%08x, 0x%04x, 0x%04x, {", g->v1, g->v2, g->v3);
73
    for(i=0;i<8;i++)
Fabrice Bellard's avatar
Fabrice Bellard committed
74 75 76
        printf(" 0x%02x,", g->v4[i]);
    printf("}\n");
}
77
#undef PRINT_IF_GUID
Fabrice Bellard's avatar
Fabrice Bellard committed
78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110
#endif

static void get_guid(ByteIOContext *s, GUID *g)
{
    int i;

    g->v1 = get_le32(s);
    g->v2 = get_le16(s);
    g->v3 = get_le16(s);
    for(i=0;i<8;i++)
        g->v4[i] = get_byte(s);
}

#if 0
static void get_str16(ByteIOContext *pb, char *buf, int buf_size)
{
    int len, c;
    char *q;

    len = get_le16(pb);
    q = buf;
    while (len > 0) {
        c = get_le16(pb);
        if ((q - buf) < buf_size - 1)
            *q++ = c;
        len--;
    }
    *q = '\0';
}
#endif

static void get_str16_nolen(ByteIOContext *pb, int len, char *buf, int buf_size)
{
Zuxy Meng's avatar
Zuxy Meng committed
111 112 113 114 115
    char* q = buf;
    len /= 2;
    while (len--) {
        uint8_t tmp;
        PUT_UTF8(get_le16(pb), tmp, if (q - buf < buf_size - 1) *q++ = tmp;)
Fabrice Bellard's avatar
Fabrice Bellard committed
116 117 118 119
    }
    *q = '\0';
}

Fabrice Bellard's avatar
Fabrice Bellard committed
120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144
static int asf_probe(AVProbeData *pd)
{
    GUID g;
    const unsigned char *p;
    int i;

    /* check file header */
    if (pd->buf_size <= 32)
        return 0;
    p = pd->buf;
    g.v1 = p[0] | (p[1] << 8) | (p[2] << 16) | (p[3] << 24);
    p += 4;
    g.v2 = p[0] | (p[1] << 8);
    p += 2;
    g.v3 = p[0] | (p[1] << 8);
    p += 2;
    for(i=0;i<8;i++)
        g.v4[i] = *p++;

    if (!memcmp(&g, &asf_header, sizeof(GUID)))
        return AVPROBE_SCORE_MAX;
    else
        return 0;
}

Fabrice Bellard's avatar
Fabrice Bellard committed
145 146
static int asf_read_header(AVFormatContext *s, AVFormatParameters *ap)
{
Fabrice Bellard's avatar
Fabrice Bellard committed
147
    ASFContext *asf = s->priv_data;
Fabrice Bellard's avatar
Fabrice Bellard committed
148 149 150 151
    GUID g;
    ByteIOContext *pb = &s->pb;
    AVStream *st;
    ASFStream *asf_st;
152
    int size, i;
153
    int64_t gsize;
Fabrice Bellard's avatar
Fabrice Bellard committed
154 155 156 157 158 159 160 161

    get_guid(pb, &g);
    if (memcmp(&g, &asf_header, sizeof(GUID)))
        goto fail;
    get_le64(pb);
    get_le32(pb);
    get_byte(pb);
    get_byte(pb);
162
    memset(&asf->asfid2avid, -1, sizeof(asf->asfid2avid));
Fabrice Bellard's avatar
Fabrice Bellard committed
163 164 165 166
    for(;;) {
        get_guid(pb, &g);
        gsize = get_le64(pb);
#ifdef DEBUG
167
        printf("%08"PRIx64": ", url_ftell(pb) - 24);
Fabrice Bellard's avatar
Fabrice Bellard committed
168
        print_guid(&g);
169
        printf("  size=0x%"PRIx64"\n", gsize);
Fabrice Bellard's avatar
Fabrice Bellard committed
170 171 172 173
#endif
        if (gsize < 24)
            goto fail;
        if (!memcmp(&g, &file_header, sizeof(GUID))) {
174
            get_guid(pb, &asf->hdr.guid);
175 176 177 178 179 180 181 182 183 184 185 186
            asf->hdr.file_size          = get_le64(pb);
            asf->hdr.create_time        = get_le64(pb);
            asf->hdr.packets_count      = get_le64(pb);
            asf->hdr.send_time          = get_le64(pb);
            asf->hdr.play_time          = get_le64(pb);
            asf->hdr.preroll            = get_le32(pb);
            asf->hdr.ignore             = get_le32(pb);
            asf->hdr.flags              = get_le32(pb);
            asf->hdr.min_pktsize        = get_le32(pb);
            asf->hdr.max_pktsize        = get_le32(pb);
            asf->hdr.max_bitrate        = get_le32(pb);
            asf->packet_size = asf->hdr.max_pktsize;
187
            asf->nb_packets = asf->hdr.packets_count;
Fabrice Bellard's avatar
Fabrice Bellard committed
188
        } else if (!memcmp(&g, &stream_header, sizeof(GUID))) {
189 190
            int type, type_specific_size, sizeX;
            uint64_t total_size;
Fabrice Bellard's avatar
Fabrice Bellard committed
191
            unsigned int tag1;
192
            int64_t pos1, pos2;
193
            int test_for_ext_stream_audio;
194

Fabrice Bellard's avatar
Fabrice Bellard committed
195 196
            pos1 = url_ftell(pb);

197
            st = av_new_stream(s, 0);
Fabrice Bellard's avatar
Fabrice Bellard committed
198 199
            if (!st)
                goto fail;
200
            av_set_pts_info(st, 32, 1, 1000); /* 32 bit pts in ms */
Fabrice Bellard's avatar
Fabrice Bellard committed
201 202 203 204
            asf_st = av_mallocz(sizeof(ASFStream));
            if (!asf_st)
                goto fail;
            st->priv_data = asf_st;
205
            st->start_time = asf->hdr.preroll;
206
            st->duration = asf->hdr.send_time /
207
                (10000000 / 1000) - st->start_time;
Fabrice Bellard's avatar
Fabrice Bellard committed
208
            get_guid(pb, &g);
209 210

            test_for_ext_stream_audio = 0;
Fabrice Bellard's avatar
Fabrice Bellard committed
211 212 213 214
            if (!memcmp(&g, &audio_stream, sizeof(GUID))) {
                type = CODEC_TYPE_AUDIO;
            } else if (!memcmp(&g, &video_stream, sizeof(GUID))) {
                type = CODEC_TYPE_VIDEO;
215 216
            } else if (!memcmp(&g, &command_stream, sizeof(GUID))) {
                type = CODEC_TYPE_UNKNOWN;
217 218 219
            } else if (!memcmp(&g, &ext_stream_embed_stream_header, sizeof(GUID))) {
                test_for_ext_stream_audio = 1;
                type = CODEC_TYPE_UNKNOWN;
Fabrice Bellard's avatar
Fabrice Bellard committed
220 221 222 223 224
            } else {
                goto fail;
            }
            get_guid(pb, &g);
            total_size = get_le64(pb);
225
            type_specific_size = get_le32(pb);
Fabrice Bellard's avatar
Fabrice Bellard committed
226
            get_le32(pb);
227
            st->id = get_le16(pb) & 0x7f; /* stream id */
228
            // mapping of asf ID to AV stream ID;
229
            asf->asfid2avid[st->id] = s->nb_streams - 1;
230

Fabrice Bellard's avatar
Fabrice Bellard committed
231
            get_le32(pb);
232 233 234 235 236 237 238 239 240 241 242 243 244 245

            if (test_for_ext_stream_audio) {
                get_guid(pb, &g);
                if (!memcmp(&g, &ext_stream_audio_stream, sizeof(GUID))) {
                    type = CODEC_TYPE_AUDIO;
                    get_guid(pb, &g);
                    get_le32(pb);
                    get_le32(pb);
                    get_le32(pb);
                    get_guid(pb, &g);
                    get_le32(pb);
                }
            }

246
            st->codec->codec_type = type;
Fabrice Bellard's avatar
Fabrice Bellard committed
247
            if (type == CODEC_TYPE_AUDIO) {
248
                get_wav_header(pb, st->codec, type_specific_size);
249
                st->need_parsing = 1;
250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266
                /* We have to init the frame size at some point .... */
                pos2 = url_ftell(pb);
                if (gsize > (pos2 + 8 - pos1 + 24)) {
                    asf_st->ds_span = get_byte(pb);
                    asf_st->ds_packet_size = get_le16(pb);
                    asf_st->ds_chunk_size = get_le16(pb);
                    asf_st->ds_data_size = get_le16(pb);
                    asf_st->ds_silence_data = get_byte(pb);
                }
                //printf("Descrambling: ps:%d cs:%d ds:%d s:%d  sd:%d\n",
                //       asf_st->ds_packet_size, asf_st->ds_chunk_size,
                //       asf_st->ds_data_size, asf_st->ds_span, asf_st->ds_silence_data);
                if (asf_st->ds_span > 1) {
                    if (!asf_st->ds_chunk_size
                        || (asf_st->ds_packet_size/asf_st->ds_chunk_size <= 1))
                        asf_st->ds_span = 0; // disable descrambling
                }
267
                switch (st->codec->codec_id) {
Fabrice Bellard's avatar
Fabrice Bellard committed
268
                case CODEC_ID_MP3:
269
                    st->codec->frame_size = MPA_FRAME_SIZE;
270 271 272 273 274 275 276 277 278
                    break;
                case CODEC_ID_PCM_S16LE:
                case CODEC_ID_PCM_S16BE:
                case CODEC_ID_PCM_U16LE:
                case CODEC_ID_PCM_U16BE:
                case CODEC_ID_PCM_S8:
                case CODEC_ID_PCM_U8:
                case CODEC_ID_PCM_ALAW:
                case CODEC_ID_PCM_MULAW:
279
                    st->codec->frame_size = 1;
280 281 282
                    break;
                default:
                    /* This is probably wrong, but it prevents a crash later */
283
                    st->codec->frame_size = 1;
284 285
                    break;
                }
286
            } else if (type == CODEC_TYPE_VIDEO) {
287
                get_le32(pb);
Fabrice Bellard's avatar
Fabrice Bellard committed
288 289 290
                get_le32(pb);
                get_byte(pb);
                size = get_le16(pb); /* size */
291
                sizeX= get_le32(pb); /* size */
292
                st->codec->width = get_le32(pb);
293
                st->codec->height = get_le32(pb);
294
                /* not available for asf */
Fabrice Bellard's avatar
Fabrice Bellard committed
295
                get_le16(pb); /* panes */
296
                st->codec->bits_per_sample = get_le16(pb); /* depth */
Fabrice Bellard's avatar
Fabrice Bellard committed
297
                tag1 = get_le32(pb);
298
                url_fskip(pb, 20);
299 300
//                av_log(NULL, AV_LOG_DEBUG, "size:%d tsize:%d sizeX:%d\n", size, total_size, sizeX);
                size= sizeX;
301 302 303 304 305
                if (size > 40) {
                    st->codec->extradata_size = size - 40;
                    st->codec->extradata = av_mallocz(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
                    get_buffer(pb, st->codec->extradata, st->codec->extradata_size);
                }
306 307 308 309

        /* Extract palette from extradata if bpp <= 8 */
        /* This code assumes that extradata contains only palette */
        /* This is true for all paletted codecs implemented in ffmpeg */
310 311
        if (st->codec->extradata_size && (st->codec->bits_per_sample <= 8)) {
            st->codec->palctrl = av_mallocz(sizeof(AVPaletteControl));
312
#ifdef WORDS_BIGENDIAN
313 314
            for (i = 0; i < FFMIN(st->codec->extradata_size, AVPALETTE_SIZE)/4; i++)
                st->codec->palctrl->palette[i] = bswap_32(((uint32_t*)st->codec->extradata)[i]);
315
#else
316 317
            memcpy(st->codec->palctrl->palette, st->codec->extradata,
                   FFMIN(st->codec->extradata_size, AVPALETTE_SIZE));
318
#endif
319
            st->codec->palctrl->palette_changed = 1;
320 321
        }

322
                st->codec->codec_tag = tag1;
323
                st->codec->codec_id = codec_get_id(codec_bmp_tags, tag1);
324 325
                if(tag1 == MKTAG('D', 'V', 'R', ' '))
                    st->need_parsing = 1;
Fabrice Bellard's avatar
Fabrice Bellard committed
326 327 328 329
            }
            pos2 = url_ftell(pb);
            url_fskip(pb, gsize - (pos2 - pos1 + 24));
        } else if (!memcmp(&g, &data_header, sizeof(GUID))) {
330 331 332 333 334 335
            asf->data_object_offset = url_ftell(pb);
            if (gsize != (uint64_t)-1 && gsize >= 24) {
                asf->data_object_size = gsize - 24;
            } else {
                asf->data_object_size = (uint64_t)-1;
            }
Fabrice Bellard's avatar
Fabrice Bellard committed
336 337 338 339 340 341 342 343 344 345 346 347 348
            break;
        } else if (!memcmp(&g, &comment_header, sizeof(GUID))) {
            int len1, len2, len3, len4, len5;

            len1 = get_le16(pb);
            len2 = get_le16(pb);
            len3 = get_le16(pb);
            len4 = get_le16(pb);
            len5 = get_le16(pb);
            get_str16_nolen(pb, len1, s->title, sizeof(s->title));
            get_str16_nolen(pb, len2, s->author, sizeof(s->author));
            get_str16_nolen(pb, len3, s->copyright, sizeof(s->copyright));
            get_str16_nolen(pb, len4, s->comment, sizeof(s->comment));
349
            url_fskip(pb, len5);
350 351 352 353 354 355
       } else if (!memcmp(&g, &extended_content_header, sizeof(GUID))) {
                int desc_count, i;

                desc_count = get_le16(pb);
                for(i=0;i<desc_count;i++)
                {
356 357
                        int name_len,value_type,value_len;
                        uint64_t value_num = 0;
358 359 360
                        char *name, *value;

                        name_len = get_le16(pb);
Zuxy Meng's avatar
Zuxy Meng committed
361 362
                        name = (char *)av_malloc(name_len * 2);
                        get_str16_nolen(pb, name_len, name, name_len * 2);
363 364 365 366
                        value_type = get_le16(pb);
                        value_len = get_le16(pb);
                        if ((value_type == 0) || (value_type == 1)) // unicode or byte
                        {
Zuxy Meng's avatar
Zuxy Meng committed
367 368 369
                                value = (char *)av_malloc(value_len * 2);
                                get_str16_nolen(pb, value_len, value,
                                        value_len * 2);
370
                                if (strcmp(name,"WM/AlbumTitle")==0) { pstrcpy(s->album, sizeof(s->album), value); }
371 372
                                av_free(value);
                        }
Michael Niedermayer's avatar
typo  
Michael Niedermayer committed
373
                        if ((value_type >= 2) && (value_type <= 5)) // boolean or DWORD or QWORD or WORD
374 375 376 377 378 379 380 381 382 383
                        {
                                if (value_type==2) value_num = get_le32(pb);
                                if (value_type==3) value_num = get_le32(pb);
                                if (value_type==4) value_num = get_le64(pb);
                                if (value_type==5) value_num = get_le16(pb);
                                if (strcmp(name,"WM/Track")==0) s->track = value_num + 1;
                                if (strcmp(name,"WM/TrackNumber")==0) s->track = value_num;
                        }
                        av_free(name);
                }
384 385 386
        } else if (!memcmp(&g, &ext_stream_header, sizeof(GUID))) {
            int ext_len, payload_ext_ct, stream_ct;
            uint32_t ext_d;
Måns Rullgård's avatar
Måns Rullgård committed
387
            int64_t pos_ex_st;
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
            pos_ex_st = url_ftell(pb);

            get_le64(pb);
            get_le64(pb);
            get_le32(pb);
            get_le32(pb);
            get_le32(pb);
            get_le32(pb);
            get_le32(pb);
            get_le32(pb);
            get_le32(pb);
            get_le32(pb);
            get_le16(pb);
            get_le16(pb);
            get_le64(pb);
            stream_ct = get_le16(pb);
            payload_ext_ct = get_le16(pb);

            for (i=0; i<stream_ct; i++){
                get_le16(pb);
                ext_len = get_le16(pb);
                url_fseek(pb, ext_len, SEEK_CUR);
            }

            for (i=0; i<payload_ext_ct; i++){
                get_guid(pb, &g);
                ext_d=get_le16(pb);
                ext_len=get_le32(pb);
                url_fseek(pb, ext_len, SEEK_CUR);
            }

            // there could be a optional stream properties object to follow
            // if so the next iteration will pick it up
Fabrice Bellard's avatar
Fabrice Bellard committed
421 422 423 424 425
        } else if (!memcmp(&g, &head1_guid, sizeof(GUID))) {
            int v1, v2;
            get_guid(pb, &g);
            v1 = get_le32(pb);
            v2 = get_le16(pb);
426
#if 0
Fabrice Bellard's avatar
Fabrice Bellard committed
427 428 429 430 431 432 433
        } else if (!memcmp(&g, &codec_comment_header, sizeof(GUID))) {
            int len, v1, n, num;
            char str[256], *q;
            char tag[16];

            get_guid(pb, &g);
            print_guid(&g);
434

Fabrice Bellard's avatar
Fabrice Bellard committed
435 436 437 438 439 440 441 442 443 444 445 446 447 448 449 450 451 452 453 454 455 456 457 458 459 460
            n = get_le32(pb);
            for(i=0;i<n;i++) {
                num = get_le16(pb); /* stream number */
                get_str16(pb, str, sizeof(str));
                get_str16(pb, str, sizeof(str));
                len = get_le16(pb);
                q = tag;
                while (len > 0) {
                    v1 = get_byte(pb);
                    if ((q - tag) < sizeof(tag) - 1)
                        *q++ = v1;
                    len--;
                }
                *q = '\0';
            }
#endif
        } else if (url_feof(pb)) {
            goto fail;
        } else {
            url_fseek(pb, gsize - 24, SEEK_CUR);
        }
    }
    get_guid(pb, &g);
    get_le64(pb);
    get_byte(pb);
    get_byte(pb);
461 462 463
    if (url_feof(pb))
        goto fail;
    asf->data_offset = url_ftell(pb);
Fabrice Bellard's avatar
Fabrice Bellard committed
464 465 466 467 468
    asf->packet_size_left = 0;

    return 0;

 fail:
Zdenek Kabelac's avatar
Zdenek Kabelac committed
469
     for(i=0;i<s->nb_streams;i++) {
Fabrice Bellard's avatar
Fabrice Bellard committed
470
        AVStream *st = s->streams[i];
471 472
        if (st) {
            av_free(st->priv_data);
473
            av_free(st->codec->extradata);
474
        }
475
        av_free(st);
Fabrice Bellard's avatar
Fabrice Bellard committed
476 477 478 479
    }
    return -1;
}

480 481 482 483 484 485 486 487 488
#define DO_2BITS(bits, var, defval) \
    switch (bits & 3) \
    { \
    case 3: var = get_le32(pb); rsize += 4; break; \
    case 2: var = get_le16(pb); rsize += 2; break; \
    case 1: var = get_byte(pb); rsize++; break; \
    default: var = defval; break; \
    }

Fabrice Bellard's avatar
Fabrice Bellard committed
489 490 491 492
static int asf_get_packet(AVFormatContext *s)
{
    ASFContext *asf = s->priv_data;
    ByteIOContext *pb = &s->pb;
493
    uint32_t packet_length, padsize;
Michael Niedermayer's avatar
Michael Niedermayer committed
494 495
    int rsize = 9;
    int c;
496

Michael Niedermayer's avatar
Michael Niedermayer committed
497
    assert((url_ftell(&s->pb) - s->data_offset) % asf->packet_size == 0);
498

Michael Niedermayer's avatar
Michael Niedermayer committed
499
    c = get_byte(pb);
500
    if (c != 0x82) {
501
        if (!url_feof(pb))
502
            av_log(s, AV_LOG_ERROR, "ff asf bad header %x  at:%"PRId64"\n", c, url_ftell(pb));
503 504
    }
    if ((c & 0x0f) == 2) { // always true for now
505
        if (get_le16(pb) != 0) {
506
            if (!url_feof(pb))
507 508 509
                av_log(s, AV_LOG_ERROR, "ff asf bad non zero\n");
            return AVERROR_IO;
        }
Michael Niedermayer's avatar
Michael Niedermayer committed
510 511 512
        rsize+=2;
/*    }else{
        if (!url_feof(pb))
513
            printf("ff asf bad header %x  at:%"PRId64"\n", c, url_ftell(pb));
514
        return AVERROR_IO;*/
Fabrice Bellard's avatar
Fabrice Bellard committed
515
    }
516 517 518 519 520 521 522 523

    asf->packet_flags = get_byte(pb);
    asf->packet_property = get_byte(pb);

    DO_2BITS(asf->packet_flags >> 5, packet_length, asf->packet_size);
    DO_2BITS(asf->packet_flags >> 1, padsize, 0); // sequence ignored
    DO_2BITS(asf->packet_flags >> 3, padsize, 0); // padding length

524 525 526 527 528 529 530 531 532 533
    //the following checks prevent overflows and infinite loops
    if(packet_length >= (1U<<29)){
        av_log(s, AV_LOG_ERROR, "invalid packet_length %d at:%"PRId64"\n", packet_length, url_ftell(pb));
        return 0; // FIXME this should be -1
    }
    if(padsize >= (1U<<29)){
        av_log(s, AV_LOG_ERROR, "invalid padsize %d at:%"PRId64"\n", padsize, url_ftell(pb));
        return 0; // FIXME this should be -1
    }

534
    asf->packet_timestamp = get_le32(pb);
Fabrice Bellard's avatar
Fabrice Bellard committed
535
    get_le16(pb); /* duration */
536
    // rsize has at least 11 bytes which have to be present
537 538

    if (asf->packet_flags & 0x01) {
539
        asf->packet_segsizetype = get_byte(pb); rsize++;
540 541
        asf->packet_segments = asf->packet_segsizetype & 0x3f;
    } else {
542
        asf->packet_segments = 1;
543 544 545
        asf->packet_segsizetype = 0x80;
    }
    asf->packet_size_left = packet_length - padsize - rsize;
546 547
    if (packet_length < asf->hdr.min_pktsize)
        padsize += asf->hdr.min_pktsize - packet_length;
548
    asf->packet_padsize = padsize;
Fabrice Bellard's avatar
Fabrice Bellard committed
549
#ifdef DEBUG
550
    printf("packet: size=%d padsize=%d  left=%d\n", asf->packet_size, asf->packet_padsize, asf->packet_size_left);
Fabrice Bellard's avatar
Fabrice Bellard committed
551 552 553 554 555 556 557
#endif
    return 0;
}

static int asf_read_packet(AVFormatContext *s, AVPacket *pkt)
{
    ASFContext *asf = s->priv_data;
558
    ASFStream *asf_st = 0;
Fabrice Bellard's avatar
Fabrice Bellard committed
559
    ByteIOContext *pb = &s->pb;
560
    //static int pc = 0;
561
    for (;;) {
562 563 564 565 566
        int rsize = 0;
        if (asf->packet_size_left < FRAME_HEADER_SIZE
            || asf->packet_segments < 1) {
            //asf->packet_size_left <= asf->packet_padsize) {
            int ret = asf->packet_size_left + asf->packet_padsize;
567
            //printf("PacketLeftSize:%d  Pad:%d Pos:%"PRId64"\n", asf->packet_size_left, asf->packet_padsize, url_ftell(pb));
568 569
            if((url_ftell(&s->pb) + ret - s->data_offset) % asf->packet_size)
                ret += asf->packet_size - ((url_ftell(&s->pb) + ret - s->data_offset) % asf->packet_size);
570
            assert(ret>=0);
571 572
            /* fail safe */
            url_fskip(pb, ret);
Michael Niedermayer's avatar
Michael Niedermayer committed
573
            asf->packet_pos= url_ftell(&s->pb);
574 575 576
            if (asf->data_object_size != (uint64_t)-1 &&
                (asf->packet_pos - asf->data_object_offset >= asf->data_object_size))
                return AVERROR_IO; /* Do not exceed the size of the data object */
577 578 579 580
            ret = asf_get_packet(s);
            //printf("READ ASF PACKET  %d   r:%d   c:%d\n", ret, asf->packet_size_left, pc++);
            if (ret < 0 || url_feof(pb))
                return AVERROR_IO;
581
            asf->packet_time_start = 0;
582
            continue;
583 584 585
        }
        if (asf->packet_time_start == 0) {
            /* read frame header */
586
            int num = get_byte(pb);
587 588 589 590 591 592 593 594
            asf->packet_segments--;
            rsize++;
            asf->packet_key_frame = (num & 0x80) >> 7;
            asf->stream_index = asf->asfid2avid[num & 0x7f];
            // sequence should be ignored!
            DO_2BITS(asf->packet_property >> 4, asf->packet_seq, 0);
            DO_2BITS(asf->packet_property >> 2, asf->packet_frag_offset, 0);
            DO_2BITS(asf->packet_property, asf->packet_replic_size, 0);
Michael Niedermayer's avatar
Michael Niedermayer committed
595
//printf("key:%d stream:%d seq:%d offset:%d replic_size:%d\n", asf->packet_key_frame, asf->stream_index, asf->packet_seq, //asf->packet_frag_offset, asf->packet_replic_size);
596
            if (asf->packet_replic_size > 1) {
597
                assert(asf->packet_replic_size >= 8);
598
                // it should be always at least 8 bytes - FIXME validate
599 600 601 602 603 604 605 606
                asf->packet_obj_size = get_le32(pb);
                asf->packet_frag_timestamp = get_le32(pb); // timestamp
                if (asf->packet_replic_size > 8)
                    url_fskip(pb, asf->packet_replic_size - 8);
                rsize += asf->packet_replic_size; // FIXME - check validity
            } else if (asf->packet_replic_size==1){
                // multipacket - frag_offset is begining timestamp
                asf->packet_time_start = asf->packet_frag_offset;
607
                asf->packet_frag_offset = 0;
608
                asf->packet_frag_timestamp = asf->packet_timestamp;
609

610
                asf->packet_time_delta = get_byte(pb);
611 612
                rsize++;
            }else{
613 614
                assert(asf->packet_replic_size==0);
            }
615 616
            if (asf->packet_flags & 0x01) {
                DO_2BITS(asf->packet_segsizetype >> 6, asf->packet_frag_size, 0); // 0 is illegal
617
#undef DO_2BITS
618 619 620 621 622 623 624 625 626
                //printf("Fragsize %d\n", asf->packet_frag_size);
            } else {
                asf->packet_frag_size = asf->packet_size_left - rsize;
                //printf("Using rest  %d %d %d\n", asf->packet_frag_size, asf->packet_size_left, rsize);
            }
            if (asf->packet_replic_size == 1) {
                asf->packet_multi_size = asf->packet_frag_size;
                if (asf->packet_multi_size > asf->packet_size_left) {
                    asf->packet_segments = 0;
627
                    continue;
628 629 630 631
                }
            }
            asf->packet_size_left -= rsize;
            //printf("___objsize____  %d   %d    rs:%d\n", asf->packet_obj_size, asf->packet_frag_offset, rsize);
632

633
            if (asf->stream_index < 0
634 635 636
                || s->streams[asf->stream_index]->discard >= AVDISCARD_ALL
                || (!asf->packet_key_frame && s->streams[asf->stream_index]->discard >= AVDISCARD_NONKEY)
                ) {
637
                asf->packet_time_start = 0;
638 639 640
                /* unhandled packet (should not happen) */
                url_fskip(pb, asf->packet_frag_size);
                asf->packet_size_left -= asf->packet_frag_size;
641 642
                if(asf->stream_index < 0)
                    av_log(s, AV_LOG_ERROR, "ff asf skip %d  %d\n", asf->packet_frag_size, num & 0x7f);
643
                continue;
644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675
            }
            asf->asf_st = s->streams[asf->stream_index]->priv_data;
        }
        asf_st = asf->asf_st;

        if ((asf->packet_frag_offset != asf_st->frag_offset
             || (asf->packet_frag_offset
                 && asf->packet_seq != asf_st->seq)) // seq should be ignored
           ) {
            /* cannot continue current packet: free it */
            // FIXME better check if packet was already allocated
            av_log(s, AV_LOG_INFO, "ff asf parser skips: %d - %d     o:%d - %d    %d %d   fl:%d\n",
                   asf_st->pkt.size,
                   asf->packet_obj_size,
                   asf->packet_frag_offset, asf_st->frag_offset,
                   asf->packet_seq, asf_st->seq, asf->packet_frag_size);
            if (asf_st->pkt.size)
                av_free_packet(&asf_st->pkt);
            asf_st->frag_offset = 0;
            if (asf->packet_frag_offset != 0) {
                url_fskip(pb, asf->packet_frag_size);
                av_log(s, AV_LOG_INFO, "ff asf parser skipping %db\n", asf->packet_frag_size);
                asf->packet_size_left -= asf->packet_frag_size;
                continue;
            }
        }
        if (asf->packet_replic_size == 1) {
            // frag_offset is here used as the begining timestamp
            asf->packet_frag_timestamp = asf->packet_time_start;
            asf->packet_time_start += asf->packet_time_delta;
            asf->packet_obj_size = asf->packet_frag_size = get_byte(pb);
            asf->packet_size_left--;
676
            asf->packet_multi_size--;
677 678 679 680 681
            if (asf->packet_multi_size < asf->packet_obj_size)
            {
                asf->packet_time_start = 0;
                url_fskip(pb, asf->packet_multi_size);
                asf->packet_size_left -= asf->packet_multi_size;
682
                continue;
683 684 685 686 687 688 689 690 691 692
            }
            asf->packet_multi_size -= asf->packet_obj_size;
            //printf("COMPRESS size  %d  %d  %d   ms:%d\n", asf->packet_obj_size, asf->packet_frag_timestamp, asf->packet_size_left, asf->packet_multi_size);
        }
        if (asf_st->frag_offset == 0) {
            /* new packet */
            av_new_packet(&asf_st->pkt, asf->packet_obj_size);
            asf_st->seq = asf->packet_seq;
            asf_st->pkt.pts = asf->packet_frag_timestamp;
            asf_st->pkt.stream_index = asf->stream_index;
693 694 695
            asf_st->pkt.pos =
            asf_st->packet_pos= asf->packet_pos;
//printf("new packet: stream:%d key:%d packet_key:%d audio:%d size:%d\n",
Michael Niedermayer's avatar
Michael Niedermayer committed
696
//asf->stream_index, asf->packet_key_frame, asf_st->pkt.flags & PKT_FLAG_KEY,
697
//s->streams[asf->stream_index]->codec->codec_type == CODEC_TYPE_AUDIO, asf->packet_obj_size);
698 699 700 701 702 703 704 705 706 707 708 709
            if (s->streams[asf->stream_index]->codec->codec_type == CODEC_TYPE_AUDIO)
                asf->packet_key_frame = 1;
            if (asf->packet_key_frame)
                asf_st->pkt.flags |= PKT_FLAG_KEY;
        }

        /* read data */
        //printf("READ PACKET s:%d  os:%d  o:%d,%d  l:%d   DATA:%p\n",
        //       asf->packet_size, asf_st->pkt.size, asf->packet_frag_offset,
        //       asf_st->frag_offset, asf->packet_frag_size, asf_st->pkt.data);
        asf->packet_size_left -= asf->packet_frag_size;
        if (asf->packet_size_left < 0)
710
            continue;
711 712 713 714 715 716 717 718
        get_buffer(pb, asf_st->pkt.data + asf->packet_frag_offset,
                   asf->packet_frag_size);
        asf_st->frag_offset += asf->packet_frag_size;
        /* test if whole packet is read */
        if (asf_st->frag_offset == asf_st->pkt.size) {
            /* return packet */
            if (asf_st->ds_span > 1) {
                /* packet descrambling */
Måns Rullgård's avatar
Måns Rullgård committed
719
                uint8_t *newdata = av_malloc(asf_st->pkt.size);
720 721 722 723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740 741 742 743
                if (newdata) {
                    int offset = 0;
                    while (offset < asf_st->pkt.size) {
                        int off = offset / asf_st->ds_chunk_size;
                        int row = off / asf_st->ds_span;
                        int col = off % asf_st->ds_span;
                        int idx = row + col * asf_st->ds_packet_size / asf_st->ds_chunk_size;
                        //printf("off:%d  row:%d  col:%d  idx:%d\n", off, row, col, idx);
                        memcpy(newdata + offset,
                               asf_st->pkt.data + idx * asf_st->ds_chunk_size,
                               asf_st->ds_chunk_size);
                        offset += asf_st->ds_chunk_size;
                    }
                    av_free(asf_st->pkt.data);
                    asf_st->pkt.data = newdata;
                }
            }
            asf_st->frag_offset = 0;
            memcpy(pkt, &asf_st->pkt, sizeof(AVPacket));
            //printf("packet %d %d\n", asf_st->pkt.size, asf->packet_frag_size);
            asf_st->pkt.size = 0;
            asf_st->pkt.data = 0;
            break; // packet completed
        }
Fabrice Bellard's avatar
Fabrice Bellard committed
744 745 746 747 748 749 750 751 752
    }
    return 0;
}

static int asf_read_close(AVFormatContext *s)
{
    int i;

    for(i=0;i<s->nb_streams;i++) {
753 754
        AVStream *st = s->streams[i];
        av_free(st->priv_data);
755
    av_free(st->codec->palctrl);
Fabrice Bellard's avatar
Fabrice Bellard committed
756 757 758 759
    }
    return 0;
}

760 761 762 763 764 765
// Added to support seeking after packets have been read
// If information is not reset, read_packet fails due to
// leftover information from previous reads
static void asf_reset_header(AVFormatContext *s)
{
    ASFContext *asf = s->priv_data;
Michael Niedermayer's avatar
Michael Niedermayer committed
766 767
    ASFStream *asf_st;
    int i;
768 769 770 771 772 773 774 775 776 777 778 779 780 781 782 783 784 785 786 787 788 789

    asf->packet_nb_frames = 0;
    asf->packet_timestamp_start = -1;
    asf->packet_timestamp_end = -1;
    asf->packet_size_left = 0;
    asf->packet_segments = 0;
    asf->packet_flags = 0;
    asf->packet_property = 0;
    asf->packet_timestamp = 0;
    asf->packet_segsizetype = 0;
    asf->packet_segments = 0;
    asf->packet_seq = 0;
    asf->packet_replic_size = 0;
    asf->packet_key_frame = 0;
    asf->packet_padsize = 0;
    asf->packet_frag_offset = 0;
    asf->packet_frag_size = 0;
    asf->packet_frag_timestamp = 0;
    asf->packet_multi_size = 0;
    asf->packet_obj_size = 0;
    asf->packet_time_delta = 0;
    asf->packet_time_start = 0;
790

Michael Niedermayer's avatar
Michael Niedermayer committed
791 792 793 794 795 796 797
    for(i=0; i<s->nb_streams; i++){
        asf_st= s->streams[i]->priv_data;
        av_free_packet(&asf_st->pkt);
        asf_st->frag_offset=0;
        asf_st->seq=0;
    }
    asf->asf_st= NULL;
798 799
}

800
static int64_t asf_read_pts(AVFormatContext *s, int stream_index, int64_t *ppos, int64_t pos_limit)
801 802 803
{
    ASFContext *asf = s->priv_data;
    AVPacket pkt1, *pkt = &pkt1;
Michael Niedermayer's avatar
Michael Niedermayer committed
804
    ASFStream *asf_st;
Michael Niedermayer's avatar
Michael Niedermayer committed
805
    int64_t pts;
Michael Niedermayer's avatar
Michael Niedermayer committed
806
    int64_t pos= *ppos;
807 808
    int i;
    int64_t start_pos[s->nb_streams];
809

810 811 812
    for(i=0; i<s->nb_streams; i++){
        start_pos[i]= pos;
    }
813

814 815 816
    pos= (pos+asf->packet_size-1-s->data_offset)/asf->packet_size*asf->packet_size+ s->data_offset;
    *ppos= pos;
    url_fseek(&s->pb, pos, SEEK_SET);
817

Michael Niedermayer's avatar
Michael Niedermayer committed
818 819
//printf("asf_read_pts\n");
    asf_reset_header(s);
820
    for(;;){
Michael Niedermayer's avatar
Michael Niedermayer committed
821
        if (av_read_frame(s, pkt) < 0){
822
            av_log(s, AV_LOG_INFO, "seek failed\n");
823
            return AV_NOPTS_VALUE;
Michael Niedermayer's avatar
Michael Niedermayer committed
824
        }
825

Michael Niedermayer's avatar
Michael Niedermayer committed
826
        pts= pkt->pts;
Michael Niedermayer's avatar
Michael Niedermayer committed
827 828

        av_free_packet(pkt);
829 830 831 832 833 834
        if(pkt->flags&PKT_FLAG_KEY){
            i= pkt->stream_index;

            asf_st= s->streams[i]->priv_data;

            assert((asf_st->packet_pos - s->data_offset) % asf->packet_size == 0);
835
            pos= asf_st->packet_pos;
836

837
            av_add_index_entry(s->streams[i], pos, pts, pkt->size, pos - start_pos[i] + 1, AVINDEX_KEYFRAME);
838
            start_pos[i]= asf_st->packet_pos + 1;
839

840 841 842 843 844 845
            if(pkt->stream_index == stream_index)
               break;
        }
    }

    *ppos= pos;
846
//printf("found keyframe at %"PRId64" stream %d stamp:%"PRId64"\n", *ppos, stream_index, pts);
Michael Niedermayer's avatar
Michael Niedermayer committed
847 848

    return pts;
849 850
}

851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869
static void asf_build_simple_index(AVFormatContext *s, int stream_index)
{
    GUID g;
    ASFContext *asf = s->priv_data;
    int64_t gsize, itime;
    int64_t pos, current_pos, index_pts;
    int i;
    int pct,ict;

    current_pos = url_ftell(&s->pb);

    url_fseek(&s->pb, asf->data_object_offset + asf->data_object_size, SEEK_SET);
    get_guid(&s->pb, &g);
    if (!memcmp(&g, &index_guid, sizeof(GUID))) {
        gsize = get_le64(&s->pb);
        get_guid(&s->pb, &g);
        itime=get_le64(&s->pb);
        pct=get_le32(&s->pb);
        ict=get_le32(&s->pb);
870
        av_log(NULL, AV_LOG_DEBUG, "itime:0x%"PRIx64", pct:%d, ict:%d\n",itime,pct,ict);
871 872 873 874 875 876 877 878 879 880 881

        for (i=0;i<ict;i++){
            int pktnum=get_le32(&s->pb);
            int pktct =get_le16(&s->pb);
            av_log(NULL, AV_LOG_DEBUG, "pktnum:%d, pktct:%d\n", pktnum, pktct);

            pos=s->data_offset + asf->packet_size*(int64_t)pktnum;
            index_pts=av_rescale(itime, i, 10000);

            av_add_index_entry(s->streams[stream_index], pos, index_pts, asf->packet_size, 0, AVINDEX_KEYFRAME);
        }
882
        asf->index_read= 1;
883 884 885 886
    }
    url_fseek(&s->pb, current_pos, SEEK_SET);
}

887
static int asf_read_seek(AVFormatContext *s, int stream_index, int64_t pts, int flags)
888
{
889
    ASFContext *asf = s->priv_data;
890 891 892
    AVStream *st = s->streams[stream_index];
    int64_t pos;
    int index;
893

Michael Niedermayer's avatar
unused  
Michael Niedermayer committed
894
    if (asf->packet_size <= 0)
895 896
        return -1;

897
    if (!asf->index_read)
898 899
        asf_build_simple_index(s, stream_index);

900
    if(!(asf->index_read && st->index_entries)){
901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933
        if(av_seek_frame_binary(s, stream_index, pts, flags)<0)
            return -1;
    }else{
        index= av_index_search_timestamp(st, pts, flags);
        if(index<0)
            return -1;

        /* find the position */
        pos = st->index_entries[index].pos;
        pts = st->index_entries[index].timestamp;

    // various attempts to find key frame have failed so far
    //    asf_reset_header(s);
    //    url_fseek(&s->pb, pos, SEEK_SET);
    //    key_pos = pos;
    //     for(i=0;i<16;i++){
    //         pos = url_ftell(&s->pb);
    //         if (av_read_frame(s, &pkt) < 0){
    //             av_log(s, AV_LOG_INFO, "seek failed\n");
    //             return -1;
    //         }
    //         asf_st = s->streams[stream_index]->priv_data;
    //         pos += st->parser->frame_offset;
    //
    //         if (pkt.size > b) {
    //             b = pkt.size;
    //             key_pos = pos;
    //         }
    //
    //         av_free_packet(&pkt);
    //     }

        /* do the seek */
934
        av_log(NULL, AV_LOG_DEBUG, "SEEKTO: %"PRId64"\n", pos);
935 936
        url_fseek(&s->pb, pos, SEEK_SET);
    }
937 938
    asf_reset_header(s);
    return 0;
939 940
}

941
AVInputFormat asf_demuxer = {
Fabrice Bellard's avatar
Fabrice Bellard committed
942 943 944 945 946 947 948
    "asf",
    "asf format",
    sizeof(ASFContext),
    asf_probe,
    asf_read_header,
    asf_read_packet,
    asf_read_close,
949
    asf_read_seek,
950
    asf_read_pts,
Fabrice Bellard's avatar
Fabrice Bellard committed
951
};