u-boot/test/dm/tee.c
Simon Glass e180c2b129 dm: Rename DM test flags to make them more generic
The test flags used by driver model are currently not available to other
tests. Rather than creating two sets of flags, make these flags generic
by changing the DM_ prefix to UT_ and moving them to the test.h header.

This will allow adding other test flags without confusion.

Signed-off-by: Simon Glass <sjg@chromium.org>
2020-08-07 22:31:32 -04:00

118 lines
2.5 KiB
C

// SPDX-License-Identifier: GPL-2.0+
/*
* Copyright (C) 2018 Linaro Limited
*/
#include <common.h>
#include <dm.h>
#include <log.h>
#include <malloc.h>
#include <dm/test.h>
#include <sandboxtee.h>
#include <tee.h>
#include <test/test.h>
#include <test/ut.h>
#include <tee/optee_ta_avb.h>
static int open_session(struct udevice *dev, u32 *session)
{
struct tee_open_session_arg arg;
const struct tee_optee_ta_uuid uuid = TA_AVB_UUID;
int rc;
memset(&arg, 0, sizeof(arg));
tee_optee_ta_uuid_to_octets(arg.uuid, &uuid);
rc = tee_open_session(dev, &arg, 0, NULL);
if (rc)
return rc;
if (arg.ret)
return -EIO;
*session = arg.session;
return 0;
}
static int invoke_func(struct udevice *dev, u32 session)
{
struct tee_param param = { .attr = TEE_PARAM_ATTR_TYPE_VALUE_OUTPUT };
struct tee_invoke_arg arg;
memset(&arg, 0, sizeof(arg));
arg.session = session;
arg.func = TA_AVB_CMD_READ_LOCK_STATE;
if (tee_invoke_func(dev, &arg, 1, &param) || arg.ret)
return -1;
return 0;
}
static int match(struct tee_version_data *vers, const void *data)
{
return vers->gen_caps & TEE_GEN_CAP_GP;
}
struct test_tee_vars {
struct tee_shm *reg_shm;
struct tee_shm *alloc_shm;
};
static int test_tee(struct unit_test_state *uts, struct test_tee_vars *vars)
{
struct tee_version_data vers;
struct udevice *dev;
struct sandbox_tee_state *state;
u32 session = 0;
int rc;
u8 data[128];
dev = tee_find_device(NULL, match, NULL, &vers);
ut_assert(dev);
state = dev_get_priv(dev);
ut_assert(!state->session);
rc = open_session(dev, &session);
ut_assert(!rc);
ut_assert(session == state->session);
rc = invoke_func(dev, session);
ut_assert(!rc);
rc = tee_close_session(dev, session);
ut_assert(!rc);
ut_assert(!state->session);
ut_assert(!state->num_shms);
rc = tee_shm_register(dev, data, sizeof(data), 0, &vars->reg_shm);
ut_assert(!rc);
ut_assert(state->num_shms == 1);
rc = tee_shm_alloc(dev, 256, 0, &vars->alloc_shm);
ut_assert(!rc);
ut_assert(state->num_shms == 2);
ut_assert(tee_shm_is_registered(vars->reg_shm, dev));
ut_assert(tee_shm_is_registered(vars->alloc_shm, dev));
tee_shm_free(vars->reg_shm);
vars->reg_shm = NULL;
tee_shm_free(vars->alloc_shm);
vars->alloc_shm = NULL;
ut_assert(!state->num_shms);
return 0;
}
static int dm_test_tee(struct unit_test_state *uts)
{
struct test_tee_vars vars = { NULL, NULL };
int rc = test_tee(uts, &vars);
/* In case test_tee() asserts these may still remain allocated */
tee_shm_free(vars.reg_shm);
tee_shm_free(vars.alloc_shm);
return rc;
}
DM_TEST(dm_test_tee, UT_TESTF_SCAN_FDT);