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