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