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