commit b857bf48133933a72f6edcd9bac3214799e093de
author Dandan Bi <dandan.bi@intel.com> Fri Nov 18 18:33:02 2016 +0800
committer Star Zeng <star.zeng@intel.com> Wed Nov 23 10:05:47 2016 +0800
tree 30f82007e0f2662a77a7edb0bb5768723c0345fb
parent 36e9e3e8ea0ce477504b6d2e21603ea94847efae

MdeModulePkg/DxeCapsuleLibFmp: Use AllocatePool for CapsuleResultVariable

Cc: Jiewen Yao <jiewen.yao@intel.com>
Contributed-under: TianoCore Contribution Agreement 1.0
Signed-off-by: Dandan Bi <dandan.bi@intel.com>
Reviewed-by: jiewen.yao@intel.com

MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleReportLib.c

diff --git a/MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleReportLib.c b/MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleReportLib.c
index a0ed2d0..2546871 100644
--- a/MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleReportLib.c
+++ b/MdeModulePkg/Library/DxeCapsuleLibFmp/DxeCapsuleReportLib.c
@@ -291,19 +291,26 @@
   IN EFI_FIRMWARE_MANAGEMENT_CAPSULE_IMAGE_HEADER  *ImageHeader

   )

 {

-  UINT8                               CapsuleResultVariable[sizeof(EFI_CAPSULE_RESULT_VARIABLE_HEADER) + sizeof(EFI_CAPSULE_RESULT_VARIABLE_FMP)];

   EFI_CAPSULE_RESULT_VARIABLE_HEADER  *CapsuleResultVariableHeader;

   EFI_CAPSULE_RESULT_VARIABLE_FMP     *CapsuleResultVariableFmp;

   EFI_STATUS                          Status;

+  UINT8                               *CapsuleResultVariable;

+  UINT32                              CapsuleResultVariableSize;

 

-  CapsuleResultVariableHeader = (VOID *)&CapsuleResultVariable[0];

-  CapsuleResultVariableHeader->VariableTotalSize = sizeof(CapsuleResultVariable);

+  CapsuleResultVariable     = NULL;

+  CapsuleResultVariableSize = sizeof(EFI_CAPSULE_RESULT_VARIABLE_HEADER) + sizeof(EFI_CAPSULE_RESULT_VARIABLE_FMP);

+  CapsuleResultVariable     = AllocatePool (CapsuleResultVariableSize);

+  if (CapsuleResultVariable == NULL) {

+    return EFI_OUT_OF_RESOURCES;

+  }

+  CapsuleResultVariableHeader = (VOID *)CapsuleResultVariable;

+  CapsuleResultVariableHeader->VariableTotalSize = CapsuleResultVariableSize;

   CopyGuid(&CapsuleResultVariableHeader->CapsuleGuid, &CapsuleHeader->CapsuleGuid);

   ZeroMem(&CapsuleResultVariableHeader->CapsuleProcessed, sizeof(CapsuleResultVariableHeader->CapsuleProcessed));

   gRT->GetTime(&CapsuleResultVariableHeader->CapsuleProcessed, NULL);

   CapsuleResultVariableHeader->CapsuleStatus = CapsuleStatus;

 

-  CapsuleResultVariableFmp = (VOID *)&CapsuleResultVariable[sizeof(EFI_CAPSULE_RESULT_VARIABLE_HEADER)];

+  CapsuleResultVariableFmp = (VOID *)(CapsuleResultVariable + sizeof(EFI_CAPSULE_RESULT_VARIABLE_HEADER));

   CapsuleResultVariableFmp->Version = 0x1;

   CapsuleResultVariableFmp->PayloadIndex = (UINT8)PayloadIndex;

   CapsuleResultVariableFmp->UpdateImageIndex = ImageHeader->UpdateImageIndex;

@@ -312,11 +319,12 @@
   //

   // Save Local Cache

   //

-  Status = WriteNewCapsuleResultVariableCache(&CapsuleResultVariable, sizeof(CapsuleResultVariable));

+  Status = WriteNewCapsuleResultVariableCache(CapsuleResultVariable, CapsuleResultVariableSize);

 

   if ((CapsuleHeader->Flags & CAPSULE_FLAGS_PERSIST_ACROSS_RESET) != 0) {

-    Status = WriteNewCapsuleResultVariable(&CapsuleResultVariable, sizeof(CapsuleResultVariable));

+    Status = WriteNewCapsuleResultVariable(CapsuleResultVariable, CapsuleResultVariableSize);

   }

+  FreePool (CapsuleResultVariable);

   return Status;

 }