898 lines
24 KiB
C
898 lines
24 KiB
C
/*****************************************************************************
|
|
* device.h: DVD device access
|
|
*****************************************************************************
|
|
* Copyright (C) 1998-2006 VideoLAN
|
|
*
|
|
* Authors: Stéphane Borel <stef@via.ecp.fr>
|
|
* Sam Hocevar <sam@zoy.org>
|
|
* Håkan Hjort <d95hjort@dtek.chalmers.se>
|
|
*
|
|
* libdvdcss is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* libdvdcss is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with libdvdcss; if not, write to the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*****************************************************************************/
|
|
|
|
/*****************************************************************************
|
|
* Preamble
|
|
*****************************************************************************/
|
|
#include "config.h"
|
|
|
|
#include <limits.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#ifdef HAVE_ERRNO_H
|
|
# include <errno.h>
|
|
#endif
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#ifdef HAVE_SYS_PARAM_H
|
|
# include <sys/param.h>
|
|
#endif
|
|
#include <fcntl.h>
|
|
|
|
#ifdef HAVE_UNISTD_H
|
|
# include <unistd.h>
|
|
#endif
|
|
|
|
#ifdef DARWIN_DVD_IOCTL
|
|
# include <paths.h>
|
|
# include <CoreFoundation/CoreFoundation.h>
|
|
# include <IOKit/IOKitLib.h>
|
|
# include <IOKit/IOBSD.h>
|
|
# include <IOKit/storage/IOMedia.h>
|
|
# include <IOKit/storage/IOCDMedia.h>
|
|
# include <IOKit/storage/IODVDMedia.h>
|
|
#endif
|
|
|
|
#ifdef __OS2__
|
|
# define INCL_DOS
|
|
# define INCL_DOSDEVIOCTL
|
|
# include <os2.h>
|
|
# include <io.h> /* setmode() */
|
|
# include <fcntl.h> /* O_BINARY */
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
# include <windows.h>
|
|
#endif
|
|
|
|
#include "dvdcss/dvdcss.h"
|
|
|
|
#include "common.h"
|
|
#include "css.h"
|
|
#include "libdvdcss.h"
|
|
#include "ioctl.h"
|
|
#include "device.h"
|
|
|
|
/*****************************************************************************
|
|
* Device reading prototypes
|
|
*****************************************************************************/
|
|
static int libc_open ( dvdcss_t, const char * );
|
|
static int libc_seek ( dvdcss_t, int );
|
|
static int libc_read ( dvdcss_t, void *, int );
|
|
static int libc_readv ( dvdcss_t, const struct iovec *, int );
|
|
|
|
static int stream_seek ( dvdcss_t, int );
|
|
static int stream_read ( dvdcss_t, void *, int );
|
|
static int stream_readv ( dvdcss_t, const struct iovec *, int );
|
|
|
|
#ifdef _WIN32
|
|
static int win2k_open ( dvdcss_t, const char * );
|
|
static int win2k_seek ( dvdcss_t, int );
|
|
static int win2k_read ( dvdcss_t, void *, int );
|
|
static int win2k_readv ( dvdcss_t, const struct iovec *, int );
|
|
|
|
#elif defined( __OS2__ )
|
|
static int os2_open ( dvdcss_t, const char * );
|
|
#endif
|
|
|
|
int dvdcss_use_ioctls( dvdcss_t dvdcss )
|
|
{
|
|
if( dvdcss->p_stream )
|
|
return 0;
|
|
|
|
#if defined( _WIN32 )
|
|
if( dvdcss->b_file )
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
/* FIXME: implement this for Windows */
|
|
return 1;
|
|
#elif defined( __OS2__ )
|
|
ULONG ulMode;
|
|
|
|
if( DosQueryFHState( dvdcss->i_fd, &ulMode ) != 0 )
|
|
return 1; /* What to do? Be conservative and try to use the ioctls */
|
|
|
|
if( ulMode & OPEN_FLAGS_DASD )
|
|
return 1;
|
|
|
|
return 0;
|
|
#else
|
|
struct stat fileinfo;
|
|
int ret;
|
|
|
|
ret = fstat( dvdcss->i_fd, &fileinfo );
|
|
if( ret < 0 )
|
|
{
|
|
return 1; /* What to do? Be conservative and try to use the ioctls */
|
|
}
|
|
|
|
/* Complete this list and check that we test for the right things
|
|
* (I've assumed for all OSs that 'r', (raw) device, are char devices
|
|
* and those that don't contain/use an 'r' in the name are block devices)
|
|
*
|
|
* Linux needs a block device
|
|
* Solaris needs a char device
|
|
* Darwin needs a char device
|
|
* OpenBSD needs a char device
|
|
* NetBSD needs a char device
|
|
* FreeBSD can use either the block or the char device
|
|
*/
|
|
|
|
/* Check if this is a block/char device */
|
|
if( S_ISBLK( fileinfo.st_mode ) ||
|
|
S_ISCHR( fileinfo.st_mode ) )
|
|
{
|
|
return 1;
|
|
}
|
|
else
|
|
{
|
|
return 0;
|
|
}
|
|
#endif
|
|
}
|
|
|
|
void dvdcss_check_device ( dvdcss_t dvdcss )
|
|
{
|
|
#if defined( _WIN32 )
|
|
DWORD drives;
|
|
int i;
|
|
#elif defined( DARWIN_DVD_IOCTL )
|
|
io_object_t next_media;
|
|
mach_port_t master_port;
|
|
kern_return_t kern_result;
|
|
io_iterator_t media_iterator;
|
|
CFMutableDictionaryRef classes_to_match;
|
|
#elif defined( __OS2__ )
|
|
#pragma pack( 1 )
|
|
struct
|
|
{
|
|
BYTE bCmdInfo;
|
|
BYTE bDrive;
|
|
} param;
|
|
|
|
struct
|
|
{
|
|
BYTE abEBPB[31];
|
|
USHORT usCylinders;
|
|
BYTE bDevType;
|
|
USHORT usDevAttr;
|
|
} data;
|
|
#pragma pack()
|
|
|
|
ULONG ulParamLen;
|
|
ULONG ulDataLen;
|
|
ULONG rc;
|
|
|
|
int i;
|
|
#else
|
|
const char *ppsz_devices[] = { "/dev/dvd", "/dev/cdrom", "/dev/hdc", NULL };
|
|
int i, i_fd;
|
|
#endif
|
|
|
|
/* If the device name is non-NULL or stream is set, return. */
|
|
if( (dvdcss->psz_device && dvdcss->psz_device[0]) || dvdcss->p_stream )
|
|
{
|
|
return;
|
|
}
|
|
|
|
#if defined( _WIN32 )
|
|
drives = GetLogicalDrives();
|
|
|
|
for( i = 0; drives; i++ )
|
|
{
|
|
char psz_device[5];
|
|
DWORD cur = 1 << i;
|
|
UINT i_ret;
|
|
|
|
if( (drives & cur) == 0 )
|
|
{
|
|
continue;
|
|
}
|
|
drives &= ~cur;
|
|
|
|
sprintf( psz_device, "%c:\\", 'A' + i );
|
|
i_ret = GetDriveType( psz_device );
|
|
if( i_ret != DRIVE_CDROM )
|
|
{
|
|
continue;
|
|
}
|
|
|
|
/* Remove trailing backslash */
|
|
psz_device[2] = '\0';
|
|
|
|
/* FIXME: we want to differentiate between CD and DVD drives
|
|
* using DeviceIoControl() */
|
|
print_debug( dvdcss, "defaulting to drive `%s'", psz_device );
|
|
free( dvdcss->psz_device );
|
|
dvdcss->psz_device = strdup( psz_device );
|
|
return;
|
|
}
|
|
#elif defined( DARWIN_DVD_IOCTL )
|
|
|
|
kern_result = IOMasterPort( MACH_PORT_NULL, &master_port );
|
|
if( kern_result != KERN_SUCCESS )
|
|
{
|
|
return;
|
|
}
|
|
|
|
classes_to_match = IOServiceMatching( kIODVDMediaClass );
|
|
if( classes_to_match == NULL )
|
|
{
|
|
return;
|
|
}
|
|
|
|
CFDictionarySetValue( classes_to_match, CFSTR( kIOMediaEjectableKey ),
|
|
kCFBooleanTrue );
|
|
|
|
kern_result = IOServiceGetMatchingServices( master_port, classes_to_match,
|
|
&media_iterator );
|
|
if( kern_result != KERN_SUCCESS )
|
|
{
|
|
return;
|
|
}
|
|
|
|
next_media = IOIteratorNext( media_iterator );
|
|
for( ; ; )
|
|
{
|
|
char psz_buf[0x32];
|
|
size_t i_pathlen;
|
|
CFTypeRef psz_path;
|
|
|
|
next_media = IOIteratorNext( media_iterator );
|
|
if( next_media == 0 )
|
|
{
|
|
break;
|
|
}
|
|
|
|
psz_path = IORegistryEntryCreateCFProperty( next_media,
|
|
CFSTR( kIOBSDNameKey ),
|
|
kCFAllocatorDefault,
|
|
0 );
|
|
if( psz_path == NULL )
|
|
{
|
|
IOObjectRelease( next_media );
|
|
continue;
|
|
}
|
|
|
|
snprintf( psz_buf, sizeof(psz_buf), "%s%c", _PATH_DEV, 'r' );
|
|
i_pathlen = strlen( psz_buf );
|
|
|
|
if( CFStringGetCString( psz_path,
|
|
(char*)&psz_buf + i_pathlen,
|
|
sizeof(psz_buf) - i_pathlen,
|
|
kCFStringEncodingASCII ) )
|
|
{
|
|
print_debug( dvdcss, "defaulting to drive `%s'", psz_buf );
|
|
CFRelease( psz_path );
|
|
IOObjectRelease( next_media );
|
|
IOObjectRelease( media_iterator );
|
|
free( dvdcss->psz_device );
|
|
dvdcss->psz_device = strdup( psz_buf );
|
|
return;
|
|
}
|
|
|
|
CFRelease( psz_path );
|
|
|
|
IOObjectRelease( next_media );
|
|
}
|
|
|
|
IOObjectRelease( media_iterator );
|
|
#elif defined( __OS2__ )
|
|
for( i = 0; i < 26; i++ )
|
|
{
|
|
param.bCmdInfo = 0;
|
|
param.bDrive = i;
|
|
|
|
rc = DosDevIOCtl( ( HFILE )-1, IOCTL_DISK, DSK_GETDEVICEPARAMS,
|
|
¶m, sizeof( param ), &ulParamLen,
|
|
&data, sizeof( data ), &ulDataLen );
|
|
|
|
if( rc == 0 )
|
|
{
|
|
/* Check for removable and for cylinders */
|
|
if( ( data.usDevAttr & 1 ) == 0 && data.usCylinders == 0xFFFF )
|
|
{
|
|
char psz_dvd[] = "A:";
|
|
|
|
psz_dvd[0] += i;
|
|
|
|
print_debug( dvdcss, "defaulting to drive `%s'", psz_dvd );
|
|
free( dvdcss->psz_device );
|
|
dvdcss->psz_device = strdup( psz_dvd );
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
#else
|
|
for( i = 0; ppsz_devices[i]; i++ )
|
|
{
|
|
i_fd = open( ppsz_devices[i], 0 );
|
|
if( i_fd != -1 )
|
|
{
|
|
print_debug( dvdcss, "defaulting to drive `%s'", ppsz_devices[i] );
|
|
close( i_fd );
|
|
free( dvdcss->psz_device );
|
|
dvdcss->psz_device = strdup( ppsz_devices[i] );
|
|
return;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
print_error( dvdcss, "could not find a suitable default drive" );
|
|
}
|
|
|
|
int dvdcss_open_device ( dvdcss_t dvdcss )
|
|
{
|
|
const char *psz_device = getenv( "DVDCSS_RAW_DEVICE" );
|
|
if( !psz_device )
|
|
{
|
|
psz_device = dvdcss->psz_device;
|
|
}
|
|
print_debug( dvdcss, "opening target `%s'", psz_device );
|
|
|
|
#if defined( _WIN32 )
|
|
/* Initialize readv temporary buffer */
|
|
dvdcss->p_readv_buffer = NULL;
|
|
dvdcss->i_readv_buf_size = 0;
|
|
#endif
|
|
|
|
/* if callback functions are initialized */
|
|
if( dvdcss->p_stream )
|
|
{
|
|
print_debug( dvdcss, "using stream API for access" );
|
|
dvdcss->pf_seek = stream_seek;
|
|
dvdcss->pf_read = stream_read;
|
|
dvdcss->pf_readv = stream_readv;
|
|
return 0;
|
|
}
|
|
|
|
#if defined( _WIN32 )
|
|
dvdcss->b_file = 1;
|
|
/* If device is "X:" or "X:\", we are not actually opening a file. */
|
|
if (psz_device[0] && psz_device[1] == ':' &&
|
|
(!psz_device[2] || (psz_device[2] == '\\' && !psz_device[3])))
|
|
dvdcss->b_file = 0;
|
|
|
|
if( !dvdcss->b_file )
|
|
{
|
|
print_debug( dvdcss, "using Win2K API for access" );
|
|
dvdcss->pf_seek = win2k_seek;
|
|
dvdcss->pf_read = win2k_read;
|
|
dvdcss->pf_readv = win2k_readv;
|
|
int i_ret = win2k_open( dvdcss, psz_device );
|
|
ioctl_LockVolume(dvdcss->i_fd);
|
|
return i_ret;
|
|
}
|
|
else
|
|
#elif defined( __OS2__ )
|
|
/* If device is "X:" or "X:\", we are not actually opening a file. */
|
|
if( psz_device[0] && psz_device[1] == ':' &&
|
|
( !psz_device[2] || ( psz_device[2] == '\\' && !psz_device[3] ) ) )
|
|
{
|
|
print_debug( dvdcss, "using OS/2 API for access" );
|
|
dvdcss->pf_seek = libc_seek;
|
|
dvdcss->pf_read = libc_read;
|
|
dvdcss->pf_readv = libc_readv;
|
|
int i_ret = os2_open( dvdcss, psz_device );
|
|
ioctl_LockVolume(dvdcss->i_fd);
|
|
return i_ret;
|
|
}
|
|
else
|
|
#endif
|
|
{
|
|
print_debug( dvdcss, "using libc API for access" );
|
|
dvdcss->pf_seek = libc_seek;
|
|
dvdcss->pf_read = libc_read;
|
|
dvdcss->pf_readv = libc_readv;
|
|
int i_ret = libc_open( dvdcss, psz_device );
|
|
ioctl_LockVolume(dvdcss->i_fd);
|
|
return i_ret;
|
|
}
|
|
}
|
|
|
|
int dvdcss_close_device ( dvdcss_t dvdcss )
|
|
{
|
|
if( dvdcss->p_stream )
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
ioctl_UnlockVolume(dvdcss->i_fd);
|
|
|
|
#if defined( _WIN32 )
|
|
/* Free readv temporary buffer */
|
|
free( dvdcss->p_readv_buffer );
|
|
dvdcss->p_readv_buffer = NULL;
|
|
dvdcss->i_readv_buf_size = 0;
|
|
|
|
if( !dvdcss->b_file )
|
|
{
|
|
CloseHandle( (HANDLE) dvdcss->i_fd );
|
|
}
|
|
else
|
|
#endif
|
|
{
|
|
int i_ret = close( dvdcss->i_fd );
|
|
if( i_ret < 0 )
|
|
{
|
|
print_error( dvdcss, "Failed to close fd, data loss possible." );
|
|
return i_ret;
|
|
}
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/* Following functions are local */
|
|
|
|
/*****************************************************************************
|
|
* Open commands.
|
|
*****************************************************************************/
|
|
static int libc_open ( dvdcss_t dvdcss, const char *psz_device )
|
|
{
|
|
#ifdef _WIN32
|
|
int wlen;
|
|
dvdcss->i_fd = -1;
|
|
wlen = MultiByteToWideChar( CP_UTF8, 0, psz_device, -1, NULL, 0 );
|
|
if( wlen > 0 ) {
|
|
wchar_t *wpath = (wchar_t*)malloc( sizeof(wchar_t) * wlen );
|
|
if( wpath ) {
|
|
if( MultiByteToWideChar(CP_UTF8, 0, psz_device, -1, wpath, wlen ) ) {
|
|
dvdcss->i_fd = _wopen( wpath, O_BINARY );
|
|
}
|
|
free( wpath );
|
|
}
|
|
}
|
|
#else
|
|
dvdcss->i_fd = open( psz_device, O_BINARY );
|
|
#endif
|
|
|
|
if( dvdcss->i_fd == -1 )
|
|
{
|
|
print_error( dvdcss, "failed to open device %s (%s)",
|
|
psz_device, strerror(errno) );
|
|
return -1;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
#if defined( _WIN32 )
|
|
static int win2k_open ( dvdcss_t dvdcss, const char *psz_device )
|
|
{
|
|
char psz_dvd[7] = "\\\\.\\\0:";
|
|
psz_dvd[4] = psz_device[0];
|
|
|
|
/* To work around an M$ bug in IOCTL_DVD_READ_STRUCTURE, we need read
|
|
* _and_ write access to the device (so we can make SCSI Pass Through
|
|
* Requests). Unfortunately this is only allowed if you have
|
|
* administrator privileges so we allow for a fallback method with
|
|
* only read access to the device (in this case ioctl_ReadCopyright()
|
|
* won't send back the right result).
|
|
* (See Microsoft Q241374: Read and Write Access Required for SCSI
|
|
* Pass Through Requests) */
|
|
dvdcss->i_fd = (int)
|
|
CreateFile( psz_dvd, GENERIC_READ | GENERIC_WRITE,
|
|
FILE_SHARE_READ | FILE_SHARE_WRITE,
|
|
NULL, OPEN_EXISTING,
|
|
FILE_FLAG_RANDOM_ACCESS, NULL );
|
|
|
|
if( (HANDLE) dvdcss->i_fd == INVALID_HANDLE_VALUE )
|
|
dvdcss->i_fd = (int)
|
|
CreateFile( psz_dvd, GENERIC_READ, FILE_SHARE_READ,
|
|
NULL, OPEN_EXISTING,
|
|
FILE_FLAG_RANDOM_ACCESS, NULL );
|
|
|
|
if( (HANDLE) dvdcss->i_fd == INVALID_HANDLE_VALUE )
|
|
{
|
|
print_error( dvdcss, "failed to open device %s", psz_device );
|
|
return -1;
|
|
}
|
|
|
|
dvdcss->i_pos = 0;
|
|
|
|
return 0;
|
|
}
|
|
#endif /* defined( _WIN32 ) */
|
|
|
|
#ifdef __OS2__
|
|
static int os2_open ( dvdcss_t dvdcss, const char *psz_device )
|
|
{
|
|
char psz_dvd[] = "X:";
|
|
HFILE hfile;
|
|
ULONG ulAction;
|
|
ULONG rc;
|
|
|
|
psz_dvd[0] = psz_device[0];
|
|
|
|
rc = DosOpenL( ( PSZ )psz_dvd, &hfile, &ulAction, 0, FILE_NORMAL,
|
|
OPEN_ACTION_OPEN_IF_EXISTS | OPEN_ACTION_FAIL_IF_NEW,
|
|
OPEN_ACCESS_READONLY | OPEN_SHARE_DENYNONE | OPEN_FLAGS_DASD,
|
|
NULL );
|
|
|
|
if( rc )
|
|
{
|
|
print_error( dvdcss, "failed to open device %s", psz_device );
|
|
return -1;
|
|
}
|
|
|
|
setmode( hfile, O_BINARY );
|
|
|
|
dvdcss->i_fd = hfile;
|
|
|
|
dvdcss->i_pos = 0;
|
|
|
|
return 0;
|
|
}
|
|
#endif /* __OS2__ */
|
|
|
|
/*****************************************************************************
|
|
* Seek commands.
|
|
*****************************************************************************/
|
|
static int libc_seek( dvdcss_t dvdcss, int i_blocks )
|
|
{
|
|
off_t i_seek;
|
|
|
|
if( dvdcss->i_pos == i_blocks )
|
|
{
|
|
/* We are already in position */
|
|
return i_blocks;
|
|
}
|
|
|
|
i_seek = (off_t)i_blocks * (off_t)DVDCSS_BLOCK_SIZE;
|
|
i_seek = lseek( dvdcss->i_fd, i_seek, SEEK_SET );
|
|
|
|
if( i_seek < 0 )
|
|
{
|
|
print_error( dvdcss, "seek error" );
|
|
dvdcss->i_pos = -1;
|
|
return i_seek;
|
|
}
|
|
|
|
dvdcss->i_pos = i_seek / DVDCSS_BLOCK_SIZE;
|
|
|
|
return dvdcss->i_pos;
|
|
}
|
|
|
|
static int stream_seek( dvdcss_t dvdcss, int i_blocks )
|
|
{
|
|
off_t i_seek = (off_t) i_blocks * (off_t) DVDCSS_BLOCK_SIZE;
|
|
|
|
if( !dvdcss->p_stream_cb->pf_seek )
|
|
return -1;
|
|
|
|
if( dvdcss->i_pos == i_blocks )
|
|
{
|
|
/* We are already in position */
|
|
return i_blocks;
|
|
}
|
|
|
|
if( dvdcss->p_stream_cb->pf_seek( dvdcss->p_stream, i_seek ) != 0 )
|
|
{
|
|
print_error( dvdcss, "seek error" );
|
|
dvdcss->i_pos = -1;
|
|
return -1;
|
|
}
|
|
|
|
dvdcss->i_pos = i_blocks;
|
|
|
|
return dvdcss->i_pos;
|
|
}
|
|
|
|
#if defined( _WIN32 )
|
|
static int win2k_seek( dvdcss_t dvdcss, int i_blocks )
|
|
{
|
|
LARGE_INTEGER li_seek;
|
|
|
|
if( dvdcss->i_pos == i_blocks )
|
|
{
|
|
/* We are already in position */
|
|
return i_blocks;
|
|
}
|
|
|
|
li_seek.QuadPart = (LONGLONG)i_blocks * DVDCSS_BLOCK_SIZE;
|
|
|
|
li_seek.LowPart = SetFilePointer( (HANDLE) dvdcss->i_fd,
|
|
li_seek.LowPart,
|
|
&li_seek.HighPart, FILE_BEGIN );
|
|
if( (li_seek.LowPart == INVALID_SET_FILE_POINTER)
|
|
&& GetLastError() != NO_ERROR)
|
|
{
|
|
dvdcss->i_pos = -1;
|
|
return -1;
|
|
}
|
|
|
|
dvdcss->i_pos = li_seek.QuadPart / DVDCSS_BLOCK_SIZE;
|
|
|
|
return dvdcss->i_pos;
|
|
}
|
|
#endif /* defined( _WIN32 ) */
|
|
|
|
/*****************************************************************************
|
|
* Read commands.
|
|
*****************************************************************************/
|
|
static int libc_read ( dvdcss_t dvdcss, void *p_buffer, int i_blocks )
|
|
{
|
|
off_t i_size, i_ret, i_ret_blocks;
|
|
|
|
i_size = (off_t)i_blocks * (off_t)DVDCSS_BLOCK_SIZE;
|
|
i_ret = read( dvdcss->i_fd, p_buffer, i_size );
|
|
|
|
if( i_ret < 0 )
|
|
{
|
|
print_error( dvdcss, "read error" );
|
|
dvdcss->i_pos = -1;
|
|
return i_ret;
|
|
}
|
|
|
|
i_ret_blocks = i_ret / DVDCSS_BLOCK_SIZE;
|
|
|
|
/* Handle partial reads */
|
|
if( i_ret != i_size )
|
|
{
|
|
int i_seek, i_set_pos;
|
|
|
|
i_set_pos = dvdcss->i_pos + i_ret_blocks;
|
|
dvdcss->i_pos = -1;
|
|
i_seek = libc_seek( dvdcss, i_set_pos );
|
|
if( i_seek < 0 )
|
|
{
|
|
return i_seek;
|
|
}
|
|
|
|
/* We have to return now so that i_pos isn't clobbered */
|
|
return i_ret_blocks;
|
|
}
|
|
|
|
dvdcss->i_pos += i_ret_blocks;
|
|
return i_ret_blocks;
|
|
}
|
|
|
|
static int stream_read ( dvdcss_t dvdcss, void *p_buffer, int i_blocks )
|
|
{
|
|
off_t i_size, i_ret, i_ret_blocks;
|
|
|
|
i_size = (off_t)i_blocks * (off_t)DVDCSS_BLOCK_SIZE;
|
|
|
|
if( !dvdcss->p_stream_cb->pf_read )
|
|
return -1;
|
|
|
|
i_ret = dvdcss->p_stream_cb->pf_read( dvdcss->p_stream, p_buffer, i_size );
|
|
|
|
if( i_ret < 0 )
|
|
{
|
|
print_error( dvdcss, "read error" );
|
|
dvdcss->i_pos = -1;
|
|
return i_ret;
|
|
}
|
|
|
|
i_ret_blocks = i_ret / DVDCSS_BLOCK_SIZE;
|
|
|
|
/* Handle partial reads */
|
|
if( i_ret != i_size )
|
|
{
|
|
int i_seek;
|
|
|
|
dvdcss->i_pos = -1;
|
|
i_seek = stream_seek( dvdcss, i_ret_blocks );
|
|
if( i_seek < 0 )
|
|
{
|
|
return i_seek;
|
|
}
|
|
|
|
/* We have to return now so that i_pos isn't clobbered */
|
|
return i_ret_blocks;
|
|
}
|
|
|
|
dvdcss->i_pos += i_ret_blocks;
|
|
return i_ret_blocks;
|
|
}
|
|
|
|
#if defined( _WIN32 )
|
|
static int win2k_read ( dvdcss_t dvdcss, void *p_buffer, int i_blocks )
|
|
{
|
|
DWORD i_bytes;
|
|
|
|
if( !ReadFile( (HANDLE) dvdcss->i_fd, p_buffer,
|
|
i_blocks * DVDCSS_BLOCK_SIZE,
|
|
&i_bytes, NULL ) )
|
|
{
|
|
dvdcss->i_pos = -1;
|
|
return -1;
|
|
}
|
|
|
|
i_bytes /= DVDCSS_BLOCK_SIZE;
|
|
dvdcss->i_pos += i_bytes;
|
|
return i_bytes;
|
|
}
|
|
#endif /* defined( _WIN32 ) */
|
|
|
|
/*****************************************************************************
|
|
* Readv commands.
|
|
*****************************************************************************/
|
|
static int libc_readv ( dvdcss_t dvdcss, const struct iovec *p_iovec,
|
|
int i_blocks )
|
|
{
|
|
#if defined( _WIN32 )
|
|
int i_index, i_len, i_total = 0;
|
|
unsigned char *p_base;
|
|
int i_bytes;
|
|
|
|
for( i_index = i_blocks;
|
|
i_index;
|
|
i_index--, p_iovec++ )
|
|
{
|
|
i_len = p_iovec->iov_len;
|
|
p_base = p_iovec->iov_base;
|
|
|
|
if( i_len <= 0 )
|
|
{
|
|
continue;
|
|
}
|
|
|
|
i_bytes = read( dvdcss->i_fd, p_base, i_len );
|
|
|
|
if( i_bytes < 0 )
|
|
{
|
|
/* One of the reads failed, too bad.
|
|
* We won't even bother returning the reads that went OK,
|
|
* and as in the POSIX spec the file position is left
|
|
* unspecified after a failure */
|
|
dvdcss->i_pos = -1;
|
|
return -1;
|
|
}
|
|
|
|
i_total += i_bytes;
|
|
i_total /= DVDCSS_BLOCK_SIZE;
|
|
|
|
if( i_bytes != i_len )
|
|
{
|
|
/* We reached the end of the file or a signal interrupted
|
|
* the read. Return a partial read. */
|
|
int i_seek;
|
|
|
|
dvdcss->i_pos = -1;
|
|
i_seek = libc_seek( dvdcss, i_total );
|
|
if( i_seek < 0 )
|
|
{
|
|
return i_seek;
|
|
}
|
|
|
|
/* We have to return now so that i_pos isn't clobbered */
|
|
return i_total;
|
|
}
|
|
}
|
|
|
|
dvdcss->i_pos += i_total;
|
|
return i_total;
|
|
#else
|
|
int i_read = readv( dvdcss->i_fd, p_iovec, i_blocks );
|
|
|
|
if( i_read < 0 )
|
|
{
|
|
dvdcss->i_pos = -1;
|
|
return i_read;
|
|
}
|
|
|
|
i_read /= DVDCSS_BLOCK_SIZE;
|
|
dvdcss->i_pos += i_read;
|
|
return i_read;
|
|
#endif
|
|
}
|
|
|
|
/*****************************************************************************
|
|
* stream_readv: vectored read
|
|
*****************************************************************************/
|
|
static int stream_readv ( dvdcss_t dvdcss, const struct iovec *p_iovec,
|
|
int i_blocks )
|
|
{
|
|
int i_read;
|
|
|
|
if( !dvdcss->p_stream_cb->pf_readv )
|
|
return -1;
|
|
|
|
i_read = dvdcss->p_stream_cb->pf_readv( dvdcss->p_stream, p_iovec,
|
|
i_blocks );
|
|
|
|
if( i_read < 0 )
|
|
{
|
|
dvdcss->i_pos = -1;
|
|
return i_read;
|
|
}
|
|
|
|
dvdcss->i_pos += i_read / DVDCSS_BLOCK_SIZE;
|
|
return i_read / DVDCSS_BLOCK_SIZE;
|
|
}
|
|
|
|
#if defined( _WIN32 )
|
|
/*****************************************************************************
|
|
* win2k_readv: vectored read using ReadFile for Win2K
|
|
*****************************************************************************/
|
|
static int win2k_readv ( dvdcss_t dvdcss, const struct iovec *p_iovec,
|
|
int i_blocks )
|
|
{
|
|
int i_index;
|
|
int i_blocks_read, i_blocks_total = 0;
|
|
DWORD i_bytes;
|
|
|
|
/* Check the size of the readv temp buffer, just in case we need to
|
|
* realloc something bigger */
|
|
if( dvdcss->i_readv_buf_size < i_blocks * DVDCSS_BLOCK_SIZE )
|
|
{
|
|
dvdcss->i_readv_buf_size = i_blocks * DVDCSS_BLOCK_SIZE;
|
|
|
|
free( dvdcss->p_readv_buffer );
|
|
|
|
/* Allocate a buffer which will be used as a temporary storage
|
|
* for readv */
|
|
dvdcss->p_readv_buffer = malloc( dvdcss->i_readv_buf_size );
|
|
if( !dvdcss->p_readv_buffer )
|
|
{
|
|
print_error( dvdcss, "scatter input (readv) failed" );
|
|
dvdcss->i_pos = -1;
|
|
return -1;
|
|
}
|
|
}
|
|
|
|
for( i_index = i_blocks; i_index; i_index-- )
|
|
{
|
|
i_blocks_total += p_iovec[i_index-1].iov_len;
|
|
}
|
|
|
|
if( i_blocks_total <= 0 ) return 0;
|
|
|
|
if( !ReadFile( (HANDLE)dvdcss->i_fd, dvdcss->p_readv_buffer,
|
|
i_blocks_total, &i_bytes, NULL ) )
|
|
{
|
|
/* The read failed... too bad.
|
|
* As in the POSIX spec the file position is left
|
|
* unspecified after a failure */
|
|
dvdcss->i_pos = -1;
|
|
return -1;
|
|
}
|
|
i_blocks_read = i_bytes / DVDCSS_BLOCK_SIZE;
|
|
|
|
/* We just have to copy the content of the temp buffer into the iovecs */
|
|
for( i_index = 0, i_blocks_total = i_blocks_read;
|
|
i_blocks_total > 0;
|
|
i_index++ )
|
|
{
|
|
memcpy( p_iovec[i_index].iov_base,
|
|
dvdcss->p_readv_buffer + (i_blocks_read - i_blocks_total)
|
|
* DVDCSS_BLOCK_SIZE,
|
|
p_iovec[i_index].iov_len );
|
|
/* if we read less blocks than asked, we'll just end up copying
|
|
* garbage, this isn't an issue as we return the number of
|
|
* blocks actually read */
|
|
i_blocks_total -= ( p_iovec[i_index].iov_len / DVDCSS_BLOCK_SIZE );
|
|
}
|
|
|
|
dvdcss->i_pos += i_blocks_read;
|
|
return i_blocks_read;
|
|
}
|
|
#endif /* defined( _WIN32 ) */
|