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