mpv/libmpdvdkit2/device.c

806 lines
22 KiB
C

/*****************************************************************************
* device.h: DVD device access
*****************************************************************************
* Copyright (C) 1998-2002 VideoLAN
* $Id$
*
* Authors: Stéphane Borel <stef@via.ecp.fr>
* Samuel Hocevar <sam@zoy.org>
* Håkan Hjort <d95hjort@dtek.chalmers.se>
*
* This program 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.
*
* This program 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 this program; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
*****************************************************************************/
/*****************************************************************************
* Preamble
*****************************************************************************/
#include "config.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <limits.h>
#if defined( WIN32 ) && !defined( SYS_CYGWIN )
# include <io.h> /* read() */
#else
# include <sys/uio.h> /* struct iovec */
#endif
#include "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, char const * );
static int libc_seek ( dvdcss_t, int );
static int libc_read ( dvdcss_t, void *, int );
static int libc_readv ( dvdcss_t, struct iovec *, int );
#ifdef WIN32
static int win2k_open ( dvdcss_t, char const * );
static int aspi_open ( dvdcss_t, char const * );
static int win2k_seek ( dvdcss_t, int );
static int aspi_seek ( dvdcss_t, int );
static int win2k_read ( dvdcss_t, void *, int );
static int aspi_read ( dvdcss_t, void *, int );
static int win_readv ( dvdcss_t, struct iovec *, int );
static int aspi_read_internal ( int, void *, int );
#endif
int _dvdcss_use_ioctls( dvdcss_t dvdcss )
{
#if defined( WIN32 )
if( dvdcss->b_file )
{
return 0;
}
/* FIXME: implement this for Windows */
if( WIN2K )
{
return 1;
}
else
{
return 1;
}
#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
* BSD/OS 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
}
int _dvdcss_open ( dvdcss_t dvdcss )
{
char psz_debug[200];
char const *psz_device = dvdcss->psz_device;
snprintf( psz_debug, 199, "opening target `%s'", psz_device );
psz_debug[199] = '\0';
_dvdcss_debug( dvdcss, psz_debug );
#if defined( WIN32 )
/* If device is not "X:", we are actually opening a file. */
dvdcss->b_file = !psz_device[0] || psz_device[1] != ':' || psz_device[3];
/* Initialize readv temporary buffer */
dvdcss->p_readv_buffer = NULL;
dvdcss->i_readv_buf_size = 0;
if( !dvdcss->b_file && WIN2K )
{
_dvdcss_debug( dvdcss, "using Win2K API for access" );
dvdcss->pf_seek = win2k_seek;
dvdcss->pf_read = win2k_read;
dvdcss->pf_readv = win_readv;
return win2k_open( dvdcss, psz_device );
}
else if( !dvdcss->b_file )
{
_dvdcss_debug( dvdcss, "using ASPI for access" );
dvdcss->pf_seek = aspi_seek;
dvdcss->pf_read = aspi_read;
dvdcss->pf_readv = win_readv;
return aspi_open( dvdcss, psz_device );
}
else
#endif
{
_dvdcss_debug( dvdcss, "using libc for access" );
dvdcss->pf_seek = libc_seek;
dvdcss->pf_read = libc_read;
dvdcss->pf_readv = libc_readv;
return libc_open( dvdcss, psz_device );
}
}
#ifndef WIN32
int _dvdcss_raw_open ( dvdcss_t dvdcss, char const *psz_device )
{
dvdcss->i_raw_fd = open( psz_device, 0 );
if( dvdcss->i_raw_fd == -1 )
{
_dvdcss_error( dvdcss, "failed opening raw device, continuing" );
return -1;
}
else
{
dvdcss->i_read_fd = dvdcss->i_raw_fd;
}
return 0;
}
#endif
int _dvdcss_close ( dvdcss_t dvdcss )
{
#if defined( WIN32 )
if( dvdcss->b_file )
{
close( dvdcss->i_fd );
}
else if( WIN2K )
{
CloseHandle( (HANDLE) dvdcss->i_fd );
}
else /* ASPI */
{
struct w32_aspidev *fd = (struct w32_aspidev *) dvdcss->i_fd;
/* Unload aspi and free w32_aspidev structure */
FreeLibrary( (HMODULE) fd->hASPI );
free( (void*) dvdcss->i_fd );
}
/* Free readv temporary buffer */
if( dvdcss->p_readv_buffer )
{
free( dvdcss->p_readv_buffer );
dvdcss->p_readv_buffer = NULL;
dvdcss->i_readv_buf_size = 0;
}
return 0;
#else
close( dvdcss->i_fd );
if( dvdcss->i_raw_fd >= 0 )
{
close( dvdcss->i_raw_fd );
dvdcss->i_raw_fd = -1;
}
return 0;
#endif
}
/* Following functions are local */
/*****************************************************************************
* Open commands.
*****************************************************************************/
static int libc_open ( dvdcss_t dvdcss, char const *psz_device )
{
#if !defined( WIN32 )
dvdcss->i_fd = dvdcss->i_read_fd = open( psz_device, 0 );
#else
dvdcss->i_fd = dvdcss->i_read_fd = open( psz_device, O_BINARY );
#endif
if( dvdcss->i_fd == -1 )
{
_dvdcss_error( dvdcss, "failed opening device" );
return -1;
}
dvdcss->i_pos = 0;
return 0;
}
#if defined( WIN32 )
static int win2k_open ( dvdcss_t dvdcss, char const *psz_device )
{
char psz_dvd[7];
_snprintf( psz_dvd, 7, "\\\\.\\%c:", 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 priviledges 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) */
(HANDLE) dvdcss->i_fd =
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 )
(HANDLE) dvdcss->i_fd =
CreateFile( psz_dvd, GENERIC_READ, FILE_SHARE_READ,
NULL, OPEN_EXISTING,
FILE_FLAG_RANDOM_ACCESS, NULL );
if( (HANDLE) dvdcss->i_fd == INVALID_HANDLE_VALUE )
{
_dvdcss_error( dvdcss, "failed opening device" );
return -1;
}
dvdcss->i_pos = 0;
return 0;
}
static int aspi_open( dvdcss_t dvdcss, char const * psz_device )
{
HMODULE hASPI;
DWORD dwSupportInfo;
struct w32_aspidev *fd;
int i, j, i_hostadapters;
long (*lpGetSupport)( void );
long (*lpSendCommand)( void* );
char c_drive = psz_device[0];
/* load aspi and init w32_aspidev structure */
hASPI = LoadLibrary( "wnaspi32.dll" );
if( hASPI == NULL )
{
_dvdcss_error( dvdcss, "unable to load wnaspi32.dll" );
return -1;
}
(FARPROC) lpGetSupport = GetProcAddress( hASPI, "GetASPI32SupportInfo" );
(FARPROC) lpSendCommand = GetProcAddress( hASPI, "SendASPI32Command" );
if(lpGetSupport == NULL || lpSendCommand == NULL )
{
_dvdcss_error( dvdcss, "unable to get aspi function pointers" );
FreeLibrary( hASPI );
return -1;
}
dwSupportInfo = lpGetSupport();
if( HIBYTE( LOWORD ( dwSupportInfo ) ) == SS_NO_ADAPTERS )
{
_dvdcss_error( dvdcss, "no ASPI adapters found" );
FreeLibrary( hASPI );
return -1;
}
if( HIBYTE( LOWORD ( dwSupportInfo ) ) != SS_COMP )
{
_dvdcss_error( dvdcss, "unable to initalize aspi layer" );
FreeLibrary( hASPI );
return -1;
}
i_hostadapters = LOBYTE( LOWORD( dwSupportInfo ) );
if( i_hostadapters == 0 )
{
_dvdcss_error( dvdcss, "no ASPI adapters ready" );
FreeLibrary( hASPI );
return -1;
}
fd = malloc( sizeof( struct w32_aspidev ) );
if( fd == NULL )
{
_dvdcss_error( dvdcss, "not enough memory" );
FreeLibrary( hASPI );
return -1;
}
fd->i_blocks = 0;
fd->hASPI = (long) hASPI;
fd->lpSendCommand = lpSendCommand;
c_drive = c_drive > 'Z' ? c_drive - 'a' : c_drive - 'A';
for( i = 0; i < i_hostadapters; i++ )
{
for( j = 0; j < 15; j++ )
{
struct SRB_GetDiskInfo srbDiskInfo;
srbDiskInfo.SRB_Cmd = SC_GET_DISK_INFO;
srbDiskInfo.SRB_HaId = i;
srbDiskInfo.SRB_Flags = 0;
srbDiskInfo.SRB_Hdr_Rsvd = 0;
srbDiskInfo.SRB_Target = j;
srbDiskInfo.SRB_Lun = 0;
lpSendCommand( (void*) &srbDiskInfo );
if( (srbDiskInfo.SRB_Status == SS_COMP) &&
(srbDiskInfo.SRB_Int13HDriveInfo == c_drive) )
{
/* Make sure this is a cdrom device */
struct SRB_GDEVBlock srbGDEVBlock;
memset( &srbGDEVBlock, 0, sizeof(struct SRB_GDEVBlock) );
srbGDEVBlock.SRB_Cmd = SC_GET_DEV_TYPE;
srbGDEVBlock.SRB_HaId = i;
srbGDEVBlock.SRB_Target = j;
lpSendCommand( (void*) &srbGDEVBlock );
if( ( srbGDEVBlock.SRB_Status == SS_COMP ) &&
( srbGDEVBlock.SRB_DeviceType == DTYPE_CDROM ) )
{
fd->i_sid = MAKEWORD( i, j );
dvdcss->i_fd = (int) fd;
dvdcss->i_pos = 0;
return 0;
}
else
{
free( (void*) fd );
FreeLibrary( hASPI );
_dvdcss_error( dvdcss,"this is not a cdrom drive" );
return -1;
}
}
}
}
free( (void*) fd );
FreeLibrary( hASPI );
_dvdcss_error( dvdcss, "unable to get haid and target (aspi)" );
return -1;
}
#endif
/*****************************************************************************
* 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 = lseek( dvdcss->i_read_fd,
(off_t)i_blocks * (off_t)DVDCSS_BLOCK_SIZE, SEEK_SET );
if( i_seek < 0 )
{
_dvdcss_error( dvdcss, "seek error" );
dvdcss->i_pos = -1;
return i_seek;
}
dvdcss->i_pos = i_seek / DVDCSS_BLOCK_SIZE;
return dvdcss->i_pos;
}
#if defined( WIN32 )
static int win2k_seek( dvdcss_t dvdcss, int i_blocks )
{
LARGE_INTEGER li_seek;
#ifndef INVALID_SET_FILE_POINTER
# define INVALID_SET_FILE_POINTER ((DWORD)-1)
#endif
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;
}
static int aspi_seek( dvdcss_t dvdcss, int i_blocks )
{
int i_old_blocks;
char sz_buf[ DVDCSS_BLOCK_SIZE ];
struct w32_aspidev *fd = (struct w32_aspidev *) dvdcss->i_fd;
if( dvdcss->i_pos == i_blocks )
{
/* We are already in position */
return i_blocks;
}
i_old_blocks = fd->i_blocks;
fd->i_blocks = i_blocks;
if( aspi_read_internal( dvdcss->i_fd, sz_buf, 1 ) == -1 )
{
fd->i_blocks = i_old_blocks;
dvdcss->i_pos = -1;
return -1;
}
(fd->i_blocks)--;
dvdcss->i_pos = fd->i_blocks;
return dvdcss->i_pos;
}
#endif
/*****************************************************************************
* Read commands.
*****************************************************************************/
static int libc_read ( dvdcss_t dvdcss, void *p_buffer, int i_blocks )
{
off_t i_ret;
i_ret = read( dvdcss->i_read_fd, p_buffer,
(off_t)i_blocks * DVDCSS_BLOCK_SIZE );
if( i_ret < 0 )
{
_dvdcss_error( dvdcss, "read error" );
dvdcss->i_pos = -1;
return i_ret;
}
/* Handle partial reads */
if( i_ret != (off_t)i_blocks * DVDCSS_BLOCK_SIZE )
{
int i_seek;
dvdcss->i_pos = -1;
i_seek = libc_seek( dvdcss, i_ret / DVDCSS_BLOCK_SIZE );
if( i_seek < 0 )
{
return i_seek;
}
/* We have to return now so that i_pos isn't clobbered */
return i_ret / DVDCSS_BLOCK_SIZE;
}
dvdcss->i_pos += i_ret / DVDCSS_BLOCK_SIZE;
return i_ret / DVDCSS_BLOCK_SIZE;
}
#if defined( WIN32 )
static int win2k_read ( dvdcss_t dvdcss, void *p_buffer, int i_blocks )
{
int i_bytes;
if( !ReadFile( (HANDLE) dvdcss->i_fd, p_buffer,
i_blocks * DVDCSS_BLOCK_SIZE,
(LPDWORD)&i_bytes, NULL ) )
{
dvdcss->i_pos = -1;
return -1;
}
dvdcss->i_pos += i_bytes / DVDCSS_BLOCK_SIZE;
return i_bytes / DVDCSS_BLOCK_SIZE;
}
static int aspi_read ( dvdcss_t dvdcss, void *p_buffer, int i_blocks )
{
int i_read = aspi_read_internal( dvdcss->i_fd, p_buffer, i_blocks );
if( i_read < 0 )
{
dvdcss->i_pos = -1;
return i_read;
}
dvdcss->i_pos += i_read;
return i_read;
}
#endif
/*****************************************************************************
* Readv commands.
*****************************************************************************/
static int libc_readv ( dvdcss_t dvdcss, 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 postition is left
* unspecified after a failure */
dvdcss->i_pos = -1;
return -1;
}
i_total += i_bytes;
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 / DVDCSS_BLOCK_SIZE );
if( i_seek < 0 )
{
return i_seek;
}
/* We have to return now so that i_pos isn't clobbered */
return i_total / DVDCSS_BLOCK_SIZE;
}
}
dvdcss->i_pos += i_total / DVDCSS_BLOCK_SIZE;
return i_total / DVDCSS_BLOCK_SIZE;
#else
int i_read = readv( dvdcss->i_read_fd, 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;
#endif
}
#if defined( WIN32 )
/*****************************************************************************
* win_readv: vectored read using ReadFile for Win2K and ASPI for win9x
*****************************************************************************/
static int win_readv ( dvdcss_t dvdcss, struct iovec *p_iovec, int i_blocks )
{
int i_index;
int i_blocks_read, i_blocks_total = 0;
/* 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;
if( dvdcss->p_readv_buffer ) 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 )
{
_dvdcss_error( dvdcss, " failed (readv)" );
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;
i_blocks_total /= DVDCSS_BLOCK_SIZE;
if( WIN2K )
{
unsigned long int i_bytes;
if( !ReadFile( (HANDLE)dvdcss->i_fd, dvdcss->p_readv_buffer,
i_blocks_total * DVDCSS_BLOCK_SIZE, &i_bytes, NULL ) )
{
/* The read failed... too bad.
* As in the posix spec the file postition is left
* unspecified after a failure */
dvdcss->i_pos = -1;
return -1;
}
i_blocks_read = i_bytes / DVDCSS_BLOCK_SIZE;
}
else /* Win9x */
{
i_blocks_read = aspi_read_internal( dvdcss->i_fd,
dvdcss->p_readv_buffer,
i_blocks_total );
if( i_blocks_read < 0 )
{
/* See above */
dvdcss->i_pos = -1;
return -1;
}
}
/* 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;
}
static int aspi_read_internal( int i_fd, void *p_data, int i_blocks )
{
HANDLE hEvent;
struct SRB_ExecSCSICmd ssc;
struct w32_aspidev *fd = (struct w32_aspidev *) i_fd;
/* Create the transfer completion event */
hEvent = CreateEvent( NULL, TRUE, FALSE, NULL );
if( hEvent == NULL )
{
return -1;
}
memset( &ssc, 0, sizeof( ssc ) );
ssc.SRB_Cmd = SC_EXEC_SCSI_CMD;
ssc.SRB_Flags = SRB_DIR_IN | SRB_EVENT_NOTIFY;
ssc.SRB_HaId = LOBYTE( fd->i_sid );
ssc.SRB_Target = HIBYTE( fd->i_sid );
ssc.SRB_SenseLen = SENSE_LEN;
ssc.SRB_PostProc = (LPVOID) hEvent;
ssc.SRB_BufPointer = p_data;
ssc.SRB_CDBLen = 12;
ssc.CDBByte[0] = 0xA8; /* RAW */
ssc.CDBByte[2] = (UCHAR) (fd->i_blocks >> 24);
ssc.CDBByte[3] = (UCHAR) (fd->i_blocks >> 16) & 0xff;
ssc.CDBByte[4] = (UCHAR) (fd->i_blocks >> 8) & 0xff;
ssc.CDBByte[5] = (UCHAR) (fd->i_blocks) & 0xff;
/* We have to break down the reads into 64kb pieces (ASPI restriction) */
if( i_blocks > 32 )
{
ssc.SRB_BufLen = 32 * DVDCSS_BLOCK_SIZE;
ssc.CDBByte[9] = 32;
fd->i_blocks += 32;
/* Initiate transfer */
ResetEvent( hEvent );
fd->lpSendCommand( (void*) &ssc );
/* transfer the next 64kb (aspi_read_internal is called recursively)
* We need to check the status of the read on return */
if( aspi_read_internal( i_fd,
(uint8_t*) p_data + 32 * DVDCSS_BLOCK_SIZE,
i_blocks - 32) < 0 )
{
return -1;
}
}
else
{
/* This is the last transfer */
ssc.SRB_BufLen = i_blocks * DVDCSS_BLOCK_SIZE;
ssc.CDBByte[9] = (UCHAR) i_blocks;
fd->i_blocks += i_blocks;
/* Initiate transfer */
ResetEvent( hEvent );
fd->lpSendCommand( (void*) &ssc );
}
/* If the command has still not been processed, wait until it's finished */
if( ssc.SRB_Status == SS_PENDING )
{
WaitForSingleObject( hEvent, INFINITE );
}
CloseHandle( hEvent );
/* check that the transfer went as planned */
if( ssc.SRB_Status != SS_COMP )
{
return -1;
}
return i_blocks;
}
#endif