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