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