rtsp.c 65.0 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 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399
static int rtsp_setup_output_streams(AVFormatContext *s)
{
    RTSPState *rt = s->priv_data;
    RTSPMessageHeader reply1, *reply = &reply1;
    char cmd[1024];
    int i;
    char *sdp;

    /* Announce the stream */
    snprintf(cmd, sizeof(cmd),
             "ANNOUNCE %s RTSP/1.0\r\n"
             "Content-Type: application/sdp\r\n",
             s->filename);
    sdp = av_mallocz(8192);
    if (sdp == NULL)
        return AVERROR(ENOMEM);
    if (avf_sdp_create(&s, 1, sdp, 8192)) {
        av_free(sdp);
        return AVERROR_INVALIDDATA;
    }
    av_log(s, AV_LOG_INFO, "SDP:\n%s\n", sdp);
    rtsp_send_cmd_with_content(s, cmd, reply, NULL, sdp, strlen(sdp));
    av_free(sdp);
    if (reply->status_code != RTSP_STATUS_OK)
        return AVERROR_INVALIDDATA;

    /* Set up the RTSPStreams for each AVStream */
    for (i = 0; i < s->nb_streams; i++) {
        RTSPStream *rtsp_st;
        AVStream *st = s->streams[i];

        rtsp_st = av_mallocz(sizeof(RTSPStream));
        if (!rtsp_st)
            return AVERROR(ENOMEM);
        dynarray_add(&rt->rtsp_streams, &rt->nb_rtsp_streams, rtsp_st);

        st->priv_data = rtsp_st;
        rtsp_st->stream_index = i;

        av_strlcpy(rtsp_st->control_url, s->filename, sizeof(rtsp_st->control_url));
        /* Note, this must match the relative uri set in the sdp content */
        av_strlcatf(rtsp_st->control_url, sizeof(rtsp_st->control_url),
                    "/streamid=%d", i);
    }

    return 0;
}

1400
static int rtsp_connect(AVFormatContext *s)
1401 1402
{
    RTSPState *rt = s->priv_data;
1403 1404
    char host[1024], path[1024], tcpname[1024], cmd[2048], auth[128];
    char *option_list, *option, *filename;
1405
    URLContext *rtsp_hd;
1406
    int port, err;
1407
    RTSPMessageHeader reply1, *reply = &reply1;
1408
    int lower_transport_mask = 0;
1409
    char real_challenge[64];
1410
redirect:
1411
    /* extract hostname and port */
1412
    url_split(NULL, 0, auth, sizeof(auth),
1413
              host, sizeof(host), &port, path, sizeof(path), s->filename);
1414 1415 1416 1417 1418 1419 1420 1421 1422 1423
    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;
        }
    }
1424 1425 1426 1427 1428 1429
    if (port < 0)
        port = RTSP_DEFAULT_PORT;

    /* search for options */
    option_list = strchr(path, '?');
    if (option_list) {
1430
        filename = strchr(s->filename, '?');
1431
        while (option_list) {
1432
            /* move the option pointer */
1433
            option = ++option_list;
1434 1435
            option_list = strchr(option_list, '&');
            if (option_list)
1436 1437
                *option_list = 0;

1438
            /* handle the options */
1439
            if (!strcmp(option, "udp")) {
1440
                lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_UDP);
1441
            } else if (!strcmp(option, "multicast")) {
1442
                lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_UDP_MULTICAST);
1443
            } else if (!strcmp(option, "tcp")) {
1444
                lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_TCP);
1445
            } else {
1446 1447 1448 1449
                strcpy(++filename, option);
                filename += strlen(option);
                if (option_list) *filename = '&';
            }
1450
        }
1451
        *filename = 0;
1452 1453
    }

1454
    if (!lower_transport_mask)
1455
        lower_transport_mask = (1 << RTSP_LOWER_TRANSPORT_NB) - 1;
1456

1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467
    if (s->oformat) {
        /* Only UDP output is supported at the moment. */
        lower_transport_mask &= 1 << RTSP_LOWER_TRANSPORT_UDP;
        if (!lower_transport_mask) {
            av_log(s, AV_LOG_ERROR, "Unsupported lower transport method, "
                                    "only UDP is supported for output.\n");
            err = AVERROR(EINVAL);
            goto fail;
        }
    }

1468 1469
    /* open the tcp connexion */
    snprintf(tcpname, sizeof(tcpname), "tcp://%s:%d", host, port);
1470 1471 1472 1473
    if (url_open(&rtsp_hd, tcpname, URL_RDWR) < 0) {
        err = AVERROR(EIO);
        goto fail;
    }
1474 1475 1476
    rt->rtsp_hd = rtsp_hd;
    rt->seq = 0;

1477 1478
    /* request options supported by the server; this also detects server
     * type */
1479 1480
    av_strlcpy(rt->control_uri, s->filename,
               sizeof(rt->control_uri));
1481 1482 1483
    for (rt->server_type = RTSP_SERVER_RTP;;) {
        snprintf(cmd, sizeof(cmd),
                 "OPTIONS %s RTSP/1.0\r\n", s->filename);
1484
        if (rt->server_type == RTSP_SERVER_REAL)
1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506
            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 */
1507 1508
        if (rt->server_type != RTSP_SERVER_REAL && reply->real_challenge[0]) {
            rt->server_type = RTSP_SERVER_REAL;
1509
            continue;
1510 1511
        } else if (!strncasecmp(reply->server, "WMServer/", 9)) {
            rt->server_type = RTSP_SERVER_WMS;
1512
        } else if (rt->server_type == RTSP_SERVER_REAL)
1513 1514 1515 1516
            strcpy(real_challenge, reply->real_challenge);
        break;
    }

1517
    if (s->iformat)
1518
    err = rtsp_setup_input_streams(s);
1519 1520
    else
        err = rtsp_setup_output_streams(s);
1521
    if (err)
1522 1523
        goto fail;

1524
    do {
1525 1526
        int lower_transport = ff_log2_tab[lower_transport_mask &
                                  ~(lower_transport_mask - 1)];
1527

1528
        err = make_setup_request(s, host, port, lower_transport,
1529
                                 rt->server_type == RTSP_SERVER_REAL ?
1530
                                     real_challenge : NULL);
1531
        if (err < 0)
1532
            goto fail;
1533 1534
        lower_transport_mask &= ~(1 << lower_transport);
        if (lower_transport_mask == 0 && err == 1) {
1535
            err = AVERROR(FF_NETERROR(EPROTONOSUPPORT));
1536 1537 1538
            goto fail;
        }
    } while (err);
1539

1540
    rt->state = RTSP_STATE_IDLE;
1541
    rt->seek_timestamp = 0; /* default is to start stream at position zero */
1542 1543
    return 0;
 fail:
1544
    rtsp_close_streams(s);
1545
    url_close(rt->rtsp_hd);
1546
    if (reply->status_code >=300 && reply->status_code < 400 && s->iformat) {
L
Luca Barbato 已提交
1547 1548 1549 1550 1551 1552
        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;
    }
1553 1554 1555
    return err;
}

1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578
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 已提交
1579 1580 1581 1582 1583 1584 1585 1586 1587
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;

1588
    for (;;) {
R
Ronald S. Bultje 已提交
1589 1590 1591 1592 1593 1594 1595 1596 1597 1598
        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;
        }
1599
        for (i = 0; i < rt->nb_rtsp_streams; i++) {
R
Ronald S. Bultje 已提交
1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613
            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) {
1614
            for (i = 0; i < rt->nb_rtsp_streams; i++) {
R
Ronald S. Bultje 已提交
1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627
                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
1628
            if (tcp_fd != -1 && FD_ISSET(tcp_fd, &rfds)) {
R
Ronald S. Bultje 已提交
1629 1630 1631 1632
                RTSPMessageHeader reply;

                rtsp_read_reply(s, &reply, NULL, 0);
                /* XXX: parse message */
1633
                if (rt->state != RTSP_STATE_STREAMING)
R
Ronald S. Bultje 已提交
1634 1635 1636 1637 1638 1639 1640
                    return 0;
            }
#endif
        }
    }
}

1641 1642
static int tcp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
                           uint8_t *buf, int buf_size)
1643 1644
{
    RTSPState *rt = s->priv_data;
F
Fabrice Bellard 已提交
1645
    int id, len, i, ret;
1646 1647
    RTSPStream *rtsp_st;

F
Fabrice Bellard 已提交
1648
#ifdef DEBUG_RTP_TCP
1649
    dprintf(s, "tcp_read_packet:\n");
F
Fabrice Bellard 已提交
1650
#endif
1651 1652
redo:
    for (;;) {
1653 1654 1655 1656
        RTSPMessageHeader reply;

        ret = rtsp_read_reply(s, &reply, NULL, 1);
        if (ret == -1)
1657
            return -1;
1658
        if (ret == 1) /* received '$' */
1659
            break;
1660
        /* XXX: parse message */
1661
        if (rt->state != RTSP_STATE_STREAMING)
1662
            return 0;
1663
    }
1664
    ret = url_read_complete(rt->rtsp_hd, buf, 3);
F
Fabrice Bellard 已提交
1665
    if (ret != 3)
1666
        return -1;
1667
    id  = buf[0];
1668
    len = AV_RB16(buf + 1);
F
Fabrice Bellard 已提交
1669
#ifdef DEBUG_RTP_TCP
1670
    dprintf(s, "id=%d len=%d\n", id, len);
F
Fabrice Bellard 已提交
1671
#endif
1672
    if (len > buf_size || len < 12)
1673 1674
        goto redo;
    /* get the data */
1675
    ret = url_read_complete(rt->rtsp_hd, buf, len);
F
Fabrice Bellard 已提交
1676
    if (ret != len)
1677
        return -1;
1678
    if (rt->transport == RTSP_TRANSPORT_RDT &&
1679
        ff_rdt_parse_header(buf, len, &id, NULL, NULL, NULL, NULL) < 0)
1680
        return -1;
1681

1682
    /* find the matching stream */
1683
    for (i = 0; i < rt->nb_rtsp_streams; i++) {
1684
        rtsp_st = rt->rtsp_streams[i];
1685 1686
        if (id >= rtsp_st->interleaved_min &&
            id <= rtsp_st->interleaved_max)
1687 1688 1689
            goto found;
    }
    goto redo;
1690
found:
1691 1692
    *prtsp_st = rtsp_st;
    return len;
1693 1694
}

R
Ronald S. Bultje 已提交
1695 1696 1697 1698 1699 1700 1701 1702 1703
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) {
1704
        if (rt->transport == RTSP_TRANSPORT_RDT) {
R
Ronald S. Bultje 已提交
1705
            ret = ff_rdt_parse_packet(rt->cur_transport_priv, pkt, NULL, 0);
1706
        } else
R
Ronald S. Bultje 已提交
1707 1708 1709 1710 1711 1712
            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;
1713
        } else
R
Ronald S. Bultje 已提交
1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736
            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;
1737
    if (rt->transport == RTSP_TRANSPORT_RDT) {
R
Ronald S. Bultje 已提交
1738
        ret = ff_rdt_parse_packet(rtsp_st->transport_priv, pkt, buf, len);
1739
    } else
R
Ronald S. Bultje 已提交
1740 1741 1742
        ret = rtp_parse_packet(rtsp_st->transport_priv, pkt, buf, len);
    if (ret < 0)
        goto redo;
1743
    if (ret == 1)
R
Ronald S. Bultje 已提交
1744 1745 1746 1747 1748 1749
        /* more packets may follow, so we save the RTP context */
        rt->cur_transport_priv = rtsp_st->transport_priv;

    return ret;
}

1750
static int rtsp_read_packet(AVFormatContext *s, AVPacket *pkt)
1751 1752
{
    RTSPState *rt = s->priv_data;
1753
    int ret;
1754 1755
    RTSPMessageHeader reply1, *reply = &reply1;
    char cmd[1024];
1756

1757
    if (rt->server_type == RTSP_SERVER_REAL) {
1758
        int i;
1759
        enum AVDiscard cache[MAX_STREAMS];
1760

1761 1762 1763 1764 1765 1766
        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)) {
1767
                snprintf(cmd, sizeof(cmd),
R
Ronald S. Bultje 已提交
1768 1769
                         "SET_PARAMETER %s RTSP/1.0\r\n"
                         "Unsubscribe: %s\r\n",
1770
                         rt->control_uri, rt->last_subscription);
1771 1772 1773 1774 1775
                rtsp_send_cmd(s, cmd, reply, NULL);
                if (reply->status_code != RTSP_STATUS_OK)
                    return AVERROR_INVALIDDATA;
                rt->need_subscription = 1;
            }
1776 1777
        }

1778 1779 1780 1781 1782 1783 1784 1785 1786 1787
        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: ",
1788
                     rt->control_uri);
1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810 1811
            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;

1812
            if (rt->state == RTSP_STATE_STREAMING)
1813 1814
                rtsp_read_play (s);
        }
1815 1816
    }

L
Luca Barbato 已提交
1817
    ret = rtsp_fetch_packet(s, pkt);
1818
    if (ret < 0)
1819
        return ret;
1820 1821 1822 1823 1824 1825 1826 1827

    /* 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",
1828
                     rt->control_uri);
1829
            rtsp_send_cmd_async(s, cmd);
1830
        } else {
1831
            rtsp_send_cmd_async(s, "OPTIONS * RTSP/1.0\r\n");
1832 1833 1834
        }
    }

1835
    return 0;
1836 1837
}

1838 1839
/* pause the stream */
static int rtsp_read_pause(AVFormatContext *s)
1840
{
1841
    RTSPState *rt = s->priv_data;
1842
    RTSPMessageHeader reply1, *reply = &reply1;
1843 1844 1845
    char cmd[1024];

    rt = s->priv_data;
1846

1847
    if (rt->state != RTSP_STATE_STREAMING)
1848
        return 0;
1849
    else if (!(rt->server_type == RTSP_SERVER_REAL && rt->need_subscription)) {
1850 1851
        snprintf(cmd, sizeof(cmd),
                 "PAUSE %s RTSP/1.0\r\n",
1852
                 rt->control_uri);
1853 1854 1855 1856
        rtsp_send_cmd(s, cmd, reply, NULL);
        if (reply->status_code != RTSP_STATUS_OK) {
            return -1;
        }
1857
    }
1858 1859
    rt->state = RTSP_STATE_PAUSED;
    return 0;
1860 1861
}

1862
static int rtsp_read_seek(AVFormatContext *s, int stream_index,
1863
                          int64_t timestamp, int flags)
1864 1865
{
    RTSPState *rt = s->priv_data;
1866

1867 1868 1869
    rt->seek_timestamp = av_rescale_q(timestamp,
                                      s->streams[stream_index]->time_base,
                                      AV_TIME_BASE_Q);
1870 1871 1872 1873
    switch(rt->state) {
    default:
    case RTSP_STATE_IDLE:
        break;
1874
    case RTSP_STATE_STREAMING:
1875 1876 1877
        if (rtsp_read_pause(s) != 0)
            return -1;
        rt->state = RTSP_STATE_SEEKING;
1878 1879 1880 1881 1882 1883 1884 1885 1886 1887
        if (rtsp_read_play(s) != 0)
            return -1;
        break;
    case RTSP_STATE_PAUSED:
        rt->state = RTSP_STATE_IDLE;
        break;
    }
    return 0;
}

1888 1889 1890 1891 1892
static int rtsp_read_close(AVFormatContext *s)
{
    RTSPState *rt = s->priv_data;
    char cmd[1024];

F
Fabrice Bellard 已提交
1893
#if 0
1894
    /* NOTE: it is valid to flush the buffer here */
1895
    if (rt->lower_transport == RTSP_LOWER_TRANSPORT_TCP) {
1896 1897
        url_fclose(&rt->rtsp_gb);
    }
F
Fabrice Bellard 已提交
1898
#endif
1899
    snprintf(cmd, sizeof(cmd),
F
Fabrice Bellard 已提交
1900
             "TEARDOWN %s RTSP/1.0\r\n",
1901
             s->filename);
1902
    rtsp_send_cmd_async(s, cmd);
1903

1904
    rtsp_close_streams(s);
1905 1906 1907 1908
    url_close(rt->rtsp_hd);
    return 0;
}

1909
AVInputFormat rtsp_demuxer = {
1910
    "rtsp",
1911
    NULL_IF_CONFIG_SMALL("RTSP input format"),
1912 1913 1914 1915 1916
    sizeof(RTSPState),
    rtsp_probe,
    rtsp_read_header,
    rtsp_read_packet,
    rtsp_read_close,
1917
    rtsp_read_seek,
1918
    .flags = AVFMT_NOFILE,
1919 1920
    .read_play = rtsp_read_play,
    .read_pause = rtsp_read_pause,
1921
};
1922
#endif
1923

1924
static int sdp_probe(AVProbeData *p1)
1925
{
1926
    const char *p = p1->buf, *p_end = p1->buf + p1->buf_size;
1927 1928

    /* we look for a line beginning "c=IN IP4" */
1929
    while (p < p_end && *p != '\0') {
1930 1931
        if (p + sizeof("c=IN IP4") - 1 < p_end &&
            av_strstart(p, "c=IN IP4", NULL))
1932
            return AVPROBE_SCORE_MAX / 2;
1933

1934
        while (p < p_end - 1 && *p != '\n') p++;
1935
        if (++p >= p_end)
1936 1937 1938 1939
            break;
        if (*p == '\r')
            p++;
    }
1940 1941 1942 1943 1944
    return 0;
}

#define SDP_MAX_SIZE 8192

1945
static int sdp_read_header(AVFormatContext *s, AVFormatParameters *ap)
1946
{
1947
    RTSPState *rt = s->priv_data;
1948 1949 1950 1951 1952 1953 1954 1955
    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);
1956
    size = get_buffer(s->pb, content, SDP_MAX_SIZE - 1);
1957 1958 1959 1960 1961 1962 1963 1964 1965 1966
    if (size <= 0) {
        av_free(content);
        return AVERROR_INVALIDDATA;
    }
    content[size] ='\0';

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

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

1970
        snprintf(url, sizeof(url), "rtp://%s:%d?localport=%d&ttl=%d",
1971
                 inet_ntoa(rtsp_st->sdp_ip),
1972
                 rtsp_st->sdp_port,
1973
                 rtsp_st->sdp_port,
1974
                 rtsp_st->sdp_ttl);
1975
        if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
1976 1977 1978
            err = AVERROR_INVALIDDATA;
            goto fail;
        }
1979
        if ((err = rtsp_open_transport_ctx(s, rtsp_st)))
1980
            goto fail;
1981 1982
    }
    return 0;
1983
fail:
1984
    rtsp_close_streams(s);
1985 1986 1987 1988 1989
    return err;
}

static int sdp_read_close(AVFormatContext *s)
{
1990
    rtsp_close_streams(s);
1991 1992 1993
    return 0;
}

1994
AVInputFormat sdp_demuxer = {
1995
    "sdp",
1996
    NULL_IF_CONFIG_SMALL("SDP"),
1997 1998 1999
    sizeof(RTSPState),
    sdp_probe,
    sdp_read_header,
L
Luca Barbato 已提交
2000
    rtsp_fetch_packet,
2001 2002
    sdp_read_close,
};