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