X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=modules%2Fdemux%2Favi%2Flibavi.c;h=678e060277f6e734bbc49217cecdfa6d8579ce20;hb=6ee1e193fd896ab9a4729fde14f009d9ce629815;hp=03d254fede3a59108e154f50a1c1288fbf1d9acd;hpb=9a5b0b8000ed8cd3e44f6162a28485edd51d2abc;p=vlc diff --git a/modules/demux/avi/libavi.c b/modules/demux/avi/libavi.c index 03d254fede..678e060277 100644 --- a/modules/demux/avi/libavi.c +++ b/modules/demux/avi/libavi.c @@ -1,8 +1,8 @@ /***************************************************************************** - * libavi.c : + * libavi.c : LibAVI ***************************************************************************** - * Copyright (C) 2001 VideoLAN - * $Id: libavi.c,v 1.14 2003/01/20 13:01:53 fenrir Exp $ + * Copyright (C) 2001 the VideoLAN team + * $Id$ * Authors: Laurent Aimar * * This program is free software; you can redistribute it and/or modify @@ -17,276 +17,72 @@ * * 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. + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA. *****************************************************************************/ -#include /* malloc(), free() */ -#include /* strdup() */ -#include -#include #include -#include +#include +#include /* BITMAPINFOHEADER */ #include "libavi.h" #define AVI_DEBUG 1 -#define AVIFOURCC_PRINT( x ) \ - (x)&0xff, \ - ( (x) >> 8 )&0xff, \ - ( (x) >> 16 )&0xff, \ - ( (x) >> 24 )&0xff - -#define FREE( p ) \ - if( p ) {free( p ); p = NULL; } #define __EVEN( x ) ( (x)&0x01 ? (x)+1 : (x) ) -/* Some functions to manipulate memory */ -static uint16_t GetWLE( uint8_t *p_buff ) -{ - return (uint16_t)p_buff[0] | ( ((uint16_t)p_buff[1]) << 8 ); -} - -static uint32_t GetDWLE( uint8_t *p_buff ) -{ - return (uint32_t)p_buff[0] | ( ((uint32_t)p_buff[1]) << 8 ) | - ( ((uint32_t)p_buff[2]) << 16 ) | ( ((uint32_t)p_buff[3]) << 24 ); -} static vlc_fourcc_t GetFOURCC( byte_t *p_buff ) { return VLC_FOURCC( p_buff[0], p_buff[1], p_buff[2], p_buff[3] ); } -/***************************************************************************** - * Some basic functions to manipulate stream more easily in vlc - * - * AVI_TellAbsolute get file position - * - * AVI_SeekAbsolute seek in the file - * - * AVI_ReadData read data from the file in a buffer - * - * AVI_SkipBytes skip bytes - * - *****************************************************************************/ -off_t AVI_TellAbsolute( input_thread_t *p_input ) -{ - off_t i_pos; - - vlc_mutex_lock( &p_input->stream.stream_lock ); - - i_pos= p_input->stream.p_selected_area->i_tell; -// - ( p_input->p_last_data - p_input->p_current_data ); - - vlc_mutex_unlock( &p_input->stream.stream_lock ); - - return i_pos; -} - -int AVI_SeekAbsolute( input_thread_t *p_input, - off_t i_pos) -{ - off_t i_filepos; - - if( i_pos >= p_input->stream.p_selected_area->i_size ) - { - return VLC_EGENERIC; - } - - i_filepos = AVI_TellAbsolute( p_input ); - - if( i_filepos == i_pos ) - { - return VLC_SUCCESS; - } - - if( p_input->stream.b_seekable && - p_input->stream.i_method == INPUT_METHOD_FILE ) - { - p_input->pf_seek( p_input, i_pos ); - input_AccessReinit( p_input ); - return VLC_SUCCESS; - } - else - { - data_packet_t *p_data; - int i_skip = i_pos - i_filepos; - - msg_Warn( p_input, "will skip %d bytes, slow", i_skip ); - if( i_skip < 0 ) - { - return VLC_EGENERIC; // failed - } - while (i_skip > 0 ) - { - int i_read; - - i_read = input_SplitBuffer( p_input, &p_data, - __MIN( 4096, i_skip ) ); - if( i_read < 0 ) - { - return VLC_EGENERIC; - } - i_skip -= i_read; - - input_DeletePacket( p_input->p_method_data, p_data ); - if( i_read == 0 && i_skip > 0 ) - { - return VLC_EGENERIC; - } - } -#if 0 - while( i_skip > 0 ) - { - i_peek = input_Peek( p_input, &p_peek, i_skip+1 ); - i_peek--; - i_skip -= i_peek; - vlc_mutex_lock( &p_input->stream.stream_lock ); - p_input->p_current_data += i_peek; // skip them - vlc_mutex_unlock( &p_input->stream.stream_lock ); - if( i_peek <= 0 ) - { - return VLC_EGENERIC; - } - } -#endif - return VLC_SUCCESS; - } -} - -/* return amount read if success, 0 if failed */ -int AVI_ReadData( input_thread_t *p_input, uint8_t *p_buff, int i_size ) -{ - data_packet_t *p_data; - int i_count; - int i_read = 0; +#define AVI_ChunkFree( a, b ) _AVI_ChunkFree( (a), (avi_chunk_t*)(b) ) +void _AVI_ChunkFree( stream_t *, avi_chunk_t *p_chk ); - - if( !i_size ) - { - return 0; - } - - do - { - i_count = input_SplitBuffer(p_input, &p_data, __MIN( i_size, 1024 ) ); - if( i_count <= 0 ) - { - return i_read; - } - memcpy( p_buff, p_data->p_payload_start, i_count ); - input_DeletePacket( p_input->p_method_data, p_data ); - - p_buff += i_count; - i_size -= i_count; - i_read += i_count; - - } while( i_size ); - - return i_read; -} - -int AVI_SkipBytes( input_thread_t *p_input, int i_count ) -{ - /* broken with new use of i_tell */ -#if 0 - int i_buff_size; - vlc_mutex_lock( &p_input->stream.stream_lock ); - i_buff_size = p_input->p_last_data - p_input->p_current_data; - vlc_mutex_unlock( &p_input->stream.stream_lock ); - - if( i_count > 0 && i_count + 1 < i_buff_size ) - { - uint8_t *p_peek; - - input_Peek( p_input, &p_peek, i_count + 1 ); - - vlc_mutex_lock( &p_input->stream.stream_lock ); - p_input->p_current_data += i_count; // skip them - vlc_mutex_unlock( &p_input->stream.stream_lock ); - return VLC_SUCCESS; - } - else -#endif - { - return AVI_SeekAbsolute( p_input, - AVI_TellAbsolute( p_input ) + i_count ); - } -} - -/***************************************************************************** - * - * AVI_TestFile: look at first bytes to see if it's a valid avi file - * - * unseekable: ok - * - *****************************************************************************/ -int AVI_TestFile( input_thread_t *p_input ) -{ - uint8_t *p_peek; - - if( input_Peek( p_input, &p_peek, 8 ) < 8 ) - { - msg_Err( p_input, "cannot peek()" ); - return VLC_EGENERIC; - } - - if( GetFOURCC( p_peek ) == AVIFOURCC_RIFF && - GetFOURCC( p_peek + 8 ) == AVIFOURCC_AVI ) - { - return VLC_SUCCESS; - } - else - { - return VLC_EGENERIC; - } -} /**************************************************************************** * * Basics functions to manipulates chunks * ****************************************************************************/ -static int AVI_ChunkReadCommon( input_thread_t *p_input, - avi_chunk_t *p_chk ) +static int AVI_ChunkReadCommon( stream_t *s, avi_chunk_t *p_chk ) { uint8_t *p_peek; int i_peek; memset( p_chk, 0, sizeof( avi_chunk_t ) ); - if( ( i_peek = input_Peek( p_input, &p_peek, 8 ) ) < 8 ) + if( ( i_peek = stream_Peek( s, &p_peek, 8 ) ) < 8 ) { return VLC_EGENERIC; } p_chk->common.i_chunk_fourcc = GetFOURCC( p_peek ); p_chk->common.i_chunk_size = GetDWLE( p_peek + 4 ); - p_chk->common.i_chunk_pos = AVI_TellAbsolute( p_input ); + p_chk->common.i_chunk_pos = stream_Tell( s ); p_chk->common.p_father = NULL; p_chk->common.p_next = NULL; p_chk->common.p_first = NULL; p_chk->common.p_next = NULL; + #ifdef AVI_DEBUG - msg_Dbg( p_input, - "Found Chunk fourcc:%8.8x (%c%c%c%c) size:"I64Fd" pos:"I64Fd, + msg_Dbg( (vlc_object_t*)s, + "found Chunk fourcc:%8.8x (%4.4s) size:"I64Fd" pos:"I64Fd, p_chk->common.i_chunk_fourcc, - AVIFOURCC_PRINT( p_chk->common.i_chunk_fourcc ), + (char*)&p_chk->common.i_chunk_fourcc, p_chk->common.i_chunk_size, p_chk->common.i_chunk_pos ); #endif return VLC_SUCCESS; } -static int AVI_NextChunk( input_thread_t *p_input, - avi_chunk_t *p_chk ) +static int AVI_NextChunk( stream_t *s, avi_chunk_t *p_chk ) { avi_chunk_t chk; if( !p_chk ) { - if( AVI_ChunkReadCommon( p_input, &chk ) ) + if( AVI_ChunkReadCommon( s, &chk ) ) { return VLC_EGENERIC; } @@ -295,79 +91,68 @@ static int AVI_NextChunk( input_thread_t *p_input, if( p_chk->common.p_father ) { - if( p_chk->common.p_father->common.i_chunk_pos + + if( p_chk->common.p_father->common.i_chunk_pos + __EVEN( p_chk->common.p_father->common.i_chunk_size ) + 8 < - p_chk->common.i_chunk_pos + + p_chk->common.i_chunk_pos + __EVEN( p_chk->common.i_chunk_size ) + 8 ) { return VLC_EGENERIC; } } - return AVI_SeekAbsolute( p_input, - p_chk->common.i_chunk_pos + + return stream_Seek( s, p_chk->common.i_chunk_pos + __EVEN( p_chk->common.i_chunk_size ) + 8 ); } -int _AVI_ChunkGoto( input_thread_t *p_input, - avi_chunk_t *p_chk ) -{ - if( !p_chk ) - { - return VLC_EGENERIC; - } - return AVI_SeekAbsolute( p_input, p_chk->common.i_chunk_pos ); -} - /**************************************************************************** * - * Functions to read chunks + * Functions to read chunks * ****************************************************************************/ -static int AVI_ChunkRead_list( input_thread_t *p_input, - avi_chunk_t *p_container, - vlc_bool_t b_seekable ) +static int AVI_ChunkRead_list( stream_t *s, avi_chunk_t *p_container ) { avi_chunk_t *p_chk; uint8_t *p_peek; + vlc_bool_t b_seekable; - if( p_container->common.i_chunk_size < 8 ) + if( p_container->common.i_chunk_size > 0 && p_container->common.i_chunk_size < 8 ) { /* empty box */ - msg_Warn( p_input, "empty list chunk" ); + msg_Warn( (vlc_object_t*)s, "empty list chunk" ); return VLC_EGENERIC; } - if( input_Peek( p_input, &p_peek, 12 ) < 12 ) + if( stream_Peek( s, &p_peek, 12 ) < 12 ) { - msg_Warn( p_input, "cannot peek while reading list chunk" ); + msg_Warn( (vlc_object_t*)s, "cannot peek while reading list chunk" ); return VLC_EGENERIC; } + + stream_Control( s, STREAM_CAN_FASTSEEK, &b_seekable ); + p_container->list.i_type = GetFOURCC( p_peek + 8 ); if( p_container->common.i_chunk_fourcc == AVIFOURCC_LIST && p_container->list.i_type == AVIFOURCC_movi ) { - msg_Dbg( p_input, "Skipping movi chunk" ); + msg_Dbg( (vlc_object_t*)s, "skipping movi chunk" ); if( b_seekable ) { - return AVI_NextChunk( p_input, p_container ); - } - else - { - return VLC_SUCCESS; // point at begining of LIST-movi + return AVI_NextChunk( s, p_container ); } + return VLC_SUCCESS; /* point at begining of LIST-movi */ } - if( AVI_SkipBytes( p_input, 12 ) ) + if( stream_Read( s, NULL, 12 ) != 12 ) { - msg_Warn( p_input, "cannot enter chunk" ); + msg_Warn( (vlc_object_t*)s, "cannot enter chunk" ); return VLC_EGENERIC; } + #ifdef AVI_DEBUG - msg_Dbg( p_input, - "found LIST chunk: \'%c%c%c%c\'", - AVIFOURCC_PRINT( p_container->list.i_type ) ); + msg_Dbg( (vlc_object_t*)s, + "found LIST chunk: \'%4.4s\'", + (char*)&p_container->list.i_type ); #endif - msg_Dbg( p_input, "", AVIFOURCC_PRINT( p_container->list.i_type ) ); + msg_Dbg( (vlc_object_t*)s, "", (char*)&p_container->list.i_type ); for( ; ; ) { p_chk = malloc( sizeof( avi_chunk_t ) ); @@ -382,22 +167,28 @@ static int AVI_ChunkRead_list( input_thread_t *p_input, } p_container->common.p_last = p_chk; - if( AVI_ChunkRead( p_input, p_chk, p_container, b_seekable ) || - ( AVI_TellAbsolute( p_input ) >= - (off_t)p_chk->common.p_father->common.i_chunk_pos + - __EVEN( p_chk->common.p_father->common.i_chunk_size ) ) ) + if( AVI_ChunkRead( s, p_chk, p_container ) ) { break; } + if( p_chk->common.p_father->common.i_chunk_size > 0 && + ( stream_Tell( s ) > + (off_t)p_chk->common.p_father->common.i_chunk_pos + + (off_t)__EVEN( p_chk->common.p_father->common.i_chunk_size ) ) ) + { + break; + } + /* If we can't seek then stop when we 've found LIST-movi */ if( p_chk->common.i_chunk_fourcc == AVIFOURCC_LIST && - p_chk->list.i_type == AVIFOURCC_movi && !b_seekable ) + p_chk->list.i_type == AVIFOURCC_movi && + ( !b_seekable || p_chk->common.i_chunk_size == 0 ) ) { break; } } - msg_Dbg( p_input, "", AVIFOURCC_PRINT( p_container->list.i_type ) ); + msg_Dbg( (vlc_object_t*)s, "", (char*)&p_container->list.i_type ); return VLC_SUCCESS; } @@ -407,37 +198,51 @@ static int AVI_ChunkRead_list( input_thread_t *p_input, uint8_t *p_read, *p_buff; \ if( !( p_read = p_buff = malloc(i_read ) ) ) \ { \ - return 0; \ + return VLC_EGENERIC; \ } \ - i_read = AVI_ReadData( p_input, p_read, i_read ); \ + i_read = stream_Read( s, p_read, i_read ); \ + if( i_read < (int64_t)__EVEN(p_chk->common.i_chunk_size ) + 8 ) \ + { \ + free( p_buff ); \ + return VLC_EGENERIC; \ + }\ p_read += 8; \ i_read -= 8 +#define AVI_READ( res, func, size ) \ + if( i_read < size ) { \ + free( p_buff); \ + return VLC_EGENERIC; \ + } \ + i_read -= size; \ + res = func( p_read ); \ + p_read += size \ + #define AVI_READCHUNK_EXIT( code ) \ free( p_buff ); \ - if( i_read < 0 ) \ - { \ - msg_Warn( p_input, "not enough data" ); \ - } \ return code + +static inline uint8_t GetB( uint8_t *ptr ) +{ + return *ptr; +} + +#define AVI_READ1BYTE( i_byte ) \ + AVI_READ( i_byte, GetB, 1 ) + #define AVI_READ2BYTES( i_word ) \ - i_word = GetWLE( p_read ); \ - p_read += 2; \ - i_read -= 2 + AVI_READ( i_word, GetWLE, 2 ) #define AVI_READ4BYTES( i_dword ) \ - i_dword = GetDWLE( p_read ); \ - p_read += 4; \ - i_read -= 4 + AVI_READ( i_dword, GetDWLE, 4 ) + +#define AVI_READ8BYTES( i_qword ) \ + AVI_READ( i_qword, GetQWLE, 8 ) #define AVI_READFOURCC( i_dword ) \ - i_dword = GetFOURCC( p_read ); \ - p_read += 4; \ - i_read -= 4 + AVI_READ( i_dword, GetFOURCC, 4 ) -static int AVI_ChunkRead_avih( input_thread_t *p_input, - avi_chunk_t *p_chk, - vlc_bool_t b_seekable ) +static int AVI_ChunkRead_avih( stream_t *s, avi_chunk_t *p_chk ) { AVI_READCHUNK_ENTER; @@ -456,21 +261,19 @@ static int AVI_ChunkRead_avih( input_thread_t *p_input, AVI_READ4BYTES( p_chk->avih.i_start ); AVI_READ4BYTES( p_chk->avih.i_length ); #ifdef AVI_DEBUG - msg_Dbg( p_input, - "avih: streams:%d flags:%s%s%s%s %dx%d", + msg_Dbg( (vlc_object_t*)s, + "avih: streams:%d flags:%s%s%s%s %dx%d", p_chk->avih.i_streams, p_chk->avih.i_flags&AVIF_HASINDEX?" HAS_INDEX":"", p_chk->avih.i_flags&AVIF_MUSTUSEINDEX?" MUST_USE_INDEX":"", p_chk->avih.i_flags&AVIF_ISINTERLEAVED?" IS_INTERLEAVED":"", p_chk->avih.i_flags&AVIF_TRUSTCKTYPE?" TRUST_CKTYPE":"", p_chk->avih.i_width, p_chk->avih.i_height ); -#endif +#endif AVI_READCHUNK_EXIT( VLC_SUCCESS ); } -static int AVI_ChunkRead_strh( input_thread_t *p_input, - avi_chunk_t *p_chk, - vlc_bool_t b_seekable ) +static int AVI_ChunkRead_strh( stream_t *s, avi_chunk_t *p_chk ) { AVI_READCHUNK_ENTER; @@ -487,40 +290,39 @@ static int AVI_ChunkRead_strh( input_thread_t *p_input, AVI_READ4BYTES( p_chk->strh.i_quality ); AVI_READ4BYTES( p_chk->strh.i_samplesize ); #ifdef AVI_DEBUG - msg_Dbg( p_input, - "strh: type:%c%c%c%c handler:0x%8.8x samplesize:%d %.2ffps", - AVIFOURCC_PRINT( p_chk->strh.i_type ), + msg_Dbg( (vlc_object_t*)s, + "strh: type:%4.4s handler:0x%8.8x samplesize:%d %.2ffps", + (char*)&p_chk->strh.i_type, p_chk->strh.i_handler, p_chk->strh.i_samplesize, - ( p_chk->strh.i_scale ? + ( p_chk->strh.i_scale ? (float)p_chk->strh.i_rate / (float)p_chk->strh.i_scale : -1) ); #endif AVI_READCHUNK_EXIT( VLC_SUCCESS ); } -static int AVI_ChunkRead_strf( input_thread_t *p_input, - avi_chunk_t *p_chk, - vlc_bool_t b_seekable ) +static int AVI_ChunkRead_strf( stream_t *s, avi_chunk_t *p_chk ) { avi_chunk_t *p_strh; AVI_READCHUNK_ENTER; if( p_chk->common.p_father == NULL ) { - msg_Err( p_input, "malformed avi file" ); + msg_Err( (vlc_object_t*)s, "malformed avi file" ); AVI_READCHUNK_EXIT( VLC_EGENERIC ); } if( !( p_strh = AVI_ChunkFind( p_chk->common.p_father, AVIFOURCC_strh, 0 ) ) ) { - msg_Err( p_input, "malformed avi file" ); + msg_Err( (vlc_object_t*)s, "malformed avi file" ); AVI_READCHUNK_EXIT( VLC_EGENERIC ); } switch( p_strh->strh.i_type ) { case( AVIFOURCC_auds ): - p_chk->strf.auds.p_wf = malloc( p_chk->common.i_chunk_size ); + p_chk->strf.auds.i_cat = AUDIO_ES; + p_chk->strf.auds.p_wf = malloc( __MAX( p_chk->common.i_chunk_size, sizeof( WAVEFORMATEX ) ) ); AVI_READ2BYTES( p_chk->strf.auds.p_wf->wFormatTag ); AVI_READ2BYTES( p_chk->strf.auds.p_wf->nChannels ); AVI_READ4BYTES( p_chk->strf.auds.p_wf->nSamplesPerSec ); @@ -538,6 +340,12 @@ static int AVI_ChunkRead_strf( input_thread_t *p_input, p_chk->strf.auds.p_wf->cbSize = p_chk->common.i_chunk_size - sizeof( WAVEFORMATEX ); } + if( p_chk->strf.auds.p_wf->wFormatTag == WAVE_FORMAT_EXTENSIBLE ) + { + /* Found an extensible header atm almost nothing uses that. */ + msg_Warn( (vlc_object_t*)s, "WAVE_FORMAT_EXTENSIBLE or " + "vorbis audio dectected: not supported" ); + } } else { @@ -546,11 +354,11 @@ static int AVI_ChunkRead_strf( input_thread_t *p_input, if( p_chk->strf.auds.p_wf->cbSize > 0 ) { memcpy( &p_chk->strf.auds.p_wf[1] , - p_buff + 8 + sizeof( WAVEFORMATEX ), // 8=fourrc+size + p_buff + 8 + sizeof( WAVEFORMATEX ), /* 8=fourrc+size */ p_chk->strf.auds.p_wf->cbSize ); } #ifdef AVI_DEBUG - msg_Dbg( p_input, + msg_Dbg( (vlc_object_t*)s, "strf: audio:0x%4.4x channels:%d %dHz %dbits/sample %dkb/s", p_chk->strf.auds.p_wf->wFormatTag, p_chk->strf.auds.p_wf->nChannels, @@ -560,7 +368,8 @@ static int AVI_ChunkRead_strf( input_thread_t *p_input, #endif break; case( AVIFOURCC_vids ): - p_strh->strh.i_samplesize = 0; // XXX for ffmpeg avi file + p_strh->strh.i_samplesize = 0; /* XXX for ffmpeg avi file */ + p_chk->strf.vids.i_cat = VIDEO_ES; p_chk->strf.vids.p_bih = malloc( p_chk->common.i_chunk_size ); AVI_READ4BYTES( p_chk->strf.vids.p_bih->biSize ); AVI_READ4BYTES( p_chk->strf.vids.p_bih->biWidth ); @@ -573,22 +382,20 @@ static int AVI_ChunkRead_strf( input_thread_t *p_input, AVI_READ4BYTES( p_chk->strf.vids.p_bih->biYPelsPerMeter ); AVI_READ4BYTES( p_chk->strf.vids.p_bih->biClrUsed ); AVI_READ4BYTES( p_chk->strf.vids.p_bih->biClrImportant ); - if( p_chk->strf.vids.p_bih->biSize > - p_chk->common.i_chunk_size ) + if( p_chk->strf.vids.p_bih->biSize > p_chk->common.i_chunk_size ) { p_chk->strf.vids.p_bih->biSize = p_chk->common.i_chunk_size; } if( p_chk->strf.vids.p_bih->biSize - sizeof(BITMAPINFOHEADER) > 0 ) { memcpy( &p_chk->strf.vids.p_bih[1], - p_buff + 8 + sizeof(BITMAPINFOHEADER), // 8=fourrc+size - p_chk->strf.vids.p_bih->biSize - - sizeof(BITMAPINFOHEADER) ); + p_buff + 8 + sizeof(BITMAPINFOHEADER), /* 8=fourrc+size */ + p_chk->common.i_chunk_size -sizeof(BITMAPINFOHEADER) ); } #ifdef AVI_DEBUG - msg_Dbg( p_input, - "strf: video:%c%c%c%c %dx%d planes:%d %dbpp", - AVIFOURCC_PRINT( p_chk->strf.vids.p_bih->biCompression ), + msg_Dbg( (vlc_object_t*)s, + "strf: video:%4.4s %dx%d planes:%d %dbpp", + (char*)&p_chk->strf.vids.p_bih->biCompression, p_chk->strf.vids.p_bih->biWidth, p_chk->strf.vids.p_bih->biHeight, p_chk->strf.vids.p_bih->biPlanes, @@ -596,38 +403,45 @@ static int AVI_ChunkRead_strf( input_thread_t *p_input, #endif break; default: - msg_Warn( p_input, "unknown stream type" ); + msg_Warn( (vlc_object_t*)s, "unknown stream type" ); + p_chk->strf.common.i_cat = UNKNOWN_ES; break; } AVI_READCHUNK_EXIT( VLC_SUCCESS ); } -static void AVI_ChunkFree_strf( input_thread_t *p_input, - avi_chunk_t *p_chk ) +static void AVI_ChunkFree_strf( avi_chunk_t *p_chk ) { - + avi_chunk_strf_t *p_strf = (avi_chunk_strf_t*)p_chk; + if( p_strf->common.i_cat == AUDIO_ES ) + { + FREENULL( p_strf->auds.p_wf ); + } + else if( p_strf->common.i_cat == VIDEO_ES ) + { + FREENULL( p_strf->vids.p_bih ); + } } -static int AVI_ChunkRead_strd( input_thread_t *p_input, - avi_chunk_t *p_chk, - vlc_bool_t b_seekable ) +static int AVI_ChunkRead_strd( stream_t *s, avi_chunk_t *p_chk ) { AVI_READCHUNK_ENTER; p_chk->strd.p_data = malloc( p_chk->common.i_chunk_size ); - memcpy( p_chk->strd.p_data, - p_buff + 8, - p_chk->common.i_chunk_size ); + memcpy( p_chk->strd.p_data, p_buff + 8, p_chk->common.i_chunk_size ); AVI_READCHUNK_EXIT( VLC_SUCCESS ); } -static int AVI_ChunkRead_idx1( input_thread_t *p_input, - avi_chunk_t *p_chk, - vlc_bool_t b_seekable ) +static void AVI_ChunkFree_strd( avi_chunk_t *p_chk ) +{ + if( p_chk->strd.p_data ) free( p_chk->strd.p_data ); +} + +static int AVI_ChunkRead_idx1( stream_t *s, avi_chunk_t *p_chk ) { unsigned int i_count, i_index; AVI_READCHUNK_ENTER; - i_count = __MIN( p_chk->common.i_chunk_size, i_read ) / 16; + i_count = __MIN( (int64_t)p_chk->common.i_chunk_size, i_read ) / 16; p_chk->idx1.i_entry_count = i_count; p_chk->idx1.i_entry_max = i_count; @@ -648,23 +462,111 @@ static int AVI_ChunkRead_idx1( input_thread_t *p_input, p_chk->idx1.entry = NULL; } #ifdef AVI_DEBUG - msg_Dbg( p_input, "idx1: index entry:%d", i_count ); + msg_Dbg( (vlc_object_t*)s, "idx1: index entry:%d", i_count ); #endif AVI_READCHUNK_EXIT( VLC_SUCCESS ); } -static void AVI_ChunkFree_idx1( input_thread_t *p_input, - avi_chunk_t *p_chk ) +static void AVI_ChunkFree_idx1( avi_chunk_t *p_chk ) { p_chk->idx1.i_entry_count = 0; p_chk->idx1.i_entry_max = 0; - FREE( p_chk->idx1.entry ) + FREENULL( p_chk->idx1.entry ) } -static struct + + +static int AVI_ChunkRead_indx( stream_t *s, avi_chunk_t *p_chk ) +{ + unsigned int i_count, i; + int32_t i_dummy; + avi_chunk_indx_t *p_indx = (avi_chunk_indx_t*)p_chk; + + AVI_READCHUNK_ENTER; + + AVI_READ2BYTES( p_indx->i_longsperentry ); + AVI_READ1BYTE ( p_indx->i_indexsubtype ); + AVI_READ1BYTE ( p_indx->i_indextype ); + AVI_READ4BYTES( p_indx->i_entriesinuse ); + + AVI_READ4BYTES( p_indx->i_id ); + p_indx->idx.std = NULL; + p_indx->idx.field = NULL; + p_indx->idx.super = NULL; + + if( p_indx->i_indextype == AVI_INDEX_OF_CHUNKS && p_indx->i_indexsubtype == 0 ) + { + AVI_READ8BYTES( p_indx->i_baseoffset ); + AVI_READ4BYTES( i_dummy ); + + i_count = __MIN( p_indx->i_entriesinuse, i_read / 8 ); + p_indx->i_entriesinuse = i_count; + p_indx->idx.std = calloc( sizeof( indx_std_entry_t ), i_count ); + + for( i = 0; i < i_count; i++ ) + { + AVI_READ4BYTES( p_indx->idx.std[i].i_offset ); + AVI_READ4BYTES( p_indx->idx.std[i].i_size ); + } + } + else if( p_indx->i_indextype == AVI_INDEX_OF_CHUNKS && p_indx->i_indexsubtype == AVI_INDEX_2FIELD ) + { + AVI_READ8BYTES( p_indx->i_baseoffset ); + AVI_READ4BYTES( i_dummy ); + + i_count = __MIN( p_indx->i_entriesinuse, i_read / 12 ); + p_indx->i_entriesinuse = i_count; + p_indx->idx.field = calloc( sizeof( indx_field_entry_t ), i_count ); + for( i = 0; i < i_count; i++ ) + { + AVI_READ4BYTES( p_indx->idx.field[i].i_offset ); + AVI_READ4BYTES( p_indx->idx.field[i].i_size ); + AVI_READ4BYTES( p_indx->idx.field[i].i_offsetfield2 ); + } + } + else if( p_indx->i_indextype == AVI_INDEX_OF_INDEXES ) + { + p_indx->i_baseoffset = 0; + AVI_READ4BYTES( i_dummy ); + AVI_READ4BYTES( i_dummy ); + AVI_READ4BYTES( i_dummy ); + + i_count = __MIN( p_indx->i_entriesinuse, i_read / 16 ); + p_indx->i_entriesinuse = i_count; + p_indx->idx.super = calloc( sizeof( indx_super_entry_t ), i_count ); + + for( i = 0; i < i_count; i++ ) + { + AVI_READ8BYTES( p_indx->idx.super[i].i_offset ); + AVI_READ4BYTES( p_indx->idx.super[i].i_size ); + AVI_READ4BYTES( p_indx->idx.super[i].i_duration ); + } + } + else + { + msg_Warn( (vlc_object_t*)s, "unknow type/subtype index" ); + } + +#ifdef AVI_DEBUG + msg_Dbg( (vlc_object_t*)s, "indx: type=%d subtype=%d entry=%d", p_indx->i_indextype, p_indx->i_indexsubtype, p_indx->i_entriesinuse ); +#endif + AVI_READCHUNK_EXIT( VLC_SUCCESS ); +} +static void AVI_ChunkFree_indx( avi_chunk_t *p_chk ) +{ + avi_chunk_indx_t *p_indx = (avi_chunk_indx_t*)p_chk; + + FREENULL( p_indx->idx.std ); + FREENULL( p_indx->idx.field ); + FREENULL( p_indx->idx.super ); +} + + + +static struct { vlc_fourcc_t i_fourcc; - char *psz_type; + const char *psz_type; } AVI_strz_type[] = { { AVIFOURCC_IARL, "archive location" }, @@ -676,7 +578,7 @@ static struct { AVIFOURCC_ICRP, "cropped" }, { AVIFOURCC_IDIM, "dimensions" }, { AVIFOURCC_IDPI, "dots per inch" }, - { AVIFOURCC_IENG, "enginner" }, + { AVIFOURCC_IENG, "engineer" }, { AVIFOURCC_IGNR, "genre" }, { AVIFOURCC_IKEY, "keywords" }, { AVIFOURCC_ILGT, "lightness" }, @@ -692,11 +594,10 @@ static struct { AVIFOURCC_ITCH, "technician" }, { AVIFOURCC_ISMP, "time code" }, { AVIFOURCC_IDIT, "digitalization time" }, + { AVIFOURCC_strn, "stream name" }, { 0, "???" } }; -static int AVI_ChunkRead_strz( input_thread_t *p_input, - avi_chunk_t *p_chk, - vlc_bool_t b_seekable ) +static int AVI_ChunkRead_strz( stream_t *s, avi_chunk_t *p_chk ) { int i_index; avi_chunk_STRING_t *p_strz = (avi_chunk_STRING_t*)p_chk; @@ -720,27 +621,23 @@ static int AVI_ChunkRead_strz( input_thread_t *p_input, p_strz->p_str[i_read] = 0; #ifdef AVI_DEBUG - msg_Dbg( p_input, "%c%c%c%c: %s : %s", - AVIFOURCC_PRINT( p_strz->i_chunk_fourcc), p_strz->p_type, p_strz->p_str); + msg_Dbg( (vlc_object_t*)s, "%4.4s: %s : %s", + (char*)&p_strz->i_chunk_fourcc, p_strz->p_type, p_strz->p_str); #endif AVI_READCHUNK_EXIT( VLC_SUCCESS ); } -static void AVI_ChunkFree_strz( input_thread_t *p_input, - avi_chunk_t *p_chk ) +static void AVI_ChunkFree_strz( avi_chunk_t *p_chk ) { avi_chunk_STRING_t *p_strz = (avi_chunk_STRING_t*)p_chk; - FREE( p_strz->p_type ); - FREE( p_strz->p_str ); + FREENULL( p_strz->p_type ); + FREENULL( p_strz->p_str ); } -static int AVI_ChunkRead_nothing( input_thread_t *p_input, - avi_chunk_t *p_chk, - vlc_bool_t b_seekable ) +static int AVI_ChunkRead_nothing( stream_t *s, avi_chunk_t *p_chk ) { - return AVI_NextChunk( p_input, p_chk ); + return AVI_NextChunk( s, p_chk ); } -static void AVI_ChunkFree_nothing( input_thread_t *p_input, - avi_chunk_t *p_chk ) +static void AVI_ChunkFree_nothing( avi_chunk_t *p_chk ) { } @@ -748,11 +645,8 @@ static void AVI_ChunkFree_nothing( input_thread_t *p_input, static struct { vlc_fourcc_t i_fourcc; - int (*AVI_ChunkRead_function)( input_thread_t *p_input, - avi_chunk_t *p_chk, - vlc_bool_t b_seekable ); - void (*AVI_ChunkFree_function)( input_thread_t *p_input, - avi_chunk_t *p_chk ); + int (*AVI_ChunkRead_function)( stream_t *s, avi_chunk_t *p_chk ); + void (*AVI_ChunkFree_function)( avi_chunk_t *p_chk ); } AVI_Chunk_Function [] = { { AVIFOURCC_RIFF, AVI_ChunkRead_list, AVI_ChunkFree_nothing }, @@ -760,8 +654,9 @@ static struct { AVIFOURCC_avih, AVI_ChunkRead_avih, AVI_ChunkFree_nothing }, { AVIFOURCC_strh, AVI_ChunkRead_strh, AVI_ChunkFree_nothing }, { AVIFOURCC_strf, AVI_ChunkRead_strf, AVI_ChunkFree_strf }, - { AVIFOURCC_strd, AVI_ChunkRead_strd, AVI_ChunkFree_nothing }, + { AVIFOURCC_strd, AVI_ChunkRead_strd, AVI_ChunkFree_strd }, { AVIFOURCC_idx1, AVI_ChunkRead_idx1, AVI_ChunkFree_idx1 }, + { AVIFOURCC_indx, AVI_ChunkRead_indx, AVI_ChunkFree_indx }, { AVIFOURCC_JUNK, AVI_ChunkRead_nothing, AVI_ChunkFree_nothing }, { AVIFOURCC_IARL, AVI_ChunkRead_strz, AVI_ChunkFree_strz }, @@ -790,6 +685,7 @@ static struct { AVIFOURCC_ITCH, AVI_ChunkRead_strz, AVI_ChunkFree_strz }, { AVIFOURCC_ISMP, AVI_ChunkRead_strz, AVI_ChunkFree_strz }, { AVIFOURCC_IDIT, AVI_ChunkRead_strz, AVI_ChunkFree_strz }, + { AVIFOURCC_strn, AVI_ChunkRead_strz, AVI_ChunkFree_strz }, { 0, NULL, NULL } }; @@ -806,10 +702,7 @@ static int AVI_ChunkFunctionFind( vlc_fourcc_t i_fourcc ) } } -int _AVI_ChunkRead( input_thread_t *p_input, - avi_chunk_t *p_chk, - avi_chunk_t *p_father, - vlc_bool_t b_seekable ) +int _AVI_ChunkRead( stream_t *s, avi_chunk_t *p_chk, avi_chunk_t *p_father ) { int i_index; @@ -818,14 +711,14 @@ int _AVI_ChunkRead( input_thread_t *p_input, return VLC_EGENERIC; } - if( AVI_ChunkReadCommon( p_input, p_chk ) ) + if( AVI_ChunkReadCommon( s, p_chk ) ) { - msg_Warn( p_input, "cannot read one chunk" ); + msg_Warn( (vlc_object_t*)s, "cannot read one chunk" ); return VLC_EGENERIC; } if( p_chk->common.i_chunk_fourcc == VLC_FOURCC( 0, 0, 0, 0 ) ) { - msg_Warn( p_input, "found null fourcc chunk (corrupted file?)" ); + msg_Warn( (vlc_object_t*)s, "found null fourcc chunk (corrupted file?)" ); return VLC_EGENERIC; } p_chk->common.p_father = p_father; @@ -833,15 +726,21 @@ int _AVI_ChunkRead( input_thread_t *p_input, i_index = AVI_ChunkFunctionFind( p_chk->common.i_chunk_fourcc ); if( AVI_Chunk_Function[i_index].AVI_ChunkRead_function ) { - return AVI_Chunk_Function[i_index]. - AVI_ChunkRead_function( p_input, p_chk, b_seekable ); + return AVI_Chunk_Function[i_index].AVI_ChunkRead_function( s, p_chk ); } - - msg_Warn( p_input, "unknown chunk (not loaded)" ); - return AVI_NextChunk( p_input, p_chk ); + else if( ( ((char*)&p_chk->common.i_chunk_fourcc)[0] == 'i' && + ((char*)&p_chk->common.i_chunk_fourcc)[1] == 'x' ) || + ( ((char*)&p_chk->common.i_chunk_fourcc)[2] == 'i' && + ((char*)&p_chk->common.i_chunk_fourcc)[3] == 'x' ) ) + { + p_chk->common.i_chunk_fourcc = AVIFOURCC_indx; + return AVI_ChunkRead_indx( s, p_chk ); + } + msg_Warn( (vlc_object_t*)s, "unknown chunk (not loaded)" ); + return AVI_NextChunk( s, p_chk ); } -void _AVI_ChunkFree( input_thread_t *p_input, +void _AVI_ChunkFree( stream_t *s, avi_chunk_t *p_chk ) { int i_index; @@ -857,7 +756,7 @@ void _AVI_ChunkFree( input_thread_t *p_input, while( p_child ) { p_next = p_child->common.p_next; - AVI_ChunkFree( p_input, p_child ); + AVI_ChunkFree( s, p_child ); free( p_child ); p_child = p_next; } @@ -866,14 +765,14 @@ void _AVI_ChunkFree( input_thread_t *p_input, if( AVI_Chunk_Function[i_index].AVI_ChunkFree_function ) { #ifdef AVI_DEBUG - msg_Dbg( p_input, "free chunk %c%c%c%c", - AVIFOURCC_PRINT( p_chk->common.i_chunk_fourcc ) ); + msg_Dbg( (vlc_object_t*)s, "free chunk %4.4s", + (char*)&p_chk->common.i_chunk_fourcc ); #endif - AVI_Chunk_Function[i_index].AVI_ChunkFree_function( p_input, p_chk); + AVI_Chunk_Function[i_index].AVI_ChunkFree_function( p_chk); } else { - msg_Warn( p_input, "unknown chunk (not unloaded)" ); + msg_Warn( (vlc_object_t*)s, "unknown chunk (not unloaded)" ); } p_chk->common.p_first = NULL; p_chk->common.p_last = NULL; @@ -881,15 +780,57 @@ void _AVI_ChunkFree( input_thread_t *p_input, return; } -int AVI_ChunkReadRoot( input_thread_t *p_input, - avi_chunk_t *p_root, - vlc_bool_t b_seekable ) +static void AVI_ChunkDumpDebug_level( vlc_object_t *p_obj, + avi_chunk_t *p_chk, int i_level ) +{ + char str[1024]; + int i; + avi_chunk_t *p_child; + + memset( str, ' ', sizeof( str ) ); + for( i = 1; i < i_level; i++ ) + { + str[i * 5] = '|'; + } + if( p_chk->common.i_chunk_fourcc == AVIFOURCC_RIFF|| + p_chk->common.i_chunk_fourcc == AVIFOURCC_LIST ) + { + sprintf( str + i_level * 5, + "%c %4.4s-%4.4s size:"I64Fu" pos:"I64Fu, + i_level ? '+' : '*', + (char*)&p_chk->common.i_chunk_fourcc, + (char*)&p_chk->list.i_type, + p_chk->common.i_chunk_size, + p_chk->common.i_chunk_pos ); + } + else + { + sprintf( str + i_level * 5, + "+ %4.4s size:"I64Fu" pos:"I64Fu, + (char*)&p_chk->common.i_chunk_fourcc, + p_chk->common.i_chunk_size, + p_chk->common.i_chunk_pos ); + } + msg_Dbg( p_obj, "%s", str ); + + p_child = p_chk->common.p_first; + while( p_child ) + { + AVI_ChunkDumpDebug_level( p_obj, p_child, i_level + 1 ); + p_child = p_child->common.p_next; + } +} + +int AVI_ChunkReadRoot( stream_t *s, avi_chunk_t *p_root ) { avi_chunk_list_t *p_list = (avi_chunk_list_t*)p_root; avi_chunk_t *p_chk; + vlc_bool_t b_seekable; + + stream_Control( s, STREAM_CAN_FASTSEEK, &b_seekable ); p_list->i_chunk_pos = 0; - p_list->i_chunk_size = p_input->stream.p_selected_area->i_size; + p_list->i_chunk_size = stream_Size( s ); p_list->i_chunk_fourcc = AVIFOURCC_LIST; p_list->p_father = NULL; p_list->p_next = NULL; @@ -912,10 +853,10 @@ int AVI_ChunkReadRoot( input_thread_t *p_input, } p_root->common.p_last = p_chk; - if( AVI_ChunkRead( p_input, p_chk, p_root, b_seekable ) || - ( AVI_TellAbsolute( p_input ) >= - (off_t)p_chk->common.p_father->common.i_chunk_pos + - __EVEN( p_chk->common.p_father->common.i_chunk_size ) ) ) + if( AVI_ChunkRead( s, p_chk, p_root ) || + ( stream_Tell( s ) >= + (off_t)p_chk->common.p_father->common.i_chunk_pos + + (off_t)__EVEN( p_chk->common.p_father->common.i_chunk_size ) ) ) { break; } @@ -925,15 +866,16 @@ int AVI_ChunkReadRoot( input_thread_t *p_input, { break; } - } + } + AVI_ChunkDumpDebug_level( (vlc_object_t*)s, p_root, 0 ); return VLC_SUCCESS; } -void AVI_ChunkFreeRoot( input_thread_t *p_input, +void AVI_ChunkFreeRoot( stream_t *s, avi_chunk_t *p_chk ) { - AVI_ChunkFree( p_input, p_chk ); + AVI_ChunkFree( s, p_chk ); } @@ -952,7 +894,7 @@ int _AVI_ChunkCount( avi_chunk_t *p_chk, vlc_fourcc_t i_fourcc ) while( p_child ) { if( p_child->common.i_chunk_fourcc == i_fourcc || - ( p_child->common.i_chunk_fourcc == AVIFOURCC_LIST && + ( p_child->common.i_chunk_fourcc == AVIFOURCC_LIST && p_child->list.i_type == i_fourcc ) ) { i_count++; @@ -962,8 +904,8 @@ int _AVI_ChunkCount( avi_chunk_t *p_chk, vlc_fourcc_t i_fourcc ) return i_count; } -avi_chunk_t *_AVI_ChunkFind( avi_chunk_t *p_chk, - vlc_fourcc_t i_fourcc, int i_number ) +void *_AVI_ChunkFind( avi_chunk_t *p_chk, + vlc_fourcc_t i_fourcc, int i_number ) { avi_chunk_t *p_child; if( !p_chk ) @@ -975,7 +917,7 @@ avi_chunk_t *_AVI_ChunkFind( avi_chunk_t *p_chk, while( p_child ) { if( p_child->common.i_chunk_fourcc == i_fourcc || - ( p_child->common.i_chunk_fourcc == AVIFOURCC_LIST && + ( p_child->common.i_chunk_fourcc == AVIFOURCC_LIST && p_child->list.i_type == i_fourcc ) ) { if( i_number == 0 ) @@ -991,49 +933,4 @@ avi_chunk_t *_AVI_ChunkFind( avi_chunk_t *p_chk, return NULL; } -static void AVI_ChunkDumpDebug_level( input_thread_t *p_input, - avi_chunk_t *p_chk, int i_level ) -{ - char str[1024]; - int i; - avi_chunk_t *p_child; - - memset( str, ' ', sizeof( str ) ); - for( i = 1; i < i_level; i++ ) - { - str[i * 5] = '|'; - } - if( p_chk->common.i_chunk_fourcc == AVIFOURCC_RIFF|| - p_chk->common.i_chunk_fourcc == AVIFOURCC_LIST ) - { - sprintf( str + i_level * 5, - "%c %c%c%c%c-%c%c%c%c size:"I64Fu" pos:"I64Fu, - i_level ? '+' : '*', - AVIFOURCC_PRINT( p_chk->common.i_chunk_fourcc ), - AVIFOURCC_PRINT( p_chk->list.i_type ), - p_chk->common.i_chunk_size, - p_chk->common.i_chunk_pos ); - } - else - { - sprintf( str + i_level * 5, - "+ %c%c%c%c size:"I64Fu" pos:"I64Fu, - AVIFOURCC_PRINT( p_chk->common.i_chunk_fourcc ), - p_chk->common.i_chunk_size, - p_chk->common.i_chunk_pos ); - } - msg_Dbg( p_input, "%s", str ); - - p_child = p_chk->common.p_first; - while( p_child ) - { - AVI_ChunkDumpDebug_level( p_input, p_child, i_level + 1 ); - p_child = p_child->common.p_next; - } -} -void _AVI_ChunkDumpDebug( input_thread_t *p_input, - avi_chunk_t *p_chk ) -{ - AVI_ChunkDumpDebug_level( p_input, p_chk, 0 ); -}