Fix memory out of band access in AsmFxSave() and AsmFxRestore().

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

diff --git a/MdePkg/Library/BaseLib/X86FxRestore.c b/MdePkg/Library/BaseLib/X86FxRestore.c
index d93cc91df3f6b7629a21f70090d47cbf2c99acc3..b2a02fe25f981b7dfe0c59f77e941aeebfab92b1 100644 (file)
--- a/MdePkg/Library/BaseLib/X86FxRestore.c
+++ b/MdePkg/Library/BaseLib/X86FxRestore.c
@@ -1,7 +1,7 @@
 /** @file\r
   IA-32/x64 AsmFxRestore()\r
 \r
-  Copyright (c) 2006 - 2010, Intel Corporation. All rights reserved.<BR>\r
+  Copyright (c) 2006 - 2012, Intel Corporation. All rights reserved.<BR>\r
   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
@@ -43,7 +43,7 @@ AsmFxRestore (
   //\r
   // Check the flag recorded by AsmFxSave()\r
   //\r
-  ASSERT (0xAA5555AA == *(UINT32 *) (&Buffer[sizeof (IA32_FX_BUFFER) - 4]));\r
+  ASSERT (0xAA5555AA == *(UINT32 *) (&Buffer->Buffer[sizeof (Buffer->Buffer) - 4]));\r
 \r
   InternalX86FxRestore (Buffer);\r
 }\r