// SPDX-License-Identifier: GPL-2.0-or-later

/*
 * SMP support functions for Microwatt
 * Copyright 2025 Paul Mackerras <paulus@ozlabs.org>
 */

#include <linux/kernel.h>
#include <linux/smp.h>
#include <linux/io.h>
#include <asm/early_ioremap.h>
#include <asm/ppc-opcode.h>
#include <asm/reg.h>
#include <asm/smp.h>
#include <asm/xics.h>

#include "microwatt.h"

static void __init microwatt_smp_probe(void)
{
	xics_smp_probe();
}

static void microwatt_smp_setup_cpu(int cpu)
{
	if (cpu != 0)
		xics_setup_cpu();
}

static struct smp_ops_t microwatt_smp_ops = {
	.probe		= microwatt_smp_probe,
	.message_pass	= NULL,		/* Use smp_muxed_ipi_message_pass */
	.kick_cpu	= smp_generic_kick_cpu,
	.setup_cpu	= microwatt_smp_setup_cpu,
};

/* XXX get from device tree */
#define SYSCON_BASE	0xc0000000
#define SYSCON_LENGTH	0x100

#define SYSCON_CPU_CTRL	0x58

void __init microwatt_init_smp(void)
{
	volatile unsigned char __iomem *syscon;
	int ncpus;
	int timeout;

	syscon = early_ioremap(SYSCON_BASE, SYSCON_LENGTH);
	if (syscon == NULL) {
		pr_err("Failed to map SYSCON\n");
		return;
	}
	ncpus = (readl(syscon + SYSCON_CPU_CTRL) >> 8) & 0xff;
	if (ncpus < 2)
		goto out;

	smp_ops = &microwatt_smp_ops;

	/*
	 * Write two instructions at location 0:
	 * mfspr r3, PIR
	 * b __secondary_hold
	 */
	*(unsigned int *)KERNELBASE = PPC_RAW_MFSPR(3, SPRN_PIR);
	*(unsigned int *)(KERNELBASE+4) = PPC_RAW_BRANCH(&__secondary_hold - (char *)(KERNELBASE+4));

	/* enable the other CPUs, they start at location 0 */
	writel((1ul << ncpus) - 1, syscon + SYSCON_CPU_CTRL);

	timeout = 10000;
	while (!__secondary_hold_acknowledge) {
		if (--timeout == 0)
			break;
		barrier();
	}

 out:
	early_iounmap((void *)syscon, SYSCON_LENGTH);
}