rtsp.c 45.6 KB
Newer Older
1
/*
2
 * RTSP/SDP client
3 4
 * Copyright (c) 2002 Fabrice Bellard.
 *
5 6 7
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
8 9
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
10
 * version 2.1 of the License, or (at your option) any later version.
11
 *
12
 * FFmpeg is distributed in the hope that it will be useful,
13 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 FFmpeg; if not, write to the Free Software
19
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20
 */
21 22

#include "libavutil/avstring.h"
23 24
#include "avformat.h"

25
#include <sys/time.h>
26
#include <unistd.h> /* for select() prototype */
27
#include "network.h"
28
#include "rtsp.h"
29

30 31
#include "rtp_internal.h"

32
//#define DEBUG
33
//#define DEBUG_RTP_TCP
34

35 36 37 38 39 40
enum RTSPClientState {
    RTSP_STATE_IDLE,
    RTSP_STATE_PLAYING,
    RTSP_STATE_PAUSED,
};

41 42
typedef struct RTSPState {
    URLContext *rtsp_hd; /* RTSP TCP connexion handle */
43 44
    int nb_rtsp_streams;
    struct RTSPStream **rtsp_streams;
45

46 47
    enum RTSPClientState state;
    int64_t seek_timestamp;
48

49 50
    /* XXX: currently we use unbuffered input */
    //    ByteIOContext rtsp_gb;
51 52 53 54
    int seq;        /* RTSP command sequence number */
    char session_id[512];
    enum RTSPProtocol protocol;
    char last_reply[2048]; /* XXX: allocate ? */
55
    RTPDemuxContext *cur_rtp;
56 57 58
} RTSPState;

typedef struct RTSPStream {
59 60
    URLContext *rtp_handle; /* RTP stream handle */
    RTPDemuxContext *rtp_ctx; /* RTP parse context */
61

62
    int stream_index; /* corresponding stream index, if any. -1 if none (MPEG2TS case) */
63
    int interleaved_min, interleaved_max;  /* interleave ids, if TCP transport */
64 65 66 67 68 69
    char control_url[1024]; /* url for this stream (from SDP) */

    int sdp_port; /* port (from SDP content - not used in RTSP) */
    struct in_addr sdp_ip; /* IP address  (from SDP content - not used in RTSP) */
    int sdp_ttl;  /* IP TTL (from SDP content - not used in RTSP) */
    int sdp_payload_type; /* payload type - only used in SDP */
70
    rtp_payload_data_t rtp_payload_data; /* rtp payload parsing infos from SDP */
71 72 73

    RTPDynamicProtocolHandler *dynamic_handler; ///< Only valid if it's a dynamic protocol. (This is the handler structure)
    void *dynamic_protocol_context; ///< Only valid if it's a dynamic protocol. (This is any private data associated with the dynamic protocol)
74 75
} RTSPStream;

76 77
static int rtsp_read_play(AVFormatContext *s);

78 79 80
/* XXX: currently, the only way to change the protocols consists in
   changing this variable */

81
#if LIBAVFORMAT_VERSION_INT < (53 << 16)
82
int rtsp_default_protocols = (1 << RTSP_PROTOCOL_RTP_UDP);
83
#endif
84

85 86
static int rtsp_probe(AVProbeData *p)
{
87
    if (av_strstart(p->filename, "rtsp:", NULL))
88 89 90 91 92 93
        return AVPROBE_SCORE_MAX;
    return 0;
}

static int redir_isspace(int c)
{
94
    return c == ' ' || c == '\t' || c == '\n' || c == '\r';
95 96 97 98 99 100 101 102 103 104 105
}

static void skip_spaces(const char **pp)
{
    const char *p;
    p = *pp;
    while (redir_isspace(*p))
        p++;
    *pp = p;
}

106
static void get_word_sep(char *buf, int buf_size, const char *sep,
107 108 109 110 111 112
                         const char **pp)
{
    const char *p;
    char *q;

    p = *pp;
113 114
    if (*p == '/')
        p++;
115 116 117 118 119 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
    skip_spaces(&p);
    q = buf;
    while (!strchr(sep, *p) && *p != '\0') {
        if ((q - buf) < buf_size - 1)
            *q++ = *p;
        p++;
    }
    if (buf_size > 0)
        *q = '\0';
    *pp = p;
}

static void get_word(char *buf, int buf_size, const char **pp)
{
    const char *p;
    char *q;

    p = *pp;
    skip_spaces(&p);
    q = buf;
    while (!redir_isspace(*p) && *p != '\0') {
        if ((q - buf) < buf_size - 1)
            *q++ = *p;
        p++;
    }
    if (buf_size > 0)
        *q = '\0';
    *pp = p;
}

145 146
/* parse the rtpmap description: <codec_name>/<clock_rate>[/<other
   params>] */
147
static int sdp_parse_rtpmap(AVCodecContext *codec, RTSPStream *rtsp_st, int payload_type, const char *p)
148 149
{
    char buf[256];
150 151
    int i;
    AVCodec *c;
152
    const char *c_name;
153

154 155
    /* Loop into AVRtpDynamicPayloadTypes[] and AVRtpPayloadTypes[] and
       see if we can handle this kind of payload */
156
    get_word_sep(buf, sizeof(buf), "/", &p);
157
    if (payload_type >= RTP_PT_PRIVATE) {
158 159 160 161 162 163 164 165
        RTPDynamicProtocolHandler *handler= RTPFirstDynamicPayloadHandler;
        while(handler) {
            if (!strcmp(buf, handler->enc_name) && (codec->codec_type == handler->codec_type)) {
                codec->codec_id = handler->codec_id;
                rtsp_st->dynamic_handler= handler;
                if(handler->open) {
                    rtsp_st->dynamic_protocol_context= handler->open();
                }
166 167
                break;
            }
168 169
            handler= handler->next;
        }
170
    } else {
171 172
        /* We are in a standard case ( from http://www.iana.org/assignments/rtp-parameters) */
        /* search into AVRtpPayloadTypes[] */
173
        codec->codec_id = ff_rtp_codec_id(buf, codec->codec_type);
174 175 176 177
    }

    c = avcodec_find_decoder(codec->codec_id);
    if (c && c->name)
178
        c_name = c->name;
179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195
    else
        c_name = (char *)NULL;

    if (c_name) {
        get_word_sep(buf, sizeof(buf), "/", &p);
        i = atoi(buf);
        switch (codec->codec_type) {
            case CODEC_TYPE_AUDIO:
                av_log(codec, AV_LOG_DEBUG, " audio codec set to : %s\n", c_name);
                codec->sample_rate = RTSP_DEFAULT_AUDIO_SAMPLERATE;
                codec->channels = RTSP_DEFAULT_NB_AUDIO_CHANNELS;
                if (i > 0) {
                    codec->sample_rate = i;
                    get_word_sep(buf, sizeof(buf), "/", &p);
                    i = atoi(buf);
                    if (i > 0)
                        codec->channels = i;
196 197
                    // TODO: there is a bug here; if it is a mono stream, and less than 22000Hz, faad upconverts to stereo and twice the
                    //  frequency.  No problem, but the sample rate is being set here by the sdp line.  Upcoming patch forthcoming. (rdm)
198 199 200 201 202 203 204 205 206 207 208
                }
                av_log(codec, AV_LOG_DEBUG, " audio samplerate set to : %i\n", codec->sample_rate);
                av_log(codec, AV_LOG_DEBUG, " audio channels set to : %i\n", codec->channels);
                break;
            case CODEC_TYPE_VIDEO:
                av_log(codec, AV_LOG_DEBUG, " video codec set to : %s\n", c_name);
                break;
            default:
                break;
        }
        return 0;
209
    }
210 211

    return -1;
212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242
}

/* return the length and optionnaly the data */
static int hex_to_data(uint8_t *data, const char *p)
{
    int c, len, v;

    len = 0;
    v = 1;
    for(;;) {
        skip_spaces(&p);
        if (p == '\0')
            break;
        c = toupper((unsigned char)*p++);
        if (c >= '0' && c <= '9')
            c = c - '0';
        else if (c >= 'A' && c <= 'F')
            c = c - 'A' + 10;
        else
            break;
        v = (v << 4) | c;
        if (v & 0x100) {
            if (data)
                data[len] = v;
            len++;
            v = 1;
        }
    }
    return len;
}

243 244 245 246
static void sdp_parse_fmtp_config(AVCodecContext *codec, char *attr, char *value)
{
    switch (codec->codec_id) {
        case CODEC_ID_MPEG4:
247
        case CODEC_ID_AAC:
248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265
            if (!strcmp(attr, "config")) {
                /* decode the hexa encoded parameter */
                int len = hex_to_data(NULL, value);
                codec->extradata = av_mallocz(len + FF_INPUT_BUFFER_PADDING_SIZE);
                if (!codec->extradata)
                    return;
                codec->extradata_size = len;
                hex_to_data(codec->extradata, value);
            }
            break;
        default:
            break;
    }
    return;
}

typedef struct attrname_map
{
266
    const char *str;
267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284
    uint16_t type;
    uint32_t offset;
} attrname_map_t;

/* All known fmtp parmeters and the corresping RTPAttrTypeEnum */
#define ATTR_NAME_TYPE_INT 0
#define ATTR_NAME_TYPE_STR 1
static attrname_map_t attr_names[]=
{
    {"SizeLength",       ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, sizelength)},
    {"IndexLength",      ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, indexlength)},
    {"IndexDeltaLength", ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, indexdeltalength)},
    {"profile-level-id", ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, profile_level_id)},
    {"StreamType",       ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, streamtype)},
    {"mode",             ATTR_NAME_TYPE_STR, offsetof(rtp_payload_data_t, mode)},
    {NULL, -1, -1},
};

285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303
/** parse the attribute line from the fmtp a line of an sdp resonse.  This is broken out as a function
* because it is used in rtp_h264.c, which is forthcoming.
*/
int rtsp_next_attr_and_value(const char **p, char *attr, int attr_size, char *value, int value_size)
{
    skip_spaces(p);
    if(**p)
    {
        get_word_sep(attr, attr_size, "=", p);
        if (**p == '=')
            (*p)++;
        get_word_sep(value, value_size, ";", p);
        if (**p == ';')
            (*p)++;
        return 1;
    }
    return 0;
}

304 305
/* parse a SDP line and save stream attributes */
static void sdp_parse_fmtp(AVStream *st, const char *p)
306 307 308
{
    char attr[256];
    char value[4096];
309 310 311
    int i;

    RTSPStream *rtsp_st = st->priv_data;
312
    AVCodecContext *codec = st->codec;
313
    rtp_payload_data_t *rtp_payload_data = &rtsp_st->rtp_payload_data;
314 315

    /* loop on each attribute */
316 317
    while(rtsp_next_attr_and_value(&p, attr, sizeof(attr), value, sizeof(value)))
    {
318
        /* grab the codec extra_data from the config parameter of the fmtp line */
319 320 321 322 323 324 325 326
        sdp_parse_fmtp_config(codec, attr, value);
        /* Looking for a known attribute */
        for (i = 0; attr_names[i].str; ++i) {
            if (!strcasecmp(attr, attr_names[i].str)) {
                if (attr_names[i].type == ATTR_NAME_TYPE_INT)
                    *(int *)((char *)rtp_payload_data + attr_names[i].offset) = atoi(value);
                else if (attr_names[i].type == ATTR_NAME_TYPE_STR)
                    *(char **)((char *)rtp_payload_data + attr_names[i].offset) = av_strdup(value);
327
            }
328 329 330 331
        }
    }
}

332 333 334 335 336 337 338 339 340
/** Parse a string \p in the form of Range:npt=xx-xx, and determine the start
 *  and end time.
 *  Used for seeking in the rtp stream.
 */
static void rtsp_parse_range_npt(const char *p, int64_t *start, int64_t *end)
{
    char buf[256];

    skip_spaces(&p);
341
    if (!av_stristart(p, "npt=", &p))
342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357
        return;

    *start = AV_NOPTS_VALUE;
    *end = AV_NOPTS_VALUE;

    get_word_sep(buf, sizeof(buf), "-", &p);
    *start = parse_date(buf, 1);
    if (*p == '-') {
        p++;
        get_word_sep(buf, sizeof(buf), "-", &p);
        *end = parse_date(buf, 1);
    }
//    av_log(NULL, AV_LOG_DEBUG, "Range Start: %lld\n", *start);
//    av_log(NULL, AV_LOG_DEBUG, "Range End: %lld\n", *end);
}

358 359 360 361 362 363 364
typedef struct SDPParseState {
    /* SDP only */
    struct in_addr default_ip;
    int default_ttl;
} SDPParseState;

static void sdp_parse_line(AVFormatContext *s, SDPParseState *s1,
365 366
                           int letter, const char *buf)
{
367
    RTSPState *rt = s->priv_data;
368 369
    char buf1[64], st_type[64];
    const char *p;
370
    int codec_type, payload_type, i;
371 372
    AVStream *st;
    RTSPStream *rtsp_st;
373 374 375
    struct in_addr sdp_ip;
    int ttl;

376 377 378 379 380 381
#ifdef DEBUG
    printf("sdp: %c='%s'\n", letter, buf);
#endif

    p = buf;
    switch(letter) {
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
    case 'c':
        get_word(buf1, sizeof(buf1), &p);
        if (strcmp(buf1, "IN") != 0)
            return;
        get_word(buf1, sizeof(buf1), &p);
        if (strcmp(buf1, "IP4") != 0)
            return;
        get_word_sep(buf1, sizeof(buf1), "/", &p);
        if (inet_aton(buf1, &sdp_ip) == 0)
            return;
        ttl = 16;
        if (*p == '/') {
            p++;
            get_word_sep(buf1, sizeof(buf1), "/", &p);
            ttl = atoi(buf1);
        }
        if (s->nb_streams == 0) {
            s1->default_ip = sdp_ip;
            s1->default_ttl = ttl;
        } else {
            st = s->streams[s->nb_streams - 1];
            rtsp_st = st->priv_data;
            rtsp_st->sdp_ip = sdp_ip;
            rtsp_st->sdp_ttl = ttl;
        }
        break;
408
    case 's':
409
        av_strlcpy(s->title, p, sizeof(s->title));
410 411 412
        break;
    case 'i':
        if (s->nb_streams == 0) {
413
            av_strlcpy(s->comment, p, sizeof(s->comment));
414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429
            break;
        }
        break;
    case 'm':
        /* new stream */
        get_word(st_type, sizeof(st_type), &p);
        if (!strcmp(st_type, "audio")) {
            codec_type = CODEC_TYPE_AUDIO;
        } else if (!strcmp(st_type, "video")) {
            codec_type = CODEC_TYPE_VIDEO;
        } else {
            return;
        }
        rtsp_st = av_mallocz(sizeof(RTSPStream));
        if (!rtsp_st)
            return;
430 431
        rtsp_st->stream_index = -1;
        dynarray_add(&rt->rtsp_streams, &rt->nb_rtsp_streams, rtsp_st);
432 433 434 435 436 437 438 439

        rtsp_st->sdp_ip = s1->default_ip;
        rtsp_st->sdp_ttl = s1->default_ttl;

        get_word(buf1, sizeof(buf1), &p); /* port */
        rtsp_st->sdp_port = atoi(buf1);

        get_word(buf1, sizeof(buf1), &p); /* protocol (ignored) */
440

441 442 443 444
        /* XXX: handle list of formats */
        get_word(buf1, sizeof(buf1), &p); /* format list */
        rtsp_st->sdp_payload_type = atoi(buf1);

445
        if (!strcmp(ff_rtp_enc_name(rtsp_st->sdp_payload_type), "MP2T")) {
446 447 448 449 450 451 452
            /* no corresponding stream */
        } else {
            st = av_new_stream(s, 0);
            if (!st)
                return;
            st->priv_data = rtsp_st;
            rtsp_st->stream_index = st->index;
453
            st->codec->codec_type = codec_type;
454
            if (rtsp_st->sdp_payload_type < RTP_PT_PRIVATE) {
455
                /* if standard payload type, we can find the codec right now */
456
                rtp_get_codec_info(st->codec, rtsp_st->sdp_payload_type);
457 458
            }
        }
459
        /* put a default control url */
460
        av_strlcpy(rtsp_st->control_url, s->filename, sizeof(rtsp_st->control_url));
461 462
        break;
    case 'a':
463
        if (av_strstart(p, "control:", &p) && s->nb_streams > 0) {
464 465 466 467
            char proto[32];
            /* get the control url */
            st = s->streams[s->nb_streams - 1];
            rtsp_st = st->priv_data;
468

469
            /* XXX: may need to add full url resolution */
470
            url_split(proto, sizeof(proto), NULL, 0, NULL, 0, NULL, NULL, 0, p);
471 472
            if (proto[0] == '\0') {
                /* relative control URL */
473 474
                av_strlcat(rtsp_st->control_url, "/", sizeof(rtsp_st->control_url));
                av_strlcat(rtsp_st->control_url, p,   sizeof(rtsp_st->control_url));
475
            } else {
476
                av_strlcpy(rtsp_st->control_url, p,   sizeof(rtsp_st->control_url));
477
            }
478
        } else if (av_strstart(p, "rtpmap:", &p)) {
479
            /* NOTE: rtpmap is only supported AFTER the 'm=' tag */
480
            get_word(buf1, sizeof(buf1), &p);
481 482 483 484 485
            payload_type = atoi(buf1);
            for(i = 0; i < s->nb_streams;i++) {
                st = s->streams[i];
                rtsp_st = st->priv_data;
                if (rtsp_st->sdp_payload_type == payload_type) {
486
                    sdp_parse_rtpmap(st->codec, rtsp_st, payload_type, p);
487 488
                }
            }
489
        } else if (av_strstart(p, "fmtp:", &p)) {
490
            /* NOTE: fmtp is only supported AFTER the 'a=rtpmap:xxx' tag */
491
            get_word(buf1, sizeof(buf1), &p);
492 493 494 495 496
            payload_type = atoi(buf1);
            for(i = 0; i < s->nb_streams;i++) {
                st = s->streams[i];
                rtsp_st = st->priv_data;
                if (rtsp_st->sdp_payload_type == payload_type) {
497 498 499 500 501
                    if(rtsp_st->dynamic_handler && rtsp_st->dynamic_handler->parse_sdp_a_line) {
                        if(!rtsp_st->dynamic_handler->parse_sdp_a_line(st, rtsp_st->dynamic_protocol_context, buf)) {
                            sdp_parse_fmtp(st, p);
                        }
                    } else {
502
                        sdp_parse_fmtp(st, p);
503
                    }
504 505
                }
            }
506
        } else if(av_strstart(p, "framesize:", &p)) {
507 508 509 510 511 512 513 514 515 516 517 518
            // let dynamic protocol handlers have a stab at the line.
            get_word(buf1, sizeof(buf1), &p);
            payload_type = atoi(buf1);
            for(i = 0; i < s->nb_streams;i++) {
                st = s->streams[i];
                rtsp_st = st->priv_data;
                if (rtsp_st->sdp_payload_type == payload_type) {
                    if(rtsp_st->dynamic_handler && rtsp_st->dynamic_handler->parse_sdp_a_line) {
                        rtsp_st->dynamic_handler->parse_sdp_a_line(st, rtsp_st->dynamic_protocol_context, buf);
                    }
                }
            }
519
        } else if(av_strstart(p, "range:", &p)) {
520 521 522 523 524 525
            int64_t start, end;

            // this is so that seeking on a streamed file can work.
            rtsp_parse_range_npt(p, &start, &end);
            s->start_time= start;
            s->duration= (end==AV_NOPTS_VALUE)?AV_NOPTS_VALUE:end-start; // AV_NOPTS_VALUE means live broadcast (and can't seek)
526 527 528 529 530
        }
        break;
    }
}

531
static int sdp_parse(AVFormatContext *s, const char *content)
532 533 534 535
{
    const char *p;
    int letter;
    char buf[1024], *q;
536
    SDPParseState sdp_parse_state, *s1 = &sdp_parse_state;
537

538
    memset(s1, 0, sizeof(SDPParseState));
539 540 541 542 543 544 545 546 547 548 549 550
    p = content;
    for(;;) {
        skip_spaces(&p);
        letter = *p;
        if (letter == '\0')
            break;
        p++;
        if (*p != '=')
            goto next_line;
        p++;
        /* get the content */
        q = buf;
551
        while (*p != '\n' && *p != '\r' && *p != '\0') {
552 553 554 555 556
            if ((q - buf) < sizeof(buf) - 1)
                *q++ = *p;
            p++;
        }
        *q = '\0';
557
        sdp_parse_line(s, s1, letter, buf);
558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595
    next_line:
        while (*p != '\n' && *p != '\0')
            p++;
        if (*p == '\n')
            p++;
    }
    return 0;
}

static void rtsp_parse_range(int *min_ptr, int *max_ptr, const char **pp)
{
    const char *p;
    int v;

    p = *pp;
    skip_spaces(&p);
    v = strtol(p, (char **)&p, 10);
    if (*p == '-') {
        p++;
        *min_ptr = v;
        v = strtol(p, (char **)&p, 10);
        *max_ptr = v;
    } else {
        *min_ptr = v;
        *max_ptr = v;
    }
    *pp = p;
}

/* XXX: only one transport specification is parsed */
static void rtsp_parse_transport(RTSPHeader *reply, const char *p)
{
    char transport_protocol[16];
    char profile[16];
    char lower_transport[16];
    char parameter[16];
    RTSPTransportField *th;
    char buf[256];
596

597
    reply->nb_transports = 0;
598

599 600 601 602 603 604 605
    for(;;) {
        skip_spaces(&p);
        if (*p == '\0')
            break;

        th = &reply->transports[reply->nb_transports];

606
        get_word_sep(transport_protocol, sizeof(transport_protocol),
607 608 609
                     "/", &p);
        if (*p == '/')
            p++;
610
        if (!strcasecmp (transport_protocol, "rtp")) {
611 612 613 614 615 616 617 618 619 620
            get_word_sep(profile, sizeof(profile), "/;,", &p);
            lower_transport[0] = '\0';
            /* rtp/avp/<protocol> */
            if (*p == '/') {
                p++;
                get_word_sep(lower_transport, sizeof(lower_transport),
                             ";,", &p);
                }
        } else if (!strcasecmp (transport_protocol, "x-pn-tng")) {
            /* x-pn-tng/<protocol> */
621 622
            get_word_sep(lower_transport, sizeof(lower_transport), "/;,", &p);
            profile[0] = '\0';
623
        }
624
        if (!strcasecmp(lower_transport, "TCP"))
625 626 627
            th->protocol = RTSP_PROTOCOL_RTP_TCP;
        else
            th->protocol = RTSP_PROTOCOL_RTP_UDP;
628

629 630 631 632 633 634 635 636 637 638 639 640 641
        if (*p == ';')
            p++;
        /* get each parameter */
        while (*p != '\0' && *p != ',') {
            get_word_sep(parameter, sizeof(parameter), "=;,", &p);
            if (!strcmp(parameter, "port")) {
                if (*p == '=') {
                    p++;
                    rtsp_parse_range(&th->port_min, &th->port_max, &p);
                }
            } else if (!strcmp(parameter, "client_port")) {
                if (*p == '=') {
                    p++;
642
                    rtsp_parse_range(&th->client_port_min,
643 644 645 646 647
                                     &th->client_port_max, &p);
                }
            } else if (!strcmp(parameter, "server_port")) {
                if (*p == '=') {
                    p++;
648
                    rtsp_parse_range(&th->server_port_min,
649 650 651 652 653
                                     &th->server_port_max, &p);
                }
            } else if (!strcmp(parameter, "interleaved")) {
                if (*p == '=') {
                    p++;
654
                    rtsp_parse_range(&th->interleaved_min,
655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670
                                     &th->interleaved_max, &p);
                }
            } else if (!strcmp(parameter, "multicast")) {
                if (th->protocol == RTSP_PROTOCOL_RTP_UDP)
                    th->protocol = RTSP_PROTOCOL_RTP_UDP_MULTICAST;
            } else if (!strcmp(parameter, "ttl")) {
                if (*p == '=') {
                    p++;
                    th->ttl = strtol(p, (char **)&p, 10);
                }
            } else if (!strcmp(parameter, "destination")) {
                struct in_addr ipaddr;

                if (*p == '=') {
                    p++;
                    get_word_sep(buf, sizeof(buf), ";,", &p);
671
                    if (inet_aton(buf, &ipaddr))
672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692
                        th->destination = ntohl(ipaddr.s_addr);
                }
            }
            while (*p != ';' && *p != '\0' && *p != ',')
                p++;
            if (*p == ';')
                p++;
        }
        if (*p == ',')
            p++;

        reply->nb_transports++;
    }
}

void rtsp_parse_line(RTSPHeader *reply, const char *buf)
{
    const char *p;

    /* NOTE: we do case independent match for broken servers */
    p = buf;
693
    if (av_stristart(p, "Session:", &p)) {
694
        get_word_sep(reply->session_id, sizeof(reply->session_id), ";", &p);
695
    } else if (av_stristart(p, "Content-Length:", &p)) {
696
        reply->content_length = strtol(p, NULL, 10);
697
    } else if (av_stristart(p, "Transport:", &p)) {
698
        rtsp_parse_transport(reply, p);
699
    } else if (av_stristart(p, "CSeq:", &p)) {
700
        reply->seq = strtol(p, NULL, 10);
701
    } else if (av_stristart(p, "Range:", &p)) {
702
        rtsp_parse_range_npt(p, &reply->range_start, &reply->range_end);
703 704 705
    }
}

706 707 708 709 710 711 712 713 714 715 716 717 718 719
static int url_readbuf(URLContext *h, unsigned char *buf, int size)
{
    int ret, len;

    len = 0;
    while (len < size) {
        ret = url_read(h, buf+len, size-len);
        if (ret < 1)
            return ret;
        len += ret;
    }
    return len;
}

720 721 722 723 724 725 726
/* skip a RTP/TCP interleaved packet */
static void rtsp_skip_packet(AVFormatContext *s)
{
    RTSPState *rt = s->priv_data;
    int ret, len, len1;
    uint8_t buf[1024];

727
    ret = url_readbuf(rt->rtsp_hd, buf, 3);
728 729
    if (ret != 3)
        return;
730
    len = AV_RB16(buf + 1);
731 732 733 734 735 736 737 738
#ifdef DEBUG
    printf("skipping RTP packet len=%d\n", len);
#endif
    /* skip payload */
    while (len > 0) {
        len1 = len;
        if (len1 > sizeof(buf))
            len1 = sizeof(buf);
739
        ret = url_readbuf(rt->rtsp_hd, buf, len1);
740 741 742 743 744
        if (ret != len1)
            return;
        len -= len1;
    }
}
745

746 747
static void rtsp_send_cmd(AVFormatContext *s,
                          const char *cmd, RTSPHeader *reply,
748 749 750 751 752 753 754 755 756 757 758 759
                          unsigned char **content_ptr)
{
    RTSPState *rt = s->priv_data;
    char buf[4096], buf1[1024], *q;
    unsigned char ch;
    const char *p;
    int content_length, line_count;
    unsigned char *content = NULL;

    memset(reply, 0, sizeof(RTSPHeader));

    rt->seq++;
760
    av_strlcpy(buf, cmd, sizeof(buf));
761
    snprintf(buf1, sizeof(buf1), "CSeq: %d\r\n", rt->seq);
762
    av_strlcat(buf, buf1, sizeof(buf));
763
    if (rt->session_id[0] != '\0' && !strstr(cmd, "\nIf-Match:")) {
764
        snprintf(buf1, sizeof(buf1), "Session: %s\r\n", rt->session_id);
765
        av_strlcat(buf, buf1, sizeof(buf));
766
    }
767
    av_strlcat(buf, "\r\n", sizeof(buf));
768 769 770 771 772 773 774 775 776 777 778
#ifdef DEBUG
    printf("Sending:\n%s--\n", buf);
#endif
    url_write(rt->rtsp_hd, buf, strlen(buf));

    /* parse reply (XXX: use buffers) */
    line_count = 0;
    rt->last_reply[0] = '\0';
    for(;;) {
        q = buf;
        for(;;) {
779
            if (url_readbuf(rt->rtsp_hd, &ch, 1) != 1)
780 781 782
                break;
            if (ch == '\n')
                break;
783 784 785 786
            if (ch == '$') {
                /* XXX: only parse it if first char on line ? */
                rtsp_skip_packet(s);
            } else if (ch != '\r') {
787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805
                if ((q - buf) < sizeof(buf) - 1)
                    *q++ = ch;
            }
        }
        *q = '\0';
#ifdef DEBUG
        printf("line='%s'\n", buf);
#endif
        /* test if last line */
        if (buf[0] == '\0')
            break;
        p = buf;
        if (line_count == 0) {
            /* get reply code */
            get_word(buf1, sizeof(buf1), &p);
            get_word(buf1, sizeof(buf1), &p);
            reply->status_code = atoi(buf1);
        } else {
            rtsp_parse_line(reply, p);
806 807
            av_strlcat(rt->last_reply, p,    sizeof(rt->last_reply));
            av_strlcat(rt->last_reply, "\n", sizeof(rt->last_reply));
808 809 810
        }
        line_count++;
    }
811

812
    if (rt->session_id[0] == '\0' && reply->session_id[0] != '\0')
813
        av_strlcpy(rt->session_id, reply->session_id, sizeof(rt->session_id));
814

815 816 817 818
    content_length = reply->content_length;
    if (content_length > 0) {
        /* leave some room for a trailing '\0' (useful for simple parsing) */
        content = av_malloc(content_length + 1);
819
        (void)url_readbuf(rt->rtsp_hd, content, content_length);
820 821 822 823
        content[content_length] = '\0';
    }
    if (content_ptr)
        *content_ptr = content;
824 825
    else
        av_free(content);
826 827 828
}


829 830 831 832 833 834 835 836 837 838 839 840 841
/* close and free RTSP streams */
static void rtsp_close_streams(RTSPState *rt)
{
    int i;
    RTSPStream *rtsp_st;

    for(i=0;i<rt->nb_rtsp_streams;i++) {
        rtsp_st = rt->rtsp_streams[i];
        if (rtsp_st) {
            if (rtsp_st->rtp_ctx)
                rtp_parse_close(rtsp_st->rtp_ctx);
            if (rtsp_st->rtp_handle)
                url_close(rtsp_st->rtp_handle);
842 843
            if (rtsp_st->dynamic_handler && rtsp_st->dynamic_protocol_context)
                rtsp_st->dynamic_handler->close(rtsp_st->dynamic_protocol_context);
844 845 846 847 848
        }
    }
    av_free(rt->rtsp_streams);
}

849 850 851 852
/**
 * @returns 0 on success, <0 on error, 1 if protocol is unavailable.
 */
static int
853
make_setup_request (AVFormatContext *s, const char *host, int port, int protocol)
854 855
{
    RTSPState *rt = s->priv_data;
856
    int j, i, err;
857
    RTSPStream *rtsp_st;
858
    AVStream *st;
859 860
    RTSPHeader reply1, *reply = &reply1;
    char cmd[2048];
861

862 863 864
    /* for each stream, make the setup request */
    /* XXX: we assume the same server is used for the control of each
       RTSP stream */
865 866

    for(j = RTSP_RTP_PORT_MIN, i = 0; i < rt->nb_rtsp_streams; ++i) {
867 868
        char transport[2048];

869
        rtsp_st = rt->rtsp_streams[i];
870 871 872 873 874

        /* compute available transports */
        transport[0] = '\0';

        /* RTP/UDP */
875
        if (protocol == RTSP_PROTOCOL_RTP_UDP) {
876 877 878
            char buf[256];

            /* first try in specified port range */
879 880
            if (RTSP_RTP_PORT_MIN != 0) {
                while(j <= RTSP_RTP_PORT_MAX) {
881
                    snprintf(buf, sizeof(buf), "rtp://%s?localport=%d", host, j);
882
                    j += 2; /* we will use two port by rtp stream (rtp and rtcp) */
883
                    if (url_open(&rtsp_st->rtp_handle, buf, URL_RDWR) == 0) {
884
                        goto rtp_opened;
885
                    }
886
                }
887
            }
888

889 890 891 892 893 894
/*            then try on any port
**            if (url_open(&rtsp_st->rtp_handle, "rtp://", URL_RDONLY) < 0) {
**                err = AVERROR_INVALIDDATA;
**                goto fail;
**            }
*/
895 896

        rtp_opened:
897
            port = rtp_get_local_port(rtsp_st->rtp_handle);
898
            if (transport[0] != '\0')
899
                av_strlcat(transport, ",", sizeof(transport));
900 901 902 903 904 905
            snprintf(transport + strlen(transport), sizeof(transport) - strlen(transport) - 1,
                     "RTP/AVP/UDP;unicast;client_port=%d-%d",
                     port, port + 1);
        }

        /* RTP/TCP */
906
        else if (protocol == RTSP_PROTOCOL_RTP_TCP) {
907
            if (transport[0] != '\0')
908
                av_strlcat(transport, ",", sizeof(transport));
909 910 911 912
            snprintf(transport + strlen(transport), sizeof(transport) - strlen(transport) - 1,
                     "RTP/AVP/TCP");
        }

913
        else if (protocol == RTSP_PROTOCOL_RTP_UDP_MULTICAST) {
914
            if (transport[0] != '\0')
915
                av_strlcat(transport, ",", sizeof(transport));
916
            snprintf(transport + strlen(transport),
917 918 919
                     sizeof(transport) - strlen(transport) - 1,
                     "RTP/AVP/UDP;multicast");
        }
920
        snprintf(cmd, sizeof(cmd),
921 922
                 "SETUP %s RTSP/1.0\r\n"
                 "Transport: %s\r\n",
923 924
                 rtsp_st->control_url, transport);
        rtsp_send_cmd(s, cmd, reply, NULL);
925 926 927
        if (reply->status_code == 461 /* Unsupported protocol */ && i == 0) {
            err = 1;
            goto fail;
928 929
        } else if (reply->status_code != RTSP_STATUS_OK ||
                   reply->nb_transports != 1) {
930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945
            err = AVERROR_INVALIDDATA;
            goto fail;
        }

        /* XXX: same protocol for all streams is required */
        if (i > 0) {
            if (reply->transports[0].protocol != rt->protocol) {
                err = AVERROR_INVALIDDATA;
                goto fail;
            }
        } else {
            rt->protocol = reply->transports[0].protocol;
        }

        /* close RTP connection if not choosen */
        if (reply->transports[0].protocol != RTSP_PROTOCOL_RTP_UDP &&
946
            (protocol == RTSP_PROTOCOL_RTP_UDP)) {
947 948
            url_close(rtsp_st->rtp_handle);
            rtsp_st->rtp_handle = NULL;
949 950 951 952 953 954 955
        }

        switch(reply->transports[0].protocol) {
        case RTSP_PROTOCOL_RTP_TCP:
            rtsp_st->interleaved_min = reply->transports[0].interleaved_min;
            rtsp_st->interleaved_max = reply->transports[0].interleaved_max;
            break;
956

957 958 959
        case RTSP_PROTOCOL_RTP_UDP:
            {
                char url[1024];
960

961
                /* XXX: also use address if specified */
962
                snprintf(url, sizeof(url), "rtp://%s:%d",
963
                         host, reply->transports[0].server_port_min);
964
                if (rtp_set_remote_url(rtsp_st->rtp_handle, url) < 0) {
965 966 967 968 969 970 971 972
                    err = AVERROR_INVALIDDATA;
                    goto fail;
                }
            }
            break;
        case RTSP_PROTOCOL_RTP_UDP_MULTICAST:
            {
                char url[1024];
973
                struct in_addr in;
974

975
                in.s_addr = htonl(reply->transports[0].destination);
976
                snprintf(url, sizeof(url), "rtp://%s:%d?ttl=%d",
977 978 979
                         inet_ntoa(in),
                         reply->transports[0].port_min,
                         reply->transports[0].ttl);
980
                if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
981 982 983 984 985 986
                    err = AVERROR_INVALIDDATA;
                    goto fail;
                }
            }
            break;
        }
987 988 989 990 991 992
        /* open the RTP context */
        st = NULL;
        if (rtsp_st->stream_index >= 0)
            st = s->streams[rtsp_st->stream_index];
        if (!st)
            s->ctx_flags |= AVFMTCTX_NOHEADER;
993
        rtsp_st->rtp_ctx = rtp_parse_open(s, st, rtsp_st->rtp_handle, rtsp_st->sdp_payload_type, &rtsp_st->rtp_payload_data);
994

995
        if (!rtsp_st->rtp_ctx) {
996
            err = AVERROR(ENOMEM);
997
            goto fail;
998 999 1000 1001 1002
        } else {
            if(rtsp_st->dynamic_handler) {
                rtsp_st->rtp_ctx->dynamic_protocol_context= rtsp_st->dynamic_protocol_context;
                rtsp_st->rtp_ctx->parse_packet= rtsp_st->dynamic_handler->parse_packet;
            }
1003
        }
1004 1005
    }

1006 1007 1008
    return 0;

fail:
1009 1010 1011 1012 1013 1014
    for (i=0; i<rt->nb_rtsp_streams; i++) {
        if (rt->rtsp_streams[i]->rtp_handle) {
            url_close(rt->rtsp_streams[i]->rtp_handle);
            rt->rtsp_streams[i]->rtp_handle = NULL;
        }
    }
1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056
    return err;
}

static int rtsp_read_header(AVFormatContext *s,
                            AVFormatParameters *ap)
{
    RTSPState *rt = s->priv_data;
    char host[1024], path[1024], tcpname[1024], cmd[2048], *option_list, *option;
    URLContext *rtsp_hd;
    int port, ret, err;
    RTSPHeader reply1, *reply = &reply1;
    unsigned char *content = NULL;
    int protocol_mask = 0;

    /* extract hostname and port */
    url_split(NULL, 0, NULL, 0,
              host, sizeof(host), &port, path, sizeof(path), s->filename);
    if (port < 0)
        port = RTSP_DEFAULT_PORT;

    /* search for options */
    option_list = strchr(path, '?');
    if (option_list) {
        /* remove the options from the path */
        *option_list++ = 0;
        while(option_list) {
            /* move the option pointer */
            option = option_list;
            option_list = strchr(option_list, '&');
            if (option_list)
                *(option_list++) = 0;
            /* handle the options */
            if (strcmp(option, "udp") == 0)
                protocol_mask = (1<< RTSP_PROTOCOL_RTP_UDP);
            else if (strcmp(option, "multicast") == 0)
                protocol_mask = (1<< RTSP_PROTOCOL_RTP_UDP_MULTICAST);
            else if (strcmp(option, "tcp") == 0)
                protocol_mask = (1<< RTSP_PROTOCOL_RTP_TCP);
        }
    }

    if (!protocol_mask)
1057
        protocol_mask = (1 << RTSP_PROTOCOL_RTP_LAST) - 1;
1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088

    /* open the tcp connexion */
    snprintf(tcpname, sizeof(tcpname), "tcp://%s:%d", host, port);
    if (url_open(&rtsp_hd, tcpname, URL_RDWR) < 0)
        return AVERROR(EIO);
    rt->rtsp_hd = rtsp_hd;
    rt->seq = 0;

    /* describe the stream */
    snprintf(cmd, sizeof(cmd),
             "DESCRIBE %s RTSP/1.0\r\n"
             "Accept: application/sdp\r\n",
             s->filename);
    rtsp_send_cmd(s, cmd, reply, &content);
    if (!content) {
        err = AVERROR_INVALIDDATA;
        goto fail;
    }
    if (reply->status_code != RTSP_STATUS_OK) {
        err = AVERROR_INVALIDDATA;
        goto fail;
    }

    /* now we got the SDP description, we parse it */
    ret = sdp_parse(s, (const char *)content);
    av_freep(&content);
    if (ret < 0) {
        err = AVERROR_INVALIDDATA;
        goto fail;
    }

1089
    do {
1090
        int protocol = ff_log2_tab[protocol_mask & ~(protocol_mask - 1)];
1091 1092 1093

        err = make_setup_request(s, host, port, protocol);
        if (err < 0)
1094
            goto fail;
1095
        protocol_mask &= ~(1 << protocol);
1096
        if (protocol_mask == 0 && err == 1) {
1097
            err = AVERROR(FF_NETERROR(EPROTONOSUPPORT));
1098 1099 1100
            goto fail;
        }
    } while (err);
1101

1102 1103 1104
    rt->state = RTSP_STATE_IDLE;
    rt->seek_timestamp = 0; /* default is to start stream at position
                               zero */
1105
    if (ap->initial_pause) {
1106 1107 1108 1109
        /* do not start immediately */
    } else {
        if (rtsp_read_play(s) < 0) {
            err = AVERROR_INVALIDDATA;
1110 1111 1112 1113 1114
            goto fail;
        }
    }
    return 0;
 fail:
1115
    rtsp_close_streams(rt);
1116 1117 1118 1119 1120
    av_freep(&content);
    url_close(rt->rtsp_hd);
    return err;
}

1121 1122
static int tcp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
                           uint8_t *buf, int buf_size)
1123 1124
{
    RTSPState *rt = s->priv_data;
1125
    int id, len, i, ret;
1126 1127
    RTSPStream *rtsp_st;

1128 1129 1130
#ifdef DEBUG_RTP_TCP
    printf("tcp_read_packet:\n");
#endif
1131 1132
 redo:
    for(;;) {
1133
        ret = url_readbuf(rt->rtsp_hd, buf, 1);
1134 1135 1136 1137
#ifdef DEBUG_RTP_TCP
        printf("ret=%d c=%02x [%c]\n", ret, buf[0], buf[0]);
#endif
        if (ret != 1)
1138
            return -1;
1139
        if (buf[0] == '$')
1140 1141
            break;
    }
1142
    ret = url_readbuf(rt->rtsp_hd, buf, 3);
1143
    if (ret != 3)
1144
        return -1;
1145
    id = buf[0];
1146
    len = AV_RB16(buf + 1);
1147 1148 1149
#ifdef DEBUG_RTP_TCP
    printf("id=%d len=%d\n", id, len);
#endif
1150
    if (len > buf_size || len < 12)
1151 1152
        goto redo;
    /* get the data */
1153
    ret = url_readbuf(rt->rtsp_hd, buf, len);
1154
    if (ret != len)
1155
        return -1;
1156

1157
    /* find the matching stream */
1158 1159
    for(i = 0; i < rt->nb_rtsp_streams; i++) {
        rtsp_st = rt->rtsp_streams[i];
1160 1161
        if (id >= rtsp_st->interleaved_min &&
            id <= rtsp_st->interleaved_max)
1162 1163 1164 1165
            goto found;
    }
    goto redo;
 found:
1166 1167
    *prtsp_st = rtsp_st;
    return len;
1168 1169
}

1170
static int udp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
1171
                           uint8_t *buf, int buf_size)
1172
{
1173
    RTSPState *rt = s->priv_data;
1174 1175 1176 1177 1178 1179
    RTSPStream *rtsp_st;
    fd_set rfds;
    int fd1, fd2, fd_max, n, i, ret;
    struct timeval tv;

    for(;;) {
1180
        if (url_interrupt_cb())
1181
            return AVERROR(EINTR);
1182 1183
        FD_ZERO(&rfds);
        fd_max = -1;
1184 1185
        for(i = 0; i < rt->nb_rtsp_streams; i++) {
            rtsp_st = rt->rtsp_streams[i];
1186
            /* currently, we cannot probe RTCP handle because of blocking restrictions */
1187
            rtp_get_file_handles(rtsp_st->rtp_handle, &fd1, &fd2);
1188 1189 1190 1191 1192
            if (fd1 > fd_max)
                fd_max = fd1;
            FD_SET(fd1, &rfds);
        }
        tv.tv_sec = 0;
1193
        tv.tv_usec = 100 * 1000;
1194 1195
        n = select(fd_max + 1, &rfds, NULL, NULL, &tv);
        if (n > 0) {
1196 1197 1198
            for(i = 0; i < rt->nb_rtsp_streams; i++) {
                rtsp_st = rt->rtsp_streams[i];
                rtp_get_file_handles(rtsp_st->rtp_handle, &fd1, &fd2);
1199
                if (FD_ISSET(fd1, &rfds)) {
1200 1201 1202
                    ret = url_read(rtsp_st->rtp_handle, buf, buf_size);
                    if (ret > 0) {
                        *prtsp_st = rtsp_st;
1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214
                        return ret;
                    }
                }
            }
        }
    }
}

static int rtsp_read_packet(AVFormatContext *s,
                            AVPacket *pkt)
{
    RTSPState *rt = s->priv_data;
1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230
    RTSPStream *rtsp_st;
    int ret, len;
    uint8_t buf[RTP_MAX_PACKET_LENGTH];

    /* get next frames from the same RTP packet */
    if (rt->cur_rtp) {
        ret = rtp_parse_packet(rt->cur_rtp, pkt, NULL, 0);
        if (ret == 0) {
            rt->cur_rtp = NULL;
            return 0;
        } else if (ret == 1) {
            return 0;
        } else {
            rt->cur_rtp = NULL;
        }
    }
1231

1232 1233
    /* read next RTP packet */
 redo:
1234 1235 1236
    switch(rt->protocol) {
    default:
    case RTSP_PROTOCOL_RTP_TCP:
1237
        len = tcp_read_packet(s, &rtsp_st, buf, sizeof(buf));
1238 1239
        break;
    case RTSP_PROTOCOL_RTP_UDP:
1240 1241
    case RTSP_PROTOCOL_RTP_UDP_MULTICAST:
        len = udp_read_packet(s, &rtsp_st, buf, sizeof(buf));
1242
        if (len >=0 && rtsp_st->rtp_ctx)
1243
            rtp_check_and_send_back_rr(rtsp_st->rtp_ctx, len);
1244 1245
        break;
    }
1246
    if (len < 0)
1247
        return len;
1248 1249 1250 1251 1252 1253 1254 1255
    ret = rtp_parse_packet(rtsp_st->rtp_ctx, pkt, buf, len);
    if (ret < 0)
        goto redo;
    if (ret == 1) {
        /* more packets may follow, so we save the RTP context */
        rt->cur_rtp = rtsp_st->rtp_ctx;
    }
    return 0;
1256 1257
}

1258
static int rtsp_read_play(AVFormatContext *s)
1259
{
1260
    RTSPState *rt = s->priv_data;
1261 1262 1263
    RTSPHeader reply1, *reply = &reply1;
    char cmd[1024];

1264
    av_log(s, AV_LOG_DEBUG, "hello state=%d\n", rt->state);
1265 1266

    if (rt->state == RTSP_STATE_PAUSED) {
1267
        snprintf(cmd, sizeof(cmd),
1268 1269 1270
                 "PLAY %s RTSP/1.0\r\n",
                 s->filename);
    } else {
1271
        snprintf(cmd, sizeof(cmd),
1272 1273 1274 1275 1276
                 "PLAY %s RTSP/1.0\r\n"
                 "Range: npt=%0.3f-\r\n",
                 s->filename,
                 (double)rt->seek_timestamp / AV_TIME_BASE);
    }
1277 1278 1279 1280
    rtsp_send_cmd(s, cmd, reply, NULL);
    if (reply->status_code != RTSP_STATUS_OK) {
        return -1;
    } else {
1281
        rt->state = RTSP_STATE_PLAYING;
1282 1283 1284 1285
        return 0;
    }
}

1286 1287
/* pause the stream */
static int rtsp_read_pause(AVFormatContext *s)
1288
{
1289
    RTSPState *rt = s->priv_data;
1290 1291 1292 1293
    RTSPHeader reply1, *reply = &reply1;
    char cmd[1024];

    rt = s->priv_data;
1294

1295 1296 1297
    if (rt->state != RTSP_STATE_PLAYING)
        return 0;

1298
    snprintf(cmd, sizeof(cmd),
1299
             "PAUSE %s RTSP/1.0\r\n",
1300 1301 1302 1303 1304
             s->filename);
    rtsp_send_cmd(s, cmd, reply, NULL);
    if (reply->status_code != RTSP_STATUS_OK) {
        return -1;
    } else {
1305
        rt->state = RTSP_STATE_PAUSED;
1306 1307 1308 1309
        return 0;
    }
}

1310
static int rtsp_read_seek(AVFormatContext *s, int stream_index,
1311
                          int64_t timestamp, int flags)
1312 1313
{
    RTSPState *rt = s->priv_data;
1314

1315
    rt->seek_timestamp = av_rescale_q(timestamp, s->streams[stream_index]->time_base, AV_TIME_BASE_Q);
1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330
    switch(rt->state) {
    default:
    case RTSP_STATE_IDLE:
        break;
    case RTSP_STATE_PLAYING:
        if (rtsp_read_play(s) != 0)
            return -1;
        break;
    case RTSP_STATE_PAUSED:
        rt->state = RTSP_STATE_IDLE;
        break;
    }
    return 0;
}

1331 1332 1333 1334 1335 1336
static int rtsp_read_close(AVFormatContext *s)
{
    RTSPState *rt = s->priv_data;
    RTSPHeader reply1, *reply = &reply1;
    char cmd[1024];

1337
#if 0
1338 1339 1340 1341
    /* NOTE: it is valid to flush the buffer here */
    if (rt->protocol == RTSP_PROTOCOL_RTP_TCP) {
        url_fclose(&rt->rtsp_gb);
    }
1342
#endif
1343
    snprintf(cmd, sizeof(cmd),
1344
             "TEARDOWN %s RTSP/1.0\r\n",
1345 1346 1347
             s->filename);
    rtsp_send_cmd(s, cmd, reply, NULL);

1348
    rtsp_close_streams(rt);
1349 1350 1351 1352
    url_close(rt->rtsp_hd);
    return 0;
}

1353
#ifdef CONFIG_RTSP_DEMUXER
1354
AVInputFormat rtsp_demuxer = {
1355
    "rtsp",
1356
    NULL_IF_CONFIG_SMALL("RTSP input format"),
1357 1358 1359 1360 1361
    sizeof(RTSPState),
    rtsp_probe,
    rtsp_read_header,
    rtsp_read_packet,
    rtsp_read_close,
1362
    rtsp_read_seek,
1363
    .flags = AVFMT_NOFILE,
1364 1365
    .read_play = rtsp_read_play,
    .read_pause = rtsp_read_pause,
1366
};
1367
#endif
1368

1369
static int sdp_probe(AVProbeData *p1)
1370
{
1371
    const char *p = p1->buf, *p_end = p1->buf + p1->buf_size;
1372 1373

    /* we look for a line beginning "c=IN IP4" */
1374
    while (p < p_end && *p != '\0') {
1375
        if (p + sizeof("c=IN IP4") - 1 < p_end && av_strstart(p, "c=IN IP4", NULL))
1376
            return AVPROBE_SCORE_MAX / 2;
1377 1378 1379

        while(p < p_end - 1 && *p != '\n') p++;
        if (++p >= p_end)
1380 1381 1382 1383
            break;
        if (*p == '\r')
            p++;
    }
1384 1385 1386 1387 1388 1389 1390 1391
    return 0;
}

#define SDP_MAX_SIZE 8192

static int sdp_read_header(AVFormatContext *s,
                           AVFormatParameters *ap)
{
1392
    RTSPState *rt = s->priv_data;
1393 1394 1395 1396
    RTSPStream *rtsp_st;
    int size, i, err;
    char *content;
    char url[1024];
1397
    AVStream *st;
1398 1399 1400 1401

    /* read the whole sdp file */
    /* XXX: better loading */
    content = av_malloc(SDP_MAX_SIZE);
1402
    size = get_buffer(s->pb, content, SDP_MAX_SIZE - 1);
1403 1404 1405 1406 1407 1408 1409 1410 1411 1412
    if (size <= 0) {
        av_free(content);
        return AVERROR_INVALIDDATA;
    }
    content[size] ='\0';

    sdp_parse(s, content);
    av_free(content);

    /* open each RTP stream */
1413 1414
    for(i=0;i<rt->nb_rtsp_streams;i++) {
        rtsp_st = rt->rtsp_streams[i];
1415

1416
        snprintf(url, sizeof(url), "rtp://%s:%d?localport=%d&ttl=%d",
1417
                 inet_ntoa(rtsp_st->sdp_ip),
1418
                 rtsp_st->sdp_port,
1419
                 rtsp_st->sdp_port,
1420
                 rtsp_st->sdp_ttl);
1421
        if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
1422 1423 1424
            err = AVERROR_INVALIDDATA;
            goto fail;
        }
1425 1426 1427 1428 1429 1430
        /* open the RTP context */
        st = NULL;
        if (rtsp_st->stream_index >= 0)
            st = s->streams[rtsp_st->stream_index];
        if (!st)
            s->ctx_flags |= AVFMTCTX_NOHEADER;
1431
        rtsp_st->rtp_ctx = rtp_parse_open(s, st, rtsp_st->rtp_handle, rtsp_st->sdp_payload_type, &rtsp_st->rtp_payload_data);
1432
        if (!rtsp_st->rtp_ctx) {
1433
            err = AVERROR(ENOMEM);
1434
            goto fail;
1435 1436 1437 1438 1439
        } else {
            if(rtsp_st->dynamic_handler) {
                rtsp_st->rtp_ctx->dynamic_protocol_context= rtsp_st->dynamic_protocol_context;
                rtsp_st->rtp_ctx->parse_packet= rtsp_st->dynamic_handler->parse_packet;
            }
1440
        }
1441 1442 1443
    }
    return 0;
 fail:
1444
    rtsp_close_streams(rt);
1445 1446 1447 1448 1449 1450
    return err;
}

static int sdp_read_packet(AVFormatContext *s,
                            AVPacket *pkt)
{
1451
    return rtsp_read_packet(s, pkt);
1452 1453 1454 1455
}

static int sdp_read_close(AVFormatContext *s)
{
1456 1457
    RTSPState *rt = s->priv_data;
    rtsp_close_streams(rt);
1458 1459 1460
    return 0;
}

1461 1462
#ifdef CONFIG_SDP_DEMUXER
AVInputFormat sdp_demuxer = {
1463
    "sdp",
1464
    NULL_IF_CONFIG_SMALL("SDP"),
1465 1466 1467 1468 1469 1470
    sizeof(RTSPState),
    sdp_probe,
    sdp_read_header,
    sdp_read_packet,
    sdp_read_close,
};
1471
#endif
1472

1473
#ifdef CONFIG_REDIR_DEMUXER
1474 1475 1476 1477 1478 1479 1480
/* dummy redirector format (used directly in av_open_input_file now) */
static int redir_probe(AVProbeData *pd)
{
    const char *p;
    p = pd->buf;
    while (redir_isspace(*p))
        p++;
1481 1482
    if (av_strstart(p, "http://", NULL) ||
        av_strstart(p, "rtsp://", NULL))
1483 1484 1485 1486
        return AVPROBE_SCORE_MAX;
    return 0;
}

1487
static int redir_read_header(AVFormatContext *s, AVFormatParameters *ap)
1488 1489 1490 1491
{
    char buf[4096], *q;
    int c;
    AVFormatContext *ic = NULL;
1492
    ByteIOContext *f = s->pb;
1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520

    /* parse each URL and try to open it */
    c = url_fgetc(f);
    while (c != URL_EOF) {
        /* skip spaces */
        for(;;) {
            if (!redir_isspace(c))
                break;
            c = url_fgetc(f);
        }
        if (c == URL_EOF)
            break;
        /* record url */
        q = buf;
        for(;;) {
            if (c == URL_EOF || redir_isspace(c))
                break;
            if ((q - buf) < sizeof(buf) - 1)
                *q++ = c;
            c = url_fgetc(f);
        }
        *q = '\0';
        //printf("URL='%s'\n", buf);
        /* try to open the media file */
        if (av_open_input_file(&ic, buf, NULL, 0, NULL) == 0)
            break;
    }
    if (!ic)
1521
        return AVERROR(EIO);
1522 1523 1524 1525 1526

    *s = *ic;
    url_fclose(f);

    return 0;
1527 1528
}

1529
AVInputFormat redir_demuxer = {
1530
    "redir",
1531
    NULL_IF_CONFIG_SMALL("Redirector format"),
1532 1533
    0,
    redir_probe,
1534
    redir_read_header,
1535 1536 1537
    NULL,
    NULL,
};
1538
#endif