// SPDX-License-Identifier: GPL-2.0-or-later
// Copyright (C) ASPEED Technology Inc.
// Copyright IBM Corp.

#include <linux/of_address.h>
#include <linux/io.h>
#include <linux/of.h>
#include <linux/smp.h>

#define BOOT_ADDR       0x00
#define BOOT_SIG        0x04

static struct device_node *secboot_node;

static int aspeed_g6_boot_secondary(unsigned int cpu, struct task_struct *idle)
{
        void __iomem *base;

        base = of_iomap(secboot_node, 0);
        if (!base) {
                pr_err("could not map the secondary boot base!");
                return -ENODEV;
        }

        writel_relaxed(0, base + BOOT_ADDR);
        writel_relaxed(__pa_symbol(secondary_startup_arm), base + BOOT_ADDR);
        writel_relaxed((0xABBAAB00 | (cpu & 0xff)), base + BOOT_SIG);

        dsb_sev();

        iounmap(base);

        return 0;
}

static void __init aspeed_g6_smp_prepare_cpus(unsigned int max_cpus)
{
        void __iomem *base;

        secboot_node = of_find_compatible_node(NULL, NULL, "aspeed,ast2600-smpmem");
        if (!secboot_node) {
                pr_err("secboot device node found!!\n");
                return;
        }

        base = of_iomap(secboot_node, 0);
        if (!base) {
                pr_err("could not map the secondary boot base!");
                return;
        }
        __raw_writel(0xBADABABA, base + BOOT_SIG);

        iounmap(base);
}

static const struct smp_operations aspeed_smp_ops __initconst = {
        .smp_prepare_cpus       = aspeed_g6_smp_prepare_cpus,
        .smp_boot_secondary     = aspeed_g6_boot_secondary,
};

CPU_METHOD_OF_DECLARE(aspeed_smp, "aspeed,ast2600-smp", &aspeed_smp_ops);