/* * 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/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 ) { 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(L".windows-label"); 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), L".windows-label", sizeof(L".windows-label") ); 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 ) { 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(L".windows-serial"); 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), L".windows-serial", sizeof(L".windows-serial") ); 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 | OBJ_PERMANENT, 0, NULL ); if (flags & DDD_REMOVE_DEFINITION) { if (!set_ntstatus( NtOpenSymbolicLinkObject( &handle, 0, &attr ) )) return FALSE; status = NtMakeTemporaryObject( handle ); 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 ); if (!(status = NtCreateSymbolicLinkObject( &handle, SYMBOLIC_LINK_ALL_ACCESS, &attr, &nt_target ))) NtClose( handle ); return set_ntstatus( status ); } /*********************************************************************** * 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; }