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