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