rtsp.c 58.4 KB
Newer Older
1
/*
2
 * RTSP/SDP client
3
 * Copyright (c) 2002 Fabrice Bellard
4
 *
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 23 24
/* needed by inet_aton() */
#define _SVID_SOURCE

25
#include "libavutil/base64.h"
26
#include "libavutil/avstring.h"
27
#include "libavutil/intreadwrite.h"
28 29
#include "avformat.h"

30
#include <sys/time.h>
31
#if HAVE_SYS_SELECT_H
32
#include <sys/select.h>
33
#endif
34
#include <strings.h>
35
#include "network.h"
36
#include "rtsp.h"
37

38
#include "rtpdec.h"
39
#include "rdt.h"
40
#include "rtp_asf.h"
41
#include "rtp_vorbis.h"
R
Ryan Martell 已提交
42

43
//#define DEBUG
F
Fabrice Bellard 已提交
44
//#define DEBUG_RTP_TCP
45

46 47
static int rtsp_read_play(AVFormatContext *s);

R
Ronald S. Bultje 已提交
48
#if LIBAVFORMAT_VERSION_INT < (53 << 16)
49
int rtsp_default_protocols = (1 << RTSP_LOWER_TRANSPORT_UDP);
R
Ronald S. Bultje 已提交
50
#endif
F
Fabrice Bellard 已提交
51

52 53
static int rtsp_probe(AVProbeData *p)
{
54
    if (av_strstart(p->filename, "rtsp:", NULL))
55 56 57 58
        return AVPROBE_SCORE_MAX;
    return 0;
}

59
#define SPACE_CHARS " \t\r\n"
60 61 62
/* we use memchr() instead of strchr() here because strchr() will return
 * the terminating '\0' of SPACE_CHARS instead of NULL if c is '\0'. */
#define redir_isspace(c) memchr(SPACE_CHARS, c, 4)
63 64 65 66 67 68 69 70 71
static void skip_spaces(const char **pp)
{
    const char *p;
    p = *pp;
    while (redir_isspace(*p))
        p++;
    *pp = p;
}

72 73
static void get_word_until_chars(char *buf, int buf_size,
                                 const char *sep, const char **pp)
74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90
{
    const char *p;
    char *q;

    p = *pp;
    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;
}

91 92
static void get_word_sep(char *buf, int buf_size, const char *sep,
                         const char **pp)
93
{
94 95 96
    if (**pp == '/') (*pp)++;
    get_word_until_chars(buf, buf_size, sep, pp);
}
97

98 99 100
static void get_word(char *buf, int buf_size, const char **pp)
{
    get_word_until_chars(buf, buf_size, SPACE_CHARS, pp);
101 102
}

103 104
/* parse the rtpmap description: <codec_name>/<clock_rate>[/<other
   params>] */
R
Ryan Martell 已提交
105
static int sdp_parse_rtpmap(AVCodecContext *codec, RTSPStream *rtsp_st, int payload_type, const char *p)
106 107
{
    char buf[256];
R
Romain Degez 已提交
108 109
    int i;
    AVCodec *c;
110
    const char *c_name;
111

R
Romain Degez 已提交
112 113
    /* Loop into AVRtpDynamicPayloadTypes[] and AVRtpPayloadTypes[] and
       see if we can handle this kind of payload */
114
    get_word_sep(buf, sizeof(buf), "/", &p);
R
Romain Degez 已提交
115
    if (payload_type >= RTP_PT_PRIVATE) {
R
Ryan Martell 已提交
116 117
        RTPDynamicProtocolHandler *handler= RTPFirstDynamicPayloadHandler;
        while(handler) {
118
            if (!strcasecmp(buf, handler->enc_name) && (codec->codec_type == handler->codec_type)) {
R
Ryan Martell 已提交
119 120 121 122 123
                codec->codec_id = handler->codec_id;
                rtsp_st->dynamic_handler= handler;
                if(handler->open) {
                    rtsp_st->dynamic_protocol_context= handler->open();
                }
R
Romain Degez 已提交
124 125
                break;
            }
R
Ryan Martell 已提交
126 127
            handler= handler->next;
        }
128
    } else {
R
Romain Degez 已提交
129 130
        /* We are in a standard case ( from http://www.iana.org/assignments/rtp-parameters) */
        /* search into AVRtpPayloadTypes[] */
131
        codec->codec_id = ff_rtp_codec_id(buf, codec->codec_type);
R
Romain Degez 已提交
132 133 134 135
    }

    c = avcodec_find_decoder(codec->codec_id);
    if (c && c->name)
136
        c_name = c->name;
R
Romain Degez 已提交
137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153
    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;
154 155
                    // 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)
R
Romain Degez 已提交
156 157 158 159 160 161 162 163 164 165 166
                }
                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;
167
    }
R
Romain Degez 已提交
168 169

    return -1;
170 171 172 173 174 175 176 177 178 179 180
}

/* 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);
181
        if (*p == '\0')
182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200
            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;
}

201 202
static void sdp_parse_fmtp_config(AVCodecContext * codec, void *ctx,
                                  char *attr, char *value)
R
Romain Degez 已提交
203 204 205
{
    switch (codec->codec_id) {
        case CODEC_ID_MPEG4:
206
        case CODEC_ID_AAC:
R
Romain Degez 已提交
207 208 209
            if (!strcmp(attr, "config")) {
                /* decode the hexa encoded parameter */
                int len = hex_to_data(NULL, value);
210 211
                if (codec->extradata)
                    av_free(codec->extradata);
R
Romain Degez 已提交
212 213 214 215 216 217 218
                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;
219 220 221
        case CODEC_ID_VORBIS:
            ff_vorbis_parse_fmtp_config(codec, ctx, attr, value);
            break;
R
Romain Degez 已提交
222 223 224 225 226 227
        default:
            break;
    }
    return;
}

L
Luca Barbato 已提交
228
typedef struct {
229
    const char *str;
R
Romain Degez 已提交
230 231
    uint16_t type;
    uint32_t offset;
L
Luca Barbato 已提交
232
} AttrNameMap;
R
Romain Degez 已提交
233 234 235 236

/* All known fmtp parmeters and the corresping RTPAttrTypeEnum */
#define ATTR_NAME_TYPE_INT 0
#define ATTR_NAME_TYPE_STR 1
L
Luca Barbato 已提交
237
static const AttrNameMap attr_names[]=
R
Romain Degez 已提交
238
{
239 240 241 242 243 244
    {"SizeLength",       ATTR_NAME_TYPE_INT, offsetof(RTPPayloadData, sizelength)},
    {"IndexLength",      ATTR_NAME_TYPE_INT, offsetof(RTPPayloadData, indexlength)},
    {"IndexDeltaLength", ATTR_NAME_TYPE_INT, offsetof(RTPPayloadData, indexdeltalength)},
    {"profile-level-id", ATTR_NAME_TYPE_INT, offsetof(RTPPayloadData, profile_level_id)},
    {"StreamType",       ATTR_NAME_TYPE_INT, offsetof(RTPPayloadData, streamtype)},
    {"mode",             ATTR_NAME_TYPE_STR, offsetof(RTPPayloadData, mode)},
R
Romain Degez 已提交
245 246 247
    {NULL, -1, -1},
};

248 249 250 251 252 253
/** 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);
254
    if(**p) {
255 256 257 258 259 260 261 262 263 264 265
        get_word_sep(attr, attr_size, "=", p);
        if (**p == '=')
            (*p)++;
        get_word_sep(value, value_size, ";", p);
        if (**p == ';')
            (*p)++;
        return 1;
    }
    return 0;
}

R
Romain Degez 已提交
266 267
/* parse a SDP line and save stream attributes */
static void sdp_parse_fmtp(AVStream *st, const char *p)
268 269
{
    char attr[256];
270 271 272
    /* Vorbis setup headers can be up to 12KB and are sent base64
     * encoded, giving a 12KB * (4/3) = 16KB FMTP line. */
    char value[16384];
R
Romain Degez 已提交
273 274 275
    int i;

    RTSPStream *rtsp_st = st->priv_data;
276
    AVCodecContext *codec = st->codec;
277
    RTPPayloadData *rtp_payload_data = &rtsp_st->rtp_payload_data;
278 279

    /* loop on each attribute */
280 281
    while(rtsp_next_attr_and_value(&p, attr, sizeof(attr), value, sizeof(value)))
    {
282
        /* grab the codec extra_data from the config parameter of the fmtp line */
283 284
        sdp_parse_fmtp_config(codec, rtsp_st->dynamic_protocol_context,
                              attr, value);
R
Romain Degez 已提交
285 286 287 288 289 290 291
        /* 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);
292
            }
293 294 295 296
        }
    }
}

297 298 299 300 301 302 303 304 305
/** 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);
306
    if (!av_stristart(p, "npt=", &p))
307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322
        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);
}

323 324 325 326
typedef struct SDPParseState {
    /* SDP only */
    struct in_addr default_ip;
    int default_ttl;
327
    int skip_media; ///< set if an unknown m= line occurs
328 329 330
} SDPParseState;

static void sdp_parse_line(AVFormatContext *s, SDPParseState *s1,
331 332
                           int letter, const char *buf)
{
333
    RTSPState *rt = s->priv_data;
334 335
    char buf1[64], st_type[64];
    const char *p;
D
Diego Pettenò 已提交
336 337
    enum CodecType codec_type;
    int payload_type, i;
338 339
    AVStream *st;
    RTSPStream *rtsp_st;
340 341 342
    struct in_addr sdp_ip;
    int ttl;

343
    dprintf(s, "sdp: %c='%s'\n", letter, buf);
344 345

    p = buf;
346 347
    if (s1->skip_media && letter != 'm')
        return;
348
    switch(letter) {
349 350 351 352 353 354 355 356 357 358 359 360 361 362 363 364 365 366 367 368 369 370 371 372 373 374
    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;
375
    case 's':
376
        av_metadata_set(&s->metadata, "title", p);
377 378 379
        break;
    case 'i':
        if (s->nb_streams == 0) {
380
            av_metadata_set(&s->metadata, "comment", p);
381 382 383 384 385
            break;
        }
        break;
    case 'm':
        /* new stream */
386
        s1->skip_media = 0;
387 388 389 390 391
        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;
392 393
        } else if (!strcmp(st_type, "application")) {
            codec_type = CODEC_TYPE_DATA;
394
        } else {
395
            s1->skip_media = 1;
396 397 398 399 400
            return;
        }
        rtsp_st = av_mallocz(sizeof(RTSPStream));
        if (!rtsp_st)
            return;
401 402
        rtsp_st->stream_index = -1;
        dynarray_add(&rt->rtsp_streams, &rt->nb_rtsp_streams, rtsp_st);
403 404 405 406 407 408 409 410

        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) */
411

412 413 414 415
        /* XXX: handle list of formats */
        get_word(buf1, sizeof(buf1), &p); /* format list */
        rtsp_st->sdp_payload_type = atoi(buf1);

416
        if (!strcmp(ff_rtp_enc_name(rtsp_st->sdp_payload_type), "MP2T")) {
417 418 419 420 421 422 423
            /* no corresponding stream */
        } else {
            st = av_new_stream(s, 0);
            if (!st)
                return;
            st->priv_data = rtsp_st;
            rtsp_st->stream_index = st->index;
424
            st->codec->codec_type = codec_type;
R
Romain Degez 已提交
425
            if (rtsp_st->sdp_payload_type < RTP_PT_PRIVATE) {
426
                /* if standard payload type, we can find the codec right now */
427
                ff_rtp_get_codec_info(st->codec, rtsp_st->sdp_payload_type);
428 429
            }
        }
430
        /* put a default control url */
M
Måns Rullgård 已提交
431
        av_strlcpy(rtsp_st->control_url, s->filename, sizeof(rtsp_st->control_url));
432 433
        break;
    case 'a':
434
        if (av_strstart(p, "control:", &p) && s->nb_streams > 0) {
435 436 437 438
            char proto[32];
            /* get the control url */
            st = s->streams[s->nb_streams - 1];
            rtsp_st = st->priv_data;
439

440
            /* XXX: may need to add full url resolution */
441
            url_split(proto, sizeof(proto), NULL, 0, NULL, 0, NULL, NULL, 0, p);
442 443
            if (proto[0] == '\0') {
                /* relative control URL */
M
Måns Rullgård 已提交
444 445
                av_strlcat(rtsp_st->control_url, "/", sizeof(rtsp_st->control_url));
                av_strlcat(rtsp_st->control_url, p,   sizeof(rtsp_st->control_url));
446
            } else {
M
Måns Rullgård 已提交
447
                av_strlcpy(rtsp_st->control_url, p,   sizeof(rtsp_st->control_url));
448
            }
449
        } else if (av_strstart(p, "rtpmap:", &p) && s->nb_streams > 0) {
450
            /* NOTE: rtpmap is only supported AFTER the 'm=' tag */
451
            get_word(buf1, sizeof(buf1), &p);
452
            payload_type = atoi(buf1);
453
            st = s->streams[s->nb_streams - 1];
R
Ronald S. Bultje 已提交
454 455
            rtsp_st = st->priv_data;
            sdp_parse_rtpmap(st->codec, rtsp_st, payload_type, p);
456
        } else if (av_strstart(p, "fmtp:", &p)) {
457
            /* NOTE: fmtp is only supported AFTER the 'a=rtpmap:xxx' tag */
458
            get_word(buf1, sizeof(buf1), &p);
459 460 461 462 463
            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) {
R
Ryan Martell 已提交
464
                    if(rtsp_st->dynamic_handler && rtsp_st->dynamic_handler->parse_sdp_a_line) {
465
                        if(!rtsp_st->dynamic_handler->parse_sdp_a_line(s, i, rtsp_st->dynamic_protocol_context, buf)) {
R
Ryan Martell 已提交
466 467 468
                            sdp_parse_fmtp(st, p);
                        }
                    } else {
469
                        sdp_parse_fmtp(st, p);
R
Ryan Martell 已提交
470
                    }
471 472
                }
            }
473
        } else if(av_strstart(p, "framesize:", &p)) {
474 475 476 477 478 479 480 481
            // 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) {
482
                        rtsp_st->dynamic_handler->parse_sdp_a_line(s, i, rtsp_st->dynamic_protocol_context, buf);
483 484 485
                    }
                }
            }
486
        } else if(av_strstart(p, "range:", &p)) {
487 488 489 490 491 492
            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)
493 494 495
        } else if (av_strstart(p, "IsRealDataType:integer;",&p)) {
            if (atoi(p) == 1)
                rt->transport = RTSP_TRANSPORT_RDT;
496 497 498 499
        } else {
            if (rt->server_type == RTSP_SERVER_WMS)
                ff_wms_parse_sdp_a_line(s, p);
            if (s->nb_streams > 0) {
R
Ronald S. Bultje 已提交
500 501 502 503 504 505 506 507
                if (rt->server_type == RTSP_SERVER_REAL)
                    ff_real_parse_sdp_a_line(s, s->nb_streams - 1, p);

                rtsp_st = s->streams[s->nb_streams - 1]->priv_data;
                if (rtsp_st->dynamic_handler &&
                    rtsp_st->dynamic_handler->parse_sdp_a_line)
                    rtsp_st->dynamic_handler->parse_sdp_a_line(s, s->nb_streams - 1,
                        rtsp_st->dynamic_protocol_context, buf);
508
            }
509 510 511 512 513
        }
        break;
    }
}

514
static int sdp_parse(AVFormatContext *s, const char *content)
515 516 517
{
    const char *p;
    int letter;
518 519 520 521
    /* Some SDP lines, particularly for Realmedia or ASF RTSP streams,
     * contain long SDP lines containing complete ASF Headers (several
     * kB) or arrays of MDPR (RM stream descriptor) headers plus
     * "rulebooks" describing their properties. Therefore, the SDP line
522 523 524 525
     * buffer is large.
     *
     * The Vorbis FMTP line can be up to 16KB - see sdp_parse_fmtp. */
    char buf[16384], *q;
526
    SDPParseState sdp_parse_state, *s1 = &sdp_parse_state;
527

528
    memset(s1, 0, sizeof(SDPParseState));
529 530 531 532 533 534 535 536 537 538 539 540
    p = content;
    for(;;) {
        skip_spaces(&p);
        letter = *p;
        if (letter == '\0')
            break;
        p++;
        if (*p != '=')
            goto next_line;
        p++;
        /* get the content */
        q = buf;
F
Fabrice Bellard 已提交
541
        while (*p != '\n' && *p != '\r' && *p != '\0') {
542 543 544 545 546
            if ((q - buf) < sizeof(buf) - 1)
                *q++ = *p;
            p++;
        }
        *q = '\0';
547
        sdp_parse_line(s, s1, letter, buf);
548 549 550 551 552 553 554 555 556 557 558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577
    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 */
578
static void rtsp_parse_transport(RTSPMessageHeader *reply, const char *p)
579 580 581 582 583 584 585
{
    char transport_protocol[16];
    char profile[16];
    char lower_transport[16];
    char parameter[16];
    RTSPTransportField *th;
    char buf[256];
586

587
    reply->nb_transports = 0;
588

589 590 591 592 593 594 595
    for(;;) {
        skip_spaces(&p);
        if (*p == '\0')
            break;

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

596
        get_word_sep(transport_protocol, sizeof(transport_protocol),
597
                     "/", &p);
598
        if (!strcasecmp (transport_protocol, "rtp")) {
599 600 601 602 603 604
            get_word_sep(profile, sizeof(profile), "/;,", &p);
            lower_transport[0] = '\0';
            /* rtp/avp/<protocol> */
            if (*p == '/') {
                get_word_sep(lower_transport, sizeof(lower_transport),
                             ";,", &p);
605 606 607 608
            }
            th->transport = RTSP_TRANSPORT_RTP;
        } else if (!strcasecmp (transport_protocol, "x-pn-tng") ||
                   !strcasecmp (transport_protocol, "x-real-rdt")) {
609
            /* x-pn-tng/<protocol> */
610 611
            get_word_sep(lower_transport, sizeof(lower_transport), "/;,", &p);
            profile[0] = '\0';
612
            th->transport = RTSP_TRANSPORT_RDT;
613
        }
F
Fabrice Bellard 已提交
614
        if (!strcasecmp(lower_transport, "TCP"))
615
            th->lower_transport = RTSP_LOWER_TRANSPORT_TCP;
616
        else
617
            th->lower_transport = RTSP_LOWER_TRANSPORT_UDP;
618

619 620 621 622 623 624 625 626 627 628 629 630 631
        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++;
632
                    rtsp_parse_range(&th->client_port_min,
633 634 635 636 637
                                     &th->client_port_max, &p);
                }
            } else if (!strcmp(parameter, "server_port")) {
                if (*p == '=') {
                    p++;
638
                    rtsp_parse_range(&th->server_port_min,
639 640 641 642 643
                                     &th->server_port_max, &p);
                }
            } else if (!strcmp(parameter, "interleaved")) {
                if (*p == '=') {
                    p++;
644
                    rtsp_parse_range(&th->interleaved_min,
645 646 647
                                     &th->interleaved_max, &p);
                }
            } else if (!strcmp(parameter, "multicast")) {
648 649
                if (th->lower_transport == RTSP_LOWER_TRANSPORT_UDP)
                    th->lower_transport = RTSP_LOWER_TRANSPORT_UDP_MULTICAST;
650 651 652 653 654 655 656 657 658 659 660
            } 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);
661
                    if (inet_aton(buf, &ipaddr))
662 663 664 665 666 667 668 669 670 671 672 673 674 675 676
                        th->destination = ntohl(ipaddr.s_addr);
                }
            }
            while (*p != ';' && *p != '\0' && *p != ',')
                p++;
            if (*p == ';')
                p++;
        }
        if (*p == ',')
            p++;

        reply->nb_transports++;
    }
}

677
void rtsp_parse_line(RTSPMessageHeader *reply, const char *buf)
678 679 680 681 682
{
    const char *p;

    /* NOTE: we do case independent match for broken servers */
    p = buf;
683
    if (av_stristart(p, "Session:", &p)) {
684
        int t;
685
        get_word_sep(reply->session_id, sizeof(reply->session_id), ";", &p);
686 687 688 689
        if (av_stristart(p, ";timeout=", &p) &&
            (t = strtol(p, NULL, 10)) > 0) {
            reply->timeout = t;
        }
690
    } else if (av_stristart(p, "Content-Length:", &p)) {
691
        reply->content_length = strtol(p, NULL, 10);
692
    } else if (av_stristart(p, "Transport:", &p)) {
693
        rtsp_parse_transport(reply, p);
694
    } else if (av_stristart(p, "CSeq:", &p)) {
695
        reply->seq = strtol(p, NULL, 10);
696
    } else if (av_stristart(p, "Range:", &p)) {
697
        rtsp_parse_range_npt(p, &reply->range_start, &reply->range_end);
698 699 700
    } else if (av_stristart(p, "RealChallenge1:", &p)) {
        skip_spaces(&p);
        av_strlcpy(reply->real_challenge, p, sizeof(reply->real_challenge));
701 702 703
    } else if (av_stristart(p, "Server:", &p)) {
        skip_spaces(&p);
        av_strlcpy(reply->server, p, sizeof(reply->server));
704 705 706
    } else if (av_stristart(p, "Notice:", &p) ||
               av_stristart(p, "X-Notice:", &p)) {
        reply->notice = strtol(p, NULL, 10);
707 708 709
    }
}

710 711 712 713 714 715 716
/* 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];

717
    ret = url_read_complete(rt->rtsp_hd, buf, 3);
718 719
    if (ret != 3)
        return;
720
    len = AV_RB16(buf + 1);
721 722 723

    dprintf(s, "skipping RTP packet len=%d\n", len);

724 725 726 727 728
    /* skip payload */
    while (len > 0) {
        len1 = len;
        if (len1 > sizeof(buf))
            len1 = sizeof(buf);
729
        ret = url_read_complete(rt->rtsp_hd, buf, len1);
730 731 732 733 734
        if (ret != len1)
            return;
        len -= len1;
    }
}
735

736 737 738 739 740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757
/**
 * Read a RTSP message from the server, or prepare to read data
 * packets if we're reading data interleaved over the TCP/RTSP
 * connection as well.
 *
 * @param s RTSP demuxer context
 * @param reply pointer where the RTSP message header will be stored
 * @param content_ptr pointer where the RTSP message body, if any, will
 *                    be stored (length is in \p reply)
 * @param return_on_interleaved_data whether the function may return if we
 *                   encounter a data marker ('$'), which precedes data
 *                   packets over interleaved TCP/RTSP connections. If this
 *                   is set, this function will return 1 after encountering
 *                   a '$'. If it is not set, the function will skip any
 *                   data packets (if they are encountered), until a reply
 *                   has been fully parsed. If no more data is available
 *                   without parsing a reply, it will return an error.
 *
 * @returns 1 if a data packets is ready to be received, -1 on error,
 *          and 0 on success.
 */
static int
758
rtsp_read_reply (AVFormatContext *s, RTSPMessageHeader *reply,
759
                 unsigned char **content_ptr, int return_on_interleaved_data)
760 761 762 763 764
{
    RTSPState *rt = s->priv_data;
    char buf[4096], buf1[1024], *q;
    unsigned char ch;
    const char *p;
765
    int ret, content_length, line_count = 0;
766 767
    unsigned char *content = NULL;

768
    memset(reply, 0, sizeof(*reply));
769 770 771 772 773 774

    /* parse reply (XXX: use buffers) */
    rt->last_reply[0] = '\0';
    for(;;) {
        q = buf;
        for(;;) {
775
            ret = url_read_complete(rt->rtsp_hd, &ch, 1);
776
#ifdef DEBUG_RTP_TCP
777
            dprintf(s, "ret=%d c=%02x [%c]\n", ret, ch, ch);
778 779 780
#endif
            if (ret != 1)
                return -1;
781 782
            if (ch == '\n')
                break;
783 784
            if (ch == '$') {
                /* XXX: only parse it if first char on line ? */
785 786 787
                if (return_on_interleaved_data) {
                    return 1;
                } else
788 789
                rtsp_skip_packet(s);
            } else if (ch != '\r') {
790 791 792 793 794
                if ((q - buf) < sizeof(buf) - 1)
                    *q++ = ch;
            }
        }
        *q = '\0';
795 796 797

        dprintf(s, "line='%s'\n", buf);

798 799 800 801 802 803 804 805 806 807 808
        /* 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);
M
Måns Rullgård 已提交
809 810
            av_strlcat(rt->last_reply, p,    sizeof(rt->last_reply));
            av_strlcat(rt->last_reply, "\n", sizeof(rt->last_reply));
811 812 813
        }
        line_count++;
    }
814

815
    if (rt->session_id[0] == '\0' && reply->session_id[0] != '\0')
M
Måns Rullgård 已提交
816
        av_strlcpy(rt->session_id, reply->session_id, sizeof(rt->session_id));
817

818 819 820 821
    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);
822
        (void)url_read_complete(rt->rtsp_hd, content, content_length);
823 824 825 826
        content[content_length] = '\0';
    }
    if (content_ptr)
        *content_ptr = content;
827 828
    else
        av_free(content);
829

830 831 832 833 834 835 836 837 838 839 840
    /* EOS */
    if (reply->notice == 2101 /* End-of-Stream Reached */      ||
        reply->notice == 2104 /* Start-of-Stream Reached */    ||
        reply->notice == 2306 /* Continuous Feed Terminated */)
        rt->state = RTSP_STATE_IDLE;
    else if (reply->notice >= 4400 && reply->notice < 5500)
        return AVERROR(EIO); /* data or server error */
    else if (reply->notice == 2401 /* Ticket Expired */ ||
             (reply->notice >= 5500 && reply->notice < 5600) /* end of term */ )
        return AVERROR(EPERM);

841
    return 0;
842 843
}

844
static void rtsp_send_cmd_async (AVFormatContext *s,
845 846 847 848 849 850 851 852 853 854 855 856 857 858
                          const char *cmd, RTSPMessageHeader *reply,
                          unsigned char **content_ptr)
{
    RTSPState *rt = s->priv_data;
    char buf[4096], buf1[1024];

    rt->seq++;
    av_strlcpy(buf, cmd, sizeof(buf));
    snprintf(buf1, sizeof(buf1), "CSeq: %d\r\n", rt->seq);
    av_strlcat(buf, buf1, sizeof(buf));
    if (rt->session_id[0] != '\0' && !strstr(cmd, "\nIf-Match:")) {
        snprintf(buf1, sizeof(buf1), "Session: %s\r\n", rt->session_id);
        av_strlcat(buf, buf1, sizeof(buf));
    }
859 860
    if (rt->auth_b64)
        av_strlcatf(buf, sizeof(buf),
861 862
                    "Authorization: Basic %s\r\n",
                    rt->auth_b64);
863
    av_strlcat(buf, "\r\n", sizeof(buf));
864 865 866

    dprintf(s, "Sending:\n%s--\n", buf);

867
    url_write(rt->rtsp_hd, buf, strlen(buf));
868 869 870 871 872 873 874 875
    rt->last_cmd_time = av_gettime();
}

static void rtsp_send_cmd (AVFormatContext *s,
                           const char *cmd, RTSPMessageHeader *reply,
                           unsigned char **content_ptr)
{
    rtsp_send_cmd_async(s, cmd, reply, content_ptr);
876

877
    rtsp_read_reply(s, reply, content_ptr, 0);
878 879
}

880

881 882 883 884 885 886 887 888 889
/* 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) {
890
            if (rtsp_st->transport_priv) {
891
                if (rt->transport == RTSP_TRANSPORT_RDT)
892
                    ff_rdt_parse_close(rtsp_st->transport_priv);
893
                else
894
                    rtp_parse_close(rtsp_st->transport_priv);
895
            }
896 897
            if (rtsp_st->rtp_handle)
                url_close(rtsp_st->rtp_handle);
R
Ryan Martell 已提交
898 899
            if (rtsp_st->dynamic_handler && rtsp_st->dynamic_protocol_context)
                rtsp_st->dynamic_handler->close(rtsp_st->dynamic_protocol_context);
900 901 902
        }
    }
    av_free(rt->rtsp_streams);
903 904 905 906
    if (rt->asf_ctx) {
        av_close_input_stream (rt->asf_ctx);
        rt->asf_ctx = NULL;
    }
907
    av_freep(&rt->auth_b64);
908 909
}

910 911 912
static int
rtsp_open_transport_ctx(AVFormatContext *s, RTSPStream *rtsp_st)
{
913
    RTSPState *rt = s->priv_data;
914 915
    AVStream *st = NULL;

916 917 918 919 920
    /* open the RTP context */
    if (rtsp_st->stream_index >= 0)
        st = s->streams[rtsp_st->stream_index];
    if (!st)
        s->ctx_flags |= AVFMTCTX_NOHEADER;
921 922

    if (rt->transport == RTSP_TRANSPORT_RDT)
923
        rtsp_st->transport_priv = ff_rdt_parse_open(s, st->index,
924 925 926
                                            rtsp_st->dynamic_protocol_context,
                                            rtsp_st->dynamic_handler);
    else
927
        rtsp_st->transport_priv = rtp_parse_open(s, st, rtsp_st->rtp_handle,
R
Ronald S. Bultje 已提交
928 929
                                         rtsp_st->sdp_payload_type,
                                         &rtsp_st->rtp_payload_data);
930

931
    if (!rtsp_st->transport_priv) {
932
         return AVERROR(ENOMEM);
933
    } else if (rt->transport != RTSP_TRANSPORT_RDT) {
934
        if(rtsp_st->dynamic_handler) {
935
            rtp_parse_set_dynamic_protocol(rtsp_st->transport_priv,
936 937
                                           rtsp_st->dynamic_protocol_context,
                                           rtsp_st->dynamic_handler);
938
        }
939
    }
940 941 942 943

    return 0;
}

944 945 946 947
/**
 * @returns 0 on success, <0 on error, 1 if protocol is unavailable.
 */
static int
948
make_setup_request (AVFormatContext *s, const char *host, int port,
949
                    int lower_transport, const char *real_challenge)
950 951
{
    RTSPState *rt = s->priv_data;
952
    int rtx, j, i, err, interleave = 0;
953
    RTSPStream *rtsp_st;
954
    RTSPMessageHeader reply1, *reply = &reply1;
955
    char cmd[2048];
956 957
    const char *trans_pref;

958
    if (rt->transport == RTSP_TRANSPORT_RDT)
959 960 961
        trans_pref = "x-pn-tng";
    else
        trans_pref = "RTP/AVP";
962

963 964 965
    /* default timeout: 1 minute */
    rt->timeout = 60;

966 967 968
    /* for each stream, make the setup request */
    /* XXX: we assume the same server is used for the control of each
       RTSP stream */
R
Romain Degez 已提交
969 970

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

973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993
        /**
         * WMS serves all UDP data over a single connection, the RTX, which
         * isn't necessarily the first in the SDP but has to be the first
         * to be set up, else the second/third SETUP will fail with a 461.
         */
        if (lower_transport == RTSP_LOWER_TRANSPORT_UDP &&
             rt->server_type == RTSP_SERVER_WMS) {
            if (i == 0) {
                /* rtx first */
                for (rtx = 0; rtx < rt->nb_rtsp_streams; rtx++) {
                    int len = strlen(rt->rtsp_streams[rtx]->control_url);
                    if (len >= 4 &&
                        !strcmp(rt->rtsp_streams[rtx]->control_url + len - 4, "/rtx"))
                        break;
                }
                if (rtx == rt->nb_rtsp_streams)
                    return -1; /* no RTX found */
                rtsp_st = rt->rtsp_streams[rtx];
            } else
                rtsp_st = rt->rtsp_streams[i > rtx ? i : i - 1];
        } else
R
Ronald S. Bultje 已提交
994
            rtsp_st = rt->rtsp_streams[i];
995 996

        /* RTP/UDP */
997
        if (lower_transport == RTSP_LOWER_TRANSPORT_UDP) {
F
Fabrice Bellard 已提交
998 999
            char buf[256];

1000 1001 1002 1003 1004
            if (rt->server_type == RTSP_SERVER_WMS && i > 1) {
                port = reply->transports[0].client_port_min;
                goto have_port;
            }

F
Fabrice Bellard 已提交
1005
            /* first try in specified port range */
R
Romain Degez 已提交
1006 1007
            if (RTSP_RTP_PORT_MIN != 0) {
                while(j <= RTSP_RTP_PORT_MAX) {
1008
                    snprintf(buf, sizeof(buf), "rtp://%s?localport=%d", host, j);
1009
                    j += 2; /* we will use two port by rtp stream (rtp and rtcp) */
1010
                    if (url_open(&rtsp_st->rtp_handle, buf, URL_RDWR) == 0) {
F
Fabrice Bellard 已提交
1011
                        goto rtp_opened;
R
Romain Degez 已提交
1012
                    }
F
Fabrice Bellard 已提交
1013
                }
1014
            }
F
Fabrice Bellard 已提交
1015

R
Romain Degez 已提交
1016 1017 1018 1019 1020 1021
/*            then try on any port
**            if (url_open(&rtsp_st->rtp_handle, "rtp://", URL_RDONLY) < 0) {
**                err = AVERROR_INVALIDDATA;
**                goto fail;
**            }
*/
F
Fabrice Bellard 已提交
1022 1023

        rtp_opened:
1024
            port = rtp_get_local_port(rtsp_st->rtp_handle);
1025
        have_port:
1026
            snprintf(transport, sizeof(transport) - 1,
1027 1028 1029 1030 1031
                     "%s/UDP;", trans_pref);
            if (rt->server_type != RTSP_SERVER_REAL)
                av_strlcat(transport, "unicast;", sizeof(transport));
            av_strlcatf(transport, sizeof(transport),
                     "client_port=%d", port);
1032 1033
            if (rt->transport == RTSP_TRANSPORT_RTP &&
                !(rt->server_type == RTSP_SERVER_WMS && i > 0))
1034
                av_strlcatf(transport, sizeof(transport), "-%d", port + 1);
1035 1036 1037
        }

        /* RTP/TCP */
1038
        else if (lower_transport == RTSP_LOWER_TRANSPORT_TCP) {
1039 1040 1041 1042 1043 1044
            /** For WMS streams, the application streams are only used for
             * UDP. When trying to set it up for TCP streams, the server
             * will return an error. Therefore, we skip those streams. */
            if (rt->server_type == RTSP_SERVER_WMS &&
                s->streams[rtsp_st->stream_index]->codec->codec_type == CODEC_TYPE_DATA)
                continue;
1045
            snprintf(transport, sizeof(transport) - 1,
1046 1047 1048 1049 1050 1051 1052
                     "%s/TCP;", trans_pref);
            if (rt->server_type == RTSP_SERVER_WMS)
                av_strlcat(transport, "unicast;", sizeof(transport));
            av_strlcatf(transport, sizeof(transport),
                        "interleaved=%d-%d",
                        interleave, interleave + 1);
            interleave += 2;
1053 1054
        }

1055
        else if (lower_transport == RTSP_LOWER_TRANSPORT_UDP_MULTICAST) {
1056
            snprintf(transport, sizeof(transport) - 1,
1057
                     "%s/UDP;multicast", trans_pref);
1058
        }
1059 1060
        if (rt->server_type == RTSP_SERVER_REAL ||
            rt->server_type == RTSP_SERVER_WMS)
1061
            av_strlcat(transport, ";mode=play", sizeof(transport));
1062
        snprintf(cmd, sizeof(cmd),
F
Fabrice Bellard 已提交
1063 1064
                 "SETUP %s RTSP/1.0\r\n"
                 "Transport: %s\r\n",
1065
                 rtsp_st->control_url, transport);
1066
        if (i == 0 && rt->server_type == RTSP_SERVER_REAL) {
1067 1068 1069 1070 1071 1072 1073 1074
            char real_res[41], real_csum[9];
            ff_rdt_calc_response_and_checksum(real_res, real_csum,
                                              real_challenge);
            av_strlcatf(cmd, sizeof(cmd),
                        "If-Match: %s\r\n"
                        "RealChallenge2: %s, sd=%s\r\n",
                        rt->session_id, real_res, real_csum);
        }
1075
        rtsp_send_cmd(s, cmd, reply, NULL);
1076 1077 1078
        if (reply->status_code == 461 /* Unsupported protocol */ && i == 0) {
            err = 1;
            goto fail;
1079 1080
        } else if (reply->status_code != RTSP_STATUS_OK ||
                   reply->nb_transports != 1) {
1081 1082 1083 1084 1085 1086
            err = AVERROR_INVALIDDATA;
            goto fail;
        }

        /* XXX: same protocol for all streams is required */
        if (i > 0) {
1087 1088
            if (reply->transports[0].lower_transport != rt->lower_transport ||
                reply->transports[0].transport != rt->transport) {
1089 1090 1091 1092
                err = AVERROR_INVALIDDATA;
                goto fail;
            }
        } else {
1093
            rt->lower_transport = reply->transports[0].lower_transport;
1094
            rt->transport = reply->transports[0].transport;
1095 1096 1097
        }

        /* close RTP connection if not choosen */
1098 1099
        if (reply->transports[0].lower_transport != RTSP_LOWER_TRANSPORT_UDP &&
            (lower_transport == RTSP_LOWER_TRANSPORT_UDP)) {
1100 1101
            url_close(rtsp_st->rtp_handle);
            rtsp_st->rtp_handle = NULL;
1102 1103
        }

1104 1105
        switch(reply->transports[0].lower_transport) {
        case RTSP_LOWER_TRANSPORT_TCP:
1106 1107 1108
            rtsp_st->interleaved_min = reply->transports[0].interleaved_min;
            rtsp_st->interleaved_max = reply->transports[0].interleaved_max;
            break;
1109

1110
        case RTSP_LOWER_TRANSPORT_UDP:
1111 1112
            {
                char url[1024];
1113

1114
                /* XXX: also use address if specified */
1115
                snprintf(url, sizeof(url), "rtp://%s:%d",
1116
                         host, reply->transports[0].server_port_min);
1117 1118
                if (!(rt->server_type == RTSP_SERVER_WMS && i > 1) &&
                    rtp_set_remote_url(rtsp_st->rtp_handle, url) < 0) {
1119 1120 1121 1122 1123
                    err = AVERROR_INVALIDDATA;
                    goto fail;
                }
            }
            break;
1124
        case RTSP_LOWER_TRANSPORT_UDP_MULTICAST:
1125 1126
            {
                char url[1024];
1127
                struct in_addr in;
1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138
                int port, ttl;

                if (reply->transports[0].destination) {
                    in.s_addr = htonl(reply->transports[0].destination);
                    port      = reply->transports[0].port_min;
                    ttl       = reply->transports[0].ttl;
                } else {
                    in        = rtsp_st->sdp_ip;
                    port      = rtsp_st->sdp_port;
                    ttl       = rtsp_st->sdp_ttl;
                }
1139
                snprintf(url, sizeof(url), "rtp://%s:%d?ttl=%d",
1140
                         inet_ntoa(in), port, ttl);
1141
                if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
1142 1143 1144 1145 1146 1147
                    err = AVERROR_INVALIDDATA;
                    goto fail;
                }
            }
            break;
        }
R
Romain Degez 已提交
1148

1149
        if ((err = rtsp_open_transport_ctx(s, rtsp_st)))
1150
            goto fail;
1151 1152
    }

1153 1154 1155
    if (reply->timeout > 0)
        rt->timeout = reply->timeout;

1156
    if (rt->server_type == RTSP_SERVER_REAL)
1157 1158
        rt->need_subscription = 1;

1159 1160 1161
    return 0;

fail:
1162 1163 1164 1165 1166 1167
    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;
        }
    }
1168 1169 1170 1171 1172 1173 1174
    return err;
}

static int rtsp_read_header(AVFormatContext *s,
                            AVFormatParameters *ap)
{
    RTSPState *rt = s->priv_data;
1175 1176
    char host[1024], path[1024], tcpname[1024], cmd[2048], auth[128];
    char *option_list, *option, *filename;
1177 1178
    URLContext *rtsp_hd;
    int port, ret, err;
1179
    RTSPMessageHeader reply1, *reply = &reply1;
1180
    unsigned char *content = NULL;
1181
    int lower_transport_mask = 0;
1182
    char real_challenge[64];
1183 1184

    /* extract hostname and port */
1185
    url_split(NULL, 0, auth, sizeof(auth),
1186
              host, sizeof(host), &port, path, sizeof(path), s->filename);
1187 1188 1189 1190 1191 1192 1193 1194 1195 1196
    if (*auth) {
        int auth_len = strlen(auth), b64_len = ((auth_len + 2) / 3) * 4 + 1;

        if (!(rt->auth_b64 = av_malloc(b64_len)))
            return AVERROR(ENOMEM);
        if (!av_base64_encode(rt->auth_b64, b64_len, auth, auth_len)) {
            err = AVERROR(EINVAL);
            goto fail;
        }
    }
1197 1198 1199 1200 1201 1202
    if (port < 0)
        port = RTSP_DEFAULT_PORT;

    /* search for options */
    option_list = strchr(path, '?');
    if (option_list) {
1203
        filename = strchr(s->filename, '?');
1204 1205
        while(option_list) {
            /* move the option pointer */
1206
            option = ++option_list;
1207 1208
            option_list = strchr(option_list, '&');
            if (option_list)
1209 1210
                *option_list = 0;

1211 1212
            /* handle the options */
            if (strcmp(option, "udp") == 0)
1213
                lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_UDP);
1214
            else if (strcmp(option, "multicast") == 0)
1215
                lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_UDP_MULTICAST);
1216
            else if (strcmp(option, "tcp") == 0)
1217
                lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_TCP);
1218 1219 1220 1221 1222
            else {
                strcpy(++filename, option);
                filename += strlen(option);
                if (option_list) *filename = '&';
            }
1223
        }
1224
        *filename = 0;
1225 1226
    }

1227
    if (!lower_transport_mask)
1228
        lower_transport_mask = (1 << RTSP_LOWER_TRANSPORT_NB) - 1;
1229 1230 1231

    /* open the tcp connexion */
    snprintf(tcpname, sizeof(tcpname), "tcp://%s:%d", host, port);
1232 1233 1234 1235
    if (url_open(&rtsp_hd, tcpname, URL_RDWR) < 0) {
        err = AVERROR(EIO);
        goto fail;
    }
1236 1237 1238
    rt->rtsp_hd = rtsp_hd;
    rt->seq = 0;

1239 1240 1241 1242
    /* request options supported by the server; this also detects server type */
    for (rt->server_type = RTSP_SERVER_RTP;;) {
        snprintf(cmd, sizeof(cmd),
                 "OPTIONS %s RTSP/1.0\r\n", s->filename);
1243
        if (rt->server_type == RTSP_SERVER_REAL)
1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265
            av_strlcat(cmd,
                       /**
                        * The following entries are required for proper
                        * streaming from a Realmedia server. They are
                        * interdependent in some way although we currently
                        * don't quite understand how. Values were copied
                        * from mplayer SVN r23589.
                        * @param CompanyID is a 16-byte ID in base64
                        * @param ClientChallenge is a 16-byte ID in hex
                        */
                       "ClientChallenge: 9e26d33f2984236010ef6253fb1887f7\r\n"
                       "PlayerStarttime: [28/03/2003:22:50:23 00:00]\r\n"
                       "CompanyID: KnKV4M4I/B2FjJ1TToLycw==\r\n"
                       "GUID: 00000000-0000-0000-0000-000000000000\r\n",
                       sizeof(cmd));
        rtsp_send_cmd(s, cmd, reply, NULL);
        if (reply->status_code != RTSP_STATUS_OK) {
            err = AVERROR_INVALIDDATA;
            goto fail;
        }

        /* detect server type if not standard-compliant RTP */
1266 1267
        if (rt->server_type != RTSP_SERVER_REAL && reply->real_challenge[0]) {
            rt->server_type = RTSP_SERVER_REAL;
1268
            continue;
1269 1270
        } else if (!strncasecmp(reply->server, "WMServer/", 9)) {
            rt->server_type = RTSP_SERVER_WMS;
1271
        } else if (rt->server_type == RTSP_SERVER_REAL) {
1272 1273 1274 1275 1276
            strcpy(real_challenge, reply->real_challenge);
        }
        break;
    }

1277 1278 1279 1280 1281
    /* describe the stream */
    snprintf(cmd, sizeof(cmd),
             "DESCRIBE %s RTSP/1.0\r\n"
             "Accept: application/sdp\r\n",
             s->filename);
1282
    if (rt->server_type == RTSP_SERVER_REAL) {
1283 1284 1285 1286 1287 1288 1289 1290
        /**
         * The Require: attribute is needed for proper streaming from
         * Realmedia servers.
         */
        av_strlcat(cmd,
                   "Require: com.real.retain-entity-for-setup\r\n",
                   sizeof(cmd));
    }
1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308
    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;
    }

1309
    do {
1310
        int lower_transport = ff_log2_tab[lower_transport_mask & ~(lower_transport_mask - 1)];
1311

1312
        err = make_setup_request(s, host, port, lower_transport,
1313
                                 rt->server_type == RTSP_SERVER_REAL ?
1314
                                     real_challenge : NULL);
1315
        if (err < 0)
1316
            goto fail;
1317 1318
        lower_transport_mask &= ~(1 << lower_transport);
        if (lower_transport_mask == 0 && err == 1) {
1319
            err = AVERROR(FF_NETERROR(EPROTONOSUPPORT));
1320 1321 1322
            goto fail;
        }
    } while (err);
1323

1324 1325 1326
    rt->state = RTSP_STATE_IDLE;
    rt->seek_timestamp = 0; /* default is to start stream at position
                               zero */
1327
    if (ap->initial_pause) {
1328 1329 1330 1331
        /* do not start immediately */
    } else {
        if (rtsp_read_play(s) < 0) {
            err = AVERROR_INVALIDDATA;
1332 1333 1334 1335 1336
            goto fail;
        }
    }
    return 0;
 fail:
1337
    rtsp_close_streams(rt);
1338 1339
    av_freep(&content);
    url_close(rt->rtsp_hd);
1340
    av_freep(&rt->auth_b64);
1341 1342 1343
    return err;
}

1344 1345
static int tcp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
                           uint8_t *buf, int buf_size)
1346 1347
{
    RTSPState *rt = s->priv_data;
F
Fabrice Bellard 已提交
1348
    int id, len, i, ret;
1349 1350
    RTSPStream *rtsp_st;

F
Fabrice Bellard 已提交
1351
#ifdef DEBUG_RTP_TCP
1352
    dprintf(s, "tcp_read_packet:\n");
F
Fabrice Bellard 已提交
1353
#endif
1354 1355
 redo:
    for(;;) {
1356 1357 1358 1359
        RTSPMessageHeader reply;

        ret = rtsp_read_reply(s, &reply, NULL, 1);
        if (ret == -1)
1360
            return -1;
1361
        if (ret == 1) /* received '$' */
1362
            break;
1363
        /* XXX: parse message */
1364 1365
        if (rt->state != RTSP_STATE_PLAYING)
            return 0;
1366
    }
1367
    ret = url_read_complete(rt->rtsp_hd, buf, 3);
F
Fabrice Bellard 已提交
1368
    if (ret != 3)
1369
        return -1;
F
Fabrice Bellard 已提交
1370
    id = buf[0];
1371
    len = AV_RB16(buf + 1);
F
Fabrice Bellard 已提交
1372
#ifdef DEBUG_RTP_TCP
1373
    dprintf(s, "id=%d len=%d\n", id, len);
F
Fabrice Bellard 已提交
1374
#endif
1375
    if (len > buf_size || len < 12)
1376 1377
        goto redo;
    /* get the data */
1378
    ret = url_read_complete(rt->rtsp_hd, buf, len);
F
Fabrice Bellard 已提交
1379
    if (ret != len)
1380
        return -1;
1381
    if (rt->transport == RTSP_TRANSPORT_RDT &&
1382
        ff_rdt_parse_header(buf, len, &id, NULL, NULL, NULL, NULL) < 0)
1383
        return -1;
1384

1385
    /* find the matching stream */
1386 1387
    for(i = 0; i < rt->nb_rtsp_streams; i++) {
        rtsp_st = rt->rtsp_streams[i];
1388 1389
        if (id >= rtsp_st->interleaved_min &&
            id <= rtsp_st->interleaved_max)
1390 1391 1392 1393
            goto found;
    }
    goto redo;
 found:
1394 1395
    *prtsp_st = rtsp_st;
    return len;
1396 1397
}

1398
static int udp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
1399
                           uint8_t *buf, int buf_size)
1400
{
1401
    RTSPState *rt = s->priv_data;
1402 1403
    RTSPStream *rtsp_st;
    fd_set rfds;
1404
    int fd, fd_max, n, i, ret, tcp_fd;
1405 1406 1407
    struct timeval tv;

    for(;;) {
1408
        if (url_interrupt_cb())
1409
            return AVERROR(EINTR);
1410
        FD_ZERO(&rfds);
L
Luca Abeni 已提交
1411 1412 1413 1414 1415 1416 1417
        if (rt->rtsp_hd) {
            tcp_fd = fd_max = url_get_file_handle(rt->rtsp_hd);
            FD_SET(tcp_fd, &rfds);
        } else {
            fd_max = 0;
            tcp_fd = -1;
        }
1418 1419
        for(i = 0; i < rt->nb_rtsp_streams; i++) {
            rtsp_st = rt->rtsp_streams[i];
1420
            if (rtsp_st->rtp_handle) {
R
Ronald S. Bultje 已提交
1421 1422
                /* currently, we cannot probe RTCP handle because of
                 * blocking restrictions */
1423 1424 1425 1426
                fd = url_get_file_handle(rtsp_st->rtp_handle);
                if (fd > fd_max)
                    fd_max = fd;
                FD_SET(fd, &rfds);
1427
            }
1428 1429
        }
        tv.tv_sec = 0;
1430
        tv.tv_usec = 100 * 1000;
1431 1432
        n = select(fd_max + 1, &rfds, NULL, NULL, &tv);
        if (n > 0) {
1433 1434
            for(i = 0; i < rt->nb_rtsp_streams; i++) {
                rtsp_st = rt->rtsp_streams[i];
1435
                if (rtsp_st->rtp_handle) {
1436 1437
                    fd = url_get_file_handle(rtsp_st->rtp_handle);
                    if (FD_ISSET(fd, &rfds)) {
R
Ronald S. Bultje 已提交
1438 1439 1440 1441 1442
                        ret = url_read(rtsp_st->rtp_handle, buf, buf_size);
                        if (ret > 0) {
                            *prtsp_st = rtsp_st;
                            return ret;
                        }
1443 1444 1445
                    }
                }
            }
1446 1447 1448 1449 1450
            if (FD_ISSET(tcp_fd, &rfds)) {
                RTSPMessageHeader reply;

                rtsp_read_reply(s, &reply, NULL, 0);
                /* XXX: parse message */
1451 1452
                if (rt->state != RTSP_STATE_PLAYING)
                    return 0;
1453
            }
1454 1455 1456 1457 1458 1459 1460 1461
        }
    }
}

static int rtsp_read_packet(AVFormatContext *s,
                            AVPacket *pkt)
{
    RTSPState *rt = s->priv_data;
1462 1463
    RTSPStream *rtsp_st;
    int ret, len;
1464
    uint8_t buf[10 * RTP_MAX_PACKET_LENGTH];
1465 1466
    RTSPMessageHeader reply1, *reply = &reply1;
    char cmd[1024];
1467

1468
    if (rt->server_type == RTSP_SERVER_REAL) {
1469
        int i;
1470
        enum AVDiscard cache[MAX_STREAMS];
1471

1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486
        for (i = 0; i < s->nb_streams; i++)
            cache[i] = s->streams[i]->discard;

        if (!rt->need_subscription) {
            if (memcmp (cache, rt->real_setup_cache,
                        sizeof(enum AVDiscard) * s->nb_streams)) {
                av_strlcatf(cmd, sizeof(cmd),
                            "SET_PARAMETER %s RTSP/1.0\r\n"
                            "Unsubscribe: %s\r\n",
                            s->filename, rt->last_subscription);
                rtsp_send_cmd(s, cmd, reply, NULL);
                if (reply->status_code != RTSP_STATUS_OK)
                    return AVERROR_INVALIDDATA;
                rt->need_subscription = 1;
            }
1487 1488
        }

1489 1490 1491 1492 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 1521 1522 1523 1524 1525
        if (rt->need_subscription) {
            int r, rule_nr, first = 1;

            memcpy(rt->real_setup_cache, cache,
                   sizeof(enum AVDiscard) * s->nb_streams);
            rt->last_subscription[0] = 0;

            snprintf(cmd, sizeof(cmd),
                     "SET_PARAMETER %s RTSP/1.0\r\n"
                     "Subscribe: ",
                     s->filename);
            for (i = 0; i < rt->nb_rtsp_streams; i++) {
                rule_nr = 0;
                for (r = 0; r < s->nb_streams; r++) {
                    if (s->streams[r]->priv_data == rt->rtsp_streams[i]) {
                        if (s->streams[r]->discard != AVDISCARD_ALL) {
                            if (!first)
                                av_strlcat(rt->last_subscription, ",",
                                           sizeof(rt->last_subscription));
                            ff_rdt_subscribe_rule(
                                rt->last_subscription,
                                sizeof(rt->last_subscription), i, rule_nr);
                            first = 0;
                        }
                        rule_nr++;
                    }
                }
            }
            av_strlcatf(cmd, sizeof(cmd), "%s\r\n", rt->last_subscription);
            rtsp_send_cmd(s, cmd, reply, NULL);
            if (reply->status_code != RTSP_STATUS_OK)
                return AVERROR_INVALIDDATA;
            rt->need_subscription = 0;

            if (rt->state == RTSP_STATE_PLAYING)
                rtsp_read_play (s);
        }
1526 1527
    }

1528
    /* get next frames from the same RTP packet */
1529
    if (rt->cur_transport_priv) {
1530
        if (rt->transport == RTSP_TRANSPORT_RDT)
1531
            ret = ff_rdt_parse_packet(rt->cur_transport_priv, pkt, NULL, 0);
1532
        else
1533
            ret = rtp_parse_packet(rt->cur_transport_priv, pkt, NULL, 0);
1534
        if (ret == 0) {
1535
            rt->cur_transport_priv = NULL;
1536 1537 1538 1539
            return 0;
        } else if (ret == 1) {
            return 0;
        } else {
1540
            rt->cur_transport_priv = NULL;
1541 1542
        }
    }
1543

1544 1545
    /* read next RTP packet */
 redo:
1546
    switch(rt->lower_transport) {
1547
    default:
1548
    case RTSP_LOWER_TRANSPORT_TCP:
1549
        len = tcp_read_packet(s, &rtsp_st, buf, sizeof(buf));
1550
        break;
1551 1552
    case RTSP_LOWER_TRANSPORT_UDP:
    case RTSP_LOWER_TRANSPORT_UDP_MULTICAST:
1553
        len = udp_read_packet(s, &rtsp_st, buf, sizeof(buf));
1554 1555
        if (len >=0 && rtsp_st->transport_priv && rt->transport == RTSP_TRANSPORT_RTP)
            rtp_check_and_send_back_rr(rtsp_st->transport_priv, len);
1556 1557
        break;
    }
1558
    if (len < 0)
1559
        return len;
1560 1561
    if (len == 0)
        return AVERROR_EOF;
1562
    if (rt->transport == RTSP_TRANSPORT_RDT)
1563
        ret = ff_rdt_parse_packet(rtsp_st->transport_priv, pkt, buf, len);
1564
    else
1565
        ret = rtp_parse_packet(rtsp_st->transport_priv, pkt, buf, len);
1566 1567 1568 1569
    if (ret < 0)
        goto redo;
    if (ret == 1) {
        /* more packets may follow, so we save the RTP context */
1570
        rt->cur_transport_priv = rtsp_st->transport_priv;
1571
    }
1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587

    /* send dummy request to keep TCP connection alive */
    if ((rt->server_type == RTSP_SERVER_WMS ||
         rt->server_type == RTSP_SERVER_REAL) &&
        (av_gettime() - rt->last_cmd_time) / 1000000 >= rt->timeout / 2) {
        if (rt->server_type == RTSP_SERVER_WMS) {
            snprintf(cmd, sizeof(cmd) - 1,
                     "GET_PARAMETER %s RTSP/1.0\r\n",
                     s->filename);
            rtsp_send_cmd_async(s, cmd, reply, NULL);
        } else {
            rtsp_send_cmd_async(s, "OPTIONS * RTSP/1.0\r\n",
                                reply, NULL);
        }
    }

1588
    return 0;
1589 1590
}

1591
static int rtsp_read_play(AVFormatContext *s)
1592
{
1593
    RTSPState *rt = s->priv_data;
1594
    RTSPMessageHeader reply1, *reply = &reply1;
1595 1596
    char cmd[1024];

1597
    av_log(s, AV_LOG_DEBUG, "hello state=%d\n", rt->state);
1598

1599
    if (!(rt->server_type == RTSP_SERVER_REAL && rt->need_subscription)) {
1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614
        if (rt->state == RTSP_STATE_PAUSED) {
            snprintf(cmd, sizeof(cmd),
                     "PLAY %s RTSP/1.0\r\n",
                     s->filename);
        } else {
            snprintf(cmd, sizeof(cmd),
                     "PLAY %s RTSP/1.0\r\n"
                     "Range: npt=%0.3f-\r\n",
                     s->filename,
                     (double)rt->seek_timestamp / AV_TIME_BASE);
        }
        rtsp_send_cmd(s, cmd, reply, NULL);
        if (reply->status_code != RTSP_STATUS_OK) {
            return -1;
        }
1615
    }
1616 1617
    rt->state = RTSP_STATE_PLAYING;
    return 0;
1618 1619
}

1620 1621
/* pause the stream */
static int rtsp_read_pause(AVFormatContext *s)
1622
{
1623
    RTSPState *rt = s->priv_data;
1624
    RTSPMessageHeader reply1, *reply = &reply1;
1625 1626 1627
    char cmd[1024];

    rt = s->priv_data;
1628

1629 1630
    if (rt->state != RTSP_STATE_PLAYING)
        return 0;
1631
    else if (!(rt->server_type == RTSP_SERVER_REAL && rt->need_subscription)) {
1632 1633 1634 1635 1636 1637 1638
        snprintf(cmd, sizeof(cmd),
                 "PAUSE %s RTSP/1.0\r\n",
                 s->filename);
        rtsp_send_cmd(s, cmd, reply, NULL);
        if (reply->status_code != RTSP_STATUS_OK) {
            return -1;
        }
1639
    }
1640 1641
    rt->state = RTSP_STATE_PAUSED;
    return 0;
1642 1643
}

1644
static int rtsp_read_seek(AVFormatContext *s, int stream_index,
1645
                          int64_t timestamp, int flags)
1646 1647
{
    RTSPState *rt = s->priv_data;
1648

1649
    rt->seek_timestamp = av_rescale_q(timestamp, s->streams[stream_index]->time_base, AV_TIME_BASE_Q);
1650 1651 1652 1653 1654
    switch(rt->state) {
    default:
    case RTSP_STATE_IDLE:
        break;
    case RTSP_STATE_PLAYING:
1655 1656 1657
        if (rtsp_read_pause(s) != 0)
            return -1;
        rt->state = RTSP_STATE_SEEKING;
1658 1659 1660 1661 1662 1663 1664 1665 1666 1667
        if (rtsp_read_play(s) != 0)
            return -1;
        break;
    case RTSP_STATE_PAUSED:
        rt->state = RTSP_STATE_IDLE;
        break;
    }
    return 0;
}

1668 1669 1670
static int rtsp_read_close(AVFormatContext *s)
{
    RTSPState *rt = s->priv_data;
1671
    RTSPMessageHeader reply1, *reply = &reply1;
1672 1673
    char cmd[1024];

F
Fabrice Bellard 已提交
1674
#if 0
1675
    /* NOTE: it is valid to flush the buffer here */
1676
    if (rt->lower_transport == RTSP_LOWER_TRANSPORT_TCP) {
1677 1678
        url_fclose(&rt->rtsp_gb);
    }
F
Fabrice Bellard 已提交
1679
#endif
1680
    snprintf(cmd, sizeof(cmd),
F
Fabrice Bellard 已提交
1681
             "TEARDOWN %s RTSP/1.0\r\n",
1682 1683 1684
             s->filename);
    rtsp_send_cmd(s, cmd, reply, NULL);

1685
    rtsp_close_streams(rt);
1686 1687 1688 1689
    url_close(rt->rtsp_hd);
    return 0;
}

1690
#if CONFIG_RTSP_DEMUXER
1691
AVInputFormat rtsp_demuxer = {
1692
    "rtsp",
1693
    NULL_IF_CONFIG_SMALL("RTSP input format"),
1694 1695 1696 1697 1698
    sizeof(RTSPState),
    rtsp_probe,
    rtsp_read_header,
    rtsp_read_packet,
    rtsp_read_close,
1699
    rtsp_read_seek,
1700
    .flags = AVFMT_NOFILE,
1701 1702
    .read_play = rtsp_read_play,
    .read_pause = rtsp_read_pause,
1703
};
1704
#endif
1705

1706
static int sdp_probe(AVProbeData *p1)
1707
{
1708
    const char *p = p1->buf, *p_end = p1->buf + p1->buf_size;
1709 1710

    /* we look for a line beginning "c=IN IP4" */
1711
    while (p < p_end && *p != '\0') {
1712
        if (p + sizeof("c=IN IP4") - 1 < p_end && av_strstart(p, "c=IN IP4", NULL))
1713
            return AVPROBE_SCORE_MAX / 2;
1714 1715 1716

        while(p < p_end - 1 && *p != '\n') p++;
        if (++p >= p_end)
1717 1718 1719 1720
            break;
        if (*p == '\r')
            p++;
    }
1721 1722 1723 1724 1725 1726 1727 1728
    return 0;
}

#define SDP_MAX_SIZE 8192

static int sdp_read_header(AVFormatContext *s,
                           AVFormatParameters *ap)
{
1729
    RTSPState *rt = s->priv_data;
1730 1731 1732 1733 1734 1735 1736 1737
    RTSPStream *rtsp_st;
    int size, i, err;
    char *content;
    char url[1024];

    /* read the whole sdp file */
    /* XXX: better loading */
    content = av_malloc(SDP_MAX_SIZE);
1738
    size = get_buffer(s->pb, content, SDP_MAX_SIZE - 1);
1739 1740 1741 1742 1743 1744 1745 1746 1747 1748
    if (size <= 0) {
        av_free(content);
        return AVERROR_INVALIDDATA;
    }
    content[size] ='\0';

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

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

1752
        snprintf(url, sizeof(url), "rtp://%s:%d?localport=%d&ttl=%d",
1753
                 inet_ntoa(rtsp_st->sdp_ip),
1754
                 rtsp_st->sdp_port,
1755
                 rtsp_st->sdp_port,
1756
                 rtsp_st->sdp_ttl);
1757
        if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
1758 1759 1760
            err = AVERROR_INVALIDDATA;
            goto fail;
        }
1761
        if ((err = rtsp_open_transport_ctx(s, rtsp_st)))
1762
            goto fail;
1763 1764 1765
    }
    return 0;
 fail:
1766
    rtsp_close_streams(rt);
1767 1768 1769 1770 1771 1772
    return err;
}

static int sdp_read_packet(AVFormatContext *s,
                            AVPacket *pkt)
{
1773
    return rtsp_read_packet(s, pkt);
1774 1775 1776 1777
}

static int sdp_read_close(AVFormatContext *s)
{
1778 1779
    RTSPState *rt = s->priv_data;
    rtsp_close_streams(rt);
1780 1781 1782
    return 0;
}

1783
#if CONFIG_SDP_DEMUXER
1784
AVInputFormat sdp_demuxer = {
1785
    "sdp",
1786
    NULL_IF_CONFIG_SMALL("SDP"),
1787 1788 1789 1790 1791 1792
    sizeof(RTSPState),
    sdp_probe,
    sdp_read_header,
    sdp_read_packet,
    sdp_read_close,
};
1793
#endif