libmpdemux/aviheader.c

   1 /*
   2  * This file is part of MPlayer.
   3  *
   4  * MPlayer is free software; you can redistribute it and/or modify
   5  * it under the terms of the GNU General Public License as published by
   6  * the Free Software Foundation; either version 2 of the License, or
   7  * (at your option) any later version.
   8  *
   9  * MPlayer is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12  * GNU General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  17  */
  18
  19 #include <stdio.h>
  20 #include <stdlib.h>
  21 #include <unistd.h>
  22 #include <errno.h>
  23
  24 #include "config.h"
  25 #include "mp_msg.h"
  26 #include "help_mp.h"
  27
  28 #include "stream/stream.h"
  29 #include "demuxer.h"
  30 #include "stheader.h"
  31
  32 #include "aviheader.h"
  33 #include "libavutil/common.h"
  34
  35 static MainAVIHeader avih;
  36
  37 void print_avih(MainAVIHeader *h, int verbose_level);
  38 void print_avih_flags(MainAVIHeader *h, int verbose_level);
  39 void print_strh(AVIStreamHeader *h, int verbose_level);
  40 void print_wave_header(WAVEFORMATEX *h, int verbose_level);
  41 void print_video_header(BITMAPINFOHEADER *h, int verbose_level);
  42 void print_index(AVIINDEXENTRY *idx,int idx_size, int verbose_level);
  43 void print_avistdindex_chunk(avistdindex_chunk *h, int verbose_level);
  44 void print_avisuperindex_chunk(avisuperindex_chunk *h, int verbose_level);
  45 void print_vprp(VideoPropHeader *vprp, int verbose_level);
  46
  47 static int odml_get_vstream_id(int id, unsigned char res[])
  48 {
  49     unsigned char *p = (unsigned char *)&id;
  50     id = le2me_32(id);
  51
  52     if (p[2] == 'd') {
  53         if (res) {
  54             res[0] = p[0];
  55             res[1] = p[1];
  56         }
  57         return 1;
  58     }
  59     return 0;
  60 }
  61
  62 static int avi_idx_cmp(const void *elem1,const void *elem2) {
  63   register off_t a = AVI_IDX_OFFSET((AVIINDEXENTRY *)elem1);
  64   register off_t b = AVI_IDX_OFFSET((AVIINDEXENTRY *)elem2);
  65   return (a > b) - (b > a);
  66 }
  67
  68 void read_avi_header(demuxer_t *demuxer,int index_mode){
  69 sh_audio_t *sh_audio=NULL;
  70 sh_video_t *sh_video=NULL;
  71 int stream_id=-1;
  72 int idxfix_videostream=0;
  73 int idxfix_divx=0;
  74 avi_priv_t* priv=demuxer->priv;
  75 off_t list_end=0;
  76
  77 //---- AVI header:
  78 priv->idx_size=0;
  79 priv->audio_streams=0;
  80 while(1){
  81   int id=stream_read_dword_le(demuxer->stream);
  82   unsigned chunksize,size2;
  83   static int last_fccType=0;
  84   static int last_fccHandler=0;
  85   char* hdr=NULL;
  86   //
  87   if(stream_eof(demuxer->stream)) break;
  88   // Imply -forceidx if -saveidx is specified
  89   if (index_file_save)
  90     index_mode = 2;
  91   //
  92   if(id==mmioFOURCC('L','I','S','T')){
  93     unsigned len=stream_read_dword_le(demuxer->stream);   // list size
  94     id=stream_read_dword_le(demuxer->stream);             // list type
  95     mp_msg(MSGT_HEADER,MSGL_DBG2,"LIST %.4s  len=%u\n",(char *) &id,len);
  96     if(len >= 4) {
  97         len -= 4;
  98         list_end=stream_tell(demuxer->stream)+((len+1)&(~1));
  99     } else {
 100         mp_tmsg(MSGT_HEADER,MSGL_WARN,"** empty list?!\n");
 101         list_end = 0;
 102     }
 103     mp_msg(MSGT_HEADER,MSGL_V,"list_end=0x%X\n",(int)list_end);
 104     if(id==listtypeAVIMOVIE){
 105       // found MOVI header
 106       if(!demuxer->movi_start) demuxer->movi_start=stream_tell(demuxer->stream);
 107       demuxer->movi_end=stream_tell(demuxer->stream)+len;
 108       mp_tmsg(MSGT_HEADER,MSGL_V,"Found movie at 0x%X - 0x%X\n",(int)demuxer->movi_start,(int)demuxer->movi_end);
 109       if(demuxer->stream->end_pos>demuxer->movi_end) demuxer->movi_end=demuxer->stream->end_pos;
 110       if(index_mode==-2 || index_mode==2 || index_mode==0)
 111         break; // reading from non-seekable source (stdin) or forced index or no index forced
 112       if(list_end>0) stream_seek(demuxer->stream,list_end); // skip movi
 113       list_end=0;
 114     }
 115     continue;
 116   }
 117   size2=stream_read_dword_le(demuxer->stream);
 118   mp_msg(MSGT_HEADER,MSGL_DBG2,"CHUNK %.4s  len=%u\n",(char *) &id,size2);
 119   chunksize=(size2+1)&(~1);
 120   switch(id){
 121
 122     // Indicates where the subject of the file is archived
 123     case mmioFOURCC('I','A','R','L'): hdr="Archival Location";break;
 124     // Lists the artist of the original subject of the file;
 125     // for example, "Michaelangelo."
 126     case mmioFOURCC('I','A','R','T'): hdr="Artist";break;
 127     // Lists the name of the person or organization that commissioned
 128     // the subject of the file; for example "Pope Julian II."
 129     case mmioFOURCC('I','C','M','S'): hdr="Commissioned";break;
 130     // Provides general comments about the file or the subject
 131     // of the file. If the comment is several sentences long, end each
 132     // sentence with a period. Do not include new-line characters.
 133     case mmioFOURCC('I','C','M','T'): hdr="Comments";break;
 134     // Records the copyright information for the file; for example,
 135     // "Copyright Encyclopedia International 1991." If there are multiple
 136     // copyrights, separate them by semicolon followed by a space.
 137     case mmioFOURCC('I','C','O','P'): hdr="Copyright";break;
 138     // Describes whether an image has been cropped and, if so, how it
 139     // was cropped; for example, "lower-right corner."
 140     case mmioFOURCC('I','C','R','D'): hdr="Creation Date";break;
 141     // Describes whether an image has been cropped and, if so, how it
 142     // was cropped; for example, "lower-right corner."
 143     case mmioFOURCC('I','C','R','P'): hdr="Cropped";break;
 144     // Specifies the size of the original subject of the file; for
 145     // example, "8.5 in h, 11 in w."
 146     case mmioFOURCC('I','D','I','M'): hdr="Dimensions";break;
 147     // Stores dots per inch setting of the digitizer used to
 148     // produce the file, such as "300."
 149     case mmioFOURCC('I','D','P','I'): hdr="Dots Per Inch";break;
 150     // Stores the of the engineer who worked on the file. If there are
 151     // multiple engineers, separate the names by a semicolon and a blank;
 152     // for example, "Smith, John; Adams, Joe."
 153     case mmioFOURCC('I','E','N','G'): hdr="Engineer";break;
 154     // Describes the original work, such as "landscape,", "portrait,"
 155     // "still liefe," etc.
 156     case mmioFOURCC('I','G','N','R'): hdr="Genre";break;
 157     // Provides a list of keywords that refer to the file or subject of the
 158     // file. Separate multiple keywords with a semicolon and a blank;
 159     // for example, "Seattle, aerial view; scenery."
 160     case mmioFOURCC('I','K','E','Y'): hdr="Keywords";break;
 161     // ILGT - Describes the changes in the lightness settings on the digitizer
 162     // required to produce the file. Note that the format of this information
 163     // depends on the hardware used.
 164     case mmioFOURCC('I','L','G','T'): hdr="Lightness";break;
 165     // IMED - Decribes the original subject of the file, such as
 166     // "computer image," "drawing," "lithograph," and so on.
 167     case mmioFOURCC('I','M','E','D'): hdr="Medium";break;
 168     // INAM - Stores the title of the subject of the file, such as
 169     // "Seattle from Above."
 170     case mmioFOURCC('I','N','A','M'): hdr="Title";break;
 171     // IPLT - Specifies the number of colors requested when digitizing
 172     // an image, such as "256."
 173     case mmioFOURCC('I','P','L','T'): hdr="Palette Setting";break;
 174     // IPRD - Specifies the name of title the file was originally intended
 175     // for, such as "Encyclopedia of Pacific Northwest Geography."
 176     case mmioFOURCC('I','P','R','D'): hdr="Product";break;
 177     // ISBJ - Decsribes the contents of the file, such as
 178     // "Aerial view of Seattle."
 179     case mmioFOURCC('I','S','B','J'): hdr="Subject";break;
 180     // ISFT - Identifies the name of the software packages used to create the
 181     // file, such as "Microsoft WaveEdit"
 182     case mmioFOURCC('I','S','F','T'): hdr="Software";break;
 183     // ISHP - Identifies the change in sharpness for the digitizer
 184     // required to produce the file (the format depends on the hardware used).
 185     case mmioFOURCC('I','S','H','P'): hdr="Sharpness";break;
 186     // ISRC - Identifies the name of the person or organization who
 187     // suplied the original subject of the file; for example, "Try Research."
 188     case mmioFOURCC('I','S','R','C'): hdr="Source";break;
 189     // ISRF - Identifies the original form of the material that was digitized,
 190     // such as "slide," "paper," "map," and so on. This is not necessarily
 191     // the same as IMED
 192     case mmioFOURCC('I','S','R','F'): hdr="Source Form";break;
 193     // ITCH - Identifies the technician who digitized the subject file;
 194     // for example, "Smith, John."
 195     case mmioFOURCC('I','T','C','H'): hdr="Technician";break;
 196     case mmioFOURCC('I','S','M','P'): hdr="Time Code";break;
 197     case mmioFOURCC('I','D','I','T'): hdr="Digitization Time";break;
 198
 199     case ckidAVIMAINHDR:          // read 'avih'
 200       stream_read(demuxer->stream,(char*) &avih,FFMIN(size2,sizeof(avih)));
 201       le2me_MainAVIHeader(&avih); // swap to machine endian
 202       chunksize-=FFMIN(size2,sizeof(avih));
 203       if( mp_msg_test(MSGT_HEADER,MSGL_V) ) print_avih(&avih,MSGL_V); // else print_avih_flags(&avih,MSGL_V);
 204       break;
 205     case ckidSTREAMHEADER: {      // read 'strh'
 206       AVIStreamHeader h;
 207       stream_read(demuxer->stream,(char*) &h,FFMIN(size2,sizeof(h)));
 208       le2me_AVIStreamHeader(&h);  // swap to machine endian
 209       chunksize-=FFMIN(size2,sizeof(h));
 210       ++stream_id;
 211       if(h.fccType==streamtypeVIDEO){
 212         sh_video=new_sh_video(demuxer,stream_id);
 213         mp_tmsg(MSGT_DEMUX, MSGL_INFO, "[%s] Video stream found, -vid %d\n", "aviheader", stream_id);
 214         memcpy(&sh_video->video,&h,sizeof(h));
 215         sh_video->stream_delay = (float)sh_video->video.dwStart * sh_video->video.dwScale/sh_video->video.dwRate;
 216       } else
 217       if(h.fccType==streamtypeAUDIO){
 218         sh_audio=new_sh_audio(demuxer,stream_id);
 219         mp_tmsg(MSGT_DEMUX, MSGL_INFO, "[%s] Audio stream found, -aid %d\n", "aviheader", stream_id);
 220         memcpy(&sh_audio->audio,&h,sizeof(h));
 221         sh_audio->stream_delay = (float)sh_audio->audio.dwStart * sh_audio->audio.dwScale/sh_audio->audio.dwRate;
 222         sh_audio->needs_parsing = 1;
 223       }
 224       last_fccType=h.fccType;
 225       last_fccHandler=h.fccHandler;
 226       if( mp_msg_test(MSGT_HEADER,MSGL_V) ) print_strh(&h,MSGL_V);
 227       break; }
 228     case mmioFOURCC('i', 'n', 'd', 'x'): {
 229       uint32_t i;
 230       avisuperindex_chunk *s;
 231
 232       if(!index_mode) break;
 233
 234       if(chunksize<=24){
 235         break;
 236       }
 237       priv->suidx_size++;
 238       priv->suidx = realloc_struct(priv->suidx, priv->suidx_size, sizeof (avisuperindex_chunk));
 239       s = &priv->suidx[priv->suidx_size-1];
 240
 241       chunksize-=24;
 242       memcpy(s->fcc, "indx", 4);
 243       s->dwSize = size2;
 244       s->wLongsPerEntry = stream_read_word_le(demuxer->stream);
 245       s->bIndexSubType = stream_read_char(demuxer->stream);
 246       s->bIndexType = stream_read_char(demuxer->stream);
 247       s->nEntriesInUse = stream_read_dword_le(demuxer->stream);
 248       *(uint32_t *)s->dwChunkId = stream_read_dword_le(demuxer->stream);
 249       stream_read(demuxer->stream, (char *)s->dwReserved, 3*4);
 250       memset(s->dwReserved, 0, 3*4);
 251
 252       print_avisuperindex_chunk(s,MSGL_V);
 253
 254       // Check and fix this useless crap
 255       if(s->wLongsPerEntry != sizeof (avisuperindex_entry)/4) {
 256           mp_msg (MSGT_HEADER, MSGL_WARN, "Broken super index chunk size: %u\n",s->wLongsPerEntry);
 257           s->wLongsPerEntry = sizeof(avisuperindex_entry)/4;
 258       }
 259       if( ((chunksize/4)/s->wLongsPerEntry) < s->nEntriesInUse){
 260         mp_msg (MSGT_HEADER, MSGL_WARN, "Broken super index chunk\n");
 261         s->nEntriesInUse = (chunksize/4)/s->wLongsPerEntry;
 262       }
 263
 264       s->aIndex = calloc(s->nEntriesInUse, sizeof (avisuperindex_entry));
 265       s->stdidx = calloc(s->nEntriesInUse, sizeof (avistdindex_chunk));
 266
 267       // now the real index of indices
 268       for (i=0; i<s->nEntriesInUse; i++) {
 269           chunksize-=16;
 270           s->aIndex[i].qwOffset = stream_read_qword_le(demuxer->stream);
 271           s->aIndex[i].dwSize = stream_read_dword_le(demuxer->stream);
 272           s->aIndex[i].dwDuration = stream_read_dword_le(demuxer->stream);
 273           mp_msg (MSGT_HEADER, MSGL_V, "ODML (%.4s): [%d] 0x%016"PRIx64" 0x%04x %u\n",
 274                   (s->dwChunkId), i,
 275                   (uint64_t)s->aIndex[i].qwOffset, s->aIndex[i].dwSize, s->aIndex[i].dwDuration);
 276       }
 277
 278       break; }
 279     case ckidSTREAMFORMAT: {      // read 'strf'
 280       if(last_fccType==streamtypeVIDEO){
 281         sh_video->bih=calloc(FFMAX(chunksize, sizeof(BITMAPINFOHEADER)), 1);
 282 //        sh_video->bih=malloc(chunksize); memset(sh_video->bih,0,chunksize);
 283         mp_tmsg(MSGT_HEADER,MSGL_V,"Found 'bih', %u bytes of %d\n",chunksize,sizeof(BITMAPINFOHEADER));
 284         stream_read(demuxer->stream,(char*) sh_video->bih,chunksize);
 285         le2me_BITMAPINFOHEADER(sh_video->bih);  // swap to machine endian
 286         if (sh_video->bih->biSize > chunksize && sh_video->bih->biSize > sizeof(BITMAPINFOHEADER))
 287                 sh_video->bih->biSize = chunksize;
 288         // fixup MS-RLE header (seems to be broken for <256 color files)
 289         if(sh_video->bih->biCompression<=1 && sh_video->bih->biSize==40)
 290             sh_video->bih->biSize=chunksize;
 291         if( mp_msg_test(MSGT_HEADER,MSGL_V) ) print_video_header(sh_video->bih,MSGL_V);
 292         chunksize=0;
 293         sh_video->fps=(float)sh_video->video.dwRate/(float)sh_video->video.dwScale;
 294         sh_video->frametime=(float)sh_video->video.dwScale/(float)sh_video->video.dwRate;
 295         sh_video->format = sh_video->bih->biCompression;
 296 //        if(demuxer->video->id==-1) demuxer->video->id=stream_id;
 297         // IdxFix:
 298         idxfix_videostream=stream_id;
 299         switch(sh_video->bih->biCompression){
 300         case mmioFOURCC('M', 'P', 'G', '4'):
 301         case mmioFOURCC('m', 'p', 'g', '4'):
 302         case mmioFOURCC('D', 'I', 'V', '1'):
 303           idxfix_divx=3; // set index recovery mpeg4 flavour: msmpeg4v1
 304           mp_tmsg(MSGT_HEADER,MSGL_V,"Regenerating keyframe table for M$ mpg4v1 video.\n");
 305           break;
 306         case mmioFOURCC('D', 'I', 'V', '3'):
 307         case mmioFOURCC('d', 'i', 'v', '3'):
 308         case mmioFOURCC('D', 'I', 'V', '4'):
 309         case mmioFOURCC('d', 'i', 'v', '4'):
 310         case mmioFOURCC('D', 'I', 'V', '5'):
 311         case mmioFOURCC('d', 'i', 'v', '5'):
 312         case mmioFOURCC('D', 'I', 'V', '6'):
 313         case mmioFOURCC('d', 'i', 'v', '6'):
 314         case mmioFOURCC('M', 'P', '4', '3'):
 315         case mmioFOURCC('m', 'p', '4', '3'):
 316         case mmioFOURCC('M', 'P', '4', '2'):
 317         case mmioFOURCC('m', 'p', '4', '2'):
 318         case mmioFOURCC('D', 'I', 'V', '2'):
 319         case mmioFOURCC('A', 'P', '4', '1'):
 320           idxfix_divx=1; // set index recovery mpeg4 flavour: msmpeg4v3
 321           mp_tmsg(MSGT_HEADER,MSGL_V,"Regenerating keyframe table for DIVX3 video.\n");
 322           break;
 323         case mmioFOURCC('D', 'I', 'V', 'X'):
 324         case mmioFOURCC('d', 'i', 'v', 'x'):
 325         case mmioFOURCC('D', 'X', '5', '0'):
 326         case mmioFOURCC('X', 'V', 'I', 'D'):
 327         case mmioFOURCC('x', 'v', 'i', 'd'):
 328         case mmioFOURCC('F', 'M', 'P', '4'):
 329         case mmioFOURCC('f', 'm', 'p', '4'):
 330           idxfix_divx=2; // set index recovery mpeg4 flavour: generic mpeg4
 331           mp_tmsg(MSGT_HEADER,MSGL_V,"Regenerating keyframe table for MPEG-4 video.\n");
 332           break;
 333         }
 334       } else
 335       if(last_fccType==streamtypeAUDIO){
 336         unsigned wf_size = chunksize<sizeof(WAVEFORMATEX)?sizeof(WAVEFORMATEX):chunksize;
 337         sh_audio->wf=calloc(wf_size,1);
 338 //        sh_audio->wf=malloc(chunksize); memset(sh_audio->wf,0,chunksize);
 339         mp_tmsg(MSGT_HEADER,MSGL_V,"Found 'wf', %d bytes of %d\n",chunksize,sizeof(WAVEFORMATEX));
 340         stream_read(demuxer->stream,(char*) sh_audio->wf,chunksize);
 341         le2me_WAVEFORMATEX(sh_audio->wf);
 342         if (sh_audio->wf->cbSize != 0 &&
 343             wf_size < sizeof(WAVEFORMATEX)+sh_audio->wf->cbSize) {
 344             sh_audio->wf=realloc(sh_audio->wf, sizeof(WAVEFORMATEX)+sh_audio->wf->cbSize);
 345         }
 346         sh_audio->format=sh_audio->wf->wFormatTag;
 347         if (sh_audio->format == 1 &&
 348             last_fccHandler == mmioFOURCC('A', 'x', 'a', 'n'))
 349             sh_audio->format = last_fccHandler;
 350         sh_audio->i_bps=sh_audio->wf->nAvgBytesPerSec;
 351         chunksize=0;
 352         if( mp_msg_test(MSGT_HEADER,MSGL_V) ) print_wave_header(sh_audio->wf,MSGL_V);
 353         ++priv->audio_streams;
 354 //        if(demuxer->audio->id==-1) demuxer->audio->id=stream_id;
 355       }
 356       break;
 357     }
 358     case mmioFOURCC('v', 'p', 'r', 'p'): {
 359         VideoPropHeader *vprp = malloc(chunksize);
 360         unsigned int i;
 361         stream_read(demuxer->stream, (void*)vprp, chunksize);
 362         le2me_VideoPropHeader(vprp);
 363         chunksize -= sizeof(*vprp)-sizeof(vprp->FieldInfo);
 364         chunksize /= sizeof(VIDEO_FIELD_DESC);
 365         if (vprp->nbFieldPerFrame > chunksize) {
 366             vprp->nbFieldPerFrame = chunksize;
 367         }
 368         chunksize = 0;
 369         for (i=0; i<vprp->nbFieldPerFrame; i++) {
 370                 le2me_VIDEO_FIELD_DESC(&vprp->FieldInfo[i]);
 371         }
 372         if (sh_video) {
 373                 sh_video->aspect = GET_AVI_ASPECT(vprp->dwFrameAspectRatio);
 374         }
 375         if( mp_msg_test(MSGT_HEADER,MSGL_V) ) print_vprp(vprp,MSGL_V);
 376         free(vprp);
 377         break;
 378     }
 379     case mmioFOURCC('d', 'm', 'l', 'h'): {
 380         // dmlh 00 00 00 04 frms
 381         unsigned int total_frames = stream_read_dword_le(demuxer->stream);
 382         mp_tmsg(MSGT_HEADER,MSGL_V,"AVI: dmlh found (size=%d) (total_frames=%d)\n", chunksize, total_frames);
 383         stream_skip(demuxer->stream, chunksize-4);
 384         chunksize = 0;
 385     }
 386     break;
 387     case ckidAVINEWINDEX:
 388     if(demuxer->movi_end>stream_tell(demuxer->stream))
 389         demuxer->movi_end=stream_tell(demuxer->stream); // fixup movi-end
 390     if(index_mode && !priv->isodml){
 391       int i;
 392       priv->idx_size=size2>>4;
 393       mp_tmsg(MSGT_HEADER,MSGL_V,"Reading INDEX block, %d chunks for %d frames (fpos=%"PRId64").\n",
 394         priv->idx_size,avih.dwTotalFrames, (int64_t)stream_tell(demuxer->stream));
 395       priv->idx=malloc(priv->idx_size<<4);
 396 //      printf("\nindex to %p !!!!! (priv=%p)\n",priv->idx,priv);
 397       stream_read(demuxer->stream,(char*)priv->idx,priv->idx_size<<4);
 398       for (i = 0; i < priv->idx_size; i++) {    // swap index to machine endian
 399         AVIINDEXENTRY *entry=(AVIINDEXENTRY*)priv->idx + i;
 400         le2me_AVIINDEXENTRY(entry);
 401         /*
 402          * We (ab)use the upper word for bits 32-47 of the offset, so
 403          * we'll clear them here.
 404          * FIXME: AFAIK no codec uses them, but if one does it will break
 405          */
 406         entry->dwFlags&=0xffff;
 407       }
 408       chunksize-=priv->idx_size<<4;
 409       if( mp_msg_test(MSGT_HEADER,MSGL_DBG2) ) print_index(priv->idx,priv->idx_size,MSGL_DBG2);
 410     }
 411     break;
 412     /* added May 2002 */
 413     case mmioFOURCC('R','I','F','F'): {
 414         char riff_type[4];
 415
 416         mp_tmsg(MSGT_HEADER, MSGL_V, "Additional RIFF header...\n");
 417         stream_read(demuxer->stream, riff_type, sizeof riff_type);
 418         if (strncmp(riff_type, "AVIX", sizeof riff_type))
 419             mp_tmsg(MSGT_HEADER, MSGL_WARN, "** Warning: this is no extended AVI header..\n");
 420         else {
 421                 /*
 422                  * We got an extended AVI header, so we need to switch to
 423                  * ODML to get seeking to work, provided we got indx chunks
 424                  * in the header (suidx_size > 0).
 425                  */
 426                 if (priv->suidx_size > 0)
 427                         priv->isodml = 1;
 428         }
 429         chunksize = 0;
 430         list_end = 0; /* a new list will follow */
 431         break; }
 432     case ckidAVIPADDING:
 433         stream_skip(demuxer->stream, chunksize);
 434         chunksize = 0;
 435         break;
 436   }
 437   if(hdr){
 438     mp_msg(MSGT_HEADER,MSGL_V,"hdr=%s  size=%u\n",hdr,size2);
 439     if(size2==3)
 440       chunksize=1; // empty
 441     else {
 442       char buf[256];
 443       int len=(size2<250)?size2:250;
 444       stream_read(demuxer->stream,buf,len);
 445       chunksize-=len;
 446       buf[len]=0;
 447       mp_msg(MSGT_HEADER,MSGL_V,"%-10s: %s\n",hdr,buf);
 448       demux_info_add(demuxer, hdr, buf);
 449     }
 450   }
 451   mp_msg(MSGT_HEADER,MSGL_DBG2,"list_end=0x%"PRIX64"  pos=0x%"PRIX64"  chunksize=0x%"PRIX64"  next=0x%"PRIX64"\n",
 452       (int64_t)list_end, (int64_t)stream_tell(demuxer->stream),
 453       (int64_t)chunksize, (int64_t)chunksize+(int64_t)stream_tell(demuxer->stream));
 454   if(list_end>0 &&
 455      chunksize+stream_tell(demuxer->stream) == list_end) list_end=0;
 456   if(list_end>0 && chunksize+stream_tell(demuxer->stream)>list_end){
 457       mp_tmsg(MSGT_HEADER,MSGL_V,"Broken chunk? chunksize=%d (id=%.4s)\n",chunksize,(char *) &id);
 458       stream_seek(demuxer->stream,list_end);
 459       list_end=0;
 460   } else
 461   if(chunksize>0) stream_skip(demuxer->stream,chunksize); else
 462   if((int)chunksize<0) mp_msg(MSGT_HEADER,MSGL_WARN,"chunksize=%u  (id=%.4s)\n",chunksize,(char *) &id);
 463
 464 }
 465
 466 if (priv->suidx_size > 0 && priv->idx_size == 0) {
 467     /*
 468      * No NEWAVIINDEX, but we got an OpenDML index.
 469      */
 470     priv->isodml = 1;
 471 }
 472
 473 if (priv->isodml && (index_mode==-1 || index_mode==0 || index_mode==1)) {
 474     int i, j, k;
 475
 476     avisuperindex_chunk *cx;
 477     AVIINDEXENTRY *idx;
 478
 479
 480     if (priv->idx_size) free(priv->idx);
 481     priv->idx_size = 0;
 482     priv->idx_offset = 0;
 483     priv->idx = NULL;
 484
 485     mp_tmsg(MSGT_HEADER, MSGL_INFO, "AVI: ODML: Building ODML index (%d superindexchunks).\n", priv->suidx_size);
 486
 487     // read the standard indices
 488     for (cx = &priv->suidx[0], i=0; i<priv->suidx_size; cx++, i++) {
 489         stream_reset(demuxer->stream);
 490         for (j=0; j<cx->nEntriesInUse; j++) {
 491             int ret1, ret2;
 492             memset(&cx->stdidx[j], 0, 32);
 493             ret1 = stream_seek(demuxer->stream, (off_t)cx->aIndex[j].qwOffset);
 494             ret2 = stream_read(demuxer->stream, (char *)&cx->stdidx[j], 32);
 495             if (ret1 != 1 || ret2 != 32 || cx->stdidx[j].nEntriesInUse==0) {
 496                 // this is a broken file (probably incomplete) let the standard
 497                 // gen_index routine handle this
 498                 priv->isodml = 0;
 499                 priv->idx_size = 0;
 500                 mp_tmsg(MSGT_HEADER, MSGL_WARN, "AVI: ODML: Broken (incomplete?) file detected. Will use traditional index.\n");
 501                 goto freeout;
 502             }
 503
 504             le2me_AVISTDIDXCHUNK(&cx->stdidx[j]);
 505             print_avistdindex_chunk(&cx->stdidx[j],MSGL_V);
 506             priv->idx_size += cx->stdidx[j].nEntriesInUse;
 507             cx->stdidx[j].aIndex = malloc(cx->stdidx[j].nEntriesInUse*sizeof(avistdindex_entry));
 508             stream_read(demuxer->stream, (char *)cx->stdidx[j].aIndex,
 509                     cx->stdidx[j].nEntriesInUse*sizeof(avistdindex_entry));
 510             for (k=0;k<cx->stdidx[j].nEntriesInUse; k++)
 511                 le2me_AVISTDIDXENTRY(&cx->stdidx[j].aIndex[k]);
 512
 513             cx->stdidx[j].dwReserved3 = 0;
 514
 515         }
 516     }
 517
 518     /*
 519      * We convert the index by translating all entries into AVIINDEXENTRYs
 520      * and sorting them by offset.  The result should be the same index
 521      * we would get with -forceidx.
 522      */
 523
 524     idx = priv->idx = malloc(priv->idx_size * sizeof (AVIINDEXENTRY));
 525
 526     for (cx = priv->suidx; cx != &priv->suidx[priv->suidx_size]; cx++) {
 527         avistdindex_chunk *sic;
 528         for (sic = cx->stdidx; sic != &cx->stdidx[cx->nEntriesInUse]; sic++) {
 529             avistdindex_entry *sie;
 530             for (sie = sic->aIndex; sie != &sic->aIndex[sic->nEntriesInUse]; sie++) {
 531                 uint64_t off = sic->qwBaseOffset + sie->dwOffset - 8;
 532                 memcpy(&idx->ckid, sic->dwChunkId, 4);
 533                 idx->dwChunkOffset = off;
 534                 idx->dwFlags = (off >> 32) << 16;
 535                 idx->dwChunkLength = sie->dwSize & 0x7fffffff;
 536                 idx->dwFlags |= (sie->dwSize&0x80000000)?0x0:AVIIF_KEYFRAME; // bit 31 denotes !keyframe
 537                 idx++;
 538             }
 539         }
 540     }
 541     qsort(priv->idx, priv->idx_size, sizeof(AVIINDEXENTRY), avi_idx_cmp);
 542
 543     /*
 544        Hack to work around a "wrong" index in some divx odml files
 545        (processor_burning.avi as an example)
 546        They have ##dc on non keyframes but the ix00 tells us they are ##db.
 547        Read the fcc of a non-keyframe vid frame and check it.
 548      */
 549
 550     {
 551         uint32_t id;
 552         uint32_t db = 0;
 553         stream_reset (demuxer->stream);
 554
 555         // find out the video stream id. I have seen files with 01db.
 556         for (idx = &((AVIINDEXENTRY *)priv->idx)[0], i=0; i<priv->idx_size; i++, idx++){
 557             unsigned char res[2];
 558             if (odml_get_vstream_id(idx->ckid, res)) {
 559                 db = mmioFOURCC(res[0], res[1], 'd', 'b');
 560                 break;
 561             }
 562         }
 563
 564         // find first non keyframe
 565         for (idx = &((AVIINDEXENTRY *)priv->idx)[0], i=0; i<priv->idx_size; i++, idx++){
 566             if (!(idx->dwFlags & AVIIF_KEYFRAME) && idx->ckid == db) break;
 567         }
 568         if (i<priv->idx_size && db) {
 569             stream_seek(demuxer->stream, AVI_IDX_OFFSET(idx));
 570             id = stream_read_dword_le(demuxer->stream);
 571             if (id && id != db) // index fcc and real fcc differ? fix it.
 572                 for (idx = &((AVIINDEXENTRY *)priv->idx)[0], i=0; i<priv->idx_size; i++, idx++){
 573                     if (!(idx->dwFlags & AVIIF_KEYFRAME) && idx->ckid == db)
 574                         idx->ckid = id;
 575             }
 576         }
 577     }
 578
 579     if ( mp_msg_test(MSGT_HEADER,MSGL_DBG2) ) print_index(priv->idx, priv->idx_size,MSGL_DBG2);
 580
 581     demuxer->movi_end=demuxer->stream->end_pos;
 582
 583 freeout:
 584
 585     // free unneeded stuff
 586     cx = &priv->suidx[0];
 587     do {
 588         for (j=0;j<cx->nEntriesInUse;j++)
 589             if (cx->stdidx[j].nEntriesInUse) free(cx->stdidx[j].aIndex);
 590         free(cx->stdidx);
 591
 592     } while (cx++ != &priv->suidx[priv->suidx_size-1]);
 593     free(priv->suidx);
 594
 595 }
 596
 597 /* Read a saved index file */
 598 if (index_file_load) {
 599   FILE *fp;
 600   char magic[7];
 601   unsigned int i;
 602
 603   if ((fp = fopen(index_file_load, "r")) == NULL) {
 604     mp_tmsg(MSGT_HEADER,MSGL_ERR, "Can't read index file %s: %s\n", index_file_load, strerror(errno));
 605     goto gen_index;
 606   }
 607   fread(&magic, 6, 1, fp);
 608   if (strncmp(magic, "MPIDX1", 6)) {
 609     mp_tmsg(MSGT_HEADER,MSGL_ERR, "%s is not a valid MPlayer index file.\n", index_file_load);
 610     goto gen_index;
 611   }
 612   fread(&priv->idx_size, sizeof(priv->idx_size), 1, fp);
 613   priv->idx=malloc(priv->idx_size*sizeof(AVIINDEXENTRY));
 614   if (!priv->idx) {
 615     mp_tmsg(MSGT_HEADER,MSGL_ERR, "Could not allocate memory for index data from %s.\n", index_file_load);
 616     priv->idx_size = 0;
 617     goto gen_index;
 618   }
 619
 620   for (i=0; i<priv->idx_size;i++) {
 621     AVIINDEXENTRY *idx;
 622     idx=&((AVIINDEXENTRY *)priv->idx)[i];
 623     fread(idx, sizeof(AVIINDEXENTRY), 1, fp);
 624     if (feof(fp)) {
 625       mp_tmsg(MSGT_HEADER,MSGL_ERR, "premature end of index file %s\n", index_file_load);
 626       free(priv->idx);
 627       priv->idx_size = 0;
 628       goto gen_index;
 629     }
 630   }
 631   fclose(fp);
 632   mp_tmsg(MSGT_HEADER,MSGL_INFO, "Loaded index file: %s\n", index_file_load);
 633 }
 634 gen_index:
 635 if(index_mode>=2 || (priv->idx_size==0 && index_mode==1)){
 636   int idx_pos = 0;
 637   // build index for file:
 638   stream_reset(demuxer->stream);
 639   stream_seek(demuxer->stream,demuxer->movi_start);
 640
 641   priv->idx_size=0;
 642   priv->idx=NULL;
 643
 644   while(1){
 645     int id;
 646     unsigned len;
 647     off_t skip;
 648     AVIINDEXENTRY* idx;
 649     unsigned int c;
 650     demuxer->filepos=stream_tell(demuxer->stream);
 651     if(demuxer->filepos>=demuxer->movi_end && demuxer->movi_start<demuxer->movi_end) break;
 652     id=stream_read_dword_le(demuxer->stream);
 653     len=stream_read_dword_le(demuxer->stream);
 654     if(id==mmioFOURCC('L','I','S','T') || id==mmioFOURCC('R', 'I', 'F', 'F')){
 655       id=stream_read_dword_le(demuxer->stream); // list or RIFF type
 656       continue;
 657     }
 658     if(stream_eof(demuxer->stream)) break;
 659     if(!id || avi_stream_id(id)==100) goto skip_chunk; // bad ID (or padding?)
 660
 661     if(idx_pos>=priv->idx_size){
 662 //      priv->idx_size+=32;
 663       priv->idx_size+=1024; // +16kB
 664       priv->idx=realloc(priv->idx,priv->idx_size*sizeof(AVIINDEXENTRY));
 665       if(!priv->idx){idx_pos=0; break;} // error!
 666     }
 667     idx=&((AVIINDEXENTRY *)priv->idx)[idx_pos++];
 668     idx->ckid=id;
 669     idx->dwFlags=AVIIF_KEYFRAME; // FIXME
 670     idx->dwFlags|=(demuxer->filepos>>16)&0xffff0000U;
 671     idx->dwChunkOffset=(unsigned long)demuxer->filepos;
 672     idx->dwChunkLength=len;
 673
 674     c=stream_read_dword(demuxer->stream);
 675
 676     if(!len) idx->dwFlags&=~AVIIF_KEYFRAME;
 677
 678     // Fix keyframes for DivX files:
 679     if(idxfix_divx)
 680       if(avi_stream_id(id)==idxfix_videostream){
 681         switch(idxfix_divx){
 682             case 3: c=stream_read_dword(demuxer->stream)<<5; //skip 32+5 bits for m$mpeg4v1
 683             case 1: if(c&0x40000000) idx->dwFlags&=~AVIIF_KEYFRAME;break; // divx 3
 684             case 2: if(c==0x1B6) idx->dwFlags&=~AVIIF_KEYFRAME;break; // divx 4
 685         }
 686       }
 687
 688     // update status line:
 689     { static off_t lastpos;
 690       off_t pos;
 691       off_t len=demuxer->movi_end-demuxer->movi_start;
 692       if(len){
 693           pos=100*(demuxer->filepos-demuxer->movi_start)/len; // %
 694       } else {
 695           pos=(demuxer->filepos-demuxer->movi_start)>>20; // MB
 696       }
 697       if(pos!=lastpos){
 698           lastpos=pos;
 699           mp_tmsg(MSGT_HEADER,MSGL_STATUS, "Generating Index: %3lu %s     \r",
 700                  (unsigned long)pos, len?"%":"MB");
 701       }
 702     }
 703     mp_dbg(MSGT_HEADER,MSGL_DBG2,"%08X %08X %.4s %08X %X\n",(unsigned int)demuxer->filepos,id,(char *) &id,(int)c,(unsigned int) idx->dwFlags);
 704 #if 0
 705     { unsigned char tmp[64];
 706       int i;
 707       stream_read(demuxer->stream,tmp,64);
 708       printf("%.4s",&id);
 709       for(i=0;i<64;i++) printf(" %02X",tmp[i]);
 710       printf("\n");
 711     }
 712 #endif
 713 skip_chunk:
 714     skip=(len+1)&(~1UL); // total bytes in this chunk
 715     stream_seek(demuxer->stream,8+demuxer->filepos+skip);
 716   }
 717   priv->idx_size=idx_pos;
 718   mp_tmsg(MSGT_HEADER,MSGL_INFO,"AVI: Generated index table for %d chunks!\n",priv->idx_size);
 719   if( mp_msg_test(MSGT_HEADER,MSGL_DBG2) ) print_index(priv->idx,priv->idx_size,MSGL_DBG2);
 720
 721   /* Write generated index to a file */
 722   if (index_file_save) {
 723     FILE *fp;
 724     unsigned int i;
 725
 726     if ((fp=fopen(index_file_save, "w")) == NULL) {
 727       mp_tmsg(MSGT_HEADER,MSGL_ERR, "Couldn't write index file %s: %s\n", index_file_save, strerror(errno));
 728       return;
 729     }
 730     fwrite("MPIDX1", 6, 1, fp);
 731     fwrite(&priv->idx_size, sizeof(priv->idx_size), 1, fp);
 732     for (i=0; i<priv->idx_size; i++) {
 733       AVIINDEXENTRY *idx = &((AVIINDEXENTRY *)priv->idx)[i];
 734       fwrite(idx, sizeof(AVIINDEXENTRY), 1, fp);
 735     }
 736     fclose(fp);
 737     mp_tmsg(MSGT_HEADER,MSGL_INFO, "Saved index file: %s\n", index_file_save);
 738   }
 739 }
 740 }