/*      $NetBSD: boot.c,v 1.20 2011/01/22 19:19:16 joerg Exp $  */

/*-
 * Copyright (c) 2003 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Jonathan Stone, Michael Hitch, Simon Burge and Wayne Knowles.
 *
 * 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) 1992, 1993
 *      The Regents of the University of California.  All rights reserved.
 *
 * This code is derived from software contributed to Berkeley by
 * Ralph Campbell.
 *
 * 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. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS 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 REGENTS 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.
 *
 *      @(#)boot.c      8.1 (Berkeley) 6/10/93
 */

#include <lib/libsa/stand.h>
#include <lib/libsa/loadfile.h>
#include <lib/libsa/dev_net.h>
#include <lib/libkern/libkern.h>

#include <sys/param.h>
#include <sys/boot_flag.h>
#include <sys/exec.h>
#include <sys/exec_elf.h>

#include <machine/cpu.h>

#include <cobalt/dev/gtreg.h>

#include "boot.h"
#include "cons.h"
#include "common.h"
#include "bootinfo.h"

char *kernelnames[] = {
        "netbsd",
        "netbsd.gz",
        "onetbsd",
        "onetbsd.gz",
        "netbsd.bak",
        "netbsd.bak.gz",
        "netbsd.old",
        "netbsd.old.gz",
        "netbsd.cobalt",
        "netbsd.cobalt.gz",
        "netbsd.elf",
        "netbsd.elf.gz",
        NULL
};

u_int cobalt_id;
static const char * const cobalt_model[] =
{
        [0]                  = "Unknown Cobalt",
        [COBALT_ID_QUBE2700] = "Cobalt Qube 2700",
        [COBALT_ID_RAQ]      = "Cobalt RaQ",
        [COBALT_ID_QUBE2]    = "Cobalt Qube 2",
        [COBALT_ID_RAQ2]     = "Cobalt RaQ 2"
};
#define COBALT_MODELS   __arraycount(cobalt_model)

extern u_long end;              /* Boot loader code end address */
void start(void);

static char *bootstring;

static int patch_bootstring(char *bootspec);
static int get_bsdbootname(char **, char **, int *);
static int parse_bootname(char *, int, char **, char **);
static void prominit(unsigned int memsize);
static void print_banner(unsigned int memsize);
static u_int read_board_id(void);

void cpu_reboot(void);

int main(unsigned int memsize);

/*
 * Perform CPU reboot.
 */
void
cpu_reboot(void)
{

        printf("rebooting...\n\n");

        *(volatile uint8_t *)MIPS_PHYS_TO_KSEG1(LED_ADDR) = LED_RESET;
        printf("WARNING: reboot failed!\n");

        for (;;)
                ;
}

/*
 * Substitute root value with NetBSD root partition name.
 */
int
patch_bootstring(char *bootspec)
{
        char *sp = bootstring;
        uint8_t unit, part;
        int dev;
        char *file;

        DPRINTF(("patch_bootstring: %s\n", bootspec));

        /* get boot parameters */
        if (devparse(bootspec, &dev, &unit, &part, (const char **)&file) != 0)
                unit = part = 0;

        DPRINTF(("patch_bootstring: unit = %d, part = %d\n", unit, part));

        /* take out the 'root=xxx' parameter */
        if ((sp = strstr(bootstring, "root=")) != NULL) {
                const char *end;
                
                end = strchr(sp, ' ');

                /* strip off leading spaces */
                for (--sp; (sp > bootstring) && (*sp == ' '); --sp)
                        ;

                if (end != NULL)
                        strcpy(++sp, end);
                else
                        *++sp = '\0';
        }

        DPRINTF(("patch_bootstring: [%s]\n", bootstring));

#define DEVNAMESIZE     (MAXDEVNAME + sizeof(" root=/dev/hd") + sizeof("0a"))
        if (strcmp(devsw[dev].dv_name, "wd") == 0 &&
            strlen(bootstring) <= (511 - DEVNAMESIZE)) {
                int len;

                /* omit "nfsroot=" arg on wd boot */
                if ((sp = strstr(bootstring, "nfsroot=")) != NULL) {
                        const char *end;

                        end = strchr(sp, ' ');

                        /* strip off leading spaces */
                        for (--sp; (sp > bootstring) && (*sp == ' '); --sp)
                                ;

                        if (end != NULL)
                                strcpy(++sp, end);
                        else
                                *++sp = '\0';
                }

                /* bsd notation -> linux notation (wd0a -> hda1) */
                strcat(bootstring, " root=/dev/hd");

                len = strlen(bootstring);
                bootstring[len++] = unit + 'a';
                bootstring[len++] = part + '1';
                bootstring[len++] = '\0';
        }

        DPRINTF(("patch_bootstring: -> %s\n", bootstring));
        return 0;
}

/*
 * Extract NetBSD boot specification
 */
static int
get_bsdbootname(char **dev, char **kname, int *howtop)
{
        int len;
        int bootunit, bootpart;
        char *bootstr_dev, *bootstr_kname;
        char *prompt_dev, *prompt_kname;
        char *ptr, *spec;
        char c, namebuf[PATH_MAX];
        static char bootdev[] = "wd0a";
        static char nfsbootdev[] = "nfs";

        bootstr_dev = prompt_dev = NULL;
        bootstr_kname = prompt_kname = NULL;

        /* first, get root device specified by the firmware */
        spec = bootstring;

        /* assume the last one is valid */
        while ((spec = strstr(spec, "root=")) != NULL) {
                spec += 5;      /* skip 'root=' */
                ptr = strchr(spec, ' ');
                len = (ptr == NULL) ? strlen(spec) : ptr - spec;
                /* decode unit and part from "/dev/hd[ab][1-4]" strings */
                if (len == 9 && memcmp("/dev/hd", spec, 7) == 0) {
                        bootunit = spec[7] - 'a';
                        bootpart = spec[8] - '1';
                        if (bootunit >= 0 && bootunit < 2 &&
                            bootpart >= 0 && bootpart < 4) {
                                bootdev[sizeof(bootdev) - 3] = '0' + bootunit;
#if 0                           /* bootpart is fdisk partition of Linux root */
                                bootdev[sizeof(bootdev) - 2] = 'a' + bootpart;
#endif
                                bootstr_dev = bootdev;
                        }
                }
                spec += len;
        }

        /* second, get bootname from bootstrings */
        if ((spec = strstr(bootstring, "nbsd=")) != NULL) {
                ptr = strchr(spec, ' ');
                spec += 5;      /* skip 'nbsd=' */
                len = (ptr == NULL) ? strlen(spec) : ptr - spec;
                if (len > 0) {
                        if (parse_bootname(spec, len,
                            &bootstr_dev, &bootstr_kname))
                                return 1;
                }
        }

        /* third, check if netboot */
        if (strstr(bootstring, "nfsroot=") != NULL) {
                bootstr_dev = nfsbootdev;
        }

        DPRINTF(("bootstr_dev = %s, bootstr_kname = %s\n",
            bootstr_dev ? bootstr_dev : "<NULL>",
            bootstr_kname ? bootstr_kname : "<NULL>"));

        spec = NULL;
        len = 0;

        memset(namebuf, 0, sizeof namebuf);
        printf("Boot [%s:%s]: ",
            bootstr_dev ? bootstr_dev : DEFBOOTDEV,
            bootstr_kname ? bootstr_kname : DEFKERNELNAME);

        if (tgets(namebuf) == -1)
                printf("\n");

        ptr = namebuf;
        while ((c = *ptr) != '\0') {
                while (c == ' ')
                        c = *++ptr;
                if (c == '\0')
                        break;
                if (c == '-') {
                        while ((c = *++ptr) && c != ' ')
                                BOOT_FLAG(c, *howtop);
                } else {
                        spec = ptr;
                        while ((c = *++ptr) && c != ' ')
                                ;
                        if (c)
                                *ptr++ = '\0';
                        len = strlen(spec);
                }
        }

        if (len > 0) {
                if (parse_bootname(spec, len, &prompt_dev, &prompt_kname))
                        return 1;
        }

        DPRINTF(("prompt_dev = %s, prompt_kname = %s\n",
            prompt_dev ? prompt_dev : "<NULL>",
            prompt_kname ? prompt_kname : "<NULL>"));

        if (prompt_dev)
                *dev = prompt_dev;
        else
                *dev = bootstr_dev;

        if (prompt_kname)
                *kname = prompt_kname;
        else
                *kname = bootstr_kname;

        DPRINTF(("dev = %s, kname = %s\n",
            *dev ? *dev : "<NULL>",
            *kname ? *kname : "<NULL>"));

        return 0;
}

static int
parse_bootname(char *spec, int len, char **dev, char **kname)
{
        char *bootname, *ptr;

        bootname = alloc(len + 1);
        if (bootname == NULL)
                return 1;
        memcpy(bootname, spec, len);
        bootname[len] = '\0';

        if ((ptr = memchr(bootname, ':', len)) != NULL) {
                /* "wdXX:kernel" */
                *ptr = '\0';
                *dev = bootname;
                if (*++ptr)
                        *kname = ptr;
        } else
                /* "kernel" */
                *kname = bootname;
        return 0;
}

/*
 * Get the bootstring from PROM.
 */
void
prominit(unsigned int memsize)
{

        bootstring = (char *)(memsize - 512);
        bootstring[511] = '\0';
}

/*
 * Print boot message.
 */
void
print_banner(unsigned int memsize)
{

        lcd_banner();

        printf("\n");
        printf(">> %s " NETBSD_VERS " Bootloader, Revision %s [@%p]\n",
                        bootprog_name, bootprog_rev, (void*)&start);
        printf(">> Model:\t\t%s\n", cobalt_model[cobalt_id]);
        printf(">> Memory:\t\t%lu k\n", (memsize - MIPS_KSEG0_START) / 1024);
        printf(">> PROM boot string:\t%s\n", bootstring);
}

u_int
read_board_id(void)
{
        uint32_t tag, reg;

#define PCIB_PCI_BUS            0
#define PCIB_PCI_DEV            9
#define PCIB_PCI_FUNC           0
#define PCIB_BOARD_ID_REG       0x94
#define COBALT_BOARD_ID(reg)    ((reg & 0x000000f0) >> 4)

        tag = (PCIB_PCI_BUS << 16) | (PCIB_PCI_DEV << 11) |
            (PCIB_PCI_FUNC << 8);
        reg = pcicfgread(tag, PCIB_BOARD_ID_REG);

        return COBALT_BOARD_ID(reg);
}

/*
 * Entry point.
 * Parse PROM boot string, load the kernel and jump into it
 */
int
main(unsigned int memsize)
{
        char **namep, *dev, *kernel, *bi_addr;
        char bootpath[PATH_MAX];
        int win;
        u_long marks[MARK_MAX];
        void (*entry)(unsigned int, u_int, char *);

        struct btinfo_flags bi_flags;
        struct btinfo_symtab bi_syms;
        struct btinfo_bootpath bi_bpath;
        struct btinfo_howto bi_howto;
        int addr, speed, howto;

        try_bootp = 1;

        /* Initialize boot info early */
        dev = NULL;
        kernel = NULL;
        howto = 0x0;
        bi_flags.bi_flags = 0x0;
        bi_addr = bi_init();

        lcd_init();
        cobalt_id = read_board_id();
        prominit(memsize);
        if (cninit(&addr, &speed) != NULL)
                bi_flags.bi_flags |= BI_SERIAL_CONSOLE;

        print_banner(memsize);

        memset(marks, 0, sizeof marks);
        get_bsdbootname(&dev, &kernel, &howto);

        if (kernel != NULL) {
                DPRINTF(("kernel: %s\n", kernel));
                kernelnames[0] = kernel;
                kernelnames[1] = NULL;
        } else {
                DPRINTF(("kernel: NULL\n"));
        }

        win = 0;
        DPRINTF(("Kernel names: %p\n", kernelnames));
        for (namep = kernelnames, win = 0; (*namep != NULL) && !win; namep++) {
                kernel = *namep;

                bootpath[0] = '\0';

                strcpy(bootpath, dev ? dev : DEFBOOTDEV);
                strcat(bootpath, ":");
                strcat(bootpath, kernel);

                lcd_loadfile(bootpath);
                printf("Loading: %s", bootpath);
                if (howto)
                        printf(" (howto 0x%x)", howto);
                printf("\n");
                patch_bootstring(bootpath);
                win = (loadfile(bootpath, marks, LOAD_ALL) != -1);
        }

        if (win) {
                strncpy(bi_bpath.bootpath, kernel, BTINFO_BOOTPATH_LEN);
                bi_add(&bi_bpath, BTINFO_BOOTPATH, sizeof(bi_bpath));

                entry = (void *)marks[MARK_ENTRY];
                bi_syms.nsym = marks[MARK_NSYM];
                bi_syms.ssym = marks[MARK_SYM];
                bi_syms.esym = marks[MARK_END];
                bi_add(&bi_syms, BTINFO_SYMTAB, sizeof(bi_syms));

                bi_add(&bi_flags, BTINFO_FLAGS, sizeof(bi_flags));

                bi_howto.bi_howto = howto;
                bi_add(&bi_howto, BTINFO_HOWTO, sizeof(bi_howto));

                entry = (void *)marks[MARK_ENTRY];

                DPRINTF(("Bootinfo @ 0x%lx\n", (u_long)bi_addr));
                printf("Starting at 0x%lx\n\n", (u_long)entry);
                (*entry)(memsize, BOOTINFO_MAGIC, bi_addr);
        }

        delay(20000);
        lcd_failed();
        (void)printf("Boot failed! Rebooting...\n");
        return 0;
}