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