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 __FBSDID("$FreeBSD$"); 28 29 #include <sys/stat.h> 30 #include <sys/vtoc.h> 31 #include <sys/param.h> 32 #include <assert.h> 33 #include <ctype.h> 34 #include <err.h> 35 #include <errno.h> 36 #include <fcntl.h> 37 #include <libgeom.h> 38 #include <paths.h> 39 #include <signal.h> 40 #include <stdint.h> 41 #include <stdio.h> 42 #include <stdlib.h> 43 #include <getopt.h> 44 #include <limits.h> 45 #include <inttypes.h> 46 #include <stdbool.h> 47 #include <string.h> 48 #include <strings.h> 49 #include <unistd.h> 50 #include <libgeom.h> 51 #include <geom/geom.h> 52 #include <geom/geom_ctl.h> 53 #include <geom/geom_int.h> 54 55 #include <efivar.h> 56 #include <efiutil.h> 57 #include <efichar.h> 58 #include <efivar-dp.h> 59 60 #ifndef LOAD_OPTION_ACTIVE 61 #define LOAD_OPTION_ACTIVE 0x00000001 62 #endif 63 64 #ifndef LOAD_OPTION_CATEGORY_BOOT 65 #define LOAD_OPTION_CATEGORY_BOOT 0x00000000 66 #endif 67 68 #define BAD_LENGTH ((size_t)-1) 69 70 typedef struct _bmgr_opts { 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 has_bootnum; 85 bool once; 86 int cp_src; 87 bool set_active; 88 bool set_bootnext; 89 bool set_inactive; 90 bool set_timeout; 91 int timeout; 92 bool verbose; 93 } bmgr_opts_t; 94 95 static struct option lopts[] = { 96 {"activate", no_argument, NULL, 'a'}, 97 {"bootnext", no_argument, NULL, 'n'}, /* set bootnext */ 98 {"bootnum", required_argument, NULL, 'b'}, 99 {"bootorder", required_argument, NULL, 'o'}, /* set order */ 100 {"copy", required_argument, NULL, 'C'}, /* Copy boot method */ 101 {"create", no_argument, NULL, 'c'}, 102 {"deactivate", no_argument, NULL, 'A'}, 103 {"del-timout", no_argument, NULL, 'T'}, 104 {"delete", no_argument, NULL, 'B'}, 105 {"delete-bootnext", no_argument, NULL, 'N'}, 106 {"dry-run", no_argument, NULL, 'D'}, 107 {"env", required_argument, NULL, 'e'}, 108 {"help", no_argument, NULL, 'h'}, 109 {"kernel", required_argument, NULL, 'k'}, 110 {"label", required_argument, NULL, 'L'}, 111 {"loader", required_argument, NULL, 'l'}, 112 {"once", no_argument, NULL, 'O'}, 113 {"set-timeout", required_argument, NULL, 't'}, 114 {"verbose", no_argument, NULL, 'v'}, 115 { NULL, 0, NULL, 0} 116 }; 117 118 /* global efibootmgr opts */ 119 static bmgr_opts_t opts; 120 121 static LIST_HEAD(efivars_head, entry) efivars = 122 LIST_HEAD_INITIALIZER(efivars); 123 124 struct entry { 125 efi_guid_t guid; 126 uint32_t attrs; 127 uint8_t *data; 128 size_t size; 129 char *name; 130 char *label; 131 int idx; 132 int flags; 133 #define SEEN 1 134 135 LIST_ENTRY(entry) entries; 136 }; 137 138 #define MAX_DP_LEN 4096 139 #define MAX_LOADOPT_LEN 8192 140 141 142 static char * 143 mangle_loader(char *loader) 144 { 145 char *c; 146 147 for (c = loader; *c; c++) 148 if (*c == '/') 149 *c = '\\'; 150 151 return loader; 152 } 153 154 155 #define COMMON_ATTRS EFI_VARIABLE_NON_VOLATILE | \ 156 EFI_VARIABLE_BOOTSERVICE_ACCESS | \ 157 EFI_VARIABLE_RUNTIME_ACCESS 158 159 /* 160 * We use global guid, and common var attrs and 161 * find it better to just delete and re-create a var. 162 */ 163 static int 164 set_bootvar(const char *name, uint8_t *data, size_t size) 165 { 166 167 return efi_set_variable(EFI_GLOBAL_GUID, name, data, size, 168 COMMON_ATTRS); 169 } 170 171 172 #define USAGE \ 173 " [-aAnB -b bootnum] [-N] [-t timeout] [-T] [-o bootorder] [-O] [--verbose] [--help]\n\ 174 [-c -l loader [-k kernel] [-L label] [--dry-run] [-b bootnum]]" 175 176 #define CREATE_USAGE \ 177 " efibootmgr -c -l loader [-k kernel] [-L label] [--dry-run] [-b bootnum] [-a]" 178 #define ORDER_USAGE \ 179 " efibootmgr -o bootvarnum1,bootvarnum2,..." 180 #define TIMEOUT_USAGE \ 181 " efibootmgr -t seconds" 182 #define DELETE_USAGE \ 183 " efibootmgr -B -b bootnum" 184 #define ACTIVE_USAGE \ 185 " efibootmgr [-a | -A] -b bootnum" 186 #define BOOTNEXT_USAGE \ 187 " efibootmgr [-n | -N] -b bootnum" 188 189 static void 190 parse_args(int argc, char *argv[]) 191 { 192 int ch; 193 194 while ((ch = getopt_long(argc, argv, "AaBb:C:cDe:hk:L:l:NnOo:Tt:v", 195 lopts, NULL)) != -1) { 196 switch (ch) { 197 case 'A': 198 opts.set_inactive = true; 199 break; 200 case 'a': 201 opts.set_active = true; 202 break; 203 case 'b': 204 opts.has_bootnum = true; 205 opts.bootnum = strtoul(optarg, NULL, 16); 206 break; 207 case 'B': 208 opts.delete = true; 209 break; 210 case 'C': 211 opts.copy = true; 212 opts.cp_src = strtoul(optarg, NULL, 16); 213 case 'c': 214 opts.create = true; 215 break; 216 case 'D': /* should be remove dups XXX */ 217 opts.dry_run = true; 218 break; 219 case 'e': 220 free(opts.env); 221 opts.env = strdup(optarg); 222 break; 223 case 'h': 224 default: 225 errx(1, "%s", USAGE); 226 break; 227 case 'k': 228 free(opts.kernel); 229 opts.kernel = strdup(optarg); 230 break; 231 case 'L': 232 free(opts.label); 233 opts.label = strdup(optarg); 234 break; 235 case 'l': 236 free(opts.loader); 237 opts.loader = strdup(optarg); 238 opts.loader = mangle_loader(opts.loader); 239 break; 240 case 'N': 241 opts.delete_bootnext = true; 242 break; 243 case 'n': 244 opts.set_bootnext = true; 245 break; 246 case 'O': 247 opts.once = true; 248 break; 249 case 'o': 250 free(opts.order); 251 opts.order = strdup(optarg); 252 break; 253 case 'T': 254 opts.del_timeout = true; 255 break; 256 case 't': 257 opts.set_timeout = true; 258 opts.timeout = strtoul(optarg, NULL, 10); 259 break; 260 case 'v': 261 opts.verbose = true; 262 break; 263 } 264 } 265 if (opts.create) { 266 if (!opts.loader) 267 errx(1, "%s",CREATE_USAGE); 268 return; 269 } 270 271 if (opts.order && !(opts.order)) 272 errx(1, "%s", ORDER_USAGE); 273 274 if ((opts.set_inactive || opts.set_active) && !opts.has_bootnum) 275 errx(1, "%s", ACTIVE_USAGE); 276 277 if (opts.delete && !opts.has_bootnum) 278 errx(1, "%s", DELETE_USAGE); 279 280 if (opts.set_bootnext && !opts.has_bootnum) 281 errx(1, "%s", BOOTNEXT_USAGE); 282 } 283 284 285 static void 286 read_vars(void) 287 { 288 289 efi_guid_t *guid; 290 char *next_name = NULL; 291 int ret = 0; 292 293 struct entry *nent; 294 295 LIST_INIT(&efivars); 296 while ((ret = efi_get_next_variable_name(&guid, &next_name)) > 0) { 297 /* 298 * Only pay attention to EFI:BootXXXX variables to get the list. 299 */ 300 if (efi_guid_cmp(guid, &EFI_GLOBAL_GUID) != 0 || 301 strlen(next_name) != 8 || 302 strncmp(next_name, "Boot", 4) != 0 || 303 !isxdigit(next_name[4]) || 304 !isxdigit(next_name[5]) || 305 !isxdigit(next_name[6]) || 306 !isxdigit(next_name[7])) 307 continue; 308 nent = malloc(sizeof(struct entry)); 309 nent->name = strdup(next_name); 310 311 ret = efi_get_variable(*guid, next_name, &nent->data, 312 &nent->size, &nent->attrs); 313 if (ret < 0) 314 err(1, "efi_get_variable"); 315 nent->guid = *guid; 316 nent->idx = strtoul(&next_name[4], NULL, 16); 317 LIST_INSERT_HEAD(&efivars, nent, entries); 318 } 319 } 320 321 322 static void 323 set_boot_order(char *order) 324 { 325 uint16_t *new_data; 326 size_t size; 327 char *next, *cp; 328 int cnt; 329 int i; 330 331 cp = order; 332 cnt = 1; 333 while (*cp) { 334 if (*cp++ == ',') 335 cnt++; 336 } 337 size = sizeof(uint16_t) * cnt; 338 new_data = malloc(size); 339 340 i = 0; 341 cp = strdup(order); 342 while ((next = strsep(&cp, ",")) != NULL) { 343 new_data[i] = strtoul(next, NULL, 16); 344 if (new_data[i] == 0 && errno == EINVAL) { 345 warnx("can't parse %s as a numb", next); 346 errx(1, "%s", ORDER_USAGE); 347 } 348 i++; 349 } 350 free(cp); 351 if (set_bootvar("BootOrder", (uint8_t*)new_data, size) < 0) 352 err(1, "Unabke to set BootOrder to %s", order); 353 free(new_data); 354 } 355 356 static void 357 handle_activity(int bootnum, bool active) 358 { 359 uint32_t attrs, load_attrs; 360 uint8_t *data; 361 size_t size; 362 char *name; 363 364 asprintf(&name, "%s%04X", "Boot", bootnum); 365 if (name == NULL) 366 err(1, "asprintf"); 367 if (efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, &attrs) < 0) 368 err(1, "No such bootvar %s\n", name); 369 370 load_attrs = le32dec(data); 371 372 if (active) 373 load_attrs |= LOAD_OPTION_ACTIVE; 374 else 375 load_attrs &= ~LOAD_OPTION_ACTIVE; 376 377 le32enc(data, load_attrs); 378 379 if (set_bootvar(name, data, size) < 0) 380 err(1, "handle activity efi_set_variable"); 381 } 382 383 384 /* 385 * add boot var to boot order. 386 * called by create boot var. There is no option 387 * to add one independent of create. 388 * 389 * Note: we currently don't support where it goes 390 * so it goes on the front, inactive. 391 * use -o 2,3,7 etc to affect order, -a to activate. 392 */ 393 static void 394 add_to_boot_order(char *bootvar) 395 { 396 size_t size; 397 uint32_t attrs; 398 uint16_t val; 399 uint8_t *data, *new; 400 401 val = strtoul(&bootvar[4], NULL, 16); 402 403 if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) < 0) { 404 if (errno == ENOENT) { /* create it and set this bootvar to active */ 405 size = 0; 406 data = NULL; 407 } else 408 err(1, "efi_get_variable BootOrder"); 409 } 410 411 /* 412 * We have BootOrder with the current order 413 * so grow the array by one, add the value 414 * and write the new variable value. 415 */ 416 size += sizeof(uint16_t); 417 new = malloc(size); 418 if (!new) 419 err(1, "malloc"); 420 421 le16enc(new, val); 422 if (size > sizeof(uint16_t)) 423 memcpy(new + sizeof(uint16_t), data, size - sizeof(uint16_t)); 424 425 if (set_bootvar("BootOrder", new, size) < 0) 426 err(1, "set_bootvar"); 427 free(new); 428 } 429 430 431 static void 432 remove_from_order(uint16_t bootnum) 433 { 434 uint32_t attrs; 435 size_t size, i, j; 436 uint8_t *new, *data; 437 438 if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) < 0) 439 return; 440 441 new = malloc(size); 442 if (new == NULL) 443 err(1, "malloc"); 444 445 for (j = i = 0; i < size; i += sizeof(uint16_t)) { 446 if (le16dec(data + i) == bootnum) 447 continue; 448 memcpy(new + j, data + i, sizeof(uint16_t)); 449 j += sizeof(uint16_t); 450 } 451 if (i == j) 452 warnx("Boot variable %04x not in BootOrder", bootnum); 453 else if (set_bootvar("BootOrder", new, j) < 0) 454 err(1, "Unable to update BootOrder with new value"); 455 free(new); 456 } 457 458 459 static void 460 delete_bootvar(int bootnum) 461 { 462 char *name; 463 int defer = 0; 464 465 /* 466 * Try to delete the boot variable and remocve it 467 * from the boot order. We always do both actions 468 * to make it easy to clean up from oopses. 469 */ 470 if (bootnum < 0 || bootnum > 0xffff) 471 errx(1, "Bad boot variable %#x", bootnum); 472 asprintf(&name, "%s%04X", "Boot", bootnum); 473 if (name == NULL) 474 err(1, "asprintf"); 475 printf("Removing boot variable '%s'\n", name); 476 if (efi_del_variable(EFI_GLOBAL_GUID, name) < 0) { 477 defer = 1; 478 warn("cannot delete variable %s", name); 479 } 480 printf("Removing 0x%x from BootOrder\n", bootnum); 481 remove_from_order(bootnum); 482 free(name); 483 if (defer) 484 exit(defer); 485 } 486 487 488 static void 489 del_bootnext(void) 490 { 491 492 if (efi_del_variable(EFI_GLOBAL_GUID, "BootNext") < 0) 493 err(1, "efi_del_variable"); 494 } 495 496 static void 497 handle_bootnext(uint16_t bootnum) 498 { 499 uint16_t num; 500 501 le16enc(&num, bootnum); 502 if (set_bootvar("BootNext", (uint8_t*)&bootnum, sizeof(uint16_t)) < 0) 503 err(1, "set_bootvar"); 504 } 505 506 507 static int 508 compare(const void *a, const void *b) 509 { 510 uint16_t c; 511 uint16_t d; 512 513 memcpy(&c, a, sizeof(uint16_t)); 514 memcpy(&d, b, sizeof(uint16_t)); 515 516 if (c < d) 517 return -1; 518 if (c == d) 519 return 0; 520 return 1; 521 } 522 523 static char * 524 make_next_boot_var_name(void) 525 { 526 struct entry *v; 527 uint16_t *vals, next_free = 0; 528 char *name; 529 int cnt = 0; 530 int i; 531 532 LIST_FOREACH(v, &efivars, entries) { 533 cnt++; 534 } 535 536 vals = malloc(sizeof(uint16_t) * cnt); 537 if (!vals) 538 return NULL; 539 540 i = 0; 541 LIST_FOREACH(v, &efivars, entries) { 542 vals[i++] = v->idx; 543 } 544 qsort(vals, cnt, sizeof(uint16_t), compare); 545 /* if the hole is at the beginning, just return zero */ 546 if (vals[0] > 0) { 547 next_free = 0; 548 } else { 549 /* now just run the list looking for the first hole */ 550 for (i = 0; i < cnt - 1 && next_free == 0; i++) 551 if (vals[i] + 1 != vals[i + 1]) 552 next_free = vals[i] + 1; 553 if (next_free == 0) 554 next_free = vals[cnt - 1] + 1; 555 /* In theory we could have used all 65k slots -- what to do? */ 556 } 557 free(vals); 558 559 asprintf(&name, "%s%04X", "Boot", next_free); 560 if (name == NULL) 561 err(1, "asprintf"); 562 return name; 563 } 564 565 static char * 566 make_boot_var_name(uint16_t bootnum) 567 { 568 struct entry *v; 569 char *name; 570 571 LIST_FOREACH(v, &efivars, entries) { 572 if (v->idx == bootnum) 573 return NULL; 574 } 575 576 asprintf(&name, "%s%04X", "Boot", bootnum); 577 if (name == NULL) 578 err(1, "asprintf"); 579 return name; 580 } 581 582 static size_t 583 create_loadopt(uint8_t *buf, size_t bufmax, uint32_t attributes, efidp dp, size_t dp_size, 584 const char *description, const uint8_t *optional_data, size_t optional_data_size) 585 { 586 efi_char *bbuf = NULL; 587 uint8_t *pos = buf; 588 size_t desc_len = 0; 589 size_t len; 590 591 if (optional_data == NULL && optional_data_size != 0) 592 return BAD_LENGTH; 593 if (dp == NULL && dp_size != 0) 594 return BAD_LENGTH; 595 596 /* 597 * Compute the length to make sure the passed in buffer is long enough. 598 */ 599 utf8_to_ucs2(description, &bbuf, &desc_len); 600 len = sizeof(uint32_t) + sizeof(uint16_t) + desc_len + dp_size + optional_data_size; 601 if (len > bufmax) { 602 free(bbuf); 603 return BAD_LENGTH; 604 } 605 606 le32enc(pos, attributes); 607 pos += sizeof (attributes); 608 609 le16enc(pos, dp_size); 610 pos += sizeof (uint16_t); 611 612 memcpy(pos, bbuf, desc_len); /* NB:desc_len includes strailing NUL */ 613 pos += desc_len; 614 free(bbuf); 615 616 memcpy(pos, dp, dp_size); 617 pos += dp_size; 618 619 if (optional_data && optional_data_size > 0) { 620 memcpy(pos, optional_data, optional_data_size); 621 pos += optional_data_size; 622 } 623 624 return pos - buf; 625 } 626 627 628 static int 629 make_boot_var(const char *label, const char *loader, const char *kernel, const char *env, bool dry_run, 630 int bootnum, bool activate) 631 { 632 struct entry *new_ent; 633 uint32_t load_attrs = 0; 634 uint8_t *load_opt_buf; 635 size_t lopt_size, llen, klen; 636 efidp dp, loaderdp, kerneldp; 637 char *bootvar = NULL; 638 int ret; 639 640 assert(label != NULL); 641 642 if (bootnum == -1) 643 bootvar = make_next_boot_var_name(); 644 else 645 bootvar = make_boot_var_name((uint16_t)bootnum); 646 if (bootvar == NULL) 647 err(1, "bootvar creation"); 648 if (loader == NULL) 649 errx(1, "Must specify boot loader"); 650 if (efivar_unix_path_to_device_path(loader, &loaderdp) != 0) 651 err(1, "Cannot translate unix loader path '%s' to UEFI", loader); 652 if (kernel != NULL) { 653 if (efivar_unix_path_to_device_path(kernel, &kerneldp) != 0) 654 err(1, "Cannot translate unix kernel path '%s' to UEFI", kernel); 655 } else { 656 kerneldp = NULL; 657 } 658 llen = efidp_size(loaderdp); 659 if (llen > MAX_DP_LEN) 660 errx(1, "Loader path too long."); 661 klen = efidp_size(kerneldp); 662 if (klen > MAX_DP_LEN) 663 errx(1, "Kernel path too long."); 664 dp = malloc(llen + klen); 665 if (dp == NULL) 666 errx(1, "Can't allocate memory for new device paths"); 667 memcpy(dp, loaderdp, llen); 668 if (kerneldp != NULL) 669 memcpy((char *)dp + llen, kerneldp, klen); 670 671 /* don't make the new bootvar active by default, use the -a option later */ 672 load_attrs = LOAD_OPTION_CATEGORY_BOOT; 673 if (activate) 674 load_attrs |= LOAD_OPTION_ACTIVE; 675 load_opt_buf = malloc(MAX_LOADOPT_LEN); 676 if (load_opt_buf == NULL) 677 err(1, "malloc"); 678 679 lopt_size = create_loadopt(load_opt_buf, MAX_LOADOPT_LEN, load_attrs, 680 dp, llen + klen, label, env, env ? strlen(env) + 1 : 0); 681 if (lopt_size == BAD_LENGTH) 682 errx(1, "Can't create loadopt"); 683 684 ret = 0; 685 if (!dry_run) { 686 ret = efi_set_variable(EFI_GLOBAL_GUID, bootvar, 687 (uint8_t*)load_opt_buf, lopt_size, COMMON_ATTRS); 688 } 689 690 if (ret) 691 err(1, "efi_set_variable"); 692 693 if (!dry_run) 694 add_to_boot_order(bootvar); /* first, still not active */ 695 new_ent = malloc(sizeof(struct entry)); 696 if (new_ent == NULL) 697 err(1, "malloc"); 698 memset(new_ent, 0, sizeof(struct entry)); 699 new_ent->name = bootvar; 700 new_ent->guid = EFI_GLOBAL_GUID; 701 LIST_INSERT_HEAD(&efivars, new_ent, entries); 702 free(load_opt_buf); 703 free(dp); 704 705 return 0; 706 } 707 708 709 static void 710 print_loadopt_str(uint8_t *data, size_t datalen) 711 { 712 char *dev, *relpath, *abspath; 713 uint32_t attr; 714 uint16_t fplen; 715 efi_char *descr; 716 uint8_t *ep = data + datalen; 717 uint8_t *walker = data; 718 efidp dp, edp; 719 char buf[1024]; 720 int len; 721 int rv; 722 int indent; 723 724 if (datalen < sizeof(attr) + sizeof(fplen) + sizeof(efi_char)) 725 return; 726 // First 4 bytes are attribute flags 727 attr = le32dec(walker); 728 walker += sizeof(attr); 729 // Next two bytes are length of the file paths 730 fplen = le16dec(walker); 731 walker += sizeof(fplen); 732 // Next we have a 0 terminated UCS2 string that we know to be aligned 733 descr = (efi_char *)(intptr_t)(void *)walker; 734 len = ucs2len(descr); // XXX need to sanity check that len < (datalen - (ep - walker) / 2) 735 walker += (len + 1) * sizeof(efi_char); 736 if (walker > ep) 737 return; 738 // Now we have fplen bytes worth of file path stuff 739 dp = (efidp)walker; 740 walker += fplen; 741 if (walker > ep) 742 return; 743 edp = (efidp)walker; 744 /* 745 * Everything left is the binary option args 746 * opt = walker; 747 * optlen = ep - walker; 748 */ 749 indent = 1; 750 while (dp < edp) { 751 efidp_format_device_path(buf, sizeof(buf), dp, 752 (intptr_t)(void *)edp - (intptr_t)(void *)dp); 753 printf("%*s%s\n", indent, "", buf); 754 indent = 10 + len + 1; 755 rv = efivar_device_path_to_unix_path(dp, &dev, &relpath, &abspath); 756 if (rv == 0) { 757 printf("%*s%s:%s %s\n", indent + 4, "", dev, relpath, abspath); 758 free(dev); 759 free(relpath); 760 free(abspath); 761 } 762 dp = (efidp)((char *)dp + efidp_size(dp)); 763 } 764 } 765 766 static char * 767 get_descr(uint8_t *data) 768 { 769 uint8_t *pos = data; 770 efi_char *desc; 771 int len; 772 char *buf; 773 int i = 0; 774 775 pos += sizeof(uint32_t) + sizeof(uint16_t); 776 desc = (efi_char*)(intptr_t)(void *)pos; 777 len = ucs2len(desc); 778 buf = malloc(len + 1); 779 memset(buf, 0, len + 1); 780 while (desc[i]) { 781 buf[i] = desc[i]; 782 i++; 783 } 784 return (char*)buf; 785 } 786 787 788 static bool 789 print_boot_var(const char *name, bool verbose, bool curboot) 790 { 791 size_t size; 792 uint32_t load_attrs; 793 uint8_t *data; 794 int ret; 795 char *d; 796 797 ret = efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, NULL); 798 if (ret < 0) 799 return false; 800 load_attrs = le32dec(data); 801 d = get_descr(data); 802 printf("%c%s%c %s", curboot ? '+' : ' ', name, 803 ((load_attrs & LOAD_OPTION_ACTIVE) ? '*': ' '), d); 804 free(d); 805 if (verbose) 806 print_loadopt_str(data, size); 807 else 808 printf("\n"); 809 return true; 810 } 811 812 813 /* Cmd epilogue, or just the default with no args. 814 * The order is [bootnext] bootcurrent, timeout, order, and the bootvars [-v] 815 */ 816 static int 817 print_boot_vars(bool verbose) 818 { 819 /* 820 * just read and print the current values 821 * as a command epilogue 822 */ 823 struct entry *v; 824 uint8_t *data; 825 size_t size; 826 uint32_t attrs; 827 int ret, bolen; 828 uint16_t *boot_order = NULL, current; 829 830 ret = efi_get_variable(EFI_GLOBAL_GUID, "BootNext", &data, &size, &attrs); 831 if (ret > 0) { 832 printf("BootNext : %04x\n", le16dec(data)); 833 } 834 835 ret = efi_get_variable(EFI_GLOBAL_GUID, "BootCurrent", &data, &size,&attrs); 836 current = le16dec(data); 837 printf("BootCurrent: %04x\n", current); 838 839 ret = efi_get_variable(EFI_GLOBAL_GUID, "Timeout", &data, &size, &attrs); 840 if (ret > 0) { 841 printf("Timeout : %d seconds\n", le16dec(data)); 842 } 843 844 if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) > 0) { 845 if (size % 2 == 1) 846 warn("Bad BootOrder variable: odd length %d", (int)size); 847 boot_order = malloc(size); 848 bolen = size / 2; 849 printf("BootOrder : "); 850 for (size_t i = 0; i < size; i += 2) { 851 boot_order[i / 2] = le16dec(data + i); 852 printf("%04X%s", boot_order[i / 2], i == size - 2 ? "\n" : ", "); 853 } 854 } 855 856 if (boot_order == NULL) { 857 /* 858 * now we want to fetch 'em all fresh again 859 * which possibly includes a newly created bootvar 860 */ 861 LIST_FOREACH(v, &efivars, entries) { 862 print_boot_var(v->name, verbose, v->idx == current); 863 } 864 } else { 865 LIST_FOREACH(v, &efivars, entries) { 866 v->flags = 0; 867 } 868 for (int i = 0; i < bolen; i++) { 869 char buffer[10]; 870 871 snprintf(buffer, sizeof(buffer), "Boot%04X", boot_order[i]); 872 if (!print_boot_var(buffer, verbose, boot_order[i] == current)) 873 printf("%s: MISSING!\n", buffer); 874 LIST_FOREACH(v, &efivars, entries) { 875 if (v->idx == boot_order[i]) { 876 v->flags |= SEEN; 877 break; 878 } 879 } 880 } 881 if (verbose) { 882 printf("\n\nUnreferenced Variables:\n"); 883 LIST_FOREACH(v, &efivars, entries) { 884 if (v->flags == 0) 885 print_boot_var(v->name, verbose, v->idx == current); 886 } 887 } 888 } 889 return 0; 890 } 891 892 static void 893 delete_timeout(void) 894 { 895 896 efi_del_variable(EFI_GLOBAL_GUID,"Timeout"); 897 } 898 899 static void 900 handle_timeout(int to) 901 { 902 uint16_t timeout; 903 904 le16enc(&timeout, to); 905 if (set_bootvar("Timeout", (uint8_t *)&timeout, sizeof(timeout)) < 0) 906 errx(1, "Can't set Timeout for booting."); 907 } 908 909 int 910 main(int argc, char *argv[]) 911 { 912 913 if (!efi_variables_supported()) 914 errx(1, "efi variables not supported on this system. root? kldload efirt?"); 915 916 memset(&opts, 0, sizeof (bmgr_opts_t)); 917 parse_args(argc, argv); 918 read_vars(); 919 920 if (opts.create) 921 /* 922 * side effect, adds to boot order, but not yet active. 923 */ 924 make_boot_var(opts.label ? opts.label : "", 925 opts.loader, opts.kernel, opts.env, opts.dry_run, 926 opts.has_bootnum ? opts.bootnum : -1, opts.set_active); 927 else if (opts.set_active || opts.set_inactive ) 928 handle_activity(opts.bootnum, opts.set_active); 929 else if (opts.order != NULL) 930 set_boot_order(opts.order); /* create a new bootorder with opts.order */ 931 else if (opts.set_bootnext) 932 handle_bootnext(opts.bootnum); 933 else if (opts.delete_bootnext) 934 del_bootnext(); 935 else if (opts.delete) 936 delete_bootvar(opts.bootnum); 937 else if (opts.del_timeout) 938 delete_timeout(); 939 else if (opts.set_timeout) 940 handle_timeout(opts.timeout); 941 942 print_boot_vars(opts.verbose); 943 } 944