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