Mercurial > mplayer.hg
annotate libdvdcss/device.c @ 24015:b636b5de36f3
Add <application> tag around MEncoder
author | reimar |
---|---|
date | Wed, 08 Aug 2007 13:13:12 +0000 |
parents | 66c2f233ccff |
children | afa2cc0166be |
rev | line source |
---|---|
20613 | 1 /***************************************************************************** |
2 * device.h: DVD device access | |
3 ***************************************************************************** | |
4 * Copyright (C) 1998-2006 VideoLAN | |
5 * $Id$ | |
6 * | |
7 * Authors: Stéphane Borel <stef@via.ecp.fr> | |
8 * Sam Hocevar <sam@zoy.org> | |
9 * Håkan Hjort <d95hjort@dtek.chalmers.se> | |
10 * | |
11 * This program is free software; you can redistribute it and/or modify | |
12 * it under the terms of the GNU General Public License as published by | |
13 * the Free Software Foundation; either version 2 of the License, or | |
14 * (at your option) any later version. | |
15 * | |
16 * This program is distributed in the hope that it will be useful, | |
17 * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
18 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
19 * GNU General Public License for more details. | |
20 * | |
21 * You should have received a copy of the GNU General Public License | |
22 * along with this program; if not, write to the Free Software | |
23 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA. | |
24 *****************************************************************************/ | |
25 | |
26 /***************************************************************************** | |
27 * Preamble | |
28 *****************************************************************************/ | |
29 #include "config.h" | |
30 | |
31 #include <stdio.h> | |
32 #include <stdlib.h> | |
33 #include <string.h> | |
34 #ifdef HAVE_ERRNO_H | |
35 # include <errno.h> | |
36 #endif | |
37 #include <sys/types.h> | |
38 #include <sys/stat.h> | |
39 #ifdef HAVE_SYS_PARAM_H | |
40 # include <sys/param.h> | |
41 #endif | |
42 #include <fcntl.h> | |
43 | |
44 #ifdef HAVE_UNISTD_H | |
45 # include <unistd.h> | |
46 #endif | |
47 | |
48 #ifdef HAVE_LIMITS_H | |
49 # include <limits.h> | |
50 #endif | |
51 | |
52 #if defined( WIN32 ) && !defined( SYS_CYGWIN ) | |
53 # include <io.h> /* read() */ | |
54 #else | |
55 # include <sys/uio.h> /* struct iovec */ | |
56 #endif | |
57 | |
58 #ifdef DARWIN_DVD_IOCTL | |
59 # include <paths.h> | |
60 # include <CoreFoundation/CoreFoundation.h> | |
61 # include <IOKit/IOKitLib.h> | |
62 # include <IOKit/IOBSD.h> | |
63 # include <IOKit/storage/IOMedia.h> | |
64 # include <IOKit/storage/IOCDMedia.h> | |
65 # include <IOKit/storage/IODVDMedia.h> | |
66 #endif | |
67 | |
68 #include "dvdcss/dvdcss.h" | |
69 | |
70 #include "common.h" | |
71 #include "css.h" | |
72 #include "libdvdcss.h" | |
73 #include "ioctl.h" | |
74 #include "device.h" | |
75 | |
76 /***************************************************************************** | |
77 * Device reading prototypes | |
78 *****************************************************************************/ | |
79 static int libc_open ( dvdcss_t, char const * ); | |
80 static int libc_seek ( dvdcss_t, int ); | |
81 static int libc_read ( dvdcss_t, void *, int ); | |
82 static int libc_readv ( dvdcss_t, struct iovec *, int ); | |
83 | |
84 #ifdef WIN32 | |
85 static int win2k_open ( dvdcss_t, char const * ); | |
86 static int aspi_open ( dvdcss_t, char const * ); | |
87 static int win2k_seek ( dvdcss_t, int ); | |
88 static int aspi_seek ( dvdcss_t, int ); | |
89 static int win2k_read ( dvdcss_t, void *, int ); | |
90 static int aspi_read ( dvdcss_t, void *, int ); | |
91 static int win_readv ( dvdcss_t, struct iovec *, int ); | |
92 | |
93 static int aspi_read_internal ( int, void *, int ); | |
94 #endif | |
95 | |
96 int _dvdcss_use_ioctls( dvdcss_t dvdcss ) | |
97 { | |
98 #if defined( WIN32 ) | |
99 if( dvdcss->b_file ) | |
100 { | |
101 return 0; | |
102 } | |
103 | |
104 /* FIXME: implement this for Windows */ | |
105 if( WIN2K ) | |
106 { | |
107 return 1; | |
108 } | |
109 else | |
110 { | |
111 return 1; | |
112 } | |
113 #else | |
114 struct stat fileinfo; | |
115 int ret; | |
116 | |
117 ret = fstat( dvdcss->i_fd, &fileinfo ); | |
118 if( ret < 0 ) | |
119 { | |
120 return 1; /* What to do? Be conservative and try to use the ioctls */ | |
121 } | |
122 | |
123 /* Complete this list and check that we test for the right things | |
124 * (I've assumed for all OSs that 'r', (raw) device, are char devices | |
125 * and those that don't contain/use an 'r' in the name are block devices) | |
126 * | |
127 * Linux needs a block device | |
128 * Solaris needs a char device | |
129 * Darwin needs a char device | |
130 * OpenBSD needs a char device | |
131 * NetBSD needs a char device | |
132 * FreeBSD can use either the block or the char device | |
133 * BSD/OS can use either the block or the char device | |
134 */ | |
135 | |
136 /* Check if this is a block/char device */ | |
137 if( S_ISBLK( fileinfo.st_mode ) || | |
138 S_ISCHR( fileinfo.st_mode ) ) | |
139 { | |
140 return 1; | |
141 } | |
142 else | |
143 { | |
144 return 0; | |
145 } | |
146 #endif | |
147 } | |
148 | |
149 void _dvdcss_check ( dvdcss_t dvdcss ) | |
150 { | |
151 #if defined( WIN32 ) | |
152 DWORD drives; | |
153 int i; | |
154 #elif defined( DARWIN_DVD_IOCTL ) | |
155 io_object_t next_media; | |
156 mach_port_t master_port; | |
157 kern_return_t kern_result; | |
158 io_iterator_t media_iterator; | |
159 CFMutableDictionaryRef classes_to_match; | |
160 #else | |
161 char *ppsz_devices[] = { "/dev/dvd", "/dev/cdrom", "/dev/hdc", NULL }; | |
162 int i, i_fd; | |
163 #endif | |
164 | |
165 /* If the device name is non-null, return */ | |
166 if( dvdcss->psz_device[0] ) | |
167 { | |
168 return; | |
169 } | |
170 | |
171 #if defined( WIN32 ) | |
172 drives = GetLogicalDrives(); | |
173 | |
174 for( i = 0; drives; i++ ) | |
175 { | |
176 char psz_device[5]; | |
177 DWORD cur = 1 << i; | |
178 UINT i_ret; | |
179 | |
180 if( (drives & cur) == 0 ) | |
181 { | |
182 continue; | |
183 } | |
184 drives &= ~cur; | |
185 | |
186 sprintf( psz_device, "%c:\\", 'A' + i ); | |
187 i_ret = GetDriveType( psz_device ); | |
188 if( i_ret != DRIVE_CDROM ) | |
189 { | |
190 continue; | |
191 } | |
192 | |
193 /* Remove trailing backslash */ | |
194 psz_device[2] = '\0'; | |
195 | |
196 /* FIXME: we want to differenciate between CD and DVD drives | |
197 * using DeviceIoControl() */ | |
198 print_debug( dvdcss, "defaulting to drive `%s'", psz_device ); | |
199 free( dvdcss->psz_device ); | |
200 dvdcss->psz_device = strdup( psz_device ); | |
201 return; | |
202 } | |
203 #elif defined( DARWIN_DVD_IOCTL ) | |
204 | |
205 kern_result = IOMasterPort( MACH_PORT_NULL, &master_port ); | |
206 if( kern_result != KERN_SUCCESS ) | |
207 { | |
208 return; | |
209 } | |
210 | |
211 classes_to_match = IOServiceMatching( kIODVDMediaClass ); | |
212 if( classes_to_match == NULL ) | |
213 { | |
214 return; | |
215 } | |
216 | |
217 CFDictionarySetValue( classes_to_match, CFSTR( kIOMediaEjectableKey ), | |
218 kCFBooleanTrue ); | |
219 | |
220 kern_result = IOServiceGetMatchingServices( master_port, classes_to_match, | |
221 &media_iterator ); | |
222 if( kern_result != KERN_SUCCESS ) | |
223 { | |
224 return; | |
225 } | |
226 | |
227 next_media = IOIteratorNext( media_iterator ); | |
228 for( ; ; ) | |
229 { | |
230 char psz_buf[0x32]; | |
231 size_t i_pathlen; | |
232 CFTypeRef psz_path; | |
233 | |
234 next_media = IOIteratorNext( media_iterator ); | |
235 if( next_media == 0 ) | |
236 { | |
237 break; | |
238 } | |
239 | |
240 psz_path = IORegistryEntryCreateCFProperty( next_media, | |
241 CFSTR( kIOBSDNameKey ), | |
242 kCFAllocatorDefault, | |
243 0 ); | |
244 if( psz_path == NULL ) | |
245 { | |
246 IOObjectRelease( next_media ); | |
247 continue; | |
248 } | |
249 | |
250 snprintf( psz_buf, sizeof(psz_buf), "%s%c", _PATH_DEV, 'r' ); | |
251 i_pathlen = strlen( psz_buf ); | |
252 | |
253 if( CFStringGetCString( psz_path, | |
254 (char*)&psz_buf + i_pathlen, | |
255 sizeof(psz_buf) - i_pathlen, | |
256 kCFStringEncodingASCII ) ) | |
257 { | |
258 print_debug( dvdcss, "defaulting to drive `%s'", psz_buf ); | |
259 CFRelease( psz_path ); | |
260 IOObjectRelease( next_media ); | |
261 IOObjectRelease( media_iterator ); | |
262 free( dvdcss->psz_device ); | |
263 dvdcss->psz_device = strdup( psz_buf ); | |
264 return; | |
265 } | |
266 | |
267 CFRelease( psz_path ); | |
268 | |
269 IOObjectRelease( next_media ); | |
270 } | |
271 | |
272 IOObjectRelease( media_iterator ); | |
273 #else | |
274 for( i = 0; ppsz_devices[i]; i++ ) | |
275 { | |
276 i_fd = open( ppsz_devices[i], 0 ); | |
277 if( i_fd != -1 ) | |
278 { | |
279 print_debug( dvdcss, "defaulting to drive `%s'", ppsz_devices[i] ); | |
280 close( i_fd ); | |
281 free( dvdcss->psz_device ); | |
282 dvdcss->psz_device = strdup( ppsz_devices[i] ); | |
283 return; | |
284 } | |
285 } | |
286 #endif | |
287 | |
288 print_error( dvdcss, "could not find a suitable default drive" ); | |
289 } | |
290 | |
291 int _dvdcss_open ( dvdcss_t dvdcss ) | |
292 { | |
293 char const *psz_device = dvdcss->psz_device; | |
294 | |
295 print_debug( dvdcss, "opening target `%s'", psz_device ); | |
296 | |
297 #if defined( WIN32 ) | |
298 dvdcss->b_file = 1; | |
299 /* If device is "X:" or "X:\", we are not actually opening a file. */ | |
300 if (psz_device[0] && psz_device[1] == ':' && | |
301 (!psz_device[2] || (psz_device[2] == '\\' && !psz_device[3]))) | |
302 dvdcss->b_file = 0; | |
303 | |
304 /* Initialize readv temporary buffer */ | |
305 dvdcss->p_readv_buffer = NULL; | |
306 dvdcss->i_readv_buf_size = 0; | |
307 | |
308 if( !dvdcss->b_file && WIN2K ) | |
309 { | |
310 print_debug( dvdcss, "using Win2K API for access" ); | |
311 dvdcss->pf_seek = win2k_seek; | |
312 dvdcss->pf_read = win2k_read; | |
313 dvdcss->pf_readv = win_readv; | |
314 return win2k_open( dvdcss, psz_device ); | |
315 } | |
316 else if( !dvdcss->b_file ) | |
317 { | |
318 print_debug( dvdcss, "using ASPI for access" ); | |
319 dvdcss->pf_seek = aspi_seek; | |
320 dvdcss->pf_read = aspi_read; | |
321 dvdcss->pf_readv = win_readv; | |
322 return aspi_open( dvdcss, psz_device ); | |
323 } | |
324 else | |
325 #endif | |
326 { | |
327 print_debug( dvdcss, "using libc for access" ); | |
328 dvdcss->pf_seek = libc_seek; | |
329 dvdcss->pf_read = libc_read; | |
330 dvdcss->pf_readv = libc_readv; | |
331 return libc_open( dvdcss, psz_device ); | |
332 } | |
333 } | |
334 | |
335 #ifndef WIN32 | |
336 int _dvdcss_raw_open ( dvdcss_t dvdcss, char const *psz_device ) | |
337 { | |
338 dvdcss->i_raw_fd = open( psz_device, 0 ); | |
339 | |
340 if( dvdcss->i_raw_fd == -1 ) | |
341 { | |
342 print_debug( dvdcss, "cannot open %s (%s)", | |
343 psz_device, strerror(errno) ); | |
344 print_error( dvdcss, "failed to open raw device, but continuing" ); | |
345 return -1; | |
346 } | |
347 else | |
348 { | |
349 dvdcss->i_read_fd = dvdcss->i_raw_fd; | |
350 } | |
351 | |
352 return 0; | |
353 } | |
354 #endif | |
355 | |
356 int _dvdcss_close ( dvdcss_t dvdcss ) | |
357 { | |
358 #if defined( WIN32 ) | |
359 if( dvdcss->b_file ) | |
360 { | |
361 close( dvdcss->i_fd ); | |
362 } | |
363 else if( WIN2K ) | |
364 { | |
365 CloseHandle( (HANDLE) dvdcss->i_fd ); | |
366 } | |
367 else /* ASPI */ | |
368 { | |
369 struct w32_aspidev *fd = (struct w32_aspidev *) dvdcss->i_fd; | |
370 | |
371 /* Unload aspi and free w32_aspidev structure */ | |
372 FreeLibrary( (HMODULE) fd->hASPI ); | |
373 free( (void*) dvdcss->i_fd ); | |
374 } | |
375 | |
376 /* Free readv temporary buffer */ | |
377 if( dvdcss->p_readv_buffer ) | |
378 { | |
379 free( dvdcss->p_readv_buffer ); | |
380 dvdcss->p_readv_buffer = NULL; | |
381 dvdcss->i_readv_buf_size = 0; | |
382 } | |
383 | |
384 return 0; | |
385 #else | |
386 close( dvdcss->i_fd ); | |
387 | |
388 if( dvdcss->i_raw_fd >= 0 ) | |
389 { | |
390 close( dvdcss->i_raw_fd ); | |
391 dvdcss->i_raw_fd = -1; | |
392 } | |
393 | |
394 return 0; | |
395 #endif | |
396 } | |
397 | |
398 /* Following functions are local */ | |
399 | |
400 /***************************************************************************** | |
401 * Open commands. | |
402 *****************************************************************************/ | |
403 static int libc_open ( dvdcss_t dvdcss, char const *psz_device ) | |
404 { | |
405 #if !defined( WIN32 ) | |
406 dvdcss->i_fd = dvdcss->i_read_fd = open( psz_device, 0 ); | |
407 #else | |
408 dvdcss->i_fd = dvdcss->i_read_fd = open( psz_device, O_BINARY ); | |
409 #endif | |
410 | |
411 if( dvdcss->i_fd == -1 ) | |
412 { | |
413 print_debug( dvdcss, "cannot open %s (%s)", | |
414 psz_device, strerror(errno) ); | |
415 print_error( dvdcss, "failed to open device" ); | |
416 return -1; | |
417 } | |
418 | |
419 dvdcss->i_pos = 0; | |
420 | |
421 return 0; | |
422 } | |
423 | |
424 #if defined( WIN32 ) | |
425 static int win2k_open ( dvdcss_t dvdcss, char const *psz_device ) | |
426 { | |
427 char psz_dvd[7]; | |
428 _snprintf( psz_dvd, 7, "\\\\.\\%c:", psz_device[0] ); | |
429 | |
430 /* To work around an M$ bug in IOCTL_DVD_READ_STRUCTURE, we need read | |
431 * _and_ write access to the device (so we can make SCSI Pass Through | |
432 * Requests). Unfortunately this is only allowed if you have | |
433 * administrator priviledges so we allow for a fallback method with | |
434 * only read access to the device (in this case ioctl_ReadCopyright() | |
435 * won't send back the right result). | |
436 * (See Microsoft Q241374: Read and Write Access Required for SCSI | |
437 * Pass Through Requests) */ | |
438 dvdcss->i_fd = (int) | |
439 CreateFile( psz_dvd, GENERIC_READ | GENERIC_WRITE, | |
440 FILE_SHARE_READ | FILE_SHARE_WRITE, | |
441 NULL, OPEN_EXISTING, | |
442 FILE_FLAG_RANDOM_ACCESS, NULL ); | |
443 | |
444 if( (HANDLE) dvdcss->i_fd == INVALID_HANDLE_VALUE ) | |
445 dvdcss->i_fd = (int) | |
446 CreateFile( psz_dvd, GENERIC_READ, FILE_SHARE_READ, | |
447 NULL, OPEN_EXISTING, | |
448 FILE_FLAG_RANDOM_ACCESS, NULL ); | |
449 | |
450 if( (HANDLE) dvdcss->i_fd == INVALID_HANDLE_VALUE ) | |
451 { | |
452 print_error( dvdcss, "failed opening device" ); | |
453 return -1; | |
454 } | |
455 | |
456 dvdcss->i_pos = 0; | |
457 | |
458 return 0; | |
459 } | |
460 | |
461 static int aspi_open( dvdcss_t dvdcss, char const * psz_device ) | |
462 { | |
463 HMODULE hASPI; | |
464 DWORD dwSupportInfo; | |
465 struct w32_aspidev *fd; | |
466 int i, j, i_hostadapters; | |
467 GETASPI32SUPPORTINFO lpGetSupport; | |
468 SENDASPI32COMMAND lpSendCommand; | |
469 char c_drive = psz_device[0]; | |
470 | |
471 /* load aspi and init w32_aspidev structure */ | |
472 hASPI = LoadLibrary( "wnaspi32.dll" ); | |
473 if( hASPI == NULL ) | |
474 { | |
475 print_error( dvdcss, "unable to load wnaspi32.dll" ); | |
476 return -1; | |
477 } | |
478 | |
479 lpGetSupport = (GETASPI32SUPPORTINFO) GetProcAddress( hASPI, "GetASPI32SupportInfo" ); | |
480 lpSendCommand = (SENDASPI32COMMAND) GetProcAddress( hASPI, "SendASPI32Command" ); | |
481 | |
482 if(lpGetSupport == NULL || lpSendCommand == NULL ) | |
483 { | |
484 print_error( dvdcss, "unable to get aspi function pointers" ); | |
485 FreeLibrary( hASPI ); | |
486 return -1; | |
487 } | |
488 | |
489 dwSupportInfo = lpGetSupport(); | |
490 | |
491 if( HIBYTE( LOWORD ( dwSupportInfo ) ) == SS_NO_ADAPTERS ) | |
492 { | |
493 print_error( dvdcss, "no ASPI adapters found" ); | |
494 FreeLibrary( hASPI ); | |
495 return -1; | |
496 } | |
497 | |
498 if( HIBYTE( LOWORD ( dwSupportInfo ) ) != SS_COMP ) | |
499 { | |
500 print_error( dvdcss, "unable to initalize aspi layer" ); | |
501 FreeLibrary( hASPI ); | |
502 return -1; | |
503 } | |
504 | |
505 i_hostadapters = LOBYTE( LOWORD( dwSupportInfo ) ); | |
506 if( i_hostadapters == 0 ) | |
507 { | |
508 print_error( dvdcss, "no ASPI adapters ready" ); | |
509 FreeLibrary( hASPI ); | |
510 return -1; | |
511 } | |
512 | |
513 fd = malloc( sizeof( struct w32_aspidev ) ); | |
514 if( fd == NULL ) | |
515 { | |
516 print_error( dvdcss, "not enough memory" ); | |
517 FreeLibrary( hASPI ); | |
518 return -1; | |
519 } | |
520 | |
521 fd->i_blocks = 0; | |
522 fd->hASPI = (long) hASPI; | |
523 fd->lpSendCommand = lpSendCommand; | |
524 | |
525 c_drive = c_drive > 'Z' ? c_drive - 'a' : c_drive - 'A'; | |
526 | |
527 for( i = 0; i < i_hostadapters; i++ ) | |
528 { | |
529 for( j = 0; j < 15; j++ ) | |
530 { | |
531 struct SRB_GetDiskInfo srbDiskInfo; | |
532 | |
533 srbDiskInfo.SRB_Cmd = SC_GET_DISK_INFO; | |
534 srbDiskInfo.SRB_HaId = i; | |
535 srbDiskInfo.SRB_Flags = 0; | |
536 srbDiskInfo.SRB_Hdr_Rsvd = 0; | |
537 srbDiskInfo.SRB_Target = j; | |
538 srbDiskInfo.SRB_Lun = 0; | |
539 | |
540 lpSendCommand( (void*) &srbDiskInfo ); | |
541 | |
542 if( (srbDiskInfo.SRB_Status == SS_COMP) && | |
543 (srbDiskInfo.SRB_Int13HDriveInfo == c_drive) ) | |
544 { | |
545 /* Make sure this is a cdrom device */ | |
546 struct SRB_GDEVBlock srbGDEVBlock; | |
547 | |
548 memset( &srbGDEVBlock, 0, sizeof(struct SRB_GDEVBlock) ); | |
549 srbGDEVBlock.SRB_Cmd = SC_GET_DEV_TYPE; | |
550 srbGDEVBlock.SRB_HaId = i; | |
551 srbGDEVBlock.SRB_Target = j; | |
552 | |
553 lpSendCommand( (void*) &srbGDEVBlock ); | |
554 | |
555 if( ( srbGDEVBlock.SRB_Status == SS_COMP ) && | |
556 ( srbGDEVBlock.SRB_DeviceType == DTYPE_CDROM ) ) | |
557 { | |
558 fd->i_sid = MAKEWORD( i, j ); | |
559 dvdcss->i_fd = (int) fd; | |
560 dvdcss->i_pos = 0; | |
561 return 0; | |
562 } | |
563 else | |
564 { | |
565 free( (void*) fd ); | |
566 FreeLibrary( hASPI ); | |
567 print_error( dvdcss,"this is not a cdrom drive" ); | |
568 return -1; | |
569 } | |
570 } | |
571 } | |
572 } | |
573 | |
574 free( (void*) fd ); | |
575 FreeLibrary( hASPI ); | |
576 print_error( dvdcss, "unable to get haid and target (aspi)" ); | |
577 return -1; | |
578 } | |
579 #endif | |
580 | |
581 /***************************************************************************** | |
582 * Seek commands. | |
583 *****************************************************************************/ | |
584 static int libc_seek( dvdcss_t dvdcss, int i_blocks ) | |
585 { | |
586 off_t i_seek; | |
587 | |
588 if( dvdcss->i_pos == i_blocks ) | |
589 { | |
590 /* We are already in position */ | |
591 return i_blocks; | |
592 } | |
593 | |
594 i_seek = (off_t)i_blocks * (off_t)DVDCSS_BLOCK_SIZE; | |
20730
66c2f233ccff
Fix linking on Cygwin and remove nonsense lseek64 --> lseek indirection,
diego
parents:
20613
diff
changeset
|
595 i_seek = lseek( dvdcss->i_read_fd, i_seek, SEEK_SET ); |
20613 | 596 |
597 if( i_seek < 0 ) | |
598 { | |
599 print_error( dvdcss, "seek error" ); | |
600 dvdcss->i_pos = -1; | |
601 return i_seek; | |
602 } | |
603 | |
604 dvdcss->i_pos = i_seek / DVDCSS_BLOCK_SIZE; | |
605 | |
606 return dvdcss->i_pos; | |
607 } | |
608 | |
609 #if defined( WIN32 ) | |
610 static int win2k_seek( dvdcss_t dvdcss, int i_blocks ) | |
611 { | |
612 LARGE_INTEGER li_seek; | |
613 | |
614 #ifndef INVALID_SET_FILE_POINTER | |
615 # define INVALID_SET_FILE_POINTER ((DWORD)-1) | |
616 #endif | |
617 | |
618 if( dvdcss->i_pos == i_blocks ) | |
619 { | |
620 /* We are already in position */ | |
621 return i_blocks; | |
622 } | |
623 | |
624 li_seek.QuadPart = (LONGLONG)i_blocks * DVDCSS_BLOCK_SIZE; | |
625 | |
626 li_seek.LowPart = SetFilePointer( (HANDLE) dvdcss->i_fd, | |
627 li_seek.LowPart, | |
628 &li_seek.HighPart, FILE_BEGIN ); | |
629 if( (li_seek.LowPart == INVALID_SET_FILE_POINTER) | |
630 && GetLastError() != NO_ERROR) | |
631 { | |
632 dvdcss->i_pos = -1; | |
633 return -1; | |
634 } | |
635 | |
636 dvdcss->i_pos = li_seek.QuadPart / DVDCSS_BLOCK_SIZE; | |
637 | |
638 return dvdcss->i_pos; | |
639 } | |
640 | |
641 static int aspi_seek( dvdcss_t dvdcss, int i_blocks ) | |
642 { | |
643 int i_old_blocks; | |
644 char sz_buf[ DVDCSS_BLOCK_SIZE ]; | |
645 struct w32_aspidev *fd = (struct w32_aspidev *) dvdcss->i_fd; | |
646 | |
647 if( dvdcss->i_pos == i_blocks ) | |
648 { | |
649 /* We are already in position */ | |
650 return i_blocks; | |
651 } | |
652 | |
653 i_old_blocks = fd->i_blocks; | |
654 fd->i_blocks = i_blocks; | |
655 | |
656 if( aspi_read_internal( dvdcss->i_fd, sz_buf, 1 ) == -1 ) | |
657 { | |
658 fd->i_blocks = i_old_blocks; | |
659 dvdcss->i_pos = -1; | |
660 return -1; | |
661 } | |
662 | |
663 (fd->i_blocks)--; | |
664 | |
665 dvdcss->i_pos = fd->i_blocks; | |
666 | |
667 return dvdcss->i_pos; | |
668 } | |
669 #endif | |
670 | |
671 /***************************************************************************** | |
672 * Read commands. | |
673 *****************************************************************************/ | |
674 static int libc_read ( dvdcss_t dvdcss, void *p_buffer, int i_blocks ) | |
675 { | |
676 off_t i_size, i_ret; | |
677 | |
678 i_size = (off_t)i_blocks * (off_t)DVDCSS_BLOCK_SIZE; | |
679 i_ret = read( dvdcss->i_read_fd, p_buffer, i_size ); | |
680 | |
681 if( i_ret < 0 ) | |
682 { | |
683 print_error( dvdcss, "read error" ); | |
684 dvdcss->i_pos = -1; | |
685 return i_ret; | |
686 } | |
687 | |
688 /* Handle partial reads */ | |
689 if( i_ret != i_size ) | |
690 { | |
691 int i_seek; | |
692 | |
693 dvdcss->i_pos = -1; | |
694 i_seek = libc_seek( dvdcss, i_ret / DVDCSS_BLOCK_SIZE ); | |
695 if( i_seek < 0 ) | |
696 { | |
697 return i_seek; | |
698 } | |
699 | |
700 /* We have to return now so that i_pos isn't clobbered */ | |
701 return i_ret / DVDCSS_BLOCK_SIZE; | |
702 } | |
703 | |
704 dvdcss->i_pos += i_ret / DVDCSS_BLOCK_SIZE; | |
705 return i_ret / DVDCSS_BLOCK_SIZE; | |
706 } | |
707 | |
708 #if defined( WIN32 ) | |
709 static int win2k_read ( dvdcss_t dvdcss, void *p_buffer, int i_blocks ) | |
710 { | |
711 int i_bytes; | |
712 | |
713 if( !ReadFile( (HANDLE) dvdcss->i_fd, p_buffer, | |
714 i_blocks * DVDCSS_BLOCK_SIZE, | |
715 (LPDWORD)&i_bytes, NULL ) ) | |
716 { | |
717 dvdcss->i_pos = -1; | |
718 return -1; | |
719 } | |
720 | |
721 dvdcss->i_pos += i_bytes / DVDCSS_BLOCK_SIZE; | |
722 return i_bytes / DVDCSS_BLOCK_SIZE; | |
723 } | |
724 | |
725 static int aspi_read ( dvdcss_t dvdcss, void *p_buffer, int i_blocks ) | |
726 { | |
727 int i_read = aspi_read_internal( dvdcss->i_fd, p_buffer, i_blocks ); | |
728 | |
729 if( i_read < 0 ) | |
730 { | |
731 dvdcss->i_pos = -1; | |
732 return i_read; | |
733 } | |
734 | |
735 dvdcss->i_pos += i_read; | |
736 return i_read; | |
737 } | |
738 #endif | |
739 | |
740 /***************************************************************************** | |
741 * Readv commands. | |
742 *****************************************************************************/ | |
743 static int libc_readv ( dvdcss_t dvdcss, struct iovec *p_iovec, int i_blocks ) | |
744 { | |
745 #if defined( WIN32 ) | |
746 int i_index, i_len, i_total = 0; | |
747 unsigned char *p_base; | |
748 int i_bytes; | |
749 | |
750 for( i_index = i_blocks; | |
751 i_index; | |
752 i_index--, p_iovec++ ) | |
753 { | |
754 i_len = p_iovec->iov_len; | |
755 p_base = p_iovec->iov_base; | |
756 | |
757 if( i_len <= 0 ) | |
758 { | |
759 continue; | |
760 } | |
761 | |
762 i_bytes = read( dvdcss->i_fd, p_base, i_len ); | |
763 | |
764 if( i_bytes < 0 ) | |
765 { | |
766 /* One of the reads failed, too bad. | |
767 * We won't even bother returning the reads that went ok, | |
768 * and as in the posix spec the file postition is left | |
769 * unspecified after a failure */ | |
770 dvdcss->i_pos = -1; | |
771 return -1; | |
772 } | |
773 | |
774 i_total += i_bytes; | |
775 | |
776 if( i_bytes != i_len ) | |
777 { | |
778 /* We reached the end of the file or a signal interrupted | |
779 * the read. Return a partial read. */ | |
780 int i_seek; | |
781 | |
782 dvdcss->i_pos = -1; | |
783 i_seek = libc_seek( dvdcss, i_total / DVDCSS_BLOCK_SIZE ); | |
784 if( i_seek < 0 ) | |
785 { | |
786 return i_seek; | |
787 } | |
788 | |
789 /* We have to return now so that i_pos isn't clobbered */ | |
790 return i_total / DVDCSS_BLOCK_SIZE; | |
791 } | |
792 } | |
793 | |
794 dvdcss->i_pos += i_total / DVDCSS_BLOCK_SIZE; | |
795 return i_total / DVDCSS_BLOCK_SIZE; | |
796 #else | |
797 int i_read = readv( dvdcss->i_read_fd, p_iovec, i_blocks ); | |
798 | |
799 if( i_read < 0 ) | |
800 { | |
801 dvdcss->i_pos = -1; | |
802 return i_read; | |
803 } | |
804 | |
805 dvdcss->i_pos += i_read / DVDCSS_BLOCK_SIZE; | |
806 return i_read / DVDCSS_BLOCK_SIZE; | |
807 #endif | |
808 } | |
809 | |
810 #if defined( WIN32 ) | |
811 /***************************************************************************** | |
812 * win_readv: vectored read using ReadFile for Win2K and ASPI for win9x | |
813 *****************************************************************************/ | |
814 static int win_readv ( dvdcss_t dvdcss, struct iovec *p_iovec, int i_blocks ) | |
815 { | |
816 int i_index; | |
817 int i_blocks_read, i_blocks_total = 0; | |
818 | |
819 /* Check the size of the readv temp buffer, just in case we need to | |
820 * realloc something bigger */ | |
821 if( dvdcss->i_readv_buf_size < i_blocks * DVDCSS_BLOCK_SIZE ) | |
822 { | |
823 dvdcss->i_readv_buf_size = i_blocks * DVDCSS_BLOCK_SIZE; | |
824 | |
825 if( dvdcss->p_readv_buffer ) free( dvdcss->p_readv_buffer ); | |
826 | |
827 /* Allocate a buffer which will be used as a temporary storage | |
828 * for readv */ | |
829 dvdcss->p_readv_buffer = malloc( dvdcss->i_readv_buf_size ); | |
830 if( !dvdcss->p_readv_buffer ) | |
831 { | |
832 print_error( dvdcss, " failed (readv)" ); | |
833 dvdcss->i_pos = -1; | |
834 return -1; | |
835 } | |
836 } | |
837 | |
838 for( i_index = i_blocks; i_index; i_index-- ) | |
839 { | |
840 i_blocks_total += p_iovec[i_index-1].iov_len; | |
841 } | |
842 | |
843 if( i_blocks_total <= 0 ) return 0; | |
844 | |
845 i_blocks_total /= DVDCSS_BLOCK_SIZE; | |
846 | |
847 if( WIN2K ) | |
848 { | |
849 unsigned long int i_bytes; | |
850 if( !ReadFile( (HANDLE)dvdcss->i_fd, dvdcss->p_readv_buffer, | |
851 i_blocks_total * DVDCSS_BLOCK_SIZE, &i_bytes, NULL ) ) | |
852 { | |
853 /* The read failed... too bad. | |
854 * As in the posix spec the file postition is left | |
855 * unspecified after a failure */ | |
856 dvdcss->i_pos = -1; | |
857 return -1; | |
858 } | |
859 i_blocks_read = i_bytes / DVDCSS_BLOCK_SIZE; | |
860 } | |
861 else /* Win9x */ | |
862 { | |
863 i_blocks_read = aspi_read_internal( dvdcss->i_fd, | |
864 dvdcss->p_readv_buffer, | |
865 i_blocks_total ); | |
866 if( i_blocks_read < 0 ) | |
867 { | |
868 /* See above */ | |
869 dvdcss->i_pos = -1; | |
870 return -1; | |
871 } | |
872 } | |
873 | |
874 /* We just have to copy the content of the temp buffer into the iovecs */ | |
875 for( i_index = 0, i_blocks_total = i_blocks_read; | |
876 i_blocks_total > 0; | |
877 i_index++ ) | |
878 { | |
879 memcpy( p_iovec[i_index].iov_base, | |
880 dvdcss->p_readv_buffer + (i_blocks_read - i_blocks_total) | |
881 * DVDCSS_BLOCK_SIZE, | |
882 p_iovec[i_index].iov_len ); | |
883 /* if we read less blocks than asked, we'll just end up copying | |
884 * garbage, this isn't an issue as we return the number of | |
885 * blocks actually read */ | |
886 i_blocks_total -= ( p_iovec[i_index].iov_len / DVDCSS_BLOCK_SIZE ); | |
887 } | |
888 | |
889 dvdcss->i_pos += i_blocks_read; | |
890 return i_blocks_read; | |
891 } | |
892 | |
893 static int aspi_read_internal( int i_fd, void *p_data, int i_blocks ) | |
894 { | |
895 HANDLE hEvent; | |
896 struct SRB_ExecSCSICmd ssc; | |
897 struct w32_aspidev *fd = (struct w32_aspidev *) i_fd; | |
898 | |
899 /* Create the transfer completion event */ | |
900 hEvent = CreateEvent( NULL, TRUE, FALSE, NULL ); | |
901 if( hEvent == NULL ) | |
902 { | |
903 return -1; | |
904 } | |
905 | |
906 memset( &ssc, 0, sizeof( ssc ) ); | |
907 | |
908 ssc.SRB_Cmd = SC_EXEC_SCSI_CMD; | |
909 ssc.SRB_Flags = SRB_DIR_IN | SRB_EVENT_NOTIFY; | |
910 ssc.SRB_HaId = LOBYTE( fd->i_sid ); | |
911 ssc.SRB_Target = HIBYTE( fd->i_sid ); | |
912 ssc.SRB_SenseLen = SENSE_LEN; | |
913 | |
914 ssc.SRB_PostProc = (LPVOID) hEvent; | |
915 ssc.SRB_BufPointer = p_data; | |
916 ssc.SRB_CDBLen = 12; | |
917 | |
918 ssc.CDBByte[0] = 0xA8; /* RAW */ | |
919 ssc.CDBByte[2] = (UCHAR) (fd->i_blocks >> 24); | |
920 ssc.CDBByte[3] = (UCHAR) (fd->i_blocks >> 16) & 0xff; | |
921 ssc.CDBByte[4] = (UCHAR) (fd->i_blocks >> 8) & 0xff; | |
922 ssc.CDBByte[5] = (UCHAR) (fd->i_blocks) & 0xff; | |
923 | |
924 /* We have to break down the reads into 64kb pieces (ASPI restriction) */ | |
925 if( i_blocks > 32 ) | |
926 { | |
927 ssc.SRB_BufLen = 32 * DVDCSS_BLOCK_SIZE; | |
928 ssc.CDBByte[9] = 32; | |
929 fd->i_blocks += 32; | |
930 | |
931 /* Initiate transfer */ | |
932 ResetEvent( hEvent ); | |
933 fd->lpSendCommand( (void*) &ssc ); | |
934 | |
935 /* transfer the next 64kb (aspi_read_internal is called recursively) | |
936 * We need to check the status of the read on return */ | |
937 if( aspi_read_internal( i_fd, | |
938 (uint8_t*) p_data + 32 * DVDCSS_BLOCK_SIZE, | |
939 i_blocks - 32) < 0 ) | |
940 { | |
941 return -1; | |
942 } | |
943 } | |
944 else | |
945 { | |
946 /* This is the last transfer */ | |
947 ssc.SRB_BufLen = i_blocks * DVDCSS_BLOCK_SIZE; | |
948 ssc.CDBByte[9] = (UCHAR) i_blocks; | |
949 fd->i_blocks += i_blocks; | |
950 | |
951 /* Initiate transfer */ | |
952 ResetEvent( hEvent ); | |
953 fd->lpSendCommand( (void*) &ssc ); | |
954 | |
955 } | |
956 | |
957 /* If the command has still not been processed, wait until it's finished */ | |
958 if( ssc.SRB_Status == SS_PENDING ) | |
959 { | |
960 WaitForSingleObject( hEvent, INFINITE ); | |
961 } | |
962 CloseHandle( hEvent ); | |
963 | |
964 /* check that the transfer went as planned */ | |
965 if( ssc.SRB_Status != SS_COMP ) | |
966 { | |
967 return -1; | |
968 } | |
969 | |
970 return i_blocks; | |
971 } | |
972 #endif | |
973 |