xref: /freebsd/stand/i386/gptboot/gptboot.c (revision 86bb84d57695fbe94455259014edeb5aacec13f4)
1ca987d46SWarner Losh /*-
2ca987d46SWarner Losh  * Copyright (c) 1998 Robert Nordier
3ca987d46SWarner Losh  * All rights reserved.
4ca987d46SWarner Losh  *
5ca987d46SWarner Losh  * Redistribution and use in source and binary forms are freely
6ca987d46SWarner Losh  * permitted provided that the above copyright notice and this
7ca987d46SWarner Losh  * paragraph and the following disclaimer are duplicated in all
8ca987d46SWarner Losh  * such forms.
9ca987d46SWarner Losh  *
10ca987d46SWarner Losh  * This software is provided "AS IS" and without any express or
11ca987d46SWarner Losh  * implied warranties, including, without limitation, the implied
12ca987d46SWarner Losh  * warranties of merchantability and fitness for a particular
13ca987d46SWarner Losh  * purpose.
14ca987d46SWarner Losh  */
15ca987d46SWarner Losh 
16ca987d46SWarner Losh #include <sys/cdefs.h>
17ca987d46SWarner Losh __FBSDID("$FreeBSD$");
18ca987d46SWarner Losh 
19ca987d46SWarner Losh #include <sys/param.h>
20ca987d46SWarner Losh #include <sys/gpt.h>
21ca987d46SWarner Losh #include <sys/dirent.h>
22ca987d46SWarner Losh #include <sys/reboot.h>
23ca987d46SWarner Losh 
24ca987d46SWarner Losh #include <machine/bootinfo.h>
25ca987d46SWarner Losh #include <machine/elf.h>
26ca987d46SWarner Losh #include <machine/pc/bios.h>
27ca987d46SWarner Losh #include <machine/psl.h>
28ca987d46SWarner Losh 
29ca987d46SWarner Losh #include <stdarg.h>
30ca987d46SWarner Losh 
31ca987d46SWarner Losh #include <a.out.h>
32ca987d46SWarner Losh 
33ca987d46SWarner Losh #include <btxv86.h>
34ca987d46SWarner Losh 
3565628439SWarner Losh #include "stand.h"
361227a4f4SWarner Losh 
37ca987d46SWarner Losh #include "bootargs.h"
38ca987d46SWarner Losh #include "lib.h"
39ca987d46SWarner Losh #include "rbx.h"
40ca987d46SWarner Losh #include "drv.h"
41ca987d46SWarner Losh #include "cons.h"
42ca987d46SWarner Losh #include "gpt.h"
43ca987d46SWarner Losh #include "paths.h"
44ca987d46SWarner Losh 
45ca987d46SWarner Losh #define ARGS		0x900
46ca987d46SWarner Losh #define NOPT		14
47ca987d46SWarner Losh #define NDEV		3
48ca987d46SWarner Losh #define MEM_BASE	0x12
49ca987d46SWarner Losh #define MEM_EXT 	0x15
50ca987d46SWarner Losh 
51ca987d46SWarner Losh #define DRV_HARD	0x80
52ca987d46SWarner Losh #define DRV_MASK	0x7f
53ca987d46SWarner Losh 
54ca987d46SWarner Losh #define TYPE_AD		0
55ca987d46SWarner Losh #define TYPE_DA		1
56ca987d46SWarner Losh #define TYPE_MAXHARD	TYPE_DA
57ca987d46SWarner Losh #define TYPE_FD		2
58ca987d46SWarner Losh 
59ca987d46SWarner Losh extern uint32_t _end;
60ca987d46SWarner Losh 
61ca987d46SWarner Losh static const uuid_t freebsd_ufs_uuid = GPT_ENT_TYPE_FREEBSD_UFS;
62ca987d46SWarner Losh static const char optstr[NOPT] = "DhaCcdgmnpqrsv"; /* Also 'P', 'S' */
63ca987d46SWarner Losh static const unsigned char flags[NOPT] = {
64ca987d46SWarner Losh 	RBX_DUAL,
65ca987d46SWarner Losh 	RBX_SERIAL,
66ca987d46SWarner Losh 	RBX_ASKNAME,
67ca987d46SWarner Losh 	RBX_CDROM,
68ca987d46SWarner Losh 	RBX_CONFIG,
69ca987d46SWarner Losh 	RBX_KDB,
70ca987d46SWarner Losh 	RBX_GDB,
71ca987d46SWarner Losh 	RBX_MUTE,
72ca987d46SWarner Losh 	RBX_NOINTR,
73ca987d46SWarner Losh 	RBX_PAUSE,
74ca987d46SWarner Losh 	RBX_QUIET,
75ca987d46SWarner Losh 	RBX_DFLTROOT,
76ca987d46SWarner Losh 	RBX_SINGLE,
77ca987d46SWarner Losh 	RBX_VERBOSE
78ca987d46SWarner Losh };
79ca987d46SWarner Losh uint32_t opts;
80ca987d46SWarner Losh 
81ca987d46SWarner Losh static const char *const dev_nm[NDEV] = {"ad", "da", "fd"};
82ca987d46SWarner Losh static const unsigned char dev_maj[NDEV] = {30, 4, 2};
83ca987d46SWarner Losh 
84ca987d46SWarner Losh static struct dsk dsk;
85ca987d46SWarner Losh static char kname[1024];
86ca987d46SWarner Losh static int comspeed = SIOSPD;
87ca987d46SWarner Losh static struct bootinfo bootinfo;
88ca987d46SWarner Losh #ifdef LOADER_GELI_SUPPORT
89ca987d46SWarner Losh static struct geli_boot_args geliargs;
90ca987d46SWarner Losh #endif
91ca987d46SWarner Losh 
92ca987d46SWarner Losh static vm_offset_t	high_heap_base;
93ca987d46SWarner Losh static uint32_t		bios_basemem, bios_extmem, high_heap_size;
94ca987d46SWarner Losh 
95ca987d46SWarner Losh static struct bios_smap smap;
96ca987d46SWarner Losh 
97ca987d46SWarner Losh /*
98ca987d46SWarner Losh  * The minimum amount of memory to reserve in bios_extmem for the heap.
99ca987d46SWarner Losh  */
100ca987d46SWarner Losh #define	HEAP_MIN	(3 * 1024 * 1024)
101ca987d46SWarner Losh 
102ca987d46SWarner Losh static char *heap_next;
103ca987d46SWarner Losh static char *heap_end;
104ca987d46SWarner Losh 
105ca987d46SWarner Losh static void load(void);
106ca987d46SWarner Losh static int parse_cmds(char *, int *);
107ca987d46SWarner Losh static int dskread(void *, daddr_t, unsigned);
108ca987d46SWarner Losh #ifdef LOADER_GELI_SUPPORT
109ca987d46SWarner Losh static int vdev_read(void *vdev __unused, void *priv, off_t off, void *buf,
110ca987d46SWarner Losh 	size_t bytes);
111ca987d46SWarner Losh #endif
112ca987d46SWarner Losh 
113ca987d46SWarner Losh #include "ufsread.c"
114ca987d46SWarner Losh #include "gpt.c"
115ca987d46SWarner Losh #ifdef LOADER_GELI_SUPPORT
116ca987d46SWarner Losh #include "geliboot.c"
117ca987d46SWarner Losh static char gelipw[GELI_PW_MAXLEN];
118ca987d46SWarner Losh static struct keybuf *gelibuf;
119ca987d46SWarner Losh #endif
120ca987d46SWarner Losh 
121ca987d46SWarner Losh static inline int
122ca987d46SWarner Losh xfsread(ufs_ino_t inode, void *buf, size_t nbyte)
123ca987d46SWarner Losh {
124ca987d46SWarner Losh 
125ca987d46SWarner Losh 	if ((size_t)fsread(inode, buf, nbyte) != nbyte) {
126ca987d46SWarner Losh 		printf("Invalid %s\n", "format");
127ca987d46SWarner Losh 		return (-1);
128ca987d46SWarner Losh 	}
129ca987d46SWarner Losh 	return (0);
130ca987d46SWarner Losh }
131ca987d46SWarner Losh 
132ca987d46SWarner Losh static void
133ca987d46SWarner Losh bios_getmem(void)
134ca987d46SWarner Losh {
135ca987d46SWarner Losh     uint64_t size;
136ca987d46SWarner Losh 
137ca987d46SWarner Losh     /* Parse system memory map */
138ca987d46SWarner Losh     v86.ebx = 0;
139ca987d46SWarner Losh     do {
140ca987d46SWarner Losh 	v86.ctl = V86_FLAGS;
141ca987d46SWarner Losh 	v86.addr = MEM_EXT;		/* int 0x15 function 0xe820*/
142ca987d46SWarner Losh 	v86.eax = 0xe820;
143ca987d46SWarner Losh 	v86.ecx = sizeof(struct bios_smap);
144ca987d46SWarner Losh 	v86.edx = SMAP_SIG;
145ca987d46SWarner Losh 	v86.es = VTOPSEG(&smap);
146ca987d46SWarner Losh 	v86.edi = VTOPOFF(&smap);
147ca987d46SWarner Losh 	v86int();
148ca987d46SWarner Losh 	if ((v86.efl & 1) || (v86.eax != SMAP_SIG))
149ca987d46SWarner Losh 	    break;
150ca987d46SWarner Losh 	/* look for a low-memory segment that's large enough */
151ca987d46SWarner Losh 	if ((smap.type == SMAP_TYPE_MEMORY) && (smap.base == 0) &&
152ca987d46SWarner Losh 	    (smap.length >= (512 * 1024)))
153ca987d46SWarner Losh 	    bios_basemem = smap.length;
154ca987d46SWarner Losh 	/* look for the first segment in 'extended' memory */
155ca987d46SWarner Losh 	if ((smap.type == SMAP_TYPE_MEMORY) && (smap.base == 0x100000)) {
156ca987d46SWarner Losh 	    bios_extmem = smap.length;
157ca987d46SWarner Losh 	}
158ca987d46SWarner Losh 
159ca987d46SWarner Losh 	/*
160ca987d46SWarner Losh 	 * Look for the largest segment in 'extended' memory beyond
161ca987d46SWarner Losh 	 * 1MB but below 4GB.
162ca987d46SWarner Losh 	 */
163ca987d46SWarner Losh 	if ((smap.type == SMAP_TYPE_MEMORY) && (smap.base > 0x100000) &&
164ca987d46SWarner Losh 	    (smap.base < 0x100000000ull)) {
165ca987d46SWarner Losh 	    size = smap.length;
166ca987d46SWarner Losh 
167ca987d46SWarner Losh 	    /*
168ca987d46SWarner Losh 	     * If this segment crosses the 4GB boundary, truncate it.
169ca987d46SWarner Losh 	     */
170ca987d46SWarner Losh 	    if (smap.base + size > 0x100000000ull)
171ca987d46SWarner Losh 		size = 0x100000000ull - smap.base;
172ca987d46SWarner Losh 
173ca987d46SWarner Losh 	    if (size > high_heap_size) {
174ca987d46SWarner Losh 		high_heap_size = size;
175ca987d46SWarner Losh 		high_heap_base = smap.base;
176ca987d46SWarner Losh 	    }
177ca987d46SWarner Losh 	}
178ca987d46SWarner Losh     } while (v86.ebx != 0);
179ca987d46SWarner Losh 
180ca987d46SWarner Losh     /* Fall back to the old compatibility function for base memory */
181ca987d46SWarner Losh     if (bios_basemem == 0) {
182ca987d46SWarner Losh 	v86.ctl = 0;
183ca987d46SWarner Losh 	v86.addr = 0x12;		/* int 0x12 */
184ca987d46SWarner Losh 	v86int();
185ca987d46SWarner Losh 
186ca987d46SWarner Losh 	bios_basemem = (v86.eax & 0xffff) * 1024;
187ca987d46SWarner Losh     }
188ca987d46SWarner Losh 
189ca987d46SWarner Losh     /* Fall back through several compatibility functions for extended memory */
190ca987d46SWarner Losh     if (bios_extmem == 0) {
191ca987d46SWarner Losh 	v86.ctl = V86_FLAGS;
192ca987d46SWarner Losh 	v86.addr = 0x15;		/* int 0x15 function 0xe801*/
193ca987d46SWarner Losh 	v86.eax = 0xe801;
194ca987d46SWarner Losh 	v86int();
195ca987d46SWarner Losh 	if (!(v86.efl & 1)) {
196ca987d46SWarner Losh 	    bios_extmem = ((v86.ecx & 0xffff) + ((v86.edx & 0xffff) * 64)) * 1024;
197ca987d46SWarner Losh 	}
198ca987d46SWarner Losh     }
199ca987d46SWarner Losh     if (bios_extmem == 0) {
200ca987d46SWarner Losh 	v86.ctl = 0;
201ca987d46SWarner Losh 	v86.addr = 0x15;		/* int 0x15 function 0x88*/
202ca987d46SWarner Losh 	v86.eax = 0x8800;
203ca987d46SWarner Losh 	v86int();
204ca987d46SWarner Losh 	bios_extmem = (v86.eax & 0xffff) * 1024;
205ca987d46SWarner Losh     }
206ca987d46SWarner Losh 
207ca987d46SWarner Losh     /*
208ca987d46SWarner Losh      * If we have extended memory and did not find a suitable heap
209ca987d46SWarner Losh      * region in the SMAP, use the last 3MB of 'extended' memory as a
210ca987d46SWarner Losh      * high heap candidate.
211ca987d46SWarner Losh      */
212ca987d46SWarner Losh     if (bios_extmem >= HEAP_MIN && high_heap_size < HEAP_MIN) {
213ca987d46SWarner Losh 	high_heap_size = HEAP_MIN;
214ca987d46SWarner Losh 	high_heap_base = bios_extmem + 0x100000 - HEAP_MIN;
215ca987d46SWarner Losh     }
216ca987d46SWarner Losh }
217ca987d46SWarner Losh 
218ca987d46SWarner Losh static int
219ca987d46SWarner Losh gptinit(void)
220ca987d46SWarner Losh {
221ca987d46SWarner Losh 
222ca987d46SWarner Losh 	if (gptread(&freebsd_ufs_uuid, &dsk, dmadat->secbuf) == -1) {
223ca987d46SWarner Losh 		printf("%s: unable to load GPT\n", BOOTPROG);
224ca987d46SWarner Losh 		return (-1);
225ca987d46SWarner Losh 	}
226ca987d46SWarner Losh 	if (gptfind(&freebsd_ufs_uuid, &dsk, dsk.part) == -1) {
227ca987d46SWarner Losh 		printf("%s: no UFS partition was found\n", BOOTPROG);
228ca987d46SWarner Losh 		return (-1);
229ca987d46SWarner Losh 	}
230ca987d46SWarner Losh #ifdef LOADER_GELI_SUPPORT
231ca987d46SWarner Losh 	if (geli_taste(vdev_read, &dsk, (gpttable[curent].ent_lba_end -
232ca987d46SWarner Losh 	    gpttable[curent].ent_lba_start)) == 0) {
2331227a4f4SWarner Losh 		if (geli_havekey(&dsk) != 0 && geli_passphrase(gelipw,
234ca987d46SWarner Losh 		    dsk.unit, 'p', curent + 1, &dsk) != 0) {
235ca987d46SWarner Losh 			printf("%s: unable to decrypt GELI key\n", BOOTPROG);
236ca987d46SWarner Losh 			return (-1);
237ca987d46SWarner Losh 		}
238ca987d46SWarner Losh 	}
239ca987d46SWarner Losh #endif
240ca987d46SWarner Losh 
241ca987d46SWarner Losh 	dsk_meta = 0;
242ca987d46SWarner Losh 	return (0);
243ca987d46SWarner Losh }
244ca987d46SWarner Losh 
2451227a4f4SWarner Losh int main(void);
2461227a4f4SWarner Losh 
247ca987d46SWarner Losh int
248ca987d46SWarner Losh main(void)
249ca987d46SWarner Losh {
250ca987d46SWarner Losh 	char cmd[512], cmdtmp[512];
251ca987d46SWarner Losh 	ssize_t sz;
252ca987d46SWarner Losh 	int autoboot, dskupdated;
253ca987d46SWarner Losh 	ufs_ino_t ino;
254ca987d46SWarner Losh 
255ca987d46SWarner Losh 	dmadat = (void *)(roundup2(__base + (int32_t)&_end, 0x10000) - __base);
256ca987d46SWarner Losh 
257ca987d46SWarner Losh 	bios_getmem();
258ca987d46SWarner Losh 
259ca987d46SWarner Losh 	if (high_heap_size > 0) {
260ca987d46SWarner Losh 		heap_end = PTOV(high_heap_base + high_heap_size);
261ca987d46SWarner Losh 		heap_next = PTOV(high_heap_base);
262ca987d46SWarner Losh 	} else {
263ca987d46SWarner Losh 		heap_next = (char *)dmadat + sizeof(*dmadat);
264ca987d46SWarner Losh 		heap_end = (char *)PTOV(bios_basemem);
265ca987d46SWarner Losh 	}
26665628439SWarner Losh 	setheap(heap_next, heap_end);
267ca987d46SWarner Losh 
268ca987d46SWarner Losh 	v86.ctl = V86_FLAGS;
269ca987d46SWarner Losh 	v86.efl = PSL_RESERVED_DEFAULT | PSL_I;
270ca987d46SWarner Losh 	dsk.drive = *(uint8_t *)PTOV(ARGS);
271ca987d46SWarner Losh 	dsk.type = dsk.drive & DRV_HARD ? TYPE_AD : TYPE_FD;
272ca987d46SWarner Losh 	dsk.unit = dsk.drive & DRV_MASK;
273ca987d46SWarner Losh 	dsk.part = -1;
274ca987d46SWarner Losh 	dsk.start = 0;
275ca987d46SWarner Losh 	bootinfo.bi_version = BOOTINFO_VERSION;
276ca987d46SWarner Losh 	bootinfo.bi_size = sizeof(bootinfo);
277ca987d46SWarner Losh 	bootinfo.bi_basemem = bios_basemem / 1024;
278ca987d46SWarner Losh 	bootinfo.bi_extmem = bios_extmem / 1024;
279ca987d46SWarner Losh 	bootinfo.bi_memsizes_valid++;
280ca987d46SWarner Losh 	bootinfo.bi_bios_dev = dsk.drive;
281ca987d46SWarner Losh 
282ca987d46SWarner Losh #ifdef LOADER_GELI_SUPPORT
283ca987d46SWarner Losh 	geli_init();
284ca987d46SWarner Losh #endif
285ca987d46SWarner Losh 	/* Process configuration file */
286ca987d46SWarner Losh 
287ca987d46SWarner Losh 	if (gptinit() != 0)
288ca987d46SWarner Losh 		return (-1);
289ca987d46SWarner Losh 
290ca987d46SWarner Losh 	autoboot = 1;
291ca987d46SWarner Losh 	*cmd = '\0';
292ca987d46SWarner Losh 
293ca987d46SWarner Losh 	for (;;) {
294ca987d46SWarner Losh 		*kname = '\0';
295ca987d46SWarner Losh 		if ((ino = lookup(PATH_CONFIG)) ||
296ca987d46SWarner Losh 		    (ino = lookup(PATH_DOTCONFIG))) {
297ca987d46SWarner Losh 			sz = fsread(ino, cmd, sizeof(cmd) - 1);
298ca987d46SWarner Losh 			cmd[(sz < 0) ? 0 : sz] = '\0';
299ca987d46SWarner Losh 		}
300ca987d46SWarner Losh 		if (*cmd != '\0') {
301ca987d46SWarner Losh 			memcpy(cmdtmp, cmd, sizeof(cmdtmp));
302ca987d46SWarner Losh 			if (parse_cmds(cmdtmp, &dskupdated))
303ca987d46SWarner Losh 				break;
304ca987d46SWarner Losh 			if (dskupdated && gptinit() != 0)
305ca987d46SWarner Losh 				break;
306ca987d46SWarner Losh 			if (!OPT_CHECK(RBX_QUIET))
307ca987d46SWarner Losh 				printf("%s: %s", PATH_CONFIG, cmd);
308ca987d46SWarner Losh 			*cmd = '\0';
309ca987d46SWarner Losh 		}
310ca987d46SWarner Losh 
311ca987d46SWarner Losh 		if (autoboot && keyhit(3)) {
312ca987d46SWarner Losh 			if (*kname == '\0')
313ca987d46SWarner Losh 				memcpy(kname, PATH_LOADER, sizeof(PATH_LOADER));
314ca987d46SWarner Losh 			break;
315ca987d46SWarner Losh 		}
316ca987d46SWarner Losh 		autoboot = 0;
317ca987d46SWarner Losh 
318ca987d46SWarner Losh 		/*
319ca987d46SWarner Losh 		 * Try to exec stage 3 boot loader. If interrupted by a
320ca987d46SWarner Losh 		 * keypress, or in case of failure, try to load a kernel
321ca987d46SWarner Losh 		 * directly instead.
322ca987d46SWarner Losh 		 */
323ca987d46SWarner Losh 		if (*kname != '\0')
324ca987d46SWarner Losh 			load();
325ca987d46SWarner Losh 		memcpy(kname, PATH_LOADER, sizeof(PATH_LOADER));
326ca987d46SWarner Losh 		load();
327ca987d46SWarner Losh 		memcpy(kname, PATH_KERNEL, sizeof(PATH_KERNEL));
328ca987d46SWarner Losh 		load();
329ca987d46SWarner Losh 		gptbootfailed(&dsk);
330ca987d46SWarner Losh 		if (gptfind(&freebsd_ufs_uuid, &dsk, -1) == -1)
331ca987d46SWarner Losh 			break;
332ca987d46SWarner Losh 		dsk_meta = 0;
333ca987d46SWarner Losh 	}
334ca987d46SWarner Losh 
335ca987d46SWarner Losh 	/* Present the user with the boot2 prompt. */
336ca987d46SWarner Losh 
337ca987d46SWarner Losh 	for (;;) {
338ca987d46SWarner Losh 		if (!OPT_CHECK(RBX_QUIET)) {
339ca987d46SWarner Losh 			printf("\nFreeBSD/x86 boot\n"
340ca987d46SWarner Losh 			    "Default: %u:%s(%up%u)%s\n"
341ca987d46SWarner Losh 			    "boot: ",
342ca987d46SWarner Losh 			    dsk.drive & DRV_MASK, dev_nm[dsk.type], dsk.unit,
343ca987d46SWarner Losh 			    dsk.part, kname);
344ca987d46SWarner Losh 		}
345ca987d46SWarner Losh 		if (ioctrl & IO_SERIAL)
346ca987d46SWarner Losh 			sio_flush();
347ca987d46SWarner Losh 		*cmd = '\0';
348ca987d46SWarner Losh 		if (keyhit(0))
349ca987d46SWarner Losh 			getstr(cmd, sizeof(cmd));
350ca987d46SWarner Losh 		else if (!OPT_CHECK(RBX_QUIET))
351ca987d46SWarner Losh 			putchar('\n');
352ca987d46SWarner Losh 		if (parse_cmds(cmd, &dskupdated)) {
353ca987d46SWarner Losh 			putchar('\a');
354ca987d46SWarner Losh 			continue;
355ca987d46SWarner Losh 		}
356ca987d46SWarner Losh 		if (dskupdated && gptinit() != 0)
357ca987d46SWarner Losh 			continue;
358ca987d46SWarner Losh 		load();
359ca987d46SWarner Losh 	}
360ca987d46SWarner Losh 	/* NOTREACHED */
361ca987d46SWarner Losh }
362ca987d46SWarner Losh 
363ca987d46SWarner Losh /* XXX - Needed for btxld to link the boot2 binary; do not remove. */
364ca987d46SWarner Losh void
365ca987d46SWarner Losh exit(int x)
366ca987d46SWarner Losh {
367*86bb84d5SWarner Losh 	while (1);
368ca987d46SWarner Losh }
369ca987d46SWarner Losh 
370ca987d46SWarner Losh static void
371ca987d46SWarner Losh load(void)
372ca987d46SWarner Losh {
373ca987d46SWarner Losh     union {
374ca987d46SWarner Losh 	struct exec ex;
375ca987d46SWarner Losh 	Elf32_Ehdr eh;
376ca987d46SWarner Losh     } hdr;
377ca987d46SWarner Losh     static Elf32_Phdr ep[2];
378ca987d46SWarner Losh     static Elf32_Shdr es[2];
379ca987d46SWarner Losh     caddr_t p;
380ca987d46SWarner Losh     ufs_ino_t ino;
381ca987d46SWarner Losh     uint32_t addr, x;
382ca987d46SWarner Losh     int fmt, i, j;
383ca987d46SWarner Losh 
384ca987d46SWarner Losh     if (!(ino = lookup(kname))) {
385ca987d46SWarner Losh 	if (!ls) {
386ca987d46SWarner Losh 	    printf("%s: No %s on %u:%s(%up%u)\n", BOOTPROG,
387ca987d46SWarner Losh 		kname, dsk.drive & DRV_MASK, dev_nm[dsk.type], dsk.unit,
388ca987d46SWarner Losh 		dsk.part);
389ca987d46SWarner Losh 	}
390ca987d46SWarner Losh 	return;
391ca987d46SWarner Losh     }
392ca987d46SWarner Losh     if (xfsread(ino, &hdr, sizeof(hdr)))
393ca987d46SWarner Losh 	return;
394ca987d46SWarner Losh     if (N_GETMAGIC(hdr.ex) == ZMAGIC)
395ca987d46SWarner Losh 	fmt = 0;
396ca987d46SWarner Losh     else if (IS_ELF(hdr.eh))
397ca987d46SWarner Losh 	fmt = 1;
398ca987d46SWarner Losh     else {
399ca987d46SWarner Losh 	printf("Invalid %s\n", "format");
400ca987d46SWarner Losh 	return;
401ca987d46SWarner Losh     }
402ca987d46SWarner Losh     if (fmt == 0) {
403ca987d46SWarner Losh 	addr = hdr.ex.a_entry & 0xffffff;
404ca987d46SWarner Losh 	p = PTOV(addr);
405ca987d46SWarner Losh 	fs_off = PAGE_SIZE;
406ca987d46SWarner Losh 	if (xfsread(ino, p, hdr.ex.a_text))
407ca987d46SWarner Losh 	    return;
408ca987d46SWarner Losh 	p += roundup2(hdr.ex.a_text, PAGE_SIZE);
409ca987d46SWarner Losh 	if (xfsread(ino, p, hdr.ex.a_data))
410ca987d46SWarner Losh 	    return;
411ca987d46SWarner Losh 	p += hdr.ex.a_data + roundup2(hdr.ex.a_bss, PAGE_SIZE);
412ca987d46SWarner Losh 	bootinfo.bi_symtab = VTOP(p);
413ca987d46SWarner Losh 	memcpy(p, &hdr.ex.a_syms, sizeof(hdr.ex.a_syms));
414ca987d46SWarner Losh 	p += sizeof(hdr.ex.a_syms);
415ca987d46SWarner Losh 	if (hdr.ex.a_syms) {
416ca987d46SWarner Losh 	    if (xfsread(ino, p, hdr.ex.a_syms))
417ca987d46SWarner Losh 		return;
418ca987d46SWarner Losh 	    p += hdr.ex.a_syms;
419ca987d46SWarner Losh 	    if (xfsread(ino, p, sizeof(int)))
420ca987d46SWarner Losh 		return;
421ca987d46SWarner Losh 	    x = *(uint32_t *)p;
422ca987d46SWarner Losh 	    p += sizeof(int);
423ca987d46SWarner Losh 	    x -= sizeof(int);
424ca987d46SWarner Losh 	    if (xfsread(ino, p, x))
425ca987d46SWarner Losh 		return;
426ca987d46SWarner Losh 	    p += x;
427ca987d46SWarner Losh 	}
428ca987d46SWarner Losh     } else {
429ca987d46SWarner Losh 	fs_off = hdr.eh.e_phoff;
430ca987d46SWarner Losh 	for (j = i = 0; i < hdr.eh.e_phnum && j < 2; i++) {
431ca987d46SWarner Losh 	    if (xfsread(ino, ep + j, sizeof(ep[0])))
432ca987d46SWarner Losh 		return;
433ca987d46SWarner Losh 	    if (ep[j].p_type == PT_LOAD)
434ca987d46SWarner Losh 		j++;
435ca987d46SWarner Losh 	}
436ca987d46SWarner Losh 	for (i = 0; i < 2; i++) {
437ca987d46SWarner Losh 	    p = PTOV(ep[i].p_paddr & 0xffffff);
438ca987d46SWarner Losh 	    fs_off = ep[i].p_offset;
439ca987d46SWarner Losh 	    if (xfsread(ino, p, ep[i].p_filesz))
440ca987d46SWarner Losh 		return;
441ca987d46SWarner Losh 	}
442ca987d46SWarner Losh 	p += roundup2(ep[1].p_memsz, PAGE_SIZE);
443ca987d46SWarner Losh 	bootinfo.bi_symtab = VTOP(p);
444ca987d46SWarner Losh 	if (hdr.eh.e_shnum == hdr.eh.e_shstrndx + 3) {
445ca987d46SWarner Losh 	    fs_off = hdr.eh.e_shoff + sizeof(es[0]) *
446ca987d46SWarner Losh 		(hdr.eh.e_shstrndx + 1);
447ca987d46SWarner Losh 	    if (xfsread(ino, &es, sizeof(es)))
448ca987d46SWarner Losh 		return;
449ca987d46SWarner Losh 	    for (i = 0; i < 2; i++) {
450ca987d46SWarner Losh 		memcpy(p, &es[i].sh_size, sizeof(es[i].sh_size));
451ca987d46SWarner Losh 		p += sizeof(es[i].sh_size);
452ca987d46SWarner Losh 		fs_off = es[i].sh_offset;
453ca987d46SWarner Losh 		if (xfsread(ino, p, es[i].sh_size))
454ca987d46SWarner Losh 		    return;
455ca987d46SWarner Losh 		p += es[i].sh_size;
456ca987d46SWarner Losh 	    }
457ca987d46SWarner Losh 	}
458ca987d46SWarner Losh 	addr = hdr.eh.e_entry & 0xffffff;
459ca987d46SWarner Losh     }
460ca987d46SWarner Losh     bootinfo.bi_esymtab = VTOP(p);
461ca987d46SWarner Losh     bootinfo.bi_kernelname = VTOP(kname);
462ca987d46SWarner Losh     bootinfo.bi_bios_dev = dsk.drive;
463ca987d46SWarner Losh #ifdef LOADER_GELI_SUPPORT
464ca987d46SWarner Losh     geliargs.size = sizeof(geliargs);
465ca987d46SWarner Losh     explicit_bzero(gelipw, sizeof(gelipw));
466ca987d46SWarner Losh     gelibuf = malloc(sizeof(struct keybuf) + (GELI_MAX_KEYS * sizeof(struct keybuf_ent)));
467ca987d46SWarner Losh     geli_fill_keybuf(gelibuf);
468ca987d46SWarner Losh     geliargs.notapw = '\0';
469ca987d46SWarner Losh     geliargs.keybuf_sentinel = KEYBUF_SENTINEL;
470ca987d46SWarner Losh     geliargs.keybuf = gelibuf;
471ca987d46SWarner Losh #endif
472ca987d46SWarner Losh     __exec((caddr_t)addr, RB_BOOTINFO | (opts & RBX_MASK),
473ca987d46SWarner Losh 	   MAKEBOOTDEV(dev_maj[dsk.type], dsk.part + 1, dsk.unit, 0xff),
474ca987d46SWarner Losh 	   KARGS_FLAGS_EXTARG, 0, 0, VTOP(&bootinfo)
475ca987d46SWarner Losh #ifdef LOADER_GELI_SUPPORT
476ca987d46SWarner Losh 	   , geliargs
477ca987d46SWarner Losh #endif
478ca987d46SWarner Losh 	   );
479ca987d46SWarner Losh }
480ca987d46SWarner Losh 
481ca987d46SWarner Losh static int
482ca987d46SWarner Losh parse_cmds(char *cmdstr, int *dskupdated)
483ca987d46SWarner Losh {
484ca987d46SWarner Losh     char *arg = cmdstr;
485ca987d46SWarner Losh     char *ep, *p, *q;
486ca987d46SWarner Losh     const char *cp;
487ca987d46SWarner Losh     unsigned int drv;
488ca987d46SWarner Losh     int c, i, j;
489ca987d46SWarner Losh 
490ca987d46SWarner Losh     *dskupdated = 0;
491ca987d46SWarner Losh     while ((c = *arg++)) {
492ca987d46SWarner Losh 	if (c == ' ' || c == '\t' || c == '\n')
493ca987d46SWarner Losh 	    continue;
494ca987d46SWarner Losh 	for (p = arg; *p && *p != '\n' && *p != ' ' && *p != '\t'; p++);
495ca987d46SWarner Losh 	ep = p;
496ca987d46SWarner Losh 	if (*p)
497ca987d46SWarner Losh 	    *p++ = 0;
498ca987d46SWarner Losh 	if (c == '-') {
499ca987d46SWarner Losh 	    while ((c = *arg++)) {
500ca987d46SWarner Losh 		if (c == 'P') {
501ca987d46SWarner Losh 		    if (*(uint8_t *)PTOV(0x496) & 0x10) {
502ca987d46SWarner Losh 			cp = "yes";
503ca987d46SWarner Losh 		    } else {
504ca987d46SWarner Losh 			opts |= OPT_SET(RBX_DUAL) | OPT_SET(RBX_SERIAL);
505ca987d46SWarner Losh 			cp = "no";
506ca987d46SWarner Losh 		    }
507ca987d46SWarner Losh 		    printf("Keyboard: %s\n", cp);
508ca987d46SWarner Losh 		    continue;
509ca987d46SWarner Losh 		} else if (c == 'S') {
510ca987d46SWarner Losh 		    j = 0;
511ca987d46SWarner Losh 		    while ((unsigned int)(i = *arg++ - '0') <= 9)
512ca987d46SWarner Losh 			j = j * 10 + i;
513ca987d46SWarner Losh 		    if (j > 0 && i == -'0') {
514ca987d46SWarner Losh 			comspeed = j;
515ca987d46SWarner Losh 			break;
516ca987d46SWarner Losh 		    }
517ca987d46SWarner Losh 		    /* Fall through to error below ('S' not in optstr[]). */
518ca987d46SWarner Losh 		}
519ca987d46SWarner Losh 		for (i = 0; c != optstr[i]; i++)
520ca987d46SWarner Losh 		    if (i == NOPT - 1)
521ca987d46SWarner Losh 			return -1;
522ca987d46SWarner Losh 		opts ^= OPT_SET(flags[i]);
523ca987d46SWarner Losh 	    }
524ca987d46SWarner Losh 	    ioctrl = OPT_CHECK(RBX_DUAL) ? (IO_SERIAL|IO_KEYBOARD) :
525ca987d46SWarner Losh 		     OPT_CHECK(RBX_SERIAL) ? IO_SERIAL : IO_KEYBOARD;
526ca987d46SWarner Losh 	    if (ioctrl & IO_SERIAL) {
527ca987d46SWarner Losh 	        if (sio_init(115200 / comspeed) != 0)
528ca987d46SWarner Losh 		    ioctrl &= ~IO_SERIAL;
529ca987d46SWarner Losh 	    }
530ca987d46SWarner Losh 	} else {
531ca987d46SWarner Losh 	    for (q = arg--; *q && *q != '('; q++);
532ca987d46SWarner Losh 	    if (*q) {
533ca987d46SWarner Losh 		drv = -1;
534ca987d46SWarner Losh 		if (arg[1] == ':') {
535ca987d46SWarner Losh 		    drv = *arg - '0';
536ca987d46SWarner Losh 		    if (drv > 9)
537ca987d46SWarner Losh 			return (-1);
538ca987d46SWarner Losh 		    arg += 2;
539ca987d46SWarner Losh 		}
540ca987d46SWarner Losh 		if (q - arg != 2)
541ca987d46SWarner Losh 		    return -1;
542ca987d46SWarner Losh 		for (i = 0; arg[0] != dev_nm[i][0] ||
543ca987d46SWarner Losh 			    arg[1] != dev_nm[i][1]; i++)
544ca987d46SWarner Losh 		    if (i == NDEV - 1)
545ca987d46SWarner Losh 			return -1;
546ca987d46SWarner Losh 		dsk.type = i;
547ca987d46SWarner Losh 		arg += 3;
548ca987d46SWarner Losh 		dsk.unit = *arg - '0';
549ca987d46SWarner Losh 		if (arg[1] != 'p' || dsk.unit > 9)
550ca987d46SWarner Losh 		    return -1;
551ca987d46SWarner Losh 		arg += 2;
552ca987d46SWarner Losh 		dsk.part = *arg - '0';
553ca987d46SWarner Losh 		if (dsk.part < 1 || dsk.part > 9)
554ca987d46SWarner Losh 		    return -1;
555ca987d46SWarner Losh 		arg++;
556ca987d46SWarner Losh 		if (arg[0] != ')')
557ca987d46SWarner Losh 		    return -1;
558ca987d46SWarner Losh 		arg++;
559ca987d46SWarner Losh 		if (drv == -1)
560ca987d46SWarner Losh 		    drv = dsk.unit;
561ca987d46SWarner Losh 		dsk.drive = (dsk.type <= TYPE_MAXHARD
562ca987d46SWarner Losh 			     ? DRV_HARD : 0) + drv;
563ca987d46SWarner Losh 		*dskupdated = 1;
564ca987d46SWarner Losh 	    }
565ca987d46SWarner Losh 	    if ((i = ep - arg)) {
566ca987d46SWarner Losh 		if ((size_t)i >= sizeof(kname))
567ca987d46SWarner Losh 		    return -1;
568ca987d46SWarner Losh 		memcpy(kname, arg, i + 1);
569ca987d46SWarner Losh 	    }
570ca987d46SWarner Losh 	}
571ca987d46SWarner Losh 	arg = p;
572ca987d46SWarner Losh     }
573ca987d46SWarner Losh     return 0;
574ca987d46SWarner Losh }
575ca987d46SWarner Losh 
576ca987d46SWarner Losh static int
577ca987d46SWarner Losh dskread(void *buf, daddr_t lba, unsigned nblk)
578ca987d46SWarner Losh {
579ca987d46SWarner Losh 	int err;
580ca987d46SWarner Losh 
581ca987d46SWarner Losh 	err = drvread(&dsk, buf, lba + dsk.start, nblk);
582ca987d46SWarner Losh 
583ca987d46SWarner Losh #ifdef LOADER_GELI_SUPPORT
584ca987d46SWarner Losh 	if (err == 0 && is_geli(&dsk) == 0) {
585ca987d46SWarner Losh 		/* Decrypt */
586ca987d46SWarner Losh 		if (geli_read(&dsk, lba * DEV_BSIZE, buf, nblk * DEV_BSIZE))
587ca987d46SWarner Losh 			return (err);
588ca987d46SWarner Losh 	}
589ca987d46SWarner Losh #endif
590ca987d46SWarner Losh 
591ca987d46SWarner Losh 	return (err);
592ca987d46SWarner Losh }
593ca987d46SWarner Losh 
594ca987d46SWarner Losh #ifdef LOADER_GELI_SUPPORT
595ca987d46SWarner Losh /*
596ca987d46SWarner Losh  * Read function compartible with the ZFS callback, required to keep the GELI
597ca987d46SWarner Losh  * Implementation the same for both UFS and ZFS
598ca987d46SWarner Losh  */
599ca987d46SWarner Losh static int
600ca987d46SWarner Losh vdev_read(void *vdev __unused, void *priv, off_t off, void *buf, size_t bytes)
601ca987d46SWarner Losh {
602ca987d46SWarner Losh 	char *p;
603ca987d46SWarner Losh 	daddr_t lba;
604ca987d46SWarner Losh 	unsigned int nb;
605ca987d46SWarner Losh 	struct dsk *dskp = (struct dsk *) priv;
606ca987d46SWarner Losh 
607ca987d46SWarner Losh 	if ((off & (DEV_BSIZE - 1)) || (bytes & (DEV_BSIZE - 1)))
608ca987d46SWarner Losh 		return (-1);
609ca987d46SWarner Losh 
610ca987d46SWarner Losh 	p = buf;
611ca987d46SWarner Losh 	lba = off / DEV_BSIZE;
612ca987d46SWarner Losh 	lba += dskp->start;
613ca987d46SWarner Losh 
614ca987d46SWarner Losh 	while (bytes > 0) {
615ca987d46SWarner Losh 		nb = bytes / DEV_BSIZE;
616ca987d46SWarner Losh 		if (nb > VBLKSIZE / DEV_BSIZE)
617ca987d46SWarner Losh 			nb = VBLKSIZE / DEV_BSIZE;
618ca987d46SWarner Losh 		if (drvread(dskp, dmadat->blkbuf, lba, nb))
619ca987d46SWarner Losh 			return (-1);
620ca987d46SWarner Losh 		memcpy(p, dmadat->blkbuf, nb * DEV_BSIZE);
621ca987d46SWarner Losh 		p += nb * DEV_BSIZE;
622ca987d46SWarner Losh 		lba += nb;
623ca987d46SWarner Losh 		bytes -= nb * DEV_BSIZE;
624ca987d46SWarner Losh 	}
625ca987d46SWarner Losh 
626ca987d46SWarner Losh 	return (0);
627ca987d46SWarner Losh }
628ca987d46SWarner Losh #endif /* LOADER_GELI_SUPPORT */
629