]> 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 31d2c9e45e1f56e7483892ab5e246a3f3e9b03cb..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
@@ -271,19 +289,19 @@ EFI_LOAD_FIXED_ADDRESS_CONFIGURATION_TABLE    *mLMFAConfigurationTable = NULL;
 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
@@ -303,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
@@ -320,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
@@ -332,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
@@ -370,7 +388,7 @@ GetSmramCacheRange (
       }\r
     }\r
   } while (FoundAjacentRange);\r
-  \r
+\r
 }\r
 \r
 /**\r
@@ -421,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
@@ -433,10 +451,10 @@ 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
@@ -510,7 +528,7 @@ SmmCommunicationCommunicate (
     }\r
 \r
     //\r
-    // Return status from software SMI \r
+    // Return status from software SMI\r
     //\r
     if (CommSize != NULL) {\r
       *CommSize = gSmmCorePrivate->BufferSize;\r
@@ -534,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
@@ -546,9 +564,9 @@ SmmCommunicationCommunicate (
   //\r
   TempCommSize -= OFFSET_OF (EFI_SMM_COMMUNICATE_HEADER, Data);\r
   Status = gSmmCorePrivate->Smst->SmiManage (\r
-                                    &CommunicateHeader->HeaderGuid, \r
-                                    NULL, \r
-                                    CommunicateHeader->Data, \r
+                                    &CommunicateHeader->HeaderGuid,\r
+                                    NULL,\r
+                                    CommunicateHeader->Data,\r
                                     &TempCommSize\r
                                     );\r
   TempCommSize += OFFSET_OF (EFI_SMM_COMMUNICATE_HEADER, Data);\r
@@ -581,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
@@ -646,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
@@ -712,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
@@ -748,7 +766,7 @@ SmmIplReadyToLockEventNotify (
   if (mSmmLocked) {\r
     return;\r
   }\r
-  \r
+\r
   //\r
   // Make sure this notification is for this handler\r
   //\r
@@ -759,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
@@ -779,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
@@ -800,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
@@ -856,7 +874,7 @@ 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 = mLMFAConfigurationTable->SmramBase;\r
@@ -887,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
@@ -923,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
@@ -949,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
@@ -976,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
@@ -996,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
@@ -1017,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
@@ -1036,7 +1054,7 @@ 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
 \r
@@ -1514,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
@@ -1540,10 +1558,11 @@ SmmIplEntry (
   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
@@ -1572,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
@@ -1599,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
@@ -1682,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
@@ -1699,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
@@ -1721,7 +1767,7 @@ SmmIplEntry (
 \r
     return EFI_UNSUPPORTED;\r
   }\r
-  \r
+\r
   //\r
   // Install SMM Base2 Protocol and SMM Communication Protocol\r
   //\r