PrmPkg/PrmContextBufferLib: Add initial library instance

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 <afish@apple.com>
Cc: Kang Gao <kang.gao@intel.com>
Cc: Michael D Kinney <michael.d.kinney@intel.com>
Cc: Michael Kubacki <michael.kubacki@microsoft.com>
Cc: Leif Lindholm <leif@nuviainc.com>
Cc: Benjamin You <benjamin.you@intel.com>
Cc: Liu Yun <yun.y.liu@intel.com>
Cc: Ankit Sinha <ankit.sinha@intel.com>
Cc: Nate DeSimone <nathaniel.l.desimone@intel.com>
Signed-off-by: Michael Kubacki <michael.kubacki@microsoft.com>
Acked-by: Michael D Kinney <michael.d.kinney@intel.com>
Acked-by: Liming Gao <gaoliming@byosoft.com.cn>
Acked-by: Leif Lindholm <quic_llindhol@quicinc.com>
Reviewed-by: Ankit Sinha <ankit.sinha@intel.com>
This commit is contained in:
Michael Kubacki 2020-04-07 11:19:49 -07:00 committed by mergify[bot]
parent 5f76c3e471
commit e189e01af2
3 changed files with 330 additions and 0 deletions

View File

@ -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

View File

@ -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;
}

View File

@ -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