rtsp.c 45.6 KB
Newer Older
1
/*
2
 * RTSP/SDP client
3 4
 * Copyright (c) 2002 Fabrice Bellard.
 *
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/avstring.h"
23 24
#include "avformat.h"

25
#include <sys/time.h>
26
#include <unistd.h> /* for select() prototype */
27
#include "network.h"
28
#include "rtsp.h"
29

R
Ryan Martell 已提交
30 31
#include "rtp_internal.h"

32
//#define DEBUG
F
Fabrice Bellard 已提交
33
//#define DEBUG_RTP_TCP
34

35 36 37 38 39 40
enum RTSPClientState {
    RTSP_STATE_IDLE,
    RTSP_STATE_PLAYING,
    RTSP_STATE_PAUSED,
};

41 42
typedef struct RTSPState {
    URLContext *rtsp_hd; /* RTSP TCP connexion handle */
43 44
    int nb_rtsp_streams;
    struct RTSPStream **rtsp_streams;
45

46 47
    enum RTSPClientState state;
    int64_t seek_timestamp;
48

F
Fabrice Bellard 已提交
49 50
    /* XXX: currently we use unbuffered input */
    //    ByteIOContext rtsp_gb;
51 52 53 54
    int seq;        /* RTSP command sequence number */
    char session_id[512];
    enum RTSPProtocol protocol;
    char last_reply[2048]; /* XXX: allocate ? */
55
    RTPDemuxContext *cur_rtp;
56 57 58
} RTSPState;

typedef struct RTSPStream {
59 60
    URLContext *rtp_handle; /* RTP stream handle */
    RTPDemuxContext *rtp_ctx; /* RTP parse context */
61

62
    int stream_index; /* corresponding stream index, if any. -1 if none (MPEG2TS case) */
63
    int interleaved_min, interleaved_max;  /* interleave ids, if TCP transport */
64 65 66 67 68 69
    char control_url[1024]; /* url for this stream (from SDP) */

    int sdp_port; /* port (from SDP content - not used in RTSP) */
    struct in_addr sdp_ip; /* IP address  (from SDP content - not used in RTSP) */
    int sdp_ttl;  /* IP TTL (from SDP content - not used in RTSP) */
    int sdp_payload_type; /* payload type - only used in SDP */
R
Romain Degez 已提交
70
    rtp_payload_data_t rtp_payload_data; /* rtp payload parsing infos from SDP */
R
Ryan Martell 已提交
71 72 73

    RTPDynamicProtocolHandler *dynamic_handler; ///< Only valid if it's a dynamic protocol. (This is the handler structure)
    void *dynamic_protocol_context; ///< Only valid if it's a dynamic protocol. (This is any private data associated with the dynamic protocol)
74 75
} RTSPStream;

76 77
static int rtsp_read_play(AVFormatContext *s);

78 79 80
/* XXX: currently, the only way to change the protocols consists in
   changing this variable */

R
Ronald S. Bultje 已提交
81
#if LIBAVFORMAT_VERSION_INT < (53 << 16)
R
Romain Degez 已提交
82
int rtsp_default_protocols = (1 << RTSP_PROTOCOL_RTP_UDP);
R
Ronald S. Bultje 已提交
83
#endif
F
Fabrice Bellard 已提交
84

85 86
static int rtsp_probe(AVProbeData *p)
{
87
    if (av_strstart(p->filename, "rtsp:", NULL))
88 89 90 91 92 93
        return AVPROBE_SCORE_MAX;
    return 0;
}

static int redir_isspace(int c)
{
94
    return c == ' ' || c == '\t' || c == '\n' || c == '\r';
95 96 97 98 99 100 101 102 103 104 105
}

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

106
static void get_word_sep(char *buf, int buf_size, const char *sep,
107 108 109 110 111 112
                         const char **pp)
{
    const char *p;
    char *q;

    p = *pp;
R
Romain Degez 已提交
113 114
    if (*p == '/')
        p++;
115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144
    skip_spaces(&p);
    q = buf;
    while (!strchr(sep, *p) && *p != '\0') {
        if ((q - buf) < buf_size - 1)
            *q++ = *p;
        p++;
    }
    if (buf_size > 0)
        *q = '\0';
    *pp = p;
}

static void get_word(char *buf, int buf_size, const char **pp)
{
    const char *p;
    char *q;

    p = *pp;
    skip_spaces(&p);
    q = buf;
    while (!redir_isspace(*p) && *p != '\0') {
        if ((q - buf) < buf_size - 1)
            *q++ = *p;
        p++;
    }
    if (buf_size > 0)
        *q = '\0';
    *pp = p;
}

145 146
/* parse the rtpmap description: <codec_name>/<clock_rate>[/<other
   params>] */
R
Ryan Martell 已提交
147
static int sdp_parse_rtpmap(AVCodecContext *codec, RTSPStream *rtsp_st, int payload_type, const char *p)
148 149
{
    char buf[256];
R
Romain Degez 已提交
150 151
    int i;
    AVCodec *c;
152
    const char *c_name;
153

R
Romain Degez 已提交
154 155
    /* Loop into AVRtpDynamicPayloadTypes[] and AVRtpPayloadTypes[] and
       see if we can handle this kind of payload */
156
    get_word_sep(buf, sizeof(buf), "/", &p);
R
Romain Degez 已提交
157
    if (payload_type >= RTP_PT_PRIVATE) {
R
Ryan Martell 已提交
158 159 160 161 162 163 164 165
        RTPDynamicProtocolHandler *handler= RTPFirstDynamicPayloadHandler;
        while(handler) {
            if (!strcmp(buf, handler->enc_name) && (codec->codec_type == handler->codec_type)) {
                codec->codec_id = handler->codec_id;
                rtsp_st->dynamic_handler= handler;
                if(handler->open) {
                    rtsp_st->dynamic_protocol_context= handler->open();
                }
R
Romain Degez 已提交
166 167
                break;
            }
R
Ryan Martell 已提交
168 169
            handler= handler->next;
        }
170
    } else {
R
Romain Degez 已提交
171 172
        /* We are in a standard case ( from http://www.iana.org/assignments/rtp-parameters) */
        /* search into AVRtpPayloadTypes[] */
173
        codec->codec_id = ff_rtp_codec_id(buf, codec->codec_type);
R
Romain Degez 已提交
174 175 176 177
    }

    c = avcodec_find_decoder(codec->codec_id);
    if (c && c->name)
178
        c_name = c->name;
R
Romain Degez 已提交
179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195
    else
        c_name = (char *)NULL;

    if (c_name) {
        get_word_sep(buf, sizeof(buf), "/", &p);
        i = atoi(buf);
        switch (codec->codec_type) {
            case CODEC_TYPE_AUDIO:
                av_log(codec, AV_LOG_DEBUG, " audio codec set to : %s\n", c_name);
                codec->sample_rate = RTSP_DEFAULT_AUDIO_SAMPLERATE;
                codec->channels = RTSP_DEFAULT_NB_AUDIO_CHANNELS;
                if (i > 0) {
                    codec->sample_rate = i;
                    get_word_sep(buf, sizeof(buf), "/", &p);
                    i = atoi(buf);
                    if (i > 0)
                        codec->channels = i;
196 197
                    // TODO: there is a bug here; if it is a mono stream, and less than 22000Hz, faad upconverts to stereo and twice the
                    //  frequency.  No problem, but the sample rate is being set here by the sdp line.  Upcoming patch forthcoming. (rdm)
R
Romain Degez 已提交
198 199 200 201 202 203 204 205 206 207 208
                }
                av_log(codec, AV_LOG_DEBUG, " audio samplerate set to : %i\n", codec->sample_rate);
                av_log(codec, AV_LOG_DEBUG, " audio channels set to : %i\n", codec->channels);
                break;
            case CODEC_TYPE_VIDEO:
                av_log(codec, AV_LOG_DEBUG, " video codec set to : %s\n", c_name);
                break;
            default:
                break;
        }
        return 0;
209
    }
R
Romain Degez 已提交
210 211

    return -1;
212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242
}

/* return the length and optionnaly the data */
static int hex_to_data(uint8_t *data, const char *p)
{
    int c, len, v;

    len = 0;
    v = 1;
    for(;;) {
        skip_spaces(&p);
        if (p == '\0')
            break;
        c = toupper((unsigned char)*p++);
        if (c >= '0' && c <= '9')
            c = c - '0';
        else if (c >= 'A' && c <= 'F')
            c = c - 'A' + 10;
        else
            break;
        v = (v << 4) | c;
        if (v & 0x100) {
            if (data)
                data[len] = v;
            len++;
            v = 1;
        }
    }
    return len;
}

R
Romain Degez 已提交
243 244 245 246
static void sdp_parse_fmtp_config(AVCodecContext *codec, char *attr, char *value)
{
    switch (codec->codec_id) {
        case CODEC_ID_MPEG4:
247
        case CODEC_ID_AAC:
R
Romain Degez 已提交
248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265
            if (!strcmp(attr, "config")) {
                /* decode the hexa encoded parameter */
                int len = hex_to_data(NULL, value);
                codec->extradata = av_mallocz(len + FF_INPUT_BUFFER_PADDING_SIZE);
                if (!codec->extradata)
                    return;
                codec->extradata_size = len;
                hex_to_data(codec->extradata, value);
            }
            break;
        default:
            break;
    }
    return;
}

typedef struct attrname_map
{
266
    const char *str;
R
Romain Degez 已提交
267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284
    uint16_t type;
    uint32_t offset;
} attrname_map_t;

/* All known fmtp parmeters and the corresping RTPAttrTypeEnum */
#define ATTR_NAME_TYPE_INT 0
#define ATTR_NAME_TYPE_STR 1
static attrname_map_t attr_names[]=
{
    {"SizeLength",       ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, sizelength)},
    {"IndexLength",      ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, indexlength)},
    {"IndexDeltaLength", ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, indexdeltalength)},
    {"profile-level-id", ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, profile_level_id)},
    {"StreamType",       ATTR_NAME_TYPE_INT, offsetof(rtp_payload_data_t, streamtype)},
    {"mode",             ATTR_NAME_TYPE_STR, offsetof(rtp_payload_data_t, mode)},
    {NULL, -1, -1},
};

285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303
/** parse the attribute line from the fmtp a line of an sdp resonse.  This is broken out as a function
* because it is used in rtp_h264.c, which is forthcoming.
*/
int rtsp_next_attr_and_value(const char **p, char *attr, int attr_size, char *value, int value_size)
{
    skip_spaces(p);
    if(**p)
    {
        get_word_sep(attr, attr_size, "=", p);
        if (**p == '=')
            (*p)++;
        get_word_sep(value, value_size, ";", p);
        if (**p == ';')
            (*p)++;
        return 1;
    }
    return 0;
}

R
Romain Degez 已提交
304 305
/* parse a SDP line and save stream attributes */
static void sdp_parse_fmtp(AVStream *st, const char *p)
306 307 308
{
    char attr[256];
    char value[4096];
R
Romain Degez 已提交
309 310 311
    int i;

    RTSPStream *rtsp_st = st->priv_data;
312
    AVCodecContext *codec = st->codec;
R
Romain Degez 已提交
313
    rtp_payload_data_t *rtp_payload_data = &rtsp_st->rtp_payload_data;
314 315

    /* loop on each attribute */
316 317
    while(rtsp_next_attr_and_value(&p, attr, sizeof(attr), value, sizeof(value)))
    {
318
        /* grab the codec extra_data from the config parameter of the fmtp line */
R
Romain Degez 已提交
319 320 321 322 323 324 325 326
        sdp_parse_fmtp_config(codec, attr, value);
        /* Looking for a known attribute */
        for (i = 0; attr_names[i].str; ++i) {
            if (!strcasecmp(attr, attr_names[i].str)) {
                if (attr_names[i].type == ATTR_NAME_TYPE_INT)
                    *(int *)((char *)rtp_payload_data + attr_names[i].offset) = atoi(value);
                else if (attr_names[i].type == ATTR_NAME_TYPE_STR)
                    *(char **)((char *)rtp_payload_data + attr_names[i].offset) = av_strdup(value);
327
            }
328 329 330 331
        }
    }
}

332 333 334 335 336 337 338 339 340
/** Parse a string \p in the form of Range:npt=xx-xx, and determine the start
 *  and end time.
 *  Used for seeking in the rtp stream.
 */
static void rtsp_parse_range_npt(const char *p, int64_t *start, int64_t *end)
{
    char buf[256];

    skip_spaces(&p);
341
    if (!av_stristart(p, "npt=", &p))
342 343 344 345 346 347 348 349 350 351 352 353 354 355 356 357
        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);
}

358 359 360 361 362 363 364
typedef struct SDPParseState {
    /* SDP only */
    struct in_addr default_ip;
    int default_ttl;
} SDPParseState;

static void sdp_parse_line(AVFormatContext *s, SDPParseState *s1,
365 366
                           int letter, const char *buf)
{
367
    RTSPState *rt = s->priv_data;
368 369
    char buf1[64], st_type[64];
    const char *p;
370
    int codec_type, payload_type, i;
371 372
    AVStream *st;
    RTSPStream *rtsp_st;
373 374 375
    struct in_addr sdp_ip;
    int ttl;

376 377 378 379 380 381
#ifdef DEBUG
    printf("sdp: %c='%s'\n", letter, buf);
#endif

    p = buf;
    switch(letter) {
382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407
    case 'c':
        get_word(buf1, sizeof(buf1), &p);
        if (strcmp(buf1, "IN") != 0)
            return;
        get_word(buf1, sizeof(buf1), &p);
        if (strcmp(buf1, "IP4") != 0)
            return;
        get_word_sep(buf1, sizeof(buf1), "/", &p);
        if (inet_aton(buf1, &sdp_ip) == 0)
            return;
        ttl = 16;
        if (*p == '/') {
            p++;
            get_word_sep(buf1, sizeof(buf1), "/", &p);
            ttl = atoi(buf1);
        }
        if (s->nb_streams == 0) {
            s1->default_ip = sdp_ip;
            s1->default_ttl = ttl;
        } else {
            st = s->streams[s->nb_streams - 1];
            rtsp_st = st->priv_data;
            rtsp_st->sdp_ip = sdp_ip;
            rtsp_st->sdp_ttl = ttl;
        }
        break;
408
    case 's':
M
Måns Rullgård 已提交
409
        av_strlcpy(s->title, p, sizeof(s->title));
410 411 412
        break;
    case 'i':
        if (s->nb_streams == 0) {
M
Måns Rullgård 已提交
413
            av_strlcpy(s->comment, p, sizeof(s->comment));
414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429
            break;
        }
        break;
    case 'm':
        /* new stream */
        get_word(st_type, sizeof(st_type), &p);
        if (!strcmp(st_type, "audio")) {
            codec_type = CODEC_TYPE_AUDIO;
        } else if (!strcmp(st_type, "video")) {
            codec_type = CODEC_TYPE_VIDEO;
        } else {
            return;
        }
        rtsp_st = av_mallocz(sizeof(RTSPStream));
        if (!rtsp_st)
            return;
430 431
        rtsp_st->stream_index = -1;
        dynarray_add(&rt->rtsp_streams, &rt->nb_rtsp_streams, rtsp_st);
432 433 434 435 436 437 438 439

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

441 442 443 444
        /* XXX: handle list of formats */
        get_word(buf1, sizeof(buf1), &p); /* format list */
        rtsp_st->sdp_payload_type = atoi(buf1);

445
        if (!strcmp(ff_rtp_enc_name(rtsp_st->sdp_payload_type), "MP2T")) {
446 447 448 449 450 451 452
            /* no corresponding stream */
        } else {
            st = av_new_stream(s, 0);
            if (!st)
                return;
            st->priv_data = rtsp_st;
            rtsp_st->stream_index = st->index;
453
            st->codec->codec_type = codec_type;
R
Romain Degez 已提交
454
            if (rtsp_st->sdp_payload_type < RTP_PT_PRIVATE) {
455
                /* if standard payload type, we can find the codec right now */
456
                rtp_get_codec_info(st->codec, rtsp_st->sdp_payload_type);
457 458
            }
        }
459
        /* put a default control url */
M
Måns Rullgård 已提交
460
        av_strlcpy(rtsp_st->control_url, s->filename, sizeof(rtsp_st->control_url));
461 462
        break;
    case 'a':
463
        if (av_strstart(p, "control:", &p) && s->nb_streams > 0) {
464 465 466 467
            char proto[32];
            /* get the control url */
            st = s->streams[s->nb_streams - 1];
            rtsp_st = st->priv_data;
468

469
            /* XXX: may need to add full url resolution */
470
            url_split(proto, sizeof(proto), NULL, 0, NULL, 0, NULL, NULL, 0, p);
471 472
            if (proto[0] == '\0') {
                /* relative control URL */
M
Måns Rullgård 已提交
473 474
                av_strlcat(rtsp_st->control_url, "/", sizeof(rtsp_st->control_url));
                av_strlcat(rtsp_st->control_url, p,   sizeof(rtsp_st->control_url));
475
            } else {
M
Måns Rullgård 已提交
476
                av_strlcpy(rtsp_st->control_url, p,   sizeof(rtsp_st->control_url));
477
            }
478
        } else if (av_strstart(p, "rtpmap:", &p)) {
479
            /* NOTE: rtpmap is only supported AFTER the 'm=' tag */
480
            get_word(buf1, sizeof(buf1), &p);
481 482 483 484 485
            payload_type = atoi(buf1);
            for(i = 0; i < s->nb_streams;i++) {
                st = s->streams[i];
                rtsp_st = st->priv_data;
                if (rtsp_st->sdp_payload_type == payload_type) {
R
Ryan Martell 已提交
486
                    sdp_parse_rtpmap(st->codec, rtsp_st, payload_type, p);
487 488
                }
            }
489
        } else if (av_strstart(p, "fmtp:", &p)) {
490
            /* NOTE: fmtp is only supported AFTER the 'a=rtpmap:xxx' tag */
491
            get_word(buf1, sizeof(buf1), &p);
492 493 494 495 496
            payload_type = atoi(buf1);
            for(i = 0; i < s->nb_streams;i++) {
                st = s->streams[i];
                rtsp_st = st->priv_data;
                if (rtsp_st->sdp_payload_type == payload_type) {
R
Ryan Martell 已提交
497 498 499 500 501
                    if(rtsp_st->dynamic_handler && rtsp_st->dynamic_handler->parse_sdp_a_line) {
                        if(!rtsp_st->dynamic_handler->parse_sdp_a_line(st, rtsp_st->dynamic_protocol_context, buf)) {
                            sdp_parse_fmtp(st, p);
                        }
                    } else {
502
                        sdp_parse_fmtp(st, p);
R
Ryan Martell 已提交
503
                    }
504 505
                }
            }
506
        } else if(av_strstart(p, "framesize:", &p)) {
507 508 509 510 511 512 513 514 515 516 517 518
            // let dynamic protocol handlers have a stab at the line.
            get_word(buf1, sizeof(buf1), &p);
            payload_type = atoi(buf1);
            for(i = 0; i < s->nb_streams;i++) {
                st = s->streams[i];
                rtsp_st = st->priv_data;
                if (rtsp_st->sdp_payload_type == payload_type) {
                    if(rtsp_st->dynamic_handler && rtsp_st->dynamic_handler->parse_sdp_a_line) {
                        rtsp_st->dynamic_handler->parse_sdp_a_line(st, rtsp_st->dynamic_protocol_context, buf);
                    }
                }
            }
519
        } else if(av_strstart(p, "range:", &p)) {
520 521 522 523 524 525
            int64_t start, end;

            // this is so that seeking on a streamed file can work.
            rtsp_parse_range_npt(p, &start, &end);
            s->start_time= start;
            s->duration= (end==AV_NOPTS_VALUE)?AV_NOPTS_VALUE:end-start; // AV_NOPTS_VALUE means live broadcast (and can't seek)
526 527 528 529 530
        }
        break;
    }
}

531
static int sdp_parse(AVFormatContext *s, const char *content)
532 533 534 535
{
    const char *p;
    int letter;
    char buf[1024], *q;
536
    SDPParseState sdp_parse_state, *s1 = &sdp_parse_state;
537

538
    memset(s1, 0, sizeof(SDPParseState));
539 540 541 542 543 544 545 546 547 548 549 550
    p = content;
    for(;;) {
        skip_spaces(&p);
        letter = *p;
        if (letter == '\0')
            break;
        p++;
        if (*p != '=')
            goto next_line;
        p++;
        /* get the content */
        q = buf;
F
Fabrice Bellard 已提交
551
        while (*p != '\n' && *p != '\r' && *p != '\0') {
552 553 554 555 556
            if ((q - buf) < sizeof(buf) - 1)
                *q++ = *p;
            p++;
        }
        *q = '\0';
557
        sdp_parse_line(s, s1, letter, buf);
558 559 560 561 562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595
    next_line:
        while (*p != '\n' && *p != '\0')
            p++;
        if (*p == '\n')
            p++;
    }
    return 0;
}

static void rtsp_parse_range(int *min_ptr, int *max_ptr, const char **pp)
{
    const char *p;
    int v;

    p = *pp;
    skip_spaces(&p);
    v = strtol(p, (char **)&p, 10);
    if (*p == '-') {
        p++;
        *min_ptr = v;
        v = strtol(p, (char **)&p, 10);
        *max_ptr = v;
    } else {
        *min_ptr = v;
        *max_ptr = v;
    }
    *pp = p;
}

/* XXX: only one transport specification is parsed */
static void rtsp_parse_transport(RTSPHeader *reply, const char *p)
{
    char transport_protocol[16];
    char profile[16];
    char lower_transport[16];
    char parameter[16];
    RTSPTransportField *th;
    char buf[256];
596

597
    reply->nb_transports = 0;
598

599 600 601 602 603 604 605
    for(;;) {
        skip_spaces(&p);
        if (*p == '\0')
            break;

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

606
        get_word_sep(transport_protocol, sizeof(transport_protocol),
607 608 609
                     "/", &p);
        if (*p == '/')
            p++;
610
        if (!strcasecmp (transport_protocol, "rtp")) {
611 612 613 614 615 616 617 618 619 620
            get_word_sep(profile, sizeof(profile), "/;,", &p);
            lower_transport[0] = '\0';
            /* rtp/avp/<protocol> */
            if (*p == '/') {
                p++;
                get_word_sep(lower_transport, sizeof(lower_transport),
                             ";,", &p);
                }
        } else if (!strcasecmp (transport_protocol, "x-pn-tng")) {
            /* x-pn-tng/<protocol> */
621 622
            get_word_sep(lower_transport, sizeof(lower_transport), "/;,", &p);
            profile[0] = '\0';
623
        }
F
Fabrice Bellard 已提交
624
        if (!strcasecmp(lower_transport, "TCP"))
625 626 627
            th->protocol = RTSP_PROTOCOL_RTP_TCP;
        else
            th->protocol = RTSP_PROTOCOL_RTP_UDP;
628

629 630 631 632 633 634 635 636 637 638 639 640 641
        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++;
642
                    rtsp_parse_range(&th->client_port_min,
643 644 645 646 647
                                     &th->client_port_max, &p);
                }
            } else if (!strcmp(parameter, "server_port")) {
                if (*p == '=') {
                    p++;
648
                    rtsp_parse_range(&th->server_port_min,
649 650 651 652 653
                                     &th->server_port_max, &p);
                }
            } else if (!strcmp(parameter, "interleaved")) {
                if (*p == '=') {
                    p++;
654
                    rtsp_parse_range(&th->interleaved_min,
655 656 657 658 659 660 661 662 663 664 665 666 667 668 669 670
                                     &th->interleaved_max, &p);
                }
            } else if (!strcmp(parameter, "multicast")) {
                if (th->protocol == RTSP_PROTOCOL_RTP_UDP)
                    th->protocol = RTSP_PROTOCOL_RTP_UDP_MULTICAST;
            } else if (!strcmp(parameter, "ttl")) {
                if (*p == '=') {
                    p++;
                    th->ttl = strtol(p, (char **)&p, 10);
                }
            } else if (!strcmp(parameter, "destination")) {
                struct in_addr ipaddr;

                if (*p == '=') {
                    p++;
                    get_word_sep(buf, sizeof(buf), ";,", &p);
671
                    if (inet_aton(buf, &ipaddr))
672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692
                        th->destination = ntohl(ipaddr.s_addr);
                }
            }
            while (*p != ';' && *p != '\0' && *p != ',')
                p++;
            if (*p == ';')
                p++;
        }
        if (*p == ',')
            p++;

        reply->nb_transports++;
    }
}

void rtsp_parse_line(RTSPHeader *reply, const char *buf)
{
    const char *p;

    /* NOTE: we do case independent match for broken servers */
    p = buf;
693
    if (av_stristart(p, "Session:", &p)) {
694
        get_word_sep(reply->session_id, sizeof(reply->session_id), ";", &p);
695
    } else if (av_stristart(p, "Content-Length:", &p)) {
696
        reply->content_length = strtol(p, NULL, 10);
697
    } else if (av_stristart(p, "Transport:", &p)) {
698
        rtsp_parse_transport(reply, p);
699
    } else if (av_stristart(p, "CSeq:", &p)) {
700
        reply->seq = strtol(p, NULL, 10);
701
    } else if (av_stristart(p, "Range:", &p)) {
702
        rtsp_parse_range_npt(p, &reply->range_start, &reply->range_end);
703 704 705
    }
}

706 707 708 709 710 711 712 713 714 715 716 717 718 719
static int url_readbuf(URLContext *h, unsigned char *buf, int size)
{
    int ret, len;

    len = 0;
    while (len < size) {
        ret = url_read(h, buf+len, size-len);
        if (ret < 1)
            return ret;
        len += ret;
    }
    return len;
}

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

727
    ret = url_readbuf(rt->rtsp_hd, buf, 3);
728 729
    if (ret != 3)
        return;
730
    len = AV_RB16(buf + 1);
731 732 733 734 735 736 737 738
#ifdef DEBUG
    printf("skipping RTP packet len=%d\n", len);
#endif
    /* skip payload */
    while (len > 0) {
        len1 = len;
        if (len1 > sizeof(buf))
            len1 = sizeof(buf);
739
        ret = url_readbuf(rt->rtsp_hd, buf, len1);
740 741 742 743 744
        if (ret != len1)
            return;
        len -= len1;
    }
}
745

746 747
static void rtsp_send_cmd(AVFormatContext *s,
                          const char *cmd, RTSPHeader *reply,
748 749 750 751 752 753 754 755 756 757 758 759
                          unsigned char **content_ptr)
{
    RTSPState *rt = s->priv_data;
    char buf[4096], buf1[1024], *q;
    unsigned char ch;
    const char *p;
    int content_length, line_count;
    unsigned char *content = NULL;

    memset(reply, 0, sizeof(RTSPHeader));

    rt->seq++;
M
Måns Rullgård 已提交
760
    av_strlcpy(buf, cmd, sizeof(buf));
F
Fabrice Bellard 已提交
761
    snprintf(buf1, sizeof(buf1), "CSeq: %d\r\n", rt->seq);
M
Måns Rullgård 已提交
762
    av_strlcat(buf, buf1, sizeof(buf));
763
    if (rt->session_id[0] != '\0' && !strstr(cmd, "\nIf-Match:")) {
F
Fabrice Bellard 已提交
764
        snprintf(buf1, sizeof(buf1), "Session: %s\r\n", rt->session_id);
M
Måns Rullgård 已提交
765
        av_strlcat(buf, buf1, sizeof(buf));
766
    }
M
Måns Rullgård 已提交
767
    av_strlcat(buf, "\r\n", sizeof(buf));
768 769 770 771 772 773 774 775 776 777 778
#ifdef DEBUG
    printf("Sending:\n%s--\n", buf);
#endif
    url_write(rt->rtsp_hd, buf, strlen(buf));

    /* parse reply (XXX: use buffers) */
    line_count = 0;
    rt->last_reply[0] = '\0';
    for(;;) {
        q = buf;
        for(;;) {
779
            if (url_readbuf(rt->rtsp_hd, &ch, 1) != 1)
780 781 782
                break;
            if (ch == '\n')
                break;
783 784 785 786
            if (ch == '$') {
                /* XXX: only parse it if first char on line ? */
                rtsp_skip_packet(s);
            } else if (ch != '\r') {
787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805
                if ((q - buf) < sizeof(buf) - 1)
                    *q++ = ch;
            }
        }
        *q = '\0';
#ifdef DEBUG
        printf("line='%s'\n", buf);
#endif
        /* test if last line */
        if (buf[0] == '\0')
            break;
        p = buf;
        if (line_count == 0) {
            /* get reply code */
            get_word(buf1, sizeof(buf1), &p);
            get_word(buf1, sizeof(buf1), &p);
            reply->status_code = atoi(buf1);
        } else {
            rtsp_parse_line(reply, p);
M
Måns Rullgård 已提交
806 807
            av_strlcat(rt->last_reply, p,    sizeof(rt->last_reply));
            av_strlcat(rt->last_reply, "\n", sizeof(rt->last_reply));
808 809 810
        }
        line_count++;
    }
811

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

815 816 817 818
    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);
819
        (void)url_readbuf(rt->rtsp_hd, content, content_length);
820 821 822 823
        content[content_length] = '\0';
    }
    if (content_ptr)
        *content_ptr = content;
824 825
    else
        av_free(content);
826 827 828
}


829 830 831 832 833 834 835 836 837 838 839 840 841
/* close and free RTSP streams */
static void rtsp_close_streams(RTSPState *rt)
{
    int i;
    RTSPStream *rtsp_st;

    for(i=0;i<rt->nb_rtsp_streams;i++) {
        rtsp_st = rt->rtsp_streams[i];
        if (rtsp_st) {
            if (rtsp_st->rtp_ctx)
                rtp_parse_close(rtsp_st->rtp_ctx);
            if (rtsp_st->rtp_handle)
                url_close(rtsp_st->rtp_handle);
R
Ryan Martell 已提交
842 843
            if (rtsp_st->dynamic_handler && rtsp_st->dynamic_protocol_context)
                rtsp_st->dynamic_handler->close(rtsp_st->dynamic_protocol_context);
844 845 846 847 848 849
        }
        av_free(rtsp_st);
    }
    av_free(rt->rtsp_streams);
}

850 851 852 853
/**
 * @returns 0 on success, <0 on error, 1 if protocol is unavailable.
 */
static int
854
make_setup_request (AVFormatContext *s, const char *host, int port, int protocol)
855 856
{
    RTSPState *rt = s->priv_data;
857
    int j, i, err;
858
    RTSPStream *rtsp_st;
859
    AVStream *st;
860 861
    RTSPHeader reply1, *reply = &reply1;
    char cmd[2048];
862

863 864 865
    /* for each stream, make the setup request */
    /* XXX: we assume the same server is used for the control of each
       RTSP stream */
R
Romain Degez 已提交
866 867

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

870
        rtsp_st = rt->rtsp_streams[i];
871 872 873 874 875

        /* compute available transports */
        transport[0] = '\0';

        /* RTP/UDP */
876
        if (protocol == RTSP_PROTOCOL_RTP_UDP) {
F
Fabrice Bellard 已提交
877 878 879
            char buf[256];

            /* first try in specified port range */
R
Romain Degez 已提交
880 881
            if (RTSP_RTP_PORT_MIN != 0) {
                while(j <= RTSP_RTP_PORT_MAX) {
882
                    snprintf(buf, sizeof(buf), "rtp://%s?localport=%d", host, j);
883
                    j += 2; /* we will use two port by rtp stream (rtp and rtcp) */
884
                    if (url_open(&rtsp_st->rtp_handle, buf, URL_RDWR) == 0) {
F
Fabrice Bellard 已提交
885
                        goto rtp_opened;
R
Romain Degez 已提交
886
                    }
F
Fabrice Bellard 已提交
887
                }
888
            }
F
Fabrice Bellard 已提交
889

R
Romain Degez 已提交
890 891 892 893 894 895
/*            then try on any port
**            if (url_open(&rtsp_st->rtp_handle, "rtp://", URL_RDONLY) < 0) {
**                err = AVERROR_INVALIDDATA;
**                goto fail;
**            }
*/
F
Fabrice Bellard 已提交
896 897

        rtp_opened:
898
            port = rtp_get_local_port(rtsp_st->rtp_handle);
899
            if (transport[0] != '\0')
M
Måns Rullgård 已提交
900
                av_strlcat(transport, ",", sizeof(transport));
901 902 903 904 905 906
            snprintf(transport + strlen(transport), sizeof(transport) - strlen(transport) - 1,
                     "RTP/AVP/UDP;unicast;client_port=%d-%d",
                     port, port + 1);
        }

        /* RTP/TCP */
907
        else if (protocol == RTSP_PROTOCOL_RTP_TCP) {
908
            if (transport[0] != '\0')
M
Måns Rullgård 已提交
909
                av_strlcat(transport, ",", sizeof(transport));
910 911 912 913
            snprintf(transport + strlen(transport), sizeof(transport) - strlen(transport) - 1,
                     "RTP/AVP/TCP");
        }

914
        else if (protocol == RTSP_PROTOCOL_RTP_UDP_MULTICAST) {
915
            if (transport[0] != '\0')
M
Måns Rullgård 已提交
916
                av_strlcat(transport, ",", sizeof(transport));
917
            snprintf(transport + strlen(transport),
918 919 920
                     sizeof(transport) - strlen(transport) - 1,
                     "RTP/AVP/UDP;multicast");
        }
921
        snprintf(cmd, sizeof(cmd),
F
Fabrice Bellard 已提交
922 923
                 "SETUP %s RTSP/1.0\r\n"
                 "Transport: %s\r\n",
924 925
                 rtsp_st->control_url, transport);
        rtsp_send_cmd(s, cmd, reply, NULL);
926 927 928
        if (reply->status_code == 461 /* Unsupported protocol */ && i == 0) {
            err = 1;
            goto fail;
929 930
        } else if (reply->status_code != RTSP_STATUS_OK ||
                   reply->nb_transports != 1) {
931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946
            err = AVERROR_INVALIDDATA;
            goto fail;
        }

        /* XXX: same protocol for all streams is required */
        if (i > 0) {
            if (reply->transports[0].protocol != rt->protocol) {
                err = AVERROR_INVALIDDATA;
                goto fail;
            }
        } else {
            rt->protocol = reply->transports[0].protocol;
        }

        /* close RTP connection if not choosen */
        if (reply->transports[0].protocol != RTSP_PROTOCOL_RTP_UDP &&
947
            (protocol == RTSP_PROTOCOL_RTP_UDP)) {
948 949
            url_close(rtsp_st->rtp_handle);
            rtsp_st->rtp_handle = NULL;
950 951 952 953 954 955 956
        }

        switch(reply->transports[0].protocol) {
        case RTSP_PROTOCOL_RTP_TCP:
            rtsp_st->interleaved_min = reply->transports[0].interleaved_min;
            rtsp_st->interleaved_max = reply->transports[0].interleaved_max;
            break;
957

958 959 960
        case RTSP_PROTOCOL_RTP_UDP:
            {
                char url[1024];
961

962
                /* XXX: also use address if specified */
963
                snprintf(url, sizeof(url), "rtp://%s:%d",
964
                         host, reply->transports[0].server_port_min);
965
                if (rtp_set_remote_url(rtsp_st->rtp_handle, url) < 0) {
966 967 968 969 970 971 972 973
                    err = AVERROR_INVALIDDATA;
                    goto fail;
                }
            }
            break;
        case RTSP_PROTOCOL_RTP_UDP_MULTICAST:
            {
                char url[1024];
974
                struct in_addr in;
975

976
                in.s_addr = htonl(reply->transports[0].destination);
977
                snprintf(url, sizeof(url), "rtp://%s:%d?ttl=%d",
978 979 980
                         inet_ntoa(in),
                         reply->transports[0].port_min,
                         reply->transports[0].ttl);
981
                if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
982 983 984 985 986 987
                    err = AVERROR_INVALIDDATA;
                    goto fail;
                }
            }
            break;
        }
988 989 990 991 992 993
        /* open the RTP context */
        st = NULL;
        if (rtsp_st->stream_index >= 0)
            st = s->streams[rtsp_st->stream_index];
        if (!st)
            s->ctx_flags |= AVFMTCTX_NOHEADER;
994
        rtsp_st->rtp_ctx = rtp_parse_open(s, st, rtsp_st->rtp_handle, rtsp_st->sdp_payload_type, &rtsp_st->rtp_payload_data);
R
Romain Degez 已提交
995

996
        if (!rtsp_st->rtp_ctx) {
997
            err = AVERROR(ENOMEM);
998
            goto fail;
R
Ryan Martell 已提交
999 1000 1001 1002 1003
        } else {
            if(rtsp_st->dynamic_handler) {
                rtsp_st->rtp_ctx->dynamic_protocol_context= rtsp_st->dynamic_protocol_context;
                rtsp_st->rtp_ctx->parse_packet= rtsp_st->dynamic_handler->parse_packet;
            }
1004
        }
1005 1006
    }

1007 1008 1009
    return 0;

fail:
1010 1011 1012 1013 1014 1015
    for (i=0; i<rt->nb_rtsp_streams; i++) {
        if (rt->rtsp_streams[i]->rtp_handle) {
            url_close(rt->rtsp_streams[i]->rtp_handle);
            rt->rtsp_streams[i]->rtp_handle = NULL;
        }
    }
1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057
    return err;
}

static int rtsp_read_header(AVFormatContext *s,
                            AVFormatParameters *ap)
{
    RTSPState *rt = s->priv_data;
    char host[1024], path[1024], tcpname[1024], cmd[2048], *option_list, *option;
    URLContext *rtsp_hd;
    int port, ret, err;
    RTSPHeader reply1, *reply = &reply1;
    unsigned char *content = NULL;
    int protocol_mask = 0;

    /* extract hostname and port */
    url_split(NULL, 0, NULL, 0,
              host, sizeof(host), &port, path, sizeof(path), s->filename);
    if (port < 0)
        port = RTSP_DEFAULT_PORT;

    /* search for options */
    option_list = strchr(path, '?');
    if (option_list) {
        /* remove the options from the path */
        *option_list++ = 0;
        while(option_list) {
            /* move the option pointer */
            option = option_list;
            option_list = strchr(option_list, '&');
            if (option_list)
                *(option_list++) = 0;
            /* handle the options */
            if (strcmp(option, "udp") == 0)
                protocol_mask = (1<< RTSP_PROTOCOL_RTP_UDP);
            else if (strcmp(option, "multicast") == 0)
                protocol_mask = (1<< RTSP_PROTOCOL_RTP_UDP_MULTICAST);
            else if (strcmp(option, "tcp") == 0)
                protocol_mask = (1<< RTSP_PROTOCOL_RTP_TCP);
        }
    }

    if (!protocol_mask)
1058
        protocol_mask = (1 << RTSP_PROTOCOL_RTP_LAST) - 1;
1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089

    /* open the tcp connexion */
    snprintf(tcpname, sizeof(tcpname), "tcp://%s:%d", host, port);
    if (url_open(&rtsp_hd, tcpname, URL_RDWR) < 0)
        return AVERROR(EIO);
    rt->rtsp_hd = rtsp_hd;
    rt->seq = 0;

    /* describe the stream */
    snprintf(cmd, sizeof(cmd),
             "DESCRIBE %s RTSP/1.0\r\n"
             "Accept: application/sdp\r\n",
             s->filename);
    rtsp_send_cmd(s, cmd, reply, &content);
    if (!content) {
        err = AVERROR_INVALIDDATA;
        goto fail;
    }
    if (reply->status_code != RTSP_STATUS_OK) {
        err = AVERROR_INVALIDDATA;
        goto fail;
    }

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

1090
    do {
1091
        int protocol = ff_log2_tab[protocol_mask & ~(protocol_mask - 1)];
1092 1093 1094

        err = make_setup_request(s, host, port, protocol);
        if (err < 0)
1095
            goto fail;
1096
        protocol_mask &= ~(1 << protocol);
1097
        if (protocol_mask == 0 && err == 1) {
1098
            err = AVERROR(FF_NETERROR(EPROTONOSUPPORT));
1099 1100 1101
            goto fail;
        }
    } while (err);
1102

1103 1104 1105
    rt->state = RTSP_STATE_IDLE;
    rt->seek_timestamp = 0; /* default is to start stream at position
                               zero */
1106
    if (ap->initial_pause) {
1107 1108 1109 1110
        /* do not start immediately */
    } else {
        if (rtsp_read_play(s) < 0) {
            err = AVERROR_INVALIDDATA;
1111 1112 1113 1114 1115
            goto fail;
        }
    }
    return 0;
 fail:
1116
    rtsp_close_streams(rt);
1117 1118 1119 1120 1121
    av_freep(&content);
    url_close(rt->rtsp_hd);
    return err;
}

1122 1123
static int tcp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
                           uint8_t *buf, int buf_size)
1124 1125
{
    RTSPState *rt = s->priv_data;
F
Fabrice Bellard 已提交
1126
    int id, len, i, ret;
1127 1128
    RTSPStream *rtsp_st;

F
Fabrice Bellard 已提交
1129 1130 1131
#ifdef DEBUG_RTP_TCP
    printf("tcp_read_packet:\n");
#endif
1132 1133
 redo:
    for(;;) {
1134
        ret = url_readbuf(rt->rtsp_hd, buf, 1);
F
Fabrice Bellard 已提交
1135 1136 1137 1138
#ifdef DEBUG_RTP_TCP
        printf("ret=%d c=%02x [%c]\n", ret, buf[0], buf[0]);
#endif
        if (ret != 1)
1139
            return -1;
F
Fabrice Bellard 已提交
1140
        if (buf[0] == '$')
1141 1142
            break;
    }
1143
    ret = url_readbuf(rt->rtsp_hd, buf, 3);
F
Fabrice Bellard 已提交
1144
    if (ret != 3)
1145
        return -1;
F
Fabrice Bellard 已提交
1146
    id = buf[0];
1147
    len = AV_RB16(buf + 1);
F
Fabrice Bellard 已提交
1148 1149 1150
#ifdef DEBUG_RTP_TCP
    printf("id=%d len=%d\n", id, len);
#endif
1151
    if (len > buf_size || len < 12)
1152 1153
        goto redo;
    /* get the data */
1154
    ret = url_readbuf(rt->rtsp_hd, buf, len);
F
Fabrice Bellard 已提交
1155
    if (ret != len)
1156
        return -1;
1157

1158
    /* find the matching stream */
1159 1160
    for(i = 0; i < rt->nb_rtsp_streams; i++) {
        rtsp_st = rt->rtsp_streams[i];
1161 1162
        if (id >= rtsp_st->interleaved_min &&
            id <= rtsp_st->interleaved_max)
1163 1164 1165 1166
            goto found;
    }
    goto redo;
 found:
1167 1168
    *prtsp_st = rtsp_st;
    return len;
1169 1170
}

1171
static int udp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
1172
                           uint8_t *buf, int buf_size)
1173
{
1174
    RTSPState *rt = s->priv_data;
1175 1176 1177 1178 1179 1180
    RTSPStream *rtsp_st;
    fd_set rfds;
    int fd1, fd2, fd_max, n, i, ret;
    struct timeval tv;

    for(;;) {
1181
        if (url_interrupt_cb())
1182
            return AVERROR(EINTR);
1183 1184
        FD_ZERO(&rfds);
        fd_max = -1;
1185 1186
        for(i = 0; i < rt->nb_rtsp_streams; i++) {
            rtsp_st = rt->rtsp_streams[i];
1187
            /* currently, we cannot probe RTCP handle because of blocking restrictions */
1188
            rtp_get_file_handles(rtsp_st->rtp_handle, &fd1, &fd2);
1189 1190 1191 1192 1193
            if (fd1 > fd_max)
                fd_max = fd1;
            FD_SET(fd1, &rfds);
        }
        tv.tv_sec = 0;
1194
        tv.tv_usec = 100 * 1000;
1195 1196
        n = select(fd_max + 1, &rfds, NULL, NULL, &tv);
        if (n > 0) {
1197 1198 1199
            for(i = 0; i < rt->nb_rtsp_streams; i++) {
                rtsp_st = rt->rtsp_streams[i];
                rtp_get_file_handles(rtsp_st->rtp_handle, &fd1, &fd2);
1200
                if (FD_ISSET(fd1, &rfds)) {
1201 1202 1203
                    ret = url_read(rtsp_st->rtp_handle, buf, buf_size);
                    if (ret > 0) {
                        *prtsp_st = rtsp_st;
1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215
                        return ret;
                    }
                }
            }
        }
    }
}

static int rtsp_read_packet(AVFormatContext *s,
                            AVPacket *pkt)
{
    RTSPState *rt = s->priv_data;
1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231
    RTSPStream *rtsp_st;
    int ret, len;
    uint8_t buf[RTP_MAX_PACKET_LENGTH];

    /* get next frames from the same RTP packet */
    if (rt->cur_rtp) {
        ret = rtp_parse_packet(rt->cur_rtp, pkt, NULL, 0);
        if (ret == 0) {
            rt->cur_rtp = NULL;
            return 0;
        } else if (ret == 1) {
            return 0;
        } else {
            rt->cur_rtp = NULL;
        }
    }
1232

1233 1234
    /* read next RTP packet */
 redo:
1235 1236 1237
    switch(rt->protocol) {
    default:
    case RTSP_PROTOCOL_RTP_TCP:
1238
        len = tcp_read_packet(s, &rtsp_st, buf, sizeof(buf));
1239 1240
        break;
    case RTSP_PROTOCOL_RTP_UDP:
1241 1242
    case RTSP_PROTOCOL_RTP_UDP_MULTICAST:
        len = udp_read_packet(s, &rtsp_st, buf, sizeof(buf));
1243
        if (len >=0 && rtsp_st->rtp_ctx)
1244
            rtp_check_and_send_back_rr(rtsp_st->rtp_ctx, len);
1245 1246
        break;
    }
1247
    if (len < 0)
1248
        return len;
1249 1250 1251 1252 1253 1254 1255 1256
    ret = rtp_parse_packet(rtsp_st->rtp_ctx, pkt, buf, len);
    if (ret < 0)
        goto redo;
    if (ret == 1) {
        /* more packets may follow, so we save the RTP context */
        rt->cur_rtp = rtsp_st->rtp_ctx;
    }
    return 0;
1257 1258
}

1259
static int rtsp_read_play(AVFormatContext *s)
1260
{
1261
    RTSPState *rt = s->priv_data;
1262 1263 1264
    RTSPHeader reply1, *reply = &reply1;
    char cmd[1024];

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

    if (rt->state == RTSP_STATE_PAUSED) {
1268
        snprintf(cmd, sizeof(cmd),
1269 1270 1271
                 "PLAY %s RTSP/1.0\r\n",
                 s->filename);
    } else {
1272
        snprintf(cmd, sizeof(cmd),
1273 1274 1275 1276 1277
                 "PLAY %s RTSP/1.0\r\n"
                 "Range: npt=%0.3f-\r\n",
                 s->filename,
                 (double)rt->seek_timestamp / AV_TIME_BASE);
    }
1278 1279 1280 1281
    rtsp_send_cmd(s, cmd, reply, NULL);
    if (reply->status_code != RTSP_STATUS_OK) {
        return -1;
    } else {
1282
        rt->state = RTSP_STATE_PLAYING;
1283 1284 1285 1286
        return 0;
    }
}

1287 1288
/* pause the stream */
static int rtsp_read_pause(AVFormatContext *s)
1289
{
1290
    RTSPState *rt = s->priv_data;
1291 1292 1293 1294
    RTSPHeader reply1, *reply = &reply1;
    char cmd[1024];

    rt = s->priv_data;
1295

1296 1297 1298
    if (rt->state != RTSP_STATE_PLAYING)
        return 0;

1299
    snprintf(cmd, sizeof(cmd),
1300
             "PAUSE %s RTSP/1.0\r\n",
1301 1302 1303 1304 1305
             s->filename);
    rtsp_send_cmd(s, cmd, reply, NULL);
    if (reply->status_code != RTSP_STATUS_OK) {
        return -1;
    } else {
1306
        rt->state = RTSP_STATE_PAUSED;
1307 1308 1309 1310
        return 0;
    }
}

1311
static int rtsp_read_seek(AVFormatContext *s, int stream_index,
1312
                          int64_t timestamp, int flags)
1313 1314
{
    RTSPState *rt = s->priv_data;
1315

1316
    rt->seek_timestamp = av_rescale_q(timestamp, s->streams[stream_index]->time_base, AV_TIME_BASE_Q);
1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331
    switch(rt->state) {
    default:
    case RTSP_STATE_IDLE:
        break;
    case RTSP_STATE_PLAYING:
        if (rtsp_read_play(s) != 0)
            return -1;
        break;
    case RTSP_STATE_PAUSED:
        rt->state = RTSP_STATE_IDLE;
        break;
    }
    return 0;
}

1332 1333 1334 1335 1336 1337
static int rtsp_read_close(AVFormatContext *s)
{
    RTSPState *rt = s->priv_data;
    RTSPHeader reply1, *reply = &reply1;
    char cmd[1024];

F
Fabrice Bellard 已提交
1338
#if 0
1339 1340 1341 1342
    /* NOTE: it is valid to flush the buffer here */
    if (rt->protocol == RTSP_PROTOCOL_RTP_TCP) {
        url_fclose(&rt->rtsp_gb);
    }
F
Fabrice Bellard 已提交
1343
#endif
1344
    snprintf(cmd, sizeof(cmd),
F
Fabrice Bellard 已提交
1345
             "TEARDOWN %s RTSP/1.0\r\n",
1346 1347 1348
             s->filename);
    rtsp_send_cmd(s, cmd, reply, NULL);

1349
    rtsp_close_streams(rt);
1350 1351 1352 1353
    url_close(rt->rtsp_hd);
    return 0;
}

1354
#ifdef CONFIG_RTSP_DEMUXER
1355
AVInputFormat rtsp_demuxer = {
1356 1357 1358 1359 1360 1361 1362
    "rtsp",
    "RTSP input format",
    sizeof(RTSPState),
    rtsp_probe,
    rtsp_read_header,
    rtsp_read_packet,
    rtsp_read_close,
1363
    rtsp_read_seek,
1364
    .flags = AVFMT_NOFILE,
1365 1366
    .read_play = rtsp_read_play,
    .read_pause = rtsp_read_pause,
1367
};
1368
#endif
1369

1370
static int sdp_probe(AVProbeData *p1)
1371
{
1372
    const char *p = p1->buf, *p_end = p1->buf + p1->buf_size;
1373 1374

    /* we look for a line beginning "c=IN IP4" */
1375
    while (p < p_end && *p != '\0') {
1376
        if (p + sizeof("c=IN IP4") - 1 < p_end && av_strstart(p, "c=IN IP4", NULL))
1377
            return AVPROBE_SCORE_MAX / 2;
1378 1379 1380

        while(p < p_end - 1 && *p != '\n') p++;
        if (++p >= p_end)
1381 1382 1383 1384
            break;
        if (*p == '\r')
            p++;
    }
1385 1386 1387 1388 1389 1390 1391 1392
    return 0;
}

#define SDP_MAX_SIZE 8192

static int sdp_read_header(AVFormatContext *s,
                           AVFormatParameters *ap)
{
1393
    RTSPState *rt = s->priv_data;
1394 1395 1396 1397
    RTSPStream *rtsp_st;
    int size, i, err;
    char *content;
    char url[1024];
1398
    AVStream *st;
1399 1400 1401 1402

    /* read the whole sdp file */
    /* XXX: better loading */
    content = av_malloc(SDP_MAX_SIZE);
1403
    size = get_buffer(s->pb, content, SDP_MAX_SIZE - 1);
1404 1405 1406 1407 1408 1409 1410 1411 1412 1413
    if (size <= 0) {
        av_free(content);
        return AVERROR_INVALIDDATA;
    }
    content[size] ='\0';

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

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

1417
        snprintf(url, sizeof(url), "rtp://%s:%d?localport=%d&ttl=%d",
1418
                 inet_ntoa(rtsp_st->sdp_ip),
1419
                 rtsp_st->sdp_port,
1420
                 rtsp_st->sdp_port,
1421
                 rtsp_st->sdp_ttl);
1422
        if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
1423 1424 1425
            err = AVERROR_INVALIDDATA;
            goto fail;
        }
1426 1427 1428 1429 1430 1431
        /* open the RTP context */
        st = NULL;
        if (rtsp_st->stream_index >= 0)
            st = s->streams[rtsp_st->stream_index];
        if (!st)
            s->ctx_flags |= AVFMTCTX_NOHEADER;
1432
        rtsp_st->rtp_ctx = rtp_parse_open(s, st, rtsp_st->rtp_handle, rtsp_st->sdp_payload_type, &rtsp_st->rtp_payload_data);
1433
        if (!rtsp_st->rtp_ctx) {
1434
            err = AVERROR(ENOMEM);
1435
            goto fail;
R
Ryan Martell 已提交
1436 1437 1438 1439 1440
        } else {
            if(rtsp_st->dynamic_handler) {
                rtsp_st->rtp_ctx->dynamic_protocol_context= rtsp_st->dynamic_protocol_context;
                rtsp_st->rtp_ctx->parse_packet= rtsp_st->dynamic_handler->parse_packet;
            }
1441
        }
1442 1443 1444
    }
    return 0;
 fail:
1445
    rtsp_close_streams(rt);
1446 1447 1448 1449 1450 1451
    return err;
}

static int sdp_read_packet(AVFormatContext *s,
                            AVPacket *pkt)
{
1452
    return rtsp_read_packet(s, pkt);
1453 1454 1455 1456
}

static int sdp_read_close(AVFormatContext *s)
{
1457 1458
    RTSPState *rt = s->priv_data;
    rtsp_close_streams(rt);
1459 1460 1461
    return 0;
}

1462 1463
#ifdef CONFIG_SDP_DEMUXER
AVInputFormat sdp_demuxer = {
1464 1465 1466 1467 1468 1469 1470 1471
    "sdp",
    "SDP",
    sizeof(RTSPState),
    sdp_probe,
    sdp_read_header,
    sdp_read_packet,
    sdp_read_close,
};
1472
#endif
1473

1474
#ifdef CONFIG_REDIR_DEMUXER
1475 1476 1477 1478 1479 1480 1481
/* dummy redirector format (used directly in av_open_input_file now) */
static int redir_probe(AVProbeData *pd)
{
    const char *p;
    p = pd->buf;
    while (redir_isspace(*p))
        p++;
1482 1483
    if (av_strstart(p, "http://", NULL) ||
        av_strstart(p, "rtsp://", NULL))
1484 1485 1486 1487
        return AVPROBE_SCORE_MAX;
    return 0;
}

1488
static int redir_read_header(AVFormatContext *s, AVFormatParameters *ap)
1489 1490 1491 1492
{
    char buf[4096], *q;
    int c;
    AVFormatContext *ic = NULL;
1493
    ByteIOContext *f = s->pb;
1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521

    /* parse each URL and try to open it */
    c = url_fgetc(f);
    while (c != URL_EOF) {
        /* skip spaces */
        for(;;) {
            if (!redir_isspace(c))
                break;
            c = url_fgetc(f);
        }
        if (c == URL_EOF)
            break;
        /* record url */
        q = buf;
        for(;;) {
            if (c == URL_EOF || redir_isspace(c))
                break;
            if ((q - buf) < sizeof(buf) - 1)
                *q++ = c;
            c = url_fgetc(f);
        }
        *q = '\0';
        //printf("URL='%s'\n", buf);
        /* try to open the media file */
        if (av_open_input_file(&ic, buf, NULL, 0, NULL) == 0)
            break;
    }
    if (!ic)
1522
        return AVERROR(EIO);
1523 1524 1525 1526 1527

    *s = *ic;
    url_fclose(f);

    return 0;
1528 1529
}

1530
AVInputFormat redir_demuxer = {
1531 1532 1533 1534
    "redir",
    "Redirector format",
    0,
    redir_probe,
1535
    redir_read_header,
1536 1537 1538
    NULL,
    NULL,
};
1539
#endif