| /* |
| * Volume management functions |
| * |
| * Copyright 1993 Erik Bos |
| * Copyright 1996, 2004 Alexandre Julliard |
| * Copyright 1999 Petr Tomasek |
| * Copyright 2000 Andreas Mohr |
| * Copyright 2003 Eric Pouech |
| * |
| * This library is free software; you can redistribute it and/or |
| * modify it under the terms of the GNU Lesser General Public |
| * License as published by the Free Software Foundation; either |
| * version 2.1 of the License, or (at your option) any later version. |
| * |
| * This library is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| * Lesser General Public License for more details. |
| * |
| * You should have received a copy of the GNU Lesser General Public |
| * License along with this library; if not, write to the Free Software |
| * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA |
| */ |
| |
| #include "config.h" |
| #include "wine/port.h" |
| |
| #include <stdarg.h> |
| #include <stdlib.h> |
| #include <stdio.h> |
| |
| #include "windef.h" |
| #include "winbase.h" |
| #include "winreg.h" |
| #include "winnls.h" |
| #include "winternl.h" |
| #include "ntstatus.h" |
| #include "winioctl.h" |
| #include "ntddstor.h" |
| #include "ntddcdrm.h" |
| #include "kernel_private.h" |
| #include "wine/library.h" |
| #include "wine/unicode.h" |
| #include "wine/debug.h" |
| |
| WINE_DEFAULT_DEBUG_CHANNEL(volume); |
| |
| #define SUPERBLOCK_SIZE 2048 |
| |
| #define CDFRAMES_PERSEC 75 |
| #define CDFRAMES_PERMIN (CDFRAMES_PERSEC * 60) |
| #define FRAME_OF_ADDR(a) ((a)[1] * CDFRAMES_PERMIN + (a)[2] * CDFRAMES_PERSEC + (a)[3]) |
| #define FRAME_OF_TOC(toc, idx) FRAME_OF_ADDR((toc)->TrackData[(idx) - (toc)->FirstTrack].Address) |
| |
| #define GETWORD(buf,off) MAKEWORD(buf[(off)],buf[(off+1)]) |
| #define GETLONG(buf,off) MAKELONG(GETWORD(buf,off),GETWORD(buf,off+2)) |
| |
| enum fs_type |
| { |
| FS_ERROR, /* error accessing the device */ |
| FS_UNKNOWN, /* unknown file system */ |
| FS_FAT1216, |
| FS_FAT32, |
| FS_ISO9660 |
| }; |
| |
| static const WCHAR drive_types[][8] = |
| { |
| { 0 }, /* DRIVE_UNKNOWN */ |
| { 0 }, /* DRIVE_NO_ROOT_DIR */ |
| {'f','l','o','p','p','y',0}, /* DRIVE_REMOVABLE */ |
| {'h','d',0}, /* DRIVE_FIXED */ |
| {'n','e','t','w','o','r','k',0}, /* DRIVE_REMOTE */ |
| {'c','d','r','o','m',0}, /* DRIVE_CDROM */ |
| {'r','a','m','d','i','s','k',0} /* DRIVE_RAMDISK */ |
| }; |
| |
| /* read a Unix symlink; returned buffer must be freed by caller */ |
| static char *read_symlink( const char *path ) |
| { |
| char *buffer; |
| int ret, size = 128; |
| |
| for (;;) |
| { |
| if (!(buffer = HeapAlloc( GetProcessHeap(), 0, size ))) |
| { |
| SetLastError( ERROR_NOT_ENOUGH_MEMORY ); |
| return 0; |
| } |
| ret = readlink( path, buffer, size ); |
| if (ret == -1) |
| { |
| FILE_SetDosError(); |
| HeapFree( GetProcessHeap(), 0, buffer ); |
| return 0; |
| } |
| if (ret != size) |
| { |
| buffer[ret] = 0; |
| return buffer; |
| } |
| HeapFree( GetProcessHeap(), 0, buffer ); |
| size *= 2; |
| } |
| } |
| |
| /* get the path of a dos device symlink in the $WINEPREFIX/dosdevices directory */ |
| static char *get_dos_device_path( LPCWSTR name ) |
| { |
| const char *config_dir = wine_get_config_dir(); |
| char *buffer, *dev; |
| int i; |
| |
| if (!(buffer = HeapAlloc( GetProcessHeap(), 0, |
| strlen(config_dir) + sizeof("/dosdevices/") + 5 ))) |
| { |
| SetLastError( ERROR_NOT_ENOUGH_MEMORY ); |
| return NULL; |
| } |
| strcpy( buffer, config_dir ); |
| strcat( buffer, "/dosdevices/" ); |
| dev = buffer + strlen(buffer); |
| /* no codepage conversion, DOS device names are ASCII anyway */ |
| for (i = 0; i < 5; i++) |
| if (!(dev[i] = (char)tolowerW(name[i]))) break; |
| dev[5] = 0; |
| return buffer; |
| } |
| |
| |
| /* open a handle to a device root */ |
| static BOOL open_device_root( LPCWSTR root, HANDLE *handle ) |
| { |
| static const WCHAR default_rootW[] = {'\\',0}; |
| UNICODE_STRING nt_name; |
| OBJECT_ATTRIBUTES attr; |
| IO_STATUS_BLOCK io; |
| NTSTATUS status; |
| |
| if (!root) root = default_rootW; |
| if (!RtlDosPathNameToNtPathName_U( root, &nt_name, NULL, NULL )) |
| { |
| SetLastError( ERROR_PATH_NOT_FOUND ); |
| return FALSE; |
| } |
| attr.Length = sizeof(attr); |
| attr.RootDirectory = 0; |
| attr.Attributes = OBJ_CASE_INSENSITIVE; |
| attr.ObjectName = &nt_name; |
| attr.SecurityDescriptor = NULL; |
| attr.SecurityQualityOfService = NULL; |
| |
| status = NtOpenFile( handle, 0, &attr, &io, 0, |
| FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT ); |
| RtlFreeUnicodeString( &nt_name ); |
| if (status != STATUS_SUCCESS) |
| { |
| SetLastError( RtlNtStatusToDosError(status) ); |
| return FALSE; |
| } |
| return TRUE; |
| } |
| |
| |
| /* fetch the type of a drive from the registry */ |
| static UINT get_registry_drive_type( const WCHAR *root ) |
| { |
| static const WCHAR drive_types_keyW[] = {'M','a','c','h','i','n','e','\\', |
| 'S','o','f','t','w','a','r','e','\\', |
| 'W','i','n','e','\\', |
| 'D','r','i','v','e','s',0 }; |
| OBJECT_ATTRIBUTES attr; |
| UNICODE_STRING nameW; |
| HKEY hkey; |
| DWORD dummy; |
| UINT ret = DRIVE_UNKNOWN; |
| char tmp[32 + sizeof(KEY_VALUE_PARTIAL_INFORMATION)]; |
| WCHAR driveW[] = {'A',':',0}; |
| |
| attr.Length = sizeof(attr); |
| attr.RootDirectory = 0; |
| attr.ObjectName = &nameW; |
| attr.Attributes = 0; |
| attr.SecurityDescriptor = NULL; |
| attr.SecurityQualityOfService = NULL; |
| RtlInitUnicodeString( &nameW, drive_types_keyW ); |
| if (NtOpenKey( &hkey, KEY_ALL_ACCESS, &attr ) != STATUS_SUCCESS) return DRIVE_UNKNOWN; |
| |
| if (root) driveW[0] = root[0]; |
| else |
| { |
| WCHAR path[MAX_PATH]; |
| GetCurrentDirectoryW( MAX_PATH, path ); |
| driveW[0] = path[0]; |
| } |
| |
| RtlInitUnicodeString( &nameW, driveW ); |
| if (!NtQueryValueKey( hkey, &nameW, KeyValuePartialInformation, tmp, sizeof(tmp), &dummy )) |
| { |
| int i; |
| WCHAR *data = (WCHAR *)((KEY_VALUE_PARTIAL_INFORMATION *)tmp)->Data; |
| |
| for (i = 0; i < sizeof(drive_types)/sizeof(drive_types[0]); i++) |
| { |
| if (!strcmpiW( data, drive_types[i] )) |
| { |
| ret = i; |
| break; |
| } |
| } |
| } |
| NtClose( hkey ); |
| return ret; |
| } |
| |
| |
| /* create symlinks for the DOS drives; helper for VOLUME_CreateDevices */ |
| static int create_drives( int devices_only ) |
| { |
| static const WCHAR PathW[] = {'P','a','t','h',0}; |
| static const WCHAR DeviceW[] = {'D','e','v','i','c','e',0}; |
| WCHAR driveW[] = {'M','a','c','h','i','n','e','\\','S','o','f','t','w','a','r','e','\\', |
| 'W','i','n','e','\\','W','i','n','e','\\', |
| 'C','o','n','f','i','g','\\','D','r','i','v','e',' ','A',0}; |
| OBJECT_ATTRIBUTES attr; |
| UNICODE_STRING nameW; |
| char tmp[1024*sizeof(WCHAR) + sizeof(KEY_VALUE_PARTIAL_INFORMATION)]; |
| char dest[1024]; |
| WCHAR *p, name[3]; |
| HKEY hkey; |
| DWORD dummy; |
| int i, count = 0; |
| |
| attr.Length = sizeof(attr); |
| attr.RootDirectory = 0; |
| attr.ObjectName = &nameW; |
| attr.Attributes = 0; |
| attr.SecurityDescriptor = NULL; |
| attr.SecurityQualityOfService = NULL; |
| |
| /* create symlinks for the drive roots */ |
| |
| if (!devices_only) for (i = 0; i < 26; i++) |
| { |
| RtlInitUnicodeString( &nameW, driveW ); |
| nameW.Buffer[(nameW.Length / sizeof(WCHAR)) - 1] = 'A' + i; |
| if (NtOpenKey( &hkey, KEY_ALL_ACCESS, &attr ) != STATUS_SUCCESS) continue; |
| |
| RtlInitUnicodeString( &nameW, PathW ); |
| if (!NtQueryValueKey( hkey, &nameW, KeyValuePartialInformation, tmp, sizeof(tmp), &dummy )) |
| { |
| WCHAR path[1024]; |
| WCHAR *data = (WCHAR *)((KEY_VALUE_PARTIAL_INFORMATION *)tmp)->Data; |
| ExpandEnvironmentStringsW( data, path, sizeof(path)/sizeof(WCHAR) ); |
| |
| p = path + strlenW(path) - 1; |
| while ((p > path) && (*p == '/')) *p-- = '\0'; |
| |
| name[0] = 'a' + i; |
| name[1] = ':'; |
| name[2] = 0; |
| |
| if (path[0] != '/') |
| { |
| /* relative paths are relative to config dir */ |
| memmove( path + 3, path, (strlenW(path) + 1) * sizeof(WCHAR) ); |
| path[0] = '.'; |
| path[1] = '.'; |
| path[2] = '/'; |
| } |
| if (DefineDosDeviceW( DDD_RAW_TARGET_PATH, name, path )) |
| { |
| WideCharToMultiByte(CP_UNIXCP, 0, path, -1, dest, sizeof(dest), NULL, NULL); |
| MESSAGE( "Created symlink %s/dosdevices/%c: -> %s\n", |
| wine_get_config_dir(), 'a' + i, dest ); |
| count++; |
| } |
| } |
| NtClose( hkey ); |
| } |
| |
| /* create symlinks for the drive devices */ |
| |
| for (i = 0; i < 26; i++) |
| { |
| RtlInitUnicodeString( &nameW, driveW ); |
| nameW.Buffer[(nameW.Length / sizeof(WCHAR)) - 1] = 'A' + i; |
| if (NtOpenKey( &hkey, KEY_ALL_ACCESS, &attr ) != STATUS_SUCCESS) continue; |
| |
| RtlInitUnicodeString( &nameW, DeviceW ); |
| if (!NtQueryValueKey( hkey, &nameW, KeyValuePartialInformation, tmp, sizeof(tmp), &dummy )) |
| { |
| char *path, *p; |
| WCHAR devname[] = {'A',':',':',0 }; |
| WCHAR *data = (WCHAR *)((KEY_VALUE_PARTIAL_INFORMATION *)tmp)->Data; |
| WideCharToMultiByte(CP_UNIXCP, 0, data, -1, dest, sizeof(dest), NULL, NULL); |
| path = get_dos_device_path( devname ); |
| p = path + strlen(path); |
| p[-3] = 'a' + i; |
| if (!symlink( dest, path )) |
| { |
| MESSAGE( "Created symlink %s/dosdevices/%c:: -> %s\n", |
| wine_get_config_dir(), 'a' + i, dest ); |
| count++; |
| } |
| HeapFree( GetProcessHeap(), 0, path ); |
| } |
| NtClose( hkey ); |
| } |
| |
| return count; |
| } |
| |
| |
| /*********************************************************************** |
| * VOLUME_CreateDevices |
| * |
| * Create the device files for the new device naming scheme. |
| * Should go away after a transition period. |
| */ |
| void VOLUME_CreateDevices(void) |
| { |
| const char *config_dir = wine_get_config_dir(); |
| char *buffer; |
| int i, count = 0; |
| |
| if (!(buffer = HeapAlloc( GetProcessHeap(), 0, |
| strlen(config_dir) + sizeof("/dosdevices/a::") ))) |
| return; |
| |
| strcpy( buffer, config_dir ); |
| strcat( buffer, "/dosdevices" ); |
| |
| if (!mkdir( buffer, 0777 )) /* we created it, so now create the devices */ |
| { |
| HKEY hkey; |
| DWORD dummy; |
| OBJECT_ATTRIBUTES attr; |
| UNICODE_STRING nameW; |
| WCHAR *p, *devnameW; |
| char tmp[128]; |
| WCHAR com[5] = {'C','O','M','1',0}; |
| WCHAR lpt[5] = {'L','P','T','1',0}; |
| |
| static const WCHAR serialportsW[] = {'M','a','c','h','i','n','e','\\', |
| 'S','o','f','t','w','a','r','e','\\', |
| 'W','i','n','e','\\','W','i','n','e','\\', |
| 'C','o','n','f','i','g','\\', |
| 'S','e','r','i','a','l','P','o','r','t','s',0}; |
| static const WCHAR parallelportsW[] = {'M','a','c','h','i','n','e','\\', |
| 'S','o','f','t','w','a','r','e','\\', |
| 'W','i','n','e','\\','W','i','n','e','\\', |
| 'C','o','n','f','i','g','\\', |
| 'P','a','r','a','l','l','e','l','P','o','r','t','s',0}; |
| |
| attr.Length = sizeof(attr); |
| attr.RootDirectory = 0; |
| attr.ObjectName = &nameW; |
| attr.Attributes = 0; |
| attr.SecurityDescriptor = NULL; |
| attr.SecurityQualityOfService = NULL; |
| RtlInitUnicodeString( &nameW, serialportsW ); |
| |
| if (!NtOpenKey( &hkey, KEY_ALL_ACCESS, &attr )) |
| { |
| RtlInitUnicodeString( &nameW, com ); |
| for (i = 1; i <= 9; i++) |
| { |
| com[3] = '0' + i; |
| if (!NtQueryValueKey( hkey, &nameW, KeyValuePartialInformation, |
| tmp, sizeof(tmp), &dummy )) |
| { |
| devnameW = (WCHAR *)((KEY_VALUE_PARTIAL_INFORMATION *)tmp)->Data; |
| if ((p = strchrW( devnameW, ',' ))) *p = 0; |
| if (DefineDosDeviceW( DDD_RAW_TARGET_PATH, com, devnameW )) |
| { |
| char devname[32]; |
| WideCharToMultiByte(CP_UNIXCP, 0, devnameW, -1, |
| devname, sizeof(devname), NULL, NULL); |
| MESSAGE( "Created symlink %s/dosdevices/com%d -> %s\n", config_dir, i, devname ); |
| count++; |
| } |
| } |
| } |
| NtClose( hkey ); |
| } |
| |
| RtlInitUnicodeString( &nameW, parallelportsW ); |
| if (!NtOpenKey( &hkey, KEY_ALL_ACCESS, &attr )) |
| { |
| RtlInitUnicodeString( &nameW, lpt ); |
| for (i = 1; i <= 9; i++) |
| { |
| lpt[3] = '0' + i; |
| if (!NtQueryValueKey( hkey, &nameW, KeyValuePartialInformation, |
| tmp, sizeof(tmp), &dummy )) |
| { |
| devnameW = (WCHAR *)((KEY_VALUE_PARTIAL_INFORMATION *)tmp)->Data; |
| if ((p = strchrW( devnameW, ',' ))) *p = 0; |
| if (DefineDosDeviceW( DDD_RAW_TARGET_PATH, lpt, devnameW )) |
| { |
| char devname[32]; |
| WideCharToMultiByte(CP_UNIXCP, 0, devnameW, -1, |
| devname, sizeof(devname), NULL, NULL); |
| MESSAGE( "Created symlink %s/dosdevices/lpt%d -> %s\n", config_dir, i, devname ); |
| count++; |
| } |
| } |
| } |
| NtClose( hkey ); |
| } |
| count += create_drives( FALSE ); |
| } |
| else |
| { |
| struct stat st; |
| int i; |
| |
| /* it is possible that the serial/parallel devices have been created but */ |
| /* not the drives; check for at least one drive symlink to catch that case */ |
| strcat( buffer, "/a:" ); |
| for (i = 0; i < 26; i++) |
| { |
| buffer[strlen(buffer)-2] = 'a' + i; |
| if (!lstat( buffer, &st )) break; |
| } |
| if (i == 26) count += create_drives( FALSE ); |
| else |
| { |
| strcat( buffer, ":" ); |
| for (i = 0; i < 26; i++) |
| { |
| buffer[strlen(buffer)-3] = 'a' + i; |
| if (!lstat( buffer, &st )) break; |
| } |
| if (i == 26) count += create_drives( TRUE ); |
| } |
| } |
| |
| if (count) |
| MESSAGE( "\nYou can now remove the [SerialPorts], [ParallelPorts], and [Drive] sections\n" |
| "in your configuration file, they are replaced by the above symlinks.\n\n" ); |
| |
| HeapFree( GetProcessHeap(), 0, buffer ); |
| } |
| |
| |
| /****************************************************************** |
| * VOLUME_FindCdRomDataBestVoldesc |
| */ |
| static DWORD VOLUME_FindCdRomDataBestVoldesc( HANDLE handle ) |
| { |
| BYTE cur_vd_type, max_vd_type = 0; |
| BYTE buffer[16]; |
| DWORD size, offs, best_offs = 0, extra_offs = 0; |
| |
| for (offs = 0x8000; offs <= 0x9800; offs += 0x800) |
| { |
| /* if 'CDROM' occurs at position 8, this is a pre-iso9660 cd, and |
| * the volume label is displaced forward by 8 |
| */ |
| if (SetFilePointer( handle, offs, NULL, FILE_BEGIN ) != offs) break; |
| if (!ReadFile( handle, buffer, sizeof(buffer), &size, NULL )) break; |
| if (size != sizeof(buffer)) break; |
| /* check for non-ISO9660 signature */ |
| if (!memcmp( buffer + 11, "ROM", 3 )) extra_offs = 8; |
| cur_vd_type = buffer[extra_offs]; |
| if (cur_vd_type == 0xff) /* voldesc set terminator */ |
| break; |
| if (cur_vd_type > max_vd_type) |
| { |
| max_vd_type = cur_vd_type; |
| best_offs = offs + extra_offs; |
| } |
| } |
| return best_offs; |
| } |
| |
| |
| /*********************************************************************** |
| * VOLUME_ReadFATSuperblock |
| */ |
| static enum fs_type VOLUME_ReadFATSuperblock( HANDLE handle, BYTE *buff ) |
| { |
| DWORD size; |
| |
| /* try a fixed disk, with a FAT partition */ |
| if (SetFilePointer( handle, 0, NULL, FILE_BEGIN ) != 0 || |
| !ReadFile( handle, buff, SUPERBLOCK_SIZE, &size, NULL ) || |
| size != SUPERBLOCK_SIZE) |
| return FS_ERROR; |
| |
| if (buff[0] == 0xE9 || (buff[0] == 0xEB && buff[2] == 0x90)) |
| { |
| /* guess which type of FAT we have */ |
| unsigned int sz, nsect, nclust; |
| sz = GETWORD(buff, 0x16); |
| if (!sz) sz = GETLONG(buff, 0x24); |
| nsect = GETWORD(buff, 0x13); |
| if (!nsect) nsect = GETLONG(buff, 0x20); |
| nsect -= GETWORD(buff, 0x0e) + buff[0x10] * sz + |
| (GETWORD(buff, 0x11) * 32 + (GETWORD(buff, 0x0b) - 1)) / GETWORD(buff, 0x0b); |
| nclust = nsect / buff[0x0d]; |
| |
| if (nclust < 65525) |
| { |
| if (buff[0x26] == 0x29 && !memcmp(buff+0x36, "FAT", 3)) |
| { |
| /* FIXME: do really all FAT have their name beginning with |
| * "FAT" ? (At least FAT12, FAT16 and FAT32 have :) |
| */ |
| return FS_FAT1216; |
| } |
| } |
| else if (!memcmp(buff+0x52, "FAT", 3)) return FS_FAT32; |
| } |
| return FS_UNKNOWN; |
| } |
| |
| |
| /*********************************************************************** |
| * VOLUME_ReadCDSuperblock |
| */ |
| static enum fs_type VOLUME_ReadCDSuperblock( HANDLE handle, BYTE *buff ) |
| { |
| DWORD size, offs = VOLUME_FindCdRomDataBestVoldesc( handle ); |
| |
| if (!offs) return FS_UNKNOWN; |
| |
| if (SetFilePointer( handle, offs, NULL, FILE_BEGIN ) != offs || |
| !ReadFile( handle, buff, SUPERBLOCK_SIZE, &size, NULL ) || |
| size != SUPERBLOCK_SIZE) |
| return FS_ERROR; |
| |
| /* check for iso9660 present */ |
| if (!memcmp(&buff[1], "CD001", 5)) return FS_ISO9660; |
| return FS_UNKNOWN; |
| } |
| |
| |
| /************************************************************************** |
| * VOLUME_GetSuperblockLabel |
| */ |
| static void VOLUME_GetSuperblockLabel( enum fs_type type, const BYTE *superblock, |
| WCHAR *label, DWORD len ) |
| { |
| const BYTE *label_ptr = NULL; |
| DWORD label_len; |
| |
| switch(type) |
| { |
| case FS_ERROR: |
| case FS_UNKNOWN: |
| label_len = 0; |
| break; |
| case FS_FAT1216: |
| label_ptr = superblock + 0x2b; |
| label_len = 11; |
| break; |
| case FS_FAT32: |
| label_ptr = superblock + 0x47; |
| label_len = 11; |
| break; |
| case FS_ISO9660: |
| { |
| BYTE ver = superblock[0x5a]; |
| |
| if (superblock[0x58] == 0x25 && superblock[0x59] == 0x2f && /* Unicode ID */ |
| ((ver == 0x40) || (ver == 0x43) || (ver == 0x45))) |
| { /* yippee, unicode */ |
| int i; |
| |
| if (len > 17) len = 17; |
| for (i = 0; i < len-1; i++) |
| label[i] = (superblock[40+2*i] << 8) | superblock[41+2*i]; |
| label[i] = 0; |
| while (i && label[i-1] == ' ') label[--i] = 0; |
| return; |
| } |
| label_ptr = superblock + 40; |
| label_len = 32; |
| break; |
| } |
| } |
| if (label_len) RtlMultiByteToUnicodeN( label, (len-1) * sizeof(WCHAR), |
| &label_len, label_ptr, label_len ); |
| label_len /= sizeof(WCHAR); |
| label[label_len] = 0; |
| while (label_len && label[label_len-1] == ' ') label[--label_len] = 0; |
| } |
| |
| |
| /************************************************************************** |
| * VOLUME_SetSuperblockLabel |
| */ |
| static BOOL VOLUME_SetSuperblockLabel( enum fs_type type, HANDLE handle, const WCHAR *label ) |
| { |
| BYTE label_data[11]; |
| DWORD offset, len; |
| |
| switch(type) |
| { |
| case FS_FAT1216: |
| offset = 0x2b; |
| break; |
| case FS_FAT32: |
| offset = 0x47; |
| break; |
| default: |
| SetLastError( ERROR_ACCESS_DENIED ); |
| return FALSE; |
| } |
| RtlUnicodeToMultiByteN( label_data, sizeof(label_data), &len, |
| label, strlenW(label) * sizeof(WCHAR) ); |
| if (len < sizeof(label_data)) |
| memset( label_data + len, ' ', sizeof(label_data) - len ); |
| |
| return (SetFilePointer( handle, offset, NULL, FILE_BEGIN ) == offset && |
| WriteFile( handle, label_data, sizeof(label_data), &len, NULL )); |
| } |
| |
| |
| /************************************************************************** |
| * VOLUME_GetSuperblockSerial |
| */ |
| static DWORD VOLUME_GetSuperblockSerial( enum fs_type type, const BYTE *superblock ) |
| { |
| switch(type) |
| { |
| case FS_ERROR: |
| case FS_UNKNOWN: |
| break; |
| case FS_FAT1216: |
| return GETLONG( superblock, 0x27 ); |
| case FS_FAT32: |
| return GETLONG( superblock, 0x33 ); |
| case FS_ISO9660: |
| { |
| BYTE sum[4]; |
| int i; |
| |
| sum[0] = sum[1] = sum[2] = sum[3] = 0; |
| for (i = 0; i < 2048; i += 4) |
| { |
| /* DON'T optimize this into DWORD !! (breaks overflow) */ |
| sum[0] += superblock[i+0]; |
| sum[1] += superblock[i+1]; |
| sum[2] += superblock[i+2]; |
| sum[3] += superblock[i+3]; |
| } |
| /* |
| * OK, another braindead one... argh. Just believe it. |
| * Me$$ysoft chose to reverse the serial number in NT4/W2K. |
| * It's true and nobody will ever be able to change it. |
| */ |
| if (GetVersion() & 0x80000000) |
| return (sum[3] << 24) | (sum[2] << 16) | (sum[1] << 8) | sum[0]; |
| else |
| return (sum[0] << 24) | (sum[1] << 16) | (sum[2] << 8) | sum[3]; |
| } |
| } |
| return 0; |
| } |
| |
| |
| /************************************************************************** |
| * VOLUME_GetAudioCDSerial |
| */ |
| static DWORD VOLUME_GetAudioCDSerial( const CDROM_TOC *toc ) |
| { |
| DWORD serial = 0; |
| int i; |
| |
| for (i = 0; i <= toc->LastTrack - toc->FirstTrack; i++) |
| serial += ((toc->TrackData[i].Address[1] << 16) | |
| (toc->TrackData[i].Address[2] << 8) | |
| toc->TrackData[i].Address[3]); |
| |
| /* |
| * dwStart, dwEnd collect the beginning and end of the disc respectively, in |
| * frames. |
| * There it is collected for correcting the serial when there are less than |
| * 3 tracks. |
| */ |
| if (toc->LastTrack - toc->FirstTrack + 1 < 3) |
| { |
| DWORD dwStart = FRAME_OF_TOC(toc, toc->FirstTrack); |
| DWORD dwEnd = FRAME_OF_TOC(toc, toc->LastTrack + 1); |
| serial += dwEnd - dwStart; |
| } |
| return serial; |
| } |
| |
| |
| /*********************************************************************** |
| * GetVolumeInformationW (KERNEL32.@) |
| */ |
| BOOL WINAPI GetVolumeInformationW( LPCWSTR root, LPWSTR label, DWORD label_len, |
| DWORD *serial, DWORD *filename_len, DWORD *flags, |
| LPWSTR fsname, DWORD fsname_len ) |
| { |
| static const WCHAR audiocdW[] = {'A','u','d','i','o',' ','C','D',0}; |
| static const WCHAR fatW[] = {'F','A','T',0}; |
| static const WCHAR ntfsW[] = {'N','T','F','S',0}; |
| static const WCHAR cdfsW[] = {'C','D','F','S',0}; |
| |
| WCHAR device[] = {'\\','\\','.','\\','A',':',0}; |
| HANDLE handle; |
| enum fs_type type = FS_UNKNOWN; |
| |
| if (!root) |
| { |
| WCHAR path[MAX_PATH]; |
| GetCurrentDirectoryW( MAX_PATH, path ); |
| device[4] = path[0]; |
| } |
| else |
| { |
| if (!root[0] || root[1] != ':') |
| { |
| SetLastError( ERROR_INVALID_NAME ); |
| return FALSE; |
| } |
| device[4] = root[0]; |
| } |
| |
| /* try to open the device */ |
| |
| handle = CreateFileW( device, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE, |
| NULL, OPEN_EXISTING, 0, 0 ); |
| if (handle != INVALID_HANDLE_VALUE) |
| { |
| BYTE superblock[SUPERBLOCK_SIZE]; |
| CDROM_TOC toc; |
| DWORD br; |
| |
| /* check for audio CD */ |
| /* FIXME: we only check the first track for now */ |
| if (DeviceIoControl( handle, IOCTL_CDROM_READ_TOC, NULL, 0, &toc, sizeof(toc), &br, 0 )) |
| { |
| if (!(toc.TrackData[0].Control & 0x04)) /* audio track */ |
| { |
| TRACE( "%s: found audio CD\n", debugstr_w(device) ); |
| if (label) lstrcpynW( label, audiocdW, label_len ); |
| if (serial) *serial = VOLUME_GetAudioCDSerial( &toc ); |
| CloseHandle( handle ); |
| type = FS_ISO9660; |
| goto fill_fs_info; |
| } |
| type = VOLUME_ReadCDSuperblock( handle, superblock ); |
| } |
| else |
| { |
| type = VOLUME_ReadFATSuperblock( handle, superblock ); |
| if (type == FS_UNKNOWN) type = VOLUME_ReadCDSuperblock( handle, superblock ); |
| } |
| CloseHandle( handle ); |
| TRACE( "%s: found fs type %d\n", debugstr_w(device), type ); |
| if (type == FS_ERROR) return FALSE; |
| |
| if (label && label_len) VOLUME_GetSuperblockLabel( type, superblock, label, label_len ); |
| if (serial) *serial = VOLUME_GetSuperblockSerial( type, superblock ); |
| goto fill_fs_info; |
| } |
| else |
| { |
| TRACE( "cannot open device %s: err %ld\n", debugstr_w(device), GetLastError() ); |
| if (GetLastError() != ERROR_FILE_NOT_FOUND) return FALSE; |
| } |
| |
| /* we couldn't open the device, fallback to default strategy */ |
| |
| switch(GetDriveTypeW( root )) |
| { |
| case DRIVE_UNKNOWN: |
| case DRIVE_NO_ROOT_DIR: |
| SetLastError( ERROR_NOT_READY ); |
| return FALSE; |
| case DRIVE_REMOVABLE: |
| case DRIVE_FIXED: |
| case DRIVE_REMOTE: |
| case DRIVE_RAMDISK: |
| type = FS_UNKNOWN; |
| break; |
| case DRIVE_CDROM: |
| type = FS_ISO9660; |
| break; |
| } |
| |
| if (label && label_len) |
| { |
| WCHAR labelW[] = {'A',':','\\','.','w','i','n','d','o','w','s','-','l','a','b','e','l',0}; |
| |
| labelW[0] = device[4]; |
| handle = CreateFileW( labelW, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, |
| OPEN_EXISTING, 0, 0 ); |
| if (handle != INVALID_HANDLE_VALUE) |
| { |
| char buffer[256], *p; |
| DWORD size; |
| |
| if (!ReadFile( handle, buffer, sizeof(buffer)-1, &size, NULL )) size = 0; |
| CloseHandle( handle ); |
| p = buffer + size; |
| while (p > buffer && (p[-1] == ' ' || p[-1] == '\r' || p[-1] == '\n')) p--; |
| *p = 0; |
| if (!MultiByteToWideChar( CP_UNIXCP, 0, buffer, -1, label, label_len )) |
| label[label_len-1] = 0; |
| } |
| else label[0] = 0; |
| } |
| if (serial) |
| { |
| WCHAR serialW[] = {'A',':','\\','.','w','i','n','d','o','w','s','-','s','e','r','i','a','l',0}; |
| |
| serialW[0] = device[4]; |
| handle = CreateFileW( serialW, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, |
| OPEN_EXISTING, 0, 0 ); |
| if (handle != INVALID_HANDLE_VALUE) |
| { |
| char buffer[32]; |
| DWORD size; |
| |
| if (!ReadFile( handle, buffer, sizeof(buffer)-1, &size, NULL )) size = 0; |
| CloseHandle( handle ); |
| buffer[size] = 0; |
| *serial = strtoul( buffer, NULL, 16 ); |
| } |
| else *serial = 0; |
| } |
| |
| fill_fs_info: /* now fill in the information that depends on the file system type */ |
| |
| switch(type) |
| { |
| case FS_ISO9660: |
| if (fsname) lstrcpynW( fsname, cdfsW, fsname_len ); |
| if (filename_len) *filename_len = 221; |
| if (flags) *flags = FILE_READ_ONLY_VOLUME; |
| break; |
| case FS_FAT1216: |
| case FS_FAT32: |
| if (fsname) lstrcpynW( fsname, fatW, fsname_len ); |
| if (filename_len) *filename_len = 255; |
| if (flags) *flags = FILE_CASE_PRESERVED_NAMES; /* FIXME */ |
| break; |
| default: |
| if (fsname) lstrcpynW( fsname, ntfsW, fsname_len ); |
| if (filename_len) *filename_len = 255; |
| if (flags) *flags = FILE_CASE_PRESERVED_NAMES; |
| break; |
| } |
| return TRUE; |
| } |
| |
| |
| /*********************************************************************** |
| * GetVolumeInformationA (KERNEL32.@) |
| */ |
| BOOL WINAPI GetVolumeInformationA( LPCSTR root, LPSTR label, |
| DWORD label_len, DWORD *serial, |
| DWORD *filename_len, DWORD *flags, |
| LPSTR fsname, DWORD fsname_len ) |
| { |
| WCHAR *rootW = NULL; |
| LPWSTR labelW, fsnameW; |
| BOOL ret; |
| |
| if (root && !(rootW = FILE_name_AtoW( root, FALSE ))) return FALSE; |
| |
| labelW = label ? HeapAlloc(GetProcessHeap(), 0, label_len * sizeof(WCHAR)) : NULL; |
| fsnameW = fsname ? HeapAlloc(GetProcessHeap(), 0, fsname_len * sizeof(WCHAR)) : NULL; |
| |
| if ((ret = GetVolumeInformationW(rootW, labelW, label_len, serial, |
| filename_len, flags, fsnameW, fsname_len))) |
| { |
| if (label) FILE_name_WtoA( labelW, -1, label, label_len ); |
| if (fsname) FILE_name_WtoA( fsnameW, -1, fsname, fsname_len ); |
| } |
| |
| if (labelW) HeapFree( GetProcessHeap(), 0, labelW ); |
| if (fsnameW) HeapFree( GetProcessHeap(), 0, fsnameW ); |
| return ret; |
| } |
| |
| |
| |
| /*********************************************************************** |
| * SetVolumeLabelW (KERNEL32.@) |
| */ |
| BOOL WINAPI SetVolumeLabelW( LPCWSTR root, LPCWSTR label ) |
| { |
| WCHAR device[] = {'\\','\\','.','\\','A',':',0}; |
| HANDLE handle; |
| enum fs_type type = FS_UNKNOWN; |
| |
| if (!root) |
| { |
| WCHAR path[MAX_PATH]; |
| GetCurrentDirectoryW( MAX_PATH, path ); |
| device[4] = path[0]; |
| } |
| else |
| { |
| if (!root[0] || root[1] != ':') |
| { |
| SetLastError( ERROR_INVALID_NAME ); |
| return FALSE; |
| } |
| device[4] = root[0]; |
| } |
| |
| /* try to open the device */ |
| |
| handle = CreateFileW( device, GENERIC_READ|GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE, |
| NULL, OPEN_EXISTING, 0, 0 ); |
| if (handle == INVALID_HANDLE_VALUE) |
| { |
| /* try read-only */ |
| handle = CreateFileW( device, GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE, |
| NULL, OPEN_EXISTING, 0, 0 ); |
| if (handle != INVALID_HANDLE_VALUE) |
| { |
| /* device can be read but not written, return error */ |
| CloseHandle( handle ); |
| SetLastError( ERROR_ACCESS_DENIED ); |
| return FALSE; |
| } |
| } |
| |
| if (handle != INVALID_HANDLE_VALUE) |
| { |
| BYTE superblock[SUPERBLOCK_SIZE]; |
| BOOL ret; |
| |
| type = VOLUME_ReadFATSuperblock( handle, superblock ); |
| ret = VOLUME_SetSuperblockLabel( type, handle, label ); |
| CloseHandle( handle ); |
| return ret; |
| } |
| else |
| { |
| TRACE( "cannot open device %s: err %ld\n", debugstr_w(device), GetLastError() ); |
| if (GetLastError() != ERROR_ACCESS_DENIED) return FALSE; |
| } |
| |
| /* we couldn't open the device, fallback to default strategy */ |
| |
| switch(GetDriveTypeW( root )) |
| { |
| case DRIVE_UNKNOWN: |
| case DRIVE_NO_ROOT_DIR: |
| SetLastError( ERROR_NOT_READY ); |
| break; |
| case DRIVE_REMOVABLE: |
| case DRIVE_FIXED: |
| { |
| WCHAR labelW[] = {'A',':','\\','.','w','i','n','d','o','w','s','-','l','a','b','e','l',0}; |
| |
| labelW[0] = device[4]; |
| handle = CreateFileW( labelW, GENERIC_WRITE, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, |
| CREATE_ALWAYS, 0, 0 ); |
| if (handle != INVALID_HANDLE_VALUE) |
| { |
| char buffer[64]; |
| DWORD size; |
| |
| if (!WideCharToMultiByte( CP_UNIXCP, 0, label, -1, buffer, sizeof(buffer), NULL, NULL )) |
| buffer[sizeof(buffer)-1] = 0; |
| WriteFile( handle, buffer, strlen(buffer), &size, NULL ); |
| CloseHandle( handle ); |
| return TRUE; |
| } |
| break; |
| } |
| case DRIVE_REMOTE: |
| case DRIVE_RAMDISK: |
| case DRIVE_CDROM: |
| SetLastError( ERROR_ACCESS_DENIED ); |
| break; |
| } |
| return FALSE; |
| } |
| |
| /*********************************************************************** |
| * SetVolumeLabelA (KERNEL32.@) |
| */ |
| BOOL WINAPI SetVolumeLabelA(LPCSTR root, LPCSTR volname) |
| { |
| WCHAR *rootW = NULL, *volnameW = NULL; |
| BOOL ret; |
| |
| if (root && !(rootW = FILE_name_AtoW( root, FALSE ))) return FALSE; |
| if (volname && !(volnameW = FILE_name_AtoW( volname, TRUE ))) return FALSE; |
| ret = SetVolumeLabelW( rootW, volnameW ); |
| if (volnameW) HeapFree( GetProcessHeap(), 0, volnameW ); |
| return ret; |
| } |
| |
| |
| /*********************************************************************** |
| * GetVolumeNameForVolumeMountPointW (KERNEL32.@) |
| */ |
| BOOL WINAPI GetVolumeNameForVolumeMountPointW(LPCWSTR str, LPWSTR dst, DWORD size) |
| { |
| FIXME("(%s, %p, %lx): stub\n", debugstr_w(str), dst, size); |
| return 0; |
| } |
| |
| |
| /*********************************************************************** |
| * DefineDosDeviceW (KERNEL32.@) |
| */ |
| BOOL WINAPI DefineDosDeviceW( DWORD flags, LPCWSTR devname, LPCWSTR targetpath ) |
| { |
| DWORD len, dosdev; |
| BOOL ret = FALSE; |
| char *path = NULL, *target, *p; |
| |
| if (!(flags & DDD_REMOVE_DEFINITION)) |
| { |
| if (!(flags & DDD_RAW_TARGET_PATH)) |
| { |
| FIXME( "(0x%08lx,%s,%s) DDD_RAW_TARGET_PATH flag not set, not supported yet\n", |
| flags, debugstr_w(devname), debugstr_w(targetpath) ); |
| SetLastError(ERROR_CALL_NOT_IMPLEMENTED); |
| return FALSE; |
| } |
| |
| len = WideCharToMultiByte( CP_UNIXCP, 0, targetpath, -1, NULL, 0, NULL, NULL ); |
| if ((target = HeapAlloc( GetProcessHeap(), 0, len ))) |
| { |
| WideCharToMultiByte( CP_UNIXCP, 0, targetpath, -1, target, len, NULL, NULL ); |
| for (p = target; *p; p++) if (*p == '\\') *p = '/'; |
| } |
| else |
| { |
| SetLastError( ERROR_NOT_ENOUGH_MEMORY ); |
| return FALSE; |
| } |
| } |
| else target = NULL; |
| |
| /* first check for a DOS device */ |
| |
| if ((dosdev = RtlIsDosDeviceName_U( devname ))) |
| { |
| WCHAR name[5]; |
| |
| memcpy( name, devname + HIWORD(dosdev)/sizeof(WCHAR), LOWORD(dosdev) ); |
| name[LOWORD(dosdev)/sizeof(WCHAR)] = 0; |
| path = get_dos_device_path( name ); |
| } |
| else if (isalphaW(devname[0]) && devname[1] == ':' && !devname[2]) /* drive mapping */ |
| { |
| path = get_dos_device_path( devname ); |
| } |
| else SetLastError( ERROR_FILE_NOT_FOUND ); |
| |
| if (path) |
| { |
| if (target) |
| { |
| TRACE( "creating symlink %s -> %s\n", path, target ); |
| unlink( path ); |
| if (!symlink( target, path )) ret = TRUE; |
| else FILE_SetDosError(); |
| } |
| else |
| { |
| TRACE( "removing symlink %s\n", path ); |
| if (!unlink( path )) ret = TRUE; |
| else FILE_SetDosError(); |
| } |
| HeapFree( GetProcessHeap(), 0, path ); |
| } |
| if (target) HeapFree( GetProcessHeap(), 0, target ); |
| return ret; |
| } |
| |
| |
| /*********************************************************************** |
| * DefineDosDeviceA (KERNEL32.@) |
| */ |
| BOOL WINAPI DefineDosDeviceA(DWORD flags, LPCSTR devname, LPCSTR targetpath) |
| { |
| WCHAR *devW, *targetW = NULL; |
| BOOL ret; |
| |
| if (!(devW = FILE_name_AtoW( devname, FALSE ))) return FALSE; |
| if (targetpath && !(targetW = FILE_name_AtoW( targetpath, TRUE ))) return FALSE; |
| ret = DefineDosDeviceW(flags, devW, targetW); |
| if (targetW) HeapFree( GetProcessHeap(), 0, targetW ); |
| return ret; |
| } |
| |
| |
| /*********************************************************************** |
| * QueryDosDeviceW (KERNEL32.@) |
| * |
| * returns array of strings terminated by \0, terminated by \0 |
| */ |
| DWORD WINAPI QueryDosDeviceW( LPCWSTR devname, LPWSTR target, DWORD bufsize ) |
| { |
| static const WCHAR auxW[] = {'A','U','X',0}; |
| static const WCHAR nulW[] = {'N','U','L',0}; |
| static const WCHAR prnW[] = {'P','R','N',0}; |
| static const WCHAR comW[] = {'C','O','M',0}; |
| static const WCHAR lptW[] = {'L','P','T',0}; |
| static const WCHAR rootW[] = {'A',':','\\',0}; |
| static const WCHAR com0W[] = {'\\','?','?','\\','C','O','M','0',0}; |
| static const WCHAR com1W[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\','C','O','M','1',0,0}; |
| static const WCHAR lpt1W[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\','L','P','T','1',0,0}; |
| |
| UNICODE_STRING nt_name; |
| ANSI_STRING unix_name; |
| WCHAR nt_buffer[10]; |
| NTSTATUS status; |
| |
| if (!bufsize) |
| { |
| SetLastError( ERROR_INSUFFICIENT_BUFFER ); |
| return 0; |
| } |
| |
| if (devname) |
| { |
| WCHAR *p, name[5]; |
| char *path, *link; |
| DWORD dosdev, ret = 0; |
| |
| if ((dosdev = RtlIsDosDeviceName_U( devname ))) |
| { |
| memcpy( name, devname + HIWORD(dosdev)/sizeof(WCHAR), LOWORD(dosdev) ); |
| name[LOWORD(dosdev)/sizeof(WCHAR)] = 0; |
| } |
| else if (devname[0] && devname[1] == ':' && !devname[2]) |
| { |
| memcpy( name, devname, 3 * sizeof(WCHAR) ); |
| } |
| else |
| { |
| SetLastError( ERROR_BAD_PATHNAME ); |
| return 0; |
| } |
| |
| if (!(path = get_dos_device_path( name ))) return 0; |
| link = read_symlink( path ); |
| HeapFree( GetProcessHeap(), 0, path ); |
| |
| if (link) |
| { |
| ret = MultiByteToWideChar( CP_UNIXCP, 0, link, -1, target, bufsize ); |
| HeapFree( GetProcessHeap(), 0, link ); |
| } |
| else if (dosdev) /* look for device defaults */ |
| { |
| if (!strcmpiW( name, auxW )) |
| { |
| if (bufsize >= sizeof(com1W)/sizeof(WCHAR)) |
| { |
| memcpy( target, com1W, sizeof(com1W) ); |
| ret = sizeof(com1W)/sizeof(WCHAR); |
| } |
| else SetLastError( ERROR_INSUFFICIENT_BUFFER ); |
| return ret; |
| } |
| if (!strcmpiW( name, prnW )) |
| { |
| if (bufsize >= sizeof(lpt1W)/sizeof(WCHAR)) |
| { |
| memcpy( target, lpt1W, sizeof(lpt1W) ); |
| ret = sizeof(lpt1W)/sizeof(WCHAR); |
| } |
| else SetLastError( ERROR_INSUFFICIENT_BUFFER ); |
| return ret; |
| } |
| |
| nt_buffer[0] = '\\'; |
| nt_buffer[1] = '?'; |
| nt_buffer[2] = '?'; |
| nt_buffer[3] = '\\'; |
| strcpyW( nt_buffer + 4, name ); |
| RtlInitUnicodeString( &nt_name, nt_buffer ); |
| status = wine_nt_to_unix_file_name( &nt_name, &unix_name, FILE_OPEN, TRUE ); |
| if (status) SetLastError( RtlNtStatusToDosError(status) ); |
| else |
| { |
| ret = MultiByteToWideChar( CP_UNIXCP, 0, unix_name.Buffer, -1, target, bufsize ); |
| RtlFreeAnsiString( &unix_name ); |
| } |
| } |
| |
| if (ret) |
| { |
| if (ret < bufsize) target[ret++] = 0; /* add an extra null */ |
| for (p = target; *p; p++) if (*p == '/') *p = '\\'; |
| } |
| |
| return ret; |
| } |
| else /* return a list of all devices */ |
| { |
| WCHAR *p = target; |
| int i; |
| |
| if (bufsize <= (sizeof(auxW)+sizeof(nulW)+sizeof(prnW))/sizeof(WCHAR)) |
| { |
| SetLastError( ERROR_INSUFFICIENT_BUFFER ); |
| return 0; |
| } |
| |
| memcpy( p, auxW, sizeof(auxW) ); |
| p += sizeof(auxW) / sizeof(WCHAR); |
| memcpy( p, nulW, sizeof(nulW) ); |
| p += sizeof(nulW) / sizeof(WCHAR); |
| memcpy( p, prnW, sizeof(prnW) ); |
| p += sizeof(prnW) / sizeof(WCHAR); |
| |
| strcpyW( nt_buffer, com0W ); |
| RtlInitUnicodeString( &nt_name, nt_buffer ); |
| |
| for (i = 1; i <= 9; i++) |
| { |
| nt_buffer[7] = '0' + i; |
| if (!wine_nt_to_unix_file_name( &nt_name, &unix_name, FILE_OPEN, TRUE )) |
| { |
| RtlFreeAnsiString( &unix_name ); |
| if (p + 5 >= target + bufsize) |
| { |
| SetLastError( ERROR_INSUFFICIENT_BUFFER ); |
| return 0; |
| } |
| strcpyW( p, comW ); |
| p[3] = '0' + i; |
| p[4] = 0; |
| p += 5; |
| } |
| } |
| strcpyW( nt_buffer + 4, lptW ); |
| for (i = 1; i <= 9; i++) |
| { |
| nt_buffer[7] = '0' + i; |
| if (!wine_nt_to_unix_file_name( &nt_name, &unix_name, FILE_OPEN, TRUE )) |
| { |
| RtlFreeAnsiString( &unix_name ); |
| if (p + 5 >= target + bufsize) |
| { |
| SetLastError( ERROR_INSUFFICIENT_BUFFER ); |
| return 0; |
| } |
| strcpyW( p, lptW ); |
| p[3] = '0' + i; |
| p[4] = 0; |
| p += 5; |
| } |
| } |
| |
| strcpyW( nt_buffer + 4, rootW ); |
| RtlInitUnicodeString( &nt_name, nt_buffer ); |
| |
| for (i = 0; i < 26; i++) |
| { |
| nt_buffer[4] = 'a' + i; |
| if (!wine_nt_to_unix_file_name( &nt_name, &unix_name, FILE_OPEN, TRUE )) |
| { |
| RtlFreeAnsiString( &unix_name ); |
| if (p + 3 >= target + bufsize) |
| { |
| SetLastError( ERROR_INSUFFICIENT_BUFFER ); |
| return 0; |
| } |
| *p++ = 'A' + i; |
| *p++ = ':'; |
| *p++ = 0; |
| } |
| } |
| *p++ = 0; /* terminating null */ |
| return p - target; |
| } |
| } |
| |
| |
| /*********************************************************************** |
| * QueryDosDeviceA (KERNEL32.@) |
| * |
| * returns array of strings terminated by \0, terminated by \0 |
| */ |
| DWORD WINAPI QueryDosDeviceA( LPCSTR devname, LPSTR target, DWORD bufsize ) |
| { |
| DWORD ret = 0, retW; |
| WCHAR *devnameW; |
| LPWSTR targetW; |
| |
| if (!(devnameW = FILE_name_AtoW( devname, FALSE ))) return 0; |
| |
| targetW = HeapAlloc( GetProcessHeap(),0, bufsize * sizeof(WCHAR) ); |
| if (!targetW) |
| { |
| SetLastError( ERROR_NOT_ENOUGH_MEMORY ); |
| return 0; |
| } |
| |
| retW = QueryDosDeviceW(devnameW, targetW, bufsize); |
| |
| ret = FILE_name_WtoA( targetW, retW, target, bufsize ); |
| |
| HeapFree(GetProcessHeap(), 0, targetW); |
| return ret; |
| } |
| |
| |
| /*********************************************************************** |
| * GetLogicalDrives (KERNEL32.@) |
| */ |
| DWORD WINAPI GetLogicalDrives(void) |
| { |
| const char *config_dir = wine_get_config_dir(); |
| struct stat st; |
| char *buffer, *dev; |
| DWORD ret = 0; |
| int i; |
| |
| if (!(buffer = HeapAlloc( GetProcessHeap(), 0, strlen(config_dir) + sizeof("/dosdevices/a:") ))) |
| { |
| SetLastError( ERROR_NOT_ENOUGH_MEMORY ); |
| return 0; |
| } |
| strcpy( buffer, config_dir ); |
| strcat( buffer, "/dosdevices/a:" ); |
| dev = buffer + strlen(buffer) - 2; |
| |
| for (i = 0; i < 26; i++) |
| { |
| *dev = 'a' + i; |
| if (!stat( buffer, &st )) ret |= (1 << i); |
| } |
| HeapFree( GetProcessHeap(), 0, buffer ); |
| return ret; |
| } |
| |
| |
| /*********************************************************************** |
| * GetLogicalDriveStringsA (KERNEL32.@) |
| */ |
| UINT WINAPI GetLogicalDriveStringsA( UINT len, LPSTR buffer ) |
| { |
| DWORD drives = GetLogicalDrives(); |
| UINT drive, count; |
| |
| for (drive = count = 0; drive < 26; drive++) if (drives & (1 << drive)) count++; |
| if ((count * 4) + 1 > len) return count * 4 + 1; |
| |
| for (drive = 0; drive < 26; drive++) |
| { |
| if (drives & (1 << drive)) |
| { |
| *buffer++ = 'a' + drive; |
| *buffer++ = ':'; |
| *buffer++ = '\\'; |
| *buffer++ = 0; |
| } |
| } |
| *buffer = 0; |
| return count * 4; |
| } |
| |
| |
| /*********************************************************************** |
| * GetLogicalDriveStringsW (KERNEL32.@) |
| */ |
| UINT WINAPI GetLogicalDriveStringsW( UINT len, LPWSTR buffer ) |
| { |
| DWORD drives = GetLogicalDrives(); |
| UINT drive, count; |
| |
| for (drive = count = 0; drive < 26; drive++) if (drives & (1 << drive)) count++; |
| if ((count * 4) + 1 > len) return count * 4 + 1; |
| |
| for (drive = 0; drive < 26; drive++) |
| { |
| if (drives & (1 << drive)) |
| { |
| *buffer++ = 'a' + drive; |
| *buffer++ = ':'; |
| *buffer++ = '\\'; |
| *buffer++ = 0; |
| } |
| } |
| *buffer = 0; |
| return count * 4; |
| } |
| |
| |
| /*********************************************************************** |
| * GetDriveTypeW (KERNEL32.@) |
| * |
| * Returns the type of the disk drive specified. If root is NULL the |
| * root of the current directory is used. |
| * |
| * RETURNS |
| * |
| * Type of drive (from Win32 SDK): |
| * |
| * DRIVE_UNKNOWN unable to find out anything about the drive |
| * DRIVE_NO_ROOT_DIR nonexistent root dir |
| * DRIVE_REMOVABLE the disk can be removed from the machine |
| * DRIVE_FIXED the disk can not be removed from the machine |
| * DRIVE_REMOTE network disk |
| * DRIVE_CDROM CDROM drive |
| * DRIVE_RAMDISK virtual disk in RAM |
| */ |
| UINT WINAPI GetDriveTypeW(LPCWSTR root) /* [in] String describing drive */ |
| { |
| FILE_FS_DEVICE_INFORMATION info; |
| IO_STATUS_BLOCK io; |
| NTSTATUS status; |
| HANDLE handle; |
| UINT ret; |
| |
| if (!open_device_root( root, &handle )) return DRIVE_NO_ROOT_DIR; |
| |
| status = NtQueryVolumeInformationFile( handle, &io, &info, sizeof(info), FileFsDeviceInformation ); |
| NtClose( handle ); |
| if (status != STATUS_SUCCESS) |
| { |
| SetLastError( RtlNtStatusToDosError(status) ); |
| ret = DRIVE_UNKNOWN; |
| } |
| else if ((ret = get_registry_drive_type( root )) == DRIVE_UNKNOWN) |
| { |
| switch (info.DeviceType) |
| { |
| case FILE_DEVICE_CD_ROM_FILE_SYSTEM: ret = DRIVE_CDROM; break; |
| case FILE_DEVICE_VIRTUAL_DISK: ret = DRIVE_RAMDISK; break; |
| case FILE_DEVICE_NETWORK_FILE_SYSTEM: ret = DRIVE_REMOTE; break; |
| case FILE_DEVICE_DISK_FILE_SYSTEM: |
| if (info.Characteristics & FILE_REMOTE_DEVICE) ret = DRIVE_REMOTE; |
| else if (info.Characteristics & FILE_REMOVABLE_MEDIA) ret = DRIVE_REMOVABLE; |
| else ret = DRIVE_FIXED; |
| break; |
| default: |
| ret = DRIVE_UNKNOWN; |
| break; |
| } |
| } |
| TRACE( "%s -> %d\n", debugstr_w(root), ret ); |
| return ret; |
| } |
| |
| |
| /*********************************************************************** |
| * GetDriveTypeA (KERNEL32.@) |
| */ |
| UINT WINAPI GetDriveTypeA( LPCSTR root ) |
| { |
| WCHAR *rootW = NULL; |
| |
| if (root && !(rootW = FILE_name_AtoW( root, FALSE ))) return DRIVE_NO_ROOT_DIR; |
| return GetDriveTypeW( rootW ); |
| } |
| |
| |
| /*********************************************************************** |
| * GetDiskFreeSpaceExW (KERNEL32.@) |
| * |
| * This function is used to acquire the size of the available and |
| * total space on a logical volume. |
| * |
| * RETURNS |
| * |
| * Zero on failure, nonzero upon success. Use GetLastError to obtain |
| * detailed error information. |
| * |
| */ |
| BOOL WINAPI GetDiskFreeSpaceExW( LPCWSTR root, PULARGE_INTEGER avail, |
| PULARGE_INTEGER total, PULARGE_INTEGER totalfree ) |
| { |
| FILE_FS_SIZE_INFORMATION info; |
| IO_STATUS_BLOCK io; |
| NTSTATUS status; |
| HANDLE handle; |
| UINT units; |
| |
| TRACE( "%s,%p,%p,%p\n", debugstr_w(root), avail, total, totalfree ); |
| |
| if (!open_device_root( root, &handle )) return FALSE; |
| |
| status = NtQueryVolumeInformationFile( handle, &io, &info, sizeof(info), FileFsSizeInformation ); |
| NtClose( handle ); |
| if (status != STATUS_SUCCESS) |
| { |
| SetLastError( RtlNtStatusToDosError(status) ); |
| return FALSE; |
| } |
| |
| units = info.SectorsPerAllocationUnit * info.BytesPerSector; |
| if (total) total->QuadPart = info.TotalAllocationUnits.QuadPart * units; |
| if (totalfree) totalfree->QuadPart = info.AvailableAllocationUnits.QuadPart * units; |
| /* FIXME: this one should take quotas into account */ |
| if (avail) avail->QuadPart = info.AvailableAllocationUnits.QuadPart * units; |
| return TRUE; |
| } |
| |
| |
| /*********************************************************************** |
| * GetDiskFreeSpaceExA (KERNEL32.@) |
| */ |
| BOOL WINAPI GetDiskFreeSpaceExA( LPCSTR root, PULARGE_INTEGER avail, |
| PULARGE_INTEGER total, PULARGE_INTEGER totalfree ) |
| { |
| WCHAR *rootW = NULL; |
| |
| if (root && !(rootW = FILE_name_AtoW( root, FALSE ))) return FALSE; |
| return GetDiskFreeSpaceExW( rootW, avail, total, totalfree ); |
| } |
| |
| |
| /*********************************************************************** |
| * GetDiskFreeSpaceW (KERNEL32.@) |
| */ |
| BOOL WINAPI GetDiskFreeSpaceW( LPCWSTR root, LPDWORD cluster_sectors, |
| LPDWORD sector_bytes, LPDWORD free_clusters, |
| LPDWORD total_clusters ) |
| { |
| FILE_FS_SIZE_INFORMATION info; |
| IO_STATUS_BLOCK io; |
| NTSTATUS status; |
| HANDLE handle; |
| UINT units; |
| |
| TRACE( "%s,%p,%p,%p,%p\n", debugstr_w(root), |
| cluster_sectors, sector_bytes, free_clusters, total_clusters ); |
| |
| if (!open_device_root( root, &handle )) return FALSE; |
| |
| status = NtQueryVolumeInformationFile( handle, &io, &info, sizeof(info), FileFsSizeInformation ); |
| NtClose( handle ); |
| if (status != STATUS_SUCCESS) |
| { |
| SetLastError( RtlNtStatusToDosError(status) ); |
| return FALSE; |
| } |
| |
| units = info.SectorsPerAllocationUnit * info.BytesPerSector; |
| |
| /* cap the size and available at 2GB as per specs */ |
| if (info.AvailableAllocationUnits.QuadPart * units > 0x7fffffff) |
| info.AvailableAllocationUnits.QuadPart = 0x7fffffff / units; |
| if (info.TotalAllocationUnits.QuadPart * units > 0x7fffffff) |
| info.TotalAllocationUnits.QuadPart = 0x7fffffff / units; |
| |
| if (cluster_sectors) *cluster_sectors = info.SectorsPerAllocationUnit; |
| if (sector_bytes) *sector_bytes = info.BytesPerSector; |
| if (free_clusters) *free_clusters = info.AvailableAllocationUnits.u.LowPart; |
| if (total_clusters) *total_clusters = info.TotalAllocationUnits.u.LowPart; |
| return TRUE; |
| } |
| |
| |
| /*********************************************************************** |
| * GetDiskFreeSpaceA (KERNEL32.@) |
| */ |
| BOOL WINAPI GetDiskFreeSpaceA( LPCSTR root, LPDWORD cluster_sectors, |
| LPDWORD sector_bytes, LPDWORD free_clusters, |
| LPDWORD total_clusters ) |
| { |
| WCHAR *rootW = NULL; |
| |
| if (root && !(rootW = FILE_name_AtoW( root, FALSE ))) return FALSE; |
| return GetDiskFreeSpaceW( rootW, cluster_sectors, sector_bytes, free_clusters, total_clusters ); |
| } |
