rtsp.c 52.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/avstring.h"
26
#include "libavutil/intreadwrite.h"
27 28
#include "avformat.h"

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

37
#include "rtpdec.h"
38
#include "rdt.h"
Ryan Martell's avatar
Ryan Martell committed
39

40
//#define DEBUG
Fabrice Bellard's avatar
Fabrice Bellard committed
41
//#define DEBUG_RTP_TCP
42

43 44
static int rtsp_read_play(AVFormatContext *s);

45 46 47
/* XXX: currently, the only way to change the protocols consists in
   changing this variable */

Ronald S. Bultje's avatar
Ronald S. Bultje committed
48
#if LIBAVFORMAT_VERSION_INT < (53 << 16)
49
int rtsp_default_protocols = (1 << RTSP_LOWER_TRANSPORT_UDP);
Ronald S. Bultje's avatar
Ronald S. Bultje committed
50
#endif
Fabrice Bellard's avatar
Fabrice Bellard committed
51

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

static int redir_isspace(int c)
{
61
    return c == ' ' || c == '\t' || c == '\n' || c == '\r';
62 63 64 65 66 67 68 69 70 71 72
}

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

73
static void get_word_sep(char *buf, int buf_size, const char *sep,
74 75 76 77 78 79
                         const char **pp)
{
    const char *p;
    char *q;

    p = *pp;
Romain Degez's avatar
Romain Degez committed
80 81
    if (*p == '/')
        p++;
82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111
    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;
}

112 113
/* parse the rtpmap description: <codec_name>/<clock_rate>[/<other
   params>] */
Ryan Martell's avatar
Ryan Martell committed
114
static int sdp_parse_rtpmap(AVCodecContext *codec, RTSPStream *rtsp_st, int payload_type, const char *p)
115 116
{
    char buf[256];
Romain Degez's avatar
Romain Degez committed
117 118
    int i;
    AVCodec *c;
119
    const char *c_name;
120

Romain Degez's avatar
Romain Degez committed
121 122
    /* Loop into AVRtpDynamicPayloadTypes[] and AVRtpPayloadTypes[] and
       see if we can handle this kind of payload */
123
    get_word_sep(buf, sizeof(buf), "/", &p);
Romain Degez's avatar
Romain Degez committed
124
    if (payload_type >= RTP_PT_PRIVATE) {
Ryan Martell's avatar
Ryan Martell committed
125 126 127 128 129 130 131 132
        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();
                }
Romain Degez's avatar
Romain Degez committed
133 134
                break;
            }
Ryan Martell's avatar
Ryan Martell committed
135 136
            handler= handler->next;
        }
137
    } else {
Romain Degez's avatar
Romain Degez committed
138 139
        /* We are in a standard case ( from http://www.iana.org/assignments/rtp-parameters) */
        /* search into AVRtpPayloadTypes[] */
140
        codec->codec_id = ff_rtp_codec_id(buf, codec->codec_type);
Romain Degez's avatar
Romain Degez committed
141 142 143 144
    }

    c = avcodec_find_decoder(codec->codec_id);
    if (c && c->name)
145
        c_name = c->name;
Romain Degez's avatar
Romain Degez committed
146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162
    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;
163 164
                    // 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)
Romain Degez's avatar
Romain Degez committed
165 166 167 168 169 170 171 172 173 174 175
                }
                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;
176
    }
Romain Degez's avatar
Romain Degez committed
177 178

    return -1;
179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209
}

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

Romain Degez's avatar
Romain Degez committed
210 211 212 213
static void sdp_parse_fmtp_config(AVCodecContext *codec, char *attr, char *value)
{
    switch (codec->codec_id) {
        case CODEC_ID_MPEG4:
214
        case CODEC_ID_AAC:
Romain Degez's avatar
Romain Degez committed
215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230
            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;
}

Luca Barbato's avatar
Luca Barbato committed
231
typedef struct {
232
    const char *str;
Romain Degez's avatar
Romain Degez committed
233 234
    uint16_t type;
    uint32_t offset;
Luca Barbato's avatar
Luca Barbato committed
235
} AttrNameMap;
Romain Degez's avatar
Romain Degez committed
236 237 238 239

/* All known fmtp parmeters and the corresping RTPAttrTypeEnum */
#define ATTR_NAME_TYPE_INT 0
#define ATTR_NAME_TYPE_STR 1
Luca Barbato's avatar
Luca Barbato committed
240
static const AttrNameMap attr_names[]=
Romain Degez's avatar
Romain Degez committed
241
{
242 243 244 245 246 247
    {"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)},
Romain Degez's avatar
Romain Degez committed
248 249 250
    {NULL, -1, -1},
};

251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269
/** 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;
}

Romain Degez's avatar
Romain Degez committed
270 271
/* parse a SDP line and save stream attributes */
static void sdp_parse_fmtp(AVStream *st, const char *p)
272 273 274
{
    char attr[256];
    char value[4096];
Romain Degez's avatar
Romain Degez committed
275 276 277
    int i;

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

    /* loop on each attribute */
282 283
    while(rtsp_next_attr_and_value(&p, attr, sizeof(attr), value, sizeof(value)))
    {
284
        /* grab the codec extra_data from the config parameter of the fmtp line */
Romain Degez's avatar
Romain Degez committed
285 286 287 288 289 290 291 292
        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);
293
            }
294 295 296 297
        }
    }
}

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

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

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

344 345 346 347 348
#ifdef DEBUG
    printf("sdp: %c='%s'\n", letter, buf);
#endif

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

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

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

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

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

501 502 503
            rtsp_st = s->streams[s->nb_streams - 1]->priv_data;
            if (rtsp_st->dynamic_handler &&
                rtsp_st->dynamic_handler->parse_sdp_a_line)
504
                rtsp_st->dynamic_handler->parse_sdp_a_line(s, s->nb_streams - 1,
505
                    rtsp_st->dynamic_protocol_context, buf);
506 507 508 509 510
        }
        break;
    }
}

511
static int sdp_parse(AVFormatContext *s, const char *content)
512 513 514
{
    const char *p;
    int letter;
515 516 517 518 519
    /* 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
     * buffer is large. */
520
    char buf[8192], *q;
521
    SDPParseState sdp_parse_state, *s1 = &sdp_parse_state;
522

523
    memset(s1, 0, sizeof(SDPParseState));
524 525 526 527 528 529 530 531 532 533 534 535
    p = content;
    for(;;) {
        skip_spaces(&p);
        letter = *p;
        if (letter == '\0')
            break;
        p++;
        if (*p != '=')
            goto next_line;
        p++;
        /* get the content */
        q = buf;
Fabrice Bellard's avatar
Fabrice Bellard committed
536
        while (*p != '\n' && *p != '\r' && *p != '\0') {
537 538 539 540 541
            if ((q - buf) < sizeof(buf) - 1)
                *q++ = *p;
            p++;
        }
        *q = '\0';
542
        sdp_parse_line(s, s1, letter, buf);
543 544 545 546 547 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 578 579 580
    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];
581

582
    reply->nb_transports = 0;
583

584 585 586 587 588 589 590
    for(;;) {
        skip_spaces(&p);
        if (*p == '\0')
            break;

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

591
        get_word_sep(transport_protocol, sizeof(transport_protocol),
592 593 594
                     "/", &p);
        if (*p == '/')
            p++;
595
        if (!strcasecmp (transport_protocol, "rtp")) {
596 597 598 599 600 601 602
            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);
603 604 605 606
            }
            th->transport = RTSP_TRANSPORT_RTP;
        } else if (!strcasecmp (transport_protocol, "x-pn-tng") ||
                   !strcasecmp (transport_protocol, "x-real-rdt")) {
607
            /* x-pn-tng/<protocol> */
608 609
            get_word_sep(lower_transport, sizeof(lower_transport), "/;,", &p);
            profile[0] = '\0';
610
            th->transport = RTSP_TRANSPORT_RDT;
611
        }
Fabrice Bellard's avatar
Fabrice Bellard committed
612
        if (!strcasecmp(lower_transport, "TCP"))
613
            th->lower_transport = RTSP_LOWER_TRANSPORT_TCP;
614
        else
615
            th->lower_transport = RTSP_LOWER_TRANSPORT_UDP;
616

617 618 619 620 621 622 623 624 625 626 627 628 629
        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++;
630
                    rtsp_parse_range(&th->client_port_min,
631 632 633 634 635
                                     &th->client_port_max, &p);
                }
            } else if (!strcmp(parameter, "server_port")) {
                if (*p == '=') {
                    p++;
636
                    rtsp_parse_range(&th->server_port_min,
637 638 639 640 641
                                     &th->server_port_max, &p);
                }
            } else if (!strcmp(parameter, "interleaved")) {
                if (*p == '=') {
                    p++;
642
                    rtsp_parse_range(&th->interleaved_min,
643 644 645
                                     &th->interleaved_max, &p);
                }
            } else if (!strcmp(parameter, "multicast")) {
646 647
                if (th->lower_transport == RTSP_LOWER_TRANSPORT_UDP)
                    th->lower_transport = RTSP_LOWER_TRANSPORT_UDP_MULTICAST;
648 649 650 651 652 653 654 655 656 657 658
            } 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);
659
                    if (inet_aton(buf, &ipaddr))
660 661 662 663 664 665 666 667 668 669 670 671 672 673 674 675 676 677 678 679 680
                        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;
681
    if (av_stristart(p, "Session:", &p)) {
682
        get_word_sep(reply->session_id, sizeof(reply->session_id), ";", &p);
683
    } else if (av_stristart(p, "Content-Length:", &p)) {
684
        reply->content_length = strtol(p, NULL, 10);
685
    } else if (av_stristart(p, "Transport:", &p)) {
686
        rtsp_parse_transport(reply, p);
687
    } else if (av_stristart(p, "CSeq:", &p)) {
688
        reply->seq = strtol(p, NULL, 10);
689
    } else if (av_stristart(p, "Range:", &p)) {
690
        rtsp_parse_range_npt(p, &reply->range_start, &reply->range_end);
691 692 693
    } else if (av_stristart(p, "RealChallenge1:", &p)) {
        skip_spaces(&p);
        av_strlcpy(reply->real_challenge, p, sizeof(reply->real_challenge));
694 695 696
    } else if (av_stristart(p, "Server:", &p)) {
        skip_spaces(&p);
        av_strlcpy(reply->server, p, sizeof(reply->server));
697 698 699
    }
}

700 701 702 703 704 705 706 707 708 709 710 711 712 713
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;
}

714 715 716 717 718 719 720
/* 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];

721
    ret = url_readbuf(rt->rtsp_hd, buf, 3);
722 723
    if (ret != 3)
        return;
724
    len = AV_RB16(buf + 1);
725 726 727 728 729 730 731 732
#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);
733
        ret = url_readbuf(rt->rtsp_hd, buf, len1);
734 735 736 737 738
        if (ret != len1)
            return;
        len -= len1;
    }
}
739

740 741
static void rtsp_send_cmd(AVFormatContext *s,
                          const char *cmd, RTSPHeader *reply,
742 743 744 745 746 747 748 749 750 751 752 753
                          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++;
Måns Rullgård's avatar
Måns Rullgård committed
754
    av_strlcpy(buf, cmd, sizeof(buf));
Fabrice Bellard's avatar
Fabrice Bellard committed
755
    snprintf(buf1, sizeof(buf1), "CSeq: %d\r\n", rt->seq);
Måns Rullgård's avatar
Måns Rullgård committed
756
    av_strlcat(buf, buf1, sizeof(buf));
757
    if (rt->session_id[0] != '\0' && !strstr(cmd, "\nIf-Match:")) {
Fabrice Bellard's avatar
Fabrice Bellard committed
758
        snprintf(buf1, sizeof(buf1), "Session: %s\r\n", rt->session_id);
Måns Rullgård's avatar
Måns Rullgård committed
759
        av_strlcat(buf, buf1, sizeof(buf));
760
    }
Måns Rullgård's avatar
Måns Rullgård committed
761
    av_strlcat(buf, "\r\n", sizeof(buf));
762 763 764 765 766 767 768 769 770 771 772
#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(;;) {
773
            if (url_readbuf(rt->rtsp_hd, &ch, 1) != 1)
774 775 776
                break;
            if (ch == '\n')
                break;
777 778 779 780
            if (ch == '$') {
                /* XXX: only parse it if first char on line ? */
                rtsp_skip_packet(s);
            } else if (ch != '\r') {
781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796 797 798 799
                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);
Måns Rullgård's avatar
Måns Rullgård committed
800 801
            av_strlcat(rt->last_reply, p,    sizeof(rt->last_reply));
            av_strlcat(rt->last_reply, "\n", sizeof(rt->last_reply));
802 803 804
        }
        line_count++;
    }
805

806
    if (rt->session_id[0] == '\0' && reply->session_id[0] != '\0')
Måns Rullgård's avatar
Måns Rullgård committed
807
        av_strlcpy(rt->session_id, reply->session_id, sizeof(rt->session_id));
808

809 810 811 812
    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);
813
        (void)url_readbuf(rt->rtsp_hd, content, content_length);
814 815 816 817
        content[content_length] = '\0';
    }
    if (content_ptr)
        *content_ptr = content;
818 819
    else
        av_free(content);
820 821 822
}


823 824 825 826 827 828 829 830 831
/* 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) {
832
            if (rtsp_st->transport_priv) {
833
                if (rt->transport == RTSP_TRANSPORT_RDT)
834
                    ff_rdt_parse_close(rtsp_st->transport_priv);
835
                else
836
                    rtp_parse_close(rtsp_st->transport_priv);
837
            }
838 839
            if (rtsp_st->rtp_handle)
                url_close(rtsp_st->rtp_handle);
Ryan Martell's avatar
Ryan Martell committed
840 841
            if (rtsp_st->dynamic_handler && rtsp_st->dynamic_protocol_context)
                rtsp_st->dynamic_handler->close(rtsp_st->dynamic_protocol_context);
842 843 844 845 846
        }
    }
    av_free(rt->rtsp_streams);
}

847 848 849
static int
rtsp_open_transport_ctx(AVFormatContext *s, RTSPStream *rtsp_st)
{
850
    RTSPState *rt = s->priv_data;
851 852
    AVStream *st = NULL;

853 854 855 856 857
    /* open the RTP context */
    if (rtsp_st->stream_index >= 0)
        st = s->streams[rtsp_st->stream_index];
    if (!st)
        s->ctx_flags |= AVFMTCTX_NOHEADER;
858 859

    if (rt->transport == RTSP_TRANSPORT_RDT)
860
        rtsp_st->transport_priv = ff_rdt_parse_open(s, st->index,
861 862 863
                                            rtsp_st->dynamic_protocol_context,
                                            rtsp_st->dynamic_handler);
    else
864
        rtsp_st->transport_priv = rtp_parse_open(s, st, rtsp_st->rtp_handle,
Ronald S. Bultje's avatar
Ronald S. Bultje committed
865 866
                                         rtsp_st->sdp_payload_type,
                                         &rtsp_st->rtp_payload_data);
867

868
    if (!rtsp_st->transport_priv) {
869
         return AVERROR(ENOMEM);
870
    } else if (rt->transport != RTSP_TRANSPORT_RDT) {
871
        if(rtsp_st->dynamic_handler) {
872
            rtp_parse_set_dynamic_protocol(rtsp_st->transport_priv,
873 874
                                           rtsp_st->dynamic_protocol_context,
                                           rtsp_st->dynamic_handler);
875
        }
876
    }
877 878 879 880

    return 0;
}

881 882 883 884
/**
 * @returns 0 on success, <0 on error, 1 if protocol is unavailable.
 */
static int
885
make_setup_request (AVFormatContext *s, const char *host, int port,
886
                    int lower_transport, const char *real_challenge)
887 888
{
    RTSPState *rt = s->priv_data;
889
    int j, i, err, interleave = 0;
890
    RTSPStream *rtsp_st;
891 892
    RTSPHeader reply1, *reply = &reply1;
    char cmd[2048];
893 894
    const char *trans_pref;

895
    if (rt->transport == RTSP_TRANSPORT_RDT)
896 897 898
        trans_pref = "x-pn-tng";
    else
        trans_pref = "RTP/AVP";
899

900 901 902
    /* for each stream, make the setup request */
    /* XXX: we assume the same server is used for the control of each
       RTSP stream */
Romain Degez's avatar
Romain Degez committed
903 904

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

907
        rtsp_st = rt->rtsp_streams[i];
908 909

        /* RTP/UDP */
910
        if (lower_transport == RTSP_LOWER_TRANSPORT_UDP) {
Fabrice Bellard's avatar
Fabrice Bellard committed
911 912 913
            char buf[256];

            /* first try in specified port range */
Romain Degez's avatar
Romain Degez committed
914 915
            if (RTSP_RTP_PORT_MIN != 0) {
                while(j <= RTSP_RTP_PORT_MAX) {
916
                    snprintf(buf, sizeof(buf), "rtp://%s?localport=%d", host, j);
917
                    j += 2; /* we will use two port by rtp stream (rtp and rtcp) */
918
                    if (url_open(&rtsp_st->rtp_handle, buf, URL_RDWR) == 0) {
Fabrice Bellard's avatar
Fabrice Bellard committed
919
                        goto rtp_opened;
Romain Degez's avatar
Romain Degez committed
920
                    }
Fabrice Bellard's avatar
Fabrice Bellard committed
921
                }
922
            }
Fabrice Bellard's avatar
Fabrice Bellard committed
923

Romain Degez's avatar
Romain Degez committed
924 925 926 927 928 929
/*            then try on any port
**            if (url_open(&rtsp_st->rtp_handle, "rtp://", URL_RDONLY) < 0) {
**                err = AVERROR_INVALIDDATA;
**                goto fail;
**            }
*/
Fabrice Bellard's avatar
Fabrice Bellard committed
930 931

        rtp_opened:
932
            port = rtp_get_local_port(rtsp_st->rtp_handle);
933
            snprintf(transport, sizeof(transport) - 1,
934 935 936 937 938
                     "%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);
939
            if (rt->transport == RTSP_TRANSPORT_RTP)
940
                av_strlcatf(transport, sizeof(transport), "-%d", port + 1);
941 942 943
        }

        /* RTP/TCP */
944
        else if (lower_transport == RTSP_LOWER_TRANSPORT_TCP) {
945
            snprintf(transport, sizeof(transport) - 1,
946 947 948 949