Data structure and interface for manipulating and handing firmware images for verified boot.

src/platform/vboot_reference/tests/firmware_image_tests.c

Index: src/platform/vboot_reference/tests/firmware_image_tests.c
diff --git a/src/platform/vboot_reference/tests/firmware_image_tests.c b/src/platform/vboot_reference/tests/firmware_image_tests.c
new file mode 100644
index 0000000000000000000000000000000000000000..db44f2e3daf5c8be984ac42bf544a4b38cb3efc7
--- /dev/null
+++ b/src/platform/vboot_reference/tests/firmware_image_tests.c
@@ -0,0 +1,178 @@
+/* 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.
+ *
+ * Tests for firmware image library.
+ */
+
+#include <stdio.h>
+#include <stdlib.h>
+
+#include "file_keys.h"
+#include "firmware_image.h"
+#include "rsa_utility.h"
+#include "sha_utility.h"
+#include "utility.h"
+
+int TEST_EQ(int result, int expected_result, char* testname) {
+  if (result == expected_result) {
+    fprintf(stderr, "%s Test \e[1;32mSUCCEEDED\e[m\n", testname);
+    return 1;
+  }
+  else {
+    fprintf(stderr, "%s Test \e[0;31mFAILED\e[m\n", testname);
+    return 0;
+  }
+}
+
+FirmwareImage* GenerateTestFirmwareImage(int algorithm,
+                                         uint8_t* sign_key,
+                                         int key_version,
+                                         int firmware_version,
+                                         int firmware_len) {
+  FirmwareImage* image = FirmwareImageNew();
+  uint8_t* header_hash;
+  DigestContext ctx;
+
+  Memcpy(image->magic, FIRMWARE_MAGIC, FIRMWARE_MAGIC_SIZE);
+  image->sign_algorithm = algorithm;
+  image->sign_key = (uint8_t*) Malloc(
+      RSAProcessedKeySize(image->sign_algorithm));
+  Memcpy(image->sign_key, sign_key, RSAProcessedKeySize(image->sign_algorithm));
+  image->key_version = key_version;
+
+  /* Calculate SHA-512 digest on header and populate header_hash. */
+  DigestInit(&ctx, ROOT_SIGNATURE_ALGORITHM);
+  DigestUpdate(&ctx, (uint8_t*) &image->header_len,
+               sizeof(image->header_len));
+  DigestUpdate(&ctx, (uint8_t*) &image->sign_algorithm,
+               sizeof(image->sign_algorithm));
+  DigestUpdate(&ctx, image->sign_key,
+               RSAProcessedKeySize(image->sign_algorithm));
+  DigestUpdate(&ctx, (uint8_t*) &image->key_version,
+               sizeof(image->key_version));
+  header_hash = DigestFinal(&ctx);
+  Memcpy(image->header_hash, header_hash, SHA512_DIGEST_SIZE);
+  Free(header_hash);
+
+  /* Update correct header length. */
+  image->header_len = (sizeof(image->header_len) +
+                       sizeof(image->sign_algorithm) +
+                       RSAProcessedKeySize(image->sign_algorithm) +
+                       sizeof(image->key_version) +
+                       sizeof(image->header_hash));
+
+  /* Populate firmware and preamble with dummy data. */
+  image->firmware_version = firmware_version;
+  image->firmware_len = firmware_len;
+  image->preamble_signature = image->firmware_signature = NULL;
+  Memset(image->preamble, 'P', FIRMWARE_PREAMBLE_SIZE);
+  image->firmware_data = Malloc(image->firmware_len);
+  Memset(image->firmware_data, 'F', image->firmware_len);
+
+  return image;
+}
+
+#define DEV_MODE_ENABLED 1
+#define DEV_MODE_DISABLED 0
+
+/* Normal Firmware Verification Tests. */
+int VerifyFirmwareTest(FirmwareImage* image, RSAPublicKey* root_key) {
+  int success = 1;
+  if (!TEST_EQ(VerifyFirmware(root_key, image, DEV_MODE_ENABLED),
+               VERIFY_SUCCESS,
+               "Normal Verification (Dev Mode)"))
+    success = 0;
+
+  if (!TEST_EQ(VerifyFirmware(root_key, image, DEV_MODE_DISABLED),
+               VERIFY_SUCCESS,
+               "Normal Verification (Trusted)"))
+    success = 0;
+  return success;
+}
+
+/* Tampered Firmware Verification Tests. */
+int VerifyFirmwareTamperTest(FirmwareImage* image, RSAPublicKey* root_key) {
+  int success = 1;
+  fprintf(stderr, "Tampering with firmware preamble....\n");
+  image->firmware_version = 0;
+  if (!TEST_EQ(VerifyFirmware(root_key, image, DEV_MODE_ENABLED),
+               VERIFY_PREAMBLE_SIGNATURE_FAILED,
+               "Firmware Preamble Tamper Verification (Dev Mode)"))
+    success = 0;
+
+  if (!TEST_EQ(VerifyFirmware(root_key, image, DEV_MODE_DISABLED),
+               VERIFY_PREAMBLE_SIGNATURE_FAILED,
+               "Firmware Preamble Tamper Verification (Trusted)"))
+    success = 0;
+  image->firmware_version = 1;
+
+  image->firmware_data[0] = 'T';
+  if (!TEST_EQ(VerifyFirmware(root_key, image, DEV_MODE_ENABLED),
+               VERIFY_FIRMWARE_SIGNATURE_FAILED,
+               "Firmware Tamper Verification (Dev Mode)"))
+    success = 0;
+  if (!TEST_EQ(VerifyFirmware(root_key, image, DEV_MODE_DISABLED),
+               VERIFY_FIRMWARE_SIGNATURE_FAILED,
+               "Firmware Tamper Verification (Trusted)"))
+    success = 0;
+  image->firmware_data[0] = 'F';
+
+
+  fprintf(stderr, "Tampering with root key signature...\n");
+  image->key_signature[0] = 0xFF;
+  image->key_signature[1] = 0x00;
+  if (!TEST_EQ(VerifyFirmware(root_key, image, DEV_MODE_ENABLED),
+               VERIFY_SUCCESS,
+               "Root Signature Tamper Verification (Dev Mode)"))
+    success = 0;
+  if (!TEST_EQ(VerifyFirmware(root_key, image, DEV_MODE_DISABLED),
+               VERIFY_ROOT_SIGNATURE_FAILED,
+               "Root Signature Tamper Verification (Trusted)"))
+    success = 0;
+
+  return success;
+}
+
+int main(int argc, char* argv[]) {
+  int len;
+  uint8_t* sign_key_buf;
+  FirmwareImage* image;
+  RSAPublicKey* root_key;
+  int success = 1;
+
+  if(argc != 6) {
+    fprintf(stderr, "Usage: %s <algorithm> <root key> <processed root pubkey>"
+            " <signing key> <processed signing key>\n", argv[0]);
+    return -1;
+  }
+
+  /* Read verification keys and create a test image. */
+  root_key = RSAPublicKeyFromFile(argv[3]);
+  sign_key_buf = BufferFromFile(argv[5], &len);
+  image = GenerateTestFirmwareImage(atoi(argv[1]), sign_key_buf, 1,
+                                    1, 1000);
+
+  /* Generate and populate signatures. */
+  if (!AddKeySignature(image, argv[2])) {
+    fprintf(stderr, "Couldn't create key signature.\n");
+    return -1;
+  }
+
+  if (!AddFirmwareSignature(image, argv[4], image->sign_algorithm)) {
+    fprintf(stderr, "Couldn't create firmware and preamble signature.\n");
+    return -1;
+  }
+
+  if (!VerifyFirmwareTest(image, root_key))
+    success = 0;
+  if (!VerifyFirmwareTamperTest(image, root_key))
+    success = 0;
+
+  /* Clean up. */
+  Free(root_key);
+  Free(sign_key_buf);
+  Free(image);
+
+  return !success;
+}