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