| 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 |
| (...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 44 image->header_version = 1; | 44 image->header_version = 1; |
| 45 image->firmware_sign_algorithm = firmware_sign_algorithm; | 45 image->firmware_sign_algorithm = firmware_sign_algorithm; |
| 46 image->kernel_sign_algorithm = kernel_sign_algorithm; | 46 image->kernel_sign_algorithm = kernel_sign_algorithm; |
| 47 image->kernel_key_version = kernel_key_version; | 47 image->kernel_key_version = kernel_key_version; |
| 48 image->kernel_sign_key = (uint8_t*) Malloc( | 48 image->kernel_sign_key = (uint8_t*) Malloc( |
| 49 RSAProcessedKeySize(image->kernel_sign_algorithm)); | 49 RSAProcessedKeySize(image->kernel_sign_algorithm)); |
| 50 Memcpy(image->kernel_sign_key, kernel_sign_key, | 50 Memcpy(image->kernel_sign_key, kernel_sign_key, |
| 51 RSAProcessedKeySize(image->kernel_sign_algorithm)); | 51 RSAProcessedKeySize(image->kernel_sign_algorithm)); |
| 52 | 52 |
| 53 /* Update correct header length. */ | 53 /* Update correct header length. */ |
| 54 image->header_len = (sizeof(image->header_version) + | 54 image->header_len = GetKernelHeaderLen(image); |
| 55 sizeof(image->header_len) + | |
| 56 sizeof(image->firmware_sign_algorithm) + | |
| 57 sizeof(image->kernel_sign_algorithm) + | |
| 58 RSAProcessedKeySize(image->kernel_sign_algorithm) + | |
| 59 sizeof(image->kernel_key_version) + | |
| 60 sizeof(image->header_checksum)); | |
| 61 | 55 |
| 62 /* Calculate SHA-512 digest on header and populate header_checksum. */ | 56 /* Calculate SHA-512 digest on header and populate header_checksum. */ |
| 63 DigestInit(&ctx, SHA512_DIGEST_ALGORITHM); | 57 DigestInit(&ctx, SHA512_DIGEST_ALGORITHM); |
| 64 DigestUpdate(&ctx, (uint8_t*) &image->header_version, | 58 DigestUpdate(&ctx, (uint8_t*) &image->header_version, |
| 65 sizeof(image->header_version)); | 59 sizeof(image->header_version)); |
| 66 DigestUpdate(&ctx, (uint8_t*) &image->header_len, | 60 DigestUpdate(&ctx, (uint8_t*) &image->header_len, |
| 67 sizeof(image->header_len)); | 61 sizeof(image->header_len)); |
| 68 DigestUpdate(&ctx, (uint8_t*) &image->firmware_sign_algorithm, | 62 DigestUpdate(&ctx, (uint8_t*) &image->firmware_sign_algorithm, |
| 69 sizeof(image->firmware_sign_algorithm)); | 63 sizeof(image->firmware_sign_algorithm)); |
| 70 DigestUpdate(&ctx, (uint8_t*) &image->kernel_sign_algorithm, | 64 DigestUpdate(&ctx, (uint8_t*) &image->kernel_sign_algorithm, |
| 71 sizeof(image->kernel_sign_algorithm)); | 65 sizeof(image->kernel_sign_algorithm)); |
| 72 DigestUpdate(&ctx, (uint8_t*) &image->kernel_key_version, | 66 DigestUpdate(&ctx, (uint8_t*) &image->kernel_key_version, |
| 73 sizeof(image->kernel_key_version)); | 67 sizeof(image->kernel_key_version)); |
| 74 DigestUpdate(&ctx, image->kernel_sign_key, | 68 DigestUpdate(&ctx, image->kernel_sign_key, |
| 75 RSAProcessedKeySize(image->kernel_sign_algorithm)); | 69 RSAProcessedKeySize(image->kernel_sign_algorithm)); |
| 76 header_checksum = DigestFinal(&ctx); | 70 header_checksum = DigestFinal(&ctx); |
| 77 Memcpy(image->header_checksum, header_checksum, SHA512_DIGEST_SIZE); | 71 Memcpy(image->header_checksum, header_checksum, SHA512_DIGEST_SIZE); |
| 78 Free(header_checksum); | 72 Free(header_checksum); |
| 79 | 73 |
| 80 /* Populate kernel options and data with dummy data. */ | 74 /* Populate kernel options and data with dummy data. */ |
| 81 image->kernel_version = kernel_version; | 75 image->kernel_version = kernel_version; |
| 82 image->options.version[0] = 1; | 76 image->options.version[0] = 1; |
| 83 image->options.version[1] = 1; | 77 image->options.version[1] = 0; |
| 78 Memset(image->options.cmd_line, 0, sizeof(image->options.cmd_line)); |
| 84 image->options.kernel_len = kernel_len; | 79 image->options.kernel_len = kernel_len; |
| 85 image->options.kernel_load_addr = 0; | 80 image->options.kernel_load_addr = 0; |
| 86 image->options.kernel_entry_addr = 0; | 81 image->options.kernel_entry_addr = 0; |
| 87 image->kernel_key_signature = image->kernel_signature = NULL; | 82 image->kernel_key_signature = image->kernel_signature = NULL; |
| 88 image->kernel_data = Malloc(kernel_len); | 83 image->kernel_data = Malloc(kernel_len); |
| 89 Memset(image->kernel_data, 'F', kernel_len); | 84 Memset(image->kernel_data, 'F', kernel_len); |
| 90 | 85 |
| 91 return image; | 86 return image; |
| 92 } | 87 } |
| 93 | 88 |
| (...skipping 143 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 237 | 232 |
| 238 failure: | 233 failure: |
| 239 Free(kernel_blob); | 234 Free(kernel_blob); |
| 240 KernelImageFree(image); | 235 KernelImageFree(image); |
| 241 Free(kernel_sign_key_buf); | 236 Free(kernel_sign_key_buf); |
| 242 Free(firmware_key_blob); | 237 Free(firmware_key_blob); |
| 243 RSAPublicKeyFree(firmware_key); | 238 RSAPublicKeyFree(firmware_key); |
| 244 | 239 |
| 245 return error_code; | 240 return error_code; |
| 246 } | 241 } |
| OLD | NEW |