git.sesse.net Git - vlc/blob - modules/mux/mp4.c

   1 /*****************************************************************************
   2  * mp4.c: mp4/mov muxer
   3  *****************************************************************************
   4  * Copyright (C) 2001, 2002, 2003 VideoLAN
   5  * $Id: mp4.c,v 1.11 2004/01/24 11:56:16 gbazin Exp $
   6  *
   7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
   8  *          Gildas Bazin <gbazin at videolan dot org>
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  23  *****************************************************************************/
  24
  25 /*****************************************************************************
  26  * Preamble
  27  *****************************************************************************/
  28 #include <stdlib.h>
  29 #include <string.h>
  30 #include <errno.h>
  31
  32 #include <vlc/vlc.h>
  33 #include <vlc/input.h>
  34 #include <vlc/sout.h>
  35
  36 #ifdef HAVE_TIME_H
  37 #include <time.h>
  38 #endif
  39
  40 #include "codecs.h"
  41
  42 /*****************************************************************************
  43  * Exported prototypes
  44  *****************************************************************************/
  45 static int  Open   ( vlc_object_t * );
  46 static void Close  ( vlc_object_t * );
  47
  48 static int Capability(sout_mux_t *, int, void *, void * );
  49 static int AddStream( sout_mux_t *, sout_input_t * );
  50 static int DelStream( sout_mux_t *, sout_input_t * );
  51 static int Mux      ( sout_mux_t * );
  52
  53 /*****************************************************************************
  54  * Module descriptor
  55  *****************************************************************************/
  56 #define FASTSTART_TEXT N_("Create \"Fast start\" files")
  57 #define FASTSTART_LONGTEXT N_( \
  58     "When this option is turned on, \"Fast start\" files will be created. " \
  59     "(\"Fast start\" files are optimized for download, allowing the user " \
  60     "to start previewing the file while it is downloading).")
  61
  62 vlc_module_begin();
  63     set_description( _("MP4/MOV muxer") );
  64
  65     add_category_hint( "MP4/MOV muxer", NULL, VLC_TRUE );
  66         add_bool( "mp4-faststart", 1, NULL, FASTSTART_TEXT, FASTSTART_LONGTEXT, VLC_TRUE );
  67
  68     set_capability( "sout mux", 5 );
  69     add_shortcut( "mp4" );
  70     add_shortcut( "mov" );
  71     set_callbacks( Open, Close );
  72 vlc_module_end();
  73
  74 /*****************************************************************************
  75  * Local prototypes
  76  *****************************************************************************/
  77 typedef struct
  78 {
  79     uint64_t i_pos;
  80     int      i_size;
  81
  82     mtime_t  i_pts;
  83     mtime_t  i_dts;
  84     mtime_t  i_length;
  85
  86 } mp4_entry_t;
  87
  88 typedef struct
  89 {
  90     es_format_t   *p_fmt;
  91     int           i_track_id;
  92
  93     /* index */
  94     unsigned int i_entry_count;
  95     unsigned int i_entry_max;
  96     mp4_entry_t  *entry;
  97
  98     /* stats */
  99     mtime_t      i_duration;
 100
 101     /* for later stco fix-up (fast start files) */
 102     uint64_t i_stco_pos;
 103     vlc_bool_t b_stco64;
 104
 105 } mp4_stream_t;
 106
 107 struct sout_mux_sys_t
 108 {
 109     vlc_bool_t b_mov;
 110     vlc_bool_t b_64_ext;
 111     vlc_bool_t b_fast_start;
 112
 113     uint64_t i_mdat_pos;
 114     uint64_t i_pos;
 115
 116     mtime_t  i_start_dts;
 117
 118     int          i_nb_streams;
 119     mp4_stream_t **pp_streams;
 120 };
 121
 122 typedef struct bo_t
 123 {
 124     vlc_bool_t b_grow;
 125
 126     int        i_buffer_size;
 127     int        i_buffer;
 128     uint8_t    *p_buffer;
 129
 130 } bo_t;
 131
 132 static void bo_init     ( bo_t *, int , uint8_t *, vlc_bool_t  );
 133 static void bo_add_8    ( bo_t *, uint8_t );
 134 static void bo_add_16be ( bo_t *, uint16_t );
 135 static void bo_add_24be ( bo_t *, uint32_t );
 136 static void bo_add_32be ( bo_t *, uint32_t );
 137 static void bo_add_64be ( bo_t *, uint64_t );
 138 static void bo_add_fourcc(bo_t *, char * );
 139 static void bo_add_bo   ( bo_t *, bo_t * );
 140 static void bo_add_mem  ( bo_t *, int , uint8_t * );
 141 static void bo_add_descr( bo_t *, uint8_t , uint32_t );
 142
 143 static void bo_fix_32be ( bo_t *, int , uint32_t );
 144
 145 static bo_t *box_new     ( char *fcc );
 146 static bo_t *box_full_new( char *fcc, uint8_t v, uint32_t f );
 147 static void  box_fix     ( bo_t *box );
 148 static void  box_free    ( bo_t *box );
 149 static void  box_gather  ( bo_t *box, bo_t *box2 );
 150
 151 static void box_send( sout_mux_t *p_mux,  bo_t *box );
 152
 153 static int64_t get_timestamp();
 154
 155 static sout_buffer_t *bo_to_sout( sout_instance_t *p_sout,  bo_t *box );
 156
 157 /*****************************************************************************
 158  * Open:
 159  *****************************************************************************/
 160 static int Open( vlc_object_t *p_this )
 161 {
 162     sout_mux_t      *p_mux = (sout_mux_t*)p_this;
 163     sout_mux_sys_t  *p_sys;
 164     bo_t            *box;
 165
 166     p_sys = malloc( sizeof( sout_mux_sys_t ) );
 167     p_sys->i_pos        = 0;
 168     p_sys->i_nb_streams = 0;
 169     p_sys->pp_streams   = NULL;
 170     p_sys->i_mdat_pos   = 0;
 171     p_sys->b_mov        = p_mux->psz_mux && !strcmp( p_mux->psz_mux, "mov" );
 172     p_sys->i_start_dts  = 0;
 173
 174     msg_Dbg( p_mux, "Open" );
 175
 176     p_mux->pf_capacity  = Capability;
 177     p_mux->pf_addstream = AddStream;
 178     p_mux->pf_delstream = DelStream;
 179     p_mux->pf_mux       = Mux;
 180     p_mux->p_sys        = p_sys;
 181
 182     if( !p_sys->b_mov )
 183     {
 184         /* Now add ftyp header */
 185         box = box_new( "ftyp" );
 186         bo_add_fourcc( box, "isom" );
 187         bo_add_32be  ( box, 0 );
 188         bo_add_fourcc( box, "mp41" );
 189         box_fix( box );
 190
 191         p_sys->i_pos += box->i_buffer;
 192         p_sys->i_mdat_pos = p_sys->i_pos;
 193
 194         box_send( p_mux, box );
 195     }
 196
 197     /* FIXME FIXME
 198      * Quicktime actually doesn't like the 64 bits extensions !!! */
 199     p_sys->b_64_ext = VLC_FALSE;
 200
 201     /* Now add mdat header */
 202     box = box_new( "mdat" );
 203     bo_add_64be  ( box, 0 ); // enough to store an extended size
 204
 205     p_sys->i_pos += box->i_buffer;
 206
 207     box_send( p_mux, box );
 208
 209     return VLC_SUCCESS;
 210 }
 211
 212 static int GetDescrLength( int i_size )
 213 {
 214     if( i_size < 0x00000080 )
 215         return 2 + i_size;
 216     else if( i_size < 0x00004000 )
 217         return 3 + i_size;
 218     else if( i_size < 0x00200000 )
 219         return 4 + i_size;
 220     else
 221         return 5 + i_size;
 222 }
 223
 224 static bo_t *GetESDS( mp4_stream_t *p_stream )
 225 {
 226     bo_t *esds;
 227     int  i_stream_type;
 228     int  i_object_type_indication;
 229     int  i_decoder_specific_info_size;
 230
 231     if( p_stream->p_fmt->i_extra > 0 )
 232     {
 233         i_decoder_specific_info_size =
 234             GetDescrLength( p_stream->p_fmt->i_extra );
 235     }
 236     else
 237     {
 238         i_decoder_specific_info_size = 0;
 239     }
 240
 241     esds = box_full_new( "esds", 0, 0 );
 242
 243     /* ES_Descr */
 244     bo_add_descr( esds, 0x03, 3 +
 245                   GetDescrLength( 13 + i_decoder_specific_info_size ) +
 246                   GetDescrLength( 1 ) );
 247     bo_add_16be( esds, p_stream->i_track_id );
 248     bo_add_8   ( esds, 0x1f );      // flags=0|streamPriority=0x1f
 249
 250     /* DecoderConfigDescr */
 251     bo_add_descr( esds, 0x04, 13 + i_decoder_specific_info_size );
 252
 253     switch( p_stream->p_fmt->i_codec )
 254     {
 255         case VLC_FOURCC( 'm', 'p', '4', 'v' ):
 256             i_object_type_indication = 0x20;
 257             break;
 258         case VLC_FOURCC( 'm', 'p', 'g', 'v' ):
 259             /* FIXME MPEG-I=0x6b, MPEG-II = 0x60 -> 0x65 */
 260             i_object_type_indication = 0x60;
 261             break;
 262         case VLC_FOURCC( 'm', 'p', '4', 'a' ):
 263             /* FIXME for mpeg2-aac == 0x66->0x68 */
 264             i_object_type_indication = 0x40;
 265             break;
 266         case VLC_FOURCC( 'm', 'p', 'g', 'a' ):
 267             i_object_type_indication =
 268                 p_stream->p_fmt->audio.i_rate < 32000 ? 0x69 : 0x6b;
 269             break;
 270         default:
 271             i_object_type_indication = 0x00;
 272             break;
 273     }
 274     i_stream_type = p_stream->p_fmt->i_cat == VIDEO_ES ? 0x04 : 0x05;
 275
 276     bo_add_8   ( esds, i_object_type_indication );
 277     bo_add_8   ( esds, ( i_stream_type << 2 ) | 1 );
 278     bo_add_24be( esds, 1024 * 1024 );       // bufferSizeDB
 279     bo_add_32be( esds, 0x7fffffff );        // maxBitrate
 280     bo_add_32be( esds, 0 );                 // avgBitrate
 281
 282     if( p_stream->p_fmt->i_extra > 0 )
 283     {
 284         int i;
 285
 286         /* DecoderSpecificInfo */
 287         bo_add_descr( esds, 0x05, p_stream->p_fmt->i_extra );
 288
 289         for( i = 0; i < p_stream->p_fmt->i_extra; i++ )
 290         {
 291             bo_add_8( esds, ((uint8_t*)p_stream->p_fmt->p_extra)[i] );
 292         }
 293     }
 294
 295     /* SL_Descr mandatory */
 296     bo_add_descr( esds, 0x06, 1 );
 297     bo_add_8    ( esds, 0x02 );  // sl_predefined
 298
 299     box_fix( esds );
 300
 301     return esds;
 302 }
 303
 304 static bo_t *GetWaveTag( mp4_stream_t *p_stream )
 305 {
 306     bo_t *wave;
 307     bo_t *box;
 308
 309     wave = box_new( "wave" );
 310
 311     box = box_new( "frma" );
 312     bo_add_fourcc( box, "mp4a" );
 313     box_fix( box );
 314     box_gather( wave, box );
 315
 316     box = box_new( "mp4a" );
 317     bo_add_32be( box, 0 );
 318     box_fix( box );
 319     box_gather( wave, box );
 320
 321     box = GetESDS( p_stream );
 322     box_fix( box );
 323     box_gather( wave, box );
 324
 325     box = box_new( "srcq" );
 326     bo_add_32be( box, 0x40 );
 327     box_fix( box );
 328     box_gather( wave, box );
 329
 330     /* wazza ? */
 331     bo_add_32be( wave, 8 ); /* new empty box */
 332     bo_add_32be( wave, 0 ); /* box label */
 333
 334     box_fix( wave );
 335
 336     return wave;
 337 }
 338
 339 /* TODO: No idea about these values */
 340 static bo_t *GetSVQ3Tag( mp4_stream_t *p_stream )
 341 {
 342     bo_t *smi = box_new( "SMI " );
 343
 344     bo_add_fourcc( smi, "SEQH" );
 345     bo_add_32be( smi, 0x5 );
 346     bo_add_32be( smi, 0xe2c0211d );
 347     bo_add_8( smi, 0xc0 );
 348     box_fix( smi );
 349
 350     return smi;
 351 }
 352
 353 static bo_t *GetUdtaTag( sout_mux_t *p_mux )
 354 {
 355     sout_mux_sys_t *p_sys = p_mux->p_sys;
 356     bo_t *udta = box_new( "udta" );
 357     int i_track;
 358
 359     /* Requirements */
 360     for( i_track = 0; i_track < p_sys->i_nb_streams; i_track++ )
 361     {
 362         mp4_stream_t *p_stream = p_sys->pp_streams[i_track];
 363
 364         if( p_stream->p_fmt->i_codec == VLC_FOURCC('m','p','4','v') ||
 365             p_stream->p_fmt->i_codec == VLC_FOURCC('m','p','4','a') )
 366         {
 367             bo_t *box = box_new( "\251req" );
 368             /* String length */
 369             bo_add_16be( box, sizeof("QuickTime 6.0 or greater") - 1);
 370             bo_add_16be( box, 0 );
 371             bo_add_mem( box, sizeof("QuickTime 6.0 or greater") - 1,
 372                         "QuickTime 6.0 or greater" );
 373             box_fix( box );
 374             box_gather( udta, box );
 375             break;
 376         }
 377     }
 378
 379     /* Encoder */
 380     {
 381         bo_t *box = box_new( "\251enc" );
 382         /* String length */
 383         bo_add_16be( box, sizeof(PACKAGE_STRING " stream output") - 1);
 384         bo_add_16be( box, 0 );
 385         bo_add_mem( box, sizeof(PACKAGE_STRING " stream output") - 1,
 386                     PACKAGE_STRING " stream output" );
 387         box_fix( box );
 388         box_gather( udta, box );
 389     }
 390
 391     box_fix( udta );
 392     return udta;
 393 }
 394
 395 static bo_t *GetSounBox( sout_mux_t *p_mux, mp4_stream_t *p_stream )
 396 {
 397     sout_mux_sys_t *p_sys = p_mux->p_sys;
 398     vlc_bool_t b_descr = VLC_FALSE;
 399     bo_t *soun;
 400     char fcc[4] = "    ";
 401     int  i;
 402
 403     switch( p_stream->p_fmt->i_codec )
 404     {
 405     case VLC_FOURCC('m','p','4','a'):
 406         memcpy( fcc, "mp4a", 4 );
 407         b_descr = VLC_TRUE;
 408         break;
 409
 410     case VLC_FOURCC('m','p','g','a'):
 411         if( p_sys->b_mov )
 412             memcpy( fcc, ".mp3", 4 );
 413         else
 414             memcpy( fcc, "mp4a", 4 );
 415         break;
 416
 417     default:
 418         memcpy( fcc, (char*)&p_stream->p_fmt->i_codec, 4 );
 419         break;
 420     }
 421
 422     soun = box_new( fcc );
 423     for( i = 0; i < 6; i++ )
 424     {
 425         bo_add_8( soun, 0 );        // reserved;
 426     }
 427     bo_add_16be( soun, 1 );         // data-reference-index
 428
 429     /* SoundDescription */
 430     if( p_sys->b_mov &&
 431         p_stream->p_fmt->i_codec == VLC_FOURCC('m','p','4','a') )
 432     {
 433         bo_add_16be( soun, 1 );     // version 1;
 434     }
 435     else
 436     {
 437         bo_add_16be( soun, 0 );     // version 0;
 438     }
 439     bo_add_16be( soun, 0 );         // revision level (0)
 440     bo_add_32be( soun, 0 );         // vendor
 441     // channel-count
 442     bo_add_16be( soun, p_stream->p_fmt->audio.i_channels );
 443     // sample size
 444     bo_add_16be( soun, p_stream->p_fmt->audio.i_bitspersample ?
 445                  p_stream->p_fmt->audio.i_bitspersample : 16 );
 446     bo_add_16be( soun, -2 );        // compression id
 447     bo_add_16be( soun, 0 );         // packet size (0)
 448     bo_add_16be( soun, p_stream->p_fmt->audio.i_rate ); // sampleratehi
 449     bo_add_16be( soun, 0 );                             // sampleratelo
 450
 451     /* Extended data for SoundDescription V1 */
 452     if( p_sys->b_mov &&
 453         p_stream->p_fmt->i_codec == VLC_FOURCC('m','p','4','a') )
 454     {
 455         /* samples per packet */
 456         bo_add_32be( soun, p_stream->p_fmt->audio.i_frame_length );
 457         bo_add_32be( soun, 1536 ); /* bytes per packet */
 458         bo_add_32be( soun, 2 );    /* bytes per frame */
 459         /* bytes per sample */
 460         bo_add_32be( soun, 2 /*p_stream->p_fmt->audio.i_bitspersample/8 */);
 461     }
 462
 463     /* Add an ES Descriptor */
 464     if( b_descr )
 465     {
 466         bo_t *box;
 467
 468         if( p_sys->b_mov &&
 469             p_stream->p_fmt->i_codec == VLC_FOURCC('m','p','4','a') )
 470         {
 471             box = GetWaveTag( p_stream );
 472         }
 473         else
 474         {
 475             box = GetESDS( p_stream );
 476         }
 477         box_fix( box );
 478         box_gather( soun, box );
 479     }
 480
 481     box_fix( soun );
 482
 483     return soun;
 484 }
 485
 486 static bo_t *GetVideBox( sout_mux_t *p_mux, mp4_stream_t *p_stream )
 487 {
 488
 489     bo_t *vide;
 490     char fcc[4] = "    ";
 491     int  i;
 492
 493     switch( p_stream->p_fmt->i_codec )
 494     {
 495     case VLC_FOURCC('m','p','4','v'):
 496     case VLC_FOURCC('m','p','g','v'):
 497         memcpy( fcc, "mp4v", 4 );
 498         break;
 499
 500     case VLC_FOURCC('M','J','P','G'):
 501         memcpy( fcc, "mjpa", 4 );
 502         break;
 503
 504     case VLC_FOURCC('S','V','Q','3'):
 505         memcpy( fcc, "SVQ3", 4 );
 506         break;
 507
 508     default:
 509         memcpy( fcc, (char*)&p_stream->p_fmt->i_codec, 4 );
 510         break;
 511     }
 512
 513     vide = box_new( fcc );
 514     for( i = 0; i < 6; i++ )
 515     {
 516         bo_add_8( vide, 0 );        // reserved;
 517     }
 518     bo_add_16be( vide, 1 );         // data-reference-index
 519
 520     bo_add_16be( vide, 0 );         // predefined;
 521     bo_add_16be( vide, 0 );         // reserved;
 522     for( i = 0; i < 3; i++ )
 523     {
 524         bo_add_32be( vide, 0 );     // predefined;
 525     }
 526
 527     bo_add_16be( vide, p_stream->p_fmt->video.i_width );  // i_width
 528     bo_add_16be( vide, p_stream->p_fmt->video.i_height ); // i_height
 529
 530     bo_add_32be( vide, 0x00480000 );                // h 72dpi
 531     bo_add_32be( vide, 0x00480000 );                // v 72dpi
 532
 533     bo_add_32be( vide, 0 );         // data size, always 0
 534     bo_add_16be( vide, 1 );         // frames count per sample
 535
 536     // compressor name;
 537     for( i = 0; i < 32; i++ )
 538     {
 539         bo_add_8( vide, 0 );
 540     }
 541
 542     bo_add_16be( vide, 0x18 );      // depth
 543     bo_add_16be( vide, 0xffff );    // predefined
 544
 545     /* add an ES Descriptor */
 546     switch( p_stream->p_fmt->i_codec )
 547     {
 548     case VLC_FOURCC('m','p','4','v'):
 549     case VLC_FOURCC('m','p','g','v'):
 550         {
 551             bo_t *esds = GetESDS( p_stream );
 552
 553             box_fix( esds );
 554             box_gather( vide, esds );
 555         }
 556         break;
 557
 558     case VLC_FOURCC('S','V','Q','3'):
 559         {
 560             bo_t *esds = GetSVQ3Tag( p_stream );
 561
 562             box_fix( esds );
 563             box_gather( vide, esds );
 564         }
 565         break;
 566
 567     default:
 568         break;
 569     }
 570
 571     box_fix( vide );
 572
 573     return vide;
 574 }
 575
 576 static bo_t *GetStblBox( sout_mux_t *p_mux, mp4_stream_t *p_stream )
 577 {
 578     sout_mux_sys_t *p_sys = p_mux->p_sys;
 579     unsigned int i_chunk, i_stsc_last_val, i_stsc_entries, i, i_index;
 580     bo_t *stbl, *stsd, *stts, *stco, *stsc, *stsz;
 581     uint32_t i_timescale;
 582
 583     stbl = box_new( "stbl" );
 584     stsd = box_full_new( "stsd", 0, 0 );
 585     bo_add_32be( stsd, 1 );
 586
 587     if( p_stream->p_fmt->i_cat == AUDIO_ES )
 588     {
 589         bo_t *soun = GetSounBox( p_mux, p_stream );
 590         box_gather( stsd, soun );
 591     }
 592     else if( p_stream->p_fmt->i_cat == VIDEO_ES )
 593     {
 594         bo_t *vide = GetVideBox( p_mux, p_stream );
 595         box_gather( stsd, vide );
 596     }
 597
 598     /* append stsd to stbl */
 599     box_fix( stsd );
 600     box_gather( stbl, stsd );
 601
 602     /* chunk offset table */
 603     p_stream->i_stco_pos = stbl->i_buffer + 16;
 604     if( p_sys->i_pos >= (((uint64_t)0x1) << 32) )
 605     {
 606         /* 64 bits version */
 607         p_stream->b_stco64 = VLC_TRUE;
 608         stco = box_full_new( "co64", 0, 0 );
 609     }
 610     else
 611     {
 612         /* 32 bits version */
 613         p_stream->b_stco64 = VLC_FALSE;
 614         stco = box_full_new( "stco", 0, 0 );
 615     }
 616     bo_add_32be( stco, 0 );     // entry-count (fixed latter)
 617
 618     /* sample to chunk table */
 619     stsc = box_full_new( "stsc", 0, 0 );
 620     bo_add_32be( stsc, 0 );     // entry-count (fixed latter)
 621
 622     for( i_chunk = 0, i_stsc_last_val = 0, i_stsc_entries = 0, i = 0;
 623          i < p_stream->i_entry_count; i_chunk++ )
 624     {
 625         int i_first = i;
 626
 627         if( p_stream->b_stco64 )
 628             bo_add_64be( stco, p_stream->entry[i].i_pos );
 629         else
 630             bo_add_32be( stco, p_stream->entry[i].i_pos );
 631
 632         while( i < p_stream->i_entry_count )
 633         {
 634             if( i + 1 < p_stream->i_entry_count &&
 635                 p_stream->entry[i].i_pos + p_stream->entry[i].i_size
 636                 != p_stream->entry[i + 1].i_pos )
 637             {
 638                 i++;
 639                 break;
 640             }
 641
 642             i++;
 643         }
 644
 645         /* Add entry to the stsc table */
 646         if( i_stsc_last_val != i - i_first )
 647         {
 648             bo_add_32be( stsc, 1 + i_chunk );   // first-chunk
 649             bo_add_32be( stsc, i - i_first ) ;  // samples-per-chunk
 650             bo_add_32be( stsc, 1 );             // sample-descr-index
 651             i_stsc_last_val = i - i_first;
 652             i_stsc_entries++;
 653         }
 654     }
 655
 656     /* Fix stco entry count */
 657     bo_fix_32be( stco, 12, i_chunk );
 658     msg_Dbg( p_mux, "created %d chunks (stco)", i_chunk );
 659
 660     /* append stco to stbl */
 661     box_fix( stco );
 662     box_gather( stbl, stco );
 663
 664     /* Fix stsc entry count */
 665     bo_fix_32be( stsc, 12, i_stsc_entries  );
 666
 667     /* append stsc to stbl */
 668     box_fix( stsc );
 669     box_gather( stbl, stsc );
 670
 671     /* add stts */
 672     stts = box_full_new( "stts", 0, 0 );
 673     bo_add_32be( stts, 0 );     // entry-count (fixed latter)
 674
 675     if( p_stream->p_fmt->i_cat == AUDIO_ES )
 676         i_timescale = p_stream->p_fmt->audio.i_rate;
 677     else
 678         i_timescale = 1001;
 679
 680     for( i = 0, i_index = 0; i < p_stream->i_entry_count; i_index++)
 681     {
 682         int64_t i_delta;
 683         int     i_first;
 684
 685         i_first = i;
 686         i_delta = p_stream->entry[i].i_length;
 687
 688         while( i < p_stream->i_entry_count )
 689         {
 690             if( i + 1 < p_stream->i_entry_count &&
 691                 p_stream->entry[i + 1].i_length != i_delta )
 692             {
 693                 i++;
 694                 break;
 695             }
 696
 697             i++;
 698         }
 699
 700         bo_add_32be( stts, i - i_first );           // sample-count
 701         bo_add_32be( stts, i_delta * (int64_t)i_timescale /
 702                      (int64_t)1000000 );            // sample-delta
 703     }
 704     bo_fix_32be( stts, 12, i_index );
 705
 706     /* append stts to stbl */
 707     box_fix( stts );
 708     box_gather( stbl, stts );
 709
 710     /* FIXME add ctts ?? FIXME */
 711
 712     stsz = box_full_new( "stsz", 0, 0 );
 713     bo_add_32be( stsz, 0 );                             // sample-size
 714     bo_add_32be( stsz, p_stream->i_entry_count );       // sample-count
 715     for( i = 0; i < p_stream->i_entry_count; i++ )
 716     {
 717         bo_add_32be( stsz, p_stream->entry[i].i_size ); // sample-size
 718     }
 719     /* append stsz to stbl */
 720     box_fix( stsz );
 721     box_gather( stbl, stsz );
 722
 723     box_fix( stbl );
 724
 725     return stbl;
 726 }
 727
 728 static uint32_t mvhd_matrix[9] =
 729     { 0x10000, 0, 0, 0, 0x10000, 0, 0, 0, 0x40000000 };
 730
 731 static bo_t *GetMoovBox( sout_mux_t *p_mux )
 732 {
 733     sout_mux_sys_t *p_sys = p_mux->p_sys;
 734
 735     bo_t            *moov, *mvhd;
 736     int             i_trak, i;
 737
 738     uint32_t        i_movie_timescale = 90000;
 739     int64_t         i_movie_duration  = 0;
 740
 741     moov = box_new( "moov" );
 742
 743     /* *** add /moov/mvhd *** */
 744     if( !p_sys->b_64_ext )
 745     {
 746         mvhd = box_full_new( "mvhd", 0, 0 );
 747         bo_add_32be( mvhd, get_timestamp() );   // creation time
 748         bo_add_32be( mvhd, get_timestamp() );   // modification time
 749         bo_add_32be( mvhd, i_movie_timescale);  // timescale
 750         bo_add_32be( mvhd, i_movie_duration );  // duration
 751     }
 752     else
 753     {
 754         mvhd = box_full_new( "mvhd", 1, 0 );
 755         bo_add_64be( mvhd, get_timestamp() );   // creation time
 756         bo_add_64be( mvhd, get_timestamp() );   // modification time
 757         bo_add_32be( mvhd, i_movie_timescale);  // timescale
 758         bo_add_64be( mvhd, i_movie_duration );  // duration
 759     }
 760     bo_add_32be( mvhd, 0x10000 );           // rate
 761     bo_add_16be( mvhd, 0x100 );             // volume
 762     bo_add_16be( mvhd, 0 );                 // reserved
 763     for( i = 0; i < 2; i++ )
 764     {
 765         bo_add_32be( mvhd, 0 );             // reserved
 766     }
 767     for( i = 0; i < 9; i++ )
 768     {
 769         bo_add_32be( mvhd, mvhd_matrix[i] );// matrix
 770     }
 771     for( i = 0; i < 6; i++ )
 772     {
 773         bo_add_32be( mvhd, 0 );             // pre-defined
 774     }
 775
 776     /* Find the 1st track id */
 777     for( i_trak = 0; i_trak < p_sys->i_nb_streams; i_trak++ )
 778     {
 779         mp4_stream_t *p_stream = p_sys->pp_streams[i_trak];
 780
 781         if( p_stream->p_fmt->i_cat == AUDIO_ES ||
 782             p_stream->p_fmt->i_cat == VIDEO_ES )
 783         {
 784             /* Found it */
 785             bo_add_32be( mvhd, p_stream->i_track_id ); // next-track-id
 786             break;
 787         }
 788     }
 789     if( i_trak == p_sys->i_nb_streams ) /* Just for sanity reasons */
 790         bo_add_32be( mvhd, 0xffffffff );
 791
 792     box_fix( mvhd );
 793     box_gather( moov, mvhd );
 794
 795     for( i_trak = 0; i_trak < p_sys->i_nb_streams; i_trak++ )
 796     {
 797         mp4_stream_t *p_stream;
 798         uint32_t     i_timescale;
 799
 800         bo_t *trak, *tkhd, *mdia, *mdhd, *hdlr;
 801         bo_t *minf, *dinf, *dref, *url, *stbl;
 802
 803         p_stream = p_sys->pp_streams[i_trak];
 804
 805         if( p_stream->p_fmt->i_cat != AUDIO_ES &&
 806             p_stream->p_fmt->i_cat != VIDEO_ES )
 807         {
 808             msg_Err( p_mux, "FIXME ignoring trak (noaudio&&novideo)" );
 809             continue;
 810         }
 811
 812         if( p_stream->p_fmt->i_cat == AUDIO_ES )
 813             i_timescale = p_stream->p_fmt->audio.i_rate;
 814         else
 815             i_timescale = 1001;
 816
 817         /* *** add /moov/trak *** */
 818         trak = box_new( "trak" );
 819
 820         /* *** add /moov/trak/tkhd *** */
 821         if( !p_sys->b_64_ext )
 822         {
 823             if( p_sys->b_mov )
 824                 tkhd = box_full_new( "tkhd", 0, 0x0f );
 825             else
 826                 tkhd = box_full_new( "tkhd", 0, 1 );
 827
 828             bo_add_32be( tkhd, get_timestamp() );       // creation time
 829             bo_add_32be( tkhd, get_timestamp() );       // modification time
 830             bo_add_32be( tkhd, p_stream->i_track_id );
 831             bo_add_32be( tkhd, 0 );                     // reserved 0
 832             bo_add_32be( tkhd, p_stream->i_duration *
 833                          (int64_t)i_movie_timescale /
 834                          (mtime_t)1000000 );            // duration
 835         }
 836         else
 837         {
 838             if( p_sys->b_mov )
 839                 tkhd = box_full_new( "tkhd", 1, 0x0f );
 840             else
 841                 tkhd = box_full_new( "tkhd", 1, 1 );
 842
 843             bo_add_64be( tkhd, get_timestamp() );       // creation time
 844             bo_add_64be( tkhd, get_timestamp() );       // modification time
 845             bo_add_32be( tkhd, p_stream->i_track_id );
 846             bo_add_32be( tkhd, 0 );                     // reserved 0
 847             bo_add_64be( tkhd, p_stream->i_duration *
 848                          (int64_t)i_movie_timescale /
 849                          (mtime_t)1000000 );            // duration
 850         }
 851
 852         for( i = 0; i < 2; i++ )
 853         {
 854             bo_add_32be( tkhd, 0 );                 // reserved
 855         }
 856         bo_add_16be( tkhd, 0 );                     // layer
 857         bo_add_16be( tkhd, 0 );                     // pre-defined
 858         // volume
 859         bo_add_16be( tkhd, p_stream->p_fmt->i_cat == AUDIO_ES ? 0x100 : 0 );
 860         bo_add_16be( tkhd, 0 );                     // reserved
 861         for( i = 0; i < 9; i++ )
 862         {
 863             bo_add_32be( tkhd, mvhd_matrix[i] );    // matrix
 864         }
 865         if( p_stream->p_fmt->i_cat == AUDIO_ES )
 866         {
 867             bo_add_32be( tkhd, 0 );                 // width (presentation)
 868             bo_add_32be( tkhd, 0 );                 // height(presentation)
 869         }
 870         else
 871         {
 872             // width (presentation)
 873             bo_add_32be( tkhd, p_stream->p_fmt->video.i_aspect *
 874                          p_stream->p_fmt->video.i_height /
 875                          VOUT_ASPECT_FACTOR << 16 );
 876             // height(presentation)
 877             bo_add_32be( tkhd, p_stream->p_fmt->video.i_height << 16 );
 878         }
 879         box_fix( tkhd );
 880         box_gather( trak, tkhd );
 881
 882         /* *** add /moov/trak/mdia *** */
 883         mdia = box_new( "mdia" );
 884
 885         /* media header */
 886         if( !p_sys->b_64_ext )
 887         {
 888             mdhd = box_full_new( "mdhd", 0, 0 );
 889             bo_add_32be( mdhd, get_timestamp() );   // creation time
 890             bo_add_32be( mdhd, get_timestamp() );   // modification time
 891             bo_add_32be( mdhd, i_timescale);        // timescale
 892             bo_add_32be( mdhd, p_stream->i_duration * (int64_t)i_timescale /
 893                                (mtime_t)1000000 );  // duration
 894         }
 895         else
 896         {
 897             mdhd = box_full_new( "mdhd", 1, 0 );
 898             bo_add_64be( mdhd, get_timestamp() );   // creation time
 899             bo_add_64be( mdhd, get_timestamp() );   // modification time
 900             bo_add_32be( mdhd, i_timescale);        // timescale
 901             bo_add_64be( mdhd, p_stream->i_duration * (int64_t)i_timescale /
 902                                (mtime_t)1000000 );  // duration
 903         }
 904
 905         bo_add_16be( mdhd, 0    );              // language   FIXME
 906         bo_add_16be( mdhd, 0    );              // predefined
 907         box_fix( mdhd );
 908         box_gather( mdia, mdhd );
 909
 910         /* handler reference */
 911         hdlr = box_full_new( "hdlr", 0, 0 );
 912
 913         bo_add_fourcc( hdlr, "mhlr" );         // media handler
 914         if( p_stream->p_fmt->i_cat == AUDIO_ES )
 915         {
 916             bo_add_fourcc( hdlr, "soun" );
 917         }
 918         else
 919         {
 920             bo_add_fourcc( hdlr, "vide" );
 921         }
 922
 923         bo_add_32be( hdlr, 0 );         // reserved
 924         bo_add_32be( hdlr, 0 );         // reserved
 925         bo_add_32be( hdlr, 0 );         // reserved
 926
 927         if( p_stream->p_fmt->i_cat == AUDIO_ES )
 928         {
 929             bo_add_8( hdlr, 13 );
 930             bo_add_mem( hdlr, 13, "SoundHandler" );
 931         }
 932         else
 933         {
 934             bo_add_8( hdlr, 13 );
 935             bo_add_mem( hdlr, 13, "VideoHandler" );
 936         }
 937
 938         box_fix( hdlr );
 939         box_gather( mdia, hdlr );
 940
 941         /* minf*/
 942         minf = box_new( "minf" );
 943
 944         /* add smhd|vmhd */
 945         if( p_stream->p_fmt->i_cat == AUDIO_ES )
 946         {
 947             bo_t *smhd;
 948
 949             smhd = box_full_new( "smhd", 0, 0 );
 950             bo_add_16be( smhd, 0 );     // balance
 951             bo_add_16be( smhd, 0 );     // reserved
 952             box_fix( smhd );
 953
 954             box_gather( minf, smhd );
 955         }
 956         else if( p_stream->p_fmt->i_cat == VIDEO_ES )
 957         {
 958             bo_t *vmhd;
 959
 960             vmhd = box_full_new( "vmhd", 0, 1 );
 961             bo_add_16be( vmhd, 0 );     // graphicsmode
 962             for( i = 0; i < 3; i++ )
 963             {
 964                 bo_add_16be( vmhd, 0 ); // opcolor
 965             }
 966             box_fix( vmhd );
 967
 968             box_gather( minf, vmhd );
 969         }
 970
 971         /* dinf */
 972         dinf = box_new( "dinf" );
 973         dref = box_full_new( "dref", 0, 0 );
 974         bo_add_32be( dref, 1 );
 975         url = box_full_new( "url ", 0, 0x01 );
 976         box_fix( url );
 977         box_gather( dref, url );
 978         box_fix( dref );
 979         box_gather( dinf, dref );
 980
 981         /* append dinf to mdia */
 982         box_fix( dinf );
 983         box_gather( minf, dinf );
 984
 985         /* add stbl */
 986         stbl = GetStblBox( p_mux, p_stream );
 987
 988         /* append stbl to minf */
 989         p_stream->i_stco_pos += minf->i_buffer;
 990         box_gather( minf, stbl );
 991
 992         /* append minf to mdia */
 993         box_fix( minf );
 994         p_stream->i_stco_pos += mdia->i_buffer;
 995         box_gather( mdia, minf );
 996
 997         /* append mdia to trak */
 998         box_fix( mdia );
 999         p_stream->i_stco_pos += trak->i_buffer;
1000         box_gather( trak, mdia );
1001
1002         /* append trak to moov */
1003         box_fix( trak );
1004         p_stream->i_stco_pos += moov->i_buffer;
1005         box_gather( moov, trak );
1006     }
1007
1008     /* Add user data tags */
1009     box_gather( moov, GetUdtaTag( p_mux ) );
1010
1011     box_fix( moov );
1012     return moov;
1013 }
1014
1015 /*****************************************************************************
1016  * Close:
1017  *****************************************************************************/
1018 static void Close( vlc_object_t * p_this )
1019 {
1020     sout_mux_t      *p_mux = (sout_mux_t*)p_this;
1021     sout_mux_sys_t  *p_sys = p_mux->p_sys;
1022     sout_buffer_t   *p_hdr;
1023     bo_t            bo, *moov;
1024     vlc_value_t     val;
1025
1026     int             i_trak;
1027     uint64_t        i_moov_pos;
1028
1029     uint32_t        i_movie_timescale = 90000;
1030     int64_t         i_movie_duration  = 0;
1031
1032     msg_Dbg( p_mux, "Close" );
1033
1034     /* Create general info */
1035     for( i_trak = 0; i_trak < p_sys->i_nb_streams; i_trak++ )
1036     {
1037         mp4_stream_t *p_stream = p_sys->pp_streams[i_trak];
1038         i_movie_duration = __MAX( i_movie_duration, p_stream->i_duration );
1039     }
1040     msg_Dbg( p_mux, "movie duration %ds",
1041              (uint32_t)( i_movie_duration / (mtime_t)1000000 ) );
1042
1043     i_movie_duration = i_movie_duration * i_movie_timescale / 1000000;
1044
1045     /* Update mdat size */
1046     bo_init( &bo, 0, NULL, VLC_TRUE );
1047     if( p_sys->i_pos - p_sys->i_mdat_pos >= (((uint64_t)1)<<32) )
1048     {
1049         /* Extended size */
1050         bo_add_32be  ( &bo, 1 );
1051         bo_add_fourcc( &bo, "mdat" );
1052         bo_add_64be  ( &bo, p_sys->i_pos - p_sys->i_mdat_pos );
1053     }
1054     else
1055     {
1056         bo_add_32be  ( &bo, 8 );
1057         bo_add_fourcc( &bo, "wide" );
1058         bo_add_32be  ( &bo, p_sys->i_pos - p_sys->i_mdat_pos - 8 );
1059         bo_add_fourcc( &bo, "mdat" );
1060     }
1061     p_hdr = bo_to_sout( p_mux->p_sout, &bo );
1062     free( bo.p_buffer );
1063
1064     sout_AccessOutSeek( p_mux->p_access, p_sys->i_mdat_pos );
1065     sout_AccessOutWrite( p_mux->p_access, p_hdr );
1066
1067     /* Create MOOV header */
1068     i_moov_pos = p_sys->i_pos;
1069     moov = GetMoovBox( p_mux );
1070
1071     /* Check we need to create "fast start" files */
1072     var_Create( p_this, "mp4-faststart", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
1073     var_Get( p_this, "mp4-faststart", &val );
1074     p_sys->b_fast_start = val.b_bool;
1075     while( p_sys->b_fast_start )
1076     {
1077         /* Move data to the end of the file so we can fit the moov header
1078          * at the start */
1079         sout_buffer_t *p_buf;
1080         int64_t i_chunk, i_size = p_sys->i_pos - p_sys->i_mdat_pos;
1081         int i_moov_size = moov->i_buffer;
1082
1083         while( i_size > 0 )
1084         {
1085             i_chunk = __MIN( 32768, i_size );
1086             p_buf = sout_BufferNew( p_mux->p_sout, i_chunk );
1087             sout_AccessOutSeek( p_mux->p_access,
1088                                 p_sys->i_mdat_pos + i_size - i_chunk );
1089             if( sout_AccessOutRead( p_mux->p_access, p_buf ) < i_chunk )
1090             {
1091                 msg_Warn( p_this, "read() not supported by acces output, "
1092                           "won't create a fast start file" );
1093                 p_sys->b_fast_start = VLC_FALSE;
1094                 break;
1095             }
1096             sout_AccessOutSeek( p_mux->p_access, p_sys->i_mdat_pos + i_size +
1097                                 i_moov_size - i_chunk );
1098             sout_AccessOutWrite( p_mux->p_access, p_buf );
1099             i_size -= i_chunk;
1100         }
1101
1102         if( !p_sys->b_fast_start ) break;
1103
1104         /* Fix-up samples to chunks table in MOOV header */
1105         for( i_trak = 0; i_trak < p_sys->i_nb_streams; i_trak++ )
1106         {
1107             mp4_stream_t *p_stream = p_sys->pp_streams[i_trak];
1108             unsigned int i;
1109             int i_chunk;
1110
1111             moov->i_buffer = p_stream->i_stco_pos;
1112             for( i_chunk = 0, i = 0; i < p_stream->i_entry_count; i_chunk++ )
1113             {
1114                 if( p_stream->b_stco64 )
1115                     bo_add_64be( moov, p_stream->entry[i].i_pos + i_moov_size);
1116                 else
1117                     bo_add_32be( moov, p_stream->entry[i].i_pos + i_moov_size);
1118
1119                 while( i < p_stream->i_entry_count )
1120                 {
1121                     if( i + 1 < p_stream->i_entry_count &&
1122                         p_stream->entry[i].i_pos + p_stream->entry[i].i_size
1123                         != p_stream->entry[i + 1].i_pos )
1124                     {
1125                         i++;
1126                         break;
1127                     }
1128
1129                     i++;
1130                 }
1131             }
1132         }
1133
1134         moov->i_buffer = i_moov_size;
1135         i_moov_pos = p_sys->i_mdat_pos;
1136         p_sys->b_fast_start = VLC_FALSE;
1137     }
1138
1139     /* Write MOOV header */
1140     sout_AccessOutSeek( p_mux->p_access, i_moov_pos );
1141     box_send( p_mux, moov );
1142
1143     /* Clean-up */
1144     for( i_trak = 0; i_trak < p_sys->i_nb_streams; i_trak++ )
1145     {
1146         mp4_stream_t *p_stream = p_sys->pp_streams[i_trak];
1147
1148         if( p_stream->p_fmt->p_extra )
1149         {
1150             free( p_stream->p_fmt->p_extra );
1151         }
1152         free( p_stream->p_fmt );
1153         free( p_stream->entry );
1154         free( p_stream );
1155     }
1156     free( p_sys );
1157 }
1158
1159 static int Capability( sout_mux_t *p_mux, int i_query, void *p_args,
1160                        void *p_answer )
1161 {
1162    switch( i_query )
1163    {
1164         case SOUT_MUX_CAP_GET_ADD_STREAM_ANY_TIME:
1165             *(vlc_bool_t*)p_answer = VLC_FALSE;
1166             return( SOUT_MUX_CAP_ERR_OK );
1167         default:
1168             return( SOUT_MUX_CAP_ERR_UNIMPLEMENTED );
1169    }
1170 }
1171
1172 static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input )
1173 {
1174     sout_mux_sys_t  *p_sys = p_mux->p_sys;
1175     mp4_stream_t    *p_stream;
1176
1177     switch( p_input->p_fmt->i_codec )
1178     {
1179         case VLC_FOURCC( 'm', 'p', '4', 'a' ):
1180         case VLC_FOURCC( 'm', 'p', '4', 'v' ):
1181         case VLC_FOURCC( 'm', 'p', 'g', 'a' ):
1182         case VLC_FOURCC( 'm', 'p', 'g', 'v' ):
1183         case VLC_FOURCC( 'M', 'J', 'P', 'G' ):
1184         case VLC_FOURCC( 'm', 'j', 'p', 'b' ):
1185         case VLC_FOURCC( 'S', 'V', 'Q', '3' ):
1186             break;
1187         default:
1188             msg_Err( p_mux, "unsupported codec %4.4s in mp4",
1189                      (char*)&p_input->p_fmt->i_codec );
1190             return VLC_EGENERIC;
1191     }
1192
1193     p_stream                = malloc( sizeof( mp4_stream_t ) );
1194     p_stream->p_fmt         = malloc( sizeof( es_format_t ) );
1195     memcpy( p_stream->p_fmt, p_input->p_fmt, sizeof( es_format_t ) );
1196     if( p_stream->p_fmt->i_extra )
1197     {
1198         p_stream->p_fmt->p_extra =
1199             malloc( p_stream->p_fmt->i_extra );
1200         memcpy( p_stream->p_fmt->p_extra,
1201                 p_input->p_fmt->p_extra,
1202                 p_input->p_fmt->i_extra );
1203     }
1204     p_stream->i_track_id    = p_sys->i_nb_streams + 1;
1205     p_stream->i_entry_count = 0;
1206     p_stream->i_entry_max   = 1000;
1207     p_stream->entry         =
1208         calloc( p_stream->i_entry_max, sizeof( mp4_entry_t ) );
1209     p_stream->i_duration    = 0;
1210
1211     p_input->p_sys          = p_stream;
1212
1213     msg_Dbg( p_mux, "adding input" );
1214
1215     TAB_APPEND( p_sys->i_nb_streams, p_sys->pp_streams, p_stream );
1216     return( VLC_SUCCESS );
1217 }
1218
1219 static int DelStream( sout_mux_t *p_mux, sout_input_t *p_input )
1220 {
1221     msg_Dbg( p_mux, "removing input" );
1222     return( 0 );
1223 }
1224
1225 /****************************************************************************/
1226
1227 static int MuxGetStream( sout_mux_t *p_mux, int *pi_stream, mtime_t *pi_dts )
1228 {
1229     mtime_t i_dts;
1230     int     i_stream;
1231     int     i;
1232
1233     for( i = 0, i_dts = 0, i_stream = -1; i < p_mux->i_nb_inputs; i++ )
1234     {
1235         sout_fifo_t  *p_fifo;
1236
1237         p_fifo = p_mux->pp_inputs[i]->p_fifo;
1238
1239         if( p_fifo->i_depth > 1 )
1240         {
1241             sout_buffer_t *p_buf;
1242
1243             p_buf = sout_FifoShow( p_fifo );
1244             if( i_stream < 0 || p_buf->i_dts < i_dts )
1245             {
1246                 i_dts = p_buf->i_dts;
1247                 i_stream = i;
1248             }
1249         }
1250         else
1251         {
1252             return( -1 ); // wait that all fifo have at least 2 packets
1253         }
1254     }
1255     if( pi_stream )
1256     {
1257         *pi_stream = i_stream;
1258     }
1259     if( pi_dts )
1260     {
1261         *pi_dts = i_dts;
1262     }
1263     return( i_stream );
1264 }
1265
1266 static int Mux( sout_mux_t *p_mux )
1267 {
1268     sout_mux_sys_t *p_sys = p_mux->p_sys;
1269
1270     for( ;; )
1271     {
1272         sout_input_t    *p_input;
1273         int             i_stream;
1274         mp4_stream_t    *p_stream;
1275         sout_buffer_t   *p_data;
1276         mtime_t         i_dts;
1277
1278         if( MuxGetStream( p_mux, &i_stream, &i_dts) < 0 )
1279         {
1280             return( VLC_SUCCESS );
1281         }
1282
1283         if( !p_sys->i_start_dts )
1284             p_sys->i_start_dts = i_dts;
1285
1286         p_input  = p_mux->pp_inputs[i_stream];
1287         p_stream = (mp4_stream_t*)p_input->p_sys;
1288
1289         p_data  = sout_FifoGet( p_input->p_fifo );
1290
1291         /* add index entry */
1292         p_stream->entry[p_stream->i_entry_count].i_pos   = p_sys->i_pos;
1293         p_stream->entry[p_stream->i_entry_count].i_size  = p_data->i_size;
1294         p_stream->entry[p_stream->i_entry_count].i_pts   = p_data->i_pts;
1295         p_stream->entry[p_stream->i_entry_count].i_dts   = p_data->i_dts;
1296         p_stream->entry[p_stream->i_entry_count].i_length=
1297             __MAX( p_data->i_length, 0 );
1298
1299         if( p_stream->i_entry_count == 0 )
1300         {
1301             /* Here is another bad hack.
1302              * To make sure audio/video are in sync, we report a corrected
1303              * length for the 1st sample. */
1304             p_stream->entry[p_stream->i_entry_count].i_length =
1305                 __MAX( p_data->i_length, 0 ) +
1306                 p_data->i_pts - p_sys->i_start_dts;
1307         }
1308
1309         p_stream->i_entry_count++;
1310         if( p_stream->i_entry_count >= p_stream->i_entry_max )
1311         {
1312             p_stream->i_entry_max += 1000;
1313             p_stream->entry =
1314                 realloc( p_stream->entry,
1315                          p_stream->i_entry_max * sizeof( mp4_entry_t ) );
1316         }
1317
1318         /* update */
1319         p_stream->i_duration += __MAX( p_data->i_length, 0 );
1320         p_sys->i_pos += p_data->i_size;
1321
1322         /* write data */
1323         sout_AccessOutWrite( p_mux->p_access, p_data );
1324     }
1325
1326     return( VLC_SUCCESS );
1327 }
1328
1329
1330 /****************************************************************************/
1331
1332 static void bo_init( bo_t *p_bo, int i_size, uint8_t *p_buffer,
1333                      vlc_bool_t b_grow )
1334 {
1335     if( !p_buffer )
1336     {
1337         p_bo->i_buffer_size = __MAX( i_size, 1024 );
1338         p_bo->p_buffer = malloc( p_bo->i_buffer_size );
1339     }
1340     else
1341     {
1342         p_bo->i_buffer_size = i_size;
1343         p_bo->p_buffer = p_buffer;
1344     }
1345
1346     p_bo->b_grow = b_grow;
1347     p_bo->i_buffer = 0;
1348 }
1349
1350 static void bo_add_8( bo_t *p_bo, uint8_t i )
1351 {
1352     if( p_bo->i_buffer < p_bo->i_buffer_size )
1353     {
1354         p_bo->p_buffer[p_bo->i_buffer] = i;
1355     }
1356     else if( p_bo->b_grow )
1357     {
1358         p_bo->i_buffer_size += 1024;
1359         p_bo->p_buffer = realloc( p_bo->p_buffer, p_bo->i_buffer_size );
1360
1361         p_bo->p_buffer[p_bo->i_buffer] = i;
1362     }
1363
1364     p_bo->i_buffer++;
1365 }
1366
1367 static void bo_add_16be( bo_t *p_bo, uint16_t i )
1368 {
1369     bo_add_8( p_bo, ( ( i >> 8) &0xff ) );
1370     bo_add_8( p_bo, i &0xff );
1371 }
1372
1373 static void bo_add_24be( bo_t *p_bo, uint32_t i )
1374 {
1375     bo_add_8( p_bo, ( ( i >> 16) &0xff ) );
1376     bo_add_8( p_bo, ( ( i >> 8) &0xff ) );
1377     bo_add_8( p_bo, (   i &0xff ) );
1378 }
1379 static void bo_add_32be( bo_t *p_bo, uint32_t i )
1380 {
1381     bo_add_16be( p_bo, ( ( i >> 16) &0xffff ) );
1382     bo_add_16be( p_bo, i &0xffff );
1383 }
1384
1385 static void bo_fix_32be ( bo_t *p_bo, int i_pos, uint32_t i)
1386 {
1387     p_bo->p_buffer[i_pos    ] = ( i >> 24 )&0xff;
1388     p_bo->p_buffer[i_pos + 1] = ( i >> 16 )&0xff;
1389     p_bo->p_buffer[i_pos + 2] = ( i >>  8 )&0xff;
1390     p_bo->p_buffer[i_pos + 3] = ( i       )&0xff;
1391 }
1392
1393 static void bo_add_64be( bo_t *p_bo, uint64_t i )
1394 {
1395     bo_add_32be( p_bo, ( ( i >> 32) &0xffffffff ) );
1396     bo_add_32be( p_bo, i &0xffffffff );
1397 }
1398
1399 static void bo_add_fourcc( bo_t *p_bo, char *fcc )
1400 {
1401     bo_add_8( p_bo, fcc[0] );
1402     bo_add_8( p_bo, fcc[1] );
1403     bo_add_8( p_bo, fcc[2] );
1404     bo_add_8( p_bo, fcc[3] );
1405 }
1406
1407 static void bo_add_mem( bo_t *p_bo, int i_size, uint8_t *p_mem )
1408 {
1409     int i;
1410
1411     for( i = 0; i < i_size; i++ )
1412     {
1413         bo_add_8( p_bo, p_mem[i] );
1414     }
1415 }
1416
1417 static void bo_add_descr( bo_t *p_bo, uint8_t tag, uint32_t i_size )
1418 {
1419     uint32_t i_length;
1420     uint8_t  vals[4];
1421
1422     i_length = i_size;
1423     vals[3] = (unsigned char)(i_length & 0x7f);
1424     i_length >>= 7;
1425     vals[2] = (unsigned char)((i_length & 0x7f) | 0x80);
1426     i_length >>= 7;
1427     vals[1] = (unsigned char)((i_length & 0x7f) | 0x80);
1428     i_length >>= 7;
1429     vals[0] = (unsigned char)((i_length & 0x7f) | 0x80);
1430
1431     bo_add_8( p_bo, tag );
1432
1433     if( i_size < 0x00000080 )
1434     {
1435         bo_add_8( p_bo, vals[3] );
1436     }
1437     else if( i_size < 0x00004000 )
1438     {
1439         bo_add_8( p_bo, vals[2] );
1440         bo_add_8( p_bo, vals[3] );
1441     }
1442     else if( i_size < 0x00200000 )
1443     {
1444         bo_add_8( p_bo, vals[1] );
1445         bo_add_8( p_bo, vals[2] );
1446         bo_add_8( p_bo, vals[3] );
1447     }
1448     else if( i_size < 0x10000000 )
1449     {
1450         bo_add_8( p_bo, vals[0] );
1451         bo_add_8( p_bo, vals[1] );
1452         bo_add_8( p_bo, vals[2] );
1453         bo_add_8( p_bo, vals[3] );
1454     }
1455 }
1456
1457 static void bo_add_bo( bo_t *p_bo, bo_t *p_bo2 )
1458 {
1459     int i;
1460
1461     for( i = 0; i < p_bo2->i_buffer; i++ )
1462     {
1463         bo_add_8( p_bo, p_bo2->p_buffer[i] );
1464     }
1465 }
1466
1467 static bo_t * box_new( char *fcc )
1468 {
1469     bo_t *box;
1470
1471     if( ( box = malloc( sizeof( bo_t ) ) ) )
1472     {
1473         bo_init( box, 0, NULL, VLC_TRUE );
1474
1475         bo_add_32be  ( box, 0 );
1476         bo_add_fourcc( box, fcc );
1477     }
1478
1479     return box;
1480 }
1481
1482 static bo_t * box_full_new( char *fcc, uint8_t v, uint32_t f )
1483 {
1484     bo_t *box;
1485
1486     if( ( box = malloc( sizeof( bo_t ) ) ) )
1487     {
1488         bo_init( box, 0, NULL, VLC_TRUE );
1489
1490         bo_add_32be  ( box, 0 );
1491         bo_add_fourcc( box, fcc );
1492         bo_add_8     ( box, v );
1493         bo_add_24be  ( box, f );
1494     }
1495
1496     return box;
1497 }
1498
1499 static void box_fix( bo_t *box )
1500 {
1501     bo_t box_tmp;
1502
1503     memcpy( &box_tmp, box, sizeof( bo_t ) );
1504
1505     box_tmp.i_buffer = 0;
1506     bo_add_32be( &box_tmp, box->i_buffer );
1507 }
1508
1509 static void box_free( bo_t *box )
1510 {
1511     if( box->p_buffer )
1512     {
1513         free( box->p_buffer );
1514     }
1515
1516     free( box );
1517 }
1518
1519 static void box_gather ( bo_t *box, bo_t *box2 )
1520 {
1521     bo_add_bo( box, box2 );
1522     box_free( box2 );
1523 }
1524
1525 static sout_buffer_t * bo_to_sout( sout_instance_t *p_sout,  bo_t *box )
1526 {
1527     sout_buffer_t *p_buf;
1528
1529     p_buf = sout_BufferNew( p_sout, box->i_buffer );
1530     if( box->i_buffer > 0 )
1531     {
1532         memcpy( p_buf->p_buffer, box->p_buffer, box->i_buffer );
1533     }
1534
1535     p_buf->i_size = box->i_buffer;
1536
1537     return p_buf;
1538 }
1539
1540 static void box_send( sout_mux_t *p_mux,  bo_t *box )
1541 {
1542     sout_buffer_t *p_buf;
1543
1544     p_buf = bo_to_sout( p_mux->p_sout, box );
1545     box_free( box );
1546
1547     sout_AccessOutWrite( p_mux->p_access, p_buf );
1548 }
1549
1550 static int64_t get_timestamp()
1551 {
1552     int64_t i_timestamp = 0;
1553
1554 #ifdef HAVE_TIME_H
1555     i_timestamp = time(NULL);
1556     i_timestamp += 2082844800; // MOV/MP4 start date is 1/1/1904
1557     // 208284480 is (((1970 - 1904) * 365) + 17) * 24 * 60 * 60
1558 #endif
1559
1560     return i_timestamp;
1561 }