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