/*++ 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 #include #include #include #include "ftlog.h" #include #include #include #include #include #include #include } #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 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 );