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