xref: /netbsd/usr.sbin/pstat/pstat.c (revision f1d601d9)
1 /*	$NetBSD: pstat.c,v 1.130 2019/12/11 19:51:36 ad Exp $	*/
2 
3 /*-
4  * Copyright (c) 1980, 1991, 1993, 1994
5  *	The Regents of the University of California.  All rights reserved.
6  *
7  * Redistribution and use in source and binary forms, with or without
8  * modification, are permitted provided that the following conditions
9  * are met:
10  * 1. Redistributions of source code must retain the above copyright
11  *    notice, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  * 3. Neither the name of the University nor the names of its contributors
16  *    may be used to endorse or promote products derived from this software
17  *    without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29  * SUCH DAMAGE.
30  */
31 
32 #include <sys/cdefs.h>
33 #ifndef lint
34 __COPYRIGHT("@(#) Copyright (c) 1980, 1991, 1993, 1994\
35  The Regents of the University of California.  All rights reserved.");
36 #endif /* not lint */
37 
38 #ifndef lint
39 #if 0
40 static char sccsid[] = "@(#)pstat.c	8.16 (Berkeley) 5/9/95";
41 #else
42 __RCSID("$NetBSD: pstat.c,v 1.130 2019/12/11 19:51:36 ad Exp $");
43 #endif
44 #endif /* not lint */
45 
46 #define _KERNEL
47 #include <sys/types.h>
48 #undef _KERNEL
49 #include <sys/param.h>
50 #include <sys/time.h>
51 #include <sys/vnode.h>
52 #include <sys/ucred.h>
53 #include <stdbool.h>
54 #define _KERNEL
55 #define NFS
56 #include <sys/mount.h>
57 #undef NFS
58 #include <sys/file.h>
59 #include <ufs/ufs/inode.h>
60 #include <ufs/ufs/ufsmount.h>
61 #include <sys/uio.h>
62 #include <miscfs/genfs/layer.h>
63 #undef _KERNEL
64 #include <sys/stat.h>
65 #include <nfs/nfsproto.h>
66 #include <nfs/rpcv2.h>
67 #include <nfs/nfs.h>
68 #include <nfs/nfsnode.h>
69 #include <sys/ioctl.h>
70 #include <sys/tty.h>
71 #include <sys/conf.h>
72 
73 #include <sys/sysctl.h>
74 
75 #include <err.h>
76 #include <errno.h>
77 #include <kvm.h>
78 #include <limits.h>
79 #include <nlist.h>
80 #include <stdio.h>
81 #include <stdlib.h>
82 #include <string.h>
83 #include <unistd.h>
84 
85 #include "swapctl.h"
86 
87 struct nlist nl[] = {
88 #define	V_LRU_LIST	0
89 	{ "_lru_list", 0, 0, 0, 0 }	,	/* address of lru lists. */
90 #define	V_NUMV		1
91 	{ "_numvnodes", 0, 0, 0, 0 },
92 #define	V_NEXT_OFFSET	2
93 	{ "_vnode_offset_next_by_lru", 0, 0, 0, 0 },
94 #define	FNL_NFILE	3
95 	{ "_nfiles", 0, 0, 0, 0 },
96 #define FNL_MAXFILE	4
97 	{ "_maxfiles", 0, 0, 0, 0 },
98 #define TTY_NTTY	5
99 	{ "_tty_count", 0, 0, 0, 0 },
100 #define TTY_TTYLIST	6
101 	{ "_ttylist", 0, 0, 0, 0 },
102 #define NLMANDATORY TTY_TTYLIST	/* names up to here are mandatory */
103 	{ "", 0, 0, 0, 0 }
104 };
105 
106 int	usenumflag;
107 int	totalflag;
108 int	kflag;
109 int	hflag;
110 char	*nlistf	= NULL;
111 char	*memf	= NULL;
112 kvm_t	*kd;
113 
114 static const char * const dtypes[] = { DTYPE_NAMES };
115 
116 
117 static const struct {
118 	u_int m_flag;
119 	u_int m_visible;
120 	const char *m_name;
121 } mnt_flags[] = {
122 	__MNT_FLAGS
123 };
124 
125 struct flagbit_desc {
126 	u_int fd_flags;
127 	char fd_mark;
128 };
129 
130 #define	SVAR(var) __STRING(var)	/* to force expansion */
131 #define	KGET(idx, var)							\
132 	KGET1(idx, &var, sizeof(var), SVAR(var))
133 #define	KGET1(idx, p, s, msg)						\
134 	KGET2(nl[idx].n_value, p, s, msg)
135 #define	KGET2(addr, p, s, msg) do {					\
136 	if (kvm_read(kd, (u_long)(addr), p, s) != s)			\
137 		warnx("cannot read %s: %s", msg, kvm_geterr(kd));	\
138 } while (/* CONSTCOND */0)
139 #define	KGETRET(addr, p, s, msg) do {					\
140 	if (kvm_read(kd, (u_long)(addr), p, s) != s) {			\
141 		warnx("cannot read %s: %s", msg, kvm_geterr(kd));	\
142 		return (0);						\
143 	}								\
144 } while (/* CONSTCOND */0)
145 
146 #if 1				/* This is copied from vmstat/vmstat.c */
147 /*
148  * Print single word.  `ovflow' is number of characters didn't fit
149  * on the last word.  `fmt' is a format string to print this word.
150  * It must contain asterisk for field width.  `width' is a width
151  * occupied by this word.  `fixed' is a number of constant chars in
152  * `fmt'.  `val' is a value to be printed using format string `fmt'.
153  */
154 #define	PRWORD(ovflw, fmt, width, fixed, val) do {	\
155 	(ovflw) += printf((fmt),			\
156 	    (width) - (fixed) - (ovflw) > 0 ?		\
157 	    (width) - (fixed) - (ovflw) : 0,		\
158 	    (val)) - (width);				\
159 	if ((ovflw) < 0)				\
160 		(ovflw) = 0;				\
161 } while (/* CONSTCOND */0)
162 #endif
163 
164 void	filemode(void);
165 int	getfiles(char **, int *, char **);
166 int	getflags(const struct flagbit_desc *, char *, u_int);
167 struct mount *
168 	getmnt(struct mount *);
169 char *	kinfo_vnodes(int *);
170 void	layer_header(void);
171 int	layer_print(struct vnode *, int);
172 char *	loadvnodes(int *);
173 void	mount_print(struct mount *);
174 void	nfs_header(void);
175 int	nfs_print(struct vnode *, int);
176 void	ttymode(void);
177 void	ttyprt(struct tty *);
178 void	ufs_header(void);
179 int	ufs_print(struct vnode *, int);
180 int	ext2fs_print(struct vnode *, int);
181 __dead void	usage(void);
182 void	vnode_header(void);
183 int	vnode_print(struct vnode *, struct vnode *);
184 void	vnodemode(void);
185 
186 int
187 main(int argc, char *argv[])
188 {
189 	int ch, i, quit, ret, use_sysctl;
190 	int fileflag, swapflag, ttyflag, vnodeflag;
191 	gid_t egid = getegid();
192 	char buf[_POSIX2_LINE_MAX];
193 
194 	setegid(getgid());
195 	fileflag = swapflag = ttyflag = vnodeflag = 0;
196 	while ((ch = getopt(argc, argv, "TM:N:fghikmnstv")) != -1)
197 		switch (ch) {
198 		case 'f':
199 			fileflag = 1;
200 			break;
201 		case 'M':
202 			memf = optarg;
203 			break;
204 		case 'N':
205 			nlistf = optarg;
206 			break;
207 		case 'n':
208 			usenumflag = 1;
209 			break;
210 		case 's':
211 			swapflag = 1;
212 			break;
213 		case 'T':
214 			totalflag = 1;
215 			break;
216 		case 't':
217 			ttyflag = 1;
218 			break;
219 		case 'k':
220 			kflag = 1;
221 			break;
222 		case 'g':
223 			kflag = 3; /* 1k ^ 3 */
224 			break;
225 		case 'h':
226 			hflag = 1;
227 			break;
228 		case 'm':
229 			kflag = 2; /* 1k ^ 2 */
230 			break;
231 		case 'v':
232 		case 'i':		/* Backward compatibility. */
233 			vnodeflag = 1;
234 			break;
235 		default:
236 			usage();
237 		}
238 	argc -= optind;
239 	argv += optind;
240 
241 	/*
242 	 * Discard setgid privileges.  If not the running kernel, we toss
243 	 * them away totally so that bad guys can't print interesting stuff
244 	 * from kernel memory, otherwise switch back to kmem for the
245 	 * duration of the kvm_openfiles() call.
246 	 */
247 	if (nlistf != NULL || memf != NULL)
248 		(void)setgid(getgid());
249 	else
250 		(void)setegid(egid);
251 
252 	use_sysctl = (nlistf == NULL && memf == NULL);
253 
254 	if ((kd = kvm_openfiles(nlistf, memf, NULL, O_RDONLY, buf)) == 0)
255 		errx(1, "kvm_openfiles: %s", buf);
256 
257 	/* get rid of it now anyway */
258 	if (nlistf == NULL && memf == NULL)
259 		(void)setgid(getgid());
260 	if ((ret = kvm_nlist(kd, nl)) != 0) {
261 		if (ret == -1)
262 			errx(1, "kvm_nlist: %s", kvm_geterr(kd));
263 		for (i = quit = 0; i <= NLMANDATORY; i++)
264 			if (!nl[i].n_value) {
265 				quit = 1;
266 				warnx("undefined symbol: %s", nl[i].n_name);
267 			}
268 		if (quit)
269 			exit(1);
270 	}
271 	if (!(fileflag | vnodeflag | ttyflag | swapflag | totalflag))
272 		usage();
273 	if (fileflag || totalflag)
274 		filemode();
275 	if (vnodeflag || totalflag)
276 		vnodemode();
277 	if (ttyflag)
278 		ttymode();
279 	if (swapflag || totalflag)
280 		if (use_sysctl)
281 			list_swap(0, kflag, 0, totalflag, 1, hflag);
282 	exit(0);
283 }
284 
285 #define	VPTRSZ  sizeof(struct vnode *)
286 #define	VNODESZ sizeof(struct vnode)
287 #define	PTRSTRWIDTH ((int)sizeof(void *) * 2) /* Width of resulting string
288 						 when pointer is printed
289 						 in hexadecimal. */
290 
291 static void
292 devprintf(char *buf, size_t buflen, dev_t dev)
293 {
294 	(void)snprintf(buf, buflen, "%llu,%llu",
295 	    (unsigned long long)major(dev),
296 	    (unsigned long long)minor(dev));
297 }
298 
299 void
300 vnodemode(void)
301 {
302 	char *e_vnodebase, *endvnode, *evp;
303 	struct vnode *vp;
304 	struct mount *maddr, *mp;
305 	int numvnodes, ovflw;
306 	int (*vnode_fsprint) (struct vnode *, int); /* per-fs data printer */
307 
308 	mp = NULL;
309 	e_vnodebase = loadvnodes(&numvnodes);
310 	if (totalflag) {
311 		(void)printf("%7d vnodes\n", numvnodes);
312 		goto out;
313 	}
314 	endvnode = e_vnodebase + numvnodes * (VPTRSZ + VNODESZ);
315 	(void)printf("%d active vnodes\n", numvnodes);
316 
317 #define	ST	mp->mnt_stat
318 #define	FSTYPE_IS(mp, name)						\
319 	(strncmp((mp)->mnt_stat.f_fstypename, (name), 			\
320 	sizeof((mp)->mnt_stat.f_fstypename)) == 0)
321 	maddr = NULL;
322 	vnode_fsprint = NULL;
323 	for (evp = e_vnodebase; evp < endvnode; evp += VPTRSZ + VNODESZ) {
324 		vp = (struct vnode *)(evp + VPTRSZ);
325 		if (vp->v_mount != maddr) {
326 			/*
327 			 * New filesystem
328 			 */
329 			if ((mp = getmnt(vp->v_mount)) == NULL)
330 				continue;
331 			maddr = vp->v_mount;
332 			mount_print(mp);
333 			vnode_header();
334 			/*
335 			 * XXX do this in a more fs-independent way
336 			 */
337 			if (FSTYPE_IS(mp, MOUNT_FFS) ||
338 			    FSTYPE_IS(mp, MOUNT_MFS)) {
339 				ufs_header();
340 				vnode_fsprint = ufs_print;
341 			} else if (FSTYPE_IS(mp, MOUNT_NFS)) {
342 				nfs_header();
343 				vnode_fsprint = nfs_print;
344 			} else if (FSTYPE_IS(mp, MOUNT_EXT2FS)) {
345 				ufs_header();
346 				vnode_fsprint = ext2fs_print;
347 			} else if (FSTYPE_IS(mp, MOUNT_NULL) ||
348 			    FSTYPE_IS(mp, MOUNT_OVERLAY) ||
349 			    FSTYPE_IS(mp, MOUNT_UMAP)) {
350 				layer_header();
351 				vnode_fsprint = layer_print;
352 			} else
353 				vnode_fsprint = NULL;
354 			(void)printf("\n");
355 		}
356 		ovflw = vnode_print(*(struct vnode **)evp, vp);
357 		if (VTOI(vp) != NULL && vnode_fsprint != NULL)
358 			(*vnode_fsprint)(vp, ovflw);
359 		(void)printf("\n");
360 	}
361 
362  out:
363 	if (e_vnodebase)
364 		free(e_vnodebase);
365 }
366 
367 int
368 getflags(const struct flagbit_desc *fd, char *p, u_int flags)
369 {
370 	char *q = p;
371 
372 	if (flags == 0) {
373 		*p++ = '-';
374 		*p = '\0';
375 		return (0);
376 	}
377 
378 	for (; fd->fd_flags != 0; fd++)
379 		if ((flags & fd->fd_flags) != 0)
380 			*p++ = fd->fd_mark;
381 	*p = '\0';
382 	return (p - q);
383 }
384 
385 const struct flagbit_desc vnode_flags[] = {
386 	{ VV_ROOT,	'R' },
387 	{ VI_TEXT,	'T' },
388 	{ VV_SYSTEM,	'S' },
389 	{ VV_ISTTY,	'I' },
390 	{ VI_EXECMAP,	'E' },
391 	{ VU_DIROP,	'D' },
392 	{ VI_ONWORKLST,	'O' },
393 	{ VV_MPSAFE,    'M' },
394 	{ 0,		'\0' },
395 };
396 
397 void
398 vnode_header(void)
399 {
400 
401 	(void)printf("%-*s TYP VFLAG  USE HOLD TAG NPAGE",
402 	    PTRSTRWIDTH, "ADDR");
403 }
404 
405 int
406 vnode_print(struct vnode *avnode, struct vnode *vp)
407 {
408 	const char *type;
409 	char flags[sizeof(vnode_flags) / sizeof(vnode_flags[0])];
410 	int ovflw;
411 
412 	/*
413 	 * set type
414 	 */
415 	switch (vp->v_type) {
416 	case VNON:
417 		type = "non"; break;
418 	case VREG:
419 		type = "reg"; break;
420 	case VDIR:
421 		type = "dir"; break;
422 	case VBLK:
423 		type = "blk"; break;
424 	case VCHR:
425 		type = "chr"; break;
426 	case VLNK:
427 		type = "lnk"; break;
428 	case VSOCK:
429 		type = "soc"; break;
430 	case VFIFO:
431 		type = "fif"; break;
432 	case VBAD:
433 		type = "bad"; break;
434 	default:
435 		type = "unk"; break;
436 	}
437 	/*
438 	 * gather flags
439 	 */
440 	(void)getflags(vnode_flags, flags,
441 	    vp->v_uflag | vp->v_iflag | vp->v_vflag);
442 
443 	ovflw = 0;
444 	PRWORD(ovflw, "%*lx", PTRSTRWIDTH, 0, (long)avnode);
445 	PRWORD(ovflw, " %*s", 4, 1, type);
446 	PRWORD(ovflw, " %*s", 6, 1, flags);
447 	PRWORD(ovflw, " %*ld", 5, 1, (long)vp->v_usecount);
448 	PRWORD(ovflw, " %*ld", 5, 1, (long)vp->v_holdcnt);
449 	PRWORD(ovflw, " %*d", 4, 1, vp->v_tag);
450 	PRWORD(ovflw, " %*d", 6, 1, vp->v_uobj.uo_npages);
451 	return (ovflw);
452 }
453 
454 const struct flagbit_desc ufs_flags[] = {
455 	{ IN_ACCESS,	'A' },
456 	{ IN_CHANGE,	'C' },
457 	{ IN_UPDATE,	'U' },
458 	{ IN_MODIFIED,	'M' },
459 	{ IN_ACCESSED,	'a' },
460 	{ IN_SHLOCK,	'S' },
461 	{ IN_EXLOCK,	'E' },
462 	{ IN_SPACECOUNTED, 's' },
463 	{ 0,		'\0' },
464 };
465 
466 void
467 ufs_header(void)
468 {
469 
470 	(void)printf(" FILEID IFLAG RDEV|SZ");
471 }
472 
473 int
474 ufs_print(struct vnode *vp, int ovflw)
475 {
476 	struct inode inode, *ip = &inode;
477 	union dinode {
478 		struct ufs1_dinode dp1;
479 		struct ufs2_dinode dp2;
480 	} dip;
481 	struct ufsmount ump;
482 	char flags[sizeof(ufs_flags) / sizeof(ufs_flags[0])];
483 	char dev[4 + 1 + 7 + 1]; /* 12bit marjor + 20bit minor */
484 	char *name;
485 	mode_t type;
486 	dev_t rdev;
487 
488 	KGETRET(VTOI(vp), &inode, sizeof(struct inode), "vnode's inode");
489 	KGETRET(ip->i_ump, &ump, sizeof(struct ufsmount),
490 	    "vnode's mount point");
491 
492 	if (ump.um_fstype == UFS1) {
493 		KGETRET(ip->i_din.ffs1_din, &dip, sizeof (struct ufs1_dinode),
494 		    "inode's dinode");
495 		rdev = (uint32_t)dip.dp1.di_rdev;
496 	} else {
497 		KGETRET(ip->i_din.ffs2_din, &dip, sizeof (struct ufs2_dinode),
498 		    "inode's UFS2 dinode");
499 		rdev = dip.dp2.di_rdev;
500 	}
501 
502 	/*
503 	 * XXX need to to locking state.
504 	 */
505 
506 	(void)getflags(ufs_flags, flags, ip->i_flag);
507 	PRWORD(ovflw, " %*llu", 7, 1, (unsigned long long)ip->i_number);
508 	PRWORD(ovflw, " %*s", 6, 1, flags);
509 	type = ip->i_mode & S_IFMT;
510 	if (S_ISCHR(ip->i_mode) || S_ISBLK(ip->i_mode)) {
511 		if (usenumflag ||
512 		    (name = devname(rdev, type)) == NULL) {
513 			devprintf(dev, sizeof(dev), rdev);
514 			name = dev;
515 		}
516 		PRWORD(ovflw, " %*s", 8, 1, name);
517 	} else
518 		PRWORD(ovflw, " %*lld", 8, 1, (long long)ip->i_size);
519 	return 0;
520 }
521 
522 int
523 ext2fs_print(struct vnode *vp, int ovflw)
524 {
525 	struct inode inode, *ip = &inode;
526 	struct ext2fs_dinode dip;
527 	char flags[sizeof(ufs_flags) / sizeof(ufs_flags[0])];
528 	char dev[4 + 1 + 7 + 1]; /* 12bit marjor + 20bit minor */
529 	char *name;
530 	mode_t type;
531 
532 	KGETRET(VTOI(vp), &inode, sizeof(struct inode), "vnode's inode");
533 	KGETRET(ip->i_din.e2fs_din, &dip, sizeof (struct ext2fs_dinode),
534 	    "inode's dinode");
535 
536 	/*
537 	 * XXX need to to locking state.
538 	 */
539 
540 	(void)getflags(ufs_flags, flags, ip->i_flag);
541 	PRWORD(ovflw, " %*llu", 7, 1, (unsigned long long)ip->i_number);
542 	PRWORD(ovflw, " %*s", 6, 1, flags);
543 	type = dip.e2di_mode & S_IFMT;
544 	if (S_ISCHR(dip.e2di_mode) || S_ISBLK(dip.e2di_mode)) {
545 		if (usenumflag ||
546 		    (name = devname(dip.e2di_rdev, type)) == NULL) {
547 			devprintf(dev, sizeof(dev), dip.e2di_rdev);
548 			name = dev;
549 		}
550 		PRWORD(ovflw, " %*s", 8, 1, name);
551 	} else
552 		PRWORD(ovflw, " %*u", 8, 1, (u_int)dip.e2di_size);
553 	return (0);
554 }
555 
556 const struct flagbit_desc nfs_flags[] = {
557 	{ NFLUSHWANT,	'W' },
558 	{ NFLUSHINPROG,	'P' },
559 	{ NMODIFIED,	'M' },
560 	{ NWRITEERR,	'E' },
561 	{ NACC,		'A' },
562 	{ NUPD,		'U' },
563 	{ NCHG,		'C' },
564 	{ 0,		'\0' },
565 };
566 
567 void
568 nfs_header(void)
569 {
570 
571 	(void)printf(" FILEID NFLAG RDEV|SZ");
572 }
573 
574 int
575 nfs_print(struct vnode *vp, int ovflw)
576 {
577 	struct nfsnode nfsnode, *np = &nfsnode;
578 	char flags[sizeof(nfs_flags) / sizeof(nfs_flags[0])];
579 	char dev[4 + 1 + 7 + 1]; /* 12bit marjor + 20bit minor */
580 	struct vattr va;
581 	char *name;
582 	mode_t type;
583 
584 	KGETRET(VTONFS(vp), &nfsnode, sizeof(nfsnode), "vnode's nfsnode");
585 	(void)getflags(nfs_flags, flags, np->n_flag);
586 
587 	KGETRET(np->n_vattr, &va, sizeof(va), "vnode attr");
588 	PRWORD(ovflw, " %*ld", 7, 1, (long)va.va_fileid);
589 	PRWORD(ovflw, " %*s", 6, 1, flags);
590 	switch (va.va_type) {
591 	case VCHR:
592 		type = S_IFCHR;
593 		goto device;
594 
595 	case VBLK:
596 		type = S_IFBLK;
597 	device:
598 		if (usenumflag || (name = devname(va.va_rdev, type)) == NULL) {
599 			devprintf(dev, sizeof(dev), va.va_rdev);
600 			name = dev;
601 		}
602 		PRWORD(ovflw, " %*s", 8, 1, name);
603 		break;
604 	default:
605 		PRWORD(ovflw, " %*lld", 8, 1, (long long)np->n_size);
606 		break;
607 	}
608 	return (0);
609 }
610 
611 void
612 layer_header(void)
613 {
614 
615 	(void)printf(" %*s", PTRSTRWIDTH, "LOWER");
616 }
617 
618 int
619 layer_print(struct vnode *vp, int ovflw)
620 {
621 	struct layer_node lnode, *lp = &lnode;
622 
623 	KGETRET(VTOLAYER(vp), &lnode, sizeof(lnode), "layer vnode");
624 
625 	PRWORD(ovflw, " %*lx", PTRSTRWIDTH + 1, 1, (long)lp->layer_lowervp);
626 	return (0);
627 }
628 
629 /*
630  * Given a pointer to a mount structure in kernel space,
631  * read it in and return a usable pointer to it.
632  */
633 struct mount *
634 getmnt(struct mount *maddr)
635 {
636 	static struct mtab {
637 		struct mtab *next;
638 		struct mount *maddr;
639 		struct mount mount;
640 	} *mhead = NULL;
641 	struct mtab *mt;
642 	struct mount mb;
643 
644 	for (mt = mhead; mt != NULL; mt = mt->next)
645 		if (maddr == mt->maddr)
646 			return (&mt->mount);
647 	KGETRET(maddr, &mb, sizeof(struct mount), "mount table");
648 	if ((mt = malloc(sizeof(struct mtab))) == NULL)
649 		err(1, "malloc");
650 	mt->mount = mb;
651 	mt->maddr = maddr;
652 	mt->next = mhead;
653 	mhead = mt;
654 	return (&mt->mount);
655 }
656 
657 void
658 mount_print(struct mount *mp)
659 {
660 	int flags;
661 
662 	(void)printf("*** MOUNT %s %s on %s", ST.f_fstypename,
663 	    ST.f_mntfromname, ST.f_mntonname);
664 	if ((flags = mp->mnt_flag) != 0) {
665 		size_t i;
666 		const char *sep = " (";
667 
668 		for (i = 0; i < sizeof mnt_flags / sizeof mnt_flags[0]; i++) {
669 			if (flags & mnt_flags[i].m_flag) {
670 				(void)printf("%s%s", sep, mnt_flags[i].m_name);
671 				flags &= ~mnt_flags[i].m_flag;
672 				sep = ",";
673 			}
674 		}
675 		if (flags)
676 			(void)printf("%sunknown_flags:%x", sep, flags);
677 		(void)printf(")");
678 	}
679 	(void)printf("\n");
680 }
681 
682 char *
683 loadvnodes(int *avnodes)
684 {
685 	int mib[2];
686 	int status;
687 	size_t copysize;
688 #if 0
689 	size_t oldsize;
690 #endif
691 	char *vnodebase;
692 
693 	if (totalflag) {
694 		KGET(V_NUMV, *avnodes);
695 		return NULL;
696 	}
697 	if (memf != NULL) {
698 		/*
699 		 * do it by hand
700 		 */
701 		return (kinfo_vnodes(avnodes));
702 	}
703 	mib[0] = CTL_KERN;
704 	mib[1] = KERN_VNODE;
705 	/*
706 	 * First sysctl call gets the necessary buffer size; second
707 	 * sysctl call gets the data.  We allow for some growth in the
708 	 * data size between the two sysctl calls (increases of a few
709 	 * thousand vnodes in between the two calls have been observed).
710 	 * We ignore ENOMEM from the second sysctl call, which can
711 	 * happen if the kernel's data grew by even more than we allowed
712 	 * for.
713 	 */
714 	if (sysctl(mib, 2, NULL, &copysize, NULL, 0) == -1)
715 		err(1, "sysctl: KERN_VNODE");
716 #if 0
717 	oldsize = copysize;
718 #endif
719 	copysize += 100 * sizeof(struct vnode) + copysize / 20;
720 	if ((vnodebase = malloc(copysize)) == NULL)
721 		err(1, "malloc");
722 	status = sysctl(mib, 2, vnodebase, &copysize, NULL, 0);
723 	if (status == -1 && errno != ENOMEM)
724 		err(1, "sysctl: KERN_VNODE");
725 #if 0 /* for debugging the amount of growth allowed for */
726 	if (copysize != oldsize) {
727 		warnx("count changed from %ld to %ld (%+ld)%s",
728 		    (long)(oldsize / sizeof(struct vnode)),
729 		    (long)(copysize / sizeof(struct vnode)),
730 		    (long)(copysize / sizeof(struct vnode)) -
731 			(long)(oldsize / sizeof(struct vnode)),
732 		    (status == 0 ? "" : ", and errno = ENOMEM"));
733 	}
734 #endif
735 	if (copysize % (VPTRSZ + VNODESZ))
736 		errx(1, "vnode size mismatch");
737 	*avnodes = copysize / (VPTRSZ + VNODESZ);
738 
739 	return (vnodebase);
740 }
741 
742 /*
743  * simulate what a running kernel does in in kinfo_vnode
744  */
745 static int
746 vnode_cmp(const void *p1, const void *p2)
747 {
748 	const char *s1 = (const char *)p1;
749 	const char *s2 = (const char *)p2;
750 	const struct vnode *v1 = (const struct vnode *)(s1 + VPTRSZ);
751 	const struct vnode *v2 = (const struct vnode *)(s2 + VPTRSZ);
752 
753 	return (v2->v_mount - v1->v_mount);
754 }
755 
756 char *
757 kinfo_vnodes(int *avnodes)
758 {
759 	int i;
760 	char *beg, *bp, *ep;
761 	int numvnodes, next_offset;
762 
763 	KGET(V_NUMV, numvnodes);
764 	if ((bp = malloc((numvnodes + 20) * (VPTRSZ + VNODESZ))) == NULL)
765 		err(1, "malloc");
766 	beg = bp;
767 	ep = bp + (numvnodes + 20) * (VPTRSZ + VNODESZ);
768 	KGET(V_NEXT_OFFSET, next_offset);
769 
770 	for (i = 0; i < 3; i++) {
771 		TAILQ_HEAD(vnodelst, vnode) lru_head;
772 		struct vnode *vp, vnode;
773 
774 		KGET2((nl[V_LRU_LIST].n_value + sizeof(lru_head) * i), &lru_head,
775 		    sizeof(lru_head), "lru_list");
776 		vp = TAILQ_FIRST(&lru_head);
777 		while (vp != NULL) {
778 			KGET2(vp, &vnode, sizeof(vnode), "vnode");
779 			if (bp + VPTRSZ + VNODESZ > ep)
780 				/* XXX - should realloc */
781 				errx(1, "no more room for vnodes");
782 			memmove(bp, &vp, VPTRSZ);
783 			bp += VPTRSZ;
784 			memmove(bp, &vnode, VNODESZ);
785 			bp += VNODESZ;
786 			KGET2((char *)vp + next_offset, &vp, sizeof(vp), "nvp");
787 		}
788 	}
789 	*avnodes = (bp - beg) / (VPTRSZ + VNODESZ);
790 	/* Sort by mount like we get it from sysctl. */
791 	qsort(beg, *avnodes, VPTRSZ + VNODESZ, vnode_cmp);
792 	return (beg);
793 }
794 
795 void
796 ttymode(void)
797 {
798 	int ntty;
799 	struct ttylist_head tty_head;
800 	struct tty *tp, tty;
801 
802 	KGET(TTY_NTTY, ntty);
803 	(void)printf("%d terminal device%s\n", ntty, ntty == 1 ? "" : "s");
804 	KGET(TTY_TTYLIST, tty_head);
805 	(void)printf(
806 	    "  LINE RAW CAN OUT  HWT LWT     COL STATE  %-*s  PGID DISC\n",
807 	    PTRSTRWIDTH, "SESS");
808 	for (tp = tty_head.tqh_first; tp; tp = tty.tty_link.tqe_next) {
809 		KGET2(tp, &tty, sizeof tty, "tty struct");
810 		ttyprt(&tty);
811 	}
812 }
813 
814 static const struct flagbit_desc ttystates[] = {
815 	{ TS_ISOPEN,	'O'},
816 	{ TS_DIALOUT,	'>'},
817 	{ TS_CARR_ON,	'C'},
818 	{ TS_TIMEOUT,	'T'},
819 	{ TS_FLUSH,	'F'},
820 	{ TS_BUSY,	'B'},
821 	{ TS_XCLUDE,	'X'},
822 	{ TS_TTSTOP,	'S'},
823 	{ TS_TBLOCK,	'K'},
824 	{ TS_ASYNC,	'Y'},
825 	{ TS_BKSL,	'D'},
826 	{ TS_ERASE,	'E'},
827 	{ TS_LNCH,	'L'},
828 	{ TS_TYPEN,	'P'},
829 	{ TS_CNTTB,	'N'},
830 	{ 0,		'\0'},
831 };
832 
833 void
834 ttyprt(struct tty *tp)
835 {
836 	char state[sizeof(ttystates) / sizeof(ttystates[0]) + 1];
837 	char dev[2 + 3 + 1 + 5 + 1]; /* 12bit major + 20bit minor */
838 	struct linesw t_linesw;
839 	const char *name;
840 	char buffer;
841 	pid_t pgid;
842 	int n, ovflw;
843 
844 	if (usenumflag || (name = devname(tp->t_dev, S_IFCHR)) == NULL) {
845 		devprintf(dev, sizeof(dev), tp->t_dev);
846 		name = dev;
847 	}
848 	ovflw = 0;
849 	PRWORD(ovflw, "%-*s", 7, 0, name);
850 	PRWORD(ovflw, " %*d", 3, 1, tp->t_rawq.c_cc);
851 	PRWORD(ovflw, " %*d", 4, 1, tp->t_canq.c_cc);
852 	PRWORD(ovflw, " %*d", 4, 1, tp->t_outq.c_cc);
853 	PRWORD(ovflw, " %*d", 5, 1, tp->t_hiwat);
854 	PRWORD(ovflw, " %*d", 4, 1, tp->t_lowat);
855 	PRWORD(ovflw, " %*d", 8, 1, tp->t_column);
856 	n = getflags(ttystates, state, tp->t_state);
857 	if (tp->t_wopen) {
858 		state[n++] = 'W';
859 		state[n] = '\0';
860 	}
861 	PRWORD(ovflw, " %-*s", 7, 1, state);
862 	PRWORD(ovflw, " %*lX", PTRSTRWIDTH + 1, 1, (u_long)tp->t_session);
863 	pgid = 0;
864 	if (tp->t_pgrp != NULL)
865 		KGET2(&tp->t_pgrp->pg_id, &pgid, sizeof(pid_t), "pgid");
866 	PRWORD(ovflw, " %*d", 6, 1, pgid);
867 	KGET2(tp->t_linesw, &t_linesw, sizeof(t_linesw),
868 	    "line discipline switch table");
869 	name = t_linesw.l_name;
870 	(void)putchar(' ');
871 	for (;;) {
872 		KGET2(name, &buffer, sizeof(buffer), "line discipline name");
873 		if (buffer == '\0')
874 			break;
875 		(void)putchar(buffer);
876 		name++;
877 	}
878 	(void)putchar('\n');
879 }
880 
881 static const struct flagbit_desc filemode_flags[] = {
882 	{ FREAD,	'R' },
883 	{ FWRITE,	'W' },
884 	{ FAPPEND,	'A' },
885 #ifdef FSHLOCK	/* currently gone */
886 	{ FSHLOCK,	'S' },
887 	{ FEXLOCK,	'X' },
888 #endif
889 	{ FASYNC,	'I' },
890 	{ 0,		'\0' },
891 };
892 
893 void
894 filemode(void)
895 {
896 	struct kinfo_file *ki;
897 	char flags[sizeof(filemode_flags) / sizeof(filemode_flags[0])];
898 	char *buf, *offset;
899 	int len, maxfile, nfile, ovflw;
900 
901 	KGET(FNL_MAXFILE, maxfile);
902 	if (totalflag) {
903 		KGET(FNL_NFILE, nfile);
904 		(void)printf("%3d/%3d files\n", nfile, maxfile);
905 		return;
906 	}
907 	if (getfiles(&buf, &len, &offset) == -1)
908 		return;
909 	/*
910 	 * Getfiles returns in malloc'd memory to an array of kinfo_file2
911 	 * structures.
912 	 */
913 	nfile = len / sizeof(struct kinfo_file);
914 
915 	(void)printf("%d/%d open files\n", nfile, maxfile);
916 	(void)printf("%*s%s%*s TYPE    FLG     CNT  MSG  %*s%s%*s IFLG OFFSET\n",
917 	    (PTRSTRWIDTH - 4) / 2, "", " LOC", (PTRSTRWIDTH - 4) / 2, "",
918 	    (PTRSTRWIDTH - 4) / 2, "", "DATA", (PTRSTRWIDTH - 4) / 2, "");
919 	for (ki = (struct kinfo_file *)offset; nfile--; ki++) {
920 		if ((unsigned)ki->ki_ftype >= sizeof(dtypes) / sizeof(dtypes[0]))
921 			continue;
922 		ovflw = 0;
923 		(void)getflags(filemode_flags, flags, ki->ki_flag);
924 		PRWORD(ovflw, "%*lx", PTRSTRWIDTH, 0, (long)ki->ki_fileaddr);
925 		PRWORD(ovflw, " %-*s", 9, 1, dtypes[ki->ki_ftype]);
926 		PRWORD(ovflw, " %*s", 6, 1, flags);
927 		PRWORD(ovflw, " %*d", 5, 1, ki->ki_count);
928 		PRWORD(ovflw, " %*d", 5, 1, ki->ki_msgcount);
929 		PRWORD(ovflw, "  %*lx", PTRSTRWIDTH + 1, 2, (long)ki->ki_fdata);
930 		PRWORD(ovflw, " %*x", 5, 1, 0);
931 		if ((off_t)ki->ki_foffset < 0)
932 			PRWORD(ovflw, "  %-*lld\n", PTRSTRWIDTH + 1, 2,
933 			    (long long)ki->ki_foffset);
934 		else
935 			PRWORD(ovflw, "  %-*lld\n", PTRSTRWIDTH + 1, 2,
936 			    (long long)ki->ki_foffset);
937 	}
938 	free(buf);
939 }
940 
941 int
942 getfiles(char **abuf, int *alen, char **aoffset)
943 {
944 	size_t len;
945 	int mib[6];
946 	char *buf;
947 	size_t offset;
948 
949 	/*
950 	 * XXX
951 	 * Add emulation of KINFO_FILE here.
952 	 */
953 	if (memf != NULL)
954 		errx(1, "files on dead kernel, not implemented");
955 
956 	mib[0] = CTL_KERN;
957 	mib[1] = KERN_FILE2;
958 	mib[2] = KERN_FILE_BYFILE;
959 	mib[3] = 0;
960 	mib[4] = sizeof(struct kinfo_file);
961 	mib[5] = 0;
962 	if (sysctl(mib, 6, NULL, &len, NULL, 0) == -1) {
963 		warn("sysctl: KERN_FILE2");
964 		return (-1);
965 	}
966 	/* We need to align (struct kinfo_file *) in the buffer. */
967 	offset = len % sizeof(off_t);
968 	mib[5] = len / sizeof(struct kinfo_file);
969 	if ((buf = malloc(len + offset)) == NULL)
970 		err(1, "malloc");
971 	if (sysctl(mib, 6, buf + offset, &len, NULL, 0) == -1) {
972 		warn("sysctl: 2nd KERN_FILE2");
973 		free(buf);
974 		return (-1);
975 	}
976 	*abuf = buf;
977 	*alen = len;
978 	*aoffset = (buf + offset);
979 	return (0);
980 }
981 
982 void
983 usage(void)
984 {
985 
986 	(void)fprintf(stderr,
987 	    "usage: %s [-T|-f|-s|-t|-v] [-ghkmn] [-M core] [-N system]\n",
988 	    getprogname());
989 	exit(1);
990 }
991