From: Michael Kubacki <michael.kuba...@microsoft.com>

REF:https://bugzilla.tianocore.org/show_bug.cgi?id=3812

This library is introduced to add  a general abstraction for PRM context
buffer management.

Cc: Andrew Fish <af...@apple.com>
Cc: Kang Gao <kang....@intel.com>
Cc: Michael D Kinney <michael.d.kin...@intel.com>
Cc: Michael Kubacki <michael.kuba...@microsoft.com>
Cc: Leif Lindholm <l...@nuviainc.com>
Cc: Benjamin You <benjamin....@intel.com>
Cc: Liu Yun <yun.y....@intel.com>
Cc: Ankit Sinha <ankit.si...@intel.com>
Cc: Nate DeSimone <nathaniel.l.desim...@intel.com>
Signed-off-by: Michael Kubacki <michael.kuba...@microsoft.com>
---
 PrmPkg/Library/DxePrmContextBufferLib/DxePrmContextBufferLib.c   | 196 
++++++++++++++++++++
 PrmPkg/Include/Library/PrmContextBufferLib.h                     |  99 
++++++++++
 PrmPkg/Library/DxePrmContextBufferLib/DxePrmContextBufferLib.inf |  35 ++++
 3 files changed, 330 insertions(+)

diff --git a/PrmPkg/Library/DxePrmContextBufferLib/DxePrmContextBufferLib.c 
b/PrmPkg/Library/DxePrmContextBufferLib/DxePrmContextBufferLib.c
new file mode 100644
index 000000000000..1a1a15b5cdbb
--- /dev/null
+++ b/PrmPkg/Library/DxePrmContextBufferLib/DxePrmContextBufferLib.c
@@ -0,0 +1,196 @@
+/** @file
+
+  The PRM Buffer Context library provides a general abstraction for context 
buffer management.
+
+  Copyright (c) Microsoft Corporation
+  SPDX-License-Identifier: BSD-2-Clause-Patent
+
+**/
+
+#include <Library/BaseLib.h>
+#include <Library/BaseMemoryLib.h>
+#include <Library/DebugLib.h>
+#include <Library/PrmContextBufferLib.h>
+#include <Library/UefiBootServicesTableLib.h>
+#include <Protocol/PrmConfig.h>
+
+#define _DBGMSGID_        "[PRMCONTEXTBUFFERLIB]"
+
+/**
+  Finds a PRM context buffer for the given PRM handler GUID.
+
+  Note: PRM_MODULE_CONTEXT_BUFFERS is at the PRM module level while 
PRM_CONTEXT_BUFFER is at the PRM handler level.
+
+  @param[in]  HandlerGuid                 A pointer to the PRM handler GUID.
+  @param[in]  ModuleContextBuffers        A pointer to the PRM context buffers 
structure for the PRM module.
+  @param[out] PrmModuleContextBuffer      A pointer to a pointer that will be 
set to the PRM context buffer
+                                          if successfully found.
+
+  @retval EFI_SUCCESS                     The PRM context buffer was found.
+  @retval EFI_INVALID_PARAMETER           A required parameter pointer is NULL.
+  @retval EFI_NOT_FOUND                   The context buffer for the given PRM 
handler GUID could not be found.
+
+**/
+EFI_STATUS
+FindContextBufferInModuleBuffers (
+  IN  CONST EFI_GUID                      *HandlerGuid,
+  IN  CONST PRM_MODULE_CONTEXT_BUFFERS    *ModuleContextBuffers,
+  OUT CONST PRM_CONTEXT_BUFFER            **ContextBuffer
+  )
+{
+  UINTN                                   Index;
+
+  DEBUG ((DEBUG_INFO, "    %a %a - Entry.\n", _DBGMSGID_, __FUNCTION__));
+
+  if (HandlerGuid == NULL || ModuleContextBuffers == NULL || ContextBuffer == 
NULL) {
+    return EFI_INVALID_PARAMETER;
+  }
+
+  for (Index = 0; Index < ModuleContextBuffers->BufferCount; Index++) {
+    if (CompareGuid (&ModuleContextBuffers->Buffer[Index].HandlerGuid, 
HandlerGuid)) {
+      *ContextBuffer = &ModuleContextBuffers->Buffer[Index];
+      return EFI_SUCCESS;
+    }
+  }
+
+  return EFI_NOT_FOUND;
+}
+
+/**
+  Returns a PRM context buffers structure for the given PRM search type.
+
+  This function allows a caller to get the context buffers structure for a PRM 
module with either the PRM module
+  GUID or the GUID for a PRM handler in the module.
+
+  Note: PRM_MODULE_CONTEXT_BUFFERS is at the PRM module level while 
PRM_CONTEXT_BUFFER is at the PRM handler level.
+
+  @param[in]  GuidSearchType              The type of GUID passed in the Guid 
argument.
+  @param[in]  Guid                        A pointer to the GUID of a PRM 
module or PRM handler. The actual GUID type
+                                          will be interpreted based on the 
value passed in GuidSearchType.
+  @param[out] PrmModuleContextBuffers     A pointer to a pointer that will be 
set to the PRM context buffers
+                                          structure if successfully found.
+
+  @retval EFI_SUCCESS                     The PRM context buffers structure 
was found.
+  @retval EFI_INVALID_PARAMETER           A required parameter pointer is NULL.
+  @retval EFI_NOT_FOUND                   The context buffers for the given 
GUID could not be found.
+
+**/
+EFI_STATUS
+GetModuleContextBuffers (
+  IN  PRM_GUID_SEARCH_TYPE                GuidSearchType,
+  IN  CONST EFI_GUID                      *Guid,
+  OUT CONST PRM_MODULE_CONTEXT_BUFFERS    **PrmModuleContextBuffers
+  )
+{
+  EFI_STATUS                  Status;
+  UINTN                       HandleCount;
+  UINTN                       Index;
+  EFI_HANDLE                  *HandleBuffer;
+  PRM_CONFIG_PROTOCOL         *PrmConfigProtocol;
+  CONST PRM_CONTEXT_BUFFER    *PrmContextBuffer;
+
+  DEBUG ((DEBUG_INFO, "    %a %a - Entry.\n", _DBGMSGID_, __FUNCTION__));
+
+  if (Guid == NULL || PrmModuleContextBuffers == NULL) {
+    return EFI_INVALID_PARAMETER;
+  }
+  *PrmModuleContextBuffers = NULL;
+
+  Status = gBS->LocateHandleBuffer (
+                  ByProtocol,
+                  &gPrmConfigProtocolGuid,
+                  NULL,
+                  &HandleCount,
+                  &HandleBuffer
+                  );
+  if (!EFI_ERROR (Status)) {
+    for (Index = 0; Index < HandleCount; Index++) {
+      Status = gBS->HandleProtocol (
+                      HandleBuffer[Index],
+                      &gPrmConfigProtocolGuid,
+                      (VOID **) &PrmConfigProtocol
+                      );
+      ASSERT_EFI_ERROR (Status);
+      if (EFI_ERROR (Status) || PrmConfigProtocol == NULL) {
+        continue;
+      }
+
+      if (GuidSearchType == ByModuleGuid) {
+        if (CompareGuid (&PrmConfigProtocol->ModuleContextBuffers.ModuleGuid, 
Guid)) {
+          DEBUG ((
+            DEBUG_INFO,
+            "      %a %a: Found a PRM configuration protocol for PRM module 
%g.\n",
+            _DBGMSGID_,
+            __FUNCTION__,
+            Guid
+            ));
+
+          *PrmModuleContextBuffers = &PrmConfigProtocol->ModuleContextBuffers;
+          return EFI_SUCCESS;
+        }
+      } else {
+        Status = FindContextBufferInModuleBuffers (Guid, 
&PrmConfigProtocol->ModuleContextBuffers, &PrmContextBuffer);
+        if (!EFI_ERROR (Status)) {
+          *PrmModuleContextBuffers = &PrmConfigProtocol->ModuleContextBuffers;
+          return EFI_SUCCESS;
+        }
+      }
+    }
+  }
+
+  DEBUG ((
+    DEBUG_INFO,
+    "      %a %a: Could not locate a PRM configuration protocol for PRM 
handler %g.\n",
+    _DBGMSGID_,
+    __FUNCTION__,
+    Guid
+    ));
+
+  return EFI_NOT_FOUND;
+}
+
+/**
+  Returns a PRM context buffer for the given PRM handler.
+
+  @param[in]  PrmHandlerGuid              A pointer to the GUID for the PRM 
handler.
+  @param[in]  PrmModuleContextBuffers     A pointer to a 
PRM_MODULE_CONTEXT_BUFFERS structure. If this optional
+                                          parameter is provided, the handler 
context buffer will be searched for in this
+                                          buffer structure which saves time by 
not performing a global search for the
+                                          module buffer structure.
+  @param[out] PrmContextBuffer            A pointer to a pointer that will be 
set to the PRM context buffer
+                                          if successfully found.
+
+  @retval EFI_SUCCESS                     The PRM context buffer was found.
+  @retval EFI_INVALID_PARAMETER           A required parameter pointer is NULL.
+  @retval EFI_NOT_FOUND                   The context buffer for the PRM 
handler could not be found.
+
+**/
+EFI_STATUS
+GetContextBuffer (
+  IN  CONST EFI_GUID                      *PrmHandlerGuid,
+  IN  CONST PRM_MODULE_CONTEXT_BUFFERS    *PrmModuleContextBuffers  OPTIONAL,
+  OUT CONST PRM_CONTEXT_BUFFER            **PrmContextBuffer
+  )
+{
+  EFI_STATUS                              Status;
+  CONST PRM_MODULE_CONTEXT_BUFFERS        *ContextBuffers;
+
+  DEBUG ((DEBUG_INFO, "    %a %a - Entry.\n", _DBGMSGID_, __FUNCTION__));
+
+  if (PrmHandlerGuid == NULL || PrmContextBuffer == NULL) {
+    return EFI_INVALID_PARAMETER;
+  }
+  *PrmContextBuffer = NULL;
+
+  if (PrmModuleContextBuffers == NULL) {
+    Status = GetModuleContextBuffers (ByHandlerGuid, PrmHandlerGuid, 
&ContextBuffers);
+    if (EFI_ERROR (Status)) {
+      return EFI_NOT_FOUND;
+    }
+  } else {
+    ContextBuffers = PrmModuleContextBuffers;
+  }
+  Status = FindContextBufferInModuleBuffers (PrmHandlerGuid, ContextBuffers, 
PrmContextBuffer);
+
+  return Status;
+}
diff --git a/PrmPkg/Include/Library/PrmContextBufferLib.h 
b/PrmPkg/Include/Library/PrmContextBufferLib.h
new file mode 100644
index 000000000000..93dcd1e76642
--- /dev/null
+++ b/PrmPkg/Include/Library/PrmContextBufferLib.h
@@ -0,0 +1,99 @@
+/** @file
+
+  The PRM Buffer Context library provides a general abstraction for context 
buffer management.
+
+  Copyright (c) Microsoft Corporation
+  SPDX-License-Identifier: BSD-2-Clause-Patent
+
+**/
+
+#ifndef PRM_CONTEXT_BUFFER_LIB_H_
+#define PRM_CONTEXT_BUFFER_LIB_H_
+
+#include <Base.h>
+#include <PrmContextBuffer.h>
+#include <Uefi.h>
+
+typedef enum {
+  ///
+  /// Search by the PRM module GUID
+  ///
+  ByModuleGuid,
+  ///
+  /// Search by the PRM handler GUID
+  ///
+  ByHandlerGuid
+} PRM_GUID_SEARCH_TYPE;
+
+/**
+  Finds a PRM context buffer for the given PRM handler GUID.
+
+  Note: PRM_MODULE_CONTEXT_BUFFERS is at the PRM module level while 
PRM_CONTEXT_BUFFER is at the PRM handler level.
+
+  @param[in]  HandlerGuid                 A pointer to the PRM handler GUID.
+  @param[in]  ModuleContextBuffers        A pointer to the PRM context buffers 
structure for the PRM module.
+  @param[out] PrmModuleContextBuffer      A pointer to a pointer that will be 
set to the PRM context buffer
+                                          if successfully found.
+
+  @retval EFI_SUCCESS                     The PRM context buffer was found.
+  @retval EFI_INVALID_PARAMETER           A required parameter pointer is NULL.
+  @retval EFI_NOT_FOUND                   The context buffer for the given PRM 
handler GUID could not be found.
+
+**/
+EFI_STATUS
+FindContextBufferInModuleBuffers (
+  IN  CONST EFI_GUID                      *HandlerGuid,
+  IN  CONST PRM_MODULE_CONTEXT_BUFFERS    *ModuleContextBuffers,
+  OUT CONST PRM_CONTEXT_BUFFER            **ContextBuffer
+  );
+
+/**
+  Returns a PRM context buffers structure for the given PRM search type.
+
+  This function allows a caller to get the context buffers structure for a PRM 
module with either the PRM module
+  GUID or the GUID for a PRM handler in the module.
+
+  Note: PRM_MODULE_CONTEXT_BUFFERS is at the PRM module level while 
PRM_CONTEXT_BUFFER is at the PRM handler level.
+
+  @param[in]  GuidSearchType              The type of GUID passed in the Guid 
argument.
+  @param[in]  Guid                        A pointer to the GUID of a PRM 
module or PRM handler. The actual GUID type
+                                          will be interpreted based on the 
value passed in GuidSearchType.
+  @param[out] PrmModuleContextBuffers     A pointer to a pointer that will be 
set to the PRM context buffers
+                                          structure if successfully found.
+
+  @retval EFI_SUCCESS                     The PRM context buffers structure 
was found.
+  @retval EFI_INVALID_PARAMETER           A required parameter pointer is NULL.
+  @retval EFI_NOT_FOUND                   The context buffers for the given 
GUID could not be found.
+
+**/
+EFI_STATUS
+GetModuleContextBuffers (
+  IN  PRM_GUID_SEARCH_TYPE                GuidSearchType,
+  IN  CONST EFI_GUID                      *Guid,
+  OUT CONST PRM_MODULE_CONTEXT_BUFFERS    **PrmModuleContextBuffers
+  );
+
+/**
+  Returns a PRM context buffer for the given PRM handler.
+
+  @param[in]  PrmHandlerGuid              A pointer to the GUID for the PRM 
handler.
+  @param[in]  PrmModuleContextBuffers     A pointer to a 
PRM_MODULE_CONTEXT_BUFFERS structure. If this optional
+                                          parameter is provided, the handler 
context buffer will be searched for in this
+                                          buffer structure which saves time by 
not performing a global search for the
+                                          module buffer structure.
+  @param[out] PrmContextBuffer            A pointer to a pointer that will be 
set to the PRM context buffer
+                                          if successfully found.
+
+  @retval EFI_SUCCESS                     The PRM context buffer was found.
+  @retval EFI_INVALID_PARAMETER           A required parameter pointer is NULL.
+  @retval EFI_NOT_FOUND                   The context buffer for the PRM 
handler could not be found.
+
+**/
+EFI_STATUS
+GetContextBuffer (
+  IN  CONST EFI_GUID                      *PrmHandlerGuid,
+  IN  CONST PRM_MODULE_CONTEXT_BUFFERS    *PrmModuleContextBuffers  OPTIONAL,
+  OUT CONST PRM_CONTEXT_BUFFER            **PrmContextBuffer
+  );
+
+#endif
diff --git a/PrmPkg/Library/DxePrmContextBufferLib/DxePrmContextBufferLib.inf 
b/PrmPkg/Library/DxePrmContextBufferLib/DxePrmContextBufferLib.inf
new file mode 100644
index 000000000000..16ed1edfe36a
--- /dev/null
+++ b/PrmPkg/Library/DxePrmContextBufferLib/DxePrmContextBufferLib.inf
@@ -0,0 +1,35 @@
+## @file
+#  PRM Context Buffer Library
+#
+#  Provides a general abstraction for PRM context buffer management.
+#
+#  Copyright (c) Microsoft Corporation
+#
+#  SPDX-License-Identifier: BSD-2-Clause-Patent
+#
+##
+
+[Defines]
+  INF_VERSION         = 0x00010005
+  BASE_NAME           = DxePrmContextBufferLib
+  FILE_GUID           = 49828E93-29FA-4665-B8B1-19BA4059D140
+  MODULE_TYPE         = DXE_DRIVER
+  VERSION_STRING      = 1.0
+  LIBRARY_CLASS       = PrmContextBufferLib|DXE_DRIVER UEFI_DRIVER 
UEFI_APPLICATION
+
+[Sources]
+  DxePrmContextBufferLib.c
+
+[Packages]
+  MdePkg/MdePkg.dec
+  MdeModulePkg/MdeModulePkg.dec
+  PrmPkg/PrmPkg.dec
+
+[Protocols]
+  gPrmConfigProtocolGuid
+
+[LibraryClasses]
+  BaseLib
+  BaseMemoryLib
+  DebugLib
+  UefiBootServicesTableLib
-- 
2.28.0.windows.1



-=-=-=-=-=-=-=-=-=-=-=-
Groups.io Links: You receive all messages sent to this group.
View/Reply Online (#87845): https://edk2.groups.io/g/devel/message/87845
Mute This Topic: https://groups.io/mt/89955955/21656
Group Owner: devel+ow...@edk2.groups.io
Unsubscribe: https://edk2.groups.io/g/devel/unsub [arch...@mail-archive.com]
-=-=-=-=-=-=-=-=-=-=-=-


Reply via email to