?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
Make GetVolumeInformation default to reporting NTFS for non-FAT
[wine/multimedia.git] / dlls / kernel / volume.c
blobf9b65bec111db18e554b665493baec876fa78153
1 /*
2 * Volume management functions
3 *
4 * Copyright 1993 Erik Bos
5 * Copyright 1996, 2004 Alexandre Julliard
6 * Copyright 1999 Petr Tomasek
7 * Copyright 2000 Andreas Mohr
8 * Copyright 2003 Eric Pouech
9 *
10 * This library is free software; you can redistribute it and/or
11 * modify it under the terms of the GNU Lesser General Public
12 * License as published by the Free Software Foundation; either
13 * version 2.1 of the License, or (at your option) any later version.
14 *
15 * This library 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 GNU
18 * Lesser General Public License for more details.
19 *
20 * You should have received a copy of the GNU Lesser General Public
21 * License along with this library; if not, write to the Free Software
22 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
23 */
24
25 #include "config.h"
26 #include "wine/port.h"
27
28 #include <stdarg.h>
29 #include <stdlib.h>
30 #include <stdio.h>
31
32 #include "windef.h"
33 #include "winbase.h"
34 #include "winreg.h"
35 #include "winnls.h"
36 #include "winternl.h"
37 #include "ntstatus.h"
38 #include "winioctl.h"
39 #include "ntddstor.h"
40 #include "ntddcdrm.h"
41 #include "kernel_private.h"
42 #include "wine/library.h"
43 #include "wine/unicode.h"
44 #include "wine/debug.h"
45
46 WINE_DEFAULT_DEBUG_CHANNEL(volume);
47
48 #define SUPERBLOCK_SIZE 2048
49
50 #define CDFRAMES_PERSEC 75
51 #define CDFRAMES_PERMIN (CDFRAMES_PERSEC * 60)
52 #define FRAME_OF_ADDR(a) ((a)[1] * CDFRAMES_PERMIN + (a)[2] * CDFRAMES_PERSEC + (a)[3])
53 #define FRAME_OF_TOC(toc, idx) FRAME_OF_ADDR((toc)->TrackData[(idx) - (toc)->FirstTrack].Address)
54
55 #define GETWORD(buf,off) MAKEWORD(buf[(off)],buf[(off+1)])
56 #define GETLONG(buf,off) MAKELONG(GETWORD(buf,off),GETWORD(buf,off+2))
57
58 enum fs_type
59 {
60 FS_ERROR, /* error accessing the device */
61 FS_UNKNOWN, /* unknown file system */
62 FS_FAT1216,
63 FS_FAT32,
64 FS_ISO9660
65 };
66
67 static const WCHAR drive_types[][8] =
68 {
69 { 0 }, /* DRIVE_UNKNOWN */
70 { 0 }, /* DRIVE_NO_ROOT_DIR */
71 {'f','l','o','p','p','y',0}, /* DRIVE_REMOVABLE */
72 {'h','d',0}, /* DRIVE_FIXED */
73 {'n','e','t','w','o','r','k',0}, /* DRIVE_REMOTE */
74 {'c','d','r','o','m',0}, /* DRIVE_CDROM */
75 {'r','a','m','d','i','s','k',0} /* DRIVE_RAMDISK */
76 };
77
78 /* read a Unix symlink; returned buffer must be freed by caller */
79 static char *read_symlink( const char *path )
80 {
81 char *buffer;
82 int ret, size = 128;
83
84 for (;;)
85 {
86 if (!(buffer = HeapAlloc( GetProcessHeap(), 0, size )))
87 {
88 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
89 return 0;
90 }
91 ret = readlink( path, buffer, size );
92 if (ret == -1)
93 {
94 FILE_SetDosError();
95 HeapFree( GetProcessHeap(), 0, buffer );
96 return 0;
97 }
98 if (ret != size)
99 {
100 buffer[ret] = 0;
101 return buffer;
102 }
103 HeapFree( GetProcessHeap(), 0, buffer );
104 size *= 2;
105 }
106 }
107
108 /* get the path of a dos device symlink in the $WINEPREFIX/dosdevices directory */
109 static char *get_dos_device_path( LPCWSTR name )
110 {
111 const char *config_dir = wine_get_config_dir();
112 char *buffer, *dev;
113 int i;
114
115 if (!(buffer = HeapAlloc( GetProcessHeap(), 0,
116 strlen(config_dir) + sizeof("/dosdevices/") + 5 )))
117 {
118 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
119 return NULL;
120 }
121 strcpy( buffer, config_dir );
122 strcat( buffer, "/dosdevices/" );
123 dev = buffer + strlen(buffer);
124 /* no codepage conversion, DOS device names are ASCII anyway */
125 for (i = 0; i < 5; i++)
126 if (!(dev[i] = (char)tolowerW(name[i]))) break;
127 dev[5] = 0;
128 return buffer;
129 }
130
131
132 /* open a handle to a device root */
133 static BOOL open_device_root( LPCWSTR root, HANDLE *handle )
134 {
135 static const WCHAR default_rootW[] = {'\\',0};
136 UNICODE_STRING nt_name;
137 OBJECT_ATTRIBUTES attr;
138 IO_STATUS_BLOCK io;
139 NTSTATUS status;
140
141 if (!root) root = default_rootW;
142 if (!RtlDosPathNameToNtPathName_U( root, &nt_name, NULL, NULL ))
143 {
144 SetLastError( ERROR_PATH_NOT_FOUND );
145 return FALSE;
146 }
147 attr.Length = sizeof(attr);
148 attr.RootDirectory = 0;
149 attr.Attributes = OBJ_CASE_INSENSITIVE;
150 attr.ObjectName = &nt_name;
151 attr.SecurityDescriptor = NULL;
152 attr.SecurityQualityOfService = NULL;
153
154 status = NtOpenFile( handle, 0, &attr, &io, 0,
155 FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT );
156 RtlFreeUnicodeString( &nt_name );
157 if (status != STATUS_SUCCESS)
158 {
159 SetLastError( RtlNtStatusToDosError(status) );
160 return FALSE;
161 }
162 return TRUE;
163 }
164
165
166 /* fetch the type of a drive from the registry */
167 static UINT get_registry_drive_type( const WCHAR *root )
168 {
169 static const WCHAR drive_types_keyW[] = {'M','a','c','h','i','n','e','\\',
170 'S','o','f','t','w','a','r','e','\\',
171 'W','i','n','e','\\',
172 'D','r','i','v','e','s',0 };
173 OBJECT_ATTRIBUTES attr;
174 UNICODE_STRING nameW;
175 HKEY hkey;
176 DWORD dummy;
177 UINT ret = DRIVE_UNKNOWN;
178 char tmp[32 + sizeof(KEY_VALUE_PARTIAL_INFORMATION)];
179 WCHAR driveW[] = {'A',':',0};
180
181 attr.Length = sizeof(attr);
182 attr.RootDirectory = 0;
183 attr.ObjectName = &nameW;
184 attr.Attributes = 0;
185 attr.SecurityDescriptor = NULL;
186 attr.SecurityQualityOfService = NULL;
187 RtlInitUnicodeString( &nameW, drive_types_keyW );
188 if (NtOpenKey( &hkey, KEY_ALL_ACCESS, &attr ) != STATUS_SUCCESS) return DRIVE_UNKNOWN;
189
190 if (root) driveW[0] = root[0];
191 else
192 {
193 WCHAR path[MAX_PATH];
194 GetCurrentDirectoryW( MAX_PATH, path );
195 driveW[0] = path[0];
196 }
197
198 RtlInitUnicodeString( &nameW, driveW );
199 if (!NtQueryValueKey( hkey, &nameW, KeyValuePartialInformation, tmp, sizeof(tmp), &dummy ))
200 {
201 int i;
202 WCHAR *data = (WCHAR *)((KEY_VALUE_PARTIAL_INFORMATION *)tmp)->Data;
203
204 for (i = 0; i < sizeof(drive_types)/sizeof(drive_types[0]); i++)
205 {
206 if (!strcmpiW( data, drive_types[i] ))
207 {
208 ret = i;
209 break;
210 }
211 }
212 }
213 NtClose( hkey );
214 return ret;
215 }
216
217
218 /* create symlinks for the DOS drives; helper for VOLUME_CreateDevices */
219 static int create_drives( int devices_only )
220 {
221 static const WCHAR PathW[] = {'P','a','t','h',0};
222 static const WCHAR DeviceW[] = {'D','e','v','i','c','e',0};
223 WCHAR driveW[] = {'M','a','c','h','i','n','e','\\','S','o','f','t','w','a','r','e','\\',
224 'W','i','n','e','\\','W','i','n','e','\\',
225 'C','o','n','f','i','g','\\','D','r','i','v','e',' ','A',0};
226 OBJECT_ATTRIBUTES attr;
227 UNICODE_STRING nameW;
228 char tmp[1024*sizeof(WCHAR) + sizeof(KEY_VALUE_PARTIAL_INFORMATION)];
229 char dest[1024];
230 WCHAR *p, name[3];
231 HKEY hkey;
232 DWORD dummy;
233 int i, count = 0;
234
235 attr.Length = sizeof(attr);
236 attr.RootDirectory = 0;
237 attr.ObjectName = &nameW;
238 attr.Attributes = 0;
239 attr.SecurityDescriptor = NULL;
240 attr.SecurityQualityOfService = NULL;
241
242 /* create symlinks for the drive roots */
243
244 if (!devices_only) for (i = 0; i < 26; i++)
245 {
246 RtlInitUnicodeString( &nameW, driveW );
247 nameW.Buffer[(nameW.Length / sizeof(WCHAR)) - 1] = 'A' + i;
248 if (NtOpenKey( &hkey, KEY_ALL_ACCESS, &attr ) != STATUS_SUCCESS) continue;
249
250 RtlInitUnicodeString( &nameW, PathW );
251 if (!NtQueryValueKey( hkey, &nameW, KeyValuePartialInformation, tmp, sizeof(tmp), &dummy ))
252 {
253 WCHAR path[1024];
254 WCHAR *data = (WCHAR *)((KEY_VALUE_PARTIAL_INFORMATION *)tmp)->Data;
255 ExpandEnvironmentStringsW( data, path, sizeof(path)/sizeof(WCHAR) );
256
257 p = path + strlenW(path) - 1;
258 while ((p > path) && (*p == '/')) *p-- = '\0';
259
260 name[0] = 'a' + i;
261 name[1] = ':';
262 name[2] = 0;
263
264 if (path[0] != '/')
265 {
266 /* relative paths are relative to config dir */
267 memmove( path + 3, path, (strlenW(path) + 1) * sizeof(WCHAR) );
268 path[0] = '.';
269 path[1] = '.';
270 path[2] = '/';
271 }
272 if (DefineDosDeviceW( DDD_RAW_TARGET_PATH, name, path ))
273 {
274 WideCharToMultiByte(CP_UNIXCP, 0, path, -1, dest, sizeof(dest), NULL, NULL);
275 MESSAGE( "Created symlink %s/dosdevices/%c: -> %s\n",
276 wine_get_config_dir(), 'a' + i, dest );
277 count++;
278 }
279 }
280 NtClose( hkey );
281 }
282
283 /* create symlinks for the drive devices */
284
285 for (i = 0; i < 26; i++)
286 {
287 RtlInitUnicodeString( &nameW, driveW );
288 nameW.Buffer[(nameW.Length / sizeof(WCHAR)) - 1] = 'A' + i;
289 if (NtOpenKey( &hkey, KEY_ALL_ACCESS, &attr ) != STATUS_SUCCESS) continue;
290
291 RtlInitUnicodeString( &nameW, DeviceW );
292 if (!NtQueryValueKey( hkey, &nameW, KeyValuePartialInformation, tmp, sizeof(tmp), &dummy ))
293 {
294 char *path, *p;
295 WCHAR devname[] = {'A',':',':',0 };
296 WCHAR *data = (WCHAR *)((KEY_VALUE_PARTIAL_INFORMATION *)tmp)->Data;
297 WideCharToMultiByte(CP_UNIXCP, 0, data, -1, dest, sizeof(dest), NULL, NULL);
298 path = get_dos_device_path( devname );
299 p = path + strlen(path);
300 p[-3] = 'a' + i;
301 if (!symlink( dest, path ))
302 {
303 MESSAGE( "Created symlink %s/dosdevices/%c:: -> %s\n",
304 wine_get_config_dir(), 'a' + i, dest );
305 count++;
306 }
307 HeapFree( GetProcessHeap(), 0, path );
308 }
309 NtClose( hkey );
310 }
311
312 return count;
313 }
314
315
316 /***********************************************************************
317 * VOLUME_CreateDevices
318 *
319 * Create the device files for the new device naming scheme.
320 * Should go away after a transition period.
321 */
322 void VOLUME_CreateDevices(void)
323 {
324 const char *config_dir = wine_get_config_dir();
325 char *buffer;
326 int i, count = 0;
327
328 if (!(buffer = HeapAlloc( GetProcessHeap(), 0,
329 strlen(config_dir) + sizeof("/dosdevices/a::") )))
330 return;
331
332 strcpy( buffer, config_dir );
333 strcat( buffer, "/dosdevices" );
334
335 if (!mkdir( buffer, 0777 )) /* we created it, so now create the devices */
336 {
337 HKEY hkey;
338 DWORD dummy;
339 OBJECT_ATTRIBUTES attr;
340 UNICODE_STRING nameW;
341 WCHAR *p, *devnameW;
342 char tmp[128];
343 WCHAR com[5] = {'C','O','M','1',0};
344 WCHAR lpt[5] = {'L','P','T','1',0};
345
346 static const WCHAR serialportsW[] = {'M','a','c','h','i','n','e','\\',
347 'S','o','f','t','w','a','r','e','\\',
348 'W','i','n','e','\\','W','i','n','e','\\',
349 'C','o','n','f','i','g','\\',
350 'S','e','r','i','a','l','P','o','r','t','s',0};
351 static const WCHAR parallelportsW[] = {'M','a','c','h','i','n','e','\\',
352 'S','o','f','t','w','a','r','e','\\',
353 'W','i','n','e','\\','W','i','n','e','\\',
354 'C','o','n','f','i','g','\\',
355 'P','a','r','a','l','l','e','l','P','o','r','t','s',0};
356
357 attr.Length = sizeof(attr);
358 attr.RootDirectory = 0;
359 attr.ObjectName = &nameW;
360 attr.Attributes = 0;
361 attr.SecurityDescriptor = NULL;
362 attr.SecurityQualityOfService = NULL;
363 RtlInitUnicodeString( &nameW, serialportsW );
364
365 if (!NtOpenKey( &hkey, KEY_ALL_ACCESS, &attr ))
366 {
367 RtlInitUnicodeString( &nameW, com );
368 for (i = 1; i <= 9; i++)
369 {
370 com[3] = '0' + i;
371 if (!NtQueryValueKey( hkey, &nameW, KeyValuePartialInformation,
372 tmp, sizeof(tmp), &dummy ))
373 {
374 devnameW = (WCHAR *)((KEY_VALUE_PARTIAL_INFORMATION *)tmp)->Data;
375 if ((p = strchrW( devnameW, ',' ))) *p = 0;
376 if (DefineDosDeviceW( DDD_RAW_TARGET_PATH, com, devnameW ))
377 {
378 char devname[32];
379 WideCharToMultiByte(CP_UNIXCP, 0, devnameW, -1,
380 devname, sizeof(devname), NULL, NULL);
381 MESSAGE( "Created symlink %s/dosdevices/com%d -> %s\n", config_dir, i, devname );
382 count++;
383 }
384 }
385 }
386 NtClose( hkey );
387 }
388
389 RtlInitUnicodeString( &nameW, parallelportsW );
390 if (!NtOpenKey( &hkey, KEY_ALL_ACCESS, &attr ))
391 {
392 RtlInitUnicodeString( &nameW, lpt );
393 for (i = 1; i <= 9; i++)
394 {
395 lpt[3] = '0' + i;
396 if (!NtQueryValueKey( hkey, &nameW, KeyValuePartialInformation,
397 tmp, sizeof(tmp), &dummy ))
398 {
399 devnameW = (WCHAR *)((KEY_VALUE_PARTIAL_INFORMATION *)tmp)->Data;
400 if ((p = strchrW( devnameW, ',' ))) *p = 0;
401 if (DefineDosDeviceW( DDD_RAW_TARGET_PATH, lpt, devnameW ))
402 {
403 char devname[32];
404 WideCharToMultiByte(CP_UNIXCP, 0, devnameW, -1,
405 devname, sizeof(devname), NULL, NULL);
406 MESSAGE( "Created symlink %s/dosdevices/lpt%d -> %s\n", config_dir, i, devname );
407 count++;
408 }
409 }
410 }
411 NtClose( hkey );
412 }
413 count += create_drives( FALSE );
414 }
415 else
416 {
417 struct stat st;
418 int i;
419
420 /* it is possible that the serial/parallel devices have been created but */
421 /* not the drives; check for at least one drive symlink to catch that case */
422 strcat( buffer, "/a:" );
423 for (i = 0; i < 26; i++)
424 {
425 buffer[strlen(buffer)-2] = 'a' + i;
426 if (!lstat( buffer, &st )) break;
427 }
428 if (i == 26) count += create_drives( FALSE );
429 else
430 {
431 strcat( buffer, ":" );
432 for (i = 0; i < 26; i++)
433 {
434 buffer[strlen(buffer)-3] = 'a' + i;
435 if (!lstat( buffer, &st )) break;
436 }
437 if (i == 26) count += create_drives( TRUE );
438 }
439 }
440
441 if (count)
442 MESSAGE( "\nYou can now remove the [SerialPorts], [ParallelPorts], and [Drive] sections\n"
443 "in your configuration file, they are replaced by the above symlinks.\n\n" );
444
445 HeapFree( GetProcessHeap(), 0, buffer );
446 }
447
448
449 /******************************************************************
450 * VOLUME_FindCdRomDataBestVoldesc
451 */
452 static DWORD VOLUME_FindCdRomDataBestVoldesc( HANDLE handle )
453 {
454 BYTE cur_vd_type, max_vd_type = 0;
455 BYTE buffer[16];
456 DWORD size, offs, best_offs = 0, extra_offs = 0;
457
458 for (offs = 0x8000; offs <= 0x9800; offs += 0x800)
459 {
460 /* if 'CDROM' occurs at position 8, this is a pre-iso9660 cd, and
461 * the volume label is displaced forward by 8
462 */
463 if (SetFilePointer( handle, offs, NULL, FILE_BEGIN ) != offs) break;
464 if (!ReadFile( handle, buffer, sizeof(buffer), &size, NULL )) break;
465 if (size != sizeof(buffer)) break;
466 /* check for non-ISO9660 signature */
467 if (!memcmp( buffer + 11, "ROM", 3 )) extra_offs = 8;
468 cur_vd_type = buffer[extra_offs];
469 if (cur_vd_type == 0xff) /* voldesc set terminator */
470 break;
471 if (cur_vd_type > max_vd_type)
472 {
473 max_vd_type = cur_vd_type;
474 best_offs = offs + extra_offs;
475 }
476 }
477 return best_offs;
478 }
479
480
481 /***********************************************************************
482 * VOLUME_ReadFATSuperblock
483 */
484 static enum fs_type VOLUME_ReadFATSuperblock( HANDLE handle, BYTE *buff )
485 {
486 DWORD size;
487
488 /* try a fixed disk, with a FAT partition */
489 if (SetFilePointer( handle, 0, NULL, FILE_BEGIN ) != 0 ||
490 !ReadFile( handle, buff, SUPERBLOCK_SIZE, &size, NULL ) ||
491 size != SUPERBLOCK_SIZE)
492 return FS_ERROR;
493
494 if (buff[0] == 0xE9 || (buff[0] == 0xEB && buff[2] == 0x90))
495 {
496 /* guess which type of FAT we have */
497 unsigned int sz, nsect, nclust;
498 sz = GETWORD(buff, 0x16);
499 if (!sz) sz = GETLONG(buff, 0x24);
500 nsect = GETWORD(buff, 0x13);
501 if (!nsect) nsect = GETLONG(buff, 0x20);
502 nsect -= GETWORD(buff, 0x0e) + buff[0x10] * sz +
503 (GETWORD(buff, 0x11) * 32 + (GETWORD(buff, 0x0b) - 1)) / GETWORD(buff, 0x0b);
504 nclust = nsect / buff[0x0d];
505
506 if (nclust < 65525)
507 {
508 if (buff[0x26] == 0x29 && !memcmp(buff+0x36, "FAT", 3))
509 {
510 /* FIXME: do really all FAT have their name beginning with
511 * "FAT" ? (At least FAT12, FAT16 and FAT32 have :)
512 */
513 return FS_FAT1216;
514 }
515 }
516 else if (!memcmp(buff+0x52, "FAT", 3)) return FS_FAT32;
517 }
518 return FS_UNKNOWN;
519 }
520
521
522 /***********************************************************************
523 * VOLUME_ReadCDSuperblock
524 */
525 static enum fs_type VOLUME_ReadCDSuperblock( HANDLE handle, BYTE *buff )
526 {
527 DWORD size, offs = VOLUME_FindCdRomDataBestVoldesc( handle );
528
529 if (!offs) return FS_UNKNOWN;
530
531 if (SetFilePointer( handle, offs, NULL, FILE_BEGIN ) != offs ||
532 !ReadFile( handle, buff, SUPERBLOCK_SIZE, &size, NULL ) ||
533 size != SUPERBLOCK_SIZE)
534 return FS_ERROR;
535
536 /* check for iso9660 present */
537 if (!memcmp(&buff[1], "CD001", 5)) return FS_ISO9660;
538 return FS_UNKNOWN;
539 }
540
541
542 /**************************************************************************
543 * VOLUME_GetSuperblockLabel
544 */
545 static void VOLUME_GetSuperblockLabel( enum fs_type type, const BYTE *superblock,
546 WCHAR *label, DWORD len )
547 {
548 const BYTE *label_ptr = NULL;
549 DWORD label_len;
550
551 switch(type)
552 {
553 case FS_ERROR:
554 case FS_UNKNOWN:
555 label_len = 0;
556 break;
557 case FS_FAT1216:
558 label_ptr = superblock + 0x2b;
559 label_len = 11;
560 break;
561 case FS_FAT32:
562 label_ptr = superblock + 0x47;
563 label_len = 11;
564 break;
565 case FS_ISO9660:
566 {
567 BYTE ver = superblock[0x5a];
568
569 if (superblock[0x58] == 0x25 && superblock[0x59] == 0x2f && /* Unicode ID */
570 ((ver == 0x40) || (ver == 0x43) || (ver == 0x45)))
571 { /* yippee, unicode */
572 int i;
573
574 if (len > 17) len = 17;
575 for (i = 0; i < len-1; i++)
576 label[i] = (superblock[40+2*i] << 8) | superblock[41+2*i];
577 label[i] = 0;
578 while (i && label[i-1] == ' ') label[--i] = 0;
579 return;
580 }
581 label_ptr = superblock + 40;
582 label_len = 32;
583 break;
584 }
585 }
586 if (label_len) RtlMultiByteToUnicodeN( label, (len-1) * sizeof(WCHAR),
587 &label_len, label_ptr, label_len );
588 label_len /= sizeof(WCHAR);
589 label[label_len] = 0;
590 while (label_len && label[label_len-1] == ' ') label[--label_len] = 0;
591 }
592
593
594 /**************************************************************************
595 * VOLUME_SetSuperblockLabel
596 */
597 static BOOL VOLUME_SetSuperblockLabel( enum fs_type type, HANDLE handle, const WCHAR *label )
598 {
599 BYTE label_data[11];
600 DWORD offset, len;
601
602 switch(type)
603 {
604 case FS_FAT1216:
605 offset = 0x2b;
606 break;
607 case FS_FAT32:
608 offset = 0x47;
609 break;
610 default:
611 SetLastError( ERROR_ACCESS_DENIED );
612 return FALSE;
613 }
614 RtlUnicodeToMultiByteN( label_data, sizeof(label_data), &len,
615 label, strlenW(label) * sizeof(WCHAR) );
616 if (len < sizeof(label_data))
617 memset( label_data + len, ' ', sizeof(label_data) - len );
618
619 return (SetFilePointer( handle, offset, NULL, FILE_BEGIN ) == offset &&
620 WriteFile( handle, label_data, sizeof(label_data), &len, NULL ));
621 }
622
623
624 /**************************************************************************
625 * VOLUME_GetSuperblockSerial
626 */
627 static DWORD VOLUME_GetSuperblockSerial( enum fs_type type, const BYTE *superblock )
628 {
629 switch(type)
630 {
631 case FS_ERROR:
632 case FS_UNKNOWN:
633 break;
634 case FS_FAT1216:
635 return GETLONG( superblock, 0x27 );
636 case FS_FAT32:
637 return GETLONG( superblock, 0x33 );
638 case FS_ISO9660:
639 {
640 BYTE sum[4];
641 int i;
642
643 sum[0] = sum[1] = sum[2] = sum[3] = 0;
644 for (i = 0; i < 2048; i += 4)
645 {
646 /* DON'T optimize this into DWORD !! (breaks overflow) */
647 sum[0] += superblock[i+0];
648 sum[1] += superblock[i+1];
649 sum[2] += superblock[i+2];
650 sum[3] += superblock[i+3];
651 }
652 /*
653 * OK, another braindead one... argh. Just believe it.
654 * Me$$ysoft chose to reverse the serial number in NT4/W2K.
655 * It's true and nobody will ever be able to change it.
656 */
657 if (GetVersion() & 0x80000000)
658 return (sum[3] << 24) | (sum[2] << 16) | (sum[1] << 8) | sum[0];
659 else
660 return (sum[0] << 24) | (sum[1] << 16) | (sum[2] << 8) | sum[3];
661 }
662 }
663 return 0;
664 }
665
666
667 /**************************************************************************
668 * VOLUME_GetAudioCDSerial
669 */
670 static DWORD VOLUME_GetAudioCDSerial( const CDROM_TOC *toc )
671 {
672 DWORD serial = 0;
673 int i;
674
675 for (i = 0; i <= toc->LastTrack - toc->FirstTrack; i++)
676 serial += ((toc->TrackData[i].Address[1] << 16) |
677 (toc->TrackData[i].Address[2] << 8) |
678 toc->TrackData[i].Address[3]);
679
680 /*
681 * dwStart, dwEnd collect the beginning and end of the disc respectively, in
682 * frames.
683 * There it is collected for correcting the serial when there are less than
684 * 3 tracks.
685 */
686 if (toc->LastTrack - toc->FirstTrack + 1 < 3)
687 {
688 DWORD dwStart = FRAME_OF_TOC(toc, toc->FirstTrack);
689 DWORD dwEnd = FRAME_OF_TOC(toc, toc->LastTrack + 1);
690 serial += dwEnd - dwStart;
691 }
692 return serial;
693 }
694
695
696 /***********************************************************************
697 * GetVolumeInformationW (KERNEL32.@)
698 */
699 BOOL WINAPI GetVolumeInformationW( LPCWSTR root, LPWSTR label, DWORD label_len,
700 DWORD *serial, DWORD *filename_len, DWORD *flags,
701 LPWSTR fsname, DWORD fsname_len )
702 {
703 static const WCHAR audiocdW[] = {'A','u','d','i','o',' ','C','D',0};
704 static const WCHAR fatW[] = {'F','A','T',0};
705 static const WCHAR ntfsW[] = {'N','T','F','S',0};
706 static const WCHAR cdfsW[] = {'C','D','F','S',0};
707
708 WCHAR device[] = {'\\','\\','.','\\','A',':',0};
709 HANDLE handle;
710 enum fs_type type = FS_UNKNOWN;
711
712 if (!root)
713 {
714 WCHAR path[MAX_PATH];
715 GetCurrentDirectoryW( MAX_PATH, path );
716 device[4] = path[0];
717 }
718 else
719 {
720 if (!root[0] || root[1] != ':')
721 {
722 SetLastError( ERROR_INVALID_NAME );
723 return FALSE;
724 }
725 device[4] = root[0];
726 }
727
728 /* try to open the device */
729
730 handle = CreateFileW( device, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE,
731 NULL, OPEN_EXISTING, 0, 0 );
732 if (handle != INVALID_HANDLE_VALUE)
733 {
734 BYTE superblock[SUPERBLOCK_SIZE];
735 CDROM_TOC toc;
736 DWORD br;
737
738 /* check for audio CD */
739 /* FIXME: we only check the first track for now */
740 if (DeviceIoControl( handle, IOCTL_CDROM_READ_TOC, NULL, 0, &toc, sizeof(toc), &br, 0 ))
741 {
742 if (!(toc.TrackData[0].Control & 0x04)) /* audio track */
743 {
744 TRACE( "%s: found audio CD\n", debugstr_w(device) );
745 if (label) lstrcpynW( label, audiocdW, label_len );
746 if (serial) *serial = VOLUME_GetAudioCDSerial( &toc );
747 CloseHandle( handle );
748 type = FS_ISO9660;
749 goto fill_fs_info;
750 }
751 type = VOLUME_ReadCDSuperblock( handle, superblock );
752 }
753 else
754 {
755 type = VOLUME_ReadFATSuperblock( handle, superblock );
756 if (type == FS_UNKNOWN) type = VOLUME_ReadCDSuperblock( handle, superblock );
757 }
758 CloseHandle( handle );
759 TRACE( "%s: found fs type %d\n", debugstr_w(device), type );
760 if (type == FS_ERROR) return FALSE;
761
762 if (label && label_len) VOLUME_GetSuperblockLabel( type, superblock, label, label_len );
763 if (serial) *serial = VOLUME_GetSuperblockSerial( type, superblock );
764 goto fill_fs_info;
765 }
766 else
767 {
768 TRACE( "cannot open device %s: err %ld\n", debugstr_w(device), GetLastError() );
769 if (GetLastError() != ERROR_FILE_NOT_FOUND) return FALSE;
770 }
771
772 /* we couldn't open the device, fallback to default strategy */
773
774 switch(GetDriveTypeW( root ))
775 {
776 case DRIVE_UNKNOWN:
777 case DRIVE_NO_ROOT_DIR:
778 SetLastError( ERROR_NOT_READY );
779 return FALSE;
780 case DRIVE_REMOVABLE:
781 case DRIVE_FIXED:
782 case DRIVE_REMOTE:
783 case DRIVE_RAMDISK:
784 type = FS_UNKNOWN;
785 break;
786 case DRIVE_CDROM:
787 type = FS_ISO9660;
788 break;
789 }
790
791 if (label && label_len)
792 {
793 WCHAR labelW[] = {'A',':','\\','.','w','i','n','d','o','w','s','-','l','a','b','e','l',0};
794
795 labelW[0] = device[4];
796 handle = CreateFileW( labelW, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL,
797 OPEN_EXISTING, 0, 0 );
798 if (handle != INVALID_HANDLE_VALUE)
799 {
800 char buffer[256], *p;
801 DWORD size;
802
803 if (!ReadFile( handle, buffer, sizeof(buffer)-1, &size, NULL )) size = 0;
804 CloseHandle( handle );
805 p = buffer + size;
806 while (p > buffer && (p[-1] == ' ' || p[-1] == '\r' || p[-1] == '\n')) p--;
807 *p = 0;
808 if (!MultiByteToWideChar( CP_UNIXCP, 0, buffer, -1, label, label_len ))
809 label[label_len-1] = 0;
810 }
811 else label[0] = 0;
812 }
813 if (serial)
814 {
815 WCHAR serialW[] = {'A',':','\\','.','w','i','n','d','o','w','s','-','s','e','r','i','a','l',0};
816
817 serialW[0] = device[4];
818 handle = CreateFileW( serialW, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL,
819 OPEN_EXISTING, 0, 0 );
820 if (handle != INVALID_HANDLE_VALUE)
821 {
822 char buffer[32];
823 DWORD size;
824
825 if (!ReadFile( handle, buffer, sizeof(buffer)-1, &size, NULL )) size = 0;
826 CloseHandle( handle );
827 buffer[size] = 0;
828 *serial = strtoul( buffer, NULL, 16 );
829 }
830 else *serial = 0;
831 }
832
833 fill_fs_info: /* now fill in the information that depends on the file system type */
834
835 switch(type)
836 {
837 case FS_ISO9660:
838 if (fsname) lstrcpynW( fsname, cdfsW, fsname_len );
839 if (filename_len) *filename_len = 221;
840 if (flags) *flags = FILE_READ_ONLY_VOLUME;
841 break;
842 case FS_FAT1216:
843 case FS_FAT32:
844 if (fsname) lstrcpynW( fsname, fatW, fsname_len );
845 if (filename_len) *filename_len = 255;
846 if (flags) *flags = FILE_CASE_PRESERVED_NAMES; /* FIXME */
847 break;
848 default:
849 if (fsname) lstrcpynW( fsname, ntfsW, fsname_len );
850 if (filename_len) *filename_len = 255;
851 if (flags) *flags = FILE_CASE_PRESERVED_NAMES;
852 break;
853 }
854 return TRUE;
855 }
856
857
858 /***********************************************************************
859 * GetVolumeInformationA (KERNEL32.@)
860 */
861 BOOL WINAPI GetVolumeInformationA( LPCSTR root, LPSTR label,
862 DWORD label_len, DWORD *serial,
863 DWORD *filename_len, DWORD *flags,
864 LPSTR fsname, DWORD fsname_len )
865 {
866 UNICODE_STRING rootW;
867 LPWSTR labelW, fsnameW;
868 BOOL ret;
869
870 if (root) RtlCreateUnicodeStringFromAsciiz(&rootW, root);
871 else rootW.Buffer = NULL;
872 labelW = label ? HeapAlloc(GetProcessHeap(), 0, label_len * sizeof(WCHAR)) : NULL;
873 fsnameW = fsname ? HeapAlloc(GetProcessHeap(), 0, fsname_len * sizeof(WCHAR)) : NULL;
874
875 if ((ret = GetVolumeInformationW(rootW.Buffer, labelW, label_len, serial,
876 filename_len, flags, fsnameW, fsname_len)))
877 {
878 if (label) WideCharToMultiByte(CP_ACP, 0, labelW, -1, label, label_len, NULL, NULL);
879 if (fsname) WideCharToMultiByte(CP_ACP, 0, fsnameW, -1, fsname, fsname_len, NULL, NULL);
880 }
881
882 RtlFreeUnicodeString(&rootW);
883 if (labelW) HeapFree( GetProcessHeap(), 0, labelW );
884 if (fsnameW) HeapFree( GetProcessHeap(), 0, fsnameW );
885 return ret;
886 }
887
888
889
890 /***********************************************************************
891 * SetVolumeLabelW (KERNEL32.@)
892 */
893 BOOL WINAPI SetVolumeLabelW( LPCWSTR root, LPCWSTR label )
894 {
895 WCHAR device[] = {'\\','\\','.','\\','A',':',0};
896 HANDLE handle;
897 enum fs_type type = FS_UNKNOWN;
898
899 if (!root)
900 {
901 WCHAR path[MAX_PATH];
902 GetCurrentDirectoryW( MAX_PATH, path );
903 device[4] = path[0];
904 }
905 else
906 {
907 if (!root[0] || root[1] != ':')
908 {
909 SetLastError( ERROR_INVALID_NAME );
910 return FALSE;
911 }
912 device[4] = root[0];
913 }
914
915 /* try to open the device */
916
917 handle = CreateFileW( device, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE,
918 NULL, OPEN_EXISTING, 0, 0 );
919 if (handle == INVALID_HANDLE_VALUE)
920 {
921 /* try read-only */
922 handle = CreateFileW( device, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE,
923 NULL, OPEN_EXISTING, 0, 0 );
924 if (handle != INVALID_HANDLE_VALUE)
925 {
926 /* device can be read but not written, return error */
927 CloseHandle( handle );
928 SetLastError( ERROR_ACCESS_DENIED );
929 return FALSE;
930 }
931 }
932
933 if (handle != INVALID_HANDLE_VALUE)
934 {
935 BYTE superblock[SUPERBLOCK_SIZE];
936 BOOL ret;
937
938 type = VOLUME_ReadFATSuperblock( handle, superblock );
939 ret = VOLUME_SetSuperblockLabel( type, handle, label );
940 CloseHandle( handle );
941 return ret;
942 }
943 else
944 {
945 TRACE( "cannot open device %s: err %ld\n", debugstr_w(device), GetLastError() );
946 if (GetLastError() != ERROR_ACCESS_DENIED) return FALSE;
947 }
948
949 /* we couldn't open the device, fallback to default strategy */
950
951 switch(GetDriveTypeW( root ))
952 {
953 case DRIVE_UNKNOWN:
954 case DRIVE_NO_ROOT_DIR:
955 SetLastError( ERROR_NOT_READY );
956 break;
957 case DRIVE_REMOVABLE:
958 case DRIVE_FIXED:
959 {
960 WCHAR labelW[] = {'A',':','\\','.','w','i','n','d','o','w','s','-','l','a','b','e','l',0};
961
962 labelW[0] = device[4];
963 handle = CreateFileW( labelW, GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL,
964 CREATE_ALWAYS, 0, 0 );
965 if (handle != INVALID_HANDLE_VALUE)
966 {
967 char buffer[64];
968 DWORD size;
969
970 if (!WideCharToMultiByte( CP_UNIXCP, 0, label, -1, buffer, sizeof(buffer), NULL, NULL ))
971 buffer[sizeof(buffer)-1] = 0;
972 WriteFile( handle, buffer, strlen(buffer), &size, NULL );
973 CloseHandle( handle );
974 return TRUE;
975 }
976 break;
977 }
978 case DRIVE_REMOTE:
979 case DRIVE_RAMDISK:
980 case DRIVE_CDROM:
981 SetLastError( ERROR_ACCESS_DENIED );
982 break;
983 }
984 return FALSE;
985 }
986
987 /***********************************************************************
988 * SetVolumeLabelA (KERNEL32.@)
989 */
990 BOOL WINAPI SetVolumeLabelA(LPCSTR root, LPCSTR volname)
991 {
992 UNICODE_STRING rootW, volnameW;
993 BOOL ret;
994
995 if (root) RtlCreateUnicodeStringFromAsciiz(&rootW, root);
996 else rootW.Buffer = NULL;
997 if (volname) RtlCreateUnicodeStringFromAsciiz(&volnameW, volname);
998 else volnameW.Buffer = NULL;
999
1000 ret = SetVolumeLabelW( rootW.Buffer, volnameW.Buffer );
1001
1002 RtlFreeUnicodeString(&rootW);
1003 RtlFreeUnicodeString(&volnameW);
1004 return ret;
1005 }
1006
1007
1008 /***********************************************************************
1009 * GetVolumeNameForVolumeMountPointW (KERNEL32.@)
1010 */
1011 BOOL WINAPI GetVolumeNameForVolumeMountPointW(LPCWSTR str, LPWSTR dst, DWORD size)
1012 {
1013 FIXME("(%s, %p, %lx): stub\n", debugstr_w(str), dst, size);
1014 return 0;
1015 }
1016
1017
1018 /***********************************************************************
1019 * DefineDosDeviceW (KERNEL32.@)
1020 */
1021 BOOL WINAPI DefineDosDeviceW( DWORD flags, LPCWSTR devname, LPCWSTR targetpath )
1022 {
1023 DWORD len, dosdev;
1024 BOOL ret = FALSE;
1025 char *path = NULL, *target, *p;
1026
1027 if (!(flags & DDD_REMOVE_DEFINITION))
1028 {
1029 if (!(flags & DDD_RAW_TARGET_PATH))
1030 {
1031 FIXME( "(0x%08lx,%s,%s) DDD_RAW_TARGET_PATH flag not set, not supported yet\n",
1032 flags, debugstr_w(devname), debugstr_w(targetpath) );
1033 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
1034 return FALSE;
1035 }
1036
1037 len = WideCharToMultiByte( CP_UNIXCP, 0, targetpath, -1, NULL, 0, NULL, NULL );
1038 if ((target = HeapAlloc( GetProcessHeap(), 0, len )))
1039 {
1040 WideCharToMultiByte( CP_UNIXCP, 0, targetpath, -1, target, len, NULL, NULL );
1041 for (p = target; *p; p++) if (*p == '\\') *p = '/';
1042 }
1043 else
1044 {
1045 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1046 return FALSE;
1047 }
1048 }
1049 else target = NULL;
1050
1051 /* first check for a DOS device */
1052
1053 if ((dosdev = RtlIsDosDeviceName_U( devname )))
1054 {
1055 WCHAR name[5];
1056
1057 memcpy( name, devname + HIWORD(dosdev)/sizeof(WCHAR), LOWORD(dosdev) );
1058 name[LOWORD(dosdev)/sizeof(WCHAR)] = 0;
1059 path = get_dos_device_path( name );
1060 }
1061 else if (isalphaW(devname[0]) && devname[1] == ':' && !devname[2]) /* drive mapping */
1062 {
1063 path = get_dos_device_path( devname );
1064 }
1065 else SetLastError( ERROR_FILE_NOT_FOUND );
1066
1067 if (path)
1068 {
1069 if (target)
1070 {
1071 TRACE( "creating symlink %s -> %s\n", path, target );
1072 unlink( path );
1073 if (!symlink( target, path )) ret = TRUE;
1074 else FILE_SetDosError();
1075 }
1076 else
1077 {
1078 TRACE( "removing symlink %s\n", path );
1079 if (!unlink( path )) ret = TRUE;
1080 else FILE_SetDosError();
1081 }
1082 HeapFree( GetProcessHeap(), 0, path );
1083 }
1084 if (target) HeapFree( GetProcessHeap(), 0, target );
1085 return ret;
1086 }
1087
1088
1089 /***********************************************************************
1090 * DefineDosDeviceA (KERNEL32.@)
1091 */
1092 BOOL WINAPI DefineDosDeviceA(DWORD flags, LPCSTR devname, LPCSTR targetpath)
1093 {
1094 UNICODE_STRING d, t;
1095 BOOL ret;
1096
1097 if (!RtlCreateUnicodeStringFromAsciiz(&d, devname))
1098 {
1099 SetLastError(ERROR_NOT_ENOUGH_MEMORY);
1100 return FALSE;
1101 }
1102 if (!RtlCreateUnicodeStringFromAsciiz(&t, targetpath))
1103 {
1104 RtlFreeUnicodeString(&d);
1105 SetLastError(ERROR_NOT_ENOUGH_MEMORY);
1106 return FALSE;
1107 }
1108 ret = DefineDosDeviceW(flags, d.Buffer, t.Buffer);
1109 RtlFreeUnicodeString(&d);
1110 RtlFreeUnicodeString(&t);
1111 return ret;
1112 }
1113
1114
1115 /***********************************************************************
1116 * QueryDosDeviceW (KERNEL32.@)
1117 *
1118 * returns array of strings terminated by \0, terminated by \0
1119 */
1120 DWORD WINAPI QueryDosDeviceW( LPCWSTR devname, LPWSTR target, DWORD bufsize )
1121 {
1122 static const WCHAR auxW[] = {'A','U','X',0};
1123 static const WCHAR nulW[] = {'N','U','L',0};
1124 static const WCHAR prnW[] = {'P','R','N',0};
1125 static const WCHAR comW[] = {'C','O','M',0};
1126 static const WCHAR lptW[] = {'L','P','T',0};
1127 static const WCHAR rootW[] = {'A',':','\\',0};
1128 static const WCHAR com0W[] = {'\\','?','?','\\','C','O','M','0',0};
1129 static const WCHAR com1W[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\','C','O','M','1',0,0};
1130 static const WCHAR lpt1W[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\','L','P','T','1',0,0};
1131
1132 UNICODE_STRING nt_name;
1133 ANSI_STRING unix_name;
1134 WCHAR nt_buffer[10];
1135 NTSTATUS status;
1136
1137 if (!bufsize)
1138 {
1139 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1140 return 0;
1141 }
1142
1143 if (devname)
1144 {
1145 WCHAR *p, name[5];
1146 char *path, *link;
1147 DWORD dosdev, ret = 0;
1148
1149 if ((dosdev = RtlIsDosDeviceName_U( devname )))
1150 {
1151 memcpy( name, devname + HIWORD(dosdev)/sizeof(WCHAR), LOWORD(dosdev) );
1152 name[LOWORD(dosdev)/sizeof(WCHAR)] = 0;
1153 }
1154 else if (devname[0] && devname[1] == ':' && !devname[2])
1155 {
1156 memcpy( name, devname, 3 * sizeof(WCHAR) );
1157 }
1158 else
1159 {
1160 SetLastError( ERROR_BAD_PATHNAME );
1161 return 0;
1162 }
1163
1164 if (!(path = get_dos_device_path( name ))) return 0;
1165 link = read_symlink( path );
1166 HeapFree( GetProcessHeap(), 0, path );
1167
1168 if (link)
1169 {
1170 ret = MultiByteToWideChar( CP_UNIXCP, 0, link, -1, target, bufsize );
1171 HeapFree( GetProcessHeap(), 0, link );
1172 }
1173 else if (dosdev) /* look for device defaults */
1174 {
1175 if (!strcmpiW( name, auxW ))
1176 {
1177 if (bufsize >= sizeof(com1W)/sizeof(WCHAR))
1178 {
1179 memcpy( target, com1W, sizeof(com1W) );
1180 ret = sizeof(com1W)/sizeof(WCHAR);
1181 }
1182 else SetLastError( ERROR_INSUFFICIENT_BUFFER );
1183 return ret;
1184 }
1185 if (!strcmpiW( name, prnW ))
1186 {
1187 if (bufsize >= sizeof(lpt1W)/sizeof(WCHAR))
1188 {
1189 memcpy( target, lpt1W, sizeof(lpt1W) );
1190 ret = sizeof(lpt1W)/sizeof(WCHAR);
1191 }
1192 else SetLastError( ERROR_INSUFFICIENT_BUFFER );
1193 return ret;
1194 }
1195
1196 nt_buffer[0] = '\\';
1197 nt_buffer[1] = '?';
1198 nt_buffer[2] = '?';
1199 nt_buffer[3] = '\\';
1200 strcpyW( nt_buffer + 4, name );
1201 RtlInitUnicodeString( &nt_name, nt_buffer );
1202 status = wine_nt_to_unix_file_name( &nt_name, &unix_name, FILE_OPEN, TRUE );
1203 if (status) SetLastError( RtlNtStatusToDosError(status) );
1204 else
1205 {
1206 ret = MultiByteToWideChar( CP_UNIXCP, 0, unix_name.Buffer, -1, target, bufsize );
1207 RtlFreeAnsiString( &unix_name );
1208 }
1209 }
1210
1211 if (ret)
1212 {
1213 if (ret < bufsize) target[ret++] = 0; /* add an extra null */
1214 for (p = target; *p; p++) if (*p == '/') *p = '\\';
1215 }
1216
1217 return ret;
1218 }
1219 else /* return a list of all devices */
1220 {
1221 WCHAR *p = target;
1222 int i;
1223
1224 if (bufsize <= (sizeof(auxW)+sizeof(nulW)+sizeof(prnW))/sizeof(WCHAR))
1225 {
1226 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1227 return 0;
1228 }
1229
1230 memcpy( p, auxW, sizeof(auxW) );
1231 p += sizeof(auxW) / sizeof(WCHAR);
1232 memcpy( p, nulW, sizeof(nulW) );
1233 p += sizeof(nulW) / sizeof(WCHAR);
1234 memcpy( p, prnW, sizeof(prnW) );
1235 p += sizeof(prnW) / sizeof(WCHAR);
1236
1237 strcpyW( nt_buffer, com0W );
1238 RtlInitUnicodeString( &nt_name, nt_buffer );
1239
1240 for (i = 1; i <= 9; i++)
1241 {
1242 nt_buffer[7] = '0' + i;
1243 if (!wine_nt_to_unix_file_name( &nt_name, &unix_name, FILE_OPEN, TRUE ))
1244 {
1245 RtlFreeAnsiString( &unix_name );
1246 if (p + 5 >= target + bufsize)
1247 {
1248 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1249 return 0;
1250 }
1251 strcpyW( p, comW );
1252 p[3] = '0' + i;
1253 p[4] = 0;
1254 p += 5;
1255 }
1256 }
1257 strcpyW( nt_buffer + 4, lptW );
1258 for (i = 1; i <= 9; i++)
1259 {
1260 nt_buffer[7] = '0' + i;
1261 if (!wine_nt_to_unix_file_name( &nt_name, &unix_name, FILE_OPEN, TRUE ))
1262 {
1263 RtlFreeAnsiString( &unix_name );
1264 if (p + 5 >= target + bufsize)
1265 {
1266 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1267 return 0;
1268 }
1269 strcpyW( p, lptW );
1270 p[3] = '0' + i;
1271 p[4] = 0;
1272 p += 5;
1273 }
1274 }
1275
1276 strcpyW( nt_buffer + 4, rootW );
1277 RtlInitUnicodeString( &nt_name, nt_buffer );
1278
1279 for (i = 0; i < 26; i++)
1280 {
1281 nt_buffer[4] = 'a' + i;
1282 if (!wine_nt_to_unix_file_name( &nt_name, &unix_name, FILE_OPEN, TRUE ))
1283 {
1284 RtlFreeAnsiString( &unix_name );
1285 if (p + 3 >= target + bufsize)
1286 {
1287 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1288 return 0;
1289 }
1290 *p++ = 'A' + i;
1291 *p++ = ':';
1292 *p++ = 0;
1293 }
1294 }
1295 *p++ = 0; /* terminating null */
1296 return p - target;
1297 }
1298 }
1299
1300
1301 /***********************************************************************
1302 * QueryDosDeviceA (KERNEL32.@)
1303 *
1304 * returns array of strings terminated by \0, terminated by \0
1305 */
1306 DWORD WINAPI QueryDosDeviceA( LPCSTR devname, LPSTR target, DWORD bufsize )
1307 {
1308 DWORD ret = 0, retW;
1309 UNICODE_STRING devnameW;
1310 LPWSTR targetW = HeapAlloc( GetProcessHeap(),0, bufsize * sizeof(WCHAR) );
1311
1312 if (!targetW)
1313 {
1314 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1315 return 0;
1316 }
1317
1318 if (devname) RtlCreateUnicodeStringFromAsciiz(&devnameW, devname);
1319 else devnameW.Buffer = NULL;
1320
1321 retW = QueryDosDeviceW(devnameW.Buffer, targetW, bufsize);
1322
1323 ret = WideCharToMultiByte(CP_ACP, 0, targetW, retW, target, bufsize, NULL, NULL);
1324
1325 RtlFreeUnicodeString(&devnameW);
1326 HeapFree(GetProcessHeap(), 0, targetW);
1327 return ret;
1328 }
1329
1330
1331 /***********************************************************************
1332 * GetLogicalDrives (KERNEL32.@)
1333 */
1334 DWORD WINAPI GetLogicalDrives(void)
1335 {
1336 const char *config_dir = wine_get_config_dir();
1337 struct stat st;
1338 char *buffer, *dev;
1339 DWORD ret = 0;
1340 int i;
1341
1342 if (!(buffer = HeapAlloc( GetProcessHeap(), 0, strlen(config_dir) + sizeof("/dosdevices/a:") )))
1343 {
1344 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1345 return 0;
1346 }
1347 strcpy( buffer, config_dir );
1348 strcat( buffer, "/dosdevices/a:" );
1349 dev = buffer + strlen(buffer) - 2;
1350
1351 for (i = 0; i < 26; i++)
1352 {
1353 *dev = 'a' + i;
1354 if (!stat( buffer, &st )) ret |= (1 << i);
1355 }
1356 HeapFree( GetProcessHeap(), 0, buffer );
1357 return ret;
1358 }
1359
1360
1361 /***********************************************************************
1362 * GetLogicalDriveStringsA (KERNEL32.@)
1363 */
1364 UINT WINAPI GetLogicalDriveStringsA( UINT len, LPSTR buffer )
1365 {
1366 DWORD drives = GetLogicalDrives();
1367 UINT drive, count;
1368
1369 for (drive = count = 0; drive < 26; drive++) if (drives & (1 << drive)) count++;
1370 if ((count * 4) + 1 > len) return count * 4 + 1;
1371
1372 for (drive = 0; drive < 26; drive++)
1373 {
1374 if (drives & (1 << drive))
1375 {
1376 *buffer++ = 'a' + drive;
1377 *buffer++ = ':';
1378 *buffer++ = '\\';
1379 *buffer++ = 0;
1380 }
1381 }
1382 *buffer = 0;
1383 return count * 4;
1384 }
1385
1386
1387 /***********************************************************************
1388 * GetLogicalDriveStringsW (KERNEL32.@)
1389 */
1390 UINT WINAPI GetLogicalDriveStringsW( UINT len, LPWSTR buffer )
1391 {
1392 DWORD drives = GetLogicalDrives();
1393 UINT drive, count;
1394
1395 for (drive = count = 0; drive < 26; drive++) if (drives & (1 << drive)) count++;
1396 if ((count * 4) + 1 > len) return count * 4 + 1;
1397
1398 for (drive = 0; drive < 26; drive++)
1399 {
1400 if (drives & (1 << drive))
1401 {
1402 *buffer++ = 'a' + drive;
1403 *buffer++ = ':';
1404 *buffer++ = '\\';
1405 *buffer++ = 0;
1406 }
1407 }
1408 *buffer = 0;
1409 return count * 4;
1410 }
1411
1412
1413 /***********************************************************************
1414 * GetDriveTypeW (KERNEL32.@)
1415 *
1416 * Returns the type of the disk drive specified. If root is NULL the
1417 * root of the current directory is used.
1418 *
1419 * RETURNS
1420 *
1421 * Type of drive (from Win32 SDK):
1422 *
1423 * DRIVE_UNKNOWN unable to find out anything about the drive
1424 * DRIVE_NO_ROOT_DIR nonexistent root dir
1425 * DRIVE_REMOVABLE the disk can be removed from the machine
1426 * DRIVE_FIXED the disk can not be removed from the machine
1427 * DRIVE_REMOTE network disk
1428 * DRIVE_CDROM CDROM drive
1429 * DRIVE_RAMDISK virtual disk in RAM
1430 */
1431 UINT WINAPI GetDriveTypeW(LPCWSTR root) /* [in] String describing drive */
1432 {
1433 FILE_FS_DEVICE_INFORMATION info;
1434 IO_STATUS_BLOCK io;
1435 NTSTATUS status;
1436 HANDLE handle;
1437 UINT ret;
1438
1439 if (!open_device_root( root, &handle )) return DRIVE_NO_ROOT_DIR;
1440
1441 status = NtQueryVolumeInformationFile( handle, &io, &info, sizeof(info), FileFsDeviceInformation );
1442 NtClose( handle );
1443 if (status != STATUS_SUCCESS)
1444 {
1445 SetLastError( RtlNtStatusToDosError(status) );
1446 ret = DRIVE_UNKNOWN;
1447 }
1448 else if ((ret = get_registry_drive_type( root )) == DRIVE_UNKNOWN)
1449 {
1450 switch (info.DeviceType)
1451 {
1452 case FILE_DEVICE_CD_ROM_FILE_SYSTEM: ret = DRIVE_CDROM; break;
1453 case FILE_DEVICE_VIRTUAL_DISK: ret = DRIVE_RAMDISK; break;
1454 case FILE_DEVICE_NETWORK_FILE_SYSTEM: ret = DRIVE_REMOTE; break;
1455 case FILE_DEVICE_DISK_FILE_SYSTEM:
1456 if (info.Characteristics & FILE_REMOTE_DEVICE) ret = DRIVE_REMOTE;
1457 else if (info.Characteristics & FILE_REMOVABLE_MEDIA) ret = DRIVE_REMOVABLE;
1458 else ret = DRIVE_FIXED;
1459 break;
1460 default:
1461 ret = DRIVE_UNKNOWN;
1462 break;
1463 }
1464 }
1465 TRACE( "%s -> %d\n", debugstr_w(root), ret );
1466 return ret;
1467 }
1468
1469
1470 /***********************************************************************
1471 * GetDriveTypeA (KERNEL32.@)
1472 */
1473 UINT WINAPI GetDriveTypeA( LPCSTR root )
1474 {
1475 UNICODE_STRING rootW;
1476 UINT ret;
1477
1478 if (root)
1479 {
1480 if( !RtlCreateUnicodeStringFromAsciiz(&rootW, root))
1481 {
1482 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1483 return DRIVE_NO_ROOT_DIR;
1484 }
1485 }
1486 else rootW.Buffer = NULL;
1487
1488 ret = GetDriveTypeW(rootW.Buffer);
1489
1490 RtlFreeUnicodeString(&rootW);
1491 return ret;
1492 }
1493
1494
1495 /***********************************************************************
1496 * GetDiskFreeSpaceExW (KERNEL32.@)
1497 *
1498 * This function is used to acquire the size of the available and
1499 * total space on a logical volume.
1500 *
1501 * RETURNS
1502 *
1503 * Zero on failure, nonzero upon success. Use GetLastError to obtain
1504 * detailed error information.
1505 *
1506 */
1507 BOOL WINAPI GetDiskFreeSpaceExW( LPCWSTR root, PULARGE_INTEGER avail,
1508 PULARGE_INTEGER total, PULARGE_INTEGER totalfree )
1509 {
1510 FILE_FS_SIZE_INFORMATION info;
1511 IO_STATUS_BLOCK io;
1512 NTSTATUS status;
1513 HANDLE handle;
1514 UINT units;
1515
1516 TRACE( "%s,%p,%p,%p\n", debugstr_w(root), avail, total, totalfree );
1517
1518 if (!open_device_root( root, &handle )) return FALSE;
1519
1520 status = NtQueryVolumeInformationFile( handle, &io, &info, sizeof(info), FileFsSizeInformation );
1521 NtClose( handle );
1522 if (status != STATUS_SUCCESS)
1523 {
1524 SetLastError( RtlNtStatusToDosError(status) );
1525 return FALSE;
1526 }
1527
1528 units = info.SectorsPerAllocationUnit * info.BytesPerSector;
1529 if (total) total->QuadPart = info.TotalAllocationUnits.QuadPart * units;
1530 if (totalfree) totalfree->QuadPart = info.AvailableAllocationUnits.QuadPart * units;
1531 /* FIXME: this one should take quotas into account */
1532 if (avail) avail->QuadPart = info.AvailableAllocationUnits.QuadPart * units;
1533 return TRUE;
1534 }
1535
1536
1537 /***********************************************************************
1538 * GetDiskFreeSpaceExA (KERNEL32.@)
1539 */
1540 BOOL WINAPI GetDiskFreeSpaceExA( LPCSTR root, PULARGE_INTEGER avail,
1541 PULARGE_INTEGER total, PULARGE_INTEGER totalfree )
1542 {
1543 UNICODE_STRING rootW;
1544 BOOL ret;
1545
1546 if (root) RtlCreateUnicodeStringFromAsciiz(&rootW, root);
1547 else rootW.Buffer = NULL;
1548
1549 ret = GetDiskFreeSpaceExW( rootW.Buffer, avail, total, totalfree);
1550
1551 RtlFreeUnicodeString(&rootW);
1552 return ret;
1553 }
1554
1555
1556 /***********************************************************************
1557 * GetDiskFreeSpaceW (KERNEL32.@)
1558 */
1559 BOOL WINAPI GetDiskFreeSpaceW( LPCWSTR root, LPDWORD cluster_sectors,
1560 LPDWORD sector_bytes, LPDWORD free_clusters,
1561 LPDWORD total_clusters )
1562 {
1563 FILE_FS_SIZE_INFORMATION info;
1564 IO_STATUS_BLOCK io;
1565 NTSTATUS status;
1566 HANDLE handle;
1567 UINT units;
1568
1569 TRACE( "%s,%p,%p,%p,%p\n", debugstr_w(root),
1570 cluster_sectors, sector_bytes, free_clusters, total_clusters );
1571
1572 if (!open_device_root( root, &handle )) return FALSE;
1573
1574 status = NtQueryVolumeInformationFile( handle, &io, &info, sizeof(info), FileFsSizeInformation );
1575 NtClose( handle );
1576 if (status != STATUS_SUCCESS)
1577 {
1578 SetLastError( RtlNtStatusToDosError(status) );
1579 return FALSE;
1580 }
1581
1582 units = info.SectorsPerAllocationUnit * info.BytesPerSector;
1583
1584 /* cap the size and available at 2GB as per specs */
1585 if (info.AvailableAllocationUnits.QuadPart * units > 0x7fffffff)
1586 info.AvailableAllocationUnits.QuadPart = 0x7fffffff / units;
1587 if (info.TotalAllocationUnits.QuadPart * units > 0x7fffffff)
1588 info.TotalAllocationUnits.QuadPart = 0x7fffffff / units;
1589
1590 if (cluster_sectors) *cluster_sectors = info.SectorsPerAllocationUnit;
1591 if (sector_bytes) *sector_bytes = info.BytesPerSector;
1592 if (free_clusters) *free_clusters = info.AvailableAllocationUnits.u.LowPart;
1593 if (total_clusters) *total_clusters = info.TotalAllocationUnits.u.LowPart;
1594 return TRUE;
1595 }
1596
1597
1598 /***********************************************************************
1599 * GetDiskFreeSpaceA (KERNEL32.@)
1600 */
1601 BOOL WINAPI GetDiskFreeSpaceA( LPCSTR root, LPDWORD cluster_sectors,
1602 LPDWORD sector_bytes, LPDWORD free_clusters,
1603 LPDWORD total_clusters )
1604 {
1605 UNICODE_STRING rootW;
1606 BOOL ret = FALSE;
1607
1608 if (root)
1609 {
1610 if(!RtlCreateUnicodeStringFromAsciiz(&rootW, root))
1611 {
1612 SetLastError(ERROR_NOT_ENOUGH_MEMORY);
1613 return FALSE;
1614 }
1615 }
1616 else rootW.Buffer = NULL;
1617
1618 ret = GetDiskFreeSpaceW(rootW.Buffer, cluster_sectors, sector_bytes,
1619 free_clusters, total_clusters );
1620 RtlFreeUnicodeString(&rootW);
1621
1622 return ret;
1623 }
Maarten's multimedia branch