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