/*
 * Copyright (c) 2015-2019, MICROTRUST Incorporated
 * 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
 * version 2 as published by the Free Software Foundation.
 *
 * 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.
 */

#ifndef SOTER_SMC_H
#define SOTER_SMC_H

#include <linux/string.h>

/* Let's reuse optee's message format */
#include "optee_msg.h"

enum {
	NQ_CMD_CLIENT_API_REQUEST,
	NQ_CMD_GPTEE_CLIENT_API_REQUEST,
	NQ_CMD_REGISTER_SHM_POOL,
};

extern struct semaphore keymaster_api_lock;

int teei_forward_call(unsigned long long cmd, unsigned long long cmd_addr,
				unsigned long long size);

static inline int soter_do_call_with_arg(struct tee_context *ctx,
					phys_addr_t parg)
{
	if (strncmp(ctx->hostname, "bta_loader", TEE_MAX_HOSTNAME_SIZE) == 0)
		return teei_forward_call(NQ_CMD_CLIENT_API_REQUEST, parg,
						sizeof(struct optee_msg_arg));

	if (strncmp(ctx->hostname, "tta", TEE_MAX_HOSTNAME_SIZE) == 0)
		return teei_forward_call(NQ_CMD_GPTEE_CLIENT_API_REQUEST, parg,
						sizeof(struct optee_msg_arg));

	IMSG_ERROR("Unrecognized hostname: '%s'\n", ctx->hostname);

	return -EINVAL;
}

static inline int soter_register_shm_pool(phys_addr_t shm_pa, int shm_size)
{
	int retVal = 0;

	retVal = teei_forward_call(NQ_CMD_REGISTER_SHM_POOL, shm_pa, shm_size);

	return retVal;

}

#endif