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 54 static int flag_a, flag_b, flag_c, flag_d; 55 static 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 64 static void usage(void); 65 66 static const char* 67 el_prompt(void) 68 { 69 70 return ("Filter: "); 71 } 72 73 int 74 main(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 f_s[0] = '\0'; 103 while ((i = getopt(argc, argv, "adcf:I:b")) != -1) { 104 switch (i) { 105 case 'a': 106 flag_a = 1; 107 break; 108 case 'b': 109 flag_b = 1; 110 break; 111 case 'c': 112 flag_c = 1; 113 break; 114 case 'd': 115 flag_d = 1; 116 break; 117 case 'f': 118 if (strlen(optarg) > sizeof(f_s) - 1) 119 errx(EX_USAGE, "Filter string too long"); 120 if (regcomp(&f_re, optarg, REG_EXTENDED) != 0) 121 errx(EX_USAGE, 122 "Invalid filter - see re_format(7)"); 123 strncpy(f_s, optarg, sizeof(f_s)); 124 break; 125 case 'I': 126 p = NULL; 127 i = strtoul(optarg, &p, 0); 128 if (p == optarg || errno == EINVAL || 129 errno == ERANGE) { 130 errx(1, "Invalid argument to -I"); 131 } else if (!strcmp(p, "s")) 132 i *= 1000000; 133 else if (!strcmp(p, "ms")) 134 i *= 1000; 135 else if (!strcmp(p, "us")) 136 i *= 1; 137 flag_I = i; 138 break; 139 case '?': 140 default: 141 usage(); 142 } 143 } 144 argc -= optind; 145 argv += optind; 146 if (argc != 0) 147 usage(); 148 149 i = geom_gettree(&gmp); 150 if (i != 0) 151 err(1, "geom_gettree = %d", i); 152 error = geom_stats_open(); 153 if (error) 154 err(1, "geom_stats_open()"); 155 sq = NULL; 156 sq = geom_stats_snapshot_get(); 157 if (sq == NULL) 158 err(1, "geom_stats_snapshot()"); 159 if (!flag_b) { 160 /* Setup curses */ 161 initscr(); 162 start_color(); 163 use_default_colors(); 164 pair_content(0, &cf, &cb); 165 init_pair(1, COLOR_GREEN, cb); 166 init_pair(2, COLOR_MAGENTA, cb); 167 init_pair(3, COLOR_RED, cb); 168 cbreak(); 169 noecho(); 170 nonl(); 171 nodelay(stdscr, 1); 172 intrflush(stdscr, FALSE); 173 keypad(stdscr, TRUE); 174 /* Setup libedit */ 175 hist = history_init(); 176 if (hist == NULL) 177 err(EX_SOFTWARE, "history_init()"); 178 history(hist, &hist_ev, H_SETSIZE, 100); 179 el = el_init("gstat", stdin, stdout, stderr); 180 if (el == NULL) 181 err(EX_SOFTWARE, "el_init"); 182 el_set(el, EL_EDITOR, "emacs"); 183 el_set(el, EL_SIGNAL, 1); 184 el_set(el, EL_HIST, history, hist); 185 el_set(el, EL_PROMPT, el_prompt); 186 if (f_s[0] != '\0') 187 history(hist, &hist_ev, H_ENTER, f_s); 188 } 189 geom_stats_snapshot_timestamp(sq, &tq); 190 for (quit = 0; !quit;) { 191 sp = geom_stats_snapshot_get(); 192 if (sp == NULL) 193 err(1, "geom_stats_snapshot()"); 194 geom_stats_snapshot_timestamp(sp, &tp); 195 dt = tp.tv_sec - tq.tv_sec; 196 dt += (tp.tv_nsec - tq.tv_nsec) * 1e-9; 197 tq = tp; 198 199 geom_stats_snapshot_reset(sp); 200 geom_stats_snapshot_reset(sq); 201 move(0,0); 202 PRINTMSG("dT: %5.3fs w: %.3fs", dt, (float)flag_I / 1000000); 203 if (f_s[0] != '\0') { 204 PRINTMSG(" filter: "); 205 if (!flag_b) { 206 getyx(stdscr, cury, curx); 207 getmaxyx(stdscr, maxy, maxx); 208 } 209 strncpy(pf_s, f_s, sizeof(pf_s)); 210 max_flen = maxx - curx - 1; 211 if ((int)strlen(f_s) > max_flen && max_flen >= 0) { 212 if (max_flen > 3) 213 pf_s[max_flen - 3] = '.'; 214 if (max_flen > 2) 215 pf_s[max_flen - 2] = '.'; 216 if (max_flen > 1) 217 pf_s[max_flen - 1] = '.'; 218 pf_s[max_flen] = '\0'; 219 } 220 PRINTMSG("%s", pf_s); 221 } 222 PRINTMSG("\n"); 223 PRINTMSG(" L(q) ops/s "); 224 PRINTMSG(" r/s kBps ms/r "); 225 PRINTMSG(" w/s kBps ms/w "); 226 if (flag_d) 227 PRINTMSG(" d/s kBps ms/d "); 228 PRINTMSG("%%busy Name\n"); 229 for (;;) { 230 gsp = geom_stats_snapshot_next(sp); 231 gsq = geom_stats_snapshot_next(sq); 232 if (gsp == NULL || gsq == NULL) 233 break; 234 if (gsp->id == NULL) 235 continue; 236 gid = geom_lookupid(&gmp, gsp->id); 237 if (gid == NULL) { 238 geom_deletetree(&gmp); 239 i = geom_gettree(&gmp); 240 if (i != 0) 241 err(1, "geom_gettree = %d", i); 242 gid = geom_lookupid(&gmp, gsp->id); 243 } 244 if (gid == NULL) 245 continue; 246 if (gid->lg_what == ISCONSUMER && !flag_c) 247 continue; 248 /* Do not print past end of window */ 249 if (!flag_b) { 250 getyx(stdscr, cury, curx); 251 if (curx > 0) 252 continue; 253 } 254 if ((gid->lg_what == ISPROVIDER 255 || gid->lg_what == ISCONSUMER) && f_s[0] != '\0') { 256 pp = gid->lg_ptr; 257 if ((regexec(&f_re, pp->lg_name, 0, NULL, 0) 258 != 0)) 259 continue; 260 } 261 if (gsp->sequence0 != gsp->sequence1) { 262 PRINTMSG("*\n"); 263 continue; 264 } 265 devstat_compute_statistics(gsp, gsq, dt, 266 DSM_QUEUE_LENGTH, &u64, 267 DSM_TRANSFERS_PER_SECOND, &ld[0], 268 269 DSM_TRANSFERS_PER_SECOND_READ, &ld[1], 270 DSM_MB_PER_SECOND_READ, &ld[2], 271 DSM_MS_PER_TRANSACTION_READ, &ld[3], 272 273 DSM_TRANSFERS_PER_SECOND_WRITE, &ld[4], 274 DSM_MB_PER_SECOND_WRITE, &ld[5], 275 DSM_MS_PER_TRANSACTION_WRITE, &ld[6], 276 277 DSM_BUSY_PCT, &ld[7], 278 DSM_TRANSFERS_PER_SECOND_FREE, &ld[8], 279 DSM_MB_PER_SECOND_FREE, &ld[9], 280 DSM_MS_PER_TRANSACTION_FREE, &ld[10], 281 DSM_NONE); 282 283 if (flag_a && ld[7] < 0.1) { 284 *gsq = *gsp; 285 continue; 286 } 287 288 PRINTMSG(" %4ju", (uintmax_t)u64); 289 PRINTMSG(" %6.0f", (double)ld[0]); 290 PRINTMSG(" %6.0f", (double)ld[1]); 291 PRINTMSG(" %6.0f", (double)ld[2] * 1024); 292 if (ld[3] > 1e3) 293 PRINTMSG(" %6.0f", (double)ld[3]); 294 else 295 PRINTMSG(" %6.1f", (double)ld[3]); 296 PRINTMSG(" %6.0f", (double)ld[4]); 297 PRINTMSG(" %6.0f", (double)ld[5] * 1024); 298 if (ld[6] > 1e3) 299 PRINTMSG(" %6.0f", (double)ld[6]); 300 else 301 PRINTMSG(" %6.1f", (double)ld[6]); 302 303 if (flag_d) { 304 PRINTMSG(" %6.0f", (double)ld[8]); 305 PRINTMSG(" %6.0f", (double)ld[9] * 1024); 306 if (ld[10] > 1e3) 307 PRINTMSG(" %6.0f", (double)ld[10]); 308 else 309 PRINTMSG(" %6.1f", (double)ld[10]); 310 } 311 312 if (ld[7] > 80) 313 i = 3; 314 else if (ld[7] > 50) 315 i = 2; 316 else 317 i = 1; 318 if (!flag_b) 319 attron(COLOR_PAIR(i)); 320 PRINTMSG(" %6.1lf", (double)ld[7]); 321 if (!flag_b) 322 attroff(COLOR_PAIR(i)); 323 PRINTMSG("|"); 324 if (gid == NULL) { 325 PRINTMSG(" ??"); 326 } else if (gid->lg_what == ISPROVIDER) { 327 pp = gid->lg_ptr; 328 PRINTMSG(" %s", pp->lg_name); 329 } else if (gid->lg_what == ISCONSUMER) { 330 cp = gid->lg_ptr; 331 PRINTMSG(" %s/%s/%s", 332 cp->lg_geom->lg_class->lg_name, 333 cp->lg_geom->lg_name, 334 cp->lg_provider->lg_name); 335 } 336 if (!flag_b) 337 clrtoeol(); 338 PRINTMSG("\n"); 339 *gsq = *gsp; 340 } 341 geom_stats_snapshot_free(sp); 342 if (flag_b) { 343 /* We loop extra to make sure we get the information. */ 344 if (!loop) 345 break; 346 loop = 0; 347 usleep(flag_I); 348 continue; 349 } 350 getyx(stdscr, cury, curx); 351 getmaxyx(stdscr, maxy, maxx); 352 clrtobot(); 353 if (maxy - 1 <= cury) 354 move(maxy - 1, 0); 355 refresh(); 356 usleep(flag_I); 357 while((i = getch()) != ERR) { 358 switch (i) { 359 case '>': 360 flag_I *= 2; 361 break; 362 case '<': 363 flag_I /= 2; 364 if (flag_I < 1000) 365 flag_I = 1000; 366 break; 367 case 'c': 368 flag_c = !flag_c; 369 break; 370 case 'f': 371 move(0,0); 372 clrtoeol(); 373 refresh(); 374 line = el_gets(el, &line_len); 375 if (line == NULL) 376 err(1, "el_gets"); 377 if (line_len > 1) 378 history(hist, &hist_ev, H_ENTER, line); 379 strncpy(tmp_f_s, line, sizeof(f_s)); 380 if ((p = strchr(tmp_f_s, '\n')) != NULL) 381 *p = '\0'; 382 /* 383 * We have to clear since we messed up 384 * curses idea of the screen by using 385 * libedit. 386 */ 387 clear(); 388 refresh(); 389 if (regcomp(&tmp_f_re, tmp_f_s, REG_EXTENDED) 390 != 0) { 391 move(0, 0); 392 printw("Invalid filter"); 393 refresh(); 394 sleep(1); 395 } else { 396 strncpy(f_s, tmp_f_s, sizeof(f_s)); 397 f_re = tmp_f_re; 398 } 399 break; 400 case 'F': 401 f_s[0] = '\0'; 402 break; 403 case 'q': 404 quit = 1; 405 break; 406 default: 407 break; 408 } 409 } 410 } 411 412 if (!flag_b) { 413 endwin(); 414 el_end(el); 415 } 416 exit(EX_OK); 417 } 418 419 static void 420 usage(void) 421 { 422 fprintf(stderr, "usage: gstat [-abcd] [-f filter] [-I interval]\n"); 423 exit(EX_USAGE); 424 /* NOTREACHED */ 425 } 426