usr.sbin/efibootmgr/efibootmgr.c

   1 /*-
   2  * Copyright (c) 2017-2018 Netflix, Inc.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer
   9  *    in this position and unchanged.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
  15  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  16  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  17  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
  18  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  19  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  20  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  21  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  23  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #include <sys/cdefs.h>
  27 #include <sys/stat.h>
  28 #include <sys/param.h>
  29 #include <assert.h>
  30 #include <ctype.h>
  31 #include <err.h>
  32 #include <errno.h>
  33 #include <fcntl.h>
  34 #include <libgeom.h>
  35 #include <paths.h>
  36 #include <signal.h>
  37 #include <stdint.h>
  38 #include <stdio.h>
  39 #include <stdlib.h>
  40 #include <getopt.h>
  41 #include <limits.h>
  42 #include <inttypes.h>
  43 #include <stdbool.h>
  44 #include <string.h>
  45 #include <strings.h>
  46 #include <unistd.h>
  47 #include <libgeom.h>
  48 #include <geom/geom.h>
  49 #include <geom/geom_ctl.h>
  50 #include <geom/geom_int.h>
  51
  52 #include <efivar.h>
  53 #include <efiutil.h>
  54 #include <efichar.h>
  55 #include <efivar-dp.h>
  56
  57 #ifndef LOAD_OPTION_ACTIVE
  58 #define LOAD_OPTION_ACTIVE 0x00000001
  59 #endif
  60
  61 #ifndef LOAD_OPTION_CATEGORY_BOOT
  62 #define LOAD_OPTION_CATEGORY_BOOT 0x00000000
  63 #endif
  64
  65 #define BAD_LENGTH      ((size_t)-1)
  66
  67 #define EFI_OS_INDICATIONS_BOOT_TO_FW_UI 0x0000000000000001
  68
  69 typedef struct _bmgr_opts {
  70         char    *dev;
  71         char    *env;
  72         char    *loader;
  73         char    *label;
  74         char    *kernel;
  75         char    *name;
  76         char    *order;
  77         int     bootnum;
  78         bool    copy;
  79         bool    create;
  80         bool    delete;
  81         bool    delete_bootnext;
  82         bool    del_timeout;
  83         bool    dry_run;
  84         bool    device_path;
  85         bool    esp_device;
  86         bool    find_dev;
  87         bool    fw_ui;
  88         bool    no_fw_ui;
  89         bool    has_bootnum;
  90         bool    once;
  91         int     cp_src;
  92         bool    set_active;
  93         bool    set_bootnext;
  94         bool    set_inactive;
  95         bool    set_timeout;
  96         int     timeout;
  97         bool    unix_path;
  98         bool    verbose;
  99 } bmgr_opts_t;
 100
 101 static struct option lopts[] = {
 102         {"activate", no_argument, NULL, 'a'},
 103         {"bootnext", no_argument, NULL, 'n'}, /* set bootnext */
 104         {"bootnum", required_argument, NULL, 'b'},
 105         {"bootorder", required_argument, NULL, 'o'}, /* set order */
 106         {"copy", required_argument, NULL, 'C'},         /* Copy boot method */
 107         {"create", no_argument, NULL, 'c'},
 108         {"deactivate", no_argument, NULL, 'A'},
 109         {"del-timeout", no_argument, NULL, 'T'},
 110         {"delete", no_argument, NULL, 'B'},
 111         {"delete-bootnext", no_argument, NULL, 'N'},
 112         {"device-path", no_argument, NULL, 'd'},
 113         {"dry-run", no_argument, NULL, 'D'},
 114         {"env", required_argument, NULL, 'e'},
 115         {"esp", no_argument, NULL, 'E'},
 116         {"efidev", required_argument, NULL, 'u'},
 117         {"fw-ui", no_argument, NULL, 'f'},
 118         {"no-fw-ui", no_argument, NULL, 'F'},
 119         {"help", no_argument, NULL, 'h'},
 120         {"kernel", required_argument, NULL, 'k'},
 121         {"label", required_argument, NULL, 'L'},
 122         {"loader", required_argument, NULL, 'l'},
 123         {"once", no_argument, NULL, 'O'},
 124         {"set-timeout", required_argument, NULL, 't'},
 125         {"unix-path", no_argument, NULL, 'p'},
 126         {"verbose", no_argument, NULL, 'v'},
 127         { NULL, 0, NULL, 0}
 128 };
 129
 130 /* global efibootmgr opts */
 131 static bmgr_opts_t opts;
 132
 133 static LIST_HEAD(efivars_head, entry) efivars =
 134         LIST_HEAD_INITIALIZER(efivars);
 135
 136 struct entry {
 137         efi_guid_t      guid;
 138         uint32_t        attrs;
 139         uint8_t         *data;
 140         size_t          size;
 141         char            *name;
 142         char            *label;
 143         int             idx;
 144         int             flags;
 145 #define SEEN    1
 146
 147         LIST_ENTRY(entry) entries;
 148 };
 149
 150 #define MAX_DP_LEN      4096
 151 #define MAX_LOADOPT_LEN 8192
 152
 153
 154 static char *
 155 mangle_loader(char *loader)
 156 {
 157         char *c;
 158
 159         for (c = loader; *c; c++)
 160                 if (*c == '/')
 161                         *c = '\\';
 162
 163         return loader;
 164 }
 165
 166
 167 #define COMMON_ATTRS EFI_VARIABLE_NON_VOLATILE | \
 168         EFI_VARIABLE_BOOTSERVICE_ACCESS | \
 169         EFI_VARIABLE_RUNTIME_ACCESS
 170
 171 /*
 172  * We use global guid, and common var attrs and
 173  * find it better to just delete and re-create a var.
 174  */
 175 static int
 176 set_bootvar(const char *name, uint8_t *data, size_t size)
 177 {
 178
 179         return efi_set_variable(EFI_GLOBAL_GUID, name, data, size,
 180             COMMON_ATTRS);
 181 }
 182
 183
 184 #define USAGE \
 185         "   [-aAnB -b bootnum] [-N] [-t timeout] [-T] [-o bootorder] [-O] [--verbose] [--help]\n\
 186   [-c -l loader [-k kernel] [-L label] [--dry-run] [-b bootnum]]"
 187
 188 #define CREATE_USAGE \
 189         "       efibootmgr -c -l loader [-k kernel] [-L label] [--dry-run] [-b bootnum] [-a]"
 190 #define ORDER_USAGE \
 191         "       efibootmgr -o bootvarnum1,bootvarnum2,..."
 192 #define TIMEOUT_USAGE \
 193         "       efibootmgr -t seconds"
 194 #define DELETE_USAGE \
 195         "       efibootmgr -B -b bootnum"
 196 #define ACTIVE_USAGE \
 197         "       efibootmgr [-a | -A] -b bootnum"
 198 #define BOOTNEXT_USAGE \
 199         "       efibootmgr [-n | -N] -b bootnum"
 200
 201 static void
 202 parse_args(int argc, char *argv[])
 203 {
 204         int ch;
 205         const char *arg;
 206
 207         while ((ch = getopt_long(argc, argv, "AaBb:C:cdDe:EFfhk:L:l:NnOo:pTt:v",
 208                     lopts, NULL)) != -1) {
 209                 switch (ch) {
 210                 case 'A':
 211                         opts.set_inactive = true;
 212                         break;
 213                 case 'a':
 214                         opts.set_active = true;
 215                         break;
 216                 case 'b':
 217                         opts.has_bootnum = true;
 218                         arg = optarg;
 219                         if (strncasecmp(arg, "boot", 4) == 0)
 220                                 arg += 4;
 221                         opts.bootnum = strtoul(arg, NULL, 16);
 222                         break;
 223                 case 'B':
 224                         opts.delete = true;
 225                         break;
 226                 case 'C':
 227                         opts.copy = true;
 228                         opts.cp_src = strtoul(optarg, NULL, 16);
 229                         errx(1, "Copy not implemented");
 230                         break;
 231                 case 'c':
 232                         opts.create = true;
 233                         break;
 234                 case 'D': /* should be remove dups XXX */
 235                         opts.dry_run = true;
 236                         break;
 237                 case 'd':
 238                         opts.device_path = true;
 239                         break;
 240                 case 'e':
 241                         free(opts.env);
 242                         opts.env = strdup(optarg);
 243                         break;
 244                 case 'E':
 245                         opts.esp_device = true;
 246                         break;
 247                 case 'F':
 248                         opts.no_fw_ui = true;
 249                         break;
 250                 case 'f':
 251                         opts.fw_ui = true;
 252                         break;
 253                 case 'h':
 254                 default:
 255                         errx(1, "%s", USAGE);
 256                         break;
 257                 case 'k':
 258                         free(opts.kernel);
 259                         opts.kernel = strdup(optarg);
 260                         break;
 261                 case 'L':
 262                         free(opts.label);
 263                         opts.label = strdup(optarg);
 264                         break;
 265                 case 'l':
 266                         free(opts.loader);
 267                         opts.loader = strdup(optarg);
 268                         opts.loader = mangle_loader(opts.loader);
 269                         break;
 270                 case 'N':
 271                         opts.delete_bootnext = true;
 272                         break;
 273                 case 'n':
 274                         opts.set_bootnext = true;
 275                         break;
 276                 case 'O':
 277                         opts.once = true;
 278                         break;
 279                 case 'o':
 280                         free(opts.order);
 281                         opts.order = strdup(optarg);
 282                         break;
 283                 case 'p':
 284                         opts.unix_path = true;
 285                         break;
 286                 case 'T':
 287                         opts.del_timeout = true;
 288                         break;
 289                 case 't':
 290                         opts.set_timeout = true;
 291                         opts.timeout = strtoul(optarg, NULL, 10);
 292                         break;
 293                 case 'u':
 294                         opts.find_dev = true;
 295                         opts.dev = strdup(optarg);
 296                         break;
 297                 case 'v':
 298                         opts.verbose = true;
 299                         break;
 300                 }
 301         }
 302         if (opts.create) {
 303                 if (!opts.loader)
 304                         errx(1, "%s",CREATE_USAGE);
 305                 return;
 306         }
 307
 308         if (opts.order != NULL && *opts.order == '\0')
 309                 errx(1, "%s", ORDER_USAGE);
 310
 311         if ((opts.set_inactive || opts.set_active) && !opts.has_bootnum)
 312                 errx(1, "%s", ACTIVE_USAGE);
 313
 314         if (opts.delete && !opts.has_bootnum)
 315                 errx(1, "%s", DELETE_USAGE);
 316
 317         if (opts.set_bootnext && !opts.has_bootnum)
 318                 errx(1, "%s", BOOTNEXT_USAGE);
 319 }
 320
 321
 322 static void
 323 read_vars(void)
 324 {
 325
 326         efi_guid_t *guid;
 327         char *next_name = NULL;
 328         int ret = 0;
 329
 330         struct entry *nent;
 331
 332         LIST_INIT(&efivars);
 333         while ((ret = efi_get_next_variable_name(&guid, &next_name)) > 0) {
 334                 /*
 335                  * Only pay attention to EFI:BootXXXX variables to get the list.
 336                  */
 337                 if (efi_guid_cmp(guid, &EFI_GLOBAL_GUID) != 0 ||
 338                     strlen(next_name) != 8 ||
 339                     strncmp(next_name, "Boot", 4) != 0 ||
 340                     !isxdigit(next_name[4]) ||
 341                     !isxdigit(next_name[5]) ||
 342                     !isxdigit(next_name[6]) ||
 343                     !isxdigit(next_name[7]))
 344                         continue;
 345                 nent = malloc(sizeof(struct entry));
 346                 nent->name = strdup(next_name);
 347
 348                 ret = efi_get_variable(*guid, next_name, &nent->data,
 349                     &nent->size, &nent->attrs);
 350                 if (ret < 0)
 351                         err(1, "efi_get_variable");
 352                 nent->guid = *guid;
 353                 nent->idx = strtoul(&next_name[4], NULL, 16);
 354                 LIST_INSERT_HEAD(&efivars, nent, entries);
 355         }
 356 }
 357
 358
 359 static void
 360 set_boot_order(char *order)
 361 {
 362         uint16_t *new_data;
 363         size_t size;
 364         char *next, *cp;
 365         int cnt;
 366         int i;
 367
 368         cp = order;
 369         cnt = 1;
 370         while (*cp) {
 371                 if (*cp++ == ',')
 372                         cnt++;
 373         }
 374         size = sizeof(uint16_t) * cnt;
 375         new_data = malloc(size);
 376
 377         i = 0;
 378         cp = strdup(order);
 379         while ((next = strsep(&cp, ",")) != NULL) {
 380                 new_data[i] = strtoul(next, NULL, 16);
 381                 if (new_data[i] == 0 && errno == EINVAL) {
 382                         warnx("can't parse %s as a numb", next);
 383                         errx(1, "%s", ORDER_USAGE);
 384                 }
 385                 i++;
 386         }
 387         free(cp);
 388         if (set_bootvar("BootOrder", (uint8_t*)new_data, size) < 0)
 389                 err(1, "Unabke to set BootOrder to %s", order);
 390         free(new_data);
 391 }
 392
 393 static void
 394 handle_activity(int bootnum, bool active)
 395 {
 396         uint32_t attrs, load_attrs;
 397         uint8_t *data;
 398         size_t size;
 399         char *name;
 400
 401         asprintf(&name, "%s%04X", "Boot", bootnum);
 402         if (name == NULL)
 403                 err(1, "asprintf");
 404         if (efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, &attrs) < 0)
 405                 err(1, "No such bootvar %s\n", name);
 406
 407         load_attrs = le32dec(data);
 408
 409         if (active)
 410                 load_attrs |= LOAD_OPTION_ACTIVE;
 411         else
 412                 load_attrs &= ~LOAD_OPTION_ACTIVE;
 413
 414         le32enc(data, load_attrs);
 415
 416         if (set_bootvar(name, data, size) < 0)
 417                 err(1, "handle activity efi_set_variable");
 418 }
 419
 420
 421 /*
 422  * add boot var to boot order.
 423  * called by create boot var. There is no option
 424  * to add one independent of create.
 425  *
 426  * Note: we currently don't support where it goes
 427  * so it goes on the front, inactive.
 428  * use -o 2,3,7 etc to affect order, -a to activate.
 429  */
 430 static void
 431 add_to_boot_order(char *bootvar)
 432 {
 433         size_t size;
 434         uint32_t attrs;
 435         uint16_t val;
 436         uint8_t *data, *new;
 437
 438         val = strtoul(&bootvar[4], NULL, 16);
 439
 440         if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) < 0) {
 441                 if (errno == ENOENT) { /* create it and set this bootvar to active */
 442                         size = 0;
 443                         data = NULL;
 444                 } else
 445                         err(1, "efi_get_variable BootOrder");
 446         }
 447
 448         /*
 449          * We have BootOrder with the current order
 450          * so grow the array by one, add the value
 451          * and write the new variable value.
 452          */
 453         size += sizeof(uint16_t);
 454         new = malloc(size);
 455         if (!new)
 456                 err(1, "malloc");
 457
 458         le16enc(new, val);
 459         if (size > sizeof(uint16_t))
 460                 memcpy(new + sizeof(uint16_t), data, size - sizeof(uint16_t));
 461
 462         if (set_bootvar("BootOrder", new, size) < 0)
 463                 err(1, "set_bootvar");
 464         free(new);
 465 }
 466
 467
 468 static void
 469 remove_from_order(uint16_t bootnum)
 470 {
 471         uint32_t attrs;
 472         size_t size, i, j;
 473         uint8_t *new, *data;
 474
 475         if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) < 0)
 476                 return;
 477
 478         new = malloc(size);
 479         if (new == NULL)
 480                 err(1, "malloc");
 481
 482         for (j = i = 0; i < size; i += sizeof(uint16_t)) {
 483                 if (le16dec(data + i) == bootnum)
 484                         continue;
 485                 memcpy(new + j, data + i, sizeof(uint16_t));
 486                 j += sizeof(uint16_t);
 487         }
 488         if (i == j)
 489                 warnx("Boot variable %04x not in BootOrder", bootnum);
 490         else if (set_bootvar("BootOrder", new, j) < 0)
 491                 err(1, "Unable to update BootOrder with new value");
 492         free(new);
 493 }
 494
 495
 496 static void
 497 delete_bootvar(int bootnum)
 498 {
 499         char *name;
 500         int defer = 0;
 501
 502         /*
 503          * Try to delete the boot variable and remocve it
 504          * from the boot order. We always do both actions
 505          * to make it easy to clean up from oopses.
 506          */
 507         if (bootnum < 0 || bootnum > 0xffff)
 508                 errx(1, "Bad boot variable %#x", bootnum);
 509         asprintf(&name, "%s%04X", "Boot", bootnum);
 510         if (name == NULL)
 511                 err(1, "asprintf");
 512         printf("Removing boot variable '%s'\n", name);
 513         if (efi_del_variable(EFI_GLOBAL_GUID, name) < 0) {
 514                 defer = 1;
 515                 warn("cannot delete variable %s", name);
 516         }
 517         printf("Removing 0x%x from BootOrder\n", bootnum);
 518         remove_from_order(bootnum);
 519         free(name);
 520         if (defer)
 521                 exit(defer);
 522 }
 523
 524
 525 static void
 526 del_bootnext(void)
 527 {
 528
 529         if (efi_del_variable(EFI_GLOBAL_GUID, "BootNext") < 0)
 530                 err(1, "efi_del_variable");
 531 }
 532
 533 static void
 534 handle_bootnext(uint16_t bootnum)
 535 {
 536         uint16_t num;
 537
 538         le16enc(&num, bootnum);
 539         if (set_bootvar("BootNext", (uint8_t*)&bootnum, sizeof(uint16_t)) < 0)
 540                 err(1, "set_bootvar");
 541 }
 542
 543
 544 static int
 545 compare(const void *a, const void *b)
 546 {
 547         uint16_t c;
 548         uint16_t d;
 549
 550         memcpy(&c, a, sizeof(uint16_t));
 551         memcpy(&d, b, sizeof(uint16_t));
 552
 553         if (c < d)
 554                 return -1;
 555         if (c == d)
 556                 return  0;
 557         return  1;
 558 }
 559
 560 static char *
 561 make_next_boot_var_name(void)
 562 {
 563         struct entry *v;
 564         uint16_t *vals;
 565         char *name;
 566         int cnt = 0;
 567         int i;
 568
 569         LIST_FOREACH(v, &efivars, entries) {
 570                 cnt++;
 571         }
 572
 573         vals = malloc(sizeof(uint16_t) * cnt);
 574         if (!vals)
 575                 return NULL;
 576
 577         i = 0;
 578         LIST_FOREACH(v, &efivars, entries) {
 579                 vals[i++] = v->idx;
 580         }
 581         qsort(vals, cnt, sizeof(uint16_t), compare);
 582         /* Find the first hole (could be at start or end) */
 583         for (i = 0; i < cnt; ++i)
 584                 if (vals[i] != i)
 585                         break;
 586         free(vals);
 587         /* In theory we could have used all 65k slots -- what to do? */
 588
 589         asprintf(&name, "%s%04X", "Boot", i);
 590         if (name == NULL)
 591                 err(1, "asprintf");
 592         return name;
 593 }
 594
 595 static char *
 596 make_boot_var_name(uint16_t bootnum)
 597 {
 598         struct entry *v;
 599         char *name;
 600
 601         LIST_FOREACH(v, &efivars, entries) {
 602                 if (v->idx == bootnum)
 603                         return NULL;
 604         }
 605
 606         asprintf(&name, "%s%04X", "Boot", bootnum);
 607         if (name == NULL)
 608                 err(1, "asprintf");
 609         return name;
 610 }
 611
 612 static size_t
 613 create_loadopt(uint8_t *buf, size_t bufmax, uint32_t attributes, efidp dp, size_t dp_size,
 614     const char *description, const uint8_t *optional_data, size_t optional_data_size)
 615 {
 616         efi_char *bbuf = NULL;
 617         uint8_t *pos = buf;
 618         size_t desc_len = 0;
 619         size_t len;
 620
 621         if (optional_data == NULL && optional_data_size != 0)
 622                 return BAD_LENGTH;
 623         if (dp == NULL && dp_size != 0)
 624                 return BAD_LENGTH;
 625
 626         /*
 627          * Compute the length to make sure the passed in buffer is long enough.
 628          */
 629         utf8_to_ucs2(description, &bbuf, &desc_len);
 630         len = sizeof(uint32_t) + sizeof(uint16_t) + desc_len + dp_size + optional_data_size;
 631         if (len > bufmax) {
 632                 free(bbuf);
 633                 return BAD_LENGTH;
 634         }
 635
 636         le32enc(pos, attributes);
 637         pos += sizeof (attributes);
 638
 639         le16enc(pos, dp_size);
 640         pos += sizeof (uint16_t);
 641
 642         memcpy(pos, bbuf, desc_len);    /* NB:desc_len includes strailing NUL */
 643         pos += desc_len;
 644         free(bbuf);
 645
 646         memcpy(pos, dp, dp_size);
 647         pos += dp_size;
 648
 649         if (optional_data && optional_data_size > 0) {
 650                 memcpy(pos, optional_data, optional_data_size);
 651                 pos += optional_data_size;
 652         }
 653
 654         return pos - buf;
 655 }
 656
 657
 658 static int
 659 make_boot_var(const char *label, const char *loader, const char *kernel, const char *env, bool dry_run,
 660     int bootnum, bool activate)
 661 {
 662         struct entry *new_ent;
 663         uint32_t load_attrs = 0;
 664         uint8_t *load_opt_buf;
 665         size_t lopt_size, llen, klen;
 666         efidp dp, loaderdp, kerneldp;
 667         char *bootvar = NULL;
 668         int ret;
 669
 670         assert(label != NULL);
 671
 672         if (bootnum == -1)
 673                 bootvar = make_next_boot_var_name();
 674         else
 675                 bootvar = make_boot_var_name((uint16_t)bootnum);
 676         if (bootvar == NULL)
 677                 err(1, "bootvar creation");
 678         if (loader == NULL)
 679                 errx(1, "Must specify boot loader");
 680         ret = efivar_unix_path_to_device_path(loader, &loaderdp);
 681         if (ret != 0)
 682                 errc(1, ret, "Cannot translate unix loader path '%s' to UEFI",
 683                     loader);
 684         if (kernel != NULL) {
 685                 ret = efivar_unix_path_to_device_path(kernel, &kerneldp);
 686                 if (ret != 0)
 687                         errc(1, ret,
 688                             "Cannot translate unix kernel path '%s' to UEFI",
 689                             kernel);
 690         } else {
 691                 kerneldp = NULL;
 692         }
 693         llen = efidp_size(loaderdp);
 694         if (llen > MAX_DP_LEN)
 695                 errx(1, "Loader path too long.");
 696         klen = efidp_size(kerneldp);
 697         if (klen > MAX_DP_LEN)
 698                 errx(1, "Kernel path too long.");
 699         dp = malloc(llen + klen);
 700         if (dp == NULL)
 701                 errx(1, "Can't allocate memory for new device paths");
 702         memcpy(dp, loaderdp, llen);
 703         if (kerneldp != NULL)
 704                 memcpy((char *)dp + llen, kerneldp, klen);
 705
 706         /* don't make the new bootvar active by default, use the -a option later */
 707         load_attrs = LOAD_OPTION_CATEGORY_BOOT;
 708         if (activate)
 709                 load_attrs |= LOAD_OPTION_ACTIVE;
 710         load_opt_buf = malloc(MAX_LOADOPT_LEN);
 711         if (load_opt_buf == NULL)
 712                 err(1, "malloc");
 713
 714         lopt_size = create_loadopt(load_opt_buf, MAX_LOADOPT_LEN, load_attrs,
 715             dp, llen + klen, label, env, env ? strlen(env) + 1 : 0);
 716         if (lopt_size == BAD_LENGTH)
 717                 errx(1, "Can't create loadopt");
 718
 719         ret = 0;
 720         if (!dry_run) {
 721                 ret = efi_set_variable(EFI_GLOBAL_GUID, bootvar,
 722                     (uint8_t*)load_opt_buf, lopt_size, COMMON_ATTRS);
 723         }
 724
 725         if (ret)
 726                 err(1, "efi_set_variable");
 727
 728         if (!dry_run)
 729                 add_to_boot_order(bootvar); /* first, still not active */
 730         new_ent = malloc(sizeof(struct entry));
 731         if (new_ent == NULL)
 732                 err(1, "malloc");
 733         memset(new_ent, 0, sizeof(struct entry));
 734         new_ent->name = bootvar;
 735         new_ent->guid = EFI_GLOBAL_GUID;
 736         LIST_INSERT_HEAD(&efivars, new_ent, entries);
 737         free(load_opt_buf);
 738         free(dp);
 739
 740         return 0;
 741 }
 742
 743
 744 static void
 745 print_loadopt_str(uint8_t *data, size_t datalen)
 746 {
 747         char *dev, *relpath, *abspath;
 748         uint32_t attr;
 749         uint16_t fplen;
 750         efi_char *descr;
 751         uint8_t *ep = data + datalen;
 752         uint8_t *walker = data;
 753         efidp dp, edp;
 754         char buf[1024];
 755         int len;
 756         int rv;
 757         int indent;
 758
 759         if (datalen < sizeof(attr) + sizeof(fplen) + sizeof(efi_char))
 760                 return;
 761         // First 4 bytes are attribute flags
 762         attr = le32dec(walker);
 763         walker += sizeof(attr);
 764         // Next two bytes are length of the file paths
 765         fplen = le16dec(walker);
 766         walker += sizeof(fplen);
 767         // Next we have a 0 terminated UCS2 string that we know to be aligned
 768         descr = (efi_char *)(intptr_t)(void *)walker;
 769         len = ucs2len(descr); // XXX need to sanity check that len < (datalen - (ep - walker) / 2)
 770         walker += (len + 1) * sizeof(efi_char);
 771         if (walker > ep)
 772                 return;
 773         // Now we have fplen bytes worth of file path stuff
 774         dp = (efidp)walker;
 775         walker += fplen;
 776         if (walker > ep)
 777                 return;
 778         edp = (efidp)walker;
 779         /*
 780          * Everything left is the binary option args
 781          * opt = walker;
 782          * optlen = ep - walker;
 783          */
 784         indent = 1;
 785         while (dp < edp) {
 786                 if (efidp_size(dp) == 0)
 787                         break;
 788                 efidp_format_device_path(buf, sizeof(buf), dp,
 789                     (intptr_t)(void *)edp - (intptr_t)(void *)dp);
 790                 printf("%*s%s\n", indent, "", buf);
 791                 indent = 10 + len + 1;
 792                 rv = efivar_device_path_to_unix_path(dp, &dev, &relpath, &abspath);
 793                 if (rv == 0) {
 794                         printf("%*s%s:%s %s\n", indent + 4, "", dev, relpath, abspath);
 795                         free(dev);
 796                         free(relpath);
 797                         free(abspath);
 798                 }
 799                 dp = (efidp)((char *)dp + efidp_size(dp));
 800         }
 801 }
 802
 803 static char *
 804 get_descr(uint8_t *data)
 805 {
 806         uint8_t *pos = data;
 807         efi_char *desc;
 808         int  len;
 809         char *buf;
 810         int i = 0;
 811
 812         pos += sizeof(uint32_t) + sizeof(uint16_t);
 813         desc = (efi_char*)(intptr_t)(void *)pos;
 814         len = ucs2len(desc);
 815         buf = malloc(len + 1);
 816         memset(buf, 0, len + 1);
 817         while (desc[i]) {
 818                 buf[i] = desc[i];
 819                 i++;
 820         }
 821         return (char*)buf;
 822 }
 823
 824
 825 static bool
 826 print_boot_var(const char *name, bool verbose, bool curboot)
 827 {
 828         size_t size;
 829         uint32_t load_attrs;
 830         uint8_t *data;
 831         int ret;
 832         char *d;
 833
 834         ret = efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, NULL);
 835         if (ret < 0)
 836                 return false;
 837         load_attrs = le32dec(data);
 838         d = get_descr(data);
 839         printf("%c%s%c %s", curboot ? '+' : ' ', name,
 840             ((load_attrs & LOAD_OPTION_ACTIVE) ? '*': ' '), d);
 841         free(d);
 842         if (verbose)
 843                 print_loadopt_str(data, size);
 844         else
 845                 printf("\n");
 846         return true;
 847 }
 848
 849
 850 static bool
 851 os_indication_supported(uint64_t indication)
 852 {
 853         uint8_t *data;
 854         size_t size;
 855         uint32_t attrs;
 856         int ret;
 857
 858         ret = efi_get_variable(EFI_GLOBAL_GUID, "OsIndicationsSupported", &data,
 859             &size, &attrs);
 860         if (ret < 0)
 861                 return false;
 862         return (le64dec(data) & indication) == indication;
 863 }
 864
 865 static uint64_t
 866 os_indications(void)
 867 {
 868         uint8_t *data;
 869         size_t size;
 870         uint32_t attrs;
 871         int ret;
 872
 873         ret = efi_get_variable(EFI_GLOBAL_GUID, "OsIndications", &data, &size,
 874             &attrs);
 875         if (ret < 0)
 876                 return 0;
 877         return le64dec(data);
 878 }
 879
 880 static int
 881 os_indications_set(uint64_t mask, uint64_t val)
 882 {
 883         uint8_t new[sizeof(uint64_t)];
 884
 885         le64enc(&new, (os_indications() & ~mask) | (val & mask));
 886         return set_bootvar("OsIndications", new, sizeof(new));
 887 }
 888
 889 /* Cmd epilogue, or just the default with no args.
 890  * The order is [bootnext] bootcurrent, timeout, order, and the bootvars [-v]
 891  */
 892 static int
 893 print_boot_vars(bool verbose)
 894 {
 895         /*
 896          * just read and print the current values
 897          * as a command epilogue
 898          */
 899         struct entry *v;
 900         uint8_t *data;
 901         size_t size;
 902         uint32_t attrs;
 903         int ret, bolen;
 904         uint16_t *boot_order = NULL, current;
 905         bool boot_to_fw_ui;
 906
 907         if (os_indication_supported(EFI_OS_INDICATIONS_BOOT_TO_FW_UI)) {
 908                 boot_to_fw_ui =
 909                     (os_indications() & EFI_OS_INDICATIONS_BOOT_TO_FW_UI) != 0;
 910                 printf("Boot to FW : %s\n", boot_to_fw_ui != 0 ?
 911                     "true" : "false");
 912         }
 913
 914         ret = efi_get_variable(EFI_GLOBAL_GUID, "BootNext", &data, &size, &attrs);
 915         if (ret > 0) {
 916                 printf("BootNext : %04x\n", le16dec(data));
 917         }
 918
 919         ret = efi_get_variable(EFI_GLOBAL_GUID, "BootCurrent", &data, &size,&attrs);
 920         current = le16dec(data);
 921         printf("BootCurrent: %04x\n", current);
 922
 923         ret = efi_get_variable(EFI_GLOBAL_GUID, "Timeout", &data, &size, &attrs);
 924         if (ret > 0) {
 925                 printf("Timeout    : %d seconds\n", le16dec(data));
 926         }
 927
 928         if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) > 0) {
 929                 if (size % 2 == 1)
 930                         warn("Bad BootOrder variable: odd length %d", (int)size);
 931                 boot_order = malloc(size);
 932                 bolen = size / 2;
 933                 printf("BootOrder  : ");
 934                 for (size_t i = 0; i < size; i += 2) {
 935                         boot_order[i / 2] = le16dec(data + i);
 936                         printf("%04X%s", boot_order[i / 2], i == size - 2 ? "\n" : ", ");
 937                 }
 938         }
 939
 940         if (boot_order == NULL) {
 941                 /*
 942                  * now we want to fetch 'em all fresh again
 943                  * which possibly includes a newly created bootvar
 944                  */
 945                 LIST_FOREACH(v, &efivars, entries) {
 946                         print_boot_var(v->name, verbose, v->idx == current);
 947                 }
 948         } else {
 949                 LIST_FOREACH(v, &efivars, entries) {
 950                         v->flags = 0;
 951                 }
 952                 for (int i = 0; i < bolen; i++) {
 953                         char buffer[10];
 954
 955                         snprintf(buffer, sizeof(buffer), "Boot%04X", boot_order[i]);
 956                         if (!print_boot_var(buffer, verbose, boot_order[i] == current))
 957                                 printf("%s: MISSING!\n", buffer);
 958                         LIST_FOREACH(v, &efivars, entries) {
 959                                 if (v->idx == boot_order[i]) {
 960                                         v->flags |= SEEN;
 961                                         break;
 962                                 }
 963                         }
 964                 }
 965                 if (verbose) {
 966                         printf("\n\nUnreferenced Variables:\n");
 967                         LIST_FOREACH(v, &efivars, entries) {
 968                                 if (v->flags == 0)
 969                                         print_boot_var(v->name, verbose, v->idx == current);
 970                         }
 971                 }
 972         }
 973         return 0;
 974 }
 975
 976 static void
 977 delete_timeout(void)
 978 {
 979
 980         efi_del_variable(EFI_GLOBAL_GUID,"Timeout");
 981 }
 982
 983 static void
 984 handle_timeout(int to)
 985 {
 986         uint16_t timeout;
 987
 988         le16enc(&timeout, to);
 989         if (set_bootvar("Timeout", (uint8_t *)&timeout, sizeof(timeout)) < 0)
 990                 errx(1, "Can't set Timeout for booting.");
 991 }
 992
 993 static void
 994 report_esp_device(bool do_dp, bool do_unix)
 995 {
 996         uint8_t *data;
 997         size_t size, len;
 998         uint32_t attrs;
 999         int ret;
1000         uint16_t current, fplen;
1001         char *name, *dev, *relpath, *abspath;
1002         uint8_t *walker, *ep;
1003         efi_char *descr;
1004         efidp dp;
1005         char buf[PATH_MAX];
1006
1007         if (do_dp && do_unix)
1008                 errx(1, "Can't report both UEFI device-path and Unix path together");
1009
1010         ret = efi_get_variable(EFI_GLOBAL_GUID, "BootCurrent", &data, &size,&attrs);
1011         if (ret < 0)
1012                 err(1, "Can't get BootCurrent");
1013         current = le16dec(data);
1014         if (asprintf(&name, "Boot%04X", current) < 0)
1015                 err(1, "Can't format boot var\n");
1016         if (efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, NULL) < 0)
1017                 err(1, "Can't retrieve EFI var %s", name);
1018         // First 4 bytes are attribute flags
1019         walker = data;
1020         ep = walker + size;
1021         walker += sizeof(uint32_t);
1022         // Next two bytes are length of the file paths
1023         fplen = le16dec(walker);
1024         walker += sizeof(fplen);
1025         // Next we have a 0 terminated UCS2 string that we know to be aligned
1026         descr = (efi_char *)(intptr_t)(void *)walker;
1027         len = ucs2len(descr); // XXX need to sanity check that len < (datalen - (ep - walker) / 2)
1028         walker += (len + 1) * sizeof(efi_char);
1029         if (walker > ep)
1030                 errx(1, "malformed boot variable %s", name);
1031         // Now we have fplen bytes worth of file path stuff
1032         dp = (efidp)walker;
1033         walker += fplen;
1034         if (walker > ep)
1035                 errx(1, "malformed boot variable %s", name);
1036         if (do_dp) {
1037                 efidp_format_device_path_node(buf, sizeof(buf), dp);
1038                 printf("%s\n", buf);
1039                 exit(0);
1040         }
1041         if (efivar_device_path_to_unix_path(dp, &dev, &relpath, &abspath) != 0)
1042                 errx(1, "Can't convert to unix path");
1043         if (do_unix) {
1044                 if (abspath == NULL)
1045                         errx(1, "Can't find where %s:%s is mounted",
1046                             dev, relpath);
1047                 abspath[strlen(abspath) - strlen(relpath) - 1] = '\0';
1048                 printf("%s\n", abspath);
1049         } else {
1050                 printf("/dev/%s\n", dev);
1051         }
1052         free(dev);
1053         free(relpath);
1054         free(abspath);
1055         exit(0);
1056 }
1057
1058 static void
1059 set_boot_to_fw_ui(bool to_fw)
1060 {
1061         int ret;
1062
1063         if (!os_indication_supported(EFI_OS_INDICATIONS_BOOT_TO_FW_UI)) {
1064                 if (to_fw)
1065                         errx(1, "boot to fw ui not supported");
1066                 else
1067                         return;
1068         }
1069         ret = os_indications_set(EFI_OS_INDICATIONS_BOOT_TO_FW_UI,
1070             to_fw ? ~0 : 0);
1071         if (ret < 0)
1072                 errx(1, "failed to set boot to fw ui");
1073 }
1074
1075 static void
1076 find_efi_device(const char *path)
1077 {
1078         efidp dp = NULL;
1079         size_t len;
1080         int ret;
1081         char buf[1024];
1082
1083         ret = efivar_unix_path_to_device_path(path, &dp);
1084         if (ret != 0)
1085                 errc(1, ret,
1086                     "Cannot translate path '%s' to UEFI", path);
1087         len = efidp_size(dp);
1088         if (len > MAX_DP_LEN)
1089                 errx(1, "Resulting device path too long.");
1090         efidp_format_device_path(buf, sizeof(buf), dp, len);
1091         printf("%s -> %s\n", path, buf);
1092         exit (0);
1093 }
1094
1095 int
1096 main(int argc, char *argv[])
1097 {
1098
1099         memset(&opts, 0, sizeof (bmgr_opts_t));
1100         parse_args(argc, argv);
1101
1102         /*
1103          * find_dev can operate without any efi variables
1104          */
1105         if (!efi_variables_supported() && !opts.find_dev) {
1106                 if (errno == EACCES && geteuid() != 0)
1107                         errx(1, "must be run as root");
1108                 errx(1, "efi variables not supported on this system. kldload efirt?");
1109         }
1110
1111         read_vars();
1112
1113         if (opts.create)
1114                 /*
1115                  * side effect, adds to boot order, but not yet active.
1116                  */
1117                 make_boot_var(opts.label ? opts.label : "",
1118                     opts.loader, opts.kernel, opts.env, opts.dry_run,
1119                     opts.has_bootnum ? opts.bootnum : -1, opts.set_active);
1120         else if (opts.set_active || opts.set_inactive )
1121                 handle_activity(opts.bootnum, opts.set_active);
1122         else if (opts.order != NULL)
1123                 set_boot_order(opts.order); /* create a new bootorder with opts.order */
1124         else if (opts.set_bootnext)
1125                 handle_bootnext(opts.bootnum);
1126         else if (opts.delete_bootnext)
1127                 del_bootnext();
1128         else if (opts.delete)
1129                 delete_bootvar(opts.bootnum);
1130         else if (opts.del_timeout)
1131                 delete_timeout();
1132         else if (opts.set_timeout)
1133                 handle_timeout(opts.timeout);
1134         else if (opts.esp_device)
1135                 report_esp_device(opts.device_path, opts.unix_path);
1136         else if (opts.fw_ui)
1137                 set_boot_to_fw_ui(true);
1138         else if (opts.no_fw_ui)
1139                 set_boot_to_fw_ui(false);
1140         else if (opts.find_dev)
1141                 find_efi_device(opts.dev);
1142
1143         print_boot_vars(opts.verbose);
1144 }