tyler.durden
/
audk
mirror of https://github.com/acidanthera/audk.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

MdePkg/UefiLib: Add a new API GetVariable3 

BZ: https://bugzilla.tianocore.org/show_bug.cgi?id=1396
Add a new API GetVariable3, which can return the attributes of a variable during reading it.

Contributed-under: TianoCore Contribution Agreement 1.1
Signed-off-by: Jiansong Xu <jiansongx.xu@intel.com>
Cc: Liming Gao <liming.gao@intel.com>
Reviewed-by: Liming Gao <liming.gao@intel.com>
This commit is contained in:
Bret Barkelew 2019-01-04 10:22:12 +08:00 committed by Liming Gao
parent ca67441558
commit 37bf678761
2 changed files with 120 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
MdePkg/Include/Library/UefiLib.h
View File

@ -737,9 +737,9 @@ GetEfiGlobalVariable (
@param[out] Value The buffer point saved the variable info.
@param[out] Size The buffer size of the variable.
@return EFI_OUT_OF_RESOURCES Allocate buffer failed.
@return EFI_SUCCESS Find the specified variable.
@return Others Errors Return errors from call to gRT->GetVariable.
@retval EFI_OUT_OF_RESOURCES Allocate buffer failed.
@retval EFI_SUCCESS Find the specified variable.
@retval Others Errors Return errors from call to gRT->GetVariable.
**/
EFI_STATUS
@ -765,9 +765,9 @@ GetVariable2 (
@param[out] Value The buffer point saved the variable info.
@param[out] Size The buffer size of the variable.
@return EFI_OUT_OF_RESOURCES Allocate buffer failed.
@return EFI_SUCCESS Find the specified variable.
@return Others Errors Return errors from call to gRT->GetVariable.
@retval EFI_OUT_OF_RESOURCES Allocate buffer failed.
@retval EFI_SUCCESS Find the specified variable.
@retval Others Errors Return errors from call to gRT->GetVariable.
**/
EFI_STATUS
@ -778,6 +778,39 @@ GetEfiGlobalVariable2 (
OUT UINTN *Size OPTIONAL
);
/** Return the attributes of the variable.
Returns the status whether get the variable success. The function retrieves
variable through the UEFI Runtime Service GetVariable(). The
returned buffer is allocated using AllocatePool(). The caller is responsible
for freeing this buffer with FreePool(). The attributes are returned if
the caller provides a valid Attribute parameter.
If Name is NULL, then ASSERT().
If Guid is NULL, then ASSERT().
If Value is NULL, then ASSERT().
@param[in] Name The pointer to a Null-terminated Unicode string.
@param[in] Guid The pointer to an EFI_GUID structure
@param[out] Value The buffer point saved the variable info.
@param[out] Size The buffer size of the variable.
@param[out] Attr The pointer to the variable attributes as found in var store
@retval EFI_OUT_OF_RESOURCES Allocate buffer failed.
@retval EFI_SUCCESS Find the specified variable.
@retval Others Errors Return errors from call to gRT->GetVariable.
**/
EFI_STATUS
EFIAPI
GetVariable3(
IN CONST CHAR16 *Name,
IN CONST EFI_GUID *Guid,
OUT VOID **Value,
OUT UINTN *Size OPTIONAL,
OUT UINT32 *Attr OPTIONAL
);
/**
Returns a pointer to an allocated buffer that contains the best matching language
from a set of supported languages.

81
MdePkg/Library/UefiLib/UefiLib.c
View File

@ -1439,6 +1439,87 @@ GetVariable2 (
return Status;
}
/** Return the attributes of the variable.
Returns the status whether get the variable success. The function retrieves
variable through the UEFI Runtime Service GetVariable(). The
returned buffer is allocated using AllocatePool(). The caller is responsible
for freeing this buffer with FreePool(). The attributes are returned if
the caller provides a valid Attribute parameter.
If Name is NULL, then ASSERT().
If Guid is NULL, then ASSERT().
If Value is NULL, then ASSERT().
@param[in] Name The pointer to a Null-terminated Unicode string.
@param[in] Guid The pointer to an EFI_GUID structure
@param[out] Value The buffer point saved the variable info.
@param[out] Size The buffer size of the variable.
@param[out] Attr The pointer to the variable attributes as found in var store
@retval EFI_OUT_OF_RESOURCES Allocate buffer failed.
@retval EFI_SUCCESS Find the specified variable.
@retval Others Errors Return errors from call to gRT->GetVariable.
**/
EFI_STATUS
EFIAPI
GetVariable3(
IN CONST CHAR16 *Name,
IN CONST EFI_GUID *Guid,
OUT VOID **Value,
OUT UINTN *Size OPTIONAL,
OUT UINT32 *Attr OPTIONAL
)
{
EFI_STATUS Status;
UINTN BufferSize;
ASSERT(Name != NULL && Guid != NULL && Value != NULL);
//
// Try to get the variable size.
//
BufferSize = 0;
*Value = NULL;
if (Size != NULL) {
*Size = 0;
}
if (Attr != NULL) {
*Attr = 0;
}
Status = gRT->GetVariable((CHAR16 *)Name, (EFI_GUID *)Guid, Attr, &BufferSize, *Value);
if (Status != EFI_BUFFER_TOO_SMALL) {
return Status;
}
//
// Allocate buffer to get the variable.
//
*Value = AllocatePool(BufferSize);
ASSERT(*Value != NULL);
if (*Value == NULL) {
return EFI_OUT_OF_RESOURCES;
}
//
// Get the variable data.
//
Status = gRT->GetVariable((CHAR16 *)Name, (EFI_GUID *)Guid, Attr, &BufferSize, *Value);
if (EFI_ERROR(Status)) {
FreePool(*Value);
*Value = NULL;
}
if (Size != NULL) {
*Size = BufferSize;
}
return Status;
}
/**
Returns a pointer to an allocated buffer that contains the contents of a
variable retrieved through the UEFI Runtime Service GetVariable(). This