mirror of
git://source.winehq.org/git/wine.git
synced 2024-09-02 14:33:38 +00:00
1197 lines
40 KiB
C
1197 lines
40 KiB
C
/*
|
|
* 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 <stdarg.h>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
|
|
#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;
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* 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;
|
|
unsigned int i;
|
|
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;
|
|
}
|
|
|
|
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)
|
|
{
|
|
TRACE( "cannot open device %s: %lx\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) );
|
|
status = NtOpenFile( &handle, SYNCHRONIZE, &attr, &io, 0,
|
|
FILE_DIRECTORY_FILE | FILE_SYNCHRONOUS_IO_NONALERT );
|
|
}
|
|
|
|
if (!set_ntstatus( status )) goto done;
|
|
|
|
ret = GetVolumeInformationByHandleW( handle, label, label_len, serial, filename_len, flags,
|
|
fsname, fsname_len );
|
|
NtClose( handle );
|
|
|
|
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, %lx)\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)) / sizeof(WCHAR) );
|
|
TRACE("read_nt_symlink got stat=%lx, 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 %ld 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;i<output->NumberOfMountPoints;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("%#lx, %s, %s\n", flags, debugstr_w(device), debugstr_w(target));
|
|
|
|
if (flags & ~(DDD_RAW_TARGET_PATH | DDD_REMOVE_DEFINITION))
|
|
FIXME("Ignoring flags %#lx.\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 )
|
|
{
|
|
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 */
|
|
{
|
|
UNICODE_STRING nt_name = RTL_CONSTANT_STRING( L"\\DosDevices" );
|
|
OBJECT_ATTRIBUTES attr;
|
|
HANDLE handle;
|
|
WCHAR *p = target;
|
|
|
|
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 = RTL_CONSTANT_STRING( L"\\DosDevices\\" );
|
|
DWORD bitmask = 0;
|
|
NTSTATUS status;
|
|
HANDLE handle;
|
|
|
|
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("%#08lx, %#08lx, %#08lx, %#08lx\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, %lu, %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 %lu %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;
|
|
}
|