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.
3839 lines
102 KiB
3839 lines
102 KiB
/*++
|
|
|
|
Copyright (c) 1995-2001 Microsoft Corporation
|
|
|
|
Module Name:
|
|
|
|
disks.c
|
|
|
|
Abstract:
|
|
|
|
Resource DLL for disks.
|
|
|
|
Author:
|
|
|
|
John Vert (jvert) 5-Dec-1995
|
|
Rod Gamache (rodga) 18-Dec-1995
|
|
|
|
Revision History:
|
|
|
|
--*/
|
|
|
|
#include "disksp.h"
|
|
#include "lm.h"
|
|
|
|
#include "arbitrat.h"
|
|
#include "newdisks.h"
|
|
#include "newmount.h"
|
|
|
|
#define UNICODE 1
|
|
|
|
#define LOG_CURRENT_MODULE LOG_MODULE_DISK
|
|
|
|
|
|
#define DISKS_REG_CLUSTER_QUORUM "Cluster\\Quorum"
|
|
#define DISKS_REG_QUORUM_PATH CLUSREG_NAME_QUORUM_PATH
|
|
|
|
#define MAX_HANDLES 10
|
|
|
|
extern PWCHAR GLOBALROOT_HARDDISK_PARTITION_FMT; // L"\\\\\?\\GLOBALROOT\\Device\\Harddisk%u\\Partition%u\\";
|
|
|
|
PDISK_RESOURCE DiskTable[MAX_DISKS] = {0};
|
|
LONG DiskCount = 0;
|
|
|
|
CRITICAL_SECTION DisksLock;
|
|
|
|
LIST_ENTRY DisksListHead;
|
|
|
|
HANDLE DisksRegistryThread = NULL;
|
|
HANDLE DisksTerminateEvent = NULL;
|
|
HANDLE DiskspClusDiskZero = NULL;
|
|
|
|
DWORD SystemDiskAddressFound = 0;
|
|
PSCSI_ADDRESS_ENTRY SysDiskAddrList = NULL;
|
|
|
|
|
|
extern CLRES_FUNCTION_TABLE DisksFunctionTable;
|
|
|
|
extern RTL_RESOURCE PnpVolumeLock;
|
|
extern RTL_RESOURCE PnpWaitingListLock;
|
|
|
|
//
|
|
// Disk resource property names
|
|
//
|
|
#define DISKS_SIGNATURE CLUSREG_NAME_PHYSDISK_SIGNATURE
|
|
#define DISKS_DRIVE CLUSREG_NAME_PHYSDISK_DRIVE // pseudonym for signature
|
|
#define DISKS_SKIPCHKDSK CLUSREG_NAME_PHYSDISK_SKIPCHKDSK
|
|
#define DISKS_CONDITIONAL_MOUNT CLUSREG_NAME_PHYSDISK_CONDITIONAL_MOUNT
|
|
#define DISKS_MPVOLGUIDS CLUSREG_NAME_PHYSDISK_MPVOLGUIDS
|
|
#define DISKS_VOLGUID CLUSREG_NAME_PHYSDISK_VOLGUID // Not saved in cluster DB
|
|
#define DISKS_SERIALNUMBER CLUSREG_NAME_PHYSDISK_SERIALNUMBER
|
|
|
|
//
|
|
// Disk resource private read-write properties.
|
|
// Allow for a pseudonym for Signature (Drive), but don't allow both
|
|
// drive and signature to be passed.
|
|
//
|
|
RESUTIL_PROPERTY_ITEM
|
|
DiskResourcePrivateProperties[] = {
|
|
{ DISKS_SIGNATURE, NULL, CLUSPROP_FORMAT_DWORD, 0, 0, 0xFFFFFFFF, RESUTIL_PROPITEM_REQUIRED, FIELD_OFFSET(DISK_PARAMS,Signature) },
|
|
{ DISKS_SKIPCHKDSK, NULL, CLUSPROP_FORMAT_DWORD, 0, 0, 1, 0, FIELD_OFFSET(DISK_PARAMS,SkipChkdsk) },
|
|
{ DISKS_CONDITIONAL_MOUNT, NULL, CLUSPROP_FORMAT_DWORD, 1, 0, 1, 0, FIELD_OFFSET(DISK_PARAMS,ConditionalMount) },
|
|
#if USEMOUNTPOINTS_KEY
|
|
{ DISKS_USEMOUNTPOINTS, NULL, CLUSPROP_FORMAT_DWORD, 1, 0, 1, 0, FIELD_OFFSET(DISK_PARAMS, UseMountPoints) },
|
|
#endif
|
|
{ DISKS_MPVOLGUIDS, NULL, CLUSPROP_FORMAT_MULTI_SZ, 0, 0, 0, 0, FIELD_OFFSET(DISK_PARAMS, MPVolGuids) },
|
|
{ DISKS_SERIALNUMBER, NULL, CLUSPROP_FORMAT_SZ, 0, 0, 0, 0, FIELD_OFFSET(DISK_PARAMS,SerialNumber) },
|
|
{ 0 }
|
|
};
|
|
|
|
RESUTIL_PROPERTY_ITEM
|
|
DiskResourcePrivatePropertiesAlt[] = {
|
|
{ DISKS_SIGNATURE, NULL, CLUSPROP_FORMAT_DWORD, 0, 0, 0xFFFFFFFF, RESUTIL_PROPITEM_REQUIRED, FIELD_OFFSET(DISK_PARAMS,Signature) },
|
|
{ DISKS_SKIPCHKDSK, NULL, CLUSPROP_FORMAT_DWORD, 0, 0, 1, 0, FIELD_OFFSET(DISK_PARAMS,SkipChkdsk) },
|
|
{ DISKS_DRIVE, NULL, CLUSPROP_FORMAT_SZ, 0, 0, 0, RESUTIL_PROPITEM_REQUIRED, FIELD_OFFSET(DISK_PARAMS,Drive) },
|
|
{ DISKS_CONDITIONAL_MOUNT, NULL, CLUSPROP_FORMAT_DWORD, 1, 0, 1, 0, FIELD_OFFSET(DISK_PARAMS,ConditionalMount) },
|
|
#if USEMOUNTPOINTS_KEY
|
|
{ DISKS_USEMOUNTPOINTS, NULL, CLUSPROP_FORMAT_DWORD, 1, 0, 1, 0, FIELD_OFFSET(DISK_PARAMS, UseMountPoints) },
|
|
#endif
|
|
{ DISKS_MPVOLGUIDS, NULL, CLUSPROP_FORMAT_MULTI_SZ, 0, 0, 0, 0, FIELD_OFFSET(DISK_PARAMS, MPVolGuids) },
|
|
{ DISKS_VOLGUID, NULL, CLUSPROP_FORMAT_SZ, 0, 0, 0, 0, FIELD_OFFSET(DISK_PARAMS, VolGuid) },
|
|
{ DISKS_SERIALNUMBER, NULL, CLUSPROP_FORMAT_SZ, 0, 0, 0, 0, FIELD_OFFSET(DISK_PARAMS,SerialNumber) },
|
|
{ 0 }
|
|
};
|
|
|
|
#define CLUSTERLOG_ENV_VARIABLE L"ClusterLog"
|
|
|
|
//
|
|
// Local functions.
|
|
//
|
|
|
|
DWORD
|
|
DisksValidatePrivateResProperties(
|
|
IN OUT PDISK_RESOURCE ResourceEntry,
|
|
IN PVOID InBuffer,
|
|
IN DWORD InBufferSize,
|
|
OUT PDISK_PARAMS Params
|
|
);
|
|
|
|
DWORD
|
|
DisksSetPrivateResProperties(
|
|
IN OUT PDISK_RESOURCE ResourceEntry,
|
|
IN PVOID InBuffer,
|
|
IN DWORD InBufferSize
|
|
);
|
|
|
|
DWORD
|
|
DisksGetPrivateResProperties(
|
|
IN OUT PDISK_RESOURCE ResourceEntry,
|
|
OUT PVOID OutBuffer,
|
|
IN DWORD OutBufferSize,
|
|
OUT LPDWORD BytesReturned
|
|
);
|
|
|
|
DWORD
|
|
ProcessDllExtension(
|
|
IN PDISK_RESOURCE ResourceEntry,
|
|
IN PVOID InBuffer,
|
|
IN DWORD InBufferSize,
|
|
OUT PVOID OutBuffer,
|
|
IN DWORD OutBufferSize,
|
|
OUT LPDWORD BytesReturned
|
|
);
|
|
|
|
DWORD
|
|
DisksOpenChkdskLogFile(
|
|
IN PDISK_RESOURCE ResourceEntry,
|
|
IN OUT PHANDLE ChkdskLogFile,
|
|
IN OUT LPWSTR *ChkdskLogFileName
|
|
);
|
|
|
|
//
|
|
// Error callbacks from disk library
|
|
//
|
|
|
|
VOID
|
|
DiskErrorFatal(
|
|
INT MessageId,
|
|
DWORD Error,
|
|
LPSTR File,
|
|
DWORD Line
|
|
)
|
|
{
|
|
DWORD Data[3];
|
|
|
|
Data[0] = MessageId;
|
|
Data[1] = Error;
|
|
Data[2] = Line;
|
|
ClusResLogSystemEventData(LOG_CRITICAL,
|
|
RES_FTSET_DISK_ERROR,
|
|
sizeof(Data),
|
|
Data);
|
|
}
|
|
|
|
VOID
|
|
DiskErrorLogInfo(
|
|
LPSTR Format,
|
|
...
|
|
)
|
|
{
|
|
}
|
|
|
|
|
|
|
|
BOOLEAN
|
|
WINAPI
|
|
DisksDllEntryPoint(
|
|
IN HINSTANCE DllHandle,
|
|
IN DWORD Reason,
|
|
IN LPVOID Reserved
|
|
)
|
|
{
|
|
DWORD status;
|
|
|
|
switch ( Reason ) {
|
|
|
|
case DLL_PROCESS_ATTACH:
|
|
InitializeCriticalSection( &DisksLock );
|
|
InitializeListHead( &DisksListHead );
|
|
RtlInitializeResource( &PnpVolumeLock );
|
|
RtlInitializeResource( &PnpWaitingListLock );
|
|
ArbitrationInitialize();
|
|
GetSystemBusInfo();
|
|
|
|
break;
|
|
|
|
case DLL_PROCESS_DETACH:
|
|
//
|
|
// only do clean up if we're not exiting the process.
|
|
// ClRtlDestroyWorkQueue waits on an event to be set and it is
|
|
// possible at this point that there are no threads to do that.
|
|
// This causes resmon to linger and generally be a pest.
|
|
//
|
|
if (DiskspClusDiskZero) {
|
|
DevfileClose(DiskspClusDiskZero);
|
|
}
|
|
CleanupSystemBusInfo();
|
|
ArbitrationCleanup();
|
|
DeleteCriticalSection( &DisksLock );
|
|
RtlDeleteResource( &PnpVolumeLock );
|
|
RtlDeleteResource( &PnpWaitingListLock );
|
|
break;
|
|
|
|
default:
|
|
break;
|
|
}
|
|
|
|
return(TRUE);
|
|
|
|
} // DisksDllEntryPoint
|
|
|
|
|
|
|
|
|
|
RESID
|
|
DiskspGetNextDisk(
|
|
VOID
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
This routine will find the next disk slot entry.
|
|
|
|
Arguments:
|
|
|
|
None.
|
|
|
|
Return Value:
|
|
|
|
A valid resource ID if one is found.
|
|
|
|
-1 if no free entry is found.
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD index;
|
|
|
|
for ( index = 1; index < MAX_DISKS; index++ ) {
|
|
if ( DiskTable[index] == NULL ) {
|
|
return((RESID)( UlongToPtr(index) ) );
|
|
}
|
|
}
|
|
|
|
return((RESID)-1);
|
|
|
|
} // DiskspGetNextDisk
|
|
|
|
|
|
|
|
VOID
|
|
WINAPI
|
|
DisksTerminate(
|
|
IN RESID Resource
|
|
)
|
|
{
|
|
ULONG uResId=PtrToUlong(Resource);
|
|
|
|
PDISK_RESOURCE resourceEntry = DiskTable[uResId];
|
|
if ( resourceEntry == NULL ) {
|
|
DISKS_PRINT("Terminate, bad resource id, %u\n", uResId);
|
|
return;
|
|
}
|
|
// Wait for offline thread to complete, if there is one //
|
|
ClusWorkerTerminate(&(resourceEntry->OfflineThread));
|
|
DisksOfflineOrTerminate(resourceEntry, TERMINATE);
|
|
}
|
|
|
|
|
|
DWORD
|
|
WINAPI
|
|
DisksArbitrate(
|
|
IN RESID Resource,
|
|
IN PQUORUM_RESOURCE_LOST LostQuorumResource
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Arbitrate for a device by performing a reservation on the device.
|
|
|
|
Arguments:
|
|
|
|
Resource - supplies resource id to be brought online.
|
|
|
|
LostQuorumResource - routine to call when quorum resource is lost.
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS if successful.
|
|
A Win32 error code if other failure.
|
|
|
|
--*/
|
|
|
|
{
|
|
PDISK_RESOURCE resourceEntry;
|
|
DWORD status;
|
|
|
|
ULONG uResId = PtrToUlong(Resource);
|
|
|
|
//
|
|
// Make sure the RESID is okay.
|
|
//
|
|
resourceEntry = DiskTable[uResId];
|
|
if ( resourceEntry == NULL ) {
|
|
DISKS_PRINT("Arbitrate, bad resource id, %u\n", uResId);
|
|
return(ERROR_RESOURCE_NOT_FOUND);
|
|
}
|
|
|
|
// [HACKHACK]
|
|
// [GorN] 10/28/1999. If Offline thread detects that
|
|
// it is being terminated, it will not set the resource status to
|
|
// offline. ArbitrateCount != 0 will give it a hint on whether
|
|
// to set the resource status or not
|
|
|
|
InterlockedIncrement(&resourceEntry->ArbitrationInfo.ArbitrateCount);
|
|
|
|
(DiskpLogEvent)(
|
|
resourceEntry->ResourceHandle,
|
|
LOG_INFORMATION,
|
|
L"[DiskArb]Wait for offline thread to complete...\n"
|
|
);
|
|
ClusWorkerTerminate(&(resourceEntry->OfflineThread));
|
|
|
|
//
|
|
// Perform DoAttach only. Do not open.
|
|
//
|
|
status = DisksOpenResourceFileHandle(resourceEntry, L"Arbitrate",0);
|
|
if (status != ERROR_SUCCESS) {
|
|
goto error_exit;
|
|
}
|
|
|
|
status = DiskArbitration( resourceEntry, DiskspClusDiskZero );
|
|
|
|
(DiskpLogEvent)(
|
|
resourceEntry->ResourceHandle,
|
|
LOG_INFORMATION,
|
|
L"[DiskArb]Arbitrate returned status %1!u!.\n",
|
|
status );
|
|
|
|
if (status == ERROR_SUCCESS) {
|
|
resourceEntry->LostQuorum = LostQuorumResource;
|
|
}
|
|
error_exit:
|
|
InterlockedDecrement(&resourceEntry->ArbitrationInfo.ArbitrateCount);
|
|
|
|
return status;
|
|
} // DisksArbitrate //
|
|
|
|
|
|
|
|
RESID
|
|
WINAPI
|
|
DisksOpen(
|
|
IN LPCWSTR ResourceName,
|
|
IN HKEY ResourceKey,
|
|
IN RESOURCE_HANDLE ResourceHandle
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Open routine for Disk resource.
|
|
|
|
Arguments:
|
|
|
|
ResourceName - supplies the resource name
|
|
|
|
ResourceKey - supplies handle to this resource's cluster
|
|
registry key
|
|
|
|
ResourceHandle - the resource handle to be supplied with SetResourceStatus
|
|
is called.
|
|
|
|
Return Value:
|
|
|
|
RESID of created resource
|
|
Zero on failure
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD diskResource = 0;
|
|
UCHAR deviceName[MAX_PATH];
|
|
DWORD status;
|
|
HKEY clusDiskParametersKey = NULL;
|
|
HKEY resourceParametersKey = NULL;
|
|
HKEY resKey = NULL;
|
|
PDISK_RESOURCE resourceEntry;
|
|
DWORD valueType;
|
|
DWORD valueLength;
|
|
LPWSTR nameOfPropInError;
|
|
DWORD previousDiskCount;
|
|
|
|
//
|
|
// Open registry parameters key for ClusDisk.
|
|
//
|
|
status = RegOpenKeyEx( HKEY_LOCAL_MACHINE,
|
|
CLUSDISK_REGISTRY_SIGNATURES,
|
|
0,
|
|
KEY_READ,
|
|
&clusDiskParametersKey );
|
|
|
|
if ( status != ERROR_SUCCESS ) {
|
|
(DiskpLogEvent)(
|
|
ResourceHandle,
|
|
LOG_ERROR,
|
|
L"Unable to open ClusDisk parameters key. Error: %1!u!.\n",
|
|
status );
|
|
return(0);
|
|
}
|
|
|
|
//
|
|
// Open the resource's parameters key.
|
|
//
|
|
|
|
status = ClusterRegOpenKey( ResourceKey,
|
|
CLUSREG_KEYNAME_PARAMETERS,
|
|
KEY_READ | KEY_WRITE,
|
|
&resourceParametersKey );
|
|
|
|
if ( status != ERROR_SUCCESS ) {
|
|
(DiskpLogEvent)(
|
|
ResourceHandle,
|
|
LOG_ERROR,
|
|
L"Unable to open resource parameters key. Error: %1!u!.\n",
|
|
status );
|
|
goto error_exit;
|
|
}
|
|
|
|
//
|
|
// Get a handle to our resource key so that we can get our name later
|
|
// if we need to log an event.
|
|
//
|
|
status = ClusterRegOpenKey( ResourceKey,
|
|
L"",
|
|
KEY_READ,
|
|
&resKey);
|
|
if (status != ERROR_SUCCESS) {
|
|
(DiskpLogEvent)(ResourceHandle,
|
|
LOG_ERROR,
|
|
L"Unable to open resource key. Error: %1!u!.\n",
|
|
status );
|
|
goto error_exit;
|
|
}
|
|
|
|
//
|
|
// Allocate and zero disk info structure.
|
|
//
|
|
resourceEntry = LocalAlloc(LMEM_FIXED, sizeof(DISK_RESOURCE));
|
|
if (!resourceEntry) {
|
|
status = GetLastError();
|
|
(DiskpLogEvent)(ResourceHandle,
|
|
LOG_ERROR,
|
|
L"Unable to allocate disk resource data. Error: %1!u!.\n",
|
|
status );
|
|
goto error_exit;
|
|
}
|
|
ZeroMemory( resourceEntry, sizeof(DISK_RESOURCE));
|
|
resourceEntry->ResourceParametersKey = resourceParametersKey;
|
|
resourceEntry->ClusDiskParametersKey = clusDiskParametersKey;
|
|
resourceEntry->ResourceKey = resKey;
|
|
resourceEntry->ResourceHandle = ResourceHandle;
|
|
//resourceEntry->Inserted = FALSE;
|
|
//resourceEntry->Attached = FALSE;
|
|
//resourceEntry->DiskInfo.Params.Signature = 0;
|
|
|
|
status = ArbitrationInfoInit(resourceEntry);
|
|
if ( status != ERROR_SUCCESS ) {
|
|
LocalFree( resourceEntry );
|
|
goto error_exit;
|
|
}
|
|
status = CreateArbWorkQueue(ResourceHandle);
|
|
if ( status != ERROR_SUCCESS ) {
|
|
LocalFree( resourceEntry );
|
|
goto error_exit;
|
|
}
|
|
|
|
#if 0
|
|
//
|
|
// GN: It seems that there is no point doing this here
|
|
// If we are on the join path, we cannot get
|
|
// any information about the disk and the call will fail
|
|
//
|
|
// If we are forming the cluster we will update the information
|
|
// when we bring the disk online
|
|
//
|
|
status = DiskspSsyncDiskInfo( L"Open", resourceEntry , 0 );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
(DiskpLogEvent)(ResourceHandle,
|
|
LOG_ERROR,
|
|
L"Unable to ssync DiskInfo. Error: %1!u!.\n",
|
|
status );
|
|
}
|
|
#endif
|
|
|
|
//
|
|
// Save disk info structure.
|
|
//
|
|
|
|
EnterCriticalSection( &DisksLock );
|
|
|
|
if (DiskspClusDiskZero == NULL) {
|
|
status = DevfileOpen(&DiskspClusDiskZero, L"\\Device\\ClusDisk0");
|
|
if (!NT_SUCCESS(status) ) {
|
|
MountieCleanup ( &resourceEntry -> MountieInfo );
|
|
ArbitrationInfoCleanup( resourceEntry );
|
|
LocalFree( resourceEntry );
|
|
LeaveCriticalSection( &DisksLock );
|
|
(DiskpLogEvent)(
|
|
ResourceHandle,
|
|
LOG_ERROR,
|
|
L"Cannot open a handle to clusdisk driver, %1!x!.\n",
|
|
status);
|
|
goto error_exit;
|
|
}
|
|
}
|
|
diskResource = PtrToUlong(DiskspGetNextDisk());
|
|
|
|
if ( diskResource >= MAX_DISKS ) {
|
|
MountieCleanup ( &resourceEntry -> MountieInfo );
|
|
ArbitrationInfoCleanup( resourceEntry );
|
|
LocalFree( resourceEntry );
|
|
LeaveCriticalSection( &DisksLock );
|
|
(DiskpLogEvent)(
|
|
ResourceHandle,
|
|
LOG_ERROR,
|
|
L"Couldn't find an available resource slot %1!u!.\n",
|
|
diskResource);
|
|
status = ERROR_ALLOTTED_SPACE_EXCEEDED;
|
|
goto error_exit;
|
|
}
|
|
|
|
DiskTable[diskResource] = resourceEntry;
|
|
previousDiskCount = InterlockedExchangeAdd(&DiskCount, 1);
|
|
|
|
LeaveCriticalSection( &DisksLock );
|
|
|
|
if (previousDiskCount == 0) {
|
|
StartNotificationWatcherThread();
|
|
}
|
|
|
|
DisksMountPointInitialize( resourceEntry );
|
|
|
|
//
|
|
// Read our disk signature from the resource parameters.
|
|
//
|
|
status = ResUtilGetPropertiesToParameterBlock( resourceEntry->ResourceParametersKey,
|
|
DiskResourcePrivateProperties,
|
|
(LPBYTE) &resourceEntry->DiskInfo.Params,
|
|
FALSE, //CheckForRequiredProperties
|
|
&nameOfPropInError );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
(DiskpLogEvent)(
|
|
resourceEntry->ResourceHandle,
|
|
LOG_ERROR,
|
|
L"Open: Unable to read the '%1' property. Error: %2!u!.\n",
|
|
(nameOfPropInError == NULL ? L"" : nameOfPropInError),
|
|
status );
|
|
}
|
|
|
|
return((RESID)( UlongToPtr(diskResource) ) );
|
|
|
|
error_exit:
|
|
|
|
if ( clusDiskParametersKey != NULL ) {
|
|
RegCloseKey( clusDiskParametersKey );
|
|
}
|
|
|
|
if ( resourceParametersKey != NULL ) {
|
|
ClusterRegCloseKey( resourceParametersKey );
|
|
}
|
|
|
|
if ( resKey != NULL ) {
|
|
ClusterRegCloseKey( resKey );
|
|
}
|
|
|
|
SetLastError( status );
|
|
return((RESID)0);
|
|
|
|
} // DisksOpen
|
|
|
|
|
|
|
|
DWORD
|
|
WINAPI
|
|
DisksRelease(
|
|
IN RESID Resource
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Release arbitration for a device by stopping the reservation thread.
|
|
|
|
Arguments:
|
|
|
|
Resource - supplies resource id to be brought online
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS if successful.
|
|
ERROR_HOST_NODE_NOT_OWNER if the resource is not owned.
|
|
A Win32 error code if other failure.
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD status;
|
|
PDISK_RESOURCE resourceEntry;
|
|
ULONG uResId=PtrToUlong(Resource);
|
|
|
|
//
|
|
// Make sure the RESID is okay.
|
|
//
|
|
|
|
resourceEntry = DiskTable[uResId];
|
|
if ( resourceEntry == NULL ) {
|
|
DISKS_PRINT("Release, bad resource id, %u\n", uResId);
|
|
return(ERROR_RESOURCE_NOT_FOUND);
|
|
}
|
|
|
|
(DiskpLogEvent)(
|
|
resourceEntry->ResourceHandle,
|
|
LOG_INFORMATION,
|
|
L"DisksRelease started, Inserted = %1!u! \n",
|
|
resourceEntry->Inserted );
|
|
|
|
if (resourceEntry->Inserted) { // [GN] #209018 //
|
|
(DiskpLogEvent)(
|
|
resourceEntry->ResourceHandle,
|
|
LOG_ERROR,
|
|
L"Cannot release, Disk is online.\n");
|
|
} else {
|
|
StopPersistentReservations(resourceEntry);
|
|
}
|
|
resourceEntry->LostQuorum = NULL;
|
|
|
|
return(ERROR_SUCCESS);
|
|
|
|
} // DisksRelease
|
|
|
|
|
|
|
|
DWORD
|
|
WINAPI
|
|
DisksOnline(
|
|
IN RESID Resource,
|
|
IN OUT PHANDLE EventHandle
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Online routine for Disk resource.
|
|
|
|
Arguments:
|
|
|
|
Resource - supplies resource id to be brought online
|
|
|
|
EventHandle - supplies a pointer to a handle to signal on error.
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS if successful.
|
|
ERROR_RESOURCE_NOT_FOUND if RESID is not valid.
|
|
ERROR_RESOURCE_NOT_AVAILABLE if resource was arbitrated but failed to
|
|
acquire 'ownership'.
|
|
Win32 error code if other failure.
|
|
|
|
--*/
|
|
|
|
{
|
|
PDISK_RESOURCE resourceEntry;
|
|
DWORD Status;
|
|
ULONG uResId=PtrToUlong(Resource);
|
|
|
|
//
|
|
// Make sure the RESID is okay.
|
|
//
|
|
|
|
resourceEntry = DiskTable[uResId];
|
|
if ( resourceEntry == NULL ) {
|
|
DISKS_PRINT("Online, bad resource id, %u\n", uResId);
|
|
return(ERROR_RESOURCE_NOT_FOUND);
|
|
}
|
|
|
|
//
|
|
// Shutdown the online thread if it's running.
|
|
//
|
|
ClusWorkerTerminate(&resourceEntry->OnlineThread);
|
|
|
|
Status = ClusWorkerCreate(&resourceEntry->OnlineThread,
|
|
DisksOnlineThread,
|
|
resourceEntry);
|
|
if (Status == ERROR_SUCCESS) {
|
|
Status = ERROR_IO_PENDING;
|
|
}
|
|
return(Status);
|
|
|
|
} // DisksOnline
|
|
|
|
DWORD
|
|
DisksOfflineThread(
|
|
IN PCLUS_WORKER Worker,
|
|
IN PDISK_RESOURCE ResourceEntry
|
|
)
|
|
{
|
|
RESOURCE_STATUS resourceStatus;
|
|
DWORD status;
|
|
|
|
ResUtilInitializeResourceStatus( &resourceStatus );
|
|
|
|
resourceStatus.ResourceState = ClusterResourceFailed;
|
|
//resourceStatus.WaitHint = 0;
|
|
resourceStatus.CheckPoint = 1;
|
|
ClusWorkerTerminate( &ResourceEntry->OnlineThread );
|
|
status = DisksOfflineOrTerminate(ResourceEntry, OFFLINE);
|
|
|
|
if (status == ERROR_SUCCESS) {
|
|
resourceStatus.ResourceState = ClusterResourceOffline;
|
|
}
|
|
|
|
//
|
|
// [HACKHACK] [GorN 10/04/1999]
|
|
// If Terminate is called when the offline is in progress,
|
|
// the terminate blocks waiting for OfflineThread to complete.
|
|
// However, offline thread is stuck trying
|
|
// to set ResourceStatus, since event list lock in the resmon
|
|
// is taken out by Terminate thread.
|
|
//
|
|
// The following code doesn't fix this deadlock completely.
|
|
// It just reduces the window during which the problem can occur.
|
|
// [Resmon times out SetResourceStatus in 3 minutes, this breaks the deadlock]
|
|
//
|
|
// [HACKHACK] [GorN 10/28/1999]
|
|
// Arbitrate is also trying to terminate the offline thread
|
|
// We need some way to distinguish between these two cases
|
|
//
|
|
// The order of setting is
|
|
// ArbitrateCount
|
|
// ClusWorkerTerminate
|
|
//
|
|
// Order of checking is ClusWorkerTerminate then ArbitrateCount.
|
|
// (Won't work with aggressive memory access reordering, but who cares <grin>)
|
|
//
|
|
if ( !ClusWorkerCheckTerminate( Worker ) ||
|
|
ResourceEntry->ArbitrationInfo.ArbitrateCount)
|
|
{
|
|
(DiskpSetResourceStatus)(ResourceEntry->ResourceHandle,
|
|
&resourceStatus );
|
|
}
|
|
return status;
|
|
}
|
|
|
|
DWORD
|
|
WINAPI DisksOffline(
|
|
IN RESID ResourceId
|
|
)
|
|
{
|
|
PDISK_RESOURCE ResourceEntry;
|
|
DWORD status = ERROR_SUCCESS;
|
|
ULONG uResId=PtrToUlong(ResourceId);
|
|
|
|
//
|
|
// Make sure the RESID is okay.
|
|
//
|
|
|
|
ResourceEntry = DiskTable[uResId];
|
|
if ( ResourceEntry == NULL ) {
|
|
DISKS_PRINT("Offline, bad resource id, %u\n", uResId);
|
|
return(ERROR_RESOURCE_NOT_FOUND);
|
|
}
|
|
|
|
status = ClusWorkerCreate( &ResourceEntry->OfflineThread,
|
|
(PWORKER_START_ROUTINE)DisksOfflineThread,
|
|
ResourceEntry );
|
|
if ( status != ERROR_SUCCESS )
|
|
{
|
|
(DiskpLogEvent)(
|
|
ResourceEntry->ResourceHandle,
|
|
LOG_ERROR,
|
|
L"Offline: Unable to start thread, status %1!u!.\n",
|
|
status
|
|
);
|
|
}
|
|
else
|
|
{
|
|
status = ERROR_IO_PENDING;
|
|
}
|
|
|
|
return status;
|
|
|
|
} // DisksOffline
|
|
|
|
|
|
BOOL
|
|
WINAPI
|
|
DisksIsAlive(
|
|
IN RESID Resource
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
IsAlive routine for Disk resource.
|
|
|
|
Arguments:
|
|
|
|
Resource - supplies the resource id to be polled.
|
|
|
|
Return Value:
|
|
|
|
TRUE - Resource is alive and well
|
|
|
|
FALSE - Resource is toast.
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD status;
|
|
PDISK_RESOURCE resourceEntry;
|
|
|
|
ULONG uResId=PtrToUlong(Resource);
|
|
|
|
//
|
|
// Make sure the RESID is okay.
|
|
//
|
|
|
|
resourceEntry = DiskTable[uResId];
|
|
if ( resourceEntry == NULL ) {
|
|
DISKS_PRINT("IsAlive, bad resource id, %u\n", uResId);
|
|
return(FALSE);
|
|
}
|
|
|
|
if ( resourceEntry->DiskInfo.FailStatus != 0 ) {
|
|
(DiskpLogEvent)(
|
|
resourceEntry->ResourceHandle,
|
|
LOG_ERROR,
|
|
L"IsAlive, error checking device, error %1!u!.\n",
|
|
resourceEntry->DiskInfo.FailStatus );
|
|
ClusResLogSystemEventByKey(resourceEntry->ResourceKey,
|
|
LOG_CRITICAL,
|
|
RES_DISK_FAILED_SCSI_CHECK);
|
|
resourceEntry->DiskInfo.FailStatus = 0;
|
|
return(FALSE);
|
|
}
|
|
|
|
//
|
|
// Check out the interesting partitions.
|
|
//
|
|
|
|
#if 0
|
|
(DiskpLogEvent)(
|
|
resourceEntry->ResourceHandle,
|
|
LOG_INFORMATION,
|
|
L"About to call DriveIsAlive!\n" );
|
|
#endif
|
|
|
|
status = DisksDriveIsAlive( resourceEntry,
|
|
FALSE);
|
|
|
|
if (status == ERROR_SUCCESS) {
|
|
return(TRUE);
|
|
} else {
|
|
ClusResLogSystemEventByKeyData(resourceEntry->ResourceKey,
|
|
LOG_CRITICAL,
|
|
RES_DISK_FILESYSTEM_FAILED,
|
|
sizeof(status),
|
|
&status);
|
|
return(FALSE);
|
|
}
|
|
|
|
} // DisksIsAlive
|
|
|
|
|
|
BOOL
|
|
WINAPI
|
|
DisksLooksAlive(
|
|
IN RESID Resource
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
LooksAlive routine for Disk resource.
|
|
|
|
Arguments:
|
|
|
|
Resource - supplies the resource id to be polled.
|
|
|
|
Return Value:
|
|
|
|
TRUE - Resource looks like it is alive and well
|
|
|
|
FALSE - Resource looks like it is toast.
|
|
|
|
--*/
|
|
|
|
{
|
|
BOOL success;
|
|
PDISK_RESOURCE resourceEntry;
|
|
|
|
ULONG uResId=PtrToUlong(Resource);
|
|
|
|
//
|
|
// Make sure the RESID is okay.
|
|
//
|
|
|
|
resourceEntry = DiskTable[uResId];
|
|
if ( resourceEntry == NULL ) {
|
|
DISKS_PRINT("Online, bad resource id, %u\n", uResId);
|
|
return(FALSE);
|
|
}
|
|
|
|
if ( resourceEntry->DiskInfo.FailStatus != 0 ) {
|
|
(DiskpLogEvent)(
|
|
resourceEntry->ResourceHandle,
|
|
LOG_ERROR,
|
|
L"LooksAlive, error checking device, error %1!u!.\n",
|
|
resourceEntry->DiskInfo.FailStatus );
|
|
return FALSE;
|
|
}
|
|
|
|
return(TRUE);
|
|
|
|
} // DisksLooksAlive
|
|
|
|
|
|
VOID
|
|
WINAPI
|
|
DisksClose(
|
|
IN RESID Resource
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Close routine for Disk resource.
|
|
|
|
Arguments:
|
|
|
|
Resource - supplies resource id to be closed.
|
|
|
|
Return Value:
|
|
|
|
None.
|
|
|
|
--*/
|
|
|
|
{
|
|
PDISK_RESOURCE resourceEntry;
|
|
|
|
ULONG uResId=PtrToUlong(Resource);
|
|
|
|
//
|
|
// Make sure the RESID is okay.
|
|
//
|
|
|
|
resourceEntry = DiskTable[uResId];
|
|
if ( resourceEntry == NULL ) {
|
|
DISKS_PRINT("Close, bad resource id, %u\n", uResId);
|
|
return;
|
|
}
|
|
|
|
//
|
|
// Wait for the online thread to finish.
|
|
//
|
|
DisksTerminate( Resource );
|
|
|
|
DisksMountPointCleanup( resourceEntry );
|
|
|
|
if ( resourceEntry->DiskInfo.Params.SerialNumber ) {
|
|
LocalFree( resourceEntry->DiskInfo.Params.SerialNumber );
|
|
resourceEntry->DiskInfo.Params.SerialNumber = NULL;
|
|
}
|
|
|
|
DiskTable[uResId] = NULL;
|
|
if ( InterlockedExchangeAdd(&DiskCount, -1) == 1 ) {
|
|
// This is the last disk //
|
|
StopNotificationWatcher();
|
|
DestroyArbWorkQueue();
|
|
}
|
|
|
|
ClusterRegCloseKey( resourceEntry->ResourceParametersKey);
|
|
ClusterRegCloseKey( resourceEntry->ResourceKey);
|
|
RegCloseKey(resourceEntry->ClusDiskParametersKey);
|
|
ArbitrationInfoCleanup(resourceEntry);
|
|
MountieCleanup( &resourceEntry->MountieInfo );
|
|
LocalFree(resourceEntry);
|
|
|
|
return;
|
|
|
|
} // DisksClose
|
|
|
|
|
|
#define TEST_WRITE_BUFFER_SIZE 512
|
|
#define TEST_WRITE_BUFFER_BLOCKS 10
|
|
|
|
|
|
DWORD
|
|
DisksWriteTestFile(
|
|
IN PWCHAR VolumeName,
|
|
IN PDISK_RESOURCE ResourceEntry
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Checks for disk corruption problem by opening a temporary file, writing
|
|
a test pattern, and saving the file. Then it reopens the file, reads
|
|
the data and verifies the data. If any part of this fails, indicate an
|
|
error to the caller so that chkdsk can run.
|
|
|
|
Arguments:
|
|
|
|
VolumeName - Supplies the device name of the form:
|
|
\\?\Volume{GUID}\ [Note trailing backslash!]
|
|
|
|
ResourceEntry - Supplies a pointer to the resource structure
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS if successful
|
|
|
|
Win32 error code otherwise
|
|
|
|
--*/
|
|
|
|
{
|
|
PTEB currentTeb;
|
|
|
|
HANDLE hFile = INVALID_HANDLE_VALUE;
|
|
|
|
PCHAR testBuffer = NULL;
|
|
PCHAR testBufferSave;
|
|
PCHAR readBuffer = NULL;
|
|
|
|
WCHAR szFilePrefix[] = L"zClusterOnlineChk.tmp";
|
|
WCHAR tempFileName[MAX_PATH * 2];
|
|
|
|
ULONG hardErrorValue;
|
|
|
|
BOOL hardErrorsDisabled = FALSE;
|
|
|
|
DWORD error = ERROR_SUCCESS;
|
|
DWORD bytesTransferred;
|
|
DWORD i;
|
|
DWORD len;
|
|
|
|
//
|
|
// Wrap in try-finally.
|
|
//
|
|
|
|
_try {
|
|
|
|
//
|
|
// Temporarily disable hard pop-ups.
|
|
//
|
|
|
|
currentTeb = NtCurrentTeb();
|
|
hardErrorValue = currentTeb->HardErrorsAreDisabled;
|
|
currentTeb->HardErrorsAreDisabled = 1;
|
|
hardErrorsDisabled = TRUE;
|
|
|
|
//
|
|
// Allocate and prepare the test buffer.
|
|
//
|
|
|
|
testBuffer = LocalAlloc( LMEM_FIXED, TEST_WRITE_BUFFER_SIZE );
|
|
|
|
if ( !testBuffer ) {
|
|
error = GetLastError();
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle, LOG_ERROR,
|
|
L"DisksWriteTestFile: Unable to allocate test buffer %1!u! \n",
|
|
error );
|
|
_leave;
|
|
}
|
|
|
|
testBufferSave = testBuffer;
|
|
for ( i = 0; i < TEST_WRITE_BUFFER_SIZE; i++ ) {
|
|
testBuffer[i] = (CHAR)i;
|
|
}
|
|
testBuffer = testBufferSave;
|
|
|
|
//
|
|
// Allocate the read buffer.
|
|
//
|
|
|
|
readBuffer = LocalAlloc( LMEM_FIXED, TEST_WRITE_BUFFER_SIZE );
|
|
|
|
if ( !readBuffer ) {
|
|
error = GetLastError();
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle, LOG_ERROR,
|
|
L"DisksWriteTestFile: Unable to allocate read buffer %1!u! \n",
|
|
error );
|
|
_leave;
|
|
}
|
|
|
|
//
|
|
// Create a temporary file name. GetTempFileNameW doesn't like the path
|
|
// \\?\Volume{GUID}\ so we build our own file name.
|
|
//
|
|
|
|
len = wcslen( VolumeName );
|
|
|
|
if ( len > sizeof(tempFileName) ) {
|
|
error = ERROR_ALLOTTED_SPACE_EXCEEDED;
|
|
_leave;
|
|
}
|
|
|
|
ZeroMemory( tempFileName, sizeof(tempFileName) );
|
|
|
|
wcsncpy( tempFileName, VolumeName, len );
|
|
wcsncat( tempFileName, szFilePrefix, wcslen( szFilePrefix ) );
|
|
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle, LOG_INFORMATION,
|
|
L"DisksWriteTestFile: Creating test file (%1!ws!) \n",
|
|
tempFileName );
|
|
|
|
//
|
|
// First, try to delete the file to make sure we didn't leave one laying around.
|
|
// We will ignore the return status as the next CreateFile will fail when we try
|
|
// to open the temporary file again.
|
|
//
|
|
|
|
DeleteFileW( tempFileName );
|
|
|
|
//
|
|
// Open temporary file. Make sure the data is actually written to the disk
|
|
// and is not cached. Do NOT specify FILE_ATTRIBUTE_TEMPORARY as the system
|
|
// will then try to keep all the file's data in memory instead of on the
|
|
// disk.
|
|
//
|
|
|
|
hFile = CreateFileW( tempFileName,
|
|
GENERIC_READ | GENERIC_WRITE,
|
|
0,
|
|
NULL,
|
|
CREATE_NEW,
|
|
FILE_FLAG_WRITE_THROUGH,
|
|
NULL );
|
|
|
|
if ( INVALID_HANDLE_VALUE == hFile ) {
|
|
error = GetLastError();
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle, LOG_ERROR,
|
|
L"DisksWriteTestFile: CreateFile for (%1!ws!) failed %2!u! \n",
|
|
tempFileName,
|
|
error );
|
|
_leave;
|
|
}
|
|
|
|
//
|
|
// Write the data pattern for some number of blocks.
|
|
//
|
|
|
|
for ( i = 0; i < TEST_WRITE_BUFFER_BLOCKS; i++ ) {
|
|
|
|
if ( !WriteFile( hFile,
|
|
testBuffer,
|
|
TEST_WRITE_BUFFER_SIZE,
|
|
&bytesTransferred,
|
|
NULL ) ) {
|
|
|
|
// Write failed...
|
|
|
|
error = GetLastError();
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle, LOG_ERROR,
|
|
L"DisksWriteTestFile: Write test file failed %1!u! \n",
|
|
error );
|
|
_leave;
|
|
}
|
|
|
|
if ( bytesTransferred != TEST_WRITE_BUFFER_SIZE ) {
|
|
error = ERROR_IO_DEVICE;
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle, LOG_ERROR,
|
|
L"DisksWriteTestFile: bytes written (%1!u!) != test buffer (%2!u!) \n",
|
|
bytesTransferred,
|
|
TEST_WRITE_BUFFER_SIZE );
|
|
_leave;
|
|
}
|
|
|
|
}
|
|
|
|
//
|
|
// Close the file.
|
|
//
|
|
|
|
CloseHandle( hFile );
|
|
hFile = INVALID_HANDLE_VALUE;
|
|
|
|
//
|
|
// Reopen the same file.
|
|
//
|
|
|
|
hFile = CreateFileW( tempFileName,
|
|
GENERIC_READ,
|
|
0,
|
|
NULL,
|
|
OPEN_EXISTING,
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
NULL );
|
|
|
|
if ( INVALID_HANDLE_VALUE == hFile ) {
|
|
error = GetLastError();
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle, LOG_ERROR,
|
|
L"DisksWriteTestFile: CreateFile (second open) for (%1!ws!) failed %2!u! \n",
|
|
tempFileName,
|
|
error );
|
|
_leave;
|
|
}
|
|
|
|
//
|
|
// Read the data and compare to original buffer.
|
|
//
|
|
|
|
for ( i = 0; i < TEST_WRITE_BUFFER_BLOCKS; i++ ) {
|
|
|
|
if ( !ReadFile( hFile,
|
|
readBuffer,
|
|
TEST_WRITE_BUFFER_SIZE,
|
|
&bytesTransferred,
|
|
NULL ) ) {
|
|
|
|
// Read failed...
|
|
|
|
error = GetLastError();
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle, LOG_ERROR,
|
|
L"DisksWriteTestFile: Read test file failed %1!u! \n",
|
|
error );
|
|
_leave;
|
|
}
|
|
|
|
if ( bytesTransferred != TEST_WRITE_BUFFER_SIZE ) {
|
|
error = ERROR_IO_DEVICE;
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle, LOG_ERROR,
|
|
L"DisksWriteTestFile: bytes read (%1!u!) != test buffer (%2!u!) \n",
|
|
bytesTransferred,
|
|
TEST_WRITE_BUFFER_SIZE );
|
|
_leave;
|
|
}
|
|
|
|
//
|
|
// Compare original buffer to that just read from file.
|
|
//
|
|
|
|
if ( !memcmp( testBuffer, readBuffer, TEST_WRITE_BUFFER_SIZE ) ) {
|
|
|
|
//
|
|
// Buffers identical, check next buffer.
|
|
//
|
|
|
|
continue;
|
|
}
|
|
|
|
// Miscompare...
|
|
|
|
error = ERROR_IO_DEVICE;
|
|
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle, LOG_ERROR,
|
|
L"DisksWriteTestFile: byte miscompare in block %1!u! \n",
|
|
i );
|
|
_leave;
|
|
|
|
}
|
|
|
|
}
|
|
_finally {
|
|
|
|
//
|
|
// Clean up all resources.
|
|
//
|
|
|
|
if ( testBuffer ) {
|
|
LocalFree( testBuffer );
|
|
}
|
|
|
|
if ( readBuffer ) {
|
|
LocalFree( readBuffer );
|
|
}
|
|
|
|
if ( hFile != INVALID_HANDLE_VALUE ) {
|
|
|
|
//
|
|
// Close must happen before the delete.
|
|
//
|
|
|
|
CloseHandle( hFile );
|
|
|
|
//
|
|
// Only report an error deleting file if the error code was not set
|
|
// in the code above. We don't want to overwrite a potentially
|
|
// informative error code.
|
|
//
|
|
|
|
if ( !DeleteFileW( tempFileName ) && ( ERROR_SUCCESS == error ) ) {
|
|
error = GetLastError();
|
|
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle, LOG_ERROR,
|
|
L"DisksWriteTestFile: Unable to delete file (%1!ws!) failed %2!u! \n",
|
|
tempFileName,
|
|
error );
|
|
}
|
|
}
|
|
|
|
//
|
|
// Re-enable hard pop-ups.
|
|
//
|
|
|
|
if ( hardErrorsDisabled ) {
|
|
currentTeb = NtCurrentTeb();
|
|
currentTeb->HardErrorsAreDisabled = hardErrorValue;
|
|
}
|
|
}
|
|
|
|
if ( ERROR_DISK_FULL == error ) {
|
|
|
|
DWORD quorumSignature;
|
|
DWORD logMsg;
|
|
DWORD logLevel;
|
|
|
|
//
|
|
// We don't want chkdsk to run because the disk is full, so indicate there
|
|
// was no error and write a message to system event log.
|
|
//
|
|
|
|
error = NO_ERROR;
|
|
|
|
if ( NO_ERROR == GetQuorumSignature( &quorumSignature ) ) {
|
|
|
|
if ( quorumSignature == ResourceEntry->DiskInfo.Params.Signature ) {
|
|
logMsg = RES_DISK_FULL_DISK_QUORUM;
|
|
logLevel = LOG_CRITICAL;
|
|
} else {
|
|
logMsg = RES_DISK_FULL_DISK_NOT_QUORUM;
|
|
logLevel = LOG_UNUSUAL;
|
|
}
|
|
} else {
|
|
logMsg = RES_DISK_FULL_DISK_UNKNOWN;
|
|
logLevel = LOG_CRITICAL;
|
|
}
|
|
|
|
ClusResLogSystemEventByKey1(ResourceEntry->ResourceKey,
|
|
logLevel,
|
|
logMsg,
|
|
VolumeName );
|
|
}
|
|
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle, LOG_INFORMATION,
|
|
L"DisksWriteTestFile: Returns status %1!u! \n",
|
|
error );
|
|
|
|
return error;
|
|
|
|
} // DisksWriteTestFile
|
|
|
|
|
|
DWORD
|
|
DisksCheckCorruption(
|
|
IN PWCHAR DeviceName,
|
|
IN PWCHAR VolumeName,
|
|
IN PDISK_RESOURCE ResourceEntry
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Checks for disk corruption problems.
|
|
|
|
Arguments:
|
|
|
|
DeviceName - Supplies name of the form:
|
|
\Device\HarddiskX\PartitionY [Note: no trailing backslash]
|
|
|
|
VolumeName - Supplies the device name of the form:
|
|
\\?\Volume{GUID}\ [Note trailing backslash!]
|
|
|
|
ResourceEntry - Supplies a pointer to the resource structure
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS if successful
|
|
|
|
Win32 error code otherwise
|
|
|
|
--*/
|
|
|
|
{
|
|
BOOL dirty;
|
|
DWORD status = ERROR_SUCCESS;
|
|
|
|
if ( ResourceEntry->DiskInfo.Params.SkipChkdsk ) {
|
|
status = ERROR_SUCCESS;
|
|
goto FnExit;
|
|
}
|
|
|
|
status = DisksIsVolumeDirty( DeviceName, ResourceEntry, &dirty );
|
|
if (status == ERROR_SUCCESS && dirty) {
|
|
|
|
status = ERROR_DISK_CORRUPT;
|
|
|
|
} else {
|
|
|
|
status = DisksWriteTestFile( VolumeName, ResourceEntry );
|
|
|
|
if ( status != ERROR_SUCCESS ) {
|
|
status = ERROR_DISK_CORRUPT;
|
|
}
|
|
}
|
|
|
|
FnExit:
|
|
|
|
return status;
|
|
|
|
} // DisksCheckCorruption
|
|
|
|
|
|
DWORD
|
|
DisksFixCorruption(
|
|
IN PWCHAR VolumeName,
|
|
IN PDISK_RESOURCE ResourceEntry,
|
|
IN DWORD CorruptStatus
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Fix file or disk corrupt problems.
|
|
|
|
Arguments:
|
|
|
|
VolumeName - Supplies the device name of the form:
|
|
\\?\Volume{GUID}\ [Note trailing backslash!]
|
|
|
|
ResourceEntry - Supplies a pointer to the disk resource entry
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS if successful
|
|
|
|
Win32 error code otherwise
|
|
|
|
Notes:
|
|
|
|
We'll need to lock the volume exclusive while we do this...
|
|
So threads that call this routine should ensure there are no
|
|
open files!
|
|
|
|
--*/
|
|
|
|
{
|
|
LPWSTR chkdskLogFileName = NULL;
|
|
|
|
DWORD Status;
|
|
DWORD len;
|
|
STARTUPINFOW StartupInfo;
|
|
PROCESS_INFORMATION ProcessInfo;
|
|
WCHAR CheckDiskInfo[MAX_PATH*2];
|
|
RESOURCE_STATUS resourceStatus;
|
|
RESOURCE_EXIT_STATE exit;
|
|
BOOL replaceBackslash;
|
|
|
|
BOOL bInheritHandles;
|
|
HANDLE chkdskLogFile = INVALID_HANDLE_VALUE;
|
|
UINT previousMode;
|
|
|
|
previousMode = SetErrorMode( SEM_FAILCRITICALERRORS|SEM_NOGPFAULTERRORBOX|SEM_NOOPENFILEERRORBOX );
|
|
|
|
//
|
|
// We need to strip the trailing backslash off so chkdsk will work.
|
|
//
|
|
|
|
len = wcslen( VolumeName );
|
|
|
|
if ( len > MAX_PATH ) {
|
|
SetErrorMode( previousMode );
|
|
return ERROR_ALLOTTED_SPACE_EXCEEDED;
|
|
}
|
|
|
|
if ( VolumeName[len-1] == L'\\') {
|
|
VolumeName[len-1] = UNICODE_NULL;
|
|
replaceBackslash = TRUE;
|
|
} else {
|
|
replaceBackslash = FALSE;
|
|
}
|
|
|
|
//
|
|
// Now handle the corruption problem by running CHKDSK.
|
|
//
|
|
|
|
_snwprintf(CheckDiskInfo, MAX_PATH * 2, L"ChkDsk /x /f %ws", VolumeName);
|
|
|
|
//
|
|
// Restore the backslash.
|
|
//
|
|
|
|
if ( replaceBackslash ) {
|
|
VolumeName[len-1] = L'\\';
|
|
}
|
|
|
|
|
|
ZeroMemory( &StartupInfo, sizeof(STARTUPINFOW) );
|
|
StartupInfo.cb = sizeof(STARTUPINFO);
|
|
StartupInfo.lpDesktop = L"WinSta0\\Default";
|
|
|
|
bInheritHandles = FALSE;
|
|
|
|
Status = DisksOpenChkdskLogFile( ResourceEntry,
|
|
&chkdskLogFile,
|
|
&chkdskLogFileName );
|
|
|
|
if ( NO_ERROR == Status && INVALID_HANDLE_VALUE != chkdskLogFile ) {
|
|
|
|
//
|
|
// When the output is redirected, we don't want to show the console window because it
|
|
// will be blank with simply a title in it. The event log message will let the user
|
|
// know to look in the chkdsk file.
|
|
//
|
|
|
|
StartupInfo.dwFlags = STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES;
|
|
StartupInfo.wShowWindow = SW_HIDE;
|
|
|
|
// Someone watching the console won't know what is happening, so show
|
|
// the window anyway...
|
|
// StartupInfo.dwFlags = STARTF_USESTDHANDLES;
|
|
|
|
StartupInfo.hStdInput = GetStdHandle(STD_INPUT_HANDLE);
|
|
|
|
//
|
|
// [169631] Chkdsk now verifies that hStdInput is not NULL.
|
|
// Since the resmon process was started with InheritHandles set to FALSE,
|
|
// GetStdHandle(STD_INPUT_HANDLE) will return NULL. When we run chkdsk with
|
|
// the options "/f /x", chkdsk should not be prompting the user and the
|
|
// input handle wouldn't be used. However, ulibs.dll was changed to always
|
|
// insure a nonzero input handle was supplied. So we have to supply some type
|
|
// of input handle. We could put INVALID_HANDLE_VALUE here, but the checks
|
|
// may change and it will fail later. For now, point input to the temporary
|
|
// output file we created.
|
|
//
|
|
|
|
if ( NULL == StartupInfo.hStdInput ) {
|
|
StartupInfo.hStdInput = chkdskLogFile;
|
|
}
|
|
|
|
StartupInfo.hStdOutput = chkdskLogFile;
|
|
StartupInfo.hStdError = chkdskLogFile;
|
|
|
|
bInheritHandles = TRUE;
|
|
}
|
|
|
|
//
|
|
// Log an event
|
|
//
|
|
if ( CorruptStatus == ERROR_DISK_CORRUPT ) {
|
|
// Must be corrupt disk
|
|
ClusResLogSystemEventByKey2(ResourceEntry->ResourceKey,
|
|
LOG_CRITICAL,
|
|
RES_DISK_CORRUPT_DISK,
|
|
VolumeName,
|
|
chkdskLogFileName);
|
|
} else {
|
|
// Must be corrupt file.
|
|
ClusResLogSystemEventByKey2(ResourceEntry->ResourceKey,
|
|
LOG_CRITICAL,
|
|
RES_DISK_CORRUPT_FILE,
|
|
VolumeName,
|
|
chkdskLogFileName);
|
|
}
|
|
|
|
if ( chkdskLogFileName ) {
|
|
LocalFree( chkdskLogFileName );
|
|
chkdskLogFileName = NULL;
|
|
}
|
|
|
|
if ( !CreateProcessW( NULL,
|
|
CheckDiskInfo,
|
|
NULL,
|
|
NULL,
|
|
bInheritHandles,
|
|
NORMAL_PRIORITY_CLASS,
|
|
NULL,
|
|
NULL,
|
|
&StartupInfo,
|
|
&ProcessInfo ) ) {
|
|
Status = GetLastError();
|
|
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle, LOG_ERROR,
|
|
L"DisksFixCorruption: CreateProcessW for chkdsk failed %1!u! \n",
|
|
Status );
|
|
|
|
if ( INVALID_HANDLE_VALUE != chkdskLogFile ) {
|
|
CloseHandle( chkdskLogFile );
|
|
}
|
|
SetErrorMode( previousMode );
|
|
return(Status);
|
|
}
|
|
|
|
CloseHandle( ProcessInfo.hThread );
|
|
|
|
//
|
|
// Wait for CHKDSK to finish.
|
|
//
|
|
#if 1
|
|
//
|
|
// Don't wait "forever"... things could get ugly if we dismount the file
|
|
// system while ChkDsk is running! But KeithKa says its okay to kill
|
|
// ChkDsk while it is running - it has to handle powerfails, crashes, etc.
|
|
//
|
|
resourceStatus.ResourceState = ClusterResourceOnlinePending;
|
|
while ( !ResourceEntry->OnlineThread.Terminate ) {
|
|
(DiskpSetResourceStatus)(ResourceEntry->ResourceHandle,
|
|
&resourceStatus );
|
|
Status = WaitForSingleObject( ProcessInfo.hProcess, 2000 );
|
|
if ( Status != WAIT_TIMEOUT ) {
|
|
break;
|
|
}
|
|
}
|
|
|
|
if ( ResourceEntry->OnlineThread.Terminate ) {
|
|
// If we were asked to terminate, make sure ChkNtfs is killed
|
|
TerminateProcess( ProcessInfo.hProcess, 999 );
|
|
CloseHandle( ProcessInfo.hProcess );
|
|
if ( INVALID_HANDLE_VALUE != chkdskLogFile ) {
|
|
CloseHandle( chkdskLogFile );
|
|
}
|
|
SetErrorMode( previousMode );
|
|
return(ERROR_SHUTDOWN_CLUSTER);
|
|
}
|
|
|
|
//
|
|
// Update our checkpoint state.
|
|
//
|
|
++resourceStatus.CheckPoint;
|
|
exit = (DiskpSetResourceStatus)(ResourceEntry->ResourceHandle,
|
|
&resourceStatus );
|
|
if ( exit == ResourceExitStateTerminate ) {
|
|
TerminateProcess( ProcessInfo.hProcess, 998 );
|
|
CloseHandle( ProcessInfo.hProcess );
|
|
if ( INVALID_HANDLE_VALUE != chkdskLogFile ) {
|
|
CloseHandle( chkdskLogFile );
|
|
}
|
|
SetErrorMode( previousMode );
|
|
return(ERROR_SHUTDOWN_CLUSTER);
|
|
}
|
|
|
|
#else
|
|
// Wait "forever"...
|
|
Status = WaitForSingleObject( ProcessInfo.hProcess, INFINITE );
|
|
#endif
|
|
|
|
if ( (Status == 0) &&
|
|
GetExitCodeProcess( ProcessInfo.hProcess, &Status ) ) {
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle,
|
|
LOG_ERROR,
|
|
L"FixCorruption: CHKDSK returned status of %1!u!.\n",
|
|
Status );
|
|
// [From supera.hxx]
|
|
//
|
|
// These symbols are used by Chkdsk functions to return an appropriate
|
|
// exit status to the chkdsk program.
|
|
// In order of most important first, the error level order are as follows:
|
|
// 3 > 1 > 2 > 0
|
|
// An error level of 3 will overwrite an error level of 1, 2, or 0.
|
|
|
|
// #define CHKDSK_EXIT_SUCCESS 0
|
|
// #define CHKDSK_EXIT_ERRS_FIXED 1
|
|
// #define CHKDSK_EXIT_MINOR_ERRS 2 // whether or not "/f"
|
|
// #define CHKDSK_EXIT_CLEANUP_WORK 2 // whether or not "/f"
|
|
// #define CHKDSK_EXIT_COULD_NOT_CHK 3
|
|
// #define CHKDSK_EXIT_ERRS_NOT_FIXED 3
|
|
// #define CHKDSK_EXIT_COULD_NOT_FIX 3
|
|
|
|
if (Status >= 3) {
|
|
Status = ERROR_DISK_CORRUPT;
|
|
} else {
|
|
Status = ERROR_SUCCESS;
|
|
}
|
|
}
|
|
|
|
CloseHandle( ProcessInfo.hProcess );
|
|
|
|
if ( INVALID_HANDLE_VALUE != chkdskLogFile ) {
|
|
CloseHandle( chkdskLogFile );
|
|
}
|
|
|
|
SetErrorMode( previousMode );
|
|
|
|
return(Status);
|
|
|
|
} // DisksFixCorruption
|
|
|
|
DWORD
|
|
DiskspGetQuorumPath(
|
|
OUT LPWSTR* lpQuorumLogPath
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Reads QuorumPath value from the registry.
|
|
|
|
|
|
Arguments:
|
|
|
|
lpQuorumLogPath - receives the poiner to a buffer containing QuorumLogPath
|
|
the buffer needs to be deallocated later via LocalFree
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS if successful
|
|
|
|
Win32 error code otherwise
|
|
|
|
--*/
|
|
{
|
|
DWORD Status;
|
|
LPWSTR QuorumLogPath = NULL;
|
|
DWORD QuorumLogSize = 0;
|
|
DWORD Type;
|
|
HKEY QuorumKey;
|
|
|
|
Status = RegOpenKey( HKEY_LOCAL_MACHINE,
|
|
DISKS_REG_CLUSTER_QUORUM,
|
|
&QuorumKey );
|
|
if ( Status == ERROR_SUCCESS ) {
|
|
Status = RegQueryValueExW(QuorumKey,
|
|
DISKS_REG_QUORUM_PATH,
|
|
0,
|
|
&Type,
|
|
NULL,
|
|
&QuorumLogSize );
|
|
if ( Status != ERROR_SUCCESS ) {
|
|
return( Status );
|
|
}
|
|
|
|
if ( (Type != REG_SZ) ||
|
|
(QuorumLogSize > (MAX_PATH - 2)) ) {
|
|
return(ERROR_INVALID_DATA);
|
|
}
|
|
if ( (Status == ERROR_SUCCESS) ||
|
|
(Status == ERROR_MORE_DATA) ) {
|
|
QuorumLogPath = LocalAlloc( LMEM_FIXED,
|
|
(QuorumLogSize + 1) * sizeof(WCHAR) );
|
|
if ( QuorumLogPath == NULL ) {
|
|
RegCloseKey( QuorumKey );
|
|
return(ERROR_NOT_ENOUGH_MEMORY); // Mostly catastrophic
|
|
}
|
|
|
|
Status = RegQueryValueExW(QuorumKey,
|
|
DISKS_REG_QUORUM_PATH,
|
|
0,
|
|
&Type,
|
|
(LPBYTE)QuorumLogPath,
|
|
&QuorumLogSize );
|
|
if (Status == ERROR_SUCCESS) {
|
|
*lpQuorumLogPath = QuorumLogPath;
|
|
} else {
|
|
LocalFree(QuorumLogPath);
|
|
*lpQuorumLogPath = 0;
|
|
}
|
|
}
|
|
RegCloseKey( QuorumKey );
|
|
}
|
|
return Status;
|
|
}
|
|
|
|
DWORD
|
|
DiskspSetQuorumPath(
|
|
IN LPWSTR QuorumLogPath
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Reads QuorumPath value from the registry.
|
|
|
|
|
|
Arguments:
|
|
|
|
lpQuorumLogPath - receives the poiner to a buffer containing QuorumLogPath
|
|
the buffer needs to be deallocated later via LocalFree
|
|
|
|
ResourceEntry - receives a drive letter of the quorum
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS if successful
|
|
|
|
Win32 error code otherwise
|
|
|
|
--*/
|
|
{
|
|
DWORD status;
|
|
HKEY QuorumKey;
|
|
HKEY ClusterKey;
|
|
|
|
HCLUSTER hCluster;
|
|
|
|
hCluster = OpenCluster(NULL);
|
|
if (!hCluster) {
|
|
status = GetLastError();
|
|
return status;
|
|
}
|
|
|
|
ClusterKey = GetClusterKey(hCluster, KEY_READ | KEY_WRITE);
|
|
if (!ClusterKey) {
|
|
status = GetLastError();
|
|
CloseCluster(hCluster);
|
|
return status;
|
|
}
|
|
|
|
status = ClusterRegOpenKey( ClusterKey,
|
|
CLUSREG_KEYNAME_QUORUM,
|
|
KEY_READ | KEY_WRITE,
|
|
&QuorumKey );
|
|
if (status != ERROR_SUCCESS) {
|
|
ClusterRegCloseKey(ClusterKey);
|
|
CloseCluster(hCluster);
|
|
return status;
|
|
}
|
|
|
|
status = ResUtilSetSzValue(
|
|
QuorumKey,
|
|
CLUSREG_NAME_QUORUM_PATH,
|
|
QuorumLogPath,
|
|
0);
|
|
|
|
ClusterRegCloseKey(QuorumKey);
|
|
ClusterRegCloseKey(ClusterKey);
|
|
CloseCluster(hCluster);
|
|
return status;
|
|
}
|
|
|
|
|
|
DWORD
|
|
DisksDriveIsAlive(
|
|
IN PDISK_RESOURCE ResourceEntry,
|
|
IN BOOL Online
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Checks out a drive partition to see if the filesystem has mounted
|
|
it and it's working. We will also run CHKDSK if the partition/certain
|
|
files are Corrupt and the Online flag is TRUE.
|
|
|
|
Arguments:
|
|
|
|
ResourceEntry - Supplies a pointer to the resource entry for this disk
|
|
|
|
Online - TRUE if the disk was just brought online.
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS if successful
|
|
|
|
Win32 error code otherwise
|
|
|
|
--*/
|
|
|
|
{
|
|
PMOUNTIE_PARTITION entry;
|
|
|
|
DWORD Status = ERROR_SUCCESS;
|
|
DWORD nPartitions = MountiePartitionCount( &ResourceEntry->MountieInfo );
|
|
DWORD physicalDrive = ResourceEntry->DiskInfo.PhysicalDrive;
|
|
DWORD i;
|
|
|
|
LPWSTR QuorumLogPath = NULL;
|
|
BOOL QuorumResource = FALSE;
|
|
|
|
WCHAR szDiskPartName[MAX_PATH];
|
|
WCHAR szGlobalDiskPartName[MAX_PATH];
|
|
WCHAR szVolumeName[MAX_PATH];
|
|
WCHAR szQuorumVolumeName[MAX_PATH];
|
|
WCHAR szQuorumDriveLetter[16];
|
|
|
|
ZeroMemory( szDiskPartName, sizeof(szDiskPartName) );
|
|
ZeroMemory( szGlobalDiskPartName, sizeof(szGlobalDiskPartName) );
|
|
ZeroMemory( szVolumeName, sizeof(szVolumeName) );
|
|
ZeroMemory( szQuorumVolumeName, sizeof(szQuorumVolumeName) );
|
|
ZeroMemory( szQuorumDriveLetter, sizeof(szQuorumDriveLetter) );
|
|
|
|
//
|
|
// Find the quorum path... this is a hack!
|
|
//
|
|
if ( Online ) {
|
|
(DiskpLogEvent)(
|
|
ResourceEntry->ResourceHandle,
|
|
LOG_INFORMATION,
|
|
L"DriveIsAlive called for Online check\n" );
|
|
|
|
{
|
|
DWORD QuorumSignature;
|
|
Status = GetQuorumSignature( &QuorumSignature );
|
|
if (Status == ERROR_SUCCESS) {
|
|
QuorumResource =
|
|
(QuorumSignature == ResourceEntry->DiskInfo.Params.Signature);
|
|
} else {
|
|
(DiskpLogEvent)(
|
|
ResourceEntry->ResourceHandle,
|
|
LOG_INFORMATION,
|
|
L"GetQuorumSignature returned %1!u!\n", Status );
|
|
}
|
|
}
|
|
|
|
|
|
Status = DiskspGetQuorumPath( &QuorumLogPath );
|
|
if (Status != ERROR_SUCCESS) {
|
|
(DiskpLogEvent)(
|
|
ResourceEntry->ResourceHandle,
|
|
LOG_ERROR,
|
|
L"DiskspGetQuorumPath returned %1!u!\n", Status );
|
|
|
|
} else {
|
|
|
|
//
|
|
// For now, quorum path will have a drive letter. Get the corresponding volume name.
|
|
//
|
|
|
|
_snwprintf( szQuorumDriveLetter,
|
|
16,
|
|
L"%wc:\\",
|
|
QuorumLogPath[0] );
|
|
|
|
if ( !GetVolumeNameForVolumeMountPointW( szQuorumDriveLetter,
|
|
szQuorumVolumeName,
|
|
sizeof(szQuorumVolumeName)/sizeof(WCHAR) )) {
|
|
|
|
Status = GetLastError();
|
|
|
|
(DiskpLogEvent)(
|
|
ResourceEntry->ResourceHandle,
|
|
LOG_ERROR,
|
|
L"DriveIsAlive: GetVolumeNameForVolumeMountPoint (quorum) returned %1!u!\n", Status );
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
#if 0
|
|
(DiskpLogEvent)(
|
|
ResourceEntry->ResourceHandle,
|
|
LOG_INFORMATION,
|
|
L"DriveIsAlive is now checking each partition\n" );
|
|
#endif
|
|
|
|
//
|
|
// Now check out each interesting partition. Since only "valid" partitions are
|
|
// saved in the MountieInfo structure, we will only look at those (ignoring those
|
|
// partitions that are not NTFS).
|
|
//
|
|
|
|
for ( i = 0; i < nPartitions; ++i ) {
|
|
|
|
entry = MountiePartition( &ResourceEntry->MountieInfo, i );
|
|
|
|
if ( !entry ) {
|
|
(DiskpLogEvent)(
|
|
ResourceEntry->ResourceHandle,
|
|
LOG_ERROR,
|
|
L"DriveIsAlive no partition entry for partition %1!u! \n", i );
|
|
|
|
//
|
|
// Something bad happened to our data structures. We have to indicate that the
|
|
// drive is not alive.
|
|
//
|
|
|
|
Status = ERROR_INVALID_DATA;
|
|
|
|
break;
|
|
}
|
|
|
|
//
|
|
// Create device name of form \Device\HarddiskX\PartitionY (no trailing backslash).
|
|
//
|
|
|
|
_snwprintf( szDiskPartName,
|
|
MAX_PATH,
|
|
DEVICE_HARDDISK_PARTITION_FMT,
|
|
physicalDrive,
|
|
entry->PartitionNumber );
|
|
|
|
//
|
|
// Given the DiskPartName, get the VolGuid name. This name must have a trailing
|
|
// backslash to work correctly.
|
|
//
|
|
|
|
_snwprintf( szGlobalDiskPartName,
|
|
MAX_PATH,
|
|
GLOBALROOT_HARDDISK_PARTITION_FMT,
|
|
physicalDrive,
|
|
entry->PartitionNumber );
|
|
|
|
if ( !GetVolumeNameForVolumeMountPointW( szGlobalDiskPartName,
|
|
szVolumeName,
|
|
sizeof(szVolumeName)/sizeof(WCHAR) )) {
|
|
|
|
Status = GetLastError();
|
|
|
|
(DiskpLogEvent)(
|
|
ResourceEntry->ResourceHandle,
|
|
LOG_ERROR,
|
|
L"DriveIsAlive: GetVolumeNameForVolumeMountPoint for %1!ws! returned %2!u!\n",
|
|
szGlobalDiskPartName,
|
|
Status );
|
|
|
|
//
|
|
// If disk is not corrupt, exit. If disk is corrupt, fall through so chkdsk runs.
|
|
//
|
|
|
|
if ( ERROR_DISK_CORRUPT != Status && ERROR_FILE_CORRUPT != Status ) {
|
|
|
|
//
|
|
// Something bad happened. We have to stop checking this disk. Return the
|
|
// error status we received.
|
|
//
|
|
|
|
break;
|
|
}
|
|
|
|
}
|
|
|
|
//
|
|
// Simple algorithm used here is to do a FindFirstFile on X:\* and see
|
|
// if it works. Then we open each file for read access. This is the
|
|
// cluster directory, and all files in it are subject to our opening.
|
|
//
|
|
|
|
Status = DiskspCheckPath( szVolumeName,
|
|
ResourceEntry,
|
|
FALSE,
|
|
Online );
|
|
|
|
//
|
|
// [HACKHACK] Ignore error 21 during periodic IsAlive/LooksAlive
|
|
//
|
|
if ( !Online && (Status == ERROR_NOT_READY) ) {
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle,
|
|
LOG_WARNING,
|
|
L"DiskpCheckPath for %1!ws!: returned status = %2!u! (ChkDsk running?)\n",
|
|
szVolumeName,
|
|
Status );
|
|
Status = ERROR_SUCCESS;
|
|
}
|
|
|
|
// if we haven't chkdsk'd yet, keep looking.
|
|
if ( Status != ERROR_SUCCESS ) {
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle,
|
|
LOG_ERROR,
|
|
L"DiskpCheckPath for %1!ws!: returned status = %2!u!\n",
|
|
szVolumeName,
|
|
Status );
|
|
}
|
|
|
|
if ( (Status == ERROR_SUCCESS) && Online &&
|
|
QuorumLogPath && QuorumResource &&
|
|
( wcslen( szVolumeName ) == wcslen( szQuorumVolumeName ) ) &&
|
|
( !wcsncmp( szVolumeName, szQuorumVolumeName, wcslen( szQuorumVolumeName ) ))) {
|
|
|
|
(DiskpLogEvent)(
|
|
ResourceEntry->ResourceHandle,
|
|
LOG_INFORMATION,
|
|
L"DriveIsAlive checking quorum drive \n" );
|
|
|
|
//
|
|
// Everything looks fine... if this is the quorum device, then
|
|
// we should check the quorum log path if given
|
|
//
|
|
|
|
Status = DiskspCheckPath( szQuorumVolumeName,
|
|
ResourceEntry,
|
|
TRUE,
|
|
Online );
|
|
}
|
|
|
|
if ( (Status == ERROR_SUCCESS) &&
|
|
Online ) {
|
|
|
|
//
|
|
// Check if the volume dirty bit is on
|
|
//
|
|
Status = DisksCheckCorruption( szDiskPartName,
|
|
szVolumeName,
|
|
ResourceEntry );
|
|
|
|
//
|
|
// If we're requested to shutdown, then do that immediately
|
|
if ( ResourceEntry->OnlineThread.Terminate ) {
|
|
Status = ERROR_SHUTDOWN_CLUSTER;
|
|
break;
|
|
}
|
|
}
|
|
|
|
|
|
if ( (Status != ERROR_SUCCESS) && Online) {
|
|
if ( ResourceEntry->DiskInfo.Params.ConditionalMount ) {
|
|
|
|
Status = DisksFixCorruption( szVolumeName,
|
|
ResourceEntry,
|
|
ERROR_DISK_CORRUPT );
|
|
|
|
//
|
|
// Since ConditionalMount is set, if we couldn't fix the corruption
|
|
// on the disk, we don't want to continue checking the other
|
|
// partitions - we want to return an error. So we fall through
|
|
// and check the status. If status wasn't successful, we break out
|
|
// of the loop to return the error.
|
|
//
|
|
|
|
} else {
|
|
if ( QuorumLogPath ) {
|
|
LocalFree( QuorumLogPath );
|
|
}
|
|
return( Status );
|
|
}
|
|
}
|
|
if ( Status != ERROR_SUCCESS ) {
|
|
break;
|
|
}
|
|
|
|
}
|
|
|
|
if ( QuorumLogPath ) {
|
|
LocalFree( QuorumLogPath );
|
|
}
|
|
|
|
return(Status);
|
|
|
|
} // DisksDriveIsAlive
|
|
|
|
|
|
|
|
DWORD
|
|
DisksMountDrives(
|
|
IN PDISK_INFO DiskInfo,
|
|
IN PDISK_RESOURCE ResourceEntry,
|
|
IN DWORD Signature
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
For each drive letter on the supplied disk, this mounts the filesystem
|
|
and checks it out.
|
|
|
|
Arguments:
|
|
|
|
DiskInfo - Supplies the disk information
|
|
|
|
ResourceEntry - Supplies a pointer to the disk resource
|
|
|
|
Signature - the signature for the disk.
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS if successful
|
|
|
|
Win32 error code otherwise
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD Status;
|
|
WCHAR wDeviceName[4];
|
|
WCHAR wShareName[4];
|
|
DWORD letterMask;
|
|
UCHAR index;
|
|
UCHAR driveLetter;
|
|
SHARE_INFO_2 shareInfo;
|
|
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle,
|
|
LOG_INFORMATION,
|
|
L"DisksMountDrives: calling IsAlive function.\n" );
|
|
|
|
//
|
|
// Call the IsAlive to see if the filesystem checks out ok.
|
|
//
|
|
Status = DisksDriveIsAlive( ResourceEntry,
|
|
TRUE);
|
|
|
|
if ( (Status != ERROR_SUCCESS) ||
|
|
(ResourceEntry->OnlineThread.Terminate) ) {
|
|
return(Status);
|
|
}
|
|
|
|
//
|
|
// Now create the drive$ share name for each drive letter;
|
|
//
|
|
letterMask = DisksGetLettersForSignature(
|
|
ResourceEntry);
|
|
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle,
|
|
LOG_INFORMATION,
|
|
L"DisksMountDrives: letter mask is %1!08x!.\n",
|
|
letterMask);
|
|
|
|
index = 0;
|
|
while ( letterMask ) {
|
|
while ( !(letterMask & 1) ) {
|
|
letterMask = letterMask >> 1;
|
|
index++;
|
|
}
|
|
driveLetter = 'A' + index;
|
|
letterMask = letterMask >> 1;
|
|
index++;
|
|
if ( isalpha(driveLetter) ) {
|
|
wsprintfW(wDeviceName, L"%c:\\", driveLetter);
|
|
wsprintfW(wShareName, L"%c$", driveLetter);
|
|
shareInfo.shi2_netname = wShareName;
|
|
shareInfo.shi2_type = STYPE_DISKTREE;
|
|
shareInfo.shi2_remark = NULL;
|
|
shareInfo.shi2_permissions = 0;
|
|
shareInfo.shi2_max_uses = (DWORD)-1;
|
|
shareInfo.shi2_current_uses = 0;
|
|
shareInfo.shi2_path = wDeviceName;
|
|
shareInfo.shi2_passwd = NULL;
|
|
|
|
Status = NetShareAdd( NULL, 2, (PBYTE)&shareInfo, NULL );
|
|
if ( Status != ERROR_SUCCESS && Status != NERR_DuplicateShare ) {
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle,
|
|
LOG_ERROR,
|
|
L"DisksMountDrives, error creating default share %1!ws!. Error: %2!u!.\n",
|
|
wShareName,
|
|
Status);
|
|
}
|
|
}
|
|
}
|
|
|
|
return(ERROR_SUCCESS);
|
|
|
|
} // DisksMountDrives
|
|
|
|
|
|
DWORD
|
|
DisksDismountDrive(
|
|
IN PDISK_RESOURCE ResourceEntry,
|
|
IN DWORD Signature
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Desccription:
|
|
|
|
Delete the default device share names for a given disk.
|
|
|
|
Arguments:
|
|
|
|
ResourceHandle - the resource handle for logging events
|
|
|
|
Signature - the disk's signature
|
|
|
|
Return Value:
|
|
|
|
WIN32 error code.
|
|
|
|
--*/
|
|
|
|
{
|
|
WCHAR shareName[8];
|
|
DWORD letterMask;
|
|
UCHAR index;
|
|
UCHAR driveLetter;
|
|
|
|
letterMask = DisksGetLettersForSignature(
|
|
ResourceEntry);
|
|
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle,
|
|
LOG_INFORMATION,
|
|
L"DisksDismountDrives: letter mask is %1!08x!.\n",
|
|
letterMask);
|
|
|
|
index = 0;
|
|
while ( letterMask ) {
|
|
while ( !(letterMask & 1) ) {
|
|
letterMask = letterMask >> 1;
|
|
index++;
|
|
}
|
|
driveLetter = 'A' + index;
|
|
letterMask = letterMask >> 1;
|
|
index++;
|
|
if ( isalpha(driveLetter) ) {
|
|
shareName[0] = (WCHAR)driveLetter;
|
|
shareName[1] = (WCHAR)'$';
|
|
shareName[2] = (WCHAR)0;
|
|
|
|
NetShareDel( NULL,
|
|
shareName,
|
|
0 );
|
|
|
|
}
|
|
}
|
|
|
|
return (ERROR_SUCCESS);
|
|
|
|
} // DisksDismountDrive
|
|
|
|
|
|
|
|
LPSTR
|
|
GetRegParameter(
|
|
IN HKEY RegKey,
|
|
IN LPCSTR ValueName
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Queries a REG_SZ parameter out of the registry and allocates the
|
|
necessary storage for it.
|
|
|
|
Arguments:
|
|
|
|
RegKey - Supplies the cluster key where the parameter is stored
|
|
|
|
ValueName - Supplies the name of the value.
|
|
|
|
Return Value:
|
|
|
|
A pointer to a buffer containing the parameter if successful.
|
|
|
|
NULL if unsuccessful.
|
|
|
|
--*/
|
|
|
|
{
|
|
LPSTR Value;
|
|
DWORD ValueLength;
|
|
DWORD ValueType;
|
|
DWORD Status;
|
|
|
|
ValueLength = 0;
|
|
Status = RegQueryValueEx(RegKey,
|
|
ValueName,
|
|
NULL,
|
|
&ValueType,
|
|
NULL,
|
|
&ValueLength);
|
|
if ( (Status != ERROR_SUCCESS) &&
|
|
(Status != ERROR_MORE_DATA) ) {
|
|
return(NULL);
|
|
}
|
|
if ( ValueType == REG_SZ ) {
|
|
ValueLength++;
|
|
}
|
|
Value = LocalAlloc(LMEM_FIXED, ValueLength);
|
|
if (Value == NULL) {
|
|
return(NULL);
|
|
}
|
|
Status = RegQueryValueEx(RegKey,
|
|
ValueName,
|
|
NULL,
|
|
&ValueType,
|
|
(LPBYTE)Value,
|
|
&ValueLength);
|
|
if (Status != ERROR_SUCCESS) {
|
|
LocalFree(Value);
|
|
Value = NULL;
|
|
}
|
|
|
|
return(Value);
|
|
|
|
} // GetRegParameter
|
|
|
|
|
|
DWORD
|
|
DisksResourceControl(
|
|
IN RESID Resource,
|
|
IN DWORD ControlCode,
|
|
IN PVOID InBuffer,
|
|
IN DWORD InBufferSize,
|
|
OUT PVOID OutBuffer,
|
|
IN DWORD OutBufferSize,
|
|
OUT LPDWORD BytesReturned
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Arguments:
|
|
|
|
Return Value:
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD status = ERROR_SUCCESS;
|
|
PDISK_RESOURCE resourceEntry;
|
|
DWORD required;
|
|
|
|
ULONG uResId=PtrToUlong(Resource);
|
|
|
|
//
|
|
// Make sure the RESID is okay.
|
|
//
|
|
|
|
resourceEntry = DiskTable[uResId];
|
|
if ( resourceEntry == NULL ) {
|
|
DISKS_PRINT("ResourceControl, bad resource id, %u\n", uResId);
|
|
return(ERROR_RESOURCE_NOT_FOUND);
|
|
}
|
|
|
|
switch ( ControlCode ) {
|
|
|
|
case CLUSCTL_RESOURCE_UNKNOWN:
|
|
*BytesReturned = 0;
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_GET_PRIVATE_PROPERTY_FMTS:
|
|
status = ResUtilGetPropertyFormats( DiskResourcePrivateProperties,
|
|
OutBuffer,
|
|
OutBufferSize,
|
|
BytesReturned,
|
|
&required );
|
|
if ( status == ERROR_MORE_DATA ) {
|
|
*BytesReturned = required;
|
|
}
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_GET_CLASS_INFO:
|
|
*BytesReturned = sizeof(CLUS_RESOURCE_CLASS_INFO);
|
|
if ( OutBufferSize < sizeof(CLUS_RESOURCE_CLASS_INFO) ) {
|
|
status = ERROR_MORE_DATA;
|
|
} else {
|
|
PCLUS_RESOURCE_CLASS_INFO ptrResClassInfo = OutBuffer;
|
|
ptrResClassInfo->rc = CLUS_RESCLASS_STORAGE;
|
|
ptrResClassInfo->SubClass = (DWORD) CLUS_RESSUBCLASS_SHARED;
|
|
}
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_STORAGE_GET_DISK_INFO:
|
|
status = GetDiskInfo( resourceEntry->DiskInfo.Params.Signature,
|
|
&OutBuffer,
|
|
OutBufferSize,
|
|
BytesReturned,
|
|
FALSE );
|
|
|
|
// Add the endmark.
|
|
if ( OutBufferSize > *BytesReturned ) {
|
|
OutBufferSize -= *BytesReturned;
|
|
} else {
|
|
OutBufferSize = 0;
|
|
}
|
|
*BytesReturned += sizeof(CLUSPROP_SYNTAX);
|
|
if ( OutBufferSize >= sizeof(CLUSPROP_SYNTAX) ) {
|
|
PCLUSPROP_SYNTAX ptrSyntax = OutBuffer;
|
|
ptrSyntax->dw = CLUSPROP_SYNTAX_ENDMARK;
|
|
}
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_ENUM_PRIVATE_PROPERTIES:
|
|
status = ResUtilEnumProperties( DiskResourcePrivateProperties,
|
|
OutBuffer,
|
|
OutBufferSize,
|
|
BytesReturned,
|
|
&required );
|
|
if ( status == ERROR_MORE_DATA ) {
|
|
*BytesReturned = required;
|
|
}
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_GET_PRIVATE_PROPERTIES:
|
|
status = DisksGetPrivateResProperties( resourceEntry,
|
|
OutBuffer,
|
|
OutBufferSize,
|
|
BytesReturned );
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_VALIDATE_PRIVATE_PROPERTIES:
|
|
status = DisksValidatePrivateResProperties( resourceEntry,
|
|
InBuffer,
|
|
InBufferSize,
|
|
NULL );
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_SET_PRIVATE_PROPERTIES:
|
|
status = DisksSetPrivateResProperties( resourceEntry,
|
|
InBuffer,
|
|
InBufferSize );
|
|
#if 0
|
|
status = ClusterRegQueryValue(resourceEntry->ResourceParametersKey,
|
|
DISKS_DISK_INFO,
|
|
NULL,
|
|
NULL,
|
|
&required);
|
|
if (status == ERROR_FILE_NOT_FOUND ) {
|
|
// If not found, then make it.
|
|
PFT_INFO diskInfo;
|
|
diskInfo = DiskGetFtInfo();
|
|
if ( diskInfo ) {
|
|
status = DiskspRestoreState( resourceEntry,
|
|
diskInfo );
|
|
DiskFreeFtInfo( diskInfo );
|
|
}
|
|
}
|
|
#else
|
|
DiskspSsyncDiskInfo(L"ResourceControl", resourceEntry, 0);
|
|
#endif
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_DELETE:
|
|
if ( resourceEntry->DiskInfo.Params.Signature ) {
|
|
(DiskpLogEvent)(
|
|
resourceEntry->ResourceHandle,
|
|
LOG_INFORMATION,
|
|
L"Delete disk resource %1!lx!\n",
|
|
resourceEntry->DiskInfo.Params.Signature );
|
|
status = DoDetach( resourceEntry->DiskInfo.Params.Signature,
|
|
resourceEntry->ResourceHandle );
|
|
}
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_GET_CHARACTERISTICS:
|
|
*BytesReturned = sizeof(DWORD);
|
|
if ( OutBufferSize < sizeof(DWORD) ) {
|
|
status = ERROR_MORE_DATA;
|
|
} else {
|
|
LPDWORD ptrDword = OutBuffer;
|
|
*ptrDword = CLUS_CHAR_QUORUM | CLUS_CHAR_DELETE_REQUIRES_ALL_NODES;
|
|
}
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_STORAGE_DLL_EXTENSION:
|
|
|
|
status = ProcessDllExtension( resourceEntry,
|
|
InBuffer,
|
|
InBufferSize,
|
|
OutBuffer,
|
|
OutBufferSize,
|
|
BytesReturned );
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_REMOVE_DEPENDENCY:
|
|
case CLUSCTL_RESOURCE_ADD_DEPENDENCY:
|
|
|
|
(DiskpLogEvent)(
|
|
resourceEntry->ResourceHandle,
|
|
LOG_INFORMATION,
|
|
L"Add/Remove dependency: source signature %1!lx! target name (%2!ws!) \n",
|
|
resourceEntry->DiskInfo.Params.Signature,
|
|
InBuffer );
|
|
|
|
status = DisksProcessMPControlCode( resourceEntry,
|
|
ControlCode );
|
|
|
|
break;
|
|
|
|
default:
|
|
status = ERROR_INVALID_FUNCTION;
|
|
break;
|
|
}
|
|
|
|
return(status);
|
|
|
|
} // DisksResourceControl
|
|
|
|
|
|
|
|
DWORD
|
|
DisksResourceTypeControl(
|
|
IN LPCWSTR ResourceTypeName,
|
|
IN DWORD ControlCode,
|
|
IN PVOID InBuffer,
|
|
IN DWORD InBufferSize,
|
|
OUT PVOID OutBuffer,
|
|
IN DWORD OutBufferSize,
|
|
OUT LPDWORD BytesReturned
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Process control requests for this resource type.
|
|
|
|
Arguments:
|
|
|
|
ResourceTypeName - the name of the resource type - not very useful!
|
|
|
|
ControlCode - the control request
|
|
|
|
InBuffer - pointer to the input buffer
|
|
|
|
InBufferSize - the size of the input buffer
|
|
|
|
OutBuffer - pointer to the output buffer
|
|
|
|
OutBufferSize - the size of the output buffer
|
|
|
|
BytesReturned - the number of bytes returned (or needed if larger than
|
|
OutBufferSize and ERROR_MORE_DATA is returned
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS if successful
|
|
|
|
A WIN32 error on failure
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD status = ERROR_SUCCESS;
|
|
DWORD required;
|
|
|
|
switch ( ControlCode ) {
|
|
|
|
case CLUSCTL_RESOURCE_TYPE_UNKNOWN:
|
|
*BytesReturned = 0;
|
|
status = ERROR_SUCCESS;
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_TYPE_GET_PRIVATE_RESOURCE_PROPERTY_FMTS:
|
|
status = ResUtilGetPropertyFormats( DiskResourcePrivateProperties,
|
|
OutBuffer,
|
|
OutBufferSize,
|
|
BytesReturned,
|
|
&required );
|
|
if ( status == ERROR_MORE_DATA ) {
|
|
*BytesReturned = required;
|
|
}
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_TYPE_HOLD_IO:
|
|
DoHoldIO();
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_TYPE_RESUME_IO:
|
|
DoResumeIO();
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_TYPE_ENUM_PRIVATE_PROPERTIES:
|
|
status = ResUtilEnumProperties( DiskResourcePrivateProperties,
|
|
OutBuffer,
|
|
OutBufferSize,
|
|
BytesReturned,
|
|
&required );
|
|
if ( status == ERROR_MORE_DATA ) {
|
|
*BytesReturned = required;
|
|
}
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_TYPE_GET_CLASS_INFO:
|
|
*BytesReturned = sizeof(CLUS_RESOURCE_CLASS_INFO);
|
|
if ( OutBufferSize < sizeof(CLUS_RESOURCE_CLASS_INFO) ) {
|
|
status = ERROR_MORE_DATA;
|
|
} else {
|
|
PCLUS_RESOURCE_CLASS_INFO ptrResClassInfo = OutBuffer;
|
|
ptrResClassInfo->rc = CLUS_RESCLASS_STORAGE;
|
|
ptrResClassInfo->SubClass = (DWORD) CLUS_RESSUBCLASS_SHARED;
|
|
status = ERROR_SUCCESS;
|
|
}
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_TYPE_STORAGE_GET_AVAILABLE_DISKS:
|
|
status = ClusDiskGetAvailableDisks( OutBuffer,
|
|
OutBufferSize,
|
|
BytesReturned,
|
|
FALSE );
|
|
break;
|
|
|
|
case CLUSCTL_RESOURCE_TYPE_GET_CHARACTERISTICS:
|
|
*BytesReturned = sizeof(DWORD);
|
|
if ( OutBufferSize < sizeof(DWORD) ) {
|
|
status = ERROR_MORE_DATA;
|
|
} else {
|
|
LPDWORD ptrDword = OutBuffer;
|
|
*ptrDword = CLUS_CHAR_QUORUM | CLUS_CHAR_DELETE_REQUIRES_ALL_NODES;
|
|
}
|
|
break;
|
|
|
|
default:
|
|
status = ERROR_INVALID_FUNCTION;
|
|
break;
|
|
}
|
|
|
|
return(status);
|
|
|
|
} // DisksResourceTypeControl
|
|
|
|
|
|
|
|
DWORD
|
|
DisksGetPrivateResProperties(
|
|
IN OUT PDISK_RESOURCE ResourceEntry,
|
|
OUT PVOID OutBuffer,
|
|
IN DWORD OutBufferSize,
|
|
OUT LPDWORD BytesReturned
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Processes the CLUSCTL_RESOURCE_GET_PRIVATE_PROPERTIES control function
|
|
for resources of type Physical Disk.
|
|
|
|
Arguments:
|
|
|
|
ResourceEntry - Supplies the resource entry on which to operate.
|
|
|
|
OutBuffer - Returns the output data.
|
|
|
|
OutBufferSize - Supplies the size, in bytes, of the data pointed
|
|
to by OutBuffer.
|
|
|
|
BytesReturned - The number of bytes returned in OutBuffer.
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS - The function completed successfully.
|
|
|
|
ERROR_INVALID_PARAMETER - The data is formatted incorrectly.
|
|
|
|
ERROR_NOT_ENOUGH_MEMORY - An error occurred allocating memory.
|
|
|
|
Win32 error code - The function failed.
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD status;
|
|
DWORD required;
|
|
|
|
status = ResUtilGetAllProperties(
|
|
ResourceEntry->ResourceParametersKey,
|
|
DiskResourcePrivateProperties,
|
|
OutBuffer,
|
|
OutBufferSize,
|
|
BytesReturned,
|
|
&required );
|
|
if ( status == ERROR_MORE_DATA ) {
|
|
*BytesReturned = required;
|
|
}
|
|
|
|
return(status);
|
|
|
|
} // DisksGetPrivateResProperties
|
|
|
|
|
|
|
|
DWORD
|
|
DisksConvertDriveToSignature(
|
|
IN OUT PDISK_PARAMS Params,
|
|
IN RESOURCE_HANDLE ResourceHandle
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Change the drive letter or VolGuid to the associated disk signature.
|
|
|
|
Arguments:
|
|
|
|
Params - Supplies the parameter block that contains the drive letter
|
|
(or VolGuid) to convert to a signature.
|
|
|
|
VolGuid syntax is returned from mountvol.exe and in the form:
|
|
\\?\Volume{e6de97f1-6f97-11d3-bb7f-806d6172696f}\
|
|
|
|
ResourceHandle -
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS - The function completed successfully.
|
|
|
|
ERROR_INVALID_PARAMETER - The data is formatted incorrectly or is invalid.
|
|
|
|
ERROR_NOT_ENOUGH_MEMORY - An error occurred allocating memory.
|
|
|
|
Win32 error code - The function failed.
|
|
|
|
--*/
|
|
|
|
{
|
|
PWCHAR deviceName = NULL;
|
|
DWORD deviceNameBytes;
|
|
DWORD lenChar;
|
|
DWORD status = ERROR_SUCCESS;
|
|
LPWSTR drive = Params->Drive;
|
|
LPWSTR volGuid = Params->VolGuid;
|
|
HANDLE fileHandle;
|
|
BOOL success;
|
|
PDRIVE_LAYOUT_INFORMATION driveLayout = NULL;
|
|
|
|
//
|
|
// Build device string for CreateFile
|
|
//
|
|
|
|
if ( drive ) {
|
|
|
|
deviceNameBytes = (( wcslen( drive ) + wcslen( L"\\\\.\\" ) ) * sizeof(WCHAR) ) +
|
|
sizeof( UNICODE_NULL );
|
|
|
|
deviceName = LocalAlloc( LPTR, deviceNameBytes );
|
|
|
|
if ( !deviceName ) {
|
|
goto FnExit;
|
|
}
|
|
|
|
_snwprintf( deviceName,
|
|
deviceNameBytes/sizeof(WCHAR),
|
|
L"\\\\.\\%ws",
|
|
drive );
|
|
|
|
} else if ( volGuid ) {
|
|
|
|
deviceNameBytes = ( wcslen( volGuid ) * sizeof( WCHAR ) ) + sizeof( UNICODE_NULL );
|
|
|
|
deviceName = LocalAlloc( LPTR, deviceNameBytes );
|
|
|
|
if ( !deviceName ) {
|
|
goto FnExit;
|
|
}
|
|
|
|
CopyMemory( deviceName,
|
|
volGuid,
|
|
deviceNameBytes );
|
|
|
|
//
|
|
// If user specified \\?\Volume{guid}\ with trailing backslash, we need to get
|
|
// rid of the backslash.
|
|
//
|
|
|
|
lenChar = wcslen( deviceName );
|
|
|
|
if ( lenChar > 1 && L'\\' == deviceName[lenChar-1] ) {
|
|
deviceName[lenChar-1] = L'\0';
|
|
}
|
|
|
|
|
|
} else {
|
|
goto FnExit;
|
|
}
|
|
|
|
fileHandle = CreateFileW( deviceName,
|
|
GENERIC_READ | GENERIC_WRITE,
|
|
FILE_SHARE_READ | FILE_SHARE_WRITE,
|
|
NULL,
|
|
OPEN_EXISTING,
|
|
0,
|
|
NULL );
|
|
if ( (fileHandle == INVALID_HANDLE_VALUE) ||
|
|
(fileHandle == NULL) ) {
|
|
status = GetLastError();
|
|
(DiskpLogEvent)(
|
|
ResourceHandle,
|
|
LOG_ERROR,
|
|
L"ConvertDriveToSignature, error opening device '%1!ws!'. Error: %2!u!\n",
|
|
deviceName,
|
|
status );
|
|
goto FnExit;
|
|
}
|
|
|
|
//
|
|
// Get drive layout - in order to get disk signature.
|
|
//
|
|
success = ClRtlGetDriveLayoutTable( fileHandle, &driveLayout, NULL );
|
|
|
|
if ( success &&
|
|
driveLayout->Signature ) {
|
|
Params->Signature = driveLayout->Signature;
|
|
|
|
if ( Params->Drive ) {
|
|
LocalFree( Params->Drive );
|
|
Params->Drive = NULL;
|
|
}
|
|
|
|
if ( Params->VolGuid ) {
|
|
LocalFree( Params->VolGuid );
|
|
Params->VolGuid = NULL;
|
|
}
|
|
|
|
} else {
|
|
status = ERROR_FILE_NOT_FOUND;
|
|
}
|
|
|
|
if ( driveLayout ) {
|
|
LocalFree( driveLayout );
|
|
}
|
|
CloseHandle( fileHandle );
|
|
|
|
FnExit:
|
|
|
|
if ( deviceName ) {
|
|
LocalFree( deviceName );
|
|
}
|
|
|
|
return(status);
|
|
|
|
} // DisksConvertDriveToSignature
|
|
|
|
|
|
DWORD
|
|
DisksValidatePrivateResProperties(
|
|
IN OUT PDISK_RESOURCE ResourceEntry,
|
|
IN PVOID InBuffer,
|
|
IN DWORD InBufferSize,
|
|
OUT PDISK_PARAMS Params
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Processes the CLUSCTL_RESOURCE_VALIDATE_PRIVATE_PROPERTIES control
|
|
function for resources of type Physical Disk.
|
|
|
|
Arguments:
|
|
|
|
ResourceEntry - Supplies the resource entry on which to operate.
|
|
|
|
InBuffer - Supplies a pointer to a buffer containing input data.
|
|
|
|
InBufferSize - Supplies the size, in bytes, of the data pointed
|
|
to by InBuffer.
|
|
|
|
Params - Supplies the parameter block to fill in.
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS - The function completed successfully.
|
|
|
|
ERROR_INVALID_PARAMETER - The data is formatted incorrectly or is invalid.
|
|
|
|
ERROR_NOT_ENOUGH_MEMORY - An error occurred allocating memory.
|
|
|
|
Win32 error code - The function failed.
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD status;
|
|
DWORD enableSanBoot;
|
|
DWORD diskNumber;
|
|
DWORD oldSerNumLen = 0;
|
|
DWORD newSerNumLen = 0;
|
|
DISK_PARAMS params;
|
|
PDISK_PARAMS pParams;
|
|
PDISK_PARAMS currentParams = &ResourceEntry->DiskInfo.Params;
|
|
|
|
SCSI_ADDRESS scsiAddress;
|
|
CLUSPROP_SCSI_ADDRESS clusScsiAddress;
|
|
|
|
//
|
|
// Check if there is input data.
|
|
//
|
|
if ( (InBuffer == NULL) ||
|
|
(InBufferSize < sizeof(DWORD)) ) {
|
|
return(ERROR_INVALID_DATA);
|
|
}
|
|
|
|
//
|
|
// Duplicate the resource parameter block.
|
|
//
|
|
if ( Params == NULL ) {
|
|
pParams = ¶ms;
|
|
} else {
|
|
pParams = Params;
|
|
}
|
|
ZeroMemory( pParams, sizeof(DISK_PARAMS) );
|
|
status = ResUtilDupParameterBlock( (LPBYTE) pParams,
|
|
(LPBYTE) currentParams,
|
|
DiskResourcePrivateProperties );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
return(status);
|
|
}
|
|
|
|
//
|
|
// Parse and validate the properties. Accept the alternate name here.
|
|
//
|
|
status = ResUtilVerifyPropertyTable( DiskResourcePrivatePropertiesAlt,
|
|
NULL,
|
|
TRUE, // Allow unknowns
|
|
InBuffer,
|
|
InBufferSize,
|
|
(LPBYTE) pParams );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
goto FnExit;
|
|
}
|
|
|
|
//
|
|
// First make sure there are no bogus properties - i.e. we don't allow
|
|
// specifying both the signature and the drive in the same request.
|
|
// We also don't allow specifying the VolGuid and signature in the same
|
|
// request.
|
|
//
|
|
if ( (pParams->Drive || pParams->VolGuid) && pParams->Signature ) {
|
|
status = ERROR_INVALID_PARAMETER;
|
|
goto FnExit;
|
|
}
|
|
|
|
if ( pParams->Drive || pParams->VolGuid ) {
|
|
//
|
|
// Convert from drive to signature.
|
|
//
|
|
status = DisksConvertDriveToSignature( pParams,
|
|
ResourceEntry->ResourceHandle );
|
|
}
|
|
|
|
if ( status != ERROR_SUCCESS ) {
|
|
goto FnExit;
|
|
}
|
|
|
|
//
|
|
// Verify the new serial number is valid and not overwriting an existing
|
|
// serial number.
|
|
//
|
|
|
|
if ( pParams->SerialNumber ) {
|
|
newSerNumLen = wcslen( pParams->SerialNumber );
|
|
}
|
|
|
|
if ( currentParams->SerialNumber ) {
|
|
oldSerNumLen = wcslen( currentParams->SerialNumber );
|
|
}
|
|
|
|
//
|
|
// If there was an old serial number, make sure the new serial number is
|
|
// the same.
|
|
//
|
|
|
|
if ( oldSerNumLen &&
|
|
( oldSerNumLen != newSerNumLen ||
|
|
0 != wcsncmp( currentParams->SerialNumber, pParams->SerialNumber, newSerNumLen ) ) ) {
|
|
status = ERROR_INVALID_PARAMETER;
|
|
goto FnExit;
|
|
}
|
|
|
|
//
|
|
// Check serial number and signature.
|
|
//
|
|
|
|
if ( 0 == oldSerNumLen &&
|
|
pParams->SerialNumber &&
|
|
0 == currentParams->Signature ) {
|
|
|
|
//
|
|
// New serial number specified and no current signature.
|
|
//
|
|
|
|
if ( 0 == pParams->Signature ) {
|
|
|
|
//
|
|
// No new signature specified, use the new serial number to
|
|
// find the new signature.
|
|
//
|
|
|
|
status = GetSignatureFromSerialNumber( pParams->SerialNumber,
|
|
&pParams->Signature );
|
|
|
|
if ( status != ERROR_SUCCESS ) {
|
|
goto FnExit;
|
|
}
|
|
|
|
} else {
|
|
|
|
//
|
|
// New signature and new serial number specified. Fail the request.
|
|
//
|
|
|
|
status = ERROR_INVALID_PARAMETER;
|
|
goto FnExit;
|
|
}
|
|
}
|
|
|
|
//
|
|
// Validate the parameter values.
|
|
//
|
|
// Make sure the disk signature is not zero.
|
|
//
|
|
|
|
if ( 0 == pParams->Signature ) {
|
|
status = ERROR_INVALID_PARAMETER;
|
|
goto FnExit;
|
|
}
|
|
|
|
// At this point, we have a valid disk signature.
|
|
|
|
//
|
|
// If no serial number, get it.
|
|
//
|
|
|
|
if ( !pParams->SerialNumber || newSerNumLen <= 1 ) {
|
|
|
|
status = GetSerialNumber( pParams->Signature,
|
|
&pParams->SerialNumber );
|
|
}
|
|
|
|
if ( ERROR_SUCCESS != status ) {
|
|
goto FnExit;
|
|
}
|
|
|
|
|
|
ZeroMemory( &clusScsiAddress, sizeof(clusScsiAddress) );
|
|
status = GetScsiAddress( pParams->Signature, &clusScsiAddress.dw, &diskNumber );
|
|
|
|
if ( ERROR_SUCCESS != status ) {
|
|
status = ERROR_INVALID_PARAMETER;
|
|
goto FnExit;
|
|
}
|
|
|
|
scsiAddress.Length = sizeof(SCSI_ADDRESS);
|
|
scsiAddress.PortNumber = clusScsiAddress.PortNumber;
|
|
scsiAddress.PathId = clusScsiAddress.PathId;
|
|
scsiAddress.TargetId = clusScsiAddress.TargetId;
|
|
scsiAddress.Lun = clusScsiAddress.Lun;
|
|
|
|
//
|
|
// Make sure the SCSI address is not system disk.
|
|
//
|
|
|
|
enableSanBoot = 0;
|
|
GetRegDwordValue( CLUSREG_KEYNAME_CLUSSVC_PARAMETERS,
|
|
CLUSREG_VALUENAME_MANAGEDISKSONSYSTEMBUSES,
|
|
&enableSanBoot );
|
|
|
|
|
|
if ( !enableSanBoot ) {
|
|
|
|
//
|
|
// Signature is valid if:
|
|
// - the signature is for a disk not on system bus
|
|
//
|
|
|
|
if ( IsDiskOnSystemBus( &scsiAddress ) ) {
|
|
status = ERROR_INVALID_PARAMETER;
|
|
goto FnExit;
|
|
}
|
|
|
|
} else {
|
|
|
|
// Allow disks on system bus to be added to cluster.
|
|
|
|
//
|
|
// Signature is valid if:
|
|
// - the signature is not for the system disk
|
|
|
|
if ( IsDiskSystemDisk( &scsiAddress ) ) {
|
|
status = ERROR_INVALID_PARAMETER;
|
|
goto FnExit;
|
|
}
|
|
|
|
}
|
|
|
|
//
|
|
// For now, we don't allow setting mount point volume GUIDs this way.
|
|
// This is a multi SZ string, so use memcmp to skip past each string's
|
|
// terminating NULL.
|
|
//
|
|
|
|
if ( ( currentParams->MPVolGuidsSize != pParams->MPVolGuidsSize ) ||
|
|
( 0 != memcmp( currentParams->MPVolGuids, pParams->MPVolGuids, currentParams->MPVolGuidsSize ) ) ) {
|
|
|
|
status = ERROR_INVALID_PARAMETER;
|
|
goto FnExit;
|
|
}
|
|
|
|
FnExit:
|
|
|
|
//
|
|
// Cleanup our parameter block.
|
|
//
|
|
if ( pParams == ¶ms ) {
|
|
ResUtilFreeParameterBlock( (LPBYTE) ¶ms,
|
|
(LPBYTE) &ResourceEntry->DiskInfo.Params,
|
|
DiskResourcePrivateProperties );
|
|
}
|
|
|
|
return(status);
|
|
|
|
} // DisksValidatePrivateResProperties
|
|
|
|
|
|
|
|
DWORD
|
|
DisksSetPrivateResProperties(
|
|
IN OUT PDISK_RESOURCE ResourceEntry,
|
|
IN PVOID InBuffer,
|
|
IN DWORD InBufferSize
|
|
)
|
|
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Processes the CLUSCTL_RESOURCE_SET_PRIVATE_PROPERTIES control function
|
|
for resources of type Physical Disk.
|
|
|
|
Arguments:
|
|
|
|
ResourceEntry - Supplies the resource entry on which to operate.
|
|
|
|
InBuffer - Supplies a pointer to a buffer containing input data.
|
|
|
|
InBufferSize - Supplies the size, in bytes, of the data pointed
|
|
to by InBuffer.
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS - The function completed successfully.
|
|
|
|
ERROR_INVALID_PARAMETER - The data is formatted incorrectly.
|
|
|
|
ERROR_NOT_ENOUGH_MEMORY - An error occurred allocating memory.
|
|
|
|
Win32 error code - The function failed.
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD status;
|
|
DISK_PARAMS params;
|
|
|
|
ZeroMemory( ¶ms, sizeof(DISK_PARAMS) );
|
|
|
|
//
|
|
// Parse and validate the properties.
|
|
//
|
|
status = DisksValidatePrivateResProperties( ResourceEntry,
|
|
InBuffer,
|
|
InBufferSize,
|
|
¶ms );
|
|
if ( status != ERROR_SUCCESS ) {
|
|
ResUtilFreeParameterBlock( (LPBYTE) ¶ms,
|
|
(LPBYTE) &ResourceEntry->DiskInfo.Params,
|
|
DiskResourcePrivateProperties );
|
|
return(status);
|
|
}
|
|
|
|
//
|
|
// We cannot allow changing the Signature 'on the fly'...
|
|
//
|
|
if ( (ResourceEntry->DiskInfo.Params.Signature != 0) &&
|
|
(params.Signature != ResourceEntry->DiskInfo.Params.Signature) ) {
|
|
ResUtilFreeParameterBlock( (LPBYTE) ¶ms,
|
|
(LPBYTE) &ResourceEntry->DiskInfo.Params,
|
|
DiskResourcePrivateProperties );
|
|
(DiskpLogEvent)(
|
|
NULL,
|
|
LOG_ERROR,
|
|
L"SetPrivateProperties doesn't allow changing signature, old %1!lx!, new %2!lx!\n",
|
|
ResourceEntry->DiskInfo.Params.Signature,
|
|
params.Signature );
|
|
return(ERROR_INVALID_STATE);
|
|
}
|
|
|
|
//
|
|
// Save the parameter values.
|
|
//
|
|
// NB: Unknown, or non-property table values are dealt with farther below. That is why InBuffer and
|
|
// InBufferSize are not used in this call. Only the propertys in the parameter block are handled here.
|
|
//
|
|
status = ResUtilSetPropertyParameterBlock( ResourceEntry->ResourceParametersKey,
|
|
DiskResourcePrivateProperties,
|
|
NULL,
|
|
(LPBYTE) ¶ms,
|
|
NULL,
|
|
0,
|
|
(LPBYTE) &ResourceEntry->DiskInfo.Params );
|
|
|
|
ResUtilFreeParameterBlock( (LPBYTE) ¶ms,
|
|
(LPBYTE) &ResourceEntry->DiskInfo.Params,
|
|
DiskResourcePrivateProperties );
|
|
|
|
if ( status != ERROR_SUCCESS ) {
|
|
(DiskpLogEvent)(
|
|
NULL,
|
|
LOG_ERROR,
|
|
L"SetPrivateResProperties: Error %1!d! saving properties\n",
|
|
status );
|
|
return(status);
|
|
}
|
|
|
|
//
|
|
// Save any unknown properties.
|
|
//
|
|
status = ResUtilSetUnknownProperties(
|
|
ResourceEntry->ResourceParametersKey,
|
|
DiskResourcePrivatePropertiesAlt,
|
|
InBuffer,
|
|
InBufferSize );
|
|
|
|
if ( status != ERROR_SUCCESS ) {
|
|
(DiskpLogEvent)(
|
|
NULL,
|
|
LOG_ERROR,
|
|
L"SetPrivateResProperties: Error %1!d! saving unknown properties\n",
|
|
status );
|
|
}
|
|
|
|
//
|
|
// Try to attach to this device if we have a signature.
|
|
//
|
|
if ( ResourceEntry->DiskInfo.Params.Signature ) {
|
|
#if 0
|
|
DiskspVerifyState( ResourceEntry );
|
|
#endif
|
|
DoAttach( ResourceEntry->DiskInfo.Params.Signature,
|
|
ResourceEntry->ResourceHandle );
|
|
// ignore status return
|
|
}
|
|
|
|
//
|
|
// If the resource is online, return a non-success status.
|
|
//
|
|
if (status == ERROR_SUCCESS) {
|
|
if ( ResourceEntry->Valid ) {
|
|
status = ERROR_RESOURCE_PROPERTIES_STORED;
|
|
} else {
|
|
status = ERROR_SUCCESS;
|
|
}
|
|
}
|
|
|
|
return status;
|
|
|
|
} // DisksSetPrivateResProperties
|
|
|
|
|
|
|
|
DWORD
|
|
ProcessDllExtension(
|
|
IN PDISK_RESOURCE ResourceEntry,
|
|
IN PVOID InBuffer,
|
|
IN DWORD InBufferSize,
|
|
OUT PVOID OutBuffer,
|
|
IN DWORD OutBufferSize,
|
|
OUT LPDWORD BytesReturned
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Processes the CLUSCTL_RESOURCE_STORAGE_DLL_EXTENSION control function
|
|
for resources of type Physical Disk. The caller will specify a DLL
|
|
and DLL entry point that should be called. This routine will call
|
|
into that entry point with the disk device name represented by the
|
|
Signature parameter.
|
|
|
|
Arguments:
|
|
|
|
ResourceEntry -
|
|
|
|
InBuffer - Supplies a pointer to a buffer containing input data.
|
|
|
|
InBufferSize - Supplies the size, in bytes, of the data pointed
|
|
to by InBuffer.
|
|
|
|
OutBuffer - Supplies a pointer to a buffer containing output data.
|
|
|
|
OutBufferSize - Supplies the size, in bytes, of the data pointed
|
|
to by OutBuffer.
|
|
|
|
BytesReturned - the number of bytes returned (or needed if larger than
|
|
OutBufferSize and ERROR_MORE_DATA is returned
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS - The function completed successfully.
|
|
|
|
ERROR_INVALID_PARAMETER - The input data is formatted incorrectly.
|
|
|
|
ERROR_REVISION_MISMATCH - The input buffer did not have the correct
|
|
revision information.
|
|
|
|
ERROR_MORE_DATA - The output buffer is not large enough to hold all
|
|
the requested data.
|
|
|
|
Win32 error code - The function failed.
|
|
|
|
--*/
|
|
{
|
|
lpPassThruFunc passThruFunc = NULL;
|
|
|
|
PCHAR dllModuleName;
|
|
PCHAR dllProcName;
|
|
PCHAR contextStr;
|
|
|
|
PDISK_DLL_EXTENSION_INFO passThru = InBuffer;
|
|
|
|
HINSTANCE dllModule = NULL;
|
|
DWORD scsiAddress;
|
|
DWORD diskNumber;
|
|
DWORD dwStatus;
|
|
DWORD signature = ResourceEntry->DiskInfo.Params.Signature;
|
|
DWORD expandedSize;
|
|
|
|
CHAR deviceName[MAX_PATH];
|
|
CHAR expandedDllModuleName[MAX_PATH];
|
|
|
|
if ( !InBuffer || !OutBuffer || !OutBufferSize ) {
|
|
dwStatus = ERROR_INVALID_PARAMETER;
|
|
goto FnExit;
|
|
}
|
|
|
|
if ( InBufferSize < sizeof(DISK_DLL_EXTENSION_INFO) ) {
|
|
dwStatus = ERROR_INVALID_PARAMETER;
|
|
goto FnExit;
|
|
}
|
|
|
|
try {
|
|
|
|
if ( passThru->MajorVersion != NT5_MAJOR_VERSION ||
|
|
passThru->MinorVersion != 0 ) {
|
|
|
|
dwStatus = ERROR_REVISION_MISMATCH;
|
|
goto FnExit;
|
|
}
|
|
|
|
//
|
|
// Get the DLL module name from the Input buffer.
|
|
//
|
|
|
|
dllModuleName = passThru->DllModuleName;
|
|
|
|
//
|
|
// Get the DLL entry point name from the Input buffer.
|
|
//
|
|
|
|
dllProcName = passThru->DllProcName;
|
|
|
|
contextStr = passThru->ContextStr;
|
|
|
|
//
|
|
// Get the SCSI address to build the device name.
|
|
//
|
|
|
|
dwStatus = GetScsiAddress( signature, &scsiAddress, &diskNumber );
|
|
|
|
if ( NO_ERROR != dwStatus ) {
|
|
goto FnExit;
|
|
}
|
|
|
|
_snprintf( deviceName, sizeof(deviceName), "\\\\.\\PhysicalDrive%d", diskNumber );
|
|
|
|
expandedSize = ExpandEnvironmentStrings( dllModuleName, expandedDllModuleName, MAX_PATH );
|
|
|
|
if ( !expandedSize ) {
|
|
dwStatus = GetLastError();
|
|
goto FnExit;
|
|
}
|
|
|
|
if ( expandedSize > MAX_PATH ) {
|
|
dwStatus = ERROR_BAD_PATHNAME;
|
|
goto FnExit;
|
|
}
|
|
|
|
//
|
|
// Assume the DLL has not yet been loaded into the address space.
|
|
// The caller should specify a fully qualified path name to make sure their
|
|
// DLL is called, and they are responsible for making sure that the DLL is
|
|
// protected.
|
|
//
|
|
|
|
dllModule = LoadLibraryEx( expandedDllModuleName, NULL, LOAD_WITH_ALTERED_SEARCH_PATH );
|
|
|
|
if ( NULL == dllModule ) {
|
|
dwStatus = GetLastError();
|
|
goto FnExit;
|
|
}
|
|
|
|
//
|
|
// The function name MUST be as defined (i.e. with exactly the same type
|
|
// and number of parameters) or we will have stack problems.
|
|
//
|
|
|
|
passThruFunc = (lpPassThruFunc)GetProcAddress( dllModule, dllProcName );
|
|
|
|
if ( NULL == passThruFunc ) {
|
|
|
|
//
|
|
// The specified function is not available in the DLL, exit now.
|
|
//
|
|
|
|
dwStatus = GetLastError();
|
|
goto FnExit;
|
|
}
|
|
|
|
//
|
|
// Call into the specified DLL.
|
|
//
|
|
|
|
dwStatus = (passThruFunc)( deviceName,
|
|
contextStr,
|
|
OutBuffer,
|
|
OutBufferSize,
|
|
BytesReturned );
|
|
|
|
} except (EXCEPTION_EXECUTE_HANDLER) {
|
|
|
|
dwStatus = GetExceptionCode();
|
|
|
|
(DiskpLogEvent)( ResourceEntry->ResourceHandle,
|
|
LOG_ERROR,
|
|
L"ProcessDllExtension: Exception occurred %1!X! \n",
|
|
dwStatus );
|
|
}
|
|
|
|
FnExit:
|
|
|
|
if ( dllModule ) {
|
|
FreeLibrary( dllModule );
|
|
}
|
|
|
|
return dwStatus;
|
|
|
|
} // ProcessDllExtension
|
|
|
|
|
|
DWORD
|
|
DisksOpenChkdskLogFile(
|
|
IN PDISK_RESOURCE ResourceEntry,
|
|
IN OUT PHANDLE ChkdskLogFile,
|
|
IN OUT LPWSTR *ChkdskLogFileName
|
|
)
|
|
/*++
|
|
|
|
Routine Description:
|
|
|
|
Creates a file to log chkdsk output and returns the handle to the caller.
|
|
The file will not be deleted on close.
|
|
|
|
Arguments:
|
|
|
|
ResourceEntry - Supplies a pointer to the resource structure
|
|
|
|
ChkdskLogFile - Returned handled of newly opened log file.
|
|
|
|
ChkdskLogFileName - String pointer to the name of the newly
|
|
opened log file. Caller is responsible
|
|
for freeing storage.
|
|
|
|
Return Value:
|
|
|
|
ERROR_SUCCESS if successful
|
|
|
|
Win32 error code otherwise
|
|
|
|
--*/
|
|
|
|
{
|
|
DWORD status = NO_ERROR;
|
|
HANDLE retHandle = INVALID_HANDLE_VALUE;
|
|
|
|
PWCHAR last;
|
|
PWCHAR current;
|
|
|
|
LPWSTR clusterDir = NULL;
|
|
LPWSTR fileName = NULL;
|
|
LPWSTR finalFileName = NULL;
|
|
|
|
DWORD clusterDirLength; // This value can change.
|
|
DWORD fileNameLength = MAX_PATH;
|
|
DWORD finalFileNameLength;
|
|
|
|
SECURITY_ATTRIBUTES sa;
|
|
|
|
if ( !ChkdskLogFile || !ChkdskLogFileName ) {
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|
|
|
|
*ChkdskLogFile = INVALID_HANDLE_VALUE;
|
|
*ChkdskLogFileName = NULL;
|
|
|
|
_try {
|
|
|
|
//
|
|
// Get the environment variable "ClusterLog". Embedded in this string
|
|
// is the cluster directory.
|
|
//
|
|
|
|
clusterDir = LocalAlloc( LMEM_FIXED, MAX_PATH * sizeof(WCHAR) );
|
|
if ( !clusterDir ) {
|
|
status = GetLastError();
|
|
_leave;
|
|
}
|
|
|
|
clusterDirLength = GetEnvironmentVariableW( CLUSTERLOG_ENV_VARIABLE,
|
|
clusterDir,
|
|
MAX_PATH );
|
|
|
|
if ( !clusterDirLength ) {
|
|
status = GetLastError();
|
|
_leave;
|
|
}
|
|
|
|
if ( clusterDirLength > MAX_PATH ) {
|
|
|
|
LocalFree( clusterDir );
|
|
clusterDir = LocalAlloc( LMEM_FIXED, clusterDirLength * sizeof( WCHAR ) );
|
|
if ( NULL == clusterDir ) {
|
|
status = GetLastError();
|
|
_leave;
|
|
}
|
|
|
|
clusterDirLength = GetEnvironmentVariableW( CLUSTERLOG_ENV_VARIABLE,
|
|
clusterDir,
|
|
clusterDirLength );
|
|
|
|
if ( !clusterDirLength ) {
|
|
status = GetLastError();
|
|
LocalFree( clusterDir );
|
|
clusterDir = NULL;
|
|
_leave;
|
|
}
|
|
}
|
|
|
|
//
|
|
// We have the log file path and name. Find the last backslash and strip off the
|
|
// log file name. This will be used as our temporary file path.
|
|
//
|
|
|
|
last = NULL;
|
|
current = (PWCHAR) clusterDir;
|
|
|
|
while ( *current != L'\0' ) {
|
|
|
|
if ( L'\\' == *current ) {
|
|
last = current;
|
|
}
|
|
current++;
|
|
}
|
|
|
|
if ( !last ) {
|
|
status = ERROR_BAD_FORMAT;
|
|
_leave;
|
|
}
|
|
|
|
//
|
|
// Change the last backslash to the end of string mark.
|
|
//
|
|
|
|
*last = L'\0';
|
|
|
|
//
|
|
// Now create a file name based on the disk signature.
|
|
//
|
|
|
|
fileName = LocalAlloc( LMEM_FIXED, fileNameLength * sizeof(WCHAR) );
|
|
if ( !fileName ) {
|
|
status = GetLastError();
|
|
_leave;
|
|
}
|
|
|
|
_snwprintf( fileName, fileNameLength,
|
|
L"ChkDsk_Disk%d_Sig%08X.log",
|
|
ResourceEntry->DiskInfo.PhysicalDrive,
|
|
ResourceEntry->DiskInfo.Params.Signature );
|
|
|
|
//
|
|
// Put it all together for the final name.
|
|
//
|
|
|
|
finalFileNameLength = fileNameLength + clusterDirLength + MAX_PATH;
|
|
|
|
finalFileName = LocalAlloc( LMEM_FIXED, finalFileNameLength * sizeof(WCHAR));
|
|
if ( !finalFileName ) {
|
|
status = GetLastError();
|
|
_leave;
|
|
}
|
|
|
|
_snwprintf( finalFileName, finalFileNameLength, L"%ws\\%ws", clusterDir, fileName );
|
|
|
|
//
|
|
// Now open up the file name to log the chkdsk info.
|
|
//
|
|
|
|
ZeroMemory( &sa, sizeof(sa) );
|
|
sa.nLength = sizeof(sa);
|
|
sa.lpSecurityDescriptor = NULL;
|
|
sa.bInheritHandle = TRUE;
|
|
|
|
retHandle = CreateFileW( finalFileName,
|
|
GENERIC_READ | GENERIC_WRITE,
|
|
FILE_SHARE_READ | FILE_SHARE_WRITE,
|
|
&sa,
|
|
CREATE_ALWAYS, // Create a new file or overwrite existing file
|
|
FILE_ATTRIBUTE_NORMAL,
|
|
NULL );
|
|
|
|
if ( INVALID_HANDLE_VALUE == retHandle ) {
|
|
status = GetLastError();
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle,
|
|
LOG_ERROR,
|
|
L"DisksOpenChkdskLogFile: CreateFile returned status of %1!u! \n",
|
|
status );
|
|
_leave;
|
|
}
|
|
|
|
(DiskpLogEvent)(ResourceEntry->ResourceHandle,
|
|
LOG_INFORMATION,
|
|
L"DisksOpenChkdskLogFile: ChkDsk output is in file: %1!s! \n",
|
|
finalFileName );
|
|
|
|
} _finally {
|
|
|
|
*ChkdskLogFile = retHandle;
|
|
|
|
if ( clusterDir ) {
|
|
LocalFree( clusterDir );
|
|
}
|
|
|
|
if ( fileName ) {
|
|
LocalFree( fileName );
|
|
}
|
|
|
|
if ( finalFileName ) {
|
|
*ChkdskLogFileName = finalFileName;
|
|
}
|
|
}
|
|
|
|
return status;
|
|
|
|
} // DisksOpenChkdskLogFile
|
|
|
|
|
|
DWORD
|
|
GetRegDwordValue(
|
|
IN LPWSTR RegKeyName,
|
|
IN LPWSTR ValueName,
|
|
OUT LPDWORD ValueBuffer
|
|
)
|
|
{
|
|
DWORD dwValue;
|
|
DWORD dwValueType;
|
|
DWORD dwDataBufferSize = sizeof( DWORD );
|
|
DWORD dwError;
|
|
|
|
HKEY hClusParmKey = NULL;
|
|
|
|
if ( !ValueBuffer ) {
|
|
dwError = ERROR_INVALID_PARAMETER;
|
|
goto FnExit;
|
|
}
|
|
|
|
*ValueBuffer = 0;
|
|
|
|
//
|
|
// Open the cluster service parameters key.
|
|
//
|
|
|
|
dwError = RegOpenKeyExW( HKEY_LOCAL_MACHINE,
|
|
RegKeyName,
|
|
0,
|
|
KEY_READ,
|
|
&hClusParmKey );
|
|
|
|
if ( ERROR_SUCCESS != dwError ) {
|
|
goto FnExit;
|
|
}
|
|
|
|
//
|
|
// Get the DWORD value.
|
|
//
|
|
|
|
dwError = RegQueryValueExW( hClusParmKey,
|
|
ValueName,
|
|
NULL,
|
|
&dwValueType,
|
|
(LPBYTE) &dwValue,
|
|
&dwDataBufferSize );
|
|
|
|
if ( ERROR_SUCCESS == dwError ) {
|
|
|
|
// Insure that a DWORD value was returned. If not, return an error.
|
|
|
|
if ( REG_DWORD == dwValueType ) {
|
|
|
|
*ValueBuffer = dwValue;
|
|
|
|
} else {
|
|
|
|
dwError = ERROR_BAD_FORMAT;
|
|
|
|
}
|
|
}
|
|
|
|
FnExit:
|
|
|
|
if ( hClusParmKey ) {
|
|
RegCloseKey( hClusParmKey );
|
|
}
|
|
|
|
|
|
return dwError;
|
|
|
|
} // GetRegDwordValue
|
|
|
|
|
|
|
|
//***********************************************************
|
|
//
|
|
// Define Function Table
|
|
//
|
|
//***********************************************************
|
|
|
|
CLRES_V1_FUNCTION_TABLE( DisksFunctionTable,
|
|
CLRES_VERSION_V1_00,
|
|
Disks,
|
|
DisksArbitrate,
|
|
DisksRelease,
|
|
DisksResourceControl,
|
|
DisksResourceTypeControl );
|