stream.c 10 KB
Newer Older
1 2 3
/*****************************************************************************
 * stream.c
 *****************************************************************************
4
 * Copyright (C) 1999-2004 VideoLAN
Clément Stenac's avatar
Clément Stenac committed
5
 * $Id: stream.c,v 1.13 2004/01/25 17:16:06 zorglub Exp $
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
 *
 * Authors: Laurent Aimar <fenrir@via.ecp.fr>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
 *****************************************************************************/

#include <stdlib.h>
#include <vlc/vlc.h>
#include <vlc/input.h>

#include "ninput.h"

Sigmund Augdal Helberg's avatar
Sigmund Augdal Helberg committed
30
#define MAX_LINE 1024
31 32 33 34 35 36
/****************************************************************************
 * stream_* :
 * XXX for now it's just a wrapper
 *
 ****************************************************************************/

37 38 39
/**
 * Handle to a stream.
 */
40 41 42 43
struct stream_t
{
    VLC_COMMON_MEMBERS

44
    /** pointer to the input thread */
45 46 47 48
    input_thread_t *p_input;

};

49 50 51
/**
 * Create a "stream_t *" from an "input_thread_t *".
 */
Gildas Bazin's avatar
 
Gildas Bazin committed
52
stream_t *stream_OpenInput( input_thread_t *p_input )
53 54 55 56 57 58 59 60 61 62 63 64
{
    stream_t *s;

    s = vlc_object_create( p_input, sizeof( stream_t ) );
    if( s )
    {
        s->p_input = p_input;
    }

    return s;
}

65 66 67
/**
 * Destroy a previously created "stream_t *" instance.
 */
Gildas Bazin's avatar
 
Gildas Bazin committed
68
void stream_Release( stream_t *s )
69 70 71 72
{
    vlc_object_destroy( s );
}

73
/**
Gildas Bazin's avatar
 
Gildas Bazin committed
74 75
 * Similar to #stream_Control(), but takes a va_list and not variable
 * arguments.
76
 */
Gildas Bazin's avatar
 
Gildas Bazin committed
77
int stream_vaControl( stream_t *s, int i_query, va_list args )
78 79 80
{
    vlc_bool_t *p_b;
    int64_t    *p_i64, i64;
81
    int        *p_int;
82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104

    switch( i_query )
    {
        case STREAM_GET_SIZE:
            p_i64 = (int64_t*) va_arg( args, int64_t * );

            vlc_mutex_lock( &s->p_input->stream.stream_lock );
            *p_i64 = s->p_input->stream.p_selected_area->i_size;
            vlc_mutex_unlock( &s->p_input->stream.stream_lock );
            return VLC_SUCCESS;

        case STREAM_CAN_SEEK:
            p_b = (vlc_bool_t*) va_arg( args, vlc_bool_t * );

            vlc_mutex_lock( &s->p_input->stream.stream_lock );
            *p_b = s->p_input->stream.b_seekable;
            vlc_mutex_unlock( &s->p_input->stream.stream_lock );
            return VLC_SUCCESS;

        case STREAM_CAN_FASTSEEK:
            p_b = (vlc_bool_t*) va_arg( args, vlc_bool_t * );

            vlc_mutex_lock( &s->p_input->stream.stream_lock );
Gildas Bazin's avatar
 
Gildas Bazin committed
105 106
            *p_b = s->p_input->stream.b_seekable &&
                   s->p_input->stream.i_method == INPUT_METHOD_FILE;
107 108 109 110 111 112 113 114 115 116 117 118
            vlc_mutex_unlock( &s->p_input->stream.stream_lock );
            return VLC_SUCCESS;

        case STREAM_GET_POSITION:
            p_i64 = (int64_t*) va_arg( args, int64_t * );

            vlc_mutex_lock( &s->p_input->stream.stream_lock );
            *p_i64 = s->p_input->stream.p_selected_area->i_tell;
            vlc_mutex_unlock( &s->p_input->stream.stream_lock );
            return VLC_SUCCESS;

        case STREAM_SET_POSITION:
119
        {
120
            i64 = (int64_t) va_arg( args, int64_t );
121
            int64_t i_skip;
122 123 124 125 126 127 128

            vlc_mutex_lock( &s->p_input->stream.stream_lock );
            if( i64 < 0 ||
                ( s->p_input->stream.p_selected_area->i_size > 0 &&
                  s->p_input->stream.p_selected_area->i_size < i64 ) )
            {
                vlc_mutex_unlock( &s->p_input->stream.stream_lock );
129
                msg_Warn( s, "seek out of bound" );
130 131 132
                return VLC_EGENERIC;
            }

133 134 135 136 137 138 139 140
            i_skip = i64 - s->p_input->stream.p_selected_area->i_tell;

            if( i_skip == 0 )
            {
                vlc_mutex_unlock( &s->p_input->stream.stream_lock );
                return VLC_SUCCESS;
            }

Clément Stenac's avatar
Clément Stenac committed
141 142
            if( i_skip > 0 && i_skip < s->p_input->p_last_data -
                            s->p_input->p_current_data - 1 )
143
            {
144 145 146 147
                /* We can skip without reading/seeking */
                s->p_input->p_current_data += i_skip;
                s->p_input->stream.p_selected_area->i_tell = i64;
                vlc_mutex_unlock( &s->p_input->stream.stream_lock );
148 149
                return VLC_SUCCESS;
            }
150
            vlc_mutex_unlock( &s->p_input->stream.stream_lock );
151 152 153

            if( s->p_input->stream.b_seekable &&
                ( s->p_input->stream.i_method == INPUT_METHOD_FILE ||
Clément Stenac's avatar
Clément Stenac committed
154 155
                  i_skip < 0 || i_skip >= ( s->p_input->i_mtu > 0 ?
                                            s->p_input->i_mtu : 4096 ) ) )
156 157 158 159 160 161
            {
                input_AccessReinit( s->p_input );
                s->p_input->pf_seek( s->p_input, i64 );
                return VLC_SUCCESS;
            }

162
            if( i_skip > 0 )
163
            {
Gildas Bazin's avatar
 
Gildas Bazin committed
164
                data_packet_t *p_data;
165

Laurent Aimar's avatar
Laurent Aimar committed
166 167
                if( i_skip > 1000 )
                {
168
                    msg_Warn( s, "will skip "I64Fd" bytes, slow", i_skip );
Laurent Aimar's avatar
Laurent Aimar committed
169
                }
170

171
                while( i_skip > 0 )
172 173 174
                {
                    int i_read;

175
                    i_read = input_SplitBuffer( s->p_input, &p_data,
Gildas Bazin's avatar
 
Gildas Bazin committed
176
                                 __MIN( (int)s->p_input->i_bufsize, i_skip ) );
177 178 179 180 181 182 183 184 185 186 187 188 189
                    if( i_read < 0 )
                    {
                        return VLC_EGENERIC;
                    }
                    i_skip -= i_read;

                    input_DeletePacket( s->p_input->p_method_data, p_data );
                    if( i_read == 0 && i_skip > 0 )
                    {
                        return VLC_EGENERIC;
                    }
                }
            }
190
            return VLC_SUCCESS;
191
        }
192

193 194 195
        case STREAM_GET_MTU:
            p_int = (int*) va_arg( args, int * );
            *p_int = s->p_input->i_mtu;
196 197 198 199 200 201 202 203
            return VLC_SUCCESS;

        default:
            msg_Err( s, "invalid stream_vaControl query=0x%x", i_query );
            return VLC_EGENERIC;
    }
}

204 205 206 207 208
/**
 * Use to control the "stream_t *". Look at #stream_query_e for
 * possible "i_query" value and format arguments.  Return VLC_SUCCESS
 * if ... succeed ;) and VLC_EGENERIC if failed or unimplemented
 */
Gildas Bazin's avatar
 
Gildas Bazin committed
209
int stream_Control( stream_t *s, int i_query, ... )
210 211 212 213 214 215 216 217 218 219 220
{
    va_list args;
    int     i_result;

    va_start( args, i_query );
    i_result = stream_vaControl( s, i_query, args );
    va_end( args );

    return i_result;
}

221 222 223 224 225 226
/**
 * Try to read "i_read" bytes into a buffer pointed by "p_read".  If
 * "p_read" is NULL then data are skipped instead of read.  The return
 * value is the real numbers of bytes read/skip. If this value is less
 * than i_read that means that it's the end of the stream.
 */
Gildas Bazin's avatar
 
Gildas Bazin committed
227
int stream_Read( stream_t *s, void *p_data, int i_data )
228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251
{
    uint8_t       *p = (uint8_t*)p_data;
    data_packet_t *p_packet;

    int i_read = 0;

    if( p_data == NULL && i_data > 0 )
    {
        int64_t i_pos;

        stream_Control( s, STREAM_GET_POSITION, &i_pos );

        i_pos += i_data;
        if( stream_Control( s, STREAM_SET_POSITION, i_pos ) )
        {
            return 0;
        }
        return i_data;
    }

    while( i_data > 0 && !s->p_input->b_die )
    {
        int i_count;

252
        i_count = input_SplitBuffer( s->p_input, &p_packet,
Gildas Bazin's avatar
 
Gildas Bazin committed
253
                      __MIN( i_data, (int)s->p_input->i_bufsize ) );
Gildas Bazin's avatar
 
Gildas Bazin committed
254

255 256
        if( i_count <= 0 )
        {
Gildas Bazin's avatar
 
Gildas Bazin committed
257 258 259
            if( i_count == 0 )
                input_DeletePacket( s->p_input->p_method_data, p_packet );

260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277
            return i_read;
        }

        if( p )
        {
            memcpy( p, p_packet->p_payload_start, i_count );
            p += i_count;
        }

        input_DeletePacket( s->p_input->p_method_data, p_packet );

        i_data -= i_count;
        i_read += i_count;
    }

    return i_read;
}

278 279
/**
 * Store in pp_peek a pointer to the next "i_peek" bytes in the stream
Sigmund Augdal Helberg's avatar
Sigmund Augdal Helberg committed
280 281 282 283 284
 * \return The real numbers of valid bytes, if it's less
 * or equal to 0, *pp_peek is invalid.
 * \note pp_peek is a pointer to internal buffer and it will be invalid as
 * soons as other stream_* functions are called.
 * \note Due to input limitation, it could be less than i_peek without meaning
285 286 287
 * the end of the stream (but only when you have i_peek >=
 * p_input->i_bufsize)
 */
Sigmund Augdal Helberg's avatar
Sigmund Augdal Helberg committed
288
int stream_Peek( stream_t *s, uint8_t **pp_peek, int i_peek )
289
{
Sigmund Augdal Helberg's avatar
Sigmund Augdal Helberg committed
290
    return input_Peek( s->p_input, pp_peek, i_peek );
291 292
}

Gildas Bazin's avatar
 
Gildas Bazin committed
293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311 312 313 314
/**
 * Read "i_size" bytes and store them in a block_t. If less than "i_size"
 * bytes are available then return what is left and if nothing is availble,
 * return NULL.
 */
block_t *stream_Block( stream_t *s, int i_size )
{
    block_t *p_block;

    if( i_size <= 0 ) return NULL;
    if( !(p_block = block_New( s->p_input, i_size ) ) ) return NULL;

    p_block->i_buffer = stream_Read( s, p_block->p_buffer, i_size );
    if( !p_block->i_buffer )
    {
        block_Release( p_block );
        p_block = NULL;
    }

    return p_block;
}

Sigmund Augdal Helberg's avatar
Sigmund Augdal Helberg committed
315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340
/**
 * Read from the stream untill first newline.
 * \param s Stream handle to read from
 * \return A null-terminated string. This must be freed,
 */
char *stream_ReadLine( stream_t *s )
{
    uint8_t *p_data;
    char    *p_line;
    int      i_data;
    int      i = 0;
    i_data = stream_Peek( s, &p_data, MAX_LINE );
    msg_Dbg( s->p_input, "i_data %d", i_data );
    while( i < i_data && p_data[i] != '\n' )
    {
        i++;
    }
    if( i_data <= 0 )
    {
        return NULL;
    }
    else
    {
        p_line = malloc( i + 1 );
        if( p_line == NULL )
        {
Clément Stenac's avatar
Clément Stenac committed
341
            msg_Err( s->p_input, "out of memory" );
Sigmund Augdal Helberg's avatar
Sigmund Augdal Helberg committed
342 343 344 345 346 347 348 349
            return NULL;
        }
        i = stream_Read( s, p_line, i + 1 );
        p_line[ i - 1 ] = '\0';
        msg_Dbg( s->p_input, "found %d chars long line", i );
        return p_line;
    }
}