git.sesse.net Git - vlc/blob - modules/demux/mpeg/system.c

   1 /*****************************************************************************
   2  * system.c: helper module for TS, PS and PES management
   3  *****************************************************************************
   4  * Copyright (C) 1998-2002 VideoLAN
   5  * $Id: system.c,v 1.19 2003/11/06 16:36:41 nitrox Exp $
   6  *
   7  * Authors: Christophe Massiot <massiot@via.ecp.fr>
   8  *          Michel Lespinasse <walken@via.ecp.fr>
   9  *          Benoît Steiner <benny@via.ecp.fr>
  10  *          Samuel Hocevar <sam@zoy.org>
  11  *          Henri Fallon <henri@via.ecp.fr>
  12  *
  13  * This program is free software; you can redistribute it and/or modify
  14  * it under the terms of the GNU General Public License as published by
  15  * the Free Software Foundation; either version 2 of the License, or
  16  * (at your option) any later version.
  17  *
  18  * This program is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU General Public License for more details.
  22  *
  23  * You should have received a copy of the GNU General Public License
  24  * along with this program; if not, write to the Free Software
  25  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  26  *****************************************************************************/
  27
  28 /*****************************************************************************
  29  * Preamble
  30  *****************************************************************************/
  31 #include <stdlib.h>
  32
  33 #include <vlc/vlc.h>
  34 #include <vlc/input.h>
  35
  36 #include "system.h"
  37
  38 /*****************************************************************************
  39  * Local prototypes
  40  *****************************************************************************/
  41 static int Activate ( vlc_object_t * );
  42
  43 static ssize_t           ReadPS  ( input_thread_t *, data_packet_t ** );
  44 static es_descriptor_t * ParsePS ( input_thread_t *, data_packet_t * );
  45 static void              DemuxPS ( input_thread_t *, data_packet_t * );
  46
  47 static ssize_t           ReadTS  ( input_thread_t *, data_packet_t ** );
  48 static void              DemuxTS ( input_thread_t *, data_packet_t *,
  49                                    psi_callback_t );
  50
  51 /*****************************************************************************
  52  * Module descriptor
  53  *****************************************************************************/
  54 vlc_module_begin();
  55     set_description( _("generic ISO 13818-1 MPEG demultiplexing") );
  56     set_capability( "mpeg-system", 100 );
  57     set_callbacks( Activate, NULL );
  58 vlc_module_end();
  59
  60 /*****************************************************************************
  61  * Activate: initializes helper functions
  62  *****************************************************************************/
  63 static int Activate ( vlc_object_t *p_this )
  64 {
  65     static mpeg_demux_t mpeg_demux =
  66                     { NULL, ReadPS, ParsePS, DemuxPS, ReadTS, DemuxTS };
  67
  68     memcpy( p_this->p_private, &mpeg_demux, sizeof( mpeg_demux ) );
  69
  70     return VLC_SUCCESS;
  71 }
  72
  73 /*
  74  * PES Packet management
  75  */
  76
  77 /*****************************************************************************
  78  * MoveChunk
  79  *****************************************************************************
  80  * Small utility function used to parse discontinuous headers safely. Copies
  81  * i_buf_len bytes of data to a buffer and returns the size copied.
  82  * It also solves some alignment problems on non-IA-32, non-PPC processors.
  83  * This is a variation on the theme of input_ext-dec.h:GetChunk().
  84  *****************************************************************************/
  85 static inline size_t MoveChunk( byte_t * p_dest, data_packet_t ** pp_data_src,
  86                                 byte_t ** pp_src, size_t i_buf_len )
  87 {
  88     size_t i_available;
  89
  90     i_available = (ptrdiff_t)((*pp_data_src)->p_payload_end - *pp_src);
  91     if( i_available >= i_buf_len )
  92     {
  93         if( p_dest != NULL )
  94             memcpy( p_dest, *pp_src, i_buf_len );
  95         *pp_src += i_buf_len;
  96         return( i_buf_len );
  97     }
  98     else
  99     {
 100         size_t i_init_len = i_buf_len;
 101
 102         do
 103         {
 104             if( p_dest != NULL )
 105                 memcpy( p_dest, *pp_src, i_available );
 106             *pp_data_src = (*pp_data_src)->p_next;
 107             i_buf_len -= i_available;
 108             p_dest += i_available;
 109             if( *pp_data_src == NULL )
 110             {
 111                 *pp_src = NULL;
 112                 return( i_init_len - i_buf_len );
 113             }
 114             *pp_src = (*pp_data_src)->p_payload_start;
 115
 116             i_available = (ptrdiff_t)((*pp_data_src)->p_payload_end - *pp_src);
 117         }
 118         while( i_available <= i_buf_len );
 119
 120         if( i_buf_len )
 121         {
 122             if( p_dest != NULL )
 123                 memcpy( p_dest, *pp_src, i_buf_len );
 124             *pp_src += i_buf_len;
 125         }
 126         return( i_init_len );
 127     }
 128 }
 129
 130 /*****************************************************************************
 131  * ParsePES
 132  *****************************************************************************
 133  * Parse a finished PES packet and analyze its header.
 134  *****************************************************************************/
 135 #define PES_HEADER_SIZE     7
 136 static void ParsePES( input_thread_t * p_input, es_descriptor_t * p_es )
 137 {
 138     data_packet_t * p_data;
 139     byte_t *        p_byte;
 140     byte_t          p_header[PES_HEADER_SIZE];
 141     int             i_done;
 142
 143 #define p_pes (p_es->p_pes)
 144
 145     /* Parse the header. The header has a variable length, but in order
 146      * to improve the algorithm, we will read the 14 bytes we may be
 147      * interested in */
 148     p_data = p_pes->p_first;
 149     p_byte = p_data->p_payload_start;
 150     i_done = 0;
 151
 152     if( MoveChunk( p_header, &p_data, &p_byte, PES_HEADER_SIZE )
 153             != PES_HEADER_SIZE )
 154     {
 155         msg_Warn( p_input, "PES packet too short to have a header" );
 156         input_DeletePES( p_input->p_method_data, p_pes );
 157         p_pes = NULL;
 158         return;
 159     }
 160
 161     /* Get the PES size if defined */
 162     p_es->i_pes_real_size = U16_AT(p_header + 4);
 163     if( p_es->i_pes_real_size )
 164     {
 165         p_es->i_pes_real_size += 6;
 166     }
 167
 168     /* First read the 6 header bytes common to all PES packets:
 169      * use them to test the PES validity */
 170     if( (p_header[0] || p_header[1] || (p_header[2] != 1)) )
 171     {
 172         /* packet_start_code_prefix != 0x000001 */
 173         msg_Err( p_input, "data loss, PES packet does not start with 000001" );
 174         input_DeletePES( p_input->p_method_data, p_pes );
 175         p_pes = NULL;
 176     }
 177     else
 178     {
 179         unsigned int i_pes_header_size, i_payload_size;
 180
 181         if ( p_es->i_pes_real_size &&
 182              (p_es->i_pes_real_size != p_pes->i_pes_size) )
 183         {
 184             /* PES_packet_length is set and != total received payload */
 185             /* Warn the decoder that the data may be corrupt. */
 186             msg_Warn( p_input, "packet corrupted, PES sizes do not match" );
 187         }
 188
 189         switch( p_header[3] )
 190         {
 191         case 0xBC:  /* Program stream map */
 192         case 0xBE:  /* Padding */
 193         case 0xBF:  /* Private stream 2 */
 194         case 0xB0:  /* ECM */
 195         case 0xB1:  /* EMM */
 196         case 0xFF:  /* Program stream directory */
 197         case 0xF2:  /* DSMCC stream */
 198         case 0xF8:  /* ITU-T H.222.1 type E stream */
 199             /* The payload begins immediately after the 6 bytes header, so
 200              * we have finished with the parsing */
 201             i_pes_header_size = 6;
 202             break;
 203
 204         default:
 205             if( (p_header[6] & 0xC0) == 0x80 )
 206             {
 207                 /* MPEG-2 : the PES header contains at least 3 more bytes. */
 208                 size_t      i_max_len;
 209                 vlc_bool_t  b_has_pts, b_has_dts;
 210                 byte_t      p_full_header[12];
 211
 212                 p_pes->b_data_alignment = p_header[6] & 0x04;
 213
 214                 i_max_len = MoveChunk( p_full_header, &p_data, &p_byte, 12 );
 215                 if( i_max_len < 2 )
 216                 {
 217                     msg_Warn( p_input,
 218                               "PES packet too short to have a MPEG-2 header" );
 219                     input_DeletePES( p_input->p_method_data,
 220                                             p_pes );
 221                     p_pes = NULL;
 222                     return;
 223                 }
 224
 225                 b_has_pts = p_full_header[0] & 0x80;
 226                 b_has_dts = p_full_header[0] & 0x40;
 227                 i_pes_header_size = p_full_header[1] + 9;
 228
 229                 /* Now parse the optional header extensions */
 230                 if( b_has_pts )
 231                 {
 232                     if( i_max_len < 7 )
 233                     {
 234                         msg_Warn( p_input,
 235                              "PES packet too short to have a MPEG-2 header" );
 236                         input_DeletePES( p_input->p_method_data,
 237                                                 p_pes );
 238                         p_pes = NULL;
 239                         return;
 240                     }
 241                     p_pes->i_pts = input_ClockGetTS( p_input, p_es->p_pgrm,
 242                     ( ((mtime_t)(p_full_header[2] & 0x0E) << 29) |
 243                       ((mtime_t)(p_full_header[3]) << 22) |
 244                       ((mtime_t)(p_full_header[4] & 0xFE) << 14) |
 245                       ((mtime_t)p_full_header[5] << 7) |
 246                       ((mtime_t)p_full_header[6] >> 1) ) );
 247
 248                     if( b_has_dts )
 249                     {
 250                         if( i_max_len < 12 )
 251                         {
 252                             msg_Warn( p_input,
 253                               "PES packet too short to have a MPEG-2 header" );
 254                             input_DeletePES( p_input->p_method_data,
 255                                                     p_pes );
 256                             p_pes = NULL;
 257                             return;
 258                         }
 259                         p_pes->i_dts = input_ClockGetTS( p_input, p_es->p_pgrm,
 260                         ( ((mtime_t)(p_full_header[7] & 0x0E) << 29) |
 261                           (((mtime_t)U16_AT(p_full_header + 8) << 14)
 262                                 - (1 << 14)) |
 263                           ((mtime_t)U16_AT(p_full_header + 10) >> 1) ) );
 264                     }
 265                 }
 266             }
 267             else
 268             {
 269                 /* Probably MPEG-1 */
 270                 vlc_bool_t      b_has_pts, b_has_dts;
 271
 272                 i_pes_header_size = 6;
 273                 p_data = p_pes->p_first;
 274                 p_byte = p_data->p_payload_start;
 275                 /* Cannot fail because the previous one succeeded. */
 276                 MoveChunk( NULL, &p_data, &p_byte, 6 );
 277
 278                 while( *p_byte == 0xFF && i_pes_header_size < 23 )
 279                 {
 280                     i_pes_header_size++;
 281                     if( MoveChunk( NULL, &p_data, &p_byte, 1 ) != 1 )
 282                     {
 283                         msg_Warn( p_input,
 284                             "PES packet too short to have a MPEG-1 header" );
 285                         input_DeletePES( p_input->p_method_data, p_pes );
 286                         p_pes = NULL;
 287                         return;
 288                     }
 289                 }
 290                 if( i_pes_header_size == 23 )
 291                 {
 292                     msg_Err( p_input, "too much MPEG-1 stuffing" );
 293                     input_DeletePES( p_input->p_method_data, p_pes );
 294                     p_pes = NULL;
 295                     return;
 296                 }
 297
 298                 if( (*p_byte & 0xC0) == 0x40 )
 299                 {
 300                     /* Don't ask why... --Meuuh */
 301                     /* Erm... why ? --Sam */
 302                     /* Well... According to the recommendation, it is for
 303                      * STD_buffer_scale and STD_buffer_size. --Meuuh */
 304                     i_pes_header_size += 2;
 305                     if( MoveChunk( NULL, &p_data, &p_byte, 2 ) != 2 )
 306                     {
 307                         msg_Warn( p_input,
 308                             "PES packet too short to have a MPEG-1 header" );
 309                         input_DeletePES( p_input->p_method_data, p_pes );
 310                         p_pes = NULL;
 311                         return;
 312                     }
 313                 }
 314
 315                 i_pes_header_size++;
 316
 317                 b_has_pts = *p_byte & 0x20;
 318                 b_has_dts = *p_byte & 0x10;
 319
 320                 if( b_has_pts )
 321                 {
 322                     byte_t      p_ts[5];
 323
 324                     i_pes_header_size += 4;
 325                     if( MoveChunk( p_ts, &p_data, &p_byte, 5 ) != 5 )
 326                     {
 327                         msg_Warn( p_input,
 328                             "PES packet too short to have a MPEG-1 header" );
 329                         input_DeletePES( p_input->p_method_data, p_pes );
 330                         p_pes = NULL;
 331                         return;
 332                     }
 333
 334                     p_pes->i_pts = input_ClockGetTS( p_input, p_es->p_pgrm,
 335                        ( ((mtime_t)(p_ts[0] & 0x0E) << 29) |
 336                          (((mtime_t)U32_AT(p_ts) & 0xFFFE00) << 6) |
 337                          ((mtime_t)p_ts[3] << 7) |
 338                          ((mtime_t)p_ts[4] >> 1) ) );
 339
 340                     if( b_has_dts )
 341                     {
 342                         i_pes_header_size += 5;
 343                         if( MoveChunk( p_ts, &p_data, &p_byte, 5 ) != 5 )
 344                         {
 345                             msg_Warn( p_input,
 346                               "PES packet too short to have a MPEG-1 header" );
 347                             input_DeletePES( p_input->p_method_data, p_pes );
 348                             p_pes = NULL;
 349                             return;
 350                         }
 351
 352                         p_pes->i_dts = input_ClockGetTS( p_input,
 353                                                          p_es->p_pgrm,
 354                             ( ((mtime_t)(p_ts[0] & 0x0E) << 29) |
 355                               (((mtime_t)U32_AT(p_ts) & 0xFFFE00) << 6) |
 356                               ((mtime_t)p_ts[3] << 7) |
 357                               ((mtime_t)p_ts[4] >> 1) ) );
 358                     }
 359                 }
 360             }
 361
 362             break;
 363         }
 364
 365         /* Welcome to the kludge area ! --Meuuh */
 366         if ( p_es->i_fourcc == VLC_FOURCC('a','5','2','b') )
 367         {
 368             /* With A/52 audio, we need to skip the first 4 bytes */
 369             i_pes_header_size += 4;
 370         }
 371
 372         if ( p_es->i_fourcc == VLC_FOURCC('l','p','c','b')
 373               || p_es->i_fourcc == VLC_FOURCC('s','p','u','b')
 374               || p_es->i_fourcc == VLC_FOURCC('d','t','s','b')
 375               || p_es->i_fourcc == VLC_FOURCC('s','d','d','b') )
 376         {
 377             /* stream_private_id */
 378             i_pes_header_size += 1;
 379         }
 380
 381         /* Now we've parsed the header, we just have to indicate in some
 382          * specific data packets where the PES payload begins (renumber
 383          * p_payload_start), so that the decoders can find the beginning
 384          * of their data right out of the box. */
 385         p_data = p_pes->p_first;
 386         i_payload_size = p_data->p_payload_end
 387                                  - p_data->p_payload_start;
 388         while( i_pes_header_size > i_payload_size )
 389         {
 390             /* These packets are entirely filled by the PES header. */
 391             i_pes_header_size -= i_payload_size;
 392             p_data->p_payload_start = p_data->p_payload_end;
 393             /* Go to the next data packet. */
 394             if( (p_data = p_data->p_next) == NULL )
 395             {
 396                 msg_Err( p_input, "PES header bigger than payload" );
 397                 input_DeletePES( p_input->p_method_data, p_pes );
 398                 p_pes = NULL;
 399                 return;
 400             }
 401             i_payload_size = p_data->p_payload_end
 402                                  - p_data->p_payload_start;
 403         }
 404         /* This last packet is partly header, partly payload. */
 405         if( i_payload_size < i_pes_header_size )
 406         {
 407             msg_Err( p_input, "PES header bigger than payload" );
 408             input_DeletePES( p_input->p_method_data, p_pes );
 409             p_pes = NULL;
 410             return;
 411         }
 412         p_data->p_payload_start += i_pes_header_size;
 413
 414
 415         /* Now we can eventually put the PES packet in the decoder's
 416          * PES fifo */
 417         if( p_es->p_decoder_fifo != NULL )
 418         {
 419             input_DecodePES( p_es->p_decoder_fifo, p_pes );
 420         }
 421         else if ( p_es->p_decoder_fifo == NULL &&
 422                    ((es_ts_data_t*)p_es->p_demux_data)->b_dvbsub)
 423         {
 424             es_descriptor_t* p_dvbsub;
 425             uint8_t          i_count;
 426             uint8_t         i;
 427             i_count = ((es_ts_data_t*)p_es->p_demux_data)->i_dvbsub_es_count;
 428
 429             // If a language is selected, we send the packet to the decoder
 430             for(i = 0; i < i_count; i++)
 431             {
 432                 p_dvbsub = ((es_ts_data_t*)p_es->p_demux_data)->p_dvbsub_es[i];
 433                 if(p_dvbsub->p_decoder_fifo!=NULL)
 434                 {
 435                     input_DecodePES ( p_dvbsub->p_decoder_fifo, p_pes );
 436                     break;
 437                 }
 438             }
 439             if(i == i_count)
 440             {
 441                 input_DeletePES( p_input->p_method_data, p_pes );
 442             }
 443         }
 444         else
 445         {
 446             msg_Err( p_input, "no fifo to receive PES %p "
 447                               "(who wrote this damn code ?)", p_pes );
 448             input_DeletePES( p_input->p_method_data, p_pes );
 449         }
 450         p_pes = NULL;
 451     }
 452 #undef p_pes
 453
 454 }
 455
 456 /*****************************************************************************
 457  * GatherPES:
 458  *****************************************************************************
 459  * Gather a PES packet.
 460  *****************************************************************************/
 461 static void GatherPES( input_thread_t * p_input, data_packet_t * p_data,
 462                        es_descriptor_t * p_es,
 463                        vlc_bool_t b_unit_start, vlc_bool_t b_packet_lost )
 464 {
 465 #define p_pes (p_es->p_pes)
 466
 467     /* If we lost data, insert a NULL data packet (philosophy : 0 is quite
 468      * often an escape sequence in decoders, so that should make them wait
 469      * for the next start code). */
 470     if( b_packet_lost  && !((es_ts_data_t*)p_es->p_demux_data)->b_dvbsub)
 471     {
 472         input_NullPacket( p_input, p_es );
 473     }
 474
 475     if( b_unit_start && p_pes != NULL )
 476     {
 477         /* If the data packet contains the begining of a new PES packet, and
 478          * if we were reassembling a PES packet, then the PES should be
 479          * complete now, so parse its header and give it to the decoders. */
 480         ParsePES( p_input, p_es );
 481     }
 482
 483     if( !b_unit_start && p_pes == NULL )
 484     {
 485         /* Random access... */
 486         input_DeletePacket( p_input->p_method_data, p_data );
 487     }
 488     else
 489     {
 490         if( b_unit_start )
 491         {
 492             /* If we are at the beginning of a new PES packet, we must fetch
 493              * a new PES buffer to begin with the reassembly of this PES
 494              * packet. This is also here that we can synchronize with the
 495              * stream if we lost packets or if the decoder has just
 496              * started. */
 497             if( (p_pes = input_NewPES( p_input->p_method_data ) ) == NULL )
 498             {
 499                 msg_Err( p_input, "out of memory" );
 500                 p_input->b_error = 1;
 501                 return;
 502             }
 503             p_pes->i_rate = p_input->stream.control.i_rate;
 504             p_pes->p_first = p_data;
 505
 506             /* If the PES header fits in the first data packet, we can
 507              * already set p_gather->i_pes_real_size. */
 508             if( p_data->p_payload_end - p_data->p_payload_start
 509                     >= PES_HEADER_SIZE )
 510             {
 511                 p_es->i_pes_real_size = ((uint16_t)p_data->p_payload_start[4] << 8)
 512                                          + p_data->p_payload_start[5];
 513                 if ( p_es->i_pes_real_size )
 514                 {
 515                     p_es->i_pes_real_size += 6;
 516                 }
 517             }
 518             else
 519             {
 520                 p_es->i_pes_real_size = 0;
 521             }
 522         }
 523         else
 524         {
 525             /* Update the relations between the data packets */
 526             p_pes->p_last->p_next = p_data;
 527         }
 528
 529         p_pes->p_last = p_data;
 530         p_pes->i_nb_data++;
 531
 532         /* Size of the payload carried in the data packet */
 533         p_pes->i_pes_size += (p_data->p_payload_end
 534                                  - p_data->p_payload_start);
 535
 536         /* We can check if the packet is finished */
 537         if( p_es->i_pes_real_size  && p_pes->i_pes_size >= p_es->i_pes_real_size )
 538         {
 539             if( p_pes->i_pes_size > p_es->i_pes_real_size )
 540             {
 541                 msg_Warn( p_input,
 542                           "Oversized PES packet for PID %d: expected %d, actual %d",
 543                           p_es->i_id, p_es->i_pes_real_size, p_pes->i_pes_size );
 544             }
 545             /* The packet is finished, parse it */
 546             ParsePES( p_input, p_es );
 547         }
 548     }
 549 #undef p_pes
 550 }
 551
 552
 553 /*
 554  * PS Demultiplexing
 555  */
 556
 557 /*****************************************************************************
 558  * GetID: Get the ID of a stream
 559  *****************************************************************************/
 560 static uint16_t GetID( data_packet_t * p_data )
 561 {
 562     uint16_t i_id;
 563
 564     i_id = p_data->p_demux_start[3];                            /* stream_id */
 565     if( i_id == 0xBD )
 566     {
 567         /* FIXME : this is not valid if the header is split in multiple
 568          * packets */
 569         /* stream_private_id */
 570         i_id |= p_data->p_demux_start[ 9 + p_data->p_demux_start[8] ] << 8;
 571     }
 572     return( i_id );
 573 }
 574
 575 /*****************************************************************************
 576  * DecodePSM: Decode the Program Stream Map information
 577  *****************************************************************************
 578  * FIXME : loads are not aligned in this function
 579  *****************************************************************************/
 580 static void DecodePSM( input_thread_t * p_input, data_packet_t * p_data )
 581 {
 582     stream_ps_data_t *  p_demux =
 583                  (stream_ps_data_t *)p_input->stream.p_demux_data;
 584     byte_t *            p_byte;
 585     byte_t *            p_end;
 586     unsigned int        i;
 587     unsigned int        i_new_es_number = 0;
 588
 589     if( p_data->p_demux_start + 10 > p_data->p_payload_end )
 590     {
 591         msg_Err( p_input, "PSM too short, packet corrupt" );
 592         return;
 593     }
 594
 595     if( p_demux->b_has_PSM
 596         && p_demux->i_PSM_version == (p_data->p_demux_start[6] & 0x1F) )
 597     {
 598         /* Already got that one. */
 599         return;
 600     }
 601
 602     p_demux->b_has_PSM = 1;
 603     p_demux->i_PSM_version = p_data->p_demux_start[6] & 0x1F;
 604
 605     /* Go to elementary_stream_map_length, jumping over
 606      * program_stream_info. */
 607     p_byte = p_data->p_demux_start + 10
 608               + U16_AT(&p_data->p_demux_start[8]);
 609     if( p_byte > p_data->p_payload_end )
 610     {
 611         msg_Err( p_input, "PSM too short, packet corrupt" );
 612         return;
 613     }
 614     /* This is the full size of the elementary_stream_map.
 615      * 2 == elementary_stream_map_length
 616      * Please note that CRC_32 is not included in the length. */
 617     p_end = p_byte + 2 + U16_AT(p_byte);
 618     p_byte += 2;
 619     if( p_end > p_data->p_payload_end )
 620     {
 621         msg_Err( p_input, "PSM too short, packet corrupt" );
 622         return;
 623     }
 624
 625     vlc_mutex_lock( &p_input->stream.stream_lock );
 626
 627     /* 4 == minimum useful size of a section */
 628     while( p_byte + 4 <= p_end )
 629     {
 630         es_descriptor_t *   p_es = NULL;
 631         uint8_t             i_stream_id = p_byte[1];
 632         /* FIXME: there will be a problem with private streams... (same
 633          * stream_id) */
 634
 635         /* Look for the ES in the ES table */
 636         for( i = i_new_es_number;
 637              i < p_input->stream.pp_programs[0]->i_es_number;
 638              i++ )
 639         {
 640             if( p_input->stream.pp_programs[0]->pp_es[i]->i_stream_id
 641                     == i_stream_id )
 642             {
 643                 p_es = p_input->stream.pp_programs[0]->pp_es[i];
 644                 /* FIXME: do something below */
 645 #if 0
 646                 if( p_es->i_type != p_byte[0] )
 647                 {
 648                     input_DelES( p_input, p_es );
 649                     p_es = NULL;
 650                 }
 651                 else
 652 #endif
 653                 {
 654                     /* Move the ES to the beginning. */
 655                     p_input->stream.pp_programs[0]->pp_es[i]
 656                         = p_input->stream.pp_programs[0]->pp_es[ i_new_es_number ];
 657                     p_input->stream.pp_programs[0]->pp_es[ i_new_es_number ]
 658                         = p_es;
 659                     i_new_es_number++;
 660                 }
 661                 break;
 662             }
 663         }
 664
 665         /* The goal is to have all the ES we have just read in the
 666          * beginning of the pp_es table, and all the others at the end,
 667          * so that we can close them more easily at the end. */
 668         if( p_es == NULL )
 669         {
 670             int i_fourcc, i_cat;
 671
 672             switch( p_byte[0] )
 673             {
 674             case MPEG1_VIDEO_ES:
 675             case MPEG2_VIDEO_ES:
 676             case MPEG2_MOTO_VIDEO_ES:
 677                 i_fourcc = VLC_FOURCC('m','p','g','v');
 678                 i_cat = VIDEO_ES;
 679                 break;
 680             case DVD_SPU_ES:
 681                 i_fourcc = VLC_FOURCC('s','p','u','b');
 682                 i_cat = SPU_ES;
 683                 break;
 684             case MPEG1_AUDIO_ES:
 685             case MPEG2_AUDIO_ES:
 686                 i_fourcc = VLC_FOURCC('m','p','g','a');
 687                 i_cat = AUDIO_ES;
 688                 break;
 689             case A52_AUDIO_ES:
 690                 i_fourcc = VLC_FOURCC('a','5','2','b');
 691                 i_cat = AUDIO_ES;
 692                 break;
 693             case LPCM_AUDIO_ES:
 694                 i_fourcc = VLC_FOURCC('l','p','c','b');
 695                 i_cat = AUDIO_ES;
 696                 break;
 697             default:
 698                 i_cat = UNKNOWN_ES;
 699                 i_fourcc = 0;
 700                 break;
 701             }
 702
 703             p_es = input_AddES( p_input, p_input->stream.pp_programs[0],
 704                                 i_stream_id, i_cat, NULL, 0 );
 705             p_es->i_fourcc = i_fourcc;
 706
 707             /* input_AddES has inserted the new element at the end. */
 708             p_input->stream.pp_programs[0]->pp_es[
 709                 p_input->stream.pp_programs[0]->i_es_number ]
 710                 = p_input->stream.pp_programs[0]->pp_es[ i_new_es_number ];
 711             p_input->stream.pp_programs[0]->pp_es[ i_new_es_number ] = p_es;
 712             i_new_es_number++;
 713         }
 714         p_byte += 4 + U16_AT(&p_byte[2]);
 715     }
 716
 717     /* Un-select the streams that are no longer parts of the program. */
 718     while( i_new_es_number < p_input->stream.pp_programs[0]->i_es_number )
 719     {
 720         /* We remove pp_es[i_new_es_member] and not pp_es[i] because the
 721          * list will be emptied starting from the end */
 722         input_DelES( p_input,
 723                      p_input->stream.pp_programs[0]->pp_es[i_new_es_number] );
 724     }
 725
 726     msg_Dbg( p_input, "the stream map after the PSM is now:" );
 727     input_DumpStream( p_input );
 728
 729     vlc_mutex_unlock( &p_input->stream.stream_lock );
 730 }
 731
 732 /*****************************************************************************
 733  * ReadPS: store a PS packet into a data_buffer_t
 734  *****************************************************************************/
 735 #define PEEK( SIZE )                                                        \
 736     i_error = input_Peek( p_input, &p_peek, SIZE );                         \
 737     if( i_error == -1 )                                                     \
 738     {                                                                       \
 739         return( -1 );                                                       \
 740     }                                                                       \
 741     else if( (size_t)i_error < SIZE )                                       \
 742     {                                                                       \
 743         /* EOF */                                                           \
 744         return( 0 );                                                        \
 745     }
 746
 747 static ssize_t ReadPS( input_thread_t * p_input, data_packet_t ** pp_data )
 748 {
 749     byte_t *            p_peek;
 750     size_t              i_packet_size;
 751     ssize_t             i_error, i_read;
 752
 753     /* Read what we believe to be a packet header. */
 754     PEEK( 4 );
 755
 756     if( p_peek[0] || p_peek[1] || p_peek[2] != 1 || p_peek[3] < 0xB9 )
 757     {
 758         if( p_peek[0] || p_peek[1] || p_peek[2] )
 759         {
 760             /* It is common for MPEG-1 streams to pad with zeros
 761              * (although it is forbidden by the recommendation), so
 762              * don't bother everybody in this case. */
 763             msg_Warn( p_input, "garbage (0x%.2x%.2x%.2x%.2x)",
 764                       p_peek[0], p_peek[1], p_peek[2], p_peek[3] );
 765         }
 766
 767         /* This is not the startcode of a packet. Read the stream
 768          * until we find one. */
 769         while( p_peek[0] || p_peek[1] || p_peek[2] != 1 || p_peek[3] < 0xB9 )
 770         {
 771             p_input->p_current_data++;
 772             PEEK( 4 );
 773             if( p_input->b_die ) return( -1 );
 774         }
 775         /* Packet found. */
 776     }
 777
 778     /* 0x1B9 == SYSTEM_END_CODE, it is only 4 bytes long. */
 779     if( p_peek[3] != 0xB9 )
 780     {
 781         /* The packet is at least 6 bytes long. */
 782         PEEK( 6 );
 783
 784         if( p_peek[3] != 0xBA )
 785         {
 786             /* That's the case for all packets, except pack header. */
 787             i_packet_size = (p_peek[4] << 8) | p_peek[5];
 788         }
 789         else
 790         {
 791             /* Pack header. */
 792             if( (p_peek[4] & 0xC0) == 0x40 )
 793             {
 794                 /* MPEG-2 */
 795                 i_packet_size = 8;
 796             }
 797             else if( (p_peek[4] & 0xF0) == 0x20 )
 798             {
 799                 /* MPEG-1 */
 800                 i_packet_size = 6;
 801             }
 802             else
 803             {
 804                 msg_Err( p_input, "unable to determine stream type" );
 805                 p_input->p_current_data++;
 806                 return( -1 );
 807             }
 808         }
 809     }
 810     else
 811     {
 812         /* System End Code */
 813         i_packet_size = -2;
 814     }
 815
 816     /* Fetch a packet of the appropriate size. */
 817     i_read = input_SplitBuffer( p_input, pp_data, i_packet_size + 6 );
 818     if( i_read <= 0 )
 819     {
 820         return( i_read );
 821     }
 822
 823     /* In MPEG-2 pack headers we still have to read stuffing bytes. */
 824     if( ((*pp_data)->p_demux_start[3] == 0xBA) && (i_packet_size == 8) )
 825     {
 826         size_t i_stuffing = ((*pp_data)->p_demux_start[13] & 0x7);
 827         /* Force refill of the input buffer - though we don't care
 828          * about p_peek. Please note that this is unoptimized. */
 829         PEEK( i_stuffing );
 830         p_input->p_current_data += i_stuffing;
 831     }
 832
 833     return( 1 );
 834 }
 835
 836 #undef PEEK
 837
 838 /*****************************************************************************
 839  * ParsePS: read the PS header
 840  *****************************************************************************/
 841 static es_descriptor_t * ParsePS( input_thread_t * p_input,
 842                                   data_packet_t * p_data )
 843 {
 844     uint32_t            i_code;
 845     es_descriptor_t *   p_es = NULL;
 846
 847     i_code = p_data->p_demux_start[3];
 848
 849     if( i_code > 0xBC ) /* ES start code */
 850     {
 851         uint16_t            i_id;
 852         unsigned int        i_dummy;
 853
 854         /* This is a PES packet. Find out if we want it or not. */
 855         i_id = GetID( p_data );
 856
 857         vlc_mutex_lock( &p_input->stream.stream_lock );
 858         if( p_input->stream.pp_programs[0]->b_is_ok )
 859         {
 860             /* Look only at the selected ES. */
 861             for( i_dummy = 0; i_dummy < p_input->stream.i_selected_es_number;
 862                  i_dummy++ )
 863             {
 864                 if( p_input->stream.pp_selected_es[i_dummy] != NULL
 865                     && p_input->stream.pp_selected_es[i_dummy]->i_id == i_id )
 866                 {
 867                     p_es = p_input->stream.pp_selected_es[i_dummy];
 868                     break;
 869                 }
 870             }
 871         }
 872         else
 873         {
 874             vlc_bool_t b_auto_spawn = VLC_FALSE;
 875             stream_ps_data_t * p_demux =
 876               (stream_ps_data_t *)p_input->stream.pp_programs[0]->p_demux_data;
 877
 878             /* Search all ES ; if not found -> AddES */
 879             p_es = input_FindES( p_input, i_id );
 880
 881             if( p_es == NULL && !p_demux->b_has_PSM )
 882             {
 883                 int i_fourcc, i_cat;
 884
 885                 /* Set stream type and auto-spawn. */
 886                 if( (i_id & 0xF0) == 0xE0 )
 887                 {
 888                     /* MPEG video */
 889                     i_fourcc = VLC_FOURCC('m','p','g','v');
 890                     i_cat = VIDEO_ES;
 891 #ifdef AUTO_SPAWN
 892                     if( !p_input->stream.b_seekable ) b_auto_spawn = VLC_TRUE;
 893 #endif
 894                 }
 895                 else if( (i_id & 0xE0) == 0xC0 )
 896                 {
 897                     /* MPEG audio */
 898                     i_fourcc = VLC_FOURCC('m','p','g','a');
 899                     i_cat = AUDIO_ES;
 900 #ifdef AUTO_SPAWN
 901                     if( !p_input->stream.b_seekable )
 902                     if( config_GetInt( p_input, "audio-channel" )
 903                             == (i_id & 0x1F) ||
 904                         ( config_GetInt( p_input, "audio-channel" ) < 0
 905                           && !(i_id & 0x1F) ) )
 906                     switch( config_GetInt( p_input, "audio-type" ) )
 907                     {
 908                     case -1:
 909                     case REQUESTED_MPEG:
 910                         b_auto_spawn = VLC_TRUE;
 911                     }
 912 #endif
 913                 }
 914                 else if( (i_id & 0xF8FF) == 0x88BD )
 915                 {
 916                     i_fourcc = VLC_FOURCC('d','t','s','b');
 917                     i_cat = AUDIO_ES;
 918 #ifdef AUTO_SPAWN
 919                     if( !p_input->stream.b_seekable )
 920                     if( config_GetInt( p_input, "audio-channel" )
 921                             == ((i_id & 0x700) >> 8) ||
 922                         ( config_GetInt( p_input, "audio-channel" ) < 0
 923                           && !((i_id & 0x700) >> 8)) )
 924                     switch( config_GetInt( p_input, "audio-type" ) )
 925                     {
 926                     case -1:
 927                     case REQUESTED_DTS:
 928                         b_auto_spawn = VLC_TRUE;
 929                     }
 930 #endif
 931                 }
 932                 else if( (i_id & 0xF0FF) == 0x80BD )
 933                 {
 934                     /* A52 audio (0x80->0x8F) */
 935                     i_fourcc = VLC_FOURCC('a','5','2','b');
 936                     i_cat = AUDIO_ES;
 937 #ifdef AUTO_SPAWN
 938                     if( !p_input->stream.b_seekable )
 939                     if( config_GetInt( p_input, "audio-channel" )
 940                             == ((i_id & 0xF00) >> 8) ||
 941                         ( config_GetInt( p_input, "audio-channel" ) < 0
 942                           && !((i_id & 0xF00) >> 8)) )
 943                     switch( config_GetInt( p_input, "audio-type" ) )
 944                     {
 945                     case -1:
 946                     case REQUESTED_A52:
 947                         b_auto_spawn = VLC_TRUE;
 948                     }
 949 #endif
 950                 }
 951                 else if( (i_id & 0xE0FF) == 0x20BD )
 952                 {
 953                     /* Subtitles video (0x20->0x3F) */
 954                     i_fourcc = VLC_FOURCC('s','p','u','b');
 955                     i_cat = SPU_ES;
 956 #ifdef AUTO_SPAWN
 957                     if( !p_input->stream.b_seekable )
 958                     if( config_GetInt( p_input, "spu-channel" )
 959                            == ((i_id & 0x1F00) >> 8) )
 960                     {
 961                         b_auto_spawn = VLC_TRUE;
 962                     }
 963 #endif
 964                 }
 965                 else if( (i_id & 0xF0FF) == 0xA0BD )
 966                 {
 967                     /* LPCM audio (0xA0->0xAF) */
 968                     i_fourcc = VLC_FOURCC('l','p','c','b');
 969                     i_cat = AUDIO_ES;
 970                 }
 971                 else
 972                 {
 973                     i_cat = UNKNOWN_ES;
 974                     i_fourcc = 0;
 975                 }
 976
 977                 p_es = input_AddES( p_input, p_input->stream.pp_programs[0],
 978                                     i_id, i_cat, NULL, 0 );
 979
 980                 p_es->i_stream_id = p_data->p_demux_start[3];
 981                 p_es->i_fourcc = i_fourcc;
 982
 983                 if( b_auto_spawn ) input_SelectES( p_input, p_es );
 984
 985                 /* Tell the interface the stream has changed */
 986                 p_input->stream.b_changed = 1;
 987             }
 988         } /* stream.b_is_ok */
 989         vlc_mutex_unlock( &p_input->stream.stream_lock );
 990     } /* i_code > 0xBC */
 991
 992     return( p_es );
 993 }
 994
 995 /*****************************************************************************
 996  * DemuxPS: first step of demultiplexing: the PS header
 997  *****************************************************************************/
 998 static void DemuxPS( input_thread_t * p_input, data_packet_t * p_data )
 999 {
1000     uint32_t i_code;
1001     vlc_bool_t b_trash = 0;
1002     es_descriptor_t * p_es = NULL;
1003
1004     i_code = ((uint32_t)p_data->p_demux_start[0] << 24)
1005                 | ((uint32_t)p_data->p_demux_start[1] << 16)
1006                 | ((uint32_t)p_data->p_demux_start[2] << 8)
1007                 | p_data->p_demux_start[3];
1008     if( i_code <= 0x1BC )
1009     {
1010         switch( i_code )
1011         {
1012         case 0x1BA: /* PACK_START_CODE */
1013             {
1014                 /* Read the SCR. */
1015                 mtime_t scr_time;
1016                 uint32_t i_mux_rate;
1017
1018                 if( (p_data->p_demux_start[4] & 0xC0) == 0x40 )
1019                 {
1020                     /* MPEG-2 */
1021                     byte_t      p_header[14];
1022                     byte_t *    p_byte;
1023                     p_byte = p_data->p_demux_start;
1024
1025                     if( MoveChunk( p_header, &p_data, &p_byte, 14 ) != 14 )
1026                     {
1027                         msg_Warn( p_input,
1028                                   "packet too short to have a header" );
1029                         b_trash = 1;
1030                         break;
1031                     }
1032                     scr_time =
1033                          ((mtime_t)(p_header[4] & 0x38) << 27) |
1034                          ((mtime_t)(U32_AT(p_header + 4) & 0x03FFF800)
1035                                         << 4) |
1036                          ((( ((mtime_t)U16_AT(p_header + 6) << 16)
1037                             | (mtime_t)U16_AT(p_header + 8) ) & 0x03FFF800)
1038                                         >> 11);
1039
1040                     /* mux_rate */
1041                     i_mux_rate = ((uint32_t)U16_AT(p_header + 10) << 6)
1042                                    | (p_header[12] >> 2);
1043                     /* FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME
1044                      * This is the biggest kludge ever !
1045                      * I don't know what's wrong with mux_rate calculation
1046                      * but this heuristic works well : */
1047                     i_mux_rate <<= 1;
1048                     i_mux_rate /= 3;
1049                 }
1050                 else
1051                 {
1052                     /* MPEG-1 SCR is like PTS. */
1053                     byte_t      p_header[12];
1054                     byte_t *    p_byte;
1055                     p_byte = p_data->p_demux_start;
1056
1057                     if( MoveChunk( p_header, &p_data, &p_byte, 12 ) != 12 )
1058                     {
1059                         msg_Warn( p_input,
1060                                   "packet too short to have a header" );
1061                         b_trash = 1;
1062                         break;
1063                     }
1064                     scr_time =
1065                          ((mtime_t)(p_header[4] & 0x0E) << 29) |
1066                          (((mtime_t)U32_AT(p_header + 4) & 0xFFFE00) << 6) |
1067                          ((mtime_t)p_header[7] << 7) |
1068                          ((mtime_t)p_header[8] >> 1);
1069
1070                     /* mux_rate */
1071                     i_mux_rate = (U32_AT(p_header + 8) & 0x7FFFFE) >> 1;
1072                 }
1073                 /* Call the pace control. */
1074                 input_ClockManageRef( p_input,
1075                                       p_input->stream.p_selected_program,
1076                                       scr_time );
1077
1078                 if( i_mux_rate != p_input->stream.i_mux_rate
1079                      && p_input->stream.i_mux_rate )
1080                 {
1081                     msg_Warn( p_input,
1082                               "mux_rate changed, expect cosmetic errors" );
1083                 }
1084                 p_input->stream.i_mux_rate = i_mux_rate;
1085
1086                 b_trash = 1;
1087             }
1088             break;
1089
1090         case 0x1BB: /* SYSTEM_START_CODE */
1091             b_trash = 1;                              /* Nothing interesting */
1092             break;
1093
1094         case 0x1BC: /* PROGRAM_STREAM_MAP_CODE */
1095             DecodePSM( p_input, p_data );
1096             b_trash = 1;
1097             break;
1098
1099         case 0x1B9: /* PROGRAM_END_CODE */
1100             b_trash = 1;
1101             break;
1102
1103         default:
1104             /* This should not happen */
1105             b_trash = 1;
1106             msg_Warn( p_input, "unwanted packet received "
1107                                "with startcode 0x%.8x", i_code );
1108         }
1109     }
1110     else
1111     {
1112         p_es = ParsePS( p_input, p_data );
1113
1114         vlc_mutex_lock( &p_input->stream.control.control_lock );
1115         if( p_es != NULL && p_es->p_decoder_fifo != NULL
1116              && (p_es->i_cat != AUDIO_ES || !p_input->stream.control.b_mute) )
1117         {
1118             vlc_mutex_unlock( &p_input->stream.control.control_lock );
1119             p_es->c_packets++;
1120             GatherPES( p_input, p_data, p_es, 1, 0 );
1121         }
1122         else
1123         {
1124             vlc_mutex_unlock( &p_input->stream.control.control_lock );
1125             b_trash = 1;
1126         }
1127     }
1128
1129     /* Trash the packet if it has no payload or if it isn't selected */
1130     if( b_trash )
1131     {
1132         input_DeletePacket( p_input->p_method_data, p_data );
1133         p_input->stream.c_packets_trashed++;
1134     }
1135 }
1136
1137
1138 /*
1139  * TS Demultiplexing
1140  */
1141
1142 /*****************************************************************************
1143  * ReadTS: store a TS packet into a data_buffer_t
1144  *****************************************************************************/
1145 #define PEEK( SIZE )                                                        \
1146     i_error = input_Peek( p_input, &p_peek, SIZE );                         \
1147     if( i_error == -1 )                                                     \
1148     {                                                                       \
1149         return( -1 );                                                       \
1150     }                                                                       \
1151     else if( i_error < SIZE )                                               \
1152     {                                                                       \
1153         /* EOF */                                                           \
1154         return( 0 );                                                        \
1155     }
1156
1157 static ssize_t ReadTS( input_thread_t * p_input, data_packet_t ** pp_data )
1158 {
1159     byte_t *            p_peek;
1160     ssize_t             i_error, i_read;
1161
1162     PEEK( 1 );
1163
1164     if( *p_peek != TS_SYNC_CODE )
1165     {
1166         msg_Warn( p_input, "garbage at input (%x)", *p_peek );
1167
1168         if( p_input->i_mtu )
1169         {
1170             while( *p_peek != TS_SYNC_CODE )
1171             {
1172                 /* Try to resync on next packet. */
1173                 PEEK( TS_PACKET_SIZE );
1174                 p_input->p_current_data += TS_PACKET_SIZE;
1175                 PEEK( 1 );
1176             }
1177         }
1178         else
1179         {
1180             /* Move forward until we find 0x47 (and hope it's the good
1181              * one... FIXME) */
1182             while( *p_peek != TS_SYNC_CODE )
1183             {
1184                 p_input->p_current_data++;
1185                 PEEK( 1 );
1186             }
1187         }
1188     }
1189
1190     i_read = input_SplitBuffer( p_input, pp_data, TS_PACKET_SIZE );
1191     if( i_read <= 0 )
1192     {
1193         return( i_read );
1194     }
1195
1196     return( 1 );
1197 }
1198
1199 /*****************************************************************************
1200  * DemuxTS: first step of demultiplexing: the TS header
1201  *****************************************************************************/
1202 static void DemuxTS( input_thread_t * p_input, data_packet_t * p_data,
1203                      psi_callback_t pf_psi_callback )
1204 {
1205     uint16_t            i_pid;
1206     unsigned int        i_dummy;
1207     vlc_bool_t          b_adaptation;         /* Adaptation field is present */
1208     vlc_bool_t          b_payload;                 /* Packet carries payload */
1209     vlc_bool_t          b_unit_start;  /* A PSI or a PES start in the packet */
1210     vlc_bool_t          b_trash = 0;             /* Is the packet unuseful ? */
1211     vlc_bool_t          b_lost = 0;             /* Was there a packet loss ? */
1212     vlc_bool_t          b_psi = 0;                        /* Is this a PSI ? */
1213     vlc_bool_t          b_dvbsub = 0;            /* Is this a dvb subtitle ? */
1214     vlc_bool_t          b_pcr = 0;                   /* Does it have a PCR ? */
1215     es_descriptor_t *   p_es = NULL;
1216     es_ts_data_t *      p_es_demux = NULL;
1217     pgrm_ts_data_t *    p_pgrm_demux = NULL;
1218     stream_ts_data_t *  p_stream_demux =
1219                           (stream_ts_data_t *)p_input->stream.p_demux_data;
1220
1221 #define p (p_data->p_demux_start)
1222     /* Extract flags values from TS common header. */
1223     i_pid = ((p[1] & 0x1F) << 8) | p[2];
1224     b_unit_start = (p[1] & 0x40);
1225     b_adaptation = (p[3] & 0x20);
1226     b_payload = (p[3] & 0x10);
1227
1228     /* Was there a transport error ? */
1229     if ( p[1] & 0x80 )
1230     {
1231         msg_Warn( p_input, "transport_error_indicator set for PID %d counter %x", i_pid, p[3] & 0x0f );
1232     }
1233
1234     /* Find out the elementary stream. */
1235     vlc_mutex_lock( &p_input->stream.stream_lock );
1236
1237     for( i_dummy = 0; i_dummy < p_input->stream.i_pgrm_number; i_dummy ++ )
1238     {
1239         if( (( pgrm_ts_data_t * ) p_input->stream.pp_programs[i_dummy]->
1240                     p_demux_data)->i_pcr_pid == i_pid )
1241         {
1242             b_pcr = 1;
1243             break;
1244         }
1245     }
1246
1247     p_es = input_FindES( p_input, i_pid );
1248
1249     if( (p_es != NULL) && (p_es->p_demux_data != NULL) )
1250     {
1251         p_es_demux = (es_ts_data_t *)p_es->p_demux_data;
1252
1253         if( p_es_demux->b_psi )
1254         {
1255             b_psi = 1;
1256         }
1257         else if ( p_es_demux->b_dvbsub )
1258         {
1259             b_dvbsub = 1;
1260         }
1261         else
1262         {
1263             p_pgrm_demux = (pgrm_ts_data_t *)p_es->p_pgrm->p_demux_data;
1264         }
1265     }
1266
1267     vlc_mutex_lock( &p_input->stream.control.control_lock );
1268     if( ( p_es == NULL ) || (p_es->i_cat == AUDIO_ES
1269                               && p_input->stream.control.b_mute) )
1270     {
1271         /* Not selected. Just read the adaptation field for a PCR. */
1272         b_trash = 1;
1273     }
1274     else if( p_es->p_decoder_fifo == NULL && !b_psi && !b_dvbsub )
1275     {
1276         b_trash = 1;
1277     }
1278
1279     vlc_mutex_unlock( &p_input->stream.control.control_lock );
1280     vlc_mutex_unlock( &p_input->stream.stream_lock );
1281
1282
1283     /* Don't change the order of the tests : if b_psi then p_pgrm_demux
1284      * may still be null. Who said it was ugly ?
1285      * I have written worse. --Meuuh */
1286     if( ( p_es ) &&
1287         ((p_es->p_decoder_fifo != NULL) || b_psi || b_pcr || b_dvbsub) )
1288     {
1289         p_es->c_packets++;
1290
1291         /* Extract adaptation field information if any */
1292
1293         if( !b_adaptation )
1294         {
1295             /* We don't have any adaptation_field, so payload starts
1296              * immediately after the 4 byte TS header */
1297             p_data->p_payload_start += 4;
1298         }
1299         else
1300         {
1301             /* p[4] is adaptation_field_length minus one */
1302             p_data->p_payload_start += 5 + p[4];
1303
1304             /* The adaptation field can be limited to the
1305              * adaptation_field_length byte, so that there is nothing to do:
1306              * skip this possibility */
1307             if( p[4] )
1308             {
1309                 /* If the packet has both adaptation_field and payload,
1310                  * adaptation_field cannot be more than 182 bytes long; if
1311                  * there is only an adaptation_field, it must fill the next
1312                  * 183 bytes. */
1313                 if( b_payload ? (p[4] > 182) : (p[4] != 183) )
1314                 {
1315                     msg_Warn( p_input, "invalid TS adaptation field for PID %d (%2x)",
1316                               i_pid, p[4] );
1317                     p_data->b_discard_payload = 1;
1318                     p_es->c_invalid_packets++;
1319
1320                     /* The length was invalid so we shouldn't have added it to
1321                      * p_payload_start above.  Ensure p_payload_start has a
1322                      * valid value by setting it equal to p_payload_end.  This
1323                      * also stops any data being processed from the packet.
1324                      */
1325                     p_data->p_payload_start = p_data->p_payload_end;
1326                 }
1327
1328                 /* Now we are sure that the byte containing flags is present:
1329                  * read it */
1330                 else
1331                 {
1332                     /* discontinuity_indicator */
1333                     if( p[5] & 0x80 )
1334                     {
1335                         msg_Warn( p_input,
1336                             "discontinuity_indicator encountered by TS demux "
1337                             "(PID %d: current %d, packet %d)",
1338                             i_pid,
1339                             ( p_es_demux->i_continuity_counter ) & 0x0f,
1340                             p[3] & 0x0f );
1341
1342                         /* If the PID carries the PCR, there will be a system
1343                          * time-based discontinuity. We let the PCR decoder
1344                          * handle that. */
1345                         p_es->p_pgrm->i_synchro_state = SYNCHRO_REINIT;
1346
1347                         /* Don't resynchronise the counter here - it will
1348                          * be checked later and b_lost will then be set if
1349                          * necessary.
1350                          */
1351                     }
1352
1353                 } /* valid TS adaptation field ? */
1354             } /* length > 0 */
1355         } /* has adaptation field */
1356         /* Check the continuity of the stream. */
1357         i_dummy = ((p[3] & 0x0f) - p_es_demux->i_continuity_counter) & 0x0f;
1358         if( b_payload &&
1359                 ( i_dummy == 1 || (b_psi && p_stream_demux->b_buggy_psi ) ) )
1360         {
1361             /* Everything is ok, just increase our counter */
1362             (p_es_demux->i_continuity_counter)++;
1363         }
1364         else
1365         {
1366             if( !b_payload && i_dummy == 0 )
1367             {
1368                 /* This is a packet without payload, this is allowed by the
1369                  * draft. As there is nothing interesting in this packet
1370                  * (except PCR that have already been handled), we can trash
1371                  * the packet. */
1372                 b_trash = 1;
1373             }
1374             else if( !b_payload )
1375             {
1376                 /* If there is no payload, the counter should be unchanged */
1377                 msg_Warn( p_input, "packet rxd for PID %d with no payload but "
1378                         "wrong counter: current %d, packet %d", i_pid,
1379                         p_es_demux->i_continuity_counter & 0x0f, p[3] & 0x0f );
1380             }
1381             else if( i_dummy <= 0 )
1382             {
1383                 /* Duplicate packet: mark it as being to be trashed. */
1384                 msg_Warn( p_input,
1385                           "duplicate packet received for PID %d (counter %d)",
1386                           p_es->i_id, p[3] & 0x0f );
1387                 b_trash = 1;
1388             }
1389             else if( p_es_demux->i_continuity_counter == 0xFF )
1390             {
1391                 /* This means that the packet is the first one we receive for
1392                  * this ES since the continuity counter ranges between 0 and
1393                  * 0x0F excepts when it has been initialized by the input:
1394                  * init the counter to the correct value. */
1395                 msg_Warn( p_input, "first packet for PID %d received "
1396                                    "by TS demux", p_es->i_id );
1397                 p_es_demux->i_continuity_counter = (p[3] & 0x0f);
1398             }
1399             else
1400             {
1401                 /* This can indicate that we missed a packet or that the
1402                  * continuity_counter wrapped and we received a dup packet:
1403                  * as we don't know, do as if we missed a packet to be sure
1404                  * to recover from this situation */
1405                 msg_Warn( p_input,
1406                           "packet lost by TS demux for PID %d: current %d, packet %d",
1407                           i_pid,
1408                           p_es_demux->i_continuity_counter & 0x0f,
1409                           p[3] & 0x0f );
1410                 b_lost = 1;
1411                 p_es_demux->i_continuity_counter = p[3] & 0x0f;
1412             } /* not continuous */
1413         } /* continuity */
1414     } /* if selected or PCR */
1415
1416     /* Handle PCR */
1417     if( b_pcr && b_adaptation && (p[5] & 0x10) && p[4]>=7 )
1418     {
1419         /* Read the PCR. */
1420         mtime_t     pcr_time;
1421         pcr_time = ( (mtime_t)p[6] << 25 ) |
1422                    ( (mtime_t)p[7] << 17 ) |
1423                    ( (mtime_t)p[8] << 9 ) |
1424                    ( (mtime_t)p[9] << 1 ) |
1425                    ( (mtime_t)p[10] >> 7 );
1426         /* Call the pace control. */
1427         for( i_dummy = 0; i_dummy < p_input->stream.i_pgrm_number;
1428                                 i_dummy ++ )
1429         {
1430             if( ( ( pgrm_ts_data_t * ) p_input->stream.pp_programs[i_dummy]->
1431                         p_demux_data )->i_pcr_pid == i_pid )
1432             {
1433                 input_ClockManageRef( p_input,
1434                     p_input->stream.pp_programs[i_dummy], pcr_time );
1435             }
1436         }
1437
1438     }
1439
1440     /* Trash the packet if it has no payload or if it isn't selected */
1441     if( b_trash )
1442     {
1443         input_DeletePacket( p_input->p_method_data, p_data );
1444         p_input->stream.c_packets_trashed++;
1445     }
1446     else
1447     {
1448         if( b_psi )
1449         {
1450             /* The payload contains PSI tables */
1451             (* pf_psi_callback) ( p_input, p_data, p_es, b_unit_start );
1452         }
1453         else
1454         {
1455             /* The payload carries a PES stream */
1456             GatherPES( p_input, p_data, p_es, b_unit_start, b_lost );
1457         }
1458
1459     }
1460
1461 #undef p
1462
1463 }