git.sesse.net Git - vlc/blob - src/input/mpeg_system.c

   1 /*****************************************************************************
   2  * mpeg_system.c: TS, PS and PES management
   3  *****************************************************************************
   4  * Copyright (C) 1998-2001 VideoLAN
   5  * $Id: mpeg_system.c,v 1.76 2001/12/19 10:00:00 massiot 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@via.ecp.fr>
  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 "defs.h"
  32
  33 #include <stdlib.h>
  34 #include <string.h>                                    /* memcpy(), memset() */
  35 #include <sys/types.h>                                              /* off_t */
  36
  37 #include "common.h"
  38 #include "intf_msg.h"
  39 #include "threads.h"
  40 #include "mtime.h"
  41
  42 #include "stream_control.h"
  43 #include "input_ext-intf.h"
  44 #include "input_ext-dec.h"
  45 #include "input_ext-plugins.h"
  46
  47 /*****************************************************************************
  48  * Local prototypes
  49  *****************************************************************************/
  50
  51 static void input_DecodePAT( input_thread_t *, es_descriptor_t *);
  52 static void input_DecodePMT( input_thread_t *, es_descriptor_t *);
  53
  54 /*
  55  * PES Packet management
  56  */
  57
  58 /*****************************************************************************
  59  * MoveChunk
  60  *****************************************************************************
  61  * Small utility function used to parse discontinuous headers safely. Copies
  62  * i_buf_len bytes of data to a buffer and returns the size copied.
  63  * It also solves some alignment problems on non-IA-32, non-PPC processors.
  64  * This is a variation on the theme of input_ext-dec.h:GetChunk().
  65  *****************************************************************************/
  66 static __inline__ size_t MoveChunk( byte_t * p_dest,
  67                                     data_packet_t ** pp_data_src,
  68                                     byte_t ** pp_src,
  69                                     size_t i_buf_len )
  70 {
  71     ptrdiff_t           i_available;
  72
  73     if( (i_available = (*pp_data_src)->p_payload_end - *pp_src)
  74             >= i_buf_len )
  75     {
  76         if( p_dest != NULL )
  77             memcpy( p_dest, *pp_src, i_buf_len );
  78         *pp_src += i_buf_len;
  79         return( i_buf_len );
  80     }
  81     else
  82     {
  83         size_t          i_init_len = i_buf_len;
  84
  85         do
  86         {
  87             if( p_dest != NULL )
  88                 memcpy( p_dest, *pp_src, i_available );
  89             *pp_data_src = (*pp_data_src)->p_next;
  90             i_buf_len -= i_available;
  91             p_dest += i_available;
  92             if( *pp_data_src == NULL )
  93             {
  94                 *pp_src = NULL;
  95                 return( i_init_len - i_buf_len );
  96             }
  97             *pp_src = (*pp_data_src)->p_payload_start;
  98         }
  99         while( (i_available = (*pp_data_src)->p_payload_end - *pp_src)
 100                 <= i_buf_len );
 101
 102         if( i_buf_len )
 103         {
 104             if( p_dest != NULL )
 105                 memcpy( p_dest, *pp_src, i_buf_len );
 106             *pp_src += i_buf_len;
 107         }
 108         return( i_init_len );
 109     }
 110 }
 111
 112 /*****************************************************************************
 113  * input_ParsePES
 114  *****************************************************************************
 115  * Parse a finished PES packet and analyze its header.
 116  *****************************************************************************/
 117 #define PES_HEADER_SIZE     7
 118 void input_ParsePES( input_thread_t * p_input, es_descriptor_t * p_es )
 119 {
 120     data_packet_t * p_data;
 121     byte_t *        p_byte;
 122     byte_t          p_header[PES_HEADER_SIZE];
 123     int             i_done;
 124
 125 #define p_pes (p_es->p_pes)
 126
 127     //intf_DbgMsg("End of PES packet %p", p_pes);
 128
 129     /* Parse the header. The header has a variable length, but in order
 130      * to improve the algorithm, we will read the 14 bytes we may be
 131      * interested in */
 132     p_data = p_pes->p_first;
 133     p_byte = p_data->p_payload_start;
 134     i_done = 0;
 135
 136     if( MoveChunk( p_header, &p_data, &p_byte, PES_HEADER_SIZE )
 137             != PES_HEADER_SIZE )
 138     {
 139         intf_WarnMsg( 1, "input: PES packet too short to have a header" );
 140         p_input->pf_delete_pes( p_input->p_method_data, p_pes );
 141         p_pes = NULL;
 142         return;
 143     }
 144
 145     /* Get the PES size if defined */
 146     p_es->i_pes_real_size = U16_AT(p_header + 4);
 147     if( p_es->i_pes_real_size )
 148     {
 149         p_es->i_pes_real_size += 6;
 150     }
 151
 152     /* First read the 6 header bytes common to all PES packets:
 153      * use them to test the PES validity */
 154     if( (p_header[0] || p_header[1] || (p_header[2] != 1)) )
 155     {
 156         /* packet_start_code_prefix != 0x000001 */
 157         intf_ErrMsg( "input error: data loss, "
 158                      "PES packet doesn't start with 0x000001" );
 159         p_input->pf_delete_pes( p_input->p_method_data, p_pes );
 160         p_pes = NULL;
 161     }
 162     else
 163     {
 164         int i_pes_header_size, i_payload_size;
 165
 166         if ( p_es->i_pes_real_size &&
 167              (p_es->i_pes_real_size != p_pes->i_pes_size) )
 168         {
 169             /* PES_packet_length is set and != total received payload */
 170             /* Warn the decoder that the data may be corrupt. */
 171             intf_WarnMsg( 1, "input: packet corrupted, "
 172                              "PES sizes do not match" );
 173         }
 174
 175         switch( p_es->i_stream_id )
 176         {
 177         case 0xBC:  /* Program stream map */
 178         case 0xBE:  /* Padding */
 179         case 0xBF:  /* Private stream 2 */
 180         case 0xB0:  /* ECM */
 181         case 0xB1:  /* EMM */
 182         case 0xFF:  /* Program stream directory */
 183         case 0xF2:  /* DSMCC stream */
 184         case 0xF8:  /* ITU-T H.222.1 type E stream */
 185             /* The payload begins immediately after the 6 bytes header, so
 186              * we have finished with the parsing */
 187             i_pes_header_size = 6;
 188             break;
 189
 190         default:
 191             if( (p_header[6] & 0xC0) == 0x80 )
 192             {
 193                 /* MPEG-2 : the PES header contains at least 3 more bytes. */
 194                 size_t      i_max_len;
 195                 boolean_t   b_has_pts, b_has_dts;
 196                 byte_t      p_full_header[12];
 197
 198                 p_pes->b_data_alignment = p_header[6] & 0x04;
 199
 200                 i_max_len = MoveChunk( p_full_header, &p_data, &p_byte, 12 );
 201                 if( i_max_len < 2 )
 202                 {
 203                     intf_WarnMsg( 1,
 204                             "PES packet too short to have a MPEG-2 header" );
 205                     p_input->pf_delete_pes( p_input->p_method_data,
 206                                             p_pes );
 207                     p_pes = NULL;
 208                     return;
 209                 }
 210
 211                 b_has_pts = p_full_header[0] & 0x80;
 212                 b_has_dts = p_full_header[0] & 0x40;
 213                 i_pes_header_size = p_full_header[1] + 9;
 214
 215                 /* Now parse the optional header extensions */
 216                 if( b_has_pts )
 217                 {
 218                     if( i_max_len < 7 )
 219                     {
 220                         intf_WarnMsg( 1,
 221                             "PES packet too short to have a MPEG-2 header" );
 222                         p_input->pf_delete_pes( p_input->p_method_data,
 223                                                 p_pes );
 224                         p_pes = NULL;
 225                         return;
 226                     }
 227                     p_pes->i_pts = input_ClockGetTS( p_input, p_es->p_pgrm,
 228                     ( ((mtime_t)(p_full_header[2] & 0x0E) << 29) |
 229                       ((mtime_t)(p_full_header[3]) << 22) |
 230                       ((mtime_t)(p_full_header[4] & 0xFE) << 14) |
 231                       ((mtime_t)p_full_header[5] << 7) |
 232                       ((mtime_t)p_full_header[6] >> 1) ) );
 233
 234                     if( b_has_dts )
 235                     {
 236                         if( i_max_len < 12 )
 237                         {
 238                             intf_WarnMsg( 1,
 239                               "PES packet too short to have a MPEG-2 header" );
 240                             p_input->pf_delete_pes( p_input->p_method_data,
 241                                                     p_pes );
 242                             p_pes = NULL;
 243                             return;
 244                         }
 245                         p_pes->i_dts = input_ClockGetTS( p_input, p_es->p_pgrm,
 246                         ( ((mtime_t)(p_full_header[7] & 0x0E) << 29) |
 247                           (((mtime_t)U16_AT(p_full_header + 8) << 14)
 248                                 - (1 << 14)) |
 249                           ((mtime_t)U16_AT(p_full_header + 10) >> 1) ) );
 250                     }
 251                 }
 252             }
 253             else
 254             {
 255                 /* Probably MPEG-1 */
 256                 boolean_t       b_has_pts, b_has_dts;
 257
 258                 i_pes_header_size = 6;
 259                 p_data = p_pes->p_first;
 260                 p_byte = p_data->p_payload_start;
 261                 /* Cannot fail because the previous one succeeded. */
 262                 MoveChunk( NULL, &p_data, &p_byte, 6 );
 263
 264                 while( *p_byte == 0xFF && i_pes_header_size < 23 )
 265                 {
 266                     i_pes_header_size++;
 267                     if( MoveChunk( NULL, &p_data, &p_byte, 1 ) != 1 )
 268                     {
 269                         intf_WarnMsg( 1,
 270                             "PES packet too short to have a MPEG-1 header" );
 271                         p_input->pf_delete_pes( p_input->p_method_data, p_pes );
 272                         p_pes = NULL;
 273                         return;
 274                     }
 275                 }
 276                 if( i_pes_header_size == 23 )
 277                 {
 278                     intf_ErrMsg( "input error: too much MPEG-1 stuffing" );
 279                     p_input->pf_delete_pes( p_input->p_method_data, p_pes );
 280                     p_pes = NULL;
 281                     return;
 282                 }
 283
 284                 if( (*p_byte & 0xC0) == 0x40 )
 285                 {
 286                     /* Don't ask why... --Meuuh */
 287                     /* Erm... why ? --Sam */
 288                     /* Well... According to the recommendation, it is for
 289                      * STD_buffer_scale and STD_buffer_size. --Meuuh */
 290                     i_pes_header_size += 2;
 291                     if( MoveChunk( NULL, &p_data, &p_byte, 2 ) != 2 )
 292                     {
 293                         intf_WarnMsg( 1, "input: PES packet too short "
 294                                          "to have a MPEG-1 header" );
 295                         p_input->pf_delete_pes( p_input->p_method_data, p_pes );
 296                         p_pes = NULL;
 297                         return;
 298                     }
 299                 }
 300
 301                 i_pes_header_size++;
 302
 303                 b_has_pts = *p_byte & 0x20;
 304                 b_has_dts = *p_byte & 0x10;
 305
 306                 if( b_has_pts )
 307                 {
 308                     byte_t      p_ts[5];
 309
 310                     i_pes_header_size += 4;
 311                     if( MoveChunk( p_ts, &p_data, &p_byte, 5 ) != 5 )
 312                     {
 313                         intf_WarnMsg( 1, "input: PES packet too short "
 314                                          "to have a MPEG-1 header" );
 315                         p_input->pf_delete_pes( p_input->p_method_data, p_pes );
 316                         p_pes = NULL;
 317                         return;
 318                     }
 319
 320                     p_pes->i_pts = input_ClockGetTS( p_input, p_es->p_pgrm,
 321                        ( ((mtime_t)(p_ts[0] & 0x0E) << 29) |
 322                          (((mtime_t)U32_AT(p_ts) & 0xFFFE00) << 6) |
 323                          ((mtime_t)p_ts[3] << 7) |
 324                          ((mtime_t)p_ts[4] >> 1) ) );
 325
 326                     if( b_has_dts )
 327                     {
 328                         i_pes_header_size += 5;
 329                         if( MoveChunk( p_ts, &p_data, &p_byte, 5 ) != 5 )
 330                         {
 331                             intf_WarnMsg( 1, "input: PES packet too short "
 332                                              "to have a MPEG-1 header" );
 333                             p_input->pf_delete_pes( p_input->p_method_data,
 334                                                     p_pes );
 335                             p_pes = NULL;
 336                             return;
 337                         }
 338
 339                         p_pes->i_dts = input_ClockGetTS( p_input,
 340                                                          p_es->p_pgrm,
 341                             ( ((mtime_t)(p_ts[0] & 0x0E) << 29) |
 342                               (((mtime_t)U32_AT(p_ts) & 0xFFFE00) << 6) |
 343                               ((mtime_t)p_ts[3] << 7) |
 344                               ((mtime_t)p_ts[4] >> 1) ) );
 345                     }
 346                 }
 347             }
 348
 349             break;
 350         }
 351
 352         if( p_es->i_stream_id == 0xbd )
 353         {
 354             /* With private stream 1, the first byte of the payload
 355              * is a stream_private_id, so skip it. */
 356             i_pes_header_size++;
 357         }
 358
 359         /* Now we've parsed the header, we just have to indicate in some
 360          * specific data packets where the PES payload begins (renumber
 361          * p_payload_start), so that the decoders can find the beginning
 362          * of their data right out of the box. */
 363         p_data = p_pes->p_first;
 364         i_payload_size = p_data->p_payload_end
 365                                  - p_data->p_payload_start;
 366         while( i_pes_header_size > i_payload_size )
 367         {
 368             /* These packets are entirely filled by the PES header. */
 369             i_pes_header_size -= i_payload_size;
 370             p_data->p_payload_start = p_data->p_payload_end;
 371             /* Go to the next data packet. */
 372             if( (p_data = p_data->p_next) == NULL )
 373             {
 374                 intf_ErrMsg( "input error: PES header bigger than payload" );
 375                 p_input->pf_delete_pes( p_input->p_method_data, p_pes );
 376                 p_pes = NULL;
 377                 return;
 378             }
 379             i_payload_size = p_data->p_payload_end
 380                                  - p_data->p_payload_start;
 381         }
 382         /* This last packet is partly header, partly payload. */
 383         if( i_payload_size < i_pes_header_size )
 384         {
 385             intf_ErrMsg( "input error: PES header bigger than payload" );
 386             p_input->pf_delete_pes( p_input->p_method_data, p_pes );
 387             p_pes = NULL;
 388             return;
 389         }
 390         p_data->p_payload_start += i_pes_header_size;
 391
 392         /* Now we can eventually put the PES packet in the decoder's
 393          * PES fifo */
 394         if( p_es->p_decoder_fifo != NULL )
 395         {
 396             input_DecodePES( p_es->p_decoder_fifo, p_pes );
 397         }
 398         else
 399         {
 400             intf_ErrMsg( "input error: no fifo to receive PES %p "
 401                          "(who wrote this damn code ?)", p_pes );
 402             p_input->pf_delete_pes( p_input->p_method_data, p_pes );
 403         }
 404         p_pes = NULL;
 405     }
 406 #undef p_pes
 407
 408 }
 409
 410 /*****************************************************************************
 411  * input_GatherPES:
 412  *****************************************************************************
 413  * Gather a PES packet.
 414  *****************************************************************************/
 415 void input_GatherPES( input_thread_t * p_input, data_packet_t * p_data,
 416                       es_descriptor_t * p_es,
 417                       boolean_t b_unit_start, boolean_t b_packet_lost )
 418 {
 419 #define p_pes (p_es->p_pes)
 420
 421     //intf_DbgMsg("PES-demultiplexing %p (%p)", p_ts_packet, p_pes);
 422
 423     /* If we lost data, insert a NULL data packet (philosophy : 0 is quite
 424      * often an escape sequence in decoders, so that should make them wait
 425      * for the next start code). */
 426     if( b_packet_lost )
 427     {
 428         input_NullPacket( p_input, p_es );
 429     }
 430
 431     if( b_unit_start && p_pes != NULL )
 432     {
 433         /* If the data packet contains the begining of a new PES packet, and
 434          * if we were reassembling a PES packet, then the PES should be
 435          * complete now, so parse its header and give it to the decoders. */
 436         input_ParsePES( p_input, p_es );
 437     }
 438
 439     if( !b_unit_start && p_pes == NULL )
 440     {
 441         /* Random access... */
 442         p_input->pf_delete_packet( p_input->p_method_data, p_data );
 443     }
 444     else
 445     {
 446         if( b_unit_start )
 447         {
 448             /* If we are at the beginning of a new PES packet, we must fetch
 449              * a new PES buffer to begin with the reassembly of this PES
 450              * packet. This is also here that we can synchronize with the
 451              * stream if we lost packets or if the decoder has just
 452              * started. */
 453             if( (p_pes = p_input->pf_new_pes( p_input->p_method_data ) ) == NULL )
 454             {
 455                 intf_ErrMsg( "input error: out of memory" );
 456                 p_input->b_error = 1;
 457                 return;
 458             }
 459             p_pes->i_rate = p_input->stream.control.i_rate;
 460             p_pes->p_first = p_data;
 461
 462             /* If the PES header fits in the first data packet, we can
 463              * already set p_gather->i_pes_real_size. */
 464             if( p_data->p_payload_end - p_data->p_payload_start
 465                     >= PES_HEADER_SIZE )
 466             {
 467                 p_es->i_pes_real_size =
 468                                 U16_AT(p_data->p_payload_start + 4) + 6;
 469
 470             }
 471             else
 472             {
 473                 p_es->i_pes_real_size = 0;
 474             }
 475         }
 476         else
 477         {
 478             /* Update the relations between the data packets */
 479             p_pes->p_last->p_next = p_data;
 480         }
 481
 482         p_pes->p_last = p_data;
 483         p_pes->i_nb_data++;
 484
 485         /* Size of the payload carried in the data packet */
 486         p_pes->i_pes_size += (p_data->p_payload_end
 487                                  - p_data->p_payload_start);
 488
 489         /* We can check if the packet is finished */
 490         if( p_pes->i_pes_size == p_es->i_pes_real_size )
 491         {
 492             /* The packet is finished, parse it */
 493             input_ParsePES( p_input, p_es );
 494         }
 495     }
 496 #undef p_pes
 497 }
 498
 499
 500 /*
 501  * PS Demultiplexing
 502  */
 503
 504 /*****************************************************************************
 505  * GetID: Get the ID of a stream
 506  *****************************************************************************/
 507 static u16 GetID( data_packet_t * p_data )
 508 {
 509     u16         i_id;
 510
 511     i_id = p_data->p_demux_start[3];                            /* stream_id */
 512     if( i_id == 0xBD )
 513     {
 514         /* FIXME : this is not valid if the header is split in multiple
 515          * packets */
 516         /* stream_private_id */
 517         i_id |= p_data->p_demux_start[ 9 + p_data->p_demux_start[8] ] << 8;
 518     }
 519     return( i_id );
 520 }
 521
 522 /*****************************************************************************
 523  * DecodePSM: Decode the Program Stream Map information
 524  *****************************************************************************
 525  * FIXME : loads are not aligned in this function
 526  *****************************************************************************/
 527 static void DecodePSM( input_thread_t * p_input, data_packet_t * p_data )
 528 {
 529     stream_ps_data_t *  p_demux =
 530                  (stream_ps_data_t *)p_input->stream.p_demux_data;
 531     byte_t *            p_byte;
 532     byte_t *            p_end;
 533     int                 i;
 534     int                 i_new_es_number = 0;
 535
 536     if( p_data->p_demux_start + 10 > p_data->p_payload_end )
 537     {
 538         intf_ErrMsg( "input error: PSM too short : packet corrupt" );
 539         return;
 540     }
 541
 542     if( p_demux->b_has_PSM
 543         && p_demux->i_PSM_version == (p_data->p_demux_start[6] & 0x1F) )
 544     {
 545         /* Already got that one. */
 546         return;
 547     }
 548
 549     intf_DbgMsg( "input: building PSM" );
 550     p_demux->b_has_PSM = 1;
 551     p_demux->i_PSM_version = p_data->p_demux_start[6] & 0x1F;
 552
 553     /* Go to elementary_stream_map_length, jumping over
 554      * program_stream_info. */
 555     p_byte = p_data->p_demux_start + 10
 556               + U16_AT(&p_data->p_demux_start[8]);
 557     if( p_byte > p_data->p_payload_end )
 558     {
 559         intf_ErrMsg( "input error: PSM too short, packet corrupt" );
 560         return;
 561     }
 562     /* This is the full size of the elementary_stream_map.
 563      * 2 == elementary_stream_map_length
 564      * Please note that CRC_32 is not included in the length. */
 565     p_end = p_byte + 2 + U16_AT(p_byte);
 566     p_byte += 2;
 567     if( p_end > p_data->p_payload_end )
 568     {
 569         intf_ErrMsg( "input error: PSM too short, packet corrupt" );
 570         return;
 571     }
 572
 573     vlc_mutex_lock( &p_input->stream.stream_lock );
 574
 575     /* 4 == minimum useful size of a section */
 576     while( p_byte + 4 <= p_end )
 577     {
 578         es_descriptor_t *   p_es = NULL;
 579         u8                  i_stream_id = p_byte[1];
 580         /* FIXME: there will be a problem with private streams... (same
 581          * stream_id) */
 582
 583         /* Look for the ES in the ES table */
 584         for( i = i_new_es_number;
 585              i < p_input->stream.pp_programs[0]->i_es_number;
 586              i++ )
 587         {
 588             if( p_input->stream.pp_programs[0]->pp_es[i]->i_stream_id
 589                     == i_stream_id )
 590             {
 591                 p_es = p_input->stream.pp_programs[0]->pp_es[i];
 592                 if( p_es->i_type != p_byte[0] )
 593                 {
 594                     input_DelES( p_input, p_es );
 595                     p_es = NULL;
 596                 }
 597                 else
 598                 {
 599                     /* Move the ES to the beginning. */
 600                     p_input->stream.pp_programs[0]->pp_es[i]
 601                         = p_input->stream.pp_programs[0]->pp_es[ i_new_es_number ];
 602                     p_input->stream.pp_programs[0]->pp_es[ i_new_es_number ]
 603                         = p_es;
 604                     i_new_es_number++;
 605                 }
 606                 break;
 607             }
 608         }
 609
 610         /* The goal is to have all the ES we have just read in the
 611          * beginning of the pp_es table, and all the others at the end,
 612          * so that we can close them more easily at the end. */
 613         if( p_es == NULL )
 614         {
 615             p_es = input_AddES( p_input, p_input->stream.pp_programs[0],
 616                                 i_stream_id, 0 );
 617             p_es->i_type = p_byte[0];
 618             p_es->b_audio = ( p_es->i_type == MPEG1_AUDIO_ES
 619                               || p_es->i_type == MPEG2_AUDIO_ES
 620                               || p_es->i_type == AC3_AUDIO_ES
 621                               || p_es->i_type == LPCM_AUDIO_ES
 622                             );
 623
 624             /* input_AddES has inserted the new element at the end. */
 625             p_input->stream.pp_programs[0]->pp_es[
 626                 p_input->stream.pp_programs[0]->i_es_number ]
 627                 = p_input->stream.pp_programs[0]->pp_es[ i_new_es_number ];
 628             p_input->stream.pp_programs[0]->pp_es[ i_new_es_number ] = p_es;
 629             i_new_es_number++;
 630         }
 631         p_byte += 4 + U16_AT(&p_byte[2]);
 632     }
 633
 634     /* Un-select the streams that are no longer parts of the program. */
 635     while( i_new_es_number < p_input->stream.pp_programs[0]->i_es_number )
 636     {
 637         /* We remove pp_es[i_new_es_member] and not pp_es[i] because the
 638          * list will be emptied starting from the end */
 639         input_DelES( p_input,
 640                      p_input->stream.pp_programs[0]->pp_es[i_new_es_number] );
 641     }
 642
 643     if( p_main->b_stats )
 644     {
 645         intf_StatMsg( "input info: The stream map after the PSM is now :" );
 646         input_DumpStream( p_input );
 647     }
 648
 649     vlc_mutex_unlock( &p_input->stream.stream_lock );
 650 }
 651
 652 /*****************************************************************************
 653  * input_ParsePS: read the PS header
 654  *****************************************************************************/
 655 es_descriptor_t * input_ParsePS( input_thread_t * p_input,
 656                                  data_packet_t * p_data )
 657 {
 658     u32                 i_code;
 659     es_descriptor_t *   p_es = NULL;
 660
 661     i_code = p_data->p_demux_start[3];
 662
 663     if( i_code > 0xBC ) /* ES start code */
 664     {
 665         u16                 i_id;
 666         int                 i_dummy;
 667
 668         /* This is a PES packet. Find out if we want it or not. */
 669         i_id = GetID( p_data );
 670
 671         vlc_mutex_lock( &p_input->stream.stream_lock );
 672         if( p_input->stream.pp_programs[0]->b_is_ok )
 673         {
 674             /* Look only at the selected ES. */
 675             for( i_dummy = 0; i_dummy < p_input->stream.i_selected_es_number;
 676                  i_dummy++ )
 677             {
 678                 if( p_input->stream.pp_selected_es[i_dummy] != NULL
 679                     && p_input->stream.pp_selected_es[i_dummy]->i_id == i_id )
 680                 {
 681                     p_es = p_input->stream.pp_selected_es[i_dummy];
 682                     break;
 683                 }
 684             }
 685         }
 686         else
 687         {
 688             stream_ps_data_t * p_demux =
 689               (stream_ps_data_t *)p_input->stream.pp_programs[0]->p_demux_data;
 690
 691             /* Search all ES ; if not found -> AddES */
 692             p_es = input_FindES( p_input, i_id );
 693
 694             if( p_es == NULL && !p_demux->b_has_PSM )
 695             {
 696                 p_es = input_AddES( p_input, p_input->stream.pp_programs[0],
 697                                     i_id, 0 );
 698                 if( p_es != NULL )
 699                 {
 700                     p_es->i_stream_id = p_data->p_demux_start[3];
 701
 702                     /* Set stream type and auto-spawn. */
 703                     if( (i_id & 0xF0) == 0xE0 )
 704                     {
 705                         /* MPEG video */
 706                         p_es->i_type = MPEG2_VIDEO_ES;
 707                         p_es->i_cat = VIDEO_ES;
 708 #ifdef AUTO_SPAWN
 709                         if( !p_input->stream.b_seekable )
 710                             input_SelectES( p_input, p_es );
 711 #endif
 712                     }
 713                     else if( (i_id & 0xE0) == 0xC0 )
 714                     {
 715                         /* MPEG audio */
 716                         p_es->i_type = MPEG2_AUDIO_ES;
 717                         p_es->b_audio = 1;
 718                         p_es->i_cat = AUDIO_ES;
 719 #ifdef AUTO_SPAWN
 720                         if( !p_input->stream.b_seekable )
 721                         if( main_GetIntVariable( INPUT_CHANNEL_VAR, 0 )
 722                                 == (p_es->i_id & 0x1F) )
 723                         switch( main_GetIntVariable( INPUT_AUDIO_VAR, 0 ) )
 724                         {
 725                         case 0:
 726                             main_PutIntVariable( INPUT_CHANNEL_VAR,
 727                                                  REQUESTED_MPEG );
 728                         case REQUESTED_MPEG:
 729                             input_SelectES( p_input, p_es );
 730                         }
 731 #endif
 732                     }
 733                     else if( (i_id & 0xF0FF) == 0x80BD )
 734                     {
 735                         /* AC3 audio (0x80->0x8F) */
 736                         p_es->i_type = AC3_AUDIO_ES;
 737                         p_es->b_audio = 1;
 738                         p_es->i_cat = AUDIO_ES;
 739 #ifdef AUTO_SPAWN
 740                         if( !p_input->stream.b_seekable )
 741                         if( main_GetIntVariable( INPUT_CHANNEL_VAR, 0 )
 742                                 == ((p_es->i_id & 0xF00) >> 8) )
 743                         switch( main_GetIntVariable( INPUT_AUDIO_VAR, 0 ) )
 744                         {
 745                         case 0:
 746                             main_PutIntVariable( INPUT_CHANNEL_VAR,
 747                                                  REQUESTED_AC3 );
 748                         case REQUESTED_AC3:
 749                             input_SelectES( p_input, p_es );
 750                         }
 751 #endif
 752                     }
 753                     else if( (i_id & 0xE0FF) == 0x20BD )
 754                     {
 755                         /* Subtitles video (0x20->0x3F) */
 756                         p_es->i_type = DVD_SPU_ES;
 757                         p_es->i_cat = SPU_ES;
 758 #ifdef AUTO_SPAWN
 759                         if( main_GetIntVariable( INPUT_SUBTITLE_VAR, -1 )
 760                                 == ((p_es->i_id & 0x1F00) >> 8) )
 761                         {
 762                             if( !p_input->stream.b_seekable )
 763                                 input_SelectES( p_input, p_es );
 764                         }
 765 #endif
 766                     }
 767                     else if( (i_id & 0xF0FF) == 0xA0BD )
 768                     {
 769                         /* LPCM audio (0xA0->0xAF) */
 770                         p_es->i_type = LPCM_AUDIO_ES;
 771                         p_es->b_audio = 1;
 772                         p_es->i_cat = AUDIO_ES;
 773                         /* FIXME : write the decoder */
 774                     }
 775                     else
 776                     {
 777                         p_es->i_type = UNKNOWN_ES;
 778                     }
 779                 }
 780
 781                 /* Tell the interface the stream has changed */
 782                 p_input->stream.b_changed = 1;
 783             }
 784         } /* stream.b_is_ok */
 785         vlc_mutex_unlock( &p_input->stream.stream_lock );
 786     } /* i_code > 0xBC */
 787
 788     return( p_es );
 789 }
 790
 791 /*****************************************************************************
 792  * input_DemuxPS: first step of demultiplexing: the PS header
 793  *****************************************************************************/
 794 void input_DemuxPS( input_thread_t * p_input, data_packet_t * p_data )
 795 {
 796     u32                 i_code;
 797     boolean_t           b_trash = 0;
 798     es_descriptor_t *   p_es = NULL;
 799
 800     i_code = ((u32)p_data->p_demux_start[0] << 24)
 801                 | ((u32)p_data->p_demux_start[1] << 16)
 802                 | ((u32)p_data->p_demux_start[2] << 8)
 803                 | p_data->p_demux_start[3];
 804     if( i_code <= 0x1BC )
 805     {
 806         switch( i_code )
 807         {
 808         case 0x1BA: /* PACK_START_CODE */
 809             {
 810                 /* Read the SCR. */
 811                 mtime_t         scr_time;
 812                 u32             i_mux_rate;
 813
 814                 if( (p_data->p_demux_start[4] & 0xC0) == 0x40 )
 815                 {
 816                     /* MPEG-2 */
 817                     byte_t      p_header[14];
 818                     byte_t *    p_byte;
 819                     p_byte = p_data->p_demux_start;
 820
 821                     if( MoveChunk( p_header, &p_data, &p_byte, 14 ) != 14 )
 822                     {
 823                         intf_WarnMsg( 1, "input: packet too short "
 824                                          "to have a header" );
 825                         b_trash = 1;
 826                         break;
 827                     }
 828                     scr_time =
 829                          ((mtime_t)(p_header[4] & 0x38) << 27) |
 830                          ((mtime_t)(U32_AT(p_header + 4) & 0x03FFF800)
 831                                         << 4) |
 832                          ((( ((mtime_t)U16_AT(p_header + 6) << 16)
 833                             | (mtime_t)U16_AT(p_header + 8) ) & 0x03FFF800)
 834                                         >> 11);
 835
 836                     /* mux_rate */
 837                     i_mux_rate = ((u32)U16_AT(p_header + 10) << 6)
 838                                    | (p_header[12] >> 2);
 839                     /* FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME
 840                      * This is the biggest kludge ever !
 841                      * I don't know what's wrong with mux_rate calculation
 842                      * but this heuristic work well : */
 843                     i_mux_rate <<= 1;
 844                     i_mux_rate /= 3;
 845                 }
 846                 else
 847                 {
 848                     /* MPEG-1 SCR is like PTS. */
 849                     byte_t      p_header[12];
 850                     byte_t *    p_byte;
 851                     p_byte = p_data->p_demux_start;
 852
 853                     if( MoveChunk( p_header, &p_data, &p_byte, 12 ) != 12 )
 854                     {
 855                         intf_WarnMsg( 1, "input: packet too short "
 856                                          "to have a header" );
 857                         b_trash = 1;
 858                         break;
 859                     }
 860                     scr_time =
 861                          ((mtime_t)(p_header[4] & 0x0E) << 29) |
 862                          (((mtime_t)U32_AT(p_header + 4) & 0xFFFE00) << 6) |
 863                          ((mtime_t)p_header[7] << 7) |
 864                          ((mtime_t)p_header[8] >> 1);
 865
 866                     /* mux_rate */
 867                     i_mux_rate = (U32_AT(p_header + 8) & 0x7FFFFE) >> 1;
 868                 }
 869                 /* Call the pace control. */
 870                 input_ClockManageRef( p_input, p_input->stream.pp_programs[0],
 871                                       scr_time );
 872
 873                 if( i_mux_rate != p_input->stream.i_mux_rate
 874                      && p_input->stream.i_mux_rate )
 875                 {
 876                     intf_WarnMsg( 2, "input: mux_rate changed, "
 877                                      "expect cosmetic errors" );
 878                 }
 879                 p_input->stream.i_mux_rate = i_mux_rate;
 880
 881                 b_trash = 1;
 882             }
 883             break;
 884
 885         case 0x1BB: /* SYSTEM_START_CODE */
 886             b_trash = 1;                              /* Nothing interesting */
 887             break;
 888
 889         case 0x1BC: /* PROGRAM_STREAM_MAP_CODE */
 890             DecodePSM( p_input, p_data );
 891             b_trash = 1;
 892             break;
 893
 894         case 0x1B9: /* PROGRAM_END_CODE */
 895             b_trash = 1;
 896             break;
 897
 898         default:
 899             /* This should not happen */
 900             b_trash = 1;
 901             intf_WarnMsg( 3, "input: unwanted packet received "
 902                              "with start code 0x%.8x", i_code );
 903         }
 904     }
 905     else
 906     {
 907         p_es = input_ParsePS( p_input, p_data );
 908
 909         vlc_mutex_lock( &p_input->stream.control.control_lock );
 910         if( p_es != NULL && p_es->p_decoder_fifo != NULL
 911              && (!p_es->b_audio || !p_input->stream.control.b_mute) )
 912         {
 913             vlc_mutex_unlock( &p_input->stream.control.control_lock );
 914             p_es->c_packets++;
 915             input_GatherPES( p_input, p_data, p_es, 1, 0 );
 916         }
 917         else
 918         {
 919             vlc_mutex_unlock( &p_input->stream.control.control_lock );
 920             b_trash = 1;
 921         }
 922     }
 923
 924     /* Trash the packet if it has no payload or if it isn't selected */
 925     if( b_trash )
 926     {
 927         p_input->pf_delete_packet( p_input->p_method_data, p_data );
 928         p_input->stream.c_packets_trashed++;
 929     }
 930 }
 931
 932
 933 /*
 934  * TS Demultiplexing
 935  */
 936
 937 /*****************************************************************************
 938  * input_DemuxTS: first step of demultiplexing: the TS header
 939  *****************************************************************************/
 940 void input_DemuxTS( input_thread_t * p_input, data_packet_t * p_data )
 941 {
 942     u16                 i_pid;
 943     int                 i_dummy;
 944     boolean_t           b_adaptation;         /* Adaptation field is present */
 945     boolean_t           b_payload;                 /* Packet carries payload */
 946     boolean_t           b_unit_start;  /* A PSI or a PES start in the packet */
 947     boolean_t           b_trash = 0;             /* Is the packet unuseful ? */
 948     boolean_t           b_lost = 0;             /* Was there a packet loss ? */
 949     boolean_t           b_psi = 0;                        /* Is this a PSI ? */
 950     es_descriptor_t *   p_es = NULL;
 951     es_ts_data_t *      p_es_demux = NULL;
 952     pgrm_ts_data_t *    p_pgrm_demux = NULL;
 953
 954 #define p (p_data->p_demux_start)
 955     /* Extract flags values from TS common header. */
 956     i_pid = ((p[1] & 0x1F) << 8) | p[2];
 957     b_unit_start = (p[1] & 0x40);
 958     b_adaptation = (p[3] & 0x20);
 959     b_payload = (p[3] & 0x10);
 960
 961     /* Find out the elementary stream. */
 962     vlc_mutex_lock( &p_input->stream.stream_lock );
 963
 964     p_es= input_FindES( p_input, i_pid );
 965
 966     if( (p_es != NULL) && (p_es->p_demux_data != NULL) )
 967     {
 968         p_es_demux = (es_ts_data_t *)p_es->p_demux_data;
 969
 970         if( p_es_demux->b_psi )
 971         {
 972             b_psi = 1;
 973         }
 974         else
 975         {
 976             p_pgrm_demux = (pgrm_ts_data_t *)p_es->p_pgrm->p_demux_data;
 977         }
 978     }
 979
 980     vlc_mutex_lock( &p_input->stream.control.control_lock );
 981     if( ( p_es == NULL ) || (p_es->b_audio && p_input->stream.control.b_mute) )
 982     {
 983         /* Not selected. Just read the adaptation field for a PCR. */
 984         b_trash = 1;
 985     }
 986     else if( p_es->p_decoder_fifo == NULL && !b_psi )
 987     {
 988         b_trash = 1;
 989     }
 990
 991     vlc_mutex_unlock( &p_input->stream.control.control_lock );
 992     vlc_mutex_unlock( &p_input->stream.stream_lock );
 993
 994
 995     /* Don't change the order of the tests : if b_psi then p_pgrm_demux
 996      * may still be null. Who said it was ugly ?
 997      * I have written worse. --Meuuh */
 998     if( ( p_es != NULL ) &&
 999         ((p_es->p_decoder_fifo != NULL) || b_psi
1000                                    || (p_pgrm_demux->i_pcr_pid == i_pid) ) )
1001     {
1002         p_es->c_packets++;
1003
1004         /* Extract adaptation field information if any */
1005
1006         if( !b_adaptation )
1007         {
1008             /* We don't have any adaptation_field, so payload starts
1009              * immediately after the 4 byte TS header */
1010             p_data->p_payload_start += 4;
1011         }
1012         else
1013         {
1014             /* p[4] is adaptation_field_length minus one */
1015             p_data->p_payload_start += 5 + p[4];
1016
1017             /* The adaptation field can be limited to the
1018              * adaptation_field_length byte, so that there is nothing to do:
1019              * skip this possibility */
1020             if( p[4] )
1021             {
1022                 /* If the packet has both adaptation_field and payload,
1023                  * adaptation_field cannot be more than 182 bytes long; if
1024                  * there is only an adaptation_field, it must fill the next
1025                  * 183 bytes. */
1026                 if( b_payload ? (p[4] > 182) : (p[4] != 183) )
1027                 {
1028                     intf_WarnMsg( 2,
1029                         "input: invalid TS adaptation field (%p)",
1030                         p_data );
1031                     p_data->b_discard_payload = 1;
1032                     p_es->c_invalid_packets++;
1033                 }
1034
1035                 /* Now we are sure that the byte containing flags is present:
1036                  * read it */
1037                 else
1038                 {
1039                     /* discontinuity_indicator */
1040                     if( p[5] & 0x80 )
1041                     {
1042                         intf_WarnMsg( 2,
1043                             "input: discontinuity_indicator"
1044                             " encountered by TS demux (position read: %d,"
1045                             " saved: %d)",
1046                             p[5] & 0x80, p_es_demux->i_continuity_counter );
1047
1048                         /* If the PID carries the PCR, there will be a system
1049                          * time-based discontinuity. We let the PCR decoder
1050                          * handle that. */
1051                         p_es->p_pgrm->i_synchro_state = SYNCHRO_REINIT;
1052
1053                         /* There also may be a continuity_counter
1054                          * discontinuity: resynchronize our counter with
1055                          * the one of the stream. */
1056                         p_es_demux->i_continuity_counter = (p[3] & 0x0f) - 1;
1057                     }
1058
1059                     /* If this is a PCR_PID, and this TS packet contains a
1060                      * PCR, we pass it along to the PCR decoder. */
1061
1062                     if( !b_psi && (p_pgrm_demux->i_pcr_pid == i_pid)
1063                         && (p[5] & 0x10) )
1064                     {
1065                         /* There should be a PCR field in the packet, check
1066                          * if the adaptation field is long enough to carry
1067                          * it. */
1068                         if( p[4] >= 7 )
1069                         {
1070                             /* Read the PCR. */
1071                             mtime_t     pcr_time;
1072                             pcr_time = ( (mtime_t)p[6] << 25 ) |
1073                                        ( (mtime_t)p[7] << 17 ) |
1074                                        ( (mtime_t)p[8] << 9 ) |
1075                                        ( (mtime_t)p[9] << 1 ) |
1076                                        ( (mtime_t)p[10] >> 7 );
1077                             /* Call the pace control. */
1078                             input_ClockManageRef( p_input, p_es->p_pgrm,
1079                                                   pcr_time );
1080                         }
1081                     } /* PCR ? */
1082                 } /* valid TS adaptation field ? */
1083             } /* length > 0 */
1084         } /* has adaptation field */
1085         /* Check the continuity of the stream. */
1086         i_dummy = ((p[3] & 0x0f) - p_es_demux->i_continuity_counter) & 0x0f;
1087         if( i_dummy == 1 )
1088         {
1089             /* Everything is ok, just increase our counter */
1090             (p_es_demux->i_continuity_counter)++;
1091         }
1092         else
1093         {
1094             if( !b_payload && i_dummy == 0 )
1095             {
1096                 /* This is a packet without payload, this is allowed by the
1097                  * draft. As there is nothing interesting in this packet
1098                  * (except PCR that have already been handled), we can trash
1099                  * the packet. */
1100                 intf_WarnMsg( 3, "input: packet without payload received "
1101                                  "by TS demux" );
1102                 b_trash = 1;
1103             }
1104             else if( i_dummy <= 0 )
1105             {
1106                 /* Duplicate packet: mark it as being to be trashed. */
1107                 intf_WarnMsg( 3, "input: duplicate packet received "
1108                                  "by TS demux" );
1109                 b_trash = 1;
1110             }
1111             else if( p_es_demux->i_continuity_counter == 0xFF )
1112             {
1113                 /* This means that the packet is the first one we receive for
1114                  * this ES since the continuity counter ranges between 0 and
1115                  * 0x0F excepts when it has been initialized by the input:
1116                  * init the counter to the correct value. */
1117                 intf_WarnMsg( 3, "input: first packet for PID %d received "
1118                              "by TS demux", p_es->i_id );
1119                 p_es_demux->i_continuity_counter = (p[3] & 0x0f);
1120             }
1121             else
1122             {
1123                 /* This can indicate that we missed a packet or that the
1124                  * continuity_counter wrapped and we received a dup packet:
1125                  * as we don't know, do as if we missed a packet to be sure
1126                  * to recover from this situation */
1127                 intf_WarnMsg( 2, "input: packet lost by TS demux: "
1128                                  "current %d, packet %d",
1129                               p_es_demux->i_continuity_counter & 0x0f,
1130                               p[3] & 0x0f );
1131                 b_lost = 1;
1132                 p_es_demux->i_continuity_counter = p[3] & 0x0f;
1133             } /* not continuous */
1134         } /* continuity */
1135     } /* if selected or PCR */
1136
1137     /* Trash the packet if it has no payload or if it isn't selected */
1138     if( b_trash )
1139     {
1140         p_input->pf_delete_packet( p_input->p_method_data, p_data );
1141         p_input->stream.c_packets_trashed++;
1142     }
1143     else
1144     {
1145         if( b_psi )
1146         {
1147             /* The payload contains PSI tables */
1148             input_DemuxPSI( p_input, p_data, p_es,
1149                             b_unit_start, b_lost );
1150
1151         }
1152         else
1153         {
1154             /* The payload carries a PES stream */
1155             input_GatherPES( p_input, p_data, p_es, b_unit_start, b_lost );
1156         }
1157
1158     }
1159
1160 #undef p
1161
1162 }
1163
1164 /*
1165  * PSI demultiplexing and decoding
1166  */
1167
1168 /*****************************************************************************
1169  * DemuxPSI : makes up complete PSI data
1170  *****************************************************************************/
1171 void input_DemuxPSI( input_thread_t * p_input, data_packet_t * p_data,
1172         es_descriptor_t * p_es, boolean_t b_unit_start, boolean_t b_lost )
1173 {
1174     es_ts_data_t  * p_demux_data;
1175
1176     p_demux_data = (es_ts_data_t *)p_es->p_demux_data;
1177
1178 #define p_psi (p_demux_data->p_psi_section)
1179 #define p (p_data->p_payload_start)
1180
1181     if( b_unit_start )
1182     {
1183         /* unit_start set to 1 -> presence of a pointer field
1184          * (see ISO/IEC 13818 (2.4.4.2) which should be set to 0x00 */
1185         if( (u8)p[0] != 0x00 )
1186         {
1187             intf_WarnMsg( 2, "input: non zero pointer field found, "
1188                              "trying to continue" );
1189             p+=(u8)p[0];
1190         }
1191         else
1192         {
1193             p++;
1194         }
1195
1196         /* This is the begining of a new section */
1197
1198         if( ((u8)(p[1]) & 0xc0) != 0x80 )
1199         {
1200             intf_WarnMsg( 2, "input: invalid PSI packet" );
1201             p_psi->b_trash = 1;
1202         }
1203         else
1204         {
1205             p_psi->i_section_length = ((p[1] & 0xF) << 8) | p[2];
1206             p_psi->b_section_complete = 0;
1207             p_psi->i_read_in_section = 0;
1208             p_psi->i_section_number = (u8)p[6];
1209
1210             if( p_psi->b_is_complete || p_psi->i_section_number == 0 )
1211             {
1212                 /* This is a new PSI packet */
1213                 p_psi->b_is_complete = 0;
1214                 p_psi->b_trash = 0;
1215                 p_psi->i_version_number = ( p[5] >> 1 ) & 0x1f;
1216                 p_psi->i_last_section_number = (u8)p[7];
1217
1218                 /* We'll write at the begining of the buffer */
1219                 p_psi->p_current = p_psi->buffer;
1220             }
1221             else
1222             {
1223                 if( p_psi->b_section_complete )
1224                 {
1225                     /* New Section of an already started PSI */
1226                     p_psi->b_section_complete = 0;
1227
1228                     if( p_psi->i_version_number != (( p[5] >> 1 ) & 0x1f) )
1229                     {
1230                         intf_WarnMsg( 2, "input: PSI version differs "
1231                                          "inside same PAT" );
1232                         p_psi->b_trash = 1;
1233                     }
1234                     if( p_psi->i_section_number + 1 != (u8)p[6] )
1235                     {
1236                         intf_WarnMsg( 2, "input: PSI Section discontinuity, "
1237                                          "packet lost ?" );
1238                         p_psi->b_trash = 1;
1239                     }
1240                     else
1241                         p_psi->i_section_number++;
1242                 }
1243                 else
1244                 {
1245                     intf_WarnMsg( 2, "input: got unexpected new PSI section" );
1246                     p_psi->b_trash = 1;
1247                 }
1248             }
1249         }
1250     } /* b_unit_start */
1251
1252     if( !p_psi->b_trash )
1253     {
1254         /* read */
1255         if( (p_data->p_payload_end - p) >=
1256             ( p_psi->i_section_length - p_psi->i_read_in_section ) )
1257         {
1258             /* The end of the section is in this TS packet */
1259             memcpy( p_psi->p_current, p,
1260             (p_psi->i_section_length - p_psi->i_read_in_section) );
1261
1262             p_psi->b_section_complete = 1;
1263             p_psi->p_current +=
1264                 (p_psi->i_section_length - p_psi->i_read_in_section);
1265
1266             if( p_psi->i_section_number == p_psi->i_last_section_number )
1267             {
1268                 /* This was the last section of PSI */
1269                 p_psi->b_is_complete = 1;
1270
1271                 switch( p_demux_data->i_psi_type)
1272                 {
1273                 case PSI_IS_PAT:
1274                     input_DecodePAT( p_input, p_es );
1275                     break;
1276                 case PSI_IS_PMT:
1277                     input_DecodePMT( p_input, p_es );
1278                     break;
1279                 default:
1280                     intf_WarnMsg(2, "Received unknown PSI in DemuxPSI");
1281                 }
1282             }
1283         }
1284         else
1285         {
1286             memcpy( p_psi->buffer, p, p_data->p_payload_end - p );
1287             p_psi->i_read_in_section += p_data->p_payload_end - p;
1288
1289             p_psi->p_current += p_data->p_payload_end - p;
1290         }
1291     }
1292
1293 #undef p_psi
1294 #undef p
1295
1296     p_input->pf_delete_packet( p_input->p_method_data, p_data );
1297
1298     return ;
1299 }
1300
1301 /*****************************************************************************
1302  * DecodePAT : Decodes Programm association table and deal with it
1303  *****************************************************************************/
1304 static void input_DecodePAT( input_thread_t * p_input, es_descriptor_t * p_es )
1305 {
1306     stream_ts_data_t  * p_stream_data;
1307     es_ts_data_t      * p_demux_data;
1308
1309     pgrm_descriptor_t * p_pgrm;
1310     es_descriptor_t   * p_current_es;
1311     byte_t            * p_current_data;
1312
1313     int                 i_section_length, i_program_id, i_pmt_pid;
1314     int                 i_loop, i_current_section;
1315
1316     boolean_t           b_changed = 0;
1317
1318     p_demux_data = (es_ts_data_t *)p_es->p_demux_data;
1319     p_stream_data = (stream_ts_data_t *)p_input->stream.p_demux_data;
1320
1321 #define p_psi (p_demux_data->p_psi_section)
1322
1323     /* Not so fast, Mike ! If the PAT version has changed, we first check
1324      * that its content has really changed before doing anything */
1325     if( p_stream_data->i_pat_version != p_psi->i_version_number )
1326     {
1327         int i_programs = p_input->stream.i_pgrm_number;
1328
1329         p_current_data = p_psi->buffer;
1330
1331         do
1332         {
1333             i_section_length = ((u32)(p_current_data[1] & 0xF) << 8) |
1334                                  p_current_data[2];
1335             i_current_section = (u8)p_current_data[6];
1336
1337             for( i_loop = 0;
1338                  ( i_loop < (i_section_length - 9) / 4 ) && !b_changed;
1339                  i_loop++ )
1340             {
1341                 i_program_id = ( (u32)*(p_current_data + i_loop * 4 + 8) << 8 )
1342                                  | *(p_current_data + i_loop * 4 + 9);
1343                 i_pmt_pid = ( ((u32)*(p_current_data + i_loop * 4 + 10) & 0x1F)
1344                                     << 8 )
1345                                | *(p_current_data + i_loop * 4 + 11);
1346
1347                 if( i_program_id )
1348                 {
1349                     if( (p_pgrm = input_FindProgram( p_input, i_program_id ))
1350                         && (p_current_es = input_FindES( p_input, i_pmt_pid ))
1351                         && p_current_es->p_pgrm == p_pgrm
1352                         && p_current_es->i_id == i_pmt_pid
1353                         && ((es_ts_data_t *)p_current_es->p_demux_data)->b_psi
1354                         && ((es_ts_data_t *)p_current_es->p_demux_data)
1355                             ->i_psi_type == PSI_IS_PMT )
1356                     {
1357                         i_programs--;
1358                     }
1359                     else
1360                     {
1361                         b_changed = 1;
1362                     }
1363                 }
1364             }
1365
1366             p_current_data += 3 + i_section_length;
1367
1368         } while( ( i_current_section < p_psi->i_last_section_number )
1369                   && !b_changed );
1370
1371         /* If we didn't find the expected amount of programs, the PAT has
1372          * changed. Otherwise, it only changed if b_changed is already != 0 */
1373         b_changed = b_changed || i_programs;
1374     }
1375
1376     if( b_changed )
1377     {
1378         /* PAT has changed. We are going to delete all programs and
1379          * create new ones. We chose not to only change what was needed
1380          * as a PAT change may mean the stream is radically changing and
1381          * this is a secure method to avoid crashes */
1382         es_ts_data_t      * p_es_demux;
1383         pgrm_ts_data_t    * p_pgrm_demux;
1384
1385         p_current_data = p_psi->buffer;
1386
1387         /* Delete all programs */
1388         while( p_input->stream.i_pgrm_number )
1389         {
1390             input_DelProgram( p_input, p_input->stream.pp_programs[0] );
1391         }
1392
1393         do
1394         {
1395             i_section_length = ((u32)(p_current_data[1] & 0xF) << 8) |
1396                                  p_current_data[2];
1397             i_current_section = (u8)p_current_data[6];
1398
1399             for( i_loop = 0; i_loop < (i_section_length - 9) / 4 ; i_loop++ )
1400             {
1401                 i_program_id = ( (u32)*(p_current_data + i_loop * 4 + 8) << 8 )
1402                                  | *(p_current_data + i_loop * 4 + 9);
1403                 i_pmt_pid = ( ((u32)*(p_current_data + i_loop * 4 + 10) & 0x1F)
1404                                     << 8 )
1405                                | *(p_current_data + i_loop * 4 + 11);
1406
1407                 /* If program = 0, we're having info about NIT not PMT */
1408                 if( i_program_id )
1409                 {
1410                     /* Add this program */
1411                     p_pgrm = input_AddProgram( p_input, i_program_id,
1412                                                sizeof( pgrm_ts_data_t ) );
1413
1414                     /* whatis the PID of the PMT of this program */
1415                     p_pgrm_demux = (pgrm_ts_data_t *)p_pgrm->p_demux_data;
1416                     p_pgrm_demux->i_pmt_version = PMT_UNINITIALIZED;
1417
1418                     /* Add the PMT ES to this program */
1419                     p_current_es = input_AddES( p_input, p_pgrm,(u16)i_pmt_pid,
1420                                         sizeof( es_ts_data_t) );
1421                     p_es_demux = (es_ts_data_t *)p_current_es->p_demux_data;
1422                     p_es_demux->b_psi = 1;
1423                     p_es_demux->i_psi_type = PSI_IS_PMT;
1424
1425                     p_es_demux->p_psi_section =
1426                                             malloc( sizeof( psi_section_t ) );
1427                     p_es_demux->p_psi_section->b_is_complete = 0;
1428                 }
1429             }
1430
1431             p_current_data += 3 + i_section_length;
1432
1433         } while( i_current_section < p_psi->i_last_section_number );
1434
1435         /* Go to the beginning of the next section */
1436         p_stream_data->i_pat_version = p_psi->i_version_number;
1437
1438     }
1439 #undef p_psi
1440
1441     /* FIXME This has nothing to do here */
1442     p_input->stream.p_selected_program = p_input->stream.pp_programs[0] ;
1443 }
1444
1445 /*****************************************************************************
1446  * DecodePMT : decode a given Program Stream Map
1447  * ***************************************************************************
1448  * When the PMT changes, it may mean a deep change in the stream, and it is
1449  * careful to delete the ES and add them again. If the PMT doesn't change,
1450  * there no need to do anything.
1451  *****************************************************************************/
1452 static void input_DecodePMT( input_thread_t * p_input, es_descriptor_t * p_es )
1453 {
1454
1455     pgrm_ts_data_t            * p_pgrm_data;
1456     es_ts_data_t              * p_demux_data;
1457
1458     p_demux_data = (es_ts_data_t *)p_es->p_demux_data;
1459     p_pgrm_data = (pgrm_ts_data_t *)p_es->p_pgrm->p_demux_data;
1460
1461 #define p_psi (p_demux_data->p_psi_section)
1462
1463     if( p_psi->i_version_number != p_pgrm_data->i_pmt_version )
1464     {
1465         es_descriptor_t   * p_new_es;
1466         es_ts_data_t      * p_es_demux;
1467         byte_t            * p_current_data, * p_current_section;
1468         int                 i_section_length,i_current_section;
1469         int                 i_prog_info_length, i_loop;
1470         int                 i_es_info_length, i_pid, i_stream_type;
1471         int                 i_audio_es, i_spu_es;
1472         int                 i_required_audio_es, i_required_spu_es;
1473
1474         p_current_section = p_psi->buffer;
1475         p_current_data = p_psi->buffer;
1476
1477         p_pgrm_data->i_pcr_pid = ( ((u32)*(p_current_section + 8) & 0x1F) << 8 ) |
1478                                     *(p_current_section + 9);
1479
1480         i_audio_es = 0;
1481         i_spu_es = 0;
1482
1483         /* Lock stream information */
1484         vlc_mutex_lock( &p_input->stream.stream_lock );
1485
1486         /* Get the number of the required audio stream */
1487         if( p_main->b_audio )
1488         {
1489             /* Default is the first one */
1490             i_required_audio_es = main_GetIntVariable( INPUT_CHANNEL_VAR, 1 );
1491             if( i_required_audio_es < 0 )
1492             {
1493                 main_PutIntVariable( INPUT_CHANNEL_VAR, 1 );
1494                 i_required_audio_es = 1;
1495             }
1496         }
1497         else
1498         {
1499             i_required_audio_es = 0;
1500         }
1501
1502         /* Same thing for subtitles */
1503         if( p_main->b_video )
1504         {
1505             /* for spu, default is none */
1506             i_required_spu_es = main_GetIntVariable( INPUT_SUBTITLE_VAR, 0 );
1507             if( i_required_spu_es < 0 )
1508             {
1509                 main_PutIntVariable( INPUT_SUBTITLE_VAR, 0 );
1510                 i_required_spu_es = 0;
1511             }
1512         }
1513         else
1514         {
1515             i_required_spu_es = 0;
1516         }
1517
1518         /* Delete all ES in this program  except the PSI. We start from the
1519          * end because i_es_number gets decremented after each deletion. */
1520         for( i_loop = p_es->p_pgrm->i_es_number ; i_loop ; )
1521         {
1522             i_loop--;
1523             p_es_demux = (es_ts_data_t *)
1524                          p_es->p_pgrm->pp_es[i_loop]->p_demux_data;
1525             if ( ! p_es_demux->b_psi )
1526             {
1527                 input_DelES( p_input, p_es->p_pgrm->pp_es[i_loop] );
1528             }
1529         }
1530
1531         /* Then add what we received in this PMT */
1532         do
1533         {
1534             i_section_length = ( ((u32)*(p_current_data + 1) & 0xF) << 8 ) |
1535                                   *(p_current_data + 2);
1536             i_current_section = (u8)p_current_data[6];
1537             i_prog_info_length = ( ((u32)*(p_current_data + 10) & 0xF) << 8 ) |
1538                                     *(p_current_data + 11);
1539
1540             /* For the moment we ignore program descriptors */
1541             p_current_data += 12 + i_prog_info_length;
1542
1543             /* The end of the section, before the CRC is at
1544              * p_current_section + i_section_length -1 */
1545             while( p_current_data < p_current_section + i_section_length -1 )
1546             {
1547                 i_stream_type = (int)p_current_data[0];
1548                 i_pid = ( ((u32)*(p_current_data + 1) & 0x1F) << 8 ) |
1549                            *(p_current_data + 2);
1550                 i_es_info_length = ( ((u32)*(p_current_data + 3) & 0xF) << 8 ) |
1551                                       *(p_current_data + 4);
1552
1553                 /* Add this ES to the program */
1554                 p_new_es = input_AddES( p_input, p_es->p_pgrm,
1555                                         (u16)i_pid, sizeof( es_ts_data_t ) );
1556
1557                 /* Tell the decoders what kind of stream it is */
1558                 p_new_es->i_type = i_stream_type;
1559
1560                 /* Tell the interface what kind of stream it is and select
1561                  * the required ones */
1562                 switch( i_stream_type )
1563                 {
1564                     case MPEG1_VIDEO_ES:
1565                     case MPEG2_VIDEO_ES:
1566                         p_new_es->i_cat = VIDEO_ES;
1567                         input_SelectES( p_input, p_new_es );
1568                         break;
1569                     case MPEG1_AUDIO_ES:
1570                     case MPEG2_AUDIO_ES:
1571                         p_new_es->i_cat = AUDIO_ES;
1572                         i_audio_es += 1;
1573                         if( i_audio_es == i_required_audio_es )
1574                             input_SelectES( p_input, p_new_es );
1575                         break;
1576                     case LPCM_AUDIO_ES :
1577                     case AC3_AUDIO_ES :
1578                         p_new_es->i_stream_id = 0xBD;
1579                         p_new_es->i_cat = AUDIO_ES;
1580                         i_audio_es += 1;
1581                         if( i_audio_es == i_required_audio_es )
1582                             input_SelectES( p_input, p_new_es );
1583                         break;
1584                     /* Not sure this one is fully specification-compliant */
1585                     case DVD_SPU_ES :
1586                         p_new_es->i_stream_id = 0xBD;
1587                         p_new_es->i_cat = SPU_ES;
1588                         i_spu_es += 1;
1589                         if( i_spu_es == i_required_spu_es )
1590                             input_SelectES( p_input, p_new_es );
1591                         break;
1592                     default :
1593                         p_new_es->i_cat = UNKNOWN_ES;
1594                         break;
1595                 }
1596
1597                 p_current_data += 5 + i_es_info_length;
1598             }
1599
1600             /* Go to the beginning of the next section*/
1601             p_current_data += 3 + i_section_length;
1602
1603             p_current_section++;
1604
1605         } while( i_current_section < p_psi->i_last_section_number );
1606
1607         if( i_required_audio_es > i_audio_es )
1608         {
1609             intf_WarnMsg( 2, "input: non-existing audio ES required" );
1610         }
1611
1612         if( i_required_spu_es > i_spu_es )
1613         {
1614             intf_WarnMsg( 2, "input: non-existing subtitles ES required" );
1615         }
1616
1617         p_pgrm_data->i_pmt_version = p_psi->i_version_number;
1618
1619         /* inform interface that stream has changed */
1620         p_input->stream.b_changed = 1;
1621         /*  Remove lock */
1622         vlc_mutex_unlock( &p_input->stream.stream_lock );
1623     }
1624
1625 #undef p_psi
1626 }
1627