/*	$NetBSD: boot.c,v 1.31 2020/01/14 03:48:10 isaki Exp $	*/

/*
 * Copyright (c) 2001 Minoura Makoto
 * 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.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR 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 AUTHOR 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.
 */

#include <sys/param.h>
#include <machine/bootinfo.h>

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

#include "libx68k.h"
#include "iocs.h"
#include "switch.h"

#include "exec_image.h"


#define HEAP_START	((void*) 0x00080000)
#define HEAP_END	((void*) 0x000fffff)
#define EXSCSI_BDID	((void*) 0x00ea0001)
#define SRAM_MEMSIZE	(*((long*) 0x00ed0008))

char default_kernel[20] =
#if defined(NETBOOT)
    "nfs:netbsd";
#else
    "sd0a:netbsd";
#endif
int mpu;
int hostadaptor;
int console_device = -1;

#ifdef DEBUG
int debug = 1;
#endif

static void help(void);
static int get_scsi_host_adapter(void);
static void doboot(const char *, int);
static void boot(char *);
static void cmd_ls(char *);
int bootmenu(void);
void bootmain(int);
extern int detectmpu(void);
extern int badbaddr(void *);

extern struct fs_ops file_system_ustarfs[];
extern struct fs_ops file_system_nfs[];

/* from boot_ufs/bootmain.c */
static int
get_scsi_host_adapter(void)
{
	uint32_t bootinf;
	char *bootrom;
	int ha;

	bootinf = IOCS_BOOTINF();
	if (bootinf < 0xa0) {
		/* boot from FD */
		return 0;
	}

	/* Or, bootinf indicates the boot address */
	bootrom = (char *)(bootinf & 0x00ffffe0);
	/*
	 * bootrom+0x24	"SCSIIN" ... Internal SCSI (spc0@)
	 *		"SCSIEX" ... External SCSI (spc1@ or mha0@)
	 */
	if (*(u_short *)(bootrom + 0x24 + 4) == 0x494e) {	/* "IN" */
		ha = (X68K_BOOT_SCSIIF_SPC << 4) | 0;
	} else if (badbaddr(EXSCSI_BDID)) {
		ha = (X68K_BOOT_SCSIIF_MHA << 4) | 0;
	} else {
		ha = (X68K_BOOT_SCSIIF_SPC << 4) | 1;
	}

	return ha;
}

static void
help(void)
{
	printf("Usage:\n");
	printf("boot [ha@][dev:][file] -[flags]\n");
	printf(" ha:    spc0, spc1, mha0\n");
	printf(" dev:   sd<ID><PART>, ID=0-7, PART=a-p\n");
	printf("        cd<ID>a, ID=0-7\n");
	printf("        fd<UNIT>a, UNIT=0-3, format is detected.\n");
	printf("        nfs, first probed NE2000 is used.\n");
	printf(" file:  netbsd, netbsd.gz, etc.\n");
	printf(" flags: abdqsv\n");
	printf("ls [dev:][directory]\n");
	printf("switch [show | key=val]\n");
	printf("halt\nreboot\n");
}

static void
doboot(const char *file, int flags)
{
	u_long		marks[MARK_MAX];
	int fd;
	int ha;		/* host adaptor */
	int dev;	/* device number in devspec[] */
	int unit;
	int part;
	int bootdev;
	int maj;
	char *name;
	short *p;
	int loadflag;
	struct fs_ops *fs;

	printf("Starting %s, flags 0x%x\n", file, flags);

	if (devparse(file, &ha, &dev, &unit, &part, &name) != 0) {
		printf("XXX: unknown corruption in /boot.\n");
	}

#ifdef DEBUG
	if (file[0] == 'n') {
		printf("dev = %x, unit = %d, name = %s\n",
		       dev, unit, name);
	} else {
		printf("ha = 0x%x, dev = %x, unit = %d, part = %c, name = %s\n",
		       ha, dev, unit, part + 'a', name);
	}
#endif

	if (dev == 3) {		/* netboot */
		bootdev = X68K_MAKEBOOTDEV(X68K_MAJOR_NE, unit, 0);
	} else if (dev == 2) {		/* FD */
		bootdev = X68K_MAKEBOOTDEV(X68K_MAJOR_FD, unit & 3, 0);
	} else {		/* SCSI */
		if (ha != 0) {
			hostadaptor = ha;
		}
		if (hostadaptor == 0) {
			printf("host adaptor must be specified.\n");
			return;
		}

		maj = (dev == 0) ? X68K_MAJOR_SD : X68K_MAJOR_CD;
		bootdev = X68K_MAKESCSIBOOTDEV(maj,
		    hostadaptor >> 4,
		    hostadaptor & 15,
		    unit & 7, 0, 0);
	}
#ifdef DEBUG
	printf("boot device = %x\n", bootdev);
	if (file[0] == 'n') {
		printf("type = %x, if = %d, unit = %d\n",
		       B_TYPE(bootdev),
		       B_X68K_SCSI_IF(bootdev),
		       B_X68K_SCSI_IF_UN(bootdev));
	} else {
		printf("type = %x, if = %d, unit = %d, id = %d, lun = %d, part = %c\n",
		       B_TYPE(bootdev),
		       B_X68K_SCSI_IF(bootdev),
		       B_X68K_SCSI_IF_UN(bootdev),
		       B_X68K_SCSI_ID(bootdev),
		       B_X68K_SCSI_LUN(bootdev),
		       B_X68K_SCSI_PART(bootdev) + 'a');
	}
#endif

	/*
	 * Choose the last entry of file_system[] at runtime.
	 *
	 * file_system[] is checked in turn from the beginning at all cases.
	 * Trying FS_OPS(ustarfs) for non-ustarfs displays "@" (as the
	 * mark which read a cylinder?).  OTOH, trying FS_OPS(nfs) for
	 * non-nfs displays "must mount first" error message.
	 * It is better that neither is displayed and in other words you
	 * should not put these two into file_system[] at the same time.
	 * Therefore I choose one of these here.
	 */
	if (file[0] == 'n') {
		fs = &file_system_nfs[0];
	} else {
		fs = &file_system_ustarfs[0];
	}
	memcpy(&file_system[nfsys - 1], fs, sizeof(*fs));

	loadflag = LOAD_KERNEL;
	if (file[0] == 'f')
		loadflag &= ~LOAD_BACKWARDS;

	marks[MARK_START] = 0x100000;
	if ((fd = loadfile(file, marks, loadflag)) == -1) {
		printf("loadfile failed\n");
		return;
	}
	close(fd);

	p = ((short*) marks[MARK_ENTRY]) - 1;
#ifdef DEBUG
	printf("Kernel Version: 0x%x\n", *p);
#endif
	if (*p != 0x4e73 && *p != 0) {
		/*
		 * XXX temporary solution; compatibility loader
		 * must be written.
		 */
		printf("This kernel is too new to be loaded by "
		       "this version of /boot.\n");
		return;
	}

	exec_image(marks[MARK_START], 0, marks[MARK_ENTRY]-marks[MARK_START],
		   marks[MARK_END]-marks[MARK_START], bootdev, flags);

	return;
}

static void
boot(char *arg)
{
	char filename[80];
	char *p;
	int flags = 0;

	if (*arg == 0 || *arg == '-') {
		strcpy(filename, default_kernel);
		if (*arg == '-')
			if (parseopts(arg, &flags) == 0) {
				help();
				return;
			}
		doboot(filename, flags);
		return;
	} else {
		p = gettrailer(arg);
		if (strchr(arg, ':')) {
			strcpy(filename, arg);
			if (arg[strlen(arg) - 1] == ':')
				strcat(filename, "netbsd");
		} else {
			strcpy(filename, default_kernel);
			strcpy(strchr(filename, ':') + 1, arg);
		}
		if (*p == '-') {
			if (parseopts(p, &flags) == 0)
				return;
		} else if (*p != 0) {
			help();
			return;
		}

		doboot(filename, flags);
		return;
	}
}

static void
cmd_ls(char *arg)
{
	char filename[80];

	devopen_open_dir = 1;
	if (*arg == 0) {
		strcpy(filename, default_kernel);
		strcpy(strchr(filename, ':')+1, "/");
	} else if (strchr(arg, ':') == 0) {
		strcpy(filename, default_kernel);
		strcpy(strchr(filename, ':')+1, arg);
	} else {
		strcpy(filename, arg);
		if (*(strchr(arg, ':')+1) == 0)
			strcat(filename, "/");
	}
	ls(filename);
	devopen_open_dir = 0;
}

int
bootmenu(void)
{
	char input[80];
	int n = 5, c;

	printf("Press return to boot now, any other key for boot menu\n");
	printf("booting %s - starting in %d seconds. ",
		default_kernel, n);
	while (n-- > 0 && (c = awaitkey_1sec()) == 0) {
		printf("\r");
		printf("booting %s - starting in %d seconds. ",
		       default_kernel, n);
	}
	printf("\r");
	printf("booting %s - starting in %d seconds. ", default_kernel, 0);
	printf("\n");
	
	if (c == 0 || c == '\r') {
		doboot(default_kernel, 0);
		printf("Could not start %s; ", default_kernel);
		strcat(default_kernel, ".gz");
		printf("trying %s.\n", default_kernel);
		doboot(default_kernel, 0);
		printf("Could not start %s; ", default_kernel);
	}

	printf("Please use the absolute unit# (e.g. SCSI ID)"
	       " instead of the NetBSD logical #.\n");
	for (;;) {
		char *p, *options;

		printf("> ");
		kgets(input, sizeof(input));

		for (p = &input[0]; p - &input[0] < 80 && *p == ' '; p++)
			;
		options = gettrailer(p);
		if (strcmp("boot", p) == 0)
			boot(options);
		else if (strcmp("help", p) == 0 ||
			 strcmp("?", p) == 0)
			help();
		else if (strcmp("halt", p) == 0 ||
			 strcmp("reboot", p) == 0)
			exit(0);
		else if (strcmp("switch", p) == 0)
			cmd_switch(options);
		else if (strcmp("ls", p) == 0)
			cmd_ls(options);
		else
			printf("Unknown command %s\n", p);
	}
}

static u_int
checkmemsize(void)
{
	u_int m;

#define MIN_MB 4
#define MAX_MB 12

	for (m = MIN_MB; m <= MAX_MB; m++) {
		if (badbaddr((void *)(m * 1024 * 1024 - 1))) {
			/* no memory */
			break;
		}
	}

	return (m - 1) * 1024 * 1024;
}

extern const char bootprog_rev[];
extern const char bootprog_name[];

/*
 * Arguments from the boot block:
 *   bootdev - specifies the device from which /boot was read, in 
 *		bootdev format.
 */
void
bootmain(int bootdev)
{
	u_int sram_memsize;
	u_int probed_memsize;

	console_device = consio_init(console_device);

	hostadaptor = get_scsi_host_adapter();
	rtc_offset = RTC_OFFSET;
	try_bootp = 1;
	mpu = detectmpu();

	if (mpu < 3) {		/* not tested on 68020 */
		printf("This MPU cannot run NetBSD.\n");
		exit(1);
	}
	sram_memsize = SRAM_MEMSIZE;
	if (sram_memsize < 4*1024*1024) {
		printf("Main memory too small.\n");
		exit(1);
	}

	setheap(HEAP_START, HEAP_END);

#if !defined(NETBOOT)
	switch (B_TYPE(bootdev)) {
	case X68K_MAJOR_FD:
		default_kernel[0] = 'f';
		default_kernel[2] = '0' + B_UNIT(bootdev);
		default_kernel[3] = 'a';
		break;
	case X68K_MAJOR_SD:
		default_kernel[2] = '0' + B_X68K_SCSI_ID(bootdev);
		default_kernel[3] =
			'a' + sd_getbsdpartition(B_X68K_SCSI_ID(bootdev),
						 B_X68K_SCSI_PART(bootdev));
		break;
	case X68K_MAJOR_CD:
		default_kernel[0] = 'c';
		default_kernel[2] = '0' + B_X68K_SCSI_ID(bootdev);
		default_kernel[3] = 'a';
		break;
	default:
		printf("Warning: unknown boot device: %x\n", bootdev);
	}
#endif
	print_title("%s, Revision %s\n", bootprog_name, bootprog_rev);

	/* check actual memory size for machines with a dead SRAM battery */
	probed_memsize = checkmemsize();
	if (sram_memsize != probed_memsize) {
		printf("\x1b[1mWarning: SRAM Memory Size (%d MB) "
		    "is different from probed Memory Size (%d MB)\n"
		    "         Check and reset SRAM values.\x1b[m\n\n",
		    sram_memsize / (1024 * 1024),
		    probed_memsize / (1024 * 1024));
	}

	bootmenu();
}