]> git.sesse.net Git - vlc/blobdiff - modules/access_filter/timeshift.c
Removes trailing spaces. Removes tabs.
[vlc] / modules / access_filter / timeshift.c
index 2d898a5f74868d6f2f675eee3a5de2e9610351d5..9693b89d38779dd149f6cac8737dfb6c34ce154d 100644 (file)
 /*****************************************************************************
  * Preamble
  *****************************************************************************/
-#include <stdlib.h>
+#include <vlc/vlc.h>
+
 
 #include <errno.h>
 
-#include <vlc/vlc.h>
-#include <vlc/input.h>
+#include <vlc_access.h>
+#include <vlc_charset.h>
+#include <vlc_input.h>
+
 #include <unistd.h>
 
 /*****************************************************************************
@@ -40,11 +43,15 @@ static int  Open ( vlc_object_t * );
 static void Close( vlc_object_t * );
 
 #define GRANULARITY_TEXT N_("Timeshift granularity")
-#define GRANULARITY_LONGTEXT N_( "Size of the temporary files use to store " \
-  "the timeshifted stream." )
+/// \bug [String] typo
+#define GRANULARITY_LONGTEXT N_( "This is the size of the temporary files " \
+  "that will be used to store the timeshifted streams." )
 #define DIR_TEXT N_("Timeshift directory")
 #define DIR_LONGTEXT N_( "Directory used to store the timeshift temporary " \
   "files." )
+#define FORCE_TEXT N_("Force use of the timeshift module")
+#define FORCE_LONGTEXT N_("Force use of the timeshift module even if the " \
+  "access declares that it can control pace or pause." )
 
 vlc_module_begin();
     set_shortname( _("Timeshift") );
@@ -58,6 +65,8 @@ vlc_module_begin();
     add_integer( "timeshift-granularity", 50, NULL, GRANULARITY_TEXT,
                  GRANULARITY_LONGTEXT, VLC_TRUE );
     add_directory( "timeshift-dir", 0, 0, DIR_TEXT, DIR_LONGTEXT, VLC_FALSE );
+    add_bool( "timeshift-force", VLC_FALSE, NULL, FORCE_TEXT, FORCE_LONGTEXT,
+              VLC_FALSE );
 vlc_module_end();
 
 /*****************************************************************************
@@ -98,6 +107,8 @@ struct access_sys_t
 
     char *psz_filename_base;
     char *psz_filename;
+
+    int64_t i_data;
 };
 
 /*****************************************************************************
@@ -110,17 +121,26 @@ static int Open( vlc_object_t *p_this )
     access_sys_t *p_sys;
     vlc_bool_t b_bool;
 
-    /* Only work with not pace controled access */
-    if( access2_Control( p_src, ACCESS_CAN_CONTROL_PACE, &b_bool ) || b_bool )
+    var_Create( p_access, "timeshift-force", VLC_VAR_BOOL|VLC_VAR_DOINHERIT );
+    if( var_GetBool( p_access, "timeshift-force" ) )
     {
-        msg_Dbg( p_src, "ACCESS_CAN_CONTROL_PACE" );
-        return VLC_EGENERIC;
+        msg_Dbg( p_access, "Forcing use of timeshift even if access can control pace or pause" );
     }
-    /* Refuse access that can be paused */
-    if( access2_Control( p_src, ACCESS_CAN_PAUSE, &b_bool ) || b_bool )
+    else
     {
-        msg_Dbg( p_src, "ACCESS_CAN_PAUSE: timeshift useless" );
-        return VLC_EGENERIC;
+        /* Only work with not pace controled access */
+        if( access2_Control( p_src, ACCESS_CAN_CONTROL_PACE, &b_bool ) ||
+            b_bool )
+        {
+            msg_Dbg( p_src, "ACCESS_CAN_CONTROL_PACE: timeshift useless" );
+            return VLC_EGENERIC;
+        }
+        /* Refuse access that can be paused */
+        if( access2_Control( p_src, ACCESS_CAN_PAUSE, &b_bool ) || b_bool )
+        {
+            msg_Dbg( p_src, "ACCESS_CAN_PAUSE: timeshift useless" );
+            return VLC_EGENERIC;
+        }
     }
 
     /* */
@@ -136,6 +156,7 @@ static int Open( vlc_object_t *p_this )
     p_sys->p_fifo = block_FifoNew( p_access );
     p_sys->i_write_size = 0;
     p_sys->i_files = 0;
+    p_sys->i_data = 0;
 
     p_sys->p_read_list = NULL;
     p_sys->pp_read_last = &p_sys->p_read_list;
@@ -209,17 +230,31 @@ static void Close( vlc_object_t *p_this )
 static block_t *Block( access_t *p_access )
 {
     access_sys_t *p_sys = p_access->p_sys;
-    block_t *p_block;
+    access_t *p_src = p_access->p_source;
+    block_t *p_block = NULL;
 
-    if( p_access->b_die )
+    /* Update info (we probably ought to be time caching that as well) */
+    if( p_src->info.i_update & INPUT_UPDATE_META )
     {
-        p_access->info.b_eof = VLC_TRUE;
-        return NULL;
+        p_src->info.i_update &= ~INPUT_UPDATE_META;
+        p_access->info.i_update |= INPUT_UPDATE_META;
     }
 
-    p_block = block_FifoGet( p_sys->p_fifo );
-    //p_access->info.i_size -= p_block->i_buffer;
-    return p_block;
+    /* Get data from timeshift fifo */
+    if( !p_access->info.b_eof )
+        p_block = block_FifoGet( p_sys->p_fifo );
+
+    if( p_block && !p_block->i_buffer ) /* Used to signal EOF */
+    { block_Release( p_block ); p_block = 0; }
+
+    if( p_block )
+    {
+        p_sys->i_data -= p_block->i_buffer;
+        return p_block;
+    }
+
+    p_access->info.b_eof = p_src->info.b_eof;
+    return NULL;
 }
 
 /*****************************************************************************
@@ -229,23 +264,14 @@ static void Thread( access_t *p_access )
 {
     access_sys_t *p_sys = p_access->p_sys;
     access_t     *p_src = p_access->p_source;
-    int i;
+    block_t      *p_block;
 
     while( !p_access->b_die )
     {
-        block_t *p_block;
-
         /* Get a new block from the source */
         if( p_src->pf_block )
         {
             p_block = p_src->pf_block( p_src );
-
-            if( p_block == NULL )
-            {
-                if( p_src->info.b_eof ) break;
-                msleep( 1000 );
-                continue;
-            }
         }
         else
         {
@@ -254,15 +280,22 @@ static void Thread( access_t *p_access )
             p_block->i_buffer =
                 p_src->pf_read( p_src, p_block->p_buffer, 2048 );
 
-            if( p_block->i_buffer < 0 )
+            if( p_block->i_buffer <= 0 )
             {
-                block_Release( p_block );
-                if( p_block->i_buffer == 0 ) break;
-                msleep( 1000 );
-                continue;
+              block_Release( p_block );
+              p_block = NULL;
             }
         }
 
+        if( p_block == NULL )
+        {
+          if( p_src->info.b_eof ) break;
+          msleep( 10000 );
+          continue;
+        }
+
+        p_sys->i_data += p_block->i_buffer;
+
         /* Write block */
         if( !p_sys->p_write_list && !p_sys->p_read_list &&
             p_sys->p_fifo->i_size < TIMESHIFT_FIFO_MAX )
@@ -270,11 +303,6 @@ static void Thread( access_t *p_access )
             /* If there isn't too much timeshifted data,
              * write directly to FIFO */
             block_FifoPut( p_sys->p_fifo, p_block );
-
-            //p_access->info.i_size += p_block->i_buffer;
-            //p_access->info.i_update |= INPUT_UPDATE_SIZE;
-
-            /* Nothing else to do */
             continue;
         }
 
@@ -287,20 +315,34 @@ static void Thread( access_t *p_access )
         {
             p_block = ReadBlockFromFile( p_access );
             if( !p_block ) break;
+
             block_FifoPut( p_sys->p_fifo, p_block );
         }
     }
 
-    msg_Dbg( p_access, "timeshift: EOF" );
+    msg_Dbg( p_access, "timeshift: no more input data" );
 
-    /* Send dummy packet to avoid deadlock in TShiftBlock */
-    for( i = 0; i < 2; i++ )
+    while( !p_access->b_die &&
+           (p_sys->p_read_list || p_sys->p_fifo->i_size) )
     {
-        block_t *p_dummy = block_New( p_access, 128 );
-        p_dummy->i_flags |= BLOCK_FLAG_DISCONTINUITY;
-        memset( p_dummy->p_buffer, 0, p_dummy->i_buffer );
-        block_FifoPut( p_sys->p_fifo, p_dummy );
+        /* Read from file to fill up the fifo */
+        while( p_sys->p_fifo->i_size < TIMESHIFT_FIFO_MIN &&
+               !p_access->b_die && p_sys->p_read_list )
+        {
+            p_block = ReadBlockFromFile( p_access );
+            if( !p_block ) break;
+
+            block_FifoPut( p_sys->p_fifo, p_block );
+        }
+
+        msleep( 100000 );
     }
+
+    msg_Dbg( p_access, "timeshift: EOF" );
+    p_src->info.b_eof = VLC_TRUE;
+
+    /* Send dummy packet to avoid deadlock in Block() */
+    block_FifoPut( p_sys->p_fifo, block_New( p_access, 0 ) );
 }
 
 /*****************************************************************************
@@ -473,53 +515,34 @@ static int Seek( access_t *p_access, int64_t i_pos )
  *****************************************************************************/
 static int Control( access_t *p_access, int i_query, va_list args )
 {
-    access_t     *p_src = p_access->p_source;
-
     vlc_bool_t   *pb_bool;
     int          *pi_int;
-    int64_t      *pi_64;
 
     switch( i_query )
     {
-        case ACCESS_CAN_SEEK:
-        case ACCESS_CAN_FASTSEEK:
-            pb_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t* );
-            *pb_bool = VLC_TRUE;
-            break;
-
-        case ACCESS_CAN_CONTROL_PACE:   /* Not really true */
-        case ACCESS_CAN_PAUSE:
-            pb_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t* );
-            *pb_bool = VLC_TRUE;
-            break;
-
-        case ACCESS_GET_MTU:
-            pi_int = (int*)va_arg( args, int * );
-            *pi_int = 0;
-            break;
-
-        case ACCESS_GET_PTS_DELAY:
-            pi_64 = (int64_t*)va_arg( args, int64_t * );
-            return access2_Control( p_src, ACCESS_GET_PTS_DELAY, pi_64 );
-
-        case ACCESS_SET_PAUSE_STATE:
-            return VLC_SUCCESS;
-
-        case ACCESS_GET_TITLE_INFO:
-        case ACCESS_SET_TITLE:
-        case ACCESS_SET_SEEKPOINT:
-        case ACCESS_GET_META:
-            return VLC_EGENERIC;
-
-        case ACCESS_SET_PRIVATE_ID_STATE:
-        case ACCESS_GET_PRIVATE_ID_STATE:
-        case ACCESS_SET_PRIVATE_ID_CA:
-            return access2_vaControl( p_src, i_query, args );
-
-        default:
-            msg_Warn( p_access, "unimplemented query in control" );
-            return VLC_EGENERIC;
-
+    case ACCESS_CAN_SEEK:
+    case ACCESS_CAN_FASTSEEK:
+        pb_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t* );
+        *pb_bool = VLC_TRUE;
+        break;
+
+    case ACCESS_CAN_CONTROL_PACE:   /* Not really true */
+    case ACCESS_CAN_PAUSE:
+        pb_bool = (vlc_bool_t*)va_arg( args, vlc_bool_t* );
+        *pb_bool = VLC_TRUE;
+        break;
+
+    case ACCESS_GET_MTU:
+        pi_int = (int*)va_arg( args, int * );
+        *pi_int = 0;
+        break;
+
+    case ACCESS_SET_PAUSE_STATE:
+        break;
+
+    /* Forward everything else to the source access */
+    default:
+        return access2_vaControl( p_access->p_source, i_query, args );
     }
     return VLC_SUCCESS;
 }
@@ -544,21 +567,25 @@ static char *GetTmpFilePath( access_t *p_access )
     if( psz_dir == NULL )
     {
 #ifdef WIN32
-        char psz_local_dir[MAX_PATH];
-        int i_size;
+        DWORD ret = GetTempPathW (0, NULL);
+        wchar_t wdir[ret + 3]; // can at least old "C:" + nul
+        const wchar_t *pwdir = wdir;
+        wchar_t *pwdir_free = NULL;
 
-        i_size = GetTempPath( MAX_PATH, psz_local_dir );
-        if( i_size <= 0 || i_size > MAX_PATH )
+        if (GetTempPathW (ret + 1, wdir) == 0)
         {
-            if( !getcwd( psz_local_dir, MAX_PATH ) )
-                strcpy( psz_local_dir, "C:" );
+            pwdir_free = pwdir = _wgetcwd (NULL, 0);
+            if (pwdir == NULL)
+                pwdir = L"C:";
         }
 
-        psz_dir = FromLocaleDup( MAX_PATH + 1 );
+        psz_dir = FromWide (pwdir);
+        if (pwdir_free != NULL)
+            free (pwdir_free);
 
-        /* remove last \\ if any */
-        if( psz_dir[strlen(psz_dir)-1] == '\\' )
-            psz_dir[strlen(psz_dir)-1] = '\0';
+        /* remove trailing antislash if any */
+        if (psz_dir[strlen (psz_dir) - 1] == '\\')
+            psz_dir[strlen (psz_dir) - 1] = '\0';
 #else
         psz_dir = strdup( "/tmp" );
 #endif