Hi Sumit,
On Tue, Dec 3, 2024 at 9:27 AM Sumit Garg <sumit.garg@xxxxxxxxxx> wrote:
>
> Hi Jens,
>
> On Thu, 28 Nov 2024 at 20:39, Jens Wiklander <jens.wiklander@xxxxxxxxxx> wrote:
> >
> > The OP-TEE backend driver has two internal function pointers to convert
> > between the subsystem type struct tee_param and the OP-TEE type struct
> > optee_msg_param.
> >
> > The conversion is done from one of the types to the other, which is then
> > involved in some operation and finally converted back to the original
> > type. When converting to prepare the parameters for the operation, all
> > fields must be taken into account, but then converting back, it's enough
> > to update only out-values and out-sizes. So, an update_out parameter is
> > added to the conversion functions to tell if all or only some fields
> > must be copied.
>
> Is this patch just a refactoring of TEE params handling? Or are we
> fixing a real ABI issue here? Also, is this required for restricted
> shared memory support to work? Just asking if we want to consider it
> as a fix for backporting or if we can handle this refactoring patch
> independently of this series?
This is needed in this series. Without this patch, it might get
confusing when converting back in from_msg_param() callback since an
allocated restricted SHM can be using the sec_world_id of the pool and
that doesn't translate back well. params_to_user() in
drivers/tee/tee_core.c already takes the direction into account so it
wouldn't spread to user-space.
Cheers,
Jens
>
> -Sumit
>
> >
> > Signed-off-by: Jens Wiklander <jens.wiklander@xxxxxxxxxx>
> > ---
> > drivers/tee/optee/call.c | 10 ++--
> > drivers/tee/optee/ffa_abi.c | 43 +++++++++++++----
> > drivers/tee/optee/optee_private.h | 42 +++++++++++------
> > drivers/tee/optee/rpc.c | 31 +++++++++----
> > drivers/tee/optee/smc_abi.c | 76 +++++++++++++++++++++++--------
> > 5 files changed, 144 insertions(+), 58 deletions(-)
> >
> > diff --git a/drivers/tee/optee/call.c b/drivers/tee/optee/call.c
> > index 16eb953e14bb..f1533b894726 100644
> > --- a/drivers/tee/optee/call.c
> > +++ b/drivers/tee/optee/call.c
> > @@ -400,7 +400,8 @@ int optee_open_session(struct tee_context *ctx,
> > export_uuid(msg_arg->params[1].u.octets, &client_uuid);
> >
> > rc = optee->ops->to_msg_param(optee, msg_arg->params + 2,
> > - arg->num_params, param);
> > + arg->num_params, param,
> > + false /*!update_out*/);
> > if (rc)
> > goto out;
> >
> > @@ -427,7 +428,8 @@ int optee_open_session(struct tee_context *ctx,
> > }
> >
> > if (optee->ops->from_msg_param(optee, param, arg->num_params,
> > - msg_arg->params + 2)) {
> > + msg_arg->params + 2,
> > + true /*update_out*/)) {
> > arg->ret = TEEC_ERROR_COMMUNICATION;
> > arg->ret_origin = TEEC_ORIGIN_COMMS;
> > /* Close session again to avoid leakage */
> > @@ -541,7 +543,7 @@ int optee_invoke_func(struct tee_context *ctx, struct tee_ioctl_invoke_arg *arg,
> > msg_arg->cancel_id = arg->cancel_id;
> >
> > rc = optee->ops->to_msg_param(optee, msg_arg->params, arg->num_params,
> > - param);
> > + param, false /*!update_out*/);
> > if (rc)
> > goto out;
> >
> > @@ -551,7 +553,7 @@ int optee_invoke_func(struct tee_context *ctx, struct tee_ioctl_invoke_arg *arg,
> > }
> >
> > if (optee->ops->from_msg_param(optee, param, arg->num_params,
> > - msg_arg->params)) {
> > + msg_arg->params, true /*update_out*/)) {
> > msg_arg->ret = TEEC_ERROR_COMMUNICATION;
> > msg_arg->ret_origin = TEEC_ORIGIN_COMMS;
> > }
> > diff --git a/drivers/tee/optee/ffa_abi.c b/drivers/tee/optee/ffa_abi.c
> > index f3af5666bb11..02e6175ac5f0 100644
> > --- a/drivers/tee/optee/ffa_abi.c
> > +++ b/drivers/tee/optee/ffa_abi.c
> > @@ -122,15 +122,21 @@ static int optee_shm_rem_ffa_handle(struct optee *optee, u64 global_id)
> > */
> >
> > static void from_msg_param_ffa_mem(struct optee *optee, struct tee_param *p,
> > - u32 attr, const struct optee_msg_param *mp)
> > + u32 attr, const struct optee_msg_param *mp,
> > + bool update_out)
> > {
> > struct tee_shm *shm = NULL;
> > u64 offs_high = 0;
> > u64 offs_low = 0;
> >
> > + if (update_out) {
> > + if (attr == OPTEE_MSG_ATTR_TYPE_FMEM_INPUT)
> > + return;
> > + goto out;
> > + }
> > +
> > p->attr = TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_INPUT +
> > attr - OPTEE_MSG_ATTR_TYPE_FMEM_INPUT;
> > - p->u.memref.size = mp->u.fmem.size;
> >
> > if (mp->u.fmem.global_id != OPTEE_MSG_FMEM_INVALID_GLOBAL_ID)
> > shm = optee_shm_from_ffa_handle(optee, mp->u.fmem.global_id);
> > @@ -141,6 +147,8 @@ static void from_msg_param_ffa_mem(struct optee *optee, struct tee_param *p,
> > offs_high = mp->u.fmem.offs_high;
> > }
> > p->u.memref.shm_offs = offs_low | offs_high << 32;
> > +out:
> > + p->u.memref.size = mp->u.fmem.size;
> > }
> >
> > /**
> > @@ -150,12 +158,14 @@ static void from_msg_param_ffa_mem(struct optee *optee, struct tee_param *p,
> > * @params: subsystem internal parameter representation
> > * @num_params: number of elements in the parameter arrays
> > * @msg_params: OPTEE_MSG parameters
> > + * @update_out: update parameter for output only
> > *
> > * Returns 0 on success or <0 on failure
> > */
> > static int optee_ffa_from_msg_param(struct optee *optee,
> > struct tee_param *params, size_t num_params,
> > - const struct optee_msg_param *msg_params)
> > + const struct optee_msg_param *msg_params,
> > + bool update_out)
> > {
> > size_t n;
> >
> > @@ -166,18 +176,20 @@ static int optee_ffa_from_msg_param(struct optee *optee,
> >
> > switch (attr) {
> > case OPTEE_MSG_ATTR_TYPE_NONE:
> > + if (update_out)
> > + break;
> > p->attr = TEE_IOCTL_PARAM_ATTR_TYPE_NONE;
> > memset(&p->u, 0, sizeof(p->u));
> > break;
> > case OPTEE_MSG_ATTR_TYPE_VALUE_INPUT:
> > case OPTEE_MSG_ATTR_TYPE_VALUE_OUTPUT:
> > case OPTEE_MSG_ATTR_TYPE_VALUE_INOUT:
> > - optee_from_msg_param_value(p, attr, mp);
> > + optee_from_msg_param_value(p, attr, mp, update_out);
> > break;
> > case OPTEE_MSG_ATTR_TYPE_FMEM_INPUT:
> > case OPTEE_MSG_ATTR_TYPE_FMEM_OUTPUT:
> > case OPTEE_MSG_ATTR_TYPE_FMEM_INOUT:
> > - from_msg_param_ffa_mem(optee, p, attr, mp);
> > + from_msg_param_ffa_mem(optee, p, attr, mp, update_out);
> > break;
> > default:
> > return -EINVAL;
> > @@ -188,10 +200,16 @@ static int optee_ffa_from_msg_param(struct optee *optee,
> > }
> >
> > static int to_msg_param_ffa_mem(struct optee_msg_param *mp,
> > - const struct tee_param *p)
> > + const struct tee_param *p, bool update_out)
> > {
> > struct tee_shm *shm = p->u.memref.shm;
> >
> > + if (update_out) {
> > + if (p->attr == TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_INPUT)
> > + return 0;
> > + goto out;
> > + }
> > +
> > mp->attr = OPTEE_MSG_ATTR_TYPE_FMEM_INPUT + p->attr -
> > TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_INPUT;
> >
> > @@ -211,6 +229,7 @@ static int to_msg_param_ffa_mem(struct optee_msg_param *mp,
> > memset(&mp->u, 0, sizeof(mp->u));
> > mp->u.fmem.global_id = OPTEE_MSG_FMEM_INVALID_GLOBAL_ID;
> > }
> > +out:
> > mp->u.fmem.size = p->u.memref.size;
> >
> > return 0;
> > @@ -222,13 +241,15 @@ static int to_msg_param_ffa_mem(struct optee_msg_param *mp,
> > * @optee: main service struct
> > * @msg_params: OPTEE_MSG parameters
> > * @num_params: number of elements in the parameter arrays
> > - * @params: subsystem itnernal parameter representation
> > + * @params: subsystem internal parameter representation
> > + * @update_out: update parameter for output only
> > * Returns 0 on success or <0 on failure
> > */
> > static int optee_ffa_to_msg_param(struct optee *optee,
> > struct optee_msg_param *msg_params,
> > size_t num_params,
> > - const struct tee_param *params)
> > + const struct tee_param *params,
> > + bool update_out)
> > {
> > size_t n;
> >
> > @@ -238,18 +259,20 @@ static int optee_ffa_to_msg_param(struct optee *optee,
> >
> > switch (p->attr) {
> > case TEE_IOCTL_PARAM_ATTR_TYPE_NONE:
> > + if (update_out)
> > + break;
> > mp->attr = TEE_IOCTL_PARAM_ATTR_TYPE_NONE;
> > memset(&mp->u, 0, sizeof(mp->u));
> > break;
> > case TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_INPUT:
> > case TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_OUTPUT:
> > case TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_INOUT:
> > - optee_to_msg_param_value(mp, p);
> > + optee_to_msg_param_value(mp, p, update_out);
> > break;
> > case TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_INPUT:
> > case TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_OUTPUT:
> > case TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_INOUT:
> > - if (to_msg_param_ffa_mem(mp, p))
> > + if (to_msg_param_ffa_mem(mp, p, update_out))
> > return -EINVAL;
> > break;
> > default:
> > diff --git a/drivers/tee/optee/optee_private.h b/drivers/tee/optee/optee_private.h
> > index dc0f355ef72a..20eda508dbac 100644
> > --- a/drivers/tee/optee/optee_private.h
> > +++ b/drivers/tee/optee/optee_private.h
> > @@ -185,10 +185,12 @@ struct optee_ops {
> > bool system_thread);
> > int (*to_msg_param)(struct optee *optee,
> > struct optee_msg_param *msg_params,
> > - size_t num_params, const struct tee_param *params);
> > + size_t num_params, const struct tee_param *params,
> > + bool update_out);
> > int (*from_msg_param)(struct optee *optee, struct tee_param *params,
> > size_t num_params,
> > - const struct optee_msg_param *msg_params);
> > + const struct optee_msg_param *msg_params,
> > + bool update_out);
> > };
> >
> > /**
> > @@ -316,23 +318,35 @@ void optee_release(struct tee_context *ctx);
> > void optee_release_supp(struct tee_context *ctx);
> >
> > static inline void optee_from_msg_param_value(struct tee_param *p, u32 attr,
> > - const struct optee_msg_param *mp)
> > + const struct optee_msg_param *mp,
> > + bool update_out)
> > {
> > - p->attr = TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_INPUT +
> > - attr - OPTEE_MSG_ATTR_TYPE_VALUE_INPUT;
> > - p->u.value.a = mp->u.value.a;
> > - p->u.value.b = mp->u.value.b;
> > - p->u.value.c = mp->u.value.c;
> > + if (!update_out)
> > + p->attr = TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_INPUT +
> > + attr - OPTEE_MSG_ATTR_TYPE_VALUE_INPUT;
> > +
> > + if (attr == OPTEE_MSG_ATTR_TYPE_VALUE_OUTPUT ||
> > + attr == OPTEE_MSG_ATTR_TYPE_VALUE_INOUT || !update_out) {
> > + p->u.value.a = mp->u.value.a;
> > + p->u.value.b = mp->u.value.b;
> > + p->u.value.c = mp->u.value.c;
> > + }
> > }
> >
> > static inline void optee_to_msg_param_value(struct optee_msg_param *mp,
> > - const struct tee_param *p)
> > + const struct tee_param *p,
> > + bool update_out)
> > {
> > - mp->attr = OPTEE_MSG_ATTR_TYPE_VALUE_INPUT + p->attr -
> > - TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_INPUT;
> > - mp->u.value.a = p->u.value.a;
> > - mp->u.value.b = p->u.value.b;
> > - mp->u.value.c = p->u.value.c;
> > + if (!update_out)
> > + mp->attr = OPTEE_MSG_ATTR_TYPE_VALUE_INPUT + p->attr -
> > + TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_INPUT;
> > +
> > + if (p->attr == TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_OUTPUT ||
> > + p->attr == TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_INOUT || !update_out) {
> > + mp->u.value.a = p->u.value.a;
> > + mp->u.value.b = p->u.value.b;
> > + mp->u.value.c = p->u.value.c;
> > + }
> > }
> >
> > void optee_cq_init(struct optee_call_queue *cq, int thread_count);
> > diff --git a/drivers/tee/optee/rpc.c b/drivers/tee/optee/rpc.c
> > index ebbbd42b0e3e..580e6b9b0606 100644
> > --- a/drivers/tee/optee/rpc.c
> > +++ b/drivers/tee/optee/rpc.c
> > @@ -63,7 +63,7 @@ static void handle_rpc_func_cmd_i2c_transfer(struct tee_context *ctx,
> > }
> >
> > if (optee->ops->from_msg_param(optee, params, arg->num_params,
> > - arg->params))
> > + arg->params, false /*!update_out*/))
> > goto bad;
> >
> > for (i = 0; i < arg->num_params; i++) {
> > @@ -107,7 +107,8 @@ static void handle_rpc_func_cmd_i2c_transfer(struct tee_context *ctx,
> > } else {
> > params[3].u.value.a = msg.len;
> > if (optee->ops->to_msg_param(optee, arg->params,
> > - arg->num_params, params))
> > + arg->num_params, params,
> > + true /*update_out*/))
> > arg->ret = TEEC_ERROR_BAD_PARAMETERS;
> > else
> > arg->ret = TEEC_SUCCESS;
> > @@ -188,6 +189,7 @@ static void handle_rpc_func_cmd_wait(struct optee_msg_arg *arg)
> > static void handle_rpc_supp_cmd(struct tee_context *ctx, struct optee *optee,
> > struct optee_msg_arg *arg)
> > {
> > + bool update_out = false;
> > struct tee_param *params;
> >
> > arg->ret_origin = TEEC_ORIGIN_COMMS;
> > @@ -200,15 +202,21 @@ static void handle_rpc_supp_cmd(struct tee_context *ctx, struct optee *optee,
> > }
> >
> > if (optee->ops->from_msg_param(optee, params, arg->num_params,
> > - arg->params)) {
> > + arg->params, update_out)) {
> > arg->ret = TEEC_ERROR_BAD_PARAMETERS;
> > goto out;
> > }
> >
> > arg->ret = optee_supp_thrd_req(ctx, arg->cmd, arg->num_params, params);
> >
> > + /*
> > + * Special treatment for OPTEE_RPC_CMD_SHM_ALLOC since input is a
> > + * value type, but the output is a memref type.
> > + */
> > + if (arg->cmd != OPTEE_RPC_CMD_SHM_ALLOC)
> > + update_out = true;
> > if (optee->ops->to_msg_param(optee, arg->params, arg->num_params,
> > - params))
> > + params, update_out))
> > arg->ret = TEEC_ERROR_BAD_PARAMETERS;
> > out:
> > kfree(params);
> > @@ -270,7 +278,7 @@ static void handle_rpc_func_rpmb_probe_reset(struct tee_context *ctx,
> >
> > if (arg->num_params != ARRAY_SIZE(params) ||
> > optee->ops->from_msg_param(optee, params, arg->num_params,
> > - arg->params) ||
> > + arg->params, false /*!update_out*/) ||
> > params[0].attr != TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_OUTPUT) {
> > arg->ret = TEEC_ERROR_BAD_PARAMETERS;
> > return;
> > @@ -280,7 +288,8 @@ static void handle_rpc_func_rpmb_probe_reset(struct tee_context *ctx,
> > params[0].u.value.b = 0;
> > params[0].u.value.c = 0;
> > if (optee->ops->to_msg_param(optee, arg->params,
> > - arg->num_params, params)) {
> > + arg->num_params, params,
> > + true /*update_out*/)) {
> > arg->ret = TEEC_ERROR_BAD_PARAMETERS;
> > return;
> > }
> > @@ -324,7 +333,7 @@ static void handle_rpc_func_rpmb_probe_next(struct tee_context *ctx,
> >
> > if (arg->num_params != ARRAY_SIZE(params) ||
> > optee->ops->from_msg_param(optee, params, arg->num_params,
> > - arg->params) ||
> > + arg->params, false /*!update_out*/) ||
> > params[0].attr != TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_OUTPUT ||
> > params[1].attr != TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_OUTPUT) {
> > arg->ret = TEEC_ERROR_BAD_PARAMETERS;
> > @@ -358,7 +367,8 @@ static void handle_rpc_func_rpmb_probe_next(struct tee_context *ctx,
> > params[0].u.value.b = rdev->descr.capacity;
> > params[0].u.value.c = rdev->descr.reliable_wr_count;
> > if (optee->ops->to_msg_param(optee, arg->params,
> > - arg->num_params, params)) {
> > + arg->num_params, params,
> > + true /*update_out*/)) {
> > arg->ret = TEEC_ERROR_BAD_PARAMETERS;
> > return;
> > }
> > @@ -384,7 +394,7 @@ static void handle_rpc_func_rpmb_frames(struct tee_context *ctx,
> >
> > if (arg->num_params != ARRAY_SIZE(params) ||
> > optee->ops->from_msg_param(optee, params, arg->num_params,
> > - arg->params) ||
> > + arg->params, false /*!update_out*/) ||
> > params[0].attr != TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_INPUT ||
> > params[1].attr != TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_OUTPUT) {
> > arg->ret = TEEC_ERROR_BAD_PARAMETERS;
> > @@ -401,7 +411,8 @@ static void handle_rpc_func_rpmb_frames(struct tee_context *ctx,
> > goto out;
> > }
> > if (optee->ops->to_msg_param(optee, arg->params,
> > - arg->num_params, params)) {
> > + arg->num_params, params,
> > + true /*update_out*/)) {
> > arg->ret = TEEC_ERROR_BAD_PARAMETERS;
> > goto out;
> > }
> > diff --git a/drivers/tee/optee/smc_abi.c b/drivers/tee/optee/smc_abi.c
> > index e9456e3e74cc..d1f79947f58a 100644
> > --- a/drivers/tee/optee/smc_abi.c
> > +++ b/drivers/tee/optee/smc_abi.c
> > @@ -81,20 +81,26 @@ static int optee_cpuhp_disable_pcpu_irq(unsigned int cpu)
> > */
> >
> > static int from_msg_param_tmp_mem(struct tee_param *p, u32 attr,
> > - const struct optee_msg_param *mp)
> > + const struct optee_msg_param *mp,
> > + bool update_out)
> > {
> > struct tee_shm *shm;
> > phys_addr_t pa;
> > int rc;
> >
> > + if (update_out) {
> > + if (attr == OPTEE_MSG_ATTR_TYPE_TMEM_INPUT)
> > + return 0;
> > + goto out;
> > + }
> > +
> > p->attr = TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_INPUT +
> > attr - OPTEE_MSG_ATTR_TYPE_TMEM_INPUT;
> > - p->u.memref.size = mp->u.tmem.size;
> > shm = (struct tee_shm *)(unsigned long)mp->u.tmem.shm_ref;
> > if (!shm) {
> > p->u.memref.shm_offs = 0;
> > p->u.memref.shm = NULL;
> > - return 0;
> > + goto out;
> > }
> >
> > rc = tee_shm_get_pa(shm, 0, &pa);
> > @@ -103,18 +109,25 @@ static int from_msg_param_tmp_mem(struct tee_param *p, u32 attr,
> >
> > p->u.memref.shm_offs = mp->u.tmem.buf_ptr - pa;
> > p->u.memref.shm = shm;
> > -
> > +out:
> > + p->u.memref.size = mp->u.tmem.size;
> > return 0;
> > }
> >
> > static void from_msg_param_reg_mem(struct tee_param *p, u32 attr,
> > - const struct optee_msg_param *mp)
> > + const struct optee_msg_param *mp,
> > + bool update_out)
> > {
> > struct tee_shm *shm;
> >
> > + if (update_out) {
> > + if (attr == OPTEE_MSG_ATTR_TYPE_RMEM_INPUT)
> > + return;
> > + goto out;
> > + }
> > +
> > p->attr = TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_INPUT +
> > attr - OPTEE_MSG_ATTR_TYPE_RMEM_INPUT;
> > - p->u.memref.size = mp->u.rmem.size;
> > shm = (struct tee_shm *)(unsigned long)mp->u.rmem.shm_ref;
> >
> > if (shm) {
> > @@ -124,6 +137,8 @@ static void from_msg_param_reg_mem(struct tee_param *p, u32 attr,
> > p->u.memref.shm_offs = 0;
> > p->u.memref.shm = NULL;
> > }
> > +out:
> > + p->u.memref.size = mp->u.rmem.size;
> > }
> >
> > /**
> > @@ -133,11 +148,13 @@ static void from_msg_param_reg_mem(struct tee_param *p, u32 attr,
> > * @params: subsystem internal parameter representation
> > * @num_params: number of elements in the parameter arrays
> > * @msg_params: OPTEE_MSG parameters
> > + * @update_out: update parameter for output only
> > * Returns 0 on success or <0 on failure
> > */
> > static int optee_from_msg_param(struct optee *optee, struct tee_param *params,
> > size_t num_params,
> > - const struct optee_msg_param *msg_params)
> > + const struct optee_msg_param *msg_params,
> > + bool update_out)
> > {
> > int rc;
> > size_t n;
> > @@ -149,25 +166,27 @@ static int optee_from_msg_param(struct optee *optee, struct tee_param *params,
> >
> > switch (attr) {
> > case OPTEE_MSG_ATTR_TYPE_NONE:
> > + if (update_out)
> > + break;
> > p->attr = TEE_IOCTL_PARAM_ATTR_TYPE_NONE;
> > memset(&p->u, 0, sizeof(p->u));
> > break;
> > case OPTEE_MSG_ATTR_TYPE_VALUE_INPUT:
> > case OPTEE_MSG_ATTR_TYPE_VALUE_OUTPUT:
> > case OPTEE_MSG_ATTR_TYPE_VALUE_INOUT:
> > - optee_from_msg_param_value(p, attr, mp);
> > + optee_from_msg_param_value(p, attr, mp, update_out);
> > break;
> > case OPTEE_MSG_ATTR_TYPE_TMEM_INPUT:
> > case OPTEE_MSG_ATTR_TYPE_TMEM_OUTPUT:
> > case OPTEE_MSG_ATTR_TYPE_TMEM_INOUT:
> > - rc = from_msg_param_tmp_mem(p, attr, mp);
> > + rc = from_msg_param_tmp_mem(p, attr, mp, update_out);
> > if (rc)
> > return rc;
> > break;
> > case OPTEE_MSG_ATTR_TYPE_RMEM_INPUT:
> > case OPTEE_MSG_ATTR_TYPE_RMEM_OUTPUT:
> > case OPTEE_MSG_ATTR_TYPE_RMEM_INOUT:
> > - from_msg_param_reg_mem(p, attr, mp);
> > + from_msg_param_reg_mem(p, attr, mp, update_out);
> > break;
> >
> > default:
> > @@ -178,20 +197,25 @@ static int optee_from_msg_param(struct optee *optee, struct tee_param *params,
> > }
> >
> > static int to_msg_param_tmp_mem(struct optee_msg_param *mp,
> > - const struct tee_param *p)
> > + const struct tee_param *p, bool update_out)
> > {
> > int rc;
> > phys_addr_t pa;
> >
> > + if (update_out) {
> > + if (p->attr == TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_INPUT)
> > + return 0;
> > + goto out;
> > + }
> > +
> > mp->attr = OPTEE_MSG_ATTR_TYPE_TMEM_INPUT + p->attr -
> > TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_INPUT;
> >
> > mp->u.tmem.shm_ref = (unsigned long)p->u.memref.shm;
> > - mp->u.tmem.size = p->u.memref.size;
> >
> > if (!p->u.memref.shm) {
> > mp->u.tmem.buf_ptr = 0;
> > - return 0;
> > + goto out;
> > }
> >
> > rc = tee_shm_get_pa(p->u.memref.shm, p->u.memref.shm_offs, &pa);
> > @@ -201,19 +225,27 @@ static int to_msg_param_tmp_mem(struct optee_msg_param *mp,
> > mp->u.tmem.buf_ptr = pa;
> > mp->attr |= OPTEE_MSG_ATTR_CACHE_PREDEFINED <<
> > OPTEE_MSG_ATTR_CACHE_SHIFT;
> > -
> > +out:
> > + mp->u.tmem.size = p->u.memref.size;
> > return 0;
> > }
> >
> > static int to_msg_param_reg_mem(struct optee_msg_param *mp,
> > - const struct tee_param *p)
> > + const struct tee_param *p, bool update_out)
> > {
> > + if (update_out) {
> > + if (p->attr == TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_INPUT)
> > + return 0;
> > + goto out;
> > + }
> > +
> > mp->attr = OPTEE_MSG_ATTR_TYPE_RMEM_INPUT + p->attr -
> > TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_INPUT;
> >
> > mp->u.rmem.shm_ref = (unsigned long)p->u.memref.shm;
> > - mp->u.rmem.size = p->u.memref.size;
> > mp->u.rmem.offs = p->u.memref.shm_offs;
> > +out:
> > + mp->u.rmem.size = p->u.memref.size;
> > return 0;
> > }
> >
> > @@ -223,11 +255,13 @@ static int to_msg_param_reg_mem(struct optee_msg_param *mp,
> > * @msg_params: OPTEE_MSG parameters
> > * @num_params: number of elements in the parameter arrays
> > * @params: subsystem itnernal parameter representation
> > + * @update_out: update parameter for output only
> > * Returns 0 on success or <0 on failure
> > */
> > static int optee_to_msg_param(struct optee *optee,
> > struct optee_msg_param *msg_params,
> > - size_t num_params, const struct tee_param *params)
> > + size_t num_params, const struct tee_param *params,
> > + bool update_out)
> > {
> > int rc;
> > size_t n;
> > @@ -238,21 +272,23 @@ static int optee_to_msg_param(struct optee *optee,
> >
> > switch (p->attr) {
> > case TEE_IOCTL_PARAM_ATTR_TYPE_NONE:
> > + if (update_out)
> > + break;
> > mp->attr = TEE_IOCTL_PARAM_ATTR_TYPE_NONE;
> > memset(&mp->u, 0, sizeof(mp->u));
> > break;
> > case TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_INPUT:
> > case TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_OUTPUT:
> > case TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_INOUT:
> > - optee_to_msg_param_value(mp, p);
> > + optee_to_msg_param_value(mp, p, update_out);
> > break;
> > case TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_INPUT:
> > case TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_OUTPUT:
> > case TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_INOUT:
> > if (tee_shm_is_dynamic(p->u.memref.shm))
> > - rc = to_msg_param_reg_mem(mp, p);
> > + rc = to_msg_param_reg_mem(mp, p, update_out);
> > else
> > - rc = to_msg_param_tmp_mem(mp, p);
> > + rc = to_msg_param_tmp_mem(mp, p, update_out);
> > if (rc)
> > return rc;
> > break;
> > --
> > 2.43.0
> >
