| 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 * Common functions used by tests. | 5 * Common functions used by tests. |
| 6 */ | 6 */ |
| 7 | 7 |
| 8 #include "test_common.h" | 8 #include "test_common.h" |
| 9 | 9 |
| 10 #include <stdio.h> | 10 #include <stdio.h> |
| (...skipping 19 matching lines...) Expand all Loading... |
| 30 fprintf(stderr, "%s Test " COL_RED "FAILED\n" COL_STOP, testname); | 30 fprintf(stderr, "%s Test " COL_RED "FAILED\n" COL_STOP, testname); |
| 31 gTestSuccess = 0; | 31 gTestSuccess = 0; |
| 32 return 0; | 32 return 0; |
| 33 } | 33 } |
| 34 } | 34 } |
| 35 | 35 |
| 36 FirmwareImage* GenerateTestFirmwareImage(int algorithm, | 36 FirmwareImage* GenerateTestFirmwareImage(int algorithm, |
| 37 const uint8_t* firmware_sign_key, | 37 const uint8_t* firmware_sign_key, |
| 38 int firmware_key_version, | 38 int firmware_key_version, |
| 39 int firmware_version, | 39 int firmware_version, |
| 40 int firmware_len, | 40 uint64_t firmware_len, |
| 41 const char* root_key_file, | 41 const char* root_key_file, |
| 42 const char* firmware_key_file, | 42 const char* firmware_key_file, |
| 43 uint8_t firmware_data_fill_char) { | 43 uint8_t firmware_data_fill_char) { |
| 44 FirmwareImage* image = FirmwareImageNew(); | 44 FirmwareImage* image = FirmwareImageNew(); |
| 45 | 45 |
| 46 Memcpy(image->magic, FIRMWARE_MAGIC, FIRMWARE_MAGIC_SIZE); | 46 Memcpy(image->magic, FIRMWARE_MAGIC, FIRMWARE_MAGIC_SIZE); |
| 47 image->firmware_sign_algorithm = algorithm; | 47 image->firmware_sign_algorithm = algorithm; |
| 48 image->firmware_sign_key = (uint8_t*) Malloc( | 48 image->firmware_sign_key = (uint8_t*) Malloc( |
| 49 RSAProcessedKeySize(image->firmware_sign_algorithm)); | 49 RSAProcessedKeySize(image->firmware_sign_algorithm)); |
| 50 Memcpy(image->firmware_sign_key, firmware_sign_key, | 50 Memcpy(image->firmware_sign_key, firmware_sign_key, |
| (...skipping 26 matching lines...) Expand all Loading... |
| 77 FirmwareImageFree(image); | 77 FirmwareImageFree(image); |
| 78 return NULL; | 78 return NULL; |
| 79 } | 79 } |
| 80 return image; | 80 return image; |
| 81 } | 81 } |
| 82 | 82 |
| 83 uint8_t* GenerateTestFirmwareBlob(int algorithm, | 83 uint8_t* GenerateTestFirmwareBlob(int algorithm, |
| 84 const uint8_t* firmware_sign_key, | 84 const uint8_t* firmware_sign_key, |
| 85 int firmware_key_version, | 85 int firmware_key_version, |
| 86 int firmware_version, | 86 int firmware_version, |
| 87 int firmware_len, | 87 uint64_t firmware_len, |
| 88 const char* root_key_file, | 88 const char* root_key_file, |
| 89 const char* firmware_key_file) { | 89 const char* firmware_key_file) { |
| 90 FirmwareImage* image = NULL; | 90 FirmwareImage* image = NULL; |
| 91 uint8_t* firmware_blob = NULL; | 91 uint8_t* firmware_blob = NULL; |
| 92 uint64_t firmware_blob_len = 0; | 92 uint64_t firmware_blob_len = 0; |
| 93 | 93 |
| 94 image = GenerateTestFirmwareImage(algorithm, | 94 image = GenerateTestFirmwareImage(algorithm, |
| 95 firmware_sign_key, | 95 firmware_sign_key, |
| 96 firmware_key_version, | 96 firmware_key_version, |
| 97 firmware_version, | 97 firmware_version, |
| (...skipping 37 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 135 FirmwareImageFree(image); | 135 FirmwareImageFree(image); |
| 136 return firmware_blob; | 136 return firmware_blob; |
| 137 } | 137 } |
| 138 | 138 |
| 139 | 139 |
| 140 KernelImage* GenerateTestKernelImage(int firmware_sign_algorithm, | 140 KernelImage* GenerateTestKernelImage(int firmware_sign_algorithm, |
| 141 int kernel_sign_algorithm, | 141 int kernel_sign_algorithm, |
| 142 const uint8_t* kernel_sign_key, | 142 const uint8_t* kernel_sign_key, |
| 143 int kernel_key_version, | 143 int kernel_key_version, |
| 144 int kernel_version, | 144 int kernel_version, |
| 145 int kernel_len, | 145 uint64_t kernel_len, |
| 146 const char* firmware_key_file, | 146 const char* firmware_key_file, |
| 147 const char* kernel_key_file, | 147 const char* kernel_key_file, |
| 148 uint8_t kernel_data_fill_char) { | 148 uint8_t kernel_data_fill_char) { |
| 149 KernelImage* image = KernelImageNew(); | 149 KernelImage* image = KernelImageNew(); |
| 150 | 150 |
| 151 Memcpy(image->magic, KERNEL_MAGIC, KERNEL_MAGIC_SIZE); | 151 Memcpy(image->magic, KERNEL_MAGIC, KERNEL_MAGIC_SIZE); |
| 152 image->header_version = 1; | 152 image->header_version = 1; |
| 153 image->firmware_sign_algorithm = firmware_sign_algorithm; | 153 image->firmware_sign_algorithm = firmware_sign_algorithm; |
| 154 image->kernel_sign_algorithm = kernel_sign_algorithm; | 154 image->kernel_sign_algorithm = kernel_sign_algorithm; |
| 155 image->kernel_key_version = kernel_key_version; | 155 image->kernel_key_version = kernel_key_version; |
| (...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 190 } | 190 } |
| 191 | 191 |
| 192 return image; | 192 return image; |
| 193 } | 193 } |
| 194 | 194 |
| 195 uint8_t* GenerateTestKernelBlob(int firmware_sign_algorithm, | 195 uint8_t* GenerateTestKernelBlob(int firmware_sign_algorithm, |
| 196 int kernel_sign_algorithm, | 196 int kernel_sign_algorithm, |
| 197 const uint8_t* kernel_sign_key, | 197 const uint8_t* kernel_sign_key, |
| 198 int kernel_key_version, | 198 int kernel_key_version, |
| 199 int kernel_version, | 199 int kernel_version, |
| 200 int kernel_len, | 200 uint64_t kernel_len, |
| 201 const char* firmware_key_file, | 201 const char* firmware_key_file, |
| 202 const char* kernel_key_file) { | 202 const char* kernel_key_file) { |
| 203 KernelImage* image = NULL; | 203 KernelImage* image = NULL; |
| 204 uint8_t* kernel_blob = NULL; | 204 uint8_t* kernel_blob = NULL; |
| 205 uint64_t kernel_blob_len = 0; | 205 uint64_t kernel_blob_len = 0; |
| 206 | 206 |
| 207 image = GenerateTestKernelImage(firmware_sign_algorithm, | 207 image = GenerateTestKernelImage(firmware_sign_algorithm, |
| 208 kernel_sign_algorithm, | 208 kernel_sign_algorithm, |
| 209 kernel_sign_key, | 209 kernel_sign_key, |
| 210 kernel_key_version, | 210 kernel_key_version, |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 242 if (!image) | 242 if (!image) |
| 243 return NULL; | 243 return NULL; |
| 244 if (is_corrupt) { | 244 if (is_corrupt) { |
| 245 /* Invalidate image. */ | 245 /* Invalidate image. */ |
| 246 Memset(image->kernel_data, 'X', image->options.kernel_len); | 246 Memset(image->kernel_data, 'X', image->options.kernel_len); |
| 247 } | 247 } |
| 248 kernel_blob = GetKernelBlob(image, &len); | 248 kernel_blob = GetKernelBlob(image, &len); |
| 249 KernelImageFree(image); | 249 KernelImageFree(image); |
| 250 return kernel_blob; | 250 return kernel_blob; |
| 251 } | 251 } |
| OLD | NEW |