// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Copyright (C) 2000, 2001, 2002, 2003 Broadcom Corporation
 */

#include <linux/init.h>
#include <linux/kernel.h>
#include <linux/linkage.h>
#include <linux/mm.h>
#include <linux/memblock.h>
#include <linux/pm.h>
#include <linux/smp.h>

#include <asm/bootinfo.h>
#include <asm/reboot.h>
#include <asm/setup.h>
#include <asm/sibyte/board.h>
#include <asm/smp-ops.h>

#include <asm/fw/cfe/cfe_api.h>
#include <asm/fw/cfe/cfe_error.h>

/* Max ram addressable in 32-bit segments */
#ifdef CONFIG_64BIT
#define MAX_RAM_SIZE (~0ULL)
#else
#ifdef CONFIG_HIGHMEM
#ifdef CONFIG_PHYS_ADDR_T_64BIT
#define MAX_RAM_SIZE (~0ULL)
#else
#define MAX_RAM_SIZE (0xffffffffULL)
#endif
#else
#define MAX_RAM_SIZE (0x1fffffffULL)
#endif
#endif

int cfe_cons_handle;

#ifdef CONFIG_BLK_DEV_INITRD
extern unsigned long initrd_start, initrd_end;
#endif

static void __noreturn cfe_linux_exit(void *arg)
{
        int warm = *(int *)arg;

        if (smp_processor_id()) {
                static int reboot_smp;

                /* Don't repeat the process from another CPU */
                if (!reboot_smp) {
                        /* Get CPU 0 to do the cfe_exit */
                        reboot_smp = 1;
                        smp_call_function(cfe_linux_exit, arg, 0);
                }
        } else {
                printk("Passing control back to CFE...\n");
                cfe_exit(warm, 0);
                printk("cfe_exit returned??\n");
        }
        while (1);
}

static void __noreturn cfe_linux_restart(char *command)
{
        static const int zero;

        cfe_linux_exit((void *)&zero);
}

static void __noreturn cfe_linux_halt(void)
{
        static const int one = 1;

        cfe_linux_exit((void *)&one);
}

static __init void prom_meminit(void)
{
        u64 addr, size, type; /* regardless of PHYS_ADDR_T_64BIT */
        int mem_flags = 0;
        unsigned int idx;
        int rd_flag;
#ifdef CONFIG_BLK_DEV_INITRD
        unsigned long initrd_pstart;
        unsigned long initrd_pend;

        initrd_pstart = CPHYSADDR(initrd_start);
        initrd_pend = CPHYSADDR(initrd_end);
        if (initrd_start &&
            ((initrd_pstart > MAX_RAM_SIZE)
             || (initrd_pend > MAX_RAM_SIZE))) {
                panic("initrd out of addressable memory");
        }

#endif /* INITRD */

        for (idx = 0; cfe_enummem(idx, mem_flags, &addr, &size, &type) != CFE_ERR_NOMORE;
             idx++) {
                rd_flag = 0;
                if (type == CFE_MI_AVAILABLE) {
                        /*
                         * See if this block contains (any portion of) the
                         * ramdisk
                         */
#ifdef CONFIG_BLK_DEV_INITRD
                        if (initrd_start) {
                                if ((initrd_pstart > addr) &&
                                    (initrd_pstart < (addr + size))) {
                                        memblock_add(addr,
                                                     initrd_pstart - addr);
                                        rd_flag = 1;
                                }
                                if ((initrd_pend > addr) &&
                                    (initrd_pend < (addr + size))) {
                                        memblock_add(initrd_pend,
                                                (addr + size) - initrd_pend);
                                        rd_flag = 1;
                                }
                        }
#endif
                        if (!rd_flag) {
                                if (addr > MAX_RAM_SIZE)
                                        continue;
                                if (addr+size > MAX_RAM_SIZE)
                                        size = MAX_RAM_SIZE - (addr+size) + 1;
                                /*
                                 * memcpy/__copy_user prefetch, which
                                 * will cause a bus error for
                                 * KSEG/KUSEG addrs not backed by RAM.
                                 * Hence, reserve some padding for the
                                 * prefetch distance.
                                 */
                                if (size > 512)
                                        size -= 512;
                                memblock_add(addr, size);
                        }
                }
        }
#ifdef CONFIG_BLK_DEV_INITRD
        if (initrd_start) {
                memblock_add(initrd_pstart, initrd_pend - initrd_pstart);
                memblock_reserve(initrd_pstart, initrd_pend - initrd_pstart);
        }
#endif
}

#ifdef CONFIG_BLK_DEV_INITRD
static int __init initrd_setup(char *str)
{
        char rdarg[64];
        int idx;
        char *tmp, *endptr;
        unsigned long initrd_size;

        /* Make a copy of the initrd argument so we can smash it up here */
        for (idx = 0; idx < sizeof(rdarg)-1; idx++) {
                if (!str[idx] || (str[idx] == ' ')) break;
                rdarg[idx] = str[idx];
        }

        rdarg[idx] = 0;
        str = rdarg;

        /*
         *Initrd location comes in the form "<hex size of ramdisk in bytes>@<location in memory>"
         *  e.g. initrd=3abfd@80010000.  This is set up by the loader.
         */
        for (tmp = str; *tmp != '@'; tmp++) {
                if (!*tmp) {
                        goto fail;
                }
        }
        *tmp = 0;
        tmp++;
        if (!*tmp) {
                goto fail;
        }
        initrd_size = simple_strtoul(str, &endptr, 16);
        if (*endptr) {
                *(tmp-1) = '@';
                goto fail;
        }
        *(tmp-1) = '@';
        initrd_start = simple_strtoul(tmp, &endptr, 16);
        if (*endptr) {
                goto fail;
        }
        initrd_end = initrd_start + initrd_size;
        printk("Found initrd of %lx@%lx\n", initrd_size, initrd_start);
        return 1;
 fail:
        printk("Bad initrd argument.  Disabling initrd\n");
        initrd_start = 0;
        initrd_end = 0;
        return 1;
}

#endif

extern const struct plat_smp_ops sb_smp_ops;
extern const struct plat_smp_ops bcm1480_smp_ops;

/*
 * prom_init is called just after the cpu type is determined, from setup_arch()
 */
void __init prom_init(void)
{
        uint64_t cfe_ept, cfe_handle;
        unsigned int cfe_eptseal;
        int argc = fw_arg0;
        char **envp = (char **) fw_arg2;
        int *prom_vec = (int *) fw_arg3;

        _machine_restart   = cfe_linux_restart;
        _machine_halt      = cfe_linux_halt;
        pm_power_off = cfe_linux_halt;

        /*
         * Check if a loader was used; if NOT, the 4 arguments are
         * what CFE gives us (handle, 0, EPT and EPTSEAL)
         */
        if (argc < 0) {
                cfe_handle = (uint64_t)(long)argc;
                cfe_ept = (long)envp;
                cfe_eptseal = (uint32_t)(unsigned long)prom_vec;
        } else {
                if ((int32_t)(long)prom_vec < 0) {
                        /*
                         * Old loader; all it gives us is the handle,
                         * so use the "known" entrypoint and assume
                         * the seal.
                         */
                        cfe_handle = (uint64_t)(long)prom_vec;
                        cfe_ept = (uint64_t)((int32_t)0x9fc00500);
                        cfe_eptseal = CFE_EPTSEAL;
                } else {
                        /*
                         * Newer loaders bundle the handle/ept/eptseal
                         * Note: prom_vec is in the loader's useg
                         * which is still alive in the TLB.
                         */
                        cfe_handle = (uint64_t)((int32_t *)prom_vec)[0];
                        cfe_ept = (uint64_t)((int32_t *)prom_vec)[2];
                        cfe_eptseal = (unsigned int)((uint32_t *)prom_vec)[3];
                }
        }
        if (cfe_eptseal != CFE_EPTSEAL) {
                /* too early for panic to do any good */
                printk("CFE's entrypoint seal doesn't match. Spinning.");
                while (1) ;
        }
        cfe_init(cfe_handle, cfe_ept);
        /*
         * Get the handle for (at least) prom_putchar, possibly for
         * boot console
         */
        cfe_cons_handle = cfe_getstdhandle(CFE_STDHANDLE_CONSOLE);
        if (cfe_getenv("LINUX_CMDLINE", arcs_cmdline, COMMAND_LINE_SIZE) < 0) {
                if (argc >= 0) {
                        /* The loader should have set the command line */
                        /* too early for panic to do any good */
                        printk("LINUX_CMDLINE not defined in cfe.");
                        while (1) ;
                }
        }

#ifdef CONFIG_BLK_DEV_INITRD
        {
                char *ptr;
                /* Need to find out early whether we've got an initrd.  So scan
                   the list looking now */
                for (ptr = arcs_cmdline; *ptr; ptr++) {
                        while (*ptr == ' ') {
                                ptr++;
                        }
                        if (!strncmp(ptr, "initrd=", 7)) {
                                initrd_setup(ptr+7);
                                break;
                        } else {
                                while (*ptr && (*ptr != ' ')) {
                                        ptr++;
                                }
                        }
                }
        }
#endif /* CONFIG_BLK_DEV_INITRD */

        /* Not sure this is needed, but it's the safe way. */
        arcs_cmdline[COMMAND_LINE_SIZE-1] = 0;

        prom_meminit();

#if defined(CONFIG_SIBYTE_BCM112X) || defined(CONFIG_SIBYTE_SB1250)
        register_smp_ops(&sb_smp_ops);
#endif
#ifdef CONFIG_SIBYTE_BCM1x80
        register_smp_ops(&bcm1480_smp_ops);
#endif
}

void prom_putchar(char c)
{
        int ret;

        while ((ret = cfe_write(cfe_cons_handle, &c, 1)) == 0)
                ;
}