libavformat/rtsp.c

   1 /*
   2  * RTSP/SDP client
   3  * Copyright (c) 2002 Fabrice Bellard
   4  *
   5  * This file is part of FFmpeg.
   6  *
   7  * FFmpeg is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * 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  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * 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 /* needed by inet_aton() */
  23 #define _SVID_SOURCE
  24
  25 #include "libavutil/base64.h"
  26 #include "libavutil/avstring.h"
  27 #include "libavutil/intreadwrite.h"
  28 #include "avformat.h"
  29
  30 #include <sys/time.h>
  31 #if HAVE_SYS_SELECT_H
  32 #include <sys/select.h>
  33 #endif
  34 #include <strings.h>
  35 #include "network.h"
  36 #include "rtsp.h"
  37
  38 #include "rtpdec.h"
  39 #include "rdt.h"
  40 #include "rtp_asf.h"
  41 #include "rtp_vorbis.h"
  42
  43 //#define DEBUG
  44 //#define DEBUG_RTP_TCP
  45
  46 static int rtsp_read_play(AVFormatContext *s);
  47
  48 #if LIBAVFORMAT_VERSION_INT < (53 << 16)
  49 int rtsp_default_protocols = (1 << RTSP_LOWER_TRANSPORT_UDP);
  50 #endif
  51
  52 static int rtsp_probe(AVProbeData *p)
  53 {
  54     if (av_strstart(p->filename, "rtsp:", NULL))
  55         return AVPROBE_SCORE_MAX;
  56     return 0;
  57 }
  58
  59 #define SPACE_CHARS " \t\r\n"
  60 /* we use memchr() instead of strchr() here because strchr() will return
  61  * the terminating '\0' of SPACE_CHARS instead of NULL if c is '\0'. */
  62 #define redir_isspace(c) memchr(SPACE_CHARS, c, 4)
  63 static void skip_spaces(const char **pp)
  64 {
  65     const char *p;
  66     p = *pp;
  67     while (redir_isspace(*p))
  68         p++;
  69     *pp = p;
  70 }
  71
  72 static void get_word_until_chars(char *buf, int buf_size,
  73                                  const char *sep, const char **pp)
  74 {
  75     const char *p;
  76     char *q;
  77
  78     p = *pp;
  79     skip_spaces(&p);
  80     q = buf;
  81     while (!strchr(sep, *p) && *p != '\0') {
  82         if ((q - buf) < buf_size - 1)
  83             *q++ = *p;
  84         p++;
  85     }
  86     if (buf_size > 0)
  87         *q = '\0';
  88     *pp = p;
  89 }
  90
  91 static void get_word_sep(char *buf, int buf_size, const char *sep,
  92                          const char **pp)
  93 {
  94     if (**pp == '/') (*pp)++;
  95     get_word_until_chars(buf, buf_size, sep, pp);
  96 }
  97
  98 static void get_word(char *buf, int buf_size, const char **pp)
  99 {
 100     get_word_until_chars(buf, buf_size, SPACE_CHARS, pp);
 101 }
 102
 103 /* parse the rtpmap description: <codec_name>/<clock_rate>[/<other
 104    params>] */
 105 static int sdp_parse_rtpmap(AVCodecContext *codec, RTSPStream *rtsp_st, int payload_type, const char *p)
 106 {
 107     char buf[256];
 108     int i;
 109     AVCodec *c;
 110     const char *c_name;
 111
 112     /* Loop into AVRtpDynamicPayloadTypes[] and AVRtpPayloadTypes[] and
 113        see if we can handle this kind of payload */
 114     get_word_sep(buf, sizeof(buf), "/", &p);
 115     if (payload_type >= RTP_PT_PRIVATE) {
 116         RTPDynamicProtocolHandler *handler= RTPFirstDynamicPayloadHandler;
 117         while(handler) {
 118             if (!strcasecmp(buf, handler->enc_name) && (codec->codec_type == handler->codec_type)) {
 119                 codec->codec_id = handler->codec_id;
 120                 rtsp_st->dynamic_handler= handler;
 121                 if(handler->open) {
 122                     rtsp_st->dynamic_protocol_context= handler->open();
 123                 }
 124                 break;
 125             }
 126             handler= handler->next;
 127         }
 128     } else {
 129         /* We are in a standard case ( from http://www.iana.org/assignments/rtp-parameters) */
 130         /* search into AVRtpPayloadTypes[] */
 131         codec->codec_id = ff_rtp_codec_id(buf, codec->codec_type);
 132     }
 133
 134     c = avcodec_find_decoder(codec->codec_id);
 135     if (c && c->name)
 136         c_name = c->name;
 137     else
 138         c_name = (char *)NULL;
 139
 140     if (c_name) {
 141         get_word_sep(buf, sizeof(buf), "/", &p);
 142         i = atoi(buf);
 143         switch (codec->codec_type) {
 144             case CODEC_TYPE_AUDIO:
 145                 av_log(codec, AV_LOG_DEBUG, " audio codec set to : %s\n", c_name);
 146                 codec->sample_rate = RTSP_DEFAULT_AUDIO_SAMPLERATE;
 147                 codec->channels = RTSP_DEFAULT_NB_AUDIO_CHANNELS;
 148                 if (i > 0) {
 149                     codec->sample_rate = i;
 150                     get_word_sep(buf, sizeof(buf), "/", &p);
 151                     i = atoi(buf);
 152                     if (i > 0)
 153                         codec->channels = i;
 154                     // TODO: there is a bug here; if it is a mono stream, and less than 22000Hz, faad upconverts to stereo and twice the
 155                     //  frequency.  No problem, but the sample rate is being set here by the sdp line.  Upcoming patch forthcoming. (rdm)
 156                 }
 157                 av_log(codec, AV_LOG_DEBUG, " audio samplerate set to : %i\n", codec->sample_rate);
 158                 av_log(codec, AV_LOG_DEBUG, " audio channels set to : %i\n", codec->channels);
 159                 break;
 160             case CODEC_TYPE_VIDEO:
 161                 av_log(codec, AV_LOG_DEBUG, " video codec set to : %s\n", c_name);
 162                 break;
 163             default:
 164                 break;
 165         }
 166         return 0;
 167     }
 168
 169     return -1;
 170 }
 171
 172 /* return the length and optionnaly the data */
 173 static int hex_to_data(uint8_t *data, const char *p)
 174 {
 175     int c, len, v;
 176
 177     len = 0;
 178     v = 1;
 179     for(;;) {
 180         skip_spaces(&p);
 181         if (*p == '\0')
 182             break;
 183         c = toupper((unsigned char)*p++);
 184         if (c >= '0' && c <= '9')
 185             c = c - '0';
 186         else if (c >= 'A' && c <= 'F')
 187             c = c - 'A' + 10;
 188         else
 189             break;
 190         v = (v << 4) | c;
 191         if (v & 0x100) {
 192             if (data)
 193                 data[len] = v;
 194             len++;
 195             v = 1;
 196         }
 197     }
 198     return len;
 199 }
 200
 201 static void sdp_parse_fmtp_config(AVCodecContext * codec, void *ctx,
 202                                   char *attr, char *value)
 203 {
 204     switch (codec->codec_id) {
 205         case CODEC_ID_MPEG4:
 206         case CODEC_ID_AAC:
 207             if (!strcmp(attr, "config")) {
 208                 /* decode the hexa encoded parameter */
 209                 int len = hex_to_data(NULL, value);
 210                 if (codec->extradata)
 211                     av_free(codec->extradata);
 212                 codec->extradata = av_mallocz(len + FF_INPUT_BUFFER_PADDING_SIZE);
 213                 if (!codec->extradata)
 214                     return;
 215                 codec->extradata_size = len;
 216                 hex_to_data(codec->extradata, value);
 217             }
 218             break;
 219         case CODEC_ID_VORBIS:
 220             ff_vorbis_parse_fmtp_config(codec, ctx, attr, value);
 221             break;
 222         default:
 223             break;
 224     }
 225     return;
 226 }
 227
 228 typedef struct {
 229     const char *str;
 230     uint16_t type;
 231     uint32_t offset;
 232 } AttrNameMap;
 233
 234 /* All known fmtp parmeters and the corresping RTPAttrTypeEnum */
 235 #define ATTR_NAME_TYPE_INT 0
 236 #define ATTR_NAME_TYPE_STR 1
 237 static const AttrNameMap attr_names[]=
 238 {
 239     {"SizeLength",       ATTR_NAME_TYPE_INT, offsetof(RTPPayloadData, sizelength)},
 240     {"IndexLength",      ATTR_NAME_TYPE_INT, offsetof(RTPPayloadData, indexlength)},
 241     {"IndexDeltaLength", ATTR_NAME_TYPE_INT, offsetof(RTPPayloadData, indexdeltalength)},
 242     {"profile-level-id", ATTR_NAME_TYPE_INT, offsetof(RTPPayloadData, profile_level_id)},
 243     {"StreamType",       ATTR_NAME_TYPE_INT, offsetof(RTPPayloadData, streamtype)},
 244     {"mode",             ATTR_NAME_TYPE_STR, offsetof(RTPPayloadData, mode)},
 245     {NULL, -1, -1},
 246 };
 247
 248 /** parse the attribute line from the fmtp a line of an sdp resonse.  This is broken out as a function
 249 * because it is used in rtp_h264.c, which is forthcoming.
 250 */
 251 int rtsp_next_attr_and_value(const char **p, char *attr, int attr_size, char *value, int value_size)
 252 {
 253     skip_spaces(p);
 254     if(**p) {
 255         get_word_sep(attr, attr_size, "=", p);
 256         if (**p == '=')
 257             (*p)++;
 258         get_word_sep(value, value_size, ";", p);
 259         if (**p == ';')
 260             (*p)++;
 261         return 1;
 262     }
 263     return 0;
 264 }
 265
 266 /* parse a SDP line and save stream attributes */
 267 static void sdp_parse_fmtp(AVStream *st, const char *p)
 268 {
 269     char attr[256];
 270     /* Vorbis setup headers can be up to 12KB and are sent base64
 271      * encoded, giving a 12KB * (4/3) = 16KB FMTP line. */
 272     char value[16384];
 273     int i;
 274
 275     RTSPStream *rtsp_st = st->priv_data;
 276     AVCodecContext *codec = st->codec;
 277     RTPPayloadData *rtp_payload_data = &rtsp_st->rtp_payload_data;
 278
 279     /* loop on each attribute */
 280     while(rtsp_next_attr_and_value(&p, attr, sizeof(attr), value, sizeof(value)))
 281     {
 282         /* grab the codec extra_data from the config parameter of the fmtp line */
 283         sdp_parse_fmtp_config(codec, rtsp_st->dynamic_protocol_context,
 284                               attr, value);
 285         /* Looking for a known attribute */
 286         for (i = 0; attr_names[i].str; ++i) {
 287             if (!strcasecmp(attr, attr_names[i].str)) {
 288                 if (attr_names[i].type == ATTR_NAME_TYPE_INT)
 289                     *(int *)((char *)rtp_payload_data + attr_names[i].offset) = atoi(value);
 290                 else if (attr_names[i].type == ATTR_NAME_TYPE_STR)
 291                     *(char **)((char *)rtp_payload_data + attr_names[i].offset) = av_strdup(value);
 292             }
 293         }
 294     }
 295 }
 296
 297 /** Parse a string \p in the form of Range:npt=xx-xx, and determine the start
 298  *  and end time.
 299  *  Used for seeking in the rtp stream.
 300  */
 301 static void rtsp_parse_range_npt(const char *p, int64_t *start, int64_t *end)
 302 {
 303     char buf[256];
 304
 305     skip_spaces(&p);
 306     if (!av_stristart(p, "npt=", &p))
 307         return;
 308
 309     *start = AV_NOPTS_VALUE;
 310     *end = AV_NOPTS_VALUE;
 311
 312     get_word_sep(buf, sizeof(buf), "-", &p);
 313     *start = parse_date(buf, 1);
 314     if (*p == '-') {
 315         p++;
 316         get_word_sep(buf, sizeof(buf), "-", &p);
 317         *end = parse_date(buf, 1);
 318     }
 319 //    av_log(NULL, AV_LOG_DEBUG, "Range Start: %lld\n", *start);
 320 //    av_log(NULL, AV_LOG_DEBUG, "Range End: %lld\n", *end);
 321 }
 322
 323 typedef struct SDPParseState {
 324     /* SDP only */
 325     struct in_addr default_ip;
 326     int default_ttl;
 327     int skip_media; ///< set if an unknown m= line occurs
 328 } SDPParseState;
 329
 330 static void sdp_parse_line(AVFormatContext *s, SDPParseState *s1,
 331                            int letter, const char *buf)
 332 {
 333     RTSPState *rt = s->priv_data;
 334     char buf1[64], st_type[64];
 335     const char *p;
 336     enum CodecType codec_type;
 337     int payload_type, i;
 338     AVStream *st;
 339     RTSPStream *rtsp_st;
 340     struct in_addr sdp_ip;
 341     int ttl;
 342
 343     dprintf(s, "sdp: %c='%s'\n", letter, buf);
 344
 345     p = buf;
 346     if (s1->skip_media && letter != 'm')
 347         return;
 348     switch(letter) {
 349     case 'c':
 350         get_word(buf1, sizeof(buf1), &p);
 351         if (strcmp(buf1, "IN") != 0)
 352             return;
 353         get_word(buf1, sizeof(buf1), &p);
 354         if (strcmp(buf1, "IP4") != 0)
 355             return;
 356         get_word_sep(buf1, sizeof(buf1), "/", &p);
 357         if (inet_aton(buf1, &sdp_ip) == 0)
 358             return;
 359         ttl = 16;
 360         if (*p == '/') {
 361             p++;
 362             get_word_sep(buf1, sizeof(buf1), "/", &p);
 363             ttl = atoi(buf1);
 364         }
 365         if (s->nb_streams == 0) {
 366             s1->default_ip = sdp_ip;
 367             s1->default_ttl = ttl;
 368         } else {
 369             st = s->streams[s->nb_streams - 1];
 370             rtsp_st = st->priv_data;
 371             rtsp_st->sdp_ip = sdp_ip;
 372             rtsp_st->sdp_ttl = ttl;
 373         }
 374         break;
 375     case 's':
 376         av_metadata_set(&s->metadata, "title", p);
 377         break;
 378     case 'i':
 379         if (s->nb_streams == 0) {
 380             av_metadata_set(&s->metadata, "comment", p);
 381             break;
 382         }
 383         break;
 384     case 'm':
 385         /* new stream */
 386         s1->skip_media = 0;
 387         get_word(st_type, sizeof(st_type), &p);
 388         if (!strcmp(st_type, "audio")) {
 389             codec_type = CODEC_TYPE_AUDIO;
 390         } else if (!strcmp(st_type, "video")) {
 391             codec_type = CODEC_TYPE_VIDEO;
 392         } else if (!strcmp(st_type, "application")) {
 393             codec_type = CODEC_TYPE_DATA;
 394         } else {
 395             s1->skip_media = 1;
 396             return;
 397         }
 398         rtsp_st = av_mallocz(sizeof(RTSPStream));
 399         if (!rtsp_st)
 400             return;
 401         rtsp_st->stream_index = -1;
 402         dynarray_add(&rt->rtsp_streams, &rt->nb_rtsp_streams, rtsp_st);
 403
 404         rtsp_st->sdp_ip = s1->default_ip;
 405         rtsp_st->sdp_ttl = s1->default_ttl;
 406
 407         get_word(buf1, sizeof(buf1), &p); /* port */
 408         rtsp_st->sdp_port = atoi(buf1);
 409
 410         get_word(buf1, sizeof(buf1), &p); /* protocol (ignored) */
 411
 412         /* XXX: handle list of formats */
 413         get_word(buf1, sizeof(buf1), &p); /* format list */
 414         rtsp_st->sdp_payload_type = atoi(buf1);
 415
 416         if (!strcmp(ff_rtp_enc_name(rtsp_st->sdp_payload_type), "MP2T")) {
 417             /* no corresponding stream */
 418         } else {
 419             st = av_new_stream(s, 0);
 420             if (!st)
 421                 return;
 422             st->priv_data = rtsp_st;
 423             rtsp_st->stream_index = st->index;
 424             st->codec->codec_type = codec_type;
 425             if (rtsp_st->sdp_payload_type < RTP_PT_PRIVATE) {
 426                 /* if standard payload type, we can find the codec right now */
 427                 ff_rtp_get_codec_info(st->codec, rtsp_st->sdp_payload_type);
 428             }
 429         }
 430         /* put a default control url */
 431         av_strlcpy(rtsp_st->control_url, s->filename, sizeof(rtsp_st->control_url));
 432         break;
 433     case 'a':
 434         if (av_strstart(p, "control:", &p) && s->nb_streams > 0) {
 435             char proto[32];
 436             /* get the control url */
 437             st = s->streams[s->nb_streams - 1];
 438             rtsp_st = st->priv_data;
 439
 440             /* XXX: may need to add full url resolution */
 441             url_split(proto, sizeof(proto), NULL, 0, NULL, 0, NULL, NULL, 0, p);
 442             if (proto[0] == '\0') {
 443                 /* relative control URL */
 444                 av_strlcat(rtsp_st->control_url, "/", sizeof(rtsp_st->control_url));
 445                 av_strlcat(rtsp_st->control_url, p,   sizeof(rtsp_st->control_url));
 446             } else {
 447                 av_strlcpy(rtsp_st->control_url, p,   sizeof(rtsp_st->control_url));
 448             }
 449         } else if (av_strstart(p, "rtpmap:", &p) && s->nb_streams > 0) {
 450             /* NOTE: rtpmap is only supported AFTER the 'm=' tag */
 451             get_word(buf1, sizeof(buf1), &p);
 452             payload_type = atoi(buf1);
 453             st = s->streams[s->nb_streams - 1];
 454             rtsp_st = st->priv_data;
 455             sdp_parse_rtpmap(st->codec, rtsp_st, payload_type, p);
 456         } else if (av_strstart(p, "fmtp:", &p)) {
 457             /* NOTE: fmtp is only supported AFTER the 'a=rtpmap:xxx' tag */
 458             get_word(buf1, sizeof(buf1), &p);
 459             payload_type = atoi(buf1);
 460             for(i = 0; i < s->nb_streams;i++) {
 461                 st = s->streams[i];
 462                 rtsp_st = st->priv_data;
 463                 if (rtsp_st->sdp_payload_type == payload_type) {
 464                     if(rtsp_st->dynamic_handler && rtsp_st->dynamic_handler->parse_sdp_a_line) {
 465                         if(!rtsp_st->dynamic_handler->parse_sdp_a_line(s, i, rtsp_st->dynamic_protocol_context, buf)) {
 466                             sdp_parse_fmtp(st, p);
 467                         }
 468                     } else {
 469                         sdp_parse_fmtp(st, p);
 470                     }
 471                 }
 472             }
 473         } else if(av_strstart(p, "framesize:", &p)) {
 474             // let dynamic protocol handlers have a stab at the line.
 475             get_word(buf1, sizeof(buf1), &p);
 476             payload_type = atoi(buf1);
 477             for(i = 0; i < s->nb_streams;i++) {
 478                 st = s->streams[i];
 479                 rtsp_st = st->priv_data;
 480                 if (rtsp_st->sdp_payload_type == payload_type) {
 481                     if(rtsp_st->dynamic_handler && rtsp_st->dynamic_handler->parse_sdp_a_line) {
 482                         rtsp_st->dynamic_handler->parse_sdp_a_line(s, i, rtsp_st->dynamic_protocol_context, buf);
 483                     }
 484                 }
 485             }
 486         } else if(av_strstart(p, "range:", &p)) {
 487             int64_t start, end;
 488
 489             // this is so that seeking on a streamed file can work.
 490             rtsp_parse_range_npt(p, &start, &end);
 491             s->start_time= start;
 492             s->duration= (end==AV_NOPTS_VALUE)?AV_NOPTS_VALUE:end-start; // AV_NOPTS_VALUE means live broadcast (and can't seek)
 493         } else if (av_strstart(p, "IsRealDataType:integer;",&p)) {
 494             if (atoi(p) == 1)
 495                 rt->transport = RTSP_TRANSPORT_RDT;
 496         } else {
 497             if (rt->server_type == RTSP_SERVER_WMS)
 498                 ff_wms_parse_sdp_a_line(s, p);
 499             if (s->nb_streams > 0) {
 500                 if (rt->server_type == RTSP_SERVER_REAL)
 501                     ff_real_parse_sdp_a_line(s, s->nb_streams - 1, p);
 502
 503                 rtsp_st = s->streams[s->nb_streams - 1]->priv_data;
 504                 if (rtsp_st->dynamic_handler &&
 505                     rtsp_st->dynamic_handler->parse_sdp_a_line)
 506                     rtsp_st->dynamic_handler->parse_sdp_a_line(s, s->nb_streams - 1,
 507                         rtsp_st->dynamic_protocol_context, buf);
 508             }
 509         }
 510         break;
 511     }
 512 }
 513
 514 static int sdp_parse(AVFormatContext *s, const char *content)
 515 {
 516     const char *p;
 517     int letter;
 518     /* Some SDP lines, particularly for Realmedia or ASF RTSP streams,
 519      * contain long SDP lines containing complete ASF Headers (several
 520      * kB) or arrays of MDPR (RM stream descriptor) headers plus
 521      * "rulebooks" describing their properties. Therefore, the SDP line
 522      * buffer is large.
 523      *
 524      * The Vorbis FMTP line can be up to 16KB - see sdp_parse_fmtp. */
 525     char buf[16384], *q;
 526     SDPParseState sdp_parse_state, *s1 = &sdp_parse_state;
 527
 528     memset(s1, 0, sizeof(SDPParseState));
 529     p = content;
 530     for(;;) {
 531         skip_spaces(&p);
 532         letter = *p;
 533         if (letter == '\0')
 534             break;
 535         p++;
 536         if (*p != '=')
 537             goto next_line;
 538         p++;
 539         /* get the content */
 540         q = buf;
 541         while (*p != '\n' && *p != '\r' && *p != '\0') {
 542             if ((q - buf) < sizeof(buf) - 1)
 543                 *q++ = *p;
 544             p++;
 545         }
 546         *q = '\0';
 547         sdp_parse_line(s, s1, letter, buf);
 548     next_line:
 549         while (*p != '\n' && *p != '\0')
 550             p++;
 551         if (*p == '\n')
 552             p++;
 553     }
 554     return 0;
 555 }
 556
 557 static void rtsp_parse_range(int *min_ptr, int *max_ptr, const char **pp)
 558 {
 559     const char *p;
 560     int v;
 561
 562     p = *pp;
 563     skip_spaces(&p);
 564     v = strtol(p, (char **)&p, 10);
 565     if (*p == '-') {
 566         p++;
 567         *min_ptr = v;
 568         v = strtol(p, (char **)&p, 10);
 569         *max_ptr = v;
 570     } else {
 571         *min_ptr = v;
 572         *max_ptr = v;
 573     }
 574     *pp = p;
 575 }
 576
 577 /* XXX: only one transport specification is parsed */
 578 static void rtsp_parse_transport(RTSPMessageHeader *reply, const char *p)
 579 {
 580     char transport_protocol[16];
 581     char profile[16];
 582     char lower_transport[16];
 583     char parameter[16];
 584     RTSPTransportField *th;
 585     char buf[256];
 586
 587     reply->nb_transports = 0;
 588
 589     for(;;) {
 590         skip_spaces(&p);
 591         if (*p == '\0')
 592             break;
 593
 594         th = &reply->transports[reply->nb_transports];
 595
 596         get_word_sep(transport_protocol, sizeof(transport_protocol),
 597                      "/", &p);
 598         if (!strcasecmp (transport_protocol, "rtp")) {
 599             get_word_sep(profile, sizeof(profile), "/;,", &p);
 600             lower_transport[0] = '\0';
 601             /* rtp/avp/<protocol> */
 602             if (*p == '/') {
 603                 get_word_sep(lower_transport, sizeof(lower_transport),
 604                              ";,", &p);
 605             }
 606             th->transport = RTSP_TRANSPORT_RTP;
 607         } else if (!strcasecmp (transport_protocol, "x-pn-tng") ||
 608                    !strcasecmp (transport_protocol, "x-real-rdt")) {
 609             /* x-pn-tng/<protocol> */
 610             get_word_sep(lower_transport, sizeof(lower_transport), "/;,", &p);
 611             profile[0] = '\0';
 612             th->transport = RTSP_TRANSPORT_RDT;
 613         }
 614         if (!strcasecmp(lower_transport, "TCP"))
 615             th->lower_transport = RTSP_LOWER_TRANSPORT_TCP;
 616         else
 617             th->lower_transport = RTSP_LOWER_TRANSPORT_UDP;
 618
 619         if (*p == ';')
 620             p++;
 621         /* get each parameter */
 622         while (*p != '\0' && *p != ',') {
 623             get_word_sep(parameter, sizeof(parameter), "=;,", &p);
 624             if (!strcmp(parameter, "port")) {
 625                 if (*p == '=') {
 626                     p++;
 627                     rtsp_parse_range(&th->port_min, &th->port_max, &p);
 628                 }
 629             } else if (!strcmp(parameter, "client_port")) {
 630                 if (*p == '=') {
 631                     p++;
 632                     rtsp_parse_range(&th->client_port_min,
 633                                      &th->client_port_max, &p);
 634                 }
 635             } else if (!strcmp(parameter, "server_port")) {
 636                 if (*p == '=') {
 637                     p++;
 638                     rtsp_parse_range(&th->server_port_min,
 639                                      &th->server_port_max, &p);
 640                 }
 641             } else if (!strcmp(parameter, "interleaved")) {
 642                 if (*p == '=') {
 643                     p++;
 644                     rtsp_parse_range(&th->interleaved_min,
 645                                      &th->interleaved_max, &p);
 646                 }
 647             } else if (!strcmp(parameter, "multicast")) {
 648                 if (th->lower_transport == RTSP_LOWER_TRANSPORT_UDP)
 649                     th->lower_transport = RTSP_LOWER_TRANSPORT_UDP_MULTICAST;
 650             } else if (!strcmp(parameter, "ttl")) {
 651                 if (*p == '=') {
 652                     p++;
 653                     th->ttl = strtol(p, (char **)&p, 10);
 654                 }
 655             } else if (!strcmp(parameter, "destination")) {
 656                 struct in_addr ipaddr;
 657
 658                 if (*p == '=') {
 659                     p++;
 660                     get_word_sep(buf, sizeof(buf), ";,", &p);
 661                     if (inet_aton(buf, &ipaddr))
 662                         th->destination = ntohl(ipaddr.s_addr);
 663                 }
 664             }
 665             while (*p != ';' && *p != '\0' && *p != ',')
 666                 p++;
 667             if (*p == ';')
 668                 p++;
 669         }
 670         if (*p == ',')
 671             p++;
 672
 673         reply->nb_transports++;
 674     }
 675 }
 676
 677 void rtsp_parse_line(RTSPMessageHeader *reply, const char *buf)
 678 {
 679     const char *p;
 680
 681     /* NOTE: we do case independent match for broken servers */
 682     p = buf;
 683     if (av_stristart(p, "Session:", &p)) {
 684         int t;
 685         get_word_sep(reply->session_id, sizeof(reply->session_id), ";", &p);
 686         if (av_stristart(p, ";timeout=", &p) &&
 687             (t = strtol(p, NULL, 10)) > 0) {
 688             reply->timeout = t;
 689         }
 690     } else if (av_stristart(p, "Content-Length:", &p)) {
 691         reply->content_length = strtol(p, NULL, 10);
 692     } else if (av_stristart(p, "Transport:", &p)) {
 693         rtsp_parse_transport(reply, p);
 694     } else if (av_stristart(p, "CSeq:", &p)) {
 695         reply->seq = strtol(p, NULL, 10);
 696     } else if (av_stristart(p, "Range:", &p)) {
 697         rtsp_parse_range_npt(p, &reply->range_start, &reply->range_end);
 698     } else if (av_stristart(p, "RealChallenge1:", &p)) {
 699         skip_spaces(&p);
 700         av_strlcpy(reply->real_challenge, p, sizeof(reply->real_challenge));
 701     } else if (av_stristart(p, "Server:", &p)) {
 702         skip_spaces(&p);
 703         av_strlcpy(reply->server, p, sizeof(reply->server));
 704     } else if (av_stristart(p, "Notice:", &p) ||
 705                av_stristart(p, "X-Notice:", &p)) {
 706         reply->notice = strtol(p, NULL, 10);
 707     }
 708 }
 709
 710 /* skip a RTP/TCP interleaved packet */
 711 static void rtsp_skip_packet(AVFormatContext *s)
 712 {
 713     RTSPState *rt = s->priv_data;
 714     int ret, len, len1;
 715     uint8_t buf[1024];
 716
 717     ret = url_read_complete(rt->rtsp_hd, buf, 3);
 718     if (ret != 3)
 719         return;
 720     len = AV_RB16(buf + 1);
 721
 722     dprintf(s, "skipping RTP packet len=%d\n", len);
 723
 724     /* skip payload */
 725     while (len > 0) {
 726         len1 = len;
 727         if (len1 > sizeof(buf))
 728             len1 = sizeof(buf);
 729         ret = url_read_complete(rt->rtsp_hd, buf, len1);
 730         if (ret != len1)
 731             return;
 732         len -= len1;
 733     }
 734 }
 735
 736 /**
 737  * Read a RTSP message from the server, or prepare to read data
 738  * packets if we're reading data interleaved over the TCP/RTSP
 739  * connection as well.
 740  *
 741  * @param s RTSP demuxer context
 742  * @param reply pointer where the RTSP message header will be stored
 743  * @param content_ptr pointer where the RTSP message body, if any, will
 744  *                    be stored (length is in \p reply)
 745  * @param return_on_interleaved_data whether the function may return if we
 746  *                   encounter a data marker ('$'), which precedes data
 747  *                   packets over interleaved TCP/RTSP connections. If this
 748  *                   is set, this function will return 1 after encountering
 749  *                   a '$'. If it is not set, the function will skip any
 750  *                   data packets (if they are encountered), until a reply
 751  *                   has been fully parsed. If no more data is available
 752  *                   without parsing a reply, it will return an error.
 753  *
 754  * @returns 1 if a data packets is ready to be received, -1 on error,
 755  *          and 0 on success.
 756  */
 757 static int
 758 rtsp_read_reply (AVFormatContext *s, RTSPMessageHeader *reply,
 759                  unsigned char **content_ptr, int return_on_interleaved_data)
 760 {
 761     RTSPState *rt = s->priv_data;
 762     char buf[4096], buf1[1024], *q;
 763     unsigned char ch;
 764     const char *p;
 765     int ret, content_length, line_count = 0;
 766     unsigned char *content = NULL;
 767
 768     memset(reply, 0, sizeof(*reply));
 769
 770     /* parse reply (XXX: use buffers) */
 771     rt->last_reply[0] = '\0';
 772     for(;;) {
 773         q = buf;
 774         for(;;) {
 775             ret = url_read_complete(rt->rtsp_hd, &ch, 1);
 776 #ifdef DEBUG_RTP_TCP
 777             dprintf(s, "ret=%d c=%02x [%c]\n", ret, ch, ch);
 778 #endif
 779             if (ret != 1)
 780                 return -1;
 781             if (ch == '\n')
 782                 break;
 783             if (ch == '$') {
 784                 /* XXX: only parse it if first char on line ? */
 785                 if (return_on_interleaved_data) {
 786                     return 1;
 787                 } else
 788                 rtsp_skip_packet(s);
 789             } else if (ch != '\r') {
 790                 if ((q - buf) < sizeof(buf) - 1)
 791                     *q++ = ch;
 792             }
 793         }
 794         *q = '\0';
 795
 796         dprintf(s, "line='%s'\n", buf);
 797
 798         /* test if last line */
 799         if (buf[0] == '\0')
 800             break;
 801         p = buf;
 802         if (line_count == 0) {
 803             /* get reply code */
 804             get_word(buf1, sizeof(buf1), &p);
 805             get_word(buf1, sizeof(buf1), &p);
 806             reply->status_code = atoi(buf1);
 807         } else {
 808             rtsp_parse_line(reply, p);
 809             av_strlcat(rt->last_reply, p,    sizeof(rt->last_reply));
 810             av_strlcat(rt->last_reply, "\n", sizeof(rt->last_reply));
 811         }
 812         line_count++;
 813     }
 814
 815     if (rt->session_id[0] == '\0' && reply->session_id[0] != '\0')
 816         av_strlcpy(rt->session_id, reply->session_id, sizeof(rt->session_id));
 817
 818     content_length = reply->content_length;
 819     if (content_length > 0) {
 820         /* leave some room for a trailing '\0' (useful for simple parsing) */
 821         content = av_malloc(content_length + 1);
 822         (void)url_read_complete(rt->rtsp_hd, content, content_length);
 823         content[content_length] = '\0';
 824     }
 825     if (content_ptr)
 826         *content_ptr = content;
 827     else
 828         av_free(content);
 829
 830     /* EOS */
 831     if (reply->notice == 2101 /* End-of-Stream Reached */      ||
 832         reply->notice == 2104 /* Start-of-Stream Reached */    ||
 833         reply->notice == 2306 /* Continuous Feed Terminated */)
 834         rt->state = RTSP_STATE_IDLE;
 835     else if (reply->notice >= 4400 && reply->notice < 5500)
 836         return AVERROR(EIO); /* data or server error */
 837     else if (reply->notice == 2401 /* Ticket Expired */ ||
 838              (reply->notice >= 5500 && reply->notice < 5600) /* end of term */ )
 839         return AVERROR(EPERM);
 840
 841     return 0;
 842 }
 843
 844 static void rtsp_send_cmd_async (AVFormatContext *s,
 845                           const char *cmd, RTSPMessageHeader *reply,
 846                           unsigned char **content_ptr)
 847 {
 848     RTSPState *rt = s->priv_data;
 849     char buf[4096], buf1[1024];
 850
 851     rt->seq++;
 852     av_strlcpy(buf, cmd, sizeof(buf));
 853     snprintf(buf1, sizeof(buf1), "CSeq: %d\r\n", rt->seq);
 854     av_strlcat(buf, buf1, sizeof(buf));
 855     if (rt->session_id[0] != '\0' && !strstr(cmd, "\nIf-Match:")) {
 856         snprintf(buf1, sizeof(buf1), "Session: %s\r\n", rt->session_id);
 857         av_strlcat(buf, buf1, sizeof(buf));
 858     }
 859     if (rt->auth_b64)
 860         av_strlcatf(buf, sizeof(buf),
 861                     "Authorization: Basic %s\r\n",
 862                     rt->auth_b64);
 863     av_strlcat(buf, "\r\n", sizeof(buf));
 864
 865     dprintf(s, "Sending:\n%s--\n", buf);
 866
 867     url_write(rt->rtsp_hd, buf, strlen(buf));
 868     rt->last_cmd_time = av_gettime();
 869 }
 870
 871 static void rtsp_send_cmd (AVFormatContext *s,
 872                            const char *cmd, RTSPMessageHeader *reply,
 873                            unsigned char **content_ptr)
 874 {
 875     rtsp_send_cmd_async(s, cmd, reply, content_ptr);
 876
 877     rtsp_read_reply(s, reply, content_ptr, 0);
 878 }
 879
 880
 881 /* close and free RTSP streams */
 882 static void rtsp_close_streams(RTSPState *rt)
 883 {
 884     int i;
 885     RTSPStream *rtsp_st;
 886
 887     for(i=0;i<rt->nb_rtsp_streams;i++) {
 888         rtsp_st = rt->rtsp_streams[i];
 889         if (rtsp_st) {
 890             if (rtsp_st->transport_priv) {
 891                 if (rt->transport == RTSP_TRANSPORT_RDT)
 892                     ff_rdt_parse_close(rtsp_st->transport_priv);
 893                 else
 894                     rtp_parse_close(rtsp_st->transport_priv);
 895             }
 896             if (rtsp_st->rtp_handle)
 897                 url_close(rtsp_st->rtp_handle);
 898             if (rtsp_st->dynamic_handler && rtsp_st->dynamic_protocol_context)
 899                 rtsp_st->dynamic_handler->close(rtsp_st->dynamic_protocol_context);
 900         }
 901     }
 902     av_free(rt->rtsp_streams);
 903     if (rt->asf_ctx) {
 904         av_close_input_stream (rt->asf_ctx);
 905         rt->asf_ctx = NULL;
 906     }
 907     av_freep(&rt->auth_b64);
 908 }
 909
 910 static int
 911 rtsp_open_transport_ctx(AVFormatContext *s, RTSPStream *rtsp_st)
 912 {
 913     RTSPState *rt = s->priv_data;
 914     AVStream *st = NULL;
 915
 916     /* open the RTP context */
 917     if (rtsp_st->stream_index >= 0)
 918         st = s->streams[rtsp_st->stream_index];
 919     if (!st)
 920         s->ctx_flags |= AVFMTCTX_NOHEADER;
 921
 922     if (rt->transport == RTSP_TRANSPORT_RDT)
 923         rtsp_st->transport_priv = ff_rdt_parse_open(s, st->index,
 924                                             rtsp_st->dynamic_protocol_context,
 925                                             rtsp_st->dynamic_handler);
 926     else
 927         rtsp_st->transport_priv = rtp_parse_open(s, st, rtsp_st->rtp_handle,
 928                                          rtsp_st->sdp_payload_type,
 929                                          &rtsp_st->rtp_payload_data);
 930
 931     if (!rtsp_st->transport_priv) {
 932          return AVERROR(ENOMEM);
 933     } else if (rt->transport != RTSP_TRANSPORT_RDT) {
 934         if(rtsp_st->dynamic_handler) {
 935             rtp_parse_set_dynamic_protocol(rtsp_st->transport_priv,
 936                                            rtsp_st->dynamic_protocol_context,
 937                                            rtsp_st->dynamic_handler);
 938         }
 939     }
 940
 941     return 0;
 942 }
 943
 944 /**
 945  * @returns 0 on success, <0 on error, 1 if protocol is unavailable.
 946  */
 947 static int
 948 make_setup_request (AVFormatContext *s, const char *host, int port,
 949                     int lower_transport, const char *real_challenge)
 950 {
 951     RTSPState *rt = s->priv_data;
 952     int rtx, j, i, err, interleave = 0;
 953     RTSPStream *rtsp_st;
 954     RTSPMessageHeader reply1, *reply = &reply1;
 955     char cmd[2048];
 956     const char *trans_pref;
 957
 958     if (rt->transport == RTSP_TRANSPORT_RDT)
 959         trans_pref = "x-pn-tng";
 960     else
 961         trans_pref = "RTP/AVP";
 962
 963     /* default timeout: 1 minute */
 964     rt->timeout = 60;
 965
 966     /* for each stream, make the setup request */
 967     /* XXX: we assume the same server is used for the control of each
 968        RTSP stream */
 969
 970     for(j = RTSP_RTP_PORT_MIN, i = 0; i < rt->nb_rtsp_streams; ++i) {
 971         char transport[2048];
 972
 973         /**
 974          * WMS serves all UDP data over a single connection, the RTX, which
 975          * isn't necessarily the first in the SDP but has to be the first
 976          * to be set up, else the second/third SETUP will fail with a 461.
 977          */
 978         if (lower_transport == RTSP_LOWER_TRANSPORT_UDP &&
 979              rt->server_type == RTSP_SERVER_WMS) {
 980             if (i == 0) {
 981                 /* rtx first */
 982                 for (rtx = 0; rtx < rt->nb_rtsp_streams; rtx++) {
 983                     int len = strlen(rt->rtsp_streams[rtx]->control_url);
 984                     if (len >= 4 &&
 985                         !strcmp(rt->rtsp_streams[rtx]->control_url + len - 4, "/rtx"))
 986                         break;
 987                 }
 988                 if (rtx == rt->nb_rtsp_streams)
 989                     return -1; /* no RTX found */
 990                 rtsp_st = rt->rtsp_streams[rtx];
 991             } else
 992                 rtsp_st = rt->rtsp_streams[i > rtx ? i : i - 1];
 993         } else
 994             rtsp_st = rt->rtsp_streams[i];
 995
 996         /* RTP/UDP */
 997         if (lower_transport == RTSP_LOWER_TRANSPORT_UDP) {
 998             char buf[256];
 999
1000             if (rt->server_type == RTSP_SERVER_WMS && i > 1) {
1001                 port = reply->transports[0].client_port_min;
1002                 goto have_port;
1003             }
1004
1005             /* first try in specified port range */
1006             if (RTSP_RTP_PORT_MIN != 0) {
1007                 while(j <= RTSP_RTP_PORT_MAX) {
1008                     snprintf(buf, sizeof(buf), "rtp://%s?localport=%d", host, j);
1009                     j += 2; /* we will use two port by rtp stream (rtp and rtcp) */
1010                     if (url_open(&rtsp_st->rtp_handle, buf, URL_RDWR) == 0) {
1011                         goto rtp_opened;
1012                     }
1013                 }
1014             }
1015
1016 /*            then try on any port
1017 **            if (url_open(&rtsp_st->rtp_handle, "rtp://", URL_RDONLY) < 0) {
1018 **                err = AVERROR_INVALIDDATA;
1019 **                goto fail;
1020 **            }
1021 */
1022
1023         rtp_opened:
1024             port = rtp_get_local_port(rtsp_st->rtp_handle);
1025         have_port:
1026             snprintf(transport, sizeof(transport) - 1,
1027                      "%s/UDP;", trans_pref);
1028             if (rt->server_type != RTSP_SERVER_REAL)
1029                 av_strlcat(transport, "unicast;", sizeof(transport));
1030             av_strlcatf(transport, sizeof(transport),
1031                      "client_port=%d", port);
1032             if (rt->transport == RTSP_TRANSPORT_RTP &&
1033                 !(rt->server_type == RTSP_SERVER_WMS && i > 0))
1034                 av_strlcatf(transport, sizeof(transport), "-%d", port + 1);
1035         }
1036
1037         /* RTP/TCP */
1038         else if (lower_transport == RTSP_LOWER_TRANSPORT_TCP) {
1039             /** For WMS streams, the application streams are only used for
1040              * UDP. When trying to set it up for TCP streams, the server
1041              * will return an error. Therefore, we skip those streams. */
1042             if (rt->server_type == RTSP_SERVER_WMS &&
1043                 s->streams[rtsp_st->stream_index]->codec->codec_type == CODEC_TYPE_DATA)
1044                 continue;
1045             snprintf(transport, sizeof(transport) - 1,
1046                      "%s/TCP;", trans_pref);
1047             if (rt->server_type == RTSP_SERVER_WMS)
1048                 av_strlcat(transport, "unicast;", sizeof(transport));
1049             av_strlcatf(transport, sizeof(transport),
1050                         "interleaved=%d-%d",
1051                         interleave, interleave + 1);
1052             interleave += 2;
1053         }
1054
1055         else if (lower_transport == RTSP_LOWER_TRANSPORT_UDP_MULTICAST) {
1056             snprintf(transport, sizeof(transport) - 1,
1057                      "%s/UDP;multicast", trans_pref);
1058         }
1059         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),
1063                  "SETUP %s RTSP/1.0\r\n"
1064                  "Transport: %s\r\n",
1065                  rtsp_st->control_url, transport);
1066         if (i == 0 && rt->server_type == RTSP_SERVER_REAL) {
1067             char real_res[41], real_csum[9];
1068             ff_rdt_calc_response_and_checksum(real_res, real_csum,
1069                                               real_challenge);
1070             av_strlcatf(cmd, sizeof(cmd),
1071                         "If-Match: %s\r\n"
1072                         "RealChallenge2: %s, sd=%s\r\n",
1073                         rt->session_id, real_res, real_csum);
1074         }
1075         rtsp_send_cmd(s, cmd, reply, NULL);
1076         if (reply->status_code == 461 /* Unsupported protocol */ && i == 0) {
1077             err = 1;
1078             goto fail;
1079         } else if (reply->status_code != RTSP_STATUS_OK ||
1080                    reply->nb_transports != 1) {
1081             err = AVERROR_INVALIDDATA;
1082             goto fail;
1083         }
1084
1085         /* XXX: same protocol for all streams is required */
1086         if (i > 0) {
1087             if (reply->transports[0].lower_transport != rt->lower_transport ||
1088                 reply->transports[0].transport != rt->transport) {
1089                 err = AVERROR_INVALIDDATA;
1090                 goto fail;
1091             }
1092         } else {
1093             rt->lower_transport = reply->transports[0].lower_transport;
1094             rt->transport = reply->transports[0].transport;
1095         }
1096
1097         /* close RTP connection if not choosen */
1098         if (reply->transports[0].lower_transport != RTSP_LOWER_TRANSPORT_UDP &&
1099             (lower_transport == RTSP_LOWER_TRANSPORT_UDP)) {
1100             url_close(rtsp_st->rtp_handle);
1101             rtsp_st->rtp_handle = NULL;
1102         }
1103
1104         switch(reply->transports[0].lower_transport) {
1105         case RTSP_LOWER_TRANSPORT_TCP:
1106             rtsp_st->interleaved_min = reply->transports[0].interleaved_min;
1107             rtsp_st->interleaved_max = reply->transports[0].interleaved_max;
1108             break;
1109
1110         case RTSP_LOWER_TRANSPORT_UDP:
1111             {
1112                 char url[1024];
1113
1114                 /* XXX: also use address if specified */
1115                 snprintf(url, sizeof(url), "rtp://%s:%d",
1116                          host, reply->transports[0].server_port_min);
1117                 if (!(rt->server_type == RTSP_SERVER_WMS && i > 1) &&
1118                     rtp_set_remote_url(rtsp_st->rtp_handle, url) < 0) {
1119                     err = AVERROR_INVALIDDATA;
1120                     goto fail;
1121                 }
1122             }
1123             break;
1124         case RTSP_LOWER_TRANSPORT_UDP_MULTICAST:
1125             {
1126                 char url[1024];
1127                 struct in_addr in;
1128                 int port, ttl;
1129
1130                 if (reply->transports[0].destination) {
1131                     in.s_addr = htonl(reply->transports[0].destination);
1132                     port      = reply->transports[0].port_min;
1133                     ttl       = reply->transports[0].ttl;
1134                 } else {
1135                     in        = rtsp_st->sdp_ip;
1136                     port      = rtsp_st->sdp_port;
1137                     ttl       = rtsp_st->sdp_ttl;
1138                 }
1139                 snprintf(url, sizeof(url), "rtp://%s:%d?ttl=%d",
1140                          inet_ntoa(in), port, ttl);
1141                 if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
1142                     err = AVERROR_INVALIDDATA;
1143                     goto fail;
1144                 }
1145             }
1146             break;
1147         }
1148
1149         if ((err = rtsp_open_transport_ctx(s, rtsp_st)))
1150             goto fail;
1151     }
1152
1153     if (reply->timeout > 0)
1154         rt->timeout = reply->timeout;
1155
1156     if (rt->server_type == RTSP_SERVER_REAL)
1157         rt->need_subscription = 1;
1158
1159     return 0;
1160
1161 fail:
1162     for (i=0; i<rt->nb_rtsp_streams; i++) {
1163         if (rt->rtsp_streams[i]->rtp_handle) {
1164             url_close(rt->rtsp_streams[i]->rtp_handle);
1165             rt->rtsp_streams[i]->rtp_handle = NULL;
1166         }
1167     }
1168     return err;
1169 }
1170
1171 static int rtsp_read_header(AVFormatContext *s,
1172                             AVFormatParameters *ap)
1173 {
1174     RTSPState *rt = s->priv_data;
1175     char host[1024], path[1024], tcpname[1024], cmd[2048], auth[128];
1176     char *option_list, *option, *filename;
1177     URLContext *rtsp_hd;
1178     int port, ret, err;
1179     RTSPMessageHeader reply1, *reply = &reply1;
1180     unsigned char *content = NULL;
1181     int lower_transport_mask = 0;
1182     char real_challenge[64];
1183
1184     /* extract hostname and port */
1185     url_split(NULL, 0, auth, sizeof(auth),
1186               host, sizeof(host), &port, path, sizeof(path), s->filename);
1187     if (*auth) {
1188         int auth_len = strlen(auth), b64_len = ((auth_len + 2) / 3) * 4 + 1;
1189
1190         if (!(rt->auth_b64 = av_malloc(b64_len)))
1191             return AVERROR(ENOMEM);
1192         if (!av_base64_encode(rt->auth_b64, b64_len, auth, auth_len)) {
1193             err = AVERROR(EINVAL);
1194             goto fail;
1195         }
1196     }
1197     if (port < 0)
1198         port = RTSP_DEFAULT_PORT;
1199
1200     /* search for options */
1201     option_list = strchr(path, '?');
1202     if (option_list) {
1203         filename = strchr(s->filename, '?');
1204         while(option_list) {
1205             /* move the option pointer */
1206             option = ++option_list;
1207             option_list = strchr(option_list, '&');
1208             if (option_list)
1209                 *option_list = 0;
1210
1211             /* handle the options */
1212             if (strcmp(option, "udp") == 0)
1213                 lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_UDP);
1214             else if (strcmp(option, "multicast") == 0)
1215                 lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_UDP_MULTICAST);
1216             else if (strcmp(option, "tcp") == 0)
1217                 lower_transport_mask = (1<< RTSP_LOWER_TRANSPORT_TCP);
1218             else {
1219                 strcpy(++filename, option);
1220                 filename += strlen(option);
1221                 if (option_list) *filename = '&';
1222             }
1223         }
1224         *filename = 0;
1225     }
1226
1227     if (!lower_transport_mask)
1228         lower_transport_mask = (1 << RTSP_LOWER_TRANSPORT_NB) - 1;
1229
1230     /* open the tcp connexion */
1231     snprintf(tcpname, sizeof(tcpname), "tcp://%s:%d", host, port);
1232     if (url_open(&rtsp_hd, tcpname, URL_RDWR) < 0) {
1233         err = AVERROR(EIO);
1234         goto fail;
1235     }
1236     rt->rtsp_hd = rtsp_hd;
1237     rt->seq = 0;
1238
1239     /* request options supported by the server; this also detects server type */
1240     for (rt->server_type = RTSP_SERVER_RTP;;) {
1241         snprintf(cmd, sizeof(cmd),
1242                  "OPTIONS %s RTSP/1.0\r\n", s->filename);
1243         if (rt->server_type == RTSP_SERVER_REAL)
1244             av_strlcat(cmd,
1245                        /**
1246                         * The following entries are required for proper
1247                         * streaming from a Realmedia server. They are
1248                         * interdependent in some way although we currently
1249                         * don't quite understand how. Values were copied
1250                         * from mplayer SVN r23589.
1251                         * @param CompanyID is a 16-byte ID in base64
1252                         * @param ClientChallenge is a 16-byte ID in hex
1253                         */
1254                        "ClientChallenge: 9e26d33f2984236010ef6253fb1887f7\r\n"
1255                        "PlayerStarttime: [28/03/2003:22:50:23 00:00]\r\n"
1256                        "CompanyID: KnKV4M4I/B2FjJ1TToLycw==\r\n"
1257                        "GUID: 00000000-0000-0000-0000-000000000000\r\n",
1258                        sizeof(cmd));
1259         rtsp_send_cmd(s, cmd, reply, NULL);
1260         if (reply->status_code != RTSP_STATUS_OK) {
1261             err = AVERROR_INVALIDDATA;
1262             goto fail;
1263         }
1264
1265         /* detect server type if not standard-compliant RTP */
1266         if (rt->server_type != RTSP_SERVER_REAL && reply->real_challenge[0]) {
1267             rt->server_type = RTSP_SERVER_REAL;
1268             continue;
1269         } else if (!strncasecmp(reply->server, "WMServer/", 9)) {
1270             rt->server_type = RTSP_SERVER_WMS;
1271         } else if (rt->server_type == RTSP_SERVER_REAL) {
1272             strcpy(real_challenge, reply->real_challenge);
1273         }
1274         break;
1275     }
1276
1277     /* describe the stream */
1278     snprintf(cmd, sizeof(cmd),
1279              "DESCRIBE %s RTSP/1.0\r\n"
1280              "Accept: application/sdp\r\n",
1281              s->filename);
1282     if (rt->server_type == RTSP_SERVER_REAL) {
1283         /**
1284          * The Require: attribute is needed for proper streaming from
1285          * Realmedia servers.
1286          */
1287         av_strlcat(cmd,
1288                    "Require: com.real.retain-entity-for-setup\r\n",
1289                    sizeof(cmd));
1290     }
1291     rtsp_send_cmd(s, cmd, reply, &content);
1292     if (!content) {
1293         err = AVERROR_INVALIDDATA;
1294         goto fail;
1295     }
1296     if (reply->status_code != RTSP_STATUS_OK) {
1297         err = AVERROR_INVALIDDATA;
1298         goto fail;
1299     }
1300
1301     /* now we got the SDP description, we parse it */
1302     ret = sdp_parse(s, (const char *)content);
1303     av_freep(&content);
1304     if (ret < 0) {
1305         err = AVERROR_INVALIDDATA;
1306         goto fail;
1307     }
1308
1309     do {
1310         int lower_transport = ff_log2_tab[lower_transport_mask & ~(lower_transport_mask - 1)];
1311
1312         err = make_setup_request(s, host, port, lower_transport,
1313                                  rt->server_type == RTSP_SERVER_REAL ?
1314                                      real_challenge : NULL);
1315         if (err < 0)
1316             goto fail;
1317         lower_transport_mask &= ~(1 << lower_transport);
1318         if (lower_transport_mask == 0 && err == 1) {
1319             err = AVERROR(FF_NETERROR(EPROTONOSUPPORT));
1320             goto fail;
1321         }
1322     } while (err);
1323
1324     rt->state = RTSP_STATE_IDLE;
1325     rt->seek_timestamp = 0; /* default is to start stream at position
1326                                zero */
1327     if (ap->initial_pause) {
1328         /* do not start immediately */
1329     } else {
1330         if (rtsp_read_play(s) < 0) {
1331             err = AVERROR_INVALIDDATA;
1332             goto fail;
1333         }
1334     }
1335     return 0;
1336  fail:
1337     rtsp_close_streams(rt);
1338     av_freep(&content);
1339     url_close(rt->rtsp_hd);
1340     av_freep(&rt->auth_b64);
1341     return err;
1342 }
1343
1344 static int tcp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
1345                            uint8_t *buf, int buf_size)
1346 {
1347     RTSPState *rt = s->priv_data;
1348     int id, len, i, ret;
1349     RTSPStream *rtsp_st;
1350
1351 #ifdef DEBUG_RTP_TCP
1352     dprintf(s, "tcp_read_packet:\n");
1353 #endif
1354  redo:
1355     for(;;) {
1356         RTSPMessageHeader reply;
1357
1358         ret = rtsp_read_reply(s, &reply, NULL, 1);
1359         if (ret == -1)
1360             return -1;
1361         if (ret == 1) /* received '$' */
1362             break;
1363         /* XXX: parse message */
1364         if (rt->state != RTSP_STATE_PLAYING)
1365             return 0;
1366     }
1367     ret = url_read_complete(rt->rtsp_hd, buf, 3);
1368     if (ret != 3)
1369         return -1;
1370     id = buf[0];
1371     len = AV_RB16(buf + 1);
1372 #ifdef DEBUG_RTP_TCP
1373     dprintf(s, "id=%d len=%d\n", id, len);
1374 #endif
1375     if (len > buf_size || len < 12)
1376         goto redo;
1377     /* get the data */
1378     ret = url_read_complete(rt->rtsp_hd, buf, len);
1379     if (ret != len)
1380         return -1;
1381     if (rt->transport == RTSP_TRANSPORT_RDT &&
1382         ff_rdt_parse_header(buf, len, &id, NULL, NULL, NULL, NULL) < 0)
1383         return -1;
1384
1385     /* find the matching stream */
1386     for(i = 0; i < rt->nb_rtsp_streams; i++) {
1387         rtsp_st = rt->rtsp_streams[i];
1388         if (id >= rtsp_st->interleaved_min &&
1389             id <= rtsp_st->interleaved_max)
1390             goto found;
1391     }
1392     goto redo;
1393  found:
1394     *prtsp_st = rtsp_st;
1395     return len;
1396 }
1397
1398 static int udp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
1399                            uint8_t *buf, int buf_size)
1400 {
1401     RTSPState *rt = s->priv_data;
1402     RTSPStream *rtsp_st;
1403     fd_set rfds;
1404     int fd, fd_max, n, i, ret, tcp_fd;
1405     struct timeval tv;
1406
1407     for(;;) {
1408         if (url_interrupt_cb())
1409             return AVERROR(EINTR);
1410         FD_ZERO(&rfds);
1411         if (rt->rtsp_hd) {
1412             tcp_fd = fd_max = url_get_file_handle(rt->rtsp_hd);
1413             FD_SET(tcp_fd, &rfds);
1414         } else {
1415             fd_max = 0;
1416             tcp_fd = -1;
1417         }
1418         for(i = 0; i < rt->nb_rtsp_streams; i++) {
1419             rtsp_st = rt->rtsp_streams[i];
1420             if (rtsp_st->rtp_handle) {
1421                 /* currently, we cannot probe RTCP handle because of
1422                  * blocking restrictions */
1423                 fd = url_get_file_handle(rtsp_st->rtp_handle);
1424                 if (fd > fd_max)
1425                     fd_max = fd;
1426                 FD_SET(fd, &rfds);
1427             }
1428         }
1429         tv.tv_sec = 0;
1430         tv.tv_usec = 100 * 1000;
1431         n = select(fd_max + 1, &rfds, NULL, NULL, &tv);
1432         if (n > 0) {
1433             for(i = 0; i < rt->nb_rtsp_streams; i++) {
1434                 rtsp_st = rt->rtsp_streams[i];
1435                 if (rtsp_st->rtp_handle) {
1436                     fd = url_get_file_handle(rtsp_st->rtp_handle);
1437                     if (FD_ISSET(fd, &rfds)) {
1438                         ret = url_read(rtsp_st->rtp_handle, buf, buf_size);
1439                         if (ret > 0) {
1440                             *prtsp_st = rtsp_st;
1441                             return ret;
1442                         }
1443                     }
1444                 }
1445             }
1446             if (FD_ISSET(tcp_fd, &rfds)) {
1447                 RTSPMessageHeader reply;
1448
1449                 rtsp_read_reply(s, &reply, NULL, 0);
1450                 /* XXX: parse message */
1451                 if (rt->state != RTSP_STATE_PLAYING)
1452                     return 0;
1453             }
1454         }
1455     }
1456 }
1457
1458 static int rtsp_read_packet(AVFormatContext *s,
1459                             AVPacket *pkt)
1460 {
1461     RTSPState *rt = s->priv_data;
1462     RTSPStream *rtsp_st;
1463     int ret, len;
1464     uint8_t buf[10 * RTP_MAX_PACKET_LENGTH];
1465     RTSPMessageHeader reply1, *reply = &reply1;
1466     char cmd[1024];
1467
1468     if (rt->server_type == RTSP_SERVER_REAL) {
1469         int i;
1470         enum AVDiscard cache[MAX_STREAMS];
1471
1472         for (i = 0; i < s->nb_streams; i++)
1473             cache[i] = s->streams[i]->discard;
1474
1475         if (!rt->need_subscription) {
1476             if (memcmp (cache, rt->real_setup_cache,
1477                         sizeof(enum AVDiscard) * s->nb_streams)) {
1478                 av_strlcatf(cmd, sizeof(cmd),
1479                             "SET_PARAMETER %s RTSP/1.0\r\n"
1480                             "Unsubscribe: %s\r\n",
1481                             s->filename, rt->last_subscription);
1482                 rtsp_send_cmd(s, cmd, reply, NULL);
1483                 if (reply->status_code != RTSP_STATUS_OK)
1484                     return AVERROR_INVALIDDATA;
1485                 rt->need_subscription = 1;
1486             }
1487         }
1488
1489         if (rt->need_subscription) {
1490             int r, rule_nr, first = 1;
1491
1492             memcpy(rt->real_setup_cache, cache,
1493                    sizeof(enum AVDiscard) * s->nb_streams);
1494             rt->last_subscription[0] = 0;
1495
1496             snprintf(cmd, sizeof(cmd),
1497                      "SET_PARAMETER %s RTSP/1.0\r\n"
1498                      "Subscribe: ",
1499                      s->filename);
1500             for (i = 0; i < rt->nb_rtsp_streams; i++) {
1501                 rule_nr = 0;
1502                 for (r = 0; r < s->nb_streams; r++) {
1503                     if (s->streams[r]->priv_data == rt->rtsp_streams[i]) {
1504                         if (s->streams[r]->discard != AVDISCARD_ALL) {
1505                             if (!first)
1506                                 av_strlcat(rt->last_subscription, ",",
1507                                            sizeof(rt->last_subscription));
1508                             ff_rdt_subscribe_rule(
1509                                 rt->last_subscription,
1510                                 sizeof(rt->last_subscription), i, rule_nr);
1511                             first = 0;
1512                         }
1513                         rule_nr++;
1514                     }
1515                 }
1516             }
1517             av_strlcatf(cmd, sizeof(cmd), "%s\r\n", rt->last_subscription);
1518             rtsp_send_cmd(s, cmd, reply, NULL);
1519             if (reply->status_code != RTSP_STATUS_OK)
1520                 return AVERROR_INVALIDDATA;
1521             rt->need_subscription = 0;
1522
1523             if (rt->state == RTSP_STATE_PLAYING)
1524                 rtsp_read_play (s);
1525         }
1526     }
1527
1528     /* get next frames from the same RTP packet */
1529     if (rt->cur_transport_priv) {
1530         if (rt->transport == RTSP_TRANSPORT_RDT)
1531             ret = ff_rdt_parse_packet(rt->cur_transport_priv, pkt, NULL, 0);
1532         else
1533             ret = rtp_parse_packet(rt->cur_transport_priv, pkt, NULL, 0);
1534         if (ret == 0) {
1535             rt->cur_transport_priv = NULL;
1536             return 0;
1537         } else if (ret == 1) {
1538             return 0;
1539         } else {
1540             rt->cur_transport_priv = NULL;
1541         }
1542     }
1543
1544     /* read next RTP packet */
1545  redo:
1546     switch(rt->lower_transport) {
1547     default:
1548     case RTSP_LOWER_TRANSPORT_TCP:
1549         len = tcp_read_packet(s, &rtsp_st, buf, sizeof(buf));
1550         break;
1551     case RTSP_LOWER_TRANSPORT_UDP:
1552     case RTSP_LOWER_TRANSPORT_UDP_MULTICAST:
1553         len = udp_read_packet(s, &rtsp_st, buf, sizeof(buf));
1554         if (len >=0 && rtsp_st->transport_priv && rt->transport == RTSP_TRANSPORT_RTP)
1555             rtp_check_and_send_back_rr(rtsp_st->transport_priv, len);
1556         break;
1557     }
1558     if (len < 0)
1559         return len;
1560     if (len == 0)
1561         return AVERROR_EOF;
1562     if (rt->transport == RTSP_TRANSPORT_RDT)
1563         ret = ff_rdt_parse_packet(rtsp_st->transport_priv, pkt, buf, len);
1564     else
1565         ret = rtp_parse_packet(rtsp_st->transport_priv, pkt, buf, len);
1566     if (ret < 0)
1567         goto redo;
1568     if (ret == 1) {
1569         /* more packets may follow, so we save the RTP context */
1570         rt->cur_transport_priv = rtsp_st->transport_priv;
1571     }
1572
1573     /* send dummy request to keep TCP connection alive */
1574     if ((rt->server_type == RTSP_SERVER_WMS ||
1575          rt->server_type == RTSP_SERVER_REAL) &&
1576         (av_gettime() - rt->last_cmd_time) / 1000000 >= rt->timeout / 2) {
1577         if (rt->server_type == RTSP_SERVER_WMS) {
1578             snprintf(cmd, sizeof(cmd) - 1,
1579                      "GET_PARAMETER %s RTSP/1.0\r\n",
1580                      s->filename);
1581             rtsp_send_cmd_async(s, cmd, reply, NULL);
1582         } else {
1583             rtsp_send_cmd_async(s, "OPTIONS * RTSP/1.0\r\n",
1584                                 reply, NULL);
1585         }
1586     }
1587
1588     return 0;
1589 }
1590
1591 static int rtsp_read_play(AVFormatContext *s)
1592 {
1593     RTSPState *rt = s->priv_data;
1594     RTSPMessageHeader reply1, *reply = &reply1;
1595     char cmd[1024];
1596
1597     av_log(s, AV_LOG_DEBUG, "hello state=%d\n", rt->state);
1598
1599     if (!(rt->server_type == RTSP_SERVER_REAL && rt->need_subscription)) {
1600         if (rt->state == RTSP_STATE_PAUSED) {
1601             snprintf(cmd, sizeof(cmd),
1602                      "PLAY %s RTSP/1.0\r\n",
1603                      s->filename);
1604         } else {
1605             snprintf(cmd, sizeof(cmd),
1606                      "PLAY %s RTSP/1.0\r\n"
1607                      "Range: npt=%0.3f-\r\n",
1608                      s->filename,
1609                      (double)rt->seek_timestamp / AV_TIME_BASE);
1610         }
1611         rtsp_send_cmd(s, cmd, reply, NULL);
1612         if (reply->status_code != RTSP_STATUS_OK) {
1613             return -1;
1614         }
1615     }
1616     rt->state = RTSP_STATE_PLAYING;
1617     return 0;
1618 }
1619
1620 /* pause the stream */
1621 static int rtsp_read_pause(AVFormatContext *s)
1622 {
1623     RTSPState *rt = s->priv_data;
1624     RTSPMessageHeader reply1, *reply = &reply1;
1625     char cmd[1024];
1626
1627     rt = s->priv_data;
1628
1629     if (rt->state != RTSP_STATE_PLAYING)
1630         return 0;
1631     else if (!(rt->server_type == RTSP_SERVER_REAL && rt->need_subscription)) {
1632         snprintf(cmd, sizeof(cmd),
1633                  "PAUSE %s RTSP/1.0\r\n",
1634                  s->filename);
1635         rtsp_send_cmd(s, cmd, reply, NULL);
1636         if (reply->status_code != RTSP_STATUS_OK) {
1637             return -1;
1638         }
1639     }
1640     rt->state = RTSP_STATE_PAUSED;
1641     return 0;
1642 }
1643
1644 static int rtsp_read_seek(AVFormatContext *s, int stream_index,
1645                           int64_t timestamp, int flags)
1646 {
1647     RTSPState *rt = s->priv_data;
1648
1649     rt->seek_timestamp = av_rescale_q(timestamp, s->streams[stream_index]->time_base, AV_TIME_BASE_Q);
1650     switch(rt->state) {
1651     default:
1652     case RTSP_STATE_IDLE:
1653         break;
1654     case RTSP_STATE_PLAYING:
1655         if (rtsp_read_pause(s) != 0)
1656             return -1;
1657         rt->state = RTSP_STATE_SEEKING;
1658         if (rtsp_read_play(s) != 0)
1659             return -1;
1660         break;
1661     case RTSP_STATE_PAUSED:
1662         rt->state = RTSP_STATE_IDLE;
1663         break;
1664     }
1665     return 0;
1666 }
1667
1668 static int rtsp_read_close(AVFormatContext *s)
1669 {
1670     RTSPState *rt = s->priv_data;
1671     RTSPMessageHeader reply1, *reply = &reply1;
1672     char cmd[1024];
1673
1674 #if 0
1675     /* NOTE: it is valid to flush the buffer here */
1676     if (rt->lower_transport == RTSP_LOWER_TRANSPORT_TCP) {
1677         url_fclose(&rt->rtsp_gb);
1678     }
1679 #endif
1680     snprintf(cmd, sizeof(cmd),
1681              "TEARDOWN %s RTSP/1.0\r\n",
1682              s->filename);
1683     rtsp_send_cmd(s, cmd, reply, NULL);
1684
1685     rtsp_close_streams(rt);
1686     url_close(rt->rtsp_hd);
1687     return 0;
1688 }
1689
1690 #if CONFIG_RTSP_DEMUXER
1691 AVInputFormat rtsp_demuxer = {
1692     "rtsp",
1693     NULL_IF_CONFIG_SMALL("RTSP input format"),
1694     sizeof(RTSPState),
1695     rtsp_probe,
1696     rtsp_read_header,
1697     rtsp_read_packet,
1698     rtsp_read_close,
1699     rtsp_read_seek,
1700     .flags = AVFMT_NOFILE,
1701     .read_play = rtsp_read_play,
1702     .read_pause = rtsp_read_pause,
1703 };
1704 #endif
1705
1706 static int sdp_probe(AVProbeData *p1)
1707 {
1708     const char *p = p1->buf, *p_end = p1->buf + p1->buf_size;
1709
1710     /* we look for a line beginning "c=IN IP4" */
1711     while (p < p_end && *p != '\0') {
1712         if (p + sizeof("c=IN IP4") - 1 < p_end && av_strstart(p, "c=IN IP4", NULL))
1713             return AVPROBE_SCORE_MAX / 2;
1714
1715         while(p < p_end - 1 && *p != '\n') p++;
1716         if (++p >= p_end)
1717             break;
1718         if (*p == '\r')
1719             p++;
1720     }
1721     return 0;
1722 }
1723
1724 #define SDP_MAX_SIZE 8192
1725
1726 static int sdp_read_header(AVFormatContext *s,
1727                            AVFormatParameters *ap)
1728 {
1729     RTSPState *rt = s->priv_data;
1730     RTSPStream *rtsp_st;
1731     int size, i, err;
1732     char *content;
1733     char url[1024];
1734
1735     /* read the whole sdp file */
1736     /* XXX: better loading */
1737     content = av_malloc(SDP_MAX_SIZE);
1738     size = get_buffer(s->pb, content, SDP_MAX_SIZE - 1);
1739     if (size <= 0) {
1740         av_free(content);
1741         return AVERROR_INVALIDDATA;
1742     }
1743     content[size] ='\0';
1744
1745     sdp_parse(s, content);
1746     av_free(content);
1747
1748     /* open each RTP stream */
1749     for(i=0;i<rt->nb_rtsp_streams;i++) {
1750         rtsp_st = rt->rtsp_streams[i];
1751
1752         snprintf(url, sizeof(url), "rtp://%s:%d?localport=%d&ttl=%d",
1753                  inet_ntoa(rtsp_st->sdp_ip),
1754                  rtsp_st->sdp_port,
1755                  rtsp_st->sdp_port,
1756                  rtsp_st->sdp_ttl);
1757         if (url_open(&rtsp_st->rtp_handle, url, URL_RDWR) < 0) {
1758             err = AVERROR_INVALIDDATA;
1759             goto fail;
1760         }
1761         if ((err = rtsp_open_transport_ctx(s, rtsp_st)))
1762             goto fail;
1763     }
1764     return 0;
1765  fail:
1766     rtsp_close_streams(rt);
1767     return err;
1768 }
1769
1770 static int sdp_read_packet(AVFormatContext *s,
1771                             AVPacket *pkt)
1772 {
1773     return rtsp_read_packet(s, pkt);
1774 }
1775
1776 static int sdp_read_close(AVFormatContext *s)
1777 {
1778     RTSPState *rt = s->priv_data;
1779     rtsp_close_streams(rt);
1780     return 0;
1781 }
1782
1783 #if CONFIG_SDP_DEMUXER
1784 AVInputFormat sdp_demuxer = {
1785     "sdp",
1786     NULL_IF_CONFIG_SMALL("SDP"),
1787     sizeof(RTSPState),
1788     sdp_probe,
1789     sdp_read_header,
1790     sdp_read_packet,
1791     sdp_read_close,
1792 };
1793 #endif