aboutsummaryrefslogtreecommitdiff
path: root/usr.sbin/efibootmgr/efibootmgr.c
diff options
context:
space:
mode:
Diffstat (limited to 'usr.sbin/efibootmgr/efibootmgr.c')
-rw-r--r--usr.sbin/efibootmgr/efibootmgr.c1144
1 files changed, 1144 insertions, 0 deletions
diff --git a/usr.sbin/efibootmgr/efibootmgr.c b/usr.sbin/efibootmgr/efibootmgr.c
new file mode 100644
index 000000000000..b919130d9c11
--- /dev/null
+++ b/usr.sbin/efibootmgr/efibootmgr.c
@@ -0,0 +1,1144 @@
+/*-
+ * Copyright (c) 2017-2018 Netflix, Inc.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer
+ * in this position and unchanged.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include <sys/cdefs.h>
+#include <sys/stat.h>
+#include <sys/param.h>
+#include <assert.h>
+#include <ctype.h>
+#include <err.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <libgeom.h>
+#include <paths.h>
+#include <signal.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <getopt.h>
+#include <limits.h>
+#include <inttypes.h>
+#include <stdbool.h>
+#include <string.h>
+#include <strings.h>
+#include <unistd.h>
+#include <libgeom.h>
+#include <geom/geom.h>
+#include <geom/geom_ctl.h>
+#include <geom/geom_int.h>
+
+#include <efivar.h>
+#include <efiutil.h>
+#include <efichar.h>
+#include <efivar-dp.h>
+
+#ifndef LOAD_OPTION_ACTIVE
+#define LOAD_OPTION_ACTIVE 0x00000001
+#endif
+
+#ifndef LOAD_OPTION_CATEGORY_BOOT
+#define LOAD_OPTION_CATEGORY_BOOT 0x00000000
+#endif
+
+#define BAD_LENGTH ((size_t)-1)
+
+#define EFI_OS_INDICATIONS_BOOT_TO_FW_UI 0x0000000000000001
+
+typedef struct _bmgr_opts {
+ char *dev;
+ char *env;
+ char *loader;
+ char *label;
+ char *kernel;
+ char *name;
+ char *order;
+ int bootnum;
+ bool copy;
+ bool create;
+ bool delete;
+ bool delete_bootnext;
+ bool del_timeout;
+ bool dry_run;
+ bool device_path;
+ bool esp_device;
+ bool find_dev;
+ bool fw_ui;
+ bool no_fw_ui;
+ bool has_bootnum;
+ bool once;
+ int cp_src;
+ bool set_active;
+ bool set_bootnext;
+ bool set_inactive;
+ bool set_timeout;
+ int timeout;
+ bool unix_path;
+ bool verbose;
+} bmgr_opts_t;
+
+static struct option lopts[] = {
+ {"activate", no_argument, NULL, 'a'},
+ {"bootnext", no_argument, NULL, 'n'}, /* set bootnext */
+ {"bootnum", required_argument, NULL, 'b'},
+ {"bootorder", required_argument, NULL, 'o'}, /* set order */
+ {"copy", required_argument, NULL, 'C'}, /* Copy boot method */
+ {"create", no_argument, NULL, 'c'},
+ {"deactivate", no_argument, NULL, 'A'},
+ {"del-timeout", no_argument, NULL, 'T'},
+ {"delete", no_argument, NULL, 'B'},
+ {"delete-bootnext", no_argument, NULL, 'N'},
+ {"device-path", no_argument, NULL, 'd'},
+ {"dry-run", no_argument, NULL, 'D'},
+ {"env", required_argument, NULL, 'e'},
+ {"esp", no_argument, NULL, 'E'},
+ {"efidev", required_argument, NULL, 'u'},
+ {"fw-ui", no_argument, NULL, 'f'},
+ {"no-fw-ui", no_argument, NULL, 'F'},
+ {"help", no_argument, NULL, 'h'},
+ {"kernel", required_argument, NULL, 'k'},
+ {"label", required_argument, NULL, 'L'},
+ {"loader", required_argument, NULL, 'l'},
+ {"once", no_argument, NULL, 'O'},
+ {"set-timeout", required_argument, NULL, 't'},
+ {"unix-path", no_argument, NULL, 'p'},
+ {"verbose", no_argument, NULL, 'v'},
+ { NULL, 0, NULL, 0}
+};
+
+/* global efibootmgr opts */
+static bmgr_opts_t opts;
+
+static LIST_HEAD(efivars_head, entry) efivars =
+ LIST_HEAD_INITIALIZER(efivars);
+
+struct entry {
+ efi_guid_t guid;
+ uint32_t attrs;
+ uint8_t *data;
+ size_t size;
+ char *name;
+ char *label;
+ int idx;
+ int flags;
+#define SEEN 1
+
+ LIST_ENTRY(entry) entries;
+};
+
+#define MAX_DP_LEN 4096
+#define MAX_LOADOPT_LEN 8192
+
+
+static char *
+mangle_loader(char *loader)
+{
+ char *c;
+
+ for (c = loader; *c; c++)
+ if (*c == '/')
+ *c = '\\';
+
+ return loader;
+}
+
+
+#define COMMON_ATTRS EFI_VARIABLE_NON_VOLATILE | \
+ EFI_VARIABLE_BOOTSERVICE_ACCESS | \
+ EFI_VARIABLE_RUNTIME_ACCESS
+
+/*
+ * We use global guid, and common var attrs and
+ * find it better to just delete and re-create a var.
+ */
+static int
+set_bootvar(const char *name, uint8_t *data, size_t size)
+{
+
+ return efi_set_variable(EFI_GLOBAL_GUID, name, data, size,
+ COMMON_ATTRS);
+}
+
+
+#define USAGE \
+ " [-aAnB -b bootnum] [-N] [-t timeout] [-T] [-o bootorder] [-O] [--verbose] [--help]\n\
+ [-c -l loader [-k kernel] [-L label] [--dry-run] [-b bootnum]]"
+
+#define CREATE_USAGE \
+ " efibootmgr -c -l loader [-k kernel] [-L label] [--dry-run] [-b bootnum] [-a]"
+#define ORDER_USAGE \
+ " efibootmgr -o bootvarnum1,bootvarnum2,..."
+#define TIMEOUT_USAGE \
+ " efibootmgr -t seconds"
+#define DELETE_USAGE \
+ " efibootmgr -B -b bootnum"
+#define ACTIVE_USAGE \
+ " efibootmgr [-a | -A] -b bootnum"
+#define BOOTNEXT_USAGE \
+ " efibootmgr [-n | -N] -b bootnum"
+
+static void
+parse_args(int argc, char *argv[])
+{
+ int ch;
+ const char *arg;
+
+ while ((ch = getopt_long(argc, argv,
+ "AaBb:C:cdDe:EFfhk:L:l:NnOo:pTt:u:v", lopts, NULL)) != -1) {
+ switch (ch) {
+ case 'A':
+ opts.set_inactive = true;
+ break;
+ case 'a':
+ opts.set_active = true;
+ break;
+ case 'b':
+ opts.has_bootnum = true;
+ arg = optarg;
+ if (strncasecmp(arg, "boot", 4) == 0)
+ arg += 4;
+ opts.bootnum = strtoul(arg, NULL, 16);
+ break;
+ case 'B':
+ opts.delete = true;
+ break;
+ case 'C':
+ opts.copy = true;
+ opts.cp_src = strtoul(optarg, NULL, 16);
+ errx(1, "Copy not implemented");
+ break;
+ case 'c':
+ opts.create = true;
+ break;
+ case 'D': /* should be remove dups XXX */
+ opts.dry_run = true;
+ break;
+ case 'd':
+ opts.device_path = true;
+ break;
+ case 'e':
+ free(opts.env);
+ opts.env = strdup(optarg);
+ break;
+ case 'E':
+ opts.esp_device = true;
+ break;
+ case 'F':
+ opts.no_fw_ui = true;
+ break;
+ case 'f':
+ opts.fw_ui = true;
+ break;
+ case 'h':
+ default:
+ errx(1, "%s", USAGE);
+ break;
+ case 'k':
+ free(opts.kernel);
+ opts.kernel = strdup(optarg);
+ break;
+ case 'L':
+ free(opts.label);
+ opts.label = strdup(optarg);
+ break;
+ case 'l':
+ free(opts.loader);
+ opts.loader = strdup(optarg);
+ opts.loader = mangle_loader(opts.loader);
+ break;
+ case 'N':
+ opts.delete_bootnext = true;
+ break;
+ case 'n':
+ opts.set_bootnext = true;
+ break;
+ case 'O':
+ opts.once = true;
+ break;
+ case 'o':
+ free(opts.order);
+ opts.order = strdup(optarg);
+ break;
+ case 'p':
+ opts.unix_path = true;
+ break;
+ case 'T':
+ opts.del_timeout = true;
+ break;
+ case 't':
+ opts.set_timeout = true;
+ opts.timeout = strtoul(optarg, NULL, 10);
+ break;
+ case 'u':
+ opts.find_dev = true;
+ opts.dev = strdup(optarg);
+ break;
+ case 'v':
+ opts.verbose = true;
+ break;
+ }
+ }
+ if (opts.create) {
+ if (!opts.loader)
+ errx(1, "%s",CREATE_USAGE);
+ return;
+ }
+
+ if (opts.order != NULL && *opts.order == '\0')
+ errx(1, "%s", ORDER_USAGE);
+
+ if ((opts.set_inactive || opts.set_active) && !opts.has_bootnum)
+ errx(1, "%s", ACTIVE_USAGE);
+
+ if (opts.delete && !opts.has_bootnum)
+ errx(1, "%s", DELETE_USAGE);
+
+ if (opts.set_bootnext && !opts.has_bootnum)
+ errx(1, "%s", BOOTNEXT_USAGE);
+}
+
+
+static void
+read_vars(void)
+{
+
+ efi_guid_t *guid;
+ char *next_name = NULL;
+ int ret = 0;
+
+ struct entry *nent;
+
+ LIST_INIT(&efivars);
+ while ((ret = efi_get_next_variable_name(&guid, &next_name)) > 0) {
+ /*
+ * Only pay attention to EFI:BootXXXX variables to get the list.
+ */
+ if (efi_guid_cmp(guid, &EFI_GLOBAL_GUID) != 0 ||
+ strlen(next_name) != 8 ||
+ strncmp(next_name, "Boot", 4) != 0 ||
+ !isxdigit(next_name[4]) ||
+ !isxdigit(next_name[5]) ||
+ !isxdigit(next_name[6]) ||
+ !isxdigit(next_name[7]))
+ continue;
+ nent = malloc(sizeof(struct entry));
+ nent->name = strdup(next_name);
+
+ ret = efi_get_variable(*guid, next_name, &nent->data,
+ &nent->size, &nent->attrs);
+ if (ret < 0)
+ err(1, "efi_get_variable");
+ nent->guid = *guid;
+ nent->idx = strtoul(&next_name[4], NULL, 16);
+ LIST_INSERT_HEAD(&efivars, nent, entries);
+ }
+}
+
+
+static void
+set_boot_order(char *order)
+{
+ uint16_t *new_data;
+ size_t size;
+ char *next, *cp;
+ int cnt;
+ int i;
+
+ cp = order;
+ cnt = 1;
+ while (*cp) {
+ if (*cp++ == ',')
+ cnt++;
+ }
+ size = sizeof(uint16_t) * cnt;
+ new_data = malloc(size);
+
+ i = 0;
+ cp = strdup(order);
+ while ((next = strsep(&cp, ",")) != NULL) {
+ new_data[i] = strtoul(next, NULL, 16);
+ if (new_data[i] == 0 && errno == EINVAL) {
+ warnx("can't parse %s as a numb", next);
+ errx(1, "%s", ORDER_USAGE);
+ }
+ i++;
+ }
+ free(cp);
+ if (set_bootvar("BootOrder", (uint8_t*)new_data, size) < 0)
+ err(1, "Unabke to set BootOrder to %s", order);
+ free(new_data);
+}
+
+static void
+handle_activity(int bootnum, bool active)
+{
+ uint32_t attrs, load_attrs;
+ uint8_t *data;
+ size_t size;
+ char *name;
+
+ asprintf(&name, "%s%04X", "Boot", bootnum);
+ if (name == NULL)
+ err(1, "asprintf");
+ if (efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, &attrs) < 0)
+ err(1, "No such bootvar %s\n", name);
+
+ load_attrs = le32dec(data);
+
+ if (active)
+ load_attrs |= LOAD_OPTION_ACTIVE;
+ else
+ load_attrs &= ~LOAD_OPTION_ACTIVE;
+
+ le32enc(data, load_attrs);
+
+ if (set_bootvar(name, data, size) < 0)
+ err(1, "handle activity efi_set_variable");
+}
+
+
+/*
+ * add boot var to boot order.
+ * called by create boot var. There is no option
+ * to add one independent of create.
+ *
+ * Note: we currently don't support where it goes
+ * so it goes on the front, inactive.
+ * use -o 2,3,7 etc to affect order, -a to activate.
+ */
+static void
+add_to_boot_order(char *bootvar)
+{
+ size_t size;
+ uint32_t attrs;
+ uint16_t val;
+ uint8_t *data, *new;
+
+ val = strtoul(&bootvar[4], NULL, 16);
+
+ if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) < 0) {
+ if (errno == ENOENT) { /* create it and set this bootvar to active */
+ size = 0;
+ data = NULL;
+ } else
+ err(1, "efi_get_variable BootOrder");
+ }
+
+ /*
+ * We have BootOrder with the current order
+ * so grow the array by one, add the value
+ * and write the new variable value.
+ */
+ size += sizeof(uint16_t);
+ new = malloc(size);
+ if (!new)
+ err(1, "malloc");
+
+ le16enc(new, val);
+ if (size > sizeof(uint16_t))
+ memcpy(new + sizeof(uint16_t), data, size - sizeof(uint16_t));
+
+ if (set_bootvar("BootOrder", new, size) < 0)
+ err(1, "set_bootvar");
+ free(new);
+}
+
+
+static void
+remove_from_order(uint16_t bootnum)
+{
+ uint32_t attrs;
+ size_t size, i, j;
+ uint8_t *new, *data;
+
+ if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) < 0)
+ return;
+
+ new = malloc(size);
+ if (new == NULL)
+ err(1, "malloc");
+
+ for (j = i = 0; i < size; i += sizeof(uint16_t)) {
+ if (le16dec(data + i) == bootnum)
+ continue;
+ memcpy(new + j, data + i, sizeof(uint16_t));
+ j += sizeof(uint16_t);
+ }
+ if (i == j)
+ warnx("Boot variable %04x not in BootOrder", bootnum);
+ else if (set_bootvar("BootOrder", new, j) < 0)
+ err(1, "Unable to update BootOrder with new value");
+ free(new);
+}
+
+
+static void
+delete_bootvar(int bootnum)
+{
+ char *name;
+ int defer = 0;
+
+ /*
+ * Try to delete the boot variable and remocve it
+ * from the boot order. We always do both actions
+ * to make it easy to clean up from oopses.
+ */
+ if (bootnum < 0 || bootnum > 0xffff)
+ errx(1, "Bad boot variable %#x", bootnum);
+ asprintf(&name, "%s%04X", "Boot", bootnum);
+ if (name == NULL)
+ err(1, "asprintf");
+ printf("Removing boot variable '%s'\n", name);
+ if (efi_del_variable(EFI_GLOBAL_GUID, name) < 0) {
+ defer = 1;
+ warn("cannot delete variable %s", name);
+ }
+ printf("Removing 0x%x from BootOrder\n", bootnum);
+ remove_from_order(bootnum);
+ free(name);
+ if (defer)
+ exit(defer);
+}
+
+
+static void
+del_bootnext(void)
+{
+
+ if (efi_del_variable(EFI_GLOBAL_GUID, "BootNext") < 0)
+ err(1, "efi_del_variable");
+}
+
+static void
+handle_bootnext(uint16_t bootnum)
+{
+ uint16_t num;
+
+ le16enc(&num, bootnum);
+ if (set_bootvar("BootNext", (uint8_t*)&bootnum, sizeof(uint16_t)) < 0)
+ err(1, "set_bootvar");
+}
+
+
+static int
+compare(const void *a, const void *b)
+{
+ uint16_t c;
+ uint16_t d;
+
+ memcpy(&c, a, sizeof(uint16_t));
+ memcpy(&d, b, sizeof(uint16_t));
+
+ if (c < d)
+ return -1;
+ if (c == d)
+ return 0;
+ return 1;
+}
+
+static char *
+make_next_boot_var_name(void)
+{
+ struct entry *v;
+ uint16_t *vals;
+ char *name;
+ int cnt = 0;
+ int i;
+
+ LIST_FOREACH(v, &efivars, entries) {
+ cnt++;
+ }
+
+ vals = malloc(sizeof(uint16_t) * cnt);
+ if (!vals)
+ return NULL;
+
+ i = 0;
+ LIST_FOREACH(v, &efivars, entries) {
+ vals[i++] = v->idx;
+ }
+ qsort(vals, cnt, sizeof(uint16_t), compare);
+ /* Find the first hole (could be at start or end) */
+ for (i = 0; i < cnt; ++i)
+ if (vals[i] != i)
+ break;
+ free(vals);
+ /* In theory we could have used all 65k slots -- what to do? */
+
+ asprintf(&name, "%s%04X", "Boot", i);
+ if (name == NULL)
+ err(1, "asprintf");
+ return name;
+}
+
+static char *
+make_boot_var_name(uint16_t bootnum)
+{
+ struct entry *v;
+ char *name;
+
+ LIST_FOREACH(v, &efivars, entries) {
+ if (v->idx == bootnum)
+ return NULL;
+ }
+
+ asprintf(&name, "%s%04X", "Boot", bootnum);
+ if (name == NULL)
+ err(1, "asprintf");
+ return name;
+}
+
+static size_t
+create_loadopt(uint8_t *buf, size_t bufmax, uint32_t attributes, efidp dp, size_t dp_size,
+ const char *description, const uint8_t *optional_data, size_t optional_data_size)
+{
+ efi_char *bbuf = NULL;
+ uint8_t *pos = buf;
+ size_t desc_len = 0;
+ size_t len;
+
+ if (optional_data == NULL && optional_data_size != 0)
+ return BAD_LENGTH;
+ if (dp == NULL && dp_size != 0)
+ return BAD_LENGTH;
+
+ /*
+ * Compute the length to make sure the passed in buffer is long enough.
+ */
+ utf8_to_ucs2(description, &bbuf, &desc_len);
+ len = sizeof(uint32_t) + sizeof(uint16_t) + desc_len + dp_size + optional_data_size;
+ if (len > bufmax) {
+ free(bbuf);
+ return BAD_LENGTH;
+ }
+
+ le32enc(pos, attributes);
+ pos += sizeof (attributes);
+
+ le16enc(pos, dp_size);
+ pos += sizeof (uint16_t);
+
+ memcpy(pos, bbuf, desc_len); /* NB:desc_len includes strailing NUL */
+ pos += desc_len;
+ free(bbuf);
+
+ memcpy(pos, dp, dp_size);
+ pos += dp_size;
+
+ if (optional_data && optional_data_size > 0) {
+ memcpy(pos, optional_data, optional_data_size);
+ pos += optional_data_size;
+ }
+
+ return pos - buf;
+}
+
+
+static int
+make_boot_var(const char *label, const char *loader, const char *kernel, const char *env, bool dry_run,
+ int bootnum, bool activate)
+{
+ struct entry *new_ent;
+ uint32_t load_attrs = 0;
+ uint8_t *load_opt_buf;
+ size_t lopt_size, llen, klen;
+ efidp dp, loaderdp, kerneldp;
+ char *bootvar = NULL;
+ int ret;
+
+ assert(label != NULL);
+
+ if (bootnum == -1)
+ bootvar = make_next_boot_var_name();
+ else
+ bootvar = make_boot_var_name((uint16_t)bootnum);
+ if (bootvar == NULL)
+ err(1, "bootvar creation");
+ if (loader == NULL)
+ errx(1, "Must specify boot loader");
+ ret = efivar_unix_path_to_device_path(loader, &loaderdp);
+ if (ret != 0)
+ errc(1, ret, "Cannot translate unix loader path '%s' to UEFI",
+ loader);
+ if (kernel != NULL) {
+ ret = efivar_unix_path_to_device_path(kernel, &kerneldp);
+ if (ret != 0)
+ errc(1, ret,
+ "Cannot translate unix kernel path '%s' to UEFI",
+ kernel);
+ } else {
+ kerneldp = NULL;
+ }
+ llen = efidp_size(loaderdp);
+ if (llen > MAX_DP_LEN)
+ errx(1, "Loader path too long.");
+ klen = efidp_size(kerneldp);
+ if (klen > MAX_DP_LEN)
+ errx(1, "Kernel path too long.");
+ dp = malloc(llen + klen);
+ if (dp == NULL)
+ errx(1, "Can't allocate memory for new device paths");
+ memcpy(dp, loaderdp, llen);
+ if (kerneldp != NULL)
+ memcpy((char *)dp + llen, kerneldp, klen);
+
+ /* don't make the new bootvar active by default, use the -a option later */
+ load_attrs = LOAD_OPTION_CATEGORY_BOOT;
+ if (activate)
+ load_attrs |= LOAD_OPTION_ACTIVE;
+ load_opt_buf = malloc(MAX_LOADOPT_LEN);
+ if (load_opt_buf == NULL)
+ err(1, "malloc");
+
+ lopt_size = create_loadopt(load_opt_buf, MAX_LOADOPT_LEN, load_attrs,
+ dp, llen + klen, label, env, env ? strlen(env) + 1 : 0);
+ if (lopt_size == BAD_LENGTH)
+ errx(1, "Can't create loadopt");
+
+ ret = 0;
+ if (!dry_run) {
+ ret = efi_set_variable(EFI_GLOBAL_GUID, bootvar,
+ (uint8_t*)load_opt_buf, lopt_size, COMMON_ATTRS);
+ }
+
+ if (ret)
+ err(1, "efi_set_variable");
+
+ if (!dry_run)
+ add_to_boot_order(bootvar); /* first, still not active */
+ new_ent = malloc(sizeof(struct entry));
+ if (new_ent == NULL)
+ err(1, "malloc");
+ memset(new_ent, 0, sizeof(struct entry));
+ new_ent->name = bootvar;
+ new_ent->guid = EFI_GLOBAL_GUID;
+ LIST_INSERT_HEAD(&efivars, new_ent, entries);
+ free(load_opt_buf);
+ free(dp);
+
+ return 0;
+}
+
+
+static void
+print_loadopt_str(uint8_t *data, size_t datalen)
+{
+ char *dev, *relpath, *abspath;
+ uint32_t attr;
+ uint16_t fplen;
+ efi_char *descr;
+ uint8_t *ep = data + datalen;
+ uint8_t *walker = data;
+ efidp dp, edp;
+ char buf[1024];
+ int len;
+ int rv;
+ int indent;
+
+ if (datalen < sizeof(attr) + sizeof(fplen) + sizeof(efi_char))
+ return;
+ // First 4 bytes are attribute flags
+ attr = le32dec(walker);
+ walker += sizeof(attr);
+ // Next two bytes are length of the file paths
+ fplen = le16dec(walker);
+ walker += sizeof(fplen);
+ // Next we have a 0 terminated UCS2 string that we know to be aligned
+ descr = (efi_char *)(intptr_t)(void *)walker;
+ len = ucs2len(descr); // XXX need to sanity check that len < (datalen - (ep - walker) / 2)
+ walker += (len + 1) * sizeof(efi_char);
+ if (walker > ep)
+ return;
+ // Now we have fplen bytes worth of file path stuff
+ dp = (efidp)walker;
+ walker += fplen;
+ if (walker > ep)
+ return;
+ edp = (efidp)walker;
+ /*
+ * Everything left is the binary option args
+ * opt = walker;
+ * optlen = ep - walker;
+ */
+ indent = 1;
+ while (dp < edp) {
+ if (efidp_size(dp) == 0)
+ break;
+ efidp_format_device_path(buf, sizeof(buf), dp,
+ (intptr_t)(void *)edp - (intptr_t)(void *)dp);
+ printf("%*s%s\n", indent, "", buf);
+ indent = 10 + len + 1;
+ rv = efivar_device_path_to_unix_path(dp, &dev, &relpath, &abspath);
+ if (rv == 0) {
+ printf("%*s%s:%s %s\n", indent + 4, "", dev, relpath, abspath);
+ free(dev);
+ free(relpath);
+ free(abspath);
+ }
+ dp = (efidp)((char *)dp + efidp_size(dp));
+ }
+}
+
+static char *
+get_descr(uint8_t *data)
+{
+ uint8_t *pos = data;
+ efi_char *desc;
+ int len;
+ char *buf;
+ int i = 0;
+
+ pos += sizeof(uint32_t) + sizeof(uint16_t);
+ desc = (efi_char*)(intptr_t)(void *)pos;
+ len = ucs2len(desc);
+ buf = malloc(len + 1);
+ memset(buf, 0, len + 1);
+ while (desc[i]) {
+ buf[i] = desc[i];
+ i++;
+ }
+ return (char*)buf;
+}
+
+
+static bool
+print_boot_var(const char *name, bool verbose, bool curboot)
+{
+ size_t size;
+ uint32_t load_attrs;
+ uint8_t *data;
+ int ret;
+ char *d;
+
+ ret = efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, NULL);
+ if (ret < 0)
+ return false;
+ load_attrs = le32dec(data);
+ d = get_descr(data);
+ printf("%c%s%c %s", curboot ? '+' : ' ', name,
+ ((load_attrs & LOAD_OPTION_ACTIVE) ? '*': ' '), d);
+ free(d);
+ if (verbose)
+ print_loadopt_str(data, size);
+ else
+ printf("\n");
+ return true;
+}
+
+
+static bool
+os_indication_supported(uint64_t indication)
+{
+ uint8_t *data;
+ size_t size;
+ uint32_t attrs;
+ int ret;
+
+ ret = efi_get_variable(EFI_GLOBAL_GUID, "OsIndicationsSupported", &data,
+ &size, &attrs);
+ if (ret < 0)
+ return false;
+ return (le64dec(data) & indication) == indication;
+}
+
+static uint64_t
+os_indications(void)
+{
+ uint8_t *data;
+ size_t size;
+ uint32_t attrs;
+ int ret;
+
+ ret = efi_get_variable(EFI_GLOBAL_GUID, "OsIndications", &data, &size,
+ &attrs);
+ if (ret < 0)
+ return 0;
+ return le64dec(data);
+}
+
+static int
+os_indications_set(uint64_t mask, uint64_t val)
+{
+ uint8_t new[sizeof(uint64_t)];
+
+ le64enc(&new, (os_indications() & ~mask) | (val & mask));
+ return set_bootvar("OsIndications", new, sizeof(new));
+}
+
+/* Cmd epilogue, or just the default with no args.
+ * The order is [bootnext] bootcurrent, timeout, order, and the bootvars [-v]
+ */
+static int
+print_boot_vars(bool verbose)
+{
+ /*
+ * just read and print the current values
+ * as a command epilogue
+ */
+ struct entry *v;
+ uint8_t *data;
+ size_t size;
+ uint32_t attrs;
+ int ret, bolen;
+ uint16_t *boot_order = NULL, current;
+ bool boot_to_fw_ui;
+
+ if (os_indication_supported(EFI_OS_INDICATIONS_BOOT_TO_FW_UI)) {
+ boot_to_fw_ui =
+ (os_indications() & EFI_OS_INDICATIONS_BOOT_TO_FW_UI) != 0;
+ printf("Boot to FW : %s\n", boot_to_fw_ui != 0 ?
+ "true" : "false");
+ }
+
+ ret = efi_get_variable(EFI_GLOBAL_GUID, "BootNext", &data, &size, &attrs);
+ if (ret > 0) {
+ printf("BootNext : %04x\n", le16dec(data));
+ }
+
+ ret = efi_get_variable(EFI_GLOBAL_GUID, "BootCurrent", &data, &size,&attrs);
+ current = le16dec(data);
+ printf("BootCurrent: %04x\n", current);
+
+ ret = efi_get_variable(EFI_GLOBAL_GUID, "Timeout", &data, &size, &attrs);
+ if (ret > 0) {
+ printf("Timeout : %d seconds\n", le16dec(data));
+ }
+
+ if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) > 0) {
+ if (size % 2 == 1)
+ warn("Bad BootOrder variable: odd length %d", (int)size);
+ boot_order = malloc(size);
+ bolen = size / 2;
+ printf("BootOrder : ");
+ for (size_t i = 0; i < size; i += 2) {
+ boot_order[i / 2] = le16dec(data + i);
+ printf("%04X%s", boot_order[i / 2], i == size - 2 ? "\n" : ", ");
+ }
+ }
+
+ if (boot_order == NULL) {
+ /*
+ * now we want to fetch 'em all fresh again
+ * which possibly includes a newly created bootvar
+ */
+ LIST_FOREACH(v, &efivars, entries) {
+ print_boot_var(v->name, verbose, v->idx == current);
+ }
+ } else {
+ LIST_FOREACH(v, &efivars, entries) {
+ v->flags = 0;
+ }
+ for (int i = 0; i < bolen; i++) {
+ char buffer[10];
+
+ snprintf(buffer, sizeof(buffer), "Boot%04X", boot_order[i]);
+ if (!print_boot_var(buffer, verbose, boot_order[i] == current))
+ printf("%s: MISSING!\n", buffer);
+ LIST_FOREACH(v, &efivars, entries) {
+ if (v->idx == boot_order[i]) {
+ v->flags |= SEEN;
+ break;
+ }
+ }
+ }
+ if (verbose) {
+ printf("\n\nUnreferenced Variables:\n");
+ LIST_FOREACH(v, &efivars, entries) {
+ if (v->flags == 0)
+ print_boot_var(v->name, verbose, v->idx == current);
+ }
+ }
+ }
+ return 0;
+}
+
+static void
+delete_timeout(void)
+{
+
+ efi_del_variable(EFI_GLOBAL_GUID,"Timeout");
+}
+
+static void
+handle_timeout(int to)
+{
+ uint16_t timeout;
+
+ le16enc(&timeout, to);
+ if (set_bootvar("Timeout", (uint8_t *)&timeout, sizeof(timeout)) < 0)
+ errx(1, "Can't set Timeout for booting.");
+}
+
+static void
+report_esp_device(bool do_dp, bool do_unix)
+{
+ uint8_t *data;
+ size_t size, len;
+ uint32_t attrs;
+ int ret;
+ uint16_t current, fplen;
+ char *name, *dev, *relpath, *abspath;
+ uint8_t *walker, *ep;
+ efi_char *descr;
+ efidp dp;
+ char buf[PATH_MAX];
+
+ if (do_dp && do_unix)
+ errx(1, "Can't report both UEFI device-path and Unix path together");
+
+ ret = efi_get_variable(EFI_GLOBAL_GUID, "BootCurrent", &data, &size,&attrs);
+ if (ret < 0)
+ err(1, "Can't get BootCurrent");
+ current = le16dec(data);
+ if (asprintf(&name, "Boot%04X", current) < 0)
+ err(1, "Can't format boot var\n");
+ if (efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, NULL) < 0)
+ err(1, "Can't retrieve EFI var %s", name);
+ // First 4 bytes are attribute flags
+ walker = data;
+ ep = walker + size;
+ walker += sizeof(uint32_t);
+ // Next two bytes are length of the file paths
+ fplen = le16dec(walker);
+ walker += sizeof(fplen);
+ // Next we have a 0 terminated UCS2 string that we know to be aligned
+ descr = (efi_char *)(intptr_t)(void *)walker;
+ len = ucs2len(descr); // XXX need to sanity check that len < (datalen - (ep - walker) / 2)
+ walker += (len + 1) * sizeof(efi_char);
+ if (walker > ep)
+ errx(1, "malformed boot variable %s", name);
+ // Now we have fplen bytes worth of file path stuff
+ dp = (efidp)walker;
+ walker += fplen;
+ if (walker > ep)
+ errx(1, "malformed boot variable %s", name);
+ if (do_dp) {
+ efidp_format_device_path_node(buf, sizeof(buf), dp);
+ printf("%s\n", buf);
+ exit(0);
+ }
+ if (efivar_device_path_to_unix_path(dp, &dev, &relpath, &abspath) != 0)
+ errx(1, "Can't convert to unix path");
+ if (do_unix) {
+ if (abspath == NULL)
+ errx(1, "Can't find where %s:%s is mounted",
+ dev, relpath);
+ abspath[strlen(abspath) - strlen(relpath) - 1] = '\0';
+ printf("%s\n", abspath);
+ } else {
+ printf("/dev/%s\n", dev);
+ }
+ free(dev);
+ free(relpath);
+ free(abspath);
+ exit(0);
+}
+
+static void
+set_boot_to_fw_ui(bool to_fw)
+{
+ int ret;
+
+ if (!os_indication_supported(EFI_OS_INDICATIONS_BOOT_TO_FW_UI)) {
+ if (to_fw)
+ errx(1, "boot to fw ui not supported");
+ else
+ return;
+ }
+ ret = os_indications_set(EFI_OS_INDICATIONS_BOOT_TO_FW_UI,
+ to_fw ? ~0 : 0);
+ if (ret < 0)
+ errx(1, "failed to set boot to fw ui");
+}
+
+static void
+find_efi_device(const char *path)
+{
+ efidp dp = NULL;
+ size_t len;
+ int ret;
+ char buf[1024];
+
+ ret = efivar_unix_path_to_device_path(path, &dp);
+ if (ret != 0)
+ errc(1, ret,
+ "Cannot translate path '%s' to UEFI", path);
+ len = efidp_size(dp);
+ if (len > MAX_DP_LEN)
+ errx(1, "Resulting device path too long.");
+ efidp_format_device_path(buf, sizeof(buf), dp, len);
+ printf("%s -> %s\n", path, buf);
+ exit (0);
+}
+
+int
+main(int argc, char *argv[])
+{
+
+ memset(&opts, 0, sizeof (bmgr_opts_t));
+ parse_args(argc, argv);
+
+ /*
+ * find_dev can operate without any efi variables
+ */
+ if (!efi_variables_supported() && !opts.find_dev) {
+ if (errno == EACCES && geteuid() != 0)
+ errx(1, "must be run as root");
+ errx(1, "efi variables not supported on this system. kldload efirt?");
+ }
+
+ read_vars();
+
+ if (opts.create)
+ /*
+ * side effect, adds to boot order, but not yet active.
+ */
+ make_boot_var(opts.label ? opts.label : "",
+ opts.loader, opts.kernel, opts.env, opts.dry_run,
+ opts.has_bootnum ? opts.bootnum : -1, opts.set_active);
+ else if (opts.set_active || opts.set_inactive )
+ handle_activity(opts.bootnum, opts.set_active);
+ else if (opts.order != NULL)
+ set_boot_order(opts.order); /* create a new bootorder with opts.order */
+ else if (opts.set_bootnext)
+ handle_bootnext(opts.bootnum);
+ else if (opts.delete_bootnext)
+ del_bootnext();
+ else if (opts.delete)
+ delete_bootvar(opts.bootnum);
+ else if (opts.del_timeout)
+ delete_timeout();
+ else if (opts.set_timeout)
+ handle_timeout(opts.timeout);
+ else if (opts.esp_device)
+ report_esp_device(opts.device_path, opts.unix_path);
+ else if (opts.fw_ui)
+ set_boot_to_fw_ui(true);
+ else if (opts.no_fw_ui)
+ set_boot_to_fw_ui(false);
+ else if (opts.find_dev)
+ find_efi_device(opts.dev);
+
+ print_boot_vars(opts.verbose);
+}