]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Core/PiSmmCore/PiSmmIpl.c
MdeModulePkg: remove PE/COFF header workaround for ELILO on IPF
[mirror_edk2.git] / MdeModulePkg / Core / PiSmmCore / PiSmmIpl.c
index 26b71f163a9aa05712db14fee8f668553294465e..f8cbe1704bfd78e715c42b983456fa25b1a1efbc 100644 (file)
@@ -1,14 +1,14 @@
 /** @file\r
   SMM IPL that produces SMM related runtime protocols and load the SMM Core into SMRAM\r
 \r
-  Copyright (c) 2009 - 2017, Intel Corporation. All rights reserved.<BR>\r
-  This program and the accompanying materials are licensed and made available \r
-  under the terms and conditions of the BSD License which accompanies this \r
-  distribution.  The full text of the license may be found at        \r
-  http://opensource.org/licenses/bsd-license.php                                            \r
+  Copyright (c) 2009 - 2018, Intel Corporation. All rights reserved.<BR>\r
+  This program and the accompanying materials are licensed and made available\r
+  under the terms and conditions of the BSD License which accompanies this\r
+  distribution.  The full text of the license may be found at\r
+  http://opensource.org/licenses/bsd-license.php\r
 \r
-  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
+  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
 **/\r
 \r
 \r
 #include "PiSmmCorePrivateData.h"\r
 \r
+#define SMRAM_CAPABILITIES  (EFI_MEMORY_WB | EFI_MEMORY_UC)\r
+\r
+#define MEMORY_CACHE_ATTRIBUTES (EFI_MEMORY_UC | EFI_MEMORY_WC | \\r
+                                 EFI_MEMORY_WT | EFI_MEMORY_WB | \\r
+                                 EFI_MEMORY_WP | EFI_MEMORY_UCE)\r
+\r
+#define MEMORY_PAGE_ATTRIBUTES  (EFI_MEMORY_XP | EFI_MEMORY_RP | EFI_MEMORY_RO)\r
+\r
 //\r
 // Function prototypes from produced protocols\r
 //\r
@@ -84,26 +92,36 @@ SmmBase2GetSmstLocation (
 \r
 /**\r
   Communicates with a registered handler.\r
-  \r
-  This function provides a service to send and receive messages from a registered \r
-  UEFI service.  This function is part of the SMM Communication Protocol that may \r
-  be called in physical mode prior to SetVirtualAddressMap() and in virtual mode \r
+\r
+  This function provides a service to send and receive messages from a registered\r
+  UEFI service.  This function is part of the SMM Communication Protocol that may\r
+  be called in physical mode prior to SetVirtualAddressMap() and in virtual mode\r
   after SetVirtualAddressMap().\r
 \r
-  @param[in]     This                The EFI_SMM_COMMUNICATION_PROTOCOL instance.\r
-  @param[in, out] CommBuffer          A pointer to the buffer to convey into SMRAM.\r
-  @param[in, out] CommSize            The size of the data buffer being passed in.On exit, the size of data\r
-                                     being returned. Zero if the handler does not wish to reply with any data.\r
+  @param[in] This                The EFI_SMM_COMMUNICATION_PROTOCOL instance.\r
+  @param[in, out] CommBuffer     A pointer to the buffer to convey into SMRAM.\r
+  @param[in, out] CommSize       The size of the data buffer being passed in. On exit, the size of data\r
+                                 being returned. Zero if the handler does not wish to reply with any data.\r
+                                 This parameter is optional and may be NULL.\r
+\r
+  @retval EFI_SUCCESS            The message was successfully posted.\r
+  @retval EFI_INVALID_PARAMETER  The CommBuffer was NULL.\r
+  @retval EFI_BAD_BUFFER_SIZE    The buffer is too large for the MM implementation.\r
+                                 If this error is returned, the MessageLength field\r
+                                 in the CommBuffer header or the integer pointed by\r
+                                 CommSize, are updated to reflect the maximum payload\r
+                                 size the implementation can accommodate.\r
+  @retval EFI_ACCESS_DENIED      The CommunicateBuffer parameter or CommSize parameter,\r
+                                 if not omitted, are in address range that cannot be\r
+                                 accessed by the MM environment.\r
 \r
-  @retval EFI_SUCCESS                The message was successfully posted.\r
-  @retval EFI_INVALID_PARAMETER      The CommBuffer was NULL.\r
 **/\r
 EFI_STATUS\r
 EFIAPI\r
 SmmCommunicationCommunicate (\r
   IN CONST EFI_SMM_COMMUNICATION_PROTOCOL  *This,\r
   IN OUT VOID                              *CommBuffer,\r
-  IN OUT UINTN                             *CommSize\r
+  IN OUT UINTN                             *CommSize OPTIONAL\r
   );\r
 \r
 /**\r
@@ -195,7 +213,7 @@ SmmIplSetVirtualAddressNotify (
   );\r
 \r
 //\r
-// Data structure used to declare a table of protocol notifications and event \r
+// Data structure used to declare a table of protocol notifications and event\r
 // notifications required by the SMM IPL\r
 //\r
 typedef struct {\r
@@ -263,6 +281,7 @@ EFI_PHYSICAL_ADDRESS       mSmramCacheBase;
 UINT64                     mSmramCacheSize;\r
 \r
 EFI_SMM_COMMUNICATE_HEADER mCommunicateHeader;\r
+EFI_LOAD_FIXED_ADDRESS_CONFIGURATION_TABLE    *mLMFAConfigurationTable = NULL;\r
 \r
 //\r
 // Table of Protocol notification and GUIDed Event notifications that the SMM IPL requires\r
@@ -270,19 +289,19 @@ EFI_SMM_COMMUNICATE_HEADER mCommunicateHeader;
 SMM_IPL_EVENT_NOTIFICATION  mSmmIplEvents[] = {\r
   //\r
   // Declare protocol notification on the SMM Configuration protocol.  When this notification is established,\r
-  // the associated event is immediately signalled, so the notification function will be executed and the \r
+  // the associated event is immediately signalled, so the notification function will be executed and the\r
   // SMM Configuration Protocol will be found if it is already in the handle database.\r
   //\r
   { TRUE,  FALSE, &gEfiSmmConfigurationProtocolGuid,  SmmIplSmmConfigurationEventNotify, &gEfiSmmConfigurationProtocolGuid,  TPL_NOTIFY,   NULL },\r
   //\r
-  // Declare protocol notification on DxeSmmReadyToLock protocols.  When this notification is established, \r
-  // the associated event is immediately signalled, so the notification function will be executed and the \r
+  // Declare protocol notification on DxeSmmReadyToLock protocols.  When this notification is established,\r
+  // the associated event is immediately signalled, so the notification function will be executed and the\r
   // DXE SMM Ready To Lock Protocol will be found if it is already in the handle database.\r
   //\r
   { TRUE,  TRUE,  &gEfiDxeSmmReadyToLockProtocolGuid, SmmIplReadyToLockEventNotify,      &gEfiDxeSmmReadyToLockProtocolGuid, TPL_CALLBACK, NULL },\r
   //\r
   // Declare event notification on EndOfDxe event.  When this notification is established,\r
-  // the associated event is immediately signalled, so the notification function will be executed and the \r
+  // the associated event is immediately signalled, so the notification function will be executed and the\r
   // SMM End Of Dxe Protocol will be found if it is already in the handle database.\r
   //\r
   { FALSE, TRUE,  &gEfiEndOfDxeEventGroupGuid,        SmmIplGuidedEventNotify,           &gEfiEndOfDxeEventGroupGuid,        TPL_CALLBACK, NULL },\r
@@ -302,8 +321,8 @@ SMM_IPL_EVENT_NOTIFICATION  mSmmIplEvents[] = {
   //\r
   { FALSE, TRUE,  &gEfiEventReadyToBootGuid,          SmmIplReadyToLockEventNotify,      &gEfiEventReadyToBootGuid,          TPL_CALLBACK, NULL },\r
   //\r
-  // Declare event notification on Legacy Boot Event Group.  This is used to inform the SMM Core that the platform \r
-  // is performing a legacy boot operation, and that the UEFI environment is no longer available and the SMM Core \r
+  // Declare event notification on Legacy Boot Event Group.  This is used to inform the SMM Core that the platform\r
+  // is performing a legacy boot operation, and that the UEFI environment is no longer available and the SMM Core\r
   // must guarantee that it does not access any UEFI related structures outside of SMRAM.\r
   // It is also to inform the SMM Core to notify SMM driver that system enter legacy boot.\r
   //\r
@@ -319,7 +338,7 @@ SMM_IPL_EVENT_NOTIFICATION  mSmmIplEvents[] = {
   //\r
   { FALSE, FALSE, &gEfiEventReadyToBootGuid,          SmmIplGuidedEventNotify,           &gEfiEventReadyToBootGuid,          TPL_CALLBACK, NULL },\r
   //\r
-  // Declare event notification on SetVirtualAddressMap() Event Group.  This is used to convert gSmmCorePrivate \r
+  // Declare event notification on SetVirtualAddressMap() Event Group.  This is used to convert gSmmCorePrivate\r
   // and mSmmControl2 from physical addresses to virtual addresses.\r
   //\r
   { FALSE, FALSE, &gEfiEventVirtualAddressChangeGuid, SmmIplSetVirtualAddressNotify,     NULL,                               TPL_CALLBACK, NULL },\r
@@ -331,7 +350,7 @@ SMM_IPL_EVENT_NOTIFICATION  mSmmIplEvents[] = {
 \r
 /**\r
   Find the maximum SMRAM cache range that covers the range specified by SmramRange.\r
-  \r
+\r
   This function searches and joins all adjacent ranges of SmramRange into a range to be cached.\r
 \r
   @param   SmramRange       The SMRAM range to search from.\r
@@ -369,7 +388,7 @@ GetSmramCacheRange (
       }\r
     }\r
   } while (FoundAjacentRange);\r
-  \r
+\r
 }\r
 \r
 /**\r
@@ -420,11 +439,11 @@ SmmBase2GetSmstLocation (
   if ((This == NULL) ||(Smst == NULL)) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
-  \r
+\r
   if (!gSmmCorePrivate->InSmm) {\r
     return EFI_UNSUPPORTED;\r
   }\r
-  \r
+\r
   *Smst = gSmmCorePrivate->Smst;\r
 \r
   return EFI_SUCCESS;\r
@@ -432,44 +451,62 @@ SmmBase2GetSmstLocation (
 \r
 /**\r
   Communicates with a registered handler.\r
-  \r
-  This function provides a service to send and receive messages from a registered \r
-  UEFI service.  This function is part of the SMM Communication Protocol that may \r
-  be called in physical mode prior to SetVirtualAddressMap() and in virtual mode \r
+\r
+  This function provides a service to send and receive messages from a registered\r
+  UEFI service.  This function is part of the SMM Communication Protocol that may\r
+  be called in physical mode prior to SetVirtualAddressMap() and in virtual mode\r
   after SetVirtualAddressMap().\r
 \r
   @param[in] This                The EFI_SMM_COMMUNICATION_PROTOCOL instance.\r
-  @param[in, out] CommBuffer          A pointer to the buffer to convey into SMRAM.\r
-  @param[in, out] CommSize            The size of the data buffer being passed in.On exit, the size of data\r
+  @param[in, out] CommBuffer     A pointer to the buffer to convey into SMRAM.\r
+  @param[in, out] CommSize       The size of the data buffer being passed in. On exit, the size of data\r
                                  being returned. Zero if the handler does not wish to reply with any data.\r
+                                 This parameter is optional and may be NULL.\r
 \r
   @retval EFI_SUCCESS            The message was successfully posted.\r
   @retval EFI_INVALID_PARAMETER  The CommBuffer was NULL.\r
+  @retval EFI_BAD_BUFFER_SIZE    The buffer is too large for the MM implementation.\r
+                                 If this error is returned, the MessageLength field\r
+                                 in the CommBuffer header or the integer pointed by\r
+                                 CommSize, are updated to reflect the maximum payload\r
+                                 size the implementation can accommodate.\r
+  @retval EFI_ACCESS_DENIED      The CommunicateBuffer parameter or CommSize parameter,\r
+                                 if not omitted, are in address range that cannot be\r
+                                 accessed by the MM environment.\r
+\r
 **/\r
 EFI_STATUS\r
 EFIAPI\r
 SmmCommunicationCommunicate (\r
   IN CONST EFI_SMM_COMMUNICATION_PROTOCOL  *This,\r
   IN OUT VOID                              *CommBuffer,\r
-  IN OUT UINTN                             *CommSize\r
+  IN OUT UINTN                             *CommSize OPTIONAL\r
   )\r
 {\r
   EFI_STATUS                  Status;\r
   EFI_SMM_COMMUNICATE_HEADER  *CommunicateHeader;\r
   BOOLEAN                     OldInSmm;\r
+  UINTN                       TempCommSize;\r
 \r
   //\r
   // Check parameters\r
   //\r
-  if ((CommBuffer == NULL) || (CommSize == NULL)) {\r
+  if (CommBuffer == NULL) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
 \r
-  //\r
-  // CommSize must hold HeaderGuid and MessageLength\r
-  //\r
-  if (*CommSize < OFFSET_OF (EFI_SMM_COMMUNICATE_HEADER, Data)) {\r
-    return EFI_INVALID_PARAMETER;\r
+  CommunicateHeader = (EFI_SMM_COMMUNICATE_HEADER *) CommBuffer;\r
+\r
+  if (CommSize == NULL) {\r
+    TempCommSize = OFFSET_OF (EFI_SMM_COMMUNICATE_HEADER, Data) + CommunicateHeader->MessageLength;\r
+  } else {\r
+    TempCommSize = *CommSize;\r
+    //\r
+    // CommSize must hold HeaderGuid and MessageLength\r
+    //\r
+    if (TempCommSize < OFFSET_OF (EFI_SMM_COMMUNICATE_HEADER, Data)) {\r
+      return EFI_INVALID_PARAMETER;\r
+    }\r
   }\r
 \r
   //\r
@@ -480,7 +517,7 @@ SmmCommunicationCommunicate (
     // Put arguments for Software SMI in gSmmCorePrivate\r
     //\r
     gSmmCorePrivate->CommunicationBuffer = CommBuffer;\r
-    gSmmCorePrivate->BufferSize          = *CommSize;\r
+    gSmmCorePrivate->BufferSize          = TempCommSize;\r
 \r
     //\r
     // Generate Software SMI\r
@@ -491,17 +528,19 @@ SmmCommunicationCommunicate (
     }\r
 \r
     //\r
-    // Return status from software SMI \r
+    // Return status from software SMI\r
     //\r
-    *CommSize = gSmmCorePrivate->BufferSize;\r
+    if (CommSize != NULL) {\r
+      *CommSize = gSmmCorePrivate->BufferSize;\r
+    }\r
     return gSmmCorePrivate->ReturnStatus;\r
   }\r
 \r
   //\r
   // If we are in SMM, then the execution mode must be physical, which means that\r
   // OS established virtual addresses can not be used.  If SetVirtualAddressMap()\r
-  // has been called, then a direct invocation of the Software SMI is not \r
-  // not allowed so return EFI_INVALID_PARAMETER.\r
+  // has been called, then a direct invocation of the Software SMI is not allowed,\r
+  // so return EFI_INVALID_PARAMETER.\r
   //\r
   if (EfiGoneVirtual()) {\r
     return EFI_INVALID_PARAMETER;\r
@@ -513,7 +552,7 @@ SmmCommunicationCommunicate (
   if ((!gSmmCorePrivate->InSmm) && (!mSmmAccess->OpenState || mSmmAccess->LockState)) {\r
     return EFI_INVALID_PARAMETER;\r
   }\r
\r
+\r
   //\r
   // Save current InSmm state and set InSmm state to TRUE\r
   //\r
@@ -523,20 +562,17 @@ SmmCommunicationCommunicate (
   //\r
   // Before SetVirtualAddressMap(), we are in SMM or SMRAM is open and unlocked, call SmiManage() directly.\r
   //\r
-  CommunicateHeader = (EFI_SMM_COMMUNICATE_HEADER *)CommBuffer;\r
-  *CommSize -= OFFSET_OF (EFI_SMM_COMMUNICATE_HEADER, Data);\r
+  TempCommSize -= OFFSET_OF (EFI_SMM_COMMUNICATE_HEADER, Data);\r
   Status = gSmmCorePrivate->Smst->SmiManage (\r
-                                    &CommunicateHeader->HeaderGuid, \r
-                                    NULL, \r
-                                    CommunicateHeader->Data, \r
-                                    CommSize\r
+                                    &CommunicateHeader->HeaderGuid,\r
+                                    NULL,\r
+                                    CommunicateHeader->Data,\r
+                                    &TempCommSize\r
                                     );\r
-\r
-  //\r
-  // Update CommunicationBuffer, BufferSize and ReturnStatus\r
-  // Communicate service finished, reset the pointer to CommBuffer to NULL\r
-  //\r
-  *CommSize += OFFSET_OF (EFI_SMM_COMMUNICATE_HEADER, Data);\r
+  TempCommSize += OFFSET_OF (EFI_SMM_COMMUNICATE_HEADER, Data);\r
+  if (CommSize != NULL) {\r
+    *CommSize = TempCommSize;\r
+  }\r
 \r
   //\r
   // Restore original InSmm state\r
@@ -563,7 +599,7 @@ SmmIplGuidedEventNotify (
   UINTN                       Size;\r
 \r
   //\r
-  // Use Guid to initialize EFI_SMM_COMMUNICATE_HEADER structure \r
+  // Use Guid to initialize EFI_SMM_COMMUNICATE_HEADER structure\r
   //\r
   CopyGuid (&mCommunicateHeader.HeaderGuid, (EFI_GUID *)Context);\r
   mCommunicateHeader.MessageLength = 1;\r
@@ -628,26 +664,26 @@ SmmIplDxeDispatchEventNotify (
     //\r
     Size = sizeof (mCommunicateHeader);\r
     SmmCommunicationCommunicate (&mSmmCommunication, &mCommunicateHeader, &Size);\r
-    \r
+\r
     //\r
     // Return if there is no request to restart the SMM Core Dispatcher\r
     //\r
     if (mCommunicateHeader.Data[0] != COMM_BUFFER_SMM_DISPATCH_RESTART) {\r
       return;\r
     }\r
-      \r
+\r
     //\r
     // Attempt to reset SMRAM cacheability to UC\r
     // Assume CPU AP is available at this time\r
     //\r
     Status = gDS->SetMemorySpaceAttributes(\r
-                    mSmramCacheBase, \r
+                    mSmramCacheBase,\r
                     mSmramCacheSize,\r
                     EFI_MEMORY_UC\r
                     );\r
     if (EFI_ERROR (Status)) {\r
       DEBUG ((DEBUG_WARN, "SMM IPL failed to reset SMRAM window to EFI_MEMORY_UC\n"));\r
-    }  \r
+    }\r
 \r
     //\r
     // Close all SMRAM ranges to protect SMRAM\r
@@ -694,7 +730,7 @@ SmmIplSmmConfigurationEventNotify (
   ASSERT_EFI_ERROR (Status);\r
 \r
   //\r
-  // Set flag to indicate that the SMM Entry Point has been registered which \r
+  // Set flag to indicate that the SMM Entry Point has been registered which\r
   // means that SMIs are now fully operational.\r
   //\r
   gSmmCorePrivate->SmmEntryPointRegistered = TRUE;\r
@@ -730,7 +766,7 @@ SmmIplReadyToLockEventNotify (
   if (mSmmLocked) {\r
     return;\r
   }\r
-  \r
+\r
   //\r
   // Make sure this notification is for this handler\r
   //\r
@@ -741,7 +777,7 @@ SmmIplReadyToLockEventNotify (
     }\r
   } else {\r
     //\r
-    // If SMM is not locked yet and we got here from gEfiEventReadyToBootGuid being \r
+    // If SMM is not locked yet and we got here from gEfiEventReadyToBootGuid being\r
     // signaled, then gEfiDxeSmmReadyToLockProtocolGuid was not installed as expected.\r
     // Print a warning on debug builds.\r
     //\r
@@ -761,10 +797,10 @@ SmmIplReadyToLockEventNotify (
   // Lock the SMRAM (Note: Locking SMRAM may not be supported on all platforms)\r
   //\r
   mSmmAccess->Lock (mSmmAccess);\r
-  \r
+\r
   //\r
-  // Close protocol and event notification events that do not apply after the \r
-  // DXE SMM Ready To Lock Protocol has been installed or the Ready To Boot \r
+  // Close protocol and event notification events that do not apply after the\r
+  // DXE SMM Ready To Lock Protocol has been installed or the Ready To Boot\r
   // event has been signalled.\r
   //\r
   for (Index = 0; mSmmIplEvents[Index].NotifyFunction != NULL; Index++) {\r
@@ -782,7 +818,7 @@ SmmIplReadyToLockEventNotify (
   // Print debug message that the SMRAM window is now locked.\r
   //\r
   DEBUG ((DEBUG_INFO, "SMM IPL locked SMRAM window\n"));\r
-  \r
+\r
   //\r
   // Set flag so this operation will not be performed again\r
   //\r
@@ -838,20 +874,18 @@ GetPeCoffImageFixLoadingAssignedAddress(
    // Build tool will calculate the smm code size and then patch the PcdLoadFixAddressSmmCodePageNumber\r
    //\r
    SmmCodeSize = EFI_PAGES_TO_SIZE (PcdGet32(PcdLoadFixAddressSmmCodePageNumber));\r
\r
+\r
    FixLoadingAddress = 0;\r
    Status = EFI_NOT_FOUND;\r
-   SmramBase = mCurrentSmramRange->CpuStart;\r
+   SmramBase = mLMFAConfigurationTable->SmramBase;\r
    //\r
    // Get PeHeader pointer\r
    //\r
    ImgHdr = (EFI_IMAGE_OPTIONAL_HEADER_UNION *)((CHAR8* )ImageContext->Handle + ImageContext->PeCoffHeaderOffset);\r
-   SectionHeaderOffset = (UINTN)(\r
-                                 ImageContext->PeCoffHeaderOffset +\r
-                                 sizeof (UINT32) +\r
-                                 sizeof (EFI_IMAGE_FILE_HEADER) +\r
-                                 ImgHdr->Pe32.FileHeader.SizeOfOptionalHeader\r
-                                 );\r
+   SectionHeaderOffset = ImageContext->PeCoffHeaderOffset +\r
+                         sizeof (UINT32) +\r
+                         sizeof (EFI_IMAGE_FILE_HEADER) +\r
+                         ImgHdr->Pe32.FileHeader.SizeOfOptionalHeader;\r
    NumberOfSections = ImgHdr->Pe32.FileHeader.NumberOfSections;\r
 \r
    //\r
@@ -871,9 +905,9 @@ GetPeCoffImageFixLoadingAssignedAddress(
      if (EFI_ERROR (Status)) {\r
        return Status;\r
      }\r
-     \r
+\r
      Status = EFI_NOT_FOUND;\r
-     \r
+\r
      if ((SectionHeader.Characteristics & EFI_IMAGE_SCN_CNT_CODE) == 0) {\r
        //\r
        // Build tool saves the offset to SMRAM base as image base in PointerToRelocations & PointerToLineNumbers fields in the\r
@@ -907,7 +941,7 @@ GetPeCoffImageFixLoadingAssignedAddress(
 /**\r
   Load the SMM Core image into SMRAM and executes the SMM Core from SMRAM.\r
 \r
-  @param[in, out] SmramRange            Descriptor for the range of SMRAM to reload the \r
+  @param[in, out] SmramRange            Descriptor for the range of SMRAM to reload the\r
                                         currently executing image, the rang of SMRAM to\r
                                         hold SMM Core will be excluded.\r
   @param[in, out] SmramRangeSmmCore     Descriptor for the range of SMRAM to hold SMM Core.\r
@@ -933,19 +967,19 @@ ExecuteSmmCoreFromSmram (
 \r
   //\r
   // Search all Firmware Volumes for a PE/COFF image in a file of type SMM_CORE\r
-  //  \r
+  //\r
   Status = GetSectionFromAnyFvByFileType (\r
-             EFI_FV_FILETYPE_SMM_CORE, \r
+             EFI_FV_FILETYPE_SMM_CORE,\r
              0,\r
-             EFI_SECTION_PE32, \r
+             EFI_SECTION_PE32,\r
              0,\r
-             &SourceBuffer, \r
+             &SourceBuffer,\r
              &SourceSize\r
              );\r
   if (EFI_ERROR (Status)) {\r
     return Status;\r
   }\r
-  \r
+\r
   //\r
   // Initilize ImageContext\r
   //\r
@@ -960,7 +994,7 @@ ExecuteSmmCoreFromSmram (
     return Status;\r
   }\r
   //\r
-  // if Loading module at Fixed Address feature is enabled, the SMM core driver will be loaded to \r
+  // if Loading module at Fixed Address feature is enabled, the SMM core driver will be loaded to\r
   // the address assigned by build tool.\r
   //\r
   if (PcdGet64(PcdLoadModuleAtFixAddressEnable) != 0) {\r
@@ -980,7 +1014,7 @@ ExecuteSmmCoreFromSmram (
     } else {\r
       DEBUG ((EFI_D_INFO, "LOADING MODULE FIXED ERROR: Loading module at fixed address at address failed\n"));\r
       //\r
-      // Allocate memory for the image being loaded from the EFI_SRAM_DESCRIPTOR \r
+      // Allocate memory for the image being loaded from the EFI_SRAM_DESCRIPTOR\r
       // specified by SmramRange\r
       //\r
       PageCount = (UINTN)EFI_SIZE_TO_PAGES((UINTN)ImageContext.ImageSize + ImageContext.SectionAlignment);\r
@@ -1001,7 +1035,7 @@ ExecuteSmmCoreFromSmram (
     }\r
   } else {\r
     //\r
-    // Allocate memory for the image being loaded from the EFI_SRAM_DESCRIPTOR \r
+    // Allocate memory for the image being loaded from the EFI_SRAM_DESCRIPTOR\r
     // specified by SmramRange\r
     //\r
     PageCount = (UINTN)EFI_SIZE_TO_PAGES((UINTN)ImageContext.ImageSize + ImageContext.SectionAlignment);\r
@@ -1020,9 +1054,9 @@ ExecuteSmmCoreFromSmram (
     //\r
     ImageContext.ImageAddress = SmramRangeSmmCore->CpuStart;\r
   }\r
-  \r
+\r
   ImageContext.ImageAddress += ImageContext.SectionAlignment - 1;\r
-  ImageContext.ImageAddress &= ~((EFI_PHYSICAL_ADDRESS)(ImageContext.SectionAlignment - 1));\r
+  ImageContext.ImageAddress &= ~((EFI_PHYSICAL_ADDRESS)ImageContext.SectionAlignment - 1);\r
 \r
   //\r
   // Print debug message showing SMM Core load address.\r
@@ -1498,10 +1532,10 @@ GetFullSmramRanges (
 /**\r
   The Entry Point for SMM IPL\r
 \r
-  Load SMM Core into SMRAM, register SMM Core entry point for SMIs, install \r
-  SMM Base 2 Protocol and SMM Communication Protocol, and register for the \r
+  Load SMM Core into SMRAM, register SMM Core entry point for SMIs, install\r
+  SMM Base 2 Protocol and SMM Communication Protocol, and register for the\r
   critical events required to coordinate between DXE and SMM environments.\r
-  \r
+\r
   @param  ImageHandle    The firmware allocated handle for the EFI image.\r
   @param  SystemTable    A pointer to the EFI System Table.\r
 \r
@@ -1521,14 +1555,14 @@ SmmIplEntry (
   UINT64                          MaxSize;\r
   VOID                            *Registration;\r
   UINT64                          SmmCodeSize;\r
-  EFI_LOAD_FIXED_ADDRESS_CONFIGURATION_TABLE    *LMFAConfigurationTable;\r
   EFI_CPU_ARCH_PROTOCOL           *CpuArch;\r
   EFI_STATUS                      SetAttrStatus;\r
   EFI_SMRAM_DESCRIPTOR            *SmramRangeSmmDriver;\r
+  EFI_GCD_MEMORY_SPACE_DESCRIPTOR MemDesc;\r
 \r
   //\r
-  // Fill in the image handle of the SMM IPL so the SMM Core can use this as the \r
-  // ParentImageHandle field of the Load Image Protocol for all SMM Drivers loaded \r
+  // Fill in the image handle of the SMM IPL so the SMM Core can use this as the\r
+  // ParentImageHandle field of the Load Image Protocol for all SMM Drivers loaded\r
   // by the SMM Core\r
   //\r
   mSmmCorePrivateData.SmmIplImageHandle = ImageHandle;\r
@@ -1557,7 +1591,7 @@ SmmIplEntry (
   // Print debug message that the SMRAM window is now open.\r
   //\r
   DEBUG ((DEBUG_INFO, "SMM IPL opened SMRAM window\n"));\r
-  \r
+\r
   //\r
   // Find the largest SMRAM range between 1MB and 4GB that is at least 256KB - 4K in size\r
   //\r
@@ -1584,28 +1618,55 @@ SmmIplEntry (
     //\r
     // Print debug message showing SMRAM window that will be used by SMM IPL and SMM Core\r
     //\r
-    DEBUG ((DEBUG_INFO, "SMM IPL found SMRAM window %p - %p\n", \r
-      (VOID *)(UINTN)mCurrentSmramRange->CpuStart, \r
+    DEBUG ((DEBUG_INFO, "SMM IPL found SMRAM window %p - %p\n",\r
+      (VOID *)(UINTN)mCurrentSmramRange->CpuStart,\r
       (VOID *)(UINTN)(mCurrentSmramRange->CpuStart + mCurrentSmramRange->PhysicalSize - 1)\r
       ));\r
 \r
     GetSmramCacheRange (mCurrentSmramRange, &mSmramCacheBase, &mSmramCacheSize);\r
     //\r
-    // If CPU AP is present, attempt to set SMRAM cacheability to WB\r
+    // Make sure we can change the desired memory attributes.\r
+    //\r
+    Status = gDS->GetMemorySpaceDescriptor (\r
+                    mSmramCacheBase,\r
+                    &MemDesc\r
+                    );\r
+    ASSERT_EFI_ERROR (Status);\r
+    if ((MemDesc.Capabilities & SMRAM_CAPABILITIES) != SMRAM_CAPABILITIES) {\r
+      gDS->SetMemorySpaceCapabilities (\r
+             mSmramCacheBase,\r
+             mSmramCacheSize,\r
+             MemDesc.Capabilities | SMRAM_CAPABILITIES\r
+             );\r
+    }\r
+    //\r
+    // If CPU AP is present, attempt to set SMRAM cacheability to WB and clear\r
+    // all paging attributes.\r
     // Note that it is expected that cacheability of SMRAM has been set to WB if CPU AP\r
     // is not available here.\r
     //\r
     CpuArch = NULL;\r
     Status = gBS->LocateProtocol (&gEfiCpuArchProtocolGuid, NULL, (VOID **)&CpuArch);\r
     if (!EFI_ERROR (Status)) {\r
-      Status = gDS->SetMemorySpaceAttributes(\r
-                      mSmramCacheBase, \r
+      MemDesc.Attributes &= ~(MEMORY_CACHE_ATTRIBUTES | MEMORY_PAGE_ATTRIBUTES);\r
+      MemDesc.Attributes |= EFI_MEMORY_WB;\r
+      Status = gDS->SetMemorySpaceAttributes (\r
+                      mSmramCacheBase,\r
                       mSmramCacheSize,\r
-                      EFI_MEMORY_WB\r
+                      MemDesc.Attributes\r
                       );\r
       if (EFI_ERROR (Status)) {\r
         DEBUG ((DEBUG_WARN, "SMM IPL failed to set SMRAM window to EFI_MEMORY_WB\n"));\r
-      }  \r
+      }\r
+\r
+      DEBUG_CODE (\r
+        gDS->GetMemorySpaceDescriptor (\r
+               mSmramCacheBase,\r
+               &MemDesc\r
+               );\r
+        DEBUG ((DEBUG_INFO, "SMRAM attributes: %016lx\n", MemDesc.Attributes));\r
+        ASSERT ((MemDesc.Attributes & MEMORY_PAGE_ATTRIBUTES) == 0);\r
+      );\r
     }\r
     //\r
     // if Loading module at Fixed Address feature is enabled, save the SMRAM base to Load\r
@@ -1625,14 +1686,14 @@ SmmIplEntry (
       //\r
       Status = EfiGetSystemConfigurationTable (\r
                 &gLoadFixedAddressConfigurationTableGuid,\r
-               (VOID **) &LMFAConfigurationTable\r
+               (VOID **) &mLMFAConfigurationTable\r
                );\r
-      if (!EFI_ERROR (Status) && LMFAConfigurationTable != NULL) {\r
-        LMFAConfigurationTable->SmramBase = mCurrentSmramRange->CpuStart;\r
+      if (!EFI_ERROR (Status) && mLMFAConfigurationTable != NULL) {\r
+        mLMFAConfigurationTable->SmramBase = mCurrentSmramRange->CpuStart;\r
         //\r
         // Print the SMRAM base\r
         //\r
-        DEBUG ((EFI_D_INFO, "LOADING MODULE FIXED INFO: TSEG BASE is %x. \n", LMFAConfigurationTable->SmramBase));\r
+        DEBUG ((EFI_D_INFO, "LOADING MODULE FIXED INFO: TSEG BASE is %x. \n", mLMFAConfigurationTable->SmramBase));\r
       }\r
 \r
       //\r
@@ -1667,13 +1728,13 @@ SmmIplEntry (
       //\r
       if (CpuArch != NULL) {\r
         SetAttrStatus = gDS->SetMemorySpaceAttributes(\r
-                               mSmramCacheBase, \r
+                               mSmramCacheBase,\r
                                mSmramCacheSize,\r
                                EFI_MEMORY_UC\r
                                );\r
         if (EFI_ERROR (SetAttrStatus)) {\r
           DEBUG ((DEBUG_WARN, "SMM IPL failed to reset SMRAM window to EFI_MEMORY_UC\n"));\r
-        }  \r
+        }\r
       }\r
     }\r
   } else {\r
@@ -1684,7 +1745,7 @@ SmmIplEntry (
   }\r
 \r
   //\r
-  // If the SMM Core could not be loaded then close SMRAM window, free allocated \r
+  // If the SMM Core could not be loaded then close SMRAM window, free allocated\r
   // resources, and return an error so SMM IPL will be unloaded.\r
   //\r
   if (mCurrentSmramRange == NULL || EFI_ERROR (Status)) {\r
@@ -1706,7 +1767,7 @@ SmmIplEntry (
 \r
     return EFI_UNSUPPORTED;\r
   }\r
-  \r
+\r
   //\r
   // Install SMM Base2 Protocol and SMM Communication Protocol\r
   //\r