newline added at end

[mirror_edk2.git] / EdkUnixPkg / Dxe / PlatformBds / Generic / BootMaint / BmLib.c

diff --git a/EdkUnixPkg/Dxe/PlatformBds/Generic/BootMaint/BmLib.c b/EdkUnixPkg/Dxe/PlatformBds/Generic/BootMaint/BmLib.c
index 6f4638662166002f970db497091ca93d714f3c35..e9245d2fb62a38af9d4c43a749bca320546d610f 100644 (file)
--- a/EdkUnixPkg/Dxe/PlatformBds/Generic/BootMaint/BmLib.c
+++ b/EdkUnixPkg/Dxe/PlatformBds/Generic/BootMaint/BmLib.c
@@ -464,12 +464,16 @@ Returns:
   EFI_GUID          HiiGuid;\r
   EFI_HII_PROTOCOL  *Hii;\r
 \r
-  HandleBufferLength  = 0x1000;\r
+  //\r
+  // Initialize params.\r
+  //\r
+  HandleBufferLength  = 0;\r
   HiiHandleBuffer     = NULL;\r
+  \r
   Status = gBS->LocateProtocol (\r
                   &gEfiHiiProtocolGuid,\r
                   NULL,\r
-                  &Hii\r
+                  (VOID**) &Hii\r
                   );\r
   if (EFI_ERROR (Status)) {\r
     *String = NULL;\r
@@ -478,12 +482,9 @@ Returns:
   //\r
   // Get all the Hii handles\r
   //\r
-  HiiHandleBuffer = AllocateZeroPool (HandleBufferLength);\r
-  ASSERT (HiiHandleBuffer != NULL);\r
-\r
-  Status = Hii->FindHandles (Hii, &HandleBufferLength, HiiHandleBuffer);\r
+  Status = BdsLibGetHiiHandles (Hii, &HandleBufferLength, &HiiHandleBuffer);\r
   ASSERT_EFI_ERROR (Status);\r
-\r
+  \r
   //\r
   // Get the Hii Handle that matches the StructureNode->ProducerName\r
   //\r
@@ -579,7 +580,7 @@ EfiLibStrFromDatahub (
   Status = gBS->LocateProtocol (\r
                   &gEfiDataHubProtocolGuid,\r
                   NULL,\r
-                  &Datahub\r
+                  (VOID**) &Datahub\r
                   );\r
   if (EFI_ERROR (Status)) {\r
     return NULL;\r