Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(199)

Unified Diff: vkernel/kernel_image.c

Issue 2815011: Remove unused files, and tidy the directory structure of the remaining ones. (Closed) Base URL: ssh://gitrw.chromium.org/vboot_reference.git
Patch Set: Created 10 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « vkernel/include/kernel_image_fw.h ('k') | vkernel/kernel_image_fw.c » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: vkernel/kernel_image.c
diff --git a/vkernel/kernel_image.c b/vkernel/kernel_image.c
deleted file mode 100644
index 392ee1fc344f2dcd8867e55fd947c9d506ca2ee4..0000000000000000000000000000000000000000
--- a/vkernel/kernel_image.c
+++ /dev/null
@@ -1,754 +0,0 @@
-/* Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
- * Use of this source code is governed by a BSD-style license that can be
- * found in the LICENSE file.
- *
- * Functions for generating and manipulating a verified boot kernel image.
- * (Userland portion)
- */
-#include "kernel_image.h"
-
-#include <fcntl.h>
-#include <stddef.h>
-#include <stdio.h>
-#include <sys/types.h>
-#include <sys/stat.h>
-#include <unistd.h>
-
-#include "cryptolib.h"
-#include "file_keys.h"
-#include "kernel_blob.h"
-#include "rollback_index.h"
-#include "signature_digest.h"
-#include "stateful_util.h"
-#include "utility.h"
-
-/* Macro to determine the size of a field structure in the KernelImage
- * structure. */
-#define FIELD_LEN(field) (sizeof(((KernelImage*)0)->field))
-
-KernelImage* KernelImageNew(void) {
- KernelImage* image = (KernelImage*) Malloc(sizeof(KernelImage));
- if (image) {
- image->kernel_sign_key = NULL;
- image->kernel_key_signature = NULL;
- image->preamble_signature = NULL;
- image->kernel_signature = NULL;
- image->kernel_data = NULL;
- image->padded_header_size = 0x4000;
- }
- return image;
-}
-
-void KernelImageFree(KernelImage* image) {
- if (image) {
- Free(image->kernel_sign_key);
- Free(image->kernel_key_signature);
- Free(image->preamble_signature);
- Free(image->kernel_signature);
- Free(image->kernel_data);
- Free(image);
- }
-}
-
-uint64_t GetHeaderSizeOnDisk(const KernelImage* image) {
- uint64_t kernel_signature_len = siglen_map[image->kernel_sign_algorithm];
- uint64_t kernel_key_signature_len =
- siglen_map[image->firmware_sign_algorithm];
-
- return FIELD_LEN(magic) +
- GetKernelHeaderLen(image) +
- kernel_key_signature_len +
- GetKernelPreambleLen(image->kernel_sign_algorithm) +
- kernel_signature_len;
-}
-
-
-KernelImage* ReadKernelImage(const char* input_file) {
- uint64_t file_size;
- uint64_t on_disk_header_size;
- uint64_t on_disk_padding;
- int header_len = 0;
- int kernel_key_signature_len;
- int kernel_sign_key_len;
- int kernel_signature_len;
- uint8_t* kernel_buf;
- uint8_t header_checksum[FIELD_LEN(header_checksum)];
- MemcpyState st;
- KernelImage* image = KernelImageNew();
-
- if (!image)
- return NULL;
-
- kernel_buf = BufferFromFile(input_file, &file_size);
-
- st.remaining_len = file_size;
- st.remaining_buf = kernel_buf;
- st.overrun = 0;
-
- /* Read and compare magic bytes. */
- StatefulMemcpy(&st, &image->magic, KERNEL_MAGIC_SIZE);
-
- if (SafeMemcmp(image->magic, KERNEL_MAGIC, KERNEL_MAGIC_SIZE)) {
- debug("Wrong Kernel Magic.\n");
- Free(kernel_buf);
- return NULL;
- }
- StatefulMemcpy(&st, &image->header_version, FIELD_LEN(header_version));
- StatefulMemcpy(&st, &image->header_len, FIELD_LEN(header_len));
- StatefulMemcpy(&st, &image->firmware_sign_algorithm,
- FIELD_LEN(firmware_sign_algorithm));
- StatefulMemcpy(&st, &image->kernel_sign_algorithm,
- FIELD_LEN(kernel_sign_algorithm));
-
- /* Valid Kernel Key signing algorithm. */
- if (image->firmware_sign_algorithm >= kNumAlgorithms) {
- Free(kernel_buf);
- return NULL;
- }
-
- /* Valid Kernel Signing Algorithm? */
- if (image->kernel_sign_algorithm >= kNumAlgorithms) {
- Free(kernel_buf);
- return NULL;
- }
-
- /* Compute size of pre-processed RSA public keys and signatures. */
- kernel_key_signature_len = siglen_map[image->firmware_sign_algorithm];
- kernel_sign_key_len = RSAProcessedKeySize(image->kernel_sign_algorithm);
- kernel_signature_len = siglen_map[image->kernel_sign_algorithm];
-
- /* Check whether key header length is correct. */
- header_len = GetKernelHeaderLen(image);
- if (header_len != image->header_len) {
- debug("Header length mismatch. Got: %d, Expected: %d\n",
- image->header_len, header_len);
- Free(kernel_buf);
- return NULL;
- }
-
- /* Read pre-processed public half of the kernel signing key. */
- StatefulMemcpy(&st, &image->kernel_key_version,
- FIELD_LEN(kernel_key_version));
- image->kernel_sign_key = (uint8_t*) Malloc(kernel_sign_key_len);
- StatefulMemcpy(&st, image->kernel_sign_key, kernel_sign_key_len);
- StatefulMemcpy(&st, image->header_checksum, FIELD_LEN(header_checksum));
-
- /* Check whether the header checksum matches. */
- CalculateKernelHeaderChecksum(image, header_checksum);
- if (SafeMemcmp(header_checksum, image->header_checksum,
- FIELD_LEN(header_checksum))) {
- debug("Invalid kernel header checksum!\n");
- Free(kernel_buf);
- return NULL;
- }
-
- /* Read key signature. */
- image->kernel_key_signature = (uint8_t*) Malloc(kernel_key_signature_len);
- StatefulMemcpy(&st, image->kernel_key_signature,
- kernel_key_signature_len);
-
- /* Read the kernel preamble. */
- StatefulMemcpy(&st, &image->kernel_version, FIELD_LEN(kernel_version));
- StatefulMemcpy(&st, &image->kernel_len, FIELD_LEN(kernel_len));
- StatefulMemcpy(&st, &image->bootloader_offset, FIELD_LEN(bootloader_offset));
- StatefulMemcpy(&st, &image->bootloader_size, FIELD_LEN(bootloader_size));
- StatefulMemcpy(&st, &image->padded_header_size,
- FIELD_LEN(padded_header_size));
-
- /* Read preamble and kernel signatures. */
- image->kernel_signature = (uint8_t*) Malloc(kernel_signature_len);
- StatefulMemcpy(&st, image->kernel_signature, kernel_signature_len);
- image->preamble_signature = (uint8_t*) Malloc(kernel_signature_len);
- StatefulMemcpy(&st, image->preamble_signature, kernel_signature_len);
-
- /* Skip over the rest of the padded header, unless we're already past it. */
- on_disk_header_size = file_size - st.remaining_len;
- if (image->padded_header_size > on_disk_header_size) {
- on_disk_padding = image->padded_header_size - on_disk_header_size;
- if (st.remaining_len < on_disk_padding)
- st.overrun = -1;
- st.remaining_buf += on_disk_padding;
- st.remaining_len -= on_disk_padding;
- }
-
- /* Read kernel image data. */
- image->kernel_data = (uint8_t*) Malloc(image->kernel_len);
- StatefulMemcpy(&st, image->kernel_data, image->kernel_len);
-
- if(st.overrun) {
- Free(kernel_buf);
- return NULL;
- }
- Free(kernel_buf);
- return image;
-}
-
-int GetKernelHeaderLen(const KernelImage* image) {
- return (FIELD_LEN(header_version) + FIELD_LEN(header_len) +
- FIELD_LEN(firmware_sign_algorithm) +
- FIELD_LEN(kernel_sign_algorithm) + FIELD_LEN(kernel_key_version) +
- RSAProcessedKeySize(image->kernel_sign_algorithm) +
- FIELD_LEN(header_checksum));
-}
-
-void CalculateKernelHeaderChecksum(const KernelImage* image,
- uint8_t* header_checksum) {
- uint8_t* checksum;
- DigestContext ctx;
- 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));
- checksum = DigestFinal(&ctx);
- Memcpy(header_checksum, checksum, FIELD_LEN(header_checksum));
- Free(checksum);
- return;
-}
-
-uint8_t* GetKernelHeaderBlob(const KernelImage* image) {
- uint8_t* header_blob = NULL;
- MemcpyState st;
-
- header_blob = (uint8_t*) Malloc(GetKernelHeaderLen(image));
- st.remaining_len = GetKernelHeaderLen(image);
- st.remaining_buf = header_blob;
- st.overrun = 0;
-
- StatefulMemcpy_r(&st, &image->header_version, FIELD_LEN(header_version));
- StatefulMemcpy_r(&st, &image->header_len, FIELD_LEN(header_len));
- StatefulMemcpy_r(&st, &image->firmware_sign_algorithm,
- FIELD_LEN(firmware_sign_algorithm));
- StatefulMemcpy_r(&st, &image->kernel_sign_algorithm,
- FIELD_LEN(kernel_sign_algorithm));
- StatefulMemcpy_r(&st, &image->kernel_key_version,
- FIELD_LEN(kernel_key_version));
- StatefulMemcpy_r(&st, image->kernel_sign_key,
- RSAProcessedKeySize(image->kernel_sign_algorithm));
- StatefulMemcpy_r(&st, &image->header_checksum, FIELD_LEN(header_checksum));
-
- if (st.overrun || st.remaining_len != 0) { /* Underrun or Overrun. */
- Free(header_blob);
- return NULL;
- }
- return header_blob;
-}
-
-uint8_t* GetKernelPreambleBlob(const KernelImage* image) {
- uint8_t* preamble_blob = NULL;
- MemcpyState st;
-
- preamble_blob = (uint8_t*) Malloc(
- GetKernelPreambleLen(image->kernel_sign_algorithm));
- st.remaining_len = GetKernelPreambleLen(image->kernel_sign_algorithm);
- st.remaining_buf = preamble_blob;
- st.overrun = 0;
-
- StatefulMemcpy_r(&st, &image->kernel_version, FIELD_LEN(kernel_version));
- StatefulMemcpy_r(&st, &image->kernel_len, FIELD_LEN(kernel_len));
- StatefulMemcpy_r(&st, &image->bootloader_offset, FIELD_LEN(bootloader_offset));
- StatefulMemcpy_r(&st, &image->bootloader_size, FIELD_LEN(bootloader_size));
- StatefulMemcpy_r(&st, &image->padded_header_size,
- FIELD_LEN(padded_header_size));
- StatefulMemcpy_r(&st, image->kernel_signature,
- siglen_map[image->kernel_sign_algorithm]);
-
- if (st.overrun || st.remaining_len != 0) { /* Overrun or Underrun. */
- Free(preamble_blob);
- return NULL;
- }
- return preamble_blob;
-}
-
-uint8_t* GetKernelBlob(const KernelImage* image, uint64_t* blob_len) {
- int kernel_key_signature_len;
- int kernel_signature_len;
- uint8_t* kernel_blob = NULL;
- uint8_t* header_blob = NULL;
- MemcpyState st;
- uint64_t on_disk_header_size;
- uint64_t on_disk_padding = 0;
-
- if (!image)
- return NULL;
- kernel_key_signature_len = siglen_map[image->firmware_sign_algorithm];
- kernel_signature_len = siglen_map[image->kernel_sign_algorithm];
- on_disk_header_size = GetHeaderSizeOnDisk(image);
- if (image->padded_header_size > on_disk_header_size)
- on_disk_padding = image->padded_header_size - on_disk_header_size;
- *blob_len = on_disk_header_size + on_disk_padding + image->kernel_len;
- kernel_blob = (uint8_t*) Malloc(*blob_len);
- st.remaining_len = *blob_len;
- st.remaining_buf = kernel_blob;
- st.overrun = 0;
- header_blob = GetKernelHeaderBlob(image);
-
- StatefulMemcpy_r(&st, image->magic, FIELD_LEN(magic));
- StatefulMemcpy_r(&st, header_blob, GetKernelHeaderLen(image));
- StatefulMemcpy_r(&st, image->kernel_key_signature, kernel_key_signature_len);
- /* Copy over kernel preamble blob (including signatures.) */
- StatefulMemcpy_r(&st, &image->kernel_version, FIELD_LEN(kernel_version));
- StatefulMemcpy_r(&st, &image->kernel_len, FIELD_LEN(kernel_len));
- StatefulMemcpy_r(&st, &image->bootloader_offset,
- FIELD_LEN(bootloader_offset));
- StatefulMemcpy_r(&st, &image->bootloader_size, FIELD_LEN(bootloader_size));
- StatefulMemcpy_r(&st, &image->padded_header_size,
- FIELD_LEN(padded_header_size));
- StatefulMemcpy_r(&st, image->kernel_signature, kernel_signature_len);
- StatefulMemcpy_r(&st, image->preamble_signature, kernel_signature_len);
- /* Copy a bunch of zeros to pad out the header */
- if (on_disk_padding)
- StatefulMemset_r(&st, 0, on_disk_padding);
- StatefulMemcpy_r(&st, image->kernel_data, image->kernel_len);
-
- Free(header_blob);
-
- if (st.overrun || st.remaining_len != 0) { /* Underrun or Overrun. */
- debug("GetKernelBlob() failed.\n");
- Free(kernel_blob);
- return NULL;
- }
- return kernel_blob;
-}
-
-int WriteKernelImage(const char* output_file,
- const KernelImage* image,
- int is_only_vblock,
- int is_subkey_out) {
- int fd;
- int success = 1;
- uint8_t* kernel_blob = NULL;
- uint8_t* subkey_out_buf = NULL;
- uint8_t* subkey_header = NULL;
- uint64_t blob_len;
-
- if (!image)
- return 0;
- if (-1 == (fd = creat(output_file, 0666))) {
- debug("Couldn't open file for writing kernel image: %s\n",
- output_file);
- return 0;
- }
- if (is_subkey_out) {
- blob_len = GetKernelHeaderLen(image) +
- siglen_map[image->firmware_sign_algorithm];
- subkey_out_buf = (uint8_t*) Malloc(blob_len);
- subkey_header = GetKernelHeaderBlob(image);
- Memcpy(subkey_out_buf, subkey_header, GetKernelHeaderLen(image));
- Memcpy(subkey_out_buf + GetKernelHeaderLen(image),
- image->kernel_key_signature,
- siglen_map[image->firmware_sign_algorithm]);
- if (blob_len != write(fd, subkey_out_buf, blob_len)) {
- debug("Couldn't write Kernel Subkey header to file: %s\n",
- output_file);
- success = 0;
- }
- Free(subkey_header);
- Free(subkey_out_buf);
- close(fd);
- return success;
- }
-
- kernel_blob = GetKernelBlob(image, &blob_len);
- if (!kernel_blob) {
- debug("Couldn't create kernel blob from KernelImage.\n");
- return 0;
- }
- if (!is_only_vblock) {
- if (blob_len != write(fd, kernel_blob, blob_len)) {
- debug("Couldn't write Kernel Image to file: %s\n",
- output_file);
- success = 0;
- }
- } else {
- /* Exclude kernel_data. */
- int vblock_len = blob_len - (image->kernel_len);
- if (vblock_len != write(fd, kernel_blob, vblock_len)) {
- debug("Couldn't write Kernel Image Verification block to file: %s\n",
- output_file);
- success = 0;
- }
- }
- Free(kernel_blob);
- close(fd);
- return success;
-}
-
-void PrintKernelImage(const KernelImage* image) {
- uint64_t header_size;
-
- if (!image)
- return;
-
- header_size = GetHeaderSizeOnDisk(image);
- if (image->padded_header_size > header_size)
- header_size = image->padded_header_size;
-
-
- /* Print header. */
- printf("Header Version = %d\n"
- "Header Length = %d\n"
- "Kernel Key Signature Algorithm = %s\n"
- "Kernel Signature Algorithm = %s\n"
- "Kernel Key Version = %d\n\n",
- image->header_version,
- image->header_len,
- algo_strings[image->firmware_sign_algorithm],
- algo_strings[image->kernel_sign_algorithm],
- image->kernel_key_version);
- /* TODO(gauravsh): Output hash and key signature here? */
- /* Print preamble. */
- printf("Kernel Version = %d\n"
- "kernel Length = %" PRId64 " (0x%" PRIx64 ")\n"
- "Bootloader Offset = %" PRId64 " (0x%" PRIx64 ")\n"
- "Bootloader Size = %" PRId64 " (0x%" PRIx64 ")\n"
- "Padded Header Size = %" PRId64 " (0x%" PRIx64 ")\n\n"
- "Actual Header Size on disk = %" PRIu64 " (0x%" PRIx64 ")\n",
- image->kernel_version,
- image->kernel_len, image->kernel_len,
- image->bootloader_offset, image->bootloader_offset,
- image->bootloader_size, image->bootloader_size,
- image->padded_header_size, image->padded_header_size,
- header_size, header_size);
- /* TODO(gauravsh): Output kernel signature here? */
-}
-
-
-int VerifyKernelImage(const RSAPublicKey* firmware_key,
- const KernelImage* image,
- const int dev_mode) {
- RSAPublicKey* kernel_sign_key = NULL;
- uint8_t* header_digest = NULL;
- uint8_t* preamble_digest = NULL;
- uint8_t* kernel_digest = NULL;
- int kernel_sign_key_size;
- int kernel_signature_size;
- int error_code = 0;
- DigestContext ctx;
- if (!image)
- return VERIFY_KERNEL_INVALID_IMAGE;
-
- /* Verify kernel key signature on the key header if we
- * are not in dev mode.
- *
- * TODO(gauravsh): Add additional sanity checks here for:
- * 1) verifying the header length is correct.
- * 2) header_checksum is correct.
- */
-
- if (image->firmware_sign_algorithm >= kNumAlgorithms)
- return VERIFY_KERNEL_INVALID_ALGORITHM;
- if (image->kernel_sign_algorithm >= kNumAlgorithms)
- return VERIFY_KERNEL_INVALID_ALGORITHM;
-
- if (!dev_mode) {
- DigestInit(&ctx, image->firmware_sign_algorithm);
- DigestUpdate(&ctx, (uint8_t*) &image->header_version,
- FIELD_LEN(header_version));
- DigestUpdate(&ctx, (uint8_t*) &image->header_len,
- FIELD_LEN(header_len));
- DigestUpdate(&ctx, (uint8_t*) &image->firmware_sign_algorithm,
- FIELD_LEN(firmware_sign_algorithm));
- DigestUpdate(&ctx, (uint8_t*) &image->kernel_sign_algorithm,
- FIELD_LEN(kernel_sign_algorithm));
- DigestUpdate(&ctx, (uint8_t*) &image->kernel_key_version,
- FIELD_LEN(kernel_key_version));
- DigestUpdate(&ctx, image->kernel_sign_key,
- RSAProcessedKeySize(image->kernel_sign_algorithm));
- DigestUpdate(&ctx, image->header_checksum,
- FIELD_LEN(header_checksum));
- header_digest = DigestFinal(&ctx);
- if (!RSAVerify(firmware_key, image->kernel_key_signature,
- siglen_map[image->firmware_sign_algorithm],
- image->firmware_sign_algorithm,
- header_digest)) {
- debug("VerifyKernelImage(): Key signature check failed.\n");
- error_code = VERIFY_KERNEL_KEY_SIGNATURE_FAILED;
- goto verify_failure;
- }
- }
-
- /* Get kernel signing key to verify the rest of the kernel. */
- kernel_sign_key_size = RSAProcessedKeySize(image->kernel_sign_algorithm);
- kernel_sign_key = RSAPublicKeyFromBuf(image->kernel_sign_key,
- kernel_sign_key_size);
- kernel_signature_size = siglen_map[image->kernel_sign_algorithm];
-
- /* Verify kernel preamble signature. */
- DigestInit(&ctx, image->kernel_sign_algorithm);
- DigestUpdate(&ctx, (uint8_t*) &image->kernel_version,
- FIELD_LEN(kernel_version));
- DigestUpdate(&ctx, (uint8_t*) &image->kernel_len,
- FIELD_LEN(kernel_len));
- DigestUpdate(&ctx, (uint8_t*) &image->bootloader_offset,
- FIELD_LEN(bootloader_offset));
- DigestUpdate(&ctx, (uint8_t*) &image->bootloader_size,
- FIELD_LEN(bootloader_size));
- DigestUpdate(&ctx, (uint8_t*) &image->padded_header_size,
- FIELD_LEN(padded_header_size));
- DigestUpdate(&ctx, (uint8_t*) image->kernel_signature,
- kernel_signature_size);
- preamble_digest = DigestFinal(&ctx);
- if (!RSAVerify(kernel_sign_key, image->preamble_signature,
- kernel_signature_size, image->kernel_sign_algorithm,
- preamble_digest)) {
- error_code = VERIFY_KERNEL_PREAMBLE_SIGNATURE_FAILED;
- goto verify_failure;
- }
-
- /* Verify kernel signature - kernel signature is computed on the contents
- * of kernel_data.
- * Association between the kernel_data and preamble is maintained by making
- * the kernel signature a part of the preamble and verifying it as part
- * of preamble signature checking. */
-
- kernel_digest = DigestBuf(image->kernel_data,
- image->kernel_len,
- image->kernel_sign_algorithm);
- if (!RSAVerify(kernel_sign_key, image->kernel_signature,
- kernel_signature_size, image->kernel_sign_algorithm,
- kernel_digest)) {
- error_code = VERIFY_KERNEL_SIGNATURE_FAILED;
- goto verify_failure;
- }
-
-verify_failure:
- RSAPublicKeyFree(kernel_sign_key);
- Free(kernel_digest);
- Free(preamble_digest);
- Free(header_digest);
- return error_code;
-}
-
-const char* VerifyKernelErrorString(int error) {
- return kVerifyKernelErrors[error];
-}
-
-int AddKernelKeySignature(KernelImage* image, const char* firmware_key_file) {
- uint8_t* header_blob = NULL;
- uint8_t* signature = NULL;
- int signature_len = siglen_map[image->firmware_sign_algorithm];
- if (!image || !firmware_key_file)
- return 0;
- header_blob = GetKernelHeaderBlob(image);
- if (!header_blob)
- return 0;
- if (!(signature = SignatureBuf(header_blob,
- GetKernelHeaderLen(image),
- firmware_key_file,
- image->firmware_sign_algorithm))) {
- Free(header_blob);
- return 0;
- }
- image->kernel_key_signature = Malloc(signature_len);
- Memcpy(image->kernel_key_signature, signature, signature_len);
- Free(signature);
- Free(header_blob);
- return 1;
-}
-
-int AddKernelSignature(KernelImage* image,
- const char* kernel_signing_key_file) {
- uint8_t* preamble_blob = NULL;
- uint8_t* preamble_signature = NULL;
- uint8_t* kernel_signature = NULL;
- uint8_t* kernel_buf;
- int algorithm = image->kernel_sign_algorithm;
- int signature_len = siglen_map[algorithm];
-
- /* Kernel signature must be calculated first as its used for computing the
- * preamble signature. */
- kernel_buf = (uint8_t*) Malloc(image->kernel_len);
- Memcpy(kernel_buf, image->kernel_data, image->kernel_len);
- if (!(kernel_signature = SignatureBuf(kernel_buf,
- image->kernel_len,
- kernel_signing_key_file,
- algorithm))) {
- Free(preamble_blob);
- Free(kernel_buf);
- debug("Could not compute signature on the kernel.\n");
- return 0;
- }
- image->kernel_signature = (uint8_t*) Malloc(signature_len);
- Memcpy(image->kernel_signature, kernel_signature, signature_len);
-
-
- preamble_blob = GetKernelPreambleBlob(image);
- if (!(preamble_signature = SignatureBuf(preamble_blob,
- GetKernelPreambleLen(algorithm),
- kernel_signing_key_file,
- algorithm))) {
- debug("Could not compute signature on the kernel preamble.\n");
- Free(preamble_blob);
- return 0;
- }
- image->preamble_signature = (uint8_t*) Malloc(signature_len);
- Memcpy(image->preamble_signature, preamble_signature, signature_len);
-
- Free(preamble_signature);
- Free(preamble_blob);
- Free(kernel_signature);
- Free(kernel_buf);
- return 1;
-}
-
-/* Return the smallest integral multiple of [alignment] that is equal to or
- * greater than [val]. Used to determine the number of
- * pages/sectors/blocks/whatever needed to contain [val] items/bytes/etc. */
-static uint64_t roundup(uint64_t val, uint64_t alignment) {
- uint64_t rem = val % alignment;
- if ( rem )
- return val + (alignment - rem);
- return val;
-}
-
-/* Match regexp /\b--\b/ to delimit the start of the kernel commandline. If we
- * don't find one, we'll use the whole thing. */
-static unsigned int find_cmdline_start(char *input, unsigned int max_len) {
- int start = 0;
- int i;
- for(i = 0; i < max_len-1 && input[i]; i++) {
- if (input[i] == '-' && input[i+1] == '-') { /* found a "--" */
- if ((i == 0 || input[i-1] == ' ') && /* nothing before it */
- (i+2 >= max_len || input[i+2] == ' ')) { /* nothing after it */
- start = i+2; /* note: hope there's a trailing '\0' */
- break;
- }
- }
- }
- while(input[start] == ' ') /* skip leading spaces */
- start++;
-
- return start;
-}
-
-uint8_t* GenerateKernelBlob(const char* kernel_file,
- const char* config_file,
- const char* bootloader_file,
- uint64_t* ret_blob_len,
- uint64_t* ret_bootloader_offset,
- uint64_t* ret_bootloader_size) {
- uint8_t* kernel_buf;
- uint8_t* config_buf;
- uint8_t* bootloader_buf;
- uint8_t* blob = 0;
- uint64_t kernel_size;
- uint64_t config_size;
- uint64_t bootloader_size;
- uint64_t blob_size;
- uint64_t kernel32_start = 0;
- uint64_t kernel32_size = 0;
- uint64_t bootloader_mem_start;
- uint64_t bootloader_mem_size;
- uint64_t now;
- struct linux_kernel_header *lh = 0;
- struct linux_kernel_params *params = 0;
- uint32_t cmdline_addr;
- uint64_t i;
-
- /* Read the input files. */
- kernel_buf = BufferFromFile(kernel_file, &kernel_size);
- if (!kernel_buf)
- goto done0;
-
- config_buf = BufferFromFile(config_file, &config_size);
- if (!config_buf)
- goto done1;
- if (config_size >= CROS_CONFIG_SIZE) { /* need room for trailing '\0' */
- error("config file %s is too large (>= %d bytes)\n",
- config_file, CROS_CONFIG_SIZE);
- goto done1;
- }
-
- /* Replace any newlines with spaces in the config file. */
- for (i=0; i < config_size; i++)
- if (config_buf[i] == '\n')
- config_buf[i] = ' ';
-
- bootloader_buf = BufferFromFile(bootloader_file, &bootloader_size);
- if (!bootloader_buf)
- goto done2;
-
- /* The first part of vmlinuz is a header, followed by a real-mode boot stub.
- * We only want the 32-bit part. */
- if (kernel_size) {
- lh = (struct linux_kernel_header *)kernel_buf;
- kernel32_start = (lh->setup_sects+1) << 9;
- kernel32_size = kernel_size - kernel32_start;
- }
-
- /* Allocate and zero the blob we need. */
- blob_size = roundup(kernel32_size, CROS_ALIGN) +
- CROS_CONFIG_SIZE +
- CROS_PARAMS_SIZE +
- roundup(bootloader_size, CROS_ALIGN);
- blob = (uint8_t *)Malloc(blob_size);
- if (!blob) {
- error("Couldn't allocate %ld bytes.\n", blob_size);
- goto done3;
- }
- Memset(blob, 0, blob_size);
- now = 0;
-
- /* Copy the 32-bit kernel. */
- if (kernel32_size)
- Memcpy(blob + now, kernel_buf + kernel32_start, kernel32_size);
- now += roundup(now + kernel32_size, CROS_ALIGN);
-
- /* Find the load address of the commandline. We'll need it later. */
- cmdline_addr = CROS_32BIT_ENTRY_ADDR + now
- + find_cmdline_start((char *)config_buf, config_size);
-
- /* Copy the config. */
- if (config_size)
- Memcpy(blob + now, config_buf, config_size);
- now += CROS_CONFIG_SIZE;
-
- /* The zeropage data is next. Overlay the linux_kernel_header onto it, and
- * tweak a few fields. */
- params = (struct linux_kernel_params *)(blob + now);
-
- if (kernel_size)
- Memcpy(&(params->setup_sects), &(lh->setup_sects),
- sizeof(*lh) - offsetof(struct linux_kernel_header, setup_sects));
- params->boot_flag = 0;
- params->ramdisk_image = 0; /* we don't support initrd */
- params->ramdisk_size = 0;
- params->type_of_loader = 0xff;
- params->cmd_line_ptr = cmdline_addr;
- now += CROS_PARAMS_SIZE;
-
- /* Finally, append the bootloader. Remember where it will load in memory, too.
- */
- bootloader_mem_start = CROS_32BIT_ENTRY_ADDR + now;
- bootloader_mem_size = roundup(bootloader_size, CROS_ALIGN);
- if (bootloader_size)
- Memcpy(blob + now, bootloader_buf, bootloader_size);
- now += bootloader_mem_size;
-
- /* Pass back some info. */
- if (ret_blob_len)
- *ret_blob_len = blob_size;
- if (ret_bootloader_offset)
- *ret_bootloader_offset = bootloader_mem_start;
- if (ret_bootloader_size)
- *ret_bootloader_size = bootloader_mem_size;
-
- /* Clean up and return the blob. */
-done3:
- Free(bootloader_buf);
-done2:
- Free(config_buf);
-done1:
- Free(kernel_buf);
-done0:
- return blob;
-}
« no previous file with comments | « vkernel/include/kernel_image_fw.h ('k') | vkernel/kernel_image_fw.c » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698