Mercurial > mplayer.hg
view libmpdvdkit2/device.c @ 10252:d275152390ee
I've found some time to implement the encoding support for the new
DivX API. Now it's possible to play and encode movies with the
latest DivX release.
One thing that doesn't work is the new Video Buffer Verifier (VBV)
multipass encoding. The encoder segfaults. Maybe it just isn't
supported with the standard profile of the released binary encoder.
Andreas Hess <jaska@gmx.net>
author | arpi |
---|---|
date | Fri, 06 Jun 2003 19:57:37 +0000 |
parents | f0f0f176d298 |
children | 002e7a679a05 |
line wrap: on
line source
/***************************************************************************** * 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