/* $NetBSD: machdep.c,v 1.6 2026/02/24 21:41:59 jmcneill Exp $ */

/*
 * Copyright (c) 2002, 2024 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Lennart Augustsson (lennart@augustsson.net) at Sandburst Corp.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

/*
 * Copyright (C) 1995, 1996 Wolfgang Solfrank.
 * Copyright (C) 1995, 1996 TooLs GmbH.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. All advertising materials mentioning features or use of this software
 *    must display the following acknowledgement:
 *      This product includes software developed by TooLs GmbH.
 * 4. The name of TooLs GmbH may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY TOOLS GMBH ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL TOOLS GMBH BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
 * OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
 * OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
 * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#define _POWERPC_BUS_DMA_PRIVATE

#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: machdep.c,v 1.6 2026/02/24 21:41:59 jmcneill Exp $");

#include "opt_compat_netbsd.h"
#include "opt_ddb.h"
#include "opt_ddbparam.h"
#include "opt_inet.h"
#include "opt_ns.h"
#include "opt_oea.h"
#include "opt_multiprocessor.h"

#include <sys/param.h>
#include <sys/buf.h>
#include <sys/bus.h>
#include <sys/conf.h>
#include <sys/device.h>
#include <sys/exec.h>
#include <sys/extent.h>
#include <sys/intr.h>
#include <sys/kernel.h>
#include <sys/kgdb.h>
#include <sys/ksyms.h>
#include <sys/mbuf.h>
#include <sys/mount.h>
#include <sys/msgbuf.h>
#include <sys/proc.h>
#include <sys/reboot.h>
#include <sys/syscallargs.h>
#include <sys/sysctl.h>
#include <sys/syslog.h>
#include <sys/systm.h>
#include <sys/cpu.h>

#include <uvm/uvm_extern.h>

#include <machine/powerpc.h>
#include <machine/wii.h>
#include <machine/wiiu.h>
#include <arch/evbppc/nintendo/dev/gecko.h>
#include <arch/evbppc/nintendo/pic_pi.h>
#include <arch/evbppc/nintendo/ipi_latte.h>

#include <powerpc/bus_funcs.h>
#include <powerpc/db_machdep.h>
#include <powerpc/pio.h>
#include <powerpc/pmap.h>
#include <powerpc/spr.h>
#include <powerpc/trap.h>

#include <powerpc/oea/bat.h>
#include <powerpc/oea/spr.h>
#include <powerpc/oea/hid.h>
#include <powerpc/pic/picvar.h>

#include <ddb/db_extern.h>
#include <ddb/db_active.h>

#include <dev/wscons/wsconsio.h>
#include <dev/wscons/wsdisplayvar.h>
#include <dev/rasops/rasops.h>
#include <dev/wsfont/wsfont.h>
#include <dev/wscons/wsdisplay_vconsvar.h>

#include <dev/usb/ukbdvar.h>

#include "ksyms.h"
#include "ukbd.h"
#include "genfb.h"
#include "gecko.h"

#ifndef WII_DEFAULT_CMDLINE
#define WII_DEFAULT_CMDLINE "root=ld0a"
#endif

#define IBM750CL_SPR_HID4       1011

#define MINI_MEM2_START         0x13f00000      /* Start of reserved MEM2 for MINI */

#define PI_INTERRUPT_CAUSE      0x0c003000
#define  RESET_SWITCH_STATE     __BIT(16)

extern u_int l2cr_config;

struct powerpc_bus_space wii_mem_tag = {
        .pbs_flags = _BUS_SPACE_BIG_ENDIAN |
                     _BUS_SPACE_MEM_TYPE,
};

static char ex_storage[1][EXTENT_FIXED_STORAGE_SIZE(EXTMAP_RANGES)]
    __attribute__((aligned(8)));

static bus_addr_t
wii_dma_phys_to_bus_mem(bus_dma_tag_t t, bus_addr_t addr)
{
        return addr;
}

static bus_addr_t
wii_dma_bus_mem_to_phys(bus_dma_tag_t t, bus_addr_t addr)
{
        return addr;
}

static int
wii_mem2_dmamem_alloc(bus_dma_tag_t t, bus_size_t size, bus_size_t alignment,
    bus_size_t boundary, bus_dma_segment_t *segs, int nsegs, int *rsegs,
    int flags)
{
        struct mem_region *mem, *avail;
        const int mem2_index = wiiu_native ? 2 : 1;

        /* Restrict memory used for DMA to ranges in MEM2 */
        mem_regions(&mem, &avail);
        if (mem[mem2_index].size == 0) {
                return ENOMEM;
        }

        return _bus_dmamem_alloc_range(t, size, alignment, boundary, segs,      
            nsegs, rsegs, flags, mem[mem2_index].start,
            mem[mem2_index].start + mem[mem2_index].size - PAGE_SIZE - 1);
}

struct powerpc_bus_dma_tag wii_bus_dma_tag = {
        0, 0,                           /* _bounce_thresh_min/max */
        _bus_dmamap_create,
        _bus_dmamap_destroy,
        _bus_dmamap_load,
        _bus_dmamap_load_mbuf,
        _bus_dmamap_load_uio,
        _bus_dmamap_load_raw,
        _bus_dmamap_unload,
        _bus_dmamap_sync,
        _bus_dmamem_alloc,
        _bus_dmamem_free,
        _bus_dmamem_map,
        _bus_dmamem_unmap,
        _bus_dmamem_mmap,
        wii_dma_phys_to_bus_mem,
        wii_dma_bus_mem_to_phys,
};

struct powerpc_bus_dma_tag wii_mem2_bus_dma_tag = {
        WII_MEM2_BASE,                          /* _bounce_thresh_min */
        WII_MEM2_BASE + WII_MEM2_SIZE - 1,      /* _bounce_thresh_max */
        _bus_dmamap_create,
        _bus_dmamap_destroy,
        _bus_dmamap_load,
        _bus_dmamap_load_mbuf,
        _bus_dmamap_load_uio,
        _bus_dmamap_load_raw,
        _bus_dmamap_unload,
        _bus_dmamap_sync,
        wii_mem2_dmamem_alloc,
        _bus_dmamem_free,
        _bus_dmamem_map,
        _bus_dmamem_unmap,
        _bus_dmamem_mmap,
        wii_dma_phys_to_bus_mem,
        wii_dma_bus_mem_to_phys,
};


/*
 * Global variables used here and there
 */
struct mem_region physmemr[4], availmemr[4];
char wii_cmdline[1024];
bool wiiu_plat;         /* Running on a Wii U */
bool wiiu_native;       /* Native Wii U mode (not vWii) */

void initppc(u_int, u_int, u_int, void *); /* Called from locore */
void wii_dolphin_elf_loader_id(void);

static void wii_setup(void);
static void wii_poweroff(void);
static void wii_reset(void);
static void wii_halt(void);
static void init_decrementer(void);

static void
system_detect(void)
{
        uint32_t val;

        val = in32(LT_CHIPREVID);
        if (__SHIFTOUT(val, LT_CHIPREVID_MAGIC) == LT_CHIPREVID_MAGIC_CAFE) {
                wiiu_plat = true;
                wiiu_native = (in32(LT_PIMCOMPAT) & PPC_COMPAT) == 0;
        }
}

static void
wii_init_cmdline(void)
{
        extern struct wii_argv wii_argv;

        if (wii_argv.magic == WII_ARGV_MAGIC) {
                void *ptr = (void *)(uintptr_t)(wii_argv.cmdline & ~0x80000000);
                if (ptr != NULL) {
                        memcpy(wii_cmdline, ptr, wii_argv.length);
                }
        } else {
                snprintf(wii_cmdline, sizeof(wii_cmdline), WII_DEFAULT_CMDLINE);
        }
}

static void
wii_init_memmap(u_int endkernel)
{
        uint32_t mem2_start, mem2_end;

        mem2_start = in32(GLOBAL_MEM2_AVAIL_START) & ~0x80000000;
        mem2_end = in32(GLOBAL_MEM2_AVAIL_END) & ~0x80000000;
        if (mem2_start < WII_MEM2_BASE) {
                /* Must have been booted from MINI. */
                mem2_start = WII_MEM2_BASE + DSP_MEM_SIZE;
                mem2_end = MINI_MEM2_START;
        }
        /*
         * Clear GLOBAL_MEM2_AVAIL_{START,END} so we can detect the correct
         * memory size when soft resetting from IOS to MINI.
         */
        out32(GLOBAL_MEM2_AVAIL_START, 0);
        out32(GLOBAL_MEM2_AVAIL_END, 0);

        /* MEM1 24MB 1T-SRAM */
        physmemr[0].start = WII_MEM1_BASE;
        physmemr[0].size = WII_MEM1_SIZE;

        /* MEM2 64MB GDDR3 */
        physmemr[1].start = WII_MEM2_BASE;
        physmemr[1].size = WII_MEM2_SIZE;

        physmemr[2].size = 0;

        /* MEM1 available memory */
        availmemr[0].start = ((endkernel & ~0x80000000) + PGOFSET) & ~PGOFSET;
        availmemr[0].size = physmemr[0].size - availmemr[0].start;
        /* External framebuffer is at the end of MEM1 */
        availmemr[0].size -= XFB_SIZE;

        /* MEM2 available memory */
        availmemr[1].start = mem2_start;
        availmemr[1].size = mem2_end - mem2_start;

        availmemr[2].size = 0;
}

static void
wiiu_init_cmdline(void)
{
        struct wiiu_argv *argv = (struct wiiu_argv *)WIIU_LOADER_DATA_ADDR;

        if (argv->magic == WIIU_LOADER_MAGIC) {
                char *p;

                /*
                 * Split the command line string into tokens to match
                 * the format used by HBC.
                 */
                memcpy(wii_cmdline, argv->cmdline, sizeof(argv->cmdline));
                wii_cmdline[sizeof(argv->cmdline) - 1] = '\0';
                p = wii_cmdline;
                for (;;) {
                        p = strchr(p, ' ');
                        if (*p == '\0') {
                                break;
                        }
                        *p++ = '\0';
                }
                p++;
                *p = '\0';
        } else {
                snprintf(wii_cmdline, sizeof(wii_cmdline), WII_DEFAULT_CMDLINE);
        }
}

static void
wiiu_init_memmap(u_int endkernel)
{
        /* MEM1 32MB */
        physmemr[0].start  = 0x00000000;
        physmemr[0].size   = 0x2000000;

        /* MEM0 3MB */
        physmemr[1].start  = 0x08000000;
        physmemr[1].size   = 0x300000;

        /* MEM2 2GB */
        physmemr[2].start  = 0x10000000;
        physmemr[2].size   = 0x80000000;

        physmemr[3].size   = 0;

        /* MEM1 available memory */
        availmemr[0].start = (endkernel + PGOFSET) & ~PGOFSET;
        availmemr[0].size  = physmemr[0].size - availmemr[0].start;

        /*
         * MEM2 available memory. Reserved regions:
         *   USB DMA                    0x10000000 - 0x103fffff
         *   Sound DMA                  0x10400000 - 0x1041ffff
         *   Framebuffer                0x17500000 - 0x17a7ffff
         */
        availmemr[1].start = 0x10420000;
        availmemr[1].size  = 0x17500000 - availmemr[1].start;
        availmemr[2].start = 0x17a80000;
        availmemr[2].size  = physmemr[2].start + physmemr[2].size -
                             availmemr[2].start;

        availmemr[3].size  = 0;
}

void
initppc(u_int startkernel, u_int endkernel, u_int args, void *btinfo)
{
        extern uint32_t ticks_per_sec;
        extern uint32_t ticks_per_msec;
        extern unsigned char edata[], end[];
        register_t scratch, spr;

        memset(&edata, 0, end - edata); /* clear BSS */
        wii_cmdline[0] = '\0';

        system_detect();

        if (wiiu_native) {
                wiiu_init_cmdline();
                wiiu_init_memmap(endkernel);
        } else {
                wii_init_cmdline();
                wii_init_memmap(endkernel);
        }

#ifdef BOOTHOWTO
        /*
         * boothowto
         */
        boothowto = BOOTHOWTO;
#endif

        cpu_model_init();

        if (!wiiu_native) {
                spr = mfspr(IBM750CL_SPR_HID4);
                spr |= HID4_L2_CCFI;
                spr |= HID4_DBP;
                mtspr(IBM750CL_SPR_HID4, spr);
                asm volatile ("isync");
        }

        if (wiiu_native) {
                spr = mfspr(SPR_IBMESPRESSO_HID5);
                mtspr(SPR_IBMESPRESSO_HID5, spr | HID5_H5A | HID5_PIRE);

                spr = mfspr(SPR_SCR);
                mtspr(SPR_SCR, (spr & ~0x40000000) | 0x80000000);

                spr = mfspr(SPR_CAR);
                mtspr(SPR_CAR, spr | 0xfc100000);

                mtspr(SPR_BCR, 0x08000000);

                asm volatile ("isync");
        }

        /* Configure L2 cache */
        l2cr_config = L2CR_L2E;

        if (bus_space_init(&wii_mem_tag, NULL,
                           ex_storage[0], sizeof(ex_storage[0]))) {
                panic("bus_space_init failed");
        }

        /*
         * Initialize the BAT registers
         */
        if (wiiu_native) {
                oea_batinit(
                    WII_IOMEM_BASE, BAT_BL_32M,
                    0);
        } else {
                oea_batinit(
                    EFB_BASE, BAT_BL_128M,
                    0);
        }

        /*
         * Set up trap vectors
         */
        oea_init(NULL);

        /*
         * Get CPU clock
         */
        ticks_per_sec = wiiu_native ? WIIU_TIMEBASE_FREQ_HZ :
                                      WII_TIMEBASE_FREQ_HZ;
        ticks_per_msec = ticks_per_sec / 1000;
        cpu_timebase = ticks_per_sec;

        wii_setup();

        uvm_md_init();

        /*
         * Initialize pmap module.
         */
        pmap_bootstrap(startkernel, endkernel);

        /* Now enable translation (and machine checks/recoverable interrupts) */
        asm volatile ("sync; mfmsr %0; ori %0,%0,%1; mtmsr %0; isync"
                      : "=r"(scratch)
                      : "K"(PSL_IR|PSL_DR|PSL_ME|PSL_RI));

        /*
         * Setup decrementer
         */
        init_decrementer();
}

void
mem_regions(struct mem_region **mem, struct mem_region **avail)
{
        *mem = physmemr;
        *avail = availmemr;
}

/*
 * Machine dependent startup code.
 */
void
cpu_startup(void)
{
        oea_startup(NULL);

        /*
         * Now that we have VM, malloc()s are OK in bus_space.
         */
        bus_space_mallocok();

        /* Set up interrupt controller */
        pic_init();
        pi_init_intr();
        oea_install_extint(pic_ext_intr);

#ifdef MULTIPROCESSOR
        if (wiiu_native) {
                ipi_latte_init();
                oea_install_extint_vec(pic_ext_intr, EXC_IPI);
        }
#endif
}

/*
 * No early console support.
 */
void
consinit(void)
{
#if NGENFB > 0
        if (wiiu_native) {
                extern void wiiufb_consinit(void);
                wiiufb_consinit();
        }
#endif
#if NGECKO > 0
        if (!wiiu_native) {
                usbgecko_consinit();
        }
#endif
#if NUKBD > 0
        ukbd_cnattach();
#endif
}

/*
 * Halt or reboot the machine after syncing/dumping according to howto.
 */
void
cpu_reboot(int howto, char *what)
{
        static int syncdone = false;

        if (cold) {
                howto |= RB_HALT;
                goto haltsys;
        }

        boothowto = howto;

        if ((howto & RB_NOSYNC) == 0 && panicstr == NULL) {
                if (!syncdone) {
                        syncdone = true;
                        /* XXX used to force unmount as well, here */
                        vfs_sync_all(curlwp);
                }

                while (vfs_unmountall1(curlwp, false, false) ||
                       config_detach_all(boothowto) ||
                       vfs_unmount_forceone(curlwp)) {
                        /* do nothing */
                }
        } else {
                if (!db_active) {
                        suspendsched();
                }
        }

        pmf_system_shutdown(boothowto);

        /* Disable interrupts */
        splhigh();

        if ((howto & (RB_DUMP | RB_HALT)) == RB_DUMP) {
                oea_dumpsys();
        }

haltsys:
        doshutdownhooks();

#ifdef MULTIPROCESSOR
        cpu_halt_others();
        delay(100000);
#endif

        if ((howto & RB_POWERDOWN) == RB_POWERDOWN) {
                printf("power off\n\n");
                wii_poweroff();
                delay(100000);
                printf("power off failed!\n\n");
        }

        if (howto & RB_HALT) {
                printf("The operating system has halted.\n");
                wii_halt();
        }

        printf("rebooting...\n\n");
        wii_reset();
        while (1);
}

static void
wii_setup(void)
{
        /* Enable PPC access to EXI bus. */
        out32(HW_AIPPROT, in32(HW_AIPPROT) | ENAHBIOPI);

        if (!wiiu_plat) {
                /* Turn on the drive slot LED. */
                wii_slot_led(true);

                /* Enable PPC access to SHUTDOWN GPIO. */
                out32(HW_GPIO_OWNER, in32(HW_GPIO_OWNER) | __BIT(GPIO_SHUTDOWN));

                /* Enable PPC access to DI_SPIN GPIO. */
                out32(HW_GPIO_OWNER, in32(HW_GPIO_OWNER) | __BIT(GPIO_DI_SPIN));

                /* Enable DVD video support. */
                out32(HW_COMPAT, in32(HW_COMPAT) & ~DVDVIDEO);
        }
}

static void
wiiu_wood_ipc(uint32_t msg)
{
        KASSERT(wiiu_native);

        out32(HW_IPCPPCMSG, msg);
        out32(HW_IPCPPCCTRL, HW_IPCPPCCTRL_X1);
        for (;;) {
                if ((in32(HW_IPCPPCCTRL) & HW_IPCPPCCTRL_X1) == 0) {
                        break;
                }
        }
}

static void
wii_poweroff(void)
{
        if (wiiu_native) {
                wiiu_wood_ipc(CMD_POWEROFF);
        } else {
                out32(HW_GPIOB_OUT, in32(HW_GPIOB_OUT) | __BIT(GPIO_SHUTDOWN));
        }
}

static void
wii_reset(void)
{
        if (wiiu_native) {
                wiiu_wood_ipc(CMD_REBOOT);
        } else {
                out32(HW_RESETS, in32(HW_RESETS) & ~RSTBINB);
        }
}

static void
wii_halt(void)
{
        printf("Please press the %s button to reboot.\n",
            wiiu_native ? "POWER" : "RESET");
        if (wiiu_native) {
                bool pressed = false;
                while (1) {
                        if ((in32(HW_GPIOB_IN) & __BIT(WIIU_GPIO_POWER)) != 0) {
                                /* Button was pressed */
                                pressed = true;
                        } else if (pressed) {
                                /* Button was released */
                                break;
                        }
                }
        } else {
                while (1) {
                        if ((in32(PI_INTERRUPT_CAUSE) & RESET_SWITCH_STATE) == 0) {
                                break;
                        }
                }
        }
}

static void
init_decrementer(void)
{
        extern uint32_t ns_per_tick;
        extern uint32_t ticks_per_intr;
        extern uint32_t ticks_per_sec;
        int scratch, msr;

        KASSERT(ticks_per_sec != 0);

        __asm volatile ("mfmsr %0; andi. %1,%0,%2; mtmsr %1"
                        : "=r"(msr), "=r"(scratch) : "K"((u_short)~PSL_EE));
        ns_per_tick = 1000000000 / ticks_per_sec;
        ticks_per_intr = ticks_per_sec / hz;
        cpu_timebase = ticks_per_sec;

        curcpu()->ci_lasttb = mftbl();

        mtspr(SPR_DEC, ticks_per_intr);
        mtmsr(msr);
}