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