git.sesse.net Git - vlc/blob - modules/demux/real.c

   1 /*****************************************************************************
   2  * real.c: Real demuxer.
   3  *****************************************************************************
   4  * Copyright (C) 2004, 2006-2007 the VideoLAN team
   5  * $Id$
   6  *
   7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
   8  *
   9  * This program is free software; you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation; either version 2 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program; if not, write to the Free Software
  21  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  22  *****************************************************************************/
  23
  24 /*****************************************************************************
  25  * Preamble
  26  *****************************************************************************/
  27 #include <vlc/vlc.h>
  28
  29 #include <stdio.h>
  30 #include <stdlib.h>                                      /* malloc(), free() */
  31
  32 #include <vlc_demux.h>
  33 #include <vlc_charset.h>
  34 #include <vlc_meta.h>
  35
  36 /*****************************************************************************
  37  * Module descriptor
  38  *****************************************************************************/
  39 static int  Open    ( vlc_object_t * );
  40 static void Close  ( vlc_object_t * );
  41
  42 vlc_module_begin();
  43     set_description( _("Real demuxer" ) );
  44     set_capability( "demux2", 15 );
  45     set_category( CAT_INPUT );
  46     set_subcategory( SUBCAT_INPUT_DEMUX );
  47     set_callbacks( Open, Close );
  48     add_shortcut( "real" );
  49     add_shortcut( "rm" );
  50 vlc_module_end();
  51
  52 /*****************************************************************************
  53  * Local prototypes
  54  *****************************************************************************/
  55
  56 typedef struct
  57 {
  58     int         i_id;
  59     es_format_t fmt;
  60
  61     es_out_id_t *p_es;
  62
  63     int         i_frame;
  64     block_t     *p_frame;
  65
  66     int         i_subpacket_h;
  67     int         i_subpacket_size;
  68     int         i_coded_frame_size;
  69     int         i_frame_size;
  70
  71     int         i_subpacket;
  72     int         i_subpackets;
  73     block_t     **p_subpackets;
  74     int         i_out_subpacket;
  75
  76 } real_track_t;
  77
  78 struct demux_sys_t
  79 {
  80     int64_t  i_data_offset;
  81     int64_t  i_data_size;
  82     uint32_t i_data_packets_count;
  83     uint32_t i_data_packets;
  84     int64_t  i_data_offset_next;
  85
  86     int  i_our_duration;
  87     int  i_mux_rate;
  88
  89     char* psz_title;
  90     char* psz_artist;
  91     char* psz_copyright;
  92     char* psz_description;
  93
  94     int          i_track;
  95     real_track_t **track;
  96
  97     uint8_t buffer[65536];
  98
  99     int64_t     i_pcr;
 100 };
 101
 102 static int Demux( demux_t *p_demux );
 103 static int Control( demux_t *p_demux, int i_query, va_list args );
 104
 105 static int HeaderRead( demux_t *p_demux );
 106 static int ReadCodecSpecificData( demux_t *p_demux, int i_len, int i_num );
 107
 108 /*****************************************************************************
 109  * Open
 110  *****************************************************************************/
 111 static int Open( vlc_object_t *p_this )
 112 {
 113     demux_t     *p_demux = (demux_t*)p_this;
 114     demux_sys_t *p_sys;
 115
 116     const uint8_t *p_peek;
 117
 118     if( stream_Peek( p_demux->s, &p_peek, 10 ) < 10 ) return VLC_EGENERIC;
 119     if( memcmp( p_peek, ".RMF", 4 ) ) return VLC_EGENERIC;
 120
 121     /* Fill p_demux field */
 122     p_demux->pf_demux = Demux;
 123     p_demux->pf_control = Control;
 124     p_demux->p_sys = p_sys = malloc( sizeof( demux_sys_t ) );
 125     memset( p_sys, 0, sizeof( demux_sys_t ) );
 126     p_sys->i_data_offset = 0;
 127     p_sys->i_track = 0;
 128     p_sys->track   = NULL;
 129     p_sys->i_pcr   = 1;
 130
 131
 132     /* Parse the headers */
 133     if( HeaderRead( p_demux ) )
 134     {
 135         int i;
 136         msg_Err( p_demux, "invalid header" );
 137         for( i = 0; i < p_sys->i_track; i++ )
 138         {
 139             real_track_t *tk = p_sys->track[i];
 140
 141             if( tk->p_es )
 142             {
 143                 es_out_Del( p_demux->out, tk->p_es );
 144             }
 145             free( tk );
 146         }
 147         if( p_sys->i_track > 0 )
 148         {
 149             free( p_sys->track );
 150         }
 151         free( p_sys );
 152         return VLC_EGENERIC;
 153     }
 154
 155     return VLC_SUCCESS;
 156 }
 157
 158 /*****************************************************************************
 159  * Close
 160  *****************************************************************************/
 161 static void Close( vlc_object_t *p_this )
 162 {
 163     demux_t *p_demux = (demux_t*)p_this;
 164     demux_sys_t *p_sys = p_demux->p_sys;
 165     int i;
 166
 167     for( i = 0; i < p_sys->i_track; i++ )
 168     {
 169         real_track_t *tk = p_sys->track[i];
 170         int j = tk->i_subpackets;
 171
 172         if( tk->p_frame ) block_Release( tk->p_frame );
 173         es_format_Clean( &tk->fmt );
 174
 175         while(  j-- )
 176         {
 177             if( tk->p_subpackets[ j ] )
 178                 block_Release( tk->p_subpackets[ j ] );
 179         }
 180         if( tk->i_subpackets ) free( tk->p_subpackets );
 181
 182         free( tk );
 183     }
 184
 185     if( p_sys->psz_title ) free( p_sys->psz_title );
 186     if( p_sys->psz_artist ) free( p_sys->psz_artist );
 187     if( p_sys->psz_copyright ) free( p_sys->psz_copyright );
 188     if( p_sys->psz_description ) free( p_sys->psz_description );
 189
 190     if( p_sys->i_track > 0 ) free( p_sys->track );
 191     free( p_sys );
 192 }
 193
 194
 195 /*****************************************************************************
 196  * Demux:
 197  *****************************************************************************/
 198 static int Demux( demux_t *p_demux )
 199 {
 200     demux_sys_t *p_sys = p_demux->p_sys;
 201     uint8_t     header[18];
 202     int         i_size, i_id, i_flags, i;
 203     int64_t     i_pts;
 204     real_track_t *tk = NULL;
 205     vlc_bool_t  b_selected;
 206
 207     if( p_sys->i_data_packets >= p_sys->i_data_packets_count &&
 208         p_sys->i_data_packets_count )
 209     {
 210         if( stream_Read( p_demux->s, header, 18 ) < 18 )
 211         {
 212             return 0;
 213         }
 214         if( strncmp( (char *)header, "DATA", 4 ) )
 215         {
 216             return 0;
 217         }
 218         p_sys->i_data_offset = stream_Tell( p_demux->s ) - 18;
 219         p_sys->i_data_size   = GetDWBE( &header[4] );
 220         p_sys->i_data_packets_count = GetDWBE( &header[10] );
 221         p_sys->i_data_packets = 0;
 222         p_sys->i_data_offset_next = GetDWBE( &header[14] );
 223
 224         msg_Dbg( p_demux, "entering new DATA packets=%d next=%u",
 225                  p_sys->i_data_packets_count,
 226                  (uint32_t)p_sys->i_data_offset_next );
 227     }
 228
 229     if( stream_Read( p_demux->s, header, 12 ) < 12 ) return 0;
 230
 231     i_size = GetWBE( &header[2] ) - 12;
 232     i_id   = GetWBE( &header[4] );
 233     i_pts  = 1000 * GetDWBE( &header[6] );
 234     i_pts += 1000; /* Avoid 0 pts */
 235     i_flags= header[11]; /* flags 0x02 -> keyframe */
 236
 237 #if 0
 238     msg_Dbg( p_demux, "packet %d size=%d id=%d pts=%u",
 239              p_sys->i_data_packets, i_size, i_id, (uint32_t)(i_pts/1000) );
 240 #endif
 241
 242     p_sys->i_data_packets++;
 243
 244     stream_Read( p_demux->s, p_sys->buffer, i_size );
 245
 246     for( i = 0; i < p_sys->i_track; i++ )
 247     {
 248         if( p_sys->track[i]->i_id == i_id ) tk = p_sys->track[i];
 249     }
 250
 251     if( tk == NULL )
 252     {
 253         msg_Warn( p_demux, "unknown track id(0x%x)", i_id );
 254         return 1;
 255     }
 256     es_out_Control( p_demux->out, ES_OUT_GET_ES_STATE, tk->p_es, &b_selected );
 257
 258     if( tk->fmt.i_cat == VIDEO_ES && b_selected )
 259     {
 260         uint8_t *p = p_sys->buffer;
 261
 262         while( p < &p_sys->buffer[i_size - 2] )
 263         {
 264             uint8_t h = *p++;
 265             int     i_len = 0;
 266             int     i_copy;
 267             int     i_subseq = 0;
 268             int     i_seqnum = 0;
 269             int     i_offset = 0;
 270
 271             if( (h&0xc0) == 0x40 )
 272             {
 273                 /* Short header */
 274                 p++;
 275                 i_len = &p_sys->buffer[i_size] - p;
 276             }
 277             else
 278             {
 279                 if( (h&0x40) == 0 )
 280                 {
 281                     i_subseq = (*p++)&0x7f;
 282                 }
 283                 i_len = (p[0] << 8)|p[1]; p += 2;
 284                 if( (i_len&0xc000) == 0 )
 285                 {
 286                     i_len <<= 16;
 287                     i_len |= (p[0] << 8)|p[1]; p += 2;
 288                     i_len &= 0x3fffffff;
 289                 }
 290                 else
 291                 {
 292                     i_len &= 0x3fff;
 293                 }
 294
 295                 i_offset = (p[0] << 8)|p[1]; p += 2;
 296                 if( (i_offset&0xc000) == 0 )
 297                 {
 298                     i_offset <<= 16;
 299                     i_offset |= (p[0] << 8)|p[1]; p += 2;
 300                     i_offset &= 0x3fffffff;
 301                 }
 302                 else
 303                 {
 304                     i_offset &= 0x3fff;
 305                 }
 306                 i_seqnum = *p++;
 307             }
 308
 309             i_copy = i_len - i_offset;
 310             if( i_copy > &p_sys->buffer[i_size] - p )
 311             {
 312                 i_copy = &p_sys->buffer[i_size] - p;
 313             }
 314             else if( i_copy < 0 )
 315             {
 316                 break;
 317             }
 318
 319             msg_Dbg( p_demux, "    - len=%d offset=%d size=%d subseq=%d seqnum=%d",
 320                      i_len, i_offset, i_copy, i_subseq, i_seqnum );
 321
 322             if( (h&0xc0) == 0x80 )
 323             {
 324                 /* last fragment -> fixes */
 325                 i_copy = i_offset;
 326                 i_offset = i_len - i_copy;
 327                 msg_Dbg( p_demux, "last fixing copy=%d offset=%d",
 328                          i_copy, i_offset );
 329             }
 330
 331             if( tk->p_frame &&
 332                 ( tk->p_frame->i_dts != i_pts ||
 333                   tk->i_frame != i_len ) )
 334             {
 335                 msg_Dbg( p_demux, "sending size=%d", tk->p_frame->i_buffer );
 336
 337                 if( p_sys->i_pcr < tk->p_frame->i_dts )
 338                 {
 339                     p_sys->i_pcr = tk->p_frame->i_dts;
 340
 341                     es_out_Control( p_demux->out, ES_OUT_SET_PCR,
 342                                     (int64_t)p_sys->i_pcr );
 343                 }
 344                 es_out_Send( p_demux->out, tk->p_es, tk->p_frame );
 345
 346                 tk->i_frame = 0;
 347                 tk->p_frame = NULL;
 348             }
 349
 350             if( (h&0xc0) != 0x80 && (h&0xc0) != 0x00 && !tk->p_frame )
 351             {
 352                 /* no fragment */
 353                 i_len = i_copy;
 354                 i_offset = 0;
 355             }
 356
 357
 358             if( tk->p_frame == NULL )
 359             {
 360                 msg_Dbg( p_demux, "new frame size=%d", i_len );
 361                 tk->i_frame = i_len;
 362                 if( !( tk->p_frame = block_New( p_demux, i_len + 8 + 1000) ) )
 363                 {
 364                     return -1;
 365                 }
 366                 memset( &tk->p_frame->p_buffer[8], 0, i_len );
 367                 tk->p_frame->i_dts = i_pts;
 368                 tk->p_frame->i_pts = i_pts;
 369
 370                 ((uint32_t*)tk->p_frame->p_buffer)[0] = i_len;  /* len */
 371                 ((uint32_t*)tk->p_frame->p_buffer)[1] = 0;      /* chunk counts */
 372             }
 373
 374             if( i_offset < tk->i_frame)
 375             {
 376                 int i_ck = ((uint32_t*)tk->p_frame->p_buffer)[1]++;
 377
 378                 msg_Dbg( p_demux, "copying new buffer n=%d offset=%d copy=%d",
 379                          i_ck, i_offset, i_copy );
 380
 381                 ((uint32_t*)(tk->p_frame->p_buffer+i_len+8))[i_ck] = i_offset;
 382
 383                 memcpy( &tk->p_frame->p_buffer[i_offset + 8], p, i_copy );
 384             }
 385
 386             p += i_copy;
 387
 388             if( (h&0xc0) != 0x80 )
 389             {
 390                 break;
 391             }
 392
 393 #if 0
 394             if( tk->p_frame )
 395             {
 396                 /* append data */
 397                 int i_ck = ((uint32_t*)tk->p_frame->p_buffer)[1]++;
 398
 399                 if( (h&0xc0) == 0x80 )
 400                 {
 401                     /* last fragment */
 402                     i_copy = i_offset;
 403                     i_offset = i_len - i_offset;
 404
 405                     ((uint32_t*)(tk->p_frame->p_buffer+i_len+8))[i_ck] = i_offset;
 406                     memcpy( &tk->p_frame->p_buffer[i_offset+ 8], p, i_copy );
 407                     p += i_copy;
 408
 409                     if( p_sys->i_pcr < tk->p_frame->i_dts )
 410                     {
 411                         p_sys->i_pcr = tk->p_frame->i_dts;
 412                         es_out_Control( p_demux->out, ES_OUT_SET_PCR,
 413                                         (int64_t)p_sys->i_pcr );
 414                     }
 415                     es_out_Send( p_demux->out, tk->p_es, tk->p_frame );
 416
 417                     tk->i_frame = 0;
 418                     tk->p_frame = NULL;
 419
 420                     continue;
 421                 }
 422
 423                 ((uint32_t*)(tk->p_frame->p_buffer+i_len+8))[i_ck] = i_offset;
 424                 memcpy( &tk->p_frame->p_buffer[i_offset + 8], p, i_copy );
 425                 break;
 426             }
 427
 428             if( (h&0xc0) != 0x00 )
 429             {
 430                 block_t *p_frame;
 431
 432                 /* not fragmented */
 433                 if( !( p_frame = block_New( p_demux, i_copy + 8 + 8 ) ) )
 434                 {
 435                     return -1;
 436                 }
 437                 p_frame->i_dts = i_pts;
 438                 p_frame->i_pts = i_pts;
 439
 440                 ((uint32_t*)p_frame->p_buffer)[0] = i_copy;
 441                 ((uint32_t*)p_frame->p_buffer)[1] = 1;
 442                 ((uint32_t*)(p_frame->p_buffer+i_copy+8))[0] = 0;
 443                 memcpy( &p_frame->p_buffer[8], p, i_copy );
 444
 445                 p += i_copy;
 446
 447                 if( p_sys->i_pcr < p_frame->i_dts )
 448                 {
 449                     p_sys->i_pcr = p_frame->i_dts;
 450                     es_out_Control( p_demux->out, ES_OUT_SET_PCR,
 451                                     (int64_t)p_sys->i_pcr );
 452                 }
 453                 es_out_Send( p_demux->out, tk->p_es, p_frame );
 454             }
 455             else
 456             {
 457                 /* First fragment */
 458                 tk->i_frame = i_len;
 459                 if( !( tk->p_frame = block_New( p_demux, i_len + 8 + 1000) ) )
 460                 {
 461                     return -1;
 462                 }
 463                 memset( &tk->p_frame->p_buffer[8], 0, i_len );
 464                 tk->p_frame->i_dts = i_pts;
 465                 tk->p_frame->i_pts = i_pts;
 466
 467                 ((uint32_t*)tk->p_frame->p_buffer)[0] = i_len;  /* len */
 468                 ((uint32_t*)tk->p_frame->p_buffer)[1] = 1;      /* chunk counts */
 469                 ((uint32_t*)(tk->p_frame->p_buffer+i_len+8))[0] = i_offset;
 470                 memcpy( &tk->p_frame->p_buffer[i_offset + 8], p, i_copy );
 471
 472                 break;
 473             }
 474 #endif
 475         }
 476     }
 477     else if( tk->fmt.i_cat == AUDIO_ES && b_selected )
 478     {
 479         /* Set PCR */
 480         if( p_sys->i_pcr < i_pts )
 481         {
 482             p_sys->i_pcr = i_pts;
 483             es_out_Control( p_demux->out, ES_OUT_SET_PCR,
 484                             (int64_t)p_sys->i_pcr );
 485         }
 486
 487         if( tk->fmt.i_codec == VLC_FOURCC( 'm', 'p', '4', 'a' ) )
 488         {
 489             int     i_sub = (p_sys->buffer[1] >> 4)&0x0f;
 490             uint8_t *p_sub = &p_sys->buffer[2+2*i_sub];
 491
 492             int i;
 493             for( i = 0; i < i_sub; i++ )
 494             {
 495                 int i_sub_size = GetWBE( &p_sys->buffer[2+i*2]);
 496                 block_t *p_block = block_New( p_demux, i_sub_size );
 497                 if( p_block )
 498                 {
 499                     memcpy( p_block->p_buffer, p_sub, i_sub_size );
 500                     p_sub += i_sub_size;
 501
 502                     p_block->i_dts =
 503                     p_block->i_pts = ( i == 0 ? i_pts : 0 );
 504
 505                     es_out_Send( p_demux->out, tk->p_es, p_block );
 506                 }
 507             }
 508         }
 509         else if( tk->fmt.i_codec == VLC_FOURCC( 'c', 'o', 'o', 'k' ) ||
 510                  tk->fmt.i_codec == VLC_FOURCC( 'a', 't', 'r', 'c') ||
 511                  tk->fmt.i_codec == VLC_FOURCC( '2', '8', '_', '8') )
 512         {
 513             uint8_t *p_buf = p_sys->buffer;
 514             int y = tk->i_subpacket / ( tk->i_frame_size /tk->i_subpacket_size);
 515             int i_index, i;
 516
 517             /* Sanity check */
 518             if( i_flags & 2 ) y = tk->i_subpacket = 0;
 519
 520             if( tk->fmt.i_codec == VLC_FOURCC( 'c', 'o', 'o', 'k' ) ||
 521                 tk->fmt.i_codec == VLC_FOURCC( 'a', 't', 'r', 'c' ) )
 522             for( i = 0; i < tk->i_frame_size / tk->i_subpacket_size; i++ )
 523             {
 524                 block_t *p_block = block_New( p_demux, tk->i_subpacket_size );
 525                 memcpy( p_block->p_buffer, p_buf, tk->i_subpacket_size );
 526                 p_buf += tk->i_subpacket_size;
 527
 528                 i_index = tk->i_subpacket_h * i +
 529                     ((tk->i_subpacket_h + 1) / 2) * (y&1) + (y>>1);
 530
 531                 p_block->i_dts = p_block->i_pts = i_pts;
 532                 tk->p_subpackets[i_index] = p_block;
 533                 tk->i_subpacket++;
 534             }
 535
 536             if( tk->fmt.i_codec == VLC_FOURCC( '2', '8', '_', '8' ) )
 537             for( i = 0; i < tk->i_subpacket_h / 2; i++ )
 538             {
 539                 block_t *p_block = block_New( p_demux, tk->i_coded_frame_size);
 540                 memcpy( p_block->p_buffer, p_buf, tk->i_coded_frame_size );
 541                 p_buf += tk->i_coded_frame_size;
 542
 543                 i_index = (i * 2 * tk->i_frame_size) /
 544                     tk->i_coded_frame_size + y;
 545
 546                 p_block->i_dts = p_block->i_pts = i_pts;
 547                 tk->p_subpackets[i_index] = p_block;
 548                 tk->i_subpacket++;
 549             }
 550
 551             while( tk->i_out_subpacket != tk->i_subpackets &&
 552                    tk->p_subpackets[tk->i_out_subpacket] )
 553             {
 554                 block_t *p_block = tk->p_subpackets[tk->i_out_subpacket];
 555                 tk->p_subpackets[tk->i_out_subpacket] = 0;
 556
 557                 if( tk->i_out_subpacket ) p_block->i_dts = p_block->i_pts = 0;
 558                 es_out_Send( p_demux->out, tk->p_es, p_block );
 559
 560                 tk->i_out_subpacket++;
 561             }
 562
 563             if( tk->i_subpacket == tk->i_subpackets &&
 564                 tk->i_out_subpacket != tk->i_subpackets )
 565             {
 566                 msg_Warn( p_demux, "i_subpacket != i_out_subpacket, "
 567                           "this shouldn't happen" );
 568             }
 569
 570             if( tk->i_subpacket == tk->i_subpackets )
 571             {
 572                 tk->i_subpacket = 0;
 573                 tk->i_out_subpacket = 0;
 574             }
 575         }
 576         else
 577         {
 578             block_t *p_block = block_New( p_demux, i_size );
 579
 580             if( tk->fmt.i_codec == VLC_FOURCC( 'a', '5', '2', ' ' ) )
 581             {
 582                 uint8_t *src = p_sys->buffer;
 583                 uint8_t *dst = p_block->p_buffer;
 584
 585                 /* byte swap data */
 586                 while( dst < &p_block->p_buffer[i_size- 1])
 587                 {
 588                     *dst++ = src[1];
 589                     *dst++ = src[0];
 590
 591                     src += 2;
 592                 }
 593             }
 594             else
 595             {
 596                 memcpy( p_block->p_buffer, p_sys->buffer, i_size );
 597             }
 598             p_block->i_dts = p_block->i_pts = i_pts;
 599             es_out_Send( p_demux->out, tk->p_es, p_block );
 600         }
 601     }
 602
 603
 604     return 1;
 605 }
 606
 607 /*****************************************************************************
 608  * Control:
 609  *****************************************************************************/
 610 static int Control( demux_t *p_demux, int i_query, va_list args )
 611 {
 612     demux_sys_t *p_sys = p_demux->p_sys;
 613 #if 0
 614     double f, *pf;
 615     int64_t i64;
 616 #endif
 617     int64_t *pi64;
 618
 619     switch( i_query )
 620     {
 621 #if 0
 622         case DEMUX_GET_POSITION:
 623             pf = (double*) va_arg( args, double* );
 624             i64 = stream_Size( p_demux->s );
 625             if( i64 > 0 )
 626             {
 627                 *pf = (double)stream_Tell( p_demux->s ) / (double)i64;
 628             }
 629             else
 630             {
 631                 *pf = 0.0;
 632             }
 633             return VLC_SUCCESS;
 634
 635         case DEMUX_SET_POSITION:
 636             f = (double) va_arg( args, double );
 637             i64 = stream_Size( p_demux->s );
 638
 639             es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
 640
 641             return stream_Seek( p_demux->s, (int64_t)(i64 * f) );
 642
 643         case DEMUX_GET_TIME:
 644             pi64 = (int64_t*)va_arg( args, int64_t * );
 645             if( p_sys->i_mux_rate > 0 )
 646             {
 647                 *pi64 = (int64_t)1000000 * ( stream_Tell( p_demux->s ) / 50 ) / p_sys->i_mux_rate;
 648                 return VLC_SUCCESS;
 649             }
 650             *pi64 = 0;
 651             return VLC_EGENERIC;
 652 #endif
 653
 654         case DEMUX_GET_LENGTH:
 655             pi64 = (int64_t*)va_arg( args, int64_t * );
 656
 657             /* the commented following lines are fen's implementation, which doesn't seem to
 658              * work for one reason or another -- FK */
 659             /*if( p_sys->i_mux_rate > 0 )
 660             {
 661                 *pi64 = (int64_t)1000000 * ( stream_Size( p_demux->s ) / 50 ) / p_sys->i_mux_rate;
 662                 return VLC_SUCCESS;
 663             }*/
 664             if( p_sys->i_our_duration > 0 )
 665             {
 666                 /* our stored duration is in ms, so... */
 667                 *pi64 = (int64_t)1000 * p_sys->i_our_duration;
 668
 669                 return VLC_SUCCESS;
 670             }
 671             *pi64 = 0;
 672             return VLC_EGENERIC;
 673
 674         case DEMUX_GET_META:
 675         {
 676             vlc_meta_t *p_meta = (vlc_meta_t*)va_arg( args, vlc_meta_t* );
 677
 678             /* the core will crash if we provide NULL strings, so check
 679              * every string first */
 680             if( p_sys->psz_title )
 681                 vlc_meta_SetTitle( p_meta, p_sys->psz_title );
 682             if( p_sys->psz_artist )
 683                 vlc_meta_SetArtist( p_meta, p_sys->psz_artist );
 684             if( p_sys->psz_copyright )
 685                 vlc_meta_SetCopyright( p_meta, p_sys->psz_copyright );
 686             if( p_sys->psz_description )
 687                 vlc_meta_SetDescription( p_meta, p_sys->psz_description );
 688             return VLC_SUCCESS;
 689         }
 690
 691         case DEMUX_SET_TIME:
 692         case DEMUX_GET_FPS:
 693         default:
 694             return VLC_EGENERIC;
 695     }
 696     return VLC_EGENERIC;
 697 }
 698
 699 /*****************************************************************************
 700  * HeaderRead:
 701  *****************************************************************************/
 702 static int HeaderRead( demux_t *p_demux )
 703 {
 704     demux_sys_t *p_sys = p_demux->p_sys;
 705     uint8_t header[100];    /* FIXME */
 706
 707     uint32_t    i_id;
 708     uint32_t    i_size;
 709     int64_t     i_skip;
 710     int         i_version;
 711
 712     for( ;; )
 713     {
 714         /* Read the header */
 715         if( stream_Read( p_demux->s, header, 10 ) < 10 )
 716         {
 717             return VLC_EGENERIC;
 718         }
 719         i_id        = VLC_FOURCC( header[0], header[1], header[2], header[3] );
 720         i_size      = GetDWBE( &header[4] );
 721         i_version   = GetWBE( &header[8] );
 722
 723         msg_Dbg( p_demux, "object %4.4s size=%d version=%d",
 724                  (char*)&i_id, i_size, i_version );
 725
 726         if( i_size < 10 && i_id != VLC_FOURCC('D','A','T','A') )
 727         {
 728             msg_Dbg( p_demux, "invalid size for object %4.4s", (char*)&i_id );
 729             return VLC_EGENERIC;
 730         }
 731         i_skip = i_size - 10;
 732
 733         if( i_id == VLC_FOURCC('.','R','M','F') )
 734         {
 735             if( stream_Read( p_demux->s, header, 8 ) < 8 ) return VLC_EGENERIC;
 736             msg_Dbg( p_demux, "    - file version=0x%x num headers=%d",
 737                      GetDWBE( &header[0] ), GetDWBE( &header[4] ) );
 738
 739             i_skip -= 8;
 740         }
 741         else if( i_id == VLC_FOURCC('P','R','O','P') )
 742         {
 743             int i_flags;
 744
 745             if( stream_Read(p_demux->s, header, 40) < 40 ) return VLC_EGENERIC;
 746
 747             msg_Dbg( p_demux, "    - max bitrate=%d avg bitrate=%d",
 748                      GetDWBE(&header[0]), GetDWBE(&header[4]) );
 749             msg_Dbg( p_demux, "    - max packet size=%d avg bitrate=%d",
 750                      GetDWBE(&header[8]), GetDWBE(&header[12]) );
 751             msg_Dbg( p_demux, "    - packets count=%d", GetDWBE(&header[16]) );
 752             msg_Dbg( p_demux, "    - duration=%d ms", GetDWBE(&header[20]) );
 753             msg_Dbg( p_demux, "    - preroll=%d ms", GetDWBE(&header[24]) );
 754             msg_Dbg( p_demux, "    - index offset=%d", GetDWBE(&header[28]) );
 755             msg_Dbg( p_demux, "    - data offset=%d", GetDWBE(&header[32]) );
 756             msg_Dbg( p_demux, "    - num streams=%d", GetWBE(&header[36]) );
 757
 758             /* set the duration for export in control */
 759             p_sys->i_our_duration = (int)GetDWBE(&header[20]);
 760
 761             i_flags = GetWBE(&header[38]);
 762             msg_Dbg( p_demux, "    - flags=0x%x %s%s%s",
 763                      i_flags,
 764                      i_flags&0x0001 ? "PN_SAVE_ENABLED " : "",
 765                      i_flags&0x0002 ? "PN_PERFECT_PLAY_ENABLED " : "",
 766                      i_flags&0x0004 ? "PN_LIVE_BROADCAST" : "" );
 767             i_skip -= 40;
 768         }
 769         else if( i_id == VLC_FOURCC('C','O','N','T') )
 770         {
 771             int i_len;
 772             char *psz;
 773
 774             /* FIXME FIXME: should convert from whatever the character
 775              * encoding of the input meta data is to UTF-8. */
 776
 777             stream_Read( p_demux->s, header, 2 );
 778             if( ( i_len = GetWBE( header ) ) > 0 )
 779             {
 780                 psz = malloc( i_len + 1 );
 781                 stream_Read( p_demux->s, psz, i_len );
 782                 psz[i_len] = '\0';
 783
 784                 msg_Dbg( p_demux, "    - title=`%s'", psz );
 785                 EnsureUTF8( psz );
 786                 asprintf( &p_sys->psz_title, psz );
 787                 free( psz );
 788                 i_skip -= i_len;
 789             }
 790             i_skip -= 2;
 791
 792             stream_Read( p_demux->s, header, 2 );
 793             if( ( i_len = GetWBE( header ) ) > 0 )
 794             {
 795                 psz = malloc( i_len + 1 );
 796                 stream_Read( p_demux->s, psz, i_len );
 797                 psz[i_len] = '\0';
 798
 799                 msg_Dbg( p_demux, "    - author=`%s'", psz );
 800                 EnsureUTF8( psz );
 801                 asprintf( &p_sys->psz_artist, psz );
 802                 free( psz );
 803                 i_skip -= i_len;
 804             }
 805             i_skip -= 2;
 806
 807             stream_Read( p_demux->s, header, 2 );
 808             if( ( i_len = GetWBE( header ) ) > 0 )
 809             {
 810                 psz = malloc( i_len + 1 );
 811                 stream_Read( p_demux->s, psz, i_len );
 812                 psz[i_len] = '\0';
 813
 814                 msg_Dbg( p_demux, "    - copyright=`%s'", psz );
 815                 EnsureUTF8( psz );
 816                 asprintf( &p_sys->psz_copyright, psz );
 817                 free( psz );
 818                 i_skip -= i_len;
 819             }
 820             i_skip -= 2;
 821
 822             stream_Read( p_demux->s, header, 2 );
 823             if( ( i_len = GetWBE( header ) ) > 0 )
 824             {
 825                 psz = malloc( i_len + 1 );
 826                 stream_Read( p_demux->s, psz, i_len );
 827                 psz[i_len] = '\0';
 828
 829                 msg_Dbg( p_demux, "    - comment=`%s'", psz );
 830                 EnsureUTF8( psz );
 831                 asprintf( &p_sys->psz_description, psz );
 832                 free( psz );
 833                 i_skip -= i_len;
 834             }
 835             i_skip -= 2;
 836         }
 837         else if( i_id == VLC_FOURCC('M','D','P','R') )
 838         {
 839             /* Media properties header */
 840             int  i_num;
 841             int  i_len;
 842             char *psz;
 843
 844             if( stream_Read(p_demux->s, header, 30) < 30 ) return VLC_EGENERIC;
 845             i_num = GetWBE( header );
 846             msg_Dbg( p_demux, "    - id=0x%x", i_num );
 847             msg_Dbg( p_demux, "    - max bitrate=%d avg bitrate=%d",
 848                      GetDWBE(&header[2]), GetDWBE(&header[6]) );
 849             msg_Dbg( p_demux, "    - max packet size=%d avg packet size=%d",
 850                      GetDWBE(&header[10]), GetDWBE(&header[14]) );
 851             msg_Dbg( p_demux, "    - start time=%d", GetDWBE(&header[18]) );
 852             msg_Dbg( p_demux, "    - preroll=%d", GetDWBE(&header[22]) );
 853             msg_Dbg( p_demux, "    - duration=%d", GetDWBE(&header[26]) );
 854
 855             i_skip -= 30;
 856
 857             stream_Read( p_demux->s, header, 1 );
 858             if( ( i_len = header[0] ) > 0 )
 859             {
 860                 psz = malloc( i_len + 1 );
 861                 stream_Read( p_demux->s, psz, i_len );
 862                 psz[i_len] = '\0';
 863
 864                 msg_Dbg( p_demux, "    - name=`%s'", psz );
 865                 free( psz );
 866                 i_skip -= i_len;
 867             }
 868             i_skip--;
 869
 870             stream_Read( p_demux->s, header, 1 );
 871             if( ( i_len = header[0] ) > 0 )
 872             {
 873                 psz = malloc( i_len + 1 );
 874                 stream_Read( p_demux->s, psz, i_len );
 875                 psz[i_len] = '\0';
 876
 877                 msg_Dbg( p_demux, "    - mime=`%s'", psz );
 878                 free( psz );
 879                 i_skip -= i_len;
 880             }
 881             i_skip--;
 882
 883             stream_Read( p_demux->s, header, 4 );
 884             if( ( i_len = GetDWBE( header ) ) > 0 )
 885             {
 886                 ReadCodecSpecificData( p_demux, i_len, i_num );
 887                 stream_Read( p_demux->s, NULL, i_len );
 888
 889                 i_skip -= i_len;
 890             }
 891             i_skip -= 4;
 892         }
 893         else if( i_id == VLC_FOURCC('D','A','T','A') )
 894         {
 895             stream_Read( p_demux->s, header, 8 );
 896
 897             p_sys->i_data_offset    = stream_Tell( p_demux->s ) - 10;
 898             p_sys->i_data_size      = i_size;
 899             p_sys->i_data_packets_count = GetDWBE( header );
 900             p_sys->i_data_packets   = 0;
 901             p_sys->i_data_offset_next = GetDWBE( &header[4] );
 902
 903             msg_Dbg( p_demux, "    - packets count=%d next=%u",
 904                      p_sys->i_data_packets_count,
 905                      (uint32_t)p_sys->i_data_offset_next );
 906
 907             /* we have finished the header */
 908             break;
 909         }
 910         else
 911         {
 912             /* unknow header */
 913             msg_Dbg( p_demux, "unknown chunk" );
 914         }
 915
 916         if( i_skip < 0 ) return VLC_EGENERIC;
 917         stream_Read( p_demux->s, NULL, i_skip );
 918     }
 919
 920     /* TODO read index if possible */
 921
 922     return VLC_SUCCESS;
 923 }
 924
 925 static int ReadCodecSpecificData( demux_t *p_demux, int i_len, int i_num )
 926 {
 927     demux_sys_t *p_sys = p_demux->p_sys;
 928     es_format_t fmt;
 929     real_track_t *tk;
 930     const uint8_t *p_peek;
 931
 932     msg_Dbg( p_demux, "    - specific data len=%d", i_len );
 933     if( stream_Peek(p_demux->s, &p_peek, i_len) < i_len ) return VLC_EGENERIC;
 934
 935     if( ( i_len >= 8 ) && !memcmp( &p_peek[4], "VIDO", 4 ) )
 936     {
 937         es_format_Init( &fmt, VIDEO_ES, VLC_FOURCC( p_peek[8], p_peek[9],
 938                         p_peek[10], p_peek[11] ) );
 939         fmt.video.i_width = GetWBE( &p_peek[12] );
 940         fmt.video.i_height= GetWBE( &p_peek[14] );
 941
 942         fmt.i_extra = 8;
 943         fmt.p_extra = malloc( 8 );
 944         ((uint32_t*)fmt.p_extra)[0] = GetDWBE( &p_peek[26] );
 945         ((uint32_t*)fmt.p_extra)[1] = GetDWBE( &p_peek[30] );
 946
 947         msg_Dbg( p_demux, "    - video 0x%08x 0x%08x",
 948                  ((uint32_t*)fmt.p_extra)[0], ((uint32_t*)fmt.p_extra)[1] );
 949
 950         if( GetDWBE( &p_peek[30] ) == 0x10003000 ||
 951             GetDWBE( &p_peek[30] ) == 0x10003001 )
 952         {
 953             fmt.i_codec = VLC_FOURCC( 'R','V','1','3' );
 954         }
 955         else if( GetDWBE( &p_peek[30] ) == 0x20001000 ||
 956                  GetDWBE( &p_peek[30] ) == 0x20100001 ||
 957                  GetDWBE( &p_peek[30] ) == 0x20200002 )
 958         {
 959             fmt.i_codec = VLC_FOURCC( 'R','V','2','0' );
 960         }
 961         else if( GetDWBE( &p_peek[30] ) == 0x30202002 )
 962         {
 963             fmt.i_codec = VLC_FOURCC( 'R','V','3','0' );
 964         }
 965         else if( GetDWBE( &p_peek[30] ) == 0x40000000 )
 966         {
 967             fmt.i_codec = VLC_FOURCC( 'R','V','4','0' );
 968         }
 969
 970         msg_Dbg( p_demux, "    - video %4.4s %dx%d",
 971                  (char*)&fmt.i_codec, fmt.video.i_width, fmt.video.i_height );
 972
 973         tk = malloc( sizeof( real_track_t ) );
 974         tk->i_out_subpacket = 0;
 975         tk->i_subpacket = 0;
 976         tk->i_subpackets = 0;
 977         tk->p_subpackets = NULL;
 978         tk->i_id = i_num;
 979         tk->fmt = fmt;
 980         tk->i_frame = 0;
 981         tk->p_frame = NULL;
 982         tk->p_es = es_out_Add( p_demux->out, &fmt );
 983
 984         TAB_APPEND( p_sys->i_track, p_sys->track, tk );
 985     }
 986     else if( !strncmp( (char *)p_peek, ".ra\xfd", 4 ) )
 987     {
 988         int i_version = GetWBE( &p_peek[4] );
 989         int i_header_size, i_flavor, i_coded_frame_size, i_subpacket_h;
 990         int i_frame_size, i_subpacket_size;
 991
 992         msg_Dbg( p_demux, "    - audio version=%d", i_version );
 993
 994         p_peek += 6;
 995         es_format_Init( &fmt, AUDIO_ES, 0 );
 996
 997         if( i_version == 3 )
 998         {
 999             msg_Dbg( p_demux, "    - audio version 3 is not supported!" );
1000             return VLC_EGENERIC;
1001         }
1002
1003         p_peek += 2; /* 00 00 */
1004         p_peek += 4; /* .ra4 or .ra5 */
1005         p_peek += 4; /* ?? */
1006         p_peek += 2; /* version (4 or 5) */
1007         i_header_size = GetDWBE( p_peek ); p_peek += 4; /* header size */
1008         i_flavor = GetWBE( p_peek ); p_peek += 2; /* codec flavor */
1009         i_coded_frame_size = GetDWBE( p_peek ); p_peek += 4;
1010         p_peek += 4; /* ?? */
1011         p_peek += 4; /* ?? */
1012         p_peek += 4; /* ?? */
1013         i_subpacket_h = GetWBE( p_peek ); p_peek += 2;
1014         i_frame_size = GetWBE( p_peek ); p_peek += 2;
1015         i_subpacket_size = GetWBE( p_peek ); p_peek += 2;
1016         p_peek += 2; /* ?? */
1017
1018         if( i_version == 5 ) p_peek += 6; /* 0, srate, 0 */
1019
1020         fmt.audio.i_rate = GetWBE( p_peek ); p_peek += 2;
1021         p_peek += 2; /* ?? */
1022         fmt.audio.i_bitspersample = GetWBE( p_peek ); p_peek += 2;
1023         fmt.audio.i_channels = GetWBE( p_peek ); p_peek += 2;
1024         fmt.audio.i_blockalign = i_frame_size;
1025
1026         if( i_version == 5 )
1027         {
1028             p_peek += 4; /* genr */
1029             memcpy( (char *)&fmt.i_codec, p_peek, 4 ); p_peek += 4;
1030         }
1031         else
1032         {
1033             p_peek += p_peek[0] + 1; /* descr 1 */
1034             memcpy( (char *)&fmt.i_codec, p_peek + 1, 4 ); /* descr 2 */
1035             p_peek += p_peek[0] + 1;
1036         }
1037
1038         msg_Dbg( p_demux, "    - audio codec=%4.4s channels=%d rate=%dHz",
1039                  (char*)&fmt.i_codec, fmt.audio.i_channels, fmt.audio.i_rate );
1040
1041         p_peek += 3; /* ?? */
1042         if( i_version == 5 ) p_peek++;
1043
1044         switch( fmt.i_codec )
1045         {
1046         case VLC_FOURCC( 'd', 'n', 'e', 't' ):
1047             fmt.i_codec = VLC_FOURCC( 'a', '5', '2', ' ' );
1048             break;
1049
1050         case VLC_FOURCC( 'r', 'a', 'a', 'c' ):
1051         case VLC_FOURCC( 'r', 'a', 'c', 'p' ):
1052             fmt.i_extra = GetDWBE( p_peek ); p_peek += 4;
1053             if( fmt.i_extra > 0 ) { fmt.i_extra--; p_peek++; }
1054             if( fmt.i_extra > 0 )
1055             {
1056                 fmt.p_extra = malloc( fmt.i_extra );
1057                 memcpy( fmt.p_extra, p_peek, fmt.i_extra );
1058             }
1059
1060             fmt.i_codec = VLC_FOURCC( 'm', 'p', '4', 'a' );
1061             break;
1062
1063         case VLC_FOURCC('c','o','o','k'):
1064         case VLC_FOURCC('a','t','r','c'):
1065             fmt.audio.i_blockalign = i_subpacket_size;
1066             if( !(fmt.i_extra = GetDWBE( p_peek )) ) break;
1067             fmt.p_extra = malloc( fmt.i_extra );
1068             memcpy( fmt.p_extra, p_peek + 4, fmt.i_extra );
1069             break;
1070
1071         case VLC_FOURCC('2','8','_','8'):
1072             fmt.audio.i_blockalign = i_coded_frame_size;
1073             break;
1074
1075         default:
1076             msg_Dbg( p_demux, "    - unknown audio codec=%4.4s",
1077                      (char*)&fmt.i_codec );
1078             break;
1079         }
1080
1081         if( fmt.i_codec != 0 )
1082         {
1083             int i;
1084
1085             msg_Dbg( p_demux, "        - extra data=%d", fmt.i_extra );
1086
1087             tk = malloc( sizeof( real_track_t ) );
1088             tk->i_id = i_num;
1089             tk->fmt = fmt;
1090             tk->i_frame = 0;
1091             tk->p_frame = NULL;
1092
1093             tk->i_subpacket_h = i_subpacket_h;
1094             tk->i_subpacket_size = i_subpacket_size;
1095             tk->i_coded_frame_size = i_coded_frame_size;
1096             tk->i_frame_size = i_frame_size;
1097
1098             tk->i_out_subpacket = 0;
1099             tk->i_subpacket = 0;
1100             tk->i_subpackets = 0;
1101             tk->p_subpackets = NULL;
1102             if( fmt.i_codec == VLC_FOURCC('c','o','o','k') )
1103             {
1104                 tk->i_subpackets =
1105                     i_subpacket_h * i_frame_size / tk->i_subpacket_size;
1106                 tk->p_subpackets =
1107                     malloc( tk->i_subpackets * sizeof(block_t *) );
1108             }
1109             else if( fmt.i_codec == VLC_FOURCC('2','8','_','8') )
1110             {
1111                 tk->i_subpackets =
1112                     i_subpacket_h * i_frame_size / tk->i_coded_frame_size;
1113                 tk->p_subpackets =
1114                     malloc( tk->i_subpackets * sizeof(block_t *) );
1115             }
1116
1117             for( i = 0; i < tk->i_subpackets; i++ ) tk->p_subpackets[i] = NULL;
1118
1119             tk->p_es = es_out_Add( p_demux->out, &fmt );
1120
1121             TAB_APPEND( p_sys->i_track, p_sys->track, tk );
1122         }
1123     }
1124
1125     return VLC_SUCCESS;
1126 }