Reviewed by: imp, zlei
Reported by: Coverity
CID: 1505695
Fixes: 9a79152994
("efibootmgr: Add --efidev (-u) to dis...")
Sponsored by: The FreeBSD Foundation
Differential Revision: https://reviews.freebsd.org/D38747
1144 lines
26 KiB
C
1144 lines
26 KiB
C
/*-
|
|
* 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>
|
|
__FBSDID("$FreeBSD$");
|
|
|
|
#include <sys/stat.h>
|
|
#include <sys/vtoc.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;
|
|
|
|
while ((ch = getopt_long(argc, argv, "AaBb:C:cdDe:EFfhk:L:l:NnOo:pTt: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;
|
|
opts.bootnum = strtoul(optarg, NULL, 16);
|
|
break;
|
|
case 'B':
|
|
opts.delete = true;
|
|
break;
|
|
case 'C':
|
|
opts.copy = true;
|
|
opts.cp_src = strtoul(optarg, NULL, 16);
|
|
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, next_free = 0;
|
|
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);
|
|
/* if the hole is at the beginning, just return zero */
|
|
if (vals[0] > 0) {
|
|
next_free = 0;
|
|
} else {
|
|
/* now just run the list looking for the first hole */
|
|
for (i = 0; i < cnt - 1 && next_free == 0; i++)
|
|
if (vals[i] + 1 != vals[i + 1])
|
|
next_free = vals[i] + 1;
|
|
if (next_free == 0)
|
|
next_free = vals[cnt - 1] + 1;
|
|
/* In theory we could have used all 65k slots -- what to do? */
|
|
}
|
|
free(vals);
|
|
|
|
asprintf(&name, "%s%04X", "Boot", next_free);
|
|
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) {
|
|
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("%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)
|
|
errx(1, "efi variables not supported on this system. root? 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);
|
|
}
|