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