ShellPkg: make automatically created shells quit automatically

[mirror_edk2.git] / ArmPlatformPkg / MemoryInitPei / MemoryInitPeiLib.c

/** @file\r
*\r
*  Copyright (c) 2011, ARM Limited. All rights reserved.\r
*\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
*\r
**/\r
\r
#include <PiPei.h>\r
\r
#include <Library/ArmPlatformLib.h>\r
#include <Library/DebugLib.h>\r
#include <Library/HobLib.h>\r
#include <Library/MemoryAllocationLib.h>\r
#include <Library/PcdLib.h>\r
\r
VOID\r
BuildMemoryTypeInformationHob (\r
  VOID\r
  );\r
\r
VOID\r
InitMmu (\r
  VOID\r
  )\r
{\r
  ARM_MEMORY_REGION_DESCRIPTOR  *MemoryTable;\r
  VOID                          *TranslationTableBase;\r
  UINTN                         TranslationTableSize;\r
\r
  // Get Virtual Memory Map from the Platform Library\r
  ArmPlatformGetVirtualMemoryMap (&MemoryTable);\r
\r
  //Note: Because we called PeiServicesInstallPeiMemory() before to call InitMmu() the MMU Page Table resides in\r
  //      DRAM (even at the top of DRAM as it is the first permanent memory allocation)\r
  ArmConfigureMmu (MemoryTable, &TranslationTableBase, &TranslationTableSize);\r
}\r
\r
/*++\r
\r
Routine Description:\r
\r
\r
\r
Arguments:\r
\r
  FileHandle  - Handle of the file being invoked.\r
  PeiServices - Describes the list of possible PEI Services.\r
\r
Returns:\r
\r
  Status -  EFI_SUCCESS if the boot mode could be set\r
\r
--*/\r
EFI_STATUS\r
EFIAPI\r
MemoryPeim (\r
  IN EFI_PHYSICAL_ADDRESS               UefiMemoryBase,\r
  IN UINT64                             UefiMemorySize\r
  )\r
{\r
  EFI_RESOURCE_ATTRIBUTE_TYPE ResourceAttributes;\r
  UINT64                      ResourceLength;\r
  EFI_PEI_HOB_POINTERS        NextHob;\r
  EFI_PHYSICAL_ADDRESS        FdTop;\r
  EFI_PHYSICAL_ADDRESS        SystemMemoryTop;\r
  EFI_PHYSICAL_ADDRESS        ResourceTop;\r
  BOOLEAN                     Found;\r
\r
  // Ensure PcdSystemMemorySize has been set\r
  ASSERT (PcdGet32 (PcdSystemMemorySize) != 0);\r
\r
  //\r
  // Now, the permanent memory has been installed, we can call AllocatePages()\r
  //\r
  ResourceAttributes = (\r
      EFI_RESOURCE_ATTRIBUTE_PRESENT |\r
      EFI_RESOURCE_ATTRIBUTE_INITIALIZED |\r
      EFI_RESOURCE_ATTRIBUTE_UNCACHEABLE |\r
      EFI_RESOURCE_ATTRIBUTE_WRITE_COMBINEABLE |\r
      EFI_RESOURCE_ATTRIBUTE_WRITE_THROUGH_CACHEABLE |\r
      EFI_RESOURCE_ATTRIBUTE_WRITE_BACK_CACHEABLE |\r
      EFI_RESOURCE_ATTRIBUTE_TESTED\r
  );\r
\r
  // Reserved the memory space occupied by the firmware volume\r
  BuildResourceDescriptorHob (\r
      EFI_RESOURCE_SYSTEM_MEMORY,\r
      ResourceAttributes,\r
      PcdGet32 (PcdSystemMemoryBase),\r
      PcdGet32 (PcdSystemMemorySize)\r
  );\r
\r
  SystemMemoryTop = PcdGet32 (PcdSystemMemoryBase) + PcdGet32 (PcdSystemMemorySize);\r
  FdTop = PcdGet32(PcdFdBaseAddress) + PcdGet32(PcdFdSize);\r
\r
  // EDK2 does not have the concept of boot firmware copied into DRAM. To avoid the DXE\r
  // core to overwrite this area we must mark the region with the attribute non-present\r
  if ((PcdGet32 (PcdFdBaseAddress) >= PcdGet32 (PcdSystemMemoryBase)) && (FdTop <= SystemMemoryTop)) {\r
    Found = FALSE;\r
\r
    // Search for System Memory Hob that contains the firmware\r
    NextHob.Raw = GetHobList ();\r
    while ((NextHob.Raw = GetNextHob (EFI_HOB_TYPE_RESOURCE_DESCRIPTOR, NextHob.Raw)) != NULL) {\r
      if ((NextHob.ResourceDescriptor->ResourceType == EFI_RESOURCE_SYSTEM_MEMORY) &&\r
          (PcdGet32(PcdFdBaseAddress) >= NextHob.ResourceDescriptor->PhysicalStart) &&\r
          (FdTop <= NextHob.ResourceDescriptor->PhysicalStart + NextHob.ResourceDescriptor->ResourceLength))\r
      {\r
        ResourceAttributes = NextHob.ResourceDescriptor->ResourceAttribute;\r
        ResourceLength = NextHob.ResourceDescriptor->ResourceLength;\r
        ResourceTop = NextHob.ResourceDescriptor->PhysicalStart + ResourceLength;\r
\r
        if (PcdGet32(PcdFdBaseAddress) == NextHob.ResourceDescriptor->PhysicalStart) {\r
          if (SystemMemoryTop == FdTop) {\r
            NextHob.ResourceDescriptor->ResourceAttribute = ResourceAttributes & ~EFI_RESOURCE_ATTRIBUTE_PRESENT;\r
          } else {\r
            // Create the System Memory HOB for the firmware with the non-present attribute\r
            BuildResourceDescriptorHob (EFI_RESOURCE_SYSTEM_MEMORY,\r
                                        ResourceAttributes & ~EFI_RESOURCE_ATTRIBUTE_PRESENT,\r
                                        PcdGet32(PcdFdBaseAddress),\r
                                        PcdGet32(PcdFdSize));\r
\r
            // Top of the FD is system memory available for UEFI\r
            NextHob.ResourceDescriptor->PhysicalStart += PcdGet32(PcdFdSize);\r
            NextHob.ResourceDescriptor->ResourceLength -= PcdGet32(PcdFdSize);\r
          }\r
        } else {\r
          // Create the System Memory HOB for the firmware with the non-present attribute\r
          BuildResourceDescriptorHob (EFI_RESOURCE_SYSTEM_MEMORY,\r
                                      ResourceAttributes & ~EFI_RESOURCE_ATTRIBUTE_PRESENT,\r
                                      PcdGet32(PcdFdBaseAddress),\r
                                      PcdGet32(PcdFdSize));\r
\r
          // Update the HOB\r
          NextHob.ResourceDescriptor->ResourceLength = PcdGet32(PcdFdBaseAddress) - NextHob.ResourceDescriptor->PhysicalStart;\r
\r
          // If there is some memory available on the top of the FD then create a HOB\r
          if (FdTop < NextHob.ResourceDescriptor->PhysicalStart + ResourceLength) {\r
            // Create the System Memory HOB for the remaining region (top of the FD)\r
            BuildResourceDescriptorHob (EFI_RESOURCE_SYSTEM_MEMORY,\r
                                        ResourceAttributes,\r
                                        FdTop,\r
                                        ResourceTop - FdTop);\r
          }\r
        }\r
        Found = TRUE;\r
        break;\r
      }\r
      NextHob.Raw = GET_NEXT_HOB (NextHob);\r
    }\r
\r
    ASSERT(Found);\r
  }\r
\r
  // Build Memory Allocation Hob\r
  InitMmu ();\r
\r
  if (FeaturePcdGet (PcdPrePiProduceMemoryTypeInformationHob)) {\r
    // Optional feature that helps prevent EFI memory map fragmentation.\r
    BuildMemoryTypeInformationHob ();\r
  }\r
\r
  return EFI_SUCCESS;\r
}\r