1 /*****************************************************************************
2 * input_es.c: Elementary Stream demux and packet management
3 *****************************************************************************
4 * Copyright (C) 2001 VideoLAN
5 * $Id: input_es.c,v 1.5 2001/05/31 03:12:49 sam Exp $
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, write to the Free Software
21 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
22 *****************************************************************************/
24 #define MODULE_NAME es
25 #include "modules_inner.h"
27 /*****************************************************************************
29 *****************************************************************************/
36 #include <sys/types.h>
41 #elif defined( _MSC_VER ) && defined( _WIN32 )
57 #include "stream_control.h"
58 #include "input_ext-intf.h"
59 #include "input_ext-dec.h"
64 #include "mpeg_system.h"
69 #include "modules_export.h"
71 /*****************************************************************************
73 *****************************************************************************/
74 static int ESProbe ( probedata_t * );
75 static int ESRead ( struct input_thread_s *,
76 data_packet_t * p_packets[INPUT_READ_ONCE] );
77 static void ESInit ( struct input_thread_s * );
78 static void ESEnd ( struct input_thread_s * );
79 static void ESSeek ( struct input_thread_s *, off_t );
80 static void ESDemux ( struct input_thread_s *, struct data_packet_s * );
81 static struct pes_packet_s * NewPES ( void * );
82 static struct data_packet_s * NewPacket ( void *, size_t );
83 static void DeletePacket( void *, struct data_packet_s * );
84 static void DeletePES ( void *, struct pes_packet_s * );
86 /*****************************************************************************
87 * Functions exported as capabilities. They are declared as static so that
88 * we don't pollute the namespace too much.
89 *****************************************************************************/
90 void _M( input_getfunctions )( function_list_t * p_function_list )
92 #define input p_function_list->functions.input
93 p_function_list->pf_probe = ESProbe;
94 input.pf_init = ESInit;
95 input.pf_open = NULL; /* Set in ESInit */
96 input.pf_close = NULL;
98 input.pf_set_area = NULL;
99 input.pf_read = ESRead;
100 input.pf_demux = ESDemux;
101 input.pf_new_packet = NewPacket;
102 input.pf_new_pes = NewPES;
103 input.pf_delete_packet = DeletePacket;
104 input.pf_delete_pes = DeletePES;
105 input.pf_rewind = NULL;
106 input.pf_seek = ESSeek;
111 * Data reading functions
114 /*****************************************************************************
115 * ESProbe: verifies that the stream is a ES stream
116 *****************************************************************************/
117 static int ESProbe( probedata_t *p_data )
119 input_thread_t * p_input = (input_thread_t *)p_data;
121 char * psz_name = p_input->p_source;
125 if( TestMethod( INPUT_METHOD_VAR, "es" ) )
130 i_handle = open( psz_name, 0 );
140 /*****************************************************************************
141 * ESInit: initializes ES structures
142 *****************************************************************************/
143 static void ESInit( input_thread_t * p_input )
145 thread_es_data_t * p_method;
148 (thread_es_data_t *)malloc( sizeof(thread_es_data_t) )) == NULL )
150 intf_ErrMsg( "Out of memory" );
151 p_input->b_error = 1;
154 p_input->p_plugin_data = (void *)p_method;
156 p_input->pf_open = p_input->pf_file_open;
157 p_input->pf_close = p_input->pf_file_close;
160 /*****************************************************************************
161 * ESEnd: frees unused data
162 *****************************************************************************/
163 static void ESEnd( input_thread_t * p_input )
167 free( p_input->p_plugin_data );
170 /*****************************************************************************
171 * SafeRead: reads a chunk of stream and correctly detects errors
172 *****************************************************************************/
173 static __inline__ int SafeRead( input_thread_t * p_input, byte_t * p_buffer,
176 thread_es_data_t * p_method;
179 p_method = (thread_es_data_t *)p_input->p_plugin_data;
180 while( fread( p_buffer, i_len, 1, p_method->stream ) != 1 )
182 if( feof( p_method->stream ) )
187 if( (i_error = ferror( p_method->stream )) )
189 intf_ErrMsg( "Read failed (%s)", strerror(i_error) );
193 vlc_mutex_lock( &p_input->stream.stream_lock );
194 p_input->stream.p_selected_area->i_tell += i_len;
195 vlc_mutex_unlock( &p_input->stream.stream_lock );
200 /*****************************************************************************
201 * ESRead: reads data packets
202 *****************************************************************************
203 * Returns -1 in case of error, 0 if everything went well, and 1 in case of
205 *****************************************************************************/
206 static int ESRead( input_thread_t * p_input,
207 data_packet_t * pp_packets[INPUT_READ_ONCE] )
214 /*****************************************************************************
215 * ESSeek: changes the stream position indicator
216 *****************************************************************************/
217 static void ESSeek( input_thread_t * p_input, off_t i_position )
219 thread_es_data_t * p_method;
221 p_method = (thread_es_data_t *)p_input->p_plugin_data;
223 /* A little bourrin but should work for a while --Meuuh */
225 fseeko( p_method->stream, i_position, SEEK_SET );
227 fseek( p_method->stream, (long)i_position, SEEK_SET );
230 p_input->stream.p_selected_area->i_tell = i_position;
233 void ESDemux( input_thread_t * p_input, data_packet_t * p_data )
239 * Packet management utilities
242 /*****************************************************************************
243 * NewPacket: allocates a data packet
244 *****************************************************************************/
245 static struct data_packet_s * NewPacket( void * p_packet_cache,
254 /*****************************************************************************
255 * NewPES: allocates a pes packet
256 *****************************************************************************/
257 static pes_packet_t * NewPES( void * p_packet_cache )
264 /*****************************************************************************
265 * DeletePacket: deletes a data packet
266 *****************************************************************************/
267 static void DeletePacket( void * p_packet_cache,
268 data_packet_t * p_data )
273 /*****************************************************************************
274 * DeletePES: deletes a PES packet and associated data packets
275 *****************************************************************************/
276 static void DeletePES( void * p_packet_cache, pes_packet_t * p_pes )