/* Copyright (c) 2010, Code Aurora Forum. 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 and
 * only 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.
 *
 * 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/module.h>
#include <linux/slab.h>

#include <soc/qcom/scm.h>
#include <soc/qcom/scm-boot.h>

#define SCM_FLAG_WARMBOOT_CPU0		0x04
#define SCM_FLAG_WARMBOOT_CPU1		0x02
#define SCM_FLAG_WARMBOOT_CPU2		0x10
#define SCM_FLAG_WARMBOOT_CPU3		0x40

struct scm_warmboot {
	int flag;
	void *entry;
};

static struct scm_warmboot scm_flags[] = {
	{ .flag = SCM_FLAG_WARMBOOT_CPU0 },
	{ .flag = SCM_FLAG_WARMBOOT_CPU1 },
	{ .flag = SCM_FLAG_WARMBOOT_CPU2 },
	{ .flag = SCM_FLAG_WARMBOOT_CPU3 },
};

/*
 * Set the cold/warm boot address for one of the CPU cores.
 */
int scm_set_boot_addr(phys_addr_t addr, int flags)
{
	struct {
		unsigned int flags;
		phys_addr_t  addr;
	} cmd;

	cmd.addr = addr;
	cmd.flags = flags;
	return scm_call(SCM_SVC_BOOT, SCM_BOOT_ADDR,
			&cmd, sizeof(cmd), NULL, 0);
}
EXPORT_SYMBOL(scm_set_boot_addr);

int scm_set_warm_boot_addr(void *entry, int cpu)
{
	int ret;

	/*
	 * Reassign only if we are switching from hotplug entry point
	 * to cpuidle entry point or vice versa.
	 */
	if (entry == scm_flags[cpu].entry)
		return 0;

	ret = scm_set_boot_addr(virt_to_phys(entry), scm_flags[cpu].flag);
	if (!ret)
		scm_flags[cpu].entry  = entry;

	return ret;
}