ts.c 124 KB
Newer Older
1
2
3
/*****************************************************************************
 * ts.c: Transport Stream input module for VLC.
 *****************************************************************************
zorglub's avatar
zorglub committed
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
11
12
13
14
15
16
17
18
19
20
21
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
dionoea's avatar
dionoea committed
22
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston MA 02110-1301, USA.
23
24
25
26
27
 *****************************************************************************/

/*****************************************************************************
 * Preamble
 *****************************************************************************/
Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
28
29
30
31

#include <vlc/vlc.h>

#include <stdio.h>
32
#include <stdlib.h>                                      /* malloc(), free() */
Laurent Aimar's avatar
Laurent Aimar committed
33
#include <ctype.h>
34

zorglub's avatar
zorglub committed
35
36
37
#include <vlc_access.h> /* DVB-specific things */
#include <vlc_demux.h>
#include <vlc_meta.h>
38
39

#include "iso_lang.h"
zorglub's avatar
zorglub committed
40
41
#include <vlc_network.h>
#include <vlc_charset.h>
42

43
44
#include "../mux/mpeg/csa.h"

45
46
47
/* Include dvbpsi headers */
#ifdef HAVE_DVBPSI_DR_H
#   include <dvbpsi/dvbpsi.h>
48
#   include <dvbpsi/demux.h>
49
50
51
#   include <dvbpsi/descriptor.h>
#   include <dvbpsi/pat.h>
#   include <dvbpsi/pmt.h>
52
#   include <dvbpsi/sdt.h>
53
54
55
56
#   include <dvbpsi/dr.h>
#   include <dvbpsi/psi.h>
#else
#   include "dvbpsi.h"
57
#   include "demux.h"
58
59
60
#   include "descriptor.h"
#   include "tables/pat.h"
#   include "tables/pmt.h"
61
#   include "tables/sdt.h"
62
63
64
65
#   include "descriptors/dr.h"
#   include "psi.h"
#endif

66
67
68
69
70
71
72
73
74
/* EIT support */
#ifdef _DVBPSI_DR_4D_H_
#   define TS_USE_DVB_SI 1
#   ifdef HAVE_DVBPSI_DR_H
#       include <dvbpsi/eit.h>
#   else
#       include "tables/eit.h"
#   endif
#endif
75

76
77
#undef TS_DEBUG

78
/* TODO:
79
 *  - XXX: do not mark options message to be translated, they are too osbcure for now ...
80
81
82
83
 *  - test it
 *  - ...
 */

84
#define vlc_meta_Add(a,b,c) fprintf(stderr, "FIXME: TS demuxer meta is broken\n" )
85
86
87
/*****************************************************************************
 * Module descriptor
 *****************************************************************************/
gbazin's avatar
   
gbazin committed
88
89
static int  Open  ( vlc_object_t * );
static void Close ( vlc_object_t * );
90

91
92
#define PMT_TEXT N_("Extra PMT")
#define PMT_LONGTEXT N_( \
93
  "Allows a user to specify an extra pmt (pmt_pid=pid:stream_type[,...])." )
94
95

#define PID_TEXT N_("Set id of ES to PID")
96
97
98
99
#define PID_LONGTEXT N_("Set the internal ID of each elementary stream" \
                       " handled by VLC to the same value as the PID in" \
                       " the TS stream, instead of 1, 2, 3, etc. Useful to" \
                       " do \'#duplicate{..., select=\"es=<pid>\"}\'.")
100
101
102

#define TSOUT_TEXT N_("Fast udp streaming")
#define TSOUT_LONGTEXT N_( \
103
  "Sends TS to specific ip:port by udp (you must know what you are doing).")
104
105

#define MTUOUT_TEXT N_("MTU for out mode")
106
#define MTUOUT_LONGTEXT N_("MTU for out mode.")
107
108

#define CSA_TEXT N_("CSA ck")
Christophe Mutricy's avatar
Typo    
Christophe Mutricy committed
109
#define CSA_LONGTEXT N_("Control word for the CSA encryption algorithm")
110
111

#define SILENT_TEXT N_("Silent mode")
112
#define SILENT_LONGTEXT N_("Do not complain on encrypted PES.")
113

114
#define CAPMT_SYSID_TEXT N_("CAPMT System ID")
115
#define CAPMT_SYSID_LONGTEXT N_("Only forward descriptors from this SysID to the CAM.")
116

117
118
119
120
121
122
#define CPKT_TEXT N_("Packet size in bytes to decrypt")
#define CPKT_LONGTEXT N_("Specify the size of the TS packet to decrypt. " \
    "The decryption routines subtract the TS-header from the value before " \
    "decrypting. " )

#define TSDUMP_TEXT N_("Filename of dump")
123
#define TSDUMP_LONGTEXT N_("Specify a filename where to dump the TS in.")
124
125
126
127
128
129
130
131
132
133
134

#define APPEND_TEXT N_("Append")
#define APPEND_LONGTEXT N_( \
    "If the file exists and this option is selected, the existing file " \
    "will not be overwritten." )

#define DUMPSIZE_TEXT N_("Dump buffer size")
#define DUMPSIZE_LONGTEXT N_( \
    "Tweak the buffer size for reading and writing an integer number of packets." \
    "Specify the size of the buffer here and not the number of packets." )

135
vlc_module_begin();
136
    set_description( _("MPEG Transport Stream demuxer") );
137
    set_shortname ( "MPEG-TS" );
zorglub's avatar
zorglub committed
138
139
    set_category( CAT_INPUT );
    set_subcategory( SUBCAT_INPUT_DEMUX );
140
141

    add_string( "ts-extra-pmt", NULL, NULL, PMT_TEXT, PMT_LONGTEXT, VLC_TRUE );
142
    add_bool( "ts-es-id-pid", 1, NULL, PID_TEXT, PID_LONGTEXT, VLC_TRUE );
143
144
145
146
    add_string( "ts-out", NULL, NULL, TSOUT_TEXT, TSOUT_LONGTEXT, VLC_TRUE );
    add_integer( "ts-out-mtu", 1500, NULL, MTUOUT_TEXT,
                 MTUOUT_LONGTEXT, VLC_TRUE );
    add_string( "ts-csa-ck", NULL, NULL, CSA_TEXT, CSA_LONGTEXT, VLC_TRUE );
147
    add_integer( "ts-csa-pkt", 188, NULL, CPKT_TEXT, CPKT_LONGTEXT, VLC_TRUE );
148
149
    add_bool( "ts-silent", 0, NULL, SILENT_TEXT, SILENT_LONGTEXT, VLC_TRUE );

150
151
152
153
154
    add_file( "ts-dump-file", NULL, NULL, TSDUMP_TEXT, TSDUMP_LONGTEXT, VLC_FALSE );
    add_bool( "ts-dump-append", 0, NULL, APPEND_TEXT, APPEND_LONGTEXT, VLC_FALSE );
    add_integer( "ts-dump-size", 16384, NULL, DUMPSIZE_TEXT,
                 DUMPSIZE_LONGTEXT, VLC_TRUE );

155
156
    set_capability( "demux2", 10 );
    set_callbacks( Open, Close );
Laurent Aimar's avatar
Laurent Aimar committed
157
    add_shortcut( "ts" );
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
vlc_module_end();

/*****************************************************************************
 * Local prototypes
 *****************************************************************************/

typedef struct
{
    uint8_t                 i_objectTypeIndication;
    uint8_t                 i_streamType;
    vlc_bool_t              b_upStream;
    uint32_t                i_bufferSizeDB;
    uint32_t                i_maxBitrate;
    uint32_t                i_avgBitrate;

    int                     i_decoder_specific_info_len;
    uint8_t                 *p_decoder_specific_info;

} decoder_config_descriptor_t;

typedef struct
{
    vlc_bool_t              b_useAccessUnitStartFlag;
    vlc_bool_t              b_useAccessUnitEndFlag;
    vlc_bool_t              b_useRandomAccessPointFlag;
    vlc_bool_t              b_useRandomAccessUnitsOnlyFlag;
    vlc_bool_t              b_usePaddingFlag;
    vlc_bool_t              b_useTimeStampsFlags;
    vlc_bool_t              b_useIdleFlag;
    vlc_bool_t              b_durationFlag;
    uint32_t                i_timeStampResolution;
    uint32_t                i_OCRResolution;
    uint8_t                 i_timeStampLength;
    uint8_t                 i_OCRLength;
    uint8_t                 i_AU_Length;
    uint8_t                 i_instantBitrateLength;
    uint8_t                 i_degradationPriorityLength;
    uint8_t                 i_AU_seqNumLength;
    uint8_t                 i_packetSeqNumLength;

    uint32_t                i_timeScale;
    uint16_t                i_accessUnitDuration;
    uint16_t                i_compositionUnitDuration;

    uint64_t                i_startDecodingTimeStamp;
    uint64_t                i_startCompositionTimeStamp;

} sl_config_descriptor_t;

typedef struct
{
    vlc_bool_t              b_ok;
    uint16_t                i_es_id;

    vlc_bool_t              b_streamDependenceFlag;
    vlc_bool_t              b_OCRStreamFlag;
    uint8_t                 i_streamPriority;

    char                    *psz_url;

    uint16_t                i_dependOn_es_id;
    uint16_t                i_OCR_es_id;

    decoder_config_descriptor_t    dec_descr;
    sl_config_descriptor_t         sl_descr;
223

224
225
226
227
} es_mpeg4_descriptor_t;

typedef struct
{
228
    uint8_t                i_iod_label, i_iod_label_scope;
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245

    /* IOD */
    uint16_t                i_od_id;
    char                    *psz_url;

    uint8_t                 i_ODProfileLevelIndication;
    uint8_t                 i_sceneProfileLevelIndication;
    uint8_t                 i_audioProfileLevelIndication;
    uint8_t                 i_visualProfileLevelIndication;
    uint8_t                 i_graphicsProfileLevelIndication;

    es_mpeg4_descriptor_t   es_descr[255];

} iod_descriptor_t;

typedef struct
{
246
247
    dvbpsi_handle   handle;

248
249
250
    int             i_version;
    int             i_number;
    int             i_pid_pcr;
251
    int             i_pid_pmt;
252
253
    /* IOD stuff (mpeg4) */
    iod_descriptor_t *iod;
254

255
256
} ts_prg_psi_t;

257
258
259
260
typedef struct
{
    /* for special PAT/SDT case */
    dvbpsi_handle   handle; /* PAT/SDT/EIT */
261
    int             i_pat_version;
262
    int             i_sdt_version;
263
264
265
266

    /* For PMT */
    int             i_prg;
    ts_prg_psi_t    **prg;
267

268
269
270
271
272
273
} ts_psi_t;

typedef struct
{
    es_format_t  fmt;
    es_out_id_t *id;
274
275
    int         i_pes_size;
    int         i_pes_gathered;
276
    block_t     *p_pes;
277
    block_t     **pp_last;
278
279

    es_mpeg4_descriptor_t *p_mpeg4desc;
280
    int         b_gather;
281

282
283
284
285
286
287
288
289
290
291
292
293
} ts_es_t;

typedef struct
{
    int         i_pid;

    vlc_bool_t  b_seen;
    vlc_bool_t  b_valid;
    int         i_cc;   /* countinuity counter */

    /* PSI owner (ie PMT -> PAT, ES -> PMT */
    ts_psi_t   *p_owner;
294
    int         i_owner_number;
295
296
297
298
299

    /* */
    ts_psi_t    *psi;
    ts_es_t     *es;

300
301
302
303
    /* Some private streams encapsulate several ES (eg. DVB subtitles)*/
    ts_es_t     **extra_es;
    int         i_extra_es;

304
305
306
307
} ts_pid_t;

struct demux_sys_t
{
308
309
310
    /* TS packet size (188, 192, 204) */
    int         i_packet_size;

311
312
313
    /* how many TS packet we read at once */
    int         i_ts_read;

314
315
316
317
318
319
    /* All pid */
    ts_pid_t    pid[8192];

    /* All PMT */
    int         i_pmt;
    ts_pid_t    **pmt;
320
321
322

    /* */
    vlc_bool_t  b_es_id_pid;
323
    csa_t       *csa;
324
    int         i_csa_pkt_size;
325
    vlc_bool_t  b_silent;
326
327
328
329

    vlc_bool_t  b_udp_out;
    int         fd; /* udp socket */
    uint8_t     *buffer;
330
331
332

    vlc_bool_t  b_dvb_control;
    int         i_dvb_program;
333
    vlc_list_t  *p_programs_list;
334

335
336
337
338
339
    /* TS dump */
    char        *psz_file;  /* file to dump data in */
    FILE        *p_file;    /* filehandle */
    uint64_t    i_write;    /* bytes written */
    vlc_bool_t  b_file_out; /* dump mode enabled */
340

341
342
    /* */
    vlc_bool_t  b_meta;
343
344
};

345
346
static int Demux    ( demux_t *p_demux );
static int DemuxFile( demux_t *p_demux );
347
348
static int Control( demux_t *p_demux, int i_query, va_list args );

349
350
351
352
static void PIDInit ( ts_pid_t *pid, vlc_bool_t b_psi, ts_psi_t *p_owner );
static void PIDClean( es_out_t *out, ts_pid_t *pid );
static int  PIDFillFormat( ts_pid_t *pid, int i_stream_type );

353
354
static void PATCallBack( demux_t *, dvbpsi_pat_t * );
static void PMTCallBack( demux_t *p_demux, dvbpsi_pmt_t *p_pmt );
355
#ifdef TS_USE_DVB_SI
356
357
static void PSINewTableCallBack( demux_t *, dvbpsi_handle,
                                 uint8_t  i_table_id, uint16_t i_extension );
358
#endif
359

360
361
362
363
364
365
366
static inline int PIDGet( block_t *p )
{
    return ( (p->p_buffer[1]&0x1f)<<8 )|p->p_buffer[2];
}

static vlc_bool_t GatherPES( demux_t *p_demux, ts_pid_t *pid, block_t *p_bk );

367
368
static void PCRHandle( demux_t *p_demux, ts_pid_t *, block_t * );

369
static iod_descriptor_t *IODNew( int , uint8_t * );
370
static void              IODFree( iod_descriptor_t * );
371

372
373
374
375
#define TS_PACKET_SIZE_188 188
#define TS_PACKET_SIZE_192 192
#define TS_PACKET_SIZE_204 204
#define TS_PACKET_SIZE_MAX 204
376
#define TS_TOPFIELD_HEADER 1320
377
378
379
380
381
382
383
384
385
386

/*****************************************************************************
 * Open
 *****************************************************************************/
static int Open( vlc_object_t *p_this )
{
    demux_t     *p_demux = (demux_t*)p_this;
    demux_sys_t *p_sys;

    uint8_t     *p_peek;
387
388
    int          i_sync, i_peek, i;
    int          i_packet_size;
389

390
    ts_pid_t    *pat;
zorglub's avatar
zorglub committed
391
    const char  *psz_mode;
392
    vlc_bool_t   b_append;
393
    vlc_bool_t   b_topfield = VLC_FALSE;
394
395
396

    vlc_value_t  val;

397
    if( stream_Peek( p_demux->s, &p_peek, TS_PACKET_SIZE_MAX ) <
398
        TS_PACKET_SIZE_MAX ) return VLC_EGENERIC;
399

400
401
402
403
404
405
406
    if( p_peek[0] == 'T' && p_peek[1] == 'F' &&
        p_peek[2] == 'r' && p_peek[3] == 'c' )
    {
        b_topfield = VLC_TRUE;
        msg_Dbg( p_demux, "this is a topfield file" );
    }

407
408
    /* Search first sync byte */
    for( i_sync = 0; i_sync < TS_PACKET_SIZE_MAX; i_sync++ )
409
410
411
    {
        if( p_peek[i_sync] == 0x47 ) break;
    }
412
    if( i_sync >= TS_PACKET_SIZE_MAX && !b_topfield )
413
    {
414
        if( strcmp( p_demux->psz_demux, "ts" ) ) return VLC_EGENERIC;
415
416
        msg_Warn( p_demux, "this does not look like a TS stream, continuing" );
    }
417

418
419
420
421
422
423
424
425
426
427
428
    if( b_topfield )
    {
        /* Read the entire Topfield header */
        i_peek = TS_TOPFIELD_HEADER;
    }
    else
    {
        /* Check next 3 sync bytes */
        i_peek = TS_PACKET_SIZE_MAX * 3 + i_sync + 1;
    }
    
429
    if( ( stream_Peek( p_demux->s, &p_peek, i_peek ) ) < i_peek )
430
    {
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
        msg_Err( p_demux, "cannot peek" );
        return VLC_EGENERIC;
    }
    if( p_peek[i_sync + TS_PACKET_SIZE_188] == 0x47 &&
        p_peek[i_sync + 2 * TS_PACKET_SIZE_188] == 0x47 &&
        p_peek[i_sync + 3 * TS_PACKET_SIZE_188] == 0x47 )
    {
        i_packet_size = TS_PACKET_SIZE_188;
    }
    else if( p_peek[i_sync + TS_PACKET_SIZE_192] == 0x47 &&
             p_peek[i_sync + 2 * TS_PACKET_SIZE_192] == 0x47 &&
             p_peek[i_sync + 3 * TS_PACKET_SIZE_192] == 0x47 )
    {
        i_packet_size = TS_PACKET_SIZE_192;
    }
    else if( p_peek[i_sync + TS_PACKET_SIZE_204] == 0x47 &&
             p_peek[i_sync + 2 * TS_PACKET_SIZE_204] == 0x47 &&
             p_peek[i_sync + 3 * TS_PACKET_SIZE_204] == 0x47 )
    {
        i_packet_size = TS_PACKET_SIZE_204;
    }
    else if( !strcmp( p_demux->psz_demux, "ts" ) )
    {
        i_packet_size = TS_PACKET_SIZE_188;
    }
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
    else if( b_topfield )
    {
        i_packet_size = TS_PACKET_SIZE_188;
#if 0
        /* I used the TF5000PVR 2004 Firmware .doc header documentation, 
         * http://www.i-topfield.com/data/product/firmware/Structure%20of%20Recorded%20File%20in%20TF5000PVR%20(Feb%2021%202004).doc
         * but after the filename the offsets seem to be incorrect.  - DJ */
        int i_duration, i_name;
        char *psz_name = malloc(25);
        char *psz_event_name;
        char *psz_event_text = malloc(130);
        char *psz_ext_text = malloc(1025);

        // 2 bytes version Uimsbf (4,5)
        // 2 bytes reserved (6,7)
        // 2 bytes duration in minutes Uimsbf (8,9(
        i_duration = (int) (p_peek[8] << 8) | p_peek[9];
        msg_Dbg( p_demux, "Topfield recording length: +/- %d minutes", i_duration);
        // 2 bytes service number in channel list (10, 11)
        // 2 bytes service type Bslbf 0=TV 1=Radio Bslb (12, 13)
        // 4 bytes of reserved + tuner info (14,15,16,17)
        // 2 bytes of Service ID  Bslbf (18,19)
        // 2 bytes of PMT PID  Uimsbf (20,21)
        // 2 bytes of PCR PID  Uimsbf (22,23)
        // 2 bytes of Video PID  Uimsbf (24,25)
        // 2 bytes of Audio PID  Uimsbf (26,27)
        // 24 bytes filename Bslbf
        memcpy( psz_name, &p_peek[28], 24 );
        psz_name[24] = '\0';
        msg_Dbg( p_demux, "recordingname=%s", psz_name );
        // 1 byte of sat index Uimsbf  (52)
        // 3 bytes (1 bit of polarity Bslbf +23 bits reserved)
        // 4 bytes of freq. Uimsbf (56,57,58,59)
        // 2 bytes of symbol rate Uimsbf (60,61)
        // 2 bytes of TS stream ID Uimsbf (62,63)
        // 4 bytes reserved 
        // 2 bytes reserved
        // 2 bytes duration Uimsbf (70,71)
        //i_duration = (int) (p_peek[70] << 8) | p_peek[71];
        //msg_Dbg( p_demux, "Topfield 2nd duration field: +/- %d minutes", i_duration);
        // 4 bytes EventID Uimsbf (72-75)
        // 8 bytes of Start and End time info (76-83)
        // 1 byte reserved (84)
        // 1 byte event name length Uimsbf (89)
        i_name = (int)(p_peek[89]&~0x81);
        msg_Dbg( p_demux, "event name length = %d", i_name);
        psz_event_name = malloc( i_name+1 );
        // 1 byte parental rating (90)
        // 129 bytes of event text
        memcpy( psz_event_name, &p_peek[91], i_name );
        psz_event_name[i_name] = '\0';
        memcpy( psz_event_text, &p_peek[91+i_name], 129-i_name );
        psz_event_text[129-i_name] = '\0';
        msg_Dbg( p_demux, "event name=%s", psz_event_name );
        msg_Dbg( p_demux, "event text=%s", psz_event_text );
        // 12 bytes reserved (220)
        // 6 bytes reserved
        // 2 bytes Event Text Length Uimsbf
        // 4 bytes EventID Uimsbf
        // FIXME We just have 613 bytes. not enough for this entire text
        // 1024 bytes Extended Event Text Bslbf
        memcpy( psz_ext_text, p_peek+372, 1024 );
        psz_ext_text[1024] = '\0';
        msg_Dbg( p_demux, "extended event text=%s", psz_ext_text );
        // 52 bytes reserved Bslbf
#endif
    }
523
524
525
526
    else
    {
        msg_Warn( p_demux, "TS module discarded (lost sync)" );
        return VLC_EGENERIC;
527
    }
528

529
530
    p_demux->p_sys = p_sys = malloc( sizeof( demux_sys_t ) );
    memset( p_sys, 0, sizeof( demux_sys_t ) );
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
    p_sys->i_packet_size = i_packet_size;

    /* Fill dump mode fields */
    p_sys->i_write = 0;
    p_sys->p_file = NULL;
    p_sys->b_file_out = VLC_FALSE;
    p_sys->psz_file = var_CreateGetString( p_demux, "ts-dump-file" );
    if( *p_sys->psz_file != '\0' )
    {
        p_sys->b_file_out = VLC_TRUE;

        var_Create( p_demux, "ts-dump-append", VLC_VAR_BOOL|VLC_VAR_DOINHERIT );
        var_Get( p_demux, "ts-dump-append", &val );
        b_append = val.b_bool;
        if ( b_append )
            psz_mode = "ab";
        else
            psz_mode = "wb";

        if( !strcmp( p_sys->psz_file, "-" ) )
        {
            msg_Info( p_demux, "dumping raw stream to standard output" );
            p_sys->p_file = stdout;
        }
555
        else if( ( p_sys->p_file = utf8_fopen( p_sys->psz_file, psz_mode ) ) == NULL )
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
        {
            msg_Err( p_demux, "cannot create `%s' for writing", p_sys->psz_file );
            p_sys->b_file_out = VLC_FALSE;
        }

        if( p_sys->b_file_out )
        {
            vlc_value_t bufsize;

            /* Determine how many packets to read. */
            var_Create( p_demux, "ts-dump-size",
                        VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
            var_Get( p_demux, "ts-dump-size", &bufsize );
            p_sys->i_ts_read = (int) (bufsize.i_int / p_sys->i_packet_size);
            if( p_sys->i_ts_read <= 0 )
            {
                p_sys->i_ts_read = 1500 / p_sys->i_packet_size;
            }
            p_sys->buffer = malloc( p_sys->i_packet_size * p_sys->i_ts_read );
            msg_Info( p_demux, "%s raw stream to file `%s' reading packets %d",
                      b_append ? "appending" : "dumping", p_sys->psz_file,
                      p_sys->i_ts_read );
        }
    }

581
    /* Fill p_demux field */
582
583
584
585
    if( p_sys->b_file_out )
        p_demux->pf_demux = DemuxFile;
    else
        p_demux->pf_demux = Demux;
586
587
588
    p_demux->pf_control = Control;

    /* Init p_sys field */
589
    p_sys->b_meta = VLC_TRUE;
590
591
    p_sys->b_dvb_control = VLC_TRUE;
    p_sys->i_dvb_program = 0;
592
593
594
595
596
597
598
599
    for( i = 0; i < 8192; i++ )
    {
        ts_pid_t *pid = &p_sys->pid[i];

        pid->i_pid      = i;
        pid->b_seen     = VLC_FALSE;
        pid->b_valid    = VLC_FALSE;
    }
600
601
    /* PID 8191 is padding */
    p_sys->pid[8191].b_seen = VLC_TRUE;
602
    p_sys->i_packet_size = i_packet_size;
603
604
    p_sys->b_udp_out = VLC_FALSE;
    p_sys->i_ts_read = 50;
605
    p_sys->csa = NULL;
606
607
608
609

    /* Init PAT handler */
    pat = &p_sys->pid[0];
    PIDInit( pat, VLC_TRUE, NULL );
610
611
    pat->psi->handle = dvbpsi_AttachPAT( (dvbpsi_pat_callback)PATCallBack,
                                         p_demux );
612
#ifdef TS_USE_DVB_SI
613
614
615
616
617
618
    if( p_sys->b_meta )
    {
        ts_pid_t *sdt = &p_sys->pid[0x11];
        ts_pid_t *eit = &p_sys->pid[0x12];

        PIDInit( sdt, VLC_TRUE, NULL );
619
620
621
        sdt->psi->handle =
            dvbpsi_AttachDemux( (dvbpsi_demux_new_cb_t)PSINewTableCallBack,
                                p_demux );
622
        PIDInit( eit, VLC_TRUE, NULL );
623
624
625
        eit->psi->handle =
            dvbpsi_AttachDemux( (dvbpsi_demux_new_cb_t)PSINewTableCallBack,
                                p_demux );
626
627
628
629
630
631
632
633
634
        if( p_sys->b_dvb_control )
        {
            stream_Control( p_demux->s, STREAM_CONTROL_ACCESS,
                            ACCESS_SET_PRIVATE_ID_STATE, 0x11, VLC_TRUE );
            stream_Control( p_demux->s, STREAM_CONTROL_ACCESS,
                            ACCESS_SET_PRIVATE_ID_STATE, 0x12, VLC_TRUE );
        }
    }
#endif
635
636
637
638
639

    /* Init PMT array */
    p_sys->i_pmt = 0;
    p_sys->pmt   = NULL;

640
    /* Read config */
641
642
    var_Create( p_demux, "ts-es-id-pid", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
    var_Get( p_demux, "ts-es-id-pid", &val );
643
    p_sys->b_es_id_pid = val.b_bool;
644

645
646
    var_Create( p_demux, "ts-out", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_demux, "ts-out", &val );
647
    if( val.psz_string && *val.psz_string && !p_sys->b_file_out )
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
    {
        vlc_value_t mtu;
        char *psz = strchr( val.psz_string, ':' );
        int   i_port = 0;

        p_sys->b_udp_out = VLC_TRUE;

        if( psz )
        {
            *psz++ = '\0';
            i_port = atoi( psz );
        }
        if( i_port <= 0 ) i_port  = 1234;
        msg_Dbg( p_demux, "resend ts to '%s:%d'", val.psz_string, i_port );

Rémi Denis-Courmont's avatar
Rémi Denis-Courmont committed
663
        p_sys->fd = net_ConnectUDP( p_demux, val.psz_string, i_port, 0 );
664
665
666
667
668
669
670
        if( p_sys->fd < 0 )
        {
            msg_Err( p_demux, "failed to open udp socket, send disabled" );
            p_sys->b_udp_out = VLC_FALSE;
        }
        else
        {
671
672
            var_Create( p_demux, "ts-out-mtu",
                        VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
673
            var_Get( p_demux, "ts-out-mtu", &mtu );
674
            p_sys->i_ts_read = mtu.i_int / p_sys->i_packet_size;
675
676
            if( p_sys->i_ts_read <= 0 )
            {
677
                p_sys->i_ts_read = 1500 / p_sys->i_packet_size;
678
            }
679
            p_sys->buffer = malloc( p_sys->i_packet_size * p_sys->i_ts_read );
680
681
682
683
684
685
686
        }
    }
    if( val.psz_string )
    {
        free( val.psz_string );
    }

687
688
689
690
691
692
693
694
695
696
697
698
    /* We handle description of an extra PMT */
    var_Create( p_demux, "ts-extra-pmt", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_demux, "ts-extra-pmt", &val );
    if( val.psz_string && strchr( val.psz_string, '=' ) != NULL )
    {
        char *psz = val.psz_string;
        int  i_pid = strtol( psz, &psz, 0 );

        if( i_pid >= 2 && i_pid < 8192 )
        {
            ts_pid_t *pmt = &p_sys->pid[i_pid];

699
            msg_Dbg( p_demux, "extra pmt specified (pid=%d)", i_pid );
700
            PIDInit( pmt, VLC_TRUE, NULL );
Laurent Aimar's avatar
Laurent Aimar committed
701
702
            pmt->psi->i_prg = 1;
            pmt->psi->prg = malloc( sizeof(ts_prg_psi_t) );
703
            /* FIXME we should also ask for a number */
704
            pmt->psi->prg[0]->handle =
705
706
                dvbpsi_AttachPMT( 1, (dvbpsi_pmt_callback)PMTCallBack,
                                  p_demux );
707
            pmt->psi->prg[0]->i_number = 0; /* special one */
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723

            psz = strchr( psz, '=' ) + 1;   /* can't failed */
            while( psz && *psz )
            {
                char *psz_next = strchr( psz, ',' );
                int i_pid, i_stream_type;

                if( psz_next )
                {
                    *psz_next++ = '\0';
                }

                i_pid = strtol( psz, &psz, 0 );
                if( *psz == ':' )
                {
                    i_stream_type = strtol( psz + 1, &psz, 0 );
724
725
                    if( i_pid >= 2 && i_pid < 8192 &&
                        !p_sys->pid[i_pid].b_valid )
726
727
728
729
                    {
                        ts_pid_t *pid = &p_sys->pid[i_pid];

                        PIDInit( pid, VLC_FALSE, pmt->psi);
730
                        if( pmt->psi->prg[0]->i_pid_pcr <= 0 )
731
                        {
732
                            pmt->psi->prg[0]->i_pid_pcr = i_pid;
733
                        }
734
                        PIDFillFormat( pid, i_stream_type);
735
736
                        if( pid->es->fmt.i_cat != UNKNOWN_ES )
                        {
737
738
739
740
                            if( p_sys->b_es_id_pid )
                            {
                                pid->es->fmt.i_id = i_pid;
                            }
741
                            msg_Dbg( p_demux, "  * es pid=%d type=%d "
742
743
744
745
                                     "fcc=%4.4s", i_pid, i_stream_type,
                                     (char*)&pid->es->fmt.i_codec );
                            pid->es->id = es_out_Add( p_demux->out,
                                                      &pid->es->fmt );
746
747
748
749
750
751
752
753
754
755
756
757
                        }
                    }
                }
                psz = psz_next;
            }
        }
    }
    if( val.psz_string )
    {
        free( val.psz_string );
    }

758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
    var_Create( p_demux, "ts-csa-ck", VLC_VAR_STRING | VLC_VAR_DOINHERIT );
    var_Get( p_demux, "ts-csa-ck", &val );
    if( val.psz_string && *val.psz_string )
    {
        char *psz = val.psz_string;
        if( psz[0] == '0' && ( psz[1] == 'x' || psz[1] == 'X' ) )
        {
            psz += 2;
        }
        if( strlen( psz ) != 16 )
        {
            msg_Warn( p_demux, "invalid csa ck (it must be 16 chars long)" );
        }
        else
        {
gbazin's avatar
gbazin committed
773
#ifndef UNDER_CE
774
            uint64_t i_ck = strtoull( psz, NULL, 16 );
gbazin's avatar
gbazin committed
775
776
777
#else
            uint64_t i_ck = strtoll( psz, NULL, 16 );
#endif
778
779
780
781
782
783
            uint8_t ck[8];
            int     i;
            for( i = 0; i < 8; i++ )
            {
                ck[i] = ( i_ck >> ( 56 - 8*i) )&0xff;
            }
784
#ifndef TS_NO_CSA_CK_MSG
785
786
            msg_Dbg( p_demux, "using CSA scrambling with "
                     "ck=%x:%x:%x:%x:%x:%x:%x:%x",
787
                     ck[0], ck[1], ck[2], ck[3], ck[4], ck[5], ck[6], ck[7] );
788
#endif
789
            p_sys->csa = csa_New();
790

791
792
            if( p_sys->csa )
            {
793
                vlc_value_t pkt_val;
794

795
                csa_SetCW( p_sys->csa, ck, ck );
796

797
                var_Create( p_demux, "ts-csa-pkt", VLC_VAR_INTEGER | VLC_VAR_DOINHERIT );
798
799
                var_Get( p_demux, "ts-csa-pkt", &pkt_val );
                if( pkt_val.i_int < 4 || pkt_val.i_int > 188 )
800
                {
801
                    msg_Err( p_demux, "wrong packet size %d specified.", pkt_val.i_int );
802
803
804
                    msg_Warn( p_demux, "using default packet size of 188 bytes" );
                    p_sys->i_csa_pkt_size = 188;
                }
805
                else p_sys->i_csa_pkt_size = pkt_val.i_int;
806
807
                msg_Dbg( p_demux, "decrypting %d bytes of packet", p_sys->i_csa_pkt_size );
            }
808
809
810
811
812
813
814
        }
    }
    if( val.psz_string )
    {
        free( val.psz_string );
    }

815
816
817
    var_Create( p_demux, "ts-silent", VLC_VAR_BOOL | VLC_VAR_DOINHERIT );
    var_Get( p_demux, "ts-silent", &val );
    p_sys->b_silent = val.b_bool;
818

819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
    return VLC_SUCCESS;
}

/*****************************************************************************
 * Close
 *****************************************************************************/
static void Close( vlc_object_t *p_this )
{
    demux_t     *p_demux = (demux_t*)p_this;
    demux_sys_t *p_sys = p_demux->p_sys;

    int          i;

    msg_Dbg( p_demux, "pid list:" );
    for( i = 0; i < 8192; i++ )
    {
        ts_pid_t *pid = &p_sys->pid[i];

        if( pid->b_valid && pid->psi )
        {
            switch( pid->i_pid )
            {
                case 0: /* PAT */
                    dvbpsi_DetachPAT( pid->psi->handle );
843
                    free( pid->psi );
844
845
                    break;
                case 1: /* CAT */
846
                    free( pid->psi );
847
                    break;
848
849
850
851
852
                case 0x11: /* SDT */
                case 0x12: /* EIT */
                    dvbpsi_DetachDemux( pid->psi->handle );
                    free( pid->psi );
                    break;
853
                default:
854
                    PIDClean( p_demux->out, pid );
855
856
857
858
859
                    break;
            }
        }
        else if( pid->b_valid && pid->es )
        {
860
            PIDClean( p_demux->out, pid );
861
        }
862

863
864
        if( pid->b_seen )
        {
865
            msg_Dbg( p_demux, "  - pid[%d] seen", pid->i_pid );
866
        }
867
868
869
870
871
872
873

        if( p_sys->b_dvb_control && pid->i_pid > 0 )
        {
            /* too much */
            stream_Control( p_demux->s, STREAM_CONTROL_ACCESS, ACCESS_SET_PRIVATE_ID_STATE, pid->i_pid, VLC_FALSE );
        }

874
875
    }

876
877
878
879
880
    if( p_sys->b_udp_out )
    {
        net_Close( p_sys->fd );
        free( p_sys->buffer );
    }
881
882
883
884
    if( p_sys->csa )
    {
        csa_Delete( p_sys->csa );
    }
885

gbazin's avatar
   
gbazin committed
886
    if( p_sys->i_pmt ) free( p_sys->pmt );
887
888
889
890
891
892
893
894

    if ( p_sys->p_programs_list )
    {
        vlc_value_t val;
        val.p_list = p_sys->p_programs_list;
        var_Change( p_demux, "programs", VLC_VAR_FREELIST, &val, NULL );
    }

895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
    /* If in dump mode, then close the file */
    if( p_sys->b_file_out )
    {
        msg_Info( p_demux ,"closing %s ("I64Fd" Kbytes dumped)", p_sys->psz_file,
                  p_sys->i_write / 1024 );

        if( p_sys->p_file != stdout )
        {
            fclose( p_sys->p_file );
            p_sys->p_file = NULL;
        }

        free( p_sys->buffer );
    }

910
911
912
    free( p_sys->psz_file );
    p_sys->psz_file = NULL;

913
914
915
    free( p_sys );
}

916
917
918
919
/*****************************************************************************
 * DemuxFile:
 *****************************************************************************/
static int DemuxFile( demux_t *p_demux )
920
{
921
922
923
924
925
926
    demux_sys_t *p_sys = p_demux->p_sys;
    uint8_t     *p_buffer = p_sys->buffer; /* Put first on sync byte */
    int i_diff= 0;
    int i_data= 0;
    int i_pos = 0;
    int i_bufsize = p_sys->i_packet_size * p_sys->i_ts_read;
927

928
929
930
    i_data = stream_Read( p_demux->s, p_sys->buffer, i_bufsize );
    if( (i_data <= 0) && (i_data < p_sys->i_packet_size) )
    {
931
        msg_Dbg( p_demux, "error reading malformed packets" );
932
933
934
935
936
937
938
939
940
941
        return i_data;
    }

    /* Test continuity counter */
    while( i_pos < i_data )
    {
        ts_pid_t    *p_pid;   /* point to a PID structure */
        vlc_bool_t b_payload; /* indicates a packet with payload */
        vlc_bool_t b_adaptation; /* adaptation field */
        int i_cc  = 0;        /* continuity counter */
942

943
944
945
946
947
948
949
950
951
952
953
954
        if( p_sys->buffer[i_pos] != 0x47 )
        {
            msg_Warn( p_demux, "lost sync" );
            while( !p_demux->b_die && (i_pos < i_data) )
            {
                i_pos++;
                if( p_sys->buffer[i_pos] == 0x47 )
                    break;
            }
            if( !p_demux->b_die )
                msg_Warn( p_demux, "sync found" );
        }
955

956
957
958
959
960
961
962
963
964
        /* continuous when (one of this):
         * diff == 1
         * diff == 0 and payload == 0
         * diff == 0 and duplicate packet (playload != 0) <- should we
         *   test the content ?
         */
        i_cc  = p_buffer[i_pos+3]&0x0f;
        b_payload = p_buffer[i_pos+3]&0x10;
        b_adaptation = p_buffer[i_pos+3]&0x20;
965

966
967
        /* Get the PID */
        p_pid = &p_sys->pid[ ((p_buffer[i_pos+1]&0x1f)<<8)|p_buffer[i_pos+2] ];
968

969
        /* Detect discontinuity indicator in adaptation field */
970
        if( b_adaptation && p_buffer[i_pos + 4] > 0 )
971
972
973
974
975
976
        {
            if( p_buffer[i_pos+5]&0x80 )
                msg_Warn( p_demux, "discontinuity indicator (pid=%d) ", p_pid->i_pid );
            if( p_buffer[i_pos+5]&0x40 )
                msg_Warn( p_demux, "random access indicator (pid=%d) ", p_pid->i_pid );
        }
977

978
979
980
        i_diff = ( i_cc - p_pid->i_cc )&0x0f;
        if( b_payload && i_diff == 1 )
        {
981
            p_pid->i_cc = ( p_pid->i_cc + 1 ) & 0xf;
982
983
984
985
986
987
988
989
990
991
992
993
994
995
        }
        else
        {
            if( p_pid->i_cc == 0xff )
            {
                msg_Warn( p_demux, "first packet for pid=%d cc=0x%x",
                        p_pid->i_pid, i_cc );
                p_pid->i_cc = i_cc;
            }
            else if( i_diff != 0 )
            {
                /* FIXME what to do when discontinuity_indicator is set ? */
                msg_Warn( p_demux, "transport error detected 0x%x instead of 0x%x",
                          i_cc, ( p_pid->i_cc + 1 )&0x0f );
996

997
998
999
1000
1001
                p_pid->i_cc = i_cc;
                /* Mark transport error in the TS packet. */
                p_buffer[i_pos+1] |= 0x80;
            }
        }
1002

1003
1004
1005
        /* Test if user wants to decrypt it first */
        if( p_sys->csa )
            csa_Decrypt( p_demux->p_sys->csa, &p_buffer[i_pos], p_demux->p_sys->i_csa_pkt_size );
1006

1007
1008
1009
1010
1011
1012
1013
1014
1015
1016
1017
1018
1019
1020
1021
1022
1023
1024
        i_pos += p_sys->i_packet_size;
    }

    /* Then write */
    i_data = fwrite( p_sys->buffer, 1, i_data, p_sys->p_file );
    if( i_data < 0 )
    {
        msg_Err( p_demux, "failed to write data" );
        return -1;
    }
#if 0
    msg_Dbg( p_demux, "dumped %d bytes", i_data );
#endif

    p_sys->i_write += i_data;
    return 1;
}

1025
1026
1027
1028
1029
1030
1031
1032
1033
/*****************************************************************************
 * Demux:
 *****************************************************************************/
static int Demux( demux_t *p_demux )
{
    demux_sys_t *p_sys = p_demux->p_sys;
    int          i_pkt;

    /* We read at most 100 TS packet or until a frame is completed */
1034
    for( i_pkt = 0; i_pkt < p_sys->i_ts_read; i_pkt++ )
1035
1036
1037
1038
1039
1040
    {
        vlc_bool_t  b_frame = VLC_FALSE;
        block_t     *p_pkt;
        ts_pid_t    *p_pid;

        /* Get a new TS packet */
1041
        if( !( p_pkt = stream_Block( p_demux->s, p_sys->i_packet_size ) ) )
1042
1043
1044
1045
        {
            msg_Dbg( p_demux, "eof ?" );
            return 0;
        }
1046
1047

        /* Check sync byte and re-sync if needed */
1048
1049
1050
1051
1052
        if( p_pkt->p_buffer[0] != 0x47 )
        {
            msg_Warn( p_demux, "lost synchro" );
            block_Release( p_pkt );

1053
            while( !p_demux->b_die )
1054
            {
1055
1056
                uint8_t *p_peek;
                int i_peek, i_skip = 0;
1057

1058
1059
1060
                i_peek = stream_Peek( p_demux->s, &p_peek,
                                      p_sys->i_packet_size * 10 );
                if( i_peek < p_sys->i_packet_size + 1 )
1061
1062
1063
1064
1065
                {
                    msg_Dbg( p_demux, "eof ?" );
                    return 0;
                }

1066
                while( i_skip < i_peek - p_sys->i_packet_size )
1067
                {
1068
1069
                    if( p_peek[i_skip] == 0x47 &&
                        p_peek[i_skip + p_sys->i_packet_size] == 0x47 )
1070
1071
1072
1073
1074
                    {
                        break;
                    }
                    i_skip++;
                }
1075
1076

                msg_Dbg( p_demux, "skipping %d bytes of garbage", i_skip );
1077
                stream_Read( p_demux->s, NULL, i_skip );
1078
1079

                if( i_skip < i_peek - p_sys->i_packet_size )
1080
1081
1082
1083
                {
                    break;
                }
            }
1084
1085

            if( !( p_pkt = stream_Block( p_demux->s, p_sys->i_packet_size ) ) )
1086
1087
1088
1089
1090
1091
            {
                msg_Dbg( p_demux, "eof ?" );
                return 0;
            }
        }

1092
1093
        if( p_sys->b_udp_out )
        {
1094
1095
            memcpy( &p_sys->buffer[i_pkt * p_sys->i_packet_size],
                    p_pkt->p_buffer, p_sys->i_packet_size );
1096
1097
        }

1098
1099
1100
1101
1102
1103
1104
        /* Parse the TS packet */
        p_pid = &p_sys->pid[PIDGet( p_pkt )];

        if( p_pid->b_valid )
        {
            if( p_pid->psi )
            {
1105
                if( p_pid->i_pid == 0 || p_pid->i_pid == 0x11 || p_pid->i_pid == 0x12 )
1106
1107
1108
1109
1110
1111
1112
1113
                {
                    dvbpsi_PushPacket( p_pid->psi->handle, p_pkt->p_buffer );
                }
                else
                {
                    int i_prg;
                    for( i_prg = 0; i_prg < p_pid->psi->i_prg; i_prg++ )
                    {
1114
1115
                        dvbpsi_PushPacket( p_pid->psi->prg[i_prg]->handle,
                                           p_pkt->p_buffer );
1116
1117
                    }
                }
1118
1119
                block_Release( p_pkt );
            }
1120
            else if( !p_sys->b_udp_out )
1121
1122
1123
            {
                b_frame = GatherPES( p_demux, p_pid, p_pkt );
            }
1124
1125
            else
            {
Laurent Aimar's avatar
Laurent Aimar committed
1126
                PCRHandle( p_demux, p_pid, p_pkt );
1127
1128
                block_Release( p_pkt );
            }
1129
1130
1131
1132
1133
        }
        else
        {
            if( !p_pid->b_seen )
            {
1134
                msg_Dbg( p_demux, "pid[%d] unknown", p_pid->i_pid );
1135
            }
1136
1137
            /* We have to handle PCR if present */
            PCRHandle( p_demux, p_pid, p_pkt );
1138
1139
1140
1141
1142
1143
1144
1145
1146
1147
            block_Release( p_pkt );
        }
        p_pid->b_seen = VLC_TRUE;

        if( b_frame )
        {
            break;
        }
    }

1148
1149
1150
    if( p_sys->b_udp_out )
    {
        /* Send the complete block */
1151
        net_Write( p_demux, p_sys->fd, NULL, p_sys->buffer,
1152
                   p_sys->i_ts_read * p_sys->i_packet_size );
1153
1154
    }

1155
1156
1157
1158
1159
1160
1161
1162
    return 1;
}

/*****************************************************************************
 * Control:
 *****************************************************************************/
static int Control( demux_t *p_demux, int i_query, va_list args )
{
1163
    demux_sys_t *p_sys = p_demux->p_sys;
1164
    double f, *pf;
1165
    int64_t i64;
1166
    int64_t *pi64;
1167
    int i_int;
1168

1169
1170
1171
    if( p_sys->b_file_out )
        return demux2_vaControlHelper( p_demux->s, 0, -1, 0, 1, i_query, args );

1172
1173
1174
1175
1176
1177
1178
1179
1180
1181
1182
1183
1184
1185
1186
1187
1188
1189
1190
1191
1192
1193
1194
1195
1196
1197
1198
1199
1200
1201
1202
1203
1204
1205
1206
1207
1208
1209
1210
1211
    switch( i_query )
    {
        case DEMUX_GET_POSITION:
            pf = (double*) va_arg( args, double* );
            i64 = stream_Size( p_demux->s );
            if( i64 > 0 )
            {
                *pf = (double)stream_Tell( p_demux->s ) / (double)i64;
            }
            else
            {
                *pf = 0.0;
            }
            return VLC_SUCCESS;
        case DEMUX_SET_POSITION:
            f = (double) va_arg( args, double );
            i64 = stream_Size( p_demux->s );

            es_out_Control( p_demux->out, ES_OUT_RESET_PCR );
            if( stream_Seek( p_demux->s, (int64_t)(i64 * f) ) )
            {
                return VLC_EGENERIC;
            }
            return VLC_SUCCESS;
#if 0

        case DEMUX_GET_TIME:
            pi64 = (int64_t*)va_arg( args, int64_t * );
            if( p_sys->i_time < 0 )
            {
                *pi64 = 0;
                return VLC_EGENERIC;
            }
            *pi64 = p_sys->i_time;
            return VLC_SUCCESS;

        case DEMUX_GET_LENGTH:
            pi64 = (int64_t*)va_arg( args, int64_t * );
            if( p_sys->i_mux_rate > 0 )
            {
1212
1213
                *pi64 = I64C(1000000) * ( stream_Size( p_demux->s ) / 50 ) /
                        p_sys->i_mux_rate;
1214
1215
1216
1217
                return VLC_SUCCESS;
            }
            *pi64 = 0;
            return VLC_EGENERIC;
1218
1219
1220
1221
1222
1223
#else
        case DEMUX_GET_TIME:
        case DEMUX_GET_LENGTH:
            pi64 = (int64_t*)va_arg( args, int64_t * );
            *pi64 = 0;
            return VLC_SUCCESS;
1224
#endif
1225
        case DEMUX_SET_GROUP:
1226
1227
1228
        {
            uint16_t i_vpid = 0, i_apid1 = 0, i_apid2 = 0, i_apid3 = 0;
            ts_prg_psi_t *p_prg = NULL;
1229
            vlc_list_t *p_list;
1230

1231
            i_int = (int)va_arg( args, int );
1232
1233
            p_list = (vlc_list_t *)va_arg( args, vlc_list_t * );
            msg_Dbg( p_demux, "DEMUX_SET_GROUP %d %p", i_int, p_list );
1234

1235
1236
1237
1238
1239
1240
1241
1242
1243
1244
1245
1246
1247
            if( p_sys->b_dvb_control && i_int > 0 && i_int != p_sys->i_dvb_program )
            {
                int i_pmt_pid = -1;
                int i;

                /* Search pmt to be unselected */
                for( i = 0; i < p_sys->i_pmt; i++ )
                {
                    ts_pid_t *pmt = p_sys->pmt[i];
                    int i_prg;

                    for( i_prg = 0; i_prg < pmt->psi->i_prg; i_prg++ )
                    {
1248
                        if( pmt->psi->prg[i_prg]->i_number == p_sys->i_dvb_program )
1249
1250
1251
1252
1253
1254
1255
1256
1257
1258
                        {
                            i_pmt_pid = p_sys->pmt[i]->i_pid;
                            break;
                        }
                    }
                    if( i_pmt_pid > 0 ) break;
                }

                if( i_pmt_pid > 0 )
                {
1259
1260
1261
                    stream_Control( p_demux->s, STREAM_CONTROL_ACCESS,
                                    ACCESS_SET_PRIVATE_ID_STATE, i_pmt_pid,
                                    VLC_FALSE );
1262
1263
1264
1265
                    /* All ES */
                    for( i = 2; i < 8192; i++ )
                    {
                        ts_pid_t *pid = &p_sys->pid[i];
Eric Petit's avatar
Eric Petit committed
1266
                        int i_prg;
1267
1268
1269
1270
1271

                        if( !pid->b_valid || pid->psi ) continue;

                        for( i_prg = 0; i_prg < pid->p_owner->i_prg; i_prg++ )
                        {
1272
                            if( pid->p_owner->prg[i_prg]->i_pid_pmt == i_pmt_pid && pid->es->id )
1273
1274
                            {
                                /* We only remove es that aren't defined by extra pmt */