1/*-
2 * Copyright (c) 2003 Poul-Henning Kamp
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 * 3. The names of the authors may not be used to endorse or promote
14 *    products derived from this software without specific prior written
15 *    permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 *
29 * $FreeBSD$
30 */
31
32
33#include <sys/devicestat.h>
34#include <sys/mman.h>
35#include <sys/resource.h>
36#include <sys/time.h>
37
38#include <curses.h>
39#include <devstat.h>
40#include <err.h>
41#include <errno.h>
42#include <fcntl.h>
43#include <histedit.h>
44#include <libgeom.h>
45#include <paths.h>
46#include <regex.h>
47#include <stdint.h>
48#include <stdio.h>
49#include <stdlib.h>
50#include <string.h>
51#include <sysexits.h>
52#include <unistd.h>
53
54static int flag_a, flag_b, flag_c, flag_d;
55static int flag_I = 1000000;
56
57#define PRINTMSG(...) do {						\
58		if (flag_b && !loop)					\
59			printf(__VA_ARGS__);				\
60		else if (!flag_b)					\
61			printw(__VA_ARGS__);				\
62	} while(0)
63
64static void usage(void);
65
66static const char*
67el_prompt(void)
68{
69
70	return ("Filter: ");
71}
72
73int
74main(int argc, char **argv)
75{
76	int error, i, quit;
77	int curx, cury, maxx, maxy, line_len, loop, max_flen;
78	struct devstat *gsp, *gsq;
79	void *sp, *sq;
80	double dt;
81	struct timespec tp, tq;
82	struct gmesh gmp;
83	struct gprovider *pp;
84	struct gconsumer *cp;
85	struct gident *gid;
86	regex_t f_re, tmp_f_re;
87	short cf, cb;
88	char *p;
89	char f_s[100], pf_s[100], tmp_f_s[100];
90	const char *line;
91	long double ld[11];
92	uint64_t u64;
93	EditLine *el;
94	History *hist;
95	HistEvent hist_ev;
96
97	hist = NULL;
98	el = NULL;
99	maxx = -1;
100	curx = -1;
101	loop = 1;
102	/* Turn on batch mode if output is not tty. */
103	if (!isatty(fileno(stdout)))
104		flag_b = 1;
105
106	f_s[0] = '\0';
107	while ((i = getopt(argc, argv, "adcf:I:b")) != -1) {
108		switch (i) {
109		case 'a':
110			flag_a = 1;
111			break;
112		case 'b':
113			flag_b = 1;
114			break;
115		case 'c':
116			flag_c = 1;
117			break;
118		case 'd':
119			flag_d = 1;
120			break;
121		case 'f':
122			if (strlen(optarg) > sizeof(f_s) - 1)
123				errx(EX_USAGE, "Filter string too long");
124			if (regcomp(&f_re, optarg, REG_EXTENDED) != 0)
125				errx(EX_USAGE,
126				    "Invalid filter - see re_format(7)");
127			strncpy(f_s, optarg, sizeof(f_s));
128			break;
129		case 'I':
130			p = NULL;
131			i = strtoul(optarg, &p, 0);
132			if (p == optarg || errno == EINVAL ||
133			    errno == ERANGE) {
134				errx(1, "Invalid argument to -I");
135			} else if (!strcmp(p, "s"))
136				i *= 1000000;
137			else if (!strcmp(p, "ms"))
138				i *= 1000;
139			else if (!strcmp(p, "us"))
140				i *= 1;
141			flag_I = i;
142			break;
143		case '?':
144		default:
145			usage();
146		}
147	}
148	argc -= optind;
149	argv += optind;
150	if (argc != 0)
151		usage();
152
153	i = geom_gettree(&gmp);
154	if (i != 0)
155		err(1, "geom_gettree = %d", i);
156	error = geom_stats_open();
157	if (error)
158		err(1, "geom_stats_open()");
159	sq = NULL;
160	sq = geom_stats_snapshot_get();
161	if (sq == NULL)
162		err(1, "geom_stats_snapshot()");
163	if (!flag_b) {
164		/* Setup curses */
165		initscr();
166		start_color();
167		use_default_colors();
168		pair_content(0, &cf, &cb);
169		init_pair(1, COLOR_GREEN, cb);
170		init_pair(2, COLOR_MAGENTA, cb);
171		init_pair(3, COLOR_RED, cb);
172		cbreak();
173		noecho();
174		nonl();
175		nodelay(stdscr, 1);
176		intrflush(stdscr, FALSE);
177		keypad(stdscr, TRUE);
178		/* Setup libedit */
179		hist = history_init();
180		if (hist == NULL)
181			err(EX_SOFTWARE, "history_init()");
182		history(hist, &hist_ev, H_SETSIZE, 100);
183		el = el_init("gstat", stdin, stdout, stderr);
184		if (el == NULL)
185			err(EX_SOFTWARE, "el_init");
186		el_set(el, EL_EDITOR, "emacs");
187		el_set(el, EL_SIGNAL, 1);
188		el_set(el, EL_HIST, history, hist);
189		el_set(el, EL_PROMPT, el_prompt);
190		if (f_s[0] != '\0')
191			history(hist, &hist_ev, H_ENTER, f_s);
192	}
193	geom_stats_snapshot_timestamp(sq, &tq);
194	for (quit = 0; !quit;) {
195		sp = geom_stats_snapshot_get();
196		if (sp == NULL)
197			err(1, "geom_stats_snapshot()");
198		geom_stats_snapshot_timestamp(sp, &tp);
199		dt = tp.tv_sec - tq.tv_sec;
200		dt += (tp.tv_nsec - tq.tv_nsec) * 1e-9;
201		tq = tp;
202
203		geom_stats_snapshot_reset(sp);
204		geom_stats_snapshot_reset(sq);
205		move(0,0);
206		PRINTMSG("dT: %5.3fs  w: %.3fs", dt, (float)flag_I / 1000000);
207		if (f_s[0] != '\0') {
208			PRINTMSG("  filter: ");
209			if (!flag_b) {
210				getyx(stdscr, cury, curx);
211				getmaxyx(stdscr, maxy, maxx);
212			}
213			strncpy(pf_s, f_s, sizeof(pf_s));
214			max_flen = maxx - curx - 1;
215			if ((int)strlen(f_s) > max_flen && max_flen >= 0) {
216				if (max_flen > 3)
217					pf_s[max_flen - 3] = '.';
218				if (max_flen > 2)
219					pf_s[max_flen - 2] = '.';
220				if (max_flen > 1)
221					pf_s[max_flen - 1] = '.';
222				pf_s[max_flen] = '\0';
223			}
224			PRINTMSG("%s", pf_s);
225		}
226		PRINTMSG("\n");
227		PRINTMSG(" L(q)  ops/s   ");
228		PRINTMSG(" r/s   kBps   ms/r   ");
229		PRINTMSG(" w/s   kBps   ms/w   ");
230		if (flag_d)
231			PRINTMSG(" d/s   kBps   ms/d   ");
232		PRINTMSG("%%busy Name\n");
233		for (;;) {
234			gsp = geom_stats_snapshot_next(sp);
235			gsq = geom_stats_snapshot_next(sq);
236			if (gsp == NULL || gsq == NULL)
237				break;
238			if (gsp->id == NULL)
239				continue;
240			gid = geom_lookupid(&gmp, gsp->id);
241			if (gid == NULL) {
242				geom_deletetree(&gmp);
243				i = geom_gettree(&gmp);
244				if (i != 0)
245					err(1, "geom_gettree = %d", i);
246				gid = geom_lookupid(&gmp, gsp->id);
247			}
248			if (gid == NULL)
249				continue;
250			if (gid->lg_what == ISCONSUMER && !flag_c)
251				continue;
252			/* Do not print past end of window */
253			if (!flag_b) {
254				getyx(stdscr, cury, curx);
255				if (curx > 0)
256					continue;
257			}
258			if ((gid->lg_what == ISPROVIDER
259			    || gid->lg_what == ISCONSUMER) && f_s[0] != '\0') {
260				pp = gid->lg_ptr;
261				if ((regexec(&f_re, pp->lg_name, 0, NULL, 0)
262				     != 0))
263				  continue;
264			}
265			if (gsp->sequence0 != gsp->sequence1) {
266				PRINTMSG("*\n");
267				continue;
268			}
269			devstat_compute_statistics(gsp, gsq, dt,
270			    DSM_QUEUE_LENGTH, &u64,
271			    DSM_TRANSFERS_PER_SECOND, &ld[0],
272
273			    DSM_TRANSFERS_PER_SECOND_READ, &ld[1],
274			    DSM_MB_PER_SECOND_READ, &ld[2],
275			    DSM_MS_PER_TRANSACTION_READ, &ld[3],
276
277			    DSM_TRANSFERS_PER_SECOND_WRITE, &ld[4],
278			    DSM_MB_PER_SECOND_WRITE, &ld[5],
279			    DSM_MS_PER_TRANSACTION_WRITE, &ld[6],
280
281			    DSM_BUSY_PCT, &ld[7],
282			    DSM_TRANSFERS_PER_SECOND_FREE, &ld[8],
283			    DSM_MB_PER_SECOND_FREE, &ld[9],
284			    DSM_MS_PER_TRANSACTION_FREE, &ld[10],
285			    DSM_NONE);
286
287			if (flag_a && ld[7] < 0.1) {
288				*gsq = *gsp;
289				continue;
290			}
291
292			PRINTMSG(" %4ju", (uintmax_t)u64);
293			PRINTMSG(" %6.0f", (double)ld[0]);
294			PRINTMSG(" %6.0f", (double)ld[1]);
295			PRINTMSG(" %6.0f", (double)ld[2] * 1024);
296			if (ld[3] > 1e3)
297				PRINTMSG(" %6.0f", (double)ld[3]);
298			else
299				PRINTMSG(" %6.1f", (double)ld[3]);
300			PRINTMSG(" %6.0f", (double)ld[4]);
301			PRINTMSG(" %6.0f", (double)ld[5] * 1024);
302			if (ld[6] > 1e3)
303				PRINTMSG(" %6.0f", (double)ld[6]);
304			else
305				PRINTMSG(" %6.1f", (double)ld[6]);
306
307			if (flag_d) {
308				PRINTMSG(" %6.0f", (double)ld[8]);
309				PRINTMSG(" %6.0f", (double)ld[9] * 1024);
310				if (ld[10] > 1e3)
311					PRINTMSG(" %6.0f", (double)ld[10]);
312				else
313					PRINTMSG(" %6.1f", (double)ld[10]);
314			}
315
316			if (ld[7] > 80)
317				i = 3;
318			else if (ld[7] > 50)
319				i = 2;
320			else
321				i = 1;
322			if (!flag_b)
323				attron(COLOR_PAIR(i));
324			PRINTMSG(" %6.1lf", (double)ld[7]);
325			if (!flag_b) {
326				attroff(COLOR_PAIR(i));
327				PRINTMSG("|");
328			} else
329				PRINTMSG(" ");
330			if (gid == NULL) {
331				PRINTMSG(" ??");
332			} else if (gid->lg_what == ISPROVIDER) {
333				pp = gid->lg_ptr;
334				PRINTMSG(" %s", pp->lg_name);
335			} else if (gid->lg_what == ISCONSUMER) {
336				cp = gid->lg_ptr;
337				PRINTMSG(" %s/%s/%s",
338				    cp->lg_geom->lg_class->lg_name,
339				    cp->lg_geom->lg_name,
340				    cp->lg_provider->lg_name);
341			}
342			if (!flag_b)
343				clrtoeol();
344			PRINTMSG("\n");
345			*gsq = *gsp;
346		}
347		geom_stats_snapshot_free(sp);
348		if (flag_b) {
349			/* We loop extra to make sure we get the information. */
350			if (!loop)
351				break;
352			loop = 0;
353			usleep(flag_I);
354			continue;
355		}
356		getyx(stdscr, cury, curx);
357		getmaxyx(stdscr, maxy, maxx);
358		clrtobot();
359		if (maxy - 1 <= cury)
360			move(maxy - 1, 0);
361		refresh();
362		usleep(flag_I);
363		while((i = getch()) != ERR) {
364			switch (i) {
365			case '>':
366				flag_I *= 2;
367				break;
368			case '<':
369				flag_I /= 2;
370				if (flag_I < 1000)
371					flag_I = 1000;
372				break;
373			case 'c':
374				flag_c = !flag_c;
375				break;
376			case 'f':
377				move(0,0);
378				clrtoeol();
379				refresh();
380				line = el_gets(el, &line_len);
381				if (line == NULL)
382					err(1, "el_gets");
383				if (line_len > 1)
384					history(hist, &hist_ev, H_ENTER, line);
385				strncpy(tmp_f_s, line, sizeof(f_s));
386				if ((p = strchr(tmp_f_s, '\n')) != NULL)
387					*p = '\0';
388				/*
389				 * We have to clear since we messed up
390				 * curses idea of the screen by using
391				 * libedit.
392				 */
393				clear();
394				refresh();
395				if (regcomp(&tmp_f_re, tmp_f_s, REG_EXTENDED)
396				    != 0) {
397					move(0, 0);
398					printw("Invalid filter");
399					refresh();
400					sleep(1);
401				} else {
402					strncpy(f_s, tmp_f_s, sizeof(f_s));
403					f_re = tmp_f_re;
404				}
405				break;
406			case 'F':
407				f_s[0] = '\0';
408				break;
409			case 'q':
410				quit = 1;
411				break;
412			default:
413				break;
414			}
415		}
416	}
417
418	if (!flag_b) {
419		endwin();
420		el_end(el);
421	}
422	exit(EX_OK);
423}
424
425static void
426usage(void)
427{
428	fprintf(stderr, "usage: gstat [-abcd] [-f filter] [-I interval]\n");
429	exit(EX_USAGE);
430        /* NOTREACHED */
431}
432