xref: /freebsd/usr.sbin/efibootmgr/efibootmgr.c (revision 83c4237258d3574b28a63c9182b7d95c428cd8ef)
11285bcc8SWarner Losh /*-
29ed08029SRebecca Cran  * Copyright (c) 2017-2018 Netflix, Inc.
31285bcc8SWarner Losh  *
41285bcc8SWarner Losh  * Redistribution and use in source and binary forms, with or without
51285bcc8SWarner Losh  * modification, are permitted provided that the following conditions
61285bcc8SWarner Losh  * are met:
71285bcc8SWarner Losh  * 1. Redistributions of source code must retain the above copyright
81285bcc8SWarner Losh  *    notice, this list of conditions and the following disclaimer
91285bcc8SWarner Losh  *    in this position and unchanged.
101285bcc8SWarner Losh  * 2. Redistributions in binary form must reproduce the above copyright
111285bcc8SWarner Losh  *    notice, this list of conditions and the following disclaimer in the
121285bcc8SWarner Losh  *    documentation and/or other materials provided with the distribution.
131285bcc8SWarner Losh  *
141285bcc8SWarner Losh  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
151285bcc8SWarner Losh  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
161285bcc8SWarner Losh  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
171285bcc8SWarner Losh  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
181285bcc8SWarner Losh  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
191285bcc8SWarner Losh  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
201285bcc8SWarner Losh  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
211285bcc8SWarner Losh  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
221285bcc8SWarner Losh  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
231285bcc8SWarner Losh  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
241285bcc8SWarner Losh  */
251285bcc8SWarner Losh 
261285bcc8SWarner Losh #include <sys/cdefs.h>
271285bcc8SWarner Losh __FBSDID("$FreeBSD$");
281285bcc8SWarner Losh 
291285bcc8SWarner Losh #include <sys/stat.h>
301285bcc8SWarner Losh #include <sys/vtoc.h>
311285bcc8SWarner Losh #include <sys/param.h>
321285bcc8SWarner Losh #include <assert.h>
331285bcc8SWarner Losh #include <ctype.h>
341285bcc8SWarner Losh #include <err.h>
351285bcc8SWarner Losh #include <errno.h>
361285bcc8SWarner Losh #include <fcntl.h>
371285bcc8SWarner Losh #include <libgeom.h>
381285bcc8SWarner Losh #include <paths.h>
391285bcc8SWarner Losh #include <signal.h>
401285bcc8SWarner Losh #include <stdint.h>
411285bcc8SWarner Losh #include <stdio.h>
421285bcc8SWarner Losh #include <stdlib.h>
431285bcc8SWarner Losh #include <getopt.h>
441285bcc8SWarner Losh #include <limits.h>
451285bcc8SWarner Losh #include <inttypes.h>
461285bcc8SWarner Losh #include <stdbool.h>
471285bcc8SWarner Losh #include <string.h>
481285bcc8SWarner Losh #include <strings.h>
491285bcc8SWarner Losh #include <unistd.h>
501285bcc8SWarner Losh #include <libgeom.h>
511285bcc8SWarner Losh #include <geom/geom.h>
521285bcc8SWarner Losh #include <geom/geom_ctl.h>
531285bcc8SWarner Losh #include <geom/geom_int.h>
541285bcc8SWarner Losh 
551285bcc8SWarner Losh #include <efivar.h>
561285bcc8SWarner Losh #include <efiutil.h>
571285bcc8SWarner Losh #include <efichar.h>
581285bcc8SWarner Losh #include <efivar-dp.h>
591285bcc8SWarner Losh 
601285bcc8SWarner Losh #ifndef LOAD_OPTION_ACTIVE
611285bcc8SWarner Losh #define LOAD_OPTION_ACTIVE 0x00000001
621285bcc8SWarner Losh #endif
631285bcc8SWarner Losh 
641285bcc8SWarner Losh #ifndef LOAD_OPTION_CATEGORY_BOOT
651285bcc8SWarner Losh #define LOAD_OPTION_CATEGORY_BOOT 0x00000000
661285bcc8SWarner Losh #endif
671285bcc8SWarner Losh 
681285bcc8SWarner Losh #define BAD_LENGTH	((size_t)-1)
691285bcc8SWarner Losh 
70*83c42372SD Scott Phillips #define EFI_OS_INDICATIONS_BOOT_TO_FW_UI 0x0000000000000001
71*83c42372SD Scott Phillips 
721285bcc8SWarner Losh typedef struct _bmgr_opts {
731285bcc8SWarner Losh 	char	*env;
741285bcc8SWarner Losh 	char	*loader;
751285bcc8SWarner Losh 	char	*label;
761285bcc8SWarner Losh 	char	*kernel;
771285bcc8SWarner Losh 	char	*name;
781285bcc8SWarner Losh 	char	*order;
791285bcc8SWarner Losh 	int     bootnum;
801285bcc8SWarner Losh 	bool	copy;
811285bcc8SWarner Losh 	bool    create;
821285bcc8SWarner Losh 	bool    delete;
831285bcc8SWarner Losh 	bool    delete_bootnext;
841285bcc8SWarner Losh 	bool    del_timeout;
851285bcc8SWarner Losh 	bool    dry_run;
861cdb8eb8SWarner Losh 	bool	device_path;
871cdb8eb8SWarner Losh 	bool	esp_device;
88*83c42372SD Scott Phillips 	bool    fw_ui;
89*83c42372SD Scott Phillips 	bool    no_fw_ui;
90de26ba4dSWarner Losh 	bool	has_bootnum;
911285bcc8SWarner Losh 	bool    once;
921285bcc8SWarner Losh 	int	cp_src;
931285bcc8SWarner Losh 	bool    set_active;
941285bcc8SWarner Losh 	bool    set_bootnext;
951285bcc8SWarner Losh 	bool    set_inactive;
961285bcc8SWarner Losh 	bool    set_timeout;
971285bcc8SWarner Losh 	int     timeout;
981cdb8eb8SWarner Losh 	bool	unix_path;
991285bcc8SWarner Losh 	bool    verbose;
1001285bcc8SWarner Losh } bmgr_opts_t;
1011285bcc8SWarner Losh 
1021285bcc8SWarner Losh static struct option lopts[] = {
1039ed08029SRebecca Cran 	{"activate", no_argument, NULL, 'a'},
1049ed08029SRebecca Cran 	{"bootnext", no_argument, NULL, 'n'}, /* set bootnext */
105a2581e80SRebecca Cran 	{"bootnum", required_argument, NULL, 'b'},
1061285bcc8SWarner Losh 	{"bootorder", required_argument, NULL, 'o'}, /* set order */
1071285bcc8SWarner Losh 	{"copy", required_argument, NULL, 'C'},		/* Copy boot method */
1081285bcc8SWarner Losh 	{"create", no_argument, NULL, 'c'},
1099ed08029SRebecca Cran 	{"deactivate", no_argument, NULL, 'A'},
1106b8b1bfeSToomas Soome 	{"del-timeout", no_argument, NULL, 'T'},
1119ed08029SRebecca Cran 	{"delete", no_argument, NULL, 'B'},
1129ed08029SRebecca Cran 	{"delete-bootnext", no_argument, NULL, 'N'},
1131cdb8eb8SWarner Losh 	{"device-path", no_argument, NULL, 'd'},
1141285bcc8SWarner Losh 	{"dry-run", no_argument, NULL, 'D'},
1151285bcc8SWarner Losh 	{"env", required_argument, NULL, 'e'},
1161cdb8eb8SWarner Losh 	{"esp", no_argument, NULL, 'E'},
117*83c42372SD Scott Phillips 	{"fw-ui", no_argument, NULL, 'f'},
118*83c42372SD Scott Phillips 	{"no-fw-ui", no_argument, NULL, 'F'},
1191285bcc8SWarner Losh 	{"help", no_argument, NULL, 'h'},
1201285bcc8SWarner Losh 	{"kernel", required_argument, NULL, 'k'},
1211285bcc8SWarner Losh 	{"label", required_argument, NULL, 'L'},
1221285bcc8SWarner Losh 	{"loader", required_argument, NULL, 'l'},
1231285bcc8SWarner Losh 	{"once", no_argument, NULL, 'O'},
1241285bcc8SWarner Losh 	{"set-timeout", required_argument, NULL, 't'},
1251cdb8eb8SWarner Losh 	{"unix-path", no_argument, NULL, 'p'},
1261285bcc8SWarner Losh 	{"verbose", no_argument, NULL, 'v'},
1271285bcc8SWarner Losh 	{ NULL, 0, NULL, 0}
1281285bcc8SWarner Losh };
1291285bcc8SWarner Losh 
1301285bcc8SWarner Losh /* global efibootmgr opts */
1311285bcc8SWarner Losh static bmgr_opts_t opts;
1321285bcc8SWarner Losh 
1331285bcc8SWarner Losh static LIST_HEAD(efivars_head, entry) efivars =
1341285bcc8SWarner Losh 	LIST_HEAD_INITIALIZER(efivars);
1351285bcc8SWarner Losh 
1361285bcc8SWarner Losh struct entry {
1371285bcc8SWarner Losh 	efi_guid_t	guid;
1381285bcc8SWarner Losh 	uint32_t	attrs;
1391285bcc8SWarner Losh 	uint8_t		*data;
1401285bcc8SWarner Losh 	size_t		size;
1411285bcc8SWarner Losh 	char		*name;
1421285bcc8SWarner Losh 	char		*label;
1431285bcc8SWarner Losh 	int		idx;
14451922c69SWarner Losh 	int		flags;
14551922c69SWarner Losh #define SEEN	1
1461285bcc8SWarner Losh 
1471285bcc8SWarner Losh 	LIST_ENTRY(entry) entries;
1481285bcc8SWarner Losh };
1491285bcc8SWarner Losh 
1501285bcc8SWarner Losh #define MAX_DP_LEN	4096
1511285bcc8SWarner Losh #define MAX_LOADOPT_LEN	8192
1521285bcc8SWarner Losh 
1531285bcc8SWarner Losh 
1541285bcc8SWarner Losh static char *
1551285bcc8SWarner Losh mangle_loader(char *loader)
1561285bcc8SWarner Losh {
1571285bcc8SWarner Losh 	char *c;
1581285bcc8SWarner Losh 
1591285bcc8SWarner Losh 	for (c = loader; *c; c++)
1601285bcc8SWarner Losh 		if (*c == '/')
1611285bcc8SWarner Losh 			*c = '\\';
1621285bcc8SWarner Losh 
1631285bcc8SWarner Losh 	return loader;
1641285bcc8SWarner Losh }
1651285bcc8SWarner Losh 
1661285bcc8SWarner Losh 
1671285bcc8SWarner Losh #define COMMON_ATTRS EFI_VARIABLE_NON_VOLATILE | \
1681285bcc8SWarner Losh 	EFI_VARIABLE_BOOTSERVICE_ACCESS | \
1691285bcc8SWarner Losh 	EFI_VARIABLE_RUNTIME_ACCESS
1701285bcc8SWarner Losh 
1711285bcc8SWarner Losh /*
1721285bcc8SWarner Losh  * We use global guid, and common var attrs and
1731285bcc8SWarner Losh  * find it better to just delete and re-create a var.
1741285bcc8SWarner Losh  */
1751285bcc8SWarner Losh static int
1761285bcc8SWarner Losh set_bootvar(const char *name, uint8_t *data, size_t size)
1771285bcc8SWarner Losh {
1781285bcc8SWarner Losh 
1791285bcc8SWarner Losh 	return efi_set_variable(EFI_GLOBAL_GUID, name, data, size,
1801285bcc8SWarner Losh 	    COMMON_ATTRS);
1811285bcc8SWarner Losh }
1821285bcc8SWarner Losh 
1831285bcc8SWarner Losh 
1841285bcc8SWarner Losh #define USAGE \
1859ed08029SRebecca Cran 	"   [-aAnB -b bootnum] [-N] [-t timeout] [-T] [-o bootorder] [-O] [--verbose] [--help]\n\
1869ed08029SRebecca Cran   [-c -l loader [-k kernel] [-L label] [--dry-run] [-b bootnum]]"
1871285bcc8SWarner Losh 
1881285bcc8SWarner Losh #define CREATE_USAGE \
1899ed08029SRebecca Cran 	"       efibootmgr -c -l loader [-k kernel] [-L label] [--dry-run] [-b bootnum] [-a]"
1901285bcc8SWarner Losh #define ORDER_USAGE \
1911285bcc8SWarner Losh 	"       efibootmgr -o bootvarnum1,bootvarnum2,..."
1921285bcc8SWarner Losh #define TIMEOUT_USAGE \
1931285bcc8SWarner Losh 	"       efibootmgr -t seconds"
1941285bcc8SWarner Losh #define DELETE_USAGE \
1959ed08029SRebecca Cran 	"       efibootmgr -B -b bootnum"
1961285bcc8SWarner Losh #define ACTIVE_USAGE \
1979ed08029SRebecca Cran 	"       efibootmgr [-a | -A] -b bootnum"
1981285bcc8SWarner Losh #define BOOTNEXT_USAGE \
1999ed08029SRebecca Cran 	"       efibootmgr [-n | -N] -b bootnum"
2001285bcc8SWarner Losh 
2011285bcc8SWarner Losh static void
2021285bcc8SWarner Losh parse_args(int argc, char *argv[])
2031285bcc8SWarner Losh {
2041285bcc8SWarner Losh 	int ch;
2051285bcc8SWarner Losh 
206*83c42372SD Scott Phillips 	while ((ch = getopt_long(argc, argv, "AaBb:C:cdDe:EFfhk:L:l:NnOo:pTt:v",
2071285bcc8SWarner Losh 		    lopts, NULL)) != -1) {
2081285bcc8SWarner Losh 		switch (ch) {
2091285bcc8SWarner Losh 		case 'A':
2101285bcc8SWarner Losh 			opts.set_inactive = true;
2111285bcc8SWarner Losh 			break;
2121285bcc8SWarner Losh 		case 'a':
2131285bcc8SWarner Losh 			opts.set_active = true;
2141285bcc8SWarner Losh 			break;
215de26ba4dSWarner Losh 		case 'b':
216de26ba4dSWarner Losh 			opts.has_bootnum = true;
217de26ba4dSWarner Losh 			opts.bootnum = strtoul(optarg, NULL, 16);
218de26ba4dSWarner Losh 			break;
2191285bcc8SWarner Losh 		case 'B':
2201285bcc8SWarner Losh 			opts.delete = true;
2211285bcc8SWarner Losh 			break;
2221285bcc8SWarner Losh 		case 'C':
2231285bcc8SWarner Losh 			opts.copy = true;
2241285bcc8SWarner Losh 			opts.cp_src = strtoul(optarg, NULL, 16);
2251285bcc8SWarner Losh 		case 'c':
2261285bcc8SWarner Losh 			opts.create = true;
2271285bcc8SWarner Losh 			break;
2281285bcc8SWarner Losh 		case 'D': /* should be remove dups XXX */
2291285bcc8SWarner Losh 			opts.dry_run = true;
2301285bcc8SWarner Losh 			break;
2311cdb8eb8SWarner Losh 		case 'd':
2321cdb8eb8SWarner Losh 			opts.device_path = true;
2331cdb8eb8SWarner Losh 			break;
2341285bcc8SWarner Losh 		case 'e':
23563b56694SWarner Losh 			free(opts.env);
2361285bcc8SWarner Losh 			opts.env = strdup(optarg);
2371285bcc8SWarner Losh 			break;
2381cdb8eb8SWarner Losh 		case 'E':
2391cdb8eb8SWarner Losh 			opts.esp_device = true;
2401cdb8eb8SWarner Losh 			break;
241*83c42372SD Scott Phillips 		case 'F':
242*83c42372SD Scott Phillips 			opts.no_fw_ui = true;
243*83c42372SD Scott Phillips 			break;
244*83c42372SD Scott Phillips 		case 'f':
245*83c42372SD Scott Phillips 			opts.fw_ui = true;
246*83c42372SD Scott Phillips 			break;
2471285bcc8SWarner Losh 		case 'h':
2481285bcc8SWarner Losh 		default:
2491285bcc8SWarner Losh 			errx(1, "%s", USAGE);
2501285bcc8SWarner Losh 			break;
2511285bcc8SWarner Losh 		case 'k':
25263b56694SWarner Losh 			free(opts.kernel);
2531285bcc8SWarner Losh 			opts.kernel = strdup(optarg);
2541285bcc8SWarner Losh 			break;
2551285bcc8SWarner Losh 		case 'L':
25663b56694SWarner Losh 			free(opts.label);
2571285bcc8SWarner Losh 			opts.label = strdup(optarg);
2581285bcc8SWarner Losh 			break;
2591285bcc8SWarner Losh 		case 'l':
26063b56694SWarner Losh 			free(opts.loader);
2611285bcc8SWarner Losh 			opts.loader = strdup(optarg);
2621285bcc8SWarner Losh 			opts.loader = mangle_loader(opts.loader);
2631285bcc8SWarner Losh 			break;
2641285bcc8SWarner Losh 		case 'N':
2651285bcc8SWarner Losh 			opts.delete_bootnext = true;
2661285bcc8SWarner Losh 			break;
2671285bcc8SWarner Losh 		case 'n':
2681285bcc8SWarner Losh 			opts.set_bootnext = true;
2691285bcc8SWarner Losh 			break;
2701285bcc8SWarner Losh 		case 'O':
2711285bcc8SWarner Losh 			opts.once = true;
2721285bcc8SWarner Losh 			break;
2731285bcc8SWarner Losh 		case 'o':
27463b56694SWarner Losh 			free(opts.order);
2751285bcc8SWarner Losh 			opts.order = strdup(optarg);
2761285bcc8SWarner Losh 			break;
2771cdb8eb8SWarner Losh 		case 'p':
2781cdb8eb8SWarner Losh 			opts.unix_path = true;
2791cdb8eb8SWarner Losh 			break;
2801285bcc8SWarner Losh 		case 'T':
2811285bcc8SWarner Losh 			opts.del_timeout = true;
2821285bcc8SWarner Losh 			break;
2831285bcc8SWarner Losh 		case 't':
2841285bcc8SWarner Losh 			opts.set_timeout = true;
2851285bcc8SWarner Losh 			opts.timeout = strtoul(optarg, NULL, 10);
2861285bcc8SWarner Losh 			break;
2871285bcc8SWarner Losh 		case 'v':
2881285bcc8SWarner Losh 			opts.verbose = true;
2891285bcc8SWarner Losh 			break;
2901285bcc8SWarner Losh 		}
2911285bcc8SWarner Losh 	}
2921285bcc8SWarner Losh 	if (opts.create) {
2931127aea3SWarner Losh 		if (!opts.loader)
2941285bcc8SWarner Losh 			errx(1, "%s",CREATE_USAGE);
2951285bcc8SWarner Losh 		return;
2961285bcc8SWarner Losh 	}
2971285bcc8SWarner Losh 
2981285bcc8SWarner Losh 	if (opts.order && !(opts.order))
2991285bcc8SWarner Losh 		errx(1, "%s", ORDER_USAGE);
3009ed08029SRebecca Cran 
3019ed08029SRebecca Cran 	if ((opts.set_inactive || opts.set_active) && !opts.has_bootnum)
3029ed08029SRebecca Cran 		errx(1, "%s", ACTIVE_USAGE);
3039ed08029SRebecca Cran 
3049ed08029SRebecca Cran 	if (opts.delete && !opts.has_bootnum)
3059ed08029SRebecca Cran 		errx(1, "%s", DELETE_USAGE);
3069ed08029SRebecca Cran 
3079ed08029SRebecca Cran 	if (opts.set_bootnext && !opts.has_bootnum)
3089ed08029SRebecca Cran 		errx(1, "%s", BOOTNEXT_USAGE);
3091285bcc8SWarner Losh }
3101285bcc8SWarner Losh 
3111285bcc8SWarner Losh 
3121285bcc8SWarner Losh static void
3131285bcc8SWarner Losh read_vars(void)
3141285bcc8SWarner Losh {
3151285bcc8SWarner Losh 
3161285bcc8SWarner Losh 	efi_guid_t *guid;
3171285bcc8SWarner Losh 	char *next_name = NULL;
3181285bcc8SWarner Losh 	int ret = 0;
3191285bcc8SWarner Losh 
3201285bcc8SWarner Losh 	struct entry *nent;
3211285bcc8SWarner Losh 
3221285bcc8SWarner Losh 	LIST_INIT(&efivars);
3231285bcc8SWarner Losh 	while ((ret = efi_get_next_variable_name(&guid, &next_name)) > 0) {
3241285bcc8SWarner Losh 		/*
325a3e6c4a6SWarner Losh 		 * Only pay attention to EFI:BootXXXX variables to get the list.
3261285bcc8SWarner Losh 		 */
3271285bcc8SWarner Losh 		if (efi_guid_cmp(guid, &EFI_GLOBAL_GUID) != 0 ||
3281285bcc8SWarner Losh 		    strlen(next_name) != 8 ||
3291285bcc8SWarner Losh 		    strncmp(next_name, "Boot", 4) != 0 ||
3301285bcc8SWarner Losh 		    !isxdigit(next_name[4]) ||
3311285bcc8SWarner Losh 		    !isxdigit(next_name[5]) ||
3321285bcc8SWarner Losh 		    !isxdigit(next_name[6]) ||
3331285bcc8SWarner Losh 		    !isxdigit(next_name[7]))
3341285bcc8SWarner Losh 			continue;
3351285bcc8SWarner Losh 		nent = malloc(sizeof(struct entry));
3361285bcc8SWarner Losh 		nent->name = strdup(next_name);
3371285bcc8SWarner Losh 
3381285bcc8SWarner Losh 		ret = efi_get_variable(*guid, next_name, &nent->data,
3391285bcc8SWarner Losh 		    &nent->size, &nent->attrs);
3401285bcc8SWarner Losh 		if (ret < 0)
3411285bcc8SWarner Losh 			err(1, "efi_get_variable");
3421285bcc8SWarner Losh 		nent->guid = *guid;
3431285bcc8SWarner Losh 		nent->idx = strtoul(&next_name[4], NULL, 16);
3441285bcc8SWarner Losh 		LIST_INSERT_HEAD(&efivars, nent, entries);
3451285bcc8SWarner Losh 	}
3461285bcc8SWarner Losh }
3471285bcc8SWarner Losh 
3481285bcc8SWarner Losh 
3491285bcc8SWarner Losh static void
3501285bcc8SWarner Losh set_boot_order(char *order)
3511285bcc8SWarner Losh {
3521285bcc8SWarner Losh 	uint16_t *new_data;
3531285bcc8SWarner Losh 	size_t size;
3541285bcc8SWarner Losh 	char *next, *cp;
3551285bcc8SWarner Losh 	int cnt;
3561285bcc8SWarner Losh 	int i;
3571285bcc8SWarner Losh 
3581285bcc8SWarner Losh 	cp = order;
3591285bcc8SWarner Losh 	cnt = 1;
3601285bcc8SWarner Losh 	while (*cp) {
3611285bcc8SWarner Losh 		if (*cp++ == ',')
3621285bcc8SWarner Losh 			cnt++;
3631285bcc8SWarner Losh 	}
3641285bcc8SWarner Losh 	size = sizeof(uint16_t) * cnt;
3651285bcc8SWarner Losh 	new_data = malloc(size);
3661285bcc8SWarner Losh 
3671285bcc8SWarner Losh 	i = 0;
3681285bcc8SWarner Losh 	cp = strdup(order);
3691285bcc8SWarner Losh 	while ((next = strsep(&cp, ",")) != NULL) {
3701285bcc8SWarner Losh 		new_data[i] = strtoul(next, NULL, 16);
3711285bcc8SWarner Losh 		if (new_data[i] == 0 && errno == EINVAL) {
3721285bcc8SWarner Losh 			warnx("can't parse %s as a numb", next);
3731285bcc8SWarner Losh 			errx(1, "%s", ORDER_USAGE);
3741285bcc8SWarner Losh 		}
3751285bcc8SWarner Losh 		i++;
3761285bcc8SWarner Losh 	}
3771285bcc8SWarner Losh 	free(cp);
3781285bcc8SWarner Losh 	if (set_bootvar("BootOrder", (uint8_t*)new_data, size) < 0)
3791285bcc8SWarner Losh 		err(1, "Unabke to set BootOrder to %s", order);
3803af42ca7SWarner Losh 	free(new_data);
3811285bcc8SWarner Losh }
3821285bcc8SWarner Losh 
3831285bcc8SWarner Losh static void
3841285bcc8SWarner Losh handle_activity(int bootnum, bool active)
3851285bcc8SWarner Losh {
3861285bcc8SWarner Losh 	uint32_t attrs, load_attrs;
3871285bcc8SWarner Losh 	uint8_t *data;
3881285bcc8SWarner Losh 	size_t size;
3891285bcc8SWarner Losh 	char *name;
3901285bcc8SWarner Losh 
3911285bcc8SWarner Losh 	asprintf(&name, "%s%04X", "Boot", bootnum);
3921285bcc8SWarner Losh 	if (name == NULL)
3931285bcc8SWarner Losh 		err(1, "asprintf");
3941285bcc8SWarner Losh 	if (efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, &attrs) < 0)
3951285bcc8SWarner Losh 		err(1, "No such bootvar %s\n", name);
3961285bcc8SWarner Losh 
3971285bcc8SWarner Losh 	load_attrs = le32dec(data);
3981285bcc8SWarner Losh 
3991285bcc8SWarner Losh 	if (active)
4001285bcc8SWarner Losh 		load_attrs |= LOAD_OPTION_ACTIVE;
4011285bcc8SWarner Losh 	else
4021285bcc8SWarner Losh 		load_attrs &= ~LOAD_OPTION_ACTIVE;
4031285bcc8SWarner Losh 
4041285bcc8SWarner Losh 	le32enc(data, load_attrs);
4051285bcc8SWarner Losh 
4061285bcc8SWarner Losh 	if (set_bootvar(name, data, size) < 0)
4071285bcc8SWarner Losh 		err(1, "handle activity efi_set_variable");
4081285bcc8SWarner Losh }
4091285bcc8SWarner Losh 
4101285bcc8SWarner Losh 
4111285bcc8SWarner Losh /*
4121285bcc8SWarner Losh  * add boot var to boot order.
4131285bcc8SWarner Losh  * called by create boot var. There is no option
4141285bcc8SWarner Losh  * to add one independent of create.
4151285bcc8SWarner Losh  *
4161285bcc8SWarner Losh  * Note: we currently don't support where it goes
4171285bcc8SWarner Losh  * so it goes on the front, inactive.
4181285bcc8SWarner Losh  * use -o 2,3,7 etc to affect order, -a to activate.
4191285bcc8SWarner Losh  */
4201285bcc8SWarner Losh static void
4211285bcc8SWarner Losh add_to_boot_order(char *bootvar)
4221285bcc8SWarner Losh {
4231285bcc8SWarner Losh 	size_t size;
4241285bcc8SWarner Losh 	uint32_t attrs;
4251285bcc8SWarner Losh 	uint16_t val;
4261285bcc8SWarner Losh 	uint8_t *data, *new;
4271285bcc8SWarner Losh 
4281285bcc8SWarner Losh 	val = strtoul(&bootvar[4], NULL, 16);
4291285bcc8SWarner Losh 
4301285bcc8SWarner Losh 	if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) < 0) {
4311285bcc8SWarner Losh 		if (errno == ENOENT) { /* create it and set this bootvar to active */
4321285bcc8SWarner Losh 			size = 0;
4331285bcc8SWarner Losh 			data = NULL;
4341285bcc8SWarner Losh 		} else
4351285bcc8SWarner Losh 			err(1, "efi_get_variable BootOrder");
4361285bcc8SWarner Losh 	}
4371285bcc8SWarner Losh 
4381285bcc8SWarner Losh 	/*
4391285bcc8SWarner Losh 	 * We have BootOrder with the current order
4401285bcc8SWarner Losh 	 * so grow the array by one, add the value
4411285bcc8SWarner Losh 	 * and write the new variable value.
4421285bcc8SWarner Losh 	 */
4431285bcc8SWarner Losh 	size += sizeof(uint16_t);
4441285bcc8SWarner Losh 	new = malloc(size);
4451285bcc8SWarner Losh 	if (!new)
4461285bcc8SWarner Losh 		err(1, "malloc");
4471285bcc8SWarner Losh 
4481285bcc8SWarner Losh 	le16enc(new, val);
4491285bcc8SWarner Losh 	if (size > sizeof(uint16_t))
4501285bcc8SWarner Losh 		memcpy(new + sizeof(uint16_t), data, size - sizeof(uint16_t));
4511285bcc8SWarner Losh 
4521285bcc8SWarner Losh 	if (set_bootvar("BootOrder", new, size) < 0)
4531285bcc8SWarner Losh 		err(1, "set_bootvar");
4541285bcc8SWarner Losh 	free(new);
4551285bcc8SWarner Losh }
4561285bcc8SWarner Losh 
4571285bcc8SWarner Losh 
4581285bcc8SWarner Losh static void
4591285bcc8SWarner Losh remove_from_order(uint16_t bootnum)
4601285bcc8SWarner Losh {
4611285bcc8SWarner Losh 	uint32_t attrs;
4621285bcc8SWarner Losh 	size_t size, i, j;
4631285bcc8SWarner Losh 	uint8_t *new, *data;
4641285bcc8SWarner Losh 
4651285bcc8SWarner Losh 	if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) < 0)
4661285bcc8SWarner Losh 		return;
4671285bcc8SWarner Losh 
4681285bcc8SWarner Losh 	new = malloc(size);
4691285bcc8SWarner Losh 	if (new == NULL)
4701285bcc8SWarner Losh 		err(1, "malloc");
4711285bcc8SWarner Losh 
4721285bcc8SWarner Losh 	for (j = i = 0; i < size; i += sizeof(uint16_t)) {
4731285bcc8SWarner Losh 		if (le16dec(data + i) == bootnum)
4741285bcc8SWarner Losh 			continue;
4751285bcc8SWarner Losh 		memcpy(new + j, data + i, sizeof(uint16_t));
4761285bcc8SWarner Losh 		j += sizeof(uint16_t);
4771285bcc8SWarner Losh 	}
4781285bcc8SWarner Losh 	if (i == j)
4791285bcc8SWarner Losh 		warnx("Boot variable %04x not in BootOrder", bootnum);
4801285bcc8SWarner Losh 	else if (set_bootvar("BootOrder", new, j) < 0)
4811285bcc8SWarner Losh 		err(1, "Unable to update BootOrder with new value");
4821285bcc8SWarner Losh 	free(new);
4831285bcc8SWarner Losh }
4841285bcc8SWarner Losh 
4851285bcc8SWarner Losh 
4861285bcc8SWarner Losh static void
4871285bcc8SWarner Losh delete_bootvar(int bootnum)
4881285bcc8SWarner Losh {
4891285bcc8SWarner Losh 	char *name;
4901285bcc8SWarner Losh 	int defer = 0;
4911285bcc8SWarner Losh 
4921285bcc8SWarner Losh 	/*
4931285bcc8SWarner Losh 	 * Try to delete the boot variable and remocve it
4941285bcc8SWarner Losh 	 * from the boot order. We always do both actions
4951285bcc8SWarner Losh 	 * to make it easy to clean up from oopses.
4961285bcc8SWarner Losh 	 */
4971285bcc8SWarner Losh 	if (bootnum < 0 || bootnum > 0xffff)
4981285bcc8SWarner Losh 		errx(1, "Bad boot variable %#x", bootnum);
4991285bcc8SWarner Losh 	asprintf(&name, "%s%04X", "Boot", bootnum);
5001285bcc8SWarner Losh 	if (name == NULL)
5011285bcc8SWarner Losh 		err(1, "asprintf");
5021285bcc8SWarner Losh 	printf("Removing boot variable '%s'\n", name);
5031285bcc8SWarner Losh 	if (efi_del_variable(EFI_GLOBAL_GUID, name) < 0) {
5041285bcc8SWarner Losh 		defer = 1;
5051285bcc8SWarner Losh 		warn("cannot delete variable %s", name);
5061285bcc8SWarner Losh 	}
5071285bcc8SWarner Losh 	printf("Removing 0x%x from BootOrder\n", bootnum);
5081285bcc8SWarner Losh 	remove_from_order(bootnum);
5091285bcc8SWarner Losh 	free(name);
5101285bcc8SWarner Losh 	if (defer)
5111285bcc8SWarner Losh 		exit(defer);
5121285bcc8SWarner Losh }
5131285bcc8SWarner Losh 
5141285bcc8SWarner Losh 
5151285bcc8SWarner Losh static void
5161285bcc8SWarner Losh del_bootnext(void)
5171285bcc8SWarner Losh {
5181285bcc8SWarner Losh 
5191285bcc8SWarner Losh 	if (efi_del_variable(EFI_GLOBAL_GUID, "BootNext") < 0)
5201285bcc8SWarner Losh 		err(1, "efi_del_variable");
5211285bcc8SWarner Losh }
5221285bcc8SWarner Losh 
5231285bcc8SWarner Losh static void
5241285bcc8SWarner Losh handle_bootnext(uint16_t bootnum)
5251285bcc8SWarner Losh {
5261285bcc8SWarner Losh 	uint16_t num;
5271285bcc8SWarner Losh 
5281285bcc8SWarner Losh 	le16enc(&num, bootnum);
5291285bcc8SWarner Losh 	if (set_bootvar("BootNext", (uint8_t*)&bootnum, sizeof(uint16_t)) < 0)
5301285bcc8SWarner Losh 		err(1, "set_bootvar");
5311285bcc8SWarner Losh }
5321285bcc8SWarner Losh 
5331285bcc8SWarner Losh 
5341285bcc8SWarner Losh static int
5351285bcc8SWarner Losh compare(const void *a, const void *b)
5361285bcc8SWarner Losh {
5371285bcc8SWarner Losh 	uint16_t c;
5381285bcc8SWarner Losh 	uint16_t d;
5391285bcc8SWarner Losh 
5401285bcc8SWarner Losh 	memcpy(&c, a, sizeof(uint16_t));
5411285bcc8SWarner Losh 	memcpy(&d, b, sizeof(uint16_t));
5421285bcc8SWarner Losh 
5431285bcc8SWarner Losh 	if (c < d)
5441285bcc8SWarner Losh 		return -1;
5451285bcc8SWarner Losh 	if (c == d)
5461285bcc8SWarner Losh 		return  0;
5471285bcc8SWarner Losh 	return  1;
5481285bcc8SWarner Losh }
5491285bcc8SWarner Losh 
5501285bcc8SWarner Losh static char *
5516604afe9SWarner Losh make_next_boot_var_name(void)
5521285bcc8SWarner Losh {
5531285bcc8SWarner Losh 	struct entry *v;
5541285bcc8SWarner Losh 	uint16_t *vals, next_free = 0;
5551285bcc8SWarner Losh 	char *name;
5561285bcc8SWarner Losh 	int cnt = 0;
5571285bcc8SWarner Losh 	int i;
5581285bcc8SWarner Losh 
5591285bcc8SWarner Losh 	LIST_FOREACH(v, &efivars, entries) {
5601285bcc8SWarner Losh 		cnt++;
5611285bcc8SWarner Losh 	}
5621285bcc8SWarner Losh 
5631285bcc8SWarner Losh 	vals = malloc(sizeof(uint16_t) * cnt);
5641285bcc8SWarner Losh 	if (!vals)
5651285bcc8SWarner Losh 		return NULL;
5661285bcc8SWarner Losh 
5671285bcc8SWarner Losh 	i = 0;
5681285bcc8SWarner Losh 	LIST_FOREACH(v, &efivars, entries) {
5691285bcc8SWarner Losh 		vals[i++] = v->idx;
5701285bcc8SWarner Losh 	}
5711285bcc8SWarner Losh 	qsort(vals, cnt, sizeof(uint16_t), compare);
5721285bcc8SWarner Losh 	/* if the hole is at the beginning, just return zero */
5731285bcc8SWarner Losh 	if (vals[0] > 0) {
5741285bcc8SWarner Losh 		next_free = 0;
5751285bcc8SWarner Losh 	} else {
5761285bcc8SWarner Losh 		/* now just run the list looking for the first hole */
5771285bcc8SWarner Losh 		for (i = 0; i < cnt - 1 && next_free == 0; i++)
578018caaaeSWarner Losh 			if (vals[i] + 1 != vals[i + 1])
5791285bcc8SWarner Losh 				next_free = vals[i] + 1;
5801285bcc8SWarner Losh 		if (next_free == 0)
5811285bcc8SWarner Losh 			next_free = vals[cnt - 1] + 1;
5821285bcc8SWarner Losh 		/* In theory we could have used all 65k slots -- what to do? */
5831285bcc8SWarner Losh 	}
5841285bcc8SWarner Losh 	free(vals);
5851285bcc8SWarner Losh 
5861285bcc8SWarner Losh 	asprintf(&name, "%s%04X", "Boot", next_free);
5871285bcc8SWarner Losh 	if (name == NULL)
5881285bcc8SWarner Losh 		err(1, "asprintf");
5891285bcc8SWarner Losh 	return name;
5901285bcc8SWarner Losh }
5911285bcc8SWarner Losh 
592de26ba4dSWarner Losh static char *
593de26ba4dSWarner Losh make_boot_var_name(uint16_t bootnum)
594de26ba4dSWarner Losh {
595de26ba4dSWarner Losh 	struct entry *v;
596de26ba4dSWarner Losh 	char *name;
597de26ba4dSWarner Losh 
598de26ba4dSWarner Losh 	LIST_FOREACH(v, &efivars, entries) {
599de26ba4dSWarner Losh 		if (v->idx == bootnum)
600de26ba4dSWarner Losh 			return NULL;
601de26ba4dSWarner Losh 	}
602de26ba4dSWarner Losh 
603de26ba4dSWarner Losh 	asprintf(&name, "%s%04X", "Boot", bootnum);
604de26ba4dSWarner Losh 	if (name == NULL)
605de26ba4dSWarner Losh 		err(1, "asprintf");
606de26ba4dSWarner Losh 	return name;
607de26ba4dSWarner Losh }
6081285bcc8SWarner Losh 
6091285bcc8SWarner Losh static size_t
6101285bcc8SWarner Losh create_loadopt(uint8_t *buf, size_t bufmax, uint32_t attributes, efidp dp, size_t dp_size,
6111285bcc8SWarner Losh     const char *description, const uint8_t *optional_data, size_t optional_data_size)
6121285bcc8SWarner Losh {
6131285bcc8SWarner Losh 	efi_char *bbuf = NULL;
6141285bcc8SWarner Losh 	uint8_t *pos = buf;
6151285bcc8SWarner Losh 	size_t desc_len = 0;
6161285bcc8SWarner Losh 	size_t len;
6171285bcc8SWarner Losh 
6181285bcc8SWarner Losh 	if (optional_data == NULL && optional_data_size != 0)
6191285bcc8SWarner Losh 		return BAD_LENGTH;
6201285bcc8SWarner Losh 	if (dp == NULL && dp_size != 0)
6211285bcc8SWarner Losh 		return BAD_LENGTH;
6221285bcc8SWarner Losh 
6231285bcc8SWarner Losh 	/*
6241285bcc8SWarner Losh 	 * Compute the length to make sure the passed in buffer is long enough.
6251285bcc8SWarner Losh 	 */
6261285bcc8SWarner Losh 	utf8_to_ucs2(description, &bbuf, &desc_len);
6271285bcc8SWarner Losh 	len = sizeof(uint32_t) + sizeof(uint16_t) + desc_len + dp_size + optional_data_size;
6281285bcc8SWarner Losh 	if (len > bufmax) {
6291285bcc8SWarner Losh 		free(bbuf);
6301285bcc8SWarner Losh 		return BAD_LENGTH;
6311285bcc8SWarner Losh 	}
6321285bcc8SWarner Losh 
6331285bcc8SWarner Losh 	le32enc(pos, attributes);
6341285bcc8SWarner Losh 	pos += sizeof (attributes);
6351285bcc8SWarner Losh 
6361285bcc8SWarner Losh 	le16enc(pos, dp_size);
6371285bcc8SWarner Losh 	pos += sizeof (uint16_t);
6381285bcc8SWarner Losh 
6391285bcc8SWarner Losh 	memcpy(pos, bbuf, desc_len);	/* NB:desc_len includes strailing NUL */
6401285bcc8SWarner Losh 	pos += desc_len;
6411285bcc8SWarner Losh 	free(bbuf);
6421285bcc8SWarner Losh 
6431285bcc8SWarner Losh 	memcpy(pos, dp, dp_size);
6441285bcc8SWarner Losh 	pos += dp_size;
6451285bcc8SWarner Losh 
6461285bcc8SWarner Losh 	if (optional_data && optional_data_size > 0) {
6471285bcc8SWarner Losh 		memcpy(pos, optional_data, optional_data_size);
6481285bcc8SWarner Losh 		pos += optional_data_size;
6491285bcc8SWarner Losh 	}
6501285bcc8SWarner Losh 
6511285bcc8SWarner Losh 	return pos - buf;
6521285bcc8SWarner Losh }
6531285bcc8SWarner Losh 
6541285bcc8SWarner Losh 
6551285bcc8SWarner Losh static int
656de26ba4dSWarner Losh make_boot_var(const char *label, const char *loader, const char *kernel, const char *env, bool dry_run,
657de6b4aa8SWarner Losh     int bootnum, bool activate)
6581285bcc8SWarner Losh {
6591285bcc8SWarner Losh 	struct entry *new_ent;
6601285bcc8SWarner Losh 	uint32_t load_attrs = 0;
6611285bcc8SWarner Losh 	uint8_t *load_opt_buf;
6621285bcc8SWarner Losh 	size_t lopt_size, llen, klen;
6631285bcc8SWarner Losh 	efidp dp, loaderdp, kerneldp;
6641285bcc8SWarner Losh 	char *bootvar = NULL;
6651285bcc8SWarner Losh 	int ret;
6661285bcc8SWarner Losh 
667da15338dSWarner Losh 	assert(label != NULL);
668da15338dSWarner Losh 
669de26ba4dSWarner Losh 	if (bootnum == -1)
6701285bcc8SWarner Losh 		bootvar = make_next_boot_var_name();
671de26ba4dSWarner Losh 	else
672de26ba4dSWarner Losh 		bootvar = make_boot_var_name((uint16_t)bootnum);
6731285bcc8SWarner Losh 	if (bootvar == NULL)
6741285bcc8SWarner Losh 		err(1, "bootvar creation");
6751285bcc8SWarner Losh 	if (loader == NULL)
6761285bcc8SWarner Losh 		errx(1, "Must specify boot loader");
6771285bcc8SWarner Losh 	if (efivar_unix_path_to_device_path(loader, &loaderdp) != 0)
6781285bcc8SWarner Losh 		err(1, "Cannot translate unix loader path '%s' to UEFI", loader);
6791285bcc8SWarner Losh 	if (kernel != NULL) {
6801285bcc8SWarner Losh 		if (efivar_unix_path_to_device_path(kernel, &kerneldp) != 0)
6811285bcc8SWarner Losh 			err(1, "Cannot translate unix kernel path '%s' to UEFI", kernel);
6821285bcc8SWarner Losh 	} else {
6831285bcc8SWarner Losh 		kerneldp = NULL;
6841285bcc8SWarner Losh 	}
6851285bcc8SWarner Losh 	llen = efidp_size(loaderdp);
686af934ea3SWarner Losh 	if (llen > MAX_DP_LEN)
687af934ea3SWarner Losh 		errx(1, "Loader path too long.");
6881285bcc8SWarner Losh 	klen = efidp_size(kerneldp);
689af934ea3SWarner Losh 	if (klen > MAX_DP_LEN)
690af934ea3SWarner Losh 		errx(1, "Kernel path too long.");
6911285bcc8SWarner Losh 	dp = malloc(llen + klen);
692af934ea3SWarner Losh 	if (dp == NULL)
693af934ea3SWarner Losh 		errx(1, "Can't allocate memory for new device paths");
6941285bcc8SWarner Losh 	memcpy(dp, loaderdp, llen);
6951285bcc8SWarner Losh 	if (kerneldp != NULL)
6961285bcc8SWarner Losh 		memcpy((char *)dp + llen, kerneldp, klen);
6971285bcc8SWarner Losh 
6981285bcc8SWarner Losh 	/* don't make the new bootvar active by default, use the -a option later */
6991285bcc8SWarner Losh 	load_attrs = LOAD_OPTION_CATEGORY_BOOT;
700de6b4aa8SWarner Losh 	if (activate)
701de6b4aa8SWarner Losh 		load_attrs |= LOAD_OPTION_ACTIVE;
7021285bcc8SWarner Losh 	load_opt_buf = malloc(MAX_LOADOPT_LEN);
7031285bcc8SWarner Losh 	if (load_opt_buf == NULL)
7041285bcc8SWarner Losh 		err(1, "malloc");
7051285bcc8SWarner Losh 
7061285bcc8SWarner Losh 	lopt_size = create_loadopt(load_opt_buf, MAX_LOADOPT_LEN, load_attrs,
7071127aea3SWarner Losh 	    dp, llen + klen, label, env, env ? strlen(env) + 1 : 0);
7081285bcc8SWarner Losh 	if (lopt_size == BAD_LENGTH)
70967a5e533SEmmanuel Vadot 		errx(1, "Can't create loadopt");
7101285bcc8SWarner Losh 
7111285bcc8SWarner Losh 	ret = 0;
7121285bcc8SWarner Losh 	if (!dry_run) {
7131285bcc8SWarner Losh 		ret = efi_set_variable(EFI_GLOBAL_GUID, bootvar,
7141285bcc8SWarner Losh 		    (uint8_t*)load_opt_buf, lopt_size, COMMON_ATTRS);
7151285bcc8SWarner Losh 	}
7161285bcc8SWarner Losh 
7171285bcc8SWarner Losh 	if (ret)
7181285bcc8SWarner Losh 		err(1, "efi_set_variable");
7191285bcc8SWarner Losh 
72067a5e533SEmmanuel Vadot 	if (!dry_run)
7211285bcc8SWarner Losh 		add_to_boot_order(bootvar); /* first, still not active */
7221285bcc8SWarner Losh 	new_ent = malloc(sizeof(struct entry));
7231285bcc8SWarner Losh 	if (new_ent == NULL)
7241285bcc8SWarner Losh 		err(1, "malloc");
7251285bcc8SWarner Losh 	memset(new_ent, 0, sizeof(struct entry));
7261285bcc8SWarner Losh 	new_ent->name = bootvar;
7271285bcc8SWarner Losh 	new_ent->guid = EFI_GLOBAL_GUID;
7281285bcc8SWarner Losh 	LIST_INSERT_HEAD(&efivars, new_ent, entries);
729a6b6f888SWarner Losh 	free(load_opt_buf);
7301285bcc8SWarner Losh 	free(dp);
731a6b6f888SWarner Losh 
7321285bcc8SWarner Losh 	return 0;
7331285bcc8SWarner Losh }
7341285bcc8SWarner Losh 
7351285bcc8SWarner Losh 
7361285bcc8SWarner Losh static void
7371285bcc8SWarner Losh print_loadopt_str(uint8_t *data, size_t datalen)
7381285bcc8SWarner Losh {
7391285bcc8SWarner Losh 	char *dev, *relpath, *abspath;
7401285bcc8SWarner Losh 	uint32_t attr;
7411285bcc8SWarner Losh 	uint16_t fplen;
7421285bcc8SWarner Losh 	efi_char *descr;
7431285bcc8SWarner Losh 	uint8_t *ep = data + datalen;
7441285bcc8SWarner Losh 	uint8_t *walker = data;
7451285bcc8SWarner Losh 	efidp dp, edp;
7461285bcc8SWarner Losh 	char buf[1024];
7471285bcc8SWarner Losh 	int len;
7481285bcc8SWarner Losh 	int rv;
749a2aa6671SWarner Losh 	int indent;
7501285bcc8SWarner Losh 
7511285bcc8SWarner Losh 	if (datalen < sizeof(attr) + sizeof(fplen) + sizeof(efi_char))
7521285bcc8SWarner Losh 		return;
7531285bcc8SWarner Losh 	// First 4 bytes are attribute flags
7541285bcc8SWarner Losh 	attr = le32dec(walker);
7551285bcc8SWarner Losh 	walker += sizeof(attr);
7561285bcc8SWarner Losh 	// Next two bytes are length of the file paths
7571285bcc8SWarner Losh 	fplen = le16dec(walker);
7581285bcc8SWarner Losh 	walker += sizeof(fplen);
7591285bcc8SWarner Losh 	// Next we have a 0 terminated UCS2 string that we know to be aligned
7601285bcc8SWarner Losh 	descr = (efi_char *)(intptr_t)(void *)walker;
7611285bcc8SWarner Losh 	len = ucs2len(descr); // XXX need to sanity check that len < (datalen - (ep - walker) / 2)
7621285bcc8SWarner Losh 	walker += (len + 1) * sizeof(efi_char);
7631285bcc8SWarner Losh 	if (walker > ep)
7641285bcc8SWarner Losh 		return;
7651285bcc8SWarner Losh 	// Now we have fplen bytes worth of file path stuff
7661285bcc8SWarner Losh 	dp = (efidp)walker;
7671285bcc8SWarner Losh 	walker += fplen;
7681285bcc8SWarner Losh 	if (walker > ep)
7691285bcc8SWarner Losh 		return;
7701285bcc8SWarner Losh 	edp = (efidp)walker;
77196289d09SAlexander Kabaev 	/*
77296289d09SAlexander Kabaev 	 * Everything left is the binary option args
77396289d09SAlexander Kabaev 	 * opt = walker;
77496289d09SAlexander Kabaev 	 * optlen = ep - walker;
77596289d09SAlexander Kabaev 	 */
776a2aa6671SWarner Losh 	indent = 1;
7771285bcc8SWarner Losh 	while (dp < edp) {
7781285bcc8SWarner Losh 		efidp_format_device_path(buf, sizeof(buf), dp,
7791285bcc8SWarner Losh 		    (intptr_t)(void *)edp - (intptr_t)(void *)dp);
780a2aa6671SWarner Losh 		printf("%*s%s\n", indent, "", buf);
781a2aa6671SWarner Losh 		indent = 10 + len + 1;
7821285bcc8SWarner Losh 		rv = efivar_device_path_to_unix_path(dp, &dev, &relpath, &abspath);
7831285bcc8SWarner Losh 		if (rv == 0) {
784a2aa6671SWarner Losh 			printf("%*s%s:%s %s\n", indent + 4, "", dev, relpath, abspath);
7851285bcc8SWarner Losh 			free(dev);
7861285bcc8SWarner Losh 			free(relpath);
7871285bcc8SWarner Losh 			free(abspath);
7881285bcc8SWarner Losh 		}
7891285bcc8SWarner Losh 		dp = (efidp)((char *)dp + efidp_size(dp));
7901285bcc8SWarner Losh 	}
7911285bcc8SWarner Losh }
7921285bcc8SWarner Losh 
7931285bcc8SWarner Losh static char *
7941285bcc8SWarner Losh get_descr(uint8_t *data)
7951285bcc8SWarner Losh {
7961285bcc8SWarner Losh 	uint8_t *pos = data;
7971285bcc8SWarner Losh 	efi_char *desc;
7981285bcc8SWarner Losh 	int  len;
7991285bcc8SWarner Losh 	char *buf;
8001285bcc8SWarner Losh 	int i = 0;
8011285bcc8SWarner Losh 
8021285bcc8SWarner Losh 	pos += sizeof(uint32_t) + sizeof(uint16_t);
8031285bcc8SWarner Losh 	desc = (efi_char*)(intptr_t)(void *)pos;
8041285bcc8SWarner Losh 	len = ucs2len(desc);
8051285bcc8SWarner Losh 	buf = malloc(len + 1);
8061285bcc8SWarner Losh 	memset(buf, 0, len + 1);
8071285bcc8SWarner Losh 	while (desc[i]) {
8081285bcc8SWarner Losh 		buf[i] = desc[i];
8091285bcc8SWarner Losh 		i++;
8101285bcc8SWarner Losh 	}
8111285bcc8SWarner Losh 	return (char*)buf;
8121285bcc8SWarner Losh }
8131285bcc8SWarner Losh 
8141285bcc8SWarner Losh 
81551922c69SWarner Losh static bool
81651922c69SWarner Losh print_boot_var(const char *name, bool verbose, bool curboot)
81751922c69SWarner Losh {
81851922c69SWarner Losh 	size_t size;
81951922c69SWarner Losh 	uint32_t load_attrs;
82051922c69SWarner Losh 	uint8_t *data;
82151922c69SWarner Losh 	int ret;
82251922c69SWarner Losh 	char *d;
82351922c69SWarner Losh 
82451922c69SWarner Losh 	ret = efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, NULL);
82551922c69SWarner Losh 	if (ret < 0)
82651922c69SWarner Losh 		return false;
82751922c69SWarner Losh 	load_attrs = le32dec(data);
82851922c69SWarner Losh 	d = get_descr(data);
82951922c69SWarner Losh 	printf("%c%s%c %s", curboot ? '+' : ' ', name,
83051922c69SWarner Losh 	    ((load_attrs & LOAD_OPTION_ACTIVE) ? '*': ' '), d);
83151922c69SWarner Losh 	free(d);
83251922c69SWarner Losh 	if (verbose)
83351922c69SWarner Losh 		print_loadopt_str(data, size);
83451922c69SWarner Losh 	else
83551922c69SWarner Losh 		printf("\n");
83651922c69SWarner Losh 	return true;
83751922c69SWarner Losh }
83851922c69SWarner Losh 
83951922c69SWarner Losh 
840*83c42372SD Scott Phillips static bool
841*83c42372SD Scott Phillips os_indication_supported(uint64_t indication)
842*83c42372SD Scott Phillips {
843*83c42372SD Scott Phillips 	uint8_t *data;
844*83c42372SD Scott Phillips 	size_t size;
845*83c42372SD Scott Phillips 	uint32_t attrs;
846*83c42372SD Scott Phillips 	int ret;
847*83c42372SD Scott Phillips 
848*83c42372SD Scott Phillips 	ret = efi_get_variable(EFI_GLOBAL_GUID, "OsIndicationsSupported", &data,
849*83c42372SD Scott Phillips 	    &size, &attrs);
850*83c42372SD Scott Phillips 	if (ret < 0)
851*83c42372SD Scott Phillips 		return false;
852*83c42372SD Scott Phillips 	return (le64dec(data) & indication) == indication;
853*83c42372SD Scott Phillips }
854*83c42372SD Scott Phillips 
855*83c42372SD Scott Phillips static uint64_t
856*83c42372SD Scott Phillips os_indications(void)
857*83c42372SD Scott Phillips {
858*83c42372SD Scott Phillips 	uint8_t *data;
859*83c42372SD Scott Phillips 	size_t size;
860*83c42372SD Scott Phillips 	uint32_t attrs;
861*83c42372SD Scott Phillips 	int ret;
862*83c42372SD Scott Phillips 
863*83c42372SD Scott Phillips 	ret = efi_get_variable(EFI_GLOBAL_GUID, "OsIndications", &data, &size,
864*83c42372SD Scott Phillips 	    &attrs);
865*83c42372SD Scott Phillips 	if (ret < 0)
866*83c42372SD Scott Phillips 		return 0;
867*83c42372SD Scott Phillips 	return le64dec(data);
868*83c42372SD Scott Phillips }
869*83c42372SD Scott Phillips 
870*83c42372SD Scott Phillips static int
871*83c42372SD Scott Phillips os_indications_set(uint64_t mask, uint64_t val)
872*83c42372SD Scott Phillips {
873*83c42372SD Scott Phillips 	uint8_t new[sizeof(uint64_t)];
874*83c42372SD Scott Phillips 
875*83c42372SD Scott Phillips 	le64enc(&new, (os_indications() & ~mask) | (val & mask));
876*83c42372SD Scott Phillips 	return set_bootvar("OsIndications", new, sizeof(new));
877*83c42372SD Scott Phillips }
878*83c42372SD Scott Phillips 
8791285bcc8SWarner Losh /* Cmd epilogue, or just the default with no args.
8801285bcc8SWarner Losh  * The order is [bootnext] bootcurrent, timeout, order, and the bootvars [-v]
8811285bcc8SWarner Losh  */
8821285bcc8SWarner Losh static int
8831285bcc8SWarner Losh print_boot_vars(bool verbose)
8841285bcc8SWarner Losh {
8851285bcc8SWarner Losh 	/*
8861285bcc8SWarner Losh 	 * just read and print the current values
8871285bcc8SWarner Losh 	 * as a command epilogue
8881285bcc8SWarner Losh 	 */
8891285bcc8SWarner Losh 	struct entry *v;
8901285bcc8SWarner Losh 	uint8_t *data;
8911285bcc8SWarner Losh 	size_t size;
89251922c69SWarner Losh 	uint32_t attrs;
89351922c69SWarner Losh 	int ret, bolen;
89451922c69SWarner Losh 	uint16_t *boot_order = NULL, current;
895*83c42372SD Scott Phillips 	bool boot_to_fw_ui;
896*83c42372SD Scott Phillips 
897*83c42372SD Scott Phillips 	if (os_indication_supported(EFI_OS_INDICATIONS_BOOT_TO_FW_UI)) {
898*83c42372SD Scott Phillips 		boot_to_fw_ui =
899*83c42372SD Scott Phillips 		    (os_indications() & EFI_OS_INDICATIONS_BOOT_TO_FW_UI) != 0;
900*83c42372SD Scott Phillips 		printf("Boot to FW : %s\n", boot_to_fw_ui != 0 ?
901*83c42372SD Scott Phillips 		    "true" : "false");
902*83c42372SD Scott Phillips 	}
9031285bcc8SWarner Losh 
9041285bcc8SWarner Losh 	ret = efi_get_variable(EFI_GLOBAL_GUID, "BootNext", &data, &size, &attrs);
9051285bcc8SWarner Losh 	if (ret > 0) {
9061285bcc8SWarner Losh 		printf("BootNext : %04x\n", le16dec(data));
9071285bcc8SWarner Losh 	}
9081285bcc8SWarner Losh 
9091285bcc8SWarner Losh 	ret = efi_get_variable(EFI_GLOBAL_GUID, "BootCurrent", &data, &size,&attrs);
91051922c69SWarner Losh 	current = le16dec(data);
91151922c69SWarner Losh 	printf("BootCurrent: %04x\n", current);
9121285bcc8SWarner Losh 
9131285bcc8SWarner Losh 	ret = efi_get_variable(EFI_GLOBAL_GUID, "Timeout", &data, &size, &attrs);
9141285bcc8SWarner Losh 	if (ret > 0) {
9151285bcc8SWarner Losh 		printf("Timeout    : %d seconds\n", le16dec(data));
9161285bcc8SWarner Losh 	}
91712f409ffSWarner Losh 
91812f409ffSWarner Losh 	if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) > 0) {
91912f409ffSWarner Losh 		if (size % 2 == 1)
92012f409ffSWarner Losh 			warn("Bad BootOrder variable: odd length %d", (int)size);
92151922c69SWarner Losh 		boot_order = malloc(size);
92251922c69SWarner Losh 		bolen = size / 2;
92312f409ffSWarner Losh 		printf("BootOrder  : ");
92451922c69SWarner Losh 		for (size_t i = 0; i < size; i += 2) {
92551922c69SWarner Losh 			boot_order[i / 2] = le16dec(data + i);
92651922c69SWarner Losh 			printf("%04X%s", boot_order[i / 2], i == size - 2 ? "\n" : ", ");
92751922c69SWarner Losh 		}
92812f409ffSWarner Losh 	}
9291285bcc8SWarner Losh 
93051922c69SWarner Losh 	if (boot_order == NULL) {
93151922c69SWarner Losh 		/*
93251922c69SWarner Losh 		 * now we want to fetch 'em all fresh again
9331285bcc8SWarner Losh 		 * which possibly includes a newly created bootvar
9341285bcc8SWarner Losh 		 */
9351285bcc8SWarner Losh 		LIST_FOREACH(v, &efivars, entries) {
93651922c69SWarner Losh 			print_boot_var(v->name, verbose, v->idx == current);
93751922c69SWarner Losh 		}
93851922c69SWarner Losh 	} else {
93951922c69SWarner Losh 		LIST_FOREACH(v, &efivars, entries) {
94051922c69SWarner Losh 			v->flags = 0;
94151922c69SWarner Losh 		}
94251922c69SWarner Losh 		for (int i = 0; i < bolen; i++) {
94351922c69SWarner Losh 			char buffer[10];
94451922c69SWarner Losh 
94551922c69SWarner Losh 			snprintf(buffer, sizeof(buffer), "Boot%04X", boot_order[i]);
94651922c69SWarner Losh 			if (!print_boot_var(buffer, verbose, boot_order[i] == current))
94751922c69SWarner Losh 				printf("%s: MISSING!\n", buffer);
94851922c69SWarner Losh 			LIST_FOREACH(v, &efivars, entries) {
94951922c69SWarner Losh 				if (v->idx == boot_order[i]) {
95051922c69SWarner Losh 					v->flags |= SEEN;
95151922c69SWarner Losh 					break;
95251922c69SWarner Losh 				}
95351922c69SWarner Losh 			}
95451922c69SWarner Losh 		}
95551922c69SWarner Losh 		if (verbose) {
95651922c69SWarner Losh 			printf("\n\nUnreferenced Variables:\n");
95751922c69SWarner Losh 			LIST_FOREACH(v, &efivars, entries) {
95851922c69SWarner Losh 				if (v->flags == 0)
95951922c69SWarner Losh 					print_boot_var(v->name, verbose, v->idx == current);
96051922c69SWarner Losh 			}
96151922c69SWarner Losh 		}
9621285bcc8SWarner Losh 	}
9631285bcc8SWarner Losh 	return 0;
9641285bcc8SWarner Losh }
9651285bcc8SWarner Losh 
9661285bcc8SWarner Losh static void
9676604afe9SWarner Losh delete_timeout(void)
9681285bcc8SWarner Losh {
9691285bcc8SWarner Losh 
9701285bcc8SWarner Losh 	efi_del_variable(EFI_GLOBAL_GUID,"Timeout");
9711285bcc8SWarner Losh }
9721285bcc8SWarner Losh 
9731285bcc8SWarner Losh static void
9741285bcc8SWarner Losh handle_timeout(int to)
9751285bcc8SWarner Losh {
9761285bcc8SWarner Losh 	uint16_t timeout;
9771285bcc8SWarner Losh 
9781285bcc8SWarner Losh 	le16enc(&timeout, to);
979c66805a5SWarner Losh 	if (set_bootvar("Timeout", (uint8_t *)&timeout, sizeof(timeout)) < 0)
980c66805a5SWarner Losh 		errx(1, "Can't set Timeout for booting.");
9811285bcc8SWarner Losh }
9821285bcc8SWarner Losh 
9831cdb8eb8SWarner Losh static void
9841cdb8eb8SWarner Losh report_esp_device(bool do_dp, bool do_unix)
9851cdb8eb8SWarner Losh {
9861cdb8eb8SWarner Losh 	uint8_t *data;
9871cdb8eb8SWarner Losh 	size_t size, len;
9881cdb8eb8SWarner Losh 	uint32_t attrs;
9891cdb8eb8SWarner Losh 	int ret;
9901cdb8eb8SWarner Losh 	uint16_t current, fplen;
9911cdb8eb8SWarner Losh 	char *name, *dev, *relpath, *abspath;
9921cdb8eb8SWarner Losh 	uint8_t *walker, *ep;
9931cdb8eb8SWarner Losh 	efi_char *descr;
9941cdb8eb8SWarner Losh 	efidp dp, edp;
9951cdb8eb8SWarner Losh 	char buf[PATH_MAX];
9961cdb8eb8SWarner Losh 
9971cdb8eb8SWarner Losh 	if (do_dp && do_unix)
9981cdb8eb8SWarner Losh 		errx(1, "Can't report both UEFI device-path and Unix path together");
9991cdb8eb8SWarner Losh 
10001cdb8eb8SWarner Losh 	ret = efi_get_variable(EFI_GLOBAL_GUID, "BootCurrent", &data, &size,&attrs);
10011cdb8eb8SWarner Losh 	if (ret < 0)
10021cdb8eb8SWarner Losh 		err(1, "Can't get BootCurrent");
10031cdb8eb8SWarner Losh 	current = le16dec(data);
10041cdb8eb8SWarner Losh 	if (asprintf(&name, "Boot%04X", current) < 0)
10051cdb8eb8SWarner Losh 		err(1, "Can't format boot var\n");
10061cdb8eb8SWarner Losh 	if (efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, NULL) < 0)
10071cdb8eb8SWarner Losh 		err(1, "Can't retrieve EFI var %s", name);
10081cdb8eb8SWarner Losh 	// First 4 bytes are attribute flags
10091cdb8eb8SWarner Losh 	walker = data;
10101cdb8eb8SWarner Losh 	ep = walker + size;
10111cdb8eb8SWarner Losh 	walker += sizeof(uint32_t);
10121cdb8eb8SWarner Losh 	// Next two bytes are length of the file paths
10131cdb8eb8SWarner Losh 	fplen = le16dec(walker);
10141cdb8eb8SWarner Losh 	walker += sizeof(fplen);
10151cdb8eb8SWarner Losh 	// Next we have a 0 terminated UCS2 string that we know to be aligned
10161cdb8eb8SWarner Losh 	descr = (efi_char *)(intptr_t)(void *)walker;
10171cdb8eb8SWarner Losh 	len = ucs2len(descr); // XXX need to sanity check that len < (datalen - (ep - walker) / 2)
10181cdb8eb8SWarner Losh 	walker += (len + 1) * sizeof(efi_char);
10191cdb8eb8SWarner Losh 	if (walker > ep)
10201cdb8eb8SWarner Losh 		errx(1, "malformed boot variable %s", name);
10211cdb8eb8SWarner Losh 	// Now we have fplen bytes worth of file path stuff
10221cdb8eb8SWarner Losh 	dp = (efidp)walker;
10231cdb8eb8SWarner Losh 	walker += fplen;
10241cdb8eb8SWarner Losh 	edp = (efidp)walker;
10251cdb8eb8SWarner Losh 	if (walker > ep)
10261cdb8eb8SWarner Losh 		errx(1, "malformed boot variable %s", name);
10271cdb8eb8SWarner Losh 	if (do_dp) {
10281cdb8eb8SWarner Losh 		efidp_format_device_path_node(buf, sizeof(buf), dp);
10291cdb8eb8SWarner Losh 		printf("%s\n", buf);
10301cdb8eb8SWarner Losh 		exit(0);
10311cdb8eb8SWarner Losh 	}
10321cdb8eb8SWarner Losh 	if (efivar_device_path_to_unix_path(dp, &dev, &relpath, &abspath) < 0)
10331cdb8eb8SWarner Losh 		errx(1, "Can't convert to unix path");
10341cdb8eb8SWarner Losh 	if (do_unix) {
10351cdb8eb8SWarner Losh 		if (abspath == NULL)
10361cdb8eb8SWarner Losh 			errx(1, "Can't find where %s:%s is mounted",
10371cdb8eb8SWarner Losh 			    dev, relpath);
10381cdb8eb8SWarner Losh 		abspath[strlen(abspath) - strlen(relpath) - 1] = '\0';
10391cdb8eb8SWarner Losh 		printf("%s\n", abspath);
10401cdb8eb8SWarner Losh 	} else {
10411cdb8eb8SWarner Losh 		printf("%s\n", dev);
10421cdb8eb8SWarner Losh 	}
10431cdb8eb8SWarner Losh 	free(dev);
10441cdb8eb8SWarner Losh 	free(relpath);
10451cdb8eb8SWarner Losh 	free(abspath);
10461cdb8eb8SWarner Losh 	exit(0);
10471cdb8eb8SWarner Losh }
10481cdb8eb8SWarner Losh 
1049*83c42372SD Scott Phillips static void
1050*83c42372SD Scott Phillips set_boot_to_fw_ui(bool to_fw)
1051*83c42372SD Scott Phillips {
1052*83c42372SD Scott Phillips 	int ret;
1053*83c42372SD Scott Phillips 
1054*83c42372SD Scott Phillips 	if (!os_indication_supported(EFI_OS_INDICATIONS_BOOT_TO_FW_UI)) {
1055*83c42372SD Scott Phillips 		if (to_fw)
1056*83c42372SD Scott Phillips 			errx(1, "boot to fw ui not supported");
1057*83c42372SD Scott Phillips 		else
1058*83c42372SD Scott Phillips 			return;
1059*83c42372SD Scott Phillips 	}
1060*83c42372SD Scott Phillips 	ret = os_indications_set(EFI_OS_INDICATIONS_BOOT_TO_FW_UI,
1061*83c42372SD Scott Phillips 	    to_fw ? ~0 : 0);
1062*83c42372SD Scott Phillips 	if (ret < 0)
1063*83c42372SD Scott Phillips 		errx(1, "failed to set boot to fw ui");
1064*83c42372SD Scott Phillips }
1065*83c42372SD Scott Phillips 
10661285bcc8SWarner Losh int
10671285bcc8SWarner Losh main(int argc, char *argv[])
10681285bcc8SWarner Losh {
10691285bcc8SWarner Losh 
10701285bcc8SWarner Losh 	if (!efi_variables_supported())
10711285bcc8SWarner Losh 		errx(1, "efi variables not supported on this system. root? kldload efirt?");
10721285bcc8SWarner Losh 
10731285bcc8SWarner Losh 	memset(&opts, 0, sizeof (bmgr_opts_t));
10741285bcc8SWarner Losh 	parse_args(argc, argv);
10751285bcc8SWarner Losh 	read_vars();
10761285bcc8SWarner Losh 
10771285bcc8SWarner Losh 	if (opts.create)
10781285bcc8SWarner Losh 		/*
10791285bcc8SWarner Losh 		 * side effect, adds to boot order, but not yet active.
10801285bcc8SWarner Losh 		 */
1081da15338dSWarner Losh 		make_boot_var(opts.label ? opts.label : "",
1082de26ba4dSWarner Losh 		    opts.loader, opts.kernel, opts.env, opts.dry_run,
1083de6b4aa8SWarner Losh 		    opts.has_bootnum ? opts.bootnum : -1, opts.set_active);
10841285bcc8SWarner Losh 	else if (opts.set_active || opts.set_inactive )
10851285bcc8SWarner Losh 		handle_activity(opts.bootnum, opts.set_active);
10861285bcc8SWarner Losh 	else if (opts.order != NULL)
10871285bcc8SWarner Losh 		set_boot_order(opts.order); /* create a new bootorder with opts.order */
10881285bcc8SWarner Losh 	else if (opts.set_bootnext)
10891285bcc8SWarner Losh 		handle_bootnext(opts.bootnum);
10901285bcc8SWarner Losh 	else if (opts.delete_bootnext)
10911285bcc8SWarner Losh 		del_bootnext();
10921285bcc8SWarner Losh 	else if (opts.delete)
10931285bcc8SWarner Losh 		delete_bootvar(opts.bootnum);
10941285bcc8SWarner Losh 	else if (opts.del_timeout)
10951285bcc8SWarner Losh 		delete_timeout();
10961285bcc8SWarner Losh 	else if (opts.set_timeout)
10971285bcc8SWarner Losh 		handle_timeout(opts.timeout);
10981cdb8eb8SWarner Losh 	else if (opts.esp_device)
10991cdb8eb8SWarner Losh 		report_esp_device(opts.device_path, opts.unix_path);
1100*83c42372SD Scott Phillips 	else if (opts.fw_ui)
1101*83c42372SD Scott Phillips 		set_boot_to_fw_ui(true);
1102*83c42372SD Scott Phillips 	else if (opts.no_fw_ui)
1103*83c42372SD Scott Phillips 		set_boot_to_fw_ui(false);
11041285bcc8SWarner Losh 
11051285bcc8SWarner Losh 	print_boot_vars(opts.verbose);
11061285bcc8SWarner Losh }
1107