OvmfPkg/PlatformPei: set the Hypervisor Features PCD

[mirror_edk2.git] / UefiPayloadPkg / FvbRuntimeDxe / FvbInfo.c

/** @file\r
\r
  Copyright (c) 2014 - 2021, Intel Corporation. All rights reserved.<BR>\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include <PiDxe.h>\r
#include <Protocol/FirmwareVolumeBlock.h>\r
#include <Library/PcdLib.h>\r
#include <Library/DebugLib.h>\r
#include <Library/BaseLib.h>\r
#include <Guid/FirmwareFileSystem2.h>\r
#include <Guid/SystemNvDataGuid.h>\r
#include <Guid/NvVariableInfoGuid.h>\r
#include <Library/HobLib.h>\r
\r
#define FVB_MEDIA_BLOCK_SIZE  0x1000\r
\r
typedef struct {\r
  EFI_FIRMWARE_VOLUME_HEADER    FvInfo;\r
  EFI_FV_BLOCK_MAP_ENTRY        End[1];\r
} EFI_FVB2_MEDIA_INFO;\r
\r
//\r
// This data structure contains a template of FV header which is used to restore\r
// Fv header if it's corrupted.\r
//\r
EFI_FVB2_MEDIA_INFO  mFvbMediaInfo = {\r
  {\r
    { 0, },         // ZeroVector[16]\r
    EFI_SYSTEM_NV_DATA_FV_GUID,\r
    0,\r
    EFI_FVH_SIGNATURE,\r
    0x0004feff,     // check PiFirmwareVolume.h for details on EFI_FVB_ATTRIBUTES_2\r
    sizeof (EFI_FIRMWARE_VOLUME_HEADER) + sizeof (EFI_FV_BLOCK_MAP_ENTRY),\r
    0,              // CheckSum which will be calucated dynamically.\r
    0,              // ExtHeaderOffset\r
    { 0, },\r
    EFI_FVH_REVISION,\r
    {\r
      {\r
        0,\r
        FVB_MEDIA_BLOCK_SIZE,\r
      }\r
    }\r
  },\r
  {\r
    {\r
      0,\r
      0\r
    }\r
  }\r
};\r
\r
/**\r
  Initialize the variable store\r
\r
  @retval     EFI_SUCCESS if initialize the store success.\r
\r
**/\r
EFI_STATUS\r
InitVariableStore (\r
  VOID\r
  )\r
{\r
  EFI_STATUS         Status;\r
  UINT32             NvStorageBase;\r
  UINT32             NvStorageSize;\r
  UINT32             NvVariableSize;\r
  UINT32             FtwWorkingSize;\r
  UINT32             FtwSpareSize;\r
  EFI_HOB_GUID_TYPE  *GuidHob;\r
  NV_VARIABLE_INFO   *NvVariableInfo;\r
\r
  //\r
  // Find SPI flash variable hob\r
  //\r
  GuidHob = GetFirstGuidHob (&gNvVariableInfoGuid);\r
  if (GuidHob == NULL) {\r
    ASSERT (FALSE);\r
    return EFI_NOT_FOUND;\r
  }\r
\r
  NvVariableInfo = (NV_VARIABLE_INFO *)GET_GUID_HOB_DATA (GuidHob);\r
\r
  //\r
  // Get variable region base and size.\r
  //\r
  NvStorageSize = NvVariableInfo->VariableStoreSize;\r
  NvStorageBase = NvVariableInfo->VariableStoreBase;\r
\r
  //\r
  // NvStorageBase needs to be 4KB aligned, NvStorageSize needs to be 8KB * n\r
  //\r
  if (((NvStorageBase & (SIZE_4KB - 1)) != 0) || ((NvStorageSize & (SIZE_8KB - 1)) != 0)) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  FtwSpareSize   = NvStorageSize / 2;\r
  FtwWorkingSize = 0x2000;\r
  NvVariableSize = NvStorageSize / 2 - FtwWorkingSize;\r
  DEBUG ((DEBUG_INFO, "NvStorageBase:0x%x, NvStorageSize:0x%x\n", NvStorageBase, NvStorageSize));\r
\r
  if (NvVariableSize >= 0x80000000) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  Status = PcdSet32S (PcdFlashNvStorageVariableSize, NvVariableSize);\r
  ASSERT_EFI_ERROR (Status);\r
  Status = PcdSet32S (PcdFlashNvStorageVariableBase, NvStorageBase);\r
  ASSERT_EFI_ERROR (Status);\r
  Status = PcdSet64S (PcdFlashNvStorageVariableBase64, NvStorageBase);\r
  ASSERT_EFI_ERROR (Status);\r
\r
  Status = PcdSet32S (PcdFlashNvStorageFtwWorkingSize, FtwWorkingSize);\r
  ASSERT_EFI_ERROR (Status);\r
  Status = PcdSet32S (PcdFlashNvStorageFtwWorkingBase, NvStorageBase + NvVariableSize);\r
  ASSERT_EFI_ERROR (Status);\r
\r
  Status = PcdSet32S (PcdFlashNvStorageFtwSpareSize, FtwSpareSize);\r
  ASSERT_EFI_ERROR (Status);\r
  Status = PcdSet32S (PcdFlashNvStorageFtwSpareBase, NvStorageBase + FtwSpareSize);\r
  ASSERT_EFI_ERROR (Status);\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Get a heathy FV header used for variable store recovery\r
\r
  @retval     The FV header.\r
\r
**/\r
EFI_FIRMWARE_VOLUME_HEADER *\r
GetFvHeaderTemplate (\r
  VOID\r
  )\r
{\r
  EFI_FIRMWARE_VOLUME_HEADER  *FvHeader;\r
  UINTN                       FvSize;\r
\r
  FvSize                          = PcdGet32 (PcdFlashNvStorageFtwSpareSize) * 2;\r
  FvHeader                        = &mFvbMediaInfo.FvInfo;\r
  FvHeader->FvLength              = FvSize;\r
  FvHeader->BlockMap[0].NumBlocks = (UINT32)(FvSize / FvHeader->BlockMap[0].Length);\r
  FvHeader->Checksum              = 0;\r
  FvHeader->Checksum              = CalculateCheckSum16 ((UINT16 *)FvHeader, FvHeader->HeaderLength);\r
\r
  return FvHeader;\r
}\r