Loading...
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 | // 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/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, ¶m) || 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, DM_TESTF_SCAN_FDT); |