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 } 2220659df6fSKonstantin Belousov error = vm_map_protect(kernel_map, start, end, prot, 0, 2230659df6fSKonstantin 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 389*5e6989baSMark Johnston case SHT_INIT_ARRAY: 390*5e6989baSMark Johnston case SHT_FINI_ARRAY: 39158c4aee0SJohn Baldwin /* Ignore sections not loaded by the loader. */ 39258c4aee0SJohn Baldwin if (shdr[i].sh_addr == 0) 39358c4aee0SJohn Baldwin break; 39470b7ffeeSIan Dowse ef->nprogtab++; 39570b7ffeeSIan Dowse break; 39670b7ffeeSIan Dowse case SHT_SYMTAB: 39770b7ffeeSIan Dowse symtabindex = i; 39870b7ffeeSIan Dowse symstrindex = shdr[i].sh_link; 39970b7ffeeSIan Dowse break; 40070b7ffeeSIan Dowse case SHT_REL: 40115746ef4SJohn Baldwin /* 40215746ef4SJohn Baldwin * Ignore relocation tables for sections not 40315746ef4SJohn Baldwin * loaded by the loader. 40415746ef4SJohn Baldwin */ 40515746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 40615746ef4SJohn Baldwin break; 407b474c780SDavid E. O'Brien ef->nreltab++; 40870b7ffeeSIan Dowse break; 40970b7ffeeSIan Dowse case SHT_RELA: 41015746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 41115746ef4SJohn Baldwin break; 412b474c780SDavid E. O'Brien ef->nrelatab++; 41370b7ffeeSIan Dowse break; 41470b7ffeeSIan Dowse } 41570b7ffeeSIan Dowse } 41670b7ffeeSIan Dowse 41770b7ffeeSIan Dowse shstrindex = hdr->e_shstrndx; 41870b7ffeeSIan Dowse if (ef->nprogtab == 0 || symstrindex < 0 || 41970b7ffeeSIan Dowse symstrindex >= hdr->e_shnum || 42070b7ffeeSIan Dowse shdr[symstrindex].sh_type != SHT_STRTAB || shstrindex == 0 || 42170b7ffeeSIan Dowse shstrindex >= hdr->e_shnum || 42270b7ffeeSIan Dowse shdr[shstrindex].sh_type != SHT_STRTAB) { 42370b7ffeeSIan Dowse printf("%s: bad/missing section headers\n", filename); 42470b7ffeeSIan Dowse error = ENOEXEC; 42570b7ffeeSIan Dowse goto out; 42670b7ffeeSIan Dowse } 42770b7ffeeSIan Dowse 42870b7ffeeSIan Dowse /* Allocate space for tracking the load chunks */ 42970b7ffeeSIan Dowse if (ef->nprogtab != 0) 43070b7ffeeSIan Dowse ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab), 43170b7ffeeSIan Dowse M_LINKER, M_WAITOK | M_ZERO); 432b474c780SDavid E. O'Brien if (ef->nreltab != 0) 433b474c780SDavid E. O'Brien ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab), 434b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 435b474c780SDavid E. O'Brien if (ef->nrelatab != 0) 436b474c780SDavid E. O'Brien ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab), 437b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 43870b7ffeeSIan Dowse if ((ef->nprogtab != 0 && ef->progtab == NULL) || 439b474c780SDavid E. O'Brien (ef->nreltab != 0 && ef->reltab == NULL) || 440b474c780SDavid E. O'Brien (ef->nrelatab != 0 && ef->relatab == NULL)) { 44170b7ffeeSIan Dowse error = ENOMEM; 44270b7ffeeSIan Dowse goto out; 44370b7ffeeSIan Dowse } 44470b7ffeeSIan Dowse 44570b7ffeeSIan Dowse /* XXX, relocate the sh_addr fields saved by the loader. */ 44670b7ffeeSIan Dowse off = 0; 44770b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 44870b7ffeeSIan Dowse if (shdr[i].sh_addr != 0 && (off == 0 || shdr[i].sh_addr < off)) 44970b7ffeeSIan Dowse off = shdr[i].sh_addr; 45070b7ffeeSIan Dowse } 45170b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 45270b7ffeeSIan Dowse if (shdr[i].sh_addr != 0) 45370b7ffeeSIan Dowse shdr[i].sh_addr = shdr[i].sh_addr - off + 45470b7ffeeSIan Dowse (Elf_Addr)ef->address; 45570b7ffeeSIan Dowse } 45670b7ffeeSIan Dowse 45770b7ffeeSIan Dowse ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym); 45870b7ffeeSIan Dowse ef->ddbsymtab = (Elf_Sym *)shdr[symtabindex].sh_addr; 45970b7ffeeSIan Dowse ef->ddbstrcnt = shdr[symstrindex].sh_size; 46070b7ffeeSIan Dowse ef->ddbstrtab = (char *)shdr[symstrindex].sh_addr; 46170b7ffeeSIan Dowse ef->shstrcnt = shdr[shstrindex].sh_size; 46270b7ffeeSIan Dowse ef->shstrtab = (char *)shdr[shstrindex].sh_addr; 46370b7ffeeSIan Dowse 46470b7ffeeSIan Dowse /* Now fill out progtab and the relocation tables. */ 46570b7ffeeSIan Dowse pb = 0; 46670b7ffeeSIan Dowse rl = 0; 46770b7ffeeSIan Dowse ra = 0; 46870b7ffeeSIan Dowse for (i = 0; i < hdr->e_shnum; i++) { 46970b7ffeeSIan Dowse switch (shdr[i].sh_type) { 47070b7ffeeSIan Dowse case SHT_PROGBITS: 47170b7ffeeSIan Dowse case SHT_NOBITS: 47213f28d96SKonstantin Belousov #ifdef __amd64__ 473b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 47413f28d96SKonstantin Belousov #endif 475*5e6989baSMark Johnston case SHT_FINI_ARRAY: 47658c4aee0SJohn Baldwin if (shdr[i].sh_addr == 0) 47758c4aee0SJohn Baldwin break; 47870b7ffeeSIan Dowse ef->progtab[pb].addr = (void *)shdr[i].sh_addr; 47970b7ffeeSIan Dowse if (shdr[i].sh_type == SHT_PROGBITS) 48070b7ffeeSIan Dowse ef->progtab[pb].name = "<<PROGBITS>>"; 48113f28d96SKonstantin Belousov #ifdef __amd64__ 482b715d9afSKonstantin Belousov else if (shdr[i].sh_type == SHT_X86_64_UNWIND) 48313f28d96SKonstantin Belousov ef->progtab[pb].name = "<<UNWIND>>"; 48413f28d96SKonstantin Belousov #endif 485*5e6989baSMark Johnston else if (shdr[i].sh_type == SHT_INIT_ARRAY) 486*5e6989baSMark Johnston ef->progtab[pb].name = "<<INIT_ARRAY>>"; 487*5e6989baSMark Johnston else if (shdr[i].sh_type == SHT_FINI_ARRAY) 488*5e6989baSMark Johnston ef->progtab[pb].name = "<<FINI_ARRAY>>"; 48970b7ffeeSIan Dowse else 49070b7ffeeSIan Dowse ef->progtab[pb].name = "<<NOBITS>>"; 49170b7ffeeSIan Dowse ef->progtab[pb].size = shdr[i].sh_size; 4921d9eae9fSMark Johnston ef->progtab[pb].flags = shdr[i].sh_flags; 49370b7ffeeSIan Dowse ef->progtab[pb].sec = i; 49470b7ffeeSIan Dowse if (ef->shstrtab && shdr[i].sh_name != 0) 49570b7ffeeSIan Dowse ef->progtab[pb].name = 49670b7ffeeSIan Dowse ef->shstrtab + shdr[i].sh_name; 49750c202c5SJeff Roberson if (ef->progtab[pb].name != NULL && 4985f67450dSDimitry Andric !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) { 49950c202c5SJeff Roberson void *dpcpu; 50050c202c5SJeff Roberson 50150c202c5SJeff Roberson dpcpu = dpcpu_alloc(shdr[i].sh_size); 50250c202c5SJeff Roberson if (dpcpu == NULL) { 5030f823b64SBjoern A. Zeeb printf("%s: pcpu module space is out " 5049afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 5059afc5684SBjoern A. Zeeb "for %s\n", __func__, 5069afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 5070f823b64SBjoern A. Zeeb filename); 50850c202c5SJeff Roberson error = ENOSPC; 50950c202c5SJeff Roberson goto out; 51050c202c5SJeff Roberson } 51150c202c5SJeff Roberson memcpy(dpcpu, ef->progtab[pb].addr, 51250c202c5SJeff Roberson ef->progtab[pb].size); 51350c202c5SJeff Roberson dpcpu_copy(dpcpu, shdr[i].sh_size); 51450c202c5SJeff Roberson ef->progtab[pb].addr = dpcpu; 515eddfbb76SRobert Watson #ifdef VIMAGE 516eddfbb76SRobert Watson } else if (ef->progtab[pb].name != NULL && 51717ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) { 518eddfbb76SRobert Watson void *vnet_data; 519eddfbb76SRobert Watson 520eddfbb76SRobert Watson vnet_data = vnet_data_alloc(shdr[i].sh_size); 521eddfbb76SRobert Watson if (vnet_data == NULL) { 5220f823b64SBjoern A. Zeeb printf("%s: vnet module space is out " 5239afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 5249afc5684SBjoern A. Zeeb "for %s\n", __func__, 5259afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 5260f823b64SBjoern A. Zeeb filename); 527eddfbb76SRobert Watson error = ENOSPC; 528eddfbb76SRobert Watson goto out; 529eddfbb76SRobert Watson } 530eddfbb76SRobert Watson memcpy(vnet_data, ef->progtab[pb].addr, 531eddfbb76SRobert Watson ef->progtab[pb].size); 532eddfbb76SRobert Watson vnet_data_copy(vnet_data, shdr[i].sh_size); 533eddfbb76SRobert Watson ef->progtab[pb].addr = vnet_data; 534eddfbb76SRobert Watson #endif 535*5e6989baSMark Johnston } else if ((ef->progtab[pb].name != NULL && 536*5e6989baSMark Johnston strcmp(ef->progtab[pb].name, ".ctors") == 0) || 537*5e6989baSMark Johnston shdr[i].sh_type == SHT_INIT_ARRAY) { 538*5e6989baSMark Johnston if (lf->ctors_addr != 0) { 539*5e6989baSMark Johnston printf( 540*5e6989baSMark Johnston "%s: multiple ctor sections in %s\n", 541*5e6989baSMark Johnston __func__, filename); 542*5e6989baSMark Johnston } else { 5430067051fSMarcel Moolenaar lf->ctors_addr = ef->progtab[pb].addr; 5440067051fSMarcel Moolenaar lf->ctors_size = shdr[i].sh_size; 54550c202c5SJeff Roberson } 546*5e6989baSMark Johnston } 54770b7ffeeSIan Dowse 54870b7ffeeSIan Dowse /* Update all symbol values with the offset. */ 54970b7ffeeSIan Dowse for (j = 0; j < ef->ddbsymcnt; j++) { 55070b7ffeeSIan Dowse es = &ef->ddbsymtab[j]; 55170b7ffeeSIan Dowse if (es->st_shndx != i) 55270b7ffeeSIan Dowse continue; 55370b7ffeeSIan Dowse es->st_value += (Elf_Addr)ef->progtab[pb].addr; 55470b7ffeeSIan Dowse } 55570b7ffeeSIan Dowse pb++; 55670b7ffeeSIan Dowse break; 55770b7ffeeSIan Dowse case SHT_REL: 55815746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 55915746ef4SJohn Baldwin break; 56070b7ffeeSIan Dowse ef->reltab[rl].rel = (Elf_Rel *)shdr[i].sh_addr; 56170b7ffeeSIan Dowse ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel); 56270b7ffeeSIan Dowse ef->reltab[rl].sec = shdr[i].sh_info; 56370b7ffeeSIan Dowse rl++; 56470b7ffeeSIan Dowse break; 56570b7ffeeSIan Dowse case SHT_RELA: 56615746ef4SJohn Baldwin if (shdr[shdr[i].sh_info].sh_addr == 0) 56715746ef4SJohn Baldwin break; 56870b7ffeeSIan Dowse ef->relatab[ra].rela = (Elf_Rela *)shdr[i].sh_addr; 56970b7ffeeSIan Dowse ef->relatab[ra].nrela = 57070b7ffeeSIan Dowse shdr[i].sh_size / sizeof(Elf_Rela); 57170b7ffeeSIan Dowse ef->relatab[ra].sec = shdr[i].sh_info; 57270b7ffeeSIan Dowse ra++; 57370b7ffeeSIan Dowse break; 57470b7ffeeSIan Dowse } 57570b7ffeeSIan Dowse } 5763cfce8e4SKonstantin Belousov if (pb != ef->nprogtab) { 5773cfce8e4SKonstantin Belousov printf("%s: lost progbits\n", filename); 5783cfce8e4SKonstantin Belousov error = ENOEXEC; 5793cfce8e4SKonstantin Belousov goto out; 5803cfce8e4SKonstantin Belousov } 5813cfce8e4SKonstantin Belousov if (rl != ef->nreltab) { 5823cfce8e4SKonstantin Belousov printf("%s: lost reltab\n", filename); 5833cfce8e4SKonstantin Belousov error = ENOEXEC; 5843cfce8e4SKonstantin Belousov goto out; 5853cfce8e4SKonstantin Belousov } 5863cfce8e4SKonstantin Belousov if (ra != ef->nrelatab) { 5873cfce8e4SKonstantin Belousov printf("%s: lost relatab\n", filename); 5883cfce8e4SKonstantin Belousov error = ENOEXEC; 5893cfce8e4SKonstantin Belousov goto out; 5903cfce8e4SKonstantin Belousov } 59170b7ffeeSIan Dowse 592f822c9e2SMark Johnston /* 593f822c9e2SMark Johnston * The file needs to be writeable and executable while applying 594f822c9e2SMark Johnston * relocations. Mapping protections are applied once relocation 595f822c9e2SMark Johnston * processing is complete. 596f822c9e2SMark Johnston */ 597f822c9e2SMark Johnston link_elf_protect_range(ef, (vm_offset_t)ef->address, 598f822c9e2SMark Johnston round_page((vm_offset_t)ef->address + ef->lf.size), VM_PROT_ALL); 599f822c9e2SMark Johnston 60070b7ffeeSIan Dowse /* Local intra-module relocations */ 601289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, false); 6023cfce8e4SKonstantin Belousov if (error != 0) 6033cfce8e4SKonstantin Belousov goto out; 60470b7ffeeSIan Dowse *result = lf; 60570b7ffeeSIan Dowse return (0); 60670b7ffeeSIan Dowse 60770b7ffeeSIan Dowse out: 60870b7ffeeSIan Dowse /* preload not done this way */ 60970b7ffeeSIan Dowse linker_file_unload(lf, LINKER_UNLOAD_FORCE); 61070b7ffeeSIan Dowse return (error); 61170b7ffeeSIan Dowse } 612de78ca7eSPeter Wemm 6130067051fSMarcel Moolenaar static void 6140067051fSMarcel Moolenaar link_elf_invoke_ctors(caddr_t addr, size_t size) 6150067051fSMarcel Moolenaar { 6160067051fSMarcel Moolenaar void (**ctor)(void); 6170067051fSMarcel Moolenaar size_t i, cnt; 6180067051fSMarcel Moolenaar 6190067051fSMarcel Moolenaar if (addr == NULL || size == 0) 6200067051fSMarcel Moolenaar return; 6210067051fSMarcel Moolenaar cnt = size / sizeof(*ctor); 6220067051fSMarcel Moolenaar ctor = (void *)addr; 6230067051fSMarcel Moolenaar for (i = 0; i < cnt; i++) { 6240067051fSMarcel Moolenaar if (ctor[i] != NULL) 6250067051fSMarcel Moolenaar (*ctor[i])(); 6260067051fSMarcel Moolenaar } 6270067051fSMarcel Moolenaar } 6280067051fSMarcel Moolenaar 629de78ca7eSPeter Wemm static int 63054823af2SPeter Wemm link_elf_link_preload_finish(linker_file_t lf) 63154823af2SPeter Wemm { 63270b7ffeeSIan Dowse elf_file_t ef; 63370b7ffeeSIan Dowse int error; 63470b7ffeeSIan Dowse 63570b7ffeeSIan Dowse ef = (elf_file_t)lf; 63670b7ffeeSIan Dowse error = relocate_file(ef); 63770b7ffeeSIan Dowse if (error) 638289ead7cSKonstantin Belousov return (error); 63970b7ffeeSIan Dowse 64070b7ffeeSIan Dowse /* Notify MD code that a module is being loaded. */ 64170b7ffeeSIan Dowse error = elf_cpu_load_file(lf); 64270b7ffeeSIan Dowse if (error) 64370b7ffeeSIan Dowse return (error); 64470b7ffeeSIan Dowse 6458f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__) 646289ead7cSKonstantin Belousov /* Now ifuncs. */ 647289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, true); 648289ead7cSKonstantin Belousov if (error != 0) 649289ead7cSKonstantin Belousov return (error); 6508f941950SKonstantin Belousov #endif 651289ead7cSKonstantin Belousov 652f822c9e2SMark Johnston /* Apply protections now that relocation processing is complete. */ 653f822c9e2SMark Johnston link_elf_protect(ef); 654f822c9e2SMark Johnston 6550067051fSMarcel Moolenaar link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size); 65670b7ffeeSIan Dowse return (0); 65754823af2SPeter Wemm } 65854823af2SPeter Wemm 65954823af2SPeter Wemm static int 660c143d6c2SMarcel Moolenaar link_elf_load_file(linker_class_t cls, const char *filename, 661c143d6c2SMarcel Moolenaar linker_file_t *result) 662a4f67738SDoug Rabson { 663a344babfSGleb Smirnoff struct nameidata *nd; 664b40ce416SJulian Elischer struct thread *td = curthread; /* XXX */ 665caab6e90SPeter Wemm Elf_Ehdr *hdr; 666e9eabf59SPeter Wemm Elf_Shdr *shdr; 667a8774e39SPeter Wemm Elf_Sym *es; 668a8774e39SPeter Wemm int nbytes, i, j; 66920947801SPeter Wemm vm_offset_t mapbase; 670fe3db7c7SDoug Rabson size_t mapsize; 671a4f67738SDoug Rabson int error = 0; 672526d0bd5SKonstantin Belousov ssize_t resid; 673526d0bd5SKonstantin Belousov int flags; 674a4f67738SDoug Rabson elf_file_t ef; 675a4f67738SDoug Rabson linker_file_t lf; 676ca65d5c7SPeter Wemm int symtabindex; 677ca65d5c7SPeter Wemm int symstrindex; 678e9eabf59SPeter Wemm int shstrindex; 679e9eabf59SPeter Wemm int nsym; 68020947801SPeter Wemm int pb, rl, ra; 681e9eabf59SPeter Wemm int alignmask; 682ca65d5c7SPeter Wemm 683ca65d5c7SPeter Wemm shdr = NULL; 684ca65d5c7SPeter Wemm lf = NULL; 685e9eabf59SPeter Wemm mapsize = 0; 686e9eabf59SPeter Wemm hdr = NULL; 687a4f67738SDoug Rabson 688a344babfSGleb Smirnoff nd = malloc(sizeof(struct nameidata), M_TEMP, M_WAITOK); 689a344babfSGleb Smirnoff NDINIT(nd, LOOKUP, FOLLOW, UIO_SYSSPACE, filename, td); 690e6796b67SKirk McKusick flags = FREAD; 691a344babfSGleb Smirnoff error = vn_open(nd, &flags, 0, NULL); 692a344babfSGleb Smirnoff if (error) { 693a344babfSGleb Smirnoff free(nd, M_TEMP); 694a4f67738SDoug Rabson return error; 695a344babfSGleb Smirnoff } 696a344babfSGleb Smirnoff NDFREE(nd, NDF_ONLY_PNBUF); 697a344babfSGleb Smirnoff if (nd->ni_vp->v_type != VREG) { 6981f49b573SRuslan Ermilov error = ENOEXEC; 6991f49b573SRuslan Ermilov goto out; 7001f49b573SRuslan Ermilov } 701a3df768bSRobert Watson #ifdef MAC 702a344babfSGleb Smirnoff error = mac_kld_check_load(td->td_ucred, nd->ni_vp); 703a3df768bSRobert Watson if (error) { 704a3df768bSRobert Watson goto out; 705a3df768bSRobert Watson } 706a3df768bSRobert Watson #endif 707a4f67738SDoug Rabson 708e9eabf59SPeter Wemm /* Read the elf header from the file. */ 709e9eabf59SPeter Wemm hdr = malloc(sizeof(*hdr), M_LINKER, M_WAITOK); 710a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)hdr, sizeof(*hdr), 0, 7119ca43589SRobert Watson UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 7129ca43589SRobert Watson &resid, td); 713a4f67738SDoug Rabson if (error) 714a4f67738SDoug Rabson goto out; 715e9eabf59SPeter Wemm if (resid != 0){ 716e9eabf59SPeter Wemm error = ENOEXEC; 717e9eabf59SPeter Wemm goto out; 718e9eabf59SPeter Wemm } 719a4f67738SDoug Rabson 720caab6e90SPeter Wemm if (!IS_ELF(*hdr)) { 721fe3db7c7SDoug Rabson error = ENOEXEC; 722a4f67738SDoug Rabson goto out; 723fe3db7c7SDoug Rabson } 724fe3db7c7SDoug Rabson 725caab6e90SPeter Wemm if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS 726caab6e90SPeter Wemm || hdr->e_ident[EI_DATA] != ELF_TARG_DATA) { 727552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported file layout"); 728fe3db7c7SDoug Rabson error = ENOEXEC; 729fe3db7c7SDoug Rabson goto out; 730fe3db7c7SDoug Rabson } 731caab6e90SPeter Wemm if (hdr->e_ident[EI_VERSION] != EV_CURRENT 732caab6e90SPeter Wemm || hdr->e_version != EV_CURRENT) { 733552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported file version"); 734fe3db7c7SDoug Rabson error = ENOEXEC; 735fe3db7c7SDoug Rabson goto out; 736fe3db7c7SDoug Rabson } 737e9eabf59SPeter Wemm if (hdr->e_type != ET_REL) { 738a1d7ce03SAttilio Rao error = ENOSYS; 739fe3db7c7SDoug Rabson goto out; 740fe3db7c7SDoug Rabson } 741caab6e90SPeter Wemm if (hdr->e_machine != ELF_TARG_MACH) { 742552f9f63SEdwin Groothuis link_elf_error(filename, "Unsupported machine"); 743fe3db7c7SDoug Rabson error = ENOEXEC; 744fe3db7c7SDoug Rabson goto out; 745fe3db7c7SDoug Rabson } 746a4f67738SDoug Rabson 747326e27d8SDoug Rabson lf = linker_make_file(filename, &link_elf_class); 748326e27d8SDoug Rabson if (!lf) { 749326e27d8SDoug Rabson error = ENOMEM; 750326e27d8SDoug Rabson goto out; 751326e27d8SDoug Rabson } 752326e27d8SDoug Rabson ef = (elf_file_t) lf; 753e9eabf59SPeter Wemm ef->nprogtab = 0; 754e9eabf59SPeter Wemm ef->e_shdr = 0; 755b474c780SDavid E. O'Brien ef->nreltab = 0; 756b474c780SDavid E. O'Brien ef->nrelatab = 0; 757e9eabf59SPeter Wemm 758e9eabf59SPeter Wemm /* Allocate and read in the section header */ 759e9eabf59SPeter Wemm nbytes = hdr->e_shnum * hdr->e_shentsize; 760e9eabf59SPeter Wemm if (nbytes == 0 || hdr->e_shoff == 0 || 761e9eabf59SPeter Wemm hdr->e_shentsize != sizeof(Elf_Shdr)) { 762e9eabf59SPeter Wemm error = ENOEXEC; 763e9eabf59SPeter Wemm goto out; 764e9eabf59SPeter Wemm } 76520947801SPeter Wemm shdr = malloc(nbytes, M_LINKER, M_WAITOK); 766e9eabf59SPeter Wemm ef->e_shdr = shdr; 767a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (caddr_t)shdr, nbytes, 768a344babfSGleb Smirnoff hdr->e_shoff, UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, 769a344babfSGleb Smirnoff NOCRED, &resid, td); 770e9eabf59SPeter Wemm if (error) 771e9eabf59SPeter Wemm goto out; 772e9eabf59SPeter Wemm if (resid) { 773e9eabf59SPeter Wemm error = ENOEXEC; 774e9eabf59SPeter Wemm goto out; 775e9eabf59SPeter Wemm } 776e9eabf59SPeter Wemm 777e9eabf59SPeter Wemm /* Scan the section header for information and table sizing. */ 778e9eabf59SPeter Wemm nsym = 0; 779e9eabf59SPeter Wemm symtabindex = -1; 780e9eabf59SPeter Wemm symstrindex = -1; 781e9eabf59SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 782676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 783676799a0SAndriy Gapon continue; 784e9eabf59SPeter Wemm switch (shdr[i].sh_type) { 785e9eabf59SPeter Wemm case SHT_PROGBITS: 786e9eabf59SPeter Wemm case SHT_NOBITS: 78713f28d96SKonstantin Belousov #ifdef __amd64__ 788b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 78913f28d96SKonstantin Belousov #endif 790*5e6989baSMark Johnston case SHT_INIT_ARRAY: 791*5e6989baSMark Johnston case SHT_FINI_ARRAY: 79258c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 79358c4aee0SJohn Baldwin break; 79420947801SPeter Wemm ef->nprogtab++; 795e9eabf59SPeter Wemm break; 796e9eabf59SPeter Wemm case SHT_SYMTAB: 797e9eabf59SPeter Wemm nsym++; 798e9eabf59SPeter Wemm symtabindex = i; 799e9eabf59SPeter Wemm symstrindex = shdr[i].sh_link; 800e9eabf59SPeter Wemm break; 801e9eabf59SPeter Wemm case SHT_REL: 80215746ef4SJohn Baldwin /* 80315746ef4SJohn Baldwin * Ignore relocation tables for unallocated 80415746ef4SJohn Baldwin * sections. 80515746ef4SJohn Baldwin */ 80615746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 80715746ef4SJohn Baldwin break; 808b474c780SDavid E. O'Brien ef->nreltab++; 809e9eabf59SPeter Wemm break; 810e9eabf59SPeter Wemm case SHT_RELA: 81115746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 81215746ef4SJohn Baldwin break; 813b474c780SDavid E. O'Brien ef->nrelatab++; 814e9eabf59SPeter Wemm break; 815e9eabf59SPeter Wemm case SHT_STRTAB: 816e9eabf59SPeter Wemm break; 817e9eabf59SPeter Wemm } 818e9eabf59SPeter Wemm } 81920947801SPeter Wemm if (ef->nprogtab == 0) { 820552f9f63SEdwin Groothuis link_elf_error(filename, "file has no contents"); 821e9eabf59SPeter Wemm error = ENOEXEC; 822e9eabf59SPeter Wemm goto out; 823e9eabf59SPeter Wemm } 824e9eabf59SPeter Wemm if (nsym != 1) { 825e9eabf59SPeter Wemm /* Only allow one symbol table for now */ 826f912a970SEd Maste link_elf_error(filename, 827f912a970SEd Maste "file must have exactly one symbol table"); 828e9eabf59SPeter Wemm error = ENOEXEC; 829e9eabf59SPeter Wemm goto out; 830e9eabf59SPeter Wemm } 831e9eabf59SPeter Wemm if (symstrindex < 0 || symstrindex > hdr->e_shnum || 832e9eabf59SPeter Wemm shdr[symstrindex].sh_type != SHT_STRTAB) { 833552f9f63SEdwin Groothuis link_elf_error(filename, "file has invalid symbol strings"); 834e9eabf59SPeter Wemm error = ENOEXEC; 835e9eabf59SPeter Wemm goto out; 836e9eabf59SPeter Wemm } 837e9eabf59SPeter Wemm 838e9eabf59SPeter Wemm /* Allocate space for tracking the load chunks */ 839e9eabf59SPeter Wemm if (ef->nprogtab != 0) 84020947801SPeter Wemm ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab), 84120947801SPeter Wemm M_LINKER, M_WAITOK | M_ZERO); 842b474c780SDavid E. O'Brien if (ef->nreltab != 0) 843b474c780SDavid E. O'Brien ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab), 844b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 845b474c780SDavid E. O'Brien if (ef->nrelatab != 0) 846b474c780SDavid E. O'Brien ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab), 847b474c780SDavid E. O'Brien M_LINKER, M_WAITOK | M_ZERO); 848e9eabf59SPeter Wemm 8493cfce8e4SKonstantin Belousov if (symtabindex == -1) { 8503cfce8e4SKonstantin Belousov link_elf_error(filename, "lost symbol table index"); 8513cfce8e4SKonstantin Belousov error = ENOEXEC; 8523cfce8e4SKonstantin Belousov goto out; 8533cfce8e4SKonstantin Belousov } 85420947801SPeter Wemm /* Allocate space for and load the symbol table */ 855e9eabf59SPeter Wemm ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym); 856e9eabf59SPeter Wemm ef->ddbsymtab = malloc(shdr[symtabindex].sh_size, M_LINKER, M_WAITOK); 857a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)ef->ddbsymtab, 85820947801SPeter Wemm shdr[symtabindex].sh_size, shdr[symtabindex].sh_offset, 85920947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 86020947801SPeter Wemm &resid, td); 86120947801SPeter Wemm if (error) 86220947801SPeter Wemm goto out; 86320947801SPeter Wemm if (resid != 0){ 86420947801SPeter Wemm error = EINVAL; 86520947801SPeter Wemm goto out; 86620947801SPeter Wemm } 867e9eabf59SPeter Wemm 86820947801SPeter Wemm /* Allocate space for and load the symbol strings */ 869e9eabf59SPeter Wemm ef->ddbstrcnt = shdr[symstrindex].sh_size; 870e9eabf59SPeter Wemm ef->ddbstrtab = malloc(shdr[symstrindex].sh_size, M_LINKER, M_WAITOK); 871a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, ef->ddbstrtab, 87220947801SPeter Wemm shdr[symstrindex].sh_size, shdr[symstrindex].sh_offset, 87320947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 87420947801SPeter Wemm &resid, td); 87520947801SPeter Wemm if (error) 87620947801SPeter Wemm goto out; 87720947801SPeter Wemm if (resid != 0){ 87820947801SPeter Wemm error = EINVAL; 87920947801SPeter Wemm goto out; 88020947801SPeter Wemm } 881e9eabf59SPeter Wemm 882e9eabf59SPeter Wemm /* Do we have a string table for the section names? */ 883e9eabf59SPeter Wemm shstrindex = -1; 88420947801SPeter Wemm if (hdr->e_shstrndx != 0 && 88520947801SPeter Wemm shdr[hdr->e_shstrndx].sh_type == SHT_STRTAB) { 886e9eabf59SPeter Wemm shstrindex = hdr->e_shstrndx; 887e9eabf59SPeter Wemm ef->shstrcnt = shdr[shstrindex].sh_size; 88820947801SPeter Wemm ef->shstrtab = malloc(shdr[shstrindex].sh_size, M_LINKER, 88920947801SPeter Wemm M_WAITOK); 890a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, ef->shstrtab, 89120947801SPeter Wemm shdr[shstrindex].sh_size, shdr[shstrindex].sh_offset, 892e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 893e9eabf59SPeter Wemm &resid, td); 894e9eabf59SPeter Wemm if (error) 895e9eabf59SPeter Wemm goto out; 89620947801SPeter Wemm if (resid != 0){ 89720947801SPeter Wemm error = EINVAL; 89820947801SPeter Wemm goto out; 89920947801SPeter Wemm } 900e9eabf59SPeter Wemm } 901e9eabf59SPeter Wemm 90220947801SPeter Wemm /* Size up code/data(progbits) and bss(nobits). */ 903e9eabf59SPeter Wemm alignmask = 0; 904e9eabf59SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 905676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 906676799a0SAndriy Gapon continue; 907e9eabf59SPeter Wemm switch (shdr[i].sh_type) { 908e9eabf59SPeter Wemm case SHT_PROGBITS: 909e9eabf59SPeter Wemm case SHT_NOBITS: 91013f28d96SKonstantin Belousov #ifdef __amd64__ 911b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 91213f28d96SKonstantin Belousov #endif 913*5e6989baSMark Johnston case SHT_INIT_ARRAY: 914*5e6989baSMark Johnston case SHT_FINI_ARRAY: 91558c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 91658c4aee0SJohn Baldwin break; 917e9eabf59SPeter Wemm alignmask = shdr[i].sh_addralign - 1; 918e9eabf59SPeter Wemm mapsize += alignmask; 919e9eabf59SPeter Wemm mapsize &= ~alignmask; 920e9eabf59SPeter Wemm mapsize += shdr[i].sh_size; 921e9eabf59SPeter Wemm break; 922e9eabf59SPeter Wemm } 923e9eabf59SPeter Wemm } 924e9eabf59SPeter Wemm 925e9eabf59SPeter Wemm /* 926e9eabf59SPeter Wemm * We know how much space we need for the text/data/bss/etc. 927e9eabf59SPeter Wemm * This stuff needs to be in a single chunk so that profiling etc 928e9eabf59SPeter Wemm * can get the bounds and gdb can associate offsets with modules 929e9eabf59SPeter Wemm */ 930fbf2a778SKonstantin Belousov ef->object = vm_pager_allocate(OBJT_PHYS, NULL, round_page(mapsize), 931fbf2a778SKonstantin Belousov VM_PROT_ALL, 0, thread0.td_ucred); 932fe3db7c7SDoug Rabson if (ef->object == NULL) { 933fe3db7c7SDoug Rabson error = ENOMEM; 934fe3db7c7SDoug Rabson goto out; 935fe3db7c7SDoug Rabson } 936d80126a6SMark Johnston #if VM_NRESERVLEVEL > 0 937d80126a6SMark Johnston vm_object_color(ef->object, 0); 938d80126a6SMark Johnston #endif 939ac68d1c9SAlan Cox 940ac68d1c9SAlan Cox /* 941ac68d1c9SAlan Cox * In order to satisfy amd64's architectural requirements on the 942ac68d1c9SAlan Cox * location of code and data in the kernel's address space, request a 943ac68d1c9SAlan Cox * mapping that is above the kernel. 9441d9eae9fSMark Johnston * 9451d9eae9fSMark Johnston * Protections will be restricted once relocations are applied. 946ac68d1c9SAlan Cox */ 9474104e835SOleksandr Tymoshenko #ifdef __amd64__ 948ac68d1c9SAlan Cox mapbase = KERNBASE; 9494104e835SOleksandr Tymoshenko #else 9504104e835SOleksandr Tymoshenko mapbase = VM_MIN_KERNEL_ADDRESS; 9514104e835SOleksandr Tymoshenko #endif 95220947801SPeter Wemm error = vm_map_find(kernel_map, ef->object, 0, &mapbase, 953edb572a3SJohn Baldwin round_page(mapsize), 0, VMFS_OPTIMAL_SPACE, VM_PROT_ALL, 954edb572a3SJohn Baldwin VM_PROT_ALL, 0); 955092bacb2SMark Johnston if (error != KERN_SUCCESS) { 956fe3db7c7SDoug Rabson vm_object_deallocate(ef->object); 957092bacb2SMark Johnston ef->object = NULL; 958092bacb2SMark Johnston error = ENOMEM; 959fe3db7c7SDoug Rabson goto out; 960fe3db7c7SDoug Rabson } 96120947801SPeter Wemm 962e9eabf59SPeter Wemm /* Wire the pages */ 9637f1ef325SAlan Cox error = vm_map_wire(kernel_map, mapbase, 96420947801SPeter Wemm mapbase + round_page(mapsize), 965e9eabf59SPeter Wemm VM_MAP_WIRE_SYSTEM|VM_MAP_WIRE_NOHOLES); 9667f1ef325SAlan Cox if (error != KERN_SUCCESS) { 9677f1ef325SAlan Cox error = ENOMEM; 9687f1ef325SAlan Cox goto out; 9697f1ef325SAlan Cox } 970a4f67738SDoug Rabson 97120947801SPeter Wemm /* Inform the kld system about the situation */ 97220947801SPeter Wemm lf->address = ef->address = (caddr_t)mapbase; 97320947801SPeter Wemm lf->size = mapsize; 974e9eabf59SPeter Wemm 975e9eabf59SPeter Wemm /* 97620947801SPeter Wemm * Now load code/data(progbits), zero bss(nobits), allocate space for 97720947801SPeter Wemm * and load relocs 978e9eabf59SPeter Wemm */ 97920947801SPeter Wemm pb = 0; 98020947801SPeter Wemm rl = 0; 98120947801SPeter Wemm ra = 0; 98220947801SPeter Wemm alignmask = 0; 98320947801SPeter Wemm for (i = 0; i < hdr->e_shnum; i++) { 984676799a0SAndriy Gapon if (shdr[i].sh_size == 0) 985676799a0SAndriy Gapon continue; 98620947801SPeter Wemm switch (shdr[i].sh_type) { 98720947801SPeter Wemm case SHT_PROGBITS: 98820947801SPeter Wemm case SHT_NOBITS: 98913f28d96SKonstantin Belousov #ifdef __amd64__ 990b715d9afSKonstantin Belousov case SHT_X86_64_UNWIND: 99113f28d96SKonstantin Belousov #endif 992*5e6989baSMark Johnston case SHT_INIT_ARRAY: 993*5e6989baSMark Johnston case SHT_FINI_ARRAY: 99458c4aee0SJohn Baldwin if ((shdr[i].sh_flags & SHF_ALLOC) == 0) 99558c4aee0SJohn Baldwin break; 99620947801SPeter Wemm alignmask = shdr[i].sh_addralign - 1; 99720947801SPeter Wemm mapbase += alignmask; 99820947801SPeter Wemm mapbase &= ~alignmask; 9990067051fSMarcel Moolenaar if (ef->shstrtab != NULL && shdr[i].sh_name != 0) { 100050c202c5SJeff Roberson ef->progtab[pb].name = 100150c202c5SJeff Roberson ef->shstrtab + shdr[i].sh_name; 1002*5e6989baSMark Johnston if (!strcmp(ef->progtab[pb].name, ".ctors") || 1003*5e6989baSMark Johnston shdr[i].sh_type == SHT_INIT_ARRAY) { 1004*5e6989baSMark Johnston if (lf->ctors_addr != 0) { 1005*5e6989baSMark Johnston printf( 1006*5e6989baSMark Johnston "%s: multiple ctor sections in %s\n", 1007*5e6989baSMark Johnston __func__, filename); 1008*5e6989baSMark Johnston } else { 1009*5e6989baSMark Johnston lf->ctors_addr = 1010*5e6989baSMark Johnston (caddr_t)mapbase; 1011*5e6989baSMark Johnston lf->ctors_size = 1012*5e6989baSMark Johnston shdr[i].sh_size; 1013*5e6989baSMark Johnston } 10140067051fSMarcel Moolenaar } 10150067051fSMarcel Moolenaar } else if (shdr[i].sh_type == SHT_PROGBITS) 101620947801SPeter Wemm ef->progtab[pb].name = "<<PROGBITS>>"; 101713f28d96SKonstantin Belousov #ifdef __amd64__ 1018b715d9afSKonstantin Belousov else if (shdr[i].sh_type == SHT_X86_64_UNWIND) 101913f28d96SKonstantin Belousov ef->progtab[pb].name = "<<UNWIND>>"; 102013f28d96SKonstantin Belousov #endif 102150c202c5SJeff Roberson else 102250c202c5SJeff Roberson ef->progtab[pb].name = "<<NOBITS>>"; 102350c202c5SJeff Roberson if (ef->progtab[pb].name != NULL && 10240f823b64SBjoern A. Zeeb !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) { 102550c202c5SJeff Roberson ef->progtab[pb].addr = 102650c202c5SJeff Roberson dpcpu_alloc(shdr[i].sh_size); 10270f823b64SBjoern A. Zeeb if (ef->progtab[pb].addr == NULL) { 10280f823b64SBjoern A. Zeeb printf("%s: pcpu module space is out " 10299afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 10309afc5684SBjoern A. Zeeb "for %s\n", __func__, 10319afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 10320f823b64SBjoern A. Zeeb filename); 10330f823b64SBjoern A. Zeeb } 10340f823b64SBjoern A. Zeeb } 1035eddfbb76SRobert Watson #ifdef VIMAGE 1036eddfbb76SRobert Watson else if (ef->progtab[pb].name != NULL && 10370f823b64SBjoern A. Zeeb !strcmp(ef->progtab[pb].name, VNET_SETNAME)) { 1038eddfbb76SRobert Watson ef->progtab[pb].addr = 1039eddfbb76SRobert Watson vnet_data_alloc(shdr[i].sh_size); 10400f823b64SBjoern A. Zeeb if (ef->progtab[pb].addr == NULL) { 10410f823b64SBjoern A. Zeeb printf("%s: vnet module space is out " 10429afc5684SBjoern A. Zeeb "of space; cannot allocate %#jx " 10439afc5684SBjoern A. Zeeb "for %s\n", __func__, 10449afc5684SBjoern A. Zeeb (uintmax_t)shdr[i].sh_size, 10450f823b64SBjoern A. Zeeb filename); 10460f823b64SBjoern A. Zeeb } 10470f823b64SBjoern A. Zeeb } 1048eddfbb76SRobert Watson #endif 104950c202c5SJeff Roberson else 105050c202c5SJeff Roberson ef->progtab[pb].addr = 105150c202c5SJeff Roberson (void *)(uintptr_t)mapbase; 105250c202c5SJeff Roberson if (ef->progtab[pb].addr == NULL) { 105350c202c5SJeff Roberson error = ENOSPC; 105450c202c5SJeff Roberson goto out; 105550c202c5SJeff Roberson } 105650c202c5SJeff Roberson ef->progtab[pb].size = shdr[i].sh_size; 10571d9eae9fSMark Johnston ef->progtab[pb].flags = shdr[i].sh_flags; 105850c202c5SJeff Roberson ef->progtab[pb].sec = i; 105913f28d96SKonstantin Belousov if (shdr[i].sh_type == SHT_PROGBITS 106013f28d96SKonstantin Belousov #ifdef __amd64__ 1061b715d9afSKonstantin Belousov || shdr[i].sh_type == SHT_X86_64_UNWIND 106213f28d96SKonstantin Belousov #endif 106313f28d96SKonstantin Belousov ) { 1064a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 106520947801SPeter Wemm ef->progtab[pb].addr, 106620947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 106720947801SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, 106820947801SPeter Wemm NOCRED, &resid, td); 106920947801SPeter Wemm if (error) 107020947801SPeter Wemm goto out; 107120947801SPeter Wemm if (resid != 0){ 107220947801SPeter Wemm error = EINVAL; 107320947801SPeter Wemm goto out; 107420947801SPeter Wemm } 1075eddfbb76SRobert Watson /* Initialize the per-cpu or vnet area. */ 1076eddfbb76SRobert Watson if (ef->progtab[pb].addr != (void *)mapbase && 10775f67450dSDimitry Andric !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) 107850c202c5SJeff Roberson dpcpu_copy(ef->progtab[pb].addr, 107950c202c5SJeff Roberson shdr[i].sh_size); 1080eddfbb76SRobert Watson #ifdef VIMAGE 1081eddfbb76SRobert Watson else if (ef->progtab[pb].addr != 1082eddfbb76SRobert Watson (void *)mapbase && 108317ef1febSRobert Watson !strcmp(ef->progtab[pb].name, VNET_SETNAME)) 1084eddfbb76SRobert Watson vnet_data_copy(ef->progtab[pb].addr, 1085eddfbb76SRobert Watson shdr[i].sh_size); 1086eddfbb76SRobert Watson #endif 108750c202c5SJeff Roberson } else 108820947801SPeter Wemm bzero(ef->progtab[pb].addr, shdr[i].sh_size); 1089a8774e39SPeter Wemm 1090a8774e39SPeter Wemm /* Update all symbol values with the offset. */ 1091a8774e39SPeter Wemm for (j = 0; j < ef->ddbsymcnt; j++) { 1092a8774e39SPeter Wemm es = &ef->ddbsymtab[j]; 1093a8774e39SPeter Wemm if (es->st_shndx != i) 1094a8774e39SPeter Wemm continue; 1095a8774e39SPeter Wemm es->st_value += (Elf_Addr)ef->progtab[pb].addr; 1096a8774e39SPeter Wemm } 109720947801SPeter Wemm mapbase += shdr[i].sh_size; 109820947801SPeter Wemm pb++; 109920947801SPeter Wemm break; 110020947801SPeter Wemm case SHT_REL: 110115746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 110215746ef4SJohn Baldwin break; 110320947801SPeter Wemm ef->reltab[rl].rel = malloc(shdr[i].sh_size, M_LINKER, 110420947801SPeter Wemm M_WAITOK); 110520947801SPeter Wemm ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel); 110620947801SPeter Wemm ef->reltab[rl].sec = shdr[i].sh_info; 1107a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 110820947801SPeter Wemm (void *)ef->reltab[rl].rel, 110920947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 1110e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 1111e9eabf59SPeter Wemm &resid, td); 1112e9eabf59SPeter Wemm if (error) 1113e9eabf59SPeter Wemm goto out; 111420947801SPeter Wemm if (resid != 0){ 111520947801SPeter Wemm error = EINVAL; 111620947801SPeter Wemm goto out; 1117e9eabf59SPeter Wemm } 111820947801SPeter Wemm rl++; 111920947801SPeter Wemm break; 112020947801SPeter Wemm case SHT_RELA: 112115746ef4SJohn Baldwin if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0) 112215746ef4SJohn Baldwin break; 112320947801SPeter Wemm ef->relatab[ra].rela = malloc(shdr[i].sh_size, M_LINKER, 112420947801SPeter Wemm M_WAITOK); 112520947801SPeter Wemm ef->relatab[ra].nrela = 112620947801SPeter Wemm shdr[i].sh_size / sizeof(Elf_Rela); 112720947801SPeter Wemm ef->relatab[ra].sec = shdr[i].sh_info; 1128a344babfSGleb Smirnoff error = vn_rdwr(UIO_READ, nd->ni_vp, 112920947801SPeter Wemm (void *)ef->relatab[ra].rela, 113020947801SPeter Wemm shdr[i].sh_size, shdr[i].sh_offset, 1131e9eabf59SPeter Wemm UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, 1132e9eabf59SPeter Wemm &resid, td); 1133e9eabf59SPeter Wemm if (error) 1134e9eabf59SPeter Wemm goto out; 113520947801SPeter Wemm if (resid != 0){ 113620947801SPeter Wemm error = EINVAL; 113720947801SPeter Wemm goto out; 1138e9eabf59SPeter Wemm } 113920947801SPeter Wemm ra++; 114020947801SPeter Wemm break; 114120947801SPeter Wemm } 114220947801SPeter Wemm } 11433cfce8e4SKonstantin Belousov if (pb != ef->nprogtab) { 11443cfce8e4SKonstantin Belousov link_elf_error(filename, "lost progbits"); 11453cfce8e4SKonstantin Belousov error = ENOEXEC; 11463cfce8e4SKonstantin Belousov goto out; 11473cfce8e4SKonstantin Belousov } 11483cfce8e4SKonstantin Belousov if (rl != ef->nreltab) { 11493cfce8e4SKonstantin Belousov link_elf_error(filename, "lost reltab"); 11503cfce8e4SKonstantin Belousov error = ENOEXEC; 11513cfce8e4SKonstantin Belousov goto out; 11523cfce8e4SKonstantin Belousov } 11533cfce8e4SKonstantin Belousov if (ra != ef->nrelatab) { 11543cfce8e4SKonstantin Belousov link_elf_error(filename, "lost relatab"); 11553cfce8e4SKonstantin Belousov error = ENOEXEC; 11563cfce8e4SKonstantin Belousov goto out; 11573cfce8e4SKonstantin Belousov } 11583cfce8e4SKonstantin Belousov if (mapbase != (vm_offset_t)ef->address + mapsize) { 11593cfce8e4SKonstantin Belousov printf( 11603cfce8e4SKonstantin Belousov "%s: mapbase 0x%lx != address %p + mapsize 0x%lx (0x%lx)\n", 11613cfce8e4SKonstantin Belousov filename != NULL ? filename : "<none>", 11626f3c6327SNeel Natu (u_long)mapbase, ef->address, (u_long)mapsize, 11636f3c6327SNeel Natu (u_long)(vm_offset_t)ef->address + mapsize); 11643cfce8e4SKonstantin Belousov error = ENOMEM; 11653cfce8e4SKonstantin Belousov goto out; 11663cfce8e4SKonstantin Belousov } 1167e9eabf59SPeter Wemm 1168e9eabf59SPeter Wemm /* Local intra-module relocations */ 1169289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, false); 11703cfce8e4SKonstantin Belousov if (error != 0) 11713cfce8e4SKonstantin Belousov goto out; 11727251b4bfSJake Burkholder 1173e9eabf59SPeter Wemm /* Pull in dependencies */ 1174b249ce48SMateusz Guzik VOP_UNLOCK(nd->ni_vp); 11757b9716baSIan Dowse error = linker_load_dependencies(lf); 1176a344babfSGleb Smirnoff vn_lock(nd->ni_vp, LK_EXCLUSIVE | LK_RETRY); 1177ca65d5c7SPeter Wemm if (error) 1178de78ca7eSPeter Wemm goto out; 1179e9eabf59SPeter Wemm 1180e9eabf59SPeter Wemm /* External relocations */ 1181326e27d8SDoug Rabson error = relocate_file(ef); 1182ca65d5c7SPeter Wemm if (error) 1183ca65d5c7SPeter Wemm goto out; 1184ca65d5c7SPeter Wemm 1185e9eabf59SPeter Wemm /* Notify MD code that a module is being loaded. */ 1186e9eabf59SPeter Wemm error = elf_cpu_load_file(lf); 1187ca65d5c7SPeter Wemm if (error) 1188ca65d5c7SPeter Wemm goto out; 1189a4f67738SDoug Rabson 11908f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__) 1191289ead7cSKonstantin Belousov /* Now ifuncs. */ 1192289ead7cSKonstantin Belousov error = link_elf_reloc_local(lf, true); 1193289ead7cSKonstantin Belousov if (error != 0) 1194289ead7cSKonstantin Belousov goto out; 11958f941950SKonstantin Belousov #endif 1196289ead7cSKonstantin Belousov 11971d9eae9fSMark Johnston link_elf_protect(ef); 11980067051fSMarcel Moolenaar link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size); 1199a4f67738SDoug Rabson *result = lf; 1200a4f67738SDoug Rabson 1201a4f67738SDoug Rabson out: 1202b249ce48SMateusz Guzik VOP_UNLOCK(nd->ni_vp); 1203a344babfSGleb Smirnoff vn_close(nd->ni_vp, FREAD, td->td_ucred, td); 1204a344babfSGleb Smirnoff free(nd, M_TEMP); 1205ca65d5c7SPeter Wemm if (error && lf) 120665a311fcSPoul-Henning Kamp linker_file_unload(lf, LINKER_UNLOAD_FORCE); 1207e9eabf59SPeter Wemm free(hdr, M_LINKER); 1208a4f67738SDoug Rabson 1209a4f67738SDoug Rabson return error; 1210a4f67738SDoug Rabson } 1211a4f67738SDoug Rabson 1212a4f67738SDoug Rabson static void 1213de78ca7eSPeter Wemm link_elf_unload_file(linker_file_t file) 1214a4f67738SDoug Rabson { 1215326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) file; 1216d821d364SPedro F. Giffuni u_int i; 1217a4f67738SDoug Rabson 12181aeb23cdSMarcel Moolenaar /* Notify MD code that a module is being unloaded. */ 12191aeb23cdSMarcel Moolenaar elf_cpu_unload_file(file); 12201aeb23cdSMarcel Moolenaar 122150c202c5SJeff Roberson if (ef->progtab) { 122250c202c5SJeff Roberson for (i = 0; i < ef->nprogtab; i++) { 122350c202c5SJeff Roberson if (ef->progtab[i].size == 0) 122450c202c5SJeff Roberson continue; 122550c202c5SJeff Roberson if (ef->progtab[i].name == NULL) 122650c202c5SJeff Roberson continue; 12275f67450dSDimitry Andric if (!strcmp(ef->progtab[i].name, DPCPU_SETNAME)) 122850c202c5SJeff Roberson dpcpu_free(ef->progtab[i].addr, 122950c202c5SJeff Roberson ef->progtab[i].size); 1230eddfbb76SRobert Watson #ifdef VIMAGE 123117ef1febSRobert Watson else if (!strcmp(ef->progtab[i].name, VNET_SETNAME)) 1232eddfbb76SRobert Watson vnet_data_free(ef->progtab[i].addr, 1233eddfbb76SRobert Watson ef->progtab[i].size); 1234eddfbb76SRobert Watson #endif 123550c202c5SJeff Roberson } 123650c202c5SJeff Roberson } 123770b7ffeeSIan Dowse if (ef->preloaded) { 123870b7ffeeSIan Dowse free(ef->reltab, M_LINKER); 123970b7ffeeSIan Dowse free(ef->relatab, M_LINKER); 124070b7ffeeSIan Dowse free(ef->progtab, M_LINKER); 1241a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 1242a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 1243a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 124473624a80SMark Johnston if (file->pathname != NULL) 124573624a80SMark Johnston preload_delete_name(file->pathname); 124670b7ffeeSIan Dowse return; 124770b7ffeeSIan Dowse } 124870b7ffeeSIan Dowse 1249b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) 125020947801SPeter Wemm free(ef->reltab[i].rel, M_LINKER); 1251b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) 125220947801SPeter Wemm free(ef->relatab[i].rela, M_LINKER); 125320947801SPeter Wemm free(ef->reltab, M_LINKER); 125420947801SPeter Wemm free(ef->relatab, M_LINKER); 125520947801SPeter Wemm free(ef->progtab, M_LINKER); 125620947801SPeter Wemm 1257092bacb2SMark Johnston if (ef->object != NULL) 1258fe3db7c7SDoug Rabson vm_map_remove(kernel_map, (vm_offset_t)ef->address, 1259092bacb2SMark Johnston (vm_offset_t)ef->address + ptoa(ef->object->size)); 1260e9eabf59SPeter Wemm free(ef->e_shdr, M_LINKER); 1261e9eabf59SPeter Wemm free(ef->ddbsymtab, M_LINKER); 1262e9eabf59SPeter Wemm free(ef->ddbstrtab, M_LINKER); 126320947801SPeter Wemm free(ef->shstrtab, M_LINKER); 1264a2024a3eSJohn Birrell free(ef->ctftab, M_LINKER); 1265a2024a3eSJohn Birrell free(ef->ctfoff, M_LINKER); 1266a2024a3eSJohn Birrell free(ef->typoff, M_LINKER); 1267de78ca7eSPeter Wemm } 1268de78ca7eSPeter Wemm 1269fe3db7c7SDoug Rabson static const char * 1270757686b1SMarcel Moolenaar symbol_name(elf_file_t ef, Elf_Size r_info) 1271a4f67738SDoug Rabson { 1272fe3db7c7SDoug Rabson const Elf_Sym *ref; 1273a4f67738SDoug Rabson 1274aa855a59SPeter Wemm if (ELF_R_SYM(r_info)) { 1275e9eabf59SPeter Wemm ref = ef->ddbsymtab + ELF_R_SYM(r_info); 1276e9eabf59SPeter Wemm return ef->ddbstrtab + ref->st_name; 1277fe3db7c7SDoug Rabson } else 1278fe3db7c7SDoug Rabson return NULL; 1279a4f67738SDoug Rabson } 1280a4f67738SDoug Rabson 1281e9eabf59SPeter Wemm static Elf_Addr 1282e9eabf59SPeter Wemm findbase(elf_file_t ef, int sec) 1283e9eabf59SPeter Wemm { 1284e9eabf59SPeter Wemm int i; 1285e9eabf59SPeter Wemm Elf_Addr base = 0; 1286e9eabf59SPeter Wemm 1287e9eabf59SPeter Wemm for (i = 0; i < ef->nprogtab; i++) { 1288a8774e39SPeter Wemm if (sec == ef->progtab[i].sec) { 1289e9eabf59SPeter Wemm base = (Elf_Addr)ef->progtab[i].addr; 1290a8774e39SPeter Wemm break; 1291e9eabf59SPeter Wemm } 1292a8774e39SPeter Wemm } 1293e9eabf59SPeter Wemm return base; 1294e9eabf59SPeter Wemm } 1295e9eabf59SPeter Wemm 1296a4f67738SDoug Rabson static int 1297326e27d8SDoug Rabson relocate_file(elf_file_t ef) 1298a4f67738SDoug Rabson { 1299fe3db7c7SDoug Rabson const Elf_Rel *rellim; 1300fe3db7c7SDoug Rabson const Elf_Rel *rel; 1301fe3db7c7SDoug Rabson const Elf_Rela *relalim; 1302fe3db7c7SDoug Rabson const Elf_Rela *rela; 1303aa855a59SPeter Wemm const char *symname; 1304e9eabf59SPeter Wemm const Elf_Sym *sym; 1305e9eabf59SPeter Wemm int i; 1306757686b1SMarcel Moolenaar Elf_Size symidx; 1307e9eabf59SPeter Wemm Elf_Addr base; 1308e9eabf59SPeter Wemm 1309fe3db7c7SDoug Rabson /* Perform relocations without addend if there are any: */ 1310b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) { 1311e9eabf59SPeter Wemm rel = ef->reltab[i].rel; 13123cfce8e4SKonstantin Belousov if (rel == NULL) { 13133cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a reltab!"); 13143cfce8e4SKonstantin Belousov return (ENOEXEC); 13153cfce8e4SKonstantin Belousov } 131620947801SPeter Wemm rellim = rel + ef->reltab[i].nrel; 1317e9eabf59SPeter Wemm base = findbase(ef, ef->reltab[i].sec); 13183cfce8e4SKonstantin Belousov if (base == 0) { 13193cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 13203cfce8e4SKonstantin Belousov return (ENOEXEC); 13213cfce8e4SKonstantin Belousov } 132220947801SPeter Wemm for ( ; rel < rellim; rel++) { 1323e9eabf59SPeter Wemm symidx = ELF_R_SYM(rel->r_info); 132420947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 132520947801SPeter Wemm continue; 1326e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 132720947801SPeter Wemm /* Local relocs are already done */ 132820947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) 132920947801SPeter Wemm continue; 133020947801SPeter Wemm if (elf_reloc(&ef->lf, base, rel, ELF_RELOC_REL, 133120947801SPeter Wemm elf_obj_lookup)) { 1332aa855a59SPeter Wemm symname = symbol_name(ef, rel->r_info); 133320947801SPeter Wemm printf("link_elf_obj: symbol %s undefined\n", 133420947801SPeter Wemm symname); 13353cfce8e4SKonstantin Belousov return (ENOENT); 1336a13ddfb6SPeter Wemm } 1337aa855a59SPeter Wemm } 1338a4f67738SDoug Rabson } 1339a4f67738SDoug Rabson 1340fe3db7c7SDoug Rabson /* Perform relocations with addend if there are any: */ 1341b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) { 1342e9eabf59SPeter Wemm rela = ef->relatab[i].rela; 13433cfce8e4SKonstantin Belousov if (rela == NULL) { 13443cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a relatab!"); 13453cfce8e4SKonstantin Belousov return (ENOEXEC); 13463cfce8e4SKonstantin Belousov } 134720947801SPeter Wemm relalim = rela + ef->relatab[i].nrela; 1348e9eabf59SPeter Wemm base = findbase(ef, ef->relatab[i].sec); 13493cfce8e4SKonstantin Belousov if (base == 0) { 13503cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, 13513cfce8e4SKonstantin Belousov "lost base for relatab"); 13523cfce8e4SKonstantin Belousov return (ENOEXEC); 13533cfce8e4SKonstantin Belousov } 135420947801SPeter Wemm for ( ; rela < relalim; rela++) { 1355e9eabf59SPeter Wemm symidx = ELF_R_SYM(rela->r_info); 135620947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 135720947801SPeter Wemm continue; 1358e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 135920947801SPeter Wemm /* Local relocs are already done */ 136020947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) 136120947801SPeter Wemm continue; 136220947801SPeter Wemm if (elf_reloc(&ef->lf, base, rela, ELF_RELOC_RELA, 136320947801SPeter Wemm elf_obj_lookup)) { 1364aa855a59SPeter Wemm symname = symbol_name(ef, rela->r_info); 136520947801SPeter Wemm printf("link_elf_obj: symbol %s undefined\n", 136620947801SPeter Wemm symname); 13673cfce8e4SKonstantin Belousov return (ENOENT); 1368a13ddfb6SPeter Wemm } 1369e9eabf59SPeter Wemm } 1370e9eabf59SPeter Wemm } 1371a4f67738SDoug Rabson 13722832cd54SKonstantin Belousov /* 1373e3043798SPedro F. Giffuni * Only clean SHN_FBSD_CACHED for successful return. If we 13742832cd54SKonstantin Belousov * modified symbol table for the object but found an 13752832cd54SKonstantin Belousov * unresolved symbol, there is no reason to roll back. 13762832cd54SKonstantin Belousov */ 13772832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(ef); 13782832cd54SKonstantin Belousov 13793cfce8e4SKonstantin Belousov return (0); 1380a4f67738SDoug Rabson } 1381a4f67738SDoug Rabson 138237c84183SPoul-Henning Kamp static int 1383d254af07SMatthew Dillon link_elf_lookup_symbol(linker_file_t lf, const char *name, c_linker_sym_t *sym) 1384a4f67738SDoug Rabson { 1385326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 13862d636ab0SPeter Wemm const Elf_Sym *symp; 13872d636ab0SPeter Wemm const char *strp; 1388a4f67738SDoug Rabson int i; 1389a4f67738SDoug Rabson 13902d636ab0SPeter Wemm for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 13912d636ab0SPeter Wemm strp = ef->ddbstrtab + symp->st_name; 1392a8774e39SPeter Wemm if (symp->st_shndx != SHN_UNDEF && strcmp(name, strp) == 0) { 1393d254af07SMatthew Dillon *sym = (c_linker_sym_t) symp; 13942d636ab0SPeter Wemm return 0; 13952d636ab0SPeter Wemm } 13962d636ab0SPeter Wemm } 1397a4f67738SDoug Rabson return ENOENT; 1398a4f67738SDoug Rabson } 1399a4f67738SDoug Rabson 1400de78ca7eSPeter Wemm static int 140120947801SPeter Wemm link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym, 140220947801SPeter Wemm linker_symval_t *symval) 1403a4f67738SDoug Rabson { 14047035cf14SKonstantin Belousov elf_file_t ef; 14057035cf14SKonstantin Belousov const Elf_Sym *es; 14067035cf14SKonstantin Belousov caddr_t val; 1407a4f67738SDoug Rabson 14087035cf14SKonstantin Belousov ef = (elf_file_t) lf; 14097035cf14SKonstantin Belousov es = (const Elf_Sym*) sym; 14107035cf14SKonstantin Belousov val = (caddr_t)es->st_value; 14115cf87418SMarcel Moolenaar if (es >= ef->ddbsymtab && es < (ef->ddbsymtab + ef->ddbsymcnt)) { 14122d636ab0SPeter Wemm symval->name = ef->ddbstrtab + es->st_name; 14137035cf14SKonstantin Belousov val = (caddr_t)es->st_value; 14147035cf14SKonstantin Belousov if (ELF_ST_TYPE(es->st_info) == STT_GNU_IFUNC) 14157035cf14SKonstantin Belousov val = ((caddr_t (*)(void))val)(); 14167035cf14SKonstantin Belousov symval->value = val; 14172d636ab0SPeter Wemm symval->size = es->st_size; 14182d636ab0SPeter Wemm return 0; 14192d636ab0SPeter Wemm } 14202d636ab0SPeter Wemm return ENOENT; 14212d636ab0SPeter Wemm } 1422a4f67738SDoug Rabson 1423a4f67738SDoug Rabson static int 1424a4f67738SDoug Rabson link_elf_search_symbol(linker_file_t lf, caddr_t value, 1425d254af07SMatthew Dillon c_linker_sym_t *sym, long *diffp) 1426a4f67738SDoug Rabson { 1427326e27d8SDoug Rabson elf_file_t ef = (elf_file_t) lf; 1428586453feSBruce Evans u_long off = (uintptr_t) (void *) value; 1429a4f67738SDoug Rabson u_long diff = off; 1430586453feSBruce Evans u_long st_value; 1431fe3db7c7SDoug Rabson const Elf_Sym *es; 1432b85f65afSPedro F. Giffuni const Elf_Sym *best = NULL; 1433a4f67738SDoug Rabson int i; 1434a4f67738SDoug Rabson 14352d636ab0SPeter Wemm for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) { 1436a4f67738SDoug Rabson if (es->st_name == 0) 1437a4f67738SDoug Rabson continue; 1438a8774e39SPeter Wemm st_value = es->st_value; 1439b5abfb70SPeter Wemm if (off >= st_value) { 1440b5abfb70SPeter Wemm if (off - st_value < diff) { 1441b5abfb70SPeter Wemm diff = off - st_value; 1442a4f67738SDoug Rabson best = es; 1443a4f67738SDoug Rabson if (diff == 0) 1444a4f67738SDoug Rabson break; 1445b5abfb70SPeter Wemm } else if (off - st_value == diff) { 1446a4f67738SDoug Rabson best = es; 1447a4f67738SDoug Rabson } 1448a4f67738SDoug Rabson } 1449a4f67738SDoug Rabson } 1450b85f65afSPedro F. Giffuni if (best == NULL) 1451a4f67738SDoug Rabson *diffp = off; 1452a4f67738SDoug Rabson else 1453a4f67738SDoug Rabson *diffp = diff; 1454d254af07SMatthew Dillon *sym = (c_linker_sym_t) best; 1455a4f67738SDoug Rabson 1456a4f67738SDoug Rabson return 0; 1457a4f67738SDoug Rabson } 1458f41325dbSPeter Wemm 1459f41325dbSPeter Wemm /* 1460f41325dbSPeter Wemm * Look up a linker set on an ELF system. 1461f41325dbSPeter Wemm */ 1462f41325dbSPeter Wemm static int 1463f41325dbSPeter Wemm link_elf_lookup_set(linker_file_t lf, const char *name, 1464f41325dbSPeter Wemm void ***startp, void ***stopp, int *countp) 1465f41325dbSPeter Wemm { 1466e9eabf59SPeter Wemm elf_file_t ef = (elf_file_t)lf; 1467f41325dbSPeter Wemm void **start, **stop; 1468e9eabf59SPeter Wemm int i, count; 1469f41325dbSPeter Wemm 1470e9eabf59SPeter Wemm /* Relative to section number */ 1471e9eabf59SPeter Wemm for (i = 0; i < ef->nprogtab; i++) { 1472e9eabf59SPeter Wemm if ((strncmp(ef->progtab[i].name, "set_", 4) == 0) && 1473e9eabf59SPeter Wemm strcmp(ef->progtab[i].name + 4, name) == 0) { 1474e9eabf59SPeter Wemm start = (void **)ef->progtab[i].addr; 147520947801SPeter Wemm stop = (void **)((char *)ef->progtab[i].addr + 147620947801SPeter Wemm ef->progtab[i].size); 1477f41325dbSPeter Wemm count = stop - start; 1478f41325dbSPeter Wemm if (startp) 1479f41325dbSPeter Wemm *startp = start; 1480f41325dbSPeter Wemm if (stopp) 1481f41325dbSPeter Wemm *stopp = stop; 1482f41325dbSPeter Wemm if (countp) 1483f41325dbSPeter Wemm *countp = count; 1484e9eabf59SPeter Wemm return (0); 1485e9eabf59SPeter Wemm } 1486e9eabf59SPeter Wemm } 1487e9eabf59SPeter Wemm return (ESRCH); 1488f41325dbSPeter Wemm } 1489bb9fe9ddSBrian Feldman 1490bb9fe9ddSBrian Feldman static int 1491bb9fe9ddSBrian Feldman link_elf_each_function_name(linker_file_t file, 1492e9eabf59SPeter Wemm int (*callback)(const char *, void *), void *opaque) 1493e9eabf59SPeter Wemm { 1494bb9fe9ddSBrian Feldman elf_file_t ef = (elf_file_t)file; 1495bb9fe9ddSBrian Feldman const Elf_Sym *symp; 1496bb9fe9ddSBrian Feldman int i, error; 1497bb9fe9ddSBrian Feldman 1498bb9fe9ddSBrian Feldman /* Exhaustive search */ 1499bb9fe9ddSBrian Feldman for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1500bb9fe9ddSBrian Feldman if (symp->st_value != 0 && 15017035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 15027035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) { 1503bb9fe9ddSBrian Feldman error = callback(ef->ddbstrtab + symp->st_name, opaque); 1504bb9fe9ddSBrian Feldman if (error) 1505bb9fe9ddSBrian Feldman return (error); 1506bb9fe9ddSBrian Feldman } 1507bb9fe9ddSBrian Feldman } 1508bb9fe9ddSBrian Feldman return (0); 1509bb9fe9ddSBrian Feldman } 151084201059SMarcel Moolenaar 1511a2024a3eSJohn Birrell static int 1512a2024a3eSJohn Birrell link_elf_each_function_nameval(linker_file_t file, 1513a2024a3eSJohn Birrell linker_function_nameval_callback_t callback, void *opaque) 1514a2024a3eSJohn Birrell { 1515a2024a3eSJohn Birrell linker_symval_t symval; 1516a2024a3eSJohn Birrell elf_file_t ef = (elf_file_t)file; 1517a2024a3eSJohn Birrell const Elf_Sym* symp; 1518a2024a3eSJohn Birrell int i, error; 1519a2024a3eSJohn Birrell 1520a2024a3eSJohn Birrell /* Exhaustive search */ 1521a2024a3eSJohn Birrell for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) { 1522a2024a3eSJohn Birrell if (symp->st_value != 0 && 15237035cf14SKonstantin Belousov (ELF_ST_TYPE(symp->st_info) == STT_FUNC || 15247035cf14SKonstantin Belousov ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) { 15257035cf14SKonstantin Belousov error = link_elf_symbol_values(file, 15267035cf14SKonstantin Belousov (c_linker_sym_t)symp, &symval); 1527a2024a3eSJohn Birrell if (error) 1528a2024a3eSJohn Birrell return (error); 1529a2024a3eSJohn Birrell error = callback(file, i, &symval, opaque); 1530a2024a3eSJohn Birrell if (error) 1531a2024a3eSJohn Birrell return (error); 1532a2024a3eSJohn Birrell } 1533a2024a3eSJohn Birrell } 1534a2024a3eSJohn Birrell return (0); 1535a2024a3eSJohn Birrell } 1536a2024a3eSJohn Birrell 15372832cd54SKonstantin Belousov static void 15382832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(elf_file_t ef) 15392832cd54SKonstantin Belousov { 15402832cd54SKonstantin Belousov Elf_Sym *sym; 15412832cd54SKonstantin Belousov Elf_Size i; 15422832cd54SKonstantin Belousov 15432832cd54SKonstantin Belousov for (i = 0; i < ef->ddbsymcnt; i++) { 15442832cd54SKonstantin Belousov sym = ef->ddbsymtab + i; 15452832cd54SKonstantin Belousov if (sym->st_shndx == SHN_FBSD_CACHED) { 15462832cd54SKonstantin Belousov sym->st_shndx = SHN_UNDEF; 15472832cd54SKonstantin Belousov sym->st_value = 0; 15482832cd54SKonstantin Belousov } 15492832cd54SKonstantin Belousov } 15502832cd54SKonstantin Belousov } 15512832cd54SKonstantin Belousov 1552d297ad16SMarcel Moolenaar /* 1553d297ad16SMarcel Moolenaar * Symbol lookup function that can be used when the symbol index is known (ie 1554d297ad16SMarcel Moolenaar * in relocations). It uses the symbol index instead of doing a fully fledged 1555d297ad16SMarcel Moolenaar * hash table based lookup when such is valid. For example for local symbols. 1556d297ad16SMarcel Moolenaar * This is not only more efficient, it's also more correct. It's not always 1557d297ad16SMarcel Moolenaar * the case that the symbol can be found through the hash table. 1558d297ad16SMarcel Moolenaar */ 1559cff8c6f2SKonstantin Belousov static int 1560cff8c6f2SKonstantin Belousov elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, Elf_Addr *res) 1561d297ad16SMarcel Moolenaar { 1562d297ad16SMarcel Moolenaar elf_file_t ef = (elf_file_t)lf; 15632832cd54SKonstantin Belousov Elf_Sym *sym; 1564d297ad16SMarcel Moolenaar const char *symbol; 1565cff8c6f2SKonstantin Belousov Elf_Addr res1; 1566d297ad16SMarcel Moolenaar 1567d297ad16SMarcel Moolenaar /* Don't even try to lookup the symbol if the index is bogus. */ 1568cff8c6f2SKonstantin Belousov if (symidx >= ef->ddbsymcnt) { 1569cff8c6f2SKonstantin Belousov *res = 0; 1570cff8c6f2SKonstantin Belousov return (EINVAL); 1571cff8c6f2SKonstantin Belousov } 1572d297ad16SMarcel Moolenaar 1573e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 1574d297ad16SMarcel Moolenaar 15754cec6f5dSPeter Wemm /* Quick answer if there is a definition included. */ 1576cff8c6f2SKonstantin Belousov if (sym->st_shndx != SHN_UNDEF) { 1577289ead7cSKonstantin Belousov res1 = (Elf_Addr)sym->st_value; 1578289ead7cSKonstantin Belousov if (ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC) 1579289ead7cSKonstantin Belousov res1 = ((Elf_Addr (*)(void))res1)(); 1580289ead7cSKonstantin Belousov *res = res1; 1581cff8c6f2SKonstantin Belousov return (0); 1582cff8c6f2SKonstantin Belousov } 15834cec6f5dSPeter Wemm 15844cec6f5dSPeter Wemm /* If we get here, then it is undefined and needs a lookup. */ 15854cec6f5dSPeter Wemm switch (ELF_ST_BIND(sym->st_info)) { 15864cec6f5dSPeter Wemm case STB_LOCAL: 15874cec6f5dSPeter Wemm /* Local, but undefined? huh? */ 1588cff8c6f2SKonstantin Belousov *res = 0; 1589cff8c6f2SKonstantin Belousov return (EINVAL); 1590d297ad16SMarcel Moolenaar 1591e9eabf59SPeter Wemm case STB_GLOBAL: 1592cff8c6f2SKonstantin Belousov case STB_WEAK: 1593e9eabf59SPeter Wemm /* Relative to Data or Function name */ 1594e9eabf59SPeter Wemm symbol = ef->ddbstrtab + sym->st_name; 1595d297ad16SMarcel Moolenaar 1596d297ad16SMarcel Moolenaar /* Force a lookup failure if the symbol name is bogus. */ 1597cff8c6f2SKonstantin Belousov if (*symbol == 0) { 1598cff8c6f2SKonstantin Belousov *res = 0; 1599cff8c6f2SKonstantin Belousov return (EINVAL); 1600cff8c6f2SKonstantin Belousov } 1601cff8c6f2SKonstantin Belousov res1 = (Elf_Addr)linker_file_lookup_symbol(lf, symbol, deps); 16022832cd54SKonstantin Belousov 16032832cd54SKonstantin Belousov /* 16042832cd54SKonstantin Belousov * Cache global lookups during module relocation. The failure 16052832cd54SKonstantin Belousov * case is particularly expensive for callers, who must scan 16062832cd54SKonstantin Belousov * through the entire globals table doing strcmp(). Cache to 16072832cd54SKonstantin Belousov * avoid doing such work repeatedly. 16082832cd54SKonstantin Belousov * 16092832cd54SKonstantin Belousov * After relocation is complete, undefined globals will be 16102832cd54SKonstantin Belousov * restored to SHN_UNDEF in elf_obj_cleanup_globals_cache(), 16112832cd54SKonstantin Belousov * above. 16122832cd54SKonstantin Belousov */ 1613cff8c6f2SKonstantin Belousov if (res1 != 0) { 16142832cd54SKonstantin Belousov sym->st_shndx = SHN_FBSD_CACHED; 1615cff8c6f2SKonstantin Belousov sym->st_value = res1; 1616cff8c6f2SKonstantin Belousov *res = res1; 1617e9eabf59SPeter Wemm return (0); 1618cff8c6f2SKonstantin Belousov } else if (ELF_ST_BIND(sym->st_info) == STB_WEAK) { 1619cff8c6f2SKonstantin Belousov sym->st_value = 0; 1620cff8c6f2SKonstantin Belousov *res = 0; 1621cff8c6f2SKonstantin Belousov return (0); 1622cff8c6f2SKonstantin Belousov } 1623cff8c6f2SKonstantin Belousov return (EINVAL); 1624e9eabf59SPeter Wemm 1625e9eabf59SPeter Wemm default: 1626cff8c6f2SKonstantin Belousov return (EINVAL); 1627d297ad16SMarcel Moolenaar } 1628e9eabf59SPeter Wemm } 1629e9eabf59SPeter Wemm 16307251b4bfSJake Burkholder static void 16317226306eSKonstantin Belousov link_elf_fix_link_set(elf_file_t ef) 16327226306eSKonstantin Belousov { 16337226306eSKonstantin Belousov static const char startn[] = "__start_"; 16347226306eSKonstantin Belousov static const char stopn[] = "__stop_"; 16357226306eSKonstantin Belousov Elf_Sym *sym; 16367226306eSKonstantin Belousov const char *sym_name, *linkset_name; 16377226306eSKonstantin Belousov Elf_Addr startp, stopp; 16387226306eSKonstantin Belousov Elf_Size symidx; 16397226306eSKonstantin Belousov int start, i; 16407226306eSKonstantin Belousov 16417226306eSKonstantin Belousov startp = stopp = 0; 16427226306eSKonstantin Belousov for (symidx = 1 /* zero entry is special */; 16437226306eSKonstantin Belousov symidx < ef->ddbsymcnt; symidx++) { 16447226306eSKonstantin Belousov sym = ef->ddbsymtab + symidx; 16457226306eSKonstantin Belousov if (sym->st_shndx != SHN_UNDEF) 16467226306eSKonstantin Belousov continue; 16477226306eSKonstantin Belousov 16487226306eSKonstantin Belousov sym_name = ef->ddbstrtab + sym->st_name; 16497226306eSKonstantin Belousov if (strncmp(sym_name, startn, sizeof(startn) - 1) == 0) { 16507226306eSKonstantin Belousov start = 1; 16517226306eSKonstantin Belousov linkset_name = sym_name + sizeof(startn) - 1; 16527226306eSKonstantin Belousov } 16537226306eSKonstantin Belousov else if (strncmp(sym_name, stopn, sizeof(stopn) - 1) == 0) { 16547226306eSKonstantin Belousov start = 0; 16557226306eSKonstantin Belousov linkset_name = sym_name + sizeof(stopn) - 1; 16567226306eSKonstantin Belousov } 16577226306eSKonstantin Belousov else 16587226306eSKonstantin Belousov continue; 16597226306eSKonstantin Belousov 16607226306eSKonstantin Belousov for (i = 0; i < ef->nprogtab; i++) { 16617226306eSKonstantin Belousov if (strcmp(ef->progtab[i].name, linkset_name) == 0) { 16627226306eSKonstantin Belousov startp = (Elf_Addr)ef->progtab[i].addr; 16637226306eSKonstantin Belousov stopp = (Elf_Addr)(startp + ef->progtab[i].size); 16647226306eSKonstantin Belousov break; 16657226306eSKonstantin Belousov } 16667226306eSKonstantin Belousov } 16677226306eSKonstantin Belousov if (i == ef->nprogtab) 16687226306eSKonstantin Belousov continue; 16697226306eSKonstantin Belousov 16707226306eSKonstantin Belousov sym->st_value = start ? startp : stopp; 16717226306eSKonstantin Belousov sym->st_shndx = i; 16727226306eSKonstantin Belousov } 16737226306eSKonstantin Belousov } 16747226306eSKonstantin Belousov 16753cfce8e4SKonstantin Belousov static int 1676289ead7cSKonstantin Belousov link_elf_reloc_local(linker_file_t lf, bool ifuncs) 16777251b4bfSJake Burkholder { 1678e9eabf59SPeter Wemm elf_file_t ef = (elf_file_t)lf; 16797251b4bfSJake Burkholder const Elf_Rel *rellim; 16807251b4bfSJake Burkholder const Elf_Rel *rel; 16817251b4bfSJake Burkholder const Elf_Rela *relalim; 16827251b4bfSJake Burkholder const Elf_Rela *rela; 1683e9eabf59SPeter Wemm const Elf_Sym *sym; 1684e9eabf59SPeter Wemm Elf_Addr base; 1685e9eabf59SPeter Wemm int i; 1686757686b1SMarcel Moolenaar Elf_Size symidx; 1687e9eabf59SPeter Wemm 16887226306eSKonstantin Belousov link_elf_fix_link_set(ef); 16897226306eSKonstantin Belousov 16907251b4bfSJake Burkholder /* Perform relocations without addend if there are any: */ 1691b474c780SDavid E. O'Brien for (i = 0; i < ef->nreltab; i++) { 1692e9eabf59SPeter Wemm rel = ef->reltab[i].rel; 16933cfce8e4SKonstantin Belousov if (rel == NULL) { 16943cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a reltab"); 16953cfce8e4SKonstantin Belousov return (ENOEXEC); 16963cfce8e4SKonstantin Belousov } 169720947801SPeter Wemm rellim = rel + ef->reltab[i].nrel; 1698e9eabf59SPeter Wemm base = findbase(ef, ef->reltab[i].sec); 16993cfce8e4SKonstantin Belousov if (base == 0) { 17003cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 17013cfce8e4SKonstantin Belousov return (ENOEXEC); 17023cfce8e4SKonstantin Belousov } 170320947801SPeter Wemm for ( ; rel < rellim; rel++) { 1704e9eabf59SPeter Wemm symidx = ELF_R_SYM(rel->r_info); 170520947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 170620947801SPeter Wemm continue; 1707e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 170820947801SPeter Wemm /* Only do local relocs */ 170920947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) != STB_LOCAL) 171020947801SPeter Wemm continue; 171136716fe2SMark Johnston if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC || 171244c705cfSMitchell Horne elf_is_ifunc_reloc(rel->r_info)) != ifuncs) 171344c705cfSMitchell Horne continue; 171444c705cfSMitchell Horne if (elf_reloc_local(lf, base, rel, ELF_RELOC_REL, 171544c705cfSMitchell Horne elf_obj_lookup) != 0) 171644c705cfSMitchell Horne return (ENOEXEC); 17177251b4bfSJake Burkholder } 1718e9eabf59SPeter Wemm } 17197251b4bfSJake Burkholder 17207251b4bfSJake Burkholder /* Perform relocations with addend if there are any: */ 1721b474c780SDavid E. O'Brien for (i = 0; i < ef->nrelatab; i++) { 1722e9eabf59SPeter Wemm rela = ef->relatab[i].rela; 17233cfce8e4SKonstantin Belousov if (rela == NULL) { 17243cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost a relatab!"); 17253cfce8e4SKonstantin Belousov return (ENOEXEC); 17263cfce8e4SKonstantin Belousov } 172720947801SPeter Wemm relalim = rela + ef->relatab[i].nrela; 1728e9eabf59SPeter Wemm base = findbase(ef, ef->relatab[i].sec); 17293cfce8e4SKonstantin Belousov if (base == 0) { 17303cfce8e4SKonstantin Belousov link_elf_error(ef->lf.filename, "lost base for reltab"); 17313cfce8e4SKonstantin Belousov return (ENOEXEC); 17323cfce8e4SKonstantin Belousov } 173320947801SPeter Wemm for ( ; rela < relalim; rela++) { 1734e9eabf59SPeter Wemm symidx = ELF_R_SYM(rela->r_info); 173520947801SPeter Wemm if (symidx >= ef->ddbsymcnt) 173620947801SPeter Wemm continue; 1737e9eabf59SPeter Wemm sym = ef->ddbsymtab + symidx; 173820947801SPeter Wemm /* Only do local relocs */ 173920947801SPeter Wemm if (ELF_ST_BIND(sym->st_info) != STB_LOCAL) 174020947801SPeter Wemm continue; 174136716fe2SMark Johnston if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC || 174244c705cfSMitchell Horne elf_is_ifunc_reloc(rela->r_info)) != ifuncs) 174344c705cfSMitchell Horne continue; 174444c705cfSMitchell Horne if (elf_reloc_local(lf, base, rela, ELF_RELOC_RELA, 174544c705cfSMitchell Horne elf_obj_lookup) != 0) 174644c705cfSMitchell Horne return (ENOEXEC); 17477251b4bfSJake Burkholder } 17487251b4bfSJake Burkholder } 17493cfce8e4SKonstantin Belousov return (0); 1750e9eabf59SPeter Wemm } 175100a5db46SStacey Son 175200a5db46SStacey Son static long 1753e76f11f4SAndriy Gapon link_elf_symtab_get(linker_file_t lf, const Elf_Sym **symtab) 175400a5db46SStacey Son { 175500a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 175600a5db46SStacey Son 175700a5db46SStacey Son *symtab = ef->ddbsymtab; 175800a5db46SStacey Son 175900a5db46SStacey Son if (*symtab == NULL) 176000a5db46SStacey Son return (0); 176100a5db46SStacey Son 176200a5db46SStacey Son return (ef->ddbsymcnt); 176300a5db46SStacey Son } 176400a5db46SStacey Son 176500a5db46SStacey Son static long 176600a5db46SStacey Son link_elf_strtab_get(linker_file_t lf, caddr_t *strtab) 176700a5db46SStacey Son { 176800a5db46SStacey Son elf_file_t ef = (elf_file_t)lf; 176900a5db46SStacey Son 177000a5db46SStacey Son *strtab = ef->ddbstrtab; 177100a5db46SStacey Son 177200a5db46SStacey Son if (*strtab == NULL) 177300a5db46SStacey Son return (0); 177400a5db46SStacey Son 177500a5db46SStacey Son return (ef->ddbstrcnt); 177600a5db46SStacey Son } 1777