diff --git a/dlls/kernel32/kernel32.spec b/dlls/kernel32/kernel32.spec index 86575c2b994..cb3edf384cb 100644 --- a/dlls/kernel32/kernel32.spec +++ b/dlls/kernel32/kernel32.spec @@ -355,7 +355,7 @@ @ stdcall DecodePointer(ptr) ntdll.RtlDecodePointer @ stdcall DecodeSystemPointer(ptr) ntdll.RtlDecodeSystemPointer @ stdcall DefineDosDeviceA(long str str) -@ stdcall DefineDosDeviceW(long wstr wstr) +@ stdcall -import DefineDosDeviceW(long wstr wstr) @ stdcall -import DelayLoadFailureHook(str str) @ stdcall DeleteAtom(long) # @ stub DeleteBoundaryDescriptor @@ -374,7 +374,7 @@ @ stdcall -arch=x86_64 DeleteUmsCompletionList(ptr) @ stdcall -arch=x86_64 DeleteUmsThreadContext(ptr) @ stdcall DeleteVolumeMountPointA(str) -@ stdcall DeleteVolumeMountPointW(wstr) +@ stdcall -import DeleteVolumeMountPointW(wstr) @ stdcall -arch=x86_64 DequeueUmsCompletionListItems(ptr long ptr) @ stdcall DeviceIoControl(long long ptr long ptr long ptr ptr) @ stdcall -import DisableThreadLibraryCalls(long) @@ -492,7 +492,7 @@ @ stdcall FindFirstVolumeA(ptr long) @ stdcall FindFirstVolumeMountPointA(str ptr long) @ stdcall FindFirstVolumeMountPointW(wstr ptr long) -@ stdcall FindFirstVolumeW(ptr long) +@ stdcall -import FindFirstVolumeW(ptr long) @ stdcall -import FindNextChangeNotification(long) @ stdcall -import FindNextFileA(long ptr) # @ stub FindNextFileNameW @@ -501,7 +501,7 @@ @ stdcall FindNextVolumeA(long ptr long) @ stub FindNextVolumeMountPointA @ stub FindNextVolumeMountPointW -@ stdcall FindNextVolumeW(long ptr long) +@ stdcall -import FindNextVolumeW(long ptr long) @ stdcall -import FindNLSString(long long wstr long wstr long ptr) @ stdcall -import FindNLSStringEx(wstr long wstr long wstr long ptr ptr ptr long) @ stdcall FindResourceA(long str str) @@ -509,7 +509,7 @@ @ stdcall -import FindResourceExW(long wstr wstr long) @ stdcall -import FindResourceW(long wstr wstr) @ stdcall -import FindStringOrdinal(long wstr long wstr long long) -@ stdcall FindVolumeClose(ptr) +@ stdcall -import FindVolumeClose(ptr) @ stdcall FindVolumeMountPointClose(ptr) @ stdcall -import FlsAlloc(ptr) @ stdcall -import FlsFree(long) @@ -645,14 +645,14 @@ @ stdcall GetDefaultCommConfigW(wstr ptr ptr) @ stub GetDefaultSortkeySize @ stdcall GetDevicePowerState(long ptr) -@ stdcall GetDiskFreeSpaceA(str ptr ptr ptr ptr) -@ stdcall GetDiskFreeSpaceExA (str ptr ptr ptr) -@ stdcall GetDiskFreeSpaceExW (wstr ptr ptr ptr) -@ stdcall GetDiskFreeSpaceW(wstr ptr ptr ptr ptr) +@ stdcall -import GetDiskFreeSpaceA(str ptr ptr ptr ptr) +@ stdcall -import GetDiskFreeSpaceExA (str ptr ptr ptr) +@ stdcall -import GetDiskFreeSpaceExW (wstr ptr ptr ptr) +@ stdcall -import GetDiskFreeSpaceW(wstr ptr ptr ptr ptr) @ stdcall GetDllDirectoryA(long ptr) @ stdcall GetDllDirectoryW(long ptr) -@ stdcall GetDriveTypeA(str) -@ stdcall GetDriveTypeW(wstr) +@ stdcall -import GetDriveTypeA(str) +@ stdcall -import GetDriveTypeW(wstr) # @ stub GetDurationFormat # @ stub GetDurationFormatEx @ stdcall -import GetDynamicTimeZoneInformation(ptr) @@ -707,8 +707,8 @@ @ stdcall -import GetLocaleInfoW(long long ptr long) @ stdcall -import GetLocaleInfoEx(wstr long ptr long) @ stdcall GetLogicalDriveStringsA(long ptr) -@ stdcall GetLogicalDriveStringsW(long ptr) -@ stdcall GetLogicalDrives() +@ stdcall -import GetLogicalDriveStringsW(long ptr) +@ stdcall -import GetLogicalDrives() @ stdcall -import GetLogicalProcessorInformation(ptr ptr) @ stdcall -import GetLogicalProcessorInformationEx(long ptr ptr) @ stdcall -import GetLongPathNameA(str ptr long) @@ -880,15 +880,15 @@ @ stdcall -import GetVersion() @ stdcall -import GetVersionExA(ptr) @ stdcall -import GetVersionExW(ptr) -@ stdcall GetVolumeInformationA(str ptr long ptr ptr ptr ptr long) -@ stdcall GetVolumeInformationByHandleW(ptr ptr long ptr ptr ptr ptr long) -@ stdcall GetVolumeInformationW(wstr ptr long ptr ptr ptr ptr long) +@ stdcall -import GetVolumeInformationA(str ptr long ptr ptr ptr ptr long) +@ stdcall -import GetVolumeInformationByHandleW(ptr ptr long ptr ptr ptr ptr long) +@ stdcall -import GetVolumeInformationW(wstr ptr long ptr ptr ptr ptr long) @ stdcall GetVolumeNameForVolumeMountPointA(str ptr long) -@ stdcall GetVolumeNameForVolumeMountPointW(wstr ptr long) +@ stdcall -import GetVolumeNameForVolumeMountPointW(wstr ptr long) @ stdcall GetVolumePathNameA(str ptr long) -@ stdcall GetVolumePathNameW(wstr ptr long) +@ stdcall -import GetVolumePathNameW(wstr ptr long) @ stdcall GetVolumePathNamesForVolumeNameA(str ptr long ptr) -@ stdcall GetVolumePathNamesForVolumeNameW(wstr ptr long ptr) +@ stdcall -import GetVolumePathNamesForVolumeNameW(wstr ptr long ptr) @ stdcall -import GetWindowsDirectoryA(ptr long) @ stdcall -import GetWindowsDirectoryW(ptr long) @ stdcall -import GetWriteWatch(long ptr long ptr ptr ptr) @@ -1164,7 +1164,7 @@ @ stdcall -import QueryActCtxW(long ptr ptr long ptr long ptr) @ stdcall QueryDepthSList(ptr) ntdll.RtlQueryDepthSList @ stdcall QueryDosDeviceA(str ptr long) -@ stdcall QueryDosDeviceW(wstr ptr long) +@ stdcall -import QueryDosDeviceW(wstr ptr long) @ stdcall QueryFullProcessImageNameA(ptr long ptr ptr) @ stdcall QueryFullProcessImageNameW(ptr long ptr ptr) # @ stub QueryIdleProcessorCycleTime diff --git a/dlls/kernel32/volume.c b/dlls/kernel32/volume.c index 7555dd9e95d..7545aedf040 100644 --- a/dlls/kernel32/volume.c +++ b/dlls/kernel32/volume.c @@ -37,24 +37,14 @@ #include "winternl.h" #include "winioctl.h" #include "ntddcdrm.h" -#define WINE_MOUNTMGR_EXTENSIONS -#include "ddk/mountmgr.h" #include "ddk/wdm.h" #include "kernel_private.h" -#include "wine/library.h" -#include "wine/unicode.h" #include "wine/debug.h" WINE_DEFAULT_DEBUG_CHANNEL(volume); #define BLOCK_SIZE 2048 #define SUPERBLOCK_SIZE BLOCK_SIZE -#define SYMBOLIC_LINK_QUERY 0x0001 - -#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)) @@ -69,171 +59,6 @@ enum fs_type FS_UDF /* For reference [E] = Ecma-167.pdf, [U] = udf260.pdf */ }; -/* read the contents of an NT symlink object */ -static NTSTATUS read_nt_symlink( const WCHAR *name, WCHAR *target, DWORD size ) -{ - NTSTATUS status; - OBJECT_ATTRIBUTES attr; - UNICODE_STRING nameW; - HANDLE handle; - - attr.Length = sizeof(attr); - attr.RootDirectory = 0; - attr.Attributes = OBJ_CASE_INSENSITIVE; - attr.ObjectName = &nameW; - attr.SecurityDescriptor = NULL; - attr.SecurityQualityOfService = NULL; - RtlInitUnicodeString( &nameW, name ); - - if (!(status = NtOpenSymbolicLinkObject( &handle, SYMBOLIC_LINK_QUERY, &attr ))) - { - UNICODE_STRING targetW; - targetW.Buffer = target; - targetW.MaximumLength = (size - 1) * sizeof(WCHAR); - status = NtQuerySymbolicLinkObject( handle, &targetW, NULL ); - if (!status) target[targetW.Length / sizeof(WCHAR)] = 0; - NtClose( handle ); - } - return status; -} - -/* 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, SYNCHRONIZE, &attr, &io, 0, - FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT ); - RtlFreeUnicodeString( &nt_name ); - return set_ntstatus( status ); -} - -/* query the type of a drive from the mount manager */ -static DWORD get_mountmgr_drive_type( LPCWSTR root ) -{ - HANDLE mgr; - struct mountmgr_unix_drive data; - DWORD br; - - memset( &data, 0, sizeof(data) ); - if (root) data.letter = root[0]; - else - { - WCHAR curdir[MAX_PATH]; - GetCurrentDirectoryW( MAX_PATH, curdir ); - if (curdir[1] != ':' || curdir[2] != '\\') return DRIVE_UNKNOWN; - data.letter = curdir[0]; - } - - mgr = CreateFileW( MOUNTMGR_DOS_DEVICE_NAME, GENERIC_READ, - FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, 0 ); - if (mgr == INVALID_HANDLE_VALUE) return DRIVE_UNKNOWN; - - if (!DeviceIoControl( mgr, IOCTL_MOUNTMGR_QUERY_UNIX_DRIVE, &data, sizeof(data), &data, - sizeof(data), &br, NULL ) && GetLastError() != ERROR_MORE_DATA) - data.type = DRIVE_UNKNOWN; - - CloseHandle( mgr ); - return data.type; -} - -/* get the label by reading it from a file at the root of the filesystem */ -static void get_filesystem_label( const UNICODE_STRING *device, WCHAR *label, DWORD len ) -{ - static const WCHAR labelW[] = {'.','w','i','n','d','o','w','s','-','l','a','b','e','l',0}; - HANDLE handle; - UNICODE_STRING name; - IO_STATUS_BLOCK io; - OBJECT_ATTRIBUTES attr; - - label[0] = 0; - - attr.Length = sizeof(attr); - attr.RootDirectory = 0; - attr.Attributes = OBJ_CASE_INSENSITIVE; - attr.ObjectName = &name; - attr.SecurityDescriptor = NULL; - attr.SecurityQualityOfService = NULL; - - name.MaximumLength = device->Length + sizeof(labelW); - name.Length = name.MaximumLength - sizeof(WCHAR); - if (!(name.Buffer = HeapAlloc( GetProcessHeap(), 0, name.MaximumLength ))) return; - - memcpy( name.Buffer, device->Buffer, device->Length ); - memcpy( name.Buffer + device->Length / sizeof(WCHAR), labelW, sizeof(labelW) ); - if (!NtOpenFile( &handle, GENERIC_READ | SYNCHRONIZE, &attr, &io, FILE_SHARE_READ|FILE_SHARE_WRITE, - FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT )) - { - 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, len )) - label[len-1] = 0; - } - RtlFreeUnicodeString( &name ); -} - -/* get the serial number by reading it from a file at the root of the filesystem */ -static DWORD get_filesystem_serial( const UNICODE_STRING *device ) -{ - static const WCHAR serialW[] = {'.','w','i','n','d','o','w','s','-','s','e','r','i','a','l',0}; - HANDLE handle; - UNICODE_STRING name; - IO_STATUS_BLOCK io; - OBJECT_ATTRIBUTES attr; - DWORD ret = 0; - - attr.Length = sizeof(attr); - attr.RootDirectory = 0; - attr.Attributes = OBJ_CASE_INSENSITIVE; - attr.ObjectName = &name; - attr.SecurityDescriptor = NULL; - attr.SecurityQualityOfService = NULL; - - name.MaximumLength = device->Length + sizeof(serialW); - name.Length = name.MaximumLength - sizeof(WCHAR); - if (!(name.Buffer = HeapAlloc( GetProcessHeap(), 0, name.MaximumLength ))) return 0; - - memcpy( name.Buffer, device->Buffer, device->Length ); - memcpy( name.Buffer + device->Length / sizeof(WCHAR), serialW, sizeof(serialW) ); - if (!NtOpenFile( &handle, GENERIC_READ | SYNCHRONIZE, &attr, &io, FILE_SHARE_READ|FILE_SHARE_WRITE, - FILE_SYNCHRONOUS_IO_NONALERT )) - { - char buffer[32]; - DWORD size; - - if (!ReadFile( handle, buffer, sizeof(buffer)-1, &size, NULL )) size = 0; - CloseHandle( handle ); - buffer[size] = 0; - ret = strtoul( buffer, NULL, 16 ); - } - RtlFreeUnicodeString( &name ); - return ret; -} - - /****************************************************************** * VOLUME_FindCdRomDataBestVoldesc */ @@ -383,461 +208,6 @@ static enum fs_type VOLUME_ReadCDSuperblock( HANDLE handle, BYTE *buff ) } -/************************************************************************** - * UDF_Find_PVD - * Find the Primary Volume Descriptor - */ -static BOOL UDF_Find_PVD( HANDLE handle, BYTE pvd[] ) -{ - unsigned int i; - DWORD offset; - INT locations[] = { 256, -1, -257, 512 }; - - for(i=0; i 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; - } - case FS_UDF: - { - BYTE pvd[BLOCK_SIZE]; - - if(!UDF_Find_PVD(handle, pvd)) - { - label_len = 0; - break; - } - - /* [E] 3/10.1.4 and [U] 2.1.1 */ - if(pvd[24]==8) - { - label_ptr = pvd + 24 + 1; - label_len = pvd[24+32-1]; - break; - } - else - { - unsigned int i; - - label_len = 1 + pvd[24+32-1]; - for(i=0; iLastTrack - 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 fat32W[] = {'F','A','T','3','2',0}; - static const WCHAR ntfsW[] = {'N','T','F','S',0}; - static const WCHAR cdfsW[] = {'C','D','F','S',0}; - static const WCHAR udfW[] = {'U','D','F',0}; - static const WCHAR default_rootW[] = {'\\',0}; - - HANDLE handle; - NTSTATUS status; - UNICODE_STRING nt_name; - IO_STATUS_BLOCK io; - OBJECT_ATTRIBUTES attr; - FILE_FS_DEVICE_INFORMATION info; - unsigned int i; - enum fs_type type = FS_UNKNOWN; - BOOL ret = FALSE; - - if (!root) root = default_rootW; - if (!RtlDosPathNameToNtPathName_U( root, &nt_name, NULL, NULL )) - { - SetLastError( ERROR_PATH_NOT_FOUND ); - return FALSE; - } - /* there must be exactly one backslash in the name, at the end */ - for (i = 4; i < nt_name.Length / sizeof(WCHAR); i++) if (nt_name.Buffer[i] == '\\') break; - if (i != nt_name.Length / sizeof(WCHAR) - 1) - { - /* check if root contains an explicit subdir */ - if (root[0] && root[1] == ':') root += 2; - while (*root == '\\') root++; - if (strchrW( root, '\\' )) - SetLastError( ERROR_DIR_NOT_ROOT ); - else - SetLastError( ERROR_INVALID_NAME ); - goto done; - } - - /* try to open the device */ - - attr.Length = sizeof(attr); - attr.RootDirectory = 0; - attr.Attributes = OBJ_CASE_INSENSITIVE; - attr.ObjectName = &nt_name; - attr.SecurityDescriptor = NULL; - attr.SecurityQualityOfService = NULL; - - nt_name.Length -= sizeof(WCHAR); /* without trailing slash */ - status = NtOpenFile( &handle, GENERIC_READ | SYNCHRONIZE, &attr, &io, FILE_SHARE_READ | FILE_SHARE_WRITE, - FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT ); - nt_name.Length += sizeof(WCHAR); - - if (status == STATUS_SUCCESS) - { - 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(nt_name.Buffer) ); - 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 ); - } - TRACE( "%s: found fs type %d\n", debugstr_w(nt_name.Buffer), type ); - if (type == FS_ERROR) - { - CloseHandle( handle ); - goto done; - } - - if (label && label_len) VOLUME_GetSuperblockLabel( &nt_name, handle, type, superblock, label, label_len ); - if (serial) *serial = VOLUME_GetSuperblockSerial( &nt_name, handle, type, superblock ); - CloseHandle( handle ); - goto fill_fs_info; - } - else - { - TRACE( "cannot open device %s: %x\n", debugstr_w(nt_name.Buffer), status ); - if (status == STATUS_ACCESS_DENIED) - MESSAGE( "wine: Read access denied for device %s, FS volume label and serial are not available.\n", debugstr_w(nt_name.Buffer) ); - } - /* we couldn't open the device, fallback to default strategy */ - - if (!set_ntstatus( NtOpenFile( &handle, SYNCHRONIZE, &attr, &io, 0, - FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT ))) - goto done; - - status = NtQueryVolumeInformationFile( handle, &io, &info, sizeof(info), FileFsDeviceInformation ); - NtClose( handle ); - if (!set_ntstatus( status )) goto done; - - if (info.DeviceType == FILE_DEVICE_CD_ROM_FILE_SYSTEM) type = FS_ISO9660; - - if (label && label_len) get_filesystem_label( &nt_name, label, label_len ); - if (serial) *serial = get_filesystem_serial( &nt_name ); - -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_UDF: - if (fsname) lstrcpynW( fsname, udfW, fsname_len ); - if (filename_len) *filename_len = 255; - if (flags) - *flags = FILE_READ_ONLY_VOLUME | FILE_UNICODE_ON_DISK | FILE_CASE_SENSITIVE_SEARCH; - break; - case FS_FAT1216: - if (fsname) lstrcpynW( fsname, fatW, fsname_len ); - case FS_FAT32: - if (type == FS_FAT32 && fsname) lstrcpynW( fsname, fat32W, 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 | FILE_PERSISTENT_ACLS; - break; - } - ret = TRUE; - -done: - RtlFreeUnicodeString( &nt_name ); - return ret; -} - - -/*********************************************************************** - * 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 ); - } - - HeapFree( GetProcessHeap(), 0, labelW ); - HeapFree( GetProcessHeap(), 0, fsnameW ); - return ret; -} - - - /*********************************************************************** * SetVolumeLabelW (KERNEL32.@) */ @@ -968,190 +338,6 @@ BOOL WINAPI GetVolumeNameForVolumeMountPointA( LPCSTR path, LPSTR volume, DWORD return ret; } -/*********************************************************************** - * GetVolumeNameForVolumeMountPointW (KERNEL32.@) - */ -BOOL WINAPI GetVolumeNameForVolumeMountPointW( LPCWSTR path, LPWSTR volume, DWORD size ) -{ - static const WCHAR prefixW[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\',0}; - static const WCHAR volumeW[] = {'\\','?','?','\\','V','o','l','u','m','e','{'}; - static const WCHAR trailingW[] = {'\\',0}; - - MOUNTMGR_MOUNT_POINT *input = NULL, *o1; - MOUNTMGR_MOUNT_POINTS *output = NULL; - WCHAR *p; - char *r; - DWORD i, i_size = 1024, o_size = 1024; - WCHAR *nonpersist_name; - WCHAR symlink_name[MAX_PATH]; - NTSTATUS status; - HANDLE mgr = INVALID_HANDLE_VALUE; - BOOL ret = FALSE; - DWORD br; - - TRACE("(%s, %p, %x)\n", debugstr_w(path), volume, size); - if (path[lstrlenW(path)-1] != '\\') - { - SetLastError( ERROR_INVALID_NAME ); - return FALSE; - } - - if (size < 50) - { - SetLastError( ERROR_FILENAME_EXCED_RANGE ); - return FALSE; - } - /* if length of input is > 3 then it must be a mounted folder */ - if (lstrlenW(path) > 3) - { - FIXME("Mounted Folders are not yet supported\n"); - SetLastError( ERROR_NOT_A_REPARSE_POINT ); - return FALSE; - } - - mgr = CreateFileW( MOUNTMGR_DOS_DEVICE_NAME, 0, FILE_SHARE_READ, - NULL, OPEN_EXISTING, 0, 0 ); - if (mgr == INVALID_HANDLE_VALUE) return FALSE; - - if (!(input = HeapAlloc( GetProcessHeap(), 0, i_size ))) - { - SetLastError( ERROR_NOT_ENOUGH_MEMORY ); - goto err_ret; - } - - if (!(output = HeapAlloc( GetProcessHeap(), 0, o_size ))) - { - SetLastError( ERROR_NOT_ENOUGH_MEMORY ); - goto err_ret; - } - - /* construct the symlink name as "\DosDevices\C:" */ - lstrcpyW( symlink_name, prefixW ); - lstrcatW( symlink_name, path ); - symlink_name[lstrlenW(symlink_name)-1] = 0; - - /* Take the mount point and get the "nonpersistent name" */ - /* We will then take that and get the volume name */ - nonpersist_name = (WCHAR *)(input + 1); - status = read_nt_symlink( symlink_name, nonpersist_name, i_size - sizeof(*input) ); - TRACE("read_nt_symlink got stat=%x, for %s, got <%s>\n", status, - debugstr_w(symlink_name), debugstr_w(nonpersist_name)); - if (status != STATUS_SUCCESS) - { - SetLastError( ERROR_FILE_NOT_FOUND ); - goto err_ret; - } - - /* Now take the "nonpersistent name" and ask the mountmgr */ - /* to give us all the mount points. One of them will be */ - /* the volume name (format of \??\Volume{). */ - memset( input, 0, sizeof(*input) ); /* clear all input parameters */ - input->DeviceNameOffset = sizeof(*input); - input->DeviceNameLength = lstrlenW( nonpersist_name) * sizeof(WCHAR); - i_size = input->DeviceNameOffset + input->DeviceNameLength; - - output->Size = o_size; - - /* now get the true volume name from the mountmgr */ - if (!DeviceIoControl( mgr, IOCTL_MOUNTMGR_QUERY_POINTS, input, i_size, - output, o_size, &br, NULL )) - goto err_ret; - - /* Verify and return the data, note string is not null terminated */ - TRACE("found %d matching mount points\n", output->NumberOfMountPoints); - if (output->NumberOfMountPoints < 1) - { - SetLastError( ERROR_NO_VOLUME_ID ); - goto err_ret; - } - o1 = &output->MountPoints[0]; - - /* look for the volume name in returned values */ - for(i=0;iNumberOfMountPoints;i++) - { - p = (WCHAR*)((char *)output + o1->SymbolicLinkNameOffset); - r = (char *)output + o1->UniqueIdOffset; - TRACE("found symlink=%s, unique=%s, devname=%s\n", - debugstr_wn(p, o1->SymbolicLinkNameLength/sizeof(WCHAR)), - debugstr_an(r, o1->UniqueIdLength), - debugstr_wn((WCHAR*)((char *)output + o1->DeviceNameOffset), - o1->DeviceNameLength/sizeof(WCHAR))); - - if (!strncmpW( p, volumeW, ARRAY_SIZE( volumeW ))) - { - /* is there space in the return variable ?? */ - if ((o1->SymbolicLinkNameLength/sizeof(WCHAR))+2 > size) - { - SetLastError( ERROR_FILENAME_EXCED_RANGE ); - goto err_ret; - } - memcpy( volume, p, o1->SymbolicLinkNameLength ); - volume[o1->SymbolicLinkNameLength / sizeof(WCHAR)] = 0; - lstrcatW( volume, trailingW ); - /* change second char from '?' to '\' */ - volume[1] = '\\'; - ret = TRUE; - break; - } - o1++; - } - -err_ret: - HeapFree( GetProcessHeap(), 0, input ); - HeapFree( GetProcessHeap(), 0, output ); - CloseHandle( mgr ); - return ret; -} - -/*********************************************************************** - * DefineDosDeviceW (KERNEL32.@) - */ -BOOL WINAPI DefineDosDeviceW( DWORD flags, const WCHAR *device, const WCHAR *target ) -{ - WCHAR link_name[15] = {'\\','D','o','s','D','e','v','i','c','e','s','\\',0}; - UNICODE_STRING nt_name, nt_target; - OBJECT_ATTRIBUTES attr; - NTSTATUS status; - HANDLE handle; - - TRACE("%#x, %s, %s\n", flags, debugstr_w(device), debugstr_w(target)); - - if (flags & ~(DDD_RAW_TARGET_PATH | DDD_REMOVE_DEFINITION)) - FIXME("Ignoring flags %#x.\n", flags & ~(DDD_RAW_TARGET_PATH | DDD_REMOVE_DEFINITION)); - - strcatW( link_name, device ); - RtlInitUnicodeString( &nt_name, link_name ); - InitializeObjectAttributes( &attr, &nt_name, OBJ_CASE_INSENSITIVE, 0, NULL ); - if (flags & DDD_REMOVE_DEFINITION) - { - if (!set_ntstatus( NtOpenSymbolicLinkObject( &handle, 0, &attr ) )) - return FALSE; - - SERVER_START_REQ( unlink_object ) - { - req->handle = wine_server_obj_handle( handle ); - status = wine_server_call( req ); - } - SERVER_END_REQ; - NtClose( handle ); - - return set_ntstatus( status ); - } - - if (!(flags & DDD_RAW_TARGET_PATH)) - { - if (!RtlDosPathNameToNtPathName_U( target, &nt_target, NULL, NULL)) - { - SetLastError( ERROR_PATH_NOT_FOUND ); - return FALSE; - } - } - else - RtlInitUnicodeString( &nt_target, target ); - - return set_ntstatus( NtCreateSymbolicLinkObject( &handle, SYMBOLIC_LINK_ALL_ACCESS, &attr, &nt_target ) ); -} - /*********************************************************************** * DefineDosDeviceA (KERNEL32.@) @@ -1169,93 +355,6 @@ BOOL WINAPI DefineDosDeviceA(DWORD flags, LPCSTR devname, LPCSTR targetpath) } -/*********************************************************************** - * QueryDosDeviceW (KERNEL32.@) - * - * returns array of strings terminated by \0, terminated by \0 - */ -DWORD WINAPI QueryDosDeviceW( LPCWSTR devname, LPWSTR target, DWORD bufsize ) -{ - static const WCHAR dosdevW[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\',0}; - - UNICODE_STRING nt_name; - NTSTATUS status; - - if (!bufsize) - { - SetLastError( ERROR_INSUFFICIENT_BUFFER ); - return 0; - } - - if (devname) - { - WCHAR name[8]; - WCHAR *buffer; - DWORD dosdev, ret = 0; - - if ((dosdev = RtlIsDosDeviceName_U( devname ))) - { - memcpy( name, devname + HIWORD(dosdev)/sizeof(WCHAR), LOWORD(dosdev) ); - name[LOWORD(dosdev)/sizeof(WCHAR)] = 0; - devname = name; - } - - if (!(buffer = HeapAlloc( GetProcessHeap(), 0, sizeof(dosdevW) + strlenW(devname)*sizeof(WCHAR) ))) - { - SetLastError( ERROR_OUTOFMEMORY ); - return 0; - } - memcpy( buffer, dosdevW, sizeof(dosdevW) ); - strcatW( buffer, devname ); - status = read_nt_symlink( buffer, target, bufsize ); - HeapFree( GetProcessHeap(), 0, buffer ); - if (!set_ntstatus( status )) return 0; - ret = strlenW( target ) + 1; - if (ret < bufsize) target[ret++] = 0; /* add an extra null */ - return ret; - } - else /* return a list of all devices */ - { - OBJECT_ATTRIBUTES attr; - HANDLE handle; - WCHAR *p = target; - - RtlInitUnicodeString( &nt_name, dosdevW ); - nt_name.Length -= sizeof(WCHAR); /* without trailing slash */ - attr.Length = sizeof(attr); - attr.RootDirectory = 0; - attr.ObjectName = &nt_name; - attr.Attributes = OBJ_CASE_INSENSITIVE; - attr.SecurityDescriptor = NULL; - attr.SecurityQualityOfService = NULL; - status = NtOpenDirectoryObject( &handle, FILE_LIST_DIRECTORY, &attr ); - if (!status) - { - char data[1024]; - DIRECTORY_BASIC_INFORMATION *info = (DIRECTORY_BASIC_INFORMATION *)data; - ULONG ctx = 0, len; - - while (!NtQueryDirectoryObject( handle, info, sizeof(data), 1, 0, &ctx, &len )) - { - if (p + info->ObjectName.Length/sizeof(WCHAR) + 1 >= target + bufsize) - { - SetLastError( ERROR_INSUFFICIENT_BUFFER ); - NtClose( handle ); - return 0; - } - memcpy( p, info->ObjectName.Buffer, info->ObjectName.Length ); - p += info->ObjectName.Length/sizeof(WCHAR); - *p++ = 0; - } - NtClose( handle ); - } - - *p++ = 0; /* terminating null */ - return p - target; - } -} - - /*********************************************************************** * QueryDosDeviceA (KERNEL32.@) * @@ -1285,44 +384,6 @@ DWORD WINAPI QueryDosDeviceA( LPCSTR devname, LPSTR target, DWORD bufsize ) } -/*********************************************************************** - * GetLogicalDrives (KERNEL32.@) - */ -DWORD WINAPI GetLogicalDrives(void) -{ - static const WCHAR dosdevW[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\',0}; - OBJECT_ATTRIBUTES attr; - UNICODE_STRING nt_name; - DWORD bitmask = 0; - NTSTATUS status; - HANDLE handle; - - RtlInitUnicodeString( &nt_name, dosdevW ); - nt_name.Length -= sizeof(WCHAR); /* without trailing slash */ - attr.Length = sizeof(attr); - attr.RootDirectory = 0; - attr.ObjectName = &nt_name; - attr.Attributes = OBJ_CASE_INSENSITIVE; - attr.SecurityDescriptor = NULL; - attr.SecurityQualityOfService = NULL; - status = NtOpenDirectoryObject( &handle, FILE_LIST_DIRECTORY, &attr ); - if (!status) - { - char data[1024]; - DIRECTORY_BASIC_INFORMATION *info = (DIRECTORY_BASIC_INFORMATION *)data; - ULONG ctx = 0, len; - - while (!NtQueryDirectoryObject( handle, info, sizeof(data), 1, 0, &ctx, &len )) - if(info->ObjectName.Length == 2*sizeof(WCHAR) && info->ObjectName.Buffer[1] == ':') - bitmask |= 1 << (info->ObjectName.Buffer[0] - 'A'); - - NtClose( handle ); - } - - return bitmask; -} - - /*********************************************************************** * GetLogicalDriveStringsA (KERNEL32.@) */ @@ -1349,226 +410,6 @@ UINT WINAPI GetLogicalDriveStringsA( UINT len, LPSTR buffer ) } -/*********************************************************************** - * 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 cannot 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 )) - { - /* CD ROM devices do not necessarily have a volume, but a drive type */ - ret = get_mountmgr_drive_type( root ); - if (ret == DRIVE_CDROM || ret == DRIVE_REMOVABLE) - return ret; - - 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 - { - 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 if ((ret = get_mountmgr_drive_type( root )) == DRIVE_UNKNOWN) ret = DRIVE_FIXED; - break; - default: - ret = DRIVE_UNKNOWN; - break; - } - } - TRACE( "%s -> %d\n", debugstr_w(root), ret ); - return ret; -} - - -/*********************************************************************** - * GetDriveTypeA (KERNEL32.@) - * - * See GetDriveTypeW. - */ -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 (!set_ntstatus( 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.@) - * - * See GetDiskFreeSpaceExW. - */ -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 (!set_ntstatus( status )) return FALSE; - - units = info.SectorsPerAllocationUnit * info.BytesPerSector; - - if( GetVersion() & 0x80000000) { /* win3.x, 9x, ME */ - /* cap the size and available at 2GB as per specs */ - if (info.TotalAllocationUnits.QuadPart * units > 0x7fffffff) { - info.TotalAllocationUnits.QuadPart = 0x7fffffff / units; - if (info.AvailableAllocationUnits.QuadPart * units > 0x7fffffff) - info.AvailableAllocationUnits.QuadPart = 0x7fffffff / units; - } - /* nr. of clusters is always <= 65335 */ - while( info.TotalAllocationUnits.QuadPart > 65535 ) { - info.TotalAllocationUnits.QuadPart /= 2; - info.AvailableAllocationUnits.QuadPart /= 2; - info.SectorsPerAllocationUnit *= 2; - } - } - - 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; - TRACE("%#08x, %#08x, %#08x, %#08x\n", info.SectorsPerAllocationUnit, info.BytesPerSector, - info.AvailableAllocationUnits.u.LowPart, 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 ); -} - /*********************************************************************** * GetVolumePathNameA (KERNEL32.@) */ @@ -1591,173 +432,6 @@ BOOL WINAPI GetVolumePathNameA(LPCSTR filename, LPSTR volumepathname, DWORD bufl return ret; } -static BOOL is_dos_path( const UNICODE_STRING *path ) -{ - static const WCHAR global_prefix[4] = {'\\','?','?','\\'}; - return path->Length >= 7 * sizeof(WCHAR) - && !memcmp(path->Buffer, global_prefix, sizeof(global_prefix)) - && path->Buffer[5] == ':' && path->Buffer[6] == '\\'; -} - -/* resolve all symlinks in a path in-place; return FALSE if allocation failed */ -static BOOL resolve_symlink( UNICODE_STRING *path ) -{ - OBJECT_NAME_INFORMATION *info; - OBJECT_ATTRIBUTES attr; - IO_STATUS_BLOCK io; - NTSTATUS status; - HANDLE file; - ULONG size; - - InitializeObjectAttributes( &attr, path, OBJ_CASE_INSENSITIVE, 0, NULL ); - if (NtOpenFile( &file, SYNCHRONIZE, &attr, &io, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, - FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT )) - return TRUE; - - if (NtQueryObject( file, ObjectNameInformation, NULL, 0, &size ) != STATUS_INFO_LENGTH_MISMATCH) - { - NtClose( file ); - return TRUE; - } - - if (!(info = HeapAlloc( GetProcessHeap(), 0, size ))) - { - NtClose( file ); - return FALSE; - } - - status = NtQueryObject( file, ObjectNameInformation, info, size, NULL ); - NtClose( file ); - if (status) - return TRUE; - - RtlFreeUnicodeString( path ); - status = RtlDuplicateUnicodeString( 0, &info->Name, path ); - HeapFree( GetProcessHeap(), 0, info ); - return !status; -} - -/*********************************************************************** - * GetVolumePathNameW (KERNEL32.@) - */ -BOOL WINAPI GetVolumePathNameW(const WCHAR *path, WCHAR *volume_path, DWORD length) -{ - static const WCHAR device_prefix[4] = {'\\','\\','.','\\'}; - static const WCHAR device_prefix2[4] = {'\\','\\','?','\\'}; - static const WCHAR global_prefix[4] = {'\\','?','?','\\'}; - FILE_ATTRIBUTE_TAG_INFORMATION attr_info; - FILE_BASIC_INFORMATION basic_info; - OBJECT_ATTRIBUTES attr; - UNICODE_STRING nt_name; - NTSTATUS status; - - if (path && !memcmp(path, global_prefix, sizeof(global_prefix))) - { - WCHAR current_drive[MAX_PATH]; - - GetCurrentDirectoryW( ARRAY_SIZE(current_drive), current_drive ); - if (length >= 3) - { - WCHAR ret_path[4] = {current_drive[0], ':', '\\', 0}; - lstrcpynW( volume_path, ret_path, length ); - return TRUE; - } - - SetLastError( ERROR_FILENAME_EXCED_RANGE ); - return FALSE; - } - - if (!volume_path || !length || !RtlDosPathNameToNtPathName_U( path, &nt_name, NULL, NULL )) - { - SetLastError( ERROR_INVALID_PARAMETER ); - return FALSE; - } - - if (!is_dos_path( &nt_name )) - { - RtlFreeUnicodeString( &nt_name ); - WARN("invalid path %s\n", debugstr_w(path)); - SetLastError( ERROR_INVALID_NAME ); - return FALSE; - } - - InitializeObjectAttributes( &attr, &nt_name, OBJ_CASE_INSENSITIVE, 0, NULL ); - - while (nt_name.Length > 7 * sizeof(WCHAR)) - { - IO_STATUS_BLOCK io; - HANDLE file; - - if (!NtQueryAttributesFile( &attr, &basic_info ) - && (basic_info.FileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) - && (basic_info.FileAttributes & FILE_ATTRIBUTE_DIRECTORY) - && !NtOpenFile( &file, SYNCHRONIZE | FILE_READ_ATTRIBUTES, &attr, &io, - FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, - FILE_OPEN_REPARSE_POINT | FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT )) - { - status = NtQueryInformationFile( file, &io, &attr_info, - sizeof(attr_info), FileAttributeTagInformation ); - NtClose( file ); - if (!status) - { - - if (attr_info.ReparseTag == IO_REPARSE_TAG_MOUNT_POINT) - break; - - if (!resolve_symlink( &nt_name )) - { - SetLastError( ERROR_OUTOFMEMORY ); - return FALSE; - } - } - } - - if (nt_name.Buffer[(nt_name.Length / sizeof(WCHAR)) - 1] == '\\') - nt_name.Length -= sizeof(WCHAR); - while (nt_name.Length && nt_name.Buffer[(nt_name.Length / sizeof(WCHAR)) - 1] != '\\') - nt_name.Length -= sizeof(WCHAR); - } - - nt_name.Buffer[nt_name.Length / sizeof(WCHAR)] = 0; - - if (NtQueryAttributesFile( &attr, &basic_info )) - { - RtlFreeUnicodeString( &nt_name ); - WARN("nonexistent path %s -> %s\n", debugstr_w(path), debugstr_w( nt_name.Buffer )); - SetLastError( ERROR_FILE_NOT_FOUND ); - return FALSE; - } - - if (!memcmp(path, device_prefix, sizeof(device_prefix)) - || !memcmp(path, device_prefix2, sizeof(device_prefix2))) - { - if (length >= nt_name.Length / sizeof(WCHAR)) - { - memcpy(volume_path, path, 4 * sizeof(WCHAR)); - lstrcpynW( volume_path + 4, nt_name.Buffer + 4, length - 4 ); - - TRACE("%s -> %s\n", debugstr_w(path), debugstr_w(volume_path)); - - RtlFreeUnicodeString( &nt_name ); - return TRUE; - } - } - else if (length >= (nt_name.Length / sizeof(WCHAR)) - 4) - { - lstrcpynW( volume_path, nt_name.Buffer + 4, length ); - volume_path[0] = toupperW(volume_path[0]); - - TRACE("%s -> %s\n", debugstr_w(path), debugstr_w(volume_path)); - - RtlFreeUnicodeString( &nt_name ); - return TRUE; - } - - RtlFreeUnicodeString( &nt_name ); - SetLastError( ERROR_FILENAME_EXCED_RANGE ); - return FALSE; -} - /*********************************************************************** * GetVolumePathNamesForVolumeNameA (KERNEL32.@) @@ -1780,7 +454,7 @@ BOOL WINAPI GetVolumePathNamesForVolumeNameA(LPCSTR volumename, LPSTR volumepath while (*pathW) { - int len = strlenW( pathW ) + 1; + int len = lstrlenW( pathW ) + 1; FILE_name_WtoA( pathW, len, path, buflen ); buflen -= len; pathW += len; @@ -1793,139 +467,6 @@ BOOL WINAPI GetVolumePathNamesForVolumeNameA(LPCSTR volumename, LPSTR volumepath return ret; } -static MOUNTMGR_MOUNT_POINTS *query_mount_points( HANDLE mgr, MOUNTMGR_MOUNT_POINT *input, DWORD insize ) -{ - MOUNTMGR_MOUNT_POINTS *output; - DWORD outsize = 1024; - DWORD br; - - for (;;) - { - if (!(output = HeapAlloc( GetProcessHeap(), 0, outsize ))) - { - SetLastError( ERROR_NOT_ENOUGH_MEMORY ); - return NULL; - } - if (DeviceIoControl( mgr, IOCTL_MOUNTMGR_QUERY_POINTS, input, insize, output, outsize, &br, NULL )) break; - outsize = output->Size; - HeapFree( GetProcessHeap(), 0, output ); - if (GetLastError() != ERROR_MORE_DATA) return NULL; - } - return output; -} -/*********************************************************************** - * GetVolumePathNamesForVolumeNameW (KERNEL32.@) - */ -BOOL WINAPI GetVolumePathNamesForVolumeNameW(LPCWSTR volumename, LPWSTR volumepathname, DWORD buflen, PDWORD returnlen) -{ - static const WCHAR dosdevicesW[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\'}; - HANDLE mgr; - DWORD len, size; - MOUNTMGR_MOUNT_POINT *spec; - MOUNTMGR_MOUNT_POINTS *link, *target = NULL; - WCHAR *name, *path; - BOOL ret = FALSE; - UINT i, j; - - TRACE("%s, %p, %u, %p\n", debugstr_w(volumename), volumepathname, buflen, returnlen); - - if (!volumename || (len = strlenW( volumename )) != 49) - { - SetLastError( ERROR_INVALID_NAME ); - return FALSE; - } - mgr = CreateFileW( MOUNTMGR_DOS_DEVICE_NAME, 0, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, 0 ); - if (mgr == INVALID_HANDLE_VALUE) return FALSE; - - size = sizeof(*spec) + sizeof(WCHAR) * (len - 1); /* remove trailing backslash */ - if (!(spec = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, size ))) goto done; - spec->SymbolicLinkNameOffset = sizeof(*spec); - spec->SymbolicLinkNameLength = size - sizeof(*spec); - name = (WCHAR *)((char *)spec + spec->SymbolicLinkNameOffset); - memcpy( name, volumename, size - sizeof(*spec) ); - name[1] = '?'; /* map \\?\ to \??\ */ - - target = query_mount_points( mgr, spec, size ); - HeapFree( GetProcessHeap(), 0, spec ); - if (!target) - { - goto done; - } - if (!target->NumberOfMountPoints) - { - SetLastError( ERROR_FILE_NOT_FOUND ); - goto done; - } - len = 0; - path = volumepathname; - for (i = 0; i < target->NumberOfMountPoints; i++) - { - link = NULL; - if (target->MountPoints[i].DeviceNameOffset) - { - const WCHAR *device = (const WCHAR *)((const char *)target + target->MountPoints[i].DeviceNameOffset); - USHORT device_len = target->MountPoints[i].DeviceNameLength; - - size = sizeof(*spec) + device_len; - if (!(spec = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, size ))) goto done; - spec->DeviceNameOffset = sizeof(*spec); - spec->DeviceNameLength = device_len; - memcpy( (char *)spec + spec->DeviceNameOffset, device, device_len ); - - link = query_mount_points( mgr, spec, size ); - HeapFree( GetProcessHeap(), 0, spec ); - } - else if (target->MountPoints[i].UniqueIdOffset) - { - const WCHAR *id = (const WCHAR *)((const char *)target + target->MountPoints[i].UniqueIdOffset); - USHORT id_len = target->MountPoints[i].UniqueIdLength; - - size = sizeof(*spec) + id_len; - if (!(spec = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, size ))) goto done; - spec->UniqueIdOffset = sizeof(*spec); - spec->UniqueIdLength = id_len; - memcpy( (char *)spec + spec->UniqueIdOffset, id, id_len ); - - link = query_mount_points( mgr, spec, size ); - HeapFree( GetProcessHeap(), 0, spec ); - } - if (!link) continue; - for (j = 0; j < link->NumberOfMountPoints; j++) - { - const WCHAR *linkname; - - if (!link->MountPoints[j].SymbolicLinkNameOffset) continue; - linkname = (const WCHAR *)((const char *)link + link->MountPoints[j].SymbolicLinkNameOffset); - - if (link->MountPoints[j].SymbolicLinkNameLength == sizeof(dosdevicesW) + 2 * sizeof(WCHAR) && - !strncmpiW( linkname, dosdevicesW, ARRAY_SIZE( dosdevicesW ))) - { - len += 4; - if (volumepathname && len < buflen) - { - path[0] = linkname[ARRAY_SIZE( dosdevicesW )]; - path[1] = ':'; - path[2] = '\\'; - path[3] = 0; - path += 4; - } - } - } - HeapFree( GetProcessHeap(), 0, link ); - } - if (buflen <= len) SetLastError( ERROR_MORE_DATA ); - else if (volumepathname) - { - volumepathname[len] = 0; - ret = TRUE; - } - if (returnlen) *returnlen = len + 1; - -done: - HeapFree( GetProcessHeap(), 0, target ); - CloseHandle( mgr ); - return ret; -} /*********************************************************************** * FindFirstVolumeA (KERNEL32.@) @@ -1947,51 +488,6 @@ HANDLE WINAPI FindFirstVolumeA(LPSTR volume, DWORD len) return handle; } -/*********************************************************************** - * FindFirstVolumeW (KERNEL32.@) - */ -HANDLE WINAPI FindFirstVolumeW( LPWSTR volume, DWORD len ) -{ - DWORD size = 1024; - DWORD br; - HANDLE mgr = CreateFileW( MOUNTMGR_DOS_DEVICE_NAME, 0, FILE_SHARE_READ|FILE_SHARE_WRITE, - NULL, OPEN_EXISTING, 0, 0 ); - if (mgr == INVALID_HANDLE_VALUE) return INVALID_HANDLE_VALUE; - - for (;;) - { - MOUNTMGR_MOUNT_POINT input; - MOUNTMGR_MOUNT_POINTS *output; - - if (!(output = HeapAlloc( GetProcessHeap(), 0, size ))) - { - SetLastError( ERROR_NOT_ENOUGH_MEMORY ); - break; - } - memset( &input, 0, sizeof(input) ); - - if (!DeviceIoControl( mgr, IOCTL_MOUNTMGR_QUERY_POINTS, &input, sizeof(input), - output, size, &br, NULL )) - { - if (GetLastError() != ERROR_MORE_DATA) break; - size = output->Size; - HeapFree( GetProcessHeap(), 0, output ); - continue; - } - CloseHandle( mgr ); - /* abuse the Size field to store the current index */ - output->Size = 0; - if (!FindNextVolumeW( output, volume, len )) - { - HeapFree( GetProcessHeap(), 0, output ); - return INVALID_HANDLE_VALUE; - } - return output; - } - CloseHandle( mgr ); - return INVALID_HANDLE_VALUE; -} - /*********************************************************************** * FindNextVolumeA (KERNEL32.@) */ @@ -2008,45 +504,6 @@ BOOL WINAPI FindNextVolumeA( HANDLE handle, LPSTR volume, DWORD len ) return ret; } -/*********************************************************************** - * FindNextVolumeW (KERNEL32.@) - */ -BOOL WINAPI FindNextVolumeW( HANDLE handle, LPWSTR volume, DWORD len ) -{ - MOUNTMGR_MOUNT_POINTS *data = handle; - - while (data->Size < data->NumberOfMountPoints) - { - static const WCHAR volumeW[] = {'\\','?','?','\\','V','o','l','u','m','e','{',}; - WCHAR *link = (WCHAR *)((char *)data + data->MountPoints[data->Size].SymbolicLinkNameOffset); - DWORD size = data->MountPoints[data->Size].SymbolicLinkNameLength; - data->Size++; - /* skip non-volumes */ - if (size < sizeof(volumeW) || memcmp( link, volumeW, sizeof(volumeW) )) continue; - if (size + sizeof(WCHAR) >= len * sizeof(WCHAR)) - { - SetLastError( ERROR_FILENAME_EXCED_RANGE ); - return FALSE; - } - memcpy( volume, link, size ); - volume[1] = '\\'; /* map \??\ to \\?\ */ - volume[size / sizeof(WCHAR)] = '\\'; /* Windows appends a backslash */ - volume[size / sizeof(WCHAR) + 1] = 0; - TRACE( "returning entry %u %s\n", data->Size - 1, debugstr_w(volume) ); - return TRUE; - } - SetLastError( ERROR_NO_MORE_FILES ); - return FALSE; -} - -/*********************************************************************** - * FindVolumeClose (KERNEL32.@) - */ -BOOL WINAPI FindVolumeClose(HANDLE handle) -{ - return HeapFree( GetProcessHeap(), 0, handle ); -} - /*********************************************************************** * FindFirstVolumeMountPointA (KERNEL32.@) */ @@ -2085,15 +542,6 @@ BOOL WINAPI DeleteVolumeMountPointA(LPCSTR mountpoint) return FALSE; } -/*********************************************************************** - * DeleteVolumeMountPointW (KERNEL32.@) - */ -BOOL WINAPI DeleteVolumeMountPointW(LPCWSTR mountpoint) -{ - FIXME("(%s), stub!\n", debugstr_w(mountpoint)); - return FALSE; -} - /*********************************************************************** * SetVolumeMountPointA (KERNEL32.@) */ @@ -2113,62 +561,3 @@ BOOL WINAPI SetVolumeMountPointW(LPCWSTR path, LPCWSTR volume) SetLastError(ERROR_CALL_NOT_IMPLEMENTED); return FALSE; } - -/*********************************************************************** - * GetVolumeInformationByHandleW (KERNEL32.@) - */ -BOOL WINAPI GetVolumeInformationByHandleW( HANDLE handle, WCHAR *label, DWORD label_len, - DWORD *serial, DWORD *filename_len, DWORD *flags, - WCHAR *fsname, DWORD fsname_len ) -{ - IO_STATUS_BLOCK io; - - TRACE( "%p\n", handle ); - - if (label || serial) - { - char buffer[sizeof(FILE_FS_VOLUME_INFORMATION) + MAX_PATH * sizeof(WCHAR)]; - FILE_FS_VOLUME_INFORMATION *info = (FILE_FS_VOLUME_INFORMATION *)buffer; - - if (!set_ntstatus( NtQueryVolumeInformationFile( handle, &io, info, sizeof(buffer), - FileFsVolumeInformation ) )) - return FALSE; - - if (label) - { - if (label_len < info->VolumeLabelLength / sizeof(WCHAR) + 1) - { - SetLastError( ERROR_BAD_LENGTH ); - return FALSE; - } - memcpy( label, info->VolumeLabel, info->VolumeLabelLength ); - label[info->VolumeLabelLength / sizeof(WCHAR)] = 0; - } - if (serial) *serial = info->VolumeSerialNumber; - } - - if (filename_len || flags || fsname) - { - char buffer[sizeof(FILE_FS_ATTRIBUTE_INFORMATION) + MAX_PATH * sizeof(WCHAR)]; - FILE_FS_ATTRIBUTE_INFORMATION *info = (FILE_FS_ATTRIBUTE_INFORMATION *)buffer; - - if (!set_ntstatus( NtQueryVolumeInformationFile( handle, &io, info, sizeof(buffer), - FileFsAttributeInformation ) )) - return FALSE; - - if (fsname) - { - if (fsname_len < info->FileSystemNameLength / sizeof(WCHAR) + 1) - { - SetLastError( ERROR_BAD_LENGTH ); - return FALSE; - } - memcpy( fsname, info->FileSystemName, info->FileSystemNameLength ); - fsname[info->FileSystemNameLength / sizeof(WCHAR)] = 0; - } - if (filename_len) *filename_len = info->MaximumComponentNameLength; - if (flags) *flags = info->FileSystemAttributes; - } - - return TRUE; -} diff --git a/dlls/kernelbase/Makefile.in b/dlls/kernelbase/Makefile.in index de2cb79ae53..14953246f24 100644 --- a/dlls/kernelbase/Makefile.in +++ b/dlls/kernelbase/Makefile.in @@ -20,6 +20,7 @@ C_SRCS = \ string.c \ sync.c \ thread.c \ - version.c + version.c \ + volume.c RC_SRCS = kernelbase.rc diff --git a/dlls/kernelbase/kernelbase.spec b/dlls/kernelbase/kernelbase.spec index fe687c132ce..a3a208e8fe2 100644 --- a/dlls/kernelbase/kernelbase.spec +++ b/dlls/kernelbase/kernelbase.spec @@ -243,7 +243,7 @@ @ stdcall DecodePointer(ptr) ntdll.RtlDecodePointer # @ stub DecodeRemotePointer @ stdcall DecodeSystemPointer(ptr) ntdll.RtlDecodeSystemPointer -@ stdcall DefineDosDeviceW(long wstr wstr) kernel32.DefineDosDeviceW +@ stdcall DefineDosDeviceW(long wstr wstr) @ stdcall DelayLoadFailureHook(str str) # @ stub DelayLoadFailureHookLookup @ stdcall DeleteAce(ptr long) @@ -259,7 +259,7 @@ # @ stub DeleteSynchronizationBarrier @ stdcall DeleteTimerQueueEx(long long) @ stdcall DeleteTimerQueueTimer(long long long) -@ stdcall DeleteVolumeMountPointW(wstr) kernel32.DeleteVolumeMountPointW +@ stdcall DeleteVolumeMountPointW(wstr) @ stdcall DestroyPrivateObjectSecurity(ptr) @ stdcall DeviceIoControl(long long ptr long ptr long ptr ptr) @ stdcall DisablePredefinedHandleTableInternal(long) @@ -361,7 +361,7 @@ @ stdcall FindFirstFileW(wstr ptr) @ stdcall FindFirstFreeAce(ptr ptr) @ stdcall FindFirstStreamW(wstr long ptr long) kernel32.FindFirstStreamW -@ stdcall FindFirstVolumeW(ptr long) kernel32.FindFirstVolumeW +@ stdcall FindFirstVolumeW(ptr long) @ stdcall FindNLSString(long long wstr long wstr long ptr) @ stdcall FindNLSStringEx(wstr long wstr long wstr long ptr ptr ptr long) @ stdcall FindNextChangeNotification(long) @@ -369,12 +369,12 @@ # @ stub FindNextFileNameW @ stdcall FindNextFileW(long ptr) @ stdcall FindNextStreamW(long ptr) kernel32.FindNextStreamW -@ stdcall FindNextVolumeW(long ptr long) kernel32.FindNextVolumeW +@ stdcall FindNextVolumeW(long ptr long) # @ stub FindPackagesByPackageFamily @ stdcall FindResourceExW(long wstr wstr long) @ stdcall FindResourceW(long wstr wstr) @ stdcall FindStringOrdinal(long wstr long wstr long long) -@ stdcall FindVolumeClose(ptr) kernel32.FindVolumeClose +@ stdcall FindVolumeClose(ptr) @ stdcall FlsAlloc(ptr) @ stdcall FlsFree(long) @ stdcall FlsGetValue(long) @@ -479,12 +479,12 @@ # @ stub GetDeviceDriverBaseNameW # @ stub GetDeviceDriverFileNameA # @ stub GetDeviceDriverFileNameW -@ stdcall GetDiskFreeSpaceA(str ptr ptr ptr ptr) kernel32.GetDiskFreeSpaceA -@ stdcall GetDiskFreeSpaceExA(str ptr ptr ptr) kernel32.GetDiskFreeSpaceExA -@ stdcall GetDiskFreeSpaceExW(wstr ptr ptr ptr) kernel32.GetDiskFreeSpaceExW -@ stdcall GetDiskFreeSpaceW(wstr ptr ptr ptr ptr) kernel32.GetDiskFreeSpaceW -@ stdcall GetDriveTypeA(str) kernel32.GetDriveTypeA -@ stdcall GetDriveTypeW(wstr) kernel32.GetDriveTypeW +@ stdcall GetDiskFreeSpaceA(str ptr ptr ptr ptr) +@ stdcall GetDiskFreeSpaceExA(str ptr ptr ptr) +@ stdcall GetDiskFreeSpaceExW(wstr ptr ptr ptr) +@ stdcall GetDiskFreeSpaceW(wstr ptr ptr ptr ptr) +@ stdcall GetDriveTypeA(str) +@ stdcall GetDriveTypeW(wstr) # @ stub GetDurationFormatEx @ stdcall GetDynamicTimeZoneInformation(ptr) @ stdcall GetDynamicTimeZoneInformationEffectiveYears(ptr ptr ptr) @@ -545,8 +545,8 @@ @ stdcall GetLocaleInfoEx(wstr long ptr long) @ stub GetLocaleInfoHelper @ stdcall GetLocaleInfoW(long long ptr long) -@ stdcall GetLogicalDriveStringsW(long ptr) kernel32.GetLogicalDriveStringsW -@ stdcall GetLogicalDrives() kernel32.GetLogicalDrives +@ stdcall GetLogicalDriveStringsW(long ptr) +@ stdcall GetLogicalDrives() @ stdcall GetLogicalProcessorInformation(ptr ptr) @ stdcall GetLogicalProcessorInformationEx(long ptr ptr) @ stdcall GetLongPathNameA(str ptr long) @@ -752,12 +752,12 @@ @ stdcall GetVersion() @ stdcall GetVersionExA(ptr) @ stdcall GetVersionExW(ptr) -@ stdcall GetVolumeInformationA(str ptr long ptr ptr ptr ptr long) kernel32.GetVolumeInformationA -@ stdcall GetVolumeInformationByHandleW(ptr ptr long ptr ptr ptr ptr long) kernel32.GetVolumeInformationByHandleW -@ stdcall GetVolumeInformationW(wstr ptr long ptr ptr ptr ptr long) kernel32.GetVolumeInformationW -@ stdcall GetVolumeNameForVolumeMountPointW(wstr ptr long) kernel32.GetVolumeNameForVolumeMountPointW -@ stdcall GetVolumePathNameW(wstr ptr long) kernel32.GetVolumePathNameW -@ stdcall GetVolumePathNamesForVolumeNameW(wstr ptr long ptr) kernel32.GetVolumePathNamesForVolumeNameW +@ stdcall GetVolumeInformationA(str ptr long ptr ptr ptr ptr long) +@ stdcall GetVolumeInformationByHandleW(ptr ptr long ptr ptr ptr ptr long) +@ stdcall GetVolumeInformationW(wstr ptr long ptr ptr ptr ptr long) +@ stdcall GetVolumeNameForVolumeMountPointW(wstr ptr long) +@ stdcall GetVolumePathNameW(wstr ptr long) +@ stdcall GetVolumePathNamesForVolumeNameW(wstr ptr long ptr) @ stdcall GetWindowsAccountDomainSid(ptr ptr ptr) @ stdcall GetWindowsDirectoryA(ptr long) @ stdcall GetWindowsDirectoryW(ptr long) @@ -1194,7 +1194,7 @@ @ stdcall QueryActCtxSettingsW(long ptr wstr wstr ptr long ptr) @ stdcall QueryActCtxW(long ptr ptr long ptr long ptr) @ stdcall QueryDepthSList(ptr) ntdll.RtlQueryDepthSList -@ stdcall QueryDosDeviceW(wstr ptr long) kernel32.QueryDosDeviceW +@ stdcall QueryDosDeviceW(wstr ptr long) @ stdcall QueryFullProcessImageNameA(ptr long ptr ptr) kernel32.QueryFullProcessImageNameA @ stdcall QueryFullProcessImageNameW(ptr long ptr ptr) kernel32.QueryFullProcessImageNameW # @ stub QueryIdleProcessorCycleTime diff --git a/dlls/kernelbase/volume.c b/dlls/kernelbase/volume.c new file mode 100644 index 00000000000..9939ea3bef4 --- /dev/null +++ b/dlls/kernelbase/volume.c @@ -0,0 +1,1777 @@ +/* + * 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA + */ + +#include +#include +#include + +#include "ntstatus.h" +#define WIN32_NO_STATUS +#include "windef.h" +#include "winbase.h" +#include "winnls.h" +#include "winternl.h" +#include "winioctl.h" +#include "ntddcdrm.h" +#define WINE_MOUNTMGR_EXTENSIONS +#include "ddk/mountmgr.h" +#include "ddk/wdm.h" +#include "kernelbase.h" +#include "wine/server.h" +#include "wine/debug.h" + +WINE_DEFAULT_DEBUG_CHANNEL(volume); + +#define BLOCK_SIZE 2048 +#define SUPERBLOCK_SIZE BLOCK_SIZE +#define SYMBOLIC_LINK_QUERY 0x0001 + +#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, + FS_UDF /* For reference [E] = Ecma-167.pdf, [U] = udf260.pdf */ +}; + +/* read the contents of an NT symlink object */ +static NTSTATUS read_nt_symlink( const WCHAR *name, WCHAR *target, DWORD size ) +{ + NTSTATUS status; + OBJECT_ATTRIBUTES attr; + UNICODE_STRING nameW; + HANDLE handle; + + attr.Length = sizeof(attr); + attr.RootDirectory = 0; + attr.Attributes = OBJ_CASE_INSENSITIVE; + attr.ObjectName = &nameW; + attr.SecurityDescriptor = NULL; + attr.SecurityQualityOfService = NULL; + RtlInitUnicodeString( &nameW, name ); + + if (!(status = NtOpenSymbolicLinkObject( &handle, SYMBOLIC_LINK_QUERY, &attr ))) + { + UNICODE_STRING targetW; + targetW.Buffer = target; + targetW.MaximumLength = (size - 1) * sizeof(WCHAR); + status = NtQuerySymbolicLinkObject( handle, &targetW, NULL ); + if (!status) target[targetW.Length / sizeof(WCHAR)] = 0; + NtClose( handle ); + } + return status; +} + +/* open a handle to a device root */ +static BOOL open_device_root( LPCWSTR root, HANDLE *handle ) +{ + UNICODE_STRING nt_name; + OBJECT_ATTRIBUTES attr; + IO_STATUS_BLOCK io; + NTSTATUS status; + + if (!root) root = L"\\"; + 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, SYNCHRONIZE, &attr, &io, 0, + FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT ); + RtlFreeUnicodeString( &nt_name ); + return set_ntstatus( status ); +} + +/* query the type of a drive from the mount manager */ +static DWORD get_mountmgr_drive_type( LPCWSTR root ) +{ + HANDLE mgr; + struct mountmgr_unix_drive data; + DWORD br; + + memset( &data, 0, sizeof(data) ); + if (root) data.letter = root[0]; + else + { + WCHAR curdir[MAX_PATH]; + GetCurrentDirectoryW( MAX_PATH, curdir ); + if (curdir[1] != ':' || curdir[2] != '\\') return DRIVE_UNKNOWN; + data.letter = curdir[0]; + } + + mgr = CreateFileW( MOUNTMGR_DOS_DEVICE_NAME, GENERIC_READ, + FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, 0 ); + if (mgr == INVALID_HANDLE_VALUE) return DRIVE_UNKNOWN; + + if (!DeviceIoControl( mgr, IOCTL_MOUNTMGR_QUERY_UNIX_DRIVE, &data, sizeof(data), &data, + sizeof(data), &br, NULL ) && GetLastError() != ERROR_MORE_DATA) + data.type = DRIVE_UNKNOWN; + + CloseHandle( mgr ); + return data.type; +} + +/* get the label by reading it from a file at the root of the filesystem */ +static void get_filesystem_label( const UNICODE_STRING *device, WCHAR *label, DWORD len ) +{ + static const WCHAR labelW[] = {'.','w','i','n','d','o','w','s','-','l','a','b','e','l',0}; + HANDLE handle; + UNICODE_STRING name; + IO_STATUS_BLOCK io; + OBJECT_ATTRIBUTES attr; + + label[0] = 0; + + attr.Length = sizeof(attr); + attr.RootDirectory = 0; + attr.Attributes = OBJ_CASE_INSENSITIVE; + attr.ObjectName = &name; + attr.SecurityDescriptor = NULL; + attr.SecurityQualityOfService = NULL; + + name.MaximumLength = device->Length + sizeof(labelW); + name.Length = name.MaximumLength - sizeof(WCHAR); + if (!(name.Buffer = HeapAlloc( GetProcessHeap(), 0, name.MaximumLength ))) return; + + memcpy( name.Buffer, device->Buffer, device->Length ); + memcpy( name.Buffer + device->Length / sizeof(WCHAR), labelW, sizeof(labelW) ); + if (!NtOpenFile( &handle, GENERIC_READ | SYNCHRONIZE, &attr, &io, FILE_SHARE_READ|FILE_SHARE_WRITE, + FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT )) + { + 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, len )) + label[len-1] = 0; + } + RtlFreeUnicodeString( &name ); +} + +/* get the serial number by reading it from a file at the root of the filesystem */ +static DWORD get_filesystem_serial( const UNICODE_STRING *device ) +{ + static const WCHAR serialW[] = {'.','w','i','n','d','o','w','s','-','s','e','r','i','a','l',0}; + HANDLE handle; + UNICODE_STRING name; + IO_STATUS_BLOCK io; + OBJECT_ATTRIBUTES attr; + DWORD ret = 0; + + attr.Length = sizeof(attr); + attr.RootDirectory = 0; + attr.Attributes = OBJ_CASE_INSENSITIVE; + attr.ObjectName = &name; + attr.SecurityDescriptor = NULL; + attr.SecurityQualityOfService = NULL; + + name.MaximumLength = device->Length + sizeof(serialW); + name.Length = name.MaximumLength - sizeof(WCHAR); + if (!(name.Buffer = HeapAlloc( GetProcessHeap(), 0, name.MaximumLength ))) return 0; + + memcpy( name.Buffer, device->Buffer, device->Length ); + memcpy( name.Buffer + device->Length / sizeof(WCHAR), serialW, sizeof(serialW) ); + if (!NtOpenFile( &handle, GENERIC_READ | SYNCHRONIZE, &attr, &io, FILE_SHARE_READ|FILE_SHARE_WRITE, + FILE_SYNCHRONOUS_IO_NONALERT )) + { + char buffer[32]; + DWORD size; + + if (!ReadFile( handle, buffer, sizeof(buffer)-1, &size, NULL )) size = 0; + CloseHandle( handle ); + buffer[size] = 0; + ret = strtoul( buffer, NULL, 16 ); + } + RtlFreeUnicodeString( &name ); + return ret; +} + + +/****************************************************************** + * find_cdrom_best_voldesc + */ +static DWORD find_cdrom_best_voldesc( HANDLE handle ) +{ + BYTE cur_vd_type, max_vd_type = 0; + BYTE buffer[0x800]; + 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; +} + + +/*********************************************************************** + * read_fat_superblock + */ +static enum fs_type read_fat_superblock( 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 )) + { + if (GetLastError() == ERROR_BAD_DEV_TYPE) return FS_UNKNOWN; /* not a real device */ + return FS_ERROR; + } + + if (size < SUPERBLOCK_SIZE) return FS_UNKNOWN; + + /* FIXME: do really all FAT have their name beginning with + * "FAT" ? (At least FAT12, FAT16 and FAT32 have :) + */ + if (!memcmp(buff+0x36, "FAT", 3) || !memcmp(buff+0x52, "FAT", 3)) + { + /* guess which type of FAT we have */ + int reasonable; + unsigned int sectors, + sect_per_fat, + total_sectors, + num_boot_sectors, + num_fats, + num_root_dir_ents, + bytes_per_sector, + sectors_per_cluster, + nclust; + sect_per_fat = GETWORD(buff, 0x16); + if (!sect_per_fat) sect_per_fat = GETLONG(buff, 0x24); + total_sectors = GETWORD(buff, 0x13); + if (!total_sectors) + total_sectors = GETLONG(buff, 0x20); + num_boot_sectors = GETWORD(buff, 0x0e); + num_fats = buff[0x10]; + num_root_dir_ents = GETWORD(buff, 0x11); + bytes_per_sector = GETWORD(buff, 0x0b); + sectors_per_cluster = buff[0x0d]; + /* check if the parameters are reasonable and will not cause + * arithmetic errors in the calculation */ + reasonable = num_boot_sectors < total_sectors && + num_fats < 16 && + bytes_per_sector >= 512 && bytes_per_sector % 512 == 0 && + sectors_per_cluster >= 1; + if (!reasonable) return FS_UNKNOWN; + sectors = total_sectors - num_boot_sectors - num_fats * sect_per_fat - + (num_root_dir_ents * 32 + bytes_per_sector - 1) / bytes_per_sector; + nclust = sectors / sectors_per_cluster; + if ((buff[0x42] == 0x28 || buff[0x42] == 0x29) && + !memcmp(buff+0x52, "FAT", 3)) return FS_FAT32; + if (nclust < 65525) + { + if ((buff[0x26] == 0x28 || buff[0x26] == 0x29) && + !memcmp(buff+0x36, "FAT", 3)) + return FS_FAT1216; + } + } + return FS_UNKNOWN; +} + + +/*********************************************************************** + * read_cd_block + */ +static BOOL read_cd_block( HANDLE handle, BYTE *buff, INT offs ) +{ + DWORD size, whence = offs >= 0 ? FILE_BEGIN : FILE_END; + + if (SetFilePointer( handle, offs, NULL, whence ) != offs || + !ReadFile( handle, buff, SUPERBLOCK_SIZE, &size, NULL ) || + size != SUPERBLOCK_SIZE) + return FALSE; + + return TRUE; +} + + +/*********************************************************************** + * read_cd_superblock + */ +static enum fs_type read_cd_superblock( HANDLE handle, BYTE *buff ) +{ + int i; + DWORD offs; + + /* Check UDF first as UDF and ISO9660 structures can coexist on the same medium + * Starting from sector 16, we may find : + * - a CD-ROM Volume Descriptor Set (ISO9660) containing one or more Volume Descriptors + * - an Extended Area (UDF) -- [E] 2/8.3.1 and [U] 2.1.7 + * There is no explicit end so read 16 sectors and then give up */ + for( i=16; i<16+16; i++) + { + if (!read_cd_block(handle, buff, i*BLOCK_SIZE)) + continue; + + /* We are supposed to check "BEA01", "NSR0x" and "TEA01" IDs + verify tag checksum + * but we assume the volume is well-formatted */ + if (!memcmp(&buff[1], "BEA01", 5)) return FS_UDF; + } + + offs = find_cdrom_best_voldesc( handle ); + if (!offs) return FS_UNKNOWN; + + if (!read_cd_block(handle, buff, offs)) + return FS_ERROR; + + /* check for the iso9660 identifier */ + if (!memcmp(&buff[1], "CD001", 5)) return FS_ISO9660; + return FS_UNKNOWN; +} + + +/************************************************************************** + * udf_find_pvd + */ +static BOOL udf_find_pvd( HANDLE handle, BYTE pvd[] ) +{ + unsigned int i; + DWORD offset; + INT locations[] = { 256, -1, -257, 512 }; + + for(i=0; i 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; + } + case FS_UDF: + { + BYTE pvd[BLOCK_SIZE]; + + if(!udf_find_pvd(handle, pvd)) + { + label_len = 0; + break; + } + + /* [E] 3/10.1.4 and [U] 2.1.1 */ + if(pvd[24]==8) + { + label_ptr = pvd + 24 + 1; + label_len = pvd[24+32-1]; + break; + } + else + { + unsigned int i; + + label_len = 1 + pvd[24+32-1]; + for(i=0; iLastTrack - 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 (kernelbase.@) + */ +BOOL WINAPI DECLSPEC_HOTPATCH GetVolumeInformationW( LPCWSTR root, LPWSTR label, DWORD label_len, + DWORD *serial, DWORD *filename_len, DWORD *flags, + LPWSTR fsname, DWORD fsname_len ) +{ + HANDLE handle; + NTSTATUS status; + UNICODE_STRING nt_name; + IO_STATUS_BLOCK io; + OBJECT_ATTRIBUTES attr; + FILE_FS_DEVICE_INFORMATION info; + unsigned int i; + enum fs_type type = FS_UNKNOWN; + BOOL ret = FALSE; + + if (!root) root = L"\\"; + if (!RtlDosPathNameToNtPathName_U( root, &nt_name, NULL, NULL )) + { + SetLastError( ERROR_PATH_NOT_FOUND ); + return FALSE; + } + /* there must be exactly one backslash in the name, at the end */ + for (i = 4; i < nt_name.Length / sizeof(WCHAR); i++) if (nt_name.Buffer[i] == '\\') break; + if (i != nt_name.Length / sizeof(WCHAR) - 1) + { + /* check if root contains an explicit subdir */ + if (root[0] && root[1] == ':') root += 2; + while (*root == '\\') root++; + if (wcschr( root, '\\' )) + SetLastError( ERROR_DIR_NOT_ROOT ); + else + SetLastError( ERROR_INVALID_NAME ); + goto done; + } + + /* try to open the device */ + + attr.Length = sizeof(attr); + attr.RootDirectory = 0; + attr.Attributes = OBJ_CASE_INSENSITIVE; + attr.ObjectName = &nt_name; + attr.SecurityDescriptor = NULL; + attr.SecurityQualityOfService = NULL; + + nt_name.Length -= sizeof(WCHAR); /* without trailing slash */ + status = NtOpenFile( &handle, GENERIC_READ | SYNCHRONIZE, &attr, &io, FILE_SHARE_READ | FILE_SHARE_WRITE, + FILE_NON_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT ); + nt_name.Length += sizeof(WCHAR); + + if (status == STATUS_SUCCESS) + { + 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(nt_name.Buffer) ); + if (label) lstrcpynW( label, L"Audio CD", label_len ); + if (serial) *serial = get_audiocd_serial( &toc ); + CloseHandle( handle ); + type = FS_ISO9660; + goto fill_fs_info; + } + type = read_cd_superblock( handle, superblock ); + } + else + { + type = read_fat_superblock( handle, superblock ); + if (type == FS_UNKNOWN) type = read_cd_superblock( handle, superblock ); + } + TRACE( "%s: found fs type %d\n", debugstr_w(nt_name.Buffer), type ); + if (type == FS_ERROR) + { + CloseHandle( handle ); + goto done; + } + + if (label && label_len) get_superblock_label( &nt_name, handle, type, superblock, label, label_len ); + if (serial) *serial = get_superblock_serial( &nt_name, handle, type, superblock ); + CloseHandle( handle ); + goto fill_fs_info; + } + else + { + TRACE( "cannot open device %s: %x\n", debugstr_w(nt_name.Buffer), status ); + if (status == STATUS_ACCESS_DENIED) + MESSAGE( "wine: Read access denied for device %s, FS volume label and serial are not available.\n", debugstr_w(nt_name.Buffer) ); + } + /* we couldn't open the device, fallback to default strategy */ + + if (!set_ntstatus( NtOpenFile( &handle, SYNCHRONIZE, &attr, &io, 0, + FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT ))) + goto done; + + status = NtQueryVolumeInformationFile( handle, &io, &info, sizeof(info), FileFsDeviceInformation ); + NtClose( handle ); + if (!set_ntstatus( status )) goto done; + + if (info.DeviceType == FILE_DEVICE_CD_ROM_FILE_SYSTEM) type = FS_ISO9660; + + if (label && label_len) get_filesystem_label( &nt_name, label, label_len ); + if (serial) *serial = get_filesystem_serial( &nt_name ); + +fill_fs_info: /* now fill in the information that depends on the file system type */ + + switch(type) + { + case FS_ISO9660: + if (fsname) lstrcpynW( fsname, L"CDFS", fsname_len ); + if (filename_len) *filename_len = 221; + if (flags) *flags = FILE_READ_ONLY_VOLUME; + break; + case FS_UDF: + if (fsname) lstrcpynW( fsname, L"UDF", fsname_len ); + if (filename_len) *filename_len = 255; + if (flags) + *flags = FILE_READ_ONLY_VOLUME | FILE_UNICODE_ON_DISK | FILE_CASE_SENSITIVE_SEARCH; + break; + case FS_FAT1216: + if (fsname) lstrcpynW( fsname, L"FAT", fsname_len ); + case FS_FAT32: + if (type == FS_FAT32 && fsname) lstrcpynW( fsname, L"FAT32", fsname_len ); + if (filename_len) *filename_len = 255; + if (flags) *flags = FILE_CASE_PRESERVED_NAMES; /* FIXME */ + break; + default: + if (fsname) lstrcpynW( fsname, L"NTFS", fsname_len ); + if (filename_len) *filename_len = 255; + if (flags) *flags = FILE_CASE_PRESERVED_NAMES | FILE_PERSISTENT_ACLS; + break; + } + ret = TRUE; + +done: + RtlFreeUnicodeString( &nt_name ); + return ret; +} + + +/*********************************************************************** + * GetVolumeInformationA (kernelbase.@) + */ +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 ); + } + + HeapFree( GetProcessHeap(), 0, labelW ); + HeapFree( GetProcessHeap(), 0, fsnameW ); + return ret; +} + + +/*********************************************************************** + * GetVolumeNameForVolumeMountPointW (kernelbase.@) + */ +BOOL WINAPI GetVolumeNameForVolumeMountPointW( LPCWSTR path, LPWSTR volume, DWORD size ) +{ + MOUNTMGR_MOUNT_POINT *input = NULL, *o1; + MOUNTMGR_MOUNT_POINTS *output = NULL; + WCHAR *p; + char *r; + DWORD i, i_size = 1024, o_size = 1024; + WCHAR *nonpersist_name; + WCHAR symlink_name[MAX_PATH]; + NTSTATUS status; + HANDLE mgr = INVALID_HANDLE_VALUE; + BOOL ret = FALSE; + DWORD br; + + TRACE("(%s, %p, %x)\n", debugstr_w(path), volume, size); + if (path[lstrlenW(path)-1] != '\\') + { + SetLastError( ERROR_INVALID_NAME ); + return FALSE; + } + + if (size < 50) + { + SetLastError( ERROR_FILENAME_EXCED_RANGE ); + return FALSE; + } + /* if length of input is > 3 then it must be a mounted folder */ + if (lstrlenW(path) > 3) + { + FIXME("Mounted Folders are not yet supported\n"); + SetLastError( ERROR_NOT_A_REPARSE_POINT ); + return FALSE; + } + + mgr = CreateFileW( MOUNTMGR_DOS_DEVICE_NAME, 0, FILE_SHARE_READ, + NULL, OPEN_EXISTING, 0, 0 ); + if (mgr == INVALID_HANDLE_VALUE) return FALSE; + + if (!(input = HeapAlloc( GetProcessHeap(), 0, i_size ))) + { + SetLastError( ERROR_NOT_ENOUGH_MEMORY ); + goto err_ret; + } + + if (!(output = HeapAlloc( GetProcessHeap(), 0, o_size ))) + { + SetLastError( ERROR_NOT_ENOUGH_MEMORY ); + goto err_ret; + } + + /* construct the symlink name as "\DosDevices\C:" */ + lstrcpyW( symlink_name, L"\\DosDevices\\" ); + lstrcatW( symlink_name, path ); + symlink_name[lstrlenW(symlink_name)-1] = 0; + + /* Take the mount point and get the "nonpersistent name" */ + /* We will then take that and get the volume name */ + nonpersist_name = (WCHAR *)(input + 1); + status = read_nt_symlink( symlink_name, nonpersist_name, i_size - sizeof(*input) ); + TRACE("read_nt_symlink got stat=%x, for %s, got <%s>\n", status, + debugstr_w(symlink_name), debugstr_w(nonpersist_name)); + if (status != STATUS_SUCCESS) + { + SetLastError( ERROR_FILE_NOT_FOUND ); + goto err_ret; + } + + /* Now take the "nonpersistent name" and ask the mountmgr */ + /* to give us all the mount points. One of them will be */ + /* the volume name (format of \??\Volume{). */ + memset( input, 0, sizeof(*input) ); /* clear all input parameters */ + input->DeviceNameOffset = sizeof(*input); + input->DeviceNameLength = lstrlenW( nonpersist_name) * sizeof(WCHAR); + i_size = input->DeviceNameOffset + input->DeviceNameLength; + + output->Size = o_size; + + /* now get the true volume name from the mountmgr */ + if (!DeviceIoControl( mgr, IOCTL_MOUNTMGR_QUERY_POINTS, input, i_size, + output, o_size, &br, NULL )) + goto err_ret; + + /* Verify and return the data, note string is not null terminated */ + TRACE("found %d matching mount points\n", output->NumberOfMountPoints); + if (output->NumberOfMountPoints < 1) + { + SetLastError( ERROR_NO_VOLUME_ID ); + goto err_ret; + } + o1 = &output->MountPoints[0]; + + /* look for the volume name in returned values */ + for(i=0;iNumberOfMountPoints;i++) + { + p = (WCHAR*)((char *)output + o1->SymbolicLinkNameOffset); + r = (char *)output + o1->UniqueIdOffset; + TRACE("found symlink=%s, unique=%s, devname=%s\n", + debugstr_wn(p, o1->SymbolicLinkNameLength/sizeof(WCHAR)), + debugstr_an(r, o1->UniqueIdLength), + debugstr_wn((WCHAR*)((char *)output + o1->DeviceNameOffset), + o1->DeviceNameLength/sizeof(WCHAR))); + + if (!wcsncmp( p, L"\\??\\Volume{", wcslen(L"\\??\\Volume{") )) + { + /* is there space in the return variable ?? */ + if ((o1->SymbolicLinkNameLength/sizeof(WCHAR))+2 > size) + { + SetLastError( ERROR_FILENAME_EXCED_RANGE ); + goto err_ret; + } + memcpy( volume, p, o1->SymbolicLinkNameLength ); + volume[o1->SymbolicLinkNameLength / sizeof(WCHAR)] = 0; + lstrcatW( volume, L"\\" ); + /* change second char from '?' to '\' */ + volume[1] = '\\'; + ret = TRUE; + break; + } + o1++; + } + +err_ret: + HeapFree( GetProcessHeap(), 0, input ); + HeapFree( GetProcessHeap(), 0, output ); + CloseHandle( mgr ); + return ret; +} + + +/*********************************************************************** + * DefineDosDeviceW (kernelbase.@) + */ +BOOL WINAPI DECLSPEC_HOTPATCH DefineDosDeviceW( DWORD flags, const WCHAR *device, const WCHAR *target ) +{ + WCHAR link_name[15] = L"\\DosDevices\\"; + UNICODE_STRING nt_name, nt_target; + OBJECT_ATTRIBUTES attr; + NTSTATUS status; + HANDLE handle; + + TRACE("%#x, %s, %s\n", flags, debugstr_w(device), debugstr_w(target)); + + if (flags & ~(DDD_RAW_TARGET_PATH | DDD_REMOVE_DEFINITION)) + FIXME("Ignoring flags %#x.\n", flags & ~(DDD_RAW_TARGET_PATH | DDD_REMOVE_DEFINITION)); + + lstrcatW( link_name, device ); + RtlInitUnicodeString( &nt_name, link_name ); + InitializeObjectAttributes( &attr, &nt_name, OBJ_CASE_INSENSITIVE, 0, NULL ); + if (flags & DDD_REMOVE_DEFINITION) + { + if (!set_ntstatus( NtOpenSymbolicLinkObject( &handle, 0, &attr ) )) + return FALSE; + + SERVER_START_REQ( unlink_object ) + { + req->handle = wine_server_obj_handle( handle ); + status = wine_server_call( req ); + } + SERVER_END_REQ; + NtClose( handle ); + + return set_ntstatus( status ); + } + + if (!(flags & DDD_RAW_TARGET_PATH)) + { + if (!RtlDosPathNameToNtPathName_U( target, &nt_target, NULL, NULL)) + { + SetLastError( ERROR_PATH_NOT_FOUND ); + return FALSE; + } + } + else + RtlInitUnicodeString( &nt_target, target ); + + return set_ntstatus( NtCreateSymbolicLinkObject( &handle, SYMBOLIC_LINK_ALL_ACCESS, &attr, &nt_target ) ); +} + + +/*********************************************************************** + * QueryDosDeviceW (kernelbase.@) + * + * returns array of strings terminated by \0, terminated by \0 + */ +DWORD WINAPI QueryDosDeviceW( LPCWSTR devname, LPWSTR target, DWORD bufsize ) +{ + UNICODE_STRING nt_name; + NTSTATUS status; + + if (!bufsize) + { + SetLastError( ERROR_INSUFFICIENT_BUFFER ); + return 0; + } + + if (devname) + { + WCHAR name[8]; + WCHAR *buffer; + DWORD dosdev, ret = 0; + + if ((dosdev = RtlIsDosDeviceName_U( devname ))) + { + memcpy( name, devname + HIWORD(dosdev)/sizeof(WCHAR), LOWORD(dosdev) ); + name[LOWORD(dosdev)/sizeof(WCHAR)] = 0; + devname = name; + } + + if (!(buffer = HeapAlloc( GetProcessHeap(), 0, sizeof(L"\\DosDevices\\") + lstrlenW(devname)*sizeof(WCHAR) ))) + { + SetLastError( ERROR_OUTOFMEMORY ); + return 0; + } + lstrcpyW( buffer, L"\\DosDevices\\" ); + lstrcatW( buffer, devname ); + status = read_nt_symlink( buffer, target, bufsize ); + HeapFree( GetProcessHeap(), 0, buffer ); + if (!set_ntstatus( status )) return 0; + ret = lstrlenW( target ) + 1; + if (ret < bufsize) target[ret++] = 0; /* add an extra null */ + return ret; + } + else /* return a list of all devices */ + { + OBJECT_ATTRIBUTES attr; + HANDLE handle; + WCHAR *p = target; + + RtlInitUnicodeString( &nt_name, L"\\DosDevices\\" ); + nt_name.Length -= sizeof(WCHAR); /* without trailing slash */ + attr.Length = sizeof(attr); + attr.RootDirectory = 0; + attr.ObjectName = &nt_name; + attr.Attributes = OBJ_CASE_INSENSITIVE; + attr.SecurityDescriptor = NULL; + attr.SecurityQualityOfService = NULL; + status = NtOpenDirectoryObject( &handle, FILE_LIST_DIRECTORY, &attr ); + if (!status) + { + char data[1024]; + DIRECTORY_BASIC_INFORMATION *info = (DIRECTORY_BASIC_INFORMATION *)data; + ULONG ctx = 0, len; + + while (!NtQueryDirectoryObject( handle, info, sizeof(data), 1, 0, &ctx, &len )) + { + if (p + info->ObjectName.Length/sizeof(WCHAR) + 1 >= target + bufsize) + { + SetLastError( ERROR_INSUFFICIENT_BUFFER ); + NtClose( handle ); + return 0; + } + memcpy( p, info->ObjectName.Buffer, info->ObjectName.Length ); + p += info->ObjectName.Length/sizeof(WCHAR); + *p++ = 0; + } + NtClose( handle ); + } + + *p++ = 0; /* terminating null */ + return p - target; + } +} + + +/*********************************************************************** + * GetLogicalDrives (kernelbase.@) + */ +DWORD WINAPI DECLSPEC_HOTPATCH GetLogicalDrives(void) +{ + OBJECT_ATTRIBUTES attr; + UNICODE_STRING nt_name; + DWORD bitmask = 0; + NTSTATUS status; + HANDLE handle; + + RtlInitUnicodeString( &nt_name, L"\\DosDevices\\" ); + nt_name.Length -= sizeof(WCHAR); /* without trailing slash */ + attr.Length = sizeof(attr); + attr.RootDirectory = 0; + attr.ObjectName = &nt_name; + attr.Attributes = OBJ_CASE_INSENSITIVE; + attr.SecurityDescriptor = NULL; + attr.SecurityQualityOfService = NULL; + status = NtOpenDirectoryObject( &handle, FILE_LIST_DIRECTORY, &attr ); + if (!status) + { + char data[1024]; + DIRECTORY_BASIC_INFORMATION *info = (DIRECTORY_BASIC_INFORMATION *)data; + ULONG ctx = 0, len; + + while (!NtQueryDirectoryObject( handle, info, sizeof(data), 1, 0, &ctx, &len )) + if(info->ObjectName.Length == 2*sizeof(WCHAR) && info->ObjectName.Buffer[1] == ':') + bitmask |= 1 << (info->ObjectName.Buffer[0] - 'A'); + + NtClose( handle ); + } + + return bitmask; +} + + +/*********************************************************************** + * GetLogicalDriveStringsW (kernelbase.@) + */ +UINT WINAPI DECLSPEC_HOTPATCH 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 (kernelbase.@) + */ +UINT WINAPI DECLSPEC_HOTPATCH GetDriveTypeW( LPCWSTR root ) +{ + FILE_FS_DEVICE_INFORMATION info; + IO_STATUS_BLOCK io; + NTSTATUS status; + HANDLE handle; + UINT ret; + + if (!open_device_root( root, &handle )) + { + /* CD ROM devices do not necessarily have a volume, but a drive type */ + ret = get_mountmgr_drive_type( root ); + if (ret == DRIVE_CDROM || ret == DRIVE_REMOVABLE) + return ret; + + 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 + { + 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 if ((ret = get_mountmgr_drive_type( root )) == DRIVE_UNKNOWN) ret = DRIVE_FIXED; + break; + default: + ret = DRIVE_UNKNOWN; + break; + } + } + TRACE( "%s -> %d\n", debugstr_w(root), ret ); + return ret; +} + + +/*********************************************************************** + * GetDriveTypeA (kernelbase.@) + */ +UINT WINAPI DECLSPEC_HOTPATCH GetDriveTypeA( LPCSTR root ) +{ + WCHAR *rootW = NULL; + + if (root && !(rootW = file_name_AtoW( root, FALSE ))) return DRIVE_NO_ROOT_DIR; + return GetDriveTypeW( rootW ); +} + + +/*********************************************************************** + * GetDiskFreeSpaceExW (kernelbase.@) + */ +BOOL WINAPI DECLSPEC_HOTPATCH 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 (!set_ntstatus( 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 (kernelbase.@) + */ +BOOL WINAPI DECLSPEC_HOTPATCH 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 (kernelbase.@) + */ +BOOL WINAPI DECLSPEC_HOTPATCH 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 (!set_ntstatus( status )) return FALSE; + + units = info.SectorsPerAllocationUnit * info.BytesPerSector; + + if( GetVersion() & 0x80000000) { /* win3.x, 9x, ME */ + /* cap the size and available at 2GB as per specs */ + if (info.TotalAllocationUnits.QuadPart * units > 0x7fffffff) { + info.TotalAllocationUnits.QuadPart = 0x7fffffff / units; + if (info.AvailableAllocationUnits.QuadPart * units > 0x7fffffff) + info.AvailableAllocationUnits.QuadPart = 0x7fffffff / units; + } + /* nr. of clusters is always <= 65335 */ + while( info.TotalAllocationUnits.QuadPart > 65535 ) { + info.TotalAllocationUnits.QuadPart /= 2; + info.AvailableAllocationUnits.QuadPart /= 2; + info.SectorsPerAllocationUnit *= 2; + } + } + + 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; + TRACE("%#08x, %#08x, %#08x, %#08x\n", info.SectorsPerAllocationUnit, info.BytesPerSector, + info.AvailableAllocationUnits.u.LowPart, info.TotalAllocationUnits.u.LowPart); + return TRUE; +} + + +/*********************************************************************** + * GetDiskFreeSpaceA (kernelbase.@) + */ +BOOL WINAPI DECLSPEC_HOTPATCH 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 ); +} + + +static BOOL is_dos_path( const UNICODE_STRING *path ) +{ + static const WCHAR global_prefix[4] = {'\\','?','?','\\'}; + return path->Length >= 7 * sizeof(WCHAR) + && !memcmp(path->Buffer, global_prefix, sizeof(global_prefix)) + && path->Buffer[5] == ':' && path->Buffer[6] == '\\'; +} + +/* resolve all symlinks in a path in-place; return FALSE if allocation failed */ +static BOOL resolve_symlink( UNICODE_STRING *path ) +{ + OBJECT_NAME_INFORMATION *info; + OBJECT_ATTRIBUTES attr; + IO_STATUS_BLOCK io; + NTSTATUS status; + HANDLE file; + ULONG size; + + InitializeObjectAttributes( &attr, path, OBJ_CASE_INSENSITIVE, 0, NULL ); + if (NtOpenFile( &file, SYNCHRONIZE, &attr, &io, FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, + FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT )) + return TRUE; + + if (NtQueryObject( file, ObjectNameInformation, NULL, 0, &size ) != STATUS_INFO_LENGTH_MISMATCH) + { + NtClose( file ); + return TRUE; + } + + if (!(info = HeapAlloc( GetProcessHeap(), 0, size ))) + { + NtClose( file ); + return FALSE; + } + + status = NtQueryObject( file, ObjectNameInformation, info, size, NULL ); + NtClose( file ); + if (status) + return TRUE; + + RtlFreeUnicodeString( path ); + status = RtlDuplicateUnicodeString( 0, &info->Name, path ); + HeapFree( GetProcessHeap(), 0, info ); + return !status; +} + +/*********************************************************************** + * GetVolumePathNameW (kernelbase.@) + */ +BOOL WINAPI DECLSPEC_HOTPATCH GetVolumePathNameW( const WCHAR *path, WCHAR *volume_path, DWORD length ) +{ + FILE_ATTRIBUTE_TAG_INFORMATION attr_info; + FILE_BASIC_INFORMATION basic_info; + OBJECT_ATTRIBUTES attr; + UNICODE_STRING nt_name; + NTSTATUS status; + + if (path && !wcsncmp(path, L"\\??\\", 4)) + { + WCHAR current_drive[MAX_PATH]; + + GetCurrentDirectoryW( ARRAY_SIZE(current_drive), current_drive ); + if (length >= 3) + { + WCHAR ret_path[4] = {current_drive[0], ':', '\\', 0}; + lstrcpynW( volume_path, ret_path, length ); + return TRUE; + } + + SetLastError( ERROR_FILENAME_EXCED_RANGE ); + return FALSE; + } + + if (!volume_path || !length || !RtlDosPathNameToNtPathName_U( path, &nt_name, NULL, NULL )) + { + SetLastError( ERROR_INVALID_PARAMETER ); + return FALSE; + } + + if (!is_dos_path( &nt_name )) + { + RtlFreeUnicodeString( &nt_name ); + WARN("invalid path %s\n", debugstr_w(path)); + SetLastError( ERROR_INVALID_NAME ); + return FALSE; + } + + InitializeObjectAttributes( &attr, &nt_name, OBJ_CASE_INSENSITIVE, 0, NULL ); + + while (nt_name.Length > 7 * sizeof(WCHAR)) + { + IO_STATUS_BLOCK io; + HANDLE file; + + if (!NtQueryAttributesFile( &attr, &basic_info ) + && (basic_info.FileAttributes & FILE_ATTRIBUTE_REPARSE_POINT) + && (basic_info.FileAttributes & FILE_ATTRIBUTE_DIRECTORY) + && !NtOpenFile( &file, SYNCHRONIZE | FILE_READ_ATTRIBUTES, &attr, &io, + FILE_SHARE_READ | FILE_SHARE_WRITE | FILE_SHARE_DELETE, + FILE_OPEN_REPARSE_POINT | FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT )) + { + status = NtQueryInformationFile( file, &io, &attr_info, + sizeof(attr_info), FileAttributeTagInformation ); + NtClose( file ); + if (!status) + { + + if (attr_info.ReparseTag == IO_REPARSE_TAG_MOUNT_POINT) + break; + + if (!resolve_symlink( &nt_name )) + { + SetLastError( ERROR_OUTOFMEMORY ); + return FALSE; + } + } + } + + if (nt_name.Buffer[(nt_name.Length / sizeof(WCHAR)) - 1] == '\\') + nt_name.Length -= sizeof(WCHAR); + while (nt_name.Length && nt_name.Buffer[(nt_name.Length / sizeof(WCHAR)) - 1] != '\\') + nt_name.Length -= sizeof(WCHAR); + } + + nt_name.Buffer[nt_name.Length / sizeof(WCHAR)] = 0; + + if (NtQueryAttributesFile( &attr, &basic_info )) + { + RtlFreeUnicodeString( &nt_name ); + WARN("nonexistent path %s -> %s\n", debugstr_w(path), debugstr_w( nt_name.Buffer )); + SetLastError( ERROR_FILE_NOT_FOUND ); + return FALSE; + } + + if (!wcsncmp(path, L"\\\\.\\", 4) || !wcsncmp(path, L"\\\\?\\", 4)) + { + if (length >= nt_name.Length / sizeof(WCHAR)) + { + memcpy(volume_path, path, 4 * sizeof(WCHAR)); + lstrcpynW( volume_path + 4, nt_name.Buffer + 4, length - 4 ); + + TRACE("%s -> %s\n", debugstr_w(path), debugstr_w(volume_path)); + + RtlFreeUnicodeString( &nt_name ); + return TRUE; + } + } + else if (length >= (nt_name.Length / sizeof(WCHAR)) - 4) + { + lstrcpynW( volume_path, nt_name.Buffer + 4, length ); + volume_path[0] = towupper(volume_path[0]); + + TRACE("%s -> %s\n", debugstr_w(path), debugstr_w(volume_path)); + + RtlFreeUnicodeString( &nt_name ); + return TRUE; + } + + RtlFreeUnicodeString( &nt_name ); + SetLastError( ERROR_FILENAME_EXCED_RANGE ); + return FALSE; +} + + +static MOUNTMGR_MOUNT_POINTS *query_mount_points( HANDLE mgr, MOUNTMGR_MOUNT_POINT *input, DWORD insize ) +{ + MOUNTMGR_MOUNT_POINTS *output; + DWORD outsize = 1024; + DWORD br; + + for (;;) + { + if (!(output = HeapAlloc( GetProcessHeap(), 0, outsize ))) + { + SetLastError( ERROR_NOT_ENOUGH_MEMORY ); + return NULL; + } + if (DeviceIoControl( mgr, IOCTL_MOUNTMGR_QUERY_POINTS, input, insize, output, outsize, &br, NULL )) break; + outsize = output->Size; + HeapFree( GetProcessHeap(), 0, output ); + if (GetLastError() != ERROR_MORE_DATA) return NULL; + } + return output; +} + +/*********************************************************************** + * GetVolumePathNamesForVolumeNameW (kernelbase.@) + */ +BOOL WINAPI DECLSPEC_HOTPATCH GetVolumePathNamesForVolumeNameW( LPCWSTR volumename, LPWSTR volumepathname, + DWORD buflen, PDWORD returnlen ) +{ + static const WCHAR dosdevicesW[] = {'\\','D','o','s','D','e','v','i','c','e','s','\\'}; + HANDLE mgr; + DWORD len, size; + MOUNTMGR_MOUNT_POINT *spec; + MOUNTMGR_MOUNT_POINTS *link, *target = NULL; + WCHAR *name, *path; + BOOL ret = FALSE; + UINT i, j; + + TRACE("%s, %p, %u, %p\n", debugstr_w(volumename), volumepathname, buflen, returnlen); + + if (!volumename || (len = lstrlenW( volumename )) != 49) + { + SetLastError( ERROR_INVALID_NAME ); + return FALSE; + } + mgr = CreateFileW( MOUNTMGR_DOS_DEVICE_NAME, 0, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, 0, 0 ); + if (mgr == INVALID_HANDLE_VALUE) return FALSE; + + size = sizeof(*spec) + sizeof(WCHAR) * (len - 1); /* remove trailing backslash */ + if (!(spec = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, size ))) goto done; + spec->SymbolicLinkNameOffset = sizeof(*spec); + spec->SymbolicLinkNameLength = size - sizeof(*spec); + name = (WCHAR *)((char *)spec + spec->SymbolicLinkNameOffset); + memcpy( name, volumename, size - sizeof(*spec) ); + name[1] = '?'; /* map \\?\ to \??\ */ + + target = query_mount_points( mgr, spec, size ); + HeapFree( GetProcessHeap(), 0, spec ); + if (!target) + { + goto done; + } + if (!target->NumberOfMountPoints) + { + SetLastError( ERROR_FILE_NOT_FOUND ); + goto done; + } + len = 0; + path = volumepathname; + for (i = 0; i < target->NumberOfMountPoints; i++) + { + link = NULL; + if (target->MountPoints[i].DeviceNameOffset) + { + const WCHAR *device = (const WCHAR *)((const char *)target + target->MountPoints[i].DeviceNameOffset); + USHORT device_len = target->MountPoints[i].DeviceNameLength; + + size = sizeof(*spec) + device_len; + if (!(spec = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, size ))) goto done; + spec->DeviceNameOffset = sizeof(*spec); + spec->DeviceNameLength = device_len; + memcpy( (char *)spec + spec->DeviceNameOffset, device, device_len ); + + link = query_mount_points( mgr, spec, size ); + HeapFree( GetProcessHeap(), 0, spec ); + } + else if (target->MountPoints[i].UniqueIdOffset) + { + const WCHAR *id = (const WCHAR *)((const char *)target + target->MountPoints[i].UniqueIdOffset); + USHORT id_len = target->MountPoints[i].UniqueIdLength; + + size = sizeof(*spec) + id_len; + if (!(spec = HeapAlloc( GetProcessHeap(), HEAP_ZERO_MEMORY, size ))) goto done; + spec->UniqueIdOffset = sizeof(*spec); + spec->UniqueIdLength = id_len; + memcpy( (char *)spec + spec->UniqueIdOffset, id, id_len ); + + link = query_mount_points( mgr, spec, size ); + HeapFree( GetProcessHeap(), 0, spec ); + } + if (!link) continue; + for (j = 0; j < link->NumberOfMountPoints; j++) + { + const WCHAR *linkname; + + if (!link->MountPoints[j].SymbolicLinkNameOffset) continue; + linkname = (const WCHAR *)((const char *)link + link->MountPoints[j].SymbolicLinkNameOffset); + + if (link->MountPoints[j].SymbolicLinkNameLength == sizeof(dosdevicesW) + 2 * sizeof(WCHAR) && + !wcsnicmp( linkname, dosdevicesW, ARRAY_SIZE( dosdevicesW ))) + { + len += 4; + if (volumepathname && len < buflen) + { + path[0] = linkname[ARRAY_SIZE( dosdevicesW )]; + path[1] = ':'; + path[2] = '\\'; + path[3] = 0; + path += 4; + } + } + } + HeapFree( GetProcessHeap(), 0, link ); + } + if (buflen <= len) SetLastError( ERROR_MORE_DATA ); + else if (volumepathname) + { + volumepathname[len] = 0; + ret = TRUE; + } + if (returnlen) *returnlen = len + 1; + +done: + HeapFree( GetProcessHeap(), 0, target ); + CloseHandle( mgr ); + return ret; +} + + +/*********************************************************************** + * FindFirstVolumeW (kernelbase.@) + */ +HANDLE WINAPI DECLSPEC_HOTPATCH FindFirstVolumeW( LPWSTR volume, DWORD len ) +{ + DWORD size = 1024; + DWORD br; + HANDLE mgr = CreateFileW( MOUNTMGR_DOS_DEVICE_NAME, 0, FILE_SHARE_READ|FILE_SHARE_WRITE, + NULL, OPEN_EXISTING, 0, 0 ); + if (mgr == INVALID_HANDLE_VALUE) return INVALID_HANDLE_VALUE; + + for (;;) + { + MOUNTMGR_MOUNT_POINT input; + MOUNTMGR_MOUNT_POINTS *output; + + if (!(output = HeapAlloc( GetProcessHeap(), 0, size ))) + { + SetLastError( ERROR_NOT_ENOUGH_MEMORY ); + break; + } + memset( &input, 0, sizeof(input) ); + + if (!DeviceIoControl( mgr, IOCTL_MOUNTMGR_QUERY_POINTS, &input, sizeof(input), + output, size, &br, NULL )) + { + if (GetLastError() != ERROR_MORE_DATA) break; + size = output->Size; + HeapFree( GetProcessHeap(), 0, output ); + continue; + } + CloseHandle( mgr ); + /* abuse the Size field to store the current index */ + output->Size = 0; + if (!FindNextVolumeW( output, volume, len )) + { + HeapFree( GetProcessHeap(), 0, output ); + return INVALID_HANDLE_VALUE; + } + return output; + } + CloseHandle( mgr ); + return INVALID_HANDLE_VALUE; +} + + +/*********************************************************************** + * FindNextVolumeW (kernelbase.@) + */ +BOOL WINAPI DECLSPEC_HOTPATCH FindNextVolumeW( HANDLE handle, LPWSTR volume, DWORD len ) +{ + MOUNTMGR_MOUNT_POINTS *data = handle; + + while (data->Size < data->NumberOfMountPoints) + { + static const WCHAR volumeW[] = {'\\','?','?','\\','V','o','l','u','m','e','{',}; + WCHAR *link = (WCHAR *)((char *)data + data->MountPoints[data->Size].SymbolicLinkNameOffset); + DWORD size = data->MountPoints[data->Size].SymbolicLinkNameLength; + data->Size++; + /* skip non-volumes */ + if (size < sizeof(volumeW) || memcmp( link, volumeW, sizeof(volumeW) )) continue; + if (size + sizeof(WCHAR) >= len * sizeof(WCHAR)) + { + SetLastError( ERROR_FILENAME_EXCED_RANGE ); + return FALSE; + } + memcpy( volume, link, size ); + volume[1] = '\\'; /* map \??\ to \\?\ */ + volume[size / sizeof(WCHAR)] = '\\'; /* Windows appends a backslash */ + volume[size / sizeof(WCHAR) + 1] = 0; + TRACE( "returning entry %u %s\n", data->Size - 1, debugstr_w(volume) ); + return TRUE; + } + SetLastError( ERROR_NO_MORE_FILES ); + return FALSE; +} + + +/*********************************************************************** + * FindVolumeClose (kernelbase.@) + */ +BOOL WINAPI DECLSPEC_HOTPATCH FindVolumeClose( HANDLE handle ) +{ + return HeapFree( GetProcessHeap(), 0, handle ); +} + + +/*********************************************************************** + * DeleteVolumeMountPointW (kernelbase.@) + */ +BOOL WINAPI /* DECLSPEC_HOTPATCH */ DeleteVolumeMountPointW( LPCWSTR mountpoint ) +{ + FIXME("(%s), stub!\n", debugstr_w(mountpoint)); + return FALSE; +} + + +/*********************************************************************** + * GetVolumeInformationByHandleW (kernelbase.@) + */ +BOOL WINAPI GetVolumeInformationByHandleW( HANDLE handle, WCHAR *label, DWORD label_len, + DWORD *serial, DWORD *filename_len, DWORD *flags, + WCHAR *fsname, DWORD fsname_len ) +{ + IO_STATUS_BLOCK io; + + TRACE( "%p\n", handle ); + + if (label || serial) + { + char buffer[sizeof(FILE_FS_VOLUME_INFORMATION) + MAX_PATH * sizeof(WCHAR)]; + FILE_FS_VOLUME_INFORMATION *info = (FILE_FS_VOLUME_INFORMATION *)buffer; + + if (!set_ntstatus( NtQueryVolumeInformationFile( handle, &io, info, sizeof(buffer), + FileFsVolumeInformation ) )) + return FALSE; + + if (label) + { + if (label_len < info->VolumeLabelLength / sizeof(WCHAR) + 1) + { + SetLastError( ERROR_BAD_LENGTH ); + return FALSE; + } + memcpy( label, info->VolumeLabel, info->VolumeLabelLength ); + label[info->VolumeLabelLength / sizeof(WCHAR)] = 0; + } + if (serial) *serial = info->VolumeSerialNumber; + } + + if (filename_len || flags || fsname) + { + char buffer[sizeof(FILE_FS_ATTRIBUTE_INFORMATION) + MAX_PATH * sizeof(WCHAR)]; + FILE_FS_ATTRIBUTE_INFORMATION *info = (FILE_FS_ATTRIBUTE_INFORMATION *)buffer; + + if (!set_ntstatus( NtQueryVolumeInformationFile( handle, &io, info, sizeof(buffer), + FileFsAttributeInformation ) )) + return FALSE; + + if (fsname) + { + if (fsname_len < info->FileSystemNameLength / sizeof(WCHAR) + 1) + { + SetLastError( ERROR_BAD_LENGTH ); + return FALSE; + } + memcpy( fsname, info->FileSystemName, info->FileSystemNameLength ); + fsname[info->FileSystemNameLength / sizeof(WCHAR)] = 0; + } + if (filename_len) *filename_len = info->MaximumComponentNameLength; + if (flags) *flags = info->FileSystemAttributes; + } + + return TRUE; +}