8a94baf9d66beeba9d1bd688befa193156036ab0
[libav.git] / libavformat / rtsp.c
1 /*
2 * RTSP/SDP client
3 * Copyright (c) 2002 Fabrice Bellard
4 *
5 * This file is part of Libav.
6 *
7 * Libav 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 * Libav 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 Libav; 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/mathematics.h"
26 #include "libavutil/parseutils.h"
27 #include "libavutil/random_seed.h"
28 #include "libavutil/dict.h"
29 #include "libavutil/opt.h"
30 #include "libavutil/time.h"
31 #include "avformat.h"
32 #include "avio_internal.h"
33
34 #if HAVE_POLL_H
35 #include <poll.h>
36 #endif
37 #include "internal.h"
38 #include "network.h"
39 #include "os_support.h"
40 #include "http.h"
41 #include "rtsp.h"
42
43 #include "rtpdec.h"
44 #include "rtpproto.h"
45 #include "rdt.h"
46 #include "rtpdec_formats.h"
47 #include "rtpenc_chain.h"
48 #include "url.h"
49 #include "rtpenc.h"
50 #include "mpegts.h"
51
52 /* Timeout values for socket poll, in ms,
53 * and read_packet(), in seconds */
54 #define POLL_TIMEOUT_MS 100
55 #define READ_PACKET_TIMEOUT_S 10
56 #define MAX_TIMEOUTS READ_PACKET_TIMEOUT_S * 1000 / POLL_TIMEOUT_MS
57 #define SDP_MAX_SIZE 16384
58 #define RECVBUF_SIZE 10 * RTP_MAX_PACKET_LENGTH
59 #define DEFAULT_REORDERING_DELAY 100000
60
61 #define OFFSET(x) offsetof(RTSPState, x)
62 #define DEC AV_OPT_FLAG_DECODING_PARAM
63 #define ENC AV_OPT_FLAG_ENCODING_PARAM
64
65 #define RTSP_FLAG_OPTS(name, longname) \
66 { name, longname, OFFSET(rtsp_flags), AV_OPT_TYPE_FLAGS, {.i64 = 0}, INT_MIN, INT_MAX, DEC, "rtsp_flags" }, \
67 { "filter_src", "Only receive packets from the negotiated peer IP", 0, AV_OPT_TYPE_CONST, {.i64 = RTSP_FLAG_FILTER_SRC}, 0, 0, DEC, "rtsp_flags" }
68
69 #define RTSP_MEDIATYPE_OPTS(name, longname) \
70 { name, longname, OFFSET(media_type_mask), AV_OPT_TYPE_FLAGS, { .i64 = (1 << (AVMEDIA_TYPE_DATA+1)) - 1 }, INT_MIN, INT_MAX, DEC, "allowed_media_types" }, \
71 { "video", "Video", 0, AV_OPT_TYPE_CONST, {.i64 = 1 << AVMEDIA_TYPE_VIDEO}, 0, 0, DEC, "allowed_media_types" }, \
72 { "audio", "Audio", 0, AV_OPT_TYPE_CONST, {.i64 = 1 << AVMEDIA_TYPE_AUDIO}, 0, 0, DEC, "allowed_media_types" }, \
73 { "data", "Data", 0, AV_OPT_TYPE_CONST, {.i64 = 1 << AVMEDIA_TYPE_DATA}, 0, 0, DEC, "allowed_media_types" }
74
75 #define COMMON_OPTS() \
76 { "reorder_queue_size", "Number of packets to buffer for handling of reordered packets", OFFSET(reordering_queue_size), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, INT_MAX, DEC }, \
77 { "buffer_size", "Underlying protocol send/receive buffer size", OFFSET(buffer_size), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, INT_MAX, DEC|ENC } \
78
79
80 const AVOption ff_rtsp_options[] = {
81 { "initial_pause", "Don't start playing the stream immediately", OFFSET(initial_pause), AV_OPT_TYPE_INT, {.i64 = 0}, 0, 1, DEC },
82 FF_RTP_FLAG_OPTS(RTSPState, rtp_muxer_flags),
83 { "rtsp_transport", "RTSP transport protocols", OFFSET(lower_transport_mask), AV_OPT_TYPE_FLAGS, {.i64 = 0}, INT_MIN, INT_MAX, DEC|ENC, "rtsp_transport" }, \
84 { "udp", "UDP", 0, AV_OPT_TYPE_CONST, {.i64 = 1 << RTSP_LOWER_TRANSPORT_UDP}, 0, 0, DEC|ENC, "rtsp_transport" }, \
85 { "tcp", "TCP", 0, AV_OPT_TYPE_CONST, {.i64 = 1 << RTSP_LOWER_TRANSPORT_TCP}, 0, 0, DEC|ENC, "rtsp_transport" }, \
86 { "udp_multicast", "UDP multicast", 0, AV_OPT_TYPE_CONST, {.i64 = 1 << RTSP_LOWER_TRANSPORT_UDP_MULTICAST}, 0, 0, DEC, "rtsp_transport" },
87 { "http", "HTTP tunneling", 0, AV_OPT_TYPE_CONST, {.i64 = (1 << RTSP_LOWER_TRANSPORT_HTTP)}, 0, 0, DEC, "rtsp_transport" },
88 RTSP_FLAG_OPTS("rtsp_flags", "RTSP flags"),
89 { "listen", "Wait for incoming connections", 0, AV_OPT_TYPE_CONST, {.i64 = RTSP_FLAG_LISTEN}, 0, 0, DEC, "rtsp_flags" },
90 RTSP_MEDIATYPE_OPTS("allowed_media_types", "Media types to accept from the server"),
91 { "min_port", "Minimum local UDP port", OFFSET(rtp_port_min), AV_OPT_TYPE_INT, {.i64 = RTSP_RTP_PORT_MIN}, 0, 65535, DEC|ENC },
92 { "max_port", "Maximum local UDP port", OFFSET(rtp_port_max), AV_OPT_TYPE_INT, {.i64 = RTSP_RTP_PORT_MAX}, 0, 65535, DEC|ENC },
93 { "timeout", "Maximum timeout (in seconds) to wait for incoming connections. -1 is infinite. Implies flag listen", OFFSET(initial_timeout), AV_OPT_TYPE_INT, {.i64 = -1}, INT_MIN, INT_MAX, DEC },
94 COMMON_OPTS(),
95 { NULL },
96 };
97
98 static const AVOption sdp_options[] = {
99 RTSP_FLAG_OPTS("sdp_flags", "SDP flags"),
100 { "custom_io", "Use custom IO", 0, AV_OPT_TYPE_CONST, {.i64 = RTSP_FLAG_CUSTOM_IO}, 0, 0, DEC, "rtsp_flags" },
101 { "rtcp_to_source", "Send RTCP packets to the source address of received packets", 0, AV_OPT_TYPE_CONST, {.i64 = RTSP_FLAG_RTCP_TO_SOURCE}, 0, 0, DEC, "rtsp_flags" },
102 RTSP_MEDIATYPE_OPTS("allowed_media_types", "Media types to accept from the server"),
103 COMMON_OPTS(),
104 { NULL },
105 };
106
107 static const AVOption rtp_options[] = {
108 RTSP_FLAG_OPTS("rtp_flags", "RTP flags"),
109 COMMON_OPTS(),
110 { NULL },
111 };
112
113
114 static AVDictionary *map_to_opts(RTSPState *rt)
115 {
116 AVDictionary *opts = NULL;
117 char buf[256];
118
119 snprintf(buf, sizeof(buf), "%d", rt->buffer_size);
120 av_dict_set(&opts, "buffer_size", buf, 0);
121
122 return opts;
123 }
124
125 static void get_word_until_chars(char *buf, int buf_size,
126 const char *sep, const char **pp)
127 {
128 const char *p;
129 char *q;
130
131 p = *pp;
132 p += strspn(p, SPACE_CHARS);
133 q = buf;
134 while (!strchr(sep, *p) && *p != '\0') {
135 if ((q - buf) < buf_size - 1)
136 *q++ = *p;
137 p++;
138 }
139 if (buf_size > 0)
140 *q = '\0';
141 *pp = p;
142 }
143
144 static void get_word_sep(char *buf, int buf_size, const char *sep,
145 const char **pp)
146 {
147 if (**pp == '/') (*pp)++;
148 get_word_until_chars(buf, buf_size, sep, pp);
149 }
150
151 static void get_word(char *buf, int buf_size, const char **pp)
152 {
153 get_word_until_chars(buf, buf_size, SPACE_CHARS, pp);
154 }
155
156 /** Parse a string p in the form of Range:npt=xx-xx, and determine the start
157 * and end time.
158 * Used for seeking in the rtp stream.
159 */
160 static void rtsp_parse_range_npt(const char *p, int64_t *start, int64_t *end)
161 {
162 char buf[256];
163
164 p += strspn(p, SPACE_CHARS);
165 if (!av_stristart(p, "npt=", &p))
166 return;
167
168 *start = AV_NOPTS_VALUE;
169 *end = AV_NOPTS_VALUE;
170
171 get_word_sep(buf, sizeof(buf), "-", &p);
172 if (av_parse_time(start, buf, 1) < 0)
173 return;
174 if (*p == '-') {
175 p++;
176 get_word_sep(buf, sizeof(buf), "-", &p);
177 av_parse_time(end, buf, 1);
178 }
179 }
180
181 static int get_sockaddr(const char *buf, struct sockaddr_storage *sock)
182 {
183 struct addrinfo hints = { 0 }, *ai = NULL;
184 hints.ai_flags = AI_NUMERICHOST;
185 if (getaddrinfo(buf, NULL, &hints, &ai))
186 return -1;
187 memcpy(sock, ai->ai_addr, FFMIN(sizeof(*sock), ai->ai_addrlen));
188 freeaddrinfo(ai);
189 return 0;
190 }
191
192 #if CONFIG_RTPDEC
193 static void init_rtp_handler(RTPDynamicProtocolHandler *handler,
194 RTSPStream *rtsp_st, AVStream *st)
195 {
196 AVCodecContext *codec = st ? st->codec : NULL;
197 if (!handler)
198 return;
199 if (codec)
200 codec->codec_id = handler->codec_id;
201 rtsp_st->dynamic_handler = handler;
202 if (st)
203 st->need_parsing = handler->need_parsing;
204 if (handler->priv_data_size) {
205 rtsp_st->dynamic_protocol_context = av_mallocz(handler->priv_data_size);
206 if (!rtsp_st->dynamic_protocol_context)
207 rtsp_st->dynamic_handler = NULL;
208 }
209 }
210
211 static void finalize_rtp_handler_init(AVFormatContext *s, RTSPStream *rtsp_st,
212 AVStream *st)
213 {
214 if (rtsp_st->dynamic_handler && rtsp_st->dynamic_handler->init) {
215 int ret = rtsp_st->dynamic_handler->init(s, st ? st->index : -1,
216 rtsp_st->dynamic_protocol_context);
217 if (ret < 0) {
218 if (rtsp_st->dynamic_protocol_context) {
219 if (rtsp_st->dynamic_handler->close)
220 rtsp_st->dynamic_handler->close(
221 rtsp_st->dynamic_protocol_context);
222 av_free(rtsp_st->dynamic_protocol_context);
223 }
224 rtsp_st->dynamic_protocol_context = NULL;
225 rtsp_st->dynamic_handler = NULL;
226 }
227 }
228 }
229
230 /* parse the rtpmap description: <codec_name>/<clock_rate>[/<other params>] */
231 static int sdp_parse_rtpmap(AVFormatContext *s,
232 AVStream *st, RTSPStream *rtsp_st,
233 int payload_type, const char *p)
234 {
235 AVCodecContext *codec = st->codec;
236 char buf[256];
237 int i;
238 AVCodec *c;
239 const char *c_name;
240
241 /* See if we can handle this kind of payload.
242 * The space should normally not be there but some Real streams or
243 * particular servers ("RealServer Version 6.1.3.970", see issue 1658)
244 * have a trailing space. */
245 get_word_sep(buf, sizeof(buf), "/ ", &p);
246 if (payload_type < RTP_PT_PRIVATE) {
247 /* We are in a standard case
248 * (from http://www.iana.org/assignments/rtp-parameters). */
249 codec->codec_id = ff_rtp_codec_id(buf, codec->codec_type);
250 }
251
252 if (codec->codec_id == AV_CODEC_ID_NONE) {
253 RTPDynamicProtocolHandler *handler =
254 ff_rtp_handler_find_by_name(buf, codec->codec_type);
255 init_rtp_handler(handler, rtsp_st, st);
256 /* If no dynamic handler was found, check with the list of standard
257 * allocated types, if such a stream for some reason happens to
258 * use a private payload type. This isn't handled in rtpdec.c, since
259 * the format name from the rtpmap line never is passed into rtpdec. */
260 if (!rtsp_st->dynamic_handler)
261 codec->codec_id = ff_rtp_codec_id(buf, codec->codec_type);
262 }
263
264 c = avcodec_find_decoder(codec->codec_id);
265 if (c && c->name)
266 c_name = c->name;
267 else
268 c_name = "(null)";
269
270 get_word_sep(buf, sizeof(buf), "/", &p);
271 i = atoi(buf);
272 switch (codec->codec_type) {
273 case AVMEDIA_TYPE_AUDIO:
274 av_log(s, AV_LOG_DEBUG, "audio codec set to: %s\n", c_name);
275 codec->sample_rate = RTSP_DEFAULT_AUDIO_SAMPLERATE;
276 codec->channels = RTSP_DEFAULT_NB_AUDIO_CHANNELS;
277 if (i > 0) {
278 codec->sample_rate = i;
279 avpriv_set_pts_info(st, 32, 1, codec->sample_rate);
280 get_word_sep(buf, sizeof(buf), "/", &p);
281 i = atoi(buf);
282 if (i > 0)
283 codec->channels = i;
284 }
285 av_log(s, AV_LOG_DEBUG, "audio samplerate set to: %i\n",
286 codec->sample_rate);
287 av_log(s, AV_LOG_DEBUG, "audio channels set to: %i\n",
288 codec->channels);
289 break;
290 case AVMEDIA_TYPE_VIDEO:
291 av_log(s, AV_LOG_DEBUG, "video codec set to: %s\n", c_name);
292 if (i > 0)
293 avpriv_set_pts_info(st, 32, 1, i);
294 break;
295 default:
296 break;
297 }
298 finalize_rtp_handler_init(s, rtsp_st, st);
299 return 0;
300 }
301
302 /* parse the attribute line from the fmtp a line of an sdp response. This
303 * is broken out as a function because it is used in rtp_h264.c, which is
304 * forthcoming. */
305 int ff_rtsp_next_attr_and_value(const char **p, char *attr, int attr_size,
306 char *value, int value_size)
307 {
308 *p += strspn(*p, SPACE_CHARS);
309 if (**p) {
310 get_word_sep(attr, attr_size, "=", p);
311 if (**p == '=')
312 (*p)++;
313 get_word_sep(value, value_size, ";", p);
314 if (**p == ';')
315 (*p)++;
316 return 1;
317 }
318 return 0;
319 }
320
321 typedef struct SDPParseState {
322 /* SDP only */
323 struct sockaddr_storage default_ip;
324 int default_ttl;
325 int skip_media; ///< set if an unknown m= line occurs
326 int nb_default_include_source_addrs; /**< Number of source-specific multicast include source IP address (from SDP content) */
327 struct RTSPSource **default_include_source_addrs; /**< Source-specific multicast include source IP address (from SDP content) */
328 int nb_default_exclude_source_addrs; /**< Number of source-specific multicast exclude source IP address (from SDP content) */
329 struct RTSPSource **default_exclude_source_addrs; /**< Source-specific multicast exclude source IP address (from SDP content) */
330 int seen_rtpmap;
331 int seen_fmtp;
332 char delayed_fmtp[2048];
333 } SDPParseState;
334
335 static void copy_default_source_addrs(struct RTSPSource **addrs, int count,
336 struct RTSPSource ***dest, int *dest_count)
337 {
338 RTSPSource *rtsp_src, *rtsp_src2;
339 int i;
340 for (i = 0; i < count; i++) {
341 rtsp_src = addrs[i];
342 rtsp_src2 = av_malloc(sizeof(*rtsp_src2));
343 if (!rtsp_src2)
344 continue;
345 memcpy(rtsp_src2, rtsp_src, sizeof(*rtsp_src));
346 dynarray_add(dest, dest_count, rtsp_src2);
347 }
348 }
349
350 static void parse_fmtp(AVFormatContext *s, RTSPState *rt,
351 int payload_type, const char *line)
352 {
353 int i;
354
355 for (i = 0; i < rt->nb_rtsp_streams; i++) {
356 RTSPStream *rtsp_st = rt->rtsp_streams[i];
357 if (rtsp_st->sdp_payload_type == payload_type &&
358 rtsp_st->dynamic_handler &&
359 rtsp_st->dynamic_handler->parse_sdp_a_line) {
360 rtsp_st->dynamic_handler->parse_sdp_a_line(s, i,
361 rtsp_st->dynamic_protocol_context, line);
362 }
363 }
364 }
365
366 static void sdp_parse_line(AVFormatContext *s, SDPParseState *s1,
367 int letter, const char *buf)
368 {
369 RTSPState *rt = s->priv_data;
370 char buf1[64], st_type[64];
371 const char *p;
372 enum AVMediaType codec_type;
373 int payload_type;
374 AVStream *st;
375 RTSPStream *rtsp_st;
376 RTSPSource *rtsp_src;
377 struct sockaddr_storage sdp_ip;
378 int ttl;
379
380 av_log(s, AV_LOG_TRACE, "sdp: %c='%s'\n", letter, buf);
381
382 p = buf;
383 if (s1->skip_media && letter != 'm')
384 return;
385 switch (letter) {
386 case 'c':
387 get_word(buf1, sizeof(buf1), &p);
388 if (strcmp(buf1, "IN") != 0)
389 return;
390 get_word(buf1, sizeof(buf1), &p);
391 if (strcmp(buf1, "IP4") && strcmp(buf1, "IP6"))
392 return;
393 get_word_sep(buf1, sizeof(buf1), "/", &p);
394 if (get_sockaddr(buf1, &sdp_ip))
395 return;
396 ttl = 16;
397 if (*p == '/') {
398 p++;
399 get_word_sep(buf1, sizeof(buf1), "/", &p);
400 ttl = atoi(buf1);
401 }
402 if (s->nb_streams == 0) {
403 s1->default_ip = sdp_ip;
404 s1->default_ttl = ttl;
405 } else {
406 rtsp_st = rt->rtsp_streams[rt->nb_rtsp_streams - 1];
407 rtsp_st->sdp_ip = sdp_ip;
408 rtsp_st->sdp_ttl = ttl;
409 }
410 break;
411 case 's':
412 av_dict_set(&s->metadata, "title", p, 0);
413 break;
414 case 'i':
415 if (s->nb_streams == 0) {
416 av_dict_set(&s->metadata, "comment", p, 0);
417 break;
418 }
419 break;
420 case 'm':
421 /* new stream */
422 s1->skip_media = 0;
423 s1->seen_fmtp = 0;
424 s1->seen_rtpmap = 0;
425 codec_type = AVMEDIA_TYPE_UNKNOWN;
426 get_word(st_type, sizeof(st_type), &p);
427 if (!strcmp(st_type, "audio")) {
428 codec_type = AVMEDIA_TYPE_AUDIO;
429 } else if (!strcmp(st_type, "video")) {
430 codec_type = AVMEDIA_TYPE_VIDEO;
431 } else if (!strcmp(st_type, "application") || !strcmp(st_type, "text")) {
432 codec_type = AVMEDIA_TYPE_DATA;
433 }
434 if (codec_type == AVMEDIA_TYPE_UNKNOWN || !(rt->media_type_mask & (1 << codec_type))) {
435 s1->skip_media = 1;
436 return;
437 }
438 rtsp_st = av_mallocz(sizeof(RTSPStream));
439 if (!rtsp_st)
440 return;
441 rtsp_st->stream_index = -1;
442 dynarray_add(&rt->rtsp_streams, &rt->nb_rtsp_streams, rtsp_st);
443
444 rtsp_st->sdp_ip = s1->default_ip;
445 rtsp_st->sdp_ttl = s1->default_ttl;
446
447 copy_default_source_addrs(s1->default_include_source_addrs,
448 s1->nb_default_include_source_addrs,
449 &rtsp_st->include_source_addrs,
450 &rtsp_st->nb_include_source_addrs);
451 copy_default_source_addrs(s1->default_exclude_source_addrs,
452 s1->nb_default_exclude_source_addrs,
453 &rtsp_st->exclude_source_addrs,
454 &rtsp_st->nb_exclude_source_addrs);
455
456 get_word(buf1, sizeof(buf1), &p); /* port */
457 rtsp_st->sdp_port = atoi(buf1);
458
459 get_word(buf1, sizeof(buf1), &p); /* protocol */
460 if (!strcmp(buf1, "udp"))
461 rt->transport = RTSP_TRANSPORT_RAW;
462 else if (strstr(buf1, "/AVPF") || strstr(buf1, "/SAVPF"))
463 rtsp_st->feedback = 1;
464
465 /* XXX: handle list of formats */
466 get_word(buf1, sizeof(buf1), &p); /* format list */
467 rtsp_st->sdp_payload_type = atoi(buf1);
468
469 if (!strcmp(ff_rtp_enc_name(rtsp_st->sdp_payload_type), "MP2T")) {
470 /* no corresponding stream */
471 if (rt->transport == RTSP_TRANSPORT_RAW) {
472 if (CONFIG_RTPDEC && !rt->ts)
473 rt->ts = ff_mpegts_parse_open(s);
474 } else {
475 RTPDynamicProtocolHandler *handler;
476 handler = ff_rtp_handler_find_by_id(
477 rtsp_st->sdp_payload_type, AVMEDIA_TYPE_DATA);
478 init_rtp_handler(handler, rtsp_st, NULL);
479 finalize_rtp_handler_init(s, rtsp_st, NULL);
480 }
481 } else if (rt->server_type == RTSP_SERVER_WMS &&
482 codec_type == AVMEDIA_TYPE_DATA) {
483 /* RTX stream, a stream that carries all the other actual
484 * audio/video streams. Don't expose this to the callers. */
485 } else {
486 st = avformat_new_stream(s, NULL);
487 if (!st)
488 return;
489 st->id = rt->nb_rtsp_streams - 1;
490 rtsp_st->stream_index = st->index;
491 st->codec->codec_type = codec_type;
492 if (rtsp_st->sdp_payload_type < RTP_PT_PRIVATE) {
493 RTPDynamicProtocolHandler *handler;
494 /* if standard payload type, we can find the codec right now */
495 ff_rtp_get_codec_info(st->codec, rtsp_st->sdp_payload_type);
496 if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO &&
497 st->codec->sample_rate > 0)
498 avpriv_set_pts_info(st, 32, 1, st->codec->sample_rate);
499 /* Even static payload types may need a custom depacketizer */
500 handler = ff_rtp_handler_find_by_id(
501 rtsp_st->sdp_payload_type, st->codec->codec_type);
502 init_rtp_handler(handler, rtsp_st, st);
503 finalize_rtp_handler_init(s, rtsp_st, st);
504 }
505 if (rt->default_lang[0])
506 av_dict_set(&st->metadata, "language", rt->default_lang, 0);
507 }
508 /* put a default control url */
509 av_strlcpy(rtsp_st->control_url, rt->control_uri,
510 sizeof(rtsp_st->control_url));
511 break;
512 case 'a':
513 if (av_strstart(p, "control:", &p)) {
514 if (s->nb_streams == 0) {
515 if (!strncmp(p, "rtsp://", 7))
516 av_strlcpy(rt->control_uri, p,
517 sizeof(rt->control_uri));
518 } else {
519 char proto[32];
520 /* get the control url */
521 rtsp_st = rt->rtsp_streams[rt->nb_rtsp_streams - 1];
522
523 /* XXX: may need to add full url resolution */
524 av_url_split(proto, sizeof(proto), NULL, 0, NULL, 0,
525 NULL, NULL, 0, p);
526 if (proto[0] == '\0') {
527 /* relative control URL */
528 if (rtsp_st->control_url[strlen(rtsp_st->control_url)-1]!='/')
529 av_strlcat(rtsp_st->control_url, "/",
530 sizeof(rtsp_st->control_url));
531 av_strlcat(rtsp_st->control_url, p,
532 sizeof(rtsp_st->control_url));
533 } else
534 av_strlcpy(rtsp_st->control_url, p,
535 sizeof(rtsp_st->control_url));
536 }
537 } else if (av_strstart(p, "rtpmap:", &p) && s->nb_streams > 0) {
538 /* NOTE: rtpmap is only supported AFTER the 'm=' tag */
539 get_word(buf1, sizeof(buf1), &p);
540 payload_type = atoi(buf1);
541 rtsp_st = rt->rtsp_streams[rt->nb_rtsp_streams - 1];
542 if (rtsp_st->stream_index >= 0) {
543 st = s->streams[rtsp_st->stream_index];
544 sdp_parse_rtpmap(s, st, rtsp_st, payload_type, p);
545 }
546 s1->seen_rtpmap = 1;
547 if (s1->seen_fmtp) {
548 parse_fmtp(s, rt, payload_type, s1->delayed_fmtp);
549 }
550 } else if (av_strstart(p, "fmtp:", &p) ||
551 av_strstart(p, "framesize:", &p)) {
552 // let dynamic protocol handlers have a stab at the line.
553 get_word(buf1, sizeof(buf1), &p);
554 payload_type = atoi(buf1);
555 if (s1->seen_rtpmap) {
556 parse_fmtp(s, rt, payload_type, buf);
557 } else {
558 s1->seen_fmtp = 1;
559 av_strlcpy(s1->delayed_fmtp, buf, sizeof(s1->delayed_fmtp));
560 }
561 } else if (av_strstart(p, "range:", &p)) {
562 int64_t start, end;
563
564 // this is so that seeking on a streamed file can work.
565 rtsp_parse_range_npt(p, &start, &end);
566 s->start_time = start;
567 /* AV_NOPTS_VALUE means live broadcast (and can't seek) */
568 s->duration = (end == AV_NOPTS_VALUE) ?
569 AV_NOPTS_VALUE : end - start;
570 } else if (av_strstart(p, "lang:", &p)) {
571 if (s->nb_streams > 0) {
572 get_word(buf1, sizeof(buf1), &p);
573 rtsp_st = rt->rtsp_streams[rt->nb_rtsp_streams - 1];
574 if (rtsp_st->stream_index >= 0) {
575 st = s->streams[rtsp_st->stream_index];
576 av_dict_set(&st->metadata, "language", buf1, 0);
577 }
578 } else
579 get_word(rt->default_lang, sizeof(rt->default_lang), &p);
580 } else if (av_strstart(p, "IsRealDataType:integer;",&p)) {
581 if (atoi(p) == 1)
582 rt->transport = RTSP_TRANSPORT_RDT;
583 } else if (av_strstart(p, "SampleRate:integer;", &p) &&
584 s->nb_streams > 0) {
585 st = s->streams[s->nb_streams - 1];
586 st->codec->sample_rate = atoi(p);
587 } else if (av_strstart(p, "crypto:", &p) && s->nb_streams > 0) {
588 // RFC 4568
589 rtsp_st = rt->rtsp_streams[rt->nb_rtsp_streams - 1];
590 get_word(buf1, sizeof(buf1), &p); // ignore tag
591 get_word(rtsp_st->crypto_suite, sizeof(rtsp_st->crypto_suite), &p);
592 p += strspn(p, SPACE_CHARS);
593 if (av_strstart(p, "inline:", &p))
594 get_word(rtsp_st->crypto_params, sizeof(rtsp_st->crypto_params), &p);
595 } else if (av_strstart(p, "source-filter:", &p)) {
596 int exclude = 0;
597 get_word(buf1, sizeof(buf1), &p);
598 if (strcmp(buf1, "incl") && strcmp(buf1, "excl"))
599 return;
600 exclude = !strcmp(buf1, "excl");
601
602 get_word(buf1, sizeof(buf1), &p);
603 if (strcmp(buf1, "IN") != 0)
604 return;
605 get_word(buf1, sizeof(buf1), &p);
606 if (strcmp(buf1, "IP4") && strcmp(buf1, "IP6") && strcmp(buf1, "*"))
607 return;
608 // not checking that the destination address actually matches or is wildcard
609 get_word(buf1, sizeof(buf1), &p);
610
611 while (*p != '\0') {
612 rtsp_src = av_mallocz(sizeof(*rtsp_src));
613 if (!rtsp_src)
614 return;
615 get_word(rtsp_src->addr, sizeof(rtsp_src->addr), &p);
616 if (exclude) {
617 if (s->nb_streams == 0) {
618 dynarray_add(&s1->default_exclude_source_addrs, &s1->nb_default_exclude_source_addrs, rtsp_src);
619 } else {
620 rtsp_st = rt->rtsp_streams[rt->nb_rtsp_streams - 1];
621 dynarray_add(&rtsp_st->exclude_source_addrs, &rtsp_st->nb_exclude_source_addrs, rtsp_src);
622 }
623 } else {
624 if (s->nb_streams == 0) {
625 dynarray_add(&s1->default_include_source_addrs, &s1->nb_default_include_source_addrs, rtsp_src);
626 } else {
627 rtsp_st = rt->rtsp_streams[rt->nb_rtsp_streams - 1];
628 dynarray_add(&rtsp_st->include_source_addrs, &rtsp_st->nb_include_source_addrs, rtsp_src);
629 }
630 }
631 }
632 } else {
633 if (rt->server_type == RTSP_SERVER_WMS)
634 ff_wms_parse_sdp_a_line(s, p);
635 if (s->nb_streams > 0) {
636 rtsp_st = rt->rtsp_streams[rt->nb_rtsp_streams - 1];
637
638 if (rt->server_type == RTSP_SERVER_REAL)
639 ff_real_parse_sdp_a_line(s, rtsp_st->stream_index, p);
640
641 if (rtsp_st->dynamic_handler &&
642 rtsp_st->dynamic_handler->parse_sdp_a_line)
643 rtsp_st->dynamic_handler->parse_sdp_a_line(s,
644 rtsp_st->stream_index,
645 rtsp_st->dynamic_protocol_context, buf);
646 }
647 }
648 break;
649 }
650 }
651
652 int ff_sdp_parse(AVFormatContext *s, const char *content)
653 {
654 RTSPState *rt = s->priv_data;
655 const char *p;
656 int letter, i;
657 /* Some SDP lines, particularly for Realmedia or ASF RTSP streams,
658 * contain long SDP lines containing complete ASF Headers (several
659 * kB) or arrays of MDPR (RM stream descriptor) headers plus
660 * "rulebooks" describing their properties. Therefore, the SDP line
661 * buffer is large.
662 *
663 * The Vorbis FMTP line can be up to 16KB - see xiph_parse_sdp_line
664 * in rtpdec_xiph.c. */
665 char buf[16384], *q;
666 SDPParseState sdp_parse_state = { { 0 } }, *s1 = &sdp_parse_state;
667
668 p = content;
669 for (;;) {
670 p += strspn(p, SPACE_CHARS);
671 letter = *p;
672 if (letter == '\0')
673 break;
674 p++;
675 if (*p != '=')
676 goto next_line;
677 p++;
678 /* get the content */
679 q = buf;
680 while (*p != '\n' && *p != '\r' && *p != '\0') {
681 if ((q - buf) < sizeof(buf) - 1)
682 *q++ = *p;
683 p++;
684 }
685 *q = '\0';
686 sdp_parse_line(s, s1, letter, buf);
687 next_line:
688 while (*p != '\n' && *p != '\0')
689 p++;
690 if (*p == '\n')
691 p++;
692 }
693
694 for (i = 0; i < s1->nb_default_include_source_addrs; i++)
695 av_free(s1->default_include_source_addrs[i]);
696 av_freep(&s1->default_include_source_addrs);
697 for (i = 0; i < s1->nb_default_exclude_source_addrs; i++)
698 av_free(s1->default_exclude_source_addrs[i]);
699 av_freep(&s1->default_exclude_source_addrs);
700
701 rt->p = av_malloc(sizeof(struct pollfd)*2*(rt->nb_rtsp_streams+1));
702 if (!rt->p) return AVERROR(ENOMEM);
703 return 0;
704 }
705 #endif /* CONFIG_RTPDEC */
706
707 void ff_rtsp_undo_setup(AVFormatContext *s, int send_packets)
708 {
709 RTSPState *rt = s->priv_data;
710 int i;
711
712 for (i = 0; i < rt->nb_rtsp_streams; i++) {
713 RTSPStream *rtsp_st = rt->rtsp_streams[i];
714 if (!rtsp_st)
715 continue;
716 if (rtsp_st->transport_priv) {
717 if (s->oformat) {
718 AVFormatContext *rtpctx = rtsp_st->transport_priv;
719 av_write_trailer(rtpctx);
720 if (rt->lower_transport == RTSP_LOWER_TRANSPORT_TCP) {
721 if (CONFIG_RTSP_MUXER && rtpctx->pb && send_packets)
722 ff_rtsp_tcp_write_packet(s, rtsp_st);
723 ffio_free_dyn_buf(&rtpctx->pb);
724 } else {
725 avio_close(rtpctx->pb);
726 }
727 avformat_free_context(rtpctx);
728 } else if (CONFIG_RTPDEC && rt->transport == RTSP_TRANSPORT_RDT)
729 ff_rdt_parse_close(rtsp_st->transport_priv);
730 else if (CONFIG_RTPDEC && rt->transport == RTSP_TRANSPORT_RTP)
731 ff_rtp_parse_close(rtsp_st->transport_priv);
732 }
733 rtsp_st->transport_priv = NULL;
734 if (rtsp_st->rtp_handle)
735 ffurl_close(rtsp_st->rtp_handle);
736 rtsp_st->rtp_handle = NULL;
737 }
738 }
739
740 /* close and free RTSP streams */
741 void ff_rtsp_close_streams(AVFormatContext *s)
742 {
743 RTSPState *rt = s->priv_data;
744 int i, j;
745 RTSPStream *rtsp_st;
746
747 ff_rtsp_undo_setup(s, 0);
748 for (i = 0; i < rt->nb_rtsp_streams; i++) {
749 rtsp_st = rt->rtsp_streams[i];
750 if (rtsp_st) {
751 if (rtsp_st->dynamic_handler && rtsp_st->dynamic_protocol_context) {
752 if (rtsp_st->dynamic_handler->close)
753 rtsp_st->dynamic_handler->close(
754 rtsp_st->dynamic_protocol_context);
755 av_free(rtsp_st->dynamic_protocol_context);
756 }
757 for (j = 0; j < rtsp_st->nb_include_source_addrs; j++)
758 av_free(rtsp_st->include_source_addrs[j]);
759 av_freep(&rtsp_st->include_source_addrs);
760 for (j = 0; j < rtsp_st->nb_exclude_source_addrs; j++)
761 av_free(rtsp_st->exclude_source_addrs[j]);
762 av_freep(&rtsp_st->exclude_source_addrs);
763
764 av_free(rtsp_st);
765 }
766 }
767 av_free(rt->rtsp_streams);
768 if (rt->asf_ctx) {
769 avformat_close_input(&rt->asf_ctx);
770 }
771 if (CONFIG_RTPDEC && rt->ts)
772 ff_mpegts_parse_close(rt->ts);
773 av_free(rt->p);
774 av_free(rt->recvbuf);
775 }
776
777 int ff_rtsp_open_transport_ctx(AVFormatContext *s, RTSPStream *rtsp_st)
778 {
779 RTSPState *rt = s->priv_data;
780 AVStream *st = NULL;
781 int reordering_queue_size = rt->reordering_queue_size;
782 if (reordering_queue_size < 0) {
783 if (rt->lower_transport == RTSP_LOWER_TRANSPORT_TCP || !s->max_delay)
784 reordering_queue_size = 0;
785 else
786 reordering_queue_size = RTP_REORDER_QUEUE_DEFAULT_SIZE;
787 }
788
789 /* open the RTP context */
790 if (rtsp_st->stream_index >= 0)
791 st = s->streams[rtsp_st->stream_index];
792 if (!st)
793 s->ctx_flags |= AVFMTCTX_NOHEADER;
794
795 if (CONFIG_RTSP_MUXER && s->oformat) {
796 int ret = ff_rtp_chain_mux_open((AVFormatContext **)&rtsp_st->transport_priv,
797 s, st, rtsp_st->rtp_handle,
798 RTSP_TCP_MAX_PACKET_SIZE,
799 rtsp_st->stream_index);
800 /* Ownership of rtp_handle is passed to the rtp mux context */
801 rtsp_st->rtp_handle = NULL;
802 if (ret < 0)
803 return ret;
804 st->time_base = ((AVFormatContext*)rtsp_st->transport_priv)->streams[0]->time_base;
805 } else if (rt->transport == RTSP_TRANSPORT_RAW) {
806 return 0; // Don't need to open any parser here
807 } else if (CONFIG_RTPDEC && rt->transport == RTSP_TRANSPORT_RDT)
808 rtsp_st->transport_priv = ff_rdt_parse_open(s, st->index,
809 rtsp_st->dynamic_protocol_context,
810 rtsp_st->dynamic_handler);
811 else if (CONFIG_RTPDEC)
812 rtsp_st->transport_priv = ff_rtp_parse_open(s, st,
813 rtsp_st->sdp_payload_type,
814 reordering_queue_size);
815
816 if (!rtsp_st->transport_priv) {
817 return AVERROR(ENOMEM);
818 } else if (CONFIG_RTPDEC && rt->transport == RTSP_TRANSPORT_RTP) {
819 if (rtsp_st->dynamic_handler) {
820 ff_rtp_parse_set_dynamic_protocol(rtsp_st->transport_priv,
821 rtsp_st->dynamic_protocol_context,
822 rtsp_st->dynamic_handler);
823 }
824 if (rtsp_st->crypto_suite[0])
825 ff_rtp_parse_set_crypto(rtsp_st->transport_priv,
826 rtsp_st->crypto_suite,
827 rtsp_st->crypto_params);
828 }
829
830 return 0;
831 }
832
833 #if CONFIG_RTSP_DEMUXER || CONFIG_RTSP_MUXER
834 static void rtsp_parse_range(int *min_ptr, int *max_ptr, const char **pp)
835 {
836 const char *q;
837 char *p;
838 int v;
839
840 q = *pp;
841 q += strspn(q, SPACE_CHARS);
842 v = strtol(q, &p, 10);
843 if (*p == '-') {
844 p++;
845 *min_ptr = v;
846 v = strtol(p, &p, 10);
847 *max_ptr = v;
848 } else {
849 *min_ptr = v;
850 *max_ptr = v;
851 }
852 *pp = p;
853 }
854
855 /* XXX: only one transport specification is parsed */
856 static void rtsp_parse_transport(RTSPMessageHeader *reply, const char *p)
857 {
858 char transport_protocol[16];
859 char profile[16];
860 char lower_transport[16];
861 char parameter[16];
862 RTSPTransportField *th;
863 char buf[256];
864
865 reply->nb_transports = 0;
866
867 for (;;) {
868 p += strspn(p, SPACE_CHARS);
869 if (*p == '\0')
870 break;
871
872 th = &reply->transports[reply->nb_transports];
873
874 get_word_sep(transport_protocol, sizeof(transport_protocol),
875 "/", &p);
876 if (!av_strcasecmp (transport_protocol, "rtp")) {
877 get_word_sep(profile, sizeof(profile), "/;,", &p);
878 lower_transport[0] = '\0';
879 /* rtp/avp/<protocol> */
880 if (*p == '/') {
881 get_word_sep(lower_transport, sizeof(lower_transport),
882 ";,", &p);
883 }
884 th->transport = RTSP_TRANSPORT_RTP;
885 } else if (!av_strcasecmp (transport_protocol, "x-pn-tng") ||
886 !av_strcasecmp (transport_protocol, "x-real-rdt")) {
887 /* x-pn-tng/<protocol> */
888 get_word_sep(lower_transport, sizeof(lower_transport), "/;,", &p);
889 profile[0] = '\0';
890 th->transport = RTSP_TRANSPORT_RDT;
891 } else if (!av_strcasecmp(transport_protocol, "raw")) {
892 get_word_sep(profile, sizeof(profile), "/;,", &p);
893 lower_transport[0] = '\0';
894 /* raw/raw/<protocol> */
895 if (*p == '/') {
896 get_word_sep(lower_transport, sizeof(lower_transport),
897 ";,", &p);
898 }
899 th->transport = RTSP_TRANSPORT_RAW;
900 }
901 if (!av_strcasecmp(lower_transport, "TCP"))
902 th->lower_transport = RTSP_LOWER_TRANSPORT_TCP;
903 else
904 th->lower_transport = RTSP_LOWER_TRANSPORT_UDP;
905
906 if (*p == ';')
907 p++;
908 /* get each parameter */
909 while (*p != '\0' && *p != ',') {
910 get_word_sep(parameter, sizeof(parameter), "=;,", &p);
911 if (!strcmp(parameter, "port")) {
912 if (*p == '=') {
913 p++;
914 rtsp_parse_range(&th->port_min, &th->port_max, &p);
915 }
916 } else if (!strcmp(parameter, "client_port")) {
917 if (*p == '=') {
918 p++;
919 rtsp_parse_range(&th->client_port_min,
920 &th->client_port_max, &p);
921 }
922 } else if (!strcmp(parameter, "server_port")) {
923 if (*p == '=') {
924 p++;
925 rtsp_parse_range(&th->server_port_min,
926 &th->server_port_max, &p);
927 }
928 } else if (!strcmp(parameter, "interleaved")) {
929 if (*p == '=') {
930 p++;
931 rtsp_parse_range(&th->interleaved_min,
932 &th->interleaved_max, &p);
933 }
934 } else if (!strcmp(parameter, "multicast")) {
935 if (th->lower_transport == RTSP_LOWER_TRANSPORT_UDP)
936 th->lower_transport = RTSP_LOWER_TRANSPORT_UDP_MULTICAST;
937 } else if (!strcmp(parameter, "ttl")) {
938 if (*p == '=') {
939 char *end;
940 p++;
941 th->ttl = strtol(p, &end, 10);
942 p = end;
943 }
944 } else if (!strcmp(parameter, "destination")) {
945 if (*p == '=') {
946 p++;
947 get_word_sep(buf, sizeof(buf), ";,", &p);
948 get_sockaddr(buf, &th->destination);
949 }
950 } else if (!strcmp(parameter, "source")) {
951 if (*p == '=') {
952 p++;
953 get_word_sep(buf, sizeof(buf), ";,", &p);
954 av_strlcpy(th->source, buf, sizeof(th->source));
955 }
956 } else if (!strcmp(parameter, "mode")) {
957 if (*p == '=') {
958 p++;
959 get_word_sep(buf, sizeof(buf), ";, ", &p);
960 if (!strcmp(buf, "record") ||
961 !strcmp(buf, "receive"))
962 th->mode_record = 1;
963 }
964 }
965
966 while (*p != ';' && *p != '\0' && *p != ',')
967 p++;
968 if (*p == ';')
969 p++;
970 }
971 if (*p == ',')
972 p++;
973
974 reply->nb_transports++;
975 if (reply->nb_transports >= RTSP_MAX_TRANSPORTS)
976 break;
977 }
978 }
979
980 static void handle_rtp_info(RTSPState *rt, const char *url,
981 uint32_t seq, uint32_t rtptime)
982 {
983 int i;
984 if (!rtptime || !url[0])
985 return;
986 if (rt->transport != RTSP_TRANSPORT_RTP)
987 return;
988 for (i = 0; i < rt->nb_rtsp_streams; i++) {
989 RTSPStream *rtsp_st = rt->rtsp_streams[i];
990 RTPDemuxContext *rtpctx = rtsp_st->transport_priv;
991 if (!rtpctx)
992 continue;
993 if (!strcmp(rtsp_st->control_url, url)) {
994 rtpctx->base_timestamp = rtptime;
995 break;
996 }
997 }
998 }
999
1000 static void rtsp_parse_rtp_info(RTSPState *rt, const char *p)
1001 {
1002 int read = 0;
1003 char key[20], value[1024], url[1024] = "";
1004 uint32_t seq = 0, rtptime = 0;
1005
1006 for (;;) {
1007 p += strspn(p, SPACE_CHARS);
1008 if (!*p)
1009 break;
1010 get_word_sep(key, sizeof(key), "=", &p);
1011 if (*p != '=')
1012 break;
1013 p++;
1014 get_word_sep(value, sizeof(value), ";, ", &p);
1015 read++;
1016 if (!strcmp(key, "url"))
1017 av_strlcpy(url, value, sizeof(url));
1018 else if (!strcmp(key, "seq"))
1019 seq = strtoul(value, NULL, 10);
1020 else if (!strcmp(key, "rtptime"))
1021 rtptime = strtoul(value, NULL, 10);
1022 if (*p == ',') {
1023 handle_rtp_info(rt, url, seq, rtptime);
1024 url[0] = '\0';
1025 seq = rtptime = 0;
1026 read = 0;
1027 }
1028 if (*p)
1029 p++;
1030 }
1031 if (read > 0)
1032 handle_rtp_info(rt, url, seq, rtptime);
1033 }
1034
1035 void ff_rtsp_parse_line(RTSPMessageHeader *reply, const char *buf,
1036 RTSPState *rt, const char *method)
1037 {
1038 const char *p;
1039
1040 /* NOTE: we do case independent match for broken servers */
1041 p = buf;
1042 if (av_stristart(p, "Session:", &p)) {
1043 int t;
1044 get_word_sep(reply->session_id, sizeof(reply->session_id), ";", &p);
1045 if (av_stristart(p, ";timeout=", &p) &&
1046 (t = strtol(p, NULL, 10)) > 0) {
1047 reply->timeout = t;
1048 }
1049 } else if (av_stristart(p, "Content-Length:", &p)) {
1050 reply->content_length = strtol(p, NULL, 10);
1051 } else if (av_stristart(p, "Transport:", &p)) {
1052 rtsp_parse_transport(reply, p);
1053 } else if (av_stristart(p, "CSeq:", &p)) {
1054 reply->seq = strtol(p, NULL, 10);
1055 } else if (av_stristart(p, "Range:", &p)) {
1056 rtsp_parse_range_npt(p, &reply->range_start, &reply->range_end);
1057 } else if (av_stristart(p, "RealChallenge1:", &p)) {
1058 p += strspn(p, SPACE_CHARS);
1059 av_strlcpy(reply->real_challenge, p, sizeof(reply->real_challenge));
1060 } else if (av_stristart(p, "Server:", &p)) {
1061 p += strspn(p, SPACE_CHARS);
1062 av_strlcpy(reply->server, p, sizeof(reply->server));
1063 } else if (av_stristart(p, "Notice:", &p) ||
1064 av_stristart(p, "X-Notice:", &p)) {
1065 reply->notice = strtol(p, NULL, 10);
1066 } else if (av_stristart(p, "Location:", &p)) {
1067 p += strspn(p, SPACE_CHARS);
1068 av_strlcpy(reply->location, p , sizeof(reply->location));
1069 } else if (av_stristart(p, "WWW-Authenticate:", &p) && rt) {
1070 p += strspn(p, SPACE_CHARS);
1071 ff_http_auth_handle_header(&rt->auth_state, "WWW-Authenticate", p);
1072 } else if (av_stristart(p, "Authentication-Info:", &p) && rt) {
1073 p += strspn(p, SPACE_CHARS);
1074 ff_http_auth_handle_header(&rt->auth_state, "Authentication-Info", p);
1075 } else if (av_stristart(p, "Content-Base:", &p) && rt) {
1076 p += strspn(p, SPACE_CHARS);
1077 if (method && !strcmp(method, "DESCRIBE"))
1078 av_strlcpy(rt->control_uri, p , sizeof(rt->control_uri));
1079 } else if (av_stristart(p, "RTP-Info:", &p) && rt) {
1080 p += strspn(p, SPACE_CHARS);
1081 if (method && !strcmp(method, "PLAY"))
1082 rtsp_parse_rtp_info(rt, p);
1083 } else if (av_stristart(p, "Public:", &p) && rt) {
1084 if (strstr(p, "GET_PARAMETER") &&
1085 method && !strcmp(method, "OPTIONS"))
1086 rt->get_parameter_supported = 1;
1087 } else if (av_stristart(p, "x-Accept-Dynamic-Rate:", &p) && rt) {
1088 p += strspn(p, SPACE_CHARS);
1089 rt->accept_dynamic_rate = atoi(p);
1090 } else if (av_stristart(p, "Content-Type:", &p)) {
1091 p += strspn(p, SPACE_CHARS);
1092 av_strlcpy(reply->content_type, p, sizeof(reply->content_type));
1093 }
1094 }
1095
1096 /* skip a RTP/TCP interleaved packet */
1097 void ff_rtsp_skip_packet(AVFormatContext *s)
1098 {
1099 RTSPState *rt = s->priv_data;
1100 int ret, len, len1;
1101 uint8_t buf[1024];
1102
1103 ret = ffurl_read_complete(rt->rtsp_hd, buf, 3);
1104 if (ret != 3)
1105 return;
1106 len = AV_RB16(buf + 1);
1107
1108 av_log(s, AV_LOG_TRACE, "skipping RTP packet len=%d\n", len);
1109
1110 /* skip payload */
1111 while (len > 0) {
1112 len1 = len;
1113 if (len1 > sizeof(buf))
1114 len1 = sizeof(buf);
1115 ret = ffurl_read_complete(rt->rtsp_hd, buf, len1);
1116 if (ret != len1)
1117 return;
1118 len -= len1;
1119 }
1120 }
1121
1122 int ff_rtsp_read_reply(AVFormatContext *s, RTSPMessageHeader *reply,
1123 unsigned char **content_ptr,
1124 int return_on_interleaved_data, const char *method)
1125 {
1126 RTSPState *rt = s->priv_data;
1127 char buf[4096], buf1[1024], *q;
1128 unsigned char ch;
1129 const char *p;
1130 int ret, content_length, line_count = 0, request = 0;
1131 unsigned char *content = NULL;
1132
1133 start:
1134 line_count = 0;
1135 request = 0;
1136 content = NULL;
1137 memset(reply, 0, sizeof(*reply));
1138
1139 /* parse reply (XXX: use buffers) */
1140 rt->last_reply[0] = '\0';
1141 for (;;) {
1142 q = buf;
1143 for (;;) {
1144 ret = ffurl_read_complete(rt->rtsp_hd, &ch, 1);
1145 av_log(s, AV_LOG_TRACE, "ret=%d c=%02x [%c]\n", ret, ch, ch);
1146 if (ret != 1)
1147 return AVERROR_EOF;
1148 if (ch == '\n')
1149 break;
1150 if (ch == '$' && q == buf) {
1151 if (return_on_interleaved_data) {
1152 return 1;
1153 } else
1154 ff_rtsp_skip_packet(s);
1155 } else if (ch != '\r') {
1156 if ((q - buf) < sizeof(buf) - 1)
1157 *q++ = ch;
1158 }
1159 }
1160 *q = '\0';
1161
1162 av_log(s, AV_LOG_TRACE, "line='%s'\n", buf);
1163
1164 /* test if last line */
1165 if (buf[0] == '\0')
1166 break;
1167 p = buf;
1168 if (line_count == 0) {
1169 /* get reply code */
1170 get_word(buf1, sizeof(buf1), &p);
1171 if (!strncmp(buf1, "RTSP/", 5)) {
1172 get_word(buf1, sizeof(buf1), &p);
1173 reply->status_code = atoi(buf1);
1174 av_strlcpy(reply->reason, p, sizeof(reply->reason));
1175 } else {
1176 av_strlcpy(reply->reason, buf1, sizeof(reply->reason)); // method
1177 get_word(buf1, sizeof(buf1), &p); // object
1178 request = 1;
1179 }
1180 } else {
1181 ff_rtsp_parse_line(reply, p, rt, method);
1182 av_strlcat(rt->last_reply, p, sizeof(rt->last_reply));
1183 av_strlcat(rt->last_reply, "\n", sizeof(rt->last_reply));
1184 }
1185 line_count++;
1186 }
1187
1188 if (rt->session_id[0] == '\0' && reply->session_id[0] != '\0' && !request)
1189 av_strlcpy(rt->session_id, reply->session_id, sizeof(rt->session_id));
1190
1191 content_length = reply->content_length;
1192 if (content_length > 0) {
1193 /* leave some room for a trailing '\0' (useful for simple parsing) */
1194 content = av_malloc(content_length + 1);
1195 if (!content)
1196 return AVERROR(ENOMEM);
1197 ffurl_read_complete(rt->rtsp_hd, content, content_length);
1198 content[content_length] = '\0';
1199 }
1200 if (content_ptr)
1201 *content_ptr = content;
1202 else
1203 av_free(content);
1204
1205 if (request) {
1206 char buf[1024];
1207 char base64buf[AV_BASE64_SIZE(sizeof(buf))];
1208 const char* ptr = buf;
1209
1210 if (!strcmp(reply->reason, "OPTIONS")) {
1211 snprintf(buf, sizeof(buf), "RTSP/1.0 200 OK\r\n");
1212 if (reply->seq)
1213 av_strlcatf(buf, sizeof(buf), "CSeq: %d\r\n", reply->seq);
1214 if (reply->session_id[0])
1215 av_strlcatf(buf, sizeof(buf), "Session: %s\r\n",
1216 reply->session_id);
1217 } else {
1218 snprintf(buf, sizeof(buf), "RTSP/1.0 501 Not Implemented\r\n");
1219 }
1220 av_strlcat(buf, "\r\n", sizeof(buf));
1221
1222 if (rt->control_transport == RTSP_MODE_TUNNEL) {
1223 av_base64_encode(base64buf, sizeof(base64buf), buf, strlen(buf));
1224 ptr = base64buf;
1225 }
1226 ffurl_write(rt->rtsp_hd_out, ptr, strlen(ptr));
1227
1228 rt->last_cmd_time = av_gettime_relative();
1229 /* Even if the request from the server had data, it is not the data
1230 * that the caller wants or expects. The memory could also be leaked
1231 * if the actual following reply has content data. */
1232 if (content_ptr)
1233 av_freep(content_ptr);
1234 /* If method is set, this is called from ff_rtsp_send_cmd,
1235 * where a reply to exactly this request is awaited. For
1236 * callers from within packet receiving, we just want to
1237 * return to the caller and go back to receiving packets. */
1238 if (method)
1239 goto start;
1240 return 0;
1241 }
1242
1243 if (rt->seq != reply->seq) {
1244 av_log(s, AV_LOG_WARNING, "CSeq %d expected, %d received.\n",
1245 rt->seq, reply->seq);
1246 }
1247
1248 /* EOS */
1249 if (reply->notice == 2101 /* End-of-Stream Reached */ ||
1250 reply->notice == 2104 /* Start-of-Stream Reached */ ||
1251 reply->notice == 2306 /* Continuous Feed Terminated */) {
1252 rt->state = RTSP_STATE_IDLE;
1253 } else if (reply->notice >= 4400 && reply->notice < 5500) {
1254 return AVERROR(EIO); /* data or server error */
1255 } else if (reply->notice == 2401 /* Ticket Expired */ ||
1256 (reply->notice >= 5500 && reply->notice < 5600) /* end of term */ )
1257 return AVERROR(EPERM);
1258
1259 return 0;
1260 }
1261
1262 /**
1263 * Send a command to the RTSP server without waiting for the reply.
1264 *
1265 * @param s RTSP (de)muxer context
1266 * @param method the method for the request
1267 * @param url the target url for the request
1268 * @param headers extra header lines to include in the request
1269 * @param send_content if non-null, the data to send as request body content
1270 * @param send_content_length the length of the send_content data, or 0 if
1271 * send_content is null
1272 *
1273 * @return zero if success, nonzero otherwise
1274 */
1275 static int rtsp_send_cmd_with_content_async(AVFormatContext *s,
1276 const char *method, const char *url,
1277 const char *headers,
1278 const unsigned char *send_content,
1279 int send_content_length)
1280 {
1281 RTSPState *rt = s->priv_data;
1282 char buf[4096], *out_buf;
1283 char base64buf[AV_BASE64_SIZE(sizeof(buf))];
1284
1285 /* Add in RTSP headers */
1286 out_buf = buf;
1287 rt->seq++;
1288 snprintf(buf, sizeof(buf), "%s %s RTSP/1.0\r\n", method, url);
1289 if (headers)
1290 av_strlcat(buf, headers, sizeof(buf));
1291 av_strlcatf(buf, sizeof(buf), "CSeq: %d\r\n", rt->seq);
1292 av_strlcatf(buf, sizeof(buf), "User-Agent: %s\r\n", LIBAVFORMAT_IDENT);
1293 if (rt->session_id[0] != '\0' && (!headers ||
1294 !strstr(headers, "\nIf-Match:"))) {
1295 av_strlcatf(buf, sizeof(buf), "Session: %s\r\n", rt->session_id);
1296 }
1297 if (rt->auth[0]) {
1298 char *str = ff_http_auth_create_response(&rt->auth_state,
1299 rt->auth, url, method);
1300 if (str)
1301 av_strlcat(buf, str, sizeof(buf));
1302 av_free(str);
1303 }
1304 if (send_content_length > 0 && send_content)
1305 av_strlcatf(buf, sizeof(buf), "Content-Length: %d\r\n", send_content_length);
1306 av_strlcat(buf, "\r\n", sizeof(buf));
1307
1308 /* base64 encode rtsp if tunneling */
1309 if (rt->control_transport == RTSP_MODE_TUNNEL) {
1310 av_base64_encode(base64buf, sizeof(base64buf), buf, strlen(buf));
1311 out_buf = base64buf;
1312 }
1313
1314 av_log(s, AV_LOG_TRACE, "Sending:\n%s--\n", buf);
1315
1316 ffurl_write(rt->rtsp_hd_out, out_buf, strlen(out_buf));
1317 if (send_content_length > 0 && send_content) {
1318 if (rt->control_transport == RTSP_MODE_TUNNEL) {
1319 av_log(s, AV_LOG_ERROR, "tunneling of RTSP requests "
1320 "with content data not supported\n");
1321 return AVERROR_PATCHWELCOME;
1322 }
1323 ffurl_write(rt->rtsp_hd_out, send_content, send_content_length);
1324 }
1325 rt->last_cmd_time = av_gettime_relative();
1326
1327 return 0;
1328 }
1329
1330 int ff_rtsp_send_cmd_async(AVFormatContext *s, const char *method,
1331 const char *url, const char *headers)
1332 {
1333 return rtsp_send_cmd_with_content_async(s, method, url, headers, NULL, 0);
1334 }
1335
1336 int ff_rtsp_send_cmd(AVFormatContext *s, const char *method, const char *url,
1337 const char *headers, RTSPMessageHeader *reply,
1338 unsigned char **content_ptr)
1339 {
1340 return ff_rtsp_send_cmd_with_content(s, method, url, headers, reply,
1341 content_ptr, NULL, 0);
1342 }
1343
1344 int ff_rtsp_send_cmd_with_content(AVFormatContext *s,
1345 const char *method, const char *url,
1346 const char *header,
1347 RTSPMessageHeader *reply,
1348 unsigned char **content_ptr,
1349 const unsigned char *send_content,
1350 int send_content_length)
1351 {
1352 RTSPState *rt = s->priv_data;
1353 HTTPAuthType cur_auth_type;
1354 int ret, attempts = 0;
1355
1356 retry:
1357 cur_auth_type = rt->auth_state.auth_type;
1358 if ((ret = rtsp_send_cmd_with_content_async(s, method, url, header,
1359 send_content,
1360 send_content_length)))
1361 return ret;
1362
1363 if ((ret = ff_rtsp_read_reply(s, reply, content_ptr, 0, method) ) < 0)
1364 return ret;
1365 attempts++;
1366
1367 if (reply->status_code == 401 &&
1368 (cur_auth_type == HTTP_AUTH_NONE || rt->auth_state.stale) &&
1369 rt->auth_state.auth_type != HTTP_AUTH_NONE && attempts < 2)
1370 goto retry;
1371
1372 if (reply->status_code > 400){
1373 av_log(s, AV_LOG_ERROR, "method %s failed: %d%s\n",
1374 method,
1375 reply->status_code,
1376 reply->reason);
1377 av_log(s, AV_LOG_DEBUG, "%s\n", rt->last_reply);
1378 }
1379
1380 return 0;
1381 }
1382
1383 int ff_rtsp_make_setup_request(AVFormatContext *s, const char *host, int port,
1384 int lower_transport, const char *real_challenge)
1385 {
1386 RTSPState *rt = s->priv_data;
1387 int rtx = 0, j, i, err, interleave = 0, port_off;
1388 RTSPStream *rtsp_st;
1389 RTSPMessageHeader reply1, *reply = &reply1;
1390 char cmd[2048];
1391 const char *trans_pref;
1392
1393 if (rt->transport == RTSP_TRANSPORT_RDT)
1394 trans_pref = "x-pn-tng";
1395 else if (rt->transport == RTSP_TRANSPORT_RAW)
1396 trans_pref = "RAW/RAW";
1397 else
1398 trans_pref = "RTP/AVP";
1399
1400 /* default timeout: 1 minute */
1401 rt->timeout = 60;
1402
1403 /* for each stream, make the setup request */
1404 /* XXX: we assume the same server is used for the control of each
1405 * RTSP stream */
1406
1407 /* Choose a random starting offset within the first half of the
1408 * port range, to allow for a number of ports to try even if the offset
1409 * happens to be at the end of the random range. */
1410 port_off = av_get_random_seed() % ((rt->rtp_port_max - rt->rtp_port_min)/2);
1411 /* even random offset */
1412 port_off -= port_off & 0x01;
1413
1414 for (j = rt->rtp_port_min + port_off, i = 0; i < rt->nb_rtsp_streams; ++i) {
1415 char transport[2048];
1416
1417 /*
1418 * WMS serves all UDP data over a single connection, the RTX, which
1419 * isn't necessarily the first in the SDP but has to be the first
1420 * to be set up, else the second/third SETUP will fail with a 461.
1421 */
1422 if (lower_transport == RTSP_LOWER_TRANSPORT_UDP &&
1423 rt->server_type == RTSP_SERVER_WMS) {
1424 if (i == 0) {
1425 /* rtx first */
1426 for (rtx = 0; rtx < rt->nb_rtsp_streams; rtx++) {
1427 int len = strlen(rt->rtsp_streams[rtx]->control_url);
1428 if (len >= 4 &&
1429 !strcmp(rt->rtsp_streams[rtx]->control_url + len - 4,
1430 "/rtx"))
1431 break;
1432 }
1433 if (rtx == rt->nb_rtsp_streams)
1434 return -1; /* no RTX found */
1435 rtsp_st = rt->rtsp_streams[rtx];
1436 } else
1437 rtsp_st = rt->rtsp_streams[i > rtx ? i : i - 1];
1438 } else
1439 rtsp_st = rt->rtsp_streams[i];
1440
1441 /* RTP/UDP */
1442 if (lower_transport == RTSP_LOWER_TRANSPORT_UDP) {
1443 char buf[256];
1444
1445 if (rt->server_type == RTSP_SERVER_WMS && i > 1) {
1446 port = reply->transports[0].client_port_min;
1447 goto have_port;
1448 }
1449
1450 /* first try in specified port range */
1451 while (j <= rt->rtp_port_max) {
1452 AVDictionary *opts = map_to_opts(rt);
1453
1454 ff_url_join(buf, sizeof(buf), "rtp", NULL, host, -1,
1455 "?localport=%d", j);
1456 /* we will use two ports per rtp stream (rtp and rtcp) */
1457 j += 2;
1458 err = ffurl_open(&rtsp_st->rtp_handle, buf, AVIO_FLAG_READ_WRITE,
1459 &s->interrupt_callback, &opts);
1460
1461 av_dict_free(&opts);
1462
1463 if (!err)
1464 goto rtp_opened;
1465 }
1466
1467 av_log(s, AV_LOG_ERROR, "Unable to open an input RTP port\n");
1468 err = AVERROR(EIO);
1469 goto fail;
1470
1471 rtp_opened:
1472 port = ff_rtp_get_local_rtp_port(rtsp_st->rtp_handle);
1473 have_port:
1474 snprintf(transport, sizeof(transport) - 1,
1475 "%s/UDP;", trans_pref);
1476 if (rt->server_type != RTSP_SERVER_REAL)
1477 av_strlcat(transport, "unicast;", sizeof(transport));
1478 av_strlcatf(transport, sizeof(transport),
1479 "client_port=%d", port);
1480 if (rt->transport == RTSP_TRANSPORT_RTP &&
1481 !(rt->server_type == RTSP_SERVER_WMS && i > 0))
1482 av_strlcatf(transport, sizeof(transport), "-%d", port + 1);
1483 }
1484
1485 /* RTP/TCP */
1486 else if (lower_transport == RTSP_LOWER_TRANSPORT_TCP) {
1487 /* For WMS streams, the application streams are only used for
1488 * UDP. When trying to set it up for TCP streams, the server
1489 * will return an error. Therefore, we skip those streams. */
1490 if (rt->server_type == RTSP_SERVER_WMS &&
1491 (rtsp_st->stream_index < 0 ||
1492 s->streams[rtsp_st->stream_index]->codec->codec_type ==
1493 AVMEDIA_TYPE_DATA))
1494 continue;
1495 snprintf(transport, sizeof(transport) - 1,
1496 "%s/TCP;", trans_pref);
1497 if (rt->transport != RTSP_TRANSPORT_RDT)
1498 av_strlcat(transport, "unicast;", sizeof(transport));
1499 av_strlcatf(transport, sizeof(transport),
1500 "interleaved=%d-%d",
1501 interleave, interleave + 1);
1502 interleave += 2;
1503 }
1504
1505 else if (lower_transport == RTSP_LOWER_TRANSPORT_UDP_MULTICAST) {
1506 snprintf(transport, sizeof(transport) - 1,
1507 "%s/UDP;multicast", trans_pref);
1508 }
1509 if (s->oformat) {
1510 av_strlcat(transport, ";mode=record", sizeof(transport));
1511 } else if (rt->server_type == RTSP_SERVER_REAL ||
1512 rt->server_type == RTSP_SERVER_WMS)
1513 av_strlcat(transport, ";mode=play", sizeof(transport));
1514 snprintf(cmd, sizeof(cmd),
1515 "Transport: %s\r\n",
1516 transport);
1517 if (rt->accept_dynamic_rate)
1518 av_strlcat(cmd, "x-Dynamic-Rate: 0\r\n", sizeof(cmd));
1519 if (CONFIG_RTPDEC && i == 0 && rt->server_type == RTSP_SERVER_REAL) {
1520 char real_res[41], real_csum[9];
1521 ff_rdt_calc_response_and_checksum(real_res, real_csum,
1522 real_challenge);
1523 av_strlcatf(cmd, sizeof(cmd),
1524 "If-Match: %s\r\n"
1525 "RealChallenge2: %s, sd=%s\r\n",
1526 rt->session_id, real_res, real_csum);
1527 }
1528 ff_rtsp_send_cmd(s, "SETUP", rtsp_st->control_url, cmd, reply, NULL);
1529 if (reply->status_code == 461 /* Unsupported protocol */ && i == 0) {
1530 err = 1;
1531 goto fail;
1532 } else if (reply->status_code != RTSP_STATUS_OK ||
1533 reply->nb_transports != 1) {
1534 err = AVERROR_INVALIDDATA;
1535 goto fail;
1536 }
1537
1538 /* XXX: same protocol for all streams is required */
1539 if (i > 0) {
1540 if (reply->transports[0].lower_transport != rt->lower_transport ||
1541 reply->transports[0].transport != rt->transport) {
1542 err = AVERROR_INVALIDDATA;
1543 goto fail;
1544 }
1545 } else {
1546 rt->lower_transport = reply->transports[0].lower_transport;
1547 rt->transport = reply->transports[0].transport;
1548 }
1549
1550 /* Fail if the server responded with another lower transport mode
1551 * than what we requested. */
1552 if (reply->transports[0].lower_transport != lower_transport) {
1553 av_log(s, AV_LOG_ERROR, "Nonmatching transport in server reply\n");
1554 err = AVERROR_INVALIDDATA;
1555 goto fail;
1556 }
1557
1558 switch(reply->transports[0].lower_transport) {
1559 case RTSP_LOWER_TRANSPORT_TCP:
1560 rtsp_st->interleaved_min = reply->transports[0].interleaved_min;
1561 rtsp_st->interleaved_max = reply->transports[0].interleaved_max;
1562 break;
1563
1564 case RTSP_LOWER_TRANSPORT_UDP: {
1565 char url[1024], options[30] = "";
1566 const char *peer = host;
1567
1568 if (rt->rtsp_flags & RTSP_FLAG_FILTER_SRC)
1569 av_strlcpy(options, "?connect=1", sizeof(options));
1570 /* Use source address if specified */
1571 if (reply->transports[0].source[0])
1572 peer = reply->transports[0].source;
1573 ff_url_join(url, sizeof(url), "rtp", NULL, peer,
1574 reply->transports[0].server_port_min, "%s", options);
1575 if (!(rt->server_type == RTSP_SERVER_WMS && i > 1) &&
1576 ff_rtp_set_remote_url(rtsp_st->rtp_handle, url) < 0) {
1577 err = AVERROR_INVALIDDATA;
1578 goto fail;
1579 }
1580 break;
1581 }
1582 case RTSP_LOWER_TRANSPORT_UDP_MULTICAST: {
1583 char url[1024], namebuf[50], optbuf[20] = "";
1584 struct sockaddr_storage addr;
1585 int port, ttl;
1586
1587 if (reply->transports[0].destination.ss_family) {
1588 addr = reply->transports[0].destination;
1589 port = reply->transports[0].port_min;
1590 ttl = reply->transports[0].ttl;
1591 } else {
1592 addr = rtsp_st->sdp_ip;
1593 port = rtsp_st->sdp_port;
1594 ttl = rtsp_st->sdp_ttl;
1595 }
1596 if (ttl > 0)
1597 snprintf(optbuf, sizeof(optbuf), "?ttl=%d", ttl);
1598 getnameinfo((struct sockaddr*) &addr, sizeof(addr),
1599 namebuf, sizeof(namebuf), NULL, 0, NI_NUMERICHOST);
1600 ff_url_join(url, sizeof(url), "rtp", NULL, namebuf,
1601 port, "%s", optbuf);
1602 if (ffurl_open(&rtsp_st->rtp_handle, url, AVIO_FLAG_READ_WRITE,
1603 &s->interrupt_callback, NULL) < 0) {
1604 err = AVERROR_INVALIDDATA;
1605 goto fail;
1606 }
1607 break;
1608 }
1609 }
1610
1611 if ((err = ff_rtsp_open_transport_ctx(s, rtsp_st)))
1612 goto fail;
1613 }
1614
1615 if (rt->nb_rtsp_streams && reply->timeout > 0)
1616 rt->timeout = reply->timeout;
1617
1618 if (rt->server_type == RTSP_SERVER_REAL)
1619 rt->need_subscription = 1;
1620
1621 return 0;
1622
1623 fail:
1624 ff_rtsp_undo_setup(s, 0);
1625 return err;
1626 }
1627
1628 void ff_rtsp_close_connections(AVFormatContext *s)
1629 {
1630 RTSPState *rt = s->priv_data;
1631 if (rt->rtsp_hd_out != rt->rtsp_hd) ffurl_close(rt->rtsp_hd_out);
1632 ffurl_close(rt->rtsp_hd);
1633 rt->rtsp_hd = rt->rtsp_hd_out = NULL;
1634 }
1635
1636 int ff_rtsp_connect(AVFormatContext *s)
1637 {
1638 RTSPState *rt = s->priv_data;
1639 char proto[128], host[1024], path[1024];
1640 char tcpname[1024], cmd[2048], auth[128];
1641 const char *lower_rtsp_proto = "tcp";
1642 int port, err, tcp_fd;
1643 RTSPMessageHeader reply1 = {0}, *reply = &reply1;
1644 int lower_transport_mask = 0;
1645 int default_port = RTSP_DEFAULT_PORT;
1646 char real_challenge[64] = "";
1647 struct sockaddr_storage peer;
1648 socklen_t peer_len = sizeof(peer);
1649
1650 if (rt->rtp_port_max < rt->rtp_port_min) {
1651 av_log(s, AV_LOG_ERROR, "Invalid UDP port range, max port %d less "
1652 "than min port %d\n", rt->rtp_port_max,
1653 rt->rtp_port_min);
1654 return AVERROR(EINVAL);
1655 }
1656
1657 if (!ff_network_init())
1658 return AVERROR(EIO);
1659
1660 if (s->max_delay < 0) /* Not set by the caller */
1661 s->max_delay = s->iformat ? DEFAULT_REORDERING_DELAY : 0;
1662
1663 rt->control_transport = RTSP_MODE_PLAIN;
1664 if (rt->lower_transport_mask & (1 << RTSP_LOWER_TRANSPORT_HTTP)) {
1665 rt->lower_transport_mask = 1 << RTSP_LOWER_TRANSPORT_TCP;
1666 rt->control_transport = RTSP_MODE_TUNNEL;
1667 }
1668 /* Only pass through valid flags from here */
1669 rt->lower_transport_mask &= (1 << RTSP_LOWER_TRANSPORT_NB) - 1;
1670
1671 redirect:
1672 /* extract hostname and port */
1673 av_url_split(proto, sizeof(proto), auth, sizeof(auth),
1674 host, sizeof(host), &port, path, sizeof(path), s->filename);
1675
1676 if (!strcmp(proto, "rtsps")) {
1677 lower_rtsp_proto = "tls";
1678 default_port = RTSPS_DEFAULT_PORT;
1679 rt->lower_transport_mask = 1 << RTSP_LOWER_TRANSPORT_TCP;
1680 }
1681
1682 if (*auth) {
1683 av_strlcpy(rt->auth, auth, sizeof(rt->auth));
1684 }
1685 if (port < 0)
1686 port = default_port;
1687
1688 lower_transport_mask = rt->lower_transport_mask;
1689
1690 if (!lower_transport_mask)
1691 lower_transport_mask = (1 << RTSP_LOWER_TRANSPORT_NB) - 1;
1692
1693 if (s->oformat) {
1694 /* Only UDP or TCP - UDP multicast isn't supported. */
1695 lower_transport_mask &= (1 << RTSP_LOWER_TRANSPORT_UDP) |
1696 (1 << RTSP_LOWER_TRANSPORT_TCP);
1697 if (!lower_transport_mask || rt->control_transport == RTSP_MODE_TUNNEL) {
1698 av_log(s, AV_LOG_ERROR, "Unsupported lower transport method, "
1699 "only UDP and TCP are supported for output.\n");
1700 err = AVERROR(EINVAL);
1701 goto fail;
1702 }
1703 }
1704
1705 /* Construct the URI used in request; this is similar to s->filename,
1706 * but with authentication credentials removed and RTSP specific options
1707 * stripped out. */
1708 ff_url_join(rt->control_uri, sizeof(rt->control_uri), proto, NULL,
1709 host, port, "%s", path);
1710
1711 if (rt->control_transport == RTSP_MODE_TUNNEL) {
1712 /* set up initial handshake for tunneling */
1713 char httpname[1024];
1714 char sessioncookie[17];
1715 char headers[1024];
1716
1717 ff_url_join(httpname, sizeof(httpname), "http", auth, host, port, "%s", path);
1718 snprintf(sessioncookie, sizeof(sessioncookie), "%08x%08x",
1719 av_get_random_seed(), av_get_random_seed());
1720
1721 /* GET requests */
1722 if (ffurl_alloc(&rt->rtsp_hd, httpname, AVIO_FLAG_READ,
1723 &s->interrupt_callback) < 0) {
1724 err = AVERROR(EIO);
1725 goto fail;
1726 }
1727
1728 /* generate GET headers */
1729 snprintf(headers, sizeof(headers),
1730 "x-sessioncookie: %s\r\n"
1731 "Accept: application/x-rtsp-tunnelled\r\n"
1732 "Pragma: no-cache\r\n"
1733 "Cache-Control: no-cache\r\n",
1734 sessioncookie);
1735 av_opt_set(rt->rtsp_hd->priv_data, "headers", headers, 0);
1736
1737 /* complete the connection */
1738 if (ffurl_connect(rt->rtsp_hd, NULL)) {
1739 err = AVERROR(EIO);
1740 goto fail;
1741 }
1742
1743 /* POST requests */
1744 if (ffurl_alloc(&rt->rtsp_hd_out, httpname, AVIO_FLAG_WRITE,
1745 &s->interrupt_callback) < 0 ) {
1746 err = AVERROR(EIO);
1747 goto fail;
1748 }
1749
1750 /* generate POST headers */
1751 snprintf(headers, sizeof(headers),
1752 "x-sessioncookie: %s\r\n"
1753 "Content-Type: application/x-rtsp-tunnelled\r\n"
1754 "Pragma: no-cache\r\n"
1755 "Cache-Control: no-cache\r\n"
1756 "Content-Length: 32767\r\n"
1757 "Expires: Sun, 9 Jan 1972 00:00:00 GMT\r\n",
1758 sessioncookie);
1759 av_opt_set(rt->rtsp_hd_out->priv_data, "headers", headers, 0);
1760 av_opt_set(rt->rtsp_hd_out->priv_data, "chunked_post", "0", 0);
1761
1762 /* Initialize the authentication state for the POST session. The HTTP
1763 * protocol implementation doesn't properly handle multi-pass
1764 * authentication for POST requests, since it would require one of
1765 * the following:
1766 * - implementing Expect: 100-continue, which many HTTP servers
1767 * don't support anyway, even less the RTSP servers that do HTTP
1768 * tunneling
1769 * - sending the whole POST data until getting a 401 reply specifying
1770 * what authentication method to use, then resending all that data
1771 * - waiting for potential 401 replies directly after sending the
1772 * POST header (waiting for some unspecified time)
1773 * Therefore, we copy the full auth state, which works for both basic
1774 * and digest. (For digest, we would have to synchronize the nonce
1775 * count variable between the two sessions, if we'd do more requests
1776 * with the original session, though.)
1777 */
1778 ff_http_init_auth_state(rt->rtsp_hd_out, rt->rtsp_hd);
1779
1780 /* complete the connection */
1781 if (ffurl_connect(rt->rtsp_hd_out, NULL)) {
1782 err = AVERROR(EIO);
1783 goto fail;
1784 }
1785 } else {
1786 /* open the tcp connection */
1787 ff_url_join(tcpname, sizeof(tcpname), lower_rtsp_proto, NULL,
1788 host, port, NULL);
1789 if (ffurl_open(&rt->rtsp_hd, tcpname, AVIO_FLAG_READ_WRITE,
1790 &s->interrupt_callback, NULL) < 0) {
1791 err = AVERROR(EIO);
1792 goto fail;
1793 }
1794 rt->rtsp_hd_out = rt->rtsp_hd;
1795 }
1796 rt->seq = 0;
1797
1798 tcp_fd = ffurl_get_file_handle(rt->rtsp_hd);
1799 if (tcp_fd < 0) {
1800 err = tcp_fd;
1801 goto fail;
1802 }
1803 if (!getpeername(tcp_fd, (struct sockaddr*) &peer, &peer_len)) {
1804 getnameinfo((struct sockaddr*) &peer, peer_len, host, sizeof(host),
1805 NULL, 0, NI_NUMERICHOST);
1806 }
1807
1808 /* request options supported by the server; this also detects server
1809 * type */
1810 for (rt->server_type = RTSP_SERVER_RTP;;) {
1811 cmd[0] = 0;
1812 if (rt->server_type == RTSP_SERVER_REAL)
1813 av_strlcat(cmd,
1814 /*
1815 * The following entries are required for proper
1816 * streaming from a Realmedia server. They are
1817 * interdependent in some way although we currently
1818 * don't quite understand how. Values were copied
1819 * from mplayer SVN r23589.
1820 * ClientChallenge is a 16-byte ID in hex
1821 * CompanyID is a 16-byte ID in base64
1822 */
1823 "ClientChallenge: 9e26d33f2984236010ef6253fb1887f7\r\n"
1824 "PlayerStarttime: [28/03/2003:22:50:23 00:00]\r\n"
1825 "CompanyID: KnKV4M4I/B2FjJ1TToLycw==\r\n"
1826 "GUID: 00000000-0000-0000-0000-000000000000\r\n",
1827 sizeof(cmd));
1828 ff_rtsp_send_cmd(s, "OPTIONS", rt->control_uri, cmd, reply, NULL);
1829 if (reply->status_code != RTSP_STATUS_OK) {
1830 err = AVERROR_INVALIDDATA;
1831 goto fail;
1832 }
1833
1834 /* detect server type if not standard-compliant RTP */
1835 if (rt->server_type != RTSP_SERVER_REAL && reply->real_challenge[0]) {
1836 rt->server_type = RTSP_SERVER_REAL;
1837 continue;
1838 } else if (!av_strncasecmp(reply->server, "WMServer/", 9)) {
1839 rt->server_type = RTSP_SERVER_WMS;
1840 } else if (rt->server_type == RTSP_SERVER_REAL)
1841 strcpy(real_challenge, reply->real_challenge);
1842 break;
1843 }
1844
1845 if (CONFIG_RTSP_DEMUXER && s->iformat)
1846 err = ff_rtsp_setup_input_streams(s, reply);
1847 else if (CONFIG_RTSP_MUXER)
1848 err = ff_rtsp_setup_output_streams(s, host);
1849 if (err)
1850 goto fail;
1851
1852 do {
1853 int lower_transport = ff_log2_tab[lower_transport_mask &
1854 ~(lower_transport_mask - 1)];
1855
1856 err = ff_rtsp_make_setup_request(s, host, port, lower_transport,
1857 rt->server_type == RTSP_SERVER_REAL ?
1858 real_challenge : NULL);
1859 if (err < 0)
1860 goto fail;
1861 lower_transport_mask &= ~(1 << lower_transport);
1862 if (lower_transport_mask == 0 && err == 1) {
1863 err = AVERROR(EPROTONOSUPPORT);
1864 goto fail;
1865 }
1866 } while (err);
1867
1868 rt->lower_transport_mask = lower_transport_mask;
1869 av_strlcpy(rt->real_challenge, real_challenge, sizeof(rt->real_challenge));
1870 rt->state = RTSP_STATE_IDLE;
1871 rt->seek_timestamp = 0; /* default is to start stream at position zero */
1872 return 0;
1873 fail:
1874 ff_rtsp_close_streams(s);
1875 ff_rtsp_close_connections(s);
1876 if (reply->status_code >=300 && reply->status_code < 400 && s->iformat) {
1877 av_strlcpy(s->filename, reply->location, sizeof(s->filename));
1878 rt->session_id[0] = '\0';
1879 av_log(s, AV_LOG_INFO, "Status %d: Redirecting to %s\n",
1880 reply->status_code,
1881 s->filename);
1882 goto redirect;
1883 }
1884 ff_network_close();
1885 return err;
1886 }
1887 #endif /* CONFIG_RTSP_DEMUXER || CONFIG_RTSP_MUXER */
1888
1889 #if CONFIG_RTPDEC
1890 static int udp_read_packet(AVFormatContext *s, RTSPStream **prtsp_st,
1891 uint8_t *buf, int buf_size, int64_t wait_end)
1892 {
1893 RTSPState *rt = s->priv_data;
1894 RTSPStream *rtsp_st;
1895 int n, i, ret, tcp_fd, timeout_cnt = 0;
1896 int max_p = 0;
1897 struct pollfd *p = rt->p;
1898 int *fds = NULL, fdsnum, fdsidx;
1899
1900 for (;;) {
1901 if (ff_check_interrupt(&s->interrupt_callback))
1902 return AVERROR_EXIT;
1903 if (wait_end && wait_end - av_gettime_relative() < 0)
1904 return AVERROR(EAGAIN);
1905 max_p = 0;
1906 if (rt->rtsp_hd) {
1907 tcp_fd = ffurl_get_file_handle(rt->rtsp_hd);
1908 p[max_p].fd = tcp_fd;
1909 p[max_p++].events = POLLIN;
1910 } else {
1911 tcp_fd = -1;
1912 }
1913 for (i = 0; i < rt->nb_rtsp_streams; i++) {
1914 rtsp_st = rt->rtsp_streams[i];
1915 if (rtsp_st->rtp_handle) {
1916 if (ret = ffurl_get_multi_file_handle(rtsp_st->rtp_handle,
1917 &fds, &fdsnum)) {
1918 av_log(s, AV_LOG_ERROR, "Unable to recover rtp ports\n");
1919 return ret;
1920 }
1921 if (fdsnum != 2) {
1922 av_log(s, AV_LOG_ERROR,
1923 "Number of fds %d not supported\n", fdsnum);
1924 return AVERROR_INVALIDDATA;
1925 }
1926 for (fdsidx = 0; fdsidx < fdsnum; fdsidx++) {
1927 p[max_p].fd = fds[fdsidx];
1928 p[max_p++].events = POLLIN;
1929 }
1930 av_free(fds);
1931 }
1932 }
1933 n = poll(p, max_p, POLL_TIMEOUT_MS);
1934 if (n > 0) {
1935 int j = 1 - (tcp_fd == -1);
1936 timeout_cnt = 0;
1937 for (i = 0; i < rt->nb_rtsp_streams; i++) {
1938 rtsp_st = rt->rtsp_streams[i];
1939 if (rtsp_st->rtp_handle) {
1940 if (p[j].revents & POLLIN || p[j+1].revents & POLLIN) {
1941 ret = ffurl_read(rtsp_st->rtp_handle, buf, buf_size);
1942 if (ret > 0) {
1943 *prtsp_st = rtsp_st;
1944 return ret;
1945 }
1946 }
1947 j+=2;
1948 }
1949 }
1950 #if CONFIG_RTSP_DEMUXER
1951 if (tcp_fd != -1 && p[0].revents & POLLIN) {
1952 if (rt->rtsp_flags & RTSP_FLAG_LISTEN) {
1953 if (rt->state == RTSP_STATE_STREAMING) {
1954 if (!ff_rtsp_parse_streaming_commands(s))
1955 return AVERROR_EOF;
1956 else
1957 av_log(s, AV_LOG_WARNING,
1958 "Unable to answer to TEARDOWN\n");
1959 } else
1960 return 0;
1961 } else {
1962 RTSPMessageHeader reply;
1963 ret = ff_rtsp_read_reply(s, &reply, NULL, 0, NULL);
1964 if (ret < 0)
1965 return ret;
1966 /* XXX: parse message */
1967 if (rt->state != RTSP_STATE_STREAMING)
1968 return 0;
1969 }
1970 }
1971 #endif
1972 } else if (n == 0 && ++timeout_cnt >= MAX_TIMEOUTS) {
1973 return AVERROR(ETIMEDOUT);
1974 } else if (n < 0 && errno != EINTR)
1975 return AVERROR(errno);
1976 }
1977 }
1978
1979 static int pick_stream(AVFormatContext *s, RTSPStream **rtsp_st,
1980 const uint8_t *buf, int len)
1981 {
1982 RTSPState *rt = s->priv_data;
1983 int i;
1984 if (len < 0)
1985 return len;
1986 if (rt->nb_rtsp_streams == 1) {
1987 *rtsp_st = rt->rtsp_streams[0];
1988 return len;
1989 }
1990 if (len >= 8 && rt->transport == RTSP_TRANSPORT_RTP) {
1991 if (RTP_PT_IS_RTCP(rt->recvbuf[1])) {
1992 int no_ssrc = 0;
1993 for (i = 0; i < rt->nb_rtsp_streams; i++) {
1994 RTPDemuxContext *rtpctx = rt->rtsp_streams[i]->transport_priv;
1995 if (!rtpctx)
1996 continue;
1997 if (rtpctx->ssrc == AV_RB32(&buf[4])) {
1998 *rtsp_st = rt->rtsp_streams[i];
1999 return len;
2000 }
2001 if (!rtpctx->ssrc)
2002 no_ssrc = 1;
2003 }
2004 if (no_ssrc) {
2005 av_log(s, AV_LOG_WARNING,
2006 "Unable to pick stream for packet - SSRC not known for "
2007 "all streams\n");
2008 return AVERROR(EAGAIN);
2009 }
2010 } else {
2011 for (i = 0; i < rt->nb_rtsp_streams; i++) {
2012 if ((buf[1] & 0x7f) == rt->rtsp_streams[i]->sdp_payload_type) {
2013 *rtsp_st = rt->rtsp_streams[i];
2014 return len;
2015 }
2016 }
2017 }
2018 }
2019 av_log(s, AV_LOG_WARNING, "Unable to pick stream for packet\n");
2020 return AVERROR(EAGAIN);
2021 }
2022
2023 int ff_rtsp_fetch_packet(AVFormatContext *s, AVPacket *pkt)
2024 {
2025 RTSPState *rt = s->priv_data;
2026 int ret, len;
2027 RTSPStream *rtsp_st, *first_queue_st = NULL;
2028 int64_t wait_end = 0;
2029
2030 if (rt->nb_byes == rt->nb_rtsp_streams)
2031 return AVERROR_EOF;
2032
2033 /* get next frames from the same RTP packet */
2034 if (rt->cur_transport_priv) {
2035 if (rt->transport == RTSP_TRANSPORT_RDT) {
2036 ret = ff_rdt_parse_packet(rt->cur_transport_priv, pkt, NULL, 0);
2037 } else if (rt->transport == RTSP_TRANSPORT_RTP) {
2038 ret = ff_rtp_parse_packet(rt->cur_transport_priv, pkt, NULL, 0);
2039 } else if (CONFIG_RTPDEC && rt->ts) {
2040 ret = ff_mpegts_parse_packet(rt->ts, pkt, rt->recvbuf + rt->recvbuf_pos, rt->recvbuf_len - rt->recvbuf_pos);
2041 if (ret >= 0) {
2042 rt->recvbuf_pos += ret;
2043 ret = rt->recvbuf_pos < rt->recvbuf_len;
2044 }
2045 } else
2046 ret = -1;
2047 if (ret == 0) {
2048 rt->cur_transport_priv = NULL;
2049 return 0;
2050 } else if (ret == 1) {
2051 return 0;
2052 } else
2053 rt->cur_transport_priv = NULL;
2054 }
2055
2056 redo:
2057 if (rt->transport == RTSP_TRANSPORT_RTP) {
2058 int i;
2059 int64_t first_queue_time = 0;
2060 for (i = 0; i < rt->nb_rtsp_streams; i++) {
2061 RTPDemuxContext *rtpctx = rt->rtsp_streams[i]->transport_priv;
2062 int64_t queue_time;
2063 if (!rtpctx)
2064 continue;
2065 queue_time = ff_rtp_queued_packet_time(rtpctx);
2066 if (queue_time && (queue_time - first_queue_time < 0 ||
2067 !first_queue_time)) {
2068 first_queue_time = queue_time;
2069 first_queue_st = rt->rtsp_streams[i];
2070 }
2071 }
2072 if (first_queue_time) {
2073 wait_end = first_queue_time + s->max_delay;
2074 } else {
2075 wait_end = 0;
2076 first_queue_st = NULL;
2077 }
2078 }
2079
2080 /* read next RTP packet */
2081 if (!rt->recvbuf) {
2082 rt->recvbuf = av_malloc(RECVBUF_SIZE);
2083 if (!rt->recvbuf)
2084 return AVERROR(ENOMEM);
2085 }
2086
2087 switch(rt->lower_transport) {
2088 default:
2089 #if CONFIG_RTSP_DEMUXER
2090 case RTSP_LOWER_TRANSPORT_TCP:
2091 len = ff_rtsp_tcp_read_packet(s, &rtsp_st, rt->recvbuf, RECVBUF_SIZE);
2092 break;
2093 #endif
2094 case RTSP_LOWER_TRANSPORT_UDP:
2095 case RTSP_LOWER_TRANSPORT_UDP_MULTICAST:
2096 len = udp_read_packet(s, &rtsp_st, rt->recvbuf, RECVBUF_SIZE, wait_end);
2097 if (len > 0 && rtsp_st->transport_priv && rt->transport == RTSP_TRANSPORT_RTP)
2098 ff_rtp_check_and_send_back_rr(rtsp_st->transport_priv, rtsp_st->rtp_handle, NULL, len);
2099 break;
2100 case RTSP_LOWER_TRANSPORT_CUSTOM:
2101 if (first_queue_st && rt->transport == RTSP_TRANSPORT_RTP &&
2102 wait_end && wait_end < av_gettime_relative())
2103 len = AVERROR(EAGAIN);
2104 else
2105 len = ffio_read_partial(s->pb, rt->recvbuf, RECVBUF_SIZE);
2106 len = pick_stream(s, &rtsp_st, rt->recvbuf, len);
2107 if (len > 0 && rtsp_st->transport_priv && rt->transport == RTSP_TRANSPORT_RTP)
2108 ff_rtp_check_and_send_back_rr(rtsp_st->transport_priv, NULL, s->pb, len);
2109 break;
2110 }
2111 if (len == AVERROR(EAGAIN) && first_queue_st &&
2112 rt->transport == RTSP_TRANSPORT_RTP) {
2113 av_log(s, AV_LOG_WARNING,
2114 "max delay reached. need to consume packet\n");
2115 rtsp_st = first_queue_st;
2116 ret = ff_rtp_parse_packet(rtsp_st->transport_priv, pkt, NULL, 0);
2117 goto end;
2118 }
2119 if (len < 0)
2120 return len;
2121 if (len == 0)
2122 return AVERROR_EOF;
2123 if (rt->transport == RTSP_TRANSPORT_RDT) {
2124 ret = ff_rdt_parse_packet(rtsp_st->transport_priv, pkt, &rt->recvbuf, len);
2125 } else if (rt->transport == RTSP_TRANSPORT_RTP) {
2126 ret = ff_rtp_parse_packet(rtsp_st->transport_priv, pkt, &rt->recvbuf, len);
2127 if (rtsp_st->feedback) {
2128 AVIOContext *pb = NULL;
2129 if (rt->lower_transport == RTSP_LOWER_TRANSPORT_CUSTOM)
2130 pb = s->pb;
2131 ff_rtp_send_rtcp_feedback(rtsp_st->transport_priv, rtsp_st->rtp_handle, pb);
2132 }
2133 if (ret < 0) {
2134 /* Either bad packet, or a RTCP packet. Check if the
2135 * first_rtcp_ntp_time field was initialized. */
2136 RTPDemuxContext *rtpctx = rtsp_st->transport_priv;
2137 if (rtpctx->first_rtcp_ntp_time != AV_NOPTS_VALUE) {
2138 /* first_rtcp_ntp_time has been initialized for this stream,
2139 * copy the same value to all other uninitialized streams,
2140 * in order to map their timestamp origin to the same ntp time
2141 * as this one. */
2142 int i;
2143 AVStream *st = NULL;
2144 if (rtsp_st->stream_index >= 0)
2145 st = s->streams[rtsp_st->stream_index];
2146 for (i = 0; i < rt->nb_rtsp_streams; i++) {
2147 RTPDemuxContext *rtpctx2 = rt->rtsp_streams[i]->transport_priv;
2148 AVStream *st2 = NULL;
2149 if (rt->rtsp_streams[i]->stream_index >= 0)
2150 st2 = s->streams[rt->rtsp_streams[i]->stream_index];
2151 if (rtpctx2 && st && st2 &&
2152 rtpctx2->first_rtcp_ntp_time == AV_NOPTS_VALUE) {
2153 rtpctx2->first_rtcp_ntp_time = rtpctx->first_rtcp_ntp_time;
2154 rtpctx2->rtcp_ts_offset = av_rescale_q(
2155 rtpctx->rtcp_ts_offset, st->time_base,
2156 st2->time_base);
2157 }
2158 }
2159 }
2160 if (ret == -RTCP_BYE) {
2161 rt->nb_byes++;
2162
2163 av_log(s, AV_LOG_DEBUG, "Received BYE for stream %d (%d/%d)\n",
2164 rtsp_st->stream_index, rt->nb_byes, rt->nb_rtsp_streams);
2165
2166 if (rt->nb_byes == rt->nb_rtsp_streams)
2167 return AVERROR_EOF;
2168 }
2169 }
2170 } else if (CONFIG_RTPDEC && rt->ts) {
2171 ret = ff_mpegts_parse_packet(rt->ts, pkt, rt->recvbuf, len);
2172 if (ret >= 0) {
2173 if (ret < len) {
2174 rt->recvbuf_len = len;
2175 rt->recvbuf_pos = ret;
2176 rt->cur_transport_priv = rt->ts;
2177 return 1;
2178 } else {
2179 ret = 0;
2180 }
2181 }
2182 } else {
2183 return AVERROR_INVALIDDATA;
2184 }
2185 end:
2186 if (ret < 0)
2187 goto redo;
2188 if (ret == 1)
2189 /* more packets may follow, so we save the RTP context */
2190 rt->cur_transport_priv = rtsp_st->transport_priv;
2191
2192 return ret;
2193 }
2194 #endif /* CONFIG_RTPDEC */
2195
2196 #if CONFIG_SDP_DEMUXER
2197 static int sdp_probe(AVProbeData *p1)
2198 {
2199 const char *p = p1->buf, *p_end = p1->buf + p1->buf_size;
2200
2201 /* we look for a line beginning "c=IN IP" */
2202 while (p < p_end && *p != '\0') {
2203 if (p + sizeof("c=IN IP") - 1 < p_end &&
2204 av_strstart(p, "c=IN IP", NULL))
2205 return AVPROBE_SCORE_EXTENSION;
2206
2207 while (p < p_end - 1 && *p != '\n') p++;
2208 if (++p >= p_end)
2209 break;
2210 if (*p == '\r')
2211 p++;
2212 }
2213 return 0;
2214 }
2215
2216 static void append_source_addrs(char *buf, int size, const char *name,
2217 int count, struct RTSPSource **addrs)
2218 {
2219 int i;
2220 if (!count)
2221 return;
2222 av_strlcatf(buf, size, "&%s=%s", name, addrs[0]->addr);
2223 for (i = 1; i < count; i++)
2224 av_strlcatf(buf, size, ",%s", addrs[i]->addr);
2225 }
2226
2227 static int sdp_read_header(AVFormatContext *s)
2228 {
2229 RTSPState *rt = s->priv_data;
2230 RTSPStream *rtsp_st;
2231 int size, i, err;
2232 char *content;
2233 char url[1024];
2234
2235 if (!ff_network_init())
2236 return AVERROR(EIO);
2237
2238 if (s->max_delay < 0) /* Not set by the caller */
2239 s->max_delay = DEFAULT_REORDERING_DELAY;
2240 if (rt->rtsp_flags & RTSP_FLAG_CUSTOM_IO)
2241 rt->lower_transport = RTSP_LOWER_TRANSPORT_CUSTOM;
2242
2243 /* read the whole sdp file */
2244 /* XXX: better loading */
2245 content = av_malloc(SDP_MAX_SIZE);
2246 if (!content)
2247 return AVERROR(ENOMEM);
2248 size = avio_read(s->pb, content, SDP_MAX_SIZE - 1);
2249 if (size <= 0) {
2250 av_free(content);
2251 return AVERROR_INVALIDDATA;
2252 }
2253 content[size] ='\0';
2254
2255 err = ff_sdp_parse(s, content);
2256 av_free(content);
2257 if (err) goto fail;
2258
2259 /* open each RTP stream */
2260 for (i = 0; i < rt->nb_rtsp_streams; i++) {
2261 char namebuf[50];
2262 rtsp_st = rt->rtsp_streams[i];
2263
2264 if (!(rt->rtsp_flags & RTSP_FLAG_CUSTOM_IO)) {
2265 AVDictionary *opts = map_to_opts(rt);
2266
2267 err = getnameinfo((struct sockaddr*) &rtsp_st->sdp_ip,
2268 sizeof(rtsp_st->sdp_ip),
2269 namebuf, sizeof(namebuf), NULL, 0, NI_NUMERICHOST);
2270 if (err) {
2271 av_log(s, AV_LOG_ERROR, "getnameinfo: %s\n", gai_strerror(err));
2272 err = AVERROR(EIO);
2273 goto fail;
2274 }
2275 ff_url_join(url, sizeof(url), "rtp", NULL,
2276 namebuf, rtsp_st->sdp_port,
2277 "?localport=%d&ttl=%d&connect=%d&write_to_source=%d",
2278 rtsp_st->sdp_port, rtsp_st->sdp_ttl,
2279 rt->rtsp_flags & RTSP_FLAG_FILTER_SRC ? 1 : 0,
2280 rt->rtsp_flags & RTSP_FLAG_RTCP_TO_SOURCE ? 1 : 0);
2281
2282 append_source_addrs(url, sizeof(url), "sources",
2283 rtsp_st->nb_include_source_addrs,
2284 rtsp_st->include_source_addrs);
2285 append_source_addrs(url, sizeof(url), "block",
2286 rtsp_st->nb_exclude_source_addrs,
2287 rtsp_st->exclude_source_addrs);
2288 err = ffurl_open(&rtsp_st->rtp_handle, url, AVIO_FLAG_READ_WRITE,
2289 &s->interrupt_callback, &opts);
2290
2291 av_dict_free(&opts);
2292
2293 if (err < 0) {
2294 err = AVERROR_INVALIDDATA;
2295 goto fail;
2296 }
2297 }
2298 if ((err = ff_rtsp_open_transport_ctx(s, rtsp_st)))
2299 goto fail;
2300 }
2301 return 0;
2302 fail:
2303 ff_rtsp_close_streams(s);
2304 ff_network_close();
2305 return err;
2306 }
2307
2308 static int sdp_read_close(AVFormatContext *s)
2309 {
2310 ff_rtsp_close_streams(s);
2311 ff_network_close();
2312 return 0;
2313 }
2314
2315 static const AVClass sdp_demuxer_class = {
2316 .class_name = "SDP demuxer",
2317 .item_name = av_default_item_name,
2318 .option = sdp_options,
2319 .version = LIBAVUTIL_VERSION_INT,
2320 };
2321
2322 AVInputFormat ff_sdp_demuxer = {
2323 .name = "sdp",
2324 .long_name = NULL_IF_CONFIG_SMALL("SDP"),
2325 .priv_data_size = sizeof(RTSPState),
2326 .read_probe = sdp_probe,
2327 .read_header = sdp_read_header,
2328 .read_packet = ff_rtsp_fetch_packet,
2329 .read_close = sdp_read_close,
2330 .priv_class = &sdp_demuxer_class,
2331 };
2332 #endif /* CONFIG_SDP_DEMUXER */
2333
2334 #if CONFIG_RTP_DEMUXER
2335 static int rtp_probe(AVProbeData *p)
2336 {
2337 if (av_strstart(p->filename, "rtp:", NULL))
2338 return AVPROBE_SCORE_MAX;
2339 return 0;
2340 }
2341
2342 static int rtp_read_header(AVFormatContext *s)
2343 {
2344 uint8_t recvbuf[RTP_MAX_PACKET_LENGTH];
2345 char host[500], sdp[500];
2346 int ret, port;
2347 URLContext* in = NULL;
2348 int payload_type;
2349 AVCodecContext codec = { 0 };
2350 struct sockaddr_storage addr;
2351 AVIOContext pb;
2352 socklen_t addrlen = sizeof(addr);
2353 RTSPState *rt = s->priv_data;
2354
2355 if (!ff_network_init())
2356 return AVERROR(EIO);
2357
2358 ret = ffurl_open(&in, s->filename, AVIO_FLAG_READ,
2359 &s->interrupt_callback, NULL);
2360 if (ret)
2361 goto fail;
2362
2363 while (1) {
2364 ret = ffurl_read(in, recvbuf, sizeof(recvbuf));
2365 if (ret == AVERROR(EAGAIN))
2366 continue;
2367 if (ret < 0)
2368 goto fail;
2369 if (ret < 12) {
2370 av_log(s, AV_LOG_WARNING, "Received too short packet\n");
2371 continue;
2372 }
2373
2374 if ((recvbuf[0] & 0xc0) != 0x80) {
2375 av_log(s, AV_LOG_WARNING, "Unsupported RTP version packet "
2376 "received\n");
2377 continue;
2378 }
2379
2380 if (RTP_PT_IS_RTCP(recvbuf[1]))
2381 continue;
2382
2383 payload_type = recvbuf[1] & 0x7f;
2384 break;
2385 }
2386 getsockname(ffurl_get_file_handle(in), (struct sockaddr*) &addr, &addrlen);
2387 ffurl_close(in);
2388 in = NULL;
2389
2390 if (ff_rtp_get_codec_info(&codec, payload_type)) {
2391 av_log(s, AV_LOG_ERROR, "Unable to receive RTP payload type %d "
2392 "without an SDP file describing it\n",
2393 payload_type);
2394 goto fail;
2395 }
2396 if (codec.codec_type != AVMEDIA_TYPE_DATA) {
2397 av_log(s, AV_LOG_WARNING, "Guessing on RTP content - if not received "
2398 "properly you need an SDP file "
2399 "describing it\n");
2400 }
2401
2402 av_url_split(NULL, 0, NULL, 0, host, sizeof(host), &port,
2403 NULL, 0, s->filename);
2404
2405 snprintf(sdp, sizeof(sdp),
2406 "v=0\r\nc=IN IP%d %s\r\nm=%s %d RTP/AVP %d\r\n",
2407 addr.ss_family == AF_INET ? 4 : 6, host,
2408 codec.codec_type == AVMEDIA_TYPE_DATA ? "application" :
2409 codec.codec_type == AVMEDIA_TYPE_VIDEO ? "video" : "audio",
2410 port, payload_type);
2411 av_log(s, AV_LOG_VERBOSE, "SDP:\n%s\n", sdp);
2412
2413 ffio_init_context(&pb, sdp, strlen(sdp), 0, NULL, NULL, NULL, NULL);
2414 s->pb = &pb;
2415
2416 /* sdp_read_header initializes this again */
2417 ff_network_close();
2418
2419 rt->media_type_mask = (1 << (AVMEDIA_TYPE_DATA+1)) - 1;
2420
2421 ret = sdp_read_header(s);
2422 s->pb = NULL;
2423 return ret;
2424
2425 fail:
2426 if (in)
2427 ffurl_close(in);
2428 ff_network_close();
2429 return ret;
2430 }
2431
2432 static const AVClass rtp_demuxer_class = {
2433 .class_name = "RTP demuxer",
2434 .item_name = av_default_item_name,
2435 .option = rtp_options,
2436 .version = LIBAVUTIL_VERSION_INT,
2437 };
2438
2439 AVInputFormat ff_rtp_demuxer = {
2440 .name = "rtp",
2441 .long_name = NULL_IF_CONFIG_SMALL("RTP input"),
2442 .priv_data_size = sizeof(RTSPState),
2443 .read_probe = rtp_probe,
2444 .read_header = rtp_read_header,
2445 .read_packet = ff_rtsp_fetch_packet,
2446 .read_close = sdp_read_close,
2447 .flags = AVFMT_NOFILE,
2448 .priv_class = &rtp_demuxer_class,
2449 };
2450 #endif /* CONFIG_RTP_DEMUXER */