1 /*****************************************************************************
2 * rtsp.c: RTSP support for RTP stream output module
3 *****************************************************************************
4 * Copyright (C) 2003-2004 the VideoLAN team
5 * Copyright © 2007 Rémi Denis-Courmont
7 * $Id: rtp.c 21407 2007-08-22 20:10:41Z courmisch $
9 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
11 * This program is free software; you can redistribute it and/or modify
12 * it under the terms of the GNU General Public License as published by
13 * the Free Software Foundation; either version 2 of the License, or
14 * (at your option) any later version.
16 * This program is distributed in the hope that it will be useful,
17 * but WITHOUT ANY WARRANTY; without even the implied warranty of
18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 * GNU General Public License for more details.
21 * You should have received a copy of the GNU General Public License
22 * along with this program; if not, write to the Free Software
23 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
24 *****************************************************************************/
26 /*****************************************************************************
28 *****************************************************************************/
32 #include <vlc_httpd.h>
34 #include <vlc_network.h>
42 typedef struct rtsp_session_t rtsp_session_t;
51 const char *track_fmt;
55 rtsp_session_t **sessionv;
59 static int RtspCallback( httpd_callback_sys_t *p_args,
60 httpd_client_t *cl, httpd_message_t *answer,
61 const httpd_message_t *query );
62 static int RtspCallbackId( httpd_callback_sys_t *p_args,
63 httpd_client_t *cl, httpd_message_t *answer,
64 const httpd_message_t *query );
65 static void RtspClientDel( rtsp_stream_t *rtsp, rtsp_session_t *session );
67 rtsp_stream_t *RtspSetup( sout_stream_t *p_stream, const vlc_url_t *url )
69 rtsp_stream_t *rtsp = malloc( sizeof( *rtsp ) );
71 if( rtsp == NULL || ( url->i_port > 99999 ) )
77 rtsp->owner = p_stream;
79 rtsp->sessionv = NULL;
82 rtsp->psz_path = NULL;
83 vlc_mutex_init( p_stream, &rtsp->lock );
85 rtsp->port = (url->i_port > 0) ? url->i_port : 554;
86 rtsp->psz_path = strdup( ( url->psz_path != NULL ) ? url->psz_path : "/" );
87 if( rtsp->psz_path == NULL )
90 assert( strlen( rtsp->psz_path ) > 0 );
91 if( rtsp->psz_path[strlen( rtsp->psz_path ) - 1] == '/' )
92 rtsp->track_fmt = "%strackID=%u";
94 rtsp->track_fmt = "%s/trackID=%u";
96 msg_Dbg( p_stream, "RTSP stream: host %s port %d at %s",
97 url->psz_host, rtsp->port, rtsp->psz_path );
99 rtsp->host = httpd_HostNew( VLC_OBJECT(p_stream), url->psz_host,
101 if( rtsp->host == NULL )
104 rtsp->url = httpd_UrlNewUnique( rtsp->host, rtsp->psz_path,
106 if( rtsp->url == NULL )
109 httpd_UrlCatch( rtsp->url, HTTPD_MSG_DESCRIBE, RtspCallback, (void*)rtsp );
110 httpd_UrlCatch( rtsp->url, HTTPD_MSG_SETUP, RtspCallback, (void*)rtsp );
111 httpd_UrlCatch( rtsp->url, HTTPD_MSG_PLAY, RtspCallback, (void*)rtsp );
112 httpd_UrlCatch( rtsp->url, HTTPD_MSG_PAUSE, RtspCallback, (void*)rtsp );
113 httpd_UrlCatch( rtsp->url, HTTPD_MSG_GETPARAMETER, RtspCallback,
115 httpd_UrlCatch( rtsp->url, HTTPD_MSG_TEARDOWN, RtspCallback, (void*)rtsp );
124 void RtspUnsetup( rtsp_stream_t *rtsp )
126 while( rtsp->sessionc > 0 )
127 RtspClientDel( rtsp, rtsp->sessionv[0] );
130 httpd_UrlDelete( rtsp->url );
133 httpd_HostDelete( rtsp->host );
135 free( rtsp->psz_path );
136 vlc_mutex_destroy( &rtsp->lock );
140 struct rtsp_stream_id_t
142 rtsp_stream_t *stream;
143 sout_stream_id_t *sout_id;
148 uint16_t loport, hiport;
152 typedef struct rtsp_strack_t rtsp_strack_t;
154 /* For unicast streaming */
155 struct rtsp_session_t
157 rtsp_stream_t *stream;
160 /* output (id-access) */
162 rtsp_strack_t *trackv;
166 /* Unicast session track */
169 sout_stream_id_t *id;
175 rtsp_stream_id_t *RtspAddId( rtsp_stream_t *rtsp, sout_stream_id_t *sid,
176 unsigned num, uint32_t ssrc,
177 /* Multicast stuff - TODO: cleanup */
178 const char *dst, int ttl,
179 unsigned loport, unsigned hiport )
181 char urlbuf[sizeof( "/trackID=123" ) + strlen( rtsp->psz_path )];
182 rtsp_stream_id_t *id = malloc( sizeof( *id ) );
191 /* TODO: can we assume that this need not be strdup'd? */
193 if( id->dst != NULL )
200 /* FIXME: num screws up if any ES has been removed and re-added */
201 snprintf( urlbuf, sizeof( urlbuf ), rtsp->track_fmt, rtsp->psz_path,
203 msg_Dbg( rtsp->owner, "RTSP: adding %s", urlbuf );
204 url = id->url = httpd_UrlNewUnique( rtsp->host, urlbuf, NULL, NULL, NULL );
212 httpd_UrlCatch( url, HTTPD_MSG_DESCRIBE, RtspCallbackId, (void *)id );
213 httpd_UrlCatch( url, HTTPD_MSG_SETUP, RtspCallbackId, (void *)id );
214 httpd_UrlCatch( url, HTTPD_MSG_PLAY, RtspCallbackId, (void *)id );
215 httpd_UrlCatch( url, HTTPD_MSG_PAUSE, RtspCallbackId, (void *)id );
216 httpd_UrlCatch( url, HTTPD_MSG_GETPARAMETER, RtspCallbackId, (void *)id );
217 httpd_UrlCatch( url, HTTPD_MSG_TEARDOWN, RtspCallbackId, (void *)id );
223 void RtspDelId( rtsp_stream_t *rtsp, rtsp_stream_id_t *id )
225 vlc_mutex_lock( &rtsp->lock );
226 for( int i = 0; i < rtsp->sessionc; i++ )
228 rtsp_session_t *ses = rtsp->sessionv[i];
230 for( int j = 0; j < ses->trackc; j++ )
232 if( ses->trackv[j].id == id->sout_id )
234 rtsp_strack_t *tr = ses->trackv + j;
236 REMOVE_ELEM( ses->trackv, ses->trackc, j );
241 vlc_mutex_unlock( &rtsp->lock );
242 httpd_UrlDelete( id->url );
247 /** rtsp must be locked */
249 rtsp_session_t *RtspClientNew( rtsp_stream_t *rtsp )
251 rtsp_session_t *s = malloc( sizeof( *s ) );
256 vlc_rand_bytes (&s->id, sizeof (s->id));
260 TAB_APPEND( rtsp->sessionc, rtsp->sessionv, s );
266 /** rtsp must be locked */
268 rtsp_session_t *RtspClientGet( rtsp_stream_t *rtsp, const char *name )
278 id = strtoull( name, &end, 0x10 );
282 /* FIXME: use a hash/dictionary */
283 for( i = 0; i < rtsp->sessionc; i++ )
285 if( rtsp->sessionv[i]->id == id )
286 return rtsp->sessionv[i];
292 /** rtsp must be locked */
294 void RtspClientDel( rtsp_stream_t *rtsp, rtsp_session_t *session )
297 TAB_REMOVE( rtsp->sessionc, rtsp->sessionv, session );
299 for( i = 0; i < session->trackc; i++ )
300 rtp_del_sink( session->trackv[i].id, session->trackv[i].fd );
302 free( session->trackv );
307 /** Finds the next transport choice */
308 static inline const char *transport_next( const char *str )
310 /* Looks for comma */
311 str = strchr( str, ',' );
313 return NULL; /* No more transport options */
315 str++; /* skips comma */
316 while( strchr( "\r\n\t ", *str ) )
319 return (*str) ? str : NULL;
323 /** Finds the next transport parameter */
324 static inline const char *parameter_next( const char *str )
326 while( strchr( ",;", *str ) == NULL )
329 return (*str == ';') ? (str + 1) : NULL;
333 /** RTSP requests handler
334 * @param id selected track for non-aggregate URLs,
335 * NULL for aggregate URLs
337 static int RtspHandler( rtsp_stream_t *rtsp, rtsp_stream_id_t *id,
339 httpd_message_t *answer,
340 const httpd_message_t *query )
342 sout_stream_t *p_stream = rtsp->owner;
344 const char *psz_session = NULL, *psz;
345 char control[sizeof("rtsp://[]:12345") + NI_MAXNUMERICHOST
346 + strlen( rtsp->psz_path )];
348 if( answer == NULL || query == NULL || cl == NULL )
352 /* Build self-referential control URL */
353 char ip[NI_MAXNUMERICHOST], *ptr;
355 httpd_ServerIP( cl, ip );
356 ptr = strchr( ip, '%' );
360 if( strchr( ip, ':' ) != NULL )
361 sprintf( control, "rtsp://[%s]:%u%s", ip, rtsp->port,
364 sprintf( control, "rtsp://%s:%u%s", ip, rtsp->port,
369 answer->i_proto = HTTPD_PROTO_RTSP;
370 answer->i_version= 0;
371 answer->i_type = HTTPD_MSG_ANSWER;
373 answer->p_body = NULL;
375 if( query->i_proto != HTTPD_PROTO_RTSP )
377 answer->i_status = 505;
380 if( httpd_MsgGet( query, "Require" ) != NULL )
382 answer->i_status = 551;
383 httpd_MsgAdd( answer, "Unsupported", "%s",
384 httpd_MsgGet( query, "Require" ) );
387 switch( query->i_type )
389 case HTTPD_MSG_DESCRIBE:
390 { /* Aggregate-only */
393 answer->i_status = 460;
397 answer->i_status = 200;
398 httpd_MsgAdd( answer, "Content-Type", "%s", "application/sdp" );
399 httpd_MsgAdd( answer, "Content-Base", "%s", control );
400 answer->p_body = (uint8_t *)SDPGenerate( rtsp->owner, control );
401 if( answer->p_body != NULL )
402 answer->i_body = strlen( (char *)answer->p_body );
404 answer->i_status = 500;
408 case HTTPD_MSG_SETUP:
409 /* Non-aggregate-only */
412 answer->i_status = 459;
416 psz_session = httpd_MsgGet( query, "Session" );
417 answer->i_status = 461;
419 for( const char *tpt = httpd_MsgGet( query, "Transport" );
421 tpt = transport_next( tpt ) )
423 vlc_bool_t b_multicast = VLC_TRUE, b_unsupp = VLC_FALSE;
424 unsigned loport = 5004, hiport = 5005; /* from RFC3551 */
426 /* Check transport protocol. */
427 /* Currently, we only support RTP/AVP over UDP */
428 if( strncmp( tpt, "RTP/AVP", 7 ) )
431 if( strncmp( tpt, "/UDP", 4 ) == 0 )
433 if( strchr( ";,", *tpt ) == NULL )
436 /* Parse transport options */
437 for( const char *opt = parameter_next( tpt );
439 opt = parameter_next( opt ) )
441 if( strncmp( opt, "multicast", 9 ) == 0)
442 b_multicast = VLC_TRUE;
444 if( strncmp( opt, "unicast", 7 ) == 0 )
445 b_multicast = VLC_FALSE;
447 if( sscanf( opt, "client_port=%u-%u", &loport, &hiport )
451 if( strncmp( opt, "mode=", 5 ) == 0 )
453 if( strncasecmp( opt + 5, "play", 4 )
454 && strncasecmp( opt + 5, "\"PLAY\"", 6 ) )
462 if( strncmp( opt,"destination=", 12 ) == 0 )
464 answer->i_status = 403;
470 * Every other option is unsupported:
472 * "source" and "append" are invalid (server-only);
473 * "ssrc" also (as clarified per RFC2326bis).
475 * For multicast, "port", "layers", "ttl" are set by the
476 * stream output configuration.
478 * For unicast, we want to decide "server_port" values.
480 * "interleaved" is not implemented.
492 const char *dst = id->dst;
496 if( psz_session == NULL )
498 /* Create a dummy session ID */
499 snprintf( psz_sesbuf, sizeof( psz_sesbuf ), "%d",
501 psz_session = psz_sesbuf;
503 answer->i_status = 200;
505 httpd_MsgAdd( answer, "Transport",
506 "RTP/AVP/UDP;destination=%s;port=%u-%u;"
508 dst, id->loport, id->hiport,
509 ( id->ttl > 0 ) ? id->ttl : 1 );
513 char ip[NI_MAXNUMERICHOST], src[NI_MAXNUMERICHOST];
514 rtsp_session_t *ses = NULL;
515 rtsp_strack_t track = { id->sout_id, -1, VLC_FALSE };
518 if( httpd_ClientIP( cl, ip ) == NULL )
520 answer->i_status = 500;
524 track.fd = net_ConnectDgram( p_stream, ip, loport, -1,
529 "cannot create RTP socket for %s port %u",
531 answer->i_status = 500;
535 net_GetSockAddress( track.fd, src, &sport );
537 vlc_mutex_lock( &rtsp->lock );
538 if( psz_session == NULL )
540 ses = RtspClientNew( rtsp );
541 snprintf( psz_sesbuf, sizeof( psz_sesbuf ), I64Fx,
543 psz_session = psz_sesbuf;
547 /* FIXME: we probably need to remove an access out,
548 * if there is already one for the same ID */
549 ses = RtspClientGet( rtsp, psz_session );
552 answer->i_status = 454;
553 vlc_mutex_unlock( &rtsp->lock );
558 INSERT_ELEM( ses->trackv, ses->trackc, ses->trackc,
560 vlc_mutex_unlock( &rtsp->lock );
562 httpd_ServerIP( cl, ip );
564 if( strcmp( src, ip ) )
566 /* Specify source IP if it is different from the RTSP
567 * control connection server address */
568 char *ptr = strchr( src, '%' );
569 if( ptr != NULL ) *ptr = '\0'; /* remove scope ID */
571 httpd_MsgAdd( answer, "Transport",
572 "RTP/AVP/UDP;unicast;source=%s;"
573 "client_port=%u-%u;server_port=%u-%u;"
574 "ssrc=%08X;mode=play",
575 src, loport, loport + 1, sport,
576 sport + 1, id->ssrc );
580 httpd_MsgAdd( answer, "Transport",
581 "RTP/AVP/UDP;unicast;"
582 "client_port=%u-%u;server_port=%u-%u;"
583 "ssrc=%08X;mode=play",
584 loport, loport + 1, sport, sport + 1,
588 answer->i_status = 200;
597 answer->i_status = 200;
599 psz_session = httpd_MsgGet( query, "Session" );
600 if( httpd_MsgGet( query, "Range" ) != NULL )
602 answer->i_status = 456; /* cannot seek */
606 vlc_mutex_lock( &rtsp->lock );
607 ses = RtspClientGet( rtsp, psz_session );
610 /* FIXME: we really need to limit the number of tracks... */
611 char info[ses->trackc * ( strlen( control )
612 + sizeof("/trackID=123;seq=65535, ") ) + 1];
615 for( int i = 0; i < ses->trackc; i++ )
617 rtsp_strack_t *tr = ses->trackv + i;
618 if( ( id == NULL ) || ( tr->id == id->sout_id ) )
622 tr->playing = VLC_TRUE;
623 rtp_add_sink( tr->id, tr->fd, VLC_FALSE );
625 infolen += sprintf( info + infolen,
626 "%s/trackID=%u;seq=%u, ", control,
627 rtp_get_num( tr->id ),
628 rtp_get_seq( tr->id ) );
633 info[infolen - 2] = '\0'; /* remove trailing ", " */
634 httpd_MsgAdd( answer, "RTP-Info", "%s", info );
637 vlc_mutex_unlock( &rtsp->lock );
639 if( httpd_MsgGet( query, "Scale" ) != NULL )
640 httpd_MsgAdd( answer, "Scale", "1." );
644 case HTTPD_MSG_PAUSE:
645 answer->i_status = 405;
646 httpd_MsgAdd( answer, "Allow",
647 "%s, TEARDOWN, PLAY, GET_PARAMETER",
648 ( id != NULL ) ? "SETUP" : "DESCRIBE" );
651 case HTTPD_MSG_GETPARAMETER:
652 if( query->i_body > 0 )
654 answer->i_status = 451;
658 psz_session = httpd_MsgGet( query, "Session" );
659 answer->i_status = 200;
662 case HTTPD_MSG_TEARDOWN:
666 answer->i_status = 200;
668 psz_session = httpd_MsgGet( query, "Session" );
670 vlc_mutex_lock( &rtsp->lock );
671 ses = RtspClientGet( rtsp, psz_session );
674 if( id == NULL ) /* Delete the entire session */
675 RtspClientDel( rtsp, ses );
676 else /* Delete one track from the session */
677 for( int i = 0; i < ses->trackc; i++ )
679 if( ses->trackv[i].id == id->sout_id )
681 rtp_del_sink( id->sout_id, ses->trackv[i].fd );
682 REMOVE_ELEM( ses->trackv, ses->trackc, i );
686 vlc_mutex_unlock( &rtsp->lock );
694 httpd_MsgAdd( answer, "Server", "%s", PACKAGE_STRING );
696 httpd_MsgAdd( answer, "Session", "%s"/*;timeout=5*/, psz_session );
698 httpd_MsgAdd( answer, "Content-Length", "%d", answer->i_body );
699 httpd_MsgAdd( answer, "Cache-Control", "no-cache" );
701 psz = httpd_MsgGet( query, "Cseq" );
703 httpd_MsgAdd( answer, "Cseq", "%s", psz );
704 psz = httpd_MsgGet( query, "Timestamp" );
706 httpd_MsgAdd( answer, "Timestamp", "%s", psz );
712 /** Aggregate RTSP callback */
713 static int RtspCallback( httpd_callback_sys_t *p_args,
715 httpd_message_t *answer,
716 const httpd_message_t *query )
718 return RtspHandler( (rtsp_stream_t *)p_args, NULL, cl, answer, query );
722 /** Non-aggregate RTSP callback */
723 static int RtspCallbackId( httpd_callback_sys_t *p_args,
725 httpd_message_t *answer,
726 const httpd_message_t *query )
728 rtsp_stream_id_t *id = (rtsp_stream_id_t *)p_args;
729 return RtspHandler( id->stream, id, cl, answer, query );