Home | History | Annotate | Line # | Download | only in kern
subr_kobj.c revision 1.22
      1  1.22        ad /*	$NetBSD: subr_kobj.c,v 1.22 2008/10/06 10:46:58 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  *
     16   1.1        ad  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
     17   1.1        ad  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
     18   1.1        ad  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
     19   1.1        ad  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
     20   1.1        ad  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
     21   1.1        ad  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
     22   1.1        ad  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
     23   1.1        ad  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
     24   1.1        ad  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
     25   1.1        ad  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
     26   1.1        ad  * POSSIBILITY OF SUCH DAMAGE.
     27   1.1        ad  */
     28   1.1        ad 
     29   1.1        ad /*-
     30   1.1        ad  * Copyright (c) 1998-2000 Doug Rabson
     31   1.1        ad  * Copyright (c) 2004 Peter Wemm
     32   1.1        ad  * All rights reserved.
     33   1.1        ad  *
     34   1.1        ad  * Redistribution and use in source and binary forms, with or without
     35   1.1        ad  * modification, are permitted provided that the following conditions
     36   1.1        ad  * are met:
     37   1.1        ad  * 1. Redistributions of source code must retain the above copyright
     38   1.1        ad  *    notice, this list of conditions and the following disclaimer.
     39   1.1        ad  * 2. Redistributions in binary form must reproduce the above copyright
     40   1.1        ad  *    notice, this list of conditions and the following disclaimer in the
     41   1.1        ad  *    documentation and/or other materials provided with the distribution.
     42   1.1        ad  *
     43   1.1        ad  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
     44   1.1        ad  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     45   1.1        ad  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     46   1.1        ad  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
     47   1.1        ad  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     48   1.1        ad  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     49   1.1        ad  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     50   1.1        ad  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     51   1.1        ad  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     52   1.1        ad  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     53   1.1        ad  * SUCH DAMAGE.
     54   1.1        ad  */
     55   1.1        ad 
     56   1.1        ad /*
     57   1.1        ad  * Kernel loader for ELF objects.
     58   1.1        ad  *
     59   1.1        ad  * TODO: adjust kmem_alloc() calls to avoid needless fragmentation.
     60   1.1        ad  */
     61   1.1        ad 
     62   1.5        ad #include "opt_modular.h"
     63   1.5        ad 
     64   1.1        ad #include <sys/cdefs.h>
     65  1.22        ad __KERNEL_RCSID(0, "$NetBSD: subr_kobj.c,v 1.22 2008/10/06 10:46:58 ad Exp $");
     66   1.1        ad 
     67   1.1        ad #define	ELFSIZE		ARCH_ELFSIZE
     68   1.1        ad 
     69  1.16        ad #include <sys/systm.h>
     70  1.16        ad #include <sys/kobj.h>
     71  1.16        ad #include <sys/errno.h>
     72  1.16        ad 
     73  1.16        ad #ifdef MODULAR
     74  1.16        ad 
     75   1.1        ad #include <sys/param.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/ksyms.h>
     83   1.1        ad #include <sys/lkm.h>
     84   1.1        ad #include <sys/exec.h>
     85   1.1        ad #include <sys/exec_elf.h>
     86   1.1        ad 
     87   1.1        ad #include <machine/stdarg.h>
     88   1.1        ad 
     89   1.1        ad #include <uvm/uvm_extern.h>
     90   1.1        ad 
     91   1.5        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.3        ad typedef enum kobjtype {
    115   1.3        ad 	KT_UNSET,
    116   1.3        ad 	KT_VNODE,
    117   1.3        ad 	KT_MEMORY
    118   1.3        ad } kobjtype_t;
    119   1.3        ad 
    120   1.1        ad struct kobj {
    121   1.1        ad 	char		ko_name[MAXLKMNAME];
    122   1.3        ad 	kobjtype_t	ko_type;
    123   1.3        ad 	void		*ko_source;
    124   1.3        ad 	ssize_t		ko_memsize;
    125   1.1        ad 	vaddr_t		ko_address;	/* Relocation address */
    126   1.1        ad 	Elf_Shdr	*ko_shdr;
    127   1.1        ad 	progent_t	*ko_progtab;
    128   1.1        ad 	relaent_t	*ko_relatab;
    129   1.1        ad 	relent_t	*ko_reltab;
    130   1.1        ad 	Elf_Sym		*ko_symtab;	/* Symbol table */
    131   1.1        ad 	char		*ko_strtab;	/* String table */
    132   1.8        ad 	char		*ko_shstrtab;	/* Section name string table */
    133   1.1        ad 	size_t		ko_size;	/* Size of text/data/bss */
    134   1.1        ad 	size_t		ko_symcnt;	/* Number of symbols */
    135   1.1        ad 	size_t		ko_strtabsz;	/* Number of bytes in string table */
    136   1.8        ad 	size_t		ko_shstrtabsz;	/* Number of bytes in scn str table */
    137   1.1        ad 	size_t		ko_shdrsz;
    138   1.1        ad 	int		ko_nrel;
    139   1.1        ad 	int		ko_nrela;
    140   1.1        ad 	int		ko_nprogtab;
    141   1.1        ad 	bool		ko_ksyms;
    142   1.3        ad 	bool		ko_loaded;
    143   1.1        ad };
    144   1.1        ad 
    145  1.18        ad static int	kobj_relocate(kobj_t, bool);
    146   1.1        ad static void	kobj_error(const char *, ...);
    147  1.12        ad static int	kobj_read(kobj_t, void **, size_t, off_t);
    148  1.12        ad static int	kobj_read_bits(kobj_t, void *, size_t, off_t);
    149  1.18        ad static void	kobj_jettison(kobj_t);
    150  1.12        ad static void	kobj_free(kobj_t, void *, size_t);
    151  1.18        ad static void	kobj_close(kobj_t);
    152  1.18        ad static int	kobj_load(kobj_t);
    153   1.1        ad 
    154   1.1        ad extern struct vm_map *lkm_map;
    155   1.1        ad 
    156   1.1        ad /*
    157  1.18        ad  * kobj_load_file:
    158   1.1        ad  *
    159  1.18        ad  *	Load an object located in the file system.
    160   1.1        ad  */
    161   1.1        ad int
    162  1.21        ad kobj_load_file(kobj_t *kop, const char *filename, const char *base,
    163  1.21        ad 	       bool autoload)
    164   1.1        ad {
    165   1.1        ad 	struct nameidata nd;
    166   1.1        ad 	kauth_cred_t cred;
    167   1.1        ad 	char *path;
    168   1.1        ad 	int error;
    169   1.1        ad 	kobj_t ko;
    170   1.1        ad 
    171   1.1        ad 	cred = kauth_cred_get();
    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.21        ad 	if (autoload) {
    179  1.19        ad 		error = ENOENT;
    180  1.19        ad 	} else {
    181  1.19        ad 		NDINIT(&nd, LOOKUP, FOLLOW, UIO_SYSSPACE, filename);
    182  1.19        ad 		error = vn_open(&nd, FREAD, 0);
    183  1.19        ad 	}
    184   1.1        ad 	if (error != 0) {
    185   1.2        ad 		if (error != ENOENT) {
    186   1.2        ad 			goto out;
    187   1.2        ad 		}
    188   1.2        ad 		path = PNBUF_GET();
    189  1.19        ad 		snprintf(path, MAXPATHLEN - 1, "%s/%s/%s.kmod", base,
    190  1.19        ad 		    filename, filename);
    191   1.2        ad 		NDINIT(&nd, LOOKUP, FOLLOW, UIO_SYSSPACE, path);
    192   1.2        ad 		error = vn_open(&nd, FREAD, 0);
    193   1.9    rumble 		if (error != 0) {
    194   1.9    rumble 			strlcat(path, ".o", MAXPATHLEN);
    195   1.9    rumble 			NDINIT(&nd, LOOKUP, FOLLOW, UIO_SYSSPACE, path);
    196   1.9    rumble 			error = vn_open(&nd, FREAD, 0);
    197   1.9    rumble 		}
    198   1.2        ad 		PNBUF_PUT(path);
    199   1.2        ad 		if (error != 0) {
    200   1.2        ad 			goto out;
    201   1.2        ad 		}
    202   1.1        ad 	}
    203   1.1        ad 
    204   1.3        ad  out:
    205   1.3        ad  	if (error != 0) {
    206   1.3        ad 	 	kmem_free(ko, sizeof(*ko));
    207  1.18        ad 	 	return error;
    208   1.3        ad 	}
    209  1.18        ad 
    210  1.18        ad 	ko->ko_type = KT_VNODE;
    211  1.18        ad 	ko->ko_source = nd.ni_vp;
    212  1.18        ad 	*kop = ko;
    213  1.18        ad 	return kobj_load(ko);
    214   1.3        ad }
    215   1.3        ad 
    216   1.3        ad /*
    217  1.18        ad  * kobj_load_mem:
    218   1.3        ad  *
    219  1.18        ad  *	Load an object already resident in memory.  If size is not -1,
    220  1.18        ad  *	the complete size of the object is known.
    221   1.3        ad  */
    222   1.3        ad int
    223  1.18        ad kobj_load_mem(kobj_t *kop, void *base, ssize_t size)
    224   1.3        ad {
    225   1.3        ad 	kobj_t ko;
    226   1.3        ad 
    227   1.3        ad 	ko = kmem_zalloc(sizeof(*ko), KM_SLEEP);
    228   1.3        ad 	if (ko == NULL) {
    229   1.3        ad 		return ENOMEM;
    230   1.3        ad 	}
    231   1.3        ad 
    232   1.3        ad 	ko->ko_type = KT_MEMORY;
    233   1.3        ad 	ko->ko_source = base;
    234   1.3        ad 	ko->ko_memsize = size;
    235   1.3        ad 	*kop = ko;
    236  1.18        ad 	return kobj_load(ko);
    237   1.3        ad }
    238   1.3        ad 
    239   1.3        ad /*
    240   1.3        ad  * kobj_close:
    241   1.3        ad  *
    242  1.18        ad  *	Close an open ELF object.
    243   1.3        ad  */
    244  1.18        ad static void
    245   1.3        ad kobj_close(kobj_t ko)
    246   1.3        ad {
    247   1.3        ad 
    248  1.18        ad 	if (ko->ko_source == NULL) {
    249  1.18        ad 		return;
    250  1.18        ad 	}
    251   1.3        ad 
    252   1.3        ad 	switch (ko->ko_type) {
    253   1.3        ad 	case KT_VNODE:
    254   1.3        ad 		VOP_UNLOCK(ko->ko_source, 0);
    255  1.10        ad 		vn_close(ko->ko_source, FREAD, kauth_cred_get());
    256   1.3        ad 		break;
    257   1.3        ad 	case KT_MEMORY:
    258   1.3        ad 		/* nothing */
    259   1.3        ad 		break;
    260   1.3        ad 	default:
    261   1.3        ad 		panic("kobj_close: unknown type");
    262   1.3        ad 		break;
    263   1.3        ad 	}
    264   1.3        ad 
    265   1.3        ad 	ko->ko_source = NULL;
    266   1.3        ad }
    267   1.3        ad 
    268   1.3        ad /*
    269   1.3        ad  * kobj_load:
    270   1.3        ad  *
    271  1.18        ad  *	Load an ELF object and prepare to link into the running kernel
    272  1.18        ad  *	image.
    273   1.3        ad  */
    274  1.18        ad static int
    275   1.3        ad kobj_load(kobj_t ko)
    276   1.3        ad {
    277   1.3        ad 	Elf_Ehdr *hdr;
    278   1.3        ad 	Elf_Shdr *shdr;
    279   1.3        ad 	Elf_Sym *es;
    280   1.3        ad 	vaddr_t mapbase;
    281   1.3        ad 	size_t mapsize;
    282   1.3        ad 	int error;
    283   1.3        ad 	int symtabindex;
    284   1.3        ad 	int symstrindex;
    285   1.3        ad 	int nsym;
    286   1.3        ad 	int pb, rl, ra;
    287   1.3        ad 	int alignmask;
    288   1.3        ad 	int i, j;
    289  1.13        ad 	void *addr;
    290   1.3        ad 
    291   1.3        ad 	KASSERT(ko->ko_type != KT_UNSET);
    292   1.3        ad 	KASSERT(ko->ko_source != NULL);
    293   1.3        ad 
    294   1.3        ad 	shdr = NULL;
    295   1.3        ad 	mapsize = 0;
    296   1.3        ad 	error = 0;
    297   1.3        ad 	hdr = NULL;
    298   1.3        ad 
    299   1.1        ad 	/*
    300   1.1        ad 	 * Read the elf header from the file.
    301   1.1        ad 	 */
    302  1.12        ad 	error = kobj_read(ko, (void **)&hdr, sizeof(*hdr), 0);
    303   1.1        ad 	if (error != 0)
    304   1.1        ad 		goto out;
    305   1.1        ad 	if (memcmp(hdr->e_ident, ELFMAG, SELFMAG) != 0) {
    306   1.3        ad 		kobj_error("not an ELF object");
    307   1.1        ad 		error = ENOEXEC;
    308   1.1        ad 		goto out;
    309   1.1        ad 	}
    310   1.1        ad 
    311   1.1        ad 	if (hdr->e_ident[EI_VERSION] != EV_CURRENT ||
    312   1.1        ad 	    hdr->e_version != EV_CURRENT) {
    313   1.1        ad 		kobj_error("unsupported file version");
    314   1.1        ad 		error = ENOEXEC;
    315   1.1        ad 		goto out;
    316   1.1        ad 	}
    317   1.1        ad 	if (hdr->e_type != ET_REL) {
    318   1.1        ad 		kobj_error("unsupported file type");
    319   1.1        ad 		error = ENOEXEC;
    320   1.1        ad 		goto out;
    321   1.1        ad 	}
    322   1.1        ad 	switch (hdr->e_machine) {
    323   1.1        ad #if ELFSIZE == 32
    324   1.1        ad 	ELF32_MACHDEP_ID_CASES
    325   1.1        ad #else
    326   1.1        ad 	ELF64_MACHDEP_ID_CASES
    327   1.1        ad #endif
    328   1.1        ad 	default:
    329   1.1        ad 		kobj_error("unsupported machine");
    330   1.1        ad 		error = ENOEXEC;
    331   1.1        ad 		goto out;
    332   1.1        ad 	}
    333   1.1        ad 
    334   1.1        ad 	ko->ko_nprogtab = 0;
    335   1.1        ad 	ko->ko_shdr = 0;
    336   1.1        ad 	ko->ko_nrel = 0;
    337   1.1        ad 	ko->ko_nrela = 0;
    338   1.1        ad 
    339   1.1        ad 	/*
    340   1.1        ad 	 * Allocate and read in the section header.
    341   1.1        ad 	 */
    342   1.1        ad 	ko->ko_shdrsz = hdr->e_shnum * hdr->e_shentsize;
    343   1.1        ad 	if (ko->ko_shdrsz == 0 || hdr->e_shoff == 0 ||
    344   1.1        ad 	    hdr->e_shentsize != sizeof(Elf_Shdr)) {
    345   1.1        ad 		error = ENOEXEC;
    346   1.1        ad 		goto out;
    347   1.1        ad 	}
    348  1.12        ad 	error = kobj_read(ko, (void **)&shdr, ko->ko_shdrsz, hdr->e_shoff);
    349  1.12        ad 	if (error != 0) {
    350   1.1        ad 		goto out;
    351   1.1        ad 	}
    352   1.1        ad 	ko->ko_shdr = shdr;
    353   1.1        ad 
    354   1.1        ad 	/*
    355   1.1        ad 	 * Scan the section header for information and table sizing.
    356   1.1        ad 	 */
    357   1.1        ad 	nsym = 0;
    358   1.1        ad 	symtabindex = -1;
    359   1.1        ad 	symstrindex = -1;
    360   1.1        ad 	for (i = 0; i < hdr->e_shnum; i++) {
    361   1.1        ad 		switch (shdr[i].sh_type) {
    362   1.1        ad 		case SHT_PROGBITS:
    363   1.1        ad 		case SHT_NOBITS:
    364   1.1        ad 			ko->ko_nprogtab++;
    365   1.1        ad 			break;
    366   1.1        ad 		case SHT_SYMTAB:
    367   1.1        ad 			nsym++;
    368   1.1        ad 			symtabindex = i;
    369   1.1        ad 			symstrindex = shdr[i].sh_link;
    370   1.1        ad 			break;
    371   1.1        ad 		case SHT_REL:
    372   1.1        ad 			ko->ko_nrel++;
    373   1.1        ad 			break;
    374   1.1        ad 		case SHT_RELA:
    375   1.1        ad 			ko->ko_nrela++;
    376   1.1        ad 			break;
    377   1.1        ad 		case SHT_STRTAB:
    378   1.1        ad 			break;
    379   1.1        ad 		}
    380   1.1        ad 	}
    381   1.1        ad 	if (ko->ko_nprogtab == 0) {
    382   1.1        ad 		kobj_error("file has no contents");
    383   1.1        ad 		error = ENOEXEC;
    384   1.1        ad 		goto out;
    385   1.1        ad 	}
    386   1.1        ad 	if (nsym != 1) {
    387   1.1        ad 		/* Only allow one symbol table for now */
    388   1.1        ad 		kobj_error("file has no valid symbol table");
    389   1.1        ad 		error = ENOEXEC;
    390   1.1        ad 		goto out;
    391   1.1        ad 	}
    392   1.1        ad 	if (symstrindex < 0 || symstrindex > hdr->e_shnum ||
    393   1.1        ad 	    shdr[symstrindex].sh_type != SHT_STRTAB) {
    394   1.1        ad 		kobj_error("file has invalid symbol strings");
    395   1.1        ad 		error = ENOEXEC;
    396   1.1        ad 		goto out;
    397   1.1        ad 	}
    398   1.1        ad 
    399   1.1        ad 	/*
    400   1.1        ad 	 * Allocate space for tracking the load chunks.
    401   1.1        ad 	 */
    402   1.1        ad 	if (ko->ko_nprogtab != 0) {
    403   1.1        ad 		ko->ko_progtab = kmem_zalloc(ko->ko_nprogtab *
    404   1.1        ad 		    sizeof(*ko->ko_progtab), KM_SLEEP);
    405   1.1        ad 		if (ko->ko_progtab == NULL) {
    406   1.1        ad 			error = ENOMEM;
    407   1.1        ad 			goto out;
    408   1.1        ad 		}
    409   1.1        ad 	}
    410   1.1        ad 	if (ko->ko_nrel != 0) {
    411   1.1        ad 		ko->ko_reltab = kmem_zalloc(ko->ko_nrel *
    412   1.1        ad 		    sizeof(*ko->ko_reltab), KM_SLEEP);
    413   1.1        ad 		if (ko->ko_reltab == NULL) {
    414   1.1        ad 			error = ENOMEM;
    415   1.1        ad 			goto out;
    416   1.1        ad 		}
    417   1.1        ad 	}
    418   1.1        ad 	if (ko->ko_nrela != 0) {
    419   1.1        ad 		ko->ko_relatab = kmem_zalloc(ko->ko_nrela *
    420   1.1        ad 		    sizeof(*ko->ko_relatab), KM_SLEEP);
    421   1.1        ad 		if (ko->ko_relatab == NULL) {
    422   1.1        ad 			error = ENOMEM;
    423   1.1        ad 			goto out;
    424   1.1        ad 		}
    425   1.1        ad 	}
    426   1.1        ad 	if (symtabindex == -1) {
    427   1.1        ad 		kobj_error("lost symbol table index");
    428   1.1        ad 		goto out;
    429   1.1        ad 	}
    430   1.1        ad 
    431   1.1        ad 	/*
    432   1.1        ad 	 * Allocate space for and load the symbol table.
    433   1.1        ad 	 */
    434   1.1        ad 	ko->ko_symcnt = shdr[symtabindex].sh_size / sizeof(Elf_Sym);
    435   1.1        ad 	if (ko->ko_symcnt == 0) {
    436   1.1        ad 		kobj_error("no symbol table");
    437   1.1        ad 		goto out;
    438   1.1        ad 	}
    439  1.12        ad 	error = kobj_read(ko, (void **)&ko->ko_symtab,
    440  1.12        ad 	    ko->ko_symcnt * sizeof(Elf_Sym),
    441   1.1        ad 	    shdr[symtabindex].sh_offset);
    442   1.1        ad 	if (error != 0) {
    443   1.1        ad 		goto out;
    444   1.1        ad 	}
    445   1.1        ad 
    446   1.1        ad 	/*
    447   1.1        ad 	 * Allocate space for and load the symbol strings.
    448   1.1        ad 	 */
    449   1.1        ad 	ko->ko_strtabsz = shdr[symstrindex].sh_size;
    450   1.1        ad 	if (ko->ko_strtabsz == 0) {
    451   1.1        ad 		kobj_error("no symbol strings");
    452   1.1        ad 		goto out;
    453   1.1        ad 	}
    454  1.12        ad 	error = kobj_read(ko, (void *)&ko->ko_strtab, ko->ko_strtabsz,
    455   1.1        ad 	    shdr[symstrindex].sh_offset);
    456   1.1        ad 	if (error != 0) {
    457   1.1        ad 		goto out;
    458   1.1        ad 	}
    459   1.1        ad 
    460   1.1        ad 	/*
    461   1.8        ad 	 * Do we have a string table for the section names?
    462   1.8        ad 	 */
    463   1.8        ad 	if (hdr->e_shstrndx != 0 && shdr[hdr->e_shstrndx].sh_size != 0 &&
    464   1.8        ad 	    shdr[hdr->e_shstrndx].sh_type == SHT_STRTAB) {
    465   1.8        ad 		ko->ko_shstrtabsz = shdr[hdr->e_shstrndx].sh_size;
    466  1.12        ad 		error = kobj_read(ko, (void *)&ko->ko_shstrtab,
    467   1.8        ad 		    shdr[hdr->e_shstrndx].sh_size,
    468   1.8        ad 		    shdr[hdr->e_shstrndx].sh_offset);
    469   1.8        ad 		if (error != 0) {
    470   1.8        ad 			goto out;
    471   1.8        ad 		}
    472   1.8        ad 	}
    473   1.8        ad 
    474   1.8        ad 	/*
    475   1.1        ad 	 * Size up code/data(progbits) and bss(nobits).
    476   1.1        ad 	 */
    477   1.1        ad 	alignmask = 0;
    478  1.12        ad 	mapbase = 0;
    479   1.1        ad 	for (i = 0; i < hdr->e_shnum; i++) {
    480   1.1        ad 		switch (shdr[i].sh_type) {
    481   1.1        ad 		case SHT_PROGBITS:
    482   1.1        ad 		case SHT_NOBITS:
    483  1.12        ad 			if (mapbase == 0)
    484  1.12        ad 				mapbase = shdr[i].sh_offset;
    485   1.1        ad 			alignmask = shdr[i].sh_addralign - 1;
    486   1.1        ad 			mapsize += alignmask;
    487   1.1        ad 			mapsize &= ~alignmask;
    488   1.1        ad 			mapsize += shdr[i].sh_size;
    489   1.1        ad 			break;
    490   1.1        ad 		}
    491   1.1        ad 	}
    492   1.1        ad 
    493   1.1        ad 	/*
    494   1.1        ad 	 * We know how much space we need for the text/data/bss/etc.
    495   1.1        ad 	 * This stuff needs to be in a single chunk so that profiling etc
    496   1.1        ad 	 * can get the bounds and gdb can associate offsets with modules.
    497   1.1        ad 	 */
    498   1.1        ad 	if (mapsize == 0) {
    499   1.1        ad 		kobj_error("no text/data/bss");
    500   1.1        ad 		goto out;
    501   1.1        ad 	}
    502  1.12        ad 	if (ko->ko_type == KT_MEMORY) {
    503  1.12        ad 		mapbase += (vaddr_t)ko->ko_source;
    504  1.12        ad 	} else {
    505  1.13        ad 		mapbase = uvm_km_alloc(lkm_map, round_page(mapsize),
    506  1.13        ad 		    0, UVM_KMF_WIRED | UVM_KMF_EXEC);
    507  1.12        ad 		if (mapbase == 0) {
    508  1.12        ad 			error = ENOMEM;
    509  1.12        ad 			goto out;
    510  1.12        ad 		}
    511   1.1        ad 	}
    512   1.1        ad 	ko->ko_address = mapbase;
    513   1.1        ad 	ko->ko_size = mapsize;
    514   1.1        ad 
    515   1.1        ad 	/*
    516   1.1        ad 	 * Now load code/data(progbits), zero bss(nobits), allocate space
    517   1.1        ad 	 * for and load relocs
    518   1.1        ad 	 */
    519   1.1        ad 	pb = 0;
    520   1.1        ad 	rl = 0;
    521   1.1        ad 	ra = 0;
    522   1.1        ad 	alignmask = 0;
    523   1.1        ad 	for (i = 0; i < hdr->e_shnum; i++) {
    524   1.1        ad 		switch (shdr[i].sh_type) {
    525   1.1        ad 		case SHT_PROGBITS:
    526   1.1        ad 		case SHT_NOBITS:
    527   1.1        ad 			alignmask = shdr[i].sh_addralign - 1;
    528  1.13        ad 			if (ko->ko_type == KT_MEMORY) {
    529  1.13        ad 				addr = (void *)(shdr[i].sh_offset +
    530  1.13        ad 				    (vaddr_t)ko->ko_source);
    531  1.13        ad 				if (((vaddr_t)addr & alignmask) != 0) {
    532  1.13        ad 					kobj_error("section %d not aligned\n",
    533  1.13        ad 					    i);
    534  1.13        ad 					goto out;
    535  1.13        ad 				}
    536  1.13        ad 			} else {
    537  1.13        ad 				mapbase += alignmask;
    538  1.13        ad 				mapbase &= ~alignmask;
    539  1.13        ad 				addr = (void *)mapbase;
    540  1.13        ad 				mapbase += shdr[i].sh_size;
    541  1.13        ad 			}
    542  1.13        ad 			ko->ko_progtab[pb].addr = addr;
    543   1.1        ad 			if (shdr[i].sh_type == SHT_PROGBITS) {
    544   1.1        ad 				ko->ko_progtab[pb].name = "<<PROGBITS>>";
    545  1.13        ad 				error = kobj_read_bits(ko, addr,
    546  1.13        ad 				    shdr[i].sh_size, shdr[i].sh_offset);
    547   1.1        ad 				if (error != 0) {
    548   1.1        ad 					goto out;
    549   1.1        ad 				}
    550  1.13        ad 			} else if (ko->ko_type == KT_MEMORY &&
    551  1.13        ad 			    shdr[i].sh_size != 0) {
    552  1.13        ad 			    	kobj_error("non-loadable BSS section in "
    553  1.13        ad 			    	    "pre-loaded module");
    554  1.17  jmcneill 				error = EINVAL;
    555  1.13        ad 			    	goto out;
    556   1.1        ad 			} else {
    557   1.1        ad 				ko->ko_progtab[pb].name = "<<NOBITS>>";
    558  1.13        ad 				memset(addr, 0, shdr[i].sh_size);
    559   1.1        ad 			}
    560   1.1        ad 			ko->ko_progtab[pb].size = shdr[i].sh_size;
    561   1.1        ad 			ko->ko_progtab[pb].sec = i;
    562   1.8        ad 			if (ko->ko_shstrtab != NULL && shdr[i].sh_name != 0) {
    563   1.8        ad 				ko->ko_progtab[pb].name =
    564   1.8        ad 				    ko->ko_shstrtab + shdr[i].sh_name;
    565   1.8        ad 			}
    566   1.1        ad 
    567   1.1        ad 			/* Update all symbol values with the offset. */
    568   1.1        ad 			for (j = 0; j < ko->ko_symcnt; j++) {
    569   1.1        ad 				es = &ko->ko_symtab[j];
    570   1.1        ad 				if (es->st_shndx != i) {
    571   1.1        ad 					continue;
    572   1.1        ad 				}
    573  1.13        ad 				es->st_value += (Elf_Addr)addr;
    574   1.1        ad 			}
    575   1.1        ad 			pb++;
    576   1.1        ad 			break;
    577   1.1        ad 		case SHT_REL:
    578   1.1        ad 			ko->ko_reltab[rl].size = shdr[i].sh_size;
    579   1.1        ad 			ko->ko_reltab[rl].size -=
    580   1.1        ad 			    shdr[i].sh_size % sizeof(Elf_Rel);
    581   1.1        ad 			if (ko->ko_reltab[rl].size != 0) {
    582   1.1        ad 				ko->ko_reltab[rl].nrel =
    583   1.1        ad 				    shdr[i].sh_size / sizeof(Elf_Rel);
    584   1.1        ad 				ko->ko_reltab[rl].sec = shdr[i].sh_info;
    585   1.3        ad 				error = kobj_read(ko,
    586  1.12        ad 				    (void **)&ko->ko_reltab[rl].rel,
    587   1.1        ad 				    ko->ko_reltab[rl].size,
    588   1.1        ad 				    shdr[i].sh_offset);
    589   1.1        ad 				if (error != 0) {
    590   1.1        ad 					goto out;
    591   1.1        ad 				}
    592   1.1        ad 			}
    593   1.1        ad 			rl++;
    594   1.1        ad 			break;
    595   1.1        ad 		case SHT_RELA:
    596   1.1        ad 			ko->ko_relatab[ra].size = shdr[i].sh_size;
    597   1.1        ad 			ko->ko_relatab[ra].size -=
    598   1.1        ad 			    shdr[i].sh_size % sizeof(Elf_Rela);
    599   1.1        ad 			if (ko->ko_relatab[ra].size != 0) {
    600   1.1        ad 				ko->ko_relatab[ra].nrela =
    601   1.1        ad 				    shdr[i].sh_size / sizeof(Elf_Rela);
    602   1.1        ad 				ko->ko_relatab[ra].sec = shdr[i].sh_info;
    603   1.3        ad 				error = kobj_read(ko,
    604  1.12        ad 				    (void **)&ko->ko_relatab[ra].rela,
    605   1.1        ad 				    shdr[i].sh_size,
    606   1.1        ad 				    shdr[i].sh_offset);
    607   1.1        ad 				if (error != 0) {
    608   1.1        ad 					goto out;
    609   1.1        ad 				}
    610   1.1        ad 			}
    611   1.1        ad 			ra++;
    612   1.1        ad 			break;
    613  1.13        ad 		default:
    614  1.13        ad 			break;
    615   1.1        ad 		}
    616   1.1        ad 	}
    617   1.1        ad 	if (pb != ko->ko_nprogtab) {
    618   1.1        ad 		panic("lost progbits");
    619   1.1        ad 	}
    620   1.1        ad 	if (rl != ko->ko_nrel) {
    621   1.1        ad 		panic("lost rel");
    622   1.1        ad 	}
    623   1.1        ad 	if (ra != ko->ko_nrela) {
    624   1.1        ad 		panic("lost rela");
    625   1.1        ad 	}
    626  1.13        ad 	if (ko->ko_type != KT_MEMORY && mapbase != ko->ko_address + mapsize) {
    627  1.13        ad 		panic("mapbase 0x%lx != address %lx + mapsize %ld (0x%lx)\n",
    628   1.1        ad 		    (long)mapbase, (long)ko->ko_address, (long)mapsize,
    629   1.1        ad 		    (long)ko->ko_address + mapsize);
    630   1.1        ad 	}
    631   1.1        ad 
    632   1.1        ad 	/*
    633  1.18        ad 	 * Perform local relocations only.  Relocations relating to global
    634  1.18        ad 	 * symbols will be done by kobj_affix().
    635   1.1        ad 	 */
    636  1.18        ad 	error = kobj_relocate(ko, true);
    637   1.1        ad  out:
    638   1.3        ad 	if (hdr != NULL) {
    639  1.12        ad 		kobj_free(ko, hdr, sizeof(*hdr));
    640   1.1        ad 	}
    641  1.18        ad 	kobj_close(ko);
    642  1.18        ad 	if (error != 0) {
    643  1.18        ad 		kobj_unload(ko);
    644  1.18        ad 	}
    645   1.1        ad 
    646   1.1        ad 	return error;
    647   1.1        ad }
    648   1.1        ad 
    649   1.1        ad /*
    650   1.1        ad  * kobj_unload:
    651   1.1        ad  *
    652   1.1        ad  *	Unload an object previously loaded by kobj_load().
    653   1.1        ad  */
    654   1.1        ad void
    655   1.1        ad kobj_unload(kobj_t ko)
    656   1.1        ad {
    657   1.1        ad 	int error;
    658   1.1        ad 
    659  1.18        ad 	kobj_close(ko);
    660  1.18        ad 	kobj_jettison(ko);
    661  1.18        ad 
    662  1.18        ad 	/*
    663  1.18        ad 	 * Notify MD code that a module has been unloaded.
    664  1.18        ad 	 */
    665  1.18        ad 	if (ko->ko_loaded) {
    666  1.18        ad 		error = kobj_machdep(ko, (void *)ko->ko_address, ko->ko_size,
    667  1.18        ad 		    false);
    668  1.18        ad 		if (error != 0) {
    669  1.18        ad 			kobj_error("machine dependent deinit failed");
    670  1.18        ad 		}
    671  1.18        ad 	}
    672  1.12        ad 	if (ko->ko_address != 0 && ko->ko_type != KT_MEMORY) {
    673   1.1        ad 		uvm_km_free(lkm_map, ko->ko_address, round_page(ko->ko_size),
    674   1.1        ad 		    UVM_KMF_WIRED);
    675   1.1        ad 	}
    676   1.1        ad 	if (ko->ko_ksyms == true) {
    677   1.1        ad 		ksyms_delsymtab(ko->ko_name);
    678   1.1        ad 	}
    679   1.1        ad 	if (ko->ko_symtab != NULL) {
    680  1.12        ad 		kobj_free(ko, ko->ko_symtab, ko->ko_symcnt * sizeof(Elf_Sym));
    681   1.1        ad 	}
    682   1.1        ad 	if (ko->ko_strtab != NULL) {
    683  1.12        ad 		kobj_free(ko, ko->ko_strtab, ko->ko_strtabsz);
    684   1.1        ad 	}
    685  1.14        ad 	if (ko->ko_progtab != NULL) {
    686  1.14        ad 		kobj_free(ko, ko->ko_progtab, ko->ko_nprogtab *
    687  1.14        ad 		    sizeof(*ko->ko_progtab));
    688  1.14        ad 		ko->ko_progtab = NULL;
    689  1.14        ad 	}
    690  1.14        ad 	if (ko->ko_shstrtab) {
    691  1.14        ad 		kobj_free(ko, ko->ko_shstrtab, ko->ko_shstrtabsz);
    692  1.14        ad 		ko->ko_shstrtab = NULL;
    693  1.14        ad 	}
    694   1.1        ad 
    695   1.3        ad 	kmem_free(ko, sizeof(*ko));
    696   1.1        ad }
    697   1.1        ad 
    698   1.1        ad /*
    699   1.2        ad  * kobj_stat:
    700   1.2        ad  *
    701   1.2        ad  *	Return size and load address of an object.
    702   1.2        ad  */
    703   1.2        ad void
    704   1.8        ad kobj_stat(kobj_t ko, vaddr_t *address, size_t *size)
    705   1.2        ad {
    706   1.2        ad 
    707   1.2        ad 	if (address != NULL) {
    708   1.2        ad 		*address = ko->ko_address;
    709   1.2        ad 	}
    710   1.2        ad 	if (size != NULL) {
    711   1.2        ad 		*size = ko->ko_size;
    712   1.2        ad 	}
    713   1.2        ad }
    714   1.2        ad 
    715   1.2        ad /*
    716  1.18        ad  * kobj_affix:
    717   1.3        ad  *
    718  1.18        ad  *	Set an object's name and perform global relocs.  May only be
    719  1.18        ad  *	called after the module and any requisite modules are loaded.
    720   1.3        ad  */
    721   1.6        ad int
    722  1.18        ad kobj_affix(kobj_t ko, const char *name)
    723   1.3        ad {
    724   1.6        ad 	int error;
    725   1.3        ad 
    726  1.18        ad 	KASSERT(ko->ko_ksyms == false);
    727  1.18        ad 	KASSERT(ko->ko_loaded == false);
    728   1.3        ad 
    729   1.3        ad 	strlcpy(ko->ko_name, name, sizeof(ko->ko_name));
    730   1.6        ad 
    731  1.18        ad 	/* Now that we know the name, register the symbol table. */
    732   1.6        ad 	error = ksyms_addsymtab(ko->ko_name, ko->ko_symtab, ko->ko_symcnt *
    733   1.6        ad 	    sizeof(Elf_Sym), ko->ko_strtab, ko->ko_strtabsz);
    734   1.6        ad 	if (error != 0) {
    735   1.6        ad 		kobj_error("unable to register module symbol table");
    736   1.6        ad 	} else {
    737   1.6        ad 		ko->ko_ksyms = true;
    738  1.18        ad 		/* Now do global relocations. */
    739  1.18        ad 		error = kobj_relocate(ko, false);
    740  1.18        ad 	}
    741  1.18        ad 
    742  1.18        ad 	/* Jettison unneeded memory post-link. */
    743  1.18        ad 	kobj_jettison(ko);
    744  1.18        ad 
    745  1.18        ad 	/* Notify MD code that a module has been loaded. */
    746  1.18        ad 	if (error == 0) {
    747  1.18        ad 		error = kobj_machdep(ko, (void *)ko->ko_address, ko->ko_size,
    748  1.18        ad 		    true);
    749  1.18        ad 		if (error != 0) {
    750  1.18        ad 			kobj_error("machine dependent init failed");
    751  1.18        ad 		}
    752  1.18        ad 		ko->ko_loaded = true;
    753  1.18        ad 	}
    754  1.18        ad 
    755  1.18        ad 	/* If there was an error, destroy the whole object. */
    756  1.18        ad 	if (error != 0) {
    757  1.18        ad 		kobj_unload(ko);
    758   1.6        ad 	}
    759   1.6        ad 
    760   1.6        ad 	return error;
    761   1.3        ad }
    762   1.3        ad 
    763   1.3        ad /*
    764   1.8        ad  * kobj_find_section:
    765   1.8        ad  *
    766   1.8        ad  *	Given a section name, search the loaded object and return
    767   1.8        ad  *	virtual address if present and loaded.
    768   1.8        ad  */
    769   1.8        ad int
    770   1.8        ad kobj_find_section(kobj_t ko, const char *name, void **addr, size_t *size)
    771   1.8        ad {
    772   1.8        ad 	int i;
    773   1.8        ad 
    774   1.8        ad 	KASSERT(ko->ko_progtab != NULL);
    775   1.8        ad 
    776   1.8        ad 	for (i = 0; i < ko->ko_nprogtab; i++) {
    777   1.8        ad 		if (strcmp(ko->ko_progtab[i].name, name) == 0) {
    778   1.8        ad 			if (addr != NULL) {
    779   1.8        ad 				*addr = ko->ko_progtab[i].addr;
    780   1.8        ad 			}
    781   1.8        ad 			if (size != NULL) {
    782   1.8        ad 				*size = ko->ko_progtab[i].size;
    783   1.8        ad 			}
    784   1.8        ad 			return 0;
    785   1.8        ad 		}
    786   1.8        ad 	}
    787   1.8        ad 
    788   1.8        ad 	return ENOENT;
    789   1.8        ad }
    790   1.8        ad 
    791   1.8        ad /*
    792  1.18        ad  * kobj_jettison:
    793   1.1        ad  *
    794  1.18        ad  *	Release object data not needed after performing relocations.
    795   1.1        ad  */
    796   1.1        ad static void
    797  1.18        ad kobj_jettison(kobj_t ko)
    798   1.1        ad {
    799   1.1        ad 	int i;
    800   1.1        ad 
    801   1.1        ad 	for (i = 0; i < ko->ko_nrel; i++) {
    802   1.1        ad 		if (ko->ko_reltab[i].rel) {
    803  1.12        ad 			kobj_free(ko, ko->ko_reltab[i].rel,
    804   1.1        ad 			    ko->ko_reltab[i].size);
    805   1.1        ad 		}
    806   1.1        ad 	}
    807   1.1        ad 	for (i = 0; i < ko->ko_nrela; i++) {
    808   1.1        ad 		if (ko->ko_relatab[i].rela) {
    809  1.12        ad 			kobj_free(ko, ko->ko_relatab[i].rela,
    810   1.1        ad 			    ko->ko_relatab[i].size);
    811   1.1        ad 		}
    812   1.1        ad 	}
    813   1.1        ad 	if (ko->ko_reltab != NULL) {
    814  1.12        ad 		kobj_free(ko, ko->ko_reltab, ko->ko_nrel *
    815   1.1        ad 		    sizeof(*ko->ko_reltab));
    816   1.1        ad 		ko->ko_reltab = NULL;
    817   1.1        ad 		ko->ko_nrel = 0;
    818   1.1        ad 	}
    819   1.1        ad 	if (ko->ko_relatab != NULL) {
    820  1.12        ad 		kobj_free(ko, ko->ko_relatab, ko->ko_nrela *
    821   1.1        ad 		    sizeof(*ko->ko_relatab));
    822   1.1        ad 		ko->ko_relatab = NULL;
    823   1.1        ad 		ko->ko_nrela = 0;
    824   1.1        ad 	}
    825   1.1        ad 	if (ko->ko_shdr != NULL) {
    826  1.12        ad 		kobj_free(ko, ko->ko_shdr, ko->ko_shdrsz);
    827   1.1        ad 		ko->ko_shdr = NULL;
    828   1.1        ad 	}
    829   1.1        ad }
    830   1.1        ad 
    831   1.1        ad /*
    832   1.1        ad  * kobj_sym_lookup:
    833   1.1        ad  *
    834   1.1        ad  *	Symbol lookup function to be used when the symbol index
    835   1.1        ad  *	is known (ie during relocation).
    836   1.1        ad  */
    837   1.1        ad uintptr_t
    838   1.1        ad kobj_sym_lookup(kobj_t ko, uintptr_t symidx)
    839   1.1        ad {
    840   1.1        ad 	const Elf_Sym *sym;
    841   1.1        ad 	const char *symbol;
    842   1.1        ad 	int error;
    843   1.1        ad 	u_long addr;
    844   1.1        ad 
    845   1.1        ad 	/* Don't even try to lookup the symbol if the index is bogus. */
    846   1.1        ad 	if (symidx >= ko->ko_symcnt)
    847   1.1        ad 		return 0;
    848   1.1        ad 
    849   1.1        ad 	sym = ko->ko_symtab + symidx;
    850   1.1        ad 
    851   1.1        ad 	/* Quick answer if there is a definition included. */
    852   1.1        ad 	if (sym->st_shndx != SHN_UNDEF) {
    853   1.1        ad 		return sym->st_value;
    854   1.1        ad 	}
    855   1.1        ad 
    856   1.1        ad 	/* If we get here, then it is undefined and needs a lookup. */
    857   1.1        ad 	switch (ELF_ST_BIND(sym->st_info)) {
    858   1.1        ad 	case STB_LOCAL:
    859   1.1        ad 		/* Local, but undefined? huh? */
    860   1.1        ad 		kobj_error("local symbol undefined");
    861   1.1        ad 		return 0;
    862   1.1        ad 
    863   1.1        ad 	case STB_GLOBAL:
    864   1.1        ad 		/* Relative to Data or Function name */
    865   1.1        ad 		symbol = ko->ko_strtab + sym->st_name;
    866   1.1        ad 
    867   1.1        ad 		/* Force a lookup failure if the symbol name is bogus. */
    868   1.1        ad 		if (*symbol == 0) {
    869   1.1        ad 			kobj_error("bad symbol name");
    870   1.1        ad 			return 0;
    871   1.1        ad 		}
    872   1.1        ad 
    873   1.1        ad 		error = ksyms_getval(NULL, symbol, &addr, KSYMS_ANY);
    874   1.1        ad 		if (error != 0) {
    875  1.18        ad 			kobj_error("symbol `%s' not found", symbol);
    876   1.1        ad 			return (uintptr_t)0;
    877   1.1        ad 		}
    878   1.1        ad 		return (uintptr_t)addr;
    879   1.1        ad 
    880   1.1        ad 	case STB_WEAK:
    881   1.1        ad 		kobj_error("weak symbols not supported\n");
    882   1.1        ad 		return 0;
    883   1.1        ad 
    884   1.1        ad 	default:
    885   1.1        ad 		return 0;
    886   1.1        ad 	}
    887   1.1        ad }
    888   1.1        ad 
    889   1.1        ad /*
    890   1.1        ad  * kobj_findbase:
    891   1.1        ad  *
    892   1.1        ad  *	Return base address of the given section.
    893   1.1        ad  */
    894   1.1        ad static uintptr_t
    895   1.1        ad kobj_findbase(kobj_t ko, int sec)
    896   1.1        ad {
    897   1.1        ad 	int i;
    898   1.1        ad 
    899   1.1        ad 	for (i = 0; i < ko->ko_nprogtab; i++) {
    900   1.1        ad 		if (sec == ko->ko_progtab[i].sec) {
    901   1.1        ad 			return (uintptr_t)ko->ko_progtab[i].addr;
    902   1.1        ad 		}
    903   1.1        ad 	}
    904   1.1        ad 	return 0;
    905   1.1        ad }
    906   1.1        ad 
    907   1.1        ad /*
    908   1.1        ad  * kobj_relocate:
    909   1.1        ad  *
    910  1.18        ad  *	Resolve relocations for the loaded object.
    911   1.1        ad  */
    912   1.1        ad static int
    913  1.18        ad kobj_relocate(kobj_t ko, bool local)
    914   1.1        ad {
    915   1.1        ad 	const Elf_Rel *rellim;
    916   1.1        ad 	const Elf_Rel *rel;
    917   1.1        ad 	const Elf_Rela *relalim;
    918   1.1        ad 	const Elf_Rela *rela;
    919   1.1        ad 	const Elf_Sym *sym;
    920   1.1        ad 	uintptr_t base;
    921   1.8        ad 	int i, error;
    922   1.1        ad 	uintptr_t symidx;
    923   1.1        ad 
    924   1.1        ad 	/*
    925   1.1        ad 	 * Perform relocations without addend if there are any.
    926   1.1        ad 	 */
    927   1.1        ad 	for (i = 0; i < ko->ko_nrel; i++) {
    928   1.1        ad 		rel = ko->ko_reltab[i].rel;
    929   1.1        ad 		if (rel == NULL) {
    930   1.1        ad 			continue;
    931   1.1        ad 		}
    932   1.1        ad 		rellim = rel + ko->ko_reltab[i].nrel;
    933   1.1        ad 		base = kobj_findbase(ko, ko->ko_reltab[i].sec);
    934   1.1        ad 		if (base == 0) {
    935   1.1        ad 			panic("lost base for e_reltab");
    936   1.1        ad 		}
    937   1.1        ad 		for (; rel < rellim; rel++) {
    938   1.1        ad 			symidx = ELF_R_SYM(rel->r_info);
    939   1.1        ad 			if (symidx >= ko->ko_symcnt) {
    940   1.1        ad 				continue;
    941   1.1        ad 			}
    942   1.1        ad 			sym = ko->ko_symtab + symidx;
    943  1.18        ad 			if (local != (ELF_ST_BIND(sym->st_info) == STB_LOCAL)) {
    944  1.18        ad 				continue;
    945  1.18        ad 			}
    946  1.18        ad 			error = kobj_reloc(ko, base, rel, false, local);
    947   1.8        ad 			if (error != 0) {
    948   1.1        ad 				return ENOENT;
    949   1.1        ad 			}
    950   1.1        ad 		}
    951   1.1        ad 	}
    952   1.1        ad 
    953   1.1        ad 	/*
    954   1.1        ad 	 * Perform relocations with addend if there are any.
    955   1.1        ad 	 */
    956   1.1        ad 	for (i = 0; i < ko->ko_nrela; i++) {
    957   1.1        ad 		rela = ko->ko_relatab[i].rela;
    958   1.1        ad 		if (rela == NULL) {
    959   1.1        ad 			continue;
    960   1.1        ad 		}
    961   1.1        ad 		relalim = rela + ko->ko_relatab[i].nrela;
    962   1.1        ad 		base = kobj_findbase(ko, ko->ko_relatab[i].sec);
    963   1.1        ad 		if (base == 0) {
    964   1.1        ad 			panic("lost base for e_relatab");
    965   1.1        ad 		}
    966   1.1        ad 		for (; rela < relalim; rela++) {
    967   1.1        ad 			symidx = ELF_R_SYM(rela->r_info);
    968   1.1        ad 			if (symidx >= ko->ko_symcnt) {
    969   1.1        ad 				continue;
    970   1.1        ad 			}
    971   1.1        ad 			sym = ko->ko_symtab + symidx;
    972  1.18        ad 			if (local != (ELF_ST_BIND(sym->st_info) == STB_LOCAL)) {
    973  1.18        ad 				continue;
    974  1.18        ad 			}
    975  1.18        ad 			error = kobj_reloc(ko, base, rela, true, local);
    976   1.8        ad 			if (error != 0) {
    977   1.1        ad 				return ENOENT;
    978   1.1        ad 			}
    979   1.1        ad 		}
    980   1.1        ad 	}
    981   1.1        ad 
    982   1.1        ad 	return 0;
    983   1.1        ad }
    984   1.1        ad 
    985   1.1        ad /*
    986   1.1        ad  * kobj_error:
    987   1.1        ad  *
    988   1.1        ad  *	Utility function: log an error.
    989   1.1        ad  */
    990   1.1        ad static void
    991   1.1        ad kobj_error(const char *fmt, ...)
    992   1.1        ad {
    993   1.1        ad 	va_list ap;
    994   1.1        ad 
    995   1.1        ad 	va_start(ap, fmt);
    996   1.1        ad 	printf("WARNING: linker error: ");
    997   1.1        ad 	vprintf(fmt, ap);
    998   1.1        ad 	printf("\n");
    999   1.1        ad 	va_end(ap);
   1000   1.1        ad }
   1001   1.1        ad 
   1002   1.1        ad /*
   1003   1.1        ad  * kobj_read:
   1004   1.1        ad  *
   1005   1.1        ad  *	Utility function: read from the object.
   1006   1.1        ad  */
   1007   1.1        ad static int
   1008  1.12        ad kobj_read(kobj_t ko, void **basep, size_t size, off_t off)
   1009   1.1        ad {
   1010   1.1        ad 	size_t resid;
   1011  1.12        ad 	void *base;
   1012   1.1        ad 	int error;
   1013   1.1        ad 
   1014   1.3        ad 	KASSERT(ko->ko_source != NULL);
   1015   1.3        ad 
   1016   1.3        ad 	switch (ko->ko_type) {
   1017   1.3        ad 	case KT_VNODE:
   1018  1.12        ad 		base = kmem_alloc(size, KM_SLEEP);
   1019  1.12        ad 		if (base == NULL) {
   1020  1.12        ad 			error = ENOMEM;
   1021  1.12        ad 			break;
   1022  1.12        ad 		}
   1023   1.3        ad 		error = vn_rdwr(UIO_READ, ko->ko_source, base, size, off,
   1024   1.3        ad 		    UIO_SYSSPACE, IO_NODELOCKED, curlwp->l_cred, &resid,
   1025   1.3        ad 		    curlwp);
   1026   1.3        ad 		if (error == 0 && resid != 0) {
   1027  1.22        ad 			error = EINVAL;
   1028  1.22        ad 		}
   1029  1.22        ad 		if (error != 0) {
   1030  1.12        ad 			kmem_free(base, size);
   1031  1.22        ad 			base = NULL;
   1032   1.3        ad 		}
   1033   1.3        ad 		break;
   1034   1.3        ad 	case KT_MEMORY:
   1035   1.4  jmcneill 		if (ko->ko_memsize != -1 && off + size > ko->ko_memsize) {
   1036   1.3        ad 			kobj_error("kobj_read: preloaded object short");
   1037   1.3        ad 			error = EINVAL;
   1038  1.12        ad 			base = NULL;
   1039  1.12        ad 		} else {
   1040  1.12        ad 			base = (uint8_t *)ko->ko_source + off;
   1041  1.12        ad 			error = 0;
   1042  1.12        ad 		}
   1043  1.12        ad 		break;
   1044  1.12        ad 	default:
   1045  1.12        ad 		panic("kobj_read: invalid type");
   1046  1.12        ad 	}
   1047  1.12        ad 
   1048  1.12        ad 	*basep = base;
   1049  1.12        ad 	return error;
   1050  1.12        ad }
   1051  1.12        ad 
   1052  1.12        ad /*
   1053  1.12        ad  * kobj_read_bits:
   1054  1.12        ad  *
   1055  1.12        ad  *	Utility function: load a section from the object.
   1056  1.12        ad  */
   1057  1.12        ad static int
   1058  1.12        ad kobj_read_bits(kobj_t ko, void *base, size_t size, off_t off)
   1059  1.12        ad {
   1060  1.12        ad 	size_t resid;
   1061  1.12        ad 	int error;
   1062  1.12        ad 
   1063  1.12        ad 	KASSERT(ko->ko_source != NULL);
   1064  1.12        ad 
   1065  1.12        ad 	switch (ko->ko_type) {
   1066  1.12        ad 	case KT_VNODE:
   1067  1.18        ad 		KASSERT((uintptr_t)base >= (uintptr_t)ko->ko_address);
   1068  1.18        ad 		KASSERT((uintptr_t)base + size <=
   1069  1.18        ad 		    (uintptr_t)ko->ko_address + ko->ko_size);
   1070  1.12        ad 		error = vn_rdwr(UIO_READ, ko->ko_source, base, size, off,
   1071  1.12        ad 		    UIO_SYSSPACE, IO_NODELOCKED, curlwp->l_cred, &resid,
   1072  1.12        ad 		    curlwp);
   1073  1.12        ad 		if (error == 0 && resid != 0) {
   1074  1.12        ad 			error = EINVAL;
   1075  1.12        ad 		}
   1076  1.12        ad 		break;
   1077  1.12        ad 	case KT_MEMORY:
   1078  1.12        ad 		if (ko->ko_memsize != -1 && off + size > ko->ko_memsize) {
   1079  1.12        ad 			kobj_error("kobj_read_bits: preloaded object short");
   1080  1.12        ad 			error = EINVAL;
   1081  1.12        ad 		} else if ((uint8_t *)base != (uint8_t *)ko->ko_source + off) {
   1082  1.12        ad 			kobj_error("kobj_read_bits: object not aligned");
   1083  1.12        ad 			kobj_error("source=%p base=%p off=%d size=%zd",
   1084  1.12        ad 			    ko->ko_source, base, (int)off, size);
   1085  1.12        ad 			error = EINVAL;
   1086   1.3        ad 		} else {
   1087  1.12        ad 			/* Nothing to do.  Loading in-situ. */
   1088   1.3        ad 			error = 0;
   1089   1.3        ad 		}
   1090   1.3        ad 		break;
   1091   1.3        ad 	default:
   1092   1.3        ad 		panic("kobj_read: invalid type");
   1093   1.3        ad 	}
   1094   1.3        ad 
   1095   1.1        ad 	return error;
   1096   1.1        ad }
   1097   1.5        ad 
   1098  1.12        ad /*
   1099  1.12        ad  * kobj_free:
   1100  1.12        ad  *
   1101  1.12        ad  *	Utility function: free memory if it was allocated from the heap.
   1102  1.12        ad  */
   1103  1.12        ad static void
   1104  1.12        ad kobj_free(kobj_t ko, void *base, size_t size)
   1105  1.12        ad {
   1106  1.12        ad 
   1107  1.12        ad 	if (ko->ko_type != KT_MEMORY)
   1108  1.12        ad 		kmem_free(base, size);
   1109  1.12        ad }
   1110  1.12        ad 
   1111   1.5        ad #else	/* MODULAR */
   1112   1.5        ad 
   1113   1.5        ad int
   1114  1.21        ad kobj_load_file(kobj_t *kop, const char *name, const char *base, bool autoload)
   1115   1.5        ad {
   1116   1.5        ad 
   1117   1.5        ad 	return ENOSYS;
   1118   1.5        ad }
   1119   1.5        ad 
   1120   1.5        ad int
   1121  1.18        ad kobj_load_mem(kobj_t *kop, void *base, ssize_t size)
   1122   1.5        ad {
   1123   1.5        ad 
   1124   1.5        ad 	return ENOSYS;
   1125   1.5        ad }
   1126   1.5        ad 
   1127   1.5        ad void
   1128   1.5        ad kobj_unload(kobj_t ko)
   1129   1.5        ad {
   1130   1.5        ad 
   1131   1.5        ad 	panic("not modular");
   1132   1.5        ad }
   1133   1.5        ad 
   1134   1.5        ad void
   1135   1.8        ad kobj_stat(kobj_t ko, vaddr_t *base, size_t *size)
   1136   1.5        ad {
   1137   1.5        ad 
   1138   1.5        ad 	panic("not modular");
   1139   1.5        ad }
   1140   1.5        ad 
   1141   1.7        ad int
   1142  1.18        ad kobj_affix(kobj_t ko, const char *name)
   1143   1.5        ad {
   1144   1.5        ad 
   1145   1.5        ad 	panic("not modular");
   1146   1.5        ad }
   1147   1.5        ad 
   1148   1.8        ad int
   1149   1.8        ad kobj_find_section(kobj_t ko, const char *name, void **addr, size_t *size)
   1150   1.8        ad {
   1151   1.8        ad 
   1152   1.8        ad 	panic("not modular");
   1153   1.8        ad }
   1154   1.8        ad 
   1155   1.5        ad #endif	/* MODULAR */
   1156