| Index: src/platform/vboot_reference/tests/kernel_image_tests.c
|
| diff --git a/src/platform/vboot_reference/tests/kernel_image_tests.c b/src/platform/vboot_reference/tests/kernel_image_tests.c
|
| index 3cf9c4ce0ce312f2bf493f681dd1b3808542cd00..bad432936a5200b355a06facaad5c33a0e86b6be 100644
|
| --- a/src/platform/vboot_reference/tests/kernel_image_tests.c
|
| +++ b/src/platform/vboot_reference/tests/kernel_image_tests.c
|
| @@ -11,7 +11,6 @@
|
| #include "file_keys.h"
|
| #include "kernel_image.h"
|
| #include "rsa_utility.h"
|
| -#include "sha_utility.h"
|
| #include "utility.h"
|
|
|
| /* ANSI Color coding sequences. */
|
| @@ -37,8 +36,6 @@ KernelImage* GenerateTestKernelImage(int firmware_sign_algorithm,
|
| int kernel_version,
|
| int kernel_len) {
|
| KernelImage* image = KernelImageNew();
|
| - uint8_t* header_checksum;
|
| - DigestContext ctx;
|
|
|
| Memcpy(image->magic, KERNEL_MAGIC, KERNEL_MAGIC_SIZE);
|
| image->header_version = 1;
|
| @@ -54,22 +51,7 @@ KernelImage* GenerateTestKernelImage(int firmware_sign_algorithm,
|
| image->header_len = GetKernelHeaderLen(image);
|
|
|
| /* Calculate SHA-512 digest on header and populate header_checksum. */
|
| - DigestInit(&ctx, SHA512_DIGEST_ALGORITHM);
|
| - DigestUpdate(&ctx, (uint8_t*) &image->header_version,
|
| - sizeof(image->header_version));
|
| - DigestUpdate(&ctx, (uint8_t*) &image->header_len,
|
| - sizeof(image->header_len));
|
| - DigestUpdate(&ctx, (uint8_t*) &image->firmware_sign_algorithm,
|
| - sizeof(image->firmware_sign_algorithm));
|
| - DigestUpdate(&ctx, (uint8_t*) &image->kernel_sign_algorithm,
|
| - sizeof(image->kernel_sign_algorithm));
|
| - DigestUpdate(&ctx, (uint8_t*) &image->kernel_key_version,
|
| - sizeof(image->kernel_key_version));
|
| - DigestUpdate(&ctx, image->kernel_sign_key,
|
| - RSAProcessedKeySize(image->kernel_sign_algorithm));
|
| - header_checksum = DigestFinal(&ctx);
|
| - Memcpy(image->header_checksum, header_checksum, SHA512_DIGEST_SIZE);
|
| - Free(header_checksum);
|
| + CalculateKernelHeaderChecksum(image, image->header_checksum);
|
|
|
| /* Populate kernel options and data with dummy data. */
|
| image->kernel_version = kernel_version;
|
|
|