rtsp.c 63.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
#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
Reimar Döffinger 已提交
47
/* Timeout values for socket select, in ms,
48 49 50 51
 * 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
52
#define SDP_MAX_SIZE 16384
53
#define RECVBUF_SIZE 10 * RTP_MAX_PACKET_LENGTH
54

55 56
static void get_word_until_chars(char *buf, int buf_size,
                                 const char *sep, const char **pp)
57 58 59 60 61
{
    const char *p;
    char *q;

    p = *pp;
62
    p += strspn(p, SPACE_CHARS);
63 64 65 66 67 68 69 70 71 72 73
    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;
}

74 75
static void get_word_sep(char *buf, int buf_size, const char *sep,
                         const char **pp)
76
{
77 78 79
    if (**pp == '/') (*pp)++;
    get_word_until_chars(buf, buf_size, sep, pp);
}
80

81 82 83
static void get_word(char *buf, int buf_size, const char **pp)
{
    get_word_until_chars(buf, buf_size, SPACE_CHARS, pp);
84 85
}

M
Martin Storsjö 已提交
86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123
/** 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;
}

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

R
Romain Degez 已提交
135
    /* Loop into AVRtpDynamicPayloadTypes[] and AVRtpPayloadTypes[] and
136 137 138 139 140
     * 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 已提交
141
    if (payload_type >= RTP_PT_PRIVATE) {
142 143 144 145 146 147 148 149 150
        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 已提交
151 152
                break;
            }
R
Ryan Martell 已提交
153
        }
154 155 156 157 158 159
        /* 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);
160
    } else {
161 162
        /* We are in a standard case
         * (from http://www.iana.org/assignments/rtp-parameters). */
R
Romain Degez 已提交
163
        /* search into AVRtpPayloadTypes[] */
164
        codec->codec_id = ff_rtp_codec_id(buf, codec->codec_type);
R
Romain Degez 已提交
165 166 167 168
    }

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

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

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

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

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

244
    dprintf(s, "sdp: %c='%s'\n", letter, buf);
245 246

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

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

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

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

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

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

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

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

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

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

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

515
    if (s->oformat && CONFIG_RTSP_MUXER) {
516 517 518
        rtsp_st->transport_priv = ff_rtp_chain_mux_open(s, st,
                                      rtsp_st->rtp_handle,
                                      RTSP_TCP_MAX_PACKET_SIZE);
R
Reimar Döffinger 已提交
519
        /* Ownership of rtp_handle is passed to the rtp mux context */
520
        rtsp_st->rtp_handle = NULL;
521
    } else if (rt->transport == RTSP_TRANSPORT_RDT && CONFIG_RTPDEC)
522 523 524
        rtsp_st->transport_priv = ff_rdt_parse_open(s, st->index,
                                            rtsp_st->dynamic_protocol_context,
                                            rtsp_st->dynamic_handler);
525
    else if (CONFIG_RTPDEC)
526
        rtsp_st->transport_priv = rtp_parse_open(s, st, rtsp_st->rtp_handle,
527 528 529
                                         rtsp_st->sdp_payload_type,
            (rt->lower_transport == RTSP_LOWER_TRANSPORT_TCP || !s->max_delay)
            ? 0 : RTP_REORDER_QUEUE_DEFAULT_SIZE);
530 531 532

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

    return 0;
}

M
Martin Storsjö 已提交
544
#if CONFIG_RTSP_DEMUXER || CONFIG_RTSP_MUXER
545 546 547 548 549 550
static void rtsp_parse_range(int *min_ptr, int *max_ptr, const char **pp)
{
    const char *p;
    int v;

    p = *pp;
551
    p += strspn(p, SPACE_CHARS);
552 553 554 555 556 557 558 559 560 561 562 563 564 565
    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 */
566
static void rtsp_parse_transport(RTSPMessageHeader *reply, const char *p)
567 568 569 570 571 572 573
{
    char transport_protocol[16];
    char profile[16];
    char lower_transport[16];
    char parameter[16];
    RTSPTransportField *th;
    char buf[256];
574

575
    reply->nb_transports = 0;
576

577
    for (;;) {
578
        p += strspn(p, SPACE_CHARS);
579 580 581 582 583
        if (*p == '\0')
            break;

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

584
        get_word_sep(transport_protocol, sizeof(transport_protocol),
585
                     "/", &p);
586
        if (!strcasecmp (transport_protocol, "rtp")) {
587 588 589 590 591 592
            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);
593 594 595 596
            }
            th->transport = RTSP_TRANSPORT_RTP;
        } else if (!strcasecmp (transport_protocol, "x-pn-tng") ||
                   !strcasecmp (transport_protocol, "x-real-rdt")) {
597
            /* x-pn-tng/<protocol> */
598 599
            get_word_sep(lower_transport, sizeof(lower_transport), "/;,", &p);
            profile[0] = '\0';
600
            th->transport = RTSP_TRANSPORT_RDT;
601
        }
F
Fabrice Bellard 已提交
602
        if (!strcasecmp(lower_transport, "TCP"))
603
            th->lower_transport = RTSP_LOWER_TRANSPORT_TCP;
604
        else
605
            th->lower_transport = RTSP_LOWER_TRANSPORT_UDP;
606

607 608 609 610 611 612 613 614 615 616 617 618 619
        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++;
620
                    rtsp_parse_range(&th->client_port_min,
621 622 623 624 625
                                     &th->client_port_max, &p);
                }
            } else if (!strcmp(parameter, "server_port")) {
                if (*p == '=') {
                    p++;
626
                    rtsp_parse_range(&th->server_port_min,
627 628 629 630 631
                                     &th->server_port_max, &p);
                }
            } else if (!strcmp(parameter, "interleaved")) {
                if (*p == '=') {
                    p++;
632
                    rtsp_parse_range(&th->interleaved_min,
633 634 635
                                     &th->interleaved_max, &p);
                }
            } else if (!strcmp(parameter, "multicast")) {
636 637
                if (th->lower_transport == RTSP_LOWER_TRANSPORT_UDP)
                    th->lower_transport = RTSP_LOWER_TRANSPORT_UDP_MULTICAST;
638 639 640 641 642 643 644 645 646
            } 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ö 已提交
647
                    get_sockaddr(buf, &th->destination);
648
                }
649 650 651 652 653 654
            } else if (!strcmp(parameter, "source")) {
                if (*p == '=') {
                    p++;
                    get_word_sep(buf, sizeof(buf), ";,", &p);
                    av_strlcpy(th->source, buf, sizeof(th->source));
                }
655
            }
656

657 658 659 660 661 662 663 664 665 666 667 668
            while (*p != ';' && *p != '\0' && *p != ',')
                p++;
            if (*p == ';')
                p++;
        }
        if (*p == ',')
            p++;

        reply->nb_transports++;
    }
}

669 670
void ff_rtsp_parse_line(RTSPMessageHeader *reply, const char *buf,
                        HTTPAuthState *auth_state)
671 672 673 674 675
{
    const char *p;

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

715
/* skip a RTP/TCP interleaved packet */
716
void ff_rtsp_skip_packet(AVFormatContext *s)
717 718 719 720 721
{
    RTSPState *rt = s->priv_data;
    int ret, len, len1;
    uint8_t buf[1024];

722
    ret = url_read_complete(rt->rtsp_hd, buf, 3);
723 724
    if (ret != 3)
        return;
725
    len = AV_RB16(buf + 1);
726 727 728

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

729 730 731 732 733
    /* skip payload */
    while (len > 0) {
        len1 = len;
        if (len1 > sizeof(buf))
            len1 = sizeof(buf);
734
        ret = url_read_complete(rt->rtsp_hd, buf, len1);
735 736 737 738 739
        if (ret != len1)
            return;
        len -= len1;
    }
}
740

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

752
    memset(reply, 0, sizeof(*reply));
753 754 755

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

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

782 783 784 785 786 787 788 789 790
        /* 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 已提交
791
            av_strlcpy(reply->reason, p, sizeof(reply->reason));
792
        } else {
793
            ff_rtsp_parse_line(reply, p, &rt->auth_state);
M
Måns Rullgård 已提交
794 795
            av_strlcat(rt->last_reply, p,    sizeof(rt->last_reply));
            av_strlcat(rt->last_reply, "\n", sizeof(rt->last_reply));
796 797 798
        }
        line_count++;
    }
799

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

803 804 805 806
    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);
807
        (void)url_read_complete(rt->rtsp_hd, content, content_length);
808 809 810 811
        content[content_length] = '\0';
    }
    if (content_ptr)
        *content_ptr = content;
812 813
    else
        av_free(content);
814

815 816 817 818 819
    if (rt->seq != reply->seq) {
        av_log(s, AV_LOG_WARNING, "CSeq %d expected, %d received.\n",
            rt->seq, reply->seq);
    }

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

831
    return 0;
832 833
}

834
int ff_rtsp_send_cmd_with_content_async(AVFormatContext *s,
M
Martin Storsjö 已提交
835 836 837 838
                                        const char *method, const char *url,
                                        const char *headers,
                                        const unsigned char *send_content,
                                        int send_content_length)
839 840
{
    RTSPState *rt = s->priv_data;
J
Josh Allmann 已提交
841 842
    char buf[4096], *out_buf;
    char base64buf[AV_BASE64_SIZE(sizeof(buf))];
843

J
Josh Allmann 已提交
844 845
    /* Add in RTSP headers */
    out_buf = buf;
846
    rt->seq++;
847 848 849
    snprintf(buf, sizeof(buf), "%s %s RTSP/1.0\r\n", method, url);
    if (headers)
        av_strlcat(buf, headers, sizeof(buf));
850
    av_strlcatf(buf, sizeof(buf), "CSeq: %d\r\n", rt->seq);
851 852
    if (rt->session_id[0] != '\0' && (!headers ||
        !strstr(headers, "\nIf-Match:"))) {
853
        av_strlcatf(buf, sizeof(buf), "Session: %s\r\n", rt->session_id);
854
    }
855 856 857 858 859 860 861
    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);
    }
862 863
    if (send_content_length > 0 && send_content)
        av_strlcatf(buf, sizeof(buf), "Content-Length: %d\r\n", send_content_length);
864
    av_strlcat(buf, "\r\n", sizeof(buf));
865

J
Josh Allmann 已提交
866 867 868 869 870 871
    /* base64 encode rtsp if tunneling */
    if (rt->control_transport == RTSP_MODE_TUNNEL) {
        av_base64_encode(base64buf, sizeof(base64buf), buf, strlen(buf));
        out_buf = base64buf;
    }

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

J
Josh Allmann 已提交
874 875 876 877 878 879 880
    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;
        }
881
        url_write(rt->rtsp_hd_out, send_content, send_content_length);
J
Josh Allmann 已提交
882
    }
883
    rt->last_cmd_time = av_gettime();
884 885

    return 0;
886 887
}

888
int ff_rtsp_send_cmd_async(AVFormatContext *s, const char *method,
M
Martin Storsjö 已提交
889
                           const char *url, const char *headers)
890
{
891
    return ff_rtsp_send_cmd_with_content_async(s, method, url, headers, NULL, 0);
892 893
}

894
int ff_rtsp_send_cmd(AVFormatContext *s, const char *method, const char *url,
M
Martin Storsjö 已提交
895 896
                     const char *headers, RTSPMessageHeader *reply,
                     unsigned char **content_ptr)
897
{
898
    return ff_rtsp_send_cmd_with_content(s, method, url, headers, reply,
M
Martin Storsjö 已提交
899
                                         content_ptr, NULL, 0);
900 901
}

902
int ff_rtsp_send_cmd_with_content(AVFormatContext *s,
M
Martin Storsjö 已提交
903 904 905 906 907 908
                                  const char *method, const char *url,
                                  const char *header,
                                  RTSPMessageHeader *reply,
                                  unsigned char **content_ptr,
                                  const unsigned char *send_content,
                                  int send_content_length)
909
{
910 911
    RTSPState *rt = s->priv_data;
    HTTPAuthType cur_auth_type;
912
    int ret;
913 914 915

retry:
    cur_auth_type = rt->auth_state.auth_type;
916
    if ((ret = ff_rtsp_send_cmd_with_content_async(s, method, url, header,
M
Martin Storsjö 已提交
917 918
                                                   send_content,
                                                   send_content_length)))
919
        return ret;
920

921 922
    if ((ret = ff_rtsp_read_reply(s, reply, content_ptr, 0) ) < 0)
        return ret;
923 924 925 926

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

928
    if (reply->status_code > 400){
L
Luca Barbato 已提交
929
        av_log(s, AV_LOG_ERROR, "method %s failed: %d%s\n",
930
               method,
L
Luca Barbato 已提交
931 932
               reply->status_code,
               reply->reason);
933 934 935
        av_log(s, AV_LOG_DEBUG, "%s\n", rt->last_reply);
    }

936
    return 0;
937 938
}

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

952
    if (rt->transport == RTSP_TRANSPORT_RDT)
953 954 955
        trans_pref = "x-pn-tng";
    else
        trans_pref = "RTP/AVP";
956

957 958 959
    /* default timeout: 1 minute */
    rt->timeout = 60;

960 961
    /* for each stream, make the setup request */
    /* XXX: we assume the same server is used for the control of each
962
     * RTSP stream */
R
Romain Degez 已提交
963

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

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

        /* RTP/UDP */
992
        if (lower_transport == RTSP_LOWER_TRANSPORT_UDP) {
F
Fabrice Bellard 已提交
993 994
            char buf[256];

995 996 997 998 999
            if (rt->server_type == RTSP_SERVER_WMS && i > 1) {
                port = reply->transports[0].client_port_min;
                goto have_port;
            }

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

1012 1013 1014 1015 1016 1017 1018
#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 已提交
1019 1020

        rtp_opened:
1021
            port = rtp_get_local_rtp_port(rtsp_st->rtp_handle);
1022
        have_port:
1023
            snprintf(transport, sizeof(transport) - 1,
1024 1025 1026 1027 1028
                     "%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);
1029 1030
            if (rt->transport == RTSP_TRANSPORT_RTP &&
                !(rt->server_type == RTSP_SERVER_WMS && i > 0))
1031
                av_strlcatf(transport, sizeof(transport), "-%d", port + 1);
1032 1033 1034
        }

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

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

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

R
Reinhard Tartler 已提交
1096
        /* close RTP connection if not chosen */
1097 1098
        if (reply->transports[0].lower_transport != RTSP_LOWER_TRANSPORT_UDP &&
            (lower_transport == RTSP_LOWER_TRANSPORT_UDP)) {
1099 1100
            url_close(rtsp_st->rtp_handle);
            rtsp_st->rtp_handle = NULL;
1101 1102
        }

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

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

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

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

1161
        if ((err = rtsp_open_transport_ctx(s, rtsp_st)))
1162
            goto fail;
1163 1164
    }

1165 1166 1167
    if (reply->timeout > 0)
        rt->timeout = reply->timeout;

1168
    if (rt->server_type == RTSP_SERVER_REAL)
1169 1170
        rt->need_subscription = 1;

1171 1172 1173
    return 0;

fail:
1174
    for (i = 0; i < rt->nb_rtsp_streams; i++) {
1175 1176 1177 1178 1179
        if (rt->rtsp_streams[i]->rtp_handle) {
            url_close(rt->rtsp_streams[i]->rtp_handle);
            rt->rtsp_streams[i]->rtp_handle = NULL;
        }
    }
1180 1181 1182
    return err;
}

1183 1184 1185 1186 1187
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);
1188
    rt->rtsp_hd = rt->rtsp_hd_out = NULL;
1189 1190
}

1191
int ff_rtsp_connect(AVFormatContext *s)
1192 1193
{
    RTSPState *rt = s->priv_data;
1194 1195
    char host[1024], path[1024], tcpname[1024], cmd[2048], auth[128];
    char *option_list, *option, *filename;
1196
    int port, err, tcp_fd;
1197
    RTSPMessageHeader reply1 = {0}, *reply = &reply1;
1198
    int lower_transport_mask = 0;
1199
    char real_challenge[64];
1200 1201
    struct sockaddr_storage peer;
    socklen_t peer_len = sizeof(peer);
1202 1203 1204

    if (!ff_network_init())
        return AVERROR(EIO);
1205
redirect:
J
Josh Allmann 已提交
1206
    rt->control_transport = RTSP_MODE_PLAIN;
1207
    /* extract hostname and port */
M
Måns Rullgård 已提交
1208
    av_url_split(NULL, 0, auth, sizeof(auth),
M
Martin Storsjö 已提交
1209
                 host, sizeof(host), &port, path, sizeof(path), s->filename);
1210
    if (*auth) {
1211
        av_strlcpy(rt->auth, auth, sizeof(rt->auth));
1212
    }
1213 1214 1215 1216
    if (port < 0)
        port = RTSP_DEFAULT_PORT;

    /* search for options */
1217
    option_list = strrchr(path, '?');
1218
    if (option_list) {
1219 1220 1221
        /* Strip out the RTSP specific options, write out the rest of
         * the options back into the same string. */
        filename = option_list;
1222
        while (option_list) {
1223
            /* move the option pointer */
1224
            option = ++option_list;
1225 1226
            option_list = strchr(option_list, '&');
            if (option_list)
1227 1228
                *option_list = 0;

1229
            /* handle the options */
1230
            if (!strcmp(option, "udp")) {
1231
                lower_transport_mask |= (1<< RTSP_LOWER_TRANSPORT_UDP);
1232
            } else if (!strcmp(option, "multicast")) {
1233
                lower_transport_mask |= (1<< RTSP_LOWER_TRANSPORT_UDP_MULTICAST);
1234
            } else if (!strcmp(option, "tcp")) {
1235
                lower_transport_mask |= (1<< RTSP_LOWER_TRANSPORT_TCP);
J
Josh Allmann 已提交
1236 1237 1238
            } else if(!strcmp(option, "http")) {
                lower_transport_mask |= (1<< RTSP_LOWER_TRANSPORT_TCP);
                rt->control_transport = RTSP_MODE_TUNNEL;
1239
            } else {
1240 1241 1242 1243 1244
                /* 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;
1245 1246
                if (option_list) *filename = '&';
            }
1247
        }
1248
        *filename = 0;
1249 1250
    }

1251
    if (!lower_transport_mask)
1252
        lower_transport_mask = (1 << RTSP_LOWER_TRANSPORT_NB) - 1;
1253

1254
    if (s->oformat) {
1255 1256 1257
        /* 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 已提交
1258
        if (!lower_transport_mask || rt->control_transport == RTSP_MODE_TUNNEL) {
1259
            av_log(s, AV_LOG_ERROR, "Unsupported lower transport method, "
1260
                                    "only UDP and TCP are supported for output.\n");
1261 1262 1263 1264 1265
            err = AVERROR(EINVAL);
            goto fail;
        }
    }

1266 1267 1268 1269 1270 1271
    /* 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 已提交
1272 1273 1274 1275 1276 1277
    if (rt->control_transport == RTSP_MODE_TUNNEL) {
        /* set up initial handshake for tunneling */
        char httpname[1024];
        char sessioncookie[17];
        char headers[1024];

1278
        ff_url_join(httpname, sizeof(httpname), "http", auth, host, port, "%s", path);
J
Josh Allmann 已提交
1279 1280 1281 1282
        snprintf(sessioncookie, sizeof(sessioncookie), "%08x%08x",
                 av_get_random_seed(), av_get_random_seed());

        /* GET requests */
1283
        if (url_alloc(&rt->rtsp_hd, httpname, URL_RDONLY) < 0) {
J
Josh Allmann 已提交
1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294
            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);
1295
        ff_http_set_headers(rt->rtsp_hd, headers);
J
Josh Allmann 已提交
1296 1297

        /* complete the connection */
1298
        if (url_connect(rt->rtsp_hd)) {
J
Josh Allmann 已提交
1299 1300 1301 1302 1303
            err = AVERROR(EIO);
            goto fail;
        }

        /* POST requests */
1304
        if (url_alloc(&rt->rtsp_hd_out, httpname, URL_WRONLY) < 0 ) {
J
Josh Allmann 已提交
1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317
            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);
1318 1319
        ff_http_set_headers(rt->rtsp_hd_out, headers);
        ff_http_set_chunked_transfer_encoding(rt->rtsp_hd_out, 0);
J
Josh Allmann 已提交
1320

1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338
        /* 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);

1339 1340 1341 1342 1343
        /* complete the connection */
        if (url_connect(rt->rtsp_hd_out)) {
            err = AVERROR(EIO);
            goto fail;
        }
J
Josh Allmann 已提交
1344
    } else {
1345
        /* open the tcp connection */
J
Josh Allmann 已提交
1346
        ff_url_join(tcpname, sizeof(tcpname), "tcp", NULL, host, port, NULL);
1347
        if (url_open(&rt->rtsp_hd, tcpname, URL_RDWR) < 0) {
J
Josh Allmann 已提交
1348 1349 1350
            err = AVERROR(EIO);
            goto fail;
        }
1351
        rt->rtsp_hd_out = rt->rtsp_hd;
J
Josh Allmann 已提交
1352
    }
1353 1354
    rt->seq = 0;

1355
    tcp_fd = url_get_file_handle(rt->rtsp_hd);
1356 1357 1358 1359 1360
    if (!getpeername(tcp_fd, (struct sockaddr*) &peer, &peer_len)) {
        getnameinfo((struct sockaddr*) &peer, peer_len, host, sizeof(host),
                    NULL, 0, NI_NUMERICHOST);
    }

1361 1362
    /* request options supported by the server; this also detects server
     * type */
1363
    for (rt->server_type = RTSP_SERVER_RTP;;) {
1364
        cmd[0] = 0;
1365
        if (rt->server_type == RTSP_SERVER_REAL)
1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380
            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));
1381
        ff_rtsp_send_cmd(s, "OPTIONS", rt->control_uri, cmd, reply, NULL);
1382 1383 1384 1385 1386 1387
        if (reply->status_code != RTSP_STATUS_OK) {
            err = AVERROR_INVALIDDATA;
            goto fail;
        }

        /* detect server type if not standard-compliant RTP */
1388 1389
        if (rt->server_type != RTSP_SERVER_REAL && reply->real_challenge[0]) {
            rt->server_type = RTSP_SERVER_REAL;
1390
            continue;
1391 1392
        } else if (!strncasecmp(reply->server, "WMServer/", 9)) {
            rt->server_type = RTSP_SERVER_WMS;
1393
        } else if (rt->server_type == RTSP_SERVER_REAL)
1394 1395 1396 1397
            strcpy(real_challenge, reply->real_challenge);
        break;
    }

1398
    if (s->iformat && CONFIG_RTSP_DEMUXER)
1399
        err = ff_rtsp_setup_input_streams(s, reply);
1400
    else if (CONFIG_RTSP_MUXER)
1401
        err = ff_rtsp_setup_output_streams(s, host);
1402
    if (err)
1403 1404
        goto fail;

1405
    do {
1406 1407
        int lower_transport = ff_log2_tab[lower_transport_mask &
                                  ~(lower_transport_mask - 1)];
1408

1409
        err = make_setup_request(s, host, port, lower_transport,
1410
                                 rt->server_type == RTSP_SERVER_REAL ?
1411
                                     real_challenge : NULL);
1412
        if (err < 0)
1413
            goto fail;
1414 1415
        lower_transport_mask &= ~(1 << lower_transport);
        if (lower_transport_mask == 0 && err == 1) {
1416
            err = FF_NETERROR(EPROTONOSUPPORT);
1417 1418 1419
            goto fail;
        }
    } while (err);
1420

1421
    rt->state = RTSP_STATE_IDLE;
1422
    rt->seek_timestamp = 0; /* default is to start stream at position zero */
1423 1424
    return 0;
 fail:
1425
    ff_rtsp_close_streams(s);
1426
    ff_rtsp_close_connections(s);
1427
    if (reply->status_code >=300 && reply->status_code < 400 && s->iformat) {
L
Luca Barbato 已提交
1428 1429 1430 1431 1432 1433
        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;
    }
1434
    ff_network_close();
1435 1436
    return err;
}
1437
#endif /* CONFIG_RTSP_DEMUXER || CONFIG_RTSP_MUXER */
1438

1439
#if CONFIG_RTPDEC
R
Ronald S. Bultje 已提交
1440
static int udp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
1441
                           uint8_t *buf, int buf_size, int64_t wait_end)
R
Ronald S. Bultje 已提交
1442 1443 1444 1445
{
    RTSPState *rt = s->priv_data;
    RTSPStream *rtsp_st;
    fd_set rfds;
1446
    int fd, fd_rtcp, fd_max, n, i, ret, tcp_fd, timeout_cnt = 0;
R
Ronald S. Bultje 已提交
1447 1448
    struct timeval tv;

1449
    for (;;) {
R
Ronald S. Bultje 已提交
1450 1451
        if (url_interrupt_cb())
            return AVERROR(EINTR);
1452 1453
        if (wait_end && wait_end - av_gettime() < 0)
            return AVERROR(EAGAIN);
R
Ronald S. Bultje 已提交
1454 1455 1456 1457 1458 1459 1460 1461
        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;
        }
1462
        for (i = 0; i < rt->nb_rtsp_streams; i++) {
R
Ronald S. Bultje 已提交
1463 1464 1465
            rtsp_st = rt->rtsp_streams[i];
            if (rtsp_st->rtp_handle) {
                fd = url_get_file_handle(rtsp_st->rtp_handle);
1466 1467 1468
                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 已提交
1469
                FD_SET(fd, &rfds);
1470
                FD_SET(fd_rtcp, &rfds);
R
Ronald S. Bultje 已提交
1471 1472 1473
            }
        }
        tv.tv_sec = 0;
1474
        tv.tv_usec = SELECT_TIMEOUT_MS * 1000;
R
Ronald S. Bultje 已提交
1475 1476
        n = select(fd_max + 1, &rfds, NULL, NULL, &tv);
        if (n > 0) {
1477
            timeout_cnt = 0;
1478
            for (i = 0; i < rt->nb_rtsp_streams; i++) {
R
Ronald S. Bultje 已提交
1479 1480 1481
                rtsp_st = rt->rtsp_streams[i];
                if (rtsp_st->rtp_handle) {
                    fd = url_get_file_handle(rtsp_st->rtp_handle);
1482 1483
                    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 已提交
1484 1485 1486 1487 1488 1489 1490 1491
                        ret = url_read(rtsp_st->rtp_handle, buf, buf_size);
                        if (ret > 0) {
                            *prtsp_st = rtsp_st;
                            return ret;
                        }
                    }
                }
            }
1492
#if CONFIG_RTSP_DEMUXER
1493
            if (tcp_fd != -1 && FD_ISSET(tcp_fd, &rfds)) {
R
Ronald S. Bultje 已提交
1494 1495
                RTSPMessageHeader reply;

1496 1497 1498
                ret = ff_rtsp_read_reply(s, &reply, NULL, 0);
                if (ret < 0)
                    return ret;
R
Ronald S. Bultje 已提交
1499
                /* XXX: parse message */
1500
                if (rt->state != RTSP_STATE_STREAMING)
R
Ronald S. Bultje 已提交
1501 1502
                    return 0;
            }
1503
#endif
1504
        } else if (n == 0 && ++timeout_cnt >= MAX_TIMEOUTS) {
1505
            return FF_NETERROR(ETIMEDOUT);
1506 1507
        } else if (n < 0 && errno != EINTR)
            return AVERROR(errno);
R
Ronald S. Bultje 已提交
1508 1509 1510
    }
}

1511
int ff_rtsp_fetch_packet(AVFormatContext *s, AVPacket *pkt)
R
Ronald S. Bultje 已提交
1512 1513 1514
{
    RTSPState *rt = s->priv_data;
    int ret, len;
1515 1516
    RTSPStream *rtsp_st, *first_queue_st = NULL;
    int64_t wait_end = 0;
R
Ronald S. Bultje 已提交
1517

1518 1519 1520
    if (rt->nb_byes == rt->nb_rtsp_streams)
        return AVERROR_EOF;

R
Ronald S. Bultje 已提交
1521 1522
    /* get next frames from the same RTP packet */
    if (rt->cur_transport_priv) {
1523
        if (rt->transport == RTSP_TRANSPORT_RDT) {
R
Ronald S. Bultje 已提交
1524
            ret = ff_rdt_parse_packet(rt->cur_transport_priv, pkt, NULL, 0);
1525
        } else
R
Ronald S. Bultje 已提交
1526 1527 1528 1529 1530 1531
            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;
1532
        } else
R
Ronald S. Bultje 已提交
1533 1534 1535
            rt->cur_transport_priv = NULL;
    }

1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551
    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 已提交
1552 1553
    /* read next RTP packet */
 redo:
1554 1555 1556 1557 1558 1559
    if (!rt->recvbuf) {
        rt->recvbuf = av_malloc(RECVBUF_SIZE);
        if (!rt->recvbuf)
            return AVERROR(ENOMEM);
    }

R
Ronald S. Bultje 已提交
1560 1561
    switch(rt->lower_transport) {
    default:
1562
#if CONFIG_RTSP_DEMUXER
R
Ronald S. Bultje 已提交
1563
    case RTSP_LOWER_TRANSPORT_TCP:
1564
        len = ff_rtsp_tcp_read_packet(s, &rtsp_st, rt->recvbuf, RECVBUF_SIZE);
R
Ronald S. Bultje 已提交
1565
        break;
1566
#endif
R
Ronald S. Bultje 已提交
1567 1568
    case RTSP_LOWER_TRANSPORT_UDP:
    case RTSP_LOWER_TRANSPORT_UDP_MULTICAST:
1569
        len = udp_read_packet(s, &rtsp_st, rt->recvbuf, RECVBUF_SIZE, wait_end);
R
Ronald S. Bultje 已提交
1570 1571 1572 1573
        if (len >=0 && rtsp_st->transport_priv && rt->transport == RTSP_TRANSPORT_RTP)
            rtp_check_and_send_back_rr(rtsp_st->transport_priv, len);
        break;
    }
1574 1575 1576 1577 1578 1579
    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 已提交
1580 1581 1582 1583
    if (len < 0)
        return len;
    if (len == 0)
        return AVERROR_EOF;
1584
    if (rt->transport == RTSP_TRANSPORT_RDT) {
1585
        ret = ff_rdt_parse_packet(rtsp_st->transport_priv, pkt, &rt->recvbuf, len);
1586
    } else {
1587
        ret = rtp_parse_packet(rtsp_st->transport_priv, pkt, &rt->recvbuf, len);
1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598
        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++) {
1599
                    RTPDemuxContext *rtpctx2 = rt->rtsp_streams[i]->transport_priv;
1600 1601 1602 1603 1604
                    if (rtpctx2 &&
                        rtpctx2->first_rtcp_ntp_time == AV_NOPTS_VALUE)
                        rtpctx2->first_rtcp_ntp_time = rtpctx->first_rtcp_ntp_time;
                }
            }
1605 1606 1607 1608 1609 1610 1611 1612 1613
            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;
            }
1614 1615
        }
    }
1616
end:
R
Ronald S. Bultje 已提交
1617 1618
    if (ret < 0)
        goto redo;
1619
    if (ret == 1)
R
Ronald S. Bultje 已提交
1620 1621 1622 1623 1624
        /* more packets may follow, so we save the RTP context */
        rt->cur_transport_priv = rtsp_st->transport_priv;

    return ret;
}
1625
#endif /* CONFIG_RTPDEC */
R
Ronald S. Bultje 已提交
1626

1627
#if CONFIG_SDP_DEMUXER
1628
static int sdp_probe(AVProbeData *p1)
1629
{
1630
    const char *p = p1->buf, *p_end = p1->buf + p1->buf_size;
1631

M
Martin Storsjö 已提交
1632
    /* we look for a line beginning "c=IN IP" */
1633
    while (p < p_end && *p != '\0') {
M
Martin Storsjö 已提交
1634 1635
        if (p + sizeof("c=IN IP") - 1 < p_end &&
            av_strstart(p, "c=IN IP", NULL))
1636
            return AVPROBE_SCORE_MAX / 2;
1637

1638
        while (p < p_end - 1 && *p != '\n') p++;
1639
        if (++p >= p_end)
1640 1641 1642 1643
            break;
        if (*p == '\r')
            p++;
    }
1644 1645 1646
    return 0;
}

1647
static int sdp_read_header(AVFormatContext *s, AVFormatParameters *ap)
1648
{
1649
    RTSPState *rt = s->priv_data;
1650 1651 1652 1653 1654
    RTSPStream *rtsp_st;
    int size, i, err;
    char *content;
    char url[1024];

1655 1656 1657
    if (!ff_network_init())
        return AVERROR(EIO);

1658 1659 1660
    /* read the whole sdp file */
    /* XXX: better loading */
    content = av_malloc(SDP_MAX_SIZE);
1661
    size = get_buffer(s->pb, content, SDP_MAX_SIZE - 1);
1662 1663 1664 1665 1666 1667
    if (size <= 0) {
        av_free(content);
        return AVERROR_INVALIDDATA;
    }
    content[size] ='\0';

1668
    ff_sdp_parse(s, content);
1669 1670 1671
    av_free(content);

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

M
Martin Storsjö 已提交
1676 1677
        getnameinfo((struct sockaddr*) &rtsp_st->sdp_ip, sizeof(rtsp_st->sdp_ip),
                    namebuf, sizeof(namebuf), NULL, 0, NI_NUMERICHOST);
1678
        ff_url_join(url, sizeof(url), "rtp", NULL,
M
Martin Storsjö 已提交
1679
                    namebuf, rtsp_st->sdp_port,
1680 1681
                    "?localport=%d&ttl=%d", rtsp_st->sdp_port,
                    rtsp_st->sdp_ttl);
1682
        if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
1683 1684 1685
            err = AVERROR_INVALIDDATA;
            goto fail;
        }
1686
        if ((err = rtsp_open_transport_ctx(s, rtsp_st)))
1687
            goto fail;
1688 1689
    }
    return 0;
1690
fail:
1691
    ff_rtsp_close_streams(s);
1692
    ff_network_close();
1693 1694 1695 1696 1697
    return err;
}

static int sdp_read_close(AVFormatContext *s)
{
1698
    ff_rtsp_close_streams(s);
1699
    ff_network_close();
1700 1701 1702
    return 0;
}

1703
AVInputFormat sdp_demuxer = {
1704
    "sdp",
1705
    NULL_IF_CONFIG_SMALL("SDP"),
1706 1707 1708
    sizeof(RTSPState),
    sdp_probe,
    sdp_read_header,
1709
    ff_rtsp_fetch_packet,
1710 1711
    sdp_read_close,
};
1712
#endif /* CONFIG_SDP_DEMUXER */
1713

1714
#if CONFIG_RTP_DEMUXER
1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738 1739 1740 1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 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
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,
1813
    ff_rtsp_fetch_packet,
1814 1815 1816
    sdp_read_close,
    .flags = AVFMT_NOFILE,
};
1817
#endif /* CONFIG_RTP_DEMUXER */
1818