1a4f67738SDoug Rabson /*- 28a36da99SPedro F. Giffuni * SPDX-License-Identifier: BSD-2-Clause-FreeBSD 38a36da99SPedro F. Giffuni * 4326e27d8SDoug Rabson * Copyright (c) 1998-2000 Doug Rabson 5e9eabf59SPeter Wemm * Copyright (c) 2004 Peter Wemm 6a4f67738SDoug Rabson * All rights reserved. 7a4f67738SDoug Rabson * 8a4f67738SDoug Rabson * Redistribution and use in source and binary forms, with or without 9a4f67738SDoug Rabson * modification, are permitted provided that the following conditions 10a4f67738SDoug Rabson * are met: 11a4f67738SDoug Rabson * 1. Redistributions of source code must retain the above copyright 12a4f67738SDoug Rabson * notice, this list of conditions and the following disclaimer. 13a4f67738SDoug Rabson * 2. Redistributions in binary form must reproduce the above copyright 14a4f67738SDoug Rabson * notice, this list of conditions and the following disclaimer in the 15a4f67738SDoug Rabson * documentation and/or other materials provided with the distribution. 16a4f67738SDoug Rabson * 17a4f67738SDoug Rabson * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 18a4f67738SDoug Rabson * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 19a4f67738SDoug Rabson * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 20a4f67738SDoug Rabson * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 21a4f67738SDoug Rabson * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 22a4f67738SDoug Rabson * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 23a4f67738SDoug Rabson * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 24a4f67738SDoug Rabson * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 25a4f67738SDoug Rabson * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 26a4f67738SDoug Rabson * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 27a4f67738SDoug Rabson * SUCH DAMAGE. 28a4f67738SDoug Rabson */ 29a4f67738SDoug Rabson 30677b542eSDavid E. O'Brien #include <sys/cdefs.h> 31677b542eSDavid E. O'Brien __FBSDID("$FreeBSD$"); 32677b542eSDavid E. O'Brien 33326e27d8SDoug Rabson #include "opt_ddb.h" 34326e27d8SDoug Rabson 35a4f67738SDoug Rabson #include <sys/param.h> 36a4f67738SDoug Rabson #include <sys/systm.h> 37fbf2a778SKonstantin Belousov #include <sys/fcntl.h> 38fb919e4dSMark Murray #include <sys/kernel.h> 39fb919e4dSMark Murray #include <sys/lock.h> 40a4f67738SDoug Rabson #include <sys/malloc.h> 41fbf2a778SKonstantin Belousov #include <sys/linker.h> 42c0b824f9SDima Dorfman #include <sys/mutex.h> 4362d615d5SJohn Baldwin #include <sys/mount.h> 44a4f67738SDoug Rabson #include <sys/namei.h> 45fbf2a778SKonstantin Belousov #include <sys/proc.h> 46fbf2a778SKonstantin Belousov #include <sys/rwlock.h> 47a4f67738SDoug Rabson #include <sys/vnode.h> 48fb919e4dSMark Murray 49a4f67738SDoug Rabson #include <machine/elf.h> 50a4f67738SDoug Rabson 51eddfbb76SRobert Watson #include <net/vnet.h> 52eddfbb76SRobert Watson 53aed55708SRobert Watson #include <security/mac/mac_framework.h> 54aed55708SRobert Watson 55fe3db7c7SDoug Rabson #include <vm/vm.h> 56fe3db7c7SDoug Rabson #include <vm/vm_param.h> 57fe3db7c7SDoug Rabson #include <vm/pmap.h> 58fbf2a778SKonstantin Belousov #include <vm/vm_extern.h> 59fbf2a778SKonstantin Belousov #include <vm/vm_kern.h> 60fe3db7c7SDoug Rabson #include <vm/vm_map.h> 61fbf2a778SKonstantin Belousov #include <vm/vm_object.h> 62fbf2a778SKonstantin Belousov #include <vm/vm_page.h> 63fbf2a778SKonstantin Belousov #include <vm/vm_pager.h> 64fb919e4dSMark Murray 65f9980387SPeter Wemm #include <sys/link_elf.h> 66fe3db7c7SDoug Rabson 67a2024a3eSJohn Birrell #ifdef DDB_CTF 6822bbc4b2SXin LI #include <contrib/zlib/zlib.h> 69a2024a3eSJohn Birrell #endif 70a2024a3eSJohn Birrell 71326e27d8SDoug Rabson #include "linker_if.h" 72a4f67738SDoug Rabson 73e9eabf59SPeter Wemm typedef struct { 74e9eabf59SPeter Wemm void *addr; 7520947801SPeter Wemm Elf_Off size; 761d9eae9fSMark Johnston int flags; /* Section flags. */ 771d9eae9fSMark Johnston int sec; /* Original section number. */ 78e9eabf59SPeter Wemm char *name; 79e9eabf59SPeter Wemm } Elf_progent; 80e9eabf59SPeter Wemm 81e9eabf59SPeter Wemm typedef struct { 82e9eabf59SPeter Wemm Elf_Rel *rel; 8320947801SPeter Wemm int nrel; 84e9eabf59SPeter Wemm int sec; 85e9eabf59SPeter Wemm } Elf_relent; 86e9eabf59SPeter Wemm 87e9eabf59SPeter Wemm typedef struct { 88e9eabf59SPeter Wemm Elf_Rela *rela; 8920947801SPeter Wemm int nrela; 90e9eabf59SPeter Wemm int sec; 91e9eabf59SPeter Wemm } Elf_relaent; 92e9eabf59SPeter Wemm 93a4f67738SDoug Rabson typedef struct elf_file { 94326e27d8SDoug Rabson struct linker_file lf; /* Common fields */ 9520947801SPeter Wemm 9670b7ffeeSIan Dowse int preloaded; 97fe3db7c7SDoug Rabson caddr_t address; /* Relocation address */ 98fe3db7c7SDoug Rabson vm_object_t object; /* VM object to hold file pages */ 99e9eabf59SPeter Wemm Elf_Shdr *e_shdr; 100e9eabf59SPeter Wemm 101e9eabf59SPeter Wemm Elf_progent *progtab; 102d821d364SPedro F. Giffuni u_int nprogtab; 103e9eabf59SPeter Wemm 104e9eabf59SPeter Wemm Elf_relaent *relatab; 105d821d364SPedro F. Giffuni u_int nrelatab; 106e9eabf59SPeter Wemm 107e9eabf59SPeter Wemm Elf_relent *reltab; 108b474c780SDavid E. O'Brien int nreltab; 109e9eabf59SPeter Wemm 110e9eabf59SPeter Wemm Elf_Sym *ddbsymtab; /* The symbol table we are using */ 1112d636ab0SPeter Wemm long ddbsymcnt; /* Number of symbols */ 1122d636ab0SPeter Wemm caddr_t ddbstrtab; /* String table */ 1132d636ab0SPeter Wemm long ddbstrcnt; /* number of bytes in string table */ 114e9eabf59SPeter Wemm 115e9eabf59SPeter Wemm caddr_t shstrtab; /* Section name string table */ 116e9eabf59SPeter Wemm long shstrcnt; /* number of bytes in string table */ 117e9eabf59SPeter Wemm 118a2024a3eSJohn Birrell caddr_t ctftab; /* CTF table */ 119a2024a3eSJohn Birrell long ctfcnt; /* number of bytes in CTF table */ 120a2024a3eSJohn Birrell caddr_t ctfoff; /* CTF offset table */ 121a2024a3eSJohn Birrell caddr_t typoff; /* Type offset table */ 122a2024a3eSJohn Birrell long typlen; /* Number of type entries. */ 123a2024a3eSJohn Birrell 124a4f67738SDoug Rabson } *elf_file_t; 125a4f67738SDoug Rabson 126a2024a3eSJohn Birrell #include <kern/kern_ctf.c> 127a2024a3eSJohn Birrell 12854823af2SPeter Wemm static int link_elf_link_preload(linker_class_t cls, 129326e27d8SDoug Rabson const char *, linker_file_t *); 13054823af2SPeter Wemm static int link_elf_link_preload_finish(linker_file_t); 13154823af2SPeter Wemm static int link_elf_load_file(linker_class_t, const char *, linker_file_t *); 132326e27d8SDoug Rabson static int link_elf_lookup_symbol(linker_file_t, const char *, 133326e27d8SDoug Rabson c_linker_sym_t *); 134e9eabf59SPeter Wemm static int link_elf_symbol_values(linker_file_t, c_linker_sym_t, 135e9eabf59SPeter Wemm linker_symval_t *); 136326e27d8SDoug Rabson static int link_elf_search_symbol(linker_file_t, caddr_t value, 137326e27d8SDoug Rabson c_linker_sym_t *sym, long *diffp); 138326e27d8SDoug Rabson 139326e27d8SDoug Rabson static void link_elf_unload_file(linker_file_t); 140f41325dbSPeter Wemm static int link_elf_lookup_set(linker_file_t, const char *, 141f41325dbSPeter Wemm void ***, void ***, int *); 142bb9fe9ddSBrian Feldman static int link_elf_each_function_name(linker_file_t, 143e9eabf59SPeter Wemm int (*)(const char *, void *), void *); 144a2024a3eSJohn Birrell static int link_elf_each_function_nameval(linker_file_t, 145a2024a3eSJohn Birrell linker_function_nameval_callback_t, 146a2024a3eSJohn Birrell void *); 147289ead7cSKonstantin Belousov static int link_elf_reloc_local(linker_file_t, bool); 148e76f11f4SAndriy Gapon static long link_elf_symtab_get(linker_file_t, const Elf_Sym **); 14900a5db46SStacey Son static long link_elf_strtab_get(linker_file_t, caddr_t *); 150326e27d8SDoug Rabson 151cff8c6f2SKonstantin Belousov static int elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, 152cff8c6f2SKonstantin Belousov Elf_Addr *); 153e9eabf59SPeter Wemm 154326e27d8SDoug Rabson static kobj_method_t link_elf_methods[] = { 155326e27d8SDoug Rabson KOBJMETHOD(linker_lookup_symbol, link_elf_lookup_symbol), 156326e27d8SDoug Rabson KOBJMETHOD(linker_symbol_values, link_elf_symbol_values), 157326e27d8SDoug Rabson KOBJMETHOD(linker_search_symbol, link_elf_search_symbol), 158326e27d8SDoug Rabson KOBJMETHOD(linker_unload, link_elf_unload_file), 15954823af2SPeter Wemm KOBJMETHOD(linker_load_file, link_elf_load_file), 16054823af2SPeter Wemm KOBJMETHOD(linker_link_preload, link_elf_link_preload), 16154823af2SPeter Wemm KOBJMETHOD(linker_link_preload_finish, link_elf_link_preload_finish), 162f41325dbSPeter Wemm KOBJMETHOD(linker_lookup_set, link_elf_lookup_set), 163bb9fe9ddSBrian Feldman KOBJMETHOD(linker_each_function_name, link_elf_each_function_name), 164a2024a3eSJohn Birrell KOBJMETHOD(linker_each_function_nameval, link_elf_each_function_nameval), 165a2024a3eSJohn Birrell KOBJMETHOD(linker_ctf_get, link_elf_ctf_get), 16600a5db46SStacey Son KOBJMETHOD(linker_symtab_get, link_elf_symtab_get), 16700a5db46SStacey Son KOBJMETHOD(linker_strtab_get, link_elf_strtab_get), 1686d775f0bSMark Johnston KOBJMETHOD_END 169326e27d8SDoug Rabson }; 170326e27d8SDoug Rabson 171326e27d8SDoug Rabson static struct linker_class link_elf_class = { 172326e27d8SDoug Rabson #if ELF_TARG_CLASS == ELFCLASS32 173e9eabf59SPeter Wemm "elf32_obj", 174326e27d8SDoug Rabson #else 175e9eabf59SPeter Wemm "elf64_obj", 176326e27d8SDoug Rabson #endif 177326e27d8SDoug Rabson link_elf_methods, sizeof(struct elf_file) 178326e27d8SDoug Rabson }; 179326e27d8SDoug Rabson 180326e27d8SDoug Rabson static int relocate_file(elf_file_t ef); 1812832cd54SKonstantin Belousov static void elf_obj_cleanup_globals_cache(elf_file_t); 182a4f67738SDoug Rabson 183a4f67738SDoug Rabson static void 184552f9f63SEdwin Groothuis link_elf_error(const char *filename, const char *s) 1851720979bSMarcel Moolenaar { 186552f9f63SEdwin Groothuis if (filename == NULL) 1871720979bSMarcel Moolenaar printf("kldload: %s\n", s); 188552f9f63SEdwin Groothuis else 189552f9f63SEdwin Groothuis printf("kldload: %s: %s\n", filename, s); 1901720979bSMarcel Moolenaar } 1911720979bSMarcel Moolenaar 1921720979bSMarcel Moolenaar static void 193a4f67738SDoug Rabson link_elf_init(void *arg) 194a4f67738SDoug Rabson { 195a4f67738SDoug Rabson 196326e27d8SDoug Rabson linker_add_class(&link_elf_class); 197a4f67738SDoug Rabson } 198891cf3edSEd Maste SYSINIT(link_elf_obj, SI_SUB_KLD, SI_ORDER_SECOND, link_elf_init, NULL); 199a4f67738SDoug Rabson 2001d9eae9fSMark Johnston static void 2011d9eae9fSMark Johnston link_elf_protect_range(elf_file_t ef, vm_offset_t start, vm_offset_t end, 2021d9eae9fSMark Johnston vm_prot_t prot) 2031d9eae9fSMark Johnston { 2041d9eae9fSMark Johnston int error __unused; 2051d9eae9fSMark Johnston 2061d9eae9fSMark Johnston KASSERT(start <= end && start >= (vm_offset_t)ef->address && 2071d9eae9fSMark Johnston end <= round_page((vm_offset_t)ef->address + ef->lf.size), 2081d9eae9fSMark Johnston ("link_elf_protect_range: invalid range %#jx-%#jx", 2091d9eae9fSMark Johnston (uintmax_t)start, (uintmax_t)end)); 2101d9eae9fSMark Johnston 2111d9eae9fSMark Johnston if (start == end) 2121d9eae9fSMark Johnston return; 213f822c9e2SMark Johnston if (ef->preloaded) { 214f822c9e2SMark Johnston #ifdef __amd64__ 215f822c9e2SMark Johnston error = pmap_change_prot(start, end - start, prot); 216f822c9e2SMark Johnston KASSERT(error == 0, 217f822c9e2SMark Johnston ("link_elf_protect_range: pmap_change_prot() returned %d", 218f822c9e2SMark Johnston error)); 219f822c9e2SMark Johnston #endif 220f822c9e2SMark Johnston return; 221f822c9e2SMark Johnston } 222*0659df6fSKonstantin Belousov error = vm_map_protect(kernel_map, start, end, prot, 0, 223*0659df6fSKonstantin Belousov VM_MAP_PROTECT_SET_PROT); 2241d9eae9fSMark Johnston KASSERT(error == KERN_SUCCESS, 2251d9eae9fSMark Johnston ("link_elf_protect_range: vm_map_protect() returned %d", error)); 2261d9eae9fSMark Johnston } 2271d9eae9fSMark Johnston 2281d9eae9fSMark Johnston /* 2291d9eae9fSMark Johnston * Restrict permissions on linker file memory based on section flags. 2301d9eae9fSMark Johnston * Sections need not be page-aligned, so overlap within a page is possible. 2311d9eae9fSMark Johnston */ 2321d9eae9fSMark Johnston static void 2331d9eae9fSMark Johnston link_elf_protect(elf_file_t ef) 2341d9eae9fSMark Johnston { 2351d9eae9fSMark Johnston vm_offset_t end, segend, segstart, start; 2361d9eae9fSMark Johnston vm_prot_t gapprot, prot, segprot; 2371d9eae9fSMark Johnston int i; 2381d9eae9fSMark Johnston 2391d9eae9fSMark Johnston /* 2401d9eae9fSMark Johnston * If the file was preloaded, the last page may contain other preloaded 2411d9eae9fSMark Johnston * data which may need to be writeable. ELF files are always 2421d9eae9fSMark Johnston * page-aligned, but other preloaded data, such as entropy or CPU 2431d9eae9fSMark Johnston * microcode may be loaded with a smaller alignment. 2441d9eae9fSMark Johnston */ 2451d9eae9fSMark Johnston gapprot = ef->preloaded ? VM_PROT_RW : VM_PROT_READ; 2461d9eae9fSMark Johnston 2471d9eae9fSMark Johnston start = end = (vm_offset_t)ef->address; 2481d9eae9fSMark Johnston prot = VM_PROT_READ; 2491d9eae9fSMark Johnston for (i = 0; i < ef->nprogtab; i++) { 2501d9eae9fSMark Johnston /* 2511d9eae9fSMark Johnston * VNET and DPCPU sections have their memory allocated by their 2521d9eae9fSMark Johnston * respective subsystems. 2531d9eae9fSMark Johnston */ 2541d9eae9fSMark Johnston if (ef->progtab[i].name != NULL && ( 2551d9eae9fSMark Johnston #ifdef VIMAGE 2561d9eae9fSMark Johnston strcmp(ef->progtab[i].name, VNET_SETNAME) == 0 || 2571d9eae9fSMark Johnston #endif 2581d9eae9fSMark Johnston strcmp(ef->progtab[i].name, DPCPU_SETNAME) == 0)) 2591d9eae9fSMark Johnston continue; 2601d9eae9fSMark Johnston 2611d9eae9fSMark Johnston segstart = trunc_page((vm_offset_t)ef->progtab[i].addr); 2621d9eae9fSMark Johnston segend = round_page((vm_offset_t)ef->progtab[i].addr + 2631d9eae9fSMark Johnston ef->progtab[i].size); 2641d9eae9fSMark Johnston segprot = VM_PROT_READ; 2651d9eae9fSMark Johnston if ((ef->progtab[i].flags & SHF_WRITE) != 0) 2661d9eae9fSMark Johnston segprot |= VM_PROT_WRITE; 2671d9eae9fSMark Johnston if ((ef->progtab[i].flags & SHF_EXECINSTR) != 0) 2681d9eae9fSMark Johnston segprot |= VM_PROT_EXECUTE; 2691d9eae9fSMark Johnston 2701d9eae9fSMark Johnston if (end <= segstart) { 2711d9eae9fSMark Johnston /* 2721d9eae9fSMark Johnston * Case 1: there is no overlap between the previous 2731d9eae9fSMark Johnston * segment and this one. Apply protections to the 2741d9eae9fSMark Johnston * previous segment, and protect the gap between the 2751d9eae9fSMark Johnston * previous and current segments, if any. 2761d9eae9fSMark Johnston */ 2771d9eae9fSMark Johnston link_elf_protect_range(ef, start, end, prot); 2781d9eae9fSMark Johnston link_elf_protect_range(ef, end, segstart, gapprot); 2791d9eae9fSMark Johnston 2801d9eae9fSMark Johnston start = segstart; 2811d9eae9fSMark Johnston end = segend; 2821d9eae9fSMark Johnston prot = segprot; 2831d9eae9fSMark Johnston } else if (start < segstart && end == segend) { 2841d9eae9fSMark Johnston /* 2851d9eae9fSMark Johnston * Case 2: the current segment is a subrange of the 2861d9eae9fSMark Johnston * previous segment. Apply protections to the 2871d9eae9fSMark Johnston * non-overlapping portion of the previous segment. 2881d9eae9fSMark Johnston */ 2891d9eae9fSMark Johnston link_elf_protect_range(ef, start, segstart, prot); 2901d9eae9fSMark Johnston 2911d9eae9fSMark Johnston start = segstart; 2921d9eae9fSMark Johnston prot |= segprot; 2931d9eae9fSMark Johnston } else if (end < segend) { 2941d9eae9fSMark Johnston /* 2951d9eae9fSMark Johnston * Case 3: there is partial overlap between the previous 2961d9eae9fSMark Johnston * and current segments. Apply protections to the 2971d9eae9fSMark Johnston * non-overlapping portion of the previous segment, and 2981d9eae9fSMark Johnston * then the overlap, which must use the union of the two 2991d9eae9fSMark Johnston * segments' protections. 3001d9eae9fSMark Johnston */ 3011d9eae9fSMark Johnston link_elf_protect_range(ef, start, segstart, prot); 3021d9eae9fSMark Johnston link_elf_protect_range(ef, segstart, end, 3031d9eae9fSMark Johnston prot | segprot); 3041d9eae9fSMark Johnston start = end; 3051d9eae9fSMark Johnston end = segend; 3061d9eae9fSMark Johnston prot = segprot; 3071d9eae9fSMark Johnston } else { 3081d9eae9fSMark Johnston /* 3091d9eae9fSMark Johnston * Case 4: the two segments reside in the same page. 3101d9eae9fSMark Johnston */ 3111d9eae9fSMark Johnston prot |= segprot; 3121d9eae9fSMark Johnston } 3131d9eae9fSMark Johnston } 3141d9eae9fSMark Johnston 3151d9eae9fSMark Johnston /* 3161d9eae9fSMark Johnston * Fix up the last unprotected segment and trailing data. 3171d9eae9fSMark Johnston */ 3181d9eae9fSMark Johnston link_elf_protect_range(ef, start, end, prot); 3191d9eae9fSMark Johnston link_elf_protect_range(ef, end, 3201d9eae9fSMark Johnston round_page((vm_offset_t)ef->address + ef->lf.size), gapprot); 3211d9eae9fSMark Johnston } 3221d9eae9fSMark Johnston 323a4f67738SDoug Rabson static int 324e9eabf59SPeter Wemm link_elf_link_preload(linker_class_t cls, const char *filename, 325e9eabf59SPeter Wemm linker_file_t *result) 3262d636ab0SPeter Wemm { 32770b7ffeeSIan Dowse Elf_Ehdr *hdr; 32870b7ffeeSIan Dowse Elf_Shdr *shdr; 32970b7ffeeSIan Dowse Elf_Sym *es; 33070b7ffeeSIan Dowse void *modptr, *baseptr, *sizeptr; 33170b7ffeeSIan Dowse char *type; 33270b7ffeeSIan Dowse elf_file_t ef; 33370b7ffeeSIan Dowse linker_file_t lf; 33470b7ffeeSIan Dowse Elf_Addr off; 33570b7ffeeSIan Dowse int error, i, j, pb, ra, rl, shstrindex, symstrindex, symtabindex; 33670b7ffeeSIan Dowse 33770b7ffeeSIan Dowse /* Look to see if we have the file preloaded */ 33870b7ffeeSIan Dowse modptr = preload_search_by_name(filename); 33970b7ffeeSIan Dowse if (modptr == NULL) 34070b7ffeeSIan Dowse return ENOENT; 34170b7ffeeSIan Dowse 34270b7ffeeSIan Dowse type = (char *)preload_search_info(modptr, MODINFO_TYPE); 34370b7ffeeSIan Dowse baseptr = preload_search_info(modptr, MODINFO_ADDR); 34470b7ffeeSIan Dowse sizeptr = preload_search_info(modptr, MODINFO_SIZE); 34570b7ffeeSIan Dowse hdr = (Elf_Ehdr *)preload_search_info(modptr, MODINFO_METADATA | 34670b7ffeeSIan Dowse MODINFOMD_ELFHDR); 34770b7ffeeSIan Dowse shdr = (Elf_Shdr *)preload_search_info(modptr, MODINFO_METADATA | 34870b7ffeeSIan Dowse MODINFOMD_SHDR); 34970b7ffeeSIan Dowse if (type == NULL || (strcmp(type, "elf" __XSTRING(__ELF_WORD_SIZE) 35070b7ffeeSIan Dowse " obj module") != 0 && 35170b7ffeeSIan Dowse strcmp(type, "elf obj module") != 0)) { 352de78ca7eSPeter Wemm return (EFTYPE); 353de78ca7eSPeter Wemm } 35470b7ffeeSIan Dowse if (baseptr == NULL || sizeptr == NULL || hdr == NULL || 35570b7ffeeSIan Dowse shdr == NULL) 35670b7ffeeSIan Dowse return (EINVAL); 35770b7ffeeSIan Dowse 35870b7ffeeSIan Dowse lf = linker_make_file(filename, &link_elf_class); 35970b7ffeeSIan Dowse if (lf == NULL) 36070b7ffeeSIan Dowse return (ENOMEM); 36170b7ffeeSIan Dowse 36270b7ffeeSIan Dowse ef = (elf_file_t)lf; 36370b7ffeeSIan Dowse ef->preloaded = 1; 36470b7ffeeSIan Dowse ef->address = *(caddr_t *)baseptr; 36570b7ffeeSIan Dowse lf->address = *(caddr_t *)baseptr; 36670b7ffeeSIan Dowse lf->size = *(size_t *)sizeptr; 36770b7ffeeSIan Dowse 36870b7ffeeSIan Dowse if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS || 36970b7ffeeSIan Dowse hdr->e_ident[EI_DATA] != ELF_TARG_DATA || 37070b7ffeeSIan Dowse hdr->e_ident[EI_VERSION] != EV_CURRENT || 37170b7ffeeSIan Dowse hdr->e_version != EV_CURRENT || 37270b7ffeeSIan Dowse hdr->e_type != ET_REL || 37370b7ffeeSIan Dowse hdr->e_machine != ELF_TARG_MACH) { 37470b7ffeeSIan Dowse error = EFTYPE; 37570b7ffeeSIan Dowse goto out; 37670b7ffeeSIan Dowse } 37770b7ffeeSIan Dowse ef->e_shdr = shdr; 37870b7ffeeSIan Dowse 37970b7ffeeSIan Dowse /* Scan the section header for information and table sizing. */ 38070b7ffeeSIan Dowse symtabindex = -1; 38170b7ffeeSIan Dowse symstrindex = -1; 38270b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 38370b7ffeeSIan Dowse switch (shdr[i].sh_type) { 38470b7ffeeSIan Dowse case SHT_PROGBITS: 38570b7ffeeSIan Dowse case SHT_NOBITS: 38613f28d96SKonstantin Belousov #ifdef __amd64__ 387b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 38813f28d96SKonstantin Belousov #endif 38958c4aee0SJohn Baldwin /* Ignore sections not loaded by the loader. */ 39058c4aee0SJohn Baldwin if (shdr[i].sh_addr == 0) 39158c4aee0SJohn Baldwin break; 39270b7ffeeSIan Dowse ef->nprogtab++; 39370b7ffeeSIan Dowse break; 39470b7ffeeSIan Dowse case SHT_SYMTAB: 39570b7ffeeSIan Dowse symtabindex = i; 39670b7ffeeSIan Dowse symstrindex = shdr[i].sh_link; 39770b7ffeeSIan Dowse break; 39870b7ffeeSIan Dowse case SHT_REL: 39915746ef4SJohn Baldwin /* 40015746ef4SJohn Baldwin * Ignore relocation tables for sections not 40115746ef4SJohn Baldwin * loaded by the loader. 40215746ef4SJohn Baldwin */ 40315746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 40415746ef4SJohn Baldwin break; 405b474c780SDavid E. O'Brien ef->nreltab++; 40670b7ffeeSIan Dowse break; 40770b7ffeeSIan Dowse case SHT_RELA: 40815746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 40915746ef4SJohn Baldwin break; 410b474c780SDavid E. O'Brien ef->nrelatab++; 41170b7ffeeSIan Dowse break; 41270b7ffeeSIan Dowse } 41370b7ffeeSIan Dowse } 41470b7ffeeSIan Dowse 41570b7ffeeSIan Dowse shstrindex = hdr->e_shstrndx; 41670b7ffeeSIan Dowse if (ef->nprogtab == 0 || symstrindex < 0 || 41770b7ffeeSIan Dowse symstrindex >= hdr->e_shnum || 41870b7ffeeSIan Dowse shdr[symstrindex].sh_type != SHT_STRTAB || shstrindex == 0 || 41970b7ffeeSIan Dowse shstrindex >= hdr->e_shnum || 42070b7ffeeSIan Dowse shdr[shstrindex].sh_type != SHT_STRTAB) { 42170b7ffeeSIan Dowse printf("%s: bad/missing section headers\n", filename); 42270b7ffeeSIan Dowse error = ENOEXEC; 42370b7ffeeSIan Dowse goto out; 42470b7ffeeSIan Dowse } 42570b7ffeeSIan Dowse 42670b7ffeeSIan Dowse /* Allocate space for tracking the load chunks */ 42770b7ffeeSIan Dowse if (ef->nprogtab != 0) 42870b7ffeeSIan Dowse ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab), 42970b7ffeeSIan Dowse M_LINKER, M_WAITOK | M_ZERO); 430b474c780SDavid E. O'Brien if (ef->nreltab != 0) 431b474c780SDavid E. O'Brien ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab), 432b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 433b474c780SDavid E. O'Brien if (ef->nrelatab != 0) 434b474c780SDavid E. O'Brien ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab), 435b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 43670b7ffeeSIan Dowse if ((ef->nprogtab != 0 && ef->progtab == NULL) || 437b474c780SDavid E. O'Brien (ef->nreltab != 0 && ef->reltab == NULL) || 438b474c780SDavid E. O'Brien (ef->nrelatab != 0 && ef->relatab == NULL)) { 43970b7ffeeSIan Dowse error = ENOMEM; 44070b7ffeeSIan Dowse goto out; 44170b7ffeeSIan Dowse } 44270b7ffeeSIan Dowse 44370b7ffeeSIan Dowse /* XXX, relocate the sh_addr fields saved by the loader. */ 44470b7ffeeSIan Dowse off = 0; 44570b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 44670b7ffeeSIan Dowse if (shdr[i].sh_addr != 0 && (off == 0 || shdr[i].sh_addr < off)) 44770b7ffeeSIan Dowse off = shdr[i].sh_addr; 44870b7ffeeSIan Dowse } 44970b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 45070b7ffeeSIan Dowse if (shdr[i].sh_addr != 0) 45170b7ffeeSIan Dowse shdr[i].sh_addr = shdr[i].sh_addr - off + 45270b7ffeeSIan Dowse (Elf_Addr)ef->address; 45370b7ffeeSIan Dowse } 45470b7ffeeSIan Dowse 45570b7ffeeSIan Dowse ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym); 45670b7ffeeSIan Dowse ef->ddbsymtab = (Elf_Sym *)shdr[symtabindex].sh_addr; 45770b7ffeeSIan Dowse ef->ddbstrcnt = shdr[symstrindex].sh_size; 45870b7ffeeSIan Dowse ef->ddbstrtab = (char *)shdr[symstrindex].sh_addr; 45970b7ffeeSIan Dowse ef->shstrcnt = shdr[shstrindex].sh_size; 46070b7ffeeSIan Dowse ef->shstrtab = (char *)shdr[shstrindex].sh_addr; 46170b7ffeeSIan Dowse 46270b7ffeeSIan Dowse /* Now fill out progtab and the relocation tables. */ 46370b7ffeeSIan Dowse pb = 0; 46470b7ffeeSIan Dowse rl = 0; 46570b7ffeeSIan Dowse ra = 0; 46670b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 46770b7ffeeSIan Dowse switch (shdr[i].sh_type) { 46870b7ffeeSIan Dowse case SHT_PROGBITS: 46970b7ffeeSIan Dowse case SHT_NOBITS: 47013f28d96SKonstantin Belousov #ifdef __amd64__ 471b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 47213f28d96SKonstantin Belousov #endif 47358c4aee0SJohn Baldwin if (shdr[i].sh_addr == 0) 47458c4aee0SJohn Baldwin break; 47570b7ffeeSIan Dowse ef->progtab[pb].addr = (void *)shdr[i].sh_addr; 47670b7ffeeSIan Dowse if (shdr[i].sh_type == SHT_PROGBITS) 47770b7ffeeSIan Dowse ef->progtab[pb].name = "<<PROGBITS>>"; 47813f28d96SKonstantin Belousov #ifdef __amd64__ 479b715d9afSKonstantin Belousov else if (shdr[i].sh_type == SHT_X86_64_UNWIND) 48013f28d96SKonstantin Belousov ef->progtab[pb].name = "<<UNWIND>>"; 48113f28d96SKonstantin Belousov #endif 48270b7ffeeSIan Dowse else 48370b7ffeeSIan Dowse ef->progtab[pb].name = "<<NOBITS>>"; 48470b7ffeeSIan Dowse ef->progtab[pb].size = shdr[i].sh_size; 4851d9eae9fSMark Johnston ef->progtab[pb].flags = shdr[i].sh_flags; 48670b7ffeeSIan Dowse ef->progtab[pb].sec = i; 48770b7ffeeSIan Dowse if (ef->shstrtab && shdr[i].sh_name != 0) 48870b7ffeeSIan Dowse ef->progtab[pb].name = 48970b7ffeeSIan Dowse ef->shstrtab + shdr[i].sh_name; 49050c202c5SJeff Roberson if (ef->progtab[pb].name != NULL && 4915f67450dSDimitry Andric !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) { 49250c202c5SJeff Roberson void *dpcpu; 49350c202c5SJeff Roberson 49450c202c5SJeff Roberson dpcpu = dpcpu_alloc(shdr[i].sh_size); 49550c202c5SJeff Roberson if (dpcpu == NULL) { 4960f823b64SBjoern A. Zeeb printf("%s: pcpu module space is out " 4979afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 4989afc5684SBjoern A. Zeeb "for %s\n", __func__, 4999afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 5000f823b64SBjoern A. Zeeb filename); 50150c202c5SJeff Roberson error = ENOSPC; 50250c202c5SJeff Roberson goto out; 50350c202c5SJeff Roberson } 50450c202c5SJeff Roberson memcpy(dpcpu, ef->progtab[pb].addr, 50550c202c5SJeff Roberson ef->progtab[pb].size); 50650c202c5SJeff Roberson dpcpu_copy(dpcpu, shdr[i].sh_size); 50750c202c5SJeff Roberson ef->progtab[pb].addr = dpcpu; 508eddfbb76SRobert Watson #ifdef VIMAGE 509eddfbb76SRobert Watson } else if (ef->progtab[pb].name != NULL && 51017ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) { 511eddfbb76SRobert Watson void *vnet_data; 512eddfbb76SRobert Watson 513eddfbb76SRobert Watson vnet_data = vnet_data_alloc(shdr[i].sh_size); 514eddfbb76SRobert Watson if (vnet_data == NULL) { 5150f823b64SBjoern A. Zeeb printf("%s: vnet module space is out " 5169afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 5179afc5684SBjoern A. Zeeb "for %s\n", __func__, 5189afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 5190f823b64SBjoern A. Zeeb filename); 520eddfbb76SRobert Watson error = ENOSPC; 521eddfbb76SRobert Watson goto out; 522eddfbb76SRobert Watson } 523eddfbb76SRobert Watson memcpy(vnet_data, ef->progtab[pb].addr, 524eddfbb76SRobert Watson ef->progtab[pb].size); 525eddfbb76SRobert Watson vnet_data_copy(vnet_data, shdr[i].sh_size); 526eddfbb76SRobert Watson ef->progtab[pb].addr = vnet_data; 527eddfbb76SRobert Watson #endif 5280067051fSMarcel Moolenaar } else if (ef->progtab[pb].name != NULL && 5290067051fSMarcel Moolenaar !strcmp(ef->progtab[pb].name, ".ctors")) { 5300067051fSMarcel Moolenaar lf->ctors_addr = ef->progtab[pb].addr; 5310067051fSMarcel Moolenaar lf->ctors_size = shdr[i].sh_size; 53250c202c5SJeff Roberson } 53370b7ffeeSIan Dowse 53470b7ffeeSIan Dowse /* Update all symbol values with the offset. */ 53570b7ffeeSIan Dowse for (j = 0; j < ef->ddbsymcnt; j++) { 53670b7ffeeSIan Dowse es = &ef->ddbsymtab[j]; 53770b7ffeeSIan Dowse if (es->st_shndx != i) 53870b7ffeeSIan Dowse continue; 53970b7ffeeSIan Dowse es->st_value += (Elf_Addr)ef->progtab[pb].addr; 54070b7ffeeSIan Dowse } 54170b7ffeeSIan Dowse pb++; 54270b7ffeeSIan Dowse break; 54370b7ffeeSIan Dowse case SHT_REL: 54415746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 54515746ef4SJohn Baldwin break; 54670b7ffeeSIan Dowse ef->reltab[rl].rel = (Elf_Rel *)shdr[i].sh_addr; 54770b7ffeeSIan Dowse ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel); 54870b7ffeeSIan Dowse ef->reltab[rl].sec = shdr[i].sh_info; 54970b7ffeeSIan Dowse rl++; 55070b7ffeeSIan Dowse break; 55170b7ffeeSIan Dowse case SHT_RELA: 55215746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 55315746ef4SJohn Baldwin break; 55470b7ffeeSIan Dowse ef->relatab[ra].rela = (Elf_Rela *)shdr[i].sh_addr; 55570b7ffeeSIan Dowse ef->relatab[ra].nrela = 55670b7ffeeSIan Dowse shdr[i].sh_size / sizeof(Elf_Rela); 55770b7ffeeSIan Dowse ef->relatab[ra].sec = shdr[i].sh_info; 55870b7ffeeSIan Dowse ra++; 55970b7ffeeSIan Dowse break; 56070b7ffeeSIan Dowse } 56170b7ffeeSIan Dowse } 5623cfce8e4SKonstantin Belousov if (pb != ef->nprogtab) { 5633cfce8e4SKonstantin Belousov printf("%s: lost progbits\n", filename); 5643cfce8e4SKonstantin Belousov error = ENOEXEC; 5653cfce8e4SKonstantin Belousov goto out; 5663cfce8e4SKonstantin Belousov } 5673cfce8e4SKonstantin Belousov if (rl != ef->nreltab) { 5683cfce8e4SKonstantin Belousov printf("%s: lost reltab\n", filename); 5693cfce8e4SKonstantin Belousov error = ENOEXEC; 5703cfce8e4SKonstantin Belousov goto out; 5713cfce8e4SKonstantin Belousov } 5723cfce8e4SKonstantin Belousov if (ra != ef->nrelatab) { 5733cfce8e4SKonstantin Belousov printf("%s: lost relatab\n", filename); 5743cfce8e4SKonstantin Belousov error = ENOEXEC; 5753cfce8e4SKonstantin Belousov goto out; 5763cfce8e4SKonstantin Belousov } 57770b7ffeeSIan Dowse 578f822c9e2SMark Johnston /* 579f822c9e2SMark Johnston * The file needs to be writeable and executable while applying 580f822c9e2SMark Johnston * relocations. Mapping protections are applied once relocation 581f822c9e2SMark Johnston * processing is complete. 582f822c9e2SMark Johnston */ 583f822c9e2SMark Johnston link_elf_protect_range(ef, (vm_offset_t)ef->address, 584f822c9e2SMark Johnston round_page((vm_offset_t)ef->address + ef->lf.size), VM_PROT_ALL); 585f822c9e2SMark Johnston 58670b7ffeeSIan Dowse /* Local intra-module relocations */ 587289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, false); 5883cfce8e4SKonstantin Belousov if (error != 0) 5893cfce8e4SKonstantin Belousov goto out; 59070b7ffeeSIan Dowse *result = lf; 59170b7ffeeSIan Dowse return (0); 59270b7ffeeSIan Dowse 59370b7ffeeSIan Dowse out: 59470b7ffeeSIan Dowse /* preload not done this way */ 59570b7ffeeSIan Dowse linker_file_unload(lf, LINKER_UNLOAD_FORCE); 59670b7ffeeSIan Dowse return (error); 59770b7ffeeSIan Dowse } 598de78ca7eSPeter Wemm 5990067051fSMarcel Moolenaar static void 6000067051fSMarcel Moolenaar link_elf_invoke_ctors(caddr_t addr, size_t size) 6010067051fSMarcel Moolenaar { 6020067051fSMarcel Moolenaar void (**ctor)(void); 6030067051fSMarcel Moolenaar size_t i, cnt; 6040067051fSMarcel Moolenaar 6050067051fSMarcel Moolenaar if (addr == NULL || size == 0) 6060067051fSMarcel Moolenaar return; 6070067051fSMarcel Moolenaar cnt = size / sizeof(*ctor); 6080067051fSMarcel Moolenaar ctor = (void *)addr; 6090067051fSMarcel Moolenaar for (i = 0; i < cnt; i++) { 6100067051fSMarcel Moolenaar if (ctor[i] != NULL) 6110067051fSMarcel Moolenaar (*ctor[i])(); 6120067051fSMarcel Moolenaar } 6130067051fSMarcel Moolenaar } 6140067051fSMarcel Moolenaar 615de78ca7eSPeter Wemm static int 61654823af2SPeter Wemm link_elf_link_preload_finish(linker_file_t lf) 61754823af2SPeter Wemm { 61870b7ffeeSIan Dowse elf_file_t ef; 61970b7ffeeSIan Dowse int error; 62070b7ffeeSIan Dowse 62170b7ffeeSIan Dowse ef = (elf_file_t)lf; 62270b7ffeeSIan Dowse error = relocate_file(ef); 62370b7ffeeSIan Dowse if (error) 624289ead7cSKonstantin Belousov return (error); 62570b7ffeeSIan Dowse 62670b7ffeeSIan Dowse /* Notify MD code that a module is being loaded. */ 62770b7ffeeSIan Dowse error = elf_cpu_load_file(lf); 62870b7ffeeSIan Dowse if (error) 62970b7ffeeSIan Dowse return (error); 63070b7ffeeSIan Dowse 6318f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__) 632289ead7cSKonstantin Belousov /* Now ifuncs. */ 633289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, true); 634289ead7cSKonstantin Belousov if (error != 0) 635289ead7cSKonstantin Belousov return (error); 6368f941950SKonstantin Belousov #endif 637289ead7cSKonstantin Belousov 638f822c9e2SMark Johnston /* Apply protections now that relocation processing is complete. */ 639f822c9e2SMark Johnston link_elf_protect(ef); 640f822c9e2SMark Johnston 6410067051fSMarcel Moolenaar link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size); 64270b7ffeeSIan Dowse return (0); 64354823af2SPeter Wemm } 64454823af2SPeter Wemm 64554823af2SPeter Wemm static int 646c143d6c2SMarcel Moolenaar link_elf_load_file(linker_class_t cls, const char *filename, 647c143d6c2SMarcel Moolenaar linker_file_t *result) 648a4f67738SDoug Rabson { 649a344babfSGleb Smirnoff struct nameidata *nd; 650b40ce416SJulian Elischer struct thread *td = curthread; /* XXX */ 651caab6e90SPeter Wemm Elf_Ehdr *hdr; 652e9eabf59SPeter Wemm Elf_Shdr *shdr; 653a8774e39SPeter Wemm Elf_Sym *es; 654a8774e39SPeter Wemm int nbytes, i, j; 65520947801SPeter Wemm vm_offset_t mapbase; 656fe3db7c7SDoug Rabson size_t mapsize; 657a4f67738SDoug Rabson int error = 0; 658526d0bd5SKonstantin Belousov ssize_t resid; 659526d0bd5SKonstantin Belousov int flags; 660a4f67738SDoug Rabson elf_file_t ef; 661a4f67738SDoug Rabson linker_file_t lf; 662ca65d5c7SPeter Wemm int symtabindex; 663ca65d5c7SPeter Wemm int symstrindex; 664e9eabf59SPeter Wemm int shstrindex; 665e9eabf59SPeter Wemm int nsym; 66620947801SPeter Wemm int pb, rl, ra; 667e9eabf59SPeter Wemm int alignmask; 668ca65d5c7SPeter Wemm 669ca65d5c7SPeter Wemm shdr = NULL; 670ca65d5c7SPeter Wemm lf = NULL; 671e9eabf59SPeter Wemm mapsize = 0; 672e9eabf59SPeter Wemm hdr = NULL; 673a4f67738SDoug Rabson 674a344babfSGleb Smirnoff nd = malloc(sizeof(struct nameidata), M_TEMP, M_WAITOK); 675a344babfSGleb Smirnoff NDINIT(nd, LOOKUP, FOLLOW, UIO_SYSSPACE, filename, td); 676e6796b67SKirk McKusick flags = FREAD; 677a344babfSGleb Smirnoff error = vn_open(nd, &flags, 0, NULL); 678a344babfSGleb Smirnoff if (error) { 679a344babfSGleb Smirnoff free(nd, M_TEMP); 680a4f67738SDoug Rabson return error; 681a344babfSGleb Smirnoff } 682a344babfSGleb Smirnoff NDFREE(nd, NDF_ONLY_PNBUF); 683a344babfSGleb Smirnoff if (nd->ni_vp->v_type != VREG) { 6841f49b573SRuslan Ermilov error = ENOEXEC; 6851f49b573SRuslan Ermilov goto out; 6861f49b573SRuslan Ermilov } 687a3df768bSRobert Watson #ifdef MAC 688a344babfSGleb Smirnoff error = mac_kld_check_load(td->td_ucred, nd->ni_vp); 689a3df768bSRobert Watson if (error) { 690a3df768bSRobert Watson goto out; 691a3df768bSRobert Watson } 692a3df768bSRobert Watson #endif 693a4f67738SDoug Rabson 694e9eabf59SPeter Wemm /* Read the elf header from the file. */ 695e9eabf59SPeter Wemm hdr = malloc(sizeof(*hdr), M_LINKER, M_WAITOK); 696a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)hdr, sizeof(*hdr), 0, 6979ca43589SRobert Watson UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 6989ca43589SRobert Watson &resid, td); 699a4f67738SDoug Rabson if (error) 700a4f67738SDoug Rabson goto out; 701e9eabf59SPeter Wemm if (resid != 0){ 702e9eabf59SPeter Wemm error = ENOEXEC; 703e9eabf59SPeter Wemm goto out; 704e9eabf59SPeter Wemm } 705a4f67738SDoug Rabson 706caab6e90SPeter Wemm if (!IS_ELF(*hdr)) { 707fe3db7c7SDoug Rabson error = ENOEXEC; 708a4f67738SDoug Rabson goto out; 709fe3db7c7SDoug Rabson } 710fe3db7c7SDoug Rabson 711caab6e90SPeter Wemm if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS 712caab6e90SPeter Wemm || hdr->e_ident[EI_DATA] != ELF_TARG_DATA) { 713552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported file layout"); 714fe3db7c7SDoug Rabson error = ENOEXEC; 715fe3db7c7SDoug Rabson goto out; 716fe3db7c7SDoug Rabson } 717caab6e90SPeter Wemm if (hdr->e_ident[EI_VERSION] != EV_CURRENT 718caab6e90SPeter Wemm || hdr->e_version != EV_CURRENT) { 719552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported file version"); 720fe3db7c7SDoug Rabson error = ENOEXEC; 721fe3db7c7SDoug Rabson goto out; 722fe3db7c7SDoug Rabson } 723e9eabf59SPeter Wemm if (hdr->e_type != ET_REL) { 724a1d7ce03SAttilio Rao error = ENOSYS; 725fe3db7c7SDoug Rabson goto out; 726fe3db7c7SDoug Rabson } 727caab6e90SPeter Wemm if (hdr->e_machine != ELF_TARG_MACH) { 728552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported machine"); 729fe3db7c7SDoug Rabson error = ENOEXEC; 730fe3db7c7SDoug Rabson goto out; 731fe3db7c7SDoug Rabson } 732a4f67738SDoug Rabson 733326e27d8SDoug Rabson lf = linker_make_file(filename, &link_elf_class); 734326e27d8SDoug Rabson if (!lf) { 735326e27d8SDoug Rabson error = ENOMEM; 736326e27d8SDoug Rabson goto out; 737326e27d8SDoug Rabson } 738326e27d8SDoug Rabson ef = (elf_file_t) lf; 739e9eabf59SPeter Wemm ef->nprogtab = 0; 740e9eabf59SPeter Wemm ef->e_shdr = 0; 741b474c780SDavid E. O'Brien ef->nreltab = 0; 742b474c780SDavid E. O'Brien ef->nrelatab = 0; 743e9eabf59SPeter Wemm 744e9eabf59SPeter Wemm /* Allocate and read in the section header */ 745e9eabf59SPeter Wemm nbytes = hdr->e_shnum * hdr->e_shentsize; 746e9eabf59SPeter Wemm if (nbytes == 0 || hdr->e_shoff == 0 || 747e9eabf59SPeter Wemm hdr->e_shentsize != sizeof(Elf_Shdr)) { 748e9eabf59SPeter Wemm error = ENOEXEC; 749e9eabf59SPeter Wemm goto out; 750e9eabf59SPeter Wemm } 75120947801SPeter Wemm shdr = malloc(nbytes, M_LINKER, M_WAITOK); 752e9eabf59SPeter Wemm ef->e_shdr = shdr; 753a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (caddr_t)shdr, nbytes, 754a344babfSGleb Smirnoff hdr->e_shoff, UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, 755a344babfSGleb Smirnoff NOCRED, &resid, td); 756e9eabf59SPeter Wemm if (error) 757e9eabf59SPeter Wemm goto out; 758e9eabf59SPeter Wemm if (resid) { 759e9eabf59SPeter Wemm error = ENOEXEC; 760e9eabf59SPeter Wemm goto out; 761e9eabf59SPeter Wemm } 762e9eabf59SPeter Wemm 763e9eabf59SPeter Wemm /* Scan the section header for information and table sizing. */ 764e9eabf59SPeter Wemm nsym = 0; 765e9eabf59SPeter Wemm symtabindex = -1; 766e9eabf59SPeter Wemm symstrindex = -1; 767e9eabf59SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 768676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 769676799a0SAndriy Gapon continue; 770e9eabf59SPeter Wemm switch (shdr[i].sh_type) { 771e9eabf59SPeter Wemm case SHT_PROGBITS: 772e9eabf59SPeter Wemm case SHT_NOBITS: 77313f28d96SKonstantin Belousov #ifdef __amd64__ 774b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 77513f28d96SKonstantin Belousov #endif 77658c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 77758c4aee0SJohn Baldwin break; 77820947801SPeter Wemm ef->nprogtab++; 779e9eabf59SPeter Wemm break; 780e9eabf59SPeter Wemm case SHT_SYMTAB: 781e9eabf59SPeter Wemm nsym++; 782e9eabf59SPeter Wemm symtabindex = i; 783e9eabf59SPeter Wemm symstrindex = shdr[i].sh_link; 784e9eabf59SPeter Wemm break; 785e9eabf59SPeter Wemm case SHT_REL: 78615746ef4SJohn Baldwin /* 78715746ef4SJohn Baldwin * Ignore relocation tables for unallocated 78815746ef4SJohn Baldwin * sections. 78915746ef4SJohn Baldwin */ 79015746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 79115746ef4SJohn Baldwin break; 792b474c780SDavid E. O'Brien ef->nreltab++; 793e9eabf59SPeter Wemm break; 794e9eabf59SPeter Wemm case SHT_RELA: 79515746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 79615746ef4SJohn Baldwin break; 797b474c780SDavid E. O'Brien ef->nrelatab++; 798e9eabf59SPeter Wemm break; 799e9eabf59SPeter Wemm case SHT_STRTAB: 800e9eabf59SPeter Wemm break; 801e9eabf59SPeter Wemm } 802e9eabf59SPeter Wemm } 80320947801SPeter Wemm if (ef->nprogtab == 0) { 804552f9f63SEdwin Groothuis link_elf_error(filename, "file has no contents"); 805e9eabf59SPeter Wemm error = ENOEXEC; 806e9eabf59SPeter Wemm goto out; 807e9eabf59SPeter Wemm } 808e9eabf59SPeter Wemm if (nsym != 1) { 809e9eabf59SPeter Wemm /* Only allow one symbol table for now */ 810f912a970SEd Maste link_elf_error(filename, 811f912a970SEd Maste "file must have exactly one symbol table"); 812e9eabf59SPeter Wemm error = ENOEXEC; 813e9eabf59SPeter Wemm goto out; 814e9eabf59SPeter Wemm } 815e9eabf59SPeter Wemm if (symstrindex < 0 || symstrindex > hdr->e_shnum || 816e9eabf59SPeter Wemm shdr[symstrindex].sh_type != SHT_STRTAB) { 817552f9f63SEdwin Groothuis link_elf_error(filename, "file has invalid symbol strings"); 818e9eabf59SPeter Wemm error = ENOEXEC; 819e9eabf59SPeter Wemm goto out; 820e9eabf59SPeter Wemm } 821e9eabf59SPeter Wemm 822e9eabf59SPeter Wemm /* Allocate space for tracking the load chunks */ 823e9eabf59SPeter Wemm if (ef->nprogtab != 0) 82420947801SPeter Wemm ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab), 82520947801SPeter Wemm M_LINKER, M_WAITOK | M_ZERO); 826b474c780SDavid E. O'Brien if (ef->nreltab != 0) 827b474c780SDavid E. O'Brien ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab), 828b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 829b474c780SDavid E. O'Brien if (ef->nrelatab != 0) 830b474c780SDavid E. O'Brien ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab), 831b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 832e9eabf59SPeter Wemm 8333cfce8e4SKonstantin Belousov if (symtabindex == -1) { 8343cfce8e4SKonstantin Belousov link_elf_error(filename, "lost symbol table index"); 8353cfce8e4SKonstantin Belousov error = ENOEXEC; 8363cfce8e4SKonstantin Belousov goto out; 8373cfce8e4SKonstantin Belousov } 83820947801SPeter Wemm /* Allocate space for and load the symbol table */ 839e9eabf59SPeter Wemm ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym); 840e9eabf59SPeter Wemm ef->ddbsymtab = malloc(shdr[symtabindex].sh_size, M_LINKER, M_WAITOK); 841a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)ef->ddbsymtab, 84220947801SPeter Wemm shdr[symtabindex].sh_size, shdr[symtabindex].sh_offset, 84320947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 84420947801SPeter Wemm &resid, td); 84520947801SPeter Wemm if (error) 84620947801SPeter Wemm goto out; 84720947801SPeter Wemm if (resid != 0){ 84820947801SPeter Wemm error = EINVAL; 84920947801SPeter Wemm goto out; 85020947801SPeter Wemm } 851e9eabf59SPeter Wemm 85220947801SPeter Wemm /* Allocate space for and load the symbol strings */ 853e9eabf59SPeter Wemm ef->ddbstrcnt = shdr[symstrindex].sh_size; 854e9eabf59SPeter Wemm ef->ddbstrtab = malloc(shdr[symstrindex].sh_size, M_LINKER, M_WAITOK); 855a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, ef->ddbstrtab, 85620947801SPeter Wemm shdr[symstrindex].sh_size, shdr[symstrindex].sh_offset, 85720947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 85820947801SPeter Wemm &resid, td); 85920947801SPeter Wemm if (error) 86020947801SPeter Wemm goto out; 86120947801SPeter Wemm if (resid != 0){ 86220947801SPeter Wemm error = EINVAL; 86320947801SPeter Wemm goto out; 86420947801SPeter Wemm } 865e9eabf59SPeter Wemm 866e9eabf59SPeter Wemm /* Do we have a string table for the section names? */ 867e9eabf59SPeter Wemm shstrindex = -1; 86820947801SPeter Wemm if (hdr->e_shstrndx != 0 && 86920947801SPeter Wemm shdr[hdr->e_shstrndx].sh_type == SHT_STRTAB) { 870e9eabf59SPeter Wemm shstrindex = hdr->e_shstrndx; 871e9eabf59SPeter Wemm ef->shstrcnt = shdr[shstrindex].sh_size; 87220947801SPeter Wemm ef->shstrtab = malloc(shdr[shstrindex].sh_size, M_LINKER, 87320947801SPeter Wemm M_WAITOK); 874a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, ef->shstrtab, 87520947801SPeter Wemm shdr[shstrindex].sh_size, shdr[shstrindex].sh_offset, 876e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 877e9eabf59SPeter Wemm &resid, td); 878e9eabf59SPeter Wemm if (error) 879e9eabf59SPeter Wemm goto out; 88020947801SPeter Wemm if (resid != 0){ 88120947801SPeter Wemm error = EINVAL; 88220947801SPeter Wemm goto out; 88320947801SPeter Wemm } 884e9eabf59SPeter Wemm } 885e9eabf59SPeter Wemm 88620947801SPeter Wemm /* Size up code/data(progbits) and bss(nobits). */ 887e9eabf59SPeter Wemm alignmask = 0; 888e9eabf59SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 889676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 890676799a0SAndriy Gapon continue; 891e9eabf59SPeter Wemm switch (shdr[i].sh_type) { 892e9eabf59SPeter Wemm case SHT_PROGBITS: 893e9eabf59SPeter Wemm case SHT_NOBITS: 89413f28d96SKonstantin Belousov #ifdef __amd64__ 895b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 89613f28d96SKonstantin Belousov #endif 89758c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 89858c4aee0SJohn Baldwin break; 899e9eabf59SPeter Wemm alignmask = shdr[i].sh_addralign - 1; 900e9eabf59SPeter Wemm mapsize += alignmask; 901e9eabf59SPeter Wemm mapsize &= ~alignmask; 902e9eabf59SPeter Wemm mapsize += shdr[i].sh_size; 903e9eabf59SPeter Wemm break; 904e9eabf59SPeter Wemm } 905e9eabf59SPeter Wemm } 906e9eabf59SPeter Wemm 907e9eabf59SPeter Wemm /* 908e9eabf59SPeter Wemm * We know how much space we need for the text/data/bss/etc. 909e9eabf59SPeter Wemm * This stuff needs to be in a single chunk so that profiling etc 910e9eabf59SPeter Wemm * can get the bounds and gdb can associate offsets with modules 911e9eabf59SPeter Wemm */ 912fbf2a778SKonstantin Belousov ef->object = vm_pager_allocate(OBJT_PHYS, NULL, round_page(mapsize), 913fbf2a778SKonstantin Belousov VM_PROT_ALL, 0, thread0.td_ucred); 914fe3db7c7SDoug Rabson if (ef->object == NULL) { 915fe3db7c7SDoug Rabson error = ENOMEM; 916fe3db7c7SDoug Rabson goto out; 917fe3db7c7SDoug Rabson } 918d80126a6SMark Johnston #if VM_NRESERVLEVEL > 0 919d80126a6SMark Johnston vm_object_color(ef->object, 0); 920d80126a6SMark Johnston #endif 921ac68d1c9SAlan Cox 922ac68d1c9SAlan Cox /* 923ac68d1c9SAlan Cox * In order to satisfy amd64's architectural requirements on the 924ac68d1c9SAlan Cox * location of code and data in the kernel's address space, request a 925ac68d1c9SAlan Cox * mapping that is above the kernel. 9261d9eae9fSMark Johnston * 9271d9eae9fSMark Johnston * Protections will be restricted once relocations are applied. 928ac68d1c9SAlan Cox */ 9294104e835SOleksandr Tymoshenko #ifdef __amd64__ 930ac68d1c9SAlan Cox mapbase = KERNBASE; 9314104e835SOleksandr Tymoshenko #else 9324104e835SOleksandr Tymoshenko mapbase = VM_MIN_KERNEL_ADDRESS; 9334104e835SOleksandr Tymoshenko #endif 93420947801SPeter Wemm error = vm_map_find(kernel_map, ef->object, 0, &mapbase, 935edb572a3SJohn Baldwin round_page(mapsize), 0, VMFS_OPTIMAL_SPACE, VM_PROT_ALL, 936edb572a3SJohn Baldwin VM_PROT_ALL, 0); 937092bacb2SMark Johnston if (error != KERN_SUCCESS) { 938fe3db7c7SDoug Rabson vm_object_deallocate(ef->object); 939092bacb2SMark Johnston ef->object = NULL; 940092bacb2SMark Johnston error = ENOMEM; 941fe3db7c7SDoug Rabson goto out; 942fe3db7c7SDoug Rabson } 94320947801SPeter Wemm 944e9eabf59SPeter Wemm /* Wire the pages */ 9457f1ef325SAlan Cox error = vm_map_wire(kernel_map, mapbase, 94620947801SPeter Wemm mapbase + round_page(mapsize), 947e9eabf59SPeter Wemm VM_MAP_WIRE_SYSTEM|VM_MAP_WIRE_NOHOLES); 9487f1ef325SAlan Cox if (error != KERN_SUCCESS) { 9497f1ef325SAlan Cox error = ENOMEM; 9507f1ef325SAlan Cox goto out; 9517f1ef325SAlan Cox } 952a4f67738SDoug Rabson 95320947801SPeter Wemm /* Inform the kld system about the situation */ 95420947801SPeter Wemm lf->address = ef->address = (caddr_t)mapbase; 95520947801SPeter Wemm lf->size = mapsize; 956e9eabf59SPeter Wemm 957e9eabf59SPeter Wemm /* 95820947801SPeter Wemm * Now load code/data(progbits), zero bss(nobits), allocate space for 95920947801SPeter Wemm * and load relocs 960e9eabf59SPeter Wemm */ 96120947801SPeter Wemm pb = 0; 96220947801SPeter Wemm rl = 0; 96320947801SPeter Wemm ra = 0; 96420947801SPeter Wemm alignmask = 0; 96520947801SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 966676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 967676799a0SAndriy Gapon continue; 96820947801SPeter Wemm switch (shdr[i].sh_type) { 96920947801SPeter Wemm case SHT_PROGBITS: 97020947801SPeter Wemm case SHT_NOBITS: 97113f28d96SKonstantin Belousov #ifdef __amd64__ 972b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 97313f28d96SKonstantin Belousov #endif 97458c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 97558c4aee0SJohn Baldwin break; 97620947801SPeter Wemm alignmask = shdr[i].sh_addralign - 1; 97720947801SPeter Wemm mapbase += alignmask; 97820947801SPeter Wemm mapbase &= ~alignmask; 9790067051fSMarcel Moolenaar if (ef->shstrtab != NULL && shdr[i].sh_name != 0) { 98050c202c5SJeff Roberson ef->progtab[pb].name = 98150c202c5SJeff Roberson ef->shstrtab + shdr[i].sh_name; 9820067051fSMarcel Moolenaar if (!strcmp(ef->progtab[pb].name, ".ctors")) { 9830067051fSMarcel Moolenaar lf->ctors_addr = (caddr_t)mapbase; 9840067051fSMarcel Moolenaar lf->ctors_size = shdr[i].sh_size; 9850067051fSMarcel Moolenaar } 9860067051fSMarcel Moolenaar } else if (shdr[i].sh_type == SHT_PROGBITS) 98720947801SPeter Wemm ef->progtab[pb].name = "<<PROGBITS>>"; 98813f28d96SKonstantin Belousov #ifdef __amd64__ 989b715d9afSKonstantin Belousov else if (shdr[i].sh_type == SHT_X86_64_UNWIND) 99013f28d96SKonstantin Belousov ef->progtab[pb].name = "<<UNWIND>>"; 99113f28d96SKonstantin Belousov #endif 99250c202c5SJeff Roberson else 99350c202c5SJeff Roberson ef->progtab[pb].name = "<<NOBITS>>"; 99450c202c5SJeff Roberson if (ef->progtab[pb].name != NULL && 9950f823b64SBjoern A. Zeeb !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) { 99650c202c5SJeff Roberson ef->progtab[pb].addr = 99750c202c5SJeff Roberson dpcpu_alloc(shdr[i].sh_size); 9980f823b64SBjoern A. Zeeb if (ef->progtab[pb].addr == NULL) { 9990f823b64SBjoern A. Zeeb printf("%s: pcpu module space is out " 10009afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 10019afc5684SBjoern A. Zeeb "for %s\n", __func__, 10029afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 10030f823b64SBjoern A. Zeeb filename); 10040f823b64SBjoern A. Zeeb } 10050f823b64SBjoern A. Zeeb } 1006eddfbb76SRobert Watson #ifdef VIMAGE 1007eddfbb76SRobert Watson else if (ef->progtab[pb].name != NULL && 10080f823b64SBjoern A. Zeeb !strcmp(ef->progtab[pb].name, VNET_SETNAME)) { 1009eddfbb76SRobert Watson ef->progtab[pb].addr = 1010eddfbb76SRobert Watson vnet_data_alloc(shdr[i].sh_size); 10110f823b64SBjoern A. Zeeb if (ef->progtab[pb].addr == NULL) { 10120f823b64SBjoern A. Zeeb printf("%s: vnet module space is out " 10139afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 10149afc5684SBjoern A. Zeeb "for %s\n", __func__, 10159afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 10160f823b64SBjoern A. Zeeb filename); 10170f823b64SBjoern A. Zeeb } 10180f823b64SBjoern A. Zeeb } 1019eddfbb76SRobert Watson #endif 102050c202c5SJeff Roberson else 102150c202c5SJeff Roberson ef->progtab[pb].addr = 102250c202c5SJeff Roberson (void *)(uintptr_t)mapbase; 102350c202c5SJeff Roberson if (ef->progtab[pb].addr == NULL) { 102450c202c5SJeff Roberson error = ENOSPC; 102550c202c5SJeff Roberson goto out; 102650c202c5SJeff Roberson } 102750c202c5SJeff Roberson ef->progtab[pb].size = shdr[i].sh_size; 10281d9eae9fSMark Johnston ef->progtab[pb].flags = shdr[i].sh_flags; 102950c202c5SJeff Roberson ef->progtab[pb].sec = i; 103013f28d96SKonstantin Belousov if (shdr[i].sh_type == SHT_PROGBITS 103113f28d96SKonstantin Belousov #ifdef __amd64__ 1032b715d9afSKonstantin Belousov || shdr[i].sh_type == SHT_X86_64_UNWIND 103313f28d96SKonstantin Belousov #endif 103413f28d96SKonstantin Belousov ) { 1035a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 103620947801SPeter Wemm ef->progtab[pb].addr, 103720947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 103820947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, 103920947801SPeter Wemm NOCRED, &resid, td); 104020947801SPeter Wemm if (error) 104120947801SPeter Wemm goto out; 104220947801SPeter Wemm if (resid != 0){ 104320947801SPeter Wemm error = EINVAL; 104420947801SPeter Wemm goto out; 104520947801SPeter Wemm } 1046eddfbb76SRobert Watson /* Initialize the per-cpu or vnet area. */ 1047eddfbb76SRobert Watson if (ef->progtab[pb].addr != (void *)mapbase && 10485f67450dSDimitry Andric !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) 104950c202c5SJeff Roberson dpcpu_copy(ef->progtab[pb].addr, 105050c202c5SJeff Roberson shdr[i].sh_size); 1051eddfbb76SRobert Watson #ifdef VIMAGE 1052eddfbb76SRobert Watson else if (ef->progtab[pb].addr != 1053eddfbb76SRobert Watson (void *)mapbase && 105417ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) 1055eddfbb76SRobert Watson vnet_data_copy(ef->progtab[pb].addr, 1056eddfbb76SRobert Watson shdr[i].sh_size); 1057eddfbb76SRobert Watson #endif 105850c202c5SJeff Roberson } else 105920947801SPeter Wemm bzero(ef->progtab[pb].addr, shdr[i].sh_size); 1060a8774e39SPeter Wemm 1061a8774e39SPeter Wemm /* Update all symbol values with the offset. */ 1062a8774e39SPeter Wemm for (j = 0; j < ef->ddbsymcnt; j++) { 1063a8774e39SPeter Wemm es = &ef->ddbsymtab[j]; 1064a8774e39SPeter Wemm if (es->st_shndx != i) 1065a8774e39SPeter Wemm continue; 1066a8774e39SPeter Wemm es->st_value += (Elf_Addr)ef->progtab[pb].addr; 1067a8774e39SPeter Wemm } 106820947801SPeter Wemm mapbase += shdr[i].sh_size; 106920947801SPeter Wemm pb++; 107020947801SPeter Wemm break; 107120947801SPeter Wemm case SHT_REL: 107215746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 107315746ef4SJohn Baldwin break; 107420947801SPeter Wemm ef->reltab[rl].rel = malloc(shdr[i].sh_size, M_LINKER, 107520947801SPeter Wemm M_WAITOK); 107620947801SPeter Wemm ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel); 107720947801SPeter Wemm ef->reltab[rl].sec = shdr[i].sh_info; 1078a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 107920947801SPeter Wemm (void *)ef->reltab[rl].rel, 108020947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 1081e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 1082e9eabf59SPeter Wemm &resid, td); 1083e9eabf59SPeter Wemm if (error) 1084e9eabf59SPeter Wemm goto out; 108520947801SPeter Wemm if (resid != 0){ 108620947801SPeter Wemm error = EINVAL; 108720947801SPeter Wemm goto out; 1088e9eabf59SPeter Wemm } 108920947801SPeter Wemm rl++; 109020947801SPeter Wemm break; 109120947801SPeter Wemm case SHT_RELA: 109215746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 109315746ef4SJohn Baldwin break; 109420947801SPeter Wemm ef->relatab[ra].rela = malloc(shdr[i].sh_size, M_LINKER, 109520947801SPeter Wemm M_WAITOK); 109620947801SPeter Wemm ef->relatab[ra].nrela = 109720947801SPeter Wemm shdr[i].sh_size / sizeof(Elf_Rela); 109820947801SPeter Wemm ef->relatab[ra].sec = shdr[i].sh_info; 1099a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 110020947801SPeter Wemm (void *)ef->relatab[ra].rela, 110120947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 1102e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 1103e9eabf59SPeter Wemm &resid, td); 1104e9eabf59SPeter Wemm if (error) 1105e9eabf59SPeter Wemm goto out; 110620947801SPeter Wemm if (resid != 0){ 110720947801SPeter Wemm error = EINVAL; 110820947801SPeter Wemm goto out; 1109e9eabf59SPeter Wemm } 111020947801SPeter Wemm ra++; 111120947801SPeter Wemm break; 111220947801SPeter Wemm } 111320947801SPeter Wemm } 11143cfce8e4SKonstantin Belousov if (pb != ef->nprogtab) { 11153cfce8e4SKonstantin Belousov link_elf_error(filename, "lost progbits"); 11163cfce8e4SKonstantin Belousov error = ENOEXEC; 11173cfce8e4SKonstantin Belousov goto out; 11183cfce8e4SKonstantin Belousov } 11193cfce8e4SKonstantin Belousov if (rl != ef->nreltab) { 11203cfce8e4SKonstantin Belousov link_elf_error(filename, "lost reltab"); 11213cfce8e4SKonstantin Belousov error = ENOEXEC; 11223cfce8e4SKonstantin Belousov goto out; 11233cfce8e4SKonstantin Belousov } 11243cfce8e4SKonstantin Belousov if (ra != ef->nrelatab) { 11253cfce8e4SKonstantin Belousov link_elf_error(filename, "lost relatab"); 11263cfce8e4SKonstantin Belousov error = ENOEXEC; 11273cfce8e4SKonstantin Belousov goto out; 11283cfce8e4SKonstantin Belousov } 11293cfce8e4SKonstantin Belousov if (mapbase != (vm_offset_t)ef->address + mapsize) { 11303cfce8e4SKonstantin Belousov printf( 11313cfce8e4SKonstantin Belousov "%s: mapbase 0x%lx != address %p + mapsize 0x%lx (0x%lx)\n", 11323cfce8e4SKonstantin Belousov filename != NULL ? filename : "<none>", 11336f3c6327SNeel Natu (u_long)mapbase, ef->address, (u_long)mapsize, 11346f3c6327SNeel Natu (u_long)(vm_offset_t)ef->address + mapsize); 11353cfce8e4SKonstantin Belousov error = ENOMEM; 11363cfce8e4SKonstantin Belousov goto out; 11373cfce8e4SKonstantin Belousov } 1138e9eabf59SPeter Wemm 1139e9eabf59SPeter Wemm /* Local intra-module relocations */ 1140289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, false); 11413cfce8e4SKonstantin Belousov if (error != 0) 11423cfce8e4SKonstantin Belousov goto out; 11437251b4bfSJake Burkholder 1144e9eabf59SPeter Wemm /* Pull in dependencies */ 1145b249ce48SMateusz Guzik VOP_UNLOCK(nd->ni_vp); 11467b9716baSIan Dowse error = linker_load_dependencies(lf); 1147a344babfSGleb Smirnoff vn_lock(nd->ni_vp, LK_EXCLUSIVE | LK_RETRY); 1148ca65d5c7SPeter Wemm if (error) 1149de78ca7eSPeter Wemm goto out; 1150e9eabf59SPeter Wemm 1151e9eabf59SPeter Wemm /* External relocations */ 1152326e27d8SDoug Rabson error = relocate_file(ef); 1153ca65d5c7SPeter Wemm if (error) 1154ca65d5c7SPeter Wemm goto out; 1155ca65d5c7SPeter Wemm 1156e9eabf59SPeter Wemm /* Notify MD code that a module is being loaded. */ 1157e9eabf59SPeter Wemm error = elf_cpu_load_file(lf); 1158ca65d5c7SPeter Wemm if (error) 1159ca65d5c7SPeter Wemm goto out; 1160a4f67738SDoug Rabson 11618f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__) 1162289ead7cSKonstantin Belousov /* Now ifuncs. */ 1163289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, true); 1164289ead7cSKonstantin Belousov if (error != 0) 1165289ead7cSKonstantin Belousov goto out; 11668f941950SKonstantin Belousov #endif 1167289ead7cSKonstantin Belousov 11681d9eae9fSMark Johnston link_elf_protect(ef); 11690067051fSMarcel Moolenaar link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size); 1170a4f67738SDoug Rabson *result = lf; 1171a4f67738SDoug Rabson 1172a4f67738SDoug Rabson out: 1173b249ce48SMateusz Guzik VOP_UNLOCK(nd->ni_vp); 1174a344babfSGleb Smirnoff vn_close(nd->ni_vp, FREAD, td->td_ucred, td); 1175a344babfSGleb Smirnoff free(nd, M_TEMP); 1176ca65d5c7SPeter Wemm if (error && lf) 117765a311fcSPoul-Henning Kamp linker_file_unload(lf, LINKER_UNLOAD_FORCE); 1178e9eabf59SPeter Wemm free(hdr, M_LINKER); 1179a4f67738SDoug Rabson 1180a4f67738SDoug Rabson return error; 1181a4f67738SDoug Rabson } 1182a4f67738SDoug Rabson 1183a4f67738SDoug Rabson static void 1184de78ca7eSPeter Wemm link_elf_unload_file(linker_file_t file) 1185a4f67738SDoug Rabson { 1186326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) file; 1187d821d364SPedro F. Giffuni u_int i; 1188a4f67738SDoug Rabson 11891aeb23cdSMarcel Moolenaar /* Notify MD code that a module is being unloaded. */ 11901aeb23cdSMarcel Moolenaar elf_cpu_unload_file(file); 11911aeb23cdSMarcel Moolenaar 119250c202c5SJeff Roberson if (ef->progtab) { 119350c202c5SJeff Roberson for (i = 0; i < ef->nprogtab; i++) { 119450c202c5SJeff Roberson if (ef->progtab[i].size == 0) 119550c202c5SJeff Roberson continue; 119650c202c5SJeff Roberson if (ef->progtab[i].name == NULL) 119750c202c5SJeff Roberson continue; 11985f67450dSDimitry Andric if (!strcmp(ef->progtab[i].name, DPCPU_SETNAME)) 119950c202c5SJeff Roberson dpcpu_free(ef->progtab[i].addr, 120050c202c5SJeff Roberson ef->progtab[i].size); 1201eddfbb76SRobert Watson #ifdef VIMAGE 120217ef1febSRobert Watson else if (!strcmp(ef->progtab[i].name, VNET_SETNAME)) 1203eddfbb76SRobert Watson vnet_data_free(ef->progtab[i].addr, 1204eddfbb76SRobert Watson ef->progtab[i].size); 1205eddfbb76SRobert Watson #endif 120650c202c5SJeff Roberson } 120750c202c5SJeff Roberson } 120870b7ffeeSIan Dowse if (ef->preloaded) { 120970b7ffeeSIan Dowse free(ef->reltab, M_LINKER); 121070b7ffeeSIan Dowse free(ef->relatab, M_LINKER); 121170b7ffeeSIan Dowse free(ef->progtab, M_LINKER); 1212a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 1213a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 1214a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 121573624a80SMark Johnston if (file->pathname != NULL) 121673624a80SMark Johnston preload_delete_name(file->pathname); 121770b7ffeeSIan Dowse return; 121870b7ffeeSIan Dowse } 121970b7ffeeSIan Dowse 1220b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) 122120947801SPeter Wemm free(ef->reltab[i].rel, M_LINKER); 1222b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) 122320947801SPeter Wemm free(ef->relatab[i].rela, M_LINKER); 122420947801SPeter Wemm free(ef->reltab, M_LINKER); 122520947801SPeter Wemm free(ef->relatab, M_LINKER); 122620947801SPeter Wemm free(ef->progtab, M_LINKER); 122720947801SPeter Wemm 1228092bacb2SMark Johnston if (ef->object != NULL) 1229fe3db7c7SDoug Rabson vm_map_remove(kernel_map, (vm_offset_t)ef->address, 1230092bacb2SMark Johnston (vm_offset_t)ef->address + ptoa(ef->object->size)); 1231e9eabf59SPeter Wemm free(ef->e_shdr, M_LINKER); 1232e9eabf59SPeter Wemm free(ef->ddbsymtab, M_LINKER); 1233e9eabf59SPeter Wemm free(ef->ddbstrtab, M_LINKER); 123420947801SPeter Wemm free(ef->shstrtab, M_LINKER); 1235a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 1236a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 1237a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 1238de78ca7eSPeter Wemm } 1239de78ca7eSPeter Wemm 1240fe3db7c7SDoug Rabson static const char * 1241757686b1SMarcel Moolenaar symbol_name(elf_file_t ef, Elf_Size r_info) 1242a4f67738SDoug Rabson { 1243fe3db7c7SDoug Rabson const Elf_Sym *ref; 1244a4f67738SDoug Rabson 1245aa855a59SPeter Wemm if (ELF_R_SYM(r_info)) { 1246e9eabf59SPeter Wemm ref = ef->ddbsymtab + ELF_R_SYM(r_info); 1247e9eabf59SPeter Wemm return ef->ddbstrtab + ref->st_name; 1248fe3db7c7SDoug Rabson } else 1249fe3db7c7SDoug Rabson return NULL; 1250a4f67738SDoug Rabson } 1251a4f67738SDoug Rabson 1252e9eabf59SPeter Wemm static Elf_Addr 1253e9eabf59SPeter Wemm findbase(elf_file_t ef, int sec) 1254e9eabf59SPeter Wemm { 1255e9eabf59SPeter Wemm int i; 1256e9eabf59SPeter Wemm Elf_Addr base = 0; 1257e9eabf59SPeter Wemm 1258e9eabf59SPeter Wemm for (i = 0; i < ef->nprogtab; i++) { 1259a8774e39SPeter Wemm if (sec == ef->progtab[i].sec) { 1260e9eabf59SPeter Wemm base = (Elf_Addr)ef->progtab[i].addr; 1261a8774e39SPeter Wemm break; 1262e9eabf59SPeter Wemm } 1263a8774e39SPeter Wemm } 1264e9eabf59SPeter Wemm return base; 1265e9eabf59SPeter Wemm } 1266e9eabf59SPeter Wemm 1267a4f67738SDoug Rabson static int 1268326e27d8SDoug Rabson relocate_file(elf_file_t ef) 1269a4f67738SDoug Rabson { 1270fe3db7c7SDoug Rabson const Elf_Rel *rellim; 1271fe3db7c7SDoug Rabson const Elf_Rel *rel; 1272fe3db7c7SDoug Rabson const Elf_Rela *relalim; 1273fe3db7c7SDoug Rabson const Elf_Rela *rela; 1274aa855a59SPeter Wemm const char *symname; 1275e9eabf59SPeter Wemm const Elf_Sym *sym; 1276e9eabf59SPeter Wemm int i; 1277757686b1SMarcel Moolenaar Elf_Size symidx; 1278e9eabf59SPeter Wemm Elf_Addr base; 1279e9eabf59SPeter Wemm 1280fe3db7c7SDoug Rabson /* Perform relocations without addend if there are any: */ 1281b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) { 1282e9eabf59SPeter Wemm rel = ef->reltab[i].rel; 12833cfce8e4SKonstantin Belousov if (rel == NULL) { 12843cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a reltab!"); 12853cfce8e4SKonstantin Belousov return (ENOEXEC); 12863cfce8e4SKonstantin Belousov } 128720947801SPeter Wemm rellim = rel + ef->reltab[i].nrel; 1288e9eabf59SPeter Wemm base = findbase(ef, ef->reltab[i].sec); 12893cfce8e4SKonstantin Belousov if (base == 0) { 12903cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 12913cfce8e4SKonstantin Belousov return (ENOEXEC); 12923cfce8e4SKonstantin Belousov } 129320947801SPeter Wemm for ( ; rel < rellim; rel++) { 1294e9eabf59SPeter Wemm symidx = ELF_R_SYM(rel->r_info); 129520947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 129620947801SPeter Wemm continue; 1297e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 129820947801SPeter Wemm /* Local relocs are already done */ 129920947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) 130020947801SPeter Wemm continue; 130120947801SPeter Wemm if (elf_reloc(&ef->lf, base, rel, ELF_RELOC_REL, 130220947801SPeter Wemm elf_obj_lookup)) { 1303aa855a59SPeter Wemm symname = symbol_name(ef, rel->r_info); 130420947801SPeter Wemm printf("link_elf_obj: symbol %s undefined\n", 130520947801SPeter Wemm symname); 13063cfce8e4SKonstantin Belousov return (ENOENT); 1307a13ddfb6SPeter Wemm } 1308aa855a59SPeter Wemm } 1309a4f67738SDoug Rabson } 1310a4f67738SDoug Rabson 1311fe3db7c7SDoug Rabson /* Perform relocations with addend if there are any: */ 1312b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) { 1313e9eabf59SPeter Wemm rela = ef->relatab[i].rela; 13143cfce8e4SKonstantin Belousov if (rela == NULL) { 13153cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a relatab!"); 13163cfce8e4SKonstantin Belousov return (ENOEXEC); 13173cfce8e4SKonstantin Belousov } 131820947801SPeter Wemm relalim = rela + ef->relatab[i].nrela; 1319e9eabf59SPeter Wemm base = findbase(ef, ef->relatab[i].sec); 13203cfce8e4SKonstantin Belousov if (base == 0) { 13213cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, 13223cfce8e4SKonstantin Belousov "lost base for relatab"); 13233cfce8e4SKonstantin Belousov return (ENOEXEC); 13243cfce8e4SKonstantin Belousov } 132520947801SPeter Wemm for ( ; rela < relalim; rela++) { 1326e9eabf59SPeter Wemm symidx = ELF_R_SYM(rela->r_info); 132720947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 132820947801SPeter Wemm continue; 1329e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 133020947801SPeter Wemm /* Local relocs are already done */ 133120947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) 133220947801SPeter Wemm continue; 133320947801SPeter Wemm if (elf_reloc(&ef->lf, base, rela, ELF_RELOC_RELA, 133420947801SPeter Wemm elf_obj_lookup)) { 1335aa855a59SPeter Wemm symname = symbol_name(ef, rela->r_info); 133620947801SPeter Wemm printf("link_elf_obj: symbol %s undefined\n", 133720947801SPeter Wemm symname); 13383cfce8e4SKonstantin Belousov return (ENOENT); 1339a13ddfb6SPeter Wemm } 1340e9eabf59SPeter Wemm } 1341e9eabf59SPeter Wemm } 1342a4f67738SDoug Rabson 13432832cd54SKonstantin Belousov /* 1344e3043798SPedro F. Giffuni * Only clean SHN_FBSD_CACHED for successful return. If we 13452832cd54SKonstantin Belousov * modified symbol table for the object but found an 13462832cd54SKonstantin Belousov * unresolved symbol, there is no reason to roll back. 13472832cd54SKonstantin Belousov */ 13482832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(ef); 13492832cd54SKonstantin Belousov 13503cfce8e4SKonstantin Belousov return (0); 1351a4f67738SDoug Rabson } 1352a4f67738SDoug Rabson 135337c84183SPoul-Henning Kamp static int 1354d254af07SMatthew Dillon link_elf_lookup_symbol(linker_file_t lf, const char *name, c_linker_sym_t *sym) 1355a4f67738SDoug Rabson { 1356326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 13572d636ab0SPeter Wemm const Elf_Sym *symp; 13582d636ab0SPeter Wemm const char *strp; 1359a4f67738SDoug Rabson int i; 1360a4f67738SDoug Rabson 13612d636ab0SPeter Wemm for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 13622d636ab0SPeter Wemm strp = ef->ddbstrtab + symp->st_name; 1363a8774e39SPeter Wemm if (symp->st_shndx != SHN_UNDEF && strcmp(name, strp) == 0) { 1364d254af07SMatthew Dillon *sym = (c_linker_sym_t) symp; 13652d636ab0SPeter Wemm return 0; 13662d636ab0SPeter Wemm } 13672d636ab0SPeter Wemm } 1368a4f67738SDoug Rabson return ENOENT; 1369a4f67738SDoug Rabson } 1370a4f67738SDoug Rabson 1371de78ca7eSPeter Wemm static int 137220947801SPeter Wemm link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym, 137320947801SPeter Wemm linker_symval_t *symval) 1374a4f67738SDoug Rabson { 13757035cf14SKonstantin Belousov elf_file_t ef; 13767035cf14SKonstantin Belousov const Elf_Sym *es; 13777035cf14SKonstantin Belousov caddr_t val; 1378a4f67738SDoug Rabson 13797035cf14SKonstantin Belousov ef = (elf_file_t) lf; 13807035cf14SKonstantin Belousov es = (const Elf_Sym*) sym; 13817035cf14SKonstantin Belousov val = (caddr_t)es->st_value; 13825cf87418SMarcel Moolenaar if (es >= ef->ddbsymtab && es < (ef->ddbsymtab + ef->ddbsymcnt)) { 13832d636ab0SPeter Wemm symval->name = ef->ddbstrtab + es->st_name; 13847035cf14SKonstantin Belousov val = (caddr_t)es->st_value; 13857035cf14SKonstantin Belousov if (ELF_ST_TYPE(es->st_info) == STT_GNU_IFUNC) 13867035cf14SKonstantin Belousov val = ((caddr_t (*)(void))val)(); 13877035cf14SKonstantin Belousov symval->value = val; 13882d636ab0SPeter Wemm symval->size = es->st_size; 13892d636ab0SPeter Wemm return 0; 13902d636ab0SPeter Wemm } 13912d636ab0SPeter Wemm return ENOENT; 13922d636ab0SPeter Wemm } 1393a4f67738SDoug Rabson 1394a4f67738SDoug Rabson static int 1395a4f67738SDoug Rabson link_elf_search_symbol(linker_file_t lf, caddr_t value, 1396d254af07SMatthew Dillon c_linker_sym_t *sym, long *diffp) 1397a4f67738SDoug Rabson { 1398326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 1399586453feSBruce Evans u_long off = (uintptr_t) (void *) value; 1400a4f67738SDoug Rabson u_long diff = off; 1401586453feSBruce Evans u_long st_value; 1402fe3db7c7SDoug Rabson const Elf_Sym *es; 1403b85f65afSPedro F. Giffuni const Elf_Sym *best = NULL; 1404a4f67738SDoug Rabson int i; 1405a4f67738SDoug Rabson 14062d636ab0SPeter Wemm for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) { 1407a4f67738SDoug Rabson if (es->st_name == 0) 1408a4f67738SDoug Rabson continue; 1409a8774e39SPeter Wemm st_value = es->st_value; 1410b5abfb70SPeter Wemm if (off >= st_value) { 1411b5abfb70SPeter Wemm if (off - st_value < diff) { 1412b5abfb70SPeter Wemm diff = off - st_value; 1413a4f67738SDoug Rabson best = es; 1414a4f67738SDoug Rabson if (diff == 0) 1415a4f67738SDoug Rabson break; 1416b5abfb70SPeter Wemm } else if (off - st_value == diff) { 1417a4f67738SDoug Rabson best = es; 1418a4f67738SDoug Rabson } 1419a4f67738SDoug Rabson } 1420a4f67738SDoug Rabson } 1421b85f65afSPedro F. Giffuni if (best == NULL) 1422a4f67738SDoug Rabson *diffp = off; 1423a4f67738SDoug Rabson else 1424a4f67738SDoug Rabson *diffp = diff; 1425d254af07SMatthew Dillon *sym = (c_linker_sym_t) best; 1426a4f67738SDoug Rabson 1427a4f67738SDoug Rabson return 0; 1428a4f67738SDoug Rabson } 1429f41325dbSPeter Wemm 1430f41325dbSPeter Wemm /* 1431f41325dbSPeter Wemm * Look up a linker set on an ELF system. 1432f41325dbSPeter Wemm */ 1433f41325dbSPeter Wemm static int 1434f41325dbSPeter Wemm link_elf_lookup_set(linker_file_t lf, const char *name, 1435f41325dbSPeter Wemm void ***startp, void ***stopp, int *countp) 1436f41325dbSPeter Wemm { 1437e9eabf59SPeter Wemm elf_file_t ef = (elf_file_t)lf; 1438f41325dbSPeter Wemm void **start, **stop; 1439e9eabf59SPeter Wemm int i, count; 1440f41325dbSPeter Wemm 1441e9eabf59SPeter Wemm /* Relative to section number */ 1442e9eabf59SPeter Wemm for (i = 0; i < ef->nprogtab; i++) { 1443e9eabf59SPeter Wemm if ((strncmp(ef->progtab[i].name, "set_", 4) == 0) && 1444e9eabf59SPeter Wemm strcmp(ef->progtab[i].name + 4, name) == 0) { 1445e9eabf59SPeter Wemm start = (void **)ef->progtab[i].addr; 144620947801SPeter Wemm stop = (void **)((char *)ef->progtab[i].addr + 144720947801SPeter Wemm ef->progtab[i].size); 1448f41325dbSPeter Wemm count = stop - start; 1449f41325dbSPeter Wemm if (startp) 1450f41325dbSPeter Wemm *startp = start; 1451f41325dbSPeter Wemm if (stopp) 1452f41325dbSPeter Wemm *stopp = stop; 1453f41325dbSPeter Wemm if (countp) 1454f41325dbSPeter Wemm *countp = count; 1455e9eabf59SPeter Wemm return (0); 1456e9eabf59SPeter Wemm } 1457e9eabf59SPeter Wemm } 1458e9eabf59SPeter Wemm return (ESRCH); 1459f41325dbSPeter Wemm } 1460bb9fe9ddSBrian Feldman 1461bb9fe9ddSBrian Feldman static int 1462bb9fe9ddSBrian Feldman link_elf_each_function_name(linker_file_t file, 1463e9eabf59SPeter Wemm int (*callback)(const char *, void *), void *opaque) 1464e9eabf59SPeter Wemm { 1465bb9fe9ddSBrian Feldman elf_file_t ef = (elf_file_t)file; 1466bb9fe9ddSBrian Feldman const Elf_Sym *symp; 1467bb9fe9ddSBrian Feldman int i, error; 1468bb9fe9ddSBrian Feldman 1469bb9fe9ddSBrian Feldman /* Exhaustive search */ 1470bb9fe9ddSBrian Feldman for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1471bb9fe9ddSBrian Feldman if (symp->st_value != 0 && 14727035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 14737035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) { 1474bb9fe9ddSBrian Feldman error = callback(ef->ddbstrtab + symp->st_name, opaque); 1475bb9fe9ddSBrian Feldman if (error) 1476bb9fe9ddSBrian Feldman return (error); 1477bb9fe9ddSBrian Feldman } 1478bb9fe9ddSBrian Feldman } 1479bb9fe9ddSBrian Feldman return (0); 1480bb9fe9ddSBrian Feldman } 148184201059SMarcel Moolenaar 1482a2024a3eSJohn Birrell static int 1483a2024a3eSJohn Birrell link_elf_each_function_nameval(linker_file_t file, 1484a2024a3eSJohn Birrell linker_function_nameval_callback_t callback, void *opaque) 1485a2024a3eSJohn Birrell { 1486a2024a3eSJohn Birrell linker_symval_t symval; 1487a2024a3eSJohn Birrell elf_file_t ef = (elf_file_t)file; 1488a2024a3eSJohn Birrell const Elf_Sym* symp; 1489a2024a3eSJohn Birrell int i, error; 1490a2024a3eSJohn Birrell 1491a2024a3eSJohn Birrell /* Exhaustive search */ 1492a2024a3eSJohn Birrell for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1493a2024a3eSJohn Birrell if (symp->st_value != 0 && 14947035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 14957035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) { 14967035cf14SKonstantin Belousov error = link_elf_symbol_values(file, 14977035cf14SKonstantin Belousov (c_linker_sym_t)symp, &symval); 1498a2024a3eSJohn Birrell if (error) 1499a2024a3eSJohn Birrell return (error); 1500a2024a3eSJohn Birrell error = callback(file, i, &symval, opaque); 1501a2024a3eSJohn Birrell if (error) 1502a2024a3eSJohn Birrell return (error); 1503a2024a3eSJohn Birrell } 1504a2024a3eSJohn Birrell } 1505a2024a3eSJohn Birrell return (0); 1506a2024a3eSJohn Birrell } 1507a2024a3eSJohn Birrell 15082832cd54SKonstantin Belousov static void 15092832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(elf_file_t ef) 15102832cd54SKonstantin Belousov { 15112832cd54SKonstantin Belousov Elf_Sym *sym; 15122832cd54SKonstantin Belousov Elf_Size i; 15132832cd54SKonstantin Belousov 15142832cd54SKonstantin Belousov for (i = 0; i < ef->ddbsymcnt; i++) { 15152832cd54SKonstantin Belousov sym = ef->ddbsymtab + i; 15162832cd54SKonstantin Belousov if (sym->st_shndx == SHN_FBSD_CACHED) { 15172832cd54SKonstantin Belousov sym->st_shndx = SHN_UNDEF; 15182832cd54SKonstantin Belousov sym->st_value = 0; 15192832cd54SKonstantin Belousov } 15202832cd54SKonstantin Belousov } 15212832cd54SKonstantin Belousov } 15222832cd54SKonstantin Belousov 1523d297ad16SMarcel Moolenaar /* 1524d297ad16SMarcel Moolenaar * Symbol lookup function that can be used when the symbol index is known (ie 1525d297ad16SMarcel Moolenaar * in relocations). It uses the symbol index instead of doing a fully fledged 1526d297ad16SMarcel Moolenaar * hash table based lookup when such is valid. For example for local symbols. 1527d297ad16SMarcel Moolenaar * This is not only more efficient, it's also more correct. It's not always 1528d297ad16SMarcel Moolenaar * the case that the symbol can be found through the hash table. 1529d297ad16SMarcel Moolenaar */ 1530cff8c6f2SKonstantin Belousov static int 1531cff8c6f2SKonstantin Belousov elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, Elf_Addr *res) 1532d297ad16SMarcel Moolenaar { 1533d297ad16SMarcel Moolenaar elf_file_t ef = (elf_file_t)lf; 15342832cd54SKonstantin Belousov Elf_Sym *sym; 1535d297ad16SMarcel Moolenaar const char *symbol; 1536cff8c6f2SKonstantin Belousov Elf_Addr res1; 1537d297ad16SMarcel Moolenaar 1538d297ad16SMarcel Moolenaar /* Don't even try to lookup the symbol if the index is bogus. */ 1539cff8c6f2SKonstantin Belousov if (symidx >= ef->ddbsymcnt) { 1540cff8c6f2SKonstantin Belousov *res = 0; 1541cff8c6f2SKonstantin Belousov return (EINVAL); 1542cff8c6f2SKonstantin Belousov } 1543d297ad16SMarcel Moolenaar 1544e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 1545d297ad16SMarcel Moolenaar 15464cec6f5dSPeter Wemm /* Quick answer if there is a definition included. */ 1547cff8c6f2SKonstantin Belousov if (sym->st_shndx != SHN_UNDEF) { 1548289ead7cSKonstantin Belousov res1 = (Elf_Addr)sym->st_value; 1549289ead7cSKonstantin Belousov if (ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC) 1550289ead7cSKonstantin Belousov res1 = ((Elf_Addr (*)(void))res1)(); 1551289ead7cSKonstantin Belousov *res = res1; 1552cff8c6f2SKonstantin Belousov return (0); 1553cff8c6f2SKonstantin Belousov } 15544cec6f5dSPeter Wemm 15554cec6f5dSPeter Wemm /* If we get here, then it is undefined and needs a lookup. */ 15564cec6f5dSPeter Wemm switch (ELF_ST_BIND(sym->st_info)) { 15574cec6f5dSPeter Wemm case STB_LOCAL: 15584cec6f5dSPeter Wemm /* Local, but undefined? huh? */ 1559cff8c6f2SKonstantin Belousov *res = 0; 1560cff8c6f2SKonstantin Belousov return (EINVAL); 1561d297ad16SMarcel Moolenaar 1562e9eabf59SPeter Wemm case STB_GLOBAL: 1563cff8c6f2SKonstantin Belousov case STB_WEAK: 1564e9eabf59SPeter Wemm /* Relative to Data or Function name */ 1565e9eabf59SPeter Wemm symbol = ef->ddbstrtab + sym->st_name; 1566d297ad16SMarcel Moolenaar 1567d297ad16SMarcel Moolenaar /* Force a lookup failure if the symbol name is bogus. */ 1568cff8c6f2SKonstantin Belousov if (*symbol == 0) { 1569cff8c6f2SKonstantin Belousov *res = 0; 1570cff8c6f2SKonstantin Belousov return (EINVAL); 1571cff8c6f2SKonstantin Belousov } 1572cff8c6f2SKonstantin Belousov res1 = (Elf_Addr)linker_file_lookup_symbol(lf, symbol, deps); 15732832cd54SKonstantin Belousov 15742832cd54SKonstantin Belousov /* 15752832cd54SKonstantin Belousov * Cache global lookups during module relocation. The failure 15762832cd54SKonstantin Belousov * case is particularly expensive for callers, who must scan 15772832cd54SKonstantin Belousov * through the entire globals table doing strcmp(). Cache to 15782832cd54SKonstantin Belousov * avoid doing such work repeatedly. 15792832cd54SKonstantin Belousov * 15802832cd54SKonstantin Belousov * After relocation is complete, undefined globals will be 15812832cd54SKonstantin Belousov * restored to SHN_UNDEF in elf_obj_cleanup_globals_cache(), 15822832cd54SKonstantin Belousov * above. 15832832cd54SKonstantin Belousov */ 1584cff8c6f2SKonstantin Belousov if (res1 != 0) { 15852832cd54SKonstantin Belousov sym->st_shndx = SHN_FBSD_CACHED; 1586cff8c6f2SKonstantin Belousov sym->st_value = res1; 1587cff8c6f2SKonstantin Belousov *res = res1; 1588e9eabf59SPeter Wemm return (0); 1589cff8c6f2SKonstantin Belousov } else if (ELF_ST_BIND(sym->st_info) == STB_WEAK) { 1590cff8c6f2SKonstantin Belousov sym->st_value = 0; 1591cff8c6f2SKonstantin Belousov *res = 0; 1592cff8c6f2SKonstantin Belousov return (0); 1593cff8c6f2SKonstantin Belousov } 1594cff8c6f2SKonstantin Belousov return (EINVAL); 1595e9eabf59SPeter Wemm 1596e9eabf59SPeter Wemm default: 1597cff8c6f2SKonstantin Belousov return (EINVAL); 1598d297ad16SMarcel Moolenaar } 1599e9eabf59SPeter Wemm } 1600e9eabf59SPeter Wemm 16017251b4bfSJake Burkholder static void 16027226306eSKonstantin Belousov link_elf_fix_link_set(elf_file_t ef) 16037226306eSKonstantin Belousov { 16047226306eSKonstantin Belousov static const char startn[] = "__start_"; 16057226306eSKonstantin Belousov static const char stopn[] = "__stop_"; 16067226306eSKonstantin Belousov Elf_Sym *sym; 16077226306eSKonstantin Belousov const char *sym_name, *linkset_name; 16087226306eSKonstantin Belousov Elf_Addr startp, stopp; 16097226306eSKonstantin Belousov Elf_Size symidx; 16107226306eSKonstantin Belousov int start, i; 16117226306eSKonstantin Belousov 16127226306eSKonstantin Belousov startp = stopp = 0; 16137226306eSKonstantin Belousov for (symidx = 1 /* zero entry is special */; 16147226306eSKonstantin Belousov symidx < ef->ddbsymcnt; symidx++) { 16157226306eSKonstantin Belousov sym = ef->ddbsymtab + symidx; 16167226306eSKonstantin Belousov if (sym->st_shndx != SHN_UNDEF) 16177226306eSKonstantin Belousov continue; 16187226306eSKonstantin Belousov 16197226306eSKonstantin Belousov sym_name = ef->ddbstrtab + sym->st_name; 16207226306eSKonstantin Belousov if (strncmp(sym_name, startn, sizeof(startn) - 1) == 0) { 16217226306eSKonstantin Belousov start = 1; 16227226306eSKonstantin Belousov linkset_name = sym_name + sizeof(startn) - 1; 16237226306eSKonstantin Belousov } 16247226306eSKonstantin Belousov else if (strncmp(sym_name, stopn, sizeof(stopn) - 1) == 0) { 16257226306eSKonstantin Belousov start = 0; 16267226306eSKonstantin Belousov linkset_name = sym_name + sizeof(stopn) - 1; 16277226306eSKonstantin Belousov } 16287226306eSKonstantin Belousov else 16297226306eSKonstantin Belousov continue; 16307226306eSKonstantin Belousov 16317226306eSKonstantin Belousov for (i = 0; i < ef->nprogtab; i++) { 16327226306eSKonstantin Belousov if (strcmp(ef->progtab[i].name, linkset_name) == 0) { 16337226306eSKonstantin Belousov startp = (Elf_Addr)ef->progtab[i].addr; 16347226306eSKonstantin Belousov stopp = (Elf_Addr)(startp + ef->progtab[i].size); 16357226306eSKonstantin Belousov break; 16367226306eSKonstantin Belousov } 16377226306eSKonstantin Belousov } 16387226306eSKonstantin Belousov if (i == ef->nprogtab) 16397226306eSKonstantin Belousov continue; 16407226306eSKonstantin Belousov 16417226306eSKonstantin Belousov sym->st_value = start ? startp : stopp; 16427226306eSKonstantin Belousov sym->st_shndx = i; 16437226306eSKonstantin Belousov } 16447226306eSKonstantin Belousov } 16457226306eSKonstantin Belousov 16463cfce8e4SKonstantin Belousov static int 1647289ead7cSKonstantin Belousov link_elf_reloc_local(linker_file_t lf, bool ifuncs) 16487251b4bfSJake Burkholder { 1649e9eabf59SPeter Wemm elf_file_t ef = (elf_file_t)lf; 16507251b4bfSJake Burkholder const Elf_Rel *rellim; 16517251b4bfSJake Burkholder const Elf_Rel *rel; 16527251b4bfSJake Burkholder const Elf_Rela *relalim; 16537251b4bfSJake Burkholder const Elf_Rela *rela; 1654e9eabf59SPeter Wemm const Elf_Sym *sym; 1655e9eabf59SPeter Wemm Elf_Addr base; 1656e9eabf59SPeter Wemm int i; 1657757686b1SMarcel Moolenaar Elf_Size symidx; 1658e9eabf59SPeter Wemm 16597226306eSKonstantin Belousov link_elf_fix_link_set(ef); 16607226306eSKonstantin Belousov 16617251b4bfSJake Burkholder /* Perform relocations without addend if there are any: */ 1662b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) { 1663e9eabf59SPeter Wemm rel = ef->reltab[i].rel; 16643cfce8e4SKonstantin Belousov if (rel == NULL) { 16653cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a reltab"); 16663cfce8e4SKonstantin Belousov return (ENOEXEC); 16673cfce8e4SKonstantin Belousov } 166820947801SPeter Wemm rellim = rel + ef->reltab[i].nrel; 1669e9eabf59SPeter Wemm base = findbase(ef, ef->reltab[i].sec); 16703cfce8e4SKonstantin Belousov if (base == 0) { 16713cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 16723cfce8e4SKonstantin Belousov return (ENOEXEC); 16733cfce8e4SKonstantin Belousov } 167420947801SPeter Wemm for ( ; rel < rellim; rel++) { 1675e9eabf59SPeter Wemm symidx = ELF_R_SYM(rel->r_info); 167620947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 167720947801SPeter Wemm continue; 1678e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 167920947801SPeter Wemm /* Only do local relocs */ 168020947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) != STB_LOCAL) 168120947801SPeter Wemm continue; 168236716fe2SMark Johnston if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC || 168344c705cfSMitchell Horne elf_is_ifunc_reloc(rel->r_info)) != ifuncs) 168444c705cfSMitchell Horne continue; 168544c705cfSMitchell Horne if (elf_reloc_local(lf, base, rel, ELF_RELOC_REL, 168644c705cfSMitchell Horne elf_obj_lookup) != 0) 168744c705cfSMitchell Horne return (ENOEXEC); 16887251b4bfSJake Burkholder } 1689e9eabf59SPeter Wemm } 16907251b4bfSJake Burkholder 16917251b4bfSJake Burkholder /* Perform relocations with addend if there are any: */ 1692b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) { 1693e9eabf59SPeter Wemm rela = ef->relatab[i].rela; 16943cfce8e4SKonstantin Belousov if (rela == NULL) { 16953cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a relatab!"); 16963cfce8e4SKonstantin Belousov return (ENOEXEC); 16973cfce8e4SKonstantin Belousov } 169820947801SPeter Wemm relalim = rela + ef->relatab[i].nrela; 1699e9eabf59SPeter Wemm base = findbase(ef, ef->relatab[i].sec); 17003cfce8e4SKonstantin Belousov if (base == 0) { 17013cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 17023cfce8e4SKonstantin Belousov return (ENOEXEC); 17033cfce8e4SKonstantin Belousov } 170420947801SPeter Wemm for ( ; rela < relalim; rela++) { 1705e9eabf59SPeter Wemm symidx = ELF_R_SYM(rela->r_info); 170620947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 170720947801SPeter Wemm continue; 1708e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 170920947801SPeter Wemm /* Only do local relocs */ 171020947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) != STB_LOCAL) 171120947801SPeter Wemm continue; 171236716fe2SMark Johnston if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC || 171344c705cfSMitchell Horne elf_is_ifunc_reloc(rela->r_info)) != ifuncs) 171444c705cfSMitchell Horne continue; 171544c705cfSMitchell Horne if (elf_reloc_local(lf, base, rela, ELF_RELOC_RELA, 171644c705cfSMitchell Horne elf_obj_lookup) != 0) 171744c705cfSMitchell Horne return (ENOEXEC); 17187251b4bfSJake Burkholder } 17197251b4bfSJake Burkholder } 17203cfce8e4SKonstantin Belousov return (0); 1721e9eabf59SPeter Wemm } 172200a5db46SStacey Son 172300a5db46SStacey Son static long 1724e76f11f4SAndriy Gapon link_elf_symtab_get(linker_file_t lf, const Elf_Sym **symtab) 172500a5db46SStacey Son { 172600a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 172700a5db46SStacey Son 172800a5db46SStacey Son *symtab = ef->ddbsymtab; 172900a5db46SStacey Son 173000a5db46SStacey Son if (*symtab == NULL) 173100a5db46SStacey Son return (0); 173200a5db46SStacey Son 173300a5db46SStacey Son return (ef->ddbsymcnt); 173400a5db46SStacey Son } 173500a5db46SStacey Son 173600a5db46SStacey Son static long 173700a5db46SStacey Son link_elf_strtab_get(linker_file_t lf, caddr_t *strtab) 173800a5db46SStacey Son { 173900a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 174000a5db46SStacey Son 174100a5db46SStacey Son *strtab = ef->ddbstrtab; 174200a5db46SStacey Son 174300a5db46SStacey Son if (*strtab == NULL) 174400a5db46SStacey Son return (0); 174500a5db46SStacey Son 174600a5db46SStacey Son return (ef->ddbstrcnt); 174700a5db46SStacey Son } 1748