Fix the bug that if code is run above 4G, AsmDisablePaging64 will be failure.

[mirror_edk2.git] / MdePkg / Library / BaseLib / SetJump.c

diff --git a/MdePkg/Library/BaseLib/SetJump.c b/MdePkg/Library/BaseLib/SetJump.c
index f03479ae81e095a1306aaf418fb38b66ee2ff688..81e2b578f95392147a603958c5b6da5946b5f0ee 100644 (file)
--- a/MdePkg/Library/BaseLib/SetJump.c
+++ b/MdePkg/Library/BaseLib/SetJump.c
@@ -1,7 +1,7 @@
 /** @file\r
   Internal ASSERT () functions for SetJump.\r
 \r
-  Copyright (c) 2006, Intel Corporation\r
+  Copyright (c) 2006 - 2008, Intel Corporation\r
   All rights reserved. This program and the accompanying materials\r
   are licensed and made available under the terms and conditions of the BSD License\r
   which accompanies this distribution.  The full text of the license may be found at\r
@@ -10,10 +10,13 @@
   THE PROGRAM IS DISTRIBUTED UNDER THE BSD LICENSE ON AN "AS IS" BASIS,\r
   WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.\r
 \r
-  Module Name:  SetJump.c\r
-\r
 **/\r
 \r
+\r
+\r
+\r
+#include "BaseLibInternals.h"\r
+\r
 /**\r
   Worker function that checks ASSERT condition for JumpBuffer\r
 \r
@@ -33,7 +36,5 @@ InternalAssertJumpBuffer (
 {\r
   ASSERT (JumpBuffer != NULL);\r
 \r
-#ifdef MDE_CPU_IPF\r
-  ASSERT (((UINTN)JumpBuffer & 0xf) == 0);\r
-#endif\r
+  ASSERT (((UINTN)JumpBuffer & (BASE_LIBRARY_JUMP_BUFFER_ALIGNMENT - 1)) == 0);\r
 }\r