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