/***************************************************************************** * ts.c: MPEG-II TS Muxer ***************************************************************************** * Copyright (C) 2001, 2002 VideoLAN * $Id: ts.c,v 1.45 2004/02/22 15:57:41 fenrir Exp $ * * Authors: Laurent Aimar <fenrir@via.ecp.fr> *          Eric Petit <titer@videolan.org> * * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA. *****************************************************************************//***************************************************************************** * Preamble *****************************************************************************/#include <stdlib.h>#include <vlc/vlc.h>#include <vlc/input.h>#include <vlc/sout.h>#include "iso_lang.h"#include "bits.h"#include "pes.h"#include "csa.h"#if defined MODULE_NAME_IS_mux_ts_dvbpsi#   ifdef HAVE_DVBPSI_DR_H#       include <dvbpsi/dvbpsi.h>#       include <dvbpsi/descriptor.h>#       include <dvbpsi/pat.h>#       include <dvbpsi/pmt.h>#       include <dvbpsi/dr.h>#       include <dvbpsi/psi.h>#   else#       include "dvbpsi.h"#       include "descriptor.h"#       include "tables/pat.h"#       include "tables/pmt.h"#       include "descriptors/dr.h"#       include "psi.h"#   endif#endif/* * TODO: *  - check PCR frequency requirement *  - check PAT/PMT  "        " *  - check PCR/PCR "soft" *  - check if "registration" descriptor : "AC-3" should be a program *    descriptor or an es one. (xine want an es one) * *  - remove creation of PAT/PMT without dvbpsi *  - ? * FIXME: *  - subtitle support is far from perfect. I expect some subtitles drop *    if they arrive a bit late *    (We cannot rely on the fact that the fifo should be full) *//***************************************************************************** * Module descriptor *****************************************************************************/static int     Open   ( vlc_object_t * );static void    Close  ( vlc_object_t * );vlc_module_begin();#if defined MODULE_NAME_IS_mux_ts    set_description( _("TS muxer") );    set_capability( "sout mux", 100 );    add_shortcut( "ts" );    add_shortcut( "ts_nodvbpsi" );#elif defined MODULE_NAME_IS_mux_ts_dvbpsi    set_description( _("TS muxer (libdvbpsi)") );    set_capability( "sout mux", 120 );    add_shortcut( "ts" );    add_shortcut( "ts_dvbpsi" );#endif    set_callbacks( Open, Close );vlc_module_end();/***************************************************************************** * Exported prototypes *****************************************************************************/static int     Capability(sout_mux_t *, int, void *, void * );static int     AddStream( sout_mux_t *, sout_input_t * );static int     DelStream( sout_mux_t *, sout_input_t * );static int     Mux      ( sout_mux_t * );/***************************************************************************** * Local prototypes *****************************************************************************/#define SOUT_BUFFER_FLAGS_PRIVATE_PCR  ( 1 << SOUT_BUFFER_FLAGS_PRIVATE_SHIFT )#define SOUT_BUFFER_FLAGS_PRIVATE_CSA  ( 2 << SOUT_BUFFER_FLAGS_PRIVATE_SHIFT )typedef struct{    int           i_depth;    sout_buffer_t *p_first;    sout_buffer_t **pp_last;} sout_buffer_chain_t;static inline void BufferChainInit  ( sout_buffer_chain_t *c ){    c->i_depth = 0;    c->p_first = NULL;    c->pp_last = &c->p_first;}static inline void BufferChainAppend( sout_buffer_chain_t *c, sout_buffer_t *b ){    *c->pp_last = b;    c->i_depth++;    while( b->p_next )    {        b = b->p_next;        c->i_depth++;    }    c->pp_last = &b->p_next;}static inline sout_buffer_t *BufferChainGet( sout_buffer_chain_t *c ){    sout_buffer_t *b = c->p_first;    if( b )    {        c->i_depth--;        c->p_first = b->p_next;        if( c->p_first == NULL )        {            c->pp_last = &c->p_first;        }        b->p_next = NULL;    }    return b;}static inline void BufferChainClean( sout_instance_t *p_sout, sout_buffer_chain_t *c ){    sout_buffer_t *b;    while( ( b = BufferChainGet( c ) ) )    {        sout_BufferDelete( p_sout, b );    }    BufferChainInit( c );}typedef struct ts_stream_s{    int             i_pid;    int             i_stream_type;    int             i_stream_id;    int             i_continuity_counter;    /* to be used for carriege of DIV3 */    vlc_fourcc_t    i_bih_codec;    int             i_bih_width, i_bih_height;    /* Specific to mpeg4 in mpeg2ts */    int             i_es_id;    int             i_decoder_specific_info;    uint8_t         *p_decoder_specific_info;    /* language is iso639-2T */    uint8_t         lang[3];    sout_buffer_chain_t chain_pes;    mtime_t             i_pes_dts;    mtime_t             i_pes_length;    int                 i_pes_used;} ts_stream_t;struct sout_mux_sys_t{    int             i_pcr_pid;    sout_input_t    *p_pcr_input;    int             i_audio_bound;    int             i_video_bound;    int             i_pid_video;    int             i_pid_audio;    int             i_pid_free; // first usable pid    int             i_pat_version_number;    ts_stream_t     pat;    int             i_pmt_version_number;    ts_stream_t     pmt;        // Up to now only one program    int             i_mpeg4_streams;    int             i_null_continuity_counter;  /* Needed ? */    /* for TS building */    int64_t             i_bitrate_min;    int64_t             i_bitrate_max;    int64_t             i_caching_delay;    int64_t             i_pcr_delay;    int64_t             i_dts_delay;    mtime_t             i_pcr;  /* last PCR emited */    csa_t               *csa;};/* Reserve a pid and return it */static int  AllocatePID( sout_mux_sys_t *p_sys, int i_cat ){    int i_pid;    if ( i_cat == VIDEO_ES && p_sys->i_pid_video )    {        i_pid = p_sys->i_pid_video;        p_sys->i_pid_video = 0;    }    else if ( i_cat == AUDIO_ES && p_sys->i_pid_audio )    {        i_pid = p_sys->i_pid_audio;        p_sys->i_pid_audio = 0;    }    else    {        i_pid = ++p_sys->i_pid_free;    }    return i_pid;}static void GetPAT( sout_mux_t *p_mux, sout_buffer_chain_t *c );static void GetPMT( sout_mux_t *p_mux, sout_buffer_chain_t *c );static sout_buffer_t *TSNew( sout_mux_t *p_mux, ts_stream_t *p_stream, vlc_bool_t b_pcr );static void TSSetPCR( sout_buffer_t *p_ts, mtime_t i_dts );static void PEStoTS  ( sout_instance_t *, sout_buffer_chain_t *, sout_buffer_t *, ts_stream_t * );/***************************************************************************** * Open: *****************************************************************************/static int Open( vlc_object_t *p_this ){    sout_mux_t          *p_mux =(sout_mux_t*)p_this;    sout_mux_sys_t      *p_sys;    char                *val;    msg_Dbg( p_mux, "Open" );    p_sys = malloc( sizeof( sout_mux_sys_t ) );    p_mux->pf_capacity  = Capability;    p_mux->pf_addstream = AddStream;    p_mux->pf_delstream = DelStream;    p_mux->pf_mux       = Mux;    p_mux->p_sys        = p_sys;    p_mux->i_preheader  = 30; // really enough for a pes header    srand( (uint32_t)mdate() );    p_sys->i_audio_bound = 0;    p_sys->i_video_bound = 0;    p_sys->i_pat_version_number = rand() % 32;    p_sys->pat.i_pid = 0;    p_sys->pat.i_continuity_counter = 0;    p_sys->i_pmt_version_number = rand() % 32;    p_sys->pmt.i_pid = 0x42;    p_sys->pmt.i_continuity_counter = 0;    p_sys->i_pid_free = 0x43;    p_sys->i_pid_video = 0;    if( ( val = sout_cfg_find_value( p_mux->p_cfg, "pid-video" ) ) )    {        p_sys->i_pid_video = strtol( val, NULL, 0 );        if ( p_sys->i_pid_video > p_sys->i_pid_free )        {            p_sys->i_pid_free = p_sys->i_pid_video + 1;        }    }    p_sys->i_pid_audio = 0;    if( ( val = sout_cfg_find_value( p_mux->p_cfg, "pid-audio" ) ) )    {        p_sys->i_pid_audio = strtol( val, NULL, 0 );        if ( p_sys->i_pid_audio > p_sys->i_pid_free )        {            p_sys->i_pid_free = p_sys->i_pid_audio + 1;        }    }    p_sys->i_pcr_pid = 0x1fff;    p_sys->p_pcr_input = NULL;    p_sys->i_mpeg4_streams = 0;    p_sys->i_null_continuity_counter = 0;    /* Allow to create constrained stream */    p_sys->i_bitrate_min = 0;    p_sys->i_bitrate_max = 0;    if( ( val = sout_cfg_find_value( p_mux->p_cfg, "bmin" ) ) )    {        p_sys->i_bitrate_min = atoll( val );    }    if( ( val = sout_cfg_find_value( p_mux->p_cfg, "bmax" ) ) )    {        p_sys->i_bitrate_max = atoll( val );    }    if( p_sys->i_bitrate_min > 0 && p_sys->i_bitrate_max > 0 &&        p_sys->i_bitrate_min > p_sys->i_bitrate_max )    {        msg_Err( p_mux, "incompatible minimum and maximum bitrate, "                 "disabling bitrate control" );        p_sys->i_bitrate_min = 0;        p_sys->i_bitrate_max = 0;    }    if( p_sys->i_bitrate_min > 0 || p_sys->i_bitrate_max > 0 )    {        msg_Err( p_mux, "bmin and bmax no more supported (if you need them report it)" );    }    p_sys->i_caching_delay = 200000;    if( ( val = sout_cfg_find_value( p_mux->p_cfg, "caching" ) ) )    {        p_sys->i_caching_delay = (int64_t)atoi( val ) * 1000;        if( p_sys->i_caching_delay <= 0 )        {            msg_Err( p_mux,                     "invalid caching ("I64Fd"ms) reseting to 200ms",                     p_sys->i_caching_delay / 1000 );            p_sys->i_caching_delay = 200000;        }    }    p_sys->i_pcr_delay = 30000;    if( ( val = sout_cfg_find_value( p_mux->p_cfg, "pcr" ) ) )    {        p_sys->i_pcr_delay = (int64_t)atoi( val ) * 1000;        if( p_sys->i_pcr_delay <= 0 ||            p_sys->i_pcr_delay >= p_sys->i_caching_delay )        {            msg_Err( p_mux,                     "invalid pcr delay ("I64Fd"ms) reseting to 30ms",                     p_sys->i_pcr_delay / 1000 );            p_sys->i_pcr_delay = 30000;        }    }    msg_Dbg( p_mux, "caching="I64Fd" pcr="I64Fd,             p_sys->i_caching_delay, p_sys->i_pcr_delay );    p_sys->i_dts_delay = 200000;    if( ( val = sout_cfg_find_value( p_mux->p_cfg, "dts-delay" ) ) )    {        p_sys->i_dts_delay = (int64_t)atoi( val ) * 1000;    }    /* for TS generation */    p_sys->i_pcr    = 0;    p_sys->csa      = NULL;    if( ( val = sout_cfg_find_value( p_mux->p_cfg, "csa-ck" ) ) )    {        /* skip 0x */        if( val[0] == '0' && ( val[1] == 'x' || val[1] == 'X' ) )        {            val += 2;        }        if( strlen( val ) != 16 )        {            msg_Dbg( p_mux, "invalid csa ck (it must be 16 chars long)" );        }        else        {            /* Avoid using strtoll */            uint64_t i_ck;            uint8_t  ck[8];            int      i;            ck[0] = val[8];            val[8] = 0;            i_ck = ((int64_t)strtol( val, NULL, 16 )) << 32;            val[8] = ck[0];            i_ck += strtol( &val[8], NULL, 16 );            for( i = 0; i < 8; i++ )            {                ck[i] = ( i_ck >> ( 56 - 8*i) )&0xff;            }            msg_Dbg( p_mux, "using CSA scrambling with ck=%x:%x:%x:%x:%x:%x:%x:%x",                     ck[0], ck[1], ck[2], ck[3], ck[4], ck[5], ck[6], ck[7] );            p_sys->csa = csa_New();            csa_SetCW( p_sys->csa, ck, ck );        }    }    return VLC_SUCCESS;}/***************************************************************************** * Close: *****************************************************************************/static void Close( vlc_object_t * p_this ){    sout_mux_t          *p_mux = (sout_mux_t*)p_this;    sout_mux_sys_t      *p_sys = p_mux->p_sys;    msg_Dbg( p_mux, "Close" );    if( p_sys->csa )    {