ref: 18b755d525681ce85e14b5b953687938d1812f1a
dir: /src/device.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" #ifdef __PLAN9__ # include <u.h> # include <libc.h> #else # include <limits.h> # include <stdio.h> # include <stdlib.h> # include <string.h> #endif #ifdef HAVE_ERRNO_H # include <errno.h> #endif #ifndef __PLAN9__ # include <sys/types.h> # include <sys/stat.h> #endif #ifdef HAVE_SYS_PARAM_H # include <sys/param.h> #endif #ifndef __PLAN9__ # include <fcntl.h> #endif #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; #elif defined( __PLAN9__ ) 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; return win2k_open( dvdcss, psz_device ); } 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; return os2_open( dvdcss, psz_device ); } 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; return libc_open( dvdcss, psz_device ); } } int dvdcss_close_device ( dvdcss_t dvdcss ) { if( dvdcss->p_stream ) { return 0; } #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 ) */