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