usr.sbin/gstat/gstat.c

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