zephyr/arch/arm64/core/smp.c
Jiafei Pan 7889364771 arm64: refine the code for primary core checking
We can find caller of z_arm64_mmu_init is on primary
core or not, so no need to check mpidr, just add a
function parameter.

Signed-off-by: Jiafei Pan <Jiafei.Pan@nxp.com>
2021-04-19 11:00:05 -04:00

160 lines
3.4 KiB
C

/*
* Copyright 2020 NXP
*
* SPDX-License-Identifier: Apache-2.0
*
*/
/**
* @file
* @brief codes required for AArch64 multicore and Zephyr smp support
*/
#include <cache.h>
#include <device.h>
#include <kernel.h>
#include <kernel_structs.h>
#include <ksched.h>
#include <soc.h>
#include <init.h>
#include <arch/arm64/arm_mmu.h>
#include <arch/cpu.h>
#include <drivers/interrupt_controller/gic.h>
#include <drivers/pm_cpu_ops.h>
#include <sys/arch_interface.h>
#define SGI_SCHED_IPI 0
#define SGI_PTABLE_IPI 1
volatile struct {
void *sp; /* Fixed at the first entry */
arch_cpustart_t fn;
void *arg;
} __aligned(L1_CACHE_BYTES) arm64_cpu_init[CONFIG_MP_NUM_CPUS];
extern void __start(void);
/* Called from Zephyr initialization */
void arch_start_cpu(int cpu_num, k_thread_stack_t *stack, int sz,
arch_cpustart_t fn, void *arg)
{
__ASSERT(sizeof(arm64_cpu_init[0]) == ARM64_CPU_INIT_SIZE,
"ARM64_CPU_INIT_SIZE != sizeof(arm64_cpu_init[0]\n");
arm64_cpu_init[cpu_num].fn = fn;
arm64_cpu_init[cpu_num].arg = arg;
arm64_cpu_init[cpu_num].sp =
(void *)(Z_THREAD_STACK_BUFFER(stack) + sz);
arch_dcache_range((void *)&arm64_cpu_init[cpu_num],
sizeof(arm64_cpu_init[cpu_num]), K_CACHE_WB_INVD);
/* TODO: get mpidr from device tree, using cpu_num */
if (pm_cpu_on(cpu_num, (uint64_t)&__start))
printk("Failed to boot CPU%d\n", cpu_num);
/* Wait secondary cores up, see z_arm64_secondary_start */
while (arm64_cpu_init[cpu_num].fn) {
wfe();
}
}
/* the C entry of secondary cores */
void z_arm64_secondary_start(void)
{
arch_cpustart_t fn;
int cpu_num = MPIDR_TO_CORE(GET_MPIDR());
/* Initialize tpidrro_el0 with our struct _cpu instance address */
write_tpidrro_el0((uintptr_t)&_kernel.cpus[cpu_num]);
z_arm64_mmu_init(false);
#ifdef CONFIG_SMP
arm_gic_secondary_init();
irq_enable(SGI_SCHED_IPI);
#ifdef CONFIG_USERSPACE
irq_enable(SGI_PTABLE_IPI);
#endif
#endif
fn = arm64_cpu_init[cpu_num].fn;
/*
* Secondary core clears .fn to announce its presence.
* Primary core is polling for this.
*/
arm64_cpu_init[cpu_num].fn = NULL;
dsb();
sev();
fn(arm64_cpu_init[cpu_num].arg);
}
#ifdef CONFIG_SMP
static void broadcast_ipi(unsigned int ipi)
{
const uint64_t mpidr = GET_MPIDR();
/*
* Send SGI to all cores except itself
* Note: Assume only one Cluster now.
*/
gic_raise_sgi(ipi, mpidr, SGIR_TGT_MASK & ~(1 << MPIDR_TO_CORE(mpidr)));
}
void sched_ipi_handler(const void *unused)
{
ARG_UNUSED(unused);
z_sched_ipi();
}
/* arch implementation of sched_ipi */
void arch_sched_ipi(void)
{
broadcast_ipi(SGI_SCHED_IPI);
}
#ifdef CONFIG_USERSPACE
void ptable_ipi_handler(const void *unused)
{
ARG_UNUSED(unused);
/*
* Make sure a domain switch by another CPU is effective on this CPU.
* This is a no-op if the page table is already the right one.
*/
z_arm64_swap_ptables(_current);
}
void z_arm64_ptable_ipi(void)
{
broadcast_ipi(SGI_PTABLE_IPI);
}
#endif
static int arm64_smp_init(const struct device *dev)
{
ARG_UNUSED(dev);
/*
* SGI0 is use for sched ipi, this might be changed to use Kconfig
* option
*/
IRQ_CONNECT(SGI_SCHED_IPI, IRQ_DEFAULT_PRIORITY, sched_ipi_handler, NULL, 0);
irq_enable(SGI_SCHED_IPI);
#ifdef CONFIG_USERSPACE
IRQ_CONNECT(SGI_PTABLE_IPI, IRQ_DEFAULT_PRIORITY, ptable_ipi_handler, NULL, 0);
irq_enable(SGI_PTABLE_IPI);
#endif
return 0;
}
SYS_INIT(arm64_smp_init, PRE_KERNEL_1, CONFIG_KERNEL_INIT_PRIORITY_DEFAULT);
#endif