| OLD | NEW |
| 1 /* Copyright (c) 2010 The Chromium OS Authors. All rights reserved. | 1 /* Copyright (c) 2010 The Chromium OS Authors. All rights reserved. |
| 2 * Use of this source code is governed by a BSD-style license that can be | 2 * Use of this source code is governed by a BSD-style license that can be |
| 3 * found in the LICENSE file. | 3 * found in the LICENSE file. |
| 4 * | 4 * |
| 5 * Tests for kernel image library. | 5 * Tests for kernel image library. |
| 6 */ | 6 */ |
| 7 | 7 |
| 8 #include <stdio.h> | 8 #include <stdio.h> |
| 9 #include <stdlib.h> | 9 #include <stdlib.h> |
| 10 | 10 |
| 11 #include "file_keys.h" | 11 #include "file_keys.h" |
| 12 #include "kernel_image.h" | 12 #include "kernel_image.h" |
| 13 #include "rsa_utility.h" | 13 #include "rsa_utility.h" |
| 14 #include "sha_utility.h" | |
| 15 #include "utility.h" | 14 #include "utility.h" |
| 16 | 15 |
| 17 /* ANSI Color coding sequences. */ | 16 /* ANSI Color coding sequences. */ |
| 18 #define COL_GREEN "\e[1;32m" | 17 #define COL_GREEN "\e[1;32m" |
| 19 #define COL_RED "\e[0;31m" | 18 #define COL_RED "\e[0;31m" |
| 20 #define COL_STOP "\e[m" | 19 #define COL_STOP "\e[m" |
| 21 | 20 |
| 22 int TEST_EQ(int result, int expected_result, char* testname) { | 21 int TEST_EQ(int result, int expected_result, char* testname) { |
| 23 if (result == expected_result) { | 22 if (result == expected_result) { |
| 24 fprintf(stderr, "%s Test " COL_GREEN " PASSED\n" COL_STOP, testname); | 23 fprintf(stderr, "%s Test " COL_GREEN " PASSED\n" COL_STOP, testname); |
| 25 return 1; | 24 return 1; |
| 26 } | 25 } |
| 27 else { | 26 else { |
| 28 fprintf(stderr, "%s Test " COL_RED " FAILED\n" COL_STOP, testname); | 27 fprintf(stderr, "%s Test " COL_RED " FAILED\n" COL_STOP, testname); |
| 29 return 0; | 28 return 0; |
| 30 } | 29 } |
| 31 } | 30 } |
| 32 | 31 |
| 33 KernelImage* GenerateTestKernelImage(int firmware_sign_algorithm, | 32 KernelImage* GenerateTestKernelImage(int firmware_sign_algorithm, |
| 34 int kernel_sign_algorithm, | 33 int kernel_sign_algorithm, |
| 35 uint8_t* kernel_sign_key, | 34 uint8_t* kernel_sign_key, |
| 36 int kernel_key_version, | 35 int kernel_key_version, |
| 37 int kernel_version, | 36 int kernel_version, |
| 38 int kernel_len) { | 37 int kernel_len) { |
| 39 KernelImage* image = KernelImageNew(); | 38 KernelImage* image = KernelImageNew(); |
| 40 uint8_t* header_checksum; | |
| 41 DigestContext ctx; | |
| 42 | 39 |
| 43 Memcpy(image->magic, KERNEL_MAGIC, KERNEL_MAGIC_SIZE); | 40 Memcpy(image->magic, KERNEL_MAGIC, KERNEL_MAGIC_SIZE); |
| 44 image->header_version = 1; | 41 image->header_version = 1; |
| 45 image->firmware_sign_algorithm = firmware_sign_algorithm; | 42 image->firmware_sign_algorithm = firmware_sign_algorithm; |
| 46 image->kernel_sign_algorithm = kernel_sign_algorithm; | 43 image->kernel_sign_algorithm = kernel_sign_algorithm; |
| 47 image->kernel_key_version = kernel_key_version; | 44 image->kernel_key_version = kernel_key_version; |
| 48 image->kernel_sign_key = (uint8_t*) Malloc( | 45 image->kernel_sign_key = (uint8_t*) Malloc( |
| 49 RSAProcessedKeySize(image->kernel_sign_algorithm)); | 46 RSAProcessedKeySize(image->kernel_sign_algorithm)); |
| 50 Memcpy(image->kernel_sign_key, kernel_sign_key, | 47 Memcpy(image->kernel_sign_key, kernel_sign_key, |
| 51 RSAProcessedKeySize(image->kernel_sign_algorithm)); | 48 RSAProcessedKeySize(image->kernel_sign_algorithm)); |
| 52 | 49 |
| 53 /* Update correct header length. */ | 50 /* Update correct header length. */ |
| 54 image->header_len = GetKernelHeaderLen(image); | 51 image->header_len = GetKernelHeaderLen(image); |
| 55 | 52 |
| 56 /* Calculate SHA-512 digest on header and populate header_checksum. */ | 53 /* Calculate SHA-512 digest on header and populate header_checksum. */ |
| 57 DigestInit(&ctx, SHA512_DIGEST_ALGORITHM); | 54 CalculateKernelHeaderChecksum(image, image->header_checksum); |
| 58 DigestUpdate(&ctx, (uint8_t*) &image->header_version, | |
| 59 sizeof(image->header_version)); | |
| 60 DigestUpdate(&ctx, (uint8_t*) &image->header_len, | |
| 61 sizeof(image->header_len)); | |
| 62 DigestUpdate(&ctx, (uint8_t*) &image->firmware_sign_algorithm, | |
| 63 sizeof(image->firmware_sign_algorithm)); | |
| 64 DigestUpdate(&ctx, (uint8_t*) &image->kernel_sign_algorithm, | |
| 65 sizeof(image->kernel_sign_algorithm)); | |
| 66 DigestUpdate(&ctx, (uint8_t*) &image->kernel_key_version, | |
| 67 sizeof(image->kernel_key_version)); | |
| 68 DigestUpdate(&ctx, image->kernel_sign_key, | |
| 69 RSAProcessedKeySize(image->kernel_sign_algorithm)); | |
| 70 header_checksum = DigestFinal(&ctx); | |
| 71 Memcpy(image->header_checksum, header_checksum, SHA512_DIGEST_SIZE); | |
| 72 Free(header_checksum); | |
| 73 | 55 |
| 74 /* Populate kernel options and data with dummy data. */ | 56 /* Populate kernel options and data with dummy data. */ |
| 75 image->kernel_version = kernel_version; | 57 image->kernel_version = kernel_version; |
| 76 image->options.version[0] = 1; | 58 image->options.version[0] = 1; |
| 77 image->options.version[1] = 0; | 59 image->options.version[1] = 0; |
| 78 Memset(image->options.cmd_line, 0, sizeof(image->options.cmd_line)); | 60 Memset(image->options.cmd_line, 0, sizeof(image->options.cmd_line)); |
| 79 image->options.kernel_len = kernel_len; | 61 image->options.kernel_len = kernel_len; |
| 80 image->options.kernel_load_addr = 0; | 62 image->options.kernel_load_addr = 0; |
| 81 image->options.kernel_entry_addr = 0; | 63 image->options.kernel_entry_addr = 0; |
| 82 image->kernel_key_signature = image->kernel_signature = NULL; | 64 image->kernel_key_signature = image->kernel_signature = NULL; |
| (...skipping 149 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 232 | 214 |
| 233 failure: | 215 failure: |
| 234 Free(kernel_blob); | 216 Free(kernel_blob); |
| 235 KernelImageFree(image); | 217 KernelImageFree(image); |
| 236 Free(kernel_sign_key_buf); | 218 Free(kernel_sign_key_buf); |
| 237 Free(firmware_key_blob); | 219 Free(firmware_key_blob); |
| 238 RSAPublicKeyFree(firmware_key); | 220 RSAPublicKeyFree(firmware_key); |
| 239 | 221 |
| 240 return error_code; | 222 return error_code; |
| 241 } | 223 } |
| OLD | NEW |