1 /*****************************************************************************
2 * es_out_timeshift.c: Es Out timeshift.
3 *****************************************************************************
4 * Copyright (C) 2008 Laurent Aimar
7 * Authors: Laurent Aimar < fenrir _AT_ videolan _DOT_ org>
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, write to the Free Software
21 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
22 *****************************************************************************/
24 /*****************************************************************************
26 *****************************************************************************/
34 #if defined (WIN32) && !defined (UNDER_CE)
37 #ifdef HAVE_SYS_STAT_H
38 # include <sys/stat.h>
41 #include <vlc_common.h>
42 #include <vlc_charset.h>
44 #include <vlc_input.h>
45 #include <vlc_es_out.h>
46 #include <vlc_block.h>
47 #include "input_internal.h"
49 #include "es_out_timeshift.h"
51 /*****************************************************************************
53 *****************************************************************************/
55 /* XXX attribute_packed is (and MUST be) used ONLY to reduce memory usage */
56 #ifdef HAVE_ATTRIBUTE_PACKED
57 # define attribute_packed __attribute__((__packed__))
59 # define attribute_packed
70 typedef struct attribute_packed
76 typedef struct attribute_packed
81 typedef struct attribute_packed
85 int i_offset; /* We do not use file > INT_MAX */
88 typedef struct attribute_packed
125 /* FIXME Really too big (double make the whole thing too big) */
138 typedef struct attribute_packed
147 ts_cmd_control_t control;
151 typedef struct ts_storage_t ts_storage_t;
154 ts_storage_t *p_next;
157 char *psz_file; /* Filename */
158 size_t i_file_max; /* Max size in bytes */
159 int64_t i_file_size;/* Current size in bytes */
160 FILE *p_filew; /* FILE handle for data writing */
161 FILE *p_filer; /* FILE handle for data reading */
175 input_thread_t *p_input;
177 int64_t i_tmp_size_max;
178 const char *psz_tmp_path;
180 /* Lock for all following fields */
186 mtime_t i_pause_date;
192 mtime_t i_rate_delay;
195 mtime_t i_buffering_delay;
198 ts_storage_t *p_storage_r;
199 ts_storage_t *p_storage_w;
212 input_thread_t *p_input;
216 int64_t i_tmp_size_max; /* Maximal temporary file size in byte */
217 char *psz_tmp_path; /* Path for temporary files */
219 /* Lock for all following fields */
224 ts_thread_t *p_thread;
228 bool b_input_paused_source;
230 int i_input_rate_source;
237 static es_out_id_t *Add ( es_out_t *, const es_format_t * );
238 static int Send ( es_out_t *, es_out_id_t *, block_t * );
239 static void Del ( es_out_t *, es_out_id_t * );
240 static int Control( es_out_t *, int i_query, va_list );
241 static void Destroy( es_out_t * );
243 static int TsStart( es_out_t * );
244 static void TsAutoStop( es_out_t * );
246 static void TsStop( ts_thread_t * );
247 static void TsPushCmd( ts_thread_t *, ts_cmd_t * );
248 static int TsPopCmdLocked( ts_thread_t *, ts_cmd_t *, bool b_flush );
249 static bool TsHasCmd( ts_thread_t * );
250 static bool TsIsUnused( ts_thread_t * );
251 static int TsChangePause( ts_thread_t *, bool b_source_paused, bool b_paused, mtime_t i_date );
252 static int TsChangeRate( ts_thread_t *, int i_src_rate, int i_rate );
254 static void *TsRun( vlc_object_t * );
256 static ts_storage_t *TsStorageNew( const char *psz_path, int64_t i_tmp_size_max );
257 static void TsStorageDelete( ts_storage_t * );
258 static void TsStoragePack( ts_storage_t *p_storage );
259 static bool TsStorageIsFull( ts_storage_t *, const ts_cmd_t *p_cmd );
260 static bool TsStorageIsEmpty( ts_storage_t * );
261 static void TsStoragePushCmd( ts_storage_t *, const ts_cmd_t *p_cmd, bool b_flush );
262 static void TsStoragePopCmd( ts_storage_t *p_storage, ts_cmd_t *p_cmd, bool b_flush );
264 static void CmdClean( ts_cmd_t * );
265 static void cmd_cleanup_routine( void *p ) { CmdClean( p ); }
267 static int CmdInitAdd ( ts_cmd_t *, es_out_id_t *, const es_format_t *, bool b_copy );
268 static void CmdInitSend ( ts_cmd_t *, es_out_id_t *, block_t * );
269 static int CmdInitDel ( ts_cmd_t *, es_out_id_t * );
270 static int CmdInitControl( ts_cmd_t *, int i_query, va_list, bool b_copy );
273 static void CmdCleanAdd ( ts_cmd_t * );
274 static void CmdCleanSend ( ts_cmd_t * );
275 static void CmdCleanControl( ts_cmd_t *p_cmd );
277 /* XXX these functions will take the destination es_out_t */
278 static void CmdExecuteAdd ( es_out_t *, ts_cmd_t * );
279 static int CmdExecuteSend ( es_out_t *, ts_cmd_t * );
280 static void CmdExecuteDel ( es_out_t *, ts_cmd_t * );
281 static int CmdExecuteControl( es_out_t *, ts_cmd_t * );
284 static char *GetTmpPath( char *psz_path );
285 static FILE *GetTmpFile( char **ppsz_file, const char *psz_path );
287 /*****************************************************************************
288 * input_EsOutTimeshiftNew:
289 *****************************************************************************/
290 es_out_t *input_EsOutTimeshiftNew( input_thread_t *p_input, es_out_t *p_next_out, int i_rate )
292 es_out_t *p_out = malloc( sizeof(*p_out) );
296 es_out_sys_t *p_sys = malloc( sizeof(*p_sys) );
305 p_out->pf_send = Send;
307 p_out->pf_control = Control;
308 p_out->pf_destroy = Destroy;
309 p_out->p_sys = p_sys;
310 p_out->b_sout = p_input->p->p_sout != NULL;
313 p_sys->b_input_paused = false;
314 p_sys->b_input_paused_source = false;
315 p_sys->p_input = p_input;
316 p_sys->i_input_rate = i_rate;
317 p_sys->i_input_rate_source = i_rate;
319 p_sys->p_out = p_next_out;
320 vlc_mutex_init_recursive( &p_sys->lock );
322 p_sys->b_delayed = false;
323 p_sys->p_thread = NULL;
325 TAB_INIT( p_sys->i_es, p_sys->pp_es );
328 const int i_tmp_size_max = var_CreateGetInteger( p_input, "input-timeshift-granularity" );
329 if( i_tmp_size_max < 0 )
330 p_sys->i_tmp_size_max = 50*1024*1024;
332 p_sys->i_tmp_size_max = __MAX( i_tmp_size_max, 1*1024*1024 );
333 msg_Dbg( p_input, "using timeshift granularity of %d MBytes",
334 (int)p_sys->i_tmp_size_max/(1024*1024) );
336 char *psz_tmp_path = var_CreateGetNonEmptyString( p_input, "input-timeshift-path" );
337 p_sys->psz_tmp_path = GetTmpPath( psz_tmp_path );
338 msg_Dbg( p_input, "using timeshift path '%s'", p_sys->psz_tmp_path );
341 #define S(t) msg_Err( p_input, "SIZEOF("#t")=%d", sizeof(t) )
353 /*****************************************************************************
355 *****************************************************************************/
356 static void Destroy( es_out_t *p_out )
358 es_out_sys_t *p_sys = p_out->p_sys;
360 if( p_sys->b_delayed )
362 TsStop( p_sys->p_thread );
363 p_sys->b_delayed = false;
366 while( p_sys->i_es > 0 )
367 Del( p_out, p_sys->pp_es[0] );
368 TAB_CLEAN( p_sys->i_es, p_sys->pp_es );
370 free( p_sys->psz_tmp_path );
371 vlc_mutex_destroy( &p_sys->lock );
376 static es_out_id_t *Add( es_out_t *p_out, const es_format_t *p_fmt )
378 es_out_sys_t *p_sys = p_out->p_sys;
381 es_out_id_t *p_es = malloc( sizeof( *p_es ) );
385 vlc_mutex_lock( &p_sys->lock );
389 if( CmdInitAdd( &cmd, p_es, p_fmt, p_sys->b_delayed ) )
391 vlc_mutex_unlock( &p_sys->lock );
396 TAB_APPEND( p_sys->i_es, p_sys->pp_es, p_es );
398 if( p_sys->b_delayed )
399 TsPushCmd( p_sys->p_thread, &cmd );
401 CmdExecuteAdd( p_sys->p_out, &cmd );
403 vlc_mutex_unlock( &p_sys->lock );
407 static int Send( es_out_t *p_out, es_out_id_t *p_es, block_t *p_block )
409 es_out_sys_t *p_sys = p_out->p_sys;
411 int i_ret = VLC_SUCCESS;
413 vlc_mutex_lock( &p_sys->lock );
417 CmdInitSend( &cmd, p_es, p_block );
418 if( p_sys->b_delayed )
419 TsPushCmd( p_sys->p_thread, &cmd );
421 i_ret = CmdExecuteSend( p_sys->p_out, &cmd) ;
423 vlc_mutex_unlock( &p_sys->lock );
427 static void Del( es_out_t *p_out, es_out_id_t *p_es )
429 es_out_sys_t *p_sys = p_out->p_sys;
432 vlc_mutex_lock( &p_sys->lock );
436 CmdInitDel( &cmd, p_es );
437 if( p_sys->b_delayed )
438 TsPushCmd( p_sys->p_thread, &cmd );
440 CmdExecuteDel( p_sys->p_out, &cmd );
442 TAB_REMOVE( p_sys->i_es, p_sys->pp_es, p_es );
444 vlc_mutex_unlock( &p_sys->lock );
447 static int ControlLockedGetEmpty( es_out_t *p_out, bool *pb_empty )
449 es_out_sys_t *p_sys = p_out->p_sys;
451 if( p_sys->b_delayed && TsHasCmd( p_sys->p_thread ) )
454 *pb_empty = es_out_GetEmpty( p_sys->p_out );
458 static int ControlLockedGetWakeup( es_out_t *p_out, mtime_t *pi_wakeup )
460 es_out_sys_t *p_sys = p_out->p_sys;
462 if( p_sys->b_delayed )
464 assert( !p_sys->p_input->p->b_can_pace_control );
469 *pi_wakeup = es_out_GetWakeup( p_sys->p_out );
474 static int ControlLockedGetBuffering( es_out_t *p_out, bool *pb_buffering )
476 es_out_sys_t *p_sys = p_out->p_sys;
478 if( p_sys->b_delayed )
479 *pb_buffering = true;
481 *pb_buffering = es_out_GetBuffering( p_sys->p_out );
485 static int ControlLockedSetPauseState( es_out_t *p_out, bool b_source_paused, bool b_paused, mtime_t i_date )
487 es_out_sys_t *p_sys = p_out->p_sys;
490 if( !p_sys->b_delayed && !b_source_paused == !b_paused )
492 i_ret = es_out_SetPauseState( p_sys->p_out, b_source_paused, b_paused, i_date );
496 i_ret = VLC_EGENERIC;
497 if( !p_sys->p_input->p->b_can_pace_control )
499 if( !p_sys->b_delayed )
501 if( p_sys->b_delayed )
502 i_ret = TsChangePause( p_sys->p_thread, b_source_paused, b_paused, i_date );
506 /* XXX we may do it BUT it would be better to finish the clock clean up+improvments
507 * and so be able to advertize correctly pace control property in access
509 msg_Err( p_sys->p_input, "EsOutTimeshift does not work with streams that have space control" );
515 p_sys->b_input_paused_source = b_source_paused;
516 p_sys->b_input_paused = b_paused;
520 static int ControlLockedSetRate( es_out_t *p_out, int i_src_rate, int i_rate )
522 es_out_sys_t *p_sys = p_out->p_sys;
525 if( !p_sys->b_delayed && i_src_rate == i_rate )
527 i_ret = es_out_SetRate( p_sys->p_out, i_src_rate, i_rate );
531 i_ret = VLC_EGENERIC;
532 if( !p_sys->p_input->p->b_can_pace_control )
534 if( !p_sys->b_delayed )
536 if( p_sys->b_delayed )
537 i_ret = TsChangeRate( p_sys->p_thread, i_src_rate, i_rate );
541 /* XXX we may do it BUT it would be better to finish the clock clean up+improvments
542 * and so be able to advertize correctly pace control property in access
544 msg_Err( p_sys->p_input, "EsOutTimeshift does not work with streams that have space control" );
551 p_sys->i_input_rate_source = i_src_rate;
552 p_sys->i_input_rate = i_rate;
556 static int ControlLockedSetTime( es_out_t *p_out, mtime_t i_date )
558 es_out_sys_t *p_sys = p_out->p_sys;
560 if( !p_sys->b_delayed )
561 return es_out_SetTime( p_sys->p_out, i_date );
564 msg_Err( p_sys->p_input, "EsOutTimeshift does not yet support time change" );
567 static int ControlLockedSetFrameNext( es_out_t *p_out )
569 es_out_sys_t *p_sys = p_out->p_sys;
571 return es_out_SetFrameNext( p_sys->p_out );
574 static int ControlLocked( es_out_t *p_out, int i_query, va_list args )
576 es_out_sys_t *p_sys = p_out->p_sys;
580 /* Invalid query for this es_out level */
581 case ES_OUT_SET_ES_BY_ID:
582 case ES_OUT_RESTART_ES_BY_ID:
583 case ES_OUT_SET_ES_DEFAULT_BY_ID:
584 case ES_OUT_SET_DELAY:
585 case ES_OUT_SET_RECORD_STATE:
589 /* Pass-through control */
590 case ES_OUT_SET_ACTIVE:
591 case ES_OUT_SET_MODE:
592 case ES_OUT_SET_GROUP:
594 case ES_OUT_SET_GROUP_PCR:
595 case ES_OUT_RESET_PCR:
596 case ES_OUT_SET_NEXT_DISPLAY_TIME:
597 case ES_OUT_SET_GROUP_META:
598 case ES_OUT_SET_GROUP_EPG:
599 case ES_OUT_SET_ES_SCRAMBLED_STATE:
600 case ES_OUT_DEL_GROUP:
601 case ES_OUT_SET_META:
603 case ES_OUT_RESTART_ES:
604 case ES_OUT_SET_ES_DEFAULT:
605 case ES_OUT_SET_ES_STATE:
606 case ES_OUT_SET_ES_FMT:
607 case ES_OUT_SET_TIMES:
608 case ES_OUT_SET_JITTER:
611 if( CmdInitControl( &cmd, i_query, args, p_sys->b_delayed ) )
613 if( p_sys->b_delayed )
615 TsPushCmd( p_sys->p_thread, &cmd );
618 return CmdExecuteControl( p_sys->p_out, &cmd );
621 /* Special control when delayed */
622 case ES_OUT_GET_ES_STATE:
624 es_out_id_t *p_es = (es_out_id_t*)va_arg( args, es_out_id_t * );
625 bool *pb_enabled = (bool*)va_arg( args, bool* );
627 if( p_sys->b_delayed )
632 return es_out_Control( p_sys->p_out, ES_OUT_GET_ES_STATE, p_es->p_es, pb_enabled );
635 /* Special internal input control */
636 case ES_OUT_GET_EMPTY:
638 bool *pb_empty = (bool*)va_arg( args, bool* );
639 return ControlLockedGetEmpty( p_out, pb_empty );
641 case ES_OUT_GET_WAKE_UP: /* TODO ? */
643 mtime_t *pi_wakeup = (mtime_t*)va_arg( args, mtime_t* );
644 return ControlLockedGetWakeup( p_out, pi_wakeup );
646 case ES_OUT_GET_BUFFERING:
648 bool *pb_buffering = (bool *)va_arg( args, bool* );
649 return ControlLockedGetBuffering( p_out, pb_buffering );
651 case ES_OUT_SET_PAUSE_STATE:
653 const bool b_source_paused = (bool)va_arg( args, int );
654 const bool b_paused = (bool)va_arg( args, int );
655 const mtime_t i_date = (mtime_t) va_arg( args, mtime_t );
657 return ControlLockedSetPauseState( p_out, b_source_paused, b_paused, i_date );
659 case ES_OUT_SET_RATE:
661 const int i_src_rate = (int)va_arg( args, int );
662 const int i_rate = (int)va_arg( args, int );
664 return ControlLockedSetRate( p_out, i_src_rate, i_rate );
666 case ES_OUT_SET_TIME:
668 const mtime_t i_date = (mtime_t)va_arg( args, mtime_t );
670 return ControlLockedSetTime( p_out, i_date );
672 case ES_OUT_SET_FRAME_NEXT:
674 return ControlLockedSetFrameNext( p_out );
676 case ES_OUT_GET_PCR_SYSTEM:
678 if( p_sys->b_delayed )
681 mtime_t *pi_system = (mtime_t*)va_arg( args, mtime_t * );
682 return es_out_ControlGetPcrSystem( p_sys->p_out, pi_system );
684 case ES_OUT_MODIFY_PCR_SYSTEM:
686 const bool b_absolute = va_arg( args, int );
687 const mtime_t i_system = va_arg( args, mtime_t );
689 if( b_absolute && p_sys->b_delayed )
692 return es_out_ControlModifyPcrSystem( p_sys->p_out, b_absolute, i_system );
695 msg_Err( p_sys->p_input, "Unknown es_out_Control query !" );
700 static int Control( es_out_t *p_out, int i_query, va_list args )
702 es_out_sys_t *p_sys = p_out->p_sys;
705 vlc_mutex_lock( &p_sys->lock );
709 i_ret = ControlLocked( p_out, i_query, args );
711 vlc_mutex_unlock( &p_sys->lock );
716 /*****************************************************************************
718 *****************************************************************************/
719 static void TsDestructor( vlc_object_t *p_this )
721 ts_thread_t *p_ts = (ts_thread_t*)p_this;
723 vlc_cond_destroy( &p_ts->wait );
724 vlc_mutex_destroy( &p_ts->lock );
726 static int TsStart( es_out_t *p_out )
728 es_out_sys_t *p_sys = p_out->p_sys;
731 assert( !p_sys->b_delayed );
733 p_sys->p_thread = p_ts = vlc_custom_create( p_sys->p_input, sizeof(ts_thread_t),
734 VLC_OBJECT_GENERIC, "es out timeshift" );
738 p_ts->i_tmp_size_max = p_sys->i_tmp_size_max;
739 p_ts->psz_tmp_path = p_sys->psz_tmp_path;
740 p_ts->p_input = p_sys->p_input;
741 p_ts->p_out = p_sys->p_out;
742 vlc_mutex_init( &p_ts->lock );
743 vlc_cond_init( &p_ts->wait );
744 p_ts->b_paused = p_sys->b_input_paused && !p_sys->b_input_paused_source;
745 p_ts->i_pause_date = p_ts->b_paused ? mdate() : -1;
746 p_ts->i_rate_source = p_sys->i_input_rate_source;
747 p_ts->i_rate = p_sys->i_input_rate;
748 p_ts->i_rate_date = -1;
749 p_ts->i_rate_delay = 0;
750 p_ts->i_buffering_delay = 0;
751 p_ts->i_cmd_delay = 0;
752 p_ts->p_storage_r = NULL;
753 p_ts->p_storage_w = NULL;
755 vlc_object_set_destructor( p_ts, TsDestructor );
757 p_sys->b_delayed = true;
758 if( vlc_thread_create( p_ts, "es out timeshift",
759 TsRun, VLC_THREAD_PRIORITY_INPUT ) )
761 msg_Err( p_sys->p_input, "cannot create input thread" );
763 vlc_object_release( p_ts );
765 p_sys->b_delayed = false;
771 static void TsAutoStop( es_out_t *p_out )
773 es_out_sys_t *p_sys = p_out->p_sys;
775 if( !p_sys->b_delayed || !TsIsUnused( p_sys->p_thread ) )
778 msg_Warn( p_sys->p_input, "es out timeshift: auto stop" );
779 TsStop( p_sys->p_thread );
781 p_sys->b_delayed = false;
783 static void TsStop( ts_thread_t *p_ts )
785 vlc_object_kill( p_ts );
786 vlc_thread_join( p_ts );
788 vlc_mutex_lock( &p_ts->lock );
793 if( TsPopCmdLocked( p_ts, &cmd, true ) )
798 assert( !p_ts->p_storage_r || !p_ts->p_storage_r->p_next );
799 if( p_ts->p_storage_r )
800 TsStorageDelete( p_ts->p_storage_r );
801 vlc_mutex_unlock( &p_ts->lock );
803 vlc_object_release( p_ts );
805 static void TsPushCmd( ts_thread_t *p_ts, ts_cmd_t *p_cmd )
807 vlc_mutex_lock( &p_ts->lock );
809 if( !p_ts->p_storage_w || TsStorageIsFull( p_ts->p_storage_w, p_cmd ) )
811 ts_storage_t *p_storage = TsStorageNew( p_ts->psz_tmp_path, p_ts->i_tmp_size_max );
816 vlc_mutex_unlock( &p_ts->lock );
817 /* TODO warn the user (but only once) */
821 if( !p_ts->p_storage_w )
823 p_ts->p_storage_r = p_ts->p_storage_w = p_storage;
827 TsStoragePack( p_ts->p_storage_w );
828 p_ts->p_storage_w->p_next = p_storage;
829 p_ts->p_storage_w = p_storage;
833 /* TODO return error and warn the user (but only once) */
834 TsStoragePushCmd( p_ts->p_storage_w, p_cmd, p_ts->p_storage_r == p_ts->p_storage_w );
836 vlc_cond_signal( &p_ts->wait );
838 vlc_mutex_unlock( &p_ts->lock );
840 static int TsPopCmdLocked( ts_thread_t *p_ts, ts_cmd_t *p_cmd, bool b_flush )
842 vlc_assert_locked( &p_ts->lock );
844 if( TsStorageIsEmpty( p_ts->p_storage_r ) )
847 TsStoragePopCmd( p_ts->p_storage_r, p_cmd, b_flush );
849 while( p_ts->p_storage_r && TsStorageIsEmpty( p_ts->p_storage_r ) )
851 ts_storage_t *p_next = p_ts->p_storage_r->p_next;
855 TsStorageDelete( p_ts->p_storage_r );
856 p_ts->p_storage_r = p_next;
861 static bool TsHasCmd( ts_thread_t *p_ts )
865 vlc_mutex_lock( &p_ts->lock );
866 b_cmd = TsStorageIsEmpty( p_ts->p_storage_r );
867 vlc_mutex_unlock( &p_ts->lock );
871 static bool TsIsUnused( ts_thread_t *p_ts )
875 vlc_mutex_lock( &p_ts->lock );
876 b_unused = !p_ts->b_paused &&
877 p_ts->i_rate == p_ts->i_rate_source &&
878 TsStorageIsEmpty( p_ts->p_storage_r );
879 vlc_mutex_unlock( &p_ts->lock );
883 static int TsChangePause( ts_thread_t *p_ts, bool b_source_paused, bool b_paused, mtime_t i_date )
885 vlc_mutex_lock( &p_ts->lock );
890 assert( !b_source_paused );
891 i_ret = es_out_SetPauseState( p_ts->p_out, true, true, i_date );
895 i_ret = es_out_SetPauseState( p_ts->p_out, false, false, i_date );
902 assert( p_ts->i_pause_date > 0 );
904 p_ts->i_cmd_delay += i_date - p_ts->i_pause_date;
907 p_ts->b_paused = b_paused;
908 p_ts->i_pause_date = i_date;
910 vlc_cond_signal( &p_ts->wait );
912 vlc_mutex_unlock( &p_ts->lock );
915 static int TsChangeRate( ts_thread_t *p_ts, int i_src_rate, int i_rate )
919 vlc_mutex_lock( &p_ts->lock );
920 p_ts->i_cmd_delay += p_ts->i_rate_delay;
922 p_ts->i_rate_date = -1;
923 p_ts->i_rate_delay = 0;
924 p_ts->i_rate = i_rate;
925 p_ts->i_rate_source = i_src_rate;
927 i_ret = es_out_SetRate( p_ts->p_out, i_rate, i_rate );
928 vlc_mutex_unlock( &p_ts->lock );
933 static void *TsRun( vlc_object_t *p_thread )
935 ts_thread_t *p_ts = (ts_thread_t*)p_thread;
936 mtime_t i_buffering_date = -1;
944 /* Pop a command to execute */
945 vlc_mutex_lock( &p_ts->lock );
946 mutex_cleanup_push( &p_ts->lock );
950 const int canc = vlc_savecancel();
951 b_buffering = es_out_GetBuffering( p_ts->p_out );
953 if( ( !p_ts->b_paused || b_buffering ) && !TsPopCmdLocked( p_ts, &cmd, false ) )
955 vlc_restorecancel( canc );
958 vlc_restorecancel( canc );
960 vlc_cond_wait( &p_ts->wait, &p_ts->lock );
963 if( b_buffering && i_buffering_date < 0 )
965 i_buffering_date = cmd.i_date;
967 else if( i_buffering_date > 0 )
969 p_ts->i_buffering_delay += i_buffering_date - cmd.i_date; /* It is < 0 */
971 i_buffering_date = cmd.i_date;
973 i_buffering_date = -1;
976 if( p_ts->i_rate_date < 0 )
977 p_ts->i_rate_date = cmd.i_date;
979 p_ts->i_rate_delay = 0;
980 if( p_ts->i_rate_source != p_ts->i_rate )
982 const mtime_t i_duration = cmd.i_date - p_ts->i_rate_date;
983 p_ts->i_rate_delay = i_duration * p_ts->i_rate / p_ts->i_rate_source - i_duration;
985 if( p_ts->i_cmd_delay + p_ts->i_rate_delay + p_ts->i_buffering_delay < 0 && p_ts->i_rate != p_ts->i_rate_source )
987 const int canc = vlc_savecancel();
989 /* Auto reset to rate 1.0 */
990 msg_Warn( p_ts->p_input, "es out timeshift: auto reset rate to %d", p_ts->i_rate_source );
992 p_ts->i_cmd_delay = 0;
993 p_ts->i_buffering_delay = 0;
995 p_ts->i_rate_delay = 0;
996 p_ts->i_rate_date = -1;
997 p_ts->i_rate = p_ts->i_rate_source;
999 if( !es_out_SetRate( p_ts->p_out, p_ts->i_rate_source, p_ts->i_rate ) )
1001 vlc_value_t val = { .i_int = p_ts->i_rate };
1003 * FIXME it is perfectly safe BUT it is ugly as it may hide a
1004 * rate change requested by user */
1005 input_ControlPush( p_ts->p_input, INPUT_CONTROL_SET_RATE, &val );
1008 vlc_restorecancel( canc );
1010 i_deadline = cmd.i_date + p_ts->i_cmd_delay + p_ts->i_rate_delay + p_ts->i_buffering_delay;
1014 /* Regulate the speed of command processing to the same one than
1016 vlc_cleanup_push( cmd_cleanup_routine, &cmd );
1018 mwait( i_deadline );
1022 /* Execute the command */
1023 const int canc = vlc_savecancel();
1024 switch( cmd.i_type )
1027 CmdExecuteAdd( p_ts->p_out, &cmd );
1028 CmdCleanAdd( &cmd );
1031 CmdExecuteSend( p_ts->p_out, &cmd );
1032 CmdCleanSend( &cmd );
1035 CmdExecuteControl( p_ts->p_out, &cmd );
1036 CmdCleanControl( &cmd );
1039 CmdExecuteDel( p_ts->p_out, &cmd );
1045 vlc_restorecancel( canc );
1051 /*****************************************************************************
1053 *****************************************************************************/
1054 static ts_storage_t *TsStorageNew( const char *psz_tmp_path, int64_t i_tmp_size_max )
1056 ts_storage_t *p_storage = calloc( 1, sizeof(ts_storage_t) );
1061 p_storage->p_next = NULL;
1064 p_storage->i_file_max = i_tmp_size_max;
1065 p_storage->i_file_size = 0;
1066 p_storage->p_filew = GetTmpFile( &p_storage->psz_file, psz_tmp_path );
1067 if( p_storage->psz_file )
1068 p_storage->p_filer = utf8_fopen( p_storage->psz_file, "rb" );
1071 p_storage->i_cmd_w = 0;
1072 p_storage->i_cmd_r = 0;
1073 p_storage->i_cmd_max = 30000;
1074 p_storage->p_cmd = malloc( p_storage->i_cmd_max * sizeof(*p_storage->p_cmd) );
1075 //fprintf( stderr, "\nSTORAGE name=%s size=%d kbytes\n", p_storage->psz_file, p_storage->i_cmd_max * sizeof(*p_storage->p_cmd) /1024 );
1077 if( !p_storage->p_cmd || !p_storage->p_filew || !p_storage->p_filer )
1079 TsStorageDelete( p_storage );
1084 static void TsStorageDelete( ts_storage_t *p_storage )
1086 while( p_storage->i_cmd_r < p_storage->i_cmd_w )
1090 TsStoragePopCmd( p_storage, &cmd, true );
1094 free( p_storage->p_cmd );
1096 if( p_storage->p_filer )
1097 fclose( p_storage->p_filer );
1098 if( p_storage->p_filew )
1099 fclose( p_storage->p_filew );
1101 if( p_storage->psz_file )
1103 utf8_unlink( p_storage->psz_file );
1104 free( p_storage->psz_file );
1109 static void TsStoragePack( ts_storage_t *p_storage )
1111 /* Try to release a bit of memory */
1112 if( p_storage->i_cmd_w >= p_storage->i_cmd_max )
1115 p_storage->i_cmd_max = __MAX( p_storage->i_cmd_w, 1 );
1117 ts_cmd_t *p_new = realloc( p_storage->p_cmd, p_storage->i_cmd_max * sizeof(*p_storage->p_cmd) );
1119 p_storage->p_cmd = p_new;
1121 static bool TsStorageIsFull( ts_storage_t *p_storage, const ts_cmd_t *p_cmd )
1123 if( p_cmd && p_cmd->i_type == C_SEND && p_storage->i_cmd_w > 0 )
1125 size_t i_size = sizeof(*p_cmd->send.p_block) + p_cmd->send.p_block->i_buffer;
1127 if( p_storage->i_file_size + i_size >= p_storage->i_file_max )
1130 return p_storage->i_cmd_w >= p_storage->i_cmd_max;
1132 static bool TsStorageIsEmpty( ts_storage_t *p_storage )
1134 return !p_storage || p_storage->i_cmd_r >= p_storage->i_cmd_w;
1136 static void TsStoragePushCmd( ts_storage_t *p_storage, const ts_cmd_t *p_cmd, bool b_flush )
1138 ts_cmd_t cmd = *p_cmd;
1140 assert( !TsStorageIsFull( p_storage, p_cmd ) );
1142 if( cmd.i_type == C_SEND )
1144 block_t *p_block = cmd.send.p_block;
1146 cmd.send.p_block = NULL;
1147 cmd.send.i_offset = ftell( p_storage->p_filew );
1149 if( fwrite( p_block, sizeof(*p_block), 1, p_storage->p_filew ) != 1 )
1151 block_Release( p_block );
1154 p_storage->i_file_size += sizeof(*p_block);
1155 if( p_block->i_buffer > 0 )
1157 if( fwrite( p_block->p_buffer, p_block->i_buffer, 1, p_storage->p_filew ) != 1 )
1159 block_Release( p_block );
1163 p_storage->i_file_size += p_block->i_buffer;
1164 block_Release( p_block );
1167 fflush( p_storage->p_filew );
1169 p_storage->p_cmd[p_storage->i_cmd_w++] = cmd;
1171 static void TsStoragePopCmd( ts_storage_t *p_storage, ts_cmd_t *p_cmd, bool b_flush )
1173 assert( !TsStorageIsEmpty( p_storage ) );
1175 *p_cmd = p_storage->p_cmd[p_storage->i_cmd_r++];
1176 if( p_cmd->i_type == C_SEND )
1181 !fseek( p_storage->p_filer, p_cmd->send.i_offset, SEEK_SET ) &&
1182 fread( &block, sizeof(block), 1, p_storage->p_filer ) == 1 )
1184 block_t *p_block = block_Alloc( block.i_buffer );
1187 p_block->i_dts = block.i_dts;
1188 p_block->i_pts = block.i_pts;
1189 p_block->i_flags = block.i_flags;
1190 p_block->i_length = block.i_length;
1191 p_block->i_rate = block.i_rate;
1192 p_block->i_nb_samples = block.i_nb_samples;
1193 p_block->i_buffer = fread( p_block->p_buffer, 1, block.i_buffer, p_storage->p_filer );
1195 p_cmd->send.p_block = p_block;
1199 //fprintf( stderr, "TsStoragePopCmd: %m\n" );
1200 p_cmd->send.p_block = block_Alloc( 1 );
1205 /*****************************************************************************
1207 *****************************************************************************/
1208 static void CmdClean( ts_cmd_t *p_cmd )
1210 switch( p_cmd->i_type )
1213 CmdCleanAdd( p_cmd );
1216 CmdCleanSend( p_cmd );
1219 CmdCleanControl( p_cmd );
1229 static int CmdInitAdd( ts_cmd_t *p_cmd, es_out_id_t *p_es, const es_format_t *p_fmt, bool b_copy )
1231 p_cmd->i_type = C_ADD;
1232 p_cmd->i_date = mdate();
1233 p_cmd->add.p_es = p_es;
1236 p_cmd->add.p_fmt = malloc( sizeof(*p_fmt) );
1237 if( !p_cmd->add.p_fmt )
1238 return VLC_EGENERIC;
1239 es_format_Copy( p_cmd->add.p_fmt, p_fmt );
1243 p_cmd->add.p_fmt = (es_format_t*)p_fmt;
1247 static void CmdExecuteAdd( es_out_t *p_out, ts_cmd_t *p_cmd )
1249 p_cmd->add.p_es->p_es = es_out_Add( p_out, p_cmd->add.p_fmt );
1251 static void CmdCleanAdd( ts_cmd_t *p_cmd )
1253 es_format_Clean( p_cmd->add.p_fmt );
1254 free( p_cmd->add.p_fmt );
1257 static void CmdInitSend( ts_cmd_t *p_cmd, es_out_id_t *p_es, block_t *p_block )
1259 p_cmd->i_type = C_SEND;
1260 p_cmd->i_date = mdate();
1261 p_cmd->send.p_es = p_es;
1262 p_cmd->send.p_block = p_block;
1264 static int CmdExecuteSend( es_out_t *p_out, ts_cmd_t *p_cmd )
1266 block_t *p_block = p_cmd->send.p_block;
1268 p_cmd->send.p_block = NULL;
1272 if( p_cmd->send.p_es->p_es )
1273 return es_out_Send( p_out, p_cmd->send.p_es->p_es, p_block );
1274 block_Release( p_block );
1276 return VLC_EGENERIC;
1278 static void CmdCleanSend( ts_cmd_t *p_cmd )
1280 if( p_cmd->send.p_block )
1281 block_Release( p_cmd->send.p_block );
1284 static int CmdInitDel( ts_cmd_t *p_cmd, es_out_id_t *p_es )
1286 p_cmd->i_type = C_DEL;
1287 p_cmd->i_date = mdate();
1288 p_cmd->del.p_es = p_es;
1291 static void CmdExecuteDel( es_out_t *p_out, ts_cmd_t *p_cmd )
1293 if( p_cmd->del.p_es->p_es )
1294 es_out_Del( p_out, p_cmd->del.p_es->p_es );
1295 free( p_cmd->del.p_es );
1298 static int CmdInitControl( ts_cmd_t *p_cmd, int i_query, va_list args, bool b_copy )
1300 p_cmd->i_type = C_CONTROL;
1301 p_cmd->i_date = mdate();
1302 p_cmd->control.i_query = i_query;
1306 /* Pass-through control */
1307 case ES_OUT_SET_ACTIVE: /* arg1= bool */
1308 p_cmd->control.b_bool = (bool)va_arg( args, int );
1311 case ES_OUT_SET_MODE: /* arg1= int */
1312 case ES_OUT_SET_GROUP: /* arg1= int */
1313 case ES_OUT_DEL_GROUP: /* arg1=int i_group */
1314 p_cmd->control.i_int = (int)va_arg( args, int );
1317 case ES_OUT_SET_PCR: /* arg1=int64_t i_pcr(microsecond!) (using default group 0)*/
1318 case ES_OUT_SET_NEXT_DISPLAY_TIME: /* arg1=int64_t i_pts(microsecond) */
1319 p_cmd->control.i_i64 = (int64_t)va_arg( args, int64_t );
1322 case ES_OUT_SET_GROUP_PCR: /* arg1= int i_group, arg2=int64_t i_pcr(microsecond!)*/
1323 p_cmd->control.int_i64.i_int = (int)va_arg( args, int );
1324 p_cmd->control.int_i64.i_i64 = (int64_t)va_arg( args, int64_t );
1327 case ES_OUT_SET_ES_SCRAMBLED_STATE:
1328 p_cmd->control.es_bool.p_es = (es_out_id_t*)va_arg( args, es_out_id_t * );
1329 p_cmd->control.es_bool.b_bool = (bool)va_arg( args, int );
1332 case ES_OUT_RESET_PCR: /* no arg */
1335 case ES_OUT_SET_META: /* arg1=const vlc_meta_t* */
1336 case ES_OUT_SET_GROUP_META: /* arg1=int i_group arg2=const vlc_meta_t* */
1338 if( i_query == ES_OUT_SET_GROUP_META )
1339 p_cmd->control.int_meta.i_int = (int)va_arg( args, int );
1340 const vlc_meta_t *p_meta = va_arg( args, const vlc_meta_t * );
1344 p_cmd->control.int_meta.p_meta = vlc_meta_New();
1345 if( !p_cmd->control.int_meta.p_meta )
1346 return VLC_EGENERIC;
1347 vlc_meta_Merge( p_cmd->control.int_meta.p_meta, p_meta );
1351 /* The cast is only needed to avoid warning */
1352 p_cmd->control.int_meta.p_meta = (vlc_meta_t*)p_meta;
1357 case ES_OUT_SET_GROUP_EPG: /* arg1=int i_group arg2=const vlc_epg_t* */
1359 p_cmd->control.int_epg.i_int = (int)va_arg( args, int );
1360 const vlc_epg_t *p_epg = va_arg( args, const vlc_epg_t * );
1364 p_cmd->control.int_epg.p_epg = vlc_epg_New( p_epg->psz_name );
1365 if( !p_cmd->control.int_epg.p_epg )
1366 return VLC_EGENERIC;
1367 for( int i = 0; i < p_epg->i_event; i++ )
1369 vlc_epg_event_t *p_evt = p_epg->pp_event[i];
1371 vlc_epg_AddEvent( p_cmd->control.int_epg.p_epg,
1372 p_evt->i_start, p_evt->i_duration,
1374 p_evt->psz_short_description, p_evt->psz_description );
1376 vlc_epg_SetCurrent( p_cmd->control.int_epg.p_epg,
1377 p_epg->p_current ? p_epg->p_current->i_start : -1 );
1381 /* The cast is only needed to avoid warning */
1382 p_cmd->control.int_epg.p_epg = (vlc_epg_t*)p_epg;
1387 /* Modified control */
1388 case ES_OUT_SET_ES: /* arg1= es_out_id_t* */
1389 case ES_OUT_RESTART_ES: /* arg1= es_out_id_t* */
1390 case ES_OUT_SET_ES_DEFAULT: /* arg1= es_out_id_t* */
1391 p_cmd->control.p_es = (es_out_id_t*)va_arg( args, es_out_id_t * );
1394 case ES_OUT_SET_ES_STATE:/* arg1= es_out_id_t* arg2=bool */
1395 p_cmd->control.es_bool.p_es = (es_out_id_t*)va_arg( args, es_out_id_t * );
1396 p_cmd->control.es_bool.b_bool = (bool)va_arg( args, int );
1399 case ES_OUT_SET_ES_FMT: /* arg1= es_out_id_t* arg2=es_format_t* */
1401 p_cmd->control.es_fmt.p_es = (es_out_id_t*)va_arg( args, es_out_id_t * );
1402 es_format_t *p_fmt = (es_format_t*)va_arg( args, es_format_t * );
1406 p_cmd->control.es_fmt.p_fmt = malloc( sizeof(*p_fmt) );
1407 if( !p_cmd->control.es_fmt.p_fmt )
1408 return VLC_EGENERIC;
1409 es_format_Copy( p_cmd->control.es_fmt.p_fmt, p_fmt );
1413 p_cmd->control.es_fmt.p_fmt = p_fmt;
1417 case ES_OUT_SET_TIMES:
1419 double f_position = (double)va_arg( args, double );
1420 mtime_t i_time = (mtime_t)va_arg( args, mtime_t );
1421 mtime_t i_length = (mtime_t)va_arg( args, mtime_t );
1423 p_cmd->control.times.f_position = f_position;
1424 p_cmd->control.times.i_time = i_time;
1425 p_cmd->control.times.i_length = i_length;
1428 case ES_OUT_SET_JITTER:
1430 mtime_t i_pts_delay = (mtime_t)va_arg( args, mtime_t );
1431 int i_cr_average = (int)va_arg( args, int );
1433 p_cmd->control.jitter.i_pts_delay = i_pts_delay;
1434 p_cmd->control.jitter.i_cr_average = i_cr_average;
1440 return VLC_EGENERIC;
1445 static int CmdExecuteControl( es_out_t *p_out, ts_cmd_t *p_cmd )
1447 const int i_query = p_cmd->control.i_query;
1451 /* Pass-through control */
1452 case ES_OUT_SET_ACTIVE: /* arg1= bool */
1453 return es_out_Control( p_out, i_query, p_cmd->control.b_bool );
1455 case ES_OUT_SET_MODE: /* arg1= int */
1456 case ES_OUT_SET_GROUP: /* arg1= int */
1457 case ES_OUT_DEL_GROUP: /* arg1=int i_group */
1458 return es_out_Control( p_out, i_query, p_cmd->control.i_int );
1460 case ES_OUT_SET_PCR: /* arg1=int64_t i_pcr(microsecond!) (using default group 0)*/
1461 case ES_OUT_SET_NEXT_DISPLAY_TIME: /* arg1=int64_t i_pts(microsecond) */
1462 return es_out_Control( p_out, i_query, p_cmd->control.i_i64 );
1464 case ES_OUT_SET_GROUP_PCR: /* arg1= int i_group, arg2=int64_t i_pcr(microsecond!)*/
1465 return es_out_Control( p_out, i_query, p_cmd->control.int_i64.i_int,
1466 p_cmd->control.int_i64.i_i64 );
1468 case ES_OUT_RESET_PCR: /* no arg */
1469 return es_out_Control( p_out, i_query );
1471 case ES_OUT_SET_GROUP_META: /* arg1=int i_group arg2=const vlc_meta_t* */
1472 return es_out_Control( p_out, i_query, p_cmd->control.int_meta.i_int,
1473 p_cmd->control.int_meta.p_meta );
1475 case ES_OUT_SET_GROUP_EPG: /* arg1=int i_group arg2=const vlc_epg_t* */
1476 return es_out_Control( p_out, i_query, p_cmd->control.int_epg.i_int,
1477 p_cmd->control.int_epg.p_epg );
1479 case ES_OUT_SET_ES_SCRAMBLED_STATE: /* arg1=int es_out_id_t* arg2=bool */
1480 return es_out_Control( p_out, i_query, p_cmd->control.es_bool.p_es->p_es,
1481 p_cmd->control.es_bool.b_bool );
1483 case ES_OUT_SET_META: /* arg1=const vlc_meta_t* */
1484 return es_out_Control( p_out, i_query, p_cmd->control.int_meta.p_meta );
1486 /* Modified control */
1487 case ES_OUT_SET_ES: /* arg1= es_out_id_t* */
1488 case ES_OUT_RESTART_ES: /* arg1= es_out_id_t* */
1489 case ES_OUT_SET_ES_DEFAULT: /* arg1= es_out_id_t* */
1490 return es_out_Control( p_out, i_query, p_cmd->control.p_es->p_es );
1492 case ES_OUT_SET_ES_STATE:/* arg1= es_out_id_t* arg2=bool */
1493 return es_out_Control( p_out, i_query, p_cmd->control.es_bool.p_es->p_es,
1494 p_cmd->control.es_bool.b_bool );
1496 case ES_OUT_SET_ES_FMT: /* arg1= es_out_id_t* arg2=es_format_t* */
1497 return es_out_Control( p_out, i_query, p_cmd->control.es_fmt.p_es->p_es,
1498 p_cmd->control.es_fmt.p_fmt );
1500 case ES_OUT_SET_TIMES:
1501 return es_out_Control( p_out, i_query, p_cmd->control.times.f_position,
1502 p_cmd->control.times.i_time,
1503 p_cmd->control.times.i_length );
1504 case ES_OUT_SET_JITTER:
1505 return es_out_Control( p_out, i_query, p_cmd->control.jitter.i_pts_delay,
1506 p_cmd->control.jitter.i_cr_average );
1510 return VLC_EGENERIC;
1513 static void CmdCleanControl( ts_cmd_t *p_cmd )
1515 if( ( p_cmd->control.i_query == ES_OUT_SET_GROUP_META ||
1516 p_cmd->control.i_query == ES_OUT_SET_META ) &&
1517 p_cmd->control.int_meta.p_meta )
1519 vlc_meta_Delete( p_cmd->control.int_meta.p_meta );
1521 else if( p_cmd->control.i_query == ES_OUT_SET_GROUP_EPG &&
1522 p_cmd->control.int_epg.p_epg )
1524 vlc_epg_Delete( p_cmd->control.int_epg.p_epg );
1526 else if( p_cmd->control.i_query == ES_OUT_SET_ES_FMT &&
1527 p_cmd->control.es_fmt.p_fmt )
1529 es_format_Clean( p_cmd->control.es_fmt.p_fmt );
1530 free( p_cmd->control.es_fmt.p_fmt );
1535 /*****************************************************************************
1537 *****************************************************************************/
1538 static char *GetTmpPath( char *psz_path )
1540 if( psz_path && *psz_path )
1542 /* Make sure that the path exists and is a directory */
1544 const int i_ret = utf8_stat( psz_path, &s );
1546 if( i_ret < 0 && !utf8_mkdir( psz_path, 0600 ) )
1548 else if( i_ret == 0 && ( s.st_mode & S_IFDIR ) )
1553 /* Create a suitable path */
1554 #if defined (WIN32) && !defined (UNDER_CE)
1555 const DWORD dwCount = GetTempPathW( 0, NULL );
1556 wchar_t *psw_path = calloc( dwCount + 1, sizeof(wchar_t) );
1559 if( GetTempPathW( dwCount + 1, psw_path ) <= 0 )
1563 psw_path = _wgetcwd( NULL, 0 );
1570 psz_path = FromWide( psw_path );
1571 while( psz_path && *psz_path && psz_path[strlen( psz_path ) - 1] == '\\' )
1572 psz_path[strlen( psz_path ) - 1] = '\0';
1577 if( !psz_path || *psz_path == '\0' )
1580 return strdup( "C:" );
1583 psz_path = strdup( "/tmp" );
1589 static FILE *GetTmpFile( char **ppsz_file, const char *psz_path )
1597 if( asprintf( &psz_name, "%s/vlc-timeshift.XXXXXX", psz_path ) < 0 )
1601 fd = utf8_mkstemp( psz_name );
1602 *ppsz_file = psz_name;
1608 f = fdopen( fd, "w+b" );