ArmVirtPkg/ArmVirtPlatformLib: eliminate unchecked PcdSetXX() calls

[mirror_edk2.git] / ArmVirtPkg / Library / ArmVirtPlatformLib / Virt.c

diff --git a/ArmVirtPkg/Library/ArmVirtPlatformLib/Virt.c b/ArmVirtPkg/Library/ArmVirtPlatformLib/Virt.c
index 7a0fc0e75e37c6c48709c1df2e9fc95eaa0ed321..fcaf3c681a977ca72c96dcd418e6675942095adb 100644 (file)
--- a/ArmVirtPkg/Library/ArmVirtPlatformLib/Virt.c
+++ b/ArmVirtPkg/Library/ArmVirtPlatformLib/Virt.c
@@ -70,13 +70,14 @@ ArmPlatformInitializeSystemMemory (
   VOID\r
   )\r
 {\r
-  VOID         *DeviceTreeBase;\r
-  INT32        Node, Prev;\r
-  UINT64       NewBase, CurBase;\r
-  UINT64       NewSize, CurSize;\r
-  CONST CHAR8  *Type;\r
-  INT32        Len;\r
-  CONST UINT64 *RegProp;\r
+  VOID          *DeviceTreeBase;\r
+  INT32         Node, Prev;\r
+  UINT64        NewBase, CurBase;\r
+  UINT64        NewSize, CurSize;\r
+  CONST CHAR8   *Type;\r
+  INT32         Len;\r
+  CONST UINT64  *RegProp;\r
+  RETURN_STATUS PcdStatus;\r
 \r
   NewBase = 0;\r
   NewSize = 0;\r
@@ -131,7 +132,8 @@ ArmPlatformInitializeSystemMemory (
   // Make sure the start of DRAM matches our expectation\r
   //\r
   ASSERT (FixedPcdGet64 (PcdSystemMemoryBase) == NewBase);\r
-  PcdSet64 (PcdSystemMemorySize, NewSize);\r
+  PcdStatus = PcdSet64S (PcdSystemMemorySize, NewSize);\r
+  ASSERT_RETURN_ERROR (PcdStatus);\r
 \r
   //\r
   // We need to make sure that the machine we are running on has at least\r