MdeModulePkg/CapsuleLib: Follow UEFI 22.2.3 to process FMP.
Previous logic does not follow UEFI 22.2.3 to process FMP strictly. It may cause FMP image not be processed in some corner case. The updated logic follows UEFI 22.2.3. The way to check if a capsule is processed is also simplified. The function - ProcessFmpCapsuleImage() is too big, so that we created sub-functions - StartFmpImage(), DumpAllFmpInfo(), GetFmpHandleBufferByType(), SetFmpImageData(), RecordFmpCapsuleStatus() to improve the readability. The function - ProcessTheseCapsules() is too big, so that we created sub-functions - InitCapsulePtr(), AreAllImagesProcessed(), PopulateCapsuleInConfigurationTable() to improve the readability. Cc: Feng Tian <feng.tian@intel.com> Cc: Star Zeng <star.zeng@intel.com> Cc: Chao Zhang <chao.b.zhang@intel.com> Contributed-under: TianoCore Contribution Agreement 1.0 Signed-off-by: Jiewen Yao <jiewen.yao@intel.com> Reviewed-by: Star Zeng <star.zeng@intel.com>
This commit is contained in:
@@ -35,17 +35,6 @@
|
||||
|
||||
#include <IndustryStandard/WindowsUxCapsule.h>
|
||||
|
||||
typedef struct {
|
||||
EFI_CAPSULE_RESULT_VARIABLE_HEADER CapsuleResultHeader;
|
||||
EFI_CAPSULE_RESULT_VARIABLE_FMP CapsuleResultFmp;
|
||||
} CAPSULE_RESULT_VARIABLE_CACHE;
|
||||
|
||||
#define CAPSULE_RESULT_VARIABLE_CACHE_COUNT 0x10
|
||||
|
||||
CAPSULE_RESULT_VARIABLE_CACHE *mCapsuleResultVariableCache;
|
||||
UINTN mCapsuleResultVariableCacheMaxCount;
|
||||
UINTN mCapsuleResultVariableCacheCount;
|
||||
|
||||
/**
|
||||
Get current capsule last variable index.
|
||||
|
||||
@@ -77,92 +66,6 @@ GetCurrentCapsuleLastIndex (
|
||||
return CurrentIndex;
|
||||
}
|
||||
|
||||
/**
|
||||
Check if this FMP capsule is processed.
|
||||
|
||||
@param[in] CapsuleHeader The capsule image header
|
||||
@param[in] PayloadIndex FMP payload index
|
||||
@param[in] ImageHeader FMP image header
|
||||
|
||||
@retval TRUE This FMP capsule is processed.
|
||||
@retval FALSE This FMP capsule is not processed.
|
||||
**/
|
||||
BOOLEAN
|
||||
IsFmpCapsuleProcessed (
|
||||
IN EFI_CAPSULE_HEADER *CapsuleHeader,
|
||||
IN UINTN PayloadIndex,
|
||||
IN EFI_FIRMWARE_MANAGEMENT_CAPSULE_IMAGE_HEADER *ImageHeader
|
||||
)
|
||||
{
|
||||
UINTN Index;
|
||||
EFI_CAPSULE_RESULT_VARIABLE_HEADER *CapsuleResult;
|
||||
EFI_CAPSULE_RESULT_VARIABLE_FMP *CapsuleResultFmp;
|
||||
|
||||
for (Index = 0; Index < mCapsuleResultVariableCacheCount; Index++) {
|
||||
//
|
||||
// Check
|
||||
//
|
||||
CapsuleResult = &mCapsuleResultVariableCache[Index].CapsuleResultHeader;
|
||||
if (CapsuleResult->VariableTotalSize >= sizeof(EFI_CAPSULE_RESULT_VARIABLE_HEADER)) {
|
||||
if (CompareGuid(&CapsuleResult->CapsuleGuid, &gEfiFmpCapsuleGuid)) {
|
||||
if (CapsuleResult->VariableTotalSize >= sizeof(EFI_CAPSULE_RESULT_VARIABLE_HEADER) + sizeof(EFI_CAPSULE_RESULT_VARIABLE_FMP) + sizeof(CHAR16) * 2) {
|
||||
CapsuleResultFmp = (EFI_CAPSULE_RESULT_VARIABLE_FMP *)(CapsuleResult + 1);
|
||||
if (CompareGuid(&CapsuleResultFmp->UpdateImageTypeId, &ImageHeader->UpdateImageTypeId) &&
|
||||
(CapsuleResultFmp->UpdateImageIndex == ImageHeader->UpdateImageIndex) &&
|
||||
(CapsuleResultFmp->PayloadIndex == PayloadIndex) ) {
|
||||
return TRUE;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
/**
|
||||
Write a new capsule status variable cache.
|
||||
|
||||
@param[in] CapsuleResult The capsule status variable
|
||||
@param[in] CapsuleResultSize The size of the capsule stauts variable in bytes
|
||||
|
||||
@retval EFI_SUCCESS The capsule status variable is cached.
|
||||
@retval EFI_OUT_OF_RESOURCES No resource to cache the capsule status variable.
|
||||
**/
|
||||
EFI_STATUS
|
||||
WriteNewCapsuleResultVariableCache (
|
||||
IN VOID *CapsuleResult,
|
||||
IN UINTN CapsuleResultSize
|
||||
)
|
||||
{
|
||||
if (CapsuleResultSize > sizeof(CAPSULE_RESULT_VARIABLE_CACHE)) {
|
||||
CapsuleResultSize = sizeof(CAPSULE_RESULT_VARIABLE_CACHE);
|
||||
}
|
||||
|
||||
if (mCapsuleResultVariableCacheCount == mCapsuleResultVariableCacheMaxCount) {
|
||||
mCapsuleResultVariableCache = ReallocatePool(
|
||||
mCapsuleResultVariableCacheMaxCount * sizeof(CAPSULE_RESULT_VARIABLE_CACHE),
|
||||
(mCapsuleResultVariableCacheMaxCount + CAPSULE_RESULT_VARIABLE_CACHE_COUNT) * sizeof(CAPSULE_RESULT_VARIABLE_CACHE),
|
||||
mCapsuleResultVariableCache
|
||||
);
|
||||
if (mCapsuleResultVariableCache == NULL) {
|
||||
return EFI_OUT_OF_RESOURCES;
|
||||
}
|
||||
mCapsuleResultVariableCacheMaxCount += CAPSULE_RESULT_VARIABLE_CACHE_COUNT;
|
||||
}
|
||||
|
||||
ASSERT(mCapsuleResultVariableCacheCount < mCapsuleResultVariableCacheMaxCount);
|
||||
ASSERT(mCapsuleResultVariableCache != NULL);
|
||||
CopyMem(
|
||||
&mCapsuleResultVariableCache[mCapsuleResultVariableCacheCount],
|
||||
CapsuleResult,
|
||||
CapsuleResultSize
|
||||
);
|
||||
mCapsuleResultVariableCacheCount++;
|
||||
|
||||
return EFI_SUCCESS;
|
||||
}
|
||||
|
||||
/**
|
||||
Get a new capsule status variable index.
|
||||
|
||||
@@ -262,11 +165,7 @@ RecordCapsuleStatusVariable (
|
||||
gRT->GetTime(&CapsuleResultVariable.CapsuleProcessed, NULL);
|
||||
CapsuleResultVariable.CapsuleStatus = CapsuleStatus;
|
||||
|
||||
//
|
||||
// Save Local Cache
|
||||
//
|
||||
Status = WriteNewCapsuleResultVariableCache(&CapsuleResultVariable, sizeof(CapsuleResultVariable));
|
||||
|
||||
Status = EFI_SUCCESS;
|
||||
if ((CapsuleHeader->Flags & CAPSULE_FLAGS_PERSIST_ACROSS_RESET) != 0) {
|
||||
Status = WriteNewCapsuleResultVariable(&CapsuleResultVariable, sizeof(CapsuleResultVariable));
|
||||
}
|
||||
@@ -338,11 +237,7 @@ RecordFmpCapsuleStatusVariable (
|
||||
DevicePathStr = NULL;
|
||||
}
|
||||
|
||||
//
|
||||
// Save Local Cache
|
||||
//
|
||||
Status = WriteNewCapsuleResultVariableCache(CapsuleResultVariable, CapsuleResultVariableSize);
|
||||
|
||||
Status = EFI_SUCCESS;
|
||||
if ((CapsuleHeader->Flags & CAPSULE_FLAGS_PERSIST_ACROSS_RESET) != 0) {
|
||||
Status = WriteNewCapsuleResultVariable(CapsuleResultVariable, CapsuleResultVariableSize);
|
||||
}
|
||||
|
Reference in New Issue
Block a user