diff options
Diffstat (limited to 'tests/vb2_host_misc_tests.c')
-rw-r--r-- | tests/vb2_host_misc_tests.c | 71 |
1 files changed, 71 insertions, 0 deletions
diff --git a/tests/vb2_host_misc_tests.c b/tests/vb2_host_misc_tests.c new file mode 100644 index 00000000..e8c2a588 --- /dev/null +++ b/tests/vb2_host_misc_tests.c @@ -0,0 +1,71 @@ +/* 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 host misc library vboot2 functions + */ + +#include <unistd.h> + +#include "2sysincludes.h" +#include "2common.h" +#include "host_common.h" +#include "host_misc.h" + +#include "test_common.h" + +static void misc_tests(void) +{ + TEST_EQ(roundup32(0), 0, "roundup32(0)"); + TEST_EQ(roundup32(15), 16, "roundup32(15)"); + TEST_EQ(roundup32(16), 16, "roundup32(16)"); +} + +static void file_tests(void) +{ + const char *testfile = "file_tests.dat"; + const uint8_t test_data[] = "Some test data"; + uint8_t *read_data; + uint32_t read_size; + + uint8_t cbuf[sizeof(struct vb2_struct_common) + 12]; + struct vb2_struct_common *c = (struct vb2_struct_common *)cbuf; + + unlink(testfile); + + TEST_EQ(vb2_read_file(testfile, &read_data, &read_size), + VB2_ERROR_READ_FILE_OPEN, "vb2_read_file() missing"); + TEST_EQ(vb2_write_file("no/such/dir", test_data, sizeof(test_data)), + VB2_ERROR_WRITE_FILE_OPEN, "vb2_write_file() open"); + + TEST_SUCC(vb2_write_file(testfile, test_data, sizeof(test_data)), + "vb2_write_file() good"); + TEST_SUCC(vb2_read_file(testfile, &read_data, &read_size), + "vb2_read_file() good"); + TEST_EQ(read_size, sizeof(test_data), " data size"); + TEST_EQ(memcmp(read_data, test_data, read_size), 0, " data"); + free(read_data); + unlink(testfile); + + memset(cbuf, 0, sizeof(cbuf)); + c->fixed_size = sizeof(*c); + c->total_size = sizeof(cbuf); + c->magic = 0x1234; + cbuf[sizeof(cbuf) - 1] = 0xed; /* Some non-zero data at the end */ + TEST_SUCC(vb2_write_object(testfile, c), "vb2_write_object() good"); + TEST_SUCC(vb2_read_file(testfile, &read_data, &read_size), + "vb2_read_file() object"); + TEST_EQ(read_size, c->total_size, " data size"); + /* Compare the entire buffer, including the non-zero data at the end */ + TEST_EQ(memcmp(read_data, c, read_size), 0, " data"); + free(read_data); + unlink(testfile); +} + +int main(int argc, char* argv[]) +{ + misc_tests(); + file_tests(); + + return gTestSuccess ? 0 : 255; +} |