diff options
-rw-r--r-- | firmware/lib/vboot_api_kernel.c | 6 | ||||
-rw-r--r-- | firmware/lib/vboot_kernel.c | 6 |
2 files changed, 4 insertions, 8 deletions
diff --git a/firmware/lib/vboot_api_kernel.c b/firmware/lib/vboot_api_kernel.c index 6f896fbf..53f8b0ed 100644 --- a/firmware/lib/vboot_api_kernel.c +++ b/firmware/lib/vboot_api_kernel.c @@ -1403,12 +1403,10 @@ VbError_t VbVerifyMemoryBootImage(VbCommonParams *cparams, retval = VBERROR_INVALID_KERNEL_FOUND; /* Allocate work buffer */ - workbuf = (uint8_t *) - VbExMalloc(VB2_VERIFY_KERNEL_PREAMBLE_WORKBUF_BYTES); + workbuf = (uint8_t *)VbExMalloc(VB2_KERNEL_WORKBUF_RECOMMENDED_SIZE); if (!workbuf) goto fail; - vb2_workbuf_init(&wb, workbuf, - VB2_VERIFY_KERNEL_PREAMBLE_WORKBUF_BYTES); + vb2_workbuf_init(&wb, workbuf, VB2_KERNEL_WORKBUF_RECOMMENDED_SIZE); /* Verify the key block. */ key_block = (VbKeyBlockHeader *)kbuf; diff --git a/firmware/lib/vboot_kernel.c b/firmware/lib/vboot_kernel.c index cf6f92bc..4e0d3e48 100644 --- a/firmware/lib/vboot_kernel.c +++ b/firmware/lib/vboot_kernel.c @@ -152,12 +152,10 @@ VbError_t LoadKernel(LoadKernelParams *params, VbCommonParams *cparams) goto bad_gpt; /* Allocate work buffer */ - workbuf = (uint8_t *) - VbExMalloc(VB2_VERIFY_KERNEL_PREAMBLE_WORKBUF_BYTES); + workbuf = (uint8_t *)VbExMalloc(VB2_KERNEL_WORKBUF_RECOMMENDED_SIZE); if (!workbuf) goto bad_gpt; - vb2_workbuf_init(&wb, workbuf, - VB2_VERIFY_KERNEL_PREAMBLE_WORKBUF_BYTES); + vb2_workbuf_init(&wb, workbuf, VB2_KERNEL_WORKBUF_RECOMMENDED_SIZE); /* Unpack kernel subkey */ struct vb2_public_key kernel_subkey2; |