OcAppleBootCompatLib: Use the original GetMemoryMap for VM pool

Include/Library/OcMemoryLib.h

@@ -393,15 +393,17 @@ typedef struct OC_VMEM_CONTEXT_ {
   in the end of BASE_4GB of RAM. Should be called while boot services are
   still usable.
 
-  @param[out]  Context   Virtual memory pool context.
-  @param[in]   NumPages  Number of pages to be allocated in the pool.
+  @param[out]  Context       Virtual memory pool context.
+  @param[in]   NumPages      Number of pages to be allocated in the pool.
+  @param[in]   GetMemoryMap  Custom GetMemoryMap implementation to use, optional.
 
   @retval EFI_SUCCESS on successful allocation.
 **/
 EFI_STATUS
 VmAllocateMemoryPool (
-  OUT OC_VMEM_CONTEXT  *Context,
-  IN  UINTN            NumPages
+  OUT OC_VMEM_CONTEXT     *Context,
+  IN  UINTN               NumPages,
+  IN  EFI_GET_MEMORY_MAP  GetMemoryMap  OPTIONAL
   );
 
 /**

Library/OcAppleBootCompatLib/KernelSupport.c

@@ -477,7 +477,8 @@ AppleMapPrepareMemoryPool (
 
   Status = VmAllocateMemoryPool (
     &BootCompat->KernelState.VmContext,
-    OC_DEFAULT_VMEM_PAGE_COUNT
+    OC_DEFAULT_VMEM_PAGE_COUNT,
+    BootCompat->ServicePtrs.GetMemoryMap
     );
 
   if (EFI_ERROR (Status)) {

Library/OcMemoryLib/VirtualMemory.c

@@ -143,8 +143,9 @@ GetPhysicalAddress (
 
 EFI_STATUS
 VmAllocateMemoryPool (
-  OUT OC_VMEM_CONTEXT  *Context,
-  IN  UINTN            NumPages
+  OUT OC_VMEM_CONTEXT     *Context,
+  IN  UINTN               NumPages,
+  IN  EFI_GET_MEMORY_MAP  GetMemoryMap  OPTIONAL
   )
 {
   EFI_STATUS              Status;
@@ -155,7 +156,7 @@ VmAllocateMemoryPool (
     EfiBootServicesData,
     NumPages,
     &Addr,
-    NULL,
+    GetMemoryMap,
     NULL
     );