1 // SPDX-License-Identifier: GPL-2.0+
14 static int open_session(struct udevice *dev, u32 *session)  in open_session()  argument
17 	const struct tee_optee_ta_uuid uuid = TA_AVB_UUID;  in open_session()  local
21 	tee_optee_ta_uuid_to_octets(arg.uuid, &uuid);  in open_session()
22 	rc = tee_open_session(dev, &arg, 0, NULL);  in open_session()
26 		return -EIO;  in open_session()
32 static int invoke_func(struct udevice *dev, u32 session)  in invoke_func()  argument
41 	if (tee_invoke_func(dev, &arg, 1, &param) || arg.ret)  in invoke_func()
42 		return -1;  in invoke_func()
49 	return vers->gen_caps & TEE_GEN_CAP_GP;  in match()
60 	struct udevice *dev;  in test_tee()  local
66 	dev = tee_find_device(NULL, match, NULL, &vers);  in test_tee()
67 	ut_assert(dev);  in test_tee()
68 	state = dev_get_priv(dev);  in test_tee()
69 	ut_assert(!state->session);  in test_tee()
71 	rc = open_session(dev, &session);  in test_tee()
73 	ut_assert(session == state->session);  in test_tee()
75 	rc = invoke_func(dev, session);  in test_tee()
78 	rc = tee_close_session(dev, session);  in test_tee()
80 	ut_assert(!state->session);  in test_tee()
82 	ut_assert(!state->num_shms);  in test_tee()
83 	rc = tee_shm_register(dev, data, sizeof(data), 0, &vars->reg_shm);  in test_tee()
85 	ut_assert(state->num_shms == 1);  in test_tee()
87 	rc = tee_shm_alloc(dev, 256, 0, &vars->alloc_shm);  in test_tee()
89 	ut_assert(state->num_shms == 2);  in test_tee()
91 	ut_assert(tee_shm_is_registered(vars->reg_shm, dev));  in test_tee()
92 	ut_assert(tee_shm_is_registered(vars->alloc_shm, dev));  in test_tee()
94 	tee_shm_free(vars->reg_shm);  in test_tee()
95 	vars->reg_shm = NULL;  in test_tee()
96 	tee_shm_free(vars->alloc_shm);  in test_tee()
97 	vars->alloc_shm = NULL;  in test_tee()
98 	ut_assert(!state->num_shms);  in test_tee()