Files
SDK_RK3288/kernel/security/tlk_driver/ote_comms.c

549 lines
13 KiB
C

/*
* Copyright (c) 2012-2014 NVIDIA Corporation. All rights reserved.
*
* This program is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
* more details.
*
* You should have received a copy of the GNU General Public License along
* with this program; if not, write to the Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/
#include <linux/atomic.h>
#include <linux/uaccess.h>
#include <linux/module.h>
#include <linux/slab.h>
#include <linux/fs.h>
#include <linux/printk.h>
#include <linux/ioctl.h>
#include <linux/sched.h>
#include <linux/mm.h>
#include <linux/pagemap.h>
#include <asm/smp_plat.h>
#include "ote_protocol.h"
bool verbose_smc;
core_param(verbose_smc, verbose_smc, bool, 0644);
#define SET_RESULT(req, r, ro) { req->result = r; req->result_origin = ro; }
static struct te_shmem_desc *te_add_shmem_desc(void *buffer, size_t size,
struct tlk_context *context)
{
struct te_shmem_desc *shmem_desc = NULL;
shmem_desc = kzalloc(sizeof(struct te_shmem_desc), GFP_KERNEL);
if (shmem_desc) {
INIT_LIST_HEAD(&(shmem_desc->list));
shmem_desc->buffer = buffer;
shmem_desc->size = size;
list_add_tail(&shmem_desc->list, &(context->shmem_alloc_list));
}
return shmem_desc;
}
static int te_pin_mem_buffers(void *buffer, size_t size,
struct tlk_context *context)
{
struct te_shmem_desc *shmem_desc = NULL;
int ret = 0;
shmem_desc = te_add_shmem_desc(buffer, size, context);
if (!shmem_desc) {
pr_err("%s: te_add_shmem_desc Failed\n", __func__);
ret = OTE_ERROR_OUT_OF_MEMORY;
goto error;
}
return OTE_SUCCESS;
error:
return ret;
}
static int te_setup_temp_buffers(struct te_request *request,
struct tlk_context *context)
{
uint32_t i;
int ret = OTE_SUCCESS;
struct te_oper_param *params = request->params;
for (i = 0; i < request->params_size; i++) {
switch (params[i].type) {
case TE_PARAM_TYPE_NONE:
case TE_PARAM_TYPE_INT_RO:
case TE_PARAM_TYPE_INT_RW:
break;
case TE_PARAM_TYPE_MEM_RO:
case TE_PARAM_TYPE_MEM_RW:
ret = te_pin_mem_buffers(
params[i].u.Mem.base,
params[i].u.Mem.len,
context);
if (ret < 0) {
pr_err("%s failed with err (%d)\n",
__func__, ret);
ret = OTE_ERROR_BAD_PARAMETERS;
break;
}
break;
default:
pr_err("%s: OTE_ERROR_BAD_PARAMETERS\n", __func__);
ret = OTE_ERROR_BAD_PARAMETERS;
break;
}
}
return ret;
}
static int te_setup_temp_buffers_compat(struct te_request_compat *request,
struct tlk_context *context)
{
uint32_t i;
int ret = OTE_SUCCESS;
struct te_oper_param_compat *params;
params = (struct te_oper_param_compat *)(uintptr_t)request->params;
for (i = 0; i < request->params_size; i++) {
switch (params[i].type) {
case TE_PARAM_TYPE_NONE:
case TE_PARAM_TYPE_INT_RO:
case TE_PARAM_TYPE_INT_RW:
break;
case TE_PARAM_TYPE_MEM_RO:
case TE_PARAM_TYPE_MEM_RW:
ret = te_pin_mem_buffers(
(void *)(uintptr_t)params[i].u.Mem.base,
params[i].u.Mem.len,
context);
if (ret < 0) {
pr_err("%s failed with err (%d)\n",
__func__, ret);
ret = OTE_ERROR_BAD_PARAMETERS;
break;
}
break;
default:
pr_err("%s: OTE_ERROR_BAD_PARAMETERS\n", __func__);
ret = OTE_ERROR_BAD_PARAMETERS;
break;
}
}
return ret;
}
static void te_del_shmem_desc(void *buffer, struct tlk_context *context)
{
struct te_shmem_desc *shmem_desc, *tmp_shmem_desc;
list_for_each_entry_safe(shmem_desc, tmp_shmem_desc,
&(context->shmem_alloc_list), list) {
if (shmem_desc->buffer == buffer) {
list_del(&shmem_desc->list);
kfree(shmem_desc);
}
}
}
/*
* Deregister previously initialized shared memory
*/
void te_unregister_memory(void *buffer,
struct tlk_context *context)
{
if (!(list_empty(&(context->shmem_alloc_list))))
te_del_shmem_desc(buffer, context);
else
pr_err("No buffers to unpin\n");
}
static void te_unpin_temp_buffers(struct te_request *request,
struct tlk_context *context)
{
uint32_t i;
struct te_oper_param *params = request->params;
for (i = 0; i < request->params_size; i++) {
switch (params[i].type) {
case TE_PARAM_TYPE_NONE:
case TE_PARAM_TYPE_INT_RO:
case TE_PARAM_TYPE_INT_RW:
break;
case TE_PARAM_TYPE_MEM_RO:
case TE_PARAM_TYPE_MEM_RW:
te_unregister_memory(params[i].u.Mem.base, context);
break;
default:
pr_err("%s: OTE_ERROR_BAD_PARAMETERS\n", __func__);
break;
}
}
}
static void te_unpin_temp_buffers_compat(struct te_request_compat *request,
struct tlk_context *context)
{
uint32_t i;
struct te_oper_param_compat *params;
params = (struct te_oper_param_compat *)(uintptr_t)request->params;
for (i = 0; i < request->params_size; i++) {
switch (params[i].type) {
case TE_PARAM_TYPE_NONE:
case TE_PARAM_TYPE_INT_RO:
case TE_PARAM_TYPE_INT_RW:
break;
case TE_PARAM_TYPE_MEM_RO:
case TE_PARAM_TYPE_MEM_RW:
te_unregister_memory(
(void *)(uintptr_t)params[i].u.Mem.base,
context);
break;
default:
pr_err("%s: OTE_ERROR_BAD_PARAMETERS\n", __func__);
break;
}
}
}
#ifdef CONFIG_SMP
cpumask_t saved_cpu_mask;
static void switch_cpumask_to_cpu0(void)
{
long ret;
cpumask_t local_cpu_mask = CPU_MASK_NONE;
cpu_set(0, local_cpu_mask);
cpumask_copy(&saved_cpu_mask, tsk_cpus_allowed(current));
ret = sched_setaffinity(0, &local_cpu_mask);
if (ret)
pr_err("sched_setaffinity #1 -> 0x%lX", ret);
}
static void restore_cpumask(void)
{
long ret = sched_setaffinity(0, &saved_cpu_mask);
if (ret)
pr_err("sched_setaffinity #2 -> 0x%lX", ret);
}
#else
static inline void switch_cpumask_to_cpu0(void) {};
static inline void restore_cpumask(void) {};
#endif
uint32_t tlk_generic_smc(uint32_t arg0, uintptr_t arg1, uintptr_t arg2)
{
uint32_t retval;
switch_cpumask_to_cpu0();
retval = _tlk_generic_smc(arg0, arg1, arg2);
while (retval == TE_ERROR_PREEMPT_BY_IRQ ||
retval == TE_ERROR_PREEMPT_BY_FS) {
if (retval == TE_ERROR_PREEMPT_BY_IRQ) {
retval = _tlk_generic_smc((60 << 24), 0, 0);
} else {
tlk_ss_op();
retval = _tlk_generic_smc(TE_SMC_SS_REQ_COMPLETE, 0, 0);
}
}
restore_cpumask();
/* Print TLK logs if any */
ote_print_logs();
return retval;
}
uint32_t tlk_extended_smc(uintptr_t *regs)
{
uint32_t retval;
switch_cpumask_to_cpu0();
retval = _tlk_extended_smc(regs);
while (retval == 0xFFFFFFFD)
retval = _tlk_generic_smc((60 << 24), 0, 0);
restore_cpumask();
/* Print TLK logs if any */
ote_print_logs();
return retval;
}
/*
* Do an SMC call
*/
static void do_smc(struct te_request *request, struct tlk_device *dev)
{
uint32_t smc_args;
uint32_t smc_params = 0;
if (dev->req_param_buf) {
smc_args = (char *)request - dev->req_param_buf;
if (request->params)
smc_params = (char *)request->params -
dev->req_param_buf;
} else {
smc_args = (uint32_t)virt_to_phys(request);
if (request->params)
smc_params = (uint32_t)virt_to_phys(request->params);
}
tlk_generic_smc(request->type, smc_args, smc_params);
}
/*
* Do an SMC call
*/
static void do_smc_compat(struct te_request_compat *request,
struct tlk_device *dev)
{
uint32_t smc_args;
uint32_t smc_params = 0;
smc_args = (char *)request - dev->req_param_buf;
if (request->params) {
smc_params =
(char *)(uintptr_t)request->params - dev->req_param_buf;
}
tlk_generic_smc(request->type, smc_args, smc_params);
}
struct tlk_smc_work_args {
uint32_t arg0;
uint32_t arg1;
uint32_t arg2;
};
static long tlk_generic_smc_on_cpu0(void *args)
{
struct tlk_smc_work_args *work;
int cpu = cpu_logical_map(smp_processor_id());
uint32_t retval;
BUG_ON(cpu != 0);
work = (struct tlk_smc_work_args *)args;
retval = _tlk_generic_smc(work->arg0, work->arg1, work->arg2);
while (retval == 0xFFFFFFFD)
retval = _tlk_generic_smc((60 << 24), 0, 0);
return retval;
}
/*
* VPR programming SMC
*
* This routine is called both from normal threads and worker threads.
* The worker threads are per-cpu and have PF_NO_SETAFFINITY set, so
* any calls to sched_setaffinity will fail.
*
* If it's a worker thread on CPU0, just invoke the SMC directly. If
* it's running on a non-CPU0, use work_on_cpu() to schedule the SMC
* on CPU0.
*/
int te_set_vpr_params(void *vpr_base, size_t vpr_size)
{
uint32_t retval;
/* Share the same lock used when request is send from user side */
mutex_lock(&smc_lock);
if (current->flags &
(PF_WQ_WORKER | PF_NO_SETAFFINITY | PF_KTHREAD)) {
struct tlk_smc_work_args work_args;
int cpu = cpu_logical_map(smp_processor_id());
work_args.arg0 = TE_SMC_PROGRAM_VPR;
work_args.arg1 = (uint32_t)vpr_base;
work_args.arg2 = vpr_size;
/* workers don't change CPU. depending on the CPU, execute
* directly or sched work */
if (cpu == 0 && (current->flags & PF_WQ_WORKER))
retval = tlk_generic_smc_on_cpu0(&work_args);
else
retval = work_on_cpu(0,
tlk_generic_smc_on_cpu0, &work_args);
} else {
retval = tlk_generic_smc(TE_SMC_PROGRAM_VPR,
(uintptr_t)vpr_base, vpr_size);
}
mutex_unlock(&smc_lock);
if (retval != OTE_SUCCESS) {
pr_err("te_set_vpr_params failed err (0x%x)\n", retval);
return -EINVAL;
}
return 0;
}
EXPORT_SYMBOL(te_set_vpr_params);
/*
* Open session SMC (supporting client-based te_open_session() calls)
*/
void te_open_session(struct te_opensession *cmd,
struct te_request *request,
struct tlk_context *context)
{
int ret;
ret = te_setup_temp_buffers(request, context);
if (ret != OTE_SUCCESS) {
pr_err("te_setup_temp_buffers failed err (0x%x)\n", ret);
SET_RESULT(request, ret, OTE_RESULT_ORIGIN_API);
return;
}
memcpy(&request->dest_uuid,
&cmd->dest_uuid,
sizeof(struct te_service_id));
pr_info("OPEN_CLIENT_SESSION: 0x%x 0x%x 0x%x 0x%x\n",
request->dest_uuid[0],
request->dest_uuid[1],
request->dest_uuid[2],
request->dest_uuid[3]);
request->type = TE_SMC_OPEN_SESSION;
do_smc(request, context->dev);
te_unpin_temp_buffers(request, context);
}
/*
* Close session SMC (supporting client-based te_close_session() calls)
*/
void te_close_session(struct te_closesession *cmd,
struct te_request *request,
struct tlk_context *context)
{
request->session_id = cmd->session_id;
request->type = TE_SMC_CLOSE_SESSION;
do_smc(request, context->dev);
if (request->result)
pr_info("Error closing session: %08x\n", request->result);
}
/*
* Launch operation SMC (supporting client-based te_launch_operation() calls)
*/
void te_launch_operation(struct te_launchop *cmd,
struct te_request *request,
struct tlk_context *context)
{
int ret;
ret = te_setup_temp_buffers(request, context);
if (ret != OTE_SUCCESS) {
pr_err("te_setup_temp_buffers failed err (0x%x)\n", ret);
SET_RESULT(request, ret, OTE_RESULT_ORIGIN_API);
return;
}
request->session_id = cmd->session_id;
request->command_id = cmd->operation.command;
request->type = TE_SMC_LAUNCH_OPERATION;
do_smc(request, context->dev);
te_unpin_temp_buffers(request, context);
}
/*
* Open session SMC (supporting client-based te_open_session() calls)
*/
void te_open_session_compat(struct te_opensession_compat *cmd,
struct te_request_compat *request,
struct tlk_context *context)
{
int ret;
ret = te_setup_temp_buffers_compat(request, context);
if (ret != OTE_SUCCESS) {
pr_err("te_setup_temp_buffers failed err (0x%x)\n", ret);
SET_RESULT(request, ret, OTE_RESULT_ORIGIN_API);
return;
}
memcpy(&request->dest_uuid,
&cmd->dest_uuid,
sizeof(struct te_service_id));
pr_info("OPEN_CLIENT_SESSION_COMPAT: 0x%x 0x%x 0x%x 0x%x\n",
request->dest_uuid[0],
request->dest_uuid[1],
request->dest_uuid[2],
request->dest_uuid[3]);
request->type = TE_SMC_OPEN_SESSION;
do_smc_compat(request, context->dev);
te_unpin_temp_buffers_compat(request, context);
}
/*
* Close session SMC (supporting client-based te_close_session() calls)
*/
void te_close_session_compat(struct te_closesession_compat *cmd,
struct te_request_compat *request,
struct tlk_context *context)
{
request->session_id = cmd->session_id;
request->type = TE_SMC_CLOSE_SESSION;
do_smc_compat(request, context->dev);
if (request->result)
pr_info("Error closing session: %08x\n", request->result);
}
/*
* Launch operation SMC (supporting client-based te_launch_operation() calls)
*/
void te_launch_operation_compat(struct te_launchop_compat *cmd,
struct te_request_compat *request,
struct tlk_context *context)
{
int ret;
ret = te_setup_temp_buffers_compat(request, context);
if (ret != OTE_SUCCESS) {
pr_err("te_setup_temp_buffers failed err (0x%x)\n", ret);
SET_RESULT(request, ret, OTE_RESULT_ORIGIN_API);
return;
}
request->session_id = cmd->session_id;
request->command_id = cmd->operation.command;
request->type = TE_SMC_LAUNCH_OPERATION;
do_smc_compat(request, context->dev);
te_unpin_temp_buffers_compat(request, context);
}
static int __init tlk_register_irq_handler(void)
{
tlk_generic_smc(TE_SMC_REGISTER_IRQ_HANDLER,
(uintptr_t)tlk_irq_handler, 0);
return 0;
}
arch_initcall(tlk_register_irq_handler);