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