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