xref: /freebsd/sys/kern/link_elf_obj.c (revision 13f28d969a91b4abe8ca0e767a7e293077860b16)
1a4f67738SDoug Rabson /*-
2326e27d8SDoug Rabson  * Copyright (c) 1998-2000 Doug Rabson
3e9eabf59SPeter Wemm  * Copyright (c) 2004 Peter Wemm
4a4f67738SDoug Rabson  * All rights reserved.
5a4f67738SDoug Rabson  *
6a4f67738SDoug Rabson  * Redistribution and use in source and binary forms, with or without
7a4f67738SDoug Rabson  * modification, are permitted provided that the following conditions
8a4f67738SDoug Rabson  * are met:
9a4f67738SDoug Rabson  * 1. Redistributions of source code must retain the above copyright
10a4f67738SDoug Rabson  *    notice, this list of conditions and the following disclaimer.
11a4f67738SDoug Rabson  * 2. Redistributions in binary form must reproduce the above copyright
12a4f67738SDoug Rabson  *    notice, this list of conditions and the following disclaimer in the
13a4f67738SDoug Rabson  *    documentation and/or other materials provided with the distribution.
14a4f67738SDoug Rabson  *
15a4f67738SDoug Rabson  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16a4f67738SDoug Rabson  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17a4f67738SDoug Rabson  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18a4f67738SDoug Rabson  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19a4f67738SDoug Rabson  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20a4f67738SDoug Rabson  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21a4f67738SDoug Rabson  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22a4f67738SDoug Rabson  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23a4f67738SDoug Rabson  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24a4f67738SDoug Rabson  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25a4f67738SDoug Rabson  * SUCH DAMAGE.
26a4f67738SDoug Rabson  */
27a4f67738SDoug Rabson 
28677b542eSDavid E. O'Brien #include <sys/cdefs.h>
29677b542eSDavid E. O'Brien __FBSDID("$FreeBSD$");
30677b542eSDavid E. O'Brien 
31326e27d8SDoug Rabson #include "opt_ddb.h"
32326e27d8SDoug Rabson 
33a4f67738SDoug Rabson #include <sys/param.h>
34a4f67738SDoug Rabson #include <sys/systm.h>
35fb919e4dSMark Murray #include <sys/kernel.h>
36fb919e4dSMark Murray #include <sys/lock.h>
37a4f67738SDoug Rabson #include <sys/malloc.h>
38c0b824f9SDima Dorfman #include <sys/mutex.h>
3962d615d5SJohn Baldwin #include <sys/mount.h>
40a4f67738SDoug Rabson #include <sys/proc.h>
41a4f67738SDoug Rabson #include <sys/namei.h>
42a4f67738SDoug Rabson #include <sys/fcntl.h>
43a4f67738SDoug Rabson #include <sys/vnode.h>
44a4f67738SDoug Rabson #include <sys/linker.h>
45fb919e4dSMark Murray 
46a4f67738SDoug Rabson #include <machine/elf.h>
47a4f67738SDoug Rabson 
48eddfbb76SRobert Watson #include <net/vnet.h>
49eddfbb76SRobert Watson 
50aed55708SRobert Watson #include <security/mac/mac_framework.h>
51aed55708SRobert Watson 
52fe3db7c7SDoug Rabson #include <vm/vm.h>
53fe3db7c7SDoug Rabson #include <vm/vm_param.h>
54fe3db7c7SDoug Rabson #include <vm/vm_object.h>
55fe3db7c7SDoug Rabson #include <vm/vm_kern.h>
56fe3db7c7SDoug Rabson #include <vm/vm_extern.h>
57fe3db7c7SDoug Rabson #include <vm/pmap.h>
58fe3db7c7SDoug Rabson #include <vm/vm_map.h>
59fb919e4dSMark Murray 
60f9980387SPeter Wemm #include <sys/link_elf.h>
61fe3db7c7SDoug Rabson 
62a2024a3eSJohn Birrell #ifdef DDB_CTF
63d9db5225SCraig Rodrigues #include <sys/zlib.h>
64a2024a3eSJohn Birrell #endif
65a2024a3eSJohn Birrell 
66326e27d8SDoug Rabson #include "linker_if.h"
67a4f67738SDoug Rabson 
68e9eabf59SPeter Wemm typedef struct {
69e9eabf59SPeter Wemm 	void		*addr;
7020947801SPeter Wemm 	Elf_Off		size;
71e9eabf59SPeter Wemm 	int		flags;
72e9eabf59SPeter Wemm 	int		sec;	/* Original section */
73e9eabf59SPeter Wemm 	char		*name;
74e9eabf59SPeter Wemm } Elf_progent;
75e9eabf59SPeter Wemm 
76e9eabf59SPeter Wemm typedef struct {
77e9eabf59SPeter Wemm 	Elf_Rel		*rel;
7820947801SPeter Wemm 	int		nrel;
79e9eabf59SPeter Wemm 	int		sec;
80e9eabf59SPeter Wemm } Elf_relent;
81e9eabf59SPeter Wemm 
82e9eabf59SPeter Wemm typedef struct {
83e9eabf59SPeter Wemm 	Elf_Rela	*rela;
8420947801SPeter Wemm 	int		nrela;
85e9eabf59SPeter Wemm 	int		sec;
86e9eabf59SPeter Wemm } Elf_relaent;
87e9eabf59SPeter Wemm 
88e9eabf59SPeter Wemm 
89a4f67738SDoug Rabson typedef struct elf_file {
90326e27d8SDoug Rabson 	struct linker_file lf;		/* Common fields */
9120947801SPeter Wemm 
9270b7ffeeSIan Dowse 	int		preloaded;
93fe3db7c7SDoug Rabson 	caddr_t		address;	/* Relocation address */
94fe3db7c7SDoug Rabson 	vm_object_t	object;		/* VM object to hold file pages */
95e9eabf59SPeter Wemm 	Elf_Shdr	*e_shdr;
96e9eabf59SPeter Wemm 
97e9eabf59SPeter Wemm 	Elf_progent	*progtab;
98e9eabf59SPeter Wemm 	int		nprogtab;
99e9eabf59SPeter Wemm 
100e9eabf59SPeter Wemm 	Elf_relaent	*relatab;
101b474c780SDavid E. O'Brien 	int		nrelatab;
102e9eabf59SPeter Wemm 
103e9eabf59SPeter Wemm 	Elf_relent	*reltab;
104b474c780SDavid E. O'Brien 	int		nreltab;
105e9eabf59SPeter Wemm 
106e9eabf59SPeter Wemm 	Elf_Sym		*ddbsymtab;	/* The symbol table we are using */
1072d636ab0SPeter Wemm 	long		ddbsymcnt;	/* Number of symbols */
1082d636ab0SPeter Wemm 	caddr_t		ddbstrtab;	/* String table */
1092d636ab0SPeter Wemm 	long		ddbstrcnt;	/* number of bytes in string table */
110e9eabf59SPeter Wemm 
111e9eabf59SPeter Wemm 	caddr_t		shstrtab;	/* Section name string table */
112e9eabf59SPeter Wemm 	long		shstrcnt;	/* number of bytes in string table */
113e9eabf59SPeter Wemm 
114a2024a3eSJohn Birrell 	caddr_t		ctftab;		/* CTF table */
115a2024a3eSJohn Birrell 	long		ctfcnt;		/* number of bytes in CTF table */
116a2024a3eSJohn Birrell 	caddr_t		ctfoff;		/* CTF offset table */
117a2024a3eSJohn Birrell 	caddr_t		typoff;		/* Type offset table */
118a2024a3eSJohn Birrell 	long		typlen;		/* Number of type entries. */
119a2024a3eSJohn Birrell 
120a4f67738SDoug Rabson } *elf_file_t;
121a4f67738SDoug Rabson 
122a2024a3eSJohn Birrell #include <kern/kern_ctf.c>
123a2024a3eSJohn Birrell 
12454823af2SPeter Wemm static int	link_elf_link_preload(linker_class_t cls,
125326e27d8SDoug Rabson 		    const char *, linker_file_t *);
12654823af2SPeter Wemm static int	link_elf_link_preload_finish(linker_file_t);
12754823af2SPeter Wemm static int	link_elf_load_file(linker_class_t, const char *, linker_file_t *);
128326e27d8SDoug Rabson static int	link_elf_lookup_symbol(linker_file_t, const char *,
129326e27d8SDoug Rabson 		    c_linker_sym_t *);
130e9eabf59SPeter Wemm static int	link_elf_symbol_values(linker_file_t, c_linker_sym_t,
131e9eabf59SPeter Wemm 		    linker_symval_t *);
132326e27d8SDoug Rabson static int	link_elf_search_symbol(linker_file_t, caddr_t value,
133326e27d8SDoug Rabson 		    c_linker_sym_t *sym, long *diffp);
134326e27d8SDoug Rabson 
135326e27d8SDoug Rabson static void	link_elf_unload_file(linker_file_t);
136f41325dbSPeter Wemm static int	link_elf_lookup_set(linker_file_t, const char *,
137f41325dbSPeter Wemm 		    void ***, void ***, int *);
138bb9fe9ddSBrian Feldman static int	link_elf_each_function_name(linker_file_t,
139e9eabf59SPeter Wemm 		    int (*)(const char *, void *), void *);
140a2024a3eSJohn Birrell static int	link_elf_each_function_nameval(linker_file_t,
141a2024a3eSJohn Birrell 				linker_function_nameval_callback_t,
142a2024a3eSJohn Birrell 				void *);
1437251b4bfSJake Burkholder static void	link_elf_reloc_local(linker_file_t);
144e76f11f4SAndriy Gapon static long	link_elf_symtab_get(linker_file_t, const Elf_Sym **);
14500a5db46SStacey Son static long	link_elf_strtab_get(linker_file_t, caddr_t *);
146326e27d8SDoug Rabson 
147cff8c6f2SKonstantin Belousov static int	elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps,
148cff8c6f2SKonstantin Belousov 		    Elf_Addr *);
149e9eabf59SPeter Wemm 
150326e27d8SDoug Rabson static kobj_method_t link_elf_methods[] = {
151326e27d8SDoug Rabson 	KOBJMETHOD(linker_lookup_symbol,	link_elf_lookup_symbol),
152326e27d8SDoug Rabson 	KOBJMETHOD(linker_symbol_values,	link_elf_symbol_values),
153326e27d8SDoug Rabson 	KOBJMETHOD(linker_search_symbol,	link_elf_search_symbol),
154326e27d8SDoug Rabson 	KOBJMETHOD(linker_unload,		link_elf_unload_file),
15554823af2SPeter Wemm 	KOBJMETHOD(linker_load_file,		link_elf_load_file),
15654823af2SPeter Wemm 	KOBJMETHOD(linker_link_preload,		link_elf_link_preload),
15754823af2SPeter Wemm 	KOBJMETHOD(linker_link_preload_finish,	link_elf_link_preload_finish),
158f41325dbSPeter Wemm 	KOBJMETHOD(linker_lookup_set,		link_elf_lookup_set),
159bb9fe9ddSBrian Feldman 	KOBJMETHOD(linker_each_function_name,	link_elf_each_function_name),
160a2024a3eSJohn Birrell 	KOBJMETHOD(linker_each_function_nameval, link_elf_each_function_nameval),
161a2024a3eSJohn Birrell 	KOBJMETHOD(linker_ctf_get,		link_elf_ctf_get),
16200a5db46SStacey Son 	KOBJMETHOD(linker_symtab_get, 		link_elf_symtab_get),
16300a5db46SStacey Son 	KOBJMETHOD(linker_strtab_get, 		link_elf_strtab_get),
164326e27d8SDoug Rabson 	{ 0, 0 }
165326e27d8SDoug Rabson };
166326e27d8SDoug Rabson 
167326e27d8SDoug Rabson static struct linker_class link_elf_class = {
168326e27d8SDoug Rabson #if ELF_TARG_CLASS == ELFCLASS32
169e9eabf59SPeter Wemm 	"elf32_obj",
170326e27d8SDoug Rabson #else
171e9eabf59SPeter Wemm 	"elf64_obj",
172326e27d8SDoug Rabson #endif
173326e27d8SDoug Rabson 	link_elf_methods, sizeof(struct elf_file)
174326e27d8SDoug Rabson };
175326e27d8SDoug Rabson 
176326e27d8SDoug Rabson static int	relocate_file(elf_file_t ef);
1772832cd54SKonstantin Belousov static void	elf_obj_cleanup_globals_cache(elf_file_t);
178a4f67738SDoug Rabson 
179a4f67738SDoug Rabson static void
180552f9f63SEdwin Groothuis link_elf_error(const char *filename, const char *s)
1811720979bSMarcel Moolenaar {
182552f9f63SEdwin Groothuis 	if (filename == NULL)
1831720979bSMarcel Moolenaar 		printf("kldload: %s\n", s);
184552f9f63SEdwin Groothuis 	else
185552f9f63SEdwin Groothuis 		printf("kldload: %s: %s\n", filename, s);
1861720979bSMarcel Moolenaar }
1871720979bSMarcel Moolenaar 
1881720979bSMarcel Moolenaar static void
189a4f67738SDoug Rabson link_elf_init(void *arg)
190a4f67738SDoug Rabson {
191a4f67738SDoug Rabson 
192326e27d8SDoug Rabson 	linker_add_class(&link_elf_class);
193a4f67738SDoug Rabson }
194a4f67738SDoug Rabson 
195e9eabf59SPeter Wemm SYSINIT(link_elf_obj, SI_SUB_KLD, SI_ORDER_SECOND, link_elf_init, 0);
196a4f67738SDoug Rabson 
197a4f67738SDoug Rabson static int
198e9eabf59SPeter Wemm link_elf_link_preload(linker_class_t cls, const char *filename,
199e9eabf59SPeter Wemm     linker_file_t *result)
2002d636ab0SPeter Wemm {
20170b7ffeeSIan Dowse 	Elf_Ehdr *hdr;
20270b7ffeeSIan Dowse 	Elf_Shdr *shdr;
20370b7ffeeSIan Dowse 	Elf_Sym *es;
20470b7ffeeSIan Dowse 	void *modptr, *baseptr, *sizeptr;
20570b7ffeeSIan Dowse 	char *type;
20670b7ffeeSIan Dowse 	elf_file_t ef;
20770b7ffeeSIan Dowse 	linker_file_t lf;
20870b7ffeeSIan Dowse 	Elf_Addr off;
20970b7ffeeSIan Dowse 	int error, i, j, pb, ra, rl, shstrindex, symstrindex, symtabindex;
21070b7ffeeSIan Dowse 
21170b7ffeeSIan Dowse 	/* Look to see if we have the file preloaded */
21270b7ffeeSIan Dowse 	modptr = preload_search_by_name(filename);
21370b7ffeeSIan Dowse 	if (modptr == NULL)
21470b7ffeeSIan Dowse 		return ENOENT;
21570b7ffeeSIan Dowse 
21670b7ffeeSIan Dowse 	type = (char *)preload_search_info(modptr, MODINFO_TYPE);
21770b7ffeeSIan Dowse 	baseptr = preload_search_info(modptr, MODINFO_ADDR);
21870b7ffeeSIan Dowse 	sizeptr = preload_search_info(modptr, MODINFO_SIZE);
21970b7ffeeSIan Dowse 	hdr = (Elf_Ehdr *)preload_search_info(modptr, MODINFO_METADATA |
22070b7ffeeSIan Dowse 	    MODINFOMD_ELFHDR);
22170b7ffeeSIan Dowse 	shdr = (Elf_Shdr *)preload_search_info(modptr, MODINFO_METADATA |
22270b7ffeeSIan Dowse 	    MODINFOMD_SHDR);
22370b7ffeeSIan Dowse 	if (type == NULL || (strcmp(type, "elf" __XSTRING(__ELF_WORD_SIZE)
22470b7ffeeSIan Dowse 	    " obj module") != 0 &&
22570b7ffeeSIan Dowse 	    strcmp(type, "elf obj module") != 0)) {
226de78ca7eSPeter Wemm 		return (EFTYPE);
227de78ca7eSPeter Wemm 	}
22870b7ffeeSIan Dowse 	if (baseptr == NULL || sizeptr == NULL || hdr == NULL ||
22970b7ffeeSIan Dowse 	    shdr == NULL)
23070b7ffeeSIan Dowse 		return (EINVAL);
23170b7ffeeSIan Dowse 
23270b7ffeeSIan Dowse 	lf = linker_make_file(filename, &link_elf_class);
23370b7ffeeSIan Dowse 	if (lf == NULL)
23470b7ffeeSIan Dowse 		return (ENOMEM);
23570b7ffeeSIan Dowse 
23670b7ffeeSIan Dowse 	ef = (elf_file_t)lf;
23770b7ffeeSIan Dowse 	ef->preloaded = 1;
23870b7ffeeSIan Dowse 	ef->address = *(caddr_t *)baseptr;
23970b7ffeeSIan Dowse 	lf->address = *(caddr_t *)baseptr;
24070b7ffeeSIan Dowse 	lf->size = *(size_t *)sizeptr;
24170b7ffeeSIan Dowse 
24270b7ffeeSIan Dowse 	if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS ||
24370b7ffeeSIan Dowse 	    hdr->e_ident[EI_DATA] != ELF_TARG_DATA ||
24470b7ffeeSIan Dowse 	    hdr->e_ident[EI_VERSION] != EV_CURRENT ||
24570b7ffeeSIan Dowse 	    hdr->e_version != EV_CURRENT ||
24670b7ffeeSIan Dowse 	    hdr->e_type != ET_REL ||
24770b7ffeeSIan Dowse 	    hdr->e_machine != ELF_TARG_MACH) {
24870b7ffeeSIan Dowse 		error = EFTYPE;
24970b7ffeeSIan Dowse 		goto out;
25070b7ffeeSIan Dowse 	}
25170b7ffeeSIan Dowse 	ef->e_shdr = shdr;
25270b7ffeeSIan Dowse 
25370b7ffeeSIan Dowse 	/* Scan the section header for information and table sizing. */
25470b7ffeeSIan Dowse 	symtabindex = -1;
25570b7ffeeSIan Dowse 	symstrindex = -1;
25670b7ffeeSIan Dowse 	for (i = 0; i < hdr->e_shnum; i++) {
25770b7ffeeSIan Dowse 		switch (shdr[i].sh_type) {
25870b7ffeeSIan Dowse 		case SHT_PROGBITS:
25970b7ffeeSIan Dowse 		case SHT_NOBITS:
260*13f28d96SKonstantin Belousov #ifdef __amd64__
261*13f28d96SKonstantin Belousov 		case SHT_AMD64_UNWIND:
262*13f28d96SKonstantin Belousov #endif
26370b7ffeeSIan Dowse 			ef->nprogtab++;
26470b7ffeeSIan Dowse 			break;
26570b7ffeeSIan Dowse 		case SHT_SYMTAB:
26670b7ffeeSIan Dowse 			symtabindex = i;
26770b7ffeeSIan Dowse 			symstrindex = shdr[i].sh_link;
26870b7ffeeSIan Dowse 			break;
26970b7ffeeSIan Dowse 		case SHT_REL:
270b474c780SDavid E. O'Brien 			ef->nreltab++;
27170b7ffeeSIan Dowse 			break;
27270b7ffeeSIan Dowse 		case SHT_RELA:
273b474c780SDavid E. O'Brien 			ef->nrelatab++;
27470b7ffeeSIan Dowse 			break;
27570b7ffeeSIan Dowse 		}
27670b7ffeeSIan Dowse 	}
27770b7ffeeSIan Dowse 
27870b7ffeeSIan Dowse 	shstrindex = hdr->e_shstrndx;
27970b7ffeeSIan Dowse 	if (ef->nprogtab == 0 || symstrindex < 0 ||
28070b7ffeeSIan Dowse 	    symstrindex >= hdr->e_shnum ||
28170b7ffeeSIan Dowse 	    shdr[symstrindex].sh_type != SHT_STRTAB || shstrindex == 0 ||
28270b7ffeeSIan Dowse 	    shstrindex >= hdr->e_shnum ||
28370b7ffeeSIan Dowse 	    shdr[shstrindex].sh_type != SHT_STRTAB) {
28470b7ffeeSIan Dowse 		printf("%s: bad/missing section headers\n", filename);
28570b7ffeeSIan Dowse 		error = ENOEXEC;
28670b7ffeeSIan Dowse 		goto out;
28770b7ffeeSIan Dowse 	}
28870b7ffeeSIan Dowse 
28970b7ffeeSIan Dowse 	/* Allocate space for tracking the load chunks */
29070b7ffeeSIan Dowse 	if (ef->nprogtab != 0)
29170b7ffeeSIan Dowse 		ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab),
29270b7ffeeSIan Dowse 		    M_LINKER, M_WAITOK | M_ZERO);
293b474c780SDavid E. O'Brien 	if (ef->nreltab != 0)
294b474c780SDavid E. O'Brien 		ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab),
295b474c780SDavid E. O'Brien 		    M_LINKER, M_WAITOK | M_ZERO);
296b474c780SDavid E. O'Brien 	if (ef->nrelatab != 0)
297b474c780SDavid E. O'Brien 		ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab),
298b474c780SDavid E. O'Brien 		    M_LINKER, M_WAITOK | M_ZERO);
29970b7ffeeSIan Dowse 	if ((ef->nprogtab != 0 && ef->progtab == NULL) ||
300b474c780SDavid E. O'Brien 	    (ef->nreltab != 0 && ef->reltab == NULL) ||
301b474c780SDavid E. O'Brien 	    (ef->nrelatab != 0 && ef->relatab == NULL)) {
30270b7ffeeSIan Dowse 		error = ENOMEM;
30370b7ffeeSIan Dowse 		goto out;
30470b7ffeeSIan Dowse 	}
30570b7ffeeSIan Dowse 
30670b7ffeeSIan Dowse 	/* XXX, relocate the sh_addr fields saved by the loader. */
30770b7ffeeSIan Dowse 	off = 0;
30870b7ffeeSIan Dowse 	for (i = 0; i < hdr->e_shnum; i++) {
30970b7ffeeSIan Dowse 		if (shdr[i].sh_addr != 0 && (off == 0 || shdr[i].sh_addr < off))
31070b7ffeeSIan Dowse 			off = shdr[i].sh_addr;
31170b7ffeeSIan Dowse 	}
31270b7ffeeSIan Dowse 	for (i = 0; i < hdr->e_shnum; i++) {
31370b7ffeeSIan Dowse 		if (shdr[i].sh_addr != 0)
31470b7ffeeSIan Dowse 			shdr[i].sh_addr = shdr[i].sh_addr - off +
31570b7ffeeSIan Dowse 			    (Elf_Addr)ef->address;
31670b7ffeeSIan Dowse 	}
31770b7ffeeSIan Dowse 
31870b7ffeeSIan Dowse 	ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym);
31970b7ffeeSIan Dowse 	ef->ddbsymtab = (Elf_Sym *)shdr[symtabindex].sh_addr;
32070b7ffeeSIan Dowse 	ef->ddbstrcnt = shdr[symstrindex].sh_size;
32170b7ffeeSIan Dowse 	ef->ddbstrtab = (char *)shdr[symstrindex].sh_addr;
32270b7ffeeSIan Dowse 	ef->shstrcnt = shdr[shstrindex].sh_size;
32370b7ffeeSIan Dowse 	ef->shstrtab = (char *)shdr[shstrindex].sh_addr;
32470b7ffeeSIan Dowse 
32570b7ffeeSIan Dowse 	/* Now fill out progtab and the relocation tables. */
32670b7ffeeSIan Dowse 	pb = 0;
32770b7ffeeSIan Dowse 	rl = 0;
32870b7ffeeSIan Dowse 	ra = 0;
32970b7ffeeSIan Dowse 	for (i = 0; i < hdr->e_shnum; i++) {
33070b7ffeeSIan Dowse 		switch (shdr[i].sh_type) {
33170b7ffeeSIan Dowse 		case SHT_PROGBITS:
33270b7ffeeSIan Dowse 		case SHT_NOBITS:
333*13f28d96SKonstantin Belousov #ifdef __amd64__
334*13f28d96SKonstantin Belousov 		case SHT_AMD64_UNWIND:
335*13f28d96SKonstantin Belousov #endif
33670b7ffeeSIan Dowse 			ef->progtab[pb].addr = (void *)shdr[i].sh_addr;
33770b7ffeeSIan Dowse 			if (shdr[i].sh_type == SHT_PROGBITS)
33870b7ffeeSIan Dowse 				ef->progtab[pb].name = "<<PROGBITS>>";
339*13f28d96SKonstantin Belousov #ifdef __amd64__
340*13f28d96SKonstantin Belousov 			else if (shdr[i].sh_type == SHT_AMD64_UNWIND)
341*13f28d96SKonstantin Belousov 				ef->progtab[pb].name = "<<UNWIND>>";
342*13f28d96SKonstantin Belousov #endif
34370b7ffeeSIan Dowse 			else
34470b7ffeeSIan Dowse 				ef->progtab[pb].name = "<<NOBITS>>";
34570b7ffeeSIan Dowse 			ef->progtab[pb].size = shdr[i].sh_size;
34670b7ffeeSIan Dowse 			ef->progtab[pb].sec = i;
34770b7ffeeSIan Dowse 			if (ef->shstrtab && shdr[i].sh_name != 0)
34870b7ffeeSIan Dowse 				ef->progtab[pb].name =
34970b7ffeeSIan Dowse 				    ef->shstrtab + shdr[i].sh_name;
35050c202c5SJeff Roberson 			if (ef->progtab[pb].name != NULL &&
3515f67450dSDimitry Andric 			    !strcmp(ef->progtab[pb].name, DPCPU_SETNAME)) {
35250c202c5SJeff Roberson 				void *dpcpu;
35350c202c5SJeff Roberson 
35450c202c5SJeff Roberson 				dpcpu = dpcpu_alloc(shdr[i].sh_size);
35550c202c5SJeff Roberson 				if (dpcpu == NULL) {
35650c202c5SJeff Roberson 					error = ENOSPC;
35750c202c5SJeff Roberson 					goto out;
35850c202c5SJeff Roberson 				}
35950c202c5SJeff Roberson 				memcpy(dpcpu, ef->progtab[pb].addr,
36050c202c5SJeff Roberson 				    ef->progtab[pb].size);
36150c202c5SJeff Roberson 				dpcpu_copy(dpcpu, shdr[i].sh_size);
36250c202c5SJeff Roberson 				ef->progtab[pb].addr = dpcpu;
363eddfbb76SRobert Watson #ifdef VIMAGE
364eddfbb76SRobert Watson 			} else if (ef->progtab[pb].name != NULL &&
36517ef1febSRobert Watson 			    !strcmp(ef->progtab[pb].name, VNET_SETNAME)) {
366eddfbb76SRobert Watson 				void *vnet_data;
367eddfbb76SRobert Watson 
368eddfbb76SRobert Watson 				vnet_data = vnet_data_alloc(shdr[i].sh_size);
369eddfbb76SRobert Watson 				if (vnet_data == NULL) {
370eddfbb76SRobert Watson 					error = ENOSPC;
371eddfbb76SRobert Watson 					goto out;
372eddfbb76SRobert Watson 				}
373eddfbb76SRobert Watson 				memcpy(vnet_data, ef->progtab[pb].addr,
374eddfbb76SRobert Watson 				    ef->progtab[pb].size);
375eddfbb76SRobert Watson 				vnet_data_copy(vnet_data, shdr[i].sh_size);
376eddfbb76SRobert Watson 				ef->progtab[pb].addr = vnet_data;
377eddfbb76SRobert Watson #endif
3780067051fSMarcel Moolenaar 			} else if (ef->progtab[pb].name != NULL &&
3790067051fSMarcel Moolenaar 			    !strcmp(ef->progtab[pb].name, ".ctors")) {
3800067051fSMarcel Moolenaar 				lf->ctors_addr = ef->progtab[pb].addr;
3810067051fSMarcel Moolenaar 				lf->ctors_size = shdr[i].sh_size;
38250c202c5SJeff Roberson 			}
38370b7ffeeSIan Dowse 
38470b7ffeeSIan Dowse 			/* Update all symbol values with the offset. */
38570b7ffeeSIan Dowse 			for (j = 0; j < ef->ddbsymcnt; j++) {
38670b7ffeeSIan Dowse 				es = &ef->ddbsymtab[j];
38770b7ffeeSIan Dowse 				if (es->st_shndx != i)
38870b7ffeeSIan Dowse 					continue;
38970b7ffeeSIan Dowse 				es->st_value += (Elf_Addr)ef->progtab[pb].addr;
39070b7ffeeSIan Dowse 			}
39170b7ffeeSIan Dowse 			pb++;
39270b7ffeeSIan Dowse 			break;
39370b7ffeeSIan Dowse 		case SHT_REL:
39470b7ffeeSIan Dowse 			ef->reltab[rl].rel = (Elf_Rel *)shdr[i].sh_addr;
39570b7ffeeSIan Dowse 			ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel);
39670b7ffeeSIan Dowse 			ef->reltab[rl].sec = shdr[i].sh_info;
39770b7ffeeSIan Dowse 			rl++;
39870b7ffeeSIan Dowse 			break;
39970b7ffeeSIan Dowse 		case SHT_RELA:
40070b7ffeeSIan Dowse 			ef->relatab[ra].rela = (Elf_Rela *)shdr[i].sh_addr;
40170b7ffeeSIan Dowse 			ef->relatab[ra].nrela =
40270b7ffeeSIan Dowse 			    shdr[i].sh_size / sizeof(Elf_Rela);
40370b7ffeeSIan Dowse 			ef->relatab[ra].sec = shdr[i].sh_info;
40470b7ffeeSIan Dowse 			ra++;
40570b7ffeeSIan Dowse 			break;
40670b7ffeeSIan Dowse 		}
40770b7ffeeSIan Dowse 	}
40870b7ffeeSIan Dowse 	if (pb != ef->nprogtab)
40970b7ffeeSIan Dowse 		panic("lost progbits");
410b474c780SDavid E. O'Brien 	if (rl != ef->nreltab)
411b474c780SDavid E. O'Brien 		panic("lost reltab");
412b474c780SDavid E. O'Brien 	if (ra != ef->nrelatab)
413b474c780SDavid E. O'Brien 		panic("lost relatab");
41470b7ffeeSIan Dowse 
41570b7ffeeSIan Dowse 	/* Local intra-module relocations */
41670b7ffeeSIan Dowse 	link_elf_reloc_local(lf);
41770b7ffeeSIan Dowse 
41870b7ffeeSIan Dowse 	*result = lf;
41970b7ffeeSIan Dowse 	return (0);
42070b7ffeeSIan Dowse 
42170b7ffeeSIan Dowse out:
42270b7ffeeSIan Dowse 	/* preload not done this way */
42370b7ffeeSIan Dowse 	linker_file_unload(lf, LINKER_UNLOAD_FORCE);
42470b7ffeeSIan Dowse 	return (error);
42570b7ffeeSIan Dowse }
426de78ca7eSPeter Wemm 
4270067051fSMarcel Moolenaar static void
4280067051fSMarcel Moolenaar link_elf_invoke_ctors(caddr_t addr, size_t size)
4290067051fSMarcel Moolenaar {
4300067051fSMarcel Moolenaar 	void (**ctor)(void);
4310067051fSMarcel Moolenaar 	size_t i, cnt;
4320067051fSMarcel Moolenaar 
4330067051fSMarcel Moolenaar 	if (addr == NULL || size == 0)
4340067051fSMarcel Moolenaar 		return;
4350067051fSMarcel Moolenaar 	cnt = size / sizeof(*ctor);
4360067051fSMarcel Moolenaar 	ctor = (void *)addr;
4370067051fSMarcel Moolenaar 	for (i = 0; i < cnt; i++) {
4380067051fSMarcel Moolenaar 		if (ctor[i] != NULL)
4390067051fSMarcel Moolenaar 			(*ctor[i])();
4400067051fSMarcel Moolenaar 	}
4410067051fSMarcel Moolenaar }
4420067051fSMarcel Moolenaar 
443de78ca7eSPeter Wemm static int
44454823af2SPeter Wemm link_elf_link_preload_finish(linker_file_t lf)
44554823af2SPeter Wemm {
44670b7ffeeSIan Dowse 	elf_file_t ef;
44770b7ffeeSIan Dowse 	int error;
44870b7ffeeSIan Dowse 
44970b7ffeeSIan Dowse 	ef = (elf_file_t)lf;
45070b7ffeeSIan Dowse 	error = relocate_file(ef);
45170b7ffeeSIan Dowse 	if (error)
45270b7ffeeSIan Dowse 		return error;
45370b7ffeeSIan Dowse 
45470b7ffeeSIan Dowse 	/* Notify MD code that a module is being loaded. */
45570b7ffeeSIan Dowse 	error = elf_cpu_load_file(lf);
45670b7ffeeSIan Dowse 	if (error)
45770b7ffeeSIan Dowse 		return (error);
45870b7ffeeSIan Dowse 
4590067051fSMarcel Moolenaar 	/* Invoke .ctors */
4600067051fSMarcel Moolenaar 	link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size);
46170b7ffeeSIan Dowse 	return (0);
46254823af2SPeter Wemm }
46354823af2SPeter Wemm 
46454823af2SPeter Wemm static int
465c143d6c2SMarcel Moolenaar link_elf_load_file(linker_class_t cls, const char *filename,
466c143d6c2SMarcel Moolenaar     linker_file_t *result)
467a4f67738SDoug Rabson {
468a4f67738SDoug Rabson 	struct nameidata nd;
469b40ce416SJulian Elischer 	struct thread *td = curthread;	/* XXX */
470caab6e90SPeter Wemm 	Elf_Ehdr *hdr;
471e9eabf59SPeter Wemm 	Elf_Shdr *shdr;
472a8774e39SPeter Wemm 	Elf_Sym *es;
473a8774e39SPeter Wemm 	int nbytes, i, j;
47420947801SPeter Wemm 	vm_offset_t mapbase;
475fe3db7c7SDoug Rabson 	size_t mapsize;
476a4f67738SDoug Rabson 	int error = 0;
477526d0bd5SKonstantin Belousov 	ssize_t resid;
478526d0bd5SKonstantin Belousov 	int flags;
479a4f67738SDoug Rabson 	elf_file_t ef;
480a4f67738SDoug Rabson 	linker_file_t lf;
481ca65d5c7SPeter Wemm 	int symtabindex;
482ca65d5c7SPeter Wemm 	int symstrindex;
483e9eabf59SPeter Wemm 	int shstrindex;
484e9eabf59SPeter Wemm 	int nsym;
48520947801SPeter Wemm 	int pb, rl, ra;
486e9eabf59SPeter Wemm 	int alignmask;
487ca65d5c7SPeter Wemm 
488ca65d5c7SPeter Wemm 	shdr = NULL;
489ca65d5c7SPeter Wemm 	lf = NULL;
490e9eabf59SPeter Wemm 	mapsize = 0;
491e9eabf59SPeter Wemm 	hdr = NULL;
492a4f67738SDoug Rabson 
4935050aa86SKonstantin Belousov 	NDINIT(&nd, LOOKUP, FOLLOW, UIO_SYSSPACE, filename, td);
494e6796b67SKirk McKusick 	flags = FREAD;
4959e223287SKonstantin Belousov 	error = vn_open(&nd, &flags, 0, NULL);
496a4f67738SDoug Rabson 	if (error)
497a4f67738SDoug Rabson 		return error;
498762e6b85SEivind Eklund 	NDFREE(&nd, NDF_ONLY_PNBUF);
4991f49b573SRuslan Ermilov 	if (nd.ni_vp->v_type != VREG) {
5001f49b573SRuslan Ermilov 		error = ENOEXEC;
5011f49b573SRuslan Ermilov 		goto out;
5021f49b573SRuslan Ermilov 	}
503a3df768bSRobert Watson #ifdef MAC
50430d239bcSRobert Watson 	error = mac_kld_check_load(td->td_ucred, nd.ni_vp);
505a3df768bSRobert Watson 	if (error) {
506a3df768bSRobert Watson 		goto out;
507a3df768bSRobert Watson 	}
508a3df768bSRobert Watson #endif
509a4f67738SDoug Rabson 
510e9eabf59SPeter Wemm 	/* Read the elf header from the file. */
511e9eabf59SPeter Wemm 	hdr = malloc(sizeof(*hdr), M_LINKER, M_WAITOK);
512e9eabf59SPeter Wemm 	error = vn_rdwr(UIO_READ, nd.ni_vp, (void *)hdr, sizeof(*hdr), 0,
5139ca43589SRobert Watson 	    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
5149ca43589SRobert Watson 	    &resid, td);
515a4f67738SDoug Rabson 	if (error)
516a4f67738SDoug Rabson 		goto out;
517e9eabf59SPeter Wemm 	if (resid != 0){
518e9eabf59SPeter Wemm 		error = ENOEXEC;
519e9eabf59SPeter Wemm 		goto out;
520e9eabf59SPeter Wemm 	}
521a4f67738SDoug Rabson 
522caab6e90SPeter Wemm 	if (!IS_ELF(*hdr)) {
523fe3db7c7SDoug Rabson 		error = ENOEXEC;
524a4f67738SDoug Rabson 		goto out;
525fe3db7c7SDoug Rabson 	}
526fe3db7c7SDoug Rabson 
527caab6e90SPeter Wemm 	if (hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS
528caab6e90SPeter Wemm 	    || hdr->e_ident[EI_DATA] != ELF_TARG_DATA) {
529552f9f63SEdwin Groothuis 		link_elf_error(filename, "Unsupported file layout");
530fe3db7c7SDoug Rabson 		error = ENOEXEC;
531fe3db7c7SDoug Rabson 		goto out;
532fe3db7c7SDoug Rabson 	}
533caab6e90SPeter Wemm 	if (hdr->e_ident[EI_VERSION] != EV_CURRENT
534caab6e90SPeter Wemm 	    || hdr->e_version != EV_CURRENT) {
535552f9f63SEdwin Groothuis 		link_elf_error(filename, "Unsupported file version");
536fe3db7c7SDoug Rabson 		error = ENOEXEC;
537fe3db7c7SDoug Rabson 		goto out;
538fe3db7c7SDoug Rabson 	}
539e9eabf59SPeter Wemm 	if (hdr->e_type != ET_REL) {
540a1d7ce03SAttilio Rao 		error = ENOSYS;
541fe3db7c7SDoug Rabson 		goto out;
542fe3db7c7SDoug Rabson 	}
543caab6e90SPeter Wemm 	if (hdr->e_machine != ELF_TARG_MACH) {
544552f9f63SEdwin Groothuis 		link_elf_error(filename, "Unsupported machine");
545fe3db7c7SDoug Rabson 		error = ENOEXEC;
546fe3db7c7SDoug Rabson 		goto out;
547fe3db7c7SDoug Rabson 	}
548a4f67738SDoug Rabson 
549326e27d8SDoug Rabson 	lf = linker_make_file(filename, &link_elf_class);
550326e27d8SDoug Rabson 	if (!lf) {
551326e27d8SDoug Rabson 		error = ENOMEM;
552326e27d8SDoug Rabson 		goto out;
553326e27d8SDoug Rabson 	}
554326e27d8SDoug Rabson 	ef = (elf_file_t) lf;
555e9eabf59SPeter Wemm 	ef->nprogtab = 0;
556e9eabf59SPeter Wemm 	ef->e_shdr = 0;
557b474c780SDavid E. O'Brien 	ef->nreltab = 0;
558b474c780SDavid E. O'Brien 	ef->nrelatab = 0;
559e9eabf59SPeter Wemm 
560e9eabf59SPeter Wemm 	/* Allocate and read in the section header */
561e9eabf59SPeter Wemm 	nbytes = hdr->e_shnum * hdr->e_shentsize;
562e9eabf59SPeter Wemm 	if (nbytes == 0 || hdr->e_shoff == 0 ||
563e9eabf59SPeter Wemm 	    hdr->e_shentsize != sizeof(Elf_Shdr)) {
564e9eabf59SPeter Wemm 		error = ENOEXEC;
565e9eabf59SPeter Wemm 		goto out;
566e9eabf59SPeter Wemm 	}
56720947801SPeter Wemm 	shdr = malloc(nbytes, M_LINKER, M_WAITOK);
568e9eabf59SPeter Wemm 	ef->e_shdr = shdr;
569e9eabf59SPeter Wemm 	error = vn_rdwr(UIO_READ, nd.ni_vp, (caddr_t)shdr, nbytes, hdr->e_shoff,
570e9eabf59SPeter Wemm 	    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED, &resid, td);
571e9eabf59SPeter Wemm 	if (error)
572e9eabf59SPeter Wemm 		goto out;
573e9eabf59SPeter Wemm 	if (resid) {
574e9eabf59SPeter Wemm 		error = ENOEXEC;
575e9eabf59SPeter Wemm 		goto out;
576e9eabf59SPeter Wemm 	}
577e9eabf59SPeter Wemm 
578e9eabf59SPeter Wemm 	/* Scan the section header for information and table sizing. */
579e9eabf59SPeter Wemm 	nsym = 0;
580e9eabf59SPeter Wemm 	symtabindex = -1;
581e9eabf59SPeter Wemm 	symstrindex = -1;
582e9eabf59SPeter Wemm 	for (i = 0; i < hdr->e_shnum; i++) {
583676799a0SAndriy Gapon 		if (shdr[i].sh_size == 0)
584676799a0SAndriy Gapon 			continue;
585e9eabf59SPeter Wemm 		switch (shdr[i].sh_type) {
586e9eabf59SPeter Wemm 		case SHT_PROGBITS:
587e9eabf59SPeter Wemm 		case SHT_NOBITS:
588*13f28d96SKonstantin Belousov #ifdef __amd64__
589*13f28d96SKonstantin Belousov 		case SHT_AMD64_UNWIND:
590*13f28d96SKonstantin Belousov #endif
59120947801SPeter Wemm 			ef->nprogtab++;
592e9eabf59SPeter Wemm 			break;
593e9eabf59SPeter Wemm 		case SHT_SYMTAB:
594e9eabf59SPeter Wemm 			nsym++;
595e9eabf59SPeter Wemm 			symtabindex = i;
596e9eabf59SPeter Wemm 			symstrindex = shdr[i].sh_link;
597e9eabf59SPeter Wemm 			break;
598e9eabf59SPeter Wemm 		case SHT_REL:
599b474c780SDavid E. O'Brien 			ef->nreltab++;
600e9eabf59SPeter Wemm 			break;
601e9eabf59SPeter Wemm 		case SHT_RELA:
602b474c780SDavid E. O'Brien 			ef->nrelatab++;
603e9eabf59SPeter Wemm 			break;
604e9eabf59SPeter Wemm 		case SHT_STRTAB:
605e9eabf59SPeter Wemm 			break;
606e9eabf59SPeter Wemm 		}
607e9eabf59SPeter Wemm 	}
60820947801SPeter Wemm 	if (ef->nprogtab == 0) {
609552f9f63SEdwin Groothuis 		link_elf_error(filename, "file has no contents");
610e9eabf59SPeter Wemm 		error = ENOEXEC;
611e9eabf59SPeter Wemm 		goto out;
612e9eabf59SPeter Wemm 	}
613e9eabf59SPeter Wemm 	if (nsym != 1) {
614e9eabf59SPeter Wemm 		/* Only allow one symbol table for now */
615552f9f63SEdwin Groothuis 		link_elf_error(filename, "file has no valid symbol table");
616e9eabf59SPeter Wemm 		error = ENOEXEC;
617e9eabf59SPeter Wemm 		goto out;
618e9eabf59SPeter Wemm 	}
619e9eabf59SPeter Wemm 	if (symstrindex < 0 || symstrindex > hdr->e_shnum ||
620e9eabf59SPeter Wemm 	    shdr[symstrindex].sh_type != SHT_STRTAB) {
621552f9f63SEdwin Groothuis 		link_elf_error(filename, "file has invalid symbol strings");
622e9eabf59SPeter Wemm 		error = ENOEXEC;
623e9eabf59SPeter Wemm 		goto out;
624e9eabf59SPeter Wemm 	}
625e9eabf59SPeter Wemm 
626e9eabf59SPeter Wemm 	/* Allocate space for tracking the load chunks */
627e9eabf59SPeter Wemm 	if (ef->nprogtab != 0)
62820947801SPeter Wemm 		ef->progtab = malloc(ef->nprogtab * sizeof(*ef->progtab),
62920947801SPeter Wemm 		    M_LINKER, M_WAITOK | M_ZERO);
630b474c780SDavid E. O'Brien 	if (ef->nreltab != 0)
631b474c780SDavid E. O'Brien 		ef->reltab = malloc(ef->nreltab * sizeof(*ef->reltab),
632b474c780SDavid E. O'Brien 		    M_LINKER, M_WAITOK | M_ZERO);
633b474c780SDavid E. O'Brien 	if (ef->nrelatab != 0)
634b474c780SDavid E. O'Brien 		ef->relatab = malloc(ef->nrelatab * sizeof(*ef->relatab),
635b474c780SDavid E. O'Brien 		    M_LINKER, M_WAITOK | M_ZERO);
636e9eabf59SPeter Wemm 
63720947801SPeter Wemm 	if (symtabindex == -1)
63820947801SPeter Wemm 		panic("lost symbol table index");
63920947801SPeter Wemm 	/* Allocate space for and load the symbol table */
640e9eabf59SPeter Wemm 	ef->ddbsymcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym);
641e9eabf59SPeter Wemm 	ef->ddbsymtab = malloc(shdr[symtabindex].sh_size, M_LINKER, M_WAITOK);
64220947801SPeter Wemm 	error = vn_rdwr(UIO_READ, nd.ni_vp, (void *)ef->ddbsymtab,
64320947801SPeter Wemm 	    shdr[symtabindex].sh_size, shdr[symtabindex].sh_offset,
64420947801SPeter Wemm 	    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
64520947801SPeter Wemm 	    &resid, td);
64620947801SPeter Wemm 	if (error)
64720947801SPeter Wemm 		goto out;
64820947801SPeter Wemm 	if (resid != 0){
64920947801SPeter Wemm 		error = EINVAL;
65020947801SPeter Wemm 		goto out;
65120947801SPeter Wemm 	}
652e9eabf59SPeter Wemm 
65320947801SPeter Wemm 	if (symstrindex == -1)
65420947801SPeter Wemm 		panic("lost symbol string index");
65520947801SPeter Wemm 	/* Allocate space for and load the symbol strings */
656e9eabf59SPeter Wemm 	ef->ddbstrcnt = shdr[symstrindex].sh_size;
657e9eabf59SPeter Wemm 	ef->ddbstrtab = malloc(shdr[symstrindex].sh_size, M_LINKER, M_WAITOK);
65820947801SPeter Wemm 	error = vn_rdwr(UIO_READ, nd.ni_vp, ef->ddbstrtab,
65920947801SPeter Wemm 	    shdr[symstrindex].sh_size, shdr[symstrindex].sh_offset,
66020947801SPeter Wemm 	    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
66120947801SPeter Wemm 	    &resid, td);
66220947801SPeter Wemm 	if (error)
66320947801SPeter Wemm 		goto out;
66420947801SPeter Wemm 	if (resid != 0){
66520947801SPeter Wemm 		error = EINVAL;
66620947801SPeter Wemm 		goto out;
66720947801SPeter Wemm 	}
668e9eabf59SPeter Wemm 
669e9eabf59SPeter Wemm 	/* Do we have a string table for the section names?  */
670e9eabf59SPeter Wemm 	shstrindex = -1;
67120947801SPeter Wemm 	if (hdr->e_shstrndx != 0 &&
67220947801SPeter Wemm 	    shdr[hdr->e_shstrndx].sh_type == SHT_STRTAB) {
673e9eabf59SPeter Wemm 		shstrindex = hdr->e_shstrndx;
674e9eabf59SPeter Wemm 		ef->shstrcnt = shdr[shstrindex].sh_size;
67520947801SPeter Wemm 		ef->shstrtab = malloc(shdr[shstrindex].sh_size, M_LINKER,
67620947801SPeter Wemm 		    M_WAITOK);
67720947801SPeter Wemm 		error = vn_rdwr(UIO_READ, nd.ni_vp, ef->shstrtab,
67820947801SPeter Wemm 		    shdr[shstrindex].sh_size, shdr[shstrindex].sh_offset,
679e9eabf59SPeter Wemm 		    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
680e9eabf59SPeter Wemm 		    &resid, td);
681e9eabf59SPeter Wemm 		if (error)
682e9eabf59SPeter Wemm 			goto out;
68320947801SPeter Wemm 		if (resid != 0){
68420947801SPeter Wemm 			error = EINVAL;
68520947801SPeter Wemm 			goto out;
68620947801SPeter Wemm 		}
687e9eabf59SPeter Wemm 	}
688e9eabf59SPeter Wemm 
68920947801SPeter Wemm 	/* Size up code/data(progbits) and bss(nobits). */
690e9eabf59SPeter Wemm 	alignmask = 0;
691e9eabf59SPeter Wemm 	for (i = 0; i < hdr->e_shnum; i++) {
692676799a0SAndriy Gapon 		if (shdr[i].sh_size == 0)
693676799a0SAndriy Gapon 			continue;
694e9eabf59SPeter Wemm 		switch (shdr[i].sh_type) {
695e9eabf59SPeter Wemm 		case SHT_PROGBITS:
696e9eabf59SPeter Wemm 		case SHT_NOBITS:
697*13f28d96SKonstantin Belousov #ifdef __amd64__
698*13f28d96SKonstantin Belousov 		case SHT_AMD64_UNWIND:
699*13f28d96SKonstantin Belousov #endif
700e9eabf59SPeter Wemm 			alignmask = shdr[i].sh_addralign - 1;
701e9eabf59SPeter Wemm 			mapsize += alignmask;
702e9eabf59SPeter Wemm 			mapsize &= ~alignmask;
703e9eabf59SPeter Wemm 			mapsize += shdr[i].sh_size;
704e9eabf59SPeter Wemm 			break;
705e9eabf59SPeter Wemm 		}
706e9eabf59SPeter Wemm 	}
707e9eabf59SPeter Wemm 
708e9eabf59SPeter Wemm 	/*
709e9eabf59SPeter Wemm 	 * We know how much space we need for the text/data/bss/etc.
710e9eabf59SPeter Wemm 	 * This stuff needs to be in a single chunk so that profiling etc
711e9eabf59SPeter Wemm 	 * can get the bounds and gdb can associate offsets with modules
712e9eabf59SPeter Wemm 	 */
71320947801SPeter Wemm 	ef->object = vm_object_allocate(OBJT_DEFAULT,
71420947801SPeter Wemm 	    round_page(mapsize) >> PAGE_SHIFT);
715fe3db7c7SDoug Rabson 	if (ef->object == NULL) {
716fe3db7c7SDoug Rabson 		error = ENOMEM;
717fe3db7c7SDoug Rabson 		goto out;
718fe3db7c7SDoug Rabson 	}
719fe3db7c7SDoug Rabson 	ef->address = (caddr_t) vm_map_min(kernel_map);
720ac68d1c9SAlan Cox 
721ac68d1c9SAlan Cox 	/*
722ac68d1c9SAlan Cox 	 * In order to satisfy amd64's architectural requirements on the
723ac68d1c9SAlan Cox 	 * location of code and data in the kernel's address space, request a
724ac68d1c9SAlan Cox 	 * mapping that is above the kernel.
725ac68d1c9SAlan Cox 	 */
7264104e835SOleksandr Tymoshenko #ifdef __amd64__
727ac68d1c9SAlan Cox 	mapbase = KERNBASE;
7284104e835SOleksandr Tymoshenko #else
7294104e835SOleksandr Tymoshenko 	mapbase = VM_MIN_KERNEL_ADDRESS;
7304104e835SOleksandr Tymoshenko #endif
73120947801SPeter Wemm 	error = vm_map_find(kernel_map, ef->object, 0, &mapbase,
732edb572a3SJohn Baldwin 	    round_page(mapsize), 0, VMFS_OPTIMAL_SPACE, VM_PROT_ALL,
733edb572a3SJohn Baldwin 	    VM_PROT_ALL, 0);
734fe3db7c7SDoug Rabson 	if (error) {
735fe3db7c7SDoug Rabson 		vm_object_deallocate(ef->object);
736326e27d8SDoug Rabson 		ef->object = 0;
737fe3db7c7SDoug Rabson 		goto out;
738fe3db7c7SDoug Rabson 	}
73920947801SPeter Wemm 
740e9eabf59SPeter Wemm 	/* Wire the pages */
7417f1ef325SAlan Cox 	error = vm_map_wire(kernel_map, mapbase,
74220947801SPeter Wemm 	    mapbase + round_page(mapsize),
743e9eabf59SPeter Wemm 	    VM_MAP_WIRE_SYSTEM|VM_MAP_WIRE_NOHOLES);
7447f1ef325SAlan Cox 	if (error != KERN_SUCCESS) {
7457f1ef325SAlan Cox 		error = ENOMEM;
7467f1ef325SAlan Cox 		goto out;
7477f1ef325SAlan Cox 	}
748a4f67738SDoug Rabson 
74920947801SPeter Wemm 	/* Inform the kld system about the situation */
75020947801SPeter Wemm 	lf->address = ef->address = (caddr_t)mapbase;
75120947801SPeter Wemm 	lf->size = mapsize;
752e9eabf59SPeter Wemm 
753e9eabf59SPeter Wemm 	/*
75420947801SPeter Wemm 	 * Now load code/data(progbits), zero bss(nobits), allocate space for
75520947801SPeter Wemm 	 * and load relocs
756e9eabf59SPeter Wemm 	 */
75720947801SPeter Wemm 	pb = 0;
75820947801SPeter Wemm 	rl = 0;
75920947801SPeter Wemm 	ra = 0;
76020947801SPeter Wemm 	alignmask = 0;
76120947801SPeter Wemm 	for (i = 0; i < hdr->e_shnum; i++) {
762676799a0SAndriy Gapon 		if (shdr[i].sh_size == 0)
763676799a0SAndriy Gapon 			continue;
76420947801SPeter Wemm 		switch (shdr[i].sh_type) {
76520947801SPeter Wemm 		case SHT_PROGBITS:
76620947801SPeter Wemm 		case SHT_NOBITS:
767*13f28d96SKonstantin Belousov #ifdef __amd64__
768*13f28d96SKonstantin Belousov 		case SHT_AMD64_UNWIND:
769*13f28d96SKonstantin Belousov #endif
77020947801SPeter Wemm 			alignmask = shdr[i].sh_addralign - 1;
77120947801SPeter Wemm 			mapbase += alignmask;
77220947801SPeter Wemm 			mapbase &= ~alignmask;
7730067051fSMarcel Moolenaar 			if (ef->shstrtab != NULL && shdr[i].sh_name != 0) {
77450c202c5SJeff Roberson 				ef->progtab[pb].name =
77550c202c5SJeff Roberson 				    ef->shstrtab + shdr[i].sh_name;
7760067051fSMarcel Moolenaar 				if (!strcmp(ef->progtab[pb].name, ".ctors")) {
7770067051fSMarcel Moolenaar 					lf->ctors_addr = (caddr_t)mapbase;
7780067051fSMarcel Moolenaar 					lf->ctors_size = shdr[i].sh_size;
7790067051fSMarcel Moolenaar 				}
7800067051fSMarcel Moolenaar 			} else if (shdr[i].sh_type == SHT_PROGBITS)
78120947801SPeter Wemm 				ef->progtab[pb].name = "<<PROGBITS>>";
782*13f28d96SKonstantin Belousov #ifdef __amd64__
783*13f28d96SKonstantin Belousov 			else if (shdr[i].sh_type == SHT_AMD64_UNWIND)
784*13f28d96SKonstantin Belousov 				ef->progtab[pb].name = "<<UNWIND>>";
785*13f28d96SKonstantin Belousov #endif
78650c202c5SJeff Roberson 			else
78750c202c5SJeff Roberson 				ef->progtab[pb].name = "<<NOBITS>>";
78850c202c5SJeff Roberson 			if (ef->progtab[pb].name != NULL &&
7895f67450dSDimitry Andric 			    !strcmp(ef->progtab[pb].name, DPCPU_SETNAME))
79050c202c5SJeff Roberson 				ef->progtab[pb].addr =
79150c202c5SJeff Roberson 				    dpcpu_alloc(shdr[i].sh_size);
792eddfbb76SRobert Watson #ifdef VIMAGE
793eddfbb76SRobert Watson 			else if (ef->progtab[pb].name != NULL &&
79417ef1febSRobert Watson 			    !strcmp(ef->progtab[pb].name, VNET_SETNAME))
795eddfbb76SRobert Watson 				ef->progtab[pb].addr =
796eddfbb76SRobert Watson 				    vnet_data_alloc(shdr[i].sh_size);
797eddfbb76SRobert Watson #endif
79850c202c5SJeff Roberson 			else
79950c202c5SJeff Roberson 				ef->progtab[pb].addr =
80050c202c5SJeff Roberson 				    (void *)(uintptr_t)mapbase;
80150c202c5SJeff Roberson 			if (ef->progtab[pb].addr == NULL) {
80250c202c5SJeff Roberson 				error = ENOSPC;
80350c202c5SJeff Roberson 				goto out;
80450c202c5SJeff Roberson 			}
80550c202c5SJeff Roberson 			ef->progtab[pb].size = shdr[i].sh_size;
80650c202c5SJeff Roberson 			ef->progtab[pb].sec = i;
807*13f28d96SKonstantin Belousov 			if (shdr[i].sh_type == SHT_PROGBITS
808*13f28d96SKonstantin Belousov #ifdef __amd64__
809*13f28d96SKonstantin Belousov 			    || shdr[i].sh_type == SHT_AMD64_UNWIND
810*13f28d96SKonstantin Belousov #endif
811*13f28d96SKonstantin Belousov 			    ) {
812e9eabf59SPeter Wemm 				error = vn_rdwr(UIO_READ, nd.ni_vp,
81320947801SPeter Wemm 				    ef->progtab[pb].addr,
81420947801SPeter Wemm 				    shdr[i].sh_size, shdr[i].sh_offset,
81520947801SPeter Wemm 				    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred,
81620947801SPeter Wemm 				    NOCRED, &resid, td);
81720947801SPeter Wemm 				if (error)
81820947801SPeter Wemm 					goto out;
81920947801SPeter Wemm 				if (resid != 0){
82020947801SPeter Wemm 					error = EINVAL;
82120947801SPeter Wemm 					goto out;
82220947801SPeter Wemm 				}
823eddfbb76SRobert Watson 				/* Initialize the per-cpu or vnet area. */
824eddfbb76SRobert Watson 				if (ef->progtab[pb].addr != (void *)mapbase &&
8255f67450dSDimitry Andric 				    !strcmp(ef->progtab[pb].name, DPCPU_SETNAME))
82650c202c5SJeff Roberson 					dpcpu_copy(ef->progtab[pb].addr,
82750c202c5SJeff Roberson 					    shdr[i].sh_size);
828eddfbb76SRobert Watson #ifdef VIMAGE
829eddfbb76SRobert Watson 				else if (ef->progtab[pb].addr !=
830eddfbb76SRobert Watson 				    (void *)mapbase &&
83117ef1febSRobert Watson 				    !strcmp(ef->progtab[pb].name, VNET_SETNAME))
832eddfbb76SRobert Watson 					vnet_data_copy(ef->progtab[pb].addr,
833eddfbb76SRobert Watson 					    shdr[i].sh_size);
834eddfbb76SRobert Watson #endif
83550c202c5SJeff Roberson 			} else
83620947801SPeter Wemm 				bzero(ef->progtab[pb].addr, shdr[i].sh_size);
837a8774e39SPeter Wemm 
838a8774e39SPeter Wemm 			/* Update all symbol values with the offset. */
839a8774e39SPeter Wemm 			for (j = 0; j < ef->ddbsymcnt; j++) {
840a8774e39SPeter Wemm 				es = &ef->ddbsymtab[j];
841a8774e39SPeter Wemm 				if (es->st_shndx != i)
842a8774e39SPeter Wemm 					continue;
843a8774e39SPeter Wemm 				es->st_value += (Elf_Addr)ef->progtab[pb].addr;
844a8774e39SPeter Wemm 			}
84520947801SPeter Wemm 			mapbase += shdr[i].sh_size;
84620947801SPeter Wemm 			pb++;
84720947801SPeter Wemm 			break;
84820947801SPeter Wemm 		case SHT_REL:
84920947801SPeter Wemm 			ef->reltab[rl].rel = malloc(shdr[i].sh_size, M_LINKER,
85020947801SPeter Wemm 			    M_WAITOK);
85120947801SPeter Wemm 			ef->reltab[rl].nrel = shdr[i].sh_size / sizeof(Elf_Rel);
85220947801SPeter Wemm 			ef->reltab[rl].sec = shdr[i].sh_info;
85320947801SPeter Wemm 			error = vn_rdwr(UIO_READ, nd.ni_vp,
85420947801SPeter Wemm 			    (void *)ef->reltab[rl].rel,
85520947801SPeter Wemm 			    shdr[i].sh_size, shdr[i].sh_offset,
856e9eabf59SPeter Wemm 			    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
857e9eabf59SPeter Wemm 			    &resid, td);
858e9eabf59SPeter Wemm 			if (error)
859e9eabf59SPeter Wemm 				goto out;
86020947801SPeter Wemm 			if (resid != 0){
86120947801SPeter Wemm 				error = EINVAL;
86220947801SPeter Wemm 				goto out;
863e9eabf59SPeter Wemm 			}
86420947801SPeter Wemm 			rl++;
86520947801SPeter Wemm 			break;
86620947801SPeter Wemm 		case SHT_RELA:
86720947801SPeter Wemm 			ef->relatab[ra].rela = malloc(shdr[i].sh_size, M_LINKER,
86820947801SPeter Wemm 			    M_WAITOK);
86920947801SPeter Wemm 			ef->relatab[ra].nrela =
87020947801SPeter Wemm 			    shdr[i].sh_size / sizeof(Elf_Rela);
87120947801SPeter Wemm 			ef->relatab[ra].sec = shdr[i].sh_info;
872e9eabf59SPeter Wemm 			error = vn_rdwr(UIO_READ, nd.ni_vp,
87320947801SPeter Wemm 			    (void *)ef->relatab[ra].rela,
87420947801SPeter Wemm 			    shdr[i].sh_size, shdr[i].sh_offset,
875e9eabf59SPeter Wemm 			    UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, NOCRED,
876e9eabf59SPeter Wemm 			    &resid, td);
877e9eabf59SPeter Wemm 			if (error)
878e9eabf59SPeter Wemm 				goto out;
87920947801SPeter Wemm 			if (resid != 0){
88020947801SPeter Wemm 				error = EINVAL;
88120947801SPeter Wemm 				goto out;
882e9eabf59SPeter Wemm 			}
88320947801SPeter Wemm 			ra++;
88420947801SPeter Wemm 			break;
88520947801SPeter Wemm 		}
88620947801SPeter Wemm 	}
88720947801SPeter Wemm 	if (pb != ef->nprogtab)
88820947801SPeter Wemm 		panic("lost progbits");
889b474c780SDavid E. O'Brien 	if (rl != ef->nreltab)
890b474c780SDavid E. O'Brien 		panic("lost reltab");
891b474c780SDavid E. O'Brien 	if (ra != ef->nrelatab)
892b474c780SDavid E. O'Brien 		panic("lost relatab");
89320947801SPeter Wemm 	if (mapbase != (vm_offset_t)ef->address + mapsize)
8945b201fdcSPeter Wemm 		panic("mapbase 0x%lx != address %p + mapsize 0x%lx (0x%lx)\n",
8956f3c6327SNeel Natu 		    (u_long)mapbase, ef->address, (u_long)mapsize,
8966f3c6327SNeel Natu 		    (u_long)(vm_offset_t)ef->address + mapsize);
897e9eabf59SPeter Wemm 
898e9eabf59SPeter Wemm 	/* Local intra-module relocations */
8997251b4bfSJake Burkholder 	link_elf_reloc_local(lf);
9007251b4bfSJake Burkholder 
901e9eabf59SPeter Wemm 	/* Pull in dependencies */
9024f7afc20SKonstantin Belousov 	VOP_UNLOCK(nd.ni_vp, 0);
9037b9716baSIan Dowse 	error = linker_load_dependencies(lf);
9044f7afc20SKonstantin Belousov 	vn_lock(nd.ni_vp, LK_EXCLUSIVE | LK_RETRY);
905ca65d5c7SPeter Wemm 	if (error)
906de78ca7eSPeter Wemm 		goto out;
907e9eabf59SPeter Wemm 
908e9eabf59SPeter Wemm 	/* External relocations */
909326e27d8SDoug Rabson 	error = relocate_file(ef);
910ca65d5c7SPeter Wemm 	if (error)
911ca65d5c7SPeter Wemm 		goto out;
912ca65d5c7SPeter Wemm 
913e9eabf59SPeter Wemm 	/* Notify MD code that a module is being loaded. */
914e9eabf59SPeter Wemm 	error = elf_cpu_load_file(lf);
915ca65d5c7SPeter Wemm 	if (error)
916ca65d5c7SPeter Wemm 		goto out;
917a4f67738SDoug Rabson 
9180067051fSMarcel Moolenaar 	/* Invoke .ctors */
9190067051fSMarcel Moolenaar 	link_elf_invoke_ctors(lf->ctors_addr, lf->ctors_size);
9200067051fSMarcel Moolenaar 
921a4f67738SDoug Rabson 	*result = lf;
922a4f67738SDoug Rabson 
923a4f67738SDoug Rabson out:
924f1d2d305SKonstantin Belousov 	VOP_UNLOCK(nd.ni_vp, 0);
925f1d2d305SKonstantin Belousov 	vn_close(nd.ni_vp, FREAD, td->td_ucred, td);
926ca65d5c7SPeter Wemm 	if (error && lf)
92765a311fcSPoul-Henning Kamp 		linker_file_unload(lf, LINKER_UNLOAD_FORCE);
928e9eabf59SPeter Wemm 	free(hdr, M_LINKER);
929a4f67738SDoug Rabson 
930a4f67738SDoug Rabson 	return error;
931a4f67738SDoug Rabson }
932a4f67738SDoug Rabson 
933a4f67738SDoug Rabson static void
934de78ca7eSPeter Wemm link_elf_unload_file(linker_file_t file)
935a4f67738SDoug Rabson {
936326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t) file;
93720947801SPeter Wemm 	int i;
938a4f67738SDoug Rabson 
9391aeb23cdSMarcel Moolenaar 	/* Notify MD code that a module is being unloaded. */
9401aeb23cdSMarcel Moolenaar 	elf_cpu_unload_file(file);
9411aeb23cdSMarcel Moolenaar 
94250c202c5SJeff Roberson 	if (ef->progtab) {
94350c202c5SJeff Roberson 		for (i = 0; i < ef->nprogtab; i++) {
94450c202c5SJeff Roberson 			if (ef->progtab[i].size == 0)
94550c202c5SJeff Roberson 				continue;
94650c202c5SJeff Roberson 			if (ef->progtab[i].name == NULL)
94750c202c5SJeff Roberson 				continue;
9485f67450dSDimitry Andric 			if (!strcmp(ef->progtab[i].name, DPCPU_SETNAME))
94950c202c5SJeff Roberson 				dpcpu_free(ef->progtab[i].addr,
95050c202c5SJeff Roberson 				    ef->progtab[i].size);
951eddfbb76SRobert Watson #ifdef VIMAGE
95217ef1febSRobert Watson 			else if (!strcmp(ef->progtab[i].name, VNET_SETNAME))
953eddfbb76SRobert Watson 				vnet_data_free(ef->progtab[i].addr,
954eddfbb76SRobert Watson 				    ef->progtab[i].size);
955eddfbb76SRobert Watson #endif
95650c202c5SJeff Roberson 		}
95750c202c5SJeff Roberson 	}
95870b7ffeeSIan Dowse 	if (ef->preloaded) {
95970b7ffeeSIan Dowse 		free(ef->reltab, M_LINKER);
96070b7ffeeSIan Dowse 		free(ef->relatab, M_LINKER);
96170b7ffeeSIan Dowse 		free(ef->progtab, M_LINKER);
962a2024a3eSJohn Birrell 		free(ef->ctftab, M_LINKER);
963a2024a3eSJohn Birrell 		free(ef->ctfoff, M_LINKER);
964a2024a3eSJohn Birrell 		free(ef->typoff, M_LINKER);
96570b7ffeeSIan Dowse 		if (file->filename != NULL)
96670b7ffeeSIan Dowse 			preload_delete_name(file->filename);
96770b7ffeeSIan Dowse 		/* XXX reclaim module memory? */
96870b7ffeeSIan Dowse 		return;
96970b7ffeeSIan Dowse 	}
97070b7ffeeSIan Dowse 
971b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nreltab; i++)
97220947801SPeter Wemm 		free(ef->reltab[i].rel, M_LINKER);
973b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nrelatab; i++)
97420947801SPeter Wemm 		free(ef->relatab[i].rela, M_LINKER);
97520947801SPeter Wemm 	free(ef->reltab, M_LINKER);
97620947801SPeter Wemm 	free(ef->relatab, M_LINKER);
97720947801SPeter Wemm 	free(ef->progtab, M_LINKER);
97820947801SPeter Wemm 
979fe3db7c7SDoug Rabson 	if (ef->object) {
980fe3db7c7SDoug Rabson 		vm_map_remove(kernel_map, (vm_offset_t) ef->address,
98120947801SPeter Wemm 		    (vm_offset_t) ef->address +
98220947801SPeter Wemm 		    (ef->object->size << PAGE_SHIFT));
983fe3db7c7SDoug Rabson 	}
984e9eabf59SPeter Wemm 	free(ef->e_shdr, M_LINKER);
985e9eabf59SPeter Wemm 	free(ef->ddbsymtab, M_LINKER);
986e9eabf59SPeter Wemm 	free(ef->ddbstrtab, M_LINKER);
98720947801SPeter Wemm 	free(ef->shstrtab, M_LINKER);
988a2024a3eSJohn Birrell 	free(ef->ctftab, M_LINKER);
989a2024a3eSJohn Birrell 	free(ef->ctfoff, M_LINKER);
990a2024a3eSJohn Birrell 	free(ef->typoff, M_LINKER);
991de78ca7eSPeter Wemm }
992de78ca7eSPeter Wemm 
993fe3db7c7SDoug Rabson static const char *
994757686b1SMarcel Moolenaar symbol_name(elf_file_t ef, Elf_Size r_info)
995a4f67738SDoug Rabson {
996fe3db7c7SDoug Rabson 	const Elf_Sym *ref;
997a4f67738SDoug Rabson 
998aa855a59SPeter Wemm 	if (ELF_R_SYM(r_info)) {
999e9eabf59SPeter Wemm 		ref = ef->ddbsymtab + ELF_R_SYM(r_info);
1000e9eabf59SPeter Wemm 		return ef->ddbstrtab + ref->st_name;
1001fe3db7c7SDoug Rabson 	} else
1002fe3db7c7SDoug Rabson 		return NULL;
1003a4f67738SDoug Rabson }
1004a4f67738SDoug Rabson 
1005e9eabf59SPeter Wemm static Elf_Addr
1006e9eabf59SPeter Wemm findbase(elf_file_t ef, int sec)
1007e9eabf59SPeter Wemm {
1008e9eabf59SPeter Wemm 	int i;
1009e9eabf59SPeter Wemm 	Elf_Addr base = 0;
1010e9eabf59SPeter Wemm 
1011e9eabf59SPeter Wemm 	for (i = 0; i < ef->nprogtab; i++) {
1012a8774e39SPeter Wemm 		if (sec == ef->progtab[i].sec) {
1013e9eabf59SPeter Wemm 			base = (Elf_Addr)ef->progtab[i].addr;
1014a8774e39SPeter Wemm 			break;
1015e9eabf59SPeter Wemm 		}
1016a8774e39SPeter Wemm 	}
1017e9eabf59SPeter Wemm 	return base;
1018e9eabf59SPeter Wemm }
1019e9eabf59SPeter Wemm 
1020a4f67738SDoug Rabson static int
1021326e27d8SDoug Rabson relocate_file(elf_file_t ef)
1022a4f67738SDoug Rabson {
1023fe3db7c7SDoug Rabson 	const Elf_Rel *rellim;
1024fe3db7c7SDoug Rabson 	const Elf_Rel *rel;
1025fe3db7c7SDoug Rabson 	const Elf_Rela *relalim;
1026fe3db7c7SDoug Rabson 	const Elf_Rela *rela;
1027aa855a59SPeter Wemm 	const char *symname;
1028e9eabf59SPeter Wemm 	const Elf_Sym *sym;
1029e9eabf59SPeter Wemm 	int i;
1030757686b1SMarcel Moolenaar 	Elf_Size symidx;
1031e9eabf59SPeter Wemm 	Elf_Addr base;
1032e9eabf59SPeter Wemm 
1033a4f67738SDoug Rabson 
1034fe3db7c7SDoug Rabson 	/* Perform relocations without addend if there are any: */
1035b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nreltab; i++) {
1036e9eabf59SPeter Wemm 		rel = ef->reltab[i].rel;
103720947801SPeter Wemm 		if (rel == NULL)
103882d0d1a0SPeter Wemm 			panic("lost a reltab!");
103920947801SPeter Wemm 		rellim = rel + ef->reltab[i].nrel;
1040e9eabf59SPeter Wemm 		base = findbase(ef, ef->reltab[i].sec);
1041a8774e39SPeter Wemm 		if (base == 0)
1042a8774e39SPeter Wemm 			panic("lost base for reltab");
104320947801SPeter Wemm 		for ( ; rel < rellim; rel++) {
1044e9eabf59SPeter Wemm 			symidx = ELF_R_SYM(rel->r_info);
104520947801SPeter Wemm 			if (symidx >= ef->ddbsymcnt)
104620947801SPeter Wemm 				continue;
1047e9eabf59SPeter Wemm 			sym = ef->ddbsymtab + symidx;
104820947801SPeter Wemm 			/* Local relocs are already done */
104920947801SPeter Wemm 			if (ELF_ST_BIND(sym->st_info) == STB_LOCAL)
105020947801SPeter Wemm 				continue;
105120947801SPeter Wemm 			if (elf_reloc(&ef->lf, base, rel, ELF_RELOC_REL,
105220947801SPeter Wemm 			    elf_obj_lookup)) {
1053aa855a59SPeter Wemm 				symname = symbol_name(ef, rel->r_info);
105420947801SPeter Wemm 				printf("link_elf_obj: symbol %s undefined\n",
105520947801SPeter Wemm 				    symname);
1056a4f67738SDoug Rabson 				return ENOENT;
1057a13ddfb6SPeter Wemm 			}
1058aa855a59SPeter Wemm 		}
1059a4f67738SDoug Rabson 	}
1060a4f67738SDoug Rabson 
1061fe3db7c7SDoug Rabson 	/* Perform relocations with addend if there are any: */
1062b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nrelatab; i++) {
1063e9eabf59SPeter Wemm 		rela = ef->relatab[i].rela;
106420947801SPeter Wemm 		if (rela == NULL)
106582d0d1a0SPeter Wemm 			panic("lost a relatab!");
106620947801SPeter Wemm 		relalim = rela + ef->relatab[i].nrela;
1067e9eabf59SPeter Wemm 		base = findbase(ef, ef->relatab[i].sec);
1068a8774e39SPeter Wemm 		if (base == 0)
1069a8774e39SPeter Wemm 			panic("lost base for relatab");
107020947801SPeter Wemm 		for ( ; rela < relalim; rela++) {
1071e9eabf59SPeter Wemm 			symidx = ELF_R_SYM(rela->r_info);
107220947801SPeter Wemm 			if (symidx >= ef->ddbsymcnt)
107320947801SPeter Wemm 				continue;
1074e9eabf59SPeter Wemm 			sym = ef->ddbsymtab + symidx;
107520947801SPeter Wemm 			/* Local relocs are already done */
107620947801SPeter Wemm 			if (ELF_ST_BIND(sym->st_info) == STB_LOCAL)
107720947801SPeter Wemm 				continue;
107820947801SPeter Wemm 			if (elf_reloc(&ef->lf, base, rela, ELF_RELOC_RELA,
107920947801SPeter Wemm 			    elf_obj_lookup)) {
1080aa855a59SPeter Wemm 				symname = symbol_name(ef, rela->r_info);
108120947801SPeter Wemm 				printf("link_elf_obj: symbol %s undefined\n",
108220947801SPeter Wemm 				    symname);
1083fe3db7c7SDoug Rabson 				return ENOENT;
1084a13ddfb6SPeter Wemm 			}
1085e9eabf59SPeter Wemm 		}
1086e9eabf59SPeter Wemm 	}
1087a4f67738SDoug Rabson 
10882832cd54SKonstantin Belousov 	/*
10892832cd54SKonstantin Belousov 	 * Only clean SHN_FBSD_CACHED for successfull return.  If we
10902832cd54SKonstantin Belousov 	 * modified symbol table for the object but found an
10912832cd54SKonstantin Belousov 	 * unresolved symbol, there is no reason to roll back.
10922832cd54SKonstantin Belousov 	 */
10932832cd54SKonstantin Belousov 	elf_obj_cleanup_globals_cache(ef);
10942832cd54SKonstantin Belousov 
1095a4f67738SDoug Rabson 	return 0;
1096a4f67738SDoug Rabson }
1097a4f67738SDoug Rabson 
109837c84183SPoul-Henning Kamp static int
1099d254af07SMatthew Dillon link_elf_lookup_symbol(linker_file_t lf, const char *name, c_linker_sym_t *sym)
1100a4f67738SDoug Rabson {
1101326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t) lf;
11022d636ab0SPeter Wemm 	const Elf_Sym *symp;
11032d636ab0SPeter Wemm 	const char *strp;
1104a4f67738SDoug Rabson 	int i;
1105a4f67738SDoug Rabson 
11062d636ab0SPeter Wemm 	for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
11072d636ab0SPeter Wemm 		strp = ef->ddbstrtab + symp->st_name;
1108a8774e39SPeter Wemm 		if (symp->st_shndx != SHN_UNDEF && strcmp(name, strp) == 0) {
1109d254af07SMatthew Dillon 			*sym = (c_linker_sym_t) symp;
11102d636ab0SPeter Wemm 			return 0;
11112d636ab0SPeter Wemm 		}
11122d636ab0SPeter Wemm 	}
1113a4f67738SDoug Rabson 	return ENOENT;
1114a4f67738SDoug Rabson }
1115a4f67738SDoug Rabson 
1116de78ca7eSPeter Wemm static int
111720947801SPeter Wemm link_elf_symbol_values(linker_file_t lf, c_linker_sym_t sym,
111820947801SPeter Wemm     linker_symval_t *symval)
1119a4f67738SDoug Rabson {
1120326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t) lf;
11218aef1712SMatthew Dillon 	const Elf_Sym *es = (const Elf_Sym*) sym;
1122a4f67738SDoug Rabson 
11235cf87418SMarcel Moolenaar 	if (es >= ef->ddbsymtab && es < (ef->ddbsymtab + ef->ddbsymcnt)) {
11242d636ab0SPeter Wemm 		symval->name = ef->ddbstrtab + es->st_name;
1125a8774e39SPeter Wemm 		symval->value = (caddr_t)es->st_value;
11262d636ab0SPeter Wemm 		symval->size = es->st_size;
11272d636ab0SPeter Wemm 		return 0;
11282d636ab0SPeter Wemm 	}
11292d636ab0SPeter Wemm 	return ENOENT;
11302d636ab0SPeter Wemm }
1131a4f67738SDoug Rabson 
1132a4f67738SDoug Rabson static int
1133a4f67738SDoug Rabson link_elf_search_symbol(linker_file_t lf, caddr_t value,
1134d254af07SMatthew Dillon     c_linker_sym_t *sym, long *diffp)
1135a4f67738SDoug Rabson {
1136326e27d8SDoug Rabson 	elf_file_t ef = (elf_file_t) lf;
1137586453feSBruce Evans 	u_long off = (uintptr_t) (void *) value;
1138a4f67738SDoug Rabson 	u_long diff = off;
1139586453feSBruce Evans 	u_long st_value;
1140fe3db7c7SDoug Rabson 	const Elf_Sym *es;
1141fe3db7c7SDoug Rabson 	const Elf_Sym *best = 0;
1142a4f67738SDoug Rabson 	int i;
1143a4f67738SDoug Rabson 
11442d636ab0SPeter Wemm 	for (i = 0, es = ef->ddbsymtab; i < ef->ddbsymcnt; i++, es++) {
1145a4f67738SDoug Rabson 		if (es->st_name == 0)
1146a4f67738SDoug Rabson 			continue;
1147a8774e39SPeter Wemm 		st_value = es->st_value;
1148b5abfb70SPeter Wemm 		if (off >= st_value) {
1149b5abfb70SPeter Wemm 			if (off - st_value < diff) {
1150b5abfb70SPeter Wemm 				diff = off - st_value;
1151a4f67738SDoug Rabson 				best = es;
1152a4f67738SDoug Rabson 				if (diff == 0)
1153a4f67738SDoug Rabson 					break;
1154b5abfb70SPeter Wemm 			} else if (off - st_value == diff) {
1155a4f67738SDoug Rabson 				best = es;
1156a4f67738SDoug Rabson 			}
1157a4f67738SDoug Rabson 		}
1158a4f67738SDoug Rabson 	}
1159a4f67738SDoug Rabson 	if (best == 0)
1160a4f67738SDoug Rabson 		*diffp = off;
1161a4f67738SDoug Rabson 	else
1162a4f67738SDoug Rabson 		*diffp = diff;
1163d254af07SMatthew Dillon 	*sym = (c_linker_sym_t) best;
1164a4f67738SDoug Rabson 
1165a4f67738SDoug Rabson 	return 0;
1166a4f67738SDoug Rabson }
1167f41325dbSPeter Wemm 
1168f41325dbSPeter Wemm /*
1169f41325dbSPeter Wemm  * Look up a linker set on an ELF system.
1170f41325dbSPeter Wemm  */
1171f41325dbSPeter Wemm static int
1172f41325dbSPeter Wemm link_elf_lookup_set(linker_file_t lf, const char *name,
1173f41325dbSPeter Wemm     void ***startp, void ***stopp, int *countp)
1174f41325dbSPeter Wemm {
1175e9eabf59SPeter Wemm 	elf_file_t ef = (elf_file_t)lf;
1176f41325dbSPeter Wemm 	void **start, **stop;
1177e9eabf59SPeter Wemm 	int i, count;
1178f41325dbSPeter Wemm 
1179e9eabf59SPeter Wemm 	/* Relative to section number */
1180e9eabf59SPeter Wemm 	for (i = 0; i < ef->nprogtab; i++) {
1181e9eabf59SPeter Wemm 		if ((strncmp(ef->progtab[i].name, "set_", 4) == 0) &&
1182e9eabf59SPeter Wemm 		    strcmp(ef->progtab[i].name + 4, name) == 0) {
1183e9eabf59SPeter Wemm 			start  = (void **)ef->progtab[i].addr;
118420947801SPeter Wemm 			stop = (void **)((char *)ef->progtab[i].addr +
118520947801SPeter Wemm 			    ef->progtab[i].size);
1186f41325dbSPeter Wemm 			count = stop - start;
1187f41325dbSPeter Wemm 			if (startp)
1188f41325dbSPeter Wemm 				*startp = start;
1189f41325dbSPeter Wemm 			if (stopp)
1190f41325dbSPeter Wemm 				*stopp = stop;
1191f41325dbSPeter Wemm 			if (countp)
1192f41325dbSPeter Wemm 				*countp = count;
1193e9eabf59SPeter Wemm 			return (0);
1194e9eabf59SPeter Wemm 		}
1195e9eabf59SPeter Wemm 	}
1196e9eabf59SPeter Wemm 	return (ESRCH);
1197f41325dbSPeter Wemm }
1198bb9fe9ddSBrian Feldman 
1199bb9fe9ddSBrian Feldman static int
1200bb9fe9ddSBrian Feldman link_elf_each_function_name(linker_file_t file,
1201e9eabf59SPeter Wemm     int (*callback)(const char *, void *), void *opaque)
1202e9eabf59SPeter Wemm {
1203bb9fe9ddSBrian Feldman 	elf_file_t ef = (elf_file_t)file;
1204bb9fe9ddSBrian Feldman 	const Elf_Sym *symp;
1205bb9fe9ddSBrian Feldman 	int i, error;
1206bb9fe9ddSBrian Feldman 
1207bb9fe9ddSBrian Feldman 	/* Exhaustive search */
1208bb9fe9ddSBrian Feldman 	for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
1209bb9fe9ddSBrian Feldman 		if (symp->st_value != 0 &&
1210bb9fe9ddSBrian Feldman 		    ELF_ST_TYPE(symp->st_info) == STT_FUNC) {
1211bb9fe9ddSBrian Feldman 			error = callback(ef->ddbstrtab + symp->st_name, opaque);
1212bb9fe9ddSBrian Feldman 			if (error)
1213bb9fe9ddSBrian Feldman 				return (error);
1214bb9fe9ddSBrian Feldman 		}
1215bb9fe9ddSBrian Feldman 	}
1216bb9fe9ddSBrian Feldman 	return (0);
1217bb9fe9ddSBrian Feldman }
121884201059SMarcel Moolenaar 
1219a2024a3eSJohn Birrell static int
1220a2024a3eSJohn Birrell link_elf_each_function_nameval(linker_file_t file,
1221a2024a3eSJohn Birrell     linker_function_nameval_callback_t callback, void *opaque)
1222a2024a3eSJohn Birrell {
1223a2024a3eSJohn Birrell 	linker_symval_t symval;
1224a2024a3eSJohn Birrell 	elf_file_t ef = (elf_file_t)file;
1225a2024a3eSJohn Birrell 	const Elf_Sym* symp;
1226a2024a3eSJohn Birrell 	int i, error;
1227a2024a3eSJohn Birrell 
1228a2024a3eSJohn Birrell 	/* Exhaustive search */
1229a2024a3eSJohn Birrell 	for (i = 0, symp = ef->ddbsymtab; i < ef->ddbsymcnt; i++, symp++) {
1230a2024a3eSJohn Birrell 		if (symp->st_value != 0 &&
1231a2024a3eSJohn Birrell 		    ELF_ST_TYPE(symp->st_info) == STT_FUNC) {
1232a2024a3eSJohn Birrell 			error = link_elf_symbol_values(file, (c_linker_sym_t) symp, &symval);
1233a2024a3eSJohn Birrell 			if (error)
1234a2024a3eSJohn Birrell 				return (error);
1235a2024a3eSJohn Birrell 			error = callback(file, i, &symval, opaque);
1236a2024a3eSJohn Birrell 			if (error)
1237a2024a3eSJohn Birrell 				return (error);
1238a2024a3eSJohn Birrell 		}
1239a2024a3eSJohn Birrell 	}
1240a2024a3eSJohn Birrell 	return (0);
1241a2024a3eSJohn Birrell }
1242a2024a3eSJohn Birrell 
12432832cd54SKonstantin Belousov static void
12442832cd54SKonstantin Belousov elf_obj_cleanup_globals_cache(elf_file_t ef)
12452832cd54SKonstantin Belousov {
12462832cd54SKonstantin Belousov 	Elf_Sym *sym;
12472832cd54SKonstantin Belousov 	Elf_Size i;
12482832cd54SKonstantin Belousov 
12492832cd54SKonstantin Belousov 	for (i = 0; i < ef->ddbsymcnt; i++) {
12502832cd54SKonstantin Belousov 		sym = ef->ddbsymtab + i;
12512832cd54SKonstantin Belousov 		if (sym->st_shndx == SHN_FBSD_CACHED) {
12522832cd54SKonstantin Belousov 			sym->st_shndx = SHN_UNDEF;
12532832cd54SKonstantin Belousov 			sym->st_value = 0;
12542832cd54SKonstantin Belousov 		}
12552832cd54SKonstantin Belousov 	}
12562832cd54SKonstantin Belousov }
12572832cd54SKonstantin Belousov 
1258d297ad16SMarcel Moolenaar /*
1259d297ad16SMarcel Moolenaar  * Symbol lookup function that can be used when the symbol index is known (ie
1260d297ad16SMarcel Moolenaar  * in relocations). It uses the symbol index instead of doing a fully fledged
1261d297ad16SMarcel Moolenaar  * hash table based lookup when such is valid. For example for local symbols.
1262d297ad16SMarcel Moolenaar  * This is not only more efficient, it's also more correct. It's not always
1263d297ad16SMarcel Moolenaar  * the case that the symbol can be found through the hash table.
1264d297ad16SMarcel Moolenaar  */
1265cff8c6f2SKonstantin Belousov static int
1266cff8c6f2SKonstantin Belousov elf_obj_lookup(linker_file_t lf, Elf_Size symidx, int deps, Elf_Addr *res)
1267d297ad16SMarcel Moolenaar {
1268d297ad16SMarcel Moolenaar 	elf_file_t ef = (elf_file_t)lf;
12692832cd54SKonstantin Belousov 	Elf_Sym *sym;
1270d297ad16SMarcel Moolenaar 	const char *symbol;
1271cff8c6f2SKonstantin Belousov 	Elf_Addr res1;
1272d297ad16SMarcel Moolenaar 
1273d297ad16SMarcel Moolenaar 	/* Don't even try to lookup the symbol if the index is bogus. */
1274cff8c6f2SKonstantin Belousov 	if (symidx >= ef->ddbsymcnt) {
1275cff8c6f2SKonstantin Belousov 		*res = 0;
1276cff8c6f2SKonstantin Belousov 		return (EINVAL);
1277cff8c6f2SKonstantin Belousov 	}
1278d297ad16SMarcel Moolenaar 
1279e9eabf59SPeter Wemm 	sym = ef->ddbsymtab + symidx;
1280d297ad16SMarcel Moolenaar 
12814cec6f5dSPeter Wemm 	/* Quick answer if there is a definition included. */
1282cff8c6f2SKonstantin Belousov 	if (sym->st_shndx != SHN_UNDEF) {
1283cff8c6f2SKonstantin Belousov 		*res = sym->st_value;
1284cff8c6f2SKonstantin Belousov 		return (0);
1285cff8c6f2SKonstantin Belousov 	}
12864cec6f5dSPeter Wemm 
12874cec6f5dSPeter Wemm 	/* If we get here, then it is undefined and needs a lookup. */
12884cec6f5dSPeter Wemm 	switch (ELF_ST_BIND(sym->st_info)) {
12894cec6f5dSPeter Wemm 	case STB_LOCAL:
12904cec6f5dSPeter Wemm 		/* Local, but undefined? huh? */
1291cff8c6f2SKonstantin Belousov 		*res = 0;
1292cff8c6f2SKonstantin Belousov 		return (EINVAL);
1293d297ad16SMarcel Moolenaar 
1294e9eabf59SPeter Wemm 	case STB_GLOBAL:
1295cff8c6f2SKonstantin Belousov 	case STB_WEAK:
1296e9eabf59SPeter Wemm 		/* Relative to Data or Function name */
1297e9eabf59SPeter Wemm 		symbol = ef->ddbstrtab + sym->st_name;
1298d297ad16SMarcel Moolenaar 
1299d297ad16SMarcel Moolenaar 		/* Force a lookup failure if the symbol name is bogus. */
1300cff8c6f2SKonstantin Belousov 		if (*symbol == 0) {
1301cff8c6f2SKonstantin Belousov 			*res = 0;
1302cff8c6f2SKonstantin Belousov 			return (EINVAL);
1303cff8c6f2SKonstantin Belousov 		}
1304cff8c6f2SKonstantin Belousov 		res1 = (Elf_Addr)linker_file_lookup_symbol(lf, symbol, deps);
13052832cd54SKonstantin Belousov 
13062832cd54SKonstantin Belousov 		/*
13072832cd54SKonstantin Belousov 		 * Cache global lookups during module relocation. The failure
13082832cd54SKonstantin Belousov 		 * case is particularly expensive for callers, who must scan
13092832cd54SKonstantin Belousov 		 * through the entire globals table doing strcmp(). Cache to
13102832cd54SKonstantin Belousov 		 * avoid doing such work repeatedly.
13112832cd54SKonstantin Belousov 		 *
13122832cd54SKonstantin Belousov 		 * After relocation is complete, undefined globals will be
13132832cd54SKonstantin Belousov 		 * restored to SHN_UNDEF in elf_obj_cleanup_globals_cache(),
13142832cd54SKonstantin Belousov 		 * above.
13152832cd54SKonstantin Belousov 		 */
1316cff8c6f2SKonstantin Belousov 		if (res1 != 0) {
13172832cd54SKonstantin Belousov 			sym->st_shndx = SHN_FBSD_CACHED;
1318cff8c6f2SKonstantin Belousov 			sym->st_value = res1;
1319cff8c6f2SKonstantin Belousov 			*res = res1;
1320e9eabf59SPeter Wemm 			return (0);
1321cff8c6f2SKonstantin Belousov 		} else if (ELF_ST_BIND(sym->st_info) == STB_WEAK) {
1322cff8c6f2SKonstantin Belousov 			sym->st_value = 0;
1323cff8c6f2SKonstantin Belousov 			*res = 0;
1324cff8c6f2SKonstantin Belousov 			return (0);
1325cff8c6f2SKonstantin Belousov 		}
1326cff8c6f2SKonstantin Belousov 		return (EINVAL);
1327e9eabf59SPeter Wemm 
1328e9eabf59SPeter Wemm 	default:
1329cff8c6f2SKonstantin Belousov 		return (EINVAL);
1330d297ad16SMarcel Moolenaar 	}
1331e9eabf59SPeter Wemm }
1332e9eabf59SPeter Wemm 
13337251b4bfSJake Burkholder static void
13347226306eSKonstantin Belousov link_elf_fix_link_set(elf_file_t ef)
13357226306eSKonstantin Belousov {
13367226306eSKonstantin Belousov 	static const char startn[] = "__start_";
13377226306eSKonstantin Belousov 	static const char stopn[] = "__stop_";
13387226306eSKonstantin Belousov 	Elf_Sym *sym;
13397226306eSKonstantin Belousov 	const char *sym_name, *linkset_name;
13407226306eSKonstantin Belousov 	Elf_Addr startp, stopp;
13417226306eSKonstantin Belousov 	Elf_Size symidx;
13427226306eSKonstantin Belousov 	int start, i;
13437226306eSKonstantin Belousov 
13447226306eSKonstantin Belousov 	startp = stopp = 0;
13457226306eSKonstantin Belousov 	for (symidx = 1 /* zero entry is special */;
13467226306eSKonstantin Belousov 		symidx < ef->ddbsymcnt; symidx++) {
13477226306eSKonstantin Belousov 		sym = ef->ddbsymtab + symidx;
13487226306eSKonstantin Belousov 		if (sym->st_shndx != SHN_UNDEF)
13497226306eSKonstantin Belousov 			continue;
13507226306eSKonstantin Belousov 
13517226306eSKonstantin Belousov 		sym_name = ef->ddbstrtab + sym->st_name;
13527226306eSKonstantin Belousov 		if (strncmp(sym_name, startn, sizeof(startn) - 1) == 0) {
13537226306eSKonstantin Belousov 			start = 1;
13547226306eSKonstantin Belousov 			linkset_name = sym_name + sizeof(startn) - 1;
13557226306eSKonstantin Belousov 		}
13567226306eSKonstantin Belousov 		else if (strncmp(sym_name, stopn, sizeof(stopn) - 1) == 0) {
13577226306eSKonstantin Belousov 			start = 0;
13587226306eSKonstantin Belousov 			linkset_name = sym_name + sizeof(stopn) - 1;
13597226306eSKonstantin Belousov 		}
13607226306eSKonstantin Belousov 		else
13617226306eSKonstantin Belousov 			continue;
13627226306eSKonstantin Belousov 
13637226306eSKonstantin Belousov 		for (i = 0; i < ef->nprogtab; i++) {
13647226306eSKonstantin Belousov 			if (strcmp(ef->progtab[i].name, linkset_name) == 0) {
13657226306eSKonstantin Belousov 				startp = (Elf_Addr)ef->progtab[i].addr;
13667226306eSKonstantin Belousov 				stopp = (Elf_Addr)(startp + ef->progtab[i].size);
13677226306eSKonstantin Belousov 				break;
13687226306eSKonstantin Belousov 			}
13697226306eSKonstantin Belousov 		}
13707226306eSKonstantin Belousov 		if (i == ef->nprogtab)
13717226306eSKonstantin Belousov 			continue;
13727226306eSKonstantin Belousov 
13737226306eSKonstantin Belousov 		sym->st_value = start ? startp : stopp;
13747226306eSKonstantin Belousov 		sym->st_shndx = i;
13757226306eSKonstantin Belousov 	}
13767226306eSKonstantin Belousov }
13777226306eSKonstantin Belousov 
13787226306eSKonstantin Belousov static void
13797251b4bfSJake Burkholder link_elf_reloc_local(linker_file_t lf)
13807251b4bfSJake Burkholder {
1381e9eabf59SPeter Wemm 	elf_file_t ef = (elf_file_t)lf;
13827251b4bfSJake Burkholder 	const Elf_Rel *rellim;
13837251b4bfSJake Burkholder 	const Elf_Rel *rel;
13847251b4bfSJake Burkholder 	const Elf_Rela *relalim;
13857251b4bfSJake Burkholder 	const Elf_Rela *rela;
1386e9eabf59SPeter Wemm 	const Elf_Sym *sym;
1387e9eabf59SPeter Wemm 	Elf_Addr base;
1388e9eabf59SPeter Wemm 	int i;
1389757686b1SMarcel Moolenaar 	Elf_Size symidx;
1390e9eabf59SPeter Wemm 
13917226306eSKonstantin Belousov 	link_elf_fix_link_set(ef);
13927226306eSKonstantin Belousov 
13937251b4bfSJake Burkholder 	/* Perform relocations without addend if there are any: */
1394b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nreltab; i++) {
1395e9eabf59SPeter Wemm 		rel = ef->reltab[i].rel;
139620947801SPeter Wemm 		if (rel == NULL)
139782d0d1a0SPeter Wemm 			panic("lost a reltab!");
139820947801SPeter Wemm 		rellim = rel + ef->reltab[i].nrel;
1399e9eabf59SPeter Wemm 		base = findbase(ef, ef->reltab[i].sec);
1400a8774e39SPeter Wemm 		if (base == 0)
1401a8774e39SPeter Wemm 			panic("lost base for reltab");
140220947801SPeter Wemm 		for ( ; rel < rellim; rel++) {
1403e9eabf59SPeter Wemm 			symidx = ELF_R_SYM(rel->r_info);
140420947801SPeter Wemm 			if (symidx >= ef->ddbsymcnt)
140520947801SPeter Wemm 				continue;
1406e9eabf59SPeter Wemm 			sym = ef->ddbsymtab + symidx;
140720947801SPeter Wemm 			/* Only do local relocs */
140820947801SPeter Wemm 			if (ELF_ST_BIND(sym->st_info) != STB_LOCAL)
140920947801SPeter Wemm 				continue;
141020947801SPeter Wemm 			elf_reloc_local(lf, base, rel, ELF_RELOC_REL,
141120947801SPeter Wemm 			    elf_obj_lookup);
14127251b4bfSJake Burkholder 		}
1413e9eabf59SPeter Wemm 	}
14147251b4bfSJake Burkholder 
14157251b4bfSJake Burkholder 	/* Perform relocations with addend if there are any: */
1416b474c780SDavid E. O'Brien 	for (i = 0; i < ef->nrelatab; i++) {
1417e9eabf59SPeter Wemm 		rela = ef->relatab[i].rela;
141820947801SPeter Wemm 		if (rela == NULL)
141982d0d1a0SPeter Wemm 			panic("lost a relatab!");
142020947801SPeter Wemm 		relalim = rela + ef->relatab[i].nrela;
1421e9eabf59SPeter Wemm 		base = findbase(ef, ef->relatab[i].sec);
1422a8774e39SPeter Wemm 		if (base == 0)
1423a8774e39SPeter Wemm 			panic("lost base for relatab");
142420947801SPeter Wemm 		for ( ; rela < relalim; rela++) {
1425e9eabf59SPeter Wemm 			symidx = ELF_R_SYM(rela->r_info);
142620947801SPeter Wemm 			if (symidx >= ef->ddbsymcnt)
142720947801SPeter Wemm 				continue;
1428e9eabf59SPeter Wemm 			sym = ef->ddbsymtab + symidx;
142920947801SPeter Wemm 			/* Only do local relocs */
143020947801SPeter Wemm 			if (ELF_ST_BIND(sym->st_info) != STB_LOCAL)
143120947801SPeter Wemm 				continue;
143220947801SPeter Wemm 			elf_reloc_local(lf, base, rela, ELF_RELOC_RELA,
143320947801SPeter Wemm 			    elf_obj_lookup);
14347251b4bfSJake Burkholder 		}
14357251b4bfSJake Burkholder 	}
1436e9eabf59SPeter Wemm }
143700a5db46SStacey Son 
143800a5db46SStacey Son static long
1439e76f11f4SAndriy Gapon link_elf_symtab_get(linker_file_t lf, const Elf_Sym **symtab)
144000a5db46SStacey Son {
144100a5db46SStacey Son     elf_file_t ef = (elf_file_t)lf;
144200a5db46SStacey Son 
144300a5db46SStacey Son     *symtab = ef->ddbsymtab;
144400a5db46SStacey Son 
144500a5db46SStacey Son     if (*symtab == NULL)
144600a5db46SStacey Son         return (0);
144700a5db46SStacey Son 
144800a5db46SStacey Son     return (ef->ddbsymcnt);
144900a5db46SStacey Son }
145000a5db46SStacey Son 
145100a5db46SStacey Son static long
145200a5db46SStacey Son link_elf_strtab_get(linker_file_t lf, caddr_t *strtab)
145300a5db46SStacey Son {
145400a5db46SStacey Son     elf_file_t ef = (elf_file_t)lf;
145500a5db46SStacey Son 
145600a5db46SStacey Son     *strtab = ef->ddbstrtab;
145700a5db46SStacey Son 
145800a5db46SStacey Son     if (*strtab == NULL)
145900a5db46SStacey Son         return (0);
146000a5db46SStacey Son 
146100a5db46SStacey Son     return (ef->ddbstrcnt);
146200a5db46SStacey Son }
1463