PrmPkg: Remove ALLOCATE_CONTEXT_BUFFER_IN_FW build flag

[mirror_edk2.git] / PrmPkg / Samples / PrmSampleContextBufferModule / Library / DxeContextBufferModuleConfigLib / DxeContextBufferModuleConfigLib.c

/** @file\r
\r
  The boot services environment configuration library for the Context Buffer Sample PRM module.\r
\r
  Copyright (c) Microsoft Corporation\r
  SPDX-License-Identifier: BSD-2-Clause-Patent\r
\r
**/\r
\r
#include <Library/BaseLib.h>\r
#include <Library/BaseMemoryLib.h>\r
#include <Library/DebugLib.h>\r
#include <Library/MemoryAllocationLib.h>\r
#include <Library/UefiBootServicesTableLib.h>\r
#include <Protocol/PrmConfig.h>\r
#include <Samples/PrmSampleContextBufferModule/Include/StaticData.h>\r
\r
#include <PrmContextBuffer.h>\r
#include <PrmDataBuffer.h>\r
\r
STATIC EFI_HANDLE  mPrmConfigProtocolHandle;\r
\r
// {5a6cf42b-8bb4-472c-a233-5c4dc4033dc7}\r
STATIC CONST EFI_GUID mPrmModuleGuid = {0x5a6cf42b, 0x8bb4, 0x472c, {0xa2, 0x33, 0x5c, 0x4d, 0xc4, 0x03, 0x3d, 0xc7}};\r
\r
// {e1466081-7562-430f-896b-b0e523dc335a}\r
STATIC CONST EFI_GUID mDumpStaticDataBufferPrmHandlerGuid = {0xe1466081, 0x7562, 0x430f, {0x89, 0x6b, 0xb0, 0xe5, 0x23, 0xdc, 0x33, 0x5a}};\r
\r
/**\r
  Populates the static data buffer for this PRM module.\r
\r
  @param[out] StaticDataBuffer  A pointer to the static data buffer.\r
\r
  @retval EFI_SUCCESS           The static data buffer was populated successfully.\r
  @retval EFI_INVALID_PARAMETER The StaticDataBuffer pointer argument is NULL.\r
\r
**/\r
EFI_STATUS\r
PopulateStaticDataBuffer (\r
  OUT STATIC_DATA_SAMPLE_CONTEXT_BUFFER_MODULE  *StaticDataBuffer\r
  )\r
{\r
  if (StaticDataBuffer == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
\r
  //\r
  // Note: In a real-world module these values would likely come from somewhere\r
  // like a Setup menu option, PCD, binary data, runtime device info, etc. Ideally,\r
  // this configuration library would be provided an API to get what it needs (the data)\r
  // and not be concerned with how the data is provided. This makes the PRM module more\r
  // portable across systems.\r
  //\r
  StaticDataBuffer->Policy1Enabled = TRUE;\r
  StaticDataBuffer->Policy2Enabled = FALSE;\r
  SetMem (StaticDataBuffer->SomeValueArray, ARRAY_SIZE (StaticDataBuffer->SomeValueArray), 'D');\r
\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Allocates and populates the static data buffer for this PRM module.\r
\r
  @param[out] StaticDataBuffer  A pointer to a pointer to the static data buffer.\r
\r
  @retval EFI_SUCCESS           The static data buffer was allocated and filled successfully.\r
  @retval EFI_INVALID_PARAMETER The StaticDataBuffer pointer argument is NULL.\r
  @retval EFI_OUT_OF_RESOURCES  Insufficient memory resources to allocate the static data buffer.\r
\r
**/\r
EFI_STATUS\r
GetStaticDataBuffer (\r
  OUT PRM_DATA_BUFFER         **StaticDataBuffer\r
  )\r
{\r
  EFI_STATUS                  Status;\r
  PRM_DATA_BUFFER             *DataBuffer;\r
  UINTN                       DataBufferLength;\r
\r
  if (StaticDataBuffer == NULL) {\r
    return EFI_INVALID_PARAMETER;\r
  }\r
  *StaticDataBuffer = NULL;\r
\r
  //\r
  // Length of the data buffer = Buffer Header Size + Buffer Data Size\r
  //\r
  DataBufferLength = sizeof (PRM_DATA_BUFFER_HEADER) + sizeof (STATIC_DATA_SAMPLE_CONTEXT_BUFFER_MODULE);\r
\r
  DataBuffer = AllocateRuntimeZeroPool (DataBufferLength);\r
  if (DataBuffer == NULL) {\r
    return EFI_OUT_OF_RESOURCES;\r
  }\r
\r
  //\r
  // Initialize the data buffer header\r
  //\r
  DataBuffer->Header.Signature = PRM_DATA_BUFFER_HEADER_SIGNATURE;\r
  DataBuffer->Header.Length = (UINT32) DataBufferLength;\r
\r
  Status = PopulateStaticDataBuffer ((STATIC_DATA_SAMPLE_CONTEXT_BUFFER_MODULE *) &DataBuffer->Data[0]);\r
  ASSERT_EFI_ERROR (Status);\r
\r
  *StaticDataBuffer = DataBuffer;\r
  return EFI_SUCCESS;\r
}\r
\r
/**\r
  Constructor of the PRM configuration library.\r
\r
  @param[in] ImageHandle        The image handle of the driver.\r
  @param[in] SystemTable        The EFI System Table pointer.\r
\r
  @retval EFI_SUCCESS           The shell command handlers were installed successfully.\r
  @retval EFI_UNSUPPORTED       The shell level required was not found.\r
**/\r
EFI_STATUS\r
EFIAPI\r
ContextBufferModuleConfigLibConstructor (\r
  IN  EFI_HANDLE              ImageHandle,\r
  IN  EFI_SYSTEM_TABLE        *SystemTable\r
  )\r
{\r
  EFI_STATUS                  Status;\r
  PRM_CONTEXT_BUFFER          *PrmContextBuffer;\r
  PRM_DATA_BUFFER             *StaticDataBuffer;\r
  PRM_CONFIG_PROTOCOL         *PrmConfigProtocol;\r
\r
  PrmContextBuffer = NULL;\r
  StaticDataBuffer = NULL;\r
  PrmConfigProtocol = NULL;\r
\r
  /*\r
    In this sample PRM module, the protocol describing this sample module's resources is simply\r
    installed in the constructor.\r
\r
    However, if some data is not available until later, this constructor could register a callback\r
    on the dependency for the data to be available (e.g. ability to communicate with some device)\r
    and then install the protocol. The requirement is that the protocol is installed before end of DXE.\r
  */\r
\r
  //\r
  // Allocate and populate the static data buffer\r
  //\r
  Status = GetStaticDataBuffer (&StaticDataBuffer);\r
  ASSERT_EFI_ERROR (Status);\r
  if (EFI_ERROR (Status) || StaticDataBuffer == NULL) {\r
    goto Done;\r
  }\r
\r
  //\r
  // Allocate and populate the context buffer\r
  //\r
\r
  //\r
  // This context buffer is not actually used by PRM handler at OS runtime. The OS will allocate\r
  // the actual context buffer passed to the PRM handler.\r
  //\r
  // This context buffer is used internally in the firmware to associate a PRM handler with a\r
  // a static data buffer and a runtime MMIO ranges array so those can be placed into the\r
  // PRM_HANDLER_INFORMATION_STRUCT and PRM_MODULE_INFORMATION_STRUCT respectively for the PRM handler.\r
  //\r
  PrmContextBuffer = AllocateZeroPool (sizeof (*PrmContextBuffer));\r
  ASSERT (PrmContextBuffer != NULL);\r
  if (PrmContextBuffer == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto Done;\r
  }\r
  CopyGuid (&PrmContextBuffer->HandlerGuid, &mDumpStaticDataBufferPrmHandlerGuid);\r
  PrmContextBuffer->Signature = PRM_CONTEXT_BUFFER_SIGNATURE;\r
  PrmContextBuffer->Version = PRM_CONTEXT_BUFFER_INTERFACE_VERSION;\r
  PrmContextBuffer->StaticDataBuffer = StaticDataBuffer;\r
\r
  PrmConfigProtocol = AllocateZeroPool (sizeof (*PrmConfigProtocol));\r
  ASSERT (PrmConfigProtocol != NULL);\r
  if (PrmConfigProtocol == NULL) {\r
    Status = EFI_OUT_OF_RESOURCES;\r
    goto Done;\r
  }\r
  CopyGuid (&PrmConfigProtocol->ModuleContextBuffers.ModuleGuid, &mPrmModuleGuid);\r
  PrmConfigProtocol->ModuleContextBuffers.BufferCount = 1;\r
  PrmConfigProtocol->ModuleContextBuffers.Buffer = PrmContextBuffer;\r
\r
  //\r
  // Install the PRM Configuration Protocol for this module. This indicates the configuration\r
  // library has completed resource initialization for the PRM module.\r
  //\r
  Status = gBS->InstallProtocolInterface (\r
                  &mPrmConfigProtocolHandle,\r
                  &gPrmConfigProtocolGuid,\r
                  EFI_NATIVE_INTERFACE,\r
                  (VOID *) PrmConfigProtocol\r
                  );\r
\r
Done:\r
  if (EFI_ERROR (Status)) {\r
    if (StaticDataBuffer != NULL) {\r
      FreePool (StaticDataBuffer);\r
    }\r
    if (PrmContextBuffer != NULL) {\r
      FreePool (PrmContextBuffer);\r
    }\r
    if (PrmConfigProtocol != NULL) {\r
      FreePool (PrmConfigProtocol);\r
    }\r
  }\r
\r
  return Status;\r
}\r