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.12 2004/01/25 13:29:04 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     /* Next available track id */
 777     bo_add_32be( mvhd, p_sys->i_nb_streams + 1 ); // next-track-id
 778
 779     box_fix( mvhd );
 780     box_gather( moov, mvhd );
 781
 782     for( i_trak = 0; i_trak < p_sys->i_nb_streams; i_trak++ )
 783     {
 784         mp4_stream_t *p_stream;
 785         uint32_t     i_timescale;
 786
 787         bo_t *trak, *tkhd, *mdia, *mdhd, *hdlr;
 788         bo_t *minf, *dinf, *dref, *url, *stbl;
 789
 790         p_stream = p_sys->pp_streams[i_trak];
 791
 792         if( p_stream->p_fmt->i_cat != AUDIO_ES &&
 793             p_stream->p_fmt->i_cat != VIDEO_ES )
 794         {
 795             msg_Err( p_mux, "FIXME ignoring trak (noaudio&&novideo)" );
 796             continue;
 797         }
 798
 799         if( p_stream->p_fmt->i_cat == AUDIO_ES )
 800             i_timescale = p_stream->p_fmt->audio.i_rate;
 801         else
 802             i_timescale = 1001;
 803
 804         /* *** add /moov/trak *** */
 805         trak = box_new( "trak" );
 806
 807         /* *** add /moov/trak/tkhd *** */
 808         if( !p_sys->b_64_ext )
 809         {
 810             if( p_sys->b_mov )
 811                 tkhd = box_full_new( "tkhd", 0, 0x0f );
 812             else
 813                 tkhd = box_full_new( "tkhd", 0, 1 );
 814
 815             bo_add_32be( tkhd, get_timestamp() );       // creation time
 816             bo_add_32be( tkhd, get_timestamp() );       // modification time
 817             bo_add_32be( tkhd, p_stream->i_track_id );
 818             bo_add_32be( tkhd, 0 );                     // reserved 0
 819             bo_add_32be( tkhd, p_stream->i_duration *
 820                          (int64_t)i_movie_timescale /
 821                          (mtime_t)1000000 );            // duration
 822         }
 823         else
 824         {
 825             if( p_sys->b_mov )
 826                 tkhd = box_full_new( "tkhd", 1, 0x0f );
 827             else
 828                 tkhd = box_full_new( "tkhd", 1, 1 );
 829
 830             bo_add_64be( tkhd, get_timestamp() );       // creation time
 831             bo_add_64be( tkhd, get_timestamp() );       // modification time
 832             bo_add_32be( tkhd, p_stream->i_track_id );
 833             bo_add_32be( tkhd, 0 );                     // reserved 0
 834             bo_add_64be( tkhd, p_stream->i_duration *
 835                          (int64_t)i_movie_timescale /
 836                          (mtime_t)1000000 );            // duration
 837         }
 838
 839         for( i = 0; i < 2; i++ )
 840         {
 841             bo_add_32be( tkhd, 0 );                 // reserved
 842         }
 843         bo_add_16be( tkhd, 0 );                     // layer
 844         bo_add_16be( tkhd, 0 );                     // pre-defined
 845         // volume
 846         bo_add_16be( tkhd, p_stream->p_fmt->i_cat == AUDIO_ES ? 0x100 : 0 );
 847         bo_add_16be( tkhd, 0 );                     // reserved
 848         for( i = 0; i < 9; i++ )
 849         {
 850             bo_add_32be( tkhd, mvhd_matrix[i] );    // matrix
 851         }
 852         if( p_stream->p_fmt->i_cat == AUDIO_ES )
 853         {
 854             bo_add_32be( tkhd, 0 );                 // width (presentation)
 855             bo_add_32be( tkhd, 0 );                 // height(presentation)
 856         }
 857         else
 858         {
 859             // width (presentation)
 860             bo_add_32be( tkhd, p_stream->p_fmt->video.i_aspect *
 861                          p_stream->p_fmt->video.i_height /
 862                          VOUT_ASPECT_FACTOR << 16 );
 863             // height(presentation)
 864             bo_add_32be( tkhd, p_stream->p_fmt->video.i_height << 16 );
 865         }
 866         box_fix( tkhd );
 867         box_gather( trak, tkhd );
 868
 869         /* *** add /moov/trak/mdia *** */
 870         mdia = box_new( "mdia" );
 871
 872         /* media header */
 873         if( !p_sys->b_64_ext )
 874         {
 875             mdhd = box_full_new( "mdhd", 0, 0 );
 876             bo_add_32be( mdhd, get_timestamp() );   // creation time
 877             bo_add_32be( mdhd, get_timestamp() );   // modification time
 878             bo_add_32be( mdhd, i_timescale);        // timescale
 879             bo_add_32be( mdhd, p_stream->i_duration * (int64_t)i_timescale /
 880                                (mtime_t)1000000 );  // duration
 881         }
 882         else
 883         {
 884             mdhd = box_full_new( "mdhd", 1, 0 );
 885             bo_add_64be( mdhd, get_timestamp() );   // creation time
 886             bo_add_64be( mdhd, get_timestamp() );   // modification time
 887             bo_add_32be( mdhd, i_timescale);        // timescale
 888             bo_add_64be( mdhd, p_stream->i_duration * (int64_t)i_timescale /
 889                                (mtime_t)1000000 );  // duration
 890         }
 891
 892         bo_add_16be( mdhd, 0    );              // language   FIXME
 893         bo_add_16be( mdhd, 0    );              // predefined
 894         box_fix( mdhd );
 895         box_gather( mdia, mdhd );
 896
 897         /* handler reference */
 898         hdlr = box_full_new( "hdlr", 0, 0 );
 899
 900         bo_add_fourcc( hdlr, "mhlr" );         // media handler
 901         if( p_stream->p_fmt->i_cat == AUDIO_ES )
 902         {
 903             bo_add_fourcc( hdlr, "soun" );
 904         }
 905         else
 906         {
 907             bo_add_fourcc( hdlr, "vide" );
 908         }
 909
 910         bo_add_32be( hdlr, 0 );         // reserved
 911         bo_add_32be( hdlr, 0 );         // reserved
 912         bo_add_32be( hdlr, 0 );         // reserved
 913
 914         if( p_stream->p_fmt->i_cat == AUDIO_ES )
 915         {
 916             bo_add_8( hdlr, 13 );
 917             bo_add_mem( hdlr, 13, "SoundHandler" );
 918         }
 919         else
 920         {
 921             bo_add_8( hdlr, 13 );
 922             bo_add_mem( hdlr, 13, "VideoHandler" );
 923         }
 924
 925         box_fix( hdlr );
 926         box_gather( mdia, hdlr );
 927
 928         /* minf*/
 929         minf = box_new( "minf" );
 930
 931         /* add smhd|vmhd */
 932         if( p_stream->p_fmt->i_cat == AUDIO_ES )
 933         {
 934             bo_t *smhd;
 935
 936             smhd = box_full_new( "smhd", 0, 0 );
 937             bo_add_16be( smhd, 0 );     // balance
 938             bo_add_16be( smhd, 0 );     // reserved
 939             box_fix( smhd );
 940
 941             box_gather( minf, smhd );
 942         }
 943         else if( p_stream->p_fmt->i_cat == VIDEO_ES )
 944         {
 945             bo_t *vmhd;
 946
 947             vmhd = box_full_new( "vmhd", 0, 1 );
 948             bo_add_16be( vmhd, 0 );     // graphicsmode
 949             for( i = 0; i < 3; i++ )
 950             {
 951                 bo_add_16be( vmhd, 0 ); // opcolor
 952             }
 953             box_fix( vmhd );
 954
 955             box_gather( minf, vmhd );
 956         }
 957
 958         /* dinf */
 959         dinf = box_new( "dinf" );
 960         dref = box_full_new( "dref", 0, 0 );
 961         bo_add_32be( dref, 1 );
 962         url = box_full_new( "url ", 0, 0x01 );
 963         box_fix( url );
 964         box_gather( dref, url );
 965         box_fix( dref );
 966         box_gather( dinf, dref );
 967
 968         /* append dinf to mdia */
 969         box_fix( dinf );
 970         box_gather( minf, dinf );
 971
 972         /* add stbl */
 973         stbl = GetStblBox( p_mux, p_stream );
 974
 975         /* append stbl to minf */
 976         p_stream->i_stco_pos += minf->i_buffer;
 977         box_gather( minf, stbl );
 978
 979         /* append minf to mdia */
 980         box_fix( minf );
 981         p_stream->i_stco_pos += mdia->i_buffer;
 982         box_gather( mdia, minf );
 983
 984         /* append mdia to trak */
 985         box_fix( mdia );
 986         p_stream->i_stco_pos += trak->i_buffer;
 987         box_gather( trak, mdia );
 988
 989         /* append trak to moov */
 990         box_fix( trak );
 991         p_stream->i_stco_pos += moov->i_buffer;
 992         box_gather( moov, trak );
 993     }
 994
 995     /* Add user data tags */
 996     box_gather( moov, GetUdtaTag( p_mux ) );
 997
 998     box_fix( moov );
 999     return moov;
1000 }
1001
1002 /*****************************************************************************
1003  * Close:
1004  *****************************************************************************/
1005 static void Close( vlc_object_t * p_this )
1006 {
1007     sout_mux_t      *p_mux = (sout_mux_t*)p_this;
1008     sout_mux_sys_t  *p_sys = p_mux->p_sys;
1009     sout_buffer_t   *p_hdr;
1010     bo_t            bo, *moov;
1011     vlc_value_t     val;
1012
1013     int             i_trak;
1014     uint64_t        i_moov_pos;
1015
1016     uint32_t        i_movie_timescale = 90000;
1017     int64_t         i_movie_duration  = 0;
1018
1019     msg_Dbg( p_mux, "Close" );
1020
1021     /* Create general info */
1022     for( i_trak = 0; i_trak < p_sys->i_nb_streams; i_trak++ )
1023     {
1024         mp4_stream_t *p_stream = p_sys->pp_streams[i_trak];
1025         i_movie_duration = __MAX( i_movie_duration, p_stream->i_duration );
1026     }
1027     msg_Dbg( p_mux, "movie duration %ds",
1028              (uint32_t)( i_movie_duration / (mtime_t)1000000 ) );
1029
1030     i_movie_duration = i_movie_duration * i_movie_timescale / 1000000;
1031
1032     /* Update mdat size */
1033     bo_init( &bo, 0, NULL, VLC_TRUE );
1034     if( p_sys->i_pos - p_sys->i_mdat_pos >= (((uint64_t)1)<<32) )
1035     {
1036         /* Extended size */
1037         bo_add_32be  ( &bo, 1 );
1038         bo_add_fourcc( &bo, "mdat" );
1039         bo_add_64be  ( &bo, p_sys->i_pos - p_sys->i_mdat_pos );
1040     }
1041     else
1042     {
1043         bo_add_32be  ( &bo, 8 );
1044         bo_add_fourcc( &bo, "wide" );
1045         bo_add_32be  ( &bo, p_sys->i_pos - p_sys->i_mdat_pos - 8 );
1046         bo_add_fourcc( &bo, "mdat" );
1047     }
1048     p_hdr = bo_to_sout( p_mux->p_sout, &bo );
1049     free( bo.p_buffer );
1050
1051     sout_AccessOutSeek( p_mux->p_access, p_sys->i_mdat_pos );
1052     sout_AccessOutWrite( p_mux->p_access, p_hdr );
1053
1054     /* Create MOOV header */
1055     i_moov_pos = p_sys->i_pos;
1056     moov = GetMoovBox( p_mux );
1057
1058     /* Check we need to create "fast start" files */
1059     var_Create( p_this, "mp4-faststart", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
1060     var_Get( p_this, "mp4-faststart", &val );
1061     p_sys->b_fast_start = val.b_bool;
1062     while( p_sys->b_fast_start )
1063     {
1064         /* Move data to the end of the file so we can fit the moov header
1065          * at the start */
1066         sout_buffer_t *p_buf;
1067         int64_t i_chunk, i_size = p_sys->i_pos - p_sys->i_mdat_pos;
1068         int i_moov_size = moov->i_buffer;
1069
1070         while( i_size > 0 )
1071         {
1072             i_chunk = __MIN( 32768, i_size );
1073             p_buf = sout_BufferNew( p_mux->p_sout, i_chunk );
1074             sout_AccessOutSeek( p_mux->p_access,
1075                                 p_sys->i_mdat_pos + i_size - i_chunk );
1076             if( sout_AccessOutRead( p_mux->p_access, p_buf ) < i_chunk )
1077             {
1078                 msg_Warn( p_this, "read() not supported by acces output, "
1079                           "won't create a fast start file" );
1080                 p_sys->b_fast_start = VLC_FALSE;
1081                 break;
1082             }
1083             sout_AccessOutSeek( p_mux->p_access, p_sys->i_mdat_pos + i_size +
1084                                 i_moov_size - i_chunk );
1085             sout_AccessOutWrite( p_mux->p_access, p_buf );
1086             i_size -= i_chunk;
1087         }
1088
1089         if( !p_sys->b_fast_start ) break;
1090
1091         /* Fix-up samples to chunks table in MOOV header */
1092         for( i_trak = 0; i_trak < p_sys->i_nb_streams; i_trak++ )
1093         {
1094             mp4_stream_t *p_stream = p_sys->pp_streams[i_trak];
1095             unsigned int i;
1096             int i_chunk;
1097
1098             moov->i_buffer = p_stream->i_stco_pos;
1099             for( i_chunk = 0, i = 0; i < p_stream->i_entry_count; i_chunk++ )
1100             {
1101                 if( p_stream->b_stco64 )
1102                     bo_add_64be( moov, p_stream->entry[i].i_pos + i_moov_size);
1103                 else
1104                     bo_add_32be( moov, p_stream->entry[i].i_pos + i_moov_size);
1105
1106                 while( i < p_stream->i_entry_count )
1107                 {
1108                     if( i + 1 < p_stream->i_entry_count &&
1109                         p_stream->entry[i].i_pos + p_stream->entry[i].i_size
1110                         != p_stream->entry[i + 1].i_pos )
1111                     {
1112                         i++;
1113                         break;
1114                     }
1115
1116                     i++;
1117                 }
1118             }
1119         }
1120
1121         moov->i_buffer = i_moov_size;
1122         i_moov_pos = p_sys->i_mdat_pos;
1123         p_sys->b_fast_start = VLC_FALSE;
1124     }
1125
1126     /* Write MOOV header */
1127     sout_AccessOutSeek( p_mux->p_access, i_moov_pos );
1128     box_send( p_mux, moov );
1129
1130     /* Clean-up */
1131     for( i_trak = 0; i_trak < p_sys->i_nb_streams; i_trak++ )
1132     {
1133         mp4_stream_t *p_stream = p_sys->pp_streams[i_trak];
1134
1135         if( p_stream->p_fmt->p_extra )
1136         {
1137             free( p_stream->p_fmt->p_extra );
1138         }
1139         free( p_stream->p_fmt );
1140         free( p_stream->entry );
1141         free( p_stream );
1142     }
1143     free( p_sys );
1144 }
1145
1146 static int Capability( sout_mux_t *p_mux, int i_query, void *p_args,
1147                        void *p_answer )
1148 {
1149    switch( i_query )
1150    {
1151         case SOUT_MUX_CAP_GET_ADD_STREAM_ANY_TIME:
1152             *(vlc_bool_t*)p_answer = VLC_FALSE;
1153             return( SOUT_MUX_CAP_ERR_OK );
1154         default:
1155             return( SOUT_MUX_CAP_ERR_UNIMPLEMENTED );
1156    }
1157 }
1158
1159 static int AddStream( sout_mux_t *p_mux, sout_input_t *p_input )
1160 {
1161     sout_mux_sys_t  *p_sys = p_mux->p_sys;
1162     mp4_stream_t    *p_stream;
1163
1164     switch( p_input->p_fmt->i_codec )
1165     {
1166         case VLC_FOURCC( 'm', 'p', '4', 'a' ):
1167         case VLC_FOURCC( 'm', 'p', '4', 'v' ):
1168         case VLC_FOURCC( 'm', 'p', 'g', 'a' ):
1169         case VLC_FOURCC( 'm', 'p', 'g', 'v' ):
1170         case VLC_FOURCC( 'M', 'J', 'P', 'G' ):
1171         case VLC_FOURCC( 'm', 'j', 'p', 'b' ):
1172         case VLC_FOURCC( 'S', 'V', 'Q', '3' ):
1173             break;
1174         default:
1175             msg_Err( p_mux, "unsupported codec %4.4s in mp4",
1176                      (char*)&p_input->p_fmt->i_codec );
1177             return VLC_EGENERIC;
1178     }
1179
1180     p_stream                = malloc( sizeof( mp4_stream_t ) );
1181     p_stream->p_fmt         = malloc( sizeof( es_format_t ) );
1182     memcpy( p_stream->p_fmt, p_input->p_fmt, sizeof( es_format_t ) );
1183     if( p_stream->p_fmt->i_extra )
1184     {
1185         p_stream->p_fmt->p_extra =
1186             malloc( p_stream->p_fmt->i_extra );
1187         memcpy( p_stream->p_fmt->p_extra,
1188                 p_input->p_fmt->p_extra,
1189                 p_input->p_fmt->i_extra );
1190     }
1191     p_stream->i_track_id    = p_sys->i_nb_streams + 1;
1192     p_stream->i_entry_count = 0;
1193     p_stream->i_entry_max   = 1000;
1194     p_stream->entry         =
1195         calloc( p_stream->i_entry_max, sizeof( mp4_entry_t ) );
1196     p_stream->i_duration    = 0;
1197
1198     p_input->p_sys          = p_stream;
1199
1200     msg_Dbg( p_mux, "adding input" );
1201
1202     TAB_APPEND( p_sys->i_nb_streams, p_sys->pp_streams, p_stream );
1203     return( VLC_SUCCESS );
1204 }
1205
1206 static int DelStream( sout_mux_t *p_mux, sout_input_t *p_input )
1207 {
1208     msg_Dbg( p_mux, "removing input" );
1209     return( 0 );
1210 }
1211
1212 /****************************************************************************/
1213
1214 static int MuxGetStream( sout_mux_t *p_mux, int *pi_stream, mtime_t *pi_dts )
1215 {
1216     mtime_t i_dts;
1217     int     i_stream;
1218     int     i;
1219
1220     for( i = 0, i_dts = 0, i_stream = -1; i < p_mux->i_nb_inputs; i++ )
1221     {
1222         sout_fifo_t  *p_fifo;
1223
1224         p_fifo = p_mux->pp_inputs[i]->p_fifo;
1225
1226         if( p_fifo->i_depth > 1 )
1227         {
1228             sout_buffer_t *p_buf;
1229
1230             p_buf = sout_FifoShow( p_fifo );
1231             if( i_stream < 0 || p_buf->i_dts < i_dts )
1232             {
1233                 i_dts = p_buf->i_dts;
1234                 i_stream = i;
1235             }
1236         }
1237         else
1238         {
1239             return( -1 ); // wait that all fifo have at least 2 packets
1240         }
1241     }
1242     if( pi_stream )
1243     {
1244         *pi_stream = i_stream;
1245     }
1246     if( pi_dts )
1247     {
1248         *pi_dts = i_dts;
1249     }
1250     return( i_stream );
1251 }
1252
1253 static int Mux( sout_mux_t *p_mux )
1254 {
1255     sout_mux_sys_t *p_sys = p_mux->p_sys;
1256
1257     for( ;; )
1258     {
1259         sout_input_t    *p_input;
1260         int             i_stream;
1261         mp4_stream_t    *p_stream;
1262         sout_buffer_t   *p_data;
1263         mtime_t         i_dts;
1264
1265         if( MuxGetStream( p_mux, &i_stream, &i_dts) < 0 )
1266         {
1267             return( VLC_SUCCESS );
1268         }
1269
1270         if( !p_sys->i_start_dts )
1271             p_sys->i_start_dts = i_dts;
1272
1273         p_input  = p_mux->pp_inputs[i_stream];
1274         p_stream = (mp4_stream_t*)p_input->p_sys;
1275
1276         p_data  = sout_FifoGet( p_input->p_fifo );
1277
1278         /* add index entry */
1279         p_stream->entry[p_stream->i_entry_count].i_pos   = p_sys->i_pos;
1280         p_stream->entry[p_stream->i_entry_count].i_size  = p_data->i_size;
1281         p_stream->entry[p_stream->i_entry_count].i_pts   = p_data->i_pts;
1282         p_stream->entry[p_stream->i_entry_count].i_dts   = p_data->i_dts;
1283         p_stream->entry[p_stream->i_entry_count].i_length=
1284             __MAX( p_data->i_length, 0 );
1285
1286         if( p_stream->i_entry_count == 0 )
1287         {
1288             /* Here is another bad hack.
1289              * To make sure audio/video are in sync, we report a corrected
1290              * length for the 1st sample. */
1291             p_stream->entry[p_stream->i_entry_count].i_length =
1292                 __MAX( p_data->i_length, 0 ) +
1293                 p_data->i_pts - p_sys->i_start_dts;
1294         }
1295
1296         p_stream->i_entry_count++;
1297         if( p_stream->i_entry_count >= p_stream->i_entry_max )
1298         {
1299             p_stream->i_entry_max += 1000;
1300             p_stream->entry =
1301                 realloc( p_stream->entry,
1302                          p_stream->i_entry_max * sizeof( mp4_entry_t ) );
1303         }
1304
1305         /* update */
1306         p_stream->i_duration += __MAX( p_data->i_length, 0 );
1307         p_sys->i_pos += p_data->i_size;
1308
1309         /* write data */
1310         sout_AccessOutWrite( p_mux->p_access, p_data );
1311     }
1312
1313     return( VLC_SUCCESS );
1314 }
1315
1316
1317 /****************************************************************************/
1318
1319 static void bo_init( bo_t *p_bo, int i_size, uint8_t *p_buffer,
1320                      vlc_bool_t b_grow )
1321 {
1322     if( !p_buffer )
1323     {
1324         p_bo->i_buffer_size = __MAX( i_size, 1024 );
1325         p_bo->p_buffer = malloc( p_bo->i_buffer_size );
1326     }
1327     else
1328     {
1329         p_bo->i_buffer_size = i_size;
1330         p_bo->p_buffer = p_buffer;
1331     }
1332
1333     p_bo->b_grow = b_grow;
1334     p_bo->i_buffer = 0;
1335 }
1336
1337 static void bo_add_8( bo_t *p_bo, uint8_t i )
1338 {
1339     if( p_bo->i_buffer < p_bo->i_buffer_size )
1340     {
1341         p_bo->p_buffer[p_bo->i_buffer] = i;
1342     }
1343     else if( p_bo->b_grow )
1344     {
1345         p_bo->i_buffer_size += 1024;
1346         p_bo->p_buffer = realloc( p_bo->p_buffer, p_bo->i_buffer_size );
1347
1348         p_bo->p_buffer[p_bo->i_buffer] = i;
1349     }
1350
1351     p_bo->i_buffer++;
1352 }
1353
1354 static void bo_add_16be( bo_t *p_bo, uint16_t i )
1355 {
1356     bo_add_8( p_bo, ( ( i >> 8) &0xff ) );
1357     bo_add_8( p_bo, i &0xff );
1358 }
1359
1360 static void bo_add_24be( bo_t *p_bo, uint32_t i )
1361 {
1362     bo_add_8( p_bo, ( ( i >> 16) &0xff ) );
1363     bo_add_8( p_bo, ( ( i >> 8) &0xff ) );
1364     bo_add_8( p_bo, (   i &0xff ) );
1365 }
1366 static void bo_add_32be( bo_t *p_bo, uint32_t i )
1367 {
1368     bo_add_16be( p_bo, ( ( i >> 16) &0xffff ) );
1369     bo_add_16be( p_bo, i &0xffff );
1370 }
1371
1372 static void bo_fix_32be ( bo_t *p_bo, int i_pos, uint32_t i)
1373 {
1374     p_bo->p_buffer[i_pos    ] = ( i >> 24 )&0xff;
1375     p_bo->p_buffer[i_pos + 1] = ( i >> 16 )&0xff;
1376     p_bo->p_buffer[i_pos + 2] = ( i >>  8 )&0xff;
1377     p_bo->p_buffer[i_pos + 3] = ( i       )&0xff;
1378 }
1379
1380 static void bo_add_64be( bo_t *p_bo, uint64_t i )
1381 {
1382     bo_add_32be( p_bo, ( ( i >> 32) &0xffffffff ) );
1383     bo_add_32be( p_bo, i &0xffffffff );
1384 }
1385
1386 static void bo_add_fourcc( bo_t *p_bo, char *fcc )
1387 {
1388     bo_add_8( p_bo, fcc[0] );
1389     bo_add_8( p_bo, fcc[1] );
1390     bo_add_8( p_bo, fcc[2] );
1391     bo_add_8( p_bo, fcc[3] );
1392 }
1393
1394 static void bo_add_mem( bo_t *p_bo, int i_size, uint8_t *p_mem )
1395 {
1396     int i;
1397
1398     for( i = 0; i < i_size; i++ )
1399     {
1400         bo_add_8( p_bo, p_mem[i] );
1401     }
1402 }
1403
1404 static void bo_add_descr( bo_t *p_bo, uint8_t tag, uint32_t i_size )
1405 {
1406     uint32_t i_length;
1407     uint8_t  vals[4];
1408
1409     i_length = i_size;
1410     vals[3] = (unsigned char)(i_length & 0x7f);
1411     i_length >>= 7;
1412     vals[2] = (unsigned char)((i_length & 0x7f) | 0x80);
1413     i_length >>= 7;
1414     vals[1] = (unsigned char)((i_length & 0x7f) | 0x80);
1415     i_length >>= 7;
1416     vals[0] = (unsigned char)((i_length & 0x7f) | 0x80);
1417
1418     bo_add_8( p_bo, tag );
1419
1420     if( i_size < 0x00000080 )
1421     {
1422         bo_add_8( p_bo, vals[3] );
1423     }
1424     else if( i_size < 0x00004000 )
1425     {
1426         bo_add_8( p_bo, vals[2] );
1427         bo_add_8( p_bo, vals[3] );
1428     }
1429     else if( i_size < 0x00200000 )
1430     {
1431         bo_add_8( p_bo, vals[1] );
1432         bo_add_8( p_bo, vals[2] );
1433         bo_add_8( p_bo, vals[3] );
1434     }
1435     else if( i_size < 0x10000000 )
1436     {
1437         bo_add_8( p_bo, vals[0] );
1438         bo_add_8( p_bo, vals[1] );
1439         bo_add_8( p_bo, vals[2] );
1440         bo_add_8( p_bo, vals[3] );
1441     }
1442 }
1443
1444 static void bo_add_bo( bo_t *p_bo, bo_t *p_bo2 )
1445 {
1446     int i;
1447
1448     for( i = 0; i < p_bo2->i_buffer; i++ )
1449     {
1450         bo_add_8( p_bo, p_bo2->p_buffer[i] );
1451     }
1452 }
1453
1454 static bo_t * box_new( char *fcc )
1455 {
1456     bo_t *box;
1457
1458     if( ( box = malloc( sizeof( bo_t ) ) ) )
1459     {
1460         bo_init( box, 0, NULL, VLC_TRUE );
1461
1462         bo_add_32be  ( box, 0 );
1463         bo_add_fourcc( box, fcc );
1464     }
1465
1466     return box;
1467 }
1468
1469 static bo_t * box_full_new( char *fcc, uint8_t v, uint32_t f )
1470 {
1471     bo_t *box;
1472
1473     if( ( box = malloc( sizeof( bo_t ) ) ) )
1474     {
1475         bo_init( box, 0, NULL, VLC_TRUE );
1476
1477         bo_add_32be  ( box, 0 );
1478         bo_add_fourcc( box, fcc );
1479         bo_add_8     ( box, v );
1480         bo_add_24be  ( box, f );
1481     }
1482
1483     return box;
1484 }
1485
1486 static void box_fix( bo_t *box )
1487 {
1488     bo_t box_tmp;
1489
1490     memcpy( &box_tmp, box, sizeof( bo_t ) );
1491
1492     box_tmp.i_buffer = 0;
1493     bo_add_32be( &box_tmp, box->i_buffer );
1494 }
1495
1496 static void box_free( bo_t *box )
1497 {
1498     if( box->p_buffer )
1499     {
1500         free( box->p_buffer );
1501     }
1502
1503     free( box );
1504 }
1505
1506 static void box_gather ( bo_t *box, bo_t *box2 )
1507 {
1508     bo_add_bo( box, box2 );
1509     box_free( box2 );
1510 }
1511
1512 static sout_buffer_t * bo_to_sout( sout_instance_t *p_sout,  bo_t *box )
1513 {
1514     sout_buffer_t *p_buf;
1515
1516     p_buf = sout_BufferNew( p_sout, box->i_buffer );
1517     if( box->i_buffer > 0 )
1518     {
1519         memcpy( p_buf->p_buffer, box->p_buffer, box->i_buffer );
1520     }
1521
1522     p_buf->i_size = box->i_buffer;
1523
1524     return p_buf;
1525 }
1526
1527 static void box_send( sout_mux_t *p_mux,  bo_t *box )
1528 {
1529     sout_buffer_t *p_buf;
1530
1531     p_buf = bo_to_sout( p_mux->p_sout, box );
1532     box_free( box );
1533
1534     sout_AccessOutWrite( p_mux->p_access, p_buf );
1535 }
1536
1537 static int64_t get_timestamp()
1538 {
1539     int64_t i_timestamp = 0;
1540
1541 #ifdef HAVE_TIME_H
1542     i_timestamp = time(NULL);
1543     i_timestamp += 2082844800; // MOV/MP4 start date is 1/1/1904
1544     // 208284480 is (((1970 - 1904) * 365) + 17) * 24 * 60 * 60
1545 #endif
1546
1547     return i_timestamp;
1548 }