1 files changed, 3 insertions, 7 deletions
diff --git a/MdeModulePkg/Library/GenericBdsLib/BdsConnect.c b/MdeModulePkg/Library/GenericBdsLib/BdsConnect.c
index 07d5c38..5e7d13f 100644
--- a/MdeModulePkg/Library/GenericBdsLib/BdsConnect.c
+++ b/MdeModulePkg/Library/GenericBdsLib/BdsConnect.c
@@ -52,10 +52,6 @@ BdsLibConnectAll (
   have. After this we should get all the device work and console avariable
   if the system have console device.
 
-  None
-
-  @return None
-
 **/
 VOID
 BdsLibGenericConnectAll (
@@ -180,7 +176,7 @@ BdsLibConnectDevicePath (
   } while (DevicePath != NULL);
 
   if (CopyOfDevicePath != NULL) {
-    gBS->FreePool (CopyOfDevicePath);
+    SafeFreePool (CopyOfDevicePath);
   }
   //
   // All handle with DevicePath exists in the handle database
@@ -224,7 +220,7 @@ BdsLibConnectAllEfi (
     Status = gBS->ConnectController (HandleBuffer[Index], NULL, NULL, TRUE);
   }
 
-  gBS->FreePool (HandleBuffer);
+  SafeFreePool (HandleBuffer);
 
   return EFI_SUCCESS;
 }
@@ -267,7 +263,7 @@ BdsLibDisconnectAllEfi (
     Status = gBS->DisconnectController (HandleBuffer[Index], NULL, NULL);
   }
 
-  gBS->FreePool (HandleBuffer);
+  SafeFreePool (HandleBuffer);
 
   return EFI_SUCCESS;
 }