544 lines
9.4 KiB
C++
544 lines
9.4 KiB
C++
/*++
|
||
|
||
Copyright (C) 1991-5 Microsoft Corporation
|
||
|
||
Module Name:
|
||
|
||
packet.cxx
|
||
|
||
Abstract:
|
||
|
||
This module contains the code specific to all types of TRANSFER_PACKETS
|
||
objects.
|
||
|
||
Author:
|
||
|
||
Norbert Kusters 2-Feb-1995
|
||
|
||
Environment:
|
||
|
||
kernel mode only
|
||
|
||
Notes:
|
||
|
||
Revision History:
|
||
|
||
--*/
|
||
|
||
extern "C" {
|
||
#include <ntddk.h>
|
||
}
|
||
|
||
#include <ftdisk.h>
|
||
|
||
static PNPAGED_LOOKASIDE_LIST StripeLookasidePackets = NULL;
|
||
static PNPAGED_LOOKASIDE_LIST MirrorLookasidePackets = NULL;
|
||
|
||
|
||
#ifdef ALLOC_PRAGMA
|
||
#pragma code_seg("PAGELK")
|
||
#endif
|
||
|
||
PVOID
|
||
TRANSFER_PACKET::operator new(
|
||
IN size_t Size
|
||
)
|
||
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
This routine is the memory allocator for all classes derived from
|
||
FT_VOLUME.
|
||
|
||
Arguments:
|
||
|
||
Size - Supplies the number of bytes to allocate.
|
||
|
||
Return Value:
|
||
|
||
A pointer to Size bytes of non-paged pool.
|
||
|
||
--*/
|
||
|
||
{
|
||
PTRANSFER_PACKET p;
|
||
|
||
if (Size <= sizeof(STRIPE_TP)) {
|
||
if (!StripeLookasidePackets) {
|
||
StripeLookasidePackets = (PNPAGED_LOOKASIDE_LIST)
|
||
ExAllocatePool(NonPagedPool,
|
||
sizeof(NPAGED_LOOKASIDE_LIST));
|
||
if (!StripeLookasidePackets) {
|
||
return NULL;
|
||
}
|
||
|
||
ExInitializeNPagedLookasideList(StripeLookasidePackets, NULL, NULL,
|
||
0, sizeof(STRIPE_TP), 'sFcS', 32);
|
||
}
|
||
|
||
p = (PTRANSFER_PACKET)
|
||
ExAllocateFromNPagedLookasideList(StripeLookasidePackets);
|
||
if (p) {
|
||
p->_allocationType = TP_ALLOCATION_STRIPE_POOL;
|
||
}
|
||
return p;
|
||
}
|
||
|
||
if (Size <= sizeof(MIRROR_TP)) {
|
||
if (!MirrorLookasidePackets) {
|
||
MirrorLookasidePackets = (PNPAGED_LOOKASIDE_LIST)
|
||
ExAllocatePool(NonPagedPool,
|
||
sizeof(NPAGED_LOOKASIDE_LIST));
|
||
if (!MirrorLookasidePackets) {
|
||
return NULL;
|
||
}
|
||
|
||
ExInitializeNPagedLookasideList(MirrorLookasidePackets, NULL, NULL,
|
||
0, sizeof(MIRROR_TP), 'mFcS', 32);
|
||
}
|
||
|
||
p = (PTRANSFER_PACKET)
|
||
ExAllocateFromNPagedLookasideList(MirrorLookasidePackets);
|
||
if (p) {
|
||
p->_allocationType = TP_ALLOCATION_MIRROR_POOL;
|
||
}
|
||
return p;
|
||
}
|
||
|
||
p = (PTRANSFER_PACKET) ExAllocatePool(NonPagedPool, Size);
|
||
if (p) {
|
||
p->_allocationType = 0;
|
||
}
|
||
return p;
|
||
}
|
||
|
||
VOID
|
||
TRANSFER_PACKET::operator delete(
|
||
IN PVOID MemPtr
|
||
)
|
||
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
This routine frees memory allocated for all classes derived from
|
||
FT_VOLUME.
|
||
|
||
Arguments:
|
||
|
||
MemPtr - Supplies a pointer to the memory to free.
|
||
|
||
Return Value:
|
||
|
||
None.
|
||
|
||
--*/
|
||
|
||
{
|
||
PTRANSFER_PACKET p = (PTRANSFER_PACKET) MemPtr;
|
||
|
||
if (!p) {
|
||
return;
|
||
}
|
||
|
||
if (p->_allocationType == TP_ALLOCATION_STRIPE_POOL) {
|
||
ExFreeToNPagedLookasideList(StripeLookasidePackets, MemPtr);
|
||
} else if (p->_allocationType == TP_ALLOCATION_MIRROR_POOL) {
|
||
ExFreeToNPagedLookasideList(MirrorLookasidePackets, MemPtr);
|
||
} else {
|
||
ExFreePool(MemPtr);
|
||
}
|
||
}
|
||
|
||
TRANSFER_PACKET::~TRANSFER_PACKET(
|
||
)
|
||
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
This is the destructor for a transfer packet. It frees up any allocated
|
||
MDL and buffer.
|
||
|
||
Arguments:
|
||
|
||
None.
|
||
|
||
Return Value:
|
||
|
||
None.
|
||
|
||
--*/
|
||
|
||
{
|
||
FreeMdl();
|
||
}
|
||
|
||
BOOLEAN
|
||
TRANSFER_PACKET::AllocateMdl(
|
||
IN PVOID Buffer,
|
||
IN ULONG Length
|
||
)
|
||
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
This routine allocates an MDL and for this transfer packet.
|
||
|
||
Arguments:
|
||
|
||
Buffer - Supplies the buffer.
|
||
|
||
Length - Supplies the buffer length.
|
||
|
||
Return Value:
|
||
|
||
FALSE - Insufficient resources.
|
||
|
||
TRUE - Success.
|
||
|
||
--*/
|
||
|
||
{
|
||
FreeMdl();
|
||
|
||
Mdl = IoAllocateMdl(Buffer, Length, FALSE, FALSE, NULL);
|
||
if (!Mdl) {
|
||
return FALSE;
|
||
}
|
||
_freeMdl = TRUE;
|
||
|
||
return TRUE;
|
||
}
|
||
|
||
BOOLEAN
|
||
TRANSFER_PACKET::AllocateMdl(
|
||
IN ULONG Length
|
||
)
|
||
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
This routine allocates an MDL and buffer for this transfer packet.
|
||
|
||
Arguments:
|
||
|
||
Length - Supplies the buffer length.
|
||
|
||
Return Value:
|
||
|
||
FALSE - Insufficient resources.
|
||
|
||
TRUE - Success.
|
||
|
||
--*/
|
||
|
||
{
|
||
PVOID buffer;
|
||
|
||
FreeMdl();
|
||
|
||
buffer = ExAllocatePool(NonPagedPoolCacheAligned,
|
||
Length < PAGE_SIZE ? PAGE_SIZE : Length);
|
||
if (!buffer) {
|
||
return FALSE;
|
||
}
|
||
_freeBuffer = TRUE;
|
||
|
||
Mdl = IoAllocateMdl(buffer, Length, FALSE, FALSE, NULL);
|
||
if (!Mdl) {
|
||
ExFreePool(buffer);
|
||
_freeBuffer = FALSE;
|
||
return FALSE;
|
||
}
|
||
_freeMdl = TRUE;
|
||
MmBuildMdlForNonPagedPool(Mdl);
|
||
|
||
return TRUE;
|
||
}
|
||
|
||
VOID
|
||
TRANSFER_PACKET::FreeMdl(
|
||
)
|
||
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
It frees up any allocated MDL and buffer.
|
||
|
||
Arguments:
|
||
|
||
None.
|
||
|
||
Return Value:
|
||
|
||
None.
|
||
|
||
--*/
|
||
|
||
{
|
||
if (_freeBuffer) {
|
||
ExFreePool(MmGetMdlVirtualAddress(Mdl));
|
||
_freeBuffer = FALSE;
|
||
}
|
||
if (_freeMdl) {
|
||
IoFreeMdl(Mdl);
|
||
_freeMdl = FALSE;
|
||
}
|
||
}
|
||
|
||
OVERLAP_TP::~OVERLAP_TP(
|
||
)
|
||
|
||
{
|
||
if (InQueue) {
|
||
OverlappedIoManager->ReleaseIoRegion(this);
|
||
}
|
||
}
|
||
|
||
MIRROR_RECOVER_TP::~MIRROR_RECOVER_TP(
|
||
)
|
||
|
||
{
|
||
FreeMdls();
|
||
}
|
||
|
||
BOOLEAN
|
||
MIRROR_RECOVER_TP::AllocateMdls(
|
||
IN ULONG Length
|
||
)
|
||
|
||
{
|
||
PVOID buffer;
|
||
|
||
FreeMdls();
|
||
|
||
PartialMdl = IoAllocateMdl((PVOID) (PAGE_SIZE - 1), Length,
|
||
FALSE, FALSE, NULL);
|
||
|
||
if (!PartialMdl) {
|
||
FreeMdls();
|
||
return FALSE;
|
||
}
|
||
|
||
buffer = ExAllocatePool(NonPagedPoolCacheAligned,
|
||
Length < PAGE_SIZE ? PAGE_SIZE : Length);
|
||
if (!buffer) {
|
||
FreeMdls();
|
||
return FALSE;
|
||
}
|
||
|
||
VerifyMdl = IoAllocateMdl(buffer, Length, FALSE, FALSE, NULL);
|
||
if (!VerifyMdl) {
|
||
ExFreePool(buffer);
|
||
FreeMdls();
|
||
return FALSE;
|
||
}
|
||
MmBuildMdlForNonPagedPool(VerifyMdl);
|
||
|
||
return TRUE;
|
||
}
|
||
|
||
VOID
|
||
MIRROR_RECOVER_TP::FreeMdls(
|
||
)
|
||
|
||
{
|
||
if (PartialMdl) {
|
||
IoFreeMdl(PartialMdl);
|
||
PartialMdl = NULL;
|
||
}
|
||
if (VerifyMdl) {
|
||
ExFreePool(MmGetMdlVirtualAddress(VerifyMdl));
|
||
IoFreeMdl(VerifyMdl);
|
||
VerifyMdl = NULL;
|
||
}
|
||
}
|
||
|
||
SWP_RECOVER_TP::~SWP_RECOVER_TP(
|
||
)
|
||
|
||
{
|
||
FreeMdls();
|
||
}
|
||
|
||
BOOLEAN
|
||
SWP_RECOVER_TP::AllocateMdls(
|
||
IN ULONG Length
|
||
)
|
||
|
||
{
|
||
PVOID buffer;
|
||
|
||
FreeMdls();
|
||
|
||
PartialMdl = IoAllocateMdl((PVOID) (PAGE_SIZE - 1), Length,
|
||
FALSE, FALSE, NULL);
|
||
|
||
if (!PartialMdl) {
|
||
FreeMdls();
|
||
return FALSE;
|
||
}
|
||
|
||
buffer = ExAllocatePool(NonPagedPoolCacheAligned,
|
||
Length < PAGE_SIZE ? PAGE_SIZE : Length);
|
||
if (!buffer) {
|
||
FreeMdls();
|
||
return FALSE;
|
||
}
|
||
|
||
VerifyMdl = IoAllocateMdl(buffer, Length, FALSE, FALSE, NULL);
|
||
if (!VerifyMdl) {
|
||
ExFreePool(buffer);
|
||
FreeMdls();
|
||
return FALSE;
|
||
}
|
||
MmBuildMdlForNonPagedPool(VerifyMdl);
|
||
|
||
return TRUE;
|
||
}
|
||
|
||
VOID
|
||
SWP_RECOVER_TP::FreeMdls(
|
||
)
|
||
|
||
{
|
||
if (PartialMdl) {
|
||
IoFreeMdl(PartialMdl);
|
||
PartialMdl = NULL;
|
||
}
|
||
if (VerifyMdl) {
|
||
ExFreePool(MmGetMdlVirtualAddress(VerifyMdl));
|
||
IoFreeMdl(VerifyMdl);
|
||
VerifyMdl = NULL;
|
||
}
|
||
}
|
||
|
||
SWP_WRITE_TP::~SWP_WRITE_TP(
|
||
)
|
||
|
||
{
|
||
FreeMdls();
|
||
}
|
||
|
||
BOOLEAN
|
||
SWP_WRITE_TP::AllocateMdls(
|
||
IN ULONG Length
|
||
)
|
||
|
||
{
|
||
PVOID buffer;
|
||
|
||
FreeMdls();
|
||
|
||
buffer = ExAllocatePool(NonPagedPoolCacheAligned,
|
||
Length < PAGE_SIZE ? PAGE_SIZE : Length);
|
||
if (!buffer) {
|
||
return FALSE;
|
||
}
|
||
|
||
ReadAndParityMdl = IoAllocateMdl(buffer, Length, FALSE, FALSE, NULL);
|
||
if (!ReadAndParityMdl) {
|
||
ExFreePool(buffer);
|
||
return FALSE;
|
||
}
|
||
MmBuildMdlForNonPagedPool(ReadAndParityMdl);
|
||
|
||
buffer = ExAllocatePool(NonPagedPoolCacheAligned,
|
||
Length < PAGE_SIZE ? PAGE_SIZE : Length);
|
||
if (!buffer) {
|
||
FreeMdls();
|
||
return FALSE;
|
||
}
|
||
|
||
WriteMdl = IoAllocateMdl(buffer, Length, FALSE, FALSE, NULL);
|
||
if (!WriteMdl) {
|
||
ExFreePool(buffer);
|
||
FreeMdls();
|
||
return FALSE;
|
||
}
|
||
MmBuildMdlForNonPagedPool(WriteMdl);
|
||
|
||
return TRUE;
|
||
}
|
||
|
||
VOID
|
||
SWP_WRITE_TP::FreeMdls(
|
||
)
|
||
|
||
{
|
||
if (ReadAndParityMdl) {
|
||
ExFreePool(MmGetMdlVirtualAddress(ReadAndParityMdl));
|
||
IoFreeMdl(ReadAndParityMdl);
|
||
ReadAndParityMdl = NULL;
|
||
}
|
||
if (WriteMdl) {
|
||
ExFreePool(MmGetMdlVirtualAddress(WriteMdl));
|
||
IoFreeMdl(WriteMdl);
|
||
WriteMdl = NULL;
|
||
}
|
||
}
|
||
|
||
REDISTRIBUTION_CW_TP::~REDISTRIBUTION_CW_TP(
|
||
)
|
||
|
||
{
|
||
FreeMdls();
|
||
}
|
||
|
||
BOOLEAN
|
||
REDISTRIBUTION_CW_TP::AllocateMdls(
|
||
IN ULONG Length
|
||
)
|
||
|
||
{
|
||
PVOID buffer;
|
||
|
||
FreeMdls();
|
||
|
||
PartialMdl = IoAllocateMdl((PVOID) (PAGE_SIZE - 1), Length,
|
||
FALSE, FALSE, NULL);
|
||
|
||
if (!PartialMdl) {
|
||
FreeMdls();
|
||
return FALSE;
|
||
}
|
||
|
||
buffer = ExAllocatePool(NonPagedPoolCacheAligned,
|
||
Length < PAGE_SIZE ? PAGE_SIZE : Length);
|
||
if (!buffer) {
|
||
FreeMdls();
|
||
return FALSE;
|
||
}
|
||
|
||
VerifyMdl = IoAllocateMdl(buffer, Length, FALSE, FALSE, NULL);
|
||
if (!VerifyMdl) {
|
||
ExFreePool(buffer);
|
||
FreeMdls();
|
||
return FALSE;
|
||
}
|
||
MmBuildMdlForNonPagedPool(VerifyMdl);
|
||
|
||
return TRUE;
|
||
}
|
||
|
||
VOID
|
||
REDISTRIBUTION_CW_TP::FreeMdls(
|
||
)
|
||
|
||
{
|
||
if (PartialMdl) {
|
||
IoFreeMdl(PartialMdl);
|
||
PartialMdl = NULL;
|
||
}
|
||
if (VerifyMdl) {
|
||
ExFreePool(MmGetMdlVirtualAddress(VerifyMdl));
|
||
IoFreeMdl(VerifyMdl);
|
||
VerifyMdl = NULL;
|
||
}
|
||
}
|