1a4f67738SDoug Rabson /*- 2326e27d8SDoug Rabson * Copyright (c) 1998-2000 Doug Rabson 3e9eabf59SPeter Wemm * Copyright (c) 2004 Peter Wemm 4a4f67738SDoug Rabson * All rights reserved. 5a4f67738SDoug Rabson * 6a4f67738SDoug Rabson * Redistribution and use in source and binary forms, with or without 7a4f67738SDoug Rabson * modification, are permitted provided that the following conditions 8a4f67738SDoug Rabson * are met: 9a4f67738SDoug Rabson * 1. Redistributions of source code must retain the above copyright 10a4f67738SDoug Rabson * notice, this list of conditions and the following disclaimer. 11a4f67738SDoug Rabson * 2. Redistributions in binary form must reproduce the above copyright 12a4f67738SDoug Rabson * notice, this list of conditions and the following disclaimer in the 13a4f67738SDoug Rabson * documentation and/or other materials provided with the distribution. 14a4f67738SDoug Rabson * 15a4f67738SDoug Rabson * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 16a4f67738SDoug Rabson * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17a4f67738SDoug Rabson * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 18a4f67738SDoug Rabson * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 19a4f67738SDoug Rabson * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20a4f67738SDoug Rabson * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 21a4f67738SDoug Rabson * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22a4f67738SDoug Rabson * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23a4f67738SDoug Rabson * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24a4f67738SDoug Rabson * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25a4f67738SDoug Rabson * SUCH DAMAGE. 26a4f67738SDoug Rabson */ 27a4f67738SDoug Rabson 28677b542eSDavid E. O'Brien #include <sys/cdefs.h> 29677b542eSDavid E. O'Brien __FBSDID("$FreeBSD$"); 30677b542eSDavid E. O'Brien 31326e27d8SDoug Rabson #include "opt_ddb.h" 32326e27d8SDoug Rabson 33a4f67738SDoug Rabson #include <sys/param.h> 34a4f67738SDoug Rabson #include <sys/systm.h> 35fb919e4dSMark Murray #include <sys/kernel.h> 36fb919e4dSMark Murray #include <sys/lock.h> 37a4f67738SDoug Rabson #include <sys/malloc.h> 38c0b824f9SDima Dorfman #include <sys/mutex.h> 3962d615d5SJohn Baldwin #include <sys/mount.h> 40a4f67738SDoug Rabson #include <sys/proc.h> 41a4f67738SDoug Rabson #include <sys/namei.h> 42a4f67738SDoug Rabson #include <sys/fcntl.h> 43a4f67738SDoug Rabson #include <sys/vnode.h> 44a4f67738SDoug Rabson #include <sys/linker.h> 45fb919e4dSMark Murray 46a4f67738SDoug Rabson #include <machine/elf.h> 47a4f67738SDoug Rabson 48eddfbb76SRobert Watson #include <net/vnet.h> 49eddfbb76SRobert Watson 50aed55708SRobert Watson #include <security/mac/mac_framework.h> 51aed55708SRobert Watson 52fe3db7c7SDoug Rabson #include <vm/vm.h> 53fe3db7c7SDoug Rabson #include <vm/vm_param.h> 54fe3db7c7SDoug Rabson #include <vm/vm_object.h> 55fe3db7c7SDoug Rabson #include <vm/vm_kern.h> 56fe3db7c7SDoug Rabson #include <vm/vm_extern.h> 57fe3db7c7SDoug Rabson #include <vm/pmap.h> 58fe3db7c7SDoug Rabson #include <vm/vm_map.h> 59fb919e4dSMark Murray 60f9980387SPeter Wemm #include <sys/link_elf.h> 61fe3db7c7SDoug Rabson 62a2024a3eSJohn Birrell #ifdef DDB_CTF 63d9db5225SCraig Rodrigues #include <sys/zlib.h> 64a2024a3eSJohn Birrell #endif 65a2024a3eSJohn Birrell 66326e27d8SDoug Rabson #include "linker_if.h" 67a4f67738SDoug Rabson 68e9eabf59SPeter Wemm typedef struct { 69e9eabf59SPeter Wemm void *addr; 7020947801SPeter Wemm Elf_Off size; 71e9eabf59SPeter Wemm int flags; 72e9eabf59SPeter Wemm int sec; /* Original section */ 73e9eabf59SPeter Wemm char *name; 74e9eabf59SPeter Wemm } Elf_progent; 75e9eabf59SPeter Wemm 76e9eabf59SPeter Wemm typedef struct { 77e9eabf59SPeter Wemm Elf_Rel *rel; 7820947801SPeter Wemm int nrel; 79e9eabf59SPeter Wemm int sec; 80e9eabf59SPeter Wemm } Elf_relent; 81e9eabf59SPeter Wemm 82e9eabf59SPeter Wemm typedef struct { 83e9eabf59SPeter Wemm Elf_Rela *rela; 8420947801SPeter Wemm int nrela; 85e9eabf59SPeter Wemm int sec; 86e9eabf59SPeter Wemm } Elf_relaent; 87e9eabf59SPeter Wemm 88e9eabf59SPeter Wemm 89a4f67738SDoug Rabson typedef struct elf_file { 90326e27d8SDoug Rabson struct linker_file lf; /* Common fields */ 9120947801SPeter Wemm 9270b7ffeeSIan Dowse int preloaded; 93fe3db7c7SDoug Rabson caddr_t address; /* Relocation address */ 94fe3db7c7SDoug Rabson vm_object_t object; /* VM object to hold file pages */ 95e9eabf59SPeter Wemm Elf_Shdr *e_shdr; 96e9eabf59SPeter Wemm 97e9eabf59SPeter Wemm Elf_progent *progtab; 98e9eabf59SPeter Wemm int nprogtab; 99e9eabf59SPeter Wemm 100e9eabf59SPeter Wemm Elf_relaent *relatab; 101b474c780SDavid E. O'Brien int nrelatab; 102e9eabf59SPeter Wemm 103e9eabf59SPeter Wemm Elf_relent *reltab; 104b474c780SDavid E. O'Brien int nreltab; 105e9eabf59SPeter Wemm 106e9eabf59SPeter Wemm Elf_Sym *ddbsymtab; /* The symbol table we are using */ 1072d636ab0SPeter Wemm long ddbsymcnt; /* Number of symbols */ 1082d636ab0SPeter Wemm caddr_t ddbstrtab; /* String table */ 1092d636ab0SPeter Wemm long ddbstrcnt; /* number of bytes in string table */ 110e9eabf59SPeter Wemm 111e9eabf59SPeter Wemm caddr_t shstrtab; /* Section name string table */ 112e9eabf59SPeter Wemm long shstrcnt; /* number of bytes in string table */ 113e9eabf59SPeter Wemm 114a2024a3eSJohn Birrell caddr_t ctftab; /* CTF table */ 115a2024a3eSJohn Birrell long ctfcnt; /* number of bytes in CTF table */ 116a2024a3eSJohn Birrell caddr_t ctfoff; /* CTF offset table */ 117a2024a3eSJohn Birrell caddr_t typoff; /* Type offset table */ 118a2024a3eSJohn Birrell long typlen; /* Number of type entries. */ 119a2024a3eSJohn Birrell 120a4f67738SDoug Rabson } *elf_file_t; 121a4f67738SDoug Rabson 122a2024a3eSJohn Birrell #include <kern/kern_ctf.c> 123a2024a3eSJohn Birrell 12454823af2SPeter Wemm static int link_elf_link_preload(linker_class_t cls, 125326e27d8SDoug Rabson const char *, linker_file_t *); 12654823af2SPeter Wemm static int link_elf_link_preload_finish(linker_file_t); 12754823af2SPeter Wemm static int link_elf_load_file(linker_class_t, const char *, linker_file_t *); 128326e27d8SDoug Rabson static int link_elf_lookup_symbol(linker_file_t, const char *, 129326e27d8SDoug Rabson c_linker_sym_t *); 130e9eabf59SPeter Wemm static int link_elf_symbol_values(linker_file_t, c_linker_sym_t, 131e9eabf59SPeter Wemm linker_symval_t *); 132326e27d8SDoug Rabson static int link_elf_search_symbol(linker_file_t, caddr_t value, 133326e27d8SDoug Rabson c_linker_sym_t *sym, long *diffp); 134326e27d8SDoug Rabson 135326e27d8SDoug Rabson static void link_elf_unload_file(linker_file_t); 136f41325dbSPeter Wemm static int link_elf_lookup_set(linker_file_t, const char *, 137f41325dbSPeter Wemm void ***, void ***, int *); 138bb9fe9ddSBrian Feldman static int link_elf_each_function_name(linker_file_t, 139e9eabf59SPeter Wemm int (*)(const char *, void *), void *); 140a2024a3eSJohn Birrell static int link_elf_each_function_nameval(linker_file_t, 141a2024a3eSJohn Birrell linker_function_nameval_callback_t, 142a2024a3eSJohn Birrell void *); 1437251b4bfSJake Burkholder static void link_elf_reloc_local(linker_file_t); 144e76f11f4SAndriy Gapon static long link_elf_symtab_get(linker_file_t, const Elf_Sym **); 14500a5db46SStacey Son static long link_elf_strtab_get(linker_file_t, caddr_t *); 146326e27d8SDoug Rabson 147*cff8c6f2SKonstantin Belousov static int elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, 148*cff8c6f2SKonstantin Belousov Elf_Addr *); 149e9eabf59SPeter Wemm 150326e27d8SDoug Rabson static kobj_method_t link_elf_methods[] = { 151326e27d8SDoug Rabson KOBJMETHOD(linker_lookup_symbol, link_elf_lookup_symbol), 152326e27d8SDoug Rabson KOBJMETHOD(linker_symbol_values, link_elf_symbol_values), 153326e27d8SDoug Rabson KOBJMETHOD(linker_search_symbol, link_elf_search_symbol), 154326e27d8SDoug Rabson KOBJMETHOD(linker_unload, link_elf_unload_file), 15554823af2SPeter Wemm KOBJMETHOD(linker_load_file, link_elf_load_file), 15654823af2SPeter Wemm KOBJMETHOD(linker_link_preload, link_elf_link_preload), 15754823af2SPeter Wemm KOBJMETHOD(linker_link_preload_finish, link_elf_link_preload_finish), 158f41325dbSPeter Wemm KOBJMETHOD(linker_lookup_set, link_elf_lookup_set), 159bb9fe9ddSBrian Feldman KOBJMETHOD(linker_each_function_name, link_elf_each_function_name), 160a2024a3eSJohn Birrell KOBJMETHOD(linker_each_function_nameval, link_elf_each_function_nameval), 161a2024a3eSJohn Birrell KOBJMETHOD(linker_ctf_get, link_elf_ctf_get), 16200a5db46SStacey Son KOBJMETHOD(linker_symtab_get, link_elf_symtab_get), 16300a5db46SStacey Son KOBJMETHOD(linker_strtab_get, link_elf_strtab_get), 164326e27d8SDoug Rabson { 0, 0 } 165326e27d8SDoug Rabson }; 166326e27d8SDoug Rabson 167326e27d8SDoug Rabson static struct linker_class link_elf_class = { 168326e27d8SDoug Rabson #if ELF_TARG_CLASS == ELFCLASS32 169e9eabf59SPeter Wemm "elf32_obj", 170326e27d8SDoug Rabson #else 171e9eabf59SPeter Wemm "elf64_obj", 172326e27d8SDoug Rabson #endif 173326e27d8SDoug Rabson link_elf_methods, sizeof(struct elf_file) 174326e27d8SDoug Rabson }; 175326e27d8SDoug Rabson 176326e27d8SDoug Rabson static int relocate_file(elf_file_t ef); 1772832cd54SKonstantin Belousov static void elf_obj_cleanup_globals_cache(elf_file_t); 178a4f67738SDoug Rabson 179a4f67738SDoug Rabson static void 180552f9f63SEdwin Groothuis link_elf_error(const char *filename, const char *s) 1811720979bSMarcel Moolenaar { 182552f9f63SEdwin Groothuis if (filename == NULL) 1831720979bSMarcel Moolenaar printf("kldload: %s\n", s); 184552f9f63SEdwin Groothuis else 185552f9f63SEdwin Groothuis printf("kldload: %s: %s\n", filename, s); 1861720979bSMarcel Moolenaar } 1871720979bSMarcel Moolenaar 1881720979bSMarcel Moolenaar static void 189a4f67738SDoug Rabson link_elf_init(void *arg) 190a4f67738SDoug Rabson { 191a4f67738SDoug Rabson 192326e27d8SDoug Rabson linker_add_class(&link_elf_class); 193a4f67738SDoug Rabson } 194a4f67738SDoug Rabson 195e9eabf59SPeter Wemm SYSINIT(link_elf_obj, SI_SUB_KLD, SI_ORDER_SECOND, link_elf_init, 0); 196a4f67738SDoug Rabson 197a4f67738SDoug Rabson static int 198e9eabf59SPeter Wemm link_elf_link_preload(linker_class_t cls, const char *filename, 199e9eabf59SPeter Wemm linker_file_t *result) 2002d636ab0SPeter Wemm { 20170b7ffeeSIan Dowse Elf_Ehdr *hdr; 20270b7ffeeSIan Dowse Elf_Shdr *shdr; 20370b7ffeeSIan Dowse Elf_Sym *es; 20470b7ffeeSIan Dowse void *modptr, *baseptr, *sizeptr; 20570b7ffeeSIan Dowse char *type; 20670b7ffeeSIan Dowse elf_file_t ef; 20770b7ffeeSIan Dowse linker_file_t lf; 20870b7ffeeSIan Dowse Elf_Addr off; 20970b7ffeeSIan Dowse int error, i, j, pb, ra, rl, shstrindex, symstrindex, symtabindex; 21070b7ffeeSIan Dowse 21170b7ffeeSIan Dowse /* Look to see if we have the file preloaded */ 21270b7ffeeSIan Dowse modptr = preload_search_by_name(filename); 21370b7ffeeSIan Dowse if (modptr == NULL) 21470b7ffeeSIan Dowse return ENOENT; 21570b7ffeeSIan Dowse 21670b7ffeeSIan Dowse type = (char *)preload_search_info(modptr, MODINFO_TYPE); 21770b7ffeeSIan Dowse baseptr = preload_search_info(modptr, MODINFO_ADDR); 21870b7ffeeSIan Dowse sizeptr = preload_search_info(modptr, MODINFO_SIZE); 21970b7ffeeSIan Dowse hdr = (Elf_Ehdr *)preload_search_info(modptr, MODINFO_METADATA | 22070b7ffeeSIan Dowse MODINFOMD_ELFHDR); 22170b7ffeeSIan Dowse shdr = (Elf_Shdr *)preload_search_info(modptr, MODINFO_METADATA | 22270b7ffeeSIan Dowse MODINFOMD_SHDR); 22370b7ffeeSIan Dowse if (type == NULL || (strcmp(type, "elf" __XSTRING(__ELF_WORD_SIZE) 22470b7ffeeSIan Dowse " obj module") != 0 && 22570b7ffeeSIan Dowse strcmp(type, "elf obj module") != 0)) { 226de78ca7eSPeter Wemm return (EFTYPE); 227de78ca7eSPeter Wemm } 22870b7ffeeSIan Dowse if (baseptr == NULL || sizeptr == NULL || hdr == NULL || 22970b7ffeeSIan Dowse shdr == NULL) 23070b7ffeeSIan Dowse return (EINVAL); 23170b7ffeeSIan Dowse 23270b7ffeeSIan Dowse lf = linker_make_file(filename, &link_elf_class); 23370b7ffeeSIan Dowse if (lf == NULL) 23470b7ffeeSIan Dowse return (ENOMEM); 23570b7ffeeSIan Dowse 23670b7ffeeSIan Dowse ef = (elf_file_t)lf; 23770b7ffeeSIan Dowse ef->preloaded = 1; 23870b7ffeeSIan Dowse ef->address = *(caddr_t *)baseptr; 23970b7ffeeSIan Dowse lf->address = *(caddr_t *)baseptr; 24070b7ffeeSIan Dowse lf->size = *(size_t *)sizeptr; 24170b7ffeeSIan Dowse 24270b7ffeeSIan Dowse if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS || 24370b7ffeeSIan Dowse hdr->e_ident[EI_DATA] != ELF_TARG_DATA || 24470b7ffeeSIan Dowse hdr->e_ident[EI_VERSION] != EV_CURRENT || 24570b7ffeeSIan Dowse hdr->e_version != EV_CURRENT || 24670b7ffeeSIan Dowse hdr->e_type != ET_REL || 24770b7ffeeSIan Dowse hdr->e_machine != ELF_TARG_MACH) { 24870b7ffeeSIan Dowse error = EFTYPE; 24970b7ffeeSIan Dowse goto out; 25070b7ffeeSIan Dowse } 25170b7ffeeSIan Dowse ef->e_shdr = shdr; 25270b7ffeeSIan Dowse 25370b7ffeeSIan Dowse /* Scan the section header for information and table sizing. */ 25470b7ffeeSIan Dowse symtabindex = -1; 25570b7ffeeSIan Dowse symstrindex = -1; 25670b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 25770b7ffeeSIan Dowse switch (shdr[i].sh_type) { 25870b7ffeeSIan Dowse case SHT_PROGBITS: 25970b7ffeeSIan Dowse case SHT_NOBITS: 26070b7ffeeSIan Dowse ef->nprogtab++; 26170b7ffeeSIan Dowse break; 26270b7ffeeSIan Dowse case SHT_SYMTAB: 26370b7ffeeSIan Dowse symtabindex = i; 26470b7ffeeSIan Dowse symstrindex = shdr[i].sh_link; 26570b7ffeeSIan Dowse break; 26670b7ffeeSIan Dowse case SHT_REL: 267b474c780SDavid E. O'Brien ef->nreltab++; 26870b7ffeeSIan Dowse break; 26970b7ffeeSIan Dowse case SHT_RELA: 270b474c780SDavid E. O'Brien ef->nrelatab++; 27170b7ffeeSIan Dowse break; 27270b7ffeeSIan Dowse } 27370b7ffeeSIan Dowse } 27470b7ffeeSIan Dowse 27570b7ffeeSIan Dowse shstrindex = hdr->e_shstrndx; 27670b7ffeeSIan Dowse if (ef->nprogtab == 0 || symstrindex < 0 || 27770b7ffeeSIan Dowse symstrindex >= hdr->e_shnum || 27870b7ffeeSIan Dowse shdr[symstrindex].sh_type != SHT_STRTAB || shstrindex == 0 || 27970b7ffeeSIan Dowse shstrindex >= hdr->e_shnum || 28070b7ffeeSIan Dowse shdr[shstrindex].sh_type != SHT_STRTAB) { 28170b7ffeeSIan Dowse printf("%s: bad/missing section headers\n", filename); 28270b7ffeeSIan Dowse error = ENOEXEC; 28370b7ffeeSIan Dowse goto out; 28470b7ffeeSIan Dowse } 28570b7ffeeSIan Dowse 28670b7ffeeSIan Dowse /* Allocate space for tracking the load chunks */ 28770b7ffeeSIan Dowse if (ef->nprogtab != 0) 28870b7ffeeSIan Dowse ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab), 28970b7ffeeSIan Dowse M_LINKER, M_WAITOK | M_ZERO); 290b474c780SDavid E. O'Brien if (ef->nreltab != 0) 291b474c780SDavid E. O'Brien ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab), 292b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 293b474c780SDavid E. O'Brien if (ef->nrelatab != 0) 294b474c780SDavid E. O'Brien ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab), 295b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 29670b7ffeeSIan Dowse if ((ef->nprogtab != 0 && ef->progtab == NULL) || 297b474c780SDavid E. O'Brien (ef->nreltab != 0 && ef->reltab == NULL) || 298b474c780SDavid E. O'Brien (ef->nrelatab != 0 && ef->relatab == NULL)) { 29970b7ffeeSIan Dowse error = ENOMEM; 30070b7ffeeSIan Dowse goto out; 30170b7ffeeSIan Dowse } 30270b7ffeeSIan Dowse 30370b7ffeeSIan Dowse /* XXX, relocate the sh_addr fields saved by the loader. */ 30470b7ffeeSIan Dowse off = 0; 30570b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 30670b7ffeeSIan Dowse if (shdr[i].sh_addr != 0 && (off == 0 || shdr[i].sh_addr < off)) 30770b7ffeeSIan Dowse off = shdr[i].sh_addr; 30870b7ffeeSIan Dowse } 30970b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 31070b7ffeeSIan Dowse if (shdr[i].sh_addr != 0) 31170b7ffeeSIan Dowse shdr[i].sh_addr = shdr[i].sh_addr - off + 31270b7ffeeSIan Dowse (Elf_Addr)ef->address; 31370b7ffeeSIan Dowse } 31470b7ffeeSIan Dowse 31570b7ffeeSIan Dowse ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym); 31670b7ffeeSIan Dowse ef->ddbsymtab = (Elf_Sym *)shdr[symtabindex].sh_addr; 31770b7ffeeSIan Dowse ef->ddbstrcnt = shdr[symstrindex].sh_size; 31870b7ffeeSIan Dowse ef->ddbstrtab = (char *)shdr[symstrindex].sh_addr; 31970b7ffeeSIan Dowse ef->shstrcnt = shdr[shstrindex].sh_size; 32070b7ffeeSIan Dowse ef->shstrtab = (char *)shdr[shstrindex].sh_addr; 32170b7ffeeSIan Dowse 32270b7ffeeSIan Dowse /* Now fill out progtab and the relocation tables. */ 32370b7ffeeSIan Dowse pb = 0; 32470b7ffeeSIan Dowse rl = 0; 32570b7ffeeSIan Dowse ra = 0; 32670b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 32770b7ffeeSIan Dowse switch (shdr[i].sh_type) { 32870b7ffeeSIan Dowse case SHT_PROGBITS: 32970b7ffeeSIan Dowse case SHT_NOBITS: 33070b7ffeeSIan Dowse ef->progtab[pb].addr = (void *)shdr[i].sh_addr; 33170b7ffeeSIan Dowse if (shdr[i].sh_type == SHT_PROGBITS) 33270b7ffeeSIan Dowse ef->progtab[pb].name = "<<PROGBITS>>"; 33370b7ffeeSIan Dowse else 33470b7ffeeSIan Dowse ef->progtab[pb].name = "<<NOBITS>>"; 33570b7ffeeSIan Dowse ef->progtab[pb].size = shdr[i].sh_size; 33670b7ffeeSIan Dowse ef->progtab[pb].sec = i; 33770b7ffeeSIan Dowse if (ef->shstrtab && shdr[i].sh_name != 0) 33870b7ffeeSIan Dowse ef->progtab[pb].name = 33970b7ffeeSIan Dowse ef->shstrtab + shdr[i].sh_name; 34050c202c5SJeff Roberson if (ef->progtab[pb].name != NULL && 3415f67450dSDimitry Andric !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) { 34250c202c5SJeff Roberson void *dpcpu; 34350c202c5SJeff Roberson 34450c202c5SJeff Roberson dpcpu = dpcpu_alloc(shdr[i].sh_size); 34550c202c5SJeff Roberson if (dpcpu == NULL) { 34650c202c5SJeff Roberson error = ENOSPC; 34750c202c5SJeff Roberson goto out; 34850c202c5SJeff Roberson } 34950c202c5SJeff Roberson memcpy(dpcpu, ef->progtab[pb].addr, 35050c202c5SJeff Roberson ef->progtab[pb].size); 35150c202c5SJeff Roberson dpcpu_copy(dpcpu, shdr[i].sh_size); 35250c202c5SJeff Roberson ef->progtab[pb].addr = dpcpu; 353eddfbb76SRobert Watson #ifdef VIMAGE 354eddfbb76SRobert Watson } else if (ef->progtab[pb].name != NULL && 35517ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) { 356eddfbb76SRobert Watson void *vnet_data; 357eddfbb76SRobert Watson 358eddfbb76SRobert Watson vnet_data = vnet_data_alloc(shdr[i].sh_size); 359eddfbb76SRobert Watson if (vnet_data == NULL) { 360eddfbb76SRobert Watson error = ENOSPC; 361eddfbb76SRobert Watson goto out; 362eddfbb76SRobert Watson } 363eddfbb76SRobert Watson memcpy(vnet_data, ef->progtab[pb].addr, 364eddfbb76SRobert Watson ef->progtab[pb].size); 365eddfbb76SRobert Watson vnet_data_copy(vnet_data, shdr[i].sh_size); 366eddfbb76SRobert Watson ef->progtab[pb].addr = vnet_data; 367eddfbb76SRobert Watson #endif 3680067051fSMarcel Moolenaar } else if (ef->progtab[pb].name != NULL && 3690067051fSMarcel Moolenaar !strcmp(ef->progtab[pb].name, ".ctors")) { 3700067051fSMarcel Moolenaar lf->ctors_addr = ef->progtab[pb].addr; 3710067051fSMarcel Moolenaar lf->ctors_size = shdr[i].sh_size; 37250c202c5SJeff Roberson } 37370b7ffeeSIan Dowse 37470b7ffeeSIan Dowse /* Update all symbol values with the offset. */ 37570b7ffeeSIan Dowse for (j = 0; j < ef->ddbsymcnt; j++) { 37670b7ffeeSIan Dowse es = &ef->ddbsymtab[j]; 37770b7ffeeSIan Dowse if (es->st_shndx != i) 37870b7ffeeSIan Dowse continue; 37970b7ffeeSIan Dowse es->st_value += (Elf_Addr)ef->progtab[pb].addr; 38070b7ffeeSIan Dowse } 38170b7ffeeSIan Dowse pb++; 38270b7ffeeSIan Dowse break; 38370b7ffeeSIan Dowse case SHT_REL: 38470b7ffeeSIan Dowse ef->reltab[rl].rel = (Elf_Rel *)shdr[i].sh_addr; 38570b7ffeeSIan Dowse ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel); 38670b7ffeeSIan Dowse ef->reltab[rl].sec = shdr[i].sh_info; 38770b7ffeeSIan Dowse rl++; 38870b7ffeeSIan Dowse break; 38970b7ffeeSIan Dowse case SHT_RELA: 39070b7ffeeSIan Dowse ef->relatab[ra].rela = (Elf_Rela *)shdr[i].sh_addr; 39170b7ffeeSIan Dowse ef->relatab[ra].nrela = 39270b7ffeeSIan Dowse shdr[i].sh_size / sizeof(Elf_Rela); 39370b7ffeeSIan Dowse ef->relatab[ra].sec = shdr[i].sh_info; 39470b7ffeeSIan Dowse ra++; 39570b7ffeeSIan Dowse break; 39670b7ffeeSIan Dowse } 39770b7ffeeSIan Dowse } 39870b7ffeeSIan Dowse if (pb != ef->nprogtab) 39970b7ffeeSIan Dowse panic("lost progbits"); 400b474c780SDavid E. O'Brien if (rl != ef->nreltab) 401b474c780SDavid E. O'Brien panic("lost reltab"); 402b474c780SDavid E. O'Brien if (ra != ef->nrelatab) 403b474c780SDavid E. O'Brien panic("lost relatab"); 40470b7ffeeSIan Dowse 40570b7ffeeSIan Dowse /* Local intra-module relocations */ 40670b7ffeeSIan Dowse link_elf_reloc_local(lf); 40770b7ffeeSIan Dowse 40870b7ffeeSIan Dowse *result = lf; 40970b7ffeeSIan Dowse return (0); 41070b7ffeeSIan Dowse 41170b7ffeeSIan Dowse out: 41270b7ffeeSIan Dowse /* preload not done this way */ 41370b7ffeeSIan Dowse linker_file_unload(lf, LINKER_UNLOAD_FORCE); 41470b7ffeeSIan Dowse return (error); 41570b7ffeeSIan Dowse } 416de78ca7eSPeter Wemm 4170067051fSMarcel Moolenaar static void 4180067051fSMarcel Moolenaar link_elf_invoke_ctors(caddr_t addr, size_t size) 4190067051fSMarcel Moolenaar { 4200067051fSMarcel Moolenaar void (**ctor)(void); 4210067051fSMarcel Moolenaar size_t i, cnt; 4220067051fSMarcel Moolenaar 4230067051fSMarcel Moolenaar if (addr == NULL || size == 0) 4240067051fSMarcel Moolenaar return; 4250067051fSMarcel Moolenaar cnt = size / sizeof(*ctor); 4260067051fSMarcel Moolenaar ctor = (void *)addr; 4270067051fSMarcel Moolenaar for (i = 0; i < cnt; i++) { 4280067051fSMarcel Moolenaar if (ctor[i] != NULL) 4290067051fSMarcel Moolenaar (*ctor[i])(); 4300067051fSMarcel Moolenaar } 4310067051fSMarcel Moolenaar } 4320067051fSMarcel Moolenaar 433de78ca7eSPeter Wemm static int 43454823af2SPeter Wemm link_elf_link_preload_finish(linker_file_t lf) 43554823af2SPeter Wemm { 43670b7ffeeSIan Dowse elf_file_t ef; 43770b7ffeeSIan Dowse int error; 43870b7ffeeSIan Dowse 43970b7ffeeSIan Dowse ef = (elf_file_t)lf; 44070b7ffeeSIan Dowse error = relocate_file(ef); 44170b7ffeeSIan Dowse if (error) 44270b7ffeeSIan Dowse return error; 44370b7ffeeSIan Dowse 44470b7ffeeSIan Dowse /* Notify MD code that a module is being loaded. */ 44570b7ffeeSIan Dowse error = elf_cpu_load_file(lf); 44670b7ffeeSIan Dowse if (error) 44770b7ffeeSIan Dowse return (error); 44870b7ffeeSIan Dowse 4490067051fSMarcel Moolenaar /* Invoke .ctors */ 4500067051fSMarcel Moolenaar link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size); 45170b7ffeeSIan Dowse return (0); 45254823af2SPeter Wemm } 45354823af2SPeter Wemm 45454823af2SPeter Wemm static int 455c143d6c2SMarcel Moolenaar link_elf_load_file(linker_class_t cls, const char *filename, 456c143d6c2SMarcel Moolenaar linker_file_t *result) 457a4f67738SDoug Rabson { 458a4f67738SDoug Rabson struct nameidata nd; 459b40ce416SJulian Elischer struct thread *td = curthread; /* XXX */ 460caab6e90SPeter Wemm Elf_Ehdr *hdr; 461e9eabf59SPeter Wemm Elf_Shdr *shdr; 462a8774e39SPeter Wemm Elf_Sym *es; 463a8774e39SPeter Wemm int nbytes, i, j; 46420947801SPeter Wemm vm_offset_t mapbase; 465fe3db7c7SDoug Rabson size_t mapsize; 466a4f67738SDoug Rabson int error = 0; 467526d0bd5SKonstantin Belousov ssize_t resid; 468526d0bd5SKonstantin Belousov int flags; 469a4f67738SDoug Rabson elf_file_t ef; 470a4f67738SDoug Rabson linker_file_t lf; 471ca65d5c7SPeter Wemm int symtabindex; 472ca65d5c7SPeter Wemm int symstrindex; 473e9eabf59SPeter Wemm int shstrindex; 474e9eabf59SPeter Wemm int nsym; 47520947801SPeter Wemm int pb, rl, ra; 476e9eabf59SPeter Wemm int alignmask; 477ca65d5c7SPeter Wemm 478ca65d5c7SPeter Wemm shdr = NULL; 479ca65d5c7SPeter Wemm lf = NULL; 480e9eabf59SPeter Wemm mapsize = 0; 481e9eabf59SPeter Wemm hdr = NULL; 482a4f67738SDoug Rabson 4835050aa86SKonstantin Belousov NDINIT(&nd, LOOKUP, FOLLOW, UIO_SYSSPACE, filename, td); 484e6796b67SKirk McKusick flags = FREAD; 4859e223287SKonstantin Belousov error = vn_open(&nd, &flags, 0, NULL); 486a4f67738SDoug Rabson if (error) 487a4f67738SDoug Rabson return error; 488762e6b85SEivind Eklund NDFREE(&nd, NDF_ONLY_PNBUF); 4891f49b573SRuslan Ermilov if (nd.ni_vp->v_type != VREG) { 4901f49b573SRuslan Ermilov error = ENOEXEC; 4911f49b573SRuslan Ermilov goto out; 4921f49b573SRuslan Ermilov } 493a3df768bSRobert Watson #ifdef MAC 49430d239bcSRobert Watson error = mac_kld_check_load(td->td_ucred, nd.ni_vp); 495a3df768bSRobert Watson if (error) { 496a3df768bSRobert Watson goto out; 497a3df768bSRobert Watson } 498a3df768bSRobert Watson #endif 499a4f67738SDoug Rabson 500e9eabf59SPeter Wemm /* Read the elf header from the file. */ 501e9eabf59SPeter Wemm hdr = malloc(sizeof(*hdr), M_LINKER, M_WAITOK); 502e9eabf59SPeter Wemm error = vn_rdwr(UIO_READ, nd.ni_vp, (void *)hdr, sizeof(*hdr), 0, 5039ca43589SRobert Watson UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 5049ca43589SRobert Watson &resid, td); 505a4f67738SDoug Rabson if (error) 506a4f67738SDoug Rabson goto out; 507e9eabf59SPeter Wemm if (resid != 0){ 508e9eabf59SPeter Wemm error = ENOEXEC; 509e9eabf59SPeter Wemm goto out; 510e9eabf59SPeter Wemm } 511a4f67738SDoug Rabson 512caab6e90SPeter Wemm if (!IS_ELF(*hdr)) { 513fe3db7c7SDoug Rabson error = ENOEXEC; 514a4f67738SDoug Rabson goto out; 515fe3db7c7SDoug Rabson } 516fe3db7c7SDoug Rabson 517caab6e90SPeter Wemm if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS 518caab6e90SPeter Wemm || hdr->e_ident[EI_DATA] != ELF_TARG_DATA) { 519552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported file layout"); 520fe3db7c7SDoug Rabson error = ENOEXEC; 521fe3db7c7SDoug Rabson goto out; 522fe3db7c7SDoug Rabson } 523caab6e90SPeter Wemm if (hdr->e_ident[EI_VERSION] != EV_CURRENT 524caab6e90SPeter Wemm || hdr->e_version != EV_CURRENT) { 525552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported file version"); 526fe3db7c7SDoug Rabson error = ENOEXEC; 527fe3db7c7SDoug Rabson goto out; 528fe3db7c7SDoug Rabson } 529e9eabf59SPeter Wemm if (hdr->e_type != ET_REL) { 530a1d7ce03SAttilio Rao error = ENOSYS; 531fe3db7c7SDoug Rabson goto out; 532fe3db7c7SDoug Rabson } 533caab6e90SPeter Wemm if (hdr->e_machine != ELF_TARG_MACH) { 534552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported machine"); 535fe3db7c7SDoug Rabson error = ENOEXEC; 536fe3db7c7SDoug Rabson goto out; 537fe3db7c7SDoug Rabson } 538a4f67738SDoug Rabson 539326e27d8SDoug Rabson lf = linker_make_file(filename, &link_elf_class); 540326e27d8SDoug Rabson if (!lf) { 541326e27d8SDoug Rabson error = ENOMEM; 542326e27d8SDoug Rabson goto out; 543326e27d8SDoug Rabson } 544326e27d8SDoug Rabson ef = (elf_file_t) lf; 545e9eabf59SPeter Wemm ef->nprogtab = 0; 546e9eabf59SPeter Wemm ef->e_shdr = 0; 547b474c780SDavid E. O'Brien ef->nreltab = 0; 548b474c780SDavid E. O'Brien ef->nrelatab = 0; 549e9eabf59SPeter Wemm 550e9eabf59SPeter Wemm /* Allocate and read in the section header */ 551e9eabf59SPeter Wemm nbytes = hdr->e_shnum * hdr->e_shentsize; 552e9eabf59SPeter Wemm if (nbytes == 0 || hdr->e_shoff == 0 || 553e9eabf59SPeter Wemm hdr->e_shentsize != sizeof(Elf_Shdr)) { 554e9eabf59SPeter Wemm error = ENOEXEC; 555e9eabf59SPeter Wemm goto out; 556e9eabf59SPeter Wemm } 55720947801SPeter Wemm shdr = malloc(nbytes, M_LINKER, M_WAITOK); 558e9eabf59SPeter Wemm ef->e_shdr = shdr; 559e9eabf59SPeter Wemm error = vn_rdwr(UIO_READ, nd.ni_vp, (caddr_t)shdr, nbytes, hdr->e_shoff, 560e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, &resid, td); 561e9eabf59SPeter Wemm if (error) 562e9eabf59SPeter Wemm goto out; 563e9eabf59SPeter Wemm if (resid) { 564e9eabf59SPeter Wemm error = ENOEXEC; 565e9eabf59SPeter Wemm goto out; 566e9eabf59SPeter Wemm } 567e9eabf59SPeter Wemm 568e9eabf59SPeter Wemm /* Scan the section header for information and table sizing. */ 569e9eabf59SPeter Wemm nsym = 0; 570e9eabf59SPeter Wemm symtabindex = -1; 571e9eabf59SPeter Wemm symstrindex = -1; 572e9eabf59SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 573676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 574676799a0SAndriy Gapon continue; 575e9eabf59SPeter Wemm switch (shdr[i].sh_type) { 576e9eabf59SPeter Wemm case SHT_PROGBITS: 577e9eabf59SPeter Wemm case SHT_NOBITS: 57820947801SPeter Wemm ef->nprogtab++; 579e9eabf59SPeter Wemm break; 580e9eabf59SPeter Wemm case SHT_SYMTAB: 581e9eabf59SPeter Wemm nsym++; 582e9eabf59SPeter Wemm symtabindex = i; 583e9eabf59SPeter Wemm symstrindex = shdr[i].sh_link; 584e9eabf59SPeter Wemm break; 585e9eabf59SPeter Wemm case SHT_REL: 586b474c780SDavid E. O'Brien ef->nreltab++; 587e9eabf59SPeter Wemm break; 588e9eabf59SPeter Wemm case SHT_RELA: 589b474c780SDavid E. O'Brien ef->nrelatab++; 590e9eabf59SPeter Wemm break; 591e9eabf59SPeter Wemm case SHT_STRTAB: 592e9eabf59SPeter Wemm break; 593e9eabf59SPeter Wemm } 594e9eabf59SPeter Wemm } 59520947801SPeter Wemm if (ef->nprogtab == 0) { 596552f9f63SEdwin Groothuis link_elf_error(filename, "file has no contents"); 597e9eabf59SPeter Wemm error = ENOEXEC; 598e9eabf59SPeter Wemm goto out; 599e9eabf59SPeter Wemm } 600e9eabf59SPeter Wemm if (nsym != 1) { 601e9eabf59SPeter Wemm /* Only allow one symbol table for now */ 602552f9f63SEdwin Groothuis link_elf_error(filename, "file has no valid symbol table"); 603e9eabf59SPeter Wemm error = ENOEXEC; 604e9eabf59SPeter Wemm goto out; 605e9eabf59SPeter Wemm } 606e9eabf59SPeter Wemm if (symstrindex < 0 || symstrindex > hdr->e_shnum || 607e9eabf59SPeter Wemm shdr[symstrindex].sh_type != SHT_STRTAB) { 608552f9f63SEdwin Groothuis link_elf_error(filename, "file has invalid symbol strings"); 609e9eabf59SPeter Wemm error = ENOEXEC; 610e9eabf59SPeter Wemm goto out; 611e9eabf59SPeter Wemm } 612e9eabf59SPeter Wemm 613e9eabf59SPeter Wemm /* Allocate space for tracking the load chunks */ 614e9eabf59SPeter Wemm if (ef->nprogtab != 0) 61520947801SPeter Wemm ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab), 61620947801SPeter Wemm M_LINKER, M_WAITOK | M_ZERO); 617b474c780SDavid E. O'Brien if (ef->nreltab != 0) 618b474c780SDavid E. O'Brien ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab), 619b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 620b474c780SDavid E. O'Brien if (ef->nrelatab != 0) 621b474c780SDavid E. O'Brien ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab), 622b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 623e9eabf59SPeter Wemm 62420947801SPeter Wemm if (symtabindex == -1) 62520947801SPeter Wemm panic("lost symbol table index"); 62620947801SPeter Wemm /* Allocate space for and load the symbol table */ 627e9eabf59SPeter Wemm ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym); 628e9eabf59SPeter Wemm ef->ddbsymtab = malloc(shdr[symtabindex].sh_size, M_LINKER, M_WAITOK); 62920947801SPeter Wemm error = vn_rdwr(UIO_READ, nd.ni_vp, (void *)ef->ddbsymtab, 63020947801SPeter Wemm shdr[symtabindex].sh_size, shdr[symtabindex].sh_offset, 63120947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 63220947801SPeter Wemm &resid, td); 63320947801SPeter Wemm if (error) 63420947801SPeter Wemm goto out; 63520947801SPeter Wemm if (resid != 0){ 63620947801SPeter Wemm error = EINVAL; 63720947801SPeter Wemm goto out; 63820947801SPeter Wemm } 639e9eabf59SPeter Wemm 64020947801SPeter Wemm if (symstrindex == -1) 64120947801SPeter Wemm panic("lost symbol string index"); 64220947801SPeter Wemm /* Allocate space for and load the symbol strings */ 643e9eabf59SPeter Wemm ef->ddbstrcnt = shdr[symstrindex].sh_size; 644e9eabf59SPeter Wemm ef->ddbstrtab = malloc(shdr[symstrindex].sh_size, M_LINKER, M_WAITOK); 64520947801SPeter Wemm error = vn_rdwr(UIO_READ, nd.ni_vp, ef->ddbstrtab, 64620947801SPeter Wemm shdr[symstrindex].sh_size, shdr[symstrindex].sh_offset, 64720947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 64820947801SPeter Wemm &resid, td); 64920947801SPeter Wemm if (error) 65020947801SPeter Wemm goto out; 65120947801SPeter Wemm if (resid != 0){ 65220947801SPeter Wemm error = EINVAL; 65320947801SPeter Wemm goto out; 65420947801SPeter Wemm } 655e9eabf59SPeter Wemm 656e9eabf59SPeter Wemm /* Do we have a string table for the section names? */ 657e9eabf59SPeter Wemm shstrindex = -1; 65820947801SPeter Wemm if (hdr->e_shstrndx != 0 && 65920947801SPeter Wemm shdr[hdr->e_shstrndx].sh_type == SHT_STRTAB) { 660e9eabf59SPeter Wemm shstrindex = hdr->e_shstrndx; 661e9eabf59SPeter Wemm ef->shstrcnt = shdr[shstrindex].sh_size; 66220947801SPeter Wemm ef->shstrtab = malloc(shdr[shstrindex].sh_size, M_LINKER, 66320947801SPeter Wemm M_WAITOK); 66420947801SPeter Wemm error = vn_rdwr(UIO_READ, nd.ni_vp, ef->shstrtab, 66520947801SPeter Wemm shdr[shstrindex].sh_size, shdr[shstrindex].sh_offset, 666e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 667e9eabf59SPeter Wemm &resid, td); 668e9eabf59SPeter Wemm if (error) 669e9eabf59SPeter Wemm goto out; 67020947801SPeter Wemm if (resid != 0){ 67120947801SPeter Wemm error = EINVAL; 67220947801SPeter Wemm goto out; 67320947801SPeter Wemm } 674e9eabf59SPeter Wemm } 675e9eabf59SPeter Wemm 67620947801SPeter Wemm /* Size up code/data(progbits) and bss(nobits). */ 677e9eabf59SPeter Wemm alignmask = 0; 678e9eabf59SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 679676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 680676799a0SAndriy Gapon continue; 681e9eabf59SPeter Wemm switch (shdr[i].sh_type) { 682e9eabf59SPeter Wemm case SHT_PROGBITS: 683e9eabf59SPeter Wemm case SHT_NOBITS: 684e9eabf59SPeter Wemm alignmask = shdr[i].sh_addralign - 1; 685e9eabf59SPeter Wemm mapsize += alignmask; 686e9eabf59SPeter Wemm mapsize &= ~alignmask; 687e9eabf59SPeter Wemm mapsize += shdr[i].sh_size; 688e9eabf59SPeter Wemm break; 689e9eabf59SPeter Wemm } 690e9eabf59SPeter Wemm } 691e9eabf59SPeter Wemm 692e9eabf59SPeter Wemm /* 693e9eabf59SPeter Wemm * We know how much space we need for the text/data/bss/etc. 694e9eabf59SPeter Wemm * This stuff needs to be in a single chunk so that profiling etc 695e9eabf59SPeter Wemm * can get the bounds and gdb can associate offsets with modules 696e9eabf59SPeter Wemm */ 69720947801SPeter Wemm ef->object = vm_object_allocate(OBJT_DEFAULT, 69820947801SPeter Wemm round_page(mapsize) >> PAGE_SHIFT); 699fe3db7c7SDoug Rabson if (ef->object == NULL) { 700fe3db7c7SDoug Rabson error = ENOMEM; 701fe3db7c7SDoug Rabson goto out; 702fe3db7c7SDoug Rabson } 703fe3db7c7SDoug Rabson ef->address = (caddr_t) vm_map_min(kernel_map); 704ac68d1c9SAlan Cox 705ac68d1c9SAlan Cox /* 706ac68d1c9SAlan Cox * In order to satisfy amd64's architectural requirements on the 707ac68d1c9SAlan Cox * location of code and data in the kernel's address space, request a 708ac68d1c9SAlan Cox * mapping that is above the kernel. 709ac68d1c9SAlan Cox */ 7104104e835SOleksandr Tymoshenko #ifdef __amd64__ 711ac68d1c9SAlan Cox mapbase = KERNBASE; 7124104e835SOleksandr Tymoshenko #else 7134104e835SOleksandr Tymoshenko mapbase = VM_MIN_KERNEL_ADDRESS; 7144104e835SOleksandr Tymoshenko #endif 71520947801SPeter Wemm error = vm_map_find(kernel_map, ef->object, 0, &mapbase, 716edb572a3SJohn Baldwin round_page(mapsize), 0, VMFS_OPTIMAL_SPACE, VM_PROT_ALL, 717edb572a3SJohn Baldwin VM_PROT_ALL, 0); 718fe3db7c7SDoug Rabson if (error) { 719fe3db7c7SDoug Rabson vm_object_deallocate(ef->object); 720326e27d8SDoug Rabson ef->object = 0; 721fe3db7c7SDoug Rabson goto out; 722fe3db7c7SDoug Rabson } 72320947801SPeter Wemm 724e9eabf59SPeter Wemm /* Wire the pages */ 7257f1ef325SAlan Cox error = vm_map_wire(kernel_map, mapbase, 72620947801SPeter Wemm mapbase + round_page(mapsize), 727e9eabf59SPeter Wemm VM_MAP_WIRE_SYSTEM|VM_MAP_WIRE_NOHOLES); 7287f1ef325SAlan Cox if (error != KERN_SUCCESS) { 7297f1ef325SAlan Cox error = ENOMEM; 7307f1ef325SAlan Cox goto out; 7317f1ef325SAlan Cox } 732a4f67738SDoug Rabson 73320947801SPeter Wemm /* Inform the kld system about the situation */ 73420947801SPeter Wemm lf->address = ef->address = (caddr_t)mapbase; 73520947801SPeter Wemm lf->size = mapsize; 736e9eabf59SPeter Wemm 737e9eabf59SPeter Wemm /* 73820947801SPeter Wemm * Now load code/data(progbits), zero bss(nobits), allocate space for 73920947801SPeter Wemm * and load relocs 740e9eabf59SPeter Wemm */ 74120947801SPeter Wemm pb = 0; 74220947801SPeter Wemm rl = 0; 74320947801SPeter Wemm ra = 0; 74420947801SPeter Wemm alignmask = 0; 74520947801SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 746676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 747676799a0SAndriy Gapon continue; 74820947801SPeter Wemm switch (shdr[i].sh_type) { 74920947801SPeter Wemm case SHT_PROGBITS: 75020947801SPeter Wemm case SHT_NOBITS: 75120947801SPeter Wemm alignmask = shdr[i].sh_addralign - 1; 75220947801SPeter Wemm mapbase += alignmask; 75320947801SPeter Wemm mapbase &= ~alignmask; 7540067051fSMarcel Moolenaar if (ef->shstrtab != NULL && shdr[i].sh_name != 0) { 75550c202c5SJeff Roberson ef->progtab[pb].name = 75650c202c5SJeff Roberson ef->shstrtab + shdr[i].sh_name; 7570067051fSMarcel Moolenaar if (!strcmp(ef->progtab[pb].name, ".ctors")) { 7580067051fSMarcel Moolenaar lf->ctors_addr = (caddr_t)mapbase; 7590067051fSMarcel Moolenaar lf->ctors_size = shdr[i].sh_size; 7600067051fSMarcel Moolenaar } 7610067051fSMarcel Moolenaar } else if (shdr[i].sh_type == SHT_PROGBITS) 76220947801SPeter Wemm ef->progtab[pb].name = "<<PROGBITS>>"; 76350c202c5SJeff Roberson else 76450c202c5SJeff Roberson ef->progtab[pb].name = "<<NOBITS>>"; 76550c202c5SJeff Roberson if (ef->progtab[pb].name != NULL && 7665f67450dSDimitry Andric !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) 76750c202c5SJeff Roberson ef->progtab[pb].addr = 76850c202c5SJeff Roberson dpcpu_alloc(shdr[i].sh_size); 769eddfbb76SRobert Watson #ifdef VIMAGE 770eddfbb76SRobert Watson else if (ef->progtab[pb].name != NULL && 77117ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) 772eddfbb76SRobert Watson ef->progtab[pb].addr = 773eddfbb76SRobert Watson vnet_data_alloc(shdr[i].sh_size); 774eddfbb76SRobert Watson #endif 77550c202c5SJeff Roberson else 77650c202c5SJeff Roberson ef->progtab[pb].addr = 77750c202c5SJeff Roberson (void *)(uintptr_t)mapbase; 77850c202c5SJeff Roberson if (ef->progtab[pb].addr == NULL) { 77950c202c5SJeff Roberson error = ENOSPC; 78050c202c5SJeff Roberson goto out; 78150c202c5SJeff Roberson } 78250c202c5SJeff Roberson ef->progtab[pb].size = shdr[i].sh_size; 78350c202c5SJeff Roberson ef->progtab[pb].sec = i; 78450c202c5SJeff Roberson if (shdr[i].sh_type == SHT_PROGBITS) { 785e9eabf59SPeter Wemm error = vn_rdwr(UIO_READ, nd.ni_vp, 78620947801SPeter Wemm ef->progtab[pb].addr, 78720947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 78820947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, 78920947801SPeter Wemm NOCRED, &resid, td); 79020947801SPeter Wemm if (error) 79120947801SPeter Wemm goto out; 79220947801SPeter Wemm if (resid != 0){ 79320947801SPeter Wemm error = EINVAL; 79420947801SPeter Wemm goto out; 79520947801SPeter Wemm } 796eddfbb76SRobert Watson /* Initialize the per-cpu or vnet area. */ 797eddfbb76SRobert Watson if (ef->progtab[pb].addr != (void *)mapbase && 7985f67450dSDimitry Andric !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) 79950c202c5SJeff Roberson dpcpu_copy(ef->progtab[pb].addr, 80050c202c5SJeff Roberson shdr[i].sh_size); 801eddfbb76SRobert Watson #ifdef VIMAGE 802eddfbb76SRobert Watson else if (ef->progtab[pb].addr != 803eddfbb76SRobert Watson (void *)mapbase && 80417ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) 805eddfbb76SRobert Watson vnet_data_copy(ef->progtab[pb].addr, 806eddfbb76SRobert Watson shdr[i].sh_size); 807eddfbb76SRobert Watson #endif 80850c202c5SJeff Roberson } else 80920947801SPeter Wemm bzero(ef->progtab[pb].addr, shdr[i].sh_size); 810a8774e39SPeter Wemm 811a8774e39SPeter Wemm /* Update all symbol values with the offset. */ 812a8774e39SPeter Wemm for (j = 0; j < ef->ddbsymcnt; j++) { 813a8774e39SPeter Wemm es = &ef->ddbsymtab[j]; 814a8774e39SPeter Wemm if (es->st_shndx != i) 815a8774e39SPeter Wemm continue; 816a8774e39SPeter Wemm es->st_value += (Elf_Addr)ef->progtab[pb].addr; 817a8774e39SPeter Wemm } 81820947801SPeter Wemm mapbase += shdr[i].sh_size; 81920947801SPeter Wemm pb++; 82020947801SPeter Wemm break; 82120947801SPeter Wemm case SHT_REL: 82220947801SPeter Wemm ef->reltab[rl].rel = malloc(shdr[i].sh_size, M_LINKER, 82320947801SPeter Wemm M_WAITOK); 82420947801SPeter Wemm ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel); 82520947801SPeter Wemm ef->reltab[rl].sec = shdr[i].sh_info; 82620947801SPeter Wemm error = vn_rdwr(UIO_READ, nd.ni_vp, 82720947801SPeter Wemm (void *)ef->reltab[rl].rel, 82820947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 829e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 830e9eabf59SPeter Wemm &resid, td); 831e9eabf59SPeter Wemm if (error) 832e9eabf59SPeter Wemm goto out; 83320947801SPeter Wemm if (resid != 0){ 83420947801SPeter Wemm error = EINVAL; 83520947801SPeter Wemm goto out; 836e9eabf59SPeter Wemm } 83720947801SPeter Wemm rl++; 83820947801SPeter Wemm break; 83920947801SPeter Wemm case SHT_RELA: 84020947801SPeter Wemm ef->relatab[ra].rela = malloc(shdr[i].sh_size, M_LINKER, 84120947801SPeter Wemm M_WAITOK); 84220947801SPeter Wemm ef->relatab[ra].nrela = 84320947801SPeter Wemm shdr[i].sh_size / sizeof(Elf_Rela); 84420947801SPeter Wemm ef->relatab[ra].sec = shdr[i].sh_info; 845e9eabf59SPeter Wemm error = vn_rdwr(UIO_READ, nd.ni_vp, 84620947801SPeter Wemm (void *)ef->relatab[ra].rela, 84720947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 848e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 849e9eabf59SPeter Wemm &resid, td); 850e9eabf59SPeter Wemm if (error) 851e9eabf59SPeter Wemm goto out; 85220947801SPeter Wemm if (resid != 0){ 85320947801SPeter Wemm error = EINVAL; 85420947801SPeter Wemm goto out; 855e9eabf59SPeter Wemm } 85620947801SPeter Wemm ra++; 85720947801SPeter Wemm break; 85820947801SPeter Wemm } 85920947801SPeter Wemm } 86020947801SPeter Wemm if (pb != ef->nprogtab) 86120947801SPeter Wemm panic("lost progbits"); 862b474c780SDavid E. O'Brien if (rl != ef->nreltab) 863b474c780SDavid E. O'Brien panic("lost reltab"); 864b474c780SDavid E. O'Brien if (ra != ef->nrelatab) 865b474c780SDavid E. O'Brien panic("lost relatab"); 86620947801SPeter Wemm if (mapbase != (vm_offset_t)ef->address + mapsize) 8675b201fdcSPeter Wemm panic("mapbase 0x%lx != address %p + mapsize 0x%lx (0x%lx)\n", 8686f3c6327SNeel Natu (u_long)mapbase, ef->address, (u_long)mapsize, 8696f3c6327SNeel Natu (u_long)(vm_offset_t)ef->address + mapsize); 870e9eabf59SPeter Wemm 871e9eabf59SPeter Wemm /* Local intra-module relocations */ 8727251b4bfSJake Burkholder link_elf_reloc_local(lf); 8737251b4bfSJake Burkholder 874e9eabf59SPeter Wemm /* Pull in dependencies */ 8754f7afc20SKonstantin Belousov VOP_UNLOCK(nd.ni_vp, 0); 8767b9716baSIan Dowse error = linker_load_dependencies(lf); 8774f7afc20SKonstantin Belousov vn_lock(nd.ni_vp, LK_EXCLUSIVE | LK_RETRY); 878ca65d5c7SPeter Wemm if (error) 879de78ca7eSPeter Wemm goto out; 880e9eabf59SPeter Wemm 881e9eabf59SPeter Wemm /* External relocations */ 882326e27d8SDoug Rabson error = relocate_file(ef); 883ca65d5c7SPeter Wemm if (error) 884ca65d5c7SPeter Wemm goto out; 885ca65d5c7SPeter Wemm 886e9eabf59SPeter Wemm /* Notify MD code that a module is being loaded. */ 887e9eabf59SPeter Wemm error = elf_cpu_load_file(lf); 888ca65d5c7SPeter Wemm if (error) 889ca65d5c7SPeter Wemm goto out; 890a4f67738SDoug Rabson 8910067051fSMarcel Moolenaar /* Invoke .ctors */ 8920067051fSMarcel Moolenaar link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size); 8930067051fSMarcel Moolenaar 894a4f67738SDoug Rabson *result = lf; 895a4f67738SDoug Rabson 896a4f67738SDoug Rabson out: 897f1d2d305SKonstantin Belousov VOP_UNLOCK(nd.ni_vp, 0); 898f1d2d305SKonstantin Belousov vn_close(nd.ni_vp, FREAD, td->td_ucred, td); 899ca65d5c7SPeter Wemm if (error && lf) 90065a311fcSPoul-Henning Kamp linker_file_unload(lf, LINKER_UNLOAD_FORCE); 901e9eabf59SPeter Wemm if (hdr) 902e9eabf59SPeter Wemm free(hdr, M_LINKER); 903a4f67738SDoug Rabson 904a4f67738SDoug Rabson return error; 905a4f67738SDoug Rabson } 906a4f67738SDoug Rabson 907a4f67738SDoug Rabson static void 908de78ca7eSPeter Wemm link_elf_unload_file(linker_file_t file) 909a4f67738SDoug Rabson { 910326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) file; 91120947801SPeter Wemm int i; 912a4f67738SDoug Rabson 9131aeb23cdSMarcel Moolenaar /* Notify MD code that a module is being unloaded. */ 9141aeb23cdSMarcel Moolenaar elf_cpu_unload_file(file); 9151aeb23cdSMarcel Moolenaar 91650c202c5SJeff Roberson if (ef->progtab) { 91750c202c5SJeff Roberson for (i = 0; i < ef->nprogtab; i++) { 91850c202c5SJeff Roberson if (ef->progtab[i].size == 0) 91950c202c5SJeff Roberson continue; 92050c202c5SJeff Roberson if (ef->progtab[i].name == NULL) 92150c202c5SJeff Roberson continue; 9225f67450dSDimitry Andric if (!strcmp(ef->progtab[i].name, DPCPU_SETNAME)) 92350c202c5SJeff Roberson dpcpu_free(ef->progtab[i].addr, 92450c202c5SJeff Roberson ef->progtab[i].size); 925eddfbb76SRobert Watson #ifdef VIMAGE 92617ef1febSRobert Watson else if (!strcmp(ef->progtab[i].name, VNET_SETNAME)) 927eddfbb76SRobert Watson vnet_data_free(ef->progtab[i].addr, 928eddfbb76SRobert Watson ef->progtab[i].size); 929eddfbb76SRobert Watson #endif 93050c202c5SJeff Roberson } 93150c202c5SJeff Roberson } 93270b7ffeeSIan Dowse if (ef->preloaded) { 93370b7ffeeSIan Dowse if (ef->reltab) 93470b7ffeeSIan Dowse free(ef->reltab, M_LINKER); 93570b7ffeeSIan Dowse if (ef->relatab) 93670b7ffeeSIan Dowse free(ef->relatab, M_LINKER); 93770b7ffeeSIan Dowse if (ef->progtab) 93870b7ffeeSIan Dowse free(ef->progtab, M_LINKER); 939a2024a3eSJohn Birrell if (ef->ctftab) 940a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 941a2024a3eSJohn Birrell if (ef->ctfoff) 942a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 943a2024a3eSJohn Birrell if (ef->typoff) 944a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 94570b7ffeeSIan Dowse if (file->filename != NULL) 94670b7ffeeSIan Dowse preload_delete_name(file->filename); 94770b7ffeeSIan Dowse /* XXX reclaim module memory? */ 94870b7ffeeSIan Dowse return; 94970b7ffeeSIan Dowse } 95070b7ffeeSIan Dowse 951b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) 95220947801SPeter Wemm if (ef->reltab[i].rel) 95320947801SPeter Wemm free(ef->reltab[i].rel, M_LINKER); 954b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) 95520947801SPeter Wemm if (ef->relatab[i].rela) 95620947801SPeter Wemm free(ef->relatab[i].rela, M_LINKER); 95720947801SPeter Wemm if (ef->reltab) 95820947801SPeter Wemm free(ef->reltab, M_LINKER); 95920947801SPeter Wemm if (ef->relatab) 96020947801SPeter Wemm free(ef->relatab, M_LINKER); 96120947801SPeter Wemm if (ef->progtab) 96220947801SPeter Wemm free(ef->progtab, M_LINKER); 96320947801SPeter Wemm 964fe3db7c7SDoug Rabson if (ef->object) { 965fe3db7c7SDoug Rabson vm_map_remove(kernel_map, (vm_offset_t) ef->address, 96620947801SPeter Wemm (vm_offset_t) ef->address + 96720947801SPeter Wemm (ef->object->size << PAGE_SHIFT)); 968fe3db7c7SDoug Rabson } 969e9eabf59SPeter Wemm if (ef->e_shdr) 970e9eabf59SPeter Wemm free(ef->e_shdr, M_LINKER); 971e9eabf59SPeter Wemm if (ef->ddbsymtab) 972e9eabf59SPeter Wemm free(ef->ddbsymtab, M_LINKER); 973e9eabf59SPeter Wemm if (ef->ddbstrtab) 974e9eabf59SPeter Wemm free(ef->ddbstrtab, M_LINKER); 97520947801SPeter Wemm if (ef->shstrtab) 97620947801SPeter Wemm free(ef->shstrtab, M_LINKER); 977a2024a3eSJohn Birrell if (ef->ctftab) 978a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 979a2024a3eSJohn Birrell if (ef->ctfoff) 980a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 981a2024a3eSJohn Birrell if (ef->typoff) 982a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 983de78ca7eSPeter Wemm } 984de78ca7eSPeter Wemm 985fe3db7c7SDoug Rabson static const char * 986757686b1SMarcel Moolenaar symbol_name(elf_file_t ef, Elf_Size r_info) 987a4f67738SDoug Rabson { 988fe3db7c7SDoug Rabson const Elf_Sym *ref; 989a4f67738SDoug Rabson 990aa855a59SPeter Wemm if (ELF_R_SYM(r_info)) { 991e9eabf59SPeter Wemm ref = ef->ddbsymtab + ELF_R_SYM(r_info); 992e9eabf59SPeter Wemm return ef->ddbstrtab + ref->st_name; 993fe3db7c7SDoug Rabson } else 994fe3db7c7SDoug Rabson return NULL; 995a4f67738SDoug Rabson } 996a4f67738SDoug Rabson 997e9eabf59SPeter Wemm static Elf_Addr 998e9eabf59SPeter Wemm findbase(elf_file_t ef, int sec) 999e9eabf59SPeter Wemm { 1000e9eabf59SPeter Wemm int i; 1001e9eabf59SPeter Wemm Elf_Addr base = 0; 1002e9eabf59SPeter Wemm 1003e9eabf59SPeter Wemm for (i = 0; i < ef->nprogtab; i++) { 1004a8774e39SPeter Wemm if (sec == ef->progtab[i].sec) { 1005e9eabf59SPeter Wemm base = (Elf_Addr)ef->progtab[i].addr; 1006a8774e39SPeter Wemm break; 1007e9eabf59SPeter Wemm } 1008a8774e39SPeter Wemm } 1009e9eabf59SPeter Wemm return base; 1010e9eabf59SPeter Wemm } 1011e9eabf59SPeter Wemm 1012a4f67738SDoug Rabson static int 1013326e27d8SDoug Rabson relocate_file(elf_file_t ef) 1014a4f67738SDoug Rabson { 1015fe3db7c7SDoug Rabson const Elf_Rel *rellim; 1016fe3db7c7SDoug Rabson const Elf_Rel *rel; 1017fe3db7c7SDoug Rabson const Elf_Rela *relalim; 1018fe3db7c7SDoug Rabson const Elf_Rela *rela; 1019aa855a59SPeter Wemm const char *symname; 1020e9eabf59SPeter Wemm const Elf_Sym *sym; 1021e9eabf59SPeter Wemm int i; 1022757686b1SMarcel Moolenaar Elf_Size symidx; 1023e9eabf59SPeter Wemm Elf_Addr base; 1024e9eabf59SPeter Wemm 1025a4f67738SDoug Rabson 1026fe3db7c7SDoug Rabson /* Perform relocations without addend if there are any: */ 1027b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) { 1028e9eabf59SPeter Wemm rel = ef->reltab[i].rel; 102920947801SPeter Wemm if (rel == NULL) 103082d0d1a0SPeter Wemm panic("lost a reltab!"); 103120947801SPeter Wemm rellim = rel + ef->reltab[i].nrel; 1032e9eabf59SPeter Wemm base = findbase(ef, ef->reltab[i].sec); 1033a8774e39SPeter Wemm if (base == 0) 1034a8774e39SPeter Wemm panic("lost base for reltab"); 103520947801SPeter Wemm for ( ; rel < rellim; rel++) { 1036e9eabf59SPeter Wemm symidx = ELF_R_SYM(rel->r_info); 103720947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 103820947801SPeter Wemm continue; 1039e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 104020947801SPeter Wemm /* Local relocs are already done */ 104120947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) 104220947801SPeter Wemm continue; 104320947801SPeter Wemm if (elf_reloc(&ef->lf, base, rel, ELF_RELOC_REL, 104420947801SPeter Wemm elf_obj_lookup)) { 1045aa855a59SPeter Wemm symname = symbol_name(ef, rel->r_info); 104620947801SPeter Wemm printf("link_elf_obj: symbol %s undefined\n", 104720947801SPeter Wemm symname); 1048a4f67738SDoug Rabson return ENOENT; 1049a13ddfb6SPeter Wemm } 1050aa855a59SPeter Wemm } 1051a4f67738SDoug Rabson } 1052a4f67738SDoug Rabson 1053fe3db7c7SDoug Rabson /* Perform relocations with addend if there are any: */ 1054b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) { 1055e9eabf59SPeter Wemm rela = ef->relatab[i].rela; 105620947801SPeter Wemm if (rela == NULL) 105782d0d1a0SPeter Wemm panic("lost a relatab!"); 105820947801SPeter Wemm relalim = rela + ef->relatab[i].nrela; 1059e9eabf59SPeter Wemm base = findbase(ef, ef->relatab[i].sec); 1060a8774e39SPeter Wemm if (base == 0) 1061a8774e39SPeter Wemm panic("lost base for relatab"); 106220947801SPeter Wemm for ( ; rela < relalim; rela++) { 1063e9eabf59SPeter Wemm symidx = ELF_R_SYM(rela->r_info); 106420947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 106520947801SPeter Wemm continue; 1066e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 106720947801SPeter Wemm /* Local relocs are already done */ 106820947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) 106920947801SPeter Wemm continue; 107020947801SPeter Wemm if (elf_reloc(&ef->lf, base, rela, ELF_RELOC_RELA, 107120947801SPeter Wemm elf_obj_lookup)) { 1072aa855a59SPeter Wemm symname = symbol_name(ef, rela->r_info); 107320947801SPeter Wemm printf("link_elf_obj: symbol %s undefined\n", 107420947801SPeter Wemm symname); 1075fe3db7c7SDoug Rabson return ENOENT; 1076a13ddfb6SPeter Wemm } 1077e9eabf59SPeter Wemm } 1078e9eabf59SPeter Wemm } 1079a4f67738SDoug Rabson 10802832cd54SKonstantin Belousov /* 10812832cd54SKonstantin Belousov * Only clean SHN_FBSD_CACHED for successfull return. If we 10822832cd54SKonstantin Belousov * modified symbol table for the object but found an 10832832cd54SKonstantin Belousov * unresolved symbol, there is no reason to roll back. 10842832cd54SKonstantin Belousov */ 10852832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(ef); 10862832cd54SKonstantin Belousov 1087a4f67738SDoug Rabson return 0; 1088a4f67738SDoug Rabson } 1089a4f67738SDoug Rabson 109037c84183SPoul-Henning Kamp static int 1091d254af07SMatthew Dillon link_elf_lookup_symbol(linker_file_t lf, const char *name, c_linker_sym_t *sym) 1092a4f67738SDoug Rabson { 1093326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 10942d636ab0SPeter Wemm const Elf_Sym *symp; 10952d636ab0SPeter Wemm const char *strp; 1096a4f67738SDoug Rabson int i; 1097a4f67738SDoug Rabson 10982d636ab0SPeter Wemm for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 10992d636ab0SPeter Wemm strp = ef->ddbstrtab + symp->st_name; 1100a8774e39SPeter Wemm if (symp->st_shndx != SHN_UNDEF && strcmp(name, strp) == 0) { 1101d254af07SMatthew Dillon *sym = (c_linker_sym_t) symp; 11022d636ab0SPeter Wemm return 0; 11032d636ab0SPeter Wemm } 11042d636ab0SPeter Wemm } 1105a4f67738SDoug Rabson return ENOENT; 1106a4f67738SDoug Rabson } 1107a4f67738SDoug Rabson 1108de78ca7eSPeter Wemm static int 110920947801SPeter Wemm link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym, 111020947801SPeter Wemm linker_symval_t *symval) 1111a4f67738SDoug Rabson { 1112326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 11138aef1712SMatthew Dillon const Elf_Sym *es = (const Elf_Sym*) sym; 1114a4f67738SDoug Rabson 11155cf87418SMarcel Moolenaar if (es >= ef->ddbsymtab && es < (ef->ddbsymtab + ef->ddbsymcnt)) { 11162d636ab0SPeter Wemm symval->name = ef->ddbstrtab + es->st_name; 1117a8774e39SPeter Wemm symval->value = (caddr_t)es->st_value; 11182d636ab0SPeter Wemm symval->size = es->st_size; 11192d636ab0SPeter Wemm return 0; 11202d636ab0SPeter Wemm } 11212d636ab0SPeter Wemm return ENOENT; 11222d636ab0SPeter Wemm } 1123a4f67738SDoug Rabson 1124a4f67738SDoug Rabson static int 1125a4f67738SDoug Rabson link_elf_search_symbol(linker_file_t lf, caddr_t value, 1126d254af07SMatthew Dillon c_linker_sym_t *sym, long *diffp) 1127a4f67738SDoug Rabson { 1128326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 1129586453feSBruce Evans u_long off = (uintptr_t) (void *) value; 1130a4f67738SDoug Rabson u_long diff = off; 1131586453feSBruce Evans u_long st_value; 1132fe3db7c7SDoug Rabson const Elf_Sym *es; 1133fe3db7c7SDoug Rabson const Elf_Sym *best = 0; 1134a4f67738SDoug Rabson int i; 1135a4f67738SDoug Rabson 11362d636ab0SPeter Wemm for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) { 1137a4f67738SDoug Rabson if (es->st_name == 0) 1138a4f67738SDoug Rabson continue; 1139a8774e39SPeter Wemm st_value = es->st_value; 1140b5abfb70SPeter Wemm if (off >= st_value) { 1141b5abfb70SPeter Wemm if (off - st_value < diff) { 1142b5abfb70SPeter Wemm diff = off - st_value; 1143a4f67738SDoug Rabson best = es; 1144a4f67738SDoug Rabson if (diff == 0) 1145a4f67738SDoug Rabson break; 1146b5abfb70SPeter Wemm } else if (off - st_value == diff) { 1147a4f67738SDoug Rabson best = es; 1148a4f67738SDoug Rabson } 1149a4f67738SDoug Rabson } 1150a4f67738SDoug Rabson } 1151a4f67738SDoug Rabson if (best == 0) 1152a4f67738SDoug Rabson *diffp = off; 1153a4f67738SDoug Rabson else 1154a4f67738SDoug Rabson *diffp = diff; 1155d254af07SMatthew Dillon *sym = (c_linker_sym_t) best; 1156a4f67738SDoug Rabson 1157a4f67738SDoug Rabson return 0; 1158a4f67738SDoug Rabson } 1159f41325dbSPeter Wemm 1160f41325dbSPeter Wemm /* 1161f41325dbSPeter Wemm * Look up a linker set on an ELF system. 1162f41325dbSPeter Wemm */ 1163f41325dbSPeter Wemm static int 1164f41325dbSPeter Wemm link_elf_lookup_set(linker_file_t lf, const char *name, 1165f41325dbSPeter Wemm void ***startp, void ***stopp, int *countp) 1166f41325dbSPeter Wemm { 1167e9eabf59SPeter Wemm elf_file_t ef = (elf_file_t)lf; 1168f41325dbSPeter Wemm void **start, **stop; 1169e9eabf59SPeter Wemm int i, count; 1170f41325dbSPeter Wemm 1171e9eabf59SPeter Wemm /* Relative to section number */ 1172e9eabf59SPeter Wemm for (i = 0; i < ef->nprogtab; i++) { 1173e9eabf59SPeter Wemm if ((strncmp(ef->progtab[i].name, "set_", 4) == 0) && 1174e9eabf59SPeter Wemm strcmp(ef->progtab[i].name + 4, name) == 0) { 1175e9eabf59SPeter Wemm start = (void **)ef->progtab[i].addr; 117620947801SPeter Wemm stop = (void **)((char *)ef->progtab[i].addr + 117720947801SPeter Wemm ef->progtab[i].size); 1178f41325dbSPeter Wemm count = stop - start; 1179f41325dbSPeter Wemm if (startp) 1180f41325dbSPeter Wemm *startp = start; 1181f41325dbSPeter Wemm if (stopp) 1182f41325dbSPeter Wemm *stopp = stop; 1183f41325dbSPeter Wemm if (countp) 1184f41325dbSPeter Wemm *countp = count; 1185e9eabf59SPeter Wemm return (0); 1186e9eabf59SPeter Wemm } 1187e9eabf59SPeter Wemm } 1188e9eabf59SPeter Wemm return (ESRCH); 1189f41325dbSPeter Wemm } 1190bb9fe9ddSBrian Feldman 1191bb9fe9ddSBrian Feldman static int 1192bb9fe9ddSBrian Feldman link_elf_each_function_name(linker_file_t file, 1193e9eabf59SPeter Wemm int (*callback)(const char *, void *), void *opaque) 1194e9eabf59SPeter Wemm { 1195bb9fe9ddSBrian Feldman elf_file_t ef = (elf_file_t)file; 1196bb9fe9ddSBrian Feldman const Elf_Sym *symp; 1197bb9fe9ddSBrian Feldman int i, error; 1198bb9fe9ddSBrian Feldman 1199bb9fe9ddSBrian Feldman /* Exhaustive search */ 1200bb9fe9ddSBrian Feldman for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1201bb9fe9ddSBrian Feldman if (symp->st_value != 0 && 1202bb9fe9ddSBrian Feldman ELF_ST_TYPE(symp->st_info) == STT_FUNC) { 1203bb9fe9ddSBrian Feldman error = callback(ef->ddbstrtab + symp->st_name, opaque); 1204bb9fe9ddSBrian Feldman if (error) 1205bb9fe9ddSBrian Feldman return (error); 1206bb9fe9ddSBrian Feldman } 1207bb9fe9ddSBrian Feldman } 1208bb9fe9ddSBrian Feldman return (0); 1209bb9fe9ddSBrian Feldman } 121084201059SMarcel Moolenaar 1211a2024a3eSJohn Birrell static int 1212a2024a3eSJohn Birrell link_elf_each_function_nameval(linker_file_t file, 1213a2024a3eSJohn Birrell linker_function_nameval_callback_t callback, void *opaque) 1214a2024a3eSJohn Birrell { 1215a2024a3eSJohn Birrell linker_symval_t symval; 1216a2024a3eSJohn Birrell elf_file_t ef = (elf_file_t)file; 1217a2024a3eSJohn Birrell const Elf_Sym* symp; 1218a2024a3eSJohn Birrell int i, error; 1219a2024a3eSJohn Birrell 1220a2024a3eSJohn Birrell /* Exhaustive search */ 1221a2024a3eSJohn Birrell for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1222a2024a3eSJohn Birrell if (symp->st_value != 0 && 1223a2024a3eSJohn Birrell ELF_ST_TYPE(symp->st_info) == STT_FUNC) { 1224a2024a3eSJohn Birrell error = link_elf_symbol_values(file, (c_linker_sym_t) symp, &symval); 1225a2024a3eSJohn Birrell if (error) 1226a2024a3eSJohn Birrell return (error); 1227a2024a3eSJohn Birrell error = callback(file, i, &symval, opaque); 1228a2024a3eSJohn Birrell if (error) 1229a2024a3eSJohn Birrell return (error); 1230a2024a3eSJohn Birrell } 1231a2024a3eSJohn Birrell } 1232a2024a3eSJohn Birrell return (0); 1233a2024a3eSJohn Birrell } 1234a2024a3eSJohn Birrell 12352832cd54SKonstantin Belousov static void 12362832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(elf_file_t ef) 12372832cd54SKonstantin Belousov { 12382832cd54SKonstantin Belousov Elf_Sym *sym; 12392832cd54SKonstantin Belousov Elf_Size i; 12402832cd54SKonstantin Belousov 12412832cd54SKonstantin Belousov for (i = 0; i < ef->ddbsymcnt; i++) { 12422832cd54SKonstantin Belousov sym = ef->ddbsymtab + i; 12432832cd54SKonstantin Belousov if (sym->st_shndx == SHN_FBSD_CACHED) { 12442832cd54SKonstantin Belousov sym->st_shndx = SHN_UNDEF; 12452832cd54SKonstantin Belousov sym->st_value = 0; 12462832cd54SKonstantin Belousov } 12472832cd54SKonstantin Belousov } 12482832cd54SKonstantin Belousov } 12492832cd54SKonstantin Belousov 1250d297ad16SMarcel Moolenaar /* 1251d297ad16SMarcel Moolenaar * Symbol lookup function that can be used when the symbol index is known (ie 1252d297ad16SMarcel Moolenaar * in relocations). It uses the symbol index instead of doing a fully fledged 1253d297ad16SMarcel Moolenaar * hash table based lookup when such is valid. For example for local symbols. 1254d297ad16SMarcel Moolenaar * This is not only more efficient, it's also more correct. It's not always 1255d297ad16SMarcel Moolenaar * the case that the symbol can be found through the hash table. 1256d297ad16SMarcel Moolenaar */ 1257*cff8c6f2SKonstantin Belousov static int 1258*cff8c6f2SKonstantin Belousov elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, Elf_Addr *res) 1259d297ad16SMarcel Moolenaar { 1260d297ad16SMarcel Moolenaar elf_file_t ef = (elf_file_t)lf; 12612832cd54SKonstantin Belousov Elf_Sym *sym; 1262d297ad16SMarcel Moolenaar const char *symbol; 1263*cff8c6f2SKonstantin Belousov Elf_Addr res1; 1264d297ad16SMarcel Moolenaar 1265d297ad16SMarcel Moolenaar /* Don't even try to lookup the symbol if the index is bogus. */ 1266*cff8c6f2SKonstantin Belousov if (symidx >= ef->ddbsymcnt) { 1267*cff8c6f2SKonstantin Belousov *res = 0; 1268*cff8c6f2SKonstantin Belousov return (EINVAL); 1269*cff8c6f2SKonstantin Belousov } 1270d297ad16SMarcel Moolenaar 1271e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 1272d297ad16SMarcel Moolenaar 12734cec6f5dSPeter Wemm /* Quick answer if there is a definition included. */ 1274*cff8c6f2SKonstantin Belousov if (sym->st_shndx != SHN_UNDEF) { 1275*cff8c6f2SKonstantin Belousov *res = sym->st_value; 1276*cff8c6f2SKonstantin Belousov return (0); 1277*cff8c6f2SKonstantin Belousov } 12784cec6f5dSPeter Wemm 12794cec6f5dSPeter Wemm /* If we get here, then it is undefined and needs a lookup. */ 12804cec6f5dSPeter Wemm switch (ELF_ST_BIND(sym->st_info)) { 12814cec6f5dSPeter Wemm case STB_LOCAL: 12824cec6f5dSPeter Wemm /* Local, but undefined? huh? */ 1283*cff8c6f2SKonstantin Belousov *res = 0; 1284*cff8c6f2SKonstantin Belousov return (EINVAL); 1285d297ad16SMarcel Moolenaar 1286e9eabf59SPeter Wemm case STB_GLOBAL: 1287*cff8c6f2SKonstantin Belousov case STB_WEAK: 1288e9eabf59SPeter Wemm /* Relative to Data or Function name */ 1289e9eabf59SPeter Wemm symbol = ef->ddbstrtab + sym->st_name; 1290d297ad16SMarcel Moolenaar 1291d297ad16SMarcel Moolenaar /* Force a lookup failure if the symbol name is bogus. */ 1292*cff8c6f2SKonstantin Belousov if (*symbol == 0) { 1293*cff8c6f2SKonstantin Belousov *res = 0; 1294*cff8c6f2SKonstantin Belousov return (EINVAL); 1295*cff8c6f2SKonstantin Belousov } 1296*cff8c6f2SKonstantin Belousov res1 = (Elf_Addr)linker_file_lookup_symbol(lf, symbol, deps); 12972832cd54SKonstantin Belousov 12982832cd54SKonstantin Belousov /* 12992832cd54SKonstantin Belousov * Cache global lookups during module relocation. The failure 13002832cd54SKonstantin Belousov * case is particularly expensive for callers, who must scan 13012832cd54SKonstantin Belousov * through the entire globals table doing strcmp(). Cache to 13022832cd54SKonstantin Belousov * avoid doing such work repeatedly. 13032832cd54SKonstantin Belousov * 13042832cd54SKonstantin Belousov * After relocation is complete, undefined globals will be 13052832cd54SKonstantin Belousov * restored to SHN_UNDEF in elf_obj_cleanup_globals_cache(), 13062832cd54SKonstantin Belousov * above. 13072832cd54SKonstantin Belousov */ 1308*cff8c6f2SKonstantin Belousov if (res1 != 0) { 13092832cd54SKonstantin Belousov sym->st_shndx = SHN_FBSD_CACHED; 1310*cff8c6f2SKonstantin Belousov sym->st_value = res1; 1311*cff8c6f2SKonstantin Belousov *res = res1; 1312e9eabf59SPeter Wemm return (0); 1313*cff8c6f2SKonstantin Belousov } else if (ELF_ST_BIND(sym->st_info) == STB_WEAK) { 1314*cff8c6f2SKonstantin Belousov sym->st_value = 0; 1315*cff8c6f2SKonstantin Belousov *res = 0; 1316*cff8c6f2SKonstantin Belousov return (0); 1317*cff8c6f2SKonstantin Belousov } 1318*cff8c6f2SKonstantin Belousov return (EINVAL); 1319e9eabf59SPeter Wemm 1320e9eabf59SPeter Wemm default: 1321*cff8c6f2SKonstantin Belousov return (EINVAL); 1322d297ad16SMarcel Moolenaar } 1323e9eabf59SPeter Wemm } 1324e9eabf59SPeter Wemm 13257251b4bfSJake Burkholder static void 13267226306eSKonstantin Belousov link_elf_fix_link_set(elf_file_t ef) 13277226306eSKonstantin Belousov { 13287226306eSKonstantin Belousov static const char startn[] = "__start_"; 13297226306eSKonstantin Belousov static const char stopn[] = "__stop_"; 13307226306eSKonstantin Belousov Elf_Sym *sym; 13317226306eSKonstantin Belousov const char *sym_name, *linkset_name; 13327226306eSKonstantin Belousov Elf_Addr startp, stopp; 13337226306eSKonstantin Belousov Elf_Size symidx; 13347226306eSKonstantin Belousov int start, i; 13357226306eSKonstantin Belousov 13367226306eSKonstantin Belousov startp = stopp = 0; 13377226306eSKonstantin Belousov for (symidx = 1 /* zero entry is special */; 13387226306eSKonstantin Belousov symidx < ef->ddbsymcnt; symidx++) { 13397226306eSKonstantin Belousov sym = ef->ddbsymtab + symidx; 13407226306eSKonstantin Belousov if (sym->st_shndx != SHN_UNDEF) 13417226306eSKonstantin Belousov continue; 13427226306eSKonstantin Belousov 13437226306eSKonstantin Belousov sym_name = ef->ddbstrtab + sym->st_name; 13447226306eSKonstantin Belousov if (strncmp(sym_name, startn, sizeof(startn) - 1) == 0) { 13457226306eSKonstantin Belousov start = 1; 13467226306eSKonstantin Belousov linkset_name = sym_name + sizeof(startn) - 1; 13477226306eSKonstantin Belousov } 13487226306eSKonstantin Belousov else if (strncmp(sym_name, stopn, sizeof(stopn) - 1) == 0) { 13497226306eSKonstantin Belousov start = 0; 13507226306eSKonstantin Belousov linkset_name = sym_name + sizeof(stopn) - 1; 13517226306eSKonstantin Belousov } 13527226306eSKonstantin Belousov else 13537226306eSKonstantin Belousov continue; 13547226306eSKonstantin Belousov 13557226306eSKonstantin Belousov for (i = 0; i < ef->nprogtab; i++) { 13567226306eSKonstantin Belousov if (strcmp(ef->progtab[i].name, linkset_name) == 0) { 13577226306eSKonstantin Belousov startp = (Elf_Addr)ef->progtab[i].addr; 13587226306eSKonstantin Belousov stopp = (Elf_Addr)(startp + ef->progtab[i].size); 13597226306eSKonstantin Belousov break; 13607226306eSKonstantin Belousov } 13617226306eSKonstantin Belousov } 13627226306eSKonstantin Belousov if (i == ef->nprogtab) 13637226306eSKonstantin Belousov continue; 13647226306eSKonstantin Belousov 13657226306eSKonstantin Belousov sym->st_value = start ? startp : stopp; 13667226306eSKonstantin Belousov sym->st_shndx = i; 13677226306eSKonstantin Belousov } 13687226306eSKonstantin Belousov } 13697226306eSKonstantin Belousov 13707226306eSKonstantin Belousov static void 13717251b4bfSJake Burkholder link_elf_reloc_local(linker_file_t lf) 13727251b4bfSJake Burkholder { 1373e9eabf59SPeter Wemm elf_file_t ef = (elf_file_t)lf; 13747251b4bfSJake Burkholder const Elf_Rel *rellim; 13757251b4bfSJake Burkholder const Elf_Rel *rel; 13767251b4bfSJake Burkholder const Elf_Rela *relalim; 13777251b4bfSJake Burkholder const Elf_Rela *rela; 1378e9eabf59SPeter Wemm const Elf_Sym *sym; 1379e9eabf59SPeter Wemm Elf_Addr base; 1380e9eabf59SPeter Wemm int i; 1381757686b1SMarcel Moolenaar Elf_Size symidx; 1382e9eabf59SPeter Wemm 13837226306eSKonstantin Belousov link_elf_fix_link_set(ef); 13847226306eSKonstantin Belousov 13857251b4bfSJake Burkholder /* Perform relocations without addend if there are any: */ 1386b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) { 1387e9eabf59SPeter Wemm rel = ef->reltab[i].rel; 138820947801SPeter Wemm if (rel == NULL) 138982d0d1a0SPeter Wemm panic("lost a reltab!"); 139020947801SPeter Wemm rellim = rel + ef->reltab[i].nrel; 1391e9eabf59SPeter Wemm base = findbase(ef, ef->reltab[i].sec); 1392a8774e39SPeter Wemm if (base == 0) 1393a8774e39SPeter Wemm panic("lost base for reltab"); 139420947801SPeter Wemm for ( ; rel < rellim; rel++) { 1395e9eabf59SPeter Wemm symidx = ELF_R_SYM(rel->r_info); 139620947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 139720947801SPeter Wemm continue; 1398e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 139920947801SPeter Wemm /* Only do local relocs */ 140020947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) != STB_LOCAL) 140120947801SPeter Wemm continue; 140220947801SPeter Wemm elf_reloc_local(lf, base, rel, ELF_RELOC_REL, 140320947801SPeter Wemm elf_obj_lookup); 14047251b4bfSJake Burkholder } 1405e9eabf59SPeter Wemm } 14067251b4bfSJake Burkholder 14077251b4bfSJake Burkholder /* Perform relocations with addend if there are any: */ 1408b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) { 1409e9eabf59SPeter Wemm rela = ef->relatab[i].rela; 141020947801SPeter Wemm if (rela == NULL) 141182d0d1a0SPeter Wemm panic("lost a relatab!"); 141220947801SPeter Wemm relalim = rela + ef->relatab[i].nrela; 1413e9eabf59SPeter Wemm base = findbase(ef, ef->relatab[i].sec); 1414a8774e39SPeter Wemm if (base == 0) 1415a8774e39SPeter Wemm panic("lost base for relatab"); 141620947801SPeter Wemm for ( ; rela < relalim; rela++) { 1417e9eabf59SPeter Wemm symidx = ELF_R_SYM(rela->r_info); 141820947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 141920947801SPeter Wemm continue; 1420e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 142120947801SPeter Wemm /* Only do local relocs */ 142220947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) != STB_LOCAL) 142320947801SPeter Wemm continue; 142420947801SPeter Wemm elf_reloc_local(lf, base, rela, ELF_RELOC_RELA, 142520947801SPeter Wemm elf_obj_lookup); 14267251b4bfSJake Burkholder } 14277251b4bfSJake Burkholder } 1428e9eabf59SPeter Wemm } 142900a5db46SStacey Son 143000a5db46SStacey Son static long 1431e76f11f4SAndriy Gapon link_elf_symtab_get(linker_file_t lf, const Elf_Sym **symtab) 143200a5db46SStacey Son { 143300a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 143400a5db46SStacey Son 143500a5db46SStacey Son *symtab = ef->ddbsymtab; 143600a5db46SStacey Son 143700a5db46SStacey Son if (*symtab == NULL) 143800a5db46SStacey Son return (0); 143900a5db46SStacey Son 144000a5db46SStacey Son return (ef->ddbsymcnt); 144100a5db46SStacey Son } 144200a5db46SStacey Son 144300a5db46SStacey Son static long 144400a5db46SStacey Son link_elf_strtab_get(linker_file_t lf, caddr_t *strtab) 144500a5db46SStacey Son { 144600a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 144700a5db46SStacey Son 144800a5db46SStacey Son *strtab = ef->ddbstrtab; 144900a5db46SStacey Son 145000a5db46SStacey Son if (*strtab == NULL) 145100a5db46SStacey Son return (0); 145200a5db46SStacey Son 145300a5db46SStacey Son return (ef->ddbstrcnt); 145400a5db46SStacey Son } 1455