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