[BOOT:MM] Begin work on memory manager

BOOT/ENVIRON/LIB/MM/mm.c

@@ -0,0 +1,87 @@
+/*++
+
+Copyright (c) 2024, Quinn Stephens.
+Provided under the BSD 3-Clause license.
+
+Module Name:
+
+   mm.c
+
+Abstract:
+
+   Provides boot library memory manager routines.
+
+--*/
+
+#include <ntrtl.h>
+#include "bootlib.h"
+#include "mm.h"
+
+#define MAX_STATIC_DESCRIPTOR_COUNT 1024
+
+BOOT_MEMORY_DESCRIPTOR MmStaticMemoryDescriptors[MAX_STATIC_DESCRIPTOR_COUNT];
+PBOOT_MEMORY_DESCRIPTOR MmGlobalMemoryDescriptors;
+ULONG MmGlobalMemoryDescriptorCount;
+
+NTSTATUS
+BlpMmInitialize (
+    IN PBOOT_MEMORY_INFO MemoryInfo,
+    IN ULONG TranslationType,
+    IN PBOOT_LIBRARY_PARAMETERS LibraryParameters
+    )
+
+/*++
+
+Routine Description:
+
+    Initializes the boot memory manager.
+
+Arguments:
+
+    MemoryInfo - pointer to the memory info structure.
+
+    TranslationType - the current translation type being used.
+
+    LibraryParameters - pointer to the library parameters structure.
+
+Return Value:
+
+    STATUS_SUCCESS if successful,
+    STATUS_INVALID_PARAMETER if TranslationType is invalid,
+    Other NTSTATUS value if an error occurs.
+
+--*/
+
+{
+    NTSTATUS Status;
+
+    DebugPrint(L"Initializing memory manager...\r\n");
+
+    //
+    // Check TranslationType.
+    //
+    if (
+        TranslationType > BOOT_TRANSLATION_TYPE_MAX ||
+        LibraryParameters->TranslationType > BOOT_TRANSLATION_TYPE_MAX
+    ) {
+        DebugPrint(L"BlpMmInitialize(): TranslationType is invalid\r\n");
+        return STATUS_INVALID_PARAMETER;
+    }
+
+    //
+    // Initialize memory descriptors.
+    //
+    MmGlobalMemoryDescriptors = &MmStaticMemoryDescriptors[0];
+    MmGlobalMemoryDescriptorCount = MAX_STATIC_DESCRIPTOR_COUNT;
+    RtlZeroMemory(MmGlobalMemoryDescriptors, MAX_STATIC_DESCRIPTOR_COUNT * sizeof(BOOT_MEMORY_DESCRIPTOR));
+
+    //
+    // Initialize the page allocator.
+    //
+    Status = MmPaInitialize(MemoryInfo, LibraryParameters->MinimumPageAllocation);
+    if (!NT_SUCCESS(Status)) {
+        return Status;
+    }
+
+    return STATUS_SUCCESS;
+}

BOOT/ENVIRON/LIB/MM/mmpa.c

@@ -0,0 +1,120 @@
+/*++
+
+Copyright (c) 2024, Quinn Stephens.
+Provided under the BSD 3-Clause license.
+
+Module Name:
+
+   mmpa.c
+
+Abstract:
+
+   Provides memory manager page routines.
+
+--*/
+
+#include "bootlib.h"
+#include "mm.h"
+
+ULONG PapMinimumAllocationCount;
+ULONGLONG PapMinimumPhysicalPage;
+ULONGLONG PapMaximumPhysicalPage;
+
+BOOT_MEMORY_DESCRIPTOR_LIST MmMdlFwAllocationTracker;
+BOOT_MEMORY_DESCRIPTOR_LIST MmMdlBadMemory;
+BOOT_MEMORY_DESCRIPTOR_LIST MmMdlTruncatedMemory;
+BOOT_MEMORY_DESCRIPTOR_LIST MmMdlPersistentMemory;
+BOOT_MEMORY_DESCRIPTOR_LIST MmMdlReservedAllocated;
+BOOT_MEMORY_DESCRIPTOR_LIST MmMdlMappedAllocated;
+BOOT_MEMORY_DESCRIPTOR_LIST MmMdlMappedUnallocated;
+BOOT_MEMORY_DESCRIPTOR_LIST MmMdlUnmappedAllocated;
+BOOT_MEMORY_DESCRIPTOR_LIST MmMdlUnmappedUnallocated;
+
+FORCEINLINE
+VOID
+InitializeList (
+    IN PBOOT_MEMORY_DESCRIPTOR_LIST List
+    )
+
+/*++
+
+Routine Description:
+
+    Initializes a MDL.
+
+Arguments:
+
+    List - the MDL to initialize.
+
+Return Value:
+
+    None.
+
+--*/
+
+{
+    List->Head = NULL;
+    List->Current = NULL;
+    List->Type = MDL_TYPE_PHYSICAL;
+}
+
+NTSTATUS
+MmPaInitialize (
+    IN PBOOT_MEMORY_INFO MemoryInfo,
+    IN ULONG MinimumAllocation
+    )
+
+/*++
+
+Routine Description:
+
+    Initializes the page allocator.
+
+Arguments:
+
+    MemoryInfo - pointer to the memory info structure.
+
+    MinimumAllocation - minimum amount of pages to grow the pool by at a time.
+
+Return Value:
+
+    STATUS_SUCCESS if successful,
+
+--*/
+
+{
+    (VOID)MemoryInfo;
+
+    DebugPrint(L"Initializing page allocator...\r\n");
+
+    //
+    // Initialize page allocator settings.
+    //
+    PapMinimumAllocationCount = MinimumAllocation;
+    PapMinimumPhysicalPage = 1;
+    PapMaximumPhysicalPage = MAXULONGLONG >> PAGE_SHIFT;
+    DebugPrintf(L"Maximum physical page: %x %x\r\n", (ULONG)(PapMaximumPhysicalPage >> 32), (ULONG)(PapMaximumPhysicalPage));
+
+    //
+    // Initialize MDLs.
+    //
+    InitializeList(&MmMdlFwAllocationTracker);
+    InitializeList(&MmMdlBadMemory);
+    InitializeList(&MmMdlTruncatedMemory);
+    InitializeList(&MmMdlPersistentMemory);
+    InitializeList(&MmMdlReservedAllocated);;
+    InitializeList(&MmMdlMappedAllocated);
+    InitializeList(&MmMdlMappedUnallocated);
+    InitializeList(&MmMdlUnmappedAllocated);
+    InitializeList(&MmMdlUnmappedUnallocated);
+
+    //
+    // Get the firmware memory map.
+    //
+    // Status = MmFwGetMemoryMap(&MmMdlUnmappedUnallocated, 0x03);
+    // if (!NT_SUCCESS(Status)) {
+    //     return Status;
+    // }
+
+    return STATUS_SUCCESS;
+}