]> git.sesse.net Git - vlc/blob - plugins/mpeg/input_ps.c
Second step in area management ; everything is in place now to have
[vlc] / plugins / mpeg / input_ps.c
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.9 2001/02/20 08:47:25 stef Exp $
6  *
7  * Authors: Christophe Massiot <massiot@via.ecp.fr>
8  *
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.
13  * 
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.
18  *
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  *****************************************************************************/
23
24 #define MODULE_NAME ps
25 #include "modules_inner.h"
26
27 /*****************************************************************************
28  * Preamble
29  *****************************************************************************/
30 #include "defs.h"
31
32 #include <stdlib.h>
33 #include <string.h>
34 #include <errno.h>
35
36 #include <sys/types.h>
37 #include <sys/stat.h>
38 #include <unistd.h>
39 #include <fcntl.h>
40
41 #include "config.h"
42 #include "common.h"
43 #include "threads.h"
44 #include "mtime.h"
45 #include "tests.h"
46
47 #include "intf_msg.h"
48
49 #include "main.h"
50
51 #include "stream_control.h"
52 #include "input_ext-intf.h"
53 #include "input_ext-dec.h"
54
55 #include "input.h"
56
57 #include "input_ps.h"
58 #include "mpeg_system.h"
59
60 #include "debug.h"
61
62 #include "modules.h"
63
64 /*****************************************************************************
65  * Local prototypes
66  *****************************************************************************/
67 static int  PSProbe     ( probedata_t * );
68 static int  PSRead      ( struct input_thread_s *,
69                           data_packet_t * p_packets[INPUT_READ_ONCE] );
70 static void PSInit      ( struct input_thread_s * );
71 static void PSEnd       ( struct input_thread_s * );
72 static void PSSeek      ( struct input_thread_s *, off_t );
73 static struct pes_packet_s *  NewPES    ( void * );
74 static struct data_packet_s * NewPacket ( void *, size_t );
75 static void DeletePacket( void *, struct data_packet_s * );
76 static void DeletePES   ( void *, struct pes_packet_s * );
77
78 /*****************************************************************************
79  * Functions exported as capabilities. They are declared as static so that
80  * we don't pollute the namespace too much.
81  *****************************************************************************/
82 void _M( input_getfunctions )( function_list_t * p_function_list )
83 {
84 #define input p_function_list->functions.input
85     p_function_list->pf_probe = PSProbe;
86     input.pf_init             = PSInit;
87     input.pf_open             = input_FileOpen;
88     input.pf_close            = input_FileClose;
89     input.pf_end              = PSEnd;
90     input.pf_set_area         = NULL;
91     input.pf_read             = PSRead;
92     input.pf_demux            = input_DemuxPS;
93     input.pf_new_packet       = NewPacket;
94     input.pf_new_pes          = NewPES;
95     input.pf_delete_packet    = DeletePacket;
96     input.pf_delete_pes       = DeletePES;
97     input.pf_rewind           = NULL;
98     input.pf_seek             = PSSeek;
99 #undef input
100 }
101
102 /*
103  * Data reading functions
104  */
105
106 /*****************************************************************************
107  * PSProbe: verifies that the stream is a PS stream
108  *****************************************************************************/
109 static int PSProbe( probedata_t *p_data )
110 {
111     input_thread_t * p_input = (input_thread_t *)p_data;
112
113     char * psz_name = p_input->p_source;
114     int i_handle;
115     int i_score = 10;
116
117     if( TestMethod( INPUT_METHOD_VAR, "ps" ) )
118     {
119         return( 999 );
120     }
121
122     if( ( strlen(psz_name) > 5 ) && !strncasecmp( psz_name, "file:", 5 ) )
123     {
124         /* If the user specified "file:" then it's probably a file */
125         i_score = 100;
126         psz_name += 5;
127     }
128
129     i_handle = open( psz_name, 0 );
130     if( i_handle == -1 )
131     {
132         return( 0 );
133     }
134     close( i_handle );
135
136     return( i_score );
137 }
138
139 /*****************************************************************************
140  * PSInit: initializes PS structures
141  *****************************************************************************/
142 static void PSInit( input_thread_t * p_input )
143 {
144     thread_ps_data_t *  p_method;
145
146     if( (p_method =
147          (thread_ps_data_t *)malloc( sizeof(thread_ps_data_t) )) == NULL )
148     {
149         intf_ErrMsg( "Out of memory" );
150         p_input->b_error = 1;
151         return;
152     }
153
154     p_input->p_plugin_data = (void *)p_method;
155     p_input->p_method_data = NULL;
156
157     /* Re-open the socket as a buffered FILE stream */
158     if( (p_method->stream = fdopen( p_input->i_handle, "r" )) == NULL )
159     {
160         intf_ErrMsg( "Cannot open file (%s)", strerror(errno) );
161         p_input->b_error = 1;
162         return;
163     }
164     rewind( p_method->stream );
165
166     /* FIXME : detect if InitStream failed */
167     input_InitStream( p_input, sizeof( stream_ps_data_t ) );
168     input_AddProgram( p_input, 0, sizeof( stream_ps_data_t ) );
169
170     if( p_input->stream.b_seekable )
171     {
172         stream_ps_data_t * p_demux_data =
173              (stream_ps_data_t *)p_input->stream.pp_programs[0]->p_demux_data;
174
175         /* Pre-parse the stream to gather stream_descriptor_t. */
176         p_input->stream.pp_programs[0]->b_is_ok = 0;
177         p_demux_data->i_PSM_version = EMPTY_PSM_VERSION;
178
179         while( !p_input->b_die && !p_input->b_error
180                 && !p_demux_data->b_has_PSM )
181         {
182             int                 i_result, i;
183             data_packet_t *     pp_packets[INPUT_READ_ONCE];
184
185             i_result = PSRead( p_input, pp_packets );
186             if( i_result == 1 )
187             {
188                 /* EOF */
189                 vlc_mutex_lock( &p_input->stream.stream_lock );
190                 p_input->stream.pp_programs[0]->b_is_ok = 1;
191                 vlc_mutex_unlock( &p_input->stream.stream_lock );
192                 break;
193             }
194             if( i_result == -1 )
195             {
196                 p_input->b_error = 1;
197                 break;
198             }
199
200             for( i = 0; i < INPUT_READ_ONCE && pp_packets[i] != NULL; i++ )
201             {
202                 /* FIXME: use i_p_config_t */
203                 input_ParsePS( p_input, pp_packets[i] );
204                 DeletePacket( p_input->p_method_data, pp_packets[i] );
205             }
206
207             /* File too big. */
208             if( p_input->stream.p_selected_area->i_tell >
209                                                     INPUT_PREPARSE_LENGTH )
210             {
211                 break;
212             }
213         }
214         rewind( p_method->stream );
215         vlc_mutex_lock( &p_input->stream.stream_lock );
216         p_input->stream.p_selected_area->i_tell = 0;
217         if( p_demux_data->b_has_PSM )
218         {
219             /* (The PSM decoder will care about spawning the decoders) */
220             p_input->stream.pp_programs[0]->b_is_ok = 1;
221         }
222 #ifdef AUTO_SPAWN
223         else
224         {
225             /* (We have to do it ourselves) */
226             int                 i_es;
227
228             /* FIXME: we should do multiple passes in case an audio type
229              * is not present */
230             for( i_es = 0;
231                  i_es < p_input->stream.pp_programs[0]->i_es_number;
232                  i_es++ )
233             {
234 #define p_es p_input->stream.pp_programs[0]->pp_es[i_es]
235                 switch( p_es->i_type )
236                 {
237                     case MPEG1_VIDEO_ES:
238                     case MPEG2_VIDEO_ES:
239                         input_SelectES( p_input, p_es );
240                         break;
241
242                     case MPEG1_AUDIO_ES:
243                     case MPEG2_AUDIO_ES:
244                         if( main_GetIntVariable( INPUT_CHANNEL_VAR, 0 )
245                                 == (p_es->i_id & 0x1F) )
246                         switch( main_GetIntVariable( INPUT_AUDIO_VAR, 0 ) )
247                         {
248                         case 0:
249                             main_PutIntVariable( INPUT_AUDIO_VAR,
250                                                  REQUESTED_MPEG );
251                         case REQUESTED_MPEG:
252                             input_SelectES( p_input, p_es );
253                         }
254                         break;
255
256                     case AC3_AUDIO_ES:
257                         if( main_GetIntVariable( INPUT_CHANNEL_VAR, 0 )
258                                 == ((p_es->i_id & 0xF00) >> 8) )
259                         switch( main_GetIntVariable( INPUT_AUDIO_VAR, 0 ) )
260                         {
261                         case 0:
262                             main_PutIntVariable( INPUT_AUDIO_VAR,
263                                                  REQUESTED_AC3 );
264                         case REQUESTED_AC3:
265                             input_SelectES( p_input, p_es );
266                         }
267                         break;
268
269                     case DVD_SPU_ES:
270                         if( main_GetIntVariable( INPUT_SUBTITLE_VAR, -1 )
271                                 == ((p_es->i_id & 0x1F00) >> 8) )
272                         {
273                             input_SelectES( p_input, p_es );
274                         }
275                         break;
276
277                     case LPCM_AUDIO_ES:
278                         /* FIXME ! */
279                         break;
280                 }
281             }
282                     
283         }
284 #endif
285 #ifdef STATS
286         input_DumpStream( p_input );
287 #endif
288         vlc_mutex_unlock( &p_input->stream.stream_lock );
289     }
290     else
291     {
292         /* The programs will be added when we read them. */
293         vlc_mutex_lock( &p_input->stream.stream_lock );
294         p_input->stream.pp_programs[0]->b_is_ok = 0;
295         vlc_mutex_unlock( &p_input->stream.stream_lock );
296     }
297 }
298
299 /*****************************************************************************
300  * PSEnd: frees unused data
301  *****************************************************************************/
302 static void PSEnd( input_thread_t * p_input )
303 {
304     free( p_input->stream.p_demux_data );
305     free( p_input->p_plugin_data );
306 }
307
308 /*****************************************************************************
309  * SafeRead: reads a chunk of stream and correctly detects errors
310  *****************************************************************************/
311 static __inline__ int SafeRead( input_thread_t * p_input, byte_t * p_buffer,
312                                 size_t i_len )
313 {
314     thread_ps_data_t *  p_method;
315     int                 i_error;
316
317     p_method = (thread_ps_data_t *)p_input->p_plugin_data;
318     while( fread( p_buffer, i_len, 1, p_method->stream ) != 1 )
319     {
320         if( feof( p_method->stream ) )
321         {
322             return( 1 );
323         }
324
325         if( (i_error = ferror( p_method->stream )) )
326         {
327             intf_ErrMsg( "Read failed (%s)", strerror(i_error) );
328             return( -1 );
329         }
330     }
331     vlc_mutex_lock( &p_input->stream.stream_lock );
332     p_input->stream.p_selected_area->i_tell += i_len;
333     vlc_mutex_unlock( &p_input->stream.stream_lock );
334     return( 0 );
335 }
336
337 /*****************************************************************************
338  * PSRead: reads data packets
339  *****************************************************************************
340  * Returns -1 in case of error, 0 if everything went well, and 1 in case of
341  * EOF.
342  *****************************************************************************/
343 static int PSRead( input_thread_t * p_input,
344                    data_packet_t * pp_packets[INPUT_READ_ONCE] )
345 {
346     byte_t              p_header[6];
347     data_packet_t *     p_data;
348     size_t              i_packet_size;
349     int                 i_packet, i_error;
350     thread_ps_data_t *  p_method;
351
352     p_method = (thread_ps_data_t *)p_input->p_plugin_data;
353
354     memset( pp_packets, 0, INPUT_READ_ONCE * sizeof(data_packet_t *) );
355     for( i_packet = 0; i_packet < INPUT_READ_ONCE; i_packet++ )
356     {
357         /* Read what we believe to be a packet header. */
358         if( (i_error = SafeRead( p_input, p_header, 6 )) )
359         {
360             return( i_error );
361         }
362
363         if( (U32_AT(p_header) & 0xFFFFFF00) != 0x100L )
364         {
365             /* This is not the startcode of a packet. Read the stream
366              * until we find one. */
367             u32         i_startcode = U32_AT(p_header);
368             int         i_dummy;
369
370             if( i_startcode )
371             {
372                 /* It is common for MPEG-1 streams to pad with zeros
373                  * (although it is forbidden by the recommendation), so
374                  * don't bother everybody in this case. */
375                 intf_WarnMsg( 1, "Garbage at input (%x)", i_startcode );
376             }
377
378             while( (i_startcode & 0xFFFFFF00) != 0x100L )
379             {
380                 i_startcode <<= 8;
381                 if( (i_dummy = getc( p_method->stream )) != EOF )
382                 {
383                     i_startcode |= i_dummy;
384                 }
385                 else
386                 {
387                     return( 1 );
388                 }
389             }
390             /* Packet found. */
391             *(u32 *)p_header = U32_AT(&i_startcode);
392             if( (i_error = SafeRead( p_input, p_header + 4, 2 )) )
393             {
394                 return( i_error );
395             }
396         }
397
398         if( U32_AT(p_header) != 0x1BA )
399         {
400             /* That's the case for all packets, except pack header. */
401             i_packet_size = U16_AT(&p_header[4]);
402         }
403         else
404         {
405             /* Pack header. */
406             if( (p_header[4] & 0xC0) == 0x40 )
407             {
408                 /* MPEG-2 */
409                 i_packet_size = 8;
410             }
411             else if( (p_header[4] & 0xF0) == 0x20 )
412             {
413                 /* MPEG-1 */
414                 i_packet_size = 6;
415             }
416             else
417             {
418                 intf_ErrMsg( "Unable to determine stream type" );
419                 return( -1 );
420             }
421         }
422
423         /* Fetch a packet of the appropriate size. */
424         if( (p_data = NewPacket( p_input, i_packet_size + 6 )) == NULL )
425         {
426             intf_ErrMsg( "Out of memory" );
427             return( -1 );
428         }
429
430         /* Copy the header we already read. */
431         memcpy( p_data->p_buffer, p_header, 6 );
432
433         /* Read the remaining of the packet. */
434         if( i_packet_size && (i_error =
435                 SafeRead( p_input, p_data->p_buffer + 6, i_packet_size )) )
436         {
437             return( i_error );
438         }
439
440         /* In MPEG-2 pack headers we still have to read stuffing bytes. */
441         if( U32_AT(p_header) == 0x1BA )
442         {
443             if( i_packet_size == 8 && (p_data->p_buffer[13] & 0x7) != 0 )
444             {
445                 /* MPEG-2 stuffing bytes */
446                 byte_t      p_garbage[8];
447                 if( (i_error = SafeRead( p_input, p_garbage,
448                                          p_data->p_buffer[13] & 0x7)) )
449                 {
450                     return( i_error );
451                 }
452             }
453         }
454
455         /* Give the packet to the other input stages. */
456         pp_packets[i_packet] = p_data;
457     }
458
459     return( 0 );
460 }
461
462 /*****************************************************************************
463  * PSSeek: changes the stream position indicator
464  *****************************************************************************/
465 static void PSSeek( input_thread_t * p_input, off_t i_position )
466 {
467     thread_ps_data_t *  p_method;
468
469     p_method = (thread_ps_data_t *)p_input->p_plugin_data;
470
471     /* A little bourrin but should work for a while --Meuuh */
472     fseeko( p_method->stream, i_position, SEEK_SET );
473
474     p_input->stream.p_selected_area->i_tell = i_position;
475 }
476
477 /*
478  * Packet management utilities
479  */
480
481 /*****************************************************************************
482  * NewPacket: allocates a data packet
483  *****************************************************************************/
484 static struct data_packet_s * NewPacket( void * p_garbage,
485                                          size_t i_size )
486 {
487     data_packet_t * p_data;
488
489     /* Safety check */
490     if( i_size > INPUT_MAX_PACKET_SIZE )
491     {
492         intf_ErrMsg( "Packet too big (%d)", i_size );
493         return NULL;
494     }
495
496     if( (p_data = (data_packet_t *)malloc( sizeof(data_packet_t) )) == NULL )
497     {
498         intf_DbgMsg( "Out of memory" );
499         return NULL;
500     }
501
502     if( (p_data->p_buffer = (byte_t *)malloc( i_size )) == NULL )
503     {
504         intf_DbgMsg( "Out of memory" );
505         free( p_data );
506         return NULL;
507     }
508
509     /* Initialize data */
510     p_data->p_next = NULL;
511     p_data->b_discard_payload = 0;
512
513     p_data->p_payload_start = p_data->p_buffer;
514     p_data->p_payload_end = p_data->p_buffer + i_size;
515
516     return( p_data );
517 }
518
519 /*****************************************************************************
520  * NewPES: allocates a pes packet
521  *****************************************************************************/
522 static pes_packet_t * NewPES( void * p_garbage )
523 {
524     pes_packet_t * p_pes;
525
526     if( (p_pes = (pes_packet_t *)malloc( sizeof(pes_packet_t) )) == NULL )
527     {
528         intf_DbgMsg( "Out of memory" );
529         return NULL;
530     }
531
532     p_pes->b_data_alignment = p_pes->b_discontinuity =
533         p_pes->i_pts = p_pes->i_dts = 0;
534     p_pes->i_pes_size = 0;
535     p_pes->p_first = NULL;
536
537     return( p_pes );
538 }
539
540 /*****************************************************************************
541  * DeletePacket: deletes a data packet
542  *****************************************************************************/
543 static void DeletePacket( void * p_garbage,
544                           data_packet_t * p_data )
545 {
546     ASSERT(p_data);
547     ASSERT(p_data->p_buffer);
548     free( p_data->p_buffer );
549     free( p_data );
550 }
551
552 /*****************************************************************************
553  * DeletePES: deletes a PES packet and associated data packets
554  *****************************************************************************/
555 static void DeletePES( void * p_garbage, pes_packet_t * p_pes )
556 {
557     data_packet_t *     p_data;
558     data_packet_t *     p_next;
559
560     p_data = p_pes->p_first;
561
562     while( p_data != NULL )
563     {
564         p_next = p_data->p_next;
565         free( p_data->p_buffer );
566         free( p_data );
567         p_data = p_next;
568     }
569
570     free( p_pes );
571 }
572
573