1 /*****************************************************************************
2 * input_es.c: Elementary Stream demux and packet management
3 *****************************************************************************
4 * Copyright (C) 2001 VideoLAN
5 * $Id: input_es.c,v 1.2 2001/04/28 03:36:25 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>
51 #include "stream_control.h"
52 #include "input_ext-intf.h"
53 #include "input_ext-dec.h"
58 #include "mpeg_system.h"
64 /*****************************************************************************
66 *****************************************************************************/
67 static int ESProbe ( probedata_t * );
68 static int ESRead ( struct input_thread_s *,
69 data_packet_t * p_packets[INPUT_READ_ONCE] );
70 static void ESInit ( struct input_thread_s * );
71 static void ESEnd ( struct input_thread_s * );
72 static void ESSeek ( struct input_thread_s *, off_t );
73 static void ESDemux ( struct input_thread_s *, struct data_packet_s * );
74 static struct pes_packet_s * NewPES ( void * );
75 static struct data_packet_s * NewPacket ( void *, size_t );
76 static void DeletePacket( void *, struct data_packet_s * );
77 static void DeletePES ( void *, struct pes_packet_s * );
79 /*****************************************************************************
80 * Functions exported as capabilities. They are declared as static so that
81 * we don't pollute the namespace too much.
82 *****************************************************************************/
83 void _M( input_getfunctions )( function_list_t * p_function_list )
85 #define input p_function_list->functions.input
86 p_function_list->pf_probe = ESProbe;
87 input.pf_init = ESInit;
88 input.pf_open = input_FileOpen;
89 input.pf_close = input_FileClose;
91 input.pf_set_area = NULL;
92 input.pf_read = ESRead;
93 input.pf_demux = ESDemux;
94 input.pf_new_packet = NewPacket;
95 input.pf_new_pes = NewPES;
96 input.pf_delete_packet = DeletePacket;
97 input.pf_delete_pes = DeletePES;
98 input.pf_rewind = NULL;
99 input.pf_seek = ESSeek;
104 * Data reading functions
107 /*****************************************************************************
108 * ESProbe: verifies that the stream is a ES stream
109 *****************************************************************************/
110 static int ESProbe( probedata_t *p_data )
112 input_thread_t * p_input = (input_thread_t *)p_data;
114 char * psz_name = p_input->p_source;
118 if( TestMethod( INPUT_METHOD_VAR, "es" ) )
123 i_handle = open( psz_name, 0 );
133 /*****************************************************************************
134 * ESInit: initializes ES structures
135 *****************************************************************************/
136 static void ESInit( input_thread_t * p_input )
138 thread_es_data_t * p_method;
141 (thread_es_data_t *)malloc( sizeof(thread_es_data_t) )) == NULL )
143 intf_ErrMsg( "Out of memory" );
144 p_input->b_error = 1;
147 p_input->p_plugin_data = (void *)p_method;
152 /*****************************************************************************
153 * ESEnd: frees unused data
154 *****************************************************************************/
155 static void ESEnd( input_thread_t * p_input )
159 free( p_input->p_plugin_data );
162 /*****************************************************************************
163 * SafeRead: reads a chunk of stream and correctly detects errors
164 *****************************************************************************/
165 static __inline__ int SafeRead( input_thread_t * p_input, byte_t * p_buffer,
168 thread_es_data_t * p_method;
171 p_method = (thread_es_data_t *)p_input->p_plugin_data;
172 while( fread( p_buffer, i_len, 1, p_method->stream ) != 1 )
174 if( feof( p_method->stream ) )
179 if( (i_error = ferror( p_method->stream )) )
181 intf_ErrMsg( "Read failed (%s)", strerror(i_error) );
185 vlc_mutex_lock( &p_input->stream.stream_lock );
186 p_input->stream.p_selected_area->i_tell += i_len;
187 vlc_mutex_unlock( &p_input->stream.stream_lock );
192 /*****************************************************************************
193 * ESRead: reads data packets
194 *****************************************************************************
195 * Returns -1 in case of error, 0 if everything went well, and 1 in case of
197 *****************************************************************************/
198 static int ESRead( input_thread_t * p_input,
199 data_packet_t * pp_packets[INPUT_READ_ONCE] )
206 /*****************************************************************************
207 * ESSeek: changes the stream position indicator
208 *****************************************************************************/
209 static void ESSeek( input_thread_t * p_input, off_t i_position )
211 thread_es_data_t * p_method;
213 p_method = (thread_es_data_t *)p_input->p_plugin_data;
215 /* A little bourrin but should work for a while --Meuuh */
217 fseeko( p_method->stream, i_position, SEEK_SET );
219 fseek( p_method->stream, (long)i_position, SEEK_SET );
222 p_input->stream.p_selected_area->i_tell = i_position;
225 void ESDemux( input_thread_t * p_input, data_packet_t * p_data )
231 * Packet management utilities
234 /*****************************************************************************
235 * NewPacket: allocates a data packet
236 *****************************************************************************/
237 static struct data_packet_s * NewPacket( void * p_packet_cache,
246 /*****************************************************************************
247 * NewPES: allocates a pes packet
248 *****************************************************************************/
249 static pes_packet_t * NewPES( void * p_packet_cache )
256 /*****************************************************************************
257 * DeletePacket: deletes a data packet
258 *****************************************************************************/
259 static void DeletePacket( void * p_packet_cache,
260 data_packet_t * p_data )
265 /*****************************************************************************
266 * DeletePES: deletes a PES packet and associated data packets
267 *****************************************************************************/
268 static void DeletePES( void * p_packet_cache, pes_packet_t * p_pes )