xref: /freebsd-src/stand/common/load_elf.c (revision 86077f4fd11070518a6d04eee7fdb93cbbfb1b52)
1ca987d46SWarner Losh /*-
2ca987d46SWarner Losh  * Copyright (c) 1998 Michael Smith <msmith@freebsd.org>
3ca987d46SWarner Losh  * Copyright (c) 1998 Peter Wemm <peter@freebsd.org>
4ca987d46SWarner Losh  * All rights reserved.
5ca987d46SWarner Losh  *
6ca987d46SWarner Losh  * Redistribution and use in source and binary forms, with or without
7ca987d46SWarner Losh  * modification, are permitted provided that the following conditions
8ca987d46SWarner Losh  * are met:
9ca987d46SWarner Losh  * 1. Redistributions of source code must retain the above copyright
10ca987d46SWarner Losh  *    notice, this list of conditions and the following disclaimer.
11ca987d46SWarner Losh  * 2. Redistributions in binary form must reproduce the above copyright
12ca987d46SWarner Losh  *    notice, this list of conditions and the following disclaimer in the
13ca987d46SWarner Losh  *    documentation and/or other materials provided with the distribution.
14ca987d46SWarner Losh  *
15ca987d46SWarner Losh  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16ca987d46SWarner Losh  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17ca987d46SWarner Losh  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18ca987d46SWarner Losh  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19ca987d46SWarner Losh  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20ca987d46SWarner Losh  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21ca987d46SWarner Losh  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22ca987d46SWarner Losh  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23ca987d46SWarner Losh  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24ca987d46SWarner Losh  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25ca987d46SWarner Losh  * SUCH DAMAGE.
26ca987d46SWarner Losh  */
27ca987d46SWarner Losh 
28ca987d46SWarner Losh #include <sys/param.h>
298de1ad0bSWojciech Macek #include <sys/endian.h>
30ca987d46SWarner Losh #include <sys/exec.h>
31ca987d46SWarner Losh #include <sys/linker.h>
32ca987d46SWarner Losh #include <sys/module.h>
33ca987d46SWarner Losh #include <sys/stdint.h>
34ca987d46SWarner Losh #include <string.h>
35ca987d46SWarner Losh #include <machine/elf.h>
36ca987d46SWarner Losh #include <stand.h>
37f38658e1SWarner Losh #include <sys/link_elf.h>
38ca987d46SWarner Losh 
39ca987d46SWarner Losh #include "bootstrap.h"
40*86077f4fSAhmad Khalifa #include "modinfo.h"
41ca987d46SWarner Losh 
42ca987d46SWarner Losh #define COPYOUT(s,d,l)	archsw.arch_copyout((vm_offset_t)(s), d, l)
43ca987d46SWarner Losh 
44ca987d46SWarner Losh #if defined(__i386__) && __ELF_WORD_SIZE == 64
45ca987d46SWarner Losh #undef ELF_TARG_CLASS
46ca987d46SWarner Losh #undef ELF_TARG_MACH
47ca987d46SWarner Losh #define ELF_TARG_CLASS  ELFCLASS64
48ca987d46SWarner Losh #define ELF_TARG_MACH   EM_X86_64
49ca987d46SWarner Losh #endif
50ca987d46SWarner Losh 
51ca987d46SWarner Losh typedef struct elf_file {
52ca987d46SWarner Losh 	Elf_Phdr	*ph;
53ca987d46SWarner Losh 	Elf_Ehdr	*ehdr;
54ca987d46SWarner Losh 	Elf_Sym		*symtab;
55ca987d46SWarner Losh 	Elf_Hashelt	*hashtab;
56ca987d46SWarner Losh 	Elf_Hashelt	nbuckets;
57ca987d46SWarner Losh 	Elf_Hashelt	nchains;
58ca987d46SWarner Losh 	Elf_Hashelt	*buckets;
59ca987d46SWarner Losh 	Elf_Hashelt	*chains;
60ca987d46SWarner Losh 	Elf_Rel	*rel;
61ca987d46SWarner Losh 	size_t	relsz;
62ca987d46SWarner Losh 	Elf_Rela	*rela;
63ca987d46SWarner Losh 	size_t	relasz;
64ca987d46SWarner Losh 	char	*strtab;
65ca987d46SWarner Losh 	size_t	strsz;
66ca987d46SWarner Losh 	int		fd;
67ca987d46SWarner Losh 	caddr_t	firstpage;
68ca987d46SWarner Losh 	size_t	firstlen;
69ca987d46SWarner Losh 	int		kernel;
7056e53cb8SWarner Losh 	uint64_t	off;
71afc571b1SSimon J. Gerraty #ifdef LOADER_VERIEXEC_VECTX
72afc571b1SSimon J. Gerraty 	struct vectx	*vctx;
73afc571b1SSimon J. Gerraty #endif
74ca987d46SWarner Losh } *elf_file_t;
75ca987d46SWarner Losh 
76afc571b1SSimon J. Gerraty #ifdef LOADER_VERIEXEC_VECTX
77afc571b1SSimon J. Gerraty #define VECTX_HANDLE(ef) (ef)->vctx
78afc571b1SSimon J. Gerraty #else
79afc571b1SSimon J. Gerraty #define VECTX_HANDLE(ef) (ef)->fd
80afc571b1SSimon J. Gerraty #endif
81afc571b1SSimon J. Gerraty 
826b632498SEd Maste static int __elfN(loadimage)(struct preloaded_file *mp, elf_file_t ef,
8356e53cb8SWarner Losh     uint64_t loadaddr);
846c7a932dSToomas Soome static int __elfN(lookup_symbol)(elf_file_t ef, const char* name,
856c7a932dSToomas Soome     Elf_Sym *sym, unsigned char type);
86ca987d46SWarner Losh static int __elfN(reloc_ptr)(struct preloaded_file *mp, elf_file_t ef,
87ca987d46SWarner Losh     Elf_Addr p, void *val, size_t len);
88ca987d46SWarner Losh static int __elfN(parse_modmetadata)(struct preloaded_file *mp, elf_file_t ef,
89ca987d46SWarner Losh     Elf_Addr p_start, Elf_Addr p_end);
90ca987d46SWarner Losh static symaddr_fn __elfN(symaddr);
91ca987d46SWarner Losh static char	*fake_modname(const char *name);
92ca987d46SWarner Losh 
9356e53cb8SWarner Losh uint64_t	__elfN(relocation_offset) = 0;
94ca987d46SWarner Losh 
95c0ecae78SWarner Losh #ifdef __powerpc__
9612363d8eSRoger Pau Monné extern void elf_wrong_field_size(void);
9712363d8eSRoger Pau Monné #define CONVERT_FIELD(b, f, e)			\
9812363d8eSRoger Pau Monné 	switch (sizeof((b)->f)) {		\
9912363d8eSRoger Pau Monné 	case 2:					\
10012363d8eSRoger Pau Monné 		(b)->f = e ## 16toh((b)->f);	\
10112363d8eSRoger Pau Monné 		break;				\
10212363d8eSRoger Pau Monné 	case 4:					\
10312363d8eSRoger Pau Monné 		(b)->f = e ## 32toh((b)->f);	\
10412363d8eSRoger Pau Monné 		break;				\
10512363d8eSRoger Pau Monné 	case 8:					\
10612363d8eSRoger Pau Monné 		(b)->f = e ## 64toh((b)->f);	\
10712363d8eSRoger Pau Monné 		break;				\
10812363d8eSRoger Pau Monné 	default:				\
10912363d8eSRoger Pau Monné 		/* Force a link time error. */	\
11012363d8eSRoger Pau Monné 		elf_wrong_field_size();		\
11112363d8eSRoger Pau Monné 		break;				\
11212363d8eSRoger Pau Monné 	}
11312363d8eSRoger Pau Monné 
11412363d8eSRoger Pau Monné #define CONVERT_SWITCH(h, d, f)			\
11512363d8eSRoger Pau Monné 	switch ((h)->e_ident[EI_DATA]) {	\
11612363d8eSRoger Pau Monné 	case ELFDATA2MSB:			\
11712363d8eSRoger Pau Monné 		f(d, be);			\
11812363d8eSRoger Pau Monné 		break;				\
11912363d8eSRoger Pau Monné 	case ELFDATA2LSB:			\
12012363d8eSRoger Pau Monné 		f(d, le);			\
12112363d8eSRoger Pau Monné 		break;				\
12212363d8eSRoger Pau Monné 	default:				\
12312363d8eSRoger Pau Monné 		return (EINVAL);		\
12412363d8eSRoger Pau Monné 	}
12512363d8eSRoger Pau Monné 
12612363d8eSRoger Pau Monné 
12712363d8eSRoger Pau Monné static int elf_header_convert(Elf_Ehdr *ehdr)
12812363d8eSRoger Pau Monné {
12912363d8eSRoger Pau Monné 	/*
13012363d8eSRoger Pau Monné 	 * Fixup ELF header endianness.
13112363d8eSRoger Pau Monné 	 *
13212363d8eSRoger Pau Monné 	 * The Xhdr structure was loaded using block read call to optimize file
13312363d8eSRoger Pau Monné 	 * accesses. It might happen, that the endianness of the system memory
13412363d8eSRoger Pau Monné 	 * is different that endianness of the ELF header.  Swap fields here to
13512363d8eSRoger Pau Monné 	 * guarantee that Xhdr always contain valid data regardless of
13612363d8eSRoger Pau Monné 	 * architecture.
13712363d8eSRoger Pau Monné 	 */
13812363d8eSRoger Pau Monné #define HEADER_FIELDS(b, e)			\
13912363d8eSRoger Pau Monné 	CONVERT_FIELD(b, e_type, e);		\
14012363d8eSRoger Pau Monné 	CONVERT_FIELD(b, e_machine, e);		\
14112363d8eSRoger Pau Monné 	CONVERT_FIELD(b, e_version, e);		\
14212363d8eSRoger Pau Monné 	CONVERT_FIELD(b, e_entry, e);		\
14312363d8eSRoger Pau Monné 	CONVERT_FIELD(b, e_phoff, e);		\
14412363d8eSRoger Pau Monné 	CONVERT_FIELD(b, e_shoff, e);		\
14512363d8eSRoger Pau Monné 	CONVERT_FIELD(b, e_flags, e);		\
14612363d8eSRoger Pau Monné 	CONVERT_FIELD(b, e_ehsize, e);		\
14712363d8eSRoger Pau Monné 	CONVERT_FIELD(b, e_phentsize, e);	\
14812363d8eSRoger Pau Monné 	CONVERT_FIELD(b, e_phnum, e);		\
14912363d8eSRoger Pau Monné 	CONVERT_FIELD(b, e_shentsize, e);	\
15012363d8eSRoger Pau Monné 	CONVERT_FIELD(b, e_shnum, e);		\
15112363d8eSRoger Pau Monné 	CONVERT_FIELD(b, e_shstrndx, e)
15212363d8eSRoger Pau Monné 
15312363d8eSRoger Pau Monné 	CONVERT_SWITCH(ehdr, ehdr, HEADER_FIELDS);
15412363d8eSRoger Pau Monné 
15512363d8eSRoger Pau Monné #undef HEADER_FIELDS
15612363d8eSRoger Pau Monné 
15712363d8eSRoger Pau Monné 	return (0);
15812363d8eSRoger Pau Monné }
15912363d8eSRoger Pau Monné 
16012363d8eSRoger Pau Monné static int elf_program_header_convert(const Elf_Ehdr *ehdr, Elf_Phdr *phdr)
16112363d8eSRoger Pau Monné {
16212363d8eSRoger Pau Monné #define PROGRAM_HEADER_FIELDS(b, e)		\
16312363d8eSRoger Pau Monné 	CONVERT_FIELD(b, p_type, e);		\
16412363d8eSRoger Pau Monné 	CONVERT_FIELD(b, p_flags, e);		\
16512363d8eSRoger Pau Monné 	CONVERT_FIELD(b, p_offset, e);		\
16612363d8eSRoger Pau Monné 	CONVERT_FIELD(b, p_vaddr, e);		\
16712363d8eSRoger Pau Monné 	CONVERT_FIELD(b, p_paddr, e);		\
16812363d8eSRoger Pau Monné 	CONVERT_FIELD(b, p_filesz, e);		\
16912363d8eSRoger Pau Monné 	CONVERT_FIELD(b, p_memsz, e);		\
17012363d8eSRoger Pau Monné 	CONVERT_FIELD(b, p_align, e)
17112363d8eSRoger Pau Monné 
17212363d8eSRoger Pau Monné 	CONVERT_SWITCH(ehdr, phdr, PROGRAM_HEADER_FIELDS);
17312363d8eSRoger Pau Monné 
17412363d8eSRoger Pau Monné #undef PROGRAM_HEADER_FIELDS
17512363d8eSRoger Pau Monné 
17612363d8eSRoger Pau Monné 	return (0);
17712363d8eSRoger Pau Monné }
17812363d8eSRoger Pau Monné 
17912363d8eSRoger Pau Monné static int elf_section_header_convert(const Elf_Ehdr *ehdr, Elf_Shdr *shdr)
18012363d8eSRoger Pau Monné {
18112363d8eSRoger Pau Monné #define SECTION_HEADER_FIELDS(b, e)		\
18212363d8eSRoger Pau Monné 	CONVERT_FIELD(b, sh_name, e);		\
18312363d8eSRoger Pau Monné 	CONVERT_FIELD(b, sh_type, e);		\
18412363d8eSRoger Pau Monné 	CONVERT_FIELD(b, sh_link, e);		\
18512363d8eSRoger Pau Monné 	CONVERT_FIELD(b, sh_info, e);		\
18612363d8eSRoger Pau Monné 	CONVERT_FIELD(b, sh_flags, e);		\
18712363d8eSRoger Pau Monné 	CONVERT_FIELD(b, sh_addr, e);		\
18812363d8eSRoger Pau Monné 	CONVERT_FIELD(b, sh_offset, e);		\
18912363d8eSRoger Pau Monné 	CONVERT_FIELD(b, sh_size, e);		\
19012363d8eSRoger Pau Monné 	CONVERT_FIELD(b, sh_addralign, e);	\
19112363d8eSRoger Pau Monné 	CONVERT_FIELD(b, sh_entsize, e)
19212363d8eSRoger Pau Monné 
19312363d8eSRoger Pau Monné 	CONVERT_SWITCH(ehdr, shdr, SECTION_HEADER_FIELDS);
19412363d8eSRoger Pau Monné 
19512363d8eSRoger Pau Monné #undef SECTION_HEADER_FIELDS
19612363d8eSRoger Pau Monné 
19712363d8eSRoger Pau Monné 	return (0);
19812363d8eSRoger Pau Monné }
19912363d8eSRoger Pau Monné #undef CONVERT_SWITCH
20012363d8eSRoger Pau Monné #undef CONVERT_FIELD
201c0ecae78SWarner Losh #else
202c0ecae78SWarner Losh static int elf_header_convert(Elf_Ehdr *ehdr)
203c0ecae78SWarner Losh {
204c0ecae78SWarner Losh 	return (0);
205c0ecae78SWarner Losh }
20612363d8eSRoger Pau Monné 
207c0ecae78SWarner Losh static int elf_program_header_convert(const Elf_Ehdr *ehdr, Elf_Phdr *phdr)
208c0ecae78SWarner Losh {
209c0ecae78SWarner Losh 	return (0);
210c0ecae78SWarner Losh }
211c0ecae78SWarner Losh 
212c0ecae78SWarner Losh static int elf_section_header_convert(const Elf_Ehdr *ehdr, Elf_Shdr *shdr)
213c0ecae78SWarner Losh {
214c0ecae78SWarner Losh 	return (0);
215c0ecae78SWarner Losh }
216c0ecae78SWarner Losh #endif
217f75caed6SKonstantin Belousov 
2189e239e67SAhmad Khalifa #if defined(__amd64__) || (defined(__i386__) && defined(EFI))
219f75caed6SKonstantin Belousov static bool
220f75caed6SKonstantin Belousov is_kernphys_relocatable(elf_file_t ef)
221f75caed6SKonstantin Belousov {
222f75caed6SKonstantin Belousov 	Elf_Sym sym;
223f75caed6SKonstantin Belousov 
2249a964ad5SMark Johnston 	return (__elfN(lookup_symbol)(ef, "kernphys", &sym, STT_OBJECT) == 0);
225f75caed6SKonstantin Belousov }
226f75caed6SKonstantin Belousov #endif
227f75caed6SKonstantin Belousov 
228aaaa5a2eSWarner Losh #ifdef __i386__
229aaaa5a2eSWarner Losh static bool
230aaaa5a2eSWarner Losh is_tg_kernel_support(struct preloaded_file *fp, elf_file_t ef)
231aaaa5a2eSWarner Losh {
232aaaa5a2eSWarner Losh 	Elf_Sym		sym;
233aaaa5a2eSWarner Losh 	Elf_Addr	p_start, p_end, v, p;
234aaaa5a2eSWarner Losh 	char		vd_name[16];
235aaaa5a2eSWarner Losh 	int		error;
236aaaa5a2eSWarner Losh 
237aaaa5a2eSWarner Losh 	if (__elfN(lookup_symbol)(ef, "__start_set_vt_drv_set", &sym, STT_NOTYPE) != 0)
238aaaa5a2eSWarner Losh 		return (false);
239aaaa5a2eSWarner Losh 	p_start = sym.st_value + ef->off;
240aaaa5a2eSWarner Losh 	if (__elfN(lookup_symbol)(ef, "__stop_set_vt_drv_set", &sym, STT_NOTYPE) != 0)
241aaaa5a2eSWarner Losh 		return (false);
242aaaa5a2eSWarner Losh 	p_end = sym.st_value + ef->off;
243aaaa5a2eSWarner Losh 
244aaaa5a2eSWarner Losh 	/*
245aaaa5a2eSWarner Losh 	 * Walk through vt_drv_set, each vt driver structure starts with
246aaaa5a2eSWarner Losh 	 * static 16 chars for driver name. If we have "vbefb", return true.
247aaaa5a2eSWarner Losh 	 */
248aaaa5a2eSWarner Losh 	for (p = p_start; p < p_end; p += sizeof(Elf_Addr)) {
249aaaa5a2eSWarner Losh 		COPYOUT(p, &v, sizeof(v));
250aaaa5a2eSWarner Losh 
251aaaa5a2eSWarner Losh 		error = __elfN(reloc_ptr)(fp, ef, p, &v, sizeof(v));
252aaaa5a2eSWarner Losh 		if (error == EOPNOTSUPP)
253aaaa5a2eSWarner Losh 			v += ef->off;
254aaaa5a2eSWarner Losh 		else if (error != 0)
255aaaa5a2eSWarner Losh 			return (false);
256aaaa5a2eSWarner Losh 		COPYOUT(v, &vd_name, sizeof(vd_name));
257aaaa5a2eSWarner Losh 		if (strncmp(vd_name, "vbefb", sizeof(vd_name)) == 0)
258aaaa5a2eSWarner Losh 			return (true);
259aaaa5a2eSWarner Losh 	}
260aaaa5a2eSWarner Losh 
261aaaa5a2eSWarner Losh 	return (false);
262aaaa5a2eSWarner Losh }
263aaaa5a2eSWarner Losh #endif
264aaaa5a2eSWarner Losh 
265ca987d46SWarner Losh static int
266ca987d46SWarner Losh __elfN(load_elf_header)(char *filename, elf_file_t ef)
267ca987d46SWarner Losh {
268ca987d46SWarner Losh 	ssize_t			 bytes_read;
269ca987d46SWarner Losh 	Elf_Ehdr		*ehdr;
270ca987d46SWarner Losh 	int			 err;
271ca987d46SWarner Losh 
272ca987d46SWarner Losh 	/*
273ca987d46SWarner Losh 	 * Open the image, read and validate the ELF header
274ca987d46SWarner Losh 	 */
275ca987d46SWarner Losh 	if (filename == NULL)	/* can't handle nameless */
276ca987d46SWarner Losh 		return (EFTYPE);
277ca987d46SWarner Losh 	if ((ef->fd = open(filename, O_RDONLY)) == -1)
278ca987d46SWarner Losh 		return (errno);
279ca987d46SWarner Losh 	ef->firstpage = malloc(PAGE_SIZE);
280ca987d46SWarner Losh 	if (ef->firstpage == NULL) {
281ca987d46SWarner Losh 		close(ef->fd);
282ca987d46SWarner Losh 		return (ENOMEM);
283ca987d46SWarner Losh 	}
284c25d9affSEmmanuel Vadot 	preload(ef->fd);
285afc571b1SSimon J. Gerraty #ifdef LOADER_VERIEXEC_VECTX
286afc571b1SSimon J. Gerraty 	{
287afc571b1SSimon J. Gerraty 		int verror;
288afc571b1SSimon J. Gerraty 
289afc571b1SSimon J. Gerraty 		ef->vctx = vectx_open(ef->fd, filename, 0L, NULL, &verror, __func__);
290afc571b1SSimon J. Gerraty 		if (verror) {
291afc571b1SSimon J. Gerraty 			printf("Unverified %s: %s\n", filename, ve_error_get());
292afc571b1SSimon J. Gerraty 			close(ef->fd);
293afc571b1SSimon J. Gerraty 			free(ef->vctx);
294afc571b1SSimon J. Gerraty 			return (EAUTH);
295afc571b1SSimon J. Gerraty 		}
296afc571b1SSimon J. Gerraty 	}
297afc571b1SSimon J. Gerraty #endif
298afc571b1SSimon J. Gerraty 	bytes_read = VECTX_READ(VECTX_HANDLE(ef), ef->firstpage, PAGE_SIZE);
299ca987d46SWarner Losh 	ef->firstlen = (size_t)bytes_read;
300ca987d46SWarner Losh 	if (bytes_read < 0 || ef->firstlen <= sizeof(Elf_Ehdr)) {
301ca987d46SWarner Losh 		err = EFTYPE; /* could be EIO, but may be small file */
302ca987d46SWarner Losh 		goto error;
303ca987d46SWarner Losh 	}
304ca987d46SWarner Losh 	ehdr = ef->ehdr = (Elf_Ehdr *)ef->firstpage;
305ca987d46SWarner Losh 
306ca987d46SWarner Losh 	/* Is it ELF? */
307ca987d46SWarner Losh 	if (!IS_ELF(*ehdr)) {
308ca987d46SWarner Losh 		err = EFTYPE;
309ca987d46SWarner Losh 		goto error;
310ca987d46SWarner Losh 	}
3118de1ad0bSWojciech Macek 
312ca987d46SWarner Losh 	if (ehdr->e_ident[EI_CLASS] != ELF_TARG_CLASS || /* Layout ? */
313ca987d46SWarner Losh 	    ehdr->e_ident[EI_DATA] != ELF_TARG_DATA ||
3148de1ad0bSWojciech Macek 	    ehdr->e_ident[EI_VERSION] != EV_CURRENT) /* Version ? */ {
3158de1ad0bSWojciech Macek 		err = EFTYPE;
3168de1ad0bSWojciech Macek 		goto error;
3178de1ad0bSWojciech Macek 	}
3188de1ad0bSWojciech Macek 
31912363d8eSRoger Pau Monné 	err = elf_header_convert(ehdr);
32012363d8eSRoger Pau Monné 	if (err)
32112363d8eSRoger Pau Monné 		goto error;
3228de1ad0bSWojciech Macek 
3236b632498SEd Maste 	if (ehdr->e_version != EV_CURRENT || ehdr->e_machine != ELF_TARG_MACH) {
3246b632498SEd Maste 		/* Machine ? */
325ca987d46SWarner Losh 		err = EFTYPE;
326ca987d46SWarner Losh 		goto error;
327ca987d46SWarner Losh 	}
328ca987d46SWarner Losh 
329afc571b1SSimon J. Gerraty #if defined(LOADER_VERIEXEC) && !defined(LOADER_VERIEXEC_VECTX)
330afc571b1SSimon J. Gerraty 	if (verify_file(ef->fd, filename, bytes_read, VE_MUST, __func__) < 0) {
3318df8b2d3SSimon J. Gerraty 		err = EAUTH;
3328df8b2d3SSimon J. Gerraty 		goto error;
3338df8b2d3SSimon J. Gerraty 	}
3348df8b2d3SSimon J. Gerraty #endif
335ca987d46SWarner Losh 	return (0);
336ca987d46SWarner Losh 
337ca987d46SWarner Losh error:
338ca987d46SWarner Losh 	if (ef->firstpage != NULL) {
339ca987d46SWarner Losh 		free(ef->firstpage);
340ca987d46SWarner Losh 		ef->firstpage = NULL;
341ca987d46SWarner Losh 	}
342ca987d46SWarner Losh 	if (ef->fd != -1) {
343afc571b1SSimon J. Gerraty #ifdef LOADER_VERIEXEC_VECTX
344afc571b1SSimon J. Gerraty 		free(ef->vctx);
345afc571b1SSimon J. Gerraty #endif
346ca987d46SWarner Losh 		close(ef->fd);
347ca987d46SWarner Losh 		ef->fd = -1;
348ca987d46SWarner Losh 	}
349ca987d46SWarner Losh 	return (err);
350ca987d46SWarner Losh }
351ca987d46SWarner Losh 
352ca987d46SWarner Losh /*
353ca987d46SWarner Losh  * Attempt to load the file (file) as an ELF module.  It will be stored at
354ca987d46SWarner Losh  * (dest), and a pointer to a module structure describing the loaded object
355ca987d46SWarner Losh  * will be saved in (result).
356ca987d46SWarner Losh  */
357ca987d46SWarner Losh int
35856e53cb8SWarner Losh __elfN(loadfile)(char *filename, uint64_t dest, struct preloaded_file **result)
359ca987d46SWarner Losh {
360ca987d46SWarner Losh 	return (__elfN(loadfile_raw)(filename, dest, result, 0));
361ca987d46SWarner Losh }
362ca987d46SWarner Losh 
363ca987d46SWarner Losh int
36456e53cb8SWarner Losh __elfN(loadfile_raw)(char *filename, uint64_t dest,
365ca987d46SWarner Losh     struct preloaded_file **result, int multiboot)
366ca987d46SWarner Losh {
367ca987d46SWarner Losh 	struct preloaded_file	*fp, *kfp;
368ca987d46SWarner Losh 	struct elf_file		ef;
369ca987d46SWarner Losh 	Elf_Ehdr		*ehdr;
370ca987d46SWarner Losh 	int			err;
371ca987d46SWarner Losh 
372ca987d46SWarner Losh 	fp = NULL;
373ca987d46SWarner Losh 	bzero(&ef, sizeof(struct elf_file));
374ca987d46SWarner Losh 	ef.fd = -1;
375ca987d46SWarner Losh 
376ca987d46SWarner Losh 	err = __elfN(load_elf_header)(filename, &ef);
377ca987d46SWarner Losh 	if (err != 0)
378ca987d46SWarner Losh 		return (err);
379ca987d46SWarner Losh 
380ca987d46SWarner Losh 	ehdr = ef.ehdr;
381ca987d46SWarner Losh 
382ca987d46SWarner Losh 	/*
383ca987d46SWarner Losh 	 * Check to see what sort of module we are.
384ca987d46SWarner Losh 	 */
385*86077f4fSAhmad Khalifa 	kfp = file_findfile(NULL, md_kerntype);
386ca987d46SWarner Losh #ifdef __powerpc__
387ca987d46SWarner Losh 	/*
388ca987d46SWarner Losh 	 * Kernels can be ET_DYN, so just assume the first loaded object is the
389ca987d46SWarner Losh 	 * kernel. This assumption will be checked later.
390ca987d46SWarner Losh 	 */
391ca987d46SWarner Losh 	if (kfp == NULL)
392ca987d46SWarner Losh 		ef.kernel = 1;
393ca987d46SWarner Losh #endif
394ca987d46SWarner Losh 	if (ef.kernel || ehdr->e_type == ET_EXEC) {
395ca987d46SWarner Losh 		/* Looks like a kernel */
396ca987d46SWarner Losh 		if (kfp != NULL) {
3976b632498SEd Maste 			printf("elf" __XSTRING(__ELF_WORD_SIZE)
3986b632498SEd Maste 			    "_loadfile: kernel already loaded\n");
399ca987d46SWarner Losh 			err = EPERM;
400ca987d46SWarner Losh 			goto oerr;
401ca987d46SWarner Losh 		}
402ca987d46SWarner Losh 		/*
403ca987d46SWarner Losh 		 * Calculate destination address based on kernel entrypoint.
404ca987d46SWarner Losh 		 *
4056b632498SEd Maste 		 * For ARM, the destination address is independent of any values
4066b632498SEd Maste 		 * in the elf header (an ARM kernel can be loaded at any 2MB
4076b632498SEd Maste 		 * boundary), so we leave dest set to the value calculated by
4086b632498SEd Maste 		 * archsw.arch_loadaddr() and passed in to this function.
409ca987d46SWarner Losh 		 */
410ca987d46SWarner Losh #ifndef __arm__
411ca987d46SWarner Losh 		if (ehdr->e_type == ET_EXEC)
412ca987d46SWarner Losh 			dest = (ehdr->e_entry & ~PAGE_MASK);
413ca987d46SWarner Losh #endif
414ca987d46SWarner Losh 		if ((ehdr->e_entry & ~PAGE_MASK) == 0) {
4156b632498SEd Maste 			printf("elf" __XSTRING(__ELF_WORD_SIZE)
4166b632498SEd Maste 			    "_loadfile: not a kernel (maybe static binary?)\n");
417ca987d46SWarner Losh 			err = EPERM;
418ca987d46SWarner Losh 			goto oerr;
419ca987d46SWarner Losh 		}
420ca987d46SWarner Losh 		ef.kernel = 1;
421ca987d46SWarner Losh 
422ca987d46SWarner Losh 	} else if (ehdr->e_type == ET_DYN) {
423ca987d46SWarner Losh 		/* Looks like a kld module */
424ca987d46SWarner Losh 		if (multiboot != 0) {
4256b632498SEd Maste 			printf("elf" __XSTRING(__ELF_WORD_SIZE)
4266b632498SEd Maste 			    "_loadfile: can't load module as multiboot\n");
427ca987d46SWarner Losh 			err = EPERM;
428ca987d46SWarner Losh 			goto oerr;
429ca987d46SWarner Losh 		}
430ca987d46SWarner Losh 		if (kfp == NULL) {
4316b632498SEd Maste 			printf("elf" __XSTRING(__ELF_WORD_SIZE)
4326b632498SEd Maste 			    "_loadfile: can't load module before kernel\n");
433ca987d46SWarner Losh 			err = EPERM;
434ca987d46SWarner Losh 			goto oerr;
435ca987d46SWarner Losh 		}
436*86077f4fSAhmad Khalifa 		if (strcmp(md_kerntype, kfp->f_type)) {
4376b632498SEd Maste 			printf("elf" __XSTRING(__ELF_WORD_SIZE)
4386b632498SEd Maste 			 "_loadfile: can't load module with kernel type '%s'\n",
4396b632498SEd Maste 			    kfp->f_type);
440ca987d46SWarner Losh 			err = EPERM;
441ca987d46SWarner Losh 			goto oerr;
442ca987d46SWarner Losh 		}
443ca987d46SWarner Losh 		/* Looks OK, got ahead */
444ca987d46SWarner Losh 		ef.kernel = 0;
445ca987d46SWarner Losh 
446ca987d46SWarner Losh 	} else {
447ca987d46SWarner Losh 		err = EFTYPE;
448ca987d46SWarner Losh 		goto oerr;
449ca987d46SWarner Losh 	}
450ca987d46SWarner Losh 
451ca987d46SWarner Losh 	if (archsw.arch_loadaddr != NULL)
452ca987d46SWarner Losh 		dest = archsw.arch_loadaddr(LOAD_ELF, ehdr, dest);
453ca987d46SWarner Losh 	else
454ca987d46SWarner Losh 		dest = roundup(dest, PAGE_SIZE);
455ca987d46SWarner Losh 
456ca987d46SWarner Losh 	/*
457ca987d46SWarner Losh 	 * Ok, we think we should handle this.
458ca987d46SWarner Losh 	 */
459ca987d46SWarner Losh 	fp = file_alloc();
460ca987d46SWarner Losh 	if (fp == NULL) {
4616b632498SEd Maste 		printf("elf" __XSTRING(__ELF_WORD_SIZE)
4626b632498SEd Maste 		    "_loadfile: cannot allocate module info\n");
463ca987d46SWarner Losh 		err = EPERM;
464ca987d46SWarner Losh 		goto out;
465ca987d46SWarner Losh 	}
466ca987d46SWarner Losh 	if (ef.kernel == 1 && multiboot == 0)
467ca987d46SWarner Losh 		setenv("kernelname", filename, 1);
468ca987d46SWarner Losh 	fp->f_name = strdup(filename);
469ca987d46SWarner Losh 	if (multiboot == 0)
470ca987d46SWarner Losh 		fp->f_type = strdup(ef.kernel ?
471*86077f4fSAhmad Khalifa 		    md_kerntype : md_modtype);
472ca987d46SWarner Losh 	else
473*86077f4fSAhmad Khalifa 		fp->f_type = strdup(md_kerntype_mb);
474ca987d46SWarner Losh 
475ec042f46SSimon J. Gerraty 	if (module_verbose >= MODULE_VERBOSE_FULL) {
476ca987d46SWarner Losh 		if (ef.kernel)
4776b632498SEd Maste 			printf("%s entry at 0x%jx\n", filename,
4786b632498SEd Maste 			    (uintmax_t)ehdr->e_entry);
479ec042f46SSimon J. Gerraty 	} else if (module_verbose > MODULE_VERBOSE_SILENT)
480ca987d46SWarner Losh 		printf("%s ", filename);
481ec042f46SSimon J. Gerraty 
482ca987d46SWarner Losh 	fp->f_size = __elfN(loadimage)(fp, &ef, dest);
483ca987d46SWarner Losh 	if (fp->f_size == 0 || fp->f_addr == 0)
484ca987d46SWarner Losh 		goto ioerr;
485ca987d46SWarner Losh 
486ca987d46SWarner Losh 	/* save exec header as metadata */
487ca987d46SWarner Losh 	file_addmetadata(fp, MODINFOMD_ELFHDR, sizeof(*ehdr), ehdr);
488ca987d46SWarner Losh 
489ca987d46SWarner Losh 	/* Load OK, return module pointer */
490ca987d46SWarner Losh 	*result = (struct preloaded_file *)fp;
491ca987d46SWarner Losh 	err = 0;
4929e239e67SAhmad Khalifa #if defined(__amd64__) || (defined(__i386__) && defined(EFI))
4935e4279a8SRoger Pau Monné 	fp->f_kernphys_relocatable = multiboot || is_kernphys_relocatable(&ef);
494f75caed6SKonstantin Belousov #endif
4959e239e67SAhmad Khalifa #if defined(__i386__) && !defined(EFI)
496aaaa5a2eSWarner Losh 	fp->f_tg_kernel_support = is_tg_kernel_support(fp, &ef);
497aaaa5a2eSWarner Losh #endif
498ca987d46SWarner Losh 	goto out;
499ca987d46SWarner Losh 
500ca987d46SWarner Losh ioerr:
501ca987d46SWarner Losh 	err = EIO;
502ca987d46SWarner Losh oerr:
503ca987d46SWarner Losh 	file_discard(fp);
504ca987d46SWarner Losh out:
505ca987d46SWarner Losh 	if (ef.firstpage)
506ca987d46SWarner Losh 		free(ef.firstpage);
507afc571b1SSimon J. Gerraty 	if (ef.fd != -1) {
508afc571b1SSimon J. Gerraty #ifdef LOADER_VERIEXEC_VECTX
509afc571b1SSimon J. Gerraty 		if (!err && ef.vctx) {
510afc571b1SSimon J. Gerraty 			int verror;
511afc571b1SSimon J. Gerraty 
512afc571b1SSimon J. Gerraty 			verror = vectx_close(ef.vctx, VE_MUST, __func__);
513afc571b1SSimon J. Gerraty 			if (verror) {
514afc571b1SSimon J. Gerraty 				err = EAUTH;
515afc571b1SSimon J. Gerraty 				file_discard(fp);
516afc571b1SSimon J. Gerraty 			}
517afc571b1SSimon J. Gerraty 		}
518afc571b1SSimon J. Gerraty #endif
519ca987d46SWarner Losh 		close(ef.fd);
520afc571b1SSimon J. Gerraty 	}
521ca987d46SWarner Losh 	return (err);
522ca987d46SWarner Losh }
523ca987d46SWarner Losh 
524ca987d46SWarner Losh /*
525ca987d46SWarner Losh  * With the file (fd) open on the image, and (ehdr) containing
526ca987d46SWarner Losh  * the Elf header, load the image at (off)
527ca987d46SWarner Losh  */
528ca987d46SWarner Losh static int
52956e53cb8SWarner Losh __elfN(loadimage)(struct preloaded_file *fp, elf_file_t ef, uint64_t off)
530ca987d46SWarner Losh {
531ca987d46SWarner Losh 	int		i;
532ca987d46SWarner Losh 	u_int		j;
533ca987d46SWarner Losh 	Elf_Ehdr	*ehdr;
534ca987d46SWarner Losh 	Elf_Phdr	*phdr, *php;
535ca987d46SWarner Losh 	Elf_Shdr	*shdr;
536ca987d46SWarner Losh 	char		*shstr;
537ca987d46SWarner Losh 	int		ret;
538ca987d46SWarner Losh 	vm_offset_t	firstaddr;
539ca987d46SWarner Losh 	vm_offset_t	lastaddr;
540ca987d46SWarner Losh 	size_t		chunk;
541ca987d46SWarner Losh 	ssize_t		result;
542ca987d46SWarner Losh 	Elf_Addr	ssym, esym;
543ca987d46SWarner Losh 	Elf_Dyn		*dp;
544ca987d46SWarner Losh 	Elf_Addr	adp;
545ca987d46SWarner Losh 	Elf_Addr	ctors;
546ca987d46SWarner Losh 	int		ndp;
547ca987d46SWarner Losh 	int		symstrindex;
548ca987d46SWarner Losh 	int		symtabindex;
549ca987d46SWarner Losh 	Elf_Size	size;
550ca987d46SWarner Losh 	u_int		fpcopy;
551ca987d46SWarner Losh 	Elf_Sym		sym;
552ca987d46SWarner Losh 	Elf_Addr	p_start, p_end;
553ca987d46SWarner Losh 
554ca987d46SWarner Losh 	dp = NULL;
555ca987d46SWarner Losh 	shdr = NULL;
556ca987d46SWarner Losh 	ret = 0;
557ca987d46SWarner Losh 	firstaddr = lastaddr = 0;
558ca987d46SWarner Losh 	ehdr = ef->ehdr;
5592f6823d4SJustin Hibbits #ifdef __powerpc__
5602f6823d4SJustin Hibbits 	if (ef->kernel) {
5612f6823d4SJustin Hibbits #else
562ca987d46SWarner Losh 	if (ehdr->e_type == ET_EXEC) {
5632f6823d4SJustin Hibbits #endif
564ca987d46SWarner Losh #if defined(__i386__) || defined(__amd64__)
565ca987d46SWarner Losh #if __ELF_WORD_SIZE == 64
5666b632498SEd Maste 		/* x86_64 relocates after locore */
5676b632498SEd Maste 		off = - (off & 0xffffffffff000000ull);
568ca987d46SWarner Losh #else
5696b632498SEd Maste 		/* i386 relocates after locore */
5706b632498SEd Maste 		off = - (off & 0xff000000u);
571ca987d46SWarner Losh #endif
572ca987d46SWarner Losh #elif defined(__powerpc__)
573ca987d46SWarner Losh 		/*
5746b632498SEd Maste 		 * On the purely virtual memory machines like e500, the kernel
5756b632498SEd Maste 		 * is linked against its final VA range, which is most often
5766b632498SEd Maste 		 * not available at the loader stage, but only after kernel
5776b632498SEd Maste 		 * initializes and completes its VM settings. In such cases we
5786b632498SEd Maste 		 * cannot use p_vaddr field directly to load ELF segments, but
5796b632498SEd Maste 		 * put them at some 'load-time' locations.
580ca987d46SWarner Losh 		 */
581ca987d46SWarner Losh 		if (off & 0xf0000000u) {
582ca987d46SWarner Losh 			off = -(off & 0xf0000000u);
583ca987d46SWarner Losh 			/*
5846b632498SEd Maste 			 * XXX the physical load address should not be
5856b632498SEd Maste 			 * hardcoded. Note that the Book-E kernel assumes that
5866b632498SEd Maste 			 * it's loaded at a 16MB boundary for now...
587ca987d46SWarner Losh 			 */
588ca987d46SWarner Losh 			off += 0x01000000;
5892f6823d4SJustin Hibbits 		}
590ca987d46SWarner Losh 		ehdr->e_entry += off;
591ec042f46SSimon J. Gerraty 		if (module_verbose >= MODULE_VERBOSE_FULL)
592ec042f46SSimon J. Gerraty 			printf("Converted entry 0x%jx\n",
593ec042f46SSimon J. Gerraty 			    (uintmax_t)ehdr->e_entry);
594ec042f46SSimon J. Gerraty 
595ca987d46SWarner Losh #elif defined(__arm__) && !defined(EFI)
596ca987d46SWarner Losh 		/*
5976b632498SEd Maste 		 * The elf headers in arm kernels specify virtual addresses in
5986b632498SEd Maste 		 * all header fields, even the ones that should be physical
5996b632498SEd Maste 		 * addresses.  We assume the entry point is in the first page,
6006b632498SEd Maste 		 * and masking the page offset will leave us with the virtual
6016b632498SEd Maste 		 * address the kernel was linked at.  We subtract that from the
6026b632498SEd Maste 		 * load offset, making 'off' into the value which, when added
6036b632498SEd Maste 		 * to a virtual address in an elf header, translates it to a
6046b632498SEd Maste 		 * physical address.  We do the va->pa conversion on the entry
6056b632498SEd Maste 		 * point address in the header now, so that later we can launch
6066b632498SEd Maste 		 * the kernel by just jumping to that address.
607ca987d46SWarner Losh 		 *
6086b632498SEd Maste 		 * When booting from UEFI the copyin and copyout functions
6096b632498SEd Maste 		 * handle adjusting the location relative to the first virtual
6106b632498SEd Maste 		 * address.  Because of this there is no need to adjust the
6116b632498SEd Maste 		 * offset or entry point address as these will both be handled
6126b632498SEd Maste 		 * by the efi code.
613ca987d46SWarner Losh 		 */
614ca987d46SWarner Losh 		off -= ehdr->e_entry & ~PAGE_MASK;
615ca987d46SWarner Losh 		ehdr->e_entry += off;
616ec042f46SSimon J. Gerraty 		if (module_verbose >= MODULE_VERBOSE_FULL)
617637b6532SEnji Cooper 			printf("ehdr->e_entry 0x%jx, va<->pa off %llx\n",
618ff7449d6SLeandro Lupori 			    (uintmax_t)ehdr->e_entry, off);
619ca987d46SWarner Losh #else
620ca987d46SWarner Losh 		off = 0;	/* other archs use direct mapped kernels */
621ca987d46SWarner Losh #endif
622ca987d46SWarner Losh 	}
623ca987d46SWarner Losh 	ef->off = off;
624ca987d46SWarner Losh 
625ca987d46SWarner Losh 	if (ef->kernel)
626ca987d46SWarner Losh 		__elfN(relocation_offset) = off;
627ca987d46SWarner Losh 
628ca987d46SWarner Losh 	if ((ehdr->e_phoff + ehdr->e_phnum * sizeof(*phdr)) > ef->firstlen) {
6296b632498SEd Maste 		printf("elf" __XSTRING(__ELF_WORD_SIZE)
6306b632498SEd Maste 		    "_loadimage: program header not within first page\n");
631ca987d46SWarner Losh 		goto out;
632ca987d46SWarner Losh 	}
633ca987d46SWarner Losh 	phdr = (Elf_Phdr *)(ef->firstpage + ehdr->e_phoff);
634ca987d46SWarner Losh 
635ca987d46SWarner Losh 	for (i = 0; i < ehdr->e_phnum; i++) {
63612363d8eSRoger Pau Monné 		if (elf_program_header_convert(ehdr, phdr))
63712363d8eSRoger Pau Monné 			continue;
6388de1ad0bSWojciech Macek 
639ca987d46SWarner Losh 		/* We want to load PT_LOAD segments only.. */
640ca987d46SWarner Losh 		if (phdr[i].p_type != PT_LOAD)
641ca987d46SWarner Losh 			continue;
642ca987d46SWarner Losh 
643ec042f46SSimon J. Gerraty 		if (module_verbose >= MODULE_VERBOSE_FULL) {
644ca987d46SWarner Losh 			printf("Segment: 0x%lx@0x%lx -> 0x%lx-0x%lx",
645ca987d46SWarner Losh 			    (long)phdr[i].p_filesz, (long)phdr[i].p_offset,
646ca987d46SWarner Losh 			    (long)(phdr[i].p_vaddr + off),
647ca987d46SWarner Losh 			    (long)(phdr[i].p_vaddr + off + phdr[i].p_memsz - 1));
648ec042f46SSimon J. Gerraty 		} else if (module_verbose > MODULE_VERBOSE_SILENT) {
649ca987d46SWarner Losh 			if ((phdr[i].p_flags & PF_W) == 0) {
650ca987d46SWarner Losh 				printf("text=0x%lx ", (long)phdr[i].p_filesz);
651ca987d46SWarner Losh 			} else {
652ca987d46SWarner Losh 				printf("data=0x%lx", (long)phdr[i].p_filesz);
653ca987d46SWarner Losh 				if (phdr[i].p_filesz < phdr[i].p_memsz)
6546b632498SEd Maste 					printf("+0x%lx", (long)(phdr[i].p_memsz -
6556b632498SEd Maste 						phdr[i].p_filesz));
656ca987d46SWarner Losh 				printf(" ");
657ca987d46SWarner Losh 			}
658ec042f46SSimon J. Gerraty 		}
659ca987d46SWarner Losh 		fpcopy = 0;
660ca987d46SWarner Losh 		if (ef->firstlen > phdr[i].p_offset) {
661ca987d46SWarner Losh 			fpcopy = ef->firstlen - phdr[i].p_offset;
662ca987d46SWarner Losh 			archsw.arch_copyin(ef->firstpage + phdr[i].p_offset,
663ca987d46SWarner Losh 			    phdr[i].p_vaddr + off, fpcopy);
664ca987d46SWarner Losh 		}
665ca987d46SWarner Losh 		if (phdr[i].p_filesz > fpcopy) {
666afc571b1SSimon J. Gerraty 			if (kern_pread(VECTX_HANDLE(ef),
667afc571b1SSimon J. Gerraty 			    phdr[i].p_vaddr + off + fpcopy,
6686b632498SEd Maste 			    phdr[i].p_filesz - fpcopy,
6696b632498SEd Maste 			    phdr[i].p_offset + fpcopy) != 0) {
670ca987d46SWarner Losh 				printf("\nelf" __XSTRING(__ELF_WORD_SIZE)
671ca987d46SWarner Losh 				    "_loadimage: read failed\n");
672ca987d46SWarner Losh 				goto out;
673ca987d46SWarner Losh 			}
674ca987d46SWarner Losh 		}
675ca987d46SWarner Losh 		/* clear space from oversized segments; eg: bss */
676ca987d46SWarner Losh 		if (phdr[i].p_filesz < phdr[i].p_memsz) {
677ec042f46SSimon J. Gerraty 			if (module_verbose >= MODULE_VERBOSE_FULL) {
678ca987d46SWarner Losh 				printf(" (bss: 0x%lx-0x%lx)",
679ca987d46SWarner Losh 				    (long)(phdr[i].p_vaddr + off + phdr[i].p_filesz),
680ca987d46SWarner Losh 				    (long)(phdr[i].p_vaddr + off + phdr[i].p_memsz -1));
681ec042f46SSimon J. Gerraty 			}
682ca987d46SWarner Losh 			kern_bzero(phdr[i].p_vaddr + off + phdr[i].p_filesz,
683ca987d46SWarner Losh 			    phdr[i].p_memsz - phdr[i].p_filesz);
684ca987d46SWarner Losh 		}
685ec042f46SSimon J. Gerraty 		if (module_verbose >= MODULE_VERBOSE_FULL)
686ca987d46SWarner Losh 			printf("\n");
687ca987d46SWarner Losh 
688ca987d46SWarner Losh 		if (archsw.arch_loadseg != NULL)
689ca987d46SWarner Losh 			archsw.arch_loadseg(ehdr, phdr + i, off);
690ca987d46SWarner Losh 
691ca987d46SWarner Losh 		if (firstaddr == 0 || firstaddr > (phdr[i].p_vaddr + off))
692ca987d46SWarner Losh 			firstaddr = phdr[i].p_vaddr + off;
6936b632498SEd Maste 		if (lastaddr == 0 || lastaddr <
6946b632498SEd Maste 		    (phdr[i].p_vaddr + off + phdr[i].p_memsz))
695ca987d46SWarner Losh 			lastaddr = phdr[i].p_vaddr + off + phdr[i].p_memsz;
696ca987d46SWarner Losh 	}
697ca987d46SWarner Losh 	lastaddr = roundup(lastaddr, sizeof(long));
698ca987d46SWarner Losh 
699ca987d46SWarner Losh 	/*
700ca987d46SWarner Losh 	 * Get the section headers.  We need this for finding the .ctors
701ca987d46SWarner Losh 	 * section as well as for loading any symbols.  Both may be hard
702ca987d46SWarner Losh 	 * to do if reading from a .gz file as it involves seeking.  I
703ca987d46SWarner Losh 	 * think the rule is going to have to be that you must strip a
704ca987d46SWarner Losh 	 * file to remove symbols before gzipping it.
705ca987d46SWarner Losh 	 */
706db711744SWarner Losh 	chunk = (size_t)ehdr->e_shnum * (size_t)ehdr->e_shentsize;
707ca987d46SWarner Losh 	if (chunk == 0 || ehdr->e_shoff == 0)
708ca987d46SWarner Losh 		goto nosyms;
709afc571b1SSimon J. Gerraty 	shdr = alloc_pread(VECTX_HANDLE(ef), ehdr->e_shoff, chunk);
710ca987d46SWarner Losh 	if (shdr == NULL) {
711ca987d46SWarner Losh 		printf("\nelf" __XSTRING(__ELF_WORD_SIZE)
712ca987d46SWarner Losh 		    "_loadimage: failed to read section headers");
713ca987d46SWarner Losh 		goto nosyms;
714ca987d46SWarner Losh 	}
7158de1ad0bSWojciech Macek 
71612363d8eSRoger Pau Monné 	for (i = 0; i < ehdr->e_shnum; i++)
71712363d8eSRoger Pau Monné 		elf_section_header_convert(ehdr, &shdr[i]);
7185cca46a9SEd Maste 
719ca987d46SWarner Losh 	file_addmetadata(fp, MODINFOMD_SHDR, chunk, shdr);
720ca987d46SWarner Losh 
721ca987d46SWarner Losh 	/*
722ca987d46SWarner Losh 	 * Read the section string table and look for the .ctors section.
723ca987d46SWarner Losh 	 * We need to tell the kernel where it is so that it can call the
724ca987d46SWarner Losh 	 * ctors.
725ca987d46SWarner Losh 	 */
726ca987d46SWarner Losh 	chunk = shdr[ehdr->e_shstrndx].sh_size;
727ca987d46SWarner Losh 	if (chunk) {
728afc571b1SSimon J. Gerraty 		shstr = alloc_pread(VECTX_HANDLE(ef),
729afc571b1SSimon J. Gerraty 		    shdr[ehdr->e_shstrndx].sh_offset, chunk);
730ca987d46SWarner Losh 		if (shstr) {
731ca987d46SWarner Losh 			for (i = 0; i < ehdr->e_shnum; i++) {
7326b632498SEd Maste 				if (strcmp(shstr + shdr[i].sh_name,
7336b632498SEd Maste 				    ".ctors") != 0)
734ca987d46SWarner Losh 					continue;
735ca987d46SWarner Losh 				ctors = shdr[i].sh_addr;
7366b632498SEd Maste 				file_addmetadata(fp, MODINFOMD_CTORS_ADDR,
7376b632498SEd Maste 				    sizeof(ctors), &ctors);
738ca987d46SWarner Losh 				size = shdr[i].sh_size;
7396b632498SEd Maste 				file_addmetadata(fp, MODINFOMD_CTORS_SIZE,
7406b632498SEd Maste 				    sizeof(size), &size);
741ca987d46SWarner Losh 				break;
742ca987d46SWarner Losh 			}
743ca987d46SWarner Losh 			free(shstr);
744ca987d46SWarner Losh 		}
745ca987d46SWarner Losh 	}
746ca987d46SWarner Losh 
747ca987d46SWarner Losh 	/*
748ca987d46SWarner Losh 	 * Now load any symbols.
749ca987d46SWarner Losh 	 */
750ca987d46SWarner Losh 	symtabindex = -1;
751ca987d46SWarner Losh 	symstrindex = -1;
752ca987d46SWarner Losh 	for (i = 0; i < ehdr->e_shnum; i++) {
753ca987d46SWarner Losh 		if (shdr[i].sh_type != SHT_SYMTAB)
754ca987d46SWarner Losh 			continue;
755ca987d46SWarner Losh 		for (j = 0; j < ehdr->e_phnum; j++) {
756ca987d46SWarner Losh 			if (phdr[j].p_type != PT_LOAD)
757ca987d46SWarner Losh 				continue;
758ca987d46SWarner Losh 			if (shdr[i].sh_offset >= phdr[j].p_offset &&
759ca987d46SWarner Losh 			    (shdr[i].sh_offset + shdr[i].sh_size <=
760ca987d46SWarner Losh 			    phdr[j].p_offset + phdr[j].p_filesz)) {
761ca987d46SWarner Losh 				shdr[i].sh_offset = 0;
762ca987d46SWarner Losh 				shdr[i].sh_size = 0;
763ca987d46SWarner Losh 				break;
764ca987d46SWarner Losh 			}
765ca987d46SWarner Losh 		}
766ca987d46SWarner Losh 		if (shdr[i].sh_offset == 0 || shdr[i].sh_size == 0)
767ca987d46SWarner Losh 			continue;	/* alread loaded in a PT_LOAD above */
768ca987d46SWarner Losh 		/* Save it for loading below */
769ca987d46SWarner Losh 		symtabindex = i;
770ca987d46SWarner Losh 		symstrindex = shdr[i].sh_link;
771ca987d46SWarner Losh 	}
772ca987d46SWarner Losh 	if (symtabindex < 0 || symstrindex < 0)
773ca987d46SWarner Losh 		goto nosyms;
774ca987d46SWarner Losh 
775ca987d46SWarner Losh 	/* Ok, committed to a load. */
776ec042f46SSimon J. Gerraty 	if (module_verbose >= MODULE_VERBOSE_FULL)
777ca987d46SWarner Losh 		printf("syms=[");
778ca987d46SWarner Losh 	ssym = lastaddr;
779ca987d46SWarner Losh 	for (i = symtabindex; i >= 0; i = symstrindex) {
780ca987d46SWarner Losh 		char	*secname;
781ca987d46SWarner Losh 
782ca987d46SWarner Losh 		switch(shdr[i].sh_type) {
783ca987d46SWarner Losh 		case SHT_SYMTAB:		/* Symbol table */
784ca987d46SWarner Losh 			secname = "symtab";
785ca987d46SWarner Losh 			break;
786ca987d46SWarner Losh 		case SHT_STRTAB:		/* String table */
787ca987d46SWarner Losh 			secname = "strtab";
788ca987d46SWarner Losh 			break;
789ca987d46SWarner Losh 		default:
790ca987d46SWarner Losh 			secname = "WHOA!!";
791ca987d46SWarner Losh 			break;
792ca987d46SWarner Losh 		}
793ca987d46SWarner Losh 		size = shdr[i].sh_size;
7945cca46a9SEd Maste 
7955cca46a9SEd Maste 		archsw.arch_copyin(&size, lastaddr, sizeof(size));
7965cca46a9SEd Maste 		lastaddr += sizeof(size);
797ca987d46SWarner Losh 
798ec042f46SSimon J. Gerraty 		if (module_verbose >= MODULE_VERBOSE_FULL) {
799ca987d46SWarner Losh 			printf("\n%s: 0x%jx@0x%jx -> 0x%jx-0x%jx", secname,
800ca987d46SWarner Losh 			    (uintmax_t)shdr[i].sh_size, (uintmax_t)shdr[i].sh_offset,
8016b632498SEd Maste 			    (uintmax_t)lastaddr,
8026b632498SEd Maste 			    (uintmax_t)(lastaddr + shdr[i].sh_size));
803ec042f46SSimon J. Gerraty 		} else if (module_verbose > MODULE_VERBOSE_SILENT) {
804ca987d46SWarner Losh 			if (i == symstrindex)
805ca987d46SWarner Losh 				printf("+");
806ca987d46SWarner Losh 			printf("0x%lx+0x%lx", (long)sizeof(size), (long)size);
807ec042f46SSimon J. Gerraty 		}
808afc571b1SSimon J. Gerraty 		if (VECTX_LSEEK(VECTX_HANDLE(ef), (off_t)shdr[i].sh_offset, SEEK_SET) == -1) {
8096b632498SEd Maste 			printf("\nelf" __XSTRING(__ELF_WORD_SIZE)
8106b632498SEd Maste 			   "_loadimage: could not seek for symbols - skipped!");
811ca987d46SWarner Losh 			lastaddr = ssym;
812ca987d46SWarner Losh 			ssym = 0;
813ca987d46SWarner Losh 			goto nosyms;
814ca987d46SWarner Losh 		}
815afc571b1SSimon J. Gerraty 		result = archsw.arch_readin(VECTX_HANDLE(ef), lastaddr, shdr[i].sh_size);
816ca987d46SWarner Losh 		if (result < 0 || (size_t)result != shdr[i].sh_size) {
8176b632498SEd Maste 			printf("\nelf" __XSTRING(__ELF_WORD_SIZE)
8186b632498SEd Maste 			    "_loadimage: could not read symbols - skipped! "
8196b632498SEd Maste 			    "(%ju != %ju)", (uintmax_t)result,
820ca987d46SWarner Losh 			    (uintmax_t)shdr[i].sh_size);
821ca987d46SWarner Losh 			lastaddr = ssym;
822ca987d46SWarner Losh 			ssym = 0;
823ca987d46SWarner Losh 			goto nosyms;
824ca987d46SWarner Losh 		}
825ca987d46SWarner Losh 		/* Reset offsets relative to ssym */
826ca987d46SWarner Losh 		lastaddr += shdr[i].sh_size;
827ca987d46SWarner Losh 		lastaddr = roundup(lastaddr, sizeof(size));
828ca987d46SWarner Losh 		if (i == symtabindex)
829ca987d46SWarner Losh 			symtabindex = -1;
830ca987d46SWarner Losh 		else if (i == symstrindex)
831ca987d46SWarner Losh 			symstrindex = -1;
832ca987d46SWarner Losh 	}
833ca987d46SWarner Losh 	esym = lastaddr;
834ec042f46SSimon J. Gerraty 	if (module_verbose >= MODULE_VERBOSE_FULL)
835ca987d46SWarner Losh 		printf("]");
836ca987d46SWarner Losh 
8375cca46a9SEd Maste 	file_addmetadata(fp, MODINFOMD_SSYM, sizeof(ssym), &ssym);
8385cca46a9SEd Maste 	file_addmetadata(fp, MODINFOMD_ESYM, sizeof(esym), &esym);
839ca987d46SWarner Losh 
840ca987d46SWarner Losh nosyms:
841ec042f46SSimon J. Gerraty 	if (module_verbose > MODULE_VERBOSE_SILENT)
842ca987d46SWarner Losh 		printf("\n");
843ca987d46SWarner Losh 
844ca987d46SWarner Losh 	ret = lastaddr - firstaddr;
845ca987d46SWarner Losh 	fp->f_addr = firstaddr;
846ca987d46SWarner Losh 
847ca987d46SWarner Losh 	php = NULL;
848ca987d46SWarner Losh 	for (i = 0; i < ehdr->e_phnum; i++) {
849ca987d46SWarner Losh 		if (phdr[i].p_type == PT_DYNAMIC) {
850ca987d46SWarner Losh 			php = phdr + i;
851ca987d46SWarner Losh 			adp = php->p_vaddr;
8526b632498SEd Maste 			file_addmetadata(fp, MODINFOMD_DYNAMIC, sizeof(adp),
8536b632498SEd Maste 			    &adp);
854ca987d46SWarner Losh 			break;
855ca987d46SWarner Losh 		}
856ca987d46SWarner Losh 	}
857ca987d46SWarner Losh 
858ca987d46SWarner Losh 	if (php == NULL) /* this is bad, we cannot get to symbols or _DYNAMIC */
859ca987d46SWarner Losh 		goto out;
860ca987d46SWarner Losh 
861ca987d46SWarner Losh 	ndp = php->p_filesz / sizeof(Elf_Dyn);
862ca987d46SWarner Losh 	if (ndp == 0)
863ca987d46SWarner Losh 		goto out;
864ca987d46SWarner Losh 	dp = malloc(php->p_filesz);
865ca987d46SWarner Losh 	if (dp == NULL)
866ca987d46SWarner Losh 		goto out;
867ca987d46SWarner Losh 	archsw.arch_copyout(php->p_vaddr + off, dp, php->p_filesz);
868ca987d46SWarner Losh 
869ca987d46SWarner Losh 	ef->strsz = 0;
870ca987d46SWarner Losh 	for (i = 0; i < ndp; i++) {
871ca987d46SWarner Losh 		if (dp[i].d_tag == 0)
872ca987d46SWarner Losh 			break;
873ca987d46SWarner Losh 		switch (dp[i].d_tag) {
874ca987d46SWarner Losh 		case DT_HASH:
8756b632498SEd Maste 			ef->hashtab =
8766b632498SEd Maste 			    (Elf_Hashelt*)(uintptr_t)(dp[i].d_un.d_ptr + off);
877ca987d46SWarner Losh 			break;
878ca987d46SWarner Losh 		case DT_STRTAB:
8796b632498SEd Maste 			ef->strtab =
8806b632498SEd Maste 			    (char *)(uintptr_t)(dp[i].d_un.d_ptr + off);
881ca987d46SWarner Losh 			break;
882ca987d46SWarner Losh 		case DT_STRSZ:
883ca987d46SWarner Losh 			ef->strsz = dp[i].d_un.d_val;
884ca987d46SWarner Losh 			break;
885ca987d46SWarner Losh 		case DT_SYMTAB:
8866b632498SEd Maste 			ef->symtab =
8876b632498SEd Maste 			    (Elf_Sym *)(uintptr_t)(dp[i].d_un.d_ptr + off);
888ca987d46SWarner Losh 			break;
889ca987d46SWarner Losh 		case DT_REL:
8906b632498SEd Maste 			ef->rel =
8916b632498SEd Maste 			    (Elf_Rel *)(uintptr_t)(dp[i].d_un.d_ptr + off);
892ca987d46SWarner Losh 			break;
893ca987d46SWarner Losh 		case DT_RELSZ:
894ca987d46SWarner Losh 			ef->relsz = dp[i].d_un.d_val;
895ca987d46SWarner Losh 			break;
896ca987d46SWarner Losh 		case DT_RELA:
8976b632498SEd Maste 			ef->rela =
8986b632498SEd Maste 			    (Elf_Rela *)(uintptr_t)(dp[i].d_un.d_ptr + off);
899ca987d46SWarner Losh 			break;
900ca987d46SWarner Losh 		case DT_RELASZ:
901ca987d46SWarner Losh 			ef->relasz = dp[i].d_un.d_val;
902ca987d46SWarner Losh 			break;
903ca987d46SWarner Losh 		default:
904ca987d46SWarner Losh 			break;
905ca987d46SWarner Losh 		}
906ca987d46SWarner Losh 	}
907ca987d46SWarner Losh 	if (ef->hashtab == NULL || ef->symtab == NULL ||
908ca987d46SWarner Losh 	    ef->strtab == NULL || ef->strsz == 0)
909ca987d46SWarner Losh 		goto out;
910ca987d46SWarner Losh 	COPYOUT(ef->hashtab, &ef->nbuckets, sizeof(ef->nbuckets));
911ca987d46SWarner Losh 	COPYOUT(ef->hashtab + 1, &ef->nchains, sizeof(ef->nchains));
912ca987d46SWarner Losh 	ef->buckets = ef->hashtab + 2;
913ca987d46SWarner Losh 	ef->chains = ef->buckets + ef->nbuckets;
914ca987d46SWarner Losh 
9156c7a932dSToomas Soome 	if (__elfN(lookup_symbol)(ef, "__start_set_modmetadata_set", &sym,
9166c7a932dSToomas Soome 	    STT_NOTYPE) != 0)
917ca987d46SWarner Losh 		return 0;
918ca987d46SWarner Losh 	p_start = sym.st_value + ef->off;
9196c7a932dSToomas Soome 	if (__elfN(lookup_symbol)(ef, "__stop_set_modmetadata_set", &sym,
9206c7a932dSToomas Soome 	    STT_NOTYPE) != 0)
9211ea87e2aSMark Johnston 		return 0;
922ca987d46SWarner Losh 	p_end = sym.st_value + ef->off;
923ca987d46SWarner Losh 
924ca987d46SWarner Losh 	if (__elfN(parse_modmetadata)(fp, ef, p_start, p_end) == 0)
925ca987d46SWarner Losh 		goto out;
926ca987d46SWarner Losh 
927ca987d46SWarner Losh 	if (ef->kernel)		/* kernel must not depend on anything */
928ca987d46SWarner Losh 		goto out;
929ca987d46SWarner Losh 
930ca987d46SWarner Losh out:
931ca987d46SWarner Losh 	if (dp)
932ca987d46SWarner Losh 		free(dp);
933ca987d46SWarner Losh 	if (shdr)
934ca987d46SWarner Losh 		free(shdr);
935ca987d46SWarner Losh 	return ret;
936ca987d46SWarner Losh }
937ca987d46SWarner Losh 
938ca987d46SWarner Losh static char invalid_name[] = "bad";
939ca987d46SWarner Losh 
940ca987d46SWarner Losh char *
941ca987d46SWarner Losh fake_modname(const char *name)
942ca987d46SWarner Losh {
943ca987d46SWarner Losh 	const char *sp, *ep;
944ca987d46SWarner Losh 	char *fp;
945ca987d46SWarner Losh 	size_t len;
946ca987d46SWarner Losh 
947ca987d46SWarner Losh 	sp = strrchr(name, '/');
948ca987d46SWarner Losh 	if (sp)
949ca987d46SWarner Losh 		sp++;
950ca987d46SWarner Losh 	else
951ca987d46SWarner Losh 		sp = name;
952fe6ee081SEd Maste 
953fe6ee081SEd Maste 	ep = strrchr(sp, '.');
954fe6ee081SEd Maste 	if (ep == NULL) {
955fe6ee081SEd Maste 		ep = sp + strlen(sp);
956fe6ee081SEd Maste 	}
957fe6ee081SEd Maste 	if (ep == sp) {
958ca987d46SWarner Losh 		sp = invalid_name;
959ca987d46SWarner Losh 		ep = invalid_name + sizeof(invalid_name) - 1;
960ca987d46SWarner Losh 	}
961fe6ee081SEd Maste 
962ca987d46SWarner Losh 	len = ep - sp;
963ca987d46SWarner Losh 	fp = malloc(len + 1);
964ca987d46SWarner Losh 	if (fp == NULL)
965ca987d46SWarner Losh 		return NULL;
966ca987d46SWarner Losh 	memcpy(fp, sp, len);
967ca987d46SWarner Losh 	fp[len] = '\0';
968ca987d46SWarner Losh 	return fp;
969ca987d46SWarner Losh }
970ca987d46SWarner Losh 
971ca987d46SWarner Losh #if (defined(__i386__) || defined(__powerpc__)) && __ELF_WORD_SIZE == 64
972ca987d46SWarner Losh struct mod_metadata64 {
973ca987d46SWarner Losh 	int		md_version;	/* structure version MDTV_* */
974ca987d46SWarner Losh 	int		md_type;	/* type of entry MDT_* */
97556e53cb8SWarner Losh 	uint64_t	md_data;	/* specific data */
97656e53cb8SWarner Losh 	uint64_t	md_cval;	/* common string label */
977ca987d46SWarner Losh };
978ca987d46SWarner Losh #endif
979ca987d46SWarner Losh #if defined(__amd64__) && __ELF_WORD_SIZE == 32
980ca987d46SWarner Losh struct mod_metadata32 {
981ca987d46SWarner Losh 	int		md_version;	/* structure version MDTV_* */
982ca987d46SWarner Losh 	int		md_type;	/* type of entry MDT_* */
98356e53cb8SWarner Losh 	uint32_t	md_data;	/* specific data */
98456e53cb8SWarner Losh 	uint32_t	md_cval;	/* common string label */
985ca987d46SWarner Losh };
986ca987d46SWarner Losh #endif
987ca987d46SWarner Losh 
988ca987d46SWarner Losh int
98956e53cb8SWarner Losh __elfN(load_modmetadata)(struct preloaded_file *fp, uint64_t dest)
990ca987d46SWarner Losh {
991ca987d46SWarner Losh 	struct elf_file		 ef;
992ca987d46SWarner Losh 	int			 err, i, j;
993ca987d46SWarner Losh 	Elf_Shdr		*sh_meta, *shdr = NULL;
994ca987d46SWarner Losh 	Elf_Shdr		*sh_data[2];
995ca987d46SWarner Losh 	char			*shstrtab = NULL;
996ca987d46SWarner Losh 	size_t			 size;
997ca987d46SWarner Losh 	Elf_Addr		 p_start, p_end;
998ca987d46SWarner Losh 
999ca987d46SWarner Losh 	bzero(&ef, sizeof(struct elf_file));
1000ca987d46SWarner Losh 	ef.fd = -1;
1001ca987d46SWarner Losh 
1002ca987d46SWarner Losh 	err = __elfN(load_elf_header)(fp->f_name, &ef);
1003ca987d46SWarner Losh 	if (err != 0)
1004ca987d46SWarner Losh 		goto out;
1005ca987d46SWarner Losh 
1006ca987d46SWarner Losh 	if (ef.kernel == 1 || ef.ehdr->e_type == ET_EXEC) {
1007ca987d46SWarner Losh 		ef.kernel = 1;
1008ca987d46SWarner Losh 	} else if (ef.ehdr->e_type != ET_DYN) {
1009ca987d46SWarner Losh 		err = EFTYPE;
1010ca987d46SWarner Losh 		goto out;
1011ca987d46SWarner Losh 	}
1012ca987d46SWarner Losh 
1013db711744SWarner Losh 	size = (size_t)ef.ehdr->e_shnum * (size_t)ef.ehdr->e_shentsize;
1014afc571b1SSimon J. Gerraty 	shdr = alloc_pread(VECTX_HANDLE(&ef), ef.ehdr->e_shoff, size);
1015ca987d46SWarner Losh 	if (shdr == NULL) {
1016ca987d46SWarner Losh 		err = ENOMEM;
1017ca987d46SWarner Losh 		goto out;
1018ca987d46SWarner Losh 	}
1019ca987d46SWarner Losh 
1020ca987d46SWarner Losh 	/* Load shstrtab. */
1021afc571b1SSimon J. Gerraty 	shstrtab = alloc_pread(VECTX_HANDLE(&ef), shdr[ef.ehdr->e_shstrndx].sh_offset,
1022ca987d46SWarner Losh 	    shdr[ef.ehdr->e_shstrndx].sh_size);
1023ca987d46SWarner Losh 	if (shstrtab == NULL) {
1024ca987d46SWarner Losh 		printf("\nelf" __XSTRING(__ELF_WORD_SIZE)
1025ca987d46SWarner Losh 		    "load_modmetadata: unable to load shstrtab\n");
1026ca987d46SWarner Losh 		err = EFTYPE;
1027ca987d46SWarner Losh 		goto out;
1028ca987d46SWarner Losh 	}
1029ca987d46SWarner Losh 
1030ca987d46SWarner Losh 	/* Find set_modmetadata_set and data sections. */
1031ca987d46SWarner Losh 	sh_data[0] = sh_data[1] = sh_meta = NULL;
1032ca987d46SWarner Losh 	for (i = 0, j = 0; i < ef.ehdr->e_shnum; i++) {
1033ca987d46SWarner Losh 		if (strcmp(&shstrtab[shdr[i].sh_name],
1034ca987d46SWarner Losh 		    "set_modmetadata_set") == 0) {
1035ca987d46SWarner Losh 			sh_meta = &shdr[i];
1036ca987d46SWarner Losh 		}
1037ca987d46SWarner Losh 		if ((strcmp(&shstrtab[shdr[i].sh_name], ".data") == 0) ||
1038ca987d46SWarner Losh 		    (strcmp(&shstrtab[shdr[i].sh_name], ".rodata") == 0)) {
1039ca987d46SWarner Losh 			sh_data[j++] = &shdr[i];
1040ca987d46SWarner Losh 		}
1041ca987d46SWarner Losh 	}
1042ca987d46SWarner Losh 	if (sh_meta == NULL || sh_data[0] == NULL || sh_data[1] == NULL) {
1043ca987d46SWarner Losh 		printf("\nelf" __XSTRING(__ELF_WORD_SIZE)
1044ca987d46SWarner Losh     "load_modmetadata: unable to find set_modmetadata_set or data sections\n");
1045ca987d46SWarner Losh 		err = EFTYPE;
1046ca987d46SWarner Losh 		goto out;
1047ca987d46SWarner Losh 	}
1048ca987d46SWarner Losh 
1049ca987d46SWarner Losh 	/* Load set_modmetadata_set into memory */
1050afc571b1SSimon J. Gerraty 	err = kern_pread(VECTX_HANDLE(&ef), dest, sh_meta->sh_size, sh_meta->sh_offset);
1051ca987d46SWarner Losh 	if (err != 0) {
1052ca987d46SWarner Losh 		printf("\nelf" __XSTRING(__ELF_WORD_SIZE)
1053ca987d46SWarner Losh     "load_modmetadata: unable to load set_modmetadata_set: %d\n", err);
1054ca987d46SWarner Losh 		goto out;
1055ca987d46SWarner Losh 	}
1056ca987d46SWarner Losh 	p_start = dest;
1057ca987d46SWarner Losh 	p_end = dest + sh_meta->sh_size;
1058ca987d46SWarner Losh 	dest += sh_meta->sh_size;
1059ca987d46SWarner Losh 
1060ca987d46SWarner Losh 	/* Load data sections into memory. */
1061afc571b1SSimon J. Gerraty 	err = kern_pread(VECTX_HANDLE(&ef), dest, sh_data[0]->sh_size,
1062ca987d46SWarner Losh 	    sh_data[0]->sh_offset);
1063ca987d46SWarner Losh 	if (err != 0) {
1064ca987d46SWarner Losh 		printf("\nelf" __XSTRING(__ELF_WORD_SIZE)
1065ca987d46SWarner Losh 		    "load_modmetadata: unable to load data: %d\n", err);
1066ca987d46SWarner Losh 		goto out;
1067ca987d46SWarner Losh 	}
1068ca987d46SWarner Losh 
1069ca987d46SWarner Losh 	/*
1070ca987d46SWarner Losh 	 * We have to increment the dest, so that the offset is the same into
1071ca987d46SWarner Losh 	 * both the .rodata and .data sections.
1072ca987d46SWarner Losh 	 */
1073ca987d46SWarner Losh 	ef.off = -(sh_data[0]->sh_addr - dest);
1074ca987d46SWarner Losh 	dest +=	(sh_data[1]->sh_addr - sh_data[0]->sh_addr);
1075ca987d46SWarner Losh 
1076afc571b1SSimon J. Gerraty 	err = kern_pread(VECTX_HANDLE(&ef), dest, sh_data[1]->sh_size,
1077ca987d46SWarner Losh 	    sh_data[1]->sh_offset);
1078ca987d46SWarner Losh 	if (err != 0) {
1079ca987d46SWarner Losh 		printf("\nelf" __XSTRING(__ELF_WORD_SIZE)
1080ca987d46SWarner Losh 		    "load_modmetadata: unable to load data: %d\n", err);
1081ca987d46SWarner Losh 		goto out;
1082ca987d46SWarner Losh 	}
1083ca987d46SWarner Losh 
1084ca987d46SWarner Losh 	err = __elfN(parse_modmetadata)(fp, &ef, p_start, p_end);
1085ca987d46SWarner Losh 	if (err != 0) {
1086ca987d46SWarner Losh 		printf("\nelf" __XSTRING(__ELF_WORD_SIZE)
1087ca987d46SWarner Losh 		    "load_modmetadata: unable to parse metadata: %d\n", err);
1088ca987d46SWarner Losh 		goto out;
1089ca987d46SWarner Losh 	}
1090ca987d46SWarner Losh 
1091ca987d46SWarner Losh out:
1092ca987d46SWarner Losh 	if (shstrtab != NULL)
1093ca987d46SWarner Losh 		free(shstrtab);
1094ca987d46SWarner Losh 	if (shdr != NULL)
1095ca987d46SWarner Losh 		free(shdr);
1096ca987d46SWarner Losh 	if (ef.firstpage != NULL)
1097ca987d46SWarner Losh 		free(ef.firstpage);
1098afc571b1SSimon J. Gerraty 	if (ef.fd != -1) {
1099afc571b1SSimon J. Gerraty #ifdef LOADER_VERIEXEC_VECTX
1100afc571b1SSimon J. Gerraty 		if (!err && ef.vctx) {
1101afc571b1SSimon J. Gerraty 			int verror;
1102afc571b1SSimon J. Gerraty 
1103afc571b1SSimon J. Gerraty 			verror = vectx_close(ef.vctx, VE_MUST, __func__);
1104afc571b1SSimon J. Gerraty 			if (verror) {
1105afc571b1SSimon J. Gerraty 				err = EAUTH;
1106afc571b1SSimon J. Gerraty 				file_discard(fp);
1107afc571b1SSimon J. Gerraty 			}
1108afc571b1SSimon J. Gerraty 		}
1109afc571b1SSimon J. Gerraty #endif
1110ca987d46SWarner Losh 		close(ef.fd);
1111afc571b1SSimon J. Gerraty 	}
1112ca987d46SWarner Losh 	return (err);
1113ca987d46SWarner Losh }
1114ca987d46SWarner Losh 
1115ca987d46SWarner Losh int
1116ca987d46SWarner Losh __elfN(parse_modmetadata)(struct preloaded_file *fp, elf_file_t ef,
1117ca987d46SWarner Losh     Elf_Addr p_start, Elf_Addr p_end)
1118ca987d46SWarner Losh {
1119ca987d46SWarner Losh 	struct mod_metadata md;
1120ca987d46SWarner Losh #if (defined(__i386__) || defined(__powerpc__)) && __ELF_WORD_SIZE == 64
1121ca987d46SWarner Losh 	struct mod_metadata64 md64;
1122ca987d46SWarner Losh #elif defined(__amd64__) && __ELF_WORD_SIZE == 32
1123ca987d46SWarner Losh 	struct mod_metadata32 md32;
1124ca987d46SWarner Losh #endif
1125ca987d46SWarner Losh 	struct mod_depend *mdepend;
1126ca987d46SWarner Losh 	struct mod_version mver;
1127ca987d46SWarner Losh 	char *s;
1128ca987d46SWarner Losh 	int error, modcnt, minfolen;
1129ca987d46SWarner Losh 	Elf_Addr v, p;
1130ca987d46SWarner Losh 
1131ca987d46SWarner Losh 	modcnt = 0;
1132ca987d46SWarner Losh 	p = p_start;
1133ca987d46SWarner Losh 	while (p < p_end) {
1134ca987d46SWarner Losh 		COPYOUT(p, &v, sizeof(v));
1135ca987d46SWarner Losh 		error = __elfN(reloc_ptr)(fp, ef, p, &v, sizeof(v));
1136ca987d46SWarner Losh 		if (error == EOPNOTSUPP)
1137ca987d46SWarner Losh 			v += ef->off;
1138ca987d46SWarner Losh 		else if (error != 0)
1139ca987d46SWarner Losh 			return (error);
1140ca987d46SWarner Losh #if (defined(__i386__) || defined(__powerpc__)) && __ELF_WORD_SIZE == 64
1141ca987d46SWarner Losh 		COPYOUT(v, &md64, sizeof(md64));
1142ca987d46SWarner Losh 		error = __elfN(reloc_ptr)(fp, ef, v, &md64, sizeof(md64));
1143ca987d46SWarner Losh 		if (error == EOPNOTSUPP) {
1144ca987d46SWarner Losh 			md64.md_cval += ef->off;
1145ca987d46SWarner Losh 			md64.md_data += ef->off;
1146ca987d46SWarner Losh 		} else if (error != 0)
1147ca987d46SWarner Losh 			return (error);
1148ca987d46SWarner Losh 		md.md_version = md64.md_version;
1149ca987d46SWarner Losh 		md.md_type = md64.md_type;
1150ca987d46SWarner Losh 		md.md_cval = (const char *)(uintptr_t)md64.md_cval;
1151ca987d46SWarner Losh 		md.md_data = (void *)(uintptr_t)md64.md_data;
1152ca987d46SWarner Losh #elif defined(__amd64__) && __ELF_WORD_SIZE == 32
1153ca987d46SWarner Losh 		COPYOUT(v, &md32, sizeof(md32));
1154ca987d46SWarner Losh 		error = __elfN(reloc_ptr)(fp, ef, v, &md32, sizeof(md32));
1155ca987d46SWarner Losh 		if (error == EOPNOTSUPP) {
1156ca987d46SWarner Losh 			md32.md_cval += ef->off;
1157ca987d46SWarner Losh 			md32.md_data += ef->off;
1158ca987d46SWarner Losh 		} else if (error != 0)
1159ca987d46SWarner Losh 			return (error);
1160ca987d46SWarner Losh 		md.md_version = md32.md_version;
1161ca987d46SWarner Losh 		md.md_type = md32.md_type;
1162ca987d46SWarner Losh 		md.md_cval = (const char *)(uintptr_t)md32.md_cval;
1163ca987d46SWarner Losh 		md.md_data = (void *)(uintptr_t)md32.md_data;
1164ca987d46SWarner Losh #else
1165ca987d46SWarner Losh 		COPYOUT(v, &md, sizeof(md));
1166ca987d46SWarner Losh 		error = __elfN(reloc_ptr)(fp, ef, v, &md, sizeof(md));
1167ca987d46SWarner Losh 		if (error == EOPNOTSUPP) {
1168ca987d46SWarner Losh 			md.md_cval += ef->off;
11696b632498SEd Maste 			md.md_data = (void *)((uintptr_t)md.md_data +
11706b632498SEd Maste 			    (uintptr_t)ef->off);
1171ca987d46SWarner Losh 		} else if (error != 0)
1172ca987d46SWarner Losh 			return (error);
1173ca987d46SWarner Losh #endif
1174ca987d46SWarner Losh 		p += sizeof(Elf_Addr);
1175ca987d46SWarner Losh 		switch(md.md_type) {
1176ca987d46SWarner Losh 		case MDT_DEPEND:
1177ca987d46SWarner Losh 			if (ef->kernel) /* kernel must not depend on anything */
1178ca987d46SWarner Losh 				break;
1179ca987d46SWarner Losh 			s = strdupout((vm_offset_t)md.md_cval);
1180ca987d46SWarner Losh 			minfolen = sizeof(*mdepend) + strlen(s) + 1;
1181ca987d46SWarner Losh 			mdepend = malloc(minfolen);
1182ca987d46SWarner Losh 			if (mdepend == NULL)
1183ca987d46SWarner Losh 				return ENOMEM;
11846b632498SEd Maste 			COPYOUT((vm_offset_t)md.md_data, mdepend,
11856b632498SEd Maste 			    sizeof(*mdepend));
1186ca987d46SWarner Losh 			strcpy((char*)(mdepend + 1), s);
1187ca987d46SWarner Losh 			free(s);
11886b632498SEd Maste 			file_addmetadata(fp, MODINFOMD_DEPLIST, minfolen,
11896b632498SEd Maste 			    mdepend);
1190ca987d46SWarner Losh 			free(mdepend);
1191ca987d46SWarner Losh 			break;
1192ca987d46SWarner Losh 		case MDT_VERSION:
1193ca987d46SWarner Losh 			s = strdupout((vm_offset_t)md.md_cval);
1194ca987d46SWarner Losh 			COPYOUT((vm_offset_t)md.md_data, &mver, sizeof(mver));
1195ca987d46SWarner Losh 			file_addmodule(fp, s, mver.mv_version, NULL);
1196ca987d46SWarner Losh 			free(s);
1197ca987d46SWarner Losh 			modcnt++;
1198ca987d46SWarner Losh 			break;
1199ca987d46SWarner Losh 		}
1200ca987d46SWarner Losh 	}
1201ca987d46SWarner Losh 	if (modcnt == 0) {
1202ca987d46SWarner Losh 		s = fake_modname(fp->f_name);
1203ca987d46SWarner Losh 		file_addmodule(fp, s, 1, NULL);
1204ca987d46SWarner Losh 		free(s);
1205ca987d46SWarner Losh 	}
1206ca987d46SWarner Losh 	return 0;
1207ca987d46SWarner Losh }
1208ca987d46SWarner Losh 
1209ca987d46SWarner Losh static unsigned long
1210ca987d46SWarner Losh elf_hash(const char *name)
1211ca987d46SWarner Losh {
1212ca987d46SWarner Losh 	const unsigned char *p = (const unsigned char *) name;
1213ca987d46SWarner Losh 	unsigned long h = 0;
1214ca987d46SWarner Losh 	unsigned long g;
1215ca987d46SWarner Losh 
1216ca987d46SWarner Losh 	while (*p != '\0') {
1217ca987d46SWarner Losh 		h = (h << 4) + *p++;
1218ca987d46SWarner Losh 		if ((g = h & 0xf0000000) != 0)
1219ca987d46SWarner Losh 			h ^= g >> 24;
1220ca987d46SWarner Losh 		h &= ~g;
1221ca987d46SWarner Losh 	}
1222ca987d46SWarner Losh 	return h;
1223ca987d46SWarner Losh }
1224ca987d46SWarner Losh 
12256b632498SEd Maste static const char __elfN(bad_symtable)[] = "elf" __XSTRING(__ELF_WORD_SIZE)
12266b632498SEd Maste     "_lookup_symbol: corrupt symbol table\n";
1227ca987d46SWarner Losh int
12286c7a932dSToomas Soome __elfN(lookup_symbol)(elf_file_t ef, const char* name, Elf_Sym *symp,
12296c7a932dSToomas Soome     unsigned char type)
1230ca987d46SWarner Losh {
1231ca987d46SWarner Losh 	Elf_Hashelt symnum;
1232ca987d46SWarner Losh 	Elf_Sym sym;
1233ca987d46SWarner Losh 	char *strp;
1234ca987d46SWarner Losh 	unsigned long hash;
1235ca987d46SWarner Losh 
12365e4279a8SRoger Pau Monné 	if (ef->nbuckets == 0) {
12375e4279a8SRoger Pau Monné 		printf(__elfN(bad_symtable));
12385e4279a8SRoger Pau Monné 		return ENOENT;
12395e4279a8SRoger Pau Monné 	}
12405e4279a8SRoger Pau Monné 
1241ca987d46SWarner Losh 	hash = elf_hash(name);
1242ca987d46SWarner Losh 	COPYOUT(&ef->buckets[hash % ef->nbuckets], &symnum, sizeof(symnum));
1243ca987d46SWarner Losh 
1244ca987d46SWarner Losh 	while (symnum != STN_UNDEF) {
1245ca987d46SWarner Losh 		if (symnum >= ef->nchains) {
1246ca987d46SWarner Losh 			printf(__elfN(bad_symtable));
1247ca987d46SWarner Losh 			return ENOENT;
1248ca987d46SWarner Losh 		}
1249ca987d46SWarner Losh 
1250ca987d46SWarner Losh 		COPYOUT(ef->symtab + symnum, &sym, sizeof(sym));
1251ca987d46SWarner Losh 		if (sym.st_name == 0) {
1252ca987d46SWarner Losh 			printf(__elfN(bad_symtable));
1253ca987d46SWarner Losh 			return ENOENT;
1254ca987d46SWarner Losh 		}
1255ca987d46SWarner Losh 
1256ca987d46SWarner Losh 		strp = strdupout((vm_offset_t)(ef->strtab + sym.st_name));
1257ca987d46SWarner Losh 		if (strcmp(name, strp) == 0) {
1258ca987d46SWarner Losh 			free(strp);
12590701dbdaSKyle Evans 			if (sym.st_shndx != SHN_UNDEF && sym.st_value != 0 &&
12600701dbdaSKyle Evans 			    ELF_ST_TYPE(sym.st_info) == type) {
1261ca987d46SWarner Losh 				*symp = sym;
1262ca987d46SWarner Losh 				return 0;
1263ca987d46SWarner Losh 			}
1264ca987d46SWarner Losh 			return ENOENT;
1265ca987d46SWarner Losh 		}
1266ca987d46SWarner Losh 		free(strp);
1267ca987d46SWarner Losh 		COPYOUT(&ef->chains[symnum], &symnum, sizeof(symnum));
1268ca987d46SWarner Losh 	}
1269ca987d46SWarner Losh 	return ENOENT;
1270ca987d46SWarner Losh }
1271ca987d46SWarner Losh 
1272ca987d46SWarner Losh /*
1273ca987d46SWarner Losh  * Apply any intra-module relocations to the value. p is the load address
1274ca987d46SWarner Losh  * of the value and val/len is the value to be modified. This does NOT modify
1275ca987d46SWarner Losh  * the image in-place, because this is done by kern_linker later on.
1276ca987d46SWarner Losh  *
1277ca987d46SWarner Losh  * Returns EOPNOTSUPP if no relocation method is supplied.
1278ca987d46SWarner Losh  */
1279ca987d46SWarner Losh static int
1280ca987d46SWarner Losh __elfN(reloc_ptr)(struct preloaded_file *mp, elf_file_t ef,
1281ca987d46SWarner Losh     Elf_Addr p, void *val, size_t len)
1282ca987d46SWarner Losh {
1283ca987d46SWarner Losh 	size_t n;
1284ca987d46SWarner Losh 	Elf_Rela a;
1285ca987d46SWarner Losh 	Elf_Rel r;
1286ca987d46SWarner Losh 	int error;
1287ca987d46SWarner Losh 
1288ca987d46SWarner Losh 	/*
1289ca987d46SWarner Losh 	 * The kernel is already relocated, but we still want to apply
1290ca987d46SWarner Losh 	 * offset adjustments.
1291ca987d46SWarner Losh 	 */
1292ca987d46SWarner Losh 	if (ef->kernel)
1293ca987d46SWarner Losh 		return (EOPNOTSUPP);
1294ca987d46SWarner Losh 
1295ca987d46SWarner Losh 	for (n = 0; n < ef->relsz / sizeof(r); n++) {
1296ca987d46SWarner Losh 		COPYOUT(ef->rel + n, &r, sizeof(r));
1297ca987d46SWarner Losh 
1298ca987d46SWarner Losh 		error = __elfN(reloc)(ef, __elfN(symaddr), &r, ELF_RELOC_REL,
1299ca987d46SWarner Losh 		    ef->off, p, val, len);
1300ca987d46SWarner Losh 		if (error != 0)
1301ca987d46SWarner Losh 			return (error);
1302ca987d46SWarner Losh 	}
1303ca987d46SWarner Losh 	for (n = 0; n < ef->relasz / sizeof(a); n++) {
1304ca987d46SWarner Losh 		COPYOUT(ef->rela + n, &a, sizeof(a));
1305ca987d46SWarner Losh 
1306ca987d46SWarner Losh 		error = __elfN(reloc)(ef, __elfN(symaddr), &a, ELF_RELOC_RELA,
1307ca987d46SWarner Losh 		    ef->off, p, val, len);
1308ca987d46SWarner Losh 		if (error != 0)
1309ca987d46SWarner Losh 			return (error);
1310ca987d46SWarner Losh 	}
1311ca987d46SWarner Losh 
1312ca987d46SWarner Losh 	return (0);
1313ca987d46SWarner Losh }
1314ca987d46SWarner Losh 
1315ca987d46SWarner Losh static Elf_Addr
1316ca987d46SWarner Losh __elfN(symaddr)(struct elf_file *ef, Elf_Size symidx)
1317ca987d46SWarner Losh {
1318ca987d46SWarner Losh 
1319ca987d46SWarner Losh 	/* Symbol lookup by index not required here. */
1320ca987d46SWarner Losh 	return (0);
1321ca987d46SWarner Losh }
1322