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

   1 /*****************************************************************************
   2  * ts.c: Transport Stream input module for VLC.
   3  *****************************************************************************
   4  * Copyright (C) 2004-2005 the VideoLAN team
   5  * $Id$
   6  *
   7  * Authors: Laurent Aimar <fenrir@via.ecp.fr>
   8  *          Jean-Paul Saman <jpsaman #_at_# m2x.nl>
   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., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
  23  *****************************************************************************/
  24
  25 /*****************************************************************************
  26  * Preamble
  27  *****************************************************************************/
  28
  29 #ifdef HAVE_CONFIG_H
  30 # include "config.h"
  31 #endif
  32
  33 #include <vlc_common.h>
  34 #include <vlc_plugin.h>
  35
  36 #include <ctype.h>
  37 #include <assert.h>
  38
  39 #include <vlc_access.h> /* DVB-specific things */
  40 #include <vlc_demux.h>
  41 #include <vlc_meta.h>
  42 #include <vlc_epg.h>
  43
  44 #include <vlc_iso_lang.h>
  45 #include <vlc_network.h>
  46 #include <vlc_charset.h>
  47
  48 #include "../mux/mpeg/csa.h"
  49
  50 /* Include dvbpsi headers */
  51 #ifdef HAVE_DVBPSI_DR_H
  52 #   include <dvbpsi/dvbpsi.h>
  53 #   include <dvbpsi/demux.h>
  54 #   include <dvbpsi/descriptor.h>
  55 #   include <dvbpsi/pat.h>
  56 #   include <dvbpsi/pmt.h>
  57 #   include <dvbpsi/sdt.h>
  58 #   include <dvbpsi/dr.h>
  59 #   include <dvbpsi/psi.h>
  60 #else
  61 #   include "dvbpsi.h"
  62 #   include "demux.h"
  63 #   include "descriptor.h"
  64 #   include "tables/pat.h"
  65 #   include "tables/pmt.h"
  66 #   include "tables/sdt.h"
  67 #   include "descriptors/dr.h"
  68 #   include "psi.h"
  69 #endif
  70
  71 /* EIT support */
  72 #ifdef _DVBPSI_DR_4D_H_
  73 #   define TS_USE_DVB_SI 1
  74 #   ifdef HAVE_DVBPSI_DR_H
  75 #       include <dvbpsi/eit.h>
  76 #   else
  77 #       include "tables/eit.h"
  78 #   endif
  79 #endif
  80 #ifdef HAVE_TIME_H
  81 #   include <time.h>
  82 #endif
  83 #undef TS_DEBUG
  84
  85 /* TODO:
  86  *  - XXX: do not mark options message to be translated, they are too osbcure for now ...
  87  *  - test it
  88  *  - ...
  89  */
  90
  91 /*****************************************************************************
  92  * Callback prototypes
  93  *****************************************************************************/
  94 static int ChangeKeyCallback    ( vlc_object_t *, char const *, vlc_value_t, vlc_value_t, void * );
  95
  96 /*****************************************************************************
  97  * Module descriptor
  98  *****************************************************************************/
  99 static int  Open  ( vlc_object_t * );
 100 static void Close ( vlc_object_t * );
 101
 102 /* TODO
 103  * - Rename "extra pmt" to "user pmt"
 104  * - Update extra pmt description
 105  *      pmt_pid[:pmt_number][=pid_description[,pid_description]]
 106  *      where pid_description could take 3 forms:
 107  *          1. pid:pcr (to force the pcr pid)
 108  *          2. pid:stream_type
 109  *          3. pid:type=fourcc where type=(video|audio|spu)
 110  */
 111 #define PMT_TEXT N_("Extra PMT")
 112 #define PMT_LONGTEXT N_( \
 113   "Allows a user to specify an extra pmt (pmt_pid=pid:stream_type[,...])." )
 114
 115 #define PID_TEXT N_("Set id of ES to PID")
 116 #define PID_LONGTEXT N_("Set the internal ID of each elementary stream" \
 117                        " handled by VLC to the same value as the PID in" \
 118                        " the TS stream, instead of 1, 2, 3, etc. Useful to" \
 119                        " do \'#duplicate{..., select=\"es=<pid>\"}\'.")
 120
 121 #define TSOUT_TEXT N_("Fast udp streaming")
 122 #define TSOUT_LONGTEXT N_( \
 123   "Sends TS to specific ip:port by udp (you must know what you are doing).")
 124
 125 #define MTUOUT_TEXT N_("MTU for out mode")
 126 #define MTUOUT_LONGTEXT N_("MTU for out mode.")
 127
 128 #define CSA_TEXT N_("CSA ck")
 129 #define CSA_LONGTEXT N_("Control word for the CSA encryption algorithm")
 130
 131 #define CSA2_TEXT N_("Second CSA Key")
 132 #define CSA2_LONGTEXT N_("The even CSA encryption key. This must be a " \
 133   "16 char string (8 hexadecimal bytes).")
 134
 135 #define SILENT_TEXT N_("Silent mode")
 136 #define SILENT_LONGTEXT N_("Do not complain on encrypted PES.")
 137
 138 #define CAPMT_SYSID_TEXT N_("CAPMT System ID")
 139 #define CAPMT_SYSID_LONGTEXT N_("Only forward descriptors from this SysID to the CAM.")
 140
 141 #define CPKT_TEXT N_("Packet size in bytes to decrypt")
 142 #define CPKT_LONGTEXT N_("Specify the size of the TS packet to decrypt. " \
 143     "The decryption routines subtract the TS-header from the value before " \
 144     "decrypting. " )
 145
 146 #define TSDUMP_TEXT N_("Filename of dump")
 147 #define TSDUMP_LONGTEXT N_("Specify a filename where to dump the TS in.")
 148
 149 #define APPEND_TEXT N_("Append")
 150 #define APPEND_LONGTEXT N_( \
 151     "If the file exists and this option is selected, the existing file " \
 152     "will not be overwritten." )
 153
 154 #define DUMPSIZE_TEXT N_("Dump buffer size")
 155 #define DUMPSIZE_LONGTEXT N_( \
 156     "Tweak the buffer size for reading and writing an integer number of packets." \
 157     "Specify the size of the buffer here and not the number of packets." )
 158
 159 vlc_module_begin ()
 160     set_description( N_("MPEG Transport Stream demuxer") )
 161     set_shortname ( "MPEG-TS" )
 162     set_category( CAT_INPUT )
 163     set_subcategory( SUBCAT_INPUT_DEMUX )
 164
 165     add_string( "ts-extra-pmt", NULL, NULL, PMT_TEXT, PMT_LONGTEXT, true )
 166     add_bool( "ts-es-id-pid", 1, NULL, PID_TEXT, PID_LONGTEXT, true )
 167     add_string( "ts-out", NULL, NULL, TSOUT_TEXT, TSOUT_LONGTEXT, true )
 168     add_integer( "ts-out-mtu", 1400, NULL, MTUOUT_TEXT,
 169                  MTUOUT_LONGTEXT, true )
 170     add_string( "ts-csa-ck", NULL, NULL, CSA_TEXT, CSA_LONGTEXT, true )
 171     add_string( "ts-csa2-ck", NULL, NULL, CSA_TEXT, CSA_LONGTEXT, true )
 172     add_integer( "ts-csa-pkt", 188, NULL, CPKT_TEXT, CPKT_LONGTEXT, true )
 173     add_bool( "ts-silent", 0, NULL, SILENT_TEXT, SILENT_LONGTEXT, true )
 174
 175     add_file( "ts-dump-file", NULL, NULL, TSDUMP_TEXT, TSDUMP_LONGTEXT, false )
 176     add_bool( "ts-dump-append", 0, NULL, APPEND_TEXT, APPEND_LONGTEXT, false )
 177     add_integer( "ts-dump-size", 16384, NULL, DUMPSIZE_TEXT,
 178                  DUMPSIZE_LONGTEXT, true )
 179
 180     set_capability( "demux", 10 )
 181     set_callbacks( Open, Close )
 182     add_shortcut( "ts" )
 183 vlc_module_end ()
 184
 185 /*****************************************************************************
 186  * Local prototypes
 187  *****************************************************************************/
 188 static const char *const ppsz_teletext_type[] = {
 189  "",
 190  N_("Teletext"),
 191  N_("Teletext subtitles"),
 192  N_("Teletext: additional information"),
 193  N_("Teletext: program schedule"),
 194  N_("Teletext subtitles: hearing impaired")
 195 };
 196
 197 typedef struct
 198 {
 199     uint8_t                 i_objectTypeIndication;
 200     uint8_t                 i_streamType;
 201     bool                    b_upStream;
 202     uint32_t                i_bufferSizeDB;
 203     uint32_t                i_maxBitrate;
 204     uint32_t                i_avgBitrate;
 205
 206     int                     i_decoder_specific_info_len;
 207     uint8_t                 *p_decoder_specific_info;
 208
 209 } decoder_config_descriptor_t;
 210
 211 typedef struct
 212 {
 213     bool                    b_useAccessUnitStartFlag;
 214     bool                    b_useAccessUnitEndFlag;
 215     bool                    b_useRandomAccessPointFlag;
 216     bool                    b_useRandomAccessUnitsOnlyFlag;
 217     bool                    b_usePaddingFlag;
 218     bool                    b_useTimeStampsFlags;
 219     bool                    b_useIdleFlag;
 220     bool                    b_durationFlag;
 221     uint32_t                i_timeStampResolution;
 222     uint32_t                i_OCRResolution;
 223     uint8_t                 i_timeStampLength;
 224     uint8_t                 i_OCRLength;
 225     uint8_t                 i_AU_Length;
 226     uint8_t                 i_instantBitrateLength;
 227     uint8_t                 i_degradationPriorityLength;
 228     uint8_t                 i_AU_seqNumLength;
 229     uint8_t                 i_packetSeqNumLength;
 230
 231     uint32_t                i_timeScale;
 232     uint16_t                i_accessUnitDuration;
 233     uint16_t                i_compositionUnitDuration;
 234
 235     uint64_t                i_startDecodingTimeStamp;
 236     uint64_t                i_startCompositionTimeStamp;
 237
 238 } sl_config_descriptor_t;
 239
 240 typedef struct
 241 {
 242     bool                    b_ok;
 243     uint16_t                i_es_id;
 244
 245     bool                    b_streamDependenceFlag;
 246     bool                    b_OCRStreamFlag;
 247     uint8_t                 i_streamPriority;
 248
 249     char                    *psz_url;
 250
 251     uint16_t                i_dependOn_es_id;
 252     uint16_t                i_OCR_es_id;
 253
 254     decoder_config_descriptor_t    dec_descr;
 255     sl_config_descriptor_t         sl_descr;
 256
 257 } es_mpeg4_descriptor_t;
 258
 259 typedef struct
 260 {
 261     uint8_t                 i_iod_label, i_iod_label_scope;
 262
 263     /* IOD */
 264     uint16_t                i_od_id;
 265     char                    *psz_url;
 266
 267     uint8_t                 i_ODProfileLevelIndication;
 268     uint8_t                 i_sceneProfileLevelIndication;
 269     uint8_t                 i_audioProfileLevelIndication;
 270     uint8_t                 i_visualProfileLevelIndication;
 271     uint8_t                 i_graphicsProfileLevelIndication;
 272
 273     es_mpeg4_descriptor_t   es_descr[255];
 274
 275 } iod_descriptor_t;
 276
 277 typedef struct
 278 {
 279     dvbpsi_handle   handle;
 280
 281     int             i_version;
 282     int             i_number;
 283     int             i_pid_pcr;
 284     int             i_pid_pmt;
 285     /* IOD stuff (mpeg4) */
 286     iod_descriptor_t *iod;
 287
 288 } ts_prg_psi_t;
 289
 290 typedef struct
 291 {
 292     /* for special PAT/SDT case */
 293     dvbpsi_handle   handle; /* PAT/SDT/EIT */
 294     int             i_pat_version;
 295     int             i_sdt_version;
 296
 297     /* For PMT */
 298     int             i_prg;
 299     ts_prg_psi_t    **prg;
 300
 301 } ts_psi_t;
 302
 303 typedef struct
 304 {
 305     es_format_t  fmt;
 306     es_out_id_t *id;
 307     int         i_pes_size;
 308     int         i_pes_gathered;
 309     block_t     *p_pes;
 310     block_t     **pp_last;
 311
 312     es_mpeg4_descriptor_t *p_mpeg4desc;
 313     int         b_gather;
 314
 315 } ts_es_t;
 316
 317 typedef struct
 318 {
 319     int         i_pid;
 320
 321     bool        b_seen;
 322     bool        b_valid;
 323     int         i_cc;   /* countinuity counter */
 324
 325     /* PSI owner (ie PMT -> PAT, ES -> PMT */
 326     ts_psi_t   *p_owner;
 327     int         i_owner_number;
 328
 329     /* */
 330     ts_psi_t    *psi;
 331     ts_es_t     *es;
 332
 333     /* Some private streams encapsulate several ES (eg. DVB subtitles)*/
 334     ts_es_t     **extra_es;
 335     int         i_extra_es;
 336
 337 } ts_pid_t;
 338
 339 struct demux_sys_t
 340 {
 341     vlc_mutex_t     csa_lock;
 342
 343     /* TS packet size (188, 192, 204) */
 344     int         i_packet_size;
 345
 346     /* how many TS packet we read at once */
 347     int         i_ts_read;
 348
 349     /* All pid */
 350     ts_pid_t    pid[8192];
 351
 352     /* All PMT */
 353     bool        b_user_pmt;
 354     int         i_pmt;
 355     ts_pid_t    **pmt;
 356
 357     /* */
 358     bool        b_es_id_pid;
 359     csa_t       *csa;
 360     int         i_csa_pkt_size;
 361     bool        b_silent;
 362
 363     bool        b_udp_out;
 364     int         fd; /* udp socket */
 365     uint8_t     *buffer;
 366
 367     bool        b_dvb_control;
 368     int         i_dvb_program;
 369     int64_t     i_dvb_start;
 370     int64_t     i_dvb_length;
 371     vlc_list_t  *p_programs_list;
 372
 373     /* TS dump */
 374     char        *psz_file;  /* file to dump data in */
 375     FILE        *p_file;    /* filehandle */
 376     uint64_t    i_write;    /* bytes written */
 377     bool        b_file_out; /* dump mode enabled */
 378
 379     /* */
 380     bool        b_meta;
 381
 382     /* */
 383     bool        b_start_record;
 384 };
 385
 386 static int Demux    ( demux_t *p_demux );
 387 static int DemuxFile( demux_t *p_demux );
 388 static int Control( demux_t *p_demux, int i_query, va_list args );
 389
 390 static void PIDInit ( ts_pid_t *pid, bool b_psi, ts_psi_t *p_owner );
 391 static void PIDClean( es_out_t *out, ts_pid_t *pid );
 392 static int  PIDFillFormat( ts_pid_t *pid, int i_stream_type );
 393
 394 static void PATCallBack( demux_t *, dvbpsi_pat_t * );
 395 static void PMTCallBack( demux_t *p_demux, dvbpsi_pmt_t *p_pmt );
 396 #ifdef TS_USE_DVB_SI
 397 static void PSINewTableCallBack( demux_t *, dvbpsi_handle,
 398                                  uint8_t  i_table_id, uint16_t i_extension );
 399 #endif
 400
 401 static inline int PIDGet( block_t *p )
 402 {
 403     return ( (p->p_buffer[1]&0x1f)<<8 )|p->p_buffer[2];
 404 }
 405
 406 static bool GatherPES( demux_t *p_demux, ts_pid_t *pid, block_t *p_bk );
 407
 408 static void PCRHandle( demux_t *p_demux, ts_pid_t *, block_t * );
 409
 410 static iod_descriptor_t *IODNew( int , uint8_t * );
 411 static void              IODFree( iod_descriptor_t * );
 412
 413 #define TS_USER_PMT_NUMBER (0)
 414 static int UserPmt( demux_t *p_demux, const char * );
 415
 416 #define TS_PACKET_SIZE_188 188
 417 #define TS_PACKET_SIZE_192 192
 418 #define TS_PACKET_SIZE_204 204
 419 #define TS_PACKET_SIZE_MAX 204
 420 #define TS_TOPFIELD_HEADER 1320
 421
 422 /*****************************************************************************
 423  * Open
 424  *****************************************************************************/
 425 static int Open( vlc_object_t *p_this )
 426 {
 427     demux_t     *p_demux = (demux_t*)p_this;
 428     demux_sys_t *p_sys;
 429
 430     const uint8_t *p_peek;
 431     int          i_sync, i_peek, i;
 432     int          i_packet_size;
 433
 434     ts_pid_t    *pat;
 435     const char  *psz_mode;
 436     bool         b_append;
 437     bool         b_topfield = false;
 438
 439     if( stream_Peek( p_demux->s, &p_peek, TS_PACKET_SIZE_MAX ) <
 440         TS_PACKET_SIZE_MAX ) return VLC_EGENERIC;
 441
 442     if( memcmp( p_peek, "TFrc", 4 ) == 0 )
 443     {
 444         b_topfield = true;
 445         msg_Dbg( p_demux, "this is a topfield file" );
 446     }
 447
 448     /* Search first sync byte */
 449     for( i_sync = 0; i_sync < TS_PACKET_SIZE_MAX; i_sync++ )
 450     {
 451         if( p_peek[i_sync] == 0x47 ) break;
 452     }
 453     if( i_sync >= TS_PACKET_SIZE_MAX && !b_topfield )
 454     {
 455         if( !p_demux->b_force )
 456             return VLC_EGENERIC;
 457         msg_Warn( p_demux, "this does not look like a TS stream, continuing" );
 458     }
 459
 460     if( b_topfield )
 461     {
 462         /* Read the entire Topfield header */
 463         i_peek = TS_TOPFIELD_HEADER;
 464     }
 465     else
 466     {
 467         /* Check next 3 sync bytes */
 468         i_peek = TS_PACKET_SIZE_MAX * 3 + i_sync + 1;
 469     }
 470
 471     if( ( stream_Peek( p_demux->s, &p_peek, i_peek ) ) < i_peek )
 472     {
 473         msg_Err( p_demux, "cannot peek" );
 474         return VLC_EGENERIC;
 475     }
 476     if( p_peek[i_sync + TS_PACKET_SIZE_188] == 0x47 &&
 477         p_peek[i_sync + 2 * TS_PACKET_SIZE_188] == 0x47 &&
 478         p_peek[i_sync + 3 * TS_PACKET_SIZE_188] == 0x47 )
 479     {
 480         i_packet_size = TS_PACKET_SIZE_188;
 481     }
 482     else if( p_peek[i_sync + TS_PACKET_SIZE_192] == 0x47 &&
 483              p_peek[i_sync + 2 * TS_PACKET_SIZE_192] == 0x47 &&
 484              p_peek[i_sync + 3 * TS_PACKET_SIZE_192] == 0x47 )
 485     {
 486         i_packet_size = TS_PACKET_SIZE_192;
 487     }
 488     else if( p_peek[i_sync + TS_PACKET_SIZE_204] == 0x47 &&
 489              p_peek[i_sync + 2 * TS_PACKET_SIZE_204] == 0x47 &&
 490              p_peek[i_sync + 3 * TS_PACKET_SIZE_204] == 0x47 )
 491     {
 492         i_packet_size = TS_PACKET_SIZE_204;
 493     }
 494     else if( p_demux->b_force )
 495     {
 496         i_packet_size = TS_PACKET_SIZE_188;
 497     }
 498     else if( b_topfield )
 499     {
 500         i_packet_size = TS_PACKET_SIZE_188;
 501 #if 0
 502         /* I used the TF5000PVR 2004 Firmware .doc header documentation,
 503          * http://www.i-topfield.com/data/product/firmware/Structure%20of%20Recorded%20File%20in%20TF5000PVR%20(Feb%2021%202004).doc
 504          * but after the filename the offsets seem to be incorrect.  - DJ */
 505         int i_duration, i_name;
 506         char *psz_name = malloc(25);
 507         char *psz_event_name;
 508         char *psz_event_text = malloc(130);
 509         char *psz_ext_text = malloc(1025);
 510
 511         // 2 bytes version Uimsbf (4,5)
 512         // 2 bytes reserved (6,7)
 513         // 2 bytes duration in minutes Uimsbf (8,9(
 514         i_duration = (int) (p_peek[8] << 8) | p_peek[9];
 515         msg_Dbg( p_demux, "Topfield recording length: +/- %d minutes", i_duration);
 516         // 2 bytes service number in channel list (10, 11)
 517         // 2 bytes service type Bslbf 0=TV 1=Radio Bslb (12, 13)
 518         // 4 bytes of reserved + tuner info (14,15,16,17)
 519         // 2 bytes of Service ID  Bslbf (18,19)
 520         // 2 bytes of PMT PID  Uimsbf (20,21)
 521         // 2 bytes of PCR PID  Uimsbf (22,23)
 522         // 2 bytes of Video PID  Uimsbf (24,25)
 523         // 2 bytes of Audio PID  Uimsbf (26,27)
 524         // 24 bytes filename Bslbf
 525         memcpy( psz_name, &p_peek[28], 24 );
 526         psz_name[24] = '\0';
 527         msg_Dbg( p_demux, "recordingname=%s", psz_name );
 528         // 1 byte of sat index Uimsbf  (52)
 529         // 3 bytes (1 bit of polarity Bslbf +23 bits reserved)
 530         // 4 bytes of freq. Uimsbf (56,57,58,59)
 531         // 2 bytes of symbol rate Uimsbf (60,61)
 532         // 2 bytes of TS stream ID Uimsbf (62,63)
 533         // 4 bytes reserved
 534         // 2 bytes reserved
 535         // 2 bytes duration Uimsbf (70,71)
 536         //i_duration = (int) (p_peek[70] << 8) | p_peek[71];
 537         //msg_Dbg( p_demux, "Topfield 2nd duration field: +/- %d minutes", i_duration);
 538         // 4 bytes EventID Uimsbf (72-75)
 539         // 8 bytes of Start and End time info (76-83)
 540         // 1 byte reserved (84)
 541         // 1 byte event name length Uimsbf (89)
 542         i_name = (int)(p_peek[89]&~0x81);
 543         msg_Dbg( p_demux, "event name length = %d", i_name);
 544         psz_event_name = malloc( i_name+1 );
 545         // 1 byte parental rating (90)
 546         // 129 bytes of event text
 547         memcpy( psz_event_name, &p_peek[91], i_name );
 548         psz_event_name[i_name] = '\0';
 549         memcpy( psz_event_text, &p_peek[91+i_name], 129-i_name );
 550         psz_event_text[129-i_name] = '\0';
 551         msg_Dbg( p_demux, "event name=%s", psz_event_name );
 552         msg_Dbg( p_demux, "event text=%s", psz_event_text );
 553         // 12 bytes reserved (220)
 554         // 6 bytes reserved
 555         // 2 bytes Event Text Length Uimsbf
 556         // 4 bytes EventID Uimsbf
 557         // FIXME We just have 613 bytes. not enough for this entire text
 558         // 1024 bytes Extended Event Text Bslbf
 559         memcpy( psz_ext_text, p_peek+372, 1024 );
 560         psz_ext_text[1024] = '\0';
 561         msg_Dbg( p_demux, "extended event text=%s", psz_ext_text );
 562         // 52 bytes reserved Bslbf
 563 #endif
 564     }
 565     else
 566     {
 567         msg_Warn( p_demux, "TS module discarded (lost sync)" );
 568         return VLC_EGENERIC;
 569     }
 570
 571     p_demux->p_sys = p_sys = malloc( sizeof( demux_sys_t ) );
 572     if( !p_sys )
 573         return VLC_ENOMEM;
 574     memset( p_sys, 0, sizeof( demux_sys_t ) );
 575     p_sys->i_packet_size = i_packet_size;
 576     vlc_mutex_init( &p_sys->csa_lock );
 577
 578     /* Fill dump mode fields */
 579     p_sys->i_write = 0;
 580     p_sys->buffer = NULL;
 581     p_sys->p_file = NULL;
 582     p_sys->b_file_out = false;
 583     p_sys->psz_file = var_CreateGetString( p_demux, "ts-dump-file" );
 584     if( *p_sys->psz_file != '\0' )
 585     {
 586         p_sys->b_file_out = true;
 587
 588         b_append = var_CreateGetBool( p_demux, "ts-dump-append" );
 589         if ( b_append )
 590             psz_mode = "ab";
 591         else
 592             psz_mode = "wb";
 593
 594         if( !strcmp( p_sys->psz_file, "-" ) )
 595         {
 596             msg_Info( p_demux, "dumping raw stream to standard output" );
 597             p_sys->p_file = stdout;
 598         }
 599         else if( ( p_sys->p_file = utf8_fopen( p_sys->psz_file, psz_mode ) ) == NULL )
 600         {
 601             msg_Err( p_demux, "cannot create `%s' for writing", p_sys->psz_file );
 602             p_sys->b_file_out = false;
 603         }
 604
 605         if( p_sys->b_file_out )
 606         {
 607             /* Determine how many packets to read. */
 608             int bufsize = var_CreateGetInteger( p_demux, "ts-dump-size" );
 609             p_sys->i_ts_read = (int) (bufsize / p_sys->i_packet_size);
 610             if( p_sys->i_ts_read <= 0 )
 611             {
 612                 p_sys->i_ts_read = 1500 / p_sys->i_packet_size;
 613             }
 614             p_sys->buffer = malloc( p_sys->i_packet_size * p_sys->i_ts_read );
 615             msg_Info( p_demux, "%s raw stream to file `%s' reading packets %d",
 616                       b_append ? "appending" : "dumping", p_sys->psz_file,
 617                       p_sys->i_ts_read );
 618         }
 619     }
 620
 621     /* Fill p_demux field */
 622     if( p_sys->b_file_out )
 623         p_demux->pf_demux = DemuxFile;
 624     else
 625         p_demux->pf_demux = Demux;
 626     p_demux->pf_control = Control;
 627
 628     /* Init p_sys field */
 629     p_sys->b_meta = true;
 630     p_sys->b_dvb_control = true;
 631     p_sys->i_dvb_program = 0;
 632     p_sys->i_dvb_start = 0;
 633     p_sys->i_dvb_length = 0;
 634
 635     for( i = 0; i < 8192; i++ )
 636     {
 637         ts_pid_t *pid = &p_sys->pid[i];
 638
 639         pid->i_pid      = i;
 640         pid->b_seen     = false;
 641         pid->b_valid    = false;
 642     }
 643     /* PID 8191 is padding */
 644     p_sys->pid[8191].b_seen = true;
 645     p_sys->i_packet_size = i_packet_size;
 646     p_sys->b_udp_out = false;
 647     p_sys->fd = -1;
 648     p_sys->i_ts_read = 50;
 649     p_sys->csa = NULL;
 650     p_sys->b_start_record = false;
 651
 652     /* Init PAT handler */
 653     pat = &p_sys->pid[0];
 654     PIDInit( pat, true, NULL );
 655     pat->psi->handle = dvbpsi_AttachPAT( (dvbpsi_pat_callback)PATCallBack,
 656                                          p_demux );
 657 #ifdef TS_USE_DVB_SI
 658     if( p_sys->b_meta )
 659     {
 660         ts_pid_t *sdt = &p_sys->pid[0x11];
 661         ts_pid_t *eit = &p_sys->pid[0x12];
 662
 663         PIDInit( sdt, true, NULL );
 664         sdt->psi->handle =
 665             dvbpsi_AttachDemux( (dvbpsi_demux_new_cb_t)PSINewTableCallBack,
 666                                 p_demux );
 667         PIDInit( eit, true, NULL );
 668         eit->psi->handle =
 669             dvbpsi_AttachDemux( (dvbpsi_demux_new_cb_t)PSINewTableCallBack,
 670                                 p_demux );
 671         if( p_sys->b_dvb_control )
 672         {
 673             if( stream_Control( p_demux->s, STREAM_CONTROL_ACCESS,
 674                                 ACCESS_SET_PRIVATE_ID_STATE, 0x11, true ) ||
 675                 stream_Control( p_demux->s, STREAM_CONTROL_ACCESS,
 676                                 ACCESS_SET_PRIVATE_ID_STATE, 0x12, true ) )
 677                 p_sys->b_dvb_control = false;
 678         }
 679     }
 680 #endif
 681
 682     /* Init PMT array */
 683     TAB_INIT( p_sys->i_pmt, p_sys->pmt );
 684
 685     /* Read config */
 686     p_sys->b_es_id_pid = var_CreateGetBool( p_demux, "ts-es-id-pid" );
 687
 688     char* psz_string = var_CreateGetString( p_demux, "ts-out" );
 689     if( psz_string && *psz_string && !p_sys->b_file_out )
 690     {
 691         char *psz = strchr( psz_string, ':' );
 692         int   i_port = 0;
 693
 694         p_sys->b_udp_out = true;
 695
 696         if( psz )
 697         {
 698             *psz++ = '\0';
 699             i_port = atoi( psz );
 700         }
 701         if( i_port <= 0 ) i_port  = 1234;
 702         msg_Dbg( p_demux, "resend ts to '%s:%d'", psz_string, i_port );
 703
 704         p_sys->fd = net_ConnectUDP( VLC_OBJECT(p_demux), psz_string, i_port, -1 );
 705         if( p_sys->fd < 0 )
 706         {
 707             msg_Err( p_demux, "failed to open udp socket, send disabled" );
 708             p_sys->b_udp_out = false;
 709         }
 710         else
 711         {
 712             int i_mtu = var_CreateGetInteger( p_demux, "ts-out-mtu" );
 713             p_sys->i_ts_read = i_mtu / p_sys->i_packet_size;
 714             if( p_sys->i_ts_read <= 0 )
 715             {
 716                 p_sys->i_ts_read = 1500 / p_sys->i_packet_size;
 717             }
 718             p_sys->buffer = malloc( p_sys->i_packet_size * p_sys->i_ts_read );
 719         }
 720     }
 721     free( psz_string );
 722
 723     /* We handle description of an extra PMT */
 724     psz_string = var_CreateGetString( p_demux, "ts-extra-pmt" );
 725     p_sys->b_user_pmt = false;
 726     if( psz_string && *psz_string )
 727         UserPmt( p_demux, psz_string );
 728     free( psz_string );
 729
 730     psz_string = var_CreateGetStringCommand( p_demux, "ts-csa-ck" );
 731     if( psz_string && *psz_string )
 732     {
 733         int i_res;
 734         char* psz_csa2;
 735
 736         p_sys->csa = csa_New();
 737
 738         psz_csa2 = var_CreateGetStringCommand( p_demux, "ts-csa2-ck" );
 739         i_res = csa_SetCW( (vlc_object_t*)p_demux, p_sys->csa, psz_string, true );
 740         if( i_res == VLC_SUCCESS && psz_csa2 && *psz_csa2 )
 741         {
 742             if( csa_SetCW( (vlc_object_t*)p_demux, p_sys->csa, psz_csa2, false ) != VLC_SUCCESS )
 743             {
 744                 csa_SetCW( (vlc_object_t*)p_demux, p_sys->csa, psz_string, false );
 745             }
 746         }
 747         else if ( i_res == VLC_SUCCESS )
 748         {
 749             csa_SetCW( (vlc_object_t*)p_demux, p_sys->csa, psz_string, false );
 750         }
 751         else
 752         {
 753             csa_Delete( p_sys->csa );
 754             p_sys->csa = NULL;
 755         }
 756
 757         if( p_sys->csa )
 758         {
 759             var_AddCallback( p_demux, "ts-csa-ck", ChangeKeyCallback, (void *)1 );
 760             var_AddCallback( p_demux, "ts-csa2-ck", ChangeKeyCallback, NULL );
 761
 762             int i_pkt = var_CreateGetInteger( p_demux, "ts-csa-pkt" );
 763             if( i_pkt < 4 || i_pkt > 188 )
 764             {
 765                 msg_Err( p_demux, "wrong packet size %d specified.", i_pkt );
 766                 msg_Warn( p_demux, "using default packet size of 188 bytes" );
 767                 p_sys->i_csa_pkt_size = 188;
 768             }
 769             else
 770                 p_sys->i_csa_pkt_size = i_pkt;
 771             msg_Dbg( p_demux, "decrypting %d bytes of packet", p_sys->i_csa_pkt_size );
 772         }
 773         free( psz_csa2 );
 774     }
 775     free( psz_string );
 776
 777     p_sys->b_silent = var_CreateGetBool( p_demux, "ts-silent" );
 778
 779     return VLC_SUCCESS;
 780 }
 781
 782 /*****************************************************************************
 783  * Close
 784  *****************************************************************************/
 785 static void Close( vlc_object_t *p_this )
 786 {
 787     demux_t     *p_demux = (demux_t*)p_this;
 788     demux_sys_t *p_sys = p_demux->p_sys;
 789
 790     int          i;
 791
 792     msg_Dbg( p_demux, "pid list:" );
 793     for( i = 0; i < 8192; i++ )
 794     {
 795         ts_pid_t *pid = &p_sys->pid[i];
 796
 797         if( pid->b_valid && pid->psi )
 798         {
 799             switch( pid->i_pid )
 800             {
 801                 case 0: /* PAT */
 802                     dvbpsi_DetachPAT( pid->psi->handle );
 803                     free( pid->psi );
 804                     break;
 805                 case 1: /* CAT */
 806                     free( pid->psi );
 807                     break;
 808                 case 0x11: /* SDT */
 809                 case 0x12: /* EIT */
 810                     dvbpsi_DetachDemux( pid->psi->handle );
 811                     free( pid->psi );
 812                     break;
 813                 default:
 814                     PIDClean( p_demux->out, pid );
 815                     break;
 816             }
 817         }
 818         else if( pid->b_valid && pid->es )
 819         {
 820             PIDClean( p_demux->out, pid );
 821         }
 822
 823         if( pid->b_seen )
 824         {
 825             msg_Dbg( p_demux, "  - pid[%d] seen", pid->i_pid );
 826         }
 827
 828         if( p_sys->b_dvb_control && pid->i_pid > 0 )
 829         {
 830             /* too much */
 831             stream_Control( p_demux->s, STREAM_CONTROL_ACCESS,
 832                             ACCESS_SET_PRIVATE_ID_STATE, pid->i_pid,
 833                             false );
 834         }
 835
 836     }
 837
 838     vlc_mutex_lock( &p_sys->csa_lock );
 839     if( p_sys->csa )
 840     {
 841         var_DelCallback( p_demux, "ts-csa-ck", ChangeKeyCallback, NULL );
 842         var_DelCallback( p_demux, "ts-csa2-ck", ChangeKeyCallback, NULL );
 843         csa_Delete( p_sys->csa );
 844     }
 845     vlc_mutex_unlock( &p_sys->csa_lock );
 846
 847     TAB_CLEAN( p_sys->i_pmt, p_sys->pmt );
 848
 849     if ( p_sys->p_programs_list )
 850     {
 851         vlc_value_t val;
 852         val.p_list = p_sys->p_programs_list;
 853         var_Change( p_demux, "programs", VLC_VAR_FREELIST, &val, NULL );
 854     }
 855
 856     /* If in dump mode, then close the file */
 857     if( p_sys->b_file_out )
 858     {
 859         msg_Info( p_demux ,"closing %s (%"PRId64" Kbytes dumped)",
 860                   p_sys->psz_file, p_sys->i_write / 1024 );
 861
 862         if( p_sys->p_file != stdout )
 863         {
 864             fclose( p_sys->p_file );
 865         }
 866     }
 867     /* When streaming, close the port */
 868     if( p_sys->fd > -1 )
 869     {
 870         net_Close( p_sys->fd );
 871     }
 872
 873     free( p_sys->buffer );
 874     free( p_sys->psz_file );
 875
 876     vlc_mutex_destroy( &p_sys->csa_lock );
 877     free( p_sys );
 878 }
 879
 880 /*****************************************************************************
 881  * ChangeKeyCallback: called when changing the odd encryption key on the fly.
 882  *****************************************************************************/
 883 static int ChangeKeyCallback( vlc_object_t *p_this, char const *psz_cmd,
 884                            vlc_value_t oldval, vlc_value_t newval,
 885                            void *p_data )
 886 {
 887     VLC_UNUSED(psz_cmd); VLC_UNUSED(oldval);
 888     demux_t     *p_demux = (demux_t*)p_this;
 889     demux_sys_t *p_sys = p_demux->p_sys;
 890     int         i_tmp = (intptr_t)p_data;
 891
 892     vlc_mutex_lock( &p_sys->csa_lock );
 893     if ( i_tmp )
 894         i_tmp = csa_SetCW( p_this, p_sys->csa, newval.psz_string, true );
 895     else
 896         i_tmp = csa_SetCW( p_this, p_sys->csa, newval.psz_string, false );
 897
 898     vlc_mutex_unlock( &p_sys->csa_lock );
 899     return i_tmp;
 900 }
 901
 902 /*****************************************************************************
 903  * DemuxFile:
 904  *****************************************************************************/
 905 static int DemuxFile( demux_t *p_demux )
 906 {
 907     demux_sys_t *p_sys = p_demux->p_sys;
 908     uint8_t     *p_buffer = p_sys->buffer; /* Put first on sync byte */
 909     int i_diff= 0;
 910     int i_data= 0;
 911     int i_pos = 0;
 912     int i_bufsize = p_sys->i_packet_size * p_sys->i_ts_read;
 913
 914     i_data = stream_Read( p_demux->s, p_sys->buffer, i_bufsize );
 915     if( (i_data <= 0) && (i_data < p_sys->i_packet_size) )
 916     {
 917         msg_Dbg( p_demux, "error reading malformed packets" );
 918         return i_data;
 919     }
 920
 921     /* Test continuity counter */
 922     while( i_pos < i_data )
 923     {
 924         ts_pid_t *p_pid;   /* point to a PID structure */
 925         bool      b_payload; /* indicates a packet with payload */
 926         bool      b_adaptation; /* adaptation field */
 927         int       i_cc  = 0;    /* continuity counter */
 928
 929         if( p_sys->buffer[i_pos] != 0x47 )
 930         {
 931             msg_Warn( p_demux, "lost sync" );
 932             while( vlc_object_alive (p_demux) && (i_pos < i_data) )
 933             {
 934                 i_pos++;
 935                 if( p_sys->buffer[i_pos] == 0x47 )
 936                     break;
 937             }
 938             if( vlc_object_alive (p_demux) )
 939                 msg_Warn( p_demux, "sync found" );
 940         }
 941
 942         /* continuous when (one of this):
 943          * diff == 1
 944          * diff == 0 and payload == 0
 945          * diff == 0 and duplicate packet (playload != 0) <- should we
 946          *   test the content ?
 947          */
 948         i_cc  = p_buffer[i_pos+3]&0x0f;
 949         b_payload = p_buffer[i_pos+3]&0x10;
 950         b_adaptation = p_buffer[i_pos+3]&0x20;
 951
 952         /* Get the PID */
 953         p_pid = &p_sys->pid[ ((p_buffer[i_pos+1]&0x1f)<<8)|p_buffer[i_pos+2] ];
 954
 955         /* Detect discontinuity indicator in adaptation field */
 956         if( b_adaptation && p_buffer[i_pos + 4] > 0 )
 957         {
 958             if( p_buffer[i_pos+5]&0x80 )
 959                 msg_Warn( p_demux, "discontinuity indicator (pid=%d) ", p_pid->i_pid );
 960             if( p_buffer[i_pos+5]&0x40 )
 961                 msg_Warn( p_demux, "random access indicator (pid=%d) ", p_pid->i_pid );
 962         }
 963
 964         i_diff = ( i_cc - p_pid->i_cc )&0x0f;
 965         if( b_payload && i_diff == 1 )
 966         {
 967             p_pid->i_cc = ( p_pid->i_cc + 1 ) & 0xf;
 968         }
 969         else
 970         {
 971             if( p_pid->i_cc == 0xff )
 972             {
 973                 msg_Warn( p_demux, "first packet for pid=%d cc=0x%x",
 974                         p_pid->i_pid, i_cc );
 975                 p_pid->i_cc = i_cc;
 976             }
 977             else if( i_diff != 0 )
 978             {
 979                 /* FIXME what to do when discontinuity_indicator is set ? */
 980                 msg_Warn( p_demux, "transport error detected 0x%x instead of 0x%x",
 981                           i_cc, ( p_pid->i_cc + 1 )&0x0f );
 982
 983                 p_pid->i_cc = i_cc;
 984                 /* Mark transport error in the TS packet. */
 985                 p_buffer[i_pos+1] |= 0x80;
 986             }
 987         }
 988
 989         /* Test if user wants to decrypt it first */
 990         if( p_sys->csa )
 991         {
 992             vlc_mutex_lock( &p_sys->csa_lock );
 993             csa_Decrypt( p_demux->p_sys->csa, &p_buffer[i_pos], p_demux->p_sys->i_csa_pkt_size );
 994             vlc_mutex_unlock( &p_sys->csa_lock );
 995         }
 996
 997         i_pos += p_sys->i_packet_size;
 998     }
 999
1000     /* Then write */
1001     i_data = fwrite( p_sys->buffer, 1, i_data, p_sys->p_file );
1002     if( i_data < 0 )
1003     {
1004         msg_Err( p_demux, "failed to write data" );
1005         return -1;
1006     }
1007 #if 0
1008     msg_Dbg( p_demux, "dumped %d bytes", i_data );
1009 #endif
1010
1011     p_sys->i_write += i_data;
1012     return 1;
1013 }
1014
1015 /*****************************************************************************
1016  * Demux:
1017  *****************************************************************************/
1018 static int Demux( demux_t *p_demux )
1019 {
1020     demux_sys_t *p_sys = p_demux->p_sys;
1021     int          i_pkt;
1022
1023     /* We read at most 100 TS packet or until a frame is completed */
1024     for( i_pkt = 0; i_pkt < p_sys->i_ts_read; i_pkt++ )
1025     {
1026         bool         b_frame = false;
1027         block_t     *p_pkt;
1028         ts_pid_t    *p_pid;
1029
1030         /* Get a new TS packet */
1031         if( !( p_pkt = stream_Block( p_demux->s, p_sys->i_packet_size ) ) )
1032         {
1033             msg_Dbg( p_demux, "eof ?" );
1034             return 0;
1035         }
1036
1037         /* Check sync byte and re-sync if needed */
1038         if( p_pkt->p_buffer[0] != 0x47 )
1039         {
1040             msg_Warn( p_demux, "lost synchro" );
1041             block_Release( p_pkt );
1042
1043             while( vlc_object_alive (p_demux) )
1044             {
1045                 const uint8_t *p_peek;
1046                 int i_peek, i_skip = 0;
1047
1048                 i_peek = stream_Peek( p_demux->s, &p_peek,
1049                                       p_sys->i_packet_size * 10 );
1050                 if( i_peek < p_sys->i_packet_size + 1 )
1051                 {
1052                     msg_Dbg( p_demux, "eof ?" );
1053                     return 0;
1054                 }
1055
1056                 while( i_skip < i_peek - p_sys->i_packet_size )
1057                 {
1058                     if( p_peek[i_skip] == 0x47 &&
1059                         p_peek[i_skip + p_sys->i_packet_size] == 0x47 )
1060                     {
1061                         break;
1062                     }
1063                     i_skip++;
1064                 }
1065
1066                 msg_Dbg( p_demux, "skipping %d bytes of garbage", i_skip );
1067                 stream_Read( p_demux->s, NULL, i_skip );
1068
1069                 if( i_skip < i_peek - p_sys->i_packet_size )
1070                 {
1071                     break;
1072                 }
1073             }
1074
1075             if( !( p_pkt = stream_Block( p_demux->s, p_sys->i_packet_size ) ) )
1076             {
1077                 msg_Dbg( p_demux, "eof ?" );
1078                 return 0;
1079             }
1080         }
1081
1082         if( p_sys->b_start_record )
1083         {
1084             /* Enable recording once synchronized */
1085             stream_Control( p_demux->s, STREAM_SET_RECORD_STATE, true, "ts" );
1086             p_sys->b_start_record = false;
1087         }
1088
1089         if( p_sys->b_udp_out )
1090         {
1091             memcpy( &p_sys->buffer[i_pkt * p_sys->i_packet_size],
1092                     p_pkt->p_buffer, p_sys->i_packet_size );
1093         }
1094
1095         /* Parse the TS packet */
1096         p_pid = &p_sys->pid[PIDGet( p_pkt )];
1097
1098         if( p_pid->b_valid )
1099         {
1100             if( p_pid->psi )
1101             {
1102                 if( p_pid->i_pid == 0 || p_pid->i_pid == 0x11 || p_pid->i_pid == 0x12 )
1103                 {
1104                     dvbpsi_PushPacket( p_pid->psi->handle, p_pkt->p_buffer );
1105                 }
1106                 else
1107                 {
1108                     int i_prg;
1109                     for( i_prg = 0; i_prg < p_pid->psi->i_prg; i_prg++ )
1110                     {
1111                         dvbpsi_PushPacket( p_pid->psi->prg[i_prg]->handle,
1112                                            p_pkt->p_buffer );
1113                     }
1114                 }
1115                 block_Release( p_pkt );
1116             }
1117             else if( !p_sys->b_udp_out )
1118             {
1119                 b_frame = GatherPES( p_demux, p_pid, p_pkt );
1120             }
1121             else
1122             {
1123                 PCRHandle( p_demux, p_pid, p_pkt );
1124                 block_Release( p_pkt );
1125             }
1126         }
1127         else
1128         {
1129             if( !p_pid->b_seen )
1130             {
1131                 msg_Dbg( p_demux, "pid[%d] unknown", p_pid->i_pid );
1132             }
1133             /* We have to handle PCR if present */
1134             PCRHandle( p_demux, p_pid, p_pkt );
1135             block_Release( p_pkt );
1136         }
1137         p_pid->b_seen = true;
1138
1139         if( b_frame )
1140         {
1141             break;
1142         }
1143     }
1144
1145     if( p_sys->b_udp_out )
1146     {
1147         /* Send the complete block */
1148         net_Write( p_demux, p_sys->fd, NULL, p_sys->buffer,
1149                    p_sys->i_ts_read * p_sys->i_packet_size );
1150     }
1151
1152     return 1;
1153 }
1154
1155 /*****************************************************************************
1156  * Control:
1157  *****************************************************************************/
1158 static int DVBEventInformation( demux_t *p_demux, int64_t *pi_time, int64_t *pi_length )
1159 {
1160     demux_sys_t *p_sys = p_demux->p_sys;
1161     if( pi_length )
1162         *pi_length = 0;
1163     if( pi_time )
1164         *pi_time = 0;
1165
1166 #ifdef HAVE_TIME_H
1167     if( p_sys->b_dvb_control && p_sys->i_dvb_length > 0 )
1168     {
1169         /* FIXME we should not use time() but read the date from the tdt */
1170         const time_t t = time( NULL );
1171         if( p_sys->i_dvb_start <= t && t < p_sys->i_dvb_start + p_sys->i_dvb_length )
1172         {
1173             if( pi_length )
1174                 *pi_length = p_sys->i_dvb_length * INT64_C(1000000);
1175             if( pi_time )
1176                 *pi_time   = (t - p_sys->i_dvb_start) * INT64_C(1000000);
1177             return VLC_SUCCESS;
1178         }
1179     }
1180 #endif
1181     return VLC_EGENERIC;
1182 }
1183
1184 static int Control( demux_t *p_demux, int i_query, va_list args )
1185 {
1186     demux_sys_t *p_sys = p_demux->p_sys;
1187     double f, *pf;
1188     bool b_bool, *pb_bool;
1189     int64_t i64;
1190     int64_t *pi64;
1191     int i_int;
1192
1193     if( p_sys->b_file_out )
1194         return demux_vaControlHelper( p_demux->s, 0, -1, 0, 1, i_query, args );
1195
1196     switch( i_query )
1197     {
1198         case DEMUX_GET_POSITION:
1199             pf = (double*) va_arg( args, double* );
1200             i64 = stream_Size( p_demux->s );
1201             if( i64 > 0 )
1202             {
1203                 *pf = (double)stream_Tell( p_demux->s ) / (double)i64;
1204             }
1205             else
1206             {
1207                 int64_t i_time, i_length;
1208                 if( !DVBEventInformation( p_demux, &i_time, &i_length ) && i_length > 0 )
1209                     *pf = (double)i_time/(double)i_length;
1210                 else
1211                     *pf = 0.0;
1212             }
1213             return VLC_SUCCESS;
1214         case DEMUX_SET_POSITION:
1215             f = (double) va_arg( args, double );
1216             i64 = stream_Size( p_demux->s );
1217
1218             if( stream_Seek( p_demux->s, (int64_t)(i64 * f) ) )
1219                 return VLC_EGENERIC;
1220
1221             return VLC_SUCCESS;
1222 #if 0
1223
1224         case DEMUX_GET_TIME:
1225             pi64 = (int64_t*)va_arg( args, int64_t * );
1226             if( p_sys->i_time < 0 )
1227             {
1228                 *pi64 = 0;
1229                 return VLC_EGENERIC;
1230             }
1231             *pi64 = p_sys->i_time;
1232             return VLC_SUCCESS;
1233
1234         case DEMUX_GET_LENGTH:
1235             pi64 = (int64_t*)va_arg( args, int64_t * );
1236             if( p_sys->i_mux_rate > 0 )
1237             {
1238                 *pi64 = INT64_C(1000000) * ( stream_Size( p_demux->s ) / 50 ) /
1239                         p_sys->i_mux_rate;
1240                 return VLC_SUCCESS;
1241             }
1242             *pi64 = 0;
1243             return VLC_EGENERIC;
1244 #else
1245         case DEMUX_GET_TIME:
1246             pi64 = (int64_t*)va_arg( args, int64_t * );
1247             if( DVBEventInformation( p_demux, pi64, NULL ) )
1248                 *pi64 = 0;
1249             return VLC_SUCCESS;
1250
1251         case DEMUX_GET_LENGTH:
1252             pi64 = (int64_t*)va_arg( args, int64_t * );
1253             if( DVBEventInformation( p_demux, NULL, pi64 ) )
1254                 *pi64 = 0;
1255             return VLC_SUCCESS;
1256 #endif
1257         case DEMUX_SET_GROUP:
1258         {
1259             uint16_t i_vpid = 0, i_apid1 = 0, i_apid2 = 0, i_apid3 = 0;
1260             ts_prg_psi_t *p_prg = NULL;
1261             vlc_list_t *p_list;
1262
1263             i_int = (int)va_arg( args, int );
1264             p_list = (vlc_list_t *)va_arg( args, vlc_list_t * );
1265             msg_Dbg( p_demux, "DEMUX_SET_GROUP %d %p", i_int, p_list );
1266
1267             if( p_sys->b_dvb_control && i_int > 0 && i_int != p_sys->i_dvb_program )
1268             {
1269                 int i_pmt_pid = -1;
1270                 int i;
1271
1272                 /* Search pmt to be unselected */
1273                 for( i = 0; i < p_sys->i_pmt; i++ )
1274                 {
1275                     ts_pid_t *pmt = p_sys->pmt[i];
1276                     int i_prg;
1277
1278                     for( i_prg = 0; i_prg < pmt->psi->i_prg; i_prg++ )
1279                     {
1280                         if( pmt->psi->prg[i_prg]->i_number == p_sys->i_dvb_program )
1281                         {
1282                             i_pmt_pid = p_sys->pmt[i]->i_pid;
1283                             break;
1284                         }
1285                     }
1286                     if( i_pmt_pid > 0 ) break;
1287                 }
1288
1289                 if( i_pmt_pid > 0 )
1290                 {
1291                     stream_Control( p_demux->s, STREAM_CONTROL_ACCESS,
1292                                     ACCESS_SET_PRIVATE_ID_STATE, i_pmt_pid,
1293                                     false );
1294                     /* All ES */
1295                     for( i = 2; i < 8192; i++ )
1296                     {
1297                         ts_pid_t *pid = &p_sys->pid[i];
1298                         int i_prg;
1299
1300                         if( !pid->b_valid || pid->psi ) continue;
1301
1302                         for( i_prg = 0; i_prg < pid->p_owner->i_prg; i_prg++ )
1303                         {
1304                             if( pid->p_owner->prg[i_prg]->i_pid_pmt == i_pmt_pid && pid->es->id )
1305                             {
1306                                 /* We only remove es that aren't defined by extra pmt */
1307                                 stream_Control( p_demux->s,
1308                                                 STREAM_CONTROL_ACCESS,
1309                                                 ACCESS_SET_PRIVATE_ID_STATE,
1310                                                 i, false );
1311                                 break;
1312                             }
1313                         }
1314                     }
1315                 }
1316
1317                 /* select new program */
1318                 p_sys->i_dvb_program = i_int;
1319                 i_pmt_pid = -1;
1320                 for( i = 0; i < p_sys->i_pmt; i++ )
1321                 {
1322                     ts_pid_t *pmt = p_sys->pmt[i];
1323                     int i_prg;
1324
1325                     for( i_prg = 0; i_prg < pmt->psi->i_prg; i_prg++ )
1326                     {
1327                         if( pmt->psi->prg[i_prg]->i_number == i_int )
1328                         {
1329                             i_pmt_pid = p_sys->pmt[i]->i_pid;
1330                             p_prg = p_sys->pmt[i]->psi->prg[i_prg];
1331                             break;
1332                         }
1333                     }
1334                     if( i_pmt_pid > 0 ) break;
1335                 }
1336                 if( i_pmt_pid > 0 )
1337                 {
1338                     stream_Control( p_demux->s, STREAM_CONTROL_ACCESS,
1339                                     ACCESS_SET_PRIVATE_ID_STATE, i_pmt_pid,
1340                                     true );
1341                     stream_Control( p_demux->s, STREAM_CONTROL_ACCESS,
1342                                     ACCESS_SET_PRIVATE_ID_STATE, p_prg->i_pid_pcr,
1343                                     true );
1344
1345                     for( i = 2; i < 8192; i++ )
1346                     {
1347                         ts_pid_t *pid = &p_sys->pid[i];
1348                         int i_prg;
1349
1350                         if( !pid->b_valid || pid->psi ) continue;
1351
1352                         for( i_prg = 0; i_prg < pid->p_owner->i_prg; i_prg++ )
1353                         {
1354                             if( pid->p_owner->prg[i_prg]->i_pid_pmt == i_pmt_pid && pid->es->id )
1355                             {
1356                                 if ( pid->es->fmt.i_cat == VIDEO_ES && !i_vpid )
1357                                     i_vpid = i;
1358                                 if ( pid->es->fmt.i_cat == AUDIO_ES && !i_apid1 )
1359                                     i_apid1 = i;
1360                                 else if ( pid->es->fmt.i_cat == AUDIO_ES && !i_apid2 )
1361                                     i_apid2 = i;
1362                                 else if ( pid->es->fmt.i_cat == AUDIO_ES && !i_apid3 )
1363                                     i_apid3 = i;
1364
1365                                 stream_Control( p_demux->s,
1366                                                 STREAM_CONTROL_ACCESS,
1367                                                 ACCESS_SET_PRIVATE_ID_STATE,
1368                                                 i, true );
1369                                 break;
1370                             }
1371                         }
1372                     }
1373                 }
1374             }
1375             else
1376             {
1377                 p_sys->i_dvb_program = -1;
1378                 p_sys->p_programs_list = p_list;
1379             }
1380             return VLC_SUCCESS;
1381         }
1382
1383         case DEMUX_CAN_RECORD:
1384             pb_bool = (bool*)va_arg( args, bool * );
1385             *pb_bool = true;
1386             return VLC_SUCCESS;
1387
1388         case DEMUX_SET_RECORD_STATE:
1389             b_bool = (bool)va_arg( args, int );
1390
1391             if( !b_bool )
1392                 stream_Control( p_demux->s, STREAM_SET_RECORD_STATE, false );
1393             p_sys->b_start_record = b_bool;
1394             return VLC_SUCCESS;
1395
1396         case DEMUX_GET_FPS:
1397         case DEMUX_SET_TIME:
1398         default:
1399             return VLC_EGENERIC;
1400     }
1401 }
1402
1403 /*****************************************************************************
1404  *
1405  *****************************************************************************/
1406 static int UserPmt( demux_t *p_demux, const char *psz_fmt )
1407 {
1408     demux_sys_t *p_sys = p_demux->p_sys;
1409     char *psz_dup = strdup( psz_fmt );
1410     char *psz = psz_dup;
1411     int  i_pid;
1412     int  i_number;
1413
1414     if( !psz_dup )
1415         return VLC_ENOMEM;
1416
1417     /* Parse PID */
1418     i_pid = strtol( psz, &psz, 0 );
1419     if( i_pid < 2 || i_pid >= 8192 )
1420         goto error;
1421
1422     /* Parse optional program number */
1423     i_number = 0;
1424     if( *psz == ':' )
1425         i_number = strtol( &psz[1], &psz, 0 );
1426
1427     /* */
1428     ts_pid_t *pmt = &p_sys->pid[i_pid];
1429     ts_prg_psi_t *prg;
1430
1431     msg_Dbg( p_demux, "user pmt specified (pid=%d,number=%d)", i_pid, i_number );
1432     PIDInit( pmt, true, NULL );
1433
1434     /* Dummy PMT */
1435     prg = malloc( sizeof( ts_prg_psi_t ) );
1436     if( !prg )
1437         goto error;
1438
1439     memset( prg, 0, sizeof( ts_prg_psi_t ) );
1440     prg->i_pid_pcr  = -1;
1441     prg->i_pid_pmt  = -1;
1442     prg->i_version  = -1;
1443     prg->i_number   = i_number != 0 ? i_number : TS_USER_PMT_NUMBER;
1444     prg->handle     = dvbpsi_AttachPMT( i_number != TS_USER_PMT_NUMBER ? i_number : 1, (dvbpsi_pmt_callback)PMTCallBack, p_demux );
1445     TAB_APPEND( pmt->psi->i_prg, pmt->psi->prg, prg );
1446
1447     psz = strchr( psz, '=' );
1448     if( psz )
1449         psz++;
1450     while( psz && *psz )
1451     {
1452         char *psz_next = strchr( psz, ',' );
1453         int i_pid;
1454
1455         if( psz_next )
1456             *psz_next++ = '\0';
1457
1458         i_pid = strtol( psz, &psz, 0 );
1459         if( *psz != ':' || i_pid < 2 || i_pid >= 8192 )
1460             goto next;
1461
1462         char *psz_opt = &psz[1];
1463         if( !strcmp( psz_opt, "pcr" ) )
1464         {
1465             prg->i_pid_pcr = i_pid;
1466         }
1467         else if( !p_sys->pid[i_pid].b_valid )
1468         {
1469             ts_pid_t *pid = &p_sys->pid[i_pid];
1470
1471             char *psz_arg = strchr( psz_opt, '=' );
1472             if( psz_arg )
1473                 *psz_arg++ = '\0';
1474
1475             PIDInit( pid, false, pmt->psi);
1476             if( prg->i_pid_pcr <= 0 )
1477                 prg->i_pid_pcr = i_pid;
1478
1479             if( psz_arg && strlen( psz_arg ) == 4 )
1480             {
1481                 const vlc_fourcc_t i_codec = VLC_FOURCC( psz_arg[0], psz_arg[1],
1482                                                          psz_arg[2], psz_arg[3] );
1483                 int i_cat = UNKNOWN_ES;
1484                 es_format_t *fmt = &pid->es->fmt;
1485
1486                 if( !strcmp( psz_opt, "video" ) )
1487                     i_cat = VIDEO_ES;
1488                 else if( !strcmp( psz_opt, "audio" ) )
1489                     i_cat = AUDIO_ES;
1490                 else if( !strcmp( psz_opt, "spu" ) )
1491                     i_cat = SPU_ES;
1492
1493                 es_format_Init( fmt, i_cat, i_codec );
1494                 fmt->b_packetized = false;
1495             }
1496             else
1497             {
1498                 const int i_stream_type = strtol( psz_opt, NULL, 0 );
1499                 PIDFillFormat( pid, i_stream_type );
1500             }
1501             pid->es->fmt.i_group = i_number;
1502             if( p_sys->b_es_id_pid )
1503                 pid->es->fmt.i_id = i_pid;
1504
1505             if( pid->es->fmt.i_cat != UNKNOWN_ES )
1506             {
1507                 msg_Dbg( p_demux, "  * es pid=%d fcc=%4.4s", i_pid,
1508                          (char*)&pid->es->fmt.i_codec );
1509                 pid->es->id = es_out_Add( p_demux->out,
1510                                           &pid->es->fmt );
1511             }
1512         }
1513
1514     next:
1515         psz = psz_next;
1516     }
1517
1518     p_sys->b_user_pmt = true;
1519     TAB_APPEND( p_sys->i_pmt, p_sys->pmt, pmt );
1520     free( psz_dup );
1521     return VLC_SUCCESS;
1522
1523 error:
1524     free( psz_dup );
1525     return VLC_EGENERIC;
1526 }
1527
1528 static void PIDInit( ts_pid_t *pid, bool b_psi, ts_psi_t *p_owner )
1529 {
1530     bool b_old_valid = pid->b_valid;
1531
1532     pid->b_valid    = true;
1533     pid->i_cc       = 0xff;
1534     pid->p_owner    = p_owner;
1535     pid->i_owner_number = 0;
1536
1537     TAB_INIT( pid->i_extra_es, pid->extra_es );
1538
1539     if( b_psi )
1540     {
1541         pid->es  = NULL;
1542
1543         if( !b_old_valid )
1544         {
1545             pid->psi = malloc( sizeof( ts_psi_t ) );
1546             if( pid->psi )
1547             {
1548                 pid->psi->handle = NULL;
1549                 TAB_INIT( pid->psi->i_prg, pid->psi->prg );
1550             }
1551         }
1552         assert( pid->psi );
1553
1554         pid->psi->i_pat_version  = -1;
1555         pid->psi->i_sdt_version  = -1;
1556         if( p_owner )
1557         {
1558             ts_prg_psi_t *prg = malloc( sizeof( ts_prg_psi_t ) );
1559             if( prg )
1560             {
1561                 /* PMT */
1562                 prg->i_version  = -1;
1563                 prg->i_number   = -1;
1564                 prg->i_pid_pcr  = -1;
1565                 prg->i_pid_pmt  = -1;
1566                 prg->iod        = NULL;
1567                 prg->handle     = NULL;
1568
1569                 TAB_APPEND( pid->psi->i_prg, pid->psi->prg, prg );
1570             }
1571         }
1572     }
1573     else
1574     {
1575         pid->psi = NULL;
1576         pid->es  = malloc( sizeof( ts_es_t ) );
1577         if( pid->es )
1578         {
1579             es_format_Init( &pid->es->fmt, UNKNOWN_ES, 0 );
1580             pid->es->id      = NULL;
1581             pid->es->p_pes   = NULL;
1582             pid->es->i_pes_size= 0;
1583             pid->es->i_pes_gathered= 0;
1584             pid->es->pp_last = &pid->es->p_pes;
1585             pid->es->p_mpeg4desc = NULL;
1586             pid->es->b_gather = false;
1587         }
1588     }
1589 }
1590
1591 static void PIDClean( es_out_t *out, ts_pid_t *pid )
1592 {
1593     if( pid->psi )
1594     {
1595         int i;
1596
1597         if( pid->psi->handle ) dvbpsi_DetachPMT( pid->psi->handle );
1598         for( i = 0; i < pid->psi->i_prg; i++ )
1599         {
1600             if( pid->psi->prg[i]->iod )
1601                 IODFree( pid->psi->prg[i]->iod );
1602             if( pid->psi->prg[i]->handle )
1603                 dvbpsi_DetachPMT( pid->psi->prg[i]->handle );
1604             free( pid->psi->prg[i] );
1605         }
1606         free( pid->psi->prg );
1607         free( pid->psi );
1608     }
1609     else
1610     {
1611         int i;
1612
1613         if( pid->es->id )
1614             es_out_Del( out, pid->es->id );
1615
1616         if( pid->es->p_pes )
1617             block_ChainRelease( pid->es->p_pes );
1618
1619         es_format_Clean( &pid->es->fmt );
1620
1621         free( pid->es );
1622
1623         for( i = 0; i < pid->i_extra_es; i++ )
1624         {
1625             if( pid->extra_es[i]->id )
1626                 es_out_Del( out, pid->extra_es[i]->id );
1627
1628             if( pid->extra_es[i]->p_pes )
1629                 block_ChainRelease( pid->extra_es[i]->p_pes );
1630
1631             es_format_Clean( &pid->extra_es[i]->fmt );
1632
1633             free( pid->extra_es[i] );
1634         }
1635         if( pid->i_extra_es ) free( pid->extra_es );
1636     }
1637
1638     pid->b_valid = false;
1639 }
1640
1641 /****************************************************************************
1642  * gathering stuff
1643  ****************************************************************************/
1644 static void ParsePES( demux_t *p_demux, ts_pid_t *pid )
1645 {
1646     block_t *p_pes = pid->es->p_pes;
1647     uint8_t header[34];
1648     int     i_pes_size = 0;
1649     int     i_skip = 0;
1650     mtime_t i_dts = -1;
1651     mtime_t i_pts = -1;
1652     mtime_t i_length = 0;
1653     int i_max;
1654
1655     /* remove the pes from pid */
1656     pid->es->p_pes = NULL;
1657     pid->es->i_pes_size= 0;
1658     pid->es->i_pes_gathered= 0;
1659     pid->es->pp_last = &pid->es->p_pes;
1660
1661     /* FIXME find real max size */
1662     i_max = block_ChainExtract( p_pes, header, 34 );
1663
1664
1665     if( header[0] != 0 || header[1] != 0 || header[2] != 1 )
1666     {
1667         if( !p_demux->p_sys->b_silent )
1668             msg_Warn( p_demux, "invalid header [0x%x:%x:%x:%x] (pid: %d)",
1669                       header[0], header[1],header[2],header[3], pid->i_pid );
1670         block_ChainRelease( p_pes );
1671         return;
1672     }
1673
1674     /* TODO check size */
1675     switch( header[3] )
1676     {
1677         case 0xBC:  /* Program stream map */
1678         case 0xBE:  /* Padding */
1679         case 0xBF:  /* Private stream 2 */
1680         case 0xF0:  /* ECM */
1681         case 0xF1:  /* EMM */
1682         case 0xFF:  /* Program stream directory */
1683         case 0xF2:  /* DSMCC stream */
1684         case 0xF8:  /* ITU-T H.222.1 type E stream */
1685             i_skip = 6;
1686             break;
1687         default:
1688             if( ( header[6]&0xC0 ) == 0x80 )
1689             {
1690                 /* mpeg2 PES */
1691                 i_skip = header[8] + 9;
1692
1693                 if( header[7]&0x80 )    /* has pts */
1694                 {
1695                     i_pts = ((mtime_t)(header[ 9]&0x0e ) << 29)|
1696                              (mtime_t)(header[10] << 22)|
1697                             ((mtime_t)(header[11]&0xfe) << 14)|
1698                              (mtime_t)(header[12] << 7)|
1699                              (mtime_t)(header[13] >> 1);
1700
1701                     if( header[7]&0x40 )    /* has dts */
1702                     {
1703                          i_dts = ((mtime_t)(header[14]&0x0e ) << 29)|
1704                                  (mtime_t)(header[15] << 22)|
1705                                 ((mtime_t)(header[16]&0xfe) << 14)|
1706                                  (mtime_t)(header[17] << 7)|
1707                                  (mtime_t)(header[18] >> 1);
1708                     }
1709                 }
1710             }
1711             else
1712             {
1713                 i_skip = 6;
1714                 while( i_skip < 23 && header[i_skip] == 0xff )
1715                 {
1716                     i_skip++;
1717                 }
1718                 if( i_skip == 23 )
1719                 {
1720                     msg_Err( p_demux, "too much MPEG-1 stuffing" );
1721                     block_ChainRelease( p_pes );
1722                     return;
1723                 }
1724                 if( ( header[i_skip] & 0xC0 ) == 0x40 )
1725                 {
1726                     i_skip += 2;
1727                 }
1728
1729                 if(  header[i_skip]&0x20 )
1730                 {
1731                      i_pts = ((mtime_t)(header[i_skip]&0x0e ) << 29)|
1732                               (mtime_t)(header[i_skip+1] << 22)|
1733                              ((mtime_t)(header[i_skip+2]&0xfe) << 14)|
1734                               (mtime_t)(header[i_skip+3] << 7)|
1735                               (mtime_t)(header[i_skip+4] >> 1);
1736
1737                     if( header[i_skip]&0x10 )    /* has dts */
1738                     {
1739                          i_dts = ((mtime_t)(header[i_skip+5]&0x0e ) << 29)|
1740                                   (mtime_t)(header[i_skip+6] << 22)|
1741                                  ((mtime_t)(header[i_skip+7]&0xfe) << 14)|
1742                                   (mtime_t)(header[i_skip+8] << 7)|
1743                                   (mtime_t)(header[i_skip+9] >> 1);
1744                          i_skip += 10;
1745                     }
1746                     else
1747                     {
1748                         i_skip += 5;
1749                     }
1750                 }
1751                 else
1752                 {
1753                     i_skip += 1;
1754                 }
1755             }
1756             break;
1757     }
1758
1759     if( pid->es->fmt.i_codec == VLC_FOURCC( 'a', '5', '2', 'b' ) ||
1760         pid->es->fmt.i_codec == VLC_FOURCC( 'd', 't', 's', 'b' ) )
1761     {
1762         i_skip += 4;
1763     }
1764     else if( pid->es->fmt.i_codec == VLC_FOURCC( 'l', 'p', 'c', 'b' ) ||
1765              pid->es->fmt.i_codec == VLC_FOURCC( 's', 'p', 'u', 'b' ) ||
1766              pid->es->fmt.i_codec == VLC_FOURCC( 's', 'd', 'd', 'b' ) )
1767     {
1768         i_skip += 1;
1769     }
1770     else if( pid->es->fmt.i_codec == VLC_FOURCC( 's', 'u', 'b', 't' ) &&
1771              pid->es->p_mpeg4desc )
1772     {
1773         decoder_config_descriptor_t *dcd = &pid->es->p_mpeg4desc->dec_descr;
1774
1775         if( dcd->i_decoder_specific_info_len > 2 &&
1776             dcd->p_decoder_specific_info[0] == 0x10 &&
1777             ( dcd->p_decoder_specific_info[1]&0x10 ) )
1778         {
1779             /* display length */
1780             if( p_pes->i_buffer + 2 <= i_skip )
1781             {
1782                 i_length = GetWBE( &p_pes->p_buffer[i_skip] );
1783             }
1784
1785             i_skip += 2;
1786         }
1787         if( p_pes->i_buffer + 2 <= i_skip )
1788         {
1789             i_pes_size = GetWBE( &p_pes->p_buffer[i_skip] );
1790         }
1791         /* */
1792         i_skip += 2;
1793     }
1794 #ifdef ZVBI_COMPILED
1795     else if( pid->es->fmt.i_codec == VLC_FOURCC( 't', 'e', 'l', 'x' ) )
1796         i_skip = 0; /*hack for zvbi support */
1797 #endif
1798     /* skip header */
1799     while( p_pes && i_skip > 0 )
1800     {
1801         if( p_pes->i_buffer <= i_skip )
1802         {
1803             block_t *p_next = p_pes->p_next;
1804
1805             i_skip -= p_pes->i_buffer;
1806             block_Release( p_pes );
1807             p_pes = p_next;
1808         }
1809         else
1810         {
1811             p_pes->i_buffer -= i_skip;
1812             p_pes->p_buffer += i_skip;
1813             break;
1814         }
1815     }
1816
1817     /* ISO/IEC 13818-1 2.7.5: if no pts and no dts, then dts == pts */
1818     if( i_pts >= 0 && i_dts < 0 )
1819         i_dts = i_pts;
1820
1821     if( p_pes )
1822     {
1823         block_t *p_block;
1824         int i;
1825
1826         if( i_dts >= 0 )
1827         {
1828             p_pes->i_dts = i_dts * 100 / 9;
1829         }
1830         if( i_pts >= 0 )
1831         {
1832             p_pes->i_pts = i_pts * 100 / 9;
1833         }
1834         p_pes->i_length = i_length * 100 / 9;
1835
1836         p_block = block_ChainGather( p_pes );
1837         if( pid->es->fmt.i_codec == VLC_FOURCC( 's', 'u', 'b', 't' ) )
1838         {
1839             if( i_pes_size > 0 && p_block->i_buffer > i_pes_size )
1840             {
1841                 p_block->i_buffer = i_pes_size;
1842             }
1843             /* Append a \0 */
1844             p_block = block_Realloc( p_block, 0, p_block->i_buffer + 1 );
1845             p_block->p_buffer[p_block->i_buffer -1] = '\0';
1846         }
1847
1848         for( i = 0; i < pid->i_extra_es; i++ )
1849         {
1850             es_out_Send( p_demux->out, pid->extra_es[i]->id,
1851                          block_Duplicate( p_block ) );
1852         }
1853
1854         es_out_Send( p_demux->out, pid->es->id, p_block );
1855     }
1856     else
1857     {
1858         msg_Warn( p_demux, "empty pes" );
1859     }
1860 }
1861
1862 static void PCRHandle( demux_t *p_demux, ts_pid_t *pid, block_t *p_bk )
1863 {
1864     demux_sys_t   *p_sys = p_demux->p_sys;
1865     const uint8_t *p = p_bk->p_buffer;
1866
1867     if( ( p[3]&0x20 ) && /* adaptation */
1868         ( p[5]&0x10 ) &&
1869         ( p[4] >= 7 ) )
1870     {
1871         int i;
1872         mtime_t i_pcr;  /* 33 bits */
1873
1874         i_pcr = ( (mtime_t)p[6] << 25 ) |
1875                 ( (mtime_t)p[7] << 17 ) |
1876                 ( (mtime_t)p[8] << 9 ) |
1877                 ( (mtime_t)p[9] << 1 ) |
1878                 ( (mtime_t)p[10] >> 7 );
1879
1880         /* Search program and set the PCR */
1881         for( i = 0; i < p_sys->i_pmt; i++ )
1882         {
1883             int i_prg;
1884             for( i_prg = 0; i_prg < p_sys->pmt[i]->psi->i_prg; i_prg++ )
1885             {
1886                 if( pid->i_pid == p_sys->pmt[i]->psi->prg[i_prg]->i_pid_pcr )
1887                 {
1888                     es_out_Control( p_demux->out, ES_OUT_SET_GROUP_PCR,
1889                                     (int)p_sys->pmt[i]->psi->prg[i_prg]->i_number,
1890                                     (int64_t)(i_pcr * 100 / 9) );
1891                 }
1892             }
1893         }
1894     }
1895 }
1896
1897 static bool GatherPES( demux_t *p_demux, ts_pid_t *pid, block_t *p_bk )
1898 {
1899     const uint8_t *p = p_bk->p_buffer;
1900     const bool b_unit_start = p[1]&0x40;
1901     const bool b_adaptation = p[3]&0x20;
1902     const bool b_payload    = p[3]&0x10;
1903     const int  i_cc         = p[3]&0x0f;   /* continuity counter */
1904     bool       b_discontinuity = false;/* discontinuity */
1905
1906     /* transport_scrambling_control is ignored */
1907     int         i_skip = 0;
1908     bool  i_ret  = false;
1909     int         i_diff;
1910
1911 #if 0
1912     msg_Dbg( p_demux, "pid=%d unit_start=%d adaptation=%d payload=%d "
1913              "cc=0x%x", pid->i_pid, b_unit_start, b_adaptation,
1914              b_payload, i_cc );
1915 #endif
1916
1917     /* For now, ignore additional error correction
1918      * TODO: handle Reed-Solomon 204,188 error correction */
1919     p_bk->i_buffer = TS_PACKET_SIZE_188;
1920
1921     if( p[1]&0x80 )
1922     {
1923         msg_Dbg( p_demux, "transport_error_indicator set (pid=%d)",
1924                  pid->i_pid );
1925         if( pid->es->p_pes ) //&& pid->es->fmt.i_cat == VIDEO_ES )
1926             pid->es->p_pes->i_flags |= BLOCK_FLAG_CORRUPTED;
1927     }
1928
1929     if( p_demux->p_sys->csa )
1930     {
1931         vlc_mutex_lock( &p_demux->p_sys->csa_lock );
1932         csa_Decrypt( p_demux->p_sys->csa, p_bk->p_buffer, p_demux->p_sys->i_csa_pkt_size );
1933         vlc_mutex_unlock( &p_demux->p_sys->csa_lock );
1934     }
1935
1936     if( !b_adaptation )
1937     {
1938         /* We don't have any adaptation_field, so payload starts
1939          * immediately after the 4 byte TS header */
1940         i_skip = 4;
1941     }
1942     else
1943     {
1944         /* p[4] is adaptation_field_length minus one */
1945         i_skip = 5 + p[4];
1946         if( p[4] > 0 )
1947         {
1948             /* discontinuity indicator found in stream */
1949             b_discontinuity = (p[5]&0x80) ? true : false;
1950             if( b_discontinuity && pid->es->p_pes )
1951             {
1952                 msg_Warn( p_demux, "discontinuity indicator (pid=%d) ",
1953                             pid->i_pid );
1954                 /* pid->es->p_pes->i_flags |= BLOCK_FLAG_DISCONTINUITY; */
1955             }
1956 #if 0
1957             if( p[5]&0x40 )
1958                 msg_Dbg( p_demux, "random access indicator (pid=%d) ", pid->i_pid );
1959 #endif
1960         }
1961     }
1962
1963     /* Test continuity counter */
1964     /* continuous when (one of this):
1965         * diff == 1
1966         * diff == 0 and payload == 0
1967         * diff == 0 and duplicate packet (playload != 0) <- should we
1968         *   test the content ?
1969      */
1970     i_diff = ( i_cc - pid->i_cc )&0x0f;
1971     if( b_payload && i_diff == 1 )
1972     {
1973         pid->i_cc = ( pid->i_cc + 1 ) & 0xf;
1974     }
1975     else
1976     {
1977         if( pid->i_cc == 0xff )
1978         {
1979             msg_Warn( p_demux, "first packet for pid=%d cc=0x%x",
1980                       pid->i_pid, i_cc );
1981             pid->i_cc = i_cc;
1982         }
1983         else if( i_diff != 0 && !b_discontinuity )
1984         {
1985             msg_Warn( p_demux, "discontinuity received 0x%x instead of 0x%x (pid=%d)",
1986                       i_cc, ( pid->i_cc + 1 )&0x0f, pid->i_pid );
1987
1988             pid->i_cc = i_cc;
1989             if( pid->es->p_pes && pid->es->fmt.i_cat != VIDEO_ES )
1990             {
1991                 /* Small video artifacts are usually better then
1992                  * dropping full frames */
1993                 pid->es->p_pes->i_flags |= BLOCK_FLAG_CORRUPTED;
1994             }
1995         }
1996     }
1997
1998     PCRHandle( p_demux, pid, p_bk );
1999
2000     if( i_skip >= 188 || pid->es->id == NULL || p_demux->p_sys->b_udp_out )
2001     {
2002         block_Release( p_bk );
2003         return i_ret;
2004     }
2005
2006     /* We have to gather it */
2007     p_bk->p_buffer += i_skip;
2008     p_bk->i_buffer -= i_skip;
2009
2010     if( b_unit_start )
2011     {
2012         if( pid->es->p_pes )
2013         {
2014             ParsePES( p_demux, pid );
2015             i_ret = true;
2016         }
2017
2018         block_ChainLastAppend( &pid->es->pp_last, p_bk );
2019         if( p_bk->i_buffer > 6 )
2020         {
2021             pid->es->i_pes_size = GetWBE( &p_bk->p_buffer[4] );
2022             if( pid->es->i_pes_size > 0 )
2023             {
2024                 pid->es->i_pes_size += 6;
2025             }
2026         }
2027         pid->es->i_pes_gathered += p_bk->i_buffer;
2028         if( pid->es->i_pes_size > 0 &&
2029             pid->es->i_pes_gathered >= pid->es->i_pes_size )
2030         {
2031             ParsePES( p_demux, pid );
2032             i_ret = true;
2033         }
2034     }
2035     else
2036     {
2037         if( pid->es->p_pes == NULL )
2038         {
2039             /* msg_Dbg( p_demux, "broken packet" ); */
2040             block_Release( p_bk );
2041         }
2042         else
2043         {
2044             block_ChainLastAppend( &pid->es->pp_last, p_bk );
2045             pid->es->i_pes_gathered += p_bk->i_buffer;
2046             if( pid->es->i_pes_size > 0 &&
2047                 pid->es->i_pes_gathered >= pid->es->i_pes_size )
2048             {
2049                 ParsePES( p_demux, pid );
2050                 i_ret = true;
2051             }
2052         }
2053     }
2054
2055     return i_ret;
2056 }
2057
2058 static int PIDFillFormat( ts_pid_t *pid, int i_stream_type )
2059 {
2060     es_format_t *fmt = &pid->es->fmt;
2061
2062     switch( i_stream_type )
2063     {
2064         case 0x01:  /* MPEG-1 video */
2065         case 0x02:  /* MPEG-2 video */
2066         case 0x80:  /* MPEG-2 MOTO video */
2067             es_format_Init( fmt, VIDEO_ES, VLC_FOURCC( 'm', 'p', 'g', 'v' ) );
2068             break;
2069         case 0x03:  /* MPEG-1 audio */
2070         case 0x04:  /* MPEG-2 audio */
2071             es_format_Init( fmt, AUDIO_ES, VLC_FOURCC( 'm', 'p', 'g', 'a' ) );
2072             break;
2073         case 0x11:  /* MPEG4 (audio) */
2074         case 0x0f:  /* ISO/IEC 13818-7 Audio with ADTS transport syntax */
2075             es_format_Init( fmt, AUDIO_ES, VLC_FOURCC( 'm', 'p', '4', 'a' ) );
2076             break;
2077         case 0x10:  /* MPEG4 (video) */
2078             es_format_Init( fmt, VIDEO_ES, VLC_FOURCC( 'm', 'p', '4', 'v' ) );
2079             pid->es->b_gather = true;
2080             break;
2081         case 0x1B:  /* H264 <- check transport syntax/needed descriptor */
2082             es_format_Init( fmt, VIDEO_ES, VLC_FOURCC( 'h', '2', '6', '4' ) );
2083             break;
2084
2085         case 0x81:  /* A52 (audio) */
2086             es_format_Init( fmt, AUDIO_ES, VLC_FOURCC( 'a', '5', '2', ' ' ) );
2087             break;
2088         case 0x82:  /* DVD_SPU (sub) */
2089             es_format_Init( fmt, SPU_ES, VLC_FOURCC( 's', 'p', 'u', ' ' ) );
2090             break;
2091         case 0x83:  /* LPCM (audio) */
2092             es_format_Init( fmt, AUDIO_ES, VLC_FOURCC( 'l', 'p', 'c', 'm' ) );
2093             break;
2094         case 0x84:  /* SDDS (audio) */
2095             es_format_Init( fmt, AUDIO_ES, VLC_FOURCC( 's', 'd', 'd', 's' ) );
2096             break;
2097         case 0x85:  /* DTS (audio) */
2098             es_format_Init( fmt, AUDIO_ES, VLC_FOURCC( 'd', 't', 's', ' ' ) );
2099             break;
2100
2101         case 0x91:  /* A52 vls (audio) */
2102             es_format_Init( fmt, AUDIO_ES, VLC_FOURCC( 'a', '5', '2', 'b' ) );
2103             break;
2104         case 0x92:  /* DVD_SPU vls (sub) */
2105             es_format_Init( fmt, SPU_ES, VLC_FOURCC( 's', 'p', 'u', 'b' ) );
2106             break;
2107
2108         case 0x94:  /* SDDS (audio) */
2109             es_format_Init( fmt, AUDIO_ES, VLC_FOURCC( 's', 'd', 'd', 'b' ) );
2110             break;
2111
2112         case 0xa0:  /* MSCODEC vlc (video) (fixed later) */
2113             es_format_Init( fmt, UNKNOWN_ES, 0 );
2114             pid->es->b_gather = true;
2115             break;
2116
2117         case 0x06:  /* PES_PRIVATE  (fixed later) */
2118         case 0x12:  /* MPEG-4 generic (sub/scene/...) (fixed later) */
2119         case 0xEA:  /* Privately managed ES (VC-1) (fixed later */
2120         default:
2121             es_format_Init( fmt, UNKNOWN_ES, 0 );
2122             break;
2123     }
2124
2125     /* PES packets usually contain truncated frames */
2126     fmt->b_packetized = false;
2127
2128     return fmt->i_cat == UNKNOWN_ES ? VLC_EGENERIC : VLC_SUCCESS ;
2129 }
2130
2131 /*****************************************************************************
2132  * MP4 specific functions (IOD parser)
2133  *****************************************************************************/
2134 static int  IODDescriptorLength( int *pi_data, uint8_t **pp_data )
2135 {
2136     unsigned int i_b;
2137     unsigned int i_len = 0;
2138     do
2139     {
2140         i_b = **pp_data;
2141         (*pp_data)++;
2142         (*pi_data)--;
2143         i_len = ( i_len << 7 ) + ( i_b&0x7f );
2144
2145     } while( i_b&0x80 );
2146
2147     return( i_len );
2148 }
2149
2150 static int IODGetByte( int *pi_data, uint8_t **pp_data )
2151 {
2152     if( *pi_data > 0 )
2153     {
2154         const int i_b = **pp_data;
2155         (*pp_data)++;
2156         (*pi_data)--;
2157         return( i_b );
2158     }
2159     return( 0 );
2160 }
2161
2162 static int IODGetWord( int *pi_data, uint8_t **pp_data )
2163 {
2164     const int i1 = IODGetByte( pi_data, pp_data );
2165     const int i2 = IODGetByte( pi_data, pp_data );
2166     return( ( i1 << 8 ) | i2 );
2167 }
2168
2169 static int IODGet3Bytes( int *pi_data, uint8_t **pp_data )
2170 {
2171     const int i1 = IODGetByte( pi_data, pp_data );
2172     const int i2 = IODGetByte( pi_data, pp_data );
2173     const int i3 = IODGetByte( pi_data, pp_data );
2174
2175     return( ( i1 << 16 ) | ( i2 << 8) | i3 );
2176 }
2177
2178 static uint32_t IODGetDWord( int *pi_data, uint8_t **pp_data )
2179 {
2180     const uint32_t i1 = IODGetWord( pi_data, pp_data );
2181     const uint32_t i2 = IODGetWord( pi_data, pp_data );
2182     return( ( i1 << 16 ) | i2 );
2183 }
2184
2185 static char* IODGetURL( int *pi_data, uint8_t **pp_data )
2186 {
2187     char *url;
2188     int i_url_len, i;
2189
2190     i_url_len = IODGetByte( pi_data, pp_data );
2191     url = malloc( i_url_len + 1 );
2192     if( !url ) return NULL;
2193     for( i = 0; i < i_url_len; i++ )
2194     {
2195         url[i] = IODGetByte( pi_data, pp_data );
2196     }
2197     url[i_url_len] = '\0';
2198     return( url );
2199 }
2200
2201 static iod_descriptor_t *IODNew( int i_data, uint8_t *p_data )
2202 {
2203     iod_descriptor_t *p_iod;
2204     int i;
2205     int i_es_index;
2206     uint8_t i_flags, i_iod_tag, byte1, byte2, byte3;
2207     bool  b_url;
2208     int   i_iod_length;
2209
2210     p_iod = malloc( sizeof( iod_descriptor_t ) );
2211     if( !p_iod ) return NULL;
2212     memset( p_iod, 0, sizeof( iod_descriptor_t ) );
2213
2214 #ifdef TS_DEBUG
2215     fprintf( stderr, "\n************ IOD ************" );
2216 #endif
2217     for( i = 0; i < 255; i++ )
2218     {
2219         p_iod->es_descr[i].b_ok = 0;
2220     }
2221     i_es_index = 0;
2222
2223     if( i_data < 3 )
2224     {
2225         return p_iod;
2226     }
2227
2228     byte1 = IODGetByte( &i_data, &p_data );
2229     byte2 = IODGetByte( &i_data, &p_data );
2230     byte3 = IODGetByte( &i_data, &p_data );
2231     if( byte2 == 0x02 ) //old vlc's buggy implementation of the IOD_descriptor
2232     {
2233         p_iod->i_iod_label_scope = 0x11;
2234         p_iod->i_iod_label = byte1;
2235         i_iod_tag = byte2;
2236     }
2237     else  //correct implementation of the IOD_descriptor
2238     {
2239         p_iod->i_iod_label_scope = byte1;
2240         p_iod->i_iod_label = byte2;
2241         i_iod_tag = byte3;
2242     }
2243 #ifdef TS_DEBUG
2244     fprintf( stderr, "\n* iod_label:%d", p_iod->i_iod_label );
2245     fprintf( stderr, "\n* ===========" );
2246     fprintf( stderr, "\n* tag:0x%x", i_iod_tag );
2247 #endif
2248     if( i_iod_tag != 0x02 )
2249     {
2250 #ifdef TS_DEBUG
2251         fprintf( stderr, "\n ERR: tag %02x != 0x02", i_iod_tag );
2252 #endif
2253         return p_iod;
2254     }
2255
2256     i_iod_length = IODDescriptorLength( &i_data, &p_data );
2257 #ifdef TS_DEBUG
2258     fprintf( stderr, "\n* length:%d", i_iod_length );
2259 #endif
2260     if( i_iod_length > i_data )
2261     {
2262         i_iod_length = i_data;
2263     }
2264
2265     p_iod->i_od_id = ( IODGetByte( &i_data, &p_data ) << 2 );
2266     i_flags = IODGetByte( &i_data, &p_data );
2267     p_iod->i_od_id |= i_flags >> 6;
2268     b_url = ( i_flags >> 5  )&0x01;
2269 #ifdef TS_DEBUG
2270     fprintf( stderr, "\n* od_id:%d", p_iod->i_od_id );
2271     fprintf( stderr, "\n* url flag:%d", b_url );
2272     fprintf( stderr, "\n* includeInlineProfileLevel flag:%d", ( i_flags >> 4 )&0x01 );
2273 #endif
2274     if( b_url )
2275     {
2276         p_iod->psz_url = IODGetURL( &i_data, &p_data );
2277 #ifdef TS_DEBUG
2278         fprintf( stderr, "\n* url string:%s", p_iod->psz_url );
2279         fprintf( stderr, "\n*****************************\n" );
2280 #endif
2281         return p_iod;
2282     }
2283     else
2284     {
2285         p_iod->psz_url = NULL;
2286     }
2287
2288     p_iod->i_ODProfileLevelIndication = IODGetByte( &i_data, &p_data );
2289     p_iod->i_sceneProfileLevelIndication = IODGetByte( &i_data, &p_data );
2290     p_iod->i_audioProfileLevelIndication = IODGetByte( &i_data, &p_data );
2291     p_iod->i_visualProfileLevelIndication = IODGetByte( &i_data, &p_data );
2292     p_iod->i_graphicsProfileLevelIndication = IODGetByte( &i_data, &p_data );
2293 #ifdef TS_DEBUG
2294     fprintf( stderr, "\n* ODProfileLevelIndication:%d", p_iod->i_ODProfileLevelIndication );
2295     fprintf( stderr, "\n* sceneProfileLevelIndication:%d", p_iod->i_sceneProfileLevelIndication );
2296     fprintf( stderr, "\n* audioProfileLevelIndication:%d", p_iod->i_audioProfileLevelIndication );
2297     fprintf( stderr, "\n* visualProfileLevelIndication:%d", p_iod->i_visualProfileLevelIndication );
2298     fprintf( stderr, "\n* graphicsProfileLevelIndication:%d", p_iod->i_graphicsProfileLevelIndication );
2299 #endif
2300
2301     while( i_data > 0 && i_es_index < 255)
2302     {
2303         int i_tag, i_length;
2304         int     i_data_sav;
2305         uint8_t *p_data_sav;
2306
2307         i_tag = IODGetByte( &i_data, &p_data );
2308         i_length = IODDescriptorLength( &i_data, &p_data );
2309
2310         i_data_sav = i_data;
2311         p_data_sav = p_data;
2312
2313         i_data = i_length;
2314
2315         switch( i_tag )
2316         {
2317             case 0x03:
2318                 {
2319 #define es_descr    p_iod->es_descr[i_es_index]
2320                     int i_decoderConfigDescr_length;
2321 #ifdef TS_DEBUG
2322                     fprintf( stderr, "\n* - ES_Descriptor length:%d", i_length );
2323 #endif
2324                     es_descr.b_ok = 1;
2325
2326                     es_descr.i_es_id = IODGetWord( &i_data, &p_data );
2327                     i_flags = IODGetByte( &i_data, &p_data );
2328                     es_descr.b_streamDependenceFlag = ( i_flags >> 7 )&0x01;
2329                     b_url = ( i_flags >> 6 )&0x01;
2330                     es_descr.b_OCRStreamFlag = ( i_flags >> 5 )&0x01;
2331                     es_descr.i_streamPriority = i_flags & 0x1f;
2332 #ifdef TS_DEBUG
2333                     fprintf( stderr, "\n*   * streamDependenceFlag:%d", es_descr.b_streamDependenceFlag );
2334                     fprintf( stderr, "\n*   * OCRStreamFlag:%d", es_descr.b_OCRStreamFlag );
2335                     fprintf( stderr, "\n*   * streamPriority:%d", es_descr.i_streamPriority );
2336 #endif
2337                     if( es_descr.b_streamDependenceFlag )
2338                     {
2339                         es_descr.i_dependOn_es_id = IODGetWord( &i_data, &p_data );
2340 #ifdef TS_DEBUG
2341                         fprintf( stderr, "\n*   * dependOn_es_id:%d", es_descr.i_dependOn_es_id );
2342 #endif
2343                     }
2344
2345                     if( b_url )
2346                     {
2347                         es_descr.psz_url = IODGetURL( &i_data, &p_data );
2348 #ifdef TS_DEBUG
2349                         fprintf( stderr, "\n* url string:%s", es_descr.psz_url );
2350 #endif
2351                     }
2352                     else
2353                     {
2354                         es_descr.psz_url = NULL;
2355                     }
2356
2357                     if( es_descr.b_OCRStreamFlag )
2358                     {
2359                         es_descr.i_OCR_es_id = IODGetWord( &i_data, &p_data );
2360 #ifdef TS_DEBUG
2361                         fprintf( stderr, "\n*   * OCR_es_id:%d", es_descr.i_OCR_es_id );
2362 #endif
2363                     }
2364
2365                     if( IODGetByte( &i_data, &p_data ) != 0x04 )
2366                     {
2367 #ifdef TS_DEBUG
2368                         fprintf( stderr, "\n* ERR missing DecoderConfigDescr" );
2369 #endif
2370                         es_descr.b_ok = 0;
2371                         break;
2372                     }
2373                     i_decoderConfigDescr_length = IODDescriptorLength( &i_data, &p_data );
2374 #ifdef TS_DEBUG
2375                     fprintf( stderr, "\n*   - DecoderConfigDesc length:%d", i_decoderConfigDescr_length );
2376 #endif
2377 #define dec_descr   es_descr.dec_descr
2378                     dec_descr.i_objectTypeIndication = IODGetByte( &i_data, &p_data );
2379                     i_flags = IODGetByte( &i_data, &p_data );
2380                     dec_descr.i_streamType = i_flags >> 2;
2381                     dec_descr.b_upStream = ( i_flags >> 1 )&0x01;
2382                     dec_descr.i_bufferSizeDB = IODGet3Bytes( &i_data, &p_data );
2383                     dec_descr.i_maxBitrate = IODGetDWord( &i_data, &p_data );
2384                     dec_descr.i_avgBitrate = IODGetDWord( &i_data, &p_data );
2385 #ifdef TS_DEBUG
2386                     fprintf( stderr, "\n*     * objectTypeIndication:0x%x", dec_descr.i_objectTypeIndication  );
2387                     fprintf( stderr, "\n*     * streamType:0x%x", dec_descr.i_streamType );
2388                     fprintf( stderr, "\n*     * upStream:%d", dec_descr.b_upStream );
2389                     fprintf( stderr, "\n*     * bufferSizeDB:%d", dec_descr.i_bufferSizeDB );
2390                     fprintf( stderr, "\n*     * maxBitrate:%d", dec_descr.i_maxBitrate );
2391                     fprintf( stderr, "\n*     * avgBitrate:%d", dec_descr.i_avgBitrate );
2392 #endif
2393                     if( i_decoderConfigDescr_length > 13 && IODGetByte( &i_data, &p_data ) == 0x05 )
2394                     {
2395                         int i;
2396                         dec_descr.i_decoder_specific_info_len =
2397                             IODDescriptorLength( &i_data, &p_data );
2398                         if( dec_descr.i_decoder_specific_info_len > 0 )
2399                         {
2400                             dec_descr.p_decoder_specific_info =
2401                                 malloc( dec_descr.i_decoder_specific_info_len );
2402                         }
2403                         for( i = 0; i < dec_descr.i_decoder_specific_info_len; i++ )
2404                         {
2405                             dec_descr.p_decoder_specific_info[i] = IODGetByte( &i_data, &p_data );
2406                         }
2407                     }
2408                     else
2409                     {
2410                         dec_descr.i_decoder_specific_info_len = 0;
2411                         dec_descr.p_decoder_specific_info = NULL;
2412                     }
2413                 }
2414 #undef  dec_descr
2415 #define sl_descr    es_descr.sl_descr
2416                 {
2417                     int i_SLConfigDescr_length;
2418                     int i_predefined;
2419
2420                     if( IODGetByte( &i_data, &p_data ) != 0x06 )
2421                     {
2422 #ifdef TS_DEBUG
2423                         fprintf( stderr, "\n* ERR missing SLConfigDescr" );
2424 #endif
2425                         es_descr.b_ok = 0;
2426                         break;
2427                     }
2428                     i_SLConfigDescr_length = IODDescriptorLength( &i_data, &p_data );
2429 #ifdef TS_DEBUG
2430                     fprintf( stderr, "\n*   - SLConfigDescr length:%d", i_SLConfigDescr_length );
2431 #endif
2432                     i_predefined = IODGetByte( &i_data, &p_data );
2433 #ifdef TS_DEBUG
2434                     fprintf( stderr, "\n*     * i_predefined:0x%x", i_predefined  );
2435 #endif
2436                     switch( i_predefined )
2437                     {
2438                         case 0x01:
2439                             {
2440                                 sl_descr.b_useAccessUnitStartFlag   = 0;
2441                                 sl_descr.b_useAccessUnitEndFlag     = 0;
2442                                 sl_descr.b_useRandomAccessPointFlag = 0;
2443                                 //sl_descr.b_useRandomAccessUnitsOnlyFlag = 0;
2444                                 sl_descr.b_usePaddingFlag           = 0;
2445                                 sl_descr.b_useTimeStampsFlags       = 0;
2446                                 sl_descr.b_useIdleFlag              = 0;
2447                                 sl_descr.b_durationFlag     = 0;    // FIXME FIXME
2448                                 sl_descr.i_timeStampResolution      = 1000;
2449                                 sl_descr.i_OCRResolution    = 0;    // FIXME FIXME
2450                                 sl_descr.i_timeStampLength          = 32;
2451                                 sl_descr.i_OCRLength        = 0;    // FIXME FIXME
2452                                 sl_descr.i_AU_Length                = 0;
2453                                 sl_descr.i_instantBitrateLength= 0; // FIXME FIXME
2454                                 sl_descr.i_degradationPriorityLength= 0;
2455                                 sl_descr.i_AU_seqNumLength          = 0;
2456                                 sl_descr.i_packetSeqNumLength       = 0;
2457                                 if( sl_descr.b_durationFlag )
2458                                 {
2459                                     sl_descr.i_timeScale            = 0;    // FIXME FIXME
2460                                     sl_descr.i_accessUnitDuration   = 0;    // FIXME FIXME
2461                                     sl_descr.i_compositionUnitDuration= 0;    // FIXME FIXME
2462                                 }
2463                                 if( !sl_descr.b_useTimeStampsFlags )
2464                                 {
2465                                     sl_descr.i_startDecodingTimeStamp   = 0;    // FIXME FIXME
2466                                     sl_descr.i_startCompositionTimeStamp= 0;    // FIXME FIXME
2467                                 }
2468                             }
2469                             break;
2470                         default:
2471 #ifdef TS_DEBUG
2472                             fprintf( stderr, "\n* ERR unsupported SLConfigDescr predefined" );
2473 #endif
2474                             es_descr.b_ok = 0;
2475                             break;
2476                     }
2477                 }
2478                 break;
2479 #undef  sl_descr
2480 #undef  es_descr
2481             default:
2482 #ifdef TS_DEBUG
2483                 fprintf( stderr, "\n* - OD tag:0x%x length:%d (Unsupported)", i_tag, i_length );
2484 #endif
2485                 break;
2486         }
2487
2488         p_data = p_data_sav + i_length;
2489         i_data = i_data_sav - i_length;
2490         i_es_index++;
2491     }
2492 #ifdef TS_DEBUG
2493     fprintf( stderr, "\n*****************************\n" );
2494 #endif
2495     return p_iod;
2496 }
2497
2498 static void IODFree( iod_descriptor_t *p_iod )
2499 {
2500     int i;
2501
2502     if( p_iod->psz_url )
2503     {
2504         free( p_iod->psz_url );
2505         p_iod->psz_url = NULL;
2506         free( p_iod );
2507         return;
2508     }
2509
2510     for( i = 0; i < 255; i++ )
2511     {
2512 #define es_descr p_iod->es_descr[i]
2513         if( es_descr.b_ok )
2514         {
2515             if( es_descr.psz_url )
2516             {
2517                 free( es_descr.psz_url );
2518                 es_descr.psz_url = NULL;
2519             }
2520             else
2521             {
2522                 free( es_descr.dec_descr.p_decoder_specific_info );
2523                 es_descr.dec_descr.p_decoder_specific_info = NULL;
2524                 es_descr.dec_descr.i_decoder_specific_info_len = 0;
2525             }
2526         }
2527         es_descr.b_ok = 0;
2528 #undef  es_descr
2529     }
2530     free( p_iod );
2531 }
2532
2533 /****************************************************************************
2534  ****************************************************************************
2535  ** libdvbpsi callbacks
2536  ****************************************************************************
2537  ****************************************************************************/
2538 static bool DVBProgramIsSelected( demux_t *p_demux, uint16_t i_pgrm )
2539 {
2540     demux_sys_t          *p_sys = p_demux->p_sys;
2541
2542     if ( !p_sys->b_dvb_control ) return false;
2543     if ( (p_sys->i_dvb_program == -1 && p_sys->p_programs_list == NULL)
2544            || p_sys->i_dvb_program == 0 )
2545         return true;
2546     if ( p_sys->i_dvb_program == i_pgrm ) return true;
2547
2548     if ( p_sys->p_programs_list != NULL )
2549     {
2550         int i;
2551         for ( i = 0; i < p_sys->p_programs_list->i_count; i++ )
2552         {
2553             if ( i_pgrm == p_sys->p_programs_list->p_values[i].i_int )
2554                 return true;
2555         }
2556     }
2557     return false;
2558 }
2559
2560 #ifdef TS_USE_DVB_SI
2561 static void SDTCallBack( demux_t *p_demux, dvbpsi_sdt_t *p_sdt )
2562 {
2563     demux_sys_t          *p_sys = p_demux->p_sys;
2564     ts_pid_t             *sdt = &p_sys->pid[0x11];
2565     dvbpsi_sdt_service_t *p_srv;
2566
2567     msg_Dbg( p_demux, "SDTCallBack called" );
2568
2569     if( sdt->psi->i_sdt_version != -1 &&
2570         ( !p_sdt->b_current_next ||
2571           p_sdt->i_version == sdt->psi->i_sdt_version ) )
2572     {
2573         dvbpsi_DeleteSDT( p_sdt );
2574         return;
2575     }
2576
2577     msg_Dbg( p_demux, "new SDT ts_id=%d version=%d current_next=%d "
2578              "network_id=%d",
2579              p_sdt->i_ts_id, p_sdt->i_version, p_sdt->b_current_next,
2580              p_sdt->i_network_id );
2581
2582     for( p_srv = p_sdt->p_first_service; p_srv; p_srv = p_srv->p_next )
2583     {
2584         vlc_meta_t          *p_meta;
2585         dvbpsi_descriptor_t *p_dr;
2586
2587         const char *psz_type = NULL;
2588         const char *psz_status = NULL;
2589
2590         msg_Dbg( p_demux, "  * service id=%d eit schedule=%d present=%d "
2591                  "running=%d free_ca=%d",
2592                  p_srv->i_service_id, p_srv->b_eit_schedule,
2593                  p_srv->b_eit_present, p_srv->i_running_status,
2594                  p_srv->b_free_ca );
2595
2596         if( p_sys->i_dvb_program != -1 && p_sys->i_dvb_program != p_srv->i_service_id )
2597             continue;
2598
2599         p_meta = vlc_meta_New();
2600         for( p_dr = p_srv->p_first_descriptor; p_dr; p_dr = p_dr->p_next )
2601         {
2602             if( p_dr->i_tag == 0x48 )
2603             {
2604                 static const char *ppsz_type[17] = {
2605                     "Reserved",
2606                     "Digital television service",
2607                     "Digital radio sound service",
2608                     "Teletext service",
2609                     "NVOD reference service",
2610                     "NVOD time-shifted service",
2611                     "Mosaic service",
2612                     "PAL coded signal",
2613                     "SECAM coded signal",
2614                     "D/D2-MAC",
2615                     "FM Radio",
2616                     "NTSC coded signal",
2617                     "Data broadcast service",
2618                     "Reserved for Common Interface Usage",
2619                     "RCS Map (see EN 301 790 [35])",
2620                     "RCS FLS (see EN 301 790 [35])",
2621                     "DVB MHP service"
2622                 };
2623                 dvbpsi_service_dr_t *pD = dvbpsi_DecodeServiceDr( p_dr );
2624                 char str1[257];
2625                 char str2[257];
2626
2627                 memcpy( str1, pD->i_service_provider_name,
2628                         pD->i_service_provider_name_length );
2629                 str1[pD->i_service_provider_name_length] = '\0';
2630                 memcpy( str2, pD->i_service_name, pD->i_service_name_length );
2631                 str2[pD->i_service_name_length] = '\0';
2632
2633                 msg_Dbg( p_demux, "    - type=%d provider=%s name=%s",
2634                          pD->i_service_type, str1, str2 );
2635
2636                 vlc_meta_SetTitle( p_meta, str2 );
2637                 vlc_meta_SetPublisher( p_meta, str1 );
2638                 if( pD->i_service_type >= 0x01 && pD->i_service_type <= 0x10 )
2639                     psz_type = ppsz_type[pD->i_service_type];
2640             }
2641         }
2642
2643         if( p_srv->i_running_status >= 0x01 && p_srv->i_running_status <= 0x04 )
2644         {
2645             static const char *ppsz_status[5] = {
2646                 "Unknown",
2647                 "Not running",
2648                 "Starts in a few seconds",
2649                 "Pausing",
2650                 "Running"
2651             };
2652             psz_status = ppsz_status[p_srv->i_running_status];
2653         }
2654
2655         if( psz_type )
2656             vlc_meta_AddExtra( p_meta, "Type", psz_type );
2657         if( psz_status )
2658             vlc_meta_AddExtra( p_meta, "Status", psz_status );
2659
2660         es_out_Control( p_demux->out, ES_OUT_SET_GROUP_META,
2661                         p_srv->i_service_id, p_meta );
2662         vlc_meta_Delete( p_meta );
2663     }
2664
2665     sdt->psi->i_sdt_version = p_sdt->i_version;
2666     dvbpsi_DeleteSDT( p_sdt );
2667 }
2668
2669 /* i_year: year - 1900  i_month: 0-11  i_mday: 1-31 i_hour: 0-23 i_minute: 0-59 i_second: 0-59 */
2670 static int64_t vlc_timegm( int i_year, int i_month, int i_mday, int i_hour, int i_minute, int i_second )
2671 {
2672     static const int pn_day[12+1] = { 0, 31, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334 };
2673     int64_t i_day;
2674     int i;
2675
2676     if( i_year < 70 ||
2677         i_month < 0 || i_month > 11 || i_mday < 1 || i_mday > 31 ||
2678         i_hour < 0 || i_hour > 23 || i_minute < 0 || i_minute > 59 || i_second < 0 || i_second > 59 )
2679         return -1;
2680
2681     /* Count the number of days */
2682     i_day = 365 * (i_year-70) + pn_day[i_month] + i_mday - 1;
2683 #define LEAP(y) ( ((y)%4) == 0 && (((y)%100) != 0 || ((y)%400) == 0) ? 1 : 0)
2684     for( i = 70; i < i_year; i++ )
2685         i_day += LEAP(1900+i);
2686     if( i_month > 1 )
2687         i_day += LEAP(1900+i_year);
2688 #undef LEAP
2689     /**/
2690     return ((24*i_day + i_hour)*60 + i_minute)*60 + i_second;
2691 }
2692
2693 static void EITDecodeMjd( int i_mjd, int *p_y, int *p_m, int *p_d )
2694 {
2695     const int yp = (int)( ( (double)i_mjd - 15078.2)/365.25 );
2696     const int mp = (int)( ((double)i_mjd - 14956.1 - (int)(yp * 365.25)) / 30.6001 );
2697     const int c = ( mp == 14 || mp == 15 ) ? 1 : 0;
2698
2699     *p_y = 1900 + yp + c*1;
2700     *p_m = mp - 1 - c*12;
2701     *p_d = i_mjd - 14956 - (int)(yp*365.25) - (int)(mp*30.6001);
2702 }
2703 #define CVT_FROM_BCD(v) ((((v) >> 4)&0xf)*10 + ((v)&0xf))
2704 static int64_t EITConvertStartTime( uint64_t i_date )
2705 {
2706     const int i_mjd = i_date >> 24;
2707     const int i_hour   = CVT_FROM_BCD(i_date >> 16);
2708     const int i_minute = CVT_FROM_BCD(i_date >>  8);
2709     const int i_second = CVT_FROM_BCD(i_date      );
2710     int i_year;
2711     int i_month;
2712     int i_day;
2713
2714     /* if all 40 bits are 1, the start is unknown */
2715     if( i_date == UINT64_C(0xffffffffff) )
2716         return -1;
2717
2718     EITDecodeMjd( i_mjd, &i_year, &i_month, &i_day );
2719     return vlc_timegm( i_year - 1900, i_month - 1, i_day, i_hour, i_minute, i_second );
2720 }
2721 static int EITConvertDuration( uint32_t i_duration )
2722 {
2723     return CVT_FROM_BCD(i_duration >> 16) * 3600 +
2724            CVT_FROM_BCD(i_duration >> 8 ) * 60 +
2725            CVT_FROM_BCD(i_duration      );
2726 }
2727 #undef CVT_FROM_BCD
2728
2729 /* FIXME same than dvbsi_to_utf8 from dvb access */
2730 static char *EITConvertToUTF8( const unsigned char *psz_instring,
2731                                size_t i_length )
2732 {
2733     const char *psz_encoding;
2734     char *psz_outstring;
2735     char psz_encbuf[sizeof( "ISO_8859-123" )];
2736     size_t i_in, i_out, offset = 1;
2737     vlc_iconv_t iconv_handle;
2738
2739     if( i_length < 1 ) return NULL;
2740     if( psz_instring[0] >= 0x20 )
2741     {
2742         psz_encoding = "ISO_8859-1";
2743         /* According to the specification, this should be ISO6937,
2744          * but it seems Latin-1 is used instead. */
2745         offset = 0;
2746     }
2747     else switch( psz_instring[0] )
2748     {
2749     case 0x01:
2750         psz_encoding = "ISO_8859-5";
2751         break;
2752     case 0x02:
2753         psz_encoding = "ISO_8859-6";
2754         break;
2755     case 0x03:
2756         psz_encoding = "ISO_8859-7";
2757         break;
2758     case 0x04:
2759         psz_encoding = "ISO_8859-8";
2760         break;
2761     case 0x05:
2762         psz_encoding = "ISO_8859-9";
2763         break;
2764     case 0x06:
2765         psz_encoding = "ISO_8859-10";
2766         break;
2767     case 0x07:
2768         psz_encoding = "ISO_8859-11";
2769         break;
2770     case 0x08:
2771         psz_encoding = "ISO_8859-12";
2772         break;
2773     case 0x09:
2774         psz_encoding = "ISO_8859-13";
2775         break;
2776     case 0x0a:
2777         psz_encoding = "ISO_8859-14";
2778         break;
2779     case 0x0b:
2780         psz_encoding = "ISO_8859-15";
2781         break;
2782     case 0x10:
2783 #warning Is Latin-10 (psz_instring[2] == 16) really illegal?
2784         if( i_length < 3 || psz_instring[1] != 0x00 || psz_instring[2] > 15
2785          || psz_instring[2] == 0 )
2786         {
2787             psz_encoding = "UTF-8";
2788             offset = 0;
2789         }
2790         else
2791         {
2792             sprintf( psz_encbuf, "ISO_8859-%u", psz_instring[2] );
2793             psz_encoding = psz_encbuf;
2794             offset = 3;
2795         }
2796         break;
2797     case 0x11:
2798 #warning Is there a BOM or do we use a fixed endianess?
2799         psz_encoding = "UTF-16";
2800         break;
2801     case 0x12:
2802         psz_encoding = "KSC5601-1987";
2803         break;
2804     case 0x13:
2805         psz_encoding = "GB2312"; /* GB-2312-1980 */
2806         break;
2807     case 0x14:
2808         psz_encoding = "BIG-5";
2809         break;
2810     case 0x15:
2811         psz_encoding = "UTF-8";
2812         break;
2813     default:
2814         /* invalid */
2815         psz_encoding = "UTF-8";
2816         offset = 0;
2817     }
2818
2819     i_in = i_length - offset;
2820     i_out = i_in * 6 + 1;
2821
2822     psz_outstring = malloc( i_out );
2823     if( !psz_outstring )
2824     {
2825         return NULL;
2826     }
2827
2828     iconv_handle = vlc_iconv_open( "UTF-8", psz_encoding );
2829     if( iconv_handle == (vlc_iconv_t)(-1) )
2830     {
2831          /* Invalid character set (e.g. ISO_8859-12) */
2832          memcpy( psz_outstring, &psz_instring[offset], i_in );
2833          psz_outstring[i_in] = '\0';
2834          EnsureUTF8( psz_outstring );
2835     }
2836     else
2837     {
2838         const char *psz_in = (const char *)&psz_instring[offset];
2839         char *psz_out = psz_outstring;
2840
2841         while( vlc_iconv( iconv_handle, &psz_in, &i_in,
2842                           &psz_out, &i_out ) == (size_t)(-1) )
2843         {
2844             /* skip naughty byte. This may fail terribly for multibyte stuff,
2845              * but what can we do anyway? */
2846             psz_in++;
2847             i_in--;
2848             vlc_iconv( iconv_handle, NULL, NULL, NULL, NULL ); /* reset */
2849         }
2850         vlc_iconv_close( iconv_handle );
2851
2852         *psz_out = '\0';
2853     }
2854     return psz_outstring;
2855 }
2856
2857 static void EITCallBack( demux_t *p_demux, dvbpsi_eit_t *p_eit )
2858 {
2859     demux_sys_t        *p_sys = p_demux->p_sys;
2860     dvbpsi_eit_event_t *p_evt;
2861     vlc_epg_t *p_epg;
2862
2863     msg_Dbg( p_demux, "EITCallBack called" );
2864     if( !p_eit->b_current_next || ( p_sys->i_dvb_program != -1 && p_sys->i_dvb_program != p_eit->i_service_id ) )
2865     {
2866         dvbpsi_DeleteEIT( p_eit );
2867         return;
2868     }
2869
2870     msg_Dbg( p_demux, "new EIT service_id=%d version=%d current_next=%d "
2871              "ts_id=%d network_id=%d segment_last_section_number=%d "
2872              "last_table_id=%d",
2873              p_eit->i_service_id, p_eit->i_version, p_eit->b_current_next,
2874              p_eit->i_ts_id, p_eit->i_network_id,
2875              p_eit->i_segment_last_section_number, p_eit->i_last_table_id );
2876
2877     p_epg = vlc_epg_New( NULL );
2878     for( p_evt = p_eit->p_first_event; p_evt; p_evt = p_evt->p_next )
2879     {
2880         dvbpsi_descriptor_t *p_dr;
2881         char                *psz_name = NULL;
2882         char                *psz_text = NULL;
2883         char                *psz_extra = strdup("");
2884         int64_t i_start;
2885         int i_duration;
2886
2887         i_start = EITConvertStartTime( p_evt->i_start_time );
2888         i_duration = EITConvertDuration( p_evt->i_duration );
2889
2890         msg_Dbg( p_demux, "  * event id=%d start_time:%d duration=%d "
2891                           "running=%d free_ca=%d",
2892                  p_evt->i_event_id, (int)i_start, (int)i_duration,
2893                  p_evt->i_running_status, p_evt->b_free_ca );
2894
2895         for( p_dr = p_evt->p_first_descriptor; p_dr; p_dr = p_dr->p_next )
2896         {
2897             if( p_dr->i_tag == 0x4d )
2898             {
2899                 dvbpsi_short_event_dr_t *pE = dvbpsi_DecodeShortEventDr( p_dr );
2900
2901                 if( pE )
2902                 {
2903                     psz_name = EITConvertToUTF8( pE->i_event_name, pE->i_event_name_length);
2904                     psz_text = EITConvertToUTF8( pE->i_text, pE->i_text_length );
2905                     msg_Dbg( p_demux, "    - short event lang=%3.3s '%s' : '%s'",
2906                              pE->i_iso_639_code, psz_name, psz_text );
2907                 }
2908             }
2909             else if( p_dr->i_tag == 0x4e )
2910             {
2911                 dvbpsi_extended_event_dr_t *pE = dvbpsi_DecodeExtendedEventDr( p_dr );
2912                 if( pE )
2913                 {
2914                     int i;
2915                     msg_Dbg( p_demux, "    - extended event lang=%3.3s [%d/%d]",
2916                              pE->i_iso_639_code,
2917                              pE->i_descriptor_number, pE->i_last_descriptor_number );
2918
2919                     if( pE->i_text_length > 0 )
2920                     {
2921                         char *psz_text = EITConvertToUTF8( pE->i_text, pE->i_text_length );
2922                         if( psz_text )
2923                         {
2924                             msg_Dbg( p_demux, "       - text='%s'", psz_text );
2925
2926                             psz_extra = realloc( psz_extra, strlen(psz_extra) + strlen(psz_text) + 1 );
2927                             strcat( psz_extra, psz_text );
2928                             free( psz_text );
2929                         }
2930                     }
2931
2932                     for( i = 0; i < pE->i_entry_count; i++ )
2933                     {
2934                         char *psz_dsc = EITConvertToUTF8( pE->i_item_description[i],
2935                                                           pE->i_item_description_length[i] );
2936                         char *psz_itm = EITConvertToUTF8( pE->i_item[i], pE->i_item_length[i] );
2937
2938                         if( psz_dsc && psz_itm )
2939                         {
2940                             msg_Dbg( p_demux, "       - desc='%s' item='%s'", psz_dsc, psz_itm );
2941
2942                             psz_extra = realloc( psz_extra, strlen(psz_extra) + strlen(psz_dsc) + strlen(psz_itm) + 3 + 1 );
2943                             strcat( psz_extra, "(" );
2944                             strcat( psz_extra, psz_dsc );
2945                             strcat( psz_extra, " " );
2946                             strcat( psz_extra, psz_itm );
2947                             strcat( psz_extra, ")" );
2948                         }
2949                         free( psz_dsc );
2950                         free( psz_itm );
2951                     }
2952                 }
2953             }
2954             else
2955             {
2956                 msg_Dbg( p_demux, "    - tag=0x%x(%d)", p_dr->i_tag, p_dr->i_tag );
2957             }
2958         }
2959
2960         /* */
2961         if( i_start > 0 )
2962             vlc_epg_AddEvent( p_epg, i_start, i_duration, psz_name, psz_text, psz_extra );
2963
2964         /* Update "now playing" field */
2965         if( p_evt->i_running_status == 0x04 && i_start > 0 )
2966             vlc_epg_SetCurrent( p_epg, i_start );
2967
2968         free( psz_name );
2969         free( psz_text );
2970
2971         free( psz_extra );
2972     }
2973     if( p_epg->i_event > 0 )
2974     {
2975         if( p_eit->i_service_id == p_sys->i_dvb_program )
2976         {
2977             p_sys->i_dvb_length = 0;
2978             p_sys->i_dvb_start = 0;
2979
2980             if( p_epg->p_current )
2981             {
2982                 p_sys->i_dvb_start = p_epg->p_current->i_start;
2983                 p_sys->i_dvb_length = p_epg->p_current->i_duration;
2984             }
2985         }
2986         es_out_Control( p_demux->out, ES_OUT_SET_GROUP_EPG, p_eit->i_service_id, p_epg );
2987     }
2988     vlc_epg_Delete( p_epg );
2989
2990     dvbpsi_DeleteEIT( p_eit );
2991 }
2992
2993 static void PSINewTableCallBack( demux_t *p_demux, dvbpsi_handle h,
2994                                  uint8_t  i_table_id, uint16_t i_extension )
2995 {
2996 #if 0
2997     msg_Dbg( p_demux, "PSINewTableCallBack: table 0x%x(%d) ext=0x%x(%d)",
2998              i_table_id, i_table_id, i_extension, i_extension );
2999 #endif
3000     if( p_demux->p_sys->pid[0].psi->i_pat_version != -1 && i_table_id == 0x42 )
3001     {
3002         msg_Dbg( p_demux, "PSINewTableCallBack: table 0x%x(%d) ext=0x%x(%d)",
3003                  i_table_id, i_table_id, i_extension, i_extension );
3004
3005         dvbpsi_AttachSDT( h, i_table_id, i_extension,
3006                           (dvbpsi_sdt_callback)SDTCallBack, p_demux );
3007     }
3008     else if( p_demux->p_sys->pid[0x11].psi->i_sdt_version != -1 &&
3009              ( i_table_id == 0x4e || /* Current/Following */
3010                (i_table_id >= 0x50 && i_table_id <= 0x5f) ) ) /* Schedule */
3011     {
3012         msg_Dbg( p_demux, "PSINewTableCallBack: table 0x%x(%d) ext=0x%x(%d)",
3013                  i_table_id, i_table_id, i_extension, i_extension );
3014
3015         dvbpsi_AttachEIT( h, i_table_id, i_extension,
3016                           (dvbpsi_eit_callback)EITCallBack, p_demux );
3017     }
3018 }
3019 #endif
3020
3021 static void PMTCallBack( demux_t *p_demux, dvbpsi_pmt_t *p_pmt )
3022 {
3023     demux_sys_t          *p_sys = p_demux->p_sys;
3024     dvbpsi_descriptor_t  *p_dr;
3025     dvbpsi_pmt_es_t      *p_es;
3026
3027     ts_pid_t             *pmt = NULL;
3028     ts_prg_psi_t         *prg = NULL;
3029
3030     ts_pid_t             **pp_clean = NULL;
3031     int                  i_clean = 0, i;
3032     bool                 b_hdmv = false;
3033
3034     msg_Dbg( p_demux, "PMTCallBack called" );
3035
3036     /* First find this PMT declared in PAT */
3037     for( i = 0; i < p_sys->i_pmt; i++ )
3038     {
3039         int i_prg;
3040         for( i_prg = 0; i_prg < p_sys->pmt[i]->psi->i_prg; i_prg++ )
3041         {
3042             const int i_pmt_number = p_sys->pmt[i]->psi->prg[i_prg]->i_number;
3043             if( i_pmt_number != TS_USER_PMT_NUMBER && i_pmt_number == p_pmt->i_program_number )
3044             {
3045                 pmt = p_sys->pmt[i];
3046                 prg = p_sys->pmt[i]->psi->prg[i_prg];
3047                 break;
3048             }
3049         }
3050         if( pmt ) break;
3051     }
3052
3053     if( pmt == NULL )
3054     {
3055         msg_Warn( p_demux, "unreferenced program (broken stream)" );
3056         dvbpsi_DeletePMT(p_pmt);
3057         return;
3058     }
3059
3060     if( prg->i_version != -1 &&
3061         ( !p_pmt->b_current_next || prg->i_version == p_pmt->i_version ) )
3062     {
3063         dvbpsi_DeletePMT( p_pmt );
3064         return;
3065     }
3066
3067     /* Clean this program (remove all es) */
3068     for( i = 0; i < 8192; i++ )
3069     {
3070         ts_pid_t *pid = &p_sys->pid[i];
3071
3072         if( pid->b_valid && pid->p_owner == pmt->psi &&
3073             pid->i_owner_number == prg->i_number && pid->psi == NULL )
3074         {
3075             TAB_APPEND( i_clean, pp_clean, pid );
3076         }
3077     }
3078     if( prg->iod )
3079     {
3080         IODFree( prg->iod );
3081         prg->iod = NULL;
3082     }
3083
3084     msg_Dbg( p_demux, "new PMT program number=%d version=%d pid_pcr=%d",
3085              p_pmt->i_program_number, p_pmt->i_version, p_pmt->i_pcr_pid );
3086     prg->i_pid_pcr = p_pmt->i_pcr_pid;
3087     prg->i_version = p_pmt->i_version;
3088
3089     if( DVBProgramIsSelected( p_demux, prg->i_number ) )
3090     {
3091         /* Set demux filter */
3092         stream_Control( p_demux->s, STREAM_CONTROL_ACCESS,
3093                         ACCESS_SET_PRIVATE_ID_STATE, prg->i_pid_pcr,
3094                         true );
3095     }
3096     else if ( p_sys->b_dvb_control )
3097     {
3098         msg_Warn( p_demux, "skipping program (not selected)" );
3099         dvbpsi_DeletePMT(p_pmt);
3100         return;
3101     }
3102
3103     /* Parse descriptor */
3104     for( p_dr = p_pmt->p_first_descriptor; p_dr != NULL; p_dr = p_dr->p_next )
3105     {
3106         if( p_dr->i_tag == 0x1d )
3107         {
3108             /* We have found an IOD descriptor */
3109             msg_Dbg( p_demux, " * descriptor : IOD (0x1d)" );
3110
3111             prg->iod = IODNew( p_dr->i_length, p_dr->p_data );
3112         }
3113         else if( p_dr->i_tag == 0x9 )
3114         {
3115             uint16_t i_sysid = ((uint16_t)p_dr->p_data[0] << 8)
3116                                 | p_dr->p_data[1];
3117             msg_Dbg( p_demux, " * descriptor : CA (0x9) SysID 0x%x", i_sysid );
3118         }
3119         else if( p_dr->i_tag == 0x05 )
3120         {
3121             if( p_dr->i_tag == 0x05 )
3122             {
3123                 /* Registration Descriptor */
3124                 if( p_dr->i_length != 4 )
3125                 {
3126                     msg_Warn( p_demux, "invalid Registration Descriptor" );
3127                 }
3128                 else
3129                 {
3130                     msg_Dbg( p_demux, " * descriptor : registration %4.4s", p_dr->p_data );
3131                     if( !memcmp( p_dr->p_data, "HDMV", 4 ) )
3132                     {
3133                         /* Blu-Ray */
3134                         b_hdmv = true;
3135                     }
3136                 }
3137             }
3138         }
3139         else
3140         {
3141             msg_Dbg( p_demux, " * descriptor : unknown (0x%x)", p_dr->i_tag );
3142         }
3143     }
3144
3145     for( p_es = p_pmt->p_first_es; p_es != NULL; p_es = p_es->p_next )
3146     {
3147         ts_pid_t tmp_pid, *old_pid = 0, *pid = &tmp_pid;
3148
3149         /* Find out if the PID was already declared */
3150         for( i = 0; i < i_clean; i++ )
3151         {
3152             if( pp_clean[i] == &p_sys->pid[p_es->i_pid] )
3153             {
3154                 old_pid = pp_clean[i];
3155                 break;
3156             }
3157         }
3158
3159         if( !old_pid && p_sys->pid[p_es->i_pid].b_valid )
3160         {
3161             ts_pid_t *pid = &p_sys->pid[p_es->i_pid];
3162             if( ( pid->i_pid == 0x11 /* SDT */ ||
3163                   pid->i_pid == 0x12 /* EDT */ ) && pid->psi )
3164             {
3165                 /* This doesn't look like a DVB stream so don't try
3166                  * parsing the SDT/EDT */
3167                 dvbpsi_DetachDemux( pid->psi->handle );
3168                 free( pid->psi );
3169                 pid->psi = NULL;
3170             }
3171             else
3172             {
3173                 msg_Warn( p_demux, "pmt error: pid=%d already defined",
3174                           p_es->i_pid );
3175                 continue;
3176             }
3177         }
3178
3179         PIDInit( pid, false, pmt->psi );
3180         PIDFillFormat( pid, p_es->i_type );
3181         pid->i_owner_number = prg->i_number;
3182         pid->i_pid          = p_es->i_pid;
3183         pid->b_seen         = p_sys->pid[p_es->i_pid].b_seen;
3184
3185         if( p_es->i_type == 0x10 || p_es->i_type == 0x11 ||
3186             p_es->i_type == 0x12 || p_es->i_type == 0x0f )
3187         {
3188             /* MPEG-4 stream: search SL_DESCRIPTOR */
3189             dvbpsi_descriptor_t *p_dr = p_es->p_first_descriptor;;
3190
3191             while( p_dr && ( p_dr->i_tag != 0x1f ) ) p_dr = p_dr->p_next;
3192
3193             if( p_dr && p_dr->i_length == 2 )
3194             {
3195                 int i_es_id = ( p_dr->p_data[0] << 8 ) | p_dr->p_data[1];
3196
3197                 msg_Warn( p_demux, "found SL_descriptor es_id=%d", i_es_id );
3198
3199                 pid->es->p_mpeg4desc = NULL;
3200
3201                 for( i = 0; i < 255; i++ )
3202                 {
3203                     iod_descriptor_t *iod = prg->iod;
3204
3205                     if( iod->es_descr[i].b_ok &&
3206                         iod->es_descr[i].i_es_id == i_es_id )
3207                     {
3208                         pid->es->p_mpeg4desc = &iod->es_descr[i];
3209                         break;
3210                     }
3211                 }
3212             }
3213
3214             if( pid->es->p_mpeg4desc != NULL )
3215             {
3216                 decoder_config_descriptor_t *dcd =
3217                     &pid->es->p_mpeg4desc->dec_descr;
3218
3219                 if( dcd->i_streamType == 0x04 )    /* VisualStream */
3220                 {
3221                     pid->es->fmt.i_cat = VIDEO_ES;
3222                     switch( dcd->i_objectTypeIndication )
3223                     {
3224                     case 0x0B: /* mpeg4 sub */
3225                         pid->es->fmt.i_cat = SPU_ES;
3226                         pid->es->fmt.i_codec = VLC_FOURCC('s','u','b','t');
3227                         break;
3228
3229                     case 0x20: /* mpeg4 */
3230                         pid->es->fmt.i_codec = VLC_FOURCC('m','p','4','v');
3231                         break;
3232                     case 0x21: /* h264 */
3233                         pid->es->fmt.i_codec = VLC_FOURCC('h','2','6','4');
3234                         break;
3235                     case 0x60:
3236                     case 0x61:
3237                     case 0x62:
3238                     case 0x63:
3239                     case 0x64:
3240                     case 0x65: /* mpeg2 */
3241                         pid->es->fmt.i_codec = VLC_FOURCC( 'm','p','g','v' );
3242                         break;
3243                     case 0x6a: /* mpeg1 */
3244                         pid->es->fmt.i_codec = VLC_FOURCC( 'm','p','g','v' );
3245                         break;
3246                     case 0x6c: /* mpeg1 */
3247                         pid->es->fmt.i_codec = VLC_FOURCC( 'j','p','e','g' );
3248                         break;
3249                     default:
3250                         pid->es->fmt.i_cat = UNKNOWN_ES;
3251                         break;
3252                     }
3253                 }
3254                 else if( dcd->i_streamType == 0x05 )    /* AudioStream */
3255                 {
3256                     pid->es->fmt.i_cat = AUDIO_ES;
3257                     switch( dcd->i_objectTypeIndication )
3258                     {
3259                     case 0x40: /* mpeg4 */
3260                         pid->es->fmt.i_codec = VLC_FOURCC('m','p','4','a');
3261                         break;
3262                     case 0x66:
3263                     case 0x67:
3264                     case 0x68: /* mpeg2 aac */
3265                         pid->es->fmt.i_codec = VLC_FOURCC('m','p','4','a');
3266                         break;
3267                     case 0x69: /* mpeg2 */
3268                         pid->es->fmt.i_codec = VLC_FOURCC('m','p','g','a');
3269                         break;
3270                     case 0x6b: /* mpeg1 */
3271                         pid->es->fmt.i_codec = VLC_FOURCC('m','p','g','a');
3272                         break;
3273                     default:
3274                         pid->es->fmt.i_cat = UNKNOWN_ES;
3275                         break;
3276                     }
3277                 }
3278                 else
3279                 {
3280                     pid->es->fmt.i_cat = UNKNOWN_ES;
3281                 }
3282
3283                 if( pid->es->fmt.i_cat != UNKNOWN_ES )
3284                 {
3285                     pid->es->fmt.i_extra = dcd->i_decoder_specific_info_len;
3286                     if( pid->es->fmt.i_extra > 0 )
3287                     {
3288                         pid->es->fmt.p_extra = malloc( pid->es->fmt.i_extra );
3289                         if( pid->es->fmt.p_extra )
3290                             memcpy( pid->es->fmt.p_extra,
3291                                     dcd->p_decoder_specific_info,
3292                                     pid->es->fmt.i_extra );
3293                     }
3294                 }
3295             }
3296         }
3297         else if( p_es->i_type == 0x06 )
3298         {
3299             dvbpsi_descriptor_t *p_dr;
3300
3301             for( p_dr = p_es->p_first_descriptor; p_dr != NULL;
3302                  p_dr = p_dr->p_next )
3303             {
3304                 msg_Dbg( p_demux, "  * es pid=%d type=%d dr->i_tag=0x%x",
3305                          p_es->i_pid, p_es->i_type, p_dr->i_tag );
3306
3307                 if( p_dr->i_tag == 0x05 )
3308                 {
3309                     /* Registration Descriptor */
3310                     if( p_dr->i_length != 4 )
3311                     {
3312                         msg_Warn( p_demux, "invalid Registration Descriptor" );
3313                     }
3314                     else
3315                     {
3316                         if( !memcmp( p_dr->p_data, "AC-3", 4 ) )
3317                         {
3318                             /* ATSC with stream_type 0x81 (but this descriptor
3319                              * is then not mandatory */
3320                             pid->es->fmt.i_cat = AUDIO_ES;
3321                             pid->es->fmt.i_codec = VLC_FOURCC('a','5','2',' ');
3322                         }
3323                         else if( !memcmp( p_dr->p_data, "DTS1", 4 ) ||
3324                                  !memcmp( p_dr->p_data, "DTS2", 4 ) ||
3325                                  !memcmp( p_dr->p_data, "DTS3", 4 ) )
3326                         {
3327                            /*registration descriptor(ETSI TS 101 154 Annex F)*/
3328                             pid->es->fmt.i_cat = AUDIO_ES;
3329                             pid->es->fmt.i_codec = VLC_FOURCC('d','t','s',' ');
3330                         }
3331                         else if( !memcmp( p_dr->p_data, "BSSD", 4 ) )
3332                         {
3333                             pid->es->fmt.i_cat = AUDIO_ES;
3334                             pid->es->fmt.b_packetized = true;
3335                             pid->es->fmt.i_codec = VLC_FOURCC('a','e','s','3');
3336                         }
3337                         else
3338                         {
3339                             msg_Warn( p_demux,
3340                                       "unknown Registration Descriptor (%4.4s)",
3341                                       p_dr->p_data );
3342                         }
3343                     }
3344
3345                 }
3346                 else if( p_dr->i_tag == 0x6a )
3347                 {
3348                     /* DVB with stream_type 0x06 */
3349                     pid->es->fmt.i_cat = AUDIO_ES;
3350                     pid->es->fmt.i_codec = VLC_FOURCC( 'a', '5', '2', ' ' );
3351                 }
3352                 else if( p_dr->i_tag == 0x81 )
3353                 {
3354                     /* ATSC with stream_type 0x06 */
3355                     pid->es->fmt.i_cat = AUDIO_ES;
3356                     pid->es->fmt.i_codec = VLC_FOURCC( 'a', '5', '2', ' ' );
3357                 }
3358                 else if( p_dr->i_tag == 0x7a )
3359                 {
3360                     /* DVB with stream_type 0x06 (ETS EN 300 468) */
3361                     pid->es->fmt.i_cat = AUDIO_ES;
3362                     pid->es->fmt.i_codec = VLC_FOURCC( 'e', 'a', 'c', '3' );
3363                 }
3364                 else if( p_dr->i_tag == 0x73 )
3365                 {
3366                     /* DTS audio descriptor (ETSI TS 101 154 Annex F) */
3367                     msg_Dbg( p_demux, "    * DTS audio descriptor not decoded" );
3368                     pid->es->fmt.i_cat = AUDIO_ES;
3369                     pid->es->fmt.i_codec = VLC_FOURCC( 'd', 't', 's', ' ' );
3370                 }
3371                 else if( p_dr->i_tag == 0x45 )
3372                 {
3373                     msg_Dbg( p_demux, "    * VBI Data descriptor" );
3374                     /* FIXME : store the information somewhere */
3375                 }
3376                 else if( p_dr->i_tag == 0x46 )
3377                 {
3378                     msg_Dbg( p_demux, "    * VBI Teletext descriptor" );
3379                     /* FIXME : store the information somewhere */
3380                 }
3381 #ifdef _DVBPSI_DR_52_H_
3382                 else if( p_dr->i_tag == 0x52 )
3383                 {
3384                     dvbpsi_stream_identifier_dr_t *si;
3385                     si = dvbpsi_DecodeStreamIdentifierDr( p_dr );
3386
3387                     msg_Dbg( p_demux, "    * Stream Component Identifier: %d", si->i_component_tag );
3388                 }
3389 #endif
3390                 else if( p_dr->i_tag == 0x56 )
3391                 {
3392                     msg_Dbg( p_demux, "    * EBU Teletext descriptor" );
3393                     pid->es->fmt.i_cat = SPU_ES;
3394                     pid->es->fmt.i_codec = VLC_FOURCC( 't', 'e', 'l', 'x' );
3395                     pid->es->fmt.i_extra = p_dr->i_length;
3396                     pid->es->fmt.p_extra = p_dr->i_length ?
3397                         malloc( p_dr->i_length ) : NULL;
3398                     if( pid->es->fmt.p_extra )
3399                         memcpy( pid->es->fmt.p_extra, p_dr->p_data,
3400                                 p_dr->i_length );
3401
3402 #if defined _DVBPSI_DR_56_H_ && defined DVBPSI_VERSION \
3403                     && DVBPSI_VERSION_INT > ((0<<16)+(1<<8)+5)
3404                     pid->es->fmt.i_group = p_pmt->i_program_number;
3405
3406                     /* In stream output mode, only enable descriptor
3407                      * pass-through. */
3408                     if( !p_demux->out->b_sout )
3409                     {
3410                         uint16_t n, i = 0;
3411                         dvbpsi_teletext_dr_t *sub;
3412
3413                         sub = dvbpsi_DecodeTeletextDr( p_dr );
3414                         if( !sub ) continue;
3415
3416                         /* Each subtitle ES contains n languages,
3417                          * We are going to create n ES for the n tracks */
3418                         for( n = 0; n < sub->i_pages_number; n++ )
3419                         {
3420                             dvbpsi_teletextpage_t *p_page = &sub->p_pages[n];
3421                             if( p_page->i_teletext_type > 0x0 &&
3422                                 p_page->i_teletext_type < 0x6 )
3423                             {
3424                                 ts_es_t *p_es;
3425
3426                                 if( i == 0 )
3427                                 {
3428                                     p_es = pid->es;
3429                                 }
3430                                 else
3431                                 {
3432                                     p_es = malloc( sizeof( ts_es_t ) );
3433                                     if( !p_es ) break;
3434
3435                                     es_format_Copy( &p_es->fmt, &pid->es->fmt );
3436                                     free( p_es->fmt.psz_language );
3437                                     free( p_es->fmt.psz_description );
3438                                     p_es->fmt.psz_language = NULL;
3439                                     p_es->fmt.psz_description = NULL;
3440                                     p_es->id = NULL;
3441                                     p_es->p_pes = NULL;
3442                                     p_es->i_pes_size = 0;
3443                                     p_es->i_pes_gathered = 0;
3444                                     p_es->pp_last = &p_es->p_pes;
3445                                     p_es->p_mpeg4desc = NULL;
3446
3447                                     TAB_APPEND( pid->i_extra_es, pid->extra_es,
3448                                                 p_es );
3449                                 }
3450
3451                                 p_es->fmt.psz_language = malloc( 4 );
3452                                 if( p_es->fmt.psz_language )
3453                                 {
3454                                     memcpy( p_es->fmt.psz_language,
3455                                             p_page->i_iso6392_language_code, 3 );
3456                                     p_es->fmt.psz_language[3] = 0;
3457                                 }
3458                                 p_es->fmt.psz_description = strdup(_(ppsz_teletext_type[p_page->i_teletext_type]));
3459
3460                                 msg_Dbg( p_demux,
3461                                              "    * ttxt type=%s lan=%s page=%d%02x",
3462                                              p_es->fmt.psz_description,
3463                                              p_es->fmt.psz_language,
3464                                              p_page->i_teletext_magazine_number,
3465                                              p_page->i_teletext_page_number );
3466
3467                                 /* This stores the initial page for this track,
3468                                    so that it can be used by the telx and zvbi decoders. */
3469                                 p_es->fmt.subs.teletext.i_magazine =
3470                                     p_page->i_teletext_magazine_number ? : 8;
3471                                 p_es->fmt.subs.teletext.i_page =
3472                                     p_page->i_teletext_page_number;
3473
3474                                 i++;
3475                             }
3476                         }
3477                         if( !i )
3478                             pid->es->fmt.i_cat = UNKNOWN_ES;
3479                     }
3480                     else
3481 #endif  /* defined _DVBPSI_DR_56_H_  && DVBPSI_VERSION(0,1,6) */
3482                     {
3483                         pid->es->fmt.subs.teletext.i_magazine = -1;
3484                         pid->es->fmt.subs.teletext.i_page = 0;
3485                         pid->es->fmt.psz_description = strdup( _(ppsz_teletext_type[1]) );
3486                     }
3487                 }
3488                 else if( p_dr->i_tag == 0x59 )
3489                 {
3490                     /* DVB subtitles */
3491                     pid->es->fmt.i_cat = SPU_ES;
3492                     pid->es->fmt.i_codec = VLC_FOURCC( 'd', 'v', 'b', 's' );
3493                     pid->es->fmt.i_extra = p_dr->i_length;
3494                     pid->es->fmt.p_extra = malloc( p_dr->i_length );
3495                     if( pid->es->fmt.p_extra )
3496                         memcpy( pid->es->fmt.p_extra, p_dr->p_data,
3497                                 p_dr->i_length );
3498
3499 #ifdef _DVBPSI_DR_59_H_
3500                     pid->es->fmt.i_group = p_pmt->i_program_number;
3501
3502                     /* In stream output mode, only enable descriptor
3503                      * pass-through. */
3504                     if( !p_demux->out->b_sout )
3505                     {
3506                         uint16_t n, i = 0;
3507                         dvbpsi_subtitling_dr_t *sub;
3508
3509                         sub = dvbpsi_DecodeSubtitlingDr( p_dr );
3510                         if( !sub ) continue;
3511
3512                         for( n = 0; n < sub->i_subtitles_number; n++ )
3513                         {
3514                             dvbpsi_subtitle_t *p_sub = &sub->p_subtitle[n];
3515                             ts_es_t *p_es;
3516
3517                             if( i == 0 )
3518                             {
3519                                 p_es = pid->es;
3520                             }
3521                             else
3522                             {
3523                                 p_es = malloc( sizeof( ts_es_t ) );
3524                                 if( !p_es ) break;
3525                                 es_format_Copy( &p_es->fmt, &pid->es->fmt );
3526                                 free( p_es->fmt.psz_language ); p_es->fmt.psz_language = NULL;
3527                                 free( p_es->fmt.psz_description ); p_es->fmt.psz_description = NULL;
3528
3529                                 p_es->id = NULL;
3530                                 p_es->p_pes = NULL;
3531                                 p_es->i_pes_size = 0;
3532                                 p_es->i_pes_gathered = 0;
3533                                 p_es->pp_last = &p_es->p_pes;
3534                                 p_es->p_mpeg4desc = NULL;
3535
3536                                 TAB_APPEND( pid->i_extra_es, pid->extra_es,
3537                                             p_es );
3538                             }
3539
3540                             p_es->fmt.psz_language = malloc( 4 );
3541                             if( p_es->fmt.psz_language )
3542                             {
3543                                 memcpy( p_es->fmt.psz_language,
3544                                         p_sub->i_iso6392_language_code, 3 );
3545                                 p_es->fmt.psz_language[3] = 0;
3546                             }
3547
3548                             switch( p_sub->i_subtitling_type )
3549                             {
3550                             case 0x10: /* unspec. */
3551                             case 0x11: /* 4:3 */
3552                             case 0x12: /* 16:9 */
3553                             case 0x13: /* 2.21:1 */
3554                                 p_es->fmt.psz_description =
3555                                     strdup(_("DVB subtitles"));
3556                                 break;
3557                             case 0x20: /* Hearing impaired unspec. */
3558                             case 0x21: /* h.i. 4:3 */
3559                             case 0x22: /* h.i. 16:9 */
3560                             case 0x23: /* h.i. 2.21:1 */
3561                                 p_es->fmt.psz_description =
3562                                     strdup(_("DVB subtitles: hearing impaired"));
3563                                 break;
3564                             default:
3565                                 break;
3566                             }
3567
3568                             p_es->fmt.subs.dvb.i_id =
3569                                 p_sub->i_composition_page_id;
3570                             /* Hack, FIXME */
3571                             p_es->fmt.subs.dvb.i_id |=
3572                               ((int)p_sub->i_ancillary_page_id << 16);
3573
3574                             i++;
3575                         }
3576
3577                         if( !i )
3578                             pid->es->fmt.i_cat = UNKNOWN_ES;
3579                     }
3580                     else
3581 #endif /* _DVBPSI_DR_59_H_ */
3582                     {
3583                         pid->es->fmt.subs.dvb.i_id = -1;
3584                         pid->es->fmt.psz_description = strdup( _("DVB subtitles") );
3585                     }
3586                 }
3587             }
3588         }
3589         else if( p_es->i_type == 0xEA )
3590         {
3591             dvbpsi_descriptor_t *p_dr;
3592
3593             for( p_dr = p_es->p_first_descriptor; p_dr != NULL;
3594                  p_dr = p_dr->p_next )
3595             {
3596                 msg_Dbg( p_demux, "  * es pid=%d type=%d dr->i_tag=0x%x",
3597                          p_es->i_pid, p_es->i_type, p_dr->i_tag );
3598
3599                 if( p_dr->i_tag == 0x05 )
3600                 {
3601                     /* Registration Descriptor */
3602                     if( p_dr->i_length < 4 ) // XXX VC-1 has extended this descriptor with sub-descriptor
3603                     {
3604                         msg_Warn( p_demux, "invalid Registration Descriptor" );
3605                     }
3606                     else
3607                     {
3608                         if( !memcmp( p_dr->p_data, "VC-1", 4 ) )
3609                         {
3610                             /* registration descriptor for VC-1 (SMPTE rp227) */
3611                             pid->es->fmt.i_cat = VIDEO_ES;
3612                             pid->es->fmt.i_codec = VLC_FOURCC('W','V','C','1');
3613
3614                             /* XXX With Simple and Main profile the SEQUENCE
3615                              * header is modified: video width and height are
3616                              * inserted just after the start code as 2 int16_t
3617                              * The packetizer will take care of that. */
3618                         }
3619                         else
3620                         {
3621                             msg_Warn( p_demux,
3622                                       "unknown Registration Descriptor (%4.4s)",
3623                                       p_dr->p_data );
3624                         }
3625                     }
3626                 }
3627             }
3628         }
3629         else if( p_es->i_type == 0xd1 )
3630         {
3631             dvbpsi_descriptor_t *p_dr;
3632
3633             for( p_dr = p_es->p_first_descriptor; p_dr != NULL;
3634                  p_dr = p_dr->p_next )
3635             {
3636                 msg_Dbg( p_demux, "  * es pid=%d type=%d dr->i_tag=0x%x",
3637                          p_es->i_pid, p_es->i_type, p_dr->i_tag );
3638
3639                 if( p_dr->i_tag == 0x05 )
3640                 {
3641                     /* Registration Descriptor */
3642                     if( !memcmp( p_dr->p_data, "drac", 4 ) )
3643                     {
3644                         /* registration descriptor for Dirac
3645                          * (backwards compatable with VC-2 (SMPTE Sxxxx:2008)) */
3646                         pid->es->fmt.i_cat = VIDEO_ES;
3647                         pid->es->fmt.i_codec = VLC_FOURCC('d','r','a','c');
3648                     }
3649                     else
3650                     {
3651                         msg_Warn( p_demux,
3652                                   "unknown Registration Descriptor (%4.4s)",
3653                                   p_dr->p_data );
3654                     }
3655                 }
3656             }
3657         }
3658         else if( p_es->i_type == 0xa0 )
3659         {
3660             /* MSCODEC sent by vlc */
3661             dvbpsi_descriptor_t *p_dr = p_es->p_first_descriptor;
3662
3663             while( p_dr && ( p_dr->i_tag != 0xa0 ) ) p_dr = p_dr->p_next;
3664
3665             if( p_dr && p_dr->i_length >= 8 )
3666             {
3667                 pid->es->fmt.i_cat = VIDEO_ES;
3668                 pid->es->fmt.i_codec =
3669                     VLC_FOURCC( p_dr->p_data[0], p_dr->p_data[1],
3670                                 p_dr->p_data[2], p_dr->p_data[3] );
3671                 pid->es->fmt.video.i_width =
3672                     ( p_dr->p_data[4] << 8 ) | p_dr->p_data[5];
3673                 pid->es->fmt.video.i_height =
3674                     ( p_dr->p_data[6] << 8 ) | p_dr->p_data[7];
3675                 pid->es->fmt.i_extra =
3676                     (p_dr->p_data[8] << 8) | p_dr->p_data[9];
3677
3678                 if( pid->es->fmt.i_extra > 0 )
3679                 {
3680                     pid->es->fmt.p_extra = malloc( pid->es->fmt.i_extra );
3681                     if( pid->es->fmt.p_extra )
3682                         memcpy( pid->es->fmt.p_extra, &p_dr->p_data[10],
3683                                 pid->es->fmt.i_extra );
3684                 }
3685             }
3686             else
3687             {
3688                 msg_Warn( p_demux, "private MSCODEC (vlc) without bih private "
3689                           "descriptor" );
3690             }
3691             /* For such stream we will gather them ourself and don't launch a
3692              * packetizer.
3693              * Yes it's ugly but it's the only way to have DIV3 working */
3694             pid->es->fmt.b_packetized = true;
3695         }
3696         else if( b_hdmv )
3697         {
3698             /* Blu-Ray mapping */
3699             switch( p_es->i_type )
3700             {
3701             case 0x80:
3702                 pid->es->fmt.i_cat = AUDIO_ES;
3703                 pid->es->fmt.i_codec = VLC_FOURCC( 'b', 'p', 'c', 'm' );
3704                 break;
3705             case 0x82:
3706             case 0x85: /* DTS-HD High resolution audio */
3707             case 0x86: /* DTS-HD Master audio */
3708             case 0xA2: /* Secondary DTS audio */
3709                 pid->es->fmt.i_cat = AUDIO_ES;
3710                 pid->es->fmt.i_codec = VLC_FOURCC( 'd', 't', 's', ' ' );
3711                 break;
3712
3713             case 0x83: /* TrueHD AC3 */
3714                 pid->es->fmt.i_cat = AUDIO_ES;
3715                 pid->es->fmt.i_codec = VLC_FOURCC( 'm', 'l', 'p', ' ' );
3716                 break;
3717
3718             case 0x84: /* E-AC3 */
3719             case 0x87: /* E-AC3 */
3720             case 0xA1: /* Secondary E-AC3 */
3721                 pid->es->fmt.i_cat = AUDIO_ES;
3722                 pid->es->fmt.i_codec = VLC_FOURCC( 'e', 'a', 'c', '3' );
3723                 break;
3724             case 0x90: /* Presentation graphics */
3725             case 0x91: /* Interactive graphics */
3726             case 0x92: /* Subtitle */
3727             default:
3728                 break;
3729             }
3730         }
3731
3732         if( pid->es->fmt.i_cat == AUDIO_ES ||
3733             ( pid->es->fmt.i_cat == SPU_ES &&
3734               pid->es->fmt.i_codec != VLC_FOURCC('d','v','b','s') &&
3735               pid->es->fmt.i_codec != VLC_FOURCC('t','e','l','x') ) )
3736         {
3737             /* get language descriptor */
3738             dvbpsi_descriptor_t *p_dr = p_es->p_first_descriptor;
3739             while( p_dr && ( p_dr->i_tag != 0x0a ) ) p_dr = p_dr->p_next;
3740
3741             if( p_dr )
3742             {
3743                 dvbpsi_iso639_dr_t *p_decoded = dvbpsi_DecodeISO639Dr( p_dr );
3744
3745                 if( p_decoded )
3746                 {
3747 #if defined(DR_0A_API_VER) && (DR_0A_API_VER >= 2)
3748                     pid->es->fmt.psz_language = malloc( 4 );
3749                     if( pid->es->fmt.psz_language )
3750                     {
3751                         memcpy( pid->es->fmt.psz_language,
3752                                 p_decoded->code[0].iso_639_code, 3 );
3753                         pid->es->fmt.psz_language[3] = 0;
3754                         msg_Dbg( p_demux, "found language: %s", pid->es->fmt.psz_language);
3755                     }
3756                     switch( p_decoded->code[0].i_audio_type ) {
3757                     case 0:
3758                         pid->es->fmt.psz_description = NULL;
3759                         break;
3760                     case 1:
3761                         pid->es->fmt.psz_description =
3762                             strdup(_("clean effects"));
3763                         break;
3764                     case 2:
3765                         pid->es->fmt.psz_description =
3766                             strdup(_("hearing impaired"));
3767                         break;
3768                     case 3:
3769                         pid->es->fmt.psz_description =
3770                             strdup(_("visual impaired commentary"));
3771                         break;
3772                     default:
3773                         msg_Dbg( p_demux, "unknown audio type: %d",
3774                                  p_decoded->code[0].i_audio_type);
3775                         pid->es->fmt.psz_description = NULL;
3776                         break;
3777                     }
3778                     pid->es->fmt.i_extra_languages = p_decoded->i_code_count-1;
3779                     if( pid->es->fmt.i_extra_languages > 0 )
3780                         pid->es->fmt.p_extra_languages =
3781                             malloc( sizeof(*pid->es->fmt.p_extra_languages) *
3782                                     pid->es->fmt.i_extra_languages );
3783                     if( pid->es->fmt.p_extra_languages )
3784                     {
3785                         for( i = 0; i < pid->es->fmt.i_extra_languages; i++ )
3786                         {
3787                             msg_Dbg( p_demux, "bang" );
3788                             pid->es->fmt.p_extra_languages[i].psz_language =
3789                                 malloc(4);
3790                             if( pid->es->fmt.p_extra_languages[i].psz_language )
3791                             {
3792                                 memcpy( pid->es->fmt.p_extra_languages[i].psz_language,
3793                                     p_decoded->code[i+1].iso_639_code, 3 );
3794                                 pid->es->fmt.p_extra_languages[i].psz_language[3] = '\0';
3795                             }
3796                             switch( p_decoded->code[i].i_audio_type ) {
3797                             case 0:
3798                                 pid->es->fmt.p_extra_languages[i].psz_description =
3799                                     NULL;
3800                                 break;
3801                             case 1:
3802                                 pid->es->fmt.p_extra_languages[i].psz_description =
3803                                     strdup(_("clean effects"));
3804                                 break;
3805                             case 2:
3806                                 pid->es->fmt.p_extra_languages[i].psz_description =
3807                                     strdup(_("hearing impaired"));
3808                                 break;
3809                             case 3:
3810                                 pid->es->fmt.p_extra_languages[i].psz_description =
3811                                     strdup(_("visual impaired commentary"));
3812                                 break;
3813                             default:
3814                                 msg_Dbg( p_demux, "unknown audio type: %d",
3815                                         p_decoded->code[i].i_audio_type);
3816                                 pid->es->fmt.psz_description = NULL;
3817                                 break;
3818                             }
3819
3820                         }
3821                     }
3822 #else
3823                     pid->es->fmt.psz_language = malloc( 4 );
3824                     if( pid->es->fmt.psz_language )
3825                     {
3826                         memcpy( pid->es->fmt.psz_language,
3827                                 p_decoded->i_iso_639_code, 3 );
3828                         pid->es->fmt.psz_language[3] = 0;
3829                     }
3830 #endif
3831                 }
3832             }
3833         }
3834
3835         pid->es->fmt.i_group = p_pmt->i_program_number;
3836         if( pid->es->fmt.i_cat == UNKNOWN_ES )
3837         {
3838             msg_Dbg( p_demux, "  * es pid=%d type=%d *unknown*",
3839                      p_es->i_pid, p_es->i_type );
3840         }
3841         else if( !p_sys->b_udp_out )
3842         {
3843             msg_Dbg( p_demux, "  * es pid=%d type=%d fcc=%4.4s",
3844                      p_es->i_pid, p_es->i_type, (char*)&pid->es->fmt.i_codec );
3845
3846             if( p_sys->b_es_id_pid ) pid->es->fmt.i_id = p_es->i_pid;
3847
3848             /* Check if we can avoid restarting the ES */
3849             if( old_pid &&
3850                 pid->es->fmt.i_codec == old_pid->es->fmt.i_codec &&
3851                 pid->es->fmt.i_extra == old_pid->es->fmt.i_extra &&
3852                 pid->es->fmt.i_extra == 0 &&
3853                 pid->i_extra_es == old_pid->i_extra_es &&
3854                 ( ( !pid->es->fmt.psz_language &&
3855                     !old_pid->es->fmt.psz_language ) ||
3856                   ( pid->es->fmt.psz_language &&
3857                     old_pid->es->fmt.psz_language &&
3858                     !strcmp( pid->es->fmt.psz_language,
3859                              old_pid->es->fmt.psz_language ) ) ) )
3860             {
3861                 pid->es->id = old_pid->es->id;
3862                 old_pid->es->id = NULL;
3863                 for( i = 0; i < pid->i_extra_es; i++ )
3864                 {
3865                     pid->extra_es[i]->id = old_pid->extra_es[i]->id;
3866                     old_pid->extra_es[i]->id = NULL;
3867                 }
3868             }
3869             else
3870             {
3871                 if( old_pid )
3872                 {
3873                     PIDClean( p_demux->out, old_pid );
3874                     TAB_REMOVE( i_clean, pp_clean, old_pid );
3875                     old_pid = 0;
3876                 }
3877
3878                 pid->es->id = es_out_Add( p_demux->out, &pid->es->fmt );
3879                 for( i = 0; i < pid->i_extra_es; i++ )
3880                 {
3881                     pid->extra_es[i]->id =
3882                         es_out_Add( p_demux->out, &pid->extra_es[i]->fmt );
3883                 }
3884             }
3885         }
3886
3887         /* Add ES to the list */
3888         if( old_pid )
3889         {
3890             PIDClean( p_demux->out, old_pid );
3891             TAB_REMOVE( i_clean, pp_clean, old_pid );
3892         }
3893         p_sys->pid[p_es->i_pid] = *pid;
3894
3895         for( p_dr = p_es->p_first_descriptor; p_dr != NULL;
3896              p_dr = p_dr->p_next )
3897         {
3898             if( p_dr->i_tag == 0x9 )
3899             {
3900                 uint16_t i_sysid = ((uint16_t)p_dr->p_data[0] << 8)
3901                                     | p_dr->p_data[1];
3902                 msg_Dbg( p_demux, "   * descriptor : CA (0x9) SysID 0x%x",
3903                          i_sysid );
3904             }
3905         }
3906
3907         if( DVBProgramIsSelected( p_demux, prg->i_number )
3908              && (pid->es->id != NULL || p_sys->b_udp_out) )
3909         {
3910             /* Set demux filter */
3911             stream_Control( p_demux->s, STREAM_CONTROL_ACCESS,
3912                             ACCESS_SET_PRIVATE_ID_STATE, p_es->i_pid,
3913                             true );
3914         }
3915     }
3916
3917     if( DVBProgramIsSelected( p_demux, prg->i_number ) )
3918     {
3919         /* Set CAM descrambling */
3920         stream_Control( p_demux->s, STREAM_CONTROL_ACCESS,
3921                         ACCESS_SET_PRIVATE_ID_CA, p_pmt );
3922     }
3923     else
3924     {
3925         dvbpsi_DeletePMT( p_pmt );
3926     }
3927
3928     for ( i = 0; i < i_clean; i++ )
3929     {
3930         if( DVBProgramIsSelected( p_demux, prg->i_number ) )
3931         {
3932             stream_Control( p_demux->s, STREAM_CONTROL_ACCESS,
3933                             ACCESS_SET_PRIVATE_ID_STATE, pp_clean[i]->i_pid,
3934                             false );
3935         }
3936
3937         PIDClean( p_demux->out, pp_clean[i] );
3938     }
3939     if( i_clean ) free( pp_clean );
3940 }
3941
3942 static void PATCallBack( demux_t *p_demux, dvbpsi_pat_t *p_pat )
3943 {
3944     demux_sys_t          *p_sys = p_demux->p_sys;
3945     dvbpsi_pat_program_t *p_program;
3946     ts_pid_t             *pat = &p_sys->pid[0];
3947     int                  i, j;
3948
3949     msg_Dbg( p_demux, "PATCallBack called" );
3950
3951     if( ( pat->psi->i_pat_version != -1 &&
3952             ( !p_pat->b_current_next ||
3953               p_pat->i_version == pat->psi->i_pat_version ) ) ||
3954         p_sys->b_user_pmt )
3955     {
3956         dvbpsi_DeletePAT( p_pat );
3957         return;
3958     }
3959
3960     msg_Dbg( p_demux, "new PAT ts_id=%d version=%d current_next=%d",
3961              p_pat->i_ts_id, p_pat->i_version, p_pat->b_current_next );
3962
3963     /* Clean old */
3964     if( p_sys->i_pmt > 0 )
3965     {
3966         int      i_pmt_rm = 0;
3967         ts_pid_t **pmt_rm = NULL;
3968
3969         /* Search pmt to be deleted */
3970         for( i = 0; i < p_sys->i_pmt; i++ )
3971         {
3972             ts_pid_t *pmt = p_sys->pmt[i];
3973             bool b_keep = false;
3974
3975             for( p_program = p_pat->p_first_program; p_program != NULL;
3976                  p_program = p_program->p_next )
3977             {
3978                 if( p_program->i_pid == pmt->i_pid )
3979                 {
3980                     int i_prg;
3981                     for( i_prg = 0; i_prg < pmt->psi->i_prg; i_prg++ )
3982                     {
3983                         if( p_program->i_number ==
3984                             pmt->psi->prg[i_prg]->i_number )
3985                         {
3986                             b_keep = true;
3987                             break;
3988                         }
3989                     }
3990                     if( b_keep ) break;
3991                 }
3992             }
3993
3994             if( !b_keep )
3995             {
3996                 TAB_APPEND( i_pmt_rm, pmt_rm, pmt );
3997             }
3998         }
3999
4000         /* Delete all ES attached to thoses PMT */
4001         for( i = 2; i < 8192; i++ )
4002         {
4003             ts_pid_t *pid = &p_sys->pid[i];
4004
4005             if( !pid->b_valid || pid->psi ) continue;
4006
4007             for( j = 0; j < i_pmt_rm && pid->b_valid; j++ )
4008             {
4009                 int i_prg;
4010                 for( i_prg = 0; i_prg < pid->p_owner->i_prg; i_prg++ )
4011                 {
4012                     /* We only remove es that aren't defined by extra pmt */
4013                     if( pid->p_owner->prg[i_prg]->i_pid_pmt !=
4014                         pmt_rm[j]->i_pid ) continue;
4015
4016                     if( p_sys->b_dvb_control && pid->es->id )
4017                     {
4018                         if( stream_Control( p_demux->s, STREAM_CONTROL_ACCESS,
4019                                             ACCESS_SET_PRIVATE_ID_STATE, i,
4020                                             false ) )
4021                             p_sys->b_dvb_control = false;
4022                     }
4023
4024                     PIDClean( p_demux->out, pid );
4025                     break;
4026                 }
4027             }
4028         }
4029
4030         /* Delete PMT pid */
4031         for( i = 0; i < i_pmt_rm; i++ )
4032         {
4033             int i_prg;
4034             if( p_sys->b_dvb_control )
4035             {
4036                 if( stream_Control( p_demux->s, STREAM_CONTROL_ACCESS,
4037                                     ACCESS_SET_PRIVATE_ID_STATE,
4038                                     pmt_rm[i]->i_pid, false ) )
4039                     p_sys->b_dvb_control = false;
4040             }
4041
4042             for( i_prg = 0; i_prg < pmt_rm[i]->psi->i_prg; i_prg++ )
4043             {
4044                 const int i_number = pmt_rm[i]->psi->prg[i_prg]->i_number;
4045                 es_out_Control( p_demux->out, ES_OUT_DEL_GROUP, i_number );
4046             }
4047
4048             PIDClean( p_demux->out, &p_sys->pid[pmt_rm[i]->i_pid] );
4049             TAB_REMOVE( p_sys->i_pmt, p_sys->pmt, pmt_rm[i] );
4050         }
4051
4052         free( pmt_rm );
4053     }
4054
4055     /* now create programs */
4056     for( p_program = p_pat->p_first_program; p_program != NULL;
4057          p_program = p_program->p_next )
4058     {
4059         msg_Dbg( p_demux, "  * number=%d pid=%d", p_program->i_number,
4060                  p_program->i_pid );
4061         if( p_program->i_number != 0 )
4062         {
4063             ts_pid_t *pmt = &p_sys->pid[p_program->i_pid];
4064             bool b_add = true;
4065
4066             if( pmt->b_valid )
4067             {
4068                 int i_prg;
4069                 for( i_prg = 0; i_prg < pmt->psi->i_prg; i_prg++ )
4070                 {
4071                     if( pmt->psi->prg[i_prg]->i_number == p_program->i_number )
4072                     {
4073                         b_add = false;
4074                         break;
4075                     }
4076                 }
4077             }
4078             else
4079             {
4080                 TAB_APPEND( p_sys->i_pmt, p_sys->pmt, pmt );
4081             }
4082
4083             if( b_add )
4084             {
4085                 PIDInit( pmt, true, pat->psi );
4086                 pmt->psi->prg[pmt->psi->i_prg-1]->handle =
4087                     dvbpsi_AttachPMT( p_program->i_number,
4088                                       (dvbpsi_pmt_callback)PMTCallBack,
4089                                       p_demux );
4090                 pmt->psi->prg[pmt->psi->i_prg-1]->i_number =
4091                     p_program->i_number;
4092                 pmt->psi->prg[pmt->psi->i_prg-1]->i_pid_pmt =
4093                     p_program->i_pid;
4094
4095                 /* Now select PID at access level */
4096                 if( p_sys->b_dvb_control )
4097                 {
4098                     if( DVBProgramIsSelected( p_demux, p_program->i_number ) )
4099                     {
4100                         if( p_sys->i_dvb_program == 0 )
4101                             p_sys->i_dvb_program = p_program->i_number;
4102
4103                         if( stream_Control( p_demux->s, STREAM_CONTROL_ACCESS,
4104                                             ACCESS_SET_PRIVATE_ID_STATE,
4105                                             p_program->i_pid, true ) )
4106                             p_sys->b_dvb_control = false;
4107                     }
4108                 }
4109             }
4110         }
4111     }
4112     pat->psi->i_pat_version = p_pat->i_version;
4113
4114     dvbpsi_DeletePAT( p_pat );
4115 }