1 /*****************************************************************************
2 * input_ps.c: PS demux and packet management
3 *****************************************************************************
4 * Copyright (C) 1998, 1999, 2000 VideoLAN
5 * $Id: input_ps.c,v 1.24 2001/05/23 17:47:34 stef Exp $
7 * Authors: Christophe Massiot <massiot@via.ecp.fr>
8 * Cyril Deguet <asmax@via.ecp.fr>
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
18 * GNU General Public License for more details.
20 * You should have received a copy of the GNU General Public License
21 * along with this program; if not, write to the Free Software
22 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
23 *****************************************************************************/
25 #define MODULE_NAME ps
26 #include "modules_inner.h"
28 /*****************************************************************************
30 *****************************************************************************/
35 #ifdef STRNCASECMP_IN_STRINGS_H
40 #include <sys/types.h>
55 #include "stream_control.h"
56 #include "input_ext-intf.h"
57 #include "input_ext-dec.h"
62 #include "mpeg_system.h"
68 /*****************************************************************************
70 *****************************************************************************/
71 static int PSProbe ( probedata_t * );
72 static int PSRead ( struct input_thread_s *,
73 data_packet_t * p_packets[INPUT_READ_ONCE] );
74 static void PSInit ( struct input_thread_s * );
75 static void PSEnd ( struct input_thread_s * );
76 static void PSSeek ( struct input_thread_s *, off_t );
77 static struct pes_packet_s * NewPES ( void * );
78 static struct data_packet_s * NewPacket ( void *, size_t );
79 static void DeletePacket( void *, struct data_packet_s * );
80 static void DeletePES ( void *, struct pes_packet_s * );
82 /*****************************************************************************
83 * Functions exported as capabilities. They are declared as static so that
84 * we don't pollute the namespace too much.
85 *****************************************************************************/
86 void _M( input_getfunctions )( function_list_t * p_function_list )
88 #define input p_function_list->functions.input
89 p_function_list->pf_probe = PSProbe;
90 input.pf_init = PSInit;
91 input.pf_open = input_FileOpen;
92 input.pf_close = input_FileClose;
94 input.pf_set_area = NULL;
95 input.pf_read = PSRead;
96 input.pf_demux = input_DemuxPS;
97 input.pf_new_packet = NewPacket;
98 input.pf_new_pes = NewPES;
99 input.pf_delete_packet = DeletePacket;
100 input.pf_delete_pes = DeletePES;
101 input.pf_rewind = NULL;
102 input.pf_seek = PSSeek;
107 * Data reading functions
110 /*****************************************************************************
111 * PSProbe: verifies that the stream is a PS stream
112 *****************************************************************************/
113 static int PSProbe( probedata_t *p_data )
115 input_thread_t * p_input = (input_thread_t *)p_data;
117 char * psz_name = p_input->p_source;
121 if( TestMethod( INPUT_METHOD_VAR, "ps" ) )
126 if( ( strlen(psz_name) > 5 ) && !strncasecmp( psz_name, "file:", 5 ) )
128 /* If the user specified "file:" then it's probably a file */
133 i_handle = open( psz_name, 0 );
143 /*****************************************************************************
144 * PSInit: initializes PS structures
145 *****************************************************************************/
146 static void PSInit( input_thread_t * p_input )
148 thread_ps_data_t * p_method;
149 packet_cache_t * p_packet_cache;
152 (thread_ps_data_t *)malloc( sizeof(thread_ps_data_t) )) == NULL )
154 intf_ErrMsg( "Out of memory" );
155 p_input->b_error = 1;
158 p_input->p_plugin_data = (void *)p_method;
160 /* creates the packet cache structure */
161 p_packet_cache = malloc( sizeof(packet_cache_t) );
162 if ( p_packet_cache == NULL )
164 intf_ErrMsg( "Out of memory" );
165 p_input->b_error = 1;
168 p_input->p_method_data = (void *)p_packet_cache;
170 /* Initialize packet cache mutex */
171 vlc_mutex_init( &p_packet_cache->lock );
173 /* allocates the data cache */
174 p_packet_cache->data.p_stack = malloc( DATA_CACHE_SIZE *
175 sizeof(data_packet_t*) );
176 if ( p_packet_cache->data.p_stack == NULL )
178 intf_ErrMsg( "Out of memory" );
179 p_input->b_error = 1;
182 p_packet_cache->data.l_index = 0;
184 /* allocates the PES cache */
185 p_packet_cache->pes.p_stack = malloc( PES_CACHE_SIZE *
186 sizeof(pes_packet_t*) );
187 if ( p_packet_cache->pes.p_stack == NULL )
189 intf_ErrMsg( "Out of memory" );
190 p_input->b_error = 1;
193 p_packet_cache->pes.l_index = 0;
195 /* allocates the small buffer cache */
196 p_packet_cache->small.p_stack = malloc( SMALL_CACHE_SIZE *
197 sizeof(packet_buffer_t) );
198 if ( p_packet_cache->small.p_stack == NULL )
200 intf_ErrMsg( "Out of memory" );
201 p_input->b_error = 1;
204 p_packet_cache->small.l_index = 0;
206 /* allocates the large buffer cache */
207 p_packet_cache->large.p_stack = malloc( LARGE_CACHE_SIZE *
208 sizeof(packet_buffer_t) );
209 if ( p_packet_cache->large.p_stack == NULL )
211 intf_ErrMsg( "Out of memory" );
212 p_input->b_error = 1;
215 p_packet_cache->large.l_index = 0;
217 /* Re-open the socket as a buffered FILE stream */
218 if( (p_method->stream = fdopen( p_input->i_handle, "r" )) == NULL )
220 intf_ErrMsg( "Cannot open file (%s)", strerror(errno) );
221 p_input->b_error = 1;
224 rewind( p_method->stream );
226 /* FIXME : detect if InitStream failed */
227 input_InitStream( p_input, sizeof( stream_ps_data_t ) );
228 input_AddProgram( p_input, 0, sizeof( stream_ps_data_t ) );
230 if( p_input->stream.b_seekable )
232 stream_ps_data_t * p_demux_data =
233 (stream_ps_data_t *)p_input->stream.pp_programs[0]->p_demux_data;
235 /* Pre-parse the stream to gather stream_descriptor_t. */
236 p_input->stream.pp_programs[0]->b_is_ok = 0;
237 p_demux_data->i_PSM_version = EMPTY_PSM_VERSION;
239 while( !p_input->b_die && !p_input->b_error
240 && !p_demux_data->b_has_PSM )
243 data_packet_t * pp_packets[INPUT_READ_ONCE];
245 i_result = PSRead( p_input, pp_packets );
249 vlc_mutex_lock( &p_input->stream.stream_lock );
250 p_input->stream.pp_programs[0]->b_is_ok = 1;
251 vlc_mutex_unlock( &p_input->stream.stream_lock );
256 p_input->b_error = 1;
260 for( i = 0; i < INPUT_READ_ONCE && pp_packets[i] != NULL; i++ )
262 /* FIXME: use i_p_config_t */
263 input_ParsePS( p_input, pp_packets[i] );
264 DeletePacket( p_input->p_method_data, pp_packets[i] );
268 if( p_input->stream.p_selected_area->i_tell >
269 INPUT_PREPARSE_LENGTH )
274 rewind( p_method->stream );
275 vlc_mutex_lock( &p_input->stream.stream_lock );
277 p_input->stream.i_method = INPUT_METHOD_FILE;
278 p_input->stream.p_selected_area->i_tell = 0;
280 if( p_demux_data->b_has_PSM )
282 /* (The PSM decoder will care about spawning the decoders) */
283 p_input->stream.pp_programs[0]->b_is_ok = 1;
288 /* (We have to do it ourselves) */
291 /* FIXME: we should do multiple passes in case an audio type
294 i_es < p_input->stream.pp_programs[0]->i_es_number;
297 #define p_es p_input->stream.pp_programs[0]->pp_es[i_es]
298 switch( p_es->i_type )
302 input_SelectES( p_input, p_es );
307 if( main_GetIntVariable( INPUT_CHANNEL_VAR, 0 )
308 == (p_es->i_id & 0x1F) )
309 switch( main_GetIntVariable( INPUT_AUDIO_VAR, 0 ) )
312 main_PutIntVariable( INPUT_AUDIO_VAR,
315 input_SelectES( p_input, p_es );
320 if( main_GetIntVariable( INPUT_CHANNEL_VAR, 0 )
321 == ((p_es->i_id & 0xF00) >> 8) )
322 switch( main_GetIntVariable( INPUT_AUDIO_VAR, 0 ) )
325 main_PutIntVariable( INPUT_AUDIO_VAR,
328 input_SelectES( p_input, p_es );
333 if( main_GetIntVariable( INPUT_SUBTITLE_VAR, -1 )
334 == ((p_es->i_id & 0x1F00) >> 8) )
336 input_SelectES( p_input, p_es );
341 if( main_GetIntVariable( INPUT_CHANNEL_VAR, 0 )
342 == ((p_es->i_id & 0x1F00) >> 8) )
343 switch( main_GetIntVariable( INPUT_AUDIO_VAR, 0 ) )
346 main_PutIntVariable( INPUT_AUDIO_VAR,
349 input_SelectES( p_input, p_es );
358 input_DumpStream( p_input );
360 vlc_mutex_unlock( &p_input->stream.stream_lock );
364 /* The programs will be added when we read them. */
365 vlc_mutex_lock( &p_input->stream.stream_lock );
366 p_input->stream.i_method = INPUT_METHOD_FILE;
367 p_input->stream.pp_programs[0]->b_is_ok = 0;
368 vlc_mutex_unlock( &p_input->stream.stream_lock );
372 /*****************************************************************************
373 * PSEnd: frees unused data
374 *****************************************************************************/
375 static void PSEnd( input_thread_t * p_input )
377 vlc_mutex_destroy( &((packet_cache_t *)p_input->p_plugin_data)->lock );
378 free( p_input->p_plugin_data );
381 /*****************************************************************************
382 * SafeRead: reads a chunk of stream and correctly detects errors
383 *****************************************************************************/
384 static __inline__ int SafeRead( input_thread_t * p_input, byte_t * p_buffer,
387 thread_ps_data_t * p_method;
390 p_method = (thread_ps_data_t *)p_input->p_plugin_data;
391 while( fread( p_buffer, i_len, 1, p_method->stream ) != 1 )
393 if( feof( p_method->stream ) )
398 if( (i_error = ferror( p_method->stream )) )
400 intf_ErrMsg( "Read failed (%s)", strerror(i_error) );
404 vlc_mutex_lock( &p_input->stream.stream_lock );
405 p_input->stream.p_selected_area->i_tell += i_len;
406 vlc_mutex_unlock( &p_input->stream.stream_lock );
410 /*****************************************************************************
411 * PSRead: reads data packets
412 *****************************************************************************
413 * Returns -1 in case of error, 0 if everything went well, and 1 in case of
415 *****************************************************************************/
416 static int PSRead( input_thread_t * p_input,
417 data_packet_t * pp_packets[INPUT_READ_ONCE] )
420 data_packet_t * p_data;
421 size_t i_packet_size;
422 int i_packet, i_error;
423 thread_ps_data_t * p_method;
425 p_method = (thread_ps_data_t *)p_input->p_plugin_data;
427 memset( pp_packets, 0, INPUT_READ_ONCE * sizeof(data_packet_t *) );
428 for( i_packet = 0; i_packet < INPUT_READ_ONCE; i_packet++ )
430 /* Read what we believe to be a packet header. */
431 if( (i_error = SafeRead( p_input, p_header, 6 )) )
436 if( (U32_AT(p_header) & 0xFFFFFF00) != 0x100L )
438 /* This is not the startcode of a packet. Read the stream
439 * until we find one. */
440 u32 i_startcode = U32_AT(p_header);
445 /* It is common for MPEG-1 streams to pad with zeros
446 * (although it is forbidden by the recommendation), so
447 * don't bother everybody in this case. */
448 intf_WarnMsg( 3, "Garbage at input (%.8x)", i_startcode );
451 while( (i_startcode & 0xFFFFFF00) != 0x100L )
454 if( (i_dummy = getc( p_method->stream )) != EOF )
456 i_startcode |= i_dummy;
464 *(u32 *)p_header = U32_AT(&i_startcode);
465 if( (i_error = SafeRead( p_input, p_header + 4, 2 )) )
471 if( U32_AT(p_header) != 0x1BA )
473 /* That's the case for all packets, except pack header. */
474 i_packet_size = U16_AT(&p_header[4]);
479 if( (p_header[4] & 0xC0) == 0x40 )
484 else if( (p_header[4] & 0xF0) == 0x20 )
491 intf_ErrMsg( "Unable to determine stream type" );
496 /* Fetch a packet of the appropriate size. */
497 if( (p_data = NewPacket( p_input->p_method_data, i_packet_size + 6 ))
500 intf_ErrMsg( "Out of memory" );
504 /* Copy the header we already read. */
505 memcpy( p_data->p_buffer, p_header, 6 );
507 /* Read the remaining of the packet. */
508 if( i_packet_size && (i_error =
509 SafeRead( p_input, p_data->p_buffer + 6, i_packet_size )) )
514 /* In MPEG-2 pack headers we still have to read stuffing bytes. */
515 if( U32_AT(p_header) == 0x1BA )
517 if( i_packet_size == 8 && (p_data->p_buffer[13] & 0x7) != 0 )
519 /* MPEG-2 stuffing bytes */
521 if( (i_error = SafeRead( p_input, p_garbage,
522 p_data->p_buffer[13] & 0x7)) )
529 /* Give the packet to the other input stages. */
530 pp_packets[i_packet] = p_data;
536 /*****************************************************************************
537 * PSSeek: changes the stream position indicator
538 *****************************************************************************/
539 static void PSSeek( input_thread_t * p_input, off_t i_position )
541 thread_ps_data_t * p_method;
543 p_method = (thread_ps_data_t *)p_input->p_plugin_data;
545 /* A little bourrin but should work for a while --Meuuh */
547 fseeko( p_method->stream, i_position, SEEK_SET );
549 fseek( p_method->stream, (long)i_position, SEEK_SET );
552 p_input->stream.p_selected_area->i_tell = i_position;
556 * Packet management utilities
560 /*****************************************************************************
561 * NewPacket: allocates a data packet
562 *****************************************************************************/
563 static struct data_packet_s * NewPacket( void * p_packet_cache,
566 packet_cache_t * p_cache;
567 data_packet_t * p_data;
570 p_cache = (packet_cache_t *)p_packet_cache;
573 if ( p_cache == NULL )
575 intf_ErrMsg( "PPacket cache not initialized" );
581 if( l_size > INPUT_MAX_PACKET_SIZE )
583 intf_ErrMsg( "Packet too big (%d)", l_size );
587 vlc_mutex_lock( &p_cache->lock );
589 /* Checks whether the data cache is empty */
590 if( p_cache->data.l_index == 0 )
592 /* Allocates a new packet */
593 if ( (p_data = malloc( sizeof(data_packet_t) )) == NULL )
595 intf_ErrMsg( "Out of memory" );
596 vlc_mutex_unlock( &p_cache->lock );
600 intf_DbgMsg( "PS input: data packet allocated" );
605 /* Takes the packet out from the cache */
606 if( (p_data = p_cache->data.p_stack[ -- p_cache->data.l_index ])
609 intf_ErrMsg( "NULL packet in the data cache" );
610 vlc_mutex_unlock( &p_cache->lock );
615 if( l_size < MAX_SMALL_SIZE )
619 /* Checks whether the buffer cache is empty */
620 if( p_cache->small.l_index == 0 )
622 /* Allocates a new packet */
623 if ( (p_data->p_buffer = malloc( l_size )) == NULL )
625 intf_DbgMsg( "Out of memory" );
627 vlc_mutex_unlock( &p_cache->lock );
631 intf_DbgMsg( "PS input: small buffer allocated" );
633 p_data->l_size = l_size;
637 /* Takes the packet out from the cache */
638 l_index = -- p_cache->small.l_index;
639 if( (p_data->p_buffer = p_cache->small.p_stack[l_index].p_data)
642 intf_ErrMsg( "NULL packet in the small buffer cache" );
644 vlc_mutex_unlock( &p_cache->lock );
647 /* Reallocates the packet if it is too small or too large */
648 if( p_cache->small.p_stack[l_index].l_size < l_size ||
649 p_cache->small.p_stack[l_index].l_size > 2*l_size )
651 p_data->p_buffer = realloc( p_data->p_buffer, l_size );
652 p_data->l_size = l_size;
656 p_data->l_size = p_cache->small.p_stack[l_index].l_size;
664 /* Checks whether the buffer cache is empty */
665 if( p_cache->large.l_index == 0 )
667 /* Allocates a new packet */
668 if ( (p_data->p_buffer = malloc( l_size )) == NULL )
670 intf_ErrMsg( "Out of memory" );
672 vlc_mutex_unlock( &p_cache->lock );
676 intf_DbgMsg( "PS input: large buffer allocated" );
678 p_data->l_size = l_size;
682 /* Takes the packet out from the cache */
683 l_index = -- p_cache->large.l_index;
684 if( (p_data->p_buffer = p_cache->large.p_stack[l_index].p_data)
687 intf_ErrMsg( "NULL packet in the small buffer cache" );
689 vlc_mutex_unlock( &p_cache->lock );
692 /* Reallocates the packet if it is too small or too large */
693 if( p_cache->large.p_stack[l_index].l_size < l_size ||
694 p_cache->large.p_stack[l_index].l_size > 2*l_size )
696 p_data->p_buffer = realloc( p_data->p_buffer, l_size );
697 p_data->l_size = l_size;
701 p_data->l_size = p_cache->large.p_stack[l_index].l_size;
706 vlc_mutex_unlock( &p_cache->lock );
708 /* Initialize data */
709 p_data->p_next = NULL;
710 p_data->b_discard_payload = 0;
711 p_data->p_payload_start = p_data->p_buffer;
712 p_data->p_payload_end = p_data->p_buffer + l_size;
719 /*****************************************************************************
720 * NewPES: allocates a pes packet
721 *****************************************************************************/
722 static pes_packet_t * NewPES( void * p_packet_cache )
724 packet_cache_t * p_cache;
725 pes_packet_t * p_pes;
727 p_cache = (packet_cache_t *)p_packet_cache;
730 if ( p_cache == NULL )
732 intf_ErrMsg( "Packet cache not initialized" );
737 vlc_mutex_lock( &p_cache->lock );
739 /* Checks whether the PES cache is empty */
740 if( p_cache->pes.l_index == 0 )
742 /* Allocates a new packet */
743 if ( (p_pes = malloc( sizeof(pes_packet_t) )) == NULL )
745 intf_DbgMsg( "Out of memory" );
746 vlc_mutex_unlock( &p_cache->lock );
750 intf_DbgMsg( "PS input: PES packet allocated" );
755 /* Takes the packet out from the cache */
756 if( (p_pes = p_cache->pes.p_stack[ -- p_cache->pes.l_index ])
759 intf_ErrMsg( "NULL packet in the data cache" );
760 vlc_mutex_unlock( &p_cache->lock );
765 vlc_mutex_unlock( &p_cache->lock );
767 p_pes->b_data_alignment = p_pes->b_discontinuity =
768 p_pes->i_pts = p_pes->i_dts = 0;
769 p_pes->i_pes_size = 0;
770 p_pes->p_first = NULL;
776 /*****************************************************************************
777 * DeletePacket: deletes a data packet
778 *****************************************************************************/
779 static void DeletePacket( void * p_packet_cache,
780 data_packet_t * p_data )
782 packet_cache_t * p_cache;
784 p_cache = (packet_cache_t *)p_packet_cache;
787 if ( p_cache == NULL )
789 intf_ErrMsg( "Packet cache not initialized" );
796 vlc_mutex_lock( &p_cache->lock );
798 /* Checks whether the data cache is full */
799 if ( p_cache->data.l_index < DATA_CACHE_SIZE )
801 /* Cache not full: store the packet in it */
802 p_cache->data.p_stack[ p_cache->data.l_index ++ ] = p_data;
803 /* Small buffer or large buffer? */
804 if ( p_data->l_size < MAX_SMALL_SIZE )
806 /* Checks whether the small buffer cache is full */
807 if ( p_cache->small.l_index < SMALL_CACHE_SIZE )
809 p_cache->small.p_stack[ p_cache->small.l_index ].l_size =
811 p_cache->small.p_stack[ p_cache->small.l_index ++ ].p_data =
816 ASSERT( p_data->p_buffer );
817 free( p_data->p_buffer );
819 intf_DbgMsg( "PS input: small buffer freed" );
825 /* Checks whether the large buffer cache is full */
826 if ( p_cache->large.l_index < LARGE_CACHE_SIZE )
828 p_cache->large.p_stack[ p_cache->large.l_index ].l_size =
830 p_cache->large.p_stack[ p_cache->large.l_index ++ ].p_data =
835 ASSERT( p_data->p_buffer );
836 free( p_data->p_buffer );
838 intf_DbgMsg( "PS input: large buffer freed" );
845 /* Cache full: the packet must be freed */
846 free( p_data->p_buffer );
849 intf_DbgMsg( "PS input: data packet freed" );
853 vlc_mutex_unlock( &p_cache->lock );
856 /*****************************************************************************
857 * DeletePES: deletes a PES packet and associated data packets
858 *****************************************************************************/
859 static void DeletePES( void * p_packet_cache, pes_packet_t * p_pes )
861 packet_cache_t * p_cache;
862 data_packet_t * p_data;
863 data_packet_t * p_next;
865 p_cache = (packet_cache_t *)p_packet_cache;
868 if ( p_cache == NULL )
870 intf_ErrMsg( "Packet cache not initialized" );
877 p_data = p_pes->p_first;
879 while( p_data != NULL )
881 p_next = p_data->p_next;
882 DeletePacket( p_cache, p_data );
886 vlc_mutex_lock( &p_cache->lock );
888 /* Checks whether the PES cache is full */
889 if ( p_cache->pes.l_index < PES_CACHE_SIZE )
891 /* Cache not full: store the packet in it */
892 p_cache->pes.p_stack[ p_cache->pes.l_index ++ ] = p_pes;
896 /* Cache full: the packet must be freed */
899 intf_DbgMsg( "PS input: PES packet freed" );
903 vlc_mutex_unlock( &p_cache->lock );