ArmPlatformPkg/Bds: Fixed update of boot entries

[mirror_edk2.git] / ArmPlatformPkg / PrePi / MainUniCore.c

diff --git a/ArmPlatformPkg/PrePi/MainUniCore.c b/ArmPlatformPkg/PrePi/MainUniCore.c
index c92d9a55b309d0a506e02cbbeefc8af14e56ceeb..43588a50ddb5f5622c7028e1ad846e6ed0118aa0 100644 (file)
--- a/ArmPlatformPkg/PrePi/MainUniCore.c
+++ b/ArmPlatformPkg/PrePi/MainUniCore.c
@@ -17,10 +17,17 @@
 VOID\r
 PrimaryMain (\r
   IN  UINTN                     UefiMemoryBase,\r
+  IN  UINTN                     StacksBase,\r
+  IN  UINTN                     GlobalVariableBase,\r
   IN  UINT64                    StartTimeStamp\r
   )\r
 {\r
-  PrePiMain (UefiMemoryBase, StartTimeStamp);\r
+  DEBUG_CODE_BEGIN();\r
+    // On MPCore system, PeiMpCore.inf should be used instead of PeiUniCore.inf\r
+    ASSERT(ArmIsMpCore() == 0);\r
+  DEBUG_CODE_END();\r
+\r
+  PrePiMain (UefiMemoryBase, StacksBase, GlobalVariableBase, StartTimeStamp);\r
 \r
   // We must never return\r
   ASSERT(FALSE);\r
@@ -28,7 +35,7 @@ PrimaryMain (
 \r
 VOID\r
 SecondaryMain (\r
-  IN  UINTN                     CoreId\r
+  IN  UINTN                     MpId\r
   )\r
 {\r
   // We must never get into this function on UniCore system\r