HII Library Class interface refine.

[mirror_edk2.git] / UnixPkg / CpuRuntimeDxe / Cpu.c

diff --git a/UnixPkg/CpuRuntimeDxe/Cpu.c b/UnixPkg/CpuRuntimeDxe/Cpu.c
index aef2c30ccd3caf01b18f51e20eac92120efb2eec..f2541ee9940de08896aa8bd4af026177eb7dc541 100644 (file)
--- a/UnixPkg/CpuRuntimeDxe/Cpu.c
+++ b/UnixPkg/CpuRuntimeDxe/Cpu.c
@@ -451,8 +451,13 @@ Returns:
   //\r
   // Initialize strings to HII database\r
   //\r
-  HiiLibAddPackages (1, &gEfiProcessorProducerGuid, NULL, &HiiHandle, CpuStrings);\r
-  \r
+  HiiHandle = HiiAddPackages (\r
+                &gEfiProcessorProducerGuid,\r
+                NULL,\r
+                CpuStrings,\r
+                NULL\r
+                );\r
+  ASSERT (HiiHandle != NULL);\r
 \r
   CopyMem (RecordBuffer.Raw, &mCpuDataRecordHeader, HeaderSize);\r
 \r