[BOOT:MM] Begin work on memory manager

BOOT/ENVIRON/LIB/EFI/efiinit.c

@@ -680,7 +680,7 @@ Return Value:
     InputParameters->Signature = BOOT_INPUT_PARAMETERS_SIGNATURE;
     InputParameters->Version = BOOT_INPUT_PARAMETERS_VERSION;
     InputParameters->MachineType = BOOT_MACHINE_TYPE;
-    InputParameters->TranslationType = BOOT_TRANSLATION_TYPE;
+    InputParameters->TranslationType = BOOT_TRANSLATION_TYPE_NONE;
     InputParameters->ImageBase = LoadedImage->ImageBase;
     InputParameters->ImageSize = LoadedImage->ImageSize;
 

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;
+}

BOOT/ENVIRON/LIB/bootlib.c

@@ -56,6 +56,7 @@ Return Value:
 
 {
     NTSTATUS Status;
+    PBOOT_MEMORY_INFO MemoryInfo;
     PBOOT_INPUT_APPLICATION_ENTRY ApplicationEntry;
     PBOOT_FIRMWARE_DATA FirmwareData;
     PBOOT_BLOCK_IDENTIFIER BlockDevice;
@@ -75,6 +76,7 @@ Return Value:
     //
     // Calculate structure addresses from offsets.
     //
+    MemoryInfo = (PBOOT_MEMORY_INFO)((PUCHAR)InputParameters + InputParameters->MemoryInfoOffset);
     ApplicationEntry = (PBOOT_INPUT_APPLICATION_ENTRY)((PUCHAR)InputParameters + InputParameters->ApplicationEntryOffset);
     BlpBootDevice = (PBOOT_DEVICE)((PUCHAR)InputParameters + InputParameters->BootDeviceOffset);
     FirmwareData = (PBOOT_FIRMWARE_DATA)((PUCHAR)InputParameters + InputParameters->FirmwareDataOffset);
@@ -99,7 +101,7 @@ Return Value:
     // Check application entry signature.
     //
     if (ApplicationEntry->Signature != BOOT_INPUT_APPLICATION_ENTRY_SIGNATURE) {
-        DebugPrint(L"Application entry signature is invalid\r\n");
+        DebugPrint(L"InitializeLibrary(): ApplicationEntry Signature is invalid\r\n");
         return STATUS_INVALID_PARAMETER_9;
     }
 
@@ -111,6 +113,14 @@ Return Value:
     RtlCopyMemory(&BlpApplicationEntry.BcdIdentifier, &ApplicationEntry->BcdIdentifier, sizeof(GUID));
     BlpApplicationEntry.Options = &ApplicationEntry->Options;
 
+    //
+    // Initialize memory manager.
+    //
+    Status = BlpMmInitialize(MemoryInfo, InputParameters->TranslationType, LibraryParameters);
+    if (!NT_SUCCESS(Status)) {
+        return Status;
+    }
+
     //
     // Print debug information.
     // TODO: Remove this once the project is more stable?