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

   1 /*****************************************************************************
   2  * input.c: input thread
   3  * Read an MPEG2 stream, demultiplex and parse it before sending it to
   4  * decoders.
   5  *****************************************************************************
   6  * Copyright (C) 1998, 1999, 2000 VideoLAN
   7  *
   8  * Authors:
   9  *
  10  * This program is free software; you can redistribute it and/or modify
  11  * it under the terms of the GNU General Public License as published by
  12  * the Free Software Foundation; either version 2 of the License, or
  13  * (at your option) any later version.
  14  *
  15  * This program is distributed in the hope that it will be useful,
  16  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18  * GNU General Public License for more details.
  19  *
  20  * You should have received a copy of the GNU General Public License
  21  * along with this program; if not, write to the Free Software
  22  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
  23  *****************************************************************************/
  24
  25 /*****************************************************************************
  26  * Preamble
  27  *****************************************************************************/
  28 #include "defs.h"
  29
  30 #include <errno.h>                                                  /* errno */
  31 #include <sys/types.h>                        /* on BSD, uio.h needs types.h */
  32 #include <sys/uio.h>                                            /* "input.h" */
  33 #include <string.h>                                            /* strerror() */
  34
  35 #include <stdlib.h>                                                /* free() */
  36 #include <netinet/in.h>                                           /* ntohs() */
  37
  38 #include "config.h"
  39 #include "common.h"
  40 #include "threads.h"
  41 #include "mtime.h"
  42 #include "intf_msg.h"
  43 #include "plugins.h"
  44 #include "debug.h"
  45
  46 #include "input.h"
  47 #include "input_psi.h"
  48 #include "input_pcr.h"
  49 #include "input_netlist.h"
  50 #include "decoder_fifo.h"
  51 #include "input_file.h"
  52 #include "input_network.h"
  53
  54 #include "audio_output.h"                                   /* aout_thread_t */
  55
  56 #include "audio_decoder.h"        /* audiodec_t (for audio_decoder_thread.h) */
  57 #include "audio_decoder_thread.h"                           /* adec_thread_t */
  58
  59 #include "ac3_decoder.h"              /* ac3dec_t (for ac3_decoder_thread.h) */
  60 #include "ac3_decoder_thread.h"                           /* ac3dec_thread_t */
  61
  62 #include "lpcm_decoder.h"
  63 #include "lpcm_decoder_thread.h"
  64
  65 #include "video.h"                          /* picture_t (for video_output.h) */
  66 #include "video_output.h"                                   /* vout_thread_t */
  67
  68 #include "vdec_idct.h"                     /* dctelem_t (for video_parser.h) */
  69 #include "vdec_motion.h"                  /* f_motion_t (for video_parser.h) */
  70 #include "vpar_blocks.h"                /* macroblock_t (for video_parser.h) */
  71 #include "vpar_headers.h"                 /* sequence_t (for video_parser.h) */
  72 #include "vpar_synchro.h"            /* video_synchro_t (for video_parser.h) */
  73 #include "video_parser.h"                                   /* vpar_thread_t */
  74
  75 #include "spu_decoder.h"                                  /* spudec_thread_t */
  76
  77 #include "main.h"
  78
  79 /*****************************************************************************
  80  * Local prototypes
  81  *****************************************************************************/
  82 static void RunThread   ( input_thread_t *p_input );
  83 static void ErrorThread ( input_thread_t *p_input );
  84 static void EndThread   ( input_thread_t *p_input );
  85
  86 static __inline__ int   input_ReadPacket( input_thread_t *p_input );
  87 static __inline__ void  input_SortPacket( input_thread_t *p_input,
  88                                           ts_packet_t *ts_packet );
  89 static __inline__ void  input_DemuxTS( input_thread_t *p_input,
  90                                        ts_packet_t *ts_packet,
  91                                        es_descriptor_t *es_descriptor );
  92 static __inline__ void  input_DemuxPES( input_thread_t *p_input,
  93                                         ts_packet_t *ts_packet,
  94                                         es_descriptor_t *p_es_descriptor,
  95                                         boolean_t b_unit_start, boolean_t b_packet_lost );
  96 static __inline__ void  input_ParsePES( input_thread_t *p_input,
  97                                         es_descriptor_t *p_es_descriptor );
  98 static __inline__ void  input_DemuxPSI( input_thread_t *p_input,
  99                                         ts_packet_t *ts_packet,
 100                                         es_descriptor_t *p_es_descriptor,
 101                                         boolean_t b_unit_start, boolean_t b_packet_lost );
 102
 103 /*****************************************************************************
 104  * input_CreateThread: creates a new input thread
 105  *****************************************************************************
 106  * This function creates a new input, and returns a pointer
 107  * to its description. On error, it returns NULL.
 108  * If pi_status is NULL, then the function will block until the thread is ready.
 109  * If not, it will be updated using one of the THREAD_* constants.
 110  *****************************************************************************/
 111 input_thread_t *input_CreateThread ( int i_method, void *p_source, int i_port, int i_vlan,
 112                                      p_vout_thread_t p_vout, p_aout_thread_t p_aout, int *pi_status )
 113 {
 114     input_thread_t *    p_input;                        /* thread descriptor */
 115     int                 i_status;                           /* thread status */
 116     int                 i_index;          /* index for tables initialization */
 117
 118     /* Allocate descriptor */
 119     intf_DbgMsg("\n");
 120     p_input = (input_thread_t *)malloc( sizeof(input_thread_t) );
 121     if( p_input == NULL )
 122     {
 123         intf_ErrMsg("error: %s\n", strerror(ENOMEM));
 124         return( NULL );
 125     }
 126
 127     /* Initialize thread properties */
 128     p_input->b_die              = 0;
 129     p_input->b_error            = 0;
 130     p_input->pi_status          = (pi_status != NULL) ? pi_status : &i_status;
 131     *p_input->pi_status         = THREAD_CREATE;
 132
 133     /* Initialize input method description */
 134     p_input->i_method           = i_method;
 135     p_input->p_source           = p_source;
 136     p_input->i_port             = i_port;
 137     p_input->i_vlan             = i_vlan;
 138     switch( i_method )
 139     {
 140     case INPUT_METHOD_TS_FILE:                               /* file methods */
 141         p_input->p_Open =   input_FileOpen;
 142         p_input->p_Read =   input_FileRead;
 143         p_input->p_Close =  input_FileClose;
 144         break;
 145     case INPUT_METHOD_TS_VLAN_BCAST:                  /* vlan network method */
 146         if( !p_main->b_vlans )
 147         {
 148             intf_ErrMsg("error: vlans are not activated\n");
 149             free( p_input );
 150             return( NULL );
 151         }
 152         /* ... pass through */
 153     case INPUT_METHOD_TS_UCAST:                           /* network methods */
 154     case INPUT_METHOD_TS_MCAST:
 155     case INPUT_METHOD_TS_BCAST:
 156         p_input->p_Open =   input_NetworkOpen;
 157         p_input->p_Read =   input_NetworkRead;
 158         p_input->p_Close =  input_NetworkClose;
 159         break;
 160     default:
 161         intf_ErrMsg("error: unknow input method\n");
 162         free( p_input );
 163         return( NULL );
 164         break;
 165     }
 166
 167     /* Initialize stream description */
 168     for( i_index = 0; i_index < INPUT_MAX_ES; i_index++ )
 169     {
 170         p_input->p_es[i_index].i_id = EMPTY_PID;
 171         p_input->pp_selected_es[i_index] = NULL;
 172     }
 173
 174     /* Initialize default settings for spawned decoders */
 175     p_input->p_aout                     = p_aout;
 176     p_input->p_vout                     = p_vout;
 177
 178 #ifdef STATS
 179     /* Initialize statistics */
 180     p_input->c_loops                    = 0;
 181     p_input->c_bytes                    = 0;
 182     p_input->c_payload_bytes            = 0;
 183     p_input->c_packets_read             = 0;
 184     p_input->c_packets_trashed          = 0;
 185 #endif
 186
 187     /* Initialize PSI and PCR decoders */
 188     if( input_PsiInit( p_input ) )
 189     {
 190         free( p_input );
 191         return( NULL );
 192     }
 193
 194     if( input_PcrInit( p_input ) )
 195     {
 196         input_PsiEnd( p_input );
 197         free( p_input );
 198         return( NULL );
 199     }
 200
 201     /* Initialize netlists */
 202     if( input_NetlistInit( p_input ) )
 203     {
 204         input_PsiEnd( p_input );
 205         input_PcrEnd( p_input );
 206         free( p_input );
 207         return( NULL );
 208     }
 209
 210     intf_DbgMsg("configuration: method=%d, source=%s, port=%d, vlan=%d\n",
 211                 i_method, p_source, i_port, i_vlan );
 212
 213     /* Let the appropriate method open the socket. */
 214     if( p_input->p_Open( p_input ) )
 215     {
 216         input_NetlistEnd( p_input );
 217         input_PsiEnd( p_input );
 218         input_PcrEnd( p_input );
 219         free( p_input );
 220         return( NULL );
 221     }
 222
 223     /* Create thread and set locks. */
 224     vlc_mutex_init( &p_input->netlist.lock );
 225     vlc_mutex_init( &p_input->programs_lock );
 226     vlc_mutex_init( &p_input->es_lock );
 227     if( vlc_thread_create(&p_input->thread_id, "input", (void *) RunThread, (void *) p_input) )
 228     {
 229         intf_ErrMsg("error: %s\n", strerror(errno) );
 230         p_input->p_Close( p_input );
 231         input_NetlistEnd( p_input );;
 232         input_PsiEnd( p_input );
 233         input_PcrEnd( p_input );
 234         free( p_input );
 235         return( NULL );
 236     }
 237
 238     intf_Msg("Input initialized\n");
 239
 240     /* If status is NULL, wait until the thread is created */
 241     if( pi_status == NULL )
 242     {
 243         do
 244         {
 245             msleep( THREAD_SLEEP );
 246         }while( (i_status != THREAD_READY) && (i_status != THREAD_ERROR)
 247                 && (i_status != THREAD_FATAL) );
 248         if( i_status != THREAD_READY )
 249         {
 250             return( NULL );
 251         }
 252     }
 253     return( p_input );
 254 }
 255
 256 /*****************************************************************************
 257  * input_DestroyThread: mark an input thread as zombie
 258  *****************************************************************************
 259  * This function should not return until the thread is effectively cancelled.
 260  *****************************************************************************/
 261 void input_DestroyThread( input_thread_t *p_input, int *pi_status )
 262 {
 263     int         i_status;                                   /* thread status */
 264
 265     /* Set status */
 266     p_input->pi_status = (pi_status != NULL) ? pi_status : &i_status;
 267     *p_input->pi_status = THREAD_DESTROY;
 268
 269     /* Request thread destruction */
 270     p_input->b_die = 1;
 271
 272     /* If status is NULL, wait until thread has been destroyed */
 273     if( pi_status == NULL )
 274     {
 275         do
 276         {
 277             msleep( THREAD_SLEEP );
 278         }while( (i_status != THREAD_OVER) && (i_status != THREAD_ERROR)
 279                 && (i_status != THREAD_FATAL) );
 280     }
 281 }
 282
 283 #if 0
 284 /*****************************************************************************
 285  * input_OpenAudioStream: open an audio stream
 286  *****************************************************************************
 287  * This function spawns an audio decoder and plugs it on the audio output
 288  * thread.
 289  *****************************************************************************/
 290 int input_OpenAudioStream( input_thread_t *p_input, int i_id )
 291 {
 292     /* XXX?? */
 293 }
 294
 295 /*****************************************************************************
 296  * input_CloseAudioStream: close an audio stream
 297  *****************************************************************************
 298  * This function destroys an audio decoder.
 299  *****************************************************************************/
 300 void input_CloseAudioStream( input_thread_t *p_input, int i_id )
 301 {
 302     /* XXX?? */
 303 }
 304
 305 /*****************************************************************************
 306  * input_OpenVideoStream: open a video stream
 307  *****************************************************************************
 308  * This function spawns a video decoder and plugs it on a video output thread.
 309  *****************************************************************************/
 310 int input_OpenVideoStream( input_thread_t *p_input,
 311                            struct vout_thread_s *p_vout, struct video_cfg_s * p_cfg )
 312 {
 313     /* XXX?? */
 314 }
 315
 316 /*****************************************************************************
 317  * input_CloseVideoStream: close a video stream
 318  *****************************************************************************
 319  * This function destroys an video decoder.
 320  *****************************************************************************/
 321 void input_CloseVideoStream( input_thread_t *p_input, int i_id )
 322 {
 323     /* XXX?? */
 324 }
 325 #endif
 326
 327 /* following functions are local */
 328
 329 /*****************************************************************************
 330  * InitThread: initialize input thread
 331  *****************************************************************************
 332  * This function is called from RunThread and performs the second step of the
 333  * initialization. It returns 0 on success. Note that the thread's flag are not
 334  * modified inside this function.
 335  *****************************************************************************/
 336 static int InitThread( input_thread_t *p_input )
 337 {
 338     /* Mark thread as running and return */
 339     intf_DbgMsg("\n");
 340     *p_input->pi_status =        THREAD_READY;
 341     intf_DbgMsg("thread ready\n");
 342     return( 0 );
 343 }
 344
 345 /*****************************************************************************
 346  * RunThread: main thread loop
 347  *****************************************************************************
 348  * Thread in charge of processing the network packets and demultiplexing.
 349  *****************************************************************************/
 350 static void RunThread( input_thread_t *p_input )
 351 {
 352     /*
 353      * Initialize thread and free configuration
 354      */
 355     p_input->b_error = InitThread( p_input );
 356     if( p_input->b_error )
 357     {
 358         free( p_input );                               /* destroy descriptor */
 359         return;
 360     }
 361
 362     /*
 363      * Main loop
 364      */
 365     intf_DbgMsg("\n");
 366     while( !p_input->b_die && !p_input->b_error )
 367     {
 368         /* Scatter read the UDP packet from the network or the file. */
 369         if( (input_ReadPacket( p_input )) == (-1) )
 370         {
 371             /* FIXME??: Normally, a thread can't kill itself, but we don't have
 372              * any method in case of an error condition ... */
 373             p_input->b_error = 1;
 374         }
 375
 376 #ifdef STATS
 377         p_input->c_loops++;
 378 #endif
 379     }
 380
 381     /*
 382      * Error loop
 383      */
 384     if( p_input->b_error )
 385     {
 386         ErrorThread( p_input );
 387     }
 388
 389     /* End of thread */
 390     EndThread( p_input );
 391     intf_DbgMsg("thread end\n");
 392 }
 393
 394
 395 /*****************************************************************************
 396  * ErrorThread: RunThread() error loop
 397  *****************************************************************************
 398  * This function is called when an error occured during thread main's loop.
 399  *****************************************************************************/
 400 static void ErrorThread( input_thread_t *p_input )
 401 {
 402     /* Wait until a `die' order */
 403     intf_DbgMsg("\n");
 404     while( !p_input->b_die )
 405     {
 406         /* Sleep a while */
 407         msleep( VOUT_IDLE_SLEEP );
 408     }
 409 }
 410
 411 /*****************************************************************************
 412  * EndThread: end the input thread
 413  *****************************************************************************/
 414 static void EndThread( input_thread_t * p_input )
 415 {
 416     int *       pi_status;                                  /* threas status */
 417     int         i_es_loop;                                       /* es index */
 418
 419     /* Store status */
 420     intf_DbgMsg("\n");
 421     pi_status = p_input->pi_status;
 422     *pi_status = THREAD_END;
 423
 424     /* Close input method */
 425     p_input->p_Close( p_input );
 426
 427     /* Destroy all decoder threads */
 428     for( i_es_loop = 0;
 429          (i_es_loop < INPUT_MAX_ES) && (p_input->pp_selected_es[i_es_loop] != NULL) ;
 430          i_es_loop++ )
 431     {
 432         switch( p_input->pp_selected_es[i_es_loop]->i_type )
 433         {
 434         case MPEG1_VIDEO_ES:
 435         case MPEG2_VIDEO_ES:
 436             vpar_DestroyThread( (vpar_thread_t*)(p_input->pp_selected_es[i_es_loop]->p_dec) /*, NULL */ );
 437             break;
 438         case MPEG1_AUDIO_ES:
 439         case MPEG2_AUDIO_ES:
 440             adec_DestroyThread( (adec_thread_t*)(p_input->pp_selected_es[i_es_loop]->p_dec) );
 441             break;
 442         case AC3_AUDIO_ES:
 443             ac3dec_DestroyThread( (ac3dec_thread_t *)(p_input->pp_selected_es[i_es_loop]->p_dec) );
 444             break;
 445         case LPCM_AUDIO_ES:
 446             lpcmdec_DestroyThread((lpcmdec_thread_t *)(p_input->pp_selected_es[i_es_loop]->p_dec) );
 447             break;
 448         case DVD_SPU_ES:
 449             spudec_DestroyThread( (spudec_thread_t *)(p_input->pp_selected_es[i_es_loop]->p_dec) );
 450             break;
 451         case 0:
 452             /* Special streams for the PSI decoder, PID 0 and 1 */
 453             break;
 454 #ifdef DEBUG
 455         default:
 456             intf_DbgMsg("error: unknown decoder type %d\n", p_input->pp_selected_es[i_es_loop]->i_type );
 457             break;
 458 #endif
 459         }
 460     }
 461
 462     input_NetlistEnd( p_input );                            /* clean netlist */
 463     input_PsiEnd( p_input );                        /* clean PSI information */
 464     input_PcrEnd( p_input );                        /* clean PCR information */
 465     free( p_input );                          /* free input_thread structure */
 466
 467     /* Update status */
 468     *pi_status = THREAD_OVER;
 469 }
 470
 471 /*****************************************************************************
 472  * input_ReadPacket: reads a packet from the network or the file
 473  *****************************************************************************/
 474 static __inline__ int input_ReadPacket( input_thread_t *p_input )
 475 {
 476     int                 i_base_index; /* index of the first free iovec */
 477     int                 i_current_index;
 478     int                 i_packet_size;
 479 #ifdef INPUT_LIFO_TS_NETLIST
 480     int                 i_meanwhile_released;
 481     int                 i_currently_removed;
 482 #endif
 483     ts_packet_t *       p_ts_packet;
 484
 485     /* In this function, we only care about the TS netlist. PES netlist
 486      * is for the demultiplexer. */
 487 #ifdef INPUT_LIFO_TS_NETLIST
 488     i_base_index = p_input->netlist.i_ts_index;
 489
 490     /* Verify that we still have packets in the TS netlist */
 491     if( (INPUT_MAX_TS + INPUT_TS_READ_ONCE - 1 - p_input->netlist.i_ts_index) <= INPUT_TS_READ_ONCE )
 492     {
 493         intf_ErrMsg("input error: TS netlist is empty !\n");
 494         return( -1 );
 495     }
 496
 497 #else /* FIFO netlist */
 498     i_base_index = p_input->netlist.i_ts_start;
 499     if( p_input->netlist.i_ts_start + INPUT_TS_READ_ONCE -1 > INPUT_MAX_TS )
 500     {
 501         /* The netlist is splitted in 2 parts. We must gather them to consolidate
 502            the FIFO (we make the loop easily in having the same iovec at the far
 503            end and in the beginning of netlist_free).
 504            That's why the netlist is (INPUT_MAX_TS +1) + (INPUT_TS_READ_ONCE -1)
 505            large. */
 506         memcpy( p_input->netlist.p_ts_free + INPUT_MAX_TS + 1,
 507                 p_input->netlist.p_ts_free,
 508                 (p_input->netlist.i_ts_start + INPUT_TS_READ_ONCE - 1 - INPUT_MAX_TS)
 509                   * sizeof(struct iovec) );
 510     }
 511
 512     /* Verify that we still have packets in the TS netlist */
 513     if( ((p_input->netlist.i_ts_end -1 - p_input->netlist.i_ts_start) & INPUT_MAX_TS) <= INPUT_TS_READ_ONCE )
 514     {
 515         intf_ErrMsg("input error: TS netlist is empty !\n");
 516         return( -1 );
 517     }
 518 #endif /* FIFO netlist */
 519
 520     /* Scatter read the buffer. */
 521     i_packet_size = (*p_input->p_Read)( p_input,
 522                            &p_input->netlist.p_ts_free[i_base_index],
 523                            INPUT_TS_READ_ONCE );
 524     if( i_packet_size == (-1) )
 525     {
 526 #if 0
 527         intf_DbgMsg("Read packet %d %p %d %d\n", i_base_index,
 528                     &p_input->netlist.p_ts_free[i_base_index],
 529                     p_input->netlist.i_ts_start,
 530                     p_input->netlist.i_ts_end);
 531 #endif
 532         intf_ErrMsg("input error: readv() failed (%s)\n", strerror(errno));
 533         return( -1 );
 534     }
 535
 536     if( i_packet_size == 0 )
 537     {
 538         /* No packet has been received, so stop here. */
 539         return( 0 );
 540     }
 541
 542     /* Demultiplex the TS packets (1..INPUT_TS_READ_ONCE) received. */
 543     for( i_current_index = i_base_index;
 544          (i_packet_size -= TS_PACKET_SIZE) >= 0;
 545          i_current_index++ )
 546     {
 547         /* BTW, something REALLY bad could happen if we receive packets with
 548            a wrong size. */
 549         p_ts_packet = (ts_packet_t*)(p_input->netlist.p_ts_free[i_current_index].iov_base);
 550         /* Don't cry :-), we are allowed to do that cast, because initially,
 551            our buffer was malloc'ed with sizeof(ts_packet_t) */
 552
 553         /* Find out if we need this packet and demultiplex. */
 554         input_SortPacket( p_input /* for current PIDs and netlist */,
 555                           p_ts_packet);
 556     }
 557
 558     if( i_packet_size > 0 )
 559     {
 560         intf_ErrMsg("input error: wrong size\n");
 561         return( -1 );
 562     }
 563
 564     /* Remove the TS packets we have just filled from the netlist */
 565 #ifdef INPUT_LIFO_TS_NETLIST
 566     /* We need to take a lock here while we're calculating index positions. */
 567     vlc_mutex_lock( &p_input->netlist.lock );
 568
 569     i_meanwhile_released = i_base_index - p_input->netlist.i_ts_index;
 570     if( i_meanwhile_released )
 571     {
 572         /* That's where it becomes funny :-). Since we didn't take locks for
 573            efficiency reasons, other threads (including ourselves, with
 574            input_DemuxPacket) might have released packets to the netlist.
 575            So we have to copy these iovec where they should go.
 576
 577            BTW, that explains why the TS netlist is
 578            (INPUT_MAX_TS +1) + (TS_READ_ONCE -1) large. */
 579
 580         i_currently_removed = i_current_index - i_base_index;
 581         if( i_meanwhile_released < i_currently_removed )
 582         {
 583             /* Copy all iovecs in that case */
 584             memcpy( &p_input->netlist.p_ts_free[p_input->netlist.i_ts_index]
 585                      + i_currently_removed,
 586                     &p_input->netlist.p_ts_free[p_input->netlist.i_ts_index],
 587                     i_meanwhile_released * sizeof(struct iovec) );
 588         }
 589         else
 590         {
 591             /* We have fewer places than items, so we only move
 592                i_currently_removed of them. */
 593             memcpy( &p_input->netlist.p_ts_free[i_base_index],
 594                     &p_input->netlist.p_ts_free[p_input->netlist.i_ts_index],
 595                     i_currently_removed * sizeof(struct iovec) );
 596         }
 597
 598         /* Update i_netlist_index with the information gathered above. */
 599         p_input->netlist.i_ts_index += i_currently_removed;
 600     }
 601     else
 602     {
 603         /* Nothing happened. */
 604         p_input->netlist.i_ts_index = i_current_index;
 605     }
 606
 607     vlc_mutex_unlock( &p_input->netlist.lock );
 608
 609 #else /* FIFO netlist */
 610     /* & is modulo ; that's where we make the loop. */
 611     p_input->netlist.i_ts_start = i_current_index & INPUT_MAX_TS;
 612 #endif
 613
 614 #ifdef STATS
 615     p_input->c_packets_read += i_current_index - i_base_index;
 616     p_input->c_bytes += (i_current_index - i_base_index) * TS_PACKET_SIZE;
 617 #endif
 618     return( 0 );
 619 }
 620
 621 /*****************************************************************************
 622  * input_SortPacket: find out whether we need that packet
 623  *****************************************************************************/
 624 static __inline__ void input_SortPacket( input_thread_t *p_input,
 625                                          ts_packet_t *p_ts_packet )
 626 {
 627     int             i_current_pid;
 628     int             i_es_loop;
 629
 630     /* Verify that sync_byte, error_indicator and scrambling_control are
 631        what we expected. */
 632     if( !(p_ts_packet->buffer[0] == 0x47) || (p_ts_packet->buffer[1] & 0x80) ||
 633         (p_ts_packet->buffer[3] & 0xc0) )
 634     {
 635         intf_DbgMsg("input debug: invalid TS header (%p)\n", p_ts_packet);
 636     }
 637     else
 638     {
 639         /* Get the PID of the packet. Note that ntohs is needed, for endianness
 640            purposes (see man page). */
 641         i_current_pid = U16_AT(&p_ts_packet->buffer[1]) & 0x1fff;
 642
 643         //intf_DbgMsg("input debug: pid %d received (%p)\n",
 644         //            i_current_pid, p_ts_packet);
 645
 646         /* Lock current ES state. */
 647         vlc_mutex_lock( &p_input->es_lock );
 648
 649     /* Verify that we actually want this PID. */
 650         for( i_es_loop = 0; i_es_loop < INPUT_MAX_SELECTED_ES; i_es_loop++ )
 651         {
 652             if( p_input->pp_selected_es[i_es_loop] != NULL)
 653             {
 654                 if( (*p_input->pp_selected_es[i_es_loop]).i_id
 655                      == i_current_pid )
 656                 {
 657                     /* Don't need the lock anymore, since the value pointed
 658                        out by p_input->pp_selected_es[i_es_loop] can only be
 659                        modified from inside the input_thread (by the PSI
 660                        decoder): interface thread is only allowed to modify
 661                        the pp_selected_es table */
 662                     vlc_mutex_unlock( &p_input->es_lock );
 663
 664                     /* We're interested. Pass it to the demultiplexer. */
 665                     input_DemuxTS( p_input, p_ts_packet,
 666                                    p_input->pp_selected_es[i_es_loop] );
 667                     return;
 668                 }
 669             }
 670             else
 671             {
 672                 /* pp_selected_es should not contain any hole. */
 673                 break;
 674             }
 675         }
 676         vlc_mutex_unlock( &p_input->es_lock );
 677     }
 678
 679     /* We weren't interested in receiving this packet. Give it back to the
 680        netlist. */
 681     //intf_DbgMsg("SortPacket: freeing unwanted TS %p (pid %d)\n", p_ts_packet,
 682     //                 U16_AT(&p_ts_packet->buffer[1]) & 0x1fff);
 683     input_NetlistFreeTS( p_input, p_ts_packet );
 684 #ifdef STATS
 685     p_input->c_packets_trashed++;
 686 #endif
 687 }
 688
 689 /*****************************************************************************
 690  * input_DemuxTS: first step of demultiplexing: the TS header
 691  *****************************************************************************
 692  * Stream must also only contain PES and PSI, so PID must have been filtered
 693  *****************************************************************************/
 694 static __inline__ void input_DemuxTS( input_thread_t *p_input,
 695                                       ts_packet_t *p_ts_packet,
 696                                       es_descriptor_t *p_es_descriptor )
 697 {
 698     int         i_dummy;
 699     boolean_t   b_adaption;                     /* Adaption field is present */
 700     boolean_t   b_payload;                         /* Packet carries payload */
 701     boolean_t   b_unit_start;          /* A PSI or a PES start in the packet */
 702     boolean_t   b_trash = 0;                 /* Must the packet be trashed ? */
 703     boolean_t   b_lost = 0;                     /* Was there a packet lost ? */
 704
 705     ASSERT(p_input);
 706     ASSERT(p_ts_packet);
 707     ASSERT(p_es_descriptor);
 708
 709 #define p (p_ts_packet->buffer)
 710
 711     //intf_DbgMsg("input debug: TS-demultiplexing packet %p, pid %d, number %d\n",
 712     //            p_ts_packet, U16_AT(&p[1]) & 0x1fff, p[3] & 0x0f);
 713
 714 #ifdef STATS
 715     p_es_descriptor->c_packets++;
 716     p_es_descriptor->c_bytes += TS_PACKET_SIZE;
 717 #endif
 718
 719     /* Extract flags values from TS common header. */
 720     b_unit_start = (p[1] & 0x40);
 721     b_adaption = (p[3] & 0x20);
 722     b_payload = (p[3] & 0x10);
 723
 724     /* Extract adaption field informations if any */
 725     if( !b_adaption )
 726     {
 727         /* We don't have any adaptation_field, so payload start immediately
 728            after the 4 byte TS header */
 729         p_ts_packet->i_payload_start = 4;
 730     }
 731     else
 732     {
 733         /* p[4] is adaptation_field_length minus one */
 734         p_ts_packet->i_payload_start = 5 + p[4];
 735
 736         /* The adaption field can be limited to the adaptation_field_length byte,
 737            so that there is nothing to do: skip this possibility */
 738         if( p[4] )
 739         {
 740             /* If the packet has both adaptation_field and payload, adaptation_field
 741                cannot be more than 182 bytes long; if there is only an
 742                adaptation_field, it must fill the next 183 bytes. */
 743             if( b_payload ? (p[4] > 182) : (p[4] != 183) )
 744             {
 745                 intf_DbgMsg("input debug: invalid TS adaptation field (%p)\n",
 746                             p_ts_packet);
 747 #ifdef STATS
 748                 p_es_descriptor->c_invalid_packets++;
 749 #endif
 750                 b_trash = 1;
 751             }
 752
 753             /* No we are sure that the byte containing flags is present: read it */
 754             else
 755             {
 756                 /* discontinuity_indicator */
 757                 if( p[5] & 0x80 )
 758                 {
 759                     intf_DbgMsg("discontinuity_indicator encountered by TS demux " \
 760                                 "(position read: %d, saved: %d)\n", p[5] & 0x80,
 761                                 p_es_descriptor->i_continuity_counter);
 762
 763                     /* If the PID carries the PCR, there will be a system time-base
 764                        discontinuity. We let the PCR decoder handle that. */
 765                     p_es_descriptor->b_discontinuity = 1;
 766
 767                     /* There also may be a continuity_counter discontinuity:
 768                resynchronise our counter with the one of the stream */
 769                     p_es_descriptor->i_continuity_counter = (p[3] & 0x0f) - 1;
 770                 }
 771
 772                 /* random_access_indicator */
 773                 p_es_descriptor->b_random |= p[5] & 0x40;
 774
 775                 /* If this is a PCR_PID, and this TS packet contains a PCR,
 776            we pass it along to the PCR decoder. */
 777                 if( (p_es_descriptor->b_pcr) && (p[5] & 0x10) )
 778                 {
 779                     /* There should be a PCR field in the packet, check if the
 780                adaption field is long enough to carry it */
 781                     if( p[4] >= 7 )
 782                     {
 783                         /* Call the PCR decoder */
 784                         input_PcrDecode( p_input, p_es_descriptor, &p[6] );
 785                     }
 786                 }
 787             }
 788         }
 789     }
 790
 791     /* Check the continuity of the stream. */
 792     i_dummy = ((p[3] & 0x0f) - p_es_descriptor->i_continuity_counter) & 0x0f;
 793     if( i_dummy == 1 )
 794     {
 795         /* Everything is ok, just increase our counter */
 796         p_es_descriptor->i_continuity_counter++;
 797     }
 798     else
 799     {
 800         if( !b_payload && i_dummy == 0 )
 801         {
 802             /* This is a packet without payload, this is allowed by the draft
 803                As there is nothing interesting in this packet (except PCR that
 804                have already been handled), we can trash the packet. */
 805             intf_DbgMsg("Packet without payload received by TS demux\n");
 806             b_trash = 1;
 807         }
 808         else if( i_dummy <= 0 )
 809         {
 810             /* Duplicate packet: mark it as being to be trashed. */
 811             intf_DbgMsg("Duplicate packet received by TS demux\n");
 812             b_trash = 1;
 813         }
 814         else if( p_es_descriptor->i_continuity_counter == 0xFF )
 815         {
 816             /* This means that the packet is the first one we receive for this
 817                ES since the continuity counter ranges between 0 and 0x0F
 818                excepts when it has been initialized by the input: Init the
 819                counter to the correct value. */
 820             intf_DbgMsg("First packet for PID %d received by TS demux\n",
 821                         p_es_descriptor->i_id);
 822             p_es_descriptor->i_continuity_counter = (p[3] & 0x0f);
 823         }
 824         else
 825         {
 826             /* This can indicate that we missed a packet or that the
 827                continuity_counter wrapped and we received a dup packet: as we
 828                don't know, do as if we missed a packet to be sure to recover
 829                from this situation */
 830             intf_DbgMsg("Packet lost by TS demux: current %d, packet %d\n",
 831                         p_es_descriptor->i_continuity_counter & 0x0f,
 832                         p[3] & 0x0f);
 833             b_lost = 1;
 834             p_es_descriptor->i_continuity_counter = p[3] & 0x0f;
 835         }
 836     }
 837
 838     /* Trash the packet if it has no payload or if it is bad */
 839     if( b_trash )
 840     {
 841         input_NetlistFreeTS( p_input, p_ts_packet );
 842 #ifdef STATS
 843         p_input->c_packets_trashed++;
 844 #endif
 845     }
 846     else
 847     {
 848         if( p_es_descriptor->b_psi )
 849         {
 850             /* The payload contains PSI tables */
 851             input_DemuxPSI( p_input, p_ts_packet, p_es_descriptor,
 852                             b_unit_start, b_lost );
 853         }
 854         else
 855         {
 856             /* The payload carries a PES stream */
 857             input_DemuxPES( p_input, p_ts_packet, p_es_descriptor,
 858                             b_unit_start, b_lost );
 859         }
 860     }
 861
 862 #undef p
 863 }
 864
 865
 866
 867
 868 /*****************************************************************************
 869  * input_DemuxPES:
 870  *****************************************************************************
 871  * Gather a PES packet.
 872  *****************************************************************************/
 873 static __inline__ void input_DemuxPES( input_thread_t *p_input,
 874                                        ts_packet_t *p_ts_packet,
 875                                        es_descriptor_t *p_es_descriptor,
 876                                        boolean_t b_unit_start,
 877                                        boolean_t b_packet_lost )
 878 {
 879     int                         i_dummy;
 880     pes_packet_t*               p_last_pes;
 881     ts_packet_t *               p_ts;
 882     int                         i_ts_payload_size;
 883
 884
 885 #define p_pes (p_es_descriptor->p_pes_packet)
 886
 887     ASSERT(p_input);
 888     ASSERT(p_ts_packet);
 889     ASSERT(p_es_descriptor);
 890
 891     //intf_DbgMsg("PES-demultiplexing %p (%p)\n", p_ts_packet, p_pes);
 892
 893     /* If we lost data, discard the PES packet we are trying to reassemble
 894        if any and wait for the beginning of a new one in order to synchronise
 895        again */
 896     if( b_packet_lost && p_pes != NULL )
 897     {
 898         intf_DbgMsg("PES %p trashed because of packet lost\n", p_pes);
 899         input_NetlistFreePES( p_input, p_pes );
 900         p_pes = NULL;
 901     }
 902
 903     /* If the TS packet contains the begining of a new PES packet, and if we
 904        were reassembling a PES packet, then the PES should be complete now,
 905        so parse its header and give it to the decoders */
 906     if( b_unit_start && p_pes != NULL )
 907     {
 908         /* Parse the header. The header has a variable length, but in order
 909            to improve the algorithm, we will read the 14 bytes we may be
 910            interested in */
 911
 912         /* If this part of the header did not fit in the current TS packet,
 913            copy the part of the header we are interested in to the
 914            p_pes_header_save buffer. The buffer is dynamicly allocated if
 915            needed so it's time expensive but this situation almost never
 916            occurs. */
 917         p_ts = p_pes->p_first_ts;
 918         i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
 919
 920         if(i_ts_payload_size < PES_HEADER_SIZE)
 921         {
 922             intf_DbgMsg("Code never tested encountered, WARNING ! (benny)\n");
 923             if( !p_pes->p_pes_header_save )
 924             {
 925                 p_pes->p_pes_header_save = malloc(PES_HEADER_SIZE);
 926             }
 927
 928             i_dummy = 0;
 929             do
 930             {
 931                 memcpy(p_pes->p_pes_header_save + i_dummy,
 932                        &p_ts->buffer[p_ts->i_payload_start], i_ts_payload_size);
 933                 i_dummy += i_ts_payload_size;
 934
 935                 p_ts = p_ts->p_next_ts;
 936                 if(!p_ts)
 937                 {
 938                   /* The payload of the PES packet is shorter than the 14 bytes
 939                      we would read. This means that high packet lost occured
 940                      so the PES won't be useful for any decoder. Moreover,
 941                      this should never happen so we can trash the packet and
 942                      exit roughly without regrets */
 943                   intf_DbgMsg("PES packet too short: trashed\n");
 944                   input_NetlistFreePES( p_input, p_pes );
 945                   p_pes = NULL;
 946                   /* XXX: Stats */
 947                   return;
 948                 }
 949
 950                 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
 951             }
 952             while(i_ts_payload_size + i_dummy < PES_HEADER_SIZE);
 953
 954             /* This last TS packet is partly header, partly payload, so just
 955                copy the header part */
 956             memcpy(p_pes->p_pes_header_save + i_dummy,
 957                    &p_ts->buffer[p_ts->i_payload_start],
 958                    PES_HEADER_SIZE - i_dummy);
 959
 960             /* The header must be read in the buffer not in any TS packet */
 961             p_pes->p_pes_header = p_pes->p_pes_header_save;
 962
 963             /* Get the PES size if defined */
 964             if( (i_dummy = U16_AT(p_pes->p_pes_header + 4)) )
 965             {
 966                 p_pes->i_pes_real_size = i_dummy + 6;
 967             }
 968         }
 969
 970         /* Now we have the part of the PES header we were interested in:
 971            p_pes_header and i_pes_real_size ; we can parse it */
 972         input_ParsePES( p_input, p_es_descriptor );
 973     }
 974
 975     /* If we are at the beginning of a new PES packet, we must fetch a new
 976        PES buffer to begin with the reassembly of this PES packet. This is
 977        also here that we can synchronise with the stream if we we lost
 978        packets or if the decoder has just started */
 979     if( b_unit_start )
 980     {
 981         p_last_pes = p_pes;
 982
 983         /* Get a new one PES from the PES netlist. */
 984         if( (p_pes = input_NetlistGetPES( p_input )) == (NULL) )
 985         {
 986             /* PES netlist is empty ! */
 987             p_input->b_error = 1;
 988         }
 989         else
 990         {
 991             //intf_DbgMsg("New PES packet %p (first TS: %p)\n", p_pes, p_ts_packet);
 992
 993             /* Init the PES fields so that the first TS packet could be
 994              * correctly added to the PES packet (see below) */
 995             p_pes->p_first_ts = p_ts_packet;
 996             p_pes->p_last_ts = NULL;
 997
 998             /* If the last pes packet was null, this means that the
 999              * synchronization was lost and so warn the decoder that he
1000              * will have to find a way to recover */
1001             if( !p_last_pes )
1002                 p_pes->b_data_loss = 1;
1003
1004             /* Read the b_random_access flag status and then reinit it */
1005             p_pes->b_random_access = p_es_descriptor->b_random;
1006             p_es_descriptor->b_random = 0;
1007         }
1008
1009         /* If the PES header fits in the first TS packet, we can
1010          * already set p_pes->p_pes_header, and in all cases we
1011          * set p_pes->i_pes_real_size */
1012         if( p_ts_packet->i_payload_end - p_ts_packet->i_payload_start
1013                 >= PES_HEADER_SIZE )
1014         {
1015             p_pes->p_pes_header = &(p_ts_packet->buffer[p_ts_packet->i_payload_start]);
1016             if( (i_dummy = U16_AT(p_pes->p_pes_header + 4)) )
1017             {
1018                 p_pes->i_pes_real_size = i_dummy + 6;
1019             }
1020         }
1021     }
1022
1023
1024     /* If we are synchronized with the stream, and so if we are ready to
1025        receive correctly the data, add the TS packet to the current PES
1026        packet */
1027     if( p_pes != NULL )
1028     {
1029         //intf_DbgMsg("Adding TS %p to PES %p\n", p_ts_packet, p_pes);
1030
1031         /* Size of the payload carried in the TS packet */
1032         i_ts_payload_size = p_ts_packet->i_payload_end -
1033                             p_ts_packet->i_payload_start;
1034
1035         /* Update the relations between the TS packets */
1036         p_ts_packet->p_prev_ts = p_pes->p_last_ts;
1037         p_ts_packet->p_next_ts = NULL;
1038         if( p_pes->i_ts_packets != 0 )
1039         {
1040             /* Regarder si il serait pas plus efficace de ne creer que
1041              * les liens precedent->suivant pour le moment, et les
1042              * liens suivant->precedent quand le paquet est termine */
1043             /* Otherwise it is the first TS packet. */
1044             p_pes->p_last_ts->p_next_ts = p_ts_packet;
1045         }
1046         /* Now add the TS to the PES packet */
1047         p_pes->p_last_ts = p_ts_packet;
1048         p_pes->i_ts_packets++;
1049         p_pes->i_pes_size += i_ts_payload_size;
1050
1051         /* Stats */
1052 #ifdef STATS
1053         i_dummy = p_ts_packet->i_payload_end - p_ts_packet->i_payload_start;
1054         p_es_descriptor->c_payload_bytes += i_dummy;
1055 #endif
1056
1057         /* We can check if the packet is finished */
1058         if( p_pes->i_pes_size == p_pes->i_pes_real_size )
1059         {
1060             /* The packet is finished, parse it */
1061             input_ParsePES( p_input, p_es_descriptor );
1062
1063             /* Tell the Demux we have parsed this PES, no need to redo it */
1064             p_pes = NULL;
1065         }
1066     }
1067     else
1068     {
1069         /* Since we don't use the TS packet to build a PES packet, we don't
1070            need it anymore, so give it back to the netlist */
1071         //intf_DbgMsg("Trashing TS %p: no PES being build\n", p_ts_packet);
1072         input_NetlistFreeTS( p_input, p_ts_packet );
1073     }
1074
1075 #undef p_pes
1076 }
1077
1078
1079
1080 /*****************************************************************************
1081  * input_ParsePES
1082  *****************************************************************************
1083  * Parse a finished PES packet and analyze its header.
1084  *****************************************************************************/
1085 static __inline__ void input_ParsePES( input_thread_t *p_input,
1086                                        es_descriptor_t *p_es_descriptor )
1087 {
1088     decoder_fifo_t *            p_fifo;
1089     u8                          i_pes_header_size;
1090     ts_packet_t *               p_ts;
1091     int                         i_ts_payload_size;
1092
1093
1094 #define p_pes (p_es_descriptor->p_pes_packet)
1095
1096     //intf_DbgMsg("End of PES packet %p\n", p_pes);
1097
1098     /* First read the 6 header bytes common to all PES packets:
1099        use them to test the PES validity */
1100     if( (p_pes->p_pes_header[0] || p_pes->p_pes_header[1] ||
1101         (p_pes->p_pes_header[2] != 1)) ||
1102                                  /* packet_start_code_prefix != 0x000001 */
1103         ((p_pes->i_pes_real_size) &&
1104          (p_pes->i_pes_real_size != p_pes->i_pes_size)) )
1105                /* PES_packet_length is set and != total received payload */
1106     {
1107       /* Trash the packet and set p_pes to NULL to be sure the next PES
1108          packet will have its b_data_lost flag set */
1109       intf_DbgMsg("Corrupted PES packet (size doesn't match) : trashed\n");
1110       input_NetlistFreePES( p_input, p_pes );
1111       p_pes = NULL;
1112       /* Stats XXX?? */
1113     }
1114     else
1115     {
1116         /* The PES packet is valid. Check its type to test if it may
1117            carry additional informations in a header extension */
1118         p_pes->i_stream_id =  p_pes->p_pes_header[3];
1119
1120         switch( p_pes->i_stream_id )
1121         {
1122         case 0xBE:  /* Padding */
1123         case 0xBC:  /* Program stream map */
1124         case 0xBF:  /* Private stream 2 */
1125         case 0xB0:  /* ECM */
1126         case 0xB1:  /* EMM */
1127         case 0xFF:  /* Program stream directory */
1128         case 0xF2:  /* DSMCC stream */
1129         case 0xF8:  /* ITU-T H.222.1 type E stream */
1130             /* The payload begins immediatly after the 6 bytes header, so
1131                we have finished with the parsing */
1132             i_pes_header_size = 6;
1133             break;
1134
1135         default:
1136             switch( p_pes->p_pes_header[8] & 0xc0 )
1137             {
1138               case 0x80: /* MPEG2: 10xx xxxx */
1139               case 0x00: /* FIXME: This shouldn't be allowed !! */
1140                 /* The PES header contains at least 3 more bytes: parse them */
1141                 p_pes->b_data_alignment = p_pes->p_pes_header[6] & 0x04;
1142                 p_pes->b_has_pts = p_pes->p_pes_header[7] & 0x80;
1143                 i_pes_header_size = p_pes->p_pes_header[8] + 9;
1144
1145                 /* Now parse the optional header extensions (in the limit of
1146                    the 14 bytes */
1147                 if( p_pes->b_has_pts )
1148                 {
1149                     pcr_descriptor_t * p_pcr;
1150
1151                     p_pcr = p_input->p_pcr;
1152
1153                     p_pes->i_pts =
1154                         ( ((mtime_t)(p_pes->p_pes_header[9] & 0x0E) << 29) |
1155                           (((mtime_t)U16_AT(p_pes->p_pes_header + 10) << 14) - (1 << 14)) |
1156                           ((mtime_t)U16_AT(p_pes->p_pes_header + 12) >> 1) ) * 300;
1157                     p_pes->i_pts /= 27;
1158
1159                     if( p_pcr->i_synchro_state )
1160                     {
1161                         switch( p_pcr->i_synchro_state )
1162                         {
1163                             case SYNCHRO_NOT_STARTED:
1164                                 p_pes->b_has_pts = 0;
1165                                 break;
1166
1167                             case SYNCHRO_START:
1168                                 p_pes->i_pts += p_pcr->delta_pcr;
1169                                 p_pcr->delta_absolute = mdate() - p_pes->i_pts + INPUT_PTS_DELAY;
1170                                 p_pes->i_pts += p_pcr->delta_absolute;
1171                                 p_pcr->i_synchro_state = 0;
1172                                 break;
1173
1174                             case SYNCHRO_REINIT: /* We skip a PES */
1175                                 p_pes->b_has_pts = 0;
1176                                 p_pcr->i_synchro_state = SYNCHRO_START;
1177                                 break;
1178                         }
1179                     }
1180                     else
1181                     {
1182                         p_pes->i_pts += p_pcr->delta_pcr + p_pcr->delta_absolute;
1183                     }
1184                 }
1185                 break;
1186
1187             default: /* MPEG1 or some strange thing */
1188                 /* since this isn't supported yet, we certainly gonna crash */
1189                 intf_ErrMsg( "FIXME: unknown PES type %.2x\n",
1190                              p_pes->p_pes_header[8] );
1191                 i_pes_header_size = 6;
1192                 break;
1193
1194             }
1195             break;
1196         }
1197
1198         /* Now we've parsed the header, we just have to indicate in some
1199          * specific TS packets where the PES payload begins (renumber
1200          * i_payload_start), so that the decoders can find the beginning
1201          * of their data right out of the box. */
1202         p_ts = p_pes->p_first_ts;
1203         i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
1204         while( i_pes_header_size > i_ts_payload_size )
1205         {
1206             /* These packets are entirely filled by the PES header. */
1207             i_pes_header_size -= i_ts_payload_size;
1208             p_ts->i_payload_start = p_ts->i_payload_end;
1209             /* Go to the next TS packet: here we won't have to test it is
1210              * not NULL because we trash the PES packets when packet lost
1211              * occurs */
1212             p_ts = p_ts->p_next_ts;
1213             i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
1214         }
1215         /* This last packet is partly header, partly payload. */
1216         p_ts->i_payload_start += i_pes_header_size;
1217
1218
1219         /* Now we can eventually put the PES packet in the decoder's
1220          * PES fifo */
1221         switch( p_es_descriptor->i_type )
1222         {
1223             case MPEG1_VIDEO_ES:
1224             case MPEG2_VIDEO_ES:
1225                 p_fifo = &(((vpar_thread_t*)(p_es_descriptor->p_dec))->fifo);
1226                 break;
1227
1228             case MPEG1_AUDIO_ES:
1229             case MPEG2_AUDIO_ES:
1230                 p_fifo = &(((adec_thread_t*)(p_es_descriptor->p_dec))->fifo);
1231                 break;
1232
1233             case AC3_AUDIO_ES:
1234                 p_fifo = &(((ac3dec_thread_t *)(p_es_descriptor->p_dec))->fifo);
1235                 break;
1236
1237             case LPCM_AUDIO_ES:
1238                 p_fifo = &(((lpcmdec_thread_t *)(p_es_descriptor->p_dec))->fifo);
1239                 break;
1240
1241             case DVD_SPU_ES:
1242                 /* we skip the first byte at the beginning of the
1243                  * subpicture payload, it only contains the SPU ID. */
1244                 p_ts->i_payload_start++;
1245                 p_fifo = &(((spudec_thread_t *)(p_es_descriptor->p_dec))->fifo);
1246                 break;
1247
1248             default:
1249                 /* This should never happen */
1250                 intf_DbgMsg("Unknown stream type (%d, %d): PES trashed\n",
1251                     p_es_descriptor->i_id, p_es_descriptor->i_type);
1252                 p_fifo = NULL;
1253                 break;
1254         }
1255
1256         if( p_fifo != NULL )
1257         {
1258             vlc_mutex_lock( &p_fifo->data_lock );
1259             if( DECODER_FIFO_ISFULL( *p_fifo ) )
1260             {
1261                 /* The FIFO is full !!! This should not happen. */
1262 #ifdef STATS
1263                 p_input->c_packets_trashed += p_pes->i_ts_packets;
1264                 p_es_descriptor->c_invalid_packets += p_pes->i_ts_packets;
1265 #endif
1266                 input_NetlistFreePES( p_input, p_pes );
1267                 intf_DbgMsg("PES trashed - fifo full ! (%d, %d)\n",
1268                            p_es_descriptor->i_id, p_es_descriptor->i_type);
1269             }
1270         else
1271             {
1272                 //intf_DbgMsg("Putting %p into fifo %p/%d\n",
1273                 //            p_pes, p_fifo, p_fifo->i_end);
1274                 p_fifo->buffer[p_fifo->i_end] = p_pes;
1275                 DECODER_FIFO_INCEND( *p_fifo );
1276
1277                 /* Warn the decoder that it's got work to do. */
1278                 vlc_cond_signal( &p_fifo->data_wait );
1279             }
1280             vlc_mutex_unlock( &p_fifo->data_lock );
1281         }
1282         else
1283         {
1284             intf_DbgMsg("No fifo to receive PES %p: trash\n", p_pes);
1285 #ifdef STATS
1286             p_input->c_packets_trashed += p_pes->i_ts_packets;
1287             p_es_descriptor->c_invalid_packets += p_pes->i_ts_packets;
1288 #endif
1289             input_NetlistFreePES( p_input, p_pes );
1290         }
1291     }
1292 #undef p_pes
1293 }
1294
1295
1296
1297 /*****************************************************************************
1298  * input_DemuxPSI:
1299  *****************************************************************************
1300  * Notice that current ES state has been locked by input_SortPacket.
1301  * (No more true, changed by benny - FIXME: See if it's ok, and definitely
1302  * change the code ?? )
1303  *****************************************************************************/
1304 static __inline__ void input_DemuxPSI( input_thread_t *p_input,
1305                                        ts_packet_t *p_ts_packet,
1306                                        es_descriptor_t *p_es_descriptor,
1307                                        boolean_t b_unit_start, boolean_t b_packet_lost )
1308 {
1309     int i_data_offset;    /* Offset of the interesting data in the TS packet */
1310     u16 i_data_length;                               /* Length of those data */
1311   //boolean_t b_first_section;         /* another section in the TS packet ? */
1312
1313     ASSERT(p_input);
1314     ASSERT(p_ts_packet);
1315     ASSERT(p_es_descriptor);
1316
1317 #define p_psi (p_es_descriptor->p_psi_section)
1318
1319     //intf_DbgMsg( "input debug: PSI demultiplexing %p (%p)\n", p_ts_packet, p_input);
1320
1321     //intf_DbgMsg( "Packet: %x %x %x %x %x %x %x %x %x %x %x %x %x %x %x %x %x %x %x %x %x (unit start: %d)\n", p_ts_packet->buffer[p_ts_packet->i_payload_start], p_ts_packet->buffer[p_ts_packet->i_payload_start+1], p_ts_packet->buffer[p_ts_packet->i_payload_start+2], p_ts_packet->buffer[p_ts_packet->i_payload_start+3], p_ts_packet->buffer[p_ts_packet->i_payload_start+4], p_ts_packet->buffer[p_ts_packet->i_payload_start+5], p_ts_packet->buffer[p_ts_packet->i_payload_start+6], p_ts_packet->buffer[p_ts_packet->i_payload_start+7], p_ts_packet->buffer[p_ts_packet->i_payload_start+8], p_ts_packet->buffer[p_ts_packet->i_payload_start+9], p_ts_packet->buffer[p_ts_packet->i_payload_start+10], p_ts_packet->buffer[p_ts_packet->i_payload_start+11], p_ts_packet->buffer[p_ts_packet->i_payload_start+12], p_ts_packet->buffer[p_ts_packet->i_payload_start+13], p_ts_packet->buffer[p_ts_packet->i_payload_start+14], p_ts_packet->buffer[p_ts_packet->i_payload_start+15], p_ts_packet->buffer[p_ts_packet->i_payload_start+16], p_ts_packet->buffer[p_ts_packet->i_payload_start+17], p_ts_packet->buffer[p_ts_packet->i_payload_start+18], p_ts_packet->buffer[p_ts_packet->i_payload_start+19], p_ts_packet->buffer[p_ts_packet->i_payload_start+20], b_unit_start);
1322
1323
1324     /* Try to find the beginning of the payload in the packet to initialise
1325      * the do-while loop that follows -> Compute the i_data_offset variable:
1326      * by default, the value is set so that we won't enter in the while loop.
1327      * It will be set to a correct value if the data are not corrupted */
1328     i_data_offset = TS_PACKET_SIZE;
1329
1330     /* Has the reassembly of a section already begun in a previous packet ? */
1331     if( p_psi->b_running_section )
1332     {
1333         /* Was data lost since the last TS packet ? */
1334         if( b_packet_lost )
1335         {
1336             /* Discard the packet and wait for the begining of a new one
1337              * to resynch */
1338             p_psi->b_running_section = 0;
1339             p_psi->i_current_position = 0;
1340             intf_DbgMsg( "PSI section(s) discarded due to packet loss\n" );
1341         }
1342         else
1343         {
1344             /* The data that complete a previously began section are always at
1345              * the beginning of the TS payload... */
1346             i_data_offset = p_ts_packet->i_payload_start;
1347             /* ...Unless there is a pointer field, that we have to bypass */
1348             if( b_unit_start )
1349                 i_data_offset++;
1350             //intf_DbgMsg( "New part of the section received at offset %d\n", i_data_offset );
1351         }
1352     }
1353     /* We are looking for the beginning of a new section */
1354     else
1355     {
1356         if( b_unit_start )
1357         {
1358             /* Get the offset at which the data for that section can be found
1359              * The offset is stored in the pointer_field since we are
1360              * interested in the first section of the TS packet. Note that
1361              * the +1 is to bypass the pointer field */
1362             i_data_offset = p_ts_packet->i_payload_start +
1363                             p_ts_packet->buffer[p_ts_packet->i_payload_start] + 1;
1364             //intf_DbgMsg( "New section beginning at offset %d in TS packet\n", i_data_offset );
1365         }
1366         else
1367         {
1368             /* This may either mean that the TS is bad or that the packet
1369              * contains the end of a section that had been discarded in a
1370              * previous loop: trash the TS packet since we cannot do
1371              * anything with those data: */
1372             p_psi->b_running_section = 0;
1373             p_psi->i_current_position = 0;
1374             intf_DbgMsg( "PSI packet discarded due to lack of synchronisation\n" );
1375         }
1376     }
1377
1378     /* The section we will deal with during the first iteration of the
1379      * following loop is the first one contained in the TS packet */
1380     //    b_first_section = 1;
1381
1382     /* Reassemble the pieces of sections contained in the TS packet and
1383      * decode the sections that could have been completed.
1384      * Stop when we reach the end of the packet or stuffing bytes */
1385     while( i_data_offset < TS_PACKET_SIZE && p_ts_packet->buffer[i_data_offset] != 0xFF )
1386     {
1387         /* If the current section is a new one, reinit the data fields of
1388          * the p_psi struct to start its decoding */
1389         if( !p_psi->b_running_section )
1390         {
1391             /* Read the length of the new section */
1392             p_psi->i_length = (U16_AT(&p_ts_packet->buffer[i_data_offset+1]) & 0xFFF) + 3;
1393             //intf_DbgMsg( "Section length %d\n", p_psi->i_length );
1394             if( p_psi->i_length > PSI_SECTION_SIZE )
1395             {
1396                 /* The TS packet is corrupted, stop here to avoid possible
1397                  * a seg fault */
1398                 intf_DbgMsg( "PSI Section size is too big, aborting its reception\n" );
1399                 break;
1400             }
1401
1402             /* Init the reassembly of that section */
1403             p_psi->b_running_section = 1;
1404             p_psi->i_current_position = 0;
1405         }
1406
1407       /* Compute the length of data related to the section in this TS packet */
1408         if( p_psi->i_length - p_psi->i_current_position > TS_PACKET_SIZE - i_data_offset)
1409             i_data_length = TS_PACKET_SIZE - i_data_offset;
1410         else
1411           i_data_length = p_psi->i_length - p_psi->i_current_position;
1412
1413         /* Copy those data in the section buffer */
1414         memcpy( &p_psi->buffer[p_psi->i_current_position], &p_ts_packet->buffer[i_data_offset],
1415                 i_data_length );
1416
1417         /* Interesting data are now after the ones we copied, since no gap is
1418          * allowed between 2 sections in a TS packets */
1419         i_data_offset += i_data_length;
1420
1421         /* Decode the packet if it is now complete */
1422         if (p_psi->i_length == p_psi->i_current_position + i_data_length)
1423         {
1424             /* Packet is complete, decode it */
1425             //intf_DbgMsg( "SECTION COMPLETE: starting decoding of its data\n" );
1426             input_PsiDecode( p_input, p_psi );
1427
1428             /* Prepare the buffer to receive a new section */
1429             p_psi->i_current_position = 0;
1430             p_psi->b_running_section = 0;
1431
1432             /* The new section won't be the first anymore */
1433             //b_first_section = 0;
1434         }
1435         else
1436         {
1437             /* Prepare the buffer to receive the next part of the section */
1438           p_psi->i_current_position += i_data_length;
1439           //intf_DbgMsg( "Section not complete, waiting for the end\n" );
1440         }
1441
1442         //intf_DbgMsg( "Must loop ? Next data offset: %d, stuffing: %d\n",
1443         //             i_data_offset, p_ts_packet->buffer[i_data_offset] );
1444     }
1445
1446     /* Relase the TS packet, we don't need it anymore */
1447     input_NetlistFreeTS( p_input, p_ts_packet );
1448
1449 #undef p_psi
1450 }
1451