1 /*- 2 * Copyright (c) 2007 Sandvine Incorporated 3 * Copyright (c) 1998 John D. Polstra 4 * All rights reserved. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions 8 * are met: 9 * 1. Redistributions of source code must retain the above copyright 10 * notice, this list of conditions and the following disclaimer. 11 * 2. Redistributions in binary form must reproduce the above copyright 12 * notice, this list of conditions and the following disclaimer in the 13 * documentation and/or other materials provided with the distribution. 14 * 15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND 16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 25 * SUCH DAMAGE. 26 */ 27 28 #include <sys/cdefs.h> 29 __FBSDID("$FreeBSD$"); 30 31 #include <sys/endian.h> 32 #include <sys/param.h> 33 #include <sys/procfs.h> 34 #include <sys/ptrace.h> 35 #include <sys/queue.h> 36 #include <sys/linker_set.h> 37 #include <sys/sbuf.h> 38 #include <sys/sysctl.h> 39 #include <sys/user.h> 40 #include <sys/wait.h> 41 #include <machine/elf.h> 42 #include <vm/vm_param.h> 43 #include <vm/vm.h> 44 #include <vm/pmap.h> 45 #include <vm/vm_map.h> 46 #include <assert.h> 47 #include <err.h> 48 #include <errno.h> 49 #include <fcntl.h> 50 #include <stdbool.h> 51 #include <stdint.h> 52 #include <stdio.h> 53 #include <stdlib.h> 54 #include <string.h> 55 #include <unistd.h> 56 #include <libutil.h> 57 58 #include "extern.h" 59 60 /* 61 * Code for generating ELF core dumps. 62 */ 63 64 typedef void (*segment_callback)(vm_map_entry_t, void *); 65 66 /* Closure for cb_put_phdr(). */ 67 struct phdr_closure { 68 Elf_Phdr *phdr; /* Program header to fill in */ 69 Elf_Off offset; /* Offset of segment in core file */ 70 }; 71 72 /* Closure for cb_size_segment(). */ 73 struct sseg_closure { 74 int count; /* Count of writable segments. */ 75 size_t size; /* Total size of all writable segments. */ 76 }; 77 78 #ifdef ELFCORE_COMPAT_32 79 typedef struct fpreg32 elfcore_fpregset_t; 80 typedef struct reg32 elfcore_gregset_t; 81 typedef struct prpsinfo32 elfcore_prpsinfo_t; 82 typedef struct prstatus32 elfcore_prstatus_t; 83 static void elf_convert_gregset(elfcore_gregset_t *rd, struct reg *rs); 84 static void elf_convert_fpregset(elfcore_fpregset_t *rd, struct fpreg *rs); 85 #else 86 typedef fpregset_t elfcore_fpregset_t; 87 typedef gregset_t elfcore_gregset_t; 88 typedef prpsinfo_t elfcore_prpsinfo_t; 89 typedef prstatus_t elfcore_prstatus_t; 90 #define elf_convert_gregset(d,s) *d = *s 91 #define elf_convert_fpregset(d,s) *d = *s 92 #endif 93 94 typedef void* (*notefunc_t)(void *, size_t *); 95 96 static void cb_put_phdr(vm_map_entry_t, void *); 97 static void cb_size_segment(vm_map_entry_t, void *); 98 static void each_dumpable_segment(vm_map_entry_t, segment_callback, 99 void *closure); 100 static void elf_detach(void); /* atexit() handler. */ 101 static void *elf_note_fpregset(void *, size_t *); 102 static void *elf_note_prpsinfo(void *, size_t *); 103 static void *elf_note_prstatus(void *, size_t *); 104 static void *elf_note_thrmisc(void *, size_t *); 105 #if defined(__i386__) || defined(__amd64__) 106 static void *elf_note_x86_xstate(void *, size_t *); 107 #endif 108 #if defined(__powerpc__) 109 static void *elf_note_powerpc_vmx(void *, size_t *); 110 #endif 111 static void *elf_note_procstat_auxv(void *, size_t *); 112 static void *elf_note_procstat_files(void *, size_t *); 113 static void *elf_note_procstat_groups(void *, size_t *); 114 static void *elf_note_procstat_osrel(void *, size_t *); 115 static void *elf_note_procstat_proc(void *, size_t *); 116 static void *elf_note_procstat_psstrings(void *, size_t *); 117 static void *elf_note_procstat_rlimit(void *, size_t *); 118 static void *elf_note_procstat_umask(void *, size_t *); 119 static void *elf_note_procstat_vmmap(void *, size_t *); 120 static void elf_puthdr(pid_t, vm_map_entry_t, void *, size_t, size_t, size_t, 121 int); 122 static void elf_putnote(int, notefunc_t, void *, struct sbuf *); 123 static void elf_putnotes(pid_t, struct sbuf *, size_t *); 124 static void freemap(vm_map_entry_t); 125 static vm_map_entry_t readmap(pid_t); 126 static void *procstat_sysctl(void *, int, size_t, size_t *sizep); 127 128 static pid_t g_pid; /* Pid being dumped, global for elf_detach */ 129 static int g_status; /* proc status after ptrace attach */ 130 131 static int 132 elf_ident(int efd, pid_t pid __unused, char *binfile __unused) 133 { 134 Elf_Ehdr hdr; 135 int cnt; 136 uint16_t machine; 137 138 cnt = read(efd, &hdr, sizeof(hdr)); 139 if (cnt != sizeof(hdr)) 140 return (0); 141 if (!IS_ELF(hdr)) 142 return (0); 143 switch (hdr.e_ident[EI_DATA]) { 144 case ELFDATA2LSB: 145 machine = le16toh(hdr.e_machine); 146 break; 147 case ELFDATA2MSB: 148 machine = be16toh(hdr.e_machine); 149 break; 150 default: 151 return (0); 152 } 153 if (!ELF_MACHINE_OK(machine)) 154 return (0); 155 156 /* Looks good. */ 157 return (1); 158 } 159 160 static void 161 elf_detach(void) 162 { 163 int sig; 164 165 if (g_pid != 0) { 166 /* 167 * Forward any pending signals. SIGSTOP is generated by ptrace 168 * itself, so ignore it. 169 */ 170 sig = WIFSTOPPED(g_status) ? WSTOPSIG(g_status) : 0; 171 if (sig == SIGSTOP) 172 sig = 0; 173 ptrace(PT_DETACH, g_pid, (caddr_t)1, sig); 174 } 175 } 176 177 /* 178 * Write an ELF coredump for the given pid to the given fd. 179 */ 180 static void 181 elf_coredump(int efd __unused, int fd, pid_t pid) 182 { 183 vm_map_entry_t map; 184 struct sseg_closure seginfo; 185 struct sbuf *sb; 186 void *hdr; 187 size_t hdrsize, notesz, segoff; 188 ssize_t n, old_len; 189 Elf_Phdr *php; 190 int i; 191 192 /* Attach to process to dump. */ 193 g_pid = pid; 194 if (atexit(elf_detach) != 0) 195 err(1, "atexit"); 196 errno = 0; 197 ptrace(PT_ATTACH, pid, NULL, 0); 198 if (errno) 199 err(1, "PT_ATTACH"); 200 if (waitpid(pid, &g_status, 0) == -1) 201 err(1, "waitpid"); 202 203 /* Get the program's memory map. */ 204 map = readmap(pid); 205 206 /* Size the program segments. */ 207 seginfo.count = 0; 208 seginfo.size = 0; 209 each_dumpable_segment(map, cb_size_segment, &seginfo); 210 211 /* 212 * Build the header and the notes using sbuf and write to the file. 213 */ 214 sb = sbuf_new_auto(); 215 hdrsize = sizeof(Elf_Ehdr) + sizeof(Elf_Phdr) * (1 + seginfo.count); 216 if (seginfo.count + 1 >= PN_XNUM) 217 hdrsize += sizeof(Elf_Shdr); 218 /* Start header + notes section. */ 219 sbuf_start_section(sb, NULL); 220 /* Make empty header subsection. */ 221 sbuf_start_section(sb, &old_len); 222 sbuf_putc(sb, 0); 223 sbuf_end_section(sb, old_len, hdrsize, 0); 224 /* Put notes. */ 225 elf_putnotes(pid, sb, ¬esz); 226 /* Align up to a page boundary for the program segments. */ 227 sbuf_end_section(sb, -1, PAGE_SIZE, 0); 228 if (sbuf_finish(sb) != 0) 229 err(1, "sbuf_finish"); 230 hdr = sbuf_data(sb); 231 segoff = sbuf_len(sb); 232 /* Fill in the header. */ 233 elf_puthdr(pid, map, hdr, hdrsize, notesz, segoff, seginfo.count); 234 235 n = write(fd, hdr, segoff); 236 if (n == -1) 237 err(1, "write"); 238 if (n < segoff) 239 errx(1, "short write"); 240 241 /* Write the contents of all of the writable segments. */ 242 php = (Elf_Phdr *)((char *)hdr + sizeof(Elf_Ehdr)) + 1; 243 for (i = 0; i < seginfo.count; i++) { 244 struct ptrace_io_desc iorequest; 245 uintmax_t nleft = php->p_filesz; 246 247 iorequest.piod_op = PIOD_READ_D; 248 iorequest.piod_offs = (caddr_t)(uintptr_t)php->p_vaddr; 249 while (nleft > 0) { 250 char buf[8*1024]; 251 size_t nwant; 252 ssize_t ngot; 253 254 if (nleft > sizeof(buf)) 255 nwant = sizeof buf; 256 else 257 nwant = nleft; 258 iorequest.piod_addr = buf; 259 iorequest.piod_len = nwant; 260 ptrace(PT_IO, pid, (caddr_t)&iorequest, 0); 261 ngot = iorequest.piod_len; 262 if ((size_t)ngot < nwant) 263 errx(1, "short read wanted %zu, got %zd", 264 nwant, ngot); 265 ngot = write(fd, buf, nwant); 266 if (ngot == -1) 267 err(1, "write of segment %d failed", i); 268 if ((size_t)ngot != nwant) 269 errx(1, "short write"); 270 nleft -= nwant; 271 iorequest.piod_offs += ngot; 272 } 273 php++; 274 } 275 sbuf_delete(sb); 276 freemap(map); 277 } 278 279 /* 280 * A callback for each_dumpable_segment() to write out the segment's 281 * program header entry. 282 */ 283 static void 284 cb_put_phdr(vm_map_entry_t entry, void *closure) 285 { 286 struct phdr_closure *phc = (struct phdr_closure *)closure; 287 Elf_Phdr *phdr = phc->phdr; 288 289 phc->offset = round_page(phc->offset); 290 291 phdr->p_type = PT_LOAD; 292 phdr->p_offset = phc->offset; 293 phdr->p_vaddr = entry->start; 294 phdr->p_paddr = 0; 295 phdr->p_filesz = phdr->p_memsz = entry->end - entry->start; 296 phdr->p_align = PAGE_SIZE; 297 phdr->p_flags = 0; 298 if (entry->protection & VM_PROT_READ) 299 phdr->p_flags |= PF_R; 300 if (entry->protection & VM_PROT_WRITE) 301 phdr->p_flags |= PF_W; 302 if (entry->protection & VM_PROT_EXECUTE) 303 phdr->p_flags |= PF_X; 304 305 phc->offset += phdr->p_filesz; 306 phc->phdr++; 307 } 308 309 /* 310 * A callback for each_dumpable_segment() to gather information about 311 * the number of segments and their total size. 312 */ 313 static void 314 cb_size_segment(vm_map_entry_t entry, void *closure) 315 { 316 struct sseg_closure *ssc = (struct sseg_closure *)closure; 317 318 ssc->count++; 319 ssc->size += entry->end - entry->start; 320 } 321 322 /* 323 * For each segment in the given memory map, call the given function 324 * with a pointer to the map entry and some arbitrary caller-supplied 325 * data. 326 */ 327 static void 328 each_dumpable_segment(vm_map_entry_t map, segment_callback func, void *closure) 329 { 330 vm_map_entry_t entry; 331 332 for (entry = map; entry != NULL; entry = entry->next) 333 (*func)(entry, closure); 334 } 335 336 static void 337 elf_putnotes(pid_t pid, struct sbuf *sb, size_t *sizep) 338 { 339 lwpid_t *tids; 340 size_t threads, old_len; 341 ssize_t size; 342 int i; 343 344 errno = 0; 345 threads = ptrace(PT_GETNUMLWPS, pid, NULL, 0); 346 if (errno) 347 err(1, "PT_GETNUMLWPS"); 348 tids = malloc(threads * sizeof(*tids)); 349 if (tids == NULL) 350 errx(1, "out of memory"); 351 errno = 0; 352 ptrace(PT_GETLWPLIST, pid, (void *)tids, threads); 353 if (errno) 354 err(1, "PT_GETLWPLIST"); 355 356 sbuf_start_section(sb, &old_len); 357 elf_putnote(NT_PRPSINFO, elf_note_prpsinfo, &pid, sb); 358 359 for (i = 0; i < threads; ++i) { 360 elf_putnote(NT_PRSTATUS, elf_note_prstatus, tids + i, sb); 361 elf_putnote(NT_FPREGSET, elf_note_fpregset, tids + i, sb); 362 elf_putnote(NT_THRMISC, elf_note_thrmisc, tids + i, sb); 363 #if defined(__i386__) || defined(__amd64__) 364 elf_putnote(NT_X86_XSTATE, elf_note_x86_xstate, tids + i, sb); 365 #endif 366 #if defined(__powerpc__) 367 elf_putnote(NT_PPC_VMX, elf_note_powerpc_vmx, tids + i, sb); 368 #endif 369 } 370 371 #ifndef ELFCORE_COMPAT_32 372 elf_putnote(NT_PROCSTAT_PROC, elf_note_procstat_proc, &pid, sb); 373 elf_putnote(NT_PROCSTAT_FILES, elf_note_procstat_files, &pid, sb); 374 elf_putnote(NT_PROCSTAT_VMMAP, elf_note_procstat_vmmap, &pid, sb); 375 elf_putnote(NT_PROCSTAT_GROUPS, elf_note_procstat_groups, &pid, sb); 376 elf_putnote(NT_PROCSTAT_UMASK, elf_note_procstat_umask, &pid, sb); 377 elf_putnote(NT_PROCSTAT_RLIMIT, elf_note_procstat_rlimit, &pid, sb); 378 elf_putnote(NT_PROCSTAT_OSREL, elf_note_procstat_osrel, &pid, sb); 379 elf_putnote(NT_PROCSTAT_PSSTRINGS, elf_note_procstat_psstrings, &pid, 380 sb); 381 elf_putnote(NT_PROCSTAT_AUXV, elf_note_procstat_auxv, &pid, sb); 382 #endif 383 384 size = sbuf_end_section(sb, old_len, 1, 0); 385 if (size == -1) 386 err(1, "sbuf_end_section"); 387 free(tids); 388 *sizep = size; 389 } 390 391 /* 392 * Emit one note section to sbuf. 393 */ 394 static void 395 elf_putnote(int type, notefunc_t notefunc, void *arg, struct sbuf *sb) 396 { 397 Elf_Note note; 398 size_t descsz; 399 ssize_t old_len; 400 void *desc; 401 402 desc = notefunc(arg, &descsz); 403 note.n_namesz = 8; /* strlen("FreeBSD") + 1 */ 404 note.n_descsz = descsz; 405 note.n_type = type; 406 407 sbuf_bcat(sb, ¬e, sizeof(note)); 408 sbuf_start_section(sb, &old_len); 409 sbuf_bcat(sb, "FreeBSD", note.n_namesz); 410 sbuf_end_section(sb, old_len, sizeof(Elf32_Size), 0); 411 if (descsz == 0) 412 return; 413 sbuf_start_section(sb, &old_len); 414 sbuf_bcat(sb, desc, descsz); 415 sbuf_end_section(sb, old_len, sizeof(Elf32_Size), 0); 416 free(desc); 417 } 418 419 /* 420 * Generate the ELF coredump header. 421 */ 422 static void 423 elf_puthdr(pid_t pid, vm_map_entry_t map, void *hdr, size_t hdrsize, 424 size_t notesz, size_t segoff, int numsegs) 425 { 426 Elf_Ehdr *ehdr; 427 Elf_Phdr *phdr; 428 Elf_Shdr *shdr; 429 struct phdr_closure phc; 430 431 ehdr = (Elf_Ehdr *)hdr; 432 433 ehdr->e_ident[EI_MAG0] = ELFMAG0; 434 ehdr->e_ident[EI_MAG1] = ELFMAG1; 435 ehdr->e_ident[EI_MAG2] = ELFMAG2; 436 ehdr->e_ident[EI_MAG3] = ELFMAG3; 437 ehdr->e_ident[EI_CLASS] = ELF_CLASS; 438 ehdr->e_ident[EI_DATA] = ELF_DATA; 439 ehdr->e_ident[EI_VERSION] = EV_CURRENT; 440 ehdr->e_ident[EI_OSABI] = ELFOSABI_FREEBSD; 441 ehdr->e_ident[EI_ABIVERSION] = 0; 442 ehdr->e_ident[EI_PAD] = 0; 443 ehdr->e_type = ET_CORE; 444 ehdr->e_machine = ELF_ARCH; 445 ehdr->e_version = EV_CURRENT; 446 ehdr->e_entry = 0; 447 ehdr->e_phoff = sizeof(Elf_Ehdr); 448 ehdr->e_flags = 0; 449 ehdr->e_ehsize = sizeof(Elf_Ehdr); 450 ehdr->e_phentsize = sizeof(Elf_Phdr); 451 ehdr->e_shentsize = sizeof(Elf_Shdr); 452 ehdr->e_shstrndx = SHN_UNDEF; 453 if (numsegs + 1 < PN_XNUM) { 454 ehdr->e_phnum = numsegs + 1; 455 ehdr->e_shnum = 0; 456 } else { 457 ehdr->e_phnum = PN_XNUM; 458 ehdr->e_shnum = 1; 459 460 ehdr->e_shoff = ehdr->e_phoff + 461 (numsegs + 1) * ehdr->e_phentsize; 462 463 shdr = (Elf_Shdr *)((char *)hdr + ehdr->e_shoff); 464 memset(shdr, 0, sizeof(*shdr)); 465 /* 466 * A special first section is used to hold large segment and 467 * section counts. This was proposed by Sun Microsystems in 468 * Solaris and has been adopted by Linux; the standard ELF 469 * tools are already familiar with the technique. 470 * 471 * See table 7-7 of the Solaris "Linker and Libraries Guide" 472 * (or 12-7 depending on the version of the document) for more 473 * details. 474 */ 475 shdr->sh_type = SHT_NULL; 476 shdr->sh_size = ehdr->e_shnum; 477 shdr->sh_link = ehdr->e_shstrndx; 478 shdr->sh_info = numsegs + 1; 479 } 480 481 /* 482 * Fill in the program header entries. 483 */ 484 phdr = (Elf_Phdr *)((char *)hdr + ehdr->e_phoff); 485 486 /* The note segement. */ 487 phdr->p_type = PT_NOTE; 488 phdr->p_offset = hdrsize; 489 phdr->p_vaddr = 0; 490 phdr->p_paddr = 0; 491 phdr->p_filesz = notesz; 492 phdr->p_memsz = 0; 493 phdr->p_flags = PF_R; 494 phdr->p_align = sizeof(Elf32_Size); 495 phdr++; 496 497 /* All the writable segments from the program. */ 498 phc.phdr = phdr; 499 phc.offset = segoff; 500 each_dumpable_segment(map, cb_put_phdr, &phc); 501 } 502 503 /* 504 * Free the memory map. 505 */ 506 static void 507 freemap(vm_map_entry_t map) 508 { 509 510 while (map != NULL) { 511 vm_map_entry_t next = map->next; 512 free(map); 513 map = next; 514 } 515 } 516 517 /* 518 * Read the process's memory map using kinfo_getvmmap(), and return a list of 519 * VM map entries. Only the non-device read/writable segments are 520 * returned. The map entries in the list aren't fully filled in; only 521 * the items we need are present. 522 */ 523 static vm_map_entry_t 524 readmap(pid_t pid) 525 { 526 vm_map_entry_t ent, *linkp, map; 527 struct kinfo_vmentry *vmentl, *kve; 528 int i, nitems; 529 530 vmentl = kinfo_getvmmap(pid, &nitems); 531 if (vmentl == NULL) 532 err(1, "cannot retrieve mappings for %u process", pid); 533 534 map = NULL; 535 linkp = ↦ 536 for (i = 0; i < nitems; i++) { 537 kve = &vmentl[i]; 538 539 /* 540 * Ignore 'malformed' segments or ones representing memory 541 * mapping with MAP_NOCORE on. 542 * If the 'full' support is disabled, just dump the most 543 * meaningful data segments. 544 */ 545 if ((kve->kve_protection & KVME_PROT_READ) == 0 || 546 (kve->kve_flags & KVME_FLAG_NOCOREDUMP) != 0 || 547 kve->kve_type == KVME_TYPE_DEAD || 548 kve->kve_type == KVME_TYPE_UNKNOWN || 549 ((pflags & PFLAGS_FULL) == 0 && 550 kve->kve_type != KVME_TYPE_DEFAULT && 551 kve->kve_type != KVME_TYPE_VNODE && 552 kve->kve_type != KVME_TYPE_SWAP && 553 kve->kve_type != KVME_TYPE_PHYS)) 554 continue; 555 556 ent = calloc(1, sizeof(*ent)); 557 if (ent == NULL) 558 errx(1, "out of memory"); 559 ent->start = (vm_offset_t)kve->kve_start; 560 ent->end = (vm_offset_t)kve->kve_end; 561 ent->protection = VM_PROT_READ | VM_PROT_WRITE; 562 if ((kve->kve_protection & KVME_PROT_EXEC) != 0) 563 ent->protection |= VM_PROT_EXECUTE; 564 565 *linkp = ent; 566 linkp = &ent->next; 567 } 568 free(vmentl); 569 return (map); 570 } 571 572 /* 573 * Miscellaneous note out functions. 574 */ 575 576 static void * 577 elf_note_prpsinfo(void *arg, size_t *sizep) 578 { 579 char *cp, *end; 580 pid_t pid; 581 elfcore_prpsinfo_t *psinfo; 582 struct kinfo_proc kip; 583 size_t len; 584 int name[4]; 585 586 pid = *(pid_t *)arg; 587 psinfo = calloc(1, sizeof(*psinfo)); 588 if (psinfo == NULL) 589 errx(1, "out of memory"); 590 psinfo->pr_version = PRPSINFO_VERSION; 591 psinfo->pr_psinfosz = sizeof(*psinfo); 592 593 name[0] = CTL_KERN; 594 name[1] = KERN_PROC; 595 name[2] = KERN_PROC_PID; 596 name[3] = pid; 597 len = sizeof(kip); 598 if (sysctl(name, 4, &kip, &len, NULL, 0) == -1) 599 err(1, "kern.proc.pid.%u", pid); 600 if (kip.ki_pid != pid) 601 err(1, "kern.proc.pid.%u", pid); 602 strlcpy(psinfo->pr_fname, kip.ki_comm, sizeof(psinfo->pr_fname)); 603 name[2] = KERN_PROC_ARGS; 604 len = sizeof(psinfo->pr_psargs) - 1; 605 if (sysctl(name, 4, psinfo->pr_psargs, &len, NULL, 0) == 0 && len > 0) { 606 cp = psinfo->pr_psargs; 607 end = cp + len - 1; 608 for (;;) { 609 cp = memchr(cp, '\0', end - cp); 610 if (cp == NULL) 611 break; 612 *cp = ' '; 613 } 614 } else 615 strlcpy(psinfo->pr_psargs, kip.ki_comm, 616 sizeof(psinfo->pr_psargs)); 617 psinfo->pr_pid = pid; 618 619 *sizep = sizeof(*psinfo); 620 return (psinfo); 621 } 622 623 static void * 624 elf_note_prstatus(void *arg, size_t *sizep) 625 { 626 lwpid_t tid; 627 elfcore_prstatus_t *status; 628 struct reg greg; 629 630 tid = *(lwpid_t *)arg; 631 status = calloc(1, sizeof(*status)); 632 if (status == NULL) 633 errx(1, "out of memory"); 634 status->pr_version = PRSTATUS_VERSION; 635 status->pr_statussz = sizeof(*status); 636 status->pr_gregsetsz = sizeof(elfcore_gregset_t); 637 status->pr_fpregsetsz = sizeof(elfcore_fpregset_t); 638 status->pr_osreldate = __FreeBSD_version; 639 status->pr_pid = tid; 640 ptrace(PT_GETREGS, tid, (void *)&greg, 0); 641 elf_convert_gregset(&status->pr_reg, &greg); 642 643 *sizep = sizeof(*status); 644 return (status); 645 } 646 647 static void * 648 elf_note_fpregset(void *arg, size_t *sizep) 649 { 650 lwpid_t tid; 651 elfcore_fpregset_t *fpregset; 652 fpregset_t fpreg; 653 654 tid = *(lwpid_t *)arg; 655 fpregset = calloc(1, sizeof(*fpregset)); 656 if (fpregset == NULL) 657 errx(1, "out of memory"); 658 ptrace(PT_GETFPREGS, tid, (void *)&fpreg, 0); 659 elf_convert_fpregset(fpregset, &fpreg); 660 661 *sizep = sizeof(*fpregset); 662 return (fpregset); 663 } 664 665 static void * 666 elf_note_thrmisc(void *arg, size_t *sizep) 667 { 668 lwpid_t tid; 669 struct ptrace_lwpinfo lwpinfo; 670 thrmisc_t *thrmisc; 671 672 tid = *(lwpid_t *)arg; 673 thrmisc = calloc(1, sizeof(*thrmisc)); 674 if (thrmisc == NULL) 675 errx(1, "out of memory"); 676 ptrace(PT_LWPINFO, tid, (void *)&lwpinfo, 677 sizeof(lwpinfo)); 678 memset(&thrmisc->_pad, 0, sizeof(thrmisc->_pad)); 679 strcpy(thrmisc->pr_tname, lwpinfo.pl_tdname); 680 681 *sizep = sizeof(*thrmisc); 682 return (thrmisc); 683 } 684 685 #if defined(__i386__) || defined(__amd64__) 686 static void * 687 elf_note_x86_xstate(void *arg, size_t *sizep) 688 { 689 lwpid_t tid; 690 char *xstate; 691 static bool xsave_checked = false; 692 static struct ptrace_xstate_info info; 693 694 tid = *(lwpid_t *)arg; 695 if (!xsave_checked) { 696 if (ptrace(PT_GETXSTATE_INFO, tid, (void *)&info, 697 sizeof(info)) != 0) 698 info.xsave_len = 0; 699 xsave_checked = true; 700 } 701 if (info.xsave_len == 0) { 702 *sizep = 0; 703 return (NULL); 704 } 705 xstate = calloc(1, info.xsave_len); 706 ptrace(PT_GETXSTATE, tid, xstate, 0); 707 *(uint64_t *)(xstate + X86_XSTATE_XCR0_OFFSET) = info.xsave_mask; 708 *sizep = info.xsave_len; 709 return (xstate); 710 } 711 #endif 712 713 #if defined(__powerpc__) 714 static void * 715 elf_note_powerpc_vmx(void *arg, size_t *sizep) 716 { 717 lwpid_t tid; 718 struct vmxreg *vmx; 719 static bool has_vmx = true; 720 struct vmxreg info; 721 722 tid = *(lwpid_t *)arg; 723 if (has_vmx) { 724 if (ptrace(PT_GETVRREGS, tid, (void *)&info, 725 sizeof(info)) != 0) 726 has_vmx = false; 727 } 728 if (!has_vmx) { 729 *sizep = 0; 730 return (NULL); 731 } 732 vmx = calloc(1, sizeof(*vmx)); 733 memcpy(vmx, &info, sizeof(*vmx)); 734 *sizep = sizeof(*vmx); 735 return (vmx); 736 } 737 #endif 738 739 static void * 740 procstat_sysctl(void *arg, int what, size_t structsz, size_t *sizep) 741 { 742 size_t len; 743 pid_t pid; 744 int name[4], structsize; 745 void *buf, *p; 746 747 pid = *(pid_t *)arg; 748 structsize = structsz; 749 name[0] = CTL_KERN; 750 name[1] = KERN_PROC; 751 name[2] = what; 752 name[3] = pid; 753 len = 0; 754 if (sysctl(name, 4, NULL, &len, NULL, 0) == -1) 755 err(1, "kern.proc.%d.%u", what, pid); 756 buf = calloc(1, sizeof(structsize) + len * 4 / 3); 757 if (buf == NULL) 758 errx(1, "out of memory"); 759 bcopy(&structsize, buf, sizeof(structsize)); 760 p = (char *)buf + sizeof(structsize); 761 if (sysctl(name, 4, p, &len, NULL, 0) == -1) 762 err(1, "kern.proc.%d.%u", what, pid); 763 764 *sizep = sizeof(structsize) + len; 765 return (buf); 766 } 767 768 static void * 769 elf_note_procstat_proc(void *arg, size_t *sizep) 770 { 771 772 return (procstat_sysctl(arg, KERN_PROC_PID | KERN_PROC_INC_THREAD, 773 sizeof(struct kinfo_proc), sizep)); 774 } 775 776 static void * 777 elf_note_procstat_files(void *arg, size_t *sizep) 778 { 779 780 return (procstat_sysctl(arg, KERN_PROC_FILEDESC, 781 sizeof(struct kinfo_file), sizep)); 782 } 783 784 static void * 785 elf_note_procstat_vmmap(void *arg, size_t *sizep) 786 { 787 788 return (procstat_sysctl(arg, KERN_PROC_VMMAP, 789 sizeof(struct kinfo_vmentry), sizep)); 790 } 791 792 static void * 793 elf_note_procstat_groups(void *arg, size_t *sizep) 794 { 795 796 return (procstat_sysctl(arg, KERN_PROC_GROUPS, sizeof(gid_t), sizep)); 797 } 798 799 static void * 800 elf_note_procstat_umask(void *arg, size_t *sizep) 801 { 802 803 return (procstat_sysctl(arg, KERN_PROC_UMASK, sizeof(u_short), sizep)); 804 } 805 806 static void * 807 elf_note_procstat_osrel(void *arg, size_t *sizep) 808 { 809 810 return (procstat_sysctl(arg, KERN_PROC_OSREL, sizeof(int), sizep)); 811 } 812 813 static void * 814 elf_note_procstat_psstrings(void *arg, size_t *sizep) 815 { 816 817 return (procstat_sysctl(arg, KERN_PROC_PS_STRINGS, 818 sizeof(vm_offset_t), sizep)); 819 } 820 821 static void * 822 elf_note_procstat_auxv(void *arg, size_t *sizep) 823 { 824 825 return (procstat_sysctl(arg, KERN_PROC_AUXV, 826 sizeof(Elf_Auxinfo), sizep)); 827 } 828 829 static void * 830 elf_note_procstat_rlimit(void *arg, size_t *sizep) 831 { 832 pid_t pid; 833 size_t len; 834 int i, name[5], structsize; 835 void *buf, *p; 836 837 pid = *(pid_t *)arg; 838 structsize = sizeof(struct rlimit) * RLIM_NLIMITS; 839 buf = calloc(1, sizeof(structsize) + structsize); 840 if (buf == NULL) 841 errx(1, "out of memory"); 842 bcopy(&structsize, buf, sizeof(structsize)); 843 p = (char *)buf + sizeof(structsize); 844 name[0] = CTL_KERN; 845 name[1] = KERN_PROC; 846 name[2] = KERN_PROC_RLIMIT; 847 name[3] = pid; 848 len = sizeof(struct rlimit); 849 for (i = 0; i < RLIM_NLIMITS; i++) { 850 name[4] = i; 851 if (sysctl(name, 5, p, &len, NULL, 0) == -1) 852 err(1, "kern.proc.rlimit.%u", pid); 853 if (len != sizeof(struct rlimit)) 854 errx(1, "kern.proc.rlimit.%u: short read", pid); 855 p += len; 856 } 857 858 *sizep = sizeof(structsize) + structsize; 859 return (buf); 860 } 861 862 struct dumpers __elfN(dump) = { elf_ident, elf_coredump }; 863 TEXT_SET(dumpset, __elfN(dump)); 864