Added es_out_ControlGetPcrSystem helper.
[vlc.git] / src / input / es_out_timeshift.c
1 /*****************************************************************************
2  * es_out_timeshift.c: Es Out timeshift.
3  *****************************************************************************
4  * Copyright (C) 2008 Laurent Aimar
5  * $Id$
6  *
7  * Authors: Laurent Aimar < fenrir _AT_ videolan _DOT_ org>
8  *
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.
13  *
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.
18  *
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  *****************************************************************************/
23
24 /*****************************************************************************
25  * Preamble
26  *****************************************************************************/
27 #ifdef HAVE_CONFIG_H
28 # include "config.h"
29 #endif
30
31 #include <stdlib.h>
32 #include <stdio.h>
33 #include <assert.h>
34 #if defined (WIN32) && !defined (UNDER_CE)
35 #  include <direct.h>
36 #endif
37 #ifdef HAVE_SYS_STAT_H
38 #   include <sys/stat.h>
39 #endif
40
41 #include <vlc_common.h>
42 #include <vlc_charset.h>
43
44 #include <vlc_input.h>
45 #include <vlc_es_out.h>
46 #include <vlc_block.h>
47 #include "input_internal.h"
48 #include "es_out.h"
49 #include "es_out_timeshift.h"
50
51 /*****************************************************************************
52  * Local prototypes
53  *****************************************************************************/
54
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__))
58 #else
59 #   define attribute_packed
60 #endif
61
62 enum
63 {
64     C_ADD,
65     C_SEND,
66     C_DEL,
67     C_CONTROL,
68 };
69
70 typedef struct attribute_packed
71 {
72     es_out_id_t *p_es;
73     es_format_t *p_fmt;
74 } ts_cmd_add_t;
75
76 typedef struct attribute_packed
77 {
78     es_out_id_t *p_es;
79 } ts_cmd_del_t;
80
81 typedef struct attribute_packed
82 {
83     es_out_id_t *p_es;
84     block_t *p_block;
85     int     i_offset;  /* We do not use file > INT_MAX */
86 } ts_cmd_send_t;
87
88 typedef struct attribute_packed
89 {
90     int  i_query;
91
92     union
93     {
94         bool b_bool;
95         int  i_int;
96         int64_t i_i64;
97         es_out_id_t *p_es;
98         struct
99         {
100             int     i_int;
101             int64_t i_i64;
102         } int_i64;
103         struct
104         {
105             int        i_int;
106             vlc_meta_t *p_meta;
107         } int_meta;
108         struct
109         {
110             int       i_int;
111             vlc_epg_t *p_epg;
112         } int_epg;
113         struct
114         {
115             es_out_id_t *p_es;
116             bool        b_bool;
117         } es_bool;
118         struct
119         {
120             es_out_id_t *p_es;
121             es_format_t *p_fmt;
122         } es_fmt;
123         struct
124         {
125             /* FIXME Really too big (double make the whole thing too big) */
126             double  f_position;
127             mtime_t i_time;
128             mtime_t i_length;
129         } times;
130         struct
131         {
132             mtime_t i_pts_delay;
133             int     i_cr_average;
134         } jitter;
135     };
136 } ts_cmd_control_t;
137
138 typedef struct attribute_packed
139 {
140     int8_t  i_type;
141     mtime_t i_date;
142     union
143     {
144         ts_cmd_add_t     add;
145         ts_cmd_del_t     del;
146         ts_cmd_send_t    send;
147         ts_cmd_control_t control;
148     };
149 } ts_cmd_t;
150
151 typedef struct ts_storage_t ts_storage_t;
152 struct ts_storage_t
153 {
154     ts_storage_t *p_next;
155
156     /* */
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 */
162
163     /* */
164     int      i_cmd_r;
165     int      i_cmd_w;
166     int      i_cmd_max;
167     ts_cmd_t *p_cmd;
168 };
169
170 typedef struct
171 {
172     VLC_COMMON_MEMBERS
173
174     /* */
175     input_thread_t *p_input;
176     es_out_t       *p_out;
177     int64_t        i_tmp_size_max;
178     const char     *psz_tmp_path;
179
180     /* Lock for all following fields */
181     vlc_mutex_t    lock;
182     vlc_cond_t     wait;
183
184     /* */
185     bool           b_paused;
186     mtime_t        i_pause_date;
187
188     /* */
189     int            i_rate;
190     int            i_rate_source;
191     mtime_t        i_rate_date;
192     mtime_t        i_rate_delay;
193
194     /* */
195     mtime_t        i_buffering_delay;
196
197     /* */
198     ts_storage_t   *p_storage_r;
199     ts_storage_t   *p_storage_w;
200
201     mtime_t        i_cmd_delay;
202
203 } ts_thread_t;
204
205 struct es_out_id_t
206 {
207     es_out_id_t *p_es;
208 };
209
210 struct es_out_sys_t
211 {
212     input_thread_t *p_input;
213         es_out_t       *p_out;
214
215     /* Configuration */
216     int64_t        i_tmp_size_max;    /* Maximal temporary file size in byte */
217     char           *psz_tmp_path;     /* Path for temporary files */
218
219     /* Lock for all following fields */
220     vlc_mutex_t    lock;
221
222     /* */
223     bool           b_delayed;
224     ts_thread_t   *p_thread;
225
226     /* */
227     bool           b_input_paused;
228     bool           b_input_paused_source;
229     int            i_input_rate;
230     int            i_input_rate_source;
231
232     /* */
233     int            i_es;
234     es_out_id_t    **pp_es;
235 };
236
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 * );
242
243 static int          TsStart( es_out_t * );
244 static void         TsAutoStop( es_out_t * );
245
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 );
253
254 static void         *TsRun( vlc_object_t * );
255
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 );
263
264 static void CmdClean( ts_cmd_t * );
265 static void cmd_cleanup_routine( void *p ) { CmdClean( p ); }
266
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 );
271
272 /* */
273 static void CmdCleanAdd    ( ts_cmd_t * );
274 static void CmdCleanSend   ( ts_cmd_t * );
275 static void CmdCleanControl( ts_cmd_t *p_cmd );
276
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 * );
282
283 /* File helpers */
284 static char *GetTmpPath( char *psz_path );
285 static FILE *GetTmpFile( char **ppsz_file, const char *psz_path );
286
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 )
291 {
292     es_out_t *p_out = malloc( sizeof(*p_out) );
293     if( !p_out )
294         return NULL;
295
296     es_out_sys_t *p_sys = malloc( sizeof(*p_sys) );
297     if( !p_sys )
298     {
299         free( p_out );
300         return NULL;
301     }
302
303     /* */
304     p_out->pf_add     = Add;
305     p_out->pf_send    = Send;
306     p_out->pf_del     = Del;
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;
311
312     /* */
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;
318
319     p_sys->p_out = p_next_out;
320     vlc_mutex_init_recursive( &p_sys->lock );
321
322     p_sys->b_delayed = false;
323     p_sys->p_thread = NULL;
324
325     TAB_INIT( p_sys->i_es, p_sys->pp_es );
326
327     /* */
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;
331     else
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) );
335
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 );
339
340 #if 0
341 #define S(t) msg_Err( p_input, "SIZEOF("#t")=%d", sizeof(t) )
342     S(ts_cmd_t);
343     S(ts_cmd_control_t);
344     S(ts_cmd_send_t);
345     S(ts_cmd_del_t);
346     S(ts_cmd_add_t);
347 #undef S
348 #endif
349
350     return p_out;
351 }
352
353 /*****************************************************************************
354  * Internal functions
355  *****************************************************************************/
356 static void Destroy( es_out_t *p_out )
357 {
358     es_out_sys_t *p_sys = p_out->p_sys;
359
360     if( p_sys->b_delayed )
361     {
362         TsStop( p_sys->p_thread );
363         p_sys->b_delayed = false;
364     }
365
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  );
369
370     free( p_sys->psz_tmp_path );
371     vlc_mutex_destroy( &p_sys->lock );
372     free( p_sys );
373     free( p_out );
374 }
375
376 static es_out_id_t *Add( es_out_t *p_out, const es_format_t *p_fmt )
377 {
378     es_out_sys_t *p_sys = p_out->p_sys;
379     ts_cmd_t cmd;
380
381     es_out_id_t *p_es = malloc( sizeof( *p_es ) );
382     if( !p_es )
383         return NULL;
384
385     vlc_mutex_lock( &p_sys->lock );
386
387     TsAutoStop( p_out );
388
389     if( CmdInitAdd( &cmd, p_es, p_fmt, p_sys->b_delayed ) )
390     {
391         vlc_mutex_unlock( &p_sys->lock );
392         free( p_es );
393         return NULL;
394     }
395
396     TAB_APPEND( p_sys->i_es, p_sys->pp_es, p_es );
397
398     if( p_sys->b_delayed )
399         TsPushCmd( p_sys->p_thread, &cmd );
400     else
401         CmdExecuteAdd( p_sys->p_out, &cmd );
402
403     vlc_mutex_unlock( &p_sys->lock );
404
405     return p_es;
406 }
407 static int Send( es_out_t *p_out, es_out_id_t *p_es, block_t *p_block )
408 {
409     es_out_sys_t *p_sys = p_out->p_sys;
410     ts_cmd_t cmd;
411     int i_ret = VLC_SUCCESS;
412
413     vlc_mutex_lock( &p_sys->lock );
414
415     TsAutoStop( p_out );
416
417     CmdInitSend( &cmd, p_es, p_block );
418     if( p_sys->b_delayed )
419         TsPushCmd( p_sys->p_thread, &cmd );
420     else
421         i_ret = CmdExecuteSend( p_sys->p_out, &cmd) ;
422
423     vlc_mutex_unlock( &p_sys->lock );
424
425     return i_ret;
426 }
427 static void Del( es_out_t *p_out, es_out_id_t *p_es )
428 {
429     es_out_sys_t *p_sys = p_out->p_sys;
430     ts_cmd_t cmd;
431
432     vlc_mutex_lock( &p_sys->lock );
433
434     TsAutoStop( p_out );
435
436     CmdInitDel( &cmd, p_es );
437     if( p_sys->b_delayed )
438         TsPushCmd( p_sys->p_thread, &cmd );
439     else
440         CmdExecuteDel( p_sys->p_out, &cmd );
441
442     TAB_REMOVE( p_sys->i_es, p_sys->pp_es, p_es );
443
444     vlc_mutex_unlock( &p_sys->lock );
445 }
446
447 static int ControlLockedGetEmpty( es_out_t *p_out, bool *pb_empty )
448 {
449     es_out_sys_t *p_sys = p_out->p_sys;
450
451     if( p_sys->b_delayed && TsHasCmd( p_sys->p_thread ) )
452         *pb_empty = false;
453     else
454         *pb_empty = es_out_GetEmpty( p_sys->p_out );
455
456     return VLC_SUCCESS;
457 }
458 static int ControlLockedGetWakeup( es_out_t *p_out, mtime_t *pi_wakeup )
459 {
460     es_out_sys_t *p_sys = p_out->p_sys;
461
462     if( p_sys->b_delayed )
463     {
464         assert( !p_sys->p_input->p->b_can_pace_control );
465         *pi_wakeup = 0;
466     }
467     else
468     {
469         *pi_wakeup = es_out_GetWakeup( p_sys->p_out );
470     }
471
472     return VLC_SUCCESS;
473 }
474 static int ControlLockedGetBuffering( es_out_t *p_out, bool *pb_buffering )
475 {
476     es_out_sys_t *p_sys = p_out->p_sys;
477
478     if( p_sys->b_delayed )
479         *pb_buffering = true;
480     else
481         *pb_buffering = es_out_GetBuffering( p_sys->p_out );
482
483     return VLC_SUCCESS;
484 }
485 static int ControlLockedSetPauseState( es_out_t *p_out, bool b_source_paused, bool b_paused, mtime_t i_date )
486 {
487     es_out_sys_t *p_sys = p_out->p_sys;
488     int i_ret;
489
490     if( !p_sys->b_delayed && !b_source_paused == !b_paused )
491     {
492         i_ret = es_out_SetPauseState( p_sys->p_out, b_source_paused, b_paused, i_date );
493     }
494     else
495     {
496         i_ret = VLC_EGENERIC;
497         if( !p_sys->p_input->p->b_can_pace_control )
498         {
499             if( !p_sys->b_delayed )
500                 TsStart( p_out );
501             if( p_sys->b_delayed )
502                 i_ret = TsChangePause( p_sys->p_thread, b_source_paused, b_paused, i_date );
503         }
504         else
505         {
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
508              * module */
509             msg_Err( p_sys->p_input, "EsOutTimeshift does not work with streams that have space control" );
510         }
511     }
512
513     if( !i_ret )
514     {
515         p_sys->b_input_paused_source = b_source_paused;
516         p_sys->b_input_paused = b_paused;
517     }
518     return i_ret;
519 }
520 static int ControlLockedSetRate( es_out_t *p_out, int i_src_rate, int i_rate )
521 {
522     es_out_sys_t *p_sys = p_out->p_sys;
523     int i_ret;
524
525     if( !p_sys->b_delayed && i_src_rate == i_rate )
526     {
527         i_ret = es_out_SetRate( p_sys->p_out, i_src_rate, i_rate );
528     }
529     else
530     {
531         i_ret = VLC_EGENERIC;
532         if( !p_sys->p_input->p->b_can_pace_control )
533         {
534             if( !p_sys->b_delayed )
535                 TsStart( p_out );
536             if( p_sys->b_delayed )
537                 i_ret = TsChangeRate( p_sys->p_thread, i_src_rate, i_rate );
538         }
539         else
540         {
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
543              * module */
544             msg_Err( p_sys->p_input, "EsOutTimeshift does not work with streams that have space control" );
545         }
546
547     }
548
549     if( !i_ret )
550     {
551         p_sys->i_input_rate_source = i_src_rate;
552         p_sys->i_input_rate = i_rate;
553     }
554     return i_ret;
555 }
556 static int ControlLockedSetTime( es_out_t *p_out, mtime_t i_date )
557 {
558     es_out_sys_t *p_sys = p_out->p_sys;
559
560     if( !p_sys->b_delayed )
561         return es_out_SetTime( p_sys->p_out, i_date );
562
563     /* TODO */
564     msg_Err( p_sys->p_input, "EsOutTimeshift does not yet support time change" );
565     return VLC_EGENERIC;
566 }
567 static int ControlLockedSetFrameNext( es_out_t *p_out )
568 {
569     es_out_sys_t *p_sys = p_out->p_sys;
570
571     return es_out_SetFrameNext( p_sys->p_out );
572 }
573
574 static int ControlLocked( es_out_t *p_out, int i_query, va_list args )
575 {
576     es_out_sys_t *p_sys = p_out->p_sys;
577
578     switch( i_query )
579     {
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:
586         assert(0);
587         return VLC_EGENERIC;
588
589     /* Pass-through control */
590     case ES_OUT_SET_ACTIVE:
591     case ES_OUT_SET_MODE:
592     case ES_OUT_SET_GROUP:
593     case ES_OUT_SET_PCR:
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:
602     case ES_OUT_SET_ES:
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:
609     {
610         ts_cmd_t cmd;
611         if( CmdInitControl( &cmd, i_query, args, p_sys->b_delayed ) )
612             return VLC_EGENERIC;
613         if( p_sys->b_delayed )
614         {
615             TsPushCmd( p_sys->p_thread, &cmd );
616             return VLC_SUCCESS;
617         }
618         return CmdExecuteControl( p_sys->p_out, &cmd );
619     }
620
621     /* Special control when delayed */
622     case ES_OUT_GET_ES_STATE:
623     {
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* );
626
627         if( p_sys->b_delayed )
628         {
629             *pb_enabled = true;
630             return VLC_SUCCESS;
631         }
632         return es_out_Control( p_sys->p_out, ES_OUT_GET_ES_STATE, p_es->p_es, pb_enabled );
633     }
634
635     /* Special internal input control */
636     case ES_OUT_GET_EMPTY:
637     {
638         bool *pb_empty = (bool*)va_arg( args, bool* );
639         return ControlLockedGetEmpty( p_out, pb_empty );
640     }
641     case ES_OUT_GET_WAKE_UP: /* TODO ? */
642     {
643         mtime_t *pi_wakeup = (mtime_t*)va_arg( args, mtime_t* );
644         return ControlLockedGetWakeup( p_out, pi_wakeup );
645     }
646     case ES_OUT_GET_BUFFERING:
647     {
648         bool *pb_buffering = (bool *)va_arg( args, bool* );
649         return ControlLockedGetBuffering( p_out, pb_buffering );
650     }
651     case ES_OUT_SET_PAUSE_STATE:
652     {
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 );
656
657         return ControlLockedSetPauseState( p_out, b_source_paused, b_paused, i_date );
658     }
659     case ES_OUT_SET_RATE:
660     {
661         const int i_src_rate = (int)va_arg( args, int );
662         const int i_rate = (int)va_arg( args, int );
663
664         return ControlLockedSetRate( p_out, i_src_rate, i_rate );
665     }
666     case ES_OUT_SET_TIME:
667     {
668         const mtime_t i_date = (mtime_t)va_arg( args, mtime_t );
669
670         return ControlLockedSetTime( p_out, i_date );
671     }
672     case ES_OUT_SET_FRAME_NEXT:
673     {
674         return ControlLockedSetFrameNext( p_out );
675     }
676     case ES_OUT_GET_PCR_SYSTEM:
677     {
678         if( p_sys->b_delayed )
679             return VLC_EGENERIC;
680
681         mtime_t *pi_system = (mtime_t*)va_arg( args, mtime_t * );
682         return es_out_ControlGetPcrSystem( p_sys->p_out, pi_system );
683     }
684     default:
685         msg_Err( p_sys->p_input, "Unknown es_out_Control query !" );
686         assert(0);
687         return VLC_EGENERIC;
688     }
689 }
690 static int Control( es_out_t *p_out, int i_query, va_list args )
691 {
692     es_out_sys_t *p_sys = p_out->p_sys;
693     int i_ret;
694
695     vlc_mutex_lock( &p_sys->lock );
696
697     TsAutoStop( p_out );
698
699     i_ret = ControlLocked( p_out, i_query, args );
700
701     vlc_mutex_unlock( &p_sys->lock );
702
703     return i_ret;
704 }
705
706 /*****************************************************************************
707  *
708  *****************************************************************************/
709 static void TsDestructor( vlc_object_t *p_this )
710 {
711     ts_thread_t *p_ts = (ts_thread_t*)p_this;
712
713     vlc_cond_destroy( &p_ts->wait );
714     vlc_mutex_destroy( &p_ts->lock );
715 }
716 static int TsStart( es_out_t *p_out )
717 {
718     es_out_sys_t *p_sys = p_out->p_sys;
719     ts_thread_t *p_ts;
720
721     assert( !p_sys->b_delayed );
722
723     p_sys->p_thread = p_ts = vlc_custom_create( p_sys->p_input, sizeof(ts_thread_t),
724                                                 VLC_OBJECT_GENERIC, "es out timeshift" );
725     if( !p_ts )
726         return VLC_EGENERIC;
727
728     p_ts->i_tmp_size_max = p_sys->i_tmp_size_max;
729     p_ts->psz_tmp_path = p_sys->psz_tmp_path;
730     p_ts->p_input = p_sys->p_input;
731     p_ts->p_out = p_sys->p_out;
732     vlc_mutex_init( &p_ts->lock );
733     vlc_cond_init( &p_ts->wait );
734     p_ts->b_paused = p_sys->b_input_paused && !p_sys->b_input_paused_source;
735     p_ts->i_pause_date = p_ts->b_paused ? mdate() : -1;
736     p_ts->i_rate_source = p_sys->i_input_rate_source;
737     p_ts->i_rate        = p_sys->i_input_rate;
738     p_ts->i_rate_date = -1;
739     p_ts->i_rate_delay = 0;
740     p_ts->i_buffering_delay = 0;
741     p_ts->i_cmd_delay = 0;
742     p_ts->p_storage_r = NULL;
743     p_ts->p_storage_w = NULL;
744
745     vlc_object_set_destructor( p_ts, TsDestructor );
746
747     p_sys->b_delayed = true;
748     if( vlc_thread_create( p_ts, "es out timeshift",
749                            TsRun, VLC_THREAD_PRIORITY_INPUT ) )
750     {
751         msg_Err( p_sys->p_input, "cannot create input thread" );
752
753         vlc_object_release( p_ts );
754
755         p_sys->b_delayed = false;
756         return VLC_EGENERIC;
757     }
758
759     return VLC_SUCCESS;
760 }
761 static void TsAutoStop( es_out_t *p_out )
762 {
763     es_out_sys_t *p_sys = p_out->p_sys;
764
765     if( !p_sys->b_delayed || !TsIsUnused( p_sys->p_thread ) )
766         return;
767
768     msg_Warn( p_sys->p_input, "es out timeshift: auto stop" );
769     TsStop( p_sys->p_thread );
770
771     p_sys->b_delayed = false;
772 }
773 static void TsStop( ts_thread_t *p_ts )
774 {
775     vlc_object_kill( p_ts );
776     vlc_thread_join( p_ts );
777
778     vlc_mutex_lock( &p_ts->lock );
779     for( ;; )
780     {
781         ts_cmd_t cmd;
782
783         if( TsPopCmdLocked( p_ts, &cmd, true ) )
784             break;
785
786         CmdClean( &cmd );
787     }
788     assert( !p_ts->p_storage_r || !p_ts->p_storage_r->p_next );
789     if( p_ts->p_storage_r )
790         TsStorageDelete( p_ts->p_storage_r );
791     vlc_mutex_unlock( &p_ts->lock );
792
793     vlc_object_release( p_ts );
794 }
795 static void TsPushCmd( ts_thread_t *p_ts, ts_cmd_t *p_cmd )
796 {
797     vlc_mutex_lock( &p_ts->lock );
798
799     if( !p_ts->p_storage_w || TsStorageIsFull( p_ts->p_storage_w, p_cmd ) )
800     {
801         ts_storage_t *p_storage = TsStorageNew( p_ts->psz_tmp_path, p_ts->i_tmp_size_max );
802
803         if( !p_storage )
804         {
805             CmdClean( p_cmd );
806             vlc_mutex_unlock( &p_ts->lock );
807             /* TODO warn the user (but only once) */
808             return;
809         }
810
811         if( !p_ts->p_storage_w )
812         {
813             p_ts->p_storage_r = p_ts->p_storage_w = p_storage;
814         }
815         else
816         {
817             TsStoragePack( p_ts->p_storage_w );
818             p_ts->p_storage_w->p_next = p_storage;
819             p_ts->p_storage_w = p_storage;
820         }
821     }
822
823     /* TODO return error and warn the user (but only once) */
824     TsStoragePushCmd( p_ts->p_storage_w, p_cmd, p_ts->p_storage_r == p_ts->p_storage_w );
825
826     vlc_cond_signal( &p_ts->wait );
827
828     vlc_mutex_unlock( &p_ts->lock );
829 }
830 static int TsPopCmdLocked( ts_thread_t *p_ts, ts_cmd_t *p_cmd, bool b_flush )
831 {
832     vlc_assert_locked( &p_ts->lock );
833
834     if( TsStorageIsEmpty( p_ts->p_storage_r ) )
835         return VLC_EGENERIC;
836
837     TsStoragePopCmd( p_ts->p_storage_r, p_cmd, b_flush );
838
839     while( p_ts->p_storage_r && TsStorageIsEmpty( p_ts->p_storage_r ) )
840     {
841         ts_storage_t *p_next = p_ts->p_storage_r->p_next;
842         if( !p_next )
843             break;
844
845         TsStorageDelete( p_ts->p_storage_r );
846         p_ts->p_storage_r = p_next;
847     }
848
849     return VLC_SUCCESS;
850 }
851 static bool TsHasCmd( ts_thread_t *p_ts )
852 {
853     bool b_cmd;
854
855     vlc_mutex_lock( &p_ts->lock );
856     b_cmd =  TsStorageIsEmpty( p_ts->p_storage_r );
857     vlc_mutex_unlock( &p_ts->lock );
858
859     return b_cmd;
860 }
861 static bool TsIsUnused( ts_thread_t *p_ts )
862 {
863     bool b_unused;
864
865     vlc_mutex_lock( &p_ts->lock );
866     b_unused = !p_ts->b_paused &&
867                p_ts->i_rate == p_ts->i_rate_source &&
868                TsStorageIsEmpty( p_ts->p_storage_r );
869     vlc_mutex_unlock( &p_ts->lock );
870
871     return b_unused;
872 }
873 static int TsChangePause( ts_thread_t *p_ts, bool b_source_paused, bool b_paused, mtime_t i_date )
874 {
875     vlc_mutex_lock( &p_ts->lock );
876
877     int i_ret;
878     if( b_paused )
879     {
880         assert( !b_source_paused );
881         i_ret = es_out_SetPauseState( p_ts->p_out, true, true, i_date );
882     }
883     else
884     {
885         i_ret = es_out_SetPauseState( p_ts->p_out, false, false, i_date );
886     }
887
888     if( !i_ret )
889     {
890         if( !b_paused )
891         {
892             assert( p_ts->i_pause_date > 0 );
893
894             p_ts->i_cmd_delay += i_date - p_ts->i_pause_date;
895         }
896
897         p_ts->b_paused = b_paused;
898         p_ts->i_pause_date = i_date;
899
900         vlc_cond_signal( &p_ts->wait );
901     }
902     vlc_mutex_unlock( &p_ts->lock );
903     return i_ret;
904 }
905 static int TsChangeRate( ts_thread_t *p_ts, int i_src_rate, int i_rate )
906 {
907     int i_ret;
908
909     vlc_mutex_lock( &p_ts->lock );
910     p_ts->i_cmd_delay += p_ts->i_rate_delay;
911
912     p_ts->i_rate_date = -1;
913     p_ts->i_rate_delay = 0;
914     p_ts->i_rate = i_rate;
915     p_ts->i_rate_source = i_src_rate;
916
917     i_ret = es_out_SetRate( p_ts->p_out, i_rate, i_rate );
918     vlc_mutex_unlock( &p_ts->lock );
919
920     return i_ret;
921 }
922
923 static void *TsRun( vlc_object_t *p_thread )
924 {
925     ts_thread_t *p_ts = (ts_thread_t*)p_thread;
926     mtime_t i_buffering_date = -1;
927
928     for( ;; )
929     {
930         ts_cmd_t cmd;
931         mtime_t  i_deadline;
932         bool b_buffering;
933
934         /* Pop a command to execute */
935         vlc_mutex_lock( &p_ts->lock );
936         mutex_cleanup_push( &p_ts->lock );
937
938         for( ;; )
939         {
940             const int canc = vlc_savecancel();
941             b_buffering = es_out_GetBuffering( p_ts->p_out );
942
943             if( ( !p_ts->b_paused || b_buffering ) && !TsPopCmdLocked( p_ts, &cmd, false ) )
944             {
945                 vlc_restorecancel( canc );
946                 break;
947             }
948             vlc_restorecancel( canc );
949
950             vlc_cond_wait( &p_ts->wait, &p_ts->lock );
951         }
952
953         if( b_buffering && i_buffering_date < 0 )
954         {
955             i_buffering_date = cmd.i_date;
956         }
957         else if( i_buffering_date > 0 )
958         {
959             p_ts->i_buffering_delay += i_buffering_date - cmd.i_date; /* It is < 0 */
960             if( b_buffering )
961                 i_buffering_date = cmd.i_date;
962             else
963                 i_buffering_date = -1;
964         }
965
966         if( p_ts->i_rate_date < 0 )
967             p_ts->i_rate_date = cmd.i_date;
968
969         p_ts->i_rate_delay = 0;
970         if( p_ts->i_rate_source != p_ts->i_rate )
971         {
972             const mtime_t i_duration = cmd.i_date - p_ts->i_rate_date;
973             p_ts->i_rate_delay = i_duration * p_ts->i_rate / p_ts->i_rate_source - i_duration;
974         }
975         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 )
976         {
977             const int canc = vlc_savecancel();
978
979             /* Auto reset to rate 1.0 */
980             msg_Warn( p_ts->p_input, "es out timeshift: auto reset rate to %d", p_ts->i_rate_source );
981
982             p_ts->i_cmd_delay = 0;
983             p_ts->i_buffering_delay = 0;
984
985             p_ts->i_rate_delay = 0;
986             p_ts->i_rate_date = -1;
987             p_ts->i_rate = p_ts->i_rate_source;
988
989             if( !es_out_SetRate( p_ts->p_out, p_ts->i_rate_source, p_ts->i_rate ) )
990             {
991                 vlc_value_t val = { .i_int = p_ts->i_rate };
992                 /* Warn back input
993                  * FIXME it is perfectly safe BUT it is ugly as it may hide a
994                  * rate change requested by user */
995                 input_ControlPush( p_ts->p_input, INPUT_CONTROL_SET_RATE, &val );
996             }
997
998             vlc_restorecancel( canc );
999         }
1000         i_deadline = cmd.i_date + p_ts->i_cmd_delay + p_ts->i_rate_delay + p_ts->i_buffering_delay;
1001
1002         vlc_cleanup_run();
1003
1004         /* Regulate the speed of command processing to the same one than
1005          * reading  */
1006         vlc_cleanup_push( cmd_cleanup_routine, &cmd );
1007
1008         mwait( i_deadline );
1009
1010         vlc_cleanup_pop();
1011
1012         /* Execute the command  */
1013         const int canc = vlc_savecancel();
1014         switch( cmd.i_type )
1015         {
1016         case C_ADD:
1017             CmdExecuteAdd( p_ts->p_out, &cmd );
1018             CmdCleanAdd( &cmd );
1019             break;
1020         case C_SEND:
1021             CmdExecuteSend( p_ts->p_out, &cmd );
1022             CmdCleanSend( &cmd );
1023             break;
1024         case C_CONTROL:
1025             CmdExecuteControl( p_ts->p_out, &cmd );
1026             CmdCleanControl( &cmd );
1027             break;
1028         case C_DEL:
1029             CmdExecuteDel( p_ts->p_out, &cmd );
1030             break;
1031         default:
1032             assert(0);
1033             break;
1034         }
1035         vlc_restorecancel( canc );
1036     }
1037
1038     return NULL;
1039 }
1040
1041 /*****************************************************************************
1042  *
1043  *****************************************************************************/
1044 static ts_storage_t *TsStorageNew( const char *psz_tmp_path, int64_t i_tmp_size_max )
1045 {
1046     ts_storage_t *p_storage = calloc( 1, sizeof(ts_storage_t) );
1047     if( !p_storage )
1048         return NULL;
1049
1050     /* */
1051     p_storage->p_next = NULL;
1052
1053     /* */
1054     p_storage->i_file_max = i_tmp_size_max;
1055     p_storage->i_file_size = 0;
1056     p_storage->p_filew = GetTmpFile( &p_storage->psz_file, psz_tmp_path );
1057     if( p_storage->psz_file )
1058         p_storage->p_filer = utf8_fopen( p_storage->psz_file, "rb" );
1059
1060     /* */
1061     p_storage->i_cmd_w = 0;
1062     p_storage->i_cmd_r = 0;
1063     p_storage->i_cmd_max = 30000;
1064     p_storage->p_cmd = malloc( p_storage->i_cmd_max * sizeof(*p_storage->p_cmd) );
1065     //fprintf( stderr, "\nSTORAGE name=%s size=%d kbytes\n", p_storage->psz_file, p_storage->i_cmd_max * sizeof(*p_storage->p_cmd) /1024 );
1066
1067     if( !p_storage->p_cmd || !p_storage->p_filew || !p_storage->p_filer )
1068     {
1069         TsStorageDelete( p_storage );
1070         return NULL;
1071     }
1072     return p_storage;
1073 }
1074 static void TsStorageDelete( ts_storage_t *p_storage )
1075 {
1076     while( p_storage->i_cmd_r < p_storage->i_cmd_w )
1077     {
1078         ts_cmd_t cmd;
1079
1080         TsStoragePopCmd( p_storage, &cmd, true );
1081
1082         CmdClean( &cmd );
1083     }
1084     free( p_storage->p_cmd );
1085
1086     if( p_storage->p_filer )
1087         fclose( p_storage->p_filer );
1088     if( p_storage->p_filew )
1089         fclose( p_storage->p_filew );
1090
1091     if( p_storage->psz_file )
1092     {
1093         utf8_unlink( p_storage->psz_file );
1094         free( p_storage->psz_file );
1095     }
1096
1097     free( p_storage );
1098 }
1099 static void TsStoragePack( ts_storage_t *p_storage )
1100 {
1101     /* Try to release a bit of memory */
1102     if( p_storage->i_cmd_w >= p_storage->i_cmd_max )
1103         return;
1104
1105     p_storage->i_cmd_max = __MAX( p_storage->i_cmd_w, 1 );
1106
1107     ts_cmd_t *p_new = realloc( p_storage->p_cmd, p_storage->i_cmd_max * sizeof(*p_storage->p_cmd) );
1108     if( p_new )
1109         p_storage->p_cmd = p_new;
1110 }
1111 static bool TsStorageIsFull( ts_storage_t *p_storage, const ts_cmd_t *p_cmd )
1112 {
1113     if( p_cmd && p_cmd->i_type == C_SEND && p_storage->i_cmd_w > 0 )
1114     {
1115         size_t i_size = sizeof(*p_cmd->send.p_block) + p_cmd->send.p_block->i_buffer;
1116
1117         if( p_storage->i_file_size + i_size >= p_storage->i_file_max )
1118             return true;
1119     }
1120     return p_storage->i_cmd_w >= p_storage->i_cmd_max;
1121 }
1122 static bool TsStorageIsEmpty( ts_storage_t *p_storage )
1123 {
1124     return !p_storage || p_storage->i_cmd_r >= p_storage->i_cmd_w;
1125 }
1126 static void TsStoragePushCmd( ts_storage_t *p_storage, const ts_cmd_t *p_cmd, bool b_flush )
1127 {
1128     ts_cmd_t cmd = *p_cmd;
1129
1130     assert( !TsStorageIsFull( p_storage, p_cmd ) );
1131
1132     if( cmd.i_type == C_SEND )
1133     {
1134         block_t *p_block = cmd.send.p_block;
1135
1136         cmd.send.p_block = NULL;
1137         cmd.send.i_offset = ftell( p_storage->p_filew );
1138
1139         if( fwrite( p_block, sizeof(*p_block), 1, p_storage->p_filew ) != 1 )
1140         {
1141             block_Release( p_block );
1142             return;
1143         }
1144         p_storage->i_file_size += sizeof(*p_block);
1145         if( p_block->i_buffer > 0 )
1146         {
1147             if( fwrite( p_block->p_buffer, p_block->i_buffer, 1, p_storage->p_filew ) != 1 )
1148             {
1149                 block_Release( p_block );
1150                 return;
1151             }
1152         }
1153         p_storage->i_file_size += p_block->i_buffer;
1154         block_Release( p_block );
1155
1156         if( b_flush )
1157             fflush( p_storage->p_filew );
1158     }
1159     p_storage->p_cmd[p_storage->i_cmd_w++] = cmd;
1160 }
1161 static void TsStoragePopCmd( ts_storage_t *p_storage, ts_cmd_t *p_cmd, bool b_flush )
1162 {
1163     assert( !TsStorageIsEmpty( p_storage ) );
1164
1165     *p_cmd = p_storage->p_cmd[p_storage->i_cmd_r++];
1166     if( p_cmd->i_type == C_SEND )
1167     {
1168         block_t block;
1169
1170         if( !b_flush &&
1171             !fseek( p_storage->p_filer, p_cmd->send.i_offset, SEEK_SET ) &&
1172             fread( &block, sizeof(block), 1, p_storage->p_filer ) == 1 )
1173         {
1174             block_t *p_block = block_Alloc( block.i_buffer );
1175             if( p_block )
1176             {
1177                 p_block->i_dts      = block.i_dts;
1178                 p_block->i_pts      = block.i_pts;
1179                 p_block->i_flags    = block.i_flags;
1180                 p_block->i_length   = block.i_length;
1181                 p_block->i_rate     = block.i_rate;
1182                 p_block->i_nb_samples = block.i_nb_samples;
1183                 p_block->i_buffer = fread( p_block->p_buffer, 1, block.i_buffer, p_storage->p_filer );
1184             }
1185             p_cmd->send.p_block = p_block;
1186         }
1187         else
1188         {
1189             //fprintf( stderr, "TsStoragePopCmd: %m\n" );
1190             p_cmd->send.p_block = block_Alloc( 1 );
1191         }
1192     }
1193 }
1194
1195 /*****************************************************************************
1196  *
1197  *****************************************************************************/
1198 static void CmdClean( ts_cmd_t *p_cmd )
1199 {
1200     switch( p_cmd->i_type )
1201     {
1202     case C_ADD:
1203         CmdCleanAdd( p_cmd );
1204         break;
1205     case C_SEND:
1206         CmdCleanSend( p_cmd );
1207         break;
1208     case C_CONTROL:
1209         CmdCleanControl( p_cmd );
1210         break;
1211     case C_DEL:
1212         break;
1213     default:
1214         assert(0);
1215         break;
1216     }
1217 }
1218
1219 static int CmdInitAdd( ts_cmd_t *p_cmd, es_out_id_t *p_es, const es_format_t *p_fmt, bool b_copy )
1220 {
1221     p_cmd->i_type = C_ADD;
1222     p_cmd->i_date = mdate();
1223     p_cmd->add.p_es = p_es;
1224     if( b_copy )
1225     {
1226         p_cmd->add.p_fmt = malloc( sizeof(*p_fmt) );
1227         if( !p_cmd->add.p_fmt )
1228             return VLC_EGENERIC;
1229         es_format_Copy( p_cmd->add.p_fmt, p_fmt );
1230     }
1231     else
1232     {
1233         p_cmd->add.p_fmt = (es_format_t*)p_fmt;
1234     }
1235     return VLC_SUCCESS;
1236 }
1237 static void CmdExecuteAdd( es_out_t *p_out, ts_cmd_t *p_cmd )
1238 {
1239     p_cmd->add.p_es->p_es = es_out_Add( p_out, p_cmd->add.p_fmt );
1240 }
1241 static void CmdCleanAdd( ts_cmd_t *p_cmd )
1242 {
1243     es_format_Clean( p_cmd->add.p_fmt );
1244     free( p_cmd->add.p_fmt );
1245 }
1246
1247 static void CmdInitSend( ts_cmd_t *p_cmd, es_out_id_t *p_es, block_t *p_block )
1248 {
1249     p_cmd->i_type = C_SEND;
1250     p_cmd->i_date = mdate();
1251     p_cmd->send.p_es = p_es;
1252     p_cmd->send.p_block = p_block;
1253 }
1254 static int CmdExecuteSend( es_out_t *p_out, ts_cmd_t *p_cmd )
1255 {
1256     block_t *p_block = p_cmd->send.p_block;
1257
1258     p_cmd->send.p_block = NULL;
1259
1260     if( p_block )
1261     {
1262         if( p_cmd->send.p_es->p_es )
1263             return es_out_Send( p_out, p_cmd->send.p_es->p_es, p_block );
1264         block_Release( p_block );
1265     }
1266     return VLC_EGENERIC;
1267 }
1268 static void CmdCleanSend( ts_cmd_t *p_cmd )
1269 {
1270     if( p_cmd->send.p_block )
1271         block_Release( p_cmd->send.p_block );
1272 }
1273
1274 static int CmdInitDel( ts_cmd_t *p_cmd, es_out_id_t *p_es )
1275 {
1276     p_cmd->i_type = C_DEL;
1277     p_cmd->i_date = mdate();
1278     p_cmd->del.p_es = p_es;
1279     return VLC_SUCCESS;
1280 }
1281 static void CmdExecuteDel( es_out_t *p_out, ts_cmd_t *p_cmd )
1282 {
1283     if( p_cmd->del.p_es->p_es )
1284         es_out_Del( p_out, p_cmd->del.p_es->p_es );
1285     free( p_cmd->del.p_es );
1286 }
1287
1288 static int CmdInitControl( ts_cmd_t *p_cmd, int i_query, va_list args, bool b_copy )
1289 {
1290     p_cmd->i_type = C_CONTROL;
1291     p_cmd->i_date = mdate();
1292     p_cmd->control.i_query = i_query;
1293
1294     switch( i_query )
1295     {
1296     /* Pass-through control */
1297     case ES_OUT_SET_ACTIVE:  /* arg1= bool                     */
1298         p_cmd->control.b_bool = (bool)va_arg( args, int );
1299         break;
1300
1301     case ES_OUT_SET_MODE:    /* arg1= int                            */
1302     case ES_OUT_SET_GROUP:   /* arg1= int                            */
1303     case ES_OUT_DEL_GROUP:   /* arg1=int i_group */
1304         p_cmd->control.i_int = (int)va_arg( args, int );
1305         break;
1306
1307     case ES_OUT_SET_PCR:                /* arg1=int64_t i_pcr(microsecond!) (using default group 0)*/
1308     case ES_OUT_SET_NEXT_DISPLAY_TIME:  /* arg1=int64_t i_pts(microsecond) */
1309         p_cmd->control.i_i64 = (int64_t)va_arg( args, int64_t );
1310         break;
1311
1312     case ES_OUT_SET_GROUP_PCR:          /* arg1= int i_group, arg2=int64_t i_pcr(microsecond!)*/
1313         p_cmd->control.int_i64.i_int = (int)va_arg( args, int );
1314         p_cmd->control.int_i64.i_i64 = (int64_t)va_arg( args, int64_t );
1315         break;
1316
1317     case ES_OUT_SET_ES_SCRAMBLED_STATE:
1318         p_cmd->control.es_bool.p_es = (es_out_id_t*)va_arg( args, es_out_id_t * );
1319         p_cmd->control.es_bool.b_bool = (bool)va_arg( args, int );
1320         break;
1321
1322     case ES_OUT_RESET_PCR:           /* no arg */
1323         break;
1324
1325     case ES_OUT_SET_META:        /* arg1=const vlc_meta_t* */
1326     case ES_OUT_SET_GROUP_META:  /* arg1=int i_group arg2=const vlc_meta_t* */
1327     {
1328         if( i_query == ES_OUT_SET_GROUP_META )
1329             p_cmd->control.int_meta.i_int = (int)va_arg( args, int );
1330         const vlc_meta_t *p_meta = va_arg( args, const vlc_meta_t * );
1331
1332         if( b_copy )
1333         {
1334             p_cmd->control.int_meta.p_meta = vlc_meta_New();
1335             if( !p_cmd->control.int_meta.p_meta )
1336                 return VLC_EGENERIC;
1337             vlc_meta_Merge( p_cmd->control.int_meta.p_meta, p_meta );
1338         }
1339         else
1340         {
1341             /* The cast is only needed to avoid warning */
1342             p_cmd->control.int_meta.p_meta = (vlc_meta_t*)p_meta;
1343         }
1344         break;
1345     }
1346
1347     case ES_OUT_SET_GROUP_EPG:   /* arg1=int i_group arg2=const vlc_epg_t* */
1348     {
1349         p_cmd->control.int_epg.i_int = (int)va_arg( args, int );
1350         const vlc_epg_t *p_epg = va_arg( args, const vlc_epg_t * );
1351
1352         if( b_copy )
1353         {
1354             p_cmd->control.int_epg.p_epg = vlc_epg_New( p_epg->psz_name );
1355             if( !p_cmd->control.int_epg.p_epg )
1356                 return VLC_EGENERIC;
1357             for( int i = 0; i < p_epg->i_event; i++ )
1358             {
1359                 vlc_epg_event_t *p_evt = p_epg->pp_event[i];
1360
1361                 vlc_epg_AddEvent( p_cmd->control.int_epg.p_epg,
1362                                   p_evt->i_start, p_evt->i_duration,
1363                                   p_evt->psz_name,
1364                                   p_evt->psz_short_description, p_evt->psz_description );
1365             }
1366             vlc_epg_SetCurrent( p_cmd->control.int_epg.p_epg,
1367                                 p_epg->p_current ? p_epg->p_current->i_start : -1 );
1368         }
1369         else
1370         {
1371             /* The cast is only needed to avoid warning */
1372             p_cmd->control.int_epg.p_epg = (vlc_epg_t*)p_epg;
1373         }
1374         break;
1375     }
1376
1377     /* Modified control */
1378     case ES_OUT_SET_ES:      /* arg1= es_out_id_t*                   */
1379     case ES_OUT_RESTART_ES:  /* arg1= es_out_id_t*                   */
1380     case ES_OUT_SET_ES_DEFAULT: /* arg1= es_out_id_t*                */
1381         p_cmd->control.p_es = (es_out_id_t*)va_arg( args, es_out_id_t * );
1382         break;
1383
1384     case ES_OUT_SET_ES_STATE:/* arg1= es_out_id_t* arg2=bool   */
1385         p_cmd->control.es_bool.p_es = (es_out_id_t*)va_arg( args, es_out_id_t * );
1386         p_cmd->control.es_bool.b_bool = (bool)va_arg( args, int );
1387         break;
1388
1389     case ES_OUT_SET_ES_FMT:     /* arg1= es_out_id_t* arg2=es_format_t* */
1390     {
1391         p_cmd->control.es_fmt.p_es = (es_out_id_t*)va_arg( args, es_out_id_t * );
1392         es_format_t *p_fmt = (es_format_t*)va_arg( args, es_format_t * );
1393
1394         if( b_copy )
1395         {
1396             p_cmd->control.es_fmt.p_fmt = malloc( sizeof(*p_fmt) );
1397             if( !p_cmd->control.es_fmt.p_fmt )
1398                 return VLC_EGENERIC;
1399             es_format_Copy( p_cmd->control.es_fmt.p_fmt, p_fmt );
1400         }
1401         else
1402         {
1403             p_cmd->control.es_fmt.p_fmt = p_fmt;
1404         }
1405         break;
1406     }
1407     case ES_OUT_SET_TIMES:
1408     {
1409         double f_position = (double)va_arg( args, double );
1410         mtime_t i_time = (mtime_t)va_arg( args, mtime_t );
1411         mtime_t i_length = (mtime_t)va_arg( args, mtime_t );
1412
1413         p_cmd->control.times.f_position = f_position;
1414         p_cmd->control.times.i_time = i_time;
1415         p_cmd->control.times.i_length = i_length;
1416         break;
1417     }
1418     case ES_OUT_SET_JITTER:
1419     {
1420         mtime_t i_pts_delay = (mtime_t)va_arg( args, mtime_t );
1421         int     i_cr_average = (int)va_arg( args, int );
1422
1423         p_cmd->control.jitter.i_pts_delay = i_pts_delay;
1424         p_cmd->control.jitter.i_cr_average = i_cr_average;
1425         break;
1426     }
1427
1428     default:
1429         assert(0);
1430         return VLC_EGENERIC;
1431     }
1432
1433     return VLC_SUCCESS;
1434 }
1435 static int CmdExecuteControl( es_out_t *p_out, ts_cmd_t *p_cmd )
1436 {
1437     const int i_query = p_cmd->control.i_query;
1438
1439     switch( i_query )
1440     {
1441     /* Pass-through control */
1442     case ES_OUT_SET_ACTIVE:  /* arg1= bool                     */
1443         return es_out_Control( p_out, i_query, p_cmd->control.b_bool );
1444
1445     case ES_OUT_SET_MODE:    /* arg1= int                            */
1446     case ES_OUT_SET_GROUP:   /* arg1= int                            */
1447     case ES_OUT_DEL_GROUP:   /* arg1=int i_group */
1448         return es_out_Control( p_out, i_query, p_cmd->control.i_int );
1449
1450     case ES_OUT_SET_PCR:                /* arg1=int64_t i_pcr(microsecond!) (using default group 0)*/
1451     case ES_OUT_SET_NEXT_DISPLAY_TIME:  /* arg1=int64_t i_pts(microsecond) */
1452         return es_out_Control( p_out, i_query, p_cmd->control.i_i64 );
1453
1454     case ES_OUT_SET_GROUP_PCR:          /* arg1= int i_group, arg2=int64_t i_pcr(microsecond!)*/
1455         return es_out_Control( p_out, i_query, p_cmd->control.int_i64.i_int,
1456                                                p_cmd->control.int_i64.i_i64 );
1457
1458     case ES_OUT_RESET_PCR:           /* no arg */
1459         return es_out_Control( p_out, i_query );
1460
1461     case ES_OUT_SET_GROUP_META:  /* arg1=int i_group arg2=const vlc_meta_t* */
1462         return es_out_Control( p_out, i_query, p_cmd->control.int_meta.i_int,
1463                                                p_cmd->control.int_meta.p_meta );
1464
1465     case ES_OUT_SET_GROUP_EPG:   /* arg1=int i_group arg2=const vlc_epg_t* */
1466         return es_out_Control( p_out, i_query, p_cmd->control.int_epg.i_int,
1467                                                p_cmd->control.int_epg.p_epg );
1468
1469     case ES_OUT_SET_ES_SCRAMBLED_STATE: /* arg1=int es_out_id_t* arg2=bool */
1470         return es_out_Control( p_out, i_query, p_cmd->control.es_bool.p_es->p_es,
1471                                                p_cmd->control.es_bool.b_bool );
1472
1473     case ES_OUT_SET_META:  /* arg1=const vlc_meta_t* */
1474         return es_out_Control( p_out, i_query, p_cmd->control.int_meta.p_meta );
1475
1476     /* Modified control */
1477     case ES_OUT_SET_ES:      /* arg1= es_out_id_t*                   */
1478     case ES_OUT_RESTART_ES:  /* arg1= es_out_id_t*                   */
1479     case ES_OUT_SET_ES_DEFAULT: /* arg1= es_out_id_t*                */
1480         return es_out_Control( p_out, i_query, p_cmd->control.p_es->p_es );
1481
1482     case ES_OUT_SET_ES_STATE:/* arg1= es_out_id_t* arg2=bool   */
1483         return es_out_Control( p_out, i_query, p_cmd->control.es_bool.p_es->p_es,
1484                                                p_cmd->control.es_bool.b_bool );
1485
1486     case ES_OUT_SET_ES_FMT:     /* arg1= es_out_id_t* arg2=es_format_t* */
1487         return es_out_Control( p_out, i_query, p_cmd->control.es_fmt.p_es->p_es,
1488                                                p_cmd->control.es_fmt.p_fmt );
1489
1490     case ES_OUT_SET_TIMES:
1491         return es_out_Control( p_out, i_query, p_cmd->control.times.f_position,
1492                                                p_cmd->control.times.i_time,
1493                                                p_cmd->control.times.i_length );
1494     case ES_OUT_SET_JITTER:
1495         return es_out_Control( p_out, i_query, p_cmd->control.jitter.i_pts_delay,
1496                                                p_cmd->control.jitter.i_cr_average );
1497
1498     default:
1499         assert(0);
1500         return VLC_EGENERIC;
1501     }
1502 }
1503 static void CmdCleanControl( ts_cmd_t *p_cmd )
1504 {
1505     if( ( p_cmd->control.i_query == ES_OUT_SET_GROUP_META ||
1506           p_cmd->control.i_query == ES_OUT_SET_META ) &&
1507         p_cmd->control.int_meta.p_meta )
1508     {
1509         vlc_meta_Delete( p_cmd->control.int_meta.p_meta );
1510     }
1511     else if( p_cmd->control.i_query == ES_OUT_SET_GROUP_EPG &&
1512              p_cmd->control.int_epg.p_epg )
1513     {
1514         vlc_epg_Delete( p_cmd->control.int_epg.p_epg );
1515     }
1516     else if( p_cmd->control.i_query == ES_OUT_SET_ES_FMT &&
1517              p_cmd->control.es_fmt.p_fmt )
1518     {
1519         es_format_Clean( p_cmd->control.es_fmt.p_fmt );
1520         free( p_cmd->control.es_fmt.p_fmt );
1521     }
1522 }
1523
1524
1525 /*****************************************************************************
1526  * GetTmpFile/Path:
1527  *****************************************************************************/
1528 static char *GetTmpPath( char *psz_path )
1529 {
1530     if( psz_path && *psz_path )
1531     {
1532         /* Make sure that the path exists and is a directory */
1533         struct stat s;
1534         const int i_ret = utf8_stat( psz_path, &s );
1535
1536         if( i_ret < 0 && !utf8_mkdir( psz_path, 0600 ) )
1537             return psz_path;
1538         else if( i_ret == 0 && ( s.st_mode & S_IFDIR ) )
1539             return psz_path;
1540     }
1541     free( psz_path );
1542
1543     /* Create a suitable path */
1544 #if defined (WIN32) && !defined (UNDER_CE)
1545     const DWORD dwCount = GetTempPathW( 0, NULL );
1546     wchar_t *psw_path = calloc( dwCount + 1, sizeof(wchar_t) );
1547     if( psw_path )
1548     {
1549         if( GetTempPathW( dwCount + 1, psw_path ) <= 0 )
1550         {
1551             free( psw_path );
1552
1553             psw_path = _wgetcwd( NULL, 0 );
1554         }
1555     }
1556
1557     psz_path = NULL;
1558     if( psw_path )
1559     {
1560         psz_path = FromWide( psw_path );
1561         while( psz_path && *psz_path && psz_path[strlen( psz_path ) - 1] == '\\' )
1562             psz_path[strlen( psz_path ) - 1] = '\0';
1563
1564         free( psw_path );
1565     }
1566
1567     if( !psz_path || *psz_path == '\0' )
1568     {
1569         free( psz_path );
1570         return strdup( "C:" );
1571     }
1572 #else
1573     psz_path = strdup( "/tmp" );
1574 #endif
1575
1576     return psz_path;
1577 }
1578
1579 static FILE *GetTmpFile( char **ppsz_file, const char *psz_path )
1580 {
1581     char *psz_name;
1582     int fd;
1583     FILE *f;
1584
1585     /* */
1586     *ppsz_file = NULL;
1587     if( asprintf( &psz_name, "%s/vlc-timeshift.XXXXXX", psz_path ) < 0 )
1588         return NULL;
1589
1590     /* */
1591     fd = utf8_mkstemp( psz_name );
1592     *ppsz_file = psz_name;
1593
1594     if( fd < 0 )
1595         return NULL;
1596
1597     /* */
1598     f = fdopen( fd, "w+b" );
1599     if( !f )
1600         close( fd );
1601
1602     return f;
1603 }
1604