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" 18 #include "2misc.h" 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 { 33 uint8_t workbuf[VB2_WORKBUF_RECOMMENDED_SIZE] 34 __attribute__ ((aligned (VB2_WORKBUF_ALIGN))); 35 struct vb2_context c = { 36 .flags = 0, 37 .workbuf = workbuf, 38 .workbuf_size = sizeof(workbuf), 39 }; 40 uint32_t v = 1; 41 42 /* Blank data is invalid */ 43 memset(c.secdata, 0xa6, sizeof(c.secdata)); 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"); 48 49 /* Create good data */ 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"); 53 test_changed(&c, 1, "Create changes data"); 54 55 /* Now corrupt it */ 56 c.secdata[2]++; 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"); 61 62 vb2_secdata_create(&c); 63 c.flags = 0; 64 65 /* Read/write flags */ 66 TEST_SUCC(vb2_secdata_get(&c, VB2_SECDATA_FLAGS, &v), "Get flags"); 67 TEST_EQ(v, 0, "Flags created 0"); 68 test_changed(&c, 0, "Get doesn't change data"); 69 TEST_SUCC(vb2_secdata_set(&c, VB2_SECDATA_FLAGS, 0x12), "Set flags"); 70 test_changed(&c, 1, "Set changes data"); 71 TEST_SUCC(vb2_secdata_set(&c, VB2_SECDATA_FLAGS, 0x12), "Set flags 2"); 72 test_changed(&c, 0, "Set again doesn't change data"); 73 TEST_SUCC(vb2_secdata_get(&c, VB2_SECDATA_FLAGS, &v), "Get flags 2"); 74 TEST_EQ(v, 0x12, "Flags changed"); 75 TEST_EQ(vb2_secdata_set(&c, VB2_SECDATA_FLAGS, 0x100), 76 VB2_ERROR_SECDATA_SET_FLAGS, "Bad flags"); 77 78 /* Read/write versions */ 79 TEST_SUCC(vb2_secdata_get(&c, VB2_SECDATA_VERSIONS, &v), 80 "Get versions"); 81 TEST_EQ(v, 0, "Versions created 0"); 82 test_changed(&c, 0, "Get doesn't change data"); 83 TEST_SUCC(vb2_secdata_set(&c, VB2_SECDATA_VERSIONS, 0x123456ff), 84 "Set versions"); 85 test_changed(&c, 1, "Set changes data"); 86 TEST_SUCC(vb2_secdata_set(&c, VB2_SECDATA_VERSIONS, 0x123456ff), 87 "Set versions 2"); 88 test_changed(&c, 0, "Set again doesn't change data"); 89 TEST_SUCC(vb2_secdata_get(&c, VB2_SECDATA_VERSIONS, &v), 90 "Get versions 2"); 91 TEST_EQ(v, 0x123456ff, "Versions changed"); 92 93 /* Invalid field fails */ 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"); 98 test_changed(&c, 0, "Set invalid field doesn't change data"); 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"); 108 } 109 110 int main(int argc, char* argv[]) 111 { 112 secdata_test(); 113 114 return gTestSuccess ? 0 : 255; 115 } 116
