pstat.c revision 178987
1/*-
2 * Copyright (c) 1980, 1991, 1993, 1994
3 *	The Regents of the University of California.  All rights reserved.
4 * Copyright (c) 2002 Networks Associates Technologies, Inc.
5 * All rights reserved.
6 *
7 * Portions of this software were developed for the FreeBSD Project by
8 * ThinkSec AS and NAI Labs, the Security Research Division of Network
9 * Associates, Inc.  under DARPA/SPAWAR contract N66001-01-C-8035
10 * ("CBOSS"), as part of the DARPA CHATS research program.
11 *
12 * Redistribution and use in source and binary forms, with or without
13 * modification, are permitted provided that the following conditions
14 * are met:
15 * 1. Redistributions of source code must retain the above copyright
16 *    notice, this list of conditions and the following disclaimer.
17 * 2. Redistributions in binary form must reproduce the above copyright
18 *    notice, this list of conditions and the following disclaimer in the
19 *    documentation and/or other materials provided with the distribution.
20 * 4. Neither the name of the University nor the names of its contributors
21 *    may be used to endorse or promote products derived from this software
22 *    without specific prior written permission.
23 *
24 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
25 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
28 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
29 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
30 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
31 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
32 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
33 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34 * SUCH DAMAGE.
35 */
36
37#if 0
38#ifndef lint
39static const char copyright[] =
40"@(#) Copyright (c) 1980, 1991, 1993, 1994\n\
41	The Regents of the University of California.  All rights reserved.\n";
42#endif /* not lint */
43
44#ifndef lint
45static char sccsid[] = "@(#)pstat.c	8.16 (Berkeley) 5/9/95";
46#endif /* not lint */
47#endif
48#include <sys/cdefs.h>
49__FBSDID("$FreeBSD: head/usr.sbin/pstat/pstat.c 178987 2008-05-14 00:22:57Z remko $");
50
51#include <sys/param.h>
52#include <sys/time.h>
53#include <sys/file.h>
54#include <sys/stat.h>
55#include <sys/stdint.h>
56#include <sys/ioctl.h>
57#include <sys/ioctl_compat.h>	/* XXX NTTYDISC is too well hidden */
58#include <sys/tty.h>
59#include <sys/blist.h>
60
61#include <sys/sysctl.h>
62#include <vm/vm_param.h>
63
64#include <err.h>
65#include <errno.h>
66#include <fcntl.h>
67#include <kvm.h>
68#include <libutil.h>
69#include <limits.h>
70#include <nlist.h>
71#include <stdio.h>
72#include <stdlib.h>
73#include <string.h>
74#include <unistd.h>
75
76enum {
77	NL_CONSTTY,
78	NL_MAXFILES,
79	NL_NFILES,
80	NL_TTY_LIST
81};
82
83static struct nlist nl[] = {
84	{ .n_name = "_constty" },
85	{ .n_name = "_maxfiles" },
86	{ .n_name = "_openfiles" },
87	{ .n_name = "_tty_list" },
88	{ .n_name = "" }
89};
90
91static int	humanflag;
92static int	usenumflag;
93static int	totalflag;
94static int	swapflag;
95static char	*nlistf;
96static char	*memf;
97static kvm_t	*kd;
98
99static char	*usagestr;
100
101static void	filemode(void);
102static int	getfiles(char **, size_t *);
103static void	swapmode(void);
104static void	ttymode(void);
105static void	ttyprt(struct xtty *);
106static void	usage(void);
107
108int
109main(int argc, char *argv[])
110{
111	int ch, i, quit, ret;
112	int fileflag, ttyflag;
113	char buf[_POSIX2_LINE_MAX],*opts;
114
115	fileflag = swapflag = ttyflag = 0;
116
117	/* We will behave like good old swapinfo if thus invoked */
118	opts = strrchr(argv[0], '/');
119	if (opts)
120		opts++;
121	else
122		opts = argv[0];
123	if (!strcmp(opts, "swapinfo")) {
124		swapflag = 1;
125		opts = "ghkmM:N:";
126		usagestr = "swapinfo [-ghkm] [-M core [-N system]]";
127	} else {
128		opts = "TM:N:fghkmnst";
129		usagestr = "pstat [-Tfghkmnst] [-M core [-N system]]";
130	}
131
132	while ((ch = getopt(argc, argv, opts)) != -1)
133		switch (ch) {
134		case 'f':
135			fileflag = 1;
136			break;
137		case 'g':
138			setenv("BLOCKSIZE", "1G", 1);
139			break;
140		case 'h':
141			humanflag = 1;
142			break;
143		case 'k':
144			setenv("BLOCKSIZE", "1K", 1);
145			break;
146		case 'm':
147			setenv("BLOCKSIZE", "1M", 1);
148			break;
149		case 'M':
150			memf = optarg;
151			break;
152		case 'N':
153			nlistf = optarg;
154			break;
155		case 'n':
156			usenumflag = 1;
157			break;
158		case 's':
159			++swapflag;
160			break;
161		case 'T':
162			totalflag = 1;
163			break;
164		case 't':
165			ttyflag = 1;
166			break;
167		default:
168			usage();
169		}
170	argc -= optind;
171	argv += optind;
172
173	if (memf != NULL) {
174		kd = kvm_openfiles(nlistf, memf, NULL, O_RDONLY, buf);
175		if (kd == NULL)
176			errx(1, "kvm_openfiles: %s", buf);
177		if ((ret = kvm_nlist(kd, nl)) != 0) {
178			if (ret == -1)
179				errx(1, "kvm_nlist: %s", kvm_geterr(kd));
180			quit = 0;
181			for (i = 0; nl[i].n_name[0] != '\0'; ++i)
182				if (nl[i].n_value == 0) {
183					quit = 1;
184					warnx("undefined symbol: %s",
185					    nl[i].n_name);
186				}
187			if (quit)
188				exit(1);
189		}
190	}
191	if (!(fileflag | ttyflag | swapflag | totalflag))
192		usage();
193	if (fileflag || totalflag)
194		filemode();
195	if (ttyflag)
196		ttymode();
197	if (swapflag || totalflag)
198		swapmode();
199	exit (0);
200}
201
202static void
203usage(void)
204{
205	fprintf(stderr, "usage: %s\n", usagestr);
206	exit (1);
207}
208
209static const char fhdr32[] =
210  "   LOC   TYPE   FLG  CNT MSG   DATA        OFFSET\n";
211/* c0000000 ------ RWAI 123 123 c0000000 1000000000000000 */
212
213static const char fhdr64[] =
214  "       LOC       TYPE   FLG  CNT MSG       DATA            OFFSET\n";
215/* c000000000000000 ------ RWAI 123 123 c000000000000000 1000000000000000 */
216
217static const char hdr[] =
218"  LINE RAW CAN OUT IHIWT ILOWT OHWT LWT     COL STATE  SESS      PGID DISC\n";
219
220static void
221ttymode_kvm(void)
222{
223	TAILQ_HEAD(, tty) tl;
224	struct tty *tp, tty;
225	struct xtty xt;
226
227	(void)printf("%s", hdr);
228	bzero(&xt, sizeof xt);
229	xt.xt_size = sizeof xt;
230	if (kvm_read(kd, nl[NL_TTY_LIST].n_value, &tl, sizeof tl) != sizeof tl)
231		errx(1, "kvm_read(): %s", kvm_geterr(kd));
232	tp = TAILQ_FIRST(&tl);
233	while (tp != NULL) {
234		if (kvm_read(kd, (u_long)tp, &tty, sizeof tty) != sizeof tty)
235			errx(1, "kvm_read(): %s", kvm_geterr(kd));
236		xt.xt_rawcc = tty.t_rawq.c_cc;
237		xt.xt_cancc = tty.t_canq.c_cc;
238		xt.xt_outcc = tty.t_outq.c_cc;
239#define XT_COPY(field) xt.xt_##field = tty.t_##field
240		XT_COPY(line);
241		XT_COPY(state);
242		XT_COPY(column);
243		XT_COPY(ihiwat);
244		XT_COPY(ilowat);
245		XT_COPY(ohiwat);
246		XT_COPY(olowat);
247#undef XT_COPY
248		ttyprt(&xt);
249		tp = TAILQ_NEXT(&tty, t_list);
250	}
251}
252
253static void
254ttymode_sysctl(void)
255{
256	struct xtty *xt, *end;
257	void *xttys;
258	size_t len;
259
260	(void)printf("%s", hdr);
261	if ((xttys = malloc(len = sizeof *xt)) == NULL)
262		err(1, "malloc()");
263	while (sysctlbyname("kern.ttys", xttys, &len, 0, 0) == -1) {
264		if (errno != ENOMEM)
265			err(1, "sysctlbyname()");
266		len *= 2;
267		if ((xttys = realloc(xttys, len)) == NULL)
268			err(1, "realloc()");
269	}
270	if (len > 0) {
271		end = (struct xtty *)((char *)xttys + len);
272		for (xt = xttys; xt < end; xt++)
273			ttyprt(xt);
274	}
275}
276
277static void
278ttymode(void)
279{
280
281	if (kd != NULL)
282		ttymode_kvm();
283	else
284		ttymode_sysctl();
285}
286
287static struct {
288	int flag;
289	char val;
290} ttystates[] = {
291#ifdef TS_WOPEN
292	{ TS_WOPEN,	'W'},
293#endif
294	{ TS_ISOPEN,	'O'},
295	{ TS_CARR_ON,	'C'},
296#ifdef TS_CONNECTED
297	{ TS_CONNECTED,	'c'},
298#endif
299	{ TS_TIMEOUT,	'T'},
300	{ TS_FLUSH,	'F'},
301	{ TS_BUSY,	'B'},
302#ifdef TS_ASLEEP
303	{ TS_ASLEEP,	'A'},
304#endif
305#ifdef TS_SO_OLOWAT
306	{ TS_SO_OLOWAT,	'A'},
307#endif
308#ifdef TS_SO_OCOMPLETE
309	{ TS_SO_OCOMPLETE, 'a'},
310#endif
311	{ TS_XCLUDE,	'X'},
312	{ TS_TTSTOP,	'S'},
313#ifdef TS_CAR_OFLOW
314	{ TS_CAR_OFLOW,	'm'},
315#endif
316#ifdef TS_CTS_OFLOW
317	{ TS_CTS_OFLOW,	'o'},
318#endif
319#ifdef TS_DSR_OFLOW
320	{ TS_DSR_OFLOW,	'd'},
321#endif
322	{ TS_TBLOCK,	'K'},
323	{ TS_ASYNC,	'Y'},
324	{ TS_BKSL,	'D'},
325	{ TS_ERASE,	'E'},
326	{ TS_LNCH,	'L'},
327	{ TS_TYPEN,	'P'},
328	{ TS_CNTTB,	'N'},
329#ifdef TS_CAN_BYPASS_L_RINT
330	{ TS_CAN_BYPASS_L_RINT, 'l'},
331#endif
332#ifdef TS_SNOOP
333	{ TS_SNOOP,     's'},
334#endif
335#ifdef TS_ZOMBIE
336	{ TS_ZOMBIE,	'Z'},
337#endif
338	{ 0,	       '\0'},
339};
340
341static void
342ttyprt(struct xtty *xt)
343{
344	int i, j;
345	char *name, state[20];
346
347	if (xt->xt_size != sizeof *xt)
348		errx(1, "struct xtty size mismatch");
349	if (usenumflag || xt->xt_dev == 0 ||
350	   (name = devname(xt->xt_dev, S_IFCHR)) == NULL)
351		printf("   %2d,%-2d", major(xt->xt_dev), minor(xt->xt_dev));
352	else
353		(void)printf("%7s ", name);
354	(void)printf("%2ld %3ld ", xt->xt_rawcc, xt->xt_cancc);
355	(void)printf("%3ld %5d %5d %4d %3d %7d ", xt->xt_outcc,
356		xt->xt_ihiwat, xt->xt_ilowat, xt->xt_ohiwat, xt->xt_olowat,
357		xt->xt_column);
358	for (i = j = 0; ttystates[i].flag; i++)
359		if (xt->xt_state & ttystates[i].flag)
360			state[j++] = ttystates[i].val;
361	if (j == 0)
362		state[j++] = '-';
363	state[j] = '\0';
364	(void)printf("%-6s %8d", state, xt->xt_sid);
365	(void)printf("%6d ", xt->xt_pgid);
366	switch (xt->xt_line) {
367	case TTYDISC:
368		(void)printf("term\n");
369		break;
370	case NTTYDISC:
371		(void)printf("ntty\n");
372		break;
373	case SLIPDISC:
374		(void)printf("slip\n");
375		break;
376	case PPPDISC:
377		(void)printf("ppp\n");
378		break;
379	default:
380		(void)printf("%d\n", xt->xt_line);
381		break;
382	}
383}
384
385static void
386filemode(void)
387{
388	struct xfile *fp;
389	char *buf, flagbuf[16], *fbp;
390	int maxf, openf;
391	size_t len;
392	static char *dtypes[] = { "???", "inode", "socket", "pipe",
393	    "fifo", "kqueue", "crypto" };
394	int i;
395	int wid;
396
397	if (kd != NULL) {
398		if (kvm_read(kd, nl[NL_MAXFILES].n_value,
399			&maxf, sizeof maxf) != sizeof maxf ||
400		    kvm_read(kd, nl[NL_NFILES].n_value,
401			&openf, sizeof openf) != sizeof openf)
402			errx(1, "kvm_read(): %s", kvm_geterr(kd));
403	} else {
404		len = sizeof(int);
405		if (sysctlbyname("kern.maxfiles", &maxf, &len, 0, 0) == -1 ||
406		    sysctlbyname("kern.openfiles", &openf, &len, 0, 0) == -1)
407			err(1, "sysctlbyname()");
408	}
409
410	if (totalflag) {
411		(void)printf("%3d/%3d files\n", openf, maxf);
412		return;
413	}
414	if (getfiles(&buf, &len) == -1)
415		return;
416	openf = len / sizeof *fp;
417
418	(void)printf("%d/%d open files\n", openf, maxf);
419	printf(sizeof(uintptr_t) == 4 ? fhdr32 : fhdr64);
420	wid = (int)sizeof(uintptr_t) * 2;
421	for (fp = (struct xfile *)buf, i = 0; i < openf; ++fp, ++i) {
422		if ((size_t)fp->xf_type >= sizeof(dtypes) / sizeof(dtypes[0]))
423			continue;
424		(void)printf("%*jx", wid, (uintmax_t)(uintptr_t)fp->xf_file);
425		(void)printf(" %-6.6s", dtypes[fp->xf_type]);
426		fbp = flagbuf;
427		if (fp->xf_flag & FREAD)
428			*fbp++ = 'R';
429		if (fp->xf_flag & FWRITE)
430			*fbp++ = 'W';
431		if (fp->xf_flag & FAPPEND)
432			*fbp++ = 'A';
433		if (fp->xf_flag & FASYNC)
434			*fbp++ = 'I';
435		*fbp = '\0';
436		(void)printf(" %4s %3d", flagbuf, fp->xf_count);
437		(void)printf(" %3d", fp->xf_msgcount);
438		(void)printf(" %*jx", wid, (uintmax_t)(uintptr_t)fp->xf_data);
439		(void)printf(" %*jx\n", (int)sizeof(fp->xf_offset) * 2,
440		    (uintmax_t)fp->xf_offset);
441	}
442	free(buf);
443}
444
445static int
446getfiles(char **abuf, size_t *alen)
447{
448	size_t len;
449	int mib[2];
450	char *buf;
451
452	/*
453	 * XXX
454	 * Add emulation of KINFO_FILE here.
455	 */
456	if (kd != NULL)
457		errx(1, "files on dead kernel, not implemented");
458
459	mib[0] = CTL_KERN;
460	mib[1] = KERN_FILE;
461	if (sysctl(mib, 2, NULL, &len, NULL, 0) == -1) {
462		warn("sysctl: KERN_FILE");
463		return (-1);
464	}
465	if ((buf = malloc(len)) == NULL)
466		errx(1, "malloc");
467	if (sysctl(mib, 2, buf, &len, NULL, 0) == -1) {
468		warn("sysctl: KERN_FILE");
469		return (-1);
470	}
471	*abuf = buf;
472	*alen = len;
473	return (0);
474}
475
476/*
477 * swapmode is based on a program called swapinfo written
478 * by Kevin Lahey <kml@rokkaku.atl.ga.us>.
479 */
480
481#define CONVERT(v)	((int64_t)(v) * pagesize / blocksize)
482static struct kvm_swap swtot;
483static int nswdev;
484
485static void
486print_swap_header(void)
487{
488	int hlen;
489	long blocksize;
490	const char *header;
491
492	header = getbsize(&hlen, &blocksize);
493	if (totalflag == 0)
494		(void)printf("%-15s %*s %8s %8s %8s\n",
495		    "Device", hlen, header,
496		    "Used", "Avail", "Capacity");
497}
498
499static void
500print_swap_line(const char *devname, intmax_t nblks, intmax_t bused,
501    intmax_t bavail, float bpercent)
502{
503	char usedbuf[5];
504	char availbuf[5];
505	int hlen, pagesize;
506	long blocksize;
507
508	pagesize = getpagesize();
509	getbsize(&hlen, &blocksize);
510
511	printf("%-15s %*jd ", devname, hlen, CONVERT(nblks));
512	if (humanflag) {
513		humanize_number(usedbuf, sizeof(usedbuf),
514		    CONVERT(blocksize * bused), "",
515		    HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL);
516		humanize_number(availbuf, sizeof(availbuf),
517		    CONVERT(blocksize * bavail), "",
518		    HN_AUTOSCALE, HN_B | HN_NOSPACE | HN_DECIMAL);
519		printf("%8s %8s %5.0f%%\n", usedbuf, availbuf, bpercent);
520	} else {
521		printf("%8jd %8jd %5.0f%%\n", (intmax_t)CONVERT(bused),
522		    (intmax_t)CONVERT(bavail), bpercent);
523	}
524}
525
526static void
527print_swap(struct kvm_swap *ksw)
528{
529
530	swtot.ksw_total += ksw->ksw_total;
531	swtot.ksw_used += ksw->ksw_used;
532	++nswdev;
533	if (totalflag == 0)
534		print_swap_line(ksw->ksw_devname, ksw->ksw_total,
535		    ksw->ksw_used, ksw->ksw_total - ksw->ksw_used,
536		    (ksw->ksw_used * 100.0) / ksw->ksw_total);
537}
538
539static void
540print_swap_total(void)
541{
542	int hlen, pagesize;
543	long blocksize;
544
545	pagesize = getpagesize();
546	getbsize(&hlen, &blocksize);
547	if (totalflag) {
548		blocksize = 1024 * 1024;
549		(void)printf("%jdM/%jdM swap space\n",
550		    CONVERT(swtot.ksw_used), CONVERT(swtot.ksw_total));
551	} else if (nswdev > 1) {
552		print_swap_line("Total", swtot.ksw_total, swtot.ksw_used,
553		    swtot.ksw_total - swtot.ksw_used,
554		    (swtot.ksw_used * 100.0) / swtot.ksw_total);
555	}
556}
557
558static void
559swapmode_kvm(void)
560{
561	struct kvm_swap kswap[16];
562	int i, n;
563
564	n = kvm_getswapinfo(kd, kswap, sizeof kswap / sizeof kswap[0],
565	    SWIF_DEV_PREFIX);
566
567	print_swap_header();
568	for (i = 0; i < n; ++i)
569		print_swap(&kswap[i]);
570	print_swap_total();
571}
572
573static void
574swapmode_sysctl(void)
575{
576	struct kvm_swap ksw;
577	struct xswdev xsw;
578	size_t mibsize, size;
579	int mib[16], n;
580
581	print_swap_header();
582	mibsize = sizeof mib / sizeof mib[0];
583	if (sysctlnametomib("vm.swap_info", mib, &mibsize) == -1)
584		err(1, "sysctlnametomib()");
585	for (n = 0; ; ++n) {
586		mib[mibsize] = n;
587		size = sizeof xsw;
588		if (sysctl(mib, mibsize + 1, &xsw, &size, NULL, 0) == -1)
589			break;
590		if (xsw.xsw_version != XSWDEV_VERSION)
591			errx(1, "xswdev version mismatch");
592		if (xsw.xsw_dev == NODEV)
593			snprintf(ksw.ksw_devname, sizeof ksw.ksw_devname,
594			    "<NFSfile>");
595		else
596			snprintf(ksw.ksw_devname, sizeof ksw.ksw_devname,
597			    "/dev/%s", devname(xsw.xsw_dev, S_IFCHR));
598		ksw.ksw_used = xsw.xsw_used;
599		ksw.ksw_total = xsw.xsw_nblks;
600		ksw.ksw_flags = xsw.xsw_flags;
601		print_swap(&ksw);
602	}
603	if (errno != ENOENT)
604		err(1, "sysctl()");
605	print_swap_total();
606}
607
608static void
609swapmode(void)
610{
611	if (kd != NULL)
612		swapmode_kvm();
613	else
614		swapmode_sysctl();
615}
616