1 /*****************************************************************************
2 * libdvdcss.c: DVD reading library.
3 *****************************************************************************
4 * Copyright (C) 1998-2001 VideoLAN
5 * $Id: libdvdcss.c,v 1.13 2001/07/30 18:56:35 gbazin Exp $
7 * Authors: Stéphane Borel <stef@via.ecp.fr>
8 * Samuel Hocevar <sam@zoy.org>
10 * This program is free software; you can redistribute it and/or modify
11 * it under the terms of the GNU General Public License as published by
12 * the Free Software Foundation; either version 2 of the License, or
13 * (at your option) any later version.
15 * This program is distributed in the hope that it will be useful,
16 * but WITHOUT ANY WARRANTY; without even the implied warranty of
17 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 * GNU General Public License for more details.
20 * You should have received a copy of the GNU General Public License
21 * along with this program; if not, write to the Free Software
22 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
23 *****************************************************************************/
25 /*****************************************************************************
27 *****************************************************************************/
32 #include <sys/types.h>
41 # include <io.h> /* read() */
43 # include <sys/uio.h> /* struct iovec */
50 # include "input_iovec.h"
53 #include "videolan/dvdcss.h"
54 #include "libdvdcss.h"
57 /*****************************************************************************
59 *****************************************************************************/
60 static int _dvdcss_open ( dvdcss_handle, char *psz_target );
61 static int _dvdcss_close ( dvdcss_handle );
62 static int _dvdcss_seek ( dvdcss_handle, int i_blocks );
63 static int _dvdcss_read ( dvdcss_handle, void *p_buffer, int i_blocks );
64 static int _dvdcss_readv ( dvdcss_handle, struct iovec *p_iovec, int i_blocks );
66 /*****************************************************************************
67 * Local prototypes, win32 specific
68 *****************************************************************************/
70 static int _win32_dvdcss_readv ( int i_fd, struct iovec *p_iovec,
71 int i_num_buffers, char *p_tmp_buffer );
72 static int _win32_dvdcss_aopen ( char c_drive, dvdcss_handle dvdcss );
73 static int _win32_dvdcss_aclose ( int i_fd );
74 static int _win32_dvdcss_aseek ( int i_fd, int i_blocks, int i_method );
75 static int _win32_dvdcss_aread ( int i_fd, void *p_data, int i_blocks );
78 /*****************************************************************************
79 * dvdcss_open: initialize library, open a DVD device, crack CSS key
80 *****************************************************************************/
81 extern dvdcss_handle dvdcss_open ( char *psz_target, int i_flags )
87 /* Allocate the library structure */
88 dvdcss = malloc( sizeof( struct dvdcss_s ) );
91 if( ! (i_flags & DVDCSS_INIT_QUIET) )
93 DVDCSS_ERROR( "could not initialize library" );
99 /* Initialize structure */
100 dvdcss->p_titles = NULL;
101 dvdcss->b_debug = i_flags & DVDCSS_INIT_DEBUG;
102 dvdcss->b_errors = !(i_flags & DVDCSS_INIT_QUIET);
103 dvdcss->psz_error = "no error";
105 i_ret = _dvdcss_open( dvdcss, psz_target );
112 i_ret = CSSTest( dvdcss );
115 _dvdcss_error( dvdcss, "css test failed" );
116 _dvdcss_close( dvdcss );
121 dvdcss->b_encrypted = i_ret;
123 /* If drive is encrypted, crack its key */
124 if( dvdcss->b_encrypted )
126 i_ret = CSSInit( dvdcss );
130 _dvdcss_close( dvdcss );
139 /*****************************************************************************
140 * dvdcss_error: return the last libdvdcss error message
141 *****************************************************************************/
142 extern char * dvdcss_error ( dvdcss_handle dvdcss )
144 return dvdcss->psz_error;
147 /*****************************************************************************
148 * dvdcss_seek: seek into the device
149 *****************************************************************************/
150 extern int dvdcss_seek ( dvdcss_handle dvdcss, int i_blocks )
152 return _dvdcss_seek( dvdcss, i_blocks );
155 /*****************************************************************************
156 * dvdcss_title: crack the current title key if needed
157 *****************************************************************************/
158 extern int dvdcss_title ( dvdcss_handle dvdcss, int i_block )
160 dvd_title_t *p_title;
164 if( ! dvdcss->b_encrypted )
169 //fprintf( stderr, "looking for a key for offset %i\n", i_block );
171 /* Check if we've already cracked this key */
172 p_title = dvdcss->p_titles;
173 while( p_title != NULL
174 && p_title->p_next != NULL
175 && p_title->p_next->i_startlb <= i_block )
177 p_title = p_title->p_next;
181 && p_title->i_startlb == i_block )
183 /* We've already cracked this key, nothing to do */
187 /* Crack CSS title key for current VTS */
188 i_ret = CSSGetKey( dvdcss, i_block, p_key );
192 _dvdcss_error( dvdcss, "fatal error in vts css key" );
197 _dvdcss_error( dvdcss, "decryption unavailable" );
201 //fprintf( stderr, "cracked key is %.2x %.2x %.2x %.2x %.2x\n",
202 // p_key[0], p_key[1], p_key[2], p_key[3], p_key[4] );
204 /* Add key to keytable if it isn't empty */
205 if( p_key[0] | p_key[1] | p_key[2] | p_key[3] | p_key[4] )
207 dvd_title_t *p_newtitle;
209 /* Find our spot in the list */
211 p_title = dvdcss->p_titles;
212 while( p_title != NULL
213 && p_title->i_startlb < i_block )
215 p_newtitle = p_title;
216 p_title = p_title->p_next;
219 /* Save the found title */
220 p_title = p_newtitle;
222 /* Write in the new title and its key */
223 p_newtitle = malloc( sizeof( dvd_title_t ) );
224 p_newtitle->i_startlb = i_block;
225 memcpy( p_newtitle->p_key, p_key, KEY_SIZE );
227 /* Link the new title, either at the beginning or inside the list */
228 if( p_title == NULL )
230 dvdcss->p_titles = p_newtitle;
231 p_newtitle->p_next = NULL;
235 p_newtitle->p_next = p_title->p_next;
236 p_title->p_next = p_newtitle;
243 /*****************************************************************************
244 * dvdcss_read: read data from the device, decrypt if requested
245 *****************************************************************************/
246 extern int dvdcss_read ( dvdcss_handle dvdcss, void *p_buffer,
250 dvd_title_t *p_title;
253 i_ret = _dvdcss_read( dvdcss, p_buffer, i_blocks );
256 || !dvdcss->b_encrypted
257 || !(i_flags & DVDCSS_READ_DECRYPT) )
263 p_title = dvdcss->p_titles;
264 while( p_title != NULL
266 && p_title->p_next->i_startlb <= dvdcss->i_seekpos )
268 p_title = p_title->p_next;
271 if( p_title == NULL )
273 /* no css key found to use, so no decryption to do */
277 /* Decrypt the blocks we managed to read */
278 for( i_index = i_ret; i_index; i_index-- )
280 CSSDescrambleSector( p_title->p_key, p_buffer );
281 ((u8*)p_buffer)[0x14] &= 0x8f;
282 (u8*)p_buffer += DVDCSS_BLOCK_SIZE;
288 /*****************************************************************************
289 * dvdcss_readv: read data to an iovec structure, decrypt if reaquested
290 *****************************************************************************/
291 extern int dvdcss_readv ( dvdcss_handle dvdcss, void *p_iovec,
295 #define P_IOVEC ((struct iovec*)p_iovec)
296 dvd_title_t *p_title;
301 i_ret = _dvdcss_readv( dvdcss, P_IOVEC, i_blocks );
304 || !dvdcss->b_encrypted
305 || !(i_flags & DVDCSS_READ_DECRYPT) )
311 p_title = dvdcss->p_titles;
312 while( p_title != NULL
313 && p_title->p_next != NULL
314 && p_title->p_next->i_startlb <= dvdcss->i_seekpos )
316 p_title = p_title->p_next;
319 if( p_title == NULL )
321 /* no css key found to use, so no decryption to do */
325 /* Initialize loop for decryption */
326 iov_base = P_IOVEC->iov_base;
327 iov_len = P_IOVEC->iov_len;
329 /* Decrypt the blocks we managed to read */
330 for( i_index = i_ret; i_index; i_index-- )
332 /* Check that iov_len is a multiple of 2048 */
333 if( iov_len & 0x7ff )
338 while( iov_len == 0 )
341 iov_base = P_IOVEC->iov_base;
342 iov_len = P_IOVEC->iov_len;
345 CSSDescrambleSector( p_title->p_key, iov_base );
346 ((u8*)iov_base)[0x14] &= 0x8f;
348 (u8*)iov_base += DVDCSS_BLOCK_SIZE;
349 (u8*)iov_len -= DVDCSS_BLOCK_SIZE;
356 /*****************************************************************************
357 * dvdcss_close: close the DVD device and clean up the library
358 *****************************************************************************/
359 extern int dvdcss_close ( dvdcss_handle dvdcss )
361 dvd_title_t *p_title;
364 /* Free our list of keys */
365 p_title = dvdcss->p_titles;
368 dvd_title_t *p_tmptitle = p_title->p_next;
370 p_title = p_tmptitle;
373 i_ret = _dvdcss_close( dvdcss );
385 /* Following functions are local */
387 static int _dvdcss_open ( dvdcss_handle dvdcss, char *psz_target )
393 _snprintf( psz_dvd, 7, "\\\\.\\%c:", psz_target[0] );
394 (HANDLE) dvdcss->i_fd =
395 CreateFile( psz_dvd, GENERIC_READ | GENERIC_WRITE,
396 FILE_SHARE_READ | FILE_SHARE_WRITE,
397 NULL, OPEN_EXISTING, 0, NULL );
398 if( (HANDLE) dvdcss->i_fd == INVALID_HANDLE_VALUE )
400 _dvdcss_error( dvdcss, "failed opening device" );
406 dvdcss->i_fd = _win32_dvdcss_aopen( psz_target[0], dvdcss );
407 if( dvdcss->i_fd == -1 )
409 _dvdcss_error( dvdcss, "failed opening device" );
414 /* initialise readv temporary buffer */
415 dvdcss->p_readv_buffer = NULL;
416 dvdcss->i_readv_buf_size = 0;
419 dvdcss->i_fd = open( psz_target, 0 );
421 if( dvdcss->i_fd == -1 )
423 _dvdcss_error( dvdcss, "failed opening device" );
432 static int _dvdcss_close ( dvdcss_handle dvdcss )
437 CloseHandle( (HANDLE) dvdcss->i_fd );
441 _win32_dvdcss_aclose( dvdcss->i_fd );
444 /* Free readv temporary buffer */
445 if( dvdcss->p_readv_buffer )
447 free( dvdcss->p_readv_buffer );
448 dvdcss->p_readv_buffer = NULL;
449 dvdcss->i_readv_buf_size = 0;
453 close( dvdcss->i_fd );
460 static int _dvdcss_seek ( dvdcss_handle dvdcss, int i_blocks )
463 dvdcss->i_seekpos = i_blocks;
467 LARGE_INTEGER li_read;
469 #ifndef INVALID_SET_FILE_POINTER
470 #define INVALID_SET_FILE_POINTER ((DWORD)-1)
473 li_read.QuadPart = (LONGLONG)i_blocks * DVDCSS_BLOCK_SIZE;
475 li_read.LowPart = SetFilePointer( (HANDLE) dvdcss->i_fd,
477 &li_read.HighPart, FILE_BEGIN );
478 if( (li_read.LowPart == INVALID_SET_FILE_POINTER)
479 && GetLastError() != NO_ERROR)
481 li_read.QuadPart = -DVDCSS_BLOCK_SIZE;
484 li_read.QuadPart /= DVDCSS_BLOCK_SIZE;
485 return (int)li_read.QuadPart;
489 return ( _win32_dvdcss_aseek( dvdcss->i_fd, i_blocks, SEEK_SET ) );
494 dvdcss->i_seekpos = i_blocks;
496 i_read = lseek( dvdcss->i_fd,
497 (off_t)i_blocks * (off_t)DVDCSS_BLOCK_SIZE, SEEK_SET );
499 return i_read / DVDCSS_BLOCK_SIZE;
504 static int _dvdcss_read ( dvdcss_handle dvdcss, void *p_buffer, int i_blocks )
511 if( !ReadFile( (HANDLE) dvdcss->i_fd, p_buffer,
512 i_blocks * DVDCSS_BLOCK_SIZE,
513 (LPDWORD)&i_bytes, NULL ) )
517 return i_bytes / DVDCSS_BLOCK_SIZE;
521 return _win32_dvdcss_aread( dvdcss->i_fd, p_buffer, i_blocks );
527 i_bytes = read( dvdcss->i_fd, p_buffer,
528 (size_t)i_blocks * DVDCSS_BLOCK_SIZE );
529 return i_bytes / DVDCSS_BLOCK_SIZE;
534 static int _dvdcss_readv ( dvdcss_handle dvdcss, struct iovec *p_iovec,
540 /* Check the size of the readv temp buffer, just in case we need to
541 * realloc something bigger */
542 if( dvdcss->i_readv_buf_size < i_blocks * DVDCSS_BLOCK_SIZE )
544 dvdcss->i_readv_buf_size = i_blocks * DVDCSS_BLOCK_SIZE;
546 if( dvdcss->p_readv_buffer ) free( dvdcss->p_readv_buffer );
548 /* Allocate a buffer which will be used as a temporary storage
550 dvdcss->p_readv_buffer = malloc( dvdcss->i_readv_buf_size );
551 if( !dvdcss->p_readv_buffer )
553 _dvdcss_error( dvdcss, " failed (readv)" );
558 i_read = _win32_dvdcss_readv( dvdcss->i_fd, p_iovec, i_blocks,
559 dvdcss->p_readv_buffer );
563 i_read = readv( dvdcss->i_fd, p_iovec, i_blocks );
564 return i_read / DVDCSS_BLOCK_SIZE;
572 /*****************************************************************************
573 * _win32_dvdcss_readv: vectored read using ReadFile for Win2K and
574 * _win32_dvdcss_aread for win9x
575 *****************************************************************************/
576 static int _win32_dvdcss_readv( int i_fd, struct iovec *p_iovec,
577 int i_num_buffers, char *p_tmp_buffer )
580 int i_blocks, i_blocks_total = 0;
582 for( i_index = i_num_buffers; i_index; i_index-- )
584 i_blocks_total += p_iovec[i_index-1].iov_len;
587 if( i_blocks_total <= 0 ) return 0;
589 i_blocks_total /= DVDCSS_BLOCK_SIZE;
593 unsigned long int i_bytes;
594 if( !ReadFile( (HANDLE)i_fd, p_tmp_buffer,
595 i_blocks_total * DVDCSS_BLOCK_SIZE, &i_bytes, NULL ) )
598 /* The read failed... too bad.
599 As in the posix spec the file postition is left
600 unspecified after a failure */
602 i_blocks = i_bytes / DVDCSS_BLOCK_SIZE;
606 i_blocks = _win32_dvdcss_aread( i_fd, p_tmp_buffer, i_blocks_total );
609 return -1; /* idem */
613 /* We just have to copy the content of the temp buffer into the iovecs */
615 i_blocks_total = i_blocks;
616 while( i_blocks_total > 0 )
618 memcpy( p_iovec[i_index].iov_base,
619 &p_tmp_buffer[(i_blocks - i_blocks_total) * DVDCSS_BLOCK_SIZE],
620 p_iovec[i_index].iov_len );
621 /* if we read less blocks than asked, we'll just end up copying
622 garbage, this isn't an issue as we return the number of
623 blocks actually read */
624 i_blocks_total -= ( p_iovec[i_index].iov_len / DVDCSS_BLOCK_SIZE );
631 /*****************************************************************************
632 * _win32_dvdcss_aopen: open dvd drive (load aspi and init w32_aspidev
634 *****************************************************************************/
635 static int _win32_dvdcss_aopen( char c_drive, dvdcss_handle dvdcss )
639 struct w32_aspidev *fd;
640 int i, j, i_hostadapters;
641 long (*lpGetSupport)( void );
642 long (*lpSendCommand)( void* );
644 hASPI = LoadLibrary( "wnaspi32.dll" );
647 _dvdcss_error( dvdcss, "unable to load wnaspi32.dll" );
651 (FARPROC) lpGetSupport = GetProcAddress( hASPI, "GetASPI32SupportInfo" );
652 (FARPROC) lpSendCommand = GetProcAddress( hASPI, "SendASPI32Command" );
654 if(lpGetSupport == NULL || lpSendCommand == NULL )
656 _dvdcss_debug( dvdcss, "unable to get aspi function pointers" );
657 FreeLibrary( hASPI );
661 dwSupportInfo = lpGetSupport();
663 if( HIBYTE( LOWORD ( dwSupportInfo ) ) == SS_NO_ADAPTERS )
665 _dvdcss_debug( dvdcss, "no host adapters found (aspi)" );
666 FreeLibrary( hASPI );
670 if( HIBYTE( LOWORD ( dwSupportInfo ) ) != SS_COMP )
672 _dvdcss_error( dvdcss, "unable to initalize aspi layer" );
673 FreeLibrary( hASPI );
677 i_hostadapters = LOBYTE( LOWORD( dwSupportInfo ) );
678 if( i_hostadapters == 0 )
680 FreeLibrary( hASPI );
684 fd = malloc( sizeof( struct w32_aspidev ) );
687 FreeLibrary( hASPI );
692 fd->hASPI = (long) hASPI;
693 fd->lpSendCommand = lpSendCommand;
695 c_drive = c_drive > 'Z' ? c_drive - 'a' : c_drive - 'A';
697 for( i = 0; i < i_hostadapters; i++ )
699 for( j = 0; j < 15; j++ )
701 struct SRB_GetDiskInfo srbDiskInfo;
703 srbDiskInfo.SRB_Cmd = SC_GET_DISK_INFO;
704 srbDiskInfo.SRB_HaId = i;
705 srbDiskInfo.SRB_Flags = 0;
706 srbDiskInfo.SRB_Hdr_Rsvd = 0;
707 srbDiskInfo.SRB_Target = j;
708 srbDiskInfo.SRB_Lun = 0;
710 lpSendCommand( (void*) &srbDiskInfo );
712 if( (srbDiskInfo.SRB_Status == SS_COMP) &&
713 (srbDiskInfo.SRB_Int13HDriveInfo == c_drive) )
715 fd->i_sid = MAKEWORD( i, j );
722 FreeLibrary( hASPI );
723 _dvdcss_debug( dvdcss, "unable to get haid and target (aspi)" );
727 /*****************************************************************************
728 * _win32_dvdcss_aclose: close dvd drive (unload aspi and free w32_aspidev
730 *****************************************************************************/
731 static int _win32_dvdcss_aclose( int i_fd )
733 struct w32_aspidev *fd = (struct w32_aspidev *) i_fd;
735 FreeLibrary( (HMODULE) fd->hASPI );
736 free( (void*) i_fd );
741 /*****************************************************************************
742 * _win32_dvdcss_aseek: aspi version of _dvdcss_seek
744 * returns the number of blocks read.
745 *****************************************************************************/
746 static int _win32_dvdcss_aseek( int i_fd, int i_blocks, int i_method )
749 char sz_buf[ DVDCSS_BLOCK_SIZE ];
750 struct w32_aspidev *fd = (struct w32_aspidev *) i_fd;
752 i_old_blocks = fd->i_blocks;
753 fd->i_blocks = i_blocks;
755 if( _win32_dvdcss_aread( i_fd, sz_buf, 1 ) == -1 )
757 fd->i_blocks = i_old_blocks;
766 /*****************************************************************************
767 * _win32_dvdcss_aread: aspi version of _dvdcss_read
769 * returns the number of blocks read.
770 *****************************************************************************/
771 static int _win32_dvdcss_aread( int i_fd, void *p_data, int i_blocks )
774 struct SRB_ExecSCSICmd ssc;
775 struct w32_aspidev *fd = (struct w32_aspidev *) i_fd;
777 /* Create the transfer completion event */
778 hEvent = CreateEvent( NULL, TRUE, FALSE, NULL );
784 memset( &ssc, 0, sizeof( ssc ) );
786 ssc.SRB_Cmd = SC_EXEC_SCSI_CMD;
787 ssc.SRB_Flags = SRB_DIR_IN | SRB_EVENT_NOTIFY;
788 ssc.SRB_HaId = LOBYTE( fd->i_sid );
789 ssc.SRB_Target = HIBYTE( fd->i_sid );
790 ssc.SRB_SenseLen = SENSE_LEN;
792 ssc.SRB_PostProc = (LPVOID) hEvent;
793 ssc.SRB_BufPointer = p_data;
796 ssc.CDBByte[0] = 0xA8; /* RAW */
797 ssc.CDBByte[2] = (UCHAR) (fd->i_blocks >> 24);
798 ssc.CDBByte[3] = (UCHAR) (fd->i_blocks >> 16) & 0xff;
799 ssc.CDBByte[4] = (UCHAR) (fd->i_blocks >> 8) & 0xff;
800 ssc.CDBByte[5] = (UCHAR) (fd->i_blocks) & 0xff;
802 /* We have to break down the reads into 64kb pieces (ASPI restriction) */
805 ssc.SRB_BufLen = 32 * DVDCSS_BLOCK_SIZE;
809 /* Initiate transfer */
810 ResetEvent( hEvent );
811 fd->lpSendCommand( (void*) &ssc );
813 /* transfer the next 64kb (_win32_dvdcss_aread is called recursively)
814 * We need to check the status of the read on return */
815 if( _win32_dvdcss_aread( i_fd, (u8*) p_data + 32 * DVDCSS_BLOCK_SIZE,
823 /* This is the last transfer */
824 ssc.SRB_BufLen = i_blocks * DVDCSS_BLOCK_SIZE;
825 ssc.CDBByte[9] = (UCHAR) i_blocks;
826 fd->i_blocks += i_blocks;
828 /* Initiate transfer */
829 ResetEvent( hEvent );
830 fd->lpSendCommand( (void*) &ssc );
834 /* If the command has still not been processed, wait until it's finished */
835 if( ssc.SRB_Status == SS_PENDING )
837 WaitForSingleObject( hEvent, INFINITE );
839 CloseHandle( hEvent );
841 /* check that the transfer went as planned */
842 if( ssc.SRB_Status != SS_COMP )