Implement core pool allocation and deallocation logic
This commit is contained in:
@@ -18,6 +18,13 @@ namespace MM
|
||||
{
|
||||
class Allocator final : private Pool
|
||||
{
|
||||
private:
|
||||
STATIC ULONG BigAllocationsInUse;
|
||||
STATIC PPOOL_TRACKER_BIG_ALLOCATIONS BigAllocationsTable;
|
||||
STATIC SIZE_T BigAllocationsTableHash;
|
||||
STATIC KSPIN_LOCK BigAllocationsTableLock;
|
||||
STATIC SIZE_T BigAllocationsTableSize;
|
||||
|
||||
public:
|
||||
STATIC XTAPI XTSTATUS AllocatePages(IN MMPOOL_TYPE PoolType,
|
||||
IN SIZE_T Bytes,
|
||||
@@ -35,16 +42,26 @@ namespace MM
|
||||
STATIC XTAPI XTSTATUS FreePool(IN PVOID VirtualAddress);
|
||||
STATIC XTAPI XTSTATUS FreePool(IN PVOID VirtualAddress,
|
||||
IN ULONG Tag);
|
||||
STATIC XTAPI VOID InitializeBigAllocationsTable(VOID);
|
||||
|
||||
private:
|
||||
STATIC XTAPI XTSTATUS AllocateNonPagedPoolPages(IN PFN_COUNT Pages,
|
||||
OUT PVOID *Memory);
|
||||
STATIC XTAPI XTSTATUS AllocatePagedPoolPages(IN PFN_COUNT Pages,
|
||||
OUT PVOID *Memory);
|
||||
STATIC XTINLINE ULONG ComputeHash(IN PVOID VirtualAddress);
|
||||
STATIC XTAPI BOOLEAN ExpandBigAllocationsTable(VOID);
|
||||
STATIC XTAPI XTSTATUS FreeNonPagedPoolPages(IN PVOID VirtualAddress,
|
||||
OUT PPFN_NUMBER PagesFreed);
|
||||
STATIC XTAPI XTSTATUS FreePagedPoolPages(IN PVOID VirtualAddress,
|
||||
OUT PPFN_NUMBER PagesFreed);
|
||||
STATIC XTAPI BOOLEAN RegisterBigAllocationTag(IN PVOID VirtualAddress,
|
||||
IN ULONG Key,
|
||||
IN ULONG NumberOfPages,
|
||||
IN MMPOOL_TYPE PoolType);
|
||||
STATIC XTAPI ULONG UnregisterBigAllocationTag(IN PVOID VirtualAddress,
|
||||
OUT PULONG_PTR NumberOfPages,
|
||||
IN MMPOOL_TYPE PoolType);
|
||||
};
|
||||
}
|
||||
|
||||
|
||||
@@ -18,20 +18,49 @@ namespace MM
|
||||
class Pool
|
||||
{
|
||||
protected:
|
||||
STATIC POOL_DESCRIPTOR NonPagedPoolDescriptor;
|
||||
STATIC PFN_NUMBER NonPagedPoolFrameEnd;
|
||||
STATIC PFN_NUMBER NonPagedPoolFrameStart;
|
||||
STATIC LIST_ENTRY NonPagedPoolFreeList[MM_MAX_FREE_PAGE_LIST_HEADS];
|
||||
STATIC ULONG PoolSecureCookie;
|
||||
STATIC PPOOL_DESCRIPTOR PoolVector[2];
|
||||
|
||||
public:
|
||||
STATIC XTAPI MMPOOL_TYPE DeterminePoolType(IN PVOID VirtualAddress);
|
||||
STATIC XTAPI VOID InitializeNonPagedPool(VOID);
|
||||
STATIC XTAPI VOID InitializePagedPool(VOID);
|
||||
STATIC XTAPI VOID InitializePoolSecurity(VOID);
|
||||
|
||||
protected:
|
||||
STATIC XTAPI PLIST_ENTRY DecodePoolLink(IN PLIST_ENTRY PoolLink);
|
||||
STATIC XTAPI PLIST_ENTRY EncodePoolLink(IN PLIST_ENTRY PoolLink);
|
||||
STATIC XTAPI PPOOL_HEADER GetPoolBlock(IN PPOOL_HEADER Header, IN SSIZE_T Index);
|
||||
STATIC XTAPI PPOOL_HEADER GetPoolEntry(IN PVOID Payload);
|
||||
STATIC XTAPI PLIST_ENTRY GetPoolFreeBlock(IN PPOOL_HEADER Header);
|
||||
STATIC XTAPI PPOOL_HEADER GetPoolNextBlock(IN PPOOL_HEADER Header);
|
||||
STATIC XTAPI PPOOL_HEADER GetPoolPreviousBlock(IN PPOOL_HEADER Header);
|
||||
STATIC XTAPI VOID InsertPoolHeadList(IN PLIST_ENTRY ListHead,
|
||||
IN PLIST_ENTRY Entry);
|
||||
STATIC XTAPI VOID InsertPoolTailList(IN PLIST_ENTRY ListHead,
|
||||
IN PLIST_ENTRY Entry);
|
||||
STATIC XTAPI BOOLEAN PoolListEmpty(IN PLIST_ENTRY ListHead);
|
||||
STATIC XTAPI VOID RemovePoolEntryList(IN PLIST_ENTRY Entry);
|
||||
STATIC XTAPI PLIST_ENTRY RemovePoolHeadList(IN PLIST_ENTRY ListHead);
|
||||
STATIC XTAPI PLIST_ENTRY RemovePoolTailList(IN PLIST_ENTRY ListHead);
|
||||
STATIC XTAPI VOID VerifyPoolBlocks(IN PVOID Block);
|
||||
STATIC XTAPI VOID VerifyPoolHeader(IN PPOOL_HEADER Entry);
|
||||
STATIC XTAPI VOID VerifyPoolLinks(IN PLIST_ENTRY ListHead);
|
||||
STATIC XTAPI VOID VerifyRunLevel(IN MMPOOL_TYPE PoolType,
|
||||
IN SIZE_T Bytes,
|
||||
IN PVOID Entry);
|
||||
|
||||
private:
|
||||
STATIC XTAPI VOID InitializePoolDescriptor(IN PPOOL_DESCRIPTOR Descriptor,
|
||||
IN MMPOOL_TYPE PoolType,
|
||||
IN ULONG Index,
|
||||
IN ULONG Threshold,
|
||||
IN PVOID LockAddress);
|
||||
STATIC XTAPI VOID InitializePoolListHead(IN PLIST_ENTRY ListHead);
|
||||
STATIC XTAPI VOID MapNonPagedPool(VOID);
|
||||
};
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user