// SPDX-License-Identifier: GPL-2.0+
/*
 * (C) Copyright 2008 Semihalf
 *
 * (C) Copyright 2000-2006
 * Wolfgang Denk, DENX Software Engineering, wd@denx.de.
 */


#include <common.h>
#include <bootstage.h>
#include <cpu_func.h>
#include <env.h>
#include <init.h>
#include <lmb.h>
#include <log.h>
#include <watchdog.h>
#include <command.h>
#include <image.h>
#include <malloc.h>
#include <asm/global_data.h>
#include <u-boot/zlib.h>
#include <bzlib.h>
#include <asm/byteorder.h>
#include <asm/mp.h>
#include <bootm.h>
#include <vxworks.h>

#if defined(CONFIG_OF_LIBFDT)
#include <linux/libfdt.h>
#include <fdt_support.h>
#endif

#ifdef CONFIG_SYS_INIT_RAM_LOCK
#include <asm/cache.h>
#endif

DECLARE_GLOBAL_DATA_PTR;

static ulong get_sp (void);
extern void ft_fixup_num_cores(void *blob);
static void set_clocks_in_mhz (struct bd_info *kbd);

#ifndef CFG_SYS_LINUX_LOWMEM_MAX_SIZE
#define CFG_SYS_LINUX_LOWMEM_MAX_SIZE   (768*1024*1024)
#endif

static void boot_jump_linux(struct bootm_headers *images)
{
        void    (*kernel)(struct bd_info *, ulong r4, ulong r5, ulong r6,
                              ulong r7, ulong r8, ulong r9);
#ifdef CONFIG_OF_LIBFDT
        char *of_flat_tree = images->ft_addr;
#endif

        kernel = (void (*)(struct bd_info *, ulong, ulong, ulong,
                           ulong, ulong, ulong))images->ep;
        debug("## Transferring control to Linux (at address %08lx) ...\n",
              (ulong)kernel);

        bootstage_mark(BOOTSTAGE_ID_RUN_OS);

#ifdef CONFIG_BOOTSTAGE_FDT
        bootstage_fdt_add_report();
#endif
#ifdef CONFIG_BOOTSTAGE_REPORT
        bootstage_report();
#endif

#if defined(CONFIG_SYS_INIT_RAM_LOCK) && !defined(CONFIG_E500)
        unlock_ram_in_cache();
#endif

#if defined(CONFIG_OF_LIBFDT)
        if (of_flat_tree) {     /* device tree; boot new style */
                /*
                 * Linux Kernel Parameters (passing device tree):
                 *   r3: pointer to the fdt
                 *   r4: 0
                 *   r5: 0
                 *   r6: epapr magic
                 *   r7: size of IMA in bytes
                 *   r8: 0
                 *   r9: 0
                 */
                debug("   Booting using OF flat tree...\n");
                schedule();
                (*kernel) ((struct bd_info *)of_flat_tree, 0, 0, EPAPR_MAGIC,
                           env_get_bootm_mapsize(), 0, 0);
                /* does not return */
        } else
#endif
        {
                /*
                 * Linux Kernel Parameters (passing board info data):
                 *   r3: ptr to board info data
                 *   r4: initrd_start or 0 if no initrd
                 *   r5: initrd_end - unused if r4 is 0
                 *   r6: Start of command line string
                 *   r7: End   of command line string
                 *   r8: 0
                 *   r9: 0
                 */
                ulong cmd_start = images->cmdline_start;
                ulong cmd_end = images->cmdline_end;
                ulong initrd_start = images->initrd_start;
                ulong initrd_end = images->initrd_end;
                struct bd_info *kbd = images->kbd;

                debug("   Booting using board info...\n");
                schedule();
                (*kernel) (kbd, initrd_start, initrd_end,
                           cmd_start, cmd_end, 0, 0);
                /* does not return */
        }
        return;
}

void arch_lmb_reserve(struct lmb *lmb)
{
        phys_size_t bootm_size;
        ulong size, bootmap_base;

        bootmap_base = env_get_bootm_low();
        bootm_size = env_get_bootm_size();

#ifdef DEBUG
        if (((u64)bootmap_base + bootm_size) >
            (CFG_SYS_SDRAM_BASE + (u64)gd->ram_size))
                puts("WARNING: bootm_low + bootm_size exceed total memory\n");
        if ((bootmap_base + bootm_size) > get_effective_memsize())
                puts("WARNING: bootm_low + bootm_size exceed eff. memory\n");
#endif

        size = min(bootm_size, get_effective_memsize());
        size = min(size, (ulong)CFG_SYS_LINUX_LOWMEM_MAX_SIZE);

        if (size < bootm_size) {
                ulong base = bootmap_base + size;
                printf("WARNING: adjusting available memory from 0x%lx to 0x%llx\n",
                       size, (unsigned long long)bootm_size);
                lmb_reserve(lmb, base, bootm_size - size);
        }

        arch_lmb_reserve_generic(lmb, get_sp(), gd->ram_top, 4096);

#ifdef CONFIG_MP
        cpu_mp_lmb_reserve(lmb);
#endif

        return;
}

static void boot_prep_linux(struct bootm_headers *images)
{
#ifdef CONFIG_MP
        /*
         * if we are MP make sure to flush the device tree so any changes are
         * made visibile to all other cores.  In AMP boot scenarios the cores
         * might not be HW cache coherent with each other.
         */
        flush_cache((unsigned long)images->ft_addr, images->ft_len);
#endif
}

static int boot_cmdline_linux(struct bootm_headers *images)
{
        ulong of_size = images->ft_len;
        struct lmb *lmb = &images->lmb;
        ulong *cmd_start = &images->cmdline_start;
        ulong *cmd_end = &images->cmdline_end;

        int ret = 0;

        if (!of_size) {
                /* allocate space and init command line */
                ret = boot_get_cmdline (lmb, cmd_start, cmd_end);
                if (ret) {
                        puts("ERROR with allocation of cmdline\n");
                        return ret;
                }
        }

        return ret;
}

static int boot_bd_t_linux(struct bootm_headers *images)
{
        ulong of_size = images->ft_len;
        struct lmb *lmb = &images->lmb;
        struct bd_info **kbd = &images->kbd;

        int ret = 0;

        if (!of_size) {
                /* allocate space for kernel copy of board info */
                ret = boot_get_kbd (lmb, kbd);
                if (ret) {
                        puts("ERROR with allocation of kernel bd\n");
                        return ret;
                }
                set_clocks_in_mhz(*kbd);
        }

        return ret;
}

static int boot_body_linux(struct bootm_headers *images)
{
        int ret;

        /* allocate space for kernel copy of board info */
        ret = boot_bd_t_linux(images);
        if (ret)
                return ret;

        if (IS_ENABLED(CONFIG_LMB)) {
                ret = image_setup_linux(images);
                if (ret)
                        return ret;
        }

        return 0;
}

noinline int do_bootm_linux(int flag, int argc, char *const argv[],
                            struct bootm_headers *images)
{
        int     ret;

        if (flag & BOOTM_STATE_OS_CMDLINE) {
                boot_cmdline_linux(images);
                return 0;
        }

        if (flag & BOOTM_STATE_OS_BD_T) {
                boot_bd_t_linux(images);
                return 0;
        }

        if (flag & BOOTM_STATE_OS_PREP) {
                boot_prep_linux(images);
                return 0;
        }

        boot_prep_linux(images);
        ret = boot_body_linux(images);
        if (ret)
                return ret;
        boot_jump_linux(images);

        return 0;
}

static ulong get_sp (void)
{
        ulong sp;

        asm( "mr %0,1": "=r"(sp) : );
        return sp;
}

static void set_clocks_in_mhz (struct bd_info *kbd)
{
        char    *s;

        s = env_get("clocks_in_mhz");
        if (s) {
                /* convert all clock information to MHz */
                kbd->bi_intfreq /= 1000000L;
                kbd->bi_busfreq /= 1000000L;
        }
}

#if defined(CONFIG_BOOTM_VXWORKS)
void boot_prep_vxworks(struct bootm_headers *images)
{
#if defined(CONFIG_OF_LIBFDT)
        int off;
        u64 base, size;

        if (!images->ft_addr)
                return;

        base = (u64)gd->ram_base;
        size = (u64)gd->ram_size;

        off = fdt_path_offset(images->ft_addr, "/memory");
        if (off < 0)
                fdt_fixup_memory(images->ft_addr, base, size);

#if defined(CONFIG_MP)
#if defined(CONFIG_MPC85xx)
        ft_fixup_cpu(images->ft_addr, base + size);
        ft_fixup_num_cores(images->ft_addr);
#elif defined(CONFIG_MPC86xx)
        off = fdt_add_mem_rsv(images->ft_addr,
                        determine_mp_bootpg(NULL), (u64)4096);
        if (off < 0)
                printf("## WARNING %s: %s\n", __func__, fdt_strerror(off));
        ft_fixup_num_cores(images->ft_addr);
#endif
        flush_cache((unsigned long)images->ft_addr, images->ft_len);
#endif
#endif
}

void boot_jump_vxworks(struct bootm_headers *images)
{
        /* PowerPC VxWorks boot interface conforms to the ePAPR standard
         * general purpuse registers:
         *
         *      r3: Effective address of the device tree image
         *      r4: 0
         *      r5: 0
         *      r6: ePAPR magic value
         *      r7: shall be the size of the boot IMA in bytes
         *      r8: 0
         *      r9: 0
         *      TCR: WRC = 0, no watchdog timer reset will occur
         */
        schedule();

        ((void (*)(void *, ulong, ulong, ulong,
                ulong, ulong, ulong))images->ep)(images->ft_addr,
                0, 0, EPAPR_MAGIC, env_get_bootm_mapsize(), 0, 0);
}
#endif