linux-imx

// SPDX-License-Identifier: GPL-2.0

/*

 * Copyright (C) 2019 Linaro Ltd.

 */

​

#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt

​

#include <linux/kernel.h>

#include <linux/slab.h>

#include <linux/tee_drv.h>

#include <linux/uuid.h>

#include "optee_private.h"

​

/*

 * Get device UUIDs

 *

 * [out]     memref[0]        Array of device UUIDs

 *

 * Return codes:

 * TEE_SUCCESS - Invoke command success

 * TEE_ERROR_BAD_PARAMETERS - Incorrect input param

 * TEE_ERROR_SHORT_BUFFER - Output buffer size less than required

 */

#define PTA_CMD_GET_DEVICES     0x0

​

static int optee_ctx_match(struct tee_ioctl_version_data *ver, const void *data)

{

    if (ver->impl_id == TEE_IMPL_ID_OPTEE)

        return 1;

    else

        return 0;

}

​

static int get_devices(struct tee_context *ctx, u32 session,

               struct tee_shm *device_shm, u32 *shm_size)

{

    int ret = 0;

    struct tee_ioctl_invoke_arg inv_arg;

    struct tee_param param[4];

​

    memset(&inv_arg, 0, sizeof(inv_arg));

    memset(&param, 0, sizeof(param));

​

    /* Invoke PTA_CMD_GET_DEVICES function */

    inv_arg.func = PTA_CMD_GET_DEVICES;

    inv_arg.session = session;

    inv_arg.num_params = 4;

​

    /* Fill invoke cmd params */

    param[0].attr = TEE_IOCTL_PARAM_ATTR_TYPE_MEMREF_OUTPUT;

    param[0].u.memref.shm = device_shm;

    param[0].u.memref.size = *shm_size;

    param[0].u.memref.shm_offs = 0;

​

    ret = tee_client_invoke_func(ctx, &inv_arg, param);

    if ((ret < 0) || ((inv_arg.ret != TEEC_SUCCESS) &&

              (inv_arg.ret != TEEC_ERROR_SHORT_BUFFER))) {

        pr_err("PTA_CMD_GET_DEVICES invoke function err: %x\n",

               inv_arg.ret);

        return -EINVAL;

    }

​

    *shm_size = param[0].u.memref.size;