text: /*
   * arch/arm/mach-mediatek/platsmp.c
   *
   * Copyright (c) 2014 Mediatek Inc.
   * Author: Shunli Wang <shunli.wang@mediatek.com>
   *         Yingjoe Chen <yingjoe.chen@mediatek.com>
   *
   * 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/io.h>
  #include <linux/memblock.h>
  #include <linux/of.h>
  #include <linux/of_address.h>
  #include <linux/string.h>
  #include <linux/threads.h>
  
  #define MTK_MAX_CPU		8
  #define MTK_SMP_REG_SIZE	0x1000
  
  struct mtk_smp_boot_info {
  	unsigned long smp_base;
  	unsigned int jump_reg;
  	unsigned int core_keys[MTK_MAX_CPU - 1];
  	unsigned int core_regs[MTK_MAX_CPU - 1];
  };
  
  static const struct mtk_smp_boot_info mtk_mt8135_tz_boot = {
  	0x80002000, 0x3fc,
  	{ 0x534c4131, 0x4c415332, 0x41534c33 },
  	{ 0x3f8, 0x3f8, 0x3f8 },
  };
  
  static const struct mtk_smp_boot_info mtk_mt6589_boot = {
  	0x10002000, 0x34,
  	{ 0x534c4131, 0x4c415332, 0x41534c33 },
  	{ 0x38, 0x3c, 0x40 },
  };
  
  static const struct of_device_id mtk_tz_smp_boot_infos[] __initconst = {
  	{ .compatible   = "mediatek,mt8135", .data = &mtk_mt8135_tz_boot },
  	{ .compatible   = "mediatek,mt8127", .data = &mtk_mt8135_tz_boot },
  	{ .compatible   = "mediatek,mt2701", .data = &mtk_mt8135_tz_boot },
  };
  
  static const struct of_device_id mtk_smp_boot_infos[] __initconst = {
  	{ .compatible   = "mediatek,mt6589", .data = &mtk_mt6589_boot },
  };
  
  static void __iomem *mtk_smp_base;
  static const struct mtk_smp_boot_info *mtk_smp_info;
  
  static int mtk_boot_secondary(unsigned int cpu, struct task_struct *idle)
  {
  	if (!mtk_smp_base)
  		return -EINVAL;
  
  	if (!mtk_smp_info->core_keys[cpu-1])
  		return -EINVAL;
  
  	writel_relaxed(mtk_smp_info->core_keys[cpu-1],
  		mtk_smp_base + mtk_smp_info->core_regs[cpu-1]);
  
  	arch_send_wakeup_ipi_mask(cpumask_of(cpu));
  
  	return 0;
  }
  
  static void __init __mtk_smp_prepare_cpus(unsigned int max_cpus, int trustzone)
  {
  	int i, num;
  	const struct of_device_id *infos;
  
  	if (trustzone) {
  		num = ARRAY_SIZE(mtk_tz_smp_boot_infos);
  		infos = mtk_tz_smp_boot_infos;
  	} else {
  		num = ARRAY_SIZE(mtk_smp_boot_infos);
  		infos = mtk_smp_boot_infos;
  	}
  
  	/* Find smp boot info for this SoC */
  	for (i = 0; i < num; i++) {
  		if (of_machine_is_compatible(infos[i].compatible)) {
  			mtk_smp_info = infos[i].data;
  			break;
  		}
  	}
  
  	if (!mtk_smp_info) {
  		pr_err("%s: Device is not supported\n", __func__);
  		return;
  	}
  
  	if (trustzone) {
  		if (memblock_reserve(mtk_smp_info->smp_base, MTK_SMP_REG_SIZE)) {
  			pr_err("%s: Can't reserve smp memory\n", __func__);
  			return;
  		}
  		mtk_smp_base = phys_to_virt(mtk_smp_info->smp_base);
  	} else {
  		mtk_smp_base = ioremap(mtk_smp_info->smp_base, MTK_SMP_REG_SIZE);
  		if (!mtk_smp_base) {
  			pr_err("%s: Can't remap %lx\n", __func__,
  				mtk_smp_info->smp_base);
  			return;
  		}
  	}
  
  	/*
  	 * write the address of slave startup address into the system-wide
  	 * jump register
  	 */
  	writel_relaxed(virt_to_phys(secondary_startup_arm),
  			mtk_smp_base + mtk_smp_info->jump_reg);
  }
  
  static void __init mtk_tz_smp_prepare_cpus(unsigned int max_cpus)
  {
  	__mtk_smp_prepare_cpus(max_cpus, 1);
  }
  
  static void __init mtk_smp_prepare_cpus(unsigned int max_cpus)
  {
  	__mtk_smp_prepare_cpus(max_cpus, 0);
  }
  
  static struct smp_operations mt81xx_tz_smp_ops __initdata = {
  	.smp_prepare_cpus = mtk_tz_smp_prepare_cpus,
  	.smp_boot_secondary = mtk_boot_secondary,
  };
  CPU_METHOD_OF_DECLARE(mt81xx_tz_smp, "mediatek,mt81xx-tz-smp", &mt81xx_tz_smp_ops);
  
  static struct smp_operations mt6589_smp_ops __initdata = {
  	.smp_prepare_cpus = mtk_smp_prepare_cpus,
  	.smp_boot_secondary = mtk_boot_secondary,
  };
  CPU_METHOD_OF_DECLARE(mt6589_smp, "mediatek,mt6589-smp", &mt6589_smp_ops);
  