IntelFrameworkPkg FrameworkUefiLib: Fix in EfiLocateProtocolBuffer()

[mirror_edk2.git] / IntelFrameworkPkg / Library / FrameworkUefiLib / UefiLib.c

diff --git a/IntelFrameworkPkg/Library/FrameworkUefiLib/UefiLib.c b/IntelFrameworkPkg/Library/FrameworkUefiLib/UefiLib.c
index 61a32e29760646bc17696b5d0be4d52a7a601abc..3eeb12110a90466192097e186575c18c60f3765c 100644 (file)
--- a/IntelFrameworkPkg/Library/FrameworkUefiLib/UefiLib.c
+++ b/IntelFrameworkPkg/Library/FrameworkUefiLib/UefiLib.c
@@ -865,6 +865,7 @@ LookupUnicodeString2 (
 **/\r
 EFI_STATUS\r
 EFIAPI\r
+AddUnicodeString (\r
   IN     CONST CHAR8               *Language,\r
   IN     CONST CHAR8               *SupportedLanguages,\r
   IN OUT EFI_UNICODE_STRING_TABLE  **UnicodeStringTable,\r
@@ -1485,7 +1486,7 @@ CHAR8 *
 EFIAPI\r
 GetBestLanguage (\r
   IN CONST CHAR8  *SupportedLanguages, \r
-  IN BOOLEAN      Iso639Language,\r
+  IN UINTN        Iso639Language,\r
   ...\r
   )\r
 {\r
@@ -1649,6 +1650,10 @@ EfiLocateProtocolBuffer (
                   (VOID **)Buffer\r
                   );\r
   if (EFI_ERROR (Status)) {\r
+    //\r
+    // Free the handle buffer\r
+    //\r
+    gBS->FreePool (HandleBuffer);\r
     return EFI_OUT_OF_RESOURCES;\r
   }\r
   ZeroMem (*Buffer, NoHandles * sizeof (VOID *));\r