libdvdread4/dvd_reader.c

   1 /*
   2  * Copyright (C) 2001-2004 Billy Biggs <vektor@dumbterm.net>,
   3  *                         Håkan Hjort <d95hjort@dtek.chalmers.se>,
   4  *                         Björn Englund <d4bjorn@dtek.chalmers.se>
   5  *
   6  * This file is part of libdvdread.
   7  *
   8  * libdvdread is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * libdvdread is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License along
  19  * with libdvdread; if not, write to the Free Software Foundation, Inc.,
  20  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  21  */
  22
  23 #include <sys/types.h>
  24 #include <sys/stat.h>
  25 #include <sys/time.h> /* For the timing of dvdcss_title crack. */
  26 #include <fcntl.h>
  27 #include <stdlib.h>
  28 #include <stdio.h>
  29 #include <errno.h>
  30 #include <string.h>
  31 #include <unistd.h>
  32 #include <limits.h>
  33 #include <dirent.h>
  34
  35 /* misc win32 helpers */
  36 #ifdef WIN32
  37 #ifndef HAVE_GETTIMEOFDAY
  38 /* replacement gettimeofday implementation */
  39 #include <sys/timeb.h>
  40 static inline int _private_gettimeofday( struct timeval *tv, void *tz )
  41 {
  42   struct timeb t;
  43   ftime( &t );
  44   tv->tv_sec = t.time;
  45   tv->tv_usec = t.millitm * 1000;
  46   return 0;
  47 }
  48 #define gettimeofday(TV, TZ) _private_gettimeofday((TV), (TZ))
  49 #endif
  50 #include <io.h> /* read() */
  51 #define lseek64 _lseeki64
  52 #endif
  53
  54 #if defined(__FreeBSD__) || defined(__OpenBSD__) || defined(__NetBSD__) || defined(__bsdi__) || defined(__APPLE__)
  55 #define SYS_BSD 1
  56 #endif
  57
  58 #if defined(__sun)
  59 #include <sys/mnttab.h>
  60 #elif defined(SYS_BSD)
  61 #include <fstab.h>
  62 #elif defined(__linux__)
  63 #include <mntent.h>
  64 #endif
  65
  66 #include "dvdread/dvd_udf.h"
  67 #include "dvd_input.h"
  68 #include "dvdread/dvd_reader.h"
  69 #include "md5.h"
  70
  71 #define DEFAULT_UDF_CACHE_LEVEL 1
  72
  73 struct dvd_reader_s {
  74   /* Basic information. */
  75   int isImageFile;
  76
  77   /* Hack for keeping track of the css status.
  78    * 0: no css, 1: perhaps (need init of keys), 2: have done init */
  79   int css_state;
  80   int css_title; /* Last title that we have called dvdinpute_title for. */
  81
  82   /* Information required for an image file. */
  83   dvd_input_t dev;
  84
  85   /* Information required for a directory path drive. */
  86   char *path_root;
  87
  88   /* Filesystem cache */
  89   int udfcache_level; /* 0 - turned off, 1 - on */
  90   void *udfcache;
  91 };
  92
  93 #define TITLES_MAX 9
  94
  95 struct dvd_file_s {
  96   /* Basic information. */
  97   dvd_reader_t *dvd;
  98
  99   /* Hack for selecting the right css title. */
 100   int css_title;
 101
 102   /* Information required for an image file. */
 103   uint32_t lb_start;
 104   uint32_t seek_pos;
 105
 106   /* Information required for a directory path drive. */
 107   size_t title_sizes[ TITLES_MAX ];
 108   dvd_input_t title_devs[ TITLES_MAX ];
 109
 110   /* Calculated at open-time, size in blocks. */
 111   ssize_t filesize;
 112 };
 113
 114 int UDFReadBlocksRaw( dvd_reader_t *device, uint32_t lb_number,
 115                       size_t block_count, unsigned char *data,
 116                       int encrypted );
 117
 118 /**
 119  * Set the level of caching on udf
 120  * level = 0 (no caching)
 121  * level = 1 (caching filesystem info)
 122  */
 123 int DVDUDFCacheLevel(dvd_reader_t *device, int level)
 124 {
 125   struct dvd_reader_s *dev = (struct dvd_reader_s *)device;
 126
 127   if(level > 0) {
 128     level = 1;
 129   } else if(level < 0) {
 130     return dev->udfcache_level;
 131   }
 132
 133   dev->udfcache_level = level;
 134
 135   return level;
 136 }
 137
 138 void *GetUDFCacheHandle(dvd_reader_t *device)
 139 {
 140   struct dvd_reader_s *dev = (struct dvd_reader_s *)device;
 141
 142   return dev->udfcache;
 143 }
 144
 145 void SetUDFCacheHandle(dvd_reader_t *device, void *cache)
 146 {
 147   struct dvd_reader_s *dev = (struct dvd_reader_s *)device;
 148
 149   dev->udfcache = cache;
 150 }
 151
 152
 153
 154 /* Loop over all titles and call dvdcss_title to crack the keys. */
 155 static int initAllCSSKeys( dvd_reader_t *dvd )
 156 {
 157   struct timeval all_s, all_e;
 158   struct timeval t_s, t_e;
 159   char filename[ MAX_UDF_FILE_NAME_LEN ];
 160   uint32_t start, len;
 161   int title;
 162
 163   char *nokeys_str = getenv("DVDREAD_NOKEYS");
 164   if(nokeys_str != NULL)
 165     return 0;
 166
 167   fprintf( stderr, "\n" );
 168   fprintf( stderr, "libdvdread: Attempting to retrieve all CSS keys\n" );
 169   fprintf( stderr, "libdvdread: This can take a _long_ time, "
 170            "please be patient\n\n" );
 171   gettimeofday(&all_s, NULL);
 172
 173   for( title = 0; title < 100; title++ ) {
 174     gettimeofday( &t_s, NULL );
 175     if( title == 0 ) {
 176       sprintf( filename, "/VIDEO_TS/VIDEO_TS.VOB" );
 177     } else {
 178       sprintf( filename, "/VIDEO_TS/VTS_%02d_%d.VOB", title, 0 );
 179     }
 180     start = UDFFindFile( dvd, filename, &len );
 181     if( start != 0 && len != 0 ) {
 182       /* Perform CSS key cracking for this title. */
 183       fprintf( stderr, "libdvdread: Get key for %s at 0x%08x\n",
 184                filename, start );
 185       if( dvdinput_title( dvd->dev, (int)start ) < 0 ) {
 186         fprintf( stderr, "libdvdread: Error cracking CSS key for %s (0x%08x)\n", filename, start);
 187       }
 188       gettimeofday( &t_e, NULL );
 189       fprintf( stderr, "libdvdread: Elapsed time %ld\n",
 190                (long int) t_e.tv_sec - t_s.tv_sec );
 191     }
 192
 193     if( title == 0 ) continue;
 194
 195     gettimeofday( &t_s, NULL );
 196     sprintf( filename, "/VIDEO_TS/VTS_%02d_%d.VOB", title, 1 );
 197     start = UDFFindFile( dvd, filename, &len );
 198     if( start == 0 || len == 0 ) break;
 199
 200     /* Perform CSS key cracking for this title. */
 201     fprintf( stderr, "libdvdread: Get key for %s at 0x%08x\n",
 202              filename, start );
 203     if( dvdinput_title( dvd->dev, (int)start ) < 0 ) {
 204       fprintf( stderr, "libdvdread: Error cracking CSS key for %s (0x%08x)!!\n", filename, start);
 205     }
 206     gettimeofday( &t_e, NULL );
 207     fprintf( stderr, "libdvdread: Elapsed time %ld\n",
 208              (long int) t_e.tv_sec - t_s.tv_sec );
 209   }
 210   title--;
 211
 212   fprintf( stderr, "libdvdread: Found %d VTS's\n", title );
 213   gettimeofday(&all_e, NULL);
 214   fprintf( stderr, "libdvdread: Elapsed time %ld\n",
 215            (long int) all_e.tv_sec - all_s.tv_sec );
 216
 217   return 0;
 218 }
 219
 220
 221
 222 /**
 223  * Open a DVD image or block device file.
 224  */
 225 static dvd_reader_t *DVDOpenImageFile( const char *location, int have_css )
 226 {
 227   dvd_reader_t *dvd;
 228   dvd_input_t dev;
 229
 230   dev = dvdinput_open( location );
 231   if( !dev ) {
 232     fprintf( stderr, "libdvdread: Can't open %s for reading\n", location );
 233     return NULL;
 234   }
 235
 236   dvd = (dvd_reader_t *) malloc( sizeof( dvd_reader_t ) );
 237   if( !dvd ) {
 238     dvdinput_close(dev);
 239     return NULL;
 240   }
 241   dvd->isImageFile = 1;
 242   dvd->dev = dev;
 243   dvd->path_root = NULL;
 244
 245   dvd->udfcache_level = DEFAULT_UDF_CACHE_LEVEL;
 246   dvd->udfcache = NULL;
 247
 248   if( have_css ) {
 249     /* Only if DVDCSS_METHOD = title, a bit if it's disc or if
 250      * DVDCSS_METHOD = key but region mismatch. Unfortunately we
 251      * don't have that information. */
 252
 253     dvd->css_state = 1; /* Need key init. */
 254   }
 255   dvd->css_title = 0;
 256
 257   return dvd;
 258 }
 259
 260 static dvd_reader_t *DVDOpenPath( const char *path_root )
 261 {
 262   dvd_reader_t *dvd;
 263
 264   dvd = (dvd_reader_t *) malloc( sizeof( dvd_reader_t ) );
 265   if( !dvd ) return NULL;
 266   dvd->isImageFile = 0;
 267   dvd->dev = 0;
 268   dvd->path_root = strdup( path_root );
 269   if(!dvd->path_root) {
 270     free(dvd);
 271     return 0;
 272   }
 273   dvd->udfcache_level = DEFAULT_UDF_CACHE_LEVEL;
 274   dvd->udfcache = NULL;
 275
 276   dvd->css_state = 0; /* Only used in the UDF path */
 277   dvd->css_title = 0; /* Only matters in the UDF path */
 278
 279   return dvd;
 280 }
 281
 282 #if defined(__sun)
 283 /* /dev/rdsk/c0t6d0s0 (link to /devices/...)
 284    /vol/dev/rdsk/c0t6d0/??
 285    /vol/rdsk/<name> */
 286 static char *sun_block2char( const char *path )
 287 {
 288   char *new_path;
 289
 290   /* Must contain "/dsk/" */
 291   if( !strstr( path, "/dsk/" ) ) return (char *) strdup( path );
 292
 293   /* Replace "/dsk/" with "/rdsk/" */
 294   new_path = malloc( strlen(path) + 2 );
 295   strcpy( new_path, path );
 296   strcpy( strstr( new_path, "/dsk/" ), "" );
 297   strcat( new_path, "/rdsk/" );
 298   strcat( new_path, strstr( path, "/dsk/" ) + strlen( "/dsk/" ) );
 299
 300   return new_path;
 301 }
 302 #endif
 303
 304 #if defined(SYS_BSD)
 305 /* FreeBSD /dev/(r)(a)cd0c (a is for atapi), recommended to _not_ use r
 306    OpenBSD /dev/rcd0c, it needs to be the raw device
 307    NetBSD  /dev/rcd0[d|c|..] d for x86, c (for non x86), perhaps others
 308    Darwin  /dev/rdisk0,  it needs to be the raw device
 309    BSD/OS  /dev/sr0c (if not mounted) or /dev/rsr0c ('c' any letter will do) */
 310 static char *bsd_block2char( const char *path )
 311 {
 312   char *new_path;
 313
 314   /* If it doesn't start with "/dev/" or does start with "/dev/r" exit */
 315   if( !strncmp( path, "/dev/",  5 ) || strncmp( path, "/dev/r", 6 ) )
 316     return (char *) strdup( path );
 317
 318   /* Replace "/dev/" with "/dev/r" */
 319   new_path = malloc( strlen(path) + 2 );
 320   strcpy( new_path, "/dev/r" );
 321   strcat( new_path, path + strlen( "/dev/" ) );
 322
 323   return new_path;
 324 }
 325 #endif
 326
 327
 328 dvd_reader_t *DVDOpen( const char *ppath )
 329 {
 330   struct stat fileinfo;
 331   int ret, have_css, retval, cdir = 0;
 332   dvd_reader_t *ret_val = NULL;
 333   char *dev_name = NULL;
 334   char *path = NULL, *new_path = NULL, *path_copy = NULL;
 335
 336 #ifdef _MSC_VER
 337       int len;
 338 #endif
 339
 340   if( ppath == NULL )
 341     goto DVDOpen_error;
 342
 343       path = strdup(ppath);
 344   if( path == NULL )
 345     goto DVDOpen_error;
 346
 347   /* Try to open libdvdcss or fall back to standard functions */
 348   have_css = dvdinput_setup();
 349
 350 #ifdef _MSC_VER
 351   /* Strip off the trailing \ if it is not a drive */
 352   len = strlen(path);
 353   if ((len > 1) &&
 354       (path[len - 1] == '\\')  &&
 355       (path[len - 2] != ':'))
 356   {
 357     path[len-1] = '\0';
 358   }
 359 #endif
 360
 361   ret = stat( path, &fileinfo );
 362
 363   if( ret < 0 ) {
 364
 365     /* maybe "host:port" url? try opening it with acCeSS library */
 366     if( strchr(path,':') ) {
 367                     ret_val = DVDOpenImageFile( path, have_css );
 368                     free(path);
 369             return ret_val;
 370     }
 371
 372     /* If we can't stat the file, give up */
 373     fprintf( stderr, "libdvdread: Can't stat %s\n", path );
 374     perror("");
 375     goto DVDOpen_error;
 376   }
 377
 378   /* First check if this is a block/char device or a file*/
 379   if( S_ISBLK( fileinfo.st_mode ) ||
 380       S_ISCHR( fileinfo.st_mode ) ||
 381       S_ISREG( fileinfo.st_mode ) ) {
 382
 383     /**
 384      * Block devices and regular files are assumed to be DVD-Video images.
 385      */
 386 #if defined(__sun)
 387     ret_val = DVDOpenImageFile( sun_block2char( path ), have_css );
 388 #elif defined(SYS_BSD)
 389     ret_val = DVDOpenImageFile( bsd_block2char( path ), have_css );
 390 #else
 391     ret_val = DVDOpenImageFile( path, have_css );
 392 #endif
 393
 394     free(path);
 395     return ret_val;
 396
 397   } else if( S_ISDIR( fileinfo.st_mode ) ) {
 398     dvd_reader_t *auth_drive = 0;
 399 #if defined(SYS_BSD)
 400     struct fstab* fe;
 401 #elif defined(__sun) || defined(__linux__)
 402     FILE *mntfile;
 403 #endif
 404
 405     /* XXX: We should scream real loud here. */
 406     if( !(path_copy = strdup( path ) ) )
 407       goto DVDOpen_error;
 408
 409 #ifndef WIN32 /* don't have fchdir, and getcwd( NULL, ... ) is strange */
 410               /* Also WIN32 does not have symlinks, so we don't need this bit of code. */
 411
 412     /* Resolve any symlinks and get the absolute dir name. */
 413     {
 414       if( ( cdir  = open( ".", O_RDONLY ) ) >= 0 ) {
 415         if( chdir( path_copy ) == -1 ) {
 416           goto DVDOpen_error;
 417         }
 418         new_path = malloc(PATH_MAX+1);
 419         if(!new_path) {
 420           goto DVDOpen_error;
 421         }
 422         if( getcwd( new_path, PATH_MAX ) == NULL ) {
 423           goto DVDOpen_error;
 424         }
 425         retval = fchdir( cdir );
 426         close( cdir );
 427         cdir = -1;
 428         if( retval == -1 ) {
 429           goto DVDOpen_error;
 430         }
 431         path_copy = new_path;
 432         new_path = NULL;
 433       }
 434     }
 435 #endif
 436
 437     /**
 438      * If we're being asked to open a directory, check if that directory
 439      * is the mount point for a DVD-ROM which we can use instead.
 440      */
 441
 442     if( strlen( path_copy ) > 1 ) {
 443       if( path_copy[ strlen( path_copy ) - 1 ] == '/' ) {
 444         path_copy[ strlen( path_copy ) - 1 ] = '\0';
 445       }
 446     }
 447
 448     if( strlen( path_copy ) > TITLES_MAX ) {
 449       if( !strcasecmp( &(path_copy[ strlen( path_copy ) - TITLES_MAX ]),
 450                        "/video_ts" ) ) {
 451         path_copy[ strlen( path_copy ) - TITLES_MAX ] = '\0';
 452       }
 453     }
 454
 455     if(path_copy[0] == '\0') {
 456       path_copy[0] = '/';
 457       path_copy[1] = '\0';
 458     }
 459
 460 #if defined(SYS_BSD)
 461     if( ( fe = getfsfile( path_copy ) ) ) {
 462       dev_name = bsd_block2char( fe->fs_spec );
 463       fprintf( stderr,
 464                "libdvdread: Attempting to use device %s"
 465                " mounted on %s for CSS authentication\n",
 466                dev_name,
 467                fe->fs_file );
 468       auth_drive = DVDOpenImageFile( dev_name, have_css );
 469     }
 470 #elif defined(__sun)
 471     mntfile = fopen( MNTTAB, "r" );
 472     if( mntfile ) {
 473       struct mnttab mp;
 474       int res;
 475
 476       while( ( res = getmntent( mntfile, &mp ) ) != -1 ) {
 477         if( res == 0 && !strcmp( mp.mnt_mountp, path_copy ) ) {
 478           dev_name = sun_block2char( mp.mnt_special );
 479           fprintf( stderr,
 480                    "libdvdread: Attempting to use device %s"
 481                    " mounted on %s for CSS authentication\n",
 482                    dev_name,
 483                    mp.mnt_mountp );
 484           auth_drive = DVDOpenImageFile( dev_name, have_css );
 485           break;
 486         }
 487       }
 488       fclose( mntfile );
 489     }
 490 #elif defined(__linux__)
 491     mntfile = fopen( MOUNTED, "r" );
 492     if( mntfile ) {
 493       struct mntent *me;
 494
 495       while( ( me = getmntent( mntfile ) ) ) {
 496         if( !strcmp( me->mnt_dir, path_copy ) ) {
 497           fprintf( stderr,
 498                    "libdvdread: Attempting to use device %s"
 499                    " mounted on %s for CSS authentication\n",
 500                    me->mnt_fsname,
 501                    me->mnt_dir );
 502           auth_drive = DVDOpenImageFile( me->mnt_fsname, have_css );
 503           dev_name = strdup(me->mnt_fsname);
 504           break;
 505         }
 506       }
 507       fclose( mntfile );
 508     }
 509 #elif defined(_MSC_VER) || defined(__OS2__)
 510     auth_drive = DVDOpenImageFile( path, have_css );
 511 #endif
 512
 513 #if !defined(_MSC_VER) && !defined(__OS2__)
 514     if( !dev_name ) {
 515       fprintf( stderr, "libdvdread: Couldn't find device name.\n" );
 516     } else if( !auth_drive ) {
 517       fprintf( stderr, "libdvdread: Device %s inaccessible, "
 518                "CSS authentication not available.\n", dev_name );
 519     }
 520 #else
 521     if( !auth_drive ) {
 522         fprintf( stderr, "libdvdread: Device %s inaccessible, "
 523                  "CSS authentication not available.\n", dev_name );
 524     }
 525 #endif
 526
 527     free( dev_name );
 528     dev_name = NULL;
 529     free( path_copy );
 530     path_copy = NULL;
 531
 532     /**
 533      * If we've opened a drive, just use that.
 534      */
 535     if( auth_drive ) {
 536       free(path);
 537       return auth_drive;
 538     }
 539     /**
 540      * Otherwise, we now try to open the directory tree instead.
 541      */
 542     ret_val = DVDOpenPath( path );
 543       free( path );
 544       return ret_val;
 545   }
 546
 547 DVDOpen_error:
 548   /* If it's none of the above, screw it. */
 549   fprintf( stderr, "libdvdread: Could not open %s\n", path );
 550   if( path != NULL )
 551     free( path );
 552   if ( path_copy != NULL )
 553     free( path_copy );
 554   if ( cdir >= 0 )
 555     close( cdir );
 556   if ( new_path != NULL )
 557     free( new_path );
 558   return NULL;
 559 }
 560
 561 void DVDClose( dvd_reader_t *dvd )
 562 {
 563   if( dvd ) {
 564     if( dvd->dev ) dvdinput_close( dvd->dev );
 565     if( dvd->path_root ) free( dvd->path_root );
 566     if( dvd->udfcache ) FreeUDFCache( dvd->udfcache );
 567     free( dvd );
 568   }
 569 }
 570
 571 /**
 572  * Open an unencrypted file on a DVD image file.
 573  */
 574 static dvd_file_t *DVDOpenFileUDF( dvd_reader_t *dvd, char *filename )
 575 {
 576   uint32_t start, len;
 577   dvd_file_t *dvd_file;
 578
 579   start = UDFFindFile( dvd, filename, &len );
 580   if( !start ) {
 581     fprintf( stderr, "libdvdnav:DVDOpenFileUDF:UDFFindFile %s failed\n", filename );
 582     return NULL;
 583   }
 584
 585   dvd_file = (dvd_file_t *) malloc( sizeof( dvd_file_t ) );
 586   if( !dvd_file ) {
 587     fprintf( stderr, "libdvdnav:DVDOpenFileUDF:malloc failed\n" );
 588     return NULL;
 589   }
 590   dvd_file->dvd = dvd;
 591   dvd_file->lb_start = start;
 592   dvd_file->seek_pos = 0;
 593   memset( dvd_file->title_sizes, 0, sizeof( dvd_file->title_sizes ) );
 594   memset( dvd_file->title_devs, 0, sizeof( dvd_file->title_devs ) );
 595   dvd_file->filesize = len / DVD_VIDEO_LB_LEN;
 596
 597   return dvd_file;
 598 }
 599
 600 /**
 601  * Searches for <file> in directory <path>, ignoring case.
 602  * Returns 0 and full filename in <filename>.
 603  *     or -1 on file not found.
 604  *     or -2 on path not found.
 605  */
 606 static int findDirFile( const char *path, const char *file, char *filename )
 607 {
 608   DIR *dir;
 609   struct dirent *ent;
 610
 611   dir = opendir( path );
 612   if( !dir ) return -2;
 613
 614   while( ( ent = readdir( dir ) ) != NULL ) {
 615     if( !strcasecmp( ent->d_name, file ) ) {
 616       sprintf( filename, "%s%s%s", path,
 617                ( ( path[ strlen( path ) - 1 ] == '/' ) ? "" : "/" ),
 618                ent->d_name );
 619       closedir(dir);
 620       return 0;
 621     }
 622   }
 623   closedir(dir);
 624   return -1;
 625 }
 626
 627 static int findDVDFile( dvd_reader_t *dvd, const char *file, char *filename )
 628 {
 629   char video_path[ PATH_MAX + 1 ];
 630   const char *nodirfile;
 631   int ret;
 632
 633   /* Strip off the directory for our search */
 634   if( !strncasecmp( "/VIDEO_TS/", file, 10 ) ) {
 635     nodirfile = &(file[ 10 ]);
 636   } else {
 637     nodirfile = file;
 638   }
 639
 640   ret = findDirFile( dvd->path_root, nodirfile, filename );
 641   if( ret < 0 ) {
 642     /* Try also with adding the path, just in case. */
 643     sprintf( video_path, "%s/VIDEO_TS/", dvd->path_root );
 644     ret = findDirFile( video_path, nodirfile, filename );
 645     if( ret < 0 ) {
 646       /* Try with the path, but in lower case. */
 647       sprintf( video_path, "%s/video_ts/", dvd->path_root );
 648       ret = findDirFile( video_path, nodirfile, filename );
 649       if( ret < 0 ) {
 650         return 0;
 651       }
 652     }
 653   }
 654
 655   return 1;
 656 }
 657
 658 /**
 659  * Open an unencrypted file from a DVD directory tree.
 660  */
 661 static dvd_file_t *DVDOpenFilePath( dvd_reader_t *dvd, char *filename )
 662 {
 663   char full_path[ PATH_MAX + 1 ];
 664   dvd_file_t *dvd_file;
 665   struct stat fileinfo;
 666   dvd_input_t dev;
 667
 668   /* Get the full path of the file. */
 669   if( !findDVDFile( dvd, filename, full_path ) ) {
 670     fprintf( stderr, "libdvdnav:DVDOpenFilePath:findDVDFile %s failed\n", filename );
 671     return NULL;
 672   }
 673
 674   dev = dvdinput_open( full_path );
 675   if( !dev ) {
 676     fprintf( stderr, "libdvdnav:DVDOpenFilePath:dvdinput_open %s failed\n", full_path );
 677     return NULL;
 678   }
 679
 680   dvd_file = (dvd_file_t *) malloc( sizeof( dvd_file_t ) );
 681   if( !dvd_file ) {
 682     fprintf( stderr, "libdvdnav:DVDOpenFilePath:dvd_file malloc failed\n" );
 683     dvdinput_close(dev);
 684     return NULL;
 685   }
 686   dvd_file->dvd = dvd;
 687   dvd_file->lb_start = 0;
 688   dvd_file->seek_pos = 0;
 689   memset( dvd_file->title_sizes, 0, sizeof( dvd_file->title_sizes ) );
 690   memset( dvd_file->title_devs, 0, sizeof( dvd_file->title_devs ) );
 691   dvd_file->filesize = 0;
 692
 693   if( stat( full_path, &fileinfo ) < 0 ) {
 694     fprintf( stderr, "libdvdread: Can't stat() %s.\n", filename );
 695     free( dvd_file );
 696     return NULL;
 697   }
 698   dvd_file->title_sizes[ 0 ] = fileinfo.st_size / DVD_VIDEO_LB_LEN;
 699   dvd_file->title_devs[ 0 ] = dev;
 700   dvd_file->filesize = dvd_file->title_sizes[ 0 ];
 701
 702   return dvd_file;
 703 }
 704
 705 static dvd_file_t *DVDOpenVOBUDF( dvd_reader_t *dvd, int title, int menu )
 706 {
 707   char filename[ MAX_UDF_FILE_NAME_LEN ];
 708   uint32_t start, len;
 709   dvd_file_t *dvd_file;
 710
 711   if( title == 0 ) {
 712     sprintf( filename, "/VIDEO_TS/VIDEO_TS.VOB" );
 713   } else {
 714     sprintf( filename, "/VIDEO_TS/VTS_%02d_%d.VOB", title, menu ? 0 : 1 );
 715   }
 716   start = UDFFindFile( dvd, filename, &len );
 717   if( start == 0 ) return NULL;
 718
 719   dvd_file = (dvd_file_t *) malloc( sizeof( dvd_file_t ) );
 720   if( !dvd_file ) return NULL;
 721   dvd_file->dvd = dvd;
 722   /*Hack*/ dvd_file->css_title = title << 1 | menu;
 723   dvd_file->lb_start = start;
 724   dvd_file->seek_pos = 0;
 725   memset( dvd_file->title_sizes, 0, sizeof( dvd_file->title_sizes ) );
 726   memset( dvd_file->title_devs, 0, sizeof( dvd_file->title_devs ) );
 727   dvd_file->filesize = len / DVD_VIDEO_LB_LEN;
 728
 729   /* Calculate the complete file size for every file in the VOBS */
 730   if( !menu ) {
 731     int cur;
 732
 733     for( cur = 2; cur < 10; cur++ ) {
 734       sprintf( filename, "/VIDEO_TS/VTS_%02d_%d.VOB", title, cur );
 735       if( !UDFFindFile( dvd, filename, &len ) ) break;
 736       dvd_file->filesize += len / DVD_VIDEO_LB_LEN;
 737     }
 738   }
 739
 740   if( dvd->css_state == 1 /* Need key init */ ) {
 741     initAllCSSKeys( dvd );
 742     dvd->css_state = 2;
 743   }
 744   /*
 745   if( dvdinput_title( dvd_file->dvd->dev, (int)start ) < 0 ) {
 746       fprintf( stderr, "libdvdread: Error cracking CSS key for %s\n",
 747                filename );
 748   }
 749   */
 750
 751   return dvd_file;
 752 }
 753
 754 static dvd_file_t *DVDOpenVOBPath( dvd_reader_t *dvd, int title, int menu )
 755 {
 756   char filename[ MAX_UDF_FILE_NAME_LEN ];
 757   char full_path[ PATH_MAX + 1 ];
 758   struct stat fileinfo;
 759   dvd_file_t *dvd_file;
 760   int i;
 761
 762   dvd_file = (dvd_file_t *) malloc( sizeof( dvd_file_t ) );
 763   if( !dvd_file ) return NULL;
 764   dvd_file->dvd = dvd;
 765   /*Hack*/ dvd_file->css_title = title << 1 | menu;
 766   dvd_file->lb_start = 0;
 767   dvd_file->seek_pos = 0;
 768   memset( dvd_file->title_sizes, 0, sizeof( dvd_file->title_sizes ) );
 769   memset( dvd_file->title_devs, 0, sizeof( dvd_file->title_devs ) );
 770   dvd_file->filesize = 0;
 771
 772   if( menu ) {
 773     dvd_input_t dev;
 774
 775     if( title == 0 ) {
 776       sprintf( filename, "VIDEO_TS.VOB" );
 777     } else {
 778       sprintf( filename, "VTS_%02i_0.VOB", title );
 779     }
 780     if( !findDVDFile( dvd, filename, full_path ) ) {
 781       free( dvd_file );
 782       return NULL;
 783     }
 784
 785     dev = dvdinput_open( full_path );
 786     if( dev == NULL ) {
 787       free( dvd_file );
 788       return NULL;
 789     }
 790
 791     if( stat( full_path, &fileinfo ) < 0 ) {
 792       fprintf( stderr, "libdvdread: Can't stat() %s.\n", filename );
 793       dvdinput_close(dev);
 794       free( dvd_file );
 795       return NULL;
 796     }
 797     dvd_file->title_sizes[ 0 ] = fileinfo.st_size / DVD_VIDEO_LB_LEN;
 798     dvd_file->title_devs[ 0 ] = dev;
 799     dvdinput_title( dvd_file->title_devs[0], 0);
 800     dvd_file->filesize = dvd_file->title_sizes[ 0 ];
 801
 802   } else {
 803     for( i = 0; i < TITLES_MAX; ++i ) {
 804
 805       sprintf( filename, "VTS_%02i_%i.VOB", title, i + 1 );
 806       if( !findDVDFile( dvd, filename, full_path ) ) {
 807         break;
 808       }
 809
 810       if( stat( full_path, &fileinfo ) < 0 ) {
 811         fprintf( stderr, "libdvdread: Can't stat() %s.\n", filename );
 812         break;
 813       }
 814
 815       dvd_file->title_sizes[ i ] = fileinfo.st_size / DVD_VIDEO_LB_LEN;
 816       dvd_file->title_devs[ i ] = dvdinput_open( full_path );
 817       dvdinput_title( dvd_file->title_devs[ i ], 0 );
 818       dvd_file->filesize += dvd_file->title_sizes[ i ];
 819     }
 820     if( !dvd_file->title_devs[ 0 ] ) {
 821       free( dvd_file );
 822       return NULL;
 823     }
 824   }
 825
 826   return dvd_file;
 827 }
 828
 829 dvd_file_t *DVDOpenFile( dvd_reader_t *dvd, int titlenum,
 830                          dvd_read_domain_t domain )
 831 {
 832   char filename[ MAX_UDF_FILE_NAME_LEN ];
 833
 834   /* Check arguments. */
 835   if( dvd == NULL || titlenum < 0 )
 836     return NULL;
 837
 838   switch( domain ) {
 839   case DVD_READ_INFO_FILE:
 840     if( titlenum == 0 ) {
 841       sprintf( filename, "/VIDEO_TS/VIDEO_TS.IFO" );
 842     } else {
 843       sprintf( filename, "/VIDEO_TS/VTS_%02i_0.IFO", titlenum );
 844     }
 845     break;
 846   case DVD_READ_INFO_BACKUP_FILE:
 847     if( titlenum == 0 ) {
 848       sprintf( filename, "/VIDEO_TS/VIDEO_TS.BUP" );
 849     } else {
 850       sprintf( filename, "/VIDEO_TS/VTS_%02i_0.BUP", titlenum );
 851     }
 852     break;
 853   case DVD_READ_MENU_VOBS:
 854     if( dvd->isImageFile ) {
 855       return DVDOpenVOBUDF( dvd, titlenum, 1 );
 856     } else {
 857       return DVDOpenVOBPath( dvd, titlenum, 1 );
 858     }
 859     break;
 860   case DVD_READ_TITLE_VOBS:
 861     if( titlenum == 0 ) return 0;
 862     if( dvd->isImageFile ) {
 863       return DVDOpenVOBUDF( dvd, titlenum, 0 );
 864     } else {
 865       return DVDOpenVOBPath( dvd, titlenum, 0 );
 866     }
 867     break;
 868   default:
 869     fprintf( stderr, "libdvdread: Invalid domain for file open.\n" );
 870     return NULL;
 871   }
 872
 873   if( dvd->isImageFile ) {
 874     return DVDOpenFileUDF( dvd, filename );
 875   } else {
 876     return DVDOpenFilePath( dvd, filename );
 877   }
 878 }
 879
 880 void DVDCloseFile( dvd_file_t *dvd_file )
 881 {
 882   int i;
 883
 884   if( dvd_file ) {
 885     if( dvd_file->dvd->isImageFile ) {
 886       ;
 887     } else {
 888       for( i = 0; i < TITLES_MAX; ++i ) {
 889         if( dvd_file->title_devs[ i ] ) {
 890           dvdinput_close( dvd_file->title_devs[i] );
 891         }
 892       }
 893     }
 894
 895     free( dvd_file );
 896     dvd_file = 0;
 897   }
 898 }
 899
 900 /* Internal, but used from dvd_udf.c */
 901 int UDFReadBlocksRaw( dvd_reader_t *device, uint32_t lb_number,
 902                       size_t block_count, unsigned char *data,
 903                       int encrypted )
 904 {
 905   int ret;
 906
 907   if( !device->dev ) {
 908     fprintf( stderr, "libdvdread: Fatal error in block read.\n" );
 909     return 0;
 910   }
 911
 912   ret = dvdinput_seek( device->dev, (int) lb_number );
 913   if( ret != (int) lb_number ) {
 914     fprintf( stderr, "libdvdread: Can't seek to block %u\n", lb_number );
 915     return 0;
 916   }
 917
 918   ret = dvdinput_read( device->dev, (char *) data,
 919                        (int) block_count, encrypted );
 920   return ret;
 921 }
 922
 923 /* This is using a single input and starting from 'dvd_file->lb_start' offset.
 924  *
 925  * Reads 'block_count' blocks from 'dvd_file' at block offset 'offset'
 926  * into the buffer located at 'data' and if 'encrypted' is set
 927  * descramble the data if it's encrypted.  Returning either an
 928  * negative error or the number of blocks read. */
 929 static int DVDReadBlocksUDF( dvd_file_t *dvd_file, uint32_t offset,
 930                              size_t block_count, unsigned char *data,
 931                              int encrypted )
 932 {
 933   return UDFReadBlocksRaw( dvd_file->dvd, dvd_file->lb_start + offset,
 934                            block_count, data, encrypted );
 935 }
 936
 937 /* This is using possibly several inputs and starting from an offset of '0'.
 938  *
 939  * Reads 'block_count' blocks from 'dvd_file' at block offset 'offset'
 940  * into the buffer located at 'data' and if 'encrypted' is set
 941  * descramble the data if it's encrypted.  Returning either an
 942  * negative error or the number of blocks read. */
 943 static int DVDReadBlocksPath( dvd_file_t *dvd_file, unsigned int offset,
 944                               size_t block_count, unsigned char *data,
 945                               int encrypted )
 946 {
 947   int i;
 948   int ret, ret2, off;
 949
 950   ret = 0;
 951   ret2 = 0;
 952   for( i = 0; i < TITLES_MAX; ++i ) {
 953     if( !dvd_file->title_sizes[ i ] ) return 0; /* Past end of file */
 954
 955     if( offset < dvd_file->title_sizes[ i ] ) {
 956       if( ( offset + block_count ) <= dvd_file->title_sizes[ i ] ) {
 957         off = dvdinput_seek( dvd_file->title_devs[ i ], (int)offset );
 958         if( off < 0 || off != (int)offset ) {
 959           fprintf( stderr, "libdvdread: Can't seek to block %d\n",
 960                    offset );
 961           return off < 0 ? off : 0;
 962         }
 963         ret = dvdinput_read( dvd_file->title_devs[ i ], data,
 964                              (int)block_count, encrypted );
 965         break;
 966       } else {
 967         size_t part1_size = dvd_file->title_sizes[ i ] - offset;
 968         /* FIXME: Really needs to be a while loop.
 969          * (This is only true if you try and read >1GB at a time) */
 970
 971         /* Read part 1 */
 972         off = dvdinput_seek( dvd_file->title_devs[ i ], (int)offset );
 973         if( off < 0 || off != (int)offset ) {
 974           fprintf( stderr, "libdvdread: Can't seek to block %d\n",
 975                    offset );
 976           return off < 0 ? off : 0;
 977         }
 978         ret = dvdinput_read( dvd_file->title_devs[ i ], data,
 979                              (int)part1_size, encrypted );
 980         if( ret < 0 ) return ret;
 981         /* FIXME: This is wrong if i is the last file in the set.
 982          * also error from this read will not show in ret. */
 983
 984         /* Does the next part exist? If not then return now. */
 985         if( i + 1 >= TITLES_MAX || !dvd_file->title_devs[ i + 1 ] )
 986           return ret;
 987
 988         /* Read part 2 */
 989         off = dvdinput_seek( dvd_file->title_devs[ i + 1 ], 0 );
 990         if( off < 0 || off != 0 ) {
 991           fprintf( stderr, "libdvdread: Can't seek to block %d\n",
 992                    0 );
 993           return off < 0 ? off : 0;
 994         }
 995         ret2 = dvdinput_read( dvd_file->title_devs[ i + 1 ],
 996                               data + ( part1_size
 997                                        * (int64_t)DVD_VIDEO_LB_LEN ),
 998                               (int)(block_count - part1_size),
 999                               encrypted );
1000         if( ret2 < 0 ) return ret2;
1001         break;
1002       }
1003     } else {
1004       offset -= dvd_file->title_sizes[ i ];
1005     }
1006   }
1007
1008   return ret + ret2;
1009 }
1010
1011 /* This is broken reading more than 2Gb at a time is ssize_t is 32-bit. */
1012 ssize_t DVDReadBlocks( dvd_file_t *dvd_file, int offset,
1013                        size_t block_count, unsigned char *data )
1014 {
1015   int ret;
1016
1017   /* Check arguments. */
1018   if( dvd_file == NULL || offset < 0 || data == NULL )
1019     return -1;
1020
1021   /* Hack, and it will still fail for multiple opens in a threaded app ! */
1022   if( dvd_file->dvd->css_title != dvd_file->css_title ) {
1023     dvd_file->dvd->css_title = dvd_file->css_title;
1024     if( dvd_file->dvd->isImageFile ) {
1025       dvdinput_title( dvd_file->dvd->dev, (int)dvd_file->lb_start );
1026     }
1027     /* Here each vobu has it's own dvdcss handle, so no need to update
1028     else {
1029       dvdinput_title( dvd_file->title_devs[ 0 ], (int)dvd_file->lb_start );
1030     }*/
1031   }
1032
1033   if( dvd_file->dvd->isImageFile ) {
1034     ret = DVDReadBlocksUDF( dvd_file, (uint32_t)offset,
1035                             block_count, data, DVDINPUT_READ_DECRYPT );
1036   } else {
1037     ret = DVDReadBlocksPath( dvd_file, (unsigned int)offset,
1038                              block_count, data, DVDINPUT_READ_DECRYPT );
1039   }
1040
1041   return (ssize_t)ret;
1042 }
1043
1044 int32_t DVDFileSeek( dvd_file_t *dvd_file, int32_t offset )
1045 {
1046   /* Check arguments. */
1047   if( dvd_file == NULL || offset < 0 )
1048     return -1;
1049
1050   if( offset > dvd_file->filesize * DVD_VIDEO_LB_LEN ) {
1051     return -1;
1052   }
1053   dvd_file->seek_pos = (uint32_t) offset;
1054   return offset;
1055 }
1056
1057 int DVDFileSeekForce(dvd_file_t *dvd_file, int offset, int force_size)
1058 {
1059   /* Check arguments. */
1060   if( dvd_file == NULL || offset <= 0 )
1061       return -1;
1062
1063   if( dvd_file->dvd->isImageFile ) {
1064     if( force_size < 0 )
1065       force_size = (offset - 1) / DVD_VIDEO_LB_LEN + 1;
1066     if( dvd_file->filesize < force_size ) {
1067       dvd_file->filesize = force_size;
1068       fprintf(stderr, "libdvdread: Ignored size of file indicated in UDF.\n");
1069     }
1070   }
1071
1072   if( offset > dvd_file->filesize * DVD_VIDEO_LB_LEN )
1073     return -1;
1074
1075   dvd_file->seek_pos = (uint32_t) offset;
1076   return offset;
1077 }
1078
1079 ssize_t DVDReadBytes( dvd_file_t *dvd_file, void *data, size_t byte_size )
1080 {
1081   unsigned char *secbuf_base, *secbuf;
1082   unsigned int numsec, seek_sector, seek_byte;
1083   int ret;
1084
1085   /* Check arguments. */
1086   if( dvd_file == NULL || data == NULL )
1087     return -1;
1088
1089   seek_sector = dvd_file->seek_pos / DVD_VIDEO_LB_LEN;
1090   seek_byte   = dvd_file->seek_pos % DVD_VIDEO_LB_LEN;
1091
1092   numsec = ( ( seek_byte + byte_size ) / DVD_VIDEO_LB_LEN ) +
1093     ( ( ( seek_byte + byte_size ) % DVD_VIDEO_LB_LEN ) ? 1 : 0 );
1094
1095   secbuf_base = (unsigned char *) malloc( numsec * DVD_VIDEO_LB_LEN + 2048 );
1096   secbuf = (unsigned char *)(((uintptr_t)secbuf_base & ~((uintptr_t)2047)) + 2048);
1097   if( !secbuf_base ) {
1098     fprintf( stderr, "libdvdread: Can't allocate memory "
1099              "for file read!\n" );
1100     return 0;
1101   }
1102
1103   if( dvd_file->dvd->isImageFile ) {
1104     ret = DVDReadBlocksUDF( dvd_file, (uint32_t) seek_sector,
1105                             (size_t) numsec, secbuf, DVDINPUT_NOFLAGS );
1106   } else {
1107     ret = DVDReadBlocksPath( dvd_file, seek_sector,
1108                              (size_t) numsec, secbuf, DVDINPUT_NOFLAGS );
1109   }
1110
1111   if( ret != (int) numsec ) {
1112     free( secbuf_base );
1113     return ret < 0 ? ret : 0;
1114   }
1115
1116   memcpy( data, &(secbuf[ seek_byte ]), byte_size );
1117   free( secbuf_base );
1118
1119   DVDFileSeekForce(dvd_file, dvd_file->seek_pos + byte_size, -1);
1120   return byte_size;
1121 }
1122
1123 ssize_t DVDFileSize( dvd_file_t *dvd_file )
1124 {
1125   /* Check arguments. */
1126   if( dvd_file == NULL )
1127     return -1;
1128
1129   return dvd_file->filesize;
1130 }
1131
1132 int DVDDiscID( dvd_reader_t *dvd, unsigned char *discid )
1133 {
1134   struct md5_ctx ctx;
1135   int title;
1136   int nr_of_files = 0;
1137
1138   /* Check arguments. */
1139   if( dvd == NULL || discid == NULL )
1140     return 0;
1141
1142   /* Go through the first 10 IFO:s, in order,
1143    * and md5sum them, i.e  VIDEO_TS.IFO and VTS_0?_0.IFO */
1144   md5_init_ctx( &ctx );
1145   for( title = 0; title < 10; title++ ) {
1146     dvd_file_t *dvd_file = DVDOpenFile( dvd, title, DVD_READ_INFO_FILE );
1147     if( dvd_file != NULL ) {
1148       ssize_t bytes_read;
1149       size_t file_size = dvd_file->filesize * DVD_VIDEO_LB_LEN;
1150       char *buffer_base = malloc( file_size + 2048 );
1151       char *buffer = (char *)(((uintptr_t)buffer_base & ~((uintptr_t)2047)) + 2048);
1152
1153       if( buffer_base == NULL ) {
1154           DVDCloseFile( dvd_file );
1155           fprintf( stderr, "libdvdread: DVDDiscId, failed to "
1156                    "allocate memory for file read!\n" );
1157           return -1;
1158       }
1159
1160       bytes_read = DVDReadBytes( dvd_file, buffer, file_size );
1161       if( bytes_read != file_size ) {
1162           fprintf( stderr, "libdvdread: DVDDiscId read returned %zd bytes"
1163                    ", wanted %zd\n", bytes_read, file_size );
1164           DVDCloseFile( dvd_file );
1165           free( buffer_base );
1166           return -1;
1167       }
1168
1169       md5_process_bytes( buffer, file_size,  &ctx );
1170
1171       DVDCloseFile( dvd_file );
1172       free( buffer_base );
1173       nr_of_files++;
1174     }
1175   }
1176   md5_finish_ctx( &ctx, discid );
1177   if(!nr_of_files)
1178     return -1;
1179
1180   return 0;
1181 }
1182
1183
1184 int DVDISOVolumeInfo( dvd_reader_t *dvd,
1185                       char *volid, unsigned int volid_size,
1186                       unsigned char *volsetid, unsigned int volsetid_size )
1187 {
1188   unsigned char *buffer, *buffer_base;
1189   int ret;
1190
1191   /* Check arguments. */
1192   if( dvd == NULL )
1193     return 0;
1194
1195   if( dvd->dev == NULL ) {
1196     /* No block access, so no ISO... */
1197     return -1;
1198   }
1199
1200   buffer_base = malloc( DVD_VIDEO_LB_LEN + 2048 );
1201   buffer = (unsigned char *)(((uintptr_t)buffer_base & ~((uintptr_t)2047)) + 2048);
1202
1203   if( buffer_base == NULL ) {
1204     fprintf( stderr, "libdvdread: DVDISOVolumeInfo, failed to "
1205              "allocate memory for file read!\n" );
1206     return -1;
1207   }
1208
1209   ret = UDFReadBlocksRaw( dvd, 16, 1, buffer, 0 );
1210   if( ret != 1 ) {
1211     fprintf( stderr, "libdvdread: DVDISOVolumeInfo, failed to "
1212              "read ISO9660 Primary Volume Descriptor!\n" );
1213     free( buffer_base );
1214     return -1;
1215   }
1216
1217   if( (volid != NULL) && (volid_size > 0) ) {
1218     unsigned int n;
1219     for(n = 0; n < 32; n++) {
1220       if(buffer[40+n] == 0x20) {
1221         break;
1222       }
1223     }
1224
1225     if(volid_size > n+1) {
1226       volid_size = n+1;
1227     }
1228
1229     memcpy(volid, &buffer[40], volid_size-1);
1230     volid[volid_size-1] = '\0';
1231   }
1232
1233   if( (volsetid != NULL) && (volsetid_size > 0) ) {
1234     if(volsetid_size > 128) {
1235       volsetid_size = 128;
1236     }
1237     memcpy(volsetid, &buffer[190], volsetid_size);
1238   }
1239   free( buffer_base );
1240   return 0;
1241 }
1242
1243
1244 int DVDUDFVolumeInfo( dvd_reader_t *dvd,
1245                       char *volid, unsigned int volid_size,
1246                       unsigned char *volsetid, unsigned int volsetid_size )
1247 {
1248   int ret;
1249   /* Check arguments. */
1250   if( dvd == NULL )
1251     return -1;
1252
1253   if( dvd->dev == NULL ) {
1254     /* No block access, so no UDF VolumeSet Identifier */
1255     return -1;
1256   }
1257
1258   if( (volid != NULL) && (volid_size > 0) ) {
1259     ret = UDFGetVolumeIdentifier(dvd, volid, volid_size);
1260     if(!ret) {
1261       return -1;
1262     }
1263   }
1264   if( (volsetid != NULL) && (volsetid_size > 0) ) {
1265     ret =  UDFGetVolumeSetIdentifier(dvd, volsetid, volsetid_size);
1266     if(!ret) {
1267       return -1;
1268     }
1269   }
1270
1271   return 0;
1272 }