// SPDX-License-Identifier: GPL-2.0-only
/*
 * linux/arch/arm/mach-vexpress/mcpm_platsmp.c
 *
 * Created by:  Nicolas Pitre, November 2012
 * Copyright:   (C) 2012-2013  Linaro Limited
 *
 * Code to handle secondary CPU bringup and hotplug for the cluster power API.
 */

#include <linux/init.h>
#include <linux/smp.h>
#include <linux/spinlock.h>

#include <asm/mcpm.h>
#include <asm/smp.h>
#include <asm/smp_plat.h>

static void cpu_to_pcpu(unsigned int cpu,
                        unsigned int *pcpu, unsigned int *pcluster)
{
        unsigned int mpidr;

        mpidr = cpu_logical_map(cpu);
        *pcpu = MPIDR_AFFINITY_LEVEL(mpidr, 0);
        *pcluster = MPIDR_AFFINITY_LEVEL(mpidr, 1);
}

static int mcpm_boot_secondary(unsigned int cpu, struct task_struct *idle)
{
        unsigned int pcpu, pcluster, ret;
        extern void secondary_startup(void);

        cpu_to_pcpu(cpu, &pcpu, &pcluster);

        pr_debug("%s: logical CPU %d is physical CPU %d cluster %d\n",
                 __func__, cpu, pcpu, pcluster);

        mcpm_set_entry_vector(pcpu, pcluster, NULL);
        ret = mcpm_cpu_power_up(pcpu, pcluster);
        if (ret)
                return ret;
        mcpm_set_entry_vector(pcpu, pcluster, secondary_startup);
        arch_send_wakeup_ipi_mask(cpumask_of(cpu));
        dsb_sev();
        return 0;
}

static void mcpm_secondary_init(unsigned int cpu)
{
        mcpm_cpu_powered_up();
}

#ifdef CONFIG_HOTPLUG_CPU

static int mcpm_cpu_kill(unsigned int cpu)
{
        unsigned int pcpu, pcluster;

        cpu_to_pcpu(cpu, &pcpu, &pcluster);

        return !mcpm_wait_for_cpu_powerdown(pcpu, pcluster);
}

static bool mcpm_cpu_can_disable(unsigned int cpu)
{
        /* We assume all CPUs may be shut down. */
        return true;
}

static void mcpm_cpu_die(unsigned int cpu)
{
        unsigned int mpidr, pcpu, pcluster;
        mpidr = read_cpuid_mpidr();
        pcpu = MPIDR_AFFINITY_LEVEL(mpidr, 0);
        pcluster = MPIDR_AFFINITY_LEVEL(mpidr, 1);
        mcpm_set_entry_vector(pcpu, pcluster, NULL);
        mcpm_cpu_power_down();
}

#endif

static const struct smp_operations mcpm_smp_ops __initconst = {
        .smp_boot_secondary     = mcpm_boot_secondary,
        .smp_secondary_init     = mcpm_secondary_init,
#ifdef CONFIG_HOTPLUG_CPU
        .cpu_kill               = mcpm_cpu_kill,
        .cpu_can_disable        = mcpm_cpu_can_disable,
        .cpu_die                = mcpm_cpu_die,
#endif
};

void __init mcpm_smp_set_ops(void)
{
        smp_set_ops(&mcpm_smp_ops);
}