?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
access: http: only warn on deflate errors
[vlc.git] / src / input / es_out_timeshift.c
blob520f7358119c10344cbec055280c9c8ebfdd6f03
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 it
10 * under the terms of the GNU Lesser General Public License as published by
11 * the Free Software Foundation; either version 2.1 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 Lesser General Public License for more details.
18 *
19 * You should have received a copy of the GNU Lesser General Public License
20 * along with this program; if not, write to the Free Software Foundation,
21 * 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)
35 # include <direct.h>
36 #endif
37 #include <sys/stat.h>
38 #ifdef HAVE_UNISTD_H
39 # include <unistd.h>
40 #endif
41
42 #include <vlc_common.h>
43 #include <vlc_fs.h>
44 #ifdef WIN32
45 # include <vlc_charset.h>
46 #endif
47 #include <vlc_input.h>
48 #include <vlc_es_out.h>
49 #include <vlc_block.h>
50 #include "input_internal.h"
51 #include "es_out.h"
52 #include "es_out_timeshift.h"
53
54 /*****************************************************************************
55 * Local prototypes
56 *****************************************************************************/
57
58 /* XXX attribute_packed is (and MUST be) used ONLY to reduce memory usage */
59 #ifdef HAVE_ATTRIBUTE_PACKED
60 # define attribute_packed __attribute__((__packed__))
61 #else
62 # define attribute_packed
63 #endif
64
65 enum
66 {
67 C_ADD,
68 C_SEND,
69 C_DEL,
70 C_CONTROL,
71 };
72
73 typedef struct attribute_packed
74 {
75 es_out_id_t *p_es;
76 es_format_t *p_fmt;
77 } ts_cmd_add_t;
78
79 typedef struct attribute_packed
80 {
81 es_out_id_t *p_es;
82 } ts_cmd_del_t;
83
84 typedef struct attribute_packed
85 {
86 es_out_id_t *p_es;
87 block_t *p_block;
88 int i_offset; /* We do not use file > INT_MAX */
89 } ts_cmd_send_t;
90
91 typedef struct attribute_packed
92 {
93 int i_query;
94
95 union
96 {
97 bool b_bool;
98 int i_int;
99 int64_t i_i64;
100 es_out_id_t *p_es;
101 struct
102 {
103 int i_int;
104 int64_t i_i64;
105 } int_i64;
106 struct
107 {
108 int i_int;
109 vlc_meta_t *p_meta;
110 } int_meta;
111 struct
112 {
113 int i_int;
114 vlc_epg_t *p_epg;
115 } int_epg;
116 struct
117 {
118 es_out_id_t *p_es;
119 bool b_bool;
120 } es_bool;
121 struct
122 {
123 es_out_id_t *p_es;
124 es_format_t *p_fmt;
125 } es_fmt;
126 struct
127 {
128 /* FIXME Really too big (double make the whole thing too big) */
129 double f_position;
130 mtime_t i_time;
131 mtime_t i_length;
132 } times;
133 struct
134 {
135 mtime_t i_pts_delay;
136 mtime_t i_pts_jitter;
137 int i_cr_average;
138 } jitter;
139 } u;
140 } ts_cmd_control_t;
141
142 typedef struct attribute_packed
143 {
144 int8_t i_type;
145 mtime_t i_date;
146 union
147 {
148 ts_cmd_add_t add;
149 ts_cmd_del_t del;
150 ts_cmd_send_t send;
151 ts_cmd_control_t control;
152 } u;
153 } ts_cmd_t;
154
155 typedef struct ts_storage_t ts_storage_t;
156 struct ts_storage_t
157 {
158 ts_storage_t *p_next;
159
160 /* */
161 char *psz_file; /* Filename */
162 size_t i_file_max; /* Max size in bytes */
163 int64_t i_file_size;/* Current size in bytes */
164 FILE *p_filew; /* FILE handle for data writing */
165 FILE *p_filer; /* FILE handle for data reading */
166
167 /* */
168 int i_cmd_r;
169 int i_cmd_w;
170 int i_cmd_max;
171 ts_cmd_t *p_cmd;
172 };
173
174 typedef struct
175 {
176 vlc_thread_t thread;
177 input_thread_t *p_input;
178 es_out_t *p_out;
179 int64_t i_tmp_size_max;
180 const char *psz_tmp_path;
181
182 /* Lock for all following fields */
183 vlc_mutex_t lock;
184 vlc_cond_t wait;
185
186 /* */
187 bool b_paused;
188 mtime_t i_pause_date;
189
190 /* */
191 int i_rate;
192 int i_rate_source;
193 mtime_t i_rate_date;
194 mtime_t i_rate_delay;
195
196 /* */
197 mtime_t i_buffering_delay;
198
199 /* */
200 ts_storage_t *p_storage_r;
201 ts_storage_t *p_storage_w;
202
203 mtime_t i_cmd_delay;
204
205 } ts_thread_t;
206
207 struct es_out_id_t
208 {
209 es_out_id_t *p_es;
210 };
211
212 struct es_out_sys_t
213 {
214 input_thread_t *p_input;
215 es_out_t *p_out;
216
217 /* Configuration */
218 int64_t i_tmp_size_max; /* Maximal temporary file size in byte */
219 char *psz_tmp_path; /* Path for temporary files */
220
221 /* Lock for all following fields */
222 vlc_mutex_t lock;
223
224 /* */
225 bool b_delayed;
226 ts_thread_t *p_ts;
227
228 /* */
229 bool b_input_paused;
230 bool b_input_paused_source;
231 int i_input_rate;
232 int i_input_rate_source;
233
234 /* */
235 int i_es;
236 es_out_id_t **pp_es;
237 };
238
239 static es_out_id_t *Add ( es_out_t *, const es_format_t * );
240 static int Send ( es_out_t *, es_out_id_t *, block_t * );
241 static void Del ( es_out_t *, es_out_id_t * );
242 static int Control( es_out_t *, int i_query, va_list );
243 static void Destroy( es_out_t * );
244
245 static int TsStart( es_out_t * );
246 static void TsAutoStop( es_out_t * );
247
248 static void TsStop( ts_thread_t * );
249 static void TsPushCmd( ts_thread_t *, ts_cmd_t * );
250 static int TsPopCmdLocked( ts_thread_t *, ts_cmd_t *, bool b_flush );
251 static bool TsHasCmd( ts_thread_t * );
252 static bool TsIsUnused( ts_thread_t * );
253 static int TsChangePause( ts_thread_t *, bool b_source_paused, bool b_paused, mtime_t i_date );
254 static int TsChangeRate( ts_thread_t *, int i_src_rate, int i_rate );
255
256 static void *TsRun( void * );
257
258 static ts_storage_t *TsStorageNew( const char *psz_path, int64_t i_tmp_size_max );
259 static void TsStorageDelete( ts_storage_t * );
260 static void TsStoragePack( ts_storage_t *p_storage );
261 static bool TsStorageIsFull( ts_storage_t *, const ts_cmd_t *p_cmd );
262 static bool TsStorageIsEmpty( ts_storage_t * );
263 static void TsStoragePushCmd( ts_storage_t *, const ts_cmd_t *p_cmd, bool b_flush );
264 static void TsStoragePopCmd( ts_storage_t *p_storage, ts_cmd_t *p_cmd, bool b_flush );
265
266 static void CmdClean( ts_cmd_t * );
267 static void cmd_cleanup_routine( void *p ) { CmdClean( p ); }
268
269 static int CmdInitAdd ( ts_cmd_t *, es_out_id_t *, const es_format_t *, bool b_copy );
270 static void CmdInitSend ( ts_cmd_t *, es_out_id_t *, block_t * );
271 static int CmdInitDel ( ts_cmd_t *, es_out_id_t * );
272 static int CmdInitControl( ts_cmd_t *, int i_query, va_list, bool b_copy );
273
274 /* */
275 static void CmdCleanAdd ( ts_cmd_t * );
276 static void CmdCleanSend ( ts_cmd_t * );
277 static void CmdCleanControl( ts_cmd_t *p_cmd );
278
279 /* XXX these functions will take the destination es_out_t */
280 static void CmdExecuteAdd ( es_out_t *, ts_cmd_t * );
281 static int CmdExecuteSend ( es_out_t *, ts_cmd_t * );
282 static void CmdExecuteDel ( es_out_t *, ts_cmd_t * );
283 static int CmdExecuteControl( es_out_t *, ts_cmd_t * );
284
285 /* File helpers */
286 static char *GetTmpPath( char *psz_path );
287 static FILE *GetTmpFile( char **ppsz_file, const char *psz_path );
288
289 /*****************************************************************************
290 * input_EsOutTimeshiftNew:
291 *****************************************************************************/
292 es_out_t *input_EsOutTimeshiftNew( input_thread_t *p_input, es_out_t *p_next_out, int i_rate )
293 {
294 es_out_t *p_out = malloc( sizeof(*p_out) );
295 if( !p_out )
296 return NULL;
297
298 es_out_sys_t *p_sys = malloc( sizeof(*p_sys) );
299 if( !p_sys )
300 {
301 free( p_out );
302 return NULL;
303 }
304
305 /* */
306 p_out->pf_add = Add;
307 p_out->pf_send = Send;
308 p_out->pf_del = Del;
309 p_out->pf_control = Control;
310 p_out->pf_destroy = Destroy;
311 p_out->p_sys = p_sys;
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_ts = 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
335 char *psz_tmp_path = var_CreateGetNonEmptyString( p_input, "input-timeshift-path" );
336 p_sys->psz_tmp_path = GetTmpPath( psz_tmp_path );
337
338 msg_Dbg( p_input, "using timeshift granularity of %d MiB, in path '%s'",
339 (int)p_sys->i_tmp_size_max/(1024*1024), 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_ts );
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_ts, &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_ts, &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_ts, &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_ts ) )
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_ts, 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 pace 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_ts, 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 pace 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_GET_ES_OBJECTS_BY_ID:
586 case ES_OUT_SET_DELAY:
587 case ES_OUT_SET_RECORD_STATE:
588 assert(0);
589 return VLC_EGENERIC;
590
591 /* Pass-through control */
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 case ES_OUT_SET_EOS:
611 {
612 ts_cmd_t cmd;
613 if( CmdInitControl( &cmd, i_query, args, p_sys->b_delayed ) )
614 return VLC_EGENERIC;
615 if( p_sys->b_delayed )
616 {
617 TsPushCmd( p_sys->p_ts, &cmd );
618 return VLC_SUCCESS;
619 }
620 return CmdExecuteControl( p_sys->p_out, &cmd );
621 }
622
623 /* Special control when delayed */
624 case ES_OUT_GET_ES_STATE:
625 {
626 es_out_id_t *p_es = (es_out_id_t*)va_arg( args, es_out_id_t * );
627 bool *pb_enabled = (bool*)va_arg( args, bool* );
628
629 if( p_sys->b_delayed )
630 {
631 *pb_enabled = true;
632 return VLC_SUCCESS;
633 }
634 return es_out_Control( p_sys->p_out, ES_OUT_GET_ES_STATE, p_es->p_es, pb_enabled );
635 }
636 /* Special internal input control */
637 case ES_OUT_GET_EMPTY:
638 {
639 bool *pb_empty = (bool*)va_arg( args, bool* );
640 return ControlLockedGetEmpty( p_out, pb_empty );
641 }
642 case ES_OUT_GET_WAKE_UP: /* TODO ? */
643 {
644 mtime_t *pi_wakeup = (mtime_t*)va_arg( args, mtime_t* );
645 return ControlLockedGetWakeup( p_out, pi_wakeup );
646 }
647 case ES_OUT_GET_BUFFERING:
648 {
649 bool *pb_buffering = (bool *)va_arg( args, bool* );
650 return ControlLockedGetBuffering( p_out, pb_buffering );
651 }
652 case ES_OUT_SET_PAUSE_STATE:
653 {
654 const bool b_source_paused = (bool)va_arg( args, int );
655 const bool b_paused = (bool)va_arg( args, int );
656 const mtime_t i_date = (mtime_t) va_arg( args, mtime_t );
657
658 return ControlLockedSetPauseState( p_out, b_source_paused, b_paused, i_date );
659 }
660 case ES_OUT_SET_RATE:
661 {
662 const int i_src_rate = (int)va_arg( args, int );
663 const int i_rate = (int)va_arg( args, int );
664
665 return ControlLockedSetRate( p_out, i_src_rate, i_rate );
666 }
667 case ES_OUT_SET_TIME:
668 {
669 const mtime_t i_date = (mtime_t)va_arg( args, mtime_t );
670
671 return ControlLockedSetTime( p_out, i_date );
672 }
673 case ES_OUT_SET_FRAME_NEXT:
674 {
675 return ControlLockedSetFrameNext( p_out );
676 }
677 case ES_OUT_GET_PCR_SYSTEM:
678 {
679 if( p_sys->b_delayed )
680 return VLC_EGENERIC;
681
682 mtime_t *pi_system = (mtime_t*)va_arg( args, mtime_t * );
683 mtime_t *pi_delay = (mtime_t*)va_arg( args, mtime_t * );
684 return es_out_ControlGetPcrSystem( p_sys->p_out, pi_system, pi_delay );
685 }
686 case ES_OUT_MODIFY_PCR_SYSTEM:
687 {
688 const bool b_absolute = va_arg( args, int );
689 const mtime_t i_system = va_arg( args, mtime_t );
690
691 if( b_absolute && p_sys->b_delayed )
692 return VLC_EGENERIC;
693
694 return es_out_ControlModifyPcrSystem( p_sys->p_out, b_absolute, i_system );
695 }
696 case ES_OUT_GET_GROUP_FORCED:
697 {
698 int *pi_group = va_arg( args, int * );
699 return es_out_Control( p_sys->p_out, ES_OUT_GET_GROUP_FORCED, pi_group );
700 }
701
702
703 default:
704 msg_Err( p_sys->p_input, "Unknown es_out_Control query !" );
705 assert(0);
706 return VLC_EGENERIC;
707 }
708 }
709 static int Control( es_out_t *p_out, int i_query, va_list args )
710 {
711 es_out_sys_t *p_sys = p_out->p_sys;
712 int i_ret;
713
714 vlc_mutex_lock( &p_sys->lock );
715
716 TsAutoStop( p_out );
717
718 i_ret = ControlLocked( p_out, i_query, args );
719
720 vlc_mutex_unlock( &p_sys->lock );
721
722 return i_ret;
723 }
724
725 /*****************************************************************************
726 *
727 *****************************************************************************/
728 static void TsDestroy( ts_thread_t *p_ts )
729 {
730 vlc_cond_destroy( &p_ts->wait );
731 vlc_mutex_destroy( &p_ts->lock );
732 free( p_ts );
733 }
734 static int TsStart( es_out_t *p_out )
735 {
736 es_out_sys_t *p_sys = p_out->p_sys;
737 ts_thread_t *p_ts;
738
739 assert( !p_sys->b_delayed );
740
741 p_sys->p_ts = p_ts = calloc(1, sizeof(*p_ts));
742 if( !p_ts )
743 return VLC_EGENERIC;
744
745 p_ts->i_tmp_size_max = p_sys->i_tmp_size_max;
746 p_ts->psz_tmp_path = p_sys->psz_tmp_path;
747 p_ts->p_input = p_sys->p_input;
748 p_ts->p_out = p_sys->p_out;
749 vlc_mutex_init( &p_ts->lock );
750 vlc_cond_init( &p_ts->wait );
751 p_ts->b_paused = p_sys->b_input_paused && !p_sys->b_input_paused_source;
752 p_ts->i_pause_date = p_ts->b_paused ? mdate() : -1;
753 p_ts->i_rate_source = p_sys->i_input_rate_source;
754 p_ts->i_rate = p_sys->i_input_rate;
755 p_ts->i_rate_date = -1;
756 p_ts->i_rate_delay = 0;
757 p_ts->i_buffering_delay = 0;
758 p_ts->i_cmd_delay = 0;
759 p_ts->p_storage_r = NULL;
760 p_ts->p_storage_w = NULL;
761
762 p_sys->b_delayed = true;
763 if( vlc_clone( &p_ts->thread, TsRun, p_ts, VLC_THREAD_PRIORITY_INPUT ) )
764 {
765 msg_Err( p_sys->p_input, "cannot create timeshift thread" );
766
767 TsDestroy( p_ts );
768
769 p_sys->b_delayed = false;
770 return VLC_EGENERIC;
771 }
772
773 return VLC_SUCCESS;
774 }
775 static void TsAutoStop( es_out_t *p_out )
776 {
777 es_out_sys_t *p_sys = p_out->p_sys;
778
779 if( !p_sys->b_delayed || !TsIsUnused( p_sys->p_ts ) )
780 return;
781
782 msg_Warn( p_sys->p_input, "es out timeshift: auto stop" );
783 TsStop( p_sys->p_ts );
784
785 p_sys->b_delayed = false;
786 }
787 static void TsStop( ts_thread_t *p_ts )
788 {
789 vlc_cancel( p_ts->thread );
790 vlc_join( p_ts->thread, NULL );
791
792 vlc_mutex_lock( &p_ts->lock );
793 for( ;; )
794 {
795 ts_cmd_t cmd;
796
797 if( TsPopCmdLocked( p_ts, &cmd, true ) )
798 break;
799
800 CmdClean( &cmd );
801 }
802 assert( !p_ts->p_storage_r || !p_ts->p_storage_r->p_next );
803 if( p_ts->p_storage_r )
804 TsStorageDelete( p_ts->p_storage_r );
805 vlc_mutex_unlock( &p_ts->lock );
806
807 TsDestroy( p_ts );
808 }
809 static void TsPushCmd( ts_thread_t *p_ts, ts_cmd_t *p_cmd )
810 {
811 vlc_mutex_lock( &p_ts->lock );
812
813 if( !p_ts->p_storage_w || TsStorageIsFull( p_ts->p_storage_w, p_cmd ) )
814 {
815 ts_storage_t *p_storage = TsStorageNew( p_ts->psz_tmp_path, p_ts->i_tmp_size_max );
816
817 if( !p_storage )
818 {
819 CmdClean( p_cmd );
820 vlc_mutex_unlock( &p_ts->lock );
821 /* TODO warn the user (but only once) */
822 return;
823 }
824
825 if( !p_ts->p_storage_w )
826 {
827 p_ts->p_storage_r = p_ts->p_storage_w = p_storage;
828 }
829 else
830 {
831 TsStoragePack( p_ts->p_storage_w );
832 p_ts->p_storage_w->p_next = p_storage;
833 p_ts->p_storage_w = p_storage;
834 }
835 }
836
837 /* TODO return error and warn the user (but only once) */
838 TsStoragePushCmd( p_ts->p_storage_w, p_cmd, p_ts->p_storage_r == p_ts->p_storage_w );
839
840 vlc_cond_signal( &p_ts->wait );
841
842 vlc_mutex_unlock( &p_ts->lock );
843 }
844 static int TsPopCmdLocked( ts_thread_t *p_ts, ts_cmd_t *p_cmd, bool b_flush )
845 {
846 vlc_assert_locked( &p_ts->lock );
847
848 if( TsStorageIsEmpty( p_ts->p_storage_r ) )
849 return VLC_EGENERIC;
850
851 TsStoragePopCmd( p_ts->p_storage_r, p_cmd, b_flush );
852
853 while( p_ts->p_storage_r && TsStorageIsEmpty( p_ts->p_storage_r ) )
854 {
855 ts_storage_t *p_next = p_ts->p_storage_r->p_next;
856 if( !p_next )
857 break;
858
859 TsStorageDelete( p_ts->p_storage_r );
860 p_ts->p_storage_r = p_next;
861 }
862
863 return VLC_SUCCESS;
864 }
865 static bool TsHasCmd( ts_thread_t *p_ts )
866 {
867 bool b_cmd;
868
869 vlc_mutex_lock( &p_ts->lock );
870 b_cmd = TsStorageIsEmpty( p_ts->p_storage_r );
871 vlc_mutex_unlock( &p_ts->lock );
872
873 return b_cmd;
874 }
875 static bool TsIsUnused( ts_thread_t *p_ts )
876 {
877 bool b_unused;
878
879 vlc_mutex_lock( &p_ts->lock );
880 b_unused = !p_ts->b_paused &&
881 p_ts->i_rate == p_ts->i_rate_source &&
882 TsStorageIsEmpty( p_ts->p_storage_r );
883 vlc_mutex_unlock( &p_ts->lock );
884
885 return b_unused;
886 }
887 static int TsChangePause( ts_thread_t *p_ts, bool b_source_paused, bool b_paused, mtime_t i_date )
888 {
889 vlc_mutex_lock( &p_ts->lock );
890
891 int i_ret;
892 if( b_paused )
893 {
894 assert( !b_source_paused );
895 i_ret = es_out_SetPauseState( p_ts->p_out, true, true, i_date );
896 }
897 else
898 {
899 i_ret = es_out_SetPauseState( p_ts->p_out, false, false, i_date );
900 }
901
902 if( !i_ret )
903 {
904 if( !b_paused )
905 {
906 assert( p_ts->i_pause_date > 0 );
907
908 p_ts->i_cmd_delay += i_date - p_ts->i_pause_date;
909 }
910
911 p_ts->b_paused = b_paused;
912 p_ts->i_pause_date = i_date;
913
914 vlc_cond_signal( &p_ts->wait );
915 }
916 vlc_mutex_unlock( &p_ts->lock );
917 return i_ret;
918 }
919 static int TsChangeRate( ts_thread_t *p_ts, int i_src_rate, int i_rate )
920 {
921 int i_ret;
922
923 vlc_mutex_lock( &p_ts->lock );
924 p_ts->i_cmd_delay += p_ts->i_rate_delay;
925
926 p_ts->i_rate_date = -1;
927 p_ts->i_rate_delay = 0;
928 p_ts->i_rate = i_rate;
929 p_ts->i_rate_source = i_src_rate;
930
931 i_ret = es_out_SetRate( p_ts->p_out, i_rate, i_rate );
932 vlc_mutex_unlock( &p_ts->lock );
933
934 return i_ret;
935 }
936
937 static void *TsRun( void *p_data )
938 {
939 ts_thread_t *p_ts = p_data;
940 mtime_t i_buffering_date = -1;
941
942 for( ;; )
943 {
944 ts_cmd_t cmd;
945 mtime_t i_deadline;
946 bool b_buffering;
947
948 /* Pop a command to execute */
949 vlc_mutex_lock( &p_ts->lock );
950 mutex_cleanup_push( &p_ts->lock );
951
952 for( ;; )
953 {
954 const int canc = vlc_savecancel();
955 b_buffering = es_out_GetBuffering( p_ts->p_out );
956
957 if( ( !p_ts->b_paused || b_buffering ) && !TsPopCmdLocked( p_ts, &cmd, false ) )
958 {
959 vlc_restorecancel( canc );
960 break;
961 }
962 vlc_restorecancel( canc );
963
964 vlc_cond_wait( &p_ts->wait, &p_ts->lock );
965 }
966
967 if( b_buffering && i_buffering_date < 0 )
968 {
969 i_buffering_date = cmd.i_date;
970 }
971 else if( i_buffering_date > 0 )
972 {
973 p_ts->i_buffering_delay += i_buffering_date - cmd.i_date; /* It is < 0 */
974 if( b_buffering )
975 i_buffering_date = cmd.i_date;
976 else
977 i_buffering_date = -1;
978 }
979
980 if( p_ts->i_rate_date < 0 )
981 p_ts->i_rate_date = cmd.i_date;
982
983 p_ts->i_rate_delay = 0;
984 if( p_ts->i_rate_source != p_ts->i_rate )
985 {
986 const mtime_t i_duration = cmd.i_date - p_ts->i_rate_date;
987 p_ts->i_rate_delay = i_duration * p_ts->i_rate / p_ts->i_rate_source - i_duration;
988 }
989 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 )
990 {
991 const int canc = vlc_savecancel();
992
993 /* Auto reset to rate 1.0 */
994 msg_Warn( p_ts->p_input, "es out timeshift: auto reset rate to %d", p_ts->i_rate_source );
995
996 p_ts->i_cmd_delay = 0;
997 p_ts->i_buffering_delay = 0;
998
999 p_ts->i_rate_delay = 0;
1000 p_ts->i_rate_date = -1;
1001 p_ts->i_rate = p_ts->i_rate_source;
1002
1003 if( !es_out_SetRate( p_ts->p_out, p_ts->i_rate_source, p_ts->i_rate ) )
1004 {
1005 vlc_value_t val = { .i_int = p_ts->i_rate };
1006 /* Warn back input
1007 * FIXME it is perfectly safe BUT it is ugly as it may hide a
1008 * rate change requested by user */
1009 input_ControlPush( p_ts->p_input, INPUT_CONTROL_SET_RATE, &val );
1010 }
1011
1012 vlc_restorecancel( canc );
1013 }
1014 i_deadline = cmd.i_date + p_ts->i_cmd_delay + p_ts->i_rate_delay + p_ts->i_buffering_delay;
1015
1016 vlc_cleanup_run();
1017
1018 /* Regulate the speed of command processing to the same one than
1019 * reading */
1020 vlc_cleanup_push( cmd_cleanup_routine, &cmd );
1021
1022 mwait( i_deadline );
1023
1024 vlc_cleanup_pop();
1025
1026 /* Execute the command */
1027 const int canc = vlc_savecancel();
1028 switch( cmd.i_type )
1029 {
1030 case C_ADD:
1031 CmdExecuteAdd( p_ts->p_out, &cmd );
1032 CmdCleanAdd( &cmd );
1033 break;
1034 case C_SEND:
1035 CmdExecuteSend( p_ts->p_out, &cmd );
1036 CmdCleanSend( &cmd );
1037 break;
1038 case C_CONTROL:
1039 CmdExecuteControl( p_ts->p_out, &cmd );
1040 CmdCleanControl( &cmd );
1041 break;
1042 case C_DEL:
1043 CmdExecuteDel( p_ts->p_out, &cmd );
1044 break;
1045 default:
1046 assert(0);
1047 break;
1048 }
1049 vlc_restorecancel( canc );
1050 }
1051
1052 return NULL;
1053 }
1054
1055 /*****************************************************************************
1056 *
1057 *****************************************************************************/
1058 static ts_storage_t *TsStorageNew( const char *psz_tmp_path, int64_t i_tmp_size_max )
1059 {
1060 ts_storage_t *p_storage = calloc( 1, sizeof(ts_storage_t) );
1061 if( !p_storage )
1062 return NULL;
1063
1064 /* */
1065 p_storage->p_next = NULL;
1066
1067 /* */
1068 p_storage->i_file_max = i_tmp_size_max;
1069 p_storage->i_file_size = 0;
1070 p_storage->p_filew = GetTmpFile( &p_storage->psz_file, psz_tmp_path );
1071 if( p_storage->psz_file )
1072 p_storage->p_filer = vlc_fopen( p_storage->psz_file, "rb" );
1073
1074 /* */
1075 p_storage->i_cmd_w = 0;
1076 p_storage->i_cmd_r = 0;
1077 p_storage->i_cmd_max = 30000;
1078 p_storage->p_cmd = malloc( p_storage->i_cmd_max * sizeof(*p_storage->p_cmd) );
1079 //fprintf( stderr, "\nSTORAGE name=%s size=%d KiB\n", p_storage->psz_file, p_storage->i_cmd_max * sizeof(*p_storage->p_cmd) /1024 );
1080
1081 if( !p_storage->p_cmd || !p_storage->p_filew || !p_storage->p_filer )
1082 {
1083 TsStorageDelete( p_storage );
1084 return NULL;
1085 }
1086 return p_storage;
1087 }
1088 static void TsStorageDelete( ts_storage_t *p_storage )
1089 {
1090 while( p_storage->i_cmd_r < p_storage->i_cmd_w )
1091 {
1092 ts_cmd_t cmd;
1093
1094 TsStoragePopCmd( p_storage, &cmd, true );
1095
1096 CmdClean( &cmd );
1097 }
1098 free( p_storage->p_cmd );
1099
1100 if( p_storage->p_filer )
1101 fclose( p_storage->p_filer );
1102 if( p_storage->p_filew )
1103 fclose( p_storage->p_filew );
1104
1105 if( p_storage->psz_file )
1106 {
1107 vlc_unlink( p_storage->psz_file );
1108 free( p_storage->psz_file );
1109 }
1110
1111 free( p_storage );
1112 }
1113 static void TsStoragePack( ts_storage_t *p_storage )
1114 {
1115 /* Try to release a bit of memory */
1116 if( p_storage->i_cmd_w >= p_storage->i_cmd_max )
1117 return;
1118
1119 p_storage->i_cmd_max = __MAX( p_storage->i_cmd_w, 1 );
1120
1121 ts_cmd_t *p_new = realloc( p_storage->p_cmd, p_storage->i_cmd_max * sizeof(*p_storage->p_cmd) );
1122 if( p_new )
1123 p_storage->p_cmd = p_new;
1124 }
1125 static bool TsStorageIsFull( ts_storage_t *p_storage, const ts_cmd_t *p_cmd )
1126 {
1127 if( p_cmd && p_cmd->i_type == C_SEND && p_storage->i_cmd_w > 0 )
1128 {
1129 size_t i_size = sizeof(*p_cmd->u.send.p_block) + p_cmd->u.send.p_block->i_buffer;
1130
1131 if( p_storage->i_file_size + i_size >= p_storage->i_file_max )
1132 return true;
1133 }
1134 return p_storage->i_cmd_w >= p_storage->i_cmd_max;
1135 }
1136 static bool TsStorageIsEmpty( ts_storage_t *p_storage )
1137 {
1138 return !p_storage || p_storage->i_cmd_r >= p_storage->i_cmd_w;
1139 }
1140 static void TsStoragePushCmd( ts_storage_t *p_storage, const ts_cmd_t *p_cmd, bool b_flush )
1141 {
1142 ts_cmd_t cmd = *p_cmd;
1143
1144 assert( !TsStorageIsFull( p_storage, p_cmd ) );
1145
1146 if( cmd.i_type == C_SEND )
1147 {
1148 block_t *p_block = cmd.u.send.p_block;
1149
1150 cmd.u.send.p_block = NULL;
1151 cmd.u.send.i_offset = ftell( p_storage->p_filew );
1152
1153 if( fwrite( p_block, sizeof(*p_block), 1, p_storage->p_filew ) != 1 )
1154 {
1155 block_Release( p_block );
1156 return;
1157 }
1158 p_storage->i_file_size += sizeof(*p_block);
1159 if( p_block->i_buffer > 0 )
1160 {
1161 if( fwrite( p_block->p_buffer, p_block->i_buffer, 1, p_storage->p_filew ) != 1 )
1162 {
1163 block_Release( p_block );
1164 return;
1165 }
1166 }
1167 p_storage->i_file_size += p_block->i_buffer;
1168 block_Release( p_block );
1169
1170 if( b_flush )
1171 fflush( p_storage->p_filew );
1172 }
1173 p_storage->p_cmd[p_storage->i_cmd_w++] = cmd;
1174 }
1175 static void TsStoragePopCmd( ts_storage_t *p_storage, ts_cmd_t *p_cmd, bool b_flush )
1176 {
1177 assert( !TsStorageIsEmpty( p_storage ) );
1178
1179 *p_cmd = p_storage->p_cmd[p_storage->i_cmd_r++];
1180 if( p_cmd->i_type == C_SEND )
1181 {
1182 block_t block;
1183
1184 if( !b_flush &&
1185 !fseek( p_storage->p_filer, p_cmd->u.send.i_offset, SEEK_SET ) &&
1186 fread( &block, sizeof(block), 1, p_storage->p_filer ) == 1 )
1187 {
1188 block_t *p_block = block_Alloc( block.i_buffer );
1189 if( p_block )
1190 {
1191 p_block->i_dts = block.i_dts;
1192 p_block->i_pts = block.i_pts;
1193 p_block->i_flags = block.i_flags;
1194 p_block->i_length = block.i_length;
1195 p_block->i_nb_samples = block.i_nb_samples;
1196 p_block->i_buffer = fread( p_block->p_buffer, 1, block.i_buffer, p_storage->p_filer );
1197 }
1198 p_cmd->u.send.p_block = p_block;
1199 }
1200 else
1201 {
1202 //fprintf( stderr, "TsStoragePopCmd: %m\n" );
1203 p_cmd->u.send.p_block = block_Alloc( 1 );
1204 }
1205 }
1206 }
1207
1208 /*****************************************************************************
1209 *
1210 *****************************************************************************/
1211 static void CmdClean( ts_cmd_t *p_cmd )
1212 {
1213 switch( p_cmd->i_type )
1214 {
1215 case C_ADD:
1216 CmdCleanAdd( p_cmd );
1217 break;
1218 case C_SEND:
1219 CmdCleanSend( p_cmd );
1220 break;
1221 case C_CONTROL:
1222 CmdCleanControl( p_cmd );
1223 break;
1224 case C_DEL:
1225 break;
1226 default:
1227 assert(0);
1228 break;
1229 }
1230 }
1231
1232 static int CmdInitAdd( ts_cmd_t *p_cmd, es_out_id_t *p_es, const es_format_t *p_fmt, bool b_copy )
1233 {
1234 p_cmd->i_type = C_ADD;
1235 p_cmd->i_date = mdate();
1236 p_cmd->u.add.p_es = p_es;
1237 if( b_copy )
1238 {
1239 p_cmd->u.add.p_fmt = malloc( sizeof(*p_fmt) );
1240 if( !p_cmd->u.add.p_fmt )
1241 return VLC_EGENERIC;
1242 es_format_Copy( p_cmd->u.add.p_fmt, p_fmt );
1243 }
1244 else
1245 {
1246 p_cmd->u.add.p_fmt = (es_format_t*)p_fmt;
1247 }
1248 return VLC_SUCCESS;
1249 }
1250 static void CmdExecuteAdd( es_out_t *p_out, ts_cmd_t *p_cmd )
1251 {
1252 p_cmd->u.add.p_es->p_es = es_out_Add( p_out, p_cmd->u.add.p_fmt );
1253 }
1254 static void CmdCleanAdd( ts_cmd_t *p_cmd )
1255 {
1256 es_format_Clean( p_cmd->u.add.p_fmt );
1257 free( p_cmd->u.add.p_fmt );
1258 }
1259
1260 static void CmdInitSend( ts_cmd_t *p_cmd, es_out_id_t *p_es, block_t *p_block )
1261 {
1262 p_cmd->i_type = C_SEND;
1263 p_cmd->i_date = mdate();
1264 p_cmd->u.send.p_es = p_es;
1265 p_cmd->u.send.p_block = p_block;
1266 }
1267 static int CmdExecuteSend( es_out_t *p_out, ts_cmd_t *p_cmd )
1268 {
1269 block_t *p_block = p_cmd->u.send.p_block;
1270
1271 p_cmd->u.send.p_block = NULL;
1272
1273 if( p_block )
1274 {
1275 if( p_cmd->u.send.p_es->p_es )
1276 return es_out_Send( p_out, p_cmd->u.send.p_es->p_es, p_block );
1277 block_Release( p_block );
1278 }
1279 return VLC_EGENERIC;
1280 }
1281 static void CmdCleanSend( ts_cmd_t *p_cmd )
1282 {
1283 if( p_cmd->u.send.p_block )
1284 block_Release( p_cmd->u.send.p_block );
1285 }
1286
1287 static int CmdInitDel( ts_cmd_t *p_cmd, es_out_id_t *p_es )
1288 {
1289 p_cmd->i_type = C_DEL;
1290 p_cmd->i_date = mdate();
1291 p_cmd->u.del.p_es = p_es;
1292 return VLC_SUCCESS;
1293 }
1294 static void CmdExecuteDel( es_out_t *p_out, ts_cmd_t *p_cmd )
1295 {
1296 if( p_cmd->u.del.p_es->p_es )
1297 es_out_Del( p_out, p_cmd->u.del.p_es->p_es );
1298 free( p_cmd->u.del.p_es );
1299 }
1300
1301 static int CmdInitControl( ts_cmd_t *p_cmd, int i_query, va_list args, bool b_copy )
1302 {
1303 p_cmd->i_type = C_CONTROL;
1304 p_cmd->i_date = mdate();
1305 p_cmd->u.control.i_query = i_query;
1306
1307 switch( i_query )
1308 {
1309 /* Pass-through control */
1310 case ES_OUT_SET_MODE: /* arg1= int */
1311 case ES_OUT_SET_GROUP: /* arg1= int */
1312 case ES_OUT_DEL_GROUP: /* arg1=int i_group */
1313 p_cmd->u.control.u.i_int = (int)va_arg( args, int );
1314 break;
1315
1316 case ES_OUT_SET_PCR: /* arg1=int64_t i_pcr(microsecond!) (using default group 0)*/
1317 case ES_OUT_SET_NEXT_DISPLAY_TIME: /* arg1=int64_t i_pts(microsecond) */
1318 p_cmd->u.control.u.i_i64 = (int64_t)va_arg( args, int64_t );
1319 break;
1320
1321 case ES_OUT_SET_GROUP_PCR: /* arg1= int i_group, arg2=int64_t i_pcr(microsecond!)*/
1322 p_cmd->u.control.u.int_i64.i_int = (int)va_arg( args, int );
1323 p_cmd->u.control.u.int_i64.i_i64 = (int64_t)va_arg( args, int64_t );
1324 break;
1325
1326 case ES_OUT_SET_ES_SCRAMBLED_STATE:
1327 p_cmd->u.control.u.es_bool.p_es = (es_out_id_t*)va_arg( args, es_out_id_t * );
1328 p_cmd->u.control.u.es_bool.b_bool = (bool)va_arg( args, int );
1329 break;
1330
1331 case ES_OUT_RESET_PCR: /* no arg */
1332 case ES_OUT_SET_EOS:
1333 break;
1334
1335 case ES_OUT_SET_META: /* arg1=const vlc_meta_t* */
1336 case ES_OUT_SET_GROUP_META: /* arg1=int i_group arg2=const vlc_meta_t* */
1337 {
1338 if( i_query == ES_OUT_SET_GROUP_META )
1339 p_cmd->u.control.u.int_meta.i_int = (int)va_arg( args, int );
1340 const vlc_meta_t *p_meta = va_arg( args, const vlc_meta_t * );
1341
1342 if( b_copy )
1343 {
1344 p_cmd->u.control.u.int_meta.p_meta = vlc_meta_New();
1345 if( !p_cmd->u.control.u.int_meta.p_meta )
1346 return VLC_EGENERIC;
1347 vlc_meta_Merge( p_cmd->u.control.u.int_meta.p_meta, p_meta );
1348 }
1349 else
1350 {
1351 /* The cast is only needed to avoid warning */
1352 p_cmd->u.control.u.int_meta.p_meta = (vlc_meta_t*)p_meta;
1353 }
1354 break;
1355 }
1356
1357 case ES_OUT_SET_GROUP_EPG: /* arg1=int i_group arg2=const vlc_epg_t* */
1358 {
1359 p_cmd->u.control.u.int_epg.i_int = (int)va_arg( args, int );
1360 const vlc_epg_t *p_epg = va_arg( args, const vlc_epg_t * );
1361
1362 if( b_copy )
1363 {
1364 p_cmd->u.control.u.int_epg.p_epg = vlc_epg_New( p_epg->psz_name );
1365 if( !p_cmd->u.control.u.int_epg.p_epg )
1366 return VLC_EGENERIC;
1367 for( int i = 0; i < p_epg->i_event; i++ )
1368 {
1369 vlc_epg_event_t *p_evt = p_epg->pp_event[i];
1370
1371 vlc_epg_AddEvent( p_cmd->u.control.u.int_epg.p_epg,
1372 p_evt->i_start, p_evt->i_duration,
1373 p_evt->psz_name,
1374 p_evt->psz_short_description,
1375 p_evt->psz_description, 0 );
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 mtime_t i_pts_jitter = (mtime_t)va_arg( args, mtime_t );
1433 int i_cr_average = (int)va_arg( args, int );
1434
1435 p_cmd->u.control.u.jitter.i_pts_delay = i_pts_delay;
1436 p_cmd->u.control.u.jitter.i_pts_jitter = i_pts_jitter;
1437 p_cmd->u.control.u.jitter.i_cr_average = i_cr_average;
1438 break;
1439 }
1440
1441 default:
1442 assert(0);
1443 return VLC_EGENERIC;
1444 }
1445
1446 return VLC_SUCCESS;
1447 }
1448 static int CmdExecuteControl( es_out_t *p_out, ts_cmd_t *p_cmd )
1449 {
1450 const int i_query = p_cmd->u.control.i_query;
1451
1452 switch( i_query )
1453 {
1454 /* Pass-through control */
1455 case ES_OUT_SET_MODE: /* arg1= int */
1456 case ES_OUT_SET_GROUP: /* arg1= int */
1457 case ES_OUT_DEL_GROUP: /* arg1=int i_group */
1458 return es_out_Control( p_out, i_query, p_cmd->u.control.u.i_int );
1459
1460 case ES_OUT_SET_PCR: /* arg1=int64_t i_pcr(microsecond!) (using default group 0)*/
1461 case ES_OUT_SET_NEXT_DISPLAY_TIME: /* arg1=int64_t i_pts(microsecond) */
1462 return es_out_Control( p_out, i_query, p_cmd->u.control.u.i_i64 );
1463
1464 case ES_OUT_SET_GROUP_PCR: /* arg1= int i_group, arg2=int64_t i_pcr(microsecond!)*/
1465 return es_out_Control( p_out, i_query, p_cmd->u.control.u.int_i64.i_int,
1466 p_cmd->u.control.u.int_i64.i_i64 );
1467
1468 case ES_OUT_RESET_PCR: /* no arg */
1469 case ES_OUT_SET_EOS:
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_pts_jitter,
1508 p_cmd->u.control.u.jitter.i_cr_average );
1509
1510 default:
1511 assert(0);
1512 return VLC_EGENERIC;
1513 }
1514 }
1515 static void CmdCleanControl( ts_cmd_t *p_cmd )
1516 {
1517 if( ( p_cmd->u.control.i_query == ES_OUT_SET_GROUP_META ||
1518 p_cmd->u.control.i_query == ES_OUT_SET_META ) &&
1519 p_cmd->u.control.u.int_meta.p_meta )
1520 {
1521 vlc_meta_Delete( p_cmd->u.control.u.int_meta.p_meta );
1522 }
1523 else if( p_cmd->u.control.i_query == ES_OUT_SET_GROUP_EPG &&
1524 p_cmd->u.control.u.int_epg.p_epg )
1525 {
1526 vlc_epg_Delete( p_cmd->u.control.u.int_epg.p_epg );
1527 }
1528 else if( p_cmd->u.control.i_query == ES_OUT_SET_ES_FMT &&
1529 p_cmd->u.control.u.es_fmt.p_fmt )
1530 {
1531 es_format_Clean( p_cmd->u.control.u.es_fmt.p_fmt );
1532 free( p_cmd->u.control.u.es_fmt.p_fmt );
1533 }
1534 }
1535
1536
1537 /*****************************************************************************
1538 * GetTmpFile/Path:
1539 *****************************************************************************/
1540 static char *GetTmpPath( char *psz_path )
1541 {
1542 if( psz_path && *psz_path )
1543 {
1544 /* Make sure that the path exists and is a directory */
1545 struct stat s;
1546 const int i_ret = vlc_stat( psz_path, &s );
1547
1548 if( i_ret < 0 && !vlc_mkdir( psz_path, 0600 ) )
1549 return psz_path;
1550 else if( i_ret == 0 && ( s.st_mode & S_IFDIR ) )
1551 return psz_path;
1552 }
1553 free( psz_path );
1554
1555 /* Create a suitable path */
1556 #if defined (WIN32)
1557 const DWORD dwCount = GetTempPathW( 0, NULL );
1558 wchar_t *psw_path = calloc( dwCount + 1, sizeof(wchar_t) );
1559 if( psw_path )
1560 {
1561 if( GetTempPathW( dwCount + 1, psw_path ) <= 0 )
1562 {
1563 free( psw_path );
1564
1565 psw_path = _wgetcwd( NULL, 0 );
1566 }
1567 }
1568
1569 psz_path = NULL;
1570 if( psw_path )
1571 {
1572 psz_path = FromWide( psw_path );
1573 while( psz_path && *psz_path && psz_path[strlen( psz_path ) - 1] == '\\' )
1574 psz_path[strlen( psz_path ) - 1] = '\0';
1575
1576 free( psw_path );
1577 }
1578
1579 if( !psz_path || *psz_path == '\0' )
1580 {
1581 free( psz_path );
1582 return strdup( "C:" );
1583 }
1584 #else
1585 psz_path = strdup( DIR_SEP"tmp" );
1586 #endif
1587
1588 return psz_path;
1589 }
1590
1591 static FILE *GetTmpFile( char **ppsz_file, const char *psz_path )
1592 {
1593 char *psz_name;
1594 int fd;
1595 FILE *f;
1596
1597 /* */
1598 *ppsz_file = NULL;
1599 if( asprintf( &psz_name, "%s/vlc-timeshift.XXXXXX", psz_path ) < 0 )
1600 return NULL;
1601
1602 /* */
1603 fd = vlc_mkstemp( psz_name );
1604 *ppsz_file = psz_name;
1605
1606 if( fd < 0 )
1607 return NULL;
1608
1609 /* */
1610 f = fdopen( fd, "w+b" );
1611 if( !f )
1612 close( fd );
1613
1614 return f;
1615 }
1616
VLC Development tree mirror