diff options
author | Joel Kitching <kitching@google.com> | 2019-10-07 23:14:33 +0800 |
---|---|---|
committer | Commit Bot <commit-bot@chromium.org> | 2019-12-24 16:28:47 +0000 |
commit | 80c1a85a87e589ed74962cad98f4892dbe6a3283 (patch) | |
tree | d61e7d365a7607c9f22a680fc2bf47375c3f96be /tests/vb20_common2_tests.c | |
parent | d9ddac88c99d8df5b7dde618e43807f3c0dd6080 (diff) | |
download | vboot-80c1a85a87e589ed74962cad98f4892dbe6a3283.tar.gz |
vboot: fold lib20/common.c into 2lib/2common.c
Also rename tests.
BUG=b:124141368, chromium:968464
TEST=make clean && make runtests
BRANCH=none
Change-Id: Ic79e7bc233ab93283d2be78ca9a08322b317d11d
Signed-off-by: Joel Kitching <kitching@google.com>
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/platform/vboot_reference/+/1844600
Tested-by: Joel Kitching <kitching@chromium.org>
Reviewed-by: Joel Kitching <kitching@chromium.org>
Commit-Queue: Joel Kitching <kitching@chromium.org>
Diffstat (limited to 'tests/vb20_common2_tests.c')
-rw-r--r-- | tests/vb20_common2_tests.c | 225 |
1 files changed, 0 insertions, 225 deletions
diff --git a/tests/vb20_common2_tests.c b/tests/vb20_common2_tests.c deleted file mode 100644 index 4f68f9b8..00000000 --- a/tests/vb20_common2_tests.c +++ /dev/null @@ -1,225 +0,0 @@ -/* Copyright (c) 2014 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 <stdint.h> -#include <stdio.h> -#include <string.h> - -#include "2rsa.h" -#include "2sysincludes.h" -#include "file_keys.h" -#include "host_common.h" -#include "host_key2.h" -#include "test_common.h" -#include "vb2_common.h" - -static const uint8_t test_data[] = "This is some test data to sign."; -static const uint32_t test_size = sizeof(test_data); - -static void test_unpack_key(const struct vb2_packed_key *key1) -{ - struct vb2_public_key pubk; - - /* - * Key data follows the header for a newly allocated key, so we can - * calculate the buffer size by looking at how far the key data goes. - */ - uint32_t size = key1->key_offset + key1->key_size; - uint8_t *buf = malloc(size); - struct vb2_packed_key *key = (struct vb2_packed_key *)buf; - - memcpy(key, key1, size); - TEST_SUCC(vb2_unpack_key_buffer(&pubk, buf, size), - "vb2_unpack_key_buffer() ok"); - - TEST_EQ(pubk.sig_alg, vb2_crypto_to_signature(key->algorithm), - "vb2_unpack_key_buffer() sig_alg"); - TEST_EQ(pubk.hash_alg, vb2_crypto_to_hash(key->algorithm), - "vb2_unpack_key_buffer() hash_alg"); - - - memcpy(key, key1, size); - key->algorithm = VB2_ALG_COUNT; - TEST_EQ(vb2_unpack_key_buffer(&pubk, buf, size), - VB2_ERROR_UNPACK_KEY_SIG_ALGORITHM, - "vb2_unpack_key_buffer() invalid algorithm"); - - memcpy(key, key1, size); - key->key_size--; - TEST_EQ(vb2_unpack_key_buffer(&pubk, buf, size), - VB2_ERROR_UNPACK_KEY_SIZE, - "vb2_unpack_key_buffer() invalid size"); - - memcpy(key, key1, size); - key->key_offset++; - TEST_EQ(vb2_unpack_key_buffer(&pubk, buf, size + 1), - VB2_ERROR_UNPACK_KEY_ALIGN, - "vb2_unpack_key_buffer() unaligned data"); - - memcpy(key, key1, size); - *(uint32_t *)(buf + key->key_offset) /= 2; - TEST_EQ(vb2_unpack_key_buffer(&pubk, buf, size), - VB2_ERROR_UNPACK_KEY_ARRAY_SIZE, - "vb2_unpack_key_buffer() invalid key array size"); - - memcpy(key, key1, size); - TEST_EQ(vb2_unpack_key_buffer(&pubk, buf, size - 1), - VB2_ERROR_INSIDE_DATA_OUTSIDE, - "vb2_unpack_key_buffer() buffer too small"); - - free(key); - - TEST_EQ(vb2_unpack_key(&pubk, NULL), - VB2_ERROR_UNPACK_KEY_BUFFER, - "vb2_unpack_key_() buffer NULL"); -} - -static void test_verify_data(const struct vb2_packed_key *key1, - const struct vb2_signature *sig) -{ - uint8_t workbuf[VB2_VERIFY_DATA_WORKBUF_BYTES] - __attribute__((aligned(VB2_WORKBUF_ALIGN))); - struct vb2_workbuf wb; - - struct vb2_public_key pubk, pubk_orig; - uint32_t sig_total_size = sig->sig_offset + sig->sig_size; - struct vb2_signature *sig2; - - vb2_workbuf_init(&wb, workbuf, sizeof(workbuf)); - - /* Allocate signature copy for tests */ - sig2 = (struct vb2_signature *)malloc(sig_total_size); - - TEST_SUCC(vb2_unpack_key(&pubk, key1), "vb2_verify_data() unpack key"); - pubk_orig = pubk; - - memcpy(sig2, sig, sig_total_size); - pubk.sig_alg = VB2_SIG_INVALID; - TEST_NEQ(vb2_verify_data(test_data, test_size, sig2, &pubk, &wb), - 0, "vb2_verify_data() bad sig alg"); - pubk.sig_alg = pubk_orig.sig_alg; - - memcpy(sig2, sig, sig_total_size); - pubk.hash_alg = VB2_HASH_INVALID; - TEST_NEQ(vb2_verify_data(test_data, test_size, sig2, &pubk, &wb), - 0, "vb2_verify_data() bad hash alg"); - pubk.hash_alg = pubk_orig.hash_alg; - - vb2_workbuf_init(&wb, workbuf, 4); - memcpy(sig2, sig, sig_total_size); - TEST_NEQ(vb2_verify_data(test_data, test_size, sig2, &pubk, &wb), - 0, "vb2_verify_data() workbuf too small"); - vb2_workbuf_init(&wb, workbuf, sizeof(workbuf)); - - memcpy(sig2, sig, sig_total_size); - TEST_EQ(vb2_verify_data(test_data, test_size, sig2, &pubk, &wb), - 0, "vb2_verify_data() ok"); - - memcpy(sig2, sig, sig_total_size); - sig2->sig_size -= 16; - TEST_NEQ(vb2_verify_data(test_data, test_size, sig2, &pubk, &wb), - 0, "vb2_verify_data() wrong sig size"); - - memcpy(sig2, sig, sig_total_size); - TEST_NEQ(vb2_verify_data(test_data, test_size - 1, sig2, &pubk, &wb), - 0, "vb2_verify_data() input buffer too small"); - - memcpy(sig2, sig, sig_total_size); - vb2_signature_data_mutable(sig2)[0] ^= 0x5A; - TEST_NEQ(vb2_verify_data(test_data, test_size, sig2, &pubk, &wb), - 0, "vb2_verify_data() wrong sig"); - - free(sig2); -} - - -static int test_algorithm(int key_algorithm, const char *keys_dir) -{ - char filename[1024]; - struct vb2_private_key *private_key = NULL; - struct vb2_signature *sig = NULL; - struct vb2_packed_key *key1 = NULL; - - int retval = 1; - - printf("***Testing algorithm: %s\n", - vb2_get_crypto_algorithm_name(key_algorithm)); - - snprintf(filename, sizeof(filename), "%s/key_%s.pem", - keys_dir, - vb2_get_crypto_algorithm_file(key_algorithm)); - private_key = vb2_read_private_key_pem(filename, key_algorithm); - if (!private_key) { - fprintf(stderr, "Error reading private_key: %s\n", filename); - goto cleanup_algorithm; - } - - snprintf(filename, sizeof(filename), "%s/key_%s.keyb", - keys_dir, - vb2_get_crypto_algorithm_file(key_algorithm)); - key1 = vb2_read_packed_keyb(filename, key_algorithm, 1); - if (!key1) { - fprintf(stderr, "Error reading public_key: %s\n", filename); - goto cleanup_algorithm; - } - - /* Calculate good signatures */ - sig = vb2_calculate_signature(test_data, sizeof(test_data), - private_key); - TEST_PTR_NEQ(sig, 0, "Calculate signature"); - if (!sig) - goto cleanup_algorithm; - - test_unpack_key(key1); - test_verify_data(key1, sig); - - retval = 0; - -cleanup_algorithm: - if (key1) - free(key1); - if (private_key) - free(private_key); - if (sig) - free(sig); - - return retval; -} - -/* Test only the algorithms we use */ -const int key_algs[] = { - VB2_ALG_RSA2048_SHA256, - VB2_ALG_RSA4096_SHA256, - VB2_ALG_RSA8192_SHA512, -}; - -int main(int argc, char *argv[]) { - - if (argc == 2) { - int i; - - for (i = 0; i < ARRAY_SIZE(key_algs); i++) { - if (test_algorithm(key_algs[i], argv[1])) - return 1; - } - - } else if (argc == 3 && !strcasecmp(argv[2], "--all")) { - /* Test all the algorithms */ - int alg; - - for (alg = 0; alg < VB2_ALG_COUNT; alg++) { - if (test_algorithm(alg, argv[1])) - return 1; - } - - } else { - fprintf(stderr, "Usage: %s <keys_dir> [--all]", argv[0]); - return -1; - } - - return gTestSuccess ? 0 : 255; -} |