]> git.sesse.net Git - vlc/blob - src/input/mpeg_system.c
04d4951739aa319643081956808a8db0fb672a36
[vlc] / src / input / mpeg_system.c
1 /*****************************************************************************
2  * mpeg_system.c: TS, PS and PES management
3  *****************************************************************************
4  * Copyright (C) 1998-2001 VideoLAN
5  * $Id: mpeg_system.c,v 1.97.2.3 2002/09/26 22:43:45 massiot Exp $
6  *
7  * Authors: Christophe Massiot <massiot@via.ecp.fr>
8  *          Michel Lespinasse <walken@via.ecp.fr>
9  *          BenoĆ®t Steiner <benny@via.ecp.fr>
10  *          Samuel Hocevar <sam@via.ecp.fr>
11  *          Henri Fallon <henri@via.ecp.fr>
12  *
13  * This program is free software; you can redistribute it and/or modify
14  * it under the terms of the GNU General Public License as published by
15  * the Free Software Foundation; either version 2 of the License, or
16  * (at your option) any later version.
17  * 
18  * This program is distributed in the hope that it will be useful,
19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
21  * GNU General Public License for more details.
22  *
23  * You should have received a copy of the GNU General Public License
24  * along with this program; if not, write to the Free Software
25  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
26  *****************************************************************************/
27
28 /*****************************************************************************
29  * Preamble
30  *****************************************************************************/
31 #include <stdlib.h>
32 #include <string.h>                                    /* memcpy(), memset() */
33 #include <sys/types.h>                                              /* off_t */
34
35 #include <videolan/vlc.h>
36
37 #include "stream_control.h"
38 #include "input_ext-intf.h"
39 #include "input_ext-dec.h"
40 #include "input_ext-plugins.h"
41
42 /*
43  * PES Packet management
44  */
45
46 /*****************************************************************************
47  * MoveChunk
48  *****************************************************************************
49  * Small utility function used to parse discontinuous headers safely. Copies
50  * i_buf_len bytes of data to a buffer and returns the size copied.
51  * It also solves some alignment problems on non-IA-32, non-PPC processors.
52  * This is a variation on the theme of input_ext-dec.h:GetChunk().
53  *****************************************************************************/
54 static inline size_t MoveChunk( byte_t * p_dest, data_packet_t ** pp_data_src,
55                                 byte_t ** pp_src, size_t i_buf_len )
56 {
57     ptrdiff_t           i_available;
58
59     if( (i_available = (*pp_data_src)->p_payload_end - *pp_src)
60             >= i_buf_len )
61     {
62         if( p_dest != NULL )
63             memcpy( p_dest, *pp_src, i_buf_len );
64         *pp_src += i_buf_len;
65         return( i_buf_len );
66     }
67     else
68     {
69         size_t          i_init_len = i_buf_len;
70
71         do
72         {
73             if( p_dest != NULL )
74                 memcpy( p_dest, *pp_src, i_available );
75             *pp_data_src = (*pp_data_src)->p_next;
76             i_buf_len -= i_available;
77             p_dest += i_available;
78             if( *pp_data_src == NULL )
79             {
80                 *pp_src = NULL;
81                 return( i_init_len - i_buf_len );
82             }
83             *pp_src = (*pp_data_src)->p_payload_start;
84         }
85         while( (i_available = (*pp_data_src)->p_payload_end - *pp_src)
86                 <= i_buf_len );
87
88         if( i_buf_len )
89         {
90             if( p_dest != NULL )
91                 memcpy( p_dest, *pp_src, i_buf_len );
92             *pp_src += i_buf_len;
93         }
94         return( i_init_len );
95     }
96 }
97
98 /*****************************************************************************
99  * input_ParsePES
100  *****************************************************************************
101  * Parse a finished PES packet and analyze its header.
102  *****************************************************************************/
103 #define PES_HEADER_SIZE     7
104 void input_ParsePES( input_thread_t * p_input, es_descriptor_t * p_es )
105 {
106     data_packet_t * p_data;
107     byte_t *        p_byte;
108     byte_t          p_header[PES_HEADER_SIZE];
109     int             i_done;
110
111 #define p_pes (p_es->p_pes)
112
113     /* Parse the header. The header has a variable length, but in order
114      * to improve the algorithm, we will read the 14 bytes we may be
115      * interested in */
116     p_data = p_pes->p_first;
117     p_byte = p_data->p_payload_start;
118     i_done = 0;
119
120     if( MoveChunk( p_header, &p_data, &p_byte, PES_HEADER_SIZE )
121             != PES_HEADER_SIZE )
122     {
123         intf_WarnMsg( 1, "input: PES packet too short to have a header" );
124         input_DeletePES( p_input->p_method_data, p_pes );
125         p_pes = NULL;
126         return;
127     }
128
129     /* Get the PES size if defined */
130     p_es->i_pes_real_size = U16_AT(p_header + 4);
131     if( p_es->i_pes_real_size )
132     {
133         p_es->i_pes_real_size += 6;
134     }
135
136     /* First read the 6 header bytes common to all PES packets:
137      * use them to test the PES validity */
138     if( (p_header[0] || p_header[1] || (p_header[2] != 1)) )
139     {
140         /* packet_start_code_prefix != 0x000001 */
141         intf_ErrMsg( "input error: data loss, "
142                      "PES packet doesn't start with 0x000001" );
143         input_DeletePES( p_input->p_method_data, p_pes );
144         p_pes = NULL;
145     }
146     else
147     {
148         int i_pes_header_size, i_payload_size;
149
150         if ( p_es->i_pes_real_size &&
151              (p_es->i_pes_real_size != p_pes->i_pes_size) )
152         {
153             /* PES_packet_length is set and != total received payload */
154             /* Warn the decoder that the data may be corrupt. */
155             intf_WarnMsg( 1, "input: packet corrupted, "
156                              "PES sizes do not match" );
157         }
158
159         switch( p_es->i_stream_id )
160         {
161         case 0xBC:  /* Program stream map */
162         case 0xBE:  /* Padding */
163         case 0xBF:  /* Private stream 2 */
164         case 0xB0:  /* ECM */
165         case 0xB1:  /* EMM */
166         case 0xFF:  /* Program stream directory */
167         case 0xF2:  /* DSMCC stream */
168         case 0xF8:  /* ITU-T H.222.1 type E stream */
169             /* The payload begins immediately after the 6 bytes header, so
170              * we have finished with the parsing */
171             i_pes_header_size = 6;
172             break;
173
174         default:
175             if( (p_header[6] & 0xC0) == 0x80 )
176             {
177                 /* MPEG-2 : the PES header contains at least 3 more bytes. */
178                 size_t      i_max_len;
179                 boolean_t   b_has_pts, b_has_dts;
180                 byte_t      p_full_header[12];
181
182                 p_pes->b_data_alignment = p_header[6] & 0x04;
183
184                 i_max_len = MoveChunk( p_full_header, &p_data, &p_byte, 12 );
185                 if( i_max_len < 2 )
186                 {
187                     intf_WarnMsg( 1,
188                             "PES packet too short to have a MPEG-2 header" );
189                     input_DeletePES( p_input->p_method_data,
190                                             p_pes );
191                     p_pes = NULL;
192                     return;
193                 }
194
195                 b_has_pts = p_full_header[0] & 0x80;
196                 b_has_dts = p_full_header[0] & 0x40;
197                 i_pes_header_size = p_full_header[1] + 9;
198
199                 /* Now parse the optional header extensions */
200                 if( b_has_pts )
201                 {
202                     if( i_max_len < 7 )
203                     {
204                         intf_WarnMsg( 1,
205                             "PES packet too short to have a MPEG-2 header" );
206                         input_DeletePES( p_input->p_method_data,
207                                                 p_pes );
208                         p_pes = NULL;
209                         return;
210                     }
211                     p_pes->i_pts = input_ClockGetTS( p_input, p_es->p_pgrm,
212                     ( ((mtime_t)(p_full_header[2] & 0x0E) << 29) |
213                       ((mtime_t)(p_full_header[3]) << 22) |
214                       ((mtime_t)(p_full_header[4] & 0xFE) << 14) |
215                       ((mtime_t)p_full_header[5] << 7) |
216                       ((mtime_t)p_full_header[6] >> 1) ) );
217
218                     if( b_has_dts )
219                     {
220                         if( i_max_len < 12 )
221                         {
222                             intf_WarnMsg( 1,
223                               "PES packet too short to have a MPEG-2 header" );
224                             input_DeletePES( p_input->p_method_data,
225                                                     p_pes );
226                             p_pes = NULL;
227                             return;
228                         }
229                         p_pes->i_dts = input_ClockGetTS( p_input, p_es->p_pgrm,
230                         ( ((mtime_t)(p_full_header[7] & 0x0E) << 29) |
231                           (((mtime_t)U16_AT(p_full_header + 8) << 14)
232                                 - (1 << 14)) |
233                           ((mtime_t)U16_AT(p_full_header + 10) >> 1) ) );
234                     }
235                 }
236             }
237             else
238             {
239                 /* Probably MPEG-1 */
240                 boolean_t       b_has_pts, b_has_dts;
241
242                 i_pes_header_size = 6;
243                 p_data = p_pes->p_first;
244                 p_byte = p_data->p_payload_start;
245                 /* Cannot fail because the previous one succeeded. */
246                 MoveChunk( NULL, &p_data, &p_byte, 6 );
247
248                 while( *p_byte == 0xFF && i_pes_header_size < 23 )
249                 {
250                     i_pes_header_size++;
251                     if( MoveChunk( NULL, &p_data, &p_byte, 1 ) != 1 )
252                     {
253                         intf_WarnMsg( 1,
254                             "PES packet too short to have a MPEG-1 header" );
255                         input_DeletePES( p_input->p_method_data, p_pes );
256                         p_pes = NULL;
257                         return;
258                     }
259                 }
260                 if( i_pes_header_size == 23 )
261                 {
262                     intf_ErrMsg( "input error: too much MPEG-1 stuffing" );
263                     input_DeletePES( p_input->p_method_data, p_pes );
264                     p_pes = NULL;
265                     return;
266                 }
267
268                 if( (*p_byte & 0xC0) == 0x40 )
269                 {
270                     /* Don't ask why... --Meuuh */
271                     /* Erm... why ? --Sam */
272                     /* Well... According to the recommendation, it is for
273                      * STD_buffer_scale and STD_buffer_size. --Meuuh */
274                     i_pes_header_size += 2;
275                     if( MoveChunk( NULL, &p_data, &p_byte, 2 ) != 2 )
276                     {
277                         intf_WarnMsg( 1, "input: PES packet too short "
278                                          "to have a MPEG-1 header" );
279                         input_DeletePES( p_input->p_method_data, p_pes );
280                         p_pes = NULL;
281                         return;
282                     }
283                 }
284
285                 i_pes_header_size++;
286
287                 b_has_pts = *p_byte & 0x20;
288                 b_has_dts = *p_byte & 0x10;
289
290                 if( b_has_pts )
291                 {
292                     byte_t      p_ts[5];
293
294                     i_pes_header_size += 4;
295                     if( MoveChunk( p_ts, &p_data, &p_byte, 5 ) != 5 )
296                     {
297                         intf_WarnMsg( 1, "input: PES packet too short "
298                                          "to have a MPEG-1 header" );
299                         input_DeletePES( p_input->p_method_data, p_pes );
300                         p_pes = NULL;
301                         return;
302                     }
303
304                     p_pes->i_pts = input_ClockGetTS( p_input, p_es->p_pgrm,
305                        ( ((mtime_t)(p_ts[0] & 0x0E) << 29) |
306                          (((mtime_t)U32_AT(p_ts) & 0xFFFE00) << 6) |
307                          ((mtime_t)p_ts[3] << 7) |
308                          ((mtime_t)p_ts[4] >> 1) ) );
309
310                     if( b_has_dts )
311                     {
312                         i_pes_header_size += 5;
313                         if( MoveChunk( p_ts, &p_data, &p_byte, 5 ) != 5 )
314                         {
315                             intf_WarnMsg( 1, "input: PES packet too short "
316                                              "to have a MPEG-1 header" );
317                             input_DeletePES( p_input->p_method_data,
318                                                     p_pes );
319                             p_pes = NULL;
320                             return;
321                         }
322
323                         p_pes->i_dts = input_ClockGetTS( p_input,
324                                                          p_es->p_pgrm,
325                             ( ((mtime_t)(p_ts[0] & 0x0E) << 29) |
326                               (((mtime_t)U32_AT(p_ts) & 0xFFFE00) << 6) |
327                               ((mtime_t)p_ts[3] << 7) |
328                               ((mtime_t)p_ts[4] >> 1) ) );
329                     }
330                 }
331             }
332
333             break;
334         }
335
336         if ( p_es->i_type == A52B_AUDIO_ES
337               || p_es->i_type == AC3_AUDIO_ES )
338         {
339             /* With A/52 audio, we need to skip the first 3 bytes */
340             i_pes_header_size += 3;
341         }
342
343         if( p_es->i_type == A52B_AUDIO_ES
344              || p_es->i_type == LPCMB_AUDIO_ES
345              || p_es->i_type == DVDB_SPU_ES )
346         {
347             /* stream_private_id */
348             i_pes_header_size += 1;
349         }
350
351         /* Now we've parsed the header, we just have to indicate in some
352          * specific data packets where the PES payload begins (renumber
353          * p_payload_start), so that the decoders can find the beginning
354          * of their data right out of the box. */
355         p_data = p_pes->p_first;
356         i_payload_size = p_data->p_payload_end
357                                  - p_data->p_payload_start;
358         while( i_pes_header_size > i_payload_size )
359         {
360             /* These packets are entirely filled by the PES header. */
361             i_pes_header_size -= i_payload_size;
362             p_data->p_payload_start = p_data->p_payload_end;
363             /* Go to the next data packet. */
364             if( (p_data = p_data->p_next) == NULL )
365             {
366                 intf_ErrMsg( "input error: PES header bigger than payload" );
367                 input_DeletePES( p_input->p_method_data, p_pes );
368                 p_pes = NULL;
369                 return;
370             }
371             i_payload_size = p_data->p_payload_end
372                                  - p_data->p_payload_start;
373         }
374         /* This last packet is partly header, partly payload. */
375         if( i_payload_size < i_pes_header_size )
376         {
377             intf_ErrMsg( "input error: PES header bigger than payload" );
378             input_DeletePES( p_input->p_method_data, p_pes );
379             p_pes = NULL;
380             return;
381         }
382         p_data->p_payload_start += i_pes_header_size;
383
384
385         /* Now we can eventually put the PES packet in the decoder's
386          * PES fifo */
387         if( p_es->p_decoder_fifo != NULL )
388         {
389             input_DecodePES( p_es->p_decoder_fifo, p_pes );
390         }
391         else
392         {
393             intf_ErrMsg( "input error: no fifo to receive PES %p "
394                          "(who wrote this damn code ?)", p_pes );
395             input_DeletePES( p_input->p_method_data, p_pes );
396         }
397         p_pes = NULL;
398     }
399 #undef p_pes
400
401 }
402
403 /*****************************************************************************
404  * input_GatherPES:
405  *****************************************************************************
406  * Gather a PES packet.
407  *****************************************************************************/
408 void input_GatherPES( input_thread_t * p_input, data_packet_t * p_data,
409                       es_descriptor_t * p_es,
410                       boolean_t b_unit_start, boolean_t b_packet_lost )
411 {
412 #define p_pes (p_es->p_pes)
413
414     /* If we lost data, insert a NULL data packet (philosophy : 0 is quite
415      * often an escape sequence in decoders, so that should make them wait
416      * for the next start code). */
417     if( b_packet_lost )
418     {
419         input_NullPacket( p_input, p_es );
420     }
421
422     if( b_unit_start && p_pes != NULL )
423     {
424         /* If the data packet contains the begining of a new PES packet, and
425          * if we were reassembling a PES packet, then the PES should be
426          * complete now, so parse its header and give it to the decoders. */
427         input_ParsePES( p_input, p_es );
428     }
429
430     if( !b_unit_start && p_pes == NULL )
431     {
432         /* Random access... */
433         input_DeletePacket( p_input->p_method_data, p_data );
434     }
435     else
436     {
437         if( b_unit_start )
438         {
439             /* If we are at the beginning of a new PES packet, we must fetch
440              * a new PES buffer to begin with the reassembly of this PES
441              * packet. This is also here that we can synchronize with the
442              * stream if we lost packets or if the decoder has just
443              * started. */
444             if( (p_pes = input_NewPES( p_input->p_method_data ) ) == NULL )
445             {
446                 intf_ErrMsg( "input error: out of memory" );
447                 p_input->b_error = 1;
448                 return;
449             }
450             p_pes->i_rate = p_input->stream.control.i_rate;
451             p_pes->p_first = p_data;
452             
453             /* If the PES header fits in the first data packet, we can
454              * already set p_gather->i_pes_real_size. */
455             if( p_data->p_payload_end - p_data->p_payload_start
456                     >= PES_HEADER_SIZE )
457             {
458                 p_es->i_pes_real_size = ((u16)p_data->p_payload_start[4] << 8)
459                                          + p_data->p_payload_start[5] + 6;
460             }
461             else
462             { 
463                 p_es->i_pes_real_size = 0;
464             } 
465         }
466         else
467         {
468             /* Update the relations between the data packets */
469             p_pes->p_last->p_next = p_data;
470         }
471
472         p_pes->p_last = p_data;
473         p_pes->i_nb_data++;
474
475         /* Size of the payload carried in the data packet */
476         p_pes->i_pes_size += (p_data->p_payload_end
477                                  - p_data->p_payload_start);
478     
479         /* We can check if the packet is finished */
480         if( p_pes->i_pes_size == p_es->i_pes_real_size )
481         {
482             /* The packet is finished, parse it */
483             input_ParsePES( p_input, p_es );
484         }
485     }
486 #undef p_pes
487 }
488
489
490 /*
491  * PS Demultiplexing
492  */
493
494 /*****************************************************************************
495  * GetID: Get the ID of a stream
496  *****************************************************************************/
497 static u16 GetID( data_packet_t * p_data )
498 {
499     u16         i_id;
500
501     i_id = p_data->p_demux_start[3];                            /* stream_id */
502     if( i_id == 0xBD )
503     {
504         /* FIXME : this is not valid if the header is split in multiple
505          * packets */
506         /* stream_private_id */
507         i_id |= p_data->p_demux_start[ 9 + p_data->p_demux_start[8] ] << 8;
508     }
509     return( i_id );
510 }
511
512 /*****************************************************************************
513  * DecodePSM: Decode the Program Stream Map information
514  *****************************************************************************
515  * FIXME : loads are not aligned in this function
516  *****************************************************************************/
517 static void DecodePSM( input_thread_t * p_input, data_packet_t * p_data )
518 {
519     stream_ps_data_t *  p_demux =
520                  (stream_ps_data_t *)p_input->stream.p_demux_data;
521     byte_t *            p_byte;
522     byte_t *            p_end;
523     int                 i;
524     int                 i_new_es_number = 0;
525
526     if( p_data->p_demux_start + 10 > p_data->p_payload_end )
527     {
528         intf_ErrMsg( "input error: PSM too short : packet corrupt" );
529         return;
530     }
531
532     if( p_demux->b_has_PSM
533         && p_demux->i_PSM_version == (p_data->p_demux_start[6] & 0x1F) )
534     {
535         /* Already got that one. */
536         return;
537     }
538
539     p_demux->b_has_PSM = 1;
540     p_demux->i_PSM_version = p_data->p_demux_start[6] & 0x1F;
541
542     /* Go to elementary_stream_map_length, jumping over
543      * program_stream_info. */
544     p_byte = p_data->p_demux_start + 10
545               + U16_AT(&p_data->p_demux_start[8]);
546     if( p_byte > p_data->p_payload_end )
547     {
548         intf_ErrMsg( "input error: PSM too short, packet corrupt" );
549         return;
550     }
551     /* This is the full size of the elementary_stream_map.
552      * 2 == elementary_stream_map_length
553      * Please note that CRC_32 is not included in the length. */
554     p_end = p_byte + 2 + U16_AT(p_byte);
555     p_byte += 2;
556     if( p_end > p_data->p_payload_end )
557     {
558         intf_ErrMsg( "input error: PSM too short, packet corrupt" );
559         return;
560     }
561
562     vlc_mutex_lock( &p_input->stream.stream_lock );
563
564     /* 4 == minimum useful size of a section */
565     while( p_byte + 4 <= p_end )
566     {
567         es_descriptor_t *   p_es = NULL;
568         u8                  i_stream_id = p_byte[1];
569         /* FIXME: there will be a problem with private streams... (same
570          * stream_id) */
571
572         /* Look for the ES in the ES table */
573         for( i = i_new_es_number;
574              i < p_input->stream.pp_programs[0]->i_es_number;
575              i++ )
576         {
577             if( p_input->stream.pp_programs[0]->pp_es[i]->i_stream_id
578                     == i_stream_id )
579             {
580                 p_es = p_input->stream.pp_programs[0]->pp_es[i];
581                 if( p_es->i_type != p_byte[0] )
582                 {
583                     input_DelES( p_input, p_es );
584                     p_es = NULL;
585                 }
586                 else
587                 {
588                     /* Move the ES to the beginning. */
589                     p_input->stream.pp_programs[0]->pp_es[i]
590                         = p_input->stream.pp_programs[0]->pp_es[ i_new_es_number ];
591                     p_input->stream.pp_programs[0]->pp_es[ i_new_es_number ]
592                         = p_es;
593                     i_new_es_number++;
594                 }
595                 break;
596             }
597         }
598
599         /* The goal is to have all the ES we have just read in the
600          * beginning of the pp_es table, and all the others at the end,
601          * so that we can close them more easily at the end. */
602         if( p_es == NULL )
603         {
604             p_es = input_AddES( p_input, p_input->stream.pp_programs[0],
605                                 i_stream_id, 0 );
606             p_es->i_type = p_byte[0];
607             p_es->b_audio = ( p_es->i_type == MPEG1_AUDIO_ES
608                               || p_es->i_type == MPEG2_AUDIO_ES
609                               || p_es->i_type == AC3_AUDIO_ES
610                               || p_es->i_type == A52B_AUDIO_ES
611                               || p_es->i_type == LPCM_AUDIO_ES
612                               || p_es->i_type == LPCMB_AUDIO_ES
613                             );
614
615             /* input_AddES has inserted the new element at the end. */
616             p_input->stream.pp_programs[0]->pp_es[
617                 p_input->stream.pp_programs[0]->i_es_number ]
618                 = p_input->stream.pp_programs[0]->pp_es[ i_new_es_number ];
619             p_input->stream.pp_programs[0]->pp_es[ i_new_es_number ] = p_es;
620             i_new_es_number++;
621         }
622         p_byte += 4 + U16_AT(&p_byte[2]);
623     }
624
625     /* Un-select the streams that are no longer parts of the program. */
626     while( i_new_es_number < p_input->stream.pp_programs[0]->i_es_number )
627     {
628         /* We remove pp_es[i_new_es_member] and not pp_es[i] because the
629          * list will be emptied starting from the end */
630         input_DelES( p_input,
631                      p_input->stream.pp_programs[0]->pp_es[i_new_es_number] );
632     }
633
634     if( p_main->b_stats )
635     {
636         intf_StatMsg( "input info: The stream map after the PSM is now :" );
637         input_DumpStream( p_input );
638     }
639
640     vlc_mutex_unlock( &p_input->stream.stream_lock );
641 }
642
643 /*****************************************************************************
644  * input_ReadPS: store a PS packet into a data_buffer_t
645  *****************************************************************************/
646 #define PEEK( SIZE )                                                        \
647     i_error = input_Peek( p_input, &p_peek, SIZE );                         \
648     if( i_error == -1 )                                                     \
649     {                                                                       \
650         return( -1 );                                                       \
651     }                                                                       \
652     else if( i_error < SIZE )                                               \
653     {                                                                       \
654         /* EOF */                                                           \
655         return( 0 );                                                        \
656     }
657
658 ssize_t input_ReadPS( input_thread_t * p_input, data_packet_t ** pp_data )
659 {
660     byte_t *            p_peek;
661     size_t              i_packet_size;
662     ssize_t             i_error, i_read;
663
664     /* Read what we believe to be a packet header. */
665     PEEK( 4 );
666
667     if( p_peek[0] || p_peek[1] || p_peek[2] != 1 || p_peek[3] < 0xB9 )
668     {
669         if( p_peek[0] || p_peek[1] || p_peek[2] )
670         {
671             /* It is common for MPEG-1 streams to pad with zeros
672              * (although it is forbidden by the recommendation), so
673              * don't bother everybody in this case. */
674             intf_WarnMsg( 3, "input warning: garbage (0x%.2x%.2x%.2x%.2x)",
675                  p_peek[0], p_peek[1], p_peek[2], p_peek[3] );
676         }
677
678         /* This is not the startcode of a packet. Read the stream
679          * until we find one. */
680         while( p_peek[0] || p_peek[1] || p_peek[2] != 1 || p_peek[3] < 0xB9 )
681         {
682             p_input->p_current_data++;
683             PEEK( 4 );
684             if( p_input->b_die ) return( -1 );
685         }
686         /* Packet found. */
687     }
688
689     /* 0x1B9 == SYSTEM_END_CODE, it is only 4 bytes long. */
690     if( p_peek[3] != 0xB9 )
691     {
692         /* The packet is at least 6 bytes long. */
693         PEEK( 6 );
694
695         if( p_peek[3] != 0xBA )
696         {
697             /* That's the case for all packets, except pack header. */
698             i_packet_size = (p_peek[4] << 8) | p_peek[5];
699         }
700         else
701         {
702             /* Pack header. */
703             if( (p_peek[4] & 0xC0) == 0x40 )
704             {
705                 /* MPEG-2 */
706                 i_packet_size = 8;
707             }
708             else if( (p_peek[4] & 0xF0) == 0x20 )
709             {
710                 /* MPEG-1 */
711                 i_packet_size = 6;
712             }
713             else
714             {
715                 intf_ErrMsg( "Unable to determine stream type" );
716                 return( -1 );
717             }
718         }
719     }
720     else
721     {
722         /* System End Code */
723         i_packet_size = -2;
724     }
725
726     /* Fetch a packet of the appropriate size. */
727     i_read = input_SplitBuffer( p_input, pp_data, i_packet_size + 6 );
728     if( i_read <= 0 )
729     {
730         return( i_read );
731     }
732
733     /* In MPEG-2 pack headers we still have to read stuffing bytes. */
734     if( ((*pp_data)->p_demux_start[3] == 0xBA) && (i_packet_size == 8) )
735     {
736         size_t i_stuffing = ((*pp_data)->p_demux_start[13] & 0x7);
737         /* Force refill of the input buffer - though we don't care
738          * about p_peek. Please note that this is unoptimized. */
739         PEEK( i_stuffing );
740         p_input->p_current_data += i_stuffing;
741     }
742
743     return( 1 );
744 }
745
746 #undef PEEK
747
748 /*****************************************************************************
749  * input_ParsePS: read the PS header
750  *****************************************************************************/
751 es_descriptor_t * input_ParsePS( input_thread_t * p_input,
752                                  data_packet_t * p_data )
753 {
754     u32                 i_code;
755     es_descriptor_t *   p_es = NULL;
756
757     i_code = p_data->p_demux_start[3];
758
759     if( i_code > 0xBC ) /* ES start code */
760     {
761         u16                 i_id;
762         int                 i_dummy;
763
764         /* This is a PES packet. Find out if we want it or not. */
765         i_id = GetID( p_data );
766
767         vlc_mutex_lock( &p_input->stream.stream_lock );
768         if( p_input->stream.pp_programs[0]->b_is_ok )
769         {
770             /* Look only at the selected ES. */
771             for( i_dummy = 0; i_dummy < p_input->stream.i_selected_es_number;
772                  i_dummy++ )
773             {
774                 if( p_input->stream.pp_selected_es[i_dummy] != NULL
775                     && p_input->stream.pp_selected_es[i_dummy]->i_id == i_id )
776                 {
777                     p_es = p_input->stream.pp_selected_es[i_dummy];
778                     break;
779                 }
780             }
781         }
782         else
783         {
784             stream_ps_data_t * p_demux =
785               (stream_ps_data_t *)p_input->stream.pp_programs[0]->p_demux_data;
786
787             /* Search all ES ; if not found -> AddES */
788             p_es = input_FindES( p_input, i_id );
789
790             if( p_es == NULL && !p_demux->b_has_PSM )
791             {
792                 p_es = input_AddES( p_input, p_input->stream.pp_programs[0],
793                                     i_id, 0 );
794                 if( p_es != NULL )
795                 {
796                     p_es->i_stream_id = p_data->p_demux_start[3];
797
798                     /* Set stream type and auto-spawn. */
799                     if( (i_id & 0xF0) == 0xE0 )
800                     {
801                         /* MPEG video */
802                         p_es->i_type = MPEG2_VIDEO_ES;
803                         p_es->i_cat = VIDEO_ES;
804 #ifdef AUTO_SPAWN
805                         if( !p_input->stream.b_seekable )
806                             input_SelectES( p_input, p_es );
807 #endif
808                     }
809                     else if( (i_id & 0xE0) == 0xC0 )
810                     {
811                         /* MPEG audio */
812                         p_es->i_type = MPEG2_AUDIO_ES;
813                         p_es->b_audio = 1;
814                         p_es->i_cat = AUDIO_ES;
815 #ifdef AUTO_SPAWN
816                         if( !p_input->stream.b_seekable )
817                         if( config_GetIntVariable( "audio-channel" )
818                                 == (p_es->i_id & 0x1F) ||
819                             ( config_GetIntVariable( "audio-channel" ) < 0
820                               && !(p_es->i_id & 0x1F) ) )
821                         switch( config_GetIntVariable( "audio-type" ) )
822                         {
823                         case -1:
824                         case REQUESTED_MPEG:
825                             input_SelectES( p_input, p_es );
826                         }
827 #endif
828                     }
829                     else if( (i_id & 0xF0FF) == 0x80BD )
830                     {
831                         /* AC3 audio (0x80->0x8F) */
832                         p_es->i_type = A52B_AUDIO_ES;
833                         p_es->b_audio = 1;
834                         p_es->i_cat = AUDIO_ES;
835 #ifdef AUTO_SPAWN
836                         if( !p_input->stream.b_seekable )
837                         if( config_GetIntVariable( "audio-channel" )
838                                 == ((p_es->i_id & 0xF00) >> 8) ||
839                             ( config_GetIntVariable( "audio-channel" ) < 0
840                               && !((p_es->i_id & 0xF00) >> 8)) )
841                         switch( config_GetIntVariable( "audio-type" ) )
842                         {
843                         case -1:
844                         case REQUESTED_AC3:
845                             input_SelectES( p_input, p_es );
846                         }
847 #endif
848                     }
849                     else if( (i_id & 0xE0FF) == 0x20BD )
850                     {
851                         /* Subtitles video (0x20->0x3F) */
852                         p_es->i_type = DVDB_SPU_ES;
853                         p_es->i_cat = SPU_ES;
854 #ifdef AUTO_SPAWN
855                         if( config_GetIntVariable( "spu-channel" )
856                                 == ((p_es->i_id & 0x1F00) >> 8) )
857                         {
858                             if( !p_input->stream.b_seekable )
859                                 input_SelectES( p_input, p_es );
860                         }
861 #endif
862                     }
863                     else if( (i_id & 0xF0FF) == 0xA0BD )
864                     {
865                         /* LPCM audio (0xA0->0xAF) */
866                         p_es->i_type = LPCMB_AUDIO_ES;
867                         p_es->b_audio = 1;
868                         p_es->i_cat = AUDIO_ES;
869                     }
870                     else
871                     {
872                         p_es->i_type = UNKNOWN_ES;
873                     }
874                 }
875
876                 /* Tell the interface the stream has changed */
877                 p_input->stream.b_changed = 1;
878             }
879         } /* stream.b_is_ok */
880         vlc_mutex_unlock( &p_input->stream.stream_lock );
881     } /* i_code > 0xBC */
882
883     return( p_es );
884 }
885
886 /*****************************************************************************
887  * input_DemuxPS: first step of demultiplexing: the PS header
888  *****************************************************************************/
889 void input_DemuxPS( input_thread_t * p_input, data_packet_t * p_data )
890 {
891     u32                 i_code;
892     boolean_t           b_trash = 0;
893     es_descriptor_t *   p_es = NULL;
894
895     i_code = ((u32)p_data->p_demux_start[0] << 24)
896                 | ((u32)p_data->p_demux_start[1] << 16)
897                 | ((u32)p_data->p_demux_start[2] << 8)
898                 | p_data->p_demux_start[3];
899     if( i_code <= 0x1BC )
900     {
901         switch( i_code )
902         {
903         case 0x1BA: /* PACK_START_CODE */
904             {
905                 /* Read the SCR. */
906                 mtime_t         scr_time;
907                 u32             i_mux_rate;
908
909                 if( (p_data->p_demux_start[4] & 0xC0) == 0x40 )
910                 {
911                     /* MPEG-2 */
912                     byte_t      p_header[14];
913                     byte_t *    p_byte;
914                     p_byte = p_data->p_demux_start;
915
916                     if( MoveChunk( p_header, &p_data, &p_byte, 14 ) != 14 )
917                     {
918                         intf_WarnMsg( 1, "input: packet too short "
919                                          "to have a header" );
920                         b_trash = 1;
921                         break;
922                     }
923                     scr_time =
924                          ((mtime_t)(p_header[4] & 0x38) << 27) |
925                          ((mtime_t)(U32_AT(p_header + 4) & 0x03FFF800)
926                                         << 4) |
927                          ((( ((mtime_t)U16_AT(p_header + 6) << 16)
928                             | (mtime_t)U16_AT(p_header + 8) ) & 0x03FFF800)
929                                         >> 11);
930
931                     /* mux_rate */
932                     i_mux_rate = ((u32)U16_AT(p_header + 10) << 6)
933                                    | (p_header[12] >> 2);
934                     /* FIXME FIXME FIXME FIXME FIXME FIXME FIXME FIXME
935                      * This is the biggest kludge ever !
936                      * I don't know what's wrong with mux_rate calculation
937                      * but this heuristic work well : */
938                     i_mux_rate <<= 1;
939                     i_mux_rate /= 3;
940                 }
941                 else
942                 {
943                     /* MPEG-1 SCR is like PTS. */
944                     byte_t      p_header[12];
945                     byte_t *    p_byte;
946                     p_byte = p_data->p_demux_start;
947
948                     if( MoveChunk( p_header, &p_data, &p_byte, 12 ) != 12 )
949                     {
950                         intf_WarnMsg( 1, "input: packet too short "
951                                          "to have a header" );
952                         b_trash = 1;
953                         break;
954                     }
955                     scr_time =
956                          ((mtime_t)(p_header[4] & 0x0E) << 29) |
957                          (((mtime_t)U32_AT(p_header + 4) & 0xFFFE00) << 6) |
958                          ((mtime_t)p_header[7] << 7) |
959                          ((mtime_t)p_header[8] >> 1);
960
961                     /* mux_rate */
962                     i_mux_rate = (U32_AT(p_header + 8) & 0x7FFFFE) >> 1;
963                 }
964                 /* Call the pace control. */
965                 input_ClockManageRef( p_input,
966                                       p_input->stream.p_selected_program,
967                                       scr_time );
968
969                 if( i_mux_rate != p_input->stream.i_mux_rate
970                      && p_input->stream.i_mux_rate )
971                 {
972                     intf_WarnMsg( 2, "input: mux_rate changed, "
973                                      "expect cosmetic errors" );
974                 }
975                 p_input->stream.i_mux_rate = i_mux_rate;
976
977                 b_trash = 1;
978             }
979             break;
980
981         case 0x1BB: /* SYSTEM_START_CODE */
982             b_trash = 1;                              /* Nothing interesting */
983             break;
984
985         case 0x1BC: /* PROGRAM_STREAM_MAP_CODE */
986             DecodePSM( p_input, p_data );
987             b_trash = 1;
988             break;
989     
990         case 0x1B9: /* PROGRAM_END_CODE */
991             b_trash = 1;
992             break;
993    
994         default:
995             /* This should not happen */
996             b_trash = 1;
997             intf_WarnMsg( 3, "input: unwanted packet received "
998                              "with start code 0x%.8x", i_code );
999         }
1000     }
1001     else
1002     {
1003         p_es = input_ParsePS( p_input, p_data );
1004
1005         vlc_mutex_lock( &p_input->stream.control.control_lock );
1006         if( p_es != NULL && p_es->p_decoder_fifo != NULL
1007              && (!p_es->b_audio || !p_input->stream.control.b_mute) )
1008         {
1009             vlc_mutex_unlock( &p_input->stream.control.control_lock );
1010             p_es->c_packets++;
1011             input_GatherPES( p_input, p_data, p_es, 1, 0 );
1012         }
1013         else
1014         {
1015             vlc_mutex_unlock( &p_input->stream.control.control_lock );
1016             b_trash = 1;
1017         }
1018     }
1019
1020     /* Trash the packet if it has no payload or if it isn't selected */
1021     if( b_trash )
1022     {
1023         input_DeletePacket( p_input->p_method_data, p_data );
1024         p_input->stream.c_packets_trashed++;
1025     }
1026 }
1027
1028  
1029 /*
1030  * TS Demultiplexing
1031  */
1032
1033 /*****************************************************************************
1034  * input_ReadTS: store a TS packet into a data_buffer_t
1035  *****************************************************************************/
1036 #define PEEK( SIZE )                                                        \
1037     i_error = input_Peek( p_input, &p_peek, SIZE );                         \
1038     if( i_error == -1 )                                                     \
1039     {                                                                       \
1040         return( -1 );                                                       \
1041     }                                                                       \
1042     else if( i_error < SIZE )                                               \
1043     {                                                                       \
1044         /* EOF */                                                           \
1045         return( 0 );                                                        \
1046     }
1047
1048 ssize_t input_ReadTS( input_thread_t * p_input, data_packet_t ** pp_data )
1049 {
1050     byte_t *            p_peek;
1051     ssize_t             i_error, i_read;
1052
1053     PEEK( 1 );
1054
1055     if( *p_peek != TS_SYNC_CODE )
1056     {
1057         intf_WarnMsg( 3, "input warning: garbage at input (%x)", *p_peek );
1058
1059         if( p_input->i_mtu )
1060         {
1061             while( *p_peek != TS_SYNC_CODE )
1062             {
1063                 /* Try to resync on next packet. */
1064                 PEEK( TS_PACKET_SIZE );
1065                 p_input->p_current_data += TS_PACKET_SIZE;
1066                 PEEK( 1 );
1067             }
1068         }
1069         else
1070         {
1071             /* Move forward until we find 0x47 (and hope it's the good
1072              * one... FIXME) */
1073             while( *p_peek != TS_SYNC_CODE )
1074             {
1075                 p_input->p_current_data++;
1076                 PEEK( 1 );
1077             }
1078         }
1079     }
1080
1081     i_read = input_SplitBuffer( p_input, pp_data, TS_PACKET_SIZE );
1082     if( i_read <= 0 )
1083     {
1084         return( i_read );
1085     }
1086
1087     return( 1 );
1088 }
1089
1090 /*****************************************************************************
1091  * input_DemuxTS: first step of demultiplexing: the TS header
1092  *****************************************************************************/
1093 void input_DemuxTS( input_thread_t * p_input, data_packet_t * p_data,
1094                         psi_callback_t pf_psi_callback )
1095 {
1096     u16                 i_pid;
1097     int                 i_dummy;
1098     boolean_t           b_adaptation;         /* Adaptation field is present */
1099     boolean_t           b_payload;                 /* Packet carries payload */
1100     boolean_t           b_unit_start;  /* A PSI or a PES start in the packet */
1101     boolean_t           b_trash = 0;             /* Is the packet unuseful ? */
1102     boolean_t           b_lost = 0;             /* Was there a packet loss ? */
1103     boolean_t           b_psi = 0;                        /* Is this a PSI ? */
1104     boolean_t           b_pcr = 0;                   /* Does it have a PCR ? */
1105     es_descriptor_t *   p_es = NULL;
1106     es_ts_data_t *      p_es_demux = NULL;
1107     pgrm_ts_data_t *    p_pgrm_demux = NULL;
1108
1109 #define p (p_data->p_demux_start)
1110     /* Extract flags values from TS common header. */
1111     i_pid = ((p[1] & 0x1F) << 8) | p[2];
1112     b_unit_start = (p[1] & 0x40);
1113     b_adaptation = (p[3] & 0x20);
1114     b_payload = (p[3] & 0x10);
1115
1116     /* Find out the elementary stream. */
1117     vlc_mutex_lock( &p_input->stream.stream_lock );
1118     
1119     for( i_dummy = 0; i_dummy < p_input->stream.i_pgrm_number; i_dummy ++ )
1120     {
1121         if( (( pgrm_ts_data_t * ) p_input->stream.pp_programs[i_dummy]->
1122                     p_demux_data)->i_pcr_pid == i_pid )
1123         {
1124             b_pcr = 1;
1125             break;
1126         }
1127     }
1128             
1129     p_es= input_FindES( p_input, i_pid );
1130     
1131     if( (p_es != NULL) && (p_es->p_demux_data != NULL) )
1132     {
1133         p_es_demux = (es_ts_data_t *)p_es->p_demux_data;
1134         
1135         if( p_es_demux->b_psi )
1136         {
1137             b_psi = 1;
1138         }
1139         else
1140         {
1141             p_pgrm_demux = (pgrm_ts_data_t *)p_es->p_pgrm->p_demux_data; 
1142         }
1143     }
1144
1145     vlc_mutex_lock( &p_input->stream.control.control_lock );
1146     if( ( p_es == NULL ) || (p_es->b_audio && p_input->stream.control.b_mute) )
1147     {
1148         /* Not selected. Just read the adaptation field for a PCR. */
1149         b_trash = 1;
1150     }
1151     else if( p_es->p_decoder_fifo == NULL && !b_psi )
1152     {
1153         b_trash = 1; 
1154     }
1155
1156     vlc_mutex_unlock( &p_input->stream.control.control_lock );
1157     vlc_mutex_unlock( &p_input->stream.stream_lock );
1158
1159
1160     /* Don't change the order of the tests : if b_psi then p_pgrm_demux 
1161      * may still be null. Who said it was ugly ?
1162      * I have written worse. --Meuuh */
1163     if( ( p_es  ) && 
1164         ((p_es->p_decoder_fifo != NULL) || b_psi || b_pcr ) )
1165     {
1166         p_es->c_packets++;
1167
1168         /* Extract adaptation field information if any */
1169
1170         if( !b_adaptation )
1171         {
1172             /* We don't have any adaptation_field, so payload starts
1173              * immediately after the 4 byte TS header */
1174             p_data->p_payload_start += 4;
1175         }
1176         else
1177         {
1178             /* p[4] is adaptation_field_length minus one */
1179             p_data->p_payload_start += 5 + p[4];
1180     
1181             /* The adaptation field can be limited to the
1182              * adaptation_field_length byte, so that there is nothing to do:
1183              * skip this possibility */
1184             if( p[4] )
1185             {
1186                 /* If the packet has both adaptation_field and payload,
1187                  * adaptation_field cannot be more than 182 bytes long; if
1188                  * there is only an adaptation_field, it must fill the next
1189                  * 183 bytes. */
1190                 if( b_payload ? (p[4] > 182) : (p[4] != 183) )
1191                 {
1192                     intf_WarnMsg( 2,
1193                         "input: invalid TS adaptation field (%p)",
1194                         p_data );
1195                     p_data->b_discard_payload = 1;
1196                     p_es->c_invalid_packets++;
1197                 }
1198     
1199                 /* Now we are sure that the byte containing flags is present:
1200                  * read it */
1201                 else
1202                 {
1203                     /* discontinuity_indicator */
1204                     if( p[5] & 0x80 )
1205                     {
1206                         intf_WarnMsg( 2,
1207                             "input: discontinuity_indicator"
1208                             " encountered by TS demux (position read: %d,"
1209                             " saved: %d)",
1210                             p[5] & 0x80, p_es_demux->i_continuity_counter );
1211     
1212                         /* If the PID carries the PCR, there will be a system
1213                          * time-based discontinuity. We let the PCR decoder
1214                          * handle that. */
1215                         p_es->p_pgrm->i_synchro_state = SYNCHRO_REINIT;
1216     
1217                         /* There also may be a continuity_counter
1218                          * discontinuity: resynchronize our counter with
1219                          * the one of the stream. */
1220                         p_es_demux->i_continuity_counter = (p[3] & 0x0f) - 1;
1221                     }
1222     
1223                 } /* valid TS adaptation field ? */
1224             } /* length > 0 */
1225         } /* has adaptation field */
1226         /* Check the continuity of the stream. */
1227         i_dummy = ((p[3] & 0x0f) - p_es_demux->i_continuity_counter) & 0x0f;
1228         if( i_dummy == 1 )
1229         {
1230             /* Everything is ok, just increase our counter */
1231             (p_es_demux->i_continuity_counter)++;
1232         }
1233         else
1234         {
1235             if( !b_payload && i_dummy == 0 )
1236             {
1237                 /* This is a packet without payload, this is allowed by the
1238                  * draft. As there is nothing interesting in this packet
1239                  * (except PCR that have already been handled), we can trash
1240                  * the packet. */
1241                 b_trash = 1;
1242             }
1243             else if( i_dummy <= 0 )
1244             {
1245                 /* Duplicate packet: mark it as being to be trashed. */
1246                 intf_WarnMsg( 3, "input: duplicate packet received "
1247                                  "by TS demux" );
1248                 b_trash = 1;
1249             }
1250             else if( p_es_demux->i_continuity_counter == 0xFF )
1251             {
1252                 /* This means that the packet is the first one we receive for
1253                  * this ES since the continuity counter ranges between 0 and
1254                  * 0x0F excepts when it has been initialized by the input:
1255                  * init the counter to the correct value. */
1256                 intf_WarnMsg( 3, "input: first packet for PID %d received "
1257                              "by TS demux", p_es->i_id );
1258                 p_es_demux->i_continuity_counter = (p[3] & 0x0f);
1259             }
1260             else
1261             {
1262                 /* This can indicate that we missed a packet or that the
1263                  * continuity_counter wrapped and we received a dup packet:
1264                  * as we don't know, do as if we missed a packet to be sure
1265                  * to recover from this situation */
1266                 intf_WarnMsg( 2, "input: packet lost by TS demux: "
1267                                  "current %d, packet %d",
1268                               p_es_demux->i_continuity_counter & 0x0f,
1269                               p[3] & 0x0f );
1270                 b_lost = 1;
1271                 p_es_demux->i_continuity_counter = p[3] & 0x0f;
1272             } /* not continuous */
1273         } /* continuity */
1274     } /* if selected or PCR */
1275     
1276     /* Handle PCR */
1277     if( b_pcr && b_adaptation && (p[5] & 0x10) && p[4]>=7 )
1278     {
1279         /* Read the PCR. */
1280         mtime_t     pcr_time;
1281         pcr_time = ( (mtime_t)p[6] << 25 ) |
1282                    ( (mtime_t)p[7] << 17 ) |
1283                    ( (mtime_t)p[8] << 9 ) |
1284                    ( (mtime_t)p[9] << 1 ) |
1285                    ( (mtime_t)p[10] >> 7 );
1286         /* Call the pace control. */
1287         for( i_dummy = 0; i_dummy < p_input->stream.i_pgrm_number; 
1288                                 i_dummy ++ )
1289         {
1290             if( ( ( pgrm_ts_data_t * ) p_input->stream.pp_programs[i_dummy]->
1291                         p_demux_data )->i_pcr_pid == i_pid )
1292             {
1293                 input_ClockManageRef( p_input,
1294                     p_input->stream.pp_programs[i_dummy], pcr_time );
1295             }
1296         }
1297
1298     }
1299     
1300     /* Trash the packet if it has no payload or if it isn't selected */
1301     if( b_trash )
1302     {
1303         input_DeletePacket( p_input->p_method_data, p_data );
1304         p_input->stream.c_packets_trashed++;
1305     }
1306     else
1307     {
1308         if( b_psi )
1309         {
1310             /* The payload contains PSI tables */
1311             (* pf_psi_callback) ( p_input, p_data, p_es, b_unit_start );
1312         }
1313         else
1314         {
1315             /* The payload carries a PES stream */
1316             input_GatherPES( p_input, p_data, p_es, b_unit_start, b_lost ); 
1317         }
1318
1319     }
1320
1321 #undef p
1322
1323 }