// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * arch/powerpc/boot/wii.c
 *
 * Nintendo Wii bootwrapper support
 * Copyright (C) 2008-2009 The GameCube Linux Team
 * Copyright (C) 2008,2009 Albert Herranz
 */

#include <stddef.h>
#include "stdio.h"
#include "types.h"
#include "io.h"
#include "ops.h"

#include "ugecon.h"

BSS_STACK(8192);

#define HW_REG(x)               ((void *)(x))

#define EXI_CTRL                HW_REG(0x0d800070)
#define EXI_CTRL_ENABLE         (1<<0)

#define MEM2_TOP                (0x10000000 + 64*1024*1024)
#define FIRMWARE_DEFAULT_SIZE   (12*1024*1024)


struct mipc_infohdr {
        char magic[3];
        u8 version;
        u32 mem2_boundary;
        u32 ipc_in;
        size_t ipc_in_size;
        u32 ipc_out;
        size_t ipc_out_size;
};

static int mipc_check_address(u32 pa)
{
        /* only MEM2 addresses */
        if (pa < 0x10000000 || pa > 0x14000000)
                return -EINVAL;
        return 0;
}

static struct mipc_infohdr *mipc_get_infohdr(void)
{
        struct mipc_infohdr **hdrp, *hdr;

        /* 'mini' header pointer is the last word of MEM2 memory */
        hdrp = (struct mipc_infohdr **)0x13fffffc;
        if (mipc_check_address((u32)hdrp)) {
                printf("mini: invalid hdrp %08X\n", (u32)hdrp);
                hdr = NULL;
                goto out;
        }

        hdr = *hdrp;
        if (mipc_check_address((u32)hdr)) {
                printf("mini: invalid hdr %08X\n", (u32)hdr);
                hdr = NULL;
                goto out;
        }
        if (memcmp(hdr->magic, "IPC", 3)) {
                printf("mini: invalid magic\n");
                hdr = NULL;
                goto out;
        }

out:
        return hdr;
}

static int mipc_get_mem2_boundary(u32 *mem2_boundary)
{
        struct mipc_infohdr *hdr;
        int error;

        hdr = mipc_get_infohdr();
        if (!hdr) {
                error = -1;
                goto out;
        }

        if (mipc_check_address(hdr->mem2_boundary)) {
                printf("mini: invalid mem2_boundary %08X\n",
                       hdr->mem2_boundary);
                error = -EINVAL;
                goto out;
        }
        *mem2_boundary = hdr->mem2_boundary;
        error = 0;
out:
        return error;

}

static void platform_fixups(void)
{
        void *mem;
        u32 reg[4];
        u32 mem2_boundary;
        int len;
        int error;

        mem = finddevice("/memory");
        if (!mem)
                fatal("Can't find memory node\n");

        /* two ranges of (address, size) words */
        len = getprop(mem, "reg", reg, sizeof(reg));
        if (len != sizeof(reg)) {
                /* nothing to do */
                goto out;
        }

        /* retrieve MEM2 boundary from 'mini' */
        error = mipc_get_mem2_boundary(&mem2_boundary);
        if (error) {
                /* if that fails use a sane value */
                mem2_boundary = MEM2_TOP - FIRMWARE_DEFAULT_SIZE;
        }

        if (mem2_boundary > reg[2] && mem2_boundary < reg[2] + reg[3]) {
                reg[3] = mem2_boundary - reg[2];
                printf("top of MEM2 @ %08X\n", reg[2] + reg[3]);
                setprop(mem, "reg", reg, sizeof(reg));
        }

out:
        return;
}

void platform_init(unsigned long r3, unsigned long r4, unsigned long r5)
{
        u32 heapsize = 24*1024*1024 - (u32)_end;

        simple_alloc_init(_end, heapsize, 32, 64);
        fdt_init(_dtb_start);

        /*
         * 'mini' boots the Broadway processor with EXI disabled.
         * We need it enabled before probing for the USB Gecko.
         */
        out_be32(EXI_CTRL, in_be32(EXI_CTRL) | EXI_CTRL_ENABLE);

        if (ug_probe())
                console_ops.write = ug_console_write;

        platform_ops.fixups = platform_fixups;
}