1e1743d02SSøren Schmidt /*- 221a3ee0eSDavid E. O'Brien * Copyright (c) 2000 David O'Brien 39a14aa01SUlrich Spörlein * Copyright (c) 1995-1996 Søren Schmidt 4e1743d02SSøren Schmidt * Copyright (c) 1996 Peter Wemm 5e1743d02SSøren Schmidt * All rights reserved. 6e1743d02SSøren Schmidt * 7e1743d02SSøren Schmidt * Redistribution and use in source and binary forms, with or without 8e1743d02SSøren Schmidt * modification, are permitted provided that the following conditions 9e1743d02SSøren Schmidt * are met: 10e1743d02SSøren Schmidt * 1. Redistributions of source code must retain the above copyright 11e1743d02SSøren Schmidt * notice, this list of conditions and the following disclaimer 12e1743d02SSøren Schmidt * in this position and unchanged. 13e1743d02SSøren Schmidt * 2. Redistributions in binary form must reproduce the above copyright 14e1743d02SSøren Schmidt * notice, this list of conditions and the following disclaimer in the 15e1743d02SSøren Schmidt * documentation and/or other materials provided with the distribution. 16e1743d02SSøren Schmidt * 3. The name of the author may not be used to endorse or promote products 1721dc7d4fSJens Schweikhardt * derived from this software without specific prior written permission 18e1743d02SSøren Schmidt * 19e1743d02SSøren Schmidt * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 20e1743d02SSøren Schmidt * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 21e1743d02SSøren Schmidt * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 22e1743d02SSøren Schmidt * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 23e1743d02SSøren Schmidt * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 24e1743d02SSøren Schmidt * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 25e1743d02SSøren Schmidt * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 26e1743d02SSøren Schmidt * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 27e1743d02SSøren Schmidt * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 28e1743d02SSøren Schmidt * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 29e1743d02SSøren Schmidt */ 30e1743d02SSøren Schmidt 31677b542eSDavid E. O'Brien #include <sys/cdefs.h> 32677b542eSDavid E. O'Brien __FBSDID("$FreeBSD$"); 33677b542eSDavid E. O'Brien 3412bc222eSJonathan Anderson #include "opt_capsicum.h" 3562919d78SPeter Wemm #include "opt_compat.h" 36aa14e9b7SMark Johnston #include "opt_gzio.h" 3762919d78SPeter Wemm 38e1743d02SSøren Schmidt #include <sys/param.h> 394a144410SRobert Watson #include <sys/capsicum.h> 40e1743d02SSøren Schmidt #include <sys/exec.h> 418c64af4fSJohn Polstra #include <sys/fcntl.h> 42aa14e9b7SMark Johnston #include <sys/gzio.h> 43e1743d02SSøren Schmidt #include <sys/imgact.h> 44e1743d02SSøren Schmidt #include <sys/imgact_elf.h> 45b96bd95bSIan Lepore #include <sys/jail.h> 46e1743d02SSøren Schmidt #include <sys/kernel.h> 47f34fa851SJohn Baldwin #include <sys/lock.h> 48e1743d02SSøren Schmidt #include <sys/malloc.h> 4968ff2a43SChristian S.J. Peron #include <sys/mount.h> 508c64af4fSJohn Polstra #include <sys/mman.h> 51a794e791SBruce Evans #include <sys/namei.h> 528c64af4fSJohn Polstra #include <sys/pioctl.h> 53a794e791SBruce Evans #include <sys/proc.h> 548c64af4fSJohn Polstra #include <sys/procfs.h> 551ba5ad42SEdward Tomasz Napierala #include <sys/racct.h> 568c64af4fSJohn Polstra #include <sys/resourcevar.h> 5789f6b863SAttilio Rao #include <sys/rwlock.h> 58bd390213SMikolaj Golub #include <sys/sbuf.h> 59da61b9a6SAlan Cox #include <sys/sf_buf.h> 60ee235befSKonstantin Belousov #include <sys/smp.h> 6136240ea5SDoug Rabson #include <sys/systm.h> 62e1743d02SSøren Schmidt #include <sys/signalvar.h> 638c64af4fSJohn Polstra #include <sys/stat.h> 641005a129SJohn Baldwin #include <sys/sx.h> 658c64af4fSJohn Polstra #include <sys/syscall.h> 66e1743d02SSøren Schmidt #include <sys/sysctl.h> 678c64af4fSJohn Polstra #include <sys/sysent.h> 68a794e791SBruce Evans #include <sys/vnode.h> 69e7228204SAlfred Perlstein #include <sys/syslog.h> 70e7228204SAlfred Perlstein #include <sys/eventhandler.h> 71f1fca82eSMikolaj Golub #include <sys/user.h> 72e7228204SAlfred Perlstein 73e1743d02SSøren Schmidt #include <vm/vm.h> 74e1743d02SSøren Schmidt #include <vm/vm_kern.h> 75e1743d02SSøren Schmidt #include <vm/vm_param.h> 76e1743d02SSøren Schmidt #include <vm/pmap.h> 77e1743d02SSøren Schmidt #include <vm/vm_map.h> 780ff27d31SJohn Polstra #include <vm/vm_object.h> 79e1743d02SSøren Schmidt #include <vm/vm_extern.h> 80e1743d02SSøren Schmidt 8152c24af7SPeter Wemm #include <machine/elf.h> 82e1743d02SSøren Schmidt #include <machine/md_var.h> 83e1743d02SSøren Schmidt 841b8388cdSMikolaj Golub #define ELF_NOTE_ROUNDSIZE 4 85c815a20cSDavid E. O'Brien #define OLD_EI_BRAND 8 86c815a20cSDavid E. O'Brien 873ebc1248SPeter Wemm static int __elfN(check_header)(const Elf_Ehdr *hdr); 8832c01de2SDmitry Chagin static Elf_Brandinfo *__elfN(get_brandinfo)(struct image_params *imgp, 89d1ae5c83SKonstantin Belousov const char *interp, int interp_name_len, int32_t *osrel); 903ebc1248SPeter Wemm static int __elfN(load_file)(struct proc *p, const char *file, u_long *addr, 913ebc1248SPeter Wemm u_long *entry, size_t pagesize); 92292177e6SAlan Cox static int __elfN(load_section)(struct image_params *imgp, vm_offset_t offset, 93292177e6SAlan Cox caddr_t vmaddr, size_t memsz, size_t filsz, vm_prot_t prot, 94292177e6SAlan Cox size_t pagesize); 953ebc1248SPeter Wemm static int __CONCAT(exec_, __elfN(imgact))(struct image_params *imgp); 9689ffc202SBjoern A. Zeeb static boolean_t __elfN(freebsd_trans_osrel)(const Elf_Note *note, 9789ffc202SBjoern A. Zeeb int32_t *osrel); 9889ffc202SBjoern A. Zeeb static boolean_t kfreebsd_trans_osrel(const Elf_Note *note, int32_t *osrel); 9932c01de2SDmitry Chagin static boolean_t __elfN(check_note)(struct image_params *imgp, 10032c01de2SDmitry Chagin Elf_Brandnote *checknote, int32_t *osrel); 101ed167eaaSKonstantin Belousov static vm_prot_t __elfN(trans_prot)(Elf_Word); 102ed167eaaSKonstantin Belousov static Elf_Word __elfN(untrans_prot)(vm_prot_t); 103e1743d02SSøren Schmidt 104a360a43dSJake Burkholder SYSCTL_NODE(_kern, OID_AUTO, __CONCAT(elf, __ELF_WORD_SIZE), CTLFLAG_RW, 0, 105a360a43dSJake Burkholder ""); 106a360a43dSJake Burkholder 107bd390213SMikolaj Golub #define CORE_BUF_SIZE (16 * 1024) 108e7228204SAlfred Perlstein 109e548a1d4SJake Burkholder int __elfN(fallback_brand) = -1; 110e548a1d4SJake Burkholder SYSCTL_INT(__CONCAT(_kern_elf, __ELF_WORD_SIZE), OID_AUTO, 111af3b2549SHans Petter Selasky fallback_brand, CTLFLAG_RWTUN, &__elfN(fallback_brand), 0, 112a360a43dSJake Burkholder __XSTRING(__CONCAT(ELF, __ELF_WORD_SIZE)) " brand of last resort"); 113a360a43dSJake Burkholder 114551d79e1SMarcel Moolenaar static int elf_legacy_coredump = 0; 115a360a43dSJake Burkholder SYSCTL_INT(_debug, OID_AUTO, __elfN(legacy_coredump), CTLFLAG_RW, 1161cbb879dSEd Maste &elf_legacy_coredump, 0, 1171cbb879dSEd Maste "include all and only RW pages in core dumps"); 118e1743d02SSøren Schmidt 11962c625fdSKonstantin Belousov int __elfN(nxstack) = 1204d22d07aSKonstantin Belousov #if defined(__amd64__) || defined(__powerpc64__) /* both 64 and 32 bit */ || \ 1214d22d07aSKonstantin Belousov (defined(__arm__) && __ARM_ARCH >= 7) || defined(__aarch64__) 12262c625fdSKonstantin Belousov 1; 12362c625fdSKonstantin Belousov #else 12462c625fdSKonstantin Belousov 0; 12562c625fdSKonstantin Belousov #endif 126291c06a1SKonstantin Belousov SYSCTL_INT(__CONCAT(_kern_elf, __ELF_WORD_SIZE), OID_AUTO, 127291c06a1SKonstantin Belousov nxstack, CTLFLAG_RW, &__elfN(nxstack), 0, 128291c06a1SKonstantin Belousov __XSTRING(__CONCAT(ELF, __ELF_WORD_SIZE)) ": enable non-executable stack"); 129291c06a1SKonstantin Belousov 130126b36a2SKonstantin Belousov #if __ELF_WORD_SIZE == 32 131e7d939bdSMarcel Moolenaar #if defined(__amd64__) 132126b36a2SKonstantin Belousov int i386_read_exec = 0; 133126b36a2SKonstantin Belousov SYSCTL_INT(_kern_elf32, OID_AUTO, read_exec, CTLFLAG_RW, &i386_read_exec, 0, 134126b36a2SKonstantin Belousov "enable execution from readable segments"); 135126b36a2SKonstantin Belousov #endif 136126b36a2SKonstantin Belousov #endif 137126b36a2SKonstantin Belousov 1383ebc1248SPeter Wemm static Elf_Brandinfo *elf_brand_list[MAX_BRANDS]; 139e1743d02SSøren Schmidt 140d9c9c81cSPedro F. Giffuni #define trunc_page_ps(va, ps) rounddown2(va, ps) 141d9c9c81cSPedro F. Giffuni #define round_page_ps(va, ps) roundup2(va, ps) 14293d1c728SKonstantin Belousov #define aligned(a, t) (trunc_page_ps((u_long)(a), sizeof(t)) == (u_long)(a)) 14393d1c728SKonstantin Belousov 14432c01de2SDmitry Chagin static const char FREEBSD_ABI_VENDOR[] = "FreeBSD"; 14532c01de2SDmitry Chagin 14632c01de2SDmitry Chagin Elf_Brandnote __elfN(freebsd_brandnote) = { 14732c01de2SDmitry Chagin .hdr.n_namesz = sizeof(FREEBSD_ABI_VENDOR), 14832c01de2SDmitry Chagin .hdr.n_descsz = sizeof(int32_t), 1494c22b468SEd Maste .hdr.n_type = NT_FREEBSD_ABI_TAG, 15032c01de2SDmitry Chagin .vendor = FREEBSD_ABI_VENDOR, 15189ffc202SBjoern A. Zeeb .flags = BN_TRANSLATE_OSREL, 15289ffc202SBjoern A. Zeeb .trans_osrel = __elfN(freebsd_trans_osrel) 15332c01de2SDmitry Chagin }; 15432c01de2SDmitry Chagin 15589ffc202SBjoern A. Zeeb static boolean_t 15689ffc202SBjoern A. Zeeb __elfN(freebsd_trans_osrel)(const Elf_Note *note, int32_t *osrel) 15789ffc202SBjoern A. Zeeb { 15889ffc202SBjoern A. Zeeb uintptr_t p; 15989ffc202SBjoern A. Zeeb 16089ffc202SBjoern A. Zeeb p = (uintptr_t)(note + 1); 1611b8388cdSMikolaj Golub p += roundup2(note->n_namesz, ELF_NOTE_ROUNDSIZE); 16289ffc202SBjoern A. Zeeb *osrel = *(const int32_t *)(p); 16389ffc202SBjoern A. Zeeb 16489ffc202SBjoern A. Zeeb return (TRUE); 16589ffc202SBjoern A. Zeeb } 16689ffc202SBjoern A. Zeeb 16789ffc202SBjoern A. Zeeb static const char GNU_ABI_VENDOR[] = "GNU"; 16889ffc202SBjoern A. Zeeb static int GNU_KFREEBSD_ABI_DESC = 3; 16989ffc202SBjoern A. Zeeb 17089ffc202SBjoern A. Zeeb Elf_Brandnote __elfN(kfreebsd_brandnote) = { 17189ffc202SBjoern A. Zeeb .hdr.n_namesz = sizeof(GNU_ABI_VENDOR), 17289ffc202SBjoern A. Zeeb .hdr.n_descsz = 16, /* XXX at least 16 */ 17389ffc202SBjoern A. Zeeb .hdr.n_type = 1, 17489ffc202SBjoern A. Zeeb .vendor = GNU_ABI_VENDOR, 17589ffc202SBjoern A. Zeeb .flags = BN_TRANSLATE_OSREL, 17689ffc202SBjoern A. Zeeb .trans_osrel = kfreebsd_trans_osrel 17789ffc202SBjoern A. Zeeb }; 17889ffc202SBjoern A. Zeeb 17989ffc202SBjoern A. Zeeb static boolean_t 18089ffc202SBjoern A. Zeeb kfreebsd_trans_osrel(const Elf_Note *note, int32_t *osrel) 18189ffc202SBjoern A. Zeeb { 18289ffc202SBjoern A. Zeeb const Elf32_Word *desc; 18389ffc202SBjoern A. Zeeb uintptr_t p; 18489ffc202SBjoern A. Zeeb 18589ffc202SBjoern A. Zeeb p = (uintptr_t)(note + 1); 1861b8388cdSMikolaj Golub p += roundup2(note->n_namesz, ELF_NOTE_ROUNDSIZE); 18789ffc202SBjoern A. Zeeb 18889ffc202SBjoern A. Zeeb desc = (const Elf32_Word *)p; 18989ffc202SBjoern A. Zeeb if (desc[0] != GNU_KFREEBSD_ABI_DESC) 19089ffc202SBjoern A. Zeeb return (FALSE); 19189ffc202SBjoern A. Zeeb 19289ffc202SBjoern A. Zeeb /* 19389ffc202SBjoern A. Zeeb * Debian GNU/kFreeBSD embed the earliest compatible kernel version 19489ffc202SBjoern A. Zeeb * (__FreeBSD_version: <major><two digit minor>Rxx) in the LSB way. 19589ffc202SBjoern A. Zeeb */ 19689ffc202SBjoern A. Zeeb *osrel = desc[1] * 100000 + desc[2] * 1000 + desc[3]; 19789ffc202SBjoern A. Zeeb 19889ffc202SBjoern A. Zeeb return (TRUE); 19989ffc202SBjoern A. Zeeb } 20089ffc202SBjoern A. Zeeb 201e1743d02SSøren Schmidt int 2023ebc1248SPeter Wemm __elfN(insert_brand_entry)(Elf_Brandinfo *entry) 203e1743d02SSøren Schmidt { 204e1743d02SSøren Schmidt int i; 205e1743d02SSøren Schmidt 2063ebc1248SPeter Wemm for (i = 0; i < MAX_BRANDS; i++) { 207ea5a2b2eSSøren Schmidt if (elf_brand_list[i] == NULL) { 208ea5a2b2eSSøren Schmidt elf_brand_list[i] = entry; 209e1743d02SSøren Schmidt break; 210e1743d02SSøren Schmidt } 211e1743d02SSøren Schmidt } 212925c8b5bSBjoern A. Zeeb if (i == MAX_BRANDS) { 213925c8b5bSBjoern A. Zeeb printf("WARNING: %s: could not insert brandinfo entry: %p\n", 214925c8b5bSBjoern A. Zeeb __func__, entry); 215a7cddfedSJake Burkholder return (-1); 216925c8b5bSBjoern A. Zeeb } 217a7cddfedSJake Burkholder return (0); 218e1743d02SSøren Schmidt } 219e1743d02SSøren Schmidt 220e1743d02SSøren Schmidt int 2213ebc1248SPeter Wemm __elfN(remove_brand_entry)(Elf_Brandinfo *entry) 222e1743d02SSøren Schmidt { 223e1743d02SSøren Schmidt int i; 224e1743d02SSøren Schmidt 2253ebc1248SPeter Wemm for (i = 0; i < MAX_BRANDS; i++) { 226ea5a2b2eSSøren Schmidt if (elf_brand_list[i] == entry) { 227ea5a2b2eSSøren Schmidt elf_brand_list[i] = NULL; 228e1743d02SSøren Schmidt break; 229e1743d02SSøren Schmidt } 230e1743d02SSøren Schmidt } 231ea5a2b2eSSøren Schmidt if (i == MAX_BRANDS) 232a7cddfedSJake Burkholder return (-1); 233a7cddfedSJake Burkholder return (0); 234e1743d02SSøren Schmidt } 235e1743d02SSøren Schmidt 236096977faSMark Newton int 2373ebc1248SPeter Wemm __elfN(brand_inuse)(Elf_Brandinfo *entry) 238096977faSMark Newton { 239096977faSMark Newton struct proc *p; 240553629ebSJake Burkholder int rval = FALSE; 241096977faSMark Newton 2421005a129SJohn Baldwin sx_slock(&allproc_lock); 2434f506694SXin LI FOREACH_PROC_IN_SYSTEM(p) { 244553629ebSJake Burkholder if (p->p_sysent == entry->sysvec) { 245553629ebSJake Burkholder rval = TRUE; 246553629ebSJake Burkholder break; 247096977faSMark Newton } 248553629ebSJake Burkholder } 2491005a129SJohn Baldwin sx_sunlock(&allproc_lock); 250096977faSMark Newton 251553629ebSJake Burkholder return (rval); 252096977faSMark Newton } 253096977faSMark Newton 2545fe3ed62SJake Burkholder static Elf_Brandinfo * 25532c01de2SDmitry Chagin __elfN(get_brandinfo)(struct image_params *imgp, const char *interp, 256d1ae5c83SKonstantin Belousov int interp_name_len, int32_t *osrel) 2575fe3ed62SJake Burkholder { 25832c01de2SDmitry Chagin const Elf_Ehdr *hdr = (const Elf_Ehdr *)imgp->image_header; 259af582aaeSKonstantin Belousov Elf_Brandinfo *bi, *bi_m; 26032c01de2SDmitry Chagin boolean_t ret; 2615fe3ed62SJake Burkholder int i; 2625fe3ed62SJake Burkholder 2635fe3ed62SJake Burkholder /* 26432c01de2SDmitry Chagin * We support four types of branding -- (1) the ELF EI_OSABI field 2655fe3ed62SJake Burkholder * that SCO added to the ELF spec, (2) FreeBSD 3.x's traditional string 26632c01de2SDmitry Chagin * branding w/in the ELF header, (3) path of the `interp_path' 26732c01de2SDmitry Chagin * field, and (4) the ".note.ABI-tag" ELF section. 2685fe3ed62SJake Burkholder */ 2695fe3ed62SJake Burkholder 27032c01de2SDmitry Chagin /* Look for an ".note.ABI-tag" ELF section */ 271af582aaeSKonstantin Belousov bi_m = NULL; 27232c01de2SDmitry Chagin for (i = 0; i < MAX_BRANDS; i++) { 27332c01de2SDmitry Chagin bi = elf_brand_list[i]; 274ecc2fda8SBjoern A. Zeeb if (bi == NULL) 275ecc2fda8SBjoern A. Zeeb continue; 276ecc2fda8SBjoern A. Zeeb if (hdr->e_machine == bi->machine && (bi->flags & 277ecc2fda8SBjoern A. Zeeb (BI_BRAND_NOTE|BI_BRAND_NOTE_MANDATORY)) != 0) { 27832c01de2SDmitry Chagin ret = __elfN(check_note)(imgp, bi->brand_note, osrel); 279f19d421aSNathan Whitehorn /* Give brand a chance to veto check_note's guess */ 280f19d421aSNathan Whitehorn if (ret && bi->header_supported) 281f19d421aSNathan Whitehorn ret = bi->header_supported(imgp); 282af582aaeSKonstantin Belousov /* 283af582aaeSKonstantin Belousov * If note checker claimed the binary, but the 284af582aaeSKonstantin Belousov * interpreter path in the image does not 285af582aaeSKonstantin Belousov * match default one for the brand, try to 286af582aaeSKonstantin Belousov * search for other brands with the same 287af582aaeSKonstantin Belousov * interpreter. Either there is better brand 288af582aaeSKonstantin Belousov * with the right interpreter, or, failing 289af582aaeSKonstantin Belousov * this, we return first brand which accepted 290af582aaeSKonstantin Belousov * our note and, optionally, header. 291af582aaeSKonstantin Belousov */ 292af582aaeSKonstantin Belousov if (ret && bi_m == NULL && (strlen(bi->interp_path) + 293af582aaeSKonstantin Belousov 1 != interp_name_len || strncmp(interp, 294af582aaeSKonstantin Belousov bi->interp_path, interp_name_len) != 0)) { 295af582aaeSKonstantin Belousov bi_m = bi; 296af582aaeSKonstantin Belousov ret = 0; 297af582aaeSKonstantin Belousov } 29832c01de2SDmitry Chagin if (ret) 29932c01de2SDmitry Chagin return (bi); 30032c01de2SDmitry Chagin } 30132c01de2SDmitry Chagin } 302af582aaeSKonstantin Belousov if (bi_m != NULL) 303af582aaeSKonstantin Belousov return (bi_m); 30432c01de2SDmitry Chagin 3055fe3ed62SJake Burkholder /* If the executable has a brand, search for it in the brand list. */ 3065fe3ed62SJake Burkholder for (i = 0; i < MAX_BRANDS; i++) { 3075fe3ed62SJake Burkholder bi = elf_brand_list[i]; 308ecc2fda8SBjoern A. Zeeb if (bi == NULL || bi->flags & BI_BRAND_NOTE_MANDATORY) 309ecc2fda8SBjoern A. Zeeb continue; 310ecc2fda8SBjoern A. Zeeb if (hdr->e_machine == bi->machine && 3115fe3ed62SJake Burkholder (hdr->e_ident[EI_OSABI] == bi->brand || 3125fe3ed62SJake Burkholder strncmp((const char *)&hdr->e_ident[OLD_EI_BRAND], 313686d2f31SNathan Whitehorn bi->compat_3_brand, strlen(bi->compat_3_brand)) == 0)) { 314686d2f31SNathan Whitehorn /* Looks good, but give brand a chance to veto */ 315686d2f31SNathan Whitehorn if (!bi->header_supported || bi->header_supported(imgp)) 3165fe3ed62SJake Burkholder return (bi); 3175fe3ed62SJake Burkholder } 318686d2f31SNathan Whitehorn } 3195fe3ed62SJake Burkholder 320817dc004SWarner Losh /* No known brand, see if the header is recognized by any brand */ 321817dc004SWarner Losh for (i = 0; i < MAX_BRANDS; i++) { 322817dc004SWarner Losh bi = elf_brand_list[i]; 323817dc004SWarner Losh if (bi == NULL || bi->flags & BI_BRAND_NOTE_MANDATORY || 324817dc004SWarner Losh bi->header_supported == NULL) 325817dc004SWarner Losh continue; 326817dc004SWarner Losh if (hdr->e_machine == bi->machine) { 327817dc004SWarner Losh ret = bi->header_supported(imgp); 328817dc004SWarner Losh if (ret) 329817dc004SWarner Losh return (bi); 330817dc004SWarner Losh } 331817dc004SWarner Losh } 332817dc004SWarner Losh 3335fe3ed62SJake Burkholder /* Lacking a known brand, search for a recognized interpreter. */ 3345fe3ed62SJake Burkholder if (interp != NULL) { 3355fe3ed62SJake Burkholder for (i = 0; i < MAX_BRANDS; i++) { 3365fe3ed62SJake Burkholder bi = elf_brand_list[i]; 337ecc2fda8SBjoern A. Zeeb if (bi == NULL || bi->flags & BI_BRAND_NOTE_MANDATORY) 338ecc2fda8SBjoern A. Zeeb continue; 339ecc2fda8SBjoern A. Zeeb if (hdr->e_machine == bi->machine && 340d1ae5c83SKonstantin Belousov /* ELF image p_filesz includes terminating zero */ 341d1ae5c83SKonstantin Belousov strlen(bi->interp_path) + 1 == interp_name_len && 342d1ae5c83SKonstantin Belousov strncmp(interp, bi->interp_path, interp_name_len) 343d1ae5c83SKonstantin Belousov == 0) 3445fe3ed62SJake Burkholder return (bi); 3455fe3ed62SJake Burkholder } 3465fe3ed62SJake Burkholder } 3475fe3ed62SJake Burkholder 3485fe3ed62SJake Burkholder /* Lacking a recognized interpreter, try the default brand */ 3495fe3ed62SJake Burkholder for (i = 0; i < MAX_BRANDS; i++) { 3505fe3ed62SJake Burkholder bi = elf_brand_list[i]; 351ecc2fda8SBjoern A. Zeeb if (bi == NULL || bi->flags & BI_BRAND_NOTE_MANDATORY) 352ecc2fda8SBjoern A. Zeeb continue; 353ecc2fda8SBjoern A. Zeeb if (hdr->e_machine == bi->machine && 354e548a1d4SJake Burkholder __elfN(fallback_brand) == bi->brand) 3555fe3ed62SJake Burkholder return (bi); 3565fe3ed62SJake Burkholder } 3575fe3ed62SJake Burkholder return (NULL); 3585fe3ed62SJake Burkholder } 3595fe3ed62SJake Burkholder 360e1743d02SSøren Schmidt static int 3613ebc1248SPeter Wemm __elfN(check_header)(const Elf_Ehdr *hdr) 362e1743d02SSøren Schmidt { 363d0ca7c29SPeter Wemm Elf_Brandinfo *bi; 3643ebc1248SPeter Wemm int i; 3653ebc1248SPeter Wemm 36652c24af7SPeter Wemm if (!IS_ELF(*hdr) || 36752c24af7SPeter Wemm hdr->e_ident[EI_CLASS] != ELF_TARG_CLASS || 36852c24af7SPeter Wemm hdr->e_ident[EI_DATA] != ELF_TARG_DATA || 3693dc19c46SJacques Vidrine hdr->e_ident[EI_VERSION] != EV_CURRENT || 3703dc19c46SJacques Vidrine hdr->e_phentsize != sizeof(Elf_Phdr) || 3713dc19c46SJacques Vidrine hdr->e_version != ELF_TARG_VER) 372a7cddfedSJake Burkholder return (ENOEXEC); 373e1743d02SSøren Schmidt 3743ebc1248SPeter Wemm /* 3753ebc1248SPeter Wemm * Make sure we have at least one brand for this machine. 3763ebc1248SPeter Wemm */ 3773ebc1248SPeter Wemm 3783ebc1248SPeter Wemm for (i = 0; i < MAX_BRANDS; i++) { 379d0ca7c29SPeter Wemm bi = elf_brand_list[i]; 380d0ca7c29SPeter Wemm if (bi != NULL && bi->machine == hdr->e_machine) 3813ebc1248SPeter Wemm break; 3823ebc1248SPeter Wemm } 3833ebc1248SPeter Wemm if (i == MAX_BRANDS) 384a7cddfedSJake Burkholder return (ENOEXEC); 385e1743d02SSøren Schmidt 386a7cddfedSJake Burkholder return (0); 387e1743d02SSøren Schmidt } 388e1743d02SSøren Schmidt 389e1743d02SSøren Schmidt static int 3903ebc1248SPeter Wemm __elfN(map_partial)(vm_map_t map, vm_object_t object, vm_ooffset_t offset, 391ff6f03c7SAlan Cox vm_offset_t start, vm_offset_t end, vm_prot_t prot) 3923ebc1248SPeter Wemm { 393da61b9a6SAlan Cox struct sf_buf *sf; 394da61b9a6SAlan Cox int error; 3953ebc1248SPeter Wemm vm_offset_t off; 3963ebc1248SPeter Wemm 3973ebc1248SPeter Wemm /* 3983ebc1248SPeter Wemm * Create the page if it doesn't exist yet. Ignore errors. 3993ebc1248SPeter Wemm */ 4003ebc1248SPeter Wemm vm_map_lock(map); 401ff6f03c7SAlan Cox vm_map_insert(map, NULL, 0, trunc_page(start), round_page(end), 402ff6f03c7SAlan Cox VM_PROT_ALL, VM_PROT_ALL, 0); 4033ebc1248SPeter Wemm vm_map_unlock(map); 4043ebc1248SPeter Wemm 4053ebc1248SPeter Wemm /* 4063ebc1248SPeter Wemm * Find the page from the underlying object. 4073ebc1248SPeter Wemm */ 4083ebc1248SPeter Wemm if (object) { 409da61b9a6SAlan Cox sf = vm_imgact_map_page(object, offset); 410da61b9a6SAlan Cox if (sf == NULL) 411da61b9a6SAlan Cox return (KERN_FAILURE); 4123ebc1248SPeter Wemm off = offset - trunc_page(offset); 413da61b9a6SAlan Cox error = copyout((caddr_t)sf_buf_kva(sf) + off, (caddr_t)start, 414ca0387efSJake Burkholder end - start); 415be996836SAttilio Rao vm_imgact_unmap_page(sf); 4163ebc1248SPeter Wemm if (error) { 417a7cddfedSJake Burkholder return (KERN_FAILURE); 4183ebc1248SPeter Wemm } 4193ebc1248SPeter Wemm } 4203ebc1248SPeter Wemm 421a7cddfedSJake Burkholder return (KERN_SUCCESS); 4223ebc1248SPeter Wemm } 4233ebc1248SPeter Wemm 4243ebc1248SPeter Wemm static int 4253ebc1248SPeter Wemm __elfN(map_insert)(vm_map_t map, vm_object_t object, vm_ooffset_t offset, 426ff6f03c7SAlan Cox vm_offset_t start, vm_offset_t end, vm_prot_t prot, int cow) 4273ebc1248SPeter Wemm { 428da61b9a6SAlan Cox struct sf_buf *sf; 429da61b9a6SAlan Cox vm_offset_t off; 430a063facbSMarcel Moolenaar vm_size_t sz; 431a063facbSMarcel Moolenaar int error, rv; 4323ebc1248SPeter Wemm 4333ebc1248SPeter Wemm if (start != trunc_page(start)) { 43481f223caSJake Burkholder rv = __elfN(map_partial)(map, object, offset, start, 435ff6f03c7SAlan Cox round_page(start), prot); 4363ebc1248SPeter Wemm if (rv) 437a7cddfedSJake Burkholder return (rv); 4383ebc1248SPeter Wemm offset += round_page(start) - start; 4393ebc1248SPeter Wemm start = round_page(start); 4403ebc1248SPeter Wemm } 4413ebc1248SPeter Wemm if (end != round_page(end)) { 44281f223caSJake Burkholder rv = __elfN(map_partial)(map, object, offset + 443ff6f03c7SAlan Cox trunc_page(end) - start, trunc_page(end), end, prot); 4443ebc1248SPeter Wemm if (rv) 445a7cddfedSJake Burkholder return (rv); 4463ebc1248SPeter Wemm end = trunc_page(end); 4473ebc1248SPeter Wemm } 4483ebc1248SPeter Wemm if (end > start) { 4493ebc1248SPeter Wemm if (offset & PAGE_MASK) { 4503ebc1248SPeter Wemm /* 4513ebc1248SPeter Wemm * The mapping is not page aligned. This means we have 4523ebc1248SPeter Wemm * to copy the data. Sigh. 4533ebc1248SPeter Wemm */ 454edb572a3SJohn Baldwin rv = vm_map_find(map, NULL, 0, &start, end - start, 0, 455edb572a3SJohn Baldwin VMFS_NO_SPACE, prot | VM_PROT_WRITE, VM_PROT_ALL, 456edb572a3SJohn Baldwin 0); 4573ebc1248SPeter Wemm if (rv) 458a7cddfedSJake Burkholder return (rv); 459da61b9a6SAlan Cox if (object == NULL) 460da61b9a6SAlan Cox return (KERN_SUCCESS); 461da61b9a6SAlan Cox for (; start < end; start += sz) { 462da61b9a6SAlan Cox sf = vm_imgact_map_page(object, offset); 463da61b9a6SAlan Cox if (sf == NULL) 464da61b9a6SAlan Cox return (KERN_FAILURE); 4653ebc1248SPeter Wemm off = offset - trunc_page(offset); 4663ebc1248SPeter Wemm sz = end - start; 467da61b9a6SAlan Cox if (sz > PAGE_SIZE - off) 468da61b9a6SAlan Cox sz = PAGE_SIZE - off; 469da61b9a6SAlan Cox error = copyout((caddr_t)sf_buf_kva(sf) + off, 4703ebc1248SPeter Wemm (caddr_t)start, sz); 471be996836SAttilio Rao vm_imgact_unmap_page(sf); 4723ebc1248SPeter Wemm if (error) { 473a7cddfedSJake Burkholder return (KERN_FAILURE); 4743ebc1248SPeter Wemm } 475da61b9a6SAlan Cox offset += sz; 4763ebc1248SPeter Wemm } 4773ebc1248SPeter Wemm rv = KERN_SUCCESS; 4783ebc1248SPeter Wemm } else { 479e5e6093bSAlan Cox vm_object_reference(object); 4803ebc1248SPeter Wemm vm_map_lock(map); 4813ebc1248SPeter Wemm rv = vm_map_insert(map, object, offset, start, end, 482ff6f03c7SAlan Cox prot, VM_PROT_ALL, cow); 4833ebc1248SPeter Wemm vm_map_unlock(map); 484e5e6093bSAlan Cox if (rv != KERN_SUCCESS) 485e5e6093bSAlan Cox vm_object_deallocate(object); 4863ebc1248SPeter Wemm } 487a7cddfedSJake Burkholder return (rv); 4883ebc1248SPeter Wemm } else { 489a7cddfedSJake Burkholder return (KERN_SUCCESS); 4903ebc1248SPeter Wemm } 4913ebc1248SPeter Wemm } 4923ebc1248SPeter Wemm 4933ebc1248SPeter Wemm static int 494292177e6SAlan Cox __elfN(load_section)(struct image_params *imgp, vm_offset_t offset, 4953ebc1248SPeter Wemm caddr_t vmaddr, size_t memsz, size_t filsz, vm_prot_t prot, 4963ebc1248SPeter Wemm size_t pagesize) 497e1743d02SSøren Schmidt { 498da61b9a6SAlan Cox struct sf_buf *sf; 499e1743d02SSøren Schmidt size_t map_len; 500292177e6SAlan Cox vm_map_t map; 501292177e6SAlan Cox vm_object_t object; 502e1743d02SSøren Schmidt vm_offset_t map_addr; 503fa7dd9c5SMatthew Dillon int error, rv, cow; 504e1743d02SSøren Schmidt size_t copy_len; 50552c24af7SPeter Wemm vm_offset_t file_addr; 50652c24af7SPeter Wemm 50725ead034SBrian Feldman /* 50825ead034SBrian Feldman * It's necessary to fail if the filsz + offset taken from the 50925ead034SBrian Feldman * header is greater than the actual file pager object's size. 51025ead034SBrian Feldman * If we were to allow this, then the vm_map_find() below would 51125ead034SBrian Feldman * walk right off the end of the file object and into the ether. 51225ead034SBrian Feldman * 51325ead034SBrian Feldman * While I'm here, might as well check for something else that 51425ead034SBrian Feldman * is invalid: filsz cannot be greater than memsz. 51525ead034SBrian Feldman */ 516292177e6SAlan Cox if ((off_t)filsz + offset > imgp->attr->va_size || filsz > memsz) { 51725ead034SBrian Feldman uprintf("elf_load_section: truncated ELF file\n"); 51825ead034SBrian Feldman return (ENOEXEC); 51925ead034SBrian Feldman } 52025ead034SBrian Feldman 521292177e6SAlan Cox object = imgp->object; 522292177e6SAlan Cox map = &imgp->proc->p_vmspace->vm_map; 5233ebc1248SPeter Wemm map_addr = trunc_page_ps((vm_offset_t)vmaddr, pagesize); 5243ebc1248SPeter Wemm file_addr = trunc_page_ps(offset, pagesize); 525e1743d02SSøren Schmidt 526e1743d02SSøren Schmidt /* 52752c24af7SPeter Wemm * We have two choices. We can either clear the data in the last page 52852c24af7SPeter Wemm * of an oversized mapping, or we can start the anon mapping a page 52952c24af7SPeter Wemm * early and copy the initialized data into that first page. We 53052c24af7SPeter Wemm * choose the second.. 53152c24af7SPeter Wemm */ 53252c24af7SPeter Wemm if (memsz > filsz) 5333ebc1248SPeter Wemm map_len = trunc_page_ps(offset + filsz, pagesize) - file_addr; 53452c24af7SPeter Wemm else 5353ebc1248SPeter Wemm map_len = round_page_ps(offset + filsz, pagesize) - file_addr; 53652c24af7SPeter Wemm 53752c24af7SPeter Wemm if (map_len != 0) { 538fa7dd9c5SMatthew Dillon /* cow flags: don't dump readonly sections in core */ 539fa7dd9c5SMatthew Dillon cow = MAP_COPY_ON_WRITE | MAP_PREFAULT | 540fa7dd9c5SMatthew Dillon (prot & VM_PROT_WRITE ? 0 : MAP_DISABLE_COREDUMP); 541fa7dd9c5SMatthew Dillon 542292177e6SAlan Cox rv = __elfN(map_insert)(map, 54352c24af7SPeter Wemm object, 54452c24af7SPeter Wemm file_addr, /* file offset */ 54552c24af7SPeter Wemm map_addr, /* virtual start */ 54652c24af7SPeter Wemm map_addr + map_len,/* virtual end */ 54752c24af7SPeter Wemm prot, 548fa7dd9c5SMatthew Dillon cow); 549e5e6093bSAlan Cox if (rv != KERN_SUCCESS) 550a7cddfedSJake Burkholder return (EINVAL); 55152c24af7SPeter Wemm 55252c24af7SPeter Wemm /* we can stop now if we've covered it all */ 55323955314SAlfred Perlstein if (memsz == filsz) { 554a7cddfedSJake Burkholder return (0); 55552c24af7SPeter Wemm } 55623955314SAlfred Perlstein } 55752c24af7SPeter Wemm 55852c24af7SPeter Wemm 55952c24af7SPeter Wemm /* 56052c24af7SPeter Wemm * We have to get the remaining bit of the file into the first part 56152c24af7SPeter Wemm * of the oversized map segment. This is normally because the .data 56252c24af7SPeter Wemm * segment in the file is extended to provide bss. It's a neat idea 56352c24af7SPeter Wemm * to try and save a page, but it's a pain in the behind to implement. 564e1743d02SSøren Schmidt */ 5653ebc1248SPeter Wemm copy_len = (offset + filsz) - trunc_page_ps(offset + filsz, pagesize); 5663ebc1248SPeter Wemm map_addr = trunc_page_ps((vm_offset_t)vmaddr + filsz, pagesize); 567ca0387efSJake Burkholder map_len = round_page_ps((vm_offset_t)vmaddr + memsz, pagesize) - 568ca0387efSJake Burkholder map_addr; 569e1743d02SSøren Schmidt 57052c24af7SPeter Wemm /* This had damn well better be true! */ 5718191d577SPeter Wemm if (map_len != 0) { 572292177e6SAlan Cox rv = __elfN(map_insert)(map, NULL, 0, map_addr, map_addr + 573292177e6SAlan Cox map_len, VM_PROT_ALL, 0); 57423955314SAlfred Perlstein if (rv != KERN_SUCCESS) { 575a7cddfedSJake Burkholder return (EINVAL); 5768191d577SPeter Wemm } 57723955314SAlfred Perlstein } 578e1743d02SSøren Schmidt 57952c24af7SPeter Wemm if (copy_len != 0) { 5803ebc1248SPeter Wemm vm_offset_t off; 581da61b9a6SAlan Cox 582da61b9a6SAlan Cox sf = vm_imgact_map_page(object, offset + filsz); 583da61b9a6SAlan Cox if (sf == NULL) 584da61b9a6SAlan Cox return (EIO); 585e1743d02SSøren Schmidt 58652c24af7SPeter Wemm /* send the page fragment to user space */ 58781f223caSJake Burkholder off = trunc_page_ps(offset + filsz, pagesize) - 58881f223caSJake Burkholder trunc_page(offset + filsz); 589da61b9a6SAlan Cox error = copyout((caddr_t)sf_buf_kva(sf) + off, 590da61b9a6SAlan Cox (caddr_t)map_addr, copy_len); 591be996836SAttilio Rao vm_imgact_unmap_page(sf); 59223955314SAlfred Perlstein if (error) { 59352c24af7SPeter Wemm return (error); 59452c24af7SPeter Wemm } 59523955314SAlfred Perlstein } 596e1743d02SSøren Schmidt 597e1743d02SSøren Schmidt /* 5983ebc1248SPeter Wemm * set it to the specified protection. 5993ebc1248SPeter Wemm * XXX had better undo the damage from pasting over the cracks here! 600e1743d02SSøren Schmidt */ 601292177e6SAlan Cox vm_map_protect(map, trunc_page(map_addr), round_page(map_addr + 602292177e6SAlan Cox map_len), prot, FALSE); 6038191d577SPeter Wemm 604ff6f03c7SAlan Cox return (0); 605e1743d02SSøren Schmidt } 606e1743d02SSøren Schmidt 607c33fe779SJohn Polstra /* 608c33fe779SJohn Polstra * Load the file "file" into memory. It may be either a shared object 609c33fe779SJohn Polstra * or an executable. 610c33fe779SJohn Polstra * 611c33fe779SJohn Polstra * The "addr" reference parameter is in/out. On entry, it specifies 612c33fe779SJohn Polstra * the address where a shared object should be loaded. If the file is 613c33fe779SJohn Polstra * an executable, this value is ignored. On exit, "addr" specifies 614c33fe779SJohn Polstra * where the file was actually loaded. 615c33fe779SJohn Polstra * 616c33fe779SJohn Polstra * The "entry" reference parameter is out only. On exit, it specifies 617c33fe779SJohn Polstra * the entry point for the loaded file. 618c33fe779SJohn Polstra */ 619e1743d02SSøren Schmidt static int 6203ebc1248SPeter Wemm __elfN(load_file)(struct proc *p, const char *file, u_long *addr, 6213ebc1248SPeter Wemm u_long *entry, size_t pagesize) 622e1743d02SSøren Schmidt { 623911c2be0SMark Peek struct { 624911c2be0SMark Peek struct nameidata nd; 625911c2be0SMark Peek struct vattr attr; 626911c2be0SMark Peek struct image_params image_params; 627911c2be0SMark Peek } *tempdata; 628d254af07SMatthew Dillon const Elf_Ehdr *hdr = NULL; 629d254af07SMatthew Dillon const Elf_Phdr *phdr = NULL; 630911c2be0SMark Peek struct nameidata *nd; 631911c2be0SMark Peek struct vattr *attr; 632911c2be0SMark Peek struct image_params *imgp; 63352c24af7SPeter Wemm vm_prot_t prot; 634c33fe779SJohn Polstra u_long rbase; 635c33fe779SJohn Polstra u_long base_addr = 0; 6365050aa86SKonstantin Belousov int error, i, numsegs; 637e1743d02SSøren Schmidt 63812bc222eSJonathan Anderson #ifdef CAPABILITY_MODE 63912bc222eSJonathan Anderson /* 64012bc222eSJonathan Anderson * XXXJA: This check can go away once we are sufficiently confident 64112bc222eSJonathan Anderson * that the checks in namei() are correct. 64212bc222eSJonathan Anderson */ 64312bc222eSJonathan Anderson if (IN_CAPABILITY_MODE(curthread)) 64412bc222eSJonathan Anderson return (ECAPMODE); 64512bc222eSJonathan Anderson #endif 64612bc222eSJonathan Anderson 647a163d034SWarner Losh tempdata = malloc(sizeof(*tempdata), M_TEMP, M_WAITOK); 648911c2be0SMark Peek nd = &tempdata->nd; 649911c2be0SMark Peek attr = &tempdata->attr; 650911c2be0SMark Peek imgp = &tempdata->image_params; 651911c2be0SMark Peek 652c8a79999SPeter Wemm /* 653c8a79999SPeter Wemm * Initialize part of the common data 654c8a79999SPeter Wemm */ 655c8a79999SPeter Wemm imgp->proc = p; 656911c2be0SMark Peek imgp->attr = attr; 657c8a79999SPeter Wemm imgp->firstpage = NULL; 65859c8bc40SAlan Cox imgp->image_header = NULL; 6590b2ed1aeSJeff Roberson imgp->object = NULL; 6606d7bdc8dSRobert Watson imgp->execlabel = NULL; 661c8a79999SPeter Wemm 6625050aa86SKonstantin Belousov NDINIT(nd, LOOKUP, LOCKLEAF | FOLLOW, UIO_SYSSPACE, file, curthread); 663911c2be0SMark Peek if ((error = namei(nd)) != 0) { 664911c2be0SMark Peek nd->ni_vp = NULL; 665e1743d02SSøren Schmidt goto fail; 666e1743d02SSøren Schmidt } 667911c2be0SMark Peek NDFREE(nd, NDF_ONLY_PNBUF); 668911c2be0SMark Peek imgp->vp = nd->ni_vp; 669c8a79999SPeter Wemm 670e1743d02SSøren Schmidt /* 671e1743d02SSøren Schmidt * Check permissions, modes, uid, etc on the file, and "open" it. 672e1743d02SSøren Schmidt */ 673c8a79999SPeter Wemm error = exec_check_permissions(imgp); 674373d1a3fSAlan Cox if (error) 675c8a79999SPeter Wemm goto fail; 676e1743d02SSøren Schmidt 677c8a79999SPeter Wemm error = exec_map_first_page(imgp); 678373d1a3fSAlan Cox if (error) 679373d1a3fSAlan Cox goto fail; 680373d1a3fSAlan Cox 68125ead034SBrian Feldman /* 68225ead034SBrian Feldman * Also make certain that the interpreter stays the same, so set 683e6e370a7SJeff Roberson * its VV_TEXT flag, too. 68425ead034SBrian Feldman */ 685877d24acSKonstantin Belousov VOP_SET_TEXT(nd->ni_vp); 686e6e370a7SJeff Roberson 6878516dd18SPoul-Henning Kamp imgp->object = nd->ni_vp->v_object; 688e1743d02SSøren Schmidt 689d254af07SMatthew Dillon hdr = (const Elf_Ehdr *)imgp->image_header; 6903ebc1248SPeter Wemm if ((error = __elfN(check_header)(hdr)) != 0) 691e1743d02SSøren Schmidt goto fail; 692c33fe779SJohn Polstra if (hdr->e_type == ET_DYN) 693c33fe779SJohn Polstra rbase = *addr; 694c33fe779SJohn Polstra else if (hdr->e_type == ET_EXEC) 695c33fe779SJohn Polstra rbase = 0; 696c33fe779SJohn Polstra else { 697c33fe779SJohn Polstra error = ENOEXEC; 698c33fe779SJohn Polstra goto fail; 699c33fe779SJohn Polstra } 700e1743d02SSøren Schmidt 701c8a79999SPeter Wemm /* Only support headers that fit within first page for now */ 70252c24af7SPeter Wemm if ((hdr->e_phoff > PAGE_SIZE) || 703d19d5bf4STijl Coosemans (u_int)hdr->e_phentsize * hdr->e_phnum > PAGE_SIZE - hdr->e_phoff) { 704c8a79999SPeter Wemm error = ENOEXEC; 705e1743d02SSøren Schmidt goto fail; 706c8a79999SPeter Wemm } 707c8a79999SPeter Wemm 708d254af07SMatthew Dillon phdr = (const Elf_Phdr *)(imgp->image_header + hdr->e_phoff); 70993d1c728SKonstantin Belousov if (!aligned(phdr, Elf_Addr)) { 71093d1c728SKonstantin Belousov error = ENOEXEC; 71193d1c728SKonstantin Belousov goto fail; 71293d1c728SKonstantin Belousov } 713e1743d02SSøren Schmidt 714c33fe779SJohn Polstra for (i = 0, numsegs = 0; i < hdr->e_phnum; i++) { 7155b33842aSKonstantin Belousov if (phdr[i].p_type == PT_LOAD && phdr[i].p_memsz != 0) { 7165b33842aSKonstantin Belousov /* Loadable segment */ 717ed167eaaSKonstantin Belousov prot = __elfN(trans_prot)(phdr[i].p_flags); 718292177e6SAlan Cox error = __elfN(load_section)(imgp, phdr[i].p_offset, 71981f223caSJake Burkholder (caddr_t)(uintptr_t)phdr[i].p_vaddr + rbase, 720292177e6SAlan Cox phdr[i].p_memsz, phdr[i].p_filesz, prot, pagesize); 721292177e6SAlan Cox if (error != 0) 722e1743d02SSøren Schmidt goto fail; 723e1743d02SSøren Schmidt /* 724c33fe779SJohn Polstra * Establish the base address if this is the 725c33fe779SJohn Polstra * first segment. 726e1743d02SSøren Schmidt */ 727c33fe779SJohn Polstra if (numsegs == 0) 728ca0387efSJake Burkholder base_addr = trunc_page(phdr[i].p_vaddr + 729ca0387efSJake Burkholder rbase); 730c33fe779SJohn Polstra numsegs++; 731e1743d02SSøren Schmidt } 732e1743d02SSøren Schmidt } 733c33fe779SJohn Polstra *addr = base_addr; 734c33fe779SJohn Polstra *entry = (unsigned long)hdr->e_entry + rbase; 735e1743d02SSøren Schmidt 736e1743d02SSøren Schmidt fail: 737c8a79999SPeter Wemm if (imgp->firstpage) 738c8a79999SPeter Wemm exec_unmap_first_page(imgp); 7390b2ed1aeSJeff Roberson 740911c2be0SMark Peek if (nd->ni_vp) 741373d1a3fSAlan Cox vput(nd->ni_vp); 742911c2be0SMark Peek 743911c2be0SMark Peek free(tempdata, M_TEMP); 744e1743d02SSøren Schmidt 745a7cddfedSJake Burkholder return (error); 746e1743d02SSøren Schmidt } 747e1743d02SSøren Schmidt 748303b270bSEivind Eklund static int 7493ebc1248SPeter Wemm __CONCAT(exec_, __elfN(imgact))(struct image_params *imgp) 750e1743d02SSøren Schmidt { 7516c775eb6SKonstantin Belousov struct thread *td; 7526c775eb6SKonstantin Belousov const Elf_Ehdr *hdr; 75332c01de2SDmitry Chagin const Elf_Phdr *phdr; 754e5e6093bSAlan Cox Elf_Auxargs *elf_auxargs; 7555856e12eSJohn Dyson struct vmspace *vmspace; 7566c775eb6SKonstantin Belousov const char *err_str, *newinterp; 7576c775eb6SKonstantin Belousov char *interp, *interp_buf, *path; 758d1dbc694SJohn Polstra Elf_Brandinfo *brand_info; 7595fe3ed62SJake Burkholder struct sysentvec *sv; 7606c775eb6SKonstantin Belousov vm_prot_t prot; 7616c775eb6SKonstantin Belousov u_long text_size, data_size, total_size, text_addr, data_addr; 7626c775eb6SKonstantin Belousov u_long seg_size, seg_addr, addr, baddr, et_dyn_addr, entry, proghdr; 7636c775eb6SKonstantin Belousov int32_t osrel; 7646c775eb6SKonstantin Belousov int error, i, n, interp_name_len, have_interp; 7656c775eb6SKonstantin Belousov 7666c775eb6SKonstantin Belousov hdr = (const Elf_Ehdr *)imgp->image_header; 767e1743d02SSøren Schmidt 768e1743d02SSøren Schmidt /* 769e1743d02SSøren Schmidt * Do we have a valid ELF header ? 770900b28f9SMaxim Sobolev * 771900b28f9SMaxim Sobolev * Only allow ET_EXEC & ET_DYN here, reject ET_DYN later 772900b28f9SMaxim Sobolev * if particular brand doesn't support it. 773e1743d02SSøren Schmidt */ 774900b28f9SMaxim Sobolev if (__elfN(check_header)(hdr) != 0 || 775900b28f9SMaxim Sobolev (hdr->e_type != ET_EXEC && hdr->e_type != ET_DYN)) 776a7cddfedSJake Burkholder return (-1); 777e1743d02SSøren Schmidt 778e1743d02SSøren Schmidt /* 779e1743d02SSøren Schmidt * From here on down, we return an errno, not -1, as we've 780e1743d02SSøren Schmidt * detected an ELF file. 781e1743d02SSøren Schmidt */ 782e1743d02SSøren Schmidt 783e1743d02SSøren Schmidt if ((hdr->e_phoff > PAGE_SIZE) || 784d19d5bf4STijl Coosemans (u_int)hdr->e_phentsize * hdr->e_phnum > PAGE_SIZE - hdr->e_phoff) { 785c8a79999SPeter Wemm /* Only support headers in first page for now */ 7866b16d664SEd Maste uprintf("Program headers not in the first page\n"); 787a7cddfedSJake Burkholder return (ENOEXEC); 788e1743d02SSøren Schmidt } 78952c24af7SPeter Wemm phdr = (const Elf_Phdr *)(imgp->image_header + hdr->e_phoff); 7906b16d664SEd Maste if (!aligned(phdr, Elf_Addr)) { 7916b16d664SEd Maste uprintf("Unaligned program headers\n"); 79293d1c728SKonstantin Belousov return (ENOEXEC); 7936b16d664SEd Maste } 7946c775eb6SKonstantin Belousov 7956c775eb6SKonstantin Belousov n = error = 0; 7967564c4adSKonstantin Belousov baddr = 0; 7976c775eb6SKonstantin Belousov osrel = 0; 7986c775eb6SKonstantin Belousov text_size = data_size = total_size = text_addr = data_addr = 0; 7996c775eb6SKonstantin Belousov entry = proghdr = 0; 8006c775eb6SKonstantin Belousov interp_name_len = 0; 8016c775eb6SKonstantin Belousov err_str = newinterp = NULL; 8026c775eb6SKonstantin Belousov interp = interp_buf = NULL; 8036c775eb6SKonstantin Belousov td = curthread; 8046c775eb6SKonstantin Belousov 8055fe3ed62SJake Burkholder for (i = 0; i < hdr->e_phnum; i++) { 806291c06a1SKonstantin Belousov switch (phdr[i].p_type) { 807291c06a1SKonstantin Belousov case PT_LOAD: 8087564c4adSKonstantin Belousov if (n == 0) 8097564c4adSKonstantin Belousov baddr = phdr[i].p_vaddr; 8107564c4adSKonstantin Belousov n++; 811291c06a1SKonstantin Belousov break; 812291c06a1SKonstantin Belousov case PT_INTERP: 813e5e6093bSAlan Cox /* Path to interpreter */ 8146c775eb6SKonstantin Belousov if (phdr[i].p_filesz > MAXPATHLEN) { 8156b16d664SEd Maste uprintf("Invalid PT_INTERP\n"); 8166c775eb6SKonstantin Belousov error = ENOEXEC; 8176c775eb6SKonstantin Belousov goto ret; 8186b16d664SEd Maste } 819d3ee0a15SJonathan T. Looney if (interp != NULL) { 820d3ee0a15SJonathan T. Looney uprintf("Multiple PT_INTERP headers\n"); 821d3ee0a15SJonathan T. Looney error = ENOEXEC; 822d3ee0a15SJonathan T. Looney goto ret; 823d3ee0a15SJonathan T. Looney } 824d1ae5c83SKonstantin Belousov interp_name_len = phdr[i].p_filesz; 8256c775eb6SKonstantin Belousov if (phdr[i].p_offset > PAGE_SIZE || 8266c775eb6SKonstantin Belousov interp_name_len > PAGE_SIZE - phdr[i].p_offset) { 8276c775eb6SKonstantin Belousov VOP_UNLOCK(imgp->vp, 0); 8286c775eb6SKonstantin Belousov interp_buf = malloc(interp_name_len + 1, M_TEMP, 8296c775eb6SKonstantin Belousov M_WAITOK); 8306c775eb6SKonstantin Belousov vn_lock(imgp->vp, LK_EXCLUSIVE | LK_RETRY); 8316c775eb6SKonstantin Belousov error = vn_rdwr(UIO_READ, imgp->vp, interp_buf, 8326c775eb6SKonstantin Belousov interp_name_len, phdr[i].p_offset, 8336c775eb6SKonstantin Belousov UIO_SYSSPACE, IO_NODELOCKED, td->td_ucred, 8346c775eb6SKonstantin Belousov NOCRED, NULL, td); 8356c775eb6SKonstantin Belousov if (error != 0) { 8366c775eb6SKonstantin Belousov uprintf("i/o error PT_INTERP\n"); 8376c775eb6SKonstantin Belousov goto ret; 8386c775eb6SKonstantin Belousov } 8396c775eb6SKonstantin Belousov interp_buf[interp_name_len] = '\0'; 8406c775eb6SKonstantin Belousov interp = interp_buf; 8416c775eb6SKonstantin Belousov } else { 8426c775eb6SKonstantin Belousov interp = __DECONST(char *, imgp->image_header) + 8436c775eb6SKonstantin Belousov phdr[i].p_offset; 8446c775eb6SKonstantin Belousov } 845291c06a1SKonstantin Belousov break; 846291c06a1SKonstantin Belousov case PT_GNU_STACK: 847291c06a1SKonstantin Belousov if (__elfN(nxstack)) 848291c06a1SKonstantin Belousov imgp->stack_prot = 849291c06a1SKonstantin Belousov __elfN(trans_prot)(phdr[i].p_flags); 850316b3843SKonstantin Belousov imgp->stack_sz = phdr[i].p_memsz; 851291c06a1SKonstantin Belousov break; 8523ebc1248SPeter Wemm } 8533ebc1248SPeter Wemm } 8543ebc1248SPeter Wemm 855d1ae5c83SKonstantin Belousov brand_info = __elfN(get_brandinfo)(imgp, interp, interp_name_len, 856d1ae5c83SKonstantin Belousov &osrel); 8575fe3ed62SJake Burkholder if (brand_info == NULL) { 8585fe3ed62SJake Burkholder uprintf("ELF binary type \"%u\" not known.\n", 8595fe3ed62SJake Burkholder hdr->e_ident[EI_OSABI]); 8606c775eb6SKonstantin Belousov error = ENOEXEC; 8616c775eb6SKonstantin Belousov goto ret; 8623ebc1248SPeter Wemm } 863ab02d85fSKonstantin Belousov if (hdr->e_type == ET_DYN) { 8646b16d664SEd Maste if ((brand_info->flags & BI_CAN_EXEC_DYN) == 0) { 8656b16d664SEd Maste uprintf("Cannot execute shared object\n"); 8666c775eb6SKonstantin Belousov error = ENOEXEC; 8676c775eb6SKonstantin Belousov goto ret; 8686b16d664SEd Maste } 8697564c4adSKonstantin Belousov /* 8707564c4adSKonstantin Belousov * Honour the base load address from the dso if it is 8717564c4adSKonstantin Belousov * non-zero for some reason. 8727564c4adSKonstantin Belousov */ 8737564c4adSKonstantin Belousov if (baddr == 0) 874ab02d85fSKonstantin Belousov et_dyn_addr = ET_DYN_LOAD_ADDR; 8757564c4adSKonstantin Belousov else 8767564c4adSKonstantin Belousov et_dyn_addr = 0; 877ab02d85fSKonstantin Belousov } else 878ab02d85fSKonstantin Belousov et_dyn_addr = 0; 8795fe3ed62SJake Burkholder sv = brand_info->sysvec; 8809b68618dSPeter Wemm if (interp != NULL && brand_info->interp_newpath != NULL) 8814113f8d7SPeter Wemm newinterp = brand_info->interp_newpath; 8823ebc1248SPeter Wemm 88360bb3943SAlan Cox /* 88460bb3943SAlan Cox * Avoid a possible deadlock if the current address space is destroyed 88560bb3943SAlan Cox * and that address space maps the locked vnode. In the common case, 88660bb3943SAlan Cox * the locked vnode's v_usecount is decremented but remains greater 88760bb3943SAlan Cox * than zero. Consequently, the vnode lock is not needed by vrele(). 88860bb3943SAlan Cox * However, in cases where the vnode lock is external, such as nullfs, 88960bb3943SAlan Cox * v_usecount may become zero. 8901dfab802SAlan Cox * 8911dfab802SAlan Cox * The VV_TEXT flag prevents modifications to the executable while 8921dfab802SAlan Cox * the vnode is unlocked. 89360bb3943SAlan Cox */ 89422db15c0SAttilio Rao VOP_UNLOCK(imgp->vp, 0); 89560bb3943SAlan Cox 89689b57fcfSKonstantin Belousov error = exec_new_vmspace(imgp, sv); 89719059a13SJohn Baldwin imgp->proc->p_sysent = sv; 898e1743d02SSøren Schmidt 899cb05b60aSAttilio Rao vn_lock(imgp->vp, LK_EXCLUSIVE | LK_RETRY); 9006c775eb6SKonstantin Belousov if (error != 0) 9016c775eb6SKonstantin Belousov goto ret; 90260bb3943SAlan Cox 903e1743d02SSøren Schmidt for (i = 0; i < hdr->e_phnum; i++) { 904e1743d02SSøren Schmidt switch (phdr[i].p_type) { 905e1743d02SSøren Schmidt case PT_LOAD: /* Loadable segment */ 9065b33842aSKonstantin Belousov if (phdr[i].p_memsz == 0) 9075b33842aSKonstantin Belousov break; 908ed167eaaSKonstantin Belousov prot = __elfN(trans_prot)(phdr[i].p_flags); 909292177e6SAlan Cox error = __elfN(load_section)(imgp, phdr[i].p_offset, 910ab02d85fSKonstantin Belousov (caddr_t)(uintptr_t)phdr[i].p_vaddr + et_dyn_addr, 91181f223caSJake Burkholder phdr[i].p_memsz, phdr[i].p_filesz, prot, 912292177e6SAlan Cox sv->sv_pagesize); 913292177e6SAlan Cox if (error != 0) 9146c775eb6SKonstantin Belousov goto ret; 915e1743d02SSøren Schmidt 916cfaf7e60SDoug Rabson /* 917cfaf7e60SDoug Rabson * If this segment contains the program headers, 918cfaf7e60SDoug Rabson * remember their virtual address for the AT_PHDR 919cfaf7e60SDoug Rabson * aux entry. Static binaries don't usually include 920cfaf7e60SDoug Rabson * a PT_PHDR entry. 921cfaf7e60SDoug Rabson */ 922cfaf7e60SDoug Rabson if (phdr[i].p_offset == 0 && 923cfaf7e60SDoug Rabson hdr->e_phoff + hdr->e_phnum * hdr->e_phentsize 924cfaf7e60SDoug Rabson <= phdr[i].p_filesz) 925ab02d85fSKonstantin Belousov proghdr = phdr[i].p_vaddr + hdr->e_phoff + 926ab02d85fSKonstantin Belousov et_dyn_addr; 927cfaf7e60SDoug Rabson 928ab02d85fSKonstantin Belousov seg_addr = trunc_page(phdr[i].p_vaddr + et_dyn_addr); 929cac45152SMatthew Dillon seg_size = round_page(phdr[i].p_memsz + 930ab02d85fSKonstantin Belousov phdr[i].p_vaddr + et_dyn_addr - seg_addr); 931cac45152SMatthew Dillon 932e1743d02SSøren Schmidt /* 933920acedbSNathan Whitehorn * Make the largest executable segment the official 934920acedbSNathan Whitehorn * text segment and all others data. 93521c2d047SMatthew Dillon * 93621c2d047SMatthew Dillon * Note that obreak() assumes that data_addr + 93721c2d047SMatthew Dillon * data_size == end of data load area, and the ELF 93821c2d047SMatthew Dillon * file format expects segments to be sorted by 93921c2d047SMatthew Dillon * address. If multiple data segments exist, the 94021c2d047SMatthew Dillon * last one will be used. 941e1743d02SSøren Schmidt */ 942920acedbSNathan Whitehorn 943920acedbSNathan Whitehorn if (phdr[i].p_flags & PF_X && text_size < seg_size) { 9449782ecbaSPeter Wemm text_size = seg_size; 9459782ecbaSPeter Wemm text_addr = seg_addr; 9469782ecbaSPeter Wemm } else { 94721c2d047SMatthew Dillon data_size = seg_size; 948cac45152SMatthew Dillon data_addr = seg_addr; 949cac45152SMatthew Dillon } 95021c2d047SMatthew Dillon total_size += seg_size; 95196725dd0SAlexander Kabaev break; 95296725dd0SAlexander Kabaev case PT_PHDR: /* Program header table info */ 953ab02d85fSKonstantin Belousov proghdr = phdr[i].p_vaddr + et_dyn_addr; 95496725dd0SAlexander Kabaev break; 95596725dd0SAlexander Kabaev default: 95696725dd0SAlexander Kabaev break; 95796725dd0SAlexander Kabaev } 95896725dd0SAlexander Kabaev } 95996725dd0SAlexander Kabaev 96096725dd0SAlexander Kabaev if (data_addr == 0 && data_size == 0) { 96196725dd0SAlexander Kabaev data_addr = text_addr; 96296725dd0SAlexander Kabaev data_size = text_size; 96396725dd0SAlexander Kabaev } 964cac45152SMatthew Dillon 965920acedbSNathan Whitehorn entry = (u_long)hdr->e_entry + et_dyn_addr; 966920acedbSNathan Whitehorn 967cac45152SMatthew Dillon /* 968cac45152SMatthew Dillon * Check limits. It should be safe to check the 96996725dd0SAlexander Kabaev * limits after loading the segments since we do 97096725dd0SAlexander Kabaev * not actually fault in all the segments pages. 971cac45152SMatthew Dillon */ 97291d5354aSJohn Baldwin PROC_LOCK(imgp->proc); 973f6f6d240SMateusz Guzik if (data_size > lim_cur_proc(imgp->proc, RLIMIT_DATA)) 9746b16d664SEd Maste err_str = "Data segment size exceeds process limit"; 9756b16d664SEd Maste else if (text_size > maxtsiz) 9766b16d664SEd Maste err_str = "Text segment size exceeds system limit"; 977f6f6d240SMateusz Guzik else if (total_size > lim_cur_proc(imgp->proc, RLIMIT_VMEM)) 9786b16d664SEd Maste err_str = "Total segment size exceeds process limit"; 9796b16d664SEd Maste else if (racct_set(imgp->proc, RACCT_DATA, data_size) != 0) 9806b16d664SEd Maste err_str = "Data segment size exceeds resource limit"; 9816b16d664SEd Maste else if (racct_set(imgp->proc, RACCT_VMEM, total_size) != 0) 9826b16d664SEd Maste err_str = "Total segment size exceeds resource limit"; 9836b16d664SEd Maste if (err_str != NULL) { 98491d5354aSJohn Baldwin PROC_UNLOCK(imgp->proc); 9856b16d664SEd Maste uprintf("%s\n", err_str); 9866c775eb6SKonstantin Belousov error = ENOMEM; 9876c775eb6SKonstantin Belousov goto ret; 988cac45152SMatthew Dillon } 989e1743d02SSøren Schmidt 990292177e6SAlan Cox vmspace = imgp->proc->p_vmspace; 991e1743d02SSøren Schmidt vmspace->vm_tsize = text_size >> PAGE_SHIFT; 9927cd99438SBruce Evans vmspace->vm_taddr = (caddr_t)(uintptr_t)text_addr; 993e1743d02SSøren Schmidt vmspace->vm_dsize = data_size >> PAGE_SHIFT; 9947cd99438SBruce Evans vmspace->vm_daddr = (caddr_t)(uintptr_t)data_addr; 995e1743d02SSøren Schmidt 996c460ac3aSPeter Wemm /* 997c460ac3aSPeter Wemm * We load the dynamic linker where a userland call 998c460ac3aSPeter Wemm * to mmap(0, ...) would put it. The rationale behind this 999c460ac3aSPeter Wemm * calculation is that it leaves room for the heap to grow to 1000c460ac3aSPeter Wemm * its maximum allowed size. 1001c460ac3aSPeter Wemm */ 10026c775eb6SKonstantin Belousov addr = round_page((vm_offset_t)vmspace->vm_daddr + lim_max(td, 1003292177e6SAlan Cox RLIMIT_DATA)); 100491d5354aSJohn Baldwin PROC_UNLOCK(imgp->proc); 1005e1743d02SSøren Schmidt 1006ea5a2b2eSSøren Schmidt imgp->entry_addr = entry; 1007ea5a2b2eSSøren Schmidt 100860bb3943SAlan Cox if (interp != NULL) { 10096c775eb6SKonstantin Belousov have_interp = FALSE; 101022db15c0SAttilio Rao VOP_UNLOCK(imgp->vp, 0); 101160bb3943SAlan Cox if (brand_info->emul_path != NULL && 10129b68618dSPeter Wemm brand_info->emul_path[0] != '\0') { 1013a163d034SWarner Losh path = malloc(MAXPATHLEN, M_TEMP, M_WAITOK); 101460bb3943SAlan Cox snprintf(path, MAXPATHLEN, "%s%s", 101560bb3943SAlan Cox brand_info->emul_path, interp); 10169b68618dSPeter Wemm error = __elfN(load_file)(imgp->proc, path, &addr, 10179b68618dSPeter Wemm &imgp->entry_addr, sv->sv_pagesize); 1018911c2be0SMark Peek free(path, M_TEMP); 10199b68618dSPeter Wemm if (error == 0) 10204113f8d7SPeter Wemm have_interp = TRUE; 10219b68618dSPeter Wemm } 102218995077SKonstantin Belousov if (!have_interp && newinterp != NULL && 102318995077SKonstantin Belousov (brand_info->interp_path == NULL || 102418995077SKonstantin Belousov strcmp(interp, brand_info->interp_path) == 0)) { 10254113f8d7SPeter Wemm error = __elfN(load_file)(imgp->proc, newinterp, &addr, 10264113f8d7SPeter Wemm &imgp->entry_addr, sv->sv_pagesize); 1027387ad998SKonstantin Belousov if (error == 0) 10284113f8d7SPeter Wemm have_interp = TRUE; 10294113f8d7SPeter Wemm } 10304113f8d7SPeter Wemm if (!have_interp) { 10319b68618dSPeter Wemm error = __elfN(load_file)(imgp->proc, interp, &addr, 10329b68618dSPeter Wemm &imgp->entry_addr, sv->sv_pagesize); 103360bb3943SAlan Cox } 1034cb05b60aSAttilio Rao vn_lock(imgp->vp, LK_EXCLUSIVE | LK_RETRY); 10359b68618dSPeter Wemm if (error != 0) { 1036d943fa35SKonstantin Belousov uprintf("ELF interpreter %s not found, error %d\n", 1037d943fa35SKonstantin Belousov interp, error); 10386c775eb6SKonstantin Belousov goto ret; 1039e1743d02SSøren Schmidt } 104095c807cfSRobert Watson } else 10417564c4adSKonstantin Belousov addr = et_dyn_addr; 1042ea5a2b2eSSøren Schmidt 1043e1743d02SSøren Schmidt /* 1044e1743d02SSøren Schmidt * Construct auxargs table (used by the fixup routine) 1045e1743d02SSøren Schmidt */ 1046a163d034SWarner Losh elf_auxargs = malloc(sizeof(Elf_Auxargs), M_TEMP, M_WAITOK); 1047e1743d02SSøren Schmidt elf_auxargs->execfd = -1; 1048e1743d02SSøren Schmidt elf_auxargs->phdr = proghdr; 1049e1743d02SSøren Schmidt elf_auxargs->phent = hdr->e_phentsize; 1050e1743d02SSøren Schmidt elf_auxargs->phnum = hdr->e_phnum; 1051e1743d02SSøren Schmidt elf_auxargs->pagesz = PAGE_SIZE; 1052e1743d02SSøren Schmidt elf_auxargs->base = addr; 1053e1743d02SSøren Schmidt elf_auxargs->flags = 0; 1054e1743d02SSøren Schmidt elf_auxargs->entry = entry; 1055d36eec69SWarner Losh elf_auxargs->hdr_eflags = hdr->e_flags; 1056e1743d02SSøren Schmidt 1057e1743d02SSøren Schmidt imgp->auxargs = elf_auxargs; 1058e1743d02SSøren Schmidt imgp->interpreted = 0; 1059a0ea661fSNathan Whitehorn imgp->reloc_base = addr; 106032c01de2SDmitry Chagin imgp->proc->p_osrel = osrel; 1061f231de47SKonstantin Belousov 10626c775eb6SKonstantin Belousov ret: 10636c775eb6SKonstantin Belousov free(interp_buf, M_TEMP); 1064a7cddfedSJake Burkholder return (error); 1065e1743d02SSøren Schmidt } 1066e1743d02SSøren Schmidt 1067a360a43dSJake Burkholder #define suword __CONCAT(suword, __ELF_WORD_SIZE) 10683ebc1248SPeter Wemm 10693ebc1248SPeter Wemm int 10703ebc1248SPeter Wemm __elfN(freebsd_fixup)(register_t **stack_base, struct image_params *imgp) 1071e1743d02SSøren Schmidt { 1072ecbb00a2SDoug Rabson Elf_Auxargs *args = (Elf_Auxargs *)imgp->auxargs; 1073a360a43dSJake Burkholder Elf_Addr *base; 1074a360a43dSJake Burkholder Elf_Addr *pos; 1075e1743d02SSøren Schmidt 1076a360a43dSJake Burkholder base = (Elf_Addr *)*stack_base; 1077610ecfe0SMaxim Sobolev pos = base + (imgp->args->argc + imgp->args->envc + 2); 1078e1743d02SSøren Schmidt 107935c2a5a8SWarner Losh if (args->execfd != -1) 1080e1743d02SSøren Schmidt AUXARGS_ENTRY(pos, AT_EXECFD, args->execfd); 1081e1743d02SSøren Schmidt AUXARGS_ENTRY(pos, AT_PHDR, args->phdr); 1082e1743d02SSøren Schmidt AUXARGS_ENTRY(pos, AT_PHENT, args->phent); 1083e1743d02SSøren Schmidt AUXARGS_ENTRY(pos, AT_PHNUM, args->phnum); 1084e1743d02SSøren Schmidt AUXARGS_ENTRY(pos, AT_PAGESZ, args->pagesz); 1085e1743d02SSøren Schmidt AUXARGS_ENTRY(pos, AT_FLAGS, args->flags); 1086e1743d02SSøren Schmidt AUXARGS_ENTRY(pos, AT_ENTRY, args->entry); 1087e1743d02SSøren Schmidt AUXARGS_ENTRY(pos, AT_BASE, args->base); 1088ee960398SWarner Losh #ifdef AT_EHDRFLAGS 1089d36eec69SWarner Losh AUXARGS_ENTRY(pos, AT_EHDRFLAGS, args->hdr_eflags); 1090d36eec69SWarner Losh #endif 10913ff06357SKonstantin Belousov if (imgp->execpathp != 0) 10923ff06357SKonstantin Belousov AUXARGS_ENTRY(pos, AT_EXECPATH, imgp->execpathp); 1093b96bd95bSIan Lepore AUXARGS_ENTRY(pos, AT_OSRELDATE, 1094b96bd95bSIan Lepore imgp->proc->p_ucred->cr_prison->pr_osreldate); 1095ee235befSKonstantin Belousov if (imgp->canary != 0) { 1096ee235befSKonstantin Belousov AUXARGS_ENTRY(pos, AT_CANARY, imgp->canary); 1097ee235befSKonstantin Belousov AUXARGS_ENTRY(pos, AT_CANARYLEN, imgp->canarylen); 1098ee235befSKonstantin Belousov } 1099ee235befSKonstantin Belousov AUXARGS_ENTRY(pos, AT_NCPUS, mp_ncpus); 1100ee235befSKonstantin Belousov if (imgp->pagesizes != 0) { 1101ee235befSKonstantin Belousov AUXARGS_ENTRY(pos, AT_PAGESIZES, imgp->pagesizes); 1102ee235befSKonstantin Belousov AUXARGS_ENTRY(pos, AT_PAGESIZESLEN, imgp->pagesizeslen); 1103ee235befSKonstantin Belousov } 1104aea81038SKonstantin Belousov if (imgp->sysent->sv_timekeep_base != 0) { 1105aea81038SKonstantin Belousov AUXARGS_ENTRY(pos, AT_TIMEKEEP, 1106aea81038SKonstantin Belousov imgp->sysent->sv_timekeep_base); 1107aea81038SKonstantin Belousov } 110826d8f3e1SKonstantin Belousov AUXARGS_ENTRY(pos, AT_STACKPROT, imgp->sysent->sv_shared_page_obj 110926d8f3e1SKonstantin Belousov != NULL && imgp->stack_prot != 0 ? imgp->stack_prot : 111026d8f3e1SKonstantin Belousov imgp->sysent->sv_stackprot); 1111e1743d02SSøren Schmidt AUXARGS_ENTRY(pos, AT_NULL, 0); 1112e1743d02SSøren Schmidt 1113e1743d02SSøren Schmidt free(imgp->auxargs, M_TEMP); 1114e1743d02SSøren Schmidt imgp->auxargs = NULL; 1115e1743d02SSøren Schmidt 11163ebc1248SPeter Wemm base--; 1117610ecfe0SMaxim Sobolev suword(base, (long)imgp->args->argc); 11183ebc1248SPeter Wemm *stack_base = (register_t *)base; 1119a7cddfedSJake Burkholder return (0); 1120e1743d02SSøren Schmidt } 1121e1743d02SSøren Schmidt 1122e1743d02SSøren Schmidt /* 11238c64af4fSJohn Polstra * Code for generating ELF core dumps. 11248c64af4fSJohn Polstra */ 11258c64af4fSJohn Polstra 11264d77a549SAlfred Perlstein typedef void (*segment_callback)(vm_map_entry_t, void *); 11270ff27d31SJohn Polstra 11280ff27d31SJohn Polstra /* Closure for cb_put_phdr(). */ 11290ff27d31SJohn Polstra struct phdr_closure { 11300ff27d31SJohn Polstra Elf_Phdr *phdr; /* Program header to fill in */ 11310ff27d31SJohn Polstra Elf_Off offset; /* Offset of segment in core file */ 11320ff27d31SJohn Polstra }; 11330ff27d31SJohn Polstra 11340ff27d31SJohn Polstra /* Closure for cb_size_segment(). */ 11350ff27d31SJohn Polstra struct sseg_closure { 11360ff27d31SJohn Polstra int count; /* Count of writable segments. */ 11370ff27d31SJohn Polstra size_t size; /* Total size of all writable segments. */ 11380ff27d31SJohn Polstra }; 11390ff27d31SJohn Polstra 1140bd390213SMikolaj Golub typedef void (*outfunc_t)(void *, struct sbuf *, size_t *); 1141bd390213SMikolaj Golub 1142bd390213SMikolaj Golub struct note_info { 1143bd390213SMikolaj Golub int type; /* Note type. */ 1144bd390213SMikolaj Golub outfunc_t outfunc; /* Output function. */ 1145bd390213SMikolaj Golub void *outarg; /* Argument for the output function. */ 1146bd390213SMikolaj Golub size_t outsize; /* Output size. */ 1147bd390213SMikolaj Golub TAILQ_ENTRY(note_info) link; /* Link to the next note info. */ 1148bd390213SMikolaj Golub }; 1149bd390213SMikolaj Golub 1150bd390213SMikolaj Golub TAILQ_HEAD(note_info_list, note_info); 1151bd390213SMikolaj Golub 1152aa14e9b7SMark Johnston /* Coredump output parameters. */ 1153aa14e9b7SMark Johnston struct coredump_params { 1154aa14e9b7SMark Johnston off_t offset; 1155aa14e9b7SMark Johnston struct ucred *active_cred; 1156aa14e9b7SMark Johnston struct ucred *file_cred; 1157aa14e9b7SMark Johnston struct thread *td; 1158aa14e9b7SMark Johnston struct vnode *vp; 1159aa14e9b7SMark Johnston struct gzio_stream *gzs; 1160aa14e9b7SMark Johnston }; 1161aa14e9b7SMark Johnston 11624d77a549SAlfred Perlstein static void cb_put_phdr(vm_map_entry_t, void *); 11634d77a549SAlfred Perlstein static void cb_size_segment(vm_map_entry_t, void *); 1164aa14e9b7SMark Johnston static int core_write(struct coredump_params *, void *, size_t, off_t, 1165aa14e9b7SMark Johnston enum uio_seg); 1166247aba24SMarcel Moolenaar static void each_writable_segment(struct thread *, segment_callback, void *); 1167aa14e9b7SMark Johnston static int __elfN(corehdr)(struct coredump_params *, int, void *, size_t, 1168aa14e9b7SMark Johnston struct note_info_list *, size_t); 1169bd390213SMikolaj Golub static void __elfN(prepare_notes)(struct thread *, struct note_info_list *, 1170bd390213SMikolaj Golub size_t *); 1171bd390213SMikolaj Golub static void __elfN(puthdr)(struct thread *, void *, size_t, int, size_t); 1172bd390213SMikolaj Golub static void __elfN(putnote)(struct note_info *, struct sbuf *); 1173bd390213SMikolaj Golub static size_t register_note(struct note_info_list *, int, outfunc_t, void *); 1174bd390213SMikolaj Golub static int sbuf_drain_core_output(void *, const char *, int); 1175f1fca82eSMikolaj Golub static int sbuf_drain_count(void *arg, const char *data, int len); 1176bd390213SMikolaj Golub 1177bd390213SMikolaj Golub static void __elfN(note_fpregset)(void *, struct sbuf *, size_t *); 1178bd390213SMikolaj Golub static void __elfN(note_prpsinfo)(void *, struct sbuf *, size_t *); 1179bd390213SMikolaj Golub static void __elfN(note_prstatus)(void *, struct sbuf *, size_t *); 1180bd390213SMikolaj Golub static void __elfN(note_threadmd)(void *, struct sbuf *, size_t *); 1181bd390213SMikolaj Golub static void __elfN(note_thrmisc)(void *, struct sbuf *, size_t *); 1182f1fca82eSMikolaj Golub static void __elfN(note_procstat_auxv)(void *, struct sbuf *, size_t *); 1183f1fca82eSMikolaj Golub static void __elfN(note_procstat_proc)(void *, struct sbuf *, size_t *); 1184f1fca82eSMikolaj Golub static void __elfN(note_procstat_psstrings)(void *, struct sbuf *, size_t *); 1185f1fca82eSMikolaj Golub static void note_procstat_files(void *, struct sbuf *, size_t *); 1186f1fca82eSMikolaj Golub static void note_procstat_groups(void *, struct sbuf *, size_t *); 1187f1fca82eSMikolaj Golub static void note_procstat_osrel(void *, struct sbuf *, size_t *); 1188f1fca82eSMikolaj Golub static void note_procstat_rlimit(void *, struct sbuf *, size_t *); 1189f1fca82eSMikolaj Golub static void note_procstat_umask(void *, struct sbuf *, size_t *); 1190f1fca82eSMikolaj Golub static void note_procstat_vmmap(void *, struct sbuf *, size_t *); 11918c64af4fSJohn Polstra 1192aa14e9b7SMark Johnston #ifdef GZIO 1193e7228204SAlfred Perlstein extern int compress_user_cores_gzlevel; 1194e7228204SAlfred Perlstein 1195aa14e9b7SMark Johnston /* 1196aa14e9b7SMark Johnston * Write out a core segment to the compression stream. 1197aa14e9b7SMark Johnston */ 1198e7228204SAlfred Perlstein static int 1199aa14e9b7SMark Johnston compress_chunk(struct coredump_params *p, char *base, char *buf, u_int len) 1200aa14e9b7SMark Johnston { 1201aa14e9b7SMark Johnston u_int chunk_len; 1202e7228204SAlfred Perlstein int error; 1203aa14e9b7SMark Johnston 1204aa14e9b7SMark Johnston while (len > 0) { 1205aa14e9b7SMark Johnston chunk_len = MIN(len, CORE_BUF_SIZE); 1206aa14e9b7SMark Johnston copyin(base, buf, chunk_len); 1207aa14e9b7SMark Johnston error = gzio_write(p->gzs, buf, chunk_len); 1208aa14e9b7SMark Johnston if (error != 0) 1209aa14e9b7SMark Johnston break; 1210aa14e9b7SMark Johnston base += chunk_len; 1211aa14e9b7SMark Johnston len -= chunk_len; 1212e7228204SAlfred Perlstein } 1213e7228204SAlfred Perlstein return (error); 1214e7228204SAlfred Perlstein } 1215e7228204SAlfred Perlstein 1216aa14e9b7SMark Johnston static int 1217aa14e9b7SMark Johnston core_gz_write(void *base, size_t len, off_t offset, void *arg) 1218aa14e9b7SMark Johnston { 1219aa14e9b7SMark Johnston 1220aa14e9b7SMark Johnston return (core_write((struct coredump_params *)arg, base, len, offset, 1221aa14e9b7SMark Johnston UIO_SYSSPACE)); 1222aa14e9b7SMark Johnston } 1223aa14e9b7SMark Johnston #endif /* GZIO */ 1224aa14e9b7SMark Johnston 1225aa14e9b7SMark Johnston static int 1226aa14e9b7SMark Johnston core_write(struct coredump_params *p, void *base, size_t len, off_t offset, 1227aa14e9b7SMark Johnston enum uio_seg seg) 1228aa14e9b7SMark Johnston { 1229aa14e9b7SMark Johnston 1230aa14e9b7SMark Johnston return (vn_rdwr_inchunks(UIO_WRITE, p->vp, base, len, offset, 1231aa14e9b7SMark Johnston seg, IO_UNIT | IO_DIRECT | IO_RANGELOCKED, 1232aa14e9b7SMark Johnston p->active_cred, p->file_cred, NULL, p->td)); 1233aa14e9b7SMark Johnston } 1234aa14e9b7SMark Johnston 1235aa14e9b7SMark Johnston static int 1236aa14e9b7SMark Johnston core_output(void *base, size_t len, off_t offset, struct coredump_params *p, 1237aa14e9b7SMark Johnston void *tmpbuf) 1238aa14e9b7SMark Johnston { 1239aa14e9b7SMark Johnston 1240aa14e9b7SMark Johnston #ifdef GZIO 1241aa14e9b7SMark Johnston if (p->gzs != NULL) 1242aa14e9b7SMark Johnston return (compress_chunk(p, base, tmpbuf, len)); 1243bd390213SMikolaj Golub #endif 1244aa14e9b7SMark Johnston return (core_write(p, base, len, offset, UIO_USERSPACE)); 1245aa14e9b7SMark Johnston } 1246bd390213SMikolaj Golub 1247bd390213SMikolaj Golub /* 1248bd390213SMikolaj Golub * Drain into a core file. 1249bd390213SMikolaj Golub */ 1250bd390213SMikolaj Golub static int 1251bd390213SMikolaj Golub sbuf_drain_core_output(void *arg, const char *data, int len) 1252bd390213SMikolaj Golub { 1253aa14e9b7SMark Johnston struct coredump_params *p; 1254f1fca82eSMikolaj Golub int error, locked; 1255bd390213SMikolaj Golub 1256aa14e9b7SMark Johnston p = (struct coredump_params *)arg; 1257f1fca82eSMikolaj Golub 1258f1fca82eSMikolaj Golub /* 1259f1fca82eSMikolaj Golub * Some kern_proc out routines that print to this sbuf may 1260f1fca82eSMikolaj Golub * call us with the process lock held. Draining with the 1261f1fca82eSMikolaj Golub * non-sleepable lock held is unsafe. The lock is needed for 1262f1fca82eSMikolaj Golub * those routines when dumping a live process. In our case we 1263f1fca82eSMikolaj Golub * can safely release the lock before draining and acquire 1264f1fca82eSMikolaj Golub * again after. 1265f1fca82eSMikolaj Golub */ 1266f1fca82eSMikolaj Golub locked = PROC_LOCKED(p->td->td_proc); 1267f1fca82eSMikolaj Golub if (locked) 1268f1fca82eSMikolaj Golub PROC_UNLOCK(p->td->td_proc); 1269aa14e9b7SMark Johnston #ifdef GZIO 1270aa14e9b7SMark Johnston if (p->gzs != NULL) 1271aa14e9b7SMark Johnston error = gzio_write(p->gzs, __DECONST(char *, data), len); 1272bd390213SMikolaj Golub else 1273bd390213SMikolaj Golub #endif 1274aa14e9b7SMark Johnston error = core_write(p, __DECONST(void *, data), len, p->offset, 1275aa14e9b7SMark Johnston UIO_SYSSPACE); 1276f1fca82eSMikolaj Golub if (locked) 1277f1fca82eSMikolaj Golub PROC_LOCK(p->td->td_proc); 1278bd390213SMikolaj Golub if (error != 0) 1279bd390213SMikolaj Golub return (-error); 1280bd390213SMikolaj Golub p->offset += len; 1281bd390213SMikolaj Golub return (len); 1282bd390213SMikolaj Golub } 1283bd390213SMikolaj Golub 1284f1fca82eSMikolaj Golub /* 1285f1fca82eSMikolaj Golub * Drain into a counter. 1286f1fca82eSMikolaj Golub */ 1287f1fca82eSMikolaj Golub static int 1288f1fca82eSMikolaj Golub sbuf_drain_count(void *arg, const char *data __unused, int len) 1289f1fca82eSMikolaj Golub { 1290f1fca82eSMikolaj Golub size_t *sizep; 1291f1fca82eSMikolaj Golub 1292f1fca82eSMikolaj Golub sizep = (size_t *)arg; 1293f1fca82eSMikolaj Golub *sizep += len; 1294f1fca82eSMikolaj Golub return (len); 1295f1fca82eSMikolaj Golub } 1296f1fca82eSMikolaj Golub 12978c64af4fSJohn Polstra int 1298e7228204SAlfred Perlstein __elfN(coredump)(struct thread *td, struct vnode *vp, off_t limit, int flags) 1299fca666a1SJulian Elischer { 1300247aba24SMarcel Moolenaar struct ucred *cred = td->td_ucred; 1301fca666a1SJulian Elischer int error = 0; 13020ff27d31SJohn Polstra struct sseg_closure seginfo; 1303bd390213SMikolaj Golub struct note_info_list notelst; 1304aa14e9b7SMark Johnston struct coredump_params params; 1305bd390213SMikolaj Golub struct note_info *ninfo; 1306aa14e9b7SMark Johnston void *hdr, *tmpbuf; 1307bd390213SMikolaj Golub size_t hdrsize, notesz, coresize; 13089a12e282SEnji Cooper #ifdef GZIO 1309aa14e9b7SMark Johnston boolean_t compress; 13108c64af4fSJohn Polstra 1311aa14e9b7SMark Johnston compress = (flags & IMGACT_CORE_COMPRESS) != 0; 13129a12e282SEnji Cooper #endif 1313e7228204SAlfred Perlstein hdr = NULL; 131402d131adSMark Johnston tmpbuf = NULL; 1315bd390213SMikolaj Golub TAILQ_INIT(¬elst); 1316e7228204SAlfred Perlstein 13170ff27d31SJohn Polstra /* Size the program segments. */ 13180ff27d31SJohn Polstra seginfo.count = 0; 13190ff27d31SJohn Polstra seginfo.size = 0; 1320247aba24SMarcel Moolenaar each_writable_segment(td, cb_size_segment, &seginfo); 13210ff27d31SJohn Polstra 13220ff27d31SJohn Polstra /* 1323bd390213SMikolaj Golub * Collect info about the core file header area. 13240ff27d31SJohn Polstra */ 1325bd390213SMikolaj Golub hdrsize = sizeof(Elf_Ehdr) + sizeof(Elf_Phdr) * (1 + seginfo.count); 1326bd390213SMikolaj Golub __elfN(prepare_notes)(td, ¬elst, ¬esz); 1327bd390213SMikolaj Golub coresize = round_page(hdrsize + notesz) + seginfo.size; 13280ff27d31SJohn Polstra 132902d131adSMark Johnston /* Set up core dump parameters. */ 133002d131adSMark Johnston params.offset = 0; 133102d131adSMark Johnston params.active_cred = cred; 133202d131adSMark Johnston params.file_cred = NOCRED; 133302d131adSMark Johnston params.td = td; 133402d131adSMark Johnston params.vp = vp; 133502d131adSMark Johnston params.gzs = NULL; 133602d131adSMark Johnston 1337afcc55f3SEdward Tomasz Napierala #ifdef RACCT 13384b5c9cf6SEdward Tomasz Napierala if (racct_enable) { 13391ba5ad42SEdward Tomasz Napierala PROC_LOCK(td->td_proc); 1340bd390213SMikolaj Golub error = racct_add(td->td_proc, RACCT_CORE, coresize); 13411ba5ad42SEdward Tomasz Napierala PROC_UNLOCK(td->td_proc); 13421ba5ad42SEdward Tomasz Napierala if (error != 0) { 13431ba5ad42SEdward Tomasz Napierala error = EFAULT; 13441ba5ad42SEdward Tomasz Napierala goto done; 13451ba5ad42SEdward Tomasz Napierala } 13464b5c9cf6SEdward Tomasz Napierala } 1347afcc55f3SEdward Tomasz Napierala #endif 1348bd390213SMikolaj Golub if (coresize >= limit) { 1349fba6b1afSAlfred Perlstein error = EFAULT; 1350fba6b1afSAlfred Perlstein goto done; 1351fba6b1afSAlfred Perlstein } 13520ff27d31SJohn Polstra 1353aa14e9b7SMark Johnston #ifdef GZIO 1354aa14e9b7SMark Johnston /* Create a compression stream if necessary. */ 1355aa14e9b7SMark Johnston if (compress) { 1356aa14e9b7SMark Johnston params.gzs = gzio_init(core_gz_write, GZIO_DEFLATE, 1357aa14e9b7SMark Johnston CORE_BUF_SIZE, compress_user_cores_gzlevel, ¶ms); 1358aa14e9b7SMark Johnston if (params.gzs == NULL) { 1359aa14e9b7SMark Johnston error = EFAULT; 1360aa14e9b7SMark Johnston goto done; 1361aa14e9b7SMark Johnston } 1362aa14e9b7SMark Johnston tmpbuf = malloc(CORE_BUF_SIZE, M_TEMP, M_WAITOK | M_ZERO); 1363aa14e9b7SMark Johnston } 1364aa14e9b7SMark Johnston #endif 1365aa14e9b7SMark Johnston 13660ff27d31SJohn Polstra /* 13670ff27d31SJohn Polstra * Allocate memory for building the header, fill it up, 1368bd390213SMikolaj Golub * and write it out following the notes. 13690ff27d31SJohn Polstra */ 1370a163d034SWarner Losh hdr = malloc(hdrsize, M_TEMP, M_WAITOK); 1371aa14e9b7SMark Johnston error = __elfN(corehdr)(¶ms, seginfo.count, hdr, hdrsize, ¬elst, 1372aa14e9b7SMark Johnston notesz); 13730ff27d31SJohn Polstra 13740ff27d31SJohn Polstra /* Write the contents of all of the writable segments. */ 13750ff27d31SJohn Polstra if (error == 0) { 13760ff27d31SJohn Polstra Elf_Phdr *php; 13772b471bc6STim J. Robbins off_t offset; 13780ff27d31SJohn Polstra int i; 13790ff27d31SJohn Polstra 13800ff27d31SJohn Polstra php = (Elf_Phdr *)((char *)hdr + sizeof(Elf_Ehdr)) + 1; 1381bd390213SMikolaj Golub offset = round_page(hdrsize + notesz); 13820ff27d31SJohn Polstra for (i = 0; i < seginfo.count; i++) { 1383aa14e9b7SMark Johnston error = core_output((caddr_t)(uintptr_t)php->p_vaddr, 1384aa14e9b7SMark Johnston php->p_filesz, offset, ¶ms, tmpbuf); 13850ff27d31SJohn Polstra if (error != 0) 13862b471bc6STim J. Robbins break; 13870ff27d31SJohn Polstra offset += php->p_filesz; 13880ff27d31SJohn Polstra php++; 13890ff27d31SJohn Polstra } 1390aa14e9b7SMark Johnston #ifdef GZIO 1391aa14e9b7SMark Johnston if (error == 0 && compress) 1392aa14e9b7SMark Johnston error = gzio_flush(params.gzs); 1393aa14e9b7SMark Johnston #endif 13940ff27d31SJohn Polstra } 1395e7228204SAlfred Perlstein if (error) { 1396e7228204SAlfred Perlstein log(LOG_WARNING, 1397e7228204SAlfred Perlstein "Failed to write core file for process %s (error %d)\n", 1398e7228204SAlfred Perlstein curproc->p_comm, error); 1399e7228204SAlfred Perlstein } 1400e7228204SAlfred Perlstein 1401e7228204SAlfred Perlstein done: 1402aa14e9b7SMark Johnston #ifdef GZIO 1403aa14e9b7SMark Johnston if (compress) { 1404aa14e9b7SMark Johnston free(tmpbuf, M_TEMP); 140502d131adSMark Johnston if (params.gzs != NULL) 1406aa14e9b7SMark Johnston gzio_fini(params.gzs); 1407aa14e9b7SMark Johnston } 14088b325009SAlfred Perlstein #endif 1409bd390213SMikolaj Golub while ((ninfo = TAILQ_FIRST(¬elst)) != NULL) { 1410bd390213SMikolaj Golub TAILQ_REMOVE(¬elst, ninfo, link); 1411bd390213SMikolaj Golub free(ninfo, M_TEMP); 1412bd390213SMikolaj Golub } 1413bd390213SMikolaj Golub if (hdr != NULL) 14140ff27d31SJohn Polstra free(hdr, M_TEMP); 14150ff27d31SJohn Polstra 1416a7cddfedSJake Burkholder return (error); 14178c64af4fSJohn Polstra } 14188c64af4fSJohn Polstra 14190ff27d31SJohn Polstra /* 14200ff27d31SJohn Polstra * A callback for each_writable_segment() to write out the segment's 14210ff27d31SJohn Polstra * program header entry. 14220ff27d31SJohn Polstra */ 14230ff27d31SJohn Polstra static void 14240ff27d31SJohn Polstra cb_put_phdr(entry, closure) 14250ff27d31SJohn Polstra vm_map_entry_t entry; 14260ff27d31SJohn Polstra void *closure; 14270ff27d31SJohn Polstra { 14280ff27d31SJohn Polstra struct phdr_closure *phc = (struct phdr_closure *)closure; 14290ff27d31SJohn Polstra Elf_Phdr *phdr = phc->phdr; 14300ff27d31SJohn Polstra 14310ff27d31SJohn Polstra phc->offset = round_page(phc->offset); 14320ff27d31SJohn Polstra 14330ff27d31SJohn Polstra phdr->p_type = PT_LOAD; 14340ff27d31SJohn Polstra phdr->p_offset = phc->offset; 14350ff27d31SJohn Polstra phdr->p_vaddr = entry->start; 14360ff27d31SJohn Polstra phdr->p_paddr = 0; 14370ff27d31SJohn Polstra phdr->p_filesz = phdr->p_memsz = entry->end - entry->start; 14380ff27d31SJohn Polstra phdr->p_align = PAGE_SIZE; 1439ed167eaaSKonstantin Belousov phdr->p_flags = __elfN(untrans_prot)(entry->protection); 14400ff27d31SJohn Polstra 14410ff27d31SJohn Polstra phc->offset += phdr->p_filesz; 14420ff27d31SJohn Polstra phc->phdr++; 14430ff27d31SJohn Polstra } 14440ff27d31SJohn Polstra 14450ff27d31SJohn Polstra /* 14460ff27d31SJohn Polstra * A callback for each_writable_segment() to gather information about 14470ff27d31SJohn Polstra * the number of segments and their total size. 14480ff27d31SJohn Polstra */ 14490ff27d31SJohn Polstra static void 14500ff27d31SJohn Polstra cb_size_segment(entry, closure) 14510ff27d31SJohn Polstra vm_map_entry_t entry; 14520ff27d31SJohn Polstra void *closure; 14530ff27d31SJohn Polstra { 14540ff27d31SJohn Polstra struct sseg_closure *ssc = (struct sseg_closure *)closure; 14550ff27d31SJohn Polstra 14560ff27d31SJohn Polstra ssc->count++; 14570ff27d31SJohn Polstra ssc->size += entry->end - entry->start; 14580ff27d31SJohn Polstra } 14590ff27d31SJohn Polstra 14600ff27d31SJohn Polstra /* 14610ff27d31SJohn Polstra * For each writable segment in the process's memory map, call the given 14620ff27d31SJohn Polstra * function with a pointer to the map entry and some arbitrary 14630ff27d31SJohn Polstra * caller-supplied data. 14640ff27d31SJohn Polstra */ 14650ff27d31SJohn Polstra static void 1466247aba24SMarcel Moolenaar each_writable_segment(td, func, closure) 1467247aba24SMarcel Moolenaar struct thread *td; 14680ff27d31SJohn Polstra segment_callback func; 14690ff27d31SJohn Polstra void *closure; 14700ff27d31SJohn Polstra { 1471247aba24SMarcel Moolenaar struct proc *p = td->td_proc; 14720ff27d31SJohn Polstra vm_map_t map = &p->p_vmspace->vm_map; 14730ff27d31SJohn Polstra vm_map_entry_t entry; 1474976a87a2SAlan Cox vm_object_t backing_object, object; 1475976a87a2SAlan Cox boolean_t ignore_entry; 14760ff27d31SJohn Polstra 1477976a87a2SAlan Cox vm_map_lock_read(map); 14780ff27d31SJohn Polstra for (entry = map->header.next; entry != &map->header; 14790ff27d31SJohn Polstra entry = entry->next) { 1480fa7dd9c5SMatthew Dillon /* 1481fa7dd9c5SMatthew Dillon * Don't dump inaccessible mappings, deal with legacy 1482fa7dd9c5SMatthew Dillon * coredump mode. 1483fa7dd9c5SMatthew Dillon * 1484fa7dd9c5SMatthew Dillon * Note that read-only segments related to the elf binary 1485fa7dd9c5SMatthew Dillon * are marked MAP_ENTRY_NOCOREDUMP now so we no longer 1486fa7dd9c5SMatthew Dillon * need to arbitrarily ignore such segments. 1487fa7dd9c5SMatthew Dillon */ 1488fa7dd9c5SMatthew Dillon if (elf_legacy_coredump) { 1489fa7dd9c5SMatthew Dillon if ((entry->protection & VM_PROT_RW) != VM_PROT_RW) 14900ff27d31SJohn Polstra continue; 1491fa7dd9c5SMatthew Dillon } else { 1492fa7dd9c5SMatthew Dillon if ((entry->protection & VM_PROT_ALL) == 0) 1493fa7dd9c5SMatthew Dillon continue; 1494fa7dd9c5SMatthew Dillon } 14950ff27d31SJohn Polstra 14969730a5daSPaul Saab /* 1497fa7dd9c5SMatthew Dillon * Dont include memory segment in the coredump if 1498fa7dd9c5SMatthew Dillon * MAP_NOCORE is set in mmap(2) or MADV_NOCORE in 1499fa7dd9c5SMatthew Dillon * madvise(2). Do not dump submaps (i.e. parts of the 1500fa7dd9c5SMatthew Dillon * kernel map). 15019730a5daSPaul Saab */ 1502fa7dd9c5SMatthew Dillon if (entry->eflags & (MAP_ENTRY_NOCOREDUMP|MAP_ENTRY_IS_SUB_MAP)) 15039730a5daSPaul Saab continue; 15049730a5daSPaul Saab 1505976a87a2SAlan Cox if ((object = entry->object.vm_object) == NULL) 15060ff27d31SJohn Polstra continue; 15070ff27d31SJohn Polstra 15080ff27d31SJohn Polstra /* Ignore memory-mapped devices and such things. */ 1509bc403f03SAttilio Rao VM_OBJECT_RLOCK(object); 1510976a87a2SAlan Cox while ((backing_object = object->backing_object) != NULL) { 1511bc403f03SAttilio Rao VM_OBJECT_RLOCK(backing_object); 1512bc403f03SAttilio Rao VM_OBJECT_RUNLOCK(object); 1513976a87a2SAlan Cox object = backing_object; 1514976a87a2SAlan Cox } 1515976a87a2SAlan Cox ignore_entry = object->type != OBJT_DEFAULT && 1516bc411bc2SJohn Baldwin object->type != OBJT_SWAP && object->type != OBJT_VNODE && 1517bc411bc2SJohn Baldwin object->type != OBJT_PHYS; 1518bc403f03SAttilio Rao VM_OBJECT_RUNLOCK(object); 1519976a87a2SAlan Cox if (ignore_entry) 15200ff27d31SJohn Polstra continue; 15210ff27d31SJohn Polstra 15220ff27d31SJohn Polstra (*func)(entry, closure); 15230ff27d31SJohn Polstra } 1524976a87a2SAlan Cox vm_map_unlock_read(map); 15250ff27d31SJohn Polstra } 15260ff27d31SJohn Polstra 15270ff27d31SJohn Polstra /* 15280ff27d31SJohn Polstra * Write the core file header to the file, including padding up to 15290ff27d31SJohn Polstra * the page boundary. 15300ff27d31SJohn Polstra */ 15318c64af4fSJohn Polstra static int 1532aa14e9b7SMark Johnston __elfN(corehdr)(struct coredump_params *p, int numsegs, void *hdr, 1533aa14e9b7SMark Johnston size_t hdrsize, struct note_info_list *notelst, size_t notesz) 15348c64af4fSJohn Polstra { 1535bd390213SMikolaj Golub struct note_info *ninfo; 1536bd390213SMikolaj Golub struct sbuf *sb; 1537bd390213SMikolaj Golub int error; 15388c64af4fSJohn Polstra 15398c64af4fSJohn Polstra /* Fill in the header. */ 15400ff27d31SJohn Polstra bzero(hdr, hdrsize); 1541aa14e9b7SMark Johnston __elfN(puthdr)(p->td, hdr, hdrsize, numsegs, notesz); 15428c64af4fSJohn Polstra 1543bd390213SMikolaj Golub sb = sbuf_new(NULL, NULL, CORE_BUF_SIZE, SBUF_FIXEDLEN); 1544aa14e9b7SMark Johnston sbuf_set_drain(sb, sbuf_drain_core_output, p); 1545bd390213SMikolaj Golub sbuf_start_section(sb, NULL); 1546bd390213SMikolaj Golub sbuf_bcat(sb, hdr, hdrsize); 1547bd390213SMikolaj Golub TAILQ_FOREACH(ninfo, notelst, link) 1548bd390213SMikolaj Golub __elfN(putnote)(ninfo, sb); 1549bd390213SMikolaj Golub /* Align up to a page boundary for the program segments. */ 1550bd390213SMikolaj Golub sbuf_end_section(sb, -1, PAGE_SIZE, 0); 1551bd390213SMikolaj Golub error = sbuf_finish(sb); 1552bd390213SMikolaj Golub sbuf_delete(sb); 1553bd390213SMikolaj Golub 1554bd390213SMikolaj Golub return (error); 1555e7228204SAlfred Perlstein } 1556bd390213SMikolaj Golub 1557bd390213SMikolaj Golub static void 1558bd390213SMikolaj Golub __elfN(prepare_notes)(struct thread *td, struct note_info_list *list, 1559bd390213SMikolaj Golub size_t *sizep) 1560bd390213SMikolaj Golub { 1561bd390213SMikolaj Golub struct proc *p; 1562bd390213SMikolaj Golub struct thread *thr; 1563bd390213SMikolaj Golub size_t size; 1564bd390213SMikolaj Golub 1565bd390213SMikolaj Golub p = td->td_proc; 1566bd390213SMikolaj Golub size = 0; 1567bd390213SMikolaj Golub 1568bd390213SMikolaj Golub size += register_note(list, NT_PRPSINFO, __elfN(note_prpsinfo), p); 1569bd390213SMikolaj Golub 1570bd390213SMikolaj Golub /* 1571bd390213SMikolaj Golub * To have the debugger select the right thread (LWP) as the initial 1572bd390213SMikolaj Golub * thread, we dump the state of the thread passed to us in td first. 1573bd390213SMikolaj Golub * This is the thread that causes the core dump and thus likely to 1574bd390213SMikolaj Golub * be the right thread one wants to have selected in the debugger. 1575bd390213SMikolaj Golub */ 1576bd390213SMikolaj Golub thr = td; 1577bd390213SMikolaj Golub while (thr != NULL) { 1578bd390213SMikolaj Golub size += register_note(list, NT_PRSTATUS, 1579bd390213SMikolaj Golub __elfN(note_prstatus), thr); 1580bd390213SMikolaj Golub size += register_note(list, NT_FPREGSET, 1581bd390213SMikolaj Golub __elfN(note_fpregset), thr); 1582bd390213SMikolaj Golub size += register_note(list, NT_THRMISC, 1583bd390213SMikolaj Golub __elfN(note_thrmisc), thr); 1584bd390213SMikolaj Golub size += register_note(list, -1, 1585bd390213SMikolaj Golub __elfN(note_threadmd), thr); 1586bd390213SMikolaj Golub 1587bd390213SMikolaj Golub thr = (thr == td) ? TAILQ_FIRST(&p->p_threads) : 1588bd390213SMikolaj Golub TAILQ_NEXT(thr, td_plist); 1589bd390213SMikolaj Golub if (thr == td) 1590bd390213SMikolaj Golub thr = TAILQ_NEXT(thr, td_plist); 1591dada0278SJohn Polstra } 1592dada0278SJohn Polstra 1593f1fca82eSMikolaj Golub size += register_note(list, NT_PROCSTAT_PROC, 1594f1fca82eSMikolaj Golub __elfN(note_procstat_proc), p); 1595f1fca82eSMikolaj Golub size += register_note(list, NT_PROCSTAT_FILES, 1596f1fca82eSMikolaj Golub note_procstat_files, p); 1597f1fca82eSMikolaj Golub size += register_note(list, NT_PROCSTAT_VMMAP, 1598f1fca82eSMikolaj Golub note_procstat_vmmap, p); 1599f1fca82eSMikolaj Golub size += register_note(list, NT_PROCSTAT_GROUPS, 1600f1fca82eSMikolaj Golub note_procstat_groups, p); 1601f1fca82eSMikolaj Golub size += register_note(list, NT_PROCSTAT_UMASK, 1602f1fca82eSMikolaj Golub note_procstat_umask, p); 1603f1fca82eSMikolaj Golub size += register_note(list, NT_PROCSTAT_RLIMIT, 1604f1fca82eSMikolaj Golub note_procstat_rlimit, p); 1605f1fca82eSMikolaj Golub size += register_note(list, NT_PROCSTAT_OSREL, 1606f1fca82eSMikolaj Golub note_procstat_osrel, p); 1607f1fca82eSMikolaj Golub size += register_note(list, NT_PROCSTAT_PSSTRINGS, 1608f1fca82eSMikolaj Golub __elfN(note_procstat_psstrings), p); 1609f1fca82eSMikolaj Golub size += register_note(list, NT_PROCSTAT_AUXV, 1610f1fca82eSMikolaj Golub __elfN(note_procstat_auxv), p); 1611f1fca82eSMikolaj Golub 1612bd390213SMikolaj Golub *sizep = size; 1613bd390213SMikolaj Golub } 1614bd390213SMikolaj Golub 1615bd390213SMikolaj Golub static void 1616bd390213SMikolaj Golub __elfN(puthdr)(struct thread *td, void *hdr, size_t hdrsize, int numsegs, 1617bd390213SMikolaj Golub size_t notesz) 1618bd390213SMikolaj Golub { 1619bd390213SMikolaj Golub Elf_Ehdr *ehdr; 1620bd390213SMikolaj Golub Elf_Phdr *phdr; 1621bd390213SMikolaj Golub struct phdr_closure phc; 1622bd390213SMikolaj Golub 1623bd390213SMikolaj Golub ehdr = (Elf_Ehdr *)hdr; 1624bd390213SMikolaj Golub phdr = (Elf_Phdr *)((char *)hdr + sizeof(Elf_Ehdr)); 1625bd390213SMikolaj Golub 1626bd390213SMikolaj Golub ehdr->e_ident[EI_MAG0] = ELFMAG0; 1627bd390213SMikolaj Golub ehdr->e_ident[EI_MAG1] = ELFMAG1; 1628bd390213SMikolaj Golub ehdr->e_ident[EI_MAG2] = ELFMAG2; 1629bd390213SMikolaj Golub ehdr->e_ident[EI_MAG3] = ELFMAG3; 1630bd390213SMikolaj Golub ehdr->e_ident[EI_CLASS] = ELF_CLASS; 1631bd390213SMikolaj Golub ehdr->e_ident[EI_DATA] = ELF_DATA; 1632bd390213SMikolaj Golub ehdr->e_ident[EI_VERSION] = EV_CURRENT; 1633bd390213SMikolaj Golub ehdr->e_ident[EI_OSABI] = ELFOSABI_FREEBSD; 1634bd390213SMikolaj Golub ehdr->e_ident[EI_ABIVERSION] = 0; 1635bd390213SMikolaj Golub ehdr->e_ident[EI_PAD] = 0; 1636bd390213SMikolaj Golub ehdr->e_type = ET_CORE; 1637bd390213SMikolaj Golub #if defined(COMPAT_FREEBSD32) && __ELF_WORD_SIZE == 32 1638bd390213SMikolaj Golub ehdr->e_machine = ELF_ARCH32; 1639bd390213SMikolaj Golub #else 1640bd390213SMikolaj Golub ehdr->e_machine = ELF_ARCH; 1641bd390213SMikolaj Golub #endif 1642bd390213SMikolaj Golub ehdr->e_version = EV_CURRENT; 1643bd390213SMikolaj Golub ehdr->e_entry = 0; 1644bd390213SMikolaj Golub ehdr->e_phoff = sizeof(Elf_Ehdr); 1645bd390213SMikolaj Golub ehdr->e_flags = 0; 1646bd390213SMikolaj Golub ehdr->e_ehsize = sizeof(Elf_Ehdr); 1647bd390213SMikolaj Golub ehdr->e_phentsize = sizeof(Elf_Phdr); 1648bd390213SMikolaj Golub ehdr->e_phnum = numsegs + 1; 1649bd390213SMikolaj Golub ehdr->e_shentsize = sizeof(Elf_Shdr); 1650bd390213SMikolaj Golub ehdr->e_shnum = 0; 1651bd390213SMikolaj Golub ehdr->e_shstrndx = SHN_UNDEF; 1652bd390213SMikolaj Golub 1653bd390213SMikolaj Golub /* 1654bd390213SMikolaj Golub * Fill in the program header entries. 1655bd390213SMikolaj Golub */ 1656bd390213SMikolaj Golub 1657bd390213SMikolaj Golub /* The note segement. */ 1658bd390213SMikolaj Golub phdr->p_type = PT_NOTE; 1659bd390213SMikolaj Golub phdr->p_offset = hdrsize; 1660bd390213SMikolaj Golub phdr->p_vaddr = 0; 1661bd390213SMikolaj Golub phdr->p_paddr = 0; 1662bd390213SMikolaj Golub phdr->p_filesz = notesz; 1663bd390213SMikolaj Golub phdr->p_memsz = 0; 1664bd390213SMikolaj Golub phdr->p_flags = PF_R; 16651b8388cdSMikolaj Golub phdr->p_align = ELF_NOTE_ROUNDSIZE; 1666bd390213SMikolaj Golub phdr++; 1667bd390213SMikolaj Golub 1668bd390213SMikolaj Golub /* All the writable segments from the program. */ 1669bd390213SMikolaj Golub phc.phdr = phdr; 1670bd390213SMikolaj Golub phc.offset = round_page(hdrsize + notesz); 1671bd390213SMikolaj Golub each_writable_segment(td, cb_put_phdr, &phc); 1672bd390213SMikolaj Golub } 1673bd390213SMikolaj Golub 1674bd390213SMikolaj Golub static size_t 1675bd390213SMikolaj Golub register_note(struct note_info_list *list, int type, outfunc_t out, void *arg) 1676bd390213SMikolaj Golub { 1677bd390213SMikolaj Golub struct note_info *ninfo; 1678bd390213SMikolaj Golub size_t size, notesize; 1679bd390213SMikolaj Golub 1680bd390213SMikolaj Golub size = 0; 1681bd390213SMikolaj Golub out(arg, NULL, &size); 1682bd390213SMikolaj Golub ninfo = malloc(sizeof(*ninfo), M_TEMP, M_ZERO | M_WAITOK); 1683bd390213SMikolaj Golub ninfo->type = type; 1684bd390213SMikolaj Golub ninfo->outfunc = out; 1685bd390213SMikolaj Golub ninfo->outarg = arg; 1686bd390213SMikolaj Golub ninfo->outsize = size; 1687bd390213SMikolaj Golub TAILQ_INSERT_TAIL(list, ninfo, link); 1688bd390213SMikolaj Golub 1689bd390213SMikolaj Golub if (type == -1) 1690bd390213SMikolaj Golub return (size); 1691bd390213SMikolaj Golub 1692bd390213SMikolaj Golub notesize = sizeof(Elf_Note) + /* note header */ 1693180e57e5SJohn Baldwin roundup2(sizeof(FREEBSD_ABI_VENDOR), ELF_NOTE_ROUNDSIZE) + 1694180e57e5SJohn Baldwin /* note name */ 1695180e57e5SJohn Baldwin roundup2(size, ELF_NOTE_ROUNDSIZE); /* note description */ 1696180e57e5SJohn Baldwin 1697180e57e5SJohn Baldwin return (notesize); 1698180e57e5SJohn Baldwin } 1699180e57e5SJohn Baldwin 1700180e57e5SJohn Baldwin static size_t 1701180e57e5SJohn Baldwin append_note_data(const void *src, void *dst, size_t len) 1702180e57e5SJohn Baldwin { 1703180e57e5SJohn Baldwin size_t padded_len; 1704180e57e5SJohn Baldwin 1705180e57e5SJohn Baldwin padded_len = roundup2(len, ELF_NOTE_ROUNDSIZE); 1706180e57e5SJohn Baldwin if (dst != NULL) { 1707180e57e5SJohn Baldwin bcopy(src, dst, len); 1708180e57e5SJohn Baldwin bzero((char *)dst + len, padded_len - len); 1709180e57e5SJohn Baldwin } 1710180e57e5SJohn Baldwin return (padded_len); 1711180e57e5SJohn Baldwin } 1712180e57e5SJohn Baldwin 1713180e57e5SJohn Baldwin size_t 1714180e57e5SJohn Baldwin __elfN(populate_note)(int type, void *src, void *dst, size_t size, void **descp) 1715180e57e5SJohn Baldwin { 1716180e57e5SJohn Baldwin Elf_Note *note; 1717180e57e5SJohn Baldwin char *buf; 1718180e57e5SJohn Baldwin size_t notesize; 1719180e57e5SJohn Baldwin 1720180e57e5SJohn Baldwin buf = dst; 1721180e57e5SJohn Baldwin if (buf != NULL) { 1722180e57e5SJohn Baldwin note = (Elf_Note *)buf; 1723180e57e5SJohn Baldwin note->n_namesz = sizeof(FREEBSD_ABI_VENDOR); 1724180e57e5SJohn Baldwin note->n_descsz = size; 1725180e57e5SJohn Baldwin note->n_type = type; 1726180e57e5SJohn Baldwin buf += sizeof(*note); 1727180e57e5SJohn Baldwin buf += append_note_data(FREEBSD_ABI_VENDOR, buf, 1728180e57e5SJohn Baldwin sizeof(FREEBSD_ABI_VENDOR)); 1729180e57e5SJohn Baldwin append_note_data(src, buf, size); 1730180e57e5SJohn Baldwin if (descp != NULL) 1731180e57e5SJohn Baldwin *descp = buf; 1732180e57e5SJohn Baldwin } 1733180e57e5SJohn Baldwin 1734180e57e5SJohn Baldwin notesize = sizeof(Elf_Note) + /* note header */ 1735180e57e5SJohn Baldwin roundup2(sizeof(FREEBSD_ABI_VENDOR), ELF_NOTE_ROUNDSIZE) + 1736180e57e5SJohn Baldwin /* note name */ 17371b8388cdSMikolaj Golub roundup2(size, ELF_NOTE_ROUNDSIZE); /* note description */ 1738bd390213SMikolaj Golub 1739bd390213SMikolaj Golub return (notesize); 1740bd390213SMikolaj Golub } 1741bd390213SMikolaj Golub 1742bd390213SMikolaj Golub static void 1743bd390213SMikolaj Golub __elfN(putnote)(struct note_info *ninfo, struct sbuf *sb) 1744bd390213SMikolaj Golub { 1745bd390213SMikolaj Golub Elf_Note note; 174614bdbaf2SConrad Meyer ssize_t old_len, sect_len; 174714bdbaf2SConrad Meyer size_t new_len, descsz, i; 1748bd390213SMikolaj Golub 1749bd390213SMikolaj Golub if (ninfo->type == -1) { 1750bd390213SMikolaj Golub ninfo->outfunc(ninfo->outarg, sb, &ninfo->outsize); 1751bd390213SMikolaj Golub return; 1752bd390213SMikolaj Golub } 1753bd390213SMikolaj Golub 1754180e57e5SJohn Baldwin note.n_namesz = sizeof(FREEBSD_ABI_VENDOR); 1755bd390213SMikolaj Golub note.n_descsz = ninfo->outsize; 1756bd390213SMikolaj Golub note.n_type = ninfo->type; 1757bd390213SMikolaj Golub 1758bd390213SMikolaj Golub sbuf_bcat(sb, ¬e, sizeof(note)); 1759bd390213SMikolaj Golub sbuf_start_section(sb, &old_len); 1760180e57e5SJohn Baldwin sbuf_bcat(sb, FREEBSD_ABI_VENDOR, sizeof(FREEBSD_ABI_VENDOR)); 17611b8388cdSMikolaj Golub sbuf_end_section(sb, old_len, ELF_NOTE_ROUNDSIZE, 0); 1762bd390213SMikolaj Golub if (note.n_descsz == 0) 1763bd390213SMikolaj Golub return; 1764bd390213SMikolaj Golub sbuf_start_section(sb, &old_len); 1765bd390213SMikolaj Golub ninfo->outfunc(ninfo->outarg, sb, &ninfo->outsize); 176614bdbaf2SConrad Meyer sect_len = sbuf_end_section(sb, old_len, ELF_NOTE_ROUNDSIZE, 0); 176714bdbaf2SConrad Meyer if (sect_len < 0) 176814bdbaf2SConrad Meyer return; 176914bdbaf2SConrad Meyer 177014bdbaf2SConrad Meyer new_len = (size_t)sect_len; 177114bdbaf2SConrad Meyer descsz = roundup(note.n_descsz, ELF_NOTE_ROUNDSIZE); 177214bdbaf2SConrad Meyer if (new_len < descsz) { 177314bdbaf2SConrad Meyer /* 177414bdbaf2SConrad Meyer * It is expected that individual note emitters will correctly 177514bdbaf2SConrad Meyer * predict their expected output size and fill up to that size 177614bdbaf2SConrad Meyer * themselves, padding in a format-specific way if needed. 177714bdbaf2SConrad Meyer * However, in case they don't, just do it here with zeros. 177814bdbaf2SConrad Meyer */ 177914bdbaf2SConrad Meyer for (i = 0; i < descsz - new_len; i++) 178014bdbaf2SConrad Meyer sbuf_putc(sb, 0); 178114bdbaf2SConrad Meyer } else if (new_len > descsz) { 178214bdbaf2SConrad Meyer /* 178314bdbaf2SConrad Meyer * We can't always truncate sb -- we may have drained some 178414bdbaf2SConrad Meyer * of it already. 178514bdbaf2SConrad Meyer */ 178614bdbaf2SConrad Meyer KASSERT(new_len == descsz, ("%s: Note type %u changed as we " 178714bdbaf2SConrad Meyer "read it (%zu > %zu). Since it is longer than " 178814bdbaf2SConrad Meyer "expected, this coredump's notes are corrupt. THIS " 178914bdbaf2SConrad Meyer "IS A BUG in the note_procstat routine for type %u.\n", 179014bdbaf2SConrad Meyer __func__, (unsigned)note.n_type, new_len, descsz, 179114bdbaf2SConrad Meyer (unsigned)note.n_type)); 179214bdbaf2SConrad Meyer } 1793bd390213SMikolaj Golub } 1794bd390213SMikolaj Golub 1795bd390213SMikolaj Golub /* 1796bd390213SMikolaj Golub * Miscellaneous note out functions. 1797bd390213SMikolaj Golub */ 1798bd390213SMikolaj Golub 1799841c0c7eSNathan Whitehorn #if defined(COMPAT_FREEBSD32) && __ELF_WORD_SIZE == 32 1800841c0c7eSNathan Whitehorn #include <compat/freebsd32/freebsd32.h> 1801841c0c7eSNathan Whitehorn 180262919d78SPeter Wemm typedef struct prstatus32 elf_prstatus_t; 180362919d78SPeter Wemm typedef struct prpsinfo32 elf_prpsinfo_t; 180462919d78SPeter Wemm typedef struct fpreg32 elf_prfpregset_t; 180562919d78SPeter Wemm typedef struct fpreg32 elf_fpregset_t; 180662919d78SPeter Wemm typedef struct reg32 elf_gregset_t; 18077f08176eSAttilio Rao typedef struct thrmisc32 elf_thrmisc_t; 1808f1fca82eSMikolaj Golub #define ELF_KERN_PROC_MASK KERN_PROC_MASK32 1809f1fca82eSMikolaj Golub typedef struct kinfo_proc32 elf_kinfo_proc_t; 1810f1fca82eSMikolaj Golub typedef uint32_t elf_ps_strings_t; 181162919d78SPeter Wemm #else 181262919d78SPeter Wemm typedef prstatus_t elf_prstatus_t; 181362919d78SPeter Wemm typedef prpsinfo_t elf_prpsinfo_t; 181462919d78SPeter Wemm typedef prfpregset_t elf_prfpregset_t; 181562919d78SPeter Wemm typedef prfpregset_t elf_fpregset_t; 181662919d78SPeter Wemm typedef gregset_t elf_gregset_t; 18177f08176eSAttilio Rao typedef thrmisc_t elf_thrmisc_t; 1818f1fca82eSMikolaj Golub #define ELF_KERN_PROC_MASK 0 1819f1fca82eSMikolaj Golub typedef struct kinfo_proc elf_kinfo_proc_t; 1820f1fca82eSMikolaj Golub typedef vm_offset_t elf_ps_strings_t; 182162919d78SPeter Wemm #endif 182262919d78SPeter Wemm 18238c64af4fSJohn Polstra static void 1824bd390213SMikolaj Golub __elfN(note_prpsinfo)(void *arg, struct sbuf *sb, size_t *sizep) 18258c64af4fSJohn Polstra { 1826*c77547d2SJohn Baldwin struct sbuf sbarg; 1827*c77547d2SJohn Baldwin size_t len; 1828*c77547d2SJohn Baldwin char *cp, *end; 1829247aba24SMarcel Moolenaar struct proc *p; 1830bd390213SMikolaj Golub elf_prpsinfo_t *psinfo; 1831*c77547d2SJohn Baldwin int error; 18328c64af4fSJohn Polstra 1833bd390213SMikolaj Golub p = (struct proc *)arg; 1834bd390213SMikolaj Golub if (sb != NULL) { 1835bd390213SMikolaj Golub KASSERT(*sizep == sizeof(*psinfo), ("invalid size")); 1836bd390213SMikolaj Golub psinfo = malloc(sizeof(*psinfo), M_TEMP, M_ZERO | M_WAITOK); 18378c9b7b2cSMarcel Moolenaar psinfo->pr_version = PRPSINFO_VERSION; 183862919d78SPeter Wemm psinfo->pr_psinfosz = sizeof(elf_prpsinfo_t); 1839ccd3953eSJohn Baldwin strlcpy(psinfo->pr_fname, p->p_comm, sizeof(psinfo->pr_fname)); 1840*c77547d2SJohn Baldwin PROC_LOCK(p); 1841*c77547d2SJohn Baldwin if (p->p_args != NULL) { 1842*c77547d2SJohn Baldwin len = sizeof(psinfo->pr_psargs) - 1; 1843*c77547d2SJohn Baldwin if (len > p->p_args->ar_length) 1844*c77547d2SJohn Baldwin len = p->p_args->ar_length; 1845*c77547d2SJohn Baldwin memcpy(psinfo->pr_psargs, p->p_args->ar_args, len); 1846*c77547d2SJohn Baldwin PROC_UNLOCK(p); 1847*c77547d2SJohn Baldwin error = 0; 1848*c77547d2SJohn Baldwin } else { 1849*c77547d2SJohn Baldwin _PHOLD(p); 1850*c77547d2SJohn Baldwin PROC_UNLOCK(p); 1851*c77547d2SJohn Baldwin sbuf_new(&sbarg, psinfo->pr_psargs, 1852*c77547d2SJohn Baldwin sizeof(psinfo->pr_psargs), SBUF_FIXEDLEN); 1853*c77547d2SJohn Baldwin error = proc_getargv(curthread, p, &sbarg); 1854*c77547d2SJohn Baldwin PRELE(p); 1855*c77547d2SJohn Baldwin if (sbuf_finish(&sbarg) == 0) 1856*c77547d2SJohn Baldwin len = sbuf_len(&sbarg) - 1; 1857*c77547d2SJohn Baldwin else 1858*c77547d2SJohn Baldwin len = sizeof(psinfo->pr_psargs) - 1; 1859*c77547d2SJohn Baldwin sbuf_delete(&sbarg); 1860*c77547d2SJohn Baldwin } 1861*c77547d2SJohn Baldwin if (error || len == 0) 1862ccd3953eSJohn Baldwin strlcpy(psinfo->pr_psargs, p->p_comm, 18638c9b7b2cSMarcel Moolenaar sizeof(psinfo->pr_psargs)); 1864*c77547d2SJohn Baldwin else { 1865*c77547d2SJohn Baldwin KASSERT(len < sizeof(psinfo->pr_psargs), 1866*c77547d2SJohn Baldwin ("len is too long: %zu vs %zu", len, 1867*c77547d2SJohn Baldwin sizeof(psinfo->pr_psargs))); 1868*c77547d2SJohn Baldwin cp = psinfo->pr_psargs; 1869*c77547d2SJohn Baldwin end = cp + len - 1; 1870*c77547d2SJohn Baldwin for (;;) { 1871*c77547d2SJohn Baldwin cp = memchr(cp, '\0', end - cp); 1872*c77547d2SJohn Baldwin if (cp == NULL) 1873*c77547d2SJohn Baldwin break; 1874*c77547d2SJohn Baldwin *cp = ' '; 1875*c77547d2SJohn Baldwin } 1876*c77547d2SJohn Baldwin } 1877bd390213SMikolaj Golub sbuf_bcat(sb, psinfo, sizeof(*psinfo)); 1878bd390213SMikolaj Golub free(psinfo, M_TEMP); 1879bd390213SMikolaj Golub } 1880bd390213SMikolaj Golub *sizep = sizeof(*psinfo); 1881bd390213SMikolaj Golub } 1882bd390213SMikolaj Golub 1883bd390213SMikolaj Golub static void 1884bd390213SMikolaj Golub __elfN(note_prstatus)(void *arg, struct sbuf *sb, size_t *sizep) 1885bd390213SMikolaj Golub { 1886bd390213SMikolaj Golub struct thread *td; 1887bd390213SMikolaj Golub elf_prstatus_t *status; 1888bd390213SMikolaj Golub 1889bd390213SMikolaj Golub td = (struct thread *)arg; 1890bd390213SMikolaj Golub if (sb != NULL) { 1891bd390213SMikolaj Golub KASSERT(*sizep == sizeof(*status), ("invalid size")); 1892bd390213SMikolaj Golub status = malloc(sizeof(*status), M_TEMP, M_ZERO | M_WAITOK); 18938c9b7b2cSMarcel Moolenaar status->pr_version = PRSTATUS_VERSION; 189462919d78SPeter Wemm status->pr_statussz = sizeof(elf_prstatus_t); 189562919d78SPeter Wemm status->pr_gregsetsz = sizeof(elf_gregset_t); 189662919d78SPeter Wemm status->pr_fpregsetsz = sizeof(elf_fpregset_t); 18978c9b7b2cSMarcel Moolenaar status->pr_osreldate = osreldate; 1898bd390213SMikolaj Golub status->pr_cursig = td->td_proc->p_sig; 1899bd390213SMikolaj Golub status->pr_pid = td->td_tid; 1900841c0c7eSNathan Whitehorn #if defined(COMPAT_FREEBSD32) && __ELF_WORD_SIZE == 32 1901bd390213SMikolaj Golub fill_regs32(td, &status->pr_reg); 190262919d78SPeter Wemm #else 1903bd390213SMikolaj Golub fill_regs(td, &status->pr_reg); 190462919d78SPeter Wemm #endif 1905bd390213SMikolaj Golub sbuf_bcat(sb, status, sizeof(*status)); 1906bd390213SMikolaj Golub free(status, M_TEMP); 19078c9b7b2cSMarcel Moolenaar } 1908bd390213SMikolaj Golub *sizep = sizeof(*status); 1909bd390213SMikolaj Golub } 1910bd390213SMikolaj Golub 1911bd390213SMikolaj Golub static void 1912bd390213SMikolaj Golub __elfN(note_fpregset)(void *arg, struct sbuf *sb, size_t *sizep) 1913bd390213SMikolaj Golub { 1914bd390213SMikolaj Golub struct thread *td; 1915bd390213SMikolaj Golub elf_prfpregset_t *fpregset; 1916bd390213SMikolaj Golub 1917bd390213SMikolaj Golub td = (struct thread *)arg; 1918bd390213SMikolaj Golub if (sb != NULL) { 1919bd390213SMikolaj Golub KASSERT(*sizep == sizeof(*fpregset), ("invalid size")); 1920bd390213SMikolaj Golub fpregset = malloc(sizeof(*fpregset), M_TEMP, M_ZERO | M_WAITOK); 1921bd390213SMikolaj Golub #if defined(COMPAT_FREEBSD32) && __ELF_WORD_SIZE == 32 1922bd390213SMikolaj Golub fill_fpregs32(td, fpregset); 1923bd390213SMikolaj Golub #else 1924bd390213SMikolaj Golub fill_fpregs(td, fpregset); 1925bd390213SMikolaj Golub #endif 1926bd390213SMikolaj Golub sbuf_bcat(sb, fpregset, sizeof(*fpregset)); 1927bd390213SMikolaj Golub free(fpregset, M_TEMP); 1928bd390213SMikolaj Golub } 1929bd390213SMikolaj Golub *sizep = sizeof(*fpregset); 1930bd390213SMikolaj Golub } 1931bd390213SMikolaj Golub 1932bd390213SMikolaj Golub static void 1933bd390213SMikolaj Golub __elfN(note_thrmisc)(void *arg, struct sbuf *sb, size_t *sizep) 1934bd390213SMikolaj Golub { 1935bd390213SMikolaj Golub struct thread *td; 1936bd390213SMikolaj Golub elf_thrmisc_t thrmisc; 1937bd390213SMikolaj Golub 1938bd390213SMikolaj Golub td = (struct thread *)arg; 1939bd390213SMikolaj Golub if (sb != NULL) { 1940bd390213SMikolaj Golub KASSERT(*sizep == sizeof(thrmisc), ("invalid size")); 1941bd390213SMikolaj Golub bzero(&thrmisc._pad, sizeof(thrmisc._pad)); 1942bd390213SMikolaj Golub strcpy(thrmisc.pr_tname, td->td_name); 1943bd390213SMikolaj Golub sbuf_bcat(sb, &thrmisc, sizeof(thrmisc)); 1944bd390213SMikolaj Golub } 1945bd390213SMikolaj Golub *sizep = sizeof(thrmisc); 1946bd390213SMikolaj Golub } 1947bd390213SMikolaj Golub 19484da47b2fSMarcel Moolenaar /* 19494da47b2fSMarcel Moolenaar * Allow for MD specific notes, as well as any MD 19504da47b2fSMarcel Moolenaar * specific preparations for writing MI notes. 19514da47b2fSMarcel Moolenaar */ 19528c64af4fSJohn Polstra static void 1953bd390213SMikolaj Golub __elfN(note_threadmd)(void *arg, struct sbuf *sb, size_t *sizep) 19548c64af4fSJohn Polstra { 1955bd390213SMikolaj Golub struct thread *td; 1956bd390213SMikolaj Golub void *buf; 1957bd390213SMikolaj Golub size_t size; 19588c64af4fSJohn Polstra 1959bd390213SMikolaj Golub td = (struct thread *)arg; 1960bd390213SMikolaj Golub size = *sizep; 1961bd390213SMikolaj Golub if (size != 0 && sb != NULL) 1962bd390213SMikolaj Golub buf = malloc(size, M_TEMP, M_ZERO | M_WAITOK); 196383a396ceSChristian Brueffer else 196483a396ceSChristian Brueffer buf = NULL; 1965bd390213SMikolaj Golub size = 0; 1966bd390213SMikolaj Golub __elfN(dump_thread)(td, buf, &size); 196764779280SKonstantin Belousov KASSERT(sb == NULL || *sizep == size, ("invalid size")); 196883a396ceSChristian Brueffer if (size != 0 && sb != NULL) 1969bd390213SMikolaj Golub sbuf_bcat(sb, buf, size); 1970a1761d73SChristian Brueffer free(buf, M_TEMP); 1971bd390213SMikolaj Golub *sizep = size; 19728c64af4fSJohn Polstra } 19738c64af4fSJohn Polstra 1974f1fca82eSMikolaj Golub #ifdef KINFO_PROC_SIZE 1975f1fca82eSMikolaj Golub CTASSERT(sizeof(struct kinfo_proc) == KINFO_PROC_SIZE); 1976f1fca82eSMikolaj Golub #endif 1977f1fca82eSMikolaj Golub 1978f1fca82eSMikolaj Golub static void 1979f1fca82eSMikolaj Golub __elfN(note_procstat_proc)(void *arg, struct sbuf *sb, size_t *sizep) 1980f1fca82eSMikolaj Golub { 1981f1fca82eSMikolaj Golub struct proc *p; 1982f1fca82eSMikolaj Golub size_t size; 1983f1fca82eSMikolaj Golub int structsize; 1984f1fca82eSMikolaj Golub 1985f1fca82eSMikolaj Golub p = (struct proc *)arg; 1986f1fca82eSMikolaj Golub size = sizeof(structsize) + p->p_numthreads * 1987f1fca82eSMikolaj Golub sizeof(elf_kinfo_proc_t); 1988f1fca82eSMikolaj Golub 1989f1fca82eSMikolaj Golub if (sb != NULL) { 1990f1fca82eSMikolaj Golub KASSERT(*sizep == size, ("invalid size")); 1991f1fca82eSMikolaj Golub structsize = sizeof(elf_kinfo_proc_t); 1992f1fca82eSMikolaj Golub sbuf_bcat(sb, &structsize, sizeof(structsize)); 19936662ce5aSMateusz Guzik sx_slock(&proctree_lock); 1994f1fca82eSMikolaj Golub PROC_LOCK(p); 1995f1fca82eSMikolaj Golub kern_proc_out(p, sb, ELF_KERN_PROC_MASK); 19966662ce5aSMateusz Guzik sx_sunlock(&proctree_lock); 1997f1fca82eSMikolaj Golub } 1998f1fca82eSMikolaj Golub *sizep = size; 1999f1fca82eSMikolaj Golub } 2000f1fca82eSMikolaj Golub 2001f1fca82eSMikolaj Golub #ifdef KINFO_FILE_SIZE 2002f1fca82eSMikolaj Golub CTASSERT(sizeof(struct kinfo_file) == KINFO_FILE_SIZE); 2003f1fca82eSMikolaj Golub #endif 2004f1fca82eSMikolaj Golub 2005f1fca82eSMikolaj Golub static void 2006f1fca82eSMikolaj Golub note_procstat_files(void *arg, struct sbuf *sb, size_t *sizep) 2007f1fca82eSMikolaj Golub { 2008f1fca82eSMikolaj Golub struct proc *p; 200914bdbaf2SConrad Meyer size_t size, sect_sz, i; 201014bdbaf2SConrad Meyer ssize_t start_len, sect_len; 201114bdbaf2SConrad Meyer int structsize, filedesc_flags; 201214bdbaf2SConrad Meyer 2013bcb60d52SConrad Meyer if (coredump_pack_fileinfo) 201414bdbaf2SConrad Meyer filedesc_flags = KERN_FILEDESC_PACK_KINFO; 201514bdbaf2SConrad Meyer else 201614bdbaf2SConrad Meyer filedesc_flags = 0; 2017f1fca82eSMikolaj Golub 2018f1fca82eSMikolaj Golub p = (struct proc *)arg; 201914bdbaf2SConrad Meyer structsize = sizeof(struct kinfo_file); 2020f1fca82eSMikolaj Golub if (sb == NULL) { 2021f1fca82eSMikolaj Golub size = 0; 2022f1fca82eSMikolaj Golub sb = sbuf_new(NULL, NULL, 128, SBUF_FIXEDLEN); 2023f1fca82eSMikolaj Golub sbuf_set_drain(sb, sbuf_drain_count, &size); 2024f1fca82eSMikolaj Golub sbuf_bcat(sb, &structsize, sizeof(structsize)); 2025f1fca82eSMikolaj Golub PROC_LOCK(p); 202614bdbaf2SConrad Meyer kern_proc_filedesc_out(p, sb, -1, filedesc_flags); 2027f1fca82eSMikolaj Golub sbuf_finish(sb); 2028f1fca82eSMikolaj Golub sbuf_delete(sb); 2029f1fca82eSMikolaj Golub *sizep = size; 2030f1fca82eSMikolaj Golub } else { 203114bdbaf2SConrad Meyer sbuf_start_section(sb, &start_len); 203214bdbaf2SConrad Meyer 2033f1fca82eSMikolaj Golub sbuf_bcat(sb, &structsize, sizeof(structsize)); 2034f1fca82eSMikolaj Golub PROC_LOCK(p); 203514bdbaf2SConrad Meyer kern_proc_filedesc_out(p, sb, *sizep - sizeof(structsize), 203614bdbaf2SConrad Meyer filedesc_flags); 203714bdbaf2SConrad Meyer 203814bdbaf2SConrad Meyer sect_len = sbuf_end_section(sb, start_len, 0, 0); 203914bdbaf2SConrad Meyer if (sect_len < 0) 204014bdbaf2SConrad Meyer return; 204114bdbaf2SConrad Meyer sect_sz = sect_len; 204214bdbaf2SConrad Meyer 204314bdbaf2SConrad Meyer KASSERT(sect_sz <= *sizep, 204414bdbaf2SConrad Meyer ("kern_proc_filedesc_out did not respect maxlen; " 204514bdbaf2SConrad Meyer "requested %zu, got %zu", *sizep - sizeof(structsize), 204614bdbaf2SConrad Meyer sect_sz - sizeof(structsize))); 204714bdbaf2SConrad Meyer 204814bdbaf2SConrad Meyer for (i = 0; i < *sizep - sect_sz && sb->s_error == 0; i++) 204914bdbaf2SConrad Meyer sbuf_putc(sb, 0); 2050f1fca82eSMikolaj Golub } 2051f1fca82eSMikolaj Golub } 2052f1fca82eSMikolaj Golub 2053f1fca82eSMikolaj Golub #ifdef KINFO_VMENTRY_SIZE 2054f1fca82eSMikolaj Golub CTASSERT(sizeof(struct kinfo_vmentry) == KINFO_VMENTRY_SIZE); 2055f1fca82eSMikolaj Golub #endif 2056f1fca82eSMikolaj Golub 2057f1fca82eSMikolaj Golub static void 2058f1fca82eSMikolaj Golub note_procstat_vmmap(void *arg, struct sbuf *sb, size_t *sizep) 2059f1fca82eSMikolaj Golub { 2060f1fca82eSMikolaj Golub struct proc *p; 2061f1fca82eSMikolaj Golub size_t size; 2062e6b95927SConrad Meyer int structsize, vmmap_flags; 2063e6b95927SConrad Meyer 2064e6b95927SConrad Meyer if (coredump_pack_vmmapinfo) 2065e6b95927SConrad Meyer vmmap_flags = KERN_VMMAP_PACK_KINFO; 2066e6b95927SConrad Meyer else 2067e6b95927SConrad Meyer vmmap_flags = 0; 2068f1fca82eSMikolaj Golub 2069f1fca82eSMikolaj Golub p = (struct proc *)arg; 2070e6b95927SConrad Meyer structsize = sizeof(struct kinfo_vmentry); 2071f1fca82eSMikolaj Golub if (sb == NULL) { 2072f1fca82eSMikolaj Golub size = 0; 2073f1fca82eSMikolaj Golub sb = sbuf_new(NULL, NULL, 128, SBUF_FIXEDLEN); 2074f1fca82eSMikolaj Golub sbuf_set_drain(sb, sbuf_drain_count, &size); 2075f1fca82eSMikolaj Golub sbuf_bcat(sb, &structsize, sizeof(structsize)); 2076f1fca82eSMikolaj Golub PROC_LOCK(p); 2077e6b95927SConrad Meyer kern_proc_vmmap_out(p, sb, -1, vmmap_flags); 2078f1fca82eSMikolaj Golub sbuf_finish(sb); 2079f1fca82eSMikolaj Golub sbuf_delete(sb); 2080f1fca82eSMikolaj Golub *sizep = size; 2081f1fca82eSMikolaj Golub } else { 2082f1fca82eSMikolaj Golub sbuf_bcat(sb, &structsize, sizeof(structsize)); 2083f1fca82eSMikolaj Golub PROC_LOCK(p); 2084e6b95927SConrad Meyer kern_proc_vmmap_out(p, sb, *sizep - sizeof(structsize), 2085e6b95927SConrad Meyer vmmap_flags); 2086f1fca82eSMikolaj Golub } 2087f1fca82eSMikolaj Golub } 2088f1fca82eSMikolaj Golub 2089f1fca82eSMikolaj Golub static void 2090f1fca82eSMikolaj Golub note_procstat_groups(void *arg, struct sbuf *sb, size_t *sizep) 2091f1fca82eSMikolaj Golub { 2092f1fca82eSMikolaj Golub struct proc *p; 2093f1fca82eSMikolaj Golub size_t size; 2094f1fca82eSMikolaj Golub int structsize; 2095f1fca82eSMikolaj Golub 2096f1fca82eSMikolaj Golub p = (struct proc *)arg; 2097f1fca82eSMikolaj Golub size = sizeof(structsize) + p->p_ucred->cr_ngroups * sizeof(gid_t); 2098f1fca82eSMikolaj Golub if (sb != NULL) { 2099f1fca82eSMikolaj Golub KASSERT(*sizep == size, ("invalid size")); 2100f1fca82eSMikolaj Golub structsize = sizeof(gid_t); 2101f1fca82eSMikolaj Golub sbuf_bcat(sb, &structsize, sizeof(structsize)); 2102f1fca82eSMikolaj Golub sbuf_bcat(sb, p->p_ucred->cr_groups, p->p_ucred->cr_ngroups * 2103f1fca82eSMikolaj Golub sizeof(gid_t)); 2104f1fca82eSMikolaj Golub } 2105f1fca82eSMikolaj Golub *sizep = size; 2106f1fca82eSMikolaj Golub } 2107f1fca82eSMikolaj Golub 2108f1fca82eSMikolaj Golub static void 2109f1fca82eSMikolaj Golub note_procstat_umask(void *arg, struct sbuf *sb, size_t *sizep) 2110f1fca82eSMikolaj Golub { 2111f1fca82eSMikolaj Golub struct proc *p; 2112f1fca82eSMikolaj Golub size_t size; 2113f1fca82eSMikolaj Golub int structsize; 2114f1fca82eSMikolaj Golub 2115f1fca82eSMikolaj Golub p = (struct proc *)arg; 2116f1fca82eSMikolaj Golub size = sizeof(structsize) + sizeof(p->p_fd->fd_cmask); 2117f1fca82eSMikolaj Golub if (sb != NULL) { 2118f1fca82eSMikolaj Golub KASSERT(*sizep == size, ("invalid size")); 2119f1fca82eSMikolaj Golub structsize = sizeof(p->p_fd->fd_cmask); 2120f1fca82eSMikolaj Golub sbuf_bcat(sb, &structsize, sizeof(structsize)); 2121f1fca82eSMikolaj Golub sbuf_bcat(sb, &p->p_fd->fd_cmask, sizeof(p->p_fd->fd_cmask)); 2122f1fca82eSMikolaj Golub } 2123f1fca82eSMikolaj Golub *sizep = size; 2124f1fca82eSMikolaj Golub } 2125f1fca82eSMikolaj Golub 2126f1fca82eSMikolaj Golub static void 2127f1fca82eSMikolaj Golub note_procstat_rlimit(void *arg, struct sbuf *sb, size_t *sizep) 2128f1fca82eSMikolaj Golub { 2129f1fca82eSMikolaj Golub struct proc *p; 2130f1fca82eSMikolaj Golub struct rlimit rlim[RLIM_NLIMITS]; 2131f1fca82eSMikolaj Golub size_t size; 2132f1fca82eSMikolaj Golub int structsize, i; 2133f1fca82eSMikolaj Golub 2134f1fca82eSMikolaj Golub p = (struct proc *)arg; 2135f1fca82eSMikolaj Golub size = sizeof(structsize) + sizeof(rlim); 2136f1fca82eSMikolaj Golub if (sb != NULL) { 2137f1fca82eSMikolaj Golub KASSERT(*sizep == size, ("invalid size")); 2138f1fca82eSMikolaj Golub structsize = sizeof(rlim); 2139f1fca82eSMikolaj Golub sbuf_bcat(sb, &structsize, sizeof(structsize)); 2140f1fca82eSMikolaj Golub PROC_LOCK(p); 2141f1fca82eSMikolaj Golub for (i = 0; i < RLIM_NLIMITS; i++) 2142f6f6d240SMateusz Guzik lim_rlimit_proc(p, i, &rlim[i]); 2143f1fca82eSMikolaj Golub PROC_UNLOCK(p); 2144f1fca82eSMikolaj Golub sbuf_bcat(sb, rlim, sizeof(rlim)); 2145f1fca82eSMikolaj Golub } 2146f1fca82eSMikolaj Golub *sizep = size; 2147f1fca82eSMikolaj Golub } 2148f1fca82eSMikolaj Golub 2149f1fca82eSMikolaj Golub static void 2150f1fca82eSMikolaj Golub note_procstat_osrel(void *arg, struct sbuf *sb, size_t *sizep) 2151f1fca82eSMikolaj Golub { 2152f1fca82eSMikolaj Golub struct proc *p; 2153f1fca82eSMikolaj Golub size_t size; 2154f1fca82eSMikolaj Golub int structsize; 2155f1fca82eSMikolaj Golub 2156f1fca82eSMikolaj Golub p = (struct proc *)arg; 2157f1fca82eSMikolaj Golub size = sizeof(structsize) + sizeof(p->p_osrel); 2158f1fca82eSMikolaj Golub if (sb != NULL) { 2159f1fca82eSMikolaj Golub KASSERT(*sizep == size, ("invalid size")); 2160f1fca82eSMikolaj Golub structsize = sizeof(p->p_osrel); 2161f1fca82eSMikolaj Golub sbuf_bcat(sb, &structsize, sizeof(structsize)); 2162f1fca82eSMikolaj Golub sbuf_bcat(sb, &p->p_osrel, sizeof(p->p_osrel)); 2163f1fca82eSMikolaj Golub } 2164f1fca82eSMikolaj Golub *sizep = size; 2165f1fca82eSMikolaj Golub } 2166f1fca82eSMikolaj Golub 2167f1fca82eSMikolaj Golub static void 2168f1fca82eSMikolaj Golub __elfN(note_procstat_psstrings)(void *arg, struct sbuf *sb, size_t *sizep) 2169f1fca82eSMikolaj Golub { 2170f1fca82eSMikolaj Golub struct proc *p; 2171f1fca82eSMikolaj Golub elf_ps_strings_t ps_strings; 2172f1fca82eSMikolaj Golub size_t size; 2173f1fca82eSMikolaj Golub int structsize; 2174f1fca82eSMikolaj Golub 2175f1fca82eSMikolaj Golub p = (struct proc *)arg; 2176f1fca82eSMikolaj Golub size = sizeof(structsize) + sizeof(ps_strings); 2177f1fca82eSMikolaj Golub if (sb != NULL) { 2178f1fca82eSMikolaj Golub KASSERT(*sizep == size, ("invalid size")); 2179f1fca82eSMikolaj Golub structsize = sizeof(ps_strings); 2180f1fca82eSMikolaj Golub #if defined(COMPAT_FREEBSD32) && __ELF_WORD_SIZE == 32 2181f1fca82eSMikolaj Golub ps_strings = PTROUT(p->p_sysent->sv_psstrings); 2182f1fca82eSMikolaj Golub #else 2183f1fca82eSMikolaj Golub ps_strings = p->p_sysent->sv_psstrings; 2184f1fca82eSMikolaj Golub #endif 2185f1fca82eSMikolaj Golub sbuf_bcat(sb, &structsize, sizeof(structsize)); 2186f1fca82eSMikolaj Golub sbuf_bcat(sb, &ps_strings, sizeof(ps_strings)); 2187f1fca82eSMikolaj Golub } 2188f1fca82eSMikolaj Golub *sizep = size; 2189f1fca82eSMikolaj Golub } 2190f1fca82eSMikolaj Golub 2191f1fca82eSMikolaj Golub static void 2192f1fca82eSMikolaj Golub __elfN(note_procstat_auxv)(void *arg, struct sbuf *sb, size_t *sizep) 2193f1fca82eSMikolaj Golub { 2194f1fca82eSMikolaj Golub struct proc *p; 2195f1fca82eSMikolaj Golub size_t size; 2196f1fca82eSMikolaj Golub int structsize; 2197f1fca82eSMikolaj Golub 2198f1fca82eSMikolaj Golub p = (struct proc *)arg; 2199f1fca82eSMikolaj Golub if (sb == NULL) { 2200f1fca82eSMikolaj Golub size = 0; 2201f1fca82eSMikolaj Golub sb = sbuf_new(NULL, NULL, 128, SBUF_FIXEDLEN); 2202f1fca82eSMikolaj Golub sbuf_set_drain(sb, sbuf_drain_count, &size); 2203f1fca82eSMikolaj Golub sbuf_bcat(sb, &structsize, sizeof(structsize)); 2204f1fca82eSMikolaj Golub PHOLD(p); 2205f1fca82eSMikolaj Golub proc_getauxv(curthread, p, sb); 2206f1fca82eSMikolaj Golub PRELE(p); 2207f1fca82eSMikolaj Golub sbuf_finish(sb); 2208f1fca82eSMikolaj Golub sbuf_delete(sb); 2209f1fca82eSMikolaj Golub *sizep = size; 2210f1fca82eSMikolaj Golub } else { 2211f1fca82eSMikolaj Golub structsize = sizeof(Elf_Auxinfo); 2212f1fca82eSMikolaj Golub sbuf_bcat(sb, &structsize, sizeof(structsize)); 2213f1fca82eSMikolaj Golub PHOLD(p); 2214f1fca82eSMikolaj Golub proc_getauxv(curthread, p, sb); 2215f1fca82eSMikolaj Golub PRELE(p); 2216f1fca82eSMikolaj Golub } 2217f1fca82eSMikolaj Golub } 2218f1fca82eSMikolaj Golub 221932c01de2SDmitry Chagin static boolean_t 22201a9c7decSKonstantin Belousov __elfN(parse_notes)(struct image_params *imgp, Elf_Brandnote *checknote, 22211a9c7decSKonstantin Belousov int32_t *osrel, const Elf_Phdr *pnote) 222232c01de2SDmitry Chagin { 2223267c52fcSKonstantin Belousov const Elf_Note *note, *note0, *note_end; 222432c01de2SDmitry Chagin const char *note_name; 22256c775eb6SKonstantin Belousov char *buf; 22266c775eb6SKonstantin Belousov int i, error; 22276c775eb6SKonstantin Belousov boolean_t res; 222832c01de2SDmitry Chagin 22296c775eb6SKonstantin Belousov /* We need some limit, might as well use PAGE_SIZE. */ 22306c775eb6SKonstantin Belousov if (pnote == NULL || pnote->p_filesz > PAGE_SIZE) 223132c01de2SDmitry Chagin return (FALSE); 22326c775eb6SKonstantin Belousov ASSERT_VOP_LOCKED(imgp->vp, "parse_notes"); 22336c775eb6SKonstantin Belousov if (pnote->p_offset > PAGE_SIZE || 22346c775eb6SKonstantin Belousov pnote->p_filesz > PAGE_SIZE - pnote->p_offset) { 22356c775eb6SKonstantin Belousov VOP_UNLOCK(imgp->vp, 0); 22366c775eb6SKonstantin Belousov buf = malloc(pnote->p_filesz, M_TEMP, M_WAITOK); 22376c775eb6SKonstantin Belousov vn_lock(imgp->vp, LK_EXCLUSIVE | LK_RETRY); 22386c775eb6SKonstantin Belousov error = vn_rdwr(UIO_READ, imgp->vp, buf, pnote->p_filesz, 22396c775eb6SKonstantin Belousov pnote->p_offset, UIO_SYSSPACE, IO_NODELOCKED, 22406c775eb6SKonstantin Belousov curthread->td_ucred, NOCRED, NULL, curthread); 22416c775eb6SKonstantin Belousov if (error != 0) { 22426c775eb6SKonstantin Belousov uprintf("i/o error PT_NOTE\n"); 22436c775eb6SKonstantin Belousov res = FALSE; 22446c775eb6SKonstantin Belousov goto ret; 22456c775eb6SKonstantin Belousov } 22466c775eb6SKonstantin Belousov note = note0 = (const Elf_Note *)buf; 22476c775eb6SKonstantin Belousov note_end = (const Elf_Note *)(buf + pnote->p_filesz); 22486c775eb6SKonstantin Belousov } else { 22496c775eb6SKonstantin Belousov note = note0 = (const Elf_Note *)(imgp->image_header + 22506c775eb6SKonstantin Belousov pnote->p_offset); 225132c01de2SDmitry Chagin note_end = (const Elf_Note *)(imgp->image_header + 225232c01de2SDmitry Chagin pnote->p_offset + pnote->p_filesz); 22536c775eb6SKonstantin Belousov buf = NULL; 22546c775eb6SKonstantin Belousov } 2255267c52fcSKonstantin Belousov for (i = 0; i < 100 && note >= note0 && note < note_end; i++) { 2256d1ae5c83SKonstantin Belousov if (!aligned(note, Elf32_Addr) || (const char *)note_end - 22576c775eb6SKonstantin Belousov (const char *)note < sizeof(Elf_Note)) { 22586c775eb6SKonstantin Belousov res = FALSE; 22596c775eb6SKonstantin Belousov goto ret; 22606c775eb6SKonstantin Belousov } 226132c01de2SDmitry Chagin if (note->n_namesz != checknote->hdr.n_namesz || 226232c01de2SDmitry Chagin note->n_descsz != checknote->hdr.n_descsz || 226332c01de2SDmitry Chagin note->n_type != checknote->hdr.n_type) 226432c01de2SDmitry Chagin goto nextnote; 226532c01de2SDmitry Chagin note_name = (const char *)(note + 1); 2266d1ae5c83SKonstantin Belousov if (note_name + checknote->hdr.n_namesz >= 2267d1ae5c83SKonstantin Belousov (const char *)note_end || strncmp(checknote->vendor, 2268d1ae5c83SKonstantin Belousov note_name, checknote->hdr.n_namesz) != 0) 226932c01de2SDmitry Chagin goto nextnote; 227032c01de2SDmitry Chagin 227132c01de2SDmitry Chagin /* 227232c01de2SDmitry Chagin * Fetch the osreldate for binary 227332c01de2SDmitry Chagin * from the ELF OSABI-note if necessary. 227432c01de2SDmitry Chagin */ 227589ffc202SBjoern A. Zeeb if ((checknote->flags & BN_TRANSLATE_OSREL) != 0 && 22766c775eb6SKonstantin Belousov checknote->trans_osrel != NULL) { 22776c775eb6SKonstantin Belousov res = checknote->trans_osrel(note, osrel); 22786c775eb6SKonstantin Belousov goto ret; 22796c775eb6SKonstantin Belousov } 22806c775eb6SKonstantin Belousov res = TRUE; 22816c775eb6SKonstantin Belousov goto ret; 228232c01de2SDmitry Chagin nextnote: 228332c01de2SDmitry Chagin note = (const Elf_Note *)((const char *)(note + 1) + 22841b8388cdSMikolaj Golub roundup2(note->n_namesz, ELF_NOTE_ROUNDSIZE) + 22851b8388cdSMikolaj Golub roundup2(note->n_descsz, ELF_NOTE_ROUNDSIZE)); 228632c01de2SDmitry Chagin } 22876c775eb6SKonstantin Belousov res = FALSE; 22886c775eb6SKonstantin Belousov ret: 22896c775eb6SKonstantin Belousov free(buf, M_TEMP); 22906c775eb6SKonstantin Belousov return (res); 229132c01de2SDmitry Chagin } 229232c01de2SDmitry Chagin 229332c01de2SDmitry Chagin /* 22941a9c7decSKonstantin Belousov * Try to find the appropriate ABI-note section for checknote, 22951a9c7decSKonstantin Belousov * fetch the osreldate for binary from the ELF OSABI-note. Only the 22961a9c7decSKonstantin Belousov * first page of the image is searched, the same as for headers. 22971a9c7decSKonstantin Belousov */ 22981a9c7decSKonstantin Belousov static boolean_t 22991a9c7decSKonstantin Belousov __elfN(check_note)(struct image_params *imgp, Elf_Brandnote *checknote, 23001a9c7decSKonstantin Belousov int32_t *osrel) 23011a9c7decSKonstantin Belousov { 23021a9c7decSKonstantin Belousov const Elf_Phdr *phdr; 23031a9c7decSKonstantin Belousov const Elf_Ehdr *hdr; 23041a9c7decSKonstantin Belousov int i; 23051a9c7decSKonstantin Belousov 23061a9c7decSKonstantin Belousov hdr = (const Elf_Ehdr *)imgp->image_header; 23071a9c7decSKonstantin Belousov phdr = (const Elf_Phdr *)(imgp->image_header + hdr->e_phoff); 23081a9c7decSKonstantin Belousov 23091a9c7decSKonstantin Belousov for (i = 0; i < hdr->e_phnum; i++) { 23101a9c7decSKonstantin Belousov if (phdr[i].p_type == PT_NOTE && 23111a9c7decSKonstantin Belousov __elfN(parse_notes)(imgp, checknote, osrel, &phdr[i])) 23121a9c7decSKonstantin Belousov return (TRUE); 23131a9c7decSKonstantin Belousov } 23141a9c7decSKonstantin Belousov return (FALSE); 23151a9c7decSKonstantin Belousov 23161a9c7decSKonstantin Belousov } 23171a9c7decSKonstantin Belousov 23181a9c7decSKonstantin Belousov /* 2319e1743d02SSøren Schmidt * Tell kern_execve.c about it, with a little help from the linker. 2320e1743d02SSøren Schmidt */ 2321a360a43dSJake Burkholder static struct execsw __elfN(execsw) = { 2322a360a43dSJake Burkholder __CONCAT(exec_, __elfN(imgact)), 2323a360a43dSJake Burkholder __XSTRING(__CONCAT(ELF, __ELF_WORD_SIZE)) 2324a360a43dSJake Burkholder }; 2325a360a43dSJake Burkholder EXEC_SET(__CONCAT(elf, __ELF_WORD_SIZE), __elfN(execsw)); 2326e7228204SAlfred Perlstein 2327ed167eaaSKonstantin Belousov static vm_prot_t 2328ed167eaaSKonstantin Belousov __elfN(trans_prot)(Elf_Word flags) 2329ed167eaaSKonstantin Belousov { 2330ed167eaaSKonstantin Belousov vm_prot_t prot; 2331ed167eaaSKonstantin Belousov 2332ed167eaaSKonstantin Belousov prot = 0; 2333ed167eaaSKonstantin Belousov if (flags & PF_X) 2334ed167eaaSKonstantin Belousov prot |= VM_PROT_EXECUTE; 2335ed167eaaSKonstantin Belousov if (flags & PF_W) 2336ed167eaaSKonstantin Belousov prot |= VM_PROT_WRITE; 2337ed167eaaSKonstantin Belousov if (flags & PF_R) 2338ed167eaaSKonstantin Belousov prot |= VM_PROT_READ; 2339676eda08SMarcel Moolenaar #if __ELF_WORD_SIZE == 32 2340e7d939bdSMarcel Moolenaar #if defined(__amd64__) 2341126b36a2SKonstantin Belousov if (i386_read_exec && (flags & PF_R)) 2342676eda08SMarcel Moolenaar prot |= VM_PROT_EXECUTE; 2343676eda08SMarcel Moolenaar #endif 2344676eda08SMarcel Moolenaar #endif 2345ed167eaaSKonstantin Belousov return (prot); 2346ed167eaaSKonstantin Belousov } 2347ed167eaaSKonstantin Belousov 2348ed167eaaSKonstantin Belousov static Elf_Word 2349ed167eaaSKonstantin Belousov __elfN(untrans_prot)(vm_prot_t prot) 2350ed167eaaSKonstantin Belousov { 2351ed167eaaSKonstantin Belousov Elf_Word flags; 2352ed167eaaSKonstantin Belousov 2353ed167eaaSKonstantin Belousov flags = 0; 2354ed167eaaSKonstantin Belousov if (prot & VM_PROT_EXECUTE) 2355ed167eaaSKonstantin Belousov flags |= PF_X; 2356ed167eaaSKonstantin Belousov if (prot & VM_PROT_READ) 2357ed167eaaSKonstantin Belousov flags |= PF_R; 2358ed167eaaSKonstantin Belousov if (prot & VM_PROT_WRITE) 2359ed167eaaSKonstantin Belousov flags |= PF_W; 2360ed167eaaSKonstantin Belousov return (flags); 2361ed167eaaSKonstantin Belousov } 2362