Home | History | Annotate | Line # | Download | only in kern
subr_kobj.c revision 1.1
      1  1.1  ad /*	$NetBSD: subr_kobj.c,v 1.1 2008/01/04 12:26:20 ad Exp $	*/
      2  1.1  ad 
      3  1.1  ad /*-
      4  1.1  ad  * Copyright (c) 2008 The NetBSD Foundation, Inc.
      5  1.1  ad  * All rights reserved.
      6  1.1  ad  *
      7  1.1  ad  * Redistribution and use in source and binary forms, with or without
      8  1.1  ad  * modification, are permitted provided that the following conditions
      9  1.1  ad  * are met:
     10  1.1  ad  * 1. Redistributions of source code must retain the above copyright
     11  1.1  ad  *    notice, this list of conditions and the following disclaimer.
     12  1.1  ad  * 2. Redistributions in binary form must reproduce the above copyright
     13  1.1  ad  *    notice, this list of conditions and the following disclaimer in the
     14  1.1  ad  *    documentation and/or other materials provided with the distribution.
     15  1.1  ad  * 3. All advertising materials mentioning features or use of this software
     16  1.1  ad  *    must display the following acknowledgement:
     17  1.1  ad  *	This product includes software developed by the NetBSD
     18  1.1  ad  *	Foundation, Inc. and its contributors.
     19  1.1  ad  * 4. Neither the name of The NetBSD Foundation nor the names of its
     20  1.1  ad  *    contributors may be used to endorse or promote products derived
     21  1.1  ad  *    from this software without specific prior written permission.
     22  1.1  ad  *
     23  1.1  ad  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     24  1.1  ad  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     25  1.1  ad  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     26  1.1  ad  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     27  1.1  ad  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     28  1.1  ad  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     29  1.1  ad  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     30  1.1  ad  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     31  1.1  ad  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     32  1.1  ad  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     33  1.1  ad  * POSSIBILITY OF SUCH DAMAGE.
     34  1.1  ad  */
     35  1.1  ad 
     36  1.1  ad /*-
     37  1.1  ad  * Copyright (c) 1998-2000 Doug Rabson
     38  1.1  ad  * Copyright (c) 2004 Peter Wemm
     39  1.1  ad  * All rights reserved.
     40  1.1  ad  *
     41  1.1  ad  * Redistribution and use in source and binary forms, with or without
     42  1.1  ad  * modification, are permitted provided that the following conditions
     43  1.1  ad  * are met:
     44  1.1  ad  * 1. Redistributions of source code must retain the above copyright
     45  1.1  ad  *    notice, this list of conditions and the following disclaimer.
     46  1.1  ad  * 2. Redistributions in binary form must reproduce the above copyright
     47  1.1  ad  *    notice, this list of conditions and the following disclaimer in the
     48  1.1  ad  *    documentation and/or other materials provided with the distribution.
     49  1.1  ad  *
     50  1.1  ad  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
     51  1.1  ad  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     52  1.1  ad  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     53  1.1  ad  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
     54  1.1  ad  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     55  1.1  ad  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     56  1.1  ad  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     57  1.1  ad  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     58  1.1  ad  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     59  1.1  ad  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     60  1.1  ad  * SUCH DAMAGE.
     61  1.1  ad  */
     62  1.1  ad 
     63  1.1  ad /*
     64  1.1  ad  * Kernel loader for ELF objects.
     65  1.1  ad  *
     66  1.1  ad  * TODO: adjust kmem_alloc() calls to avoid needless fragmentation.
     67  1.1  ad  */
     68  1.1  ad 
     69  1.1  ad #include <sys/cdefs.h>
     70  1.1  ad __KERNEL_RCSID(0, "$NetBSD: subr_kobj.c,v 1.1 2008/01/04 12:26:20 ad Exp $");
     71  1.1  ad 
     72  1.1  ad #define	ELFSIZE		ARCH_ELFSIZE
     73  1.1  ad 
     74  1.1  ad #include <sys/param.h>
     75  1.1  ad #include <sys/systm.h>
     76  1.1  ad #include <sys/kernel.h>
     77  1.1  ad #include <sys/kmem.h>
     78  1.1  ad #include <sys/proc.h>
     79  1.1  ad #include <sys/namei.h>
     80  1.1  ad #include <sys/vnode.h>
     81  1.1  ad #include <sys/fcntl.h>
     82  1.1  ad #include <sys/kobj.h>
     83  1.1  ad #include <sys/ksyms.h>
     84  1.1  ad #include <sys/lkm.h>
     85  1.1  ad #include <sys/exec.h>
     86  1.1  ad #include <sys/exec_elf.h>
     87  1.1  ad 
     88  1.1  ad #include <machine/stdarg.h>
     89  1.1  ad 
     90  1.1  ad #include <uvm/uvm_extern.h>
     91  1.1  ad 
     92  1.1  ad typedef struct {
     93  1.1  ad 	void		*addr;
     94  1.1  ad 	Elf_Off		size;
     95  1.1  ad 	int		flags;
     96  1.1  ad 	int		sec;		/* Original section */
     97  1.1  ad 	const char	*name;
     98  1.1  ad } progent_t;
     99  1.1  ad 
    100  1.1  ad typedef struct {
    101  1.1  ad 	Elf_Rel		*rel;
    102  1.1  ad 	int 		nrel;
    103  1.1  ad 	int 		sec;
    104  1.1  ad 	size_t		size;
    105  1.1  ad } relent_t;
    106  1.1  ad 
    107  1.1  ad typedef struct {
    108  1.1  ad 	Elf_Rela	*rela;
    109  1.1  ad 	int		nrela;
    110  1.1  ad 	int		sec;
    111  1.1  ad 	size_t		size;
    112  1.1  ad } relaent_t;
    113  1.1  ad 
    114  1.1  ad struct kobj {
    115  1.1  ad 	char		ko_name[MAXLKMNAME];
    116  1.1  ad 	vaddr_t		ko_address;	/* Relocation address */
    117  1.1  ad 	Elf_Shdr	*ko_shdr;
    118  1.1  ad 	progent_t	*ko_progtab;
    119  1.1  ad 	relaent_t	*ko_relatab;
    120  1.1  ad 	relent_t	*ko_reltab;
    121  1.1  ad 	Elf_Sym		*ko_symtab;	/* Symbol table */
    122  1.1  ad 	char		*ko_strtab;	/* String table */
    123  1.1  ad 	size_t		ko_size;	/* Size of text/data/bss */
    124  1.1  ad 	size_t		ko_symcnt;	/* Number of symbols */
    125  1.1  ad 	size_t		ko_strtabsz;	/* Number of bytes in string table */
    126  1.1  ad 	size_t		ko_shdrsz;
    127  1.1  ad 	int		ko_nrel;
    128  1.1  ad 	int		ko_nrela;
    129  1.1  ad 	int		ko_nprogtab;
    130  1.1  ad 	bool		ko_ksyms;
    131  1.1  ad };
    132  1.1  ad 
    133  1.1  ad static int	kobj_relocate(kobj_t);
    134  1.1  ad static void	kobj_error(const char *, ...);
    135  1.1  ad static int	kobj_read(vnode_t *, void *, size_t, off_t);
    136  1.1  ad static void	kobj_release_mem(kobj_t);
    137  1.1  ad 
    138  1.1  ad extern struct vm_map *lkm_map;
    139  1.1  ad static const char	*kobj_path = "/modules";	/* XXX ??? */
    140  1.1  ad 
    141  1.1  ad /*
    142  1.1  ad  * kobj_load:
    143  1.1  ad  *
    144  1.1  ad  *	Load an ELF object from the file system and link into the
    145  1.1  ad  *	running	kernel image.
    146  1.1  ad  */
    147  1.1  ad int
    148  1.1  ad kobj_load(const char *name, kobj_t *kop)
    149  1.1  ad {
    150  1.1  ad 	struct nameidata nd;
    151  1.1  ad 	kauth_cred_t cred;
    152  1.1  ad 	Elf_Ehdr *hdr;
    153  1.1  ad 	Elf_Shdr *shdr;
    154  1.1  ad 	Elf_Sym *es;
    155  1.1  ad 	vaddr_t mapbase;
    156  1.1  ad 	size_t mapsize;
    157  1.1  ad 	char *path;
    158  1.1  ad 	int error;
    159  1.1  ad 	int symtabindex;
    160  1.1  ad 	int symstrindex;
    161  1.1  ad 	int nsym;
    162  1.1  ad 	int pb, rl, ra;
    163  1.1  ad 	int alignmask;
    164  1.1  ad 	int i, j;
    165  1.1  ad 	kobj_t ko;
    166  1.1  ad 
    167  1.1  ad 	cred = kauth_cred_get();
    168  1.1  ad 	shdr = NULL;
    169  1.1  ad 	mapsize = 0;
    170  1.1  ad 	error = 0;
    171  1.1  ad 	hdr = NULL;
    172  1.1  ad 
    173  1.1  ad 	ko = kmem_zalloc(sizeof(*ko), KM_SLEEP);
    174  1.1  ad 	if (ko == NULL) {
    175  1.1  ad 		return ENOMEM;
    176  1.1  ad 	}
    177  1.1  ad 
    178  1.1  ad 	if (strlcpy(ko->ko_name, name, sizeof(ko->ko_name)) >=
    179  1.1  ad 	    sizeof(ko->ko_name)) {
    180  1.1  ad 	    	error = EINVAL;
    181  1.1  ad 	    	goto out;
    182  1.1  ad 	}
    183  1.1  ad 
    184  1.1  ad 	path = PNBUF_GET();
    185  1.1  ad 	snprintf(path, MAXPATHLEN, "%s/%s", kobj_path, name);
    186  1.1  ad 	NDINIT(&nd, LOOKUP, FOLLOW, UIO_SYSSPACE, path);
    187  1.1  ad 	error = vn_open(&nd, FREAD, 0);
    188  1.1  ad 	PNBUF_PUT(path);
    189  1.1  ad 	if (error != 0) {
    190  1.1  ad 		goto out;
    191  1.1  ad 	}
    192  1.1  ad 
    193  1.1  ad 	/*
    194  1.1  ad 	 * Read the elf header from the file.
    195  1.1  ad 	 */
    196  1.1  ad 	hdr = kmem_alloc(sizeof(*hdr), KM_SLEEP);
    197  1.1  ad 	if (hdr == NULL) {
    198  1.1  ad 		error = ENOMEM;
    199  1.1  ad 		goto out;
    200  1.1  ad 	}
    201  1.1  ad 	error = kobj_read(nd.ni_vp, hdr, sizeof(*hdr), 0);
    202  1.1  ad 	if (error != 0)
    203  1.1  ad 		goto out;
    204  1.1  ad 	if (memcmp(hdr->e_ident, ELFMAG, SELFMAG) != 0) {
    205  1.1  ad 		error = ENOEXEC;
    206  1.1  ad 		goto out;
    207  1.1  ad 	}
    208  1.1  ad 
    209  1.1  ad 	if (hdr->e_ident[EI_VERSION] != EV_CURRENT ||
    210  1.1  ad 	    hdr->e_version != EV_CURRENT) {
    211  1.1  ad 		kobj_error("unsupported file version");
    212  1.1  ad 		error = ENOEXEC;
    213  1.1  ad 		goto out;
    214  1.1  ad 	}
    215  1.1  ad 	if (hdr->e_type != ET_REL) {
    216  1.1  ad 		kobj_error("unsupported file type");
    217  1.1  ad 		error = ENOEXEC;
    218  1.1  ad 		goto out;
    219  1.1  ad 	}
    220  1.1  ad 	switch (hdr->e_machine) {
    221  1.1  ad #if ELFSIZE == 32
    222  1.1  ad 	ELF32_MACHDEP_ID_CASES
    223  1.1  ad #else
    224  1.1  ad 	ELF64_MACHDEP_ID_CASES
    225  1.1  ad #endif
    226  1.1  ad 	default:
    227  1.1  ad 		kobj_error("unsupported machine");
    228  1.1  ad 		error = ENOEXEC;
    229  1.1  ad 		goto out;
    230  1.1  ad 	}
    231  1.1  ad 
    232  1.1  ad 	ko->ko_nprogtab = 0;
    233  1.1  ad 	ko->ko_shdr = 0;
    234  1.1  ad 	ko->ko_nrel = 0;
    235  1.1  ad 	ko->ko_nrela = 0;
    236  1.1  ad 
    237  1.1  ad 	/*
    238  1.1  ad 	 * Allocate and read in the section header.
    239  1.1  ad 	 */
    240  1.1  ad 	ko->ko_shdrsz = hdr->e_shnum * hdr->e_shentsize;
    241  1.1  ad 	if (ko->ko_shdrsz == 0 || hdr->e_shoff == 0 ||
    242  1.1  ad 	    hdr->e_shentsize != sizeof(Elf_Shdr)) {
    243  1.1  ad 		error = ENOEXEC;
    244  1.1  ad 		goto out;
    245  1.1  ad 	}
    246  1.1  ad 	shdr = kmem_alloc(ko->ko_shdrsz, KM_SLEEP);
    247  1.1  ad 	if (shdr == NULL) {
    248  1.1  ad 		error = ENOMEM;
    249  1.1  ad 		goto out;
    250  1.1  ad 	}
    251  1.1  ad 	ko->ko_shdr = shdr;
    252  1.1  ad 	error = kobj_read(nd.ni_vp, shdr, ko->ko_shdrsz, hdr->e_shoff);
    253  1.1  ad 	if (error != 0) {
    254  1.1  ad 		goto out;
    255  1.1  ad 	}
    256  1.1  ad 
    257  1.1  ad 	/*
    258  1.1  ad 	 * Scan the section header for information and table sizing.
    259  1.1  ad 	 */
    260  1.1  ad 	nsym = 0;
    261  1.1  ad 	symtabindex = -1;
    262  1.1  ad 	symstrindex = -1;
    263  1.1  ad 	for (i = 0; i < hdr->e_shnum; i++) {
    264  1.1  ad 		switch (shdr[i].sh_type) {
    265  1.1  ad 		case SHT_PROGBITS:
    266  1.1  ad 		case SHT_NOBITS:
    267  1.1  ad 			ko->ko_nprogtab++;
    268  1.1  ad 			break;
    269  1.1  ad 		case SHT_SYMTAB:
    270  1.1  ad 			nsym++;
    271  1.1  ad 			symtabindex = i;
    272  1.1  ad 			symstrindex = shdr[i].sh_link;
    273  1.1  ad 			break;
    274  1.1  ad 		case SHT_REL:
    275  1.1  ad 			ko->ko_nrel++;
    276  1.1  ad 			break;
    277  1.1  ad 		case SHT_RELA:
    278  1.1  ad 			ko->ko_nrela++;
    279  1.1  ad 			break;
    280  1.1  ad 		case SHT_STRTAB:
    281  1.1  ad 			break;
    282  1.1  ad 		}
    283  1.1  ad 	}
    284  1.1  ad 	if (ko->ko_nprogtab == 0) {
    285  1.1  ad 		kobj_error("file has no contents");
    286  1.1  ad 		error = ENOEXEC;
    287  1.1  ad 		goto out;
    288  1.1  ad 	}
    289  1.1  ad 	if (nsym != 1) {
    290  1.1  ad 		/* Only allow one symbol table for now */
    291  1.1  ad 		kobj_error("file has no valid symbol table");
    292  1.1  ad 		error = ENOEXEC;
    293  1.1  ad 		goto out;
    294  1.1  ad 	}
    295  1.1  ad 	if (symstrindex < 0 || symstrindex > hdr->e_shnum ||
    296  1.1  ad 	    shdr[symstrindex].sh_type != SHT_STRTAB) {
    297  1.1  ad 		kobj_error("file has invalid symbol strings");
    298  1.1  ad 		error = ENOEXEC;
    299  1.1  ad 		goto out;
    300  1.1  ad 	}
    301  1.1  ad 
    302  1.1  ad 	/*
    303  1.1  ad 	 * Allocate space for tracking the load chunks.
    304  1.1  ad 	 */
    305  1.1  ad 	if (ko->ko_nprogtab != 0) {
    306  1.1  ad 		ko->ko_progtab = kmem_zalloc(ko->ko_nprogtab *
    307  1.1  ad 		    sizeof(*ko->ko_progtab), KM_SLEEP);
    308  1.1  ad 		if (ko->ko_progtab == NULL) {
    309  1.1  ad 			error = ENOMEM;
    310  1.1  ad 			goto out;
    311  1.1  ad 		}
    312  1.1  ad 	}
    313  1.1  ad 	if (ko->ko_nrel != 0) {
    314  1.1  ad 		ko->ko_reltab = kmem_zalloc(ko->ko_nrel *
    315  1.1  ad 		    sizeof(*ko->ko_reltab), KM_SLEEP);
    316  1.1  ad 		if (ko->ko_reltab == NULL) {
    317  1.1  ad 			error = ENOMEM;
    318  1.1  ad 			goto out;
    319  1.1  ad 		}
    320  1.1  ad 	}
    321  1.1  ad 	if (ko->ko_nrela != 0) {
    322  1.1  ad 		ko->ko_relatab = kmem_zalloc(ko->ko_nrela *
    323  1.1  ad 		    sizeof(*ko->ko_relatab), KM_SLEEP);
    324  1.1  ad 		if (ko->ko_relatab == NULL) {
    325  1.1  ad 			error = ENOMEM;
    326  1.1  ad 			goto out;
    327  1.1  ad 		}
    328  1.1  ad 	}
    329  1.1  ad 	if (symtabindex == -1) {
    330  1.1  ad 		kobj_error("lost symbol table index");
    331  1.1  ad 		goto out;
    332  1.1  ad 	}
    333  1.1  ad 
    334  1.1  ad 	/*
    335  1.1  ad 	 * Allocate space for and load the symbol table.
    336  1.1  ad 	 */
    337  1.1  ad 	ko->ko_symcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym);
    338  1.1  ad 	if (ko->ko_symcnt == 0) {
    339  1.1  ad 		kobj_error("no symbol table");
    340  1.1  ad 		goto out;
    341  1.1  ad 	}
    342  1.1  ad 	ko->ko_symtab = kmem_alloc(ko->ko_symcnt * sizeof(Elf_Sym), KM_SLEEP);
    343  1.1  ad 	if (ko->ko_symtab == NULL) {
    344  1.1  ad 		error = ENOMEM;
    345  1.1  ad 		goto out;
    346  1.1  ad 	}
    347  1.1  ad 	error = kobj_read(nd.ni_vp, ko->ko_symtab, shdr[symtabindex].sh_size,
    348  1.1  ad 	    shdr[symtabindex].sh_offset);
    349  1.1  ad 	if (error != 0) {
    350  1.1  ad 		goto out;
    351  1.1  ad 	}
    352  1.1  ad 
    353  1.1  ad 	/*
    354  1.1  ad 	 * Allocate space for and load the symbol strings.
    355  1.1  ad 	 */
    356  1.1  ad 	ko->ko_strtabsz = shdr[symstrindex].sh_size;
    357  1.1  ad 	if (ko->ko_strtabsz == 0) {
    358  1.1  ad 		kobj_error("no symbol strings");
    359  1.1  ad 		goto out;
    360  1.1  ad 	}
    361  1.1  ad 	ko->ko_strtab = kmem_alloc(ko->ko_strtabsz, KM_SLEEP);
    362  1.1  ad 	if (ko->ko_strtab == NULL) {
    363  1.1  ad 		error = ENOMEM;
    364  1.1  ad 		goto out;
    365  1.1  ad 	}
    366  1.1  ad 	error = kobj_read(nd.ni_vp, ko->ko_strtab, shdr[symstrindex].sh_size,
    367  1.1  ad 	    shdr[symstrindex].sh_offset);
    368  1.1  ad 	if (error != 0) {
    369  1.1  ad 		goto out;
    370  1.1  ad 	}
    371  1.1  ad 
    372  1.1  ad 	/*
    373  1.1  ad 	 * Size up code/data(progbits) and bss(nobits).
    374  1.1  ad 	 */
    375  1.1  ad 	alignmask = 0;
    376  1.1  ad 	for (i = 0; i < hdr->e_shnum; i++) {
    377  1.1  ad 		switch (shdr[i].sh_type) {
    378  1.1  ad 		case SHT_PROGBITS:
    379  1.1  ad 		case SHT_NOBITS:
    380  1.1  ad 			alignmask = shdr[i].sh_addralign - 1;
    381  1.1  ad 			mapsize += alignmask;
    382  1.1  ad 			mapsize &= ~alignmask;
    383  1.1  ad 			mapsize += shdr[i].sh_size;
    384  1.1  ad 			break;
    385  1.1  ad 		}
    386  1.1  ad 	}
    387  1.1  ad 
    388  1.1  ad 	/*
    389  1.1  ad 	 * We know how much space we need for the text/data/bss/etc.
    390  1.1  ad 	 * This stuff needs to be in a single chunk so that profiling etc
    391  1.1  ad 	 * can get the bounds and gdb can associate offsets with modules.
    392  1.1  ad 	 */
    393  1.1  ad 	if (mapsize == 0) {
    394  1.1  ad 		kobj_error("no text/data/bss");
    395  1.1  ad 		goto out;
    396  1.1  ad 	}
    397  1.1  ad 	mapbase = uvm_km_alloc(lkm_map, round_page(mapsize), 0,
    398  1.1  ad 	    UVM_KMF_WIRED | UVM_KMF_EXEC);
    399  1.1  ad 	if (mapbase == 0) {
    400  1.1  ad 		error = ENOMEM;
    401  1.1  ad 		goto out;
    402  1.1  ad 	}
    403  1.1  ad 	ko->ko_address = mapbase;
    404  1.1  ad 	ko->ko_size = mapsize;
    405  1.1  ad 
    406  1.1  ad 	/*
    407  1.1  ad 	 * Now load code/data(progbits), zero bss(nobits), allocate space
    408  1.1  ad 	 * for and load relocs
    409  1.1  ad 	 */
    410  1.1  ad 	pb = 0;
    411  1.1  ad 	rl = 0;
    412  1.1  ad 	ra = 0;
    413  1.1  ad 	alignmask = 0;
    414  1.1  ad 	for (i = 0; i < hdr->e_shnum; i++) {
    415  1.1  ad 		switch (shdr[i].sh_type) {
    416  1.1  ad 		case SHT_PROGBITS:
    417  1.1  ad 		case SHT_NOBITS:
    418  1.1  ad 			alignmask = shdr[i].sh_addralign - 1;
    419  1.1  ad 			mapbase += alignmask;
    420  1.1  ad 			mapbase &= ~alignmask;
    421  1.1  ad 			ko->ko_progtab[pb].addr = (void *)mapbase;
    422  1.1  ad 			if (shdr[i].sh_type == SHT_PROGBITS) {
    423  1.1  ad 				ko->ko_progtab[pb].name = "<<PROGBITS>>";
    424  1.1  ad 				error = kobj_read(nd.ni_vp,
    425  1.1  ad 				    ko->ko_progtab[pb].addr, shdr[i].sh_size,
    426  1.1  ad 				    shdr[i].sh_offset);
    427  1.1  ad 				if (error != 0) {
    428  1.1  ad 					goto out;
    429  1.1  ad 				}
    430  1.1  ad 			} else {
    431  1.1  ad 				ko->ko_progtab[pb].name = "<<NOBITS>>";
    432  1.1  ad 				memset(ko->ko_progtab[pb].addr, 0,
    433  1.1  ad 				    shdr[i].sh_size);
    434  1.1  ad 			}
    435  1.1  ad 			ko->ko_progtab[pb].size = shdr[i].sh_size;
    436  1.1  ad 			ko->ko_progtab[pb].sec = i;
    437  1.1  ad 
    438  1.1  ad 			/* Update all symbol values with the offset. */
    439  1.1  ad 			for (j = 0; j < ko->ko_symcnt; j++) {
    440  1.1  ad 				es = &ko->ko_symtab[j];
    441  1.1  ad 				if (es->st_shndx != i) {
    442  1.1  ad 					continue;
    443  1.1  ad 				}
    444  1.1  ad 				es->st_value +=
    445  1.1  ad 				    (Elf_Addr)ko->ko_progtab[pb].addr;
    446  1.1  ad 			}
    447  1.1  ad 			mapbase += shdr[i].sh_size;
    448  1.1  ad 			pb++;
    449  1.1  ad 			break;
    450  1.1  ad 		case SHT_REL:
    451  1.1  ad 			ko->ko_reltab[rl].size = shdr[i].sh_size;
    452  1.1  ad 			ko->ko_reltab[rl].size -=
    453  1.1  ad 			    shdr[i].sh_size % sizeof(Elf_Rel);
    454  1.1  ad 			if (ko->ko_reltab[rl].size != 0) {
    455  1.1  ad 				ko->ko_reltab[rl].rel =
    456  1.1  ad 				    kmem_alloc(ko->ko_reltab[rl].size,
    457  1.1  ad 				    KM_SLEEP);
    458  1.1  ad 				ko->ko_reltab[rl].nrel =
    459  1.1  ad 				    shdr[i].sh_size / sizeof(Elf_Rel);
    460  1.1  ad 				ko->ko_reltab[rl].sec = shdr[i].sh_info;
    461  1.1  ad 				error = kobj_read(nd.ni_vp,
    462  1.1  ad 				    ko->ko_reltab[rl].rel,
    463  1.1  ad 				    ko->ko_reltab[rl].size,
    464  1.1  ad 				    shdr[i].sh_offset);
    465  1.1  ad 				if (error != 0) {
    466  1.1  ad 					goto out;
    467  1.1  ad 				}
    468  1.1  ad 			}
    469  1.1  ad 			rl++;
    470  1.1  ad 			break;
    471  1.1  ad 		case SHT_RELA:
    472  1.1  ad 			ko->ko_relatab[ra].size = shdr[i].sh_size;
    473  1.1  ad 			ko->ko_relatab[ra].size -=
    474  1.1  ad 			    shdr[i].sh_size % sizeof(Elf_Rela);
    475  1.1  ad 			if (ko->ko_relatab[ra].size != 0) {
    476  1.1  ad 				ko->ko_relatab[ra].rela =
    477  1.1  ad 				    kmem_alloc(ko->ko_relatab[ra].size,
    478  1.1  ad 				    KM_SLEEP);
    479  1.1  ad 				ko->ko_relatab[ra].nrela =
    480  1.1  ad 				    shdr[i].sh_size / sizeof(Elf_Rela);
    481  1.1  ad 				ko->ko_relatab[ra].sec = shdr[i].sh_info;
    482  1.1  ad 				error = kobj_read(nd.ni_vp,
    483  1.1  ad 				    ko->ko_relatab[ra].rela,
    484  1.1  ad 				    shdr[i].sh_size,
    485  1.1  ad 				    shdr[i].sh_offset);
    486  1.1  ad 				if (error != 0) {
    487  1.1  ad 					goto out;
    488  1.1  ad 				}
    489  1.1  ad 			}
    490  1.1  ad 			ra++;
    491  1.1  ad 			break;
    492  1.1  ad 		}
    493  1.1  ad 	}
    494  1.1  ad 	if (pb != ko->ko_nprogtab) {
    495  1.1  ad 		panic("lost progbits");
    496  1.1  ad 	}
    497  1.1  ad 	if (rl != ko->ko_nrel) {
    498  1.1  ad 		panic("lost rel");
    499  1.1  ad 	}
    500  1.1  ad 	if (ra != ko->ko_nrela) {
    501  1.1  ad 		panic("lost rela");
    502  1.1  ad 	}
    503  1.1  ad 	if (mapbase != ko->ko_address + mapsize) {
    504  1.1  ad 		panic("mapbase 0x%lx != address %lx + mapsize 0x%lx (0x%lx)\n",
    505  1.1  ad 		    (long)mapbase, (long)ko->ko_address, (long)mapsize,
    506  1.1  ad 		    (long)ko->ko_address + mapsize);
    507  1.1  ad 	}
    508  1.1  ad 
    509  1.1  ad 	/*
    510  1.1  ad 	 * Perform relocations.  Done before registering with ksyms,
    511  1.1  ad 	 * which will pack our symbol table.
    512  1.1  ad 	 */
    513  1.1  ad 	error = kobj_relocate(ko);
    514  1.1  ad 	if (error != 0) {
    515  1.1  ad 		goto out;
    516  1.1  ad 	}
    517  1.1  ad 
    518  1.1  ad 	/*
    519  1.1  ad 	 * Register symbol table with ksyms.
    520  1.1  ad 	 */
    521  1.1  ad 	error = ksyms_addsymtab(ko->ko_name, ko->ko_symtab, ko->ko_symcnt *
    522  1.1  ad 	    sizeof(Elf_Sym), ko->ko_strtab, ko->ko_strtabsz);
    523  1.1  ad 	if (error != 0) {
    524  1.1  ad 		kobj_error("unable to register module symbol table");
    525  1.1  ad 		goto out;
    526  1.1  ad 	}
    527  1.1  ad 	ko->ko_ksyms = true;
    528  1.1  ad 
    529  1.1  ad 	/*
    530  1.1  ad 	 * Notify MD code that a module has been loaded.
    531  1.1  ad 	 */
    532  1.1  ad 	error = kobj_machdep(ko, (void *)ko->ko_address, ko->ko_size, true);
    533  1.1  ad 	if (error != 0) {
    534  1.1  ad 		kobj_error("machine dependent init failed");
    535  1.1  ad 		goto out;
    536  1.1  ad 	}
    537  1.1  ad 	*kop = ko;
    538  1.1  ad  out:
    539  1.1  ad 	kobj_release_mem(ko);
    540  1.1  ad 	if (error != 0 && ko)
    541  1.1  ad 		kobj_unload(ko);
    542  1.1  ad 	if (hdr != NULL)
    543  1.1  ad 		kmem_free(hdr, sizeof(*hdr));
    544  1.1  ad 	if (nd.ni_vp != NULL) {
    545  1.1  ad 		VOP_UNLOCK(nd.ni_vp, 0);
    546  1.1  ad 		vn_close(nd.ni_vp, FREAD, cred, curlwp);
    547  1.1  ad 	}
    548  1.1  ad 
    549  1.1  ad 	return error;
    550  1.1  ad }
    551  1.1  ad 
    552  1.1  ad /*
    553  1.1  ad  * kobj_unload:
    554  1.1  ad  *
    555  1.1  ad  *	Unload an object previously loaded by kobj_load().
    556  1.1  ad  */
    557  1.1  ad void
    558  1.1  ad kobj_unload(kobj_t ko)
    559  1.1  ad {
    560  1.1  ad 	int error;
    561  1.1  ad 
    562  1.1  ad 	if (ko->ko_address != 0) {
    563  1.1  ad 		uvm_km_free(lkm_map, ko->ko_address, round_page(ko->ko_size),
    564  1.1  ad 		    UVM_KMF_WIRED);
    565  1.1  ad 	}
    566  1.1  ad 	if (ko->ko_ksyms == true) {
    567  1.1  ad 		ksyms_delsymtab(ko->ko_name);
    568  1.1  ad 	}
    569  1.1  ad 	if (ko->ko_symtab != NULL) {
    570  1.1  ad 		kmem_free(ko->ko_symtab, ko->ko_symcnt * sizeof(Elf_Sym));
    571  1.1  ad 	}
    572  1.1  ad 	if (ko->ko_strtab != NULL) {
    573  1.1  ad 		kmem_free(ko->ko_strtab, ko->ko_strtabsz);
    574  1.1  ad 	}
    575  1.1  ad 	kmem_free(ko, sizeof(*ko));
    576  1.1  ad 
    577  1.1  ad 	/*
    578  1.1  ad 	 * Notify MD code that a module has been unloaded.
    579  1.1  ad 	 */
    580  1.1  ad 	error = kobj_machdep(ko, (void *)ko->ko_address, ko->ko_size, false);
    581  1.1  ad 	if (error != 0) {
    582  1.1  ad 		kobj_error("machine dependent deinit failed");
    583  1.1  ad 	}
    584  1.1  ad }
    585  1.1  ad 
    586  1.1  ad /*
    587  1.1  ad  * kobj_release_mem:
    588  1.1  ad  *
    589  1.1  ad  *	Release object data not needed after loading.
    590  1.1  ad  */
    591  1.1  ad static void
    592  1.1  ad kobj_release_mem(kobj_t ko)
    593  1.1  ad {
    594  1.1  ad 	int i;
    595  1.1  ad 
    596  1.1  ad 	for (i = 0; i < ko->ko_nrel; i++) {
    597  1.1  ad 		if (ko->ko_reltab[i].rel) {
    598  1.1  ad 			kmem_free(ko->ko_reltab[i].rel,
    599  1.1  ad 			    ko->ko_reltab[i].size);
    600  1.1  ad 		}
    601  1.1  ad 	}
    602  1.1  ad 	for (i = 0; i < ko->ko_nrela; i++) {
    603  1.1  ad 		if (ko->ko_relatab[i].rela) {
    604  1.1  ad 			kmem_free(ko->ko_relatab[i].rela,
    605  1.1  ad 			    ko->ko_relatab[i].size);
    606  1.1  ad 		}
    607  1.1  ad 	}
    608  1.1  ad 	if (ko->ko_reltab != NULL) {
    609  1.1  ad 		kmem_free(ko->ko_reltab, ko->ko_nrel *
    610  1.1  ad 		    sizeof(*ko->ko_reltab));
    611  1.1  ad 		ko->ko_reltab = NULL;
    612  1.1  ad 		ko->ko_nrel = 0;
    613  1.1  ad 	}
    614  1.1  ad 	if (ko->ko_relatab != NULL) {
    615  1.1  ad 		kmem_free(ko->ko_relatab, ko->ko_nrela *
    616  1.1  ad 		    sizeof(*ko->ko_relatab));
    617  1.1  ad 		ko->ko_relatab = NULL;
    618  1.1  ad 		ko->ko_nrela = 0;
    619  1.1  ad 	}
    620  1.1  ad 	if (ko->ko_progtab != NULL) {
    621  1.1  ad 		kmem_free(ko->ko_progtab, ko->ko_nprogtab *
    622  1.1  ad 		    sizeof(*ko->ko_progtab));
    623  1.1  ad 		ko->ko_progtab = NULL;
    624  1.1  ad 	}
    625  1.1  ad 	if (ko->ko_shdr != NULL) {
    626  1.1  ad 		kmem_free(ko->ko_shdr, ko->ko_shdrsz);
    627  1.1  ad 		ko->ko_shdr = NULL;
    628  1.1  ad 	}
    629  1.1  ad }
    630  1.1  ad 
    631  1.1  ad /*
    632  1.1  ad  * kobj_sym_lookup:
    633  1.1  ad  *
    634  1.1  ad  *	Symbol lookup function to be used when the symbol index
    635  1.1  ad  *	is known (ie during relocation).
    636  1.1  ad  */
    637  1.1  ad uintptr_t
    638  1.1  ad kobj_sym_lookup(kobj_t ko, uintptr_t symidx)
    639  1.1  ad {
    640  1.1  ad 	const Elf_Sym *sym;
    641  1.1  ad 	const char *symbol;
    642  1.1  ad 	int error;
    643  1.1  ad 	u_long addr;
    644  1.1  ad 
    645  1.1  ad 	/* Don't even try to lookup the symbol if the index is bogus. */
    646  1.1  ad 	if (symidx >= ko->ko_symcnt)
    647  1.1  ad 		return 0;
    648  1.1  ad 
    649  1.1  ad 	sym = ko->ko_symtab + symidx;
    650  1.1  ad 
    651  1.1  ad 	/* Quick answer if there is a definition included. */
    652  1.1  ad 	if (sym->st_shndx != SHN_UNDEF) {
    653  1.1  ad 		return sym->st_value;
    654  1.1  ad 	}
    655  1.1  ad 
    656  1.1  ad 	/* If we get here, then it is undefined and needs a lookup. */
    657  1.1  ad 	switch (ELF_ST_BIND(sym->st_info)) {
    658  1.1  ad 	case STB_LOCAL:
    659  1.1  ad 		/* Local, but undefined? huh? */
    660  1.1  ad 		kobj_error("local symbol undefined");
    661  1.1  ad 		return 0;
    662  1.1  ad 
    663  1.1  ad 	case STB_GLOBAL:
    664  1.1  ad 		/* Relative to Data or Function name */
    665  1.1  ad 		symbol = ko->ko_strtab + sym->st_name;
    666  1.1  ad 
    667  1.1  ad 		/* Force a lookup failure if the symbol name is bogus. */
    668  1.1  ad 		if (*symbol == 0) {
    669  1.1  ad 			kobj_error("bad symbol name");
    670  1.1  ad 			return 0;
    671  1.1  ad 		}
    672  1.1  ad 
    673  1.1  ad 		error = ksyms_getval(NULL, symbol, &addr, KSYMS_ANY);
    674  1.1  ad 		if (error != 0) {
    675  1.1  ad 			kobj_error("symbol %s undefined", symbol);
    676  1.1  ad 			return (uintptr_t)0;
    677  1.1  ad 		}
    678  1.1  ad 		return (uintptr_t)addr;
    679  1.1  ad 
    680  1.1  ad 	case STB_WEAK:
    681  1.1  ad 		kobj_error("weak symbols not supported\n");
    682  1.1  ad 		return 0;
    683  1.1  ad 
    684  1.1  ad 	default:
    685  1.1  ad 		return 0;
    686  1.1  ad 	}
    687  1.1  ad }
    688  1.1  ad 
    689  1.1  ad /*
    690  1.1  ad  * kobj_findbase:
    691  1.1  ad  *
    692  1.1  ad  *	Return base address of the given section.
    693  1.1  ad  */
    694  1.1  ad static uintptr_t
    695  1.1  ad kobj_findbase(kobj_t ko, int sec)
    696  1.1  ad {
    697  1.1  ad 	int i;
    698  1.1  ad 
    699  1.1  ad 	for (i = 0; i < ko->ko_nprogtab; i++) {
    700  1.1  ad 		if (sec == ko->ko_progtab[i].sec) {
    701  1.1  ad 			return (uintptr_t)ko->ko_progtab[i].addr;
    702  1.1  ad 		}
    703  1.1  ad 	}
    704  1.1  ad 	return 0;
    705  1.1  ad }
    706  1.1  ad 
    707  1.1  ad /*
    708  1.1  ad  * kobj_relocate:
    709  1.1  ad  *
    710  1.1  ad  *	Resolve all relocations for the loaded object.
    711  1.1  ad  */
    712  1.1  ad static int
    713  1.1  ad kobj_relocate(kobj_t ko)
    714  1.1  ad {
    715  1.1  ad 	const Elf_Rel *rellim;
    716  1.1  ad 	const Elf_Rel *rel;
    717  1.1  ad 	const Elf_Rela *relalim;
    718  1.1  ad 	const Elf_Rela *rela;
    719  1.1  ad 	const Elf_Sym *sym;
    720  1.1  ad 	uintptr_t base;
    721  1.1  ad 	int i;
    722  1.1  ad 	uintptr_t symidx;
    723  1.1  ad 
    724  1.1  ad 	/*
    725  1.1  ad 	 * Perform relocations without addend if there are any.
    726  1.1  ad 	 */
    727  1.1  ad 	for (i = 0; i < ko->ko_nrel; i++) {
    728  1.1  ad 		rel = ko->ko_reltab[i].rel;
    729  1.1  ad 		if (rel == NULL) {
    730  1.1  ad 			continue;
    731  1.1  ad 		}
    732  1.1  ad 		rellim = rel + ko->ko_reltab[i].nrel;
    733  1.1  ad 		base = kobj_findbase(ko, ko->ko_reltab[i].sec);
    734  1.1  ad 		if (base == 0) {
    735  1.1  ad 			panic("lost base for e_reltab");
    736  1.1  ad 		}
    737  1.1  ad 		for (; rel < rellim; rel++) {
    738  1.1  ad 			symidx = ELF_R_SYM(rel->r_info);
    739  1.1  ad 			if (symidx >= ko->ko_symcnt) {
    740  1.1  ad 				continue;
    741  1.1  ad 			}
    742  1.1  ad 			sym = ko->ko_symtab + symidx;
    743  1.1  ad 			/* Only do local relocs */
    744  1.1  ad 			if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) {
    745  1.1  ad 				kobj_reloc(ko, base, rel, false, true);
    746  1.1  ad 				continue;
    747  1.1  ad 			}
    748  1.1  ad 			if (kobj_reloc(ko, base, rel, false, false)) {
    749  1.1  ad 				return ENOENT;
    750  1.1  ad 			}
    751  1.1  ad 		}
    752  1.1  ad 	}
    753  1.1  ad 
    754  1.1  ad 	/*
    755  1.1  ad 	 * Perform relocations with addend if there are any.
    756  1.1  ad 	 */
    757  1.1  ad 	for (i = 0; i < ko->ko_nrela; i++) {
    758  1.1  ad 		rela = ko->ko_relatab[i].rela;
    759  1.1  ad 		if (rela == NULL) {
    760  1.1  ad 			continue;
    761  1.1  ad 		}
    762  1.1  ad 		relalim = rela + ko->ko_relatab[i].nrela;
    763  1.1  ad 		base = kobj_findbase(ko, ko->ko_relatab[i].sec);
    764  1.1  ad 		if (base == 0) {
    765  1.1  ad 			panic("lost base for e_relatab");
    766  1.1  ad 		}
    767  1.1  ad 		for (; rela < relalim; rela++) {
    768  1.1  ad 			symidx = ELF_R_SYM(rela->r_info);
    769  1.1  ad 			if (symidx >= ko->ko_symcnt) {
    770  1.1  ad 				continue;
    771  1.1  ad 			}
    772  1.1  ad 			sym = ko->ko_symtab + symidx;
    773  1.1  ad 			/* Only do local relocs */
    774  1.1  ad 			if (ELF_ST_BIND(sym->st_info) == STB_LOCAL) {
    775  1.1  ad 				kobj_reloc(ko, base, rela, true, true);
    776  1.1  ad 				continue;
    777  1.1  ad 			}
    778  1.1  ad 			if (kobj_reloc(ko, base, rela, true, false)) {
    779  1.1  ad 				return ENOENT;
    780  1.1  ad 			}
    781  1.1  ad 		}
    782  1.1  ad 	}
    783  1.1  ad 
    784  1.1  ad 	return 0;
    785  1.1  ad }
    786  1.1  ad 
    787  1.1  ad /*
    788  1.1  ad  * kobj_error:
    789  1.1  ad  *
    790  1.1  ad  *	Utility function: log an error.
    791  1.1  ad  */
    792  1.1  ad static void
    793  1.1  ad kobj_error(const char *fmt, ...)
    794  1.1  ad {
    795  1.1  ad 	va_list ap;
    796  1.1  ad 
    797  1.1  ad 	va_start(ap, fmt);
    798  1.1  ad 	printf("WARNING: linker error: ");
    799  1.1  ad 	vprintf(fmt, ap);
    800  1.1  ad 	printf("\n");
    801  1.1  ad 	va_end(ap);
    802  1.1  ad }
    803  1.1  ad 
    804  1.1  ad /*
    805  1.1  ad  * kobj_read:
    806  1.1  ad  *
    807  1.1  ad  *	Utility function: read from the object.
    808  1.1  ad  */
    809  1.1  ad static int
    810  1.1  ad kobj_read(vnode_t *vp, void *base, size_t size, off_t off)
    811  1.1  ad {
    812  1.1  ad 	size_t resid;
    813  1.1  ad 	int error;
    814  1.1  ad 
    815  1.1  ad 	error = vn_rdwr(UIO_READ, vp, base, size, off, UIO_SYSSPACE,
    816  1.1  ad 	    IO_NODELOCKED, curlwp->l_cred, &resid, curlwp);
    817  1.1  ad 	if (error == 0 && resid != 0)
    818  1.1  ad 		error = EINVAL;
    819  1.1  ad 	return error;
    820  1.1  ad }
    821