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 7083c42372SD Scott Phillips #define EFI_OS_INDICATIONS_BOOT_TO_FW_UI 0x0000000000000001 7183c42372SD Scott Phillips 721285bcc8SWarner Losh typedef struct _bmgr_opts { 739a791529SWarner Losh char *dev; 741285bcc8SWarner Losh char *env; 751285bcc8SWarner Losh char *loader; 761285bcc8SWarner Losh char *label; 771285bcc8SWarner Losh char *kernel; 781285bcc8SWarner Losh char *name; 791285bcc8SWarner Losh char *order; 801285bcc8SWarner Losh int bootnum; 811285bcc8SWarner Losh bool copy; 821285bcc8SWarner Losh bool create; 831285bcc8SWarner Losh bool delete; 841285bcc8SWarner Losh bool delete_bootnext; 851285bcc8SWarner Losh bool del_timeout; 861285bcc8SWarner Losh bool dry_run; 871cdb8eb8SWarner Losh bool device_path; 881cdb8eb8SWarner Losh bool esp_device; 899a791529SWarner Losh bool find_dev; 9083c42372SD Scott Phillips bool fw_ui; 9183c42372SD Scott Phillips bool no_fw_ui; 92de26ba4dSWarner Losh bool has_bootnum; 931285bcc8SWarner Losh bool once; 941285bcc8SWarner Losh int cp_src; 951285bcc8SWarner Losh bool set_active; 961285bcc8SWarner Losh bool set_bootnext; 971285bcc8SWarner Losh bool set_inactive; 981285bcc8SWarner Losh bool set_timeout; 991285bcc8SWarner Losh int timeout; 1001cdb8eb8SWarner Losh bool unix_path; 1011285bcc8SWarner Losh bool verbose; 1021285bcc8SWarner Losh } bmgr_opts_t; 1031285bcc8SWarner Losh 1041285bcc8SWarner Losh static struct option lopts[] = { 1059ed08029SRebecca Cran {"activate", no_argument, NULL, 'a'}, 1069ed08029SRebecca Cran {"bootnext", no_argument, NULL, 'n'}, /* set bootnext */ 107a2581e80SRebecca Cran {"bootnum", required_argument, NULL, 'b'}, 1081285bcc8SWarner Losh {"bootorder", required_argument, NULL, 'o'}, /* set order */ 1091285bcc8SWarner Losh {"copy", required_argument, NULL, 'C'}, /* Copy boot method */ 1101285bcc8SWarner Losh {"create", no_argument, NULL, 'c'}, 1119ed08029SRebecca Cran {"deactivate", no_argument, NULL, 'A'}, 1126b8b1bfeSToomas Soome {"del-timeout", no_argument, NULL, 'T'}, 1139ed08029SRebecca Cran {"delete", no_argument, NULL, 'B'}, 1149ed08029SRebecca Cran {"delete-bootnext", no_argument, NULL, 'N'}, 1151cdb8eb8SWarner Losh {"device-path", no_argument, NULL, 'd'}, 1161285bcc8SWarner Losh {"dry-run", no_argument, NULL, 'D'}, 1171285bcc8SWarner Losh {"env", required_argument, NULL, 'e'}, 1181cdb8eb8SWarner Losh {"esp", no_argument, NULL, 'E'}, 1199a791529SWarner Losh {"efidev", required_argument, NULL, 'u'}, 12083c42372SD Scott Phillips {"fw-ui", no_argument, NULL, 'f'}, 12183c42372SD Scott Phillips {"no-fw-ui", no_argument, NULL, 'F'}, 1221285bcc8SWarner Losh {"help", no_argument, NULL, 'h'}, 1231285bcc8SWarner Losh {"kernel", required_argument, NULL, 'k'}, 1241285bcc8SWarner Losh {"label", required_argument, NULL, 'L'}, 1251285bcc8SWarner Losh {"loader", required_argument, NULL, 'l'}, 1261285bcc8SWarner Losh {"once", no_argument, NULL, 'O'}, 1271285bcc8SWarner Losh {"set-timeout", required_argument, NULL, 't'}, 1281cdb8eb8SWarner Losh {"unix-path", no_argument, NULL, 'p'}, 1291285bcc8SWarner Losh {"verbose", no_argument, NULL, 'v'}, 1301285bcc8SWarner Losh { NULL, 0, NULL, 0} 1311285bcc8SWarner Losh }; 1321285bcc8SWarner Losh 1331285bcc8SWarner Losh /* global efibootmgr opts */ 1341285bcc8SWarner Losh static bmgr_opts_t opts; 1351285bcc8SWarner Losh 1361285bcc8SWarner Losh static LIST_HEAD(efivars_head, entry) efivars = 1371285bcc8SWarner Losh LIST_HEAD_INITIALIZER(efivars); 1381285bcc8SWarner Losh 1391285bcc8SWarner Losh struct entry { 1401285bcc8SWarner Losh efi_guid_t guid; 1411285bcc8SWarner Losh uint32_t attrs; 1421285bcc8SWarner Losh uint8_t *data; 1431285bcc8SWarner Losh size_t size; 1441285bcc8SWarner Losh char *name; 1451285bcc8SWarner Losh char *label; 1461285bcc8SWarner Losh int idx; 14751922c69SWarner Losh int flags; 14851922c69SWarner Losh #define SEEN 1 1491285bcc8SWarner Losh 1501285bcc8SWarner Losh LIST_ENTRY(entry) entries; 1511285bcc8SWarner Losh }; 1521285bcc8SWarner Losh 1531285bcc8SWarner Losh #define MAX_DP_LEN 4096 1541285bcc8SWarner Losh #define MAX_LOADOPT_LEN 8192 1551285bcc8SWarner Losh 1561285bcc8SWarner Losh 1571285bcc8SWarner Losh static char * 1581285bcc8SWarner Losh mangle_loader(char *loader) 1591285bcc8SWarner Losh { 1601285bcc8SWarner Losh char *c; 1611285bcc8SWarner Losh 1621285bcc8SWarner Losh for (c = loader; *c; c++) 1631285bcc8SWarner Losh if (*c == '/') 1641285bcc8SWarner Losh *c = '\\'; 1651285bcc8SWarner Losh 1661285bcc8SWarner Losh return loader; 1671285bcc8SWarner Losh } 1681285bcc8SWarner Losh 1691285bcc8SWarner Losh 1701285bcc8SWarner Losh #define COMMON_ATTRS EFI_VARIABLE_NON_VOLATILE | \ 1711285bcc8SWarner Losh EFI_VARIABLE_BOOTSERVICE_ACCESS | \ 1721285bcc8SWarner Losh EFI_VARIABLE_RUNTIME_ACCESS 1731285bcc8SWarner Losh 1741285bcc8SWarner Losh /* 1751285bcc8SWarner Losh * We use global guid, and common var attrs and 1761285bcc8SWarner Losh * find it better to just delete and re-create a var. 1771285bcc8SWarner Losh */ 1781285bcc8SWarner Losh static int 1791285bcc8SWarner Losh set_bootvar(const char *name, uint8_t *data, size_t size) 1801285bcc8SWarner Losh { 1811285bcc8SWarner Losh 1821285bcc8SWarner Losh return efi_set_variable(EFI_GLOBAL_GUID, name, data, size, 1831285bcc8SWarner Losh COMMON_ATTRS); 1841285bcc8SWarner Losh } 1851285bcc8SWarner Losh 1861285bcc8SWarner Losh 1871285bcc8SWarner Losh #define USAGE \ 1889ed08029SRebecca Cran " [-aAnB -b bootnum] [-N] [-t timeout] [-T] [-o bootorder] [-O] [--verbose] [--help]\n\ 1899ed08029SRebecca Cran [-c -l loader [-k kernel] [-L label] [--dry-run] [-b bootnum]]" 1901285bcc8SWarner Losh 1911285bcc8SWarner Losh #define CREATE_USAGE \ 1929ed08029SRebecca Cran " efibootmgr -c -l loader [-k kernel] [-L label] [--dry-run] [-b bootnum] [-a]" 1931285bcc8SWarner Losh #define ORDER_USAGE \ 1941285bcc8SWarner Losh " efibootmgr -o bootvarnum1,bootvarnum2,..." 1951285bcc8SWarner Losh #define TIMEOUT_USAGE \ 1961285bcc8SWarner Losh " efibootmgr -t seconds" 1971285bcc8SWarner Losh #define DELETE_USAGE \ 1989ed08029SRebecca Cran " efibootmgr -B -b bootnum" 1991285bcc8SWarner Losh #define ACTIVE_USAGE \ 2009ed08029SRebecca Cran " efibootmgr [-a | -A] -b bootnum" 2011285bcc8SWarner Losh #define BOOTNEXT_USAGE \ 2029ed08029SRebecca Cran " efibootmgr [-n | -N] -b bootnum" 2031285bcc8SWarner Losh 2041285bcc8SWarner Losh static void 2051285bcc8SWarner Losh parse_args(int argc, char *argv[]) 2061285bcc8SWarner Losh { 2071285bcc8SWarner Losh int ch; 2081285bcc8SWarner Losh 20983c42372SD Scott Phillips while ((ch = getopt_long(argc, argv, "AaBb:C:cdDe:EFfhk:L:l:NnOo:pTt:v", 2101285bcc8SWarner Losh lopts, NULL)) != -1) { 2111285bcc8SWarner Losh switch (ch) { 2121285bcc8SWarner Losh case 'A': 2131285bcc8SWarner Losh opts.set_inactive = true; 2141285bcc8SWarner Losh break; 2151285bcc8SWarner Losh case 'a': 2161285bcc8SWarner Losh opts.set_active = true; 2171285bcc8SWarner Losh break; 218de26ba4dSWarner Losh case 'b': 219de26ba4dSWarner Losh opts.has_bootnum = true; 220de26ba4dSWarner Losh opts.bootnum = strtoul(optarg, NULL, 16); 221de26ba4dSWarner Losh break; 2221285bcc8SWarner Losh case 'B': 2231285bcc8SWarner Losh opts.delete = true; 2241285bcc8SWarner Losh break; 2251285bcc8SWarner Losh case 'C': 2261285bcc8SWarner Losh opts.copy = true; 2271285bcc8SWarner Losh opts.cp_src = strtoul(optarg, NULL, 16); 2281285bcc8SWarner Losh case 'c': 2291285bcc8SWarner Losh opts.create = true; 2301285bcc8SWarner Losh break; 2311285bcc8SWarner Losh case 'D': /* should be remove dups XXX */ 2321285bcc8SWarner Losh opts.dry_run = true; 2331285bcc8SWarner Losh break; 2341cdb8eb8SWarner Losh case 'd': 2351cdb8eb8SWarner Losh opts.device_path = true; 2361cdb8eb8SWarner Losh break; 2371285bcc8SWarner Losh case 'e': 23863b56694SWarner Losh free(opts.env); 2391285bcc8SWarner Losh opts.env = strdup(optarg); 2401285bcc8SWarner Losh break; 2411cdb8eb8SWarner Losh case 'E': 2421cdb8eb8SWarner Losh opts.esp_device = true; 2431cdb8eb8SWarner Losh break; 24483c42372SD Scott Phillips case 'F': 24583c42372SD Scott Phillips opts.no_fw_ui = true; 24683c42372SD Scott Phillips break; 24783c42372SD Scott Phillips case 'f': 24883c42372SD Scott Phillips opts.fw_ui = true; 24983c42372SD Scott Phillips break; 2501285bcc8SWarner Losh case 'h': 2511285bcc8SWarner Losh default: 2521285bcc8SWarner Losh errx(1, "%s", USAGE); 2531285bcc8SWarner Losh break; 2541285bcc8SWarner Losh case 'k': 25563b56694SWarner Losh free(opts.kernel); 2561285bcc8SWarner Losh opts.kernel = strdup(optarg); 2571285bcc8SWarner Losh break; 2581285bcc8SWarner Losh case 'L': 25963b56694SWarner Losh free(opts.label); 2601285bcc8SWarner Losh opts.label = strdup(optarg); 2611285bcc8SWarner Losh break; 2621285bcc8SWarner Losh case 'l': 26363b56694SWarner Losh free(opts.loader); 2641285bcc8SWarner Losh opts.loader = strdup(optarg); 2651285bcc8SWarner Losh opts.loader = mangle_loader(opts.loader); 2661285bcc8SWarner Losh break; 2671285bcc8SWarner Losh case 'N': 2681285bcc8SWarner Losh opts.delete_bootnext = true; 2691285bcc8SWarner Losh break; 2701285bcc8SWarner Losh case 'n': 2711285bcc8SWarner Losh opts.set_bootnext = true; 2721285bcc8SWarner Losh break; 2731285bcc8SWarner Losh case 'O': 2741285bcc8SWarner Losh opts.once = true; 2751285bcc8SWarner Losh break; 2761285bcc8SWarner Losh case 'o': 27763b56694SWarner Losh free(opts.order); 2781285bcc8SWarner Losh opts.order = strdup(optarg); 2791285bcc8SWarner Losh break; 2801cdb8eb8SWarner Losh case 'p': 2811cdb8eb8SWarner Losh opts.unix_path = true; 2821cdb8eb8SWarner Losh break; 2831285bcc8SWarner Losh case 'T': 2841285bcc8SWarner Losh opts.del_timeout = true; 2851285bcc8SWarner Losh break; 2861285bcc8SWarner Losh case 't': 2871285bcc8SWarner Losh opts.set_timeout = true; 2881285bcc8SWarner Losh opts.timeout = strtoul(optarg, NULL, 10); 2891285bcc8SWarner Losh break; 2909a791529SWarner Losh case 'u': 2919a791529SWarner Losh opts.find_dev = true; 2929a791529SWarner Losh opts.dev = strdup(optarg); 293*17a5a290SEd Maste break; 2941285bcc8SWarner Losh case 'v': 2951285bcc8SWarner Losh opts.verbose = true; 2961285bcc8SWarner Losh break; 2971285bcc8SWarner Losh } 2981285bcc8SWarner Losh } 2991285bcc8SWarner Losh if (opts.create) { 3001127aea3SWarner Losh if (!opts.loader) 3011285bcc8SWarner Losh errx(1, "%s",CREATE_USAGE); 3021285bcc8SWarner Losh return; 3031285bcc8SWarner Losh } 3041285bcc8SWarner Losh 305b9580775SToomas Soome if (opts.order != NULL && *opts.order == '\0') 3061285bcc8SWarner Losh errx(1, "%s", ORDER_USAGE); 3079ed08029SRebecca Cran 3089ed08029SRebecca Cran if ((opts.set_inactive || opts.set_active) && !opts.has_bootnum) 3099ed08029SRebecca Cran errx(1, "%s", ACTIVE_USAGE); 3109ed08029SRebecca Cran 3119ed08029SRebecca Cran if (opts.delete && !opts.has_bootnum) 3129ed08029SRebecca Cran errx(1, "%s", DELETE_USAGE); 3139ed08029SRebecca Cran 3149ed08029SRebecca Cran if (opts.set_bootnext && !opts.has_bootnum) 3159ed08029SRebecca Cran errx(1, "%s", BOOTNEXT_USAGE); 3161285bcc8SWarner Losh } 3171285bcc8SWarner Losh 3181285bcc8SWarner Losh 3191285bcc8SWarner Losh static void 3201285bcc8SWarner Losh read_vars(void) 3211285bcc8SWarner Losh { 3221285bcc8SWarner Losh 3231285bcc8SWarner Losh efi_guid_t *guid; 3241285bcc8SWarner Losh char *next_name = NULL; 3251285bcc8SWarner Losh int ret = 0; 3261285bcc8SWarner Losh 3271285bcc8SWarner Losh struct entry *nent; 3281285bcc8SWarner Losh 3291285bcc8SWarner Losh LIST_INIT(&efivars); 3301285bcc8SWarner Losh while ((ret = efi_get_next_variable_name(&guid, &next_name)) > 0) { 3311285bcc8SWarner Losh /* 332a3e6c4a6SWarner Losh * Only pay attention to EFI:BootXXXX variables to get the list. 3331285bcc8SWarner Losh */ 3341285bcc8SWarner Losh if (efi_guid_cmp(guid, &EFI_GLOBAL_GUID) != 0 || 3351285bcc8SWarner Losh strlen(next_name) != 8 || 3361285bcc8SWarner Losh strncmp(next_name, "Boot", 4) != 0 || 3371285bcc8SWarner Losh !isxdigit(next_name[4]) || 3381285bcc8SWarner Losh !isxdigit(next_name[5]) || 3391285bcc8SWarner Losh !isxdigit(next_name[6]) || 3401285bcc8SWarner Losh !isxdigit(next_name[7])) 3411285bcc8SWarner Losh continue; 3421285bcc8SWarner Losh nent = malloc(sizeof(struct entry)); 3431285bcc8SWarner Losh nent->name = strdup(next_name); 3441285bcc8SWarner Losh 3451285bcc8SWarner Losh ret = efi_get_variable(*guid, next_name, &nent->data, 3461285bcc8SWarner Losh &nent->size, &nent->attrs); 3471285bcc8SWarner Losh if (ret < 0) 3481285bcc8SWarner Losh err(1, "efi_get_variable"); 3491285bcc8SWarner Losh nent->guid = *guid; 3501285bcc8SWarner Losh nent->idx = strtoul(&next_name[4], NULL, 16); 3511285bcc8SWarner Losh LIST_INSERT_HEAD(&efivars, nent, entries); 3521285bcc8SWarner Losh } 3531285bcc8SWarner Losh } 3541285bcc8SWarner Losh 3551285bcc8SWarner Losh 3561285bcc8SWarner Losh static void 3571285bcc8SWarner Losh set_boot_order(char *order) 3581285bcc8SWarner Losh { 3591285bcc8SWarner Losh uint16_t *new_data; 3601285bcc8SWarner Losh size_t size; 3611285bcc8SWarner Losh char *next, *cp; 3621285bcc8SWarner Losh int cnt; 3631285bcc8SWarner Losh int i; 3641285bcc8SWarner Losh 3651285bcc8SWarner Losh cp = order; 3661285bcc8SWarner Losh cnt = 1; 3671285bcc8SWarner Losh while (*cp) { 3681285bcc8SWarner Losh if (*cp++ == ',') 3691285bcc8SWarner Losh cnt++; 3701285bcc8SWarner Losh } 3711285bcc8SWarner Losh size = sizeof(uint16_t) * cnt; 3721285bcc8SWarner Losh new_data = malloc(size); 3731285bcc8SWarner Losh 3741285bcc8SWarner Losh i = 0; 3751285bcc8SWarner Losh cp = strdup(order); 3761285bcc8SWarner Losh while ((next = strsep(&cp, ",")) != NULL) { 3771285bcc8SWarner Losh new_data[i] = strtoul(next, NULL, 16); 3781285bcc8SWarner Losh if (new_data[i] == 0 && errno == EINVAL) { 3791285bcc8SWarner Losh warnx("can't parse %s as a numb", next); 3801285bcc8SWarner Losh errx(1, "%s", ORDER_USAGE); 3811285bcc8SWarner Losh } 3821285bcc8SWarner Losh i++; 3831285bcc8SWarner Losh } 3841285bcc8SWarner Losh free(cp); 3851285bcc8SWarner Losh if (set_bootvar("BootOrder", (uint8_t*)new_data, size) < 0) 3861285bcc8SWarner Losh err(1, "Unabke to set BootOrder to %s", order); 3873af42ca7SWarner Losh free(new_data); 3881285bcc8SWarner Losh } 3891285bcc8SWarner Losh 3901285bcc8SWarner Losh static void 3911285bcc8SWarner Losh handle_activity(int bootnum, bool active) 3921285bcc8SWarner Losh { 3931285bcc8SWarner Losh uint32_t attrs, load_attrs; 3941285bcc8SWarner Losh uint8_t *data; 3951285bcc8SWarner Losh size_t size; 3961285bcc8SWarner Losh char *name; 3971285bcc8SWarner Losh 3981285bcc8SWarner Losh asprintf(&name, "%s%04X", "Boot", bootnum); 3991285bcc8SWarner Losh if (name == NULL) 4001285bcc8SWarner Losh err(1, "asprintf"); 4011285bcc8SWarner Losh if (efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, &attrs) < 0) 4021285bcc8SWarner Losh err(1, "No such bootvar %s\n", name); 4031285bcc8SWarner Losh 4041285bcc8SWarner Losh load_attrs = le32dec(data); 4051285bcc8SWarner Losh 4061285bcc8SWarner Losh if (active) 4071285bcc8SWarner Losh load_attrs |= LOAD_OPTION_ACTIVE; 4081285bcc8SWarner Losh else 4091285bcc8SWarner Losh load_attrs &= ~LOAD_OPTION_ACTIVE; 4101285bcc8SWarner Losh 4111285bcc8SWarner Losh le32enc(data, load_attrs); 4121285bcc8SWarner Losh 4131285bcc8SWarner Losh if (set_bootvar(name, data, size) < 0) 4141285bcc8SWarner Losh err(1, "handle activity efi_set_variable"); 4151285bcc8SWarner Losh } 4161285bcc8SWarner Losh 4171285bcc8SWarner Losh 4181285bcc8SWarner Losh /* 4191285bcc8SWarner Losh * add boot var to boot order. 4201285bcc8SWarner Losh * called by create boot var. There is no option 4211285bcc8SWarner Losh * to add one independent of create. 4221285bcc8SWarner Losh * 4231285bcc8SWarner Losh * Note: we currently don't support where it goes 4241285bcc8SWarner Losh * so it goes on the front, inactive. 4251285bcc8SWarner Losh * use -o 2,3,7 etc to affect order, -a to activate. 4261285bcc8SWarner Losh */ 4271285bcc8SWarner Losh static void 4281285bcc8SWarner Losh add_to_boot_order(char *bootvar) 4291285bcc8SWarner Losh { 4301285bcc8SWarner Losh size_t size; 4311285bcc8SWarner Losh uint32_t attrs; 4321285bcc8SWarner Losh uint16_t val; 4331285bcc8SWarner Losh uint8_t *data, *new; 4341285bcc8SWarner Losh 4351285bcc8SWarner Losh val = strtoul(&bootvar[4], NULL, 16); 4361285bcc8SWarner Losh 4371285bcc8SWarner Losh if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) < 0) { 4381285bcc8SWarner Losh if (errno == ENOENT) { /* create it and set this bootvar to active */ 4391285bcc8SWarner Losh size = 0; 4401285bcc8SWarner Losh data = NULL; 4411285bcc8SWarner Losh } else 4421285bcc8SWarner Losh err(1, "efi_get_variable BootOrder"); 4431285bcc8SWarner Losh } 4441285bcc8SWarner Losh 4451285bcc8SWarner Losh /* 4461285bcc8SWarner Losh * We have BootOrder with the current order 4471285bcc8SWarner Losh * so grow the array by one, add the value 4481285bcc8SWarner Losh * and write the new variable value. 4491285bcc8SWarner Losh */ 4501285bcc8SWarner Losh size += sizeof(uint16_t); 4511285bcc8SWarner Losh new = malloc(size); 4521285bcc8SWarner Losh if (!new) 4531285bcc8SWarner Losh err(1, "malloc"); 4541285bcc8SWarner Losh 4551285bcc8SWarner Losh le16enc(new, val); 4561285bcc8SWarner Losh if (size > sizeof(uint16_t)) 4571285bcc8SWarner Losh memcpy(new + sizeof(uint16_t), data, size - sizeof(uint16_t)); 4581285bcc8SWarner Losh 4591285bcc8SWarner Losh if (set_bootvar("BootOrder", new, size) < 0) 4601285bcc8SWarner Losh err(1, "set_bootvar"); 4611285bcc8SWarner Losh free(new); 4621285bcc8SWarner Losh } 4631285bcc8SWarner Losh 4641285bcc8SWarner Losh 4651285bcc8SWarner Losh static void 4661285bcc8SWarner Losh remove_from_order(uint16_t bootnum) 4671285bcc8SWarner Losh { 4681285bcc8SWarner Losh uint32_t attrs; 4691285bcc8SWarner Losh size_t size, i, j; 4701285bcc8SWarner Losh uint8_t *new, *data; 4711285bcc8SWarner Losh 4721285bcc8SWarner Losh if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) < 0) 4731285bcc8SWarner Losh return; 4741285bcc8SWarner Losh 4751285bcc8SWarner Losh new = malloc(size); 4761285bcc8SWarner Losh if (new == NULL) 4771285bcc8SWarner Losh err(1, "malloc"); 4781285bcc8SWarner Losh 4791285bcc8SWarner Losh for (j = i = 0; i < size; i += sizeof(uint16_t)) { 4801285bcc8SWarner Losh if (le16dec(data + i) == bootnum) 4811285bcc8SWarner Losh continue; 4821285bcc8SWarner Losh memcpy(new + j, data + i, sizeof(uint16_t)); 4831285bcc8SWarner Losh j += sizeof(uint16_t); 4841285bcc8SWarner Losh } 4851285bcc8SWarner Losh if (i == j) 4861285bcc8SWarner Losh warnx("Boot variable %04x not in BootOrder", bootnum); 4871285bcc8SWarner Losh else if (set_bootvar("BootOrder", new, j) < 0) 4881285bcc8SWarner Losh err(1, "Unable to update BootOrder with new value"); 4891285bcc8SWarner Losh free(new); 4901285bcc8SWarner Losh } 4911285bcc8SWarner Losh 4921285bcc8SWarner Losh 4931285bcc8SWarner Losh static void 4941285bcc8SWarner Losh delete_bootvar(int bootnum) 4951285bcc8SWarner Losh { 4961285bcc8SWarner Losh char *name; 4971285bcc8SWarner Losh int defer = 0; 4981285bcc8SWarner Losh 4991285bcc8SWarner Losh /* 5001285bcc8SWarner Losh * Try to delete the boot variable and remocve it 5011285bcc8SWarner Losh * from the boot order. We always do both actions 5021285bcc8SWarner Losh * to make it easy to clean up from oopses. 5031285bcc8SWarner Losh */ 5041285bcc8SWarner Losh if (bootnum < 0 || bootnum > 0xffff) 5051285bcc8SWarner Losh errx(1, "Bad boot variable %#x", bootnum); 5061285bcc8SWarner Losh asprintf(&name, "%s%04X", "Boot", bootnum); 5071285bcc8SWarner Losh if (name == NULL) 5081285bcc8SWarner Losh err(1, "asprintf"); 5091285bcc8SWarner Losh printf("Removing boot variable '%s'\n", name); 5101285bcc8SWarner Losh if (efi_del_variable(EFI_GLOBAL_GUID, name) < 0) { 5111285bcc8SWarner Losh defer = 1; 5121285bcc8SWarner Losh warn("cannot delete variable %s", name); 5131285bcc8SWarner Losh } 5141285bcc8SWarner Losh printf("Removing 0x%x from BootOrder\n", bootnum); 5151285bcc8SWarner Losh remove_from_order(bootnum); 5161285bcc8SWarner Losh free(name); 5171285bcc8SWarner Losh if (defer) 5181285bcc8SWarner Losh exit(defer); 5191285bcc8SWarner Losh } 5201285bcc8SWarner Losh 5211285bcc8SWarner Losh 5221285bcc8SWarner Losh static void 5231285bcc8SWarner Losh del_bootnext(void) 5241285bcc8SWarner Losh { 5251285bcc8SWarner Losh 5261285bcc8SWarner Losh if (efi_del_variable(EFI_GLOBAL_GUID, "BootNext") < 0) 5271285bcc8SWarner Losh err(1, "efi_del_variable"); 5281285bcc8SWarner Losh } 5291285bcc8SWarner Losh 5301285bcc8SWarner Losh static void 5311285bcc8SWarner Losh handle_bootnext(uint16_t bootnum) 5321285bcc8SWarner Losh { 5331285bcc8SWarner Losh uint16_t num; 5341285bcc8SWarner Losh 5351285bcc8SWarner Losh le16enc(&num, bootnum); 5361285bcc8SWarner Losh if (set_bootvar("BootNext", (uint8_t*)&bootnum, sizeof(uint16_t)) < 0) 5371285bcc8SWarner Losh err(1, "set_bootvar"); 5381285bcc8SWarner Losh } 5391285bcc8SWarner Losh 5401285bcc8SWarner Losh 5411285bcc8SWarner Losh static int 5421285bcc8SWarner Losh compare(const void *a, const void *b) 5431285bcc8SWarner Losh { 5441285bcc8SWarner Losh uint16_t c; 5451285bcc8SWarner Losh uint16_t d; 5461285bcc8SWarner Losh 5471285bcc8SWarner Losh memcpy(&c, a, sizeof(uint16_t)); 5481285bcc8SWarner Losh memcpy(&d, b, sizeof(uint16_t)); 5491285bcc8SWarner Losh 5501285bcc8SWarner Losh if (c < d) 5511285bcc8SWarner Losh return -1; 5521285bcc8SWarner Losh if (c == d) 5531285bcc8SWarner Losh return 0; 5541285bcc8SWarner Losh return 1; 5551285bcc8SWarner Losh } 5561285bcc8SWarner Losh 5571285bcc8SWarner Losh static char * 5586604afe9SWarner Losh make_next_boot_var_name(void) 5591285bcc8SWarner Losh { 5601285bcc8SWarner Losh struct entry *v; 5611285bcc8SWarner Losh uint16_t *vals, next_free = 0; 5621285bcc8SWarner Losh char *name; 5631285bcc8SWarner Losh int cnt = 0; 5641285bcc8SWarner Losh int i; 5651285bcc8SWarner Losh 5661285bcc8SWarner Losh LIST_FOREACH(v, &efivars, entries) { 5671285bcc8SWarner Losh cnt++; 5681285bcc8SWarner Losh } 5691285bcc8SWarner Losh 5701285bcc8SWarner Losh vals = malloc(sizeof(uint16_t) * cnt); 5711285bcc8SWarner Losh if (!vals) 5721285bcc8SWarner Losh return NULL; 5731285bcc8SWarner Losh 5741285bcc8SWarner Losh i = 0; 5751285bcc8SWarner Losh LIST_FOREACH(v, &efivars, entries) { 5761285bcc8SWarner Losh vals[i++] = v->idx; 5771285bcc8SWarner Losh } 5781285bcc8SWarner Losh qsort(vals, cnt, sizeof(uint16_t), compare); 5791285bcc8SWarner Losh /* if the hole is at the beginning, just return zero */ 5801285bcc8SWarner Losh if (vals[0] > 0) { 5811285bcc8SWarner Losh next_free = 0; 5821285bcc8SWarner Losh } else { 5831285bcc8SWarner Losh /* now just run the list looking for the first hole */ 5841285bcc8SWarner Losh for (i = 0; i < cnt - 1 && next_free == 0; i++) 585018caaaeSWarner Losh if (vals[i] + 1 != vals[i + 1]) 5861285bcc8SWarner Losh next_free = vals[i] + 1; 5871285bcc8SWarner Losh if (next_free == 0) 5881285bcc8SWarner Losh next_free = vals[cnt - 1] + 1; 5891285bcc8SWarner Losh /* In theory we could have used all 65k slots -- what to do? */ 5901285bcc8SWarner Losh } 5911285bcc8SWarner Losh free(vals); 5921285bcc8SWarner Losh 5931285bcc8SWarner Losh asprintf(&name, "%s%04X", "Boot", next_free); 5941285bcc8SWarner Losh if (name == NULL) 5951285bcc8SWarner Losh err(1, "asprintf"); 5961285bcc8SWarner Losh return name; 5971285bcc8SWarner Losh } 5981285bcc8SWarner Losh 599de26ba4dSWarner Losh static char * 600de26ba4dSWarner Losh make_boot_var_name(uint16_t bootnum) 601de26ba4dSWarner Losh { 602de26ba4dSWarner Losh struct entry *v; 603de26ba4dSWarner Losh char *name; 604de26ba4dSWarner Losh 605de26ba4dSWarner Losh LIST_FOREACH(v, &efivars, entries) { 606de26ba4dSWarner Losh if (v->idx == bootnum) 607de26ba4dSWarner Losh return NULL; 608de26ba4dSWarner Losh } 609de26ba4dSWarner Losh 610de26ba4dSWarner Losh asprintf(&name, "%s%04X", "Boot", bootnum); 611de26ba4dSWarner Losh if (name == NULL) 612de26ba4dSWarner Losh err(1, "asprintf"); 613de26ba4dSWarner Losh return name; 614de26ba4dSWarner Losh } 6151285bcc8SWarner Losh 6161285bcc8SWarner Losh static size_t 6171285bcc8SWarner Losh create_loadopt(uint8_t *buf, size_t bufmax, uint32_t attributes, efidp dp, size_t dp_size, 6181285bcc8SWarner Losh const char *description, const uint8_t *optional_data, size_t optional_data_size) 6191285bcc8SWarner Losh { 6201285bcc8SWarner Losh efi_char *bbuf = NULL; 6211285bcc8SWarner Losh uint8_t *pos = buf; 6221285bcc8SWarner Losh size_t desc_len = 0; 6231285bcc8SWarner Losh size_t len; 6241285bcc8SWarner Losh 6251285bcc8SWarner Losh if (optional_data == NULL && optional_data_size != 0) 6261285bcc8SWarner Losh return BAD_LENGTH; 6271285bcc8SWarner Losh if (dp == NULL && dp_size != 0) 6281285bcc8SWarner Losh return BAD_LENGTH; 6291285bcc8SWarner Losh 6301285bcc8SWarner Losh /* 6311285bcc8SWarner Losh * Compute the length to make sure the passed in buffer is long enough. 6321285bcc8SWarner Losh */ 6331285bcc8SWarner Losh utf8_to_ucs2(description, &bbuf, &desc_len); 6341285bcc8SWarner Losh len = sizeof(uint32_t) + sizeof(uint16_t) + desc_len + dp_size + optional_data_size; 6351285bcc8SWarner Losh if (len > bufmax) { 6361285bcc8SWarner Losh free(bbuf); 6371285bcc8SWarner Losh return BAD_LENGTH; 6381285bcc8SWarner Losh } 6391285bcc8SWarner Losh 6401285bcc8SWarner Losh le32enc(pos, attributes); 6411285bcc8SWarner Losh pos += sizeof (attributes); 6421285bcc8SWarner Losh 6431285bcc8SWarner Losh le16enc(pos, dp_size); 6441285bcc8SWarner Losh pos += sizeof (uint16_t); 6451285bcc8SWarner Losh 6461285bcc8SWarner Losh memcpy(pos, bbuf, desc_len); /* NB:desc_len includes strailing NUL */ 6471285bcc8SWarner Losh pos += desc_len; 6481285bcc8SWarner Losh free(bbuf); 6491285bcc8SWarner Losh 6501285bcc8SWarner Losh memcpy(pos, dp, dp_size); 6511285bcc8SWarner Losh pos += dp_size; 6521285bcc8SWarner Losh 6531285bcc8SWarner Losh if (optional_data && optional_data_size > 0) { 6541285bcc8SWarner Losh memcpy(pos, optional_data, optional_data_size); 6551285bcc8SWarner Losh pos += optional_data_size; 6561285bcc8SWarner Losh } 6571285bcc8SWarner Losh 6581285bcc8SWarner Losh return pos - buf; 6591285bcc8SWarner Losh } 6601285bcc8SWarner Losh 6611285bcc8SWarner Losh 6621285bcc8SWarner Losh static int 663de26ba4dSWarner Losh make_boot_var(const char *label, const char *loader, const char *kernel, const char *env, bool dry_run, 664de6b4aa8SWarner Losh int bootnum, bool activate) 6651285bcc8SWarner Losh { 6661285bcc8SWarner Losh struct entry *new_ent; 6671285bcc8SWarner Losh uint32_t load_attrs = 0; 6681285bcc8SWarner Losh uint8_t *load_opt_buf; 6691285bcc8SWarner Losh size_t lopt_size, llen, klen; 6701285bcc8SWarner Losh efidp dp, loaderdp, kerneldp; 6711285bcc8SWarner Losh char *bootvar = NULL; 6721285bcc8SWarner Losh int ret; 6731285bcc8SWarner Losh 674da15338dSWarner Losh assert(label != NULL); 675da15338dSWarner Losh 676de26ba4dSWarner Losh if (bootnum == -1) 6771285bcc8SWarner Losh bootvar = make_next_boot_var_name(); 678de26ba4dSWarner Losh else 679de26ba4dSWarner Losh bootvar = make_boot_var_name((uint16_t)bootnum); 6801285bcc8SWarner Losh if (bootvar == NULL) 6811285bcc8SWarner Losh err(1, "bootvar creation"); 6821285bcc8SWarner Losh if (loader == NULL) 6831285bcc8SWarner Losh errx(1, "Must specify boot loader"); 684b8adbe1cSRyan Moeller ret = efivar_unix_path_to_device_path(loader, &loaderdp); 685b8adbe1cSRyan Moeller if (ret != 0) 686b8adbe1cSRyan Moeller errc(1, ret, "Cannot translate unix loader path '%s' to UEFI", 687b8adbe1cSRyan Moeller loader); 6881285bcc8SWarner Losh if (kernel != NULL) { 689b8adbe1cSRyan Moeller ret = efivar_unix_path_to_device_path(kernel, &kerneldp); 690b8adbe1cSRyan Moeller if (ret != 0) 691b8adbe1cSRyan Moeller errc(1, ret, 692b8adbe1cSRyan Moeller "Cannot translate unix kernel path '%s' to UEFI", 693b8adbe1cSRyan Moeller kernel); 6941285bcc8SWarner Losh } else { 6951285bcc8SWarner Losh kerneldp = NULL; 6961285bcc8SWarner Losh } 6971285bcc8SWarner Losh llen = efidp_size(loaderdp); 698af934ea3SWarner Losh if (llen > MAX_DP_LEN) 699af934ea3SWarner Losh errx(1, "Loader path too long."); 7001285bcc8SWarner Losh klen = efidp_size(kerneldp); 701af934ea3SWarner Losh if (klen > MAX_DP_LEN) 702af934ea3SWarner Losh errx(1, "Kernel path too long."); 7031285bcc8SWarner Losh dp = malloc(llen + klen); 704af934ea3SWarner Losh if (dp == NULL) 705af934ea3SWarner Losh errx(1, "Can't allocate memory for new device paths"); 7061285bcc8SWarner Losh memcpy(dp, loaderdp, llen); 7071285bcc8SWarner Losh if (kerneldp != NULL) 7081285bcc8SWarner Losh memcpy((char *)dp + llen, kerneldp, klen); 7091285bcc8SWarner Losh 7101285bcc8SWarner Losh /* don't make the new bootvar active by default, use the -a option later */ 7111285bcc8SWarner Losh load_attrs = LOAD_OPTION_CATEGORY_BOOT; 712de6b4aa8SWarner Losh if (activate) 713de6b4aa8SWarner Losh load_attrs |= LOAD_OPTION_ACTIVE; 7141285bcc8SWarner Losh load_opt_buf = malloc(MAX_LOADOPT_LEN); 7151285bcc8SWarner Losh if (load_opt_buf == NULL) 7161285bcc8SWarner Losh err(1, "malloc"); 7171285bcc8SWarner Losh 7181285bcc8SWarner Losh lopt_size = create_loadopt(load_opt_buf, MAX_LOADOPT_LEN, load_attrs, 7191127aea3SWarner Losh dp, llen + klen, label, env, env ? strlen(env) + 1 : 0); 7201285bcc8SWarner Losh if (lopt_size == BAD_LENGTH) 72167a5e533SEmmanuel Vadot errx(1, "Can't create loadopt"); 7221285bcc8SWarner Losh 7231285bcc8SWarner Losh ret = 0; 7241285bcc8SWarner Losh if (!dry_run) { 7251285bcc8SWarner Losh ret = efi_set_variable(EFI_GLOBAL_GUID, bootvar, 7261285bcc8SWarner Losh (uint8_t*)load_opt_buf, lopt_size, COMMON_ATTRS); 7271285bcc8SWarner Losh } 7281285bcc8SWarner Losh 7291285bcc8SWarner Losh if (ret) 7301285bcc8SWarner Losh err(1, "efi_set_variable"); 7311285bcc8SWarner Losh 73267a5e533SEmmanuel Vadot if (!dry_run) 7331285bcc8SWarner Losh add_to_boot_order(bootvar); /* first, still not active */ 7341285bcc8SWarner Losh new_ent = malloc(sizeof(struct entry)); 7351285bcc8SWarner Losh if (new_ent == NULL) 7361285bcc8SWarner Losh err(1, "malloc"); 7371285bcc8SWarner Losh memset(new_ent, 0, sizeof(struct entry)); 7381285bcc8SWarner Losh new_ent->name = bootvar; 7391285bcc8SWarner Losh new_ent->guid = EFI_GLOBAL_GUID; 7401285bcc8SWarner Losh LIST_INSERT_HEAD(&efivars, new_ent, entries); 741a6b6f888SWarner Losh free(load_opt_buf); 7421285bcc8SWarner Losh free(dp); 743a6b6f888SWarner Losh 7441285bcc8SWarner Losh return 0; 7451285bcc8SWarner Losh } 7461285bcc8SWarner Losh 7471285bcc8SWarner Losh 7481285bcc8SWarner Losh static void 7491285bcc8SWarner Losh print_loadopt_str(uint8_t *data, size_t datalen) 7501285bcc8SWarner Losh { 7511285bcc8SWarner Losh char *dev, *relpath, *abspath; 7521285bcc8SWarner Losh uint32_t attr; 7531285bcc8SWarner Losh uint16_t fplen; 7541285bcc8SWarner Losh efi_char *descr; 7551285bcc8SWarner Losh uint8_t *ep = data + datalen; 7561285bcc8SWarner Losh uint8_t *walker = data; 7571285bcc8SWarner Losh efidp dp, edp; 7581285bcc8SWarner Losh char buf[1024]; 7591285bcc8SWarner Losh int len; 7601285bcc8SWarner Losh int rv; 761a2aa6671SWarner Losh int indent; 7621285bcc8SWarner Losh 7631285bcc8SWarner Losh if (datalen < sizeof(attr) + sizeof(fplen) + sizeof(efi_char)) 7641285bcc8SWarner Losh return; 7651285bcc8SWarner Losh // First 4 bytes are attribute flags 7661285bcc8SWarner Losh attr = le32dec(walker); 7671285bcc8SWarner Losh walker += sizeof(attr); 7681285bcc8SWarner Losh // Next two bytes are length of the file paths 7691285bcc8SWarner Losh fplen = le16dec(walker); 7701285bcc8SWarner Losh walker += sizeof(fplen); 7711285bcc8SWarner Losh // Next we have a 0 terminated UCS2 string that we know to be aligned 7721285bcc8SWarner Losh descr = (efi_char *)(intptr_t)(void *)walker; 7731285bcc8SWarner Losh len = ucs2len(descr); // XXX need to sanity check that len < (datalen - (ep - walker) / 2) 7741285bcc8SWarner Losh walker += (len + 1) * sizeof(efi_char); 7751285bcc8SWarner Losh if (walker > ep) 7761285bcc8SWarner Losh return; 7771285bcc8SWarner Losh // Now we have fplen bytes worth of file path stuff 7781285bcc8SWarner Losh dp = (efidp)walker; 7791285bcc8SWarner Losh walker += fplen; 7801285bcc8SWarner Losh if (walker > ep) 7811285bcc8SWarner Losh return; 7821285bcc8SWarner Losh edp = (efidp)walker; 78396289d09SAlexander Kabaev /* 78496289d09SAlexander Kabaev * Everything left is the binary option args 78596289d09SAlexander Kabaev * opt = walker; 78696289d09SAlexander Kabaev * optlen = ep - walker; 78796289d09SAlexander Kabaev */ 788a2aa6671SWarner Losh indent = 1; 7891285bcc8SWarner Losh while (dp < edp) { 7901285bcc8SWarner Losh efidp_format_device_path(buf, sizeof(buf), dp, 7911285bcc8SWarner Losh (intptr_t)(void *)edp - (intptr_t)(void *)dp); 792a2aa6671SWarner Losh printf("%*s%s\n", indent, "", buf); 793a2aa6671SWarner Losh indent = 10 + len + 1; 7941285bcc8SWarner Losh rv = efivar_device_path_to_unix_path(dp, &dev, &relpath, &abspath); 7951285bcc8SWarner Losh if (rv == 0) { 796a2aa6671SWarner Losh printf("%*s%s:%s %s\n", indent + 4, "", dev, relpath, abspath); 7971285bcc8SWarner Losh free(dev); 7981285bcc8SWarner Losh free(relpath); 7991285bcc8SWarner Losh free(abspath); 8001285bcc8SWarner Losh } 8011285bcc8SWarner Losh dp = (efidp)((char *)dp + efidp_size(dp)); 8021285bcc8SWarner Losh } 8031285bcc8SWarner Losh } 8041285bcc8SWarner Losh 8051285bcc8SWarner Losh static char * 8061285bcc8SWarner Losh get_descr(uint8_t *data) 8071285bcc8SWarner Losh { 8081285bcc8SWarner Losh uint8_t *pos = data; 8091285bcc8SWarner Losh efi_char *desc; 8101285bcc8SWarner Losh int len; 8111285bcc8SWarner Losh char *buf; 8121285bcc8SWarner Losh int i = 0; 8131285bcc8SWarner Losh 8141285bcc8SWarner Losh pos += sizeof(uint32_t) + sizeof(uint16_t); 8151285bcc8SWarner Losh desc = (efi_char*)(intptr_t)(void *)pos; 8161285bcc8SWarner Losh len = ucs2len(desc); 8171285bcc8SWarner Losh buf = malloc(len + 1); 8181285bcc8SWarner Losh memset(buf, 0, len + 1); 8191285bcc8SWarner Losh while (desc[i]) { 8201285bcc8SWarner Losh buf[i] = desc[i]; 8211285bcc8SWarner Losh i++; 8221285bcc8SWarner Losh } 8231285bcc8SWarner Losh return (char*)buf; 8241285bcc8SWarner Losh } 8251285bcc8SWarner Losh 8261285bcc8SWarner Losh 82751922c69SWarner Losh static bool 82851922c69SWarner Losh print_boot_var(const char *name, bool verbose, bool curboot) 82951922c69SWarner Losh { 83051922c69SWarner Losh size_t size; 83151922c69SWarner Losh uint32_t load_attrs; 83251922c69SWarner Losh uint8_t *data; 83351922c69SWarner Losh int ret; 83451922c69SWarner Losh char *d; 83551922c69SWarner Losh 83651922c69SWarner Losh ret = efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, NULL); 83751922c69SWarner Losh if (ret < 0) 83851922c69SWarner Losh return false; 83951922c69SWarner Losh load_attrs = le32dec(data); 84051922c69SWarner Losh d = get_descr(data); 84151922c69SWarner Losh printf("%c%s%c %s", curboot ? '+' : ' ', name, 84251922c69SWarner Losh ((load_attrs & LOAD_OPTION_ACTIVE) ? '*': ' '), d); 84351922c69SWarner Losh free(d); 84451922c69SWarner Losh if (verbose) 84551922c69SWarner Losh print_loadopt_str(data, size); 84651922c69SWarner Losh else 84751922c69SWarner Losh printf("\n"); 84851922c69SWarner Losh return true; 84951922c69SWarner Losh } 85051922c69SWarner Losh 85151922c69SWarner Losh 85283c42372SD Scott Phillips static bool 85383c42372SD Scott Phillips os_indication_supported(uint64_t indication) 85483c42372SD Scott Phillips { 85583c42372SD Scott Phillips uint8_t *data; 85683c42372SD Scott Phillips size_t size; 85783c42372SD Scott Phillips uint32_t attrs; 85883c42372SD Scott Phillips int ret; 85983c42372SD Scott Phillips 86083c42372SD Scott Phillips ret = efi_get_variable(EFI_GLOBAL_GUID, "OsIndicationsSupported", &data, 86183c42372SD Scott Phillips &size, &attrs); 86283c42372SD Scott Phillips if (ret < 0) 86383c42372SD Scott Phillips return false; 86483c42372SD Scott Phillips return (le64dec(data) & indication) == indication; 86583c42372SD Scott Phillips } 86683c42372SD Scott Phillips 86783c42372SD Scott Phillips static uint64_t 86883c42372SD Scott Phillips os_indications(void) 86983c42372SD Scott Phillips { 87083c42372SD Scott Phillips uint8_t *data; 87183c42372SD Scott Phillips size_t size; 87283c42372SD Scott Phillips uint32_t attrs; 87383c42372SD Scott Phillips int ret; 87483c42372SD Scott Phillips 87583c42372SD Scott Phillips ret = efi_get_variable(EFI_GLOBAL_GUID, "OsIndications", &data, &size, 87683c42372SD Scott Phillips &attrs); 87783c42372SD Scott Phillips if (ret < 0) 87883c42372SD Scott Phillips return 0; 87983c42372SD Scott Phillips return le64dec(data); 88083c42372SD Scott Phillips } 88183c42372SD Scott Phillips 88283c42372SD Scott Phillips static int 88383c42372SD Scott Phillips os_indications_set(uint64_t mask, uint64_t val) 88483c42372SD Scott Phillips { 88583c42372SD Scott Phillips uint8_t new[sizeof(uint64_t)]; 88683c42372SD Scott Phillips 88783c42372SD Scott Phillips le64enc(&new, (os_indications() & ~mask) | (val & mask)); 88883c42372SD Scott Phillips return set_bootvar("OsIndications", new, sizeof(new)); 88983c42372SD Scott Phillips } 89083c42372SD Scott Phillips 8911285bcc8SWarner Losh /* Cmd epilogue, or just the default with no args. 8921285bcc8SWarner Losh * The order is [bootnext] bootcurrent, timeout, order, and the bootvars [-v] 8931285bcc8SWarner Losh */ 8941285bcc8SWarner Losh static int 8951285bcc8SWarner Losh print_boot_vars(bool verbose) 8961285bcc8SWarner Losh { 8971285bcc8SWarner Losh /* 8981285bcc8SWarner Losh * just read and print the current values 8991285bcc8SWarner Losh * as a command epilogue 9001285bcc8SWarner Losh */ 9011285bcc8SWarner Losh struct entry *v; 9021285bcc8SWarner Losh uint8_t *data; 9031285bcc8SWarner Losh size_t size; 90451922c69SWarner Losh uint32_t attrs; 90551922c69SWarner Losh int ret, bolen; 90651922c69SWarner Losh uint16_t *boot_order = NULL, current; 90783c42372SD Scott Phillips bool boot_to_fw_ui; 90883c42372SD Scott Phillips 90983c42372SD Scott Phillips if (os_indication_supported(EFI_OS_INDICATIONS_BOOT_TO_FW_UI)) { 91083c42372SD Scott Phillips boot_to_fw_ui = 91183c42372SD Scott Phillips (os_indications() & EFI_OS_INDICATIONS_BOOT_TO_FW_UI) != 0; 91283c42372SD Scott Phillips printf("Boot to FW : %s\n", boot_to_fw_ui != 0 ? 91383c42372SD Scott Phillips "true" : "false"); 91483c42372SD Scott Phillips } 9151285bcc8SWarner Losh 9161285bcc8SWarner Losh ret = efi_get_variable(EFI_GLOBAL_GUID, "BootNext", &data, &size, &attrs); 9171285bcc8SWarner Losh if (ret > 0) { 9181285bcc8SWarner Losh printf("BootNext : %04x\n", le16dec(data)); 9191285bcc8SWarner Losh } 9201285bcc8SWarner Losh 9211285bcc8SWarner Losh ret = efi_get_variable(EFI_GLOBAL_GUID, "BootCurrent", &data, &size,&attrs); 92251922c69SWarner Losh current = le16dec(data); 92351922c69SWarner Losh printf("BootCurrent: %04x\n", current); 9241285bcc8SWarner Losh 9251285bcc8SWarner Losh ret = efi_get_variable(EFI_GLOBAL_GUID, "Timeout", &data, &size, &attrs); 9261285bcc8SWarner Losh if (ret > 0) { 9271285bcc8SWarner Losh printf("Timeout : %d seconds\n", le16dec(data)); 9281285bcc8SWarner Losh } 92912f409ffSWarner Losh 93012f409ffSWarner Losh if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) > 0) { 93112f409ffSWarner Losh if (size % 2 == 1) 93212f409ffSWarner Losh warn("Bad BootOrder variable: odd length %d", (int)size); 93351922c69SWarner Losh boot_order = malloc(size); 93451922c69SWarner Losh bolen = size / 2; 93512f409ffSWarner Losh printf("BootOrder : "); 93651922c69SWarner Losh for (size_t i = 0; i < size; i += 2) { 93751922c69SWarner Losh boot_order[i / 2] = le16dec(data + i); 93851922c69SWarner Losh printf("%04X%s", boot_order[i / 2], i == size - 2 ? "\n" : ", "); 93951922c69SWarner Losh } 94012f409ffSWarner Losh } 9411285bcc8SWarner Losh 94251922c69SWarner Losh if (boot_order == NULL) { 94351922c69SWarner Losh /* 94451922c69SWarner Losh * now we want to fetch 'em all fresh again 9451285bcc8SWarner Losh * which possibly includes a newly created bootvar 9461285bcc8SWarner Losh */ 9471285bcc8SWarner Losh LIST_FOREACH(v, &efivars, entries) { 94851922c69SWarner Losh print_boot_var(v->name, verbose, v->idx == current); 94951922c69SWarner Losh } 95051922c69SWarner Losh } else { 95151922c69SWarner Losh LIST_FOREACH(v, &efivars, entries) { 95251922c69SWarner Losh v->flags = 0; 95351922c69SWarner Losh } 95451922c69SWarner Losh for (int i = 0; i < bolen; i++) { 95551922c69SWarner Losh char buffer[10]; 95651922c69SWarner Losh 95751922c69SWarner Losh snprintf(buffer, sizeof(buffer), "Boot%04X", boot_order[i]); 95851922c69SWarner Losh if (!print_boot_var(buffer, verbose, boot_order[i] == current)) 95951922c69SWarner Losh printf("%s: MISSING!\n", buffer); 96051922c69SWarner Losh LIST_FOREACH(v, &efivars, entries) { 96151922c69SWarner Losh if (v->idx == boot_order[i]) { 96251922c69SWarner Losh v->flags |= SEEN; 96351922c69SWarner Losh break; 96451922c69SWarner Losh } 96551922c69SWarner Losh } 96651922c69SWarner Losh } 96751922c69SWarner Losh if (verbose) { 96851922c69SWarner Losh printf("\n\nUnreferenced Variables:\n"); 96951922c69SWarner Losh LIST_FOREACH(v, &efivars, entries) { 97051922c69SWarner Losh if (v->flags == 0) 97151922c69SWarner Losh print_boot_var(v->name, verbose, v->idx == current); 97251922c69SWarner Losh } 97351922c69SWarner Losh } 9741285bcc8SWarner Losh } 9751285bcc8SWarner Losh return 0; 9761285bcc8SWarner Losh } 9771285bcc8SWarner Losh 9781285bcc8SWarner Losh static void 9796604afe9SWarner Losh delete_timeout(void) 9801285bcc8SWarner Losh { 9811285bcc8SWarner Losh 9821285bcc8SWarner Losh efi_del_variable(EFI_GLOBAL_GUID,"Timeout"); 9831285bcc8SWarner Losh } 9841285bcc8SWarner Losh 9851285bcc8SWarner Losh static void 9861285bcc8SWarner Losh handle_timeout(int to) 9871285bcc8SWarner Losh { 9881285bcc8SWarner Losh uint16_t timeout; 9891285bcc8SWarner Losh 9901285bcc8SWarner Losh le16enc(&timeout, to); 991c66805a5SWarner Losh if (set_bootvar("Timeout", (uint8_t *)&timeout, sizeof(timeout)) < 0) 992c66805a5SWarner Losh errx(1, "Can't set Timeout for booting."); 9931285bcc8SWarner Losh } 9941285bcc8SWarner Losh 9951cdb8eb8SWarner Losh static void 9961cdb8eb8SWarner Losh report_esp_device(bool do_dp, bool do_unix) 9971cdb8eb8SWarner Losh { 9981cdb8eb8SWarner Losh uint8_t *data; 9991cdb8eb8SWarner Losh size_t size, len; 10001cdb8eb8SWarner Losh uint32_t attrs; 10011cdb8eb8SWarner Losh int ret; 10021cdb8eb8SWarner Losh uint16_t current, fplen; 10031cdb8eb8SWarner Losh char *name, *dev, *relpath, *abspath; 10041cdb8eb8SWarner Losh uint8_t *walker, *ep; 10051cdb8eb8SWarner Losh efi_char *descr; 10069a791529SWarner Losh efidp dp; 10071cdb8eb8SWarner Losh char buf[PATH_MAX]; 10081cdb8eb8SWarner Losh 10091cdb8eb8SWarner Losh if (do_dp && do_unix) 10101cdb8eb8SWarner Losh errx(1, "Can't report both UEFI device-path and Unix path together"); 10111cdb8eb8SWarner Losh 10121cdb8eb8SWarner Losh ret = efi_get_variable(EFI_GLOBAL_GUID, "BootCurrent", &data, &size,&attrs); 10131cdb8eb8SWarner Losh if (ret < 0) 10141cdb8eb8SWarner Losh err(1, "Can't get BootCurrent"); 10151cdb8eb8SWarner Losh current = le16dec(data); 10161cdb8eb8SWarner Losh if (asprintf(&name, "Boot%04X", current) < 0) 10171cdb8eb8SWarner Losh err(1, "Can't format boot var\n"); 10181cdb8eb8SWarner Losh if (efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, NULL) < 0) 10191cdb8eb8SWarner Losh err(1, "Can't retrieve EFI var %s", name); 10201cdb8eb8SWarner Losh // First 4 bytes are attribute flags 10211cdb8eb8SWarner Losh walker = data; 10221cdb8eb8SWarner Losh ep = walker + size; 10231cdb8eb8SWarner Losh walker += sizeof(uint32_t); 10241cdb8eb8SWarner Losh // Next two bytes are length of the file paths 10251cdb8eb8SWarner Losh fplen = le16dec(walker); 10261cdb8eb8SWarner Losh walker += sizeof(fplen); 10271cdb8eb8SWarner Losh // Next we have a 0 terminated UCS2 string that we know to be aligned 10281cdb8eb8SWarner Losh descr = (efi_char *)(intptr_t)(void *)walker; 10291cdb8eb8SWarner Losh len = ucs2len(descr); // XXX need to sanity check that len < (datalen - (ep - walker) / 2) 10301cdb8eb8SWarner Losh walker += (len + 1) * sizeof(efi_char); 10311cdb8eb8SWarner Losh if (walker > ep) 10321cdb8eb8SWarner Losh errx(1, "malformed boot variable %s", name); 10331cdb8eb8SWarner Losh // Now we have fplen bytes worth of file path stuff 10341cdb8eb8SWarner Losh dp = (efidp)walker; 10351cdb8eb8SWarner Losh walker += fplen; 10361cdb8eb8SWarner Losh if (walker > ep) 10371cdb8eb8SWarner Losh errx(1, "malformed boot variable %s", name); 10381cdb8eb8SWarner Losh if (do_dp) { 10391cdb8eb8SWarner Losh efidp_format_device_path_node(buf, sizeof(buf), dp); 10401cdb8eb8SWarner Losh printf("%s\n", buf); 10411cdb8eb8SWarner Losh exit(0); 10421cdb8eb8SWarner Losh } 10432e085395SMitchell Horne if (efivar_device_path_to_unix_path(dp, &dev, &relpath, &abspath) != 0) 10441cdb8eb8SWarner Losh errx(1, "Can't convert to unix path"); 10451cdb8eb8SWarner Losh if (do_unix) { 10461cdb8eb8SWarner Losh if (abspath == NULL) 10471cdb8eb8SWarner Losh errx(1, "Can't find where %s:%s is mounted", 10481cdb8eb8SWarner Losh dev, relpath); 10491cdb8eb8SWarner Losh abspath[strlen(abspath) - strlen(relpath) - 1] = '\0'; 10501cdb8eb8SWarner Losh printf("%s\n", abspath); 10511cdb8eb8SWarner Losh } else { 10521cdb8eb8SWarner Losh printf("%s\n", dev); 10531cdb8eb8SWarner Losh } 10541cdb8eb8SWarner Losh free(dev); 10551cdb8eb8SWarner Losh free(relpath); 10561cdb8eb8SWarner Losh free(abspath); 10571cdb8eb8SWarner Losh exit(0); 10581cdb8eb8SWarner Losh } 10591cdb8eb8SWarner Losh 106083c42372SD Scott Phillips static void 106183c42372SD Scott Phillips set_boot_to_fw_ui(bool to_fw) 106283c42372SD Scott Phillips { 106383c42372SD Scott Phillips int ret; 106483c42372SD Scott Phillips 106583c42372SD Scott Phillips if (!os_indication_supported(EFI_OS_INDICATIONS_BOOT_TO_FW_UI)) { 106683c42372SD Scott Phillips if (to_fw) 106783c42372SD Scott Phillips errx(1, "boot to fw ui not supported"); 106883c42372SD Scott Phillips else 106983c42372SD Scott Phillips return; 107083c42372SD Scott Phillips } 107183c42372SD Scott Phillips ret = os_indications_set(EFI_OS_INDICATIONS_BOOT_TO_FW_UI, 107283c42372SD Scott Phillips to_fw ? ~0 : 0); 107383c42372SD Scott Phillips if (ret < 0) 107483c42372SD Scott Phillips errx(1, "failed to set boot to fw ui"); 107583c42372SD Scott Phillips } 107683c42372SD Scott Phillips 10779a791529SWarner Losh static void 10789a791529SWarner Losh find_efi_device(const char *path) 10799a791529SWarner Losh { 10809a791529SWarner Losh efidp dp = NULL; 10819a791529SWarner Losh size_t len; 10829a791529SWarner Losh int ret; 10839a791529SWarner Losh char buf[1024]; 10849a791529SWarner Losh 10859a791529SWarner Losh ret = efivar_unix_path_to_device_path(path, &dp); 10869a791529SWarner Losh if (ret != 0) 10879a791529SWarner Losh errc(1, ret, 10889a791529SWarner Losh "Cannot translate path '%s' to UEFI", path); 10899a791529SWarner Losh len = efidp_size(dp); 10909a791529SWarner Losh if (len > MAX_DP_LEN) 10919a791529SWarner Losh errx(1, "Resulting device path too long."); 10929a791529SWarner Losh efidp_format_device_path(buf, sizeof(buf), dp, len); 10939a791529SWarner Losh printf("%s -> %s\n", path, buf); 10949a791529SWarner Losh exit (0); 10959a791529SWarner Losh } 10969a791529SWarner Losh 10971285bcc8SWarner Losh int 10981285bcc8SWarner Losh main(int argc, char *argv[]) 10991285bcc8SWarner Losh { 11001285bcc8SWarner Losh 11017fe2f504SWarner Losh memset(&opts, 0, sizeof (bmgr_opts_t)); 11027fe2f504SWarner Losh parse_args(argc, argv); 11037fe2f504SWarner Losh 11049a791529SWarner Losh /* 11059a791529SWarner Losh * find_dev can operate without any efi variables 11069a791529SWarner Losh */ 11079a791529SWarner Losh if (!efi_variables_supported() && !opts.find_dev) 11081285bcc8SWarner Losh errx(1, "efi variables not supported on this system. root? kldload efirt?"); 11091285bcc8SWarner Losh 11101285bcc8SWarner Losh read_vars(); 11111285bcc8SWarner Losh 11121285bcc8SWarner Losh if (opts.create) 11131285bcc8SWarner Losh /* 11141285bcc8SWarner Losh * side effect, adds to boot order, but not yet active. 11151285bcc8SWarner Losh */ 1116da15338dSWarner Losh make_boot_var(opts.label ? opts.label : "", 1117de26ba4dSWarner Losh opts.loader, opts.kernel, opts.env, opts.dry_run, 1118de6b4aa8SWarner Losh opts.has_bootnum ? opts.bootnum : -1, opts.set_active); 11191285bcc8SWarner Losh else if (opts.set_active || opts.set_inactive ) 11201285bcc8SWarner Losh handle_activity(opts.bootnum, opts.set_active); 11211285bcc8SWarner Losh else if (opts.order != NULL) 11221285bcc8SWarner Losh set_boot_order(opts.order); /* create a new bootorder with opts.order */ 11231285bcc8SWarner Losh else if (opts.set_bootnext) 11241285bcc8SWarner Losh handle_bootnext(opts.bootnum); 11251285bcc8SWarner Losh else if (opts.delete_bootnext) 11261285bcc8SWarner Losh del_bootnext(); 11271285bcc8SWarner Losh else if (opts.delete) 11281285bcc8SWarner Losh delete_bootvar(opts.bootnum); 11291285bcc8SWarner Losh else if (opts.del_timeout) 11301285bcc8SWarner Losh delete_timeout(); 11311285bcc8SWarner Losh else if (opts.set_timeout) 11321285bcc8SWarner Losh handle_timeout(opts.timeout); 11331cdb8eb8SWarner Losh else if (opts.esp_device) 11341cdb8eb8SWarner Losh report_esp_device(opts.device_path, opts.unix_path); 113583c42372SD Scott Phillips else if (opts.fw_ui) 113683c42372SD Scott Phillips set_boot_to_fw_ui(true); 113783c42372SD Scott Phillips else if (opts.no_fw_ui) 113883c42372SD Scott Phillips set_boot_to_fw_ui(false); 11399a791529SWarner Losh else if (opts.find_dev) 11409a791529SWarner Losh find_efi_device(opts.dev); 11411285bcc8SWarner Losh 11421285bcc8SWarner Losh print_boot_vars(opts.verbose); 11431285bcc8SWarner Losh } 1144