Remove SafeFreePool from MemoryAllocationLib as this API's name is misleading. Its...

[mirror_edk2.git] / EdkCompatibilityPkg / Compatibility / FrameworkHiiToUefiHiiThunk / Package.c

diff --git a/EdkCompatibilityPkg/Compatibility/FrameworkHiiToUefiHiiThunk/Package.c b/EdkCompatibilityPkg/Compatibility/FrameworkHiiToUefiHiiThunk/Package.c
index 5f2aa18b845feec3b8fbfa59c399ed747d9ef1da..d87b2a28d16d787125b792742a31dbf5898b6be4 100644 (file)
--- a/EdkCompatibilityPkg/Compatibility/FrameworkHiiToUefiHiiThunk/Package.c
+++ b/EdkCompatibilityPkg/Compatibility/FrameworkHiiToUefiHiiThunk/Package.c
@@ -322,7 +322,7 @@ UefiRegisterPackageList(
               &ThunkContext->UefiHiiHandle\r
               );\r
   if (Status == EFI_INVALID_PARAMETER) {\r
-    SafeFreePool (PackageListHeader);\r
+    FreePool (PackageListHeader);\r
     \r
     //\r
     // UEFI HII database does not allow two package list with the same GUID.\r
@@ -386,7 +386,9 @@ Done:
     *Handle = ThunkContext->FwHiiHandle;\r
   }\r
 \r
-  SafeFreePool (PackageListHeader);\r
+       if (PackageListHeader != NULL) {\r
+    FreePool (PackageListHeader);\r
+  }\r
   \r
   return Status;\r
 }\r