diff --git a/PayloadPkg/FirmwareUpdate/FirmwareUpdate.c b/PayloadPkg/FirmwareUpdate/FirmwareUpdate.c index 6a522b5e..eb3e5599 100644 --- a/PayloadPkg/FirmwareUpdate/FirmwareUpdate.c +++ b/PayloadPkg/FirmwareUpdate/FirmwareUpdate.c @@ -1452,6 +1452,7 @@ InitFirmwareUpdate ( if (EFI_ERROR (Status)) { // // Error condition + FreePool(CapsuleImage); return Status; } diff --git a/PayloadPkg/FirmwareUpdate/FirmwareUpdateHelper.c b/PayloadPkg/FirmwareUpdate/FirmwareUpdateHelper.c index 403bdd14..e6f55696 100644 --- a/PayloadPkg/FirmwareUpdate/FirmwareUpdateHelper.c +++ b/PayloadPkg/FirmwareUpdate/FirmwareUpdateHelper.c @@ -577,6 +577,7 @@ UpdateSystemFirmware ( Status = VerifyFwStruct (ImageHdr); if (EFI_ERROR (Status)) { DEBUG ((DEBUG_ERROR, " VerifyFwStruct failed with Status = 0x%x\n", Status)); + FreePool(UpdatePartition); return Status; } @@ -655,6 +656,7 @@ UpdateSingleComponent ( DEBUG ((DEBUG_ERROR, "Updating component %4a failed with status = %r\n", (CHAR8 *)&CompName, Status)); } + FreePool(UpdatePartition); return Status; }