1 /*- 2 * Copyright (c) 1997 Doug Rabson 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 * $FreeBSD: src/sys/kern/kern_linker.c,v 1.41.2.3 2001/11/21 17:50:35 luigi Exp $ 27 * $DragonFly: src/sys/kern/kern_linker.c,v 1.41 2008/01/06 16:55:51 swildner Exp $ 28 */ 29 30 #include "opt_ddb.h" 31 32 #include <sys/param.h> 33 #include <sys/kernel.h> 34 #include <sys/systm.h> 35 #include <sys/malloc.h> 36 #include <sys/sysproto.h> 37 #include <sys/sysent.h> 38 #include <sys/proc.h> 39 #include <sys/lock.h> 40 #include <sys/module.h> 41 #include <sys/linker.h> 42 #include <sys/fcntl.h> 43 #include <sys/libkern.h> 44 #include <sys/nlookup.h> 45 #include <sys/vnode.h> 46 #include <sys/sysctl.h> 47 48 #include <vm/vm_zone.h> 49 50 #ifdef _KERNEL_VIRTUAL 51 #include <dlfcn.h> 52 #endif 53 54 #ifdef KLD_DEBUG 55 int kld_debug = 0; 56 #endif 57 58 /* Metadata from the static kernel */ 59 SET_DECLARE(modmetadata_set, struct mod_metadata); 60 MALLOC_DEFINE(M_LINKER, "kld", "kernel linker"); 61 62 linker_file_t linker_current_file; 63 linker_file_t linker_kernel_file; 64 65 static struct lock lock; /* lock for the file list */ 66 static linker_class_list_t classes; 67 static linker_file_list_t linker_files; 68 static int next_file_id = 1; 69 70 static void 71 linker_init(void* arg) 72 { 73 lockinit(&lock, "klink", 0, 0); 74 TAILQ_INIT(&classes); 75 TAILQ_INIT(&linker_files); 76 } 77 78 SYSINIT(linker, SI_BOOT2_KLD, SI_ORDER_FIRST, linker_init, 0); 79 80 int 81 linker_add_class(const char* desc, void* priv, 82 struct linker_class_ops* ops) 83 { 84 linker_class_t lc; 85 86 lc = kmalloc(sizeof(struct linker_class), M_LINKER, M_NOWAIT | M_ZERO); 87 if (!lc) 88 return ENOMEM; 89 90 lc->desc = desc; 91 lc->priv = priv; 92 lc->ops = ops; 93 TAILQ_INSERT_HEAD(&classes, lc, link); 94 95 return 0; 96 } 97 98 static int 99 linker_file_sysinit(linker_file_t lf) 100 { 101 struct sysinit** start, ** stop; 102 struct sysinit** sipp; 103 struct sysinit** xipp; 104 struct sysinit* save; 105 const moduledata_t *moddata; 106 int error; 107 108 KLD_DPF(FILE, ("linker_file_sysinit: calling SYSINITs for %s\n", 109 lf->filename)); 110 111 if (linker_file_lookup_set(lf, "sysinit_set", &start, &stop, NULL) != 0) 112 return 0; /* XXX is this correct ? No sysinit ? */ 113 114 /* HACK ALERT! */ 115 for (sipp = start; sipp < stop; sipp++) { 116 if ((*sipp)->func == module_register_init) { 117 moddata = (*sipp)->udata; 118 error = module_register(moddata, lf); 119 if (error) { 120 kprintf("linker_file_sysinit \"%s\" failed to register! %d\n", 121 lf->filename, error); 122 return error; 123 } 124 } 125 } 126 127 /* 128 * Perform a bubble sort of the system initialization objects by 129 * their subsystem (primary key) and order (secondary key). 130 * 131 * Since some things care about execution order, this is the 132 * operation which ensures continued function. 133 */ 134 for (sipp = start; sipp < stop; sipp++) { 135 for (xipp = sipp + 1; xipp < stop; xipp++) { 136 if ((*sipp)->subsystem < (*xipp)->subsystem || 137 ((*sipp)->subsystem == (*xipp)->subsystem && 138 (*sipp)->order <= (*xipp)->order)) 139 continue; /* skip*/ 140 save = *sipp; 141 *sipp = *xipp; 142 *xipp = save; 143 } 144 } 145 146 147 /* 148 * Traverse the (now) ordered list of system initialization tasks. 149 * Perform each task, and continue on to the next task. 150 */ 151 for (sipp = start; sipp < stop; sipp++) { 152 if ((*sipp)->subsystem == SI_SPECIAL_DUMMY) 153 continue; /* skip dummy task(s)*/ 154 155 /* Call function */ 156 (*((*sipp)->func))((*sipp)->udata); 157 } 158 return 0; /* no errors */ 159 } 160 161 static void 162 linker_file_sysuninit(linker_file_t lf) 163 { 164 struct sysinit** start, ** stop; 165 struct sysinit** sipp; 166 struct sysinit** xipp; 167 struct sysinit* save; 168 169 KLD_DPF(FILE, ("linker_file_sysuninit: calling SYSUNINITs for %s\n", 170 lf->filename)); 171 172 if (linker_file_lookup_set(lf, "sysuninit_set", &start, &stop, NULL) != 0) 173 return; 174 175 /* 176 * Perform a reverse bubble sort of the system initialization objects 177 * by their subsystem (primary key) and order (secondary key). 178 * 179 * Since some things care about execution order, this is the 180 * operation which ensures continued function. 181 */ 182 for (sipp = start; sipp < stop; sipp++) { 183 for (xipp = sipp + 1; xipp < stop; xipp++) { 184 if ((*sipp)->subsystem > (*xipp)->subsystem || 185 ((*sipp)->subsystem == (*xipp)->subsystem && 186 (*sipp)->order >= (*xipp)->order)) 187 continue; /* skip*/ 188 save = *sipp; 189 *sipp = *xipp; 190 *xipp = save; 191 } 192 } 193 194 195 /* 196 * Traverse the (now) ordered list of system initialization tasks. 197 * Perform each task, and continue on to the next task. 198 */ 199 for (sipp = start; sipp < stop; sipp++) { 200 if ((*sipp)->subsystem == SI_SPECIAL_DUMMY) 201 continue; /* skip dummy task(s)*/ 202 203 /* Call function */ 204 (*((*sipp)->func))((*sipp)->udata); 205 } 206 } 207 208 static void 209 linker_file_register_sysctls(linker_file_t lf) 210 { 211 struct sysctl_oid **start, **stop, **oidp; 212 213 KLD_DPF(FILE, ("linker_file_register_sysctls: registering SYSCTLs for %s\n", 214 lf->filename)); 215 216 if (linker_file_lookup_set(lf, "sysctl_set", &start, &stop, NULL) != 0) 217 return; 218 for (oidp = start; oidp < stop; oidp++) 219 sysctl_register_oid(*oidp); 220 } 221 222 static void 223 linker_file_unregister_sysctls(linker_file_t lf) 224 { 225 struct sysctl_oid **start, **stop, **oidp; 226 227 KLD_DPF(FILE, ("linker_file_unregister_sysctls: registering SYSCTLs for %s\n", 228 lf->filename)); 229 230 if (linker_file_lookup_set(lf, "sysctl_set", &start, &stop, NULL) != 0) 231 return; 232 for (oidp = start; oidp < stop; oidp++) 233 sysctl_unregister_oid(*oidp); 234 } 235 236 int 237 linker_load_file(const char* filename, linker_file_t* result) 238 { 239 linker_class_t lc; 240 linker_file_t lf; 241 int foundfile, error = 0; 242 char *koname = NULL; 243 244 /* Refuse to load modules if securelevel raised */ 245 if (securelevel > 0 || kernel_mem_readonly) 246 return EPERM; 247 248 lf = linker_find_file_by_name(filename); 249 if (lf) { 250 KLD_DPF(FILE, ("linker_load_file: file %s is already loaded, incrementing refs\n", filename)); 251 *result = lf; 252 lf->refs++; 253 goto out; 254 } 255 if (find_mod_metadata(filename)) { 256 if (linker_kernel_file) 257 ++linker_kernel_file->refs; 258 *result = linker_kernel_file; 259 goto out; 260 } 261 262 koname = kmalloc(strlen(filename) + 4, M_LINKER, M_WAITOK); 263 ksprintf(koname, "%s.ko", filename); 264 lf = NULL; 265 foundfile = 0; 266 for (lc = TAILQ_FIRST(&classes); lc; lc = TAILQ_NEXT(lc, link)) { 267 KLD_DPF(FILE, ("linker_load_file: trying to load %s as %s\n", 268 filename, lc->desc)); 269 270 error = lc->ops->load_file(koname, &lf); /* First with .ko */ 271 if (lf == NULL && error == ENOENT) 272 error = lc->ops->load_file(filename, &lf); /* Then try without */ 273 /* 274 * If we got something other than ENOENT, then it exists but we cannot 275 * load it for some other reason. 276 */ 277 if (error != ENOENT) 278 foundfile = 1; 279 if (lf) { 280 linker_file_register_sysctls(lf); 281 error = linker_file_sysinit(lf); 282 283 *result = lf; 284 goto out; 285 } 286 } 287 /* 288 * Less than ideal, but tells the user whether it failed to load or 289 * the module was not found. 290 */ 291 if (foundfile) { 292 /* 293 * Format not recognized or otherwise unloadable. 294 * When loading a module that is statically built into 295 * the kernel EEXIST percolates back up as the return 296 * value. Preserve this so that apps can recognize this 297 * special case. 298 */ 299 if (error != EEXIST) 300 error = ENOEXEC; 301 } else { 302 error = ENOENT; /* Nothing found */ 303 } 304 305 out: 306 if (koname) 307 kfree(koname, M_LINKER); 308 return error; 309 } 310 311 linker_file_t 312 linker_find_file_by_name(const char* filename) 313 { 314 linker_file_t lf = 0; 315 char *koname; 316 int i; 317 318 for (i = strlen(filename); i > 0 && filename[i-1] != '/'; --i) 319 ; 320 filename += i; 321 322 koname = kmalloc(strlen(filename) + 4, M_LINKER, M_WAITOK); 323 ksprintf(koname, "%s.ko", filename); 324 325 lockmgr(&lock, LK_SHARED); 326 for (lf = TAILQ_FIRST(&linker_files); lf; lf = TAILQ_NEXT(lf, link)) { 327 if (!strcmp(lf->filename, koname)) 328 break; 329 if (!strcmp(lf->filename, filename)) 330 break; 331 } 332 lockmgr(&lock, LK_RELEASE); 333 334 if (koname) 335 kfree(koname, M_LINKER); 336 return lf; 337 } 338 339 linker_file_t 340 linker_find_file_by_id(int fileid) 341 { 342 linker_file_t lf = 0; 343 344 lockmgr(&lock, LK_SHARED); 345 for (lf = TAILQ_FIRST(&linker_files); lf; lf = TAILQ_NEXT(lf, link)) 346 if (lf->id == fileid) 347 break; 348 lockmgr(&lock, LK_RELEASE); 349 350 return lf; 351 } 352 353 linker_file_t 354 linker_make_file(const char* pathname, void* priv, struct linker_file_ops* ops) 355 { 356 linker_file_t lf = 0; 357 int namelen; 358 const char *filename; 359 360 filename = rindex(pathname, '/'); 361 if (filename && filename[1]) 362 filename++; 363 else 364 filename = pathname; 365 366 KLD_DPF(FILE, ("linker_make_file: new file, filename=%s\n", filename)); 367 lockmgr(&lock, LK_EXCLUSIVE); 368 namelen = strlen(filename) + 1; 369 lf = kmalloc(sizeof(struct linker_file) + namelen, M_LINKER, M_WAITOK); 370 bzero(lf, sizeof(*lf)); 371 372 lf->refs = 1; 373 lf->userrefs = 0; 374 lf->flags = 0; 375 lf->filename = (char*) (lf + 1); 376 strcpy(lf->filename, filename); 377 lf->id = next_file_id++; 378 lf->ndeps = 0; 379 lf->deps = NULL; 380 STAILQ_INIT(&lf->common); 381 TAILQ_INIT(&lf->modules); 382 383 lf->priv = priv; 384 lf->ops = ops; 385 TAILQ_INSERT_TAIL(&linker_files, lf, link); 386 387 lockmgr(&lock, LK_RELEASE); 388 return lf; 389 } 390 391 int 392 linker_file_unload(linker_file_t file) 393 { 394 module_t mod, next; 395 struct common_symbol* cp; 396 int error = 0; 397 int i; 398 399 /* Refuse to unload modules if securelevel raised */ 400 if (securelevel > 0 || kernel_mem_readonly) 401 return EPERM; 402 403 KLD_DPF(FILE, ("linker_file_unload: lf->refs=%d\n", file->refs)); 404 lockmgr(&lock, LK_EXCLUSIVE); 405 if (file->refs == 1) { 406 KLD_DPF(FILE, ("linker_file_unload: file is unloading, informing modules\n")); 407 /* 408 * Temporarily bump file->refs to prevent recursive unloading 409 */ 410 ++file->refs; 411 412 /* 413 * Inform any modules associated with this file. 414 */ 415 mod = TAILQ_FIRST(&file->modules); 416 while (mod) { 417 /* 418 * Give the module a chance to veto the unload. Note that the 419 * act of unloading the module may cause other modules in the 420 * same file list to be unloaded recursively. 421 */ 422 if ((error = module_unload(mod)) != 0) { 423 KLD_DPF(FILE, ("linker_file_unload: module %x vetoes unload\n", 424 mod)); 425 lockmgr(&lock, LK_RELEASE); 426 file->refs--; 427 goto out; 428 } 429 430 /* 431 * Recursive relationships may prevent the release from 432 * immediately removing the module, or may remove other 433 * modules in the list. 434 */ 435 next = module_getfnext(mod); 436 module_release(mod); 437 mod = next; 438 } 439 440 /* 441 * Since we intend to destroy the file structure, we expect all 442 * modules to have been removed by now. 443 */ 444 for (mod = TAILQ_FIRST(&file->modules); 445 mod; 446 mod = module_getfnext(mod) 447 ) { 448 kprintf("linker_file_unload: module %p still has refs!\n", mod); 449 } 450 --file->refs; 451 } 452 453 file->refs--; 454 if (file->refs > 0) { 455 lockmgr(&lock, LK_RELEASE); 456 goto out; 457 } 458 459 /* Don't try to run SYSUNINITs if we are unloaded due to a link error */ 460 if (file->flags & LINKER_FILE_LINKED) { 461 linker_file_sysuninit(file); 462 linker_file_unregister_sysctls(file); 463 } 464 465 TAILQ_REMOVE(&linker_files, file, link); 466 lockmgr(&lock, LK_RELEASE); 467 468 for (i = 0; i < file->ndeps; i++) 469 linker_file_unload(file->deps[i]); 470 kfree(file->deps, M_LINKER); 471 472 for (cp = STAILQ_FIRST(&file->common); cp; 473 cp = STAILQ_FIRST(&file->common)) { 474 STAILQ_REMOVE(&file->common, cp, common_symbol, link); 475 kfree(cp, M_LINKER); 476 } 477 478 file->ops->unload(file); 479 kfree(file, M_LINKER); 480 481 out: 482 return error; 483 } 484 485 int 486 linker_file_add_dependancy(linker_file_t file, linker_file_t dep) 487 { 488 linker_file_t* newdeps; 489 490 newdeps = kmalloc((file->ndeps + 1) * sizeof(linker_file_t*), 491 M_LINKER, M_WAITOK | M_ZERO); 492 493 if (file->deps) { 494 bcopy(file->deps, newdeps, file->ndeps * sizeof(linker_file_t*)); 495 kfree(file->deps, M_LINKER); 496 } 497 file->deps = newdeps; 498 file->deps[file->ndeps] = dep; 499 file->ndeps++; 500 501 return 0; 502 } 503 504 /* 505 * Locate a linker set and its contents. 506 * This is a helper function to avoid linker_if.h exposure elsewhere. 507 * Note: firstp and lastp are really void *** 508 */ 509 int 510 linker_file_lookup_set(linker_file_t file, const char *name, 511 void *firstp, void *lastp, int *countp) 512 { 513 return file->ops->lookup_set(file, name, firstp, lastp, countp); 514 } 515 516 int 517 linker_file_lookup_symbol(linker_file_t file, const char* name, int deps, caddr_t *raddr) 518 { 519 c_linker_sym_t sym; 520 linker_symval_t symval; 521 linker_file_t lf; 522 size_t common_size = 0; 523 int i; 524 525 KLD_DPF(SYM, ("linker_file_lookup_symbol: file=%x, name=%s, deps=%d\n", 526 file, name, deps)); 527 528 if (file->ops->lookup_symbol(file, name, &sym) == 0) { 529 file->ops->symbol_values(file, sym, &symval); 530 531 /* 532 * XXX Assume a common symbol if its value is 0 and it has a non-zero 533 * size, otherwise it could be an absolute symbol with a value of 0. 534 */ 535 if (symval.value == 0 && symval.size != 0) { 536 /* 537 * For commons, first look them up in the dependancies and 538 * only allocate space if not found there. 539 */ 540 common_size = symval.size; 541 } else { 542 KLD_DPF(SYM, ("linker_file_lookup_symbol: symbol.value=%x\n", symval.value)); 543 *raddr = symval.value; 544 return 0; 545 } 546 } 547 if (deps) { 548 for (i = 0; i < file->ndeps; i++) { 549 if (linker_file_lookup_symbol(file->deps[i], name, 0, raddr) == 0) { 550 KLD_DPF(SYM, ("linker_file_lookup_symbol: deps value=%x\n", *raddr)); 551 return 0; 552 } 553 } 554 555 /* If we have not found it in the dependencies, search globally */ 556 for (lf = TAILQ_FIRST(&linker_files); lf; lf = TAILQ_NEXT(lf, link)) { 557 /* But skip the current file if it's on the list */ 558 if (lf == file) 559 continue; 560 /* And skip the files we searched above */ 561 for (i = 0; i < file->ndeps; i++) 562 if (lf == file->deps[i]) 563 break; 564 if (i < file->ndeps) 565 continue; 566 if (linker_file_lookup_symbol(lf, name, 0, raddr) == 0) { 567 KLD_DPF(SYM, ("linker_file_lookup_symbol: global value=%x\n", *raddr)); 568 return 0; 569 } 570 } 571 } 572 573 if (common_size > 0) { 574 /* 575 * This is a common symbol which was not found in the 576 * dependancies. We maintain a simple common symbol table in 577 * the file object. 578 */ 579 struct common_symbol* cp; 580 581 for (cp = STAILQ_FIRST(&file->common); cp; 582 cp = STAILQ_NEXT(cp, link)) 583 if (!strcmp(cp->name, name)) { 584 KLD_DPF(SYM, ("linker_file_lookup_symbol: old common value=%x\n", cp->address)); 585 *raddr = cp->address; 586 return 0; 587 } 588 589 /* 590 * Round the symbol size up to align. 591 */ 592 common_size = (common_size + sizeof(int) - 1) & -sizeof(int); 593 cp = kmalloc(sizeof(struct common_symbol) 594 + common_size 595 + strlen(name) + 1, 596 M_LINKER, M_WAITOK | M_ZERO); 597 598 cp->address = (caddr_t) (cp + 1); 599 cp->name = cp->address + common_size; 600 strcpy(cp->name, name); 601 bzero(cp->address, common_size); 602 STAILQ_INSERT_TAIL(&file->common, cp, link); 603 604 KLD_DPF(SYM, ("linker_file_lookup_symbol: new common value=%x\n", cp->address)); 605 *raddr = cp->address; 606 return 0; 607 } 608 609 #ifdef _KERNEL_VIRTUAL 610 *raddr = dlsym(RTLD_NEXT, name); 611 if (*raddr != NULL) { 612 KLD_DPF(SYM, ("linker_file_lookup_symbol: found dlsym=%x\n", *raddr)); 613 return 0; 614 } 615 #endif 616 617 KLD_DPF(SYM, ("linker_file_lookup_symbol: fail\n")); 618 return ENOENT; 619 } 620 621 #ifdef DDB 622 /* 623 * DDB Helpers. DDB has to look across multiple files with their own 624 * symbol tables and string tables. 625 * 626 * Note that we do not obey list locking protocols here. We really don't 627 * need DDB to hang because somebody's got the lock held. We'll take the 628 * chance that the files list is inconsistant instead. 629 */ 630 631 int 632 linker_ddb_lookup(const char *symstr, c_linker_sym_t *sym) 633 { 634 linker_file_t lf; 635 636 for (lf = TAILQ_FIRST(&linker_files); lf; lf = TAILQ_NEXT(lf, link)) { 637 if (lf->ops->lookup_symbol(lf, symstr, sym) == 0) 638 return 0; 639 } 640 return ENOENT; 641 } 642 643 int 644 linker_ddb_search_symbol(caddr_t value, c_linker_sym_t *sym, long *diffp) 645 { 646 linker_file_t lf; 647 u_long off = (uintptr_t)value; 648 u_long diff, bestdiff; 649 c_linker_sym_t best; 650 c_linker_sym_t es; 651 652 best = 0; 653 bestdiff = off; 654 for (lf = TAILQ_FIRST(&linker_files); lf; lf = TAILQ_NEXT(lf, link)) { 655 if (lf->ops->search_symbol(lf, value, &es, &diff) != 0) 656 continue; 657 if (es != 0 && diff < bestdiff) { 658 best = es; 659 bestdiff = diff; 660 } 661 if (bestdiff == 0) 662 break; 663 } 664 if (best) { 665 *sym = best; 666 *diffp = bestdiff; 667 return 0; 668 } else { 669 *sym = 0; 670 *diffp = off; 671 return ENOENT; 672 } 673 } 674 675 int 676 linker_ddb_symbol_values(c_linker_sym_t sym, linker_symval_t *symval) 677 { 678 linker_file_t lf; 679 680 for (lf = TAILQ_FIRST(&linker_files); lf; lf = TAILQ_NEXT(lf, link)) { 681 if (lf->ops->symbol_values(lf, sym, symval) == 0) 682 return 0; 683 } 684 return ENOENT; 685 } 686 687 #endif 688 689 /* 690 * Syscalls. 691 */ 692 693 int 694 sys_kldload(struct kldload_args *uap) 695 { 696 struct thread *td = curthread; 697 char* filename = NULL, *modulename; 698 linker_file_t lf; 699 int error = 0; 700 701 uap->sysmsg_result = -1; 702 703 if (securelevel > 0 || kernel_mem_readonly) /* redundant, but that's OK */ 704 return EPERM; 705 706 if ((error = suser(td)) != 0) 707 return error; 708 709 filename = kmalloc(MAXPATHLEN, M_TEMP, M_WAITOK); 710 if ((error = copyinstr(uap->file, filename, MAXPATHLEN, NULL)) != 0) 711 goto out; 712 713 /* Can't load more than one module with the same name */ 714 modulename = rindex(filename, '/'); 715 if (modulename == NULL) 716 modulename = filename; 717 else 718 modulename++; 719 if (linker_find_file_by_name(modulename)) { 720 error = EEXIST; 721 goto out; 722 } 723 724 if ((error = linker_load_file(filename, &lf)) != 0) 725 goto out; 726 727 lf->userrefs++; 728 uap->sysmsg_result = lf->id; 729 730 out: 731 if (filename) 732 kfree(filename, M_TEMP); 733 return error; 734 } 735 736 int 737 sys_kldunload(struct kldunload_args *uap) 738 { 739 struct thread *td = curthread; 740 linker_file_t lf; 741 int error = 0; 742 743 if (securelevel > 0 || kernel_mem_readonly) /* redundant, but that's OK */ 744 return EPERM; 745 746 if ((error = suser(td)) != 0) 747 return error; 748 749 lf = linker_find_file_by_id(uap->fileid); 750 if (lf) { 751 KLD_DPF(FILE, ("kldunload: lf->userrefs=%d\n", lf->userrefs)); 752 if (lf->userrefs == 0) { 753 kprintf("linkerunload: attempt to unload file that was loaded by the kernel\n"); 754 error = EBUSY; 755 goto out; 756 } 757 lf->userrefs--; 758 error = linker_file_unload(lf); 759 if (error) 760 lf->userrefs++; 761 } else 762 error = ENOENT; 763 764 out: 765 return error; 766 } 767 768 int 769 sys_kldfind(struct kldfind_args *uap) 770 { 771 char *filename = NULL, *modulename; 772 linker_file_t lf; 773 int error = 0; 774 775 uap->sysmsg_result = -1; 776 777 filename = kmalloc(MAXPATHLEN, M_TEMP, M_WAITOK); 778 if ((error = copyinstr(uap->file, filename, MAXPATHLEN, NULL)) != 0) 779 goto out; 780 781 modulename = rindex(filename, '/'); 782 if (modulename == NULL) 783 modulename = filename; 784 785 lf = linker_find_file_by_name(modulename); 786 if (lf) 787 uap->sysmsg_result = lf->id; 788 else 789 error = ENOENT; 790 791 out: 792 if (filename) 793 kfree(filename, M_TEMP); 794 return error; 795 } 796 797 int 798 sys_kldnext(struct kldnext_args *uap) 799 { 800 linker_file_t lf; 801 int error = 0; 802 803 if (uap->fileid == 0) { 804 if (TAILQ_FIRST(&linker_files)) 805 uap->sysmsg_result = TAILQ_FIRST(&linker_files)->id; 806 else 807 uap->sysmsg_result = 0; 808 return 0; 809 } 810 811 lf = linker_find_file_by_id(uap->fileid); 812 if (lf) { 813 if (TAILQ_NEXT(lf, link)) 814 uap->sysmsg_result = TAILQ_NEXT(lf, link)->id; 815 else 816 uap->sysmsg_result = 0; 817 } else 818 error = ENOENT; 819 820 return error; 821 } 822 823 int 824 sys_kldstat(struct kldstat_args *uap) 825 { 826 linker_file_t lf; 827 int error = 0; 828 int version; 829 struct kld_file_stat* stat; 830 int namelen; 831 832 lf = linker_find_file_by_id(uap->fileid); 833 if (!lf) { 834 error = ENOENT; 835 goto out; 836 } 837 838 stat = uap->stat; 839 840 /* 841 * Check the version of the user's structure. 842 */ 843 if ((error = copyin(&stat->version, &version, sizeof(version))) != 0) 844 goto out; 845 if (version != sizeof(struct kld_file_stat)) { 846 error = EINVAL; 847 goto out; 848 } 849 850 namelen = strlen(lf->filename) + 1; 851 if (namelen > MAXPATHLEN) 852 namelen = MAXPATHLEN; 853 if ((error = copyout(lf->filename, &stat->name[0], namelen)) != 0) 854 goto out; 855 if ((error = copyout(&lf->refs, &stat->refs, sizeof(int))) != 0) 856 goto out; 857 if ((error = copyout(&lf->id, &stat->id, sizeof(int))) != 0) 858 goto out; 859 if ((error = copyout(&lf->address, &stat->address, sizeof(caddr_t))) != 0) 860 goto out; 861 if ((error = copyout(&lf->size, &stat->size, sizeof(size_t))) != 0) 862 goto out; 863 864 uap->sysmsg_result = 0; 865 866 out: 867 return error; 868 } 869 870 int 871 sys_kldfirstmod(struct kldfirstmod_args *uap) 872 { 873 linker_file_t lf; 874 int error = 0; 875 876 lf = linker_find_file_by_id(uap->fileid); 877 if (lf) { 878 if (TAILQ_FIRST(&lf->modules)) 879 uap->sysmsg_result = module_getid(TAILQ_FIRST(&lf->modules)); 880 else 881 uap->sysmsg_result = 0; 882 } else 883 error = ENOENT; 884 885 return error; 886 } 887 888 int 889 sys_kldsym(struct kldsym_args *uap) 890 { 891 char *symstr = NULL; 892 c_linker_sym_t sym; 893 linker_symval_t symval; 894 linker_file_t lf; 895 struct kld_sym_lookup lookup; 896 int error = 0; 897 898 if ((error = copyin(uap->data, &lookup, sizeof(lookup))) != 0) 899 goto out; 900 if (lookup.version != sizeof(lookup) || uap->cmd != KLDSYM_LOOKUP) { 901 error = EINVAL; 902 goto out; 903 } 904 905 symstr = kmalloc(MAXPATHLEN, M_TEMP, M_WAITOK); 906 if ((error = copyinstr(lookup.symname, symstr, MAXPATHLEN, NULL)) != 0) 907 goto out; 908 909 if (uap->fileid != 0) { 910 lf = linker_find_file_by_id(uap->fileid); 911 if (lf == NULL) { 912 error = ENOENT; 913 goto out; 914 } 915 if (lf->ops->lookup_symbol(lf, symstr, &sym) == 0 && 916 lf->ops->symbol_values(lf, sym, &symval) == 0) { 917 lookup.symvalue = (uintptr_t)symval.value; 918 lookup.symsize = symval.size; 919 error = copyout(&lookup, uap->data, sizeof(lookup)); 920 } else 921 error = ENOENT; 922 } else { 923 for (lf = TAILQ_FIRST(&linker_files); lf; lf = TAILQ_NEXT(lf, link)) { 924 if (lf->ops->lookup_symbol(lf, symstr, &sym) == 0 && 925 lf->ops->symbol_values(lf, sym, &symval) == 0) { 926 lookup.symvalue = (uintptr_t)symval.value; 927 lookup.symsize = symval.size; 928 error = copyout(&lookup, uap->data, sizeof(lookup)); 929 break; 930 } 931 } 932 if (!lf) 933 error = ENOENT; 934 } 935 out: 936 if (symstr) 937 kfree(symstr, M_TEMP); 938 return error; 939 } 940 941 /* 942 * Look for module metadata in the static kernel 943 */ 944 struct mod_metadata * 945 find_mod_metadata(const char *modname) 946 { 947 int len; 948 struct mod_metadata **mdp; 949 struct mod_metadata *mdt; 950 951 /* 952 * Strip path prefixes and any dot extension. MDT_MODULE names 953 * are just the module name without a path or ".ko". 954 */ 955 for (len = strlen(modname) - 1; len >= 0; --len) { 956 if (modname[len] == '/') 957 break; 958 } 959 modname += len + 1; 960 for (len = 0; modname[len] && modname[len] != '.'; ++len) 961 ; 962 963 /* 964 * Look for the module declaration 965 */ 966 SET_FOREACH(mdp, modmetadata_set) { 967 mdt = *mdp; 968 if (mdt->md_type != MDT_MODULE) 969 continue; 970 if (strlen(mdt->md_cval) == len && 971 strncmp(mdt->md_cval, modname, len) == 0) { 972 return(mdt); 973 } 974 } 975 return(NULL); 976 } 977 978 /* 979 * Preloaded module support 980 */ 981 static void 982 linker_preload(void* arg) 983 { 984 caddr_t modptr; 985 char *modname; 986 char *modtype; 987 linker_file_t lf; 988 linker_class_t lc; 989 int error; 990 struct sysinit **sipp; 991 const moduledata_t *moddata; 992 struct sysinit **si_start, **si_stop; 993 994 modptr = NULL; 995 while ((modptr = preload_search_next_name(modptr)) != NULL) { 996 modname = (char *)preload_search_info(modptr, MODINFO_NAME); 997 modtype = (char *)preload_search_info(modptr, MODINFO_TYPE); 998 if (modname == NULL) { 999 kprintf("Preloaded module at %p does not have a name!\n", modptr); 1000 continue; 1001 } 1002 if (modtype == NULL) { 1003 kprintf("Preloaded module at %p does not have a type!\n", modptr); 1004 continue; 1005 } 1006 1007 /* 1008 * This is a hack at the moment, but what's in FreeBSD-5 is even 1009 * worse so I'd rather the hack. 1010 */ 1011 kprintf("Preloaded %s \"%s\" at %p", modtype, modname, modptr); 1012 if (find_mod_metadata(modname)) { 1013 kprintf(" (ignored, already in static kernel)\n"); 1014 continue; 1015 } 1016 kprintf(".\n"); 1017 1018 lf = linker_find_file_by_name(modname); 1019 if (lf) { 1020 lf->refs++; 1021 lf->userrefs++; 1022 continue; 1023 } 1024 lf = NULL; 1025 for (lc = TAILQ_FIRST(&classes); lc; lc = TAILQ_NEXT(lc, link)) { 1026 error = lc->ops->load_file(modname, &lf); 1027 if (error) { 1028 lf = NULL; 1029 break; 1030 } 1031 } 1032 if (lf) { 1033 lf->userrefs++; 1034 1035 if (linker_file_lookup_set(lf, "sysinit_set", &si_start, &si_stop, NULL) == 0) { 1036 /* HACK ALERT! 1037 * This is to set the sysinit moduledata so that the module 1038 * can attach itself to the correct containing file. 1039 * The sysinit could be run at *any* time. 1040 */ 1041 for (sipp = si_start; sipp < si_stop; sipp++) { 1042 if ((*sipp)->func == module_register_init) { 1043 moddata = (*sipp)->udata; 1044 error = module_register(moddata, lf); 1045 if (error) 1046 kprintf("Preloaded %s \"%s\" failed to register: %d\n", 1047 modtype, modname, error); 1048 } 1049 } 1050 sysinit_add(si_start, si_stop); 1051 } 1052 linker_file_register_sysctls(lf); 1053 } 1054 } 1055 } 1056 1057 SYSINIT(preload, SI_BOOT2_KLD, SI_ORDER_MIDDLE, linker_preload, 0); 1058 1059 /* 1060 * Search for a not-loaded module by name. 1061 * 1062 * Modules may be found in the following locations: 1063 * 1064 * - preloaded (result is just the module name) 1065 * - on disk (result is full path to module) 1066 * 1067 * If the module name is qualified in any way (contains path, etc.) 1068 * the we simply return a copy of it. 1069 * 1070 * The search path can be manipulated via sysctl. Note that we use the ';' 1071 * character as a separator to be consistent with the bootloader. 1072 */ 1073 1074 static char linker_path[MAXPATHLEN] = "/;/boot;/modules"; 1075 1076 SYSCTL_STRING(_kern, OID_AUTO, module_path, CTLFLAG_RW, linker_path, 1077 sizeof(linker_path), "module load search path"); 1078 TUNABLE_STR("module_path", linker_path, sizeof(linker_path)); 1079 1080 static char * 1081 linker_strdup(const char *str) 1082 { 1083 char *result; 1084 1085 result = kmalloc(strlen(str) + 1, M_LINKER, M_WAITOK); 1086 strcpy(result, str); 1087 return(result); 1088 } 1089 1090 char * 1091 linker_search_path(const char *name) 1092 { 1093 struct nlookupdata nd; 1094 char *cp, *ep, *result; 1095 size_t name_len, prefix_len; 1096 int sep; 1097 int error; 1098 enum vtype type; 1099 1100 /* qualified at all? */ 1101 if (index(name, '/')) 1102 return(linker_strdup(name)); 1103 1104 /* traverse the linker path */ 1105 cp = linker_path; 1106 name_len = strlen(name); 1107 for (;;) { 1108 1109 /* find the end of this component */ 1110 for (ep = cp; (*ep != 0) && (*ep != ';'); ep++) 1111 ; 1112 prefix_len = ep - cp; 1113 /* if this component doesn't end with a slash, add one */ 1114 if (ep == cp || *(ep - 1) != '/') 1115 sep = 1; 1116 else 1117 sep = 0; 1118 1119 /* 1120 * +2 : possible separator, plus terminator. 1121 */ 1122 result = kmalloc(prefix_len + name_len + 2, M_LINKER, M_WAITOK); 1123 1124 strncpy(result, cp, prefix_len); 1125 if (sep) 1126 result[prefix_len++] = '/'; 1127 strcpy(result + prefix_len, name); 1128 1129 /* 1130 * Attempt to open the file, and return the path if we succeed and it's 1131 * a regular file. 1132 */ 1133 error = nlookup_init(&nd, result, UIO_SYSSPACE, NLC_FOLLOW|NLC_LOCKVP); 1134 if (error == 0) 1135 error = vn_open(&nd, NULL, FREAD, 0); 1136 if (error == 0) { 1137 type = nd.nl_open_vp->v_type; 1138 if (type == VREG) { 1139 nlookup_done(&nd); 1140 return (result); 1141 } 1142 } 1143 nlookup_done(&nd); 1144 kfree(result, M_LINKER); 1145 1146 if (*ep == 0) 1147 break; 1148 cp = ep + 1; 1149 } 1150 return(NULL); 1151 } 1152