mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
1188 lines
34 KiB
1188 lines
34 KiB
/*++
|
|
|
|
Copyright (c) 1998 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
mountie.c
|
|
|
|
Abstract:
|
|
|
|
Abstract
|
|
|
|
Author:
|
|
|
|
Rod Gamache (rodga) 4-Mar-1998
|
|
|
|
Environment:
|
|
|
|
User Mode
|
|
|
|
Revision History:
|
|
|
|
|
|
--*/
|
|
|
|
#include <nt.h>
|
|
#include <ntdef.h>
|
|
#include <ntrtl.h>
|
|
#include <nturtl.h>
|
|
#include <windows.h>
|
|
|
|
#include <devioctl.h>
|
|
//#include <ntdddisk.h>
|
|
//#include <ntddscsi.h>
|
|
#include <stdio.h>
|
|
#include <stddef.h>
|
|
#include <stdlib.h>
|
|
#include <cfgmgr32.h>
|
|
#include <mountmgr.h>
|
|
|
|
#include "disksp.h"
|
|
#include "mountie.h"
|
|
|
|
#define OUTPUT_BUFFER_LEN 1024
|
|
|
|
|
|
/*
|
|
* DevfileOpen - open a device file given a pathname
|
|
*
|
|
* Return a non-zero code for error.
|
|
*/
|
|
DWORD
|
|
DevfileOpen(
|
|
OUT HANDLE *Handle,
|
|
IN wchar_t *pathname
|
|
)
|
|
{
|
|
HANDLE fh;
|
|
OBJECT_ATTRIBUTES objattrs;
|
|
UNICODE_STRING cwspath;
|
|
NTSTATUS status;
|
|
IO_STATUS_BLOCK iostatus;
|
|
|
|
RtlInitUnicodeString(&cwspath, pathname);
|
|
InitializeObjectAttributes(&objattrs, &cwspath, OBJ_CASE_INSENSITIVE,
|
|
NULL, NULL);
|
|
fh = NULL;
|
|
status = NtOpenFile(&fh,
|
|
SYNCHRONIZE | FILE_READ_DATA | FILE_WRITE_DATA,
|
|
&objattrs, &iostatus,
|
|
FILE_SHARE_READ | FILE_SHARE_WRITE,
|
|
FILE_SYNCHRONOUS_IO_ALERT);
|
|
if (status != STATUS_SUCCESS) {
|
|
return status;
|
|
}
|
|
|
|
if (iostatus.Status != STATUS_SUCCESS) {
|
|
if (fh) {
|
|
NtClose(fh);
|
|
}
|
|
return iostatus.Status;
|
|
}
|
|
|
|
*Handle = fh;
|
|
return STATUS_SUCCESS;
|
|
|
|
} // DevfileOpen
|
|
|
|
|
|
/*
|
|
* DevfileClose - close a file
|
|
*/
|
|
VOID
|
|
DevfileClose(
|
|
IN HANDLE Handle
|
|
)
|
|
{
|
|
|
|
NtClose(Handle);
|
|
|
|
} // DevFileClose
|
|
|
|
|
|
/*
|
|
* DevfileIoctl - issue an ioctl to a device
|
|
*/
|
|
DWORD
|
|
DevfileIoctl(
|
|
IN HANDLE Handle,
|
|
IN DWORD Ioctl,
|
|
IN PVOID InBuf,
|
|
IN ULONG InBufSize,
|
|
IN OUT PVOID OutBuf,
|
|
IN DWORD OutBufSize,
|
|
OUT LPDWORD returnLength
|
|
)
|
|
{
|
|
NTSTATUS status;
|
|
IO_STATUS_BLOCK ioStatus;
|
|
|
|
status = NtDeviceIoControlFile(Handle,
|
|
(HANDLE) NULL,
|
|
(PIO_APC_ROUTINE) NULL,
|
|
NULL,
|
|
&ioStatus,
|
|
Ioctl,
|
|
InBuf, InBufSize,
|
|
OutBuf, OutBufSize);
|
|
if ( status == STATUS_PENDING ) {
|
|
status = NtWaitForSingleObject( Handle, FALSE, NULL );
|
|
}
|
|
|
|
if ( NT_SUCCESS(status) ) {
|
|
status = ioStatus.Status;
|
|
}
|
|
|
|
if ( ARGUMENT_PRESENT(returnLength) ) {
|
|
*returnLength = (DWORD)ioStatus.Information;
|
|
}
|
|
|
|
return status;
|
|
|
|
} // DevfileIoctl
|
|
|
|
|
|
|
|
DWORD
|
|
DisksAssignDosDevice(
|
|
PCHAR MountName,
|
|
PWCHAR VolumeDevName
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Inputs:
|
|
MountName -
|
|
VolumeDevName -
|
|
|
|
Return value:
|
|
|
|
A Win32 error code.
|
|
|
|
--*/
|
|
|
|
{
|
|
WCHAR mount_device[MAX_PATH];
|
|
USHORT mount_point_len;
|
|
USHORT dev_name_len;
|
|
HANDLE handle;
|
|
DWORD status;
|
|
USHORT inputlength;
|
|
PMOUNTMGR_CREATE_POINT_INPUT input;
|
|
|
|
status = DevfileOpen(&handle, MOUNTMGR_DEVICE_NAME);
|
|
if (status) {
|
|
return status;
|
|
}
|
|
|
|
swprintf(mount_device, L"\\DosDevices\\%S", MountName);
|
|
mount_point_len = wcslen(mount_device) * sizeof(WCHAR);
|
|
dev_name_len = wcslen(VolumeDevName) * sizeof(WCHAR);
|
|
inputlength = sizeof(MOUNTMGR_CREATE_POINT_INPUT) +
|
|
mount_point_len + dev_name_len;
|
|
|
|
input = (PMOUNTMGR_CREATE_POINT_INPUT)malloc(inputlength);
|
|
if (!input) {
|
|
DevfileClose(handle);
|
|
return ERROR_NOT_ENOUGH_MEMORY;
|
|
}
|
|
input->SymbolicLinkNameOffset = sizeof(MOUNTMGR_CREATE_POINT_INPUT);
|
|
input->SymbolicLinkNameLength = mount_point_len;
|
|
input->DeviceNameOffset = input->SymbolicLinkNameOffset +
|
|
input->SymbolicLinkNameLength;
|
|
input->DeviceNameLength = dev_name_len;
|
|
RtlCopyMemory((PCHAR)input + input->SymbolicLinkNameOffset,
|
|
mount_device, mount_point_len);
|
|
RtlCopyMemory((PCHAR)input + input->DeviceNameOffset,
|
|
VolumeDevName, dev_name_len);
|
|
status = DevfileIoctl(handle, IOCTL_MOUNTMGR_CREATE_POINT,
|
|
input, inputlength, NULL, 0, NULL);
|
|
free(input);
|
|
DevfileClose(handle);
|
|
return status;
|
|
|
|
} // DisksAssignDosDevice
|
|
|
|
|
|
|
|
DWORD
|
|
DisksRemoveDosDevice(
|
|
PCHAR MountName
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Inputs:
|
|
MountName -
|
|
|
|
Return value:
|
|
|
|
|
|
--*/
|
|
|
|
{
|
|
WCHAR mount_device[MAX_PATH];
|
|
USHORT mount_point_len;
|
|
USHORT dev_name_len;
|
|
HANDLE handle;
|
|
DWORD status;
|
|
USHORT inputlength;
|
|
PMOUNTMGR_MOUNT_POINT input;
|
|
|
|
UCHAR bogusBuffer[128];
|
|
|
|
status = DevfileOpen(&handle, MOUNTMGR_DEVICE_NAME);
|
|
if (status) {
|
|
return status;
|
|
}
|
|
|
|
swprintf(mount_device, L"\\DosDevices\\%S", MountName);
|
|
mount_point_len = wcslen(mount_device) * sizeof(WCHAR);
|
|
inputlength = sizeof(MOUNTMGR_MOUNT_POINT) + mount_point_len;
|
|
|
|
input = (PMOUNTMGR_MOUNT_POINT)malloc(inputlength);
|
|
if (!input) {
|
|
DevfileClose(handle);
|
|
return ERROR_NOT_ENOUGH_MEMORY;
|
|
}
|
|
input->UniqueIdOffset = 0;
|
|
input->UniqueIdLength = 0;
|
|
input->DeviceNameOffset = 0;
|
|
input->DeviceNameLength = 0;
|
|
input->SymbolicLinkNameOffset = sizeof(MOUNTMGR_MOUNT_POINT);
|
|
input->SymbolicLinkNameLength = mount_point_len;
|
|
RtlCopyMemory((PCHAR)input + input->SymbolicLinkNameOffset,
|
|
mount_device, mount_point_len);
|
|
status = DevfileIoctl(handle, IOCTL_MOUNTMGR_DELETE_POINTS,
|
|
input, inputlength, bogusBuffer, 128, NULL);
|
|
free(input);
|
|
DevfileClose(handle);
|
|
return status;
|
|
|
|
} // DisksRemoveDosDevice
|
|
|
|
|
|
|
|
|
|
DWORD
|
|
FindFirstVolumeForSignature(
|
|
IN HANDLE MountMgrHandle,
|
|
IN DWORD Signature,
|
|
OUT LPSTR VolumeName,
|
|
IN DWORD BufferLength,
|
|
OUT LPHANDLE Handle,
|
|
OUT PVOID UniqueId OPTIONAL,
|
|
IN OUT LPDWORD IdLength,
|
|
OUT PUCHAR DriveLetter OPTIONAL
|
|
)
|
|
|
|
/*++
|
|
|
|
Inputs:
|
|
|
|
MountMgrHandle - a handle to the mount manager.
|
|
|
|
Signature - the signature we are looking for.
|
|
|
|
VolumeName - must be a valid buffer of at least MAX_PATH characters.
|
|
|
|
BufferLength - the length of VolumeName.
|
|
|
|
Handle - pointer to receive the FindFirstVolume/FindNextVolume enum handle.
|
|
|
|
UniqueId - optional pointer to buffer to receive the UniqueId.
|
|
|
|
IdLength - pointer to length of the UniqueId buffer. Must be valid if
|
|
UniqueId is present.
|
|
|
|
DriveLetter - returns the drive letter if present.
|
|
|
|
Return Value:
|
|
|
|
Win32 error code
|
|
|
|
--*/
|
|
|
|
{
|
|
HANDLE handle;
|
|
BOOL success;
|
|
DWORD status;
|
|
LPDWORD idSignature;
|
|
DWORD bufLength;
|
|
LPWSTR wVolumeName;
|
|
DWORD inputlength;
|
|
DWORD outputlength;
|
|
DWORD returnlength;
|
|
UCHAR outputBuffer[OUTPUT_BUFFER_LEN];
|
|
PMOUNTMGR_MOUNT_POINT input;
|
|
PMOUNTMGR_MOUNT_POINTS output;
|
|
PUCHAR byteBuffer;
|
|
DWORD mountPoints;
|
|
|
|
if ( !ARGUMENT_PRESENT( VolumeName ) ) {
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
handle = FindFirstVolume( VolumeName, BufferLength );
|
|
|
|
if ( handle == INVALID_HANDLE_VALUE ) {
|
|
return(ERROR_FILE_NOT_FOUND);
|
|
}
|
|
|
|
do {
|
|
bufLength = strlen( VolumeName );
|
|
VolumeName[bufLength-1] = '\0';
|
|
if ( VolumeName[1] != '\\' ) {
|
|
status = ERROR_INVALID_NAME;
|
|
break;
|
|
} else {
|
|
VolumeName[1] = '?';
|
|
wVolumeName = malloc( bufLength * sizeof(WCHAR) );
|
|
if (!wVolumeName) {
|
|
status = ERROR_NOT_ENOUGH_MEMORY;
|
|
break;
|
|
}
|
|
|
|
mbstowcs( wVolumeName, VolumeName, bufLength );
|
|
bufLength--;
|
|
printf( "\nFound volume %ws\n", wVolumeName );
|
|
inputlength = sizeof(MOUNTMGR_MOUNT_POINT) +
|
|
(bufLength*sizeof(WCHAR)) + (2*sizeof(WCHAR));
|
|
|
|
input = (PMOUNTMGR_MOUNT_POINT)malloc(inputlength);
|
|
if (!input) {
|
|
free( wVolumeName );
|
|
status = ERROR_NOT_ENOUGH_MEMORY;
|
|
break;
|
|
}
|
|
|
|
input->SymbolicLinkNameOffset = 0;
|
|
input->SymbolicLinkNameLength = 0;
|
|
input->UniqueIdOffset = 0;
|
|
input->UniqueIdLength = 0;
|
|
input->DeviceNameOffset = sizeof(MOUNTMGR_MOUNT_POINT);
|
|
input->DeviceNameLength = (USHORT)(bufLength * sizeof(WCHAR));
|
|
RtlCopyMemory((PCHAR)input + input->DeviceNameOffset,
|
|
wVolumeName, bufLength * sizeof(WCHAR) );
|
|
outputlength = OUTPUT_BUFFER_LEN;
|
|
|
|
status = DevfileIoctl(MountMgrHandle, IOCTL_MOUNTMGR_QUERY_POINTS,
|
|
input, inputlength, outputBuffer, outputlength, &returnlength);
|
|
|
|
if ( status != ERROR_SUCCESS ) {
|
|
printf( "Query points for %ws failed, error %u\n",
|
|
input->DeviceNameOffset, status );
|
|
free( wVolumeName );
|
|
free(input);
|
|
wVolumeName = NULL;
|
|
input = NULL;
|
|
break;
|
|
} else {
|
|
output = (PMOUNTMGR_MOUNT_POINTS)outputBuffer;
|
|
mountPoints = output->NumberOfMountPoints;
|
|
if ( !mountPoints ) {
|
|
return ERROR_INVALID_DATA;
|
|
}
|
|
byteBuffer = outputBuffer + output->MountPoints[0].UniqueIdOffset;
|
|
idSignature = (LPDWORD)byteBuffer;
|
|
if ( !Signature ||
|
|
(Signature == *idSignature) ) {
|
|
NTSTATUS ntStatus;
|
|
UNICODE_STRING unicodeString;
|
|
OEM_STRING oemString;
|
|
DWORD count;
|
|
UCHAR driveLetter;
|
|
UCHAR devName[ MAX_PATH ];
|
|
PWCHAR wideBuffer;
|
|
LPDWORD dwordBuffer;
|
|
|
|
free( wVolumeName );
|
|
free(input);
|
|
input = NULL;
|
|
wVolumeName = NULL;
|
|
*Handle = handle;
|
|
if ( ARGUMENT_PRESENT(UniqueId) ) {
|
|
if ( *IdLength > output->MountPoints[0].UniqueIdLength ) {
|
|
*IdLength = output->MountPoints[0].UniqueIdLength;
|
|
}
|
|
RtlCopyMemory( UniqueId, byteBuffer, *IdLength );
|
|
}
|
|
|
|
//
|
|
// Print the ID
|
|
//
|
|
count = output->MountPoints[0].UniqueIdLength;
|
|
count = (count + 3) / 4;
|
|
dwordBuffer = (LPDWORD)(outputBuffer + output->MountPoints[0].UniqueIdOffset);
|
|
printf( "Id = " );
|
|
while ( count-- ) {
|
|
printf( "%08lx ", *(dwordBuffer++) );
|
|
}
|
|
printf( "\n" );
|
|
|
|
if ( ARGUMENT_PRESENT(DriveLetter) ) {
|
|
*DriveLetter = 0;
|
|
while ( mountPoints-- ) {
|
|
byteBuffer = outputBuffer +
|
|
output->MountPoints[mountPoints].SymbolicLinkNameOffset;
|
|
//
|
|
// Covert UNICODE name to OEM string upper case
|
|
//
|
|
unicodeString.Buffer = (PWCHAR)byteBuffer;
|
|
unicodeString.MaximumLength = output->MountPoints[mountPoints].SymbolicLinkNameLength + sizeof(WCHAR);
|
|
unicodeString.Length = output->MountPoints[mountPoints].SymbolicLinkNameLength;
|
|
oemString.Buffer = devName;
|
|
oemString.MaximumLength = sizeof(devName);
|
|
ntStatus = RtlUpcaseUnicodeStringToOemString(
|
|
&oemString,
|
|
&unicodeString,
|
|
FALSE );
|
|
if ( ntStatus != STATUS_SUCCESS ) {
|
|
status = RtlNtStatusToDosError( ntStatus );
|
|
return status;
|
|
}
|
|
devName[oemString.Length] = '\0';
|
|
count = sscanf( devName, "\\DOSDEVICES\\%c:", &driveLetter );
|
|
wideBuffer = (PWCHAR)byteBuffer;
|
|
wideBuffer[(output->MountPoints[mountPoints].SymbolicLinkNameLength)/2] = L'\0';
|
|
if ( count ) {
|
|
*DriveLetter = driveLetter;
|
|
printf( "Symbolic name = %ws, letter = %c:\\\n",
|
|
byteBuffer,
|
|
driveLetter );
|
|
if ( Signature ) {
|
|
break;
|
|
}
|
|
} else {
|
|
printf( "Symbolic name = %ws\n",
|
|
byteBuffer );
|
|
}
|
|
}
|
|
}
|
|
if ( Signature ) {
|
|
return ERROR_SUCCESS;
|
|
}
|
|
}
|
|
}
|
|
|
|
free(wVolumeName);
|
|
free(input);
|
|
}
|
|
|
|
success = FindNextVolume( handle,
|
|
VolumeName,
|
|
BufferLength );
|
|
if ( !success ) {
|
|
status = GetLastError();
|
|
}
|
|
|
|
} while ( status == ERROR_SUCCESS );
|
|
|
|
FindVolumeClose( handle );
|
|
return status;
|
|
|
|
} // FindFirstVolumeForSignature
|
|
|
|
|
|
|
|
DWORD
|
|
FindNextVolumeForSignature(
|
|
IN HANDLE MountMgrHandle,
|
|
IN DWORD Signature,
|
|
IN HANDLE Handle,
|
|
OUT LPSTR VolumeName,
|
|
IN DWORD BufferLength,
|
|
OUT PVOID UniqueId OPTIONAL,
|
|
IN OUT LPDWORD IdLength,
|
|
OUT PUCHAR DriveLetter OPTIONAL
|
|
)
|
|
|
|
/*++
|
|
|
|
Inputs:
|
|
|
|
MountMgrHandle - a handle to the mount manager.
|
|
|
|
Signature - the signature we are looking for.
|
|
|
|
Handle - the FindFirstVolume/FindNextVolume enum handle.
|
|
|
|
VolumeName - must be a valid buffer of at least MAX_PATH characters.
|
|
|
|
BufferLength - the length of VolumeName.
|
|
|
|
UniqueId - optional pointer to buffer to receive the UniqueId.
|
|
|
|
IdLength - point to the length of the UniqueId buffer.
|
|
|
|
DriveLetter - returns the drive letter if present.
|
|
|
|
Return Value:
|
|
|
|
Win32 error code
|
|
|
|
--*/
|
|
|
|
{
|
|
BOOL success;
|
|
DWORD status;
|
|
LPDWORD idSignature;
|
|
DWORD bufLength;
|
|
LPWSTR wVolumeName;
|
|
DWORD inputlength;
|
|
DWORD outputlength;
|
|
DWORD returnlength;
|
|
UCHAR outputBuffer[OUTPUT_BUFFER_LEN];
|
|
PMOUNTMGR_MOUNT_POINT input;
|
|
PMOUNTMGR_MOUNT_POINTS output;
|
|
PUCHAR byteBuffer;
|
|
DWORD mountPoints;
|
|
|
|
|
|
if ( !ARGUMENT_PRESENT( VolumeName ) ) {
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
do {
|
|
success = FindNextVolume( Handle, VolumeName, BufferLength );
|
|
|
|
if ( !success ) {
|
|
status = GetLastError();
|
|
break;
|
|
}
|
|
|
|
bufLength = strlen( VolumeName );
|
|
|
|
VolumeName[bufLength-1] = '\0';
|
|
if ( VolumeName[1] != '\\' ) {
|
|
status = ERROR_INVALID_NAME;
|
|
break;
|
|
} else {
|
|
VolumeName[1] = '?';
|
|
bufLength--;
|
|
wVolumeName = malloc( bufLength * sizeof(WCHAR) );
|
|
if (!wVolumeName) {
|
|
status = ERROR_NOT_ENOUGH_MEMORY;
|
|
break;
|
|
}
|
|
|
|
mbstowcs( wVolumeName, VolumeName, bufLength );
|
|
inputlength = sizeof(MOUNTMGR_MOUNT_POINT) +
|
|
(bufLength*sizeof(WCHAR)) + (2*sizeof(WCHAR));
|
|
|
|
input = (PMOUNTMGR_MOUNT_POINT)malloc(inputlength);
|
|
if (!input) {
|
|
free( wVolumeName );
|
|
status = ERROR_NOT_ENOUGH_MEMORY;
|
|
break;
|
|
}
|
|
|
|
input->SymbolicLinkNameOffset = 0;
|
|
input->SymbolicLinkNameLength = 0;
|
|
input->UniqueIdOffset = 0;
|
|
input->UniqueIdLength = 0;
|
|
input->DeviceNameOffset = sizeof(MOUNTMGR_MOUNT_POINT);
|
|
input->DeviceNameLength = (USHORT)(bufLength * sizeof(WCHAR));
|
|
RtlCopyMemory((PCHAR)input + input->DeviceNameOffset,
|
|
wVolumeName, bufLength * sizeof(WCHAR) );
|
|
outputlength = OUTPUT_BUFFER_LEN;
|
|
|
|
status = DevfileIoctl(MountMgrHandle, IOCTL_MOUNTMGR_QUERY_POINTS,
|
|
input, inputlength, outputBuffer, outputlength, &returnlength);
|
|
|
|
if ( status != ERROR_SUCCESS ) {
|
|
free( wVolumeName );
|
|
free(input);
|
|
break;
|
|
} else {
|
|
output = (PMOUNTMGR_MOUNT_POINTS)outputBuffer;
|
|
mountPoints = output->NumberOfMountPoints;
|
|
if ( !mountPoints ) {
|
|
return ERROR_INVALID_DATA;
|
|
}
|
|
byteBuffer = outputBuffer + output->MountPoints[0].UniqueIdOffset;
|
|
idSignature = (LPDWORD)byteBuffer;
|
|
if ( Signature == *idSignature ) {
|
|
NTSTATUS ntStatus;
|
|
UNICODE_STRING unicodeString;
|
|
OEM_STRING oemString;
|
|
DWORD count;
|
|
UCHAR driveLetter;
|
|
UCHAR devName[ MAX_PATH ];
|
|
|
|
free( wVolumeName );
|
|
free(input);
|
|
if ( ARGUMENT_PRESENT(UniqueId) ) {
|
|
if ( *IdLength > output->MountPoints[0].UniqueIdLength ) {
|
|
*IdLength = output->MountPoints[0].UniqueIdLength;
|
|
}
|
|
RtlCopyMemory( UniqueId, byteBuffer, *IdLength );
|
|
}
|
|
|
|
if ( ARGUMENT_PRESENT(DriveLetter) ) {
|
|
*DriveLetter = 0;
|
|
while ( mountPoints-- ) {
|
|
byteBuffer = outputBuffer +
|
|
output->MountPoints[mountPoints].SymbolicLinkNameOffset;
|
|
//
|
|
// Covert UNICODE name to OEM string upper case
|
|
//
|
|
unicodeString.Buffer = (PWCHAR)byteBuffer;
|
|
unicodeString.MaximumLength = output->MountPoints[mountPoints].SymbolicLinkNameLength + sizeof(WCHAR);
|
|
unicodeString.Length = output->MountPoints[mountPoints].SymbolicLinkNameLength;
|
|
oemString.Buffer = devName;
|
|
oemString.MaximumLength = sizeof(devName);
|
|
ntStatus = RtlUpcaseUnicodeStringToOemString(
|
|
&oemString,
|
|
&unicodeString,
|
|
FALSE );
|
|
if ( ntStatus != STATUS_SUCCESS ) {
|
|
status = RtlNtStatusToDosError( ntStatus );
|
|
return status;
|
|
}
|
|
devName[oemString.Length] = '\0';
|
|
count = sscanf( devName, "\\DOSDEVICES\\%c:", &driveLetter );
|
|
if ( count ) {
|
|
*DriveLetter = driveLetter;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
return ERROR_SUCCESS;
|
|
}
|
|
}
|
|
|
|
free(wVolumeName);
|
|
free(input);
|
|
}
|
|
|
|
success = FindNextVolume( Handle,
|
|
VolumeName,
|
|
BufferLength );
|
|
if ( !success ) {
|
|
status = GetLastError();
|
|
}
|
|
|
|
} while ( status == ERROR_SUCCESS );
|
|
|
|
return status;
|
|
|
|
} // FindNextVolumeForSignature
|
|
|
|
|
|
#if 0
|
|
|
|
DWORD
|
|
DisksSetDiskInfo(
|
|
IN HKEY RegistryKey,
|
|
IN DWORD Signature
|
|
)
|
|
|
|
/*++
|
|
|
|
Inputs:
|
|
|
|
Return Value:
|
|
|
|
A Win32 error code.
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD status;
|
|
UCHAR driveLetter;
|
|
UCHAR volumeName[MAX_PATH];
|
|
HANDLE handle;
|
|
HANDLE mHandle;
|
|
UCHAR uniqueId[MAX_PATH];
|
|
UCHAR smashedId[MAX_PATH+1];
|
|
DWORD idLength;
|
|
DWORD i;
|
|
WCHAR indexName[16];
|
|
HKEY registryKey;
|
|
DWORD disposition;
|
|
|
|
status = DevfileOpen( &mHandle, MOUNTMGR_DEVICE_NAME );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
printf( "SetDiskInfo: DevfileOpen failed, status = %u\n", status);
|
|
return status;
|
|
}
|
|
|
|
status = ClusterRegDeleteKey( RegistryKey, L"MountMgr" );
|
|
if ( (status != ERROR_SUCCESS) && (status != ERROR_FILE_NOT_FOUND) ) {
|
|
DevfileClose( mHandle );
|
|
printf( "DiskInfo: ClusterRegDeleteKey failed, status = %1!u!\n", status);
|
|
return status;
|
|
}
|
|
|
|
status = ClusterRegCreateKey( RegistryKey,
|
|
L"MountMgr",
|
|
0,
|
|
KEY_READ | KEY_WRITE,
|
|
NULL,
|
|
®istryKey,
|
|
&disposition );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
DevfileClose( mHandle );
|
|
(DiskpLogEvent)(
|
|
ResourceHandle,
|
|
LOG_ERROR,
|
|
L"SetDiskInfo: ClusterRegCreateKey failed, status = %1!u!\n", status);
|
|
return status;
|
|
}
|
|
|
|
idLength = MAX_PATH;
|
|
status = FindFirstVolumeForSignature( ResourceHandle,
|
|
mHandle,
|
|
Signature,
|
|
volumeName,
|
|
MAX_PATH,
|
|
&handle,
|
|
uniqueId,
|
|
&idLength,
|
|
&driveLetter );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
DevfileClose( mHandle );
|
|
ClusterRegCloseKey( registryKey );
|
|
(DiskpLogEvent)(
|
|
ResourceHandle,
|
|
LOG_ERROR,
|
|
L"SetDiskInfo: FindFirstVolume failed, status = %1!u!\n", status);
|
|
return status;
|
|
}
|
|
|
|
i = 0;
|
|
while ( status == ERROR_SUCCESS ) {
|
|
wsprintfW( indexName, L"%0.5u", i++ );
|
|
|
|
smashedId[0] = driveLetter;
|
|
RtlCopyMemory( &smashedId[1], uniqueId, idLength );
|
|
status = ClusterRegSetValue( registryKey,
|
|
indexName,
|
|
REG_BINARY,
|
|
(CONST BYTE *)smashedId,
|
|
idLength + 1);
|
|
if ( status != ERROR_SUCCESS ) {
|
|
//printf("DiskSetDiskInfo, error setting value %s\n", indexName );
|
|
}
|
|
|
|
idLength = MAX_PATH;
|
|
status = FindNextVolumeForSignature( mHandle,
|
|
Signature,
|
|
handle,
|
|
volumeName,
|
|
MAX_PATH,
|
|
uniqueId,
|
|
&idLength,
|
|
&driveLetter );
|
|
}
|
|
|
|
FindVolumeClose( handle );
|
|
DevfileClose( mHandle );
|
|
ClusterRegCloseKey( registryKey );
|
|
|
|
return ERROR_SUCCESS;
|
|
|
|
} // DisksSetDiskInfo
|
|
|
|
|
|
|
|
DWORD
|
|
DisksSetMountMgr(
|
|
IN HKEY RegistryKey,
|
|
IN DWORD Signature
|
|
)
|
|
|
|
/*++
|
|
|
|
Inputs:
|
|
|
|
Return Value:
|
|
|
|
A Win32 error code.
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD status;
|
|
UCHAR volumeName[MAX_PATH];
|
|
LPWSTR wVolumeName;
|
|
HANDLE mHandle;
|
|
HANDLE handle = NULL;
|
|
UCHAR storedId[MAX_PATH+1];
|
|
DWORD storedIdSize;
|
|
DWORD i;
|
|
WCHAR indexName[16];
|
|
HKEY registryKey;
|
|
DWORD type;
|
|
DWORD bufLength;
|
|
UCHAR driveLetter[4];
|
|
NTSTATUS ntStatus;
|
|
|
|
status = DevfileOpen( &mHandle, MOUNTMGR_DEVICE_NAME );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
(DiskpLogEvent)(
|
|
ResourceHandle,
|
|
LOG_ERROR,
|
|
L"SetMountMgr: DevfileOpen failed, status = %1!u!\n", status);
|
|
return status;
|
|
}
|
|
|
|
status = ClusterRegOpenKey( RegistryKey,
|
|
L"MountMgr",
|
|
KEY_READ | KEY_WRITE,
|
|
®istryKey );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
DevfileClose( mHandle );
|
|
return status;
|
|
(DiskpLogEvent)(
|
|
ResourceHandle,
|
|
LOG_ERROR,
|
|
L"SetMountMgr: ClusterRegOpenKey failed, status = %1!u!\n", status);
|
|
}
|
|
|
|
i = 0;
|
|
do {
|
|
wsprintfW( indexName, L"%0.5u", i++ );
|
|
storedIdSize = MAX_PATH;
|
|
status = ClusterRegQueryValue( registryKey,
|
|
indexName,
|
|
&type,
|
|
(PUCHAR)storedId,
|
|
&storedIdSize);
|
|
|
|
(DiskpLogEvent)(
|
|
ResourceHandle,
|
|
LOG_ERROR,
|
|
L"SetMountMgr: ClusterRegQueryValue returned status = %1!u!\n", status);
|
|
if ( status != ERROR_SUCCESS ) {
|
|
break;
|
|
}
|
|
|
|
storedId[1] = ':';
|
|
storedId[2] = '\0';
|
|
ntStatus = DisksRemoveDosDevice( storedId );
|
|
status = RtlNtStatusToDosError( ntStatus );
|
|
(DiskpLogEvent)(
|
|
ResourceHandle,
|
|
LOG_ERROR,
|
|
L"SetMountMgr: RemoveDosDevice for %1!x! returned status = %2!u!\n", *storedId, status);
|
|
if ( status == ERROR_FILE_NOT_FOUND ) {
|
|
status = ERROR_SUCCESS;
|
|
}
|
|
|
|
} while ( status == ERROR_SUCCESS );
|
|
|
|
status = FindFirstVolumeForSignature( ResourceHandle,
|
|
mHandle,
|
|
Signature,
|
|
volumeName,
|
|
MAX_PATH,
|
|
&handle,
|
|
NULL,
|
|
NULL,
|
|
&driveLetter[0] );
|
|
|
|
if ( status != ERROR_SUCCESS ) {
|
|
(DiskpLogEvent)(
|
|
ResourceHandle,
|
|
LOG_ERROR,
|
|
L"SetMountMgr: FindFirstVolume failed for Signature %1!08lx!, status = %2!u!\n", Signature, status);
|
|
}
|
|
|
|
i = 0;
|
|
while ( status == ERROR_SUCCESS ) {
|
|
wsprintfW( indexName, L"%0.5u", i++ );
|
|
storedIdSize = MAX_PATH;
|
|
status = ClusterRegQueryValue( registryKey,
|
|
indexName,
|
|
&type,
|
|
(PUCHAR)storedId,
|
|
&storedIdSize );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
break;
|
|
}
|
|
|
|
//
|
|
// Remove current drive letter
|
|
//
|
|
driveLetter[1] = ':';
|
|
driveLetter[2] = '\0';
|
|
ntStatus = DisksRemoveDosDevice( driveLetter );
|
|
status = RtlNtStatusToDosError( ntStatus );
|
|
(DiskpLogEvent)(
|
|
ResourceHandle,
|
|
LOG_ERROR,
|
|
L"SetMountMgr: RemoveDosDevice for %1!x! returned status = %2!u!\n", driveLetter[0], status);
|
|
|
|
bufLength = strlen( volumeName );
|
|
wVolumeName = malloc( (bufLength + 1) * sizeof(WCHAR) );
|
|
if (!wVolumeName) {
|
|
status = ERROR_NOT_ENOUGH_MEMORY;
|
|
break;
|
|
}
|
|
|
|
mbstowcs( wVolumeName, volumeName, bufLength + 1 );
|
|
|
|
storedId[1] = ':';
|
|
storedId[2] = '\0';
|
|
status = DisksAssignDosDevice( storedId, wVolumeName );
|
|
(DiskpLogEvent)(
|
|
ResourceHandle,
|
|
LOG_ERROR,
|
|
L"SetMountMgr: AssignDosDevice for %1!x! (%2!ws!) returned status = %3!u!\n", *storedId, wVolumeName, status);
|
|
free( wVolumeName );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
break;
|
|
}
|
|
|
|
status = FindNextVolumeForSignature( mHandle,
|
|
Signature,
|
|
handle,
|
|
volumeName,
|
|
MAX_PATH,
|
|
NULL,
|
|
NULL,
|
|
&driveLetter[0] );
|
|
|
|
if ( status != ERROR_SUCCESS ) {
|
|
(DiskpLogEvent)(
|
|
ResourceHandle,
|
|
LOG_ERROR,
|
|
L"SetMountMgr: FindNextVolume failed, status = %1!u!\n", status);
|
|
}
|
|
if ( status == ERROR_NO_MORE_FILES ) {
|
|
status = ERROR_SUCCESS;
|
|
break;
|
|
}
|
|
|
|
}
|
|
|
|
DevfileClose( mHandle );
|
|
ClusterRegCloseKey( registryKey );
|
|
if ( handle ) {
|
|
FindVolumeClose( handle );
|
|
}
|
|
|
|
return status;
|
|
|
|
} // DisksSetMountMgr
|
|
|
|
|
|
|
|
BOOL
|
|
DisksDoesDiskInfoMatch(
|
|
IN HKEY RegistryKey,
|
|
IN DWORD Signature
|
|
)
|
|
|
|
/*++
|
|
|
|
Inputs:
|
|
|
|
Return Value:
|
|
|
|
A Win32 error code.
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD status;
|
|
UCHAR driveLetter;
|
|
UCHAR volumeName[MAX_PATH];
|
|
HANDLE handle;
|
|
HANDLE mHandle;
|
|
UCHAR uniqueId[MAX_PATH];
|
|
UCHAR smashedId[MAX_PATH+1];
|
|
UCHAR storedId[MAX_PATH+1];
|
|
DWORD idLength;
|
|
DWORD storedIdSize;
|
|
DWORD i;
|
|
WCHAR indexName[16];
|
|
HKEY registryKey;
|
|
DWORD type;
|
|
|
|
|
|
status = DevfileOpen( &mHandle, MOUNTMGR_DEVICE_NAME );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
return FALSE;
|
|
}
|
|
|
|
status = ClusterRegOpenKey( RegistryKey,
|
|
L"MountMgr",
|
|
KEY_READ | KEY_WRITE,
|
|
®istryKey );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
DevfileClose( mHandle );
|
|
return FALSE;
|
|
}
|
|
|
|
idLength = MAX_PATH;
|
|
status = FindFirstVolumeForSignature( ResourceHandle,
|
|
mHandle,
|
|
Signature,
|
|
volumeName,
|
|
MAX_PATH,
|
|
&handle,
|
|
uniqueId,
|
|
&idLength,
|
|
&driveLetter );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
DevfileClose( mHandle );
|
|
ClusterRegCloseKey( registryKey );
|
|
return FALSE;
|
|
}
|
|
|
|
i = 0;
|
|
while ( status == ERROR_SUCCESS ) {
|
|
wsprintfW( indexName, L"%0.5u", i++ );
|
|
|
|
smashedId[0] = driveLetter;
|
|
RtlCopyMemory( &smashedId[1], uniqueId, idLength );
|
|
storedIdSize = MAX_PATH;
|
|
status = ClusterRegQueryValue( registryKey,
|
|
indexName,
|
|
&type,
|
|
(PUCHAR)storedId,
|
|
&storedIdSize);
|
|
|
|
if ( (status != ERROR_SUCCESS) ||
|
|
(type != REG_BINARY) ||
|
|
(storedIdSize != (idLength+1)) ||
|
|
(RtlCompareMemory( smashedId, storedId, idLength ) != idLength) ) {
|
|
FindVolumeClose( handle );
|
|
DevfileClose( mHandle );
|
|
ClusterRegCloseKey( registryKey );
|
|
return FALSE;
|
|
}
|
|
|
|
idLength = MAX_PATH;
|
|
status = FindNextVolumeForSignature( mHandle,
|
|
Signature,
|
|
handle,
|
|
volumeName,
|
|
MAX_PATH,
|
|
uniqueId,
|
|
&idLength,
|
|
&driveLetter );
|
|
}
|
|
|
|
FindVolumeClose( handle );
|
|
DevfileClose( mHandle );
|
|
ClusterRegCloseKey( registryKey );
|
|
|
|
if ( status != ERROR_NO_MORE_FILES ) {
|
|
return FALSE;
|
|
}
|
|
|
|
return TRUE;
|
|
|
|
} // DisksDoesDiskInfoMatch
|
|
|
|
|
|
|
|
BOOL
|
|
DisksIsDiskInfoValid(
|
|
IN HKEY RegistryKey,
|
|
IN DWORD Signature
|
|
)
|
|
|
|
/*++
|
|
|
|
Inputs:
|
|
|
|
Return Value:
|
|
|
|
A Win32 error code.
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD status;
|
|
UCHAR volumeName[MAX_PATH];
|
|
UCHAR storedId[MAX_PATH+1];
|
|
DWORD storedIdSize;
|
|
WCHAR indexName[16];
|
|
HKEY registryKey;
|
|
DWORD i;
|
|
DWORD type;
|
|
HANDLE handle;
|
|
HANDLE mHandle;
|
|
|
|
|
|
status = DevfileOpen( &mHandle, MOUNTMGR_DEVICE_NAME );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
return FALSE;
|
|
}
|
|
|
|
status = ClusterRegOpenKey( RegistryKey,
|
|
L"MountMgr",
|
|
KEY_READ | KEY_WRITE,
|
|
®istryKey );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
DevfileClose( mHandle );
|
|
return FALSE;
|
|
}
|
|
|
|
status = FindFirstVolumeForSignature( ResourceHandle,
|
|
mHandle,
|
|
Signature,
|
|
volumeName,
|
|
MAX_PATH,
|
|
&handle,
|
|
NULL,
|
|
NULL,
|
|
NULL );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
DevfileClose( mHandle );
|
|
ClusterRegCloseKey( registryKey );
|
|
return TRUE;
|
|
}
|
|
|
|
i = 0;
|
|
while ( status == ERROR_SUCCESS ) {
|
|
wsprintfW( indexName, L"%0.5u", i++ );
|
|
|
|
storedIdSize = MAX_PATH;
|
|
status = ClusterRegQueryValue( registryKey,
|
|
indexName,
|
|
&type,
|
|
(PUCHAR)storedId,
|
|
&storedIdSize);
|
|
if ( (status != ERROR_SUCCESS) ||
|
|
(type != REG_BINARY) ) {
|
|
FindVolumeClose( handle );
|
|
DevfileClose( mHandle );
|
|
ClusterRegCloseKey( registryKey );
|
|
if ( status == ERROR_FILE_NOT_FOUND ) {
|
|
return TRUE;
|
|
} else {
|
|
return FALSE;
|
|
}
|
|
}
|
|
|
|
status = FindNextVolumeForSignature( mHandle,
|
|
Signature,
|
|
handle,
|
|
volumeName,
|
|
MAX_PATH,
|
|
NULL,
|
|
NULL,
|
|
NULL );
|
|
}
|
|
|
|
FindVolumeClose( handle );
|
|
DevfileClose( mHandle );
|
|
ClusterRegCloseKey( registryKey );
|
|
|
|
return TRUE;
|
|
|
|
} // DisksIsDiskInfoValid
|
|
|
|
#endif
|
|
|