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