?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
msxml3: Fix ignorable whitespace detection.
[wine.git] / dlls / kernel32 / volume.c
blob2e0cc8a2c26037ccac0b838003834b51f45576be
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 #define WINE_MOUNTMGR_EXTENSIONS
41 #include "ddk/mountmgr.h"
42 #include "kernel_private.h"
43 #include "wine/library.h"
44 #include "wine/unicode.h"
45 #include "wine/debug.h"
46
47 WINE_DEFAULT_DEBUG_CHANNEL(volume);
48
49 #define BLOCK_SIZE 2048
50 #define SUPERBLOCK_SIZE BLOCK_SIZE
51 #define SYMBOLIC_LINK_QUERY 0x0001
52
53 #define CDFRAMES_PERSEC 75
54 #define CDFRAMES_PERMIN (CDFRAMES_PERSEC * 60)
55 #define FRAME_OF_ADDR(a) ((a)[1] * CDFRAMES_PERMIN + (a)[2] * CDFRAMES_PERSEC + (a)[3])
56 #define FRAME_OF_TOC(toc, idx) FRAME_OF_ADDR((toc)->TrackData[(idx) - (toc)->FirstTrack].Address)
57
58 #define GETWORD(buf,off) MAKEWORD(buf[(off)],buf[(off+1)])
59 #define GETLONG(buf,off) MAKELONG(GETWORD(buf,off),GETWORD(buf,off+2))
60
61 enum fs_type
62 {
63 FS_ERROR, /* error accessing the device */
64 FS_UNKNOWN, /* unknown file system */
65 FS_FAT1216,
66 FS_FAT32,
67 FS_ISO9660,
68 FS_UDF /* For reference [E] = Ecma-167.pdf, [U] = udf260.pdf */
69 };
70
71 /* read a Unix symlink; returned buffer must be freed by caller */
72 static char *read_symlink( const char *path )
73 {
74 char *buffer;
75 int ret, size = 128;
76
77 for (;;)
78 {
79 if (!(buffer = HeapAlloc( GetProcessHeap(), 0, size )))
80 {
81 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
82 return 0;
83 }
84 ret = readlink( path, buffer, size );
85 if (ret == -1)
86 {
87 FILE_SetDosError();
88 HeapFree( GetProcessHeap(), 0, buffer );
89 return 0;
90 }
91 if (ret != size)
92 {
93 buffer[ret] = 0;
94 return buffer;
95 }
96 HeapFree( GetProcessHeap(), 0, buffer );
97 size *= 2;
98 }
99 }
100
101 /* get the path of a dos device symlink in the $WINEPREFIX/dosdevices directory */
102 static char *get_dos_device_path( LPCWSTR name )
103 {
104 const char *config_dir = wine_get_config_dir();
105 char *buffer, *dev;
106 int i;
107
108 if (!(buffer = HeapAlloc( GetProcessHeap(), 0,
109 strlen(config_dir) + sizeof("/dosdevices/") + 5 )))
110 {
111 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
112 return NULL;
113 }
114 strcpy( buffer, config_dir );
115 strcat( buffer, "/dosdevices/" );
116 dev = buffer + strlen(buffer);
117 /* no codepage conversion, DOS device names are ASCII anyway */
118 for (i = 0; i < 5; i++)
119 if (!(dev[i] = (char)tolowerW(name[i]))) break;
120 dev[5] = 0;
121 return buffer;
122 }
123
124 /* read the contents of an NT symlink object */
125 static NTSTATUS read_nt_symlink( const WCHAR *name, WCHAR *target, DWORD size )
126 {
127 NTSTATUS status;
128 OBJECT_ATTRIBUTES attr;
129 UNICODE_STRING nameW;
130 HANDLE handle;
131
132 attr.Length = sizeof(attr);
133 attr.RootDirectory = 0;
134 attr.Attributes = OBJ_CASE_INSENSITIVE;
135 attr.ObjectName = &nameW;
136 attr.SecurityDescriptor = NULL;
137 attr.SecurityQualityOfService = NULL;
138 RtlInitUnicodeString( &nameW, name );
139
140 if (!(status = NtOpenSymbolicLinkObject( &handle, SYMBOLIC_LINK_QUERY, &attr )))
141 {
142 UNICODE_STRING targetW;
143 targetW.Buffer = target;
144 targetW.MaximumLength = (size - 1) * sizeof(WCHAR);
145 status = NtQuerySymbolicLinkObject( handle, &targetW, NULL );
146 if (!status) target[targetW.Length / sizeof(WCHAR)] = 0;
147 NtClose( handle );
148 }
149 return status;
150 }
151
152 /* open a handle to a device root */
153 static BOOL open_device_root( LPCWSTR root, HANDLE *handle )
154 {
155 static const WCHAR default_rootW[] = {'\\',0};
156 UNICODE_STRING nt_name;
157 OBJECT_ATTRIBUTES attr;
158 IO_STATUS_BLOCK io;
159 NTSTATUS status;
160
161 if (!root) root = default_rootW;
162 if (!RtlDosPathNameToNtPathName_U( root, &nt_name, NULL, NULL ))
163 {
164 SetLastError( ERROR_PATH_NOT_FOUND );
165 return FALSE;
166 }
167 attr.Length = sizeof(attr);
168 attr.RootDirectory = 0;
169 attr.Attributes = OBJ_CASE_INSENSITIVE;
170 attr.ObjectName = &nt_name;
171 attr.SecurityDescriptor = NULL;
172 attr.SecurityQualityOfService = NULL;
173
174 status = NtOpenFile( handle, 0, &attr, &io, 0,
175 FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT );
176 RtlFreeUnicodeString( &nt_name );
177 if (status != STATUS_SUCCESS)
178 {
179 SetLastError( RtlNtStatusToDosError(status) );
180 return FALSE;
181 }
182 return TRUE;
183 }
184
185 /* query the type of a drive from the mount manager */
186 static DWORD get_mountmgr_drive_type( LPCWSTR root )
187 {
188 HANDLE mgr;
189 struct mountmgr_unix_drive data;
190
191 memset( &data, 0, sizeof(data) );
192 if (root) data.letter = root[0];
193 else
194 {
195 WCHAR curdir[MAX_PATH];
196 GetCurrentDirectoryW( MAX_PATH, curdir );
197 if (curdir[1] != ':' || curdir[2] != '\\') return DRIVE_UNKNOWN;
198 data.letter = curdir[0];
199 }
200
201 mgr = CreateFileW( MOUNTMGR_DOS_DEVICE_NAME, GENERIC_READ,
202 FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, 0 );
203 if (mgr == INVALID_HANDLE_VALUE) return DRIVE_UNKNOWN;
204
205 if (!DeviceIoControl( mgr, IOCTL_MOUNTMGR_QUERY_UNIX_DRIVE, &data, sizeof(data), &data,
206 sizeof(data), NULL, NULL ) && GetLastError() != ERROR_MORE_DATA)
207 data.type = DRIVE_UNKNOWN;
208
209 CloseHandle( mgr );
210 return data.type;
211 }
212
213 /* get the label by reading it from a file at the root of the filesystem */
214 static void get_filesystem_label( const UNICODE_STRING *device, WCHAR *label, DWORD len )
215 {
216 static const WCHAR labelW[] = {'.','w','i','n','d','o','w','s','-','l','a','b','e','l',0};
217 HANDLE handle;
218 UNICODE_STRING name;
219 IO_STATUS_BLOCK io;
220 OBJECT_ATTRIBUTES attr;
221
222 label[0] = 0;
223
224 attr.Length = sizeof(attr);
225 attr.RootDirectory = 0;
226 attr.Attributes = OBJ_CASE_INSENSITIVE;
227 attr.ObjectName = &name;
228 attr.SecurityDescriptor = NULL;
229 attr.SecurityQualityOfService = NULL;
230
231 name.MaximumLength = device->Length + sizeof(labelW);
232 name.Length = name.MaximumLength - sizeof(WCHAR);
233 if (!(name.Buffer = HeapAlloc( GetProcessHeap(), 0, name.MaximumLength ))) return;
234
235 memcpy( name.Buffer, device->Buffer, device->Length );
236 memcpy( name.Buffer + device->Length / sizeof(WCHAR), labelW, sizeof(labelW) );
237 if (!NtOpenFile( &handle, GENERIC_READ, &attr, &io, FILE_SHARE_READ|FILE_SHARE_WRITE,
238 FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT ))
239 {
240 char buffer[256], *p;
241 DWORD size;
242
243 if (!ReadFile( handle, buffer, sizeof(buffer)-1, &size, NULL )) size = 0;
244 CloseHandle( handle );
245 p = buffer + size;
246 while (p > buffer && (p[-1] == ' ' || p[-1] == '\r' || p[-1] == '\n')) p--;
247 *p = 0;
248 if (!MultiByteToWideChar( CP_UNIXCP, 0, buffer, -1, label, len ))
249 label[len-1] = 0;
250 }
251 RtlFreeUnicodeString( &name );
252 }
253
254 /* get the serial number by reading it from a file at the root of the filesystem */
255 static DWORD get_filesystem_serial( const UNICODE_STRING *device )
256 {
257 static const WCHAR serialW[] = {'.','w','i','n','d','o','w','s','-','s','e','r','i','a','l',0};
258 HANDLE handle;
259 UNICODE_STRING name;
260 IO_STATUS_BLOCK io;
261 OBJECT_ATTRIBUTES attr;
262 DWORD ret = 0;
263
264 attr.Length = sizeof(attr);
265 attr.RootDirectory = 0;
266 attr.Attributes = OBJ_CASE_INSENSITIVE;
267 attr.ObjectName = &name;
268 attr.SecurityDescriptor = NULL;
269 attr.SecurityQualityOfService = NULL;
270
271 name.MaximumLength = device->Length + sizeof(serialW);
272 name.Length = name.MaximumLength - sizeof(WCHAR);
273 if (!(name.Buffer = HeapAlloc( GetProcessHeap(), 0, name.MaximumLength ))) return 0;
274
275 memcpy( name.Buffer, device->Buffer, device->Length );
276 memcpy( name.Buffer + device->Length / sizeof(WCHAR), serialW, sizeof(serialW) );
277 if (!NtOpenFile( &handle, GENERIC_READ, &attr, &io, FILE_SHARE_READ|FILE_SHARE_WRITE,
278 FILE_SYNCHRONOUS_IO_NONALERT ))
279 {
280 char buffer[32];
281 DWORD size;
282
283 if (!ReadFile( handle, buffer, sizeof(buffer)-1, &size, NULL )) size = 0;
284 CloseHandle( handle );
285 buffer[size] = 0;
286 ret = strtoul( buffer, NULL, 16 );
287 }
288 RtlFreeUnicodeString( &name );
289 return ret;
290 }
291
292
293 /******************************************************************
294 * VOLUME_FindCdRomDataBestVoldesc
295 */
296 static DWORD VOLUME_FindCdRomDataBestVoldesc( HANDLE handle )
297 {
298 BYTE cur_vd_type, max_vd_type = 0;
299 BYTE buffer[0x800];
300 DWORD size, offs, best_offs = 0, extra_offs = 0;
301
302 for (offs = 0x8000; offs <= 0x9800; offs += 0x800)
303 {
304 /* if 'CDROM' occurs at position 8, this is a pre-iso9660 cd, and
305 * the volume label is displaced forward by 8
306 */
307 if (SetFilePointer( handle, offs, NULL, FILE_BEGIN ) != offs) break;
308 if (!ReadFile( handle, buffer, sizeof(buffer), &size, NULL )) break;
309 if (size != sizeof(buffer)) break;
310 /* check for non-ISO9660 signature */
311 if (!memcmp( buffer + 11, "ROM", 3 )) extra_offs = 8;
312 cur_vd_type = buffer[extra_offs];
313 if (cur_vd_type == 0xff) /* voldesc set terminator */
314 break;
315 if (cur_vd_type > max_vd_type)
316 {
317 max_vd_type = cur_vd_type;
318 best_offs = offs + extra_offs;
319 }
320 }
321 return best_offs;
322 }
323
324
325 /***********************************************************************
326 * VOLUME_ReadFATSuperblock
327 */
328 static enum fs_type VOLUME_ReadFATSuperblock( HANDLE handle, BYTE *buff )
329 {
330 DWORD size;
331
332 /* try a fixed disk, with a FAT partition */
333 if (SetFilePointer( handle, 0, NULL, FILE_BEGIN ) != 0 ||
334 !ReadFile( handle, buff, SUPERBLOCK_SIZE, &size, NULL ))
335 {
336 if (GetLastError() == ERROR_BAD_DEV_TYPE) return FS_UNKNOWN; /* not a real device */
337 return FS_ERROR;
338 }
339
340 if (size < SUPERBLOCK_SIZE) return FS_UNKNOWN;
341
342 /* FIXME: do really all FAT have their name beginning with
343 * "FAT" ? (At least FAT12, FAT16 and FAT32 have :)
344 */
345 if (!memcmp(buff+0x36, "FAT", 3) || !memcmp(buff+0x52, "FAT", 3))
346 {
347 /* guess which type of FAT we have */
348 int reasonable;
349 unsigned int sectors,
350 sect_per_fat,
351 total_sectors,
352 num_boot_sectors,
353 num_fats,
354 num_root_dir_ents,
355 bytes_per_sector,
356 sectors_per_cluster,
357 nclust;
358 sect_per_fat = GETWORD(buff, 0x16);
359 if (!sect_per_fat) sect_per_fat = GETLONG(buff, 0x24);
360 total_sectors = GETWORD(buff, 0x13);
361 if (!total_sectors)
362 total_sectors = GETLONG(buff, 0x20);
363 num_boot_sectors = GETWORD(buff, 0x0e);
364 num_fats = buff[0x10];
365 num_root_dir_ents = GETWORD(buff, 0x11);
366 bytes_per_sector = GETWORD(buff, 0x0b);
367 sectors_per_cluster = buff[0x0d];
368 /* check if the parameters are reasonable and will not cause
369 * arithmetic errors in the calculation */
370 reasonable = num_boot_sectors < total_sectors &&
371 num_fats < 16 &&
372 bytes_per_sector >= 512 && bytes_per_sector % 512 == 0 &&
373 sectors_per_cluster >= 1;
374 if (!reasonable) return FS_UNKNOWN;
375 sectors = total_sectors - num_boot_sectors - num_fats * sect_per_fat -
376 (num_root_dir_ents * 32 + bytes_per_sector - 1) / bytes_per_sector;
377 nclust = sectors / sectors_per_cluster;
378 if ((buff[0x42] == 0x28 || buff[0x42] == 0x29) &&
379 !memcmp(buff+0x52, "FAT", 3)) return FS_FAT32;
380 if (nclust < 65525)
381 {
382 if ((buff[0x26] == 0x28 || buff[0x26] == 0x29) &&
383 !memcmp(buff+0x36, "FAT", 3))
384 return FS_FAT1216;
385 }
386 }
387 return FS_UNKNOWN;
388 }
389
390
391 /***********************************************************************
392 * VOLUME_ReadCDBlock
393 */
394 static BOOL VOLUME_ReadCDBlock( HANDLE handle, BYTE *buff, INT offs )
395 {
396 DWORD size, whence = offs >= 0 ? FILE_BEGIN : FILE_END;
397
398 if (SetFilePointer( handle, offs, NULL, whence ) != offs ||
399 !ReadFile( handle, buff, SUPERBLOCK_SIZE, &size, NULL ) ||
400 size != SUPERBLOCK_SIZE)
401 return FALSE;
402
403 return TRUE;
404 }
405
406
407 /***********************************************************************
408 * VOLUME_ReadCDSuperblock
409 */
410 static enum fs_type VOLUME_ReadCDSuperblock( HANDLE handle, BYTE *buff )
411 {
412 int i;
413 DWORD offs;
414
415 /* Check UDF first as UDF and ISO9660 structures can coexist on the same medium
416 * Starting from sector 16, we may find :
417 * - a CD-ROM Volume Descriptor Set (ISO9660) containing one or more Volume Descriptors
418 * - an Extented Area (UDF) -- [E] 2/8.3.1 and [U] 2.1.7
419 * There is no explicit end so read 16 sectors and then give up */
420 for( i=16; i<16+16; i++)
421 {
422 if (!VOLUME_ReadCDBlock(handle, buff, i*BLOCK_SIZE))
423 continue;
424
425 /* We are supposed to check "BEA01", "NSR0x" and "TEA01" IDs + verify tag checksum
426 * but we assume the volume is well-formatted */
427 if (!memcmp(&buff[1], "BEA01", 5)) return FS_UDF;
428 }
429
430 offs = VOLUME_FindCdRomDataBestVoldesc( handle );
431 if (!offs) return FS_UNKNOWN;
432
433 if (!VOLUME_ReadCDBlock(handle, buff, offs))
434 return FS_ERROR;
435
436 /* check for the iso9660 identifier */
437 if (!memcmp(&buff[1], "CD001", 5)) return FS_ISO9660;
438 return FS_UNKNOWN;
439 }
440
441
442 /**************************************************************************
443 * UDF_Find_PVD
444 * Find the Primary Volume Descriptor
445 */
446 static BOOL UDF_Find_PVD( HANDLE handle, BYTE pvd[] )
447 {
448 unsigned int i;
449 DWORD offset;
450 INT locations[] = { 256, -1, -257, 512 };
451
452 for(i=0; i<sizeof(locations)/sizeof(locations[0]); i++)
453 {
454 if (!VOLUME_ReadCDBlock(handle, pvd, locations[i]*BLOCK_SIZE))
455 return FALSE;
456
457 /* Tag Identifier of Anchor Volume Descriptor Pointer is 2 -- [E] 3/10.2.1 */
458 if (pvd[0]==2 && pvd[1]==0)
459 {
460 /* Tag location (Uint32) at offset 12, little-endian */
461 offset = pvd[20 + 0];
462 offset |= pvd[20 + 1] << 8;
463 offset |= pvd[20 + 2] << 16;
464 offset |= pvd[20 + 3] << 24;
465 offset *= BLOCK_SIZE;
466
467 if (!VOLUME_ReadCDBlock(handle, pvd, offset))
468 return FALSE;
469
470 /* Check for the Primary Volume Descriptor Tag Id -- [E] 3/10.1.1 */
471 if (pvd[0]!=1 || pvd[1]!=0)
472 return FALSE;
473
474 /* 8 or 16 bits per character -- [U] 2.1.1 */
475 if (!(pvd[24]==8 || pvd[24]==16))
476 return FALSE;
477
478 return TRUE;
479 }
480 }
481
482 return FALSE;
483 }
484
485
486 /**************************************************************************
487 * VOLUME_GetSuperblockLabel
488 */
489 static void VOLUME_GetSuperblockLabel( const UNICODE_STRING *device, HANDLE handle,
490 enum fs_type type, const BYTE *superblock,
491 WCHAR *label, DWORD len )
492 {
493 const BYTE *label_ptr = NULL;
494 DWORD label_len;
495
496 switch(type)
497 {
498 case FS_ERROR:
499 label_len = 0;
500 break;
501 case FS_UNKNOWN:
502 get_filesystem_label( device, label, len );
503 return;
504 case FS_FAT1216:
505 label_ptr = superblock + 0x2b;
506 label_len = 11;
507 break;
508 case FS_FAT32:
509 label_ptr = superblock + 0x47;
510 label_len = 11;
511 break;
512 case FS_ISO9660:
513 {
514 BYTE ver = superblock[0x5a];
515
516 if (superblock[0x58] == 0x25 && superblock[0x59] == 0x2f && /* Unicode ID */
517 ((ver == 0x40) || (ver == 0x43) || (ver == 0x45)))
518 { /* yippee, unicode */
519 unsigned int i;
520
521 if (len > 17) len = 17;
522 for (i = 0; i < len-1; i++)
523 label[i] = (superblock[40+2*i] << 8) | superblock[41+2*i];
524 label[i] = 0;
525 while (i && label[i-1] == ' ') label[--i] = 0;
526 return;
527 }
528 label_ptr = superblock + 40;
529 label_len = 32;
530 break;
531 }
532 case FS_UDF:
533 {
534 BYTE pvd[BLOCK_SIZE];
535
536 if(!UDF_Find_PVD(handle, pvd))
537 {
538 label_len = 0;
539 break;
540 }
541
542 /* [E] 3/10.1.4 and [U] 2.1.1 */
543 if(pvd[24]==8)
544 {
545 label_ptr = pvd + 24 + 1;
546 label_len = pvd[24+32-1];
547 break;
548 }
549 else
550 {
551 unsigned int i;
552
553 label_len = 1 + pvd[24+32-1];
554 for(i=0; i<label_len && i<len; i+=2)
555 label[i/2] = (pvd[24+1 +i] << 8) | pvd[24+1 +i+1];
556 label[label_len] = 0;
557 return;
558 }
559 }
560 }
561 if (label_len) RtlMultiByteToUnicodeN( label, (len-1) * sizeof(WCHAR),
562 &label_len, (LPCSTR)label_ptr, label_len );
563 label_len /= sizeof(WCHAR);
564 label[label_len] = 0;
565 while (label_len && label[label_len-1] == ' ') label[--label_len] = 0;
566 }
567
568
569 /**************************************************************************
570 * UDF_Find_FSD_Sector
571 * Find the File Set Descriptor used to compute the serial of a UDF volume
572 */
573 static int UDF_Find_FSD_Sector( HANDLE handle, BYTE block[] )
574 {
575 int i, PVD_sector, PD_sector, PD_length;
576
577 if(!UDF_Find_PVD(handle,block))
578 goto default_sector;
579
580 /* Retrieve the tag location of the PVD -- [E] 3/7.2 */
581 PVD_sector = block[12 + 0];
582 PVD_sector |= block[12 + 1] << 8;
583 PVD_sector |= block[12 + 2] << 16;
584 PVD_sector |= block[12 + 3] << 24;
585
586 /* Find the Partition Descriptor */
587 for(i=PVD_sector+1; ; i++)
588 {
589 if(!VOLUME_ReadCDBlock(handle, block, i*BLOCK_SIZE))
590 goto default_sector;
591
592 /* Partition Descriptor Tag Id -- [E] 3/10.5.1 */
593 if(block[0]==5 && block[1]==0)
594 break;
595
596 /* Terminating Descriptor Tag Id -- [E] 3/10.9.1 */
597 if(block[0]==8 && block[1]==0)
598 goto default_sector;
599 }
600
601 /* Find the partition starting location -- [E] 3/10.5.8 */
602 PD_sector = block[188 + 0];
603 PD_sector |= block[188 + 1] << 8;
604 PD_sector |= block[188 + 2] << 16;
605 PD_sector |= block[188 + 3] << 24;
606
607 /* Find the partition length -- [E] 3/10.5.9 */
608 PD_length = block[192 + 0];
609 PD_length |= block[192 + 1] << 8;
610 PD_length |= block[192 + 2] << 16;
611 PD_length |= block[192 + 3] << 24;
612
613 for(i=PD_sector; i<PD_sector+PD_length; i++)
614 {
615 if(!VOLUME_ReadCDBlock(handle, block, i*BLOCK_SIZE))
616 goto default_sector;
617
618 /* File Set Descriptor Tag Id -- [E] 3/14.1.1 */
619 if(block[0]==0 && block[1]==1)
620 return i;
621 }
622
623 default_sector:
624 WARN("FSD sector not found, serial may be incorrect\n");
625 return 257;
626 }
627
628
629 /**************************************************************************
630 * VOLUME_GetSuperblockSerial
631 */
632 static DWORD VOLUME_GetSuperblockSerial( const UNICODE_STRING *device, HANDLE handle,
633 enum fs_type type, const BYTE *superblock )
634 {
635 int FSD_sector;
636 BYTE block[BLOCK_SIZE];
637
638 switch(type)
639 {
640 case FS_ERROR:
641 break;
642 case FS_UNKNOWN:
643 return get_filesystem_serial( device );
644 case FS_FAT1216:
645 return GETLONG( superblock, 0x27 );
646 case FS_FAT32:
647 return GETLONG( superblock, 0x33 );
648 case FS_UDF:
649 FSD_sector = UDF_Find_FSD_Sector(handle, block);
650 if (!VOLUME_ReadCDBlock(handle, block, FSD_sector*BLOCK_SIZE))
651 break;
652 superblock = block;
653 /* fallthrough */
654 case FS_ISO9660:
655 {
656 BYTE sum[4];
657 int i;
658
659 sum[0] = sum[1] = sum[2] = sum[3] = 0;
660 for (i = 0; i < 2048; i += 4)
661 {
662 /* DON'T optimize this into DWORD !! (breaks overflow) */
663 sum[0] += superblock[i+0];
664 sum[1] += superblock[i+1];
665 sum[2] += superblock[i+2];
666 sum[3] += superblock[i+3];
667 }
668 /*
669 * OK, another braindead one... argh. Just believe it.
670 * Me$$ysoft chose to reverse the serial number in NT4/W2K.
671 * It's true and nobody will ever be able to change it.
672 */
673 if ((GetVersion() & 0x80000000) || type == FS_UDF)
674 return (sum[3] << 24) | (sum[2] << 16) | (sum[1] << 8) | sum[0];
675 else
676 return (sum[0] << 24) | (sum[1] << 16) | (sum[2] << 8) | sum[3];
677 }
678 }
679 return 0;
680 }
681
682
683 /**************************************************************************
684 * VOLUME_GetAudioCDSerial
685 */
686 static DWORD VOLUME_GetAudioCDSerial( const CDROM_TOC *toc )
687 {
688 DWORD serial = 0;
689 int i;
690
691 for (i = 0; i <= toc->LastTrack - toc->FirstTrack; i++)
692 serial += ((toc->TrackData[i].Address[1] << 16) |
693 (toc->TrackData[i].Address[2] << 8) |
694 toc->TrackData[i].Address[3]);
695
696 /*
697 * dwStart, dwEnd collect the beginning and end of the disc respectively, in
698 * frames.
699 * There it is collected for correcting the serial when there are less than
700 * 3 tracks.
701 */
702 if (toc->LastTrack - toc->FirstTrack + 1 < 3)
703 {
704 DWORD dwStart = FRAME_OF_TOC(toc, toc->FirstTrack);
705 DWORD dwEnd = FRAME_OF_TOC(toc, toc->LastTrack + 1);
706 serial += dwEnd - dwStart;
707 }
708 return serial;
709 }
710
711
712 /***********************************************************************
713 * GetVolumeInformationW (KERNEL32.@)
714 */
715 BOOL WINAPI GetVolumeInformationW( LPCWSTR root, LPWSTR label, DWORD label_len,
716 DWORD *serial, DWORD *filename_len, DWORD *flags,
717 LPWSTR fsname, DWORD fsname_len )
718 {
719 static const WCHAR audiocdW[] = {'A','u','d','i','o',' ','C','D',0};
720 static const WCHAR fatW[] = {'F','A','T',0};
721 static const WCHAR fat32W[] = {'F','A','T','3','2',0};
722 static const WCHAR ntfsW[] = {'N','T','F','S',0};
723 static const WCHAR cdfsW[] = {'C','D','F','S',0};
724 static const WCHAR udfW[] = {'U','D','F',0};
725 static const WCHAR default_rootW[] = {'\\',0};
726
727 HANDLE handle;
728 NTSTATUS status;
729 UNICODE_STRING nt_name;
730 IO_STATUS_BLOCK io;
731 OBJECT_ATTRIBUTES attr;
732 FILE_FS_DEVICE_INFORMATION info;
733 WCHAR *p;
734 enum fs_type type = FS_UNKNOWN;
735 BOOL ret = FALSE;
736
737 if (!root) root = default_rootW;
738 if (!RtlDosPathNameToNtPathName_U( root, &nt_name, NULL, NULL ))
739 {
740 SetLastError( ERROR_PATH_NOT_FOUND );
741 return FALSE;
742 }
743 /* there must be exactly one backslash in the name, at the end */
744 p = memchrW( nt_name.Buffer + 4, '\\', (nt_name.Length - 4) / sizeof(WCHAR) );
745 if (p != nt_name.Buffer + nt_name.Length / sizeof(WCHAR) - 1)
746 {
747 /* check if root contains an explicit subdir */
748 if (root[0] && root[1] == ':') root += 2;
749 while (*root == '\\') root++;
750 if (strchrW( root, '\\' ))
751 SetLastError( ERROR_DIR_NOT_ROOT );
752 else
753 SetLastError( ERROR_INVALID_NAME );
754 goto done;
755 }
756
757 /* try to open the device */
758
759 attr.Length = sizeof(attr);
760 attr.RootDirectory = 0;
761 attr.Attributes = OBJ_CASE_INSENSITIVE;
762 attr.ObjectName = &nt_name;
763 attr.SecurityDescriptor = NULL;
764 attr.SecurityQualityOfService = NULL;
765
766 nt_name.Length -= sizeof(WCHAR); /* without trailing slash */
767 status = NtOpenFile( &handle, GENERIC_READ, &attr, &io, FILE_SHARE_READ | FILE_SHARE_WRITE,
768 FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT );
769 nt_name.Length += sizeof(WCHAR);
770
771 if (status == STATUS_SUCCESS)
772 {
773 BYTE superblock[SUPERBLOCK_SIZE];
774 CDROM_TOC toc;
775 DWORD br;
776
777 /* check for audio CD */
778 /* FIXME: we only check the first track for now */
779 if (DeviceIoControl( handle, IOCTL_CDROM_READ_TOC, NULL, 0, &toc, sizeof(toc), &br, 0 ))
780 {
781 if (!(toc.TrackData[0].Control & 0x04)) /* audio track */
782 {
783 TRACE( "%s: found audio CD\n", debugstr_w(nt_name.Buffer) );
784 if (label) lstrcpynW( label, audiocdW, label_len );
785 if (serial) *serial = VOLUME_GetAudioCDSerial( &toc );
786 CloseHandle( handle );
787 type = FS_ISO9660;
788 goto fill_fs_info;
789 }
790 type = VOLUME_ReadCDSuperblock( handle, superblock );
791 }
792 else
793 {
794 type = VOLUME_ReadFATSuperblock( handle, superblock );
795 if (type == FS_UNKNOWN) type = VOLUME_ReadCDSuperblock( handle, superblock );
796 }
797 TRACE( "%s: found fs type %d\n", debugstr_w(nt_name.Buffer), type );
798 if (type == FS_ERROR)
799 {
800 CloseHandle( handle );
801 goto done;
802 }
803
804 if (label && label_len) VOLUME_GetSuperblockLabel( &nt_name, handle, type, superblock, label, label_len );
805 if (serial) *serial = VOLUME_GetSuperblockSerial( &nt_name, handle, type, superblock );
806 CloseHandle( handle );
807 goto fill_fs_info;
808 }
809 else TRACE( "cannot open device %s: %x\n", debugstr_w(nt_name.Buffer), status );
810
811 /* we couldn't open the device, fallback to default strategy */
812
813 status = NtOpenFile( &handle, 0, &attr, &io, 0, FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT );
814 if (status != STATUS_SUCCESS)
815 {
816 SetLastError( RtlNtStatusToDosError(status) );
817 goto done;
818 }
819 status = NtQueryVolumeInformationFile( handle, &io, &info, sizeof(info), FileFsDeviceInformation );
820 NtClose( handle );
821 if (status != STATUS_SUCCESS)
822 {
823 SetLastError( RtlNtStatusToDosError(status) );
824 goto done;
825 }
826 if (info.DeviceType == FILE_DEVICE_CD_ROM_FILE_SYSTEM) type = FS_ISO9660;
827
828 if (label && label_len) get_filesystem_label( &nt_name, label, label_len );
829 if (serial) *serial = get_filesystem_serial( &nt_name );
830
831 fill_fs_info: /* now fill in the information that depends on the file system type */
832
833 switch(type)
834 {
835 case FS_ISO9660:
836 if (fsname) lstrcpynW( fsname, cdfsW, fsname_len );
837 if (filename_len) *filename_len = 221;
838 if (flags) *flags = FILE_READ_ONLY_VOLUME;
839 break;
840 case FS_UDF:
841 if (fsname) lstrcpynW( fsname, udfW, fsname_len );
842 if (filename_len) *filename_len = 255;
843 if (flags)
844 *flags = FILE_READ_ONLY_VOLUME | FILE_UNICODE_ON_DISK | FILE_CASE_SENSITIVE_SEARCH;
845 break;
846 case FS_FAT1216:
847 if (fsname) lstrcpynW( fsname, fatW, fsname_len );
848 case FS_FAT32:
849 if (type == FS_FAT32 && fsname) lstrcpynW( fsname, fat32W, fsname_len );
850 if (filename_len) *filename_len = 255;
851 if (flags) *flags = FILE_CASE_PRESERVED_NAMES; /* FIXME */
852 break;
853 default:
854 if (fsname) lstrcpynW( fsname, ntfsW, fsname_len );
855 if (filename_len) *filename_len = 255;
856 if (flags) *flags = FILE_CASE_PRESERVED_NAMES | FILE_PERSISTENT_ACLS;
857 break;
858 }
859 ret = TRUE;
860
861 done:
862 RtlFreeUnicodeString( &nt_name );
863 return ret;
864 }
865
866
867 /***********************************************************************
868 * GetVolumeInformationA (KERNEL32.@)
869 */
870 BOOL WINAPI GetVolumeInformationA( LPCSTR root, LPSTR label,
871 DWORD label_len, DWORD *serial,
872 DWORD *filename_len, DWORD *flags,
873 LPSTR fsname, DWORD fsname_len )
874 {
875 WCHAR *rootW = NULL;
876 LPWSTR labelW, fsnameW;
877 BOOL ret;
878
879 if (root && !(rootW = FILE_name_AtoW( root, FALSE ))) return FALSE;
880
881 labelW = label ? HeapAlloc(GetProcessHeap(), 0, label_len * sizeof(WCHAR)) : NULL;
882 fsnameW = fsname ? HeapAlloc(GetProcessHeap(), 0, fsname_len * sizeof(WCHAR)) : NULL;
883
884 if ((ret = GetVolumeInformationW(rootW, labelW, label_len, serial,
885 filename_len, flags, fsnameW, fsname_len)))
886 {
887 if (label) FILE_name_WtoA( labelW, -1, label, label_len );
888 if (fsname) FILE_name_WtoA( fsnameW, -1, fsname, fsname_len );
889 }
890
891 HeapFree( GetProcessHeap(), 0, labelW );
892 HeapFree( GetProcessHeap(), 0, fsnameW );
893 return ret;
894 }
895
896
897
898 /***********************************************************************
899 * SetVolumeLabelW (KERNEL32.@)
900 */
901 BOOL WINAPI SetVolumeLabelW( LPCWSTR root, LPCWSTR label )
902 {
903 WCHAR device[] = {'\\','\\','.','\\','A',':',0};
904 HANDLE handle;
905 enum fs_type type = FS_UNKNOWN;
906
907 if (!root)
908 {
909 WCHAR path[MAX_PATH];
910 GetCurrentDirectoryW( MAX_PATH, path );
911 device[4] = path[0];
912 }
913 else
914 {
915 if (!root[0] || root[1] != ':')
916 {
917 SetLastError( ERROR_INVALID_NAME );
918 return FALSE;
919 }
920 device[4] = root[0];
921 }
922
923 /* try to open the device */
924
925 handle = CreateFileW( device, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE,
926 NULL, OPEN_EXISTING, 0, 0 );
927 if (handle != INVALID_HANDLE_VALUE)
928 {
929 BYTE superblock[SUPERBLOCK_SIZE];
930
931 type = VOLUME_ReadFATSuperblock( handle, superblock );
932 if (type == FS_UNKNOWN) type = VOLUME_ReadCDSuperblock( handle, superblock );
933 CloseHandle( handle );
934 if (type != FS_UNKNOWN)
935 {
936 /* we can't set the label on FAT or CDROM file systems */
937 TRACE( "cannot set label on device %s type %d\n", debugstr_w(device), type );
938 SetLastError( ERROR_ACCESS_DENIED );
939 return FALSE;
940 }
941 }
942 else
943 {
944 TRACE( "cannot open device %s: err %d\n", debugstr_w(device), GetLastError() );
945 if (GetLastError() == ERROR_ACCESS_DENIED) return FALSE;
946 }
947
948 /* we couldn't open the device, fallback to default strategy */
949
950 switch(GetDriveTypeW( root ))
951 {
952 case DRIVE_UNKNOWN:
953 case DRIVE_NO_ROOT_DIR:
954 SetLastError( ERROR_NOT_READY );
955 break;
956 case DRIVE_REMOVABLE:
957 case DRIVE_FIXED:
958 {
959 WCHAR labelW[] = {'A',':','\\','.','w','i','n','d','o','w','s','-','l','a','b','e','l',0};
960
961 labelW[0] = device[4];
962
963 if (!label[0]) /* delete label file when setting an empty label */
964 return DeleteFileW( labelW ) || GetLastError() == ERROR_FILE_NOT_FOUND;
965
966 handle = CreateFileW( labelW, GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL,
967 CREATE_ALWAYS, 0, 0 );
968 if (handle != INVALID_HANDLE_VALUE)
969 {
970 char buffer[64];
971 DWORD size;
972
973 if (!WideCharToMultiByte( CP_UNIXCP, 0, label, -1, buffer, sizeof(buffer)-1, NULL, NULL ))
974 buffer[sizeof(buffer)-2] = 0;
975 strcat( buffer, "\n" );
976 WriteFile( handle, buffer, strlen(buffer), &size, NULL );
977 CloseHandle( handle );
978 return TRUE;
979 }
980 break;
981 }
982 case DRIVE_REMOTE:
983 case DRIVE_RAMDISK:
984 case DRIVE_CDROM:
985 SetLastError( ERROR_ACCESS_DENIED );
986 break;
987 }
988 return FALSE;
989 }
990
991 /***********************************************************************
992 * SetVolumeLabelA (KERNEL32.@)
993 */
994 BOOL WINAPI SetVolumeLabelA(LPCSTR root, LPCSTR volname)
995 {
996 WCHAR *rootW = NULL, *volnameW = NULL;
997 BOOL ret;
998
999 if (root && !(rootW = FILE_name_AtoW( root, FALSE ))) return FALSE;
1000 if (volname && !(volnameW = FILE_name_AtoW( volname, TRUE ))) return FALSE;
1001 ret = SetVolumeLabelW( rootW, volnameW );
1002 HeapFree( GetProcessHeap(), 0, volnameW );
1003 return ret;
1004 }
1005
1006
1007 /***********************************************************************
1008 * GetVolumeNameForVolumeMountPointA (KERNEL32.@)
1009 */
1010 BOOL WINAPI GetVolumeNameForVolumeMountPointA( LPCSTR path, LPSTR volume, DWORD size )
1011 {
1012 BOOL ret;
1013 WCHAR volumeW[50], *pathW = NULL;
1014 DWORD len = min( sizeof(volumeW) / sizeof(WCHAR), size );
1015
1016 TRACE("(%s, %p, %x)\n", debugstr_a(path), volume, size);
1017
1018 if (!path || !(pathW = FILE_name_AtoW( path, TRUE )))
1019 return FALSE;
1020
1021 if ((ret = GetVolumeNameForVolumeMountPointW( pathW, volumeW, len )))
1022 FILE_name_WtoA( volumeW, -1, volume, len );
1023
1024 HeapFree( GetProcessHeap(), 0, pathW );
1025 return ret;
1026 }
1027
1028 /***********************************************************************
1029 * GetVolumeNameForVolumeMountPointW (KERNEL32.@)
1030 */
1031 BOOL WINAPI GetVolumeNameForVolumeMountPointW( LPCWSTR path, LPWSTR volume, DWORD size )
1032 {
1033 static const WCHAR prefixW[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\',0};
1034 static const WCHAR volumeW[] = {'\\','?','?','\\','V','o','l','u','m','e','{',0};
1035 static const WCHAR trailingW[] = {'\\',0};
1036
1037 MOUNTMGR_MOUNT_POINT *input = NULL, *o1;
1038 MOUNTMGR_MOUNT_POINTS *output = NULL;
1039 WCHAR *p;
1040 char *r;
1041 DWORD i, i_size = 1024, o_size = 1024;
1042 WCHAR *nonpersist_name;
1043 WCHAR symlink_name[MAX_PATH];
1044 NTSTATUS status;
1045 HANDLE mgr = INVALID_HANDLE_VALUE;
1046 BOOL ret = FALSE;
1047
1048 TRACE("(%s, %p, %x)\n", debugstr_w(path), volume, size);
1049 if (path[lstrlenW(path)-1] != '\\')
1050 {
1051 SetLastError( ERROR_INVALID_NAME );
1052 return FALSE;
1053 }
1054
1055 if (size < 50)
1056 {
1057 SetLastError( ERROR_FILENAME_EXCED_RANGE );
1058 return FALSE;
1059 }
1060 /* if length of input is > 3 then it must be a mounted folder */
1061 if (lstrlenW(path) > 3)
1062 {
1063 FIXME("Mounted Folders are not yet supported\n");
1064 SetLastError( ERROR_NOT_A_REPARSE_POINT );
1065 return FALSE;
1066 }
1067
1068 mgr = CreateFileW( MOUNTMGR_DOS_DEVICE_NAME, 0, FILE_SHARE_READ,
1069 NULL, OPEN_EXISTING, 0, 0 );
1070 if (mgr == INVALID_HANDLE_VALUE) return FALSE;
1071
1072 if (!(input = HeapAlloc( GetProcessHeap(), 0, i_size )))
1073 {
1074 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1075 goto err_ret;
1076 }
1077
1078 if (!(output = HeapAlloc( GetProcessHeap(), 0, o_size )))
1079 {
1080 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1081 goto err_ret;
1082 }
1083
1084 /* construct the symlink name as "\DosDevices\C:" */
1085 lstrcpyW( symlink_name, prefixW );
1086 lstrcatW( symlink_name, path );
1087 symlink_name[lstrlenW(symlink_name)-1] = 0;
1088
1089 /* Take the mount point and get the "nonpersistent name" */
1090 /* We will then take that and get the volume name */
1091 nonpersist_name = (WCHAR *)(input + 1);
1092 status = read_nt_symlink( symlink_name, nonpersist_name, i_size - sizeof(*input) );
1093 TRACE("read_nt_symlink got stat=%x, for %s, got <%s>\n", status,
1094 debugstr_w(symlink_name), debugstr_w(nonpersist_name));
1095 if (status != STATUS_SUCCESS)
1096 {
1097 SetLastError( ERROR_FILE_NOT_FOUND );
1098 goto err_ret;
1099 }
1100
1101 /* Now take the "nonpersistent name" and ask the mountmgr */
1102 /* to give us all the mount points. One of them will be */
1103 /* the volume name (format of \??\Volume{). */
1104 memset( input, 0, sizeof(*input) ); /* clear all input parameters */
1105 input->DeviceNameOffset = sizeof(*input);
1106 input->DeviceNameLength = lstrlenW( nonpersist_name) * sizeof(WCHAR);
1107 i_size = input->DeviceNameOffset + input->DeviceNameLength;
1108
1109 output->Size = o_size;
1110
1111 /* now get the true volume name from the mountmgr */
1112 if (!DeviceIoControl( mgr, IOCTL_MOUNTMGR_QUERY_POINTS, input, i_size,
1113 output, o_size, NULL, NULL ))
1114 goto err_ret;
1115
1116 /* Verify and return the data, note string is not null terminated */
1117 TRACE("found %d matching mount points\n", output->NumberOfMountPoints);
1118 if (output->NumberOfMountPoints < 1)
1119 {
1120 SetLastError( ERROR_NO_VOLUME_ID );
1121 goto err_ret;
1122 }
1123 o1 = &output->MountPoints[0];
1124
1125 /* look for the volume name in returned values */
1126 for(i=0;i<output->NumberOfMountPoints;i++)
1127 {
1128 p = (WCHAR*)((char *)output + o1->SymbolicLinkNameOffset);
1129 r = (char *)output + o1->UniqueIdOffset;
1130 TRACE("found symlink=%s, unique=%s, devname=%s\n",
1131 debugstr_wn(p, o1->SymbolicLinkNameLength/sizeof(WCHAR)),
1132 debugstr_an(r, o1->UniqueIdLength),
1133 debugstr_wn((WCHAR*)((char *)output + o1->DeviceNameOffset),
1134 o1->DeviceNameLength/sizeof(WCHAR)));
1135
1136 if (!strncmpW( p, volumeW, (sizeof(volumeW)-1)/sizeof(WCHAR) ))
1137 {
1138 /* is there space in the return variable ?? */
1139 if ((o1->SymbolicLinkNameLength/sizeof(WCHAR))+2 > size)
1140 {
1141 SetLastError( ERROR_FILENAME_EXCED_RANGE );
1142 goto err_ret;
1143 }
1144 memcpy( volume, p, o1->SymbolicLinkNameLength );
1145 volume[o1->SymbolicLinkNameLength / sizeof(WCHAR)] = 0;
1146 lstrcatW( volume, trailingW );
1147 /* change second char from '?' to '\' */
1148 volume[1] = '\\';
1149 ret = TRUE;
1150 break;
1151 }
1152 o1++;
1153 }
1154
1155 err_ret:
1156 HeapFree( GetProcessHeap(), 0, input );
1157 HeapFree( GetProcessHeap(), 0, output );
1158 CloseHandle( mgr );
1159 return ret;
1160 }
1161
1162 /***********************************************************************
1163 * DefineDosDeviceW (KERNEL32.@)
1164 */
1165 BOOL WINAPI DefineDosDeviceW( DWORD flags, LPCWSTR devname, LPCWSTR targetpath )
1166 {
1167 DWORD len, dosdev;
1168 BOOL ret = FALSE;
1169 char *path = NULL, *target, *p;
1170
1171 TRACE("%x, %s, %s\n", flags, debugstr_w(devname), debugstr_w(targetpath));
1172
1173 if (!(flags & DDD_REMOVE_DEFINITION))
1174 {
1175 if (!(flags & DDD_RAW_TARGET_PATH))
1176 {
1177 FIXME( "(0x%08x,%s,%s) DDD_RAW_TARGET_PATH flag not set, not supported yet\n",
1178 flags, debugstr_w(devname), debugstr_w(targetpath) );
1179 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
1180 return FALSE;
1181 }
1182
1183 len = WideCharToMultiByte( CP_UNIXCP, 0, targetpath, -1, NULL, 0, NULL, NULL );
1184 if ((target = HeapAlloc( GetProcessHeap(), 0, len )))
1185 {
1186 WideCharToMultiByte( CP_UNIXCP, 0, targetpath, -1, target, len, NULL, NULL );
1187 for (p = target; *p; p++) if (*p == '\\') *p = '/';
1188 }
1189 else
1190 {
1191 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1192 return FALSE;
1193 }
1194 }
1195 else target = NULL;
1196
1197 /* first check for a DOS device */
1198
1199 if ((dosdev = RtlIsDosDeviceName_U( devname )))
1200 {
1201 WCHAR name[5];
1202
1203 memcpy( name, devname + HIWORD(dosdev)/sizeof(WCHAR), LOWORD(dosdev) );
1204 name[LOWORD(dosdev)/sizeof(WCHAR)] = 0;
1205 path = get_dos_device_path( name );
1206 }
1207 else if (isalphaW(devname[0]) && devname[1] == ':' && !devname[2]) /* drive mapping */
1208 {
1209 path = get_dos_device_path( devname );
1210 }
1211 else SetLastError( ERROR_FILE_NOT_FOUND );
1212
1213 if (path)
1214 {
1215 if (target)
1216 {
1217 TRACE( "creating symlink %s -> %s\n", path, target );
1218 unlink( path );
1219 if (!symlink( target, path )) ret = TRUE;
1220 else FILE_SetDosError();
1221 }
1222 else
1223 {
1224 TRACE( "removing symlink %s\n", path );
1225 if (!unlink( path )) ret = TRUE;
1226 else FILE_SetDosError();
1227 }
1228 HeapFree( GetProcessHeap(), 0, path );
1229 }
1230 HeapFree( GetProcessHeap(), 0, target );
1231 return ret;
1232 }
1233
1234
1235 /***********************************************************************
1236 * DefineDosDeviceA (KERNEL32.@)
1237 */
1238 BOOL WINAPI DefineDosDeviceA(DWORD flags, LPCSTR devname, LPCSTR targetpath)
1239 {
1240 WCHAR *devW, *targetW = NULL;
1241 BOOL ret;
1242
1243 if (!(devW = FILE_name_AtoW( devname, FALSE ))) return FALSE;
1244 if (targetpath && !(targetW = FILE_name_AtoW( targetpath, TRUE ))) return FALSE;
1245 ret = DefineDosDeviceW(flags, devW, targetW);
1246 HeapFree( GetProcessHeap(), 0, targetW );
1247 return ret;
1248 }
1249
1250
1251 /***********************************************************************
1252 * QueryDosDeviceW (KERNEL32.@)
1253 *
1254 * returns array of strings terminated by \0, terminated by \0
1255 */
1256 DWORD WINAPI QueryDosDeviceW( LPCWSTR devname, LPWSTR target, DWORD bufsize )
1257 {
1258 static const WCHAR auxW[] = {'A','U','X',0};
1259 static const WCHAR nulW[] = {'N','U','L',0};
1260 static const WCHAR prnW[] = {'P','R','N',0};
1261 static const WCHAR comW[] = {'C','O','M',0};
1262 static const WCHAR lptW[] = {'L','P','T',0};
1263 static const WCHAR com0W[] = {'\\','?','?','\\','C','O','M','0',0};
1264 static const WCHAR com1W[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\','C','O','M','1',0,0};
1265 static const WCHAR lpt1W[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\','L','P','T','1',0,0};
1266 static const WCHAR dosdevW[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\',0};
1267
1268 UNICODE_STRING nt_name;
1269 ANSI_STRING unix_name;
1270 WCHAR nt_buffer[10];
1271 NTSTATUS status;
1272
1273 if (!bufsize)
1274 {
1275 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1276 return 0;
1277 }
1278
1279 if (devname)
1280 {
1281 WCHAR *p, name[5];
1282 char *path, *link;
1283 DWORD dosdev, ret = 0;
1284
1285 if ((dosdev = RtlIsDosDeviceName_U( devname )))
1286 {
1287 memcpy( name, devname + HIWORD(dosdev)/sizeof(WCHAR), LOWORD(dosdev) );
1288 name[LOWORD(dosdev)/sizeof(WCHAR)] = 0;
1289 }
1290 else
1291 {
1292 WCHAR *buffer;
1293
1294 if (!(buffer = HeapAlloc( GetProcessHeap(), 0, sizeof(dosdevW) + strlenW(devname)*sizeof(WCHAR) )))
1295 {
1296 SetLastError( ERROR_OUTOFMEMORY );
1297 return 0;
1298 }
1299 memcpy( buffer, dosdevW, sizeof(dosdevW) );
1300 strcatW( buffer, devname );
1301 status = read_nt_symlink( buffer, target, bufsize );
1302 HeapFree( GetProcessHeap(), 0, buffer );
1303 if (status)
1304 {
1305 SetLastError( RtlNtStatusToDosError(status) );
1306 return 0;
1307 }
1308 ret = strlenW( target ) + 1;
1309 goto done;
1310 }
1311
1312 /* FIXME: should read NT symlink for all devices */
1313
1314 if (!(path = get_dos_device_path( name ))) return 0;
1315 link = read_symlink( path );
1316 HeapFree( GetProcessHeap(), 0, path );
1317
1318 if (link)
1319 {
1320 ret = MultiByteToWideChar( CP_UNIXCP, 0, link, -1, target, bufsize );
1321 HeapFree( GetProcessHeap(), 0, link );
1322 }
1323 else if (dosdev) /* look for device defaults */
1324 {
1325 if (!strcmpiW( name, auxW ))
1326 {
1327 if (bufsize >= sizeof(com1W)/sizeof(WCHAR))
1328 {
1329 memcpy( target, com1W, sizeof(com1W) );
1330 ret = sizeof(com1W)/sizeof(WCHAR);
1331 }
1332 else SetLastError( ERROR_INSUFFICIENT_BUFFER );
1333 return ret;
1334 }
1335 if (!strcmpiW( name, prnW ))
1336 {
1337 if (bufsize >= sizeof(lpt1W)/sizeof(WCHAR))
1338 {
1339 memcpy( target, lpt1W, sizeof(lpt1W) );
1340 ret = sizeof(lpt1W)/sizeof(WCHAR);
1341 }
1342 else SetLastError( ERROR_INSUFFICIENT_BUFFER );
1343 return ret;
1344 }
1345
1346 nt_buffer[0] = '\\';
1347 nt_buffer[1] = '?';
1348 nt_buffer[2] = '?';
1349 nt_buffer[3] = '\\';
1350 strcpyW( nt_buffer + 4, name );
1351 RtlInitUnicodeString( &nt_name, nt_buffer );
1352 status = wine_nt_to_unix_file_name( &nt_name, &unix_name, FILE_OPEN, TRUE );
1353 if (status) SetLastError( RtlNtStatusToDosError(status) );
1354 else
1355 {
1356 ret = MultiByteToWideChar( CP_UNIXCP, 0, unix_name.Buffer, -1, target, bufsize );
1357 RtlFreeAnsiString( &unix_name );
1358 }
1359 }
1360 done:
1361 if (ret)
1362 {
1363 if (ret < bufsize) target[ret++] = 0; /* add an extra null */
1364 for (p = target; *p; p++) if (*p == '/') *p = '\\';
1365 }
1366
1367 return ret;
1368 }
1369 else /* return a list of all devices */
1370 {
1371 OBJECT_ATTRIBUTES attr;
1372 HANDLE handle;
1373 WCHAR *p = target;
1374 int i;
1375
1376 if (bufsize <= (sizeof(auxW)+sizeof(nulW)+sizeof(prnW))/sizeof(WCHAR))
1377 {
1378 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1379 return 0;
1380 }
1381
1382 /* FIXME: these should be NT symlinks too */
1383
1384 memcpy( p, auxW, sizeof(auxW) );
1385 p += sizeof(auxW) / sizeof(WCHAR);
1386 memcpy( p, nulW, sizeof(nulW) );
1387 p += sizeof(nulW) / sizeof(WCHAR);
1388 memcpy( p, prnW, sizeof(prnW) );
1389 p += sizeof(prnW) / sizeof(WCHAR);
1390
1391 strcpyW( nt_buffer, com0W );
1392 RtlInitUnicodeString( &nt_name, nt_buffer );
1393
1394 for (i = 1; i <= 9; i++)
1395 {
1396 nt_buffer[7] = '0' + i;
1397 if (!wine_nt_to_unix_file_name( &nt_name, &unix_name, FILE_OPEN, TRUE ))
1398 {
1399 RtlFreeAnsiString( &unix_name );
1400 if (p + 5 >= target + bufsize)
1401 {
1402 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1403 return 0;
1404 }
1405 strcpyW( p, comW );
1406 p[3] = '0' + i;
1407 p[4] = 0;
1408 p += 5;
1409 }
1410 }
1411 strcpyW( nt_buffer + 4, lptW );
1412 for (i = 1; i <= 9; i++)
1413 {
1414 nt_buffer[7] = '0' + i;
1415 if (!wine_nt_to_unix_file_name( &nt_name, &unix_name, FILE_OPEN, TRUE ))
1416 {
1417 RtlFreeAnsiString( &unix_name );
1418 if (p + 5 >= target + bufsize)
1419 {
1420 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1421 return 0;
1422 }
1423 strcpyW( p, lptW );
1424 p[3] = '0' + i;
1425 p[4] = 0;
1426 p += 5;
1427 }
1428 }
1429
1430 RtlInitUnicodeString( &nt_name, dosdevW );
1431 nt_name.Length -= sizeof(WCHAR); /* without trailing slash */
1432 attr.Length = sizeof(attr);
1433 attr.RootDirectory = 0;
1434 attr.ObjectName = &nt_name;
1435 attr.Attributes = OBJ_CASE_INSENSITIVE;
1436 attr.SecurityDescriptor = NULL;
1437 attr.SecurityQualityOfService = NULL;
1438 status = NtOpenDirectoryObject( &handle, FILE_LIST_DIRECTORY, &attr );
1439 if (!status)
1440 {
1441 char data[1024];
1442 DIRECTORY_BASIC_INFORMATION *info = (DIRECTORY_BASIC_INFORMATION *)data;
1443 ULONG ctx = 0, len;
1444
1445 while (!NtQueryDirectoryObject( handle, info, sizeof(data), 1, 0, &ctx, &len ))
1446 {
1447 if (p + info->ObjectName.Length/sizeof(WCHAR) + 1 >= target + bufsize)
1448 {
1449 SetLastError( ERROR_INSUFFICIENT_BUFFER );
1450 NtClose( handle );
1451 return 0;
1452 }
1453 memcpy( p, info->ObjectName.Buffer, info->ObjectName.Length );
1454 p += info->ObjectName.Length/sizeof(WCHAR);
1455 *p++ = 0;
1456 }
1457 NtClose( handle );
1458 }
1459
1460 *p++ = 0; /* terminating null */
1461 return p - target;
1462 }
1463 }
1464
1465
1466 /***********************************************************************
1467 * QueryDosDeviceA (KERNEL32.@)
1468 *
1469 * returns array of strings terminated by \0, terminated by \0
1470 */
1471 DWORD WINAPI QueryDosDeviceA( LPCSTR devname, LPSTR target, DWORD bufsize )
1472 {
1473 DWORD ret = 0, retW;
1474 WCHAR *devnameW = NULL;
1475 LPWSTR targetW;
1476
1477 if (devname && !(devnameW = FILE_name_AtoW( devname, FALSE ))) return 0;
1478
1479 targetW = HeapAlloc( GetProcessHeap(),0, bufsize * sizeof(WCHAR) );
1480 if (!targetW)
1481 {
1482 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1483 return 0;
1484 }
1485
1486 retW = QueryDosDeviceW(devnameW, targetW, bufsize);
1487
1488 ret = FILE_name_WtoA( targetW, retW, target, bufsize );
1489
1490 HeapFree(GetProcessHeap(), 0, targetW);
1491 return ret;
1492 }
1493
1494
1495 /***********************************************************************
1496 * GetLogicalDrives (KERNEL32.@)
1497 */
1498 DWORD WINAPI GetLogicalDrives(void)
1499 {
1500 static const WCHAR dosdevW[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\',0};
1501 OBJECT_ATTRIBUTES attr;
1502 UNICODE_STRING nt_name;
1503 DWORD bitmask = 0;
1504 NTSTATUS status;
1505 HANDLE handle;
1506
1507 RtlInitUnicodeString( &nt_name, dosdevW );
1508 nt_name.Length -= sizeof(WCHAR); /* without trailing slash */
1509 attr.Length = sizeof(attr);
1510 attr.RootDirectory = 0;
1511 attr.ObjectName = &nt_name;
1512 attr.Attributes = OBJ_CASE_INSENSITIVE;
1513 attr.SecurityDescriptor = NULL;
1514 attr.SecurityQualityOfService = NULL;
1515 status = NtOpenDirectoryObject( &handle, FILE_LIST_DIRECTORY, &attr );
1516 if (!status)
1517 {
1518 char data[1024];
1519 DIRECTORY_BASIC_INFORMATION *info = (DIRECTORY_BASIC_INFORMATION *)data;
1520 ULONG ctx = 0, len;
1521
1522 while (!NtQueryDirectoryObject( handle, info, sizeof(data), 1, 0, &ctx, &len ))
1523 if(info->ObjectName.Length == 2*sizeof(WCHAR) && info->ObjectName.Buffer[1] == ':')
1524 bitmask |= 1 << (info->ObjectName.Buffer[0] - 'A');
1525
1526 NtClose( handle );
1527 }
1528
1529 return bitmask;
1530 }
1531
1532
1533 /***********************************************************************
1534 * GetLogicalDriveStringsA (KERNEL32.@)
1535 */
1536 UINT WINAPI GetLogicalDriveStringsA( UINT len, LPSTR buffer )
1537 {
1538 DWORD drives = GetLogicalDrives();
1539 UINT drive, count;
1540
1541 for (drive = count = 0; drive < 26; drive++) if (drives & (1 << drive)) count++;
1542 if ((count * 4) + 1 > len) return count * 4 + 1;
1543
1544 for (drive = 0; drive < 26; drive++)
1545 {
1546 if (drives & (1 << drive))
1547 {
1548 *buffer++ = 'A' + drive;
1549 *buffer++ = ':';
1550 *buffer++ = '\\';
1551 *buffer++ = 0;
1552 }
1553 }
1554 *buffer = 0;
1555 return count * 4;
1556 }
1557
1558
1559 /***********************************************************************
1560 * GetLogicalDriveStringsW (KERNEL32.@)
1561 */
1562 UINT WINAPI GetLogicalDriveStringsW( UINT len, LPWSTR buffer )
1563 {
1564 DWORD drives = GetLogicalDrives();
1565 UINT drive, count;
1566
1567 for (drive = count = 0; drive < 26; drive++) if (drives & (1 << drive)) count++;
1568 if ((count * 4) + 1 > len) return count * 4 + 1;
1569
1570 for (drive = 0; drive < 26; drive++)
1571 {
1572 if (drives & (1 << drive))
1573 {
1574 *buffer++ = 'A' + drive;
1575 *buffer++ = ':';
1576 *buffer++ = '\\';
1577 *buffer++ = 0;
1578 }
1579 }
1580 *buffer = 0;
1581 return count * 4;
1582 }
1583
1584
1585 /***********************************************************************
1586 * GetDriveTypeW (KERNEL32.@)
1587 *
1588 * Returns the type of the disk drive specified. If root is NULL the
1589 * root of the current directory is used.
1590 *
1591 * RETURNS
1592 *
1593 * Type of drive (from Win32 SDK):
1594 *
1595 * DRIVE_UNKNOWN unable to find out anything about the drive
1596 * DRIVE_NO_ROOT_DIR nonexistent root dir
1597 * DRIVE_REMOVABLE the disk can be removed from the machine
1598 * DRIVE_FIXED the disk cannot be removed from the machine
1599 * DRIVE_REMOTE network disk
1600 * DRIVE_CDROM CDROM drive
1601 * DRIVE_RAMDISK virtual disk in RAM
1602 */
1603 UINT WINAPI GetDriveTypeW(LPCWSTR root) /* [in] String describing drive */
1604 {
1605 FILE_FS_DEVICE_INFORMATION info;
1606 IO_STATUS_BLOCK io;
1607 NTSTATUS status;
1608 HANDLE handle;
1609 UINT ret;
1610
1611 if (!open_device_root( root, &handle )) return DRIVE_NO_ROOT_DIR;
1612
1613 status = NtQueryVolumeInformationFile( handle, &io, &info, sizeof(info), FileFsDeviceInformation );
1614 NtClose( handle );
1615 if (status != STATUS_SUCCESS)
1616 {
1617 SetLastError( RtlNtStatusToDosError(status) );
1618 ret = DRIVE_UNKNOWN;
1619 }
1620 else
1621 {
1622 switch (info.DeviceType)
1623 {
1624 case FILE_DEVICE_CD_ROM_FILE_SYSTEM: ret = DRIVE_CDROM; break;
1625 case FILE_DEVICE_VIRTUAL_DISK: ret = DRIVE_RAMDISK; break;
1626 case FILE_DEVICE_NETWORK_FILE_SYSTEM: ret = DRIVE_REMOTE; break;
1627 case FILE_DEVICE_DISK_FILE_SYSTEM:
1628 if (info.Characteristics & FILE_REMOTE_DEVICE) ret = DRIVE_REMOTE;
1629 else if (info.Characteristics & FILE_REMOVABLE_MEDIA) ret = DRIVE_REMOVABLE;
1630 else if ((ret = get_mountmgr_drive_type( root )) == DRIVE_UNKNOWN) ret = DRIVE_FIXED;
1631 break;
1632 default:
1633 ret = DRIVE_UNKNOWN;
1634 break;
1635 }
1636 }
1637 TRACE( "%s -> %d\n", debugstr_w(root), ret );
1638 return ret;
1639 }
1640
1641
1642 /***********************************************************************
1643 * GetDriveTypeA (KERNEL32.@)
1644 *
1645 * See GetDriveTypeW.
1646 */
1647 UINT WINAPI GetDriveTypeA( LPCSTR root )
1648 {
1649 WCHAR *rootW = NULL;
1650
1651 if (root && !(rootW = FILE_name_AtoW( root, FALSE ))) return DRIVE_NO_ROOT_DIR;
1652 return GetDriveTypeW( rootW );
1653 }
1654
1655
1656 /***********************************************************************
1657 * GetDiskFreeSpaceExW (KERNEL32.@)
1658 *
1659 * This function is used to acquire the size of the available and
1660 * total space on a logical volume.
1661 *
1662 * RETURNS
1663 *
1664 * Zero on failure, nonzero upon success. Use GetLastError to obtain
1665 * detailed error information.
1666 *
1667 */
1668 BOOL WINAPI GetDiskFreeSpaceExW( LPCWSTR root, PULARGE_INTEGER avail,
1669 PULARGE_INTEGER total, PULARGE_INTEGER totalfree )
1670 {
1671 FILE_FS_SIZE_INFORMATION info;
1672 IO_STATUS_BLOCK io;
1673 NTSTATUS status;
1674 HANDLE handle;
1675 UINT units;
1676
1677 TRACE( "%s,%p,%p,%p\n", debugstr_w(root), avail, total, totalfree );
1678
1679 if (!open_device_root( root, &handle )) return FALSE;
1680
1681 status = NtQueryVolumeInformationFile( handle, &io, &info, sizeof(info), FileFsSizeInformation );
1682 NtClose( handle );
1683 if (status != STATUS_SUCCESS)
1684 {
1685 SetLastError( RtlNtStatusToDosError(status) );
1686 return FALSE;
1687 }
1688
1689 units = info.SectorsPerAllocationUnit * info.BytesPerSector;
1690 if (total) total->QuadPart = info.TotalAllocationUnits.QuadPart * units;
1691 if (totalfree) totalfree->QuadPart = info.AvailableAllocationUnits.QuadPart * units;
1692 /* FIXME: this one should take quotas into account */
1693 if (avail) avail->QuadPart = info.AvailableAllocationUnits.QuadPart * units;
1694 return TRUE;
1695 }
1696
1697
1698 /***********************************************************************
1699 * GetDiskFreeSpaceExA (KERNEL32.@)
1700 *
1701 * See GetDiskFreeSpaceExW.
1702 */
1703 BOOL WINAPI GetDiskFreeSpaceExA( LPCSTR root, PULARGE_INTEGER avail,
1704 PULARGE_INTEGER total, PULARGE_INTEGER totalfree )
1705 {
1706 WCHAR *rootW = NULL;
1707
1708 if (root && !(rootW = FILE_name_AtoW( root, FALSE ))) return FALSE;
1709 return GetDiskFreeSpaceExW( rootW, avail, total, totalfree );
1710 }
1711
1712
1713 /***********************************************************************
1714 * GetDiskFreeSpaceW (KERNEL32.@)
1715 */
1716 BOOL WINAPI GetDiskFreeSpaceW( LPCWSTR root, LPDWORD cluster_sectors,
1717 LPDWORD sector_bytes, LPDWORD free_clusters,
1718 LPDWORD total_clusters )
1719 {
1720 FILE_FS_SIZE_INFORMATION info;
1721 IO_STATUS_BLOCK io;
1722 NTSTATUS status;
1723 HANDLE handle;
1724 UINT units;
1725
1726 TRACE( "%s,%p,%p,%p,%p\n", debugstr_w(root),
1727 cluster_sectors, sector_bytes, free_clusters, total_clusters );
1728
1729 if (!open_device_root( root, &handle )) return FALSE;
1730
1731 status = NtQueryVolumeInformationFile( handle, &io, &info, sizeof(info), FileFsSizeInformation );
1732 NtClose( handle );
1733 if (status != STATUS_SUCCESS)
1734 {
1735 SetLastError( RtlNtStatusToDosError(status) );
1736 return FALSE;
1737 }
1738
1739 units = info.SectorsPerAllocationUnit * info.BytesPerSector;
1740
1741 if( GetVersion() & 0x80000000) { /* win3.x, 9x, ME */
1742 /* cap the size and available at 2GB as per specs */
1743 if (info.TotalAllocationUnits.QuadPart * units > 0x7fffffff) {
1744 info.TotalAllocationUnits.QuadPart = 0x7fffffff / units;
1745 if (info.AvailableAllocationUnits.QuadPart * units > 0x7fffffff)
1746 info.AvailableAllocationUnits.QuadPart = 0x7fffffff / units;
1747 }
1748 /* nr. of clusters is always <= 65335 */
1749 while( info.TotalAllocationUnits.QuadPart > 65535 ) {
1750 info.TotalAllocationUnits.QuadPart /= 2;
1751 info.AvailableAllocationUnits.QuadPart /= 2;
1752 info.SectorsPerAllocationUnit *= 2;
1753 }
1754 }
1755
1756 if (cluster_sectors) *cluster_sectors = info.SectorsPerAllocationUnit;
1757 if (sector_bytes) *sector_bytes = info.BytesPerSector;
1758 if (free_clusters) *free_clusters = info.AvailableAllocationUnits.u.LowPart;
1759 if (total_clusters) *total_clusters = info.TotalAllocationUnits.u.LowPart;
1760 TRACE("%#08x, %#08x, %#08x, %#08x\n", info.SectorsPerAllocationUnit, info.BytesPerSector,
1761 info.AvailableAllocationUnits.u.LowPart, info.TotalAllocationUnits.u.LowPart);
1762 return TRUE;
1763 }
1764
1765
1766 /***********************************************************************
1767 * GetDiskFreeSpaceA (KERNEL32.@)
1768 */
1769 BOOL WINAPI GetDiskFreeSpaceA( LPCSTR root, LPDWORD cluster_sectors,
1770 LPDWORD sector_bytes, LPDWORD free_clusters,
1771 LPDWORD total_clusters )
1772 {
1773 WCHAR *rootW = NULL;
1774
1775 if (root && !(rootW = FILE_name_AtoW( root, FALSE ))) return FALSE;
1776 return GetDiskFreeSpaceW( rootW, cluster_sectors, sector_bytes, free_clusters, total_clusters );
1777 }
1778
1779 /***********************************************************************
1780 * GetVolumePathNameA (KERNEL32.@)
1781 */
1782 BOOL WINAPI GetVolumePathNameA(LPCSTR filename, LPSTR volumepathname, DWORD buflen)
1783 {
1784 BOOL ret;
1785 WCHAR *filenameW = NULL, *volumeW;
1786
1787 FIXME("(%s, %p, %d), stub!\n", debugstr_a(filename), volumepathname, buflen);
1788
1789 if (filename && !(filenameW = FILE_name_AtoW( filename, FALSE ))) return FALSE;
1790 if (!(volumeW = HeapAlloc( GetProcessHeap(), 0, buflen * sizeof(WCHAR) ))) return FALSE;
1791
1792 if ((ret = GetVolumePathNameW( filenameW, volumeW, buflen )))
1793 FILE_name_WtoA( volumeW, -1, volumepathname, buflen );
1794
1795 HeapFree( GetProcessHeap(), 0, volumeW );
1796 return ret;
1797 }
1798
1799 /***********************************************************************
1800 * GetVolumePathNameW (KERNEL32.@)
1801 */
1802 BOOL WINAPI GetVolumePathNameW(LPCWSTR filename, LPWSTR volumepathname, DWORD buflen)
1803 {
1804 const WCHAR *p = filename;
1805
1806 FIXME("(%s, %p, %d), stub!\n", debugstr_w(filename), volumepathname, buflen);
1807
1808 if (p && tolowerW(p[0]) >= 'a' && tolowerW(p[0]) <= 'z' && p[1] ==':' && p[2] == '\\' && buflen >= 4)
1809 {
1810 volumepathname[0] = p[0];
1811 volumepathname[1] = ':';
1812 volumepathname[2] = '\\';
1813 volumepathname[3] = 0;
1814 return TRUE;
1815 }
1816 return FALSE;
1817 }
1818
1819 /***********************************************************************
1820 * GetVolumePathNamesForVolumeNameA (KERNEL32.@)
1821 */
1822 BOOL WINAPI GetVolumePathNamesForVolumeNameA(LPCSTR volumename, LPSTR volumepathname, DWORD buflen, PDWORD returnlen)
1823 {
1824 BOOL ret;
1825 WCHAR *volumenameW = NULL, *volumepathnameW;
1826
1827 if (volumename && !(volumenameW = FILE_name_AtoW( volumename, TRUE ))) return FALSE;
1828 if (!(volumepathnameW = HeapAlloc( GetProcessHeap(), 0, buflen * sizeof(WCHAR) )))
1829 {
1830 HeapFree( GetProcessHeap(), 0, volumenameW );
1831 return FALSE;
1832 }
1833 if ((ret = GetVolumePathNamesForVolumeNameW( volumenameW, volumepathnameW, buflen, returnlen )))
1834 {
1835 char *path = volumepathname;
1836 const WCHAR *pathW = volumepathnameW;
1837
1838 while (*pathW)
1839 {
1840 int len = strlenW( pathW ) + 1;
1841 FILE_name_WtoA( pathW, len, path, buflen );
1842 buflen -= len;
1843 pathW += len;
1844 path += len;
1845 }
1846 path[0] = 0;
1847 }
1848 HeapFree( GetProcessHeap(), 0, volumenameW );
1849 HeapFree( GetProcessHeap(), 0, volumepathnameW );
1850 return ret;
1851 }
1852
1853 static MOUNTMGR_MOUNT_POINTS *query_mount_points( HANDLE mgr, MOUNTMGR_MOUNT_POINT *input, DWORD insize )
1854 {
1855 MOUNTMGR_MOUNT_POINTS *output;
1856 DWORD outsize = 1024;
1857
1858 for (;;)
1859 {
1860 if (!(output = HeapAlloc( GetProcessHeap(), 0, outsize )))
1861 {
1862 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
1863 return NULL;
1864 }
1865 if (DeviceIoControl( mgr, IOCTL_MOUNTMGR_QUERY_POINTS, input, insize, output, outsize, NULL, NULL )) break;
1866 outsize = output->Size;
1867 HeapFree( GetProcessHeap(), 0, output );
1868 if (GetLastError() != ERROR_MORE_DATA) return NULL;
1869 }
1870 return output;
1871 }
1872 /***********************************************************************
1873 * GetVolumePathNamesForVolumeNameW (KERNEL32.@)
1874 */
1875 BOOL WINAPI GetVolumePathNamesForVolumeNameW(LPCWSTR volumename, LPWSTR volumepathname, DWORD buflen, PDWORD returnlen)
1876 {
1877 static const WCHAR dosdevicesW[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\'};
1878 HANDLE mgr;
1879 DWORD len, size;
1880 MOUNTMGR_MOUNT_POINT *spec;
1881 MOUNTMGR_MOUNT_POINTS *link, *target = NULL;
1882 WCHAR *name, *path;
1883 BOOL ret = FALSE;
1884 UINT i, j;
1885
1886 TRACE("%s, %p, %u, %p\n", debugstr_w(volumename), volumepathname, buflen, returnlen);
1887
1888 if (!volumename || (len = strlenW( volumename )) != 49)
1889 {
1890 SetLastError( ERROR_INVALID_NAME );
1891 return FALSE;
1892 }
1893 mgr = CreateFileW( MOUNTMGR_DOS_DEVICE_NAME, 0, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, 0 );
1894 if (mgr == INVALID_HANDLE_VALUE) return FALSE;
1895
1896 size = sizeof(*spec) + sizeof(WCHAR) * (len - 1); /* remove trailing backslash */
1897 if (!(spec = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, size ))) goto done;
1898 spec->SymbolicLinkNameOffset = sizeof(*spec);
1899 spec->SymbolicLinkNameLength = size - sizeof(*spec);
1900 name = (WCHAR *)((char *)spec + spec->SymbolicLinkNameOffset);
1901 memcpy( name, volumename, size - sizeof(*spec) );
1902 name[1] = '?'; /* map \\?\ to \??\ */
1903
1904 target = query_mount_points( mgr, spec, size );
1905 HeapFree( GetProcessHeap(), 0, spec );
1906 if (!target)
1907 {
1908 goto done;
1909 }
1910 if (!target->NumberOfMountPoints)
1911 {
1912 SetLastError( ERROR_FILE_NOT_FOUND );
1913 goto done;
1914 }
1915 len = 0;
1916 path = volumepathname;
1917 for (i = 0; i < target->NumberOfMountPoints; i++)
1918 {
1919 link = NULL;
1920 if (target->MountPoints[i].DeviceNameOffset)
1921 {
1922 const WCHAR *device = (const WCHAR *)((const char *)target + target->MountPoints[i].DeviceNameOffset);
1923 USHORT device_len = target->MountPoints[i].DeviceNameLength;
1924
1925 size = sizeof(*spec) + device_len;
1926 if (!(spec = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, size ))) goto done;
1927 spec->DeviceNameOffset = sizeof(*spec);
1928 spec->DeviceNameLength = device_len;
1929 memcpy( (char *)spec + spec->DeviceNameOffset, device, device_len );
1930
1931 link = query_mount_points( mgr, spec, size );
1932 HeapFree( GetProcessHeap(), 0, spec );
1933 }
1934 else if (target->MountPoints[i].UniqueIdOffset)
1935 {
1936 const WCHAR *id = (const WCHAR *)((const char *)target + target->MountPoints[i].UniqueIdOffset);
1937 USHORT id_len = target->MountPoints[i].UniqueIdLength;
1938
1939 size = sizeof(*spec) + id_len;
1940 if (!(spec = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, size ))) goto done;
1941 spec->UniqueIdOffset = sizeof(*spec);
1942 spec->UniqueIdLength = id_len;
1943 memcpy( (char *)spec + spec->UniqueIdOffset, id, id_len );
1944
1945 link = query_mount_points( mgr, spec, size );
1946 HeapFree( GetProcessHeap(), 0, spec );
1947 }
1948 if (!link) continue;
1949 for (j = 0; j < link->NumberOfMountPoints; j++)
1950 {
1951 const WCHAR *linkname;
1952
1953 if (!link->MountPoints[j].SymbolicLinkNameOffset) continue;
1954 linkname = (const WCHAR *)((const char *)link + link->MountPoints[j].SymbolicLinkNameOffset);
1955
1956 if (link->MountPoints[j].SymbolicLinkNameLength == sizeof(dosdevicesW) + 2 * sizeof(WCHAR) &&
1957 !memicmpW( linkname, dosdevicesW, sizeof(dosdevicesW) / sizeof(WCHAR) ))
1958 {
1959 len += 4;
1960 if (volumepathname && len < buflen)
1961 {
1962 path[0] = linkname[sizeof(dosdevicesW) / sizeof(WCHAR)];
1963 path[1] = ':';
1964 path[2] = '\\';
1965 path[3] = 0;
1966 path += 4;
1967 }
1968 }
1969 }
1970 HeapFree( GetProcessHeap(), 0, link );
1971 }
1972 if (buflen <= len) SetLastError( ERROR_MORE_DATA );
1973 else if (volumepathname)
1974 {
1975 volumepathname[len] = 0;
1976 ret = TRUE;
1977 }
1978 if (returnlen) *returnlen = len + 1;
1979
1980 done:
1981 HeapFree( GetProcessHeap(), 0, target );
1982 CloseHandle( mgr );
1983 return ret;
1984 }
1985
1986 /***********************************************************************
1987 * FindFirstVolumeA (KERNEL32.@)
1988 */
1989 HANDLE WINAPI FindFirstVolumeA(LPSTR volume, DWORD len)
1990 {
1991 WCHAR *buffer = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) );
1992 HANDLE handle = FindFirstVolumeW( buffer, len );
1993
1994 if (handle != INVALID_HANDLE_VALUE)
1995 {
1996 if (!WideCharToMultiByte( CP_ACP, 0, buffer, -1, volume, len, NULL, NULL ))
1997 {
1998 FindVolumeClose( handle );
1999 handle = INVALID_HANDLE_VALUE;
2000 }
2001 }
2002 HeapFree( GetProcessHeap(), 0, buffer );
2003 return handle;
2004 }
2005
2006 /***********************************************************************
2007 * FindFirstVolumeW (KERNEL32.@)
2008 */
2009 HANDLE WINAPI FindFirstVolumeW( LPWSTR volume, DWORD len )
2010 {
2011 DWORD size = 1024;
2012 HANDLE mgr = CreateFileW( MOUNTMGR_DOS_DEVICE_NAME, 0, FILE_SHARE_READ|FILE_SHARE_WRITE,
2013 NULL, OPEN_EXISTING, 0, 0 );
2014 if (mgr == INVALID_HANDLE_VALUE) return INVALID_HANDLE_VALUE;
2015
2016 for (;;)
2017 {
2018 MOUNTMGR_MOUNT_POINT input;
2019 MOUNTMGR_MOUNT_POINTS *output;
2020
2021 if (!(output = HeapAlloc( GetProcessHeap(), 0, size )))
2022 {
2023 SetLastError( ERROR_NOT_ENOUGH_MEMORY );
2024 break;
2025 }
2026 memset( &input, 0, sizeof(input) );
2027
2028 if (!DeviceIoControl( mgr, IOCTL_MOUNTMGR_QUERY_POINTS, &input, sizeof(input),
2029 output, size, NULL, NULL ))
2030 {
2031 if (GetLastError() != ERROR_MORE_DATA) break;
2032 size = output->Size;
2033 HeapFree( GetProcessHeap(), 0, output );
2034 continue;
2035 }
2036 CloseHandle( mgr );
2037 /* abuse the Size field to store the current index */
2038 output->Size = 0;
2039 if (!FindNextVolumeW( output, volume, len ))
2040 {
2041 HeapFree( GetProcessHeap(), 0, output );
2042 return INVALID_HANDLE_VALUE;
2043 }
2044 return output;
2045 }
2046 CloseHandle( mgr );
2047 return INVALID_HANDLE_VALUE;
2048 }
2049
2050 /***********************************************************************
2051 * FindNextVolumeA (KERNEL32.@)
2052 */
2053 BOOL WINAPI FindNextVolumeA( HANDLE handle, LPSTR volume, DWORD len )
2054 {
2055 WCHAR *buffer = HeapAlloc( GetProcessHeap(), 0, len * sizeof(WCHAR) );
2056 BOOL ret;
2057
2058 if ((ret = FindNextVolumeW( handle, buffer, len )))
2059 {
2060 if (!WideCharToMultiByte( CP_ACP, 0, buffer, -1, volume, len, NULL, NULL )) ret = FALSE;
2061 }
2062 HeapFree( GetProcessHeap(), 0, buffer );
2063 return ret;
2064 }
2065
2066 /***********************************************************************
2067 * FindNextVolumeW (KERNEL32.@)
2068 */
2069 BOOL WINAPI FindNextVolumeW( HANDLE handle, LPWSTR volume, DWORD len )
2070 {
2071 MOUNTMGR_MOUNT_POINTS *data = handle;
2072
2073 while (data->Size < data->NumberOfMountPoints)
2074 {
2075 static const WCHAR volumeW[] = {'\\','?','?','\\','V','o','l','u','m','e','{',};
2076 WCHAR *link = (WCHAR *)((char *)data + data->MountPoints[data->Size].SymbolicLinkNameOffset);
2077 DWORD size = data->MountPoints[data->Size].SymbolicLinkNameLength;
2078 data->Size++;
2079 /* skip non-volumes */
2080 if (size < sizeof(volumeW) || memcmp( link, volumeW, sizeof(volumeW) )) continue;
2081 if (size + sizeof(WCHAR) >= len * sizeof(WCHAR))
2082 {
2083 SetLastError( ERROR_FILENAME_EXCED_RANGE );
2084 return FALSE;
2085 }
2086 memcpy( volume, link, size );
2087 volume[1] = '\\'; /* map \??\ to \\?\ */
2088 volume[size / sizeof(WCHAR)] = '\\'; /* Windows appends a backslash */
2089 volume[size / sizeof(WCHAR) + 1] = 0;
2090 TRACE( "returning entry %u %s\n", data->Size - 1, debugstr_w(volume) );
2091 return TRUE;
2092 }
2093 SetLastError( ERROR_NO_MORE_FILES );
2094 return FALSE;
2095 }
2096
2097 /***********************************************************************
2098 * FindVolumeClose (KERNEL32.@)
2099 */
2100 BOOL WINAPI FindVolumeClose(HANDLE handle)
2101 {
2102 return HeapFree( GetProcessHeap(), 0, handle );
2103 }
2104
2105 /***********************************************************************
2106 * FindFirstVolumeMountPointA (KERNEL32.@)
2107 */
2108 HANDLE WINAPI FindFirstVolumeMountPointA(LPCSTR root, LPSTR mount_point, DWORD len)
2109 {
2110 FIXME("(%s, %p, %d), stub!\n", debugstr_a(root), mount_point, len);
2111 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
2112 return INVALID_HANDLE_VALUE;
2113 }
2114
2115 /***********************************************************************
2116 * FindFirstVolumeMountPointW (KERNEL32.@)
2117 */
2118 HANDLE WINAPI FindFirstVolumeMountPointW(LPCWSTR root, LPWSTR mount_point, DWORD len)
2119 {
2120 FIXME("(%s, %p, %d), stub!\n", debugstr_w(root), mount_point, len);
2121 SetLastError(ERROR_CALL_NOT_IMPLEMENTED);
2122 return INVALID_HANDLE_VALUE;
2123 }
2124
2125 /***********************************************************************
2126 * FindVolumeMountPointClose (KERNEL32.@)
2127 */
2128 BOOL WINAPI FindVolumeMountPointClose(HANDLE h)
2129 {
2130 FIXME("(%p), stub!\n", h);
2131 return FALSE;
2132 }
2133
2134 /***********************************************************************
2135 * DeleteVolumeMountPointA (KERNEL32.@)
2136 */
2137 BOOL WINAPI DeleteVolumeMountPointA(LPCSTR mountpoint)
2138 {
2139 FIXME("(%s), stub!\n", debugstr_a(mountpoint));
2140 return FALSE;
2141 }
2142
2143 /***********************************************************************
2144 * DeleteVolumeMountPointW (KERNEL32.@)
2145 */
2146 BOOL WINAPI DeleteVolumeMountPointW(LPCWSTR mountpoint)
2147 {
2148 FIXME("(%s), stub!\n", debugstr_w(mountpoint));
2149 return FALSE;
2150 }
Windows API implementation