1 /*******************************************************************************
2 * input.c: input thread
4 *******************************************************************************
5 * Read an MPEG2 stream, demultiplex and parse it before sending it to
7 *******************************************************************************/
9 /*******************************************************************************
11 *******************************************************************************/
14 #include <sys/uio.h> /* iovec */
18 #include <X11/extensions/XShm.h>
19 #include <sys/soundcard.h>
21 #include <stdlib.h> /* atoi(), malloc(), free() */
23 #include <sys/ioctl.h> /* ioctl() */
24 #include <net/if.h> /* ifreq */
25 #include <netinet/in.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"
41 #include "audio_output.h"
42 #include "audio_decoder.h"
45 #include "video_output.h"
46 #include "video_decoder.h"
48 /******************************************************************************
50 ******************************************************************************/
51 static void input_Thread( input_thread_t *p_input );
52 static void ErrorThread( input_thread_t *p_input );
53 static void EndThread( input_thread_t *p_input );
54 static __inline__ int input_ReadPacket( input_thread_t *p_input );
55 static __inline__ void input_SortPacket( input_thread_t *p_input,
56 ts_packet_t *ts_packet );
57 static __inline__ void input_DemuxTS( input_thread_t *p_input,
58 ts_packet_t *ts_packet,
59 es_descriptor_t *es_descriptor );
60 static __inline__ void input_DemuxPES( input_thread_t *p_input,
61 ts_packet_t *ts_packet,
62 es_descriptor_t *p_es_descriptor,
63 boolean_t b_unit_start, boolean_t b_packet_lost );
64 static __inline__ void input_DemuxPSI( input_thread_t *p_input,
65 ts_packet_t *ts_packet,
66 es_descriptor_t *p_es_descriptor,
67 boolean_t b_unit_start, boolean_t b_packet_lost );
69 /*******************************************************************************
70 * input_CreateThread: initialize and spawn an input thread
71 *******************************************************************************
72 * This function initializes and spawns an input thread. It returns NULL on
73 * failure. If you want a better understanding of the input thread, don't start
74 * by reading this function :-).
75 *******************************************************************************/
76 input_thread_t *input_CreateThread( input_cfg_t *p_cfg )
78 input_thread_t * p_input;
81 intf_DbgMsg("input debug 1-1: creating thread (cfg : %p)\n", p_cfg );
83 /* Allocate input_thread_t structure. */
84 if( !( p_input = (input_thread_t *)malloc(sizeof(input_thread_t)) ) )
86 intf_ErrMsg("input error: can't allocate input thread structure (%s)\n",
91 bzero( p_input, sizeof(input_thread_t));
92 for( i_index = 0; i_index < INPUT_MAX_ES; i_index++ )
94 p_input->p_es[i_index].i_id = EMPTY_PID;
97 /* Find out which method we are gonna use and retrieve pointers. */
98 if( !((p_cfg->i_properties) & INPUT_CFG_METHOD) )
100 /* i_method is not set. */
101 intf_DbgMsg("input debug: using default method (%d)\n",
102 INPUT_DEFAULT_METHOD);
103 p_cfg->i_method = INPUT_DEFAULT_METHOD;
104 p_cfg->i_properties |= INPUT_CFG_METHOD;
106 p_input->i_method = p_cfg->i_method;
107 switch( p_cfg->i_method )
110 case INPUT_METHOD_TS_FILE:
111 p_input->p_open = &input_FileCreateMethod;
112 p_input->p_read = &input_FileRead;
113 p_input->p_clean = &input_FileDestroyMethod;
116 /* Network methods */
117 case INPUT_METHOD_TS_UCAST:
118 case INPUT_METHOD_TS_MCAST:
119 case INPUT_METHOD_TS_BCAST:
120 case INPUT_METHOD_TS_VLAN_BCAST:
121 p_input->p_open = &input_NetworkCreateMethod;
122 p_input->p_read = &input_NetworkRead;
123 p_input->p_clean = &input_NetworkDestroyMethod;
126 case INPUT_METHOD_NONE:
129 /* Internal error, which should never happen */
130 intf_DbgMsg("input debug: unknow method type %d\n",
137 /* Initialize PSI decoder. */
138 intf_DbgMsg("Initializing PSI decoder\n");
139 if( input_PsiInit( p_input ) == -1 )
145 /* Initialize PCR decoder. */
146 intf_DbgMsg("Initializing PCR decoder\n");
147 if( input_PcrInit( p_input ) == -1 )
149 input_PsiClean( p_input );
154 /* Initialize netlists. */
155 if( input_NetlistOpen( p_input ) )
157 input_PsiClean( p_input );
158 input_PcrClean( p_input );
164 /* Initialize counters. */
165 p_input->c_bytes = 0;
166 p_input->c_payload_bytes = 0;
167 p_input->c_ts_packets_read = 0;
168 p_input->c_ts_packets_trashed = 0;
170 p_input->c_loops = 0;
174 /* Let the appropriate method open the socket. */
175 if( (*(p_input->p_open))( p_input, p_cfg ) == -1 )
177 input_NetlistClean( p_input );
178 input_PsiClean( p_input );
179 input_PcrClean( p_input );
184 intf_DbgMsg("input debug: method %d properly initialized the socket\n",
187 /* Create thread and set locks. */
189 pthread_mutex_init( &p_input->netlist.lock, NULL );
190 pthread_mutex_init( &p_input->programs_lock, NULL );
191 pthread_mutex_init( &p_input->es_lock, NULL );
193 input_Thread( p_input );
195 if( pthread_create(&p_input->thread_id, NULL, (void *) input_Thread,
198 intf_ErrMsg("input error: can't spawn input thread (%s)\n",
200 (*p_input->p_clean)( p_input );
201 input_NetlistClean( p_input );;
202 input_PsiClean( p_input );
203 input_PcrClean( p_input );
209 /* Default setting for new decoders */
210 p_input->p_aout = p_cfg->p_aout;
215 /******************************************************************************
216 * input_DestroyThread: mark an input thread as zombie
217 ******************************************************************************
218 * This function should not return until the thread is effectively cancelled.
219 ******************************************************************************/
220 void input_DestroyThread( input_thread_t *p_input )
222 intf_DbgMsg("input debug: requesting termination of input thread\n");
223 p_input->b_die = 1; /* ask thread to kill itself */
225 /* Remove this as soon as the "status" flag is implemented */
226 pthread_join( p_input->thread_id, NULL ); /* wait until it's done */
230 /*******************************************************************************
231 * input_OpenAudioStream: open an audio stream
232 *******************************************************************************
233 * This function spawns an audio decoder and plugs it on the audio output
235 *******************************************************************************/
236 int input_OpenAudioStream( input_thread_t *p_input, int i_id )
241 /*******************************************************************************
242 * input_CloseAudioStream: close an audio stream
243 *******************************************************************************
244 * This function destroys an audio decoder.
245 *******************************************************************************/
246 void input_CloseAudioStream( input_thread_t *p_input, int i_id )
251 /*******************************************************************************
252 * input_OpenVideoStream: open a video stream
253 *******************************************************************************
254 * This function spawns a video decoder and plugs it on a video output thread.
255 *******************************************************************************/
256 int input_OpenVideoStream( input_thread_t *p_input,
257 struct vout_thread_s *p_vout, struct video_cfg_s * p_cfg )
262 /*******************************************************************************
263 * input_CloseVideoStream: close a video stream
264 *******************************************************************************
265 * This function destroys an video decoder.
266 *******************************************************************************/
267 void input_CloseVideoStream( input_thread_t *p_input, int i_id )
273 /* following functions are local */
275 /*******************************************************************************
276 * input_Thread: input thread
277 *******************************************************************************
278 * Thread in charge of processing the network packets and demultiplexing.
279 *******************************************************************************/
280 static void input_Thread( input_thread_t *p_input )
282 intf_DbgMsg("input debug 11-1: thread %p is active\n", p_input);
283 while( !p_input->b_die && !p_input->b_error )
285 /* Scatter read the UDP packet from the network or the file. */
286 if( (input_ReadPacket( p_input )) == (-1) )
288 /* ??? Normally, a thread can't kill itself, but we don't have
289 * any method in case of an error condition ... */
290 p_input->b_error = 1;
298 if( p_input->b_error )
300 ErrorThread( p_input );
303 /* Ohoh, we have to die as soon as possible. */
304 EndThread( p_input );
306 intf_DbgMsg("input debug: thread %p destroyed\n", p_input);
311 /******************************************************************************
312 * ErrorThread: RunThread() error loop
313 ******************************************************************************/
314 static void ErrorThread( input_thread_t *p_input )
316 while( !p_input->b_die )
318 msleep( INPUT_IDLE_SLEEP );
322 /*******************************************************************************
323 * EndThread: end the input thread
324 *******************************************************************************/
325 static void EndThread( input_thread_t *p_input )
329 (*p_input->p_clean)( p_input ); /* close input method */
331 /* Destroy all decoder threads. */
332 for( i_es_loop = 0; i_es_loop < INPUT_MAX_ES; i_es_loop++ )
334 if( p_input->pp_selected_es[i_es_loop] )
336 switch( p_input->pp_selected_es[i_es_loop]->i_type )
340 vdec_DestroyThread( (vdec_thread_t*)(p_input->pp_selected_es[i_es_loop]->p_dec) /*, NULL */ );
345 adec_DestroyThread( (adec_thread_t*)(p_input->pp_selected_es[i_es_loop]->p_dec) );
354 /* pp_selected_es should not contain any hole. */
359 input_NetlistClean( p_input ); /* clean netlist */
360 input_PsiClean( p_input ); /* clean PSI information */
361 input_PcrClean( p_input ); /* clean PCR information */
362 free( p_input ); /* free input_thread structure */
364 intf_DbgMsg("input debug: EndThread(%p)\n", p_input);
367 /*******************************************************************************
368 * input_ReadPacket: reads a packet from the network or the file
369 *******************************************************************************/
370 static __inline__ int input_ReadPacket( input_thread_t *p_input )
372 int i_base_index; /* index of the first free iovec */
375 #ifdef INPUT_LIFO_TS_NETLIST
376 int i_meanwhile_released;
377 int i_currently_removed;
379 ts_packet_t * p_ts_packet;
381 /* In this function, we only care about the TS netlist. PES netlist
382 * is for the demultiplexer. */
383 #ifdef INPUT_LIFO_TS_NETLIST
384 i_base_index = p_input->netlist.i_ts_index;
386 /* Verify that we still have packets in the TS netlist */
387 if( (INPUT_MAX_TS + INPUT_TS_READ_ONCE - 1 - p_input->netlist.i_ts_index) <= INPUT_TS_READ_ONCE )
389 intf_ErrMsg("input error: TS netlist is empty !\n");
393 #else /* FIFO netlist */
394 i_base_index = p_input->netlist.i_ts_start;
395 if( p_input->netlist.i_ts_start + INPUT_TS_READ_ONCE -1 > INPUT_MAX_TS )
397 /* The netlist is splitted in 2 parts. We must gather them to consolidate
398 the FIFO (we make the loop easily in having the same iovec at the far
399 end and in the beginning of netlist_free).
400 That's why the netlist is (INPUT_MAX_TS +1) + (INPUT_TS_READ_ONCE -1)
402 memcpy( p_input->netlist.p_ts_free + INPUT_MAX_TS + 1,
403 p_input->netlist.p_ts_free,
404 (p_input->netlist.i_ts_start + INPUT_TS_READ_ONCE - 1 - INPUT_MAX_TS)
405 * sizeof(struct iovec) );
408 /* Verify that we still have packets in the TS netlist */
409 if( ((p_input->netlist.i_ts_end -1 - p_input->netlist.i_ts_start) & INPUT_MAX_TS) <= INPUT_TS_READ_ONCE )
411 intf_ErrMsg("input error: TS netlist is empty !\n");
414 #endif /* FIFO netlist */
416 /* Scatter read the buffer. */
417 i_packet_size = (*p_input->p_read)( p_input,
418 &p_input->netlist.p_ts_free[i_base_index],
419 INPUT_TS_READ_ONCE );
420 if( i_packet_size == (-1) )
422 // intf_DbgMsg("Read packet %d %p %d %d\n", i_base_index,
423 // &p_input->netlist.p_ts_free[i_base_index],
424 // p_input->netlist.i_ts_start,
425 // p_input->netlist.i_ts_end);
426 intf_ErrMsg("input error: readv() failed (%s)\n", strerror(errno));
430 if( i_packet_size == 0 )
432 /* No packet has been received, so stop here. */
436 /* Demultiplex the TS packets (1..INPUT_TS_READ_ONCE) received. */
437 for( i_current_index = i_base_index;
438 (i_packet_size -= TS_PACKET_SIZE) >= 0;
441 /* BTW, something REALLY bad could happen if we receive packets with
443 p_ts_packet = (ts_packet_t*)(p_input->netlist.p_ts_free[i_current_index].iov_base);
444 /* Don't cry :-), we are allowed to do that cast, because initially,
445 our buffer was malloc'ed with sizeof(ts_packet_t) */
447 /* Find out if we need this packet and demultiplex. */
448 input_SortPacket( p_input /* for current PIDs and netlist */,
452 if( i_packet_size > 0 )
454 intf_ErrMsg("input error: wrong size\n");
458 /* Remove the TS packets we have just filled from the netlist */
459 #ifdef INPUT_LIFO_TS_NETLIST
460 /* We need to take a lock here while we're calculating index positions. */
461 pthread_mutex_lock( &p_input->netlist.lock );
463 i_meanwhile_released = i_base_index - p_input->netlist.i_ts_index;
464 if( i_meanwhile_released )
466 /* That's where it becomes funny :-). Since we didn't take locks for
467 efficiency reasons, other threads (including ourselves, with
468 input_DemuxPacket) might have released packets to the netlist.
469 So we have to copy these iovec where they should go.
471 BTW, that explains why the TS netlist is
472 (INPUT_MAX_TS +1) + (TS_READ_ONCE -1) large. */
474 i_currently_removed = i_current_index - i_base_index;
475 if( i_meanwhile_released < i_currently_removed )
477 /* Copy all iovecs in that case */
478 memcpy( &p_input->netlist.p_ts_free[p_input->netlist.i_ts_index]
479 + i_currently_removed,
480 &p_input->netlist.p_ts_free[p_input->netlist.i_ts_index],
481 i_meanwhile_released * sizeof(struct iovec) );
485 /* We have fewer places than items, so we only move
486 i_currently_removed of them. */
487 memcpy( &p_input->netlist.p_ts_free[i_base_index],
488 &p_input->netlist.p_ts_free[p_input->netlist.i_ts_index],
489 i_currently_removed * sizeof(struct iovec) );
492 /* Update i_netlist_index with the information gathered above. */
493 p_input->netlist.i_ts_index += i_currently_removed;
497 /* Nothing happened. */
498 p_input->netlist.i_ts_index = i_current_index;
501 pthread_mutex_unlock( &p_input->netlist.lock );
503 #else /* FIFO netlist */
504 /* & is modulo ; that's where we make the loop. */
505 p_input->netlist.i_ts_start = i_current_index & INPUT_MAX_TS;
509 p_input->c_ts_packets_read += i_current_index - i_base_index;
510 p_input->c_bytes += (i_current_index - i_base_index) * TS_PACKET_SIZE;
515 /*******************************************************************************
516 * input_SortPacket: find out whether we need that packet
517 *******************************************************************************/
518 static __inline__ void input_SortPacket( input_thread_t *p_input,
519 ts_packet_t *p_ts_packet )
524 /* Verify that sync_byte, error_indicator and scrambling_control are
526 if( !(p_ts_packet->buffer[0] == 0x47) || (p_ts_packet->buffer[1] & 0x80) ||
527 (p_ts_packet->buffer[3] & 0xc0) )
529 intf_DbgMsg("input debug: invalid TS header (%p)\n", p_ts_packet);
533 /* Get the PID of the packet. Note that ntohs is needed, for endianness
534 purposes (see man page). */
535 i_current_pid = U16_AT(&p_ts_packet->buffer[1]) & 0x1fff;
537 // intf_DbgMsg("input debug: pid %d received (%p)\n",
538 // i_current_pid, p_ts_packet);
540 /* Lock current ES state. */
541 pthread_mutex_lock( &p_input->es_lock );
543 /* Verify that we actually want this PID. */
544 for( i_es_loop = 0; i_es_loop < INPUT_MAX_SELECTED_ES; i_es_loop++ )
546 if( p_input->pp_selected_es[i_es_loop] != NULL)
548 if( (*p_input->pp_selected_es[i_es_loop]).i_id
551 /* Don't need the lock anymore, since the value pointed
552 out by p_input->pp_selected_es[i_es_loop] can only be
553 modified from inside the input_thread (by the PSI
554 decoder): interface thread is only allowed to modify
555 the pp_selected_es table */
556 pthread_mutex_unlock( &p_input->es_lock );
558 /* We're interested. Pass it to the demultiplexer. */
559 input_DemuxTS( p_input, p_ts_packet,
560 p_input->pp_selected_es[i_es_loop] );
566 /* pp_selected_es should not contain any hole. */
570 pthread_mutex_unlock( &p_input->es_lock );
573 /* We weren't interested in receiving this packet. Give it back to the
575 // intf_DbgMsg("SortPacket: freeing unwanted TS %p (pid %d)\n", p_ts_packet,
576 // U16_AT(&p_ts_packet->buffer[1]) & 0x1fff);
577 input_NetlistFreeTS( p_input, p_ts_packet );
579 p_input->c_ts_packets_trashed++;
583 /*******************************************************************************
584 * input_DemuxTS: first step of demultiplexing: the TS header
585 *******************************************************************************
586 * Stream must also only contain PES and PSI, so PID must have been filtered
587 *******************************************************************************/
588 static __inline__ void input_DemuxTS( input_thread_t *p_input,
589 ts_packet_t *p_ts_packet,
590 es_descriptor_t *p_es_descriptor )
593 boolean_t b_adaption; /* Adaption field is present */
594 boolean_t b_payload; /* Packet carries payload */
595 boolean_t b_unit_start; /* A PSI or a PES start in the packet */
596 boolean_t b_trash = 0; /* Must the packet be trashed ? */
597 boolean_t b_lost = 0; /* Was there a packet lost ? */
601 ASSERT(p_es_descriptor);
603 #define p (p_ts_packet->buffer)
605 // intf_DbgMsg("input debug: TS-demultiplexing packet %p, pid %d, number %d\n",
606 // p_ts_packet, U16_AT(&p[1]) & 0x1fff, p[3] & 0x0f);
609 p_es_descriptor->c_packets++;
610 p_es_descriptor->c_bytes += TS_PACKET_SIZE;
613 /* Extract flags values from TS common header. */
614 b_unit_start = (p[1] & 0x40);
615 b_adaption = (p[3] & 0x20);
616 b_payload = (p[3] & 0x10);
618 /* Extract adaption field informations if any */
621 /* We don't have any adaptation_field, so payload start immediately
622 after the 4 byte TS header */
623 p_ts_packet->i_payload_start = 4;
627 /* p[4] is adaptation_field_length minus one */
628 p_ts_packet->i_payload_start = 5 + p[4];
630 /* The adaption field can be limited to the adaptation_field_length byte,
631 so that there is nothing to do: skip this possibility */
634 /* If the packet has both adaptation_field and payload, adaptation_field
635 cannot be more than 182 bytes long; if there is only an adaptation_field,
636 it must fill the next 183 bytes. */
637 if( b_payload ? (p[4] > 182) : (p[4] != 183) )
639 intf_DbgMsg("input debug: invalid TS adaptation field (%p)\n",
642 p_es_descriptor->c_invalid_packets++;
647 /* No we are sure that the byte containing flags is present: read it */
650 /* discontinuity_indicator */
653 intf_DbgMsg("discontinuity_indicator encountered by TS demux " \
654 "(position read: %d, saved: %d)\n", p[5] & 0x80,
655 p_es_descriptor->i_continuity_counter);
657 /* If the PID carries the PCR, there will be a system time-base
658 discontinuity. We let the PCR decoder handle that. */
659 p_es_descriptor->b_discontinuity = 1;
661 /* There also may be a continuity_counter discontinuity: resynchronise
662 our counter with the one of the stream */
663 p_es_descriptor->i_continuity_counter = (p[3] & 0x0f) - 1;
666 /* random_access_indicator */
667 p_es_descriptor->b_random |= p[5] & 0x40;
669 /* If this is a PCR_PID, and this TS packet contains a PCR, we pass it
670 along to the PCR decoder. */
671 if( (p_es_descriptor->b_pcr) && (p[5] & 0x10) )
673 /* There should be a PCR field in the packet, check if the adaption
674 field is long enough to carry it */
677 /* Call the PCR decoder */
678 input_PcrDecode( p_input, p_es_descriptor, &p[6] );
685 /* Check the continuity of the stream. */
686 i_dummy = ((p[3] & 0x0f) - p_es_descriptor->i_continuity_counter) & 0x0f;
689 /* Everything is ok, just increase our counter */
690 p_es_descriptor->i_continuity_counter++;
694 if( !b_payload && i_dummy == 0 )
696 /* This is a packet without payload, this is allowed by the draft
697 As there is nothing interessant in this packet (except PCR that
698 have already been handled), we can trash the packet. */
699 intf_DbgMsg("Packet without payload received by TS demux\n");
702 else if( i_dummy <= 0 )
704 /* Duplicate packet: mark it as being to be trashed. */
705 intf_DbgMsg("Duplicate packet received by TS demux\n");
710 /* This can indicate that we missed a packet or that the
711 continuity_counter wrapped and we received a dup packet: as we
712 don't know, do as if we missed a packet to be sure to recover
713 from this situation */
714 intf_DbgMsg("Packet lost by TS demux: current %d, packet %d\n",
715 p_es_descriptor->i_continuity_counter & 0x0f,
718 p_es_descriptor->i_continuity_counter = p[3] & 0x0f;
722 /* Trash the packet if it has no payload or if it is bad */
725 input_NetlistFreeTS( p_input, p_ts_packet );
727 p_input->c_ts_packets_trashed++;
732 if( p_es_descriptor->b_psi )
734 /* The payload contains PSI tables */
735 input_DemuxPSI( p_input, p_ts_packet, p_es_descriptor,
736 b_unit_start, b_lost );
740 /* The payload carries a PES stream */
741 input_DemuxPES( p_input, p_ts_packet, p_es_descriptor,
742 b_unit_start, b_lost );
752 /*******************************************************************************
754 *******************************************************************************
755 * Gather a PES packet and analyzes its header.
756 *******************************************************************************/
757 static __inline__ void input_DemuxPES( input_thread_t *p_input,
758 ts_packet_t *p_ts_packet,
759 es_descriptor_t *p_es_descriptor,
760 boolean_t b_unit_start,
761 boolean_t b_packet_lost )
763 decoder_fifo_t * p_fifo;
764 u8 i_pes_header_size;
766 pes_packet_t* p_last_pes;
768 int i_ts_payload_size;
771 #define p_pes (p_es_descriptor->p_pes_packet)
775 ASSERT(p_es_descriptor);
777 // intf_DbgMsg("PES-demultiplexing %p (%p)\n", p_ts_packet, p_pes);
779 /* If we lost data, discard the PES packet we are trying to reassemble
780 if any and wait for the beginning of a new one in order to synchronise
782 if( b_packet_lost && p_pes != NULL )
784 intf_DbgMsg("PES %p trashed because of packet lost\n", p_pes);
785 input_NetlistFreePES( p_input, p_pes );
789 /* If the TS packet contains the begining of a new PES packet, and if we
790 were reassembling a PES packet, then the PES should be complete now,
791 so parse its header and give it to the decoders */
792 if( b_unit_start && p_pes != NULL )
794 // intf_DbgMsg("End of PES packet %p\n", p_pes);
796 /* Parse the header. The header has a variable length, but in order
797 to improve the algorithm, we will read the 14 bytes we may be
799 p_ts = p_pes->p_first_ts;
800 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
803 if(i_ts_payload_size >= PES_HEADER_SIZE)
805 /* This part of the header entirely fits in the payload of
806 the first TS packet */
807 p_pes->p_pes_header = &(p_ts->buffer[p_ts->i_payload_start]);
811 /* This part of the header does not fit in the current TS packet:
812 copy the part of the header we are interested in to the
813 p_pes_header_save buffer */
814 intf_DbgMsg("Code never tested encourtered, WARNING ! (benny)\n");
817 memcpy(p_pes->p_pes_header_save + i_dummy,
818 &p_ts->buffer[p_ts->i_payload_start], i_ts_payload_size);
819 i_dummy += i_ts_payload_size;
821 p_ts = p_ts->p_next_ts;
824 /* The payload of the PES packet is shorter than the 14 bytes
825 we would read. This means that high packet lost occured
826 so the PES won't be usefull for any decoder. Moreover,
827 this should never happen so we can trash the packet and
828 exit roughly without regrets */
829 intf_DbgMsg("PES packet too short: trashed\n");
830 input_NetlistFreePES( p_input, p_pes );
836 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
838 while(i_ts_payload_size + i_dummy < PES_HEADER_SIZE);
840 /* This last TS packet is partly header, partly payload, so just
841 copy the header part */
842 memcpy(p_pes->p_pes_header_save + i_dummy,
843 &p_ts->buffer[p_ts->i_payload_start],
844 PES_HEADER_SIZE - i_dummy);
846 /* The header must be read in the buffer not in any TS packet */
847 p_pes->p_pes_header = p_pes->p_pes_header_save;
850 /* Now we have the part of the PES header we were interested in:
853 /* First read the 6 header bytes common to all PES packets:
854 use them to test the PES validity */
855 if( (p_pes->p_pes_header[0] || p_pes->p_pes_header[1] ||
856 (p_pes->p_pes_header[2] != 1)) ||
857 /* packet_start_code_prefix != 0x000001 */
858 ((i_dummy = U16_AT(p_pes->p_pes_header + 4)) &&
859 (i_dummy + 6 != p_pes->i_pes_size)) )
860 /* PES_packet_length is set and != total received payload */
862 /* Trash the packet and set p_pes to NULL to be sure the next PES
863 packet will have its b_data_lost flag set */
864 intf_DbgMsg("Corrupted PES packet received: trashed\n");
865 input_NetlistFreePES( p_input, p_pes );
871 /* The PES packet is valid. Check its type to test if it may
872 carry additional informations in a header extension */
873 p_pes->i_stream_id = p_pes->p_pes_header[3];
875 switch( p_pes->i_stream_id )
877 case 0xBE: /* Padding */
878 case 0xBC: /* Program stream map */
879 case 0xBF: /* Private stream 2 */
882 case 0xFF: /* Program stream directory */
883 case 0xF2: /* DSMCC stream */
884 case 0xF8: /* ITU-T H.222.1 type E stream */
885 /* The payload begins immediatly after the 6 bytes header, so
886 we have finished with the parsing */
887 i_pes_header_size = 6;
891 /* The PES header contains at least 3 more bytes: parse them */
892 p_pes->b_data_alignment = p_pes->p_pes_header[6] & 0x10;
893 p_pes->b_has_pts = p_pes->p_pes_header[7] & 0x4;
894 i_pes_header_size = 9 + p_pes->p_pes_header[8];
896 /* Now parse the optional header extensions (in the limit of
898 if( p_pes->b_has_pts )
900 pcr_descriptor_t *p_pcr;
901 /* The PTS field is split in 3 bit records. We have to add
902 them, and thereafter we substract the 2 marker_bits */
904 p_pcr = p_input->p_pcr;
905 pthread_mutex_lock( &p_pcr->lock );
906 if( p_pcr->delta_clock == 0 )
912 p_pes->i_pts = ( ((mtime_t)p_pes->p_pes_header[9] << 29) +
913 ((mtime_t)U16_AT(p_pes->p_pes_header + 10) << 14) +
914 ((mtime_t)U16_AT(p_pes->p_pes_header + 12) >> 1) -
915 (1 << 14) - (1 << 29) );
918 p_pes->i_pts += p_pcr->delta_clock;
919 if( p_pcr->c_pts == 0 )
921 p_pcr->delta_decode = mdate() - p_pes->i_pts + 500000;
923 p_pes->i_pts += p_pcr->delta_decode;
926 pthread_mutex_unlock( &p_pcr->lock );
931 /* Now we've parsed the header, we just have to indicate in some
932 specific TS packets where the PES payload begins (renumber
933 i_payload_start), so that the decoders can find the beginning
934 of their data right out of the box. */
935 p_ts = p_pes->p_first_ts;
936 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
937 while( i_pes_header_size > i_ts_payload_size )
939 /* These packets are entirely filled by the PES header. */
940 i_pes_header_size -= i_ts_payload_size;
941 p_ts->i_payload_start = p_ts->i_payload_end;
942 /* Go to the next TS packet: here we won't have to test it is
943 not NULL because we trash the PES packets when packet lost
945 p_ts = p_ts->p_next_ts;
946 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
948 /* This last packet is partly header, partly payload. */
949 p_ts->i_payload_start += i_pes_header_size;
951 /* Now we can eventually put the PES packet in the decoder's
953 switch( p_es_descriptor->i_type )
957 p_fifo = &(((vdec_thread_t*)(p_es_descriptor->p_dec))->fifo);
961 p_fifo = &(((adec_thread_t*)(p_es_descriptor->p_dec))->fifo);
964 /* This should never happen. */
965 intf_DbgMsg("Unknown stream type (%d, %d): PES trashed\n",
966 p_es_descriptor->i_id, p_es_descriptor->i_type);
973 pthread_mutex_lock( &p_fifo->data_lock );
974 if( DECODER_FIFO_ISFULL( *p_fifo ) )
976 /* The FIFO is full !!! This should not happen. */
978 p_input->c_ts_packets_trashed += p_pes->i_ts_packets;
979 p_es_descriptor->c_invalid_packets += p_pes->i_ts_packets;
981 input_NetlistFreePES( p_input, p_pes );
982 intf_DbgMsg("PES trashed - fifo full ! (%d, %d)\n",
983 p_es_descriptor->i_id, p_es_descriptor->i_type);
987 // intf_DbgMsg("Putting %p into fifo %p/%d\n",
988 // p_pes, p_fifo, p_fifo->i_end);
989 p_fifo->buffer[p_fifo->i_end] = p_pes;
990 DECODER_FIFO_INCEND( *p_fifo );
992 /* Warn the decoder that it's got work to do. */
993 pthread_cond_signal( &p_fifo->data_wait );
995 pthread_mutex_unlock( &p_fifo->data_lock );
999 intf_DbgMsg("No fifo to receive PES %p: trash\n", p_pes);
1001 p_input->c_ts_packets_trashed += p_pes->i_ts_packets;
1002 p_es_descriptor->c_invalid_packets += p_pes->i_ts_packets;
1004 input_NetlistFreePES( p_input, p_pes );
1010 /* If we are at the beginning of a new PES packet, we must fetch a new
1011 PES buffer to begin with the reassembly of this PES packet. This is
1012 also here that we can synchronise with the stream if we we lost
1013 packets or if the decoder has just started */
1018 /* Get a new one PES from the PES netlist. */
1019 if( (p_pes = input_NetlistGetPES( p_input )) == (NULL) )
1021 /* PES netlist is empty ! */
1022 p_input->b_error = 1;
1026 // intf_DbgMsg("New PES packet %p (first TS: %p)\n", p_pes, p_ts_packet);
1028 /* Init the PES fields so that the first TS packet could be correctly
1029 added to the PES packet (see below) */
1030 p_pes->p_first_ts = p_ts_packet;
1031 p_pes->p_last_ts = NULL;
1033 /* If the last pes packet was null, this means that the synchronisation
1034 was lost and so warn the decoder that he will have to find a way to
1037 p_pes->b_data_loss = 1;
1039 /* Read the b_random_access flag status and then reinit it */
1040 p_pes->b_random_access = p_es_descriptor->b_random;
1041 p_es_descriptor->b_random = 0;
1046 /* If we are synchronised with the stream, and so if we are ready to
1047 receive correctly the data, add the TS packet to the current PES
1051 // intf_DbgMsg("Adding TS %p to PES %p\n", p_ts_packet, p_pes);
1053 /* Size of the payload carried in the TS packet */
1054 i_ts_payload_size = p_ts_packet->i_payload_end -
1055 p_ts_packet->i_payload_start;
1057 /* Update the relations between the TS packets */
1058 p_ts_packet->p_prev_ts = p_pes->p_last_ts;
1059 p_ts_packet->p_next_ts = NULL;
1060 if( p_pes->i_ts_packets != 0 )
1062 /* Regarder si il serait pas plus efficace de ne creer que les liens
1063 precedent->suivant pour le moment, et les liens suivant->precedent
1064 quand le paquet est termine */
1065 /* Otherwise it is the first TS packet. */
1066 p_pes->p_last_ts->p_next_ts = p_ts_packet;
1068 /* Now add the TS to the PES packet */
1069 p_pes->p_last_ts = p_ts_packet;
1070 p_pes->i_ts_packets++;
1071 p_pes->i_pes_size += i_ts_payload_size;
1075 i_dummy = p_ts_packet->i_payload_end - p_ts_packet->i_payload_start;
1076 p_es_descriptor->c_payload_bytes += i_dummy;
1081 /* Since we don't use the TS packet to build a PES packet, we don't
1082 need it anymore, so give it back to the netlist */
1083 // intf_DbgMsg("Trashing TS %p: no PES being build\n", p_ts_packet);
1084 input_NetlistFreeTS( p_input, p_ts_packet );
1093 /*******************************************************************************
1095 *******************************************************************************
1096 * Notice that current ES state has been locked by input_SortPacket. (No more true,
1097 * changed by benny - See if it'a ok, and definitely change the code ???????? )
1098 *******************************************************************************/
1099 static __inline__ void input_DemuxPSI( input_thread_t *p_input,
1100 ts_packet_t *p_ts_packet,
1101 es_descriptor_t *p_es_descriptor,
1102 boolean_t b_unit_start, boolean_t b_packet_lost )
1104 int i_data_offset; /* Offset of the interesting data in the TS packet */
1105 u16 i_data_length; /* Length of those data */
1106 boolean_t b_first_section; /* Was there another section in the TS packet ? */
1109 ASSERT(p_ts_packet);
1110 ASSERT(p_es_descriptor);
1112 #define p_psi (p_es_descriptor->p_psi_section)
1114 // intf_DbgMsg( "input debug: PSI demultiplexing %p (%p)\n", p_ts_packet, p_input);
1116 // 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);
1118 /* The section we will deal with during the first iteration of the following
1119 loop is the first one contained in the TS packet */
1120 b_first_section = 1;
1122 /* Reassemble the pieces of sections contained in the TS packet and decode
1123 the sections that could have been completed */
1126 /* Has the reassembly of a section already began in a previous packet ? */
1127 if( p_psi->b_running_section )
1129 /* Was data lost since the last TS packet ? */
1132 /* Discard the section and wait for the begining of a new one to resynch */
1133 p_psi->b_running_section = 0;
1134 intf_DbgMsg( "Section discarded due to packet loss\n" );
1138 /* The data that complete a previously began section are always at
1139 the beginning of the TS payload... */
1140 i_data_offset = p_ts_packet->i_payload_start;
1141 /* ...Unless there is a pointer field, that we have to bypass */
1144 // intf_DbgMsg( "New part of the section received at offset %d\n", i_data_offset );
1147 /* We are looking for the beginning of a new section */
1152 /* Cannot do anything with those data: trash both PSI section and TS packet */
1153 p_psi->b_running_section = 0;
1158 /* Get the offset at which the data for that section can be found */
1159 if( b_first_section )
1161 /* The offset is stored in the pointer_field since we are interested in
1162 the first section of the TS packet. Note that the +1 is to bypass
1163 the pointer field */
1164 i_data_offset = p_ts_packet->i_payload_start +
1165 p_ts_packet->buffer[p_ts_packet->i_payload_start] + 1;
1169 /* Since no gap is allowed between 2 sections in a TS packet, the
1170 offset is given by the end of the previous section. In fact, there
1171 is nothing to do, i_offset was set to the right value in the
1172 previous iteration */
1174 // intf_DbgMsg( "New section beginning at offset %d in TS packet\n", i_data_offset );
1176 /* Read the length of that section */
1177 p_psi->i_length = (U16_AT(&p_ts_packet->buffer[i_data_offset+1]) & 0xFFF) + 3;
1178 // intf_DbgMsg( "Section length %d\n", p_psi->i_length );
1179 if( p_psi->i_length > PSI_SECTION_SIZE )
1181 /* The TS packet is corrupted, stop here to avoid possible a seg fault */
1182 intf_DbgMsg( "Section size is too big, aborting its reception\n" );
1186 /* Init the reassembly of that section */
1187 p_psi->b_running_section = 1;
1188 p_psi->i_current_position = 0;
1192 /* Compute the length of data related to the section in this TS packet */
1193 if( p_psi->i_length - p_psi->i_current_position > TS_PACKET_SIZE - i_data_offset)
1194 i_data_length = TS_PACKET_SIZE - i_data_offset;
1196 i_data_length = p_psi->i_length - p_psi->i_current_position;
1198 /* Copy those data in the section buffer */
1199 memcpy( &p_psi->buffer[p_psi->i_current_position], &p_ts_packet->buffer[i_data_offset],
1202 /* Interesting data are now after the ones we copied */
1203 i_data_offset += i_data_length;
1205 /* Decode the packet if it is now complete */
1206 if (p_psi->i_length == p_psi->i_current_position + i_data_length)
1208 /* Packet is complete, decode it */
1209 // intf_DbgMsg( "SECTION COMPLETE: starting decoding of its data\n" );
1210 input_PsiDecode( p_input, p_psi );
1212 /* Prepare the buffer to receive a new section */
1213 p_psi->i_current_position = 0;
1214 p_psi->b_running_section = 0;
1216 /* The new section won't be the first anymore */
1217 b_first_section = 0;
1221 /* Prepare the buffer to receive the next part of the section */
1222 p_psi->i_current_position += i_data_length;
1223 // intf_DbgMsg( "Section not complete, waiting for the end\n" );
1226 // intf_DbgMsg( "Must loop ? Next data offset: %d, stuffing: %d\n",
1227 // i_data_offset, p_ts_packet->buffer[i_data_offset] );
1229 /* Stop if we reached the end of the packet or stuffing bytes */
1230 while( i_data_offset < TS_PACKET_SIZE && p_ts_packet->buffer[i_data_offset] != 0xFF );
1232 /* Relase the TS packet, we don't need it anymore */
1233 input_NetlistFreeTS( p_input, p_ts_packet );