Home | History | Annotate | Line # | Download | only in efiboot
boot.c revision 1.5.2.3
      1  1.5.2.3  pgoyette /*	$NetBSD: boot.c,v 1.5.2.3 2018/09/30 01:45:57 pgoyette Exp $	*/
      2  1.5.2.2  pgoyette 
      3  1.5.2.2  pgoyette /*-
      4  1.5.2.2  pgoyette  * Copyright (c) 2016 Kimihiro Nonaka <nonaka (at) netbsd.org>
      5  1.5.2.2  pgoyette  * Copyright (c) 2018 Jared McNeill <jmcneill (at) invisible.ca>
      6  1.5.2.2  pgoyette  * All rights reserved.
      7  1.5.2.2  pgoyette  *
      8  1.5.2.2  pgoyette  * Redistribution and use in source and binary forms, with or without
      9  1.5.2.2  pgoyette  * modification, are permitted provided that the following conditions
     10  1.5.2.2  pgoyette  * are met:
     11  1.5.2.2  pgoyette  * 1. Redistributions of source code must retain the above copyright
     12  1.5.2.2  pgoyette  *    notice, this list of conditions and the following disclaimer.
     13  1.5.2.2  pgoyette  * 2. Redistributions in binary form must reproduce the above copyright
     14  1.5.2.2  pgoyette  *    notice, this list of conditions and the following disclaimer in the
     15  1.5.2.2  pgoyette  *    documentation and/or other materials provided with the distribution.
     16  1.5.2.2  pgoyette  *
     17  1.5.2.2  pgoyette  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     18  1.5.2.2  pgoyette  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     19  1.5.2.2  pgoyette  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     20  1.5.2.2  pgoyette  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     21  1.5.2.2  pgoyette  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     22  1.5.2.2  pgoyette  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     23  1.5.2.2  pgoyette  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     24  1.5.2.2  pgoyette  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     25  1.5.2.2  pgoyette  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     26  1.5.2.2  pgoyette  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     27  1.5.2.2  pgoyette  * SUCH DAMAGE.
     28  1.5.2.2  pgoyette  */
     29  1.5.2.2  pgoyette 
     30  1.5.2.2  pgoyette #include "efiboot.h"
     31  1.5.2.2  pgoyette #include "efiblock.h"
     32  1.5.2.2  pgoyette #include "efifdt.h"
     33  1.5.2.3  pgoyette #include "efienv.h"
     34  1.5.2.2  pgoyette 
     35  1.5.2.2  pgoyette #include <sys/bootblock.h>
     36  1.5.2.2  pgoyette #include <sys/boot_flag.h>
     37  1.5.2.2  pgoyette #include <machine/limits.h>
     38  1.5.2.2  pgoyette 
     39  1.5.2.2  pgoyette #include <loadfile.h>
     40  1.5.2.2  pgoyette 
     41  1.5.2.2  pgoyette extern const char bootprog_name[], bootprog_rev[], bootprog_kernrev[];
     42  1.5.2.2  pgoyette 
     43  1.5.2.2  pgoyette extern char twiddle_toggle;
     44  1.5.2.2  pgoyette 
     45  1.5.2.2  pgoyette static const char * const names[][2] = {
     46  1.5.2.2  pgoyette 	{ "netbsd", "netbsd.gz" },
     47  1.5.2.2  pgoyette 	{ "onetbsd", "onetbsd.gz" },
     48  1.5.2.2  pgoyette 	{ "netbsd.old", "netbsd.old.gz" },
     49  1.5.2.2  pgoyette };
     50  1.5.2.2  pgoyette 
     51  1.5.2.2  pgoyette #define NUMNAMES	__arraycount(names)
     52  1.5.2.2  pgoyette #define DEFFILENAME	names[0][0]
     53  1.5.2.2  pgoyette 
     54  1.5.2.2  pgoyette #define	DEFTIMEOUT	5
     55  1.5.2.2  pgoyette 
     56  1.5.2.2  pgoyette static char default_device[32];
     57  1.5.2.3  pgoyette static char initrd_path[255];
     58  1.5.2.3  pgoyette static char dtb_path[255];
     59  1.5.2.2  pgoyette 
     60  1.5.2.2  pgoyette void	command_boot(char *);
     61  1.5.2.2  pgoyette void	command_dev(char *);
     62  1.5.2.3  pgoyette void	command_dtb(char *);
     63  1.5.2.3  pgoyette void	command_initrd(char *);
     64  1.5.2.2  pgoyette void	command_ls(char *);
     65  1.5.2.3  pgoyette void	command_printenv(char *);
     66  1.5.2.3  pgoyette void	command_setenv(char *);
     67  1.5.2.3  pgoyette void	command_clearenv(char *);
     68  1.5.2.3  pgoyette void	command_resetenv(char *);
     69  1.5.2.2  pgoyette void	command_reset(char *);
     70  1.5.2.2  pgoyette void	command_version(char *);
     71  1.5.2.2  pgoyette void	command_quit(char *);
     72  1.5.2.2  pgoyette 
     73  1.5.2.2  pgoyette const struct boot_command commands[] = {
     74  1.5.2.3  pgoyette 	{ "boot",	command_boot,		"boot [dev:][filename] [args]\n     (ex. \"hd0a:\\netbsd.old -s\"" },
     75  1.5.2.2  pgoyette 	{ "dev",	command_dev,		"dev" },
     76  1.5.2.3  pgoyette 	{ "dtb",	command_dtb,		"dtb [dev:][filename]" },
     77  1.5.2.3  pgoyette 	{ "initrd",	command_initrd,		"initrd [dev:][filename]" },
     78  1.5.2.2  pgoyette 	{ "ls",		command_ls,		"ls [hdNn:/path]" },
     79  1.5.2.3  pgoyette 	{ "printenv",	command_printenv,	"printenv [key]" },
     80  1.5.2.3  pgoyette 	{ "setenv",	command_setenv,		"setenv <key> <value>" },
     81  1.5.2.3  pgoyette 	{ "clearenv",	command_clearenv,	"clearenv <key>" },
     82  1.5.2.3  pgoyette 	{ "resetenv",	command_resetenv,	"resetenv" },
     83  1.5.2.3  pgoyette 	{ "reboot",	command_reset,		"reboot|reset" },
     84  1.5.2.3  pgoyette 	{ "reset",	command_reset,		NULL },
     85  1.5.2.2  pgoyette 	{ "version",	command_version,	"version" },
     86  1.5.2.2  pgoyette 	{ "help",	command_help,		"help|?" },
     87  1.5.2.2  pgoyette 	{ "?",		command_help,		NULL },
     88  1.5.2.2  pgoyette 	{ "quit",	command_quit,		"quit" },
     89  1.5.2.2  pgoyette 	{ NULL,		NULL },
     90  1.5.2.2  pgoyette };
     91  1.5.2.2  pgoyette 
     92  1.5.2.2  pgoyette void
     93  1.5.2.2  pgoyette command_help(char *arg)
     94  1.5.2.2  pgoyette {
     95  1.5.2.2  pgoyette 	int n;
     96  1.5.2.2  pgoyette 
     97  1.5.2.2  pgoyette 	printf("commands are:\n");
     98  1.5.2.2  pgoyette 	for (n = 0; commands[n].c_name; n++) {
     99  1.5.2.2  pgoyette 		if (commands[n].c_help)
    100  1.5.2.2  pgoyette 			printf("%s\n", commands[n].c_help);
    101  1.5.2.2  pgoyette 	}
    102  1.5.2.2  pgoyette }
    103  1.5.2.2  pgoyette 
    104  1.5.2.2  pgoyette void
    105  1.5.2.2  pgoyette command_boot(char *arg)
    106  1.5.2.2  pgoyette {
    107  1.5.2.2  pgoyette 	char *fname = arg;
    108  1.5.2.2  pgoyette 	char *bootargs = gettrailer(arg);
    109  1.5.2.2  pgoyette 
    110  1.5.2.2  pgoyette 	exec_netbsd(*fname ? fname : DEFFILENAME, bootargs);
    111  1.5.2.2  pgoyette }
    112  1.5.2.2  pgoyette 
    113  1.5.2.2  pgoyette void
    114  1.5.2.2  pgoyette command_dev(char *arg)
    115  1.5.2.2  pgoyette {
    116  1.5.2.2  pgoyette 	if (arg && *arg) {
    117  1.5.2.2  pgoyette 		set_default_device(arg);
    118  1.5.2.2  pgoyette 	} else {
    119  1.5.2.2  pgoyette 		efi_block_show();
    120  1.5.2.2  pgoyette 		efi_net_show();
    121  1.5.2.2  pgoyette 		efi_pxe_show();
    122  1.5.2.2  pgoyette 	}
    123  1.5.2.2  pgoyette 
    124  1.5.2.2  pgoyette 	if (strlen(default_device) > 0) {
    125  1.5.2.2  pgoyette 		printf("\n");
    126  1.5.2.2  pgoyette 		printf("default: %s\n", default_device);
    127  1.5.2.2  pgoyette 	}
    128  1.5.2.2  pgoyette }
    129  1.5.2.2  pgoyette 
    130  1.5.2.2  pgoyette void
    131  1.5.2.3  pgoyette command_dtb(char *arg)
    132  1.5.2.3  pgoyette {
    133  1.5.2.3  pgoyette 	set_dtb_path(arg);
    134  1.5.2.3  pgoyette }
    135  1.5.2.3  pgoyette 
    136  1.5.2.3  pgoyette void
    137  1.5.2.3  pgoyette command_initrd(char *arg)
    138  1.5.2.3  pgoyette {
    139  1.5.2.3  pgoyette 	set_initrd_path(arg);
    140  1.5.2.3  pgoyette }
    141  1.5.2.3  pgoyette 
    142  1.5.2.3  pgoyette void
    143  1.5.2.2  pgoyette command_ls(char *arg)
    144  1.5.2.2  pgoyette {
    145  1.5.2.2  pgoyette 	ls(arg);
    146  1.5.2.2  pgoyette }
    147  1.5.2.2  pgoyette 
    148  1.5.2.2  pgoyette void
    149  1.5.2.3  pgoyette command_printenv(char *arg)
    150  1.5.2.3  pgoyette {
    151  1.5.2.3  pgoyette 	char *val;
    152  1.5.2.3  pgoyette 
    153  1.5.2.3  pgoyette 	if (arg && *arg) {
    154  1.5.2.3  pgoyette 		val = efi_env_get(arg);
    155  1.5.2.3  pgoyette 		if (val) {
    156  1.5.2.3  pgoyette 			printf("\"%s\" = \"%s\"\n", arg, val);
    157  1.5.2.3  pgoyette 			FreePool(val);
    158  1.5.2.3  pgoyette 		}
    159  1.5.2.3  pgoyette 	} else {
    160  1.5.2.3  pgoyette 		efi_env_print();
    161  1.5.2.3  pgoyette 	}
    162  1.5.2.3  pgoyette }
    163  1.5.2.3  pgoyette 
    164  1.5.2.3  pgoyette void
    165  1.5.2.3  pgoyette command_setenv(char *arg)
    166  1.5.2.3  pgoyette {
    167  1.5.2.3  pgoyette 	char *spc;
    168  1.5.2.3  pgoyette 
    169  1.5.2.3  pgoyette 	spc = strchr(arg, ' ');
    170  1.5.2.3  pgoyette 	if (spc == NULL || spc[1] == '\0') {
    171  1.5.2.3  pgoyette 		command_help("");
    172  1.5.2.3  pgoyette 		return;
    173  1.5.2.3  pgoyette 	}
    174  1.5.2.3  pgoyette 
    175  1.5.2.3  pgoyette 	*spc = '\0';
    176  1.5.2.3  pgoyette 	efi_env_set(arg, spc + 1);
    177  1.5.2.3  pgoyette }
    178  1.5.2.3  pgoyette 
    179  1.5.2.3  pgoyette void
    180  1.5.2.3  pgoyette command_clearenv(char *arg)
    181  1.5.2.3  pgoyette {
    182  1.5.2.3  pgoyette 	if (*arg == '\0') {
    183  1.5.2.3  pgoyette 		command_help("");
    184  1.5.2.3  pgoyette 		return;
    185  1.5.2.3  pgoyette 	}
    186  1.5.2.3  pgoyette 	efi_env_clear(arg);
    187  1.5.2.3  pgoyette }
    188  1.5.2.3  pgoyette 
    189  1.5.2.3  pgoyette void
    190  1.5.2.3  pgoyette command_resetenv(char *arg)
    191  1.5.2.3  pgoyette {
    192  1.5.2.3  pgoyette 	efi_env_reset();
    193  1.5.2.3  pgoyette }
    194  1.5.2.3  pgoyette 
    195  1.5.2.3  pgoyette void
    196  1.5.2.2  pgoyette command_version(char *arg)
    197  1.5.2.2  pgoyette {
    198  1.5.2.2  pgoyette 	char *ufirmware;
    199  1.5.2.2  pgoyette 	int rv;
    200  1.5.2.2  pgoyette 
    201  1.5.2.2  pgoyette 	printf("EFI version: %d.%02d\n",
    202  1.5.2.2  pgoyette 	    ST->Hdr.Revision >> 16, ST->Hdr.Revision & 0xffff);
    203  1.5.2.2  pgoyette 	ufirmware = NULL;
    204  1.5.2.2  pgoyette 	rv = ucs2_to_utf8(ST->FirmwareVendor, &ufirmware);
    205  1.5.2.2  pgoyette 	if (rv == 0) {
    206  1.5.2.2  pgoyette 		printf("EFI Firmware: %s (rev %d.%02d)\n", ufirmware,
    207  1.5.2.2  pgoyette 		    ST->FirmwareRevision >> 16,
    208  1.5.2.2  pgoyette 		    ST->FirmwareRevision & 0xffff);
    209  1.5.2.2  pgoyette 		FreePool(ufirmware);
    210  1.5.2.2  pgoyette 	}
    211  1.5.2.2  pgoyette 
    212  1.5.2.2  pgoyette 	efi_fdt_show();
    213  1.5.2.2  pgoyette }
    214  1.5.2.2  pgoyette 
    215  1.5.2.2  pgoyette void
    216  1.5.2.2  pgoyette command_quit(char *arg)
    217  1.5.2.2  pgoyette {
    218  1.5.2.2  pgoyette 	efi_exit();
    219  1.5.2.2  pgoyette }
    220  1.5.2.2  pgoyette 
    221  1.5.2.3  pgoyette void
    222  1.5.2.3  pgoyette command_reset(char *arg)
    223  1.5.2.3  pgoyette {
    224  1.5.2.3  pgoyette 	efi_reboot();
    225  1.5.2.3  pgoyette }
    226  1.5.2.3  pgoyette 
    227  1.5.2.2  pgoyette int
    228  1.5.2.2  pgoyette set_default_device(char *arg)
    229  1.5.2.2  pgoyette {
    230  1.5.2.2  pgoyette 	if (strlen(arg) + 1 > sizeof(default_device))
    231  1.5.2.2  pgoyette 		return ERANGE;
    232  1.5.2.2  pgoyette 	strcpy(default_device, arg);
    233  1.5.2.2  pgoyette 	return 0;
    234  1.5.2.2  pgoyette }
    235  1.5.2.2  pgoyette 
    236  1.5.2.2  pgoyette char *
    237  1.5.2.2  pgoyette get_default_device(void)
    238  1.5.2.2  pgoyette {
    239  1.5.2.2  pgoyette 	return default_device;
    240  1.5.2.2  pgoyette }
    241  1.5.2.2  pgoyette 
    242  1.5.2.3  pgoyette int
    243  1.5.2.3  pgoyette set_initrd_path(char *arg)
    244  1.5.2.3  pgoyette {
    245  1.5.2.3  pgoyette 	if (strlen(arg) + 1 > sizeof(initrd_path))
    246  1.5.2.3  pgoyette 		return ERANGE;
    247  1.5.2.3  pgoyette 	strcpy(initrd_path, arg);
    248  1.5.2.3  pgoyette 	return 0;
    249  1.5.2.3  pgoyette }
    250  1.5.2.3  pgoyette 
    251  1.5.2.3  pgoyette char *
    252  1.5.2.3  pgoyette get_initrd_path(void)
    253  1.5.2.3  pgoyette {
    254  1.5.2.3  pgoyette 	return initrd_path;
    255  1.5.2.3  pgoyette }
    256  1.5.2.3  pgoyette 
    257  1.5.2.3  pgoyette int
    258  1.5.2.3  pgoyette set_dtb_path(char *arg)
    259  1.5.2.3  pgoyette {
    260  1.5.2.3  pgoyette 	if (strlen(arg) + 1 > sizeof(dtb_path))
    261  1.5.2.3  pgoyette 		return ERANGE;
    262  1.5.2.3  pgoyette 	strcpy(dtb_path, arg);
    263  1.5.2.3  pgoyette 	return 0;
    264  1.5.2.3  pgoyette }
    265  1.5.2.3  pgoyette 
    266  1.5.2.3  pgoyette char *
    267  1.5.2.3  pgoyette get_dtb_path(void)
    268  1.5.2.3  pgoyette {
    269  1.5.2.3  pgoyette 	return dtb_path;
    270  1.5.2.3  pgoyette }
    271  1.5.2.3  pgoyette 
    272  1.5.2.2  pgoyette void
    273  1.5.2.2  pgoyette print_banner(void)
    274  1.5.2.2  pgoyette {
    275  1.5.2.2  pgoyette 	printf("\n\n"
    276  1.5.2.2  pgoyette 	    ">> %s, Revision %s (from NetBSD %s)\n",
    277  1.5.2.2  pgoyette 	    bootprog_name, bootprog_rev, bootprog_kernrev);
    278  1.5.2.2  pgoyette }
    279  1.5.2.2  pgoyette 
    280  1.5.2.3  pgoyette static void
    281  1.5.2.3  pgoyette read_env(void)
    282  1.5.2.3  pgoyette {
    283  1.5.2.3  pgoyette 	char *s;
    284  1.5.2.3  pgoyette 
    285  1.5.2.3  pgoyette 	s = efi_env_get("fdtfile");
    286  1.5.2.3  pgoyette 	if (s) {
    287  1.5.2.3  pgoyette #ifdef EFIBOOT_DEBUG
    288  1.5.2.3  pgoyette 		printf(">> Setting DTB path to '%s' from environment\n", s);
    289  1.5.2.3  pgoyette #endif
    290  1.5.2.3  pgoyette 		set_dtb_path(s);
    291  1.5.2.3  pgoyette 		FreePool(s);
    292  1.5.2.3  pgoyette 	}
    293  1.5.2.3  pgoyette 
    294  1.5.2.3  pgoyette 	s = efi_env_get("initrd");
    295  1.5.2.3  pgoyette 	if (s) {
    296  1.5.2.3  pgoyette #ifdef EFIBOOT_DEBUG
    297  1.5.2.3  pgoyette 		printf(">> Setting initrd path to '%s' from environment\n", s);
    298  1.5.2.3  pgoyette #endif
    299  1.5.2.3  pgoyette 		set_initrd_path(s);
    300  1.5.2.3  pgoyette 		FreePool(s);
    301  1.5.2.3  pgoyette 	}
    302  1.5.2.3  pgoyette 
    303  1.5.2.3  pgoyette 	s = efi_env_get("rootdev");
    304  1.5.2.3  pgoyette 	if (s) {
    305  1.5.2.3  pgoyette #ifdef EFIBOOT_DEBUG
    306  1.5.2.3  pgoyette 		printf(">> Setting default device to '%s' from environment\n", s);
    307  1.5.2.3  pgoyette #endif
    308  1.5.2.3  pgoyette 		set_default_device(s);
    309  1.5.2.3  pgoyette 		FreePool(s);
    310  1.5.2.3  pgoyette 	}
    311  1.5.2.3  pgoyette }
    312  1.5.2.3  pgoyette 
    313  1.5.2.2  pgoyette void
    314  1.5.2.2  pgoyette boot(void)
    315  1.5.2.2  pgoyette {
    316  1.5.2.2  pgoyette 	int currname, c;
    317  1.5.2.2  pgoyette 
    318  1.5.2.3  pgoyette 	read_env();
    319  1.5.2.3  pgoyette 
    320  1.5.2.2  pgoyette 	print_banner();
    321  1.5.2.2  pgoyette 
    322  1.5.2.2  pgoyette 	printf("Press return to boot now, any other key for boot prompt\n");
    323  1.5.2.2  pgoyette 	for (currname = 0; currname < NUMNAMES; currname++) {
    324  1.5.2.2  pgoyette 		printf("booting %s - starting in ", names[currname][0]);
    325  1.5.2.2  pgoyette 
    326  1.5.2.2  pgoyette 		c = awaitkey(DEFTIMEOUT, 1);
    327  1.5.2.2  pgoyette 		if ((c != '\r') && (c != '\n') && (c != '\0')) {
    328  1.5.2.2  pgoyette 			bootprompt(); /* does not return */
    329  1.5.2.2  pgoyette 		}
    330  1.5.2.2  pgoyette 
    331  1.5.2.2  pgoyette 		/*
    332  1.5.2.2  pgoyette 		 * try pairs of names[] entries, foo and foo.gz
    333  1.5.2.2  pgoyette 		 */
    334  1.5.2.2  pgoyette 		exec_netbsd(names[currname][0], "");
    335  1.5.2.2  pgoyette 		exec_netbsd(names[currname][1], "");
    336  1.5.2.2  pgoyette 	}
    337  1.5.2.2  pgoyette 
    338  1.5.2.2  pgoyette 	bootprompt();	/* does not return */
    339  1.5.2.2  pgoyette }
    340