rtsp.c 74.9 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
#include "libavutil/base64.h"
23
#include "libavutil/avstring.h"
24
#include "libavutil/intreadwrite.h"
J
Josh Allmann 已提交
25
#include "libavutil/random_seed.h"
26 27
#include "avformat.h"

28
#include <sys/time.h>
29
#if HAVE_SYS_SELECT_H
30
#include <sys/select.h>
31
#endif
32
#include <strings.h>
33
#include "internal.h"
34
#include "network.h"
35
#include "os_support.h"
J
Josh Allmann 已提交
36
#include "http.h"
37
#include "rtsp.h"
38

39
#include "rtpdec.h"
40
#include "rdt.h"
41
#include "rtpdec_formats.h"
42
#include "rtpenc_chain.h"
R
Ryan Martell 已提交
43

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

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

R
Reimar Döffinger 已提交
51
/* Timeout values for socket select, in ms,
52 53 54 55
 * and read_packet(), in seconds  */
#define SELECT_TIMEOUT_MS 100
#define READ_PACKET_TIMEOUT_S 10
#define MAX_TIMEOUTS READ_PACKET_TIMEOUT_S * 1000 / SELECT_TIMEOUT_MS
56
#define SDP_MAX_SIZE 16384
57
#define RECVBUF_SIZE 10 * RTP_MAX_PACKET_LENGTH
58

59 60
static void get_word_until_chars(char *buf, int buf_size,
                                 const char *sep, const char **pp)
61 62 63 64 65
{
    const char *p;
    char *q;

    p = *pp;
66
    p += strspn(p, SPACE_CHARS);
67 68 69 70 71 72 73 74 75 76 77
    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;
}

78 79
static void get_word_sep(char *buf, int buf_size, const char *sep,
                         const char **pp)
80
{
81 82 83
    if (**pp == '/') (*pp)++;
    get_word_until_chars(buf, buf_size, sep, pp);
}
84

85 86 87
static void get_word(char *buf, int buf_size, const char **pp)
{
    get_word_until_chars(buf, buf_size, SPACE_CHARS, pp);
88 89
}

M
Martin Storsjö 已提交
90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127
/** Parse a string p in the form of Range:npt=xx-xx, and determine the start
 *  and end time.
 *  Used for seeking in the rtp stream.
 */
static void rtsp_parse_range_npt(const char *p, int64_t *start, int64_t *end)
{
    char buf[256];

    p += strspn(p, SPACE_CHARS);
    if (!av_stristart(p, "npt=", &p))
        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);
}

static int get_sockaddr(const char *buf, struct sockaddr_storage *sock)
{
    struct addrinfo hints, *ai = NULL;
    memset(&hints, 0, sizeof(hints));
    hints.ai_flags = AI_NUMERICHOST;
    if (getaddrinfo(buf, NULL, &hints, &ai))
        return -1;
    memcpy(sock, ai->ai_addr, FFMIN(sizeof(*sock), ai->ai_addrlen));
    freeaddrinfo(ai);
    return 0;
}

128
/* parse the rtpmap description: <codec_name>/<clock_rate>[/<other params>] */
129 130
static int sdp_parse_rtpmap(AVFormatContext *s,
                            AVCodecContext *codec, RTSPStream *rtsp_st,
131
                            int payload_type, const char *p)
132 133
{
    char buf[256];
R
Romain Degez 已提交
134 135
    int i;
    AVCodec *c;
136
    const char *c_name;
137

R
Romain Degez 已提交
138
    /* Loop into AVRtpDynamicPayloadTypes[] and AVRtpPayloadTypes[] and
139 140 141 142 143
     * 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 已提交
144
    if (payload_type >= RTP_PT_PRIVATE) {
145 146 147 148 149 150 151 152 153
        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 已提交
154 155
                break;
            }
R
Ryan Martell 已提交
156
        }
157 158 159 160 161 162
        /* If no dynamic handler was found, check with the list of standard
         * allocated types, if such a stream for some reason happens to
         * use a private payload type. This isn't handled in rtpdec.c, since
         * the format name from the rtpmap line never is passed into rtpdec. */
        if (!rtsp_st->dynamic_handler)
            codec->codec_id = ff_rtp_codec_id(buf, codec->codec_type);
163
    } else {
164 165
        /* We are in a standard case
         * (from http://www.iana.org/assignments/rtp-parameters). */
R
Romain Degez 已提交
166
        /* search into AVRtpPayloadTypes[] */
167
        codec->codec_id = ff_rtp_codec_id(buf, codec->codec_type);
R
Romain Degez 已提交
168 169 170 171
    }

    c = avcodec_find_decoder(codec->codec_id);
    if (c && c->name)
172
        c_name = c->name;
R
Romain Degez 已提交
173
    else
174
        c_name = "(null)";
R
Romain Degez 已提交
175

R
Ronald S. Bultje 已提交
176 177 178
    get_word_sep(buf, sizeof(buf), "/", &p);
    i = atoi(buf);
    switch (codec->codec_type) {
179
    case AVMEDIA_TYPE_AUDIO:
R
Ronald S. Bultje 已提交
180 181 182 183 184 185 186 187 188 189 190 191 192
        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 已提交
193
        }
R
Ronald S. Bultje 已提交
194 195 196 197 198
        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;
199
    case AVMEDIA_TYPE_VIDEO:
R
Ronald S. Bultje 已提交
200 201 202 203 204 205
        av_log(s, AV_LOG_DEBUG, "video codec set to: %s\n", c_name);
        break;
    default:
        break;
    }
    return 0;
206 207
}

R
Reimar Döffinger 已提交
208
/* parse the attribute line from the fmtp a line of an sdp response. This
209 210
 * is broken out as a function because it is used in rtp_h264.c, which is
 * forthcoming. */
211
int ff_rtsp_next_attr_and_value(const char **p, char *attr, int attr_size,
M
Martin Storsjö 已提交
212
                                char *value, int value_size)
213
{
214
    *p += strspn(*p, SPACE_CHARS);
215
    if (**p) {
216 217 218 219 220 221 222 223 224 225 226
        get_word_sep(attr, attr_size, "=", p);
        if (**p == '=')
            (*p)++;
        get_word_sep(value, value_size, ";", p);
        if (**p == ';')
            (*p)++;
        return 1;
    }
    return 0;
}

227 228
typedef struct SDPParseState {
    /* SDP only */
M
Martin Storsjö 已提交
229
    struct sockaddr_storage default_ip;
230 231
    int            default_ttl;
    int            skip_media;  ///< set if an unknown m= line occurs
232 233 234
} SDPParseState;

static void sdp_parse_line(AVFormatContext *s, SDPParseState *s1,
235 236
                           int letter, const char *buf)
{
237
    RTSPState *rt = s->priv_data;
238 239
    char buf1[64], st_type[64];
    const char *p;
240
    enum AVMediaType codec_type;
D
Diego Pettenò 已提交
241
    int payload_type, i;
242 243
    AVStream *st;
    RTSPStream *rtsp_st;
M
Martin Storsjö 已提交
244
    struct sockaddr_storage sdp_ip;
245 246
    int ttl;

247
    dprintf(s, "sdp: %c='%s'\n", letter, buf);
248 249

    p = buf;
250 251
    if (s1->skip_media && letter != 'm')
        return;
252
    switch (letter) {
253 254 255 256 257
    case 'c':
        get_word(buf1, sizeof(buf1), &p);
        if (strcmp(buf1, "IN") != 0)
            return;
        get_word(buf1, sizeof(buf1), &p);
M
Martin Storsjö 已提交
258
        if (strcmp(buf1, "IP4") && strcmp(buf1, "IP6"))
259 260
            return;
        get_word_sep(buf1, sizeof(buf1), "/", &p);
M
Martin Storsjö 已提交
261
        if (get_sockaddr(buf1, &sdp_ip))
262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278
            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;
279
    case 's':
280
        av_metadata_set2(&s->metadata, "title", p, 0);
281 282 283
        break;
    case 'i':
        if (s->nb_streams == 0) {
284
            av_metadata_set2(&s->metadata, "comment", p, 0);
285 286 287 288 289
            break;
        }
        break;
    case 'm':
        /* new stream */
290
        s1->skip_media = 0;
291 292
        get_word(st_type, sizeof(st_type), &p);
        if (!strcmp(st_type, "audio")) {
293
            codec_type = AVMEDIA_TYPE_AUDIO;
294
        } else if (!strcmp(st_type, "video")) {
295
            codec_type = AVMEDIA_TYPE_VIDEO;
296
        } else if (!strcmp(st_type, "application")) {
297
            codec_type = AVMEDIA_TYPE_DATA;
298
        } else {
299
            s1->skip_media = 1;
300 301 302 303 304
            return;
        }
        rtsp_st = av_mallocz(sizeof(RTSPStream));
        if (!rtsp_st)
            return;
305 306
        rtsp_st->stream_index = -1;
        dynarray_add(&rt->rtsp_streams, &rt->nb_rtsp_streams, rtsp_st);
307 308 309 310 311 312 313 314

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

316 317 318 319
        /* XXX: handle list of formats */
        get_word(buf1, sizeof(buf1), &p); /* format list */
        rtsp_st->sdp_payload_type = atoi(buf1);

320
        if (!strcmp(ff_rtp_enc_name(rtsp_st->sdp_payload_type), "MP2T")) {
321 322 323 324 325 326 327
            /* no corresponding stream */
        } else {
            st = av_new_stream(s, 0);
            if (!st)
                return;
            st->priv_data = rtsp_st;
            rtsp_st->stream_index = st->index;
328
            st->codec->codec_type = codec_type;
R
Romain Degez 已提交
329
            if (rtsp_st->sdp_payload_type < RTP_PT_PRIVATE) {
330
                /* if standard payload type, we can find the codec right now */
331
                ff_rtp_get_codec_info(st->codec, rtsp_st->sdp_payload_type);
332 333
            }
        }
334
        /* put a default control url */
335
        av_strlcpy(rtsp_st->control_url, rt->control_uri,
336
                   sizeof(rtsp_st->control_url));
337 338
        break;
    case 'a':
339 340 341 342 343 344
        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 {
345 346 347 348
            char proto[32];
            /* get the control url */
            st = s->streams[s->nb_streams - 1];
            rtsp_st = st->priv_data;
349

350
            /* XXX: may need to add full url resolution */
M
Måns Rullgård 已提交
351
            av_url_split(proto, sizeof(proto), NULL, 0, NULL, 0,
M
Martin Storsjö 已提交
352
                         NULL, NULL, 0, p);
353 354
            if (proto[0] == '\0') {
                /* relative control URL */
355
                if (rtsp_st->control_url[strlen(rtsp_st->control_url)-1]!='/')
356 357 358 359 360 361 362
                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));
363
            }
364
        } else if (av_strstart(p, "rtpmap:", &p) && s->nb_streams > 0) {
365
            /* NOTE: rtpmap is only supported AFTER the 'm=' tag */
366
            get_word(buf1, sizeof(buf1), &p);
367
            payload_type = atoi(buf1);
368
            st = s->streams[s->nb_streams - 1];
R
Ronald S. Bultje 已提交
369
            rtsp_st = st->priv_data;
370
            sdp_parse_rtpmap(s, st->codec, rtsp_st, payload_type, p);
371 372
        } else if (av_strstart(p, "fmtp:", &p) ||
                   av_strstart(p, "framesize:", &p)) {
373
            /* NOTE: fmtp is only supported AFTER the 'a=rtpmap:xxx' tag */
374 375 376
            // let dynamic protocol handlers have a stab at the line.
            get_word(buf1, sizeof(buf1), &p);
            payload_type = atoi(buf1);
377 378
            for (i = 0; i < s->nb_streams; i++) {
                st      = s->streams[i];
379
                rtsp_st = st->priv_data;
380 381 382 383 384
                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);
385
            }
386
        } else if (av_strstart(p, "range:", &p)) {
387 388 389 390
            int64_t start, end;

            // this is so that seeking on a streamed file can work.
            rtsp_parse_range_npt(p, &start, &end);
391 392 393 394
            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;
395 396 397
        } else if (av_strstart(p, "IsRealDataType:integer;",&p)) {
            if (atoi(p) == 1)
                rt->transport = RTSP_TRANSPORT_RDT;
398 399 400 401
        } 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 已提交
402 403 404 405 406 407
                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)
408 409
                    rtsp_st->dynamic_handler->parse_sdp_a_line(s,
                        s->nb_streams - 1,
R
Ronald S. Bultje 已提交
410
                        rtsp_st->dynamic_protocol_context, buf);
411
            }
412 413 414 415 416
        }
        break;
    }
}

417
static int sdp_parse(AVFormatContext *s, const char *content)
418 419 420
{
    const char *p;
    int letter;
421 422 423 424
    /* 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
425 426
     * buffer is large.
     *
427 428
     * The Vorbis FMTP line can be up to 16KB - see xiph_parse_sdp_line
     * in rtpdec_xiph.c. */
429
    char buf[16384], *q;
430
    SDPParseState sdp_parse_state, *s1 = &sdp_parse_state;
431

432
    memset(s1, 0, sizeof(SDPParseState));
433
    p = content;
434
    for (;;) {
435
        p += strspn(p, SPACE_CHARS);
436 437 438 439 440 441 442 443 444
        letter = *p;
        if (letter == '\0')
            break;
        p++;
        if (*p != '=')
            goto next_line;
        p++;
        /* get the content */
        q = buf;
F
Fabrice Bellard 已提交
445
        while (*p != '\n' && *p != '\r' && *p != '\0') {
446 447 448 449 450
            if ((q - buf) < sizeof(buf) - 1)
                *q++ = *p;
            p++;
        }
        *q = '\0';
451
        sdp_parse_line(s, s1, letter, buf);
452 453 454 455 456 457 458 459 460
    next_line:
        while (*p != '\n' && *p != '\0')
            p++;
        if (*p == '\n')
            p++;
    }
    return 0;
}

461
/* close and free RTSP streams */
462
void ff_rtsp_close_streams(AVFormatContext *s)
463
{
464
    RTSPState *rt = s->priv_data;
465 466 467
    int i;
    RTSPStream *rtsp_st;

468
    for (i = 0; i < rt->nb_rtsp_streams; i++) {
469 470 471
        rtsp_st = rt->rtsp_streams[i];
        if (rtsp_st) {
            if (rtsp_st->transport_priv) {
472 473 474
                if (s->oformat) {
                    AVFormatContext *rtpctx = rtsp_st->transport_priv;
                    av_write_trailer(rtpctx);
475 476 477 478 479
                    if (rt->lower_transport == RTSP_LOWER_TRANSPORT_TCP) {
                        uint8_t *ptr;
                        url_close_dyn_buf(rtpctx->pb, &ptr);
                        av_free(ptr);
                    } else {
M
Martin Storsjö 已提交
480
                        url_fclose(rtpctx->pb);
481
                    }
482 483
                    av_metadata_free(&rtpctx->streams[0]->metadata);
                    av_metadata_free(&rtpctx->metadata);
484 485 486
                    av_free(rtpctx->streams[0]);
                    av_free(rtpctx);
                } else if (rt->transport == RTSP_TRANSPORT_RDT)
487 488 489 490 491 492 493
                    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)
494 495
                rtsp_st->dynamic_handler->close(
                    rtsp_st->dynamic_protocol_context);
496 497 498 499 500 501 502
        }
    }
    av_free(rt->rtsp_streams);
    if (rt->asf_ctx) {
        av_close_input_stream (rt->asf_ctx);
        rt->asf_ctx = NULL;
    }
503
    av_free(rt->recvbuf);
504 505
}

506
static int rtsp_open_transport_ctx(AVFormatContext *s, RTSPStream *rtsp_st)
507 508 509 510 511 512 513 514 515 516
{
    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;

517
    if (s->oformat) {
518 519 520
        rtsp_st->transport_priv = ff_rtp_chain_mux_open(s, st,
                                      rtsp_st->rtp_handle,
                                      RTSP_TCP_MAX_PACKET_SIZE);
R
Reimar Döffinger 已提交
521
        /* Ownership of rtp_handle is passed to the rtp mux context */
522 523
        rtsp_st->rtp_handle = NULL;
    } else if (rt->transport == RTSP_TRANSPORT_RDT)
524 525 526 527 528
        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,
529 530 531
                                         rtsp_st->sdp_payload_type,
            (rt->lower_transport == RTSP_LOWER_TRANSPORT_TCP || !s->max_delay)
            ? 0 : RTP_REORDER_QUEUE_DEFAULT_SIZE);
532 533 534 535

    if (!rtsp_st->transport_priv) {
         return AVERROR(ENOMEM);
    } else if (rt->transport != RTSP_TRANSPORT_RDT) {
536
        if (rtsp_st->dynamic_handler) {
537 538 539 540 541 542 543 544 545
            rtp_parse_set_dynamic_protocol(rtsp_st->transport_priv,
                                           rtsp_st->dynamic_protocol_context,
                                           rtsp_st->dynamic_handler);
        }
    }

    return 0;
}

M
Martin Storsjö 已提交
546
#if CONFIG_RTSP_DEMUXER || CONFIG_RTSP_MUXER
547 548 549 550 551 552 553
static int rtsp_probe(AVProbeData *p)
{
    if (av_strstart(p->filename, "rtsp:", NULL))
        return AVPROBE_SCORE_MAX;
    return 0;
}

554 555 556 557 558 559
static void rtsp_parse_range(int *min_ptr, int *max_ptr, const char **pp)
{
    const char *p;
    int v;

    p = *pp;
560
    p += strspn(p, SPACE_CHARS);
561 562 563 564 565 566 567 568 569 570 571 572 573 574
    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 */
575
static void rtsp_parse_transport(RTSPMessageHeader *reply, const char *p)
576 577 578 579 580 581 582
{
    char transport_protocol[16];
    char profile[16];
    char lower_transport[16];
    char parameter[16];
    RTSPTransportField *th;
    char buf[256];
583

584
    reply->nb_transports = 0;
585

586
    for (;;) {
587
        p += strspn(p, SPACE_CHARS);
588 589 590 591 592
        if (*p == '\0')
            break;

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

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

616 617 618 619 620 621 622 623 624 625 626 627 628
        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++;
629
                    rtsp_parse_range(&th->client_port_min,
630 631 632 633 634
                                     &th->client_port_max, &p);
                }
            } else if (!strcmp(parameter, "server_port")) {
                if (*p == '=') {
                    p++;
635
                    rtsp_parse_range(&th->server_port_min,
636 637 638 639 640
                                     &th->server_port_max, &p);
                }
            } else if (!strcmp(parameter, "interleaved")) {
                if (*p == '=') {
                    p++;
641
                    rtsp_parse_range(&th->interleaved_min,
642 643 644
                                     &th->interleaved_max, &p);
                }
            } else if (!strcmp(parameter, "multicast")) {
645 646
                if (th->lower_transport == RTSP_LOWER_TRANSPORT_UDP)
                    th->lower_transport = RTSP_LOWER_TRANSPORT_UDP_MULTICAST;
647 648 649 650 651 652 653 654 655
            } else if (!strcmp(parameter, "ttl")) {
                if (*p == '=') {
                    p++;
                    th->ttl = strtol(p, (char **)&p, 10);
                }
            } else if (!strcmp(parameter, "destination")) {
                if (*p == '=') {
                    p++;
                    get_word_sep(buf, sizeof(buf), ";,", &p);
M
Martin Storsjö 已提交
656
                    get_sockaddr(buf, &th->destination);
657
                }
658 659 660 661 662 663
            } else if (!strcmp(parameter, "source")) {
                if (*p == '=') {
                    p++;
                    get_word_sep(buf, sizeof(buf), ";,", &p);
                    av_strlcpy(th->source, buf, sizeof(th->source));
                }
664
            }
665

666 667 668 669 670 671 672 673 674 675 676 677
            while (*p != ';' && *p != '\0' && *p != ',')
                p++;
            if (*p == ';')
                p++;
        }
        if (*p == ',')
            p++;

        reply->nb_transports++;
    }
}

678 679
void ff_rtsp_parse_line(RTSPMessageHeader *reply, const char *buf,
                        HTTPAuthState *auth_state)
680 681 682 683 684
{
    const char *p;

    /* NOTE: we do case independent match for broken servers */
    p = buf;
685
    if (av_stristart(p, "Session:", &p)) {
686
        int t;
687
        get_word_sep(reply->session_id, sizeof(reply->session_id), ";", &p);
688 689 690 691
        if (av_stristart(p, ";timeout=", &p) &&
            (t = strtol(p, NULL, 10)) > 0) {
            reply->timeout = t;
        }
692
    } else if (av_stristart(p, "Content-Length:", &p)) {
693
        reply->content_length = strtol(p, NULL, 10);
694
    } else if (av_stristart(p, "Transport:", &p)) {
695
        rtsp_parse_transport(reply, p);
696
    } else if (av_stristart(p, "CSeq:", &p)) {
697
        reply->seq = strtol(p, NULL, 10);
698
    } else if (av_stristart(p, "Range:", &p)) {
699
        rtsp_parse_range_npt(p, &reply->range_start, &reply->range_end);
700
    } else if (av_stristart(p, "RealChallenge1:", &p)) {
701
        p += strspn(p, SPACE_CHARS);
702
        av_strlcpy(reply->real_challenge, p, sizeof(reply->real_challenge));
703
    } else if (av_stristart(p, "Server:", &p)) {
704
        p += strspn(p, SPACE_CHARS);
705
        av_strlcpy(reply->server, p, sizeof(reply->server));
706 707 708
    } else if (av_stristart(p, "Notice:", &p) ||
               av_stristart(p, "X-Notice:", &p)) {
        reply->notice = strtol(p, NULL, 10);
L
Luca Barbato 已提交
709
    } else if (av_stristart(p, "Location:", &p)) {
710
        p += strspn(p, SPACE_CHARS);
L
Luca Barbato 已提交
711
        av_strlcpy(reply->location, p , sizeof(reply->location));
712
    } else if (av_stristart(p, "WWW-Authenticate:", &p) && auth_state) {
713
        p += strspn(p, SPACE_CHARS);
714 715
        ff_http_auth_handle_header(auth_state, "WWW-Authenticate", p);
    } else if (av_stristart(p, "Authentication-Info:", &p) && auth_state) {
716
        p += strspn(p, SPACE_CHARS);
717
        ff_http_auth_handle_header(auth_state, "Authentication-Info", p);
718 719 720
    }
}

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

728
    ret = url_read_complete(rt->rtsp_hd, buf, 3);
729 730
    if (ret != 3)
        return;
731
    len = AV_RB16(buf + 1);
732 733 734

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

735 736 737 738 739
    /* skip payload */
    while (len > 0) {
        len1 = len;
        if (len1 > sizeof(buf))
            len1 = sizeof(buf);
740
        ret = url_read_complete(rt->rtsp_hd, buf, len1);
741 742 743 744 745
        if (ret != len1)
            return;
        len -= len1;
    }
}
746

747
int ff_rtsp_read_reply(AVFormatContext *s, RTSPMessageHeader *reply,
M
Martin Storsjö 已提交
748 749
                       unsigned char **content_ptr,
                       int return_on_interleaved_data)
750 751 752 753 754
{
    RTSPState *rt = s->priv_data;
    char buf[4096], buf1[1024], *q;
    unsigned char ch;
    const char *p;
755
    int ret, content_length, line_count = 0;
756 757
    unsigned char *content = NULL;

758
    memset(reply, 0, sizeof(*reply));
759 760 761

    /* parse reply (XXX: use buffers) */
    rt->last_reply[0] = '\0';
762
    for (;;) {
763
        q = buf;
764
        for (;;) {
765
            ret = url_read_complete(rt->rtsp_hd, &ch, 1);
766
#ifdef DEBUG_RTP_TCP
767
            dprintf(s, "ret=%d c=%02x [%c]\n", ret, ch, ch);
768 769
#endif
            if (ret != 1)
770
                return AVERROR_EOF;
771 772
            if (ch == '\n')
                break;
773 774
            if (ch == '$') {
                /* XXX: only parse it if first char on line ? */
775 776 777
                if (return_on_interleaved_data) {
                    return 1;
                } else
778
                    ff_rtsp_skip_packet(s);
779
            } else if (ch != '\r') {
780 781 782 783 784
                if ((q - buf) < sizeof(buf) - 1)
                    *q++ = ch;
            }
        }
        *q = '\0';
785 786 787

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

788 789 790 791 792 793 794 795 796
        /* 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);
L
Luca Barbato 已提交
797
            av_strlcpy(reply->reason, p, sizeof(reply->reason));
798
        } else {
799
            ff_rtsp_parse_line(reply, p, &rt->auth_state);
M
Måns Rullgård 已提交
800 801
            av_strlcat(rt->last_reply, p,    sizeof(rt->last_reply));
            av_strlcat(rt->last_reply, "\n", sizeof(rt->last_reply));
802 803 804
        }
        line_count++;
    }
805

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

809 810 811 812
    content_length = reply->content_length;
    if (content_length > 0) {
        /* leave some room for a trailing '\0' (useful for simple parsing) */
        content = av_malloc(content_length + 1);
813
        (void)url_read_complete(rt->rtsp_hd, content, content_length);
814 815 816 817
        content[content_length] = '\0';
    }
    if (content_ptr)
        *content_ptr = content;
818 819
    else
        av_free(content);
820

821 822 823 824 825
    if (rt->seq != reply->seq) {
        av_log(s, AV_LOG_WARNING, "CSeq %d expected, %d received.\n",
            rt->seq, reply->seq);
    }

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

837
    return 0;
838 839
}

840
int ff_rtsp_send_cmd_with_content_async(AVFormatContext *s,
M
Martin Storsjö 已提交
841 842 843 844
                                        const char *method, const char *url,
                                        const char *headers,
                                        const unsigned char *send_content,
                                        int send_content_length)
845 846
{
    RTSPState *rt = s->priv_data;
J
Josh Allmann 已提交
847 848
    char buf[4096], *out_buf;
    char base64buf[AV_BASE64_SIZE(sizeof(buf))];
849

J
Josh Allmann 已提交
850 851
    /* Add in RTSP headers */
    out_buf = buf;
852
    rt->seq++;
853 854 855
    snprintf(buf, sizeof(buf), "%s %s RTSP/1.0\r\n", method, url);
    if (headers)
        av_strlcat(buf, headers, sizeof(buf));
856
    av_strlcatf(buf, sizeof(buf), "CSeq: %d\r\n", rt->seq);
857 858
    if (rt->session_id[0] != '\0' && (!headers ||
        !strstr(headers, "\nIf-Match:"))) {
859
        av_strlcatf(buf, sizeof(buf), "Session: %s\r\n", rt->session_id);
860
    }
861 862 863 864 865 866 867
    if (rt->auth[0]) {
        char *str = ff_http_auth_create_response(&rt->auth_state,
                                                 rt->auth, url, method);
        if (str)
            av_strlcat(buf, str, sizeof(buf));
        av_free(str);
    }
868 869
    if (send_content_length > 0 && send_content)
        av_strlcatf(buf, sizeof(buf), "Content-Length: %d\r\n", send_content_length);
870
    av_strlcat(buf, "\r\n", sizeof(buf));
871

J
Josh Allmann 已提交
872 873 874 875 876 877
    /* base64 encode rtsp if tunneling */
    if (rt->control_transport == RTSP_MODE_TUNNEL) {
        av_base64_encode(base64buf, sizeof(base64buf), buf, strlen(buf));
        out_buf = base64buf;
    }

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

J
Josh Allmann 已提交
880 881 882 883 884 885 886
    url_write(rt->rtsp_hd_out, out_buf, strlen(out_buf));
    if (send_content_length > 0 && send_content) {
        if (rt->control_transport == RTSP_MODE_TUNNEL) {
            av_log(s, AV_LOG_ERROR, "tunneling of RTSP requests "
                                    "with content data not supported\n");
            return AVERROR_PATCHWELCOME;
        }
887
        url_write(rt->rtsp_hd_out, send_content, send_content_length);
J
Josh Allmann 已提交
888
    }
889
    rt->last_cmd_time = av_gettime();
890 891

    return 0;
892 893
}

894
int ff_rtsp_send_cmd_async(AVFormatContext *s, const char *method,
M
Martin Storsjö 已提交
895
                           const char *url, const char *headers)
896
{
897
    return ff_rtsp_send_cmd_with_content_async(s, method, url, headers, NULL, 0);
898 899
}

900
int ff_rtsp_send_cmd(AVFormatContext *s, const char *method, const char *url,
M
Martin Storsjö 已提交
901 902
                     const char *headers, RTSPMessageHeader *reply,
                     unsigned char **content_ptr)
903
{
904
    return ff_rtsp_send_cmd_with_content(s, method, url, headers, reply,
M
Martin Storsjö 已提交
905
                                         content_ptr, NULL, 0);
906 907
}

908
int ff_rtsp_send_cmd_with_content(AVFormatContext *s,
M
Martin Storsjö 已提交
909 910 911 912 913 914
                                  const char *method, const char *url,
                                  const char *header,
                                  RTSPMessageHeader *reply,
                                  unsigned char **content_ptr,
                                  const unsigned char *send_content,
                                  int send_content_length)
915
{
916 917
    RTSPState *rt = s->priv_data;
    HTTPAuthType cur_auth_type;
918
    int ret;
919 920 921

retry:
    cur_auth_type = rt->auth_state.auth_type;
922
    if ((ret = ff_rtsp_send_cmd_with_content_async(s, method, url, header,
M
Martin Storsjö 已提交
923 924
                                                   send_content,
                                                   send_content_length)))
925
        return ret;
926

927 928
    if ((ret = ff_rtsp_read_reply(s, reply, content_ptr, 0) ) < 0)
        return ret;
929 930 931 932

    if (reply->status_code == 401 && cur_auth_type == HTTP_AUTH_NONE &&
        rt->auth_state.auth_type != HTTP_AUTH_NONE)
        goto retry;
933

934
    if (reply->status_code > 400){
L
Luca Barbato 已提交
935
        av_log(s, AV_LOG_ERROR, "method %s failed: %d%s\n",
936
               method,
L
Luca Barbato 已提交
937 938
               reply->status_code,
               reply->reason);
939 940 941
        av_log(s, AV_LOG_DEBUG, "%s\n", rt->last_reply);
    }

942
    return 0;
943 944
}

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

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

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

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

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

973 974 975 976 977 978 979 980 981 982 983 984
        /**
         * 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 &&
985 986
                        !strcmp(rt->rtsp_streams[rtx]->control_url + len - 4,
                                "/rtx"))
987 988 989 990 991 992 993 994
                        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 已提交
995
            rtsp_st = rt->rtsp_streams[i];
996 997

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

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

F
Fabrice Bellard 已提交
1006
            /* first try in specified port range */
R
Romain Degez 已提交
1007
            if (RTSP_RTP_PORT_MIN != 0) {
1008
                while (j <= RTSP_RTP_PORT_MAX) {
1009 1010
                    ff_url_join(buf, sizeof(buf), "rtp", NULL, host, -1,
                                "?localport=%d", j);
1011 1012 1013
                    /* 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 已提交
1014 1015
                        goto rtp_opened;
                }
1016
            }
F
Fabrice Bellard 已提交
1017

1018 1019 1020 1021 1022 1023 1024
#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 已提交
1025 1026

        rtp_opened:
1027
            port = rtp_get_local_port(rtsp_st->rtp_handle);
1028
        have_port:
1029
            snprintf(transport, sizeof(transport) - 1,
1030 1031 1032 1033 1034
                     "%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);
1035 1036
            if (rt->transport == RTSP_TRANSPORT_RTP &&
                !(rt->server_type == RTSP_SERVER_WMS && i > 0))
1037
                av_strlcatf(transport, sizeof(transport), "-%d", port + 1);
1038 1039 1040
        }

        /* RTP/TCP */
1041
        else if (lower_transport == RTSP_LOWER_TRANSPORT_TCP) {
1042 1043 1044 1045
            /** 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 &&
1046
                s->streams[rtsp_st->stream_index]->codec->codec_type ==
1047
                    AVMEDIA_TYPE_DATA)
1048
                continue;
1049
            snprintf(transport, sizeof(transport) - 1,
1050 1051 1052 1053 1054 1055 1056
                     "%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;
1057 1058
        }

1059
        else if (lower_transport == RTSP_LOWER_TRANSPORT_UDP_MULTICAST) {
1060
            snprintf(transport, sizeof(transport) - 1,
1061
                     "%s/UDP;multicast", trans_pref);
1062
        }
1063 1064 1065
        if (s->oformat) {
            av_strlcat(transport, ";mode=receive", sizeof(transport));
        } else if (rt->server_type == RTSP_SERVER_REAL ||
1066
                   rt->server_type == RTSP_SERVER_WMS)
1067
            av_strlcat(transport, ";mode=play", sizeof(transport));
1068
        snprintf(cmd, sizeof(cmd),
F
Fabrice Bellard 已提交
1069
                 "Transport: %s\r\n",
1070
                 transport);
1071
        if (i == 0 && rt->server_type == RTSP_SERVER_REAL) {
1072 1073 1074 1075 1076 1077 1078 1079
            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);
        }
1080
        ff_rtsp_send_cmd(s, "SETUP", rtsp_st->control_url, cmd, reply, NULL);
1081 1082 1083
        if (reply->status_code == 461 /* Unsupported protocol */ && i == 0) {
            err = 1;
            goto fail;
1084 1085
        } else if (reply->status_code != RTSP_STATUS_OK ||
                   reply->nb_transports != 1) {
1086 1087 1088 1089 1090 1091
            err = AVERROR_INVALIDDATA;
            goto fail;
        }

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

R
Reinhard Tartler 已提交
1102
        /* close RTP connection if not chosen */
1103 1104
        if (reply->transports[0].lower_transport != RTSP_LOWER_TRANSPORT_UDP &&
            (lower_transport == RTSP_LOWER_TRANSPORT_UDP)) {
1105 1106
            url_close(rtsp_st->rtp_handle);
            rtsp_st->rtp_handle = NULL;
1107 1108
        }

1109 1110
        switch(reply->transports[0].lower_transport) {
        case RTSP_LOWER_TRANSPORT_TCP:
1111 1112 1113
            rtsp_st->interleaved_min = reply->transports[0].interleaved_min;
            rtsp_st->interleaved_max = reply->transports[0].interleaved_max;
            break;
1114

1115 1116 1117
        case RTSP_LOWER_TRANSPORT_UDP: {
            char url[1024];

1118 1119 1120 1121 1122 1123
            /* Use source address if specified */
            if (reply->transports[0].source[0]) {
                ff_url_join(url, sizeof(url), "rtp", NULL,
                            reply->transports[0].source,
                            reply->transports[0].server_port_min, NULL);
            } else {
R
Ronald S. Bultje 已提交
1124 1125
                ff_url_join(url, sizeof(url), "rtp", NULL, host,
                            reply->transports[0].server_port_min, NULL);
1126
            }
1127 1128 1129 1130
            if (!(rt->server_type == RTSP_SERVER_WMS && i > 1) &&
                rtp_set_remote_url(rtsp_st->rtp_handle, url) < 0) {
                err = AVERROR_INVALIDDATA;
                goto fail;
1131
            }
1132 1133 1134 1135
            /* Try to initialize the connection state in a
             * potential NAT router by sending dummy packets.
             * RTP/RTCP dummy packets are used for RDT, too.
             */
1136
            if (!(rt->server_type == RTSP_SERVER_WMS && i > 1) && s->iformat)
1137
                rtp_send_punch_packets(rtsp_st->rtp_handle);
1138
            break;
1139 1140
        }
        case RTSP_LOWER_TRANSPORT_UDP_MULTICAST: {
M
Martin Storsjö 已提交
1141 1142
            char url[1024], namebuf[50];
            struct sockaddr_storage addr;
1143 1144
            int port, ttl;

M
Martin Storsjö 已提交
1145 1146
            if (reply->transports[0].destination.ss_family) {
                addr      = reply->transports[0].destination;
1147 1148 1149
                port      = reply->transports[0].port_min;
                ttl       = reply->transports[0].ttl;
            } else {
M
Martin Storsjö 已提交
1150
                addr      = rtsp_st->sdp_ip;
1151 1152 1153
                port      = rtsp_st->sdp_port;
                ttl       = rtsp_st->sdp_ttl;
            }
M
Martin Storsjö 已提交
1154 1155 1156
            getnameinfo((struct sockaddr*) &addr, sizeof(addr),
                        namebuf, sizeof(namebuf), NULL, 0, NI_NUMERICHOST);
            ff_url_join(url, sizeof(url), "rtp", NULL, namebuf,
1157
                        port, "?ttl=%d", ttl);
1158 1159 1160
            if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
                err = AVERROR_INVALIDDATA;
                goto fail;
1161 1162 1163
            }
            break;
        }
1164
        }
R
Romain Degez 已提交
1165

1166
        if ((err = rtsp_open_transport_ctx(s, rtsp_st)))
1167
            goto fail;
1168 1169
    }

1170 1171 1172
    if (reply->timeout > 0)
        rt->timeout = reply->timeout;

1173
    if (rt->server_type == RTSP_SERVER_REAL)
1174 1175
        rt->need_subscription = 1;

1176 1177 1178
    return 0;

fail:
1179
    for (i = 0; i < rt->nb_rtsp_streams; i++) {
1180 1181 1182 1183 1184
        if (rt->rtsp_streams[i]->rtp_handle) {
            url_close(rt->rtsp_streams[i]->rtp_handle);
            rt->rtsp_streams[i]->rtp_handle = NULL;
        }
    }
1185 1186 1187
    return err;
}

1188 1189 1190 1191
static int rtsp_read_play(AVFormatContext *s)
{
    RTSPState *rt = s->priv_data;
    RTSPMessageHeader reply1, *reply = &reply1;
1192
    int i;
1193 1194 1195
    char cmd[1024];

    av_log(s, AV_LOG_DEBUG, "hello state=%d\n", rt->state);
1196
    rt->nb_byes = 0;
1197 1198 1199

    if (!(rt->server_type == RTSP_SERVER_REAL && rt->need_subscription)) {
        if (rt->state == RTSP_STATE_PAUSED) {
1200
            cmd[0] = 0;
1201 1202 1203 1204 1205
        } else {
            snprintf(cmd, sizeof(cmd),
                     "Range: npt=%0.3f-\r\n",
                     (double)rt->seek_timestamp / AV_TIME_BASE);
        }
1206
        ff_rtsp_send_cmd(s, "PLAY", rt->control_uri, cmd, reply, NULL);
1207 1208 1209
        if (reply->status_code != RTSP_STATUS_OK) {
            return -1;
        }
M
Martin Storsjö 已提交
1210
        if (rt->transport == RTSP_TRANSPORT_RTP) {
1211 1212 1213 1214
            for (i = 0; i < rt->nb_rtsp_streams; i++) {
                RTSPStream *rtsp_st = rt->rtsp_streams[i];
                RTPDemuxContext *rtpctx = rtsp_st->transport_priv;
                AVStream *st = NULL;
1215 1216
                if (!rtpctx)
                    continue;
1217 1218
                if (rtsp_st->stream_index >= 0)
                    st = s->streams[rtsp_st->stream_index];
1219
                ff_rtp_reset_packet_queue(rtpctx);
1220
                if (reply->range_start != AV_NOPTS_VALUE) {
M
Martin Storsjö 已提交
1221 1222 1223 1224 1225 1226
                    rtpctx->last_rtcp_ntp_time  = AV_NOPTS_VALUE;
                    rtpctx->first_rtcp_ntp_time = AV_NOPTS_VALUE;
                    if (st)
                        rtpctx->range_start_offset =
                            av_rescale_q(reply->range_start, AV_TIME_BASE_Q,
                                         st->time_base);
1227
                }
1228 1229
            }
        }
1230
    }
1231
    rt->state = RTSP_STATE_STREAMING;
1232 1233 1234
    return 0;
}

1235
static int rtsp_setup_input_streams(AVFormatContext *s, RTSPMessageHeader *reply)
1236 1237 1238 1239 1240 1241 1242 1243
{
    RTSPState *rt = s->priv_data;
    char cmd[1024];
    unsigned char *content = NULL;
    int ret;

    /* describe the stream */
    snprintf(cmd, sizeof(cmd),
1244
             "Accept: application/sdp\r\n");
1245 1246 1247 1248 1249 1250 1251 1252 1253
    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));
    }
1254
    ff_rtsp_send_cmd(s, "DESCRIBE", rt->control_uri, cmd, reply, &content);
1255 1256 1257 1258 1259 1260 1261
    if (!content)
        return AVERROR_INVALIDDATA;
    if (reply->status_code != RTSP_STATUS_OK) {
        av_freep(&content);
        return AVERROR_INVALIDDATA;
    }

M
Martin Storsjö 已提交
1262
    av_log(s, AV_LOG_VERBOSE, "SDP:\n%s\n", content);
1263 1264 1265 1266 1267 1268 1269 1270 1271
    /* 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;
}

1272
static int rtsp_setup_output_streams(AVFormatContext *s, const char *addr)
1273 1274 1275 1276 1277
{
    RTSPState *rt = s->priv_data;
    RTSPMessageHeader reply1, *reply = &reply1;
    int i;
    char *sdp;
1278
    AVFormatContext sdp_ctx, *ctx_array[1];
1279

1280
    s->start_time_realtime = av_gettime();
1281 1282

    /* Announce the stream */
1283
    sdp = av_mallocz(SDP_MAX_SIZE);
1284 1285
    if (sdp == NULL)
        return AVERROR(ENOMEM);
1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301
    /* We create the SDP based on the RTSP AVFormatContext where we
     * aren't allowed to change the filename field. (We create the SDP
     * based on the RTSP context since the contexts for the RTP streams
     * don't exist yet.) In order to specify a custom URL with the actual
     * peer IP instead of the originally specified hostname, we create
     * a temporary copy of the AVFormatContext, where the custom URL is set.
     *
     * FIXME: Create the SDP without copying the AVFormatContext.
     * This either requires setting up the RTP stream AVFormatContexts
     * already here (complicating things immensely) or getting a more
     * flexible SDP creation interface.
     */
    sdp_ctx = *s;
    ff_url_join(sdp_ctx.filename, sizeof(sdp_ctx.filename),
                "rtsp", NULL, addr, -1, NULL);
    ctx_array[0] = &sdp_ctx;
1302
    if (avf_sdp_create(ctx_array, 1, sdp, SDP_MAX_SIZE)) {
1303 1304 1305
        av_free(sdp);
        return AVERROR_INVALIDDATA;
    }
1306
    av_log(s, AV_LOG_VERBOSE, "SDP:\n%s\n", sdp);
1307 1308 1309
    ff_rtsp_send_cmd_with_content(s, "ANNOUNCE", rt->control_uri,
                                  "Content-Type: application/sdp\r\n",
                                  reply, NULL, sdp, strlen(sdp));
1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326
    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;

1327
        av_strlcpy(rtsp_st->control_url, rt->control_uri, sizeof(rtsp_st->control_url));
1328 1329 1330 1331 1332 1333 1334 1335
        /* 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;
}

1336 1337 1338 1339 1340
void ff_rtsp_close_connections(AVFormatContext *s)
{
    RTSPState *rt = s->priv_data;
    if (rt->rtsp_hd_out != rt->rtsp_hd) url_close(rt->rtsp_hd_out);
    url_close(rt->rtsp_hd);
1341
    rt->rtsp_hd = rt->rtsp_hd_out = NULL;
1342 1343
}

1344
int ff_rtsp_connect(AVFormatContext *s)
1345 1346
{
    RTSPState *rt = s->priv_data;
1347 1348
    char host[1024], path[1024], tcpname[1024], cmd[2048], auth[128];
    char *option_list, *option, *filename;
1349
    int port, err, tcp_fd;
1350
    RTSPMessageHeader reply1 = {0}, *reply = &reply1;
1351
    int lower_transport_mask = 0;
1352
    char real_challenge[64];
1353 1354
    struct sockaddr_storage peer;
    socklen_t peer_len = sizeof(peer);
1355 1356 1357

    if (!ff_network_init())
        return AVERROR(EIO);
1358
redirect:
J
Josh Allmann 已提交
1359
    rt->control_transport = RTSP_MODE_PLAIN;
1360
    /* extract hostname and port */
M
Måns Rullgård 已提交
1361
    av_url_split(NULL, 0, auth, sizeof(auth),
M
Martin Storsjö 已提交
1362
                 host, sizeof(host), &port, path, sizeof(path), s->filename);
1363
    if (*auth) {
1364
        av_strlcpy(rt->auth, auth, sizeof(rt->auth));
1365
    }
1366 1367 1368 1369
    if (port < 0)
        port = RTSP_DEFAULT_PORT;

    /* search for options */
1370
    option_list = strrchr(path, '?');
1371
    if (option_list) {
1372 1373 1374
        /* Strip out the RTSP specific options, write out the rest of
         * the options back into the same string. */
        filename = option_list;
1375
        while (option_list) {
1376
            /* move the option pointer */
1377
            option = ++option_list;
1378 1379
            option_list = strchr(option_list, '&');
            if (option_list)
1380 1381
                *option_list = 0;

1382
            /* handle the options */
1383
            if (!strcmp(option, "udp")) {
1384
                lower_transport_mask |= (1<< RTSP_LOWER_TRANSPORT_UDP);
1385
            } else if (!strcmp(option, "multicast")) {
1386
                lower_transport_mask |= (1<< RTSP_LOWER_TRANSPORT_UDP_MULTICAST);
1387
            } else if (!strcmp(option, "tcp")) {
1388
                lower_transport_mask |= (1<< RTSP_LOWER_TRANSPORT_TCP);
J
Josh Allmann 已提交
1389 1390 1391
            } else if(!strcmp(option, "http")) {
                lower_transport_mask |= (1<< RTSP_LOWER_TRANSPORT_TCP);
                rt->control_transport = RTSP_MODE_TUNNEL;
1392
            } else {
1393 1394 1395 1396 1397
                /* Write options back into the buffer, using memmove instead
                 * of strcpy since the strings may overlap. */
                int len = strlen(option);
                memmove(++filename, option, len);
                filename += len;
1398 1399
                if (option_list) *filename = '&';
            }
1400
        }
1401
        *filename = 0;
1402 1403
    }

1404
    if (!lower_transport_mask)
1405
        lower_transport_mask = (1 << RTSP_LOWER_TRANSPORT_NB) - 1;
1406

1407
    if (s->oformat) {
1408 1409 1410
        /* Only UDP or TCP - UDP multicast isn't supported. */
        lower_transport_mask &= (1 << RTSP_LOWER_TRANSPORT_UDP) |
                                (1 << RTSP_LOWER_TRANSPORT_TCP);
J
Josh Allmann 已提交
1411
        if (!lower_transport_mask || rt->control_transport == RTSP_MODE_TUNNEL) {
1412
            av_log(s, AV_LOG_ERROR, "Unsupported lower transport method, "
1413
                                    "only UDP and TCP are supported for output.\n");
1414 1415 1416 1417 1418
            err = AVERROR(EINVAL);
            goto fail;
        }
    }

1419 1420 1421 1422 1423 1424
    /* Construct the URI used in request; this is similar to s->filename,
     * but with authentication credentials removed and RTSP specific options
     * stripped out. */
    ff_url_join(rt->control_uri, sizeof(rt->control_uri), "rtsp", NULL,
                host, port, "%s", path);

J
Josh Allmann 已提交
1425 1426 1427 1428 1429 1430
    if (rt->control_transport == RTSP_MODE_TUNNEL) {
        /* set up initial handshake for tunneling */
        char httpname[1024];
        char sessioncookie[17];
        char headers[1024];

1431
        ff_url_join(httpname, sizeof(httpname), "http", auth, host, port, "%s", path);
J
Josh Allmann 已提交
1432 1433 1434 1435
        snprintf(sessioncookie, sizeof(sessioncookie), "%08x%08x",
                 av_get_random_seed(), av_get_random_seed());

        /* GET requests */
1436
        if (url_alloc(&rt->rtsp_hd, httpname, URL_RDONLY) < 0) {
J
Josh Allmann 已提交
1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447
            err = AVERROR(EIO);
            goto fail;
        }

        /* generate GET headers */
        snprintf(headers, sizeof(headers),
                 "x-sessioncookie: %s\r\n"
                 "Accept: application/x-rtsp-tunnelled\r\n"
                 "Pragma: no-cache\r\n"
                 "Cache-Control: no-cache\r\n",
                 sessioncookie);
1448
        ff_http_set_headers(rt->rtsp_hd, headers);
J
Josh Allmann 已提交
1449 1450

        /* complete the connection */
1451
        if (url_connect(rt->rtsp_hd)) {
J
Josh Allmann 已提交
1452 1453 1454 1455 1456
            err = AVERROR(EIO);
            goto fail;
        }

        /* POST requests */
1457
        if (url_alloc(&rt->rtsp_hd_out, httpname, URL_WRONLY) < 0 ) {
J
Josh Allmann 已提交
1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470
            err = AVERROR(EIO);
            goto fail;
        }

        /* generate POST headers */
        snprintf(headers, sizeof(headers),
                 "x-sessioncookie: %s\r\n"
                 "Content-Type: application/x-rtsp-tunnelled\r\n"
                 "Pragma: no-cache\r\n"
                 "Cache-Control: no-cache\r\n"
                 "Content-Length: 32767\r\n"
                 "Expires: Sun, 9 Jan 1972 00:00:00 GMT\r\n",
                 sessioncookie);
1471 1472
        ff_http_set_headers(rt->rtsp_hd_out, headers);
        ff_http_set_chunked_transfer_encoding(rt->rtsp_hd_out, 0);
J
Josh Allmann 已提交
1473

1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491
        /* Initialize the authentication state for the POST session. The HTTP
         * protocol implementation doesn't properly handle multi-pass
         * authentication for POST requests, since it would require one of
         * the following:
         * - implementing Expect: 100-continue, which many HTTP servers
         *   don't support anyway, even less the RTSP servers that do HTTP
         *   tunneling
         * - sending the whole POST data until getting a 401 reply specifying
         *   what authentication method to use, then resending all that data
         * - waiting for potential 401 replies directly after sending the
         *   POST header (waiting for some unspecified time)
         * Therefore, we copy the full auth state, which works for both basic
         * and digest. (For digest, we would have to synchronize the nonce
         * count variable between the two sessions, if we'd do more requests
         * with the original session, though.)
         */
        ff_http_init_auth_state(rt->rtsp_hd_out, rt->rtsp_hd);

1492 1493 1494 1495 1496
        /* complete the connection */
        if (url_connect(rt->rtsp_hd_out)) {
            err = AVERROR(EIO);
            goto fail;
        }
J
Josh Allmann 已提交
1497
    } else {
1498
        /* open the tcp connection */
J
Josh Allmann 已提交
1499
        ff_url_join(tcpname, sizeof(tcpname), "tcp", NULL, host, port, NULL);
1500
        if (url_open(&rt->rtsp_hd, tcpname, URL_RDWR) < 0) {
J
Josh Allmann 已提交
1501 1502 1503
            err = AVERROR(EIO);
            goto fail;
        }
1504
        rt->rtsp_hd_out = rt->rtsp_hd;
J
Josh Allmann 已提交
1505
    }
1506 1507
    rt->seq = 0;

1508
    tcp_fd = url_get_file_handle(rt->rtsp_hd);
1509 1510 1511 1512 1513
    if (!getpeername(tcp_fd, (struct sockaddr*) &peer, &peer_len)) {
        getnameinfo((struct sockaddr*) &peer, peer_len, host, sizeof(host),
                    NULL, 0, NI_NUMERICHOST);
    }

1514 1515
    /* request options supported by the server; this also detects server
     * type */
1516
    for (rt->server_type = RTSP_SERVER_RTP;;) {
1517
        cmd[0] = 0;
1518
        if (rt->server_type == RTSP_SERVER_REAL)
1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533
            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));
1534
        ff_rtsp_send_cmd(s, "OPTIONS", rt->control_uri, cmd, reply, NULL);
1535 1536 1537 1538 1539 1540
        if (reply->status_code != RTSP_STATUS_OK) {
            err = AVERROR_INVALIDDATA;
            goto fail;
        }

        /* detect server type if not standard-compliant RTP */
1541 1542
        if (rt->server_type != RTSP_SERVER_REAL && reply->real_challenge[0]) {
            rt->server_type = RTSP_SERVER_REAL;
1543
            continue;
1544 1545
        } else if (!strncasecmp(reply->server, "WMServer/", 9)) {
            rt->server_type = RTSP_SERVER_WMS;
1546
        } else if (rt->server_type == RTSP_SERVER_REAL)
1547 1548 1549 1550
            strcpy(real_challenge, reply->real_challenge);
        break;
    }

1551
    if (s->iformat)
1552
        err = rtsp_setup_input_streams(s, reply);
1553
    else
1554
        err = rtsp_setup_output_streams(s, host);
1555
    if (err)
1556 1557
        goto fail;

1558
    do {
1559 1560
        int lower_transport = ff_log2_tab[lower_transport_mask &
                                  ~(lower_transport_mask - 1)];
1561

1562
        err = make_setup_request(s, host, port, lower_transport,
1563
                                 rt->server_type == RTSP_SERVER_REAL ?
1564
                                     real_challenge : NULL);
1565
        if (err < 0)
1566
            goto fail;
1567 1568
        lower_transport_mask &= ~(1 << lower_transport);
        if (lower_transport_mask == 0 && err == 1) {
1569
            err = FF_NETERROR(EPROTONOSUPPORT);
1570 1571 1572
            goto fail;
        }
    } while (err);
1573

1574
    rt->state = RTSP_STATE_IDLE;
1575
    rt->seek_timestamp = 0; /* default is to start stream at position zero */
1576 1577
    return 0;
 fail:
1578
    ff_rtsp_close_streams(s);
1579
    ff_rtsp_close_connections(s);
1580
    if (reply->status_code >=300 && reply->status_code < 400 && s->iformat) {
L
Luca Barbato 已提交
1581 1582 1583 1584 1585 1586
        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;
    }
1587
    ff_network_close();
1588 1589
    return err;
}
1590
#endif /* CONFIG_RTSP_DEMUXER || CONFIG_RTSP_MUXER */
1591

R
Ronald S. Bultje 已提交
1592
static int udp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
1593
                           uint8_t *buf, int buf_size, int64_t wait_end)
R
Ronald S. Bultje 已提交
1594 1595 1596 1597
{
    RTSPState *rt = s->priv_data;
    RTSPStream *rtsp_st;
    fd_set rfds;
1598
    int fd, fd_rtcp, fd_max, n, i, ret, tcp_fd, timeout_cnt = 0;
R
Ronald S. Bultje 已提交
1599 1600
    struct timeval tv;

1601
    for (;;) {
R
Ronald S. Bultje 已提交
1602 1603
        if (url_interrupt_cb())
            return AVERROR(EINTR);
1604 1605
        if (wait_end && wait_end - av_gettime() < 0)
            return AVERROR(EAGAIN);
R
Ronald S. Bultje 已提交
1606 1607 1608 1609 1610 1611 1612 1613
        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;
        }
1614
        for (i = 0; i < rt->nb_rtsp_streams; i++) {
R
Ronald S. Bultje 已提交
1615 1616 1617
            rtsp_st = rt->rtsp_streams[i];
            if (rtsp_st->rtp_handle) {
                fd = url_get_file_handle(rtsp_st->rtp_handle);
1618 1619 1620
                fd_rtcp = rtp_get_rtcp_file_handle(rtsp_st->rtp_handle);
                if (FFMAX(fd, fd_rtcp) > fd_max)
                    fd_max = FFMAX(fd, fd_rtcp);
R
Ronald S. Bultje 已提交
1621
                FD_SET(fd, &rfds);
1622
                FD_SET(fd_rtcp, &rfds);
R
Ronald S. Bultje 已提交
1623 1624 1625
            }
        }
        tv.tv_sec = 0;
1626
        tv.tv_usec = SELECT_TIMEOUT_MS * 1000;
R
Ronald S. Bultje 已提交
1627 1628
        n = select(fd_max + 1, &rfds, NULL, NULL, &tv);
        if (n > 0) {
1629
            timeout_cnt = 0;
1630
            for (i = 0; i < rt->nb_rtsp_streams; i++) {
R
Ronald S. Bultje 已提交
1631 1632 1633
                rtsp_st = rt->rtsp_streams[i];
                if (rtsp_st->rtp_handle) {
                    fd = url_get_file_handle(rtsp_st->rtp_handle);
1634 1635
                    fd_rtcp = rtp_get_rtcp_file_handle(rtsp_st->rtp_handle);
                    if (FD_ISSET(fd_rtcp, &rfds) || FD_ISSET(fd, &rfds)) {
R
Ronald S. Bultje 已提交
1636 1637 1638 1639 1640 1641 1642 1643
                        ret = url_read(rtsp_st->rtp_handle, buf, buf_size);
                        if (ret > 0) {
                            *prtsp_st = rtsp_st;
                            return ret;
                        }
                    }
                }
            }
1644
#if CONFIG_RTSP_DEMUXER
1645
            if (tcp_fd != -1 && FD_ISSET(tcp_fd, &rfds)) {
R
Ronald S. Bultje 已提交
1646 1647
                RTSPMessageHeader reply;

1648 1649 1650
                ret = ff_rtsp_read_reply(s, &reply, NULL, 0);
                if (ret < 0)
                    return ret;
R
Ronald S. Bultje 已提交
1651
                /* XXX: parse message */
1652
                if (rt->state != RTSP_STATE_STREAMING)
R
Ronald S. Bultje 已提交
1653 1654
                    return 0;
            }
1655
#endif
1656
        } else if (n == 0 && ++timeout_cnt >= MAX_TIMEOUTS) {
1657
            return FF_NETERROR(ETIMEDOUT);
1658 1659
        } else if (n < 0 && errno != EINTR)
            return AVERROR(errno);
R
Ronald S. Bultje 已提交
1660 1661 1662
    }
}

1663
static int tcp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
1664
                           uint8_t *buf, int buf_size);
1665

R
Ronald S. Bultje 已提交
1666 1667 1668 1669
static int rtsp_fetch_packet(AVFormatContext *s, AVPacket *pkt)
{
    RTSPState *rt = s->priv_data;
    int ret, len;
1670 1671
    RTSPStream *rtsp_st, *first_queue_st = NULL;
    int64_t wait_end = 0;
R
Ronald S. Bultje 已提交
1672

1673 1674 1675
    if (rt->nb_byes == rt->nb_rtsp_streams)
        return AVERROR_EOF;

R
Ronald S. Bultje 已提交
1676 1677
    /* get next frames from the same RTP packet */
    if (rt->cur_transport_priv) {
1678
        if (rt->transport == RTSP_TRANSPORT_RDT) {
R
Ronald S. Bultje 已提交
1679
            ret = ff_rdt_parse_packet(rt->cur_transport_priv, pkt, NULL, 0);
1680
        } else
R
Ronald S. Bultje 已提交
1681 1682 1683 1684 1685 1686
            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;
1687
        } else
R
Ronald S. Bultje 已提交
1688 1689 1690
            rt->cur_transport_priv = NULL;
    }

1691 1692 1693 1694 1695 1696 1697 1698 1699 1700 1701 1702 1703 1704 1705 1706
    if (rt->transport == RTSP_TRANSPORT_RTP) {
        int i;
        int64_t first_queue_time = 0;
        for (i = 0; i < rt->nb_rtsp_streams; i++) {
            RTPDemuxContext *rtpctx = rt->rtsp_streams[i]->transport_priv;
            int64_t queue_time = ff_rtp_queued_packet_time(rtpctx);
            if (queue_time && (queue_time - first_queue_time < 0 ||
                               !first_queue_time)) {
                first_queue_time = queue_time;
                first_queue_st   = rt->rtsp_streams[i];
            }
        }
        if (first_queue_time)
            wait_end = first_queue_time + s->max_delay;
    }

R
Ronald S. Bultje 已提交
1707 1708
    /* read next RTP packet */
 redo:
1709 1710 1711 1712 1713 1714
    if (!rt->recvbuf) {
        rt->recvbuf = av_malloc(RECVBUF_SIZE);
        if (!rt->recvbuf)
            return AVERROR(ENOMEM);
    }

R
Ronald S. Bultje 已提交
1715 1716
    switch(rt->lower_transport) {
    default:
1717
#if CONFIG_RTSP_DEMUXER
R
Ronald S. Bultje 已提交
1718
    case RTSP_LOWER_TRANSPORT_TCP:
1719
        len = tcp_read_packet(s, &rtsp_st, rt->recvbuf, RECVBUF_SIZE);
R
Ronald S. Bultje 已提交
1720
        break;
1721
#endif
R
Ronald S. Bultje 已提交
1722 1723
    case RTSP_LOWER_TRANSPORT_UDP:
    case RTSP_LOWER_TRANSPORT_UDP_MULTICAST:
1724
        len = udp_read_packet(s, &rtsp_st, rt->recvbuf, RECVBUF_SIZE, wait_end);
R
Ronald S. Bultje 已提交
1725 1726 1727 1728
        if (len >=0 && rtsp_st->transport_priv && rt->transport == RTSP_TRANSPORT_RTP)
            rtp_check_and_send_back_rr(rtsp_st->transport_priv, len);
        break;
    }
1729 1730 1731 1732 1733 1734
    if (len == AVERROR(EAGAIN) && first_queue_st &&
        rt->transport == RTSP_TRANSPORT_RTP) {
        rtsp_st = first_queue_st;
        ret = rtp_parse_packet(rtsp_st->transport_priv, pkt, NULL, 0);
        goto end;
    }
R
Ronald S. Bultje 已提交
1735 1736 1737 1738
    if (len < 0)
        return len;
    if (len == 0)
        return AVERROR_EOF;
1739
    if (rt->transport == RTSP_TRANSPORT_RDT) {
1740
        ret = ff_rdt_parse_packet(rtsp_st->transport_priv, pkt, &rt->recvbuf, len);
1741
    } else {
1742
        ret = rtp_parse_packet(rtsp_st->transport_priv, pkt, &rt->recvbuf, len);
1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753
        if (ret < 0) {
            /* Either bad packet, or a RTCP packet. Check if the
             * first_rtcp_ntp_time field was initialized. */
            RTPDemuxContext *rtpctx = rtsp_st->transport_priv;
            if (rtpctx->first_rtcp_ntp_time != AV_NOPTS_VALUE) {
                /* first_rtcp_ntp_time has been initialized for this stream,
                 * copy the same value to all other uninitialized streams,
                 * in order to map their timestamp origin to the same ntp time
                 * as this one. */
                int i;
                for (i = 0; i < rt->nb_rtsp_streams; i++) {
1754
                    RTPDemuxContext *rtpctx2 = rt->rtsp_streams[i]->transport_priv;
1755 1756 1757 1758 1759
                    if (rtpctx2 &&
                        rtpctx2->first_rtcp_ntp_time == AV_NOPTS_VALUE)
                        rtpctx2->first_rtcp_ntp_time = rtpctx->first_rtcp_ntp_time;
                }
            }
1760 1761 1762 1763 1764 1765 1766 1767 1768
            if (ret == -RTCP_BYE) {
                rt->nb_byes++;

                av_log(s, AV_LOG_DEBUG, "Received BYE for stream %d (%d/%d)\n",
                       rtsp_st->stream_index, rt->nb_byes, rt->nb_rtsp_streams);

                if (rt->nb_byes == rt->nb_rtsp_streams)
                    return AVERROR_EOF;
            }
1769 1770
        }
    }
1771
end:
R
Ronald S. Bultje 已提交
1772 1773
    if (ret < 0)
        goto redo;
1774
    if (ret == 1)
R
Ronald S. Bultje 已提交
1775 1776 1777 1778 1779 1780
        /* more packets may follow, so we save the RTP context */
        rt->cur_transport_priv = rtsp_st->transport_priv;

    return ret;
}

1781 1782 1783 1784 1785 1786 1787 1788 1789 1790 1791 1792 1793 1794 1795 1796 1797 1798 1799 1800 1801 1802 1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819 1820 1821 1822 1823 1824 1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862
#if CONFIG_RTSP_DEMUXER
static int rtsp_read_header(AVFormatContext *s,
                            AVFormatParameters *ap)
{
    RTSPState *rt = s->priv_data;
    int ret;

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

    rt->real_setup_cache = av_mallocz(2 * s->nb_streams * sizeof(*rt->real_setup_cache));
    if (!rt->real_setup_cache)
        return AVERROR(ENOMEM);
    rt->real_setup = rt->real_setup_cache + s->nb_streams * sizeof(*rt->real_setup);

    if (ap->initial_pause) {
         /* do not start immediately */
    } else {
         if (rtsp_read_play(s) < 0) {
            ff_rtsp_close_streams(s);
            ff_rtsp_close_connections(s);
            return AVERROR_INVALIDDATA;
        }
    }

    return 0;
}

static int tcp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
                           uint8_t *buf, int buf_size)
{
    RTSPState *rt = s->priv_data;
    int id, len, i, ret;
    RTSPStream *rtsp_st;

#ifdef DEBUG_RTP_TCP
    dprintf(s, "tcp_read_packet:\n");
#endif
redo:
    for (;;) {
        RTSPMessageHeader reply;

        ret = ff_rtsp_read_reply(s, &reply, NULL, 1);
        if (ret < 0)
            return ret;
        if (ret == 1) /* received '$' */
            break;
        /* XXX: parse message */
        if (rt->state != RTSP_STATE_STREAMING)
            return 0;
    }
    ret = url_read_complete(rt->rtsp_hd, buf, 3);
    if (ret != 3)
        return -1;
    id  = buf[0];
    len = AV_RB16(buf + 1);
#ifdef DEBUG_RTP_TCP
    dprintf(s, "id=%d len=%d\n", id, len);
#endif
    if (len > buf_size || len < 12)
        goto redo;
    /* get the data */
    ret = url_read_complete(rt->rtsp_hd, buf, len);
    if (ret != len)
        return -1;
    if (rt->transport == RTSP_TRANSPORT_RDT &&
        ff_rdt_parse_header(buf, len, &id, NULL, NULL, NULL, NULL) < 0)
        return -1;

    /* find the matching stream */
    for (i = 0; i < rt->nb_rtsp_streams; i++) {
        rtsp_st = rt->rtsp_streams[i];
        if (id >= rtsp_st->interleaved_min &&
            id <= rtsp_st->interleaved_max)
            goto found;
    }
    goto redo;
found:
    *prtsp_st = rtsp_st;
    return len;
}
1863
static int rtsp_read_packet(AVFormatContext *s, AVPacket *pkt)
1864 1865
{
    RTSPState *rt = s->priv_data;
1866
    int ret;
1867 1868
    RTSPMessageHeader reply1, *reply = &reply1;
    char cmd[1024];
1869

1870
    if (rt->server_type == RTSP_SERVER_REAL) {
1871 1872
        int i;

1873
        for (i = 0; i < s->nb_streams; i++)
1874
            rt->real_setup[i] = s->streams[i]->discard;
1875 1876

        if (!rt->need_subscription) {
1877
            if (memcmp (rt->real_setup, rt->real_setup_cache,
1878
                        sizeof(enum AVDiscard) * s->nb_streams)) {
1879
                snprintf(cmd, sizeof(cmd),
R
Ronald S. Bultje 已提交
1880
                         "Unsubscribe: %s\r\n",
1881 1882 1883
                         rt->last_subscription);
                ff_rtsp_send_cmd(s, "SET_PARAMETER", rt->control_uri,
                                 cmd, reply, NULL);
1884 1885 1886 1887
                if (reply->status_code != RTSP_STATUS_OK)
                    return AVERROR_INVALIDDATA;
                rt->need_subscription = 1;
            }
1888 1889
        }

1890 1891 1892
        if (rt->need_subscription) {
            int r, rule_nr, first = 1;

1893
            memcpy(rt->real_setup_cache, rt->real_setup,
1894 1895 1896 1897
                   sizeof(enum AVDiscard) * s->nb_streams);
            rt->last_subscription[0] = 0;

            snprintf(cmd, sizeof(cmd),
1898
                     "Subscribe: ");
1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909 1910 1911 1912 1913 1914 1915 1916
            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);
1917 1918
            ff_rtsp_send_cmd(s, "SET_PARAMETER", rt->control_uri,
                             cmd, reply, NULL);
1919 1920 1921 1922
            if (reply->status_code != RTSP_STATUS_OK)
                return AVERROR_INVALIDDATA;
            rt->need_subscription = 0;

1923
            if (rt->state == RTSP_STATE_STREAMING)
1924 1925
                rtsp_read_play (s);
        }
1926 1927
    }

L
Luca Barbato 已提交
1928
    ret = rtsp_fetch_packet(s, pkt);
1929
    if (ret < 0)
1930
        return ret;
1931 1932

    /* send dummy request to keep TCP connection alive */
1933
    if ((av_gettime() - rt->last_cmd_time) / 1000000 >= rt->timeout / 2) {
1934
        if (rt->server_type == RTSP_SERVER_WMS) {
1935
            ff_rtsp_send_cmd_async(s, "GET_PARAMETER", rt->control_uri, NULL);
1936
        } else {
1937
            ff_rtsp_send_cmd_async(s, "OPTIONS", "*", NULL);
1938 1939 1940
        }
    }

1941
    return 0;
1942 1943
}

1944 1945
/* pause the stream */
static int rtsp_read_pause(AVFormatContext *s)
1946
{
1947
    RTSPState *rt = s->priv_data;
1948
    RTSPMessageHeader reply1, *reply = &reply1;
1949

1950
    if (rt->state != RTSP_STATE_STREAMING)
1951
        return 0;
1952
    else if (!(rt->server_type == RTSP_SERVER_REAL && rt->need_subscription)) {
1953
        ff_rtsp_send_cmd(s, "PAUSE", rt->control_uri, NULL, reply, NULL);
1954 1955 1956
        if (reply->status_code != RTSP_STATUS_OK) {
            return -1;
        }
1957
    }
1958 1959
    rt->state = RTSP_STATE_PAUSED;
    return 0;
1960 1961
}

1962
static int rtsp_read_seek(AVFormatContext *s, int stream_index,
1963
                          int64_t timestamp, int flags)
1964 1965
{
    RTSPState *rt = s->priv_data;
1966

1967 1968 1969
    rt->seek_timestamp = av_rescale_q(timestamp,
                                      s->streams[stream_index]->time_base,
                                      AV_TIME_BASE_Q);
1970 1971 1972 1973
    switch(rt->state) {
    default:
    case RTSP_STATE_IDLE:
        break;
1974
    case RTSP_STATE_STREAMING:
1975 1976 1977
        if (rtsp_read_pause(s) != 0)
            return -1;
        rt->state = RTSP_STATE_SEEKING;
1978 1979 1980 1981 1982 1983 1984 1985 1986 1987
        if (rtsp_read_play(s) != 0)
            return -1;
        break;
    case RTSP_STATE_PAUSED:
        rt->state = RTSP_STATE_IDLE;
        break;
    }
    return 0;
}

1988 1989 1990 1991
static int rtsp_read_close(AVFormatContext *s)
{
    RTSPState *rt = s->priv_data;

F
Fabrice Bellard 已提交
1992
#if 0
1993
    /* NOTE: it is valid to flush the buffer here */
1994
    if (rt->lower_transport == RTSP_LOWER_TRANSPORT_TCP) {
1995 1996
        url_fclose(&rt->rtsp_gb);
    }
F
Fabrice Bellard 已提交
1997
#endif
1998
    ff_rtsp_send_cmd_async(s, "TEARDOWN", rt->control_uri, NULL);
1999

2000
    ff_rtsp_close_streams(s);
2001
    ff_rtsp_close_connections(s);
2002
    ff_network_close();
2003 2004
    rt->real_setup = NULL;
    av_freep(&rt->real_setup_cache);
2005 2006 2007
    return 0;
}

2008
AVInputFormat rtsp_demuxer = {
2009
    "rtsp",
2010
    NULL_IF_CONFIG_SMALL("RTSP input format"),
2011 2012 2013 2014 2015
    sizeof(RTSPState),
    rtsp_probe,
    rtsp_read_header,
    rtsp_read_packet,
    rtsp_read_close,
2016
    rtsp_read_seek,
2017
    .flags = AVFMT_NOFILE,
2018 2019
    .read_play = rtsp_read_play,
    .read_pause = rtsp_read_pause,
2020
};
2021
#endif /* CONFIG_RTSP_DEMUXER */
2022

2023
static int sdp_probe(AVProbeData *p1)
2024
{
2025
    const char *p = p1->buf, *p_end = p1->buf + p1->buf_size;
2026

M
Martin Storsjö 已提交
2027
    /* we look for a line beginning "c=IN IP" */
2028
    while (p < p_end && *p != '\0') {
M
Martin Storsjö 已提交
2029 2030
        if (p + sizeof("c=IN IP") - 1 < p_end &&
            av_strstart(p, "c=IN IP", NULL))
2031
            return AVPROBE_SCORE_MAX / 2;
2032

2033
        while (p < p_end - 1 && *p != '\n') p++;
2034
        if (++p >= p_end)
2035 2036 2037 2038
            break;
        if (*p == '\r')
            p++;
    }
2039 2040 2041
    return 0;
}

2042
static int sdp_read_header(AVFormatContext *s, AVFormatParameters *ap)
2043
{
2044
    RTSPState *rt = s->priv_data;
2045 2046 2047 2048 2049
    RTSPStream *rtsp_st;
    int size, i, err;
    char *content;
    char url[1024];

2050 2051 2052
    if (!ff_network_init())
        return AVERROR(EIO);

2053 2054 2055
    /* read the whole sdp file */
    /* XXX: better loading */
    content = av_malloc(SDP_MAX_SIZE);
2056
    size = get_buffer(s->pb, content, SDP_MAX_SIZE - 1);
2057 2058 2059 2060 2061 2062 2063 2064 2065 2066
    if (size <= 0) {
        av_free(content);
        return AVERROR_INVALIDDATA;
    }
    content[size] ='\0';

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

    /* open each RTP stream */
2067
    for (i = 0; i < rt->nb_rtsp_streams; i++) {
M
Martin Storsjö 已提交
2068
        char namebuf[50];
2069
        rtsp_st = rt->rtsp_streams[i];
2070

M
Martin Storsjö 已提交
2071 2072
        getnameinfo((struct sockaddr*) &rtsp_st->sdp_ip, sizeof(rtsp_st->sdp_ip),
                    namebuf, sizeof(namebuf), NULL, 0, NI_NUMERICHOST);
2073
        ff_url_join(url, sizeof(url), "rtp", NULL,
M
Martin Storsjö 已提交
2074
                    namebuf, rtsp_st->sdp_port,
2075 2076
                    "?localport=%d&ttl=%d", rtsp_st->sdp_port,
                    rtsp_st->sdp_ttl);
2077
        if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
2078 2079 2080
            err = AVERROR_INVALIDDATA;
            goto fail;
        }
2081
        if ((err = rtsp_open_transport_ctx(s, rtsp_st)))
2082
            goto fail;
2083 2084
    }
    return 0;
2085
fail:
2086
    ff_rtsp_close_streams(s);
2087
    ff_network_close();
2088 2089 2090 2091 2092
    return err;
}

static int sdp_read_close(AVFormatContext *s)
{
2093
    ff_rtsp_close_streams(s);
2094
    ff_network_close();
2095 2096 2097
    return 0;
}

2098
AVInputFormat sdp_demuxer = {
2099
    "sdp",
2100
    NULL_IF_CONFIG_SMALL("SDP"),
2101 2102 2103
    sizeof(RTSPState),
    sdp_probe,
    sdp_read_header,
L
Luca Barbato 已提交
2104
    rtsp_fetch_packet,
2105 2106
    sdp_read_close,
};
2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160 2161 2162 2163 2164 2165 2166 2167 2168 2169 2170 2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210

static int rtp_probe(AVProbeData *p)
{
    if (av_strstart(p->filename, "rtp:", NULL))
        return AVPROBE_SCORE_MAX;
    return 0;
}

static int rtp_read_header(AVFormatContext *s,
                           AVFormatParameters *ap)
{
    uint8_t recvbuf[1500];
    char host[500], sdp[500];
    int ret, port;
    URLContext* in = NULL;
    int payload_type;
    AVCodecContext codec;
    struct sockaddr_storage addr;
    ByteIOContext pb;
    socklen_t addrlen = sizeof(addr);

    if (!ff_network_init())
        return AVERROR(EIO);

    ret = url_open(&in, s->filename, URL_RDONLY);
    if (ret)
        goto fail;

    while (1) {
        ret = url_read(in, recvbuf, sizeof(recvbuf));
        if (ret == AVERROR(EAGAIN))
            continue;
        if (ret < 0)
            goto fail;
        if (ret < 12) {
            av_log(s, AV_LOG_WARNING, "Received too short packet\n");
            continue;
        }

        if ((recvbuf[0] & 0xc0) != 0x80) {
            av_log(s, AV_LOG_WARNING, "Unsupported RTP version packet "
                                      "received\n");
            continue;
        }

        payload_type = recvbuf[1] & 0x7f;
        break;
    }
    getsockname(url_get_file_handle(in), (struct sockaddr*) &addr, &addrlen);
    url_close(in);
    in = NULL;

    memset(&codec, 0, sizeof(codec));
    if (ff_rtp_get_codec_info(&codec, payload_type)) {
        av_log(s, AV_LOG_ERROR, "Unable to receive RTP payload type %d "
                                "without an SDP file describing it\n",
                                 payload_type);
        goto fail;
    }
    if (codec.codec_type != AVMEDIA_TYPE_DATA) {
        av_log(s, AV_LOG_WARNING, "Guessing on RTP content - if not received "
                                  "properly you need an SDP file "
                                  "describing it\n");
    }

    av_url_split(NULL, 0, NULL, 0, host, sizeof(host), &port,
                 NULL, 0, s->filename);

    snprintf(sdp, sizeof(sdp),
             "v=0\r\nc=IN IP%d %s\r\nm=%s %d RTP/AVP %d\r\n",
             addr.ss_family == AF_INET ? 4 : 6, host,
             codec.codec_type == AVMEDIA_TYPE_DATA  ? "application" :
             codec.codec_type == AVMEDIA_TYPE_VIDEO ? "video" : "audio",
             port, payload_type);
    av_log(s, AV_LOG_VERBOSE, "SDP:\n%s\n", sdp);

    init_put_byte(&pb, sdp, strlen(sdp), 0, NULL, NULL, NULL, NULL);
    s->pb = &pb;

    /* sdp_read_header initializes this again */
    ff_network_close();

    ret = sdp_read_header(s, ap);
    s->pb = NULL;
    return ret;

fail:
    if (in)
        url_close(in);
    ff_network_close();
    return ret;
}

AVInputFormat rtp_demuxer = {
    "rtp",
    NULL_IF_CONFIG_SMALL("RTP input format"),
    sizeof(RTSPState),
    rtp_probe,
    rtp_read_header,
    rtsp_fetch_packet,
    sdp_read_close,
    .flags = AVFMT_NOFILE,
};