Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 1 | /* Copyright (c) 2014 The Chromium OS Authors. All rights reserved. |
| 2 | * Use of this source code is governed by a BSD-style license that can be |
| 3 | * found in the LICENSE file. |
| 4 | * |
| 5 | * Tests for firmware secure storage library. |
| 6 | */ |
| 7 | |
| 8 | #include <stdint.h> |
| 9 | #include <stdio.h> |
| 10 | #include <stdlib.h> |
| 11 | #include <string.h> |
| 12 | |
| 13 | #include "test_common.h" |
| 14 | #include "vboot_common.h" |
| 15 | |
| 16 | #include "2common.h" |
| 17 | #include "2api.h" |
Julius Werner | b550fb1 | 2015-01-30 14:27:54 -0800 | [diff] [blame] | 18 | #include "2misc.h" |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 19 | #include "2secdata.h" |
| 20 | |
| 21 | static void test_changed(struct vb2_context *ctx, int changed, const char *why) |
| 22 | { |
| 23 | if (changed) |
| 24 | TEST_NEQ(ctx->flags & VB2_CONTEXT_SECDATA_CHANGED, 0, why); |
| 25 | else |
| 26 | TEST_EQ(ctx->flags & VB2_CONTEXT_SECDATA_CHANGED, 0, why); |
| 27 | |
| 28 | ctx->flags &= ~VB2_CONTEXT_SECDATA_CHANGED; |
| 29 | }; |
| 30 | |
| 31 | static void secdata_test(void) |
| 32 | { |
Bill Richardson | 73e5eb3 | 2015-01-26 12:18:25 -0800 | [diff] [blame] | 33 | uint8_t workbuf[VB2_WORKBUF_RECOMMENDED_SIZE] |
| 34 | __attribute__ ((aligned (VB2_WORKBUF_ALIGN))); |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 35 | struct vb2_context c = { |
| 36 | .flags = 0, |
| 37 | .workbuf = workbuf, |
| 38 | .workbuf_size = sizeof(workbuf), |
| 39 | }; |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 40 | uint32_t v = 1; |
| 41 | |
| 42 | /* Blank data is invalid */ |
| 43 | memset(c.secdata, 0xa6, sizeof(c.secdata)); |
Randall Spangler | b9be536 | 2014-06-05 13:32:11 -0700 | [diff] [blame] | 44 | TEST_EQ(vb2_secdata_check_crc(&c), |
| 45 | VB2_ERROR_SECDATA_CRC, "Check blank CRC"); |
| 46 | TEST_EQ(vb2_secdata_init(&c), |
| 47 | VB2_ERROR_SECDATA_CRC, "Init blank CRC"); |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 48 | |
| 49 | /* Create good data */ |
Randall Spangler | b9be536 | 2014-06-05 13:32:11 -0700 | [diff] [blame] | 50 | TEST_SUCC(vb2_secdata_create(&c), "Create"); |
| 51 | TEST_SUCC(vb2_secdata_check_crc(&c), "Check created CRC"); |
| 52 | TEST_SUCC(vb2_secdata_init(&c), "Init created CRC"); |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 53 | test_changed(&c, 1, "Create changes data"); |
| 54 | |
| 55 | /* Now corrupt it */ |
| 56 | c.secdata[2]++; |
Randall Spangler | b9be536 | 2014-06-05 13:32:11 -0700 | [diff] [blame] | 57 | TEST_EQ(vb2_secdata_check_crc(&c), |
| 58 | VB2_ERROR_SECDATA_CRC, "Check invalid CRC"); |
| 59 | TEST_EQ(vb2_secdata_init(&c), |
| 60 | VB2_ERROR_SECDATA_CRC, "Init invalid CRC"); |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 61 | |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 62 | vb2_secdata_create(&c); |
| 63 | c.flags = 0; |
| 64 | |
| 65 | /* Read/write flags */ |
Randall Spangler | b9be536 | 2014-06-05 13:32:11 -0700 | [diff] [blame] | 66 | TEST_SUCC(vb2_secdata_get(&c, VB2_SECDATA_FLAGS, &v), "Get flags"); |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 67 | TEST_EQ(v, 0, "Flags created 0"); |
| 68 | test_changed(&c, 0, "Get doesn't change data"); |
Randall Spangler | b9be536 | 2014-06-05 13:32:11 -0700 | [diff] [blame] | 69 | TEST_SUCC(vb2_secdata_set(&c, VB2_SECDATA_FLAGS, 0x12), "Set flags"); |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 70 | test_changed(&c, 1, "Set changes data"); |
Randall Spangler | b9be536 | 2014-06-05 13:32:11 -0700 | [diff] [blame] | 71 | TEST_SUCC(vb2_secdata_set(&c, VB2_SECDATA_FLAGS, 0x12), "Set flags 2"); |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 72 | test_changed(&c, 0, "Set again doesn't change data"); |
Randall Spangler | b9be536 | 2014-06-05 13:32:11 -0700 | [diff] [blame] | 73 | TEST_SUCC(vb2_secdata_get(&c, VB2_SECDATA_FLAGS, &v), "Get flags 2"); |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 74 | TEST_EQ(v, 0x12, "Flags changed"); |
Randall Spangler | b9be536 | 2014-06-05 13:32:11 -0700 | [diff] [blame] | 75 | TEST_EQ(vb2_secdata_set(&c, VB2_SECDATA_FLAGS, 0x100), |
| 76 | VB2_ERROR_SECDATA_SET_FLAGS, "Bad flags"); |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 77 | |
| 78 | /* Read/write versions */ |
Randall Spangler | b9be536 | 2014-06-05 13:32:11 -0700 | [diff] [blame] | 79 | TEST_SUCC(vb2_secdata_get(&c, VB2_SECDATA_VERSIONS, &v), |
| 80 | "Get versions"); |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 81 | TEST_EQ(v, 0, "Versions created 0"); |
| 82 | test_changed(&c, 0, "Get doesn't change data"); |
Randall Spangler | b9be536 | 2014-06-05 13:32:11 -0700 | [diff] [blame] | 83 | TEST_SUCC(vb2_secdata_set(&c, VB2_SECDATA_VERSIONS, 0x123456ff), |
| 84 | "Set versions"); |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 85 | test_changed(&c, 1, "Set changes data"); |
Randall Spangler | b9be536 | 2014-06-05 13:32:11 -0700 | [diff] [blame] | 86 | TEST_SUCC(vb2_secdata_set(&c, VB2_SECDATA_VERSIONS, 0x123456ff), |
| 87 | "Set versions 2"); |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 88 | test_changed(&c, 0, "Set again doesn't change data"); |
Randall Spangler | b9be536 | 2014-06-05 13:32:11 -0700 | [diff] [blame] | 89 | TEST_SUCC(vb2_secdata_get(&c, VB2_SECDATA_VERSIONS, &v), |
| 90 | "Get versions 2"); |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 91 | TEST_EQ(v, 0x123456ff, "Versions changed"); |
| 92 | |
| 93 | /* Invalid field fails */ |
Randall Spangler | b9be536 | 2014-06-05 13:32:11 -0700 | [diff] [blame] | 94 | TEST_EQ(vb2_secdata_get(&c, -1, &v), |
| 95 | VB2_ERROR_SECDATA_GET_PARAM, "Get invalid"); |
| 96 | TEST_EQ(vb2_secdata_set(&c, -1, 456), |
| 97 | VB2_ERROR_SECDATA_SET_PARAM, "Set invalid"); |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 98 | test_changed(&c, 0, "Set invalid field doesn't change data"); |
Julius Werner | b550fb1 | 2015-01-30 14:27:54 -0800 | [diff] [blame] | 99 | |
| 100 | /* Read/write uninitialized data fails */ |
| 101 | vb2_get_sd(&c)->status &= ~VB2_SD_STATUS_SECDATA_INIT; |
| 102 | TEST_EQ(vb2_secdata_get(&c, VB2_SECDATA_VERSIONS, &v), |
| 103 | VB2_ERROR_SECDATA_GET_UNINITIALIZED, "Get uninitialized"); |
| 104 | test_changed(&c, 0, "Get uninitialized doesn't change data"); |
| 105 | TEST_EQ(vb2_secdata_set(&c, VB2_SECDATA_VERSIONS, 0x123456ff), |
| 106 | VB2_ERROR_SECDATA_SET_UNINITIALIZED, "Set uninitialized"); |
| 107 | test_changed(&c, 0, "Set uninitialized doesn't change data"); |
Randall Spangler | 3333e57 | 2014-05-14 11:37:52 -0700 | [diff] [blame] | 108 | } |
| 109 | |
| 110 | int main(int argc, char* argv[]) |
| 111 | { |
| 112 | secdata_test(); |
| 113 | |
| 114 | return gTestSuccess ? 0 : 255; |
| 115 | } |