]> git.proxmox.com Git - mirror_edk2.git/blobdiff - MdeModulePkg/Core/Pei/Dispatcher/Dispatcher.c
MdeModulePkg/PeiCore: Remove MigrateSecModulesInFv()
[mirror_edk2.git] / MdeModulePkg / Core / Pei / Dispatcher / Dispatcher.c
index 68670f43e0b3f97a0e6e7c2ae1152e1e3f882a16..61d4b67c64b9ae3298f74c3ac7638ef22d8c1959 100644 (file)
@@ -1,15 +1,9 @@
 /** @file\r
   EFI PEI Core dispatch services\r
 \r
-Copyright (c) 2006 - 2018, Intel Corporation. All rights reserved.<BR>\r
+Copyright (c) 2006 - 2019, Intel Corporation. All rights reserved.<BR>\r
 (C) Copyright 2016 Hewlett Packard Enterprise Development LP<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
-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
+SPDX-License-Identifier: BSD-2-Clause-Patent\r
 \r
 **/\r
 \r
@@ -17,7 +11,7 @@ WITHOUT WARRANTIES OR REPRESENTATIONS OF ANY KIND, EITHER EXPRESS OR IMPLIED.
 \r
 /**\r
 \r
-  Discover all Peims and optional Apriori file in one FV. There is at most one\r
+  Discover all PEIMs and optional Apriori file in one FV. There is at most one\r
   Apriori file in one FV.\r
 \r
 \r
@@ -55,7 +49,7 @@ DiscoverPeimsAndOrderWithApriori (
   Guid = NULL;\r
 \r
   //\r
-  // If the current Fv has been scanned, directly get its cached records.\r
+  // If the current FV has been scanned, directly get its cached records.\r
   //\r
   if (CoreFileHandle->ScanFv) {\r
     Private->CurrentFvFileHandles = CoreFileHandle->FvFileHandles;\r
@@ -66,7 +60,7 @@ DiscoverPeimsAndOrderWithApriori (
   TempFileGuid    = Private->TempFileGuid;\r
 \r
   //\r
-  // Go ahead to scan this Fv, get PeimCount and cache FileHandles within it to TempFileHandles.\r
+  // Go ahead to scan this FV, get PeimCount and cache FileHandles within it to TempFileHandles.\r
   //\r
   PeimCount = 0;\r
   FileHandle = NULL;\r
@@ -153,7 +147,7 @@ DiscoverPeimsAndOrderWithApriori (
 \r
       for (Index = 0; Index < PeimCount; Index++) {\r
         //\r
-        // Make an array of file name guids that matches the FileHandle array so we can convert\r
+        // Make an array of file name GUIDs that matches the FileHandle array so we can convert\r
         // quickly from file name to file handle\r
         //\r
         Status = FvPpi->GetFileInfo (FvPpi, TempFileHandles[Index], &FileInfo);\r
@@ -162,7 +156,7 @@ DiscoverPeimsAndOrderWithApriori (
       }\r
 \r
       //\r
-      // Walk through TempFileGuid array to find out who is invalid PEIM guid in Apriori file.\r
+      // Walk through TempFileGuid array to find out who is invalid PEIM GUID in Apriori file.\r
       // Add available PEIMs in Apriori file into FvFileHandles array.\r
       //\r
       Index = 0;\r
@@ -200,8 +194,8 @@ DiscoverPeimsAndOrderWithApriori (
   }\r
 \r
   //\r
-  // The current Fv File Handles have been cached. So that we don't have to scan the Fv again.\r
-  // Instead, we can retrieve the file handles within this Fv from cached records.\r
+  // The current FV File Handles have been cached. So that we don't have to scan the FV again.\r
+  // Instead, we can retrieve the file handles within this FV from cached records.\r
   //\r
   CoreFileHandle->ScanFv = TRUE;\r
   Private->CurrentFvFileHandles = CoreFileHandle->FvFileHandles;\r
@@ -263,7 +257,7 @@ PeiLoadFixAddressIsMemoryRangeAvailable (
 \r
   This function should only be invoked when Loading Module at Fixed Address(LMFA) feature is enabled. When feature is\r
   configured as Load Modules at Fix Absolute Address, this function is to validate the top address assigned by user. When\r
-  feature is configured as Load Modules at Fixed Offset, the functino is to find the top address which is TOLM-TSEG in general.\r
+  feature is configured as Load Modules at Fixed Offset, the function is to find the top address which is TOLM-TSEG in general.\r
   And also the function will re-install PEI memory.\r
 \r
   @param PrivateData         Pointer to the private data passed in from caller\r
@@ -313,7 +307,7 @@ PeiLoadFixAddressHook(
   DEBUG ((EFI_D_INFO, "LOADING MODULE FIXED INFO: PcdLoadFixAddressPeiCodePageNumber= 0x%x.\n", PcdGet32(PcdLoadFixAddressPeiCodePageNumber)));\r
   DEBUG ((EFI_D_INFO, "LOADING MODULE FIXED INFO: Total Reserved Memory Size = 0x%lx.\n", TotalReservedMemorySize));\r
   //\r
-  // Loop through the system memory typed hob to merge the adjacent memory range\r
+  // Loop through the system memory typed HOB to merge the adjacent memory range\r
   //\r
   for (Hob.Raw = PrivateData->HobList.Raw; !END_OF_HOB_LIST(Hob); Hob.Raw = GET_NEXT_HOB(Hob)) {\r
     //\r
@@ -323,7 +317,7 @@ PeiLoadFixAddressHook(
 \r
       ResourceHob = Hob.ResourceDescriptor;\r
       //\r
-      // If range described in this hob is not system memory or heigher than MAX_ADDRESS, ignored.\r
+      // If range described in this HOB is not system memory or higher than MAX_ADDRESS, ignored.\r
       //\r
       if (ResourceHob->ResourceType != EFI_RESOURCE_SYSTEM_MEMORY ||\r
           ResourceHob->PhysicalStart + ResourceHob->ResourceLength > MAX_ADDRESS)   {\r
@@ -393,18 +387,18 @@ PeiLoadFixAddressHook(
         if (GET_HOB_TYPE (NextHob) == EFI_HOB_TYPE_RESOURCE_DESCRIPTOR) {\r
           NextResourceHob = NextHob.ResourceDescriptor;\r
           //\r
-          // If range described in this hob is not system memory or heigher than MAX_ADDRESS, ignored.\r
+          // If range described in this HOB is not system memory or higher than MAX_ADDRESS, ignored.\r
           //\r
           if (NextResourceHob->ResourceType != EFI_RESOURCE_SYSTEM_MEMORY || NextResourceHob->PhysicalStart + NextResourceHob->ResourceLength > MAX_ADDRESS) {\r
             continue;\r
           }\r
           //\r
-          // If the range describe in memory allocation HOB  belongs to the memroy range described by the resource hob\r
+          // If the range describe in memory allocation HOB belongs to the memory range described by the resource HOB\r
           //\r
           if (MemoryHob->AllocDescriptor.MemoryBaseAddress >= NextResourceHob->PhysicalStart &&\r
               MemoryHob->AllocDescriptor.MemoryBaseAddress + MemoryHob->AllocDescriptor.MemoryLength <= NextResourceHob->PhysicalStart + NextResourceHob->ResourceLength) {\r
              //\r
-             // Build seperate resource hob for this allocated range\r
+             // Build separate resource HOB for this allocated range\r
              //\r
              if (MemoryHob->AllocDescriptor.MemoryBaseAddress > NextResourceHob->PhysicalStart) {\r
                BuildResourceDescriptorHob (\r
@@ -458,7 +452,7 @@ PeiLoadFixAddressHook(
 \r
         ResourceHob = Hob.ResourceDescriptor;\r
         //\r
-        // See if this resource descrior HOB describes tested system memory below MAX_ADDRESS\r
+        // See if this resource descriptor HOB describes tested system memory below MAX_ADDRESS\r
         //\r
         if (ResourceHob->ResourceType == EFI_RESOURCE_SYSTEM_MEMORY &&\r
             ResourceHob->PhysicalStart + ResourceHob->ResourceLength <= MAX_ADDRESS) {\r
@@ -482,7 +476,7 @@ PeiLoadFixAddressHook(
       DEBUG ((EFI_D_INFO, "LOADING MODULE FIXED ERROR:Top Address 0x%lx is invalid \n",  TopLoadingAddress));\r
       DEBUG ((EFI_D_INFO, "LOADING MODULE FIXED ERROR:The recommended Top Address for the platform is: \n"));\r
       //\r
-      // Print the recomended Top address range.\r
+      // Print the recommended Top address range.\r
       //\r
       for (Hob.Raw = PrivateData->HobList.Raw; !END_OF_HOB_LIST(Hob); Hob.Raw = GET_NEXT_HOB(Hob)) {\r
         //\r
@@ -492,7 +486,7 @@ PeiLoadFixAddressHook(
 \r
           ResourceHob = Hob.ResourceDescriptor;\r
           //\r
-          // See if this resource descrior HOB describes tested system memory below MAX_ADDRESS\r
+          // See if this resource descriptor HOB describes tested system memory below MAX_ADDRESS\r
           //\r
           if (ResourceHob->ResourceType == EFI_RESOURCE_SYSTEM_MEMORY &&\r
               ResourceHob->PhysicalStart + ResourceHob->ResourceLength <= MAX_ADDRESS) {\r
@@ -530,7 +524,7 @@ PeiLoadFixAddressHook(
 \r
         ResourceHob = Hob.ResourceDescriptor;\r
         //\r
-        // See if this resource descrior HOB describes tested system memory below MAX_ADDRESS\r
+        // See if this resource descriptor HOB describes tested system memory below MAX_ADDRESS\r
         //\r
         if (ResourceHob->ResourceType == EFI_RESOURCE_SYSTEM_MEMORY &&\r
             ResourceHob->PhysicalStart + ResourceHob->ResourceLength <= MAX_ADDRESS &&\r
@@ -560,7 +554,7 @@ PeiLoadFixAddressHook(
 \r
   if (CurrentResourceHob != NULL) {\r
     //\r
-    // rebuild resource HOB for PEI memmory and reserved memory\r
+    // rebuild resource HOB for PEI memory and reserved memory\r
     //\r
     BuildResourceDescriptorHob (\r
       EFI_RESOURCE_SYSTEM_MEMORY,\r
@@ -610,7 +604,7 @@ PeiLoadFixAddressHook(
        );\r
     }\r
     //\r
-    // Delete CurrentHob by marking it as unused since the the memory range described by is rebuilt.\r
+    // Delete CurrentHob by marking it as unused since the memory range described by is rebuilt.\r
     //\r
     GET_HOB_TYPE (CurrentHob) = EFI_HOB_TYPE_UNUSED;\r
   }\r
@@ -756,8 +750,8 @@ PeiCheckAndSwitchStack (
     ASSERT (NewStackSize >= SecCoreData->StackSize);\r
 \r
     //\r
-    // Calculate stack offset and heap offset between temporary memory and new permement\r
-    // memory seperately.\r
+    // Calculate stack offset and heap offset between temporary memory and new permanent\r
+    // memory separately.\r
     //\r
     TopOfOldStack = (UINTN)SecCoreData->StackBase + SecCoreData->StackSize;\r
     TopOfNewStack = Private->PhysicalMemoryBegin + NewStackSize;\r
@@ -958,6 +952,341 @@ PeiCheckAndSwitchStack (
   }\r
 }\r
 \r
+/**\r
+  Migrate a PEIM from temporary RAM to permanent memory.\r
+\r
+  @param PeimFileHandle       Pointer to the FFS file header of the image.\r
+  @param MigratedFileHandle   Pointer to the FFS file header of the migrated image.\r
+\r
+  @retval EFI_SUCCESS         Sucessfully migrated the PEIM to permanent memory.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+MigratePeim (\r
+  IN  EFI_PEI_FILE_HANDLE     FileHandle,\r
+  IN  EFI_PEI_FILE_HANDLE     MigratedFileHandle\r
+  )\r
+{\r
+  EFI_STATUS                Status;\r
+  EFI_FFS_FILE_HEADER       *FileHeader;\r
+  VOID                      *Pe32Data;\r
+  VOID                      *ImageAddress;\r
+  CHAR8                     *AsciiString;\r
+  UINTN                     Index;\r
+\r
+  Status = EFI_SUCCESS;\r
+\r
+  FileHeader = (EFI_FFS_FILE_HEADER *) FileHandle;\r
+  ASSERT (!IS_FFS_FILE2 (FileHeader));\r
+\r
+  ImageAddress = NULL;\r
+  PeiGetPe32Data (MigratedFileHandle, &ImageAddress);\r
+  if (ImageAddress != NULL) {\r
+    DEBUG_CODE_BEGIN ();\r
+    AsciiString = PeCoffLoaderGetPdbPointer (ImageAddress);\r
+    for (Index = 0; AsciiString[Index] != 0; Index++) {\r
+      if (AsciiString[Index] == '\\' || AsciiString[Index] == '/') {\r
+        AsciiString = AsciiString + Index + 1;\r
+        Index = 0;\r
+      } else if (AsciiString[Index] == '.') {\r
+        AsciiString[Index] = 0;\r
+      }\r
+    }\r
+    DEBUG ((DEBUG_INFO, "%a", AsciiString));\r
+    DEBUG_CODE_END ();\r
+\r
+    Pe32Data = (VOID *) ((UINTN) ImageAddress - (UINTN) MigratedFileHandle + (UINTN) FileHandle);\r
+    Status = LoadAndRelocatePeCoffImageInPlace (Pe32Data, ImageAddress);\r
+    ASSERT_EFI_ERROR (Status);\r
+  }\r
+\r
+  return Status;\r
+}\r
+\r
+/**\r
+  Migrate Status Code Callback function pointers inside an FV from temporary memory to permanent memory.\r
+\r
+  @param OrgFvHandle      Address of FV handle in temporary memory.\r
+  @param FvHandle         Address of FV handle in permanent memory.\r
+  @param FvSize           Size of the FV.\r
+\r
+**/\r
+VOID\r
+ConvertStatusCodeCallbacks (\r
+  IN  UINTN                   OrgFvHandle,\r
+  IN  UINTN                   FvHandle,\r
+  IN  UINTN                   FvSize\r
+  )\r
+{\r
+  EFI_PEI_HOB_POINTERS    Hob;\r
+  UINTN                   *NumberOfEntries;\r
+  UINTN                   *CallbackEntry;\r
+  UINTN                   Index;\r
+\r
+  Hob.Raw  = GetFirstGuidHob (&gStatusCodeCallbackGuid);\r
+  while (Hob.Raw != NULL) {\r
+    NumberOfEntries = GET_GUID_HOB_DATA (Hob);\r
+    CallbackEntry   = NumberOfEntries + 1;\r
+    for (Index = 0; Index < *NumberOfEntries; Index++) {\r
+      if (((VOID *) CallbackEntry[Index]) != NULL) {\r
+        if ((CallbackEntry[Index] >= OrgFvHandle) && (CallbackEntry[Index] < (OrgFvHandle + FvSize))) {\r
+          DEBUG ((\r
+            DEBUG_INFO,\r
+            "Migrating CallbackEntry[%Lu] from 0x%0*Lx to ",\r
+            (UINT64)Index,\r
+            (sizeof CallbackEntry[Index]) * 2,\r
+            (UINT64)CallbackEntry[Index]\r
+            ));\r
+          if (OrgFvHandle > FvHandle) {\r
+            CallbackEntry[Index] = CallbackEntry[Index] - (OrgFvHandle - FvHandle);\r
+          } else {\r
+            CallbackEntry[Index] = CallbackEntry[Index] + (FvHandle - OrgFvHandle);\r
+          }\r
+          DEBUG ((\r
+            DEBUG_INFO,\r
+            "0x%0*Lx\n",\r
+            (sizeof CallbackEntry[Index]) * 2,\r
+            (UINT64)CallbackEntry[Index]\r
+            ));\r
+        }\r
+      }\r
+    }\r
+    Hob.Raw = GET_NEXT_HOB (Hob);\r
+    Hob.Raw = GetNextGuidHob (&gStatusCodeCallbackGuid, Hob.Raw);\r
+  }\r
+}\r
+\r
+/**\r
+  Migrates PEIMs in the given firmware volume.\r
+\r
+  @param Private          Pointer to the PeiCore's private data structure.\r
+  @param FvIndex          The firmware volume index to migrate.\r
+  @param OrgFvHandle      The handle to the firmware volume in temporary memory.\r
+  @param FvHandle         The handle to the firmware volume in permanent memory.\r
+\r
+  @retval   EFI_SUCCESS           The PEIMs in the FV were migrated successfully\r
+  @retval   EFI_INVALID_PARAMETER The Private pointer is NULL or FvCount is invalid.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+MigratePeimsInFv (\r
+  IN PEI_CORE_INSTANCE    *Private,\r
+  IN  UINTN               FvIndex,\r
+  IN  UINTN               OrgFvHandle,\r
+  IN  UINTN               FvHandle\r
+  )\r
+{\r
+  EFI_STATUS              Status;\r
+  volatile UINTN          FileIndex;\r
+  EFI_PEI_FILE_HANDLE     MigratedFileHandle;\r
+  EFI_PEI_FILE_HANDLE     FileHandle;\r
+\r
+  if (Private == NULL || FvIndex >= Private->FvCount) {\r
+    return EFI_INVALID_PARAMETER;\r
+  }\r
+\r
+  if (Private->Fv[FvIndex].ScanFv) {\r
+    for (FileIndex = 0; FileIndex < Private->Fv[FvIndex].PeimCount; FileIndex++) {\r
+      if (Private->Fv[FvIndex].FvFileHandles[FileIndex] != NULL) {\r
+        FileHandle = Private->Fv[FvIndex].FvFileHandles[FileIndex];\r
+\r
+        MigratedFileHandle = (EFI_PEI_FILE_HANDLE) ((UINTN) FileHandle - OrgFvHandle + FvHandle);\r
+\r
+        DEBUG ((DEBUG_VERBOSE, "    Migrating FileHandle %2d ", FileIndex));\r
+        Status = MigratePeim (FileHandle, MigratedFileHandle);\r
+        DEBUG ((DEBUG_VERBOSE, "\n"));\r
+        ASSERT_EFI_ERROR (Status);\r
+\r
+        if (!EFI_ERROR (Status)) {\r
+          Private->Fv[FvIndex].FvFileHandles[FileIndex] = MigratedFileHandle;\r
+          if (FvIndex == Private->CurrentPeimFvCount) {\r
+            Private->CurrentFvFileHandles[FileIndex] = MigratedFileHandle;\r
+          }\r
+        }\r
+      }\r
+    }\r
+  }\r
+\r
+  return EFI_SUCCESS;\r
+}\r
+\r
+/**\r
+  Migrate FVs out of temporary RAM before the cache is flushed.\r
+\r
+  @param Private         PeiCore's private data structure\r
+  @param SecCoreData     Points to a data structure containing information about the PEI core's operating\r
+                         environment, such as the size and location of temporary RAM, the stack location and\r
+                         the BFV location.\r
+\r
+  @retval EFI_SUCCESS           Succesfully migrated installed FVs from temporary RAM to permanent memory.\r
+  @retval EFI_OUT_OF_RESOURCES  Insufficient memory exists to allocate needed pages.\r
+\r
+**/\r
+EFI_STATUS\r
+EFIAPI\r
+EvacuateTempRam (\r
+  IN PEI_CORE_INSTANCE            *Private,\r
+  IN CONST EFI_SEC_PEI_HAND_OFF   *SecCoreData\r
+  )\r
+{\r
+  EFI_STATUS                    Status;\r
+  volatile UINTN                FvIndex;\r
+  volatile UINTN                FvChildIndex;\r
+  UINTN                         ChildFvOffset;\r
+  EFI_FIRMWARE_VOLUME_HEADER    *FvHeader;\r
+  EFI_FIRMWARE_VOLUME_HEADER    *ChildFvHeader;\r
+  EFI_FIRMWARE_VOLUME_HEADER    *MigratedFvHeader;\r
+  EFI_FIRMWARE_VOLUME_HEADER    *RawDataFvHeader;\r
+  EFI_FIRMWARE_VOLUME_HEADER    *MigratedChildFvHeader;\r
+\r
+  PEI_CORE_FV_HANDLE            PeiCoreFvHandle;\r
+  EFI_PEI_CORE_FV_LOCATION_PPI  *PeiCoreFvLocationPpi;\r
+  EDKII_MIGRATED_FV_INFO        MigratedFvInfo;\r
+\r
+  ASSERT (Private->PeiMemoryInstalled);\r
+\r
+  DEBUG ((DEBUG_VERBOSE, "Beginning evacuation of content in temporary RAM.\n"));\r
+\r
+  //\r
+  // Migrate PPI Pointers of PEI_CORE from temporary memory to newly loaded PEI_CORE in permanent memory.\r
+  //\r
+  Status = PeiLocatePpi ((CONST EFI_PEI_SERVICES **) &Private->Ps, &gEfiPeiCoreFvLocationPpiGuid, 0, NULL, (VOID **) &PeiCoreFvLocationPpi);\r
+  if (!EFI_ERROR (Status) && (PeiCoreFvLocationPpi->PeiCoreFvLocation != NULL)) {\r
+    PeiCoreFvHandle.FvHandle = (EFI_PEI_FV_HANDLE) PeiCoreFvLocationPpi->PeiCoreFvLocation;\r
+  } else {\r
+    PeiCoreFvHandle.FvHandle = (EFI_PEI_FV_HANDLE) SecCoreData->BootFirmwareVolumeBase;\r
+  }\r
+  for (FvIndex = 0; FvIndex < Private->FvCount; FvIndex++) {\r
+    if (Private->Fv[FvIndex].FvHandle == PeiCoreFvHandle.FvHandle) {\r
+      CopyMem (&PeiCoreFvHandle, &Private->Fv[FvIndex], sizeof (PEI_CORE_FV_HANDLE));\r
+      break;\r
+    }\r
+  }\r
+  Status = EFI_SUCCESS;\r
+\r
+  ConvertPeiCorePpiPointers (Private, &PeiCoreFvHandle);\r
+\r
+  for (FvIndex = 0; FvIndex < Private->FvCount; FvIndex++) {\r
+    FvHeader = Private->Fv[FvIndex].FvHeader;\r
+    ASSERT (FvHeader != NULL);\r
+    ASSERT (FvIndex < Private->FvCount);\r
+\r
+    DEBUG ((DEBUG_VERBOSE, "FV[%02d] at 0x%x.\n", FvIndex, (UINTN) FvHeader));\r
+    if (\r
+      !(\r
+        ((EFI_PHYSICAL_ADDRESS)(UINTN) FvHeader >= Private->PhysicalMemoryBegin) &&\r
+        (((EFI_PHYSICAL_ADDRESS)(UINTN) FvHeader + (FvHeader->FvLength - 1)) < Private->FreePhysicalMemoryTop)\r
+        )\r
+      ) {\r
+      //\r
+      // Allocate page to save the rebased PEIMs, the PEIMs will get dispatched later.\r
+      //\r
+      Status =  PeiServicesAllocatePages (\r
+                  EfiBootServicesCode,\r
+                  EFI_SIZE_TO_PAGES ((UINTN) FvHeader->FvLength),\r
+                  (EFI_PHYSICAL_ADDRESS *) &MigratedFvHeader\r
+                  );\r
+      ASSERT_EFI_ERROR (Status);\r
+\r
+      //\r
+      // Allocate pool to save the raw PEIMs, which is used to keep consistent context across\r
+      // multiple boot and PCR0 will keep the same no matter if the address of allocated page is changed.\r
+      //\r
+      Status =  PeiServicesAllocatePages (\r
+                  EfiBootServicesCode,\r
+                  EFI_SIZE_TO_PAGES ((UINTN) FvHeader->FvLength),\r
+                  (EFI_PHYSICAL_ADDRESS *) &RawDataFvHeader\r
+                  );\r
+      ASSERT_EFI_ERROR (Status);\r
+\r
+      DEBUG ((\r
+        DEBUG_VERBOSE,\r
+        "  Migrating FV[%d] from 0x%08X to 0x%08X\n",\r
+        FvIndex,\r
+        (UINTN) FvHeader,\r
+        (UINTN) MigratedFvHeader\r
+        ));\r
+\r
+      //\r
+      // Copy the context to the rebased pages and raw pages, and create hob to save the\r
+      // information. The MigratedFvInfo HOB will never be produced when\r
+      // PcdMigrateTemporaryRamFirmwareVolumes is FALSE, because the PCD control the\r
+      // feature.\r
+      //\r
+      CopyMem (MigratedFvHeader, FvHeader, (UINTN) FvHeader->FvLength);\r
+      CopyMem (RawDataFvHeader, MigratedFvHeader, (UINTN) FvHeader->FvLength);\r
+      MigratedFvInfo.FvOrgBase  = (UINT32) (UINTN) FvHeader;\r
+      MigratedFvInfo.FvNewBase  = (UINT32) (UINTN) MigratedFvHeader;\r
+      MigratedFvInfo.FvDataBase = (UINT32) (UINTN) RawDataFvHeader;\r
+      MigratedFvInfo.FvLength   = (UINT32) (UINTN) FvHeader->FvLength;\r
+      BuildGuidDataHob (&gEdkiiMigratedFvInfoGuid, &MigratedFvInfo, sizeof (MigratedFvInfo));\r
+\r
+      //\r
+      // Migrate any children for this FV now\r
+      //\r
+      for (FvChildIndex = FvIndex; FvChildIndex < Private->FvCount; FvChildIndex++) {\r
+        ChildFvHeader = Private->Fv[FvChildIndex].FvHeader;\r
+        if (\r
+          ((UINTN) ChildFvHeader > (UINTN) FvHeader) &&\r
+          (((UINTN) ChildFvHeader + ChildFvHeader->FvLength) < ((UINTN) FvHeader) + FvHeader->FvLength)\r
+          ) {\r
+          DEBUG ((DEBUG_VERBOSE, "    Child FV[%02d] is being migrated.\n", FvChildIndex));\r
+          ChildFvOffset = (UINTN) ChildFvHeader - (UINTN) FvHeader;\r
+          DEBUG ((DEBUG_VERBOSE, "    Child FV offset = 0x%x.\n", ChildFvOffset));\r
+          MigratedChildFvHeader = (EFI_FIRMWARE_VOLUME_HEADER *) ((UINTN) MigratedFvHeader + ChildFvOffset);\r
+          Private->Fv[FvChildIndex].FvHeader = MigratedChildFvHeader;\r
+          Private->Fv[FvChildIndex].FvHandle = (EFI_PEI_FV_HANDLE) MigratedChildFvHeader;\r
+          DEBUG ((DEBUG_VERBOSE, "    Child migrated FV header at 0x%x.\n", (UINTN) MigratedChildFvHeader));\r
+\r
+          Status =  MigratePeimsInFv (Private, FvChildIndex, (UINTN) ChildFvHeader, (UINTN) MigratedChildFvHeader);\r
+          ASSERT_EFI_ERROR (Status);\r
+\r
+          ConvertPpiPointersFv (\r
+            Private,\r
+            (UINTN) ChildFvHeader,\r
+            (UINTN) MigratedChildFvHeader,\r
+            (UINTN) ChildFvHeader->FvLength - 1\r
+            );\r
+\r
+          ConvertStatusCodeCallbacks (\r
+            (UINTN) ChildFvHeader,\r
+            (UINTN) MigratedChildFvHeader,\r
+            (UINTN) ChildFvHeader->FvLength - 1\r
+            );\r
+\r
+          ConvertFvHob (Private, (UINTN) ChildFvHeader, (UINTN) MigratedChildFvHeader);\r
+        }\r
+      }\r
+      Private->Fv[FvIndex].FvHeader = MigratedFvHeader;\r
+      Private->Fv[FvIndex].FvHandle = (EFI_PEI_FV_HANDLE) MigratedFvHeader;\r
+\r
+      Status = MigratePeimsInFv (Private, FvIndex, (UINTN) FvHeader, (UINTN) MigratedFvHeader);\r
+      ASSERT_EFI_ERROR (Status);\r
+\r
+      ConvertPpiPointersFv (\r
+        Private,\r
+        (UINTN) FvHeader,\r
+        (UINTN) MigratedFvHeader,\r
+        (UINTN) FvHeader->FvLength - 1\r
+        );\r
+\r
+      ConvertStatusCodeCallbacks (\r
+        (UINTN) FvHeader,\r
+        (UINTN) MigratedFvHeader,\r
+        (UINTN) FvHeader->FvLength - 1\r
+        );\r
+\r
+      ConvertFvHob (Private, (UINTN) FvHeader, (UINTN) MigratedFvHeader);\r
+    }\r
+  }\r
+\r
+  RemoveFvHobsInTemporaryMemory (Private);\r
+\r
+  return Status;\r
+}\r
+\r
 /**\r
   Conduct PEIM dispatch.\r
 \r
@@ -994,7 +1323,11 @@ PeiDispatcher (
   PeimFileHandle = NULL;\r
   EntryPoint     = 0;\r
 \r
-  if ((Private->PeiMemoryInstalled) && (Private->HobList.HandoffInformationTable->BootMode != BOOT_ON_S3_RESUME || PcdGetBool (PcdShadowPeimOnS3Boot))) {\r
+  if ((Private->PeiMemoryInstalled) &&\r
+      (PcdGetBool (PcdMigrateTemporaryRamFirmwareVolumes) ||\r
+       (Private->HobList.HandoffInformationTable->BootMode != BOOT_ON_S3_RESUME) ||\r
+       PcdGetBool (PcdShadowPeimOnS3Boot))\r
+    ) {\r
     //\r
     // Once real memory is available, shadow the RegisterForShadow modules. And meanwhile\r
     // update the modules' status from PEIM_STATE_REGISTER_FOR_SHADOW to PEIM_STATE_DONE.\r
@@ -1048,10 +1381,10 @@ PeiDispatcher (
   //\r
   // This is the main dispatch loop.  It will search known FVs for PEIMs and\r
   // attempt to dispatch them.  If any PEIM gets dispatched through a single\r
-  // pass of the dispatcher, it will start over from the Bfv again to see\r
+  // pass of the dispatcher, it will start over from the BFV again to see\r
   // if any new PEIMs dependencies got satisfied.  With a well ordered\r
   // FV where PEIMs are found in the order their dependencies are also\r
-  // satisfied, this dipatcher should run only once.\r
+  // satisfied, this dispatcher should run only once.\r
   //\r
   do {\r
     //\r
@@ -1087,7 +1420,7 @@ PeiDispatcher (
       }\r
 \r
       //\r
-      // Start to dispatch all modules within the current Fv.\r
+      // Start to dispatch all modules within the current FV.\r
       //\r
       for (PeimCount = Private->CurrentPeimCount;\r
            PeimCount < Private->Fv[FvCount].PeimCount;\r
@@ -1103,7 +1436,7 @@ PeiDispatcher (
             ASSERT_EFI_ERROR (Status);\r
             if (FvFileInfo.FileType == EFI_FV_FILETYPE_FIRMWARE_VOLUME_IMAGE) {\r
               //\r
-              // For Fv type file, Produce new FvInfo PPI and FV hob\r
+              // For FV type file, Produce new FvInfo PPI and FV HOB\r
               //\r
               Status = ProcessFvFile (Private, &Private->Fv[FvCount], PeimFileHandle);\r
               if (Status == EFI_SUCCESS) {\r
@@ -1193,13 +1526,17 @@ PeiDispatcher (
             PeiCheckAndSwitchStack (SecCoreData, Private);\r
 \r
             if ((Private->PeiMemoryInstalled) && (Private->Fv[FvCount].PeimState[PeimCount] == PEIM_STATE_REGISTER_FOR_SHADOW) &&   \\r
-                (Private->HobList.HandoffInformationTable->BootMode != BOOT_ON_S3_RESUME || PcdGetBool (PcdShadowPeimOnS3Boot))) {\r
+                (PcdGetBool (PcdMigrateTemporaryRamFirmwareVolumes) ||\r
+                 (Private->HobList.HandoffInformationTable->BootMode != BOOT_ON_S3_RESUME) ||\r
+                 PcdGetBool (PcdShadowPeimOnS3Boot))\r
+              ) {\r
               //\r
               // If memory is available we shadow images by default for performance reasons.\r
               // We call the entry point a 2nd time so the module knows it's shadowed.\r
               //\r
               //PERF_START (PeiServices, L"PEIM", PeimFileHandle, 0);\r
-              if ((Private->HobList.HandoffInformationTable->BootMode != BOOT_ON_S3_RESUME) && !PcdGetBool (PcdShadowPeimOnBoot)) {\r
+              if ((Private->HobList.HandoffInformationTable->BootMode != BOOT_ON_S3_RESUME) && !PcdGetBool (PcdShadowPeimOnBoot) &&\r
+                  !PcdGetBool (PcdMigrateTemporaryRamFirmwareVolumes)) {\r
                 //\r
                 // Load PEIM into Memory for Register for shadow PEIM.\r
                 //\r
@@ -1235,7 +1572,7 @@ PeiDispatcher (
 \r
       //\r
       // Before walking through the next FV, we should set them to NULL/0 to\r
-      // start at the begining of the next FV.\r
+      // start at the beginning of the next FV.\r
       //\r
       Private->CurrentFileHandle = NULL;\r
       Private->CurrentPeimCount = 0;\r
@@ -1265,7 +1602,7 @@ PeiDispatcher (
 \r
   @param PrivateData     PeiCore's private data structure\r
   @param OldCoreData     Old data from SecCore\r
-                         NULL if being run in non-permament memory mode.\r
+                         NULL if being run in non-permanent memory mode.\r
   @param SecCoreData     Points to a data structure containing information about the PEI core's operating\r
                          environment, such as the size and location of temporary RAM, the stack location and\r
                          the BFV location.\r
@@ -1323,7 +1660,7 @@ DepexSatisfied (
 \r
   if (PeimCount < Private->AprioriCount) {\r
     //\r
-    // If it's in the Apriori file then we set Depex to TRUE\r
+    // If it's in the Apriori file then we set DEPEX to TRUE\r
     //\r
     DEBUG ((DEBUG_DISPATCH, "  RESULT = TRUE (Apriori)\n"));\r
     return TRUE;\r
@@ -1353,8 +1690,8 @@ DepexSatisfied (
 }\r
 \r
 /**\r
-  This routine enable a PEIM to register itself to shadow when PEI Foundation\r
-  discovery permanent memory.\r
+  This routine enables a PEIM to register itself for shadow when the PEI Foundation\r
+  discovers permanent memory.\r
 \r
   @param FileHandle             File handle of a PEIM.\r
 \r