xref: /freebsd/lib/libprocstat/libprocstat.c (revision 8d03b99b9dafe92896f405c79f846667637c0194)
10daf62d9SStanislav Sedov /*-
2df57947fSPedro F. Giffuni  * SPDX-License-Identifier: BSD-4-Clause
3df57947fSPedro F. Giffuni  *
486be94fcSTycho Nightingale  * Copyright (c) 2017 Dell EMC
50daf62d9SStanislav Sedov  * Copyright (c) 2009 Stanislav Sedov <stas@FreeBSD.org>
60daf62d9SStanislav Sedov  * Copyright (c) 1988, 1993
70daf62d9SStanislav Sedov  *      The Regents of the University of California.  All rights reserved.
80daf62d9SStanislav Sedov  *
90daf62d9SStanislav Sedov  * Redistribution and use in source and binary forms, with or without
100daf62d9SStanislav Sedov  * modification, are permitted provided that the following conditions
110daf62d9SStanislav Sedov  * are met:
120daf62d9SStanislav Sedov  * 1. Redistributions of source code must retain the above copyright
130daf62d9SStanislav Sedov  *    notice, this list of conditions and the following disclaimer.
140daf62d9SStanislav Sedov  * 2. Redistributions in binary form must reproduce the above copyright
150daf62d9SStanislav Sedov  *    notice, this list of conditions and the following disclaimer in the
160daf62d9SStanislav Sedov  *    documentation and/or other materials provided with the distribution.
170daf62d9SStanislav Sedov  * 3. All advertising materials mentioning features or use of this software
180daf62d9SStanislav Sedov  *    must display the following acknowledgement:
190daf62d9SStanislav Sedov  *      This product includes software developed by the University of
200daf62d9SStanislav Sedov  *      California, Berkeley and its contributors.
210daf62d9SStanislav Sedov  * 4. Neither the name of the University nor the names of its contributors
220daf62d9SStanislav Sedov  *    may be used to endorse or promote products derived from this software
230daf62d9SStanislav Sedov  *    without specific prior written permission.
240daf62d9SStanislav Sedov  *
250daf62d9SStanislav Sedov  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
260daf62d9SStanislav Sedov  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
270daf62d9SStanislav Sedov  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
280daf62d9SStanislav Sedov  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
290daf62d9SStanislav Sedov  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
300daf62d9SStanislav Sedov  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
310daf62d9SStanislav Sedov  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
320daf62d9SStanislav Sedov  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
330daf62d9SStanislav Sedov  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
340daf62d9SStanislav Sedov  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
350daf62d9SStanislav Sedov  * SUCH DAMAGE.
360daf62d9SStanislav Sedov  */
370daf62d9SStanislav Sedov 
380daf62d9SStanislav Sedov #include <sys/cdefs.h>
390daf62d9SStanislav Sedov __FBSDID("$FreeBSD$");
400daf62d9SStanislav Sedov 
410daf62d9SStanislav Sedov #include <sys/param.h>
422ff020d3SMikolaj Golub #include <sys/elf.h>
430daf62d9SStanislav Sedov #include <sys/time.h>
447cc0ebfdSMikolaj Golub #include <sys/resourcevar.h>
451f84c47eSMikolaj Golub #define	_WANT_UCRED
461f84c47eSMikolaj Golub #include <sys/ucred.h>
471f84c47eSMikolaj Golub #undef _WANT_UCRED
480daf62d9SStanislav Sedov #include <sys/proc.h>
490daf62d9SStanislav Sedov #include <sys/user.h>
500daf62d9SStanislav Sedov #include <sys/stat.h>
510daf62d9SStanislav Sedov #include <sys/vnode.h>
520daf62d9SStanislav Sedov #include <sys/socket.h>
530e229f34SGleb Smirnoff #define	_WANT_SOCKET
540daf62d9SStanislav Sedov #include <sys/socketvar.h>
550daf62d9SStanislav Sedov #include <sys/domain.h>
560daf62d9SStanislav Sedov #include <sys/protosw.h>
570daf62d9SStanislav Sedov #include <sys/un.h>
580e229f34SGleb Smirnoff #define	_WANT_UNPCB
590daf62d9SStanislav Sedov #include <sys/unpcb.h>
600daf62d9SStanislav Sedov #include <sys/sysctl.h>
610daf62d9SStanislav Sedov #include <sys/tty.h>
620daf62d9SStanislav Sedov #include <sys/filedesc.h>
630daf62d9SStanislav Sedov #include <sys/queue.h>
640daf62d9SStanislav Sedov #define	_WANT_FILE
650daf62d9SStanislav Sedov #include <sys/file.h>
660daf62d9SStanislav Sedov #include <sys/conf.h>
67958aa575SJohn Baldwin #include <sys/ksem.h>
68e506e182SJohn Baldwin #include <sys/mman.h>
69b881b8beSRobert Watson #include <sys/capsicum.h>
70a4ba6502STycho Nightingale #include <sys/ptrace.h>
710daf62d9SStanislav Sedov #define	_KERNEL
720daf62d9SStanislav Sedov #include <sys/mount.h>
730daf62d9SStanislav Sedov #include <sys/pipe.h>
740daf62d9SStanislav Sedov #include <ufs/ufs/quota.h>
750daf62d9SStanislav Sedov #include <ufs/ufs/inode.h>
760daf62d9SStanislav Sedov #include <fs/devfs/devfs.h>
770daf62d9SStanislav Sedov #include <fs/devfs/devfs_int.h>
780daf62d9SStanislav Sedov #undef _KERNEL
790daf62d9SStanislav Sedov #include <nfs/nfsproto.h>
800daf62d9SStanislav Sedov #include <nfsclient/nfs.h>
810daf62d9SStanislav Sedov #include <nfsclient/nfsnode.h>
820daf62d9SStanislav Sedov 
830daf62d9SStanislav Sedov #include <vm/vm.h>
840daf62d9SStanislav Sedov #include <vm/vm_map.h>
850daf62d9SStanislav Sedov #include <vm/vm_object.h>
860daf62d9SStanislav Sedov 
870daf62d9SStanislav Sedov #include <net/route.h>
880daf62d9SStanislav Sedov #include <netinet/in.h>
890daf62d9SStanislav Sedov #include <netinet/in_systm.h>
900daf62d9SStanislav Sedov #include <netinet/ip.h>
91cc65eb4eSGleb Smirnoff #define	_WANT_INPCB
920daf62d9SStanislav Sedov #include <netinet/in_pcb.h>
930daf62d9SStanislav Sedov 
940daf62d9SStanislav Sedov #include <assert.h>
950daf62d9SStanislav Sedov #include <ctype.h>
960daf62d9SStanislav Sedov #include <err.h>
970daf62d9SStanislav Sedov #include <fcntl.h>
980daf62d9SStanislav Sedov #include <kvm.h>
990daf62d9SStanislav Sedov #include <libutil.h>
1000daf62d9SStanislav Sedov #include <limits.h>
1010daf62d9SStanislav Sedov #include <paths.h>
1020daf62d9SStanislav Sedov #include <pwd.h>
1030daf62d9SStanislav Sedov #include <stdio.h>
1040daf62d9SStanislav Sedov #include <stdlib.h>
1050daf62d9SStanislav Sedov #include <stddef.h>
1060daf62d9SStanislav Sedov #include <string.h>
1070daf62d9SStanislav Sedov #include <unistd.h>
1080daf62d9SStanislav Sedov #include <netdb.h>
1090daf62d9SStanislav Sedov 
1100daf62d9SStanislav Sedov #include <libprocstat.h>
1110daf62d9SStanislav Sedov #include "libprocstat_internal.h"
1120daf62d9SStanislav Sedov #include "common_kvm.h"
1137153ad2bSMikolaj Golub #include "core.h"
1140daf62d9SStanislav Sedov 
1150daf62d9SStanislav Sedov int     statfs(const char *, struct statfs *);	/* XXX */
1160daf62d9SStanislav Sedov 
1170daf62d9SStanislav Sedov #define	PROCSTAT_KVM	1
1180daf62d9SStanislav Sedov #define	PROCSTAT_SYSCTL	2
1197153ad2bSMikolaj Golub #define	PROCSTAT_CORE	3
1200daf62d9SStanislav Sedov 
1214482b5e3SMikolaj Golub static char	**getargv(struct procstat *procstat, struct kinfo_proc *kp,
1224482b5e3SMikolaj Golub     size_t nchr, int env);
1230daf62d9SStanislav Sedov static char	*getmnton(kvm_t *kd, struct mount *m);
12439680c7bSMikolaj Golub static struct kinfo_vmentry *	kinfo_getvmmap_core(struct procstat_core *core,
12539680c7bSMikolaj Golub     int *cntp);
1262ff020d3SMikolaj Golub static Elf_Auxinfo	*procstat_getauxv_core(struct procstat_core *core,
1272ff020d3SMikolaj Golub     unsigned int *cntp);
1282ff020d3SMikolaj Golub static Elf_Auxinfo	*procstat_getauxv_sysctl(pid_t pid, unsigned int *cntp);
1290daf62d9SStanislav Sedov static struct filestat_list	*procstat_getfiles_kvm(
1300daf62d9SStanislav Sedov     struct procstat *procstat, struct kinfo_proc *kp, int mmapped);
1310daf62d9SStanislav Sedov static struct filestat_list	*procstat_getfiles_sysctl(
1320daf62d9SStanislav Sedov     struct procstat *procstat, struct kinfo_proc *kp, int mmapped);
1330daf62d9SStanislav Sedov static int	procstat_get_pipe_info_sysctl(struct filestat *fst,
1340daf62d9SStanislav Sedov     struct pipestat *pipe, char *errbuf);
1350daf62d9SStanislav Sedov static int	procstat_get_pipe_info_kvm(kvm_t *kd, struct filestat *fst,
1360daf62d9SStanislav Sedov     struct pipestat *pipe, char *errbuf);
1370daf62d9SStanislav Sedov static int	procstat_get_pts_info_sysctl(struct filestat *fst,
1380daf62d9SStanislav Sedov     struct ptsstat *pts, char *errbuf);
1390daf62d9SStanislav Sedov static int	procstat_get_pts_info_kvm(kvm_t *kd, struct filestat *fst,
1400daf62d9SStanislav Sedov     struct ptsstat *pts, char *errbuf);
141958aa575SJohn Baldwin static int	procstat_get_sem_info_sysctl(struct filestat *fst,
142958aa575SJohn Baldwin     struct semstat *sem, char *errbuf);
143958aa575SJohn Baldwin static int	procstat_get_sem_info_kvm(kvm_t *kd, struct filestat *fst,
144958aa575SJohn Baldwin     struct semstat *sem, char *errbuf);
145e506e182SJohn Baldwin static int	procstat_get_shm_info_sysctl(struct filestat *fst,
146e506e182SJohn Baldwin     struct shmstat *shm, char *errbuf);
147e506e182SJohn Baldwin static int	procstat_get_shm_info_kvm(kvm_t *kd, struct filestat *fst,
148e506e182SJohn Baldwin     struct shmstat *shm, char *errbuf);
1490daf62d9SStanislav Sedov static int	procstat_get_socket_info_sysctl(struct filestat *fst,
1500daf62d9SStanislav Sedov     struct sockstat *sock, char *errbuf);
1510daf62d9SStanislav Sedov static int	procstat_get_socket_info_kvm(kvm_t *kd, struct filestat *fst,
1520daf62d9SStanislav Sedov     struct sockstat *sock, char *errbuf);
1530daf62d9SStanislav Sedov static int	to_filestat_flags(int flags);
1540daf62d9SStanislav Sedov static int	procstat_get_vnode_info_kvm(kvm_t *kd, struct filestat *fst,
1550daf62d9SStanislav Sedov     struct vnstat *vn, char *errbuf);
1560daf62d9SStanislav Sedov static int	procstat_get_vnode_info_sysctl(struct filestat *fst,
1570daf62d9SStanislav Sedov     struct vnstat *vn, char *errbuf);
1587f1d14e6SMikolaj Golub static gid_t	*procstat_getgroups_core(struct procstat_core *core,
1597f1d14e6SMikolaj Golub     unsigned int *count);
1601f84c47eSMikolaj Golub static gid_t *	procstat_getgroups_kvm(kvm_t *kd, struct kinfo_proc *kp,
1611f84c47eSMikolaj Golub     unsigned int *count);
1627f1d14e6SMikolaj Golub static gid_t	*procstat_getgroups_sysctl(pid_t pid, unsigned int *count);
16389358231SMikolaj Golub static struct kinfo_kstack	*procstat_getkstack_sysctl(pid_t pid,
16489358231SMikolaj Golub     int *cntp);
1651f84c47eSMikolaj Golub static int	procstat_getosrel_core(struct procstat_core *core,
1661f84c47eSMikolaj Golub     int *osrelp);
1671f84c47eSMikolaj Golub static int	procstat_getosrel_kvm(kvm_t *kd, struct kinfo_proc *kp,
1681f84c47eSMikolaj Golub     int *osrelp);
1691f84c47eSMikolaj Golub static int	procstat_getosrel_sysctl(pid_t pid, int *osrelp);
1704cdf9796SMikolaj Golub static int	procstat_getpathname_core(struct procstat_core *core,
1714cdf9796SMikolaj Golub     char *pathname, size_t maxlen);
1724cdf9796SMikolaj Golub static int	procstat_getpathname_sysctl(pid_t pid, char *pathname,
1734cdf9796SMikolaj Golub     size_t maxlen);
1747cc0ebfdSMikolaj Golub static int	procstat_getrlimit_core(struct procstat_core *core, int which,
1757cc0ebfdSMikolaj Golub     struct rlimit* rlimit);
1761f84c47eSMikolaj Golub static int	procstat_getrlimit_kvm(kvm_t *kd, struct kinfo_proc *kp,
1771f84c47eSMikolaj Golub     int which, struct rlimit* rlimit);
1787cc0ebfdSMikolaj Golub static int	procstat_getrlimit_sysctl(pid_t pid, int which,
1797cc0ebfdSMikolaj Golub     struct rlimit* rlimit);
1805b9bcba9SMikolaj Golub static int	procstat_getumask_core(struct procstat_core *core,
1815b9bcba9SMikolaj Golub     unsigned short *maskp);
1821f84c47eSMikolaj Golub static int	procstat_getumask_kvm(kvm_t *kd, struct kinfo_proc *kp,
1831f84c47eSMikolaj Golub     unsigned short *maskp);
1845b9bcba9SMikolaj Golub static int	procstat_getumask_sysctl(pid_t pid, unsigned short *maskp);
1850daf62d9SStanislav Sedov static int	vntype2psfsttype(int type);
1860daf62d9SStanislav Sedov 
1870daf62d9SStanislav Sedov void
1880daf62d9SStanislav Sedov procstat_close(struct procstat *procstat)
1890daf62d9SStanislav Sedov {
1900daf62d9SStanislav Sedov 
1910daf62d9SStanislav Sedov 	assert(procstat);
1920daf62d9SStanislav Sedov 	if (procstat->type == PROCSTAT_KVM)
1930daf62d9SStanislav Sedov 		kvm_close(procstat->kd);
1947153ad2bSMikolaj Golub 	else if (procstat->type == PROCSTAT_CORE)
1957153ad2bSMikolaj Golub 		procstat_core_close(procstat->core);
1964482b5e3SMikolaj Golub 	procstat_freeargv(procstat);
1974482b5e3SMikolaj Golub 	procstat_freeenvv(procstat);
198d7b666aeSSergey Kandaurov 	free(procstat);
1990daf62d9SStanislav Sedov }
2000daf62d9SStanislav Sedov 
2010daf62d9SStanislav Sedov struct procstat *
2020daf62d9SStanislav Sedov procstat_open_sysctl(void)
2030daf62d9SStanislav Sedov {
2040daf62d9SStanislav Sedov 	struct procstat *procstat;
2050daf62d9SStanislav Sedov 
2060daf62d9SStanislav Sedov 	procstat = calloc(1, sizeof(*procstat));
2070daf62d9SStanislav Sedov 	if (procstat == NULL) {
2080daf62d9SStanislav Sedov 		warn("malloc()");
2090daf62d9SStanislav Sedov 		return (NULL);
2100daf62d9SStanislav Sedov 	}
2110daf62d9SStanislav Sedov 	procstat->type = PROCSTAT_SYSCTL;
2120daf62d9SStanislav Sedov 	return (procstat);
2130daf62d9SStanislav Sedov }
2140daf62d9SStanislav Sedov 
2150daf62d9SStanislav Sedov struct procstat *
2160daf62d9SStanislav Sedov procstat_open_kvm(const char *nlistf, const char *memf)
2170daf62d9SStanislav Sedov {
2180daf62d9SStanislav Sedov 	struct procstat *procstat;
2190daf62d9SStanislav Sedov 	kvm_t *kd;
2200daf62d9SStanislav Sedov 	char buf[_POSIX2_LINE_MAX];
2210daf62d9SStanislav Sedov 
2220daf62d9SStanislav Sedov 	procstat = calloc(1, sizeof(*procstat));
2230daf62d9SStanislav Sedov 	if (procstat == NULL) {
2240daf62d9SStanislav Sedov 		warn("malloc()");
2250daf62d9SStanislav Sedov 		return (NULL);
2260daf62d9SStanislav Sedov 	}
2270daf62d9SStanislav Sedov 	kd = kvm_openfiles(nlistf, memf, NULL, O_RDONLY, buf);
2280daf62d9SStanislav Sedov 	if (kd == NULL) {
2290daf62d9SStanislav Sedov 		warnx("kvm_openfiles(): %s", buf);
2300daf62d9SStanislav Sedov 		free(procstat);
2310daf62d9SStanislav Sedov 		return (NULL);
2320daf62d9SStanislav Sedov 	}
2330daf62d9SStanislav Sedov 	procstat->type = PROCSTAT_KVM;
2340daf62d9SStanislav Sedov 	procstat->kd = kd;
2350daf62d9SStanislav Sedov 	return (procstat);
2360daf62d9SStanislav Sedov }
2370daf62d9SStanislav Sedov 
2387153ad2bSMikolaj Golub struct procstat *
2397153ad2bSMikolaj Golub procstat_open_core(const char *filename)
2407153ad2bSMikolaj Golub {
2417153ad2bSMikolaj Golub 	struct procstat *procstat;
2427153ad2bSMikolaj Golub 	struct procstat_core *core;
2437153ad2bSMikolaj Golub 
2447153ad2bSMikolaj Golub 	procstat = calloc(1, sizeof(*procstat));
2457153ad2bSMikolaj Golub 	if (procstat == NULL) {
2467153ad2bSMikolaj Golub 		warn("malloc()");
2477153ad2bSMikolaj Golub 		return (NULL);
2487153ad2bSMikolaj Golub 	}
2497153ad2bSMikolaj Golub 	core = procstat_core_open(filename);
2507153ad2bSMikolaj Golub 	if (core == NULL) {
2517153ad2bSMikolaj Golub 		free(procstat);
2527153ad2bSMikolaj Golub 		return (NULL);
2537153ad2bSMikolaj Golub 	}
2547153ad2bSMikolaj Golub 	procstat->type = PROCSTAT_CORE;
2557153ad2bSMikolaj Golub 	procstat->core = core;
2567153ad2bSMikolaj Golub 	return (procstat);
2577153ad2bSMikolaj Golub }
2587153ad2bSMikolaj Golub 
2590daf62d9SStanislav Sedov struct kinfo_proc *
2600daf62d9SStanislav Sedov procstat_getprocs(struct procstat *procstat, int what, int arg,
2610daf62d9SStanislav Sedov     unsigned int *count)
2620daf62d9SStanislav Sedov {
2630daf62d9SStanislav Sedov 	struct kinfo_proc *p0, *p;
264608203fdSJohn Baldwin 	size_t len, olen;
2650daf62d9SStanislav Sedov 	int name[4];
266391bdfb8SAndriy Gapon 	int cnt;
2670daf62d9SStanislav Sedov 	int error;
2680daf62d9SStanislav Sedov 
2690daf62d9SStanislav Sedov 	assert(procstat);
2700daf62d9SStanislav Sedov 	assert(count);
2710daf62d9SStanislav Sedov 	p = NULL;
2720daf62d9SStanislav Sedov 	if (procstat->type == PROCSTAT_KVM) {
273391bdfb8SAndriy Gapon 		*count = 0;
274391bdfb8SAndriy Gapon 		p0 = kvm_getprocs(procstat->kd, what, arg, &cnt);
275391bdfb8SAndriy Gapon 		if (p0 == NULL || cnt <= 0)
2760daf62d9SStanislav Sedov 			return (NULL);
277391bdfb8SAndriy Gapon 		*count = cnt;
2780daf62d9SStanislav Sedov 		len = *count * sizeof(*p);
2790daf62d9SStanislav Sedov 		p = malloc(len);
2800daf62d9SStanislav Sedov 		if (p == NULL) {
28165869214SJilles Tjoelker 			warnx("malloc(%zu)", len);
2820daf62d9SStanislav Sedov 			goto fail;
2830daf62d9SStanislav Sedov 		}
2840daf62d9SStanislav Sedov 		bcopy(p0, p, len);
2850daf62d9SStanislav Sedov 		return (p);
2860daf62d9SStanislav Sedov 	} else if (procstat->type == PROCSTAT_SYSCTL) {
2870daf62d9SStanislav Sedov 		len = 0;
2880daf62d9SStanislav Sedov 		name[0] = CTL_KERN;
2890daf62d9SStanislav Sedov 		name[1] = KERN_PROC;
2900daf62d9SStanislav Sedov 		name[2] = what;
2910daf62d9SStanislav Sedov 		name[3] = arg;
29275b6a179SEnji Cooper 		error = sysctl(name, nitems(name), NULL, &len, NULL, 0);
2930daf62d9SStanislav Sedov 		if (error < 0 && errno != EPERM) {
2940daf62d9SStanislav Sedov 			warn("sysctl(kern.proc)");
2950daf62d9SStanislav Sedov 			goto fail;
2960daf62d9SStanislav Sedov 		}
2970daf62d9SStanislav Sedov 		if (len == 0) {
2980daf62d9SStanislav Sedov 			warnx("no processes?");
2990daf62d9SStanislav Sedov 			goto fail;
3000daf62d9SStanislav Sedov 		}
301608203fdSJohn Baldwin 		do {
302608203fdSJohn Baldwin 			len += len / 10;
303608203fdSJohn Baldwin 			p = reallocf(p, len);
3040daf62d9SStanislav Sedov 			if (p == NULL) {
305608203fdSJohn Baldwin 				warnx("reallocf(%zu)", len);
3060daf62d9SStanislav Sedov 				goto fail;
3070daf62d9SStanislav Sedov 			}
308608203fdSJohn Baldwin 			olen = len;
30975b6a179SEnji Cooper 			error = sysctl(name, nitems(name), p, &len, NULL, 0);
310608203fdSJohn Baldwin 		} while (error < 0 && errno == ENOMEM && olen == len);
3110daf62d9SStanislav Sedov 		if (error < 0 && errno != EPERM) {
3120daf62d9SStanislav Sedov 			warn("sysctl(kern.proc)");
3130daf62d9SStanislav Sedov 			goto fail;
3140daf62d9SStanislav Sedov 		}
3150daf62d9SStanislav Sedov 		/* Perform simple consistency checks. */
3160daf62d9SStanislav Sedov 		if ((len % sizeof(*p)) != 0 || p->ki_structsize != sizeof(*p)) {
3177153ad2bSMikolaj Golub 			warnx("kinfo_proc structure size mismatch (len = %zu)", len);
3187153ad2bSMikolaj Golub 			goto fail;
3197153ad2bSMikolaj Golub 		}
3207153ad2bSMikolaj Golub 		*count = len / sizeof(*p);
3217153ad2bSMikolaj Golub 		return (p);
3227153ad2bSMikolaj Golub 	} else if (procstat->type == PROCSTAT_CORE) {
3237153ad2bSMikolaj Golub 		p = procstat_core_get(procstat->core, PSC_TYPE_PROC, NULL,
3247153ad2bSMikolaj Golub 		    &len);
3257153ad2bSMikolaj Golub 		if ((len % sizeof(*p)) != 0 || p->ki_structsize != sizeof(*p)) {
3260daf62d9SStanislav Sedov 			warnx("kinfo_proc structure size mismatch");
3270daf62d9SStanislav Sedov 			goto fail;
3280daf62d9SStanislav Sedov 		}
3290daf62d9SStanislav Sedov 		*count = len / sizeof(*p);
3300daf62d9SStanislav Sedov 		return (p);
3310daf62d9SStanislav Sedov 	} else {
33280905c35SJilles Tjoelker 		warnx("unknown access method: %d", procstat->type);
3330daf62d9SStanislav Sedov 		return (NULL);
3340daf62d9SStanislav Sedov 	}
3350daf62d9SStanislav Sedov fail:
3360daf62d9SStanislav Sedov 	if (p)
3370daf62d9SStanislav Sedov 		free(p);
3380daf62d9SStanislav Sedov 	return (NULL);
3390daf62d9SStanislav Sedov }
3400daf62d9SStanislav Sedov 
3410daf62d9SStanislav Sedov void
3420daf62d9SStanislav Sedov procstat_freeprocs(struct procstat *procstat __unused, struct kinfo_proc *p)
3430daf62d9SStanislav Sedov {
3440daf62d9SStanislav Sedov 
3450daf62d9SStanislav Sedov 	if (p != NULL)
3460daf62d9SStanislav Sedov 		free(p);
3470daf62d9SStanislav Sedov 	p = NULL;
3480daf62d9SStanislav Sedov }
3490daf62d9SStanislav Sedov 
3500daf62d9SStanislav Sedov struct filestat_list *
3510daf62d9SStanislav Sedov procstat_getfiles(struct procstat *procstat, struct kinfo_proc *kp, int mmapped)
3520daf62d9SStanislav Sedov {
3530daf62d9SStanislav Sedov 
3547153ad2bSMikolaj Golub 	switch(procstat->type) {
3557153ad2bSMikolaj Golub 	case PROCSTAT_KVM:
3560daf62d9SStanislav Sedov 		return (procstat_getfiles_kvm(procstat, kp, mmapped));
3577153ad2bSMikolaj Golub 	case PROCSTAT_SYSCTL:
3587153ad2bSMikolaj Golub 	case PROCSTAT_CORE:
3597153ad2bSMikolaj Golub 		return (procstat_getfiles_sysctl(procstat, kp, mmapped));
3607153ad2bSMikolaj Golub 	default:
3617153ad2bSMikolaj Golub 		warnx("unknown access method: %d", procstat->type);
3620daf62d9SStanislav Sedov 		return (NULL);
3630daf62d9SStanislav Sedov 	}
3647153ad2bSMikolaj Golub }
3650daf62d9SStanislav Sedov 
3660daf62d9SStanislav Sedov void
3670daf62d9SStanislav Sedov procstat_freefiles(struct procstat *procstat, struct filestat_list *head)
3680daf62d9SStanislav Sedov {
3690daf62d9SStanislav Sedov 	struct filestat *fst, *tmp;
3700daf62d9SStanislav Sedov 
3710daf62d9SStanislav Sedov 	STAILQ_FOREACH_SAFE(fst, head, next, tmp) {
3720daf62d9SStanislav Sedov 		if (fst->fs_path != NULL)
3730daf62d9SStanislav Sedov 			free(fst->fs_path);
3740daf62d9SStanislav Sedov 		free(fst);
3750daf62d9SStanislav Sedov 	}
3760daf62d9SStanislav Sedov 	free(head);
3770daf62d9SStanislav Sedov 	if (procstat->vmentries != NULL) {
3780daf62d9SStanislav Sedov 		free(procstat->vmentries);
3790daf62d9SStanislav Sedov 		procstat->vmentries = NULL;
3800daf62d9SStanislav Sedov 	}
3810daf62d9SStanislav Sedov 	if (procstat->files != NULL) {
3820daf62d9SStanislav Sedov 		free(procstat->files);
3830daf62d9SStanislav Sedov 		procstat->files = NULL;
3840daf62d9SStanislav Sedov 	}
3850daf62d9SStanislav Sedov }
3860daf62d9SStanislav Sedov 
3870daf62d9SStanislav Sedov static struct filestat *
3880daf62d9SStanislav Sedov filestat_new_entry(void *typedep, int type, int fd, int fflags, int uflags,
3897008be5bSPawel Jakub Dawidek     int refcount, off_t offset, char *path, cap_rights_t *cap_rightsp)
3900daf62d9SStanislav Sedov {
3910daf62d9SStanislav Sedov 	struct filestat *entry;
3920daf62d9SStanislav Sedov 
3930daf62d9SStanislav Sedov 	entry = calloc(1, sizeof(*entry));
3940daf62d9SStanislav Sedov 	if (entry == NULL) {
3950daf62d9SStanislav Sedov 		warn("malloc()");
3960daf62d9SStanislav Sedov 		return (NULL);
3970daf62d9SStanislav Sedov 	}
3980daf62d9SStanislav Sedov 	entry->fs_typedep = typedep;
3990daf62d9SStanislav Sedov 	entry->fs_fflags = fflags;
4000daf62d9SStanislav Sedov 	entry->fs_uflags = uflags;
4010daf62d9SStanislav Sedov 	entry->fs_fd = fd;
4020daf62d9SStanislav Sedov 	entry->fs_type = type;
4030daf62d9SStanislav Sedov 	entry->fs_ref_count = refcount;
4040daf62d9SStanislav Sedov 	entry->fs_offset = offset;
4050daf62d9SStanislav Sedov 	entry->fs_path = path;
406772f6645SPawel Jakub Dawidek 	if (cap_rightsp != NULL)
4077008be5bSPawel Jakub Dawidek 		entry->fs_cap_rights = *cap_rightsp;
408772f6645SPawel Jakub Dawidek 	else
409772f6645SPawel Jakub Dawidek 		cap_rights_init(&entry->fs_cap_rights);
4100daf62d9SStanislav Sedov 	return (entry);
4110daf62d9SStanislav Sedov }
4120daf62d9SStanislav Sedov 
4130daf62d9SStanislav Sedov static struct vnode *
4140daf62d9SStanislav Sedov getctty(kvm_t *kd, struct kinfo_proc *kp)
4150daf62d9SStanislav Sedov {
4160daf62d9SStanislav Sedov 	struct pgrp pgrp;
4170daf62d9SStanislav Sedov 	struct proc proc;
4180daf62d9SStanislav Sedov 	struct session sess;
4190daf62d9SStanislav Sedov 	int error;
4200daf62d9SStanislav Sedov 
4210daf62d9SStanislav Sedov 	assert(kp);
4220daf62d9SStanislav Sedov 	error = kvm_read_all(kd, (unsigned long)kp->ki_paddr, &proc,
4230daf62d9SStanislav Sedov 	    sizeof(proc));
4240daf62d9SStanislav Sedov 	if (error == 0) {
4250daf62d9SStanislav Sedov 		warnx("can't read proc struct at %p for pid %d",
4260daf62d9SStanislav Sedov 		    kp->ki_paddr, kp->ki_pid);
4270daf62d9SStanislav Sedov 		return (NULL);
4280daf62d9SStanislav Sedov 	}
4290daf62d9SStanislav Sedov 	if (proc.p_pgrp == NULL)
4300daf62d9SStanislav Sedov 		return (NULL);
4310daf62d9SStanislav Sedov 	error = kvm_read_all(kd, (unsigned long)proc.p_pgrp, &pgrp,
4320daf62d9SStanislav Sedov 	    sizeof(pgrp));
4330daf62d9SStanislav Sedov 	if (error == 0) {
4340daf62d9SStanislav Sedov 		warnx("can't read pgrp struct at %p for pid %d",
4350daf62d9SStanislav Sedov 		    proc.p_pgrp, kp->ki_pid);
4360daf62d9SStanislav Sedov 		return (NULL);
4370daf62d9SStanislav Sedov 	}
4380daf62d9SStanislav Sedov 	error = kvm_read_all(kd, (unsigned long)pgrp.pg_session, &sess,
4390daf62d9SStanislav Sedov 	    sizeof(sess));
4400daf62d9SStanislav Sedov 	if (error == 0) {
4410daf62d9SStanislav Sedov 		warnx("can't read session struct at %p for pid %d",
4420daf62d9SStanislav Sedov 		    pgrp.pg_session, kp->ki_pid);
4430daf62d9SStanislav Sedov 		return (NULL);
4440daf62d9SStanislav Sedov 	}
4450daf62d9SStanislav Sedov 	return (sess.s_ttyvp);
4460daf62d9SStanislav Sedov }
4470daf62d9SStanislav Sedov 
4488b75b1adSDoug Moore static int
4498b75b1adSDoug Moore procstat_vm_map_reader(void *token, vm_map_entry_t addr, vm_map_entry_t dest)
4508b75b1adSDoug Moore {
4518b75b1adSDoug Moore 	kvm_t *kd;
4528b75b1adSDoug Moore 
4538b75b1adSDoug Moore 	kd = (kvm_t *)token;
4548b75b1adSDoug Moore 	return (kvm_read_all(kd, (unsigned long)addr, dest, sizeof(*dest)));
4558b75b1adSDoug Moore }
4568b75b1adSDoug Moore 
4570daf62d9SStanislav Sedov static struct filestat_list *
4580daf62d9SStanislav Sedov procstat_getfiles_kvm(struct procstat *procstat, struct kinfo_proc *kp, int mmapped)
4590daf62d9SStanislav Sedov {
4600daf62d9SStanislav Sedov 	struct file file;
4610daf62d9SStanislav Sedov 	struct filedesc filed;
462*8d03b99bSMateusz Guzik 	struct pwd pwd;
4630daf62d9SStanislav Sedov 	struct vm_map_entry vmentry;
4640daf62d9SStanislav Sedov 	struct vm_object object;
4650daf62d9SStanislav Sedov 	struct vmspace vmspace;
4660daf62d9SStanislav Sedov 	vm_map_entry_t entryp;
4670daf62d9SStanislav Sedov 	vm_object_t objp;
4680daf62d9SStanislav Sedov 	struct vnode *vp;
4690daf62d9SStanislav Sedov 	struct file **ofiles;
4700daf62d9SStanislav Sedov 	struct filestat *entry;
4710daf62d9SStanislav Sedov 	struct filestat_list *head;
4720daf62d9SStanislav Sedov 	kvm_t *kd;
4730daf62d9SStanislav Sedov 	void *data;
4740daf62d9SStanislav Sedov 	int i, fflags;
4750daf62d9SStanislav Sedov 	int prot, type;
4760daf62d9SStanislav Sedov 	unsigned int nfiles;
477*8d03b99bSMateusz Guzik 	bool haspwd;
4780daf62d9SStanislav Sedov 
4790daf62d9SStanislav Sedov 	assert(procstat);
4800daf62d9SStanislav Sedov 	kd = procstat->kd;
4810daf62d9SStanislav Sedov 	if (kd == NULL)
4820daf62d9SStanislav Sedov 		return (NULL);
4830daf62d9SStanislav Sedov 	if (kp->ki_fd == NULL)
4840daf62d9SStanislav Sedov 		return (NULL);
4850daf62d9SStanislav Sedov 	if (!kvm_read_all(kd, (unsigned long)kp->ki_fd, &filed,
4860daf62d9SStanislav Sedov 	    sizeof(filed))) {
4870daf62d9SStanislav Sedov 		warnx("can't read filedesc at %p", (void *)kp->ki_fd);
4880daf62d9SStanislav Sedov 		return (NULL);
4890daf62d9SStanislav Sedov 	}
490*8d03b99bSMateusz Guzik 	haspwd = false;
491*8d03b99bSMateusz Guzik 	if (filed.fd_pwd != NULL) {
492*8d03b99bSMateusz Guzik 		if (!kvm_read_all(kd, (unsigned long)filed.fd_pwd, &pwd,
493*8d03b99bSMateusz Guzik 		    sizeof(pwd))) {
494*8d03b99bSMateusz Guzik 			warnx("can't read fd_pwd at %p", (void *)filed.fd_pwd);
495*8d03b99bSMateusz Guzik 			return (NULL);
496*8d03b99bSMateusz Guzik 		}
497*8d03b99bSMateusz Guzik 		haspwd = true;
498*8d03b99bSMateusz Guzik 	}
4990daf62d9SStanislav Sedov 
5000daf62d9SStanislav Sedov 	/*
5010daf62d9SStanislav Sedov 	 * Allocate list head.
5020daf62d9SStanislav Sedov 	 */
5030daf62d9SStanislav Sedov 	head = malloc(sizeof(*head));
5040daf62d9SStanislav Sedov 	if (head == NULL)
5050daf62d9SStanislav Sedov 		return (NULL);
5060daf62d9SStanislav Sedov 	STAILQ_INIT(head);
5070daf62d9SStanislav Sedov 
5080daf62d9SStanislav Sedov 	/* root directory vnode, if one. */
509*8d03b99bSMateusz Guzik 	if (haspwd) {
510*8d03b99bSMateusz Guzik 		if (pwd.pwd_rdir) {
511*8d03b99bSMateusz Guzik 			entry = filestat_new_entry(pwd.pwd_rdir, PS_FST_TYPE_VNODE, -1,
512772f6645SPawel Jakub Dawidek 			    PS_FST_FFLAG_READ, PS_FST_UFLAG_RDIR, 0, 0, NULL, NULL);
5130daf62d9SStanislav Sedov 			if (entry != NULL)
5140daf62d9SStanislav Sedov 				STAILQ_INSERT_TAIL(head, entry, next);
5150daf62d9SStanislav Sedov 		}
5160daf62d9SStanislav Sedov 		/* current working directory vnode. */
517*8d03b99bSMateusz Guzik 		if (pwd.pwd_cdir) {
518*8d03b99bSMateusz Guzik 			entry = filestat_new_entry(pwd.pwd_cdir, PS_FST_TYPE_VNODE, -1,
519772f6645SPawel Jakub Dawidek 			    PS_FST_FFLAG_READ, PS_FST_UFLAG_CDIR, 0, 0, NULL, NULL);
5200daf62d9SStanislav Sedov 			if (entry != NULL)
5210daf62d9SStanislav Sedov 				STAILQ_INSERT_TAIL(head, entry, next);
5220daf62d9SStanislav Sedov 		}
5230daf62d9SStanislav Sedov 		/* jail root, if any. */
524*8d03b99bSMateusz Guzik 		if (pwd.pwd_jdir) {
525*8d03b99bSMateusz Guzik 			entry = filestat_new_entry(pwd.pwd_jdir, PS_FST_TYPE_VNODE, -1,
526772f6645SPawel Jakub Dawidek 			    PS_FST_FFLAG_READ, PS_FST_UFLAG_JAIL, 0, 0, NULL, NULL);
5270daf62d9SStanislav Sedov 			if (entry != NULL)
5280daf62d9SStanislav Sedov 				STAILQ_INSERT_TAIL(head, entry, next);
5290daf62d9SStanislav Sedov 		}
530*8d03b99bSMateusz Guzik 	}
5310daf62d9SStanislav Sedov 	/* ktrace vnode, if one */
5320daf62d9SStanislav Sedov 	if (kp->ki_tracep) {
5330daf62d9SStanislav Sedov 		entry = filestat_new_entry(kp->ki_tracep, PS_FST_TYPE_VNODE, -1,
5340daf62d9SStanislav Sedov 		    PS_FST_FFLAG_READ | PS_FST_FFLAG_WRITE,
535772f6645SPawel Jakub Dawidek 		    PS_FST_UFLAG_TRACE, 0, 0, NULL, NULL);
5360daf62d9SStanislav Sedov 		if (entry != NULL)
5370daf62d9SStanislav Sedov 			STAILQ_INSERT_TAIL(head, entry, next);
5380daf62d9SStanislav Sedov 	}
5390daf62d9SStanislav Sedov 	/* text vnode, if one */
5400daf62d9SStanislav Sedov 	if (kp->ki_textvp) {
5410daf62d9SStanislav Sedov 		entry = filestat_new_entry(kp->ki_textvp, PS_FST_TYPE_VNODE, -1,
542772f6645SPawel Jakub Dawidek 		    PS_FST_FFLAG_READ, PS_FST_UFLAG_TEXT, 0, 0, NULL, NULL);
5430daf62d9SStanislav Sedov 		if (entry != NULL)
5440daf62d9SStanislav Sedov 			STAILQ_INSERT_TAIL(head, entry, next);
5450daf62d9SStanislav Sedov 	}
5460daf62d9SStanislav Sedov 	/* Controlling terminal. */
5470daf62d9SStanislav Sedov 	if ((vp = getctty(kd, kp)) != NULL) {
5480daf62d9SStanislav Sedov 		entry = filestat_new_entry(vp, PS_FST_TYPE_VNODE, -1,
5490daf62d9SStanislav Sedov 		    PS_FST_FFLAG_READ | PS_FST_FFLAG_WRITE,
550772f6645SPawel Jakub Dawidek 		    PS_FST_UFLAG_CTTY, 0, 0, NULL, NULL);
5510daf62d9SStanislav Sedov 		if (entry != NULL)
5520daf62d9SStanislav Sedov 			STAILQ_INSERT_TAIL(head, entry, next);
5530daf62d9SStanislav Sedov 	}
5540daf62d9SStanislav Sedov 
5550daf62d9SStanislav Sedov 	nfiles = filed.fd_lastfile + 1;
5560daf62d9SStanislav Sedov 	ofiles = malloc(nfiles * sizeof(struct file *));
5570daf62d9SStanislav Sedov 	if (ofiles == NULL) {
55865869214SJilles Tjoelker 		warn("malloc(%zu)", nfiles * sizeof(struct file *));
5590daf62d9SStanislav Sedov 		goto do_mmapped;
5600daf62d9SStanislav Sedov 	}
5610daf62d9SStanislav Sedov 	if (!kvm_read_all(kd, (unsigned long)filed.fd_ofiles, ofiles,
5620daf62d9SStanislav Sedov 	    nfiles * sizeof(struct file *))) {
5630daf62d9SStanislav Sedov 		warnx("cannot read file structures at %p",
5640daf62d9SStanislav Sedov 		    (void *)filed.fd_ofiles);
5650daf62d9SStanislav Sedov 		free(ofiles);
5660daf62d9SStanislav Sedov 		goto do_mmapped;
5670daf62d9SStanislav Sedov 	}
5680daf62d9SStanislav Sedov 	for (i = 0; i <= filed.fd_lastfile; i++) {
5690daf62d9SStanislav Sedov 		if (ofiles[i] == NULL)
5700daf62d9SStanislav Sedov 			continue;
5710daf62d9SStanislav Sedov 		if (!kvm_read_all(kd, (unsigned long)ofiles[i], &file,
5720daf62d9SStanislav Sedov 		    sizeof(struct file))) {
5730daf62d9SStanislav Sedov 			warnx("can't read file %d at %p", i,
5740daf62d9SStanislav Sedov 			    (void *)ofiles[i]);
5750daf62d9SStanislav Sedov 			continue;
5760daf62d9SStanislav Sedov 		}
5770daf62d9SStanislav Sedov 		switch (file.f_type) {
5780daf62d9SStanislav Sedov 		case DTYPE_VNODE:
5790daf62d9SStanislav Sedov 			type = PS_FST_TYPE_VNODE;
5800daf62d9SStanislav Sedov 			data = file.f_vnode;
5810daf62d9SStanislav Sedov 			break;
5820daf62d9SStanislav Sedov 		case DTYPE_SOCKET:
5830daf62d9SStanislav Sedov 			type = PS_FST_TYPE_SOCKET;
5840daf62d9SStanislav Sedov 			data = file.f_data;
5850daf62d9SStanislav Sedov 			break;
5860daf62d9SStanislav Sedov 		case DTYPE_PIPE:
5870daf62d9SStanislav Sedov 			type = PS_FST_TYPE_PIPE;
5880daf62d9SStanislav Sedov 			data = file.f_data;
5890daf62d9SStanislav Sedov 			break;
5900daf62d9SStanislav Sedov 		case DTYPE_FIFO:
5910daf62d9SStanislav Sedov 			type = PS_FST_TYPE_FIFO;
5920daf62d9SStanislav Sedov 			data = file.f_vnode;
5930daf62d9SStanislav Sedov 			break;
5940daf62d9SStanislav Sedov #ifdef DTYPE_PTS
5950daf62d9SStanislav Sedov 		case DTYPE_PTS:
5960daf62d9SStanislav Sedov 			type = PS_FST_TYPE_PTS;
5970daf62d9SStanislav Sedov 			data = file.f_data;
5980daf62d9SStanislav Sedov 			break;
5990daf62d9SStanislav Sedov #endif
600958aa575SJohn Baldwin 		case DTYPE_SEM:
601958aa575SJohn Baldwin 			type = PS_FST_TYPE_SEM;
602958aa575SJohn Baldwin 			data = file.f_data;
603958aa575SJohn Baldwin 			break;
604e506e182SJohn Baldwin 		case DTYPE_SHM:
605e506e182SJohn Baldwin 			type = PS_FST_TYPE_SHM;
606e506e182SJohn Baldwin 			data = file.f_data;
607e506e182SJohn Baldwin 			break;
6083cfa7c6eSEdward Tomasz Napierala 		case DTYPE_PROCDESC:
6093cfa7c6eSEdward Tomasz Napierala 			type = PS_FST_TYPE_PROCDESC;
6103cfa7c6eSEdward Tomasz Napierala 			data = file.f_data;
6113cfa7c6eSEdward Tomasz Napierala 			break;
612a66732deSKonstantin Belousov 		case DTYPE_DEV:
613a66732deSKonstantin Belousov 			type = PS_FST_TYPE_DEV;
614a66732deSKonstantin Belousov 			data = file.f_data;
615a66732deSKonstantin Belousov 			break;
6160daf62d9SStanislav Sedov 		default:
6170daf62d9SStanislav Sedov 			continue;
6180daf62d9SStanislav Sedov 		}
619d57486e2SRobert Watson 		/* XXXRW: No capability rights support for kvm yet. */
6200daf62d9SStanislav Sedov 		entry = filestat_new_entry(data, type, i,
621772f6645SPawel Jakub Dawidek 		    to_filestat_flags(file.f_flag), 0, 0, 0, NULL, NULL);
6220daf62d9SStanislav Sedov 		if (entry != NULL)
6230daf62d9SStanislav Sedov 			STAILQ_INSERT_TAIL(head, entry, next);
6240daf62d9SStanislav Sedov 	}
6250daf62d9SStanislav Sedov 	free(ofiles);
6260daf62d9SStanislav Sedov 
6270daf62d9SStanislav Sedov do_mmapped:
6280daf62d9SStanislav Sedov 
6290daf62d9SStanislav Sedov 	/*
6300daf62d9SStanislav Sedov 	 * Process mmapped files if requested.
6310daf62d9SStanislav Sedov 	 */
6320daf62d9SStanislav Sedov 	if (mmapped) {
6330daf62d9SStanislav Sedov 		if (!kvm_read_all(kd, (unsigned long)kp->ki_vmspace, &vmspace,
6340daf62d9SStanislav Sedov 		    sizeof(vmspace))) {
6350daf62d9SStanislav Sedov 			warnx("can't read vmspace at %p",
6360daf62d9SStanislav Sedov 			    (void *)kp->ki_vmspace);
6370daf62d9SStanislav Sedov 			goto exit;
6380daf62d9SStanislav Sedov 		}
6390daf62d9SStanislav Sedov 
6408b75b1adSDoug Moore 		vmentry = vmspace.vm_map.header;
6418b75b1adSDoug Moore 		for (entryp = vm_map_entry_read_succ(kd, &vmentry, procstat_vm_map_reader);
6428b75b1adSDoug Moore 		    entryp != NULL && entryp != &kp->ki_vmspace->vm_map.header;
6438b75b1adSDoug Moore 		     entryp = vm_map_entry_read_succ(kd, &vmentry, procstat_vm_map_reader)) {
6440daf62d9SStanislav Sedov 			if (vmentry.eflags & MAP_ENTRY_IS_SUB_MAP)
6450daf62d9SStanislav Sedov 				continue;
6460daf62d9SStanislav Sedov 			if ((objp = vmentry.object.vm_object) == NULL)
6470daf62d9SStanislav Sedov 				continue;
6480daf62d9SStanislav Sedov 			for (; objp; objp = object.backing_object) {
6490daf62d9SStanislav Sedov 				if (!kvm_read_all(kd, (unsigned long)objp,
6500daf62d9SStanislav Sedov 				    &object, sizeof(object))) {
6510daf62d9SStanislav Sedov 					warnx("can't read vm_object at %p",
6520daf62d9SStanislav Sedov 					    (void *)objp);
6530daf62d9SStanislav Sedov 					break;
6540daf62d9SStanislav Sedov 				}
6550daf62d9SStanislav Sedov 			}
6560daf62d9SStanislav Sedov 
6570daf62d9SStanislav Sedov 			/* We want only vnode objects. */
6580daf62d9SStanislav Sedov 			if (object.type != OBJT_VNODE)
6590daf62d9SStanislav Sedov 				continue;
6600daf62d9SStanislav Sedov 
6610daf62d9SStanislav Sedov 			prot = vmentry.protection;
6620daf62d9SStanislav Sedov 			fflags = 0;
6630daf62d9SStanislav Sedov 			if (prot & VM_PROT_READ)
6640daf62d9SStanislav Sedov 				fflags = PS_FST_FFLAG_READ;
665279a233dSJilles Tjoelker 			if ((vmentry.eflags & MAP_ENTRY_COW) == 0 &&
666279a233dSJilles Tjoelker 			    prot & VM_PROT_WRITE)
6670daf62d9SStanislav Sedov 				fflags |= PS_FST_FFLAG_WRITE;
6680daf62d9SStanislav Sedov 
6690daf62d9SStanislav Sedov 			/*
6700daf62d9SStanislav Sedov 			 * Create filestat entry.
6710daf62d9SStanislav Sedov 			 */
6720daf62d9SStanislav Sedov 			entry = filestat_new_entry(object.handle,
6730daf62d9SStanislav Sedov 			    PS_FST_TYPE_VNODE, -1, fflags,
674772f6645SPawel Jakub Dawidek 			    PS_FST_UFLAG_MMAP, 0, 0, NULL, NULL);
6750daf62d9SStanislav Sedov 			if (entry != NULL)
6760daf62d9SStanislav Sedov 				STAILQ_INSERT_TAIL(head, entry, next);
6770daf62d9SStanislav Sedov 		}
6788b75b1adSDoug Moore 		if (entryp == NULL)
6798b75b1adSDoug Moore 			warnx("can't read vm_map_entry");
6800daf62d9SStanislav Sedov 	}
6810daf62d9SStanislav Sedov exit:
6820daf62d9SStanislav Sedov 	return (head);
6830daf62d9SStanislav Sedov }
6840daf62d9SStanislav Sedov 
6850daf62d9SStanislav Sedov /*
6860daf62d9SStanislav Sedov  * kinfo types to filestat translation.
6870daf62d9SStanislav Sedov  */
6880daf62d9SStanislav Sedov static int
6890daf62d9SStanislav Sedov kinfo_type2fst(int kftype)
6900daf62d9SStanislav Sedov {
6910daf62d9SStanislav Sedov 	static struct {
6920daf62d9SStanislav Sedov 		int	kf_type;
6930daf62d9SStanislav Sedov 		int	fst_type;
6940daf62d9SStanislav Sedov 	} kftypes2fst[] = {
6953cfa7c6eSEdward Tomasz Napierala 		{ KF_TYPE_PROCDESC, PS_FST_TYPE_PROCDESC },
6960daf62d9SStanislav Sedov 		{ KF_TYPE_CRYPTO, PS_FST_TYPE_CRYPTO },
697a66732deSKonstantin Belousov 		{ KF_TYPE_DEV, PS_FST_TYPE_DEV },
6980daf62d9SStanislav Sedov 		{ KF_TYPE_FIFO, PS_FST_TYPE_FIFO },
6990daf62d9SStanislav Sedov 		{ KF_TYPE_KQUEUE, PS_FST_TYPE_KQUEUE },
7000daf62d9SStanislav Sedov 		{ KF_TYPE_MQUEUE, PS_FST_TYPE_MQUEUE },
7010daf62d9SStanislav Sedov 		{ KF_TYPE_NONE, PS_FST_TYPE_NONE },
7020daf62d9SStanislav Sedov 		{ KF_TYPE_PIPE, PS_FST_TYPE_PIPE },
7030daf62d9SStanislav Sedov 		{ KF_TYPE_PTS, PS_FST_TYPE_PTS },
7040daf62d9SStanislav Sedov 		{ KF_TYPE_SEM, PS_FST_TYPE_SEM },
7050daf62d9SStanislav Sedov 		{ KF_TYPE_SHM, PS_FST_TYPE_SHM },
7060daf62d9SStanislav Sedov 		{ KF_TYPE_SOCKET, PS_FST_TYPE_SOCKET },
7070daf62d9SStanislav Sedov 		{ KF_TYPE_VNODE, PS_FST_TYPE_VNODE },
7080daf62d9SStanislav Sedov 		{ KF_TYPE_UNKNOWN, PS_FST_TYPE_UNKNOWN }
7090daf62d9SStanislav Sedov 	};
7100daf62d9SStanislav Sedov #define NKFTYPES	(sizeof(kftypes2fst) / sizeof(*kftypes2fst))
7110daf62d9SStanislav Sedov 	unsigned int i;
7120daf62d9SStanislav Sedov 
7130daf62d9SStanislav Sedov 	for (i = 0; i < NKFTYPES; i++)
7140daf62d9SStanislav Sedov 		if (kftypes2fst[i].kf_type == kftype)
7150daf62d9SStanislav Sedov 			break;
7160daf62d9SStanislav Sedov 	if (i == NKFTYPES)
7170daf62d9SStanislav Sedov 		return (PS_FST_TYPE_UNKNOWN);
7180daf62d9SStanislav Sedov 	return (kftypes2fst[i].fst_type);
7190daf62d9SStanislav Sedov }
7200daf62d9SStanislav Sedov 
7210daf62d9SStanislav Sedov /*
7220daf62d9SStanislav Sedov  * kinfo flags to filestat translation.
7230daf62d9SStanislav Sedov  */
7240daf62d9SStanislav Sedov static int
7250daf62d9SStanislav Sedov kinfo_fflags2fst(int kfflags)
7260daf62d9SStanislav Sedov {
7270daf62d9SStanislav Sedov 	static struct {
7280daf62d9SStanislav Sedov 		int	kf_flag;
7290daf62d9SStanislav Sedov 		int	fst_flag;
7300daf62d9SStanislav Sedov 	} kfflags2fst[] = {
7310daf62d9SStanislav Sedov 		{ KF_FLAG_APPEND, PS_FST_FFLAG_APPEND },
7320daf62d9SStanislav Sedov 		{ KF_FLAG_ASYNC, PS_FST_FFLAG_ASYNC },
7330daf62d9SStanislav Sedov 		{ KF_FLAG_CREAT, PS_FST_FFLAG_CREAT },
7340daf62d9SStanislav Sedov 		{ KF_FLAG_DIRECT, PS_FST_FFLAG_DIRECT },
7350daf62d9SStanislav Sedov 		{ KF_FLAG_EXCL, PS_FST_FFLAG_EXCL },
7360daf62d9SStanislav Sedov 		{ KF_FLAG_EXEC, PS_FST_FFLAG_EXEC },
7370daf62d9SStanislav Sedov 		{ KF_FLAG_EXLOCK, PS_FST_FFLAG_EXLOCK },
7380daf62d9SStanislav Sedov 		{ KF_FLAG_FSYNC, PS_FST_FFLAG_SYNC },
7390daf62d9SStanislav Sedov 		{ KF_FLAG_HASLOCK, PS_FST_FFLAG_HASLOCK },
7400daf62d9SStanislav Sedov 		{ KF_FLAG_NOFOLLOW, PS_FST_FFLAG_NOFOLLOW },
7410daf62d9SStanislav Sedov 		{ KF_FLAG_NONBLOCK, PS_FST_FFLAG_NONBLOCK },
7420daf62d9SStanislav Sedov 		{ KF_FLAG_READ, PS_FST_FFLAG_READ },
7430daf62d9SStanislav Sedov 		{ KF_FLAG_SHLOCK, PS_FST_FFLAG_SHLOCK },
7440daf62d9SStanislav Sedov 		{ KF_FLAG_TRUNC, PS_FST_FFLAG_TRUNC },
7450daf62d9SStanislav Sedov 		{ KF_FLAG_WRITE, PS_FST_FFLAG_WRITE }
7460daf62d9SStanislav Sedov 	};
7470daf62d9SStanislav Sedov #define NKFFLAGS	(sizeof(kfflags2fst) / sizeof(*kfflags2fst))
7480daf62d9SStanislav Sedov 	unsigned int i;
7490daf62d9SStanislav Sedov 	int flags;
7500daf62d9SStanislav Sedov 
7510daf62d9SStanislav Sedov 	flags = 0;
7520daf62d9SStanislav Sedov 	for (i = 0; i < NKFFLAGS; i++)
7530daf62d9SStanislav Sedov 		if ((kfflags & kfflags2fst[i].kf_flag) != 0)
7540daf62d9SStanislav Sedov 			flags |= kfflags2fst[i].fst_flag;
7550daf62d9SStanislav Sedov 	return (flags);
7560daf62d9SStanislav Sedov }
7570daf62d9SStanislav Sedov 
7580daf62d9SStanislav Sedov static int
7590daf62d9SStanislav Sedov kinfo_uflags2fst(int fd)
7600daf62d9SStanislav Sedov {
7610daf62d9SStanislav Sedov 
7620daf62d9SStanislav Sedov 	switch (fd) {
7630daf62d9SStanislav Sedov 	case KF_FD_TYPE_CTTY:
7640daf62d9SStanislav Sedov 		return (PS_FST_UFLAG_CTTY);
7650daf62d9SStanislav Sedov 	case KF_FD_TYPE_CWD:
7660daf62d9SStanislav Sedov 		return (PS_FST_UFLAG_CDIR);
7670daf62d9SStanislav Sedov 	case KF_FD_TYPE_JAIL:
7680daf62d9SStanislav Sedov 		return (PS_FST_UFLAG_JAIL);
7690daf62d9SStanislav Sedov 	case KF_FD_TYPE_TEXT:
7700daf62d9SStanislav Sedov 		return (PS_FST_UFLAG_TEXT);
7710daf62d9SStanislav Sedov 	case KF_FD_TYPE_TRACE:
7720daf62d9SStanislav Sedov 		return (PS_FST_UFLAG_TRACE);
7730daf62d9SStanislav Sedov 	case KF_FD_TYPE_ROOT:
7740daf62d9SStanislav Sedov 		return (PS_FST_UFLAG_RDIR);
7750daf62d9SStanislav Sedov 	}
7760daf62d9SStanislav Sedov 	return (0);
7770daf62d9SStanislav Sedov }
7780daf62d9SStanislav Sedov 
7797153ad2bSMikolaj Golub static struct kinfo_file *
7807153ad2bSMikolaj Golub kinfo_getfile_core(struct procstat_core *core, int *cntp)
7817153ad2bSMikolaj Golub {
7827153ad2bSMikolaj Golub 	int cnt;
7837153ad2bSMikolaj Golub 	size_t len;
7847153ad2bSMikolaj Golub 	char *buf, *bp, *eb;
7857153ad2bSMikolaj Golub 	struct kinfo_file *kif, *kp, *kf;
7867153ad2bSMikolaj Golub 
7877153ad2bSMikolaj Golub 	buf = procstat_core_get(core, PSC_TYPE_FILES, NULL, &len);
7887153ad2bSMikolaj Golub 	if (buf == NULL)
7897153ad2bSMikolaj Golub 		return (NULL);
7907153ad2bSMikolaj Golub 	/*
7917153ad2bSMikolaj Golub 	 * XXXMG: The code below is just copy&past from libutil.
7927153ad2bSMikolaj Golub 	 * The code duplication can be avoided if libutil
7937153ad2bSMikolaj Golub 	 * is extended to provide something like:
7947153ad2bSMikolaj Golub 	 *   struct kinfo_file *kinfo_getfile_from_buf(const char *buf,
7957153ad2bSMikolaj Golub 	 *       size_t len, int *cntp);
7967153ad2bSMikolaj Golub 	 */
7977153ad2bSMikolaj Golub 
7987153ad2bSMikolaj Golub 	/* Pass 1: count items */
7997153ad2bSMikolaj Golub 	cnt = 0;
8007153ad2bSMikolaj Golub 	bp = buf;
8017153ad2bSMikolaj Golub 	eb = buf + len;
8027153ad2bSMikolaj Golub 	while (bp < eb) {
8037153ad2bSMikolaj Golub 		kf = (struct kinfo_file *)(uintptr_t)bp;
80414bdbaf2SConrad Meyer 		if (kf->kf_structsize == 0)
80514bdbaf2SConrad Meyer 			break;
8067153ad2bSMikolaj Golub 		bp += kf->kf_structsize;
8077153ad2bSMikolaj Golub 		cnt++;
8087153ad2bSMikolaj Golub 	}
8097153ad2bSMikolaj Golub 
8107153ad2bSMikolaj Golub 	kif = calloc(cnt, sizeof(*kif));
8117153ad2bSMikolaj Golub 	if (kif == NULL) {
8127153ad2bSMikolaj Golub 		free(buf);
8137153ad2bSMikolaj Golub 		return (NULL);
8147153ad2bSMikolaj Golub 	}
8157153ad2bSMikolaj Golub 	bp = buf;
8167153ad2bSMikolaj Golub 	eb = buf + len;
8177153ad2bSMikolaj Golub 	kp = kif;
8187153ad2bSMikolaj Golub 	/* Pass 2: unpack */
8197153ad2bSMikolaj Golub 	while (bp < eb) {
8207153ad2bSMikolaj Golub 		kf = (struct kinfo_file *)(uintptr_t)bp;
82114bdbaf2SConrad Meyer 		if (kf->kf_structsize == 0)
82214bdbaf2SConrad Meyer 			break;
8237153ad2bSMikolaj Golub 		/* Copy/expand into pre-zeroed buffer */
8247153ad2bSMikolaj Golub 		memcpy(kp, kf, kf->kf_structsize);
8257153ad2bSMikolaj Golub 		/* Advance to next packed record */
8267153ad2bSMikolaj Golub 		bp += kf->kf_structsize;
8277153ad2bSMikolaj Golub 		/* Set field size to fixed length, advance */
8287153ad2bSMikolaj Golub 		kp->kf_structsize = sizeof(*kp);
8297153ad2bSMikolaj Golub 		kp++;
8307153ad2bSMikolaj Golub 	}
8317153ad2bSMikolaj Golub 	free(buf);
8327153ad2bSMikolaj Golub 	*cntp = cnt;
8337153ad2bSMikolaj Golub 	return (kif);	/* Caller must free() return value */
8347153ad2bSMikolaj Golub }
8357153ad2bSMikolaj Golub 
8360daf62d9SStanislav Sedov static struct filestat_list *
8377153ad2bSMikolaj Golub procstat_getfiles_sysctl(struct procstat *procstat, struct kinfo_proc *kp,
8387153ad2bSMikolaj Golub     int mmapped)
8390daf62d9SStanislav Sedov {
8400daf62d9SStanislav Sedov 	struct kinfo_file *kif, *files;
8410daf62d9SStanislav Sedov 	struct kinfo_vmentry *kve, *vmentries;
8420daf62d9SStanislav Sedov 	struct filestat_list *head;
8430daf62d9SStanislav Sedov 	struct filestat *entry;
8440daf62d9SStanislav Sedov 	char *path;
8450daf62d9SStanislav Sedov 	off_t offset;
8460daf62d9SStanislav Sedov 	int cnt, fd, fflags;
8470daf62d9SStanislav Sedov 	int i, type, uflags;
8480daf62d9SStanislav Sedov 	int refcount;
849d57486e2SRobert Watson 	cap_rights_t cap_rights;
8500daf62d9SStanislav Sedov 
8510daf62d9SStanislav Sedov 	assert(kp);
8520daf62d9SStanislav Sedov 	if (kp->ki_fd == NULL)
8530daf62d9SStanislav Sedov 		return (NULL);
8547153ad2bSMikolaj Golub 	switch(procstat->type) {
8557153ad2bSMikolaj Golub 	case PROCSTAT_SYSCTL:
8560daf62d9SStanislav Sedov 		files = kinfo_getfile(kp->ki_pid, &cnt);
8577153ad2bSMikolaj Golub 		break;
8587153ad2bSMikolaj Golub 	case PROCSTAT_CORE:
8597153ad2bSMikolaj Golub 		files = kinfo_getfile_core(procstat->core, &cnt);
8607153ad2bSMikolaj Golub 		break;
8617153ad2bSMikolaj Golub 	default:
8627153ad2bSMikolaj Golub 		assert(!"invalid type");
8637153ad2bSMikolaj Golub 	}
8640daf62d9SStanislav Sedov 	if (files == NULL && errno != EPERM) {
8650daf62d9SStanislav Sedov 		warn("kinfo_getfile()");
8660daf62d9SStanislav Sedov 		return (NULL);
8670daf62d9SStanislav Sedov 	}
8680daf62d9SStanislav Sedov 	procstat->files = files;
8690daf62d9SStanislav Sedov 
8700daf62d9SStanislav Sedov 	/*
8710daf62d9SStanislav Sedov 	 * Allocate list head.
8720daf62d9SStanislav Sedov 	 */
8730daf62d9SStanislav Sedov 	head = malloc(sizeof(*head));
8740daf62d9SStanislav Sedov 	if (head == NULL)
8750daf62d9SStanislav Sedov 		return (NULL);
8760daf62d9SStanislav Sedov 	STAILQ_INIT(head);
8770daf62d9SStanislav Sedov 	for (i = 0; i < cnt; i++) {
8780daf62d9SStanislav Sedov 		kif = &files[i];
8790daf62d9SStanislav Sedov 
8800daf62d9SStanislav Sedov 		type = kinfo_type2fst(kif->kf_type);
8810daf62d9SStanislav Sedov 		fd = kif->kf_fd >= 0 ? kif->kf_fd : -1;
8820daf62d9SStanislav Sedov 		fflags = kinfo_fflags2fst(kif->kf_flags);
8830daf62d9SStanislav Sedov 		uflags = kinfo_uflags2fst(kif->kf_fd);
8840daf62d9SStanislav Sedov 		refcount = kif->kf_ref_count;
8850daf62d9SStanislav Sedov 		offset = kif->kf_offset;
8860daf62d9SStanislav Sedov 		if (*kif->kf_path != '\0')
8870daf62d9SStanislav Sedov 			path = strdup(kif->kf_path);
8880daf62d9SStanislav Sedov 		else
8890daf62d9SStanislav Sedov 			path = NULL;
890d57486e2SRobert Watson 		cap_rights = kif->kf_cap_rights;
8910daf62d9SStanislav Sedov 
8920daf62d9SStanislav Sedov 		/*
8930daf62d9SStanislav Sedov 		 * Create filestat entry.
8940daf62d9SStanislav Sedov 		 */
8950daf62d9SStanislav Sedov 		entry = filestat_new_entry(kif, type, fd, fflags, uflags,
8967008be5bSPawel Jakub Dawidek 		    refcount, offset, path, &cap_rights);
8970daf62d9SStanislav Sedov 		if (entry != NULL)
8980daf62d9SStanislav Sedov 			STAILQ_INSERT_TAIL(head, entry, next);
8990daf62d9SStanislav Sedov 	}
9000daf62d9SStanislav Sedov 	if (mmapped != 0) {
90139680c7bSMikolaj Golub 		vmentries = procstat_getvmmap(procstat, kp, &cnt);
9020daf62d9SStanislav Sedov 		procstat->vmentries = vmentries;
9030daf62d9SStanislav Sedov 		if (vmentries == NULL || cnt == 0)
9040daf62d9SStanislav Sedov 			goto fail;
9050daf62d9SStanislav Sedov 		for (i = 0; i < cnt; i++) {
9060daf62d9SStanislav Sedov 			kve = &vmentries[i];
9070daf62d9SStanislav Sedov 			if (kve->kve_type != KVME_TYPE_VNODE)
9080daf62d9SStanislav Sedov 				continue;
9090daf62d9SStanislav Sedov 			fflags = 0;
9100daf62d9SStanislav Sedov 			if (kve->kve_protection & KVME_PROT_READ)
9110daf62d9SStanislav Sedov 				fflags = PS_FST_FFLAG_READ;
912279a233dSJilles Tjoelker 			if ((kve->kve_flags & KVME_FLAG_COW) == 0 &&
913279a233dSJilles Tjoelker 			    kve->kve_protection & KVME_PROT_WRITE)
9140daf62d9SStanislav Sedov 				fflags |= PS_FST_FFLAG_WRITE;
9150daf62d9SStanislav Sedov 			offset = kve->kve_offset;
9160daf62d9SStanislav Sedov 			refcount = kve->kve_ref_count;
9170daf62d9SStanislav Sedov 			if (*kve->kve_path != '\0')
9180daf62d9SStanislav Sedov 				path = strdup(kve->kve_path);
9190daf62d9SStanislav Sedov 			else
9200daf62d9SStanislav Sedov 				path = NULL;
9210daf62d9SStanislav Sedov 			entry = filestat_new_entry(kve, PS_FST_TYPE_VNODE, -1,
922d57486e2SRobert Watson 			    fflags, PS_FST_UFLAG_MMAP, refcount, offset, path,
923772f6645SPawel Jakub Dawidek 			    NULL);
9240daf62d9SStanislav Sedov 			if (entry != NULL)
9250daf62d9SStanislav Sedov 				STAILQ_INSERT_TAIL(head, entry, next);
9260daf62d9SStanislav Sedov 		}
9270daf62d9SStanislav Sedov 	}
9280daf62d9SStanislav Sedov fail:
9290daf62d9SStanislav Sedov 	return (head);
9300daf62d9SStanislav Sedov }
9310daf62d9SStanislav Sedov 
9320daf62d9SStanislav Sedov int
9330daf62d9SStanislav Sedov procstat_get_pipe_info(struct procstat *procstat, struct filestat *fst,
9340daf62d9SStanislav Sedov     struct pipestat *ps, char *errbuf)
9350daf62d9SStanislav Sedov {
9360daf62d9SStanislav Sedov 
9370daf62d9SStanislav Sedov 	assert(ps);
9380daf62d9SStanislav Sedov 	if (procstat->type == PROCSTAT_KVM) {
9390daf62d9SStanislav Sedov 		return (procstat_get_pipe_info_kvm(procstat->kd, fst, ps,
9400daf62d9SStanislav Sedov 		    errbuf));
9417153ad2bSMikolaj Golub 	} else if (procstat->type == PROCSTAT_SYSCTL ||
9427153ad2bSMikolaj Golub 		procstat->type == PROCSTAT_CORE) {
9430daf62d9SStanislav Sedov 		return (procstat_get_pipe_info_sysctl(fst, ps, errbuf));
9440daf62d9SStanislav Sedov 	} else {
94580905c35SJilles Tjoelker 		warnx("unknown access method: %d", procstat->type);
946467112b4SMikolaj Golub 		if (errbuf != NULL)
9470daf62d9SStanislav Sedov 			snprintf(errbuf, _POSIX2_LINE_MAX, "error");
9480daf62d9SStanislav Sedov 		return (1);
9490daf62d9SStanislav Sedov 	}
9500daf62d9SStanislav Sedov }
9510daf62d9SStanislav Sedov 
9520daf62d9SStanislav Sedov static int
9530daf62d9SStanislav Sedov procstat_get_pipe_info_kvm(kvm_t *kd, struct filestat *fst,
9540daf62d9SStanislav Sedov     struct pipestat *ps, char *errbuf)
9550daf62d9SStanislav Sedov {
9560daf62d9SStanislav Sedov 	struct pipe pi;
9570daf62d9SStanislav Sedov 	void *pipep;
9580daf62d9SStanislav Sedov 
9590daf62d9SStanislav Sedov 	assert(kd);
9600daf62d9SStanislav Sedov 	assert(ps);
9610daf62d9SStanislav Sedov 	assert(fst);
9620daf62d9SStanislav Sedov 	bzero(ps, sizeof(*ps));
9630daf62d9SStanislav Sedov 	pipep = fst->fs_typedep;
9640daf62d9SStanislav Sedov 	if (pipep == NULL)
9650daf62d9SStanislav Sedov 		goto fail;
9660daf62d9SStanislav Sedov 	if (!kvm_read_all(kd, (unsigned long)pipep, &pi, sizeof(struct pipe))) {
9670daf62d9SStanislav Sedov 		warnx("can't read pipe at %p", (void *)pipep);
9680daf62d9SStanislav Sedov 		goto fail;
9690daf62d9SStanislav Sedov 	}
9700daf62d9SStanislav Sedov 	ps->addr = (uintptr_t)pipep;
9710daf62d9SStanislav Sedov 	ps->peer = (uintptr_t)pi.pipe_peer;
9720daf62d9SStanislav Sedov 	ps->buffer_cnt = pi.pipe_buffer.cnt;
9730daf62d9SStanislav Sedov 	return (0);
9740daf62d9SStanislav Sedov 
9750daf62d9SStanislav Sedov fail:
976467112b4SMikolaj Golub 	if (errbuf != NULL)
9770daf62d9SStanislav Sedov 		snprintf(errbuf, _POSIX2_LINE_MAX, "error");
9780daf62d9SStanislav Sedov 	return (1);
9790daf62d9SStanislav Sedov }
9800daf62d9SStanislav Sedov 
9810daf62d9SStanislav Sedov static int
9820daf62d9SStanislav Sedov procstat_get_pipe_info_sysctl(struct filestat *fst, struct pipestat *ps,
9830daf62d9SStanislav Sedov     char *errbuf __unused)
9840daf62d9SStanislav Sedov {
9850daf62d9SStanislav Sedov 	struct kinfo_file *kif;
9860daf62d9SStanislav Sedov 
9870daf62d9SStanislav Sedov 	assert(ps);
9880daf62d9SStanislav Sedov 	assert(fst);
9890daf62d9SStanislav Sedov 	bzero(ps, sizeof(*ps));
9900daf62d9SStanislav Sedov 	kif = fst->fs_typedep;
9910daf62d9SStanislav Sedov 	if (kif == NULL)
9920daf62d9SStanislav Sedov 		return (1);
9930daf62d9SStanislav Sedov 	ps->addr = kif->kf_un.kf_pipe.kf_pipe_addr;
9940daf62d9SStanislav Sedov 	ps->peer = kif->kf_un.kf_pipe.kf_pipe_peer;
9950daf62d9SStanislav Sedov 	ps->buffer_cnt = kif->kf_un.kf_pipe.kf_pipe_buffer_cnt;
9960daf62d9SStanislav Sedov 	return (0);
9970daf62d9SStanislav Sedov }
9980daf62d9SStanislav Sedov 
9990daf62d9SStanislav Sedov int
10000daf62d9SStanislav Sedov procstat_get_pts_info(struct procstat *procstat, struct filestat *fst,
10010daf62d9SStanislav Sedov     struct ptsstat *pts, char *errbuf)
10020daf62d9SStanislav Sedov {
10030daf62d9SStanislav Sedov 
10040daf62d9SStanislav Sedov 	assert(pts);
10050daf62d9SStanislav Sedov 	if (procstat->type == PROCSTAT_KVM) {
10060daf62d9SStanislav Sedov 		return (procstat_get_pts_info_kvm(procstat->kd, fst, pts,
10070daf62d9SStanislav Sedov 		    errbuf));
10087153ad2bSMikolaj Golub 	} else if (procstat->type == PROCSTAT_SYSCTL ||
10097153ad2bSMikolaj Golub 		procstat->type == PROCSTAT_CORE) {
10100daf62d9SStanislav Sedov 		return (procstat_get_pts_info_sysctl(fst, pts, errbuf));
10110daf62d9SStanislav Sedov 	} else {
101280905c35SJilles Tjoelker 		warnx("unknown access method: %d", procstat->type);
1013467112b4SMikolaj Golub 		if (errbuf != NULL)
10140daf62d9SStanislav Sedov 			snprintf(errbuf, _POSIX2_LINE_MAX, "error");
10150daf62d9SStanislav Sedov 		return (1);
10160daf62d9SStanislav Sedov 	}
10170daf62d9SStanislav Sedov }
10180daf62d9SStanislav Sedov 
10190daf62d9SStanislav Sedov static int
10200daf62d9SStanislav Sedov procstat_get_pts_info_kvm(kvm_t *kd, struct filestat *fst,
10210daf62d9SStanislav Sedov     struct ptsstat *pts, char *errbuf)
10220daf62d9SStanislav Sedov {
10230daf62d9SStanislav Sedov 	struct tty tty;
10240daf62d9SStanislav Sedov 	void *ttyp;
10250daf62d9SStanislav Sedov 
10260daf62d9SStanislav Sedov 	assert(kd);
10270daf62d9SStanislav Sedov 	assert(pts);
10280daf62d9SStanislav Sedov 	assert(fst);
10290daf62d9SStanislav Sedov 	bzero(pts, sizeof(*pts));
10300daf62d9SStanislav Sedov 	ttyp = fst->fs_typedep;
10310daf62d9SStanislav Sedov 	if (ttyp == NULL)
10320daf62d9SStanislav Sedov 		goto fail;
10330daf62d9SStanislav Sedov 	if (!kvm_read_all(kd, (unsigned long)ttyp, &tty, sizeof(struct tty))) {
10340daf62d9SStanislav Sedov 		warnx("can't read tty at %p", (void *)ttyp);
10350daf62d9SStanislav Sedov 		goto fail;
10360daf62d9SStanislav Sedov 	}
10370daf62d9SStanislav Sedov 	pts->dev = dev2udev(kd, tty.t_dev);
10380daf62d9SStanislav Sedov 	(void)kdevtoname(kd, tty.t_dev, pts->devname);
10390daf62d9SStanislav Sedov 	return (0);
10400daf62d9SStanislav Sedov 
10410daf62d9SStanislav Sedov fail:
1042467112b4SMikolaj Golub 	if (errbuf != NULL)
10430daf62d9SStanislav Sedov 		snprintf(errbuf, _POSIX2_LINE_MAX, "error");
10440daf62d9SStanislav Sedov 	return (1);
10450daf62d9SStanislav Sedov }
10460daf62d9SStanislav Sedov 
10470daf62d9SStanislav Sedov static int
10480daf62d9SStanislav Sedov procstat_get_pts_info_sysctl(struct filestat *fst, struct ptsstat *pts,
10490daf62d9SStanislav Sedov     char *errbuf __unused)
10500daf62d9SStanislav Sedov {
10510daf62d9SStanislav Sedov 	struct kinfo_file *kif;
10520daf62d9SStanislav Sedov 
10530daf62d9SStanislav Sedov 	assert(pts);
10540daf62d9SStanislav Sedov 	assert(fst);
10550daf62d9SStanislav Sedov 	bzero(pts, sizeof(*pts));
10560daf62d9SStanislav Sedov 	kif = fst->fs_typedep;
10570daf62d9SStanislav Sedov 	if (kif == NULL)
10580daf62d9SStanislav Sedov 		return (0);
10590daf62d9SStanislav Sedov 	pts->dev = kif->kf_un.kf_pts.kf_pts_dev;
10600daf62d9SStanislav Sedov 	strlcpy(pts->devname, kif->kf_path, sizeof(pts->devname));
10610daf62d9SStanislav Sedov 	return (0);
10620daf62d9SStanislav Sedov }
10630daf62d9SStanislav Sedov 
10640daf62d9SStanislav Sedov int
1065958aa575SJohn Baldwin procstat_get_sem_info(struct procstat *procstat, struct filestat *fst,
1066958aa575SJohn Baldwin     struct semstat *sem, char *errbuf)
1067958aa575SJohn Baldwin {
1068958aa575SJohn Baldwin 
1069958aa575SJohn Baldwin 	assert(sem);
1070958aa575SJohn Baldwin 	if (procstat->type == PROCSTAT_KVM) {
1071958aa575SJohn Baldwin 		return (procstat_get_sem_info_kvm(procstat->kd, fst, sem,
1072958aa575SJohn Baldwin 		    errbuf));
1073958aa575SJohn Baldwin 	} else if (procstat->type == PROCSTAT_SYSCTL ||
1074958aa575SJohn Baldwin 	    procstat->type == PROCSTAT_CORE) {
1075958aa575SJohn Baldwin 		return (procstat_get_sem_info_sysctl(fst, sem, errbuf));
1076958aa575SJohn Baldwin 	} else {
1077958aa575SJohn Baldwin 		warnx("unknown access method: %d", procstat->type);
1078467112b4SMikolaj Golub 		if (errbuf != NULL)
1079958aa575SJohn Baldwin 			snprintf(errbuf, _POSIX2_LINE_MAX, "error");
1080958aa575SJohn Baldwin 		return (1);
1081958aa575SJohn Baldwin 	}
1082958aa575SJohn Baldwin }
1083958aa575SJohn Baldwin 
1084958aa575SJohn Baldwin static int
1085958aa575SJohn Baldwin procstat_get_sem_info_kvm(kvm_t *kd, struct filestat *fst,
1086958aa575SJohn Baldwin     struct semstat *sem, char *errbuf)
1087958aa575SJohn Baldwin {
1088958aa575SJohn Baldwin 	struct ksem ksem;
1089958aa575SJohn Baldwin 	void *ksemp;
1090958aa575SJohn Baldwin 	char *path;
1091958aa575SJohn Baldwin 	int i;
1092958aa575SJohn Baldwin 
1093958aa575SJohn Baldwin 	assert(kd);
1094958aa575SJohn Baldwin 	assert(sem);
1095958aa575SJohn Baldwin 	assert(fst);
1096958aa575SJohn Baldwin 	bzero(sem, sizeof(*sem));
1097958aa575SJohn Baldwin 	ksemp = fst->fs_typedep;
1098958aa575SJohn Baldwin 	if (ksemp == NULL)
1099958aa575SJohn Baldwin 		goto fail;
1100958aa575SJohn Baldwin 	if (!kvm_read_all(kd, (unsigned long)ksemp, &ksem,
1101958aa575SJohn Baldwin 	    sizeof(struct ksem))) {
1102958aa575SJohn Baldwin 		warnx("can't read ksem at %p", (void *)ksemp);
1103958aa575SJohn Baldwin 		goto fail;
1104958aa575SJohn Baldwin 	}
1105958aa575SJohn Baldwin 	sem->mode = S_IFREG | ksem.ks_mode;
1106958aa575SJohn Baldwin 	sem->value = ksem.ks_value;
1107958aa575SJohn Baldwin 	if (fst->fs_path == NULL && ksem.ks_path != NULL) {
1108958aa575SJohn Baldwin 		path = malloc(MAXPATHLEN);
1109958aa575SJohn Baldwin 		for (i = 0; i < MAXPATHLEN - 1; i++) {
1110958aa575SJohn Baldwin 			if (!kvm_read_all(kd, (unsigned long)ksem.ks_path + i,
1111958aa575SJohn Baldwin 			    path + i, 1))
1112958aa575SJohn Baldwin 				break;
1113958aa575SJohn Baldwin 			if (path[i] == '\0')
1114958aa575SJohn Baldwin 				break;
1115958aa575SJohn Baldwin 		}
1116958aa575SJohn Baldwin 		path[i] = '\0';
1117958aa575SJohn Baldwin 		if (i == 0)
1118958aa575SJohn Baldwin 			free(path);
1119958aa575SJohn Baldwin 		else
1120958aa575SJohn Baldwin 			fst->fs_path = path;
1121958aa575SJohn Baldwin 	}
1122958aa575SJohn Baldwin 	return (0);
1123958aa575SJohn Baldwin 
1124958aa575SJohn Baldwin fail:
1125467112b4SMikolaj Golub 	if (errbuf != NULL)
1126958aa575SJohn Baldwin 		snprintf(errbuf, _POSIX2_LINE_MAX, "error");
1127958aa575SJohn Baldwin 	return (1);
1128958aa575SJohn Baldwin }
1129958aa575SJohn Baldwin 
1130958aa575SJohn Baldwin static int
1131958aa575SJohn Baldwin procstat_get_sem_info_sysctl(struct filestat *fst, struct semstat *sem,
1132958aa575SJohn Baldwin     char *errbuf __unused)
1133958aa575SJohn Baldwin {
1134958aa575SJohn Baldwin 	struct kinfo_file *kif;
1135958aa575SJohn Baldwin 
1136958aa575SJohn Baldwin 	assert(sem);
1137958aa575SJohn Baldwin 	assert(fst);
1138958aa575SJohn Baldwin 	bzero(sem, sizeof(*sem));
1139958aa575SJohn Baldwin 	kif = fst->fs_typedep;
1140958aa575SJohn Baldwin 	if (kif == NULL)
1141958aa575SJohn Baldwin 		return (0);
1142958aa575SJohn Baldwin 	sem->value = kif->kf_un.kf_sem.kf_sem_value;
1143958aa575SJohn Baldwin 	sem->mode = kif->kf_un.kf_sem.kf_sem_mode;
1144958aa575SJohn Baldwin 	return (0);
1145958aa575SJohn Baldwin }
1146958aa575SJohn Baldwin 
1147958aa575SJohn Baldwin int
1148e506e182SJohn Baldwin procstat_get_shm_info(struct procstat *procstat, struct filestat *fst,
1149e506e182SJohn Baldwin     struct shmstat *shm, char *errbuf)
1150e506e182SJohn Baldwin {
1151e506e182SJohn Baldwin 
1152e506e182SJohn Baldwin 	assert(shm);
1153e506e182SJohn Baldwin 	if (procstat->type == PROCSTAT_KVM) {
1154e506e182SJohn Baldwin 		return (procstat_get_shm_info_kvm(procstat->kd, fst, shm,
1155e506e182SJohn Baldwin 		    errbuf));
11567153ad2bSMikolaj Golub 	} else if (procstat->type == PROCSTAT_SYSCTL ||
11577153ad2bSMikolaj Golub 	    procstat->type == PROCSTAT_CORE) {
1158e506e182SJohn Baldwin 		return (procstat_get_shm_info_sysctl(fst, shm, errbuf));
1159e506e182SJohn Baldwin 	} else {
1160e506e182SJohn Baldwin 		warnx("unknown access method: %d", procstat->type);
1161467112b4SMikolaj Golub 		if (errbuf != NULL)
1162e506e182SJohn Baldwin 			snprintf(errbuf, _POSIX2_LINE_MAX, "error");
1163e506e182SJohn Baldwin 		return (1);
1164e506e182SJohn Baldwin 	}
1165e506e182SJohn Baldwin }
1166e506e182SJohn Baldwin 
1167e506e182SJohn Baldwin static int
1168e506e182SJohn Baldwin procstat_get_shm_info_kvm(kvm_t *kd, struct filestat *fst,
1169e506e182SJohn Baldwin     struct shmstat *shm, char *errbuf)
1170e506e182SJohn Baldwin {
1171e506e182SJohn Baldwin 	struct shmfd shmfd;
1172e506e182SJohn Baldwin 	void *shmfdp;
1173a9f5e425SJohn Baldwin 	char *path;
1174a9f5e425SJohn Baldwin 	int i;
1175e506e182SJohn Baldwin 
1176e506e182SJohn Baldwin 	assert(kd);
1177e506e182SJohn Baldwin 	assert(shm);
1178e506e182SJohn Baldwin 	assert(fst);
1179e506e182SJohn Baldwin 	bzero(shm, sizeof(*shm));
1180e506e182SJohn Baldwin 	shmfdp = fst->fs_typedep;
1181e506e182SJohn Baldwin 	if (shmfdp == NULL)
1182e506e182SJohn Baldwin 		goto fail;
1183e506e182SJohn Baldwin 	if (!kvm_read_all(kd, (unsigned long)shmfdp, &shmfd,
1184e506e182SJohn Baldwin 	    sizeof(struct shmfd))) {
1185e506e182SJohn Baldwin 		warnx("can't read shmfd at %p", (void *)shmfdp);
1186e506e182SJohn Baldwin 		goto fail;
1187e506e182SJohn Baldwin 	}
1188e506e182SJohn Baldwin 	shm->mode = S_IFREG | shmfd.shm_mode;
1189e506e182SJohn Baldwin 	shm->size = shmfd.shm_size;
1190a9f5e425SJohn Baldwin 	if (fst->fs_path == NULL && shmfd.shm_path != NULL) {
1191a9f5e425SJohn Baldwin 		path = malloc(MAXPATHLEN);
1192a9f5e425SJohn Baldwin 		for (i = 0; i < MAXPATHLEN - 1; i++) {
1193a9f5e425SJohn Baldwin 			if (!kvm_read_all(kd, (unsigned long)shmfd.shm_path + i,
1194a9f5e425SJohn Baldwin 			    path + i, 1))
1195a9f5e425SJohn Baldwin 				break;
1196a9f5e425SJohn Baldwin 			if (path[i] == '\0')
1197a9f5e425SJohn Baldwin 				break;
1198a9f5e425SJohn Baldwin 		}
1199a9f5e425SJohn Baldwin 		path[i] = '\0';
1200a9f5e425SJohn Baldwin 		if (i == 0)
1201a9f5e425SJohn Baldwin 			free(path);
1202a9f5e425SJohn Baldwin 		else
1203a9f5e425SJohn Baldwin 			fst->fs_path = path;
1204a9f5e425SJohn Baldwin 	}
1205e506e182SJohn Baldwin 	return (0);
1206e506e182SJohn Baldwin 
1207e506e182SJohn Baldwin fail:
1208467112b4SMikolaj Golub 	if (errbuf != NULL)
1209e506e182SJohn Baldwin 		snprintf(errbuf, _POSIX2_LINE_MAX, "error");
1210e506e182SJohn Baldwin 	return (1);
1211e506e182SJohn Baldwin }
1212e506e182SJohn Baldwin 
1213e506e182SJohn Baldwin static int
1214e506e182SJohn Baldwin procstat_get_shm_info_sysctl(struct filestat *fst, struct shmstat *shm,
1215e506e182SJohn Baldwin     char *errbuf __unused)
1216e506e182SJohn Baldwin {
1217e506e182SJohn Baldwin 	struct kinfo_file *kif;
1218e506e182SJohn Baldwin 
1219e506e182SJohn Baldwin 	assert(shm);
1220e506e182SJohn Baldwin 	assert(fst);
1221e506e182SJohn Baldwin 	bzero(shm, sizeof(*shm));
1222e506e182SJohn Baldwin 	kif = fst->fs_typedep;
1223e506e182SJohn Baldwin 	if (kif == NULL)
1224e506e182SJohn Baldwin 		return (0);
1225e506e182SJohn Baldwin 	shm->size = kif->kf_un.kf_file.kf_file_size;
1226e506e182SJohn Baldwin 	shm->mode = kif->kf_un.kf_file.kf_file_mode;
1227e506e182SJohn Baldwin 	return (0);
1228e506e182SJohn Baldwin }
1229e506e182SJohn Baldwin 
1230e506e182SJohn Baldwin int
12310daf62d9SStanislav Sedov procstat_get_vnode_info(struct procstat *procstat, struct filestat *fst,
12320daf62d9SStanislav Sedov     struct vnstat *vn, char *errbuf)
12330daf62d9SStanislav Sedov {
12340daf62d9SStanislav Sedov 
12350daf62d9SStanislav Sedov 	assert(vn);
12360daf62d9SStanislav Sedov 	if (procstat->type == PROCSTAT_KVM) {
12370daf62d9SStanislav Sedov 		return (procstat_get_vnode_info_kvm(procstat->kd, fst, vn,
12380daf62d9SStanislav Sedov 		    errbuf));
12397153ad2bSMikolaj Golub 	} else if (procstat->type == PROCSTAT_SYSCTL ||
12407153ad2bSMikolaj Golub 		procstat->type == PROCSTAT_CORE) {
12410daf62d9SStanislav Sedov 		return (procstat_get_vnode_info_sysctl(fst, vn, errbuf));
12420daf62d9SStanislav Sedov 	} else {
124380905c35SJilles Tjoelker 		warnx("unknown access method: %d", procstat->type);
1244467112b4SMikolaj Golub 		if (errbuf != NULL)
12450daf62d9SStanislav Sedov 			snprintf(errbuf, _POSIX2_LINE_MAX, "error");
12460daf62d9SStanislav Sedov 		return (1);
12470daf62d9SStanislav Sedov 	}
12480daf62d9SStanislav Sedov }
12490daf62d9SStanislav Sedov 
12500daf62d9SStanislav Sedov static int
12510daf62d9SStanislav Sedov procstat_get_vnode_info_kvm(kvm_t *kd, struct filestat *fst,
12520daf62d9SStanislav Sedov     struct vnstat *vn, char *errbuf)
12530daf62d9SStanislav Sedov {
12540daf62d9SStanislav Sedov 	/* Filesystem specific handlers. */
12550daf62d9SStanislav Sedov 	#define FSTYPE(fst)     {#fst, fst##_filestat}
12560daf62d9SStanislav Sedov 	struct {
12570daf62d9SStanislav Sedov 		const char	*tag;
12580daf62d9SStanislav Sedov 		int		(*handler)(kvm_t *kd, struct vnode *vp,
12590daf62d9SStanislav Sedov 		    struct vnstat *vn);
12600daf62d9SStanislav Sedov 	} fstypes[] = {
12610daf62d9SStanislav Sedov 		FSTYPE(devfs),
12620daf62d9SStanislav Sedov 		FSTYPE(isofs),
12630daf62d9SStanislav Sedov 		FSTYPE(msdosfs),
12640daf62d9SStanislav Sedov 		FSTYPE(nfs),
1265237abf0cSDavide Italiano 		FSTYPE(smbfs),
12660daf62d9SStanislav Sedov 		FSTYPE(udf),
12670daf62d9SStanislav Sedov 		FSTYPE(ufs),
126859ea47d0SStanislav Sedov #ifdef LIBPROCSTAT_ZFS
12690daf62d9SStanislav Sedov 		FSTYPE(zfs),
12700daf62d9SStanislav Sedov #endif
12710daf62d9SStanislav Sedov 	};
12720daf62d9SStanislav Sedov #define	NTYPES	(sizeof(fstypes) / sizeof(*fstypes))
12730daf62d9SStanislav Sedov 	struct vnode vnode;
12740daf62d9SStanislav Sedov 	char tagstr[12];
12750daf62d9SStanislav Sedov 	void *vp;
1276a770b4d8SMarcelo Araujo 	int error;
12770daf62d9SStanislav Sedov 	unsigned int i;
12780daf62d9SStanislav Sedov 
12790daf62d9SStanislav Sedov 	assert(kd);
12800daf62d9SStanislav Sedov 	assert(vn);
12810daf62d9SStanislav Sedov 	assert(fst);
12820daf62d9SStanislav Sedov 	vp = fst->fs_typedep;
12830daf62d9SStanislav Sedov 	if (vp == NULL)
12840daf62d9SStanislav Sedov 		goto fail;
12850daf62d9SStanislav Sedov 	error = kvm_read_all(kd, (unsigned long)vp, &vnode, sizeof(vnode));
12860daf62d9SStanislav Sedov 	if (error == 0) {
12870daf62d9SStanislav Sedov 		warnx("can't read vnode at %p", (void *)vp);
12880daf62d9SStanislav Sedov 		goto fail;
12890daf62d9SStanislav Sedov 	}
12900daf62d9SStanislav Sedov 	bzero(vn, sizeof(*vn));
12910daf62d9SStanislav Sedov 	vn->vn_type = vntype2psfsttype(vnode.v_type);
12920daf62d9SStanislav Sedov 	if (vnode.v_type == VNON || vnode.v_type == VBAD)
12930daf62d9SStanislav Sedov 		return (0);
1294630cb9c5SMateusz Guzik 	error = kvm_read_all(kd, (unsigned long)vnode.v_lock.lock_object.lo_name,
1295630cb9c5SMateusz Guzik 	    tagstr, sizeof(tagstr));
12960daf62d9SStanislav Sedov 	if (error == 0) {
1297630cb9c5SMateusz Guzik 		warnx("can't read lo_name at %p", (void *)vp);
12980daf62d9SStanislav Sedov 		goto fail;
12990daf62d9SStanislav Sedov 	}
13000daf62d9SStanislav Sedov 	tagstr[sizeof(tagstr) - 1] = '\0';
13010daf62d9SStanislav Sedov 
13020daf62d9SStanislav Sedov 	/*
13030daf62d9SStanislav Sedov 	 * Find appropriate handler.
13040daf62d9SStanislav Sedov 	 */
1305a770b4d8SMarcelo Araujo 	for (i = 0; i < NTYPES; i++)
13060daf62d9SStanislav Sedov 		if (!strcmp(fstypes[i].tag, tagstr)) {
13070daf62d9SStanislav Sedov 			if (fstypes[i].handler(kd, &vnode, vn) != 0) {
13080daf62d9SStanislav Sedov 				goto fail;
13090daf62d9SStanislav Sedov 			}
13100daf62d9SStanislav Sedov 			break;
13110daf62d9SStanislav Sedov 		}
13120daf62d9SStanislav Sedov 	if (i == NTYPES) {
1313467112b4SMikolaj Golub 		if (errbuf != NULL)
13140daf62d9SStanislav Sedov 			snprintf(errbuf, _POSIX2_LINE_MAX, "?(%s)", tagstr);
13150daf62d9SStanislav Sedov 		return (1);
13160daf62d9SStanislav Sedov 	}
13170daf62d9SStanislav Sedov 	vn->vn_mntdir = getmnton(kd, vnode.v_mount);
13180daf62d9SStanislav Sedov 	if ((vnode.v_type == VBLK || vnode.v_type == VCHR) &&
13190daf62d9SStanislav Sedov 	    vnode.v_rdev != NULL){
13200daf62d9SStanislav Sedov 		vn->vn_dev = dev2udev(kd, vnode.v_rdev);
13210daf62d9SStanislav Sedov 		(void)kdevtoname(kd, vnode.v_rdev, vn->vn_devname);
13220daf62d9SStanislav Sedov 	} else {
13230daf62d9SStanislav Sedov 		vn->vn_dev = -1;
13240daf62d9SStanislav Sedov 	}
13250daf62d9SStanislav Sedov 	return (0);
13260daf62d9SStanislav Sedov 
13270daf62d9SStanislav Sedov fail:
1328467112b4SMikolaj Golub 	if (errbuf != NULL)
13290daf62d9SStanislav Sedov 		snprintf(errbuf, _POSIX2_LINE_MAX, "error");
13300daf62d9SStanislav Sedov 	return (1);
13310daf62d9SStanislav Sedov }
13320daf62d9SStanislav Sedov 
13330daf62d9SStanislav Sedov /*
13340daf62d9SStanislav Sedov  * kinfo vnode type to filestat translation.
13350daf62d9SStanislav Sedov  */
13360daf62d9SStanislav Sedov static int
13370daf62d9SStanislav Sedov kinfo_vtype2fst(int kfvtype)
13380daf62d9SStanislav Sedov {
13390daf62d9SStanislav Sedov 	static struct {
13400daf62d9SStanislav Sedov 		int	kf_vtype;
13410daf62d9SStanislav Sedov 		int	fst_vtype;
13420daf62d9SStanislav Sedov 	} kfvtypes2fst[] = {
13430daf62d9SStanislav Sedov 		{ KF_VTYPE_VBAD, PS_FST_VTYPE_VBAD },
13440daf62d9SStanislav Sedov 		{ KF_VTYPE_VBLK, PS_FST_VTYPE_VBLK },
13450daf62d9SStanislav Sedov 		{ KF_VTYPE_VCHR, PS_FST_VTYPE_VCHR },
13460daf62d9SStanislav Sedov 		{ KF_VTYPE_VDIR, PS_FST_VTYPE_VDIR },
13470daf62d9SStanislav Sedov 		{ KF_VTYPE_VFIFO, PS_FST_VTYPE_VFIFO },
13480daf62d9SStanislav Sedov 		{ KF_VTYPE_VLNK, PS_FST_VTYPE_VLNK },
13490daf62d9SStanislav Sedov 		{ KF_VTYPE_VNON, PS_FST_VTYPE_VNON },
13500daf62d9SStanislav Sedov 		{ KF_VTYPE_VREG, PS_FST_VTYPE_VREG },
13510daf62d9SStanislav Sedov 		{ KF_VTYPE_VSOCK, PS_FST_VTYPE_VSOCK }
13520daf62d9SStanislav Sedov 	};
13530daf62d9SStanislav Sedov #define	NKFVTYPES	(sizeof(kfvtypes2fst) / sizeof(*kfvtypes2fst))
13540daf62d9SStanislav Sedov 	unsigned int i;
13550daf62d9SStanislav Sedov 
13560daf62d9SStanislav Sedov 	for (i = 0; i < NKFVTYPES; i++)
13570daf62d9SStanislav Sedov 		if (kfvtypes2fst[i].kf_vtype == kfvtype)
13580daf62d9SStanislav Sedov 			break;
13590daf62d9SStanislav Sedov 	if (i == NKFVTYPES)
13600daf62d9SStanislav Sedov 		return (PS_FST_VTYPE_UNKNOWN);
13610daf62d9SStanislav Sedov 	return (kfvtypes2fst[i].fst_vtype);
13620daf62d9SStanislav Sedov }
13630daf62d9SStanislav Sedov 
13640daf62d9SStanislav Sedov static int
13650daf62d9SStanislav Sedov procstat_get_vnode_info_sysctl(struct filestat *fst, struct vnstat *vn,
13660daf62d9SStanislav Sedov     char *errbuf)
13670daf62d9SStanislav Sedov {
13680daf62d9SStanislav Sedov 	struct statfs stbuf;
13690daf62d9SStanislav Sedov 	struct kinfo_file *kif;
13700daf62d9SStanislav Sedov 	struct kinfo_vmentry *kve;
137169921123SKonstantin Belousov 	char *name, *path;
13720daf62d9SStanislav Sedov 	uint64_t fileid;
13730daf62d9SStanislav Sedov 	uint64_t size;
137469921123SKonstantin Belousov 	uint64_t fsid;
137569921123SKonstantin Belousov 	uint64_t rdev;
13760daf62d9SStanislav Sedov 	uint16_t mode;
13770daf62d9SStanislav Sedov 	int vntype;
13780daf62d9SStanislav Sedov 	int status;
13790daf62d9SStanislav Sedov 
13800daf62d9SStanislav Sedov 	assert(fst);
13810daf62d9SStanislav Sedov 	assert(vn);
13820daf62d9SStanislav Sedov 	bzero(vn, sizeof(*vn));
13830daf62d9SStanislav Sedov 	if (fst->fs_typedep == NULL)
13840daf62d9SStanislav Sedov 		return (1);
13850daf62d9SStanislav Sedov 	if (fst->fs_uflags & PS_FST_UFLAG_MMAP) {
13860daf62d9SStanislav Sedov 		kve = fst->fs_typedep;
13870daf62d9SStanislav Sedov 		fileid = kve->kve_vn_fileid;
13880daf62d9SStanislav Sedov 		fsid = kve->kve_vn_fsid;
13890daf62d9SStanislav Sedov 		mode = kve->kve_vn_mode;
13900daf62d9SStanislav Sedov 		path = kve->kve_path;
13910daf62d9SStanislav Sedov 		rdev = kve->kve_vn_rdev;
13920daf62d9SStanislav Sedov 		size = kve->kve_vn_size;
13930daf62d9SStanislav Sedov 		vntype = kinfo_vtype2fst(kve->kve_vn_type);
13940daf62d9SStanislav Sedov 		status = kve->kve_status;
13950daf62d9SStanislav Sedov 	} else {
13960daf62d9SStanislav Sedov 		kif = fst->fs_typedep;
13970daf62d9SStanislav Sedov 		fileid = kif->kf_un.kf_file.kf_file_fileid;
13980daf62d9SStanislav Sedov 		fsid = kif->kf_un.kf_file.kf_file_fsid;
13990daf62d9SStanislav Sedov 		mode = kif->kf_un.kf_file.kf_file_mode;
14000daf62d9SStanislav Sedov 		path = kif->kf_path;
14010daf62d9SStanislav Sedov 		rdev = kif->kf_un.kf_file.kf_file_rdev;
14020daf62d9SStanislav Sedov 		size = kif->kf_un.kf_file.kf_file_size;
14030daf62d9SStanislav Sedov 		vntype = kinfo_vtype2fst(kif->kf_vnode_type);
14040daf62d9SStanislav Sedov 		status = kif->kf_status;
14050daf62d9SStanislav Sedov 	}
14060daf62d9SStanislav Sedov 	vn->vn_type = vntype;
14070daf62d9SStanislav Sedov 	if (vntype == PS_FST_VTYPE_VNON || vntype == PS_FST_VTYPE_VBAD)
14080daf62d9SStanislav Sedov 		return (0);
14090daf62d9SStanislav Sedov 	if ((status & KF_ATTR_VALID) == 0) {
1410467112b4SMikolaj Golub 		if (errbuf != NULL) {
1411467112b4SMikolaj Golub 			snprintf(errbuf, _POSIX2_LINE_MAX,
1412467112b4SMikolaj Golub 			    "? (no info available)");
1413467112b4SMikolaj Golub 		}
14140daf62d9SStanislav Sedov 		return (1);
14150daf62d9SStanislav Sedov 	}
14160daf62d9SStanislav Sedov 	if (path && *path) {
14170daf62d9SStanislav Sedov 		statfs(path, &stbuf);
14180daf62d9SStanislav Sedov 		vn->vn_mntdir = strdup(stbuf.f_mntonname);
14190daf62d9SStanislav Sedov 	} else
14200daf62d9SStanislav Sedov 		vn->vn_mntdir = strdup("-");
14210daf62d9SStanislav Sedov 	vn->vn_dev = rdev;
14220daf62d9SStanislav Sedov 	if (vntype == PS_FST_VTYPE_VBLK) {
14230daf62d9SStanislav Sedov 		name = devname(rdev, S_IFBLK);
14240daf62d9SStanislav Sedov 		if (name != NULL)
14250daf62d9SStanislav Sedov 			strlcpy(vn->vn_devname, name,
14260daf62d9SStanislav Sedov 			    sizeof(vn->vn_devname));
14270daf62d9SStanislav Sedov 	} else if (vntype == PS_FST_VTYPE_VCHR) {
14280daf62d9SStanislav Sedov 		name = devname(vn->vn_dev, S_IFCHR);
14290daf62d9SStanislav Sedov 		if (name != NULL)
14300daf62d9SStanislav Sedov 			strlcpy(vn->vn_devname, name,
14310daf62d9SStanislav Sedov 			    sizeof(vn->vn_devname));
14320daf62d9SStanislav Sedov 	}
14330daf62d9SStanislav Sedov 	vn->vn_fsid = fsid;
14340daf62d9SStanislav Sedov 	vn->vn_fileid = fileid;
14350daf62d9SStanislav Sedov 	vn->vn_size = size;
14360daf62d9SStanislav Sedov 	vn->vn_mode = mode;
14370daf62d9SStanislav Sedov 	return (0);
14380daf62d9SStanislav Sedov }
14390daf62d9SStanislav Sedov 
14400daf62d9SStanislav Sedov int
14410daf62d9SStanislav Sedov procstat_get_socket_info(struct procstat *procstat, struct filestat *fst,
14420daf62d9SStanislav Sedov     struct sockstat *sock, char *errbuf)
14430daf62d9SStanislav Sedov {
14440daf62d9SStanislav Sedov 
14450daf62d9SStanislav Sedov 	assert(sock);
14460daf62d9SStanislav Sedov 	if (procstat->type == PROCSTAT_KVM) {
14470daf62d9SStanislav Sedov 		return (procstat_get_socket_info_kvm(procstat->kd, fst, sock,
14480daf62d9SStanislav Sedov 		    errbuf));
14497153ad2bSMikolaj Golub 	} else if (procstat->type == PROCSTAT_SYSCTL ||
14507153ad2bSMikolaj Golub 		procstat->type == PROCSTAT_CORE) {
14510daf62d9SStanislav Sedov 		return (procstat_get_socket_info_sysctl(fst, sock, errbuf));
14520daf62d9SStanislav Sedov 	} else {
145380905c35SJilles Tjoelker 		warnx("unknown access method: %d", procstat->type);
1454467112b4SMikolaj Golub 		if (errbuf != NULL)
14550daf62d9SStanislav Sedov 			snprintf(errbuf, _POSIX2_LINE_MAX, "error");
14560daf62d9SStanislav Sedov 		return (1);
14570daf62d9SStanislav Sedov 	}
14580daf62d9SStanislav Sedov }
14590daf62d9SStanislav Sedov 
14600daf62d9SStanislav Sedov static int
14610daf62d9SStanislav Sedov procstat_get_socket_info_kvm(kvm_t *kd, struct filestat *fst,
14620daf62d9SStanislav Sedov     struct sockstat *sock, char *errbuf)
14630daf62d9SStanislav Sedov {
14640daf62d9SStanislav Sedov 	struct domain dom;
14650daf62d9SStanislav Sedov 	struct inpcb inpcb;
14660daf62d9SStanislav Sedov 	struct protosw proto;
14670daf62d9SStanislav Sedov 	struct socket s;
14680daf62d9SStanislav Sedov 	struct unpcb unpcb;
14690daf62d9SStanislav Sedov 	ssize_t len;
14700daf62d9SStanislav Sedov 	void *so;
14710daf62d9SStanislav Sedov 
14720daf62d9SStanislav Sedov 	assert(kd);
14730daf62d9SStanislav Sedov 	assert(sock);
14740daf62d9SStanislav Sedov 	assert(fst);
14750daf62d9SStanislav Sedov 	bzero(sock, sizeof(*sock));
14760daf62d9SStanislav Sedov 	so = fst->fs_typedep;
14770daf62d9SStanislav Sedov 	if (so == NULL)
14780daf62d9SStanislav Sedov 		goto fail;
14790daf62d9SStanislav Sedov 	sock->so_addr = (uintptr_t)so;
14800daf62d9SStanislav Sedov 	/* fill in socket */
14810daf62d9SStanislav Sedov 	if (!kvm_read_all(kd, (unsigned long)so, &s,
14820daf62d9SStanislav Sedov 	    sizeof(struct socket))) {
14830daf62d9SStanislav Sedov 		warnx("can't read sock at %p", (void *)so);
14840daf62d9SStanislav Sedov 		goto fail;
14850daf62d9SStanislav Sedov 	}
14860daf62d9SStanislav Sedov 	/* fill in protosw entry */
14870daf62d9SStanislav Sedov 	if (!kvm_read_all(kd, (unsigned long)s.so_proto, &proto,
14880daf62d9SStanislav Sedov 	    sizeof(struct protosw))) {
14890daf62d9SStanislav Sedov 		warnx("can't read protosw at %p", (void *)s.so_proto);
14900daf62d9SStanislav Sedov 		goto fail;
14910daf62d9SStanislav Sedov 	}
14920daf62d9SStanislav Sedov 	/* fill in domain */
14930daf62d9SStanislav Sedov 	if (!kvm_read_all(kd, (unsigned long)proto.pr_domain, &dom,
14940daf62d9SStanislav Sedov 	    sizeof(struct domain))) {
14950daf62d9SStanislav Sedov 		warnx("can't read domain at %p",
14960daf62d9SStanislav Sedov 		    (void *)proto.pr_domain);
14970daf62d9SStanislav Sedov 		goto fail;
14980daf62d9SStanislav Sedov 	}
14990daf62d9SStanislav Sedov 	if ((len = kvm_read(kd, (unsigned long)dom.dom_name, sock->dname,
15000daf62d9SStanislav Sedov 	    sizeof(sock->dname) - 1)) < 0) {
15010daf62d9SStanislav Sedov 		warnx("can't read domain name at %p", (void *)dom.dom_name);
15020daf62d9SStanislav Sedov 		sock->dname[0] = '\0';
15030daf62d9SStanislav Sedov 	}
15040daf62d9SStanislav Sedov 	else
15050daf62d9SStanislav Sedov 		sock->dname[len] = '\0';
15060daf62d9SStanislav Sedov 
15070daf62d9SStanislav Sedov 	/*
15080daf62d9SStanislav Sedov 	 * Fill in known data.
15090daf62d9SStanislav Sedov 	 */
15100daf62d9SStanislav Sedov 	sock->type = s.so_type;
15110daf62d9SStanislav Sedov 	sock->proto = proto.pr_protocol;
15120daf62d9SStanislav Sedov 	sock->dom_family = dom.dom_family;
15130daf62d9SStanislav Sedov 	sock->so_pcb = (uintptr_t)s.so_pcb;
15140daf62d9SStanislav Sedov 
15150daf62d9SStanislav Sedov 	/*
15160daf62d9SStanislav Sedov 	 * Protocol specific data.
15170daf62d9SStanislav Sedov 	 */
15180daf62d9SStanislav Sedov 	switch(dom.dom_family) {
15190daf62d9SStanislav Sedov 	case AF_INET:
15200daf62d9SStanislav Sedov 	case AF_INET6:
15210daf62d9SStanislav Sedov 		if (proto.pr_protocol == IPPROTO_TCP) {
15220daf62d9SStanislav Sedov 			if (s.so_pcb) {
15230daf62d9SStanislav Sedov 				if (kvm_read(kd, (u_long)s.so_pcb,
15240daf62d9SStanislav Sedov 				    (char *)&inpcb, sizeof(struct inpcb))
15250daf62d9SStanislav Sedov 				    != sizeof(struct inpcb)) {
15260daf62d9SStanislav Sedov 					warnx("can't read inpcb at %p",
15270daf62d9SStanislav Sedov 					    (void *)s.so_pcb);
15280daf62d9SStanislav Sedov 				} else
15290daf62d9SStanislav Sedov 					sock->inp_ppcb =
15300daf62d9SStanislav Sedov 					    (uintptr_t)inpcb.inp_ppcb;
153195b97895SConrad Meyer 				sock->sendq = s.so_snd.sb_ccc;
153295b97895SConrad Meyer 				sock->recvq = s.so_rcv.sb_ccc;
15330daf62d9SStanislav Sedov 			}
15340daf62d9SStanislav Sedov 		}
15350daf62d9SStanislav Sedov 		break;
15360daf62d9SStanislav Sedov 	case AF_UNIX:
15370daf62d9SStanislav Sedov 		if (s.so_pcb) {
15380daf62d9SStanislav Sedov 			if (kvm_read(kd, (u_long)s.so_pcb, (char *)&unpcb,
15390daf62d9SStanislav Sedov 			    sizeof(struct unpcb)) != sizeof(struct unpcb)){
15400daf62d9SStanislav Sedov 				warnx("can't read unpcb at %p",
15410daf62d9SStanislav Sedov 				    (void *)s.so_pcb);
15420daf62d9SStanislav Sedov 			} else if (unpcb.unp_conn) {
15430daf62d9SStanislav Sedov 				sock->so_rcv_sb_state = s.so_rcv.sb_state;
15440daf62d9SStanislav Sedov 				sock->so_snd_sb_state = s.so_snd.sb_state;
15450daf62d9SStanislav Sedov 				sock->unp_conn = (uintptr_t)unpcb.unp_conn;
154695b97895SConrad Meyer 				sock->sendq = s.so_snd.sb_ccc;
154795b97895SConrad Meyer 				sock->recvq = s.so_rcv.sb_ccc;
15480daf62d9SStanislav Sedov 			}
15490daf62d9SStanislav Sedov 		}
15500daf62d9SStanislav Sedov 		break;
15510daf62d9SStanislav Sedov 	default:
15520daf62d9SStanislav Sedov 		break;
15530daf62d9SStanislav Sedov 	}
15540daf62d9SStanislav Sedov 	return (0);
15550daf62d9SStanislav Sedov 
15560daf62d9SStanislav Sedov fail:
1557467112b4SMikolaj Golub 	if (errbuf != NULL)
15580daf62d9SStanislav Sedov 		snprintf(errbuf, _POSIX2_LINE_MAX, "error");
15590daf62d9SStanislav Sedov 	return (1);
15600daf62d9SStanislav Sedov }
15610daf62d9SStanislav Sedov 
15620daf62d9SStanislav Sedov static int
15630daf62d9SStanislav Sedov procstat_get_socket_info_sysctl(struct filestat *fst, struct sockstat *sock,
15640daf62d9SStanislav Sedov     char *errbuf __unused)
15650daf62d9SStanislav Sedov {
15660daf62d9SStanislav Sedov 	struct kinfo_file *kif;
15670daf62d9SStanislav Sedov 
15680daf62d9SStanislav Sedov 	assert(sock);
15690daf62d9SStanislav Sedov 	assert(fst);
15700daf62d9SStanislav Sedov 	bzero(sock, sizeof(*sock));
15710daf62d9SStanislav Sedov 	kif = fst->fs_typedep;
15720daf62d9SStanislav Sedov 	if (kif == NULL)
15730daf62d9SStanislav Sedov 		return (0);
15740daf62d9SStanislav Sedov 
15750daf62d9SStanislav Sedov 	/*
15760daf62d9SStanislav Sedov 	 * Fill in known data.
15770daf62d9SStanislav Sedov 	 */
15780daf62d9SStanislav Sedov 	sock->type = kif->kf_sock_type;
15790daf62d9SStanislav Sedov 	sock->proto = kif->kf_sock_protocol;
15800daf62d9SStanislav Sedov 	sock->dom_family = kif->kf_sock_domain;
15810daf62d9SStanislav Sedov 	sock->so_pcb = kif->kf_un.kf_sock.kf_sock_pcb;
15820daf62d9SStanislav Sedov 	strlcpy(sock->dname, kif->kf_path, sizeof(sock->dname));
158369921123SKonstantin Belousov 	bcopy(&kif->kf_un.kf_sock.kf_sa_local, &sock->sa_local,
158469921123SKonstantin Belousov 	    kif->kf_un.kf_sock.kf_sa_local.ss_len);
158569921123SKonstantin Belousov 	bcopy(&kif->kf_un.kf_sock.kf_sa_peer, &sock->sa_peer,
158669921123SKonstantin Belousov 	    kif->kf_un.kf_sock.kf_sa_peer.ss_len);
15870daf62d9SStanislav Sedov 
15880daf62d9SStanislav Sedov 	/*
15890daf62d9SStanislav Sedov 	 * Protocol specific data.
15900daf62d9SStanislav Sedov 	 */
15910daf62d9SStanislav Sedov 	switch(sock->dom_family) {
15920daf62d9SStanislav Sedov 	case AF_INET:
15930daf62d9SStanislav Sedov 	case AF_INET6:
159495b97895SConrad Meyer 		if (sock->proto == IPPROTO_TCP) {
15950daf62d9SStanislav Sedov 			sock->inp_ppcb = kif->kf_un.kf_sock.kf_sock_inpcb;
159695b97895SConrad Meyer 			sock->sendq = kif->kf_un.kf_sock.kf_sock_sendq;
159795b97895SConrad Meyer 			sock->recvq = kif->kf_un.kf_sock.kf_sock_recvq;
159895b97895SConrad Meyer 		}
15990daf62d9SStanislav Sedov 		break;
16000daf62d9SStanislav Sedov 	case AF_UNIX:
16010daf62d9SStanislav Sedov 		if (kif->kf_un.kf_sock.kf_sock_unpconn != 0) {
16020daf62d9SStanislav Sedov 			sock->so_rcv_sb_state =
16030daf62d9SStanislav Sedov 			    kif->kf_un.kf_sock.kf_sock_rcv_sb_state;
16040daf62d9SStanislav Sedov 			sock->so_snd_sb_state =
16050daf62d9SStanislav Sedov 			    kif->kf_un.kf_sock.kf_sock_snd_sb_state;
16060daf62d9SStanislav Sedov 			sock->unp_conn =
16070daf62d9SStanislav Sedov 			    kif->kf_un.kf_sock.kf_sock_unpconn;
160895b97895SConrad Meyer 			sock->sendq = kif->kf_un.kf_sock.kf_sock_sendq;
160995b97895SConrad Meyer 			sock->recvq = kif->kf_un.kf_sock.kf_sock_recvq;
16100daf62d9SStanislav Sedov 		}
16110daf62d9SStanislav Sedov 		break;
16120daf62d9SStanislav Sedov 	default:
16130daf62d9SStanislav Sedov 		break;
16140daf62d9SStanislav Sedov 	}
16150daf62d9SStanislav Sedov 	return (0);
16160daf62d9SStanislav Sedov }
16170daf62d9SStanislav Sedov 
16180daf62d9SStanislav Sedov /*
16190daf62d9SStanislav Sedov  * Descriptor flags to filestat translation.
16200daf62d9SStanislav Sedov  */
16210daf62d9SStanislav Sedov static int
16220daf62d9SStanislav Sedov to_filestat_flags(int flags)
16230daf62d9SStanislav Sedov {
16240daf62d9SStanislav Sedov 	static struct {
16250daf62d9SStanislav Sedov 		int flag;
16260daf62d9SStanislav Sedov 		int fst_flag;
16270daf62d9SStanislav Sedov 	} fstflags[] = {
16280daf62d9SStanislav Sedov 		{ FREAD, PS_FST_FFLAG_READ },
16290daf62d9SStanislav Sedov 		{ FWRITE, PS_FST_FFLAG_WRITE },
16300daf62d9SStanislav Sedov 		{ O_APPEND, PS_FST_FFLAG_APPEND },
16310daf62d9SStanislav Sedov 		{ O_ASYNC, PS_FST_FFLAG_ASYNC },
16320daf62d9SStanislav Sedov 		{ O_CREAT, PS_FST_FFLAG_CREAT },
16330daf62d9SStanislav Sedov 		{ O_DIRECT, PS_FST_FFLAG_DIRECT },
16340daf62d9SStanislav Sedov 		{ O_EXCL, PS_FST_FFLAG_EXCL },
16350daf62d9SStanislav Sedov 		{ O_EXEC, PS_FST_FFLAG_EXEC },
16360daf62d9SStanislav Sedov 		{ O_EXLOCK, PS_FST_FFLAG_EXLOCK },
16370daf62d9SStanislav Sedov 		{ O_NOFOLLOW, PS_FST_FFLAG_NOFOLLOW },
16380daf62d9SStanislav Sedov 		{ O_NONBLOCK, PS_FST_FFLAG_NONBLOCK },
16390daf62d9SStanislav Sedov 		{ O_SHLOCK, PS_FST_FFLAG_SHLOCK },
16400daf62d9SStanislav Sedov 		{ O_SYNC, PS_FST_FFLAG_SYNC },
16410daf62d9SStanislav Sedov 		{ O_TRUNC, PS_FST_FFLAG_TRUNC }
16420daf62d9SStanislav Sedov 	};
16430daf62d9SStanislav Sedov #define NFSTFLAGS	(sizeof(fstflags) / sizeof(*fstflags))
16440daf62d9SStanislav Sedov 	int fst_flags;
16450daf62d9SStanislav Sedov 	unsigned int i;
16460daf62d9SStanislav Sedov 
16470daf62d9SStanislav Sedov 	fst_flags = 0;
16480daf62d9SStanislav Sedov 	for (i = 0; i < NFSTFLAGS; i++)
16490daf62d9SStanislav Sedov 		if (flags & fstflags[i].flag)
16500daf62d9SStanislav Sedov 			fst_flags |= fstflags[i].fst_flag;
16510daf62d9SStanislav Sedov 	return (fst_flags);
16520daf62d9SStanislav Sedov }
16530daf62d9SStanislav Sedov 
16540daf62d9SStanislav Sedov /*
16550daf62d9SStanislav Sedov  * Vnode type to filestate translation.
16560daf62d9SStanislav Sedov  */
16570daf62d9SStanislav Sedov static int
16580daf62d9SStanislav Sedov vntype2psfsttype(int type)
16590daf62d9SStanislav Sedov {
16600daf62d9SStanislav Sedov 	static struct {
16610daf62d9SStanislav Sedov 		int	vtype;
16620daf62d9SStanislav Sedov 		int	fst_vtype;
16630daf62d9SStanislav Sedov 	} vt2fst[] = {
16640daf62d9SStanislav Sedov 		{ VBAD, PS_FST_VTYPE_VBAD },
16650daf62d9SStanislav Sedov 		{ VBLK, PS_FST_VTYPE_VBLK },
16660daf62d9SStanislav Sedov 		{ VCHR, PS_FST_VTYPE_VCHR },
16670daf62d9SStanislav Sedov 		{ VDIR, PS_FST_VTYPE_VDIR },
16680daf62d9SStanislav Sedov 		{ VFIFO, PS_FST_VTYPE_VFIFO },
16690daf62d9SStanislav Sedov 		{ VLNK, PS_FST_VTYPE_VLNK },
16700daf62d9SStanislav Sedov 		{ VNON, PS_FST_VTYPE_VNON },
16710daf62d9SStanislav Sedov 		{ VREG, PS_FST_VTYPE_VREG },
16720daf62d9SStanislav Sedov 		{ VSOCK, PS_FST_VTYPE_VSOCK }
16730daf62d9SStanislav Sedov 	};
16740daf62d9SStanislav Sedov #define	NVFTYPES	(sizeof(vt2fst) / sizeof(*vt2fst))
16750daf62d9SStanislav Sedov 	unsigned int i, fst_type;
16760daf62d9SStanislav Sedov 
16770daf62d9SStanislav Sedov 	fst_type = PS_FST_VTYPE_UNKNOWN;
16780daf62d9SStanislav Sedov 	for (i = 0; i < NVFTYPES; i++) {
16790daf62d9SStanislav Sedov 		if (type == vt2fst[i].vtype) {
16800daf62d9SStanislav Sedov 			fst_type = vt2fst[i].fst_vtype;
16810daf62d9SStanislav Sedov 			break;
16820daf62d9SStanislav Sedov 		}
16830daf62d9SStanislav Sedov 	}
16840daf62d9SStanislav Sedov 	return (fst_type);
16850daf62d9SStanislav Sedov }
16860daf62d9SStanislav Sedov 
16870daf62d9SStanislav Sedov static char *
16880daf62d9SStanislav Sedov getmnton(kvm_t *kd, struct mount *m)
16890daf62d9SStanislav Sedov {
16905f301949SBen Laurie 	struct mount mnt;
16910daf62d9SStanislav Sedov 	static struct mtab {
16920daf62d9SStanislav Sedov 		struct mtab *next;
16930daf62d9SStanislav Sedov 		struct mount *m;
16940daf62d9SStanislav Sedov 		char mntonname[MNAMELEN + 1];
16950daf62d9SStanislav Sedov 	} *mhead = NULL;
16960daf62d9SStanislav Sedov 	struct mtab *mt;
16970daf62d9SStanislav Sedov 
16980daf62d9SStanislav Sedov 	for (mt = mhead; mt != NULL; mt = mt->next)
16990daf62d9SStanislav Sedov 		if (m == mt->m)
17000daf62d9SStanislav Sedov 			return (mt->mntonname);
17010daf62d9SStanislav Sedov 	if (!kvm_read_all(kd, (unsigned long)m, &mnt, sizeof(struct mount))) {
17020daf62d9SStanislav Sedov 		warnx("can't read mount table at %p", (void *)m);
17030daf62d9SStanislav Sedov 		return (NULL);
17040daf62d9SStanislav Sedov 	}
17050daf62d9SStanislav Sedov 	if ((mt = malloc(sizeof (struct mtab))) == NULL)
17060daf62d9SStanislav Sedov 		err(1, NULL);
17070daf62d9SStanislav Sedov 	mt->m = m;
17080daf62d9SStanislav Sedov 	bcopy(&mnt.mnt_stat.f_mntonname[0], &mt->mntonname[0], MNAMELEN);
17095f301949SBen Laurie 	mt->mntonname[MNAMELEN] = '\0';
17100daf62d9SStanislav Sedov 	mt->next = mhead;
17110daf62d9SStanislav Sedov 	mhead = mt;
17120daf62d9SStanislav Sedov 	return (mt->mntonname);
17130daf62d9SStanislav Sedov }
17147153ad2bSMikolaj Golub 
17154482b5e3SMikolaj Golub /*
17164482b5e3SMikolaj Golub  * Auxiliary structures and functions to get process environment or
17174482b5e3SMikolaj Golub  * command line arguments.
17184482b5e3SMikolaj Golub  */
17194482b5e3SMikolaj Golub struct argvec {
17204482b5e3SMikolaj Golub 	char	*buf;
17214482b5e3SMikolaj Golub 	size_t	bufsize;
17224482b5e3SMikolaj Golub 	char	**argv;
17234482b5e3SMikolaj Golub 	size_t	argc;
17244482b5e3SMikolaj Golub };
17254482b5e3SMikolaj Golub 
17264482b5e3SMikolaj Golub static struct argvec *
17274482b5e3SMikolaj Golub argvec_alloc(size_t bufsize)
17284482b5e3SMikolaj Golub {
17294482b5e3SMikolaj Golub 	struct argvec *av;
17304482b5e3SMikolaj Golub 
17314482b5e3SMikolaj Golub 	av = malloc(sizeof(*av));
17324482b5e3SMikolaj Golub 	if (av == NULL)
17334482b5e3SMikolaj Golub 		return (NULL);
17344482b5e3SMikolaj Golub 	av->bufsize = bufsize;
17354482b5e3SMikolaj Golub 	av->buf = malloc(av->bufsize);
17364482b5e3SMikolaj Golub 	if (av->buf == NULL) {
17374482b5e3SMikolaj Golub 		free(av);
17384482b5e3SMikolaj Golub 		return (NULL);
17394482b5e3SMikolaj Golub 	}
17404482b5e3SMikolaj Golub 	av->argc = 32;
17414482b5e3SMikolaj Golub 	av->argv = malloc(sizeof(char *) * av->argc);
17424482b5e3SMikolaj Golub 	if (av->argv == NULL) {
17434482b5e3SMikolaj Golub 		free(av->buf);
17444482b5e3SMikolaj Golub 		free(av);
17454482b5e3SMikolaj Golub 		return (NULL);
17464482b5e3SMikolaj Golub 	}
17474482b5e3SMikolaj Golub 	return av;
17484482b5e3SMikolaj Golub }
17494482b5e3SMikolaj Golub 
17504482b5e3SMikolaj Golub static void
17514482b5e3SMikolaj Golub argvec_free(struct argvec * av)
17524482b5e3SMikolaj Golub {
17534482b5e3SMikolaj Golub 
17544482b5e3SMikolaj Golub 	free(av->argv);
17554482b5e3SMikolaj Golub 	free(av->buf);
17564482b5e3SMikolaj Golub 	free(av);
17574482b5e3SMikolaj Golub }
17584482b5e3SMikolaj Golub 
17594482b5e3SMikolaj Golub static char **
17604482b5e3SMikolaj Golub getargv(struct procstat *procstat, struct kinfo_proc *kp, size_t nchr, int env)
17614482b5e3SMikolaj Golub {
17624482b5e3SMikolaj Golub 	int error, name[4], argc, i;
17634482b5e3SMikolaj Golub 	struct argvec *av, **avp;
17644482b5e3SMikolaj Golub 	enum psc_type type;
17654482b5e3SMikolaj Golub 	size_t len;
17664482b5e3SMikolaj Golub 	char *p, **argv;
17674482b5e3SMikolaj Golub 
17684482b5e3SMikolaj Golub 	assert(procstat);
17694482b5e3SMikolaj Golub 	assert(kp);
17704482b5e3SMikolaj Golub 	if (procstat->type == PROCSTAT_KVM) {
17714482b5e3SMikolaj Golub 		warnx("can't use kvm access method");
17724482b5e3SMikolaj Golub 		return (NULL);
17734482b5e3SMikolaj Golub 	}
17744482b5e3SMikolaj Golub 	if (procstat->type != PROCSTAT_SYSCTL &&
17754482b5e3SMikolaj Golub 	    procstat->type != PROCSTAT_CORE) {
17764482b5e3SMikolaj Golub 		warnx("unknown access method: %d", procstat->type);
17774482b5e3SMikolaj Golub 		return (NULL);
17784482b5e3SMikolaj Golub 	}
17794482b5e3SMikolaj Golub 
17804482b5e3SMikolaj Golub 	if (nchr == 0 || nchr > ARG_MAX)
17814482b5e3SMikolaj Golub 		nchr = ARG_MAX;
17824482b5e3SMikolaj Golub 
17834482b5e3SMikolaj Golub 	avp = (struct argvec **)(env ? &procstat->argv : &procstat->envv);
17844482b5e3SMikolaj Golub 	av = *avp;
17854482b5e3SMikolaj Golub 
17864482b5e3SMikolaj Golub 	if (av == NULL)
17874482b5e3SMikolaj Golub 	{
17884482b5e3SMikolaj Golub 		av = argvec_alloc(nchr);
17894482b5e3SMikolaj Golub 		if (av == NULL)
17904482b5e3SMikolaj Golub 		{
17914482b5e3SMikolaj Golub 			warn("malloc(%zu)", nchr);
17924482b5e3SMikolaj Golub 			return (NULL);
17934482b5e3SMikolaj Golub 		}
17944482b5e3SMikolaj Golub 		*avp = av;
17954482b5e3SMikolaj Golub 	} else if (av->bufsize < nchr) {
17964482b5e3SMikolaj Golub 		av->buf = reallocf(av->buf, nchr);
17974482b5e3SMikolaj Golub 		if (av->buf == NULL) {
17984482b5e3SMikolaj Golub 			warn("malloc(%zu)", nchr);
17994482b5e3SMikolaj Golub 			return (NULL);
18004482b5e3SMikolaj Golub 		}
18014482b5e3SMikolaj Golub 	}
18024482b5e3SMikolaj Golub 	if (procstat->type == PROCSTAT_SYSCTL) {
18034482b5e3SMikolaj Golub 		name[0] = CTL_KERN;
18044482b5e3SMikolaj Golub 		name[1] = KERN_PROC;
18054482b5e3SMikolaj Golub 		name[2] = env ? KERN_PROC_ENV : KERN_PROC_ARGS;
18064482b5e3SMikolaj Golub 		name[3] = kp->ki_pid;
18074482b5e3SMikolaj Golub 		len = nchr;
180875b6a179SEnji Cooper 		error = sysctl(name, nitems(name), av->buf, &len, NULL, 0);
18094482b5e3SMikolaj Golub 		if (error != 0 && errno != ESRCH && errno != EPERM)
18104482b5e3SMikolaj Golub 			warn("sysctl(kern.proc.%s)", env ? "env" : "args");
18114482b5e3SMikolaj Golub 		if (error != 0 || len == 0)
18124482b5e3SMikolaj Golub 			return (NULL);
18134482b5e3SMikolaj Golub 	} else /* procstat->type == PROCSTAT_CORE */ {
18144482b5e3SMikolaj Golub 		type = env ? PSC_TYPE_ENVV : PSC_TYPE_ARGV;
18154482b5e3SMikolaj Golub 		len = nchr;
18164482b5e3SMikolaj Golub 		if (procstat_core_get(procstat->core, type, av->buf, &len)
18174482b5e3SMikolaj Golub 		    == NULL) {
18184482b5e3SMikolaj Golub 			return (NULL);
18194482b5e3SMikolaj Golub 		}
18204482b5e3SMikolaj Golub 	}
18214482b5e3SMikolaj Golub 
18224482b5e3SMikolaj Golub 	argv = av->argv;
18234482b5e3SMikolaj Golub 	argc = av->argc;
18244482b5e3SMikolaj Golub 	i = 0;
18254482b5e3SMikolaj Golub 	for (p = av->buf; p < av->buf + len; p += strlen(p) + 1) {
18264482b5e3SMikolaj Golub 		argv[i++] = p;
18274482b5e3SMikolaj Golub 		if (i < argc)
18284482b5e3SMikolaj Golub 			continue;
18294482b5e3SMikolaj Golub 		/* Grow argv. */
18304482b5e3SMikolaj Golub 		argc += argc;
18314482b5e3SMikolaj Golub 		argv = realloc(argv, sizeof(char *) * argc);
18324482b5e3SMikolaj Golub 		if (argv == NULL) {
18334482b5e3SMikolaj Golub 			warn("malloc(%zu)", sizeof(char *) * argc);
18344482b5e3SMikolaj Golub 			return (NULL);
18354482b5e3SMikolaj Golub 		}
18364482b5e3SMikolaj Golub 		av->argv = argv;
18374482b5e3SMikolaj Golub 		av->argc = argc;
18384482b5e3SMikolaj Golub 	}
18394482b5e3SMikolaj Golub 	argv[i] = NULL;
18404482b5e3SMikolaj Golub 
18414482b5e3SMikolaj Golub 	return (argv);
18424482b5e3SMikolaj Golub }
18434482b5e3SMikolaj Golub 
18444482b5e3SMikolaj Golub /*
18454482b5e3SMikolaj Golub  * Return process command line arguments.
18464482b5e3SMikolaj Golub  */
18474482b5e3SMikolaj Golub char **
18484482b5e3SMikolaj Golub procstat_getargv(struct procstat *procstat, struct kinfo_proc *p, size_t nchr)
18494482b5e3SMikolaj Golub {
18504482b5e3SMikolaj Golub 
18514482b5e3SMikolaj Golub 	return (getargv(procstat, p, nchr, 0));
18524482b5e3SMikolaj Golub }
18534482b5e3SMikolaj Golub 
18544482b5e3SMikolaj Golub /*
18554482b5e3SMikolaj Golub  * Free the buffer allocated by procstat_getargv().
18564482b5e3SMikolaj Golub  */
18574482b5e3SMikolaj Golub void
18584482b5e3SMikolaj Golub procstat_freeargv(struct procstat *procstat)
18594482b5e3SMikolaj Golub {
18604482b5e3SMikolaj Golub 
18614482b5e3SMikolaj Golub 	if (procstat->argv != NULL) {
18624482b5e3SMikolaj Golub 		argvec_free(procstat->argv);
18634482b5e3SMikolaj Golub 		procstat->argv = NULL;
18644482b5e3SMikolaj Golub 	}
18654482b5e3SMikolaj Golub }
18664482b5e3SMikolaj Golub 
18674482b5e3SMikolaj Golub /*
18684482b5e3SMikolaj Golub  * Return process environment.
18694482b5e3SMikolaj Golub  */
18704482b5e3SMikolaj Golub char **
18714482b5e3SMikolaj Golub procstat_getenvv(struct procstat *procstat, struct kinfo_proc *p, size_t nchr)
18724482b5e3SMikolaj Golub {
18734482b5e3SMikolaj Golub 
18744482b5e3SMikolaj Golub 	return (getargv(procstat, p, nchr, 1));
18754482b5e3SMikolaj Golub }
18764482b5e3SMikolaj Golub 
18774482b5e3SMikolaj Golub /*
18784482b5e3SMikolaj Golub  * Free the buffer allocated by procstat_getenvv().
18794482b5e3SMikolaj Golub  */
18804482b5e3SMikolaj Golub void
18814482b5e3SMikolaj Golub procstat_freeenvv(struct procstat *procstat)
18824482b5e3SMikolaj Golub {
18834482b5e3SMikolaj Golub 	if (procstat->envv != NULL) {
18844482b5e3SMikolaj Golub 		argvec_free(procstat->envv);
18854482b5e3SMikolaj Golub 		procstat->envv = NULL;
18864482b5e3SMikolaj Golub 	}
18874482b5e3SMikolaj Golub }
18884482b5e3SMikolaj Golub 
188939680c7bSMikolaj Golub static struct kinfo_vmentry *
189039680c7bSMikolaj Golub kinfo_getvmmap_core(struct procstat_core *core, int *cntp)
189139680c7bSMikolaj Golub {
189239680c7bSMikolaj Golub 	int cnt;
189339680c7bSMikolaj Golub 	size_t len;
189439680c7bSMikolaj Golub 	char *buf, *bp, *eb;
189539680c7bSMikolaj Golub 	struct kinfo_vmentry *kiv, *kp, *kv;
189639680c7bSMikolaj Golub 
189739680c7bSMikolaj Golub 	buf = procstat_core_get(core, PSC_TYPE_VMMAP, NULL, &len);
189839680c7bSMikolaj Golub 	if (buf == NULL)
189939680c7bSMikolaj Golub 		return (NULL);
190039680c7bSMikolaj Golub 
190139680c7bSMikolaj Golub 	/*
190239680c7bSMikolaj Golub 	 * XXXMG: The code below is just copy&past from libutil.
190339680c7bSMikolaj Golub 	 * The code duplication can be avoided if libutil
190439680c7bSMikolaj Golub 	 * is extended to provide something like:
190539680c7bSMikolaj Golub 	 *   struct kinfo_vmentry *kinfo_getvmmap_from_buf(const char *buf,
190639680c7bSMikolaj Golub 	 *       size_t len, int *cntp);
190739680c7bSMikolaj Golub 	 */
190839680c7bSMikolaj Golub 
190939680c7bSMikolaj Golub 	/* Pass 1: count items */
191039680c7bSMikolaj Golub 	cnt = 0;
191139680c7bSMikolaj Golub 	bp = buf;
191239680c7bSMikolaj Golub 	eb = buf + len;
191339680c7bSMikolaj Golub 	while (bp < eb) {
191439680c7bSMikolaj Golub 		kv = (struct kinfo_vmentry *)(uintptr_t)bp;
1915e6b95927SConrad Meyer 		if (kv->kve_structsize == 0)
1916e6b95927SConrad Meyer 			break;
191739680c7bSMikolaj Golub 		bp += kv->kve_structsize;
191839680c7bSMikolaj Golub 		cnt++;
191939680c7bSMikolaj Golub 	}
192039680c7bSMikolaj Golub 
192139680c7bSMikolaj Golub 	kiv = calloc(cnt, sizeof(*kiv));
192239680c7bSMikolaj Golub 	if (kiv == NULL) {
192339680c7bSMikolaj Golub 		free(buf);
192439680c7bSMikolaj Golub 		return (NULL);
192539680c7bSMikolaj Golub 	}
192639680c7bSMikolaj Golub 	bp = buf;
192739680c7bSMikolaj Golub 	eb = buf + len;
192839680c7bSMikolaj Golub 	kp = kiv;
192939680c7bSMikolaj Golub 	/* Pass 2: unpack */
193039680c7bSMikolaj Golub 	while (bp < eb) {
193139680c7bSMikolaj Golub 		kv = (struct kinfo_vmentry *)(uintptr_t)bp;
1932e6b95927SConrad Meyer 		if (kv->kve_structsize == 0)
1933e6b95927SConrad Meyer 			break;
193439680c7bSMikolaj Golub 		/* Copy/expand into pre-zeroed buffer */
193539680c7bSMikolaj Golub 		memcpy(kp, kv, kv->kve_structsize);
193639680c7bSMikolaj Golub 		/* Advance to next packed record */
193739680c7bSMikolaj Golub 		bp += kv->kve_structsize;
193839680c7bSMikolaj Golub 		/* Set field size to fixed length, advance */
193939680c7bSMikolaj Golub 		kp->kve_structsize = sizeof(*kp);
194039680c7bSMikolaj Golub 		kp++;
194139680c7bSMikolaj Golub 	}
194239680c7bSMikolaj Golub 	free(buf);
194339680c7bSMikolaj Golub 	*cntp = cnt;
194439680c7bSMikolaj Golub 	return (kiv);	/* Caller must free() return value */
194539680c7bSMikolaj Golub }
194639680c7bSMikolaj Golub 
194739680c7bSMikolaj Golub struct kinfo_vmentry *
194839680c7bSMikolaj Golub procstat_getvmmap(struct procstat *procstat, struct kinfo_proc *kp,
194939680c7bSMikolaj Golub     unsigned int *cntp)
195039680c7bSMikolaj Golub {
195189358231SMikolaj Golub 
195239680c7bSMikolaj Golub 	switch(procstat->type) {
195339680c7bSMikolaj Golub 	case PROCSTAT_KVM:
195439680c7bSMikolaj Golub 		warnx("kvm method is not supported");
195539680c7bSMikolaj Golub 		return (NULL);
195639680c7bSMikolaj Golub 	case PROCSTAT_SYSCTL:
195739680c7bSMikolaj Golub 		return (kinfo_getvmmap(kp->ki_pid, cntp));
195839680c7bSMikolaj Golub 	case PROCSTAT_CORE:
195939680c7bSMikolaj Golub 		return (kinfo_getvmmap_core(procstat->core, cntp));
196039680c7bSMikolaj Golub 	default:
196139680c7bSMikolaj Golub 		warnx("unknown access method: %d", procstat->type);
196239680c7bSMikolaj Golub 		return (NULL);
196339680c7bSMikolaj Golub 	}
196439680c7bSMikolaj Golub }
196539680c7bSMikolaj Golub 
196639680c7bSMikolaj Golub void
196739680c7bSMikolaj Golub procstat_freevmmap(struct procstat *procstat __unused,
196839680c7bSMikolaj Golub     struct kinfo_vmentry *vmmap)
196939680c7bSMikolaj Golub {
197039680c7bSMikolaj Golub 
197139680c7bSMikolaj Golub 	free(vmmap);
197239680c7bSMikolaj Golub }
19737f1d14e6SMikolaj Golub 
19747f1d14e6SMikolaj Golub static gid_t *
19751f84c47eSMikolaj Golub procstat_getgroups_kvm(kvm_t *kd, struct kinfo_proc *kp, unsigned int *cntp)
19761f84c47eSMikolaj Golub {
19771f84c47eSMikolaj Golub 	struct proc proc;
19781f84c47eSMikolaj Golub 	struct ucred ucred;
19791f84c47eSMikolaj Golub 	gid_t *groups;
19801f84c47eSMikolaj Golub 	size_t len;
19811f84c47eSMikolaj Golub 
19821f84c47eSMikolaj Golub 	assert(kd != NULL);
19831f84c47eSMikolaj Golub 	assert(kp != NULL);
19841f84c47eSMikolaj Golub 	if (!kvm_read_all(kd, (unsigned long)kp->ki_paddr, &proc,
19851f84c47eSMikolaj Golub 	    sizeof(proc))) {
19861f84c47eSMikolaj Golub 		warnx("can't read proc struct at %p for pid %d",
19871f84c47eSMikolaj Golub 		    kp->ki_paddr, kp->ki_pid);
19881f84c47eSMikolaj Golub 		return (NULL);
19891f84c47eSMikolaj Golub 	}
19901f84c47eSMikolaj Golub 	if (proc.p_ucred == NOCRED)
19911f84c47eSMikolaj Golub 		return (NULL);
19921f84c47eSMikolaj Golub 	if (!kvm_read_all(kd, (unsigned long)proc.p_ucred, &ucred,
19931f84c47eSMikolaj Golub 	    sizeof(ucred))) {
19941f84c47eSMikolaj Golub 		warnx("can't read ucred struct at %p for pid %d",
19951f84c47eSMikolaj Golub 		    proc.p_ucred, kp->ki_pid);
19961f84c47eSMikolaj Golub 		return (NULL);
19971f84c47eSMikolaj Golub 	}
19981f84c47eSMikolaj Golub 	len = ucred.cr_ngroups * sizeof(gid_t);
19991f84c47eSMikolaj Golub 	groups = malloc(len);
20001f84c47eSMikolaj Golub 	if (groups == NULL) {
20011f84c47eSMikolaj Golub 		warn("malloc(%zu)", len);
20021f84c47eSMikolaj Golub 		return (NULL);
20031f84c47eSMikolaj Golub 	}
20041f84c47eSMikolaj Golub 	if (!kvm_read_all(kd, (unsigned long)ucred.cr_groups, groups, len)) {
20051f84c47eSMikolaj Golub 		warnx("can't read groups at %p for pid %d",
20061f84c47eSMikolaj Golub 		    ucred.cr_groups, kp->ki_pid);
20071f84c47eSMikolaj Golub 		free(groups);
20081f84c47eSMikolaj Golub 		return (NULL);
20091f84c47eSMikolaj Golub 	}
20101f84c47eSMikolaj Golub 	*cntp = ucred.cr_ngroups;
20111f84c47eSMikolaj Golub 	return (groups);
20121f84c47eSMikolaj Golub }
20131f84c47eSMikolaj Golub 
20141f84c47eSMikolaj Golub static gid_t *
20157f1d14e6SMikolaj Golub procstat_getgroups_sysctl(pid_t pid, unsigned int *cntp)
20167f1d14e6SMikolaj Golub {
20177f1d14e6SMikolaj Golub 	int mib[4];
20187f1d14e6SMikolaj Golub 	size_t len;
20197f1d14e6SMikolaj Golub 	gid_t *groups;
20207f1d14e6SMikolaj Golub 
20217f1d14e6SMikolaj Golub 	mib[0] = CTL_KERN;
20227f1d14e6SMikolaj Golub 	mib[1] = KERN_PROC;
20237f1d14e6SMikolaj Golub 	mib[2] = KERN_PROC_GROUPS;
20247f1d14e6SMikolaj Golub 	mib[3] = pid;
20257f1d14e6SMikolaj Golub 	len = (sysconf(_SC_NGROUPS_MAX) + 1) * sizeof(gid_t);
20267f1d14e6SMikolaj Golub 	groups = malloc(len);
20277f1d14e6SMikolaj Golub 	if (groups == NULL) {
20287f1d14e6SMikolaj Golub 		warn("malloc(%zu)", len);
20297f1d14e6SMikolaj Golub 		return (NULL);
20307f1d14e6SMikolaj Golub 	}
203175b6a179SEnji Cooper 	if (sysctl(mib, nitems(mib), groups, &len, NULL, 0) == -1) {
20327f1d14e6SMikolaj Golub 		warn("sysctl: kern.proc.groups: %d", pid);
20337f1d14e6SMikolaj Golub 		free(groups);
20347f1d14e6SMikolaj Golub 		return (NULL);
20357f1d14e6SMikolaj Golub 	}
20367f1d14e6SMikolaj Golub 	*cntp = len / sizeof(gid_t);
20377f1d14e6SMikolaj Golub 	return (groups);
20387f1d14e6SMikolaj Golub }
20397f1d14e6SMikolaj Golub 
20407f1d14e6SMikolaj Golub static gid_t *
20417f1d14e6SMikolaj Golub procstat_getgroups_core(struct procstat_core *core, unsigned int *cntp)
20427f1d14e6SMikolaj Golub {
20437f1d14e6SMikolaj Golub 	size_t len;
20447f1d14e6SMikolaj Golub 	gid_t *groups;
20457f1d14e6SMikolaj Golub 
20467f1d14e6SMikolaj Golub 	groups = procstat_core_get(core, PSC_TYPE_GROUPS, NULL, &len);
20477f1d14e6SMikolaj Golub 	if (groups == NULL)
20487f1d14e6SMikolaj Golub 		return (NULL);
20497f1d14e6SMikolaj Golub 	*cntp = len / sizeof(gid_t);
20507f1d14e6SMikolaj Golub 	return (groups);
20517f1d14e6SMikolaj Golub }
20527f1d14e6SMikolaj Golub 
20537f1d14e6SMikolaj Golub gid_t *
20547f1d14e6SMikolaj Golub procstat_getgroups(struct procstat *procstat, struct kinfo_proc *kp,
20557f1d14e6SMikolaj Golub     unsigned int *cntp)
20567f1d14e6SMikolaj Golub {
20577f1d14e6SMikolaj Golub 	switch(procstat->type) {
20587f1d14e6SMikolaj Golub 	case PROCSTAT_KVM:
20591f84c47eSMikolaj Golub 		return (procstat_getgroups_kvm(procstat->kd, kp, cntp));
20607f1d14e6SMikolaj Golub 	case PROCSTAT_SYSCTL:
20617f1d14e6SMikolaj Golub 		return (procstat_getgroups_sysctl(kp->ki_pid, cntp));
20627f1d14e6SMikolaj Golub 	case PROCSTAT_CORE:
20637f1d14e6SMikolaj Golub 		return (procstat_getgroups_core(procstat->core, cntp));
20647f1d14e6SMikolaj Golub 	default:
20657f1d14e6SMikolaj Golub 		warnx("unknown access method: %d", procstat->type);
20667f1d14e6SMikolaj Golub 		return (NULL);
20677f1d14e6SMikolaj Golub 	}
20687f1d14e6SMikolaj Golub }
20697f1d14e6SMikolaj Golub 
20707f1d14e6SMikolaj Golub void
20717f1d14e6SMikolaj Golub procstat_freegroups(struct procstat *procstat __unused, gid_t *groups)
20727f1d14e6SMikolaj Golub {
20737f1d14e6SMikolaj Golub 
20747f1d14e6SMikolaj Golub 	free(groups);
20757f1d14e6SMikolaj Golub }
20765b9bcba9SMikolaj Golub 
20775b9bcba9SMikolaj Golub static int
20781f84c47eSMikolaj Golub procstat_getumask_kvm(kvm_t *kd, struct kinfo_proc *kp, unsigned short *maskp)
20791f84c47eSMikolaj Golub {
20801f84c47eSMikolaj Golub 	struct filedesc fd;
20811f84c47eSMikolaj Golub 
20821f84c47eSMikolaj Golub 	assert(kd != NULL);
20831f84c47eSMikolaj Golub 	assert(kp != NULL);
20841f84c47eSMikolaj Golub 	if (kp->ki_fd == NULL)
20851f84c47eSMikolaj Golub 		return (-1);
20861f84c47eSMikolaj Golub 	if (!kvm_read_all(kd, (unsigned long)kp->ki_fd, &fd, sizeof(fd))) {
20871f84c47eSMikolaj Golub 		warnx("can't read filedesc at %p for pid %d", kp->ki_fd,
20881f84c47eSMikolaj Golub 		    kp->ki_pid);
20891f84c47eSMikolaj Golub 		return (-1);
20901f84c47eSMikolaj Golub 	}
20911f84c47eSMikolaj Golub 	*maskp = fd.fd_cmask;
20921f84c47eSMikolaj Golub 	return (0);
20931f84c47eSMikolaj Golub }
20941f84c47eSMikolaj Golub 
20951f84c47eSMikolaj Golub static int
20965b9bcba9SMikolaj Golub procstat_getumask_sysctl(pid_t pid, unsigned short *maskp)
20975b9bcba9SMikolaj Golub {
20985b9bcba9SMikolaj Golub 	int error;
20995b9bcba9SMikolaj Golub 	int mib[4];
21005b9bcba9SMikolaj Golub 	size_t len;
21015b9bcba9SMikolaj Golub 
21025b9bcba9SMikolaj Golub 	mib[0] = CTL_KERN;
21035b9bcba9SMikolaj Golub 	mib[1] = KERN_PROC;
21045b9bcba9SMikolaj Golub 	mib[2] = KERN_PROC_UMASK;
21055b9bcba9SMikolaj Golub 	mib[3] = pid;
21065b9bcba9SMikolaj Golub 	len = sizeof(*maskp);
210775b6a179SEnji Cooper 	error = sysctl(mib, nitems(mib), maskp, &len, NULL, 0);
210809b46be1SRobert Watson 	if (error != 0 && errno != ESRCH && errno != EPERM)
21095b9bcba9SMikolaj Golub 		warn("sysctl: kern.proc.umask: %d", pid);
21105b9bcba9SMikolaj Golub 	return (error);
21115b9bcba9SMikolaj Golub }
21125b9bcba9SMikolaj Golub 
21135b9bcba9SMikolaj Golub static int
21145b9bcba9SMikolaj Golub procstat_getumask_core(struct procstat_core *core, unsigned short *maskp)
21155b9bcba9SMikolaj Golub {
21165b9bcba9SMikolaj Golub 	size_t len;
21175b9bcba9SMikolaj Golub 	unsigned short *buf;
21185b9bcba9SMikolaj Golub 
21195b9bcba9SMikolaj Golub 	buf = procstat_core_get(core, PSC_TYPE_UMASK, NULL, &len);
21205b9bcba9SMikolaj Golub 	if (buf == NULL)
21215b9bcba9SMikolaj Golub 		return (-1);
21225b9bcba9SMikolaj Golub 	if (len < sizeof(*maskp)) {
21235b9bcba9SMikolaj Golub 		free(buf);
21245b9bcba9SMikolaj Golub 		return (-1);
21255b9bcba9SMikolaj Golub 	}
21265b9bcba9SMikolaj Golub 	*maskp = *buf;
21275b9bcba9SMikolaj Golub 	free(buf);
21285b9bcba9SMikolaj Golub 	return (0);
21295b9bcba9SMikolaj Golub }
21305b9bcba9SMikolaj Golub 
21315b9bcba9SMikolaj Golub int
21325b9bcba9SMikolaj Golub procstat_getumask(struct procstat *procstat, struct kinfo_proc *kp,
21335b9bcba9SMikolaj Golub     unsigned short *maskp)
21345b9bcba9SMikolaj Golub {
21355b9bcba9SMikolaj Golub 	switch(procstat->type) {
21365b9bcba9SMikolaj Golub 	case PROCSTAT_KVM:
21371f84c47eSMikolaj Golub 		return (procstat_getumask_kvm(procstat->kd, kp, maskp));
21385b9bcba9SMikolaj Golub 	case PROCSTAT_SYSCTL:
21395b9bcba9SMikolaj Golub 		return (procstat_getumask_sysctl(kp->ki_pid, maskp));
21405b9bcba9SMikolaj Golub 	case PROCSTAT_CORE:
21415b9bcba9SMikolaj Golub 		return (procstat_getumask_core(procstat->core, maskp));
21425b9bcba9SMikolaj Golub 	default:
21435b9bcba9SMikolaj Golub 		warnx("unknown access method: %d", procstat->type);
21445b9bcba9SMikolaj Golub 		return (-1);
21455b9bcba9SMikolaj Golub 	}
21465b9bcba9SMikolaj Golub }
21477cc0ebfdSMikolaj Golub 
21487cc0ebfdSMikolaj Golub static int
21491f84c47eSMikolaj Golub procstat_getrlimit_kvm(kvm_t *kd, struct kinfo_proc *kp, int which,
21501f84c47eSMikolaj Golub     struct rlimit* rlimit)
21511f84c47eSMikolaj Golub {
21521f84c47eSMikolaj Golub 	struct proc proc;
21531f84c47eSMikolaj Golub 	unsigned long offset;
21541f84c47eSMikolaj Golub 
21551f84c47eSMikolaj Golub 	assert(kd != NULL);
21561f84c47eSMikolaj Golub 	assert(kp != NULL);
21571f84c47eSMikolaj Golub 	assert(which >= 0 && which < RLIM_NLIMITS);
21581f84c47eSMikolaj Golub 	if (!kvm_read_all(kd, (unsigned long)kp->ki_paddr, &proc,
21591f84c47eSMikolaj Golub 	    sizeof(proc))) {
21601f84c47eSMikolaj Golub 		warnx("can't read proc struct at %p for pid %d",
21611f84c47eSMikolaj Golub 		    kp->ki_paddr, kp->ki_pid);
21621f84c47eSMikolaj Golub 		return (-1);
21631f84c47eSMikolaj Golub 	}
21641f84c47eSMikolaj Golub 	if (proc.p_limit == NULL)
21651f84c47eSMikolaj Golub 		return (-1);
21661f84c47eSMikolaj Golub 	offset = (unsigned long)proc.p_limit + sizeof(struct rlimit) * which;
21671f84c47eSMikolaj Golub 	if (!kvm_read_all(kd, offset, rlimit, sizeof(*rlimit))) {
21681f84c47eSMikolaj Golub 		warnx("can't read rlimit struct at %p for pid %d",
21691f84c47eSMikolaj Golub 		    (void *)offset, kp->ki_pid);
21701f84c47eSMikolaj Golub 		return (-1);
21711f84c47eSMikolaj Golub 	}
21721f84c47eSMikolaj Golub 	return (0);
21731f84c47eSMikolaj Golub }
21741f84c47eSMikolaj Golub 
21751f84c47eSMikolaj Golub static int
21767cc0ebfdSMikolaj Golub procstat_getrlimit_sysctl(pid_t pid, int which, struct rlimit* rlimit)
21777cc0ebfdSMikolaj Golub {
21787cc0ebfdSMikolaj Golub 	int error, name[5];
21797cc0ebfdSMikolaj Golub 	size_t len;
21807cc0ebfdSMikolaj Golub 
21817cc0ebfdSMikolaj Golub 	name[0] = CTL_KERN;
21827cc0ebfdSMikolaj Golub 	name[1] = KERN_PROC;
21837cc0ebfdSMikolaj Golub 	name[2] = KERN_PROC_RLIMIT;
21847cc0ebfdSMikolaj Golub 	name[3] = pid;
21857cc0ebfdSMikolaj Golub 	name[4] = which;
21867cc0ebfdSMikolaj Golub 	len = sizeof(struct rlimit);
218775b6a179SEnji Cooper 	error = sysctl(name, nitems(name), rlimit, &len, NULL, 0);
21887cc0ebfdSMikolaj Golub 	if (error < 0 && errno != ESRCH) {
21897cc0ebfdSMikolaj Golub 		warn("sysctl: kern.proc.rlimit: %d", pid);
21907cc0ebfdSMikolaj Golub 		return (-1);
21917cc0ebfdSMikolaj Golub 	}
21927cc0ebfdSMikolaj Golub 	if (error < 0 || len != sizeof(struct rlimit))
21937cc0ebfdSMikolaj Golub 		return (-1);
21947cc0ebfdSMikolaj Golub 	return (0);
21957cc0ebfdSMikolaj Golub }
21967cc0ebfdSMikolaj Golub 
21977cc0ebfdSMikolaj Golub static int
21987cc0ebfdSMikolaj Golub procstat_getrlimit_core(struct procstat_core *core, int which,
21997cc0ebfdSMikolaj Golub     struct rlimit* rlimit)
22007cc0ebfdSMikolaj Golub {
22017cc0ebfdSMikolaj Golub 	size_t len;
22027cc0ebfdSMikolaj Golub 	struct rlimit* rlimits;
22037cc0ebfdSMikolaj Golub 
22047cc0ebfdSMikolaj Golub 	if (which < 0 || which >= RLIM_NLIMITS) {
22057cc0ebfdSMikolaj Golub 		errno = EINVAL;
22067cc0ebfdSMikolaj Golub 		warn("getrlimit: which");
22077cc0ebfdSMikolaj Golub 		return (-1);
22087cc0ebfdSMikolaj Golub 	}
22097cc0ebfdSMikolaj Golub 	rlimits = procstat_core_get(core, PSC_TYPE_RLIMIT, NULL, &len);
22107cc0ebfdSMikolaj Golub 	if (rlimits == NULL)
22117cc0ebfdSMikolaj Golub 		return (-1);
22127cc0ebfdSMikolaj Golub 	if (len < sizeof(struct rlimit) * RLIM_NLIMITS) {
22137cc0ebfdSMikolaj Golub 		free(rlimits);
22147cc0ebfdSMikolaj Golub 		return (-1);
22157cc0ebfdSMikolaj Golub 	}
22167cc0ebfdSMikolaj Golub 	*rlimit = rlimits[which];
22179b207441SEric van Gyzen 	free(rlimits);
22187cc0ebfdSMikolaj Golub 	return (0);
22197cc0ebfdSMikolaj Golub }
22207cc0ebfdSMikolaj Golub 
22217cc0ebfdSMikolaj Golub int
22227cc0ebfdSMikolaj Golub procstat_getrlimit(struct procstat *procstat, struct kinfo_proc *kp, int which,
22237cc0ebfdSMikolaj Golub     struct rlimit* rlimit)
22247cc0ebfdSMikolaj Golub {
22257cc0ebfdSMikolaj Golub 	switch(procstat->type) {
22267cc0ebfdSMikolaj Golub 	case PROCSTAT_KVM:
22271f84c47eSMikolaj Golub 		return (procstat_getrlimit_kvm(procstat->kd, kp, which,
22281f84c47eSMikolaj Golub 		    rlimit));
22297cc0ebfdSMikolaj Golub 	case PROCSTAT_SYSCTL:
22307cc0ebfdSMikolaj Golub 		return (procstat_getrlimit_sysctl(kp->ki_pid, which, rlimit));
22317cc0ebfdSMikolaj Golub 	case PROCSTAT_CORE:
22327cc0ebfdSMikolaj Golub 		return (procstat_getrlimit_core(procstat->core, which, rlimit));
22337cc0ebfdSMikolaj Golub 	default:
22347cc0ebfdSMikolaj Golub 		warnx("unknown access method: %d", procstat->type);
22357cc0ebfdSMikolaj Golub 		return (-1);
22367cc0ebfdSMikolaj Golub 	}
22377cc0ebfdSMikolaj Golub }
22384cdf9796SMikolaj Golub 
22394cdf9796SMikolaj Golub static int
22404cdf9796SMikolaj Golub procstat_getpathname_sysctl(pid_t pid, char *pathname, size_t maxlen)
22414cdf9796SMikolaj Golub {
22424cdf9796SMikolaj Golub 	int error, name[4];
22434cdf9796SMikolaj Golub 	size_t len;
22444cdf9796SMikolaj Golub 
22454cdf9796SMikolaj Golub 	name[0] = CTL_KERN;
22464cdf9796SMikolaj Golub 	name[1] = KERN_PROC;
22474cdf9796SMikolaj Golub 	name[2] = KERN_PROC_PATHNAME;
22484cdf9796SMikolaj Golub 	name[3] = pid;
22494cdf9796SMikolaj Golub 	len = maxlen;
225075b6a179SEnji Cooper 	error = sysctl(name, nitems(name), pathname, &len, NULL, 0);
22514cdf9796SMikolaj Golub 	if (error != 0 && errno != ESRCH)
22524cdf9796SMikolaj Golub 		warn("sysctl: kern.proc.pathname: %d", pid);
22534cdf9796SMikolaj Golub 	if (len == 0)
22544cdf9796SMikolaj Golub 		pathname[0] = '\0';
22554cdf9796SMikolaj Golub 	return (error);
22564cdf9796SMikolaj Golub }
22574cdf9796SMikolaj Golub 
22584cdf9796SMikolaj Golub static int
22594cdf9796SMikolaj Golub procstat_getpathname_core(struct procstat_core *core, char *pathname,
22604cdf9796SMikolaj Golub     size_t maxlen)
22614cdf9796SMikolaj Golub {
22624cdf9796SMikolaj Golub 	struct kinfo_file *files;
22634cdf9796SMikolaj Golub 	int cnt, i, result;
22644cdf9796SMikolaj Golub 
22654cdf9796SMikolaj Golub 	files = kinfo_getfile_core(core, &cnt);
22664cdf9796SMikolaj Golub 	if (files == NULL)
22674cdf9796SMikolaj Golub 		return (-1);
22684cdf9796SMikolaj Golub 	result = -1;
22694cdf9796SMikolaj Golub 	for (i = 0; i < cnt; i++) {
22704cdf9796SMikolaj Golub 		if (files[i].kf_fd != KF_FD_TYPE_TEXT)
22714cdf9796SMikolaj Golub 			continue;
22724cdf9796SMikolaj Golub 		strncpy(pathname, files[i].kf_path, maxlen);
22734cdf9796SMikolaj Golub 		result = 0;
22744cdf9796SMikolaj Golub 		break;
22754cdf9796SMikolaj Golub 	}
22764cdf9796SMikolaj Golub 	free(files);
22774cdf9796SMikolaj Golub 	return (result);
22784cdf9796SMikolaj Golub }
22794cdf9796SMikolaj Golub 
22804cdf9796SMikolaj Golub int
22814cdf9796SMikolaj Golub procstat_getpathname(struct procstat *procstat, struct kinfo_proc *kp,
22824cdf9796SMikolaj Golub     char *pathname, size_t maxlen)
22834cdf9796SMikolaj Golub {
22844cdf9796SMikolaj Golub 	switch(procstat->type) {
22854cdf9796SMikolaj Golub 	case PROCSTAT_KVM:
2286dd70ad64SMikolaj Golub 		/* XXX: Return empty string. */
2287dd70ad64SMikolaj Golub 		if (maxlen > 0)
2288dd70ad64SMikolaj Golub 			pathname[0] = '\0';
2289dd70ad64SMikolaj Golub 		return (0);
22904cdf9796SMikolaj Golub 	case PROCSTAT_SYSCTL:
22914cdf9796SMikolaj Golub 		return (procstat_getpathname_sysctl(kp->ki_pid, pathname,
22924cdf9796SMikolaj Golub 		    maxlen));
22934cdf9796SMikolaj Golub 	case PROCSTAT_CORE:
22944cdf9796SMikolaj Golub 		return (procstat_getpathname_core(procstat->core, pathname,
22954cdf9796SMikolaj Golub 		    maxlen));
22964cdf9796SMikolaj Golub 	default:
22974cdf9796SMikolaj Golub 		warnx("unknown access method: %d", procstat->type);
22984cdf9796SMikolaj Golub 		return (-1);
22994cdf9796SMikolaj Golub 	}
23004cdf9796SMikolaj Golub }
2301eec6cb1cSMikolaj Golub 
2302eec6cb1cSMikolaj Golub static int
23031f84c47eSMikolaj Golub procstat_getosrel_kvm(kvm_t *kd, struct kinfo_proc *kp, int *osrelp)
23041f84c47eSMikolaj Golub {
23051f84c47eSMikolaj Golub 	struct proc proc;
23061f84c47eSMikolaj Golub 
23071f84c47eSMikolaj Golub 	assert(kd != NULL);
23081f84c47eSMikolaj Golub 	assert(kp != NULL);
23091f84c47eSMikolaj Golub 	if (!kvm_read_all(kd, (unsigned long)kp->ki_paddr, &proc,
23101f84c47eSMikolaj Golub 	    sizeof(proc))) {
23111f84c47eSMikolaj Golub 		warnx("can't read proc struct at %p for pid %d",
23121f84c47eSMikolaj Golub 		    kp->ki_paddr, kp->ki_pid);
23131f84c47eSMikolaj Golub 		return (-1);
23141f84c47eSMikolaj Golub 	}
23151f84c47eSMikolaj Golub 	*osrelp = proc.p_osrel;
23161f84c47eSMikolaj Golub 	return (0);
23171f84c47eSMikolaj Golub }
23181f84c47eSMikolaj Golub 
23191f84c47eSMikolaj Golub static int
2320eec6cb1cSMikolaj Golub procstat_getosrel_sysctl(pid_t pid, int *osrelp)
2321eec6cb1cSMikolaj Golub {
2322eec6cb1cSMikolaj Golub 	int error, name[4];
2323eec6cb1cSMikolaj Golub 	size_t len;
2324eec6cb1cSMikolaj Golub 
2325eec6cb1cSMikolaj Golub 	name[0] = CTL_KERN;
2326eec6cb1cSMikolaj Golub 	name[1] = KERN_PROC;
2327eec6cb1cSMikolaj Golub 	name[2] = KERN_PROC_OSREL;
2328eec6cb1cSMikolaj Golub 	name[3] = pid;
2329eec6cb1cSMikolaj Golub 	len = sizeof(*osrelp);
233075b6a179SEnji Cooper 	error = sysctl(name, nitems(name), osrelp, &len, NULL, 0);
2331eec6cb1cSMikolaj Golub 	if (error != 0 && errno != ESRCH)
2332eec6cb1cSMikolaj Golub 		warn("sysctl: kern.proc.osrel: %d", pid);
2333eec6cb1cSMikolaj Golub 	return (error);
2334eec6cb1cSMikolaj Golub }
2335eec6cb1cSMikolaj Golub 
2336eec6cb1cSMikolaj Golub static int
2337eec6cb1cSMikolaj Golub procstat_getosrel_core(struct procstat_core *core, int *osrelp)
2338eec6cb1cSMikolaj Golub {
2339eec6cb1cSMikolaj Golub 	size_t len;
2340eec6cb1cSMikolaj Golub 	int *buf;
2341eec6cb1cSMikolaj Golub 
2342eec6cb1cSMikolaj Golub 	buf = procstat_core_get(core, PSC_TYPE_OSREL, NULL, &len);
2343eec6cb1cSMikolaj Golub 	if (buf == NULL)
2344eec6cb1cSMikolaj Golub 		return (-1);
2345eec6cb1cSMikolaj Golub 	if (len < sizeof(*osrelp)) {
2346eec6cb1cSMikolaj Golub 		free(buf);
2347eec6cb1cSMikolaj Golub 		return (-1);
2348eec6cb1cSMikolaj Golub 	}
2349eec6cb1cSMikolaj Golub 	*osrelp = *buf;
2350eec6cb1cSMikolaj Golub 	free(buf);
2351eec6cb1cSMikolaj Golub 	return (0);
2352eec6cb1cSMikolaj Golub }
2353eec6cb1cSMikolaj Golub 
2354eec6cb1cSMikolaj Golub int
2355eec6cb1cSMikolaj Golub procstat_getosrel(struct procstat *procstat, struct kinfo_proc *kp, int *osrelp)
2356eec6cb1cSMikolaj Golub {
2357eec6cb1cSMikolaj Golub 	switch(procstat->type) {
2358eec6cb1cSMikolaj Golub 	case PROCSTAT_KVM:
23591f84c47eSMikolaj Golub 		return (procstat_getosrel_kvm(procstat->kd, kp, osrelp));
2360eec6cb1cSMikolaj Golub 	case PROCSTAT_SYSCTL:
2361eec6cb1cSMikolaj Golub 		return (procstat_getosrel_sysctl(kp->ki_pid, osrelp));
2362eec6cb1cSMikolaj Golub 	case PROCSTAT_CORE:
2363eec6cb1cSMikolaj Golub 		return (procstat_getosrel_core(procstat->core, osrelp));
2364eec6cb1cSMikolaj Golub 	default:
2365eec6cb1cSMikolaj Golub 		warnx("unknown access method: %d", procstat->type);
2366eec6cb1cSMikolaj Golub 		return (-1);
2367eec6cb1cSMikolaj Golub 	}
2368eec6cb1cSMikolaj Golub }
23692ff020d3SMikolaj Golub 
23702ff020d3SMikolaj Golub #define PROC_AUXV_MAX	256
23712ff020d3SMikolaj Golub 
23722ff020d3SMikolaj Golub #if __ELF_WORD_SIZE == 64
23732ff020d3SMikolaj Golub static const char *elf32_sv_names[] = {
23742ff020d3SMikolaj Golub 	"Linux ELF32",
23752ff020d3SMikolaj Golub 	"FreeBSD ELF32",
23762ff020d3SMikolaj Golub };
23772ff020d3SMikolaj Golub 
23782ff020d3SMikolaj Golub static int
23792ff020d3SMikolaj Golub is_elf32_sysctl(pid_t pid)
23802ff020d3SMikolaj Golub {
23812ff020d3SMikolaj Golub 	int error, name[4];
23822ff020d3SMikolaj Golub 	size_t len, i;
23832ff020d3SMikolaj Golub 	static char sv_name[256];
23842ff020d3SMikolaj Golub 
23852ff020d3SMikolaj Golub 	name[0] = CTL_KERN;
23862ff020d3SMikolaj Golub 	name[1] = KERN_PROC;
23872ff020d3SMikolaj Golub 	name[2] = KERN_PROC_SV_NAME;
23882ff020d3SMikolaj Golub 	name[3] = pid;
23892ff020d3SMikolaj Golub 	len = sizeof(sv_name);
239075b6a179SEnji Cooper 	error = sysctl(name, nitems(name), sv_name, &len, NULL, 0);
23912ff020d3SMikolaj Golub 	if (error != 0 || len == 0)
23922ff020d3SMikolaj Golub 		return (0);
23932ff020d3SMikolaj Golub 	for (i = 0; i < sizeof(elf32_sv_names) / sizeof(*elf32_sv_names); i++) {
23942ff020d3SMikolaj Golub 		if (strncmp(sv_name, elf32_sv_names[i], sizeof(sv_name)) == 0)
23952ff020d3SMikolaj Golub 			return (1);
23962ff020d3SMikolaj Golub 	}
23972ff020d3SMikolaj Golub 	return (0);
23982ff020d3SMikolaj Golub }
23992ff020d3SMikolaj Golub 
24002ff020d3SMikolaj Golub static Elf_Auxinfo *
24012ff020d3SMikolaj Golub procstat_getauxv32_sysctl(pid_t pid, unsigned int *cntp)
24022ff020d3SMikolaj Golub {
24032ff020d3SMikolaj Golub 	Elf_Auxinfo *auxv;
24042ff020d3SMikolaj Golub 	Elf32_Auxinfo *auxv32;
24052ff020d3SMikolaj Golub 	void *ptr;
24062ff020d3SMikolaj Golub 	size_t len;
24072ff020d3SMikolaj Golub 	unsigned int i, count;
24082ff020d3SMikolaj Golub 	int name[4];
24092ff020d3SMikolaj Golub 
24102ff020d3SMikolaj Golub 	name[0] = CTL_KERN;
24112ff020d3SMikolaj Golub 	name[1] = KERN_PROC;
24122ff020d3SMikolaj Golub 	name[2] = KERN_PROC_AUXV;
24132ff020d3SMikolaj Golub 	name[3] = pid;
24142ff020d3SMikolaj Golub 	len = PROC_AUXV_MAX * sizeof(Elf32_Auxinfo);
24152ff020d3SMikolaj Golub 	auxv = NULL;
24162ff020d3SMikolaj Golub 	auxv32 = malloc(len);
24172ff020d3SMikolaj Golub 	if (auxv32 == NULL) {
24182ff020d3SMikolaj Golub 		warn("malloc(%zu)", len);
24192ff020d3SMikolaj Golub 		goto out;
24202ff020d3SMikolaj Golub 	}
242175b6a179SEnji Cooper 	if (sysctl(name, nitems(name), auxv32, &len, NULL, 0) == -1) {
24222ff020d3SMikolaj Golub 		if (errno != ESRCH && errno != EPERM)
24232ff020d3SMikolaj Golub 			warn("sysctl: kern.proc.auxv: %d: %d", pid, errno);
24242ff020d3SMikolaj Golub 		goto out;
24252ff020d3SMikolaj Golub 	}
24262ff020d3SMikolaj Golub 	count = len / sizeof(Elf_Auxinfo);
24272ff020d3SMikolaj Golub 	auxv = malloc(count  * sizeof(Elf_Auxinfo));
24282ff020d3SMikolaj Golub 	if (auxv == NULL) {
24292ff020d3SMikolaj Golub 		warn("malloc(%zu)", count * sizeof(Elf_Auxinfo));
24302ff020d3SMikolaj Golub 		goto out;
24312ff020d3SMikolaj Golub 	}
24322ff020d3SMikolaj Golub 	for (i = 0; i < count; i++) {
24332ff020d3SMikolaj Golub 		/*
24342ff020d3SMikolaj Golub 		 * XXX: We expect that values for a_type on a 32-bit platform
24352ff020d3SMikolaj Golub 		 * are directly mapped to values on 64-bit one, which is not
24362ff020d3SMikolaj Golub 		 * necessarily true.
24372ff020d3SMikolaj Golub 		 */
24382ff020d3SMikolaj Golub 		auxv[i].a_type = auxv32[i].a_type;
24392ff020d3SMikolaj Golub 		ptr = &auxv32[i].a_un;
24402ff020d3SMikolaj Golub 		auxv[i].a_un.a_val = *((uint32_t *)ptr);
24412ff020d3SMikolaj Golub 	}
24422ff020d3SMikolaj Golub 	*cntp = count;
24432ff020d3SMikolaj Golub out:
24442ff020d3SMikolaj Golub 	free(auxv32);
24452ff020d3SMikolaj Golub 	return (auxv);
24462ff020d3SMikolaj Golub }
24472ff020d3SMikolaj Golub #endif /* __ELF_WORD_SIZE == 64 */
24482ff020d3SMikolaj Golub 
24492ff020d3SMikolaj Golub static Elf_Auxinfo *
24502ff020d3SMikolaj Golub procstat_getauxv_sysctl(pid_t pid, unsigned int *cntp)
24512ff020d3SMikolaj Golub {
24522ff020d3SMikolaj Golub 	Elf_Auxinfo *auxv;
24532ff020d3SMikolaj Golub 	int name[4];
24542ff020d3SMikolaj Golub 	size_t len;
24552ff020d3SMikolaj Golub 
24562ff020d3SMikolaj Golub #if __ELF_WORD_SIZE == 64
24572ff020d3SMikolaj Golub 	if (is_elf32_sysctl(pid))
24582ff020d3SMikolaj Golub 		return (procstat_getauxv32_sysctl(pid, cntp));
24592ff020d3SMikolaj Golub #endif
24602ff020d3SMikolaj Golub 	name[0] = CTL_KERN;
24612ff020d3SMikolaj Golub 	name[1] = KERN_PROC;
24622ff020d3SMikolaj Golub 	name[2] = KERN_PROC_AUXV;
24632ff020d3SMikolaj Golub 	name[3] = pid;
24642ff020d3SMikolaj Golub 	len = PROC_AUXV_MAX * sizeof(Elf_Auxinfo);
24652ff020d3SMikolaj Golub 	auxv = malloc(len);
24662ff020d3SMikolaj Golub 	if (auxv == NULL) {
24672ff020d3SMikolaj Golub 		warn("malloc(%zu)", len);
24682ff020d3SMikolaj Golub 		return (NULL);
24692ff020d3SMikolaj Golub 	}
247075b6a179SEnji Cooper 	if (sysctl(name, nitems(name), auxv, &len, NULL, 0) == -1) {
24712ff020d3SMikolaj Golub 		if (errno != ESRCH && errno != EPERM)
24722ff020d3SMikolaj Golub 			warn("sysctl: kern.proc.auxv: %d: %d", pid, errno);
24732ff020d3SMikolaj Golub 		free(auxv);
24742ff020d3SMikolaj Golub 		return (NULL);
24752ff020d3SMikolaj Golub 	}
24762ff020d3SMikolaj Golub 	*cntp = len / sizeof(Elf_Auxinfo);
24772ff020d3SMikolaj Golub 	return (auxv);
24782ff020d3SMikolaj Golub }
24792ff020d3SMikolaj Golub 
24802ff020d3SMikolaj Golub static Elf_Auxinfo *
24812ff020d3SMikolaj Golub procstat_getauxv_core(struct procstat_core *core, unsigned int *cntp)
24822ff020d3SMikolaj Golub {
24832ff020d3SMikolaj Golub 	Elf_Auxinfo *auxv;
24842ff020d3SMikolaj Golub 	size_t len;
24852ff020d3SMikolaj Golub 
24862ff020d3SMikolaj Golub 	auxv = procstat_core_get(core, PSC_TYPE_AUXV, NULL, &len);
24872ff020d3SMikolaj Golub 	if (auxv == NULL)
24882ff020d3SMikolaj Golub 		return (NULL);
24892ff020d3SMikolaj Golub 	*cntp = len / sizeof(Elf_Auxinfo);
24902ff020d3SMikolaj Golub 	return (auxv);
24912ff020d3SMikolaj Golub }
24922ff020d3SMikolaj Golub 
24932ff020d3SMikolaj Golub Elf_Auxinfo *
24942ff020d3SMikolaj Golub procstat_getauxv(struct procstat *procstat, struct kinfo_proc *kp,
24952ff020d3SMikolaj Golub     unsigned int *cntp)
24962ff020d3SMikolaj Golub {
24972ff020d3SMikolaj Golub 	switch(procstat->type) {
24982ff020d3SMikolaj Golub 	case PROCSTAT_KVM:
24992ff020d3SMikolaj Golub 		warnx("kvm method is not supported");
25002ff020d3SMikolaj Golub 		return (NULL);
25012ff020d3SMikolaj Golub 	case PROCSTAT_SYSCTL:
25022ff020d3SMikolaj Golub 		return (procstat_getauxv_sysctl(kp->ki_pid, cntp));
25032ff020d3SMikolaj Golub 	case PROCSTAT_CORE:
25042ff020d3SMikolaj Golub 		return (procstat_getauxv_core(procstat->core, cntp));
25052ff020d3SMikolaj Golub 	default:
25062ff020d3SMikolaj Golub 		warnx("unknown access method: %d", procstat->type);
25072ff020d3SMikolaj Golub 		return (NULL);
25082ff020d3SMikolaj Golub 	}
25092ff020d3SMikolaj Golub }
25102ff020d3SMikolaj Golub 
25112ff020d3SMikolaj Golub void
25122ff020d3SMikolaj Golub procstat_freeauxv(struct procstat *procstat __unused, Elf_Auxinfo *auxv)
25132ff020d3SMikolaj Golub {
25142ff020d3SMikolaj Golub 
25152ff020d3SMikolaj Golub 	free(auxv);
25162ff020d3SMikolaj Golub }
251789358231SMikolaj Golub 
251886be94fcSTycho Nightingale static struct ptrace_lwpinfo *
251986be94fcSTycho Nightingale procstat_getptlwpinfo_core(struct procstat_core *core, unsigned int *cntp)
252086be94fcSTycho Nightingale {
252186be94fcSTycho Nightingale 	void *buf;
252286be94fcSTycho Nightingale 	struct ptrace_lwpinfo *pl;
252386be94fcSTycho Nightingale 	unsigned int cnt;
252486be94fcSTycho Nightingale 	size_t len;
252586be94fcSTycho Nightingale 
252686be94fcSTycho Nightingale 	cnt = procstat_core_note_count(core, PSC_TYPE_PTLWPINFO);
252786be94fcSTycho Nightingale 	if (cnt == 0)
252886be94fcSTycho Nightingale 		return (NULL);
252986be94fcSTycho Nightingale 
253086be94fcSTycho Nightingale 	len = cnt * sizeof(*pl);
253186be94fcSTycho Nightingale 	buf = calloc(1, len);
253286be94fcSTycho Nightingale 	pl = procstat_core_get(core, PSC_TYPE_PTLWPINFO, buf, &len);
253386be94fcSTycho Nightingale 	if (pl == NULL) {
253486be94fcSTycho Nightingale 		free(buf);
253586be94fcSTycho Nightingale 		return (NULL);
253686be94fcSTycho Nightingale 	}
253786be94fcSTycho Nightingale 	*cntp = len / sizeof(*pl);
253886be94fcSTycho Nightingale 	return (pl);
253986be94fcSTycho Nightingale }
254086be94fcSTycho Nightingale 
254186be94fcSTycho Nightingale struct ptrace_lwpinfo *
254286be94fcSTycho Nightingale procstat_getptlwpinfo(struct procstat *procstat, unsigned int *cntp)
254386be94fcSTycho Nightingale {
254486be94fcSTycho Nightingale 	switch (procstat->type) {
2545a2ae08e7SEnji Cooper 	case PROCSTAT_KVM:
2546a2ae08e7SEnji Cooper 		warnx("kvm method is not supported");
2547a2ae08e7SEnji Cooper 		return (NULL);
2548a2ae08e7SEnji Cooper 	case PROCSTAT_SYSCTL:
2549a2ae08e7SEnji Cooper 		warnx("sysctl method is not supported");
2550a2ae08e7SEnji Cooper 		return (NULL);
255186be94fcSTycho Nightingale 	case PROCSTAT_CORE:
255286be94fcSTycho Nightingale 	 	return (procstat_getptlwpinfo_core(procstat->core, cntp));
255386be94fcSTycho Nightingale 	default:
255486be94fcSTycho Nightingale 		warnx("unknown access method: %d", procstat->type);
255586be94fcSTycho Nightingale 		return (NULL);
255686be94fcSTycho Nightingale 	}
255786be94fcSTycho Nightingale }
255886be94fcSTycho Nightingale 
255986be94fcSTycho Nightingale void
256086be94fcSTycho Nightingale procstat_freeptlwpinfo(struct procstat *procstat __unused,
256186be94fcSTycho Nightingale     struct ptrace_lwpinfo *pl)
256286be94fcSTycho Nightingale {
256386be94fcSTycho Nightingale 	free(pl);
256486be94fcSTycho Nightingale }
256586be94fcSTycho Nightingale 
256689358231SMikolaj Golub static struct kinfo_kstack *
256789358231SMikolaj Golub procstat_getkstack_sysctl(pid_t pid, int *cntp)
256889358231SMikolaj Golub {
256989358231SMikolaj Golub 	struct kinfo_kstack *kkstp;
257089358231SMikolaj Golub 	int error, name[4];
257189358231SMikolaj Golub 	size_t len;
257289358231SMikolaj Golub 
257389358231SMikolaj Golub 	name[0] = CTL_KERN;
257489358231SMikolaj Golub 	name[1] = KERN_PROC;
257589358231SMikolaj Golub 	name[2] = KERN_PROC_KSTACK;
257689358231SMikolaj Golub 	name[3] = pid;
257789358231SMikolaj Golub 
257889358231SMikolaj Golub 	len = 0;
257975b6a179SEnji Cooper 	error = sysctl(name, nitems(name), NULL, &len, NULL, 0);
258089358231SMikolaj Golub 	if (error < 0 && errno != ESRCH && errno != EPERM && errno != ENOENT) {
258189358231SMikolaj Golub 		warn("sysctl: kern.proc.kstack: %d", pid);
258289358231SMikolaj Golub 		return (NULL);
258389358231SMikolaj Golub 	}
258489358231SMikolaj Golub 	if (error == -1 && errno == ENOENT) {
258589358231SMikolaj Golub 		warnx("sysctl: kern.proc.kstack unavailable"
258689358231SMikolaj Golub 		    " (options DDB or options STACK required in kernel)");
258789358231SMikolaj Golub 		return (NULL);
258889358231SMikolaj Golub 	}
258989358231SMikolaj Golub 	if (error == -1)
259089358231SMikolaj Golub 		return (NULL);
259189358231SMikolaj Golub 	kkstp = malloc(len);
259289358231SMikolaj Golub 	if (kkstp == NULL) {
259389358231SMikolaj Golub 		warn("malloc(%zu)", len);
259489358231SMikolaj Golub 		return (NULL);
259589358231SMikolaj Golub 	}
259675b6a179SEnji Cooper 	if (sysctl(name, nitems(name), kkstp, &len, NULL, 0) == -1) {
259789358231SMikolaj Golub 		warn("sysctl: kern.proc.pid: %d", pid);
259889358231SMikolaj Golub 		free(kkstp);
259989358231SMikolaj Golub 		return (NULL);
260089358231SMikolaj Golub 	}
260189358231SMikolaj Golub 	*cntp = len / sizeof(*kkstp);
260289358231SMikolaj Golub 
260389358231SMikolaj Golub 	return (kkstp);
260489358231SMikolaj Golub }
260589358231SMikolaj Golub 
260689358231SMikolaj Golub struct kinfo_kstack *
260789358231SMikolaj Golub procstat_getkstack(struct procstat *procstat, struct kinfo_proc *kp,
260889358231SMikolaj Golub     unsigned int *cntp)
260989358231SMikolaj Golub {
261089358231SMikolaj Golub 	switch(procstat->type) {
261189358231SMikolaj Golub 	case PROCSTAT_KVM:
261289358231SMikolaj Golub 		warnx("kvm method is not supported");
261389358231SMikolaj Golub 		return (NULL);
261489358231SMikolaj Golub 	case PROCSTAT_SYSCTL:
261589358231SMikolaj Golub 		return (procstat_getkstack_sysctl(kp->ki_pid, cntp));
261689358231SMikolaj Golub 	case PROCSTAT_CORE:
261789358231SMikolaj Golub 		warnx("core method is not supported");
261889358231SMikolaj Golub 		return (NULL);
261989358231SMikolaj Golub 	default:
262089358231SMikolaj Golub 		warnx("unknown access method: %d", procstat->type);
262189358231SMikolaj Golub 		return (NULL);
262289358231SMikolaj Golub 	}
262389358231SMikolaj Golub }
262489358231SMikolaj Golub 
262589358231SMikolaj Golub void
262689358231SMikolaj Golub procstat_freekstack(struct procstat *procstat __unused,
262789358231SMikolaj Golub     struct kinfo_kstack *kkstp)
262889358231SMikolaj Golub {
262989358231SMikolaj Golub 
263089358231SMikolaj Golub 	free(kkstp);
263189358231SMikolaj Golub }
2632