8ef41156383a6b7149d24f08d0f2a4776fe0cfa1
[vlc.git] / modules / demux / avi / libavi.c
1 /*****************************************************************************
2  * libavi.c : LibAVI
3  *****************************************************************************
4  * Copyright (C) 2001 the VideoLAN team
5  * $Id$
6  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation; either version 2 of the License, or
11  * (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  *
18  * You should have received a copy of the GNU General Public License
19  * along with this program; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
21  *****************************************************************************/
22
23
24 #include <vlc/vlc.h>
25 #include <vlc_demux.h>
26 #include <vlc_codecs.h>                                /* BITMAPINFOHEADER */
27
28 #include "libavi.h"
29
30 #define AVI_DEBUG 1
31
32 #define __EVEN( x ) ( (x)&0x01 ? (x)+1 : (x) )
33
34 static vlc_fourcc_t GetFOURCC( const byte_t *p_buff )
35 {
36     return VLC_FOURCC( p_buff[0], p_buff[1], p_buff[2], p_buff[3] );
37 }
38
39 #define AVI_ChunkFree( a, b ) _AVI_ChunkFree( (a), (avi_chunk_t*)(b) )
40 void    _AVI_ChunkFree( stream_t *, avi_chunk_t *p_chk );
41
42 /****************************************************************************
43  *
44  * Basics functions to manipulates chunks
45  *
46  ****************************************************************************/
47 static int AVI_ChunkReadCommon( stream_t *s, avi_chunk_t *p_chk )
48 {
49     const uint8_t *p_peek;
50     int i_peek;
51
52     memset( p_chk, 0, sizeof( avi_chunk_t ) );
53
54     if( ( i_peek = stream_Peek( s, &p_peek, 8 ) ) < 8 )
55     {
56         return VLC_EGENERIC;
57     }
58
59     p_chk->common.i_chunk_fourcc = GetFOURCC( p_peek );
60     p_chk->common.i_chunk_size   = GetDWLE( p_peek + 4 );
61     p_chk->common.i_chunk_pos    = stream_Tell( s );
62
63     p_chk->common.p_father = NULL;
64     p_chk->common.p_next = NULL;
65     p_chk->common.p_first = NULL;
66     p_chk->common.p_next = NULL;
67
68 #ifdef AVI_DEBUG
69     msg_Dbg( (vlc_object_t*)s,
70              "found Chunk fourcc:%8.8x (%4.4s) size:"I64Fd" pos:"I64Fd,
71              p_chk->common.i_chunk_fourcc,
72              (char*)&p_chk->common.i_chunk_fourcc,
73              p_chk->common.i_chunk_size,
74              p_chk->common.i_chunk_pos );
75 #endif
76     return VLC_SUCCESS;
77 }
78
79 static int AVI_NextChunk( stream_t *s, avi_chunk_t *p_chk )
80 {
81     avi_chunk_t chk;
82
83     if( !p_chk )
84     {
85         if( AVI_ChunkReadCommon( s, &chk ) )
86         {
87             return VLC_EGENERIC;
88         }
89         p_chk = &chk;
90     }
91
92     if( p_chk->common.p_father )
93     {
94         if( p_chk->common.p_father->common.i_chunk_pos +
95                 __EVEN( p_chk->common.p_father->common.i_chunk_size ) + 8 <
96             p_chk->common.i_chunk_pos +
97                 __EVEN( p_chk->common.i_chunk_size ) + 8 )
98         {
99             return VLC_EGENERIC;
100         }
101     }
102     return stream_Seek( s, p_chk->common.i_chunk_pos +
103                                  __EVEN( p_chk->common.i_chunk_size ) + 8 );
104 }
105
106 /****************************************************************************
107  *
108  * Functions to read chunks
109  *
110  ****************************************************************************/
111 static int AVI_ChunkRead_list( stream_t *s, avi_chunk_t *p_container )
112 {
113     avi_chunk_t *p_chk;
114     const uint8_t *p_peek;
115     vlc_bool_t b_seekable;
116
117     if( p_container->common.i_chunk_size > 0 && p_container->common.i_chunk_size < 8 )
118     {
119         /* empty box */
120         msg_Warn( (vlc_object_t*)s, "empty list chunk" );
121         return VLC_EGENERIC;
122     }
123     if( stream_Peek( s, &p_peek, 12 ) < 12 )
124     {
125         msg_Warn( (vlc_object_t*)s, "cannot peek while reading list chunk" );
126         return VLC_EGENERIC;
127     }
128
129     stream_Control( s, STREAM_CAN_FASTSEEK, &b_seekable );
130
131     p_container->list.i_type = GetFOURCC( p_peek + 8 );
132
133     if( p_container->common.i_chunk_fourcc == AVIFOURCC_LIST &&
134         p_container->list.i_type == AVIFOURCC_movi )
135     {
136         msg_Dbg( (vlc_object_t*)s, "skipping movi chunk" );
137         if( b_seekable )
138         {
139             return AVI_NextChunk( s, p_container );
140         }
141         return VLC_SUCCESS; /* point at begining of LIST-movi */
142     }
143
144     if( stream_Read( s, NULL, 12 ) != 12 )
145     {
146         msg_Warn( (vlc_object_t*)s, "cannot enter chunk" );
147         return VLC_EGENERIC;
148     }
149
150 #ifdef AVI_DEBUG
151     msg_Dbg( (vlc_object_t*)s,
152              "found LIST chunk: \'%4.4s\'",
153              (char*)&p_container->list.i_type );
154 #endif
155     msg_Dbg( (vlc_object_t*)s, "<list \'%4.4s\'>", (char*)&p_container->list.i_type );
156     for( ; ; )
157     {
158         p_chk = malloc( sizeof( avi_chunk_t ) );
159         memset( p_chk, 0, sizeof( avi_chunk_t ) );
160         if( !p_container->common.p_first )
161         {
162             p_container->common.p_first = p_chk;
163         }
164         else
165         {
166             p_container->common.p_last->common.p_next = p_chk;
167         }
168         p_container->common.p_last = p_chk;
169
170         if( AVI_ChunkRead( s, p_chk, p_container ) )
171         {
172             break;
173         }
174         if( p_chk->common.p_father->common.i_chunk_size > 0 &&
175            ( stream_Tell( s ) >
176               (off_t)p_chk->common.p_father->common.i_chunk_pos +
177                (off_t)__EVEN( p_chk->common.p_father->common.i_chunk_size ) ) )
178         {
179             break;
180         }
181
182         /* If we can't seek then stop when we 've found LIST-movi */
183         if( p_chk->common.i_chunk_fourcc == AVIFOURCC_LIST &&
184             p_chk->list.i_type == AVIFOURCC_movi &&
185             ( !b_seekable || p_chk->common.i_chunk_size == 0 ) )
186         {
187             break;
188         }
189
190     }
191     msg_Dbg( (vlc_object_t*)s, "</list \'%4.4s\'>", (char*)&p_container->list.i_type );
192
193     return VLC_SUCCESS;
194 }
195
196 #define AVI_READCHUNK_ENTER \
197     int64_t i_read = __EVEN(p_chk->common.i_chunk_size ) + 8; \
198     uint8_t  *p_read, *p_buff;    \
199     if( !( p_read = p_buff = malloc(i_read ) ) ) \
200     { \
201         return VLC_EGENERIC; \
202     } \
203     i_read = stream_Read( s, p_read, i_read ); \
204     if( i_read < (int64_t)__EVEN(p_chk->common.i_chunk_size ) + 8 ) \
205     { \
206         free( p_buff ); \
207         return VLC_EGENERIC; \
208     }\
209     p_read += 8; \
210     i_read -= 8
211
212 #define AVI_READ( res, func, size ) \
213     if( i_read < size ) { \
214         free( p_buff); \
215         return VLC_EGENERIC; \
216     } \
217     i_read -= size; \
218     res = func( p_read ); \
219     p_read += size \
220
221 #define AVI_READCHUNK_EXIT( code ) \
222     free( p_buff ); \
223     return code
224
225 static inline uint8_t GetB( uint8_t *ptr )
226 {
227     return *ptr;
228 }
229
230 #define AVI_READ1BYTE( i_byte ) \
231     AVI_READ( i_byte, GetB, 1 )
232
233 #define AVI_READ2BYTES( i_word ) \
234     AVI_READ( i_word, GetWLE, 2 )
235
236 #define AVI_READ4BYTES( i_dword ) \
237     AVI_READ( i_dword, GetDWLE, 4 )
238
239 #define AVI_READ8BYTES( i_qword ) \
240     AVI_READ( i_qword, GetQWLE, 8 )
241
242 #define AVI_READFOURCC( i_dword ) \
243     AVI_READ( i_dword, GetFOURCC, 4 )
244
245 static int AVI_ChunkRead_avih( stream_t *s, avi_chunk_t *p_chk )
246 {
247     AVI_READCHUNK_ENTER;
248
249     AVI_READ4BYTES( p_chk->avih.i_microsecperframe);
250     AVI_READ4BYTES( p_chk->avih.i_maxbytespersec );
251     AVI_READ4BYTES( p_chk->avih.i_reserved1 );
252     AVI_READ4BYTES( p_chk->avih.i_flags );
253     AVI_READ4BYTES( p_chk->avih.i_totalframes );
254     AVI_READ4BYTES( p_chk->avih.i_initialframes );
255     AVI_READ4BYTES( p_chk->avih.i_streams );
256     AVI_READ4BYTES( p_chk->avih.i_suggestedbuffersize );
257     AVI_READ4BYTES( p_chk->avih.i_width );
258     AVI_READ4BYTES( p_chk->avih.i_height );
259     AVI_READ4BYTES( p_chk->avih.i_scale );
260     AVI_READ4BYTES( p_chk->avih.i_rate );
261     AVI_READ4BYTES( p_chk->avih.i_start );
262     AVI_READ4BYTES( p_chk->avih.i_length );
263 #ifdef AVI_DEBUG
264     msg_Dbg( (vlc_object_t*)s,
265              "avih: streams:%d flags:%s%s%s%s %dx%d",
266              p_chk->avih.i_streams,
267              p_chk->avih.i_flags&AVIF_HASINDEX?" HAS_INDEX":"",
268              p_chk->avih.i_flags&AVIF_MUSTUSEINDEX?" MUST_USE_INDEX":"",
269              p_chk->avih.i_flags&AVIF_ISINTERLEAVED?" IS_INTERLEAVED":"",
270              p_chk->avih.i_flags&AVIF_TRUSTCKTYPE?" TRUST_CKTYPE":"",
271              p_chk->avih.i_width, p_chk->avih.i_height );
272 #endif
273     AVI_READCHUNK_EXIT( VLC_SUCCESS );
274 }
275
276 static int AVI_ChunkRead_strh( stream_t *s, avi_chunk_t *p_chk )
277 {
278     AVI_READCHUNK_ENTER;
279
280     AVI_READFOURCC( p_chk->strh.i_type );
281     AVI_READFOURCC( p_chk->strh.i_handler );
282     AVI_READ4BYTES( p_chk->strh.i_flags );
283     AVI_READ4BYTES( p_chk->strh.i_reserved1 );
284     AVI_READ4BYTES( p_chk->strh.i_initialframes );
285     AVI_READ4BYTES( p_chk->strh.i_scale );
286     AVI_READ4BYTES( p_chk->strh.i_rate );
287     AVI_READ4BYTES( p_chk->strh.i_start );
288     AVI_READ4BYTES( p_chk->strh.i_length );
289     AVI_READ4BYTES( p_chk->strh.i_suggestedbuffersize );
290     AVI_READ4BYTES( p_chk->strh.i_quality );
291     AVI_READ4BYTES( p_chk->strh.i_samplesize );
292 #ifdef AVI_DEBUG
293     msg_Dbg( (vlc_object_t*)s,
294              "strh: type:%4.4s handler:0x%8.8x samplesize:%d %.2ffps",
295              (char*)&p_chk->strh.i_type,
296              p_chk->strh.i_handler,
297              p_chk->strh.i_samplesize,
298              ( p_chk->strh.i_scale ?
299                 (float)p_chk->strh.i_rate / (float)p_chk->strh.i_scale : -1) );
300 #endif
301
302     AVI_READCHUNK_EXIT( VLC_SUCCESS );
303 }
304
305 static int AVI_ChunkRead_strf( stream_t *s, avi_chunk_t *p_chk )
306 {
307     avi_chunk_t *p_strh;
308
309     AVI_READCHUNK_ENTER;
310     if( p_chk->common.p_father == NULL )
311     {
312         msg_Err( (vlc_object_t*)s, "malformed avi file" );
313         AVI_READCHUNK_EXIT( VLC_EGENERIC );
314     }
315     if( !( p_strh = AVI_ChunkFind( p_chk->common.p_father, AVIFOURCC_strh, 0 ) ) )
316     {
317         msg_Err( (vlc_object_t*)s, "malformed avi file" );
318         AVI_READCHUNK_EXIT( VLC_EGENERIC );
319     }
320
321     switch( p_strh->strh.i_type )
322     {
323         case( AVIFOURCC_auds ):
324             p_chk->strf.auds.i_cat = AUDIO_ES;
325             p_chk->strf.auds.p_wf = malloc( __MAX( p_chk->common.i_chunk_size, sizeof( WAVEFORMATEX ) ) );
326             AVI_READ2BYTES( p_chk->strf.auds.p_wf->wFormatTag );
327             AVI_READ2BYTES( p_chk->strf.auds.p_wf->nChannels );
328             AVI_READ4BYTES( p_chk->strf.auds.p_wf->nSamplesPerSec );
329             AVI_READ4BYTES( p_chk->strf.auds.p_wf->nAvgBytesPerSec );
330             AVI_READ2BYTES( p_chk->strf.auds.p_wf->nBlockAlign );
331             AVI_READ2BYTES( p_chk->strf.auds.p_wf->wBitsPerSample );
332             if( p_chk->strf.auds.p_wf->wFormatTag != WAVE_FORMAT_PCM
333                  && p_chk->common.i_chunk_size > sizeof( WAVEFORMATEX ) )
334             {
335                 AVI_READ2BYTES( p_chk->strf.auds.p_wf->cbSize );
336                 /* prevent segfault */
337                 if( p_chk->strf.auds.p_wf->cbSize >
338                         p_chk->common.i_chunk_size - sizeof( WAVEFORMATEX ) )
339                 {
340                     p_chk->strf.auds.p_wf->cbSize =
341                         p_chk->common.i_chunk_size - sizeof( WAVEFORMATEX );
342                 }
343                 if( p_chk->strf.auds.p_wf->wFormatTag == WAVE_FORMAT_EXTENSIBLE )
344                 {
345                     /* Found an extensible header atm almost nothing uses that. */
346                     msg_Warn( (vlc_object_t*)s, "WAVE_FORMAT_EXTENSIBLE or "
347                               "vorbis audio dectected: not supported" );
348                 }
349             }
350             else
351             {
352                 p_chk->strf.auds.p_wf->cbSize = 0;
353             }
354             if( p_chk->strf.auds.p_wf->cbSize > 0 )
355             {
356                 memcpy( &p_chk->strf.auds.p_wf[1] ,
357                         p_buff + 8 + sizeof( WAVEFORMATEX ),    /*  8=fourrc+size */
358                         p_chk->strf.auds.p_wf->cbSize );
359             }
360 #ifdef AVI_DEBUG
361             msg_Dbg( (vlc_object_t*)s,
362                      "strf: audio:0x%4.4x channels:%d %dHz %dbits/sample %dkb/s",
363                      p_chk->strf.auds.p_wf->wFormatTag,
364                      p_chk->strf.auds.p_wf->nChannels,
365                      p_chk->strf.auds.p_wf->nSamplesPerSec,
366                      p_chk->strf.auds.p_wf->wBitsPerSample,
367                      p_chk->strf.auds.p_wf->nAvgBytesPerSec * 8 / 1024 );
368 #endif
369             break;
370         case( AVIFOURCC_vids ):
371             p_strh->strh.i_samplesize = 0; /* XXX for ffmpeg avi file */
372             p_chk->strf.vids.i_cat = VIDEO_ES;
373             p_chk->strf.vids.p_bih = malloc( p_chk->common.i_chunk_size );
374             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biSize );
375             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biWidth );
376             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biHeight );
377             AVI_READ2BYTES( p_chk->strf.vids.p_bih->biPlanes );
378             AVI_READ2BYTES( p_chk->strf.vids.p_bih->biBitCount );
379             AVI_READFOURCC( p_chk->strf.vids.p_bih->biCompression );
380             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biSizeImage );
381             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biXPelsPerMeter );
382             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biYPelsPerMeter );
383             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biClrUsed );
384             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biClrImportant );
385             if( p_chk->strf.vids.p_bih->biSize > p_chk->common.i_chunk_size )
386             {
387                 p_chk->strf.vids.p_bih->biSize = p_chk->common.i_chunk_size;
388             }
389             if( p_chk->strf.vids.p_bih->biSize - sizeof(BITMAPINFOHEADER) > 0 )
390             {
391                 memcpy( &p_chk->strf.vids.p_bih[1],
392                         p_buff + 8 + sizeof(BITMAPINFOHEADER), /* 8=fourrc+size */
393                         p_chk->common.i_chunk_size -sizeof(BITMAPINFOHEADER) );
394             }
395 #ifdef AVI_DEBUG
396             msg_Dbg( (vlc_object_t*)s,
397                      "strf: video:%4.4s %dx%d planes:%d %dbpp",
398                      (char*)&p_chk->strf.vids.p_bih->biCompression,
399                      p_chk->strf.vids.p_bih->biWidth,
400                      p_chk->strf.vids.p_bih->biHeight,
401                      p_chk->strf.vids.p_bih->biPlanes,
402                      p_chk->strf.vids.p_bih->biBitCount );
403 #endif
404             break;
405         default:
406             msg_Warn( (vlc_object_t*)s, "unknown stream type" );
407             p_chk->strf.common.i_cat = UNKNOWN_ES;
408             break;
409     }
410     AVI_READCHUNK_EXIT( VLC_SUCCESS );
411 }
412 static void AVI_ChunkFree_strf( avi_chunk_t *p_chk )
413 {
414     avi_chunk_strf_t *p_strf = (avi_chunk_strf_t*)p_chk;
415     if( p_strf->common.i_cat == AUDIO_ES )
416     {
417         FREENULL( p_strf->auds.p_wf );
418     }
419     else if( p_strf->common.i_cat == VIDEO_ES )
420     {
421         FREENULL( p_strf->vids.p_bih );
422     }
423 }
424
425 static int AVI_ChunkRead_strd( stream_t *s, avi_chunk_t *p_chk )
426 {
427     AVI_READCHUNK_ENTER;
428     p_chk->strd.p_data = malloc( p_chk->common.i_chunk_size );
429     memcpy( p_chk->strd.p_data, p_buff + 8, p_chk->common.i_chunk_size );
430     AVI_READCHUNK_EXIT( VLC_SUCCESS );
431 }
432
433 static void AVI_ChunkFree_strd( avi_chunk_t *p_chk )
434 {
435     if( p_chk->strd.p_data ) free( p_chk->strd.p_data );
436 }
437
438 static int AVI_ChunkRead_idx1( stream_t *s, avi_chunk_t *p_chk )
439 {
440     unsigned int i_count, i_index;
441
442     AVI_READCHUNK_ENTER;
443
444     i_count = __MIN( (int64_t)p_chk->common.i_chunk_size, i_read ) / 16;
445
446     p_chk->idx1.i_entry_count = i_count;
447     p_chk->idx1.i_entry_max   = i_count;
448     if( i_count > 0 )
449     {
450         p_chk->idx1.entry = calloc( i_count, sizeof( idx1_entry_t ) );
451
452         for( i_index = 0; i_index < i_count ; i_index++ )
453         {
454             AVI_READFOURCC( p_chk->idx1.entry[i_index].i_fourcc );
455             AVI_READ4BYTES( p_chk->idx1.entry[i_index].i_flags );
456             AVI_READ4BYTES( p_chk->idx1.entry[i_index].i_pos );
457             AVI_READ4BYTES( p_chk->idx1.entry[i_index].i_length );
458         }
459     }
460     else
461     {
462         p_chk->idx1.entry = NULL;
463     }
464 #ifdef AVI_DEBUG
465     msg_Dbg( (vlc_object_t*)s, "idx1: index entry:%d", i_count );
466 #endif
467     AVI_READCHUNK_EXIT( VLC_SUCCESS );
468 }
469
470 static void AVI_ChunkFree_idx1( avi_chunk_t *p_chk )
471 {
472     p_chk->idx1.i_entry_count = 0;
473     p_chk->idx1.i_entry_max   = 0;
474     FREENULL( p_chk->idx1.entry );
475 }
476
477
478
479 static int AVI_ChunkRead_indx( stream_t *s, avi_chunk_t *p_chk )
480 {
481     unsigned int i_count, i;
482     int32_t      i_dummy;
483     avi_chunk_indx_t *p_indx = (avi_chunk_indx_t*)p_chk;
484
485     AVI_READCHUNK_ENTER;
486
487     AVI_READ2BYTES( p_indx->i_longsperentry );
488     AVI_READ1BYTE ( p_indx->i_indexsubtype );
489     AVI_READ1BYTE ( p_indx->i_indextype );
490     AVI_READ4BYTES( p_indx->i_entriesinuse );
491
492     AVI_READ4BYTES( p_indx->i_id );
493     p_indx->idx.std     = NULL;
494     p_indx->idx.field   = NULL;
495     p_indx->idx.super   = NULL;
496
497     if( p_indx->i_indextype == AVI_INDEX_OF_CHUNKS && p_indx->i_indexsubtype == 0 )
498     {
499         AVI_READ8BYTES( p_indx->i_baseoffset );
500         AVI_READ4BYTES( i_dummy );
501
502         i_count = __MIN( p_indx->i_entriesinuse, i_read / 8 );
503         p_indx->i_entriesinuse = i_count;
504         p_indx->idx.std = calloc( sizeof( indx_std_entry_t ), i_count );
505
506         for( i = 0; i < i_count; i++ )
507         {
508             AVI_READ4BYTES( p_indx->idx.std[i].i_offset );
509             AVI_READ4BYTES( p_indx->idx.std[i].i_size );
510         }
511     }
512     else if( p_indx->i_indextype == AVI_INDEX_OF_CHUNKS && p_indx->i_indexsubtype == AVI_INDEX_2FIELD )
513     {
514         AVI_READ8BYTES( p_indx->i_baseoffset );
515         AVI_READ4BYTES( i_dummy );
516
517         i_count = __MIN( p_indx->i_entriesinuse, i_read / 12 );
518         p_indx->i_entriesinuse = i_count;
519         p_indx->idx.field = calloc( sizeof( indx_field_entry_t ), i_count );
520         for( i = 0; i < i_count; i++ )
521         {
522             AVI_READ4BYTES( p_indx->idx.field[i].i_offset );
523             AVI_READ4BYTES( p_indx->idx.field[i].i_size );
524             AVI_READ4BYTES( p_indx->idx.field[i].i_offsetfield2 );
525         }
526     }
527     else if( p_indx->i_indextype == AVI_INDEX_OF_INDEXES )
528     {
529         p_indx->i_baseoffset = 0;
530         AVI_READ4BYTES( i_dummy );
531         AVI_READ4BYTES( i_dummy );
532         AVI_READ4BYTES( i_dummy );
533
534         i_count = __MIN( p_indx->i_entriesinuse, i_read / 16 );
535         p_indx->i_entriesinuse = i_count;
536         p_indx->idx.super = calloc( sizeof( indx_super_entry_t ), i_count );
537
538         for( i = 0; i < i_count; i++ )
539         {
540             AVI_READ8BYTES( p_indx->idx.super[i].i_offset );
541             AVI_READ4BYTES( p_indx->idx.super[i].i_size );
542             AVI_READ4BYTES( p_indx->idx.super[i].i_duration );
543         }
544     }
545     else
546     {
547         msg_Warn( (vlc_object_t*)s, "unknow type/subtype index" );
548     }
549
550 #ifdef AVI_DEBUG
551     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 );
552 #endif
553     AVI_READCHUNK_EXIT( VLC_SUCCESS );
554 }
555 static void AVI_ChunkFree_indx( avi_chunk_t *p_chk )
556 {
557     avi_chunk_indx_t *p_indx = (avi_chunk_indx_t*)p_chk;
558
559     FREENULL( p_indx->idx.std );
560     FREENULL( p_indx->idx.field );
561     FREENULL( p_indx->idx.super );
562 }
563
564
565
566 static struct
567 {
568     vlc_fourcc_t i_fourcc;
569     const char *psz_type;
570 } AVI_strz_type[] =
571 {
572     { AVIFOURCC_IARL, "archive location" },
573     { AVIFOURCC_IART, "artist" },
574     { AVIFOURCC_ICMS, "commisioned" },
575     { AVIFOURCC_ICMT, "comments" },
576     { AVIFOURCC_ICOP, "copyright" },
577     { AVIFOURCC_ICRD, "creation date" },
578     { AVIFOURCC_ICRP, "cropped" },
579     { AVIFOURCC_IDIM, "dimensions" },
580     { AVIFOURCC_IDPI, "dots per inch" },
581     { AVIFOURCC_IENG, "engineer" },
582     { AVIFOURCC_IGNR, "genre" },
583     { AVIFOURCC_IKEY, "keywords" },
584     { AVIFOURCC_ILGT, "lightness" },
585     { AVIFOURCC_IMED, "medium" },
586     { AVIFOURCC_INAM, "name" },
587     { AVIFOURCC_IPLT, "palette setting" },
588     { AVIFOURCC_IPRD, "product" },
589     { AVIFOURCC_ISBJ, "subject" },
590     { AVIFOURCC_ISFT, "software" },
591     { AVIFOURCC_ISHP, "sharpness" },
592     { AVIFOURCC_ISRC, "source" },
593     { AVIFOURCC_ISRF, "source form" },
594     { AVIFOURCC_ITCH, "technician" },
595     { AVIFOURCC_ISMP, "time code" },
596     { AVIFOURCC_IDIT, "digitalization time" },
597     { AVIFOURCC_strn, "stream name" },
598     { 0,              "???" }
599 };
600 static int AVI_ChunkRead_strz( stream_t *s, avi_chunk_t *p_chk )
601 {
602     int i_index;
603     avi_chunk_STRING_t *p_strz = (avi_chunk_STRING_t*)p_chk;
604     AVI_READCHUNK_ENTER;
605
606     for( i_index = 0;; i_index++)
607     {
608         if( !AVI_strz_type[i_index].i_fourcc ||
609             AVI_strz_type[i_index].i_fourcc == p_strz->i_chunk_fourcc )
610         {
611             break;
612         }
613     }
614     p_strz->p_type = strdup( AVI_strz_type[i_index].psz_type );
615     p_strz->p_str = malloc( i_read + 1);
616
617     if( p_strz->i_chunk_size )
618     {
619         memcpy( p_strz->p_str, p_read, i_read );
620     }
621     p_strz->p_str[i_read] = 0;
622
623 #ifdef AVI_DEBUG
624     msg_Dbg( (vlc_object_t*)s, "%4.4s: %s : %s",
625              (char*)&p_strz->i_chunk_fourcc, p_strz->p_type, p_strz->p_str);
626 #endif
627     AVI_READCHUNK_EXIT( VLC_SUCCESS );
628 }
629 static void AVI_ChunkFree_strz( avi_chunk_t *p_chk )
630 {
631     avi_chunk_STRING_t *p_strz = (avi_chunk_STRING_t*)p_chk;
632     FREENULL( p_strz->p_type );
633     FREENULL( p_strz->p_str );
634 }
635
636 static int AVI_ChunkRead_nothing( stream_t *s, avi_chunk_t *p_chk )
637 {
638     return AVI_NextChunk( s, p_chk );
639 }
640 static void AVI_ChunkFree_nothing( avi_chunk_t *p_chk )
641 {
642
643 }
644
645 static struct
646 {
647     vlc_fourcc_t i_fourcc;
648     int   (*AVI_ChunkRead_function)( stream_t *s, avi_chunk_t *p_chk );
649     void  (*AVI_ChunkFree_function)( avi_chunk_t *p_chk );
650 } AVI_Chunk_Function [] =
651 {
652     { AVIFOURCC_RIFF, AVI_ChunkRead_list, AVI_ChunkFree_nothing },
653     { AVIFOURCC_LIST, AVI_ChunkRead_list, AVI_ChunkFree_nothing },
654     { AVIFOURCC_avih, AVI_ChunkRead_avih, AVI_ChunkFree_nothing },
655     { AVIFOURCC_strh, AVI_ChunkRead_strh, AVI_ChunkFree_nothing },
656     { AVIFOURCC_strf, AVI_ChunkRead_strf, AVI_ChunkFree_strf },
657     { AVIFOURCC_strd, AVI_ChunkRead_strd, AVI_ChunkFree_strd },
658     { AVIFOURCC_idx1, AVI_ChunkRead_idx1, AVI_ChunkFree_idx1 },
659     { AVIFOURCC_indx, AVI_ChunkRead_indx, AVI_ChunkFree_indx },
660     { AVIFOURCC_JUNK, AVI_ChunkRead_nothing, AVI_ChunkFree_nothing },
661
662     { AVIFOURCC_IARL, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
663     { AVIFOURCC_IARL, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
664     { AVIFOURCC_IART, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
665     { AVIFOURCC_ICMS, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
666     { AVIFOURCC_ICMT, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
667     { AVIFOURCC_ICOP, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
668     { AVIFOURCC_ICRD, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
669     { AVIFOURCC_ICRP, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
670     { AVIFOURCC_IDIM, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
671     { AVIFOURCC_IDPI, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
672     { AVIFOURCC_IENG, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
673     { AVIFOURCC_IGNR, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
674     { AVIFOURCC_IKEY, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
675     { AVIFOURCC_ILGT, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
676     { AVIFOURCC_IMED, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
677     { AVIFOURCC_INAM, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
678     { AVIFOURCC_IPLT, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
679     { AVIFOURCC_IPRD, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
680     { AVIFOURCC_ISBJ, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
681     { AVIFOURCC_ISFT, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
682     { AVIFOURCC_ISHP, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
683     { AVIFOURCC_ISRC, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
684     { AVIFOURCC_ISRF, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
685     { AVIFOURCC_ITCH, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
686     { AVIFOURCC_ISMP, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
687     { AVIFOURCC_IDIT, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
688     { AVIFOURCC_strn, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
689     { 0,           NULL,               NULL }
690 };
691
692 static int AVI_ChunkFunctionFind( vlc_fourcc_t i_fourcc )
693 {
694     unsigned int i_index;
695     for( i_index = 0; ; i_index++ )
696     {
697         if( ( AVI_Chunk_Function[i_index].i_fourcc == i_fourcc )||
698             ( AVI_Chunk_Function[i_index].i_fourcc == 0 ) )
699         {
700             return i_index;
701         }
702     }
703 }
704
705 int  _AVI_ChunkRead( stream_t *s, avi_chunk_t *p_chk, avi_chunk_t *p_father )
706 {
707     int i_index;
708
709     if( !p_chk )
710     {
711         return VLC_EGENERIC;
712     }
713
714     if( AVI_ChunkReadCommon( s, p_chk ) )
715     {
716         msg_Warn( (vlc_object_t*)s, "cannot read one chunk" );
717         return VLC_EGENERIC;
718     }
719     if( p_chk->common.i_chunk_fourcc == VLC_FOURCC( 0, 0, 0, 0 ) )
720     {
721         msg_Warn( (vlc_object_t*)s, "found null fourcc chunk (corrupted file?)" );
722         return VLC_EGENERIC;
723     }
724     p_chk->common.p_father = p_father;
725
726     i_index = AVI_ChunkFunctionFind( p_chk->common.i_chunk_fourcc );
727     if( AVI_Chunk_Function[i_index].AVI_ChunkRead_function )
728     {
729         return AVI_Chunk_Function[i_index].AVI_ChunkRead_function( s, p_chk );
730     }
731     else if( ( ((char*)&p_chk->common.i_chunk_fourcc)[0] == 'i' &&
732                ((char*)&p_chk->common.i_chunk_fourcc)[1] == 'x' ) ||
733              ( ((char*)&p_chk->common.i_chunk_fourcc)[2] == 'i' &&
734                ((char*)&p_chk->common.i_chunk_fourcc)[3] == 'x' ) )
735     {
736         p_chk->common.i_chunk_fourcc = AVIFOURCC_indx;
737         return AVI_ChunkRead_indx( s, p_chk );
738     }
739     msg_Warn( (vlc_object_t*)s, "unknown chunk (not loaded)" );
740     return AVI_NextChunk( s, p_chk );
741 }
742
743 void _AVI_ChunkFree( stream_t *s,
744                      avi_chunk_t *p_chk )
745 {
746     int i_index;
747     avi_chunk_t *p_child, *p_next;
748
749     if( !p_chk )
750     {
751         return;
752     }
753
754     /* Free all child chunk */
755     p_child = p_chk->common.p_first;
756     while( p_child )
757     {
758         p_next = p_child->common.p_next;
759         AVI_ChunkFree( s, p_child );
760         free( p_child );
761         p_child = p_next;
762     }
763
764     i_index = AVI_ChunkFunctionFind( p_chk->common.i_chunk_fourcc );
765     if( AVI_Chunk_Function[i_index].AVI_ChunkFree_function )
766     {
767 #ifdef AVI_DEBUG
768         msg_Dbg( (vlc_object_t*)s, "free chunk %4.4s",
769                  (char*)&p_chk->common.i_chunk_fourcc );
770 #endif
771         AVI_Chunk_Function[i_index].AVI_ChunkFree_function( p_chk);
772     }
773     else
774     {
775         msg_Warn( (vlc_object_t*)s, "unknown chunk (not unloaded)" );
776     }
777     p_chk->common.p_first = NULL;
778     p_chk->common.p_last  = NULL;
779
780     return;
781 }
782
783 static void AVI_ChunkDumpDebug_level( vlc_object_t *p_obj,
784                                       avi_chunk_t  *p_chk, int i_level )
785 {
786     char str[1024];
787     int i;
788     avi_chunk_t *p_child;
789
790     memset( str, ' ', sizeof( str ) );
791     for( i = 1; i < i_level; i++ )
792     {
793         str[i * 5] = '|';
794     }
795     if( p_chk->common.i_chunk_fourcc == AVIFOURCC_RIFF||
796         p_chk->common.i_chunk_fourcc == AVIFOURCC_LIST )
797     {
798         sprintf( str + i_level * 5,
799                  "%c %4.4s-%4.4s size:"I64Fu" pos:"I64Fu,
800                  i_level ? '+' : '*',
801                  (char*)&p_chk->common.i_chunk_fourcc,
802                  (char*)&p_chk->list.i_type,
803                  p_chk->common.i_chunk_size,
804                  p_chk->common.i_chunk_pos );
805     }
806     else
807     {
808         sprintf( str + i_level * 5,
809                  "+ %4.4s size:"I64Fu" pos:"I64Fu,
810                  (char*)&p_chk->common.i_chunk_fourcc,
811                  p_chk->common.i_chunk_size,
812                  p_chk->common.i_chunk_pos );
813     }
814     msg_Dbg( p_obj, "%s", str );
815
816     p_child = p_chk->common.p_first;
817     while( p_child )
818     {
819         AVI_ChunkDumpDebug_level( p_obj, p_child, i_level + 1 );
820         p_child = p_child->common.p_next;
821     }
822 }
823
824 int AVI_ChunkReadRoot( stream_t *s, avi_chunk_t *p_root )
825 {
826     avi_chunk_list_t *p_list = (avi_chunk_list_t*)p_root;
827     avi_chunk_t      *p_chk;
828     vlc_bool_t b_seekable;
829
830     stream_Control( s, STREAM_CAN_FASTSEEK, &b_seekable );
831
832     p_list->i_chunk_pos  = 0;
833     p_list->i_chunk_size = stream_Size( s );
834     p_list->i_chunk_fourcc = AVIFOURCC_LIST;
835     p_list->p_father = NULL;
836     p_list->p_next  = NULL;
837     p_list->p_first = NULL;
838     p_list->p_last  = NULL;
839
840     p_list->i_type = VLC_FOURCC( 'r', 'o', 'o', 't' );
841
842     for( ; ; )
843     {
844         p_chk = malloc( sizeof( avi_chunk_t ) );
845         memset( p_chk, 0, sizeof( avi_chunk_t ) );
846         if( !p_root->common.p_first )
847         {
848             p_root->common.p_first = p_chk;
849         }
850         else
851         {
852             p_root->common.p_last->common.p_next = p_chk;
853         }
854         p_root->common.p_last = p_chk;
855
856         if( AVI_ChunkRead( s, p_chk, p_root ) ||
857            ( stream_Tell( s ) >=
858               (off_t)p_chk->common.p_father->common.i_chunk_pos +
859                (off_t)__EVEN( p_chk->common.p_father->common.i_chunk_size ) ) )
860         {
861             break;
862         }
863         /* If we can't seek then stop when we 've found first RIFF-AVI */
864         if( p_chk->common.i_chunk_fourcc == AVIFOURCC_RIFF &&
865             p_chk->list.i_type == AVIFOURCC_AVI && !b_seekable )
866         {
867             break;
868         }
869     }
870
871     AVI_ChunkDumpDebug_level( (vlc_object_t*)s, p_root, 0 );
872     return VLC_SUCCESS;
873 }
874
875 void AVI_ChunkFreeRoot( stream_t *s,
876                         avi_chunk_t  *p_chk )
877 {
878     AVI_ChunkFree( s, p_chk );
879 }
880
881
882 int  _AVI_ChunkCount( avi_chunk_t *p_chk, vlc_fourcc_t i_fourcc )
883 {
884     int i_count;
885     avi_chunk_t *p_child;
886
887     if( !p_chk )
888     {
889         return 0;
890     }
891
892     i_count = 0;
893     p_child = p_chk->common.p_first;
894     while( p_child )
895     {
896         if( p_child->common.i_chunk_fourcc == i_fourcc ||
897             ( p_child->common.i_chunk_fourcc == AVIFOURCC_LIST &&
898               p_child->list.i_type == i_fourcc ) )
899         {
900             i_count++;
901         }
902         p_child = p_child->common.p_next;
903     }
904     return i_count;
905 }
906
907 void *_AVI_ChunkFind( avi_chunk_t *p_chk,
908                       vlc_fourcc_t i_fourcc, int i_number )
909 {
910     avi_chunk_t *p_child;
911     if( !p_chk )
912     {
913         return NULL;
914     }
915     p_child = p_chk->common.p_first;
916
917     while( p_child )
918     {
919         if( p_child->common.i_chunk_fourcc == i_fourcc ||
920             ( p_child->common.i_chunk_fourcc == AVIFOURCC_LIST &&
921               p_child->list.i_type == i_fourcc ) )
922         {
923             if( i_number == 0 )
924             {
925                 /* We found it */
926                 return p_child;
927             }
928
929             i_number--;
930         }
931         p_child = p_child->common.p_next;
932     }
933     return NULL;
934 }
935
936