]> git.sesse.net Git - vlc/blob - plugins/dvd/input_dvd.c
ea8c23cce4a7a29a5cb93fbd0608f199588b9619
[vlc] / plugins / dvd / input_dvd.c
1 /*****************************************************************************
2  * input_dvd.c: DVD raw reading plugin.
3  *****************************************************************************
4  * This plugins should handle all the known specificities of the DVD format,
5  * especially the 2048 bytes logical block size.
6  * It depends on:
7  *  -input_netlist used to read packets
8  *  -libdvdcss for access and unscrambling
9  *  -dvd_ifo for ifo parsing and analyse
10  *  -dvd_udf to find files
11  *****************************************************************************
12  * Copyright (C) 1998-2001 VideoLAN
13  * $Id: input_dvd.c,v 1.98 2001/11/19 15:13:11 stef Exp $
14  *
15  * Author: Stéphane Borel <stef@via.ecp.fr>
16  *
17  * This program is free software; you can redistribute it and/or modify
18  * it under the terms of the GNU General Public License as published by
19  * the Free Software Foundation; either version 2 of the License, or
20  * (at your option) any later version.
21  * 
22  * This program is distributed in the hope that it will be useful,
23  * but WITHOUT ANY WARRANTY; without even the implied warranty of
24  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
25  * GNU General Public License for more details.
26  *
27  * You should have received a copy of the GNU General Public License
28  * along with this program; if not, write to the Free Software
29  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
30  *****************************************************************************/
31
32 #define MODULE_NAME dvd
33 #include "modules_inner.h"
34
35 /*****************************************************************************
36  * Preamble
37  *****************************************************************************/
38 #include "defs.h"
39
40 #include <stdio.h>
41 #include <stdlib.h>
42
43 #ifdef HAVE_UNISTD_H
44 #   include <unistd.h>
45 #endif
46
47 #include <fcntl.h>
48 #include <sys/types.h>
49 #include <string.h>
50 #include <errno.h>
51
52 #ifdef STRNCASECMP_IN_STRINGS_H
53 #   include <strings.h>
54 #endif
55
56 #if defined( WIN32 )
57 #   include <io.h>                                                 /* read() */
58 #else
59 #   include <sys/uio.h>                                      /* struct iovec */
60 #endif
61
62 #ifdef GOD_DAMN_DMCA
63 #   include "dummy_dvdcss.h"
64 #else
65 #   include <videolan/dvdcss.h>
66 #endif
67
68 #include "config.h"
69 #include "common.h"
70 #include "threads.h"
71 #include "mtime.h"
72 #include "iso_lang.h"
73 #include "tests.h"
74
75 #if defined( WIN32 )
76 #   include "input_iovec.h"
77 #endif
78
79 #include "intf_msg.h"
80
81 #include "main.h"
82
83 #include "stream_control.h"
84 #include "input_ext-intf.h"
85 #include "input_ext-dec.h"
86 #include "input_ext-plugins.h"
87
88 #include "input_dvd.h"
89 #include "dvd_ifo.h"
90 #include "dvd_summary.h"
91
92 #include "debug.h"
93
94 #include "modules.h"
95 #include "modules_export.h"
96
97 /* how many blocks DVDRead will read in each loop */
98 #define DVD_BLOCK_READ_ONCE 64
99 #define DVD_DATA_READ_ONCE  (4 * DVD_BLOCK_READ_ONCE)
100
101 /* Size of netlist */
102 #define DVD_NETLIST_SIZE    256
103
104 /*****************************************************************************
105  * Local prototypes
106  *****************************************************************************/
107 /* called from outside */
108 static int  DVDProbe    ( probedata_t *p_data );
109 static void DVDInit     ( struct input_thread_s * );
110 static void DVDEnd      ( struct input_thread_s * );
111 static void DVDOpen     ( struct input_thread_s * );
112 static void DVDClose    ( struct input_thread_s * );
113 static int  DVDSetArea  ( struct input_thread_s *, struct input_area_s * );
114 static int  DVDRead     ( struct input_thread_s *, data_packet_t ** );
115 static void DVDSeek     ( struct input_thread_s *, off_t );
116 static int  DVDRewind   ( struct input_thread_s * );
117
118 /* called only inside */
119 static int  DVDChooseAngle( thread_dvd_data_t * );
120 static int  DVDFindCell( thread_dvd_data_t * );
121 static int  DVDFindSector( thread_dvd_data_t * );
122 static int  DVDChapterSelect( thread_dvd_data_t *, int );
123
124 /*****************************************************************************
125  * Functions exported as capabilities. They are declared as static so that
126  * we don't pollute the namespace too much.
127  *****************************************************************************/
128 void _M( input_getfunctions )( function_list_t * p_function_list )
129 {
130 #define input p_function_list->functions.input
131     p_function_list->pf_probe = DVDProbe;
132     input.pf_init             = DVDInit;
133     input.pf_open             = DVDOpen;
134     input.pf_close            = DVDClose;
135     input.pf_end              = DVDEnd;
136     input.pf_init_bit_stream  = InitBitstream;
137     input.pf_read             = DVDRead;
138     input.pf_set_area         = DVDSetArea;
139     input.pf_demux            = input_DemuxPS;
140     input.pf_new_packet       = input_NetlistNewPacket;
141     input.pf_new_pes          = input_NetlistNewPES;
142     input.pf_delete_packet    = input_NetlistDeletePacket;
143     input.pf_delete_pes       = input_NetlistDeletePES;
144     input.pf_rewind           = DVDRewind;
145     input.pf_seek             = DVDSeek;
146 #undef input
147 }
148
149 /*
150  * Data reading functions
151  */
152
153 /*****************************************************************************
154  * DVDProbe: verifies that the stream is a PS stream
155  *****************************************************************************/
156 static int DVDProbe( probedata_t *p_data )
157 {
158     input_thread_t * p_input = (input_thread_t *)p_data;
159
160     char * psz_name = p_input->p_source;
161     int i_score = 5;
162
163     if( TestMethod( INPUT_METHOD_VAR, "dvd" ) )
164     {
165         return( 999 );
166     }
167
168     if( ( strlen(psz_name) > 4 ) && !strncasecmp( psz_name, "dvd:", 4 ) )
169     {
170         /* If the user specified "dvd:" then it's probably a DVD */
171         i_score = 100;
172         psz_name += 4;
173     }
174
175     return( i_score );
176 }
177
178 /*****************************************************************************
179  * DVDInit: initializes DVD structures
180  *****************************************************************************/
181 static void DVDInit( input_thread_t * p_input )
182 {
183     thread_dvd_data_t *  p_dvd;
184     input_area_t *       p_area;
185     int                  i_title;
186     int                  i_chapter;
187     int                  i;
188
189     p_dvd = malloc( sizeof(thread_dvd_data_t) );
190     if( p_dvd == NULL )
191     {
192         intf_ErrMsg( "dvd error: out of memory" );
193         p_input->b_error = 1;
194         return;
195     }
196
197     p_input->p_plugin_data = (void *)p_dvd;
198     p_input->p_method_data = NULL;
199
200     p_dvd->dvdhandle = (dvdcss_handle) p_input->p_handle;
201
202     if( dvdcss_seek( p_dvd->dvdhandle, 0, DVDCSS_NOFLAGS ) < 0 )
203     {
204         intf_ErrMsg( "dvd error: %s", dvdcss_error( p_dvd->dvdhandle ) );
205         p_input->b_error = 1;
206         return;
207     }
208
209     /* We read DVD_BLOCK_READ_ONCE in each loop, so the input will receive
210      * DVD_DATA_READ_ONCE at most */
211     p_dvd->i_block_once = DVD_BLOCK_READ_ONCE;
212     /* this value mustn't be modifed */
213     p_input->i_read_once = DVD_DATA_READ_ONCE;
214
215     /* Reading structures initialisation */
216     input_NetlistInit( p_input, DVD_NETLIST_SIZE, 2 * DVD_NETLIST_SIZE,
217                        DVD_NETLIST_SIZE, DVD_LB_SIZE, p_dvd->i_block_once );
218     intf_WarnMsg( 2, "dvd info: netlist initialized" );
219
220     /* Ifo allocation & initialisation */
221     if( IfoCreate( p_dvd ) < 0 )
222     {
223         intf_ErrMsg( "dvd error: allcation error in ifo" );
224         dvdcss_close( p_dvd->dvdhandle );
225         free( p_dvd );
226         p_input->b_error = 1;
227         return;
228     }
229
230     if( IfoInit( p_dvd->p_ifo ) < 0 )
231     {
232         intf_ErrMsg( "dvd error: fatal failure in ifo" );
233         IfoDestroy( p_dvd->p_ifo );
234         dvdcss_close( p_dvd->dvdhandle );
235         free( p_dvd );
236         p_input->b_error = 1;
237         return;
238     }
239
240     /* Set stream and area data */
241     vlc_mutex_lock( &p_input->stream.stream_lock );
242
243     /* Initialize ES structures */
244     input_InitStream( p_input, sizeof( stream_ps_data_t ) );
245
246 #define title_inf p_dvd->p_ifo->vmg.title_inf
247     intf_WarnMsg( 2, "dvd info: number of titles: %d", title_inf.i_title_nb );
248
249 #define area p_input->stream.pp_areas
250     /* We start from 1 here since the default area 0
251      * is reserved for video_ts.vob */
252     for( i = 1 ; i <= title_inf.i_title_nb ; i++ )
253     {
254         input_AddArea( p_input );
255
256         /* Titles are Program Chains */
257         area[i]->i_id = i;
258
259         /* Absolute start offset and size 
260          * We can only set that with vts ifo, so we do it during the
261          * first call to DVDSetArea */
262         area[i]->i_start = 0;
263         area[i]->i_size = 0;
264
265         /* Number of chapters */
266         area[i]->i_part_nb = title_inf.p_attr[i-1].i_chapter_nb;
267         area[i]->i_part = 1;
268
269         /* Number of angles */
270         area[i]->i_angle_nb = 0;
271         area[i]->i_angle = 1;
272
273         /* Offset to vts_i_0.ifo */
274         area[i]->i_plugin_data = p_dvd->p_ifo->i_start +
275                        title_inf.p_attr[i-1].i_start_sector;
276     }   
277 #undef area
278
279     /* Get requested title - if none try the first title */
280     i_title = main_GetIntVariable( INPUT_TITLE_VAR, 1 );
281     if( i_title <= 0 || i_title > title_inf.i_title_nb )
282     {
283         i_title = 1;
284     }
285
286 #undef title_inf
287
288     /* Get requested chapter - if none defaults to first one */
289     i_chapter = main_GetIntVariable( INPUT_CHAPTER_VAR, 1 );
290     if( i_chapter <= 0 )
291     {
292         i_chapter = 1;
293     }
294
295     p_input->stream.pp_areas[i_title]->i_part = i_chapter;
296
297     p_area = p_input->stream.pp_areas[i_title];
298
299     /* set title, chapter, audio and subpic */
300     DVDSetArea( p_input, p_area );
301
302     vlc_mutex_unlock( &p_input->stream.stream_lock );
303
304     return;
305 }
306
307 /*****************************************************************************
308  * DVDOpen: open dvd
309  *****************************************************************************/
310 static void DVDOpen( struct input_thread_s *p_input )
311 {
312     dvdcss_handle dvdhandle;
313
314     vlc_mutex_lock( &p_input->stream.stream_lock );
315
316     /* If we are here we can control the pace... */
317     p_input->stream.b_pace_control = 1;
318
319     p_input->stream.b_seekable = 1;
320     p_input->stream.p_selected_area->i_size = 0;
321
322     p_input->stream.p_selected_area->i_tell = 0;
323
324     vlc_mutex_unlock( &p_input->stream.stream_lock );
325
326     /* XXX: put this shit in an access plugin */
327     if( strlen( p_input->p_source ) > 4
328          && !strncasecmp( p_input->p_source, "dvd:", 4 ) )
329     {
330         dvdhandle = dvdcss_open( p_input->p_source + 4 );
331     }
332     else
333     {
334         dvdhandle = dvdcss_open( p_input->p_source );
335     }
336
337     if( dvdhandle == NULL )
338     {
339         intf_ErrMsg( "dvd error: dvdcss can't open device" );
340         p_input->b_error = 1;
341         return;
342     }
343
344     p_input->p_handle = (void *) dvdhandle;
345 }
346
347 /*****************************************************************************
348  * DVDClose: close dvd
349  *****************************************************************************/
350 static void DVDClose( struct input_thread_s *p_input )
351 {
352     /* Clean up libdvdcss */
353     dvdcss_close( (dvdcss_handle) p_input->p_handle );
354 }
355
356 /*****************************************************************************
357  * DVDEnd: frees unused data
358  *****************************************************************************/
359 static void DVDEnd( input_thread_t * p_input )
360 {
361     thread_dvd_data_t *     p_dvd;
362
363     p_dvd = (thread_dvd_data_t*)p_input->p_plugin_data;
364
365     IfoDestroy( p_dvd->p_ifo );
366
367     free( p_dvd );
368
369     input_NetlistEnd( p_input );
370 }
371
372 /*****************************************************************************
373  * DVDSetArea: initialize input data for title x, chapter y.
374  * It should be called for each user navigation request.
375  *****************************************************************************
376  * Take care that i_title starts from 0 (vmg) and i_chapter start from 1.
377  * Note that you have to take the lock before entering here.
378  *****************************************************************************/
379 static int DVDSetArea( input_thread_t * p_input, input_area_t * p_area )
380 {
381     thread_dvd_data_t *  p_dvd;
382     es_descriptor_t *    p_es;
383     u16                  i_id;
384     int                  i_vts_title;
385     int                  i_audio_nb = 0;
386     int                  i_spu_nb = 0;
387     int                  i_audio;
388     int                  i_spu;
389     int                  i;
390
391     p_dvd = (thread_dvd_data_t*)p_input->p_plugin_data;
392
393     /* we can't use the interface slider until initilization is complete */
394     p_input->stream.b_seekable = 0;
395
396     if( p_area != p_input->stream.p_selected_area )
397     {
398         /* Reset the Chapter position of the old title */
399         p_input->stream.p_selected_area->i_part = 0;
400
401         /*
402          *  We have to load all title information
403          */
404         /* Change the default area */
405         p_input->stream.p_selected_area =
406                     p_input->stream.pp_areas[p_area->i_id];
407
408         /* title number: it is not vts nb!,
409          * it is what appears in the interface list */
410         p_dvd->i_title = p_area->i_id;
411         p_dvd->p_ifo->i_title = p_dvd->i_title;
412
413         /* set number of chapters of current title */
414         p_dvd->i_chapter_nb = p_area->i_part_nb;
415
416         /* ifo vts */
417         if( IfoTitleSet( p_dvd->p_ifo ) < 0 )
418         {
419             intf_ErrMsg( "dvd error: fatal error in vts ifo" );
420             free( p_dvd );
421             p_input->b_error = 1;
422             return -1;
423         }
424
425 #define vmg p_dvd->p_ifo->vmg
426 #define vts p_dvd->p_ifo->vts
427         /* title position inside the selected vts */
428         i_vts_title = vmg.title_inf.p_attr[p_dvd->i_title-1].i_title_num;
429         p_dvd->i_title_id =
430             vts.title_inf.p_title_start[i_vts_title-1].i_title_id;
431
432         intf_WarnMsgImm( 3, "dvd: title %d vts_title %d pgc %d",
433                          p_dvd->i_title, i_vts_title, p_dvd->i_title_id );
434
435
436         /*
437          * Angle management
438          */
439         p_dvd->i_angle_nb = vmg.title_inf.p_attr[p_dvd->i_title-1].i_angle_nb;
440         p_dvd->i_angle = main_GetIntVariable( INPUT_ANGLE_VAR, 1 );
441         if( ( p_dvd->i_angle <= 0 ) || p_dvd->i_angle > p_dvd->i_angle_nb )
442         {
443             p_dvd->i_angle = 1;
444         }
445     
446         /*
447          * Set selected title start and size
448          */
449         
450         /* title set offset XXX: convert to block values */
451         p_dvd->i_title_start =
452             vts.i_pos + vts.manager_inf.i_title_vob_start_sector;
453
454         /* last video cell */
455         p_dvd->i_cell = 0;
456         p_dvd->i_prg_cell = -1 +
457             vts.title_unit.p_title[p_dvd->i_title_id-1].title.i_cell_nb;
458
459         if( DVDFindCell( p_dvd ) < 0 )
460         {
461             intf_ErrMsg( "dvd error: can't find title end" );
462             p_input->b_error = 1;
463             return -1;
464         }
465         
466         /* temporary hack to fix size in some dvds */
467         if( p_dvd->i_cell >= vts.cell_inf.i_cell_nb )
468         {
469             p_dvd->i_cell = vts.cell_inf.i_cell_nb - 1;
470         }
471
472         p_dvd->i_sector = 0;
473         p_dvd->i_size = vts.cell_inf.p_cell_map[p_dvd->i_cell].i_end_sector;
474
475         if( DVDChapterSelect( p_dvd, 1 ) < 0 )
476         {
477             intf_ErrMsg( "dvd error: can't find first chapter" );
478             p_input->b_error = 1;
479             return -1;
480         }
481         
482         /* Force libdvdcss to check its title key.
483          * It is only useful for title cracking method. Methods using the
484          * decrypted disc key are fast enough to check the key at each seek */
485
486         if( dvdcss_seek( p_dvd->dvdhandle, p_dvd->i_start,
487                                 DVDCSS_SEEK_KEY ) < 0 )
488         {
489             intf_ErrMsg( "dvd error: %s", dvdcss_error( p_dvd->dvdhandle ) );
490             return -1;
491         }
492
493         p_dvd->i_size -= p_dvd->i_sector + 1;
494
495         IfoPrintTitle( p_dvd );
496
497         /* Area definition */
498         p_input->stream.p_selected_area->i_start = LB2OFF( p_dvd->i_start );
499         p_input->stream.p_selected_area->i_size = LB2OFF( p_dvd->i_size );
500         p_input->stream.p_selected_area->i_angle_nb = p_dvd->i_angle_nb;
501         p_input->stream.p_selected_area->i_angle = p_dvd->i_angle;
502
503 #if 0
504         /* start at the beginning of the title */
505         /* FIXME: create a conf option to select whether to restart
506          * title or not */
507         p_input->stream.p_selected_area->i_tell = 0;
508         p_input->stream.p_selected_area->i_part = 1;
509 #endif
510
511         /*
512          * Destroy obsolete ES by reinitializing program 0
513          * and find all ES in title with ifo data
514          */
515         if( p_input->stream.pp_programs != NULL )
516         {
517             /* We don't use input_EndStream here since
518              * we keep area structures */
519
520             for( i = 0 ; i < p_input->stream.i_selected_es_number ; i++ )
521             {
522                 input_UnselectES( p_input, p_input->stream.pp_selected_es[i] );
523             }
524
525             free( p_input->stream.pp_selected_es );
526             input_DelProgram( p_input, p_input->stream.pp_programs[0] );
527
528             p_input->stream.pp_selected_es = NULL;
529             p_input->stream.i_selected_es_number = 0;
530         }
531
532         input_AddProgram( p_input, 0, sizeof( stream_ps_data_t ) );
533
534         /* No PSM to read in DVD mode, we already have all information */
535         p_input->stream.pp_programs[0]->b_is_ok = 1;
536
537         p_es = NULL;
538
539         /* ES 0 -> video MPEG2 */
540         IfoPrintVideo( p_dvd );
541
542         p_es = input_AddES( p_input, p_input->stream.pp_programs[0], 0xe0, 0 );
543         p_es->i_stream_id = 0xe0;
544         p_es->i_type = MPEG2_VIDEO_ES;
545         p_es->i_cat = VIDEO_ES;
546         if( p_main->b_video )
547         {
548             input_SelectES( p_input, p_es );
549         }
550
551 #define audio_status \
552     vts.title_unit.p_title[p_dvd->i_title_id-1].title.pi_audio_status[i-1]
553         /* Audio ES, in the order they appear in .ifo */
554         for( i = 1 ; i <= vts.manager_inf.i_audio_nb ; i++ )
555         {
556             IfoPrintAudio( p_dvd, i );
557
558             /* audio channel is active if first byte is 0x80 */
559             if( audio_status.i_available )
560             {
561                 i_audio_nb++;
562
563                 switch( vts.manager_inf.p_audio_attr[i-1].i_coding_mode )
564                 {
565                 case 0x00:              /* AC3 */
566                     i_id = ( ( 0x80 + audio_status.i_position ) << 8 ) | 0xbd;
567                     p_es = input_AddES( p_input,
568                                p_input->stream.pp_programs[0], i_id, 0 );
569                     p_es->i_stream_id = 0xbd;
570                     p_es->i_type = AC3_AUDIO_ES;
571                     p_es->b_audio = 1;
572                     p_es->i_cat = AUDIO_ES;
573                     strcpy( p_es->psz_desc, DecodeLanguage( hton16(
574                         vts.manager_inf.p_audio_attr[i-1].i_lang_code ) ) ); 
575                     strcat( p_es->psz_desc, " (ac3)" );
576     
577                     break;
578                 case 0x02:
579                 case 0x03:              /* MPEG audio */
580                     i_id = 0xc0 + audio_status.i_position;
581                     p_es = input_AddES( p_input,
582                                     p_input->stream.pp_programs[0], i_id, 0 );
583                     p_es->i_stream_id = i_id;
584                     p_es->i_type = MPEG2_AUDIO_ES;
585                     p_es->b_audio = 1;
586                     p_es->i_cat = AUDIO_ES;
587                     strcpy( p_es->psz_desc, DecodeLanguage( hton16(
588                         vts.manager_inf.p_audio_attr[i-1].i_lang_code ) ) ); 
589                     strcat( p_es->psz_desc, " (mpeg)" );
590     
591                     break;
592                 case 0x04:              /* LPCM */
593     
594                     i_id = ( ( 0xa0 + audio_status.i_position ) << 8 ) | 0xbd;
595                     p_es = input_AddES( p_input,
596                                     p_input->stream.pp_programs[0], i_id, 0 );
597                     p_es->i_stream_id = 0xbd;
598                     p_es->i_type = LPCM_AUDIO_ES;
599                     p_es->b_audio = 1;
600                     p_es->i_cat = AUDIO_ES;
601                     strcpy( p_es->psz_desc, DecodeLanguage( hton16(
602                         vts.manager_inf.p_audio_attr[i-1].i_lang_code ) ) ); 
603                     strcat( p_es->psz_desc, " (lpcm)" );
604     
605                     break;
606                 case 0x06:              /* DTS */
607                     i_id = ( ( 0x88 + audio_status.i_position ) << 8 ) | 0xbd;
608                     intf_ErrMsg( "dvd warning: DTS audio not handled yet"
609                                  "(0x%x)", i_id );
610                     break;
611                 default:
612                     i_id = 0;
613                     intf_ErrMsg( "dvd warning: unknown audio type %.2x",
614                              vts.manager_inf.p_audio_attr[i-1].i_coding_mode );
615                 }
616             }
617         }
618 #undef audio_status
619 #define spu_status \
620     vts.title_unit.p_title[p_dvd->i_title_id-1].title.pi_spu_status[i-1]
621
622         /* Sub Picture ES */
623            
624         for( i = 1 ; i <= vts.manager_inf.i_spu_nb; i++ )
625         {
626             IfoPrintSpu( p_dvd, i );
627
628             if( spu_status.i_available )
629             {
630                 i_spu_nb++;
631
632                 /*  there are several streams for one spu */
633                 if(  vts.manager_inf.video_attr.i_ratio )
634                 {
635                     /* 16:9 */
636                     switch( vts.manager_inf.video_attr.i_perm_displ )
637                     {
638                     case 1:
639                         i_id = ( ( 0x20 + spu_status.i_position_pan ) << 8 )
640                                | 0xbd;
641                         break;
642                     case 2:
643                         i_id = ( ( 0x20 + spu_status.i_position_letter ) << 8 )
644                                | 0xbd;
645                         break;
646                     default:
647                         i_id = ( ( 0x20 + spu_status.i_position_wide ) << 8 )
648                                | 0xbd;
649                         break;
650                     }
651                 }
652                 else
653                 {
654                     /* 4:3 */
655                     i_id = ( ( 0x20 + spu_status.i_position_43 ) << 8 )
656                            | 0xbd;
657                 }
658                 p_es = input_AddES( p_input,
659                                     p_input->stream.pp_programs[0], i_id, 0 );
660                 p_es->i_stream_id = 0xbd;
661                 p_es->i_type = DVD_SPU_ES;
662                 p_es->i_cat = SPU_ES;
663                 strcpy( p_es->psz_desc, DecodeLanguage( hton16(
664                     vts.manager_inf.p_spu_attr[i-1].i_lang_code ) ) ); 
665             }
666         }
667 #undef spu_status
668         if( p_main->b_audio )
669         {
670             /* For audio: first one if none or a not existing one specified */
671             i_audio = main_GetIntVariable( INPUT_CHANNEL_VAR, 1 );
672             if( i_audio < 0 || i_audio > i_audio_nb )
673             {
674                 main_PutIntVariable( INPUT_CHANNEL_VAR, 1 );
675                 i_audio = 1;
676             }
677             if( i_audio > 0 && i_audio_nb > 0 )
678             {
679                 if( main_GetIntVariable( AOUT_SPDIF_VAR, 0 ) ||
680                     ( main_GetIntVariable( INPUT_AUDIO_VAR, 0 ) ==
681                       REQUESTED_AC3 ) )
682                 {
683                     int     i_ac3 = i_audio;
684                     while( ( p_input->stream.pp_es[i_ac3]->i_type !=
685                              AC3_AUDIO_ES ) && ( i_ac3 <=
686                              vts.manager_inf.i_audio_nb ) )
687                     {
688                         i_ac3++;
689                     }
690                     if( p_input->stream.pp_es[i_ac3]->i_type == AC3_AUDIO_ES )
691                     {
692                         input_SelectES( p_input,
693                                         p_input->stream.pp_es[i_ac3] );
694                     }
695                 }
696                 else
697                 {
698                     input_SelectES( p_input,
699                                     p_input->stream.pp_es[i_audio] );
700                 }
701             }
702         }
703
704         if( p_main->b_video )
705         {
706             /* for spu, default is none */
707             i_spu = main_GetIntVariable( INPUT_SUBTITLE_VAR, 0 );
708             if( i_spu < 0 || i_spu > i_spu_nb )
709             {
710                 main_PutIntVariable( INPUT_SUBTITLE_VAR, 0 );
711                 i_spu = 0;
712             }
713             if( i_spu > 0 && i_spu_nb > 0 )
714             {
715                 i_spu += vts.manager_inf.i_audio_nb;
716                 input_SelectES( p_input, p_input->stream.pp_es[i_spu] );
717             }
718         }
719     } /* i_title >= 0 */
720     else
721     {
722         p_area = p_input->stream.p_selected_area;
723     }
724 #undef vts
725 #undef vmg
726
727     /*
728      * Chapter selection
729      */
730
731     if( p_area->i_part != p_dvd->i_chapter )
732     {
733         if( ( p_area->i_part > 0 ) &&
734             ( p_area->i_part <= p_area->i_part_nb ))
735         {
736             if( DVDChapterSelect( p_dvd, p_area->i_part ) < 0 )
737             {
738                 intf_ErrMsg( "dvd error: can't set chapter in area" );
739                 p_input->b_error = 1;
740                 return -1;
741             }
742     
743             p_input->stream.p_selected_area->i_tell =
744                                    LB2OFF( p_dvd->i_start ) - p_area->i_start;
745             p_input->stream.p_selected_area->i_part = p_dvd->i_chapter;
746     
747             intf_WarnMsg( 4, "dvd info: chapter %d start at: %lld",
748                                         p_area->i_part, p_area->i_tell );
749         }
750         else
751         {
752             p_area->i_part = 1;
753             p_dvd->i_chapter = 1;
754         }
755     }
756
757 #define title \
758     p_dvd->p_ifo->vts.title_unit.p_title[p_dvd->i_title_id-1].title
759     if( p_area->i_angle != p_dvd->i_angle )
760     {
761         if( title.p_cell_play[p_dvd->i_prg_cell].i_category & 0xf000 )
762         {
763             if( ( p_area->i_angle - p_dvd->i_angle ) < 0 )
764             {
765                 p_dvd->i_cell = 0;
766             }
767             p_dvd->i_prg_cell += ( p_area->i_angle - p_dvd->i_angle );
768             p_dvd->i_angle = p_area->i_angle;
769     
770             DVDFindSector( p_dvd );
771             p_dvd->i_cell += p_dvd->i_angle_cell;
772         }
773         else
774         {
775             p_dvd->i_angle = p_area->i_angle;
776         }
777
778         intf_WarnMsg( 3, "dvd info: angle %d selected", p_area->i_angle );
779     }
780
781     /* warn interface that something has changed */
782     p_input->stream.b_seekable = 1;
783     p_input->stream.b_changed = 1;
784
785     return 0;
786 }
787
788
789 /*****************************************************************************
790  * DVDRead: reads data packets into the netlist.
791  *****************************************************************************
792  * Returns -1 in case of error, 0 if everything went well, and 1 in case of
793  * EOF.
794  *****************************************************************************/
795 static int DVDRead( input_thread_t * p_input,
796                     data_packet_t ** pp_packets )
797 {
798     thread_dvd_data_t *     p_dvd;
799     netlist_t *             p_netlist;
800     struct iovec *          p_vec;
801     struct data_packet_s *  pp_data[DVD_DATA_READ_ONCE];
802     u8 *                    pi_cur;
803     int                     i_block_once;
804     int                     i_packet_size;
805     int                     i_iovec;
806     int                     i_packet;
807     int                     i_pos;
808     int                     i_read_blocks;
809     int                     i_sector;
810     boolean_t               b_eof;
811     boolean_t               b_eot;
812     boolean_t               b_eoc;
813
814     p_dvd = (thread_dvd_data_t *)p_input->p_plugin_data;
815     p_netlist = (netlist_t *)p_input->p_method_data;
816
817     b_eoc = 0;
818     i_sector = p_dvd->i_title_start + p_dvd->i_sector;
819     i_block_once = p_dvd->i_end_sector - p_dvd->i_sector + 1;
820
821
822     /* Get the position of the next cell if we're at cell end */
823     if( i_block_once <= 0 )
824     {
825         int     i_angle;
826
827         p_dvd->i_cell++;
828         p_dvd->i_angle_cell++;
829
830         /* Find cell index in adress map */
831         if( DVDFindSector( p_dvd ) < 0 )
832         {
833             pp_packets[0] = NULL;
834             intf_ErrMsg( "dvd error: can't find next cell" );
835             return 1;
836         }
837
838         /* Position the fd pointer on the right address */
839         if( ( i_sector = dvdcss_seek( p_dvd->dvdhandle,
840                                       p_dvd->i_title_start + p_dvd->i_sector,
841                                       DVDCSS_SEEK_MPEG ) ) < 0 )
842         {
843             intf_ErrMsg( "dvd error: %s", dvdcss_error( p_dvd->dvdhandle ) );
844             return -1;
845         }
846
847         /* update chapter : it will be easier when we have navigation
848          * ES support */
849         if( p_dvd->i_chapter < ( p_dvd->i_chapter_nb - 1 ) )
850         {
851             if( title.p_cell_play[p_dvd->i_prg_cell].i_category & 0xf000 )
852             {
853                 i_angle = p_dvd->i_angle - 1;
854             }
855             else
856             {
857                 i_angle = 0;
858             }
859             if( title.chapter_map.pi_start_cell[p_dvd->i_chapter] <=
860                 ( p_dvd->i_prg_cell - i_angle + 1 ) )
861             {
862                 p_dvd->i_chapter++;
863                 b_eoc = 1;
864             }
865         }
866
867         i_block_once = p_dvd->i_end_sector - p_dvd->i_sector + 1;
868     }
869
870     /* The number of blocks read is the max between the requested
871      * value and the leaving block in the cell */
872     if( i_block_once > p_dvd->i_block_once )
873     {
874         i_block_once = p_dvd->i_block_once;
875     }
876 /*
877 intf_WarnMsg( 2, "Sector: 0x%x Read: %d Chapter: %d", p_dvd->i_sector, i_block_once, p_dvd->i_chapter );
878 */
879     p_netlist->i_read_once = i_block_once;
880
881     /* Get an iovec pointer */
882     if( ( p_vec = input_NetlistGetiovec( p_netlist ) ) == NULL )
883     {
884         intf_ErrMsg( "dvd error: can't get iovec" );
885         return -1;
886     }
887
888     /* Reads from DVD */
889     i_read_blocks = dvdcss_readv( p_dvd->dvdhandle, p_vec,
890                                   i_block_once, DVDCSS_READ_DECRYPT );
891
892     /* Update netlist indexes: we don't do it in DVDGetiovec since we
893      * need know the real number of blocks read */
894     input_NetlistMviovec( p_netlist, i_read_blocks, pp_data );
895
896     /* Update global position */
897     p_dvd->i_sector += i_read_blocks;
898
899     i_packet = 0;
900
901     /* Read headers to compute payload length */
902     for( i_iovec = 0 ; i_iovec < i_read_blocks ; i_iovec++ )
903     {
904         i_pos = 0;
905
906         while( i_pos < p_netlist->i_buffer_size )
907         {
908             pi_cur = (u8*)p_vec[i_iovec].iov_base + i_pos;
909
910             /*default header */
911             if( U32_AT( pi_cur ) != 0x1BA )
912             {
913                 /* That's the case for all packets, except pack header. */
914                 i_packet_size = U16_AT( pi_cur + 4 );
915                 pp_packets[i_packet] = input_NetlistNewPtr( p_netlist );
916                 (*pp_data[i_iovec]->pi_refcount)++;
917                 pp_packets[i_packet]->pi_refcount =
918                     pp_data[i_iovec]->pi_refcount;
919                 pp_packets[i_packet]->p_buffer = pp_data[i_iovec]->p_buffer;
920             }
921             else
922             {
923                 /* MPEG-2 Pack header. */
924                 i_packet_size = 8;
925                 pp_packets[i_packet] = pp_data[i_iovec];
926             }
927
928             pp_packets[i_packet]->p_payload_start =
929                     pp_packets[i_packet]->p_buffer + i_pos;
930
931             pp_packets[i_packet]->p_payload_end =
932                     pp_packets[i_packet]->p_payload_start + i_packet_size + 6;
933
934             i_packet++;
935             i_pos += i_packet_size + 6;
936         }
937     }
938
939     pp_packets[i_packet] = NULL;
940
941     vlc_mutex_lock( &p_input->stream.stream_lock );
942
943     p_input->stream.p_selected_area->i_tell =
944         LB2OFF( i_sector + i_read_blocks ) -
945         p_input->stream.p_selected_area->i_start;
946     if( b_eoc )
947     {
948         /* We modify i_part only at end of chapter not to erase
949          * some modification from the interface */
950         p_input->stream.p_selected_area->i_part = p_dvd->i_chapter;
951     }
952
953     b_eot = !( p_input->stream.p_selected_area->i_tell
954                   < p_input->stream.p_selected_area->i_size );
955     b_eof = b_eot && ( ( p_dvd->i_title + 1 ) >= p_input->stream.i_area_nb );
956
957     if( b_eof )
958     {
959         vlc_mutex_unlock( &p_input->stream.stream_lock );
960         return 1;
961     }
962
963     if( b_eot )
964     {
965         intf_WarnMsg( 4, "dvd info: new title" );
966         p_dvd->i_title++;
967         DVDSetArea( p_input, p_input->stream.pp_areas[p_dvd->i_title] );
968         vlc_mutex_unlock( &p_input->stream.stream_lock );
969         return 0;
970     }
971
972     vlc_mutex_unlock( &p_input->stream.stream_lock );
973
974     if( i_read_blocks == i_block_once )
975     {
976         return 0;
977     }
978
979     return -1;
980 }
981
982 /*****************************************************************************
983  * DVDRewind : reads a stream backward
984  *****************************************************************************/
985 static int DVDRewind( input_thread_t * p_input )
986 {
987     return( -1 );
988 }
989
990 /*****************************************************************************
991  * DVDSeek : Goes to a given position on the stream.
992  *****************************************************************************
993  * This one is used by the input and translate chronological position from
994  * input to logical position on the device.
995  * The lock should be taken before calling this function.
996  *****************************************************************************/
997 static void DVDSeek( input_thread_t * p_input, off_t i_off )
998 {
999     thread_dvd_data_t *     p_dvd;
1000     int                     i_block;
1001     int                     i_prg_cell;
1002     int                     i_cell;
1003     int                     i_chapter;
1004     int                     i_angle;
1005     
1006     p_dvd = ( thread_dvd_data_t * )p_input->p_plugin_data;
1007
1008     /* we have to take care of offset of beginning of title */
1009     p_dvd->i_sector = OFF2LB(i_off + p_input->stream.p_selected_area->i_start)
1010                        - p_dvd->i_title_start;
1011
1012     i_prg_cell = 0;
1013     i_chapter = 0;
1014
1015     /* parse vobu address map to find program cell */
1016     while( title.p_cell_play[i_prg_cell].i_end_sector < p_dvd->i_sector  )
1017     {
1018         i_prg_cell++;
1019     }
1020
1021     p_dvd->i_prg_cell = i_prg_cell;
1022
1023     if( DVDChooseAngle( p_dvd ) < 0 )
1024     {
1025         p_input->b_error = 1;
1026         return;        
1027     }
1028
1029     p_dvd->i_cell = 0;
1030
1031     /* Find first title cell which is inside program cell */
1032     if( DVDFindCell( p_dvd ) < 0 )
1033     {
1034         /* no following cell : we're at eof */
1035         intf_ErrMsg( "dvd error: cell seeking failed" );
1036         p_input->b_error = 1;
1037         return;
1038     }
1039
1040     i_cell = p_dvd->i_cell;
1041
1042 #define cell p_dvd->p_ifo->vts.cell_inf.p_cell_map[i_cell]
1043     /* parse cell address map to find title cell containing sector */
1044     while( cell.i_end_sector < p_dvd->i_sector )
1045     {
1046         i_cell++;
1047     }
1048
1049     p_dvd->i_cell = i_cell;
1050
1051     /* if we're inside a multi-angle zone, we have to choose i_sector
1052      * in the current angle ; we can't do it all the time since cells
1053      * can be very wide out of such zones */
1054     if( title.p_cell_play[p_dvd->i_prg_cell].i_category & 0xf000 )
1055     {
1056         p_dvd->i_sector = MAX(
1057                 cell.i_start_sector,
1058                 title.p_cell_play[p_dvd->i_prg_cell].i_start_sector );
1059     }
1060
1061     p_dvd->i_end_sector = MIN(
1062             cell.i_end_sector,
1063             title.p_cell_play[p_dvd->i_prg_cell].i_end_sector );
1064 #undef cell
1065     /* update chapter */
1066     if( title.p_cell_play[p_dvd->i_prg_cell].i_category & 0xf000 )
1067     {
1068         i_angle = p_dvd->i_angle - 1;
1069     }
1070     else
1071     {
1072         i_angle = 0;
1073     }
1074     if( p_dvd->i_chapter_nb > 1 )
1075     {
1076         while( ( title.chapter_map.pi_start_cell[i_chapter] <=
1077                     ( p_dvd->i_prg_cell - i_angle + 1 ) ) &&
1078                ( i_chapter < ( p_dvd->i_chapter_nb - 1 ) ) )
1079         {
1080             i_chapter++;
1081         }
1082     }
1083     else
1084     {
1085         i_chapter = 1;
1086     }
1087
1088     p_dvd->i_chapter = i_chapter;
1089     p_input->stream.p_selected_area->i_part = p_dvd->i_chapter;
1090
1091     if( ( i_block = dvdcss_seek( p_dvd->dvdhandle,
1092                                  p_dvd->i_title_start + p_dvd->i_sector,
1093                                  DVDCSS_SEEK_MPEG ) ) < 0 )
1094     {
1095         intf_ErrMsg( "dvd error: %s", dvdcss_error( p_dvd->dvdhandle ) );
1096         p_input->b_error = 1;
1097         return;
1098     }
1099
1100     p_input->stream.p_selected_area->i_tell =
1101         LB2OFF ( i_block ) - p_input->stream.p_selected_area->i_start;
1102
1103     intf_WarnMsg( 7, "Program Cell: %d Cell: %d Chapter: %d",
1104                      p_dvd->i_prg_cell, p_dvd->i_cell, p_dvd->i_chapter );
1105
1106
1107     return;
1108 }
1109
1110 #define cell  p_dvd->p_ifo->vts.cell_inf
1111
1112 /*****************************************************************************
1113  * DVDFindCell: adjust the title cell index with the program cell
1114  *****************************************************************************/
1115 static int DVDFindCell( thread_dvd_data_t * p_dvd )
1116 {
1117     int                 i_cell;
1118     int                 i_index;
1119
1120     i_cell = p_dvd->i_cell;
1121     i_index = p_dvd->i_prg_cell;
1122
1123     if( i_cell >= cell.i_cell_nb )
1124     {
1125         return -1;
1126     }
1127
1128     while( ( ( title.p_cell_pos[i_index].i_vob_id !=
1129                    cell.p_cell_map[i_cell].i_vob_id ) ||
1130       ( title.p_cell_pos[i_index].i_cell_id !=
1131                    cell.p_cell_map[i_cell].i_cell_id ) ) &&
1132            ( i_cell < cell.i_cell_nb - 1 ) )
1133     {
1134         i_cell++;
1135     }
1136
1137 /*
1138 intf_WarnMsg( 12, "FindCell: i_cell %d i_index %d found %d nb %d",
1139                     p_dvd->i_cell,
1140                     p_dvd->i_prg_cell,
1141                     i_cell,
1142                     cell.i_cell_nb );
1143 */
1144
1145     p_dvd->i_cell = i_cell;
1146
1147     return 0;    
1148 }
1149
1150 #undef cell
1151
1152 /*****************************************************************************
1153  * DVDFindSector: find cell index in adress map from index in
1154  * information table program map and give corresponding sectors.
1155  *****************************************************************************/
1156 static int DVDFindSector( thread_dvd_data_t * p_dvd )
1157 {
1158
1159     if( p_dvd->i_sector > title.p_cell_play[p_dvd->i_prg_cell].i_end_sector )
1160     {
1161         p_dvd->i_prg_cell++;
1162
1163         if( DVDChooseAngle( p_dvd ) < 0 )
1164         {
1165             return -1;
1166         }
1167     }
1168
1169     if( DVDFindCell( p_dvd ) < 0 )
1170     {
1171         intf_ErrMsg( "dvd error: can't find sector" );
1172         return -1;
1173     }
1174     
1175     /* Find start and end sectors of new cell */
1176 #if 1
1177     p_dvd->i_sector = MAX(
1178          p_dvd->p_ifo->vts.cell_inf.p_cell_map[p_dvd->i_cell].i_start_sector,
1179          title.p_cell_play[p_dvd->i_prg_cell].i_start_sector );
1180     p_dvd->i_end_sector = MIN(
1181          p_dvd->p_ifo->vts.cell_inf.p_cell_map[p_dvd->i_cell].i_end_sector,
1182          title.p_cell_play[p_dvd->i_prg_cell].i_end_sector );
1183 #else
1184     p_dvd->i_sector = title.p_cell_play[p_dvd->i_prg_cell].i_start_sector;
1185     p_dvd->i_end_sector = title.p_cell_play[p_dvd->i_prg_cell].i_end_sector;
1186 #endif
1187
1188 /*
1189     intf_WarnMsg( 12, "cell: %d sector1: 0x%x end1: 0x%x\n"
1190                    "index: %d sector2: 0x%x end2: 0x%x\n"
1191                    "category: 0x%x ilvu end: 0x%x vobu start 0x%x", 
1192         p_dvd->i_cell,
1193         p_dvd->p_ifo->vts.cell_inf.p_cell_map[p_dvd->i_cell].i_start_sector,
1194         p_dvd->p_ifo->vts.cell_inf.p_cell_map[p_dvd->i_cell].i_end_sector,
1195         p_dvd->i_prg_cell,
1196         title.p_cell_play[p_dvd->i_prg_cell].i_start_sector,
1197         title.p_cell_play[p_dvd->i_prg_cell].i_end_sector,
1198         title.p_cell_play[p_dvd->i_prg_cell].i_category, 
1199         title.p_cell_play[p_dvd->i_prg_cell].i_first_ilvu_vobu_esector,
1200         title.p_cell_play[p_dvd->i_prg_cell].i_last_vobu_start_sector );
1201 */
1202
1203     return 0;
1204 }
1205
1206 /*****************************************************************************
1207  * DVDChapterSelect: find the cell corresponding to requested chapter
1208  *****************************************************************************/
1209 static int DVDChapterSelect( thread_dvd_data_t * p_dvd, int i_chapter )
1210 {
1211
1212     /* Find cell index in Program chain for current chapter */
1213     p_dvd->i_prg_cell = title.chapter_map.pi_start_cell[i_chapter-1] - 1;
1214     p_dvd->i_cell = 0;
1215     p_dvd->i_sector = 0;
1216
1217     DVDChooseAngle( p_dvd );
1218
1219     /* Search for cell_index in cell adress_table and initialize
1220      * start sector */
1221     if( DVDFindSector( p_dvd ) < 0 )
1222     {
1223         intf_ErrMsg( "dvd error: can't select chapter" );
1224         return -1;
1225     }
1226
1227     /* start is : beginning of vts vobs + offset to vob x */
1228     p_dvd->i_start = p_dvd->i_title_start + p_dvd->i_sector;
1229
1230     /* Position the fd pointer on the right address */
1231     if( ( p_dvd->i_start = dvdcss_seek( p_dvd->dvdhandle,
1232                                         p_dvd->i_start,
1233                                         DVDCSS_SEEK_MPEG ) ) < 0 )
1234     {
1235         intf_ErrMsg( "dvd error: %s", dvdcss_error( p_dvd->dvdhandle ) );
1236         return -1;
1237     }
1238
1239     p_dvd->i_chapter = i_chapter;
1240     return 0;
1241 }
1242
1243 /*****************************************************************************
1244  * DVDChooseAngle: select the cell corresponding to the selected angle
1245  *****************************************************************************/
1246 static int DVDChooseAngle( thread_dvd_data_t * p_dvd )
1247 {
1248     /* basic handling of angles */
1249     switch( ( ( title.p_cell_play[p_dvd->i_prg_cell].i_category & 0xf000 )
1250                     >> 12 ) )
1251     {
1252         /* we enter a muli-angle section */
1253         case 0x5:
1254             p_dvd->i_prg_cell += p_dvd->i_angle - 1;
1255             p_dvd->i_angle_cell = 0;
1256             break;
1257         /* we exit a multi-angle section */
1258         case 0x9:
1259         case 0xd:
1260             p_dvd->i_prg_cell += p_dvd->i_angle_nb - p_dvd->i_angle;
1261             break;
1262     }
1263
1264     return 0;
1265 }
1266
1267 #undef title