diff options
Diffstat (limited to 'tests/vb2_secdata_firmware_tests.c')
-rw-r--r-- | tests/vb2_secdata_firmware_tests.c | 93 |
1 files changed, 45 insertions, 48 deletions
diff --git a/tests/vb2_secdata_firmware_tests.c b/tests/vb2_secdata_firmware_tests.c index 84cb2741..68a5ce21 100644 --- a/tests/vb2_secdata_firmware_tests.c +++ b/tests/vb2_secdata_firmware_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_firmware *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_firmware *)ctx.secdata_firmware; + sec = (struct vb2_secdata_firmware *)ctx->secdata_firmware; } static void test_changed(struct vb2_context *c, int changed, const char *why) @@ -57,89 +54,89 @@ static void secdata_firmware_test(void) "Struct size constant"); /* Blank data is invalid */ - memset(ctx.secdata_firmware, 0xa6, sizeof(ctx.secdata_firmware)); - TEST_EQ(vb2api_secdata_firmware_check(&ctx), + memset(&ctx->secdata_firmware, 0xa6, sizeof(ctx->secdata_firmware)); + TEST_EQ(vb2api_secdata_firmware_check(ctx), VB2_ERROR_SECDATA_FIRMWARE_CRC, "Check blank CRC"); - TEST_EQ(vb2_secdata_firmware_init(&ctx), + TEST_EQ(vb2_secdata_firmware_init(ctx), VB2_ERROR_SECDATA_FIRMWARE_CRC, "Init blank CRC"); /* Ensure zeroed buffers are invalid (coreboot relies on this) */ - memset(ctx.secdata_firmware, 0, sizeof(ctx.secdata_firmware)); - TEST_EQ(vb2_secdata_firmware_init(&ctx), + memset(&ctx->secdata_firmware, 0, sizeof(ctx->secdata_firmware)); + TEST_EQ(vb2_secdata_firmware_init(ctx), VB2_ERROR_SECDATA_FIRMWARE_VERSION, "Zeroed buffer (invalid version)"); /* Try with bad version */ - TEST_EQ(vb2api_secdata_firmware_create(&ctx), VB2_SECDATA_FIRMWARE_SIZE, + TEST_EQ(vb2api_secdata_firmware_create(ctx), VB2_SECDATA_FIRMWARE_SIZE, "Create"); sec->struct_version -= 1; sec->crc8 = vb2_crc8(sec, offsetof(struct vb2_secdata_firmware, crc8)); - TEST_EQ(vb2api_secdata_firmware_check(&ctx), + TEST_EQ(vb2api_secdata_firmware_check(ctx), VB2_ERROR_SECDATA_FIRMWARE_VERSION, "Check invalid version"); - TEST_EQ(vb2_secdata_firmware_init(&ctx), + TEST_EQ(vb2_secdata_firmware_init(ctx), VB2_ERROR_SECDATA_FIRMWARE_VERSION, "Init invalid version"); /* Create good data */ - vb2api_secdata_firmware_create(&ctx); - TEST_SUCC(vb2api_secdata_firmware_check(&ctx), "Check created CRC"); - TEST_SUCC(vb2_secdata_firmware_init(&ctx), "Init created CRC"); + vb2api_secdata_firmware_create(ctx); + TEST_SUCC(vb2api_secdata_firmware_check(ctx), "Check created CRC"); + TEST_SUCC(vb2_secdata_firmware_init(ctx), "Init created CRC"); TEST_NEQ(sd->status & VB2_SD_STATUS_SECDATA_FIRMWARE_INIT, 0, "Init set SD status"); sd->status &= ~VB2_SD_STATUS_SECDATA_FIRMWARE_INIT; - test_changed(&ctx, 1, "Create changes data"); + test_changed(ctx, 1, "Create changes data"); /* Now corrupt it */ - ctx.secdata_firmware[2]++; - TEST_EQ(vb2api_secdata_firmware_check(&ctx), + ctx->secdata_firmware[2]++; + TEST_EQ(vb2api_secdata_firmware_check(ctx), VB2_ERROR_SECDATA_FIRMWARE_CRC, "Check invalid CRC"); - TEST_EQ(vb2_secdata_firmware_init(&ctx), + TEST_EQ(vb2_secdata_firmware_init(ctx), VB2_ERROR_SECDATA_FIRMWARE_CRC, "Init invalid CRC"); /* Read/write flags */ - vb2api_secdata_firmware_create(&ctx); - vb2_secdata_firmware_init(&ctx); - ctx.flags = 0; - v = vb2_secdata_firmware_get(&ctx, VB2_SECDATA_FIRMWARE_FLAGS); + vb2api_secdata_firmware_create(ctx); + vb2_secdata_firmware_init(ctx); + ctx->flags = 0; + v = vb2_secdata_firmware_get(ctx, VB2_SECDATA_FIRMWARE_FLAGS); TEST_EQ(v, 0, "Flags created 0"); - test_changed(&ctx, 0, "Get doesn't change data"); - vb2_secdata_firmware_set(&ctx, VB2_SECDATA_FIRMWARE_FLAGS, 0x12); - test_changed(&ctx, 1, "Set changes data"); - vb2_secdata_firmware_set(&ctx, VB2_SECDATA_FIRMWARE_FLAGS, 0x12); - test_changed(&ctx, 0, "Set again doesn't change data"); - v = vb2_secdata_firmware_get(&ctx, VB2_SECDATA_FIRMWARE_FLAGS); + test_changed(ctx, 0, "Get doesn't change data"); + vb2_secdata_firmware_set(ctx, VB2_SECDATA_FIRMWARE_FLAGS, 0x12); + test_changed(ctx, 1, "Set changes data"); + vb2_secdata_firmware_set(ctx, VB2_SECDATA_FIRMWARE_FLAGS, 0x12); + test_changed(ctx, 0, "Set again doesn't change data"); + v = vb2_secdata_firmware_get(ctx, VB2_SECDATA_FIRMWARE_FLAGS); TEST_EQ(v, 0x12, "Flags changed"); - TEST_ABORT(vb2_secdata_firmware_set(&ctx, VB2_SECDATA_FIRMWARE_FLAGS, + TEST_ABORT(vb2_secdata_firmware_set(ctx, VB2_SECDATA_FIRMWARE_FLAGS, 0x100), "Bad flags"); /* Read/write versions */ - v = vb2_secdata_firmware_get(&ctx, VB2_SECDATA_FIRMWARE_VERSIONS); + v = vb2_secdata_firmware_get(ctx, VB2_SECDATA_FIRMWARE_VERSIONS); TEST_EQ(v, 0, "Versions created 0"); - test_changed(&ctx, 0, "Get doesn't change data"); - vb2_secdata_firmware_set(&ctx, VB2_SECDATA_FIRMWARE_VERSIONS, + test_changed(ctx, 0, "Get doesn't change data"); + vb2_secdata_firmware_set(ctx, VB2_SECDATA_FIRMWARE_VERSIONS, 0x123456ff); - test_changed(&ctx, 1, "Set changes data"); - vb2_secdata_firmware_set(&ctx, VB2_SECDATA_FIRMWARE_VERSIONS, + test_changed(ctx, 1, "Set changes data"); + vb2_secdata_firmware_set(ctx, VB2_SECDATA_FIRMWARE_VERSIONS, 0x123456ff); - test_changed(&ctx, 0, "Set again doesn't change data"); - v = vb2_secdata_firmware_get(&ctx, VB2_SECDATA_FIRMWARE_VERSIONS); + test_changed(ctx, 0, "Set again doesn't change data"); + v = vb2_secdata_firmware_get(ctx, VB2_SECDATA_FIRMWARE_VERSIONS); TEST_EQ(v, 0x123456ff, "Versions changed"); /* Invalid field fails */ - TEST_ABORT(vb2_secdata_firmware_get(&ctx, -1), "Get invalid"); - TEST_ABORT(vb2_secdata_firmware_set(&ctx, -1, 456), "Set invalid"); - test_changed(&ctx, 0, "Set invalid field doesn't change data"); + TEST_ABORT(vb2_secdata_firmware_get(ctx, -1), "Get invalid"); + TEST_ABORT(vb2_secdata_firmware_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_FIRMWARE_INIT; - TEST_ABORT(vb2_secdata_firmware_get(&ctx, + TEST_ABORT(vb2_secdata_firmware_get(ctx, VB2_SECDATA_FIRMWARE_VERSIONS), "Get uninitialized"); - test_changed(&ctx, 0, "Get uninitialized doesn't change data"); - TEST_ABORT(vb2_secdata_firmware_set(&ctx, VB2_SECDATA_FIRMWARE_VERSIONS, + test_changed(ctx, 0, "Get uninitialized doesn't change data"); + TEST_ABORT(vb2_secdata_firmware_set(ctx, VB2_SECDATA_FIRMWARE_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[]) |