1 /*****************************************************************************
2 * vod.c: rtsp VoD server module
3 *****************************************************************************
4 * Copyright (C) 2003-2006, 2010 VLC authors and VideoLAN
7 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
8 * Gildas Bazin <gbazin@videolan.org>
11 * This program is free software; you can redistribute it and/or modify it
12 * under the terms of the GNU Lesser General Public License as published by
13 * the Free Software Foundation; either version 2.1 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 Lesser General Public License for more details.
21 * You should have received a copy of the GNU Lesser General Public License
22 * along with this program; if not, write to the Free Software Foundation,
23 * Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
24 *****************************************************************************/
26 /*****************************************************************************
28 *****************************************************************************/
34 #include <vlc_common.h>
35 #include <vlc_plugin.h>
36 #include <vlc_input.h>
38 #include <vlc_block.h>
42 #include <vlc_network.h>
48 /*****************************************************************************
50 *****************************************************************************/
52 typedef struct media_es_t media_es_t;
58 rtsp_stream_id_t *rtsp_id;
84 block_fifo_t *p_fifo_cmd;
87 /* rtsp delayed command (to avoid deadlock between vlm/httpd) */
103 static vod_media_t *MediaNew( vod_t *, const char *, input_item_t * );
104 static void MediaDel( vod_t *, vod_media_t * );
105 static void MediaAskDel ( vod_t *, vod_media_t * );
107 static void* CommandThread( void *obj );
108 static void CommandPush( vod_t *, rtsp_cmd_type_t, vod_media_t *,
109 const char *psz_arg );
111 /*****************************************************************************
112 * Open: Starts the RTSP server module
113 *****************************************************************************/
114 int OpenVoD( vlc_object_t *p_this )
116 vod_t *p_vod = (vod_t *)p_this;
117 vod_sys_t *p_sys = NULL;
120 p_vod->p_sys = p_sys = malloc( sizeof( vod_sys_t ) );
121 if( !p_sys ) goto error;
123 psz_url = var_InheritString( p_vod, "rtsp-host" );
125 if( psz_url == NULL )
126 p_sys->psz_rtsp_path = strdup( "/" );
130 vlc_UrlParse( &url, psz_url, 0 );
133 if( url.psz_path == NULL )
134 p_sys->psz_rtsp_path = strdup( "/" );
136 if( !( strlen( url.psz_path ) > 0
137 && url.psz_path[strlen( url.psz_path ) - 1] == '/' ) )
139 if( asprintf( &p_sys->psz_rtsp_path, "%s/", url.psz_path ) == -1 )
141 p_sys->psz_rtsp_path = NULL;
142 vlc_UrlClean( &url );
147 p_sys->psz_rtsp_path = strdup( url.psz_path );
149 vlc_UrlClean( &url );
152 p_vod->pf_media_new = MediaNew;
153 p_vod->pf_media_del = MediaAskDel;
155 p_sys->p_fifo_cmd = block_FifoNew();
156 if( vlc_clone( &p_sys->thread, CommandThread, p_vod, VLC_THREAD_PRIORITY_LOW ) )
158 msg_Err( p_vod, "cannot spawn rtsp vod thread" );
159 block_FifoRelease( p_sys->p_fifo_cmd );
168 free( p_sys->psz_rtsp_path );
175 /*****************************************************************************
177 *****************************************************************************/
178 void CloseVoD( vlc_object_t * p_this )
180 vod_t *p_vod = (vod_t *)p_this;
181 vod_sys_t *p_sys = p_vod->p_sys;
183 /* Stop command thread */
184 vlc_cancel( p_sys->thread );
185 vlc_join( p_sys->thread, NULL );
187 while( block_FifoCount( p_sys->p_fifo_cmd ) > 0 )
190 block_t *p_block_cmd = block_FifoGet( p_sys->p_fifo_cmd );
191 memcpy( &cmd, p_block_cmd->p_buffer, sizeof(cmd) );
192 block_Release( p_block_cmd );
193 if ( cmd.i_type == RTSP_CMD_TYPE_DEL )
194 MediaDel(p_vod, cmd.p_media);
197 block_FifoRelease( p_sys->p_fifo_cmd );
199 free( p_sys->psz_rtsp_path );
203 /*****************************************************************************
205 *****************************************************************************/
206 static vod_media_t *MediaNew( vod_t *p_vod, const char *psz_name,
207 input_item_t *p_item )
209 vod_media_t *p_media = calloc( 1, sizeof(vod_media_t) );
213 p_media->p_vod = p_vod;
214 p_media->rtsp = NULL;
215 TAB_INIT( p_media->i_es, p_media->es );
216 p_media->psz_mux = NULL;
217 p_media->i_length = input_item_GetDuration( p_item );
219 vlc_mutex_lock( &p_item->lock );
220 msg_Dbg( p_vod, "media '%s' has %i declared ES", psz_name, p_item->i_es );
221 for( int i = 0; i < p_item->i_es; i++ )
223 es_format_t *p_fmt = p_item->es[i];
225 switch( p_fmt->i_codec )
227 case VLC_FOURCC( 'm', 'p', '2', 't' ):
228 p_media->psz_mux = "ts";
230 case VLC_FOURCC( 'm', 'p', '2', 'p' ):
231 p_media->psz_mux = "ps";
234 assert(p_media->psz_mux == NULL || p_item->i_es == 1);
236 media_es_t *p_es = calloc( 1, sizeof(media_es_t) );
240 p_es->es_id = p_fmt->i_id;
241 p_es->rtsp_id = NULL;
243 if (rtp_get_fmt(VLC_OBJECT(p_vod), p_fmt, p_media->psz_mux,
244 &p_es->rtp_fmt) != VLC_SUCCESS)
250 TAB_APPEND( p_media->i_es, p_media->es, p_es );
251 msg_Dbg(p_vod, " - added ES %u %s (%4.4s)",
252 p_es->rtp_fmt.payload_type, p_es->rtp_fmt.ptname,
253 (char *)&p_fmt->i_codec);
255 vlc_mutex_unlock( &p_item->lock );
257 if (p_media->i_es == 0)
259 msg_Err(p_vod, "no ES was added to the media, aborting");
263 msg_Dbg(p_vod, "adding media '%s'", psz_name);
265 CommandPush( p_vod, RTSP_CMD_TYPE_ADD, p_media, psz_name );
269 MediaDel(p_vod, p_media);
273 static void MediaSetup( vod_t *p_vod, vod_media_t *p_media,
274 const char *psz_name )
276 vod_sys_t *p_sys = p_vod->p_sys;
279 if( asprintf( &psz_path, "%s%s", p_sys->psz_rtsp_path, psz_name ) < 0 )
282 p_media->rtsp = RtspSetup(VLC_OBJECT(p_vod), p_media, psz_path);
285 if (p_media->rtsp == NULL)
288 for (int i = 0; i < p_media->i_es; i++)
290 media_es_t *p_es = p_media->es[i];
291 p_es->rtsp_id = RtspAddId(p_media->rtsp, NULL, 0,
292 p_es->rtp_fmt.clock_rate, -1);
296 static void MediaAskDel ( vod_t *p_vod, vod_media_t *p_media )
298 msg_Dbg( p_vod, "deleting media" );
299 CommandPush( p_vod, RTSP_CMD_TYPE_DEL, p_media, NULL );
302 static void MediaDel( vod_t *p_vod, vod_media_t *p_media )
306 if (p_media->rtsp != NULL)
308 for (int i = 0; i < p_media->i_es; i++)
310 media_es_t *p_es = p_media->es[i];
311 if (p_es->rtsp_id != NULL)
312 RtspDelId(p_media->rtsp, p_es->rtsp_id);
314 RtspUnsetup(p_media->rtsp);
317 while( p_media->i_es )
319 media_es_t *p_es = p_media->es[0];
320 TAB_REMOVE( p_media->i_es, p_media->es, p_es );
321 free( p_es->rtp_fmt.fmtp );
325 TAB_CLEAN( p_media->i_es, p_media->es );
329 static void CommandPush( vod_t *p_vod, rtsp_cmd_type_t i_type,
330 vod_media_t *p_media, const char *psz_arg )
336 cmd.p_media = p_media;
338 cmd.psz_arg = strdup(psz_arg);
342 p_cmd = block_Alloc( sizeof(rtsp_cmd_t) );
343 memcpy( p_cmd->p_buffer, &cmd, sizeof(cmd) );
345 block_FifoPut( p_vod->p_sys->p_fifo_cmd, p_cmd );
348 static void* CommandThread( void *obj )
350 vod_t *p_vod = (vod_t*)obj;
351 vod_sys_t *p_sys = p_vod->p_sys;
355 block_t *p_block_cmd = block_FifoGet( p_sys->p_fifo_cmd );
361 int canc = vlc_savecancel ();
362 memcpy( &cmd, p_block_cmd->p_buffer, sizeof(cmd) );
363 block_Release( p_block_cmd );
368 case RTSP_CMD_TYPE_ADD:
369 MediaSetup(p_vod, cmd.p_media, cmd.psz_arg);
371 case RTSP_CMD_TYPE_DEL:
372 MediaDel(p_vod, cmd.p_media);
374 case RTSP_CMD_TYPE_STOP:
375 vod_MediaControl( p_vod, cmd.p_media, cmd.psz_arg, VOD_MEDIA_STOP );
383 vlc_restorecancel (canc);
389 /*****************************************************************************
391 * FIXME: needs to be merged more?
392 *****************************************************************************/
393 char *SDPGenerateVoD( const vod_media_t *p_media, const char *rtsp_url )
397 assert(rtsp_url != NULL);
398 /* Check against URL format rtsp://[<ipv6>]:<port>/<path> */
399 bool ipv6 = strlen( rtsp_url ) > 7 && rtsp_url[7] == '[';
401 /* Dummy destination address for RTSP */
402 struct sockaddr_storage dst;
403 socklen_t dstlen = ipv6 ? sizeof( struct sockaddr_in6 )
404 : sizeof( struct sockaddr_in );
405 memset (&dst, 0, dstlen);
406 dst.ss_family = ipv6 ? AF_INET6 : AF_INET;
411 psz_sdp = vlc_sdp_Start( VLC_OBJECT( p_media->p_vod ), "sout-rtp-",
412 NULL, 0, (struct sockaddr *)&dst, dstlen );
413 if( psz_sdp == NULL )
416 if( p_media->i_length > 0 )
418 lldiv_t d = lldiv( p_media->i_length / 1000, 1000 );
419 sdp_AddAttribute( &psz_sdp, "range"," npt=0-%lld.%03u", d.quot,
423 sdp_AddAttribute ( &psz_sdp, "control", "%s", rtsp_url );
425 /* No locking needed, the ES table can't be modified now */
426 for( int i = 0; i < p_media->i_es; i++ )
428 media_es_t *p_es = p_media->es[i];
429 rtp_format_t *rtp_fmt = &p_es->rtp_fmt;
430 const char *mime_major; /* major MIME type */
432 switch( rtp_fmt->cat )
435 mime_major = "video";
438 mime_major = "audio";
447 sdp_AddMedia( &psz_sdp, mime_major, "RTP/AVP", 0,
448 rtp_fmt->payload_type, false, 0,
449 rtp_fmt->ptname, rtp_fmt->clock_rate, rtp_fmt->channels,
452 char *track_url = RtspAppendTrackPath( p_es->rtsp_id, rtsp_url );
453 if( track_url != NULL )
455 sdp_AddAttribute ( &psz_sdp, "control", "%s", track_url );
463 int vod_check_range(vod_media_t *p_media, const char *psz_session,
464 int64_t start, int64_t end)
468 if (p_media->i_length > 0 && (start > p_media->i_length
469 || end > p_media->i_length))
475 /* TODO: add support in the VLM for queueing proper PLAY requests with
476 * start and end times, fetch whether the input is seekable... and then
478 void vod_play(vod_media_t *p_media, const char *psz_session,
479 int64_t *start, int64_t end)
481 if (vod_check_range(p_media, psz_session, *start, end) != VLC_SUCCESS)
484 /* We're passing the #vod{} sout chain here */
485 vod_MediaControl(p_media->p_vod, p_media, psz_session,
486 VOD_MEDIA_PLAY, "vod", start);
489 void vod_pause(vod_media_t *p_media, const char *psz_session, int64_t *npt)
491 vod_MediaControl(p_media->p_vod, p_media, psz_session,
492 VOD_MEDIA_PAUSE, npt);
495 void vod_stop(vod_media_t *p_media, const char *psz_session)
497 CommandPush(p_media->p_vod, RTSP_CMD_TYPE_STOP, p_media, psz_session);
501 const char *vod_get_mux(const vod_media_t *p_media)
503 return p_media->psz_mux;
507 /* Match an RTP id to a VoD media ES and RTSP track to initialize it
508 * with the data that was already set up */
509 int vod_init_id(vod_media_t *p_media, const char *psz_session, int es_id,
510 sout_stream_id_t *sout_id, rtp_format_t *rtp_fmt,
511 uint32_t *ssrc, uint16_t *seq_init)
515 if (p_media->psz_mux != NULL)
517 assert(p_media->i_es == 1);
518 p_es = p_media->es[0];
523 /* No locking needed, the ES table can't be modified now */
524 for (int i = 0; i < p_media->i_es; i++)
526 if (p_media->es[i]->es_id == es_id)
528 p_es = p_media->es[i];
536 memcpy(rtp_fmt, &p_es->rtp_fmt, sizeof(*rtp_fmt));
537 if (p_es->rtp_fmt.fmtp != NULL)
538 rtp_fmt->fmtp = strdup(p_es->rtp_fmt.fmtp);
540 return RtspTrackAttach(p_media->rtsp, psz_session, p_es->rtsp_id,
541 sout_id, ssrc, seq_init);
544 /* Remove references to the RTP id from its RTSP track */
545 void vod_detach_id(vod_media_t *p_media, const char *psz_session,
546 sout_stream_id_t *sout_id)
548 RtspTrackDetach(p_media->rtsp, psz_session, sout_id);