/***************************************************************************** * ts.c: MPEG-II TS Muxer ***************************************************************************** * Copyright (C) 2001-2005 VideoLAN (Centrale Réseaux) and its contributors * $Id: ts.c 19719 2007-04-06 18:03:48Z massiot $ * * Authors: Laurent Aimar <fenrir@via.ecp.fr> *          Eric Petit <titer@videolan.org> *          Jean-Paul Saman <jpsaman #_at_# m2x.nl> *          Wallace Wadge <wwadge #_at_# gmail.com> * * 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., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, 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"#ifdef HAVE_DVBPSI_DR_H#   include <dvbpsi/dvbpsi.h>#   include <dvbpsi/demux.h>#   include <dvbpsi/descriptor.h>#   include <dvbpsi/pat.h>#   include <dvbpsi/pmt.h>#   include <dvbpsi/sdt.h>#   include <dvbpsi/dr.h>#   include <dvbpsi/psi.h>#else#   include "dvbpsi.h"#   include "demux.h"#   include "descriptor.h"#   include "tables/pat.h"#   include "tables/pmt.h"#   include "tables/sdt.h"#   include "descriptors/dr.h"#   include "psi.h"#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 * );#define VPID_TEXT N_("Video PID")#define VPID_LONGTEXT N_("Assign a fixed PID to the video stream. The PCR " \  "PID will automatically be the video.")#define APID_TEXT N_("Audio PID")#define APID_LONGTEXT N_("Assign a fixed PID to the audio stream.")#define SPUPID_TEXT N_("SPU PID")#define SPUPID_LONGTEXT N_("Assign a fixed PID to the SPU.")#define PMTPID_TEXT N_("PMT PID")#define PMTPID_LONGTEXT N_("Assign a fixed PID to the PMT")#define TSID_TEXT N_("TS ID")#define TSID_LONGTEXT N_("Assign a fixed Transport Stream ID.")#define NETID_TEXT N_("NET ID")#define NETID_LONGTEXT N_("Assign a fixed Network ID (for SDT table)")#define PMTPROG_TEXT N_("PMT Program numbers")#define PMTPROG_LONGTEXT N_("Assign a program number to each PMT. This " \                            "requires \"Set PID to ID of ES\" to be enabled." )#define MUXPMT_TEXT N_("Mux PMT (requires --sout-ts-es-id-pid)")#define MUXPMT_LONGTEXT N_("Define the pids to add to each pmt. This " \                           "requires \"Set PID to ID of ES\" to be enabled." )#define SDTDESC_TEXT N_("SDT Descriptors (requires --sout-ts-es-id-pid)")#define SDTDESC_LONGTEXT N_("Defines the descriptors of each SDT. This" \                        "requires \"Set PID to ID of ES\" to be enabled." )#define PID_TEXT N_("Set PID to ID of ES")#define PID_LONGTEXT N_("Sets PID to the ID if the incoming ES. This is for " \  "use with --ts-es-id-pid, and allows to have the same PIDs in the input " \  "and output streams.")#define ALIGNMENT_TEXT N_("Data alignment")#define ALIGNMENT_LONGTEXT N_("Enforces alignment of all access units on " \  "PES boundaries. Disabling this might save some bandwidth but introduce incompatibilities.")#define SHAPING_TEXT N_("Shaping delay (ms)")#define SHAPING_LONGTEXT N_("Cut the " \  "stream in slices of the given duration, and ensure a constant bitrate " \  "between the two boundaries. This avoids having huge bitrate peaks, " \  "especially for reference frames." )#define KEYF_TEXT N_("Use keyframes")#define KEYF_LONGTEXT N_("If enabled, and shaping is specified, " \  "the TS muxer will place the boundaries at the end of I pictures. In " \  "that case, the shaping duration given by the user is a worse case " \  "used when no reference frame is available. This enhances the efficiency " \  "of the shaping algorithm, since I frames are usually the biggest " \  "frames in the stream.")#define PCR_TEXT N_("PCR delay (ms)")#define PCR_LONGTEXT N_("Set at which interval " \  "PCRs (Program Clock Reference) will be sent (in milliseconds). " \  "This value should be below 100ms. (default is 70ms).")#define BMIN_TEXT N_( "Minimum B (deprecated)")#define BMIN_LONGTEXT N_( "This setting is deprecated and not used anymore" )#define BMAX_TEXT N_( "Maximum B (deprecated)")#define BMAX_LONGTEXT N_( "This setting is deprecated and not used anymore")#define DTS_TEXT N_("DTS delay (ms)")#define DTS_LONGTEXT N_("Delay the DTS (decoding time " \  "stamps) and PTS (presentation timestamps) of the data in the " \  "stream, compared to the PCRs. This allows for some buffering inside " \  "the client decoder.")#define ACRYPT_TEXT N_("Crypt audio")#define ACRYPT_LONGTEXT N_("Crypt audio using CSA")#define VCRYPT_TEXT N_("Crypt video")#define VCRYPT_LONGTEXT N_("Crypt video using CSA")#define CK_TEXT N_("CSA Key")#define CK_LONGTEXT N_("CSA encryption key. This must be a " \  "16 char string (8 hexadecimal bytes).")#define CPKT_TEXT N_("Packet size in bytes to encrypt")/// \bug [String] Extra space#define CPKT_LONGTEXT N_("Size of the TS packet to encrypt. " \    "The encryption routines subtract the TS-header from the value before " \    "encrypting. " )#define SOUT_CFG_PREFIX "sout-ts-"#ifdef HAVE_BSEARCH#   define MAX_PMT 64       /* Maximum number of programs. FIXME: I just chose an arbitary number. Where is the maximum in the spec? */#else#   define MAX_PMT 1#endif#define MAX_PMT_PID 64       /* Maximum pids in each pmt.  FIXME: I just chose an arbitary number. Where is the maximum in the spec? */vlc_module_begin();    set_description( _("TS muxer (libdvbpsi)") );    set_shortname( "MPEG-TS");    set_category( CAT_SOUT );    set_subcategory( SUBCAT_SOUT_MUX );    set_capability( "sout mux", 120 );    add_shortcut( "ts" );    add_integer( SOUT_CFG_PREFIX "pid-video", 0, NULL,VPID_TEXT, VPID_LONGTEXT,                                  VLC_TRUE );    add_integer( SOUT_CFG_PREFIX "pid-audio", 0, NULL, APID_TEXT,                 APID_LONGTEXT, VLC_TRUE );    add_integer( SOUT_CFG_PREFIX "pid-spu", 0, NULL, SPUPID_TEXT,                 SPUPID_LONGTEXT, VLC_TRUE );    add_integer( SOUT_CFG_PREFIX "pid-pmt", 0, NULL, PMTPID_TEXT,                 PMTPID_LONGTEXT, VLC_TRUE );    add_integer( SOUT_CFG_PREFIX "tsid", 0, NULL, TSID_TEXT,                 TSID_LONGTEXT, VLC_TRUE );#ifdef HAVE_DVBPSI_SDT    add_integer( SOUT_CFG_PREFIX "netid", 0, NULL, NETID_TEXT,                 NETID_LONGTEXT, VLC_TRUE );#endif    add_string( SOUT_CFG_PREFIX "program-pmt", NULL, NULL, PMTPROG_TEXT,                PMTPROG_LONGTEXT, VLC_TRUE );    add_bool( SOUT_CFG_PREFIX "es-id-pid", 0, NULL, PID_TEXT, PID_LONGTEXT,              VLC_TRUE );    add_string( SOUT_CFG_PREFIX "muxpmt", NULL, NULL, MUXPMT_TEXT, MUXPMT_LONGTEXT, VLC_TRUE );#ifdef HAVE_DVBPSI_SDT    add_string( SOUT_CFG_PREFIX "sdtdesc", NULL, NULL, SDTDESC_TEXT, SDTDESC_LONGTEXT, VLC_TRUE );#endif    add_bool( SOUT_CFG_PREFIX "alignment", VLC_TRUE, NULL, ALIGNMENT_TEXT,              ALIGNMENT_LONGTEXT, VLC_TRUE );    add_integer( SOUT_CFG_PREFIX "shaping", 200, NULL,SHAPING_TEXT,                 SHAPING_LONGTEXT, VLC_TRUE );    add_bool( SOUT_CFG_PREFIX "use-key-frames", VLC_FALSE, NULL, KEYF_TEXT,              KEYF_LONGTEXT, VLC_TRUE );    add_integer( SOUT_CFG_PREFIX "pcr", 70, NULL, PCR_TEXT, PCR_LONGTEXT,                 VLC_TRUE );    add_integer( SOUT_CFG_PREFIX "bmin", 0, NULL, BMIN_TEXT, BMIN_LONGTEXT,                 VLC_TRUE );    add_integer( SOUT_CFG_PREFIX "bmax", 0, NULL, BMAX_TEXT, BMAX_LONGTEXT,                 VLC_TRUE );    add_integer( SOUT_CFG_PREFIX "dts-delay", 400, NULL, DTS_TEXT,                 DTS_LONGTEXT, VLC_TRUE );    add_bool( SOUT_CFG_PREFIX "crypt-audio", VLC_TRUE, NULL, ACRYPT_TEXT,              ACRYPT_LONGTEXT, VLC_TRUE );    add_bool( SOUT_CFG_PREFIX "crypt-video", VLC_TRUE, NULL, VCRYPT_TEXT,              VCRYPT_LONGTEXT, VLC_TRUE );    add_string( SOUT_CFG_PREFIX "csa-ck", NULL, NULL, CK_TEXT, CK_LONGTEXT,                VLC_TRUE );    add_integer( SOUT_CFG_PREFIX "csa-pkt", 188, NULL, CPKT_TEXT, CPKT_LONGTEXT, VLC_TRUE );    set_callbacks( Open, Close );vlc_module_end();/***************************************************************************** * Local data structures *****************************************************************************/static const char *ppsz_sout_options[] = {    "pid-video", "pid-audio", "pid-spu", "pid-pmt", "tsid", "netid",    "es-id-pid", "shaping", "pcr", "bmin", "bmax", "use-key-frames",    "dts-delay", "csa-ck", "csa-pkt", "crypt-audio", "crypt-video",    "muxpmt", "sdtdesc", "program-pmt", "alignment",    NULL};typedef struct pmt_map_t   /* Holds the mapping between the pmt-pid/pmt table */{    int i_pid;    unsigned long i_prog;} pmt_map_t;typedef struct sdt_desc_t{    char *psz_provider;    char *psz_service_name;  /* name of program */} sdt_desc_t;typedef struct{    int     i_depth;    block_t *p_first;    block_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, block_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 block_t *BufferChainGet( sout_buffer_chain_t *c ){    block_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 block_t *BufferChainPeek( sout_buffer_chain_t *c ){    block_t *b = c->p_first;    return b;}static inline void BufferChainClean( sout_instance_t *p_sout,                                     sout_buffer_chain_t *c ){    block_t *b;    while( ( b = BufferChainGet( c ) ) )    {        block_Release( b );    }    BufferChainInit( c );}typedef struct ts_stream_t{    int             i_pid;    vlc_fourcc_t    i_codec;    int             i_stream_type;    int             i_stream_id;    int             i_continuity_counter;    vlc_bool_t      b_discontinuity;    /* 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 */    int             i_langs;    uint8_t         *lang;    sout_buffer_chain_t chain_pes;    mtime_t             i_pes_dts;    mtime_t             i_pes_length;    int                 i_pes_used;    vlc_bool_t          b_key_frame;} 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;    vlc_bool_t      b_es_id_pid;    vlc_bool_t      b_sdt;    int             i_pid_video;    int             i_pid_audio;    int             i_pid_spu;    int             i_pid_free; /* first usable pid */    int             i_tsid;    int             i_netid;    int             i_num_pmt;    int             i_pmtslots;    int             i_pat_version_number;    ts_stream_t     pat;    int             i_pmt_version_number;    ts_stream_t     pmt[MAX_PMT];    pmt_map_t       pmtmap[MAX_PMT_PID];    int             i_pmt_program_number[MAX_PMT];