]> git.sesse.net Git - vlc/blobdiff - modules/demux/ts.c
MKV: the Block duration is for the whole Block, not each frame
[vlc] / modules / demux / ts.c
index 8d46f8f4788f720aa30dcbabd754d353040b41b4..4603f2d9b97436a4382439491f0653901f382c0b 100644 (file)
@@ -240,10 +240,21 @@ typedef struct
     int             i_number;
     int             i_pid_pcr;
     int             i_pid_pmt;
-    mtime_t         i_pcr_value;
     /* IOD stuff (mpeg4) */
     iod_descriptor_t *iod;
 
+    struct
+    {
+        mtime_t i_current;
+        mtime_t i_first; // seen <> != -1
+        /* broken PCR handling */
+        mtime_t i_first_dts;
+        mtime_t i_pcroffset;
+        bool    b_disable; /* ignore PCR field, use dts */
+    } pcr;
+
+    mtime_t i_last_dts;
+
 } ts_prg_psi_t;
 
 typedef struct
@@ -283,6 +294,7 @@ typedef struct
 
     es_mpeg4_descriptor_t *p_mpeg4desc;
 
+    block_t *   p_prepcr_outqueue;
 } ts_es_t;
 
 typedef struct
@@ -310,8 +322,9 @@ typedef struct
     {
         vlc_fourcc_t i_fourcc;
         int i_type;
-        bool b_haspcr;
+        int i_pcr_count;
     } probed;
+
 } ts_pid_t;
 
 typedef struct
@@ -329,6 +342,7 @@ struct demux_sys_t
 {
     stream_t   *stream;
     bool        b_canseek;
+    bool        b_canfastseek;
     vlc_mutex_t     csa_lock;
 
     /* TS packet size (188, 192, 204) */
@@ -340,15 +354,7 @@ struct demux_sys_t
     /* how many TS packet we read at once */
     int         i_ts_read;
 
-    /* to determine length and time */
-    int         i_pid_ref_pcr;
-    mtime_t     i_first_pcr;
-    mtime_t     i_current_pcr;
-    mtime_t     i_last_pcr;
     bool        b_force_seek_per_percent;
-    int         i_pcrs_num;
-    mtime_t     *p_pcrs;
-    int64_t     *p_pos;
 
     struct
     {
@@ -367,7 +373,14 @@ struct demux_sys_t
     int         i_pmt;
     ts_pid_t    **pmt;
     int         i_pmt_es;
-    bool        b_delay_es_creation;
+
+    enum
+    {
+        NO_ES, /* for preparse */
+        DELAY_ES,
+        CREATE_ES
+    } es_creation;
+    #define PREPARSING p_sys->es_creation == NO_ES
 
     /* */
     bool        b_es_id_pid;
@@ -376,10 +389,10 @@ struct demux_sys_t
     bool        b_split_es;
 
     bool        b_trust_pcr;
-    bool        b_disable_pcr;
 
     /* */
     bool        b_access_control;
+    bool        b_end_preparse;
 
     /* */
     bool        b_dvb_meta;
@@ -388,16 +401,9 @@ struct demux_sys_t
     int64_t     i_dvb_length;
     bool        b_broken_charset; /* True if broken encoding is used in EPG/SDT */
 
-    /* */
-    int         i_current_program;
-    vlc_list_t  programs_list;
-
-    struct
-    {
-        /* broken PCR handling */
-        bool        b_program_pcr_seen;
-        mtime_t     i_first_dts;
-    } pcrfix;
+    /* Selected programs */
+    DECL_ARRAY( int ) programs; /* List of selected/access-filtered programs */
+    bool        b_default_selection; /* True if set by default to first pmt seen (to get data from filtered access) */
 
     struct
     {
@@ -431,6 +437,9 @@ static void PSINewTableCallBack( demux_t *, dvbpsi_handle,
 
 static int ChangeKeyCallback( vlc_object_t *, char const *, vlc_value_t, vlc_value_t, void * );
 
+
+/* Helpers */
+static ts_prg_psi_t * GetProgramByID( demux_sys_t *, int i_program );
 static inline int PIDGet( block_t *p )
 {
     return ( (p->p_buffer[1]&0x1f)<<8 )|p->p_buffer[2];
@@ -438,15 +447,16 @@ static inline int PIDGet( block_t *p )
 
 static bool GatherData( demux_t *p_demux, ts_pid_t *pid, block_t *p_bk );
 static void AddAndCreateES( demux_t *p_demux, ts_pid_t *pid );
+static void ProgramSetPCR( demux_t *p_demux, ts_prg_psi_t *p_prg, mtime_t i_pcr );
 
 static block_t* ReadTSPacket( demux_t *p_demux );
-static int Seek( demux_t *p_demux, double f_percent );
-static void GetFirstPCR( demux_t *p_demux );
-static void GetLastPCR( demux_t *p_demux );
-static void CheckPCR( demux_t *p_demux );
+static int ProbeStart( demux_t *p_demux, int i_program );
+static int ProbeEnd( demux_t *p_demux, int i_program );
+static int SeekToTime( demux_t *p_demux, ts_prg_psi_t *, int64_t time );
+static void ReadyQueuesPostSeek( demux_sys_t *p_sys );
 static void PCRHandle( demux_t *p_demux, ts_pid_t *, block_t * );
-static void PCRFixHandle( demux_t *, block_t * );
-static mtime_t AdjustPTSWrapAround( demux_t *, mtime_t );
+static void PCRFixHandle( demux_t *, ts_prg_psi_t *, block_t * );
+static int64_t TimeStampWrapAround( ts_prg_psi_t *, int64_t );
 
 static void              IODFree( iod_descriptor_t * );
 
@@ -684,7 +694,8 @@ static void ProbePES( demux_t *p_demux, ts_pid_t *pid, const uint8_t *p_pesstart
         {
             if( i_data < len )
                 return;
-            pid->probed.b_haspcr = ( len >= 7 && (p_pes[1] & 0x10) );
+            if( len >= 7 && (p_pes[1] & 0x10) )
+                pid->probed.i_pcr_count++;
             p_pes += len;
             i_data -= len;
         }
@@ -884,7 +895,7 @@ static void MissingPATPMTFixup( demux_t *p_demux )
             continue;
 
         if( i_pcr_pid == 0x1FFF && ( p_sys->pid[i].probed.i_type == 0x03 ||
-                                     p_sys->pid[i].probed.b_haspcr ) )
+                                     p_sys->pid[i].probed.i_pcr_count ) )
             i_pcr_pid = p_sys->pid[i].i_pid;
 
         i_num_pes++;
@@ -987,9 +998,9 @@ static int Open( vlc_object_t *p_this )
     /* Init p_sys field */
     p_sys->b_dvb_meta = true;
     p_sys->b_access_control = true;
-    p_sys->i_current_program = 0;
-    p_sys->programs_list.i_count = 0;
-    p_sys->programs_list.p_values = NULL;
+    p_sys->b_end_preparse = false;
+    ARRAY_INIT( p_sys->programs );
+    p_sys->b_default_selection = false;
     p_sys->i_tdt_delta = 0;
     p_sys->i_dvb_start = 0;
     p_sys->i_dvb_length = 0;
@@ -1108,7 +1119,6 @@ static int Open( vlc_object_t *p_this )
     /* Init PMT array */
     TAB_INIT( p_sys->i_pmt, p_sys->pmt );
     p_sys->i_pmt_es = 0;
-    p_sys->b_delay_es_creation = !p_sys->b_access_control;
 
     /* Read config */
     p_sys->b_es_id_pid = var_CreateGetBool( p_demux, "ts-es-id-pid" );
@@ -1172,52 +1182,24 @@ static int Open( vlc_object_t *p_this )
     p_sys->b_split_es = var_InheritBool( p_demux, "ts-split-es" );
 
     p_sys->b_canseek = false;
-    p_sys->i_pid_ref_pcr = -1;
-    p_sys->i_first_pcr = -1;
-    p_sys->i_current_pcr = -1;
-    p_sys->i_last_pcr = -1;
+    p_sys->b_canfastseek = false;
     p_sys->b_force_seek_per_percent = var_InheritBool( p_demux, "ts-seek-percent" );
-    p_sys->i_pcrs_num = 10;
-    p_sys->p_pcrs = (mtime_t *)calloc( p_sys->i_pcrs_num, sizeof( mtime_t ) );
-    p_sys->p_pos = (int64_t *)calloc( p_sys->i_pcrs_num, sizeof( int64_t ) );
 
     p_sys->arib.e_mode = var_InheritInteger( p_demux, "ts-arib" );
 
-    if( !p_sys->p_pcrs || !p_sys->p_pos )
-    {
-        Close( p_this );
-        return VLC_ENOMEM;
-    }
-
-    bool b_can_fastseek = false;
     stream_Control( p_sys->stream, STREAM_CAN_SEEK, &p_sys->b_canseek );
-    stream_Control( p_sys->stream, STREAM_CAN_FASTSEEK, &b_can_fastseek );
-    if ( p_sys->b_canseek )
-    {
-        if( b_can_fastseek )
-        {
-            GetFirstPCR( p_demux );
-            CheckPCR( p_demux );
-            GetLastPCR( p_demux );
-        }
-
-        if( p_sys->i_first_pcr < 0 || p_sys->i_last_pcr < 0 )
-        {
-            msg_Dbg( p_demux, "Force Seek Per Percent: PCR's not found,");
-            p_sys->b_force_seek_per_percent = true;
-        }
-    }
+    stream_Control( p_sys->stream, STREAM_CAN_FASTSEEK, &p_sys->b_canfastseek );
 
-    while( p_sys->i_pmt_es <= 0
-           && (!p_sys->pid[0].b_seen && !p_sys->patfix.b_pat_deadline) )
+    /* Preparse time */
+    if( p_sys->b_canseek )
     {
-        if( Demux( p_demux ) != 1 )
-            break;
+        p_sys->es_creation = NO_ES;
+        while( !p_sys->i_pmt_es && !p_sys->b_end_preparse )
+            if( Demux( p_demux ) != VLC_DEMUXER_SUCCESS )
+                break;
     }
 
-    /* If we had no PAT within 140ms, create PAT/PMT from probed streams */
-    if( p_sys->i_pmt_es == 0 && !p_sys->pid[0].b_seen )
-        MissingPATPMTFixup( p_demux );
+    p_sys->es_creation = ( p_sys->b_access_control ? CREATE_ES : DELAY_ES );
 
     return VLC_SUCCESS;
 }
@@ -1297,10 +1279,7 @@ static void Close( vlc_object_t *p_this )
 
     TAB_CLEAN( p_sys->i_pmt, p_sys->pmt );
 
-    free( p_sys->programs_list.p_values );
-
-    free( p_sys->p_pcrs );
-    free( p_sys->p_pos );
+    ARRAY_RESET( p_sys->programs );
 
 #ifdef HAVE_ARIBB24
     if ( p_sys->arib.p_instance )
@@ -1347,6 +1326,10 @@ static int Demux( demux_t *p_demux )
     demux_sys_t *p_sys = p_demux->p_sys;
     bool b_wait_es = p_sys->i_pmt_es <= 0;
 
+    /* If we had no PAT within 140ms, create PAT/PMT from probed streams */
+    if( p_sys->i_pmt_es == 0 && !p_sys->pid[0].b_seen && p_sys->patfix.b_pat_deadline )
+        MissingPATPMTFixup( p_demux );
+
     /* We read at most 100 TS packet or until a frame is completed */
     for( int i_pkt = 0; i_pkt < p_sys->i_ts_read; i_pkt++ )
     {
@@ -1354,7 +1337,7 @@ static int Demux( demux_t *p_demux )
         block_t     *p_pkt;
         if( !(p_pkt = ReadTSPacket( p_demux )) )
         {
-            return 0;
+            return VLC_DEMUXER_EOF;
         }
 
         if( p_sys->b_start_record )
@@ -1397,9 +1380,24 @@ static int Demux( demux_t *p_demux )
             }
             else
             {
-                if(p_sys->b_delay_es_creation) /* No longer delay ES since that pid's program sends data */
+                p_sys->b_end_preparse = true;
+                if( p_sys->es_creation == DELAY_ES ) /* No longer delay ES since that pid's program sends data */
+                {
                     AddAndCreateES( p_demux, NULL );
+                }
                 b_frame = GatherData( p_demux, p_pid, p_pkt );
+
+                if( p_sys->b_default_selection )
+                {
+                    p_sys->b_default_selection = false;
+                    assert(p_sys->programs.i_size == 1);
+                    if( p_sys->programs.p_elems[0] != p_pid->i_owner_number )
+                    {
+                        SetPrgFilter( p_demux, p_sys->programs.p_elems[0], false );
+                        SetPrgFilter( p_demux, p_pid->i_owner_number, true );
+                        p_sys->programs.p_elems[0] = p_pid->i_owner_number;
+                    }
+                }
             }
         }
         else
@@ -1419,7 +1417,7 @@ static int Demux( demux_t *p_demux )
     }
 
     demux_UpdateTitleFromStream( p_demux );
-    return 1;
+    return VLC_DEMUXER_SUCCESS;
 }
 
 /*****************************************************************************
@@ -1457,92 +1455,163 @@ static int Control( demux_t *p_demux, int i_query, va_list args )
     int64_t i64;
     int64_t *pi64;
     int i_int;
+    ts_prg_psi_t *p_prg;
+    int i_first_program = ( p_sys->programs.i_size ) ? p_sys->programs.p_elems[0] : 0;
+
+    if( PREPARSING || !i_first_program || p_sys->b_default_selection )
+    {
+        /* Set default program for preparse time (no program has been selected) */
+        for( int i=0; i<p_sys->i_pmt; i++ )
+        {
+            for( int j=0; j<p_sys->pmt[i]->psi->i_prg; j++ )
+            {
+                p_prg = p_sys->pmt[i]->psi->prg[j];
+                if( ( p_prg->pcr.i_first > -1 || p_prg->pcr.i_first_dts > VLC_TS_INVALID ) && p_prg->i_last_dts > 0 )
+                {
+                    i_first_program = p_prg->i_number;
+                    break;
+                }
+            }
+        }
+    }
 
     switch( i_query )
     {
     case DEMUX_GET_POSITION:
         pf = (double*) va_arg( args, double* );
 
-        if( p_sys->b_force_seek_per_percent ||
-            (p_sys->b_dvb_meta && p_sys->b_access_control) ||
-            p_sys->i_current_pcr - p_sys->i_first_pcr < 0 ||
-            p_sys->i_last_pcr - p_sys->i_first_pcr <= 0 )
+        /* Access control test is because EPG for recordings is not relevant */
+        if( p_sys->b_dvb_meta && p_sys->b_access_control )
         {
             int64_t i_time, i_length;
             if( !DVBEventInformation( p_demux, &i_time, &i_length ) && i_length > 0 )
-                *pf = (double)i_time/(double)i_length;
-            else if( (i64 = stream_Size( p_sys->stream) ) > 0 )
             {
-                int64_t offset = stream_Tell( p_sys->stream );
-
-                *pf = (double)offset / (double)i64;
+                *pf = (double)i_time/(double)i_length;
+                return VLC_SUCCESS;
             }
-            else
-                *pf = 0.0;
         }
-        else
+
+        if( (p_prg = GetProgramByID( p_sys, i_first_program )) &&
+             p_prg->pcr.i_first > -1 && p_prg->i_last_dts > VLC_TS_INVALID &&
+             p_prg->pcr.i_current > -1 )
         {
-            *pf = (double)(p_sys->i_current_pcr - p_sys->i_first_pcr) / (double)(p_sys->i_last_pcr - p_sys->i_first_pcr);
+            double i_length = TimeStampWrapAround( p_prg,
+                                                   p_prg->i_last_dts ) - p_prg->pcr.i_first;
+            i_length += p_prg->pcr.i_pcroffset;
+            double i_pos = TimeStampWrapAround( p_prg,
+                                                p_prg->pcr.i_current ) - p_prg->pcr.i_first;
+            *pf = i_pos / i_length;
+            return VLC_SUCCESS;
         }
-        return VLC_SUCCESS;
+
+        if( (i64 = stream_Size( p_sys->stream) ) > 0 )
+        {
+            int64_t offset = stream_Tell( p_sys->stream );
+            *pf = (double)offset / (double)i64;
+            return VLC_SUCCESS;
+        }
+        break;
 
     case DEMUX_SET_POSITION:
         f = (double) va_arg( args, double );
 
         if(!p_sys->b_canseek)
-            return VLC_EGENERIC;
+            break;
 
-        if( p_sys->b_force_seek_per_percent ||
-            (p_sys->b_dvb_meta && p_sys->b_access_control) ||
-            p_sys->i_last_pcr - p_sys->i_first_pcr <= 0 )
+        if( p_sys->b_dvb_meta && p_sys->b_access_control &&
+           !p_sys->b_force_seek_per_percent &&
+           (p_prg = GetProgramByID( p_sys, i_first_program )) )
         {
-            i64 = stream_Size( p_sys->stream );
-            if( stream_Seek( p_sys->stream, (int64_t)(i64 * f) ) )
-                return VLC_EGENERIC;
+            int64_t i_time, i_length;
+            if( !DVBEventInformation( p_demux, &i_time, &i_length ) &&
+                 i_length > 0 && !SeekToTime( p_demux, p_prg, TO_SCALE(i_length) * f ) )
+            {
+                ReadyQueuesPostSeek( p_sys );
+                es_out_Control( p_demux->out, ES_OUT_SET_NEXT_DISPLAY_TIME,
+                                TO_SCALE(i_length) * f );
+                return VLC_SUCCESS;
+            }
         }
-        else
+
+        if( !p_sys->b_force_seek_per_percent &&
+            (p_prg = GetProgramByID( p_sys, i_first_program )) &&
+             p_prg->pcr.i_first > -1 && p_prg->i_last_dts > VLC_TS_INVALID &&
+             p_prg->pcr.i_current > -1 )
         {
-            if( Seek( p_demux, f ) )
+            double i_length = TimeStampWrapAround( p_prg,
+                                                   p_prg->i_last_dts ) - p_prg->pcr.i_first;
+            if( !SeekToTime( p_demux, p_prg, p_prg->pcr.i_first + i_length * f ) )
             {
-                p_sys->b_force_seek_per_percent = true;
-                return VLC_EGENERIC;
+                ReadyQueuesPostSeek( p_sys );
+                es_out_Control( p_demux->out, ES_OUT_SET_NEXT_DISPLAY_TIME,
+                                FROM_SCALE(p_prg->pcr.i_first + i_length * f) );
+                return VLC_SUCCESS;
             }
         }
-        return VLC_SUCCESS;
+
+        i64 = stream_Size( p_sys->stream );
+        if( i64 > 0 &&
+            stream_Seek( p_sys->stream, (int64_t)(i64 * f) ) == VLC_SUCCESS )
+        {
+            ReadyQueuesPostSeek( p_sys );
+            return VLC_SUCCESS;
+        }
+        break;
+
+    case DEMUX_SET_TIME:
+        i64 = (int64_t)va_arg( args, int64_t );
+
+        if( p_sys->b_canseek &&
+           (p_prg = GetProgramByID( p_sys, i_first_program )) &&
+            p_prg->pcr.i_first > -1 &&
+           !SeekToTime( p_demux, p_prg, p_prg->pcr.i_first + TO_SCALE(i64) ) )
+        {
+            ReadyQueuesPostSeek( p_sys );
+            es_out_Control( p_demux->out, ES_OUT_SET_NEXT_DISPLAY_TIME,
+                            FROM_SCALE(p_prg->pcr.i_first) + i64 - VLC_TS_0 );
+            return VLC_SUCCESS;
+        }
+        break;
 
     case DEMUX_GET_TIME:
         pi64 = (int64_t*)va_arg( args, int64_t * );
-        if( (p_sys->b_dvb_meta && p_sys->b_access_control) ||
-            p_sys->b_force_seek_per_percent ||
-            p_sys->i_current_pcr - p_sys->i_first_pcr < 0 )
+
+        if( p_sys->b_dvb_meta && p_sys->b_access_control )
         {
-            if( DVBEventInformation( p_demux, pi64, NULL ) )
-            {
-                *pi64 = 0;
-            }
+            if( !DVBEventInformation( p_demux, pi64, NULL ) )
+                return VLC_SUCCESS;
         }
-        else
+
+        if( (p_prg = GetProgramByID( p_sys, i_first_program )) &&
+             p_prg->pcr.i_current > -1 && p_prg->pcr.i_first > -1 )
         {
-            *pi64 = (p_sys->i_current_pcr - p_sys->i_first_pcr) * 100 / 9;
+            int64_t i_pcr = TimeStampWrapAround( p_prg, p_prg->pcr.i_current );
+            *pi64 = FROM_SCALE(i_pcr - p_prg->pcr.i_first);
+            return VLC_SUCCESS;
         }
-        return VLC_SUCCESS;
+        break;
 
     case DEMUX_GET_LENGTH:
         pi64 = (int64_t*)va_arg( args, int64_t * );
-        if( (p_sys->b_dvb_meta && p_sys->b_access_control) ||
-            p_sys->b_force_seek_per_percent ||
-            p_sys->i_last_pcr - p_sys->i_first_pcr <= 0 )
+
+        if( p_sys->b_dvb_meta && p_sys->b_access_control )
         {
-            if( DVBEventInformation( p_demux, NULL, pi64 ) )
-            {
-                *pi64 = 0;
-            }
+            if( !DVBEventInformation( p_demux, NULL, pi64 ) )
+                return VLC_SUCCESS;
         }
-        else
+
+        if( (p_prg = GetProgramByID( p_sys, i_first_program )) &&
+           ( p_prg->pcr.i_first > -1 || p_prg->pcr.i_first_dts > VLC_TS_INVALID ) &&
+             p_prg->i_last_dts > 0 )
         {
-            *pi64 = (p_sys->i_last_pcr - p_sys->i_first_pcr) * 100 / 9;
+            int64_t i_start = (p_prg->pcr.i_first > -1) ? p_prg->pcr.i_first :
+                              TO_SCALE(p_prg->pcr.i_first_dts);
+            int64_t i_last = TimeStampWrapAround( p_prg, p_prg->i_last_dts );
+            i_last += p_prg->pcr.i_pcroffset;
+            *pi64 = FROM_SCALE(i_last - i_start);
+            return VLC_SUCCESS;
         }
-        return VLC_SUCCESS;
+        break;
 
     case DEMUX_SET_GROUP:
     {
@@ -1552,47 +1621,33 @@ static int Control( demux_t *p_demux, int i_query, va_list args )
         p_list = (vlc_list_t *)va_arg( args, vlc_list_t * );
         msg_Dbg( p_demux, "DEMUX_SET_GROUP %d %p", i_int, p_list );
 
-        if( i_int == 0 && p_sys->i_current_program > 0 )
-            i_int = p_sys->i_current_program;
-
-        if( p_sys->i_current_program > 0 )
-        {
-            if( p_sys->i_current_program != i_int )
-                SetPrgFilter( p_demux, p_sys->i_current_program, false );
-        }
-        else if( p_sys->i_current_program < 0 )
+        if( i_int != 0 ) /* If not default program */
         {
-            for( int i = 0; i < p_sys->programs_list.i_count; i++ )
-                SetPrgFilter( p_demux, p_sys->programs_list.p_values[i].i_int, false );
-        }
+            /* Deselect/filter current ones */
+            for( int i=0; i<p_sys->programs.i_size; i++ )
+                SetPrgFilter( p_demux, p_sys->programs.p_elems[i], false );
+            ARRAY_RESET( p_sys->programs );
 
-        if( i_int > 0 )
-        {
-            p_sys->i_current_program = i_int;
-            SetPrgFilter( p_demux, p_sys->i_current_program, true );
-        }
-        else if( i_int < 0 )
-        {
-            p_sys->i_current_program = -1;
-            p_sys->programs_list.i_count = 0;
-            if( p_list )
+            if( i_int != -1 )
+            {
+                ARRAY_APPEND( p_sys->programs, i_int );
+            }
+            else if( likely( p_list != NULL ) )
             {
-                vlc_list_t *p_dst = &p_sys->programs_list;
-                free( p_dst->p_values );
+                for( int i = 0; i < p_list->i_count; i++ )
+                   ARRAY_APPEND( p_sys->programs, p_list->p_values[i].i_int );
+            }
 
-                p_dst->p_values = calloc( p_list->i_count,
-                                          sizeof(*p_dst->p_values) );
-                if( p_dst->p_values )
-                {
-                    p_dst->i_count = p_list->i_count;
-                    for( int i = 0; i < p_list->i_count; i++ )
-                    {
-                        p_dst->p_values[i] = p_list->p_values[i];
-                        SetPrgFilter( p_demux, p_dst->p_values[i].i_int, true );
-                    }
-                }
+            /* Select/filter current ones */
+            for( int i=0; i<p_sys->programs.i_size; i++ )
+            {
+                msg_Dbg( p_demux, "Program %d in new selection", p_sys->programs.p_elems[i] );
+                SetPrgFilter( p_demux, p_sys->programs.p_elems[i], true );
             }
+
+            p_sys->b_default_selection = false;
         }
+
         return VLC_SUCCESS;
     }
 
@@ -1632,8 +1687,10 @@ static int Control( demux_t *p_demux, int i_query, va_list args )
         return stream_vaControl( p_sys->stream, STREAM_GET_SIGNAL, args );
 
     default:
-        return VLC_EGENERIC;
+        break;
     }
+
+    return VLC_EGENERIC;
 }
 
 /*****************************************************************************
@@ -1796,8 +1853,6 @@ static void SetPrgFilter( demux_t *p_demux, int i_prg_id, bool b_selected )
     ts_prg_psi_t *p_prg = NULL;
     int i_pmt_pid = -1;
 
-    p_sys->b_disable_pcr = !p_sys->b_trust_pcr;
-
     /* Search pmt to be unselected */
     for( int i = 0; i < p_sys->i_pmt; i++ )
     {
@@ -1819,6 +1874,8 @@ static void SetPrgFilter( demux_t *p_demux, int i_prg_id, bool b_selected )
         return;
     assert( p_prg );
 
+    p_prg->pcr.b_disable = !p_sys->b_trust_pcr;
+
     SetPIDFilter( p_demux, i_pmt_pid, b_selected );
     if( p_prg->i_pid_pcr > 0 )
         SetPIDFilter( p_demux, p_prg->i_pid_pcr, b_selected );
@@ -1879,10 +1936,16 @@ static void PIDInit( ts_pid_t *pid, bool b_psi, ts_psi_t *p_owner )
             prg->i_number   = -1;
             prg->i_pid_pcr  = -1;
             prg->i_pid_pmt  = -1;
-            prg->i_pcr_value= -1;
             prg->iod        = NULL;
             prg->handle     = NULL;
 
+            prg->i_last_dts = -1;
+
+            prg->pcr.i_current = -1;
+            prg->pcr.i_first  = -1;
+            prg->pcr.b_disable = false;
+            prg->pcr.i_first_dts = VLC_TS_INVALID;
+            prg->pcr.i_pcroffset = -1;
             TAB_APPEND( pid->psi->i_prg, pid->psi->prg, prg );
         }
     }
@@ -1947,6 +2010,9 @@ static void PIDClean( demux_t *p_demux, ts_pid_t *pid )
         if( pid->es->p_data )
             block_ChainRelease( pid->es->p_data );
 
+        if( pid->es->p_prepcr_outqueue )
+            block_ChainRelease( pid->es->p_prepcr_outqueue );
+
         es_format_Clean( &pid->es->fmt );
 
         free( pid->es );
@@ -2077,11 +2143,14 @@ static block_t *Opus_Parse(demux_t *demux, block_t *block)
 /****************************************************************************
  * gathering stuff
  ****************************************************************************/
-static int ParsePESHeader( demux_t *p_demux, const uint8_t *p_header,
+static int ParsePESHeader( demux_t *p_demux, const uint8_t *p_header, size_t i_header,
                            unsigned *pi_skip, mtime_t *pi_dts, mtime_t *pi_pts )
 {
     unsigned i_skip;
 
+    if ( i_header < 9 )
+        return VLC_EGENERIC;
+
     switch( p_header[3] )
     {
     case 0xBC:  /* Program stream map */
@@ -2099,21 +2168,33 @@ static int ParsePESHeader( demux_t *p_demux, const uint8_t *p_header,
         {
             /* mpeg2 PES */
             i_skip = p_header[8] + 9;
+            if( i_header < i_skip )
+                return VLC_EGENERIC;
 
             if( p_header[7]&0x80 )    /* has pts */
             {
+                if( i_header < 9 + 5 )
+                    return VLC_EGENERIC;
                 *pi_pts = ExtractPESTimestamp( &p_header[9] );
 
                 if( p_header[7]&0x40 )    /* has dts */
+                {
+                    if( i_header < 14 + 5 )
+                        return VLC_EGENERIC;
                     *pi_dts = ExtractPESTimestamp( &p_header[14] );
+                }
             }
         }
         else
         {
             i_skip = 6;
+            if( i_header < i_skip + 1 )
+                return VLC_EGENERIC;
             while( i_skip < 23 && p_header[i_skip] == 0xff )
             {
                 i_skip++;
+                if( i_header < i_skip + 1 )
+                    return VLC_EGENERIC;
             }
             if( i_skip == 23 )
             {
@@ -2125,12 +2206,19 @@ static int ParsePESHeader( demux_t *p_demux, const uint8_t *p_header,
                 i_skip += 2;
             }
 
+            if( i_header < i_skip + 1 )
+                return VLC_EGENERIC;
+
             if(  p_header[i_skip]&0x20 )
             {
+                if( i_header < i_skip + 5 )
+                    return VLC_EGENERIC;
                 *pi_pts = ExtractPESTimestamp( &p_header[i_skip] );
 
                 if( p_header[i_skip]&0x10 )    /* has dts */
                 {
+                    if( i_header < i_skip + 10 )
+                        return VLC_EGENERIC;
                     *pi_dts = ExtractPESTimestamp( &p_header[i_skip+5] );
                     i_skip += 10;
                 }
@@ -2147,6 +2235,9 @@ static int ParsePESHeader( demux_t *p_demux, const uint8_t *p_header,
         break;
     }
 
+    if( i_header < i_skip )
+        return VLC_EGENERIC;
+
     *pi_skip = i_skip;
     return VLC_SUCCESS;
 }
@@ -2161,10 +2252,8 @@ static void ParsePES( demux_t *p_demux, ts_pid_t *pid, block_t *p_pes )
     mtime_t i_pts = -1;
     mtime_t i_length = 0;
 
-    /* FIXME find real max size */
     const int i_max = block_ChainExtract( p_pes, header, 34 );
-    assert(i_max >= 34);
-    if (unlikely(i_max < 34))
+    if ( i_max < 4 )
     {
         block_ChainRelease( p_pes );
         return;
@@ -2179,8 +2268,7 @@ static void ParsePES( demux_t *p_demux, ts_pid_t *pid, block_t *p_pes )
         return;
     }
 
-    /* TODO check size */
-    if( ParsePESHeader( p_demux, (uint8_t*)&header, &i_skip, &i_dts, &i_pts ) == VLC_EGENERIC )
+    if( ParsePESHeader( p_demux, (uint8_t*)&header, i_max, &i_skip, &i_dts, &i_pts ) == VLC_EGENERIC )
     {
         block_ChainRelease( p_pes );
         return;
@@ -2188,9 +2276,9 @@ static void ParsePES( demux_t *p_demux, ts_pid_t *pid, block_t *p_pes )
     else
     {
         if( i_pts != -1 )
-            i_pts = AdjustPTSWrapAround( p_demux, i_pts );
+            i_pts = TimeStampWrapAround( GetProgramByID(p_sys, pid->i_owner_number), i_pts );
         if( i_dts != -1 )
-            i_dts = AdjustPTSWrapAround( p_demux, i_dts );
+            i_dts = TimeStampWrapAround( GetProgramByID(p_sys, pid->i_owner_number), i_dts );
     }
 
     if( pid->es->fmt.i_codec == VLC_FOURCC( 'a', '5', '2', 'b' ) ||
@@ -2283,7 +2371,7 @@ static void ParsePES( demux_t *p_demux, ts_pid_t *pid, block_t *p_pes )
                 {
                     if( pid->i_owner_number == pid->p_owner->prg[i]->i_number )
                     {
-                        mtime_t i_pcr = pid->p_owner->prg[i]->i_pcr_value;
+                        mtime_t i_pcr = pid->p_owner->prg[i]->pcr.i_current;
                         if( i_pcr > VLC_TS_INVALID )
                             p_block->i_pts = VLC_TS_0 + i_pcr * 100 / 9 + 40000;
                         break;
@@ -2312,27 +2400,69 @@ static void ParsePES( demux_t *p_demux, ts_pid_t *pid, block_t *p_pes )
             p_block = Opus_Parse(p_demux, p_block);
         }
 
+        ts_prg_psi_t *p_program = GetProgramByID( p_sys, pid->i_owner_number );
+
         while (p_block) {
             block_t *p_next = p_block->p_next;
             p_block->p_next = NULL;
-            for( int i = 0; i < pid->i_extra_es; i++ )
+
+            if( p_program->pcr.i_first == -1 ) /* Not seen yet */
+                PCRFixHandle( p_demux, p_program, p_block );
+
+            if( p_program->pcr.i_current > -1 || p_program->pcr.b_disable )
             {
-                es_out_Send( p_demux->out, pid->extra_es[i]->id,
-                        block_Duplicate( p_block ) );
-            }
+                if( pid->es->p_prepcr_outqueue )
+                {
+                    block_ChainAppend( &pid->es->p_prepcr_outqueue, p_block );
+                    p_block = pid->es->p_prepcr_outqueue;
+                    pid->es->p_prepcr_outqueue = NULL;
+                }
+
+                if ( p_program->pcr.b_disable && p_block->i_dts > VLC_TS_INVALID &&
+                     ( p_program->i_pid_pcr == pid->i_pid || p_program->i_pid_pcr == 0x1FFF ) )
+                {
+                    ProgramSetPCR( p_demux, p_program, (p_block->i_dts - VLC_TS_0) * 9 / 100 - 120000 );
+                }
+
+                /* Compute PCR/DTS offset if any */
+                if( p_program->pcr.i_pcroffset == -1 && p_block->i_dts > VLC_TS_INVALID &&
+                    p_program->pcr.i_current > VLC_TS_INVALID )
+                {
+                    int64_t i_dts27 = (p_block->i_dts - VLC_TS_0) * 9 / 100;
+                    int64_t i_pcr = TimeStampWrapAround( p_program, p_program->pcr.i_current );
+                    if( i_dts27 < i_pcr )
+                    {
+                        p_program->pcr.i_pcroffset = i_pcr - i_dts27 + 80000;
+                        msg_Warn( p_demux, "Broken stream: pid %d sends packets with dts %"PRId64
+                                           "us later than pcr, applying delay",
+                                  pid->i_pid, p_program->pcr.i_pcroffset * 100 / 9 );
+                    }
+                    else p_program->pcr.i_pcroffset = 0;
+                }
 
-            PCRFixHandle( p_demux, p_block );
+                if( p_program->pcr.i_pcroffset != -1 )
+                {
+                    if( p_block->i_dts > VLC_TS_INVALID )
+                        p_block->i_dts += (p_program->pcr.i_pcroffset * 100 / 9);
+                    if( p_block->i_pts > VLC_TS_INVALID )
+                        p_block->i_pts += (p_program->pcr.i_pcroffset * 100 / 9);
+                }
 
-            if ( p_sys->b_disable_pcr && p_block->i_dts > VLC_TS_INVALID )
-                es_out_Control( p_demux->out, ES_OUT_SET_GROUP_PCR,
-                        pid->i_owner_number, p_block->i_dts);
+                for( int i = 0; i < pid->i_extra_es; i++ )
+                {
+                    es_out_Send( p_demux->out, pid->extra_es[i]->id,
+                            block_Duplicate( p_block ) );
+                }
 
-            if( !p_sys->b_disable_pcr && p_block->i_dts > VLC_TS_INVALID &&
-                 p_block->i_dts < (VLC_TS_0 + p_sys->i_current_pcr * 100 / 9) )
-                msg_Warn( p_demux, "Broken stream: pid %d sends packets with dts %"PRId64"us later than pcr",
-                          pid->i_pid, (p_sys->i_current_pcr * 100 / 9) - p_block->i_dts + VLC_TS_0  );
+                es_out_Send( p_demux->out, pid->es->id, p_block );
+            }
+            else
+            {
+                if( p_program->pcr.i_first == -1 ) /* Not seen yet */
+                    PCRFixHandle( p_demux, p_program, p_block );
 
-            es_out_Send( p_demux->out, pid->es->id, p_block );
+                block_ChainAppend( &pid->es->p_prepcr_outqueue, p_block );
+            }
 
             p_block = p_next;
         }
@@ -2346,18 +2476,13 @@ static void ParsePES( demux_t *p_demux, ts_pid_t *pid, block_t *p_pes )
 static void ParseTableSection( demux_t *p_demux, ts_pid_t *pid, block_t *p_data )
 {
     block_t *p_content = block_ChainGather( p_data );
-    mtime_t i_date = -1;
-    for( int i = 0; pid->p_owner && i < pid->p_owner->i_prg; i++ )
-    {
-        if( pid->i_owner_number == pid->p_owner->prg[i]->i_number )
-        {
-            i_date = pid->p_owner->prg[i]->i_pcr_value;
-            if( i_date >= 0 )
-                break;
-        }
-    }
-    if( i_date >= 0 )
+    ts_prg_psi_t *p_program = NULL;
+
+    if ( pid->p_owner &&
+         (p_program = GetProgramByID( p_demux->p_sys, pid->i_owner_number )) )
     {
+        mtime_t i_date = p_program->pcr.i_current;
+
         if( pid->es->fmt.i_codec == VLC_CODEC_SCTE_27 )
         {
             /* We need to extract the truncated pts stored inside the payload */
@@ -2368,7 +2493,7 @@ static void ParseTableSection( demux_t *p_demux, ts_pid_t *pid, block_t *p_data
                 if( p_content->p_buffer[3] & 0x40 )
                 {
                     i_index = ((p_content->p_buffer[7] & 0x0f) << 8) |
-                              p_content->p_buffer[8];
+                             p_content->p_buffer[8];
                     i_offset = 9;
                 }
                 if( i_index == 0 && p_content->i_buffer > i_offset + 8 )
@@ -2386,13 +2511,14 @@ static void ParseTableSection( demux_t *p_demux, ts_pid_t *pid, block_t *p_data
                 }
             }
         }
-        p_content->i_dts =
-        p_content->i_pts = VLC_TS_0 + i_date * 100 / 9;
+
+        p_content->i_dts = p_content->i_pts = VLC_TS_0 + i_date * 100 / 9;
+        PCRFixHandle( p_demux, p_program, p_content );
     }
-    es_out_Send( p_demux->out, pid->es->id, p_content );
 
-    PCRFixHandle( p_demux, p_content );
+    es_out_Send( p_demux->out, pid->es->id, p_content );
 }
+
 static void ParseData( demux_t *p_demux, ts_pid_t *pid )
 {
     block_t *p_data = pid->es->p_data;
@@ -2490,51 +2616,13 @@ static block_t* ReadTSPacket( demux_t *p_demux )
     return p_pkt;
 }
 
-static mtime_t AdjustPCRWrapAround( demux_t *p_demux, mtime_t i_pcr )
+static int64_t TimeStampWrapAround( ts_prg_psi_t *p_prg, int64_t i_time )
 {
-    demux_sys_t   *p_sys = p_demux->p_sys;
-    /*
-     * PCR is 33bit. If PCR reaches to 0x1FFFFFFFF (26:30:43.717), ressets from 0.
-     * So, need to add 0x1FFFFFFFF, for calculating duration or current position.
-     */
-    mtime_t i_adjust = 0;
-    int64_t i_pos = stream_Tell( p_sys->stream );
-    int i;
-    for( i = 1; i < p_sys->i_pcrs_num && p_sys->p_pos[i] <= i_pos; ++i )
-    {
-        if( p_sys->p_pcrs[i-1] > p_sys->p_pcrs[i] )
-            i_adjust += 0x1FFFFFFFF;
-    }
-    if( p_sys->p_pcrs[i-1] > i_pcr )
-        i_adjust += 0x1FFFFFFFF;
-
-    return i_pcr + i_adjust;
-}
-
-static mtime_t AdjustPTSWrapAround( demux_t *p_demux, mtime_t i_pts )
-{
-    demux_sys_t *p_sys = p_demux->p_sys;
-    mtime_t i_pcr = p_sys->i_current_pcr;
-
-    mtime_t i_adjustremain = i_pcr % 0x1FFFFFFFF;
-    mtime_t i_adjustbase = i_pcr - i_adjustremain;
-
-    mtime_t i_pts_adjust = i_adjustbase;
-
-    /* PTS has rolled first */
-    if( i_adjustremain >= 0xFFFFFFFF && i_pts < 0xFFFFFFFF )
-    {
-        i_pts_adjust += 0x1FFFFFFFF;
-    }
-    /* PCR has rolled first (PTS is late!) */
-    else if( i_adjustremain < 0xFFFFFFFF && i_pts >= 0xFFFFFFFF )
-    {
-        if( i_adjustbase >= 0x1FFFFFFFF ) /* we need to remove current roll */
-            i_pts_adjust -= 0x1FFFFFFFF;
-    }
+    int64_t i_adjust = 0;
+    if( p_prg && p_prg->pcr.i_first > 0x0FFFFFFFF && i_time < 0x0FFFFFFFF )
+        i_adjust = 0x1FFFFFFFF;
 
-    i_pts += i_pts_adjust;
-    return i_pts;
+    return i_time + i_adjust;
 }
 
 static mtime_t GetPCR( block_t *p_pkt )
@@ -2557,225 +2645,344 @@ static mtime_t GetPCR( block_t *p_pkt )
     return i_pcr;
 }
 
-static int SeekToPCR( demux_t *p_demux, int64_t i_pos )
+static inline void FlushESBuffer( ts_es_t *p_es )
 {
-    demux_sys_t *p_sys = p_demux->p_sys;
-
-    mtime_t i_pcr = -1;
-    const int64_t i_initial_pos = stream_Tell( p_sys->stream );
-
-    if( i_pos < 0 )
-        return VLC_EGENERIC;
+    if( p_es->p_data )
+    {
+        p_es->i_data_gathered = p_es->i_data_size = 0;
+        block_ChainRelease( p_es->p_data );
+        p_es->p_data = NULL;
+        p_es->pp_last = &p_es->p_data;
+    }
+}
 
-    int64_t i_last_pos = stream_Size( p_sys->stream ) - p_sys->i_packet_size;
-    if( i_pos > i_last_pos )
-        i_pos = i_last_pos;
+static void ReadyQueuesPostSeek( demux_sys_t *p_sys )
+{
+    for( int i=MIN_ES_PID; i<=MAX_ES_PID; i++ )
+    {
+        ts_pid_t *pid = &p_sys->pid[i];
 
-    if( stream_Seek( p_sys->stream, i_pos ) )
-        return VLC_EGENERIC;
+        if( !pid->es )
+            continue;
 
-    for( ;; )
-    {
-        block_t *p_pkt;
+        pid->i_cc = 0xff;
 
-        if( !( p_pkt = ReadTSPacket( p_demux ) ) )
+        if( pid->es->p_prepcr_outqueue )
         {
-            break;
+            block_ChainRelease( pid->es->p_prepcr_outqueue );
+            pid->es->p_prepcr_outqueue = NULL;
         }
-        if( PIDGet( p_pkt ) == p_sys->i_pid_ref_pcr )
-        {
-            i_pcr = GetPCR( p_pkt );
-        }
-        block_Release( p_pkt );
-        if( i_pcr >= 0 )
-            break;
-        if( stream_Tell( p_sys->stream ) >= i_last_pos )
-            break;
-    }
-    if( i_pcr < 0 )
-    {
-        stream_Seek( p_sys->stream, i_initial_pos );
-        assert( i_initial_pos == stream_Tell( p_sys->stream ) );
-        return VLC_EGENERIC;
+
+        FlushESBuffer( pid->es );
+
+        for( int j = 0; j < pid->i_extra_es; j++ )
+            FlushESBuffer( pid->extra_es[j] );
     }
 
-    p_sys->i_current_pcr = i_pcr;
-    return VLC_SUCCESS;
+    for( int i=0; i<p_sys->i_pmt; i++ )
+        for( int j=0; j<p_sys->pmt[i]->psi->i_prg; j++ )
+            p_sys->pmt[i]->psi->prg[j]->pcr.i_current = -1;
 }
 
-static int Seek( demux_t *p_demux, double f_percent )
+static int SeekToTime( demux_t *p_demux, ts_prg_psi_t *p_prg, int64_t i_scaledtime )
 {
     demux_sys_t *p_sys = p_demux->p_sys;
 
-    int64_t i_initial_pos = stream_Tell( p_sys->stream );
-    mtime_t i_initial_pcr = p_sys->i_current_pcr;
+    /* Deal with common but worst binary search case */
+    if( p_prg->pcr.i_first == i_scaledtime && p_sys->b_canseek )
+        return stream_Seek( p_sys->stream, 0 );
 
-    /*
-     * Find the time position by using binary search algorithm.
-     */
-    mtime_t i_target_pcr = (p_sys->i_last_pcr - p_sys->i_first_pcr) * f_percent + p_sys->i_first_pcr;
+    if( !p_sys->b_canfastseek )
+        return VLC_EGENERIC;
+
+    int64_t i_initial_pos = stream_Tell( p_sys->stream );
 
+    /* Find the time position by using binary search algorithm. */
     int64_t i_head_pos = 0;
-    int64_t i_tail_pos;
-    {
-        mtime_t i_adjust = 0;
-        int i;
-        for( i = 1; i < p_sys->i_pcrs_num; ++i )
-        {
-            if( p_sys->p_pcrs[i-1] > p_sys->p_pcrs[i] )
-                i_adjust += 0x1FFFFFFFF;
-            if( p_sys->p_pcrs[i] + i_adjust > i_target_pcr )
-                break;
-        }
-        i_head_pos = p_sys->p_pos[i-1];
-        i_tail_pos = ( i < p_sys->i_pcrs_num ) ?  p_sys->p_pos[i] : stream_Size( p_sys->stream );
-    }
-    msg_Dbg( p_demux, "Seek():i_head_pos:%"PRId64", i_tail_pos:%"PRId64, i_head_pos, i_tail_pos);
+    int64_t i_tail_pos = stream_Size( p_sys->stream ) - p_sys->i_packet_size;
+    if( i_head_pos >= i_tail_pos )
+        return VLC_EGENERIC;
 
     bool b_found = false;
-    int i_cnt = 0;
-    while( i_head_pos <= i_tail_pos )
+    while( (i_head_pos + p_sys->i_packet_size) <= i_tail_pos && !b_found )
     {
         /* Round i_pos to a multiple of p_sys->i_packet_size */
-        int64_t i_pos = i_head_pos + (i_tail_pos - i_head_pos) / 2;
-        int64_t i_div = i_pos % p_sys->i_packet_size;
-        i_pos -= i_div;
-        if( SeekToPCR( p_demux, i_pos ) )
+        int64_t i_splitpos = i_head_pos + (i_tail_pos - i_head_pos) / 2;
+        int64_t i_div = i_splitpos % p_sys->i_packet_size;
+        i_splitpos -= i_div;
+
+        if ( stream_Seek( p_sys->stream, i_splitpos ) != VLC_SUCCESS )
             break;
-        p_sys->i_current_pcr = AdjustPCRWrapAround( p_demux, p_sys->i_current_pcr );
-        int64_t i_diff_msec = (p_sys->i_current_pcr - i_target_pcr) * 100 / 9 / 1000;
-        if( i_diff_msec > 500 )
-        {
-            i_tail_pos = i_pos - p_sys->i_packet_size;
-        }
-        else if( i_diff_msec < -500 )
-        {
-            i_head_pos = i_pos + p_sys->i_packet_size;
-        }
-        else
+
+        int64_t i_pos = i_splitpos;
+        while( i_pos > -1 && i_pos < i_tail_pos )
         {
-            // diff time <= 500msec
-            b_found = true;
-            break;
+            int64_t i_pcr = -1;
+            block_t *p_pkt = ReadTSPacket( p_demux );
+            if( !p_pkt )
+            {
+                i_head_pos = i_tail_pos;
+                break;
+            }
+            else
+                i_pos = stream_Tell( p_sys->stream );
+
+            int i_pid = PIDGet( p_pkt );
+            if( i_pid != 0x1FFF && p_sys->pid[i_pid].b_valid &&
+                p_sys->pid[i_pid].i_owner_number == p_prg->i_number &&
+               (p_pkt->p_buffer[1] & 0xC0) == 0x40 && /* Payload start but not corrupt */
+               (p_pkt->p_buffer[3] & 0xD0) == 0x10    /* Has payload but is not encrypted */
+            )
+            {
+                unsigned i_skip = 4;
+                if ( p_pkt->p_buffer[3] & 0x20 ) // adaptation field
+                {
+                    if( p_pkt->i_buffer >= 4 + 2 + 5 )
+                    {
+                        i_pcr = GetPCR( p_pkt );
+                        i_skip += 1 + p_pkt->p_buffer[4];
+                    }
+                }
+                else
+                {
+                    mtime_t i_dts = -1;
+                    mtime_t i_pts = -1;
+                    if ( VLC_SUCCESS == ParsePESHeader( p_demux, &p_pkt->p_buffer[i_skip],
+                                                        p_pkt->i_buffer - i_skip, &i_skip, &i_dts, &i_pts ) )
+                    {
+                        if( i_dts > -1 )
+                            i_pcr = i_dts;
+                    }
+                }
+            }
+            block_Release( p_pkt );
+
+            if( i_pcr != -1 )
+            {
+                int64_t i_diff = i_scaledtime - TimeStampWrapAround( p_prg, i_pcr );
+                if ( i_diff < 0 )
+                    i_tail_pos = i_splitpos - p_sys->i_packet_size;
+                else if( i_diff < TO_SCALE(VLC_TS_0 + CLOCK_FREQ / 2) ) // 500ms
+                    b_found = true;
+                else
+                    i_head_pos = i_pos;
+                break;
+            }
         }
-        ++i_cnt;
+
     }
+
     if( !b_found )
     {
-        msg_Dbg( p_demux, "Seek():cannot find a time position. i_cnt:%d", i_cnt );
+        msg_Dbg( p_demux, "Seek():cannot find a time position." );
         stream_Seek( p_sys->stream, i_initial_pos );
-        p_sys->i_current_pcr = i_initial_pcr;
         return VLC_EGENERIC;
     }
-    else
+    return VLC_SUCCESS;
+}
+
+static ts_prg_psi_t * GetProgramByID( demux_sys_t *p_sys, int i_program )
+{
+    for( int i = 0; i < p_sys->i_pmt; i++ )
     {
-        msg_Dbg( p_demux, "Seek():can find a time position. i_cnt:%d", i_cnt );
-        p_demux->p_sys->pcrfix.i_first_dts = 0;
-        return VLC_SUCCESS;
+        for( int i_prg = 0; i_prg < p_sys->pmt[i]->psi->i_prg; i_prg++ )
+        {
+            ts_prg_psi_t *p_prg = p_sys->pmt[i]->psi->prg[i_prg];
+            if( p_prg->i_number == i_program )
+                return p_prg;
+        }
     }
+    return NULL;
 }
 
-static void GetFirstPCR( demux_t *p_demux )
+#define PROBE_CHUNK_COUNT 250
+
+static int ProbeChunk( demux_t *p_demux, int i_program, bool b_end, int64_t *pi_pcr, bool *pb_found )
 {
     demux_sys_t *p_sys = p_demux->p_sys;
-
-    int64_t i_initial_pos = stream_Tell( p_sys->stream );
-
-    if( stream_Seek( p_sys->stream, 0 ) )
-        return;
+    int i_count = 0;
+    block_t *p_pkt = NULL;
+    *pi_pcr = -1;
 
     for( ;; )
     {
-        block_t     *p_pkt;
-
-        if( !( p_pkt = ReadTSPacket( p_demux ) ) )
+        if( i_count++ > PROBE_CHUNK_COUNT || !( p_pkt = ReadTSPacket( p_demux ) ) )
         {
             break;
         }
-        mtime_t i_pcr = GetPCR( p_pkt );
-        if( i_pcr >= 0 )
+
+        int i_pid = PIDGet( p_pkt );
+        p_sys->pid[i_pid].b_seen = true;
+
+        if( i_pid != 0x1FFF && p_sys->pid[i_pid].b_valid && p_sys->pid[i_pid].p_owner &&
+           (p_pkt->p_buffer[1] & 0xC0) == 0x40 && /* Payload start but not corrupt */
+           (p_pkt->p_buffer[3] & 0xD0) == 0x10    /* Has payload but is not encrypted */
+          )
         {
-            p_sys->i_pid_ref_pcr = PIDGet( p_pkt );
-            p_sys->i_first_pcr = i_pcr;
-            p_sys->i_current_pcr = i_pcr;
+            bool b_pcrresult = true;
+
+            if( p_pkt->i_buffer >= 4 + 2 + 5 )
+                *pi_pcr = GetPCR( p_pkt );
+
+            if( *pi_pcr == -1 )
+            {
+                b_pcrresult = false;
+                mtime_t i_dts = -1;
+                mtime_t i_pts = -1;
+                unsigned i_skip = 4;
+                if ( p_pkt->p_buffer[3] & 0x20 ) // adaptation field
+                    i_skip += 1 + p_pkt->p_buffer[4];
+
+                if ( VLC_SUCCESS == ParsePESHeader( p_demux, &p_pkt->p_buffer[i_skip],
+                                                    p_pkt->i_buffer - i_skip,
+                                                    &i_skip, &i_dts, &i_pts ) )
+                {
+                    if( i_dts != -1 )
+                        *pi_pcr = i_dts;
+                    else if( i_pts != -1 )
+                        *pi_pcr = i_pts;
+                }
+            }
+
+            if( *pi_pcr != -1 )
+            {
+                ts_psi_t *p_prg = p_sys->pid[i_pid].p_owner;
+                for(int i=0; i<p_prg->i_prg; i++)
+                {
+                    if( i_program == 0 || i_program == p_prg->prg[i]->i_number )
+                        *pb_found = true;
+
+                    if( b_end )
+                    {
+                        p_prg->prg[i]->i_last_dts = *pi_pcr;
+                    }
+                    /* Start, only keep first */
+                    else if( b_pcrresult && p_prg->prg[i]->pcr.i_first == -1 )
+                    {
+                        p_prg->prg[i]->pcr.i_first = *pi_pcr;
+                    }
+                    else if( p_prg->prg[i]->pcr.i_first_dts < VLC_TS_0 )
+                    {
+                        p_prg->prg[i]->pcr.i_first_dts = VLC_TS_0 + *pi_pcr * 100 / 9;
+                    }
+                }
+            }
         }
+
         block_Release( p_pkt );
-        if( p_sys->i_first_pcr >= 0 )
-            break;
     }
-    stream_Seek( p_sys->stream, i_initial_pos );
+
+    return i_count;
 }
 
-static void GetLastPCR( demux_t *p_demux )
+static int ProbeStart( demux_t *p_demux, int i_program )
 {
     demux_sys_t *p_sys = p_demux->p_sys;
-
     const int64_t i_initial_pos = stream_Tell( p_sys->stream );
-    mtime_t i_initial_pcr = p_sys->i_current_pcr;
-
     int64_t i_stream_size = stream_Size( p_sys->stream );
-    int64_t i_last_pos = i_stream_size - p_sys->i_packet_size;
-    /* Round i_pos to a multiple of p_sys->i_packet_size */
-    int64_t i_pos = i_last_pos - p_sys->i_packet_size * 4500; /* FIXME if the value is not reasonable, please change it. */
-    int64_t i_div = i_pos % p_sys->i_packet_size;
-    i_pos -= i_div;
-
-    if( i_pos <= i_initial_pos && i_pos >= i_stream_size )
-        i_pos = i_initial_pos + p_sys->i_packet_size;
-    if( i_pos < 0 && i_pos >= i_stream_size )
-        return;
 
-    for( ;; )
+    int i_probe_count = 0;
+    int64_t i_pos;
+    mtime_t i_pcr = -1;
+    bool b_found = false;
+
+    do
     {
-        if( SeekToPCR( p_demux, i_pos ) )
-            break;
-        p_sys->i_last_pcr = AdjustPCRWrapAround( p_demux, p_sys->i_current_pcr );
-        if( ( i_pos = stream_Tell( p_sys->stream ) ) >= i_last_pos )
-            break;
-    }
+        i_pos = p_sys->i_packet_size * i_probe_count;
+        i_pos = __MIN( i_pos, i_stream_size );
+
+        if( stream_Seek( p_sys->stream, i_pos ) )
+            return VLC_EGENERIC;
+
+        ProbeChunk( p_demux, i_program, false, &i_pcr, &b_found );
+
+        /* Go ahead one more chunk if end of file contained only stuffing packets */
+        i_probe_count += PROBE_CHUNK_COUNT;
+    } while( i_pos > 0 && (i_pcr == -1 || !b_found) && i_probe_count < (2 * PROBE_CHUNK_COUNT) );
 
     stream_Seek( p_sys->stream, i_initial_pos );
-    assert( i_initial_pos == stream_Tell( p_sys->stream ) );
-    p_sys->i_current_pcr = i_initial_pcr;
+
+    return (b_found) ? VLC_SUCCESS : VLC_EGENERIC;
 }
 
-static void CheckPCR( demux_t *p_demux )
+static int ProbeEnd( demux_t *p_demux, int i_program )
 {
-    demux_sys_t   *p_sys = p_demux->p_sys;
+    demux_sys_t *p_sys = p_demux->p_sys;
+    const int64_t i_initial_pos = stream_Tell( p_sys->stream );
+    int64_t i_stream_size = stream_Size( p_sys->stream );
 
-    int64_t i_initial_pos = stream_Tell( p_sys->stream );
-    mtime_t i_initial_pcr = p_sys->i_current_pcr;
+    int i_probe_count = PROBE_CHUNK_COUNT;
+    int64_t i_pos;
+    mtime_t i_pcr = -1;
+    bool b_found = false;
+
+    do
+    {
+        i_pos = i_stream_size - (p_sys->i_packet_size * i_probe_count);
+        i_pos = __MAX( i_pos, 0 );
+
+        if( stream_Seek( p_sys->stream, i_pos ) )
+            return VLC_EGENERIC;
+
+        ProbeChunk( p_demux, i_program, true, &i_pcr, &b_found );
+
+        /* Go ahead one more chunk if end of file contained only stuffing packets */
+        i_probe_count += PROBE_CHUNK_COUNT;
+    } while( i_pos > 0 && (i_pcr == -1 || !b_found) && i_probe_count < (6 * PROBE_CHUNK_COUNT) );
+
+    stream_Seek( p_sys->stream, i_initial_pos );
 
-    int64_t i_size = stream_Size( p_sys->stream );
+    return (b_found) ? VLC_SUCCESS : VLC_EGENERIC;
+}
 
-    int i = 0;
-    p_sys->p_pcrs[0] = p_sys->i_first_pcr;
-    p_sys->p_pos[0] = i_initial_pos;
+static void ProgramSetPCR( demux_t *p_demux, ts_prg_psi_t *p_prg, mtime_t i_pcr )
+{
+    demux_sys_t *p_sys = p_demux->p_sys;
 
-    for( i = 1; i < p_sys->i_pcrs_num; ++i )
+    /* Check if we have enqueued blocks waiting the/before the
+       PCR barrier, and then adapt pcr so they have valid PCR when dequeuing */
+    if( p_prg->pcr.i_current == -1 )
     {
-        /* Round i_pos to a multiple of p_sys->i_packet_size */
-        int64_t i_pos = i_size / p_sys->i_pcrs_num * i;
-        int64_t i_div = i_pos % p_sys->i_packet_size;
-        i_pos -= i_div;
-        if( SeekToPCR( p_demux, i_pos ) )
-            break;
-        p_sys->p_pcrs[i] = p_sys->i_current_pcr;
-        p_sys->p_pos[i] = stream_Tell( p_sys->stream );
-        if( p_sys->p_pcrs[i-1] > p_sys->p_pcrs[i] )
+        mtime_t i_mindts = -1;
+        for( int i=MIN_ES_PID; i<=MAX_ES_PID; i++ )
+        {
+            ts_pid_t *p_pid = &p_sys->pid[i];
+            if( p_pid->b_valid && p_pid->i_owner_number == p_prg->i_number && p_pid->es )
+            {
+                block_t *p_block = p_pid->es->p_prepcr_outqueue;
+                while( p_block && p_block->i_dts == VLC_TS_INVALID )
+                    p_block = p_block->p_next;
+
+                if( p_block )
+                    msg_Warn( p_demux, "Program %ld %ld *******", i_mindts, p_block->i_dts );
+
+                if( p_block && ( i_mindts == -1 || p_block->i_dts < i_mindts ) )
+                {
+                    i_mindts = p_block->i_dts;
+                    msg_Warn( p_demux, "Program %ld %ld *******", i_mindts, p_block->i_dts );
+                }
+            }
+        }
+
+        if( i_mindts > VLC_TS_INVALID )
         {
-            msg_Dbg( p_demux, "PCR Wrap Around found between %d%% and %d%% (pcr:%"PRId64"(0x%09"PRIx64") pcr:%"PRId64"(0x%09"PRIx64"))",
-                    (int)((i-1)*100/p_sys->i_pcrs_num), (int)(i*100/p_sys->i_pcrs_num), p_sys->p_pcrs[i-1], p_sys->p_pcrs[i-1], p_sys->p_pcrs[i], p_sys->p_pcrs[i] );
+            msg_Dbg( p_demux, "Program %d PCR prequeue fixup %"PRId64"->%"PRId64,
+                     p_prg->i_number, TO_SCALE(i_mindts), i_pcr );
+            i_pcr = TO_SCALE(i_mindts);
         }
     }
-    if( i < p_sys->i_pcrs_num )
+
+    p_prg->pcr.i_current = i_pcr;
+    if( p_prg->pcr.i_first == -1 )
     {
-        msg_Dbg( p_demux, "Force Seek Per Percent: Seeking failed at %d%%.", (int)(i*100/p_sys->i_pcrs_num) );
-        p_sys->b_force_seek_per_percent = true;
+        p_prg->pcr.i_first = i_pcr; // now seen
     }
 
-    stream_Seek( p_sys->stream, i_initial_pos );
-    p_sys->i_current_pcr = i_initial_pcr;
+    if ( p_sys->i_pmt_es )
+    {
+        es_out_Control( p_demux->out, ES_OUT_SET_GROUP_PCR,
+                        p_prg->i_number, VLC_TS_0 + i_pcr * 100 / 9 );
+    }
 }
 
 static void PCRHandle( demux_t *p_demux, ts_pid_t *pid, block_t *p_bk )
@@ -2789,28 +2996,22 @@ static void PCRHandle( demux_t *p_demux, ts_pid_t *pid, block_t *p_bk )
     if( i_pcr < 0 )
         return;
 
-    i_pcr = AdjustPCRWrapAround( p_demux, i_pcr );
-
-    if( p_sys->i_pid_ref_pcr == pid->i_pid )
-        p_sys->i_current_pcr = i_pcr;
+    pid->probed.i_pcr_count++;
 
     /* Search program and set the PCR */
     for( int i = 0; i < p_sys->i_pmt; i++ )
     {
-        int i_group = -1;
         for( int i_prg = 0; i_prg < p_sys->pmt[i]->psi->i_prg; i_prg++ )
         {
             ts_prg_psi_t *p_prg = p_sys->pmt[i]->psi->prg[i_prg];
+            mtime_t i_program_pcr = TimeStampWrapAround( p_prg, i_pcr );
 
             if( p_prg->i_pid_pcr == 0x1FFF ) /* That program has no dedicated PCR pid ISO/IEC 13818-1 2.4.4.9 */
             {
                 if( pid->p_owner ) /* PCR shall be on pid itself */
                 {
-                    p_prg->i_pcr_value = i_pcr; /* ? update PCR for the whole group program */
-                    i_group = p_prg->i_number;
-                    p_sys->pcrfix.b_program_pcr_seen = true;
-                    p_sys->pcrfix.i_first_dts = 0;
-                    p_sys->b_disable_pcr = !p_sys->b_trust_pcr;
+                    /* ? update PCR for the whole group program ? */
+                    ProgramSetPCR( p_demux, p_prg, i_program_pcr );
                 }
                 else
                 {
@@ -2823,38 +3024,70 @@ static void PCRHandle( demux_t *p_demux, ts_pid_t *pid, block_t *p_bk )
                 /* Can be dedicated PCR pid (no owned then) or another pid (owner == pmt) */
                 if( p_prg->i_pid_pcr == pid->i_pid ) /* If that program references current pid as PCR */
                 {
-                    p_prg->i_pcr_value = i_pcr;
-                    i_group = p_prg->i_number; /* We've found a target group for update */
-                    p_sys->pcrfix.b_program_pcr_seen = true;
-                    p_sys->pcrfix.i_first_dts = 0;
-                    p_sys->b_disable_pcr = !p_sys->b_trust_pcr;
+                    /* We've found a target group for update */
+                    ProgramSetPCR( p_demux, p_prg, i_program_pcr );
                 }
             }
         }
-        if ( !p_sys->b_disable_pcr && i_group > 0 && p_sys->i_pmt_es )
+    }
+}
+
+static int FindPCRCandidate( demux_sys_t *p_sys, ts_prg_psi_t *p_prg )
+{
+    ts_pid_t *p_cand = NULL;
+    int i_previous = p_prg->i_pid_pcr;
+    for( int i=MIN_ES_PID; i<=MAX_ES_PID; i++ )
+    {
+        ts_pid_t *p_pid = &p_sys->pid[i];
+        if( p_pid->b_seen && p_pid->es && p_pid->es->id &&
+            p_pid->i_owner_number == p_prg->i_number &&
+            p_cand->i_pid != i_previous )
         {
-            es_out_Control( p_demux->out, ES_OUT_SET_GROUP_PCR,
-              i_group, VLC_TS_0 + i_pcr * 100 / 9 );
+            if( p_pid->probed.i_pcr_count ) /* check PCR frequency first */
+            {
+                if( !p_cand || p_pid->probed.i_pcr_count > p_cand->probed.i_pcr_count )
+                {
+                    p_cand = p_pid;
+                    continue;
+                }
+            }
+
+            if( p_pid->es->fmt.i_cat == AUDIO_ES )
+            {
+                if( !p_cand )
+                    p_cand = p_pid;
+            }
+            else if ( p_pid->es->fmt.i_cat == VIDEO_ES ) /* Otherwise prioritize video dts */
+            {
+                if( !p_cand || p_cand->es->fmt.i_cat == AUDIO_ES )
+                    p_cand = p_pid;
+            }
         }
     }
+
+    if( p_cand )
+        return p_cand->i_pid;
+    else
+        return 0x1FFF;
 }
 
-static void PCRFixHandle( demux_t *p_demux, block_t *p_block )
+static void PCRFixHandle( demux_t *p_demux, ts_prg_psi_t *p_prg, block_t *p_block )
 {
-    demux_sys_t *p_sys = p_demux->p_sys;
+    if( p_prg->pcr.i_first > -1 || p_prg->pcr.b_disable )
+        return;
+
     /* Record the first data packet timestamp in case there wont be any PCR */
-    if( !p_sys->pcrfix.b_program_pcr_seen && !p_sys->b_disable_pcr )
+    if( !p_prg->pcr.i_first_dts )
     {
-        if( !p_sys->pcrfix.i_first_dts )
-        {
-            p_sys->pcrfix.i_first_dts = p_block->i_dts;
-        }
-        else if( p_block->i_dts - p_sys->pcrfix.i_first_dts > CLOCK_FREQ / 10 ) /* "shall not exceed 100ms" */
-        {
-            p_sys->b_disable_pcr = true;
-            msg_Warn( p_demux, "No PCR received for program %d, set up workaround",
-                      p_sys->i_current_program );
-        }
+        p_prg->pcr.i_first_dts = p_block->i_dts;
+    }
+    else if( p_block->i_dts - p_prg->pcr.i_first_dts > CLOCK_FREQ / 2 ) /* "shall not exceed 100ms" */
+    {
+        int i_cand = FindPCRCandidate( p_demux->p_sys, p_prg );
+        p_prg->i_pid_pcr = i_cand;
+        p_prg->pcr.b_disable = true; /* So we do not wait packet PCR flag as there might be none on the pid */
+        msg_Warn( p_demux, "No PCR received for program %d, set up workaround using pid %d",
+                  p_prg->i_number, i_cand );
     }
 }
 
@@ -3366,20 +3599,10 @@ static bool ProgramIsSelected( demux_t *p_demux, uint16_t i_pgrm )
 {
     demux_sys_t          *p_sys = p_demux->p_sys;
 
-    if( ( p_sys->i_current_program == -1 && p_sys->programs_list.i_count == 0 ) ||
-        p_sys->i_current_program == 0 )
-        return true;
-    if( p_sys->i_current_program == i_pgrm )
-        return true;
+    for(int i=0; i<p_sys->programs.i_size; i++)
+        if( p_sys->programs.p_elems[i] == i_pgrm )
+            return true;
 
-    if( p_sys->programs_list.i_count != 0 )
-    {
-        for( int i = 0; i < p_sys->programs_list.i_count; i++ )
-        {
-            if( i_pgrm == p_sys->programs_list.p_values[i].i_int )
-                return true;
-        }
-    }
     return false;
 }
 
@@ -3475,7 +3698,7 @@ static void SDTCallBack( demux_t *p_demux, dvbpsi_sdt_t *p_sdt )
 
     msg_Dbg( p_demux, "SDTCallBack called" );
 
-    if( p_sys->b_delay_es_creation ||
+    if( p_sys->es_creation != CREATE_ES ||
        !p_sdt->b_current_next ||
         p_sdt->i_version == sdt->psi->i_sdt_version )
     {
@@ -3873,8 +4096,8 @@ static void EITCallBack( demux_t *p_demux,
     if( p_epg->i_event > 0 )
     {
         if( b_current_following &&
-            (  p_sys->i_current_program == -1 ||
-               p_sys->i_current_program ==
+            (  p_sys->programs.i_size == 0 ||
+               p_sys->programs.p_elems[0] ==
 #if (DVBPSI_VERSION_INT >= DVBPSI_VERSION_WANTED(1,0,0))
                     p_eit->i_extension
 #else
@@ -4907,13 +5130,13 @@ static void AddAndCreateES( demux_t *p_demux, ts_pid_t *pid )
 
     if( pid )
     {
-        if( pid->b_seen && p_sys->b_delay_es_creation )
+        if( pid->b_seen && p_sys->es_creation == DELAY_ES )
         {
-            p_sys->b_delay_es_creation = false;
+            p_sys->es_creation = CREATE_ES;
             b_create_delayed = true;
         }
 
-        if( !p_sys->b_delay_es_creation )
+        if( p_sys->es_creation == CREATE_ES )
         {
             pid->es->id = es_out_Add( p_demux->out, &pid->es->fmt );
             for( int i = 0; i < pid->i_extra_es; i++ )
@@ -4924,9 +5147,9 @@ static void AddAndCreateES( demux_t *p_demux, ts_pid_t *pid )
             p_sys->i_pmt_es += 1 + pid->i_extra_es;
         }
     }
-    else if( p_sys->b_delay_es_creation )
+    else if( p_sys->es_creation == DELAY_ES )
     {
-        p_sys->b_delay_es_creation = false;
+        p_sys->es_creation = CREATE_ES;
         b_create_delayed = true;
     }
 
@@ -4944,6 +5167,9 @@ static void AddAndCreateES( demux_t *p_demux, ts_pid_t *pid )
                     es_out_Add( p_demux->out, &pid->extra_es[j]->fmt );
             }
             p_sys->i_pmt_es += 1 + pid->i_extra_es;
+
+            if( pid->es->id != NULL && ProgramIsSelected( p_demux, pid->i_owner_number ) )
+                SetPIDFilter( p_demux, pid->i_pid, true );
         }
     }
 }
@@ -5344,6 +5570,23 @@ static void PMTCallBack( void *data, dvbpsi_pmt_t *p_pmt )
     }
     if( i_clean )
         free( pp_clean );
+
+    if( !p_sys->b_trust_pcr )
+    {
+        int i_cand = FindPCRCandidate( p_demux->p_sys, prg );
+        prg->i_pid_pcr = i_cand;
+        prg->pcr.b_disable = true;
+        msg_Warn( p_demux, "PCR not trusted for program %d, set up workaround using pid %d",
+                  prg->i_number, i_cand );
+    }
+
+    /* Probe Boundaries */
+    if( p_sys->b_canfastseek && prg->i_last_dts == -1 )
+    {
+        prg->i_last_dts = 0;
+        ProbeStart( p_demux, prg->i_number );
+        ProbeEnd( p_demux, prg->i_number );
+    }
 }
 
 static int PATCheck( demux_t *p_demux, dvbpsi_pat_t *p_pat )
@@ -5506,13 +5749,19 @@ static void PATCallBack( void *data, dvbpsi_pat_t *p_pat )
         prg->i_pid_pmt = p_program->i_pid;
 
         /* Now select PID at access level */
-        if( ProgramIsSelected( p_demux, p_program->i_number ) )
+        if( p_sys->programs.i_size == 0 ||
+            ProgramIsSelected( p_demux, p_program->i_number ) )
         {
-            if( p_sys->i_current_program == 0 )
-                p_sys->i_current_program = p_program->i_number;
+            if( p_sys->programs.i_size == 0 )
+            {
+                p_sys->b_default_selection = true;
+                ARRAY_APPEND( p_sys->programs, p_program->i_number );
+            }
 
             if( SetPIDFilter( p_demux, p_program->i_pid, true ) )
                 p_sys->b_access_control = false;
+            else if ( p_sys->es_creation == DELAY_ES )
+                p_sys->es_creation = CREATE_ES;
         }
     }
     pat->psi->i_pat_version = p_pat->i_version;