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