libprocstat.c revision 223269
1139749Simp/*- 2116491Sharti * Copyright (c) 2009 Stanislav Sedov <stas@FreeBSD.org> 3116491Sharti * Copyright (c) 1988, 1993 4116491Sharti * The Regents of the University of California. All rights reserved. 5116491Sharti * 6116491Sharti * Redistribution and use in source and binary forms, with or without 7116491Sharti * modification, are permitted provided that the following conditions 8116491Sharti * are met: 9116491Sharti * 1. Redistributions of source code must retain the above copyright 10116491Sharti * notice, this list of conditions and the following disclaimer. 11116491Sharti * 2. Redistributions in binary form must reproduce the above copyright 12116491Sharti * notice, this list of conditions and the following disclaimer in the 13116491Sharti * documentation and/or other materials provided with the distribution. 14116491Sharti * 3. All advertising materials mentioning features or use of this software 15116491Sharti * must display the following acknowledgement: 16116491Sharti * This product includes software developed by the University of 17116491Sharti * California, Berkeley and its contributors. 18116491Sharti * 4. Neither the name of the University nor the names of its contributors 19116491Sharti * may be used to endorse or promote products derived from this software 20116491Sharti * without specific prior written permission. 21116491Sharti * 22116491Sharti * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND 23116491Sharti * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 24116491Sharti * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 25116491Sharti * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE 26116491Sharti * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 27116491Sharti * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 28116491Sharti * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 29116491Sharti * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 30116491Sharti * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 31116491Sharti * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 32116491Sharti * SUCH DAMAGE. 33116491Sharti */ 34116519Sharti 35116519Sharti#include <sys/cdefs.h> 36116519Sharti__FBSDID("$FreeBSD: head/lib/libprocstat/libprocstat.c 223269 2011-06-18 21:29:25Z jilles $"); 37116491Sharti 38116491Sharti#include <sys/param.h> 39116491Sharti#include <sys/time.h> 40116491Sharti#include <sys/proc.h> 41116491Sharti#include <sys/user.h> 42116491Sharti#include <sys/stat.h> 43116491Sharti#include <sys/vnode.h> 44116491Sharti#include <sys/socket.h> 45116491Sharti#include <sys/socketvar.h> 46116491Sharti#include <sys/domain.h> 47116491Sharti#include <sys/protosw.h> 48116491Sharti#include <sys/un.h> 49116491Sharti#include <sys/unpcb.h> 50116491Sharti#include <sys/sysctl.h> 51116491Sharti#include <sys/tty.h> 52116491Sharti#include <sys/filedesc.h> 53116491Sharti#include <sys/queue.h> 54116491Sharti#define _WANT_FILE 55116491Sharti#include <sys/file.h> 56116491Sharti#include <sys/conf.h> 57116491Sharti#define _KERNEL 58116491Sharti#include <sys/mount.h> 59116491Sharti#include <sys/pipe.h> 60116491Sharti#include <ufs/ufs/quota.h> 61116491Sharti#include <ufs/ufs/inode.h> 62116491Sharti#include <fs/devfs/devfs.h> 63116491Sharti#include <fs/devfs/devfs_int.h> 64116491Sharti#undef _KERNEL 65116491Sharti#include <nfs/nfsproto.h> 66116491Sharti#include <nfsclient/nfs.h> 67116491Sharti#include <nfsclient/nfsnode.h> 68116491Sharti 69116491Sharti#include <vm/vm.h> 70116491Sharti#include <vm/vm_map.h> 71116491Sharti#include <vm/vm_object.h> 72116491Sharti 73119280Simp#include <net/route.h> 74119280Simp#include <netinet/in.h> 75116491Sharti#include <netinet/in_systm.h> 76116491Sharti#include <netinet/ip.h> 77116491Sharti#include <netinet/in_pcb.h> 78116491Sharti 79116491Sharti#include <assert.h> 80116491Sharti#include <ctype.h> 81122112Sharti#include <err.h> 82116491Sharti#include <fcntl.h> 83122112Sharti#include <kvm.h> 84122112Sharti#include <libutil.h> 85122112Sharti#include <limits.h> 86122112Sharti#include <paths.h> 87122112Sharti#include <pwd.h> 88122112Sharti#include <stdio.h> 89122112Sharti#include <stdlib.h> 90122112Sharti#include <stddef.h> 91122112Sharti#include <string.h> 92122112Sharti#include <unistd.h> 93122112Sharti#include <netdb.h> 94122112Sharti 95122112Sharti#include <libprocstat.h> 96122112Sharti#include "libprocstat_internal.h" 97146617Sharti#include "common_kvm.h" 98122112Sharti 99122112Shartiint statfs(const char *, struct statfs *); /* XXX */ 100122112Sharti 101122112Sharti#define PROCSTAT_KVM 1 102122112Sharti#define PROCSTAT_SYSCTL 2 103122112Sharti 104122112Shartistatic char *getmnton(kvm_t *kd, struct mount *m); 105122112Shartistatic struct filestat_list *procstat_getfiles_kvm( 106122112Sharti struct procstat *procstat, struct kinfo_proc *kp, int mmapped); 107116491Shartistatic struct filestat_list *procstat_getfiles_sysctl( 108116491Sharti struct procstat *procstat, struct kinfo_proc *kp, int mmapped); 109116491Shartistatic int procstat_get_pipe_info_sysctl(struct filestat *fst, 110116491Sharti struct pipestat *pipe, char *errbuf); 111116491Shartistatic int procstat_get_pipe_info_kvm(kvm_t *kd, struct filestat *fst, 112116491Sharti struct pipestat *pipe, char *errbuf); 113116491Shartistatic int procstat_get_pts_info_sysctl(struct filestat *fst, 114116491Sharti struct ptsstat *pts, char *errbuf); 115116491Shartistatic int procstat_get_pts_info_kvm(kvm_t *kd, struct filestat *fst, 116116491Sharti struct ptsstat *pts, char *errbuf); 117116491Shartistatic int procstat_get_socket_info_sysctl(struct filestat *fst, 118116491Sharti struct sockstat *sock, char *errbuf); 119116491Shartistatic int procstat_get_socket_info_kvm(kvm_t *kd, struct filestat *fst, 120116491Sharti struct sockstat *sock, char *errbuf); 121116491Shartistatic int to_filestat_flags(int flags); 122116491Shartistatic int procstat_get_vnode_info_kvm(kvm_t *kd, struct filestat *fst, 123116491Sharti struct vnstat *vn, char *errbuf); 124116491Shartistatic int procstat_get_vnode_info_sysctl(struct filestat *fst, 125116491Sharti struct vnstat *vn, char *errbuf); 126116491Shartistatic int vntype2psfsttype(int type); 127116491Sharti 128116491Shartivoid 129116491Shartiprocstat_close(struct procstat *procstat) 130116491Sharti{ 131116491Sharti 132116491Sharti assert(procstat); 133116491Sharti if (procstat->type == PROCSTAT_KVM) 134116491Sharti kvm_close(procstat->kd); 135116491Sharti free(procstat); 136116491Sharti} 137116491Sharti 138116491Shartistruct procstat * 139116491Shartiprocstat_open_sysctl(void) 140116491Sharti{ 141116491Sharti struct procstat *procstat; 142116491Sharti 143116491Sharti procstat = calloc(1, sizeof(*procstat)); 144116491Sharti if (procstat == NULL) { 145116491Sharti warn("malloc()"); 146116491Sharti return (NULL); 147116491Sharti } 148116491Sharti procstat->type = PROCSTAT_SYSCTL; 149116491Sharti return (procstat); 150116491Sharti} 151116491Sharti 152122112Shartistruct procstat * 153116491Shartiprocstat_open_kvm(const char *nlistf, const char *memf) 154116491Sharti{ 155116491Sharti struct procstat *procstat; 156116491Sharti kvm_t *kd; 157116491Sharti char buf[_POSIX2_LINE_MAX]; 158116491Sharti 159116491Sharti procstat = calloc(1, sizeof(*procstat)); 160116491Sharti if (procstat == NULL) { 161116491Sharti warn("malloc()"); 162116491Sharti return (NULL); 163116491Sharti } 164116491Sharti kd = kvm_openfiles(nlistf, memf, NULL, O_RDONLY, buf); 165116491Sharti if (kd == NULL) { 166116491Sharti warnx("kvm_openfiles(): %s", buf); 167116491Sharti free(procstat); 168116491Sharti return (NULL); 169116491Sharti } 170116491Sharti procstat->type = PROCSTAT_KVM; 171116491Sharti procstat->kd = kd; 172116491Sharti return (procstat); 173116491Sharti} 174116491Sharti 175116491Shartistruct kinfo_proc * 176116491Shartiprocstat_getprocs(struct procstat *procstat, int what, int arg, 177116491Sharti unsigned int *count) 178116491Sharti{ 179116491Sharti struct kinfo_proc *p0, *p; 180116491Sharti size_t len; 181116491Sharti int name[4]; 182116491Sharti int error; 183116491Sharti 184116491Sharti assert(procstat); 185116491Sharti assert(count); 186116491Sharti p = NULL; 187116491Sharti if (procstat->type == PROCSTAT_KVM) { 188116491Sharti p0 = kvm_getprocs(procstat->kd, what, arg, count); 189116491Sharti if (p0 == NULL || count == 0) 190116491Sharti return (NULL); 191116491Sharti len = *count * sizeof(*p); 192116491Sharti p = malloc(len); 193116491Sharti if (p == NULL) { 194116491Sharti warnx("malloc(%zu)", len); 195116491Sharti goto fail; 196147256Sbrooks } 197116491Sharti bcopy(p0, p, len); 198116491Sharti return (p); 199116491Sharti } else if (procstat->type == PROCSTAT_SYSCTL) { 200116491Sharti len = 0; 201116491Sharti name[0] = CTL_KERN; 202116491Sharti name[1] = KERN_PROC; 203116491Sharti name[2] = what; 204116491Sharti name[3] = arg; 205116491Sharti error = sysctl(name, 4, NULL, &len, NULL, 0); 206116491Sharti if (error < 0 && errno != EPERM) { 207116491Sharti warn("sysctl(kern.proc)"); 208116491Sharti goto fail; 209116491Sharti } 210116491Sharti if (len == 0) { 211116491Sharti warnx("no processes?"); 212116491Sharti goto fail; 213116491Sharti } 214116491Sharti p = malloc(len); 215116491Sharti if (p == NULL) { 216116491Sharti warnx("malloc(%zu)", len); 217116491Sharti goto fail; 218116491Sharti } 219116491Sharti error = sysctl(name, 4, p, &len, NULL, 0); 220116491Sharti if (error < 0 && errno != EPERM) { 221116491Sharti warn("sysctl(kern.proc)"); 222116491Sharti goto fail; 223116491Sharti } 224116491Sharti /* Perform simple consistency checks. */ 225116491Sharti if ((len % sizeof(*p)) != 0 || p->ki_structsize != sizeof(*p)) { 226116491Sharti warnx("kinfo_proc structure size mismatch"); 227116491Sharti goto fail; 228116491Sharti } 229116491Sharti *count = len / sizeof(*p); 230116491Sharti return (p); 231116491Sharti } else { 232116491Sharti warnx("unknown access method"); 233116491Sharti return (NULL); 234116491Sharti } 235116491Shartifail: 236116491Sharti if (p) 237116491Sharti free(p); 238116491Sharti return (NULL); 239116491Sharti} 240116491Sharti 241116491Shartivoid 242116491Shartiprocstat_freeprocs(struct procstat *procstat __unused, struct kinfo_proc *p) 243116491Sharti{ 244116491Sharti 245116491Sharti if (p != NULL) 246116491Sharti free(p); 247116491Sharti p = NULL; 248116491Sharti} 249116491Sharti 250116491Shartistruct filestat_list * 251116491Shartiprocstat_getfiles(struct procstat *procstat, struct kinfo_proc *kp, int mmapped) 252116491Sharti{ 253116491Sharti 254116491Sharti if (procstat->type == PROCSTAT_SYSCTL) 255116491Sharti return (procstat_getfiles_sysctl(procstat, kp, mmapped)); 256116491Sharti else if (procstat->type == PROCSTAT_KVM) 257116491Sharti return (procstat_getfiles_kvm(procstat, kp, mmapped)); 258116491Sharti else 259116491Sharti return (NULL); 260116491Sharti} 261116491Sharti 262116491Shartivoid 263116491Shartiprocstat_freefiles(struct procstat *procstat, struct filestat_list *head) 264116491Sharti{ 265147256Sbrooks struct filestat *fst, *tmp; 266116491Sharti 267116491Sharti STAILQ_FOREACH_SAFE(fst, head, next, tmp) { 268116491Sharti if (fst->fs_path != NULL) 269116491Sharti free(fst->fs_path); 270116491Sharti free(fst); 271116491Sharti } 272116491Sharti free(head); 273116491Sharti if (procstat->vmentries != NULL) { 274116491Sharti free (procstat->vmentries); 275116491Sharti procstat->vmentries = NULL; 276116491Sharti } 277116491Sharti if (procstat->files != NULL) { 278116491Sharti free (procstat->files); 279116491Sharti procstat->files = NULL; 280147256Sbrooks } 281118170Sharti} 282116491Sharti 283116491Shartistatic struct filestat * 284116491Shartifilestat_new_entry(void *typedep, int type, int fd, int fflags, int uflags, 285116491Sharti int refcount, off_t offset, char *path) 286116491Sharti{ 287116491Sharti struct filestat *entry; 288116491Sharti 289116491Sharti entry = calloc(1, sizeof(*entry)); 290116491Sharti if (entry == NULL) { 291116491Sharti warn("malloc()"); 292116491Sharti return (NULL); 293116491Sharti } 294116491Sharti entry->fs_typedep = typedep; 295116491Sharti entry->fs_fflags = fflags; 296116491Sharti entry->fs_uflags = uflags; 297116491Sharti entry->fs_fd = fd; 298116491Sharti entry->fs_type = type; 299116491Sharti entry->fs_ref_count = refcount; 300116491Sharti entry->fs_offset = offset; 301116491Sharti entry->fs_path = path; 302116491Sharti return (entry); 303116491Sharti} 304116491Sharti 305116491Shartistatic struct vnode * 306116491Shartigetctty(kvm_t *kd, struct kinfo_proc *kp) 307116491Sharti{ 308116491Sharti struct pgrp pgrp; 309116491Sharti struct proc proc; 310116491Sharti struct session sess; 311116491Sharti int error; 312116491Sharti 313116491Sharti assert(kp); 314116491Sharti error = kvm_read_all(kd, (unsigned long)kp->ki_paddr, &proc, 315116491Sharti sizeof(proc)); 316116491Sharti if (error == 0) { 317116491Sharti warnx("can't read proc struct at %p for pid %d", 318116491Sharti kp->ki_paddr, kp->ki_pid); 319116491Sharti return (NULL); 320116491Sharti } 321116491Sharti if (proc.p_pgrp == NULL) 322116491Sharti return (NULL); 323116491Sharti error = kvm_read_all(kd, (unsigned long)proc.p_pgrp, &pgrp, 324116491Sharti sizeof(pgrp)); 325116491Sharti if (error == 0) { 326116491Sharti warnx("can't read pgrp struct at %p for pid %d", 327116491Sharti proc.p_pgrp, kp->ki_pid); 328116491Sharti return (NULL); 329116491Sharti } 330116491Sharti error = kvm_read_all(kd, (unsigned long)pgrp.pg_session, &sess, 331116491Sharti sizeof(sess)); 332116491Sharti if (error == 0) { 333116491Sharti warnx("can't read session struct at %p for pid %d", 334116491Sharti pgrp.pg_session, kp->ki_pid); 335116491Sharti return (NULL); 336116491Sharti } 337116491Sharti return (sess.s_ttyvp); 338116491Sharti} 339116491Sharti 340116491Shartistatic struct filestat_list * 341116491Shartiprocstat_getfiles_kvm(struct procstat *procstat, struct kinfo_proc *kp, int mmapped) 342116491Sharti{ 343116491Sharti struct file file; 344116491Sharti struct filedesc filed; 345116491Sharti struct vm_map_entry vmentry; 346116491Sharti struct vm_object object; 347116491Sharti struct vmspace vmspace; 348116491Sharti vm_map_entry_t entryp; 349116491Sharti vm_map_t map; 350116491Sharti vm_object_t objp; 351116491Sharti struct vnode *vp; 352116491Sharti struct file **ofiles; 353116491Sharti struct filestat *entry; 354116491Sharti struct filestat_list *head; 355116491Sharti kvm_t *kd; 356116491Sharti void *data; 357116491Sharti int i, fflags; 358116491Sharti int prot, type; 359116491Sharti unsigned int nfiles; 360116491Sharti 361116491Sharti assert(procstat); 362116491Sharti kd = procstat->kd; 363116491Sharti if (kd == NULL) 364116491Sharti return (NULL); 365116491Sharti if (kp->ki_fd == NULL) 366116491Sharti return (NULL); 367116491Sharti if (!kvm_read_all(kd, (unsigned long)kp->ki_fd, &filed, 368116491Sharti sizeof(filed))) { 369116491Sharti warnx("can't read filedesc at %p", (void *)kp->ki_fd); 370116491Sharti return (NULL); 371116491Sharti } 372116491Sharti 373122112Sharti /* 374122112Sharti * Allocate list head. 375116491Sharti */ 376122112Sharti head = malloc(sizeof(*head)); 377122112Sharti if (head == NULL) 378116491Sharti return (NULL); 379122112Sharti STAILQ_INIT(head); 380116491Sharti 381116491Sharti /* root directory vnode, if one. */ 382116491Sharti if (filed.fd_rdir) { 383116491Sharti entry = filestat_new_entry(filed.fd_rdir, PS_FST_TYPE_VNODE, -1, 384116491Sharti PS_FST_FFLAG_READ, PS_FST_UFLAG_RDIR, 0, 0, NULL); 385116491Sharti if (entry != NULL) 386116491Sharti STAILQ_INSERT_TAIL(head, entry, next); 387122112Sharti } 388116491Sharti /* current working directory vnode. */ 389116491Sharti if (filed.fd_cdir) { 390116491Sharti entry = filestat_new_entry(filed.fd_cdir, PS_FST_TYPE_VNODE, -1, 391116491Sharti PS_FST_FFLAG_READ, PS_FST_UFLAG_CDIR, 0, 0, NULL); 392116491Sharti if (entry != NULL) 393116491Sharti STAILQ_INSERT_TAIL(head, entry, next); 394122112Sharti } 395116491Sharti /* jail root, if any. */ 396116491Sharti if (filed.fd_jdir) { 397116491Sharti entry = filestat_new_entry(filed.fd_jdir, PS_FST_TYPE_VNODE, -1, 398116491Sharti PS_FST_FFLAG_READ, PS_FST_UFLAG_JAIL, 0, 0, NULL); 399116491Sharti if (entry != NULL) 400116491Sharti STAILQ_INSERT_TAIL(head, entry, next); 401116491Sharti } 402122112Sharti /* ktrace vnode, if one */ 403116491Sharti if (kp->ki_tracep) { 404116491Sharti entry = filestat_new_entry(kp->ki_tracep, PS_FST_TYPE_VNODE, -1, 405116491Sharti PS_FST_FFLAG_READ | PS_FST_FFLAG_WRITE, 406116491Sharti PS_FST_UFLAG_TRACE, 0, 0, NULL); 407122112Sharti if (entry != NULL) 408116491Sharti STAILQ_INSERT_TAIL(head, entry, next); 409116491Sharti } 410116491Sharti /* text vnode, if one */ 411116491Sharti if (kp->ki_textvp) { 412116491Sharti entry = filestat_new_entry(kp->ki_textvp, PS_FST_TYPE_VNODE, -1, 413116491Sharti PS_FST_FFLAG_READ, PS_FST_UFLAG_TEXT, 0, 0, NULL); 414116491Sharti if (entry != NULL) 415116491Sharti STAILQ_INSERT_TAIL(head, entry, next); 416121687Sharti } 417122112Sharti /* Controlling terminal. */ 418116491Sharti if ((vp = getctty(kd, kp)) != NULL) { 419116491Sharti entry = filestat_new_entry(vp, PS_FST_TYPE_VNODE, -1, 420116491Sharti PS_FST_FFLAG_READ | PS_FST_FFLAG_WRITE, 421121687Sharti PS_FST_UFLAG_CTTY, 0, 0, NULL); 422121687Sharti if (entry != NULL) 423121687Sharti STAILQ_INSERT_TAIL(head, entry, next); 424121687Sharti } 425121687Sharti 426122112Sharti nfiles = filed.fd_lastfile + 1; 427122112Sharti ofiles = malloc(nfiles * sizeof(struct file *)); 428116491Sharti if (ofiles == NULL) { 429122112Sharti warn("malloc(%zu)", nfiles * sizeof(struct file *)); 430116491Sharti goto do_mmapped; 431116491Sharti } 432116491Sharti if (!kvm_read_all(kd, (unsigned long)filed.fd_ofiles, ofiles, 433116491Sharti nfiles * sizeof(struct file *))) { 434121687Sharti warnx("cannot read file structures at %p", 435121687Sharti (void *)filed.fd_ofiles); 436121687Sharti free(ofiles); 437121687Sharti goto do_mmapped; 438121687Sharti } 439116491Sharti for (i = 0; i <= filed.fd_lastfile; i++) { 440116491Sharti if (ofiles[i] == NULL) 441116491Sharti continue; 442116491Sharti if (!kvm_read_all(kd, (unsigned long)ofiles[i], &file, 443118540Sharti sizeof(struct file))) { 444116491Sharti warnx("can't read file %d at %p", i, 445116491Sharti (void *)ofiles[i]); 446116491Sharti continue; 447116491Sharti } 448116491Sharti switch (file.f_type) { 449116491Sharti case DTYPE_VNODE: 450116491Sharti type = PS_FST_TYPE_VNODE; 451116491Sharti data = file.f_vnode; 452122112Sharti break; 453116491Sharti case DTYPE_SOCKET: 454147256Sbrooks type = PS_FST_TYPE_SOCKET; 455116491Sharti data = file.f_data; 456116491Sharti break; 457116491Sharti case DTYPE_PIPE: 458116491Sharti type = PS_FST_TYPE_PIPE; 459116491Sharti data = file.f_data; 460116491Sharti break; 461116491Sharti case DTYPE_FIFO: 462116491Sharti type = PS_FST_TYPE_FIFO; 463116491Sharti data = file.f_vnode; 464117382Sharti break; 465116491Sharti#ifdef DTYPE_PTS 466116491Sharti case DTYPE_PTS: 467116491Sharti type = PS_FST_TYPE_PTS; 468116491Sharti data = file.f_data; 469116491Sharti break; 470116491Sharti#endif 471122112Sharti default: 472122112Sharti continue; 473122112Sharti } 474147256Sbrooks entry = filestat_new_entry(data, type, i, 475116491Sharti to_filestat_flags(file.f_flag), 0, 0, 0, NULL); 476116491Sharti if (entry != NULL) 477116491Sharti STAILQ_INSERT_TAIL(head, entry, next); 478116491Sharti } 479117382Sharti free(ofiles); 480116491Sharti 481116491Shartido_mmapped: 482116491Sharti 483147256Sbrooks /* 484116491Sharti * Process mmapped files if requested. 485116491Sharti */ 486147256Sbrooks if (mmapped) { 487116491Sharti if (!kvm_read_all(kd, (unsigned long)kp->ki_vmspace, &vmspace, 488116491Sharti sizeof(vmspace))) { 489116491Sharti warnx("can't read vmspace at %p", 490116491Sharti (void *)kp->ki_vmspace); 491147256Sbrooks goto exit; 492116491Sharti } 493116491Sharti map = &vmspace.vm_map; 494116491Sharti 495116491Sharti for (entryp = map->header.next; 496147256Sbrooks entryp != &kp->ki_vmspace->vm_map.header; 497116491Sharti entryp = vmentry.next) { 498116491Sharti if (!kvm_read_all(kd, (unsigned long)entryp, &vmentry, 499116491Sharti sizeof(vmentry))) { 500116491Sharti warnx("can't read vm_map_entry at %p", 501116491Sharti (void *)entryp); 502116491Sharti continue; 503116491Sharti } 504116491Sharti if (vmentry.eflags & MAP_ENTRY_IS_SUB_MAP) 505116491Sharti continue; 506116491Sharti if ((objp = vmentry.object.vm_object) == NULL) 507116491Sharti continue; 508116491Sharti for (; objp; objp = object.backing_object) { 509116491Sharti if (!kvm_read_all(kd, (unsigned long)objp, 510116491Sharti &object, sizeof(object))) { 511116491Sharti warnx("can't read vm_object at %p", 512118540Sharti (void *)objp); 513116491Sharti break; 514116491Sharti } 515116491Sharti } 516116491Sharti 517116491Sharti /* We want only vnode objects. */ 518116491Sharti if (object.type != OBJT_VNODE) 519116491Sharti continue; 520116491Sharti 521116491Sharti prot = vmentry.protection; 522116491Sharti fflags = 0; 523116491Sharti if (prot & VM_PROT_READ) 524116491Sharti fflags = PS_FST_FFLAG_READ; 525116491Sharti if (prot & VM_PROT_WRITE) 526116491Sharti fflags |= PS_FST_FFLAG_WRITE; 527116491Sharti 528116491Sharti /* 529116491Sharti * Create filestat entry. 530116491Sharti */ 531147256Sbrooks entry = filestat_new_entry(object.handle, 532118170Sharti PS_FST_TYPE_VNODE, -1, fflags, 533116491Sharti PS_FST_UFLAG_MMAP, 0, 0, NULL); 534116491Sharti if (entry != NULL) 535116491Sharti STAILQ_INSERT_TAIL(head, entry, next); 536116491Sharti } 537116491Sharti } 538116491Shartiexit: 539116491Sharti return (head); 540116491Sharti} 541116491Sharti 542116491Sharti/* 543116491Sharti * kinfo types to filestat translation. 544116491Sharti */ 545116491Shartistatic int 546116491Shartikinfo_type2fst(int kftype) 547116491Sharti{ 548116491Sharti static struct { 549116491Sharti int kf_type; 550116491Sharti int fst_type; 551116491Sharti } kftypes2fst[] = { 552116491Sharti { KF_TYPE_CRYPTO, PS_FST_TYPE_CRYPTO }, 553116491Sharti { KF_TYPE_FIFO, PS_FST_TYPE_FIFO }, 554116491Sharti { KF_TYPE_KQUEUE, PS_FST_TYPE_KQUEUE }, 555116491Sharti { KF_TYPE_MQUEUE, PS_FST_TYPE_MQUEUE }, 556116491Sharti { KF_TYPE_NONE, PS_FST_TYPE_NONE }, 557116491Sharti { KF_TYPE_PIPE, PS_FST_TYPE_PIPE }, 558116491Sharti { KF_TYPE_PTS, PS_FST_TYPE_PTS }, 559116491Sharti { KF_TYPE_SEM, PS_FST_TYPE_SEM }, 560116491Sharti { KF_TYPE_SHM, PS_FST_TYPE_SHM }, 561116491Sharti { KF_TYPE_SOCKET, PS_FST_TYPE_SOCKET }, 562116491Sharti { KF_TYPE_VNODE, PS_FST_TYPE_VNODE }, 563116491Sharti { KF_TYPE_UNKNOWN, PS_FST_TYPE_UNKNOWN } 564116491Sharti }; 565116491Sharti#define NKFTYPES (sizeof(kftypes2fst) / sizeof(*kftypes2fst)) 566116491Sharti unsigned int i; 567116491Sharti 568116491Sharti for (i = 0; i < NKFTYPES; i++) 569116491Sharti if (kftypes2fst[i].kf_type == kftype) 570116491Sharti break; 571116491Sharti if (i == NKFTYPES) 572147256Sbrooks return (PS_FST_TYPE_UNKNOWN); 573116491Sharti return (kftypes2fst[i].fst_type); 574116491Sharti} 575116491Sharti 576116491Sharti/* 577116491Sharti * kinfo flags to filestat translation. 578147256Sbrooks */ 579116491Shartistatic int 580116491Shartikinfo_fflags2fst(int kfflags) 581116491Sharti{ 582116491Sharti static struct { 583116491Sharti int kf_flag; 584116491Sharti int fst_flag; 585116491Sharti } kfflags2fst[] = { 586116491Sharti { KF_FLAG_APPEND, PS_FST_FFLAG_APPEND }, 587116491Sharti { KF_FLAG_ASYNC, PS_FST_FFLAG_ASYNC }, 588116491Sharti { KF_FLAG_CREAT, PS_FST_FFLAG_CREAT }, 589116491Sharti { KF_FLAG_DIRECT, PS_FST_FFLAG_DIRECT }, 590116491Sharti { KF_FLAG_EXCL, PS_FST_FFLAG_EXCL }, 591116491Sharti { KF_FLAG_EXEC, PS_FST_FFLAG_EXEC }, 592116491Sharti { KF_FLAG_EXLOCK, PS_FST_FFLAG_EXLOCK }, 593116491Sharti { KF_FLAG_FSYNC, PS_FST_FFLAG_SYNC }, 594116491Sharti { KF_FLAG_HASLOCK, PS_FST_FFLAG_HASLOCK }, 595116491Sharti { KF_FLAG_NOFOLLOW, PS_FST_FFLAG_NOFOLLOW }, 596116491Sharti { KF_FLAG_NONBLOCK, PS_FST_FFLAG_NONBLOCK }, 597116491Sharti { KF_FLAG_READ, PS_FST_FFLAG_READ }, 598116491Sharti { KF_FLAG_SHLOCK, PS_FST_FFLAG_SHLOCK }, 599116491Sharti { KF_FLAG_TRUNC, PS_FST_FFLAG_TRUNC }, 600116491Sharti { KF_FLAG_WRITE, PS_FST_FFLAG_WRITE } 601116491Sharti }; 602116491Sharti#define NKFFLAGS (sizeof(kfflags2fst) / sizeof(*kfflags2fst)) 603116491Sharti unsigned int i; 604116491Sharti int flags; 605116491Sharti 606116491Sharti flags = 0; 607116491Sharti for (i = 0; i < NKFFLAGS; i++) 608116491Sharti if ((kfflags & kfflags2fst[i].kf_flag) != 0) 609116491Sharti flags |= kfflags2fst[i].fst_flag; 610116491Sharti return (flags); 611116491Sharti} 612116491Sharti 613116491Shartistatic int 614116491Shartikinfo_uflags2fst(int fd) 615116491Sharti{ 616116491Sharti 617116491Sharti switch (fd) { 618116491Sharti case KF_FD_TYPE_CTTY: 619116491Sharti return (PS_FST_UFLAG_CTTY); 620116491Sharti case KF_FD_TYPE_CWD: 621116491Sharti return (PS_FST_UFLAG_CDIR); 622116491Sharti case KF_FD_TYPE_JAIL: 623116491Sharti return (PS_FST_UFLAG_JAIL); 624116491Sharti case KF_FD_TYPE_TEXT: 625116491Sharti return (PS_FST_UFLAG_TEXT); 626118598Sharti case KF_FD_TYPE_TRACE: 627118598Sharti return (PS_FST_UFLAG_TRACE); 628118598Sharti case KF_FD_TYPE_ROOT: 629118598Sharti return (PS_FST_UFLAG_RDIR); 630116491Sharti } 631116491Sharti return (0); 632116491Sharti} 633116491Sharti 634116491Shartistatic struct filestat_list * 635116491Shartiprocstat_getfiles_sysctl(struct procstat *procstat, struct kinfo_proc *kp, int mmapped) 636116491Sharti{ 637116491Sharti struct kinfo_file *kif, *files; 638116491Sharti struct kinfo_vmentry *kve, *vmentries; 639116491Sharti struct filestat_list *head; 640116491Sharti struct filestat *entry; 641116491Sharti char *path; 642116491Sharti off_t offset; 643116491Sharti int cnt, fd, fflags; 644116491Sharti int i, type, uflags; 645116491Sharti int refcount; 646116491Sharti 647116491Sharti assert(kp); 648116491Sharti if (kp->ki_fd == NULL) 649116491Sharti return (NULL); 650116491Sharti 651116491Sharti files = kinfo_getfile(kp->ki_pid, &cnt); 652116491Sharti if (files == NULL && errno != EPERM) { 653116491Sharti warn("kinfo_getfile()"); 654116491Sharti return (NULL); 655116491Sharti } 656116491Sharti procstat->files = files; 657116491Sharti 658116491Sharti /* 659116491Sharti * Allocate list head. 660116491Sharti */ 661116491Sharti head = malloc(sizeof(*head)); 662116491Sharti if (head == NULL) 663116491Sharti return (NULL); 664116491Sharti STAILQ_INIT(head); 665116491Sharti for (i = 0; i < cnt; i++) { 666116491Sharti kif = &files[i]; 667116491Sharti 668116491Sharti type = kinfo_type2fst(kif->kf_type); 669116491Sharti fd = kif->kf_fd >= 0 ? kif->kf_fd : -1; 670116491Sharti fflags = kinfo_fflags2fst(kif->kf_flags); 671116491Sharti uflags = kinfo_uflags2fst(kif->kf_fd); 672116491Sharti refcount = kif->kf_ref_count; 673116491Sharti offset = kif->kf_offset; 674116491Sharti if (*kif->kf_path != '\0') 675116491Sharti path = strdup(kif->kf_path); 676116491Sharti else 677116491Sharti path = NULL; 678116491Sharti 679116491Sharti /* 680116491Sharti * Create filestat entry. 681116491Sharti */ 682116491Sharti entry = filestat_new_entry(kif, type, fd, fflags, uflags, 683116491Sharti refcount, offset, path); 684116491Sharti if (entry != NULL) 685116491Sharti STAILQ_INSERT_TAIL(head, entry, next); 686116491Sharti } 687116491Sharti if (mmapped != 0) { 688116491Sharti vmentries = kinfo_getvmmap(kp->ki_pid, &cnt); 689116491Sharti procstat->vmentries = vmentries; 690116491Sharti if (vmentries == NULL || cnt == 0) 691116491Sharti goto fail; 692116491Sharti for (i = 0; i < cnt; i++) { 693116491Sharti kve = &vmentries[i]; 694116491Sharti if (kve->kve_type != KVME_TYPE_VNODE) 695116491Sharti continue; 696116491Sharti fflags = 0; 697116491Sharti if (kve->kve_protection & KVME_PROT_READ) 698116491Sharti fflags = PS_FST_FFLAG_READ; 699116491Sharti if (kve->kve_protection & KVME_PROT_WRITE) 700118598Sharti fflags |= PS_FST_FFLAG_WRITE; 701118598Sharti offset = kve->kve_offset; 702118598Sharti refcount = kve->kve_ref_count; 703118598Sharti if (*kve->kve_path != '\0') 704116491Sharti path = strdup(kve->kve_path); 705116491Sharti else 706116491Sharti path = NULL; 707116491Sharti entry = filestat_new_entry(kve, PS_FST_TYPE_VNODE, -1, 708116491Sharti fflags, PS_FST_UFLAG_MMAP, refcount, offset, path); 709116491Sharti if (entry != NULL) 710116491Sharti STAILQ_INSERT_TAIL(head, entry, next); 711116491Sharti } 712116491Sharti } 713116491Shartifail: 714116491Sharti return (head); 715116491Sharti} 716116491Sharti 717116491Shartiint 718116491Shartiprocstat_get_pipe_info(struct procstat *procstat, struct filestat *fst, 719116491Sharti struct pipestat *ps, char *errbuf) 720116491Sharti{ 721116491Sharti 722116491Sharti assert(ps); 723116491Sharti if (procstat->type == PROCSTAT_KVM) { 724116491Sharti return (procstat_get_pipe_info_kvm(procstat->kd, fst, ps, 725116491Sharti errbuf)); 726116491Sharti } else if (procstat->type == PROCSTAT_SYSCTL) { 727116491Sharti return (procstat_get_pipe_info_sysctl(fst, ps, errbuf)); 728116491Sharti } else { 729116491Sharti warnx("unknow access method: %d", procstat->type); 730116491Sharti snprintf(errbuf, _POSIX2_LINE_MAX, "error"); 731116491Sharti return (1); 732116491Sharti } 733116491Sharti} 734116491Sharti 735116491Shartistatic int 736116491Shartiprocstat_get_pipe_info_kvm(kvm_t *kd, struct filestat *fst, 737116491Sharti struct pipestat *ps, char *errbuf) 738116491Sharti{ 739116491Sharti struct pipe pi; 740116491Sharti void *pipep; 741116491Sharti 742116491Sharti assert(kd); 743116491Sharti assert(ps); 744116491Sharti assert(fst); 745116491Sharti bzero(ps, sizeof(*ps)); 746116491Sharti pipep = fst->fs_typedep; 747116491Sharti if (pipep == NULL) 748116491Sharti goto fail; 749116491Sharti if (!kvm_read_all(kd, (unsigned long)pipep, &pi, sizeof(struct pipe))) { 750116491Sharti warnx("can't read pipe at %p", (void *)pipep); 751116491Sharti goto fail; 752116491Sharti } 753116491Sharti ps->addr = (uintptr_t)pipep; 754116491Sharti ps->peer = (uintptr_t)pi.pipe_peer; 755116491Sharti ps->buffer_cnt = pi.pipe_buffer.cnt; 756116491Sharti return (0); 757116491Sharti 758116491Shartifail: 759116491Sharti snprintf(errbuf, _POSIX2_LINE_MAX, "error"); 760116491Sharti return (1); 761116491Sharti} 762116491Sharti 763116491Shartistatic int 764116491Shartiprocstat_get_pipe_info_sysctl(struct filestat *fst, struct pipestat *ps, 765116491Sharti char *errbuf __unused) 766116491Sharti{ 767116491Sharti struct kinfo_file *kif; 768116491Sharti 769116491Sharti assert(ps); 770116491Sharti assert(fst); 771116491Sharti bzero(ps, sizeof(*ps)); 772116491Sharti kif = fst->fs_typedep; 773116491Sharti if (kif == NULL) 774116491Sharti return (1); 775116491Sharti ps->addr = kif->kf_un.kf_pipe.kf_pipe_addr; 776116491Sharti ps->peer = kif->kf_un.kf_pipe.kf_pipe_peer; 777116491Sharti ps->buffer_cnt = kif->kf_un.kf_pipe.kf_pipe_buffer_cnt; 778116491Sharti return (0); 779116491Sharti} 780116491Sharti 781116491Shartiint 782116491Shartiprocstat_get_pts_info(struct procstat *procstat, struct filestat *fst, 783116491Sharti struct ptsstat *pts, char *errbuf) 784116491Sharti{ 785116491Sharti 786116491Sharti assert(pts); 787116491Sharti if (procstat->type == PROCSTAT_KVM) { 788116491Sharti return (procstat_get_pts_info_kvm(procstat->kd, fst, pts, 789116491Sharti errbuf)); 790116491Sharti } else if (procstat->type == PROCSTAT_SYSCTL) { 791116491Sharti return (procstat_get_pts_info_sysctl(fst, pts, errbuf)); 792116491Sharti } else { 793116491Sharti warnx("unknow access method: %d", procstat->type); 794116491Sharti snprintf(errbuf, _POSIX2_LINE_MAX, "error"); 795116491Sharti return (1); 796116491Sharti } 797116491Sharti} 798116491Sharti 799116491Shartistatic int 800116491Shartiprocstat_get_pts_info_kvm(kvm_t *kd, struct filestat *fst, 801116491Sharti struct ptsstat *pts, char *errbuf) 802116491Sharti{ 803116491Sharti struct tty tty; 804116491Sharti void *ttyp; 805116491Sharti 806116491Sharti assert(kd); 807116491Sharti assert(pts); 808116491Sharti assert(fst); 809116491Sharti bzero(pts, sizeof(*pts)); 810116491Sharti ttyp = fst->fs_typedep; 811116491Sharti if (ttyp == NULL) 812116491Sharti goto fail; 813116491Sharti if (!kvm_read_all(kd, (unsigned long)ttyp, &tty, sizeof(struct tty))) { 814116491Sharti warnx("can't read tty at %p", (void *)ttyp); 815116491Sharti goto fail; 816116491Sharti } 817116491Sharti pts->dev = dev2udev(kd, tty.t_dev); 818116491Sharti (void)kdevtoname(kd, tty.t_dev, pts->devname); 819116491Sharti return (0); 820116491Sharti 821116491Shartifail: 822116491Sharti snprintf(errbuf, _POSIX2_LINE_MAX, "error"); 823116491Sharti return (1); 824116491Sharti} 825116491Sharti 826116491Shartistatic int 827procstat_get_pts_info_sysctl(struct filestat *fst, struct ptsstat *pts, 828 char *errbuf __unused) 829{ 830 struct kinfo_file *kif; 831 832 assert(pts); 833 assert(fst); 834 bzero(pts, sizeof(*pts)); 835 kif = fst->fs_typedep; 836 if (kif == NULL) 837 return (0); 838 pts->dev = kif->kf_un.kf_pts.kf_pts_dev; 839 strlcpy(pts->devname, kif->kf_path, sizeof(pts->devname)); 840 return (0); 841} 842 843int 844procstat_get_vnode_info(struct procstat *procstat, struct filestat *fst, 845 struct vnstat *vn, char *errbuf) 846{ 847 848 assert(vn); 849 if (procstat->type == PROCSTAT_KVM) { 850 return (procstat_get_vnode_info_kvm(procstat->kd, fst, vn, 851 errbuf)); 852 } else if (procstat->type == PROCSTAT_SYSCTL) { 853 return (procstat_get_vnode_info_sysctl(fst, vn, errbuf)); 854 } else { 855 warnx("unknow access method: %d", procstat->type); 856 snprintf(errbuf, _POSIX2_LINE_MAX, "error"); 857 return (1); 858 } 859} 860 861static int 862procstat_get_vnode_info_kvm(kvm_t *kd, struct filestat *fst, 863 struct vnstat *vn, char *errbuf) 864{ 865 /* Filesystem specific handlers. */ 866 #define FSTYPE(fst) {#fst, fst##_filestat} 867 struct { 868 const char *tag; 869 int (*handler)(kvm_t *kd, struct vnode *vp, 870 struct vnstat *vn); 871 } fstypes[] = { 872 FSTYPE(devfs), 873 FSTYPE(isofs), 874 FSTYPE(msdosfs), 875 FSTYPE(nfs), 876 FSTYPE(ntfs), 877#ifdef LIBPROCSTAT_NWFS 878 FSTYPE(nwfs), 879#endif 880 FSTYPE(smbfs), 881 FSTYPE(udf), 882 FSTYPE(ufs), 883#ifdef LIBPROCSTAT_ZFS 884 FSTYPE(zfs), 885#endif 886 }; 887#define NTYPES (sizeof(fstypes) / sizeof(*fstypes)) 888 struct vnode vnode; 889 char tagstr[12]; 890 void *vp; 891 int error, found; 892 unsigned int i; 893 894 assert(kd); 895 assert(vn); 896 assert(fst); 897 vp = fst->fs_typedep; 898 if (vp == NULL) 899 goto fail; 900 error = kvm_read_all(kd, (unsigned long)vp, &vnode, sizeof(vnode)); 901 if (error == 0) { 902 warnx("can't read vnode at %p", (void *)vp); 903 goto fail; 904 } 905 bzero(vn, sizeof(*vn)); 906 vn->vn_type = vntype2psfsttype(vnode.v_type); 907 if (vnode.v_type == VNON || vnode.v_type == VBAD) 908 return (0); 909 error = kvm_read_all(kd, (unsigned long)vnode.v_tag, tagstr, 910 sizeof(tagstr)); 911 if (error == 0) { 912 warnx("can't read v_tag at %p", (void *)vp); 913 goto fail; 914 } 915 tagstr[sizeof(tagstr) - 1] = '\0'; 916 917 /* 918 * Find appropriate handler. 919 */ 920 for (i = 0, found = 0; i < NTYPES; i++) 921 if (!strcmp(fstypes[i].tag, tagstr)) { 922 if (fstypes[i].handler(kd, &vnode, vn) != 0) { 923 goto fail; 924 } 925 break; 926 } 927 if (i == NTYPES) { 928 snprintf(errbuf, _POSIX2_LINE_MAX, "?(%s)", tagstr); 929 return (1); 930 } 931 vn->vn_mntdir = getmnton(kd, vnode.v_mount); 932 if ((vnode.v_type == VBLK || vnode.v_type == VCHR) && 933 vnode.v_rdev != NULL){ 934 vn->vn_dev = dev2udev(kd, vnode.v_rdev); 935 (void)kdevtoname(kd, vnode.v_rdev, vn->vn_devname); 936 } else { 937 vn->vn_dev = -1; 938 } 939 return (0); 940 941fail: 942 snprintf(errbuf, _POSIX2_LINE_MAX, "error"); 943 return (1); 944} 945 946/* 947 * kinfo vnode type to filestat translation. 948 */ 949static int 950kinfo_vtype2fst(int kfvtype) 951{ 952 static struct { 953 int kf_vtype; 954 int fst_vtype; 955 } kfvtypes2fst[] = { 956 { KF_VTYPE_VBAD, PS_FST_VTYPE_VBAD }, 957 { KF_VTYPE_VBLK, PS_FST_VTYPE_VBLK }, 958 { KF_VTYPE_VCHR, PS_FST_VTYPE_VCHR }, 959 { KF_VTYPE_VDIR, PS_FST_VTYPE_VDIR }, 960 { KF_VTYPE_VFIFO, PS_FST_VTYPE_VFIFO }, 961 { KF_VTYPE_VLNK, PS_FST_VTYPE_VLNK }, 962 { KF_VTYPE_VNON, PS_FST_VTYPE_VNON }, 963 { KF_VTYPE_VREG, PS_FST_VTYPE_VREG }, 964 { KF_VTYPE_VSOCK, PS_FST_VTYPE_VSOCK } 965 }; 966#define NKFVTYPES (sizeof(kfvtypes2fst) / sizeof(*kfvtypes2fst)) 967 unsigned int i; 968 969 for (i = 0; i < NKFVTYPES; i++) 970 if (kfvtypes2fst[i].kf_vtype == kfvtype) 971 break; 972 if (i == NKFVTYPES) 973 return (PS_FST_VTYPE_UNKNOWN); 974 return (kfvtypes2fst[i].fst_vtype); 975} 976 977static int 978procstat_get_vnode_info_sysctl(struct filestat *fst, struct vnstat *vn, 979 char *errbuf) 980{ 981 struct statfs stbuf; 982 struct kinfo_file *kif; 983 struct kinfo_vmentry *kve; 984 uint64_t fileid; 985 uint64_t size; 986 char *name, *path; 987 uint32_t fsid; 988 uint16_t mode; 989 uint32_t rdev; 990 int vntype; 991 int status; 992 993 assert(fst); 994 assert(vn); 995 bzero(vn, sizeof(*vn)); 996 if (fst->fs_typedep == NULL) 997 return (1); 998 if (fst->fs_uflags & PS_FST_UFLAG_MMAP) { 999 kve = fst->fs_typedep; 1000 fileid = kve->kve_vn_fileid; 1001 fsid = kve->kve_vn_fsid; 1002 mode = kve->kve_vn_mode; 1003 path = kve->kve_path; 1004 rdev = kve->kve_vn_rdev; 1005 size = kve->kve_vn_size; 1006 vntype = kinfo_vtype2fst(kve->kve_vn_type); 1007 status = kve->kve_status; 1008 } else { 1009 kif = fst->fs_typedep; 1010 fileid = kif->kf_un.kf_file.kf_file_fileid; 1011 fsid = kif->kf_un.kf_file.kf_file_fsid; 1012 mode = kif->kf_un.kf_file.kf_file_mode; 1013 path = kif->kf_path; 1014 rdev = kif->kf_un.kf_file.kf_file_rdev; 1015 size = kif->kf_un.kf_file.kf_file_size; 1016 vntype = kinfo_vtype2fst(kif->kf_vnode_type); 1017 status = kif->kf_status; 1018 } 1019 vn->vn_type = vntype; 1020 if (vntype == PS_FST_VTYPE_VNON || vntype == PS_FST_VTYPE_VBAD) 1021 return (0); 1022 if ((status & KF_ATTR_VALID) == 0) { 1023 snprintf(errbuf, _POSIX2_LINE_MAX, "? (no info available)"); 1024 return (1); 1025 } 1026 if (path && *path) { 1027 statfs(path, &stbuf); 1028 vn->vn_mntdir = strdup(stbuf.f_mntonname); 1029 } else 1030 vn->vn_mntdir = strdup("-"); 1031 vn->vn_dev = rdev; 1032 if (vntype == PS_FST_VTYPE_VBLK) { 1033 name = devname(rdev, S_IFBLK); 1034 if (name != NULL) 1035 strlcpy(vn->vn_devname, name, 1036 sizeof(vn->vn_devname)); 1037 } else if (vntype == PS_FST_VTYPE_VCHR) { 1038 name = devname(vn->vn_dev, S_IFCHR); 1039 if (name != NULL) 1040 strlcpy(vn->vn_devname, name, 1041 sizeof(vn->vn_devname)); 1042 } 1043 vn->vn_fsid = fsid; 1044 vn->vn_fileid = fileid; 1045 vn->vn_size = size; 1046 vn->vn_mode = mode; 1047 return (0); 1048} 1049 1050int 1051procstat_get_socket_info(struct procstat *procstat, struct filestat *fst, 1052 struct sockstat *sock, char *errbuf) 1053{ 1054 1055 assert(sock); 1056 if (procstat->type == PROCSTAT_KVM) { 1057 return (procstat_get_socket_info_kvm(procstat->kd, fst, sock, 1058 errbuf)); 1059 } else if (procstat->type == PROCSTAT_SYSCTL) { 1060 return (procstat_get_socket_info_sysctl(fst, sock, errbuf)); 1061 } else { 1062 warnx("unknow access method: %d", procstat->type); 1063 snprintf(errbuf, _POSIX2_LINE_MAX, "error"); 1064 return (1); 1065 } 1066} 1067 1068static int 1069procstat_get_socket_info_kvm(kvm_t *kd, struct filestat *fst, 1070 struct sockstat *sock, char *errbuf) 1071{ 1072 struct domain dom; 1073 struct inpcb inpcb; 1074 struct protosw proto; 1075 struct socket s; 1076 struct unpcb unpcb; 1077 ssize_t len; 1078 void *so; 1079 1080 assert(kd); 1081 assert(sock); 1082 assert(fst); 1083 bzero(sock, sizeof(*sock)); 1084 so = fst->fs_typedep; 1085 if (so == NULL) 1086 goto fail; 1087 sock->so_addr = (uintptr_t)so; 1088 /* fill in socket */ 1089 if (!kvm_read_all(kd, (unsigned long)so, &s, 1090 sizeof(struct socket))) { 1091 warnx("can't read sock at %p", (void *)so); 1092 goto fail; 1093 } 1094 /* fill in protosw entry */ 1095 if (!kvm_read_all(kd, (unsigned long)s.so_proto, &proto, 1096 sizeof(struct protosw))) { 1097 warnx("can't read protosw at %p", (void *)s.so_proto); 1098 goto fail; 1099 } 1100 /* fill in domain */ 1101 if (!kvm_read_all(kd, (unsigned long)proto.pr_domain, &dom, 1102 sizeof(struct domain))) { 1103 warnx("can't read domain at %p", 1104 (void *)proto.pr_domain); 1105 goto fail; 1106 } 1107 if ((len = kvm_read(kd, (unsigned long)dom.dom_name, sock->dname, 1108 sizeof(sock->dname) - 1)) < 0) { 1109 warnx("can't read domain name at %p", (void *)dom.dom_name); 1110 sock->dname[0] = '\0'; 1111 } 1112 else 1113 sock->dname[len] = '\0'; 1114 1115 /* 1116 * Fill in known data. 1117 */ 1118 sock->type = s.so_type; 1119 sock->proto = proto.pr_protocol; 1120 sock->dom_family = dom.dom_family; 1121 sock->so_pcb = (uintptr_t)s.so_pcb; 1122 1123 /* 1124 * Protocol specific data. 1125 */ 1126 switch(dom.dom_family) { 1127 case AF_INET: 1128 case AF_INET6: 1129 if (proto.pr_protocol == IPPROTO_TCP) { 1130 if (s.so_pcb) { 1131 if (kvm_read(kd, (u_long)s.so_pcb, 1132 (char *)&inpcb, sizeof(struct inpcb)) 1133 != sizeof(struct inpcb)) { 1134 warnx("can't read inpcb at %p", 1135 (void *)s.so_pcb); 1136 } else 1137 sock->inp_ppcb = 1138 (uintptr_t)inpcb.inp_ppcb; 1139 } 1140 } 1141 break; 1142 case AF_UNIX: 1143 if (s.so_pcb) { 1144 if (kvm_read(kd, (u_long)s.so_pcb, (char *)&unpcb, 1145 sizeof(struct unpcb)) != sizeof(struct unpcb)){ 1146 warnx("can't read unpcb at %p", 1147 (void *)s.so_pcb); 1148 } else if (unpcb.unp_conn) { 1149 sock->so_rcv_sb_state = s.so_rcv.sb_state; 1150 sock->so_snd_sb_state = s.so_snd.sb_state; 1151 sock->unp_conn = (uintptr_t)unpcb.unp_conn; 1152 } 1153 } 1154 break; 1155 default: 1156 break; 1157 } 1158 return (0); 1159 1160fail: 1161 snprintf(errbuf, _POSIX2_LINE_MAX, "error"); 1162 return (1); 1163} 1164 1165static int 1166procstat_get_socket_info_sysctl(struct filestat *fst, struct sockstat *sock, 1167 char *errbuf __unused) 1168{ 1169 struct kinfo_file *kif; 1170 1171 assert(sock); 1172 assert(fst); 1173 bzero(sock, sizeof(*sock)); 1174 kif = fst->fs_typedep; 1175 if (kif == NULL) 1176 return (0); 1177 1178 /* 1179 * Fill in known data. 1180 */ 1181 sock->type = kif->kf_sock_type; 1182 sock->proto = kif->kf_sock_protocol; 1183 sock->dom_family = kif->kf_sock_domain; 1184 sock->so_pcb = kif->kf_un.kf_sock.kf_sock_pcb; 1185 strlcpy(sock->dname, kif->kf_path, sizeof(sock->dname)); 1186 bcopy(&kif->kf_sa_local, &sock->sa_local, kif->kf_sa_local.ss_len); 1187 bcopy(&kif->kf_sa_peer, &sock->sa_peer, kif->kf_sa_peer.ss_len); 1188 1189 /* 1190 * Protocol specific data. 1191 */ 1192 switch(sock->dom_family) { 1193 case AF_INET: 1194 case AF_INET6: 1195 if (sock->proto == IPPROTO_TCP) 1196 sock->inp_ppcb = kif->kf_un.kf_sock.kf_sock_inpcb; 1197 break; 1198 case AF_UNIX: 1199 if (kif->kf_un.kf_sock.kf_sock_unpconn != 0) { 1200 sock->so_rcv_sb_state = 1201 kif->kf_un.kf_sock.kf_sock_rcv_sb_state; 1202 sock->so_snd_sb_state = 1203 kif->kf_un.kf_sock.kf_sock_snd_sb_state; 1204 sock->unp_conn = 1205 kif->kf_un.kf_sock.kf_sock_unpconn; 1206 } 1207 break; 1208 default: 1209 break; 1210 } 1211 return (0); 1212} 1213 1214/* 1215 * Descriptor flags to filestat translation. 1216 */ 1217static int 1218to_filestat_flags(int flags) 1219{ 1220 static struct { 1221 int flag; 1222 int fst_flag; 1223 } fstflags[] = { 1224 { FREAD, PS_FST_FFLAG_READ }, 1225 { FWRITE, PS_FST_FFLAG_WRITE }, 1226 { O_APPEND, PS_FST_FFLAG_APPEND }, 1227 { O_ASYNC, PS_FST_FFLAG_ASYNC }, 1228 { O_CREAT, PS_FST_FFLAG_CREAT }, 1229 { O_DIRECT, PS_FST_FFLAG_DIRECT }, 1230 { O_EXCL, PS_FST_FFLAG_EXCL }, 1231 { O_EXEC, PS_FST_FFLAG_EXEC }, 1232 { O_EXLOCK, PS_FST_FFLAG_EXLOCK }, 1233 { O_NOFOLLOW, PS_FST_FFLAG_NOFOLLOW }, 1234 { O_NONBLOCK, PS_FST_FFLAG_NONBLOCK }, 1235 { O_SHLOCK, PS_FST_FFLAG_SHLOCK }, 1236 { O_SYNC, PS_FST_FFLAG_SYNC }, 1237 { O_TRUNC, PS_FST_FFLAG_TRUNC } 1238 }; 1239#define NFSTFLAGS (sizeof(fstflags) / sizeof(*fstflags)) 1240 int fst_flags; 1241 unsigned int i; 1242 1243 fst_flags = 0; 1244 for (i = 0; i < NFSTFLAGS; i++) 1245 if (flags & fstflags[i].flag) 1246 fst_flags |= fstflags[i].fst_flag; 1247 return (fst_flags); 1248} 1249 1250/* 1251 * Vnode type to filestate translation. 1252 */ 1253static int 1254vntype2psfsttype(int type) 1255{ 1256 static struct { 1257 int vtype; 1258 int fst_vtype; 1259 } vt2fst[] = { 1260 { VBAD, PS_FST_VTYPE_VBAD }, 1261 { VBLK, PS_FST_VTYPE_VBLK }, 1262 { VCHR, PS_FST_VTYPE_VCHR }, 1263 { VDIR, PS_FST_VTYPE_VDIR }, 1264 { VFIFO, PS_FST_VTYPE_VFIFO }, 1265 { VLNK, PS_FST_VTYPE_VLNK }, 1266 { VNON, PS_FST_VTYPE_VNON }, 1267 { VREG, PS_FST_VTYPE_VREG }, 1268 { VSOCK, PS_FST_VTYPE_VSOCK } 1269 }; 1270#define NVFTYPES (sizeof(vt2fst) / sizeof(*vt2fst)) 1271 unsigned int i, fst_type; 1272 1273 fst_type = PS_FST_VTYPE_UNKNOWN; 1274 for (i = 0; i < NVFTYPES; i++) { 1275 if (type == vt2fst[i].vtype) { 1276 fst_type = vt2fst[i].fst_vtype; 1277 break; 1278 } 1279 } 1280 return (fst_type); 1281} 1282 1283static char * 1284getmnton(kvm_t *kd, struct mount *m) 1285{ 1286 struct mount mnt; 1287 static struct mtab { 1288 struct mtab *next; 1289 struct mount *m; 1290 char mntonname[MNAMELEN + 1]; 1291 } *mhead = NULL; 1292 struct mtab *mt; 1293 1294 for (mt = mhead; mt != NULL; mt = mt->next) 1295 if (m == mt->m) 1296 return (mt->mntonname); 1297 if (!kvm_read_all(kd, (unsigned long)m, &mnt, sizeof(struct mount))) { 1298 warnx("can't read mount table at %p", (void *)m); 1299 return (NULL); 1300 } 1301 if ((mt = malloc(sizeof (struct mtab))) == NULL) 1302 err(1, NULL); 1303 mt->m = m; 1304 bcopy(&mnt.mnt_stat.f_mntonname[0], &mt->mntonname[0], MNAMELEN); 1305 mt->mntonname[MNAMELEN] = '\0'; 1306 mt->next = mhead; 1307 mhead = mt; 1308 return (mt->mntonname); 1309} 1310