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