X-Git-Url: https://git.proxmox.com/?p=mirror_edk2.git;a=blobdiff_plain;f=MdeModulePkg%2FCore%2FDxe%2FHand%2FDriverSupport.c;h=9db124c9f3f672190342a4274af8d42e808aec67;hp=773165232fb8ed79559bc034f9b3e2581e4df341;hb=9c4ac31cca01b4a503c36616770ea3157bf3bb9e;hpb=fb53e6cc348e96709da625a2334faeabd4f31523

diff --git a/MdeModulePkg/Core/Dxe/Hand/DriverSupport.c b/MdeModulePkg/Core/Dxe/Hand/DriverSupport.c
index 773165232f..9db124c9f3 100644
--- a/MdeModulePkg/Core/Dxe/Hand/DriverSupport.c
+++ b/MdeModulePkg/Core/Dxe/Hand/DriverSupport.c
@@ -12,7 +12,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 
 **/
 
-#include <DxeMain.h>
+#include "DxeMain.h"
 
 
 //
@@ -72,7 +72,7 @@ CoreConnectController (
   //
   AlignedRemainingDevicePath = NULL;
   if (RemainingDevicePath != NULL) {
-    AlignedRemainingDevicePath = CoreDuplicateDevicePath (RemainingDevicePath);
+    AlignedRemainingDevicePath = DuplicateDevicePath (RemainingDevicePath);
   }
 
   //
@@ -137,7 +137,7 @@ CoreConnectController (
     //
     // Allocate a handle buffer for ControllerHandle's children
     //
-    ChildHandleBuffer = CoreAllocateBootServicesPool (ChildHandleCount * sizeof(EFI_HANDLE));
+    ChildHandleBuffer = AllocatePool (ChildHandleCount * sizeof(EFI_HANDLE));
 
     //
     // Fill in a handle buffer with ControllerHandle's children
@@ -376,7 +376,7 @@ CoreConnectSingleController (
   //
   // Allocate a duplicate array for the sorted Driver Binding Protocol Instances
   //
-  SortedDriverBindingProtocols = CoreAllocateBootServicesPool (sizeof (VOID *) * DriverBindingHandleCount);
+  SortedDriverBindingProtocols = AllocatePool (sizeof (VOID *) * DriverBindingHandleCount);
   if (SortedDriverBindingProtocols == NULL) {
     CoreFreePool (DriverBindingHandleBuffer);
     return EFI_OUT_OF_RESOURCES;
@@ -707,7 +707,7 @@ CoreDisconnectController (
       goto Done;
     }
 
-    DriverImageHandleBuffer = CoreAllocateBootServicesPool (sizeof (EFI_HANDLE) * DriverImageHandleCount);
+    DriverImageHandleBuffer = AllocatePool (sizeof (EFI_HANDLE) * DriverImageHandleCount);
     if (DriverImageHandleBuffer == NULL) {
       Status = EFI_OUT_OF_RESOURCES;
       goto Done;
@@ -789,7 +789,7 @@ CoreDisconnectController (
       ChildHandleValid = FALSE;
       ChildBuffer = NULL;
       if (ChildBufferCount != 0) {
-        ChildBuffer = CoreAllocateBootServicesPool (sizeof (EFI_HANDLE) * ChildBufferCount);
+        ChildBuffer = AllocatePool (sizeof (EFI_HANDLE) * ChildBufferCount);
         if (ChildBuffer == NULL) {
           Status = EFI_OUT_OF_RESOURCES;
           goto Done;