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_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     } u;
 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     } u;
 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_GET_ES_OBJECTS_BY_ID:
 585     case ES_OUT_SET_DELAY:
 586     case ES_OUT_SET_RECORD_STATE:
 587         assert(0);
 588         return VLC_EGENERIC;
 589
 590     /* Pass-through control */
 591     case ES_OUT_SET_ACTIVE:
 592     case ES_OUT_SET_MODE:
 593     case ES_OUT_SET_GROUP:
 594     case ES_OUT_SET_PCR:
 595     case ES_OUT_SET_GROUP_PCR:
 596     case ES_OUT_RESET_PCR:
 597     case ES_OUT_SET_NEXT_DISPLAY_TIME:
 598     case ES_OUT_SET_GROUP_META:
 599     case ES_OUT_SET_GROUP_EPG:
 600     case ES_OUT_SET_ES_SCRAMBLED_STATE:
 601     case ES_OUT_DEL_GROUP:
 602     case ES_OUT_SET_META:
 603     case ES_OUT_SET_ES:
 604     case ES_OUT_RESTART_ES:
 605     case ES_OUT_SET_ES_DEFAULT:
 606     case ES_OUT_SET_ES_STATE:
 607     case ES_OUT_SET_ES_FMT:
 608     case ES_OUT_SET_TIMES:
 609     case ES_OUT_SET_JITTER:
 610     {
 611         ts_cmd_t cmd;
 612         if( CmdInitControl( &cmd, i_query, args, p_sys->b_delayed ) )
 613             return VLC_EGENERIC;
 614         if( p_sys->b_delayed )
 615         {
 616             TsPushCmd( p_sys->p_thread, &cmd );
 617             return VLC_SUCCESS;
 618         }
 619         return CmdExecuteControl( p_sys->p_out, &cmd );
 620     }
 621
 622     /* Special control when delayed */
 623     case ES_OUT_GET_ES_STATE:
 624     {
 625         es_out_id_t *p_es = (es_out_id_t*)va_arg( args, es_out_id_t * );
 626         bool *pb_enabled = (bool*)va_arg( args, bool* );
 627
 628         if( p_sys->b_delayed )
 629         {
 630             *pb_enabled = true;
 631             return VLC_SUCCESS;
 632         }
 633         return es_out_Control( p_sys->p_out, ES_OUT_GET_ES_STATE, p_es->p_es, pb_enabled );
 634     }
 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         return es_out_ControlGetPcrSystem( p_sys->p_out, pi_system );
 684     }
 685     case ES_OUT_MODIFY_PCR_SYSTEM:
 686     {
 687         const bool    b_absolute = va_arg( args, int );
 688         const mtime_t i_system   = va_arg( args, mtime_t );
 689
 690         if( b_absolute && p_sys->b_delayed )
 691             return VLC_EGENERIC;
 692
 693         return es_out_ControlModifyPcrSystem( p_sys->p_out, b_absolute, i_system );
 694     }
 695     default:
 696         msg_Err( p_sys->p_input, "Unknown es_out_Control query !" );
 697         assert(0);
 698         return VLC_EGENERIC;
 699     }
 700 }
 701 static int Control( es_out_t *p_out, int i_query, va_list args )
 702 {
 703     es_out_sys_t *p_sys = p_out->p_sys;
 704     int i_ret;
 705
 706     vlc_mutex_lock( &p_sys->lock );
 707
 708     TsAutoStop( p_out );
 709
 710     i_ret = ControlLocked( p_out, i_query, args );
 711
 712     vlc_mutex_unlock( &p_sys->lock );
 713
 714     return i_ret;
 715 }
 716
 717 /*****************************************************************************
 718  *
 719  *****************************************************************************/
 720 static void TsDestructor( vlc_object_t *p_this )
 721 {
 722     ts_thread_t *p_ts = (ts_thread_t*)p_this;
 723
 724     vlc_cond_destroy( &p_ts->wait );
 725     vlc_mutex_destroy( &p_ts->lock );
 726 }
 727 static int TsStart( es_out_t *p_out )
 728 {
 729     es_out_sys_t *p_sys = p_out->p_sys;
 730     ts_thread_t *p_ts;
 731
 732     assert( !p_sys->b_delayed );
 733
 734     p_sys->p_thread = p_ts = vlc_custom_create( p_sys->p_input, sizeof(ts_thread_t),
 735                                                 VLC_OBJECT_GENERIC, "es out timeshift" );
 736     if( !p_ts )
 737         return VLC_EGENERIC;
 738
 739     p_ts->i_tmp_size_max = p_sys->i_tmp_size_max;
 740     p_ts->psz_tmp_path = p_sys->psz_tmp_path;
 741     p_ts->p_input = p_sys->p_input;
 742     p_ts->p_out = p_sys->p_out;
 743     vlc_mutex_init( &p_ts->lock );
 744     vlc_cond_init( &p_ts->wait );
 745     p_ts->b_paused = p_sys->b_input_paused && !p_sys->b_input_paused_source;
 746     p_ts->i_pause_date = p_ts->b_paused ? mdate() : -1;
 747     p_ts->i_rate_source = p_sys->i_input_rate_source;
 748     p_ts->i_rate        = p_sys->i_input_rate;
 749     p_ts->i_rate_date = -1;
 750     p_ts->i_rate_delay = 0;
 751     p_ts->i_buffering_delay = 0;
 752     p_ts->i_cmd_delay = 0;
 753     p_ts->p_storage_r = NULL;
 754     p_ts->p_storage_w = NULL;
 755
 756     vlc_object_set_destructor( p_ts, TsDestructor );
 757
 758     p_sys->b_delayed = true;
 759     if( vlc_thread_create( p_ts, "es out timeshift",
 760                            TsRun, VLC_THREAD_PRIORITY_INPUT ) )
 761     {
 762         msg_Err( p_sys->p_input, "cannot create input thread" );
 763
 764         vlc_object_release( p_ts );
 765
 766         p_sys->b_delayed = false;
 767         return VLC_EGENERIC;
 768     }
 769
 770     return VLC_SUCCESS;
 771 }
 772 static void TsAutoStop( es_out_t *p_out )
 773 {
 774     es_out_sys_t *p_sys = p_out->p_sys;
 775
 776     if( !p_sys->b_delayed || !TsIsUnused( p_sys->p_thread ) )
 777         return;
 778
 779     msg_Warn( p_sys->p_input, "es out timeshift: auto stop" );
 780     TsStop( p_sys->p_thread );
 781
 782     p_sys->b_delayed = false;
 783 }
 784 static void TsStop( ts_thread_t *p_ts )
 785 {
 786     vlc_object_kill( p_ts );
 787     vlc_thread_join( p_ts );
 788
 789     vlc_mutex_lock( &p_ts->lock );
 790     for( ;; )
 791     {
 792         ts_cmd_t cmd;
 793
 794         if( TsPopCmdLocked( p_ts, &cmd, true ) )
 795             break;
 796
 797         CmdClean( &cmd );
 798     }
 799     assert( !p_ts->p_storage_r || !p_ts->p_storage_r->p_next );
 800     if( p_ts->p_storage_r )
 801         TsStorageDelete( p_ts->p_storage_r );
 802     vlc_mutex_unlock( &p_ts->lock );
 803
 804     vlc_object_release( p_ts );
 805 }
 806 static void TsPushCmd( ts_thread_t *p_ts, ts_cmd_t *p_cmd )
 807 {
 808     vlc_mutex_lock( &p_ts->lock );
 809
 810     if( !p_ts->p_storage_w || TsStorageIsFull( p_ts->p_storage_w, p_cmd ) )
 811     {
 812         ts_storage_t *p_storage = TsStorageNew( p_ts->psz_tmp_path, p_ts->i_tmp_size_max );
 813
 814         if( !p_storage )
 815         {
 816             CmdClean( p_cmd );
 817             vlc_mutex_unlock( &p_ts->lock );
 818             /* TODO warn the user (but only once) */
 819             return;
 820         }
 821
 822         if( !p_ts->p_storage_w )
 823         {
 824             p_ts->p_storage_r = p_ts->p_storage_w = p_storage;
 825         }
 826         else
 827         {
 828             TsStoragePack( p_ts->p_storage_w );
 829             p_ts->p_storage_w->p_next = p_storage;
 830             p_ts->p_storage_w = p_storage;
 831         }
 832     }
 833
 834     /* TODO return error and warn the user (but only once) */
 835     TsStoragePushCmd( p_ts->p_storage_w, p_cmd, p_ts->p_storage_r == p_ts->p_storage_w );
 836
 837     vlc_cond_signal( &p_ts->wait );
 838
 839     vlc_mutex_unlock( &p_ts->lock );
 840 }
 841 static int TsPopCmdLocked( ts_thread_t *p_ts, ts_cmd_t *p_cmd, bool b_flush )
 842 {
 843     vlc_assert_locked( &p_ts->lock );
 844
 845     if( TsStorageIsEmpty( p_ts->p_storage_r ) )
 846         return VLC_EGENERIC;
 847
 848     TsStoragePopCmd( p_ts->p_storage_r, p_cmd, b_flush );
 849
 850     while( p_ts->p_storage_r && TsStorageIsEmpty( p_ts->p_storage_r ) )
 851     {
 852         ts_storage_t *p_next = p_ts->p_storage_r->p_next;
 853         if( !p_next )
 854             break;
 855
 856         TsStorageDelete( p_ts->p_storage_r );
 857         p_ts->p_storage_r = p_next;
 858     }
 859
 860     return VLC_SUCCESS;
 861 }
 862 static bool TsHasCmd( ts_thread_t *p_ts )
 863 {
 864     bool b_cmd;
 865
 866     vlc_mutex_lock( &p_ts->lock );
 867     b_cmd =  TsStorageIsEmpty( p_ts->p_storage_r );
 868     vlc_mutex_unlock( &p_ts->lock );
 869
 870     return b_cmd;
 871 }
 872 static bool TsIsUnused( ts_thread_t *p_ts )
 873 {
 874     bool b_unused;
 875
 876     vlc_mutex_lock( &p_ts->lock );
 877     b_unused = !p_ts->b_paused &&
 878                p_ts->i_rate == p_ts->i_rate_source &&
 879                TsStorageIsEmpty( p_ts->p_storage_r );
 880     vlc_mutex_unlock( &p_ts->lock );
 881
 882     return b_unused;
 883 }
 884 static int TsChangePause( ts_thread_t *p_ts, bool b_source_paused, bool b_paused, mtime_t i_date )
 885 {
 886     vlc_mutex_lock( &p_ts->lock );
 887
 888     int i_ret;
 889     if( b_paused )
 890     {
 891         assert( !b_source_paused );
 892         i_ret = es_out_SetPauseState( p_ts->p_out, true, true, i_date );
 893     }
 894     else
 895     {
 896         i_ret = es_out_SetPauseState( p_ts->p_out, false, false, i_date );
 897     }
 898
 899     if( !i_ret )
 900     {
 901         if( !b_paused )
 902         {
 903             assert( p_ts->i_pause_date > 0 );
 904
 905             p_ts->i_cmd_delay += i_date - p_ts->i_pause_date;
 906         }
 907
 908         p_ts->b_paused = b_paused;
 909         p_ts->i_pause_date = i_date;
 910
 911         vlc_cond_signal( &p_ts->wait );
 912     }
 913     vlc_mutex_unlock( &p_ts->lock );
 914     return i_ret;
 915 }
 916 static int TsChangeRate( ts_thread_t *p_ts, int i_src_rate, int i_rate )
 917 {
 918     int i_ret;
 919
 920     vlc_mutex_lock( &p_ts->lock );
 921     p_ts->i_cmd_delay += p_ts->i_rate_delay;
 922
 923     p_ts->i_rate_date = -1;
 924     p_ts->i_rate_delay = 0;
 925     p_ts->i_rate = i_rate;
 926     p_ts->i_rate_source = i_src_rate;
 927
 928     i_ret = es_out_SetRate( p_ts->p_out, i_rate, i_rate );
 929     vlc_mutex_unlock( &p_ts->lock );
 930
 931     return i_ret;
 932 }
 933
 934 static void *TsRun( vlc_object_t *p_thread )
 935 {
 936     ts_thread_t *p_ts = (ts_thread_t*)p_thread;
 937     mtime_t i_buffering_date = -1;
 938
 939     for( ;; )
 940     {
 941         ts_cmd_t cmd;
 942         mtime_t  i_deadline;
 943         bool b_buffering;
 944
 945         /* Pop a command to execute */
 946         vlc_mutex_lock( &p_ts->lock );
 947         mutex_cleanup_push( &p_ts->lock );
 948
 949         for( ;; )
 950         {
 951             const int canc = vlc_savecancel();
 952             b_buffering = es_out_GetBuffering( p_ts->p_out );
 953
 954             if( ( !p_ts->b_paused || b_buffering ) && !TsPopCmdLocked( p_ts, &cmd, false ) )
 955             {
 956                 vlc_restorecancel( canc );
 957                 break;
 958             }
 959             vlc_restorecancel( canc );
 960
 961             vlc_cond_wait( &p_ts->wait, &p_ts->lock );
 962         }
 963
 964         if( b_buffering && i_buffering_date < 0 )
 965         {
 966             i_buffering_date = cmd.i_date;
 967         }
 968         else if( i_buffering_date > 0 )
 969         {
 970             p_ts->i_buffering_delay += i_buffering_date - cmd.i_date; /* It is < 0 */
 971             if( b_buffering )
 972                 i_buffering_date = cmd.i_date;
 973             else
 974                 i_buffering_date = -1;
 975         }
 976
 977         if( p_ts->i_rate_date < 0 )
 978             p_ts->i_rate_date = cmd.i_date;
 979
 980         p_ts->i_rate_delay = 0;
 981         if( p_ts->i_rate_source != p_ts->i_rate )
 982         {
 983             const mtime_t i_duration = cmd.i_date - p_ts->i_rate_date;
 984             p_ts->i_rate_delay = i_duration * p_ts->i_rate / p_ts->i_rate_source - i_duration;
 985         }
 986         if( p_ts->i_cmd_delay + p_ts->i_rate_delay + p_ts->i_buffering_delay < 0 && p_ts->i_rate != p_ts->i_rate_source )
 987         {
 988             const int canc = vlc_savecancel();
 989
 990             /* Auto reset to rate 1.0 */
 991             msg_Warn( p_ts->p_input, "es out timeshift: auto reset rate to %d", p_ts->i_rate_source );
 992
 993             p_ts->i_cmd_delay = 0;
 994             p_ts->i_buffering_delay = 0;
 995
 996             p_ts->i_rate_delay = 0;
 997             p_ts->i_rate_date = -1;
 998             p_ts->i_rate = p_ts->i_rate_source;
 999
1000             if( !es_out_SetRate( p_ts->p_out, p_ts->i_rate_source, p_ts->i_rate ) )
1001             {
1002                 vlc_value_t val = { .i_int = p_ts->i_rate };
1003                 /* Warn back input
1004                  * FIXME it is perfectly safe BUT it is ugly as it may hide a
1005                  * rate change requested by user */
1006                 input_ControlPush( p_ts->p_input, INPUT_CONTROL_SET_RATE, &val );
1007             }
1008
1009             vlc_restorecancel( canc );
1010         }
1011         i_deadline = cmd.i_date + p_ts->i_cmd_delay + p_ts->i_rate_delay + p_ts->i_buffering_delay;
1012
1013         vlc_cleanup_run();
1014
1015         /* Regulate the speed of command processing to the same one than
1016          * reading  */
1017         vlc_cleanup_push( cmd_cleanup_routine, &cmd );
1018
1019         mwait( i_deadline );
1020
1021         vlc_cleanup_pop();
1022
1023         /* Execute the command  */
1024         const int canc = vlc_savecancel();
1025         switch( cmd.i_type )
1026         {
1027         case C_ADD:
1028             CmdExecuteAdd( p_ts->p_out, &cmd );
1029             CmdCleanAdd( &cmd );
1030             break;
1031         case C_SEND:
1032             CmdExecuteSend( p_ts->p_out, &cmd );
1033             CmdCleanSend( &cmd );
1034             break;
1035         case C_CONTROL:
1036             CmdExecuteControl( p_ts->p_out, &cmd );
1037             CmdCleanControl( &cmd );
1038             break;
1039         case C_DEL:
1040             CmdExecuteDel( p_ts->p_out, &cmd );
1041             break;
1042         default:
1043             assert(0);
1044             break;
1045         }
1046         vlc_restorecancel( canc );
1047     }
1048
1049     return NULL;
1050 }
1051
1052 /*****************************************************************************
1053  *
1054  *****************************************************************************/
1055 static ts_storage_t *TsStorageNew( const char *psz_tmp_path, int64_t i_tmp_size_max )
1056 {
1057     ts_storage_t *p_storage = calloc( 1, sizeof(ts_storage_t) );
1058     if( !p_storage )
1059         return NULL;
1060
1061     /* */
1062     p_storage->p_next = NULL;
1063
1064     /* */
1065     p_storage->i_file_max = i_tmp_size_max;
1066     p_storage->i_file_size = 0;
1067     p_storage->p_filew = GetTmpFile( &p_storage->psz_file, psz_tmp_path );
1068     if( p_storage->psz_file )
1069         p_storage->p_filer = utf8_fopen( p_storage->psz_file, "rb" );
1070
1071     /* */
1072     p_storage->i_cmd_w = 0;
1073     p_storage->i_cmd_r = 0;
1074     p_storage->i_cmd_max = 30000;
1075     p_storage->p_cmd = malloc( p_storage->i_cmd_max * sizeof(*p_storage->p_cmd) );
1076     //fprintf( stderr, "\nSTORAGE name=%s size=%d kbytes\n", p_storage->psz_file, p_storage->i_cmd_max * sizeof(*p_storage->p_cmd) /1024 );
1077
1078     if( !p_storage->p_cmd || !p_storage->p_filew || !p_storage->p_filer )
1079     {
1080         TsStorageDelete( p_storage );
1081         return NULL;
1082     }
1083     return p_storage;
1084 }
1085 static void TsStorageDelete( ts_storage_t *p_storage )
1086 {
1087     while( p_storage->i_cmd_r < p_storage->i_cmd_w )
1088     {
1089         ts_cmd_t cmd;
1090
1091         TsStoragePopCmd( p_storage, &cmd, true );
1092
1093         CmdClean( &cmd );
1094     }
1095     free( p_storage->p_cmd );
1096
1097     if( p_storage->p_filer )
1098         fclose( p_storage->p_filer );
1099     if( p_storage->p_filew )
1100         fclose( p_storage->p_filew );
1101
1102     if( p_storage->psz_file )
1103     {
1104         utf8_unlink( p_storage->psz_file );
1105         free( p_storage->psz_file );
1106     }
1107
1108     free( p_storage );
1109 }
1110 static void TsStoragePack( ts_storage_t *p_storage )
1111 {
1112     /* Try to release a bit of memory */
1113     if( p_storage->i_cmd_w >= p_storage->i_cmd_max )
1114         return;
1115
1116     p_storage->i_cmd_max = __MAX( p_storage->i_cmd_w, 1 );
1117
1118     ts_cmd_t *p_new = realloc( p_storage->p_cmd, p_storage->i_cmd_max * sizeof(*p_storage->p_cmd) );
1119     if( p_new )
1120         p_storage->p_cmd = p_new;
1121 }
1122 static bool TsStorageIsFull( ts_storage_t *p_storage, const ts_cmd_t *p_cmd )
1123 {
1124     if( p_cmd && p_cmd->i_type == C_SEND && p_storage->i_cmd_w > 0 )
1125     {
1126         size_t i_size = sizeof(*p_cmd->u.send.p_block) + p_cmd->u.send.p_block->i_buffer;
1127
1128         if( p_storage->i_file_size + i_size >= p_storage->i_file_max )
1129             return true;
1130     }
1131     return p_storage->i_cmd_w >= p_storage->i_cmd_max;
1132 }
1133 static bool TsStorageIsEmpty( ts_storage_t *p_storage )
1134 {
1135     return !p_storage || p_storage->i_cmd_r >= p_storage->i_cmd_w;
1136 }
1137 static void TsStoragePushCmd( ts_storage_t *p_storage, const ts_cmd_t *p_cmd, bool b_flush )
1138 {
1139     ts_cmd_t cmd = *p_cmd;
1140
1141     assert( !TsStorageIsFull( p_storage, p_cmd ) );
1142
1143     if( cmd.i_type == C_SEND )
1144     {
1145         block_t *p_block = cmd.u.send.p_block;
1146
1147         cmd.u.send.p_block = NULL;
1148         cmd.u.send.i_offset = ftell( p_storage->p_filew );
1149
1150         if( fwrite( p_block, sizeof(*p_block), 1, p_storage->p_filew ) != 1 )
1151         {
1152             block_Release( p_block );
1153             return;
1154         }
1155         p_storage->i_file_size += sizeof(*p_block);
1156         if( p_block->i_buffer > 0 )
1157         {
1158             if( fwrite( p_block->p_buffer, p_block->i_buffer, 1, p_storage->p_filew ) != 1 )
1159             {
1160                 block_Release( p_block );
1161                 return;
1162             }
1163         }
1164         p_storage->i_file_size += p_block->i_buffer;
1165         block_Release( p_block );
1166
1167         if( b_flush )
1168             fflush( p_storage->p_filew );
1169     }
1170     p_storage->p_cmd[p_storage->i_cmd_w++] = cmd;
1171 }
1172 static void TsStoragePopCmd( ts_storage_t *p_storage, ts_cmd_t *p_cmd, bool b_flush )
1173 {
1174     assert( !TsStorageIsEmpty( p_storage ) );
1175
1176     *p_cmd = p_storage->p_cmd[p_storage->i_cmd_r++];
1177     if( p_cmd->i_type == C_SEND )
1178     {
1179         block_t block;
1180
1181         if( !b_flush &&
1182             !fseek( p_storage->p_filer, p_cmd->u.send.i_offset, SEEK_SET ) &&
1183             fread( &block, sizeof(block), 1, p_storage->p_filer ) == 1 )
1184         {
1185             block_t *p_block = block_Alloc( block.i_buffer );
1186             if( p_block )
1187             {
1188                 p_block->i_dts      = block.i_dts;
1189                 p_block->i_pts      = block.i_pts;
1190                 p_block->i_flags    = block.i_flags;
1191                 p_block->i_length   = block.i_length;
1192                 p_block->i_rate     = block.i_rate;
1193                 p_block->i_nb_samples = block.i_nb_samples;
1194                 p_block->i_buffer = fread( p_block->p_buffer, 1, block.i_buffer, p_storage->p_filer );
1195             }
1196             p_cmd->u.send.p_block = p_block;
1197         }
1198         else
1199         {
1200             //fprintf( stderr, "TsStoragePopCmd: %m\n" );
1201             p_cmd->u.send.p_block = block_Alloc( 1 );
1202         }
1203     }
1204 }
1205
1206 /*****************************************************************************
1207  *
1208  *****************************************************************************/
1209 static void CmdClean( ts_cmd_t *p_cmd )
1210 {
1211     switch( p_cmd->i_type )
1212     {
1213     case C_ADD:
1214         CmdCleanAdd( p_cmd );
1215         break;
1216     case C_SEND:
1217         CmdCleanSend( p_cmd );
1218         break;
1219     case C_CONTROL:
1220         CmdCleanControl( p_cmd );
1221         break;
1222     case C_DEL:
1223         break;
1224     default:
1225         assert(0);
1226         break;
1227     }
1228 }
1229
1230 static int CmdInitAdd( ts_cmd_t *p_cmd, es_out_id_t *p_es, const es_format_t *p_fmt, bool b_copy )
1231 {
1232     p_cmd->i_type = C_ADD;
1233     p_cmd->i_date = mdate();
1234     p_cmd->u.add.p_es = p_es;
1235     if( b_copy )
1236     {
1237         p_cmd->u.add.p_fmt = malloc( sizeof(*p_fmt) );
1238         if( !p_cmd->u.add.p_fmt )
1239             return VLC_EGENERIC;
1240         es_format_Copy( p_cmd->u.add.p_fmt, p_fmt );
1241     }
1242     else
1243     {
1244         p_cmd->u.add.p_fmt = (es_format_t*)p_fmt;
1245     }
1246     return VLC_SUCCESS;
1247 }
1248 static void CmdExecuteAdd( es_out_t *p_out, ts_cmd_t *p_cmd )
1249 {
1250     p_cmd->u.add.p_es->p_es = es_out_Add( p_out, p_cmd->u.add.p_fmt );
1251 }
1252 static void CmdCleanAdd( ts_cmd_t *p_cmd )
1253 {
1254     es_format_Clean( p_cmd->u.add.p_fmt );
1255     free( p_cmd->u.add.p_fmt );
1256 }
1257
1258 static void CmdInitSend( ts_cmd_t *p_cmd, es_out_id_t *p_es, block_t *p_block )
1259 {
1260     p_cmd->i_type = C_SEND;
1261     p_cmd->i_date = mdate();
1262     p_cmd->u.send.p_es = p_es;
1263     p_cmd->u.send.p_block = p_block;
1264 }
1265 static int CmdExecuteSend( es_out_t *p_out, ts_cmd_t *p_cmd )
1266 {
1267     block_t *p_block = p_cmd->u.send.p_block;
1268
1269     p_cmd->u.send.p_block = NULL;
1270
1271     if( p_block )
1272     {
1273         if( p_cmd->u.send.p_es->p_es )
1274             return es_out_Send( p_out, p_cmd->u.send.p_es->p_es, p_block );
1275         block_Release( p_block );
1276     }
1277     return VLC_EGENERIC;
1278 }
1279 static void CmdCleanSend( ts_cmd_t *p_cmd )
1280 {
1281     if( p_cmd->u.send.p_block )
1282         block_Release( p_cmd->u.send.p_block );
1283 }
1284
1285 static int CmdInitDel( ts_cmd_t *p_cmd, es_out_id_t *p_es )
1286 {
1287     p_cmd->i_type = C_DEL;
1288     p_cmd->i_date = mdate();
1289     p_cmd->u.del.p_es = p_es;
1290     return VLC_SUCCESS;
1291 }
1292 static void CmdExecuteDel( es_out_t *p_out, ts_cmd_t *p_cmd )
1293 {
1294     if( p_cmd->u.del.p_es->p_es )
1295         es_out_Del( p_out, p_cmd->u.del.p_es->p_es );
1296     free( p_cmd->u.del.p_es );
1297 }
1298
1299 static int CmdInitControl( ts_cmd_t *p_cmd, int i_query, va_list args, bool b_copy )
1300 {
1301     p_cmd->i_type = C_CONTROL;
1302     p_cmd->i_date = mdate();
1303     p_cmd->u.control.i_query = i_query;
1304
1305     switch( i_query )
1306     {
1307     /* Pass-through control */
1308     case ES_OUT_SET_ACTIVE:  /* arg1= bool                     */
1309         p_cmd->u.control.u.b_bool = (bool)va_arg( args, int );
1310         break;
1311
1312     case ES_OUT_SET_MODE:    /* arg1= int                            */
1313     case ES_OUT_SET_GROUP:   /* arg1= int                            */
1314     case ES_OUT_DEL_GROUP:   /* arg1=int i_group */
1315         p_cmd->u.control.u.i_int = (int)va_arg( args, int );
1316         break;
1317
1318     case ES_OUT_SET_PCR:                /* arg1=int64_t i_pcr(microsecond!) (using default group 0)*/
1319     case ES_OUT_SET_NEXT_DISPLAY_TIME:  /* arg1=int64_t i_pts(microsecond) */
1320         p_cmd->u.control.u.i_i64 = (int64_t)va_arg( args, int64_t );
1321         break;
1322
1323     case ES_OUT_SET_GROUP_PCR:          /* arg1= int i_group, arg2=int64_t i_pcr(microsecond!)*/
1324         p_cmd->u.control.u.int_i64.i_int = (int)va_arg( args, int );
1325         p_cmd->u.control.u.int_i64.i_i64 = (int64_t)va_arg( args, int64_t );
1326         break;
1327
1328     case ES_OUT_SET_ES_SCRAMBLED_STATE:
1329         p_cmd->u.control.u.es_bool.p_es = (es_out_id_t*)va_arg( args, es_out_id_t * );
1330         p_cmd->u.control.u.es_bool.b_bool = (bool)va_arg( args, int );
1331         break;
1332
1333     case ES_OUT_RESET_PCR:           /* no arg */
1334         break;
1335
1336     case ES_OUT_SET_META:        /* arg1=const vlc_meta_t* */
1337     case ES_OUT_SET_GROUP_META:  /* arg1=int i_group arg2=const vlc_meta_t* */
1338     {
1339         if( i_query == ES_OUT_SET_GROUP_META )
1340             p_cmd->u.control.u.int_meta.i_int = (int)va_arg( args, int );
1341         const vlc_meta_t *p_meta = va_arg( args, const vlc_meta_t * );
1342
1343         if( b_copy )
1344         {
1345             p_cmd->u.control.u.int_meta.p_meta = vlc_meta_New();
1346             if( !p_cmd->u.control.u.int_meta.p_meta )
1347                 return VLC_EGENERIC;
1348             vlc_meta_Merge( p_cmd->u.control.u.int_meta.p_meta, p_meta );
1349         }
1350         else
1351         {
1352             /* The cast is only needed to avoid warning */
1353             p_cmd->u.control.u.int_meta.p_meta = (vlc_meta_t*)p_meta;
1354         }
1355         break;
1356     }
1357
1358     case ES_OUT_SET_GROUP_EPG:   /* arg1=int i_group arg2=const vlc_epg_t* */
1359     {
1360         p_cmd->u.control.u.int_epg.i_int = (int)va_arg( args, int );
1361         const vlc_epg_t *p_epg = va_arg( args, const vlc_epg_t * );
1362
1363         if( b_copy )
1364         {
1365             p_cmd->u.control.u.int_epg.p_epg = vlc_epg_New( p_epg->psz_name );
1366             if( !p_cmd->u.control.u.int_epg.p_epg )
1367                 return VLC_EGENERIC;
1368             for( int i = 0; i < p_epg->i_event; i++ )
1369             {
1370                 vlc_epg_event_t *p_evt = p_epg->pp_event[i];
1371
1372                 vlc_epg_AddEvent( p_cmd->u.control.u.int_epg.p_epg,
1373                                   p_evt->i_start, p_evt->i_duration,
1374                                   p_evt->psz_name,
1375                                   p_evt->psz_short_description, p_evt->psz_description );
1376             }
1377             vlc_epg_SetCurrent( p_cmd->u.control.u.int_epg.p_epg,
1378                                 p_epg->p_current ? p_epg->p_current->i_start : -1 );
1379         }
1380         else
1381         {
1382             /* The cast is only needed to avoid warning */
1383             p_cmd->u.control.u.int_epg.p_epg = (vlc_epg_t*)p_epg;
1384         }
1385         break;
1386     }
1387
1388     /* Modified control */
1389     case ES_OUT_SET_ES:      /* arg1= es_out_id_t*                   */
1390     case ES_OUT_RESTART_ES:  /* arg1= es_out_id_t*                   */
1391     case ES_OUT_SET_ES_DEFAULT: /* arg1= es_out_id_t*                */
1392         p_cmd->u.control.u.p_es = (es_out_id_t*)va_arg( args, es_out_id_t * );
1393         break;
1394
1395     case ES_OUT_SET_ES_STATE:/* arg1= es_out_id_t* arg2=bool   */
1396         p_cmd->u.control.u.es_bool.p_es = (es_out_id_t*)va_arg( args, es_out_id_t * );
1397         p_cmd->u.control.u.es_bool.b_bool = (bool)va_arg( args, int );
1398         break;
1399
1400     case ES_OUT_SET_ES_FMT:     /* arg1= es_out_id_t* arg2=es_format_t* */
1401     {
1402         p_cmd->u.control.u.es_fmt.p_es = (es_out_id_t*)va_arg( args, es_out_id_t * );
1403         es_format_t *p_fmt = (es_format_t*)va_arg( args, es_format_t * );
1404
1405         if( b_copy )
1406         {
1407             p_cmd->u.control.u.es_fmt.p_fmt = malloc( sizeof(*p_fmt) );
1408             if( !p_cmd->u.control.u.es_fmt.p_fmt )
1409                 return VLC_EGENERIC;
1410             es_format_Copy( p_cmd->u.control.u.es_fmt.p_fmt, p_fmt );
1411         }
1412         else
1413         {
1414             p_cmd->u.control.u.es_fmt.p_fmt = p_fmt;
1415         }
1416         break;
1417     }
1418     case ES_OUT_SET_TIMES:
1419     {
1420         double f_position = (double)va_arg( args, double );
1421         mtime_t i_time = (mtime_t)va_arg( args, mtime_t );
1422         mtime_t i_length = (mtime_t)va_arg( args, mtime_t );
1423
1424         p_cmd->u.control.u.times.f_position = f_position;
1425         p_cmd->u.control.u.times.i_time = i_time;
1426         p_cmd->u.control.u.times.i_length = i_length;
1427         break;
1428     }
1429     case ES_OUT_SET_JITTER:
1430     {
1431         mtime_t i_pts_delay = (mtime_t)va_arg( args, mtime_t );
1432         int     i_cr_average = (int)va_arg( args, int );
1433
1434         p_cmd->u.control.u.jitter.i_pts_delay = i_pts_delay;
1435         p_cmd->u.control.u.jitter.i_cr_average = i_cr_average;
1436         break;
1437     }
1438
1439     default:
1440         assert(0);
1441         return VLC_EGENERIC;
1442     }
1443
1444     return VLC_SUCCESS;
1445 }
1446 static int CmdExecuteControl( es_out_t *p_out, ts_cmd_t *p_cmd )
1447 {
1448     const int i_query = p_cmd->u.control.i_query;
1449
1450     switch( i_query )
1451     {
1452     /* Pass-through control */
1453     case ES_OUT_SET_ACTIVE:  /* arg1= bool                     */
1454         return es_out_Control( p_out, i_query, p_cmd->u.control.u.b_bool );
1455
1456     case ES_OUT_SET_MODE:    /* arg1= int                            */
1457     case ES_OUT_SET_GROUP:   /* arg1= int                            */
1458     case ES_OUT_DEL_GROUP:   /* arg1=int i_group */
1459         return es_out_Control( p_out, i_query, p_cmd->u.control.u.i_int );
1460
1461     case ES_OUT_SET_PCR:                /* arg1=int64_t i_pcr(microsecond!) (using default group 0)*/
1462     case ES_OUT_SET_NEXT_DISPLAY_TIME:  /* arg1=int64_t i_pts(microsecond) */
1463         return es_out_Control( p_out, i_query, p_cmd->u.control.u.i_i64 );
1464
1465     case ES_OUT_SET_GROUP_PCR:          /* arg1= int i_group, arg2=int64_t i_pcr(microsecond!)*/
1466         return es_out_Control( p_out, i_query, p_cmd->u.control.u.int_i64.i_int,
1467                                                p_cmd->u.control.u.int_i64.i_i64 );
1468
1469     case ES_OUT_RESET_PCR:           /* no arg */
1470         return es_out_Control( p_out, i_query );
1471
1472     case ES_OUT_SET_GROUP_META:  /* arg1=int i_group arg2=const vlc_meta_t* */
1473         return es_out_Control( p_out, i_query, p_cmd->u.control.u.int_meta.i_int,
1474                                                p_cmd->u.control.u.int_meta.p_meta );
1475
1476     case ES_OUT_SET_GROUP_EPG:   /* arg1=int i_group arg2=const vlc_epg_t* */
1477         return es_out_Control( p_out, i_query, p_cmd->u.control.u.int_epg.i_int,
1478                                                p_cmd->u.control.u.int_epg.p_epg );
1479
1480     case ES_OUT_SET_ES_SCRAMBLED_STATE: /* arg1=int es_out_id_t* arg2=bool */
1481         return es_out_Control( p_out, i_query, p_cmd->u.control.u.es_bool.p_es->p_es,
1482                                                p_cmd->u.control.u.es_bool.b_bool );
1483
1484     case ES_OUT_SET_META:  /* arg1=const vlc_meta_t* */
1485         return es_out_Control( p_out, i_query, p_cmd->u.control.u.int_meta.p_meta );
1486
1487     /* Modified control */
1488     case ES_OUT_SET_ES:      /* arg1= es_out_id_t*                   */
1489     case ES_OUT_RESTART_ES:  /* arg1= es_out_id_t*                   */
1490     case ES_OUT_SET_ES_DEFAULT: /* arg1= es_out_id_t*                */
1491         return es_out_Control( p_out, i_query, p_cmd->u.control.u.p_es->p_es );
1492
1493     case ES_OUT_SET_ES_STATE:/* arg1= es_out_id_t* arg2=bool   */
1494         return es_out_Control( p_out, i_query, p_cmd->u.control.u.es_bool.p_es->p_es,
1495                                                p_cmd->u.control.u.es_bool.b_bool );
1496
1497     case ES_OUT_SET_ES_FMT:     /* arg1= es_out_id_t* arg2=es_format_t* */
1498         return es_out_Control( p_out, i_query, p_cmd->u.control.u.es_fmt.p_es->p_es,
1499                                                p_cmd->u.control.u.es_fmt.p_fmt );
1500
1501     case ES_OUT_SET_TIMES:
1502         return es_out_Control( p_out, i_query, p_cmd->u.control.u.times.f_position,
1503                                                p_cmd->u.control.u.times.i_time,
1504                                                p_cmd->u.control.u.times.i_length );
1505     case ES_OUT_SET_JITTER:
1506         return es_out_Control( p_out, i_query, p_cmd->u.control.u.jitter.i_pts_delay,
1507                                                p_cmd->u.control.u.jitter.i_cr_average );
1508
1509     default:
1510         assert(0);
1511         return VLC_EGENERIC;
1512     }
1513 }
1514 static void CmdCleanControl( ts_cmd_t *p_cmd )
1515 {
1516     if( ( p_cmd->u.control.i_query == ES_OUT_SET_GROUP_META ||
1517           p_cmd->u.control.i_query == ES_OUT_SET_META ) &&
1518         p_cmd->u.control.u.int_meta.p_meta )
1519     {
1520         vlc_meta_Delete( p_cmd->u.control.u.int_meta.p_meta );
1521     }
1522     else if( p_cmd->u.control.i_query == ES_OUT_SET_GROUP_EPG &&
1523              p_cmd->u.control.u.int_epg.p_epg )
1524     {
1525         vlc_epg_Delete( p_cmd->u.control.u.int_epg.p_epg );
1526     }
1527     else if( p_cmd->u.control.i_query == ES_OUT_SET_ES_FMT &&
1528              p_cmd->u.control.u.es_fmt.p_fmt )
1529     {
1530         es_format_Clean( p_cmd->u.control.u.es_fmt.p_fmt );
1531         free( p_cmd->u.control.u.es_fmt.p_fmt );
1532     }
1533 }
1534
1535
1536 /*****************************************************************************
1537  * GetTmpFile/Path:
1538  *****************************************************************************/
1539 static char *GetTmpPath( char *psz_path )
1540 {
1541     if( psz_path && *psz_path )
1542     {
1543         /* Make sure that the path exists and is a directory */
1544         struct stat s;
1545         const int i_ret = utf8_stat( psz_path, &s );
1546
1547         if( i_ret < 0 && !utf8_mkdir( psz_path, 0600 ) )
1548             return psz_path;
1549         else if( i_ret == 0 && ( s.st_mode & S_IFDIR ) )
1550             return psz_path;
1551     }
1552     free( psz_path );
1553
1554     /* Create a suitable path */
1555 #if defined (WIN32) && !defined (UNDER_CE)
1556     const DWORD dwCount = GetTempPathW( 0, NULL );
1557     wchar_t *psw_path = calloc( dwCount + 1, sizeof(wchar_t) );
1558     if( psw_path )
1559     {
1560         if( GetTempPathW( dwCount + 1, psw_path ) <= 0 )
1561         {
1562             free( psw_path );
1563
1564             psw_path = _wgetcwd( NULL, 0 );
1565         }
1566     }
1567
1568     psz_path = NULL;
1569     if( psw_path )
1570     {
1571         psz_path = FromWide( psw_path );
1572         while( psz_path && *psz_path && psz_path[strlen( psz_path ) - 1] == '\\' )
1573             psz_path[strlen( psz_path ) - 1] = '\0';
1574
1575         free( psw_path );
1576     }
1577
1578     if( !psz_path || *psz_path == '\0' )
1579     {
1580         free( psz_path );
1581         return strdup( "C:" );
1582     }
1583 #else
1584     psz_path = strdup( "/tmp" );
1585 #endif
1586
1587     return psz_path;
1588 }
1589
1590 static FILE *GetTmpFile( char **ppsz_file, const char *psz_path )
1591 {
1592     char *psz_name;
1593     int fd;
1594     FILE *f;
1595
1596     /* */
1597     *ppsz_file = NULL;
1598     if( asprintf( &psz_name, "%s/vlc-timeshift.XXXXXX", psz_path ) < 0 )
1599         return NULL;
1600
1601     /* */
1602     fd = utf8_mkstemp( psz_name );
1603     *ppsz_file = psz_name;
1604
1605     if( fd < 0 )
1606         return NULL;
1607
1608     /* */
1609     f = fdopen( fd, "w+b" );
1610     if( !f )
1611         close( fd );
1612
1613     return f;
1614 }
1615