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 { 73*9a791529SWarner 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; 89*9a791529SWarner 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'}, 119*9a791529SWarner 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; 290*9a791529SWarner Losh case 'u': 291*9a791529SWarner Losh opts.find_dev = true; 292*9a791529SWarner Losh opts.dev = strdup(optarg); 2931285bcc8SWarner Losh case 'v': 2941285bcc8SWarner Losh opts.verbose = true; 2951285bcc8SWarner Losh break; 2961285bcc8SWarner Losh } 2971285bcc8SWarner Losh } 2981285bcc8SWarner Losh if (opts.create) { 2991127aea3SWarner Losh if (!opts.loader) 3001285bcc8SWarner Losh errx(1, "%s",CREATE_USAGE); 3011285bcc8SWarner Losh return; 3021285bcc8SWarner Losh } 3031285bcc8SWarner Losh 304b9580775SToomas Soome if (opts.order != NULL && *opts.order == '\0') 3051285bcc8SWarner Losh errx(1, "%s", ORDER_USAGE); 3069ed08029SRebecca Cran 3079ed08029SRebecca Cran if ((opts.set_inactive || opts.set_active) && !opts.has_bootnum) 3089ed08029SRebecca Cran errx(1, "%s", ACTIVE_USAGE); 3099ed08029SRebecca Cran 3109ed08029SRebecca Cran if (opts.delete && !opts.has_bootnum) 3119ed08029SRebecca Cran errx(1, "%s", DELETE_USAGE); 3129ed08029SRebecca Cran 3139ed08029SRebecca Cran if (opts.set_bootnext && !opts.has_bootnum) 3149ed08029SRebecca Cran errx(1, "%s", BOOTNEXT_USAGE); 3151285bcc8SWarner Losh } 3161285bcc8SWarner Losh 3171285bcc8SWarner Losh 3181285bcc8SWarner Losh static void 3191285bcc8SWarner Losh read_vars(void) 3201285bcc8SWarner Losh { 3211285bcc8SWarner Losh 3221285bcc8SWarner Losh efi_guid_t *guid; 3231285bcc8SWarner Losh char *next_name = NULL; 3241285bcc8SWarner Losh int ret = 0; 3251285bcc8SWarner Losh 3261285bcc8SWarner Losh struct entry *nent; 3271285bcc8SWarner Losh 3281285bcc8SWarner Losh LIST_INIT(&efivars); 3291285bcc8SWarner Losh while ((ret = efi_get_next_variable_name(&guid, &next_name)) > 0) { 3301285bcc8SWarner Losh /* 331a3e6c4a6SWarner Losh * Only pay attention to EFI:BootXXXX variables to get the list. 3321285bcc8SWarner Losh */ 3331285bcc8SWarner Losh if (efi_guid_cmp(guid, &EFI_GLOBAL_GUID) != 0 || 3341285bcc8SWarner Losh strlen(next_name) != 8 || 3351285bcc8SWarner Losh strncmp(next_name, "Boot", 4) != 0 || 3361285bcc8SWarner Losh !isxdigit(next_name[4]) || 3371285bcc8SWarner Losh !isxdigit(next_name[5]) || 3381285bcc8SWarner Losh !isxdigit(next_name[6]) || 3391285bcc8SWarner Losh !isxdigit(next_name[7])) 3401285bcc8SWarner Losh continue; 3411285bcc8SWarner Losh nent = malloc(sizeof(struct entry)); 3421285bcc8SWarner Losh nent->name = strdup(next_name); 3431285bcc8SWarner Losh 3441285bcc8SWarner Losh ret = efi_get_variable(*guid, next_name, &nent->data, 3451285bcc8SWarner Losh &nent->size, &nent->attrs); 3461285bcc8SWarner Losh if (ret < 0) 3471285bcc8SWarner Losh err(1, "efi_get_variable"); 3481285bcc8SWarner Losh nent->guid = *guid; 3491285bcc8SWarner Losh nent->idx = strtoul(&next_name[4], NULL, 16); 3501285bcc8SWarner Losh LIST_INSERT_HEAD(&efivars, nent, entries); 3511285bcc8SWarner Losh } 3521285bcc8SWarner Losh } 3531285bcc8SWarner Losh 3541285bcc8SWarner Losh 3551285bcc8SWarner Losh static void 3561285bcc8SWarner Losh set_boot_order(char *order) 3571285bcc8SWarner Losh { 3581285bcc8SWarner Losh uint16_t *new_data; 3591285bcc8SWarner Losh size_t size; 3601285bcc8SWarner Losh char *next, *cp; 3611285bcc8SWarner Losh int cnt; 3621285bcc8SWarner Losh int i; 3631285bcc8SWarner Losh 3641285bcc8SWarner Losh cp = order; 3651285bcc8SWarner Losh cnt = 1; 3661285bcc8SWarner Losh while (*cp) { 3671285bcc8SWarner Losh if (*cp++ == ',') 3681285bcc8SWarner Losh cnt++; 3691285bcc8SWarner Losh } 3701285bcc8SWarner Losh size = sizeof(uint16_t) * cnt; 3711285bcc8SWarner Losh new_data = malloc(size); 3721285bcc8SWarner Losh 3731285bcc8SWarner Losh i = 0; 3741285bcc8SWarner Losh cp = strdup(order); 3751285bcc8SWarner Losh while ((next = strsep(&cp, ",")) != NULL) { 3761285bcc8SWarner Losh new_data[i] = strtoul(next, NULL, 16); 3771285bcc8SWarner Losh if (new_data[i] == 0 && errno == EINVAL) { 3781285bcc8SWarner Losh warnx("can't parse %s as a numb", next); 3791285bcc8SWarner Losh errx(1, "%s", ORDER_USAGE); 3801285bcc8SWarner Losh } 3811285bcc8SWarner Losh i++; 3821285bcc8SWarner Losh } 3831285bcc8SWarner Losh free(cp); 3841285bcc8SWarner Losh if (set_bootvar("BootOrder", (uint8_t*)new_data, size) < 0) 3851285bcc8SWarner Losh err(1, "Unabke to set BootOrder to %s", order); 3863af42ca7SWarner Losh free(new_data); 3871285bcc8SWarner Losh } 3881285bcc8SWarner Losh 3891285bcc8SWarner Losh static void 3901285bcc8SWarner Losh handle_activity(int bootnum, bool active) 3911285bcc8SWarner Losh { 3921285bcc8SWarner Losh uint32_t attrs, load_attrs; 3931285bcc8SWarner Losh uint8_t *data; 3941285bcc8SWarner Losh size_t size; 3951285bcc8SWarner Losh char *name; 3961285bcc8SWarner Losh 3971285bcc8SWarner Losh asprintf(&name, "%s%04X", "Boot", bootnum); 3981285bcc8SWarner Losh if (name == NULL) 3991285bcc8SWarner Losh err(1, "asprintf"); 4001285bcc8SWarner Losh if (efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, &attrs) < 0) 4011285bcc8SWarner Losh err(1, "No such bootvar %s\n", name); 4021285bcc8SWarner Losh 4031285bcc8SWarner Losh load_attrs = le32dec(data); 4041285bcc8SWarner Losh 4051285bcc8SWarner Losh if (active) 4061285bcc8SWarner Losh load_attrs |= LOAD_OPTION_ACTIVE; 4071285bcc8SWarner Losh else 4081285bcc8SWarner Losh load_attrs &= ~LOAD_OPTION_ACTIVE; 4091285bcc8SWarner Losh 4101285bcc8SWarner Losh le32enc(data, load_attrs); 4111285bcc8SWarner Losh 4121285bcc8SWarner Losh if (set_bootvar(name, data, size) < 0) 4131285bcc8SWarner Losh err(1, "handle activity efi_set_variable"); 4141285bcc8SWarner Losh } 4151285bcc8SWarner Losh 4161285bcc8SWarner Losh 4171285bcc8SWarner Losh /* 4181285bcc8SWarner Losh * add boot var to boot order. 4191285bcc8SWarner Losh * called by create boot var. There is no option 4201285bcc8SWarner Losh * to add one independent of create. 4211285bcc8SWarner Losh * 4221285bcc8SWarner Losh * Note: we currently don't support where it goes 4231285bcc8SWarner Losh * so it goes on the front, inactive. 4241285bcc8SWarner Losh * use -o 2,3,7 etc to affect order, -a to activate. 4251285bcc8SWarner Losh */ 4261285bcc8SWarner Losh static void 4271285bcc8SWarner Losh add_to_boot_order(char *bootvar) 4281285bcc8SWarner Losh { 4291285bcc8SWarner Losh size_t size; 4301285bcc8SWarner Losh uint32_t attrs; 4311285bcc8SWarner Losh uint16_t val; 4321285bcc8SWarner Losh uint8_t *data, *new; 4331285bcc8SWarner Losh 4341285bcc8SWarner Losh val = strtoul(&bootvar[4], NULL, 16); 4351285bcc8SWarner Losh 4361285bcc8SWarner Losh if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) < 0) { 4371285bcc8SWarner Losh if (errno == ENOENT) { /* create it and set this bootvar to active */ 4381285bcc8SWarner Losh size = 0; 4391285bcc8SWarner Losh data = NULL; 4401285bcc8SWarner Losh } else 4411285bcc8SWarner Losh err(1, "efi_get_variable BootOrder"); 4421285bcc8SWarner Losh } 4431285bcc8SWarner Losh 4441285bcc8SWarner Losh /* 4451285bcc8SWarner Losh * We have BootOrder with the current order 4461285bcc8SWarner Losh * so grow the array by one, add the value 4471285bcc8SWarner Losh * and write the new variable value. 4481285bcc8SWarner Losh */ 4491285bcc8SWarner Losh size += sizeof(uint16_t); 4501285bcc8SWarner Losh new = malloc(size); 4511285bcc8SWarner Losh if (!new) 4521285bcc8SWarner Losh err(1, "malloc"); 4531285bcc8SWarner Losh 4541285bcc8SWarner Losh le16enc(new, val); 4551285bcc8SWarner Losh if (size > sizeof(uint16_t)) 4561285bcc8SWarner Losh memcpy(new + sizeof(uint16_t), data, size - sizeof(uint16_t)); 4571285bcc8SWarner Losh 4581285bcc8SWarner Losh if (set_bootvar("BootOrder", new, size) < 0) 4591285bcc8SWarner Losh err(1, "set_bootvar"); 4601285bcc8SWarner Losh free(new); 4611285bcc8SWarner Losh } 4621285bcc8SWarner Losh 4631285bcc8SWarner Losh 4641285bcc8SWarner Losh static void 4651285bcc8SWarner Losh remove_from_order(uint16_t bootnum) 4661285bcc8SWarner Losh { 4671285bcc8SWarner Losh uint32_t attrs; 4681285bcc8SWarner Losh size_t size, i, j; 4691285bcc8SWarner Losh uint8_t *new, *data; 4701285bcc8SWarner Losh 4711285bcc8SWarner Losh if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) < 0) 4721285bcc8SWarner Losh return; 4731285bcc8SWarner Losh 4741285bcc8SWarner Losh new = malloc(size); 4751285bcc8SWarner Losh if (new == NULL) 4761285bcc8SWarner Losh err(1, "malloc"); 4771285bcc8SWarner Losh 4781285bcc8SWarner Losh for (j = i = 0; i < size; i += sizeof(uint16_t)) { 4791285bcc8SWarner Losh if (le16dec(data + i) == bootnum) 4801285bcc8SWarner Losh continue; 4811285bcc8SWarner Losh memcpy(new + j, data + i, sizeof(uint16_t)); 4821285bcc8SWarner Losh j += sizeof(uint16_t); 4831285bcc8SWarner Losh } 4841285bcc8SWarner Losh if (i == j) 4851285bcc8SWarner Losh warnx("Boot variable %04x not in BootOrder", bootnum); 4861285bcc8SWarner Losh else if (set_bootvar("BootOrder", new, j) < 0) 4871285bcc8SWarner Losh err(1, "Unable to update BootOrder with new value"); 4881285bcc8SWarner Losh free(new); 4891285bcc8SWarner Losh } 4901285bcc8SWarner Losh 4911285bcc8SWarner Losh 4921285bcc8SWarner Losh static void 4931285bcc8SWarner Losh delete_bootvar(int bootnum) 4941285bcc8SWarner Losh { 4951285bcc8SWarner Losh char *name; 4961285bcc8SWarner Losh int defer = 0; 4971285bcc8SWarner Losh 4981285bcc8SWarner Losh /* 4991285bcc8SWarner Losh * Try to delete the boot variable and remocve it 5001285bcc8SWarner Losh * from the boot order. We always do both actions 5011285bcc8SWarner Losh * to make it easy to clean up from oopses. 5021285bcc8SWarner Losh */ 5031285bcc8SWarner Losh if (bootnum < 0 || bootnum > 0xffff) 5041285bcc8SWarner Losh errx(1, "Bad boot variable %#x", bootnum); 5051285bcc8SWarner Losh asprintf(&name, "%s%04X", "Boot", bootnum); 5061285bcc8SWarner Losh if (name == NULL) 5071285bcc8SWarner Losh err(1, "asprintf"); 5081285bcc8SWarner Losh printf("Removing boot variable '%s'\n", name); 5091285bcc8SWarner Losh if (efi_del_variable(EFI_GLOBAL_GUID, name) < 0) { 5101285bcc8SWarner Losh defer = 1; 5111285bcc8SWarner Losh warn("cannot delete variable %s", name); 5121285bcc8SWarner Losh } 5131285bcc8SWarner Losh printf("Removing 0x%x from BootOrder\n", bootnum); 5141285bcc8SWarner Losh remove_from_order(bootnum); 5151285bcc8SWarner Losh free(name); 5161285bcc8SWarner Losh if (defer) 5171285bcc8SWarner Losh exit(defer); 5181285bcc8SWarner Losh } 5191285bcc8SWarner Losh 5201285bcc8SWarner Losh 5211285bcc8SWarner Losh static void 5221285bcc8SWarner Losh del_bootnext(void) 5231285bcc8SWarner Losh { 5241285bcc8SWarner Losh 5251285bcc8SWarner Losh if (efi_del_variable(EFI_GLOBAL_GUID, "BootNext") < 0) 5261285bcc8SWarner Losh err(1, "efi_del_variable"); 5271285bcc8SWarner Losh } 5281285bcc8SWarner Losh 5291285bcc8SWarner Losh static void 5301285bcc8SWarner Losh handle_bootnext(uint16_t bootnum) 5311285bcc8SWarner Losh { 5321285bcc8SWarner Losh uint16_t num; 5331285bcc8SWarner Losh 5341285bcc8SWarner Losh le16enc(&num, bootnum); 5351285bcc8SWarner Losh if (set_bootvar("BootNext", (uint8_t*)&bootnum, sizeof(uint16_t)) < 0) 5361285bcc8SWarner Losh err(1, "set_bootvar"); 5371285bcc8SWarner Losh } 5381285bcc8SWarner Losh 5391285bcc8SWarner Losh 5401285bcc8SWarner Losh static int 5411285bcc8SWarner Losh compare(const void *a, const void *b) 5421285bcc8SWarner Losh { 5431285bcc8SWarner Losh uint16_t c; 5441285bcc8SWarner Losh uint16_t d; 5451285bcc8SWarner Losh 5461285bcc8SWarner Losh memcpy(&c, a, sizeof(uint16_t)); 5471285bcc8SWarner Losh memcpy(&d, b, sizeof(uint16_t)); 5481285bcc8SWarner Losh 5491285bcc8SWarner Losh if (c < d) 5501285bcc8SWarner Losh return -1; 5511285bcc8SWarner Losh if (c == d) 5521285bcc8SWarner Losh return 0; 5531285bcc8SWarner Losh return 1; 5541285bcc8SWarner Losh } 5551285bcc8SWarner Losh 5561285bcc8SWarner Losh static char * 5576604afe9SWarner Losh make_next_boot_var_name(void) 5581285bcc8SWarner Losh { 5591285bcc8SWarner Losh struct entry *v; 5601285bcc8SWarner Losh uint16_t *vals, next_free = 0; 5611285bcc8SWarner Losh char *name; 5621285bcc8SWarner Losh int cnt = 0; 5631285bcc8SWarner Losh int i; 5641285bcc8SWarner Losh 5651285bcc8SWarner Losh LIST_FOREACH(v, &efivars, entries) { 5661285bcc8SWarner Losh cnt++; 5671285bcc8SWarner Losh } 5681285bcc8SWarner Losh 5691285bcc8SWarner Losh vals = malloc(sizeof(uint16_t) * cnt); 5701285bcc8SWarner Losh if (!vals) 5711285bcc8SWarner Losh return NULL; 5721285bcc8SWarner Losh 5731285bcc8SWarner Losh i = 0; 5741285bcc8SWarner Losh LIST_FOREACH(v, &efivars, entries) { 5751285bcc8SWarner Losh vals[i++] = v->idx; 5761285bcc8SWarner Losh } 5771285bcc8SWarner Losh qsort(vals, cnt, sizeof(uint16_t), compare); 5781285bcc8SWarner Losh /* if the hole is at the beginning, just return zero */ 5791285bcc8SWarner Losh if (vals[0] > 0) { 5801285bcc8SWarner Losh next_free = 0; 5811285bcc8SWarner Losh } else { 5821285bcc8SWarner Losh /* now just run the list looking for the first hole */ 5831285bcc8SWarner Losh for (i = 0; i < cnt - 1 && next_free == 0; i++) 584018caaaeSWarner Losh if (vals[i] + 1 != vals[i + 1]) 5851285bcc8SWarner Losh next_free = vals[i] + 1; 5861285bcc8SWarner Losh if (next_free == 0) 5871285bcc8SWarner Losh next_free = vals[cnt - 1] + 1; 5881285bcc8SWarner Losh /* In theory we could have used all 65k slots -- what to do? */ 5891285bcc8SWarner Losh } 5901285bcc8SWarner Losh free(vals); 5911285bcc8SWarner Losh 5921285bcc8SWarner Losh asprintf(&name, "%s%04X", "Boot", next_free); 5931285bcc8SWarner Losh if (name == NULL) 5941285bcc8SWarner Losh err(1, "asprintf"); 5951285bcc8SWarner Losh return name; 5961285bcc8SWarner Losh } 5971285bcc8SWarner Losh 598de26ba4dSWarner Losh static char * 599de26ba4dSWarner Losh make_boot_var_name(uint16_t bootnum) 600de26ba4dSWarner Losh { 601de26ba4dSWarner Losh struct entry *v; 602de26ba4dSWarner Losh char *name; 603de26ba4dSWarner Losh 604de26ba4dSWarner Losh LIST_FOREACH(v, &efivars, entries) { 605de26ba4dSWarner Losh if (v->idx == bootnum) 606de26ba4dSWarner Losh return NULL; 607de26ba4dSWarner Losh } 608de26ba4dSWarner Losh 609de26ba4dSWarner Losh asprintf(&name, "%s%04X", "Boot", bootnum); 610de26ba4dSWarner Losh if (name == NULL) 611de26ba4dSWarner Losh err(1, "asprintf"); 612de26ba4dSWarner Losh return name; 613de26ba4dSWarner Losh } 6141285bcc8SWarner Losh 6151285bcc8SWarner Losh static size_t 6161285bcc8SWarner Losh create_loadopt(uint8_t *buf, size_t bufmax, uint32_t attributes, efidp dp, size_t dp_size, 6171285bcc8SWarner Losh const char *description, const uint8_t *optional_data, size_t optional_data_size) 6181285bcc8SWarner Losh { 6191285bcc8SWarner Losh efi_char *bbuf = NULL; 6201285bcc8SWarner Losh uint8_t *pos = buf; 6211285bcc8SWarner Losh size_t desc_len = 0; 6221285bcc8SWarner Losh size_t len; 6231285bcc8SWarner Losh 6241285bcc8SWarner Losh if (optional_data == NULL && optional_data_size != 0) 6251285bcc8SWarner Losh return BAD_LENGTH; 6261285bcc8SWarner Losh if (dp == NULL && dp_size != 0) 6271285bcc8SWarner Losh return BAD_LENGTH; 6281285bcc8SWarner Losh 6291285bcc8SWarner Losh /* 6301285bcc8SWarner Losh * Compute the length to make sure the passed in buffer is long enough. 6311285bcc8SWarner Losh */ 6321285bcc8SWarner Losh utf8_to_ucs2(description, &bbuf, &desc_len); 6331285bcc8SWarner Losh len = sizeof(uint32_t) + sizeof(uint16_t) + desc_len + dp_size + optional_data_size; 6341285bcc8SWarner Losh if (len > bufmax) { 6351285bcc8SWarner Losh free(bbuf); 6361285bcc8SWarner Losh return BAD_LENGTH; 6371285bcc8SWarner Losh } 6381285bcc8SWarner Losh 6391285bcc8SWarner Losh le32enc(pos, attributes); 6401285bcc8SWarner Losh pos += sizeof (attributes); 6411285bcc8SWarner Losh 6421285bcc8SWarner Losh le16enc(pos, dp_size); 6431285bcc8SWarner Losh pos += sizeof (uint16_t); 6441285bcc8SWarner Losh 6451285bcc8SWarner Losh memcpy(pos, bbuf, desc_len); /* NB:desc_len includes strailing NUL */ 6461285bcc8SWarner Losh pos += desc_len; 6471285bcc8SWarner Losh free(bbuf); 6481285bcc8SWarner Losh 6491285bcc8SWarner Losh memcpy(pos, dp, dp_size); 6501285bcc8SWarner Losh pos += dp_size; 6511285bcc8SWarner Losh 6521285bcc8SWarner Losh if (optional_data && optional_data_size > 0) { 6531285bcc8SWarner Losh memcpy(pos, optional_data, optional_data_size); 6541285bcc8SWarner Losh pos += optional_data_size; 6551285bcc8SWarner Losh } 6561285bcc8SWarner Losh 6571285bcc8SWarner Losh return pos - buf; 6581285bcc8SWarner Losh } 6591285bcc8SWarner Losh 6601285bcc8SWarner Losh 6611285bcc8SWarner Losh static int 662de26ba4dSWarner Losh make_boot_var(const char *label, const char *loader, const char *kernel, const char *env, bool dry_run, 663de6b4aa8SWarner Losh int bootnum, bool activate) 6641285bcc8SWarner Losh { 6651285bcc8SWarner Losh struct entry *new_ent; 6661285bcc8SWarner Losh uint32_t load_attrs = 0; 6671285bcc8SWarner Losh uint8_t *load_opt_buf; 6681285bcc8SWarner Losh size_t lopt_size, llen, klen; 6691285bcc8SWarner Losh efidp dp, loaderdp, kerneldp; 6701285bcc8SWarner Losh char *bootvar = NULL; 6711285bcc8SWarner Losh int ret; 6721285bcc8SWarner Losh 673da15338dSWarner Losh assert(label != NULL); 674da15338dSWarner Losh 675de26ba4dSWarner Losh if (bootnum == -1) 6761285bcc8SWarner Losh bootvar = make_next_boot_var_name(); 677de26ba4dSWarner Losh else 678de26ba4dSWarner Losh bootvar = make_boot_var_name((uint16_t)bootnum); 6791285bcc8SWarner Losh if (bootvar == NULL) 6801285bcc8SWarner Losh err(1, "bootvar creation"); 6811285bcc8SWarner Losh if (loader == NULL) 6821285bcc8SWarner Losh errx(1, "Must specify boot loader"); 683b8adbe1cSRyan Moeller ret = efivar_unix_path_to_device_path(loader, &loaderdp); 684b8adbe1cSRyan Moeller if (ret != 0) 685b8adbe1cSRyan Moeller errc(1, ret, "Cannot translate unix loader path '%s' to UEFI", 686b8adbe1cSRyan Moeller loader); 6871285bcc8SWarner Losh if (kernel != NULL) { 688b8adbe1cSRyan Moeller ret = efivar_unix_path_to_device_path(kernel, &kerneldp); 689b8adbe1cSRyan Moeller if (ret != 0) 690b8adbe1cSRyan Moeller errc(1, ret, 691b8adbe1cSRyan Moeller "Cannot translate unix kernel path '%s' to UEFI", 692b8adbe1cSRyan Moeller kernel); 6931285bcc8SWarner Losh } else { 6941285bcc8SWarner Losh kerneldp = NULL; 6951285bcc8SWarner Losh } 6961285bcc8SWarner Losh llen = efidp_size(loaderdp); 697af934ea3SWarner Losh if (llen > MAX_DP_LEN) 698af934ea3SWarner Losh errx(1, "Loader path too long."); 6991285bcc8SWarner Losh klen = efidp_size(kerneldp); 700af934ea3SWarner Losh if (klen > MAX_DP_LEN) 701af934ea3SWarner Losh errx(1, "Kernel path too long."); 7021285bcc8SWarner Losh dp = malloc(llen + klen); 703af934ea3SWarner Losh if (dp == NULL) 704af934ea3SWarner Losh errx(1, "Can't allocate memory for new device paths"); 7051285bcc8SWarner Losh memcpy(dp, loaderdp, llen); 7061285bcc8SWarner Losh if (kerneldp != NULL) 7071285bcc8SWarner Losh memcpy((char *)dp + llen, kerneldp, klen); 7081285bcc8SWarner Losh 7091285bcc8SWarner Losh /* don't make the new bootvar active by default, use the -a option later */ 7101285bcc8SWarner Losh load_attrs = LOAD_OPTION_CATEGORY_BOOT; 711de6b4aa8SWarner Losh if (activate) 712de6b4aa8SWarner Losh load_attrs |= LOAD_OPTION_ACTIVE; 7131285bcc8SWarner Losh load_opt_buf = malloc(MAX_LOADOPT_LEN); 7141285bcc8SWarner Losh if (load_opt_buf == NULL) 7151285bcc8SWarner Losh err(1, "malloc"); 7161285bcc8SWarner Losh 7171285bcc8SWarner Losh lopt_size = create_loadopt(load_opt_buf, MAX_LOADOPT_LEN, load_attrs, 7181127aea3SWarner Losh dp, llen + klen, label, env, env ? strlen(env) + 1 : 0); 7191285bcc8SWarner Losh if (lopt_size == BAD_LENGTH) 72067a5e533SEmmanuel Vadot errx(1, "Can't create loadopt"); 7211285bcc8SWarner Losh 7221285bcc8SWarner Losh ret = 0; 7231285bcc8SWarner Losh if (!dry_run) { 7241285bcc8SWarner Losh ret = efi_set_variable(EFI_GLOBAL_GUID, bootvar, 7251285bcc8SWarner Losh (uint8_t*)load_opt_buf, lopt_size, COMMON_ATTRS); 7261285bcc8SWarner Losh } 7271285bcc8SWarner Losh 7281285bcc8SWarner Losh if (ret) 7291285bcc8SWarner Losh err(1, "efi_set_variable"); 7301285bcc8SWarner Losh 73167a5e533SEmmanuel Vadot if (!dry_run) 7321285bcc8SWarner Losh add_to_boot_order(bootvar); /* first, still not active */ 7331285bcc8SWarner Losh new_ent = malloc(sizeof(struct entry)); 7341285bcc8SWarner Losh if (new_ent == NULL) 7351285bcc8SWarner Losh err(1, "malloc"); 7361285bcc8SWarner Losh memset(new_ent, 0, sizeof(struct entry)); 7371285bcc8SWarner Losh new_ent->name = bootvar; 7381285bcc8SWarner Losh new_ent->guid = EFI_GLOBAL_GUID; 7391285bcc8SWarner Losh LIST_INSERT_HEAD(&efivars, new_ent, entries); 740a6b6f888SWarner Losh free(load_opt_buf); 7411285bcc8SWarner Losh free(dp); 742a6b6f888SWarner Losh 7431285bcc8SWarner Losh return 0; 7441285bcc8SWarner Losh } 7451285bcc8SWarner Losh 7461285bcc8SWarner Losh 7471285bcc8SWarner Losh static void 7481285bcc8SWarner Losh print_loadopt_str(uint8_t *data, size_t datalen) 7491285bcc8SWarner Losh { 7501285bcc8SWarner Losh char *dev, *relpath, *abspath; 7511285bcc8SWarner Losh uint32_t attr; 7521285bcc8SWarner Losh uint16_t fplen; 7531285bcc8SWarner Losh efi_char *descr; 7541285bcc8SWarner Losh uint8_t *ep = data + datalen; 7551285bcc8SWarner Losh uint8_t *walker = data; 7561285bcc8SWarner Losh efidp dp, edp; 7571285bcc8SWarner Losh char buf[1024]; 7581285bcc8SWarner Losh int len; 7591285bcc8SWarner Losh int rv; 760a2aa6671SWarner Losh int indent; 7611285bcc8SWarner Losh 7621285bcc8SWarner Losh if (datalen < sizeof(attr) + sizeof(fplen) + sizeof(efi_char)) 7631285bcc8SWarner Losh return; 7641285bcc8SWarner Losh // First 4 bytes are attribute flags 7651285bcc8SWarner Losh attr = le32dec(walker); 7661285bcc8SWarner Losh walker += sizeof(attr); 7671285bcc8SWarner Losh // Next two bytes are length of the file paths 7681285bcc8SWarner Losh fplen = le16dec(walker); 7691285bcc8SWarner Losh walker += sizeof(fplen); 7701285bcc8SWarner Losh // Next we have a 0 terminated UCS2 string that we know to be aligned 7711285bcc8SWarner Losh descr = (efi_char *)(intptr_t)(void *)walker; 7721285bcc8SWarner Losh len = ucs2len(descr); // XXX need to sanity check that len < (datalen - (ep - walker) / 2) 7731285bcc8SWarner Losh walker += (len + 1) * sizeof(efi_char); 7741285bcc8SWarner Losh if (walker > ep) 7751285bcc8SWarner Losh return; 7761285bcc8SWarner Losh // Now we have fplen bytes worth of file path stuff 7771285bcc8SWarner Losh dp = (efidp)walker; 7781285bcc8SWarner Losh walker += fplen; 7791285bcc8SWarner Losh if (walker > ep) 7801285bcc8SWarner Losh return; 7811285bcc8SWarner Losh edp = (efidp)walker; 78296289d09SAlexander Kabaev /* 78396289d09SAlexander Kabaev * Everything left is the binary option args 78496289d09SAlexander Kabaev * opt = walker; 78596289d09SAlexander Kabaev * optlen = ep - walker; 78696289d09SAlexander Kabaev */ 787a2aa6671SWarner Losh indent = 1; 7881285bcc8SWarner Losh while (dp < edp) { 7891285bcc8SWarner Losh efidp_format_device_path(buf, sizeof(buf), dp, 7901285bcc8SWarner Losh (intptr_t)(void *)edp - (intptr_t)(void *)dp); 791a2aa6671SWarner Losh printf("%*s%s\n", indent, "", buf); 792a2aa6671SWarner Losh indent = 10 + len + 1; 7931285bcc8SWarner Losh rv = efivar_device_path_to_unix_path(dp, &dev, &relpath, &abspath); 7941285bcc8SWarner Losh if (rv == 0) { 795a2aa6671SWarner Losh printf("%*s%s:%s %s\n", indent + 4, "", dev, relpath, abspath); 7961285bcc8SWarner Losh free(dev); 7971285bcc8SWarner Losh free(relpath); 7981285bcc8SWarner Losh free(abspath); 7991285bcc8SWarner Losh } 8001285bcc8SWarner Losh dp = (efidp)((char *)dp + efidp_size(dp)); 8011285bcc8SWarner Losh } 8021285bcc8SWarner Losh } 8031285bcc8SWarner Losh 8041285bcc8SWarner Losh static char * 8051285bcc8SWarner Losh get_descr(uint8_t *data) 8061285bcc8SWarner Losh { 8071285bcc8SWarner Losh uint8_t *pos = data; 8081285bcc8SWarner Losh efi_char *desc; 8091285bcc8SWarner Losh int len; 8101285bcc8SWarner Losh char *buf; 8111285bcc8SWarner Losh int i = 0; 8121285bcc8SWarner Losh 8131285bcc8SWarner Losh pos += sizeof(uint32_t) + sizeof(uint16_t); 8141285bcc8SWarner Losh desc = (efi_char*)(intptr_t)(void *)pos; 8151285bcc8SWarner Losh len = ucs2len(desc); 8161285bcc8SWarner Losh buf = malloc(len + 1); 8171285bcc8SWarner Losh memset(buf, 0, len + 1); 8181285bcc8SWarner Losh while (desc[i]) { 8191285bcc8SWarner Losh buf[i] = desc[i]; 8201285bcc8SWarner Losh i++; 8211285bcc8SWarner Losh } 8221285bcc8SWarner Losh return (char*)buf; 8231285bcc8SWarner Losh } 8241285bcc8SWarner Losh 8251285bcc8SWarner Losh 82651922c69SWarner Losh static bool 82751922c69SWarner Losh print_boot_var(const char *name, bool verbose, bool curboot) 82851922c69SWarner Losh { 82951922c69SWarner Losh size_t size; 83051922c69SWarner Losh uint32_t load_attrs; 83151922c69SWarner Losh uint8_t *data; 83251922c69SWarner Losh int ret; 83351922c69SWarner Losh char *d; 83451922c69SWarner Losh 83551922c69SWarner Losh ret = efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, NULL); 83651922c69SWarner Losh if (ret < 0) 83751922c69SWarner Losh return false; 83851922c69SWarner Losh load_attrs = le32dec(data); 83951922c69SWarner Losh d = get_descr(data); 84051922c69SWarner Losh printf("%c%s%c %s", curboot ? '+' : ' ', name, 84151922c69SWarner Losh ((load_attrs & LOAD_OPTION_ACTIVE) ? '*': ' '), d); 84251922c69SWarner Losh free(d); 84351922c69SWarner Losh if (verbose) 84451922c69SWarner Losh print_loadopt_str(data, size); 84551922c69SWarner Losh else 84651922c69SWarner Losh printf("\n"); 84751922c69SWarner Losh return true; 84851922c69SWarner Losh } 84951922c69SWarner Losh 85051922c69SWarner Losh 85183c42372SD Scott Phillips static bool 85283c42372SD Scott Phillips os_indication_supported(uint64_t indication) 85383c42372SD Scott Phillips { 85483c42372SD Scott Phillips uint8_t *data; 85583c42372SD Scott Phillips size_t size; 85683c42372SD Scott Phillips uint32_t attrs; 85783c42372SD Scott Phillips int ret; 85883c42372SD Scott Phillips 85983c42372SD Scott Phillips ret = efi_get_variable(EFI_GLOBAL_GUID, "OsIndicationsSupported", &data, 86083c42372SD Scott Phillips &size, &attrs); 86183c42372SD Scott Phillips if (ret < 0) 86283c42372SD Scott Phillips return false; 86383c42372SD Scott Phillips return (le64dec(data) & indication) == indication; 86483c42372SD Scott Phillips } 86583c42372SD Scott Phillips 86683c42372SD Scott Phillips static uint64_t 86783c42372SD Scott Phillips os_indications(void) 86883c42372SD Scott Phillips { 86983c42372SD Scott Phillips uint8_t *data; 87083c42372SD Scott Phillips size_t size; 87183c42372SD Scott Phillips uint32_t attrs; 87283c42372SD Scott Phillips int ret; 87383c42372SD Scott Phillips 87483c42372SD Scott Phillips ret = efi_get_variable(EFI_GLOBAL_GUID, "OsIndications", &data, &size, 87583c42372SD Scott Phillips &attrs); 87683c42372SD Scott Phillips if (ret < 0) 87783c42372SD Scott Phillips return 0; 87883c42372SD Scott Phillips return le64dec(data); 87983c42372SD Scott Phillips } 88083c42372SD Scott Phillips 88183c42372SD Scott Phillips static int 88283c42372SD Scott Phillips os_indications_set(uint64_t mask, uint64_t val) 88383c42372SD Scott Phillips { 88483c42372SD Scott Phillips uint8_t new[sizeof(uint64_t)]; 88583c42372SD Scott Phillips 88683c42372SD Scott Phillips le64enc(&new, (os_indications() & ~mask) | (val & mask)); 88783c42372SD Scott Phillips return set_bootvar("OsIndications", new, sizeof(new)); 88883c42372SD Scott Phillips } 88983c42372SD Scott Phillips 8901285bcc8SWarner Losh /* Cmd epilogue, or just the default with no args. 8911285bcc8SWarner Losh * The order is [bootnext] bootcurrent, timeout, order, and the bootvars [-v] 8921285bcc8SWarner Losh */ 8931285bcc8SWarner Losh static int 8941285bcc8SWarner Losh print_boot_vars(bool verbose) 8951285bcc8SWarner Losh { 8961285bcc8SWarner Losh /* 8971285bcc8SWarner Losh * just read and print the current values 8981285bcc8SWarner Losh * as a command epilogue 8991285bcc8SWarner Losh */ 9001285bcc8SWarner Losh struct entry *v; 9011285bcc8SWarner Losh uint8_t *data; 9021285bcc8SWarner Losh size_t size; 90351922c69SWarner Losh uint32_t attrs; 90451922c69SWarner Losh int ret, bolen; 90551922c69SWarner Losh uint16_t *boot_order = NULL, current; 90683c42372SD Scott Phillips bool boot_to_fw_ui; 90783c42372SD Scott Phillips 90883c42372SD Scott Phillips if (os_indication_supported(EFI_OS_INDICATIONS_BOOT_TO_FW_UI)) { 90983c42372SD Scott Phillips boot_to_fw_ui = 91083c42372SD Scott Phillips (os_indications() & EFI_OS_INDICATIONS_BOOT_TO_FW_UI) != 0; 91183c42372SD Scott Phillips printf("Boot to FW : %s\n", boot_to_fw_ui != 0 ? 91283c42372SD Scott Phillips "true" : "false"); 91383c42372SD Scott Phillips } 9141285bcc8SWarner Losh 9151285bcc8SWarner Losh ret = efi_get_variable(EFI_GLOBAL_GUID, "BootNext", &data, &size, &attrs); 9161285bcc8SWarner Losh if (ret > 0) { 9171285bcc8SWarner Losh printf("BootNext : %04x\n", le16dec(data)); 9181285bcc8SWarner Losh } 9191285bcc8SWarner Losh 9201285bcc8SWarner Losh ret = efi_get_variable(EFI_GLOBAL_GUID, "BootCurrent", &data, &size,&attrs); 92151922c69SWarner Losh current = le16dec(data); 92251922c69SWarner Losh printf("BootCurrent: %04x\n", current); 9231285bcc8SWarner Losh 9241285bcc8SWarner Losh ret = efi_get_variable(EFI_GLOBAL_GUID, "Timeout", &data, &size, &attrs); 9251285bcc8SWarner Losh if (ret > 0) { 9261285bcc8SWarner Losh printf("Timeout : %d seconds\n", le16dec(data)); 9271285bcc8SWarner Losh } 92812f409ffSWarner Losh 92912f409ffSWarner Losh if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) > 0) { 93012f409ffSWarner Losh if (size % 2 == 1) 93112f409ffSWarner Losh warn("Bad BootOrder variable: odd length %d", (int)size); 93251922c69SWarner Losh boot_order = malloc(size); 93351922c69SWarner Losh bolen = size / 2; 93412f409ffSWarner Losh printf("BootOrder : "); 93551922c69SWarner Losh for (size_t i = 0; i < size; i += 2) { 93651922c69SWarner Losh boot_order[i / 2] = le16dec(data + i); 93751922c69SWarner Losh printf("%04X%s", boot_order[i / 2], i == size - 2 ? "\n" : ", "); 93851922c69SWarner Losh } 93912f409ffSWarner Losh } 9401285bcc8SWarner Losh 94151922c69SWarner Losh if (boot_order == NULL) { 94251922c69SWarner Losh /* 94351922c69SWarner Losh * now we want to fetch 'em all fresh again 9441285bcc8SWarner Losh * which possibly includes a newly created bootvar 9451285bcc8SWarner Losh */ 9461285bcc8SWarner Losh LIST_FOREACH(v, &efivars, entries) { 94751922c69SWarner Losh print_boot_var(v->name, verbose, v->idx == current); 94851922c69SWarner Losh } 94951922c69SWarner Losh } else { 95051922c69SWarner Losh LIST_FOREACH(v, &efivars, entries) { 95151922c69SWarner Losh v->flags = 0; 95251922c69SWarner Losh } 95351922c69SWarner Losh for (int i = 0; i < bolen; i++) { 95451922c69SWarner Losh char buffer[10]; 95551922c69SWarner Losh 95651922c69SWarner Losh snprintf(buffer, sizeof(buffer), "Boot%04X", boot_order[i]); 95751922c69SWarner Losh if (!print_boot_var(buffer, verbose, boot_order[i] == current)) 95851922c69SWarner Losh printf("%s: MISSING!\n", buffer); 95951922c69SWarner Losh LIST_FOREACH(v, &efivars, entries) { 96051922c69SWarner Losh if (v->idx == boot_order[i]) { 96151922c69SWarner Losh v->flags |= SEEN; 96251922c69SWarner Losh break; 96351922c69SWarner Losh } 96451922c69SWarner Losh } 96551922c69SWarner Losh } 96651922c69SWarner Losh if (verbose) { 96751922c69SWarner Losh printf("\n\nUnreferenced Variables:\n"); 96851922c69SWarner Losh LIST_FOREACH(v, &efivars, entries) { 96951922c69SWarner Losh if (v->flags == 0) 97051922c69SWarner Losh print_boot_var(v->name, verbose, v->idx == current); 97151922c69SWarner Losh } 97251922c69SWarner Losh } 9731285bcc8SWarner Losh } 9741285bcc8SWarner Losh return 0; 9751285bcc8SWarner Losh } 9761285bcc8SWarner Losh 9771285bcc8SWarner Losh static void 9786604afe9SWarner Losh delete_timeout(void) 9791285bcc8SWarner Losh { 9801285bcc8SWarner Losh 9811285bcc8SWarner Losh efi_del_variable(EFI_GLOBAL_GUID,"Timeout"); 9821285bcc8SWarner Losh } 9831285bcc8SWarner Losh 9841285bcc8SWarner Losh static void 9851285bcc8SWarner Losh handle_timeout(int to) 9861285bcc8SWarner Losh { 9871285bcc8SWarner Losh uint16_t timeout; 9881285bcc8SWarner Losh 9891285bcc8SWarner Losh le16enc(&timeout, to); 990c66805a5SWarner Losh if (set_bootvar("Timeout", (uint8_t *)&timeout, sizeof(timeout)) < 0) 991c66805a5SWarner Losh errx(1, "Can't set Timeout for booting."); 9921285bcc8SWarner Losh } 9931285bcc8SWarner Losh 9941cdb8eb8SWarner Losh static void 9951cdb8eb8SWarner Losh report_esp_device(bool do_dp, bool do_unix) 9961cdb8eb8SWarner Losh { 9971cdb8eb8SWarner Losh uint8_t *data; 9981cdb8eb8SWarner Losh size_t size, len; 9991cdb8eb8SWarner Losh uint32_t attrs; 10001cdb8eb8SWarner Losh int ret; 10011cdb8eb8SWarner Losh uint16_t current, fplen; 10021cdb8eb8SWarner Losh char *name, *dev, *relpath, *abspath; 10031cdb8eb8SWarner Losh uint8_t *walker, *ep; 10041cdb8eb8SWarner Losh efi_char *descr; 1005*9a791529SWarner Losh efidp dp; 10061cdb8eb8SWarner Losh char buf[PATH_MAX]; 10071cdb8eb8SWarner Losh 10081cdb8eb8SWarner Losh if (do_dp && do_unix) 10091cdb8eb8SWarner Losh errx(1, "Can't report both UEFI device-path and Unix path together"); 10101cdb8eb8SWarner Losh 10111cdb8eb8SWarner Losh ret = efi_get_variable(EFI_GLOBAL_GUID, "BootCurrent", &data, &size,&attrs); 10121cdb8eb8SWarner Losh if (ret < 0) 10131cdb8eb8SWarner Losh err(1, "Can't get BootCurrent"); 10141cdb8eb8SWarner Losh current = le16dec(data); 10151cdb8eb8SWarner Losh if (asprintf(&name, "Boot%04X", current) < 0) 10161cdb8eb8SWarner Losh err(1, "Can't format boot var\n"); 10171cdb8eb8SWarner Losh if (efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, NULL) < 0) 10181cdb8eb8SWarner Losh err(1, "Can't retrieve EFI var %s", name); 10191cdb8eb8SWarner Losh // First 4 bytes are attribute flags 10201cdb8eb8SWarner Losh walker = data; 10211cdb8eb8SWarner Losh ep = walker + size; 10221cdb8eb8SWarner Losh walker += sizeof(uint32_t); 10231cdb8eb8SWarner Losh // Next two bytes are length of the file paths 10241cdb8eb8SWarner Losh fplen = le16dec(walker); 10251cdb8eb8SWarner Losh walker += sizeof(fplen); 10261cdb8eb8SWarner Losh // Next we have a 0 terminated UCS2 string that we know to be aligned 10271cdb8eb8SWarner Losh descr = (efi_char *)(intptr_t)(void *)walker; 10281cdb8eb8SWarner Losh len = ucs2len(descr); // XXX need to sanity check that len < (datalen - (ep - walker) / 2) 10291cdb8eb8SWarner Losh walker += (len + 1) * sizeof(efi_char); 10301cdb8eb8SWarner Losh if (walker > ep) 10311cdb8eb8SWarner Losh errx(1, "malformed boot variable %s", name); 10321cdb8eb8SWarner Losh // Now we have fplen bytes worth of file path stuff 10331cdb8eb8SWarner Losh dp = (efidp)walker; 10341cdb8eb8SWarner Losh walker += fplen; 10351cdb8eb8SWarner Losh if (walker > ep) 10361cdb8eb8SWarner Losh errx(1, "malformed boot variable %s", name); 10371cdb8eb8SWarner Losh if (do_dp) { 10381cdb8eb8SWarner Losh efidp_format_device_path_node(buf, sizeof(buf), dp); 10391cdb8eb8SWarner Losh printf("%s\n", buf); 10401cdb8eb8SWarner Losh exit(0); 10411cdb8eb8SWarner Losh } 10422e085395SMitchell Horne if (efivar_device_path_to_unix_path(dp, &dev, &relpath, &abspath) != 0) 10431cdb8eb8SWarner Losh errx(1, "Can't convert to unix path"); 10441cdb8eb8SWarner Losh if (do_unix) { 10451cdb8eb8SWarner Losh if (abspath == NULL) 10461cdb8eb8SWarner Losh errx(1, "Can't find where %s:%s is mounted", 10471cdb8eb8SWarner Losh dev, relpath); 10481cdb8eb8SWarner Losh abspath[strlen(abspath) - strlen(relpath) - 1] = '\0'; 10491cdb8eb8SWarner Losh printf("%s\n", abspath); 10501cdb8eb8SWarner Losh } else { 10511cdb8eb8SWarner Losh printf("%s\n", dev); 10521cdb8eb8SWarner Losh } 10531cdb8eb8SWarner Losh free(dev); 10541cdb8eb8SWarner Losh free(relpath); 10551cdb8eb8SWarner Losh free(abspath); 10561cdb8eb8SWarner Losh exit(0); 10571cdb8eb8SWarner Losh } 10581cdb8eb8SWarner Losh 105983c42372SD Scott Phillips static void 106083c42372SD Scott Phillips set_boot_to_fw_ui(bool to_fw) 106183c42372SD Scott Phillips { 106283c42372SD Scott Phillips int ret; 106383c42372SD Scott Phillips 106483c42372SD Scott Phillips if (!os_indication_supported(EFI_OS_INDICATIONS_BOOT_TO_FW_UI)) { 106583c42372SD Scott Phillips if (to_fw) 106683c42372SD Scott Phillips errx(1, "boot to fw ui not supported"); 106783c42372SD Scott Phillips else 106883c42372SD Scott Phillips return; 106983c42372SD Scott Phillips } 107083c42372SD Scott Phillips ret = os_indications_set(EFI_OS_INDICATIONS_BOOT_TO_FW_UI, 107183c42372SD Scott Phillips to_fw ? ~0 : 0); 107283c42372SD Scott Phillips if (ret < 0) 107383c42372SD Scott Phillips errx(1, "failed to set boot to fw ui"); 107483c42372SD Scott Phillips } 107583c42372SD Scott Phillips 1076*9a791529SWarner Losh static void 1077*9a791529SWarner Losh find_efi_device(const char *path) 1078*9a791529SWarner Losh { 1079*9a791529SWarner Losh efidp dp = NULL; 1080*9a791529SWarner Losh size_t len; 1081*9a791529SWarner Losh int ret; 1082*9a791529SWarner Losh char buf[1024]; 1083*9a791529SWarner Losh 1084*9a791529SWarner Losh ret = efivar_unix_path_to_device_path(path, &dp); 1085*9a791529SWarner Losh if (ret != 0) 1086*9a791529SWarner Losh errc(1, ret, 1087*9a791529SWarner Losh "Cannot translate path '%s' to UEFI", path); 1088*9a791529SWarner Losh len = efidp_size(dp); 1089*9a791529SWarner Losh if (len > MAX_DP_LEN) 1090*9a791529SWarner Losh errx(1, "Resulting device path too long."); 1091*9a791529SWarner Losh efidp_format_device_path(buf, sizeof(buf), dp, len); 1092*9a791529SWarner Losh printf("%s -> %s\n", path, buf); 1093*9a791529SWarner Losh exit (0); 1094*9a791529SWarner Losh } 1095*9a791529SWarner Losh 10961285bcc8SWarner Losh int 10971285bcc8SWarner Losh main(int argc, char *argv[]) 10981285bcc8SWarner Losh { 10991285bcc8SWarner Losh 11007fe2f504SWarner Losh memset(&opts, 0, sizeof (bmgr_opts_t)); 11017fe2f504SWarner Losh parse_args(argc, argv); 11027fe2f504SWarner Losh 1103*9a791529SWarner Losh /* 1104*9a791529SWarner Losh * find_dev can operate without any efi variables 1105*9a791529SWarner Losh */ 1106*9a791529SWarner Losh if (!efi_variables_supported() && !opts.find_dev) 11071285bcc8SWarner Losh errx(1, "efi variables not supported on this system. root? kldload efirt?"); 11081285bcc8SWarner Losh 11091285bcc8SWarner Losh read_vars(); 11101285bcc8SWarner Losh 11111285bcc8SWarner Losh if (opts.create) 11121285bcc8SWarner Losh /* 11131285bcc8SWarner Losh * side effect, adds to boot order, but not yet active. 11141285bcc8SWarner Losh */ 1115da15338dSWarner Losh make_boot_var(opts.label ? opts.label : "", 1116de26ba4dSWarner Losh opts.loader, opts.kernel, opts.env, opts.dry_run, 1117de6b4aa8SWarner Losh opts.has_bootnum ? opts.bootnum : -1, opts.set_active); 11181285bcc8SWarner Losh else if (opts.set_active || opts.set_inactive ) 11191285bcc8SWarner Losh handle_activity(opts.bootnum, opts.set_active); 11201285bcc8SWarner Losh else if (opts.order != NULL) 11211285bcc8SWarner Losh set_boot_order(opts.order); /* create a new bootorder with opts.order */ 11221285bcc8SWarner Losh else if (opts.set_bootnext) 11231285bcc8SWarner Losh handle_bootnext(opts.bootnum); 11241285bcc8SWarner Losh else if (opts.delete_bootnext) 11251285bcc8SWarner Losh del_bootnext(); 11261285bcc8SWarner Losh else if (opts.delete) 11271285bcc8SWarner Losh delete_bootvar(opts.bootnum); 11281285bcc8SWarner Losh else if (opts.del_timeout) 11291285bcc8SWarner Losh delete_timeout(); 11301285bcc8SWarner Losh else if (opts.set_timeout) 11311285bcc8SWarner Losh handle_timeout(opts.timeout); 11321cdb8eb8SWarner Losh else if (opts.esp_device) 11331cdb8eb8SWarner Losh report_esp_device(opts.device_path, opts.unix_path); 113483c42372SD Scott Phillips else if (opts.fw_ui) 113583c42372SD Scott Phillips set_boot_to_fw_ui(true); 113683c42372SD Scott Phillips else if (opts.no_fw_ui) 113783c42372SD Scott Phillips set_boot_to_fw_ui(false); 1138*9a791529SWarner Losh else if (opts.find_dev) 1139*9a791529SWarner Losh find_efi_device(opts.dev); 11401285bcc8SWarner Losh 11411285bcc8SWarner Losh print_boot_vars(opts.verbose); 11421285bcc8SWarner Losh } 1143