rtsp.c 63.1 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

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

50
#define SPACE_CHARS " \t\r\n"
51 52 53
/* 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)
54 55 56 57 58 59 60 61 62
static void skip_spaces(const char **pp)
{
    const char *p;
    p = *pp;
    while (redir_isspace(*p))
        p++;
    *pp = p;
}

63 64
static void get_word_until_chars(char *buf, int buf_size,
                                 const char *sep, const char **pp)
65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
{
    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;
}

82 83
static void get_word_sep(char *buf, int buf_size, const char *sep,
                         const char **pp)
84
{
85 86 87
    if (**pp == '/') (*pp)++;
    get_word_until_chars(buf, buf_size, sep, pp);
}
88

89 90 91
static void get_word(char *buf, int buf_size, const char **pp)
{
    get_word_until_chars(buf, buf_size, SPACE_CHARS, pp);
92 93
}

94
/* parse the rtpmap description: <codec_name>/<clock_rate>[/<other params>] */
95 96
static int sdp_parse_rtpmap(AVFormatContext *s,
                            AVCodecContext *codec, RTSPStream *rtsp_st,
97
                            int payload_type, const char *p)
98 99
{
    char buf[256];
R
Romain Degez 已提交
100 101
    int i;
    AVCodec *c;
102
    const char *c_name;
103

R
Romain Degez 已提交
104
    /* Loop into AVRtpDynamicPayloadTypes[] and AVRtpPayloadTypes[] and
105 106 107 108 109
     * see if we can handle this kind of payload.
     * The space should normally not be there but some Real streams or
     * particular servers ("RealServer Version 6.1.3.970", see issue 1658)
     * have a trailing space. */
    get_word_sep(buf, sizeof(buf), "/ ", &p);
R
Romain Degez 已提交
110
    if (payload_type >= RTP_PT_PRIVATE) {
111 112 113 114 115 116 117 118 119
        RTPDynamicProtocolHandler *handler;
        for (handler = RTPFirstDynamicPayloadHandler;
             handler; handler = handler->next) {
            if (!strcasecmp(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();
R
Romain Degez 已提交
120 121
                break;
            }
R
Ryan Martell 已提交
122
        }
123
    } else {
124 125
        /* We are in a standard case
         * (from http://www.iana.org/assignments/rtp-parameters). */
R
Romain Degez 已提交
126
        /* search into AVRtpPayloadTypes[] */
127
        codec->codec_id = ff_rtp_codec_id(buf, codec->codec_type);
R
Romain Degez 已提交
128 129 130 131
    }

    c = avcodec_find_decoder(codec->codec_id);
    if (c && c->name)
132
        c_name = c->name;
R
Romain Degez 已提交
133
    else
134
        c_name = "(null)";
R
Romain Degez 已提交
135

R
Ronald S. Bultje 已提交
136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152
    get_word_sep(buf, sizeof(buf), "/", &p);
    i = atoi(buf);
    switch (codec->codec_type) {
    case CODEC_TYPE_AUDIO:
        av_log(s, 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;
            // 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. Patch on its way. (rdm)
R
Romain Degez 已提交
153
        }
R
Ronald S. Bultje 已提交
154 155 156 157 158 159 160 161 162 163 164 165
        av_log(s, AV_LOG_DEBUG, "audio samplerate set to: %i\n",
               codec->sample_rate);
        av_log(s, AV_LOG_DEBUG, "audio channels set to: %i\n",
               codec->channels);
        break;
    case CODEC_TYPE_VIDEO:
        av_log(s, AV_LOG_DEBUG, "video codec set to: %s\n", c_name);
        break;
    default:
        break;
    }
    return 0;
166 167
}

168
/* return the length and optionally the data */
169 170 171 172 173 174
static int hex_to_data(uint8_t *data, const char *p)
{
    int c, len, v;

    len = 0;
    v = 1;
175
    for (;;) {
176
        skip_spaces(&p);
177
        if (*p == '\0')
178
            break;
179
        c = toupper((unsigned char) *p++);
180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196
        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;
}

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

L
Luca Barbato 已提交
224
typedef struct {
225
    const char *str;
226 227
    uint16_t    type;
    uint32_t    offset;
L
Luca Barbato 已提交
228
} AttrNameMap;
R
Romain Degez 已提交
229 230 231 232

/* All known fmtp parmeters and the corresping RTPAttrTypeEnum */
#define ATTR_NAME_TYPE_INT 0
#define ATTR_NAME_TYPE_STR 1
L
Luca Barbato 已提交
233
static const AttrNameMap attr_names[]=
R
Romain Degez 已提交
234
{
235 236 237 238 239 240 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) },
    { NULL, -1, -1 },
R
Romain Degez 已提交
248 249
};

250 251 252 253 254
/* 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)
255 256
{
    skip_spaces(p);
257
    if (**p) {
258 259 260 261 262 263 264 265 266 267 268
        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 已提交
269 270
/* parse a SDP line and save stream attributes */
static void sdp_parse_fmtp(AVStream *st, const char *p)
271 272
{
    char attr[256];
273 274 275
    /* 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 已提交
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 284 285
    while (rtsp_next_attr_and_value(&p, attr, sizeof(attr),
                                    value, sizeof(value))) {
        /* grab the codec extra_data from the config parameter of the fmtp
         * line */
286 287
        sdp_parse_fmtp_config(codec, rtsp_st->dynamic_protocol_context,
                              attr, value);
R
Romain Degez 已提交
288 289 290
        /* Looking for a known attribute */
        for (i = 0; attr_names[i].str; ++i) {
            if (!strcasecmp(attr, attr_names[i].str)) {
291 292 293 294 295 296
                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);
297
            }
298 299 300 301
        }
    }
}

S
Stefano Sabatini 已提交
302
/** Parse a string p in the form of Range:npt=xx-xx, and determine the start
303 304 305 306 307 308 309 310
 *  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);
311
    if (!av_stristart(p, "npt=", &p))
312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327
        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);
}

328 329 330
typedef struct SDPParseState {
    /* SDP only */
    struct in_addr default_ip;
331 332
    int            default_ttl;
    int            skip_media;  ///< set if an unknown m= line occurs
333 334 335
} SDPParseState;

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

348
    dprintf(s, "sdp: %c='%s'\n", letter, buf);
349 350

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

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

417 418 419 420
        /* XXX: handle list of formats */
        get_word(buf1, sizeof(buf1), &p); /* format list */
        rtsp_st->sdp_payload_type = atoi(buf1);

421
        if (!strcmp(ff_rtp_enc_name(rtsp_st->sdp_payload_type), "MP2T")) {
422 423 424 425 426 427 428
            /* no corresponding stream */
        } else {
            st = av_new_stream(s, 0);
            if (!st)
                return;
            st->priv_data = rtsp_st;
            rtsp_st->stream_index = st->index;
429
            st->codec->codec_type = codec_type;
R
Romain Degez 已提交
430
            if (rtsp_st->sdp_payload_type < RTP_PT_PRIVATE) {
431
                /* if standard payload type, we can find the codec right now */
432
                ff_rtp_get_codec_info(st->codec, rtsp_st->sdp_payload_type);
433 434
            }
        }
435
        /* put a default control url */
436
        av_strlcpy(rtsp_st->control_url, rt->control_uri,
437
                   sizeof(rtsp_st->control_url));
438 439
        break;
    case 'a':
440 441 442 443 444 445
        if (av_strstart(p, "control:", &p)) {
            if (s->nb_streams == 0) {
                if (!strncmp(p, "rtsp://", 7))
                    av_strlcpy(rt->control_uri, p,
                               sizeof(rt->control_uri));
            } else {
446 447 448 449
            char proto[32];
            /* get the control url */
            st = s->streams[s->nb_streams - 1];
            rtsp_st = st->priv_data;
450

451
            /* XXX: may need to add full url resolution */
452 453
            url_split(proto, sizeof(proto), NULL, 0, NULL, 0,
                      NULL, NULL, 0, p);
454 455
            if (proto[0] == '\0') {
                /* relative control URL */
456
                if (rtsp_st->control_url[strlen(rtsp_st->control_url)-1]!='/')
457 458 459 460 461 462 463
                av_strlcat(rtsp_st->control_url, "/",
                           sizeof(rtsp_st->control_url));
                av_strlcat(rtsp_st->control_url, p,
                           sizeof(rtsp_st->control_url));
            } else
                av_strlcpy(rtsp_st->control_url, p,
                           sizeof(rtsp_st->control_url));
464
            }
465
        } else if (av_strstart(p, "rtpmap:", &p) && s->nb_streams > 0) {
466
            /* NOTE: rtpmap is only supported AFTER the 'm=' tag */
467
            get_word(buf1, sizeof(buf1), &p);
468
            payload_type = atoi(buf1);
469
            st = s->streams[s->nb_streams - 1];
R
Ronald S. Bultje 已提交
470
            rtsp_st = st->priv_data;
471
            sdp_parse_rtpmap(s, st->codec, rtsp_st, payload_type, p);
472
        } else if (av_strstart(p, "fmtp:", &p)) {
473
            /* NOTE: fmtp is only supported AFTER the 'a=rtpmap:xxx' tag */
474
            get_word(buf1, sizeof(buf1), &p);
475
            payload_type = atoi(buf1);
476 477
            for (i = 0; i < s->nb_streams; i++) {
                st      = s->streams[i];
478 479
                rtsp_st = st->priv_data;
                if (rtsp_st->sdp_payload_type == payload_type) {
480 481 482 483
                    if (!(rtsp_st->dynamic_handler &&
                          rtsp_st->dynamic_handler->parse_sdp_a_line &&
                          rtsp_st->dynamic_handler->parse_sdp_a_line(s,
                              i, rtsp_st->dynamic_protocol_context, buf)))
484
                        sdp_parse_fmtp(st, p);
485 486
                }
            }
487
        } else if (av_strstart(p, "framesize:", &p)) {
488 489 490
            // let dynamic protocol handlers have a stab at the line.
            get_word(buf1, sizeof(buf1), &p);
            payload_type = atoi(buf1);
491 492
            for (i = 0; i < s->nb_streams; i++) {
                st      = s->streams[i];
493
                rtsp_st = st->priv_data;
494 495 496 497 498
                if (rtsp_st->sdp_payload_type == payload_type &&
                    rtsp_st->dynamic_handler &&
                    rtsp_st->dynamic_handler->parse_sdp_a_line)
                    rtsp_st->dynamic_handler->parse_sdp_a_line(s, i,
                        rtsp_st->dynamic_protocol_context, buf);
499
            }
500
        } else if (av_strstart(p, "range:", &p)) {
501 502 503 504
            int64_t start, end;

            // this is so that seeking on a streamed file can work.
            rtsp_parse_range_npt(p, &start, &end);
505 506 507 508
            s->start_time = start;
            /* AV_NOPTS_VALUE means live broadcast (and can't seek) */
            s->duration   = (end == AV_NOPTS_VALUE) ?
                            AV_NOPTS_VALUE : end - start;
509 510 511
        } else if (av_strstart(p, "IsRealDataType:integer;",&p)) {
            if (atoi(p) == 1)
                rt->transport = RTSP_TRANSPORT_RDT;
512 513 514 515
        } 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 已提交
516 517 518 519 520 521
                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)
522 523
                    rtsp_st->dynamic_handler->parse_sdp_a_line(s,
                        s->nb_streams - 1,
R
Ronald S. Bultje 已提交
524
                        rtsp_st->dynamic_protocol_context, buf);
525
            }
526 527 528 529 530
        }
        break;
    }
}

531
static int sdp_parse(AVFormatContext *s, const char *content)
532 533 534
{
    const char *p;
    int letter;
535 536 537 538
    /* 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
539 540 541 542
     * buffer is large.
     *
     * The Vorbis FMTP line can be up to 16KB - see sdp_parse_fmtp. */
    char buf[16384], *q;
543
    SDPParseState sdp_parse_state, *s1 = &sdp_parse_state;
544

545
    memset(s1, 0, sizeof(SDPParseState));
546
    p = content;
547
    for (;;) {
548 549 550 551 552 553 554 555 556 557
        skip_spaces(&p);
        letter = *p;
        if (letter == '\0')
            break;
        p++;
        if (*p != '=')
            goto next_line;
        p++;
        /* get the content */
        q = buf;
F
Fabrice Bellard 已提交
558
        while (*p != '\n' && *p != '\r' && *p != '\0') {
559 560 561 562 563
            if ((q - buf) < sizeof(buf) - 1)
                *q++ = *p;
            p++;
        }
        *q = '\0';
564
        sdp_parse_line(s, s1, letter, buf);
565 566 567 568 569 570 571 572 573
    next_line:
        while (*p != '\n' && *p != '\0')
            p++;
        if (*p == '\n')
            p++;
    }
    return 0;
}

574
/* close and free RTSP streams */
575
static void rtsp_close_streams(AVFormatContext *s)
576
{
577
    RTSPState *rt = s->priv_data;
578 579 580
    int i;
    RTSPStream *rtsp_st;

581
    for (i = 0; i < rt->nb_rtsp_streams; i++) {
582 583 584
        rtsp_st = rt->rtsp_streams[i];
        if (rtsp_st) {
            if (rtsp_st->transport_priv) {
585 586 587 588 589 590 591
                if (s->oformat) {
                    AVFormatContext *rtpctx = rtsp_st->transport_priv;
                    av_write_trailer(rtpctx);
                    url_fclose(rtpctx->pb);
                    av_free(rtpctx->streams[0]);
                    av_free(rtpctx);
                } else if (rt->transport == RTSP_TRANSPORT_RDT)
592 593 594 595 596 597 598
                    ff_rdt_parse_close(rtsp_st->transport_priv);
                else
                    rtp_parse_close(rtsp_st->transport_priv);
            }
            if (rtsp_st->rtp_handle)
                url_close(rtsp_st->rtp_handle);
            if (rtsp_st->dynamic_handler && rtsp_st->dynamic_protocol_context)
599 600
                rtsp_st->dynamic_handler->close(
                    rtsp_st->dynamic_protocol_context);
601 602 603 604 605 606 607 608 609 610
        }
    }
    av_free(rt->rtsp_streams);
    if (rt->asf_ctx) {
        av_close_input_stream (rt->asf_ctx);
        rt->asf_ctx = NULL;
    }
    av_freep(&rt->auth_b64);
}

611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654
static void *rtsp_rtp_mux_open(AVFormatContext *s, AVStream *st, URLContext *handle) {
    AVFormatContext *rtpctx;
    int ret;
    AVOutputFormat *rtp_format = av_guess_format("rtp", NULL, NULL);

    if (!rtp_format)
        return NULL;

    /* Allocate an AVFormatContext for each output stream */
    rtpctx = avformat_alloc_context();
    if (!rtpctx)
        return NULL;

    rtpctx->oformat = rtp_format;
    if (!av_new_stream(rtpctx, 0)) {
        av_free(rtpctx);
        return NULL;
    }
    /* Copy the max delay setting; the rtp muxer reads this. */
    rtpctx->max_delay = s->max_delay;
    /* Copy other stream parameters. */
    rtpctx->streams[0]->sample_aspect_ratio = st->sample_aspect_ratio;

    /* Remove the local codec, link to the original codec
     * context instead, to give the rtp muxer access to
     * codec parameters. */
    av_free(rtpctx->streams[0]->codec);
    rtpctx->streams[0]->codec = st->codec;

    url_fdopen(&rtpctx->pb, handle);
    ret = av_write_header(rtpctx);

    if (ret) {
        url_fclose(rtpctx->pb);
        av_free(rtpctx->streams[0]);
        av_free(rtpctx);
        return NULL;
    }

    /* Copy the RTP AVStream timebase back to the original AVStream */
    st->time_base = rtpctx->streams[0]->time_base;
    return rtpctx;
}

655
static int rtsp_open_transport_ctx(AVFormatContext *s, RTSPStream *rtsp_st)
656 657 658 659 660 661 662 663 664 665
{
    RTSPState *rt = s->priv_data;
    AVStream *st = NULL;

    /* open the RTP context */
    if (rtsp_st->stream_index >= 0)
        st = s->streams[rtsp_st->stream_index];
    if (!st)
        s->ctx_flags |= AVFMTCTX_NOHEADER;

666 667 668 669 670
    if (s->oformat) {
        rtsp_st->transport_priv = rtsp_rtp_mux_open(s, st, rtsp_st->rtp_handle);
        /* Ownage of rtp_handle is passed to the rtp mux context */
        rtsp_st->rtp_handle = NULL;
    } else if (rt->transport == RTSP_TRANSPORT_RDT)
671 672 673 674 675 676 677 678 679 680 681
        rtsp_st->transport_priv = ff_rdt_parse_open(s, st->index,
                                            rtsp_st->dynamic_protocol_context,
                                            rtsp_st->dynamic_handler);
    else
        rtsp_st->transport_priv = rtp_parse_open(s, st, rtsp_st->rtp_handle,
                                         rtsp_st->sdp_payload_type,
                                         &rtsp_st->rtp_payload_data);

    if (!rtsp_st->transport_priv) {
         return AVERROR(ENOMEM);
    } else if (rt->transport != RTSP_TRANSPORT_RDT) {
682
        if (rtsp_st->dynamic_handler) {
683 684 685 686 687 688 689 690 691
            rtp_parse_set_dynamic_protocol(rtsp_st->transport_priv,
                                           rtsp_st->dynamic_protocol_context,
                                           rtsp_st->dynamic_handler);
        }
    }

    return 0;
}

692
#if CONFIG_RTSP_DEMUXER
693 694 695 696 697 698 699
static int rtsp_probe(AVProbeData *p)
{
    if (av_strstart(p->filename, "rtsp:", NULL))
        return AVPROBE_SCORE_MAX;
    return 0;
}

700 701 702 703 704 705 706 707 708 709 710 711 712 713 714 715 716 717 718 719 720
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 */
721
static void rtsp_parse_transport(RTSPMessageHeader *reply, const char *p)
722 723 724 725 726 727 728
{
    char transport_protocol[16];
    char profile[16];
    char lower_transport[16];
    char parameter[16];
    RTSPTransportField *th;
    char buf[256];
729

730
    reply->nb_transports = 0;
731

732
    for (;;) {
733 734 735 736 737 738
        skip_spaces(&p);
        if (*p == '\0')
            break;

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

739
        get_word_sep(transport_protocol, sizeof(transport_protocol),
740
                     "/", &p);
741
        if (!strcasecmp (transport_protocol, "rtp")) {
742 743 744 745 746 747
            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);
748 749 750 751
            }
            th->transport = RTSP_TRANSPORT_RTP;
        } else if (!strcasecmp (transport_protocol, "x-pn-tng") ||
                   !strcasecmp (transport_protocol, "x-real-rdt")) {
752
            /* x-pn-tng/<protocol> */
753 754
            get_word_sep(lower_transport, sizeof(lower_transport), "/;,", &p);
            profile[0] = '\0';
755
            th->transport = RTSP_TRANSPORT_RDT;
756
        }
F
Fabrice Bellard 已提交
757
        if (!strcasecmp(lower_transport, "TCP"))
758
            th->lower_transport = RTSP_LOWER_TRANSPORT_TCP;
759
        else
760
            th->lower_transport = RTSP_LOWER_TRANSPORT_UDP;
761

762 763 764 765 766 767 768 769 770 771 772 773 774
        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++;
775
                    rtsp_parse_range(&th->client_port_min,
776 777 778 779 780
                                     &th->client_port_max, &p);
                }
            } else if (!strcmp(parameter, "server_port")) {
                if (*p == '=') {
                    p++;
781
                    rtsp_parse_range(&th->server_port_min,
782 783 784 785 786
                                     &th->server_port_max, &p);
                }
            } else if (!strcmp(parameter, "interleaved")) {
                if (*p == '=') {
                    p++;
787
                    rtsp_parse_range(&th->interleaved_min,
788 789 790
                                     &th->interleaved_max, &p);
                }
            } else if (!strcmp(parameter, "multicast")) {
791 792
                if (th->lower_transport == RTSP_LOWER_TRANSPORT_UDP)
                    th->lower_transport = RTSP_LOWER_TRANSPORT_UDP_MULTICAST;
793 794 795 796 797 798 799 800 801 802 803
            } 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);
804
                    if (inet_aton(buf, &ipaddr))
805 806 807 808 809 810 811 812 813 814 815 816 817 818 819
                        th->destination = ntohl(ipaddr.s_addr);
                }
            }
            while (*p != ';' && *p != '\0' && *p != ',')
                p++;
            if (*p == ';')
                p++;
        }
        if (*p == ',')
            p++;

        reply->nb_transports++;
    }
}

820
void rtsp_parse_line(RTSPMessageHeader *reply, const char *buf)
821 822 823 824 825
{
    const char *p;

    /* NOTE: we do case independent match for broken servers */
    p = buf;
826
    if (av_stristart(p, "Session:", &p)) {
827
        int t;
828
        get_word_sep(reply->session_id, sizeof(reply->session_id), ";", &p);
829 830 831 832
        if (av_stristart(p, ";timeout=", &p) &&
            (t = strtol(p, NULL, 10)) > 0) {
            reply->timeout = t;
        }
833
    } else if (av_stristart(p, "Content-Length:", &p)) {
834
        reply->content_length = strtol(p, NULL, 10);
835
    } else if (av_stristart(p, "Transport:", &p)) {
836
        rtsp_parse_transport(reply, p);
837
    } else if (av_stristart(p, "CSeq:", &p)) {
838
        reply->seq = strtol(p, NULL, 10);
839
    } else if (av_stristart(p, "Range:", &p)) {
840
        rtsp_parse_range_npt(p, &reply->range_start, &reply->range_end);
841 842 843
    } else if (av_stristart(p, "RealChallenge1:", &p)) {
        skip_spaces(&p);
        av_strlcpy(reply->real_challenge, p, sizeof(reply->real_challenge));
844 845 846
    } else if (av_stristart(p, "Server:", &p)) {
        skip_spaces(&p);
        av_strlcpy(reply->server, p, sizeof(reply->server));
847 848 849
    } else if (av_stristart(p, "Notice:", &p) ||
               av_stristart(p, "X-Notice:", &p)) {
        reply->notice = strtol(p, NULL, 10);
L
Luca Barbato 已提交
850 851 852
    } else if (av_stristart(p, "Location:", &p)) {
        skip_spaces(&p);
        av_strlcpy(reply->location, p , sizeof(reply->location));
853 854 855
    }
}

856 857 858 859 860 861 862
/* 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];

863
    ret = url_read_complete(rt->rtsp_hd, buf, 3);
864 865
    if (ret != 3)
        return;
866
    len = AV_RB16(buf + 1);
867 868 869

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

870 871 872 873 874
    /* skip payload */
    while (len > 0) {
        len1 = len;
        if (len1 > sizeof(buf))
            len1 = sizeof(buf);
875
        ret = url_read_complete(rt->rtsp_hd, buf, len1);
876 877 878 879 880
        if (ret != len1)
            return;
        len -= len1;
    }
}
881

882 883 884 885 886 887 888 889
/**
 * 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
890
 *                    be stored (length is in reply)
891 892 893 894 895 896 897 898 899 900 901 902
 * @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.
 */
903 904 905
static int rtsp_read_reply(AVFormatContext *s, RTSPMessageHeader *reply,
                           unsigned char **content_ptr,
                           int return_on_interleaved_data)
906 907 908 909 910
{
    RTSPState *rt = s->priv_data;
    char buf[4096], buf1[1024], *q;
    unsigned char ch;
    const char *p;
911
    int ret, content_length, line_count = 0;
912 913
    unsigned char *content = NULL;

914
    memset(reply, 0, sizeof(*reply));
915 916 917

    /* parse reply (XXX: use buffers) */
    rt->last_reply[0] = '\0';
918
    for (;;) {
919
        q = buf;
920
        for (;;) {
921
            ret = url_read_complete(rt->rtsp_hd, &ch, 1);
922
#ifdef DEBUG_RTP_TCP
923
            dprintf(s, "ret=%d c=%02x [%c]\n", ret, ch, ch);
924 925 926
#endif
            if (ret != 1)
                return -1;
927 928
            if (ch == '\n')
                break;
929 930
            if (ch == '$') {
                /* XXX: only parse it if first char on line ? */
931 932 933
                if (return_on_interleaved_data) {
                    return 1;
                } else
934 935
                rtsp_skip_packet(s);
            } else if (ch != '\r') {
936 937 938 939 940
                if ((q - buf) < sizeof(buf) - 1)
                    *q++ = ch;
            }
        }
        *q = '\0';
941 942 943

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

944 945 946 947 948 949 950 951 952 953 954
        /* 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 已提交
955 956
            av_strlcat(rt->last_reply, p,    sizeof(rt->last_reply));
            av_strlcat(rt->last_reply, "\n", sizeof(rt->last_reply));
957 958 959
        }
        line_count++;
    }
960

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

964 965 966 967
    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);
968
        (void)url_read_complete(rt->rtsp_hd, content, content_length);
969 970 971 972
        content[content_length] = '\0';
    }
    if (content_ptr)
        *content_ptr = content;
973 974
    else
        av_free(content);
975

976 977 978
    /* EOS */
    if (reply->notice == 2101 /* End-of-Stream Reached */      ||
        reply->notice == 2104 /* Start-of-Stream Reached */    ||
979
        reply->notice == 2306 /* Continuous Feed Terminated */) {
980
        rt->state = RTSP_STATE_IDLE;
981
    } else if (reply->notice >= 4400 && reply->notice < 5500) {
982
        return AVERROR(EIO); /* data or server error */
983
    } else if (reply->notice == 2401 /* Ticket Expired */ ||
984 985 986
             (reply->notice >= 5500 && reply->notice < 5600) /* end of term */ )
        return AVERROR(EPERM);

987
    return 0;
988 989
}

990 991 992 993
static void rtsp_send_cmd_with_content_async(AVFormatContext *s,
                                             const char *cmd,
                                             const unsigned char *send_content,
                                             int send_content_length)
994 995 996 997 998 999 1000 1001 1002 1003 1004 1005
{
    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));
    }
1006 1007
    if (rt->auth_b64)
        av_strlcatf(buf, sizeof(buf),
1008 1009
                    "Authorization: Basic %s\r\n",
                    rt->auth_b64);
1010 1011
    if (send_content_length > 0 && send_content)
        av_strlcatf(buf, sizeof(buf), "Content-Length: %d\r\n", send_content_length);
1012
    av_strlcat(buf, "\r\n", sizeof(buf));
1013 1014 1015

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

1016
    url_write(rt->rtsp_hd, buf, strlen(buf));
1017 1018
    if (send_content_length > 0 && send_content)
        url_write(rt->rtsp_hd, send_content, send_content_length);
1019 1020 1021
    rt->last_cmd_time = av_gettime();
}

1022 1023 1024 1025 1026
static void rtsp_send_cmd_async(AVFormatContext *s, const char *cmd)
{
    rtsp_send_cmd_with_content_async(s, cmd, NULL, 0);
}

1027 1028 1029
static void rtsp_send_cmd(AVFormatContext *s,
                          const char *cmd, RTSPMessageHeader *reply,
                          unsigned char **content_ptr)
1030
{
1031
    rtsp_send_cmd_async(s, cmd);
1032

1033
    rtsp_read_reply(s, reply, content_ptr, 0);
1034 1035
}

1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047
static void rtsp_send_cmd_with_content(AVFormatContext *s,
                                       const char *cmd,
                                       RTSPMessageHeader *reply,
                                       unsigned char **content_ptr,
                                       const unsigned char *send_content,
                                       int send_content_length)
{
    rtsp_send_cmd_with_content_async(s, cmd, send_content, send_content_length);

    rtsp_read_reply(s, reply, content_ptr, 0);
}

1048 1049 1050
/**
 * @returns 0 on success, <0 on error, 1 if protocol is unavailable.
 */
1051 1052
static int make_setup_request(AVFormatContext *s, const char *host, int port,
                              int lower_transport, const char *real_challenge)
1053 1054
{
    RTSPState *rt = s->priv_data;
1055
    int rtx, j, i, err, interleave = 0;
1056
    RTSPStream *rtsp_st;
1057
    RTSPMessageHeader reply1, *reply = &reply1;
1058
    char cmd[2048];
1059 1060
    const char *trans_pref;

1061
    if (rt->transport == RTSP_TRANSPORT_RDT)
1062 1063 1064
        trans_pref = "x-pn-tng";
    else
        trans_pref = "RTP/AVP";
1065

1066 1067 1068
    /* default timeout: 1 minute */
    rt->timeout = 60;

1069 1070
    /* for each stream, make the setup request */
    /* XXX: we assume the same server is used for the control of each
1071
     * RTSP stream */
R
Romain Degez 已提交
1072

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

1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087
        /**
         * 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 &&
1088 1089
                        !strcmp(rt->rtsp_streams[rtx]->control_url + len - 4,
                                "/rtx"))
1090 1091 1092 1093 1094 1095 1096 1097
                        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 已提交
1098
            rtsp_st = rt->rtsp_streams[i];
1099 1100

        /* RTP/UDP */
1101
        if (lower_transport == RTSP_LOWER_TRANSPORT_UDP) {
F
Fabrice Bellard 已提交
1102 1103
            char buf[256];

1104 1105 1106 1107 1108
            if (rt->server_type == RTSP_SERVER_WMS && i > 1) {
                port = reply->transports[0].client_port_min;
                goto have_port;
            }

F
Fabrice Bellard 已提交
1109
            /* first try in specified port range */
R
Romain Degez 已提交
1110
            if (RTSP_RTP_PORT_MIN != 0) {
1111 1112 1113 1114 1115 1116
                while (j <= RTSP_RTP_PORT_MAX) {
                    snprintf(buf, sizeof(buf), "rtp://%s?localport=%d",
                             host, j);
                    /* we will use two ports per rtp stream (rtp and rtcp) */
                    j += 2;
                    if (url_open(&rtsp_st->rtp_handle, buf, URL_RDWR) == 0)
F
Fabrice Bellard 已提交
1117 1118
                        goto rtp_opened;
                }
1119
            }
F
Fabrice Bellard 已提交
1120

1121 1122 1123 1124 1125 1126 1127
#if 0
            /* then try on any port */
            if (url_open(&rtsp_st->rtp_handle, "rtp://", URL_RDONLY) < 0) {
                err = AVERROR_INVALIDDATA;
                goto fail;
            }
#endif
F
Fabrice Bellard 已提交
1128 1129

        rtp_opened:
1130
            port = rtp_get_local_port(rtsp_st->rtp_handle);
1131
        have_port:
1132
            snprintf(transport, sizeof(transport) - 1,
1133 1134 1135 1136 1137
                     "%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);
1138 1139
            if (rt->transport == RTSP_TRANSPORT_RTP &&
                !(rt->server_type == RTSP_SERVER_WMS && i > 0))
1140
                av_strlcatf(transport, sizeof(transport), "-%d", port + 1);
1141 1142 1143
        }

        /* RTP/TCP */
1144
        else if (lower_transport == RTSP_LOWER_TRANSPORT_TCP) {
1145 1146 1147 1148
            /** 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 &&
1149 1150
                s->streams[rtsp_st->stream_index]->codec->codec_type ==
                    CODEC_TYPE_DATA)
1151
                continue;
1152
            snprintf(transport, sizeof(transport) - 1,
1153 1154 1155 1156 1157 1158 1159
                     "%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;
1160 1161
        }

1162
        else if (lower_transport == RTSP_LOWER_TRANSPORT_UDP_MULTICAST) {
1163
            snprintf(transport, sizeof(transport) - 1,
1164
                     "%s/UDP;multicast", trans_pref);
1165
        }
1166 1167 1168
        if (s->oformat) {
            av_strlcat(transport, ";mode=receive", sizeof(transport));
        } else if (rt->server_type == RTSP_SERVER_REAL ||
1169
            rt->server_type == RTSP_SERVER_WMS)
1170
            av_strlcat(transport, ";mode=play", sizeof(transport));
1171
        snprintf(cmd, sizeof(cmd),
F
Fabrice Bellard 已提交
1172 1173
                 "SETUP %s RTSP/1.0\r\n"
                 "Transport: %s\r\n",
1174
                 rtsp_st->control_url, transport);
1175
        if (i == 0 && rt->server_type == RTSP_SERVER_REAL) {
1176 1177 1178 1179 1180 1181 1182 1183
            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);
        }
1184
        rtsp_send_cmd(s, cmd, reply, NULL);
1185 1186 1187
        if (reply->status_code == 461 /* Unsupported protocol */ && i == 0) {
            err = 1;
            goto fail;
1188 1189
        } else if (reply->status_code != RTSP_STATUS_OK ||
                   reply->nb_transports != 1) {
1190 1191 1192 1193 1194 1195
            err = AVERROR_INVALIDDATA;
            goto fail;
        }

        /* XXX: same protocol for all streams is required */
        if (i > 0) {
1196 1197
            if (reply->transports[0].lower_transport != rt->lower_transport ||
                reply->transports[0].transport != rt->transport) {
1198 1199 1200 1201
                err = AVERROR_INVALIDDATA;
                goto fail;
            }
        } else {
1202
            rt->lower_transport = reply->transports[0].lower_transport;
1203
            rt->transport = reply->transports[0].transport;
1204 1205 1206
        }

        /* close RTP connection if not choosen */
1207 1208
        if (reply->transports[0].lower_transport != RTSP_LOWER_TRANSPORT_UDP &&
            (lower_transport == RTSP_LOWER_TRANSPORT_UDP)) {
1209 1210
            url_close(rtsp_st->rtp_handle);
            rtsp_st->rtp_handle = NULL;
1211 1212
        }

1213 1214
        switch(reply->transports[0].lower_transport) {
        case RTSP_LOWER_TRANSPORT_TCP:
1215 1216 1217
            rtsp_st->interleaved_min = reply->transports[0].interleaved_min;
            rtsp_st->interleaved_max = reply->transports[0].interleaved_max;
            break;
1218

1219 1220 1221 1222 1223 1224 1225 1226 1227 1228
        case RTSP_LOWER_TRANSPORT_UDP: {
            char url[1024];

            /* XXX: also use address if specified */
            snprintf(url, sizeof(url), "rtp://%s:%d",
                     host, reply->transports[0].server_port_min);
            if (!(rt->server_type == RTSP_SERVER_WMS && i > 1) &&
                rtp_set_remote_url(rtsp_st->rtp_handle, url) < 0) {
                err = AVERROR_INVALIDDATA;
                goto fail;
1229
            }
1230 1231 1232 1233
            /* Try to initialize the connection state in a
             * potential NAT router by sending dummy packets.
             * RTP/RTCP dummy packets are used for RDT, too.
             */
1234
            if (!(rt->server_type == RTSP_SERVER_WMS && i > 1) && s->iformat)
1235
                rtp_send_punch_packets(rtsp_st->rtp_handle);
1236
            break;
1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253 1254 1255 1256
        }
        case RTSP_LOWER_TRANSPORT_UDP_MULTICAST: {
            char url[1024];
            struct in_addr in;
            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;
            }
            snprintf(url, sizeof(url), "rtp://%s:%d?ttl=%d",
                     inet_ntoa(in), port, ttl);
            if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
                err = AVERROR_INVALIDDATA;
                goto fail;
1257 1258 1259
            }
            break;
        }
1260
        }
R
Romain Degez 已提交
1261

1262
        if ((err = rtsp_open_transport_ctx(s, rtsp_st)))
1263
            goto fail;
1264 1265
    }

1266 1267 1268
    if (reply->timeout > 0)
        rt->timeout = reply->timeout;

1269
    if (rt->server_type == RTSP_SERVER_REAL)
1270 1271
        rt->need_subscription = 1;

1272 1273 1274
    return 0;

fail:
1275
    for (i = 0; i < rt->nb_rtsp_streams; i++) {
1276 1277 1278 1279 1280
        if (rt->rtsp_streams[i]->rtp_handle) {
            url_close(rt->rtsp_streams[i]->rtp_handle);
            rt->rtsp_streams[i]->rtp_handle = NULL;
        }
    }
1281 1282 1283
    return err;
}

1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295
static int rtsp_read_play(AVFormatContext *s)
{
    RTSPState *rt = s->priv_data;
    RTSPMessageHeader reply1, *reply = &reply1;
    char cmd[1024];

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

    if (!(rt->server_type == RTSP_SERVER_REAL && rt->need_subscription)) {
        if (rt->state == RTSP_STATE_PAUSED) {
            snprintf(cmd, sizeof(cmd),
                     "PLAY %s RTSP/1.0\r\n",
1296
                     rt->control_uri);
1297 1298 1299 1300
        } else {
            snprintf(cmd, sizeof(cmd),
                     "PLAY %s RTSP/1.0\r\n"
                     "Range: npt=%0.3f-\r\n",
1301
                     rt->control_uri,
1302 1303 1304 1305 1306 1307 1308
                     (double)rt->seek_timestamp / AV_TIME_BASE);
        }
        rtsp_send_cmd(s, cmd, reply, NULL);
        if (reply->status_code != RTSP_STATUS_OK) {
            return -1;
        }
    }
1309
    rt->state = RTSP_STATE_STREAMING;
1310 1311 1312
    return 0;
}

1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351
static int rtsp_setup_input_streams(AVFormatContext *s)
{
    RTSPState *rt = s->priv_data;
    RTSPMessageHeader reply1, *reply = &reply1;
    char cmd[1024];
    unsigned char *content = NULL;
    int ret;

    /* describe the stream */
    snprintf(cmd, sizeof(cmd),
             "DESCRIBE %s RTSP/1.0\r\n"
             "Accept: application/sdp\r\n",
             s->filename);
    if (rt->server_type == RTSP_SERVER_REAL) {
        /**
         * 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));
    }
    rtsp_send_cmd(s, cmd, reply, &content);
    if (!content)
        return AVERROR_INVALIDDATA;
    if (reply->status_code != RTSP_STATUS_OK) {
        av_freep(&content);
        return AVERROR_INVALIDDATA;
    }

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

    return 0;
}

1352
static int rtsp_connect(AVFormatContext *s)
1353 1354
{
    RTSPState *rt = s->priv_data;
1355 1356
    char host[1024], path[1024], tcpname[1024], cmd[2048], auth[128];
    char *option_list, *option, *filename;
1357
    URLContext *rtsp_hd;
1358
    int port, err;
1359
    RTSPMessageHeader reply1, *reply = &reply1;
1360
    int lower_transport_mask = 0;
1361
    char real_challenge[64];
1362
redirect:
1363
    /* extract hostname and port */
1364
    url_split(NULL, 0, auth, sizeof(auth),
1365
              host, sizeof(host), &port, path, sizeof(path), s->filename);
1366 1367 1368 1369 1370 1371 1372 1373 1374 1375
    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;
        }
    }
1376 1377 1378 1379 1380 1381
    if (port < 0)
        port = RTSP_DEFAULT_PORT;

    /* search for options */
    option_list = strchr(path, '?');
    if (option_list) {
1382
        filename = strchr(s->filename, '?');
1383
        while (option_list) {
1384
            /* move the option pointer */
1385
            option = ++option_list;
1386 1387
            option_list = strchr(option_list, '&');
            if (option_list)
1388 1389
                *option_list = 0;

1390
            /* handle the options */
1391
            if (!strcmp(option, "udp")) {
1392
                lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_UDP);
1393
            } else if (!strcmp(option, "multicast")) {
1394
                lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_UDP_MULTICAST);
1395
            } else if (!strcmp(option, "tcp")) {
1396
                lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_TCP);
1397
            } else {
1398 1399 1400 1401
                strcpy(++filename, option);
                filename += strlen(option);
                if (option_list) *filename = '&';
            }
1402
        }
1403
        *filename = 0;
1404 1405
    }

1406
    if (!lower_transport_mask)
1407
        lower_transport_mask = (1 << RTSP_LOWER_TRANSPORT_NB) - 1;
1408 1409 1410

    /* open the tcp connexion */
    snprintf(tcpname, sizeof(tcpname), "tcp://%s:%d", host, port);
1411 1412 1413 1414
    if (url_open(&rtsp_hd, tcpname, URL_RDWR) < 0) {
        err = AVERROR(EIO);
        goto fail;
    }
1415 1416 1417
    rt->rtsp_hd = rtsp_hd;
    rt->seq = 0;

1418 1419
    /* request options supported by the server; this also detects server
     * type */
1420 1421
    av_strlcpy(rt->control_uri, s->filename,
               sizeof(rt->control_uri));
1422 1423 1424
    for (rt->server_type = RTSP_SERVER_RTP;;) {
        snprintf(cmd, sizeof(cmd),
                 "OPTIONS %s RTSP/1.0\r\n", s->filename);
1425
        if (rt->server_type == RTSP_SERVER_REAL)
1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447
            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 */
1448 1449
        if (rt->server_type != RTSP_SERVER_REAL && reply->real_challenge[0]) {
            rt->server_type = RTSP_SERVER_REAL;
1450
            continue;
1451 1452
        } else if (!strncasecmp(reply->server, "WMServer/", 9)) {
            rt->server_type = RTSP_SERVER_WMS;
1453
        } else if (rt->server_type == RTSP_SERVER_REAL)
1454 1455 1456 1457
            strcpy(real_challenge, reply->real_challenge);
        break;
    }

1458 1459
    err = rtsp_setup_input_streams(s);
    if (err)
1460 1461
        goto fail;

1462
    do {
1463 1464
        int lower_transport = ff_log2_tab[lower_transport_mask &
                                  ~(lower_transport_mask - 1)];
1465

1466
        err = make_setup_request(s, host, port, lower_transport,
1467
                                 rt->server_type == RTSP_SERVER_REAL ?
1468
                                     real_challenge : NULL);
1469
        if (err < 0)
1470
            goto fail;
1471 1472
        lower_transport_mask &= ~(1 << lower_transport);
        if (lower_transport_mask == 0 && err == 1) {
1473
            err = AVERROR(FF_NETERROR(EPROTONOSUPPORT));
1474 1475 1476
            goto fail;
        }
    } while (err);
1477

1478
    rt->state = RTSP_STATE_IDLE;
1479
    rt->seek_timestamp = 0; /* default is to start stream at position zero */
1480 1481
    return 0;
 fail:
1482
    rtsp_close_streams(s);
1483
    url_close(rt->rtsp_hd);
L
Luca Barbato 已提交
1484 1485 1486 1487 1488 1489 1490
    if (reply->status_code >=300 && reply->status_code < 400) {
        av_strlcpy(s->filename, reply->location, sizeof(s->filename));
        av_log(s, AV_LOG_INFO, "Status %d: Redirecting to %s\n",
               reply->status_code,
               s->filename);
        goto redirect;
    }
1491 1492 1493
    return err;
}

1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516
static int rtsp_read_header(AVFormatContext *s,
                            AVFormatParameters *ap)
{
    RTSPState *rt = s->priv_data;
    int ret;

    ret = rtsp_connect(s);
    if (ret)
        return ret;

    if (ap->initial_pause) {
         /* do not start immediately */
    } else {
         if (rtsp_read_play(s) < 0) {
            rtsp_close_streams(s);
            url_close(rt->rtsp_hd);
            return AVERROR_INVALIDDATA;
        }
    }

    return 0;
}

R
Ronald S. Bultje 已提交
1517 1518 1519 1520 1521 1522 1523 1524 1525
static int udp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
                           uint8_t *buf, int buf_size)
{
    RTSPState *rt = s->priv_data;
    RTSPStream *rtsp_st;
    fd_set rfds;
    int fd, fd_max, n, i, ret, tcp_fd;
    struct timeval tv;

1526
    for (;;) {
R
Ronald S. Bultje 已提交
1527 1528 1529 1530 1531 1532 1533 1534 1535 1536
        if (url_interrupt_cb())
            return AVERROR(EINTR);
        FD_ZERO(&rfds);
        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;
        }
1537
        for (i = 0; i < rt->nb_rtsp_streams; i++) {
R
Ronald S. Bultje 已提交
1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551
            rtsp_st = rt->rtsp_streams[i];
            if (rtsp_st->rtp_handle) {
                /* currently, we cannot probe RTCP handle because of
                 * blocking restrictions */
                fd = url_get_file_handle(rtsp_st->rtp_handle);
                if (fd > fd_max)
                    fd_max = fd;
                FD_SET(fd, &rfds);
            }
        }
        tv.tv_sec = 0;
        tv.tv_usec = 100 * 1000;
        n = select(fd_max + 1, &rfds, NULL, NULL, &tv);
        if (n > 0) {
1552
            for (i = 0; i < rt->nb_rtsp_streams; i++) {
R
Ronald S. Bultje 已提交
1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565
                rtsp_st = rt->rtsp_streams[i];
                if (rtsp_st->rtp_handle) {
                    fd = url_get_file_handle(rtsp_st->rtp_handle);
                    if (FD_ISSET(fd, &rfds)) {
                        ret = url_read(rtsp_st->rtp_handle, buf, buf_size);
                        if (ret > 0) {
                            *prtsp_st = rtsp_st;
                            return ret;
                        }
                    }
                }
            }
#if CONFIG_RTSP_DEMUXER
1566
            if (tcp_fd != -1 && FD_ISSET(tcp_fd, &rfds)) {
R
Ronald S. Bultje 已提交
1567 1568 1569 1570
                RTSPMessageHeader reply;

                rtsp_read_reply(s, &reply, NULL, 0);
                /* XXX: parse message */
1571
                if (rt->state != RTSP_STATE_STREAMING)
R
Ronald S. Bultje 已提交
1572 1573 1574 1575 1576 1577 1578
                    return 0;
            }
#endif
        }
    }
}

1579 1580
static int tcp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
                           uint8_t *buf, int buf_size)
1581 1582
{
    RTSPState *rt = s->priv_data;
F
Fabrice Bellard 已提交
1583
    int id, len, i, ret;
1584 1585
    RTSPStream *rtsp_st;

F
Fabrice Bellard 已提交
1586
#ifdef DEBUG_RTP_TCP
1587
    dprintf(s, "tcp_read_packet:\n");
F
Fabrice Bellard 已提交
1588
#endif
1589 1590
redo:
    for (;;) {
1591 1592 1593 1594
        RTSPMessageHeader reply;

        ret = rtsp_read_reply(s, &reply, NULL, 1);
        if (ret == -1)
1595
            return -1;
1596
        if (ret == 1) /* received '$' */
1597
            break;
1598
        /* XXX: parse message */
1599
        if (rt->state != RTSP_STATE_STREAMING)
1600
            return 0;
1601
    }
1602
    ret = url_read_complete(rt->rtsp_hd, buf, 3);
F
Fabrice Bellard 已提交
1603
    if (ret != 3)
1604
        return -1;
1605
    id  = buf[0];
1606
    len = AV_RB16(buf + 1);
F
Fabrice Bellard 已提交
1607
#ifdef DEBUG_RTP_TCP
1608
    dprintf(s, "id=%d len=%d\n", id, len);
F
Fabrice Bellard 已提交
1609
#endif
1610
    if (len > buf_size || len < 12)
1611 1612
        goto redo;
    /* get the data */
1613
    ret = url_read_complete(rt->rtsp_hd, buf, len);
F
Fabrice Bellard 已提交
1614
    if (ret != len)
1615
        return -1;
1616
    if (rt->transport == RTSP_TRANSPORT_RDT &&
1617
        ff_rdt_parse_header(buf, len, &id, NULL, NULL, NULL, NULL) < 0)
1618
        return -1;
1619

1620
    /* find the matching stream */
1621
    for (i = 0; i < rt->nb_rtsp_streams; i++) {
1622
        rtsp_st = rt->rtsp_streams[i];
1623 1624
        if (id >= rtsp_st->interleaved_min &&
            id <= rtsp_st->interleaved_max)
1625 1626 1627
            goto found;
    }
    goto redo;
1628
found:
1629 1630
    *prtsp_st = rtsp_st;
    return len;
1631 1632
}

R
Ronald S. Bultje 已提交
1633 1634 1635 1636 1637 1638 1639 1640 1641
static int rtsp_fetch_packet(AVFormatContext *s, AVPacket *pkt)
{
    RTSPState *rt = s->priv_data;
    int ret, len;
    uint8_t buf[10 * RTP_MAX_PACKET_LENGTH];
    RTSPStream *rtsp_st;

    /* get next frames from the same RTP packet */
    if (rt->cur_transport_priv) {
1642
        if (rt->transport == RTSP_TRANSPORT_RDT) {
R
Ronald S. Bultje 已提交
1643
            ret = ff_rdt_parse_packet(rt->cur_transport_priv, pkt, NULL, 0);
1644
        } else
R
Ronald S. Bultje 已提交
1645 1646 1647 1648 1649 1650
            ret = rtp_parse_packet(rt->cur_transport_priv, pkt, NULL, 0);
        if (ret == 0) {
            rt->cur_transport_priv = NULL;
            return 0;
        } else if (ret == 1) {
            return 0;
1651
        } else
R
Ronald S. Bultje 已提交
1652 1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674
            rt->cur_transport_priv = NULL;
    }

    /* read next RTP packet */
 redo:
    switch(rt->lower_transport) {
    default:
#if CONFIG_RTSP_DEMUXER
    case RTSP_LOWER_TRANSPORT_TCP:
        len = tcp_read_packet(s, &rtsp_st, buf, sizeof(buf));
        break;
#endif
    case RTSP_LOWER_TRANSPORT_UDP:
    case RTSP_LOWER_TRANSPORT_UDP_MULTICAST:
        len = udp_read_packet(s, &rtsp_st, buf, sizeof(buf));
        if (len >=0 && rtsp_st->transport_priv && rt->transport == RTSP_TRANSPORT_RTP)
            rtp_check_and_send_back_rr(rtsp_st->transport_priv, len);
        break;
    }
    if (len < 0)
        return len;
    if (len == 0)
        return AVERROR_EOF;
1675
    if (rt->transport == RTSP_TRANSPORT_RDT) {
R
Ronald S. Bultje 已提交
1676
        ret = ff_rdt_parse_packet(rtsp_st->transport_priv, pkt, buf, len);
1677
    } else
R
Ronald S. Bultje 已提交
1678 1679 1680
        ret = rtp_parse_packet(rtsp_st->transport_priv, pkt, buf, len);
    if (ret < 0)
        goto redo;
1681
    if (ret == 1)
R
Ronald S. Bultje 已提交
1682 1683 1684 1685 1686 1687
        /* more packets may follow, so we save the RTP context */
        rt->cur_transport_priv = rtsp_st->transport_priv;

    return ret;
}

1688
static int rtsp_read_packet(AVFormatContext *s, AVPacket *pkt)
1689 1690
{
    RTSPState *rt = s->priv_data;
1691
    int ret;
1692 1693
    RTSPMessageHeader reply1, *reply = &reply1;
    char cmd[1024];
1694

1695
    if (rt->server_type == RTSP_SERVER_REAL) {
1696
        int i;
1697
        enum AVDiscard cache[MAX_STREAMS];
1698

1699 1700 1701 1702 1703 1704
        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)) {
1705
                snprintf(cmd, sizeof(cmd),
R
Ronald S. Bultje 已提交
1706 1707
                         "SET_PARAMETER %s RTSP/1.0\r\n"
                         "Unsubscribe: %s\r\n",
1708
                         rt->control_uri, rt->last_subscription);
1709 1710 1711 1712 1713
                rtsp_send_cmd(s, cmd, reply, NULL);
                if (reply->status_code != RTSP_STATUS_OK)
                    return AVERROR_INVALIDDATA;
                rt->need_subscription = 1;
            }
1714 1715
        }

1716 1717 1718 1719 1720 1721 1722 1723 1724 1725
        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: ",
1726
                     rt->control_uri);
1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749
            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;

1750
            if (rt->state == RTSP_STATE_STREAMING)
1751 1752
                rtsp_read_play (s);
        }
1753 1754
    }

L
Luca Barbato 已提交
1755
    ret = rtsp_fetch_packet(s, pkt);
1756
    if (ret < 0)
1757
        return ret;
1758 1759 1760 1761 1762 1763 1764 1765

    /* 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",
1766
                     rt->control_uri);
1767
            rtsp_send_cmd_async(s, cmd);
1768
        } else {
1769
            rtsp_send_cmd_async(s, "OPTIONS * RTSP/1.0\r\n");
1770 1771 1772
        }
    }

1773
    return 0;
1774 1775
}

1776 1777
/* pause the stream */
static int rtsp_read_pause(AVFormatContext *s)
1778
{
1779
    RTSPState *rt = s->priv_data;
1780
    RTSPMessageHeader reply1, *reply = &reply1;
1781 1782 1783
    char cmd[1024];

    rt = s->priv_data;
1784

1785
    if (rt->state != RTSP_STATE_STREAMING)
1786
        return 0;
1787
    else if (!(rt->server_type == RTSP_SERVER_REAL && rt->need_subscription)) {
1788 1789
        snprintf(cmd, sizeof(cmd),
                 "PAUSE %s RTSP/1.0\r\n",
1790
                 rt->control_uri);
1791 1792 1793 1794
        rtsp_send_cmd(s, cmd, reply, NULL);
        if (reply->status_code != RTSP_STATUS_OK) {
            return -1;
        }
1795
    }
1796 1797
    rt->state = RTSP_STATE_PAUSED;
    return 0;
1798 1799
}

1800
static int rtsp_read_seek(AVFormatContext *s, int stream_index,
1801
                          int64_t timestamp, int flags)
1802 1803
{
    RTSPState *rt = s->priv_data;
1804

1805 1806 1807
    rt->seek_timestamp = av_rescale_q(timestamp,
                                      s->streams[stream_index]->time_base,
                                      AV_TIME_BASE_Q);
1808 1809 1810 1811
    switch(rt->state) {
    default:
    case RTSP_STATE_IDLE:
        break;
1812
    case RTSP_STATE_STREAMING:
1813 1814 1815
        if (rtsp_read_pause(s) != 0)
            return -1;
        rt->state = RTSP_STATE_SEEKING;
1816 1817 1818 1819 1820 1821 1822 1823 1824 1825
        if (rtsp_read_play(s) != 0)
            return -1;
        break;
    case RTSP_STATE_PAUSED:
        rt->state = RTSP_STATE_IDLE;
        break;
    }
    return 0;
}

1826 1827 1828 1829 1830
static int rtsp_read_close(AVFormatContext *s)
{
    RTSPState *rt = s->priv_data;
    char cmd[1024];

F
Fabrice Bellard 已提交
1831
#if 0
1832
    /* NOTE: it is valid to flush the buffer here */
1833
    if (rt->lower_transport == RTSP_LOWER_TRANSPORT_TCP) {
1834 1835
        url_fclose(&rt->rtsp_gb);
    }
F
Fabrice Bellard 已提交
1836
#endif
1837
    snprintf(cmd, sizeof(cmd),
F
Fabrice Bellard 已提交
1838
             "TEARDOWN %s RTSP/1.0\r\n",
1839
             s->filename);
1840
    rtsp_send_cmd_async(s, cmd);
1841

1842
    rtsp_close_streams(s);
1843 1844 1845 1846
    url_close(rt->rtsp_hd);
    return 0;
}

1847
AVInputFormat rtsp_demuxer = {
1848
    "rtsp",
1849
    NULL_IF_CONFIG_SMALL("RTSP input format"),
1850 1851 1852 1853 1854
    sizeof(RTSPState),
    rtsp_probe,
    rtsp_read_header,
    rtsp_read_packet,
    rtsp_read_close,
1855
    rtsp_read_seek,
1856
    .flags = AVFMT_NOFILE,
1857 1858
    .read_play = rtsp_read_play,
    .read_pause = rtsp_read_pause,
1859
};
1860
#endif
1861

1862
static int sdp_probe(AVProbeData *p1)
1863
{
1864
    const char *p = p1->buf, *p_end = p1->buf + p1->buf_size;
1865 1866

    /* we look for a line beginning "c=IN IP4" */
1867
    while (p < p_end && *p != '\0') {
1868 1869
        if (p + sizeof("c=IN IP4") - 1 < p_end &&
            av_strstart(p, "c=IN IP4", NULL))
1870
            return AVPROBE_SCORE_MAX / 2;
1871

1872
        while (p < p_end - 1 && *p != '\n') p++;
1873
        if (++p >= p_end)
1874 1875 1876 1877
            break;
        if (*p == '\r')
            p++;
    }
1878 1879 1880 1881 1882
    return 0;
}

#define SDP_MAX_SIZE 8192

1883
static int sdp_read_header(AVFormatContext *s, AVFormatParameters *ap)
1884
{
1885
    RTSPState *rt = s->priv_data;
1886 1887 1888 1889 1890 1891 1892 1893
    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);
1894
    size = get_buffer(s->pb, content, SDP_MAX_SIZE - 1);
1895 1896 1897 1898 1899 1900 1901 1902 1903 1904
    if (size <= 0) {
        av_free(content);
        return AVERROR_INVALIDDATA;
    }
    content[size] ='\0';

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

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

1908
        snprintf(url, sizeof(url), "rtp://%s:%d?localport=%d&ttl=%d",
1909
                 inet_ntoa(rtsp_st->sdp_ip),
1910
                 rtsp_st->sdp_port,
1911
                 rtsp_st->sdp_port,
1912
                 rtsp_st->sdp_ttl);
1913
        if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
1914 1915 1916
            err = AVERROR_INVALIDDATA;
            goto fail;
        }
1917
        if ((err = rtsp_open_transport_ctx(s, rtsp_st)))
1918
            goto fail;
1919 1920
    }
    return 0;
1921
fail:
1922
    rtsp_close_streams(s);
1923 1924 1925 1926 1927
    return err;
}

static int sdp_read_close(AVFormatContext *s)
{
1928
    rtsp_close_streams(s);
1929 1930 1931
    return 0;
}

1932
AVInputFormat sdp_demuxer = {
1933
    "sdp",
1934
    NULL_IF_CONFIG_SMALL("SDP"),
1935 1936 1937
    sizeof(RTSPState),
    sdp_probe,
    sdp_read_header,
L
Luca Barbato 已提交
1938
    rtsp_fetch_packet,
1939 1940
    sdp_read_close,
};