move header files in MdeModulePkg\Core\Dxe except DxeMain.h into their corresponding...

[mirror_edk2.git] / MdeModulePkg / Core / Dxe / Hand / DriverSupport.c

diff --git a/MdeModulePkg/Core/Dxe/Hand/DriverSupport.c b/MdeModulePkg/Core/Dxe/Hand/DriverSupport.c
index 829215ba300a7e50eb8797a4f9ec4991967ab57d..203a6e6b044fc04d5737855c9cef27b397553570 100644 (file)
--- a/MdeModulePkg/Core/Dxe/Hand/DriverSupport.c
+++ b/MdeModulePkg/Core/Dxe/Hand/DriverSupport.c
@@ -13,6 +13,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 **/\r
 \r
 #include "DxeMain.h"\r
+#include "Handle.h"\r
 \r
 \r
 //\r
@@ -143,6 +144,7 @@ CoreConnectController (
     //\r
     ChildHandleBuffer = AllocatePool (ChildHandleCount * sizeof(EFI_HANDLE));\r
     if (ChildHandleBuffer == NULL) {\r
+      CoreReleaseProtocolLock ();\r
       return EFI_OUT_OF_RESOURCES;\r
     }\r
 \r