43#endif 44#include <vm/vm_param.h> 45#include <sys/lock.h> 46#ifdef SPARSE_MAPPING 47#include <vm/vm_object.h> 48#include <vm/vm_kern.h> 49#include <vm/vm_extern.h> 50#endif 51#include <vm/pmap.h> 52#include <vm/vm_map.h> 53 54static int link_elf_load_module(const char*, linker_file_t*); 55static int link_elf_load_file(const char*, linker_file_t*); 56static int link_elf_lookup_symbol(linker_file_t, const char*, 57 c_linker_sym_t*); 58static int link_elf_symbol_values(linker_file_t, c_linker_sym_t, linker_symval_t*); 59static int link_elf_search_symbol(linker_file_t, caddr_t value, 60 c_linker_sym_t* sym, long* diffp); 61 62static void link_elf_unload_file(linker_file_t); 63static void link_elf_unload_module(linker_file_t); 64 65static struct linker_class_ops link_elf_class_ops = { 66 link_elf_load_module, 67}; 68 69static struct linker_file_ops link_elf_file_ops = { 70 link_elf_lookup_symbol, 71 link_elf_symbol_values, 72 link_elf_search_symbol, 73 link_elf_unload_file, 74}; 75 76static struct linker_file_ops link_elf_module_ops = { 77 link_elf_lookup_symbol, 78 link_elf_symbol_values, 79 link_elf_search_symbol, 80 link_elf_unload_module, 81}; 82typedef struct elf_file { 83 caddr_t address; /* Relocation address */ 84#ifdef SPARSE_MAPPING 85 vm_object_t object; /* VM object to hold file pages */ 86#endif 87 const Elf_Dyn* dynamic; /* Symbol table etc. */ 88 Elf_Off nbuckets; /* DT_HASH info */ 89 Elf_Off nchains; 90 const Elf_Off* buckets; 91 const Elf_Off* chains; 92 caddr_t hash; 93 caddr_t strtab; /* DT_STRTAB */ 94 int strsz; /* DT_STRSZ */ 95 const Elf_Sym* symtab; /* DT_SYMTAB */ 96 Elf_Addr* got; /* DT_PLTGOT */ 97 const Elf_Rel* pltrel; /* DT_JMPREL */ 98 int pltrelsize; /* DT_PLTRELSZ */ 99 const Elf_Rela* pltrela; /* DT_JMPREL */ 100 int pltrelasize; /* DT_PLTRELSZ */ 101 const Elf_Rel* rel; /* DT_REL */ 102 int relsize; /* DT_RELSZ */ 103 const Elf_Rela* rela; /* DT_RELA */ 104 int relasize; /* DT_RELASZ */ 105 caddr_t modptr; 106 const Elf_Sym* ddbsymtab; /* The symbol table we are using */ 107 long ddbsymcnt; /* Number of symbols */ 108 caddr_t ddbstrtab; /* String table */ 109 long ddbstrcnt; /* number of bytes in string table */ 110 caddr_t symbase; /* malloc'ed symbold base */ 111 caddr_t strbase; /* malloc'ed string base */ 112} *elf_file_t; 113 114static int parse_dynamic(linker_file_t lf); 115static int load_dependancies(linker_file_t lf); 116static int relocate_file(linker_file_t lf); 117static int parse_module_symbols(linker_file_t lf); 118 119/* 120 * The kernel symbol table starts here. 121 */ 122extern struct _dynamic _DYNAMIC; 123 124static void 125link_elf_init(void* arg) 126{ 127#ifdef __ELF__ 128 Elf_Dyn *dp; 129 caddr_t modptr, baseptr, sizeptr; 130 elf_file_t ef; 131 char *modname; 132#endif 133 134#if ELF_TARG_CLASS == ELFCLASS32 135 linker_add_class("elf32", NULL, &link_elf_class_ops); 136#else 137 linker_add_class("elf64", NULL, &link_elf_class_ops); 138#endif 139 140#ifdef __ELF__ 141 dp = (Elf_Dyn*) &_DYNAMIC; 142 if (dp) { 143 ef = malloc(sizeof(struct elf_file), M_LINKER, M_NOWAIT); 144 if (ef == NULL) 145 panic("link_elf_init: Can't create linker structures for kernel"); 146 bzero(ef, sizeof(*ef)); 147 148 ef->address = 0; 149#ifdef SPARSE_MAPPING 150 ef->object = 0; 151#endif 152 ef->dynamic = dp; 153 modname = NULL; 154 modptr = preload_search_by_type("elf kernel"); 155 if (modptr) 156 modname = (char *)preload_search_info(modptr, MODINFO_NAME); 157 if (modname == NULL) 158 modname = "kernel"; 159 linker_kernel_file = linker_make_file(modname, ef, &link_elf_file_ops); 160 if (linker_kernel_file == NULL) 161 panic("link_elf_init: Can't create linker structures for kernel"); 162 parse_dynamic(linker_kernel_file); 163 linker_kernel_file->address = (caddr_t) KERNBASE; 164 linker_kernel_file->size = -(intptr_t)linker_kernel_file->address; 165 166 if (modptr) { 167 ef->modptr = modptr; 168 baseptr = preload_search_info(modptr, MODINFO_ADDR); 169 if (baseptr) 170 linker_kernel_file->address = *(caddr_t *)baseptr; 171 sizeptr = preload_search_info(modptr, MODINFO_SIZE); 172 if (sizeptr) 173 linker_kernel_file->size = *(size_t *)sizeptr; 174 } 175 (void)parse_module_symbols(linker_kernel_file); 176 linker_current_file = linker_kernel_file; 177 linker_kernel_file->flags |= LINKER_FILE_LINKED; 178 } 179#endif 180} 181 182SYSINIT(link_elf, SI_SUB_KLD, SI_ORDER_SECOND, link_elf_init, 0); 183 184static int 185parse_module_symbols(linker_file_t lf) 186{ 187 elf_file_t ef = lf->priv; 188 caddr_t pointer; 189 caddr_t ssym, esym, base; 190 caddr_t strtab; 191 int strcnt; 192 Elf_Sym* symtab; 193 int symcnt; 194 195 if (ef->modptr == NULL) 196 return 0; 197 pointer = preload_search_info(ef->modptr, MODINFO_METADATA|MODINFOMD_SSYM); 198 if (pointer == NULL) 199 return 0; 200 ssym = *(caddr_t *)pointer; 201 pointer = preload_search_info(ef->modptr, MODINFO_METADATA|MODINFOMD_ESYM); 202 if (pointer == NULL) 203 return 0; 204 esym = *(caddr_t *)pointer; 205 206 base = ssym; 207 208 symcnt = *(long *)base; 209 base += sizeof(long); 210 symtab = (Elf_Sym *)base; 211 base += roundup(symcnt, sizeof(long)); 212 213 if (base > esym || base < ssym) { 214 printf("Symbols are corrupt!\n"); 215 return EINVAL; 216 } 217 218 strcnt = *(long *)base; 219 base += sizeof(long); 220 strtab = base; 221 base += roundup(strcnt, sizeof(long)); 222 223 if (base > esym || base < ssym) { 224 printf("Symbols are corrupt!\n"); 225 return EINVAL; 226 } 227 228 ef->ddbsymtab = symtab; 229 ef->ddbsymcnt = symcnt / sizeof(Elf_Sym); 230 ef->ddbstrtab = strtab; 231 ef->ddbstrcnt = strcnt; 232 233 return 0; 234} 235 236static int 237parse_dynamic(linker_file_t lf) 238{ 239 elf_file_t ef = lf->priv; 240 const Elf_Dyn *dp; 241 int plttype = DT_REL; 242 243 for (dp = ef->dynamic; dp->d_tag != DT_NULL; dp++) { 244 switch (dp->d_tag) { 245 case DT_HASH: 246 { 247 /* From src/libexec/rtld-elf/rtld.c */ 248 const Elf_Off *hashtab = (const Elf_Off *) 249 (ef->address + dp->d_un.d_ptr); 250 ef->nbuckets = hashtab[0]; 251 ef->nchains = hashtab[1]; 252 ef->buckets = hashtab + 2; 253 ef->chains = ef->buckets + ef->nbuckets; 254 break; 255 } 256 case DT_STRTAB: 257 ef->strtab = (caddr_t) (ef->address + dp->d_un.d_ptr); 258 break; 259 case DT_STRSZ: 260 ef->strsz = dp->d_un.d_val; 261 break; 262 case DT_SYMTAB: 263 ef->symtab = (Elf_Sym*) (ef->address + dp->d_un.d_ptr); 264 break; 265 case DT_SYMENT: 266 if (dp->d_un.d_val != sizeof(Elf_Sym)) 267 return ENOEXEC; 268 break; 269 case DT_PLTGOT: 270 ef->got = (Elf_Addr *) (ef->address + dp->d_un.d_ptr); 271 break; 272 case DT_REL: 273 ef->rel = (const Elf_Rel *) (ef->address + dp->d_un.d_ptr); 274 break; 275 case DT_RELSZ: 276 ef->relsize = dp->d_un.d_val; 277 break; 278 case DT_RELENT: 279 if (dp->d_un.d_val != sizeof(Elf_Rel)) 280 return ENOEXEC; 281 break; 282 case DT_JMPREL: 283 ef->pltrel = (const Elf_Rel *) (ef->address + dp->d_un.d_ptr); 284 break; 285 case DT_PLTRELSZ: 286 ef->pltrelsize = dp->d_un.d_val; 287 break; 288 case DT_RELA: 289 ef->rela = (const Elf_Rela *) (ef->address + dp->d_un.d_ptr); 290 break; 291 case DT_RELASZ: 292 ef->relasize = dp->d_un.d_val; 293 break; 294 case DT_RELAENT: 295 if (dp->d_un.d_val != sizeof(Elf_Rela)) 296 return ENOEXEC; 297 break; 298 case DT_PLTREL: 299 plttype = dp->d_un.d_val; 300 if (plttype != DT_REL && plttype != DT_RELA) 301 return ENOEXEC; 302 break; 303 } 304 } 305 306 if (plttype == DT_RELA) { 307 ef->pltrela = (const Elf_Rela *) ef->pltrel; 308 ef->pltrel = NULL; 309 ef->pltrelasize = ef->pltrelsize; 310 ef->pltrelsize = 0; 311 } 312 313 ef->ddbsymtab = ef->symtab; 314 ef->ddbsymcnt = ef->nchains; 315 ef->ddbstrtab = ef->strtab; 316 ef->ddbstrcnt = ef->strsz; 317 318 return 0; 319} 320 321static void 322link_elf_error(const char *s) 323{ 324 printf("kldload: %s\n", s); 325} 326 327static int 328link_elf_load_module(const char *filename, linker_file_t *result) 329{ 330 caddr_t modptr, baseptr, sizeptr, dynptr; 331 char *type; 332 elf_file_t ef; 333 linker_file_t lf; 334 int error; 335 vm_offset_t dp; 336 337 /* Look to see if we have the module preloaded */ 338 modptr = preload_search_by_name(filename); 339 if (modptr == NULL) 340 return (link_elf_load_file(filename, result)); 341 342 /* It's preloaded, check we can handle it and collect information */ 343 type = (char *)preload_search_info(modptr, MODINFO_TYPE); 344 baseptr = preload_search_info(modptr, MODINFO_ADDR); 345 sizeptr = preload_search_info(modptr, MODINFO_SIZE); 346 dynptr = preload_search_info(modptr, MODINFO_METADATA|MODINFOMD_DYNAMIC); 347 if (type == NULL || strcmp(type, "elf module") != 0) 348 return (EFTYPE); 349 if (baseptr == NULL || sizeptr == NULL || dynptr == NULL) 350 return (EINVAL); 351 352 ef = malloc(sizeof(struct elf_file), M_LINKER, M_WAITOK); 353 if (ef == NULL) 354 return (ENOMEM); 355 bzero(ef, sizeof(*ef)); 356 ef->modptr = modptr; 357 ef->address = *(caddr_t *)baseptr; 358#ifdef SPARSE_MAPPING 359 ef->object = 0; 360#endif 361 dp = (vm_offset_t)ef->address + *(vm_offset_t *)dynptr; 362 ef->dynamic = (Elf_Dyn *)dp; 363 lf = linker_make_file(filename, ef, &link_elf_module_ops); 364 if (lf == NULL) { 365 free(ef, M_LINKER); 366 return ENOMEM; 367 } 368 lf->address = ef->address; 369 lf->size = *(size_t *)sizeptr; 370 371 error = parse_dynamic(lf); 372 if (error) { 373 linker_file_unload(lf); 374 return error; 375 } 376 error = load_dependancies(lf); 377 if (error) { 378 linker_file_unload(lf); 379 return error; 380 } 381 error = relocate_file(lf); 382 if (error) { 383 linker_file_unload(lf); 384 return error; 385 } 386 (void)parse_module_symbols(lf); 387 lf->flags |= LINKER_FILE_LINKED; 388 *result = lf; 389 return (0); 390} 391 392static int 393link_elf_load_file(const char* filename, linker_file_t* result) 394{ 395 struct nameidata nd; 396 struct proc* p = curproc; /* XXX */ 397 Elf_Ehdr *hdr; 398 caddr_t firstpage; 399 int nbytes, i; 400 Elf_Phdr *phdr; 401 Elf_Phdr *phlimit; 402 Elf_Phdr *segs[2]; 403 int nsegs; 404 Elf_Phdr *phdyn; 405 Elf_Phdr *phphdr; 406 caddr_t mapbase; 407 size_t mapsize; 408 Elf_Off base_offset; 409 Elf_Addr base_vaddr; 410 Elf_Addr base_vlimit; 411 int error = 0; 412 int resid; 413 elf_file_t ef; 414 linker_file_t lf; 415 char *pathname; 416 Elf_Shdr *shdr; 417 int symtabindex; 418 int symstrindex; 419 int symcnt; 420 int strcnt; 421 422 shdr = NULL; 423 lf = NULL; 424 425 pathname = linker_search_path(filename); 426 if (pathname == NULL) 427 return ENOENT; 428 429 NDINIT(&nd, LOOKUP, FOLLOW, UIO_SYSSPACE, pathname, p); 430 error = vn_open(&nd, FREAD, 0); 431 free(pathname, M_LINKER); 432 if (error) 433 return error; 434 435 /* 436 * Read the elf header from the file. 437 */ 438 firstpage = malloc(PAGE_SIZE, M_LINKER, M_WAITOK); 439 if (firstpage == NULL) { 440 error = ENOMEM; 441 goto out; 442 } 443 hdr = (Elf_Ehdr *)firstpage; 444 error = vn_rdwr(UIO_READ, nd.ni_vp, firstpage, PAGE_SIZE, 0, 445 UIO_SYSSPACE, IO_NODELOCKED, p->p_ucred, &resid, p); 446 nbytes = PAGE_SIZE - resid; 447 if (error) 448 goto out; 449 450 if (!IS_ELF(*hdr)) { 451 error = ENOEXEC; 452 goto out; 453 } 454 455 if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS 456 || hdr->e_ident[EI_DATA] != ELF_TARG_DATA) { 457 link_elf_error("Unsupported file layout"); 458 error = ENOEXEC; 459 goto out; 460 } 461 if (hdr->e_ident[EI_VERSION] != EV_CURRENT 462 || hdr->e_version != EV_CURRENT) { 463 link_elf_error("Unsupported file version"); 464 error = ENOEXEC; 465 goto out; 466 } 467 if (hdr->e_type != ET_EXEC && hdr->e_type != ET_DYN) { 468 link_elf_error("Unsupported file type"); 469 error = ENOEXEC; 470 goto out; 471 } 472 if (hdr->e_machine != ELF_TARG_MACH) { 473 link_elf_error("Unsupported machine"); 474 error = ENOEXEC; 475 goto out; 476 } 477 478 /* 479 * We rely on the program header being in the first page. This is 480 * not strictly required by the ABI specification, but it seems to 481 * always true in practice. And, it simplifies things considerably. 482 */ 483 if (!((hdr->e_phentsize == sizeof(Elf_Phdr)) && 484 (hdr->e_phoff + hdr->e_phnum*sizeof(Elf_Phdr) <= PAGE_SIZE) && 485 (hdr->e_phoff + hdr->e_phnum*sizeof(Elf_Phdr) <= nbytes))) 486 link_elf_error("Unreadable program headers"); 487 488 /* 489 * Scan the program header entries, and save key information. 490 * 491 * We rely on there being exactly two load segments, text and data, 492 * in that order. 493 */ 494 phdr = (Elf_Phdr *) (firstpage + hdr->e_phoff); 495 phlimit = phdr + hdr->e_phnum; 496 nsegs = 0; 497 phdyn = NULL; 498 phphdr = NULL; 499 while (phdr < phlimit) { 500 switch (phdr->p_type) { 501 502 case PT_LOAD: 503 if (nsegs == 2) { 504 link_elf_error("Too many sections"); 505 error = ENOEXEC; 506 goto out; 507 } 508 segs[nsegs] = phdr; 509 ++nsegs; 510 break; 511 512 case PT_PHDR: 513 phphdr = phdr; 514 break; 515 516 case PT_DYNAMIC: 517 phdyn = phdr; 518 break; 519 } 520 521 ++phdr; 522 } 523 if (phdyn == NULL) { 524 link_elf_error("Object is not dynamically-linked"); 525 error = ENOEXEC; 526 goto out; 527 } 528 529 /* 530 * Allocate the entire address space of the object, to stake out our 531 * contiguous region, and to establish the base address for relocation. 532 */ 533 base_offset = trunc_page(segs[0]->p_offset); 534 base_vaddr = trunc_page(segs[0]->p_vaddr); 535 base_vlimit = round_page(segs[1]->p_vaddr + segs[1]->p_memsz); 536 mapsize = base_vlimit - base_vaddr; 537 538 ef = malloc(sizeof(struct elf_file), M_LINKER, M_WAITOK); 539 bzero(ef, sizeof(*ef)); 540#ifdef SPARSE_MAPPING 541 ef->object = vm_object_allocate(OBJT_DEFAULT, mapsize >> PAGE_SHIFT); 542 if (ef->object == NULL) { 543 free(ef, M_LINKER); 544 error = ENOMEM; 545 goto out; 546 } 547 vm_object_reference(ef->object); 548 ef->address = (caddr_t) vm_map_min(kernel_map); 549 error = vm_map_find(kernel_map, ef->object, 0, 550 (vm_offset_t *) &ef->address, 551 mapsize, 1, 552 VM_PROT_ALL, VM_PROT_ALL, 0); 553 if (error) { 554 vm_object_deallocate(ef->object); 555 free(ef, M_LINKER); 556 goto out; 557 } 558#else 559 ef->address = malloc(mapsize, M_LINKER, M_WAITOK); 560#endif 561 mapbase = ef->address; 562 563 /* 564 * Read the text and data sections and zero the bss. 565 */ 566 for (i = 0; i < 2; i++) { 567 caddr_t segbase = mapbase + segs[i]->p_vaddr - base_vaddr; 568 error = vn_rdwr(UIO_READ, nd.ni_vp, 569 segbase, segs[i]->p_filesz, segs[i]->p_offset, 570 UIO_SYSSPACE, IO_NODELOCKED, p->p_ucred, &resid, p); 571 if (error) { 572#ifdef SPARSE_MAPPING 573 vm_map_remove(kernel_map, (vm_offset_t) ef->address, 574 (vm_offset_t) ef->address 575 + (ef->object->size << PAGE_SHIFT)); 576 vm_object_deallocate(ef->object); 577#else 578 free(ef->address, M_LINKER); 579#endif 580 free(ef, M_LINKER); 581 goto out; 582 } 583 bzero(segbase + segs[i]->p_filesz, 584 segs[i]->p_memsz - segs[i]->p_filesz); 585 586#ifdef SPARSE_MAPPING 587 /* 588 * Wire down the pages 589 */ 590 vm_map_pageable(kernel_map, 591 (vm_offset_t) segbase, 592 (vm_offset_t) segbase + segs[i]->p_memsz, 593 FALSE); 594#endif 595 } 596 597 ef->dynamic = (const Elf_Dyn *) (mapbase + phdyn->p_vaddr - base_vaddr); 598 599 lf = linker_make_file(filename, ef, &link_elf_file_ops); 600 if (lf == NULL) { 601#ifdef SPARSE_MAPPING 602 vm_map_remove(kernel_map, (vm_offset_t) ef->address, 603 (vm_offset_t) ef->address 604 + (ef->object->size << PAGE_SHIFT)); 605 vm_object_deallocate(ef->object); 606#else 607 free(ef->address, M_LINKER); 608#endif 609 free(ef, M_LINKER); 610 error = ENOMEM; 611 goto out; 612 } 613 lf->address = ef->address; 614 lf->size = mapsize; 615 616 error = parse_dynamic(lf); 617 if (error) 618 goto out; 619 error = load_dependancies(lf); 620 if (error) 621 goto out; 622 error = relocate_file(lf); 623 if (error) 624 goto out; 625 626 /* Try and load the symbol table if it's present. (you can strip it!) */ 627 nbytes = hdr->e_shnum * hdr->e_shentsize; 628 if (nbytes == 0 || hdr->e_shoff == 0) 629 goto nosyms; 630 shdr = malloc(nbytes, M_LINKER, M_WAITOK); 631 if (shdr == NULL) { 632 error = ENOMEM; 633 goto out; 634 } 635 bzero(shdr, nbytes); 636 error = vn_rdwr(UIO_READ, nd.ni_vp, 637 (caddr_t)shdr, nbytes, hdr->e_shoff, 638 UIO_SYSSPACE, IO_NODELOCKED, p->p_ucred, &resid, p); 639 if (error) 640 goto out; 641 symtabindex = -1; 642 symstrindex = -1; 643 for (i = 0; i < hdr->e_shnum; i++) { 644 if (shdr[i].sh_type == SHT_SYMTAB) { 645 symtabindex = i; 646 symstrindex = shdr[i].sh_link; 647 } 648 } 649 if (symtabindex < 0 || symstrindex < 0) 650 goto nosyms; 651 652 symcnt = shdr[symtabindex].sh_size; 653 ef->symbase = malloc(symcnt, M_LINKER, M_WAITOK); 654 strcnt = shdr[symstrindex].sh_size; 655 ef->strbase = malloc(strcnt, M_LINKER, M_WAITOK); 656 657 if (ef->symbase == NULL || ef->strbase == NULL) { 658 error = ENOMEM; 659 goto out; 660 } 661 error = vn_rdwr(UIO_READ, nd.ni_vp, 662 ef->symbase, symcnt, shdr[symtabindex].sh_offset, 663 UIO_SYSSPACE, IO_NODELOCKED, p->p_ucred, &resid, p); 664 if (error) 665 goto out; 666 error = vn_rdwr(UIO_READ, nd.ni_vp, 667 ef->strbase, strcnt, shdr[symstrindex].sh_offset, 668 UIO_SYSSPACE, IO_NODELOCKED, p->p_ucred, &resid, p); 669 if (error) 670 goto out; 671 672 ef->ddbsymcnt = symcnt / sizeof(Elf_Sym); 673 ef->ddbsymtab = (const Elf_Sym *)ef->symbase; 674 ef->ddbstrcnt = strcnt; 675 ef->ddbstrtab = ef->strbase; 676 677 lf->flags |= LINKER_FILE_LINKED; 678 679nosyms: 680 681 *result = lf; 682 683out: 684 if (error && lf) 685 linker_file_unload(lf); 686 if (shdr) 687 free(shdr, M_LINKER); 688 if (firstpage) 689 free(firstpage, M_LINKER); 690 VOP_UNLOCK(nd.ni_vp, 0, p); 691 vn_close(nd.ni_vp, FREAD, p->p_ucred, p); 692 693 return error; 694} 695 696static void 697link_elf_unload_file(linker_file_t file) 698{ 699 elf_file_t ef = file->priv; 700 701 if (ef) { 702#ifdef SPARSE_MAPPING 703 if (ef->object) { 704 vm_map_remove(kernel_map, (vm_offset_t) ef->address, 705 (vm_offset_t) ef->address 706 + (ef->object->size << PAGE_SHIFT)); 707 vm_object_deallocate(ef->object); 708 } 709#else 710 if (ef->address) 711 free(ef->address, M_LINKER); 712#endif 713 if (ef->symbase) 714 free(ef->symbase, M_LINKER); 715 if (ef->strbase) 716 free(ef->strbase, M_LINKER); 717 free(ef, M_LINKER); 718 } 719} 720 721static void 722link_elf_unload_module(linker_file_t file) 723{ 724 elf_file_t ef = file->priv; 725 726 if (ef) 727 free(ef, M_LINKER); 728 if (file->filename) 729 preload_delete_name(file->filename); 730} 731 732static int 733load_dependancies(linker_file_t lf) 734{ 735 elf_file_t ef = lf->priv; 736 linker_file_t lfdep; 737 char* name; 738 const Elf_Dyn *dp; 739 int error = 0; 740 741 /* 742 * All files are dependant on /kernel. 743 */ 744 if (linker_kernel_file) { 745 linker_kernel_file->refs++; 746 linker_file_add_dependancy(lf, linker_kernel_file); 747 } 748 749 for (dp = ef->dynamic; dp->d_tag != DT_NULL; dp++) { 750 if (dp->d_tag == DT_NEEDED) { 751 name = ef->strtab + dp->d_un.d_val; 752 753 error = linker_load_file(name, &lfdep); 754 if (error) 755 goto out; 756 error = linker_file_add_dependancy(lf, lfdep); 757 if (error) 758 goto out; 759 } 760 } 761 762out: 763 return error; 764} 765 766static const char * 767symbol_name(elf_file_t ef, Elf_Word r_info) 768{ 769 const Elf_Sym *ref; 770 771 if (ELF_R_SYM(r_info)) { 772 ref = ef->symtab + ELF_R_SYM(r_info); 773 return ef->strtab + ref->st_name; 774 } else 775 return NULL; 776} 777 778static int 779relocate_file(linker_file_t lf) 780{ 781 elf_file_t ef = lf->priv; 782 const Elf_Rel *rellim; 783 const Elf_Rel *rel; 784 const Elf_Rela *relalim; 785 const Elf_Rela *rela; 786 const char *symname; 787 788 /* Perform relocations without addend if there are any: */ 789 rel = ef->rel; 790 if (rel) { 791 rellim = (const Elf_Rel *)((const char *)ef->rel + ef->relsize); 792 while (rel < rellim) { 793 symname = symbol_name(ef, rel->r_info); 794 if (elf_reloc(lf, rel, ELF_RELOC_REL, symname)) { 795 printf("link_elf: symbol %s undefined\n", symname); 796 return ENOENT; 797 } 798 rel++; 799 } 800 } 801 802 /* Perform relocations with addend if there are any: */ 803 rela = ef->rela; 804 if (rela) { 805 relalim = (const Elf_Rela *)((const char *)ef->rela + ef->relasize); 806 while (rela < relalim) { 807 symname = symbol_name(ef, rela->r_info); 808 if (elf_reloc(lf, rela, ELF_RELOC_RELA, symname)) { 809 printf("link_elf: symbol %s undefined\n", symname); 810 return ENOENT; 811 } 812 rela++; 813 } 814 } 815 816 /* Perform PLT relocations without addend if there are any: */ 817 rel = ef->pltrel; 818 if (rel) { 819 rellim = (const Elf_Rel *)((const char *)ef->pltrel + ef->pltrelsize); 820 while (rel < rellim) { 821 symname = symbol_name(ef, rel->r_info); 822 if (elf_reloc(lf, rel, ELF_RELOC_REL, symname)) { 823 printf("link_elf: symbol %s undefined\n", symname); 824 return ENOENT; 825 } 826 rel++; 827 } 828 } 829 830 /* Perform relocations with addend if there are any: */ 831 rela = ef->pltrela; 832 if (rela) { 833 relalim = (const Elf_Rela *)((const char *)ef->pltrela + ef->pltrelasize); 834 while (rela < relalim) { 835 symname = symbol_name(ef, rela->r_info); 836 if (elf_reloc(lf, rela, ELF_RELOC_RELA, symname)) { 837 printf("link_elf: symbol %s undefined\n", symname); 838 return ENOENT; 839 } 840 rela++; 841 } 842 } 843 844 return 0; 845} 846 847/* 848 * Hash function for symbol table lookup. Don't even think about changing 849 * this. It is specified by the System V ABI. 850 */ 851static unsigned long 852elf_hash(const char *name) 853{ 854 const unsigned char *p = (const unsigned char *) name; 855 unsigned long h = 0; 856 unsigned long g; 857 858 while (*p != '\0') { 859 h = (h << 4) + *p++; 860 if ((g = h & 0xf0000000) != 0) 861 h ^= g >> 24; 862 h &= ~g; 863 } 864 return h; 865} 866 867int 868link_elf_lookup_symbol(linker_file_t lf, const char* name, c_linker_sym_t* sym) 869{ 870 elf_file_t ef = lf->priv; 871 unsigned long symnum; 872 const Elf_Sym* symp; 873 const char *strp; 874 unsigned long hash; 875 int i; 876 877 /* First, search hashed global symbols */ 878 hash = elf_hash(name); 879 symnum = ef->buckets[hash % ef->nbuckets]; 880 881 while (symnum != STN_UNDEF) { 882 if (symnum >= ef->nchains) { 883 printf("link_elf_lookup_symbol: corrupt symbol table\n"); 884 return ENOENT; 885 } 886 887 symp = ef->symtab + symnum; 888 if (symp->st_name == 0) { 889 printf("link_elf_lookup_symbol: corrupt symbol table\n"); 890 return ENOENT; 891 } 892 893 strp = ef->strtab + symp->st_name; 894 895 if (strcmp(name, strp) == 0) { 896 if (symp->st_shndx != SHN_UNDEF || 897 (symp->st_value != 0 && 898 ELF_ST_TYPE(symp->st_info) == STT_FUNC)) { 899 *sym = (c_linker_sym_t) symp; 900 return 0; 901 } else 902 return ENOENT; 903 } 904 905 symnum = ef->chains[symnum]; 906 } 907 908 /* If we have not found it, look at the full table (if loaded) */ 909 if (ef->symtab == ef->ddbsymtab) 910 return ENOENT; 911 912 /* Exhaustive search */ 913 for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 914 strp = ef->ddbstrtab + symp->st_name; 915 if (strcmp(name, strp) == 0) { 916 if (symp->st_shndx != SHN_UNDEF || 917 (symp->st_value != 0 && 918 ELF_ST_TYPE(symp->st_info) == STT_FUNC)) { 919 *sym = (c_linker_sym_t) symp; 920 return 0; 921 } else 922 return ENOENT; 923 } 924 } 925 926 return ENOENT; 927} 928 929static int 930link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym, linker_symval_t* symval) 931{ 932 elf_file_t ef = lf->priv; 933 const Elf_Sym* es = (const Elf_Sym*) sym; 934 935 if (es >= ef->symtab && ((es - ef->symtab) < ef->nchains)) { 936 symval->name = ef->strtab + es->st_name; 937 symval->value = (caddr_t) ef->address + es->st_value; 938 symval->size = es->st_size; 939 return 0; 940 } 941 if (ef->symtab == ef->ddbsymtab) 942 return ENOENT; 943 if (es >= ef->ddbsymtab && ((es - ef->ddbsymtab) < ef->ddbsymcnt)) { 944 symval->name = ef->ddbstrtab + es->st_name; 945 symval->value = (caddr_t) ef->address + es->st_value; 946 symval->size = es->st_size; 947 return 0; 948 } 949 return ENOENT; 950} 951 952static int 953link_elf_search_symbol(linker_file_t lf, caddr_t value, 954 c_linker_sym_t* sym, long* diffp) 955{ 956 elf_file_t ef = lf->priv; 957 u_long off = (uintptr_t) value; 958 u_long diff = off; 959 const Elf_Sym* es; 960 const Elf_Sym* best = 0; 961 int i; 962 963 for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) { 964 if (es->st_name == 0) 965 continue; 966 if (off >= es->st_value) { 967 if (off - es->st_value < diff) { 968 diff = off - es->st_value; 969 best = es; 970 if (diff == 0) 971 break; 972 } else if (off - es->st_value == diff) { 973 best = es; 974 } 975 } 976 } 977 if (best == 0) 978 *diffp = off; 979 else 980 *diffp = diff; 981 *sym = (c_linker_sym_t) best; 982 983 return 0; 984}
| 42#endif 43#include <vm/vm_param.h> 44#include <sys/lock.h> 45#ifdef SPARSE_MAPPING 46#include <vm/vm_object.h> 47#include <vm/vm_kern.h> 48#include <vm/vm_extern.h> 49#endif 50#include <vm/pmap.h> 51#include <vm/vm_map.h> 52 53static int link_elf_load_module(const char*, linker_file_t*); 54static int link_elf_load_file(const char*, linker_file_t*); 55static int link_elf_lookup_symbol(linker_file_t, const char*, 56 c_linker_sym_t*); 57static int link_elf_symbol_values(linker_file_t, c_linker_sym_t, linker_symval_t*); 58static int link_elf_search_symbol(linker_file_t, caddr_t value, 59 c_linker_sym_t* sym, long* diffp); 60 61static void link_elf_unload_file(linker_file_t); 62static void link_elf_unload_module(linker_file_t); 63 64static struct linker_class_ops link_elf_class_ops = { 65 link_elf_load_module, 66}; 67 68static struct linker_file_ops link_elf_file_ops = { 69 link_elf_lookup_symbol, 70 link_elf_symbol_values, 71 link_elf_search_symbol, 72 link_elf_unload_file, 73}; 74 75static struct linker_file_ops link_elf_module_ops = { 76 link_elf_lookup_symbol, 77 link_elf_symbol_values, 78 link_elf_search_symbol, 79 link_elf_unload_module, 80}; 81typedef struct elf_file { 82 caddr_t address; /* Relocation address */ 83#ifdef SPARSE_MAPPING 84 vm_object_t object; /* VM object to hold file pages */ 85#endif 86 const Elf_Dyn* dynamic; /* Symbol table etc. */ 87 Elf_Off nbuckets; /* DT_HASH info */ 88 Elf_Off nchains; 89 const Elf_Off* buckets; 90 const Elf_Off* chains; 91 caddr_t hash; 92 caddr_t strtab; /* DT_STRTAB */ 93 int strsz; /* DT_STRSZ */ 94 const Elf_Sym* symtab; /* DT_SYMTAB */ 95 Elf_Addr* got; /* DT_PLTGOT */ 96 const Elf_Rel* pltrel; /* DT_JMPREL */ 97 int pltrelsize; /* DT_PLTRELSZ */ 98 const Elf_Rela* pltrela; /* DT_JMPREL */ 99 int pltrelasize; /* DT_PLTRELSZ */ 100 const Elf_Rel* rel; /* DT_REL */ 101 int relsize; /* DT_RELSZ */ 102 const Elf_Rela* rela; /* DT_RELA */ 103 int relasize; /* DT_RELASZ */ 104 caddr_t modptr; 105 const Elf_Sym* ddbsymtab; /* The symbol table we are using */ 106 long ddbsymcnt; /* Number of symbols */ 107 caddr_t ddbstrtab; /* String table */ 108 long ddbstrcnt; /* number of bytes in string table */ 109 caddr_t symbase; /* malloc'ed symbold base */ 110 caddr_t strbase; /* malloc'ed string base */ 111} *elf_file_t; 112 113static int parse_dynamic(linker_file_t lf); 114static int load_dependancies(linker_file_t lf); 115static int relocate_file(linker_file_t lf); 116static int parse_module_symbols(linker_file_t lf); 117 118/* 119 * The kernel symbol table starts here. 120 */ 121extern struct _dynamic _DYNAMIC; 122 123static void 124link_elf_init(void* arg) 125{ 126#ifdef __ELF__ 127 Elf_Dyn *dp; 128 caddr_t modptr, baseptr, sizeptr; 129 elf_file_t ef; 130 char *modname; 131#endif 132 133#if ELF_TARG_CLASS == ELFCLASS32 134 linker_add_class("elf32", NULL, &link_elf_class_ops); 135#else 136 linker_add_class("elf64", NULL, &link_elf_class_ops); 137#endif 138 139#ifdef __ELF__ 140 dp = (Elf_Dyn*) &_DYNAMIC; 141 if (dp) { 142 ef = malloc(sizeof(struct elf_file), M_LINKER, M_NOWAIT); 143 if (ef == NULL) 144 panic("link_elf_init: Can't create linker structures for kernel"); 145 bzero(ef, sizeof(*ef)); 146 147 ef->address = 0; 148#ifdef SPARSE_MAPPING 149 ef->object = 0; 150#endif 151 ef->dynamic = dp; 152 modname = NULL; 153 modptr = preload_search_by_type("elf kernel"); 154 if (modptr) 155 modname = (char *)preload_search_info(modptr, MODINFO_NAME); 156 if (modname == NULL) 157 modname = "kernel"; 158 linker_kernel_file = linker_make_file(modname, ef, &link_elf_file_ops); 159 if (linker_kernel_file == NULL) 160 panic("link_elf_init: Can't create linker structures for kernel"); 161 parse_dynamic(linker_kernel_file); 162 linker_kernel_file->address = (caddr_t) KERNBASE; 163 linker_kernel_file->size = -(intptr_t)linker_kernel_file->address; 164 165 if (modptr) { 166 ef->modptr = modptr; 167 baseptr = preload_search_info(modptr, MODINFO_ADDR); 168 if (baseptr) 169 linker_kernel_file->address = *(caddr_t *)baseptr; 170 sizeptr = preload_search_info(modptr, MODINFO_SIZE); 171 if (sizeptr) 172 linker_kernel_file->size = *(size_t *)sizeptr; 173 } 174 (void)parse_module_symbols(linker_kernel_file); 175 linker_current_file = linker_kernel_file; 176 linker_kernel_file->flags |= LINKER_FILE_LINKED; 177 } 178#endif 179} 180 181SYSINIT(link_elf, SI_SUB_KLD, SI_ORDER_SECOND, link_elf_init, 0); 182 183static int 184parse_module_symbols(linker_file_t lf) 185{ 186 elf_file_t ef = lf->priv; 187 caddr_t pointer; 188 caddr_t ssym, esym, base; 189 caddr_t strtab; 190 int strcnt; 191 Elf_Sym* symtab; 192 int symcnt; 193 194 if (ef->modptr == NULL) 195 return 0; 196 pointer = preload_search_info(ef->modptr, MODINFO_METADATA|MODINFOMD_SSYM); 197 if (pointer == NULL) 198 return 0; 199 ssym = *(caddr_t *)pointer; 200 pointer = preload_search_info(ef->modptr, MODINFO_METADATA|MODINFOMD_ESYM); 201 if (pointer == NULL) 202 return 0; 203 esym = *(caddr_t *)pointer; 204 205 base = ssym; 206 207 symcnt = *(long *)base; 208 base += sizeof(long); 209 symtab = (Elf_Sym *)base; 210 base += roundup(symcnt, sizeof(long)); 211 212 if (base > esym || base < ssym) { 213 printf("Symbols are corrupt!\n"); 214 return EINVAL; 215 } 216 217 strcnt = *(long *)base; 218 base += sizeof(long); 219 strtab = base; 220 base += roundup(strcnt, sizeof(long)); 221 222 if (base > esym || base < ssym) { 223 printf("Symbols are corrupt!\n"); 224 return EINVAL; 225 } 226 227 ef->ddbsymtab = symtab; 228 ef->ddbsymcnt = symcnt / sizeof(Elf_Sym); 229 ef->ddbstrtab = strtab; 230 ef->ddbstrcnt = strcnt; 231 232 return 0; 233} 234 235static int 236parse_dynamic(linker_file_t lf) 237{ 238 elf_file_t ef = lf->priv; 239 const Elf_Dyn *dp; 240 int plttype = DT_REL; 241 242 for (dp = ef->dynamic; dp->d_tag != DT_NULL; dp++) { 243 switch (dp->d_tag) { 244 case DT_HASH: 245 { 246 /* From src/libexec/rtld-elf/rtld.c */ 247 const Elf_Off *hashtab = (const Elf_Off *) 248 (ef->address + dp->d_un.d_ptr); 249 ef->nbuckets = hashtab[0]; 250 ef->nchains = hashtab[1]; 251 ef->buckets = hashtab + 2; 252 ef->chains = ef->buckets + ef->nbuckets; 253 break; 254 } 255 case DT_STRTAB: 256 ef->strtab = (caddr_t) (ef->address + dp->d_un.d_ptr); 257 break; 258 case DT_STRSZ: 259 ef->strsz = dp->d_un.d_val; 260 break; 261 case DT_SYMTAB: 262 ef->symtab = (Elf_Sym*) (ef->address + dp->d_un.d_ptr); 263 break; 264 case DT_SYMENT: 265 if (dp->d_un.d_val != sizeof(Elf_Sym)) 266 return ENOEXEC; 267 break; 268 case DT_PLTGOT: 269 ef->got = (Elf_Addr *) (ef->address + dp->d_un.d_ptr); 270 break; 271 case DT_REL: 272 ef->rel = (const Elf_Rel *) (ef->address + dp->d_un.d_ptr); 273 break; 274 case DT_RELSZ: 275 ef->relsize = dp->d_un.d_val; 276 break; 277 case DT_RELENT: 278 if (dp->d_un.d_val != sizeof(Elf_Rel)) 279 return ENOEXEC; 280 break; 281 case DT_JMPREL: 282 ef->pltrel = (const Elf_Rel *) (ef->address + dp->d_un.d_ptr); 283 break; 284 case DT_PLTRELSZ: 285 ef->pltrelsize = dp->d_un.d_val; 286 break; 287 case DT_RELA: 288 ef->rela = (const Elf_Rela *) (ef->address + dp->d_un.d_ptr); 289 break; 290 case DT_RELASZ: 291 ef->relasize = dp->d_un.d_val; 292 break; 293 case DT_RELAENT: 294 if (dp->d_un.d_val != sizeof(Elf_Rela)) 295 return ENOEXEC; 296 break; 297 case DT_PLTREL: 298 plttype = dp->d_un.d_val; 299 if (plttype != DT_REL && plttype != DT_RELA) 300 return ENOEXEC; 301 break; 302 } 303 } 304 305 if (plttype == DT_RELA) { 306 ef->pltrela = (const Elf_Rela *) ef->pltrel; 307 ef->pltrel = NULL; 308 ef->pltrelasize = ef->pltrelsize; 309 ef->pltrelsize = 0; 310 } 311 312 ef->ddbsymtab = ef->symtab; 313 ef->ddbsymcnt = ef->nchains; 314 ef->ddbstrtab = ef->strtab; 315 ef->ddbstrcnt = ef->strsz; 316 317 return 0; 318} 319 320static void 321link_elf_error(const char *s) 322{ 323 printf("kldload: %s\n", s); 324} 325 326static int 327link_elf_load_module(const char *filename, linker_file_t *result) 328{ 329 caddr_t modptr, baseptr, sizeptr, dynptr; 330 char *type; 331 elf_file_t ef; 332 linker_file_t lf; 333 int error; 334 vm_offset_t dp; 335 336 /* Look to see if we have the module preloaded */ 337 modptr = preload_search_by_name(filename); 338 if (modptr == NULL) 339 return (link_elf_load_file(filename, result)); 340 341 /* It's preloaded, check we can handle it and collect information */ 342 type = (char *)preload_search_info(modptr, MODINFO_TYPE); 343 baseptr = preload_search_info(modptr, MODINFO_ADDR); 344 sizeptr = preload_search_info(modptr, MODINFO_SIZE); 345 dynptr = preload_search_info(modptr, MODINFO_METADATA|MODINFOMD_DYNAMIC); 346 if (type == NULL || strcmp(type, "elf module") != 0) 347 return (EFTYPE); 348 if (baseptr == NULL || sizeptr == NULL || dynptr == NULL) 349 return (EINVAL); 350 351 ef = malloc(sizeof(struct elf_file), M_LINKER, M_WAITOK); 352 if (ef == NULL) 353 return (ENOMEM); 354 bzero(ef, sizeof(*ef)); 355 ef->modptr = modptr; 356 ef->address = *(caddr_t *)baseptr; 357#ifdef SPARSE_MAPPING 358 ef->object = 0; 359#endif 360 dp = (vm_offset_t)ef->address + *(vm_offset_t *)dynptr; 361 ef->dynamic = (Elf_Dyn *)dp; 362 lf = linker_make_file(filename, ef, &link_elf_module_ops); 363 if (lf == NULL) { 364 free(ef, M_LINKER); 365 return ENOMEM; 366 } 367 lf->address = ef->address; 368 lf->size = *(size_t *)sizeptr; 369 370 error = parse_dynamic(lf); 371 if (error) { 372 linker_file_unload(lf); 373 return error; 374 } 375 error = load_dependancies(lf); 376 if (error) { 377 linker_file_unload(lf); 378 return error; 379 } 380 error = relocate_file(lf); 381 if (error) { 382 linker_file_unload(lf); 383 return error; 384 } 385 (void)parse_module_symbols(lf); 386 lf->flags |= LINKER_FILE_LINKED; 387 *result = lf; 388 return (0); 389} 390 391static int 392link_elf_load_file(const char* filename, linker_file_t* result) 393{ 394 struct nameidata nd; 395 struct proc* p = curproc; /* XXX */ 396 Elf_Ehdr *hdr; 397 caddr_t firstpage; 398 int nbytes, i; 399 Elf_Phdr *phdr; 400 Elf_Phdr *phlimit; 401 Elf_Phdr *segs[2]; 402 int nsegs; 403 Elf_Phdr *phdyn; 404 Elf_Phdr *phphdr; 405 caddr_t mapbase; 406 size_t mapsize; 407 Elf_Off base_offset; 408 Elf_Addr base_vaddr; 409 Elf_Addr base_vlimit; 410 int error = 0; 411 int resid; 412 elf_file_t ef; 413 linker_file_t lf; 414 char *pathname; 415 Elf_Shdr *shdr; 416 int symtabindex; 417 int symstrindex; 418 int symcnt; 419 int strcnt; 420 421 shdr = NULL; 422 lf = NULL; 423 424 pathname = linker_search_path(filename); 425 if (pathname == NULL) 426 return ENOENT; 427 428 NDINIT(&nd, LOOKUP, FOLLOW, UIO_SYSSPACE, pathname, p); 429 error = vn_open(&nd, FREAD, 0); 430 free(pathname, M_LINKER); 431 if (error) 432 return error; 433 434 /* 435 * Read the elf header from the file. 436 */ 437 firstpage = malloc(PAGE_SIZE, M_LINKER, M_WAITOK); 438 if (firstpage == NULL) { 439 error = ENOMEM; 440 goto out; 441 } 442 hdr = (Elf_Ehdr *)firstpage; 443 error = vn_rdwr(UIO_READ, nd.ni_vp, firstpage, PAGE_SIZE, 0, 444 UIO_SYSSPACE, IO_NODELOCKED, p->p_ucred, &resid, p); 445 nbytes = PAGE_SIZE - resid; 446 if (error) 447 goto out; 448 449 if (!IS_ELF(*hdr)) { 450 error = ENOEXEC; 451 goto out; 452 } 453 454 if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS 455 || hdr->e_ident[EI_DATA] != ELF_TARG_DATA) { 456 link_elf_error("Unsupported file layout"); 457 error = ENOEXEC; 458 goto out; 459 } 460 if (hdr->e_ident[EI_VERSION] != EV_CURRENT 461 || hdr->e_version != EV_CURRENT) { 462 link_elf_error("Unsupported file version"); 463 error = ENOEXEC; 464 goto out; 465 } 466 if (hdr->e_type != ET_EXEC && hdr->e_type != ET_DYN) { 467 link_elf_error("Unsupported file type"); 468 error = ENOEXEC; 469 goto out; 470 } 471 if (hdr->e_machine != ELF_TARG_MACH) { 472 link_elf_error("Unsupported machine"); 473 error = ENOEXEC; 474 goto out; 475 } 476 477 /* 478 * We rely on the program header being in the first page. This is 479 * not strictly required by the ABI specification, but it seems to 480 * always true in practice. And, it simplifies things considerably. 481 */ 482 if (!((hdr->e_phentsize == sizeof(Elf_Phdr)) && 483 (hdr->e_phoff + hdr->e_phnum*sizeof(Elf_Phdr) <= PAGE_SIZE) && 484 (hdr->e_phoff + hdr->e_phnum*sizeof(Elf_Phdr) <= nbytes))) 485 link_elf_error("Unreadable program headers"); 486 487 /* 488 * Scan the program header entries, and save key information. 489 * 490 * We rely on there being exactly two load segments, text and data, 491 * in that order. 492 */ 493 phdr = (Elf_Phdr *) (firstpage + hdr->e_phoff); 494 phlimit = phdr + hdr->e_phnum; 495 nsegs = 0; 496 phdyn = NULL; 497 phphdr = NULL; 498 while (phdr < phlimit) { 499 switch (phdr->p_type) { 500 501 case PT_LOAD: 502 if (nsegs == 2) { 503 link_elf_error("Too many sections"); 504 error = ENOEXEC; 505 goto out; 506 } 507 segs[nsegs] = phdr; 508 ++nsegs; 509 break; 510 511 case PT_PHDR: 512 phphdr = phdr; 513 break; 514 515 case PT_DYNAMIC: 516 phdyn = phdr; 517 break; 518 } 519 520 ++phdr; 521 } 522 if (phdyn == NULL) { 523 link_elf_error("Object is not dynamically-linked"); 524 error = ENOEXEC; 525 goto out; 526 } 527 528 /* 529 * Allocate the entire address space of the object, to stake out our 530 * contiguous region, and to establish the base address for relocation. 531 */ 532 base_offset = trunc_page(segs[0]->p_offset); 533 base_vaddr = trunc_page(segs[0]->p_vaddr); 534 base_vlimit = round_page(segs[1]->p_vaddr + segs[1]->p_memsz); 535 mapsize = base_vlimit - base_vaddr; 536 537 ef = malloc(sizeof(struct elf_file), M_LINKER, M_WAITOK); 538 bzero(ef, sizeof(*ef)); 539#ifdef SPARSE_MAPPING 540 ef->object = vm_object_allocate(OBJT_DEFAULT, mapsize >> PAGE_SHIFT); 541 if (ef->object == NULL) { 542 free(ef, M_LINKER); 543 error = ENOMEM; 544 goto out; 545 } 546 vm_object_reference(ef->object); 547 ef->address = (caddr_t) vm_map_min(kernel_map); 548 error = vm_map_find(kernel_map, ef->object, 0, 549 (vm_offset_t *) &ef->address, 550 mapsize, 1, 551 VM_PROT_ALL, VM_PROT_ALL, 0); 552 if (error) { 553 vm_object_deallocate(ef->object); 554 free(ef, M_LINKER); 555 goto out; 556 } 557#else 558 ef->address = malloc(mapsize, M_LINKER, M_WAITOK); 559#endif 560 mapbase = ef->address; 561 562 /* 563 * Read the text and data sections and zero the bss. 564 */ 565 for (i = 0; i < 2; i++) { 566 caddr_t segbase = mapbase + segs[i]->p_vaddr - base_vaddr; 567 error = vn_rdwr(UIO_READ, nd.ni_vp, 568 segbase, segs[i]->p_filesz, segs[i]->p_offset, 569 UIO_SYSSPACE, IO_NODELOCKED, p->p_ucred, &resid, p); 570 if (error) { 571#ifdef SPARSE_MAPPING 572 vm_map_remove(kernel_map, (vm_offset_t) ef->address, 573 (vm_offset_t) ef->address 574 + (ef->object->size << PAGE_SHIFT)); 575 vm_object_deallocate(ef->object); 576#else 577 free(ef->address, M_LINKER); 578#endif 579 free(ef, M_LINKER); 580 goto out; 581 } 582 bzero(segbase + segs[i]->p_filesz, 583 segs[i]->p_memsz - segs[i]->p_filesz); 584 585#ifdef SPARSE_MAPPING 586 /* 587 * Wire down the pages 588 */ 589 vm_map_pageable(kernel_map, 590 (vm_offset_t) segbase, 591 (vm_offset_t) segbase + segs[i]->p_memsz, 592 FALSE); 593#endif 594 } 595 596 ef->dynamic = (const Elf_Dyn *) (mapbase + phdyn->p_vaddr - base_vaddr); 597 598 lf = linker_make_file(filename, ef, &link_elf_file_ops); 599 if (lf == NULL) { 600#ifdef SPARSE_MAPPING 601 vm_map_remove(kernel_map, (vm_offset_t) ef->address, 602 (vm_offset_t) ef->address 603 + (ef->object->size << PAGE_SHIFT)); 604 vm_object_deallocate(ef->object); 605#else 606 free(ef->address, M_LINKER); 607#endif 608 free(ef, M_LINKER); 609 error = ENOMEM; 610 goto out; 611 } 612 lf->address = ef->address; 613 lf->size = mapsize; 614 615 error = parse_dynamic(lf); 616 if (error) 617 goto out; 618 error = load_dependancies(lf); 619 if (error) 620 goto out; 621 error = relocate_file(lf); 622 if (error) 623 goto out; 624 625 /* Try and load the symbol table if it's present. (you can strip it!) */ 626 nbytes = hdr->e_shnum * hdr->e_shentsize; 627 if (nbytes == 0 || hdr->e_shoff == 0) 628 goto nosyms; 629 shdr = malloc(nbytes, M_LINKER, M_WAITOK); 630 if (shdr == NULL) { 631 error = ENOMEM; 632 goto out; 633 } 634 bzero(shdr, nbytes); 635 error = vn_rdwr(UIO_READ, nd.ni_vp, 636 (caddr_t)shdr, nbytes, hdr->e_shoff, 637 UIO_SYSSPACE, IO_NODELOCKED, p->p_ucred, &resid, p); 638 if (error) 639 goto out; 640 symtabindex = -1; 641 symstrindex = -1; 642 for (i = 0; i < hdr->e_shnum; i++) { 643 if (shdr[i].sh_type == SHT_SYMTAB) { 644 symtabindex = i; 645 symstrindex = shdr[i].sh_link; 646 } 647 } 648 if (symtabindex < 0 || symstrindex < 0) 649 goto nosyms; 650 651 symcnt = shdr[symtabindex].sh_size; 652 ef->symbase = malloc(symcnt, M_LINKER, M_WAITOK); 653 strcnt = shdr[symstrindex].sh_size; 654 ef->strbase = malloc(strcnt, M_LINKER, M_WAITOK); 655 656 if (ef->symbase == NULL || ef->strbase == NULL) { 657 error = ENOMEM; 658 goto out; 659 } 660 error = vn_rdwr(UIO_READ, nd.ni_vp, 661 ef->symbase, symcnt, shdr[symtabindex].sh_offset, 662 UIO_SYSSPACE, IO_NODELOCKED, p->p_ucred, &resid, p); 663 if (error) 664 goto out; 665 error = vn_rdwr(UIO_READ, nd.ni_vp, 666 ef->strbase, strcnt, shdr[symstrindex].sh_offset, 667 UIO_SYSSPACE, IO_NODELOCKED, p->p_ucred, &resid, p); 668 if (error) 669 goto out; 670 671 ef->ddbsymcnt = symcnt / sizeof(Elf_Sym); 672 ef->ddbsymtab = (const Elf_Sym *)ef->symbase; 673 ef->ddbstrcnt = strcnt; 674 ef->ddbstrtab = ef->strbase; 675 676 lf->flags |= LINKER_FILE_LINKED; 677 678nosyms: 679 680 *result = lf; 681 682out: 683 if (error && lf) 684 linker_file_unload(lf); 685 if (shdr) 686 free(shdr, M_LINKER); 687 if (firstpage) 688 free(firstpage, M_LINKER); 689 VOP_UNLOCK(nd.ni_vp, 0, p); 690 vn_close(nd.ni_vp, FREAD, p->p_ucred, p); 691 692 return error; 693} 694 695static void 696link_elf_unload_file(linker_file_t file) 697{ 698 elf_file_t ef = file->priv; 699 700 if (ef) { 701#ifdef SPARSE_MAPPING 702 if (ef->object) { 703 vm_map_remove(kernel_map, (vm_offset_t) ef->address, 704 (vm_offset_t) ef->address 705 + (ef->object->size << PAGE_SHIFT)); 706 vm_object_deallocate(ef->object); 707 } 708#else 709 if (ef->address) 710 free(ef->address, M_LINKER); 711#endif 712 if (ef->symbase) 713 free(ef->symbase, M_LINKER); 714 if (ef->strbase) 715 free(ef->strbase, M_LINKER); 716 free(ef, M_LINKER); 717 } 718} 719 720static void 721link_elf_unload_module(linker_file_t file) 722{ 723 elf_file_t ef = file->priv; 724 725 if (ef) 726 free(ef, M_LINKER); 727 if (file->filename) 728 preload_delete_name(file->filename); 729} 730 731static int 732load_dependancies(linker_file_t lf) 733{ 734 elf_file_t ef = lf->priv; 735 linker_file_t lfdep; 736 char* name; 737 const Elf_Dyn *dp; 738 int error = 0; 739 740 /* 741 * All files are dependant on /kernel. 742 */ 743 if (linker_kernel_file) { 744 linker_kernel_file->refs++; 745 linker_file_add_dependancy(lf, linker_kernel_file); 746 } 747 748 for (dp = ef->dynamic; dp->d_tag != DT_NULL; dp++) { 749 if (dp->d_tag == DT_NEEDED) { 750 name = ef->strtab + dp->d_un.d_val; 751 752 error = linker_load_file(name, &lfdep); 753 if (error) 754 goto out; 755 error = linker_file_add_dependancy(lf, lfdep); 756 if (error) 757 goto out; 758 } 759 } 760 761out: 762 return error; 763} 764 765static const char * 766symbol_name(elf_file_t ef, Elf_Word r_info) 767{ 768 const Elf_Sym *ref; 769 770 if (ELF_R_SYM(r_info)) { 771 ref = ef->symtab + ELF_R_SYM(r_info); 772 return ef->strtab + ref->st_name; 773 } else 774 return NULL; 775} 776 777static int 778relocate_file(linker_file_t lf) 779{ 780 elf_file_t ef = lf->priv; 781 const Elf_Rel *rellim; 782 const Elf_Rel *rel; 783 const Elf_Rela *relalim; 784 const Elf_Rela *rela; 785 const char *symname; 786 787 /* Perform relocations without addend if there are any: */ 788 rel = ef->rel; 789 if (rel) { 790 rellim = (const Elf_Rel *)((const char *)ef->rel + ef->relsize); 791 while (rel < rellim) { 792 symname = symbol_name(ef, rel->r_info); 793 if (elf_reloc(lf, rel, ELF_RELOC_REL, symname)) { 794 printf("link_elf: symbol %s undefined\n", symname); 795 return ENOENT; 796 } 797 rel++; 798 } 799 } 800 801 /* Perform relocations with addend if there are any: */ 802 rela = ef->rela; 803 if (rela) { 804 relalim = (const Elf_Rela *)((const char *)ef->rela + ef->relasize); 805 while (rela < relalim) { 806 symname = symbol_name(ef, rela->r_info); 807 if (elf_reloc(lf, rela, ELF_RELOC_RELA, symname)) { 808 printf("link_elf: symbol %s undefined\n", symname); 809 return ENOENT; 810 } 811 rela++; 812 } 813 } 814 815 /* Perform PLT relocations without addend if there are any: */ 816 rel = ef->pltrel; 817 if (rel) { 818 rellim = (const Elf_Rel *)((const char *)ef->pltrel + ef->pltrelsize); 819 while (rel < rellim) { 820 symname = symbol_name(ef, rel->r_info); 821 if (elf_reloc(lf, rel, ELF_RELOC_REL, symname)) { 822 printf("link_elf: symbol %s undefined\n", symname); 823 return ENOENT; 824 } 825 rel++; 826 } 827 } 828 829 /* Perform relocations with addend if there are any: */ 830 rela = ef->pltrela; 831 if (rela) { 832 relalim = (const Elf_Rela *)((const char *)ef->pltrela + ef->pltrelasize); 833 while (rela < relalim) { 834 symname = symbol_name(ef, rela->r_info); 835 if (elf_reloc(lf, rela, ELF_RELOC_RELA, symname)) { 836 printf("link_elf: symbol %s undefined\n", symname); 837 return ENOENT; 838 } 839 rela++; 840 } 841 } 842 843 return 0; 844} 845 846/* 847 * Hash function for symbol table lookup. Don't even think about changing 848 * this. It is specified by the System V ABI. 849 */ 850static unsigned long 851elf_hash(const char *name) 852{ 853 const unsigned char *p = (const unsigned char *) name; 854 unsigned long h = 0; 855 unsigned long g; 856 857 while (*p != '\0') { 858 h = (h << 4) + *p++; 859 if ((g = h & 0xf0000000) != 0) 860 h ^= g >> 24; 861 h &= ~g; 862 } 863 return h; 864} 865 866int 867link_elf_lookup_symbol(linker_file_t lf, const char* name, c_linker_sym_t* sym) 868{ 869 elf_file_t ef = lf->priv; 870 unsigned long symnum; 871 const Elf_Sym* symp; 872 const char *strp; 873 unsigned long hash; 874 int i; 875 876 /* First, search hashed global symbols */ 877 hash = elf_hash(name); 878 symnum = ef->buckets[hash % ef->nbuckets]; 879 880 while (symnum != STN_UNDEF) { 881 if (symnum >= ef->nchains) { 882 printf("link_elf_lookup_symbol: corrupt symbol table\n"); 883 return ENOENT; 884 } 885 886 symp = ef->symtab + symnum; 887 if (symp->st_name == 0) { 888 printf("link_elf_lookup_symbol: corrupt symbol table\n"); 889 return ENOENT; 890 } 891 892 strp = ef->strtab + symp->st_name; 893 894 if (strcmp(name, strp) == 0) { 895 if (symp->st_shndx != SHN_UNDEF || 896 (symp->st_value != 0 && 897 ELF_ST_TYPE(symp->st_info) == STT_FUNC)) { 898 *sym = (c_linker_sym_t) symp; 899 return 0; 900 } else 901 return ENOENT; 902 } 903 904 symnum = ef->chains[symnum]; 905 } 906 907 /* If we have not found it, look at the full table (if loaded) */ 908 if (ef->symtab == ef->ddbsymtab) 909 return ENOENT; 910 911 /* Exhaustive search */ 912 for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 913 strp = ef->ddbstrtab + symp->st_name; 914 if (strcmp(name, strp) == 0) { 915 if (symp->st_shndx != SHN_UNDEF || 916 (symp->st_value != 0 && 917 ELF_ST_TYPE(symp->st_info) == STT_FUNC)) { 918 *sym = (c_linker_sym_t) symp; 919 return 0; 920 } else 921 return ENOENT; 922 } 923 } 924 925 return ENOENT; 926} 927 928static int 929link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym, linker_symval_t* symval) 930{ 931 elf_file_t ef = lf->priv; 932 const Elf_Sym* es = (const Elf_Sym*) sym; 933 934 if (es >= ef->symtab && ((es - ef->symtab) < ef->nchains)) { 935 symval->name = ef->strtab + es->st_name; 936 symval->value = (caddr_t) ef->address + es->st_value; 937 symval->size = es->st_size; 938 return 0; 939 } 940 if (ef->symtab == ef->ddbsymtab) 941 return ENOENT; 942 if (es >= ef->ddbsymtab && ((es - ef->ddbsymtab) < ef->ddbsymcnt)) { 943 symval->name = ef->ddbstrtab + es->st_name; 944 symval->value = (caddr_t) ef->address + es->st_value; 945 symval->size = es->st_size; 946 return 0; 947 } 948 return ENOENT; 949} 950 951static int 952link_elf_search_symbol(linker_file_t lf, caddr_t value, 953 c_linker_sym_t* sym, long* diffp) 954{ 955 elf_file_t ef = lf->priv; 956 u_long off = (uintptr_t) value; 957 u_long diff = off; 958 const Elf_Sym* es; 959 const Elf_Sym* best = 0; 960 int i; 961 962 for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) { 963 if (es->st_name == 0) 964 continue; 965 if (off >= es->st_value) { 966 if (off - es->st_value < diff) { 967 diff = off - es->st_value; 968 best = es; 969 if (diff == 0) 970 break; 971 } else if (off - es->st_value == diff) { 972 best = es; 973 } 974 } 975 } 976 if (best == 0) 977 *diffp = off; 978 else 979 *diffp = diff; 980 *sym = (c_linker_sym_t) best; 981 982 return 0; 983}
|