| 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 // Utility for manipulating verified boot kernel images. | 5 // Utility for manipulating verified boot kernel images. |
| 6 // | 6 // |
| 7 | 7 |
| 8 #include "kernel_utility.h" | 8 #include "kernel_utility.h" |
| 9 | 9 |
| 10 #include <errno.h> | 10 #include <errno.h> |
| (...skipping 18 matching lines...) Expand all Loading... |
| 29 namespace vboot_reference { | 29 namespace vboot_reference { |
| 30 | 30 |
| 31 KernelUtility::KernelUtility(): image_(NULL), | 31 KernelUtility::KernelUtility(): image_(NULL), |
| 32 firmware_key_pub_(NULL), | 32 firmware_key_pub_(NULL), |
| 33 header_version_(1), | 33 header_version_(1), |
| 34 firmware_sign_algorithm_(-1), | 34 firmware_sign_algorithm_(-1), |
| 35 kernel_sign_algorithm_(-1), | 35 kernel_sign_algorithm_(-1), |
| 36 kernel_key_version_(-1), | 36 kernel_key_version_(-1), |
| 37 kernel_version_(-1), | 37 kernel_version_(-1), |
| 38 kernel_len_(0), | 38 kernel_len_(0), |
| 39 kernel_config_(NULL), | |
| 40 is_generate_(false), | 39 is_generate_(false), |
| 41 is_verify_(false), | 40 is_verify_(false), |
| 42 is_describe_(false), | 41 is_describe_(false), |
| 43 is_only_vblock_(false) { | 42 is_only_vblock_(false) { |
| 44 } | 43 } |
| 45 | 44 |
| 46 KernelUtility::~KernelUtility() { | 45 KernelUtility::~KernelUtility() { |
| 47 Free(kernel_config_); | |
| 48 RSAPublicKeyFree(firmware_key_pub_); | 46 RSAPublicKeyFree(firmware_key_pub_); |
| 49 KernelImageFree(image_); | 47 KernelImageFree(image_); |
| 50 } | 48 } |
| 51 | 49 |
| 52 void KernelUtility::PrintUsage(void) { | 50 void KernelUtility::PrintUsage(void) { |
| 53 cerr << | 51 cerr << |
| 54 "Utility to generate/verify/describe a verified boot kernel image\n\n" | 52 "Utility to generate/verify/describe a verified boot kernel image\n\n" |
| 55 "Usage: kernel_utility <--generate|--verify|--describe> [OPTIONS]\n\n" | 53 "Usage: kernel_utility <--generate|--verify|--describe> [OPTIONS]\n\n" |
| 56 "For \"--verify\", required OPTIONS are:\n" | 54 "For \"--verify\", required OPTIONS are:\n" |
| 57 "--in <infile>\t\t\tVerified boot kernel image to verify.\n" | 55 "--in <infile>\t\t\tVerified boot kernel image to verify.\n" |
| 58 "--firmware_key_pub <pubkeyfile>\tPre-processed public firmware key " | 56 "--firmware_key_pub <pubkeyfile>\tPre-processed public firmware key " |
| 59 "to use for verification.\n\n" | 57 "to use for verification.\n\n" |
| 60 "For \"--generate\", required OPTIONS are:\n" | 58 "For \"--generate\", required OPTIONS are:\n" |
| 61 "--firmware_key <privkeyfile>\tPrivate firmware signing key file\n" | 59 "--firmware_key <privkeyfile>\tPrivate firmware signing key file\n" |
| 62 "--kernel_key <privkeyfile>\tPrivate kernel signing key file\n" | 60 "--kernel_key <privkeyfile>\tPrivate kernel signing key file\n" |
| 63 "--kernel_key_pub <pubkeyfile>\tPre-processed public kernel signing" | 61 "--kernel_key_pub <pubkeyfile>\tPre-processed public kernel signing" |
| 64 " key\n" | 62 " key\n" |
| 65 "--firmware_sign_algorithm <algoid>\tSigning algorithm used by " | 63 "--firmware_sign_algorithm <algoid>\tSigning algorithm used by " |
| 66 "the firmware\n" | 64 "the firmware\n" |
| 67 "--kernel_sign_algorithm <algoid>\tSigning algorithm to use for kernel\n" | 65 "--kernel_sign_algorithm <algoid>\tSigning algorithm to use for kernel\n" |
| 68 "--kernel_key_version <version#>\tKernel signing Key Version#\n" | 66 "--kernel_key_version <version#>\tKernel signing Key Version#\n" |
| 69 "--kernel_version <version#>\tKernel Version#\n" | 67 "--kernel_version <version#>\tKernel Version#\n" |
| 70 "--in <infile>\t\tKernel Image to sign\n" | 68 "--in <infile>\t\tKernel Image to sign\n" |
| 71 "--out <outfile>\t\tOutput file for verified boot Kernel image\n\n" | 69 "--out <outfile>\t\tOutput file for verified boot Kernel image\n\n" |
| 72 "Optional arguments for \"--generate\" include:\n" | 70 "Optional arguments for \"--generate\" include:\n" |
| 73 "--config <file>\t\t\tPopulate contents of kernel config from a file\n" | |
| 74 "--vblock\t\t\tJust output the verification block\n\n" | 71 "--vblock\t\t\tJust output the verification block\n\n" |
| 75 "<algoid> (for --*_sign_algorithm) is one of the following:\n"; | 72 "<algoid> (for --*_sign_algorithm) is one of the following:\n"; |
| 76 for (int i = 0; i < kNumAlgorithms; i++) { | 73 for (int i = 0; i < kNumAlgorithms; i++) { |
| 77 cerr << i << " for " << algo_strings[i] << "\n"; | 74 cerr << i << " for " << algo_strings[i] << "\n"; |
| 78 } | 75 } |
| 79 cerr << "\n\n"; | 76 cerr << "\n\n"; |
| 80 } | 77 } |
| 81 | 78 |
| 82 bool KernelUtility::ParseCmdLineOptions(int argc, char* argv[]) { | 79 bool KernelUtility::ParseCmdLineOptions(int argc, char* argv[]) { |
| 83 int option_index, i; | 80 int option_index, i; |
| 84 enum { | 81 enum { |
| 85 OPT_FIRMWARE_KEY = 1000, | 82 OPT_FIRMWARE_KEY = 1000, |
| 86 OPT_FIRMWARE_KEY_PUB, | 83 OPT_FIRMWARE_KEY_PUB, |
| 87 OPT_KERNEL_KEY, | 84 OPT_KERNEL_KEY, |
| 88 OPT_KERNEL_KEY_PUB, | 85 OPT_KERNEL_KEY_PUB, |
| 89 OPT_FIRMWARE_SIGN_ALGORITHM, | 86 OPT_FIRMWARE_SIGN_ALGORITHM, |
| 90 OPT_KERNEL_SIGN_ALGORITHM, | 87 OPT_KERNEL_SIGN_ALGORITHM, |
| 91 OPT_KERNEL_KEY_VERSION, | 88 OPT_KERNEL_KEY_VERSION, |
| 92 OPT_KERNEL_VERSION, | 89 OPT_KERNEL_VERSION, |
| 93 OPT_IN, | 90 OPT_IN, |
| 94 OPT_OUT, | 91 OPT_OUT, |
| 95 OPT_GENERATE, | 92 OPT_GENERATE, |
| 96 OPT_VERIFY, | 93 OPT_VERIFY, |
| 97 OPT_DESCRIBE, | 94 OPT_DESCRIBE, |
| 98 OPT_CONFIG, | |
| 99 OPT_VBLOCK, | 95 OPT_VBLOCK, |
| 100 }; | 96 }; |
| 101 static struct option long_options[] = { | 97 static struct option long_options[] = { |
| 102 {"firmware_key", 1, 0, OPT_FIRMWARE_KEY }, | 98 {"firmware_key", 1, 0, OPT_FIRMWARE_KEY }, |
| 103 {"firmware_key_pub", 1, 0, OPT_FIRMWARE_KEY_PUB }, | 99 {"firmware_key_pub", 1, 0, OPT_FIRMWARE_KEY_PUB }, |
| 104 {"kernel_key", 1, 0, OPT_KERNEL_KEY }, | 100 {"kernel_key", 1, 0, OPT_KERNEL_KEY }, |
| 105 {"kernel_key_pub", 1, 0, OPT_KERNEL_KEY_PUB }, | 101 {"kernel_key_pub", 1, 0, OPT_KERNEL_KEY_PUB }, |
| 106 {"firmware_sign_algorithm", 1, 0, OPT_FIRMWARE_SIGN_ALGORITHM }, | 102 {"firmware_sign_algorithm", 1, 0, OPT_FIRMWARE_SIGN_ALGORITHM }, |
| 107 {"kernel_sign_algorithm", 1, 0, OPT_KERNEL_SIGN_ALGORITHM }, | 103 {"kernel_sign_algorithm", 1, 0, OPT_KERNEL_SIGN_ALGORITHM }, |
| 108 {"kernel_key_version", 1, 0, OPT_KERNEL_KEY_VERSION }, | 104 {"kernel_key_version", 1, 0, OPT_KERNEL_KEY_VERSION }, |
| 109 {"kernel_version", 1, 0, OPT_KERNEL_VERSION }, | 105 {"kernel_version", 1, 0, OPT_KERNEL_VERSION }, |
| 110 {"in", 1, 0, OPT_IN }, | 106 {"in", 1, 0, OPT_IN }, |
| 111 {"out", 1, 0, OPT_OUT }, | 107 {"out", 1, 0, OPT_OUT }, |
| 112 {"generate", 0, 0, OPT_GENERATE }, | 108 {"generate", 0, 0, OPT_GENERATE }, |
| 113 {"verify", 0, 0, OPT_VERIFY }, | 109 {"verify", 0, 0, OPT_VERIFY }, |
| 114 {"describe", 0, 0, OPT_DESCRIBE }, | 110 {"describe", 0, 0, OPT_DESCRIBE }, |
| 115 {"config", 1, 0, OPT_CONFIG }, | |
| 116 {"vblock", 0, 0, OPT_VBLOCK }, | 111 {"vblock", 0, 0, OPT_VBLOCK }, |
| 117 {NULL, 0, 0, 0} | 112 {NULL, 0, 0, 0} |
| 118 }; | 113 }; |
| 119 while ((i = getopt_long(argc, argv, "", long_options, &option_index)) != -1) { | 114 while ((i = getopt_long(argc, argv, "", long_options, &option_index)) != -1) { |
| 120 switch (i) { | 115 switch (i) { |
| 121 case '?': | 116 case '?': |
| 122 return false; | 117 return false; |
| 123 break; | 118 break; |
| 124 case OPT_FIRMWARE_KEY: | 119 case OPT_FIRMWARE_KEY: |
| 125 firmware_key_file_ = optarg; | 120 firmware_key_file_ = optarg; |
| (...skipping 43 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 169 break; | 164 break; |
| 170 case OPT_GENERATE: | 165 case OPT_GENERATE: |
| 171 is_generate_ = true; | 166 is_generate_ = true; |
| 172 break; | 167 break; |
| 173 case OPT_VERIFY: | 168 case OPT_VERIFY: |
| 174 is_verify_ = true; | 169 is_verify_ = true; |
| 175 break; | 170 break; |
| 176 case OPT_DESCRIBE: | 171 case OPT_DESCRIBE: |
| 177 is_describe_ = true; | 172 is_describe_ = true; |
| 178 break; | 173 break; |
| 179 case OPT_CONFIG: | |
| 180 config_file_ = optarg; | |
| 181 break; | |
| 182 case OPT_VBLOCK: | 174 case OPT_VBLOCK: |
| 183 is_only_vblock_ = true; | 175 is_only_vblock_ = true; |
| 184 break; | 176 break; |
| 185 } | 177 } |
| 186 } | 178 } |
| 187 return CheckOptions(); | 179 return CheckOptions(); |
| 188 } | 180 } |
| 189 | 181 |
| 190 void KernelUtility::OutputSignedImage(void) { | 182 void KernelUtility::OutputSignedImage(void) { |
| 191 if (image_) { | 183 if (image_) { |
| 192 if (!WriteKernelImage(out_file_.c_str(), image_, is_only_vblock_)) { | 184 if (!WriteKernelImage(out_file_.c_str(), image_, is_only_vblock_)) { |
| 193 cerr << "Couldn't write verified boot kernel image to file " | 185 cerr << "Couldn't write verified boot kernel image to file " |
| 194 << out_file_ <<".\n"; | 186 << out_file_ <<".\n"; |
| 195 } | 187 } |
| 196 } | 188 } |
| 197 } | 189 } |
| 198 | 190 |
| 199 void KernelUtility::DescribeSignedImage(void) { | 191 void KernelUtility::DescribeSignedImage(void) { |
| 200 image_ = ReadKernelImage(in_file_.c_str()); | 192 image_ = ReadKernelImage(in_file_.c_str()); |
| 201 if (!image_) { | 193 if (!image_) { |
| 202 cerr << "Couldn't read kernel image or malformed image.\n"; | 194 cerr << "Couldn't read kernel image or malformed image.\n"; |
| 203 return; | 195 return; |
| 204 } | 196 } |
| 205 PrintKernelImage(image_); | 197 PrintKernelImage(image_); |
| 206 } | 198 } |
| 207 | 199 |
| 208 bool KernelUtility::GenerateSignedImage(void) { | 200 bool KernelUtility::GenerateSignedImage(void) { |
| 209 uint64_t len; | |
| 210 uint64_t kernel_key_pub_len; | 201 uint64_t kernel_key_pub_len; |
| 211 image_ = KernelImageNew(); | 202 image_ = KernelImageNew(); |
| 212 | 203 |
| 213 Memcpy(image_->magic, KERNEL_MAGIC, KERNEL_MAGIC_SIZE); | 204 Memcpy(image_->magic, KERNEL_MAGIC, KERNEL_MAGIC_SIZE); |
| 214 | 205 |
| 215 // TODO(gauravsh): make this a command line option. | 206 // TODO(gauravsh): make this a command line option. |
| 216 image_->header_version = 1; | 207 image_->header_version = 1; |
| 217 image_->firmware_sign_algorithm = (uint16_t) firmware_sign_algorithm_; | 208 image_->firmware_sign_algorithm = (uint16_t) firmware_sign_algorithm_; |
| 218 // Copy pre-processed public signing key. | 209 // Copy pre-processed public signing key. |
| 219 image_->kernel_sign_algorithm = (uint16_t) kernel_sign_algorithm_; | 210 image_->kernel_sign_algorithm = (uint16_t) kernel_sign_algorithm_; |
| 220 image_->kernel_sign_key = BufferFromFile(kernel_key_pub_file_.c_str(), | 211 image_->kernel_sign_key = BufferFromFile(kernel_key_pub_file_.c_str(), |
| 221 &kernel_key_pub_len); | 212 &kernel_key_pub_len); |
| 222 if (!image_->kernel_sign_key) | 213 if (!image_->kernel_sign_key) |
| 223 return false; | 214 return false; |
| 224 image_->kernel_key_version = kernel_key_version_; | 215 image_->kernel_key_version = kernel_key_version_; |
| 225 | 216 |
| 226 // Update header length. | 217 // Update header length. |
| 227 image_->header_len = GetKernelHeaderLen(image_); | 218 image_->header_len = GetKernelHeaderLen(image_); |
| 228 | 219 |
| 229 // Calculate header checksum. | 220 // Calculate header checksum. |
| 230 CalculateKernelHeaderChecksum(image_, image_->header_checksum); | 221 CalculateKernelHeaderChecksum(image_, image_->header_checksum); |
| 231 | 222 |
| 232 image_->kernel_version = kernel_version_; | 223 image_->kernel_version = kernel_version_; |
| 233 if (!config_file_.empty()) { | |
| 234 kernel_config_ = BufferFromFile(config_file_.c_str(), &len); | |
| 235 if (len >= sizeof(image_->kernel_config)) { | |
| 236 cerr << "Input kernel config file is too big!"; | |
| 237 return false; | |
| 238 } | |
| 239 Memcpy(image_->kernel_config, | |
| 240 kernel_config_, len); | |
| 241 } else { | |
| 242 Memset(image_->kernel_config, 0, | |
| 243 sizeof(image_->kernel_config)); | |
| 244 } | |
| 245 image_->kernel_data = BufferFromFile(in_file_.c_str(), | 224 image_->kernel_data = BufferFromFile(in_file_.c_str(), |
| 246 &image_->kernel_len); | 225 &image_->kernel_len); |
| 247 if (!image_->kernel_data) | 226 if (!image_->kernel_data) |
| 248 return false; | 227 return false; |
| 249 // Generate and add the signatures. | 228 // Generate and add the signatures. |
| 250 if (!AddKernelKeySignature(image_, firmware_key_file_.c_str())) { | 229 if (!AddKernelKeySignature(image_, firmware_key_file_.c_str())) { |
| 251 cerr << "Couldn't write key signature to verified boot kernel image.\n"; | 230 cerr << "Couldn't write key signature to verified boot kernel image.\n"; |
| 252 return false; | 231 return false; |
| 253 } | 232 } |
| 254 | 233 |
| (...skipping 101 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 356 } | 335 } |
| 357 else if (ku.is_verify()) { | 336 else if (ku.is_verify()) { |
| 358 cerr << "Verification "; | 337 cerr << "Verification "; |
| 359 if (ku.VerifySignedImage()) | 338 if (ku.VerifySignedImage()) |
| 360 cerr << "SUCCESS.\n"; | 339 cerr << "SUCCESS.\n"; |
| 361 else | 340 else |
| 362 cerr << "FAILURE.\n"; | 341 cerr << "FAILURE.\n"; |
| 363 } | 342 } |
| 364 return 0; | 343 return 0; |
| 365 } | 344 } |
| OLD | NEW |