1 /*******************************************************************************
2 * input.c: input thread
4 *******************************************************************************
5 * Read an MPEG2 stream, demultiplex and parse it before sending it to
7 ******************************************************************************/
9 /*******************************************************************************
11 *******************************************************************************/
21 #include <sys/ioctl.h>
23 #include <netinet/in.h>
28 #include "vlc_thread.h"
33 #include "input_psi.h"
34 #include "input_pcr.h"
35 #include "input_netlist.h"
36 #include "decoder_fifo.h"
37 #include "input_file.h"
38 #include "input_network.h"
40 #include "audio_output.h"
41 #include "audio_decoder.h"
44 #include "video_output.h"
45 #include "video_decoder.h"
48 /******************************************************************************
50 ******************************************************************************/
51 static void RunThread ( input_thread_t *p_input );
52 static void ErrorThread ( input_thread_t *p_input );
53 static void EndThread ( input_thread_t *p_input );
55 static __inline__ int input_ReadPacket( input_thread_t *p_input );
56 static __inline__ void input_SortPacket( input_thread_t *p_input,
57 ts_packet_t *ts_packet );
58 static __inline__ void input_DemuxTS( input_thread_t *p_input,
59 ts_packet_t *ts_packet,
60 es_descriptor_t *es_descriptor );
61 static __inline__ void input_DemuxPES( input_thread_t *p_input,
62 ts_packet_t *ts_packet,
63 es_descriptor_t *p_es_descriptor,
64 boolean_t b_unit_start, boolean_t b_packet_lost );
65 static __inline__ void input_DemuxPSI( 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 );
70 /*******************************************************************************
71 * input_CreateThread: creates a new input thread
72 *******************************************************************************
73 * This function creates a new input, and returns a pointer
74 * to its description. On error, it returns NULL.
75 * If pi_status is NULL, then the function will block until the thread is ready.
76 * If not, it will be updated using one of the THREAD_* constants.
77 *******************************************************************************/
78 input_thread_t *input_CreateThread ( int i_method, char *psz_source, int i_port, int i_vlan,
79 p_vout_thread_t p_vout, p_aout_thread_t p_aout, int *pi_status )
81 input_thread_t * p_input; /* thread descriptor */
82 int i_status; /* thread status */
83 int i_index; /* index for tables initialization */
85 /* Allocate descriptor */
87 p_input = (input_thread_t *)malloc( sizeof(input_thread_t) );
90 intf_ErrMsg("error: %s\n", strerror(ENOMEM));
94 /* Initialize thread properties */
97 p_input->pi_status = (pi_status != NULL) ? pi_status : &i_status;
98 *p_input->pi_status = THREAD_CREATE;
100 /* Initialize input method description */
101 p_input->i_method = i_method;
102 p_input->psz_source = psz_source;
103 p_input->i_port = i_port;
104 p_input->i_vlan = i_vlan;
107 case INPUT_METHOD_TS_FILE: /* file methods */
108 p_input->p_Open = input_FileOpen;
109 p_input->p_Read = input_FileRead;
110 p_input->p_Close = input_FileClose;
112 case INPUT_METHOD_TS_VLAN_BCAST: /* vlan network method */
113 if( !p_main->b_vlans )
115 intf_ErrMsg("error: vlans are not activated\n");
119 /* ... pass through */
120 case INPUT_METHOD_TS_UCAST: /* network methods */
121 case INPUT_METHOD_TS_MCAST:
122 case INPUT_METHOD_TS_BCAST:
123 p_input->p_Open = input_NetworkOpen;
124 p_input->p_Read = input_NetworkRead;
125 p_input->p_Close = input_NetworkClose;
128 intf_ErrMsg("error: unknow input method\n");
134 /* Initialize stream description */
135 for( i_index = 0; i_index < INPUT_MAX_ES; i_index++ )
137 p_input->p_es[i_index].i_id = EMPTY_PID;
138 p_input->pp_selected_es[i_index] = NULL;
141 /* Initialize default settings for spawned decoders */
142 p_input->p_aout = p_aout;
143 p_input->p_vout = p_vout;
146 /* Initialize statistics */
147 p_input->c_loops = 0;
148 p_input->c_bytes = 0;
149 p_input->c_payload_bytes = 0;
150 p_input->c_packets_read = 0;
151 p_input->c_packets_trashed = 0;
154 /* Initialize PSI and PCR decoders */
155 if( input_PsiInit( p_input ) )
161 if( input_PcrInit( p_input ) )
163 input_PsiEnd( p_input );
168 /* Initialize netlists */
169 if( input_NetlistInit( p_input ) )
171 input_PsiEnd( p_input );
172 input_PcrEnd( p_input );
177 intf_DbgMsg("configuration: method=%d, source=%s, port=%d, vlan=%d\n",
178 i_method, psz_source, i_port, i_vlan );
180 /* Let the appropriate method open the socket. */
181 if( p_input->p_Open( p_input ) )
183 input_NetlistEnd( p_input );
184 input_PsiEnd( p_input );
185 input_PcrEnd( p_input );
190 /* Create thread and set locks. */
191 vlc_mutex_init( &p_input->netlist.lock );
192 vlc_mutex_init( &p_input->programs_lock );
193 vlc_mutex_init( &p_input->es_lock );
194 if( vlc_thread_create(&p_input->thread_id, "input", (void *) RunThread, (void *) p_input) )
196 intf_ErrMsg("error: %s\n", strerror(errno) );
197 p_input->p_Close( p_input );
198 input_NetlistEnd( p_input );;
199 input_PsiEnd( p_input );
200 input_PcrEnd( p_input );
205 intf_Msg("Input initialized\n");
207 /* If status is NULL, wait until the thread is created */
208 if( pi_status == NULL )
212 msleep( THREAD_SLEEP );
213 }while( (i_status != THREAD_READY) && (i_status != THREAD_ERROR)
214 && (i_status != THREAD_FATAL) );
215 if( i_status != THREAD_READY )
223 /******************************************************************************
224 * input_DestroyThread: mark an input thread as zombie
225 ******************************************************************************
226 * This function should not return until the thread is effectively cancelled.
227 ******************************************************************************/
228 void input_DestroyThread( input_thread_t *p_input, int *pi_status )
230 int i_status; /* thread status */
233 p_input->pi_status = (pi_status != NULL) ? pi_status : &i_status;
234 *p_input->pi_status = THREAD_DESTROY;
236 /* Request thread destruction */
239 /* If status is NULL, wait until thread has been destroyed */
240 if( pi_status == NULL )
244 msleep( THREAD_SLEEP );
245 }while( (i_status != THREAD_OVER) && (i_status != THREAD_ERROR)
246 && (i_status != THREAD_FATAL) );
251 /*******************************************************************************
252 * input_OpenAudioStream: open an audio stream
253 *******************************************************************************
254 * This function spawns an audio decoder and plugs it on the audio output
256 *******************************************************************************/
257 int input_OpenAudioStream( input_thread_t *p_input, int i_id )
262 /*******************************************************************************
263 * input_CloseAudioStream: close an audio stream
264 *******************************************************************************
265 * This function destroys an audio decoder.
266 *******************************************************************************/
267 void input_CloseAudioStream( input_thread_t *p_input, int i_id )
272 /*******************************************************************************
273 * input_OpenVideoStream: open a video stream
274 *******************************************************************************
275 * This function spawns a video decoder and plugs it on a video output thread.
276 *******************************************************************************/
277 int input_OpenVideoStream( input_thread_t *p_input,
278 struct vout_thread_s *p_vout, struct video_cfg_s * p_cfg )
283 /*******************************************************************************
284 * input_CloseVideoStream: close a video stream
285 *******************************************************************************
286 * This function destroys an video decoder.
287 *******************************************************************************/
288 void input_CloseVideoStream( input_thread_t *p_input, int i_id )
294 /* following functions are local */
296 /*******************************************************************************
297 * InitThread: initialize input thread
298 *******************************************************************************
299 * This function is called from RunThread and performs the second step of the
300 * initialization. It returns 0 on success. Note that the thread's flag are not
301 * modified inside this function.
302 *******************************************************************************/
303 static int InitThread( input_thread_t *p_input )
305 /* Mark thread as running and return */
307 *p_input->pi_status = THREAD_READY;
308 intf_DbgMsg("thread ready\n");
312 /*******************************************************************************
313 * RunThread: main thread loop
314 *******************************************************************************
315 * Thread in charge of processing the network packets and demultiplexing.
316 *******************************************************************************/
317 static void RunThread( input_thread_t *p_input )
320 * Initialize thread and free configuration
322 p_input->b_error = InitThread( p_input );
323 if( p_input->b_error )
325 free( p_input ); /* destroy descriptor */
333 while( !p_input->b_die && !p_input->b_error )
335 /* Scatter read the UDP packet from the network or the file. */
336 if( (input_ReadPacket( p_input )) == (-1) )
338 /* ??? Normally, a thread can't kill itself, but we don't have
339 * any method in case of an error condition ... */
340 p_input->b_error = 1;
351 if( p_input->b_error )
353 ErrorThread( p_input );
357 EndThread( p_input );
358 intf_DbgMsg("thread end\n");
362 /******************************************************************************
363 * ErrorThread: RunThread() error loop
364 *******************************************************************************
365 * This function is called when an error occured during thread main's loop.
366 ******************************************************************************/
367 static void ErrorThread( input_thread_t *p_input )
369 /* Wait until a `die' order */
371 while( !p_input->b_die )
374 msleep( VOUT_IDLE_SLEEP );
378 /*******************************************************************************
379 * EndThread: end the input thread
380 *******************************************************************************/
381 static void EndThread( input_thread_t * p_input )
383 int * pi_status; /* threas status */
384 int i_es_loop; /* es index */
388 pi_status = p_input->pi_status;
389 *pi_status = THREAD_END;
391 /* Close input method */
392 p_input->p_Close( p_input );
394 /* Destroy all decoder threads */
396 (i_es_loop < INPUT_MAX_ES) && (p_input->pp_selected_es[i_es_loop] != NULL) ;
399 switch( p_input->pp_selected_es[i_es_loop]->i_type )
404 vdec_DestroyThread( (vdec_thread_t*)(p_input->pp_selected_es[i_es_loop]->p_dec) /*, NULL */ );
406 vpar_DestroyThread( (vpar_thread_t*)(p_input->pp_selected_es[i_es_loop]->p_dec) /*, NULL */ );
411 adec_DestroyThread( (adec_thread_t*)(p_input->pp_selected_es[i_es_loop]->p_dec) );
414 ac3dec_DestroyThread( (ac3dec_thread_t *)(p_input->pp_selected_es[i_es_loop]->p_dec) );
417 fprintf(stderr, "input.h : destroying spudec\n");
418 spudec_DestroyThread( (spudec_thread_t *)(p_input->pp_selected_es[i_es_loop]->p_dec) );
421 /* Special streams for the PSI decoder, PID 0 and 1 */
425 intf_DbgMsg("error: unknown decoder type %d\n", p_input->pp_selected_es[i_es_loop]->i_type );
431 input_NetlistEnd( p_input ); /* clean netlist */
432 input_PsiEnd( p_input ); /* clean PSI information */
433 input_PcrEnd( p_input ); /* clean PCR information */
434 free( p_input ); /* free input_thread structure */
437 *pi_status = THREAD_OVER;
440 /*******************************************************************************
441 * input_ReadPacket: reads a packet from the network or the file
442 *******************************************************************************/
443 static __inline__ int input_ReadPacket( input_thread_t *p_input )
445 int i_base_index; /* index of the first free iovec */
448 #ifdef INPUT_LIFO_TS_NETLIST
449 int i_meanwhile_released;
450 int i_currently_removed;
452 ts_packet_t * p_ts_packet;
454 /* In this function, we only care about the TS netlist. PES netlist
455 * is for the demultiplexer. */
456 #ifdef INPUT_LIFO_TS_NETLIST
457 i_base_index = p_input->netlist.i_ts_index;
459 /* Verify that we still have packets in the TS netlist */
460 if( (INPUT_MAX_TS + INPUT_TS_READ_ONCE - 1 - p_input->netlist.i_ts_index) <= INPUT_TS_READ_ONCE )
462 intf_ErrMsg("input error: TS netlist is empty !\n");
466 #else /* FIFO netlist */
467 i_base_index = p_input->netlist.i_ts_start;
468 if( p_input->netlist.i_ts_start + INPUT_TS_READ_ONCE -1 > INPUT_MAX_TS )
470 /* The netlist is splitted in 2 parts. We must gather them to consolidate
471 the FIFO (we make the loop easily in having the same iovec at the far
472 end and in the beginning of netlist_free).
473 That's why the netlist is (INPUT_MAX_TS +1) + (INPUT_TS_READ_ONCE -1)
475 memcpy( p_input->netlist.p_ts_free + INPUT_MAX_TS + 1,
476 p_input->netlist.p_ts_free,
477 (p_input->netlist.i_ts_start + INPUT_TS_READ_ONCE - 1 - INPUT_MAX_TS)
478 * sizeof(struct iovec) );
481 /* Verify that we still have packets in the TS netlist */
482 if( ((p_input->netlist.i_ts_end -1 - p_input->netlist.i_ts_start) & INPUT_MAX_TS) <= INPUT_TS_READ_ONCE )
484 intf_ErrMsg("input error: TS netlist is empty !\n");
487 #endif /* FIFO netlist */
489 /* Scatter read the buffer. */
490 i_packet_size = (*p_input->p_Read)( p_input,
491 &p_input->netlist.p_ts_free[i_base_index],
492 INPUT_TS_READ_ONCE );
493 if( i_packet_size == (-1) )
495 // intf_DbgMsg("Read packet %d %p %d %d\n", i_base_index,
496 // &p_input->netlist.p_ts_free[i_base_index],
497 // p_input->netlist.i_ts_start,
498 // p_input->netlist.i_ts_end);
499 intf_ErrMsg("input error: readv() failed (%s)\n", strerror(errno));
503 if( i_packet_size == 0 )
505 /* No packet has been received, so stop here. */
509 /* Demultiplex the TS packets (1..INPUT_TS_READ_ONCE) received. */
510 for( i_current_index = i_base_index;
511 (i_packet_size -= TS_PACKET_SIZE) >= 0;
514 /* BTW, something REALLY bad could happen if we receive packets with
516 p_ts_packet = (ts_packet_t*)(p_input->netlist.p_ts_free[i_current_index].iov_base);
517 /* Don't cry :-), we are allowed to do that cast, because initially,
518 our buffer was malloc'ed with sizeof(ts_packet_t) */
520 /* Find out if we need this packet and demultiplex. */
521 input_SortPacket( p_input /* for current PIDs and netlist */,
525 if( i_packet_size > 0 )
527 intf_ErrMsg("input error: wrong size\n");
531 /* Remove the TS packets we have just filled from the netlist */
532 #ifdef INPUT_LIFO_TS_NETLIST
533 /* We need to take a lock here while we're calculating index positions. */
534 vlc_mutex_lock( &p_input->netlist.lock );
536 i_meanwhile_released = i_base_index - p_input->netlist.i_ts_index;
537 if( i_meanwhile_released )
539 /* That's where it becomes funny :-). Since we didn't take locks for
540 efficiency reasons, other threads (including ourselves, with
541 input_DemuxPacket) might have released packets to the netlist.
542 So we have to copy these iovec where they should go.
544 BTW, that explains why the TS netlist is
545 (INPUT_MAX_TS +1) + (TS_READ_ONCE -1) large. */
547 i_currently_removed = i_current_index - i_base_index;
548 if( i_meanwhile_released < i_currently_removed )
550 /* Copy all iovecs in that case */
551 memcpy( &p_input->netlist.p_ts_free[p_input->netlist.i_ts_index]
552 + i_currently_removed,
553 &p_input->netlist.p_ts_free[p_input->netlist.i_ts_index],
554 i_meanwhile_released * sizeof(struct iovec) );
558 /* We have fewer places than items, so we only move
559 i_currently_removed of them. */
560 memcpy( &p_input->netlist.p_ts_free[i_base_index],
561 &p_input->netlist.p_ts_free[p_input->netlist.i_ts_index],
562 i_currently_removed * sizeof(struct iovec) );
565 /* Update i_netlist_index with the information gathered above. */
566 p_input->netlist.i_ts_index += i_currently_removed;
570 /* Nothing happened. */
571 p_input->netlist.i_ts_index = i_current_index;
574 vlc_mutex_unlock( &p_input->netlist.lock );
576 #else /* FIFO netlist */
577 /* & is modulo ; that's where we make the loop. */
578 p_input->netlist.i_ts_start = i_current_index & INPUT_MAX_TS;
582 p_input->c_packets_read += i_current_index - i_base_index;
583 p_input->c_bytes += (i_current_index - i_base_index) * TS_PACKET_SIZE;
588 /*******************************************************************************
589 * input_SortPacket: find out whether we need that packet
590 *******************************************************************************/
591 static __inline__ void input_SortPacket( input_thread_t *p_input,
592 ts_packet_t *p_ts_packet )
597 /* Verify that sync_byte, error_indicator and scrambling_control are
599 if( !(p_ts_packet->buffer[0] == 0x47) || (p_ts_packet->buffer[1] & 0x80) ||
600 (p_ts_packet->buffer[3] & 0xc0) )
602 intf_DbgMsg("input debug: invalid TS header (%p)\n", p_ts_packet);
606 /* Get the PID of the packet. Note that ntohs is needed, for endianness
607 purposes (see man page). */
608 i_current_pid = U16_AT(&p_ts_packet->buffer[1]) & 0x1fff;
610 // intf_DbgMsg("input debug: pid %d received (%p)\n",
611 // i_current_pid, p_ts_packet);
613 /* Lock current ES state. */
614 vlc_mutex_lock( &p_input->es_lock );
616 /* Verify that we actually want this PID. */
617 for( i_es_loop = 0; i_es_loop < INPUT_MAX_SELECTED_ES; i_es_loop++ )
619 if( p_input->pp_selected_es[i_es_loop] != NULL)
621 if( (*p_input->pp_selected_es[i_es_loop]).i_id
624 /* Don't need the lock anymore, since the value pointed
625 out by p_input->pp_selected_es[i_es_loop] can only be
626 modified from inside the input_thread (by the PSI
627 decoder): interface thread is only allowed to modify
628 the pp_selected_es table */
629 vlc_mutex_unlock( &p_input->es_lock );
631 /* We're interested. Pass it to the demultiplexer. */
632 input_DemuxTS( p_input, p_ts_packet,
633 p_input->pp_selected_es[i_es_loop] );
639 /* pp_selected_es should not contain any hole. */
643 vlc_mutex_unlock( &p_input->es_lock );
646 /* We weren't interested in receiving this packet. Give it back to the
648 // intf_DbgMsg("SortPacket: freeing unwanted TS %p (pid %d)\n", p_ts_packet,
649 // U16_AT(&p_ts_packet->buffer[1]) & 0x1fff);
650 input_NetlistFreeTS( p_input, p_ts_packet );
652 p_input->c_packets_trashed++;
656 /*******************************************************************************
657 * input_DemuxTS: first step of demultiplexing: the TS header
658 *******************************************************************************
659 * Stream must also only contain PES and PSI, so PID must have been filtered
660 *******************************************************************************/
661 static __inline__ void input_DemuxTS( input_thread_t *p_input,
662 ts_packet_t *p_ts_packet,
663 es_descriptor_t *p_es_descriptor )
666 boolean_t b_adaption; /* Adaption field is present */
667 boolean_t b_payload; /* Packet carries payload */
668 boolean_t b_unit_start; /* A PSI or a PES start in the packet */
669 boolean_t b_trash = 0; /* Must the packet be trashed ? */
670 boolean_t b_lost = 0; /* Was there a packet lost ? */
674 ASSERT(p_es_descriptor);
676 #define p (p_ts_packet->buffer)
678 // intf_DbgMsg("input debug: TS-demultiplexing packet %p, pid %d, number %d\n",
679 // p_ts_packet, U16_AT(&p[1]) & 0x1fff, p[3] & 0x0f);
682 p_es_descriptor->c_packets++;
683 p_es_descriptor->c_bytes += TS_PACKET_SIZE;
686 /* Extract flags values from TS common header. */
687 b_unit_start = (p[1] & 0x40);
688 b_adaption = (p[3] & 0x20);
689 b_payload = (p[3] & 0x10);
691 /* Extract adaption field informations if any */
694 /* We don't have any adaptation_field, so payload start immediately
695 after the 4 byte TS header */
696 p_ts_packet->i_payload_start = 4;
700 /* p[4] is adaptation_field_length minus one */
701 p_ts_packet->i_payload_start = 5 + p[4];
703 /* The adaption field can be limited to the adaptation_field_length byte,
704 so that there is nothing to do: skip this possibility */
707 /* If the packet has both adaptation_field and payload, adaptation_field
708 cannot be more than 182 bytes long; if there is only an
709 adaptation_field, it must fill the next 183 bytes. */
710 if( b_payload ? (p[4] > 182) : (p[4] != 183) )
712 intf_DbgMsg("input debug: invalid TS adaptation field (%p)\n",
715 p_es_descriptor->c_invalid_packets++;
720 /* No we are sure that the byte containing flags is present: read it */
723 /* discontinuity_indicator */
726 intf_DbgMsg("discontinuity_indicator encountered by TS demux " \
727 "(position read: %d, saved: %d)\n", p[5] & 0x80,
728 p_es_descriptor->i_continuity_counter);
730 /* If the PID carries the PCR, there will be a system time-base
731 discontinuity. We let the PCR decoder handle that. */
732 p_es_descriptor->b_discontinuity = 1;
734 /* There also may be a continuity_counter discontinuity:
735 resynchronise our counter with the one of the stream */
736 p_es_descriptor->i_continuity_counter = (p[3] & 0x0f) - 1;
739 /* random_access_indicator */
740 p_es_descriptor->b_random |= p[5] & 0x40;
742 /* If this is a PCR_PID, and this TS packet contains a PCR,
743 we pass it along to the PCR decoder. */
744 if( (p_es_descriptor->b_pcr) && (p[5] & 0x10) )
746 /* There should be a PCR field in the packet, check if the
747 adaption field is long enough to carry it */
750 /* Call the PCR decoder */
751 input_PcrDecode( p_input, p_es_descriptor, &p[6] );
758 /* Check the continuity of the stream. */
759 i_dummy = ((p[3] & 0x0f) - p_es_descriptor->i_continuity_counter) & 0x0f;
762 /* Everything is ok, just increase our counter */
763 p_es_descriptor->i_continuity_counter++;
767 if( !b_payload && i_dummy == 0 )
769 /* This is a packet without payload, this is allowed by the draft
770 As there is nothing interessant in this packet (except PCR that
771 have already been handled), we can trash the packet. */
772 intf_DbgMsg("Packet without payload received by TS demux\n");
775 else if( i_dummy <= 0 )
777 /* Duplicate packet: mark it as being to be trashed. */
778 intf_DbgMsg("Duplicate packet received by TS demux\n");
781 else if( p_es_descriptor->i_continuity_counter == 0xFF )
783 /* This means that the packet is the first one we receive for this
784 ES since the continuity counter ranges between 0 and 0x0F
785 excepts when it has been initialized by the input: Init the
786 counter to the correct value. */
787 intf_DbgMsg("First packet for PID %d received by TS demux\n",
788 p_es_descriptor->i_id);
789 p_es_descriptor->i_continuity_counter = (p[3] & 0x0f);
793 /* This can indicate that we missed a packet or that the
794 continuity_counter wrapped and we received a dup packet: as we
795 don't know, do as if we missed a packet to be sure to recover
796 from this situation */
797 intf_DbgMsg("Packet lost by TS demux: current %d, packet %d\n",
798 p_es_descriptor->i_continuity_counter & 0x0f,
801 p_es_descriptor->i_continuity_counter = p[3] & 0x0f;
805 /* Trash the packet if it has no payload or if it is bad */
808 input_NetlistFreeTS( p_input, p_ts_packet );
810 p_input->c_packets_trashed++;
815 if( p_es_descriptor->b_psi )
817 /* The payload contains PSI tables */
818 input_DemuxPSI( p_input, p_ts_packet, p_es_descriptor,
819 b_unit_start, b_lost );
823 /* The payload carries a PES stream */
824 input_DemuxPES( p_input, p_ts_packet, p_es_descriptor,
825 b_unit_start, b_lost );
835 /*******************************************************************************
837 *******************************************************************************
838 * Gather a PES packet and analyzes its header.
839 *******************************************************************************/
840 static __inline__ void input_DemuxPES( input_thread_t *p_input,
841 ts_packet_t *p_ts_packet,
842 es_descriptor_t *p_es_descriptor,
843 boolean_t b_unit_start,
844 boolean_t b_packet_lost )
846 decoder_fifo_t * p_fifo;
847 u8 i_pes_header_size;
849 pes_packet_t* p_last_pes;
851 int i_ts_payload_size;
854 #define p_pes (p_es_descriptor->p_pes_packet)
858 ASSERT(p_es_descriptor);
860 // intf_DbgMsg("PES-demultiplexing %p (%p)\n", p_ts_packet, p_pes);
862 /* If we lost data, discard the PES packet we are trying to reassemble
863 if any and wait for the beginning of a new one in order to synchronise
865 if( b_packet_lost && p_pes != NULL )
867 intf_DbgMsg("PES %p trashed because of packet lost\n", p_pes);
868 input_NetlistFreePES( p_input, p_pes );
872 /* If the TS packet contains the begining of a new PES packet, and if we
873 were reassembling a PES packet, then the PES should be complete now,
874 so parse its header and give it to the decoders */
875 if( b_unit_start && p_pes != NULL )
877 // intf_DbgMsg("End of PES packet %p\n", p_pes);
879 /* Parse the header. The header has a variable length, but in order
880 to improve the algorithm, we will read the 14 bytes we may be
882 p_ts = p_pes->p_first_ts;
883 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
886 if(i_ts_payload_size >= PES_HEADER_SIZE)
888 /* This part of the header entirely fits in the payload of
889 the first TS packet */
890 p_pes->p_pes_header = &(p_ts->buffer[p_ts->i_payload_start]);
894 /* This part of the header does not fit in the current TS packet:
895 copy the part of the header we are interested in to the
896 p_pes_header_save buffer. The buffer is dynamicly allocated if
897 needed so it's time expensive but this situation almost never
899 intf_DbgMsg("Code never tested encountered, WARNING ! (benny)\n");
900 if( !p_pes->p_pes_header_save )
901 p_pes->p_pes_header_save = malloc(PES_HEADER_SIZE);
905 memcpy(p_pes->p_pes_header_save + i_dummy,
906 &p_ts->buffer[p_ts->i_payload_start], i_ts_payload_size);
907 i_dummy += i_ts_payload_size;
909 p_ts = p_ts->p_next_ts;
912 /* The payload of the PES packet is shorter than the 14 bytes
913 we would read. This means that high packet lost occured
914 so the PES won't be usefull for any decoder. Moreover,
915 this should never happen so we can trash the packet and
916 exit roughly without regrets */
917 intf_DbgMsg("PES packet too short: trashed\n");
918 input_NetlistFreePES( p_input, p_pes );
924 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
926 while(i_ts_payload_size + i_dummy < PES_HEADER_SIZE);
928 /* This last TS packet is partly header, partly payload, so just
929 copy the header part */
930 memcpy(p_pes->p_pes_header_save + i_dummy,
931 &p_ts->buffer[p_ts->i_payload_start],
932 PES_HEADER_SIZE - i_dummy);
934 /* The header must be read in the buffer not in any TS packet */
935 p_pes->p_pes_header = p_pes->p_pes_header_save;
938 /* Now we have the part of the PES header we were interested in:
941 /* First read the 6 header bytes common to all PES packets:
942 use them to test the PES validity */
943 if( (p_pes->p_pes_header[0] || p_pes->p_pes_header[1] ||
944 (p_pes->p_pes_header[2] != 1)) ||
945 /* packet_start_code_prefix != 0x000001 */
946 ((i_dummy = U16_AT(p_pes->p_pes_header + 4)) &&
947 (i_dummy + 6 != p_pes->i_pes_size)) )
948 /* PES_packet_length is set and != total received payload */
950 /* Trash the packet and set p_pes to NULL to be sure the next PES
951 packet will have its b_data_lost flag set */
952 intf_DbgMsg("Corrupted PES packet received: trashed\n");
953 input_NetlistFreePES( p_input, p_pes );
959 /* The PES packet is valid. Check its type to test if it may
960 carry additional informations in a header extension */
961 p_pes->i_stream_id = p_pes->p_pes_header[3];
963 switch( p_pes->i_stream_id )
965 case 0xBE: /* Padding */
966 case 0xBC: /* Program stream map */
967 case 0xBF: /* Private stream 2 */
970 case 0xFF: /* Program stream directory */
971 case 0xF2: /* DSMCC stream */
972 case 0xF8: /* ITU-T H.222.1 type E stream */
973 /* The payload begins immediatly after the 6 bytes header, so
974 we have finished with the parsing */
975 i_pes_header_size = 6;
979 /* The PES header contains at least 3 more bytes: parse them */
980 p_pes->b_data_alignment = p_pes->p_pes_header[6] & 0x04;
981 p_pes->b_has_pts = p_pes->p_pes_header[7] & 0x80;
982 i_pes_header_size = p_pes->p_pes_header[8] + 9;
984 /* Now parse the optional header extensions (in the limit of
986 if( p_pes->b_has_pts )
988 pcr_descriptor_t * p_pcr;
990 p_pcr = p_input->p_pcr;
993 ( ((mtime_t)(p_pes->p_pes_header[9] & 0x0E) << 29) |
994 (((mtime_t)U16_AT(p_pes->p_pes_header + 10) << 14) - (1 << 14)) |
995 ((mtime_t)U16_AT(p_pes->p_pes_header + 12) >> 1) ) * 300;
998 if( p_pcr->i_synchro_state )
1000 switch( p_pcr->i_synchro_state )
1002 case SYNCHRO_NOT_STARTED:
1003 p_pes->b_has_pts = 0;
1007 p_pes->i_pts += p_pcr->delta_pcr;
1008 p_pcr->delta_absolute = mdate() - p_pes->i_pts + INPUT_PTS_DELAY;
1009 p_pes->i_pts += p_pcr->delta_absolute;
1010 p_pcr->i_synchro_state = 0;
1013 case SYNCHRO_REINIT: /* We skip a PES */
1014 p_pes->b_has_pts = 0;
1015 p_pcr->i_synchro_state = SYNCHRO_START;
1021 p_pes->i_pts += p_pcr->delta_pcr + p_pcr->delta_absolute;
1027 /* Now we've parsed the header, we just have to indicate in some
1028 specific TS packets where the PES payload begins (renumber
1029 i_payload_start), so that the decoders can find the beginning
1030 of their data right out of the box. */
1031 p_ts = p_pes->p_first_ts;
1032 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
1033 while( i_pes_header_size > i_ts_payload_size )
1035 /* These packets are entirely filled by the PES header. */
1036 i_pes_header_size -= i_ts_payload_size;
1037 p_ts->i_payload_start = p_ts->i_payload_end;
1038 /* Go to the next TS packet: here we won't have to test it is
1039 not NULL because we trash the PES packets when packet lost
1041 p_ts = p_ts->p_next_ts;
1042 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
1044 /* This last packet is partly header, partly payload. */
1045 p_ts->i_payload_start += i_pes_header_size;
1048 /* Now we can eventually put the PES packet in the decoder's
1050 switch( p_es_descriptor->i_type )
1052 case MPEG1_VIDEO_ES:
1053 case MPEG2_VIDEO_ES:
1055 p_fifo = &(((vdec_thread_t*)(p_es_descriptor->p_dec))->fifo);
1057 p_fifo = &(((vpar_thread_t*)(p_es_descriptor->p_dec))->fifo);
1061 case MPEG1_AUDIO_ES:
1062 case MPEG2_AUDIO_ES:
1063 p_fifo = &(((adec_thread_t*)(p_es_descriptor->p_dec))->fifo);
1067 /* we skip 4 bytes at the beginning of the AC3 payload */
1068 p_ts->i_payload_start += 4;
1069 p_fifo = &(((ac3dec_thread_t *)(p_es_descriptor->p_dec))->fifo);
1073 /* we skip 4 bytes at the beginning of the subpicture payload */
1074 p_ts->i_payload_start += 4;
1075 fprintf(stderr, "input.h : launching spudec\n");
1076 p_fifo = &(((spudec_thread_t *)(p_es_descriptor->p_dec))->fifo);
1080 /* This should never happen */
1081 intf_DbgMsg("Unknown stream type (%d, %d): PES trashed\n",
1082 p_es_descriptor->i_id, p_es_descriptor->i_type);
1087 if( p_fifo != NULL )
1089 vlc_mutex_lock( &p_fifo->data_lock );
1090 if( DECODER_FIFO_ISFULL( *p_fifo ) )
1092 /* The FIFO is full !!! This should not happen. */
1094 p_input->c_packets_trashed += p_pes->i_ts_packets;
1095 p_es_descriptor->c_invalid_packets += p_pes->i_ts_packets;
1097 input_NetlistFreePES( p_input, p_pes );
1098 intf_DbgMsg("PES trashed - fifo full ! (%d, %d)\n",
1099 p_es_descriptor->i_id, p_es_descriptor->i_type);
1103 // intf_DbgMsg("Putting %p into fifo %p/%d\n",
1104 // p_pes, p_fifo, p_fifo->i_end);
1105 p_fifo->buffer[p_fifo->i_end] = p_pes;
1106 DECODER_FIFO_INCEND( *p_fifo );
1108 /* Warn the decoder that it's got work to do. */
1109 vlc_cond_signal( &p_fifo->data_wait );
1111 vlc_mutex_unlock( &p_fifo->data_lock );
1115 intf_DbgMsg("No fifo to receive PES %p: trash\n", p_pes);
1117 p_input->c_packets_trashed += p_pes->i_ts_packets;
1118 p_es_descriptor->c_invalid_packets += p_pes->i_ts_packets;
1120 input_NetlistFreePES( p_input, p_pes );
1126 /* If we are at the beginning of a new PES packet, we must fetch a new
1127 PES buffer to begin with the reassembly of this PES packet. This is
1128 also here that we can synchronise with the stream if we we lost
1129 packets or if the decoder has just started */
1134 /* Get a new one PES from the PES netlist. */
1135 if( (p_pes = input_NetlistGetPES( p_input )) == (NULL) )
1137 /* PES netlist is empty ! */
1138 p_input->b_error = 1;
1142 // intf_DbgMsg("New PES packet %p (first TS: %p)\n", p_pes, p_ts_packet);
1144 /* Init the PES fields so that the first TS packet could be correctly
1145 added to the PES packet (see below) */
1146 p_pes->p_first_ts = p_ts_packet;
1147 p_pes->p_last_ts = NULL;
1149 /* If the last pes packet was null, this means that the synchronisation
1150 was lost and so warn the decoder that he will have to find a way to
1153 p_pes->b_data_loss = 1;
1155 /* Read the b_random_access flag status and then reinit it */
1156 p_pes->b_random_access = p_es_descriptor->b_random;
1157 p_es_descriptor->b_random = 0;
1162 /* If we are synchronised with the stream, and so if we are ready to
1163 receive correctly the data, add the TS packet to the current PES
1167 // intf_DbgMsg("Adding TS %p to PES %p\n", p_ts_packet, p_pes);
1169 /* Size of the payload carried in the TS packet */
1170 i_ts_payload_size = p_ts_packet->i_payload_end -
1171 p_ts_packet->i_payload_start;
1173 /* Update the relations between the TS packets */
1174 p_ts_packet->p_prev_ts = p_pes->p_last_ts;
1175 p_ts_packet->p_next_ts = NULL;
1176 if( p_pes->i_ts_packets != 0 )
1178 /* Regarder si il serait pas plus efficace de ne creer que les liens
1179 precedent->suivant pour le moment, et les liens suivant->precedent
1180 quand le paquet est termine */
1181 /* Otherwise it is the first TS packet. */
1182 p_pes->p_last_ts->p_next_ts = p_ts_packet;
1184 /* Now add the TS to the PES packet */
1185 p_pes->p_last_ts = p_ts_packet;
1186 p_pes->i_ts_packets++;
1187 p_pes->i_pes_size += i_ts_payload_size;
1191 i_dummy = p_ts_packet->i_payload_end - p_ts_packet->i_payload_start;
1192 p_es_descriptor->c_payload_bytes += i_dummy;
1197 /* Since we don't use the TS packet to build a PES packet, we don't
1198 need it anymore, so give it back to the netlist */
1199 // intf_DbgMsg("Trashing TS %p: no PES being build\n", p_ts_packet);
1200 input_NetlistFreeTS( p_input, p_ts_packet );
1209 /*******************************************************************************
1211 *******************************************************************************
1212 * Notice that current ES state has been locked by input_SortPacket. (No more true,
1213 * changed by benny - See if it's ok, and definitely change the code ???????? )
1214 *******************************************************************************/
1215 static __inline__ void input_DemuxPSI( input_thread_t *p_input,
1216 ts_packet_t *p_ts_packet,
1217 es_descriptor_t *p_es_descriptor,
1218 boolean_t b_unit_start, boolean_t b_packet_lost )
1220 int i_data_offset; /* Offset of the interesting data in the TS packet */
1221 u16 i_data_length; /* Length of those data */
1222 //boolean_t b_first_section; /* Was there another section in the TS packet ? */
1225 ASSERT(p_ts_packet);
1226 ASSERT(p_es_descriptor);
1228 #define p_psi (p_es_descriptor->p_psi_section)
1230 // intf_DbgMsg( "input debug: PSI demultiplexing %p (%p)\n", p_ts_packet, p_input);
1232 // 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);
1235 /* Try to find the beginning of the payload in the packet to initialise
1236 the do-while loop that follows -> Compute the i_data_offset variable:
1237 by default, the value is set so that we won't enter in the while loop.
1238 It will be set to a correct value if the data are not corrupted */
1239 i_data_offset = TS_PACKET_SIZE;
1241 /* Has the reassembly of a section already began in a previous packet ? */
1242 if( p_psi->b_running_section )
1244 /* Was data lost since the last TS packet ? */
1247 /* Discard the packet and wait for the begining of a new one to resynch */
1248 p_psi->b_running_section = 0;
1249 p_psi->i_current_position = 0;
1250 intf_DbgMsg( "PSI section(s) discarded due to packet loss\n" );
1254 /* The data that complete a previously began section are always at
1255 the beginning of the TS payload... */
1256 i_data_offset = p_ts_packet->i_payload_start;
1257 /* ...Unless there is a pointer field, that we have to bypass */
1260 // intf_DbgMsg( "New part of the section received at offset %d\n", i_data_offset );
1263 /* We are looking for the beginning of a new section */
1268 /* Get the offset at which the data for that section can be found
1269 The offset is stored in the pointer_field since we are interested in
1270 the first section of the TS packet. Note that the +1 is to bypass
1271 the pointer field */
1272 i_data_offset = p_ts_packet->i_payload_start +
1273 p_ts_packet->buffer[p_ts_packet->i_payload_start] + 1;
1274 // intf_DbgMsg( "New section beginning at offset %d in TS packet\n", i_data_offset );
1278 /* This may either mean that the TS is bad or that the packet contains
1279 the end of a section that had been discarded in a previous loop:
1280 trash the TS packet since we cannot do anything with those data: */
1281 p_psi->b_running_section = 0;
1282 p_psi->i_current_position = 0;
1283 intf_DbgMsg( "PSI packet discarded due to lack of synchronisation\n" );
1287 /* The section we will deal with during the first iteration of the following
1288 loop is the first one contained in the TS packet */
1289 // b_first_section = 1;
1291 /* Reassemble the pieces of sections contained in the TS packet and decode
1292 the sections that could have been completed.
1293 Stop when we reach the end of the packet or stuffing bytes */
1294 while( i_data_offset < TS_PACKET_SIZE && p_ts_packet->buffer[i_data_offset] != 0xFF )
1296 /* If the current section is a new one, reinit the data fields of the p_psi
1297 struct to start its decoding */
1298 if( !p_psi->b_running_section )
1300 /* Read the length of the new section */
1301 p_psi->i_length = (U16_AT(&p_ts_packet->buffer[i_data_offset+1]) & 0xFFF) + 3;
1302 // intf_DbgMsg( "Section length %d\n", p_psi->i_length );
1303 if( p_psi->i_length > PSI_SECTION_SIZE )
1305 /* The TS packet is corrupted, stop here to avoid possible a seg fault */
1306 intf_DbgMsg( "PSI Section size is too big, aborting its reception\n" );
1310 /* Init the reassembly of that section */
1311 p_psi->b_running_section = 1;
1312 p_psi->i_current_position = 0;
1315 /* Compute the length of data related to the section in this TS packet */
1316 if( p_psi->i_length - p_psi->i_current_position > TS_PACKET_SIZE - i_data_offset)
1317 i_data_length = TS_PACKET_SIZE - i_data_offset;
1319 i_data_length = p_psi->i_length - p_psi->i_current_position;
1321 /* Copy those data in the section buffer */
1322 memcpy( &p_psi->buffer[p_psi->i_current_position], &p_ts_packet->buffer[i_data_offset],
1325 /* Interesting data are now after the ones we copied, since no gap is
1326 allowed between 2 sections in a TS packets */
1327 i_data_offset += i_data_length;
1329 /* Decode the packet if it is now complete */
1330 if (p_psi->i_length == p_psi->i_current_position + i_data_length)
1332 /* Packet is complete, decode it */
1333 // intf_DbgMsg( "SECTION COMPLETE: starting decoding of its data\n" );
1334 input_PsiDecode( p_input, p_psi );
1336 /* Prepare the buffer to receive a new section */
1337 p_psi->i_current_position = 0;
1338 p_psi->b_running_section = 0;
1340 /* The new section won't be the first anymore */
1341 //b_first_section = 0;
1345 /* Prepare the buffer to receive the next part of the section */
1346 p_psi->i_current_position += i_data_length;
1347 // intf_DbgMsg( "Section not complete, waiting for the end\n" );
1350 // intf_DbgMsg( "Must loop ? Next data offset: %d, stuffing: %d\n",
1351 // i_data_offset, p_ts_packet->buffer[i_data_offset] );
1354 /* Relase the TS packet, we don't need it anymore */
1355 input_NetlistFreeTS( p_input, p_ts_packet );