]> git.sesse.net Git - vlc/blobdiff - src/input/stream.c
Cosmetics.
[vlc] / src / input / stream.c
index 8c669d0ddeab0104ef6a93b465d562aeeef6bc87..4688ceaf4d10e7cd60f6c65a0a957392b270a297 100644 (file)
 # include "config.h"
 #endif
 
+#include <dirent.h>
+#include <assert.h>
+
 #include <vlc_common.h>
+#include <vlc_strings.h>
+#include <vlc_osd.h>
+#include <vlc_charset.h>
 
-#include <assert.h>
+#include <libvlc.h>
+
+#include "access.h"
+#include "stream.h"
 
 #include "input_internal.h"
 
@@ -71,7 +80,6 @@
  *  We get blocks and put them in the linked list.
  *  We release blocks once the total size is bigger than CACHE_BLOCK_SIZE
  */
-#define STREAM_DATA_WAIT 40000       /* Time between before a pf_block retry */
 
 /* Method2: A bit more complex, for pf_read
  *  - We use ring buffers, only one if unseekable, all if seekable
@@ -109,9 +117,9 @@ typedef struct
 
 typedef enum stream_read_method_t
 {
-    Immediate,
-    Block,
-    Stream
+    STREAM_METHOD_IMMEDIATE,
+    STREAM_METHOD_BLOCK,
+    STREAM_METHOD_STREAM
 } stream_read_method_t;
 
 struct stream_sys_t
@@ -126,7 +134,7 @@ struct stream_sys_t
     struct
     {
         int64_t i_start;        /* Offset of block for p_first */
-        int     i_offset;       /* Offset for data in p_current */
+        int64_t i_offset;       /* Offset for data in p_current */
         block_t *p_current;     /* Current block */
 
         int     i_size;         /* Total amount of data in the list */
@@ -159,7 +167,7 @@ struct stream_sys_t
     } immediate;
 
     /* Peek temporary buffer */
-    int     i_peek;
+    unsigned int i_peek;
     uint8_t *p_peek;
 
     /* Stat for both method */
@@ -186,25 +194,34 @@ struct stream_sys_t
 
     /* Preparse mode ? */
     bool      b_quick;
+
+    /* */
+    struct
+    {
+        bool b_active;
+
+        FILE *f;        /* TODO it could be replaced by access_output_t one day */
+        bool b_error;
+    } record;
 };
 
 /* Method 1: */
-static int  AStreamReadBlock( stream_t *s, void *p_read, int i_read );
-static int  AStreamPeekBlock( stream_t *s, const uint8_t **p_peek, int i_read );
+static int  AStreamReadBlock( stream_t *s, void *p_read, unsigned int i_read );
+static int  AStreamPeekBlock( stream_t *s, const uint8_t **p_peek, unsigned int i_read );
 static int  AStreamSeekBlock( stream_t *s, int64_t i_pos );
 static void AStreamPrebufferBlock( stream_t *s );
 static block_t *AReadBlock( stream_t *s, bool *pb_eof );
 
 /* Method 2 */
-static int  AStreamReadStream( stream_t *s, void *p_read, int i_read );
-static int  AStreamPeekStream( stream_t *s, const uint8_t **pp_peek, int i_read );
+static int  AStreamReadStream( stream_t *s, void *p_read, unsigned int i_read );
+static int  AStreamPeekStream( stream_t *s, const uint8_t **pp_peek, unsigned int i_read );
 static int  AStreamSeekStream( stream_t *s, int64_t i_pos );
 static void AStreamPrebufferStream( stream_t *s );
-static int  AReadStream( stream_t *s, void *p_read, int i_read );
+static int  AReadStream( stream_t *s, void *p_read, unsigned int i_read );
 
 /* Method 3 */
-static int  AStreamReadImmediate( stream_t *s, void *p_read, int i_read );
-static int  AStreamPeekImmediate( stream_t *s, const uint8_t **pp_peek, int i_read );
+static int  AStreamReadImmediate( stream_t *s, void *p_read, unsigned int i_read );
+static int  AStreamPeekImmediate( stream_t *s, const uint8_t **pp_peek, unsigned int i_read );
 static int  AStreamSeekImmediate( stream_t *s, int64_t i_pos );
 
 /* Common */
@@ -212,6 +229,8 @@ static int AStreamControl( stream_t *s, int i_query, va_list );
 static void AStreamDestroy( stream_t *s );
 static void UStreamDestroy( stream_t *s );
 static int  ASeek( stream_t *s, int64_t i_pos );
+static int  ARecordSetState( stream_t *s, bool b_record, const char *psz_extension );
+static void ARecordWrite( stream_t *s, const uint8_t *p_buffer, size_t i_buffer );
 
 /****************************************************************************
  * Method 3 helpers:
@@ -244,6 +263,15 @@ static inline uint8_t * stream_buffer( stream_t *s )
     return s->p_sys->immediate.p_buffer;
 }
 
+/****************************************************************************
+ * stream_CommonNew: create an empty stream structure
+ ****************************************************************************/
+stream_t *stream_CommonNew( vlc_object_t *p_obj )
+{
+    return (stream_t *)vlc_custom_create( p_obj, sizeof(stream_t),
+                                          VLC_OBJECT_GENERIC, "stream" );
+}
+
 /****************************************************************************
  * stream_UrlNew: create a stream from a access
  ****************************************************************************/
@@ -254,11 +282,12 @@ stream_t *__stream_UrlNew( vlc_object_t *p_parent, const char *psz_url )
     access_t *p_access;
     stream_t *p_res;
 
-    if( !psz_url ) return 0;
+    if( !psz_url )
+        return NULL;
 
-    char psz_dup[strlen (psz_url) + 1];
-    strcpy (psz_dup, psz_url);;
-    MRLSplit( psz_dup, &psz_access, &psz_demux, &psz_path );
+    char psz_dup[strlen( psz_url ) + 1];
+    strcpy( psz_dup, psz_url );
+    input_SplitMRL( &psz_access, &psz_demux, &psz_path, psz_dup );
 
     /* Now try a real access */
     p_access = access_New( p_parent, psz_access, psz_demux, psz_path );
@@ -281,11 +310,19 @@ stream_t *__stream_UrlNew( vlc_object_t *p_parent, const char *psz_url )
 
 stream_t *stream_AccessNew( access_t *p_access, bool b_quick )
 {
-    stream_t *s = vlc_stream_create( VLC_OBJECT(p_access) );
+    stream_t *s = stream_CommonNew( VLC_OBJECT(p_access) );
     stream_sys_t *p_sys;
     char *psz_list = NULL;
 
-    if( !s ) return NULL;
+    if( !s )
+        return NULL;
+
+    s->p_sys = p_sys = malloc( sizeof( stream_sys_t ) );
+    if( !p_sys )
+    {
+        vlc_object_release( s );
+        return NULL;
+    }
 
     /* Attach it now, needed for b_die */
     vlc_object_attach( s, p_access );
@@ -295,10 +332,6 @@ stream_t *stream_AccessNew( access_t *p_access, bool b_quick )
     s->pf_control = AStreamControl;
     s->pf_destroy = AStreamDestroy;
 
-    s->p_sys = p_sys = malloc( sizeof( stream_sys_t ) );
-    if( p_sys == NULL )
-        goto error;
-
     /* UTF16 and UTF32 text file conversion */
     s->i_char_width = 1;
     s->b_little_endian = false;
@@ -307,11 +340,13 @@ stream_t *stream_AccessNew( access_t *p_access, bool b_quick )
     /* Common field */
     p_sys->p_access = p_access;
     if( p_access->pf_block )
-        p_sys->method = Block;
-    else if (var_CreateGetBool( s, "use-stream-immediate"))
-        p_sys->method = Immediate;
+        p_sys->method = STREAM_METHOD_BLOCK;
+    else if( var_CreateGetBool( s, "use-stream-immediate" ) )
+        p_sys->method = STREAM_METHOD_IMMEDIATE;
     else
-        p_sys->method = Stream;
+        p_sys->method = STREAM_METHOD_STREAM;
+
+    p_sys->record.b_active = false;
 
     p_sys->i_pos = p_access->info.i_pos;
 
@@ -391,7 +426,7 @@ stream_t *stream_AccessNew( access_t *p_access, bool b_quick )
     p_sys->i_peek = 0;
     p_sys->p_peek = NULL;
 
-    if( p_sys->method == Block )
+    if( p_sys->method == STREAM_METHOD_BLOCK )
     {
         msg_Dbg( s, "Using AStream*Block" );
         s->pf_read = AStreamReadBlock;
@@ -414,7 +449,7 @@ stream_t *stream_AccessNew( access_t *p_access, bool b_quick )
             goto error;
         }
     }
-    else if (p_sys->method == Immediate)
+    else if( p_sys->method == STREAM_METHOD_IMMEDIATE )
     {
         msg_Dbg( s, "Using AStream*Immediate" );
 
@@ -425,8 +460,9 @@ stream_t *stream_AccessNew( access_t *p_access, bool b_quick )
         p_sys->immediate.i_end = 0;
         p_sys->immediate.p_buffer = malloc( STREAM_CACHE_SIZE );
 
-        msg_Dbg( s, "p_buffer %p-%p", p_sys->immediate.p_buffer,
-                p_sys->immediate.p_buffer + STREAM_CACHE_SIZE );
+        msg_Dbg( s, "p_buffer %p-%p",
+                 p_sys->immediate.p_buffer,
+                 &p_sys->immediate.p_buffer[STREAM_CACHE_SIZE] );
 
         if( p_sys->immediate.p_buffer == NULL )
         {
@@ -435,10 +471,12 @@ stream_t *stream_AccessNew( access_t *p_access, bool b_quick )
             goto error;
         }
     }
-    else /* ( p_sys->method == Stream ) */
+    else
     {
         int i;
 
+        assert( p_sys->method == STREAM_METHOD_STREAM );
+
         msg_Dbg( s, "Using AStream*Stream" );
 
         s->pf_read = AStreamReadStream;
@@ -484,7 +522,7 @@ stream_t *stream_AccessNew( access_t *p_access, bool b_quick )
     return s;
 
 error:
-    if( p_sys->method == Block )
+    if( p_sys->method == STREAM_METHOD_BLOCK )
     {
         /* Nothing yet */
     }
@@ -511,8 +549,15 @@ static void AStreamDestroy( stream_t *s )
 
     vlc_object_detach( s );
 
-    if( p_sys->method == Block ) block_ChainRelease( p_sys->block.p_first );
-    else free( p_sys->stream.p_buffer );
+    if( p_sys->record.b_active )
+        ARecordSetState( s, false, NULL );
+
+    if( p_sys->method == STREAM_METHOD_BLOCK )
+        block_ChainRelease( p_sys->block.p_first );
+    else if( p_sys->method == STREAM_METHOD_IMMEDIATE )
+        free( p_sys->immediate.p_buffer );
+    else
+        free( p_sys->stream.p_buffer );
 
     free( p_sys->p_peek );
 
@@ -535,7 +580,6 @@ static void UStreamDestroy( stream_t *s )
 {
     access_t *p_access = (access_t *)s->p_parent;
     AStreamDestroy( s );
-    vlc_object_release( p_access );
     access_Delete( p_access );
 }
 
@@ -548,7 +592,7 @@ void stream_AccessReset( stream_t *s )
 
     p_sys->i_pos = p_sys->p_access->info.i_pos;
 
-    if( p_sys->method == Block )
+    if( p_sys->method == STREAM_METHOD_BLOCK )
     {
         block_ChainRelease( p_sys->block.p_first );
 
@@ -563,14 +607,16 @@ void stream_AccessReset( stream_t *s )
         /* Do the prebuffering */
         AStreamPrebufferBlock( s );
     }
-    else if( p_sys->method == Immediate )
+    else if( p_sys->method == STREAM_METHOD_IMMEDIATE )
     {
         stream_buffer_empty( s, stream_buffered_size( s ) );
     }
-    else /* ( p_sys->method == Stream ) */
+    else
     {
         int i;
 
+        assert( p_sys->method == STREAM_METHOD_STREAM );
+
         /* Setup our tracks */
         p_sys->stream.i_offset = 0;
         p_sys->stream.i_tk     = 0;
@@ -616,6 +662,8 @@ static int AStreamControl( stream_t *s, int i_query, va_list args )
     access_t     *p_access = p_sys->p_access;
 
     bool *p_bool;
+    bool b_bool;
+    const char *psz_string;
     int64_t    *pi_64, i_64;
     int        i_int;
 
@@ -651,12 +699,18 @@ static int AStreamControl( stream_t *s, int i_query, va_list args )
 
         case STREAM_SET_POSITION:
             i_64 = (int64_t)va_arg( args, int64_t );
-            if( p_sys->method == Block )
+            switch( p_sys->method )
+            {
+            case STREAM_METHOD_BLOCK:
                 return AStreamSeekBlock( s, i_64 );
-            else if( p_sys->method == Immediate )
+            case STREAM_METHOD_IMMEDIATE:
                 return AStreamSeekImmediate( s, i_64 );
-            else /* ( p_sys->method == Stream ) */
+            case STREAM_METHOD_STREAM:
                 return AStreamSeekStream( s, i_64 );
+            default:
+                assert(0);
+                return VLC_EGENERIC;
+            }
 
         case STREAM_GET_MTU:
             return VLC_EGENERIC;
@@ -676,6 +730,12 @@ static int AStreamControl( stream_t *s, int i_query, va_list args )
         case STREAM_GET_CONTENT_TYPE:
             return access_Control( p_access, ACCESS_GET_CONTENT_TYPE,
                                     va_arg( args, char ** ) );
+        case STREAM_SET_RECORD_STATE:
+            b_bool = (bool)va_arg( args, int );
+            psz_string = NULL;
+            if( b_bool )
+                psz_string = (const char*)va_arg( args, const char* );
+            return ARecordSetState( s, b_bool, psz_string );
 
         default:
             msg_Err( s, "invalid stream_vaControl query=0x%x", i_query );
@@ -684,7 +744,99 @@ static int AStreamControl( stream_t *s, int i_query, va_list args )
     return VLC_SUCCESS;
 }
 
+/****************************************************************************
+ * ARecord*: record stream functions
+ ****************************************************************************/
+static int  ARecordStart( stream_t *s, const char *psz_extension )
+{
+    stream_sys_t *p_sys = s->p_sys;
+
+    char *psz_file;
+    FILE *f;
+
+    /* */
+    if( !psz_extension )
+        psz_extension = "dat";
+
+    /* Retreive path */
+    char *psz_path = var_CreateGetString( s, "input-record-path" );
+    if( !psz_path || *psz_path == '\0' )
+    {
+        free( psz_path );
+        psz_path = strdup( config_GetHomeDir() );
+    }
+
+    if( !psz_path )
+        return VLC_ENOMEM;
+
+    /* Create file name
+     * TODO allow prefix configuration */
+    psz_file = input_CreateFilename( VLC_OBJECT(s), psz_path, INPUT_RECORD_PREFIX, psz_extension );
+
+    free( psz_path );
+
+    if( !psz_file )
+        return VLC_ENOMEM;
+
+    f = utf8_fopen( psz_file, "wb" );
+    if( !f )
+    {
+        free( psz_file );
+        return VLC_EGENERIC;
+    }
+    msg_Dbg( s, "Recording into %s", psz_file );
+    free( psz_file );
+
+    /* */
+    p_sys->record.f = f;
+    p_sys->record.b_active = true;
+    p_sys->record.b_error = false;
+    return VLC_SUCCESS;
+}
+static int  ARecordStop( stream_t *s )
+{
+    stream_sys_t *p_sys = s->p_sys;
+
+    assert( p_sys->record.b_active );
+
+    msg_Dbg( s, "Recording completed" );
+    fclose( p_sys->record.f );
+    p_sys->record.b_active = false;
+    return VLC_SUCCESS;
+}
+
+static int  ARecordSetState( stream_t *s, bool b_record, const char *psz_extension )
+{
+    stream_sys_t *p_sys = s->p_sys;
+
+    if( !!p_sys->record.b_active == !!b_record )
+        return VLC_SUCCESS;
+
+    if( b_record )
+        return ARecordStart( s, psz_extension );
+    else
+        return ARecordStop( s );
+}
+static void ARecordWrite( stream_t *s, const uint8_t *p_buffer, size_t i_buffer )
+{
+    stream_sys_t *p_sys = s->p_sys;
+
+    assert( p_sys->record.b_active );
 
+    if( i_buffer > 0 )
+    {
+        const bool b_previous_error = p_sys->record.b_error;
+        const size_t i_written = fwrite( p_buffer, 1, i_buffer, p_sys->record.f );
+
+        p_sys->record.b_error = i_written != i_buffer;
+
+        /* TODO maybe a intf_UserError or something like that ? */
+        if( p_sys->record.b_error && !b_previous_error )
+            msg_Err( s, "Failed to record data (begin)" );
+        else if( !p_sys->record.b_error && b_previous_error )
+            msg_Err( s, "Failed to record data (end)" );
+    }
+}
 
 /****************************************************************************
  * Method 1:
@@ -727,9 +879,8 @@ static void AStreamPrebufferBlock( stream_t *s )
         /* Fetch a block */
         if( ( b = AReadBlock( s, &b_eof ) ) == NULL )
         {
-            if( b_eof ) break;
-
-            msleep( STREAM_DATA_WAIT );
+            if( b_eof )
+                break;
             continue;
         }
 
@@ -765,18 +916,22 @@ static void AStreamPrebufferBlock( stream_t *s )
 
 static int AStreamRefillBlock( stream_t *s );
 
-static int AStreamReadBlock( stream_t *s, void *p_read, int i_read )
+static int AStreamReadBlock( stream_t *s, void *p_read, unsigned int i_read )
 {
     stream_sys_t *p_sys = s->p_sys;
 
-    uint8_t *p_data= (uint8_t*)p_read;
-    int     i_data = 0;
+    uint8_t *p_data = p_read;
+    uint8_t *p_record = p_data;
+    unsigned int i_data = 0;
 
     /* It means EOF */
     if( p_sys->block.p_current == NULL )
         return 0;
 
-    if( p_read == NULL )
+    if( p_sys->record.b_active && !p_data )
+        p_record = p_data = malloc( i_read );
+
+    if( p_data == NULL )
     {
         /* seek within this stream if possible, else use plain old read and discard */
         stream_sys_t *p_sys = s->p_sys;
@@ -791,7 +946,7 @@ static int AStreamReadBlock( stream_t *s, void *p_read, int i_read )
     {
         int i_current =
             p_sys->block.p_current->i_buffer - p_sys->block.i_offset;
-        int i_copy = __MIN( i_current, i_read - i_data);
+        unsigned int i_copy = __MIN( (unsigned int)__MAX(i_current,0), i_read - i_data);
 
         /* Copy data */
         if( p_data )
@@ -820,17 +975,25 @@ static int AStreamReadBlock( stream_t *s, void *p_read, int i_read )
         }
     }
 
+    if( p_sys->record.b_active )
+    {
+        if( i_data > 0 && p_record != NULL)
+            ARecordWrite( s, p_record, i_data );
+        if( !p_read )
+            free( p_record );
+    }
+
     p_sys->i_pos += i_data;
     return i_data;
 }
 
-static int AStreamPeekBlock( stream_t *s, const uint8_t **pp_peek, int i_read )
+static int AStreamPeekBlock( stream_t *s, const uint8_t **pp_peek, unsigned int i_read )
 {
     stream_sys_t *p_sys = s->p_sys;
     uint8_t *p_data;
-    int      i_data = 0;
+    unsigned int i_data = 0;
     block_t *b;
-    int      i_offset;
+    unsigned int i_offset;
 
     if( p_sys->block.p_current == NULL ) return 0; /* EOF */
 
@@ -872,7 +1035,7 @@ static int AStreamPeekBlock( stream_t *s, const uint8_t **pp_peek, int i_read )
 
     while( b && i_data < i_read )
     {
-        int i_current = b->i_buffer - i_offset;
+        unsigned int i_current = __MAX(b->i_buffer - i_offset,0);
         int i_copy = __MIN( i_current, i_read - i_data );
 
         memcpy( p_data, &b->p_buffer[i_offset], i_copy );
@@ -996,19 +1159,21 @@ static int AStreamSeekBlock( stream_t *s, int64_t i_pos )
     }
     else
     {
-        /* Read enough data */
-        while( p_sys->block.i_start + p_sys->block.i_size < i_pos )
+        do
         {
+            /* Read and skip enough data */
             if( AStreamRefillBlock( s ) )
                 return VLC_EGENERIC;
 
             while( p_sys->block.p_current &&
-                   p_sys->i_pos + p_sys->block.p_current->i_buffer < i_pos )
+                   p_sys->i_pos + p_sys->block.p_current->i_buffer - p_sys->block.i_offset < i_pos )
             {
-                p_sys->i_pos += p_sys->block.p_current->i_buffer;
+                p_sys->i_pos += p_sys->block.p_current->i_buffer - p_sys->block.i_offset;
                 p_sys->block.p_current = p_sys->block.p_current->p_next;
+                p_sys->block.i_offset = 0;
             }
         }
+        while( p_sys->block.i_start + p_sys->block.i_size < i_pos );
 
         p_sys->block.i_offset = i_pos - p_sys->i_pos;
         p_sys->i_pos = i_pos;
@@ -1051,15 +1216,14 @@ static int AStreamRefillBlock( stream_t *s )
     {
         bool b_eof;
 
-        if( s->b_die ) return VLC_EGENERIC;
-
+        if( s->b_die )
+            return VLC_EGENERIC;
 
         /* Fetch a block */
-        if( ( b = AReadBlock( s, &b_eof ) ) ) break;
-
-        if( b_eof ) return VLC_EGENERIC;
-
-        msleep( STREAM_DATA_WAIT );
+        if( ( b = AReadBlock( s, &b_eof ) ) )
+            break;
+        if( b_eof )
+            return VLC_EGENERIC;
     }
 
     while( b )
@@ -1092,17 +1256,22 @@ static int AStreamRefillBlock( stream_t *s )
  ****************************************************************************/
 static int AStreamRefillStream( stream_t *s );
 
-static int AStreamReadStream( stream_t *s, void *p_read, int i_read )
+static int AStreamReadStream( stream_t *s, void *p_read, unsigned int i_read )
 {
     stream_sys_t *p_sys = s->p_sys;
     stream_track_t *tk = &p_sys->stream.tk[p_sys->stream.i_tk];
 
     uint8_t *p_data = (uint8_t *)p_read;
-    int      i_data = 0;
+    uint8_t *p_record = p_data;
+    unsigned int i_data = 0;
 
-    if( tk->i_start >= tk->i_end ) return 0; /* EOF */
+    if( tk->i_start >= tk->i_end )
+        return 0; /* EOF */
+
+    if( p_sys->record.b_active && !p_data )
+        p_record = p_data = malloc( i_read );
 
-    if( p_read == NULL )
+    if( p_data == NULL )
     {
         /* seek within this stream if possible, else use plain old read and discard */
         stream_sys_t *p_sys = s->p_sys;
@@ -1129,9 +1298,9 @@ static int AStreamReadStream( stream_t *s, void *p_read, int i_read )
     {
         int i_off = (tk->i_start + p_sys->stream.i_offset) %
                     STREAM_CACHE_TRACK_SIZE;
-        int i_current =
-            __MIN( tk->i_end - tk->i_start - p_sys->stream.i_offset,
-                   STREAM_CACHE_TRACK_SIZE - i_off );
+        unsigned int i_current =
+            __MAX(0,__MIN( tk->i_end - tk->i_start - p_sys->stream.i_offset,
+                   STREAM_CACHE_TRACK_SIZE - i_off ));
         int i_copy = __MIN( i_current, i_read - i_data );
 
         if( i_copy <= 0 ) break; /* EOF */
@@ -1162,10 +1331,18 @@ static int AStreamReadStream( stream_t *s, void *p_read, int i_read )
         }
     }
 
+    if( p_sys->record.b_active )
+    {
+        if( i_data > 0 && p_record != NULL)
+            ARecordWrite( s, p_record, i_data );
+        if( !p_read )
+            free( p_record );
+    }
+
     return i_data;
 }
 
-static int AStreamPeekStream( stream_t *s, const uint8_t **pp_peek, int i_read )
+static int AStreamPeekStream( stream_t *s, const uint8_t **pp_peek, unsigned int i_read )
 {
     stream_sys_t *p_sys = s->p_sys;
     stream_track_t *tk = &p_sys->stream.tk[p_sys->stream.i_tk];
@@ -1400,12 +1577,12 @@ static int AStreamRefillStream( stream_t *s )
         /* msg_Dbg( s, "AStreamRefillStream: read=%d", i_read ); */
         if( i_read <  0 )
         {
-            msleep( STREAM_DATA_WAIT );
             continue;
         }
         else if( i_read == 0 )
         {
-            if( !b_read ) return VLC_EGENERIC;
+            if( !b_read )
+                return VLC_EGENERIC;
             return VLC_SUCCESS;
         }
         b_read = true;
@@ -1479,15 +1656,9 @@ static void AStreamPrebufferStream( stream_t *s )
         i_read = __MIN( p_sys->stream.i_read_size, i_read );
         i_read = AReadStream( s, &tk->p_buffer[tk->i_end], i_read );
         if( i_read <  0 )
-        {
-            msleep( STREAM_DATA_WAIT );
             continue;
-        }
         else if( i_read == 0 )
-        {
-            /* EOF */
-            break;
-        }
+            break;  /* EOF */
 
         if( i_first == 0 )
         {
@@ -1505,63 +1676,91 @@ static void AStreamPrebufferStream( stream_t *s )
  * Method 3:
  ****************************************************************************/
 
-static int AStreamReadImmediate( stream_t *s, void *p_read, int i_read )
+static int AStreamReadImmediate( stream_t *s, void *p_read, unsigned int i_read )
 {
     stream_sys_t *p_sys = s->p_sys;
+    uint8_t *p_data= (uint8_t*)p_read;
+    uint8_t *p_record = p_data;
+    unsigned int i_data;
 
 #ifdef STREAM_DEBUG
     msg_Dbg( s, "AStreamReadImmediate p_read=%p i_read=%d",
              p_read, i_read );
 #endif
 
+    if( p_sys->record.b_active && !p_data )
+        p_record = p_data = malloc( i_read );
+
     /* First, check if we already have some data in the buffer,
      * that we could copy directly */
-    int i_copy = __MIN( stream_buffered_size( s ), i_read );
-    if( i_copy )
+    i_data = __MIN( stream_buffered_size( s ), i_read );
+    if( i_data > 0 )
     {
 #ifdef STREAM_DEBUG
-        msg_Dbg( s, "AStreamReadImmediate: copy %d from %p", i_copy, stream_buffer( s ) );
+        msg_Dbg( s, "AStreamReadImmediate: copy %u from %p", i_data, stream_buffer( s ) );
 #endif
 
-        assert( i_copy <= STREAM_CACHE_SIZE );
+        assert( i_data <= STREAM_CACHE_SIZE );
 
-        if( p_read )
+        if( p_data )
         {
-            memcpy( p_read, stream_buffer( s ), i_copy );
-            p_read = (uint8_t *)p_read + i_copy;
+            memcpy( p_data, stream_buffer( s ), i_data );
+            p_data += i_data;
         }
     }
 
     /* Now that we've read our buffer we don't need its i_copy bytes */
-    stream_buffer_empty( s, i_copy );
+    stream_buffer_empty( s, i_data );
 
     /* Now check if we have still to really read some data */
-    int i_to_read = i_read - i_copy;
-    if( i_to_read )
+    while( i_data < i_read )
     {
-        if( p_read )
-            i_to_read = AReadStream( s, p_read, i_to_read );
+        const unsigned int i_to_read = i_read - i_data;
+        int i_result;
+
+        if( p_data )
+        {
+            i_result = AReadStream( s, p_data, i_to_read );
+        }
         else
         {
-            void * dummy = malloc(i_to_read);
-            i_to_read = AReadStream( s, dummy, i_to_read );
-            free(dummy);
+            void *p_dummy = malloc( i_to_read );
+
+            if( !p_dummy )
+                break;
+
+            i_result = AReadStream( s, p_dummy, i_to_read );
+
+            free( p_dummy );
         }
-    }
+        if( i_result <= 0 )
+            break;
 
-    p_sys->i_pos += i_to_read;
+        p_sys->i_pos += i_data;
+        if( p_data )
+            p_data += i_result;
+        i_data += i_result;
+    }
 
-    return i_to_read + i_copy;
+    if( p_sys->record.b_active )
+    {
+        if( i_data > 0 && p_record != NULL)
+            ARecordWrite( s, p_record, i_data );
+        if( !p_read )
+            free( p_record );
+    }
+    return i_data;
 }
 
-static int AStreamPeekImmediate( stream_t *s, const uint8_t **pp_peek, int i_read )
+static int AStreamPeekImmediate( stream_t *s, const uint8_t **pp_peek, unsigned int i_read )
 {
 #ifdef STREAM_DEBUG
     msg_Dbg( s, "AStreamPeekImmediate: %d  size=%"PRId64,
              i_read, size_buffered_size( s ) );
 #endif
 
-    /* Avoid problem, but that shouldn't happen */
+    /* Avoid problem, but that shouldn't happen
+     * FIXME yes it can */
     if( i_read > STREAM_CACHE_SIZE / 2 )
         i_read = STREAM_CACHE_SIZE / 2;
 
@@ -1581,7 +1780,7 @@ static int AStreamPeekImmediate( stream_t *s, const uint8_t **pp_peek, int i_rea
 
     *pp_peek = stream_buffer( s );
 
-    return __MIN(stream_buffered_size( s ), i_read);
+    return __MIN( stream_buffered_size( s ), i_read );
 }
 
 static int AStreamSeekImmediate( stream_t *s, int64_t i_pos )
@@ -1606,7 +1805,8 @@ static int AStreamSeekImmediate( stream_t *s, int64_t i_pos )
     /* Just reset our buffer */
     stream_buffer_empty( s, stream_buffered_size( s ) );
 
-    if( ASeek( s, i_pos ) ) return VLC_EGENERIC;
+    if( ASeek( s, i_pos ) )
+        return VLC_EGENERIC;
 
     return VLC_SUCCESS;
 }
@@ -1621,7 +1821,7 @@ static int AStreamSeekImmediate( stream_t *s, int64_t i_pos )
  */
 #define STREAM_PROBE_LINE 2048
 #define STREAM_LINE_MAX (2048*100)
-char * stream_ReadLine( stream_t *s )
+char *stream_ReadLine( stream_t *s )
 {
     char *p_line = NULL;
     int i_line = 0, i_read = 0;
@@ -1639,45 +1839,26 @@ char * stream_ReadLine( stream_t *s )
 
         /* BOM detection */
         i_pos = stream_Tell( s );
-        if( i_pos == 0 && i_data > 4 )
+        if( i_pos == 0 && i_data >= 3 )
         {
             int i_bom_size = 0;
-            char *psz_encoding = NULL;
+            const char *psz_encoding = NULL;
 
-            if( p_data[0] == 0xEF && p_data[1] == 0xBB && p_data[2] == 0xBF )
+            if( !memcmp( p_data, "\xEF\xBB\xBF", 3 ) )
             {
-                psz_encoding = strdup( "UTF-8" );
+                psz_encoding = "UTF-8";
                 i_bom_size = 3;
             }
-            else if( p_data[0] == 0x00 && p_data[1] == 0x00 )
+            else if( !memcmp( p_data, "\xFF\xFE", 2 ) )
             {
-                if( p_data[2] == 0xFE && p_data[3] == 0xFF )
-                {
-                    psz_encoding = strdup( "UTF-32BE" );
-                    s->i_char_width = 4;
-                    i_bom_size = 4;
-                }
-            }
-            else if( p_data[0] == 0xFF && p_data[1] == 0xFE )
-            {
-                if( p_data[2] == 0x00 && p_data[3] == 0x00 )
-                {
-                    psz_encoding = strdup( "UTF-32LE" );
-                    s->i_char_width = 4;
-                    s->b_little_endian = true;
-                    i_bom_size = 4;
-                }
-                else
-                {
-                    psz_encoding = strdup( "UTF-16LE" );
-                    s->b_little_endian = true;
-                    s->i_char_width = 2;
-                    i_bom_size = 2;
-                }
+                psz_encoding = "UTF-16LE";
+                s->b_little_endian = true;
+                s->i_char_width = 2;
+                i_bom_size = 2;
             }
-            else if( p_data[0] == 0xFE && p_data[1] == 0xFF )
+            else if( !memcmp( p_data, "\xFE\xFF", 2 ) )
             {
-                psz_encoding = strdup( "UTF-16BE" );
+                psz_encoding = "UTF-16BE";
                 s->i_char_width = 2;
                 i_bom_size = 2;
             }
@@ -1710,7 +1891,6 @@ char * stream_ReadLine( stream_t *s )
                     var_SetString( p_input, "subsdec-encoding", "UTF-8" );
                     vlc_object_release( p_input );
                 }
-                free( psz_encoding );
             }
         }
 
@@ -1750,23 +1930,6 @@ char * stream_ReadLine( stream_t *s )
                         p += 2;
                 }
             }
-            else if( s->i_char_width == 4 )
-            {
-                if( s->b_little_endian == true)
-                {
-                    /* UTF-32LE: 0A 00 00 00 <LF> */
-                    while( p <= p_last && ( p[0] != 0x0A || p[1] != 0x00 ||
-                           p[2] != 0x00 || p[3] != 0x00 ) )
-                        p += 4;
-                }
-                else
-                {
-                    /* UTF-32BE: 00 00 00 0A <LF> */
-                    while( p <= p_last && ( p[3] != 0x0A || p[2] != 0x00 ||
-                           p[1] != 0x00 || p[0] != 0x00 ) )
-                        p += 4;
-                }
-            }
 
             if( p > p_last )
             {
@@ -1857,7 +2020,7 @@ error:
 /****************************************************************************
  * Access reading/seeking wrappers to handle concatenated streams.
  ****************************************************************************/
-static int AReadStream( stream_t *s, void *p_read, int i_read )
+static int AReadStream( stream_t *s, void *p_read, unsigned int i_read )
 {
     stream_sys_t *p_sys = s->p_sys;
     access_t *p_access = p_sys->p_access;
@@ -1958,7 +2121,7 @@ static block_t *AReadBlock( stream_t *s, bool *pb_eof )
         return p_block;
     }
 
-    p_block = p_sys->p_list_access->pf_block( p_access );
+    p_block = p_sys->p_list_access->pf_block( p_sys->p_list_access );
     if( p_access->b_die )
         vlc_object_kill( s );
     b_eof = p_sys->p_list_access->info.b_eof;
@@ -2100,7 +2263,7 @@ int stream_Control( stream_t *s, int i_query, ... )
     va_list args;
     int     i_result;
 
-    if ( s == NULL )
+    if( s == NULL )
         return VLC_EGENERIC;
 
     va_start( args, i_query );
@@ -2122,12 +2285,13 @@ block_t *stream_Block( stream_t *s, int i_size )
     block_t *p_bk = block_New( s, i_size );
     if( p_bk )
     {
-        p_bk->i_buffer = stream_Read( s, p_bk->p_buffer, i_size );
-        if( p_bk->i_buffer > 0 )
+        int i_read = stream_Read( s, p_bk->p_buffer, i_size );
+        if( i_read > 0 )
         {
+            p_bk->i_buffer = i_read;
             return p_bk;
         }
+        block_Release( p_bk );
     }
-    if( p_bk ) block_Release( p_bk );
     return NULL;
 }