OS_Develop/Windows-Server-2003
OS_Develop/Windows-Server-2003
1
0
Fork 0
Code Pull Requests Activity
master
Windows-Server-2003/drivers/storage/newft/ftdisk.h

3135 lines
67 KiB
C
Raw Permalink Normal View History

Initiall commit
2024-08-04 01:28:15 +02:00
/*++
Copyright (C) 1991-5 Microsoft Corporation
Module Name:
ftdisk.h
Abstract:
These are the structures that FtDisk driver
uses to support IO to NTFT volumes.
Notes:
Revision History:
--*/
extern "C" {
#include "stdio.h"
#include <ntdskreg.h>
#include <ntddft.h>
#include <ntddft2.h>
#include <ntdddisk.h>
#include "ftlog.h"
#include <wdmguid.h>
#include <devguid.h>
#include <stdarg.h>
#include <volmgr.h>
#include <mountdev.h>
#include <ntddvol.h>
#include <wmilib.h>
}
#ifdef POOL_TAGGING
#undef ExAllocatePool
#undef ExAllocatePoolWithQuota
#define ExAllocatePool(a,b) ExAllocatePoolWithTag(a,b,'tFcS')
#define ExAllocatePoolWithQuota(a,b) ExAllocatePoolWithQuotaTag(a,b,'tFcS')
#define FTDISK_TAG_IOCTL_BUFFER 'iFcS'
#endif
#define STRIPE_SIZE ((ULONG) 0x00010000)
#define DISK_REGISTRY_KEY_W L"\\Registry\\Machine\\System\\DISK"
#define FT_STATE_REGISTRY_KEY L"\\Registry\\Machine\\System\\DISK\\FtState"
//
// NEC98 machines have drive-letter A and B which are non FD too.
//
#define FirstDriveLetter (IsNEC_98 ? 'A' : 'C')
#define LastDriveLetter 'Z'
class TRANSFER_PACKET;
typedef TRANSFER_PACKET* PTRANSFER_PACKET;
typedef
VOID
(*FT_TRANSFER_COMPLETION_ROUTINE)(
IN PTRANSFER_PACKET TransferPacket
);
typedef
VOID
(*FT_COMPLETION_ROUTINE)(
IN PVOID Context,
IN NTSTATUS Status
);
class FT_VOLUME;
typedef FT_VOLUME* PFT_VOLUME;
typedef struct _FT_COMPLETION_ROUTINE_CONTEXT {
KSPIN_LOCK SpinLock;
NTSTATUS Status;
LONG RefCount;
FT_COMPLETION_ROUTINE CompletionRoutine;
PVOID Context;
PFT_VOLUME ParentVolume;
} FT_COMPLETION_ROUTINE_CONTEXT, *PFT_COMPLETION_ROUTINE_CONTEXT;
class FT_BASE_CLASS;
typedef FT_BASE_CLASS* PFT_BASE_CLASS;
class FT_BASE_CLASS {
public:
static
PVOID
operator new(
IN size_t Size
);
static
VOID
operator delete(
IN PVOID MemPtr
);
};
class VOLUME_EXTENSION;
typedef VOLUME_EXTENSION* PVOLUME_EXTENSION;
class ROOT_EXTENSION;
typedef ROOT_EXTENSION* PROOT_EXTENSION;
#include <ondisk.hxx>
class TRANSFER_PACKET : public FT_BASE_CLASS {
public:
static
PVOID
operator new(
IN size_t Size
);
static
VOID
operator delete(
IN PVOID MemPtr
);
TRANSFER_PACKET(
) { _freeMdl = FALSE; _freeBuffer = FALSE; SpecialRead = 0;
OriginalIrp = NULL; };
virtual
~TRANSFER_PACKET(
);
BOOLEAN
AllocateMdl(
IN PVOID Buffer,
IN ULONG Length
);
BOOLEAN
AllocateMdl(
IN ULONG Length
);
VOID
FreeMdl(
);
// These fields must be filled in by the caller.
PMDL Mdl;
PIRP OriginalIrp;
ULONG Length;
LONGLONG Offset;
FT_TRANSFER_COMPLETION_ROUTINE CompletionRoutine;
PFT_VOLUME TargetVolume;
PETHREAD Thread;
UCHAR IrpFlags;
BOOLEAN ReadPacket;
UCHAR SpecialRead;
// A spin lock which may be used to resolve contention for the
// fields below. This spin lock must be initialized by the callee.
KSPIN_LOCK SpinLock;
// This field must be filled in by the callee.
IO_STATUS_BLOCK IoStatus;
// These fields are for use by the callee.
LONG RefCount;
LIST_ENTRY QueueEntry;
private:
BOOLEAN _freeMdl;
BOOLEAN _freeBuffer;
UCHAR _allocationType;
};
#define TP_SPECIAL_READ_PRIMARY (1)
#define TP_SPECIAL_READ_SECONDARY (2)
#define TP_ALLOCATION_STRIPE_POOL (1)
#define TP_ALLOCATION_MIRROR_POOL (2)
class DISPATCH_TP;
typedef DISPATCH_TP* PDISPATCH_TP;
class DISPATCH_TP : public TRANSFER_PACKET {
public:
DISPATCH_TP(
) {};
PIRP Irp;
PVOLUME_EXTENSION Extension;
};
class VOLUME_SET;
typedef VOLUME_SET* PVOLUME_SET;
class VOLSET_TP;
typedef VOLSET_TP* PVOLSET_TP;
class VOLSET_TP : public TRANSFER_PACKET {
public:
VOLSET_TP(
) {};
PTRANSFER_PACKET MasterPacket;
PVOLUME_SET VolumeSet;
USHORT WhichMember;
};
class STRIPE;
typedef STRIPE* PSTRIPE;
class STRIPE_TP;
typedef STRIPE_TP* PSTRIPE_TP;
class STRIPE_TP : public TRANSFER_PACKET {
public:
STRIPE_TP(
) {};
PTRANSFER_PACKET MasterPacket;
PSTRIPE Stripe;
USHORT WhichMember;
};
class OVERLAPPED_IO_MANAGER;
typedef OVERLAPPED_IO_MANAGER* POVERLAPPED_IO_MANAGER;
class OVERLAP_TP;
typedef OVERLAP_TP* POVERLAP_TP;
class OVERLAP_TP : public TRANSFER_PACKET {
friend class OVERLAPPED_IO_MANAGER;
public:
OVERLAP_TP(
) { InQueue = FALSE; };
virtual
~OVERLAP_TP(
);
private:
BOOLEAN AllMembers;
BOOLEAN InQueue;
LIST_ENTRY OverlapQueue;
LIST_ENTRY CompletionList;
POVERLAPPED_IO_MANAGER OverlappedIoManager;
};
class MIRROR;
typedef MIRROR* PMIRROR;
class MIRROR_TP;
typedef MIRROR_TP* PMIRROR_TP;
class MIRROR_TP : public OVERLAP_TP {
public:
MIRROR_TP(
) { OneReadFailed = FALSE; };
PTRANSFER_PACKET MasterPacket;
PMIRROR Mirror;
USHORT WhichMember;
BOOLEAN OneReadFailed;
PMIRROR_TP SecondWritePacket;
FT_TRANSFER_COMPLETION_ROUTINE SavedCompletionRoutine;
};
class MIRROR_RECOVER_TP;
typedef MIRROR_RECOVER_TP* PMIRROR_RECOVER_TP;
class MIRROR_RECOVER_TP : public MIRROR_TP {
public:
MIRROR_RECOVER_TP(
) { PartialMdl = NULL; VerifyMdl = NULL; };
virtual
~MIRROR_RECOVER_TP(
);
BOOLEAN
AllocateMdls(
IN ULONG Length
);
VOID
FreeMdls(
);
PMDL PartialMdl;
PMDL VerifyMdl;
};
class PARITY_IO_MANAGER;
typedef PARITY_IO_MANAGER* PPARITY_IO_MANAGER;
class PARITY_TP;
typedef PARITY_TP* PPARITY_TP;
class PARITY_TP : public TRANSFER_PACKET {
friend class PARITY_IO_MANAGER;
friend
VOID
UpdateParityCompletionRoutine(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
ParityCarefulWritePhase1(
IN OUT PTRANSFER_PACKET TransferPacket
);
public:
PARITY_TP(
) {};
private:
BOOLEAN Idle;
BOOLEAN OneWriteFailed;
LIST_ENTRY OverlapQueue;
LIST_ENTRY UpdateQueue;
PPARITY_IO_MANAGER ParityIoManager;
LONGLONG BucketNumber;
};
class PARITY_RECOVER_TP;
typedef PARITY_RECOVER_TP* PPARITY_RECOVER_TP;
class PARITY_RECOVER_TP : public PARITY_TP {
public:
PARITY_RECOVER_TP(
) {};
PPARITY_TP MasterPacket;
};
class STRIPE_WP;
typedef STRIPE_WP* PSTRIPE_WP;
class SWP_TP;
typedef SWP_TP* PSWP_TP;
class SWP_TP : public OVERLAP_TP {
public:
SWP_TP(
) {};
PTRANSFER_PACKET MasterPacket;
PSTRIPE_WP StripeWithParity;
USHORT WhichMember;
FT_TRANSFER_COMPLETION_ROUTINE SavedCompletionRoutine;
BOOLEAN OneReadFailed;
};
class SWP_RECOVER_TP;
typedef SWP_RECOVER_TP* PSWP_RECOVER_TP;
class SWP_RECOVER_TP : public SWP_TP {
public:
SWP_RECOVER_TP(
) { PartialMdl = NULL; VerifyMdl = NULL; };
virtual
~SWP_RECOVER_TP(
);
BOOLEAN
AllocateMdls(
IN ULONG Length
);
VOID
FreeMdls(
);
PMDL PartialMdl;
PMDL VerifyMdl;
PARITY_TP ParityPacket;
};
class SWP_WRITE_TP;
typedef SWP_WRITE_TP* PSWP_WRITE_TP;
class SWP_WRITE_TP : public SWP_TP {
public:
SWP_WRITE_TP(
) { ReadAndParityMdl = NULL; WriteMdl = NULL; };
virtual
~SWP_WRITE_TP(
);
BOOLEAN
AllocateMdls(
IN ULONG Length
);
VOID
FreeMdls(
);
PMDL ReadAndParityMdl;
PMDL WriteMdl;
FT_MEMBER_STATE TargetState;
USHORT ParityMember;
SWP_TP ReadWritePacket;
PARITY_TP ParityPacket;
};
class SWP_REGENERATE_TP;
typedef SWP_REGENERATE_TP *PSWP_REGENERATE_TP;
class SWP_REGENERATE_TP : public TRANSFER_PACKET {
public:
SWP_REGENERATE_TP(
) {};
PSWP_TP MasterPacket;
USHORT WhichMember;
LIST_ENTRY RegenPacketList;
};
class SWP_REBUILD_TP;
typedef SWP_REBUILD_TP *PSWP_REBUILD_TP;
class SWP_REBUILD_TP : public SWP_TP {
public:
SWP_REBUILD_TP(
) {};
PFT_COMPLETION_ROUTINE_CONTEXT Context;
BOOLEAN Initialize;
};
class REDISTRIBUTION;
typedef REDISTRIBUTION *PREDISTRIBUTION;
class REDISTRIBUTION_TP;
typedef REDISTRIBUTION_TP* PREDISTRIBUTION_TP;
class REDISTRIBUTION_TP : public TRANSFER_PACKET {
public:
REDISTRIBUTION_TP(
) {};
PTRANSFER_PACKET MasterPacket;
PREDISTRIBUTION Redistribution;
USHORT WhichMember;
};
class REDISTRIBUTION_LOCK_TP;
typedef REDISTRIBUTION_LOCK_TP* PREDISTRIBUTION_LOCK_TP;
class REDISTRIBUTION_LOCK_TP : public OVERLAP_TP {
public:
REDISTRIBUTION_LOCK_TP(
) {};
PTRANSFER_PACKET MasterPacket;
PREDISTRIBUTION Redistribution;
};
class REDISTRIBUTION_SYNC_TP;
typedef REDISTRIBUTION_SYNC_TP* PREDISTRIBUTION_SYNC_TP;
class REDISTRIBUTION_SYNC_TP : public OVERLAP_TP {
public:
REDISTRIBUTION_SYNC_TP(
) {};
PFT_COMPLETION_ROUTINE_CONTEXT Context;
PREDISTRIBUTION Redistribution;
REDISTRIBUTION_TP IoPacket;
};
class REDISTRIBUTION_CW_TP;
typedef REDISTRIBUTION_CW_TP* PREDISTRIBUTION_CW_TP;
class REDISTRIBUTION_CW_TP : public REDISTRIBUTION_TP {
public:
REDISTRIBUTION_CW_TP(
) { PartialMdl = NULL; VerifyMdl = NULL; };
virtual
~REDISTRIBUTION_CW_TP(
);
BOOLEAN
AllocateMdls(
IN ULONG Length
);
VOID
FreeMdls(
);
PMDL PartialMdl;
PMDL VerifyMdl;
};
class OVERLAPPED_IO_MANAGER : public FT_BASE_CLASS {
public:
NTSTATUS
Initialize(
IN ULONG BucketSize
);
VOID
AcquireIoRegion(
IN OUT POVERLAP_TP TransferPacket,
IN BOOLEAN AllMembers
);
VOID
ReleaseIoRegion(
IN OUT POVERLAP_TP TransferPacket
);
VOID
PromoteToAllMembers(
IN OUT POVERLAP_TP TransferPacket
);
OVERLAPPED_IO_MANAGER(
) { _spinLock = NULL; _ioQueue = NULL; };
~OVERLAPPED_IO_MANAGER(
);
private:
ULONG _numQueues;
ULONG _bucketSize;
PKSPIN_LOCK _spinLock;
PLIST_ENTRY _ioQueue;
};
class PARITY_IO_MANAGER : public FT_BASE_CLASS {
friend
VOID
UpdateParityCompletionRoutine(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
ParityCarefulWritePhase1(
IN OUT PTRANSFER_PACKET TransferPacket
);
public:
NTSTATUS
Initialize(
IN ULONG BucketSize,
IN ULONG SectorSize
);
VOID
StartReadForUpdateParity(
IN LONGLONG Offset,
IN ULONG Length,
IN PFT_VOLUME TargetVolume,
IN PETHREAD Thread,
IN UCHAR IrpFlags
);
VOID
UpdateParity(
IN OUT PPARITY_TP TransferPacket
);
PARITY_IO_MANAGER(
) { _spinLock = NULL; _ioQueue = NULL; _ePacket = NULL; };
~PARITY_IO_MANAGER(
);
private:
VOID
CarefulWrite(
IN OUT PPARITY_TP TransferPacket
);
ULONG _numQueues;
ULONG _bucketSize;
ULONG _sectorSize;
PKSPIN_LOCK _spinLock;
PLIST_ENTRY _ioQueue;
//
// Emergency packet.
//
PPARITY_TP _ePacket;
BOOLEAN _ePacketInUse;
BOOLEAN _ePacketQueueBeingServiced;
LIST_ENTRY _ePacketQueue;
KSPIN_LOCK _ePacketSpinLock;
};
class PARTITION;
typedef PARTITION* PPARTITION;
class FT_VOLUME : public FT_BASE_CLASS {
friend
VOID
SetMemberStateWorker(
IN PVOID Context
);
friend
VOID
FtVolumeNotifyWorker(
IN PVOID FtVolume
);
public:
VOID
Initialize(
IN OUT PROOT_EXTENSION RootExtension,
IN FT_LOGICAL_DISK_ID LogicalDiskId
);
FT_LOGICAL_DISK_ID
QueryLogicalDiskId(
);
VOID
SetLogicalDiskId(
IN FT_LOGICAL_DISK_ID NewLogicalDiskId
);
virtual
VOID
Notify(
);
virtual
FT_LOGICAL_DISK_TYPE
QueryLogicalDiskType(
) = 0;
virtual
USHORT
QueryNumberOfMembers(
) = 0;
virtual
PFT_VOLUME
GetMember(
IN USHORT MemberNumber
) = 0;
virtual
NTSTATUS
OrphanMember(
IN USHORT MemberNumber,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
) = 0;
virtual
NTSTATUS
RegenerateMember(
IN USHORT MemberNumber,
IN OUT PFT_VOLUME NewMember,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
) = 0;
virtual
VOID
Transfer(
IN OUT PTRANSFER_PACKET TransferPacket
) = 0;
virtual
VOID
ReplaceBadSector(
IN OUT PTRANSFER_PACKET TransferPacket
) = 0;
virtual
VOID
StartSyncOperations(
IN BOOLEAN RegenerateOrphans,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
) = 0;
virtual
VOID
StopSyncOperations(
) = 0;
virtual
VOID
BroadcastIrp(
IN PIRP Irp,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
) = 0;
virtual
ULONG
QuerySectorSize(
) = 0;
virtual
LONGLONG
QueryVolumeSize(
) = 0;
virtual
PFT_VOLUME
GetContainedLogicalDisk(
IN FT_LOGICAL_DISK_ID LogicalDiskId
) = 0;
virtual
PFT_VOLUME
GetContainedLogicalDisk(
IN PDEVICE_OBJECT TargetObject
) = 0;
virtual
PFT_VOLUME
GetContainedLogicalDisk(
IN ULONG Signature,
IN LONGLONG Offset
) = 0;
virtual
PFT_VOLUME
GetParentLogicalDisk(
IN PFT_VOLUME Volume
) = 0;
virtual
VOID
SetDirtyBit(
IN BOOLEAN IsDirty,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
) = 0;
virtual
VOID
SetMember(
IN USHORT MemberNumber,
IN PFT_VOLUME Member
) = 0;
virtual
BOOLEAN
IsComplete(
IN BOOLEAN IoPending
) = 0;
virtual
VOID
CompleteNotification(
IN BOOLEAN IoPending
) = 0;
virtual
NTSTATUS
CheckIo(
OUT PBOOLEAN IsIoOk
) = 0;
virtual
ULONG
QueryNumberOfPartitions(
) = 0;
virtual
NTSTATUS
SetPartitionType(
IN UCHAR PartitionType
) = 0;
virtual
UCHAR
QueryPartitionType(
) = 0;
virtual
UCHAR
QueryStackSize(
) = 0;
virtual
VOID
CreateLegacyNameLinks(
IN PUNICODE_STRING DeviceName
) = 0;
virtual
BOOLEAN
IsVolumeSuitableForRegenerate(
IN USHORT MemberNumber,
IN PFT_VOLUME Volume
);
virtual
VOID
NewStateArrival(
IN PVOID NewStateInstance
);
virtual
PDEVICE_OBJECT
GetLeftmostPartitionObject(
) = 0;
virtual
NTSTATUS
QueryDiskExtents(
OUT PDISK_EXTENT* DiskExtents,
OUT PULONG NumberOfDiskExtents
) = 0;
virtual
BOOLEAN
QueryVolumeState(
IN PFT_VOLUME Volume,
OUT PFT_MEMBER_STATE State
) = 0;
virtual
NTSTATUS
QueryPhysicalOffsets(
IN LONGLONG LogicalOffset,
OUT PVOLUME_PHYSICAL_OFFSET* PhysicalOffsets,
OUT PULONG NumberOfPhysicalOffsets
) = 0;
virtual
NTSTATUS
QueryLogicalOffset(
IN PVOLUME_PHYSICAL_OFFSET PhysicalOffset,
OUT PLONGLONG LogicalOffset
) = 0;
virtual
VOID
ModifyStateForUser(
IN OUT PVOID State
);
virtual
~FT_VOLUME(
);
FT_VOLUME(
) { _refCount = 1; };
LONG _refCount;
protected:
KSPIN_LOCK _spinLock;
PFT_LOGICAL_DISK_INFORMATION_SET _diskInfoSet;
PROOT_EXTENSION _rootExtension;
private:
FT_LOGICAL_DISK_ID _logicalDiskId;
};
#define TRANSFER(a) ((a)->TargetVolume->Transfer((a)))
class PARTITION : public FT_VOLUME {
friend
NTSTATUS
PartitionTransferCompletionRoutine(
IN PDEVICE_OBJECT DeviceObject,
IN PIRP Irp,
IN PVOID TransferPacket
);
public:
PARTITION(
) { _emergencyIrp = NULL; };
virtual
~PARTITION(
);
NTSTATUS
Initialize(
IN OUT PROOT_EXTENSION RootExtension,
IN FT_LOGICAL_DISK_ID LogicalDiskId,
IN OUT PDEVICE_OBJECT TargetObject,
IN OUT PDEVICE_OBJECT WholeDiskPdo
);
virtual
FT_LOGICAL_DISK_TYPE
QueryLogicalDiskType(
);
virtual
USHORT
QueryNumberOfMembers(
);
virtual
PFT_VOLUME
GetMember(
IN USHORT MemberNumber
);
virtual
NTSTATUS
OrphanMember(
IN USHORT MemberNumber,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
NTSTATUS
RegenerateMember(
IN USHORT MemberNumber,
IN OUT PFT_VOLUME NewMember,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
VOID
Transfer(
IN OUT PTRANSFER_PACKET TransferPacket
);
virtual
VOID
ReplaceBadSector(
IN OUT PTRANSFER_PACKET TransferPacket
);
virtual
VOID
StartSyncOperations(
IN BOOLEAN RegenerateOrphans,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
VOID
StopSyncOperations(
);
virtual
VOID
BroadcastIrp(
IN PIRP Irp,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
ULONG
QuerySectorSize(
);
virtual
LONGLONG
QueryVolumeSize(
);
virtual
PFT_VOLUME
GetContainedLogicalDisk(
IN FT_LOGICAL_DISK_ID LogicalDiskId
);
virtual
PFT_VOLUME
GetContainedLogicalDisk(
IN PDEVICE_OBJECT TargetObject
);
virtual
PFT_VOLUME
GetContainedLogicalDisk(
IN ULONG Signature,
IN LONGLONG Offset
);
virtual
PFT_VOLUME
GetParentLogicalDisk(
IN PFT_VOLUME Volume
);
virtual
VOID
SetDirtyBit(
IN BOOLEAN IsDirty,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
VOID
SetMember(
IN USHORT MemberNumber,
IN PFT_VOLUME Member
);
virtual
BOOLEAN
IsComplete(
IN BOOLEAN IoPending
);
virtual
VOID
CompleteNotification(
IN BOOLEAN IoPending
);
virtual
NTSTATUS
CheckIo(
OUT PBOOLEAN IsIoOk
);
virtual
ULONG
QueryNumberOfPartitions(
);
virtual
NTSTATUS
SetPartitionType(
IN UCHAR PartitionType
);
virtual
UCHAR
QueryPartitionType(
);
virtual
UCHAR
QueryStackSize(
);
virtual
VOID
CreateLegacyNameLinks(
IN PUNICODE_STRING DeviceName
);
virtual
PDEVICE_OBJECT
GetLeftmostPartitionObject(
);
virtual
NTSTATUS
QueryDiskExtents(
OUT PDISK_EXTENT* DiskExtents,
OUT PULONG NumberOfDiskExtents
);
virtual
BOOLEAN
QueryVolumeState(
IN PFT_VOLUME Volume,
OUT PFT_MEMBER_STATE State
);
virtual
NTSTATUS
QueryPhysicalOffsets(
IN LONGLONG LogicalOffset,
OUT PVOLUME_PHYSICAL_OFFSET* PhysicalOffsets,
OUT PULONG NumberOfPhysicalOffsets
);
virtual
NTSTATUS
QueryLogicalOffset(
IN PVOLUME_PHYSICAL_OFFSET PhysicalOffset,
OUT PLONGLONG LogicalOffset
);
PDEVICE_OBJECT
GetTargetObject(
) { return _targetObject; };
PDEVICE_OBJECT
GetWholeDiskPdo(
) { return _wholeDiskPdo; };
private:
PDEVICE_OBJECT _targetObject;
PDEVICE_OBJECT _wholeDiskPdo;
ULONG _sectorSize;
LONGLONG _partitionOffset;
LONGLONG _partitionLength;
PIRP _emergencyIrp;
BOOLEAN _emergencyIrpInUse;
LIST_ENTRY _emergencyIrpQueue;
};
class COMPOSITE_FT_VOLUME;
typedef COMPOSITE_FT_VOLUME *PCOMPOSITE_FT_VOLUME;
class COMPOSITE_FT_VOLUME : public FT_VOLUME {
public:
virtual
NTSTATUS
Initialize(
IN OUT PROOT_EXTENSION RootExtension,
IN FT_LOGICAL_DISK_ID LogicalDiskId,
IN OUT PFT_VOLUME* VolumeArray,
IN USHORT ArraySize,
IN PVOID ConfigInfo,
IN PVOID StateInfo
);
virtual
FT_LOGICAL_DISK_TYPE
QueryLogicalDiskType(
) = 0;
virtual
USHORT
QueryNumberOfMembers(
);
virtual
PFT_VOLUME
GetMember(
IN USHORT MemberNumber
);
virtual
NTSTATUS
OrphanMember(
IN USHORT MemberNumber,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
NTSTATUS
RegenerateMember(
IN USHORT MemberNumber,
IN OUT PFT_VOLUME NewMember,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
VOID
Transfer(
IN OUT PTRANSFER_PACKET TransferPacket
) = 0;
virtual
VOID
ReplaceBadSector(
IN OUT PTRANSFER_PACKET TransferPacket
) = 0;
virtual
VOID
StartSyncOperations(
IN BOOLEAN RegenerateOrphans,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
VOID
StopSyncOperations(
);
virtual
VOID
BroadcastIrp(
IN PIRP Irp,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
ULONG
QuerySectorSize(
);
virtual
LONGLONG
QueryVolumeSize(
) = 0;
virtual
PFT_VOLUME
GetContainedLogicalDisk(
IN FT_LOGICAL_DISK_ID LogicalDiskId
);
virtual
PFT_VOLUME
GetContainedLogicalDisk(
IN PDEVICE_OBJECT TargetObject
);
virtual
PFT_VOLUME
GetContainedLogicalDisk(
IN ULONG Signature,
IN LONGLONG Offset
);
virtual
PFT_VOLUME
GetParentLogicalDisk(
IN PFT_VOLUME Volume
);
virtual
VOID
SetDirtyBit(
IN BOOLEAN IsDirty,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
VOID
SetMember(
IN USHORT MemberNumber,
IN PFT_VOLUME Member
);
virtual
BOOLEAN
IsComplete(
IN BOOLEAN IoPending
);
virtual
VOID
CompleteNotification(
IN BOOLEAN IoPending
);
virtual
NTSTATUS
CheckIo(
OUT PBOOLEAN IsIoOk
) = 0;
virtual
ULONG
QueryNumberOfPartitions(
);
virtual
NTSTATUS
SetPartitionType(
IN UCHAR PartitionType
);
virtual
UCHAR
QueryPartitionType(
);
virtual
UCHAR
QueryStackSize(
);
virtual
VOID
CreateLegacyNameLinks(
IN PUNICODE_STRING DeviceName
);
virtual
PDEVICE_OBJECT
GetLeftmostPartitionObject(
);
virtual
NTSTATUS
QueryDiskExtents(
OUT PDISK_EXTENT* DiskExtents,
OUT PULONG NumberOfDiskExtents
);
virtual
BOOLEAN
QueryVolumeState(
IN PFT_VOLUME Volume,
OUT PFT_MEMBER_STATE State
);
virtual
NTSTATUS
QueryPhysicalOffsets(
IN LONGLONG LogicalOffset,
OUT PVOLUME_PHYSICAL_OFFSET* PhysicalOffsets,
OUT PULONG NumberOfPhysicalOffsets
) = 0;
virtual
NTSTATUS
QueryLogicalOffset(
IN PVOLUME_PHYSICAL_OFFSET PhysicalOffset,
OUT PLONGLONG LogicalOffset
) = 0;
COMPOSITE_FT_VOLUME(
) { _volumeArray = NULL; };
virtual
~COMPOSITE_FT_VOLUME(
);
protected:
PFT_VOLUME
GetMemberUnprotected(
IN USHORT MemberNumber
) { return _volumeArray[MemberNumber]; };
VOID
SetMemberUnprotected(
IN USHORT MemberNumber,
IN PFT_VOLUME NewVolume
) { _volumeArray[MemberNumber] = NewVolume; };
USHORT
QueryNumMembers(
) { return _arraySize; };
VOID
SetSectorSize(
IN ULONG SectorSize
) { _sectorSize = SectorSize; };
private:
PFT_VOLUME* _volumeArray;
USHORT _arraySize;
ULONG _sectorSize;
};
class VOLUME_SET : public COMPOSITE_FT_VOLUME {
friend
VOID
VolsetTransferSequentialCompletionRoutine(
IN PTRANSFER_PACKET TransferPacket
);
public:
VOLUME_SET(
) { _ePacket = NULL; };
virtual
~VOLUME_SET(
);
virtual
NTSTATUS
Initialize(
IN OUT PROOT_EXTENSION RootExtension,
IN FT_LOGICAL_DISK_ID LogicalDiskId,
IN OUT PFT_VOLUME* VolumeArray,
IN USHORT ArraySize,
IN PVOID ConfigInfo,
IN PVOID StateInfo
);
virtual
FT_LOGICAL_DISK_TYPE
QueryLogicalDiskType(
);
virtual
VOID
Transfer(
IN OUT PTRANSFER_PACKET TransferPacket
);
virtual
VOID
ReplaceBadSector(
IN OUT PTRANSFER_PACKET TransferPacket
);
virtual
LONGLONG
QueryVolumeSize(
);
virtual
VOID
CompleteNotification(
IN BOOLEAN IoPending
);
virtual
NTSTATUS
CheckIo(
OUT PBOOLEAN IsIoOk
);
virtual
NTSTATUS
QueryPhysicalOffsets(
IN LONGLONG LogicalOffset,
OUT PVOLUME_PHYSICAL_OFFSET* PhysicalOffsets,
OUT PULONG NumberOfPhysicalOffsets
);
virtual
NTSTATUS
QueryLogicalOffset(
IN PVOLUME_PHYSICAL_OFFSET PhysicalOffset,
OUT PLONGLONG LogicalOffset
);
private:
BOOLEAN
LaunchParallel(
IN OUT PTRANSFER_PACKET TransferPacket
);
VOID
LaunchSequential(
IN OUT PTRANSFER_PACKET TransferPacket
);
LONGLONG _volumeSize;
PVOLSET_TP _ePacket;
BOOLEAN _ePacketInUse;
LIST_ENTRY _ePacketQueue;
};
class STRIPE : public COMPOSITE_FT_VOLUME {
friend
VOID
StripeSequentialTransferCompletionRoutine(
IN PTRANSFER_PACKET TransferPacket
);
public:
STRIPE(
) { _ePacket = NULL; };
virtual
~STRIPE(
);
virtual
NTSTATUS
Initialize(
IN OUT PROOT_EXTENSION RootExtension,
IN FT_LOGICAL_DISK_ID LogicalDiskId,
IN OUT PFT_VOLUME* VolumeArray,
IN USHORT ArraySize,
IN PVOID ConfigInfo,
IN PVOID StateInfo
);
virtual
FT_LOGICAL_DISK_TYPE
QueryLogicalDiskType(
);
virtual
VOID
Transfer(
IN OUT PTRANSFER_PACKET TransferPacket
);
virtual
VOID
ReplaceBadSector(
IN OUT PTRANSFER_PACKET TransferPacket
);
virtual
LONGLONG
QueryVolumeSize(
);
virtual
VOID
CompleteNotification(
IN BOOLEAN IoPending
);
virtual
NTSTATUS
CheckIo(
OUT PBOOLEAN IsIoOk
);
virtual
NTSTATUS
QueryPhysicalOffsets(
IN LONGLONG LogicalOffset,
OUT PVOLUME_PHYSICAL_OFFSET* PhysicalOffsets,
OUT PULONG NumberOfPhysicalOffsets
);
virtual
NTSTATUS
QueryLogicalOffset(
IN PVOLUME_PHYSICAL_OFFSET PhysicalOffset,
OUT PLONGLONG LogicalOffset
);
private:
BOOLEAN
LaunchParallel(
IN OUT PTRANSFER_PACKET TransferPacket
);
VOID
LaunchSequential(
IN OUT PTRANSFER_PACKET TransferPacket
);
ULONG _stripeSize;
ULONG _stripeShift;
ULONG _stripeMask;
LONGLONG _memberSize;
LONGLONG _volumeSize;
PSTRIPE_TP _ePacket;
BOOLEAN _ePacketInUse;
LIST_ENTRY _ePacketQueue;
};
class MIRROR : public COMPOSITE_FT_VOLUME {
friend
VOID
FinishRegenerate(
IN PMIRROR Mirror,
IN PFT_COMPLETION_ROUTINE_CONTEXT RegenContext,
IN PMIRROR_TP TransferPacket
);
friend
VOID
MirrorRegenerateCompletionRoutine(
IN PTRANSFER_PACKET TransferPacket
);
friend
VOID
StartRegeneration(
IN PVOID Context,
IN NTSTATUS Status
);
friend
VOID
MirrorTransferCompletionRoutine(
IN PTRANSFER_PACKET TransferPacket
);
friend
VOID
MirrorRecoverPhase8(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
MirrorRecoverPhase7(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
MirrorRecoverPhase6(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
MirrorRecoverPhase5(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
MirrorRecoverPhase4(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
MirrorRecoverPhase3(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
MirrorRecoverPhase2(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
MirrorRecoverEmergencyCompletion(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
MirrorRecoverPhase1(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
MirrorMaxTransferCompletionRoutine(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
MirrorMaxTransferEmergencyCompletion(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
MirrorPropogateStateChangesWorker(
IN PVOID Mirror
);
friend
VOID
MirrorCarefulWritePhase1(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
MirrorCarefulWritePhase2(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
MirrorCarefulWriteEmergencyCompletion(
IN OUT PTRANSFER_PACKET TransferPacket
);
public:
virtual
~MIRROR(
);
virtual
NTSTATUS
Initialize(
IN OUT PROOT_EXTENSION RootExtension,
IN FT_LOGICAL_DISK_ID LogicalDiskId,
IN OUT PFT_VOLUME* VolumeArray,
IN USHORT ArraySize,
IN PVOID ConfigInfo,
IN PVOID StateInfo
);
virtual
FT_LOGICAL_DISK_TYPE
QueryLogicalDiskType(
);
virtual
NTSTATUS
OrphanMember(
IN USHORT MemberNumber,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
NTSTATUS
RegenerateMember(
IN USHORT MemberNumber,
IN OUT PFT_VOLUME NewMember,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
VOID
Transfer(
IN OUT PTRANSFER_PACKET TransferPacket
);
virtual
VOID
ReplaceBadSector(
IN OUT PTRANSFER_PACKET TransferPacket
);
virtual
VOID
StartSyncOperations(
IN BOOLEAN RegenerateOrphans,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
VOID
StopSyncOperations(
);
virtual
LONGLONG
QueryVolumeSize(
);
virtual
VOID
SetDirtyBit(
IN BOOLEAN IsDirty,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
BOOLEAN
IsComplete(
IN BOOLEAN IoPending
);
virtual
VOID
CompleteNotification(
IN BOOLEAN IoPending
);
virtual
NTSTATUS
CheckIo(
OUT PBOOLEAN IsIoOk
);
virtual
BOOLEAN
IsVolumeSuitableForRegenerate(
IN USHORT MemberNumber,
IN PFT_VOLUME Volume
);
virtual
VOID
NewStateArrival(
IN PVOID NewStateInstance
);
virtual
PDEVICE_OBJECT
GetLeftmostPartitionObject(
);
virtual
BOOLEAN
QueryVolumeState(
IN PFT_VOLUME Volume,
OUT PFT_MEMBER_STATE State
);
virtual
NTSTATUS
QueryPhysicalOffsets(
IN LONGLONG LogicalOffset,
OUT PVOLUME_PHYSICAL_OFFSET* PhysicalOffsets,
OUT PULONG NumberOfPhysicalOffsets
);
virtual
NTSTATUS
QueryLogicalOffset(
IN PVOLUME_PHYSICAL_OFFSET PhysicalOffset,
OUT PLONGLONG LogicalOffset
);
virtual
VOID
ModifyStateForUser(
IN OUT PVOID State
);
MIRROR(
) { _ePacket = NULL; _ePacket2 = NULL; _eRecoverPacket = NULL; };
private:
FT_MEMBER_STATE
QueryMemberState(
IN USHORT MemberNumber
) { return MemberNumber == _state.UnhealthyMemberNumber ?
_state.UnhealthyMemberState : FtMemberHealthy; }
BOOLEAN
SetMemberState(
IN USHORT MemberNumber,
IN FT_MEMBER_STATE MemberState
);
BOOLEAN
LaunchRead(
IN OUT PTRANSFER_PACKET TransferPacket,
IN OUT PMIRROR_TP Packet1
);
BOOLEAN
LaunchWrite(
IN OUT PTRANSFER_PACKET TransferPacket,
IN OUT PMIRROR_TP Packet1,
IN OUT PMIRROR_TP Packet2
);
VOID
Recycle(
IN OUT PMIRROR_TP TransferPacket,
IN BOOLEAN ServiceEmergencyQueue
);
VOID
Recover(
IN OUT PMIRROR_TP TransferPacket
);
VOID
MaxTransfer(
IN OUT PMIRROR_TP TransferPacket
);
VOID
PropogateStateChanges(
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
VOID
CarefulWrite(
IN OUT PMIRROR_TP TransferPacket
);
LONGLONG _volumeSize;
//
// Keep track of requests for load balancing.
//
LONG _requestCount[2];
LONGLONG _lastPosition[2];
//
// The dynamic state of this volume.
//
FT_MIRROR_AND_SWP_STATE_INFORMATION _state;
BOOLEAN _originalDirtyBit;
BOOLEAN _orphanedBecauseOfMissingMember;
//
// Indicates whether or not 'StartSyncOperations' or
// 'RegenerateMember' is ok.
//
BOOLEAN _syncOk;
//
// Indicate whether or not balanced reads are allowed.
//
BOOLEAN _balancedReads;
//
// Indicates whether or not to stop syncs.
//
BOOLEAN _stopSyncs;
//
// Emergency packet.
//
PMIRROR_TP _ePacket, _ePacket2;
BOOLEAN _ePacketInUse;
LIST_ENTRY _ePacketQueue;
//
// Emergency recover packet.
//
PMIRROR_RECOVER_TP _eRecoverPacket;
BOOLEAN _eRecoverPacketInUse;
LIST_ENTRY _eRecoverPacketQueue;
//
// Overlapped io manager.
//
OVERLAPPED_IO_MANAGER _overlappedIoManager;
};
class STRIPE_WP : public COMPOSITE_FT_VOLUME {
friend
VOID
StripeWpSyncFinalCompletion(
IN PVOID Context,
IN NTSTATUS Status
);
friend
VOID
StripeWpSyncCompletionRoutine(
IN PTRANSFER_PACKET TransferPacket
);
friend
VOID
StartStripeRegeneration(
IN PVOID Context,
IN NTSTATUS Status
);
friend
VOID
StripeWpParallelTransferCompletionRoutine(
IN PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpSequentialTransferCompletionRoutine(
IN PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpWritePhase31(
IN OUT PTRANSFER_PACKET Packet
);
friend
VOID
StripeWpWritePhase30(
IN OUT PTRANSFER_PACKET Packet
);
friend
VOID
StripeWpWriteRecover(
IN OUT PTRANSFER_PACKET MasterPacket
);
friend
VOID
StripeWpWritePhase2(
IN OUT PTRANSFER_PACKET ReadPacket
);
friend
VOID
StripeWpWritePhase1(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpSequentialRegenerateCompletion(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpSequentialEmergencyCompletion(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpParallelRegenerateCompletion(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpRegeneratePacketPhase1(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpRecoverPhase8(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpRecoverPhase7(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpRecoverPhase6(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpRecoverPhase5(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpRecoverPhase4(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpRecoverPhase3(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpRecoverPhase2(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpRecoverEmergencyCompletion(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpRecoverPhase1(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpMaxTransferCompletionRoutine(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpMaxTransferEmergencyCompletion(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpPropogateStateChangesWorker(
IN PVOID StripeWp
);
friend
VOID
StripeWpCompleteWritePhase4(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpCompleteWritePhase3(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpCompleteWritePhase2(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpCompleteWritePhase1(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpCarefulWritePhase2(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpCarefulWritePhase1(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpCarefulWriteEmergencyCompletion(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpCarefulUpdateCompletion(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
StripeWpCarefulUpdateEmergencyCompletion(
IN OUT PTRANSFER_PACKET TransferPacket
);
public:
virtual
NTSTATUS
Initialize(
IN OUT PROOT_EXTENSION RootExtension,
IN FT_LOGICAL_DISK_ID LogicalDiskId,
IN OUT PFT_VOLUME* VolumeArray,
IN USHORT ArraySize,
IN PVOID ConfigInfo,
IN PVOID StateInfo
);
virtual
FT_LOGICAL_DISK_TYPE
QueryLogicalDiskType(
);
virtual
NTSTATUS
OrphanMember(
IN USHORT MemberNumber,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
NTSTATUS
RegenerateMember(
IN USHORT MemberNumber,
IN OUT PFT_VOLUME NewMember,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
VOID
Transfer(
IN OUT PTRANSFER_PACKET TransferPacket
);
virtual
VOID
ReplaceBadSector(
IN OUT PTRANSFER_PACKET TransferPacket
);
virtual
VOID
StartSyncOperations(
IN BOOLEAN RegenerateOrphans,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
VOID
StopSyncOperations(
);
virtual
LONGLONG
QueryVolumeSize(
);
virtual
VOID
SetDirtyBit(
IN BOOLEAN IsDirty,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
BOOLEAN
IsComplete(
IN BOOLEAN IoPending
);
virtual
VOID
CompleteNotification(
IN BOOLEAN IoPending
);
virtual
NTSTATUS
CheckIo(
OUT PBOOLEAN IsIoOk
);
virtual
BOOLEAN
IsVolumeSuitableForRegenerate(
IN USHORT MemberNumber,
IN PFT_VOLUME Volume
);
virtual
VOID
NewStateArrival(
IN PVOID NewStateInstance
);
virtual
BOOLEAN
QueryVolumeState(
IN PFT_VOLUME Volume,
OUT PFT_MEMBER_STATE State
);
virtual
NTSTATUS
QueryPhysicalOffsets(
IN LONGLONG LogicalOffset,
OUT PVOLUME_PHYSICAL_OFFSET* PhysicalOffsets,
OUT PULONG NumberOfPhysicalOffsets
);
virtual
NTSTATUS
QueryLogicalOffset(
IN PVOLUME_PHYSICAL_OFFSET PhysicalOffset,
OUT PLONGLONG LogicalOffset
);
virtual
VOID
ModifyStateForUser(
IN OUT PVOID State
);
STRIPE_WP(
);
virtual
~STRIPE_WP(
);
private:
FT_MEMBER_STATE
QueryMemberState(
IN USHORT MemberNumber
) { return MemberNumber == _state.UnhealthyMemberNumber ?
_state.UnhealthyMemberState : FtMemberHealthy; }
BOOLEAN
SetMemberState(
IN USHORT MemberNumber,
IN FT_MEMBER_STATE MemberState
);
BOOLEAN
LaunchParallel(
IN OUT PTRANSFER_PACKET TransferPacket
);
VOID
LaunchSequential(
IN OUT PTRANSFER_PACKET TransferPacket
);
VOID
ReadPacket(
IN OUT PSWP_TP TransferPacket
);
VOID
WritePacket(
IN OUT PSWP_WRITE_TP TransferPacket
);
VOID
RegeneratePacket(
IN OUT PSWP_TP TransferPacket,
IN BOOLEAN AllocateRegion
);
VOID
Recover(
IN OUT PSWP_TP TransferPacket,
IN BOOLEAN NeedAcquire
);
VOID
MaxTransfer(
IN OUT PSWP_TP TransferPacket
);
VOID
RecycleRecoverTp(
IN OUT PSWP_RECOVER_TP TransferPacket
);
VOID
PropogateStateChanges(
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
VOID
CompleteWrite(
IN OUT PSWP_WRITE_TP TransferPacket
);
VOID
CarefulWrite(
IN OUT PSWP_TP TransferPacket
);
VOID
CarefulUpdate(
IN OUT PSWP_TP ParityPacket
);
ULONG _stripeSize;
LONGLONG _memberSize;
LONGLONG _volumeSize;
//
// The dynamic state of this volume.
//
FT_MIRROR_AND_SWP_STATE_INFORMATION _state;
BOOLEAN _originalDirtyBit;
BOOLEAN _orphanedBecauseOfMissingMember;
//
// Indicates whether or not 'StartSyncOperations' or
// 'RegenerateMember' is ok.
//
BOOLEAN _syncOk;
//
// Indicates whether or not to stop syncs.
//
BOOLEAN _stopSyncs;
//
// State for keeping track of overlapping write requests.
// One OVERLAPPED_IO_MANAGER for each member.
//
OVERLAPPED_IO_MANAGER _overlappedIoManager;
//
// State for serializing parity I/O.
//
PARITY_IO_MANAGER _parityIoManager;
//
// Emergency read/write packet.
//
PSWP_WRITE_TP _ePacket;
BOOLEAN _ePacketInUse;
BOOLEAN _ePacketQueueBeingServiced;
LIST_ENTRY _ePacketQueue;
//
// Emergency regenerate packet.
//
PSWP_REGENERATE_TP _eRegeneratePacket;
BOOLEAN _eRegeneratePacketInUse;
LIST_ENTRY _eRegeneratePacketQueue;
//
// Emergency recover packet.
//
PSWP_RECOVER_TP _eRecoverPacket;
BOOLEAN _eRecoverPacketInUse;
LIST_ENTRY _eRecoverPacketQueue;
};
class REDISTRIBUTION : public COMPOSITE_FT_VOLUME {
friend
VOID
RedistributionSyncPhase6(
IN OUT PVOID SyncPacket,
IN NTSTATUS Status
);
friend
VOID
RedistributionSyncPhase5(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
RedistributionSyncPhase4(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
RedistributionSyncPhase3(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
RedistributionSyncPhase2(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
RedistributionSyncPhase1(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
RedistributionRegionLockCompletion(
IN OUT PTRANSFER_PACKET LockPacket
);
friend
VOID
RedistributionLockReplaceCompletion(
IN OUT PTRANSFER_PACKET TransferPacket
);
friend
VOID
RedistributionPropogateStateChangesWorker(
IN PVOID WorkItem
);
public:
virtual
~REDISTRIBUTION(
);
virtual
NTSTATUS
Initialize(
IN OUT PROOT_EXTENSION RootExtension,
IN FT_LOGICAL_DISK_ID LogicalDiskId,
IN OUT PFT_VOLUME* VolumeArray,
IN USHORT ArraySize,
IN PVOID ConfigInfo,
IN PVOID StateInfo
);
virtual
FT_LOGICAL_DISK_TYPE
QueryLogicalDiskType(
);
virtual
VOID
Transfer(
IN OUT PTRANSFER_PACKET TransferPacket
);
virtual
VOID
ReplaceBadSector(
IN OUT PTRANSFER_PACKET TransferPacket
);
virtual
VOID
StartSyncOperations(
IN BOOLEAN RegenerateOrphans,
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
virtual
VOID
StopSyncOperations(
);
virtual
LONGLONG
QueryVolumeSize(
);
virtual
VOID
CompleteNotification(
IN BOOLEAN IoPending
);
virtual
NTSTATUS
CheckIo(
OUT PBOOLEAN IsIoOk
);
virtual
VOID
NewStateArrival(
IN PVOID NewStateInstance
);
virtual
NTSTATUS
QueryPhysicalOffsets(
IN LONGLONG LogicalOffset,
OUT PVOLUME_PHYSICAL_OFFSET* PhysicalOffsets,
OUT PULONG NumberOfPhysicalOffsets
);
virtual
NTSTATUS
QueryLogicalOffset(
IN PVOLUME_PHYSICAL_OFFSET PhysicalOffset,
OUT PLONGLONG LogicalOffset
);
private:
VOID
RedistributeTransfer(
IN OUT PTRANSFER_PACKET TransferPacket
);
VOID
RedistributeReplaceBadSector(
IN OUT PTRANSFER_PACKET TransferPacket
);
VOID
PropogateStateChanges(
IN FT_COMPLETION_ROUTINE CompletionRoutine,
IN PVOID Context
);
VOID
MaxTransfer(
IN OUT PREDISTRIBUTION_TP TransferPacket
);
VOID
VerifyWrite(
IN OUT PREDISTRIBUTION_TP TransferPacket
);
VOID
CarefulWrite(
IN OUT PREDISTRIBUTION_TP TransferPacket
);
ULONG _stripeSize;
USHORT _firstWidth;
USHORT _totalWidth;
LONGLONG _firstSize;
LONGLONG _totalSize;
BOOLEAN _syncOk;
BOOLEAN _stopSyncs;
FT_REDISTRIBUTION_STATE_INFORMATION _state;
BOOLEAN _redistributionComplete;
OVERLAPPED_IO_MANAGER _overlappedIoManager;
};
typedef struct _FTP_GPT_ATTRIBUTE_REVERT_ENTRY {
GUID PartitionUniqueId;
ULONGLONG GptAttributes;
ULONG MbrSignature;
ULONG Reserved;
} FTP_GPT_ATTRIBUTE_REVERT_ENTRY, *PFTP_GPT_ATTRIBUTE_REVERT_ENTRY;
struct DEVICE_EXTENSION {
//
// Pointer to the device object for this extension.
//
PDEVICE_OBJECT DeviceObject; // 00
//
// Pointer to the root device extension.
//
PROOT_EXTENSION Root; // 04
//
// The type of device extension.
//
ULONG DeviceExtensionType; // 08
//
// A spinlock for synchronization.
//
KSPIN_LOCK SpinLock; // 0C
};
class ROOT_EXTENSION : public DEVICE_EXTENSION {
public:
//
// Pointer to the driver object.
//
PDRIVER_OBJECT DriverObject; // 10
//
// Pointer to the next device in the stack.
//
PDEVICE_OBJECT TargetObject; // 14
//
// Pointer to the PDO.
//
PDEVICE_OBJECT Pdo; // 18
//
// List of volumes in the system. Protect with 'Semaphore'.
//
LIST_ENTRY VolumeList; // 1C
//
// List of dead volumes. Protect with 'Semaphore'.
//
LIST_ENTRY DeadVolumeList; // 2C
//
// The next volume number. Protect with 'Semaphore'.
//
ULONG NextVolumeNumber; // 34
//
// The disk information set for the on disk storage of FT sets.
// Protect with 'Semaphore'.
//
PFT_LOGICAL_DISK_INFORMATION_SET DiskInfoSet; // 38
//
// Private worker thread and queue. Protect queue with 'SpinLock'.
//
PVOID WorkerThread; // 3C
LIST_ENTRY WorkerQueue; // 40
KSEMAPHORE WorkerSemaphore; // 44
LONG TerminateThread; // 58
//
// Change notify Irp list. Protect with cancel spin lock.
//
LIST_ENTRY ChangeNotifyIrpList; // 5C
//
// A semaphore for synchronization.
//
KSEMAPHORE Mutex; // 64
//
// Device Interface name.
//
UNICODE_STRING VolumeManagerInterfaceName; // 78
//
// Whether or not we are past the boot reinitialize code.
//
BOOLEAN PastBootReinitialize; // 80
//
// Whether or not the FT specific code has been locked down.
//
BOOLEAN FtCodeLocked; // 81
//
// Whether or not we are past the reinitialize code.
//
BOOLEAN PastReinitialize; // 82
//
// Registry Path.
//
UNICODE_STRING DiskPerfRegistryPath;
//
// Table of PmWmiCounter Functions.
//
PMWMICOUNTERLIB_CONTEXT PmWmiCounterLibContext;
//
// The Unique partition type GUID of the ESP.
//
GUID ESPUniquePartitionGUID;
//
// An array of gpt attribute revert records retrieved from the
// registry at boot up.
//
ULONG NumberOfAttributeRevertEntries;
PFTP_GPT_ATTRIBUTE_REVERT_ENTRY GptAttributeRevertEntries;
//
// The number of pre-exposures. Protect with 'Semaphore'.
//
ULONG PreExposureCount;
};
typedef DEVICE_EXTENSION *PDEVICE_EXTENSION;
typedef
VOID
(*ZERO_REF_CALLBACK)(
IN PVOLUME_EXTENSION VolumeExtension
);
class VOLUME_EXTENSION : public DEVICE_EXTENSION {
public:
//
// A pointer to the target object or the FT volume for this
// volume. Protect these with 'SpinLock'.
//
PDEVICE_OBJECT TargetObject; // 10
PFT_VOLUME FtVolume; // 14
LONG RefCount; // 18
ZERO_REF_CALLBACK ZeroRefCallback; // 1C
PVOID ZeroRefContext; // 20
LIST_ENTRY ZeroRefHoldQueue; // 24
BOOLEAN IsStarted; // 2C
BOOLEAN IsComplete; // 2D
BOOLEAN InPagingPath; // 2E
BOOLEAN RemoveInProgress; // 2F
BOOLEAN IsOffline; // 30
BOOLEAN DeadToPnp; // 31
BOOLEAN DeviceDeleted; // 32
BOOLEAN IsPreExposure; // 33
BOOLEAN IsGpt; // 34
BOOLEAN IsHidden; // 35
BOOLEAN IsSuperFloppy; // 36
BOOLEAN IsReadOnly; // 37
BOOLEAN IsEspType; // 38
BOOLEAN IsInstalled;
LONG AllSystemsGo; // 38
GUID* OfflineOwner;
//
// List entry for volume list or dead volume list.
// Protect with 'Root->Semaphore'.
//
LIST_ENTRY ListEntry; // 3C
//
// The volume number.
//
ULONG VolumeNumber; // 44
//
// Emergency queue for a transfer packet. Protect with 'SpinLock'.
//
PDISPATCH_TP EmergencyTransferPacket; // 48
LIST_ENTRY EmergencyTransferPacketQueue; // 4C
BOOLEAN EmergencyTransferPacketInUse; // 54
//
// List of unique id change notify IRPs.
//
LIST_ENTRY ChangeNotifyIrps; // 58
//
// The dev node name for this device.
//
UNICODE_STRING DeviceNodeName; // 60
//
// Whole disk PDO, if this extension is a partition and
// not an FT volume.
//
PDEVICE_OBJECT WholeDiskPdo; // 68
PDEVICE_OBJECT WholeDisk; // 6C
LONGLONG PartitionOffset; // 70
LONGLONG PartitionLength; // 78
ULONG Signature;
//
// Device Interface name.
//
UNICODE_STRING MountedDeviceInterfaceName; // 80
//
// A semaphore to protect 'ZeroRefCallback'.
//
KSEMAPHORE Semaphore; // 8C
//
// Old whole disk PDO to facilitate debugging.
//
PDEVICE_OBJECT OldWholeDiskPdo; // A0
//
// Unique Id Guid in the GPT case.
//
GUID UniqueIdGuid; // A4
//
// A copy of the current power state. Protect with 'SpinLock'.
//
DEVICE_POWER_STATE PowerState; // B4
//
// Whether or not the counters are running.
//
BOOLEAN CountersEnabled;
//
// Leave counters always enabled if we see IOCTL_DISK_PERFORMANCE
//
LONG EnableAlways;
//
// Counter structure.
//
PVOID PmWmiCounterContext;
//
// Table of Wmi Functions.
//
PWMILIB_CONTEXT WmilibContext;
//
// State for reverting GPT attributes. Protect with 'Root->Semaphore'.
//
ULONGLONG GptAttributesToRevertTo;
PFILE_OBJECT RevertOnCloseFileObject;
BOOLEAN ApplyToAllConnectedVolumes;
//
// Cached MBR GPT attributes.
//
ULONGLONG MbrGptAttributes;
};
#define DEVICE_EXTENSION_ROOT (0)
#define DEVICE_EXTENSION_VOLUME (1)
BOOLEAN
FtpIsWorseStatus(
IN NTSTATUS Status1,
IN NTSTATUS Status2
);
VOID
FtpComputeParity(
IN PVOID TargetBuffer,
IN PVOID SourceBuffer,
IN ULONG BufferLength
);
VOID
FtpLogError(
IN PDEVICE_EXTENSION Extension,
IN FT_LOGICAL_DISK_ID LogicalDiskId,
IN NTSTATUS SpecificIoStatus,
IN NTSTATUS FinalStatus,
IN ULONG UniqueErrorValue
);
VOID
FtpQueueWorkItem(
IN PROOT_EXTENSION RootExtension,
IN PWORK_QUEUE_ITEM WorkItem
);
VOID
FtpNotify(
IN OUT PROOT_EXTENSION RootExtension,
IN PVOLUME_EXTENSION Extension
);
VOID
FtpAcquire(
IN OUT PROOT_EXTENSION RootExtension
);
NTSTATUS
FtpAcquireWithTimeout(
IN OUT PROOT_EXTENSION RootExtension
);
VOID
FtpRelease(
IN OUT PROOT_EXTENSION RootExtension
);
VOID
FtpDecrementRefCount(
IN OUT PVOLUME_EXTENSION Extension
);
FT_LOGICAL_DISK_ID
GenerateNewLogicalDiskId(
);
NTSTATUS
FtpQueryPartitionInformation(
IN PROOT_EXTENSION RootExtension,
IN PDEVICE_OBJECT Partition,
OUT PULONG DiskNumber,
OUT PLONGLONG Offset,
OUT PULONG PartitionNumber,
OUT PUCHAR PartitionType,
OUT PLONGLONG PartitionLength,
OUT GUID* PartitionTypeGuid,
OUT GUID* PartitionUniqueIdGuid,
OUT PBOOLEAN IsGpt,
OUT PULONGLONG GptAttributes
);
ULONG
FtpQueryDiskSignature(
IN PDEVICE_OBJECT WholeDiskPdo
);
NTSTATUS
FtpDiskRegistryQueryRoutine(
IN PWSTR ValueName,
IN ULONG ValueType,
IN PVOID ValueData,
IN ULONG ValueLength,
IN PVOID Context,
IN PVOID EntryContext
);
PDISK_PARTITION
FtpFindDiskPartition(
IN PDISK_REGISTRY DiskRegistry,
IN ULONG Signature,
IN LONGLONG Offset
);
VOID
FtpCopyStateToRegistry(
IN FT_LOGICAL_DISK_ID LogicalDiskId,
IN PVOID LogicalDiskState,
IN USHORT LogicalDiskStateSize
);
BOOLEAN
FtpQueryStateFromRegistry(
IN FT_LOGICAL_DISK_ID LogicalDiskId,
IN PVOID LogicalDiskState,
IN USHORT BufferSize,
OUT PUSHORT LogicalDiskStateSize
);
VOID
FtpDeleteStateInRegistry(
IN FT_LOGICAL_DISK_ID LogicalDiskId
);
PVOLUME_EXTENSION
FtpFindExtensionCoveringDiskId(
IN PROOT_EXTENSION RootExtension,
IN FT_LOGICAL_DISK_ID LogicalDiskId
);
NTSTATUS
FtpReadPartitionTableEx(
IN PDEVICE_OBJECT DeviceObject,
IN PDRIVE_LAYOUT_INFORMATION_EX* DriveLayout
);
NTSTATUS
FtpApplyESPProtection(
IN PUNICODE_STRING PartitionName
);
Copy Permalink