1 // SPDX-License-Identifier: BSD-2-Clause
2 /*
3 * Copyright (c) 2020 Foundries.io Ltd
4 */
5
6 #include <dm.h>
7 #include <i2c.h>
8 #include <tee.h>
9 #include "optee_msg.h"
10 #include "optee_private.h"
11
check_xfer_flags(struct udevice * chip,uint tee_flags)12 static int check_xfer_flags(struct udevice *chip, uint tee_flags)
13 {
14 uint flags;
15 int ret;
16
17 ret = i2c_get_chip_flags(chip, &flags);
18 if (ret)
19 return ret;
20
21 if (tee_flags & OPTEE_MSG_RPC_CMD_I2C_FLAGS_TEN_BIT) {
22 if (!(flags & DM_I2C_CHIP_10BIT))
23 return -EINVAL;
24 } else {
25 if (flags & DM_I2C_CHIP_10BIT)
26 return -EINVAL;
27 }
28
29 return 0;
30 }
31
optee_suppl_cmd_i2c_transfer(struct optee_msg_arg * arg)32 void optee_suppl_cmd_i2c_transfer(struct optee_msg_arg *arg)
33 {
34 const u8 attr[] = {
35 OPTEE_MSG_ATTR_TYPE_VALUE_INPUT,
36 OPTEE_MSG_ATTR_TYPE_VALUE_INPUT,
37 OPTEE_MSG_ATTR_TYPE_RMEM_INOUT,
38 OPTEE_MSG_ATTR_TYPE_VALUE_OUTPUT,
39 };
40 struct udevice *chip_dev;
41 struct tee_shm *shm;
42 u8 *buf;
43 int ret;
44
45 if (arg->num_params != ARRAY_SIZE(attr) ||
46 arg->params[0].attr != attr[0] ||
47 arg->params[1].attr != attr[1] ||
48 arg->params[2].attr != attr[2] ||
49 arg->params[3].attr != attr[3]) {
50 goto bad;
51 }
52
53 shm = (struct tee_shm *)(unsigned long)arg->params[2].u.rmem.shm_ref;
54 buf = shm->addr;
55 if (!buf)
56 goto bad;
57
58 if (i2c_get_chip_for_busnum((int)arg->params[0].u.value.b,
59 (int)arg->params[0].u.value.c,
60 0, &chip_dev))
61 goto bad;
62
63 if (check_xfer_flags(chip_dev, arg->params[1].u.value.a))
64 goto bad;
65
66 switch (arg->params[0].u.value.a) {
67 case OPTEE_MSG_RPC_CMD_I2C_TRANSFER_RD:
68 ret = dm_i2c_read(chip_dev, 0, buf,
69 (size_t)arg->params[2].u.rmem.size);
70 break;
71 case OPTEE_MSG_RPC_CMD_I2C_TRANSFER_WR:
72 ret = dm_i2c_write(chip_dev, 0, buf,
73 (size_t)arg->params[2].u.rmem.size);
74 break;
75 default:
76 goto bad;
77 }
78
79 if (ret) {
80 arg->ret = TEE_ERROR_COMMUNICATION;
81 } else {
82 arg->params[3].u.value.a = arg->params[2].u.rmem.size;
83 arg->ret = TEE_SUCCESS;
84 }
85
86 return;
87 bad:
88 arg->ret = TEE_ERROR_BAD_PARAMETERS;
89 }
90