diff options
Diffstat (limited to 'tests/vb2_secdata_kernel_tests.c')
| -rw-r--r-- | tests/vb2_secdata_kernel_tests.c | 81 |
1 files changed, 39 insertions, 42 deletions
diff --git a/tests/vb2_secdata_kernel_tests.c b/tests/vb2_secdata_kernel_tests.c index 89184d99..d4ad3d2f 100644 --- a/tests/vb2_secdata_kernel_tests.c +++ b/tests/vb2_secdata_kernel_tests.c @@ -17,22 +17,19 @@ static uint8_t workbuf[VB2_FIRMWARE_WORKBUF_RECOMMENDED_SIZE] __attribute__ ((aligned (VB2_WORKBUF_ALIGN))); -static struct vb2_context ctx; +static struct vb2_context *ctx; static struct vb2_shared_data *sd; static struct vb2_secdata_kernel *sec; static void reset_common_data(void) { memset(workbuf, 0xaa, sizeof(workbuf)); + TEST_SUCC(vb2api_init(workbuf, sizeof(workbuf), &ctx), + "vb2api_init failed"); - memset(&ctx, 0, sizeof(ctx)); - ctx.workbuf = workbuf; - ctx.workbuf_size = sizeof(workbuf); + sd = vb2_get_sd(ctx); - vb2_init_context(&ctx); - sd = vb2_get_sd(&ctx); - - sec = (struct vb2_secdata_kernel *)ctx.secdata_kernel; + sec = (struct vb2_secdata_kernel *)ctx->secdata_kernel; } static void test_changed(struct vb2_context *c, int changed, const char *why) @@ -55,79 +52,79 @@ static void secdata_kernel_test(void) "Struct size constant"); /* Blank data is invalid */ - memset(ctx.secdata_kernel, 0xa6, sizeof(ctx.secdata_kernel)); - TEST_EQ(vb2api_secdata_kernel_check(&ctx), + memset(&ctx->secdata_kernel, 0xa6, sizeof(ctx->secdata_kernel)); + TEST_EQ(vb2api_secdata_kernel_check(ctx), VB2_ERROR_SECDATA_KERNEL_CRC, "Check blank CRC"); - TEST_EQ(vb2_secdata_kernel_init(&ctx), + TEST_EQ(vb2_secdata_kernel_init(ctx), VB2_ERROR_SECDATA_KERNEL_CRC, "Init blank CRC"); /* Ensure zeroed buffers are invalid */ - memset(ctx.secdata_kernel, 0, sizeof(ctx.secdata_kernel)); - TEST_EQ(vb2_secdata_kernel_init(&ctx), VB2_ERROR_SECDATA_KERNEL_VERSION, + memset(&ctx->secdata_kernel, 0, sizeof(ctx->secdata_kernel)); + TEST_EQ(vb2_secdata_kernel_init(ctx), VB2_ERROR_SECDATA_KERNEL_VERSION, "Zeroed buffer (invalid version)"); /* Try with bad version */ - TEST_EQ(vb2api_secdata_kernel_create(&ctx), VB2_SECDATA_KERNEL_SIZE, + TEST_EQ(vb2api_secdata_kernel_create(ctx), VB2_SECDATA_KERNEL_SIZE, "Create"); sec->struct_version -= 1; sec->crc8 = vb2_crc8(sec, offsetof(struct vb2_secdata_kernel, crc8)); - TEST_EQ(vb2api_secdata_kernel_check(&ctx), + TEST_EQ(vb2api_secdata_kernel_check(ctx), VB2_ERROR_SECDATA_KERNEL_VERSION, "Check invalid version"); - TEST_EQ(vb2_secdata_kernel_init(&ctx), + TEST_EQ(vb2_secdata_kernel_init(ctx), VB2_ERROR_SECDATA_KERNEL_VERSION, "Init invalid version"); /* Create good data */ - vb2api_secdata_kernel_create(&ctx); - TEST_SUCC(vb2api_secdata_kernel_check(&ctx), "Check created CRC"); - TEST_SUCC(vb2_secdata_kernel_init(&ctx), "Init created CRC"); + vb2api_secdata_kernel_create(ctx); + TEST_SUCC(vb2api_secdata_kernel_check(ctx), "Check created CRC"); + TEST_SUCC(vb2_secdata_kernel_init(ctx), "Init created CRC"); TEST_NEQ(sd->status & VB2_SD_STATUS_SECDATA_KERNEL_INIT, 0, "Init set SD status"); sd->status &= ~VB2_SD_STATUS_SECDATA_KERNEL_INIT; - test_changed(&ctx, 1, "Create changes data"); + test_changed(ctx, 1, "Create changes data"); /* Now corrupt it */ - ctx.secdata_kernel[2]++; - TEST_EQ(vb2api_secdata_kernel_check(&ctx), + ctx->secdata_kernel[2]++; + TEST_EQ(vb2api_secdata_kernel_check(ctx), VB2_ERROR_SECDATA_KERNEL_CRC, "Check invalid CRC"); - TEST_EQ(vb2_secdata_kernel_init(&ctx), + TEST_EQ(vb2_secdata_kernel_init(ctx), VB2_ERROR_SECDATA_KERNEL_CRC, "Init invalid CRC"); /* Make sure UID is checked */ - vb2api_secdata_kernel_create(&ctx); + vb2api_secdata_kernel_create(ctx); sec->uid++; sec->crc8 = vb2_crc8(sec, offsetof(struct vb2_secdata_kernel, crc8)); - TEST_EQ(vb2_secdata_kernel_init(&ctx), VB2_ERROR_SECDATA_KERNEL_UID, + TEST_EQ(vb2_secdata_kernel_init(ctx), VB2_ERROR_SECDATA_KERNEL_UID, "Init invalid struct UID"); /* Read/write versions */ - vb2api_secdata_kernel_create(&ctx); - vb2_secdata_kernel_init(&ctx); - ctx.flags = 0; - v = vb2_secdata_kernel_get(&ctx, VB2_SECDATA_KERNEL_VERSIONS); + vb2api_secdata_kernel_create(ctx); + vb2_secdata_kernel_init(ctx); + ctx->flags = 0; + v = vb2_secdata_kernel_get(ctx, VB2_SECDATA_KERNEL_VERSIONS); TEST_EQ(v, 0, "Versions created 0"); - test_changed(&ctx, 0, "Get doesn't change data"); - vb2_secdata_kernel_set(&ctx, VB2_SECDATA_KERNEL_VERSIONS, 0x123456ff); - test_changed(&ctx, 1, "Set changes data"); - vb2_secdata_kernel_set(&ctx, VB2_SECDATA_KERNEL_VERSIONS, 0x123456ff); - test_changed(&ctx, 0, "Set again doesn't change data"); - v = vb2_secdata_kernel_get(&ctx, VB2_SECDATA_KERNEL_VERSIONS); + test_changed(ctx, 0, "Get doesn't change data"); + vb2_secdata_kernel_set(ctx, VB2_SECDATA_KERNEL_VERSIONS, 0x123456ff); + test_changed(ctx, 1, "Set changes data"); + vb2_secdata_kernel_set(ctx, VB2_SECDATA_KERNEL_VERSIONS, 0x123456ff); + test_changed(ctx, 0, "Set again doesn't change data"); + v = vb2_secdata_kernel_get(ctx, VB2_SECDATA_KERNEL_VERSIONS); TEST_EQ(v, 0x123456ff, "Versions changed"); /* Invalid field fails */ - TEST_ABORT(vb2_secdata_kernel_get(&ctx, -1), "Get invalid"); - TEST_ABORT(vb2_secdata_kernel_set(&ctx, -1, 456), "Set invalid"); - test_changed(&ctx, 0, "Set invalid field doesn't change data"); + TEST_ABORT(vb2_secdata_kernel_get(ctx, -1), "Get invalid"); + TEST_ABORT(vb2_secdata_kernel_set(ctx, -1, 456), "Set invalid"); + test_changed(ctx, 0, "Set invalid field doesn't change data"); /* Read/write uninitialized data fails */ sd->status &= ~VB2_SD_STATUS_SECDATA_KERNEL_INIT; - TEST_ABORT(vb2_secdata_kernel_get(&ctx, VB2_SECDATA_KERNEL_VERSIONS), + TEST_ABORT(vb2_secdata_kernel_get(ctx, VB2_SECDATA_KERNEL_VERSIONS), "Get uninitialized"); - test_changed(&ctx, 0, "Get uninitialized doesn't change data"); - TEST_ABORT(vb2_secdata_kernel_set(&ctx, VB2_SECDATA_KERNEL_VERSIONS, + test_changed(ctx, 0, "Get uninitialized doesn't change data"); + TEST_ABORT(vb2_secdata_kernel_set(ctx, VB2_SECDATA_KERNEL_VERSIONS, 0x123456ff), "Set uninitialized"); - test_changed(&ctx, 0, "Set uninitialized doesn't change data"); + test_changed(ctx, 0, "Set uninitialized doesn't change data"); } int main(int argc, char* argv[]) |