2cb0a2aefbef260a72d6b40fd1f6a64b49d5028d
[vlc.git] / modules / demux / avi / libavi.c
1 /*****************************************************************************
2  * libavi.c :
3  *****************************************************************************
4  * Copyright (C) 2001 VideoLAN
5  * $Id: libavi.c,v 1.22 2003/08/17 23:02:52 fenrir Exp $
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., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
21  *****************************************************************************/
22
23 #include <stdlib.h>                                      /* malloc(), free() */
24 #include <string.h>                                              /* strdup() */
25 #include <errno.h>
26 #include <sys/types.h>
27
28 #include <vlc/vlc.h>
29 #include <vlc/input.h>
30
31 #include "libavi.h"
32
33 #define AVI_DEBUG 1
34
35 #define FREE( p ) \
36     if( p ) {free( p ); p = NULL; }
37
38 #define __EVEN( x ) ( (x)&0x01 ? (x)+1 : (x) )
39
40 static vlc_fourcc_t GetFOURCC( byte_t *p_buff )
41 {
42     return VLC_FOURCC( p_buff[0], p_buff[1], p_buff[2], p_buff[3] );
43 }
44 /*****************************************************************************
45  * Some basic functions to manipulate stream more easily in vlc
46  *
47  * AVI_TellAbsolute get file position
48  *
49  * AVI_SeekAbsolute seek in the file
50  *
51  * AVI_ReadData read data from the file in a buffer
52  *
53  * AVI_SkipBytes skip bytes
54  *
55  *****************************************************************************/
56 off_t AVI_TellAbsolute( input_thread_t *p_input )
57 {
58     off_t i_pos;
59
60     vlc_mutex_lock( &p_input->stream.stream_lock );
61
62     i_pos= p_input->stream.p_selected_area->i_tell;
63
64     vlc_mutex_unlock( &p_input->stream.stream_lock );
65
66     return i_pos;
67 }
68
69 int AVI_SeekAbsolute( input_thread_t *p_input,
70                       off_t i_pos)
71 {
72     off_t i_filepos;
73
74     if( p_input->stream.p_selected_area->i_size > 0 &&
75         i_pos >= p_input->stream.p_selected_area->i_size )
76     {
77         return VLC_EGENERIC;
78     }
79
80     i_filepos = AVI_TellAbsolute( p_input );
81
82     if( i_filepos == i_pos )
83     {
84         return VLC_SUCCESS;
85     }
86
87     if( p_input->stream.b_seekable &&
88         ( p_input->stream.i_method == INPUT_METHOD_FILE ||
89           i_pos - i_filepos < 0 ||
90           i_pos - i_filepos > 1024 ) )
91     {
92         input_AccessReinit( p_input );
93         p_input->pf_seek( p_input, i_pos );
94         return VLC_SUCCESS;
95     }
96     else if( i_pos - i_filepos > 0 )
97     {
98         data_packet_t   *p_data;
99         int             i_skip = i_pos - i_filepos;
100
101         msg_Warn( p_input, "will skip %d bytes, slow", i_skip );
102         if( i_skip < 0 )
103         {
104             return VLC_EGENERIC; // failed
105         }
106         while (i_skip > 0 )
107         {
108             int i_read;
109
110             i_read = input_SplitBuffer( p_input, &p_data, 
111                                         __MIN( 4096, i_skip ) );
112             if( i_read <= 0 )
113             {
114                 /* Error or eof */
115                 return VLC_EGENERIC;
116             }
117             i_skip -= i_read;
118
119             input_DeletePacket( p_input->p_method_data, p_data );
120         }
121         return VLC_SUCCESS;
122     }
123     else
124     {
125         return VLC_EGENERIC;
126     }
127 }
128
129 /* return amount read if success, 0 if failed */
130 int AVI_ReadData( input_thread_t *p_input, uint8_t *p_buff, int i_size )
131 {
132     data_packet_t *p_data;
133
134     int i_count;
135     int i_read = 0;
136
137
138     if( !i_size )
139     {
140         return 0;
141     }
142
143     do
144     {
145         i_count = input_SplitBuffer(p_input, &p_data, __MIN( i_size, 1024 ) );
146         if( i_count <= 0 )
147         {
148             return i_read;
149         }
150         memcpy( p_buff, p_data->p_payload_start, i_count );
151         input_DeletePacket( p_input->p_method_data, p_data );
152
153         p_buff += i_count;
154         i_size -= i_count;
155         i_read += i_count;
156
157     } while( i_size );
158
159     return i_read;
160 }
161
162 int  AVI_SkipBytes( input_thread_t *p_input, int64_t i_count )
163 {
164     /* broken with new use of i_tell */
165 #if 0
166     int i_buff_size;
167     vlc_mutex_lock( &p_input->stream.stream_lock );
168     i_buff_size = p_input->p_last_data - p_input->p_current_data;
169     vlc_mutex_unlock( &p_input->stream.stream_lock );
170
171     if( i_count > 0 && i_count + 1 < i_buff_size )
172     {
173         uint8_t *p_peek;
174
175         input_Peek( p_input, &p_peek, i_count + 1 );
176
177         vlc_mutex_lock( &p_input->stream.stream_lock );
178         p_input->p_current_data += i_count;  // skip them
179         vlc_mutex_unlock( &p_input->stream.stream_lock );
180         return VLC_SUCCESS;
181     }
182     else
183 #endif
184     {
185         return AVI_SeekAbsolute( p_input,
186                                  AVI_TellAbsolute( p_input ) + i_count );
187     }
188 }
189
190 /*****************************************************************************
191  *
192  * AVI_TestFile: look at first bytes to see if it's a valid avi file
193  *
194  * unseekable: ok
195  *
196  *****************************************************************************/
197 int AVI_TestFile( input_thread_t *p_input )
198 {
199     uint8_t  *p_peek;
200
201     if( input_Peek( p_input, &p_peek, 8 ) < 8 )
202     {
203         msg_Err( p_input, "cannot peek()" );
204         return VLC_EGENERIC;
205     }
206
207     if( GetFOURCC( p_peek ) == AVIFOURCC_RIFF && 
208         GetFOURCC( p_peek + 8 ) == AVIFOURCC_AVI )
209     {
210         return VLC_SUCCESS;
211     }
212     else
213     {
214         return VLC_EGENERIC;
215     }
216 }
217 /****************************************************************************
218  *
219  * Basics functions to manipulates chunks
220  *
221  ****************************************************************************/
222 static int AVI_ChunkReadCommon( input_thread_t *p_input,
223                                 avi_chunk_t *p_chk )
224 {
225     uint8_t  *p_peek;
226     int i_peek;
227
228     memset( p_chk, 0, sizeof( avi_chunk_t ) );
229
230     if( ( i_peek = input_Peek( p_input, &p_peek, 8 ) ) < 8 )
231     {
232         return VLC_EGENERIC;
233     }
234
235     p_chk->common.i_chunk_fourcc = GetFOURCC( p_peek );
236     p_chk->common.i_chunk_size   = GetDWLE( p_peek + 4 );
237     p_chk->common.i_chunk_pos    = AVI_TellAbsolute( p_input );
238
239     p_chk->common.p_father = NULL;
240     p_chk->common.p_next = NULL;
241     p_chk->common.p_first = NULL;
242     p_chk->common.p_next = NULL;
243 #ifdef AVI_DEBUG
244     msg_Dbg( p_input, 
245              "Found Chunk fourcc:%8.8x (%4.4s) size:"I64Fd" pos:"I64Fd,
246              p_chk->common.i_chunk_fourcc,
247              (char*)&p_chk->common.i_chunk_fourcc,
248              p_chk->common.i_chunk_size,
249              p_chk->common.i_chunk_pos );
250 #endif
251     return VLC_SUCCESS;
252 }
253
254 static int AVI_NextChunk( input_thread_t *p_input,
255                           avi_chunk_t *p_chk )
256 {
257     avi_chunk_t chk;
258
259     if( !p_chk )
260     {
261         if( AVI_ChunkReadCommon( p_input, &chk ) )
262         {
263             return VLC_EGENERIC;
264         }
265         p_chk = &chk;
266     }
267
268     if( p_chk->common.p_father )
269     {
270         if( p_chk->common.p_father->common.i_chunk_pos + 
271                 __EVEN( p_chk->common.p_father->common.i_chunk_size ) + 8 <
272             p_chk->common.i_chunk_pos + 
273                 __EVEN( p_chk->common.i_chunk_size ) + 8 )
274         {
275             return VLC_EGENERIC;
276         }
277     }
278     return AVI_SeekAbsolute( p_input,
279                              p_chk->common.i_chunk_pos + 
280                                  __EVEN( p_chk->common.i_chunk_size ) + 8 );
281 }
282
283 int _AVI_ChunkGoto( input_thread_t *p_input,
284                    avi_chunk_t *p_chk )
285 {
286     if( !p_chk )
287     {
288         return VLC_EGENERIC;
289     }
290     return AVI_SeekAbsolute( p_input, p_chk->common.i_chunk_pos );
291 }
292
293 /****************************************************************************
294  *
295  * Functions to read chunks 
296  *
297  ****************************************************************************/
298 static int AVI_ChunkRead_list( input_thread_t *p_input,
299                                avi_chunk_t *p_container,
300                                vlc_bool_t b_seekable )
301 {
302     avi_chunk_t *p_chk;
303     uint8_t *p_peek;
304
305     if( p_container->common.i_chunk_size < 8 )
306     {
307         /* empty box */
308         msg_Warn( p_input, "empty list chunk" );
309         return VLC_EGENERIC;
310     }
311     if( input_Peek( p_input, &p_peek, 12 ) < 12 )
312     {
313         msg_Warn( p_input, "cannot peek while reading list chunk" );
314         return VLC_EGENERIC;
315     }
316     p_container->list.i_type = GetFOURCC( p_peek + 8 );
317
318     if( p_container->common.i_chunk_fourcc == AVIFOURCC_LIST &&
319         p_container->list.i_type == AVIFOURCC_movi )
320     {
321         msg_Dbg( p_input, "Skipping movi chunk" );
322         if( b_seekable )
323         {
324             return AVI_NextChunk( p_input, p_container );
325         }
326         else
327         {
328             return VLC_SUCCESS; // point at begining of LIST-movi 
329         }
330     }
331
332     if( AVI_SkipBytes( p_input, 12 ) )
333     {
334         msg_Warn( p_input, "cannot enter chunk" );
335         return VLC_EGENERIC;
336     }
337 #ifdef AVI_DEBUG
338     msg_Dbg( p_input, 
339              "found LIST chunk: \'%4.4s\'",
340              (char*)&p_container->list.i_type );
341 #endif
342     msg_Dbg( p_input, "<list \'%4.4s\'>", (char*)&p_container->list.i_type );
343     for( ; ; )
344     {
345         p_chk = malloc( sizeof( avi_chunk_t ) );
346         memset( p_chk, 0, sizeof( avi_chunk_t ) );
347         if( !p_container->common.p_first )
348         {
349             p_container->common.p_first = p_chk;
350         }
351         else
352         {
353             p_container->common.p_last->common.p_next = p_chk;
354         }
355         p_container->common.p_last = p_chk;
356
357         if( AVI_ChunkRead( p_input, p_chk, p_container, b_seekable ) ||
358            ( AVI_TellAbsolute( p_input ) >=
359               (off_t)p_chk->common.p_father->common.i_chunk_pos +
360                (off_t)__EVEN( p_chk->common.p_father->common.i_chunk_size ) ) )
361         {
362             break;
363         }
364         /* If we can't seek then stop when we 've found LIST-movi */
365         if( p_chk->common.i_chunk_fourcc == AVIFOURCC_LIST &&
366             p_chk->list.i_type == AVIFOURCC_movi && !b_seekable )
367         {
368             break;
369         }
370
371     }
372     msg_Dbg( p_input, "</list \'%4.4s\'>", (char*)&p_container->list.i_type );
373
374     return VLC_SUCCESS;
375 }
376
377 #define AVI_READCHUNK_ENTER \
378     int64_t i_read = __EVEN(p_chk->common.i_chunk_size ) + 8; \
379     uint8_t  *p_read, *p_buff;    \
380     if( !( p_read = p_buff = malloc(i_read ) ) ) \
381     { \
382         return 0; \
383     } \
384     i_read = AVI_ReadData( p_input, p_read, i_read ); \
385     p_read += 8; \
386     i_read -= 8
387
388 #define AVI_READCHUNK_EXIT( code ) \
389     free( p_buff ); \
390     if( i_read < 0 ) \
391     { \
392         msg_Warn( p_input, "not enough data" ); \
393     } \
394     return code
395
396 #define AVI_READ1BYTE( i_byte ) \
397     i_byte = *p_read; \
398     p_read++; \
399     i_read--
400
401 #define AVI_READ2BYTES( i_word ) \
402     i_word = GetWLE( p_read ); \
403     p_read += 2; \
404     i_read -= 2
405
406 #define AVI_READ4BYTES( i_dword ) \
407     i_dword = GetDWLE( p_read ); \
408     p_read += 4; \
409     i_read -= 4
410
411 #define AVI_READ8BYTES( i_dword ) \
412     i_dword = GetQWLE( p_read ); \
413     p_read += 8; \
414     i_read -= 8
415
416 #define AVI_READFOURCC( i_dword ) \
417     i_dword = GetFOURCC( p_read ); \
418     p_read += 4; \
419     i_read -= 4
420
421 static int AVI_ChunkRead_avih( input_thread_t *p_input,
422                                avi_chunk_t *p_chk,
423                                vlc_bool_t b_seekable )
424 {
425     AVI_READCHUNK_ENTER;
426
427     AVI_READ4BYTES( p_chk->avih.i_microsecperframe);
428     AVI_READ4BYTES( p_chk->avih.i_maxbytespersec );
429     AVI_READ4BYTES( p_chk->avih.i_reserved1 );
430     AVI_READ4BYTES( p_chk->avih.i_flags );
431     AVI_READ4BYTES( p_chk->avih.i_totalframes );
432     AVI_READ4BYTES( p_chk->avih.i_initialframes );
433     AVI_READ4BYTES( p_chk->avih.i_streams );
434     AVI_READ4BYTES( p_chk->avih.i_suggestedbuffersize );
435     AVI_READ4BYTES( p_chk->avih.i_width );
436     AVI_READ4BYTES( p_chk->avih.i_height );
437     AVI_READ4BYTES( p_chk->avih.i_scale );
438     AVI_READ4BYTES( p_chk->avih.i_rate );
439     AVI_READ4BYTES( p_chk->avih.i_start );
440     AVI_READ4BYTES( p_chk->avih.i_length );
441 #ifdef AVI_DEBUG
442     msg_Dbg( p_input, 
443              "avih: streams:%d flags:%s%s%s%s %dx%d", 
444              p_chk->avih.i_streams,
445              p_chk->avih.i_flags&AVIF_HASINDEX?" HAS_INDEX":"",
446              p_chk->avih.i_flags&AVIF_MUSTUSEINDEX?" MUST_USE_INDEX":"",
447              p_chk->avih.i_flags&AVIF_ISINTERLEAVED?" IS_INTERLEAVED":"",
448              p_chk->avih.i_flags&AVIF_TRUSTCKTYPE?" TRUST_CKTYPE":"",
449              p_chk->avih.i_width, p_chk->avih.i_height );
450 #endif 
451     AVI_READCHUNK_EXIT( VLC_SUCCESS );
452 }
453
454 static int AVI_ChunkRead_strh( input_thread_t *p_input,
455                                avi_chunk_t *p_chk,
456                                vlc_bool_t b_seekable )
457 {
458     AVI_READCHUNK_ENTER;
459
460     AVI_READFOURCC( p_chk->strh.i_type );
461     AVI_READFOURCC( p_chk->strh.i_handler );
462     AVI_READ4BYTES( p_chk->strh.i_flags );
463     AVI_READ4BYTES( p_chk->strh.i_reserved1 );
464     AVI_READ4BYTES( p_chk->strh.i_initialframes );
465     AVI_READ4BYTES( p_chk->strh.i_scale );
466     AVI_READ4BYTES( p_chk->strh.i_rate );
467     AVI_READ4BYTES( p_chk->strh.i_start );
468     AVI_READ4BYTES( p_chk->strh.i_length );
469     AVI_READ4BYTES( p_chk->strh.i_suggestedbuffersize );
470     AVI_READ4BYTES( p_chk->strh.i_quality );
471     AVI_READ4BYTES( p_chk->strh.i_samplesize );
472 #ifdef AVI_DEBUG
473     msg_Dbg( p_input, 
474              "strh: type:%4.4s handler:0x%8.8x samplesize:%d %.2ffps",
475              (char*)&p_chk->strh.i_type,
476              p_chk->strh.i_handler,
477              p_chk->strh.i_samplesize,
478              ( p_chk->strh.i_scale ? 
479                 (float)p_chk->strh.i_rate / (float)p_chk->strh.i_scale : -1) );
480 #endif
481
482     AVI_READCHUNK_EXIT( VLC_SUCCESS );
483 }
484
485 static int AVI_ChunkRead_strf( input_thread_t *p_input,
486                                avi_chunk_t *p_chk,
487                                vlc_bool_t b_seekable )
488 {
489     avi_chunk_t *p_strh;
490
491     AVI_READCHUNK_ENTER;
492     if( p_chk->common.p_father == NULL )
493     {
494         msg_Err( p_input, "malformed avi file" );
495         AVI_READCHUNK_EXIT( VLC_EGENERIC );
496     }
497     if( !( p_strh = AVI_ChunkFind( p_chk->common.p_father, AVIFOURCC_strh, 0 ) ) )
498     {
499         msg_Err( p_input, "malformed avi file" );
500         AVI_READCHUNK_EXIT( VLC_EGENERIC );
501     }
502
503     switch( p_strh->strh.i_type )
504     {
505         case( AVIFOURCC_auds ):
506             p_chk->strf.auds.i_cat = AUDIO_ES;
507             p_chk->strf.auds.p_wf = malloc( __MAX( p_chk->common.i_chunk_size, sizeof( WAVEFORMATEX ) ) );
508             AVI_READ2BYTES( p_chk->strf.auds.p_wf->wFormatTag );
509             AVI_READ2BYTES( p_chk->strf.auds.p_wf->nChannels );
510             AVI_READ4BYTES( p_chk->strf.auds.p_wf->nSamplesPerSec );
511             AVI_READ4BYTES( p_chk->strf.auds.p_wf->nAvgBytesPerSec );
512             AVI_READ2BYTES( p_chk->strf.auds.p_wf->nBlockAlign );
513             AVI_READ2BYTES( p_chk->strf.auds.p_wf->wBitsPerSample );
514             if( p_chk->strf.auds.p_wf->wFormatTag != WAVE_FORMAT_PCM
515                  && p_chk->common.i_chunk_size > sizeof( WAVEFORMATEX ) )
516             {
517                 AVI_READ2BYTES( p_chk->strf.auds.p_wf->cbSize );
518                 /* prevent segfault */
519                 if( p_chk->strf.auds.p_wf->cbSize >
520                         p_chk->common.i_chunk_size - sizeof( WAVEFORMATEX ) )
521                 {
522                     p_chk->strf.auds.p_wf->cbSize =
523                         p_chk->common.i_chunk_size - sizeof( WAVEFORMATEX );
524                 }
525             }
526             else
527             {
528                 p_chk->strf.auds.p_wf->cbSize = 0;
529             }
530             if( p_chk->strf.auds.p_wf->cbSize > 0 )
531             {
532                 memcpy( &p_chk->strf.auds.p_wf[1] ,
533                         p_buff + 8 + sizeof( WAVEFORMATEX ),    // 8=fourrc+size
534                         p_chk->strf.auds.p_wf->cbSize );
535             }
536 #ifdef AVI_DEBUG
537             msg_Dbg( p_input, 
538                      "strf: audio:0x%4.4x channels:%d %dHz %dbits/sample %dkb/s",
539                      p_chk->strf.auds.p_wf->wFormatTag,
540                      p_chk->strf.auds.p_wf->nChannels,
541                      p_chk->strf.auds.p_wf->nSamplesPerSec,
542                      p_chk->strf.auds.p_wf->wBitsPerSample,
543                      p_chk->strf.auds.p_wf->nAvgBytesPerSec * 8 / 1024 );
544 #endif
545             break;
546         case( AVIFOURCC_vids ):
547             p_strh->strh.i_samplesize = 0; // XXX for ffmpeg avi file
548             p_chk->strf.vids.i_cat = VIDEO_ES;
549             p_chk->strf.vids.p_bih = malloc( p_chk->common.i_chunk_size );
550             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biSize );
551             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biWidth );
552             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biHeight );
553             AVI_READ2BYTES( p_chk->strf.vids.p_bih->biPlanes );
554             AVI_READ2BYTES( p_chk->strf.vids.p_bih->biBitCount );
555             AVI_READFOURCC( p_chk->strf.vids.p_bih->biCompression );
556             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biSizeImage );
557             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biXPelsPerMeter );
558             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biYPelsPerMeter );
559             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biClrUsed );
560             AVI_READ4BYTES( p_chk->strf.vids.p_bih->biClrImportant );
561             if( p_chk->strf.vids.p_bih->biSize >
562                         p_chk->common.i_chunk_size )
563             {
564                 p_chk->strf.vids.p_bih->biSize = p_chk->common.i_chunk_size;
565             }
566             if( p_chk->strf.vids.p_bih->biSize - sizeof(BITMAPINFOHEADER) > 0 )
567             {
568                 memcpy( &p_chk->strf.vids.p_bih[1],
569                         p_buff + 8 + sizeof(BITMAPINFOHEADER), // 8=fourrc+size
570                         p_chk->strf.vids.p_bih->biSize -
571                                                     sizeof(BITMAPINFOHEADER) );
572             }
573 #ifdef AVI_DEBUG
574             msg_Dbg( p_input,
575                      "strf: video:%4.4s %dx%d planes:%d %dbpp",
576                      (char*)&p_chk->strf.vids.p_bih->biCompression,
577                      p_chk->strf.vids.p_bih->biWidth,
578                      p_chk->strf.vids.p_bih->biHeight,
579                      p_chk->strf.vids.p_bih->biPlanes,
580                      p_chk->strf.vids.p_bih->biBitCount );
581 #endif
582             break;
583         default:
584             msg_Warn( p_input, "unknown stream type" );
585             p_chk->strf.common.i_cat = UNKNOWN_ES;
586             break;
587     }
588     AVI_READCHUNK_EXIT( VLC_SUCCESS );
589 }
590 static void AVI_ChunkFree_strf( input_thread_t *p_input,
591                                avi_chunk_t *p_chk )
592 {
593     avi_chunk_strf_t *p_strf = (avi_chunk_strf_t*)p_chk;
594     switch( p_strf->common.i_cat )
595     {
596         case AUDIO_ES:
597             FREE( p_strf->auds.p_wf );
598             break;
599         case VIDEO_ES:
600             FREE( p_strf->vids.p_bih );
601             break;
602         default:
603             break;
604     }
605 }
606
607 static int AVI_ChunkRead_strd( input_thread_t *p_input,
608                                avi_chunk_t *p_chk,
609                                vlc_bool_t b_seekable )
610 {
611     AVI_READCHUNK_ENTER;
612     p_chk->strd.p_data = malloc( p_chk->common.i_chunk_size );
613     memcpy( p_chk->strd.p_data,
614             p_buff + 8,
615             p_chk->common.i_chunk_size );
616     AVI_READCHUNK_EXIT( VLC_SUCCESS );
617 }
618
619 static int AVI_ChunkRead_idx1( input_thread_t *p_input,
620                                avi_chunk_t *p_chk,
621                                vlc_bool_t b_seekable )
622 {
623     unsigned int i_count, i_index;
624
625     AVI_READCHUNK_ENTER;
626
627     i_count = __MIN( (int64_t)p_chk->common.i_chunk_size, i_read ) / 16;
628
629     p_chk->idx1.i_entry_count = i_count;
630     p_chk->idx1.i_entry_max   = i_count;
631     if( i_count > 0 )
632     {
633         p_chk->idx1.entry = calloc( i_count, sizeof( idx1_entry_t ) );
634
635         for( i_index = 0; i_index < i_count ; i_index++ )
636         {
637             AVI_READFOURCC( p_chk->idx1.entry[i_index].i_fourcc );
638             AVI_READ4BYTES( p_chk->idx1.entry[i_index].i_flags );
639             AVI_READ4BYTES( p_chk->idx1.entry[i_index].i_pos );
640             AVI_READ4BYTES( p_chk->idx1.entry[i_index].i_length );
641         }
642     }
643     else
644     {
645         p_chk->idx1.entry = NULL;
646     }
647 #ifdef AVI_DEBUG
648     msg_Dbg( p_input, "idx1: index entry:%d", i_count );
649 #endif
650     AVI_READCHUNK_EXIT( VLC_SUCCESS );
651 }
652
653 static void AVI_ChunkFree_idx1( input_thread_t *p_input,
654                                avi_chunk_t *p_chk )
655 {
656     p_chk->idx1.i_entry_count = 0;
657     p_chk->idx1.i_entry_max   = 0;
658     FREE( p_chk->idx1.entry )
659 }
660
661
662
663 static int AVI_ChunkRead_indx( input_thread_t *p_input,
664                                avi_chunk_t *p_chk,
665                                vlc_bool_t b_seekable )
666 {
667     unsigned int i_count, i;
668     int32_t      i_dummy;
669     avi_chunk_indx_t *p_indx = (avi_chunk_indx_t*)p_chk;
670
671     AVI_READCHUNK_ENTER;
672
673     AVI_READ2BYTES( p_indx->i_longsperentry );
674     AVI_READ1BYTE ( p_indx->i_indexsubtype );
675     AVI_READ1BYTE ( p_indx->i_indextype );
676     AVI_READ4BYTES( p_indx->i_entriesinuse );
677
678     AVI_READ4BYTES( p_indx->i_id );
679     p_indx->idx.std     = NULL;
680     p_indx->idx.field   = NULL;
681     p_indx->idx.super   = NULL;
682
683     if( p_indx->i_indextype == AVI_INDEX_OF_CHUNKS && p_indx->i_indexsubtype == 0 )
684     {
685         AVI_READ8BYTES( p_indx->i_baseoffset );
686         AVI_READ4BYTES( i_dummy );
687
688         i_count = __MIN( p_indx->i_entriesinuse, i_read / 8 );
689         p_indx->i_entriesinuse = i_count;
690         p_indx->idx.std = calloc( sizeof( indx_std_entry_t ), i_count );
691
692         for( i = 0; i < i_count; i++ )
693         {
694             AVI_READ4BYTES( p_indx->idx.std[i].i_offset );
695             AVI_READ4BYTES( p_indx->idx.std[i].i_size );
696         }
697     }
698     else if( p_indx->i_indextype == AVI_INDEX_OF_CHUNKS && p_indx->i_indexsubtype == AVI_INDEX_2FIELD )
699     {
700         AVI_READ8BYTES( p_indx->i_baseoffset );
701         AVI_READ4BYTES( i_dummy );
702
703         i_count = __MIN( p_indx->i_entriesinuse, i_read / 12 );
704         p_indx->i_entriesinuse = i_count;
705         p_indx->idx.field = calloc( sizeof( indx_field_entry_t ), i_count );
706         for( i = 0; i < i_count; i++ )
707         {
708             AVI_READ4BYTES( p_indx->idx.field[i].i_offset );
709             AVI_READ4BYTES( p_indx->idx.field[i].i_size );
710             AVI_READ4BYTES( p_indx->idx.field[i].i_offsetfield2 );
711         }
712     }
713     else if( p_indx->i_indextype == AVI_INDEX_OF_INDEXES )
714     {
715         p_indx->i_baseoffset = 0;
716         AVI_READ4BYTES( i_dummy );
717         AVI_READ4BYTES( i_dummy );
718         AVI_READ4BYTES( i_dummy );
719
720         i_count = __MIN( p_indx->i_entriesinuse, i_read / 16 );
721         p_indx->i_entriesinuse = i_count;
722         p_indx->idx.super = calloc( sizeof( indx_super_entry_t ), i_count );
723
724         for( i = 0; i < i_count; i++ )
725         {
726             AVI_READ8BYTES( p_indx->idx.super[i].i_offset );
727             AVI_READ4BYTES( p_indx->idx.super[i].i_size );
728             AVI_READ4BYTES( p_indx->idx.super[i].i_duration );
729         }
730     }
731     else
732     {
733         msg_Warn( p_input, "unknow type/subtype index" );
734     }
735
736 #ifdef AVI_DEBUG
737     msg_Dbg( p_input, "indx: type=%d subtype=%d entry=%d", p_indx->i_indextype, p_indx->i_indexsubtype, p_indx->i_entriesinuse );
738 #endif
739     AVI_READCHUNK_EXIT( VLC_SUCCESS );
740 }
741 static void AVI_ChunkFree_indx( input_thread_t *p_input,
742                                avi_chunk_t *p_chk )
743 {
744     avi_chunk_indx_t *p_indx = (avi_chunk_indx_t*)p_chk;
745
746     FREE( p_indx->idx.std );
747     FREE( p_indx->idx.field );
748     FREE( p_indx->idx.super );
749 }
750
751
752
753 static struct
754 {
755     vlc_fourcc_t i_fourcc;
756     char *psz_type;
757 } AVI_strz_type[] =
758 {
759     { AVIFOURCC_IARL, "archive location" },
760     { AVIFOURCC_IART, "artist" },
761     { AVIFOURCC_ICMS, "commisioned" },
762     { AVIFOURCC_ICMT, "comments" },
763     { AVIFOURCC_ICOP, "copyright" },
764     { AVIFOURCC_ICRD, "creation date" },
765     { AVIFOURCC_ICRP, "cropped" },
766     { AVIFOURCC_IDIM, "dimensions" },
767     { AVIFOURCC_IDPI, "dots per inch" },
768     { AVIFOURCC_IENG, "enginner" },
769     { AVIFOURCC_IGNR, "genre" },
770     { AVIFOURCC_IKEY, "keywords" },
771     { AVIFOURCC_ILGT, "lightness" },
772     { AVIFOURCC_IMED, "medium" },
773     { AVIFOURCC_INAM, "name" },
774     { AVIFOURCC_IPLT, "palette setting" },
775     { AVIFOURCC_IPRD, "product" },
776     { AVIFOURCC_ISBJ, "subject" },
777     { AVIFOURCC_ISFT, "software" },
778     { AVIFOURCC_ISHP, "sharpness" },
779     { AVIFOURCC_ISRC, "source" },
780     { AVIFOURCC_ISRF, "source form" },
781     { AVIFOURCC_ITCH, "technician" },
782     { AVIFOURCC_ISMP, "time code" },
783     { AVIFOURCC_IDIT, "digitalization time" },
784     { AVIFOURCC_strn, "stream name" },
785     { 0,              "???" }
786 };
787 static int AVI_ChunkRead_strz( input_thread_t *p_input,
788                                avi_chunk_t *p_chk,
789                                vlc_bool_t b_seekable )
790 {
791     int i_index;
792     avi_chunk_STRING_t *p_strz = (avi_chunk_STRING_t*)p_chk;
793     AVI_READCHUNK_ENTER;
794
795     for( i_index = 0;; i_index++)
796     {
797         if( !AVI_strz_type[i_index].i_fourcc ||
798             AVI_strz_type[i_index].i_fourcc == p_strz->i_chunk_fourcc )
799         {
800             break;
801         }
802     }
803     p_strz->p_type = strdup( AVI_strz_type[i_index].psz_type );
804     p_strz->p_str = malloc( i_read + 1);
805
806     if( p_strz->i_chunk_size )
807     {
808         memcpy( p_strz->p_str, p_read, i_read );
809     }
810     p_strz->p_str[i_read] = 0;
811
812 #ifdef AVI_DEBUG
813     msg_Dbg( p_input, "%4.4s: %s : %s", 
814              (char*)&p_strz->i_chunk_fourcc, p_strz->p_type, p_strz->p_str);
815 #endif
816     AVI_READCHUNK_EXIT( VLC_SUCCESS );
817 }
818 static void AVI_ChunkFree_strz( input_thread_t *p_input,
819                                 avi_chunk_t *p_chk )
820 {
821     avi_chunk_STRING_t *p_strz = (avi_chunk_STRING_t*)p_chk;
822     FREE( p_strz->p_type );
823     FREE( p_strz->p_str );
824 }
825
826 static int AVI_ChunkRead_nothing( input_thread_t *p_input,
827                                avi_chunk_t *p_chk,
828                                vlc_bool_t b_seekable )
829 {
830     return AVI_NextChunk( p_input, p_chk );
831 }
832 static void AVI_ChunkFree_nothing( input_thread_t *p_input,
833                                avi_chunk_t *p_chk )
834 {
835
836 }
837
838 static struct
839 {
840     vlc_fourcc_t i_fourcc;
841     int   (*AVI_ChunkRead_function)( input_thread_t *p_input, 
842                                      avi_chunk_t *p_chk,
843                                      vlc_bool_t b_seekable );
844     void  (*AVI_ChunkFree_function)( input_thread_t *p_input,
845                                      avi_chunk_t *p_chk );
846 } AVI_Chunk_Function [] =
847 {
848     { AVIFOURCC_RIFF, AVI_ChunkRead_list, AVI_ChunkFree_nothing },
849     { AVIFOURCC_LIST, AVI_ChunkRead_list, AVI_ChunkFree_nothing },
850     { AVIFOURCC_avih, AVI_ChunkRead_avih, AVI_ChunkFree_nothing },
851     { AVIFOURCC_strh, AVI_ChunkRead_strh, AVI_ChunkFree_nothing },
852     { AVIFOURCC_strf, AVI_ChunkRead_strf, AVI_ChunkFree_strf },
853     { AVIFOURCC_strd, AVI_ChunkRead_strd, AVI_ChunkFree_nothing },
854     { AVIFOURCC_idx1, AVI_ChunkRead_idx1, AVI_ChunkFree_idx1 },
855     { AVIFOURCC_indx, AVI_ChunkRead_indx, AVI_ChunkFree_indx },
856     { AVIFOURCC_JUNK, AVI_ChunkRead_nothing, AVI_ChunkFree_nothing },
857
858     { AVIFOURCC_IARL, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
859     { AVIFOURCC_IARL, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
860     { AVIFOURCC_IART, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
861     { AVIFOURCC_ICMS, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
862     { AVIFOURCC_ICMT, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
863     { AVIFOURCC_ICOP, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
864     { AVIFOURCC_ICRD, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
865     { AVIFOURCC_ICRP, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
866     { AVIFOURCC_IDIM, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
867     { AVIFOURCC_IDPI, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
868     { AVIFOURCC_IENG, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
869     { AVIFOURCC_IGNR, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
870     { AVIFOURCC_IKEY, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
871     { AVIFOURCC_ILGT, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
872     { AVIFOURCC_IMED, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
873     { AVIFOURCC_INAM, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
874     { AVIFOURCC_IPLT, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
875     { AVIFOURCC_IPRD, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
876     { AVIFOURCC_ISBJ, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
877     { AVIFOURCC_ISFT, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
878     { AVIFOURCC_ISHP, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
879     { AVIFOURCC_ISRC, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
880     { AVIFOURCC_ISRF, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
881     { AVIFOURCC_ITCH, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
882     { AVIFOURCC_ISMP, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
883     { AVIFOURCC_IDIT, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
884     { AVIFOURCC_strn, AVI_ChunkRead_strz, AVI_ChunkFree_strz },
885     { 0,           NULL,               NULL }
886 };
887
888 static int AVI_ChunkFunctionFind( vlc_fourcc_t i_fourcc )
889 {
890     unsigned int i_index;
891     for( i_index = 0; ; i_index++ )
892     {
893         if( ( AVI_Chunk_Function[i_index].i_fourcc == i_fourcc )||
894             ( AVI_Chunk_Function[i_index].i_fourcc == 0 ) )
895         {
896             return i_index;
897         }
898     }
899 }
900
901 int  _AVI_ChunkRead( input_thread_t *p_input,
902                      avi_chunk_t *p_chk,
903                      avi_chunk_t *p_father,
904                      vlc_bool_t b_seekable )
905 {
906     int i_index;
907
908     if( !p_chk )
909     {
910         return VLC_EGENERIC;
911     }
912
913     if( AVI_ChunkReadCommon( p_input, p_chk ) )
914     {
915         msg_Warn( p_input, "cannot read one chunk" );
916         return VLC_EGENERIC;
917     }
918     if( p_chk->common.i_chunk_fourcc == VLC_FOURCC( 0, 0, 0, 0 ) )
919     {
920         msg_Warn( p_input, "found null fourcc chunk (corrupted file?)" );
921         return VLC_EGENERIC;
922     }
923     p_chk->common.p_father = p_father;
924
925     i_index = AVI_ChunkFunctionFind( p_chk->common.i_chunk_fourcc );
926     if( AVI_Chunk_Function[i_index].AVI_ChunkRead_function )
927     {
928         return AVI_Chunk_Function[i_index].
929                         AVI_ChunkRead_function( p_input, p_chk, b_seekable );
930     }
931     else if( ((char*)&p_chk->common.i_chunk_fourcc)[0] == 'i' &&
932              ((char*)&p_chk->common.i_chunk_fourcc)[1] == 'x' )
933     {
934         p_chk->common.i_chunk_fourcc = AVIFOURCC_indx;
935         return AVI_ChunkRead_indx( p_input, p_chk, b_seekable );
936     }
937     msg_Warn( p_input, "unknown chunk (not loaded)" );
938     return AVI_NextChunk( p_input, p_chk );
939 }
940
941 void _AVI_ChunkFree( input_thread_t *p_input,
942                      avi_chunk_t *p_chk )
943 {
944     int i_index;
945     avi_chunk_t *p_child, *p_next;
946
947     if( !p_chk )
948     {
949         return;
950     }
951
952     /* Free all child chunk */
953     p_child = p_chk->common.p_first;
954     while( p_child )
955     {
956         p_next = p_child->common.p_next;
957         AVI_ChunkFree( p_input, p_child );
958         free( p_child );
959         p_child = p_next;
960     }
961
962     i_index = AVI_ChunkFunctionFind( p_chk->common.i_chunk_fourcc );
963     if( AVI_Chunk_Function[i_index].AVI_ChunkFree_function )
964     {
965 #ifdef AVI_DEBUG
966         msg_Dbg( p_input, "free chunk %4.4s", 
967                  (char*)&p_chk->common.i_chunk_fourcc );
968 #endif
969         AVI_Chunk_Function[i_index].AVI_ChunkFree_function( p_input, p_chk);
970     }
971     else
972     {
973         msg_Warn( p_input, "unknown chunk (not unloaded)" );
974     }
975     p_chk->common.p_first = NULL;
976     p_chk->common.p_last  = NULL;
977
978     return;
979 }
980
981 int AVI_ChunkReadRoot( input_thread_t *p_input,
982                        avi_chunk_t *p_root,
983                        vlc_bool_t b_seekable )
984 {
985     avi_chunk_list_t *p_list = (avi_chunk_list_t*)p_root;
986     avi_chunk_t      *p_chk;
987
988     p_list->i_chunk_pos  = 0;
989     p_list->i_chunk_size = p_input->stream.p_selected_area->i_size;
990     p_list->i_chunk_fourcc = AVIFOURCC_LIST;
991     p_list->p_father = NULL;
992     p_list->p_next  = NULL;
993     p_list->p_first = NULL;
994     p_list->p_last  = NULL;
995
996     p_list->i_type = VLC_FOURCC( 'r', 'o', 'o', 't' );
997
998     for( ; ; )
999     {
1000         p_chk = malloc( sizeof( avi_chunk_t ) );
1001         memset( p_chk, 0, sizeof( avi_chunk_t ) );
1002         if( !p_root->common.p_first )
1003         {
1004             p_root->common.p_first = p_chk;
1005         }
1006         else
1007         {
1008             p_root->common.p_last->common.p_next = p_chk;
1009         }
1010         p_root->common.p_last = p_chk;
1011
1012         if( AVI_ChunkRead( p_input, p_chk, p_root, b_seekable ) ||
1013            ( AVI_TellAbsolute( p_input ) >=
1014               (off_t)p_chk->common.p_father->common.i_chunk_pos +
1015                (off_t)__EVEN( p_chk->common.p_father->common.i_chunk_size ) ) )
1016         {
1017             break;
1018         }
1019         /* If we can't seek then stop when we 've found first RIFF-AVI */
1020         if( p_chk->common.i_chunk_fourcc == AVIFOURCC_RIFF &&
1021             p_chk->list.i_type == AVIFOURCC_AVI && !b_seekable )
1022         {
1023             break;
1024         }
1025     } 
1026
1027     return VLC_SUCCESS;
1028 }
1029
1030 void AVI_ChunkFreeRoot( input_thread_t *p_input,
1031                         avi_chunk_t  *p_chk )
1032 {
1033     AVI_ChunkFree( p_input, p_chk );
1034 }
1035
1036
1037 int  _AVI_ChunkCount( avi_chunk_t *p_chk, vlc_fourcc_t i_fourcc )
1038 {
1039     int i_count;
1040     avi_chunk_t *p_child;
1041
1042     if( !p_chk )
1043     {
1044         return 0;
1045     }
1046
1047     i_count = 0;
1048     p_child = p_chk->common.p_first;
1049     while( p_child )
1050     {
1051         if( p_child->common.i_chunk_fourcc == i_fourcc ||
1052             ( p_child->common.i_chunk_fourcc == AVIFOURCC_LIST && 
1053               p_child->list.i_type == i_fourcc ) )
1054         {
1055             i_count++;
1056         }
1057         p_child = p_child->common.p_next;
1058     }
1059     return i_count;
1060 }
1061
1062 avi_chunk_t *_AVI_ChunkFind( avi_chunk_t *p_chk,
1063                              vlc_fourcc_t i_fourcc, int i_number )
1064 {
1065     avi_chunk_t *p_child;
1066     if( !p_chk )
1067     {
1068         return NULL;
1069     }
1070     p_child = p_chk->common.p_first;
1071
1072     while( p_child )
1073     {
1074         if( p_child->common.i_chunk_fourcc == i_fourcc ||
1075             ( p_child->common.i_chunk_fourcc == AVIFOURCC_LIST && 
1076               p_child->list.i_type == i_fourcc ) )
1077         {
1078             if( i_number == 0 )
1079             {
1080                 /* We found it */
1081                 return p_child;
1082             }
1083
1084             i_number--;
1085         }
1086         p_child = p_child->common.p_next;
1087     }
1088     return NULL;
1089 }
1090
1091 static void AVI_ChunkDumpDebug_level( input_thread_t *p_input,
1092                                       avi_chunk_t  *p_chk, int i_level )
1093 {
1094     char str[1024];
1095     int i;
1096     avi_chunk_t *p_child;
1097     
1098     memset( str, ' ', sizeof( str ) );
1099     for( i = 1; i < i_level; i++ )
1100     {
1101         str[i * 5] = '|';
1102     }
1103     if( p_chk->common.i_chunk_fourcc == AVIFOURCC_RIFF||
1104         p_chk->common.i_chunk_fourcc == AVIFOURCC_LIST )
1105     {
1106         sprintf( str + i_level * 5, 
1107                  "%c %4.4s-%4.4s size:"I64Fu" pos:"I64Fu,
1108                  i_level ? '+' : '*',
1109                  (char*)&p_chk->common.i_chunk_fourcc,
1110                  (char*)&p_chk->list.i_type,
1111                  p_chk->common.i_chunk_size,
1112                  p_chk->common.i_chunk_pos );
1113     }
1114     else
1115     {
1116         sprintf( str + i_level * 5, 
1117                  "+ %4.4s size:"I64Fu" pos:"I64Fu,
1118                  (char*)&p_chk->common.i_chunk_fourcc,
1119                  p_chk->common.i_chunk_size,
1120                  p_chk->common.i_chunk_pos );
1121     }
1122     msg_Dbg( p_input, "%s", str );
1123
1124     p_child = p_chk->common.p_first;
1125     while( p_child )
1126     {
1127         AVI_ChunkDumpDebug_level( p_input, p_child, i_level + 1 );
1128         p_child = p_child->common.p_next;
1129     }
1130 }
1131 void _AVI_ChunkDumpDebug( input_thread_t *p_input,
1132                          avi_chunk_t  *p_chk )
1133 {
1134     AVI_ChunkDumpDebug_level( p_input, p_chk, 0 );
1135 }
1136