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) 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 load_opt_buf = malloc(MAX_LOADOPT_LEN); 669 if (load_opt_buf == NULL) 670 err(1, "malloc"); 671 672 lopt_size = create_loadopt(load_opt_buf, MAX_LOADOPT_LEN, load_attrs, 673 dp, llen + klen, label, env, env ? strlen(env) + 1 : 0); 674 if (lopt_size == BAD_LENGTH) 675 errx(1, "Can't crate loadopt"); 676 677 ret = 0; 678 if (!dry_run) { 679 ret = efi_set_variable(EFI_GLOBAL_GUID, bootvar, 680 (uint8_t*)load_opt_buf, lopt_size, COMMON_ATTRS); 681 } 682 683 if (ret) 684 err(1, "efi_set_variable"); 685 686 add_to_boot_order(bootvar); /* first, still not active */ 687 new_ent = malloc(sizeof(struct entry)); 688 if (new_ent == NULL) 689 err(1, "malloc"); 690 memset(new_ent, 0, sizeof(struct entry)); 691 new_ent->name = bootvar; 692 new_ent->guid = EFI_GLOBAL_GUID; 693 LIST_INSERT_HEAD(&efivars, new_ent, entries); 694 free(load_opt_buf); 695 free(dp); 696 697 return 0; 698 } 699 700 701 static void 702 print_loadopt_str(uint8_t *data, size_t datalen) 703 { 704 char *dev, *relpath, *abspath; 705 uint32_t attr; 706 uint16_t fplen; 707 efi_char *descr; 708 uint8_t *ep = data + datalen; 709 uint8_t *walker = data; 710 efidp dp, edp; 711 char buf[1024]; 712 int len; 713 int rv; 714 int indent; 715 716 if (datalen < sizeof(attr) + sizeof(fplen) + sizeof(efi_char)) 717 return; 718 // First 4 bytes are attribute flags 719 attr = le32dec(walker); 720 walker += sizeof(attr); 721 // Next two bytes are length of the file paths 722 fplen = le16dec(walker); 723 walker += sizeof(fplen); 724 // Next we have a 0 terminated UCS2 string that we know to be aligned 725 descr = (efi_char *)(intptr_t)(void *)walker; 726 len = ucs2len(descr); // XXX need to sanity check that len < (datalen - (ep - walker) / 2) 727 walker += (len + 1) * sizeof(efi_char); 728 if (walker > ep) 729 return; 730 // Now we have fplen bytes worth of file path stuff 731 dp = (efidp)walker; 732 walker += fplen; 733 if (walker > ep) 734 return; 735 edp = (efidp)walker; 736 /* 737 * Everything left is the binary option args 738 * opt = walker; 739 * optlen = ep - walker; 740 */ 741 indent = 1; 742 while (dp < edp) { 743 efidp_format_device_path(buf, sizeof(buf), dp, 744 (intptr_t)(void *)edp - (intptr_t)(void *)dp); 745 printf("%*s%s\n", indent, "", buf); 746 indent = 10 + len + 1; 747 rv = efivar_device_path_to_unix_path(dp, &dev, &relpath, &abspath); 748 if (rv == 0) { 749 printf("%*s%s:%s %s\n", indent + 4, "", dev, relpath, abspath); 750 free(dev); 751 free(relpath); 752 free(abspath); 753 } 754 dp = (efidp)((char *)dp + efidp_size(dp)); 755 } 756 } 757 758 static char * 759 get_descr(uint8_t *data) 760 { 761 uint8_t *pos = data; 762 efi_char *desc; 763 int len; 764 char *buf; 765 int i = 0; 766 767 pos += sizeof(uint32_t) + sizeof(uint16_t); 768 desc = (efi_char*)(intptr_t)(void *)pos; 769 len = ucs2len(desc); 770 buf = malloc(len + 1); 771 memset(buf, 0, len + 1); 772 while (desc[i]) { 773 buf[i] = desc[i]; 774 i++; 775 } 776 return (char*)buf; 777 } 778 779 780 static bool 781 print_boot_var(const char *name, bool verbose, bool curboot) 782 { 783 size_t size; 784 uint32_t load_attrs; 785 uint8_t *data; 786 int ret; 787 char *d; 788 789 ret = efi_get_variable(EFI_GLOBAL_GUID, name, &data, &size, NULL); 790 if (ret < 0) 791 return false; 792 load_attrs = le32dec(data); 793 d = get_descr(data); 794 printf("%c%s%c %s", curboot ? '+' : ' ', name, 795 ((load_attrs & LOAD_OPTION_ACTIVE) ? '*': ' '), d); 796 free(d); 797 if (verbose) 798 print_loadopt_str(data, size); 799 else 800 printf("\n"); 801 return true; 802 } 803 804 805 /* Cmd epilogue, or just the default with no args. 806 * The order is [bootnext] bootcurrent, timeout, order, and the bootvars [-v] 807 */ 808 static int 809 print_boot_vars(bool verbose) 810 { 811 /* 812 * just read and print the current values 813 * as a command epilogue 814 */ 815 struct entry *v; 816 uint8_t *data; 817 size_t size; 818 uint32_t attrs; 819 int ret, bolen; 820 uint16_t *boot_order = NULL, current; 821 822 ret = efi_get_variable(EFI_GLOBAL_GUID, "BootNext", &data, &size, &attrs); 823 if (ret > 0) { 824 printf("BootNext : %04x\n", le16dec(data)); 825 } 826 827 ret = efi_get_variable(EFI_GLOBAL_GUID, "BootCurrent", &data, &size,&attrs); 828 current = le16dec(data); 829 printf("BootCurrent: %04x\n", current); 830 831 ret = efi_get_variable(EFI_GLOBAL_GUID, "Timeout", &data, &size, &attrs); 832 if (ret > 0) { 833 printf("Timeout : %d seconds\n", le16dec(data)); 834 } 835 836 if (efi_get_variable(EFI_GLOBAL_GUID, "BootOrder", &data, &size, &attrs) > 0) { 837 if (size % 2 == 1) 838 warn("Bad BootOrder variable: odd length %d", (int)size); 839 boot_order = malloc(size); 840 bolen = size / 2; 841 printf("BootOrder : "); 842 for (size_t i = 0; i < size; i += 2) { 843 boot_order[i / 2] = le16dec(data + i); 844 printf("%04X%s", boot_order[i / 2], i == size - 2 ? "\n" : ", "); 845 } 846 } 847 848 if (boot_order == NULL) { 849 /* 850 * now we want to fetch 'em all fresh again 851 * which possibly includes a newly created bootvar 852 */ 853 LIST_FOREACH(v, &efivars, entries) { 854 print_boot_var(v->name, verbose, v->idx == current); 855 } 856 } else { 857 LIST_FOREACH(v, &efivars, entries) { 858 v->flags = 0; 859 } 860 for (int i = 0; i < bolen; i++) { 861 char buffer[10]; 862 863 snprintf(buffer, sizeof(buffer), "Boot%04X", boot_order[i]); 864 if (!print_boot_var(buffer, verbose, boot_order[i] == current)) 865 printf("%s: MISSING!\n", buffer); 866 LIST_FOREACH(v, &efivars, entries) { 867 if (v->idx == boot_order[i]) { 868 v->flags |= SEEN; 869 break; 870 } 871 } 872 } 873 if (verbose) { 874 printf("\n\nUnreferenced Variables:\n"); 875 LIST_FOREACH(v, &efivars, entries) { 876 if (v->flags == 0) 877 print_boot_var(v->name, verbose, v->idx == current); 878 } 879 } 880 } 881 return 0; 882 } 883 884 static void 885 delete_timeout(void) 886 { 887 888 efi_del_variable(EFI_GLOBAL_GUID,"Timeout"); 889 } 890 891 static void 892 handle_timeout(int to) 893 { 894 uint16_t timeout; 895 896 le16enc(&timeout, to); 897 if (set_bootvar("Timeout", (uint8_t *)&timeout, sizeof(timeout)) < 0) 898 errx(1, "Can't set Timeout for booting."); 899 } 900 901 int 902 main(int argc, char *argv[]) 903 { 904 905 if (!efi_variables_supported()) 906 errx(1, "efi variables not supported on this system. root? kldload efirt?"); 907 908 memset(&opts, 0, sizeof (bmgr_opts_t)); 909 parse_args(argc, argv); 910 read_vars(); 911 912 if (opts.create) 913 /* 914 * side effect, adds to boot order, but not yet active. 915 */ 916 make_boot_var(opts.label ? opts.label : "", 917 opts.loader, opts.kernel, opts.env, opts.dry_run, 918 opts.has_bootnum ? opts.bootnum : -1); 919 else if (opts.set_active || opts.set_inactive ) 920 handle_activity(opts.bootnum, opts.set_active); 921 else if (opts.order != NULL) 922 set_boot_order(opts.order); /* create a new bootorder with opts.order */ 923 else if (opts.set_bootnext) 924 handle_bootnext(opts.bootnum); 925 else if (opts.delete_bootnext) 926 del_bootnext(); 927 else if (opts.delete) 928 delete_bootvar(opts.bootnum); 929 else if (opts.del_timeout) 930 delete_timeout(); 931 else if (opts.set_timeout) 932 handle_timeout(opts.timeout); 933 934 print_boot_vars(opts.verbose); 935 } 936