core.c 16 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654
  1. // SPDX-License-Identifier: GPL-2.0+
  2. /*
  3. * Copyright (c) 2018 Linaro Limited
  4. */
  5. #include <common.h>
  6. #include <dm.h>
  7. #include <log.h>
  8. #include <tee.h>
  9. #include <linux/arm-smccc.h>
  10. #include <linux/io.h>
  11. #include "optee_smc.h"
  12. #include "optee_msg.h"
  13. #include "optee_private.h"
  14. #define PAGELIST_ENTRIES_PER_PAGE \
  15. ((OPTEE_MSG_NONCONTIG_PAGE_SIZE / sizeof(u64)) - 1)
  16. typedef void (optee_invoke_fn)(unsigned long, unsigned long, unsigned long,
  17. unsigned long, unsigned long, unsigned long,
  18. unsigned long, unsigned long,
  19. struct arm_smccc_res *);
  20. struct optee_pdata {
  21. optee_invoke_fn *invoke_fn;
  22. };
  23. struct rpc_param {
  24. u32 a0;
  25. u32 a1;
  26. u32 a2;
  27. u32 a3;
  28. u32 a4;
  29. u32 a5;
  30. u32 a6;
  31. u32 a7;
  32. };
  33. /**
  34. * reg_pair_to_ptr() - Make a pointer of 2 32-bit values
  35. * @reg0: High bits of the pointer
  36. * @reg1: Low bits of the pointer
  37. *
  38. * Returns the combined result, note that if a pointer is 32-bit wide @reg0
  39. * will be discarded.
  40. */
  41. static void *reg_pair_to_ptr(u32 reg0, u32 reg1)
  42. {
  43. return (void *)(ulong)(((u64)reg0 << 32) | reg1);
  44. }
  45. /**
  46. * reg_pair_from_64() - Split a 64-bit value into two 32-bit values
  47. * @reg0: High bits of @val
  48. * @reg1: Low bits of @val
  49. * @val: The value to split
  50. */
  51. static void reg_pair_from_64(u32 *reg0, u32 *reg1, u64 val)
  52. {
  53. *reg0 = val >> 32;
  54. *reg1 = val;
  55. }
  56. /**
  57. * optee_alloc_and_init_page_list() - Provide page list of memory buffer
  58. * @buf: Start of buffer
  59. * @len: Length of buffer
  60. * @phys_buf_ptr Physical pointer with coded offset to page list
  61. *
  62. * Secure world doesn't share mapping with Normal world (U-Boot in this case)
  63. * so physical pointers are needed when sharing pointers.
  64. *
  65. * Returns a pointer page list on success or NULL on failure
  66. */
  67. void *optee_alloc_and_init_page_list(void *buf, ulong len, u64 *phys_buf_ptr)
  68. {
  69. const unsigned int page_size = OPTEE_MSG_NONCONTIG_PAGE_SIZE;
  70. const phys_addr_t page_mask = page_size - 1;
  71. u8 *buf_base;
  72. unsigned int page_offset;
  73. unsigned int num_pages;
  74. unsigned int list_size;
  75. unsigned int n;
  76. void *page_list;
  77. struct {
  78. u64 pages_list[PAGELIST_ENTRIES_PER_PAGE];
  79. u64 next_page_data;
  80. } *pages_data;
  81. /*
  82. * A Memory buffer is described in chunks of 4k. The list of
  83. * physical addresses has to be represented by a physical pointer
  84. * too and a single list has to start at a 4k page and fit into
  85. * that page. In order to be able to describe large memory buffers
  86. * these 4k pages carrying physical addresses are linked together
  87. * in a list. See OPTEE_MSG_ATTR_NONCONTIG in
  88. * drivers/tee/optee/optee_msg.h for more information.
  89. */
  90. page_offset = (ulong)buf & page_mask;
  91. num_pages = roundup(page_offset + len, page_size) / page_size;
  92. list_size = DIV_ROUND_UP(num_pages, PAGELIST_ENTRIES_PER_PAGE) *
  93. page_size;
  94. page_list = memalign(page_size, list_size);
  95. if (!page_list)
  96. return NULL;
  97. pages_data = page_list;
  98. buf_base = (u8 *)rounddown((ulong)buf, page_size);
  99. n = 0;
  100. while (num_pages) {
  101. pages_data->pages_list[n] = virt_to_phys(buf_base);
  102. n++;
  103. buf_base += page_size;
  104. num_pages--;
  105. if (n == PAGELIST_ENTRIES_PER_PAGE) {
  106. pages_data->next_page_data =
  107. virt_to_phys(pages_data + 1);
  108. pages_data++;
  109. n = 0;
  110. }
  111. }
  112. *phys_buf_ptr = virt_to_phys(page_list) | page_offset;
  113. return page_list;
  114. }
  115. static void optee_get_version(struct udevice *dev,
  116. struct tee_version_data *vers)
  117. {
  118. struct tee_version_data v = {
  119. .gen_caps = TEE_GEN_CAP_GP | TEE_GEN_CAP_REG_MEM,
  120. };
  121. *vers = v;
  122. }
  123. static int get_msg_arg(struct udevice *dev, uint num_params,
  124. struct tee_shm **shmp, struct optee_msg_arg **msg_arg)
  125. {
  126. int rc;
  127. struct optee_msg_arg *ma;
  128. rc = __tee_shm_add(dev, OPTEE_MSG_NONCONTIG_PAGE_SIZE, NULL,
  129. OPTEE_MSG_GET_ARG_SIZE(num_params), TEE_SHM_ALLOC,
  130. shmp);
  131. if (rc)
  132. return rc;
  133. ma = (*shmp)->addr;
  134. memset(ma, 0, OPTEE_MSG_GET_ARG_SIZE(num_params));
  135. ma->num_params = num_params;
  136. *msg_arg = ma;
  137. return 0;
  138. }
  139. static int to_msg_param(struct optee_msg_param *msg_params, uint num_params,
  140. const struct tee_param *params)
  141. {
  142. uint n;
  143. for (n = 0; n < num_params; n++) {
  144. const struct tee_param *p = params + n;
  145. struct optee_msg_param *mp = msg_params + n;
  146. switch (p->attr) {
  147. case TEE_PARAM_ATTR_TYPE_NONE:
  148. mp->attr = OPTEE_MSG_ATTR_TYPE_NONE;
  149. memset(&mp->u, 0, sizeof(mp->u));
  150. break;
  151. case TEE_PARAM_ATTR_TYPE_VALUE_INPUT:
  152. case TEE_PARAM_ATTR_TYPE_VALUE_OUTPUT:
  153. case TEE_PARAM_ATTR_TYPE_VALUE_INOUT:
  154. mp->attr = OPTEE_MSG_ATTR_TYPE_VALUE_INPUT + p->attr -
  155. TEE_PARAM_ATTR_TYPE_VALUE_INPUT;
  156. mp->u.value.a = p->u.value.a;
  157. mp->u.value.b = p->u.value.b;
  158. mp->u.value.c = p->u.value.c;
  159. break;
  160. case TEE_PARAM_ATTR_TYPE_MEMREF_INPUT:
  161. case TEE_PARAM_ATTR_TYPE_MEMREF_OUTPUT:
  162. case TEE_PARAM_ATTR_TYPE_MEMREF_INOUT:
  163. mp->attr = OPTEE_MSG_ATTR_TYPE_RMEM_INPUT + p->attr -
  164. TEE_PARAM_ATTR_TYPE_MEMREF_INPUT;
  165. mp->u.rmem.shm_ref = (ulong)p->u.memref.shm;
  166. mp->u.rmem.size = p->u.memref.size;
  167. mp->u.rmem.offs = p->u.memref.shm_offs;
  168. break;
  169. default:
  170. return -EINVAL;
  171. }
  172. }
  173. return 0;
  174. }
  175. static int from_msg_param(struct tee_param *params, uint num_params,
  176. const struct optee_msg_param *msg_params)
  177. {
  178. uint n;
  179. struct tee_shm *shm;
  180. for (n = 0; n < num_params; n++) {
  181. struct tee_param *p = params + n;
  182. const struct optee_msg_param *mp = msg_params + n;
  183. u32 attr = mp->attr & OPTEE_MSG_ATTR_TYPE_MASK;
  184. switch (attr) {
  185. case OPTEE_MSG_ATTR_TYPE_NONE:
  186. p->attr = TEE_PARAM_ATTR_TYPE_NONE;
  187. memset(&p->u, 0, sizeof(p->u));
  188. break;
  189. case OPTEE_MSG_ATTR_TYPE_VALUE_INPUT:
  190. case OPTEE_MSG_ATTR_TYPE_VALUE_OUTPUT:
  191. case OPTEE_MSG_ATTR_TYPE_VALUE_INOUT:
  192. p->attr = TEE_PARAM_ATTR_TYPE_VALUE_INPUT + attr -
  193. OPTEE_MSG_ATTR_TYPE_VALUE_INPUT;
  194. p->u.value.a = mp->u.value.a;
  195. p->u.value.b = mp->u.value.b;
  196. p->u.value.c = mp->u.value.c;
  197. break;
  198. case OPTEE_MSG_ATTR_TYPE_RMEM_INPUT:
  199. case OPTEE_MSG_ATTR_TYPE_RMEM_OUTPUT:
  200. case OPTEE_MSG_ATTR_TYPE_RMEM_INOUT:
  201. p->attr = TEE_PARAM_ATTR_TYPE_MEMREF_INPUT + attr -
  202. OPTEE_MSG_ATTR_TYPE_RMEM_INPUT;
  203. p->u.memref.size = mp->u.rmem.size;
  204. shm = (struct tee_shm *)(ulong)mp->u.rmem.shm_ref;
  205. if (!shm) {
  206. p->u.memref.shm_offs = 0;
  207. p->u.memref.shm = NULL;
  208. break;
  209. }
  210. p->u.memref.shm_offs = mp->u.rmem.offs;
  211. p->u.memref.shm = shm;
  212. break;
  213. default:
  214. return -EINVAL;
  215. }
  216. }
  217. return 0;
  218. }
  219. static void handle_rpc(struct udevice *dev, struct rpc_param *param,
  220. void *page_list)
  221. {
  222. struct tee_shm *shm;
  223. switch (OPTEE_SMC_RETURN_GET_RPC_FUNC(param->a0)) {
  224. case OPTEE_SMC_RPC_FUNC_ALLOC:
  225. if (!__tee_shm_add(dev, OPTEE_MSG_NONCONTIG_PAGE_SIZE, NULL,
  226. param->a1, TEE_SHM_ALLOC | TEE_SHM_REGISTER,
  227. &shm)) {
  228. reg_pair_from_64(&param->a1, &param->a2,
  229. virt_to_phys(shm->addr));
  230. /* "cookie" */
  231. reg_pair_from_64(&param->a4, &param->a5, (ulong)shm);
  232. } else {
  233. param->a1 = 0;
  234. param->a2 = 0;
  235. param->a4 = 0;
  236. param->a5 = 0;
  237. }
  238. break;
  239. case OPTEE_SMC_RPC_FUNC_FREE:
  240. shm = reg_pair_to_ptr(param->a1, param->a2);
  241. tee_shm_free(shm);
  242. break;
  243. case OPTEE_SMC_RPC_FUNC_FOREIGN_INTR:
  244. break;
  245. case OPTEE_SMC_RPC_FUNC_CMD:
  246. shm = reg_pair_to_ptr(param->a1, param->a2);
  247. optee_suppl_cmd(dev, shm, page_list);
  248. break;
  249. default:
  250. break;
  251. }
  252. param->a0 = OPTEE_SMC_CALL_RETURN_FROM_RPC;
  253. }
  254. static u32 call_err_to_res(u32 call_err)
  255. {
  256. switch (call_err) {
  257. case OPTEE_SMC_RETURN_OK:
  258. return TEE_SUCCESS;
  259. default:
  260. return TEE_ERROR_BAD_PARAMETERS;
  261. }
  262. }
  263. static u32 do_call_with_arg(struct udevice *dev, struct optee_msg_arg *arg)
  264. {
  265. struct optee_pdata *pdata = dev_get_platdata(dev);
  266. struct rpc_param param = { .a0 = OPTEE_SMC_CALL_WITH_ARG };
  267. void *page_list = NULL;
  268. reg_pair_from_64(&param.a1, &param.a2, virt_to_phys(arg));
  269. while (true) {
  270. struct arm_smccc_res res;
  271. pdata->invoke_fn(param.a0, param.a1, param.a2, param.a3,
  272. param.a4, param.a5, param.a6, param.a7, &res);
  273. free(page_list);
  274. page_list = NULL;
  275. if (OPTEE_SMC_RETURN_IS_RPC(res.a0)) {
  276. param.a0 = res.a0;
  277. param.a1 = res.a1;
  278. param.a2 = res.a2;
  279. param.a3 = res.a3;
  280. handle_rpc(dev, &param, &page_list);
  281. } else {
  282. return call_err_to_res(res.a0);
  283. }
  284. }
  285. }
  286. static int optee_close_session(struct udevice *dev, u32 session)
  287. {
  288. int rc;
  289. struct tee_shm *shm;
  290. struct optee_msg_arg *msg_arg;
  291. rc = get_msg_arg(dev, 0, &shm, &msg_arg);
  292. if (rc)
  293. return rc;
  294. msg_arg->cmd = OPTEE_MSG_CMD_CLOSE_SESSION;
  295. msg_arg->session = session;
  296. do_call_with_arg(dev, msg_arg);
  297. tee_shm_free(shm);
  298. return 0;
  299. }
  300. static int optee_open_session(struct udevice *dev,
  301. struct tee_open_session_arg *arg,
  302. uint num_params, struct tee_param *params)
  303. {
  304. int rc;
  305. struct tee_shm *shm;
  306. struct optee_msg_arg *msg_arg;
  307. rc = get_msg_arg(dev, num_params + 2, &shm, &msg_arg);
  308. if (rc)
  309. return rc;
  310. msg_arg->cmd = OPTEE_MSG_CMD_OPEN_SESSION;
  311. /*
  312. * Initialize and add the meta parameters needed when opening a
  313. * session.
  314. */
  315. msg_arg->params[0].attr = OPTEE_MSG_ATTR_TYPE_VALUE_INPUT |
  316. OPTEE_MSG_ATTR_META;
  317. msg_arg->params[1].attr = OPTEE_MSG_ATTR_TYPE_VALUE_INPUT |
  318. OPTEE_MSG_ATTR_META;
  319. memcpy(&msg_arg->params[0].u.value, arg->uuid, sizeof(arg->uuid));
  320. memcpy(&msg_arg->params[1].u.value, arg->uuid, sizeof(arg->clnt_uuid));
  321. msg_arg->params[1].u.value.c = arg->clnt_login;
  322. rc = to_msg_param(msg_arg->params + 2, num_params, params);
  323. if (rc)
  324. goto out;
  325. arg->ret = do_call_with_arg(dev, msg_arg);
  326. if (arg->ret) {
  327. arg->ret_origin = TEE_ORIGIN_COMMS;
  328. goto out;
  329. }
  330. if (from_msg_param(params, num_params, msg_arg->params + 2)) {
  331. arg->ret = TEE_ERROR_COMMUNICATION;
  332. arg->ret_origin = TEE_ORIGIN_COMMS;
  333. /* Close session again to avoid leakage */
  334. optee_close_session(dev, msg_arg->session);
  335. goto out;
  336. }
  337. arg->session = msg_arg->session;
  338. arg->ret = msg_arg->ret;
  339. arg->ret_origin = msg_arg->ret_origin;
  340. out:
  341. tee_shm_free(shm);
  342. return rc;
  343. }
  344. static int optee_invoke_func(struct udevice *dev, struct tee_invoke_arg *arg,
  345. uint num_params, struct tee_param *params)
  346. {
  347. struct tee_shm *shm;
  348. struct optee_msg_arg *msg_arg;
  349. int rc;
  350. rc = get_msg_arg(dev, num_params, &shm, &msg_arg);
  351. if (rc)
  352. return rc;
  353. msg_arg->cmd = OPTEE_MSG_CMD_INVOKE_COMMAND;
  354. msg_arg->func = arg->func;
  355. msg_arg->session = arg->session;
  356. rc = to_msg_param(msg_arg->params, num_params, params);
  357. if (rc)
  358. goto out;
  359. arg->ret = do_call_with_arg(dev, msg_arg);
  360. if (arg->ret) {
  361. arg->ret_origin = TEE_ORIGIN_COMMS;
  362. goto out;
  363. }
  364. if (from_msg_param(params, num_params, msg_arg->params)) {
  365. arg->ret = TEE_ERROR_COMMUNICATION;
  366. arg->ret_origin = TEE_ORIGIN_COMMS;
  367. goto out;
  368. }
  369. arg->ret = msg_arg->ret;
  370. arg->ret_origin = msg_arg->ret_origin;
  371. out:
  372. tee_shm_free(shm);
  373. return rc;
  374. }
  375. static int optee_shm_register(struct udevice *dev, struct tee_shm *shm)
  376. {
  377. struct tee_shm *shm_arg;
  378. struct optee_msg_arg *msg_arg;
  379. void *pl;
  380. u64 ph_ptr;
  381. int rc;
  382. rc = get_msg_arg(dev, 1, &shm_arg, &msg_arg);
  383. if (rc)
  384. return rc;
  385. pl = optee_alloc_and_init_page_list(shm->addr, shm->size, &ph_ptr);
  386. if (!pl) {
  387. rc = -ENOMEM;
  388. goto out;
  389. }
  390. msg_arg->cmd = OPTEE_MSG_CMD_REGISTER_SHM;
  391. msg_arg->params->attr = OPTEE_MSG_ATTR_TYPE_TMEM_OUTPUT |
  392. OPTEE_MSG_ATTR_NONCONTIG;
  393. msg_arg->params->u.tmem.buf_ptr = ph_ptr;
  394. msg_arg->params->u.tmem.shm_ref = (ulong)shm;
  395. msg_arg->params->u.tmem.size = shm->size;
  396. if (do_call_with_arg(dev, msg_arg) || msg_arg->ret)
  397. rc = -EINVAL;
  398. free(pl);
  399. out:
  400. tee_shm_free(shm_arg);
  401. return rc;
  402. }
  403. static int optee_shm_unregister(struct udevice *dev, struct tee_shm *shm)
  404. {
  405. struct tee_shm *shm_arg;
  406. struct optee_msg_arg *msg_arg;
  407. int rc;
  408. rc = get_msg_arg(dev, 1, &shm_arg, &msg_arg);
  409. if (rc)
  410. return rc;
  411. msg_arg->cmd = OPTEE_MSG_CMD_UNREGISTER_SHM;
  412. msg_arg->params[0].attr = OPTEE_MSG_ATTR_TYPE_RMEM_INPUT;
  413. msg_arg->params[0].u.rmem.shm_ref = (ulong)shm;
  414. if (do_call_with_arg(dev, msg_arg) || msg_arg->ret)
  415. rc = -EINVAL;
  416. tee_shm_free(shm_arg);
  417. return rc;
  418. }
  419. static const struct tee_driver_ops optee_ops = {
  420. .get_version = optee_get_version,
  421. .open_session = optee_open_session,
  422. .close_session = optee_close_session,
  423. .invoke_func = optee_invoke_func,
  424. .shm_register = optee_shm_register,
  425. .shm_unregister = optee_shm_unregister,
  426. };
  427. static bool is_optee_api(optee_invoke_fn *invoke_fn)
  428. {
  429. struct arm_smccc_res res;
  430. invoke_fn(OPTEE_SMC_CALLS_UID, 0, 0, 0, 0, 0, 0, 0, &res);
  431. return res.a0 == OPTEE_MSG_UID_0 && res.a1 == OPTEE_MSG_UID_1 &&
  432. res.a2 == OPTEE_MSG_UID_2 && res.a3 == OPTEE_MSG_UID_3;
  433. }
  434. static void print_os_revision(optee_invoke_fn *invoke_fn)
  435. {
  436. union {
  437. struct arm_smccc_res smccc;
  438. struct optee_smc_call_get_os_revision_result result;
  439. } res = {
  440. .result = {
  441. .build_id = 0
  442. }
  443. };
  444. invoke_fn(OPTEE_SMC_CALL_GET_OS_REVISION, 0, 0, 0, 0, 0, 0, 0,
  445. &res.smccc);
  446. if (res.result.build_id)
  447. debug("OP-TEE revision %lu.%lu (%08lx)\n", res.result.major,
  448. res.result.minor, res.result.build_id);
  449. else
  450. debug("OP-TEE revision %lu.%lu\n", res.result.major,
  451. res.result.minor);
  452. }
  453. static bool api_revision_is_compatible(optee_invoke_fn *invoke_fn)
  454. {
  455. union {
  456. struct arm_smccc_res smccc;
  457. struct optee_smc_calls_revision_result result;
  458. } res;
  459. invoke_fn(OPTEE_SMC_CALLS_REVISION, 0, 0, 0, 0, 0, 0, 0, &res.smccc);
  460. return res.result.major == OPTEE_MSG_REVISION_MAJOR &&
  461. (int)res.result.minor >= OPTEE_MSG_REVISION_MINOR;
  462. }
  463. static bool exchange_capabilities(optee_invoke_fn *invoke_fn, u32 *sec_caps)
  464. {
  465. union {
  466. struct arm_smccc_res smccc;
  467. struct optee_smc_exchange_capabilities_result result;
  468. } res;
  469. invoke_fn(OPTEE_SMC_EXCHANGE_CAPABILITIES,
  470. OPTEE_SMC_NSEC_CAP_UNIPROCESSOR, 0, 0, 0, 0, 0, 0,
  471. &res.smccc);
  472. if (res.result.status != OPTEE_SMC_RETURN_OK)
  473. return false;
  474. *sec_caps = res.result.capabilities;
  475. return true;
  476. }
  477. /* Simple wrapper functions to be able to use a function pointer */
  478. static void optee_smccc_smc(unsigned long a0, unsigned long a1,
  479. unsigned long a2, unsigned long a3,
  480. unsigned long a4, unsigned long a5,
  481. unsigned long a6, unsigned long a7,
  482. struct arm_smccc_res *res)
  483. {
  484. arm_smccc_smc(a0, a1, a2, a3, a4, a5, a6, a7, res);
  485. }
  486. static void optee_smccc_hvc(unsigned long a0, unsigned long a1,
  487. unsigned long a2, unsigned long a3,
  488. unsigned long a4, unsigned long a5,
  489. unsigned long a6, unsigned long a7,
  490. struct arm_smccc_res *res)
  491. {
  492. arm_smccc_hvc(a0, a1, a2, a3, a4, a5, a6, a7, res);
  493. }
  494. static optee_invoke_fn *get_invoke_func(struct udevice *dev)
  495. {
  496. const char *method;
  497. debug("optee: looking for conduit method in DT.\n");
  498. method = ofnode_get_property(dev->node, "method", NULL);
  499. if (!method) {
  500. debug("optee: missing \"method\" property\n");
  501. return ERR_PTR(-ENXIO);
  502. }
  503. if (!strcmp("hvc", method))
  504. return optee_smccc_hvc;
  505. else if (!strcmp("smc", method))
  506. return optee_smccc_smc;
  507. debug("optee: invalid \"method\" property: %s\n", method);
  508. return ERR_PTR(-EINVAL);
  509. }
  510. static int optee_ofdata_to_platdata(struct udevice *dev)
  511. {
  512. struct optee_pdata *pdata = dev_get_platdata(dev);
  513. pdata->invoke_fn = get_invoke_func(dev);
  514. if (IS_ERR(pdata->invoke_fn))
  515. return PTR_ERR(pdata->invoke_fn);
  516. return 0;
  517. }
  518. static int optee_probe(struct udevice *dev)
  519. {
  520. struct optee_pdata *pdata = dev_get_platdata(dev);
  521. u32 sec_caps;
  522. if (!is_optee_api(pdata->invoke_fn)) {
  523. debug("%s: OP-TEE api uid mismatch\n", __func__);
  524. return -ENOENT;
  525. }
  526. print_os_revision(pdata->invoke_fn);
  527. if (!api_revision_is_compatible(pdata->invoke_fn)) {
  528. debug("%s: OP-TEE api revision mismatch\n", __func__);
  529. return -ENOENT;
  530. }
  531. /*
  532. * OP-TEE can use both shared memory via predefined pool or as
  533. * dynamic shared memory provided by normal world. To keep things
  534. * simple we're only using dynamic shared memory in this driver.
  535. */
  536. if (!exchange_capabilities(pdata->invoke_fn, &sec_caps) ||
  537. !(sec_caps & OPTEE_SMC_SEC_CAP_DYNAMIC_SHM)) {
  538. debug("%s: OP-TEE capabilities mismatch\n", __func__);
  539. return -ENOENT;
  540. }
  541. return 0;
  542. }
  543. static const struct udevice_id optee_match[] = {
  544. { .compatible = "linaro,optee-tz" },
  545. {},
  546. };
  547. U_BOOT_DRIVER(optee) = {
  548. .name = "optee",
  549. .id = UCLASS_TEE,
  550. .of_match = optee_match,
  551. .ofdata_to_platdata = optee_ofdata_to_platdata,
  552. .probe = optee_probe,
  553. .ops = &optee_ops,
  554. .platdata_auto_alloc_size = sizeof(struct optee_pdata),
  555. };