1 /*******************************************************************************
2 * input.c: input thread
4 *******************************************************************************
5 * Read an MPEG2 stream, demultiplex and parse it before sending it to
7 ******************************************************************************/
9 /*******************************************************************************
11 *******************************************************************************/
20 #include <X11/extensions/XShm.h>
21 #include <sys/soundcard.h>
25 #include <sys/ioctl.h>
27 #include <netinet/in.h>
32 #include "vlc_thread.h"
37 #include "input_psi.h"
38 #include "input_pcr.h"
39 #include "input_netlist.h"
40 #include "decoder_fifo.h"
41 #include "input_file.h"
42 #include "input_network.h"
44 #include "audio_output.h"
45 #include "audio_decoder.h"
48 #include "video_output.h"
49 #include "video_decoder.h"
52 /******************************************************************************
54 ******************************************************************************/
55 static void RunThread ( input_thread_t *p_input );
56 static void ErrorThread ( input_thread_t *p_input );
57 static void EndThread ( input_thread_t *p_input );
59 static __inline__ int input_ReadPacket( input_thread_t *p_input );
60 static __inline__ void input_SortPacket( input_thread_t *p_input,
61 ts_packet_t *ts_packet );
62 static __inline__ void input_DemuxTS( input_thread_t *p_input,
63 ts_packet_t *ts_packet,
64 es_descriptor_t *es_descriptor );
65 static __inline__ void input_DemuxPES( input_thread_t *p_input,
66 ts_packet_t *ts_packet,
67 es_descriptor_t *p_es_descriptor,
68 boolean_t b_unit_start, boolean_t b_packet_lost );
69 static __inline__ void input_DemuxPSI( input_thread_t *p_input,
70 ts_packet_t *ts_packet,
71 es_descriptor_t *p_es_descriptor,
72 boolean_t b_unit_start, boolean_t b_packet_lost );
74 /*******************************************************************************
75 * input_CreateThread: creates a new input thread
76 *******************************************************************************
77 * This function creates a new input, and returns a pointer
78 * to its description. On error, it returns NULL.
79 * If pi_status is NULL, then the function will block until the thread is ready.
80 * If not, it will be updated using one of the THREAD_* constants.
81 *******************************************************************************/
82 input_thread_t *input_CreateThread ( int i_method, char *psz_source, int i_port, int i_vlan,
83 p_vout_thread_t p_vout, p_aout_thread_t p_aout, int *pi_status )
85 input_thread_t * p_input; /* thread descriptor */
86 int i_status; /* thread status */
87 int i_index; /* index for tables initialization */
89 /* Allocate descriptor */
91 p_input = (input_thread_t *)malloc( sizeof(input_thread_t) );
94 intf_ErrMsg("error: %s\n", strerror(ENOMEM));
98 /* Initialize thread properties */
100 p_input->b_error = 0;
101 p_input->pi_status = (pi_status != NULL) ? pi_status : &i_status;
102 *p_input->pi_status = THREAD_CREATE;
104 /* Initialize input method description */
105 p_input->i_method = i_method;
106 p_input->psz_source = psz_source;
107 p_input->i_port = i_port;
108 p_input->i_vlan = i_vlan;
111 case INPUT_METHOD_TS_FILE: /* file methods */
112 p_input->p_Open = input_FileOpen;
113 p_input->p_Read = input_FileRead;
114 p_input->p_Close = input_FileClose;
116 case INPUT_METHOD_TS_VLAN_BCAST: /* vlan network method */
117 if( !p_main->b_vlans )
119 intf_ErrMsg("error: vlans are not activated\n");
123 /* ... pass through */
124 case INPUT_METHOD_TS_UCAST: /* network methods */
125 case INPUT_METHOD_TS_MCAST:
126 case INPUT_METHOD_TS_BCAST:
127 p_input->p_Open = input_NetworkOpen;
128 p_input->p_Read = input_NetworkRead;
129 p_input->p_Close = input_NetworkClose;
132 intf_ErrMsg("error: unknow input method\n");
138 /* Initialize stream description */
139 for( i_index = 0; i_index < INPUT_MAX_ES; i_index++ )
141 p_input->p_es[i_index].i_id = EMPTY_PID;
144 /* Initialize default settings for spawned decoders */
145 p_input->p_aout = p_aout;
146 p_input->p_vout = p_vout;
149 /* Initialize statistics */
150 p_input->c_loops = 0;
151 p_input->c_bytes = 0;
152 p_input->c_payload_bytes = 0;
153 p_input->c_packets_read = 0;
154 p_input->c_packets_trashed = 0;
157 /* Initialize PSI and PCR decoders */
158 if( input_PsiInit( p_input ) )
164 if( input_PcrInit( p_input ) )
166 input_PsiEnd( p_input );
171 /* Initialize netlists */
172 if( input_NetlistInit( p_input ) )
174 input_PsiEnd( p_input );
175 input_PcrEnd( p_input );
180 intf_DbgMsg("configuration: method=%d, source=%s, port=%d, vlan=%d\n",
181 i_method, psz_source, i_port, i_vlan );
183 /* Let the appropriate method open the socket. */
184 if( p_input->p_Open( p_input ) )
186 input_NetlistEnd( p_input );
187 input_PsiEnd( p_input );
188 input_PcrEnd( p_input );
193 /* Create thread and set locks. */
194 vlc_mutex_init( &p_input->netlist.lock );
195 vlc_mutex_init( &p_input->programs_lock );
196 vlc_mutex_init( &p_input->es_lock );
197 if( vlc_thread_create(&p_input->thread_id, "input", (void *) RunThread, (void *) p_input) )
199 intf_ErrMsg("error: %s\n", strerror(errno) );
200 p_input->p_Close( p_input );
201 input_NetlistEnd( p_input );;
202 input_PsiEnd( p_input );
203 input_PcrEnd( p_input );
208 intf_Msg("Input initialized\n");
210 /* If status is NULL, wait until the thread is created */
211 if( pi_status == NULL )
215 msleep( THREAD_SLEEP );
216 }while( (i_status != THREAD_READY) && (i_status != THREAD_ERROR)
217 && (i_status != THREAD_FATAL) );
218 if( i_status != THREAD_READY )
226 /******************************************************************************
227 * input_DestroyThread: mark an input thread as zombie
228 ******************************************************************************
229 * This function should not return until the thread is effectively cancelled.
230 ******************************************************************************/
231 void input_DestroyThread( input_thread_t *p_input, int *pi_status )
233 int i_status; /* thread status */
236 p_input->pi_status = (pi_status != NULL) ? pi_status : &i_status;
237 *p_input->pi_status = THREAD_DESTROY;
239 /* Request thread destruction */
242 /* If status is NULL, wait until thread has been destroyed */
243 if( pi_status == NULL )
247 msleep( THREAD_SLEEP );
248 }while( (i_status != THREAD_OVER) && (i_status != THREAD_ERROR)
249 && (i_status != THREAD_FATAL) );
254 /*******************************************************************************
255 * input_OpenAudioStream: open an audio stream
256 *******************************************************************************
257 * This function spawns an audio decoder and plugs it on the audio output
259 *******************************************************************************/
260 int input_OpenAudioStream( input_thread_t *p_input, int i_id )
265 /*******************************************************************************
266 * input_CloseAudioStream: close an audio stream
267 *******************************************************************************
268 * This function destroys an audio decoder.
269 *******************************************************************************/
270 void input_CloseAudioStream( input_thread_t *p_input, int i_id )
275 /*******************************************************************************
276 * input_OpenVideoStream: open a video stream
277 *******************************************************************************
278 * This function spawns a video decoder and plugs it on a video output thread.
279 *******************************************************************************/
280 int input_OpenVideoStream( input_thread_t *p_input,
281 struct vout_thread_s *p_vout, struct video_cfg_s * p_cfg )
286 /*******************************************************************************
287 * input_CloseVideoStream: close a video stream
288 *******************************************************************************
289 * This function destroys an video decoder.
290 *******************************************************************************/
291 void input_CloseVideoStream( input_thread_t *p_input, int i_id )
297 /* following functions are local */
299 /*******************************************************************************
300 * InitThread: initialize input thread
301 *******************************************************************************
302 * This function is called from RunThread and performs the second step of the
303 * initialization. It returns 0 on success. Note that the thread's flag are not
304 * modified inside this function.
305 *******************************************************************************/
306 static int InitThread( input_thread_t *p_input )
308 /* Mark thread as running and return */
310 *p_input->pi_status = THREAD_READY;
311 intf_DbgMsg("thread ready\n");
315 /*******************************************************************************
316 * RunThread: main thread loop
317 *******************************************************************************
318 * Thread in charge of processing the network packets and demultiplexing.
319 *******************************************************************************/
320 static void RunThread( input_thread_t *p_input )
323 * Initialize thread and free configuration
325 p_input->b_error = InitThread( p_input );
326 if( p_input->b_error )
328 free( p_input ); /* destroy descriptor */
336 while( !p_input->b_die && !p_input->b_error )
338 /* Scatter read the UDP packet from the network or the file. */
339 if( (input_ReadPacket( p_input )) == (-1) )
341 /* ??? Normally, a thread can't kill itself, but we don't have
342 * any method in case of an error condition ... */
343 p_input->b_error = 1;
354 if( p_input->b_error )
356 ErrorThread( p_input );
360 EndThread( p_input );
361 intf_DbgMsg("thread end\n");
365 /******************************************************************************
366 * ErrorThread: RunThread() error loop
367 *******************************************************************************
368 * This function is called when an error occured during thread main's loop.
369 ******************************************************************************/
370 static void ErrorThread( input_thread_t *p_input )
372 /* Wait until a `die' order */
374 while( !p_input->b_die )
377 msleep( VOUT_IDLE_SLEEP );
381 /*******************************************************************************
382 * EndThread: end the input thread
383 *******************************************************************************/
384 static void EndThread( input_thread_t * p_input )
386 int * pi_status; /* threas status */
387 int i_es_loop; /* es index */
391 pi_status = p_input->pi_status;
392 *pi_status = THREAD_END;
394 /* Close input method */
395 p_input->p_Close( p_input );
397 /* Destroy all decoder threads */
399 (i_es_loop < INPUT_MAX_ES) && (p_input->pp_selected_es[i_es_loop] != NULL) ;
402 switch( p_input->pp_selected_es[i_es_loop]->i_type )
407 vdec_DestroyThread( (vdec_thread_t*)(p_input->pp_selected_es[i_es_loop]->p_dec) /*, NULL */ );
409 vpar_DestroyThread( (vpar_thread_t*)(p_input->pp_selected_es[i_es_loop]->p_dec) /*, NULL */ );
414 adec_DestroyThread( (adec_thread_t*)(p_input->pp_selected_es[i_es_loop]->p_dec) );
417 ac3dec_DestroyThread( (ac3dec_thread_t *)(p_input->pp_selected_es[i_es_loop]->p_dec) );
421 intf_DbgMsg("error: unknown decoder type %d\n", p_input->pp_selected_es[i_es_loop]->i_type );
427 input_NetlistEnd( p_input ); /* clean netlist */
428 input_PsiEnd( p_input ); /* clean PSI information */
429 input_PcrEnd( p_input ); /* clean PCR information */
430 free( p_input ); /* free input_thread structure */
433 *pi_status = THREAD_OVER;
436 /*******************************************************************************
437 * input_ReadPacket: reads a packet from the network or the file
438 *******************************************************************************/
439 static __inline__ int input_ReadPacket( input_thread_t *p_input )
441 int i_base_index; /* index of the first free iovec */
444 #ifdef INPUT_LIFO_TS_NETLIST
445 int i_meanwhile_released;
446 int i_currently_removed;
448 ts_packet_t * p_ts_packet;
450 /* In this function, we only care about the TS netlist. PES netlist
451 * is for the demultiplexer. */
452 #ifdef INPUT_LIFO_TS_NETLIST
453 i_base_index = p_input->netlist.i_ts_index;
455 /* Verify that we still have packets in the TS netlist */
456 if( (INPUT_MAX_TS + INPUT_TS_READ_ONCE - 1 - p_input->netlist.i_ts_index) <= INPUT_TS_READ_ONCE )
458 intf_ErrMsg("input error: TS netlist is empty !\n");
462 #else /* FIFO netlist */
463 i_base_index = p_input->netlist.i_ts_start;
464 if( p_input->netlist.i_ts_start + INPUT_TS_READ_ONCE -1 > INPUT_MAX_TS )
466 /* The netlist is splitted in 2 parts. We must gather them to consolidate
467 the FIFO (we make the loop easily in having the same iovec at the far
468 end and in the beginning of netlist_free).
469 That's why the netlist is (INPUT_MAX_TS +1) + (INPUT_TS_READ_ONCE -1)
471 memcpy( p_input->netlist.p_ts_free + INPUT_MAX_TS + 1,
472 p_input->netlist.p_ts_free,
473 (p_input->netlist.i_ts_start + INPUT_TS_READ_ONCE - 1 - INPUT_MAX_TS)
474 * sizeof(struct iovec) );
477 /* Verify that we still have packets in the TS netlist */
478 if( ((p_input->netlist.i_ts_end -1 - p_input->netlist.i_ts_start) & INPUT_MAX_TS) <= INPUT_TS_READ_ONCE )
480 intf_ErrMsg("input error: TS netlist is empty !\n");
483 #endif /* FIFO netlist */
485 /* Scatter read the buffer. */
486 i_packet_size = (*p_input->p_Read)( p_input,
487 &p_input->netlist.p_ts_free[i_base_index],
488 INPUT_TS_READ_ONCE );
489 if( i_packet_size == (-1) )
491 // intf_DbgMsg("Read packet %d %p %d %d\n", i_base_index,
492 // &p_input->netlist.p_ts_free[i_base_index],
493 // p_input->netlist.i_ts_start,
494 // p_input->netlist.i_ts_end);
495 intf_ErrMsg("input error: readv() failed (%s)\n", strerror(errno));
499 if( i_packet_size == 0 )
501 /* No packet has been received, so stop here. */
505 /* Demultiplex the TS packets (1..INPUT_TS_READ_ONCE) received. */
506 for( i_current_index = i_base_index;
507 (i_packet_size -= TS_PACKET_SIZE) >= 0;
510 /* BTW, something REALLY bad could happen if we receive packets with
512 p_ts_packet = (ts_packet_t*)(p_input->netlist.p_ts_free[i_current_index].iov_base);
513 /* Don't cry :-), we are allowed to do that cast, because initially,
514 our buffer was malloc'ed with sizeof(ts_packet_t) */
516 /* Find out if we need this packet and demultiplex. */
517 input_SortPacket( p_input /* for current PIDs and netlist */,
521 if( i_packet_size > 0 )
523 intf_ErrMsg("input error: wrong size\n");
527 /* Remove the TS packets we have just filled from the netlist */
528 #ifdef INPUT_LIFO_TS_NETLIST
529 /* We need to take a lock here while we're calculating index positions. */
530 vlc_mutex_lock( &p_input->netlist.lock );
532 i_meanwhile_released = i_base_index - p_input->netlist.i_ts_index;
533 if( i_meanwhile_released )
535 /* That's where it becomes funny :-). Since we didn't take locks for
536 efficiency reasons, other threads (including ourselves, with
537 input_DemuxPacket) might have released packets to the netlist.
538 So we have to copy these iovec where they should go.
540 BTW, that explains why the TS netlist is
541 (INPUT_MAX_TS +1) + (TS_READ_ONCE -1) large. */
543 i_currently_removed = i_current_index - i_base_index;
544 if( i_meanwhile_released < i_currently_removed )
546 /* Copy all iovecs in that case */
547 memcpy( &p_input->netlist.p_ts_free[p_input->netlist.i_ts_index]
548 + i_currently_removed,
549 &p_input->netlist.p_ts_free[p_input->netlist.i_ts_index],
550 i_meanwhile_released * sizeof(struct iovec) );
554 /* We have fewer places than items, so we only move
555 i_currently_removed of them. */
556 memcpy( &p_input->netlist.p_ts_free[i_base_index],
557 &p_input->netlist.p_ts_free[p_input->netlist.i_ts_index],
558 i_currently_removed * sizeof(struct iovec) );
561 /* Update i_netlist_index with the information gathered above. */
562 p_input->netlist.i_ts_index += i_currently_removed;
566 /* Nothing happened. */
567 p_input->netlist.i_ts_index = i_current_index;
570 vlc_mutex_unlock( &p_input->netlist.lock );
572 #else /* FIFO netlist */
573 /* & is modulo ; that's where we make the loop. */
574 p_input->netlist.i_ts_start = i_current_index & INPUT_MAX_TS;
578 p_input->c_packets_read += i_current_index - i_base_index;
579 p_input->c_bytes += (i_current_index - i_base_index) * TS_PACKET_SIZE;
584 /*******************************************************************************
585 * input_SortPacket: find out whether we need that packet
586 *******************************************************************************/
587 static __inline__ void input_SortPacket( input_thread_t *p_input,
588 ts_packet_t *p_ts_packet )
593 /* Verify that sync_byte, error_indicator and scrambling_control are
595 if( !(p_ts_packet->buffer[0] == 0x47) || (p_ts_packet->buffer[1] & 0x80) ||
596 (p_ts_packet->buffer[3] & 0xc0) )
598 intf_DbgMsg("input debug: invalid TS header (%p)\n", p_ts_packet);
602 /* Get the PID of the packet. Note that ntohs is needed, for endianness
603 purposes (see man page). */
604 i_current_pid = U16_AT(&p_ts_packet->buffer[1]) & 0x1fff;
606 // intf_DbgMsg("input debug: pid %d received (%p)\n",
607 // i_current_pid, p_ts_packet);
609 /* Lock current ES state. */
610 vlc_mutex_lock( &p_input->es_lock );
612 /* Verify that we actually want this PID. */
613 for( i_es_loop = 0; i_es_loop < INPUT_MAX_SELECTED_ES; i_es_loop++ )
615 if( p_input->pp_selected_es[i_es_loop] != NULL)
617 if( (*p_input->pp_selected_es[i_es_loop]).i_id
620 /* Don't need the lock anymore, since the value pointed
621 out by p_input->pp_selected_es[i_es_loop] can only be
622 modified from inside the input_thread (by the PSI
623 decoder): interface thread is only allowed to modify
624 the pp_selected_es table */
625 vlc_mutex_unlock( &p_input->es_lock );
627 /* We're interested. Pass it to the demultiplexer. */
628 input_DemuxTS( p_input, p_ts_packet,
629 p_input->pp_selected_es[i_es_loop] );
635 /* pp_selected_es should not contain any hole. */
639 vlc_mutex_unlock( &p_input->es_lock );
642 /* We weren't interested in receiving this packet. Give it back to the
644 // intf_DbgMsg("SortPacket: freeing unwanted TS %p (pid %d)\n", p_ts_packet,
645 // U16_AT(&p_ts_packet->buffer[1]) & 0x1fff);
646 input_NetlistFreeTS( p_input, p_ts_packet );
648 p_input->c_packets_trashed++;
652 /*******************************************************************************
653 * input_DemuxTS: first step of demultiplexing: the TS header
654 *******************************************************************************
655 * Stream must also only contain PES and PSI, so PID must have been filtered
656 *******************************************************************************/
657 static __inline__ void input_DemuxTS( input_thread_t *p_input,
658 ts_packet_t *p_ts_packet,
659 es_descriptor_t *p_es_descriptor )
662 boolean_t b_adaption; /* Adaption field is present */
663 boolean_t b_payload; /* Packet carries payload */
664 boolean_t b_unit_start; /* A PSI or a PES start in the packet */
665 boolean_t b_trash = 0; /* Must the packet be trashed ? */
666 boolean_t b_lost = 0; /* Was there a packet lost ? */
670 ASSERT(p_es_descriptor);
672 #define p (p_ts_packet->buffer)
674 // intf_DbgMsg("input debug: TS-demultiplexing packet %p, pid %d, number %d\n",
675 // p_ts_packet, U16_AT(&p[1]) & 0x1fff, p[3] & 0x0f);
678 p_es_descriptor->c_packets++;
679 p_es_descriptor->c_bytes += TS_PACKET_SIZE;
682 /* Extract flags values from TS common header. */
683 b_unit_start = (p[1] & 0x40);
684 b_adaption = (p[3] & 0x20);
685 b_payload = (p[3] & 0x10);
687 /* Extract adaption field informations if any */
690 /* We don't have any adaptation_field, so payload start immediately
691 after the 4 byte TS header */
692 p_ts_packet->i_payload_start = 4;
696 /* p[4] is adaptation_field_length minus one */
697 p_ts_packet->i_payload_start = 5 + p[4];
699 /* The adaption field can be limited to the adaptation_field_length byte,
700 so that there is nothing to do: skip this possibility */
703 /* If the packet has both adaptation_field and payload, adaptation_field
704 cannot be more than 182 bytes long; if there is only an
705 adaptation_field, it must fill the next 183 bytes. */
706 if( b_payload ? (p[4] > 182) : (p[4] != 183) )
708 intf_DbgMsg("input debug: invalid TS adaptation field (%p)\n",
711 p_es_descriptor->c_invalid_packets++;
716 /* No we are sure that the byte containing flags is present: read it */
719 /* discontinuity_indicator */
722 intf_DbgMsg("discontinuity_indicator encountered by TS demux " \
723 "(position read: %d, saved: %d)\n", p[5] & 0x80,
724 p_es_descriptor->i_continuity_counter);
726 /* If the PID carries the PCR, there will be a system time-base
727 discontinuity. We let the PCR decoder handle that. */
728 p_es_descriptor->b_discontinuity = 1;
730 /* There also may be a continuity_counter discontinuity:
731 resynchronise our counter with the one of the stream */
732 p_es_descriptor->i_continuity_counter = (p[3] & 0x0f) - 1;
735 /* random_access_indicator */
736 p_es_descriptor->b_random |= p[5] & 0x40;
738 /* If this is a PCR_PID, and this TS packet contains a PCR,
739 we pass it along to the PCR decoder. */
740 if( (p_es_descriptor->b_pcr) && (p[5] & 0x10) )
742 /* There should be a PCR field in the packet, check if the
743 adaption field is long enough to carry it */
746 /* Call the PCR decoder */
747 input_PcrDecode( p_input, p_es_descriptor, &p[6] );
754 /* Check the continuity of the stream. */
755 i_dummy = ((p[3] & 0x0f) - p_es_descriptor->i_continuity_counter) & 0x0f;
758 /* Everything is ok, just increase our counter */
759 p_es_descriptor->i_continuity_counter++;
763 if( !b_payload && i_dummy == 0 )
765 /* This is a packet without payload, this is allowed by the draft
766 As there is nothing interessant in this packet (except PCR that
767 have already been handled), we can trash the packet. */
768 intf_DbgMsg("Packet without payload received by TS demux\n");
771 else if( i_dummy <= 0 )
773 /* Duplicate packet: mark it as being to be trashed. */
774 intf_DbgMsg("Duplicate packet received by TS demux\n");
777 else if( p_es_descriptor->i_continuity_counter == 0xFF )
779 /* This means that the packet is the first one we receive for this
780 ES since the continuity counter ranges between 0 and 0x0F
781 excepts when it has been initialized by the input: Init the
782 counter to the correct value. */
783 intf_DbgMsg("First packet for PID %d received by TS demux\n",
784 p_es_descriptor->i_id);
785 p_es_descriptor->i_continuity_counter = (p[3] & 0x0f);
789 /* This can indicate that we missed a packet or that the
790 continuity_counter wrapped and we received a dup packet: as we
791 don't know, do as if we missed a packet to be sure to recover
792 from this situation */
793 intf_DbgMsg("Packet lost by TS demux: current %d, packet %d\n",
794 p_es_descriptor->i_continuity_counter & 0x0f,
797 p_es_descriptor->i_continuity_counter = p[3] & 0x0f;
801 /* Trash the packet if it has no payload or if it is bad */
804 input_NetlistFreeTS( p_input, p_ts_packet );
806 p_input->c_packets_trashed++;
811 if( p_es_descriptor->b_psi )
813 /* The payload contains PSI tables */
814 input_DemuxPSI( p_input, p_ts_packet, p_es_descriptor,
815 b_unit_start, b_lost );
819 /* The payload carries a PES stream */
820 input_DemuxPES( p_input, p_ts_packet, p_es_descriptor,
821 b_unit_start, b_lost );
831 /*******************************************************************************
833 *******************************************************************************
834 * Gather a PES packet and analyzes its header.
835 *******************************************************************************/
836 static __inline__ void input_DemuxPES( input_thread_t *p_input,
837 ts_packet_t *p_ts_packet,
838 es_descriptor_t *p_es_descriptor,
839 boolean_t b_unit_start,
840 boolean_t b_packet_lost )
842 decoder_fifo_t * p_fifo;
843 u8 i_pes_header_size;
845 pes_packet_t* p_last_pes;
847 int i_ts_payload_size;
850 #define p_pes (p_es_descriptor->p_pes_packet)
854 ASSERT(p_es_descriptor);
856 // intf_DbgMsg("PES-demultiplexing %p (%p)\n", p_ts_packet, p_pes);
858 /* If we lost data, discard the PES packet we are trying to reassemble
859 if any and wait for the beginning of a new one in order to synchronise
861 if( b_packet_lost && p_pes != NULL )
863 intf_DbgMsg("PES %p trashed because of packet lost\n", p_pes);
864 input_NetlistFreePES( p_input, p_pes );
868 /* If the TS packet contains the begining of a new PES packet, and if we
869 were reassembling a PES packet, then the PES should be complete now,
870 so parse its header and give it to the decoders */
871 if( b_unit_start && p_pes != NULL )
873 // intf_DbgMsg("End of PES packet %p\n", p_pes);
875 /* Parse the header. The header has a variable length, but in order
876 to improve the algorithm, we will read the 14 bytes we may be
878 p_ts = p_pes->p_first_ts;
879 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
882 if(i_ts_payload_size >= PES_HEADER_SIZE)
884 /* This part of the header entirely fits in the payload of
885 the first TS packet */
886 p_pes->p_pes_header = &(p_ts->buffer[p_ts->i_payload_start]);
890 /* This part of the header does not fit in the current TS packet:
891 copy the part of the header we are interested in to the
892 p_pes_header_save buffer. The buffer is dynamicly allocated if
893 needed so it's time expensive but this situation almost never
895 intf_DbgMsg("Code never tested encountered, WARNING ! (benny)\n");
896 if( !p_pes->p_pes_header_save )
897 p_pes->p_pes_header_save = malloc(PES_HEADER_SIZE);
901 memcpy(p_pes->p_pes_header_save + i_dummy,
902 &p_ts->buffer[p_ts->i_payload_start], i_ts_payload_size);
903 i_dummy += i_ts_payload_size;
905 p_ts = p_ts->p_next_ts;
908 /* The payload of the PES packet is shorter than the 14 bytes
909 we would read. This means that high packet lost occured
910 so the PES won't be usefull for any decoder. Moreover,
911 this should never happen so we can trash the packet and
912 exit roughly without regrets */
913 intf_DbgMsg("PES packet too short: trashed\n");
914 input_NetlistFreePES( p_input, p_pes );
920 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
922 while(i_ts_payload_size + i_dummy < PES_HEADER_SIZE);
924 /* This last TS packet is partly header, partly payload, so just
925 copy the header part */
926 memcpy(p_pes->p_pes_header_save + i_dummy,
927 &p_ts->buffer[p_ts->i_payload_start],
928 PES_HEADER_SIZE - i_dummy);
930 /* The header must be read in the buffer not in any TS packet */
931 p_pes->p_pes_header = p_pes->p_pes_header_save;
934 /* Now we have the part of the PES header we were interested in:
937 /* First read the 6 header bytes common to all PES packets:
938 use them to test the PES validity */
939 if( (p_pes->p_pes_header[0] || p_pes->p_pes_header[1] ||
940 (p_pes->p_pes_header[2] != 1)) ||
941 /* packet_start_code_prefix != 0x000001 */
942 ((i_dummy = U16_AT(p_pes->p_pes_header + 4)) &&
943 (i_dummy + 6 != p_pes->i_pes_size)) )
944 /* PES_packet_length is set and != total received payload */
946 /* Trash the packet and set p_pes to NULL to be sure the next PES
947 packet will have its b_data_lost flag set */
948 intf_DbgMsg("Corrupted PES packet received: trashed\n");
949 input_NetlistFreePES( p_input, p_pes );
955 /* The PES packet is valid. Check its type to test if it may
956 carry additional informations in a header extension */
957 p_pes->i_stream_id = p_pes->p_pes_header[3];
959 switch( p_pes->i_stream_id )
961 case 0xBE: /* Padding */
962 case 0xBC: /* Program stream map */
963 case 0xBF: /* Private stream 2 */
966 case 0xFF: /* Program stream directory */
967 case 0xF2: /* DSMCC stream */
968 case 0xF8: /* ITU-T H.222.1 type E stream */
969 /* The payload begins immediatly after the 6 bytes header, so
970 we have finished with the parsing */
971 i_pes_header_size = 6;
975 /* The PES header contains at least 3 more bytes: parse them */
976 p_pes->b_data_alignment = p_pes->p_pes_header[6] & 0x04;
977 p_pes->b_has_pts = p_pes->p_pes_header[7] & 0x80;
978 i_pes_header_size = p_pes->p_pes_header[8] + 9;
980 /* Now parse the optional header extensions (in the limit of
982 if( p_pes->b_has_pts )
984 pcr_descriptor_t * p_pcr;
986 p_pcr = p_input->p_pcr;
989 ( ((mtime_t)(p_pes->p_pes_header[9] & 0x0E) << 29) |
990 (((mtime_t)U16_AT(p_pes->p_pes_header + 10) << 14) - (1 << 14)) |
991 ((mtime_t)U16_AT(p_pes->p_pes_header + 12) >> 1) ) * 300;
994 if( p_pcr->i_synchro_state )
996 switch( p_pcr->i_synchro_state )
998 case SYNCHRO_NOT_STARTED:
999 p_pes->b_has_pts = 0;
1003 p_pes->i_pts += p_pcr->delta_pcr;
1004 p_pcr->delta_absolute = mdate() - p_pes->i_pts + 500000;
1005 p_pes->i_pts += p_pcr->delta_absolute;
1006 p_pcr->i_synchro_state = 0;
1009 case SYNCHRO_REINIT: /* We skip a PES */
1010 p_pes->b_has_pts = 0;
1011 p_pcr->i_synchro_state = SYNCHRO_START;
1017 p_pes->i_pts += p_pcr->delta_pcr + p_pcr->delta_absolute;
1023 /* Now we've parsed the header, we just have to indicate in some
1024 specific TS packets where the PES payload begins (renumber
1025 i_payload_start), so that the decoders can find the beginning
1026 of their data right out of the box. */
1027 p_ts = p_pes->p_first_ts;
1028 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
1029 while( i_pes_header_size > i_ts_payload_size )
1031 /* These packets are entirely filled by the PES header. */
1032 i_pes_header_size -= i_ts_payload_size;
1033 p_ts->i_payload_start = p_ts->i_payload_end;
1034 /* Go to the next TS packet: here we won't have to test it is
1035 not NULL because we trash the PES packets when packet lost
1037 p_ts = p_ts->p_next_ts;
1038 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
1040 /* This last packet is partly header, partly payload. */
1041 p_ts->i_payload_start += i_pes_header_size;
1044 /* Now we can eventually put the PES packet in the decoder's
1046 switch( p_es_descriptor->i_type )
1048 case MPEG1_VIDEO_ES:
1049 case MPEG2_VIDEO_ES:
1051 p_fifo = &(((vdec_thread_t*)(p_es_descriptor->p_dec))->fifo);
1053 p_fifo = &(((vpar_thread_t*)(p_es_descriptor->p_dec))->fifo);
1057 case MPEG1_AUDIO_ES:
1058 case MPEG2_AUDIO_ES:
1059 p_fifo = &(((adec_thread_t*)(p_es_descriptor->p_dec))->fifo);
1063 /* we skip 4 bytes at the beginning of the AC3 payload */
1064 p_ts->i_payload_start += 4;
1065 p_fifo = &(((ac3dec_thread_t *)(p_es_descriptor->p_dec))->fifo);
1069 /* This should never happen */
1070 intf_DbgMsg("Unknown stream type (%d, %d): PES trashed\n",
1071 p_es_descriptor->i_id, p_es_descriptor->i_type);
1076 if( p_fifo != NULL )
1078 vlc_mutex_lock( &p_fifo->data_lock );
1079 if( DECODER_FIFO_ISFULL( *p_fifo ) )
1081 /* The FIFO is full !!! This should not happen. */
1083 p_input->c_packets_trashed += p_pes->i_ts_packets;
1084 p_es_descriptor->c_invalid_packets += p_pes->i_ts_packets;
1086 input_NetlistFreePES( p_input, p_pes );
1087 intf_DbgMsg("PES trashed - fifo full ! (%d, %d)\n",
1088 p_es_descriptor->i_id, p_es_descriptor->i_type);
1092 // intf_DbgMsg("Putting %p into fifo %p/%d\n",
1093 // p_pes, p_fifo, p_fifo->i_end);
1094 p_fifo->buffer[p_fifo->i_end] = p_pes;
1095 DECODER_FIFO_INCEND( *p_fifo );
1097 /* Warn the decoder that it's got work to do. */
1098 vlc_cond_signal( &p_fifo->data_wait );
1100 vlc_mutex_unlock( &p_fifo->data_lock );
1104 intf_DbgMsg("No fifo to receive PES %p: trash\n", p_pes);
1106 p_input->c_packets_trashed += p_pes->i_ts_packets;
1107 p_es_descriptor->c_invalid_packets += p_pes->i_ts_packets;
1109 input_NetlistFreePES( p_input, p_pes );
1115 /* If we are at the beginning of a new PES packet, we must fetch a new
1116 PES buffer to begin with the reassembly of this PES packet. This is
1117 also here that we can synchronise with the stream if we we lost
1118 packets or if the decoder has just started */
1123 /* Get a new one PES from the PES netlist. */
1124 if( (p_pes = input_NetlistGetPES( p_input )) == (NULL) )
1126 /* PES netlist is empty ! */
1127 p_input->b_error = 1;
1131 // intf_DbgMsg("New PES packet %p (first TS: %p)\n", p_pes, p_ts_packet);
1133 /* Init the PES fields so that the first TS packet could be correctly
1134 added to the PES packet (see below) */
1135 p_pes->p_first_ts = p_ts_packet;
1136 p_pes->p_last_ts = NULL;
1138 /* If the last pes packet was null, this means that the synchronisation
1139 was lost and so warn the decoder that he will have to find a way to
1142 p_pes->b_data_loss = 1;
1144 /* Read the b_random_access flag status and then reinit it */
1145 p_pes->b_random_access = p_es_descriptor->b_random;
1146 p_es_descriptor->b_random = 0;
1151 /* If we are synchronised with the stream, and so if we are ready to
1152 receive correctly the data, add the TS packet to the current PES
1156 // intf_DbgMsg("Adding TS %p to PES %p\n", p_ts_packet, p_pes);
1158 /* Size of the payload carried in the TS packet */
1159 i_ts_payload_size = p_ts_packet->i_payload_end -
1160 p_ts_packet->i_payload_start;
1162 /* Update the relations between the TS packets */
1163 p_ts_packet->p_prev_ts = p_pes->p_last_ts;
1164 p_ts_packet->p_next_ts = NULL;
1165 if( p_pes->i_ts_packets != 0 )
1167 /* Regarder si il serait pas plus efficace de ne creer que les liens
1168 precedent->suivant pour le moment, et les liens suivant->precedent
1169 quand le paquet est termine */
1170 /* Otherwise it is the first TS packet. */
1171 p_pes->p_last_ts->p_next_ts = p_ts_packet;
1173 /* Now add the TS to the PES packet */
1174 p_pes->p_last_ts = p_ts_packet;
1175 p_pes->i_ts_packets++;
1176 p_pes->i_pes_size += i_ts_payload_size;
1180 i_dummy = p_ts_packet->i_payload_end - p_ts_packet->i_payload_start;
1181 p_es_descriptor->c_payload_bytes += i_dummy;
1186 /* Since we don't use the TS packet to build a PES packet, we don't
1187 need it anymore, so give it back to the netlist */
1188 // intf_DbgMsg("Trashing TS %p: no PES being build\n", p_ts_packet);
1189 input_NetlistFreeTS( p_input, p_ts_packet );
1198 /*******************************************************************************
1200 *******************************************************************************
1201 * Notice that current ES state has been locked by input_SortPacket. (No more true,
1202 * changed by benny - See if it's ok, and definitely change the code ???????? )
1203 *******************************************************************************/
1204 static __inline__ void input_DemuxPSI( input_thread_t *p_input,
1205 ts_packet_t *p_ts_packet,
1206 es_descriptor_t *p_es_descriptor,
1207 boolean_t b_unit_start, boolean_t b_packet_lost )
1209 int i_data_offset; /* Offset of the interesting data in the TS packet */
1210 u16 i_data_length; /* Length of those data */
1211 //boolean_t b_first_section; /* Was there another section in the TS packet ? */
1214 ASSERT(p_ts_packet);
1215 ASSERT(p_es_descriptor);
1217 #define p_psi (p_es_descriptor->p_psi_section)
1219 // intf_DbgMsg( "input debug: PSI demultiplexing %p (%p)\n", p_ts_packet, p_input);
1221 // 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);
1224 /* Try to find the beginning of the payload in the packet to initialise
1225 the do-while loop that follows -> Compute the i_data_offset variable:
1226 by default, the value is set so that we won't enter in the while loop.
1227 It will be set to a correct value if the data are not corrupted */
1228 i_data_offset = TS_PACKET_SIZE;
1230 /* Has the reassembly of a section already began in a previous packet ? */
1231 if( p_psi->b_running_section )
1233 /* Was data lost since the last TS packet ? */
1236 /* Discard the packet and wait for the begining of a new one to resynch */
1237 p_psi->b_running_section = 0;
1238 p_psi->i_current_position = 0;
1239 intf_DbgMsg( "PSI section(s) discarded due to packet loss\n" );
1243 /* The data that complete a previously began section are always at
1244 the beginning of the TS payload... */
1245 i_data_offset = p_ts_packet->i_payload_start;
1246 /* ...Unless there is a pointer field, that we have to bypass */
1249 // intf_DbgMsg( "New part of the section received at offset %d\n", i_data_offset );
1252 /* We are looking for the beginning of a new section */
1257 /* Get the offset at which the data for that section can be found
1258 The offset is stored in the pointer_field since we are interested in
1259 the first section of the TS packet. Note that the +1 is to bypass
1260 the pointer field */
1261 i_data_offset = p_ts_packet->i_payload_start +
1262 p_ts_packet->buffer[p_ts_packet->i_payload_start] + 1;
1263 // intf_DbgMsg( "New section beginning at offset %d in TS packet\n", i_data_offset );
1267 /* This may either mean that the TS is bad or that the packet contains
1268 the end of a section that had been discarded in a previous loop:
1269 trash the TS packet since we cannot do anything with those data: */
1270 p_psi->b_running_section = 0;
1271 p_psi->i_current_position = 0;
1272 intf_DbgMsg( "PSI packet discarded due to lack of synchronisation\n" );
1276 /* The section we will deal with during the first iteration of the following
1277 loop is the first one contained in the TS packet */
1278 // b_first_section = 1;
1280 /* Reassemble the pieces of sections contained in the TS packet and decode
1281 the sections that could have been completed.
1282 Stop when we reach the end of the packet or stuffing bytes */
1283 while( i_data_offset < TS_PACKET_SIZE && p_ts_packet->buffer[i_data_offset] != 0xFF )
1285 /* If the current section is a new one, reinit the data fields of the p_psi
1286 struct to start its decoding */
1287 if( !p_psi->b_running_section )
1289 /* Read the length of the new section */
1290 p_psi->i_length = (U16_AT(&p_ts_packet->buffer[i_data_offset+1]) & 0xFFF) + 3;
1291 // intf_DbgMsg( "Section length %d\n", p_psi->i_length );
1292 if( p_psi->i_length > PSI_SECTION_SIZE )
1294 /* The TS packet is corrupted, stop here to avoid possible a seg fault */
1295 intf_DbgMsg( "PSI Section size is too big, aborting its reception\n" );
1299 /* Init the reassembly of that section */
1300 p_psi->b_running_section = 1;
1301 p_psi->i_current_position = 0;
1304 /* Compute the length of data related to the section in this TS packet */
1305 if( p_psi->i_length - p_psi->i_current_position > TS_PACKET_SIZE - i_data_offset)
1306 i_data_length = TS_PACKET_SIZE - i_data_offset;
1308 i_data_length = p_psi->i_length - p_psi->i_current_position;
1310 /* Copy those data in the section buffer */
1311 memcpy( &p_psi->buffer[p_psi->i_current_position], &p_ts_packet->buffer[i_data_offset],
1314 /* Interesting data are now after the ones we copied, since no gap is
1315 allowed between 2 sections in a TS packets */
1316 i_data_offset += i_data_length;
1318 /* Decode the packet if it is now complete */
1319 if (p_psi->i_length == p_psi->i_current_position + i_data_length)
1321 /* Packet is complete, decode it */
1322 // intf_DbgMsg( "SECTION COMPLETE: starting decoding of its data\n" );
1323 input_PsiDecode( p_input, p_psi );
1325 /* Prepare the buffer to receive a new section */
1326 p_psi->i_current_position = 0;
1327 p_psi->b_running_section = 0;
1329 /* The new section won't be the first anymore */
1330 //b_first_section = 0;
1334 /* Prepare the buffer to receive the next part of the section */
1335 p_psi->i_current_position += i_data_length;
1336 // intf_DbgMsg( "Section not complete, waiting for the end\n" );
1339 // intf_DbgMsg( "Must loop ? Next data offset: %d, stuffing: %d\n",
1340 // i_data_offset, p_ts_packet->buffer[i_data_offset] );
1343 /* Relase the TS packet, we don't need it anymore */
1344 input_NetlistFreeTS( p_input, p_ts_packet );