/*
 * Copyright (C) 2015 MediaTek Inc.
 *
 * 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.
 */


#include <linux/module.h>
#include <linux/platform_device.h>
#include <linux/of.h>
#include <linux/of_platform.h>
#include <linux/pm_runtime.h>

#include "trustzone/kree/system.h"
#include "tz_cross/ta_pm.h"
#include "tz_cross/ree_service.h"
#include "kree_int.h"

static KREE_SESSION_HANDLE pm_session;

void kree_pm_init(void)
{
	int ret;

	ret = KREE_CreateSession(TZ_TA_PM_UUID, &pm_session);
	if (ret != TZ_RESULT_SUCCESS)
		pr_warn("CreateSession error %d\n", ret);
}

#ifndef CONFIG_ARM64
void kree_pm_cpu_lowpower(int *ppen_release, int logical_cpuid)
{
	union MTEEC_PARAM param[4];
	int ret;

	param[0].mem.buffer = (void *)ppen_release;
	param[0].mem.size = sizeof(int);
	param[1].value.a = logical_cpuid;
	ret = KREE_TeeServiceCall(pm_session, TZCMD_PM_CPU_LOWPOWER,
					TZ_ParamTypes2(TZPT_MEM_INPUT,
							TZPT_VALUE_INPUT),
					param);
	if (ret != TZ_RESULT_SUCCESS)
		pr_warn("%s error: %s\n", __func__, TZ_GetErrorString(ret));
}

int kree_pm_cpu_dormant(int mode)
{
	union MTEEC_PARAM param[4];
	int ret;

	param[0].value.a = mode;
	ret = KREE_TeeServiceCall(pm_session, TZCMD_PM_CPU_DORMANT,
				  TZ_ParamTypes1(TZPT_VALUE_INPUT), param);
	if (ret != TZ_RESULT_SUCCESS)
		pr_warn("%s error: %s\n", __func__, TZ_GetErrorString(ret));

	return 0;
}

int kree_pm_cpu_dormant_workaround_wake(int workaround_wake)
{
	union MTEEC_PARAM param[4];
	int ret;

	param[0].value.a = workaround_wake;
	ret = KREE_TeeServiceCall(pm_session, TZCMD_PM_CPU_ERRATA_802022_WA,
				  TZ_ParamTypes1(TZPT_VALUE_INPUT), param);
	if (ret != TZ_RESULT_SUCCESS)
		pr_warn("%s error: %s\n", __func__, TZ_GetErrorString(ret));

	return 0;
}
#endif

int kree_pm_device_ops(int state)
{
	union MTEEC_PARAM param[4];
	int ret;

	param[0].value.a = (uint32_t) state;
	ret = KREE_TeeServiceCall(pm_session, TZCMD_PM_DEVICE_OPS,
				  TZ_ParamTypes1(TZPT_VALUE_INPUT), param);
	if (ret != TZ_RESULT_SUCCESS)
		pr_warn("%s error: %s\n", __func__, TZ_GetErrorString(ret));

	return ret;
}

int KREE_ServPMGet(u32 op, u8 uparam[REE_SERVICE_BUFFER_SIZE])
{
	struct device *dev;
	int *res = (int *)uparam;

	dev = mtee_pmdev_get(uparam);

	if (dev) {
		*res = pm_runtime_get_sync(dev);
		return TZ_RESULT_SUCCESS;
	}

	return TZ_RESULT_ERROR_GENERIC;
}

int KREE_ServPMPut(u32 op, u8 uparam[REE_SERVICE_BUFFER_SIZE])
{
	struct device *dev;
	int *res = (int *)uparam;

	dev = mtee_pmdev_get(uparam);

	if (dev) {
		*res = pm_runtime_put_sync(dev);
		return TZ_RESULT_SUCCESS;
	}

	return TZ_RESULT_ERROR_GENERIC;
}