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