1) Update some comment.

2) Add in handle for the out_of_resource cases
3) Add in Done label to clean up the code.

git-svn-id: https://edk2.svn.sourceforge.net/svnroot/edk2/trunk/edk2@5853 6f19259b-4bc3-4df7-8a09-765794883524

2 files changed, 26 insertions, 14 deletions

diff --git a/MdeModulePkg/Core/Dxe/Hand/DriverSupport.c b/MdeModulePkg/Core/Dxe/Hand/DriverSupport.c
index 9db124c..829215b 100644
--- a/MdeModulePkg/Core/Dxe/Hand/DriverSupport.c
+++ b/MdeModulePkg/Core/Dxe/Hand/DriverSupport.c
@@ -73,6 +73,10 @@ CoreConnectController (
   AlignedRemainingDevicePath = NULL;

   if (RemainingDevicePath != NULL) {

     AlignedRemainingDevicePath = DuplicateDevicePath (RemainingDevicePath);

+

+    if (AlignedRemainingDevicePath == NULL) {

+      return EFI_OUT_OF_RESOURCES;

+    }

   }

 

   //

@@ -138,6 +142,9 @@ CoreConnectController (
     // Allocate a handle buffer for ControllerHandle's children

     //

     ChildHandleBuffer = AllocatePool (ChildHandleCount * sizeof(EFI_HANDLE));

+    if (ChildHandleBuffer == NULL) {

+      return EFI_OUT_OF_RESOURCES;

+    }

 

     //

     // Fill in a handle buffer with ControllerHandle's children

diff --git a/MdeModulePkg/Core/Dxe/Hand/Notify.c b/MdeModulePkg/Core/Dxe/Hand/Notify.c
index 5f4ae19..407e3ff 100644
--- a/MdeModulePkg/Core/Dxe/Hand/Notify.c
+++ b/MdeModulePkg/Core/Dxe/Hand/Notify.c
@@ -1,5 +1,5 @@
 /** @file

-  UEFI notify infrastructure

+  Support functions for UEFI protocol notification infrastructure.

 

 Copyright (c) 2006 - 2008, Intel Corporation. <BR>

 All rights reserved. This program and the accompanying materials

@@ -213,8 +213,8 @@ CoreReinstallProtocolInterface (
   //

   Prot = CoreFindProtocolInterface (UserHandle, Protocol, OldInterface);

   if (Prot == NULL) {

-    CoreReleaseProtocolLock ();

-    return EFI_NOT_FOUND;

+    Status = EFI_NOT_FOUND;

+    goto Done;

   }

 

   //

@@ -228,8 +228,7 @@ CoreReinstallProtocolInterface (
     //

     // One or more drivers refused to release, so return the error

     //

-    CoreReleaseProtocolLock ();

-    return Status;

+    goto Done;

   }

 

   //

@@ -238,8 +237,8 @@ CoreReinstallProtocolInterface (
   Prot = CoreRemoveInterfaceFromProtocol (Handle, Protocol, OldInterface);

 

   if (Prot == NULL) {

-    CoreReleaseProtocolLock ();

-    return EFI_NOT_FOUND;

+    Status = EFI_NOT_FOUND;

+    goto Done;

   }

 

   ProtEntry = Prot->Protocol;

@@ -265,12 +264,15 @@ CoreReinstallProtocolInterface (
   // Release the lock and connect all drivers to UserHandle

   //

   CoreReleaseProtocolLock ();

-  Status = CoreConnectController (

-             UserHandle,

-             NULL,

-             NULL,

-             TRUE

-             );

+  //

+  // Return code is ignored on purpose.

+  //

+  CoreConnectController (

+    UserHandle,

+    NULL,

+    NULL,

+    TRUE

+    );

   CoreAcquireProtocolLock ();

 

   //

@@ -280,5 +282,8 @@ CoreReinstallProtocolInterface (
 

   CoreReleaseProtocolLock ();

 

-  return EFI_SUCCESS;

+  Status = EFI_SUCCESS;

+

+Done:

+  return Status;

 }