1 /*- 2 * Copyright (c) 1998 Michael Smith <[email protected]> 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 * 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 AND CONTRIBUTORS ``AS IS'' AND 15 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 16 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 17 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 18 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 19 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 20 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 21 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 22 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 23 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 24 * SUCH DAMAGE. 25 */ 26 27 #include <sys/cdefs.h> 28 __FBSDID("$FreeBSD$"); 29 30 /* 31 * file/module function dispatcher, support, etc. 32 */ 33 34 #include <stand.h> 35 #include <string.h> 36 #include <sys/param.h> 37 #include <sys/linker.h> 38 #include <sys/module.h> 39 #include <sys/queue.h> 40 #include <sys/stdint.h> 41 #include <sys/font.h> 42 #include <gfx_fb.h> 43 44 #if defined(LOADER_FDT_SUPPORT) 45 #include <fdt_platform.h> 46 #endif 47 48 #include "bootstrap.h" 49 50 #define MDIR_REMOVED 0x0001 51 #define MDIR_NOHINTS 0x0002 52 53 struct moduledir { 54 char *d_path; /* path of modules directory */ 55 u_char *d_hints; /* content of linker.hints file */ 56 int d_hintsz; /* size of hints data */ 57 int d_flags; 58 STAILQ_ENTRY(moduledir) d_link; 59 }; 60 61 static int file_load(char *filename, vm_offset_t dest, struct preloaded_file **result); 62 static int file_load_dependencies(struct preloaded_file *base_mod); 63 static char * file_search(const char *name, char **extlist); 64 static struct kernel_module * file_findmodule(struct preloaded_file *fp, char *modname, struct mod_depend *verinfo); 65 static int file_havepath(const char *name); 66 static char *mod_searchmodule(char *name, struct mod_depend *verinfo); 67 static char * mod_searchmodule_pnpinfo(const char *bus, const char *pnpinfo); 68 static void file_insert_tail(struct preloaded_file *mp); 69 static void file_remove(struct preloaded_file *fp); 70 struct file_metadata* metadata_next(struct file_metadata *base_mp, int type); 71 static void moduledir_readhints(struct moduledir *mdp); 72 static void moduledir_rebuild(void); 73 74 /* load address should be tweaked by first module loaded (kernel) */ 75 static vm_offset_t loadaddr = 0; 76 77 #if defined(LOADER_FDT_SUPPORT) 78 static const char *default_searchpath = 79 "/boot/kernel;/boot/modules;/boot/dtb"; 80 #else 81 static const char *default_searchpath = "/boot/kernel;/boot/modules"; 82 #endif 83 84 static STAILQ_HEAD(, moduledir) moduledir_list = 85 STAILQ_HEAD_INITIALIZER(moduledir_list); 86 87 struct preloaded_file *preloaded_files = NULL; 88 89 static char *kld_ext_list[] = { 90 ".ko", 91 "", 92 ".debug", 93 NULL 94 }; 95 96 /* 97 * load an object, either a disk file or code module. 98 * 99 * To load a file, the syntax is: 100 * 101 * load -t <type> <path> 102 * 103 * code modules are loaded as: 104 * 105 * load <path> <options> 106 */ 107 108 COMMAND_SET(load, "load", "load a kernel or module", command_load); 109 110 static int 111 command_load(int argc, char *argv[]) 112 { 113 struct preloaded_file *fp; 114 char *typestr; 115 char *prefix; 116 char *skip; 117 int dflag, dofile, dokld, ch, error; 118 119 dflag = dokld = dofile = 0; 120 optind = 1; 121 optreset = 1; 122 typestr = NULL; 123 if (argc == 1) { 124 command_errmsg = "no filename specified"; 125 return (CMD_CRIT); 126 } 127 prefix = skip = NULL; 128 while ((ch = getopt(argc, argv, "dkp:s:t:")) != -1) { 129 switch(ch) { 130 case 'd': 131 dflag++; 132 break; 133 case 'k': 134 dokld = 1; 135 break; 136 case 'p': 137 prefix = optarg; 138 break; 139 case 's': 140 skip = optarg; 141 break; 142 case 't': 143 typestr = optarg; 144 dofile = 1; 145 break; 146 case '?': 147 default: 148 /* getopt has already reported an error */ 149 return (CMD_OK); 150 } 151 } 152 argv += (optind - 1); 153 argc -= (optind - 1); 154 155 /* 156 * Request to load a raw file? 157 */ 158 if (dofile) { 159 if ((argc != 2) || (typestr == NULL) || (*typestr == 0)) { 160 command_errmsg = "invalid load type"; 161 return (CMD_CRIT); 162 } 163 164 #ifdef LOADER_VERIEXEC 165 if (strncmp(typestr, "manifest", 8) == 0) { 166 if (dflag > 0) 167 ve_debug_set(dflag); 168 return (load_manifest(argv[1], prefix, skip, NULL)); 169 } 170 #ifdef LOADER_VERIEXEC_PASS_MANIFEST 171 if (strncmp(typestr, "pass_manifest", 13) == 0) { 172 if (dflag > 0) 173 ve_debug_set(dflag); 174 return (pass_manifest(argv[1], prefix)); 175 } 176 #endif 177 #endif 178 179 fp = file_findfile(argv[1], typestr); 180 if (fp) { 181 snprintf(command_errbuf, sizeof(command_errbuf), 182 "warning: file '%s' already loaded", argv[1]); 183 return (CMD_WARN); 184 } 185 186 if (file_loadraw(argv[1], typestr, 1) != NULL) 187 return (CMD_OK); 188 189 /* Failing to load mfs_root is never going to end well! */ 190 if (strcmp("mfs_root", typestr) == 0) 191 return (CMD_FATAL); 192 193 return (CMD_ERROR); 194 } 195 /* 196 * Do we have explicit KLD load ? 197 */ 198 if (dokld || file_havepath(argv[1])) { 199 error = mod_loadkld(argv[1], argc - 2, argv + 2); 200 if (error == EEXIST) { 201 snprintf(command_errbuf, sizeof(command_errbuf), 202 "warning: KLD '%s' already loaded", argv[1]); 203 return (CMD_WARN); 204 } 205 206 return (error == 0 ? CMD_OK : CMD_CRIT); 207 } 208 /* 209 * Looks like a request for a module. 210 */ 211 error = mod_load(argv[1], NULL, argc - 2, argv + 2); 212 if (error == EEXIST) { 213 snprintf(command_errbuf, sizeof(command_errbuf), 214 "warning: module '%s' already loaded", argv[1]); 215 return (CMD_WARN); 216 } 217 218 return (error == 0 ? CMD_OK : CMD_CRIT); 219 } 220 221 #ifdef LOADER_GELI_SUPPORT 222 COMMAND_SET(load_geli, "load_geli", "load a geli key", command_load_geli); 223 224 static int 225 command_load_geli(int argc, char *argv[]) 226 { 227 char typestr[80]; 228 char *cp; 229 int ch, num; 230 231 if (argc < 3) { 232 command_errmsg = "usage is [-n key#] <prov> <file>"; 233 return(CMD_ERROR); 234 } 235 236 num = 0; 237 optind = 1; 238 optreset = 1; 239 while ((ch = getopt(argc, argv, "n:")) != -1) { 240 switch(ch) { 241 case 'n': 242 num = strtol(optarg, &cp, 0); 243 if (cp == optarg) { 244 snprintf(command_errbuf, sizeof(command_errbuf), 245 "bad key index '%s'", optarg); 246 return(CMD_ERROR); 247 } 248 break; 249 case '?': 250 default: 251 /* getopt has already reported an error */ 252 return(CMD_OK); 253 } 254 } 255 argv += (optind - 1); 256 argc -= (optind - 1); 257 sprintf(typestr, "%s:geli_keyfile%d", argv[1], num); 258 return (file_loadraw(argv[2], typestr, 1) ? CMD_OK : CMD_ERROR); 259 } 260 #endif 261 262 void 263 unload(void) 264 { 265 struct preloaded_file *fp; 266 267 while (preloaded_files != NULL) { 268 fp = preloaded_files; 269 preloaded_files = preloaded_files->f_next; 270 file_discard(fp); 271 } 272 loadaddr = 0; 273 unsetenv("kernelname"); 274 /* Reset tg_kernel_supported to allow next load to check it again. */ 275 gfx_state.tg_kernel_supported = false; 276 } 277 278 COMMAND_SET(unload, "unload", "unload all modules", command_unload); 279 280 static int 281 command_unload(int argc, char *argv[]) 282 { 283 unload(); 284 return(CMD_OK); 285 } 286 287 COMMAND_SET(lsmod, "lsmod", "list loaded modules", command_lsmod); 288 289 static int 290 command_lsmod(int argc, char *argv[]) 291 { 292 struct preloaded_file *fp; 293 struct kernel_module *mp; 294 struct file_metadata *md; 295 char lbuf[80]; 296 int ch, verbose, ret = 0; 297 298 verbose = 0; 299 optind = 1; 300 optreset = 1; 301 while ((ch = getopt(argc, argv, "v")) != -1) { 302 switch(ch) { 303 case 'v': 304 verbose = 1; 305 break; 306 case '?': 307 default: 308 /* getopt has already reported an error */ 309 return(CMD_OK); 310 } 311 } 312 313 pager_open(); 314 for (fp = preloaded_files; fp; fp = fp->f_next) { 315 snprintf(lbuf, sizeof(lbuf), " %p: ", (void *) fp->f_addr); 316 pager_output(lbuf); 317 pager_output(fp->f_name); 318 snprintf(lbuf, sizeof(lbuf), " (%s, 0x%lx)\n", fp->f_type, 319 (long)fp->f_size); 320 if (pager_output(lbuf)) 321 break; 322 if (fp->f_args != NULL) { 323 pager_output(" args: "); 324 pager_output(fp->f_args); 325 if (pager_output("\n")) 326 break; 327 } 328 if (fp->f_modules) { 329 pager_output(" modules: "); 330 for (mp = fp->f_modules; mp; mp = mp->m_next) { 331 snprintf(lbuf, sizeof(lbuf), "%s.%d ", mp->m_name, 332 mp->m_version); 333 pager_output(lbuf); 334 } 335 if (pager_output("\n")) 336 break; 337 } 338 if (verbose) { 339 /* XXX could add some formatting smarts here to display some better */ 340 for (md = fp->f_metadata; md != NULL; md = md->md_next) { 341 snprintf(lbuf, sizeof(lbuf), " 0x%04x, 0x%lx\n", 342 md->md_type, (long) md->md_size); 343 if (pager_output(lbuf)) 344 break; 345 } 346 } 347 if (ret) 348 break; 349 } 350 pager_close(); 351 return(CMD_OK); 352 } 353 354 COMMAND_SET(pnpmatch, "pnpmatch", "list matched modules based on pnpinfo", command_pnpmatch); 355 356 static int pnp_dump_flag = 0; 357 static int pnp_unbound_flag = 0; 358 static int pnp_verbose_flag = 0; 359 360 static int 361 command_pnpmatch(int argc, char *argv[]) 362 { 363 char *module; 364 int ch; 365 366 pnp_verbose_flag = 0; 367 pnp_dump_flag = 0; 368 optind = 1; 369 optreset = 1; 370 while ((ch = getopt(argc, argv, "vd")) != -1) { 371 switch(ch) { 372 case 'v': 373 pnp_verbose_flag = 1; 374 break; 375 case 'd': 376 pnp_dump_flag = 1; 377 break; 378 case '?': 379 default: 380 /* getopt has already reported an error */ 381 return(CMD_OK); 382 } 383 } 384 argv += optind; 385 argc -= optind; 386 387 if (argc != 2) { 388 command_errmsg = "Usage: pnpmatch <busname> compat=<compatdata>"; 389 return (CMD_CRIT); 390 } 391 392 module = mod_searchmodule_pnpinfo(argv[0], argv[1]); 393 if (module) 394 printf("Matched module: %s\n", module); 395 else 396 printf("No module matches %s on bus %s\n", argv[1], argv[0]); 397 398 return (CMD_OK); 399 } 400 401 COMMAND_SET(pnpload, "pnpload", "load matched modules based on pnpinfo", command_pnpload); 402 403 static int 404 command_pnpload(int argc, char *argv[]) 405 { 406 char *module; 407 int ch, error; 408 409 pnp_verbose_flag = 0; 410 pnp_dump_flag = 0; 411 optind = 1; 412 optreset = 1; 413 while ((ch = getopt(argc, argv, "vd")) != -1) { 414 switch(ch) { 415 case 'v': 416 pnp_verbose_flag = 1; 417 break; 418 case 'd': 419 pnp_dump_flag = 1; 420 break; 421 case '?': 422 default: 423 /* getopt has already reported an error */ 424 return(CMD_OK); 425 } 426 } 427 argv += optind; 428 argc -= optind; 429 430 if (argc != 2) { 431 command_errmsg = "Usage: pnpload <busname> compat=<compatdata>"; 432 return (CMD_ERROR); 433 } 434 435 module = mod_searchmodule_pnpinfo(argv[0], argv[1]); 436 437 error = mod_load(module, NULL, 0, NULL); 438 if (error == EEXIST) { 439 snprintf(command_errbuf, sizeof(command_errbuf), 440 "warning: module '%s' already loaded", argv[1]); 441 return (CMD_WARN); 442 } 443 444 return (error == 0 ? CMD_OK : CMD_CRIT); 445 } 446 447 #if defined(LOADER_FDT_SUPPORT) 448 static void 449 pnpautoload_simplebus(void) { 450 const char *pnpstring; 451 const char *compatstr; 452 char *pnpinfo = NULL; 453 char *module = NULL; 454 int tag = 0, len, pnplen; 455 int error; 456 457 while (1) { 458 pnpstring = fdt_devmatch_next(&tag, &len); 459 if (pnpstring == NULL) 460 return; 461 462 compatstr = pnpstring; 463 for (pnplen = 0; pnplen != len; compatstr = pnpstring + pnplen) { 464 pnplen += strlen(compatstr) + 1; 465 asprintf(&pnpinfo, "compat=%s", compatstr); 466 467 module = mod_searchmodule_pnpinfo("simplebus", pnpinfo); 468 if (module) { 469 error = mod_loadkld(module, 0, NULL); 470 if (error) 471 printf("Cannot load module %s\n", module); 472 break; 473 } 474 } 475 free(pnpinfo); 476 free(module); 477 } 478 } 479 #endif 480 481 struct pnp_bus { 482 const char *name; 483 void (*load)(void); 484 }; 485 486 struct pnp_bus pnp_buses[] = { 487 #if defined(LOADER_FDT_SUPPORT) 488 {"simplebus", pnpautoload_simplebus}, 489 #endif 490 }; 491 492 COMMAND_SET(pnpautoload, "pnpautoload", "auto load modules based on pnpinfo", command_pnpautoload); 493 494 static int 495 command_pnpautoload(int argc, char *argv[]) 496 { 497 int i; 498 int verbose; 499 int ch, match; 500 501 pnp_verbose_flag = 0; 502 pnp_dump_flag = 0; 503 verbose = 0; 504 optind = 1; 505 optreset = 1; 506 match = 0; 507 while ((ch = getopt(argc, argv, "v")) != -1) { 508 switch(ch) { 509 case 'v': 510 verbose = 1; 511 break; 512 case '?': 513 default: 514 /* getopt has already reported an error */ 515 return(CMD_OK); 516 } 517 } 518 argv += (optind - 1); 519 argc -= (optind - 1); 520 521 if (argc > 2) 522 return (CMD_ERROR); 523 524 for (i = 0; i < nitems(pnp_buses); i++) { 525 if (argc == 2 && strcmp(argv[1], pnp_buses[i].name) != 0) { 526 if (verbose) 527 printf("Skipping bus %s\n", pnp_buses[i].name); 528 continue; 529 } 530 if (verbose) 531 printf("Autoloading modules for simplebus\n"); 532 pnp_buses[i].load(); 533 match = 1; 534 } 535 if (match == 0) 536 printf("Unsupported bus %s\n", argv[1]); 537 538 return (CMD_OK); 539 } 540 541 /* 542 * File level interface, functions file_* 543 */ 544 int 545 file_load(char *filename, vm_offset_t dest, struct preloaded_file **result) 546 { 547 static int last_file_format = 0; 548 struct preloaded_file *fp; 549 int error; 550 int i; 551 552 if (archsw.arch_loadaddr != NULL) 553 dest = archsw.arch_loadaddr(LOAD_RAW, filename, dest); 554 555 error = EFTYPE; 556 for (i = last_file_format, fp = NULL; 557 file_formats[i] && fp == NULL; i++) { 558 error = (file_formats[i]->l_load)(filename, dest, &fp); 559 if (error == 0) { 560 fp->f_loader = last_file_format = i; /* remember the loader */ 561 *result = fp; 562 break; 563 } else if (last_file_format == i && i != 0) { 564 /* Restart from the beginning */ 565 i = -1; 566 last_file_format = 0; 567 fp = NULL; 568 continue; 569 } 570 if (error == EFTYPE) 571 continue; /* Unknown to this handler? */ 572 if (error) { 573 snprintf(command_errbuf, sizeof(command_errbuf), 574 "can't load file '%s': %s", filename, strerror(error)); 575 break; 576 } 577 } 578 return (error); 579 } 580 581 static int 582 file_load_dependencies(struct preloaded_file *base_file) 583 { 584 struct file_metadata *md; 585 struct preloaded_file *fp; 586 struct mod_depend *verinfo; 587 struct kernel_module *mp; 588 char *dmodname; 589 int error; 590 591 md = file_findmetadata(base_file, MODINFOMD_DEPLIST); 592 if (md == NULL) 593 return (0); 594 error = 0; 595 do { 596 verinfo = (struct mod_depend*)md->md_data; 597 dmodname = (char *)(verinfo + 1); 598 if (file_findmodule(NULL, dmodname, verinfo) == NULL) { 599 printf("loading required module '%s'\n", dmodname); 600 error = mod_load(dmodname, verinfo, 0, NULL); 601 if (error) 602 break; 603 /* 604 * If module loaded via kld name which isn't listed 605 * in the linker.hints file, we should check if it have 606 * required version. 607 */ 608 mp = file_findmodule(NULL, dmodname, verinfo); 609 if (mp == NULL) { 610 snprintf(command_errbuf, sizeof(command_errbuf), 611 "module '%s' exists but with wrong version", dmodname); 612 error = ENOENT; 613 break; 614 } 615 } 616 md = metadata_next(md, MODINFOMD_DEPLIST); 617 } while (md); 618 if (!error) 619 return (0); 620 /* Load failed; discard everything */ 621 while (base_file != NULL) { 622 fp = base_file; 623 base_file = base_file->f_next; 624 file_discard(fp); 625 } 626 return (error); 627 } 628 629 vm_offset_t 630 build_font_module(vm_offset_t addr) 631 { 632 vt_font_bitmap_data_t *bd; 633 struct vt_font *fd; 634 struct preloaded_file *fp; 635 size_t size; 636 uint32_t checksum; 637 int i; 638 struct font_info fi; 639 struct fontlist *fl; 640 uint64_t fontp; 641 642 if (STAILQ_EMPTY(&fonts)) 643 return (addr); 644 645 /* We can't load first */ 646 if ((file_findfile(NULL, NULL)) == NULL) { 647 printf("Can not load font module: %s\n", 648 "the kernel is not loaded"); 649 return (addr); 650 } 651 652 /* helper pointers */ 653 bd = NULL; 654 STAILQ_FOREACH(fl, &fonts, font_next) { 655 if (gfx_state.tg_font.vf_width == fl->font_data->vfbd_width && 656 gfx_state.tg_font.vf_height == fl->font_data->vfbd_height) { 657 /* 658 * Kernel does have better built in font. 659 */ 660 if (fl->font_flags == FONT_BUILTIN) 661 return (addr); 662 663 bd = fl->font_data; 664 break; 665 } 666 } 667 if (bd == NULL) 668 return (addr); 669 fd = bd->vfbd_font; 670 671 fi.fi_width = fd->vf_width; 672 checksum = fi.fi_width; 673 fi.fi_height = fd->vf_height; 674 checksum += fi.fi_height; 675 fi.fi_bitmap_size = bd->vfbd_uncompressed_size; 676 checksum += fi.fi_bitmap_size; 677 678 size = roundup2(sizeof (struct font_info), 8); 679 for (i = 0; i < VFNT_MAPS; i++) { 680 fi.fi_map_count[i] = fd->vf_map_count[i]; 681 checksum += fi.fi_map_count[i]; 682 size += fd->vf_map_count[i] * sizeof (struct vfnt_map); 683 size += roundup2(size, 8); 684 } 685 size += bd->vfbd_uncompressed_size; 686 687 fi.fi_checksum = -checksum; 688 689 fp = file_findfile(NULL, "elf kernel"); 690 if (fp == NULL) 691 fp = file_findfile(NULL, "elf64 kernel"); 692 if (fp == NULL) 693 panic("can't find kernel file"); 694 695 fontp = addr; 696 addr += archsw.arch_copyin(&fi, addr, sizeof (struct font_info)); 697 addr = roundup2(addr, 8); 698 699 /* Copy maps. */ 700 for (i = 0; i < VFNT_MAPS; i++) { 701 if (fd->vf_map_count[i] != 0) { 702 addr += archsw.arch_copyin(fd->vf_map[i], addr, 703 fd->vf_map_count[i] * sizeof (struct vfnt_map)); 704 addr = roundup2(addr, 8); 705 } 706 } 707 708 /* Copy the bitmap. */ 709 addr += archsw.arch_copyin(fd->vf_bytes, addr, fi.fi_bitmap_size); 710 711 /* Looks OK so far; populate control structure */ 712 file_addmetadata(fp, MODINFOMD_FONT, sizeof(fontp), &fontp); 713 return (addr); 714 } 715 716 #ifdef LOADER_VERIEXEC_VECTX 717 #define VECTX_HANDLE(fd) vctx 718 #else 719 #define VECTX_HANDLE(fd) fd 720 #endif 721 722 723 /* 724 * We've been asked to load (fname) as (type), so just suck it in, 725 * no arguments or anything. 726 */ 727 struct preloaded_file * 728 file_loadraw(const char *fname, char *type, int insert) 729 { 730 struct preloaded_file *fp; 731 char *name; 732 int fd, got; 733 vm_offset_t laddr; 734 #ifdef LOADER_VERIEXEC_VECTX 735 struct vectx *vctx; 736 int verror; 737 #endif 738 739 /* We can't load first */ 740 if ((file_findfile(NULL, NULL)) == NULL) { 741 command_errmsg = "can't load file before kernel"; 742 return(NULL); 743 } 744 745 /* locate the file on the load path */ 746 name = file_search(fname, NULL); 747 if (name == NULL) { 748 snprintf(command_errbuf, sizeof(command_errbuf), 749 "can't find '%s'", fname); 750 return(NULL); 751 } 752 753 if ((fd = open(name, O_RDONLY)) < 0) { 754 snprintf(command_errbuf, sizeof(command_errbuf), 755 "can't open '%s': %s", name, strerror(errno)); 756 free(name); 757 return(NULL); 758 } 759 760 #ifdef LOADER_VERIEXEC_VECTX 761 vctx = vectx_open(fd, name, 0L, NULL, &verror, __func__); 762 if (verror) { 763 sprintf(command_errbuf, "can't verify '%s': %s", 764 name, ve_error_get()); 765 free(name); 766 free(vctx); 767 close(fd); 768 return(NULL); 769 } 770 #else 771 #ifdef LOADER_VERIEXEC 772 if (verify_file(fd, name, 0, VE_MUST, __func__) < 0) { 773 sprintf(command_errbuf, "can't verify '%s': %s", 774 name, ve_error_get()); 775 free(name); 776 close(fd); 777 return(NULL); 778 } 779 #endif 780 #endif 781 782 if (archsw.arch_loadaddr != NULL) 783 loadaddr = archsw.arch_loadaddr(LOAD_RAW, name, loadaddr); 784 785 printf("%s ", name); 786 787 laddr = loadaddr; 788 for (;;) { 789 /* read in 4k chunks; size is not really important */ 790 got = archsw.arch_readin(VECTX_HANDLE(fd), laddr, 4096); 791 if (got == 0) /* end of file */ 792 break; 793 if (got < 0) { /* error */ 794 snprintf(command_errbuf, sizeof(command_errbuf), 795 "error reading '%s': %s", name, strerror(errno)); 796 free(name); 797 close(fd); 798 #ifdef LOADER_VERIEXEC_VECTX 799 free(vctx); 800 #endif 801 return(NULL); 802 } 803 laddr += got; 804 } 805 806 printf("size=%#jx\n", (uintmax_t)(laddr - loadaddr)); 807 #ifdef LOADER_VERIEXEC_VECTX 808 verror = vectx_close(vctx, VE_MUST, __func__); 809 if (verror) { 810 free(name); 811 close(fd); 812 free(vctx); 813 return(NULL); 814 } 815 #endif 816 817 /* Looks OK so far; create & populate control structure */ 818 fp = file_alloc(); 819 if (fp == NULL) { 820 snprintf(command_errbuf, sizeof (command_errbuf), 821 "no memory to load %s", name); 822 free(name); 823 close(fd); 824 return (NULL); 825 } 826 fp->f_name = name; 827 fp->f_type = strdup(type); 828 fp->f_args = NULL; 829 fp->f_metadata = NULL; 830 fp->f_loader = -1; 831 fp->f_addr = loadaddr; 832 fp->f_size = laddr - loadaddr; 833 834 if (fp->f_type == NULL) { 835 snprintf(command_errbuf, sizeof (command_errbuf), 836 "no memory to load %s", name); 837 free(name); 838 close(fd); 839 return (NULL); 840 } 841 /* recognise space consumption */ 842 loadaddr = laddr; 843 844 /* Add to the list of loaded files */ 845 if (insert != 0) 846 file_insert_tail(fp); 847 close(fd); 848 return(fp); 849 } 850 851 /* 852 * Load the module (name), pass it (argc),(argv), add container file 853 * to the list of loaded files. 854 * If module is already loaded just assign new argc/argv. 855 */ 856 int 857 mod_load(char *modname, struct mod_depend *verinfo, int argc, char *argv[]) 858 { 859 struct kernel_module *mp; 860 int err; 861 char *filename; 862 863 if (file_havepath(modname)) { 864 printf("Warning: mod_load() called instead of mod_loadkld() for module '%s'\n", modname); 865 return (mod_loadkld(modname, argc, argv)); 866 } 867 /* see if module is already loaded */ 868 mp = file_findmodule(NULL, modname, verinfo); 869 if (mp) { 870 #ifdef moduleargs 871 free(mp->m_args); 872 mp->m_args = unargv(argc, argv); 873 #endif 874 snprintf(command_errbuf, sizeof(command_errbuf), 875 "warning: module '%s' already loaded", mp->m_name); 876 return (0); 877 } 878 /* locate file with the module on the search path */ 879 filename = mod_searchmodule(modname, verinfo); 880 if (filename == NULL) { 881 snprintf(command_errbuf, sizeof(command_errbuf), 882 "can't find '%s'", modname); 883 return (ENOENT); 884 } 885 err = mod_loadkld(filename, argc, argv); 886 free(filename); 887 return (err); 888 } 889 890 /* 891 * Load specified KLD. If path is omitted, then try to locate it via 892 * search path. 893 */ 894 int 895 mod_loadkld(const char *kldname, int argc, char *argv[]) 896 { 897 struct preloaded_file *fp; 898 int err; 899 char *filename; 900 vm_offset_t loadaddr_saved; 901 902 /* 903 * Get fully qualified KLD name 904 */ 905 filename = file_search(kldname, kld_ext_list); 906 if (filename == NULL) { 907 snprintf(command_errbuf, sizeof(command_errbuf), 908 "can't find '%s'", kldname); 909 return (ENOENT); 910 } 911 /* 912 * Check if KLD already loaded 913 */ 914 fp = file_findfile(filename, NULL); 915 if (fp) { 916 snprintf(command_errbuf, sizeof(command_errbuf), 917 "warning: KLD '%s' already loaded", filename); 918 free(filename); 919 return (0); 920 } 921 922 do { 923 err = file_load(filename, loadaddr, &fp); 924 if (err) 925 break; 926 fp->f_args = unargv(argc, argv); 927 loadaddr_saved = loadaddr; 928 loadaddr = fp->f_addr + fp->f_size; 929 file_insert_tail(fp); /* Add to the list of loaded files */ 930 if (file_load_dependencies(fp) != 0) { 931 err = ENOENT; 932 file_remove(fp); 933 loadaddr = loadaddr_saved; 934 fp = NULL; 935 break; 936 } 937 } while(0); 938 if (err == EFTYPE) { 939 snprintf(command_errbuf, sizeof(command_errbuf), 940 "don't know how to load module '%s'", filename); 941 } 942 if (err) 943 file_discard(fp); 944 free(filename); 945 return (err); 946 } 947 948 /* 949 * Find a file matching (name) and (type). 950 * NULL may be passed as a wildcard to either. 951 */ 952 struct preloaded_file * 953 file_findfile(const char *name, const char *type) 954 { 955 struct preloaded_file *fp; 956 957 for (fp = preloaded_files; fp != NULL; fp = fp->f_next) { 958 if (((name == NULL) || !strcmp(name, fp->f_name)) && 959 ((type == NULL) || !strcmp(type, fp->f_type))) 960 break; 961 } 962 return (fp); 963 } 964 965 /* 966 * Find a module matching (name) inside of given file. 967 * NULL may be passed as a wildcard. 968 */ 969 struct kernel_module * 970 file_findmodule(struct preloaded_file *fp, char *modname, 971 struct mod_depend *verinfo) 972 { 973 struct kernel_module *mp, *best; 974 int bestver, mver; 975 976 if (fp == NULL) { 977 for (fp = preloaded_files; fp; fp = fp->f_next) { 978 mp = file_findmodule(fp, modname, verinfo); 979 if (mp) 980 return (mp); 981 } 982 return (NULL); 983 } 984 best = NULL; 985 bestver = 0; 986 for (mp = fp->f_modules; mp; mp = mp->m_next) { 987 if (strcmp(modname, mp->m_name) == 0) { 988 if (verinfo == NULL) 989 return (mp); 990 mver = mp->m_version; 991 if (mver == verinfo->md_ver_preferred) 992 return (mp); 993 if (mver >= verinfo->md_ver_minimum && 994 mver <= verinfo->md_ver_maximum && 995 mver > bestver) { 996 best = mp; 997 bestver = mver; 998 } 999 } 1000 } 1001 return (best); 1002 } 1003 /* 1004 * Make a copy of (size) bytes of data from (p), and associate them as 1005 * metadata of (type) to the module (mp). 1006 */ 1007 void 1008 file_addmetadata(struct preloaded_file *fp, int type, size_t size, void *p) 1009 { 1010 struct file_metadata *md; 1011 1012 md = malloc(sizeof(struct file_metadata) - sizeof(md->md_data) + size); 1013 if (md != NULL) { 1014 md->md_size = size; 1015 md->md_type = type; 1016 bcopy(p, md->md_data, size); 1017 md->md_next = fp->f_metadata; 1018 } 1019 fp->f_metadata = md; 1020 } 1021 1022 /* 1023 * Find a metadata object of (type) associated with the file (fp) 1024 */ 1025 struct file_metadata * 1026 file_findmetadata(struct preloaded_file *fp, int type) 1027 { 1028 struct file_metadata *md; 1029 1030 for (md = fp->f_metadata; md != NULL; md = md->md_next) 1031 if (md->md_type == type) 1032 break; 1033 return(md); 1034 } 1035 1036 /* 1037 * Remove all metadata from the file. 1038 */ 1039 void 1040 file_removemetadata(struct preloaded_file *fp) 1041 { 1042 struct file_metadata *md, *next; 1043 1044 for (md = fp->f_metadata; md != NULL; md = next) 1045 { 1046 next = md->md_next; 1047 free(md); 1048 } 1049 fp->f_metadata = NULL; 1050 } 1051 1052 /* 1053 * Add a buffer to the list of preloaded "files". 1054 */ 1055 int 1056 file_addbuf(const char *name, const char *type, size_t len, void *buf) 1057 { 1058 struct preloaded_file *fp; 1059 vm_offset_t dest; 1060 1061 /* We can't load first */ 1062 if ((file_findfile(NULL, NULL)) == NULL) { 1063 command_errmsg = "can't load file before kernel"; 1064 return (-1); 1065 } 1066 1067 /* Figure out where to load the data. */ 1068 dest = loadaddr; 1069 if (archsw.arch_loadaddr != NULL) 1070 dest = archsw.arch_loadaddr(LOAD_RAW, (void *)name, dest); 1071 1072 /* Create & populate control structure */ 1073 fp = file_alloc(); 1074 if (fp == NULL) { 1075 snprintf(command_errbuf, sizeof (command_errbuf), 1076 "no memory to load %s", name); 1077 return (-1); 1078 } 1079 fp->f_name = strdup(name); 1080 fp->f_type = strdup(type); 1081 fp->f_args = NULL; 1082 fp->f_metadata = NULL; 1083 fp->f_loader = -1; 1084 fp->f_addr = dest; 1085 fp->f_size = len; 1086 if ((fp->f_name == NULL) || (fp->f_type == NULL)) { 1087 snprintf(command_errbuf, sizeof (command_errbuf), 1088 "no memory to load %s", name); 1089 free(fp->f_name); 1090 free(fp->f_type); 1091 return (-1); 1092 } 1093 1094 /* Copy the data in. */ 1095 archsw.arch_copyin(buf, fp->f_addr, len); 1096 loadaddr = fp->f_addr + len; 1097 1098 /* Add to the list of loaded files */ 1099 file_insert_tail(fp); 1100 return(0); 1101 } 1102 1103 struct file_metadata * 1104 metadata_next(struct file_metadata *md, int type) 1105 { 1106 1107 if (md == NULL) 1108 return (NULL); 1109 while((md = md->md_next) != NULL) 1110 if (md->md_type == type) 1111 break; 1112 return (md); 1113 } 1114 1115 static char *emptyextlist[] = { "", NULL }; 1116 1117 /* 1118 * Check if the given file is in place and return full path to it. 1119 */ 1120 static char * 1121 file_lookup(const char *path, const char *name, int namelen, char **extlist) 1122 { 1123 struct stat st; 1124 char *result, *cp, **cpp; 1125 int pathlen, extlen, len; 1126 1127 pathlen = strlen(path); 1128 extlen = 0; 1129 if (extlist == NULL) 1130 extlist = emptyextlist; 1131 for (cpp = extlist; *cpp; cpp++) { 1132 len = strlen(*cpp); 1133 if (len > extlen) 1134 extlen = len; 1135 } 1136 result = malloc(pathlen + namelen + extlen + 2); 1137 if (result == NULL) 1138 return (NULL); 1139 bcopy(path, result, pathlen); 1140 if (pathlen > 0 && result[pathlen - 1] != '/') 1141 result[pathlen++] = '/'; 1142 cp = result + pathlen; 1143 bcopy(name, cp, namelen); 1144 cp += namelen; 1145 for (cpp = extlist; *cpp; cpp++) { 1146 strcpy(cp, *cpp); 1147 if (stat(result, &st) == 0 && S_ISREG(st.st_mode)) 1148 return result; 1149 } 1150 free(result); 1151 return NULL; 1152 } 1153 1154 /* 1155 * Check if file name have any qualifiers 1156 */ 1157 static int 1158 file_havepath(const char *name) 1159 { 1160 const char *cp; 1161 1162 archsw.arch_getdev(NULL, name, &cp); 1163 return (cp != name || strchr(name, '/') != NULL); 1164 } 1165 1166 /* 1167 * Attempt to find the file (name) on the module searchpath. 1168 * If (name) is qualified in any way, we simply check it and 1169 * return it or NULL. If it is not qualified, then we attempt 1170 * to construct a path using entries in the environment variable 1171 * module_path. 1172 * 1173 * The path we return a pointer to need never be freed, as we manage 1174 * it internally. 1175 */ 1176 static char * 1177 file_search(const char *name, char **extlist) 1178 { 1179 struct moduledir *mdp; 1180 struct stat sb; 1181 char *result; 1182 int namelen; 1183 1184 /* Don't look for nothing */ 1185 if (name == NULL) 1186 return(NULL); 1187 1188 if (*name == 0) 1189 return(strdup(name)); 1190 1191 if (file_havepath(name)) { 1192 /* Qualified, so just see if it exists */ 1193 if (stat(name, &sb) == 0) 1194 return(strdup(name)); 1195 return(NULL); 1196 } 1197 moduledir_rebuild(); 1198 result = NULL; 1199 namelen = strlen(name); 1200 STAILQ_FOREACH(mdp, &moduledir_list, d_link) { 1201 result = file_lookup(mdp->d_path, name, namelen, extlist); 1202 if (result) 1203 break; 1204 } 1205 return(result); 1206 } 1207 1208 #define INT_ALIGN(base, ptr) ptr = \ 1209 (base) + roundup2((ptr) - (base), sizeof(int)) 1210 1211 static char * 1212 mod_search_hints(struct moduledir *mdp, const char *modname, 1213 struct mod_depend *verinfo) 1214 { 1215 u_char *cp, *recptr, *bufend, *best; 1216 char *result; 1217 int *intp, bestver, blen, clen, found, ival, modnamelen, reclen; 1218 1219 moduledir_readhints(mdp); 1220 modnamelen = strlen(modname); 1221 found = 0; 1222 result = NULL; 1223 bestver = 0; 1224 if (mdp->d_hints == NULL) 1225 goto bad; 1226 recptr = mdp->d_hints; 1227 bufend = recptr + mdp->d_hintsz; 1228 clen = blen = 0; 1229 best = cp = NULL; 1230 while (recptr < bufend && !found) { 1231 intp = (int*)recptr; 1232 reclen = *intp++; 1233 ival = *intp++; 1234 cp = (u_char*)intp; 1235 switch (ival) { 1236 case MDT_VERSION: 1237 clen = *cp++; 1238 if (clen != modnamelen || bcmp(cp, modname, clen) != 0) 1239 break; 1240 cp += clen; 1241 INT_ALIGN(mdp->d_hints, cp); 1242 ival = *(int*)cp; 1243 cp += sizeof(int); 1244 clen = *cp++; 1245 if (verinfo == NULL || ival == verinfo->md_ver_preferred) { 1246 found = 1; 1247 break; 1248 } 1249 if (ival >= verinfo->md_ver_minimum && 1250 ival <= verinfo->md_ver_maximum && 1251 ival > bestver) { 1252 bestver = ival; 1253 best = cp; 1254 blen = clen; 1255 } 1256 break; 1257 default: 1258 break; 1259 } 1260 recptr += reclen + sizeof(int); 1261 } 1262 /* 1263 * Finally check if KLD is in the place 1264 */ 1265 if (found) 1266 result = file_lookup(mdp->d_path, (const char *)cp, clen, NULL); 1267 else if (best) 1268 result = file_lookup(mdp->d_path, (const char *)best, blen, NULL); 1269 bad: 1270 /* 1271 * If nothing found or hints is absent - fallback to the old way 1272 * by using "kldname[.ko]" as module name. 1273 */ 1274 if (!found && !bestver && result == NULL) 1275 result = file_lookup(mdp->d_path, modname, modnamelen, kld_ext_list); 1276 return result; 1277 } 1278 1279 static int 1280 getint(void **ptr) 1281 { 1282 int *p = *ptr; 1283 int rv; 1284 1285 p = (int *)roundup2((intptr_t)p, sizeof(int)); 1286 rv = *p++; 1287 *ptr = p; 1288 return rv; 1289 } 1290 1291 static void 1292 getstr(void **ptr, char *val) 1293 { 1294 int *p = *ptr; 1295 char *c = (char *)p; 1296 int len = *(uint8_t *)c; 1297 1298 memcpy(val, c + 1, len); 1299 val[len] = 0; 1300 c += len + 1; 1301 *ptr = (void *)c; 1302 } 1303 1304 static int 1305 pnpval_as_int(const char *val, const char *pnpinfo) 1306 { 1307 int rv; 1308 char key[256]; 1309 char *cp; 1310 1311 if (pnpinfo == NULL) 1312 return -1; 1313 1314 cp = strchr(val, ';'); 1315 key[0] = ' '; 1316 if (cp == NULL) 1317 strlcpy(key + 1, val, sizeof(key) - 1); 1318 else { 1319 memcpy(key + 1, val, cp - val); 1320 key[cp - val + 1] = '\0'; 1321 } 1322 strlcat(key, "=", sizeof(key)); 1323 if (strncmp(key + 1, pnpinfo, strlen(key + 1)) == 0) 1324 rv = strtol(pnpinfo + strlen(key + 1), NULL, 0); 1325 else { 1326 cp = strstr(pnpinfo, key); 1327 if (cp == NULL) 1328 rv = -1; 1329 else 1330 rv = strtol(cp + strlen(key), NULL, 0); 1331 } 1332 return rv; 1333 } 1334 1335 static void 1336 quoted_strcpy(char *dst, const char *src) 1337 { 1338 char q = ' '; 1339 1340 if (*src == '\'' || *src == '"') 1341 q = *src++; 1342 while (*src && *src != q) 1343 *dst++ = *src++; // XXX backtick quoting 1344 *dst++ = '\0'; 1345 // XXX overflow 1346 } 1347 1348 static char * 1349 pnpval_as_str(const char *val, const char *pnpinfo) 1350 { 1351 static char retval[256]; 1352 char key[256]; 1353 char *cp; 1354 1355 if (pnpinfo == NULL) { 1356 *retval = '\0'; 1357 return retval; 1358 } 1359 1360 cp = strchr(val, ';'); 1361 key[0] = ' '; 1362 if (cp == NULL) 1363 strlcpy(key + 1, val, sizeof(key) - 1); 1364 else { 1365 memcpy(key + 1, val, cp - val); 1366 key[cp - val + 1] = '\0'; 1367 } 1368 strlcat(key, "=", sizeof(key)); 1369 if (strncmp(key + 1, pnpinfo, strlen(key + 1)) == 0) 1370 quoted_strcpy(retval, pnpinfo + strlen(key + 1)); 1371 else { 1372 cp = strstr(pnpinfo, key); 1373 if (cp == NULL) 1374 strcpy(retval, "MISSING"); 1375 else 1376 quoted_strcpy(retval, cp + strlen(key)); 1377 } 1378 return retval; 1379 } 1380 1381 static char * 1382 devmatch_search_hints(struct moduledir *mdp, const char *bus, const char *dev, const char *pnpinfo) 1383 { 1384 char val1[256], val2[256]; 1385 int ival, len, ents, i, notme, mask, bit, v, found; 1386 void *ptr, *walker, *hints_end; 1387 char *lastmod = NULL, *cp, *s; 1388 1389 moduledir_readhints(mdp); 1390 found = 0; 1391 if (mdp->d_hints == NULL) 1392 goto bad; 1393 walker = mdp->d_hints; 1394 hints_end = walker + mdp->d_hintsz; 1395 while (walker < hints_end && !found) { 1396 len = getint(&walker); 1397 ival = getint(&walker); 1398 ptr = walker; 1399 switch (ival) { 1400 case MDT_VERSION: 1401 getstr(&ptr, val1); 1402 ival = getint(&ptr); 1403 getstr(&ptr, val2); 1404 if (pnp_dump_flag || pnp_verbose_flag) 1405 printf("Version: if %s.%d kmod %s\n", val1, ival, val2); 1406 break; 1407 case MDT_MODULE: 1408 getstr(&ptr, val1); 1409 getstr(&ptr, val2); 1410 if (lastmod) 1411 free(lastmod); 1412 lastmod = strdup(val2); 1413 if (pnp_dump_flag || pnp_verbose_flag) 1414 printf("module %s in %s\n", val1, val1); 1415 break; 1416 case MDT_PNP_INFO: 1417 if (!pnp_dump_flag && !pnp_unbound_flag && lastmod && strcmp(lastmod, "kernel") == 0) 1418 break; 1419 getstr(&ptr, val1); 1420 getstr(&ptr, val2); 1421 ents = getint(&ptr); 1422 if (pnp_dump_flag || pnp_verbose_flag) 1423 printf("PNP info for bus %s format %s %d entries (%s)\n", 1424 val1, val2, ents, lastmod); 1425 if (strcmp(val1, "usb") == 0) { 1426 if (pnp_verbose_flag) 1427 printf("Treating usb as uhub -- bug in source table still?\n"); 1428 strcpy(val1, "uhub"); 1429 } 1430 if (bus && strcmp(val1, bus) != 0) { 1431 if (pnp_verbose_flag) 1432 printf("Skipped because table for bus %s, looking for %s\n", 1433 val1, bus); 1434 break; 1435 } 1436 for (i = 0; i < ents; i++) { 1437 if (pnp_verbose_flag) 1438 printf("---------- Entry %d ----------\n", i); 1439 if (pnp_dump_flag) 1440 printf(" "); 1441 cp = val2; 1442 notme = 0; 1443 mask = -1; 1444 bit = -1; 1445 do { 1446 switch (*cp) { 1447 /* All integer fields */ 1448 case 'I': 1449 case 'J': 1450 case 'G': 1451 case 'L': 1452 case 'M': 1453 ival = getint(&ptr); 1454 if (pnp_dump_flag) { 1455 printf("%#x:", ival); 1456 break; 1457 } 1458 if (bit >= 0 && ((1 << bit) & mask) == 0) 1459 break; 1460 v = pnpval_as_int(cp + 2, pnpinfo); 1461 if (pnp_verbose_flag) 1462 printf("Matching %s (%c) table=%#x tomatch=%#x\n", 1463 cp + 2, *cp, v, ival); 1464 switch (*cp) { 1465 case 'J': 1466 if (ival == -1) 1467 break; 1468 /*FALLTHROUGH*/ 1469 case 'I': 1470 if (v != ival) 1471 notme++; 1472 break; 1473 case 'G': 1474 if (v < ival) 1475 notme++; 1476 break; 1477 case 'L': 1478 if (v > ival) 1479 notme++; 1480 break; 1481 case 'M': 1482 mask = ival; 1483 break; 1484 } 1485 break; 1486 /* String fields */ 1487 case 'D': 1488 case 'Z': 1489 getstr(&ptr, val1); 1490 if (pnp_dump_flag) { 1491 printf("'%s':", val1); 1492 break; 1493 } 1494 if (*cp == 'D') 1495 break; 1496 s = pnpval_as_str(cp + 2, pnpinfo); 1497 if (strcmp(s, val1) != 0) 1498 notme++; 1499 break; 1500 /* Key override fields, required to be last in the string */ 1501 case 'T': 1502 /* 1503 * This is imperfect and only does one key and will be redone 1504 * to be more general for multiple keys. Currently, nothing 1505 * does that. 1506 */ 1507 if (pnp_dump_flag) /* No per-row data stored */ 1508 break; 1509 if (cp[strlen(cp) - 1] == ';') /* Skip required ; at end */ 1510 cp[strlen(cp) - 1] = '\0'; /* in case it's not there */ 1511 if ((s = strstr(pnpinfo, cp + 2)) == NULL) 1512 notme++; 1513 else if (s > pnpinfo && s[-1] != ' ') 1514 notme++; 1515 break; 1516 default: 1517 printf("Unknown field type %c\n:", *cp); 1518 break; 1519 } 1520 bit++; 1521 cp = strchr(cp, ';'); 1522 if (cp) 1523 cp++; 1524 } while (cp && *cp); 1525 if (pnp_dump_flag) 1526 printf("\n"); 1527 else if (!notme) { 1528 if (!pnp_unbound_flag) { 1529 if (pnp_verbose_flag) 1530 printf("Matches --- %s ---\n", lastmod); 1531 } 1532 found++; 1533 } 1534 } 1535 break; 1536 default: 1537 break; 1538 } 1539 walker = (void *)(len - sizeof(int) + (intptr_t)walker); 1540 } 1541 if (pnp_unbound_flag && found == 0 && *pnpinfo) { 1542 if (pnp_verbose_flag) 1543 printf("------------------------- "); 1544 printf("%s on %s pnpinfo %s", *dev ? dev : "unattached", bus, pnpinfo); 1545 if (pnp_verbose_flag) 1546 printf(" -------------------------"); 1547 printf("\n"); 1548 } 1549 if (found != 0) 1550 return (lastmod); 1551 free(lastmod); 1552 1553 bad: 1554 return (NULL); 1555 } 1556 1557 /* 1558 * Attempt to locate the file containing the module (name) 1559 */ 1560 static char * 1561 mod_searchmodule(char *name, struct mod_depend *verinfo) 1562 { 1563 struct moduledir *mdp; 1564 char *result; 1565 1566 moduledir_rebuild(); 1567 /* 1568 * Now we ready to lookup module in the given directories 1569 */ 1570 result = NULL; 1571 STAILQ_FOREACH(mdp, &moduledir_list, d_link) { 1572 result = mod_search_hints(mdp, name, verinfo); 1573 if (result) 1574 break; 1575 } 1576 1577 return(result); 1578 } 1579 1580 static char * 1581 mod_searchmodule_pnpinfo(const char *bus, const char *pnpinfo) 1582 { 1583 struct moduledir *mdp; 1584 char *result; 1585 1586 moduledir_rebuild(); 1587 /* 1588 * Now we ready to lookup module in the given directories 1589 */ 1590 result = NULL; 1591 STAILQ_FOREACH(mdp, &moduledir_list, d_link) { 1592 result = devmatch_search_hints(mdp, bus, NULL, pnpinfo); 1593 if (result) 1594 break; 1595 } 1596 1597 return(result); 1598 } 1599 1600 int 1601 file_addmodule(struct preloaded_file *fp, char *modname, int version, 1602 struct kernel_module **newmp) 1603 { 1604 struct kernel_module *mp; 1605 struct mod_depend mdepend; 1606 1607 bzero(&mdepend, sizeof(mdepend)); 1608 mdepend.md_ver_preferred = version; 1609 mp = file_findmodule(fp, modname, &mdepend); 1610 if (mp) 1611 return (EEXIST); 1612 mp = calloc(1, sizeof(struct kernel_module)); 1613 if (mp == NULL) 1614 return (ENOMEM); 1615 mp->m_name = strdup(modname); 1616 if (mp->m_name == NULL) { 1617 free(mp); 1618 return (ENOMEM); 1619 } 1620 mp->m_version = version; 1621 mp->m_fp = fp; 1622 mp->m_next = fp->f_modules; 1623 fp->f_modules = mp; 1624 if (newmp) 1625 *newmp = mp; 1626 return (0); 1627 } 1628 1629 /* 1630 * Throw a file away 1631 */ 1632 void 1633 file_discard(struct preloaded_file *fp) 1634 { 1635 struct file_metadata *md, *md1; 1636 struct kernel_module *mp, *mp1; 1637 if (fp == NULL) 1638 return; 1639 md = fp->f_metadata; 1640 while (md) { 1641 md1 = md; 1642 md = md->md_next; 1643 free(md1); 1644 } 1645 mp = fp->f_modules; 1646 while (mp) { 1647 free(mp->m_name); 1648 mp1 = mp; 1649 mp = mp->m_next; 1650 free(mp1); 1651 } 1652 free(fp->f_name); 1653 free(fp->f_type); 1654 free(fp->f_args); 1655 free(fp); 1656 } 1657 1658 /* 1659 * Allocate a new file; must be used instead of malloc() 1660 * to ensure safe initialisation. 1661 */ 1662 struct preloaded_file * 1663 file_alloc(void) 1664 { 1665 1666 return (calloc(1, sizeof(struct preloaded_file))); 1667 } 1668 1669 /* 1670 * Add a module to the chain 1671 */ 1672 static void 1673 file_insert_tail(struct preloaded_file *fp) 1674 { 1675 struct preloaded_file *cm; 1676 1677 /* Append to list of loaded file */ 1678 fp->f_next = NULL; 1679 if (preloaded_files == NULL) { 1680 preloaded_files = fp; 1681 } else { 1682 for (cm = preloaded_files; cm->f_next != NULL; cm = cm->f_next) 1683 ; 1684 cm->f_next = fp; 1685 } 1686 } 1687 1688 /* 1689 * Remove module from the chain 1690 */ 1691 static void 1692 file_remove(struct preloaded_file *fp) 1693 { 1694 struct preloaded_file *cm; 1695 1696 if (preloaded_files == NULL) 1697 return; 1698 1699 if (preloaded_files == fp) { 1700 preloaded_files = fp->f_next; 1701 return; 1702 } 1703 for (cm = preloaded_files; cm->f_next != NULL; cm = cm->f_next) { 1704 if (cm->f_next == fp) { 1705 cm->f_next = fp->f_next; 1706 return; 1707 } 1708 } 1709 } 1710 1711 static char * 1712 moduledir_fullpath(struct moduledir *mdp, const char *fname) 1713 { 1714 char *cp; 1715 1716 cp = malloc(strlen(mdp->d_path) + strlen(fname) + 2); 1717 if (cp == NULL) 1718 return NULL; 1719 strcpy(cp, mdp->d_path); 1720 strcat(cp, "/"); 1721 strcat(cp, fname); 1722 return (cp); 1723 } 1724 1725 /* 1726 * Read linker.hints file into memory performing some sanity checks. 1727 */ 1728 static void 1729 moduledir_readhints(struct moduledir *mdp) 1730 { 1731 struct stat st; 1732 char *path; 1733 int fd, size, version; 1734 1735 if (mdp->d_hints != NULL || (mdp->d_flags & MDIR_NOHINTS)) 1736 return; 1737 path = moduledir_fullpath(mdp, "linker.hints"); 1738 if (stat(path, &st) != 0 || 1739 st.st_size < (ssize_t)(sizeof(version) + sizeof(int)) || 1740 st.st_size > LINKER_HINTS_MAX || (fd = open(path, O_RDONLY)) < 0) { 1741 free(path); 1742 mdp->d_flags |= MDIR_NOHINTS; 1743 return; 1744 } 1745 free(path); 1746 size = read(fd, &version, sizeof(version)); 1747 if (size != sizeof(version) || version != LINKER_HINTS_VERSION) 1748 goto bad; 1749 size = st.st_size - size; 1750 mdp->d_hints = malloc(size); 1751 if (mdp->d_hints == NULL) 1752 goto bad; 1753 if (read(fd, mdp->d_hints, size) != size) 1754 goto bad; 1755 mdp->d_hintsz = size; 1756 close(fd); 1757 return; 1758 bad: 1759 close(fd); 1760 free(mdp->d_hints); 1761 mdp->d_hints = NULL; 1762 mdp->d_flags |= MDIR_NOHINTS; 1763 return; 1764 } 1765 1766 /* 1767 * Extract directories from the ';' separated list, remove duplicates. 1768 */ 1769 static void 1770 moduledir_rebuild(void) 1771 { 1772 struct moduledir *mdp, *mtmp; 1773 const char *path, *cp, *ep; 1774 size_t cplen; 1775 1776 path = getenv("module_path"); 1777 if (path == NULL) 1778 path = default_searchpath; 1779 /* 1780 * Rebuild list of module directories if it changed 1781 */ 1782 STAILQ_FOREACH(mdp, &moduledir_list, d_link) 1783 mdp->d_flags |= MDIR_REMOVED; 1784 1785 for (ep = path; *ep != 0; ep++) { 1786 cp = ep; 1787 for (; *ep != 0 && *ep != ';'; ep++) 1788 ; 1789 /* 1790 * Ignore trailing slashes 1791 */ 1792 for (cplen = ep - cp; cplen > 1 && cp[cplen - 1] == '/'; cplen--) 1793 ; 1794 STAILQ_FOREACH(mdp, &moduledir_list, d_link) { 1795 if (strlen(mdp->d_path) != cplen || bcmp(cp, mdp->d_path, cplen) != 0) 1796 continue; 1797 mdp->d_flags &= ~MDIR_REMOVED; 1798 break; 1799 } 1800 if (mdp == NULL) { 1801 mdp = malloc(sizeof(*mdp) + cplen + 1); 1802 if (mdp == NULL) 1803 return; 1804 mdp->d_path = (char*)(mdp + 1); 1805 bcopy(cp, mdp->d_path, cplen); 1806 mdp->d_path[cplen] = 0; 1807 mdp->d_hints = NULL; 1808 mdp->d_flags = 0; 1809 STAILQ_INSERT_TAIL(&moduledir_list, mdp, d_link); 1810 } 1811 if (*ep == 0) 1812 break; 1813 } 1814 /* 1815 * Delete unused directories if any 1816 */ 1817 mdp = STAILQ_FIRST(&moduledir_list); 1818 while (mdp) { 1819 if ((mdp->d_flags & MDIR_REMOVED) == 0) { 1820 mdp = STAILQ_NEXT(mdp, d_link); 1821 } else { 1822 free(mdp->d_hints); 1823 mtmp = mdp; 1824 mdp = STAILQ_NEXT(mdp, d_link); 1825 STAILQ_REMOVE(&moduledir_list, mtmp, moduledir, d_link); 1826 free(mtmp); 1827 } 1828 } 1829 return; 1830 } 1831