xref: /freebsd/sys/kern/link_elf_obj.c (revision 95c20faf11a1af6924f97ec4aafc32d899fea8b0)
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 *);
134*95c20fafSKonstantin Belousov static int	link_elf_lookup_debug_symbol(linker_file_t, const char *,
135*95c20fafSKonstantin Belousov 		    c_linker_sym_t *);
136e9eabf59SPeter Wemm static int	link_elf_symbol_values(linker_file_t, c_linker_sym_t,
137e9eabf59SPeter Wemm 		    linker_symval_t *);
138*95c20fafSKonstantin Belousov static int	link_elf_debug_symbol_values(linker_file_t, c_linker_sym_t,
139*95c20fafSKonstantin Belousov 		    linker_symval_t *);
140326e27d8SDoug Rabson static int	link_elf_search_symbol(linker_file_t, caddr_t value,
141326e27d8SDoug Rabson 		    c_linker_sym_t *sym, long *diffp);
142326e27d8SDoug Rabson 
143326e27d8SDoug Rabson static void	link_elf_unload_file(linker_file_t);
144f41325dbSPeter Wemm static int	link_elf_lookup_set(linker_file_t, const char *,
145f41325dbSPeter Wemm 		    void ***, void ***, int *);
146bb9fe9ddSBrian Feldman static int	link_elf_each_function_name(linker_file_t,
147e9eabf59SPeter Wemm 		    int (*)(const char *, void *), void *);
148a2024a3eSJohn Birrell static int	link_elf_each_function_nameval(linker_file_t,
149a2024a3eSJohn Birrell 				linker_function_nameval_callback_t,
150a2024a3eSJohn Birrell 				void *);
151289ead7cSKonstantin Belousov static int	link_elf_reloc_local(linker_file_t, bool);
152e76f11f4SAndriy Gapon static long	link_elf_symtab_get(linker_file_t, const Elf_Sym **);
15300a5db46SStacey Son static long	link_elf_strtab_get(linker_file_t, caddr_t *);
154326e27d8SDoug Rabson 
155cff8c6f2SKonstantin Belousov static int	elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps,
156cff8c6f2SKonstantin Belousov 		    Elf_Addr *);
157e9eabf59SPeter Wemm 
158326e27d8SDoug Rabson static kobj_method_t link_elf_methods[] = {
159326e27d8SDoug Rabson 	KOBJMETHOD(linker_lookup_symbol,	link_elf_lookup_symbol),
160*95c20fafSKonstantin Belousov 	KOBJMETHOD(linker_lookup_debug_symbol,	link_elf_lookup_debug_symbol),
161326e27d8SDoug Rabson 	KOBJMETHOD(linker_symbol_values,	link_elf_symbol_values),
162*95c20fafSKonstantin Belousov 	KOBJMETHOD(linker_debug_symbol_values,	link_elf_debug_symbol_values),
163326e27d8SDoug Rabson 	KOBJMETHOD(linker_search_symbol,	link_elf_search_symbol),
164326e27d8SDoug Rabson 	KOBJMETHOD(linker_unload,		link_elf_unload_file),
16554823af2SPeter Wemm 	KOBJMETHOD(linker_load_file,		link_elf_load_file),
16654823af2SPeter Wemm 	KOBJMETHOD(linker_link_preload,		link_elf_link_preload),
16754823af2SPeter Wemm 	KOBJMETHOD(linker_link_preload_finish,	link_elf_link_preload_finish),
168f41325dbSPeter Wemm 	KOBJMETHOD(linker_lookup_set,		link_elf_lookup_set),
169bb9fe9ddSBrian Feldman 	KOBJMETHOD(linker_each_function_name,	link_elf_each_function_name),
170a2024a3eSJohn Birrell 	KOBJMETHOD(linker_each_function_nameval, link_elf_each_function_nameval),
171a2024a3eSJohn Birrell 	KOBJMETHOD(linker_ctf_get,		link_elf_ctf_get),
17200a5db46SStacey Son 	KOBJMETHOD(linker_symtab_get, 		link_elf_symtab_get),
17300a5db46SStacey Son 	KOBJMETHOD(linker_strtab_get, 		link_elf_strtab_get),
1746d775f0bSMark Johnston 	KOBJMETHOD_END
175326e27d8SDoug Rabson };
176326e27d8SDoug Rabson 
177326e27d8SDoug Rabson static struct linker_class link_elf_class = {
178326e27d8SDoug Rabson #if ELF_TARG_CLASS == ELFCLASS32
179e9eabf59SPeter Wemm 	"elf32_obj",
180326e27d8SDoug Rabson #else
181e9eabf59SPeter Wemm 	"elf64_obj",
182326e27d8SDoug Rabson #endif
183326e27d8SDoug Rabson 	link_elf_methods, sizeof(struct elf_file)
184326e27d8SDoug Rabson };
185326e27d8SDoug Rabson 
186326e27d8SDoug Rabson static int	relocate_file(elf_file_t ef);
1872832cd54SKonstantin Belousov static void	elf_obj_cleanup_globals_cache(elf_file_t);
188a4f67738SDoug Rabson 
189a4f67738SDoug Rabson static void
190552f9f63SEdwin Groothuis link_elf_error(const char *filename, const char *s)
1911720979bSMarcel Moolenaar {
192552f9f63SEdwin Groothuis 	if (filename == NULL)
1931720979bSMarcel Moolenaar 		printf("kldload: %s\n", s);
194552f9f63SEdwin Groothuis 	else
195552f9f63SEdwin Groothuis 		printf("kldload: %s: %s\n", filename, s);
1961720979bSMarcel Moolenaar }
1971720979bSMarcel Moolenaar 
1981720979bSMarcel Moolenaar static void
199a4f67738SDoug Rabson link_elf_init(void *arg)
200a4f67738SDoug Rabson {
201a4f67738SDoug Rabson 
202326e27d8SDoug Rabson 	linker_add_class(&link_elf_class);
203a4f67738SDoug Rabson }
204891cf3edSEd Maste SYSINIT(link_elf_obj, SI_SUB_KLD, SI_ORDER_SECOND, link_elf_init, NULL);
205a4f67738SDoug Rabson 
2061d9eae9fSMark Johnston static void
2071d9eae9fSMark Johnston link_elf_protect_range(elf_file_t ef, vm_offset_t start, vm_offset_t end,
2081d9eae9fSMark Johnston     vm_prot_t prot)
2091d9eae9fSMark Johnston {
2101d9eae9fSMark Johnston 	int error __unused;
2111d9eae9fSMark Johnston 
2121d9eae9fSMark Johnston 	KASSERT(start <= end && start >= (vm_offset_t)ef->address &&
2131d9eae9fSMark Johnston 	    end <= round_page((vm_offset_t)ef->address + ef->lf.size),
2141d9eae9fSMark Johnston 	    ("link_elf_protect_range: invalid range %#jx-%#jx",
2151d9eae9fSMark Johnston 	    (uintmax_t)start, (uintmax_t)end));
2161d9eae9fSMark Johnston 
2171d9eae9fSMark Johnston 	if (start == end)
2181d9eae9fSMark Johnston 		return;
219f822c9e2SMark Johnston 	if (ef->preloaded) {
220f822c9e2SMark Johnston #ifdef __amd64__
221f822c9e2SMark Johnston 		error = pmap_change_prot(start, end - start, prot);
222f822c9e2SMark Johnston 		KASSERT(error == 0,
223f822c9e2SMark Johnston 		    ("link_elf_protect_range: pmap_change_prot() returned %d",
224f822c9e2SMark Johnston 		    error));
225f822c9e2SMark Johnston #endif
226f822c9e2SMark Johnston 		return;
227f822c9e2SMark Johnston 	}
2280659df6fSKonstantin Belousov 	error = vm_map_protect(kernel_map, start, end, prot, 0,
2290659df6fSKonstantin Belousov 	    VM_MAP_PROTECT_SET_PROT);
2301d9eae9fSMark Johnston 	KASSERT(error == KERN_SUCCESS,
2311d9eae9fSMark Johnston 	    ("link_elf_protect_range: vm_map_protect() returned %d", error));
2321d9eae9fSMark Johnston }
2331d9eae9fSMark Johnston 
2341d9eae9fSMark Johnston /*
2351d9eae9fSMark Johnston  * Restrict permissions on linker file memory based on section flags.
2361d9eae9fSMark Johnston  * Sections need not be page-aligned, so overlap within a page is possible.
2371d9eae9fSMark Johnston  */
2381d9eae9fSMark Johnston static void
2391d9eae9fSMark Johnston link_elf_protect(elf_file_t ef)
2401d9eae9fSMark Johnston {
2411d9eae9fSMark Johnston 	vm_offset_t end, segend, segstart, start;
2421d9eae9fSMark Johnston 	vm_prot_t gapprot, prot, segprot;
2431d9eae9fSMark Johnston 	int i;
2441d9eae9fSMark Johnston 
2451d9eae9fSMark Johnston 	/*
2461d9eae9fSMark Johnston 	 * If the file was preloaded, the last page may contain other preloaded
2471d9eae9fSMark Johnston 	 * data which may need to be writeable.  ELF files are always
2481d9eae9fSMark Johnston 	 * page-aligned, but other preloaded data, such as entropy or CPU
2491d9eae9fSMark Johnston 	 * microcode may be loaded with a smaller alignment.
2501d9eae9fSMark Johnston 	 */
2511d9eae9fSMark Johnston 	gapprot = ef->preloaded ? VM_PROT_RW : VM_PROT_READ;
2521d9eae9fSMark Johnston 
2531d9eae9fSMark Johnston 	start = end = (vm_offset_t)ef->address;
2541d9eae9fSMark Johnston 	prot = VM_PROT_READ;
2551d9eae9fSMark Johnston 	for (i = 0; i < ef->nprogtab; i++) {
2561d9eae9fSMark Johnston 		/*
2571d9eae9fSMark Johnston 		 * VNET and DPCPU sections have their memory allocated by their
2581d9eae9fSMark Johnston 		 * respective subsystems.
2591d9eae9fSMark Johnston 		 */
2601d9eae9fSMark Johnston 		if (ef->progtab[i].name != NULL && (
2611d9eae9fSMark Johnston #ifdef VIMAGE
2621d9eae9fSMark Johnston 		    strcmp(ef->progtab[i].name, VNET_SETNAME) == 0 ||
2631d9eae9fSMark Johnston #endif
2641d9eae9fSMark Johnston 		    strcmp(ef->progtab[i].name, DPCPU_SETNAME) == 0))
2651d9eae9fSMark Johnston 			continue;
2661d9eae9fSMark Johnston 
2671d9eae9fSMark Johnston 		segstart = trunc_page((vm_offset_t)ef->progtab[i].addr);
2681d9eae9fSMark Johnston 		segend = round_page((vm_offset_t)ef->progtab[i].addr +
2691d9eae9fSMark Johnston 		    ef->progtab[i].size);
2701d9eae9fSMark Johnston 		segprot = VM_PROT_READ;
2711d9eae9fSMark Johnston 		if ((ef->progtab[i].flags & SHF_WRITE) != 0)
2721d9eae9fSMark Johnston 			segprot |= VM_PROT_WRITE;
2731d9eae9fSMark Johnston 		if ((ef->progtab[i].flags & SHF_EXECINSTR) != 0)
2741d9eae9fSMark Johnston 			segprot |= VM_PROT_EXECUTE;
2751d9eae9fSMark Johnston 
2761d9eae9fSMark Johnston 		if (end <= segstart) {
2771d9eae9fSMark Johnston 			/*
2781d9eae9fSMark Johnston 			 * Case 1: there is no overlap between the previous
2791d9eae9fSMark Johnston 			 * segment and this one.  Apply protections to the
2801d9eae9fSMark Johnston 			 * previous segment, and protect the gap between the
2811d9eae9fSMark Johnston 			 * previous and current segments, if any.
2821d9eae9fSMark Johnston 			 */
2831d9eae9fSMark Johnston 			link_elf_protect_range(ef, start, end, prot);
2841d9eae9fSMark Johnston 			link_elf_protect_range(ef, end, segstart, gapprot);
2851d9eae9fSMark Johnston 
2861d9eae9fSMark Johnston 			start = segstart;
2871d9eae9fSMark Johnston 			end = segend;
2881d9eae9fSMark Johnston 			prot = segprot;
2891d9eae9fSMark Johnston 		} else if (start < segstart && end == segend) {
2901d9eae9fSMark Johnston 			/*
2911d9eae9fSMark Johnston 			 * Case 2: the current segment is a subrange of the
2921d9eae9fSMark Johnston 			 * previous segment.  Apply protections to the
2931d9eae9fSMark Johnston 			 * non-overlapping portion of the previous segment.
2941d9eae9fSMark Johnston 			 */
2951d9eae9fSMark Johnston 			link_elf_protect_range(ef, start, segstart, prot);
2961d9eae9fSMark Johnston 
2971d9eae9fSMark Johnston 			start = segstart;
2981d9eae9fSMark Johnston 			prot |= segprot;
2991d9eae9fSMark Johnston 		} else if (end < segend) {
3001d9eae9fSMark Johnston 			/*
3011d9eae9fSMark Johnston 			 * Case 3: there is partial overlap between the previous
3021d9eae9fSMark Johnston 			 * and current segments.  Apply protections to the
3031d9eae9fSMark Johnston 			 * non-overlapping portion of the previous segment, and
3041d9eae9fSMark Johnston 			 * then the overlap, which must use the union of the two
3051d9eae9fSMark Johnston 			 * segments' protections.
3061d9eae9fSMark Johnston 			 */
3071d9eae9fSMark Johnston 			link_elf_protect_range(ef, start, segstart, prot);
3081d9eae9fSMark Johnston 			link_elf_protect_range(ef, segstart, end,
3091d9eae9fSMark Johnston 			    prot | segprot);
3101d9eae9fSMark Johnston 			start = end;
3111d9eae9fSMark Johnston 			end = segend;
3121d9eae9fSMark Johnston 			prot = segprot;
3131d9eae9fSMark Johnston 		} else {
3141d9eae9fSMark Johnston 			/*
3151d9eae9fSMark Johnston 			 * Case 4: the two segments reside in the same page.
3161d9eae9fSMark Johnston 			 */
3171d9eae9fSMark Johnston 			prot |= segprot;
3181d9eae9fSMark Johnston 		}
3191d9eae9fSMark Johnston 	}
3201d9eae9fSMark Johnston 
3211d9eae9fSMark Johnston 	/*
3221d9eae9fSMark Johnston 	 * Fix up the last unprotected segment and trailing data.
3231d9eae9fSMark Johnston 	 */
3241d9eae9fSMark Johnston 	link_elf_protect_range(ef, start, end, prot);
3251d9eae9fSMark Johnston 	link_elf_protect_range(ef, end,
3261d9eae9fSMark Johnston 	    round_page((vm_offset_t)ef->address + ef->lf.size), gapprot);
3271d9eae9fSMark Johnston }
3281d9eae9fSMark Johnston 
329a4f67738SDoug Rabson static int
330e9eabf59SPeter Wemm link_elf_link_preload(linker_class_t cls, const char *filename,
331e9eabf59SPeter Wemm     linker_file_t *result)
3322d636ab0SPeter Wemm {
33370b7ffeeSIan Dowse 	Elf_Ehdr *hdr;
33470b7ffeeSIan Dowse 	Elf_Shdr *shdr;
33570b7ffeeSIan Dowse 	Elf_Sym *es;
33670b7ffeeSIan Dowse 	void *modptr, *baseptr, *sizeptr;
33770b7ffeeSIan Dowse 	char *type;
33870b7ffeeSIan Dowse 	elf_file_t ef;
33970b7ffeeSIan Dowse 	linker_file_t lf;
34070b7ffeeSIan Dowse 	Elf_Addr off;
34170b7ffeeSIan Dowse 	int error, i, j, pb, ra, rl, shstrindex, symstrindex, symtabindex;
34270b7ffeeSIan Dowse 
34370b7ffeeSIan Dowse 	/* Look to see if we have the file preloaded */
34470b7ffeeSIan Dowse 	modptr = preload_search_by_name(filename);
34570b7ffeeSIan Dowse 	if (modptr == NULL)
34670b7ffeeSIan Dowse 		return ENOENT;
34770b7ffeeSIan Dowse 
34870b7ffeeSIan Dowse 	type = (char *)preload_search_info(modptr, MODINFO_TYPE);
34970b7ffeeSIan Dowse 	baseptr = preload_search_info(modptr, MODINFO_ADDR);
35070b7ffeeSIan Dowse 	sizeptr = preload_search_info(modptr, MODINFO_SIZE);
35170b7ffeeSIan Dowse 	hdr = (Elf_Ehdr *)preload_search_info(modptr, MODINFO_METADATA |
35270b7ffeeSIan Dowse 	    MODINFOMD_ELFHDR);
35370b7ffeeSIan Dowse 	shdr = (Elf_Shdr *)preload_search_info(modptr, MODINFO_METADATA |
35470b7ffeeSIan Dowse 	    MODINFOMD_SHDR);
35570b7ffeeSIan Dowse 	if (type == NULL || (strcmp(type, "elf" __XSTRING(__ELF_WORD_SIZE)
35670b7ffeeSIan Dowse 	    " obj module") != 0 &&
35770b7ffeeSIan Dowse 	    strcmp(type, "elf obj module") != 0)) {
358de78ca7eSPeter Wemm 		return (EFTYPE);
359de78ca7eSPeter Wemm 	}
36070b7ffeeSIan Dowse 	if (baseptr == NULL || sizeptr == NULL || hdr == NULL ||
36170b7ffeeSIan Dowse 	    shdr == NULL)
36270b7ffeeSIan Dowse 		return (EINVAL);
36370b7ffeeSIan Dowse 
36470b7ffeeSIan Dowse 	lf = linker_make_file(filename, &link_elf_class);
36570b7ffeeSIan Dowse 	if (lf == NULL)
36670b7ffeeSIan Dowse 		return (ENOMEM);
36770b7ffeeSIan Dowse 
36870b7ffeeSIan Dowse 	ef = (elf_file_t)lf;
36970b7ffeeSIan Dowse 	ef->preloaded = 1;
37070b7ffeeSIan Dowse 	ef->address = *(caddr_t *)baseptr;
37170b7ffeeSIan Dowse 	lf->address = *(caddr_t *)baseptr;
37270b7ffeeSIan Dowse 	lf->size = *(size_t *)sizeptr;
37370b7ffeeSIan Dowse 
37470b7ffeeSIan Dowse 	if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS ||
37570b7ffeeSIan Dowse 	    hdr->e_ident[EI_DATA] != ELF_TARG_DATA ||
37670b7ffeeSIan Dowse 	    hdr->e_ident[EI_VERSION] != EV_CURRENT ||
37770b7ffeeSIan Dowse 	    hdr->e_version != EV_CURRENT ||
37870b7ffeeSIan Dowse 	    hdr->e_type != ET_REL ||
37970b7ffeeSIan Dowse 	    hdr->e_machine != ELF_TARG_MACH) {
38070b7ffeeSIan Dowse 		error = EFTYPE;
38170b7ffeeSIan Dowse 		goto out;
38270b7ffeeSIan Dowse 	}
38370b7ffeeSIan Dowse 	ef->e_shdr = shdr;
38470b7ffeeSIan Dowse 
38570b7ffeeSIan Dowse 	/* Scan the section header for information and table sizing. */
38670b7ffeeSIan Dowse 	symtabindex = -1;
38770b7ffeeSIan Dowse 	symstrindex = -1;
38870b7ffeeSIan Dowse 	for (i = 0; i < hdr->e_shnum; i++) {
38970b7ffeeSIan Dowse 		switch (shdr[i].sh_type) {
39070b7ffeeSIan Dowse 		case SHT_PROGBITS:
39170b7ffeeSIan Dowse 		case SHT_NOBITS:
39213f28d96SKonstantin Belousov #ifdef __amd64__
393b715d9afSKonstantin Belousov 		case SHT_X86_64_UNWIND:
39413f28d96SKonstantin Belousov #endif
3955e6989baSMark Johnston 		case SHT_INIT_ARRAY:
3965e6989baSMark Johnston 		case SHT_FINI_ARRAY:
39758c4aee0SJohn Baldwin 			/* Ignore sections not loaded by the loader. */
39858c4aee0SJohn Baldwin 			if (shdr[i].sh_addr == 0)
39958c4aee0SJohn Baldwin 				break;
40070b7ffeeSIan Dowse 			ef->nprogtab++;
40170b7ffeeSIan Dowse 			break;
40270b7ffeeSIan Dowse 		case SHT_SYMTAB:
40370b7ffeeSIan Dowse 			symtabindex = i;
40470b7ffeeSIan Dowse 			symstrindex = shdr[i].sh_link;
40570b7ffeeSIan Dowse 			break;
40670b7ffeeSIan Dowse 		case SHT_REL:
40715746ef4SJohn Baldwin 			/*
40815746ef4SJohn Baldwin 			 * Ignore relocation tables for sections not
40915746ef4SJohn Baldwin 			 * loaded by the loader.
41015746ef4SJohn Baldwin 			 */
41115746ef4SJohn Baldwin 			if (shdr[shdr[i].sh_info].sh_addr == 0)
41215746ef4SJohn Baldwin 				break;
413b474c780SDavid E. O'Brien 			ef->nreltab++;
41470b7ffeeSIan Dowse 			break;
41570b7ffeeSIan Dowse 		case SHT_RELA:
41615746ef4SJohn Baldwin 			if (shdr[shdr[i].sh_info].sh_addr == 0)
41715746ef4SJohn Baldwin 				break;
418b474c780SDavid E. O'Brien 			ef->nrelatab++;
41970b7ffeeSIan Dowse 			break;
42070b7ffeeSIan Dowse 		}
42170b7ffeeSIan Dowse 	}
42270b7ffeeSIan Dowse 
42370b7ffeeSIan Dowse 	shstrindex = hdr->e_shstrndx;
42470b7ffeeSIan Dowse 	if (ef->nprogtab == 0 || symstrindex < 0 ||
42570b7ffeeSIan Dowse 	    symstrindex >= hdr->e_shnum ||
42670b7ffeeSIan Dowse 	    shdr[symstrindex].sh_type != SHT_STRTAB || shstrindex == 0 ||
42770b7ffeeSIan Dowse 	    shstrindex >= hdr->e_shnum ||
42870b7ffeeSIan Dowse 	    shdr[shstrindex].sh_type != SHT_STRTAB) {
42970b7ffeeSIan Dowse 		printf("%s: bad/missing section headers\n", filename);
43070b7ffeeSIan Dowse 		error = ENOEXEC;
43170b7ffeeSIan Dowse 		goto out;
43270b7ffeeSIan Dowse 	}
43370b7ffeeSIan Dowse 
43470b7ffeeSIan Dowse 	/* Allocate space for tracking the load chunks */
43570b7ffeeSIan Dowse 	if (ef->nprogtab != 0)
43670b7ffeeSIan Dowse 		ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab),
43770b7ffeeSIan Dowse 		    M_LINKER, M_WAITOK | M_ZERO);
438b474c780SDavid E. O'Brien 	if (ef->nreltab != 0)
439b474c780SDavid E. O'Brien 		ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab),
440b474c780SDavid E. O'Brien 		    M_LINKER, M_WAITOK | M_ZERO);
441b474c780SDavid E. O'Brien 	if (ef->nrelatab != 0)
442b474c780SDavid E. O'Brien 		ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab),
443b474c780SDavid E. O'Brien 		    M_LINKER, M_WAITOK | M_ZERO);
44470b7ffeeSIan Dowse 	if ((ef->nprogtab != 0 && ef->progtab == NULL) ||
445b474c780SDavid E. O'Brien 	    (ef->nreltab != 0 && ef->reltab == NULL) ||
446b474c780SDavid E. O'Brien 	    (ef->nrelatab != 0 && ef->relatab == NULL)) {
44770b7ffeeSIan Dowse 		error = ENOMEM;
44870b7ffeeSIan Dowse 		goto out;
44970b7ffeeSIan Dowse 	}
45070b7ffeeSIan Dowse 
45170b7ffeeSIan Dowse 	/* XXX, relocate the sh_addr fields saved by the loader. */
45270b7ffeeSIan Dowse 	off = 0;
45370b7ffeeSIan Dowse 	for (i = 0; i < hdr->e_shnum; i++) {
45470b7ffeeSIan Dowse 		if (shdr[i].sh_addr != 0 && (off == 0 || shdr[i].sh_addr < off))
45570b7ffeeSIan Dowse 			off = shdr[i].sh_addr;
45670b7ffeeSIan Dowse 	}
45770b7ffeeSIan Dowse 	for (i = 0; i < hdr->e_shnum; i++) {
45870b7ffeeSIan Dowse 		if (shdr[i].sh_addr != 0)
45970b7ffeeSIan Dowse 			shdr[i].sh_addr = shdr[i].sh_addr - off +
46070b7ffeeSIan Dowse 			    (Elf_Addr)ef->address;
46170b7ffeeSIan Dowse 	}
46270b7ffeeSIan Dowse 
46370b7ffeeSIan Dowse 	ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym);
46470b7ffeeSIan Dowse 	ef->ddbsymtab = (Elf_Sym *)shdr[symtabindex].sh_addr;
46570b7ffeeSIan Dowse 	ef->ddbstrcnt = shdr[symstrindex].sh_size;
46670b7ffeeSIan Dowse 	ef->ddbstrtab = (char *)shdr[symstrindex].sh_addr;
46770b7ffeeSIan Dowse 	ef->shstrcnt = shdr[shstrindex].sh_size;
46870b7ffeeSIan Dowse 	ef->shstrtab = (char *)shdr[shstrindex].sh_addr;
46970b7ffeeSIan Dowse 
47070b7ffeeSIan Dowse 	/* Now fill out progtab and the relocation tables. */
47170b7ffeeSIan Dowse 	pb = 0;
47270b7ffeeSIan Dowse 	rl = 0;
47370b7ffeeSIan Dowse 	ra = 0;
47470b7ffeeSIan Dowse 	for (i = 0; i < hdr->e_shnum; i++) {
47570b7ffeeSIan Dowse 		switch (shdr[i].sh_type) {
47670b7ffeeSIan Dowse 		case SHT_PROGBITS:
47770b7ffeeSIan Dowse 		case SHT_NOBITS:
47813f28d96SKonstantin Belousov #ifdef __amd64__
479b715d9afSKonstantin Belousov 		case SHT_X86_64_UNWIND:
48013f28d96SKonstantin Belousov #endif
4814aa157ddSMark Johnston 		case SHT_INIT_ARRAY:
4825e6989baSMark Johnston 		case SHT_FINI_ARRAY:
48358c4aee0SJohn Baldwin 			if (shdr[i].sh_addr == 0)
48458c4aee0SJohn Baldwin 				break;
48570b7ffeeSIan Dowse 			ef->progtab[pb].addr = (void *)shdr[i].sh_addr;
48670b7ffeeSIan Dowse 			if (shdr[i].sh_type == SHT_PROGBITS)
48770b7ffeeSIan Dowse 				ef->progtab[pb].name = "<<PROGBITS>>";
48813f28d96SKonstantin Belousov #ifdef __amd64__
489b715d9afSKonstantin Belousov 			else if (shdr[i].sh_type == SHT_X86_64_UNWIND)
49013f28d96SKonstantin Belousov 				ef->progtab[pb].name = "<<UNWIND>>";
49113f28d96SKonstantin Belousov #endif
4925e6989baSMark Johnston 			else if (shdr[i].sh_type == SHT_INIT_ARRAY)
4935e6989baSMark Johnston 				ef->progtab[pb].name = "<<INIT_ARRAY>>";
4945e6989baSMark Johnston 			else if (shdr[i].sh_type == SHT_FINI_ARRAY)
4955e6989baSMark Johnston 				ef->progtab[pb].name = "<<FINI_ARRAY>>";
49670b7ffeeSIan Dowse 			else
49770b7ffeeSIan Dowse 				ef->progtab[pb].name = "<<NOBITS>>";
49870b7ffeeSIan Dowse 			ef->progtab[pb].size = shdr[i].sh_size;
4991d9eae9fSMark Johnston 			ef->progtab[pb].flags = shdr[i].sh_flags;
50070b7ffeeSIan Dowse 			ef->progtab[pb].sec = i;
50170b7ffeeSIan Dowse 			if (ef->shstrtab && shdr[i].sh_name != 0)
50270b7ffeeSIan Dowse 				ef->progtab[pb].name =
50370b7ffeeSIan Dowse 				    ef->shstrtab + shdr[i].sh_name;
50450c202c5SJeff Roberson 			if (ef->progtab[pb].name != NULL &&
5055f67450dSDimitry Andric 			    !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) {
50650c202c5SJeff Roberson 				void *dpcpu;
50750c202c5SJeff Roberson 
50850c202c5SJeff Roberson 				dpcpu = dpcpu_alloc(shdr[i].sh_size);
50950c202c5SJeff Roberson 				if (dpcpu == NULL) {
5100f823b64SBjoern A. Zeeb 					printf("%s: pcpu module space is out "
5119afc5684SBjoern A. Zeeb 					    "of space; cannot allocate %#jx "
5129afc5684SBjoern A. Zeeb 					    "for %s\n", __func__,
5139afc5684SBjoern A. Zeeb 					    (uintmax_t)shdr[i].sh_size,
5140f823b64SBjoern A. Zeeb 					    filename);
51550c202c5SJeff Roberson 					error = ENOSPC;
51650c202c5SJeff Roberson 					goto out;
51750c202c5SJeff Roberson 				}
51850c202c5SJeff Roberson 				memcpy(dpcpu, ef->progtab[pb].addr,
51950c202c5SJeff Roberson 				    ef->progtab[pb].size);
52050c202c5SJeff Roberson 				dpcpu_copy(dpcpu, shdr[i].sh_size);
52150c202c5SJeff Roberson 				ef->progtab[pb].addr = dpcpu;
522eddfbb76SRobert Watson #ifdef VIMAGE
523eddfbb76SRobert Watson 			} else if (ef->progtab[pb].name != NULL &&
52417ef1febSRobert Watson 			    !strcmp(ef->progtab[pb].name, VNET_SETNAME)) {
525eddfbb76SRobert Watson 				void *vnet_data;
526eddfbb76SRobert Watson 
527eddfbb76SRobert Watson 				vnet_data = vnet_data_alloc(shdr[i].sh_size);
528eddfbb76SRobert Watson 				if (vnet_data == NULL) {
5290f823b64SBjoern A. Zeeb 					printf("%s: vnet module space is out "
5309afc5684SBjoern A. Zeeb 					    "of space; cannot allocate %#jx "
5319afc5684SBjoern A. Zeeb 					    "for %s\n", __func__,
5329afc5684SBjoern A. Zeeb 					    (uintmax_t)shdr[i].sh_size,
5330f823b64SBjoern A. Zeeb 					    filename);
534eddfbb76SRobert Watson 					error = ENOSPC;
535eddfbb76SRobert Watson 					goto out;
536eddfbb76SRobert Watson 				}
537eddfbb76SRobert Watson 				memcpy(vnet_data, ef->progtab[pb].addr,
538eddfbb76SRobert Watson 				    ef->progtab[pb].size);
539eddfbb76SRobert Watson 				vnet_data_copy(vnet_data, shdr[i].sh_size);
540eddfbb76SRobert Watson 				ef->progtab[pb].addr = vnet_data;
541eddfbb76SRobert Watson #endif
5425e6989baSMark Johnston 			} else if ((ef->progtab[pb].name != NULL &&
5435e6989baSMark Johnston 			    strcmp(ef->progtab[pb].name, ".ctors") == 0) ||
5445e6989baSMark Johnston 			    shdr[i].sh_type == SHT_INIT_ARRAY) {
5455e6989baSMark Johnston 				if (lf->ctors_addr != 0) {
5465e6989baSMark Johnston 					printf(
5475e6989baSMark Johnston 				    "%s: multiple ctor sections in %s\n",
5485e6989baSMark Johnston 					    __func__, filename);
5495e6989baSMark Johnston 				} else {
5500067051fSMarcel Moolenaar 					lf->ctors_addr = ef->progtab[pb].addr;
5510067051fSMarcel Moolenaar 					lf->ctors_size = shdr[i].sh_size;
55250c202c5SJeff Roberson 				}
5539e575fadSMark Johnston 			} else if ((ef->progtab[pb].name != NULL &&
5549e575fadSMark Johnston 			    strcmp(ef->progtab[pb].name, ".dtors") == 0) ||
5559e575fadSMark Johnston 			    shdr[i].sh_type == SHT_FINI_ARRAY) {
5569e575fadSMark Johnston 				if (lf->dtors_addr != 0) {
5579e575fadSMark Johnston 					printf(
5589e575fadSMark Johnston 				    "%s: multiple dtor sections in %s\n",
5599e575fadSMark Johnston 					    __func__, filename);
5609e575fadSMark Johnston 				} else {
5619e575fadSMark Johnston 					lf->dtors_addr = ef->progtab[pb].addr;
5629e575fadSMark Johnston 					lf->dtors_size = shdr[i].sh_size;
5639e575fadSMark Johnston 				}
5645e6989baSMark Johnston 			}
56570b7ffeeSIan Dowse 
56670b7ffeeSIan Dowse 			/* Update all symbol values with the offset. */
56770b7ffeeSIan Dowse 			for (j = 0; j < ef->ddbsymcnt; j++) {
56870b7ffeeSIan Dowse 				es = &ef->ddbsymtab[j];
56970b7ffeeSIan Dowse 				if (es->st_shndx != i)
57070b7ffeeSIan Dowse 					continue;
57170b7ffeeSIan Dowse 				es->st_value += (Elf_Addr)ef->progtab[pb].addr;
57270b7ffeeSIan Dowse 			}
57370b7ffeeSIan Dowse 			pb++;
57470b7ffeeSIan Dowse 			break;
57570b7ffeeSIan Dowse 		case SHT_REL:
57615746ef4SJohn Baldwin 			if (shdr[shdr[i].sh_info].sh_addr == 0)
57715746ef4SJohn Baldwin 				break;
57870b7ffeeSIan Dowse 			ef->reltab[rl].rel = (Elf_Rel *)shdr[i].sh_addr;
57970b7ffeeSIan Dowse 			ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel);
58070b7ffeeSIan Dowse 			ef->reltab[rl].sec = shdr[i].sh_info;
58170b7ffeeSIan Dowse 			rl++;
58270b7ffeeSIan Dowse 			break;
58370b7ffeeSIan Dowse 		case SHT_RELA:
58415746ef4SJohn Baldwin 			if (shdr[shdr[i].sh_info].sh_addr == 0)
58515746ef4SJohn Baldwin 				break;
58670b7ffeeSIan Dowse 			ef->relatab[ra].rela = (Elf_Rela *)shdr[i].sh_addr;
58770b7ffeeSIan Dowse 			ef->relatab[ra].nrela =
58870b7ffeeSIan Dowse 			    shdr[i].sh_size / sizeof(Elf_Rela);
58970b7ffeeSIan Dowse 			ef->relatab[ra].sec = shdr[i].sh_info;
59070b7ffeeSIan Dowse 			ra++;
59170b7ffeeSIan Dowse 			break;
59270b7ffeeSIan Dowse 		}
59370b7ffeeSIan Dowse 	}
5943cfce8e4SKonstantin Belousov 	if (pb != ef->nprogtab) {
5953cfce8e4SKonstantin Belousov 		printf("%s: lost progbits\n", filename);
5963cfce8e4SKonstantin Belousov 		error = ENOEXEC;
5973cfce8e4SKonstantin Belousov 		goto out;
5983cfce8e4SKonstantin Belousov 	}
5993cfce8e4SKonstantin Belousov 	if (rl != ef->nreltab) {
6003cfce8e4SKonstantin Belousov 		printf("%s: lost reltab\n", filename);
6013cfce8e4SKonstantin Belousov 		error = ENOEXEC;
6023cfce8e4SKonstantin Belousov 		goto out;
6033cfce8e4SKonstantin Belousov 	}
6043cfce8e4SKonstantin Belousov 	if (ra != ef->nrelatab) {
6053cfce8e4SKonstantin Belousov 		printf("%s: lost relatab\n", filename);
6063cfce8e4SKonstantin Belousov 		error = ENOEXEC;
6073cfce8e4SKonstantin Belousov 		goto out;
6083cfce8e4SKonstantin Belousov 	}
60970b7ffeeSIan Dowse 
610f822c9e2SMark Johnston 	/*
611f822c9e2SMark Johnston 	 * The file needs to be writeable and executable while applying
612f822c9e2SMark Johnston 	 * relocations.  Mapping protections are applied once relocation
613f822c9e2SMark Johnston 	 * processing is complete.
614f822c9e2SMark Johnston 	 */
615f822c9e2SMark Johnston 	link_elf_protect_range(ef, (vm_offset_t)ef->address,
616f822c9e2SMark Johnston 	    round_page((vm_offset_t)ef->address + ef->lf.size), VM_PROT_ALL);
617f822c9e2SMark Johnston 
61870b7ffeeSIan Dowse 	/* Local intra-module relocations */
619289ead7cSKonstantin Belousov 	error = link_elf_reloc_local(lf, false);
6203cfce8e4SKonstantin Belousov 	if (error != 0)
6213cfce8e4SKonstantin Belousov 		goto out;
62270b7ffeeSIan Dowse 	*result = lf;
62370b7ffeeSIan Dowse 	return (0);
62470b7ffeeSIan Dowse 
62570b7ffeeSIan Dowse out:
62670b7ffeeSIan Dowse 	/* preload not done this way */
62770b7ffeeSIan Dowse 	linker_file_unload(lf, LINKER_UNLOAD_FORCE);
62870b7ffeeSIan Dowse 	return (error);
62970b7ffeeSIan Dowse }
630de78ca7eSPeter Wemm 
6310067051fSMarcel Moolenaar static void
6329e575fadSMark Johnston link_elf_invoke_cbs(caddr_t addr, size_t size)
6330067051fSMarcel Moolenaar {
6340067051fSMarcel Moolenaar 	void (**ctor)(void);
6350067051fSMarcel Moolenaar 	size_t i, cnt;
6360067051fSMarcel Moolenaar 
6370067051fSMarcel Moolenaar 	if (addr == NULL || size == 0)
6380067051fSMarcel Moolenaar 		return;
6390067051fSMarcel Moolenaar 	cnt = size / sizeof(*ctor);
6400067051fSMarcel Moolenaar 	ctor = (void *)addr;
6410067051fSMarcel Moolenaar 	for (i = 0; i < cnt; i++) {
6420067051fSMarcel Moolenaar 		if (ctor[i] != NULL)
6430067051fSMarcel Moolenaar 			(*ctor[i])();
6440067051fSMarcel Moolenaar 	}
6450067051fSMarcel Moolenaar }
6460067051fSMarcel Moolenaar 
647de78ca7eSPeter Wemm static int
64854823af2SPeter Wemm link_elf_link_preload_finish(linker_file_t lf)
64954823af2SPeter Wemm {
65070b7ffeeSIan Dowse 	elf_file_t ef;
65170b7ffeeSIan Dowse 	int error;
65270b7ffeeSIan Dowse 
65370b7ffeeSIan Dowse 	ef = (elf_file_t)lf;
65470b7ffeeSIan Dowse 	error = relocate_file(ef);
65570b7ffeeSIan Dowse 	if (error)
656289ead7cSKonstantin Belousov 		return (error);
65770b7ffeeSIan Dowse 
65870b7ffeeSIan Dowse 	/* Notify MD code that a module is being loaded. */
65970b7ffeeSIan Dowse 	error = elf_cpu_load_file(lf);
66070b7ffeeSIan Dowse 	if (error)
66170b7ffeeSIan Dowse 		return (error);
66270b7ffeeSIan Dowse 
6638f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__)
664289ead7cSKonstantin Belousov 	/* Now ifuncs. */
665289ead7cSKonstantin Belousov 	error = link_elf_reloc_local(lf, true);
666289ead7cSKonstantin Belousov 	if (error != 0)
667289ead7cSKonstantin Belousov 		return (error);
6688f941950SKonstantin Belousov #endif
669289ead7cSKonstantin Belousov 
670f822c9e2SMark Johnston 	/* Apply protections now that relocation processing is complete. */
671f822c9e2SMark Johnston 	link_elf_protect(ef);
672f822c9e2SMark Johnston 
6739e575fadSMark Johnston 	link_elf_invoke_cbs(lf->ctors_addr, lf->ctors_size);
67470b7ffeeSIan Dowse 	return (0);
67554823af2SPeter Wemm }
67654823af2SPeter Wemm 
67754823af2SPeter Wemm static int
678c143d6c2SMarcel Moolenaar link_elf_load_file(linker_class_t cls, const char *filename,
679c143d6c2SMarcel Moolenaar     linker_file_t *result)
680a4f67738SDoug Rabson {
681a344babfSGleb Smirnoff 	struct nameidata *nd;
682b40ce416SJulian Elischer 	struct thread *td = curthread;	/* XXX */
683caab6e90SPeter Wemm 	Elf_Ehdr *hdr;
684e9eabf59SPeter Wemm 	Elf_Shdr *shdr;
685a8774e39SPeter Wemm 	Elf_Sym *es;
686a8774e39SPeter Wemm 	int nbytes, i, j;
68720947801SPeter Wemm 	vm_offset_t mapbase;
688fe3db7c7SDoug Rabson 	size_t mapsize;
689a4f67738SDoug Rabson 	int error = 0;
690526d0bd5SKonstantin Belousov 	ssize_t resid;
691526d0bd5SKonstantin Belousov 	int flags;
692a4f67738SDoug Rabson 	elf_file_t ef;
693a4f67738SDoug Rabson 	linker_file_t lf;
694ca65d5c7SPeter Wemm 	int symtabindex;
695ca65d5c7SPeter Wemm 	int symstrindex;
696e9eabf59SPeter Wemm 	int shstrindex;
697e9eabf59SPeter Wemm 	int nsym;
69820947801SPeter Wemm 	int pb, rl, ra;
699e9eabf59SPeter Wemm 	int alignmask;
700ca65d5c7SPeter Wemm 
701ca65d5c7SPeter Wemm 	shdr = NULL;
702ca65d5c7SPeter Wemm 	lf = NULL;
703e9eabf59SPeter Wemm 	mapsize = 0;
704e9eabf59SPeter Wemm 	hdr = NULL;
705a4f67738SDoug Rabson 
706a344babfSGleb Smirnoff 	nd = malloc(sizeof(struct nameidata), M_TEMP, M_WAITOK);
7077e1d3eefSMateusz Guzik 	NDINIT(nd, LOOKUP, FOLLOW, UIO_SYSSPACE, filename);
708e6796b67SKirk McKusick 	flags = FREAD;
709a344babfSGleb Smirnoff 	error = vn_open(nd, &flags, 0, NULL);
710a344babfSGleb Smirnoff 	if (error) {
711a344babfSGleb Smirnoff 		free(nd, M_TEMP);
712a4f67738SDoug Rabson 		return error;
713a344babfSGleb Smirnoff 	}
714a344babfSGleb Smirnoff 	NDFREE(nd, NDF_ONLY_PNBUF);
715a344babfSGleb Smirnoff 	if (nd->ni_vp->v_type != VREG) {
7161f49b573SRuslan Ermilov 		error = ENOEXEC;
7171f49b573SRuslan Ermilov 		goto out;
7181f49b573SRuslan Ermilov 	}
719a3df768bSRobert Watson #ifdef MAC
720a344babfSGleb Smirnoff 	error = mac_kld_check_load(td->td_ucred, nd->ni_vp);
721a3df768bSRobert Watson 	if (error) {
722a3df768bSRobert Watson 		goto out;
723a3df768bSRobert Watson 	}
724a3df768bSRobert Watson #endif
725a4f67738SDoug Rabson 
726e9eabf59SPeter Wemm 	/* Read the elf header from the file. */
727e9eabf59SPeter Wemm 	hdr = malloc(sizeof(*hdr), M_LINKER, M_WAITOK);
728a344babfSGleb Smirnoff 	error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)hdr, sizeof(*hdr), 0,
7299ca43589SRobert Watson 	    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
7309ca43589SRobert Watson 	    &resid, td);
731a4f67738SDoug Rabson 	if (error)
732a4f67738SDoug Rabson 		goto out;
733e9eabf59SPeter Wemm 	if (resid != 0){
734e9eabf59SPeter Wemm 		error = ENOEXEC;
735e9eabf59SPeter Wemm 		goto out;
736e9eabf59SPeter Wemm 	}
737a4f67738SDoug Rabson 
738caab6e90SPeter Wemm 	if (!IS_ELF(*hdr)) {
739fe3db7c7SDoug Rabson 		error = ENOEXEC;
740a4f67738SDoug Rabson 		goto out;
741fe3db7c7SDoug Rabson 	}
742fe3db7c7SDoug Rabson 
743caab6e90SPeter Wemm 	if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS
744caab6e90SPeter Wemm 	    || hdr->e_ident[EI_DATA] != ELF_TARG_DATA) {
745552f9f63SEdwin Groothuis 		link_elf_error(filename, "Unsupported file layout");
746fe3db7c7SDoug Rabson 		error = ENOEXEC;
747fe3db7c7SDoug Rabson 		goto out;
748fe3db7c7SDoug Rabson 	}
749caab6e90SPeter Wemm 	if (hdr->e_ident[EI_VERSION] != EV_CURRENT
750caab6e90SPeter Wemm 	    || hdr->e_version != EV_CURRENT) {
751552f9f63SEdwin Groothuis 		link_elf_error(filename, "Unsupported file version");
752fe3db7c7SDoug Rabson 		error = ENOEXEC;
753fe3db7c7SDoug Rabson 		goto out;
754fe3db7c7SDoug Rabson 	}
755e9eabf59SPeter Wemm 	if (hdr->e_type != ET_REL) {
756a1d7ce03SAttilio Rao 		error = ENOSYS;
757fe3db7c7SDoug Rabson 		goto out;
758fe3db7c7SDoug Rabson 	}
759caab6e90SPeter Wemm 	if (hdr->e_machine != ELF_TARG_MACH) {
760552f9f63SEdwin Groothuis 		link_elf_error(filename, "Unsupported machine");
761fe3db7c7SDoug Rabson 		error = ENOEXEC;
762fe3db7c7SDoug Rabson 		goto out;
763fe3db7c7SDoug Rabson 	}
764a4f67738SDoug Rabson 
765326e27d8SDoug Rabson 	lf = linker_make_file(filename, &link_elf_class);
766326e27d8SDoug Rabson 	if (!lf) {
767326e27d8SDoug Rabson 		error = ENOMEM;
768326e27d8SDoug Rabson 		goto out;
769326e27d8SDoug Rabson 	}
770326e27d8SDoug Rabson 	ef = (elf_file_t) lf;
771e9eabf59SPeter Wemm 	ef->nprogtab = 0;
772e9eabf59SPeter Wemm 	ef->e_shdr = 0;
773b474c780SDavid E. O'Brien 	ef->nreltab = 0;
774b474c780SDavid E. O'Brien 	ef->nrelatab = 0;
775e9eabf59SPeter Wemm 
776e9eabf59SPeter Wemm 	/* Allocate and read in the section header */
777e9eabf59SPeter Wemm 	nbytes = hdr->e_shnum * hdr->e_shentsize;
778e9eabf59SPeter Wemm 	if (nbytes == 0 || hdr->e_shoff == 0 ||
779e9eabf59SPeter Wemm 	    hdr->e_shentsize != sizeof(Elf_Shdr)) {
780e9eabf59SPeter Wemm 		error = ENOEXEC;
781e9eabf59SPeter Wemm 		goto out;
782e9eabf59SPeter Wemm 	}
78320947801SPeter Wemm 	shdr = malloc(nbytes, M_LINKER, M_WAITOK);
784e9eabf59SPeter Wemm 	ef->e_shdr = shdr;
785a344babfSGleb Smirnoff 	error = vn_rdwr(UIO_READ, nd->ni_vp, (caddr_t)shdr, nbytes,
786a344babfSGleb Smirnoff 	    hdr->e_shoff, UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred,
787a344babfSGleb Smirnoff 	    NOCRED, &resid, td);
788e9eabf59SPeter Wemm 	if (error)
789e9eabf59SPeter Wemm 		goto out;
790e9eabf59SPeter Wemm 	if (resid) {
791e9eabf59SPeter Wemm 		error = ENOEXEC;
792e9eabf59SPeter Wemm 		goto out;
793e9eabf59SPeter Wemm 	}
794e9eabf59SPeter Wemm 
795e9eabf59SPeter Wemm 	/* Scan the section header for information and table sizing. */
796e9eabf59SPeter Wemm 	nsym = 0;
797e9eabf59SPeter Wemm 	symtabindex = -1;
798e9eabf59SPeter Wemm 	symstrindex = -1;
799e9eabf59SPeter Wemm 	for (i = 0; i < hdr->e_shnum; i++) {
800676799a0SAndriy Gapon 		if (shdr[i].sh_size == 0)
801676799a0SAndriy Gapon 			continue;
802e9eabf59SPeter Wemm 		switch (shdr[i].sh_type) {
803e9eabf59SPeter Wemm 		case SHT_PROGBITS:
804e9eabf59SPeter Wemm 		case SHT_NOBITS:
80513f28d96SKonstantin Belousov #ifdef __amd64__
806b715d9afSKonstantin Belousov 		case SHT_X86_64_UNWIND:
80713f28d96SKonstantin Belousov #endif
8085e6989baSMark Johnston 		case SHT_INIT_ARRAY:
8095e6989baSMark Johnston 		case SHT_FINI_ARRAY:
81058c4aee0SJohn Baldwin 			if ((shdr[i].sh_flags & SHF_ALLOC) == 0)
81158c4aee0SJohn Baldwin 				break;
81220947801SPeter Wemm 			ef->nprogtab++;
813e9eabf59SPeter Wemm 			break;
814e9eabf59SPeter Wemm 		case SHT_SYMTAB:
815e9eabf59SPeter Wemm 			nsym++;
816e9eabf59SPeter Wemm 			symtabindex = i;
817e9eabf59SPeter Wemm 			symstrindex = shdr[i].sh_link;
818e9eabf59SPeter Wemm 			break;
819e9eabf59SPeter Wemm 		case SHT_REL:
82015746ef4SJohn Baldwin 			/*
82115746ef4SJohn Baldwin 			 * Ignore relocation tables for unallocated
82215746ef4SJohn Baldwin 			 * sections.
82315746ef4SJohn Baldwin 			 */
82415746ef4SJohn Baldwin 			if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0)
82515746ef4SJohn Baldwin 				break;
826b474c780SDavid E. O'Brien 			ef->nreltab++;
827e9eabf59SPeter Wemm 			break;
828e9eabf59SPeter Wemm 		case SHT_RELA:
82915746ef4SJohn Baldwin 			if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0)
83015746ef4SJohn Baldwin 				break;
831b474c780SDavid E. O'Brien 			ef->nrelatab++;
832e9eabf59SPeter Wemm 			break;
833e9eabf59SPeter Wemm 		case SHT_STRTAB:
834e9eabf59SPeter Wemm 			break;
835e9eabf59SPeter Wemm 		}
836e9eabf59SPeter Wemm 	}
83720947801SPeter Wemm 	if (ef->nprogtab == 0) {
838552f9f63SEdwin Groothuis 		link_elf_error(filename, "file has no contents");
839e9eabf59SPeter Wemm 		error = ENOEXEC;
840e9eabf59SPeter Wemm 		goto out;
841e9eabf59SPeter Wemm 	}
842e9eabf59SPeter Wemm 	if (nsym != 1) {
843e9eabf59SPeter Wemm 		/* Only allow one symbol table for now */
844f912a970SEd Maste 		link_elf_error(filename,
845f912a970SEd Maste 		    "file must have exactly one symbol table");
846e9eabf59SPeter Wemm 		error = ENOEXEC;
847e9eabf59SPeter Wemm 		goto out;
848e9eabf59SPeter Wemm 	}
849e9eabf59SPeter Wemm 	if (symstrindex < 0 || symstrindex > hdr->e_shnum ||
850e9eabf59SPeter Wemm 	    shdr[symstrindex].sh_type != SHT_STRTAB) {
851552f9f63SEdwin Groothuis 		link_elf_error(filename, "file has invalid symbol strings");
852e9eabf59SPeter Wemm 		error = ENOEXEC;
853e9eabf59SPeter Wemm 		goto out;
854e9eabf59SPeter Wemm 	}
855e9eabf59SPeter Wemm 
856e9eabf59SPeter Wemm 	/* Allocate space for tracking the load chunks */
857e9eabf59SPeter Wemm 	if (ef->nprogtab != 0)
85820947801SPeter Wemm 		ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab),
85920947801SPeter Wemm 		    M_LINKER, M_WAITOK | M_ZERO);
860b474c780SDavid E. O'Brien 	if (ef->nreltab != 0)
861b474c780SDavid E. O'Brien 		ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab),
862b474c780SDavid E. O'Brien 		    M_LINKER, M_WAITOK | M_ZERO);
863b474c780SDavid E. O'Brien 	if (ef->nrelatab != 0)
864b474c780SDavid E. O'Brien 		ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab),
865b474c780SDavid E. O'Brien 		    M_LINKER, M_WAITOK | M_ZERO);
866e9eabf59SPeter Wemm 
8673cfce8e4SKonstantin Belousov 	if (symtabindex == -1) {
8683cfce8e4SKonstantin Belousov 		link_elf_error(filename, "lost symbol table index");
8693cfce8e4SKonstantin Belousov 		error = ENOEXEC;
8703cfce8e4SKonstantin Belousov 		goto out;
8713cfce8e4SKonstantin Belousov 	}
87220947801SPeter Wemm 	/* Allocate space for and load the symbol table */
873e9eabf59SPeter Wemm 	ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym);
874e9eabf59SPeter Wemm 	ef->ddbsymtab = malloc(shdr[symtabindex].sh_size, M_LINKER, M_WAITOK);
875a344babfSGleb Smirnoff 	error = vn_rdwr(UIO_READ, nd->ni_vp, (void *)ef->ddbsymtab,
87620947801SPeter Wemm 	    shdr[symtabindex].sh_size, shdr[symtabindex].sh_offset,
87720947801SPeter Wemm 	    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
87820947801SPeter Wemm 	    &resid, td);
87920947801SPeter Wemm 	if (error)
88020947801SPeter Wemm 		goto out;
88120947801SPeter Wemm 	if (resid != 0){
88220947801SPeter Wemm 		error = EINVAL;
88320947801SPeter Wemm 		goto out;
88420947801SPeter Wemm 	}
885e9eabf59SPeter Wemm 
88620947801SPeter Wemm 	/* Allocate space for and load the symbol strings */
887e9eabf59SPeter Wemm 	ef->ddbstrcnt = shdr[symstrindex].sh_size;
888e9eabf59SPeter Wemm 	ef->ddbstrtab = malloc(shdr[symstrindex].sh_size, M_LINKER, M_WAITOK);
889a344babfSGleb Smirnoff 	error = vn_rdwr(UIO_READ, nd->ni_vp, ef->ddbstrtab,
89020947801SPeter Wemm 	    shdr[symstrindex].sh_size, shdr[symstrindex].sh_offset,
89120947801SPeter Wemm 	    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
89220947801SPeter Wemm 	    &resid, td);
89320947801SPeter Wemm 	if (error)
89420947801SPeter Wemm 		goto out;
89520947801SPeter Wemm 	if (resid != 0){
89620947801SPeter Wemm 		error = EINVAL;
89720947801SPeter Wemm 		goto out;
89820947801SPeter Wemm 	}
899e9eabf59SPeter Wemm 
900e9eabf59SPeter Wemm 	/* Do we have a string table for the section names?  */
901e9eabf59SPeter Wemm 	shstrindex = -1;
90220947801SPeter Wemm 	if (hdr->e_shstrndx != 0 &&
90320947801SPeter Wemm 	    shdr[hdr->e_shstrndx].sh_type == SHT_STRTAB) {
904e9eabf59SPeter Wemm 		shstrindex = hdr->e_shstrndx;
905e9eabf59SPeter Wemm 		ef->shstrcnt = shdr[shstrindex].sh_size;
90620947801SPeter Wemm 		ef->shstrtab = malloc(shdr[shstrindex].sh_size, M_LINKER,
90720947801SPeter Wemm 		    M_WAITOK);
908a344babfSGleb Smirnoff 		error = vn_rdwr(UIO_READ, nd->ni_vp, ef->shstrtab,
90920947801SPeter Wemm 		    shdr[shstrindex].sh_size, shdr[shstrindex].sh_offset,
910e9eabf59SPeter Wemm 		    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
911e9eabf59SPeter Wemm 		    &resid, td);
912e9eabf59SPeter Wemm 		if (error)
913e9eabf59SPeter Wemm 			goto out;
91420947801SPeter Wemm 		if (resid != 0){
91520947801SPeter Wemm 			error = EINVAL;
91620947801SPeter Wemm 			goto out;
91720947801SPeter Wemm 		}
918e9eabf59SPeter Wemm 	}
919e9eabf59SPeter Wemm 
92020947801SPeter Wemm 	/* Size up code/data(progbits) and bss(nobits). */
921e9eabf59SPeter Wemm 	alignmask = 0;
922e9eabf59SPeter Wemm 	for (i = 0; i < hdr->e_shnum; i++) {
923676799a0SAndriy Gapon 		if (shdr[i].sh_size == 0)
924676799a0SAndriy Gapon 			continue;
925e9eabf59SPeter Wemm 		switch (shdr[i].sh_type) {
926e9eabf59SPeter Wemm 		case SHT_PROGBITS:
927e9eabf59SPeter Wemm 		case SHT_NOBITS:
92813f28d96SKonstantin Belousov #ifdef __amd64__
929b715d9afSKonstantin Belousov 		case SHT_X86_64_UNWIND:
93013f28d96SKonstantin Belousov #endif
9315e6989baSMark Johnston 		case SHT_INIT_ARRAY:
9325e6989baSMark Johnston 		case SHT_FINI_ARRAY:
93358c4aee0SJohn Baldwin 			if ((shdr[i].sh_flags & SHF_ALLOC) == 0)
93458c4aee0SJohn Baldwin 				break;
935e9eabf59SPeter Wemm 			alignmask = shdr[i].sh_addralign - 1;
936e9eabf59SPeter Wemm 			mapsize += alignmask;
937e9eabf59SPeter Wemm 			mapsize &= ~alignmask;
938e9eabf59SPeter Wemm 			mapsize += shdr[i].sh_size;
939e9eabf59SPeter Wemm 			break;
940e9eabf59SPeter Wemm 		}
941e9eabf59SPeter Wemm 	}
942e9eabf59SPeter Wemm 
943e9eabf59SPeter Wemm 	/*
944e9eabf59SPeter Wemm 	 * We know how much space we need for the text/data/bss/etc.
945e9eabf59SPeter Wemm 	 * This stuff needs to be in a single chunk so that profiling etc
946e9eabf59SPeter Wemm 	 * can get the bounds and gdb can associate offsets with modules
947e9eabf59SPeter Wemm 	 */
948fbf2a778SKonstantin Belousov 	ef->object = vm_pager_allocate(OBJT_PHYS, NULL, round_page(mapsize),
949fbf2a778SKonstantin Belousov 	    VM_PROT_ALL, 0, thread0.td_ucred);
950fe3db7c7SDoug Rabson 	if (ef->object == NULL) {
951fe3db7c7SDoug Rabson 		error = ENOMEM;
952fe3db7c7SDoug Rabson 		goto out;
953fe3db7c7SDoug Rabson 	}
954d80126a6SMark Johnston #if VM_NRESERVLEVEL > 0
955d80126a6SMark Johnston 	vm_object_color(ef->object, 0);
956d80126a6SMark Johnston #endif
957ac68d1c9SAlan Cox 
958ac68d1c9SAlan Cox 	/*
959ac68d1c9SAlan Cox 	 * In order to satisfy amd64's architectural requirements on the
960ac68d1c9SAlan Cox 	 * location of code and data in the kernel's address space, request a
961ac68d1c9SAlan Cox 	 * mapping that is above the kernel.
9621d9eae9fSMark Johnston 	 *
9631d9eae9fSMark Johnston 	 * Protections will be restricted once relocations are applied.
964ac68d1c9SAlan Cox 	 */
9654104e835SOleksandr Tymoshenko #ifdef __amd64__
966ac68d1c9SAlan Cox 	mapbase = KERNBASE;
9674104e835SOleksandr Tymoshenko #else
9684104e835SOleksandr Tymoshenko 	mapbase = VM_MIN_KERNEL_ADDRESS;
9694104e835SOleksandr Tymoshenko #endif
97020947801SPeter Wemm 	error = vm_map_find(kernel_map, ef->object, 0, &mapbase,
971edb572a3SJohn Baldwin 	    round_page(mapsize), 0, VMFS_OPTIMAL_SPACE, VM_PROT_ALL,
972edb572a3SJohn Baldwin 	    VM_PROT_ALL, 0);
973092bacb2SMark Johnston 	if (error != KERN_SUCCESS) {
974fe3db7c7SDoug Rabson 		vm_object_deallocate(ef->object);
975092bacb2SMark Johnston 		ef->object = NULL;
976092bacb2SMark Johnston 		error = ENOMEM;
977fe3db7c7SDoug Rabson 		goto out;
978fe3db7c7SDoug Rabson 	}
97920947801SPeter Wemm 
980e9eabf59SPeter Wemm 	/* Wire the pages */
9817f1ef325SAlan Cox 	error = vm_map_wire(kernel_map, mapbase,
98220947801SPeter Wemm 	    mapbase + round_page(mapsize),
983e9eabf59SPeter Wemm 	    VM_MAP_WIRE_SYSTEM|VM_MAP_WIRE_NOHOLES);
9847f1ef325SAlan Cox 	if (error != KERN_SUCCESS) {
9857f1ef325SAlan Cox 		error = ENOMEM;
9867f1ef325SAlan Cox 		goto out;
9877f1ef325SAlan Cox 	}
988a4f67738SDoug Rabson 
98920947801SPeter Wemm 	/* Inform the kld system about the situation */
99020947801SPeter Wemm 	lf->address = ef->address = (caddr_t)mapbase;
99120947801SPeter Wemm 	lf->size = mapsize;
992e9eabf59SPeter Wemm 
993e9eabf59SPeter Wemm 	/*
99420947801SPeter Wemm 	 * Now load code/data(progbits), zero bss(nobits), allocate space for
99520947801SPeter Wemm 	 * and load relocs
996e9eabf59SPeter Wemm 	 */
99720947801SPeter Wemm 	pb = 0;
99820947801SPeter Wemm 	rl = 0;
99920947801SPeter Wemm 	ra = 0;
100020947801SPeter Wemm 	alignmask = 0;
100120947801SPeter Wemm 	for (i = 0; i < hdr->e_shnum; i++) {
1002676799a0SAndriy Gapon 		if (shdr[i].sh_size == 0)
1003676799a0SAndriy Gapon 			continue;
100420947801SPeter Wemm 		switch (shdr[i].sh_type) {
100520947801SPeter Wemm 		case SHT_PROGBITS:
100620947801SPeter Wemm 		case SHT_NOBITS:
100713f28d96SKonstantin Belousov #ifdef __amd64__
1008b715d9afSKonstantin Belousov 		case SHT_X86_64_UNWIND:
100913f28d96SKonstantin Belousov #endif
10105e6989baSMark Johnston 		case SHT_INIT_ARRAY:
10115e6989baSMark Johnston 		case SHT_FINI_ARRAY:
101258c4aee0SJohn Baldwin 			if ((shdr[i].sh_flags & SHF_ALLOC) == 0)
101358c4aee0SJohn Baldwin 				break;
101420947801SPeter Wemm 			alignmask = shdr[i].sh_addralign - 1;
101520947801SPeter Wemm 			mapbase += alignmask;
101620947801SPeter Wemm 			mapbase &= ~alignmask;
10170067051fSMarcel Moolenaar 			if (ef->shstrtab != NULL && shdr[i].sh_name != 0) {
101850c202c5SJeff Roberson 				ef->progtab[pb].name =
101950c202c5SJeff Roberson 				    ef->shstrtab + shdr[i].sh_name;
10205e6989baSMark Johnston 				if (!strcmp(ef->progtab[pb].name, ".ctors") ||
10215e6989baSMark Johnston 				    shdr[i].sh_type == SHT_INIT_ARRAY) {
10225e6989baSMark Johnston 					if (lf->ctors_addr != 0) {
10235e6989baSMark Johnston 						printf(
10245e6989baSMark Johnston 				    "%s: multiple ctor sections in %s\n",
10255e6989baSMark Johnston 						    __func__, filename);
10265e6989baSMark Johnston 					} else {
10275e6989baSMark Johnston 						lf->ctors_addr =
10285e6989baSMark Johnston 						    (caddr_t)mapbase;
10295e6989baSMark Johnston 						lf->ctors_size =
10305e6989baSMark Johnston 						    shdr[i].sh_size;
10315e6989baSMark Johnston 					}
10329e575fadSMark Johnston 				} else if (!strcmp(ef->progtab[pb].name,
10339e575fadSMark Johnston 				    ".dtors") ||
10349e575fadSMark Johnston 				    shdr[i].sh_type == SHT_FINI_ARRAY) {
10359e575fadSMark Johnston 					if (lf->dtors_addr != 0) {
10369e575fadSMark Johnston 						printf(
10379e575fadSMark Johnston 				    "%s: multiple dtor sections in %s\n",
10389e575fadSMark Johnston 						    __func__, filename);
10399e575fadSMark Johnston 					} else {
10409e575fadSMark Johnston 						lf->dtors_addr =
10419e575fadSMark Johnston 						    (caddr_t)mapbase;
10429e575fadSMark Johnston 						lf->dtors_size =
10439e575fadSMark Johnston 						    shdr[i].sh_size;
10449e575fadSMark Johnston 					}
10450067051fSMarcel Moolenaar 				}
10460067051fSMarcel Moolenaar 			} else if (shdr[i].sh_type == SHT_PROGBITS)
104720947801SPeter Wemm 				ef->progtab[pb].name = "<<PROGBITS>>";
104813f28d96SKonstantin Belousov #ifdef __amd64__
1049b715d9afSKonstantin Belousov 			else if (shdr[i].sh_type == SHT_X86_64_UNWIND)
105013f28d96SKonstantin Belousov 				ef->progtab[pb].name = "<<UNWIND>>";
105113f28d96SKonstantin Belousov #endif
105250c202c5SJeff Roberson 			else
105350c202c5SJeff Roberson 				ef->progtab[pb].name = "<<NOBITS>>";
105450c202c5SJeff Roberson 			if (ef->progtab[pb].name != NULL &&
10550f823b64SBjoern A. Zeeb 			    !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) {
105650c202c5SJeff Roberson 				ef->progtab[pb].addr =
105750c202c5SJeff Roberson 				    dpcpu_alloc(shdr[i].sh_size);
10580f823b64SBjoern A. Zeeb 				if (ef->progtab[pb].addr == NULL) {
10590f823b64SBjoern A. Zeeb 					printf("%s: pcpu module space is out "
10609afc5684SBjoern A. Zeeb 					    "of space; cannot allocate %#jx "
10619afc5684SBjoern A. Zeeb 					    "for %s\n", __func__,
10629afc5684SBjoern A. Zeeb 					    (uintmax_t)shdr[i].sh_size,
10630f823b64SBjoern A. Zeeb 					    filename);
10640f823b64SBjoern A. Zeeb 				}
10650f823b64SBjoern A. Zeeb 			}
1066eddfbb76SRobert Watson #ifdef VIMAGE
1067eddfbb76SRobert Watson 			else if (ef->progtab[pb].name != NULL &&
10680f823b64SBjoern A. Zeeb 			    !strcmp(ef->progtab[pb].name, VNET_SETNAME)) {
1069eddfbb76SRobert Watson 				ef->progtab[pb].addr =
1070eddfbb76SRobert Watson 				    vnet_data_alloc(shdr[i].sh_size);
10710f823b64SBjoern A. Zeeb 				if (ef->progtab[pb].addr == NULL) {
10720f823b64SBjoern A. Zeeb 					printf("%s: vnet module space is out "
10739afc5684SBjoern A. Zeeb 					    "of space; cannot allocate %#jx "
10749afc5684SBjoern A. Zeeb 					    "for %s\n", __func__,
10759afc5684SBjoern A. Zeeb 					    (uintmax_t)shdr[i].sh_size,
10760f823b64SBjoern A. Zeeb 					    filename);
10770f823b64SBjoern A. Zeeb 				}
10780f823b64SBjoern A. Zeeb 			}
1079eddfbb76SRobert Watson #endif
108050c202c5SJeff Roberson 			else
108150c202c5SJeff Roberson 				ef->progtab[pb].addr =
108250c202c5SJeff Roberson 				    (void *)(uintptr_t)mapbase;
108350c202c5SJeff Roberson 			if (ef->progtab[pb].addr == NULL) {
108450c202c5SJeff Roberson 				error = ENOSPC;
108550c202c5SJeff Roberson 				goto out;
108650c202c5SJeff Roberson 			}
108750c202c5SJeff Roberson 			ef->progtab[pb].size = shdr[i].sh_size;
10881d9eae9fSMark Johnston 			ef->progtab[pb].flags = shdr[i].sh_flags;
108950c202c5SJeff Roberson 			ef->progtab[pb].sec = i;
109013f28d96SKonstantin Belousov 			if (shdr[i].sh_type == SHT_PROGBITS
109113f28d96SKonstantin Belousov #ifdef __amd64__
1092b715d9afSKonstantin Belousov 			    || shdr[i].sh_type == SHT_X86_64_UNWIND
109313f28d96SKonstantin Belousov #endif
109413f28d96SKonstantin Belousov 			    ) {
1095a344babfSGleb Smirnoff 				error = vn_rdwr(UIO_READ, nd->ni_vp,
109620947801SPeter Wemm 				    ef->progtab[pb].addr,
109720947801SPeter Wemm 				    shdr[i].sh_size, shdr[i].sh_offset,
109820947801SPeter Wemm 				    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred,
109920947801SPeter Wemm 				    NOCRED, &resid, td);
110020947801SPeter Wemm 				if (error)
110120947801SPeter Wemm 					goto out;
110220947801SPeter Wemm 				if (resid != 0){
110320947801SPeter Wemm 					error = EINVAL;
110420947801SPeter Wemm 					goto out;
110520947801SPeter Wemm 				}
1106eddfbb76SRobert Watson 				/* Initialize the per-cpu or vnet area. */
1107eddfbb76SRobert Watson 				if (ef->progtab[pb].addr != (void *)mapbase &&
11085f67450dSDimitry Andric 				    !strcmp(ef->progtab[pb].name, DPCPU_SETNAME))
110950c202c5SJeff Roberson 					dpcpu_copy(ef->progtab[pb].addr,
111050c202c5SJeff Roberson 					    shdr[i].sh_size);
1111eddfbb76SRobert Watson #ifdef VIMAGE
1112eddfbb76SRobert Watson 				else if (ef->progtab[pb].addr !=
1113eddfbb76SRobert Watson 				    (void *)mapbase &&
111417ef1febSRobert Watson 				    !strcmp(ef->progtab[pb].name, VNET_SETNAME))
1115eddfbb76SRobert Watson 					vnet_data_copy(ef->progtab[pb].addr,
1116eddfbb76SRobert Watson 					    shdr[i].sh_size);
1117eddfbb76SRobert Watson #endif
111850c202c5SJeff Roberson 			} else
111920947801SPeter Wemm 				bzero(ef->progtab[pb].addr, shdr[i].sh_size);
1120a8774e39SPeter Wemm 
1121a8774e39SPeter Wemm 			/* Update all symbol values with the offset. */
1122a8774e39SPeter Wemm 			for (j = 0; j < ef->ddbsymcnt; j++) {
1123a8774e39SPeter Wemm 				es = &ef->ddbsymtab[j];
1124a8774e39SPeter Wemm 				if (es->st_shndx != i)
1125a8774e39SPeter Wemm 					continue;
1126a8774e39SPeter Wemm 				es->st_value += (Elf_Addr)ef->progtab[pb].addr;
1127a8774e39SPeter Wemm 			}
112820947801SPeter Wemm 			mapbase += shdr[i].sh_size;
112920947801SPeter Wemm 			pb++;
113020947801SPeter Wemm 			break;
113120947801SPeter Wemm 		case SHT_REL:
113215746ef4SJohn Baldwin 			if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0)
113315746ef4SJohn Baldwin 				break;
113420947801SPeter Wemm 			ef->reltab[rl].rel = malloc(shdr[i].sh_size, M_LINKER,
113520947801SPeter Wemm 			    M_WAITOK);
113620947801SPeter Wemm 			ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel);
113720947801SPeter Wemm 			ef->reltab[rl].sec = shdr[i].sh_info;
1138a344babfSGleb Smirnoff 			error = vn_rdwr(UIO_READ, nd->ni_vp,
113920947801SPeter Wemm 			    (void *)ef->reltab[rl].rel,
114020947801SPeter Wemm 			    shdr[i].sh_size, shdr[i].sh_offset,
1141e9eabf59SPeter Wemm 			    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
1142e9eabf59SPeter Wemm 			    &resid, td);
1143e9eabf59SPeter Wemm 			if (error)
1144e9eabf59SPeter Wemm 				goto out;
114520947801SPeter Wemm 			if (resid != 0){
114620947801SPeter Wemm 				error = EINVAL;
114720947801SPeter Wemm 				goto out;
1148e9eabf59SPeter Wemm 			}
114920947801SPeter Wemm 			rl++;
115020947801SPeter Wemm 			break;
115120947801SPeter Wemm 		case SHT_RELA:
115215746ef4SJohn Baldwin 			if ((shdr[shdr[i].sh_info].sh_flags & SHF_ALLOC) == 0)
115315746ef4SJohn Baldwin 				break;
115420947801SPeter Wemm 			ef->relatab[ra].rela = malloc(shdr[i].sh_size, M_LINKER,
115520947801SPeter Wemm 			    M_WAITOK);
115620947801SPeter Wemm 			ef->relatab[ra].nrela =
115720947801SPeter Wemm 			    shdr[i].sh_size / sizeof(Elf_Rela);
115820947801SPeter Wemm 			ef->relatab[ra].sec = shdr[i].sh_info;
1159a344babfSGleb Smirnoff 			error = vn_rdwr(UIO_READ, nd->ni_vp,
116020947801SPeter Wemm 			    (void *)ef->relatab[ra].rela,
116120947801SPeter Wemm 			    shdr[i].sh_size, shdr[i].sh_offset,
1162e9eabf59SPeter Wemm 			    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
1163e9eabf59SPeter Wemm 			    &resid, td);
1164e9eabf59SPeter Wemm 			if (error)
1165e9eabf59SPeter Wemm 				goto out;
116620947801SPeter Wemm 			if (resid != 0){
116720947801SPeter Wemm 				error = EINVAL;
116820947801SPeter Wemm 				goto out;
1169e9eabf59SPeter Wemm 			}
117020947801SPeter Wemm 			ra++;
117120947801SPeter Wemm 			break;
117220947801SPeter Wemm 		}
117320947801SPeter Wemm 	}
11743cfce8e4SKonstantin Belousov 	if (pb != ef->nprogtab) {
11753cfce8e4SKonstantin Belousov 		link_elf_error(filename, "lost progbits");
11763cfce8e4SKonstantin Belousov 		error = ENOEXEC;
11773cfce8e4SKonstantin Belousov 		goto out;
11783cfce8e4SKonstantin Belousov 	}
11793cfce8e4SKonstantin Belousov 	if (rl != ef->nreltab) {
11803cfce8e4SKonstantin Belousov 		link_elf_error(filename, "lost reltab");
11813cfce8e4SKonstantin Belousov 		error = ENOEXEC;
11823cfce8e4SKonstantin Belousov 		goto out;
11833cfce8e4SKonstantin Belousov 	}
11843cfce8e4SKonstantin Belousov 	if (ra != ef->nrelatab) {
11853cfce8e4SKonstantin Belousov 		link_elf_error(filename, "lost relatab");
11863cfce8e4SKonstantin Belousov 		error = ENOEXEC;
11873cfce8e4SKonstantin Belousov 		goto out;
11883cfce8e4SKonstantin Belousov 	}
11893cfce8e4SKonstantin Belousov 	if (mapbase != (vm_offset_t)ef->address + mapsize) {
11903cfce8e4SKonstantin Belousov 		printf(
11913cfce8e4SKonstantin Belousov 		    "%s: mapbase 0x%lx != address %p + mapsize 0x%lx (0x%lx)\n",
11923cfce8e4SKonstantin Belousov 		    filename != NULL ? filename : "<none>",
11936f3c6327SNeel Natu 		    (u_long)mapbase, ef->address, (u_long)mapsize,
11946f3c6327SNeel Natu 		    (u_long)(vm_offset_t)ef->address + mapsize);
11953cfce8e4SKonstantin Belousov 		error = ENOMEM;
11963cfce8e4SKonstantin Belousov 		goto out;
11973cfce8e4SKonstantin Belousov 	}
1198e9eabf59SPeter Wemm 
1199e9eabf59SPeter Wemm 	/* Local intra-module relocations */
1200289ead7cSKonstantin Belousov 	error = link_elf_reloc_local(lf, false);
12013cfce8e4SKonstantin Belousov 	if (error != 0)
12023cfce8e4SKonstantin Belousov 		goto out;
12037251b4bfSJake Burkholder 
1204e9eabf59SPeter Wemm 	/* Pull in dependencies */
1205b249ce48SMateusz Guzik 	VOP_UNLOCK(nd->ni_vp);
12067b9716baSIan Dowse 	error = linker_load_dependencies(lf);
1207a344babfSGleb Smirnoff 	vn_lock(nd->ni_vp, LK_EXCLUSIVE | LK_RETRY);
1208ca65d5c7SPeter Wemm 	if (error)
1209de78ca7eSPeter Wemm 		goto out;
1210e9eabf59SPeter Wemm 
1211e9eabf59SPeter Wemm 	/* External relocations */
1212326e27d8SDoug Rabson 	error = relocate_file(ef);
1213ca65d5c7SPeter Wemm 	if (error)
1214ca65d5c7SPeter Wemm 		goto out;
1215ca65d5c7SPeter Wemm 
1216e9eabf59SPeter Wemm 	/* Notify MD code that a module is being loaded. */
1217e9eabf59SPeter Wemm 	error = elf_cpu_load_file(lf);
1218ca65d5c7SPeter Wemm 	if (error)
1219ca65d5c7SPeter Wemm 		goto out;
1220a4f67738SDoug Rabson 
12218f941950SKonstantin Belousov #if defined(__i386__) || defined(__amd64__)
1222289ead7cSKonstantin Belousov 	/* Now ifuncs. */
1223289ead7cSKonstantin Belousov 	error = link_elf_reloc_local(lf, true);
1224289ead7cSKonstantin Belousov 	if (error != 0)
1225289ead7cSKonstantin Belousov 		goto out;
12268f941950SKonstantin Belousov #endif
1227289ead7cSKonstantin Belousov 
12281d9eae9fSMark Johnston 	link_elf_protect(ef);
12299e575fadSMark Johnston 	link_elf_invoke_cbs(lf->ctors_addr, lf->ctors_size);
1230a4f67738SDoug Rabson 	*result = lf;
1231a4f67738SDoug Rabson 
1232a4f67738SDoug Rabson out:
1233b249ce48SMateusz Guzik 	VOP_UNLOCK(nd->ni_vp);
1234a344babfSGleb Smirnoff 	vn_close(nd->ni_vp, FREAD, td->td_ucred, td);
1235a344babfSGleb Smirnoff 	free(nd, M_TEMP);
1236ca65d5c7SPeter Wemm 	if (error && lf)
123765a311fcSPoul-Henning Kamp 		linker_file_unload(lf, LINKER_UNLOAD_FORCE);
1238e9eabf59SPeter Wemm 	free(hdr, M_LINKER);
1239a4f67738SDoug Rabson 
1240a4f67738SDoug Rabson 	return error;
1241a4f67738SDoug Rabson }
1242a4f67738SDoug Rabson 
1243a4f67738SDoug Rabson static void
1244de78ca7eSPeter Wemm link_elf_unload_file(linker_file_t file)
1245a4f67738SDoug Rabson {
1246326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t) file;
1247d821d364SPedro F. Giffuni 	u_int i;
1248a4f67738SDoug Rabson 
12499e575fadSMark Johnston 	link_elf_invoke_cbs(file->dtors_addr, file->dtors_size);
12509e575fadSMark Johnston 
12511aeb23cdSMarcel Moolenaar 	/* Notify MD code that a module is being unloaded. */
12521aeb23cdSMarcel Moolenaar 	elf_cpu_unload_file(file);
12531aeb23cdSMarcel Moolenaar 
125450c202c5SJeff Roberson 	if (ef->progtab) {
125550c202c5SJeff Roberson 		for (i = 0; i < ef->nprogtab; i++) {
125650c202c5SJeff Roberson 			if (ef->progtab[i].size == 0)
125750c202c5SJeff Roberson 				continue;
125850c202c5SJeff Roberson 			if (ef->progtab[i].name == NULL)
125950c202c5SJeff Roberson 				continue;
12605f67450dSDimitry Andric 			if (!strcmp(ef->progtab[i].name, DPCPU_SETNAME))
126150c202c5SJeff Roberson 				dpcpu_free(ef->progtab[i].addr,
126250c202c5SJeff Roberson 				    ef->progtab[i].size);
1263eddfbb76SRobert Watson #ifdef VIMAGE
126417ef1febSRobert Watson 			else if (!strcmp(ef->progtab[i].name, VNET_SETNAME))
1265eddfbb76SRobert Watson 				vnet_data_free(ef->progtab[i].addr,
1266eddfbb76SRobert Watson 				    ef->progtab[i].size);
1267eddfbb76SRobert Watson #endif
126850c202c5SJeff Roberson 		}
126950c202c5SJeff Roberson 	}
127070b7ffeeSIan Dowse 	if (ef->preloaded) {
127170b7ffeeSIan Dowse 		free(ef->reltab, M_LINKER);
127270b7ffeeSIan Dowse 		free(ef->relatab, M_LINKER);
127370b7ffeeSIan Dowse 		free(ef->progtab, M_LINKER);
1274a2024a3eSJohn Birrell 		free(ef->ctftab, M_LINKER);
1275a2024a3eSJohn Birrell 		free(ef->ctfoff, M_LINKER);
1276a2024a3eSJohn Birrell 		free(ef->typoff, M_LINKER);
127773624a80SMark Johnston 		if (file->pathname != NULL)
127873624a80SMark Johnston 			preload_delete_name(file->pathname);
127970b7ffeeSIan Dowse 		return;
128070b7ffeeSIan Dowse 	}
128170b7ffeeSIan Dowse 
1282b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nreltab; i++)
128320947801SPeter Wemm 		free(ef->reltab[i].rel, M_LINKER);
1284b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nrelatab; i++)
128520947801SPeter Wemm 		free(ef->relatab[i].rela, M_LINKER);
128620947801SPeter Wemm 	free(ef->reltab, M_LINKER);
128720947801SPeter Wemm 	free(ef->relatab, M_LINKER);
128820947801SPeter Wemm 	free(ef->progtab, M_LINKER);
128920947801SPeter Wemm 
1290092bacb2SMark Johnston 	if (ef->object != NULL)
1291fe3db7c7SDoug Rabson 		vm_map_remove(kernel_map, (vm_offset_t)ef->address,
1292092bacb2SMark Johnston 		    (vm_offset_t)ef->address + ptoa(ef->object->size));
1293e9eabf59SPeter Wemm 	free(ef->e_shdr, M_LINKER);
1294e9eabf59SPeter Wemm 	free(ef->ddbsymtab, M_LINKER);
1295e9eabf59SPeter Wemm 	free(ef->ddbstrtab, M_LINKER);
129620947801SPeter Wemm 	free(ef->shstrtab, M_LINKER);
1297a2024a3eSJohn Birrell 	free(ef->ctftab, M_LINKER);
1298a2024a3eSJohn Birrell 	free(ef->ctfoff, M_LINKER);
1299a2024a3eSJohn Birrell 	free(ef->typoff, M_LINKER);
1300de78ca7eSPeter Wemm }
1301de78ca7eSPeter Wemm 
1302fe3db7c7SDoug Rabson static const char *
1303757686b1SMarcel Moolenaar symbol_name(elf_file_t ef, Elf_Size r_info)
1304a4f67738SDoug Rabson {
1305fe3db7c7SDoug Rabson 	const Elf_Sym *ref;
1306a4f67738SDoug Rabson 
1307aa855a59SPeter Wemm 	if (ELF_R_SYM(r_info)) {
1308e9eabf59SPeter Wemm 		ref = ef->ddbsymtab + ELF_R_SYM(r_info);
1309e9eabf59SPeter Wemm 		return ef->ddbstrtab + ref->st_name;
1310fe3db7c7SDoug Rabson 	} else
1311fe3db7c7SDoug Rabson 		return NULL;
1312a4f67738SDoug Rabson }
1313a4f67738SDoug Rabson 
1314e9eabf59SPeter Wemm static Elf_Addr
1315e9eabf59SPeter Wemm findbase(elf_file_t ef, int sec)
1316e9eabf59SPeter Wemm {
1317e9eabf59SPeter Wemm 	int i;
1318e9eabf59SPeter Wemm 	Elf_Addr base = 0;
1319e9eabf59SPeter Wemm 
1320e9eabf59SPeter Wemm 	for (i = 0; i < ef->nprogtab; i++) {
1321a8774e39SPeter Wemm 		if (sec == ef->progtab[i].sec) {
1322e9eabf59SPeter Wemm 			base = (Elf_Addr)ef->progtab[i].addr;
1323a8774e39SPeter Wemm 			break;
1324e9eabf59SPeter Wemm 		}
1325a8774e39SPeter Wemm 	}
1326e9eabf59SPeter Wemm 	return base;
1327e9eabf59SPeter Wemm }
1328e9eabf59SPeter Wemm 
1329a4f67738SDoug Rabson static int
1330b11e6fd7SMark Johnston relocate_file1(elf_file_t ef, bool ifuncs)
1331a4f67738SDoug Rabson {
1332fe3db7c7SDoug Rabson 	const Elf_Rel *rellim;
1333fe3db7c7SDoug Rabson 	const Elf_Rel *rel;
1334fe3db7c7SDoug Rabson 	const Elf_Rela *relalim;
1335fe3db7c7SDoug Rabson 	const Elf_Rela *rela;
1336aa855a59SPeter Wemm 	const char *symname;
1337e9eabf59SPeter Wemm 	const Elf_Sym *sym;
1338e9eabf59SPeter Wemm 	int i;
1339757686b1SMarcel Moolenaar 	Elf_Size symidx;
1340e9eabf59SPeter Wemm 	Elf_Addr base;
1341e9eabf59SPeter Wemm 
1342fe3db7c7SDoug Rabson 	/* Perform relocations without addend if there are any: */
1343b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nreltab; i++) {
1344e9eabf59SPeter Wemm 		rel = ef->reltab[i].rel;
13453cfce8e4SKonstantin Belousov 		if (rel == NULL) {
13463cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost a reltab!");
13473cfce8e4SKonstantin Belousov 			return (ENOEXEC);
13483cfce8e4SKonstantin Belousov 		}
134920947801SPeter Wemm 		rellim = rel + ef->reltab[i].nrel;
1350e9eabf59SPeter Wemm 		base = findbase(ef, ef->reltab[i].sec);
13513cfce8e4SKonstantin Belousov 		if (base == 0) {
13523cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost base for reltab");
13533cfce8e4SKonstantin Belousov 			return (ENOEXEC);
13543cfce8e4SKonstantin Belousov 		}
135520947801SPeter Wemm 		for ( ; rel < rellim; rel++) {
1356e9eabf59SPeter Wemm 			symidx = ELF_R_SYM(rel->r_info);
135720947801SPeter Wemm 			if (symidx >= ef->ddbsymcnt)
135820947801SPeter Wemm 				continue;
1359e9eabf59SPeter Wemm 			sym = ef->ddbsymtab + symidx;
136020947801SPeter Wemm 			/* Local relocs are already done */
136120947801SPeter Wemm 			if (ELF_ST_BIND(sym->st_info) == STB_LOCAL)
136220947801SPeter Wemm 				continue;
1363b11e6fd7SMark Johnston 			if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC ||
1364b11e6fd7SMark Johnston 			    elf_is_ifunc_reloc(rel->r_info)) != ifuncs)
1365b11e6fd7SMark Johnston 				continue;
136620947801SPeter Wemm 			if (elf_reloc(&ef->lf, base, rel, ELF_RELOC_REL,
136720947801SPeter Wemm 			    elf_obj_lookup)) {
1368aa855a59SPeter Wemm 				symname = symbol_name(ef, rel->r_info);
136920947801SPeter Wemm 				printf("link_elf_obj: symbol %s undefined\n",
137020947801SPeter Wemm 				    symname);
13713cfce8e4SKonstantin Belousov 				return (ENOENT);
1372a13ddfb6SPeter Wemm 			}
1373aa855a59SPeter Wemm 		}
1374a4f67738SDoug Rabson 	}
1375a4f67738SDoug Rabson 
1376fe3db7c7SDoug Rabson 	/* Perform relocations with addend if there are any: */
1377b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nrelatab; i++) {
1378e9eabf59SPeter Wemm 		rela = ef->relatab[i].rela;
13793cfce8e4SKonstantin Belousov 		if (rela == NULL) {
13803cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost a relatab!");
13813cfce8e4SKonstantin Belousov 			return (ENOEXEC);
13823cfce8e4SKonstantin Belousov 		}
138320947801SPeter Wemm 		relalim = rela + ef->relatab[i].nrela;
1384e9eabf59SPeter Wemm 		base = findbase(ef, ef->relatab[i].sec);
13853cfce8e4SKonstantin Belousov 		if (base == 0) {
13863cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename,
13873cfce8e4SKonstantin Belousov 			    "lost base for relatab");
13883cfce8e4SKonstantin Belousov 			return (ENOEXEC);
13893cfce8e4SKonstantin Belousov 		}
139020947801SPeter Wemm 		for ( ; rela < relalim; rela++) {
1391e9eabf59SPeter Wemm 			symidx = ELF_R_SYM(rela->r_info);
139220947801SPeter Wemm 			if (symidx >= ef->ddbsymcnt)
139320947801SPeter Wemm 				continue;
1394e9eabf59SPeter Wemm 			sym = ef->ddbsymtab + symidx;
139520947801SPeter Wemm 			/* Local relocs are already done */
139620947801SPeter Wemm 			if (ELF_ST_BIND(sym->st_info) == STB_LOCAL)
139720947801SPeter Wemm 				continue;
1398b11e6fd7SMark Johnston 			if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC ||
1399b11e6fd7SMark Johnston 			    elf_is_ifunc_reloc(rela->r_info)) != ifuncs)
1400b11e6fd7SMark Johnston 				continue;
140120947801SPeter Wemm 			if (elf_reloc(&ef->lf, base, rela, ELF_RELOC_RELA,
140220947801SPeter Wemm 			    elf_obj_lookup)) {
1403aa855a59SPeter Wemm 				symname = symbol_name(ef, rela->r_info);
140420947801SPeter Wemm 				printf("link_elf_obj: symbol %s undefined\n",
140520947801SPeter Wemm 				    symname);
14063cfce8e4SKonstantin Belousov 				return (ENOENT);
1407a13ddfb6SPeter Wemm 			}
1408e9eabf59SPeter Wemm 		}
1409e9eabf59SPeter Wemm 	}
1410a4f67738SDoug Rabson 
14112832cd54SKonstantin Belousov 	/*
1412e3043798SPedro F. Giffuni 	 * Only clean SHN_FBSD_CACHED for successful return.  If we
14132832cd54SKonstantin Belousov 	 * modified symbol table for the object but found an
14142832cd54SKonstantin Belousov 	 * unresolved symbol, there is no reason to roll back.
14152832cd54SKonstantin Belousov 	 */
14162832cd54SKonstantin Belousov 	elf_obj_cleanup_globals_cache(ef);
14172832cd54SKonstantin Belousov 
14183cfce8e4SKonstantin Belousov 	return (0);
1419a4f67738SDoug Rabson }
1420a4f67738SDoug Rabson 
142137c84183SPoul-Henning Kamp static int
1422b11e6fd7SMark Johnston relocate_file(elf_file_t ef)
1423b11e6fd7SMark Johnston {
1424b11e6fd7SMark Johnston 	int error;
1425b11e6fd7SMark Johnston 
1426b11e6fd7SMark Johnston 	error = relocate_file1(ef, false);
1427b11e6fd7SMark Johnston 	if (error == 0)
1428b11e6fd7SMark Johnston 		error = relocate_file1(ef, true);
1429b11e6fd7SMark Johnston 	return (error);
1430b11e6fd7SMark Johnston }
1431b11e6fd7SMark Johnston 
1432b11e6fd7SMark Johnston static int
1433*95c20fafSKonstantin Belousov link_elf_lookup_symbol1(linker_file_t lf, const char *name, c_linker_sym_t *sym,
1434*95c20fafSKonstantin Belousov     bool see_local)
1435a4f67738SDoug Rabson {
1436326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t)lf;
14372d636ab0SPeter Wemm 	const Elf_Sym *symp;
14382d636ab0SPeter Wemm 	const char *strp;
1439a4f67738SDoug Rabson 	int i;
1440a4f67738SDoug Rabson 
14412d636ab0SPeter Wemm 	for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
14422d636ab0SPeter Wemm 		strp = ef->ddbstrtab + symp->st_name;
1443a8774e39SPeter Wemm 		if (symp->st_shndx != SHN_UNDEF && strcmp(name, strp) == 0) {
1444*95c20fafSKonstantin Belousov 			if (see_local ||
1445*95c20fafSKonstantin Belousov 			    ELF_ST_BIND(symp->st_info) == STB_GLOBAL) {
1446d254af07SMatthew Dillon 				*sym = (c_linker_sym_t) symp;
1447*95c20fafSKonstantin Belousov 				return (0);
1448*95c20fafSKonstantin Belousov 			}
1449*95c20fafSKonstantin Belousov 			return (ENOENT);
14502d636ab0SPeter Wemm 		}
14512d636ab0SPeter Wemm 	}
1452a7e4eb14SKonstantin Belousov 	return (ENOENT);
1453a4f67738SDoug Rabson }
1454a4f67738SDoug Rabson 
1455de78ca7eSPeter Wemm static int
1456*95c20fafSKonstantin Belousov link_elf_lookup_symbol(linker_file_t lf, const char *name, c_linker_sym_t *sym)
1457*95c20fafSKonstantin Belousov {
1458*95c20fafSKonstantin Belousov 	return (link_elf_lookup_symbol1(lf, name, sym, false));
1459*95c20fafSKonstantin Belousov }
1460*95c20fafSKonstantin Belousov 
1461*95c20fafSKonstantin Belousov static int
1462*95c20fafSKonstantin Belousov link_elf_lookup_debug_symbol(linker_file_t lf, const char *name,
1463*95c20fafSKonstantin Belousov     c_linker_sym_t *sym)
1464*95c20fafSKonstantin Belousov {
1465*95c20fafSKonstantin Belousov 	return (link_elf_lookup_symbol1(lf, name, sym, true));
1466*95c20fafSKonstantin Belousov }
1467*95c20fafSKonstantin Belousov 
1468*95c20fafSKonstantin Belousov static int
1469*95c20fafSKonstantin Belousov link_elf_symbol_values1(linker_file_t lf, c_linker_sym_t sym,
1470*95c20fafSKonstantin Belousov     linker_symval_t *symval, bool see_local)
1471a4f67738SDoug Rabson {
14727035cf14SKonstantin Belousov 	elf_file_t ef;
14737035cf14SKonstantin Belousov 	const Elf_Sym *es;
14747035cf14SKonstantin Belousov 	caddr_t val;
1475a4f67738SDoug Rabson 
14767035cf14SKonstantin Belousov 	ef = (elf_file_t) lf;
14777035cf14SKonstantin Belousov 	es = (const Elf_Sym*) sym;
14787035cf14SKonstantin Belousov 	val = (caddr_t)es->st_value;
14795cf87418SMarcel Moolenaar 	if (es >= ef->ddbsymtab && es < (ef->ddbsymtab + ef->ddbsymcnt)) {
1480*95c20fafSKonstantin Belousov 		if (!see_local && ELF_ST_BIND(es->st_info) == STB_LOCAL)
1481*95c20fafSKonstantin Belousov 			return (ENOENT);
14822d636ab0SPeter Wemm 		symval->name = ef->ddbstrtab + es->st_name;
14837035cf14SKonstantin Belousov 		val = (caddr_t)es->st_value;
14847035cf14SKonstantin Belousov 		if (ELF_ST_TYPE(es->st_info) == STT_GNU_IFUNC)
14857035cf14SKonstantin Belousov 			val = ((caddr_t (*)(void))val)();
14867035cf14SKonstantin Belousov 		symval->value = val;
14872d636ab0SPeter Wemm 		symval->size = es->st_size;
1488a7e4eb14SKonstantin Belousov 		return (0);
14892d636ab0SPeter Wemm 	}
1490a7e4eb14SKonstantin Belousov 	return (ENOENT);
14912d636ab0SPeter Wemm }
1492a4f67738SDoug Rabson 
1493a4f67738SDoug Rabson static int
1494*95c20fafSKonstantin Belousov link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym,
1495*95c20fafSKonstantin Belousov     linker_symval_t *symval)
1496*95c20fafSKonstantin Belousov {
1497*95c20fafSKonstantin Belousov 	return (link_elf_symbol_values1(lf, sym, symval, false));
1498*95c20fafSKonstantin Belousov }
1499*95c20fafSKonstantin Belousov 
1500*95c20fafSKonstantin Belousov static int
1501*95c20fafSKonstantin Belousov link_elf_debug_symbol_values(linker_file_t lf, c_linker_sym_t sym,
1502*95c20fafSKonstantin Belousov     linker_symval_t *symval)
1503*95c20fafSKonstantin Belousov {
1504*95c20fafSKonstantin Belousov 	return (link_elf_symbol_values1(lf, sym, symval, true));
1505*95c20fafSKonstantin Belousov }
1506*95c20fafSKonstantin Belousov 
1507*95c20fafSKonstantin Belousov static int
1508a4f67738SDoug Rabson link_elf_search_symbol(linker_file_t lf, caddr_t value,
1509d254af07SMatthew Dillon     c_linker_sym_t *sym, long *diffp)
1510a4f67738SDoug Rabson {
1511326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t)lf;
1512586453feSBruce Evans 	u_long off = (uintptr_t)(void *)value;
1513a4f67738SDoug Rabson 	u_long diff = off;
1514586453feSBruce Evans 	u_long st_value;
1515fe3db7c7SDoug Rabson 	const Elf_Sym *es;
1516b85f65afSPedro F. Giffuni 	const Elf_Sym *best = NULL;
1517a4f67738SDoug Rabson 	int i;
1518a4f67738SDoug Rabson 
15192d636ab0SPeter Wemm 	for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) {
1520a4f67738SDoug Rabson 		if (es->st_name == 0)
1521a4f67738SDoug Rabson 			continue;
1522a8774e39SPeter Wemm 		st_value = es->st_value;
1523b5abfb70SPeter Wemm 		if (off >= st_value) {
1524b5abfb70SPeter Wemm 			if (off - st_value < diff) {
1525b5abfb70SPeter Wemm 				diff = off - st_value;
1526a4f67738SDoug Rabson 				best = es;
1527a4f67738SDoug Rabson 				if (diff == 0)
1528a4f67738SDoug Rabson 					break;
1529b5abfb70SPeter Wemm 			} else if (off - st_value == diff) {
1530a4f67738SDoug Rabson 				best = es;
1531a4f67738SDoug Rabson 			}
1532a4f67738SDoug Rabson 		}
1533a4f67738SDoug Rabson 	}
1534b85f65afSPedro F. Giffuni 	if (best == NULL)
1535a4f67738SDoug Rabson 		*diffp = off;
1536a4f67738SDoug Rabson 	else
1537a4f67738SDoug Rabson 		*diffp = diff;
1538d254af07SMatthew Dillon 	*sym = (c_linker_sym_t) best;
1539a4f67738SDoug Rabson 
1540a7e4eb14SKonstantin Belousov 	return (0);
1541a4f67738SDoug Rabson }
1542f41325dbSPeter Wemm 
1543f41325dbSPeter Wemm /*
1544f41325dbSPeter Wemm  * Look up a linker set on an ELF system.
1545f41325dbSPeter Wemm  */
1546f41325dbSPeter Wemm static int
1547f41325dbSPeter Wemm link_elf_lookup_set(linker_file_t lf, const char *name,
1548f41325dbSPeter Wemm     void ***startp, void ***stopp, int *countp)
1549f41325dbSPeter Wemm {
1550e9eabf59SPeter Wemm 	elf_file_t ef = (elf_file_t)lf;
1551f41325dbSPeter Wemm 	void **start, **stop;
1552e9eabf59SPeter Wemm 	int i, count;
1553f41325dbSPeter Wemm 
1554e9eabf59SPeter Wemm 	/* Relative to section number */
1555e9eabf59SPeter Wemm 	for (i = 0; i < ef->nprogtab; i++) {
1556e9eabf59SPeter Wemm 		if ((strncmp(ef->progtab[i].name, "set_", 4) == 0) &&
1557e9eabf59SPeter Wemm 		    strcmp(ef->progtab[i].name + 4, name) == 0) {
1558e9eabf59SPeter Wemm 			start  = (void **)ef->progtab[i].addr;
155920947801SPeter Wemm 			stop = (void **)((char *)ef->progtab[i].addr +
156020947801SPeter Wemm 			    ef->progtab[i].size);
1561f41325dbSPeter Wemm 			count = stop - start;
1562f41325dbSPeter Wemm 			if (startp)
1563f41325dbSPeter Wemm 				*startp = start;
1564f41325dbSPeter Wemm 			if (stopp)
1565f41325dbSPeter Wemm 				*stopp = stop;
1566f41325dbSPeter Wemm 			if (countp)
1567f41325dbSPeter Wemm 				*countp = count;
1568e9eabf59SPeter Wemm 			return (0);
1569e9eabf59SPeter Wemm 		}
1570e9eabf59SPeter Wemm 	}
1571e9eabf59SPeter Wemm 	return (ESRCH);
1572f41325dbSPeter Wemm }
1573bb9fe9ddSBrian Feldman 
1574bb9fe9ddSBrian Feldman static int
1575bb9fe9ddSBrian Feldman link_elf_each_function_name(linker_file_t file,
1576e9eabf59SPeter Wemm     int (*callback)(const char *, void *), void *opaque)
1577e9eabf59SPeter Wemm {
1578bb9fe9ddSBrian Feldman 	elf_file_t ef = (elf_file_t)file;
1579bb9fe9ddSBrian Feldman 	const Elf_Sym *symp;
1580bb9fe9ddSBrian Feldman 	int i, error;
1581bb9fe9ddSBrian Feldman 
1582bb9fe9ddSBrian Feldman 	/* Exhaustive search */
1583bb9fe9ddSBrian Feldman 	for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
1584bb9fe9ddSBrian Feldman 		if (symp->st_value != 0 &&
15857035cf14SKonstantin Belousov 		    (ELF_ST_TYPE(symp->st_info) == STT_FUNC ||
15867035cf14SKonstantin Belousov 		    ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) {
1587bb9fe9ddSBrian Feldman 			error = callback(ef->ddbstrtab + symp->st_name, opaque);
1588bb9fe9ddSBrian Feldman 			if (error)
1589bb9fe9ddSBrian Feldman 				return (error);
1590bb9fe9ddSBrian Feldman 		}
1591bb9fe9ddSBrian Feldman 	}
1592bb9fe9ddSBrian Feldman 	return (0);
1593bb9fe9ddSBrian Feldman }
159484201059SMarcel Moolenaar 
1595a2024a3eSJohn Birrell static int
1596a2024a3eSJohn Birrell link_elf_each_function_nameval(linker_file_t file,
1597a2024a3eSJohn Birrell     linker_function_nameval_callback_t callback, void *opaque)
1598a2024a3eSJohn Birrell {
1599a2024a3eSJohn Birrell 	linker_symval_t symval;
1600a2024a3eSJohn Birrell 	elf_file_t ef = (elf_file_t)file;
1601a2024a3eSJohn Birrell 	const Elf_Sym *symp;
1602a2024a3eSJohn Birrell 	int i, error;
1603a2024a3eSJohn Birrell 
1604a2024a3eSJohn Birrell 	/* Exhaustive search */
1605a2024a3eSJohn Birrell 	for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
1606a2024a3eSJohn Birrell 		if (symp->st_value != 0 &&
16077035cf14SKonstantin Belousov 		    (ELF_ST_TYPE(symp->st_info) == STT_FUNC ||
16087035cf14SKonstantin Belousov 		    ELF_ST_TYPE(symp->st_info) == STT_GNU_IFUNC)) {
1609*95c20fafSKonstantin Belousov 			error = link_elf_debug_symbol_values(file,
16107035cf14SKonstantin Belousov 			    (c_linker_sym_t)symp, &symval);
1611*95c20fafSKonstantin Belousov 			if (error == 0)
1612a2024a3eSJohn Birrell 				error = callback(file, i, &symval, opaque);
1613*95c20fafSKonstantin Belousov 			if (error != 0)
1614a2024a3eSJohn Birrell 				return (error);
1615a2024a3eSJohn Birrell 		}
1616a2024a3eSJohn Birrell 	}
1617a2024a3eSJohn Birrell 	return (0);
1618a2024a3eSJohn Birrell }
1619a2024a3eSJohn Birrell 
16202832cd54SKonstantin Belousov static void
16212832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(elf_file_t ef)
16222832cd54SKonstantin Belousov {
16232832cd54SKonstantin Belousov 	Elf_Sym *sym;
16242832cd54SKonstantin Belousov 	Elf_Size i;
16252832cd54SKonstantin Belousov 
16262832cd54SKonstantin Belousov 	for (i = 0; i < ef->ddbsymcnt; i++) {
16272832cd54SKonstantin Belousov 		sym = ef->ddbsymtab + i;
16282832cd54SKonstantin Belousov 		if (sym->st_shndx == SHN_FBSD_CACHED) {
16292832cd54SKonstantin Belousov 			sym->st_shndx = SHN_UNDEF;
16302832cd54SKonstantin Belousov 			sym->st_value = 0;
16312832cd54SKonstantin Belousov 		}
16322832cd54SKonstantin Belousov 	}
16332832cd54SKonstantin Belousov }
16342832cd54SKonstantin Belousov 
1635d297ad16SMarcel Moolenaar /*
1636d297ad16SMarcel Moolenaar  * Symbol lookup function that can be used when the symbol index is known (ie
1637d297ad16SMarcel Moolenaar  * in relocations). It uses the symbol index instead of doing a fully fledged
1638d297ad16SMarcel Moolenaar  * hash table based lookup when such is valid. For example for local symbols.
1639d297ad16SMarcel Moolenaar  * This is not only more efficient, it's also more correct. It's not always
1640d297ad16SMarcel Moolenaar  * the case that the symbol can be found through the hash table.
1641d297ad16SMarcel Moolenaar  */
1642cff8c6f2SKonstantin Belousov static int
1643cff8c6f2SKonstantin Belousov elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, Elf_Addr *res)
1644d297ad16SMarcel Moolenaar {
1645d297ad16SMarcel Moolenaar 	elf_file_t ef = (elf_file_t)lf;
16462832cd54SKonstantin Belousov 	Elf_Sym *sym;
1647d297ad16SMarcel Moolenaar 	const char *symbol;
1648cff8c6f2SKonstantin Belousov 	Elf_Addr res1;
1649d297ad16SMarcel Moolenaar 
1650d297ad16SMarcel Moolenaar 	/* Don't even try to lookup the symbol if the index is bogus. */
1651cff8c6f2SKonstantin Belousov 	if (symidx >= ef->ddbsymcnt) {
1652cff8c6f2SKonstantin Belousov 		*res = 0;
1653cff8c6f2SKonstantin Belousov 		return (EINVAL);
1654cff8c6f2SKonstantin Belousov 	}
1655d297ad16SMarcel Moolenaar 
1656e9eabf59SPeter Wemm 	sym = ef->ddbsymtab + symidx;
1657d297ad16SMarcel Moolenaar 
16584cec6f5dSPeter Wemm 	/* Quick answer if there is a definition included. */
1659cff8c6f2SKonstantin Belousov 	if (sym->st_shndx != SHN_UNDEF) {
1660289ead7cSKonstantin Belousov 		res1 = (Elf_Addr)sym->st_value;
1661289ead7cSKonstantin Belousov 		if (ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC)
1662289ead7cSKonstantin Belousov 			res1 = ((Elf_Addr (*)(void))res1)();
1663289ead7cSKonstantin Belousov 		*res = res1;
1664cff8c6f2SKonstantin Belousov 		return (0);
1665cff8c6f2SKonstantin Belousov 	}
16664cec6f5dSPeter Wemm 
16674cec6f5dSPeter Wemm 	/* If we get here, then it is undefined and needs a lookup. */
16684cec6f5dSPeter Wemm 	switch (ELF_ST_BIND(sym->st_info)) {
16694cec6f5dSPeter Wemm 	case STB_LOCAL:
16704cec6f5dSPeter Wemm 		/* Local, but undefined? huh? */
1671cff8c6f2SKonstantin Belousov 		*res = 0;
1672cff8c6f2SKonstantin Belousov 		return (EINVAL);
1673d297ad16SMarcel Moolenaar 
1674e9eabf59SPeter Wemm 	case STB_GLOBAL:
1675cff8c6f2SKonstantin Belousov 	case STB_WEAK:
1676e9eabf59SPeter Wemm 		/* Relative to Data or Function name */
1677e9eabf59SPeter Wemm 		symbol = ef->ddbstrtab + sym->st_name;
1678d297ad16SMarcel Moolenaar 
1679d297ad16SMarcel Moolenaar 		/* Force a lookup failure if the symbol name is bogus. */
1680cff8c6f2SKonstantin Belousov 		if (*symbol == 0) {
1681cff8c6f2SKonstantin Belousov 			*res = 0;
1682cff8c6f2SKonstantin Belousov 			return (EINVAL);
1683cff8c6f2SKonstantin Belousov 		}
1684cff8c6f2SKonstantin Belousov 		res1 = (Elf_Addr)linker_file_lookup_symbol(lf, symbol, deps);
16852832cd54SKonstantin Belousov 
16862832cd54SKonstantin Belousov 		/*
16872832cd54SKonstantin Belousov 		 * Cache global lookups during module relocation. The failure
16882832cd54SKonstantin Belousov 		 * case is particularly expensive for callers, who must scan
16892832cd54SKonstantin Belousov 		 * through the entire globals table doing strcmp(). Cache to
16902832cd54SKonstantin Belousov 		 * avoid doing such work repeatedly.
16912832cd54SKonstantin Belousov 		 *
16922832cd54SKonstantin Belousov 		 * After relocation is complete, undefined globals will be
16932832cd54SKonstantin Belousov 		 * restored to SHN_UNDEF in elf_obj_cleanup_globals_cache(),
16942832cd54SKonstantin Belousov 		 * above.
16952832cd54SKonstantin Belousov 		 */
1696cff8c6f2SKonstantin Belousov 		if (res1 != 0) {
16972832cd54SKonstantin Belousov 			sym->st_shndx = SHN_FBSD_CACHED;
1698cff8c6f2SKonstantin Belousov 			sym->st_value = res1;
1699cff8c6f2SKonstantin Belousov 			*res = res1;
1700e9eabf59SPeter Wemm 			return (0);
1701cff8c6f2SKonstantin Belousov 		} else if (ELF_ST_BIND(sym->st_info) == STB_WEAK) {
1702cff8c6f2SKonstantin Belousov 			sym->st_value = 0;
1703cff8c6f2SKonstantin Belousov 			*res = 0;
1704cff8c6f2SKonstantin Belousov 			return (0);
1705cff8c6f2SKonstantin Belousov 		}
1706cff8c6f2SKonstantin Belousov 		return (EINVAL);
1707e9eabf59SPeter Wemm 
1708e9eabf59SPeter Wemm 	default:
1709cff8c6f2SKonstantin Belousov 		return (EINVAL);
1710d297ad16SMarcel Moolenaar 	}
1711e9eabf59SPeter Wemm }
1712e9eabf59SPeter Wemm 
17137251b4bfSJake Burkholder static void
17147226306eSKonstantin Belousov link_elf_fix_link_set(elf_file_t ef)
17157226306eSKonstantin Belousov {
17167226306eSKonstantin Belousov 	static const char startn[] = "__start_";
17177226306eSKonstantin Belousov 	static const char stopn[] = "__stop_";
17187226306eSKonstantin Belousov 	Elf_Sym *sym;
17197226306eSKonstantin Belousov 	const char *sym_name, *linkset_name;
17207226306eSKonstantin Belousov 	Elf_Addr startp, stopp;
17217226306eSKonstantin Belousov 	Elf_Size symidx;
17227226306eSKonstantin Belousov 	int start, i;
17237226306eSKonstantin Belousov 
17247226306eSKonstantin Belousov 	startp = stopp = 0;
17257226306eSKonstantin Belousov 	for (symidx = 1 /* zero entry is special */;
17267226306eSKonstantin Belousov 		symidx < ef->ddbsymcnt; symidx++) {
17277226306eSKonstantin Belousov 		sym = ef->ddbsymtab + symidx;
17287226306eSKonstantin Belousov 		if (sym->st_shndx != SHN_UNDEF)
17297226306eSKonstantin Belousov 			continue;
17307226306eSKonstantin Belousov 
17317226306eSKonstantin Belousov 		sym_name = ef->ddbstrtab + sym->st_name;
17327226306eSKonstantin Belousov 		if (strncmp(sym_name, startn, sizeof(startn) - 1) == 0) {
17337226306eSKonstantin Belousov 			start = 1;
17347226306eSKonstantin Belousov 			linkset_name = sym_name + sizeof(startn) - 1;
17357226306eSKonstantin Belousov 		}
17367226306eSKonstantin Belousov 		else if (strncmp(sym_name, stopn, sizeof(stopn) - 1) == 0) {
17377226306eSKonstantin Belousov 			start = 0;
17387226306eSKonstantin Belousov 			linkset_name = sym_name + sizeof(stopn) - 1;
17397226306eSKonstantin Belousov 		}
17407226306eSKonstantin Belousov 		else
17417226306eSKonstantin Belousov 			continue;
17427226306eSKonstantin Belousov 
17437226306eSKonstantin Belousov 		for (i = 0; i < ef->nprogtab; i++) {
17447226306eSKonstantin Belousov 			if (strcmp(ef->progtab[i].name, linkset_name) == 0) {
17457226306eSKonstantin Belousov 				startp = (Elf_Addr)ef->progtab[i].addr;
17467226306eSKonstantin Belousov 				stopp = (Elf_Addr)(startp + ef->progtab[i].size);
17477226306eSKonstantin Belousov 				break;
17487226306eSKonstantin Belousov 			}
17497226306eSKonstantin Belousov 		}
17507226306eSKonstantin Belousov 		if (i == ef->nprogtab)
17517226306eSKonstantin Belousov 			continue;
17527226306eSKonstantin Belousov 
17537226306eSKonstantin Belousov 		sym->st_value = start ? startp : stopp;
17547226306eSKonstantin Belousov 		sym->st_shndx = i;
17557226306eSKonstantin Belousov 	}
17567226306eSKonstantin Belousov }
17577226306eSKonstantin Belousov 
17583cfce8e4SKonstantin Belousov static int
1759289ead7cSKonstantin Belousov link_elf_reloc_local(linker_file_t lf, bool ifuncs)
17607251b4bfSJake Burkholder {
1761e9eabf59SPeter Wemm 	elf_file_t ef = (elf_file_t)lf;
17627251b4bfSJake Burkholder 	const Elf_Rel *rellim;
17637251b4bfSJake Burkholder 	const Elf_Rel *rel;
17647251b4bfSJake Burkholder 	const Elf_Rela *relalim;
17657251b4bfSJake Burkholder 	const Elf_Rela *rela;
1766e9eabf59SPeter Wemm 	const Elf_Sym *sym;
1767e9eabf59SPeter Wemm 	Elf_Addr base;
1768e9eabf59SPeter Wemm 	int i;
1769757686b1SMarcel Moolenaar 	Elf_Size symidx;
1770e9eabf59SPeter Wemm 
17717226306eSKonstantin Belousov 	link_elf_fix_link_set(ef);
17727226306eSKonstantin Belousov 
17737251b4bfSJake Burkholder 	/* Perform relocations without addend if there are any: */
1774b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nreltab; i++) {
1775e9eabf59SPeter Wemm 		rel = ef->reltab[i].rel;
17763cfce8e4SKonstantin Belousov 		if (rel == NULL) {
17773cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost a reltab");
17783cfce8e4SKonstantin Belousov 			return (ENOEXEC);
17793cfce8e4SKonstantin Belousov 		}
178020947801SPeter Wemm 		rellim = rel + ef->reltab[i].nrel;
1781e9eabf59SPeter Wemm 		base = findbase(ef, ef->reltab[i].sec);
17823cfce8e4SKonstantin Belousov 		if (base == 0) {
17833cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost base for reltab");
17843cfce8e4SKonstantin Belousov 			return (ENOEXEC);
17853cfce8e4SKonstantin Belousov 		}
178620947801SPeter Wemm 		for ( ; rel < rellim; rel++) {
1787e9eabf59SPeter Wemm 			symidx = ELF_R_SYM(rel->r_info);
178820947801SPeter Wemm 			if (symidx >= ef->ddbsymcnt)
178920947801SPeter Wemm 				continue;
1790e9eabf59SPeter Wemm 			sym = ef->ddbsymtab + symidx;
179120947801SPeter Wemm 			/* Only do local relocs */
179220947801SPeter Wemm 			if (ELF_ST_BIND(sym->st_info) != STB_LOCAL)
179320947801SPeter Wemm 				continue;
179436716fe2SMark Johnston 			if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC ||
179544c705cfSMitchell Horne 			    elf_is_ifunc_reloc(rel->r_info)) != ifuncs)
179644c705cfSMitchell Horne 				continue;
179744c705cfSMitchell Horne 			if (elf_reloc_local(lf, base, rel, ELF_RELOC_REL,
179844c705cfSMitchell Horne 			    elf_obj_lookup) != 0)
179944c705cfSMitchell Horne 				return (ENOEXEC);
18007251b4bfSJake Burkholder 		}
1801e9eabf59SPeter Wemm 	}
18027251b4bfSJake Burkholder 
18037251b4bfSJake Burkholder 	/* Perform relocations with addend if there are any: */
1804b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nrelatab; i++) {
1805e9eabf59SPeter Wemm 		rela = ef->relatab[i].rela;
18063cfce8e4SKonstantin Belousov 		if (rela == NULL) {
18073cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost a relatab!");
18083cfce8e4SKonstantin Belousov 			return (ENOEXEC);
18093cfce8e4SKonstantin Belousov 		}
181020947801SPeter Wemm 		relalim = rela + ef->relatab[i].nrela;
1811e9eabf59SPeter Wemm 		base = findbase(ef, ef->relatab[i].sec);
18123cfce8e4SKonstantin Belousov 		if (base == 0) {
18133cfce8e4SKonstantin Belousov 			link_elf_error(ef->lf.filename, "lost base for reltab");
18143cfce8e4SKonstantin Belousov 			return (ENOEXEC);
18153cfce8e4SKonstantin Belousov 		}
181620947801SPeter Wemm 		for ( ; rela < relalim; rela++) {
1817e9eabf59SPeter Wemm 			symidx = ELF_R_SYM(rela->r_info);
181820947801SPeter Wemm 			if (symidx >= ef->ddbsymcnt)
181920947801SPeter Wemm 				continue;
1820e9eabf59SPeter Wemm 			sym = ef->ddbsymtab + symidx;
182120947801SPeter Wemm 			/* Only do local relocs */
182220947801SPeter Wemm 			if (ELF_ST_BIND(sym->st_info) != STB_LOCAL)
182320947801SPeter Wemm 				continue;
182436716fe2SMark Johnston 			if ((ELF_ST_TYPE(sym->st_info) == STT_GNU_IFUNC ||
182544c705cfSMitchell Horne 			    elf_is_ifunc_reloc(rela->r_info)) != ifuncs)
182644c705cfSMitchell Horne 				continue;
182744c705cfSMitchell Horne 			if (elf_reloc_local(lf, base, rela, ELF_RELOC_RELA,
182844c705cfSMitchell Horne 			    elf_obj_lookup) != 0)
182944c705cfSMitchell Horne 				return (ENOEXEC);
18307251b4bfSJake Burkholder 		}
18317251b4bfSJake Burkholder 	}
18323cfce8e4SKonstantin Belousov 	return (0);
1833e9eabf59SPeter Wemm }
183400a5db46SStacey Son 
183500a5db46SStacey Son static long
1836e76f11f4SAndriy Gapon link_elf_symtab_get(linker_file_t lf, const Elf_Sym **symtab)
183700a5db46SStacey Son {
183800a5db46SStacey Son 	elf_file_t ef = (elf_file_t)lf;
183900a5db46SStacey Son 
184000a5db46SStacey Son 	*symtab = ef->ddbsymtab;
184100a5db46SStacey Son 	if (*symtab == NULL)
184200a5db46SStacey Son 		return (0);
184300a5db46SStacey Son 	return (ef->ddbsymcnt);
184400a5db46SStacey Son }
184500a5db46SStacey Son 
184600a5db46SStacey Son static long
184700a5db46SStacey Son link_elf_strtab_get(linker_file_t lf, caddr_t *strtab)
184800a5db46SStacey Son {
184900a5db46SStacey Son 	elf_file_t ef = (elf_file_t)lf;
185000a5db46SStacey Son 
185100a5db46SStacey Son 	*strtab = ef->ddbstrtab;
185200a5db46SStacey Son 	if (*strtab == NULL)
185300a5db46SStacey Son 		return (0);
185400a5db46SStacey Son 	return (ef->ddbstrcnt);
185500a5db46SStacey Son }
1856