/**
 * 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 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,
                                                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 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);
    };
}

#endif /* __XTOSKRNL_MM_ALLOC_HH */