[FREELDR] Initialize a global module list head for freeldr

Author: tkreuzer

boot/freeldr/freeldr/freeldr.c

@@ -64,6 +64,13 @@ VOID __cdecl BootMain(IN PCCH CmdLine)
     /* Initialize I/O subsystem */
     FsInit();
 
+    /* Initialize the module list */
+    if (!PeLdrInitializeModuleList())
+    {
+        UiMessageBoxCritical("Unable to initialize module list.");
+        goto Quit;
+    }
+
     RunLoader();
 
 Quit:

boot/freeldr/freeldr/include/peloader.h

@@ -26,6 +26,9 @@ typedef VOID
 
 extern PELDR_IMPORTDLL_LOAD_CALLBACK PeLdrImportDllLoadCallback;
 
+BOOLEAN
+PeLdrInitializeModuleList(VOID);
+
 BOOLEAN
 PeLdrLoadImage(
     _In_ PCSTR FilePath,

boot/freeldr/freeldr/lib/peloader.c

@@ -25,6 +25,8 @@ DBG_DEFAULT_CHANNEL(PELOADER);
 
 /* GLOBALS *******************************************************************/
 
+LIST_ENTRY FrLdrModuleList;
+
 PELDR_IMPORTDLL_LOAD_CALLBACK PeLdrImportDllLoadCallback = NULL;
 
 #ifdef _WIN64
@@ -518,6 +520,29 @@ PeLdrpScanImportAddressTable(
 
 /* FUNCTIONS *****************************************************************/
 
+BOOLEAN
+PeLdrInitializeModuleList(VOID)
+{
+    PLDR_DATA_TABLE_ENTRY FreeldrDTE;
+
+    InitializeListHead(&FrLdrModuleList);
+
+    /* Allocate a data table entry for freeldr.sys.
+       The base name is scsiport.sys for imports from ntbootdd.sys */
+    if (!PeLdrAllocateDataTableEntry(&FrLdrModuleList,
+                                     "scsiport.sys",
+                                     "freeldr.sys",
+                                     &__ImageBase,
+                                     &FreeldrDTE))
+    {
+        /* Cleanup and bail out */
+        ERR("Failed to allocate DTE for freeldr\n");
+        return FALSE;
+    }
+
+    return TRUE;
+}
+
 PVOID
 PeLdrInitSecurityCookie(PLDR_DATA_TABLE_ENTRY LdrEntry)
 {