ArmVirtPkg/PlatformBootManagerLib: unload image on EFI_SECURITY_VIOLATION

[mirror_edk2.git] / ArmVirtPkg / Library / PlatformPeiLib / PlatformPeiLib.c

/** @file\r
*\r
*  Copyright (c) 2011-2014, ARM Limited. All rights reserved.\r
*  Copyright (c) 2014, Linaro Limited. All rights reserved.\r
*\r
*  SPDX-License-Identifier: BSD-2-Clause-Patent\r
*\r
**/\r
\r
#include <PiPei.h>\r
\r
#include <Library/MemoryAllocationLib.h>\r
#include <Library/DebugLib.h>\r
#include <Library/HobLib.h>\r
#include <Library/PcdLib.h>\r
#include <libfdt.h>\r
\r
#include <Guid/EarlyPL011BaseAddress.h>\r
#include <Guid/FdtHob.h>\r
\r
EFI_STATUS\r
EFIAPI\r
PlatformPeim (\r
  VOID\r
  )\r
{\r
  VOID               *Base;\r
  VOID               *NewBase;\r
  UINTN              FdtSize;\r
  UINTN              FdtPages;\r
  UINT64             *FdtHobData;\r
  UINT64             *UartHobData;\r
  INT32              Node, Prev;\r
  CONST CHAR8        *Compatible;\r
  CONST CHAR8        *CompItem;\r
  CONST CHAR8        *NodeStatus;\r
  INT32              Len;\r
  INT32              StatusLen;\r
  CONST UINT64       *RegProp;\r
  UINT64             UartBase;\r
\r
\r
  Base = (VOID*)(UINTN)PcdGet64 (PcdDeviceTreeInitialBaseAddress);\r
  ASSERT (Base != NULL);\r
  ASSERT (fdt_check_header (Base) == 0);\r
\r
  FdtSize = fdt_totalsize (Base) + PcdGet32 (PcdDeviceTreeAllocationPadding);\r
  FdtPages = EFI_SIZE_TO_PAGES (FdtSize);\r
  NewBase = AllocatePages (FdtPages);\r
  ASSERT (NewBase != NULL);\r
  fdt_open_into (Base, NewBase, EFI_PAGES_TO_SIZE (FdtPages));\r
\r
  FdtHobData = BuildGuidHob (&gFdtHobGuid, sizeof *FdtHobData);\r
  ASSERT (FdtHobData != NULL);\r
  *FdtHobData = (UINTN)NewBase;\r
\r
  UartHobData = BuildGuidHob (&gEarlyPL011BaseAddressGuid, sizeof *UartHobData);\r
  ASSERT (UartHobData != NULL);\r
  *UartHobData = 0;\r
\r
  //\r
  // Look for a UART node\r
  //\r
  for (Prev = 0;; Prev = Node) {\r
    Node = fdt_next_node (Base, Prev, NULL);\r
    if (Node < 0) {\r
      break;\r
    }\r
\r
    //\r
    // Check for UART node\r
    //\r
    Compatible = fdt_getprop (Base, Node, "compatible", &Len);\r
\r
    //\r
    // Iterate over the NULL-separated items in the compatible string\r
    //\r
    for (CompItem = Compatible; CompItem != NULL && CompItem < Compatible + Len;\r
      CompItem += 1 + AsciiStrLen (CompItem)) {\r
\r
      if (AsciiStrCmp (CompItem, "arm,pl011") == 0) {\r
        NodeStatus = fdt_getprop (Base, Node, "status", &StatusLen);\r
        if (NodeStatus != NULL && AsciiStrCmp (NodeStatus, "okay") != 0) {\r
          continue;\r
        }\r
\r
        RegProp = fdt_getprop (Base, Node, "reg", &Len);\r
        ASSERT (Len == 16);\r
\r
        UartBase = fdt64_to_cpu (ReadUnaligned64 (RegProp));\r
\r
        DEBUG ((EFI_D_INFO, "%a: PL011 UART @ 0x%lx\n", __FUNCTION__, UartBase));\r
\r
        *UartHobData = UartBase;\r
        break;\r
      }\r
    }\r
  }\r
\r
  BuildFvHob (PcdGet64 (PcdFvBaseAddress), PcdGet32 (PcdFvSize));\r
\r
  return EFI_SUCCESS;\r
}\r