/*- * Copyright (c) 1998 Michael Smith * Copyright (c) 1998 Peter Wemm * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions * are met: * 1. Redistributions of source code must retain the above copyright * notice, this list of conditions and the following disclaimer. * 2. Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF * SUCH DAMAGE. * * $Id: load_elf.c,v 1.5 1998/10/13 09:25:27 peter Exp $ */ #include #include #include #include #include #include #include #include #define FREEBSD_ELF #include #include "bootstrap.h" static int elf_loadimage(struct loaded_module *mp, int fd, vm_offset_t loadaddr, Elf_Ehdr *ehdr, int kernel); char *elf_kerneltype = "elf kernel"; char *elf_moduletype = "elf module"; /* * Attempt to load the file (file) as an ELF module. It will be stored at * (dest), and a pointer to a module structure describing the loaded object * will be saved in (result). */ int elf_loadmodule(char *filename, vm_offset_t dest, struct loaded_module **result) { struct loaded_module *mp, *kmp; Elf_Ehdr ehdr; int fd; int err, kernel; u_int pad; char *s; mp = NULL; /* * Open the image, read and validate the ELF header */ if (filename == NULL) /* can't handle nameless */ return(EFTYPE); if ((fd = open(filename, O_RDONLY)) == -1) return(errno); if (read(fd, &ehdr, sizeof(ehdr)) != sizeof(ehdr)) { err = EFTYPE; /* could be EIO, but may be small file */ goto oerr; } /* Is it ELF? */ if (!IS_ELF(ehdr)) { err = EFTYPE; goto oerr; } if (ehdr.e_ident[EI_CLASS] != ELF_TARG_CLASS || /* Layout ? */ ehdr.e_ident[EI_DATA] != ELF_TARG_DATA || ehdr.e_ident[EI_VERSION] != EV_CURRENT || /* Version ? */ ehdr.e_version != EV_CURRENT || ehdr.e_machine != ELF_TARG_MACH) { /* Machine ? */ err = EFTYPE; goto oerr; } /* * Check to see what sort of module we are. */ kmp = mod_findmodule(NULL, NULL); if (ehdr.e_type == ET_DYN) { /* Looks like a kld module */ if (kmp == NULL) { printf("elf_loadmodule: can't load module before kernel\n"); err = EPERM; goto oerr; } if (strcmp(elf_kerneltype, kmp->m_type)) { printf("elf_loadmodule: can't load module with kernel type '%s'\n", kmp->m_type); err = EPERM; goto oerr; } /* Looks OK, got ahead */ kernel = 0; /* Page-align the load address */ pad = (u_int)dest & PAGE_MASK; if (pad != 0) { pad = PAGE_SIZE - pad; dest += pad; } } else if (ehdr.e_type == ET_EXEC) { /* Looks like a kernel */ if (kmp != NULL) { printf("elf_loadmodule: kernel already loaded\n"); err = EPERM; goto oerr; } /* * Calculate destination address based on kernel entrypoint */ dest = (vm_offset_t) ehdr.e_entry; if (dest == 0) { printf("elf_loadmodule: not a kernel (maybe static binary?)\n"); err = EPERM; goto oerr; } kernel = 1; } else { err = EFTYPE; goto oerr; } /* * Ok, we think we should handle this. */ mp = mod_allocmodule(); if (mp == NULL) { printf("elf_loadmodule: cannot allocate module info\n"); err = EPERM; goto out; } if (kernel) setenv("kernelname", filename, 1); s = strrchr(filename, '/'); if (s) mp->m_name = strdup(s + 1); else mp->m_name = strdup(filename); mp->m_type = strdup(kernel ? elf_kerneltype : elf_moduletype); #ifdef ELF_VERBOSE if (kernel) printf("%s entry at %p\n", filename, (void *) dest); #else printf("%s ", filename); #endif mp->m_size = elf_loadimage(mp, fd, dest, &ehdr, kernel); if (mp->m_size == 0 || mp->m_addr == 0) goto ioerr; /* save exec header as metadata */ mod_addmetadata(mp, MODINFOMD_ELFHDR, sizeof(ehdr), &ehdr); /* Load OK, return module pointer */ *result = (struct loaded_module *)mp; err = 0; goto out; ioerr: err = EIO; oerr: mod_discard(mp); out: close(fd); return(err); } /* * With the file (fd) open on the image, and (ehdr) containing * the Elf header, load the image at (off) */ static int elf_loadimage(struct loaded_module *mp, int fd, vm_offset_t off, Elf_Ehdr *ehdr, int kernel) { int i, j; Elf_Phdr *phdr; Elf_Shdr *shdr; int ret; vm_offset_t firstaddr; vm_offset_t lastaddr; void *buf; size_t resid, chunk; vm_offset_t dest; vm_offset_t ssym, esym; Elf_Dyn *dp; int ndp; int deplen; char *depdata; char *s; int len; char *strtab; size_t strsz; int symstrindex; int symtabindex; long size; dp = NULL; shdr = NULL; ret = 0; firstaddr = lastaddr = 0; if (kernel) { #ifdef __i386__ off = 0x10000000; /* -0xf0000000 - i386 relocates after locore */ #else off = 0; /* alpha is direct mapped for kernels */ #endif } phdr = malloc(ehdr->e_phnum * sizeof(*phdr)); if (phdr == NULL) goto out; if (lseek(fd, ehdr->e_phoff, SEEK_SET) == -1) { printf("elf_loadexec: lseek for phdr failed\n"); goto out; } if (read(fd, phdr, ehdr->e_phnum * sizeof(*phdr)) != ehdr->e_phnum * sizeof(*phdr)) { printf("elf_loadexec: cannot read program header\n"); goto out; } for (i = 0; i < ehdr->e_phnum; i++) { /* We want to load PT_LOAD segments only.. */ if (phdr[i].p_type != PT_LOAD) continue; #ifdef ELF_VERBOSE printf("Segment: 0x%lx@0x%lx -> 0x%lx-0x%lx", (long)phdr[i].p_filesz, (long)phdr[i].p_offset, (long)(phdr[i].p_vaddr + off), (long)(phdr[i].p_vaddr + off + phdr[i].p_memsz - 1)); #else if ((phdr[i].p_flags & PF_W) == 0) { printf("text=0x%lx ", (long)phdr[i].p_filesz); } else { printf("data=0x%lx", (long)phdr[i].p_filesz); if (phdr[i].p_filesz < phdr[i].p_memsz) printf("+0x%lx", (long)(phdr[i].p_memsz -phdr[i].p_filesz)); printf(" "); } #endif if (lseek(fd, phdr[i].p_offset, SEEK_SET) == -1) { printf("\nelf_loadexec: cannot seek\n"); goto out; } if (archsw.arch_readin(fd, phdr[i].p_vaddr + off, phdr[i].p_filesz) != phdr[i].p_filesz) { printf("\nelf_loadexec: archsw.readin failed\n"); goto out; } /* clear space from oversized segments; eg: bss */ if (phdr[i].p_filesz < phdr[i].p_memsz) { #ifdef ELF_VERBOSE printf(" (bss: 0x%lx-0x%lx)", (long)(phdr[i].p_vaddr + off + phdr[i].p_filesz), (long)(phdr[i].p_vaddr + off + phdr[i].p_memsz - 1)); #endif /* no archsw.arch_bzero */ buf = malloc(PAGE_SIZE); bzero(buf, PAGE_SIZE); resid = phdr[i].p_memsz - phdr[i].p_filesz; dest = phdr[i].p_vaddr + off + phdr[i].p_filesz; while (resid > 0) { chunk = min(PAGE_SIZE, resid); archsw.arch_copyin(buf, dest, chunk); resid -= chunk; dest += chunk; } free(buf); } #ifdef ELF_VERBOSE printf("\n"); #endif if (firstaddr == 0 || firstaddr > (phdr[i].p_vaddr + off)) firstaddr = phdr[i].p_vaddr + off; if (lastaddr == 0 || lastaddr < (phdr[i].p_vaddr + off + phdr[i].p_memsz)) lastaddr = phdr[i].p_vaddr + off + phdr[i].p_memsz; } lastaddr = roundup(lastaddr, sizeof(long)); /* * Now grab the symbol tables. This isn't easy if we're reading a * .gz file. I think the rule is going to have to be that you must * strip a file to remove symbols before gzipping it so that we do not * try to lseek() on it. */ chunk = ehdr->e_shnum * ehdr->e_shentsize; if (chunk == 0 || ehdr->e_shoff == 0) goto nosyms; shdr = malloc(chunk); if (shdr == NULL) goto nosyms; if (lseek(fd, ehdr->e_shoff, SEEK_SET) == -1) { printf("\nelf_loadimage: cannot lseek() to section headers\n"); goto nosyms; } if (read(fd, shdr, chunk) != chunk) { printf("\nelf_loadimage: read section headers failed\n"); goto nosyms; } symtabindex = -1; symstrindex = -1; for (i = 0; i < ehdr->e_shnum; i++) { if (shdr[i].sh_type != SHT_SYMTAB) continue; for (j = 0; j < ehdr->e_phnum; j++) { if (phdr[j].p_type != PT_LOAD) continue; if (shdr[i].sh_offset >= phdr[j].p_offset && (shdr[i].sh_offset + shdr[i].sh_size <= phdr[j].p_offset + phdr[j].p_filesz)) { shdr[i].sh_offset = 0; shdr[i].sh_size = 0; break; } } if (shdr[i].sh_offset == 0 || shdr[i].sh_size == 0) continue; /* alread loaded in a PT_LOAD above */ /* Save it for loading below */ symtabindex = i; symstrindex = shdr[i].sh_link; } if (symtabindex < 0 || symstrindex < 0) goto nosyms; /* Ok, committed to a load. */ #ifndef ELF_VERBOSE printf("syms=["); #endif ssym = lastaddr; for (i = symtabindex; i >= 0; i = symstrindex) { #ifdef ELF_VERBOSE char *secname; switch(shdr[i].sh_type) { case SHT_SYMTAB: /* Symbol table */ secname = "symtab"; break; case SHT_STRTAB: /* String table */ secname = "strtab"; break; default: secname = "WHOA!!"; break; } #endif size = shdr[i].sh_size; archsw.arch_copyin(&size, lastaddr, sizeof(size)); lastaddr += sizeof(long); #ifdef ELF_VERBOSE printf("%s: 0x%x@0x%x -> 0x%x-0x%x\n", secname, shdr[i].sh_size, shdr[i].sh_offset, lastaddr, lastaddr + shdr[i].sh_size); #else if (i == symstrindex) printf("+"); printf("0x%x+0x%lx", sizeof(size), size); #endif if (lseek(fd, shdr[i].sh_offset, SEEK_SET) == -1) { printf("\nelf_loadimage: could not seek for symbols - skipped!\n"); lastaddr = ssym; ssym = 0; goto nosyms; } if (archsw.arch_readin(fd, lastaddr, shdr[i].sh_size) != shdr[i].sh_size) { printf("\nelf_loadimage: could not read symbols - skipped!\n"); lastaddr = ssym; ssym = 0; goto nosyms; } /* Reset offsets relative to ssym */ lastaddr += shdr[i].sh_size; lastaddr = roundup(lastaddr, sizeof(long)); if (i == symtabindex) symtabindex = -1; else if (i == symstrindex) symstrindex = -1; } esym = lastaddr; #ifndef ELF_VERBOSE printf("]\n"); #endif mod_addmetadata(mp, MODINFOMD_SSYM, sizeof(ssym), &ssym); mod_addmetadata(mp, MODINFOMD_ESYM, sizeof(esym), &esym); nosyms: ret = lastaddr - firstaddr; mp->m_addr = firstaddr; for (i = 0; i < ehdr->e_phnum; i++) { if (phdr[i].p_type == PT_DYNAMIC) { dp = (Elf_Dyn *)(phdr[i].p_vaddr); mod_addmetadata(mp, MODINFOMD_DYNAMIC, sizeof(dp), &dp); dp = NULL; break; } } if (kernel) /* kernel must not depend on anything */ goto out; ndp = 0; for (i = 0; i < ehdr->e_phnum; i++) { if (phdr[i].p_type == PT_DYNAMIC) { ndp = phdr[i].p_filesz / sizeof(Elf_Dyn); dp = malloc(phdr[i].p_filesz); archsw.arch_copyout(phdr[i].p_vaddr + off, dp, phdr[i].p_filesz); } } if (dp == NULL || ndp == 0) goto out; strtab = NULL; strsz = 0; deplen = 0; for (i = 0; i < ndp; i++) { if (dp[i].d_tag == NULL) break; switch (dp[i].d_tag) { case DT_STRTAB: strtab = (char *)(dp[i].d_un.d_ptr + off); break; case DT_STRSZ: strsz = dp[i].d_un.d_val; break; default: break; } } if (strtab == NULL || strsz == 0) goto out; deplen = 0; for (i = 0; i < ndp; i++) { if (dp[i].d_tag == NULL) break; switch (dp[i].d_tag) { case DT_NEEDED: /* count size for dependency list */ j = dp[i].d_un.d_ptr; if (j < 1 || j > (strsz - 2)) continue; /* bad symbol name index */ deplen += strlenout((vm_offset_t)&strtab[j]) + 1; break; default: break; } } if (deplen > 0) { depdata = malloc(deplen); if (depdata == NULL) goto out; s = depdata; for (i = 0; i < ndp; i++) { if (dp[i].d_tag == NULL) break; switch (dp[i].d_tag) { case DT_NEEDED: /* dependency list */ j = dp[i].d_un.d_ptr; len = strlenout((vm_offset_t)&strtab[j]) + 1; archsw.arch_copyout((vm_offset_t)&strtab[j], s, len); s += len; break; default: break; } } if ((s - depdata) > 0) mod_addmetadata(mp, MODINFOMD_DEPLIST, s - depdata, depdata); free(depdata); } out: if (dp) free(dp); if (shdr) free(shdr); if (phdr) free(phdr); return ret; }