1 /*****************************************************************************
2 * input.c: input thread
3 * Read an MPEG2 stream, demultiplex and parse it before sending it to
5 *****************************************************************************
6 * Copyright (C) 1998, 1999, 2000 VideoLAN
10 * This program is free software; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation; either version 2 of the License, or
13 * (at your option) any later version.
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 * General Public License for more details.
20 * You should have received a copy of the GNU General Public
21 * License along with this program; if not, write to the
22 * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
23 * Boston, MA 02111-1307, USA.
24 *****************************************************************************/
26 /*****************************************************************************
28 *****************************************************************************/
31 #include <errno.h> /* errno */
32 #include <sys/types.h> /* on BSD, uio.h needs types.h */
33 #include <sys/uio.h> /* "input.h" */
34 #include <string.h> /* strerror() */
36 #include <stdlib.h> /* free() */
37 #include <netinet/in.h> /* ntohs() */
48 #include "input_psi.h"
49 #include "input_pcr.h"
50 #include "input_netlist.h"
51 #include "decoder_fifo.h"
52 #include "input_file.h"
53 #include "input_network.h"
55 #include "audio_output.h" /* aout_thread_t */
57 #include "audio_decoder.h" /* audiodec_t (for audio_decoder_thread.h) */
58 #include "audio_decoder_thread.h" /* adec_thread_t */
60 #include "ac3_decoder.h" /* ac3dec_t (for ac3_decoder_thread.h) */
61 #include "ac3_decoder_thread.h" /* ac3dec_thread_t */
63 #include "video.h" /* picture_t (for video_output.h) */
64 #include "video_output.h" /* vout_thread_t */
66 #include "vdec_idct.h" /* dctelem_t (for video_parser.h) */
67 #include "vdec_motion.h" /* f_motion_t (for video_parser.h) */
68 #include "vpar_blocks.h" /* macroblock_t (for video_parser.h) */
69 #include "vpar_headers.h" /* sequence_t (for video_parser.h) */
70 #include "vpar_synchro.h" /* video_synchro_t (for video_parser.h) */
71 #include "video_parser.h" /* vpar_thread_t */
73 #include "spu_decoder.h" /* spudec_thread_t */
77 /*****************************************************************************
79 *****************************************************************************/
80 static void RunThread ( input_thread_t *p_input );
81 static void ErrorThread ( input_thread_t *p_input );
82 static void EndThread ( input_thread_t *p_input );
84 static __inline__ int input_ReadPacket( input_thread_t *p_input );
85 static __inline__ void input_SortPacket( input_thread_t *p_input,
86 ts_packet_t *ts_packet );
87 static __inline__ void input_DemuxTS( input_thread_t *p_input,
88 ts_packet_t *ts_packet,
89 es_descriptor_t *es_descriptor );
90 static __inline__ void input_DemuxPES( input_thread_t *p_input,
91 ts_packet_t *ts_packet,
92 es_descriptor_t *p_es_descriptor,
93 boolean_t b_unit_start, boolean_t b_packet_lost );
94 static __inline__ void input_ParsePES( input_thread_t *p_input,
95 es_descriptor_t *p_es_descriptor );
96 static __inline__ void input_DemuxPSI( input_thread_t *p_input,
97 ts_packet_t *ts_packet,
98 es_descriptor_t *p_es_descriptor,
99 boolean_t b_unit_start, boolean_t b_packet_lost );
101 /*****************************************************************************
102 * input_CreateThread: creates a new input thread
103 *****************************************************************************
104 * This function creates a new input, and returns a pointer
105 * to its description. On error, it returns NULL.
106 * If pi_status is NULL, then the function will block until the thread is ready.
107 * If not, it will be updated using one of the THREAD_* constants.
108 *****************************************************************************/
109 input_thread_t *input_CreateThread ( int i_method, char *psz_source, int i_port, int i_vlan,
110 p_vout_thread_t p_vout, p_aout_thread_t p_aout, int *pi_status )
112 input_thread_t * p_input; /* thread descriptor */
113 int i_status; /* thread status */
114 int i_index; /* index for tables initialization */
116 /* Allocate descriptor */
118 p_input = (input_thread_t *)malloc( sizeof(input_thread_t) );
119 if( p_input == NULL )
121 intf_ErrMsg("error: %s\n", strerror(ENOMEM));
125 /* Initialize thread properties */
127 p_input->b_error = 0;
128 p_input->pi_status = (pi_status != NULL) ? pi_status : &i_status;
129 *p_input->pi_status = THREAD_CREATE;
131 /* Initialize input method description */
132 p_input->i_method = i_method;
133 p_input->psz_source = psz_source;
134 p_input->i_port = i_port;
135 p_input->i_vlan = i_vlan;
138 case INPUT_METHOD_TS_FILE: /* file methods */
139 p_input->p_Open = input_FileOpen;
140 p_input->p_Read = input_FileRead;
141 p_input->p_Close = input_FileClose;
143 case INPUT_METHOD_TS_VLAN_BCAST: /* vlan network method */
144 if( !p_main->b_vlans )
146 intf_ErrMsg("error: vlans are not activated\n");
150 /* ... pass through */
151 case INPUT_METHOD_TS_UCAST: /* network methods */
152 case INPUT_METHOD_TS_MCAST:
153 case INPUT_METHOD_TS_BCAST:
154 p_input->p_Open = input_NetworkOpen;
155 p_input->p_Read = input_NetworkRead;
156 p_input->p_Close = input_NetworkClose;
159 intf_ErrMsg("error: unknow input method\n");
165 /* Initialize stream description */
166 for( i_index = 0; i_index < INPUT_MAX_ES; i_index++ )
168 p_input->p_es[i_index].i_id = EMPTY_PID;
169 p_input->pp_selected_es[i_index] = NULL;
172 /* Initialize default settings for spawned decoders */
173 p_input->p_aout = p_aout;
174 p_input->p_vout = p_vout;
177 /* Initialize statistics */
178 p_input->c_loops = 0;
179 p_input->c_bytes = 0;
180 p_input->c_payload_bytes = 0;
181 p_input->c_packets_read = 0;
182 p_input->c_packets_trashed = 0;
185 /* Initialize PSI and PCR decoders */
186 if( input_PsiInit( p_input ) )
192 if( input_PcrInit( p_input ) )
194 input_PsiEnd( p_input );
199 /* Initialize netlists */
200 if( input_NetlistInit( p_input ) )
202 input_PsiEnd( p_input );
203 input_PcrEnd( p_input );
208 intf_DbgMsg("configuration: method=%d, source=%s, port=%d, vlan=%d\n",
209 i_method, psz_source, i_port, i_vlan );
211 /* Let the appropriate method open the socket. */
212 if( p_input->p_Open( p_input ) )
214 input_NetlistEnd( p_input );
215 input_PsiEnd( p_input );
216 input_PcrEnd( p_input );
221 /* Create thread and set locks. */
222 vlc_mutex_init( &p_input->netlist.lock );
223 vlc_mutex_init( &p_input->programs_lock );
224 vlc_mutex_init( &p_input->es_lock );
225 if( vlc_thread_create(&p_input->thread_id, "input", (void *) RunThread, (void *) p_input) )
227 intf_ErrMsg("error: %s\n", strerror(errno) );
228 p_input->p_Close( p_input );
229 input_NetlistEnd( p_input );;
230 input_PsiEnd( p_input );
231 input_PcrEnd( p_input );
236 intf_Msg("Input initialized\n");
238 /* If status is NULL, wait until the thread is created */
239 if( pi_status == NULL )
243 msleep( THREAD_SLEEP );
244 }while( (i_status != THREAD_READY) && (i_status != THREAD_ERROR)
245 && (i_status != THREAD_FATAL) );
246 if( i_status != THREAD_READY )
254 /*****************************************************************************
255 * input_DestroyThread: mark an input thread as zombie
256 *****************************************************************************
257 * This function should not return until the thread is effectively cancelled.
258 *****************************************************************************/
259 void input_DestroyThread( input_thread_t *p_input, int *pi_status )
261 int i_status; /* thread status */
264 p_input->pi_status = (pi_status != NULL) ? pi_status : &i_status;
265 *p_input->pi_status = THREAD_DESTROY;
267 /* Request thread destruction */
270 /* If status is NULL, wait until thread has been destroyed */
271 if( pi_status == NULL )
275 msleep( THREAD_SLEEP );
276 }while( (i_status != THREAD_OVER) && (i_status != THREAD_ERROR)
277 && (i_status != THREAD_FATAL) );
282 /*****************************************************************************
283 * input_OpenAudioStream: open an audio stream
284 *****************************************************************************
285 * This function spawns an audio decoder and plugs it on the audio output
287 *****************************************************************************/
288 int input_OpenAudioStream( input_thread_t *p_input, int i_id )
293 /*****************************************************************************
294 * input_CloseAudioStream: close an audio stream
295 *****************************************************************************
296 * This function destroys an audio decoder.
297 *****************************************************************************/
298 void input_CloseAudioStream( input_thread_t *p_input, int i_id )
303 /*****************************************************************************
304 * input_OpenVideoStream: open a video stream
305 *****************************************************************************
306 * This function spawns a video decoder and plugs it on a video output thread.
307 *****************************************************************************/
308 int input_OpenVideoStream( input_thread_t *p_input,
309 struct vout_thread_s *p_vout, struct video_cfg_s * p_cfg )
314 /*****************************************************************************
315 * input_CloseVideoStream: close a video stream
316 *****************************************************************************
317 * This function destroys an video decoder.
318 *****************************************************************************/
319 void input_CloseVideoStream( input_thread_t *p_input, int i_id )
325 /* following functions are local */
327 /*****************************************************************************
328 * InitThread: initialize input thread
329 *****************************************************************************
330 * This function is called from RunThread and performs the second step of the
331 * initialization. It returns 0 on success. Note that the thread's flag are not
332 * modified inside this function.
333 *****************************************************************************/
334 static int InitThread( input_thread_t *p_input )
336 /* Mark thread as running and return */
338 *p_input->pi_status = THREAD_READY;
339 intf_DbgMsg("thread ready\n");
343 /*****************************************************************************
344 * RunThread: main thread loop
345 *****************************************************************************
346 * Thread in charge of processing the network packets and demultiplexing.
347 *****************************************************************************/
348 static void RunThread( input_thread_t *p_input )
351 * Initialize thread and free configuration
353 p_input->b_error = InitThread( p_input );
354 if( p_input->b_error )
356 free( p_input ); /* destroy descriptor */
364 while( !p_input->b_die && !p_input->b_error )
366 /* Scatter read the UDP packet from the network or the file. */
367 if( (input_ReadPacket( p_input )) == (-1) )
369 /* FIXME??: Normally, a thread can't kill itself, but we don't have
370 * any method in case of an error condition ... */
371 p_input->b_error = 1;
382 if( p_input->b_error )
384 ErrorThread( p_input );
388 EndThread( p_input );
389 intf_DbgMsg("thread end\n");
393 /*****************************************************************************
394 * ErrorThread: RunThread() error loop
395 *****************************************************************************
396 * This function is called when an error occured during thread main's loop.
397 *****************************************************************************/
398 static void ErrorThread( input_thread_t *p_input )
400 /* Wait until a `die' order */
402 while( !p_input->b_die )
405 msleep( VOUT_IDLE_SLEEP );
409 /*****************************************************************************
410 * EndThread: end the input thread
411 *****************************************************************************/
412 static void EndThread( input_thread_t * p_input )
414 int * pi_status; /* threas status */
415 int i_es_loop; /* es index */
419 pi_status = p_input->pi_status;
420 *pi_status = THREAD_END;
422 /* Close input method */
423 p_input->p_Close( p_input );
425 /* Destroy all decoder threads */
427 (i_es_loop < INPUT_MAX_ES) && (p_input->pp_selected_es[i_es_loop] != NULL) ;
430 switch( p_input->pp_selected_es[i_es_loop]->i_type )
435 vdec_DestroyThread( (vdec_thread_t*)(p_input->pp_selected_es[i_es_loop]->p_dec) /*, NULL */ );
437 vpar_DestroyThread( (vpar_thread_t*)(p_input->pp_selected_es[i_es_loop]->p_dec) /*, NULL */ );
442 adec_DestroyThread( (adec_thread_t*)(p_input->pp_selected_es[i_es_loop]->p_dec) );
445 ac3dec_DestroyThread( (ac3dec_thread_t *)(p_input->pp_selected_es[i_es_loop]->p_dec) );
448 spudec_DestroyThread( (spudec_thread_t *)(p_input->pp_selected_es[i_es_loop]->p_dec) );
451 /* Special streams for the PSI decoder, PID 0 and 1 */
455 intf_DbgMsg("error: unknown decoder type %d\n", p_input->pp_selected_es[i_es_loop]->i_type );
461 input_NetlistEnd( p_input ); /* clean netlist */
462 input_PsiEnd( p_input ); /* clean PSI information */
463 input_PcrEnd( p_input ); /* clean PCR information */
464 free( p_input ); /* free input_thread structure */
467 *pi_status = THREAD_OVER;
470 /*****************************************************************************
471 * input_ReadPacket: reads a packet from the network or the file
472 *****************************************************************************/
473 static __inline__ int input_ReadPacket( input_thread_t *p_input )
475 int i_base_index; /* index of the first free iovec */
478 #ifdef INPUT_LIFO_TS_NETLIST
479 int i_meanwhile_released;
480 int i_currently_removed;
482 ts_packet_t * p_ts_packet;
484 /* In this function, we only care about the TS netlist. PES netlist
485 * is for the demultiplexer. */
486 #ifdef INPUT_LIFO_TS_NETLIST
487 i_base_index = p_input->netlist.i_ts_index;
489 /* Verify that we still have packets in the TS netlist */
490 if( (INPUT_MAX_TS + INPUT_TS_READ_ONCE - 1 - p_input->netlist.i_ts_index) <= INPUT_TS_READ_ONCE )
492 intf_ErrMsg("input error: TS netlist is empty !\n");
496 #else /* FIFO netlist */
497 i_base_index = p_input->netlist.i_ts_start;
498 if( p_input->netlist.i_ts_start + INPUT_TS_READ_ONCE -1 > INPUT_MAX_TS )
500 /* The netlist is splitted in 2 parts. We must gather them to consolidate
501 the FIFO (we make the loop easily in having the same iovec at the far
502 end and in the beginning of netlist_free).
503 That's why the netlist is (INPUT_MAX_TS +1) + (INPUT_TS_READ_ONCE -1)
505 memcpy( p_input->netlist.p_ts_free + INPUT_MAX_TS + 1,
506 p_input->netlist.p_ts_free,
507 (p_input->netlist.i_ts_start + INPUT_TS_READ_ONCE - 1 - INPUT_MAX_TS)
508 * sizeof(struct iovec) );
511 /* Verify that we still have packets in the TS netlist */
512 if( ((p_input->netlist.i_ts_end -1 - p_input->netlist.i_ts_start) & INPUT_MAX_TS) <= INPUT_TS_READ_ONCE )
514 intf_ErrMsg("input error: TS netlist is empty !\n");
517 #endif /* FIFO netlist */
519 /* Scatter read the buffer. */
520 i_packet_size = (*p_input->p_Read)( p_input,
521 &p_input->netlist.p_ts_free[i_base_index],
522 INPUT_TS_READ_ONCE );
523 if( i_packet_size == (-1) )
526 intf_DbgMsg("Read packet %d %p %d %d\n", i_base_index,
527 &p_input->netlist.p_ts_free[i_base_index],
528 p_input->netlist.i_ts_start,
529 p_input->netlist.i_ts_end);
531 intf_ErrMsg("input error: readv() failed (%s)\n", strerror(errno));
535 if( i_packet_size == 0 )
537 /* No packet has been received, so stop here. */
541 /* Demultiplex the TS packets (1..INPUT_TS_READ_ONCE) received. */
542 for( i_current_index = i_base_index;
543 (i_packet_size -= TS_PACKET_SIZE) >= 0;
546 /* BTW, something REALLY bad could happen if we receive packets with
548 p_ts_packet = (ts_packet_t*)(p_input->netlist.p_ts_free[i_current_index].iov_base);
549 /* Don't cry :-), we are allowed to do that cast, because initially,
550 our buffer was malloc'ed with sizeof(ts_packet_t) */
552 /* Find out if we need this packet and demultiplex. */
553 input_SortPacket( p_input /* for current PIDs and netlist */,
557 if( i_packet_size > 0 )
559 intf_ErrMsg("input error: wrong size\n");
563 /* Remove the TS packets we have just filled from the netlist */
564 #ifdef INPUT_LIFO_TS_NETLIST
565 /* We need to take a lock here while we're calculating index positions. */
566 vlc_mutex_lock( &p_input->netlist.lock );
568 i_meanwhile_released = i_base_index - p_input->netlist.i_ts_index;
569 if( i_meanwhile_released )
571 /* That's where it becomes funny :-). Since we didn't take locks for
572 efficiency reasons, other threads (including ourselves, with
573 input_DemuxPacket) might have released packets to the netlist.
574 So we have to copy these iovec where they should go.
576 BTW, that explains why the TS netlist is
577 (INPUT_MAX_TS +1) + (TS_READ_ONCE -1) large. */
579 i_currently_removed = i_current_index - i_base_index;
580 if( i_meanwhile_released < i_currently_removed )
582 /* Copy all iovecs in that case */
583 memcpy( &p_input->netlist.p_ts_free[p_input->netlist.i_ts_index]
584 + i_currently_removed,
585 &p_input->netlist.p_ts_free[p_input->netlist.i_ts_index],
586 i_meanwhile_released * sizeof(struct iovec) );
590 /* We have fewer places than items, so we only move
591 i_currently_removed of them. */
592 memcpy( &p_input->netlist.p_ts_free[i_base_index],
593 &p_input->netlist.p_ts_free[p_input->netlist.i_ts_index],
594 i_currently_removed * sizeof(struct iovec) );
597 /* Update i_netlist_index with the information gathered above. */
598 p_input->netlist.i_ts_index += i_currently_removed;
602 /* Nothing happened. */
603 p_input->netlist.i_ts_index = i_current_index;
606 vlc_mutex_unlock( &p_input->netlist.lock );
608 #else /* FIFO netlist */
609 /* & is modulo ; that's where we make the loop. */
610 p_input->netlist.i_ts_start = i_current_index & INPUT_MAX_TS;
614 p_input->c_packets_read += i_current_index - i_base_index;
615 p_input->c_bytes += (i_current_index - i_base_index) * TS_PACKET_SIZE;
620 /*****************************************************************************
621 * input_SortPacket: find out whether we need that packet
622 *****************************************************************************/
623 static __inline__ void input_SortPacket( input_thread_t *p_input,
624 ts_packet_t *p_ts_packet )
629 /* Verify that sync_byte, error_indicator and scrambling_control are
631 if( !(p_ts_packet->buffer[0] == 0x47) || (p_ts_packet->buffer[1] & 0x80) ||
632 (p_ts_packet->buffer[3] & 0xc0) )
634 intf_DbgMsg("input debug: invalid TS header (%p)\n", p_ts_packet);
638 /* Get the PID of the packet. Note that ntohs is needed, for endianness
639 purposes (see man page). */
640 i_current_pid = U16_AT(&p_ts_packet->buffer[1]) & 0x1fff;
642 //intf_DbgMsg("input debug: pid %d received (%p)\n",
643 // i_current_pid, p_ts_packet);
645 /* Lock current ES state. */
646 vlc_mutex_lock( &p_input->es_lock );
648 /* Verify that we actually want this PID. */
649 for( i_es_loop = 0; i_es_loop < INPUT_MAX_SELECTED_ES; i_es_loop++ )
651 if( p_input->pp_selected_es[i_es_loop] != NULL)
653 if( (*p_input->pp_selected_es[i_es_loop]).i_id
656 /* Don't need the lock anymore, since the value pointed
657 out by p_input->pp_selected_es[i_es_loop] can only be
658 modified from inside the input_thread (by the PSI
659 decoder): interface thread is only allowed to modify
660 the pp_selected_es table */
661 vlc_mutex_unlock( &p_input->es_lock );
663 /* We're interested. Pass it to the demultiplexer. */
664 input_DemuxTS( p_input, p_ts_packet,
665 p_input->pp_selected_es[i_es_loop] );
671 /* pp_selected_es should not contain any hole. */
675 vlc_mutex_unlock( &p_input->es_lock );
678 /* We weren't interested in receiving this packet. Give it back to the
680 //intf_DbgMsg("SortPacket: freeing unwanted TS %p (pid %d)\n", p_ts_packet,
681 // U16_AT(&p_ts_packet->buffer[1]) & 0x1fff);
682 input_NetlistFreeTS( p_input, p_ts_packet );
684 p_input->c_packets_trashed++;
688 /*****************************************************************************
689 * input_DemuxTS: first step of demultiplexing: the TS header
690 *****************************************************************************
691 * Stream must also only contain PES and PSI, so PID must have been filtered
692 *****************************************************************************/
693 static __inline__ void input_DemuxTS( input_thread_t *p_input,
694 ts_packet_t *p_ts_packet,
695 es_descriptor_t *p_es_descriptor )
698 boolean_t b_adaption; /* Adaption field is present */
699 boolean_t b_payload; /* Packet carries payload */
700 boolean_t b_unit_start; /* A PSI or a PES start in the packet */
701 boolean_t b_trash = 0; /* Must the packet be trashed ? */
702 boolean_t b_lost = 0; /* Was there a packet lost ? */
706 ASSERT(p_es_descriptor);
708 #define p (p_ts_packet->buffer)
710 //intf_DbgMsg("input debug: TS-demultiplexing packet %p, pid %d, number %d\n",
711 // p_ts_packet, U16_AT(&p[1]) & 0x1fff, p[3] & 0x0f);
714 p_es_descriptor->c_packets++;
715 p_es_descriptor->c_bytes += TS_PACKET_SIZE;
718 /* Extract flags values from TS common header. */
719 b_unit_start = (p[1] & 0x40);
720 b_adaption = (p[3] & 0x20);
721 b_payload = (p[3] & 0x10);
723 /* Extract adaption field informations if any */
726 /* We don't have any adaptation_field, so payload start immediately
727 after the 4 byte TS header */
728 p_ts_packet->i_payload_start = 4;
732 /* p[4] is adaptation_field_length minus one */
733 p_ts_packet->i_payload_start = 5 + p[4];
735 /* The adaption field can be limited to the adaptation_field_length byte,
736 so that there is nothing to do: skip this possibility */
739 /* If the packet has both adaptation_field and payload, adaptation_field
740 cannot be more than 182 bytes long; if there is only an
741 adaptation_field, it must fill the next 183 bytes. */
742 if( b_payload ? (p[4] > 182) : (p[4] != 183) )
744 intf_DbgMsg("input debug: invalid TS adaptation field (%p)\n",
747 p_es_descriptor->c_invalid_packets++;
752 /* No we are sure that the byte containing flags is present: read it */
755 /* discontinuity_indicator */
758 intf_DbgMsg("discontinuity_indicator encountered by TS demux " \
759 "(position read: %d, saved: %d)\n", p[5] & 0x80,
760 p_es_descriptor->i_continuity_counter);
762 /* If the PID carries the PCR, there will be a system time-base
763 discontinuity. We let the PCR decoder handle that. */
764 p_es_descriptor->b_discontinuity = 1;
766 /* There also may be a continuity_counter discontinuity:
767 resynchronise our counter with the one of the stream */
768 p_es_descriptor->i_continuity_counter = (p[3] & 0x0f) - 1;
771 /* random_access_indicator */
772 p_es_descriptor->b_random |= p[5] & 0x40;
774 /* If this is a PCR_PID, and this TS packet contains a PCR,
775 we pass it along to the PCR decoder. */
776 if( (p_es_descriptor->b_pcr) && (p[5] & 0x10) )
778 /* There should be a PCR field in the packet, check if the
779 adaption field is long enough to carry it */
782 /* Call the PCR decoder */
783 input_PcrDecode( p_input, p_es_descriptor, &p[6] );
790 /* Check the continuity of the stream. */
791 i_dummy = ((p[3] & 0x0f) - p_es_descriptor->i_continuity_counter) & 0x0f;
794 /* Everything is ok, just increase our counter */
795 p_es_descriptor->i_continuity_counter++;
799 if( !b_payload && i_dummy == 0 )
801 /* This is a packet without payload, this is allowed by the draft
802 As there is nothing interesting in this packet (except PCR that
803 have already been handled), we can trash the packet. */
804 intf_DbgMsg("Packet without payload received by TS demux\n");
807 else if( i_dummy <= 0 )
809 /* Duplicate packet: mark it as being to be trashed. */
810 intf_DbgMsg("Duplicate packet received by TS demux\n");
813 else if( p_es_descriptor->i_continuity_counter == 0xFF )
815 /* This means that the packet is the first one we receive for this
816 ES since the continuity counter ranges between 0 and 0x0F
817 excepts when it has been initialized by the input: Init the
818 counter to the correct value. */
819 intf_DbgMsg("First packet for PID %d received by TS demux\n",
820 p_es_descriptor->i_id);
821 p_es_descriptor->i_continuity_counter = (p[3] & 0x0f);
825 /* This can indicate that we missed a packet or that the
826 continuity_counter wrapped and we received a dup packet: as we
827 don't know, do as if we missed a packet to be sure to recover
828 from this situation */
829 intf_DbgMsg("Packet lost by TS demux: current %d, packet %d\n",
830 p_es_descriptor->i_continuity_counter & 0x0f,
833 p_es_descriptor->i_continuity_counter = p[3] & 0x0f;
837 /* Trash the packet if it has no payload or if it is bad */
840 input_NetlistFreeTS( p_input, p_ts_packet );
842 p_input->c_packets_trashed++;
847 if( p_es_descriptor->b_psi )
849 /* The payload contains PSI tables */
850 input_DemuxPSI( p_input, p_ts_packet, p_es_descriptor,
851 b_unit_start, b_lost );
855 /* The payload carries a PES stream */
856 input_DemuxPES( p_input, p_ts_packet, p_es_descriptor,
857 b_unit_start, b_lost );
867 /*****************************************************************************
869 *****************************************************************************
870 * Gather a PES packet.
871 *****************************************************************************/
872 static __inline__ void input_DemuxPES( input_thread_t *p_input,
873 ts_packet_t *p_ts_packet,
874 es_descriptor_t *p_es_descriptor,
875 boolean_t b_unit_start,
876 boolean_t b_packet_lost )
879 pes_packet_t* p_last_pes;
881 int i_ts_payload_size;
884 #define p_pes (p_es_descriptor->p_pes_packet)
888 ASSERT(p_es_descriptor);
890 //intf_DbgMsg("PES-demultiplexing %p (%p)\n", p_ts_packet, p_pes);
892 /* If we lost data, discard the PES packet we are trying to reassemble
893 if any and wait for the beginning of a new one in order to synchronise
895 if( b_packet_lost && p_pes != NULL )
897 intf_DbgMsg("PES %p trashed because of packet lost\n", p_pes);
898 input_NetlistFreePES( p_input, p_pes );
902 /* If the TS packet contains the begining of a new PES packet, and if we
903 were reassembling a PES packet, then the PES should be complete now,
904 so parse its header and give it to the decoders */
905 if( b_unit_start && p_pes != NULL )
907 /* Parse the header. The header has a variable length, but in order
908 to improve the algorithm, we will read the 14 bytes we may be
911 /* If this part of the header did not fit in the current TS packet,
912 copy the part of the header we are interested in to the
913 p_pes_header_save buffer. The buffer is dynamicly allocated if
914 needed so it's time expensive but this situation almost never
916 p_ts = p_pes->p_first_ts;
917 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
919 if(i_ts_payload_size < PES_HEADER_SIZE)
921 intf_DbgMsg("Code never tested encountered, WARNING ! (benny)\n");
922 if( !p_pes->p_pes_header_save )
924 p_pes->p_pes_header_save = malloc(PES_HEADER_SIZE);
930 memcpy(p_pes->p_pes_header_save + i_dummy,
931 &p_ts->buffer[p_ts->i_payload_start], i_ts_payload_size);
932 i_dummy += i_ts_payload_size;
934 p_ts = p_ts->p_next_ts;
937 /* The payload of the PES packet is shorter than the 14 bytes
938 we would read. This means that high packet lost occured
939 so the PES won't be useful for any decoder. Moreover,
940 this should never happen so we can trash the packet and
941 exit roughly without regrets */
942 intf_DbgMsg("PES packet too short: trashed\n");
943 input_NetlistFreePES( p_input, p_pes );
949 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
951 while(i_ts_payload_size + i_dummy < PES_HEADER_SIZE);
953 /* This last TS packet is partly header, partly payload, so just
954 copy the header part */
955 memcpy(p_pes->p_pes_header_save + i_dummy,
956 &p_ts->buffer[p_ts->i_payload_start],
957 PES_HEADER_SIZE - i_dummy);
959 /* The header must be read in the buffer not in any TS packet */
960 p_pes->p_pes_header = p_pes->p_pes_header_save;
962 /* Get the PES size if defined */
963 if( (i_dummy = U16_AT(p_pes->p_pes_header + 4)) )
965 p_pes->i_pes_real_size = i_dummy + 6;
969 /* Now we have the part of the PES header we were interested in:
970 p_pes_header and i_pes_real_size ; we can parse it */
971 input_ParsePES( p_input, p_es_descriptor );
974 /* If we are at the beginning of a new PES packet, we must fetch a new
975 PES buffer to begin with the reassembly of this PES packet. This is
976 also here that we can synchronise with the stream if we we lost
977 packets or if the decoder has just started */
982 /* Get a new one PES from the PES netlist. */
983 if( (p_pes = input_NetlistGetPES( p_input )) == (NULL) )
985 /* PES netlist is empty ! */
986 p_input->b_error = 1;
990 //intf_DbgMsg("New PES packet %p (first TS: %p)\n", p_pes, p_ts_packet);
992 /* Init the PES fields so that the first TS packet could be
993 * correctly added to the PES packet (see below) */
994 p_pes->p_first_ts = p_ts_packet;
995 p_pes->p_last_ts = NULL;
997 /* If the last pes packet was null, this means that the
998 * synchronization was lost and so warn the decoder that he
999 * will have to find a way to recover */
1001 p_pes->b_data_loss = 1;
1003 /* Tell the Demux we haven't yet parsed this PES */
1004 p_pes->b_already_parsed = 0;
1006 /* Read the b_random_access flag status and then reinit it */
1007 p_pes->b_random_access = p_es_descriptor->b_random;
1008 p_es_descriptor->b_random = 0;
1011 /* If the PES header fits in the first TS packet, we can
1012 * already set p_pes->p_pes_header, and in all cases we
1013 * set p_pes->i_pes_real_size */
1014 if( p_ts_packet->i_payload_end - p_ts_packet->i_payload_start
1015 >= PES_HEADER_SIZE )
1017 p_pes->p_pes_header = &(p_ts_packet->buffer[p_ts_packet->i_payload_start]);
1018 if( (i_dummy = U16_AT(p_pes->p_pes_header + 4)) )
1020 p_pes->i_pes_real_size = i_dummy + 6;
1026 /* If we are synchronized with the stream, and so if we are ready to
1027 receive correctly the data, add the TS packet to the current PES
1031 //intf_DbgMsg("Adding TS %p to PES %p\n", p_ts_packet, p_pes);
1033 /* Size of the payload carried in the TS packet */
1034 i_ts_payload_size = p_ts_packet->i_payload_end -
1035 p_ts_packet->i_payload_start;
1037 /* Update the relations between the TS packets */
1038 p_ts_packet->p_prev_ts = p_pes->p_last_ts;
1039 p_ts_packet->p_next_ts = NULL;
1040 if( p_pes->i_ts_packets != 0 )
1042 /* Regarder si il serait pas plus efficace de ne creer que
1043 * les liens precedent->suivant pour le moment, et les
1044 * liens suivant->precedent quand le paquet est termine */
1045 /* Otherwise it is the first TS packet. */
1046 p_pes->p_last_ts->p_next_ts = p_ts_packet;
1048 /* Now add the TS to the PES packet */
1049 p_pes->p_last_ts = p_ts_packet;
1050 p_pes->i_ts_packets++;
1051 p_pes->i_pes_size += i_ts_payload_size;
1055 i_dummy = p_ts_packet->i_payload_end - p_ts_packet->i_payload_start;
1056 p_es_descriptor->c_payload_bytes += i_dummy;
1059 /* We can check if the packet is finished */
1060 if( p_pes->i_pes_size == p_pes->i_pes_real_size )
1062 /* The packet is finished, parse it */
1063 input_ParsePES( p_input, p_es_descriptor );
1065 /* Tell the Demux we have parsed this PES, no need to redo it */
1071 /* Since we don't use the TS packet to build a PES packet, we don't
1072 need it anymore, so give it back to the netlist */
1073 //intf_DbgMsg("Trashing TS %p: no PES being build\n", p_ts_packet);
1074 input_NetlistFreeTS( p_input, p_ts_packet );
1082 /*****************************************************************************
1084 *****************************************************************************
1085 * Parse a finished PES packet and analyze its header.
1086 *****************************************************************************/
1087 static __inline__ void input_ParsePES( input_thread_t *p_input,
1088 es_descriptor_t *p_es_descriptor )
1090 decoder_fifo_t * p_fifo;
1091 u8 i_pes_header_size;
1093 int i_ts_payload_size;
1096 #define p_pes (p_es_descriptor->p_pes_packet)
1098 //intf_DbgMsg("End of PES packet %p\n", p_pes);
1100 /* First read the 6 header bytes common to all PES packets:
1101 use them to test the PES validity */
1102 if( (p_pes->p_pes_header[0] || p_pes->p_pes_header[1] ||
1103 (p_pes->p_pes_header[2] != 1)) ||
1104 /* packet_start_code_prefix != 0x000001 */
1105 ((p_pes->i_pes_real_size) &&
1106 (p_pes->i_pes_real_size != p_pes->i_pes_size)) )
1107 /* PES_packet_length is set and != total received payload */
1109 /* Trash the packet and set p_pes to NULL to be sure the next PES
1110 packet will have its b_data_lost flag set */
1111 intf_DbgMsg("Corrupted PES packet received: trashed\n");
1112 input_NetlistFreePES( p_input, p_pes );
1118 /* The PES packet is valid. Check its type to test if it may
1119 carry additional informations in a header extension */
1120 p_pes->i_stream_id = p_pes->p_pes_header[3];
1122 switch( p_pes->i_stream_id )
1124 case 0xBE: /* Padding */
1125 case 0xBC: /* Program stream map */
1126 case 0xBF: /* Private stream 2 */
1127 case 0xB0: /* ECM */
1128 case 0xB1: /* EMM */
1129 case 0xFF: /* Program stream directory */
1130 case 0xF2: /* DSMCC stream */
1131 case 0xF8: /* ITU-T H.222.1 type E stream */
1132 /* The payload begins immediatly after the 6 bytes header, so
1133 we have finished with the parsing */
1134 i_pes_header_size = 6;
1138 /* The PES header contains at least 3 more bytes: parse them */
1139 p_pes->b_data_alignment = p_pes->p_pes_header[6] & 0x04;
1140 p_pes->b_has_pts = p_pes->p_pes_header[7] & 0x80;
1141 i_pes_header_size = p_pes->p_pes_header[8] + 9;
1143 /* Now parse the optional header extensions (in the limit of
1145 if( p_pes->b_has_pts )
1147 pcr_descriptor_t * p_pcr;
1149 p_pcr = p_input->p_pcr;
1152 ( ((mtime_t)(p_pes->p_pes_header[9] & 0x0E) << 29) |
1153 (((mtime_t)U16_AT(p_pes->p_pes_header + 10) << 14) - (1 << 14)) |
1154 ((mtime_t)U16_AT(p_pes->p_pes_header + 12) >> 1) ) * 300;
1157 if( p_pcr->i_synchro_state )
1159 switch( p_pcr->i_synchro_state )
1161 case SYNCHRO_NOT_STARTED:
1162 p_pes->b_has_pts = 0;
1166 p_pes->i_pts += p_pcr->delta_pcr;
1167 p_pcr->delta_absolute = mdate() - p_pes->i_pts + INPUT_PTS_DELAY;
1168 p_pes->i_pts += p_pcr->delta_absolute;
1169 p_pcr->i_synchro_state = 0;
1172 case SYNCHRO_REINIT: /* We skip a PES */
1173 p_pes->b_has_pts = 0;
1174 p_pcr->i_synchro_state = SYNCHRO_START;
1180 p_pes->i_pts += p_pcr->delta_pcr + p_pcr->delta_absolute;
1186 /* Now we've parsed the header, we just have to indicate in some
1187 specific TS packets where the PES payload begins (renumber
1188 i_payload_start), so that the decoders can find the beginning
1189 of their data right out of the box. */
1190 p_ts = p_pes->p_first_ts;
1191 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
1192 while( i_pes_header_size > i_ts_payload_size )
1194 /* These packets are entirely filled by the PES header. */
1195 i_pes_header_size -= i_ts_payload_size;
1196 p_ts->i_payload_start = p_ts->i_payload_end;
1197 /* Go to the next TS packet: here we won't have to test it is
1198 not NULL because we trash the PES packets when packet lost
1200 p_ts = p_ts->p_next_ts;
1201 i_ts_payload_size = p_ts->i_payload_end - p_ts->i_payload_start;
1203 /* This last packet is partly header, partly payload. */
1204 p_ts->i_payload_start += i_pes_header_size;
1207 /* Now we can eventually put the PES packet in the decoder's
1209 switch( p_es_descriptor->i_type )
1211 case MPEG1_VIDEO_ES:
1212 case MPEG2_VIDEO_ES:
1214 p_fifo = &(((vdec_thread_t*)(p_es_descriptor->p_dec))->fifo);
1216 p_fifo = &(((vpar_thread_t*)(p_es_descriptor->p_dec))->fifo);
1220 case MPEG1_AUDIO_ES:
1221 case MPEG2_AUDIO_ES:
1222 p_fifo = &(((adec_thread_t*)(p_es_descriptor->p_dec))->fifo);
1226 /* we skip 4 bytes at the beginning of the AC3 payload */
1227 //p_ts->i_payload_start += 4;
1228 p_fifo = &(((ac3dec_thread_t *)(p_es_descriptor->p_dec))->fifo);
1232 /* we skip 4 bytes at the beginning of the subpicture payload */
1233 //p_ts->i_payload_start += 4;
1234 p_fifo = &(((spudec_thread_t *)(p_es_descriptor->p_dec))->fifo);
1238 /* This should never happen */
1239 intf_DbgMsg("Unknown stream type (%d, %d): PES trashed\n",
1240 p_es_descriptor->i_id, p_es_descriptor->i_type);
1245 if( p_fifo != NULL )
1247 vlc_mutex_lock( &p_fifo->data_lock );
1248 if( DECODER_FIFO_ISFULL( *p_fifo ) )
1250 /* The FIFO is full !!! This should not happen. */
1252 p_input->c_packets_trashed += p_pes->i_ts_packets;
1253 p_es_descriptor->c_invalid_packets += p_pes->i_ts_packets;
1255 input_NetlistFreePES( p_input, p_pes );
1256 intf_DbgMsg("PES trashed - fifo full ! (%d, %d)\n",
1257 p_es_descriptor->i_id, p_es_descriptor->i_type);
1261 //intf_DbgMsg("Putting %p into fifo %p/%d\n",
1262 // p_pes, p_fifo, p_fifo->i_end);
1263 p_fifo->buffer[p_fifo->i_end] = p_pes;
1264 DECODER_FIFO_INCEND( *p_fifo );
1266 /* Warn the decoder that it's got work to do. */
1267 vlc_cond_signal( &p_fifo->data_wait );
1269 vlc_mutex_unlock( &p_fifo->data_lock );
1273 intf_DbgMsg("No fifo to receive PES %p: trash\n", p_pes);
1275 p_input->c_packets_trashed += p_pes->i_ts_packets;
1276 p_es_descriptor->c_invalid_packets += p_pes->i_ts_packets;
1278 input_NetlistFreePES( p_input, p_pes );
1286 /*****************************************************************************
1288 *****************************************************************************
1289 * Notice that current ES state has been locked by input_SortPacket. (No more true,
1290 * changed by benny - FIXME: See if it's ok, and definitely change the code ?? )
1291 *****************************************************************************/
1292 static __inline__ void input_DemuxPSI( input_thread_t *p_input,
1293 ts_packet_t *p_ts_packet,
1294 es_descriptor_t *p_es_descriptor,
1295 boolean_t b_unit_start, boolean_t b_packet_lost )
1297 int i_data_offset; /* Offset of the interesting data in the TS packet */
1298 u16 i_data_length; /* Length of those data */
1299 //boolean_t b_first_section; /* Was there another section in the TS packet ? */
1302 ASSERT(p_ts_packet);
1303 ASSERT(p_es_descriptor);
1305 #define p_psi (p_es_descriptor->p_psi_section)
1307 //intf_DbgMsg( "input debug: PSI demultiplexing %p (%p)\n", p_ts_packet, p_input);
1309 //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);
1312 /* Try to find the beginning of the payload in the packet to initialise
1313 the do-while loop that follows -> Compute the i_data_offset variable:
1314 by default, the value is set so that we won't enter in the while loop.
1315 It will be set to a correct value if the data are not corrupted */
1316 i_data_offset = TS_PACKET_SIZE;
1318 /* Has the reassembly of a section already began in a previous packet ? */
1319 if( p_psi->b_running_section )
1321 /* Was data lost since the last TS packet ? */
1324 /* Discard the packet and wait for the begining of a new one to resynch */
1325 p_psi->b_running_section = 0;
1326 p_psi->i_current_position = 0;
1327 intf_DbgMsg( "PSI section(s) discarded due to packet loss\n" );
1331 /* The data that complete a previously began section are always at
1332 the beginning of the TS payload... */
1333 i_data_offset = p_ts_packet->i_payload_start;
1334 /* ...Unless there is a pointer field, that we have to bypass */
1337 //intf_DbgMsg( "New part of the section received at offset %d\n", i_data_offset );
1340 /* We are looking for the beginning of a new section */
1345 /* Get the offset at which the data for that section can be found
1346 The offset is stored in the pointer_field since we are interested in
1347 the first section of the TS packet. Note that the +1 is to bypass
1348 the pointer field */
1349 i_data_offset = p_ts_packet->i_payload_start +
1350 p_ts_packet->buffer[p_ts_packet->i_payload_start] + 1;
1351 //intf_DbgMsg( "New section beginning at offset %d in TS packet\n", i_data_offset );
1355 /* This may either mean that the TS is bad or that the packet contains
1356 the end of a section that had been discarded in a previous loop:
1357 trash the TS packet since we cannot do anything with those data: */
1358 p_psi->b_running_section = 0;
1359 p_psi->i_current_position = 0;
1360 intf_DbgMsg( "PSI packet discarded due to lack of synchronisation\n" );
1364 /* The section we will deal with during the first iteration of the following
1365 loop is the first one contained in the TS packet */
1366 // b_first_section = 1;
1368 /* Reassemble the pieces of sections contained in the TS packet and decode
1369 the sections that could have been completed.
1370 Stop when we reach the end of the packet or stuffing bytes */
1371 while( i_data_offset < TS_PACKET_SIZE && p_ts_packet->buffer[i_data_offset] != 0xFF )
1373 /* If the current section is a new one, reinit the data fields of the p_psi
1374 struct to start its decoding */
1375 if( !p_psi->b_running_section )
1377 /* Read the length of the new section */
1378 p_psi->i_length = (U16_AT(&p_ts_packet->buffer[i_data_offset+1]) & 0xFFF) + 3;
1379 //intf_DbgMsg( "Section length %d\n", p_psi->i_length );
1380 if( p_psi->i_length > PSI_SECTION_SIZE )
1382 /* The TS packet is corrupted, stop here to avoid possible a seg fault */
1383 intf_DbgMsg( "PSI Section size is too big, aborting its reception\n" );
1387 /* Init the reassembly of that section */
1388 p_psi->b_running_section = 1;
1389 p_psi->i_current_position = 0;
1392 /* Compute the length of data related to the section in this TS packet */
1393 if( p_psi->i_length - p_psi->i_current_position > TS_PACKET_SIZE - i_data_offset)
1394 i_data_length = TS_PACKET_SIZE - i_data_offset;
1396 i_data_length = p_psi->i_length - p_psi->i_current_position;
1398 /* Copy those data in the section buffer */
1399 memcpy( &p_psi->buffer[p_psi->i_current_position], &p_ts_packet->buffer[i_data_offset],
1402 /* Interesting data are now after the ones we copied, since no gap is
1403 allowed between 2 sections in a TS packets */
1404 i_data_offset += i_data_length;
1406 /* Decode the packet if it is now complete */
1407 if (p_psi->i_length == p_psi->i_current_position + i_data_length)
1409 /* Packet is complete, decode it */
1410 //intf_DbgMsg( "SECTION COMPLETE: starting decoding of its data\n" );
1411 input_PsiDecode( p_input, p_psi );
1413 /* Prepare the buffer to receive a new section */
1414 p_psi->i_current_position = 0;
1415 p_psi->b_running_section = 0;
1417 /* The new section won't be the first anymore */
1418 //b_first_section = 0;
1422 /* Prepare the buffer to receive the next part of the section */
1423 p_psi->i_current_position += i_data_length;
1424 //intf_DbgMsg( "Section not complete, waiting for the end\n" );
1427 //intf_DbgMsg( "Must loop ? Next data offset: %d, stuffing: %d\n",
1428 // i_data_offset, p_ts_packet->buffer[i_data_offset] );
1431 /* Relase the TS packet, we don't need it anymore */
1432 input_NetlistFreeTS( p_input, p_ts_packet );