91 lines
4.8 KiB
C++
91 lines
4.8 KiB
C++
/**
|
|
* PROJECT: ExectOS
|
|
* COPYRIGHT: See COPYING.md in the top level directory
|
|
* FILE: xtoskrnl/includes/mm/alloc.hh
|
|
* DESCRIPTION: Memory Manager pool allocator
|
|
* DEVELOPERS: Aiken Harris <harraiken91@gmail.com>
|
|
*/
|
|
|
|
#ifndef __XTOSKRNL_MM_ALLOC_HH
|
|
#define __XTOSKRNL_MM_ALLOC_HH
|
|
|
|
#include <xtos.hh>
|
|
#include <mm/pool.hh>
|
|
|
|
|
|
/* Memory Manager */
|
|
namespace MM
|
|
{
|
|
class Allocator final : private Pool
|
|
{
|
|
private:
|
|
STATIC PPOOL_TRACKING_TABLE AllocationsTrackingExpansionTable;
|
|
STATIC SIZE_T AllocationsTrackingExpansionTableSize;
|
|
STATIC PPOOL_TRACKING_TABLE AllocationsTrackingTable;
|
|
STATIC KSPIN_LOCK AllocationsTrackingTableLock;
|
|
STATIC SIZE_T AllocationsTrackingTableMask;
|
|
STATIC SIZE_T AllocationsTrackingTableSize;
|
|
STATIC ULONG BigAllocationsInUse;
|
|
STATIC PPOOL_TRACKING_BIG_ALLOCATIONS BigAllocationsTrackingTable;
|
|
STATIC SIZE_T BigAllocationsTrackingTableHash;
|
|
STATIC KSPIN_LOCK BigAllocationsTrackingTableLock;
|
|
STATIC SIZE_T BigAllocationsTrackingTableSize;
|
|
STATIC PPOOL_TRACKING_TABLE TagTables[MM_POOL_TRACKING_TABLES];
|
|
|
|
public:
|
|
STATIC XTAPI XTSTATUS AllocatePages(IN MMPOOL_TYPE PoolType,
|
|
IN SIZE_T Bytes,
|
|
OUT PVOID *Memory);
|
|
STATIC XTAPI XTSTATUS AllocatePool(IN MMPOOL_TYPE PoolType,
|
|
IN SIZE_T Bytes,
|
|
OUT PVOID *Memory);
|
|
STATIC XTAPI XTSTATUS AllocatePool(IN MMPOOL_TYPE PoolType,
|
|
IN SIZE_T Bytes,
|
|
OUT PVOID *Memory,
|
|
IN ULONG Tag);
|
|
STATIC XTAPI XTSTATUS FreePages(IN PVOID VirtualAddress);
|
|
STATIC XTAPI XTSTATUS FreePages(IN PVOID VirtualAddress,
|
|
OUT PPFN_NUMBER PagesFreed);
|
|
STATIC XTAPI XTSTATUS FreePool(IN PVOID VirtualAddress);
|
|
STATIC XTAPI XTSTATUS FreePool(IN PVOID VirtualAddress,
|
|
IN ULONG Tag);
|
|
STATIC XTAPI VOID InitializeAllocationsTracking(VOID);
|
|
STATIC XTAPI VOID InitializeBigAllocationsTracking(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 XTINLINE ULONG ComputeHash(IN ULONG Tag,
|
|
IN ULONG TableMask);
|
|
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 VOID RegisterAllocationTag(IN ULONG Tag,
|
|
IN SIZE_T Bytes,
|
|
IN MMPOOL_TYPE PoolType);
|
|
STATIC XTAPI VOID RegisterAllocationTagExpansion(IN ULONG Tag,
|
|
IN SIZE_T Bytes,
|
|
IN MMPOOL_TYPE PoolType);
|
|
STATIC XTAPI BOOLEAN RegisterBigAllocationTag(IN PVOID VirtualAddress,
|
|
IN ULONG Tag,
|
|
IN ULONG Pages,
|
|
IN MMPOOL_TYPE PoolType);
|
|
STATIC XTAPI VOID UnregisterAllocationTag(IN ULONG Tag,
|
|
IN SIZE_T Bytes,
|
|
IN MMPOOL_TYPE PoolType);
|
|
STATIC XTAPI VOID UnregisterAllocationTagExpansion(IN ULONG Tag,
|
|
IN SIZE_T Bytes,
|
|
IN MMPOOL_TYPE PoolType);
|
|
STATIC XTAPI ULONG UnregisterBigAllocationTag(IN PVOID VirtualAddress,
|
|
OUT PULONG_PTR Pages,
|
|
IN MMPOOL_TYPE PoolType);
|
|
};
|
|
}
|
|
|
|
#endif /* __XTOSKRNL_MM_ALLOC_HH */
|