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