iowatcher/main.c

   1 /*
   2  * Copyright (C) 2012 Fusion-io
   3  *
   4  *  This program is free software; you can redistribute it and/or
   5  *  modify it under the terms of the GNU General Public
   6  *  License v2 as published by the Free Software Foundation.
   7  *
   8  *  This program is distributed in the hope that it will be useful,
   9  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11  *  GNU General Public License for more details.
  12  *
  13  *  You should have received a copy of the GNU General Public License
  14  *  along with this program; if not, write to the Free Software
  15  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  16  *
  17  *  Parts of this file were imported from Jens Axboe's blktrace sources (also GPL)
  18  */
  19 #include <sys/types.h>
  20 #include <sys/stat.h>
  21 #include <fcntl.h>
  22 #include <unistd.h>
  23 #include <stdlib.h>
  24 #include <stdio.h>
  25 #include <math.h>
  26 #include <inttypes.h>
  27 #include <string.h>
  28 #include <asm/types.h>
  29 #include <errno.h>
  30 #include <sys/mman.h>
  31 #include <time.h>
  32 #include <math.h>
  33 #include <getopt.h>
  34
  35 #include "plot.h"
  36 #include "blkparse.h"
  37 #include "list.h"
  38 #include "tracers.h"
  39 #include "mpstat.h"
  40
  41 LIST_HEAD(all_traces);
  42
  43 static char line[1024];
  44 static int line_len = 1024;
  45 static int found_mpstat = 0;
  46 static int cpu_color_index = 0;
  47 static int color_index = 0;
  48 static int make_movie = 0;
  49 static int opt_graph_width = 0;
  50 static int opt_graph_height = 0;
  51
  52 char *colors[] = {
  53         "blue", "darkgreen",
  54         "red", "aqua",
  55         "orange", "darkviolet",
  56         "brown", "#00FF00",
  57         "yellow", "coral",
  58         "black", "darkred",
  59         "fuchsia", "crimson",
  60         NULL };
  61
  62 char *pick_color(void) {
  63         char *ret = colors[color_index];
  64         if (!ret) {
  65                 color_index = 0;
  66                 ret = colors[color_index];
  67         }
  68         color_index++;
  69         return ret;
  70 }
  71
  72 char *pick_cpu_color(void) {
  73         char *ret = colors[cpu_color_index];
  74         if (!ret) {
  75                 color_index = 0;
  76                 ret = colors[cpu_color_index];
  77         }
  78         cpu_color_index++;
  79         return ret;
  80 }
  81
  82 enum {
  83         IO_GRAPH_INDEX = 0,
  84         TPUT_GRAPH_INDEX,
  85         CPU_SYS_GRAPH_INDEX,
  86         CPU_IO_GRAPH_INDEX,
  87         CPU_IRQ_GRAPH_INDEX,
  88         CPU_SOFT_GRAPH_INDEX,
  89         CPU_USER_GRAPH_INDEX,
  90         LATENCY_GRAPH_INDEX,
  91         QUEUE_DEPTH_GRAPH_INDEX,
  92         IOPS_GRAPH_INDEX,
  93         TOTAL_GRAPHS
  94 };
  95
  96 enum {
  97         MPSTAT_SYS = 0,
  98         MPSTAT_IRQ,
  99         MPSTAT_IO,
 100         MPSTAT_SOFT,
 101         MPSTAT_USER,
 102         MPSTAT_GRAPHS
 103 };
 104
 105 static char *graphs_by_name[] = {
 106         "io",
 107         "tput",
 108         "cpu-sys",
 109         "cpu-io",
 110         "cpu-irq",
 111         "cpu-soft",
 112         "cpu-user",
 113         "latency",
 114         "queue-depth",
 115         "iops",
 116 };
 117
 118 enum {
 119         MOVIE_SPINDLE,
 120         MOVIE_RECT,
 121         NUM_MOVIE_STYLES,
 122 };
 123
 124 char *movie_styles[] = {
 125         "spindle",
 126         "rect",
 127         NULL
 128 };
 129
 130 static int movie_style = 0;
 131
 132 static int lookup_movie_style(char *str)
 133 {
 134         int i;
 135
 136         for (i = 0; i < NUM_MOVIE_STYLES; i++) {
 137                 if (strcmp(str, movie_styles[i]) == 0)
 138                         return i;
 139         }
 140         return -1;
 141 }
 142
 143 static int active_graphs[TOTAL_GRAPHS];
 144 static int last_active_graph = IOPS_GRAPH_INDEX;
 145
 146 static int label_index = 0;
 147 static int num_traces = 0;
 148 static int longest_label = 0;
 149
 150 struct trace_file {
 151         struct list_head list;
 152         char *filename;
 153         char *label;
 154         struct trace *trace;
 155         int seconds;
 156         int stop_seconds;
 157         u64 max_offset;
 158
 159         char *read_color;
 160         char *write_color;
 161
 162         struct graph_line_data *tput_gld;
 163         struct graph_line_data *iop_gld;
 164         struct graph_line_data *latency_gld;
 165         struct graph_line_data *queue_depth_gld;
 166         struct graph_dot_data *gdd_writes;
 167         struct graph_dot_data *gdd_reads;
 168
 169         int mpstat_seconds;
 170         int mpstat_stop_seconds;
 171         struct graph_line_data **mpstat_gld;
 172 };
 173
 174 static void alloc_mpstat_gld(struct trace_file *tf)
 175 {
 176         struct graph_line_data **ptr;
 177
 178         if (tf->trace->mpstat_num_cpus == 0)
 179                 return;
 180
 181         ptr = calloc((tf->trace->mpstat_num_cpus + 1) * MPSTAT_GRAPHS,
 182                      sizeof(struct graph_line_data *));
 183         if (!ptr) {
 184                 perror("Unable to allocate mpstat arrays\n");
 185                 exit(1);
 186         }
 187         tf->mpstat_gld = ptr;
 188 }
 189
 190 static void enable_all_graphs(void)
 191 {
 192         int i;
 193         for (i = 0; i < TOTAL_GRAPHS; i++)
 194                 active_graphs[i] = 1;
 195 }
 196
 197 static void disable_all_graphs(void)
 198 {
 199         int i;
 200         for (i = 0; i < TOTAL_GRAPHS; i++)
 201                 active_graphs[i] = 0;
 202 }
 203
 204 static int enable_one_graph(char *name)
 205 {
 206         int i;
 207         for (i = 0; i < TOTAL_GRAPHS; i++) {
 208                 if (strcmp(name, graphs_by_name[i]) == 0) {
 209                         active_graphs[i] = 1;
 210                         return 0;
 211                 }
 212         }
 213         return -ENOENT;
 214 }
 215
 216 static int disable_one_graph(char *name)
 217 {
 218         int i;
 219         for (i = 0; i < TOTAL_GRAPHS; i++) {
 220                 if (strcmp(name, graphs_by_name[i]) == 0) {
 221                         active_graphs[i] = 0;
 222                         return 0;
 223                 }
 224         }
 225         return -ENOENT;
 226 }
 227
 228 static int last_graph(void)
 229 {
 230         int i;
 231         for (i = TOTAL_GRAPHS - 1; i >= 0; i--) {
 232                 if (active_graphs[i]) {
 233                         return i;
 234                 }
 235         }
 236         return -ENOENT;
 237 }
 238 static void add_trace_file(char *filename)
 239 {
 240         struct trace_file *tf;
 241
 242         tf = calloc(1, sizeof(*tf));
 243         if (!tf) {
 244                 fprintf(stderr, "Unable to allocate memory\n");
 245                 exit(1);
 246         }
 247         tf->label = "";
 248         tf->filename = strdup(filename);
 249         list_add_tail(&tf->list, &all_traces);
 250         tf->read_color = pick_color();
 251         tf->write_color = pick_color();
 252         num_traces++;
 253 }
 254
 255 static void setup_trace_file_graphs(void)
 256 {
 257         struct trace_file *tf;
 258         int i;
 259
 260         list_for_each_entry(tf, &all_traces, list) {
 261                 tf->tput_gld = alloc_line_data(tf->seconds, tf->stop_seconds);
 262                 tf->latency_gld = alloc_line_data(tf->seconds, tf->stop_seconds);
 263                 tf->queue_depth_gld = alloc_line_data(tf->seconds, tf->stop_seconds);
 264                 tf->iop_gld = alloc_line_data(tf->seconds, tf->stop_seconds);
 265                 tf->gdd_writes = alloc_dot_data(tf->seconds, tf->max_offset, tf->stop_seconds);
 266                 tf->gdd_reads = alloc_dot_data(tf->seconds, tf->max_offset, tf->stop_seconds);
 267
 268                 if (tf->trace->mpstat_num_cpus == 0)
 269                         continue;
 270
 271                 alloc_mpstat_gld(tf);
 272                 for (i = 0; i < (tf->trace->mpstat_num_cpus + 1) * MPSTAT_GRAPHS; i++) {
 273                         tf->mpstat_gld[i] =
 274                                 alloc_line_data(tf->mpstat_seconds,
 275                                                 tf->mpstat_seconds);
 276                         tf->mpstat_gld[i]->max = 100;
 277                 }
 278         }
 279 }
 280
 281 static void read_traces(void)
 282 {
 283         struct trace_file *tf;
 284         struct trace *trace;
 285         u64 last_time;
 286         u64 ymin;
 287         u64 ymax;
 288         u64 max_bank;
 289         u64 max_bank_offset;
 290
 291         list_for_each_entry(tf, &all_traces, list) {
 292                 trace = open_trace(tf->filename);
 293                 if (!trace)
 294                         exit(1);
 295
 296                 last_time = find_last_time(trace);
 297                 tf->trace = trace;
 298                 tf->seconds = SECONDS(last_time);
 299                 tf->stop_seconds = SECONDS(last_time);
 300                 find_highest_offset(trace, &tf->max_offset, &max_bank,
 301                                     &max_bank_offset);
 302                 filter_outliers(trace, tf->max_offset, &ymin, &ymax);
 303                 tf->max_offset = ymax;
 304
 305                 read_mpstat(trace, tf->filename);
 306                 tf->mpstat_stop_seconds = trace->mpstat_seconds;
 307                 tf->mpstat_seconds = trace->mpstat_seconds;
 308                 if (tf->mpstat_seconds)
 309                         found_mpstat = 1;
 310         }
 311 }
 312
 313 static void read_trace_events(void)
 314 {
 315
 316         struct trace_file *tf;
 317         struct trace *trace;
 318         int ret;
 319         int i;
 320         int time;
 321         double user, sys, iowait, irq, soft;
 322         double max_user = 0, max_sys = 0, max_iowait = 0,
 323                max_irq = 0, max_soft = 0;
 324
 325         list_for_each_entry(tf, &all_traces, list) {
 326                 trace = tf->trace;
 327                 first_record(trace);
 328                 while (1) {
 329                         check_record(trace);
 330                         add_tput(trace, tf->tput_gld);
 331                         add_iop(trace, tf->iop_gld);
 332                         add_io(trace, tf->gdd_writes, tf->gdd_reads);
 333                         add_pending_io(trace, tf->queue_depth_gld);
 334                         add_completed_io(trace, tf->latency_gld);
 335                         ret = next_record(trace);
 336                         if (ret)
 337                                 break;
 338                 }
 339         }
 340         list_for_each_entry(tf, &all_traces, list) {
 341                 trace = tf->trace;
 342
 343                 if (trace->mpstat_num_cpus == 0)
 344                         continue;
 345
 346                 first_mpstat(trace);
 347
 348                 for (time = 0; time < tf->mpstat_stop_seconds; time ++) {
 349                         for (i = 0; i < (trace->mpstat_num_cpus + 1) * MPSTAT_GRAPHS; i += MPSTAT_GRAPHS) {
 350                                 ret = read_mpstat_event(trace, &user, &sys,
 351                                                         &iowait, &irq, &soft);
 352                                 if (ret)
 353                                         goto mpstat_done;
 354                                 if (next_mpstat_line(trace))
 355                                         goto mpstat_done;
 356
 357                                 if (sys > max_sys)
 358                                         max_sys = sys;
 359                                 if (user > max_user)
 360                                         max_user = user;
 361                                 if (irq > max_irq)
 362                                         max_irq = irq;
 363                                 if (iowait > max_iowait)
 364                                         max_iowait = iowait;
 365
 366                                 add_mpstat_gld(time, sys, tf->mpstat_gld[i + MPSTAT_SYS]);
 367                                 add_mpstat_gld(time, irq, tf->mpstat_gld[i + MPSTAT_IRQ]);
 368                                 add_mpstat_gld(time, soft, tf->mpstat_gld[i + MPSTAT_SOFT]);
 369                                 add_mpstat_gld(time, user, tf->mpstat_gld[i + MPSTAT_USER]);
 370                                 add_mpstat_gld(time, iowait, tf->mpstat_gld[i + MPSTAT_IO]);
 371                         }
 372                         if (next_mpstat(trace) == NULL)
 373                                 break;
 374                 }
 375         }
 376
 377 mpstat_done:
 378         list_for_each_entry(tf, &all_traces, list) {
 379                 trace = tf->trace;
 380
 381                 if (trace->mpstat_num_cpus == 0)
 382                         continue;
 383
 384                 tf->mpstat_gld[MPSTAT_SYS]->max = max_sys;
 385                 tf->mpstat_gld[MPSTAT_IRQ]->max = max_irq;
 386                 tf->mpstat_gld[MPSTAT_SOFT]->max = max_soft;
 387                 tf->mpstat_gld[MPSTAT_USER]->max = max_user;
 388                 tf->mpstat_gld[MPSTAT_IO]->max = max_iowait;;
 389         }
 390         return;
 391 }
 392
 393 static void set_trace_label(char *label)
 394 {
 395         int cur = 0;
 396         struct trace_file *tf;
 397         int len = strlen(label);
 398
 399         if (len > longest_label)
 400                 longest_label = len;
 401
 402         list_for_each_entry(tf, &all_traces, list) {
 403                 if (cur == label_index) {
 404                         tf->label = strdup(label);
 405                         label_index++;
 406                         break;
 407                 }
 408                 cur++;
 409         }
 410 }
 411
 412 static char *graph_title = "";
 413 static char *output_filename = "trace.svg";
 414 static char *blktrace_device = NULL;
 415 static char *blktrace_outfile = "trace";
 416 static char *blktrace_dest_dir = ".";
 417 static char *program_to_run = NULL;
 418
 419 static void set_blktrace_outfile(char *arg)
 420 {
 421         char *s = strdup(arg);
 422         char *last_dot = strrchr(s, '.');
 423
 424         if (last_dot) {
 425                 if (strcmp(last_dot, ".dump") == 0)
 426                         *last_dot = '\0';
 427         }
 428         blktrace_outfile = s;
 429 }
 430
 431
 432 static void compare_max_tf(struct trace_file *tf, int *seconds, u64 *max_offset)
 433 {
 434         if (tf->seconds > *seconds)
 435                 *seconds = tf->seconds;
 436         if (tf->max_offset > *max_offset)
 437                 *max_offset = tf->max_offset;
 438 }
 439
 440 static void set_all_max_tf(int seconds, u64 max_offset)
 441 {
 442         struct trace_file *tf;
 443
 444         list_for_each_entry(tf, &all_traces, list) {
 445                 tf->seconds = seconds;
 446                 tf->max_offset = max_offset;
 447         }
 448 }
 449
 450 static char *create_movie_temp_dir(void)
 451 {
 452         char *ret;
 453         char *pattern = strdup("io-movie-XXXXXX");;
 454
 455         ret = mkdtemp(pattern);
 456         if (!ret) {
 457                 perror("Unable to create temp directory for movie files");
 458                 exit(1);
 459         }
 460         return ret;
 461 }
 462
 463 static struct plot_history *alloc_plot_history(char *color)
 464 {
 465         struct plot_history *ph = calloc(1, sizeof(struct plot_history));
 466
 467         if (!ph) {
 468                 perror("memory allocation failed");
 469                 exit(1);
 470         }
 471         ph->history = calloc(4096, sizeof(double));
 472         if (!ph->history) {
 473                 perror("memory allocation failed");
 474                 exit(1);
 475         }
 476         ph->history_len = 4096;
 477         ph->color = color;
 478         return ph;
 479 }
 480
 481 LIST_HEAD(movie_history_writes);
 482 LIST_HEAD(movie_history_reads);
 483 int num_histories = 0;
 484
 485 static void add_history(struct plot_history *ph, struct list_head *list)
 486 {
 487         struct plot_history *entry;
 488
 489         list_add_tail(&ph->list, list);
 490         num_histories++;
 491
 492         if (num_histories > 12) {
 493                 num_histories--;
 494                 entry = list_entry(list->next, struct plot_history, list);
 495                 list_del(&entry->list);
 496                 free(entry->history);
 497                 free(entry);
 498         }
 499 }
 500
 501 static void plot_movie_history(struct plot *plot, struct list_head *list)
 502 {
 503         struct plot_history *ph;
 504
 505         if (num_histories > 2)
 506                 rewind_spindle_steps(num_histories - 1);
 507
 508         list_for_each_entry(ph, list, list) {
 509                 if (movie_style == MOVIE_SPINDLE)
 510                         svg_io_graph_movie_array_spindle(plot, ph);
 511                 else
 512                         svg_io_graph_movie_array(plot, ph);
 513          }
 514 }
 515
 516 static void free_all_plot_history(struct list_head *head)
 517 {
 518         struct plot_history *entry;
 519         while (!list_empty(head)) {
 520                 entry = list_entry(head->next, struct plot_history, list);
 521                 list_del(&entry->list);
 522                 free(entry->history);
 523                 free(entry);
 524         }
 525 }
 526
 527 static void __plot_io(struct plot *plot, int seconds, u64 max_offset)
 528 {
 529         struct trace_file *tf;
 530
 531         setup_axis(plot);
 532
 533         svg_alloc_legend(plot, num_traces * 2);
 534
 535         set_plot_label(plot, "Device IO");
 536         set_ylabel(plot, "Offset (MB)");
 537         set_yticks(plot, 4, 0, max_offset / (1024 * 1024), "");
 538         set_xticks(plot, 9, 0, seconds);
 539
 540         list_for_each_entry(tf, &all_traces, list) {
 541                 char *label = tf->label;
 542
 543                 if (!label)
 544                         label = "";
 545                 svg_io_graph(plot, tf->gdd_reads, tf->read_color);
 546                 if (tf->gdd_reads->total_ios)
 547                         svg_add_legend(plot, label, " Reads", tf->read_color);
 548
 549                 svg_io_graph(plot, tf->gdd_writes, tf->write_color);
 550                 if (tf->gdd_writes->total_ios) {
 551                         svg_add_legend(plot, label, " Writes", tf->write_color);
 552                 }
 553         }
 554         if (plot->add_xlabel)
 555                 set_xlabel(plot, "Time (seconds)");
 556         svg_write_legend(plot);
 557 }
 558
 559 static void plot_io(struct plot *plot, int seconds, u64 max_offset)
 560 {
 561         if (active_graphs[IO_GRAPH_INDEX] == 0)
 562                 return;
 563
 564         plot->add_xlabel = last_active_graph == IO_GRAPH_INDEX;
 565         __plot_io(plot, seconds, max_offset);
 566         close_plot(plot);
 567 }
 568
 569 static void __plot_tput(struct plot *plot, int seconds)
 570 {
 571         struct trace_file *tf;
 572         char *units;
 573         char line[128];
 574         u64 max = 0;
 575
 576         if (num_traces > 1)
 577                 svg_alloc_legend(plot, num_traces);
 578         list_for_each_entry(tf, &all_traces, list) {
 579                 if (tf->tput_gld->max > max)
 580                         max = tf->tput_gld->max;
 581         }
 582         list_for_each_entry(tf, &all_traces, list)
 583                 tf->tput_gld->max = max;
 584
 585         setup_axis(plot);
 586         set_plot_label(plot, "Throughput");
 587
 588         tf = list_entry(all_traces.next, struct trace_file, list);
 589
 590         scale_line_graph_bytes(&max, &units, 1024);
 591         sprintf(line, "%sB/s", units);
 592         set_ylabel(plot, line);
 593         set_yticks(plot, 4, 0, max, "");
 594         set_xticks(plot, 9, 0, seconds);
 595
 596         list_for_each_entry(tf, &all_traces, list) {
 597                 svg_line_graph(plot, tf->tput_gld, tf->read_color, 0, 0);
 598                 if (num_traces > 1)
 599                         svg_add_legend(plot, tf->label, "", tf->read_color);
 600         }
 601
 602         if (plot->add_xlabel)
 603                 set_xlabel(plot, "Time (seconds)");
 604         if (num_traces > 1)
 605                 svg_write_legend(plot);
 606 }
 607
 608 static void plot_tput(struct plot *plot, int seconds)
 609 {
 610         if (active_graphs[TPUT_GRAPH_INDEX] == 0)
 611                 return;
 612
 613         plot->add_xlabel = last_active_graph == TPUT_GRAPH_INDEX;
 614         __plot_tput(plot, seconds);
 615         close_plot(plot);
 616 }
 617
 618 static int __plot_cpu(struct plot *plot, int seconds, char *label,
 619                      int active_index, int gld_index)
 620 {
 621         struct trace_file *tf;
 622         int max = 0;
 623         int i;
 624         int gld_i;
 625         char *color;
 626         double avg = 0;
 627         int ymax;
 628         int plotted = 0;
 629
 630         list_for_each_entry(tf, &all_traces, list) {
 631                 if (tf->trace->mpstat_num_cpus > max)
 632                         max = tf->trace->mpstat_num_cpus;
 633         }
 634         if (max == 0)
 635                 return 1;
 636
 637         tf = list_entry(all_traces.next, struct trace_file, list);
 638
 639         ymax = tf->mpstat_gld[gld_index]->max;
 640         if (ymax == 0)
 641                 return 1;
 642
 643         svg_alloc_legend(plot, num_traces * max);
 644
 645         plot->add_xlabel = last_active_graph == active_index;
 646         setup_axis(plot);
 647         set_plot_label(plot, label);
 648
 649         seconds = tf->mpstat_seconds;
 650
 651         set_yticks(plot, 4, 0, tf->mpstat_gld[gld_index]->max, "");
 652         set_ylabel(plot, "Percent");
 653         set_xticks(plot, 9, 0, seconds);
 654
 655         cpu_color_index = 0;
 656         list_for_each_entry(tf, &all_traces, list) {
 657                 for (i = 0; i < tf->mpstat_gld[0]->stop_seconds; i++) {
 658                         if (tf->mpstat_gld[gld_index]->data[i].count) {
 659                                 avg += (tf->mpstat_gld[gld_index]->data[i].sum /
 660                                         tf->mpstat_gld[gld_index]->data[i].count);
 661                         }
 662                 }
 663                 avg /= tf->mpstat_gld[gld_index]->stop_seconds;
 664                 color = pick_cpu_color();
 665                 svg_line_graph(plot, tf->mpstat_gld[0], color, 0, 0);
 666                 svg_add_legend(plot, tf->label, " avg", color);
 667
 668                 for (i = 1; i < tf->trace->mpstat_num_cpus + 1; i++) {
 669                         struct graph_line_data *gld = tf->mpstat_gld[i * MPSTAT_GRAPHS + gld_index];
 670                         double this_avg = 0;
 671
 672                         for (gld_i = 0; gld_i < gld->stop_seconds; gld_i++) {
 673                                 if (gld->data[i].count) {
 674                                         this_avg += gld->data[i].sum /
 675                                                 gld->data[i].count;
 676                                 }
 677                         }
 678
 679                         this_avg /= gld->stop_seconds;
 680
 681                         for (gld_i = 0; gld_i < gld->stop_seconds; gld_i++) {
 682                                 double val;
 683
 684                                 if (gld->data[gld_i].count == 0)
 685                                         continue;
 686                                 val = (double)gld->data[gld_i].sum /
 687                                         gld->data[gld_i].count;
 688
 689                                 if (this_avg > avg + 30 || val > 95) {
 690                                         color = pick_cpu_color();
 691                                         svg_line_graph(plot, gld, color, avg + 30, 95);
 692                                         snprintf(line, line_len, " CPU %d\n", i - 1);
 693                                         svg_add_legend(plot, tf->label, line, color);
 694                                         plotted++;
 695                                         break;
 696                                 }
 697
 698                         }
 699                 }
 700         }
 701
 702         if (plot->add_xlabel)
 703                 set_xlabel(plot, "Time (seconds)");
 704
 705         if (!plot->no_legend) {
 706                 svg_write_legend(plot);
 707                 svg_free_legend(plot);
 708         }
 709         return 0;
 710 }
 711
 712 static void plot_cpu(struct plot *plot, int seconds, char *label,
 713                      int active_index, int gld_index)
 714 {
 715         if (active_graphs[active_index] == 0)
 716                 return;
 717
 718         plot->add_xlabel = last_active_graph == active_index;
 719         if (!__plot_cpu(plot, seconds, label, active_index, gld_index))
 720                 close_plot(plot);
 721 }
 722
 723 static void __plot_queue_depth(struct plot *plot, int seconds)
 724 {
 725         struct trace_file *tf;
 726
 727         plot->add_xlabel = last_active_graph == QUEUE_DEPTH_GRAPH_INDEX;
 728
 729         setup_axis(plot);
 730         set_plot_label(plot, "Queue Depth");
 731         if (num_traces > 1)
 732                 svg_alloc_legend(plot, num_traces);
 733
 734         tf = list_entry(all_traces.next, struct trace_file, list);
 735         set_ylabel(plot, "Pending IO");
 736         set_yticks(plot, 4, 0, tf->queue_depth_gld->max, "");
 737         set_xticks(plot, 9, 0, seconds);
 738
 739         list_for_each_entry(tf, &all_traces, list) {
 740                 svg_line_graph(plot, tf->queue_depth_gld, tf->read_color, 0, 0);
 741                 if (num_traces > 1)
 742                         svg_add_legend(plot, tf->label, "", tf->read_color);
 743         }
 744
 745         if (plot->add_xlabel)
 746                 set_xlabel(plot, "Time (seconds)");
 747         if (num_traces > 1)
 748                 svg_write_legend(plot);
 749 }
 750
 751 static void plot_queue_depth(struct plot *plot, int seconds)
 752 {
 753         if (active_graphs[QUEUE_DEPTH_GRAPH_INDEX] == 0)
 754                 return;
 755         __plot_queue_depth(plot, seconds);
 756         close_plot(plot);
 757 }
 758
 759 static void convert_movie_files(char *movie_dir)
 760 {
 761         fprintf(stderr, "Converting svg files in %s\n", movie_dir);
 762         snprintf(line, line_len, "find %s -name \\*.svg | xargs -I{} -n 1 -P 8 rsvg-convert -o {}.png {}",
 763                  movie_dir);
 764         system(line);
 765 }
 766
 767 static void mencode_movie(char *movie_dir)
 768 {
 769         fprintf(stderr, "Creating movie %s\n", movie_dir);
 770         snprintf(line, line_len, "ffmpeg -r 20 -y -i %s/%%10d-%s.svg.png -b:v 250k "
 771                  "-vcodec libx264 %s", movie_dir, output_filename, output_filename);
 772         system(line);
 773 }
 774
 775 static void cleanup_movie(char *movie_dir)
 776 {
 777         fprintf(stderr, "Removing movie dir %s\n", movie_dir);
 778         snprintf(line, line_len, "rm %s/*", movie_dir);
 779         system(line);
 780
 781         snprintf(line, line_len, "rmdir %s", movie_dir);
 782         system(line);
 783 }
 784
 785 static void plot_io_movie(struct plot *plot)
 786 {
 787         struct trace_file *tf;
 788         char *movie_dir = create_movie_temp_dir();
 789         int i;
 790         struct plot_history *read_history;
 791         struct plot_history *write_history;
 792         int batch_i;
 793         int movie_len = 30;
 794         int movie_frames_per_sec = 20;
 795         int total_frames = movie_len * movie_frames_per_sec;
 796         int rows, cols;
 797         int batch_count;
 798         int graph_width_factor = 5;
 799         int orig_y_offset;
 800
 801         get_graph_size(&cols, &rows);
 802         batch_count = cols / total_frames;
 803
 804         if (batch_count == 0)
 805                 batch_count = 1;
 806
 807         list_for_each_entry(tf, &all_traces, list) {
 808                 char *label = tf->label;
 809                 if (!label)
 810                         label = "";
 811
 812                 i = 0;
 813                 while (i < cols) {
 814                         snprintf(line, line_len, "%s/%010d-%s.svg", movie_dir, i, output_filename);
 815                         set_plot_output(plot, line);
 816                         set_plot_title(plot, graph_title);
 817                         orig_y_offset = plot->start_y_offset;
 818
 819                         plot->no_legend = 1;
 820
 821                         set_graph_size(cols / graph_width_factor, rows / 8);
 822
 823                         __plot_tput(plot, tf->gdd_reads->seconds);
 824                         svg_write_time_line(plot, i / graph_width_factor);
 825                         close_plot(plot);
 826
 827                         if (!__plot_cpu(plot, tf->gdd_reads->seconds,
 828                                    "CPU System Time", CPU_SYS_GRAPH_INDEX, MPSTAT_SYS)) {
 829                                 svg_write_time_line(plot, i / graph_width_factor);
 830                                 close_plot(plot);
 831                         }
 832
 833                         __plot_queue_depth(plot, tf->gdd_reads->seconds);
 834                         svg_write_time_line(plot, i / graph_width_factor);
 835
 836                         close_plot_col(plot);
 837
 838                         /* movie graph starts here */
 839                         plot->start_y_offset = orig_y_offset;
 840                         set_graph_size(cols - cols / graph_width_factor, rows);
 841                         plot->no_legend = 0;
 842
 843                         if (movie_style == MOVIE_SPINDLE)
 844                                 setup_axis_spindle(plot);
 845                         else
 846                                 setup_axis(plot);
 847
 848                         svg_alloc_legend(plot, num_traces * 2);
 849
 850                         read_history = alloc_plot_history(tf->read_color);
 851                         write_history = alloc_plot_history(tf->write_color);
 852                         read_history->col = i;
 853                         write_history->col = i;
 854
 855                         if (tf->gdd_reads->total_ios)
 856                                 svg_add_legend(plot, label, " Reads", tf->read_color);
 857                         if (tf->gdd_writes->total_ios)
 858                                 svg_add_legend(plot, label, " Writes", tf->write_color);
 859
 860                         batch_i = 0;
 861                         while (i < cols && batch_i < batch_count) {
 862                                 svg_io_graph_movie(tf->gdd_reads, read_history, i);
 863                                 svg_io_graph_movie(tf->gdd_writes, write_history, i);
 864                                 i++;
 865                                 batch_i++;
 866                         }
 867
 868                         add_history(read_history, &movie_history_reads);
 869                         add_history(write_history, &movie_history_writes);
 870
 871                         plot_movie_history(plot, &movie_history_reads);
 872                         plot_movie_history(plot, &movie_history_writes);
 873
 874                         svg_write_legend(plot);
 875                         close_plot(plot);
 876                         close_plot(plot);
 877
 878                         close_plot_file(plot);
 879                 }
 880                 free_all_plot_history(&movie_history_reads);
 881                 free_all_plot_history(&movie_history_writes);
 882         }
 883         convert_movie_files(movie_dir);
 884         mencode_movie(movie_dir);
 885         cleanup_movie(movie_dir);
 886         free(movie_dir);
 887 }
 888
 889 static void plot_latency(struct plot *plot, int seconds)
 890 {
 891         struct trace_file *tf;
 892         char *units;
 893         char line[128];
 894         u64 max = 0;
 895
 896         if (active_graphs[LATENCY_GRAPH_INDEX] == 0)
 897                 return;
 898
 899         if (num_traces > 1)
 900                 svg_alloc_legend(plot, num_traces);
 901         list_for_each_entry(tf, &all_traces, list) {
 902                 if (tf->latency_gld->max > max)
 903                         max = tf->latency_gld->max;
 904         }
 905         list_for_each_entry(tf, &all_traces, list)
 906                 tf->latency_gld->max = max;
 907
 908         plot->add_xlabel = last_active_graph == TPUT_GRAPH_INDEX;
 909         setup_axis(plot);
 910         set_plot_label(plot, "IO Latency");
 911
 912         tf = list_entry(all_traces.next, struct trace_file, list);
 913
 914         scale_line_graph_time(&max, &units);
 915         sprintf(line, "latency (%ss)", units);
 916         set_ylabel(plot, line);
 917         set_yticks(plot, 4, 0, max, "");
 918         set_xticks(plot, 9, 0, seconds);
 919
 920         list_for_each_entry(tf, &all_traces, list) {
 921                 svg_line_graph(plot, tf->latency_gld, tf->read_color, 0, 0);
 922                 if (num_traces > 1)
 923                         svg_add_legend(plot, tf->label, "", tf->read_color);
 924         }
 925
 926         if (plot->add_xlabel)
 927                 set_xlabel(plot, "Time (seconds)");
 928         if (num_traces > 1)
 929                 svg_write_legend(plot);
 930         close_plot(plot);
 931 }
 932
 933 static void plot_iops(struct plot *plot, int seconds)
 934 {
 935         struct trace_file *tf;
 936         char *units;
 937         u64 max = 0;
 938
 939         if (active_graphs[IOPS_GRAPH_INDEX] == 0)
 940                 return;
 941
 942         list_for_each_entry(tf, &all_traces, list) {
 943                 if (tf->iop_gld->max > max)
 944                         max = tf->iop_gld->max;
 945         }
 946
 947         list_for_each_entry(tf, &all_traces, list)
 948                 tf->iop_gld->max = max;
 949
 950
 951         plot->add_xlabel = last_active_graph == IOPS_GRAPH_INDEX;
 952         setup_axis(plot);
 953         set_plot_label(plot, "IOPs");
 954         if (num_traces > 1)
 955                 svg_alloc_legend(plot, num_traces);
 956
 957         tf = list_entry(all_traces.next, struct trace_file, list);
 958
 959         scale_line_graph_bytes(&max, &units, 1000);
 960         set_ylabel(plot, "IO/s");
 961
 962         set_yticks(plot, 4, 0, max, units);
 963         set_xticks(plot, 9, 0, seconds);
 964
 965         list_for_each_entry(tf, &all_traces, list) {
 966                 svg_line_graph(plot, tf->iop_gld, tf->read_color, 0, 0);
 967                 if (num_traces > 1)
 968                         svg_add_legend(plot, tf->label, "", tf->read_color);
 969         }
 970
 971         if (plot->add_xlabel)
 972                 set_xlabel(plot, "Time (seconds)");
 973         if (num_traces > 1)
 974                 svg_write_legend(plot);
 975
 976         close_plot(plot);
 977 }
 978
 979 enum {
 980         HELP_LONG_OPT = 1,
 981 };
 982
 983 char *option_string = "T:t:o:l:r:O:N:d:p:m::h:w:";
 984 static struct option long_options[] = {
 985         {"title", required_argument, 0, 'T'},
 986         {"trace", required_argument, 0, 't'},
 987         {"output", required_argument, 0, 'o'},
 988         {"label", required_argument, 0, 'l'},
 989         {"rolling", required_argument, 0, 'r'},
 990         {"no-graph", required_argument, 0, 'N'},
 991         {"only-graph", required_argument, 0, 'O'},
 992         {"device", required_argument, 0, 'd'},
 993         {"prog", required_argument, 0, 'p'},
 994         {"movie", optional_argument, 0, 'm'},
 995         {"width", required_argument, 0, 'w'},
 996         {"height", required_argument, 0, 'h'},
 997         {"help", no_argument, 0, HELP_LONG_OPT},
 998         {0, 0, 0, 0}
 999 };
1000
1001 static void print_usage(void)
1002 {
1003         fprintf(stderr, "iowatcher usage:\n"
1004                 "\t-d (--device): device for blktrace to trace\n"
1005                 "\t-t (--trace): trace file name (more than one allowed)\n"
1006                 "\t-l (--label): trace label in the graph\n"
1007                 "\t-o (--output): output file name (SVG only)\n"
1008                 "\t-p (--prog): program to run while blktrace is run\n"
1009                 "\t-p (--movie [=spindle|rect]): create IO animations\n"
1010                 "\t-r (--rolling): number of seconds in the rolling averge\n"
1011                 "\t-T (--title): graph title\n"
1012                 "\t-N (--no-graph): skip a single graph (io, tput, latency, queue_depth, \n"
1013                 "\t\t\tiops, cpu-sys, cpu-io, cpu-irq cpu-soft cpu-user)\n"
1014                 "\t-O (--only-graph): add a single graph to the output\n"
1015                 "\t-h (--height): set the height of each graph\n"
1016                 "\t-w (--width): set the width of each graph\n"
1017                );
1018         exit(1);
1019 }
1020
1021 static int parse_options(int ac, char **av)
1022 {
1023         int c;
1024         int disabled = 0;
1025
1026         while (1) {
1027                 // int this_option_optind = optind ? optind : 1;
1028                 int option_index = 0;
1029
1030                 c = getopt_long(ac, av, option_string,
1031                                 long_options, &option_index);
1032
1033                 if (c == -1)
1034                         break;
1035
1036                 switch(c) {
1037                 case 'T':
1038                         graph_title = strdup(optarg);
1039                         break;
1040                 case 't':
1041                         add_trace_file(optarg);
1042                         set_blktrace_outfile(optarg);
1043                         break;
1044                 case 'o':
1045                         output_filename = strdup(optarg);
1046                         break;
1047                 case 'l':
1048                         set_trace_label(optarg);
1049                         break;
1050                 case 'r':
1051                         set_rolling_avg(atoi(optarg));
1052                         break;
1053                 case 'O':
1054                         if (!disabled) {
1055                                 disable_all_graphs();
1056                                 disabled = 1;
1057                         }
1058                         enable_one_graph(optarg);
1059                         break;
1060                 case 'N':
1061                         disable_one_graph(optarg);
1062                         break;
1063                 case 'd':
1064                         blktrace_device = strdup(optarg);
1065                         break;
1066                 case 'p':
1067                         program_to_run = strdup(optarg);
1068                         break;
1069                 case 'm':
1070                         make_movie = 1;
1071                         if (optarg) {
1072                                 movie_style = lookup_movie_style(optarg);
1073                                 if (movie_style < 0) {
1074                                         fprintf(stderr, "Unknown movie style %s\n", optarg);
1075                                         print_usage();
1076                                 }
1077                         }
1078                         fprintf(stderr, "Using movie style: %s\n",
1079                                 movie_styles[movie_style]);
1080                         break;
1081                 case 'h':
1082                         opt_graph_height = atoi(optarg);
1083                         break;
1084                 case 'w':
1085                         opt_graph_width = atoi(optarg);
1086                         break;
1087                 case '?':
1088                 case HELP_LONG_OPT:
1089                         print_usage();
1090                         break;
1091                 default:
1092                         break;
1093                 }
1094         }
1095         return 0;
1096 }
1097
1098
1099 int main(int ac, char **av)
1100 {
1101         struct plot *plot;
1102         int seconds = 0;
1103         u64 max_offset = 0;
1104         struct trace_file *tf;
1105         int ret;
1106
1107         init_io_hash_table();
1108
1109         enable_all_graphs();
1110
1111         parse_options(ac, av);
1112
1113         last_active_graph = last_graph();
1114         if (make_movie) {
1115                 set_io_graph_scale(256);
1116                 if (movie_style == MOVIE_SPINDLE)
1117                         set_graph_size(750, 550);
1118                 else
1119                         set_graph_size(700, 400);
1120         }
1121         if (opt_graph_height)
1122                 set_graph_height(opt_graph_height);
1123
1124         if (opt_graph_width)
1125                 set_graph_width(opt_graph_height);
1126
1127         if (list_empty(&all_traces)) {
1128                 fprintf(stderr, "No traces found, exiting\n");
1129                 exit(1);
1130         }
1131
1132         if (blktrace_device) {
1133                 ret = start_blktrace(blktrace_device, blktrace_outfile,
1134                                      blktrace_dest_dir);
1135                 if (ret) {
1136                         fprintf(stderr, "exiting due to blktrace failure\n");
1137                         exit(1);
1138                 }
1139                 start_mpstat(blktrace_outfile);
1140                 if (program_to_run) {
1141                         ret = run_program(program_to_run);
1142                         if (ret) {
1143                                 fprintf(stderr, "failed to run %s\n",
1144                                         program_to_run);
1145                                 exit(1);
1146                         }
1147                         wait_for_tracers();
1148                         blktrace_to_dump(blktrace_outfile);
1149                 } else {
1150                         /* no program specified, just wait for
1151                          * blktrace to exit
1152                          */
1153                         wait_for_tracers();
1154                 }
1155         }
1156
1157         /* step one, read all the traces */
1158         read_traces();
1159
1160         /* step two, find the maxes for time and offset */
1161         list_for_each_entry(tf, &all_traces, list)
1162                 compare_max_tf(tf, &seconds, &max_offset);
1163
1164         /* push the max we found into all the tfs */
1165         set_all_max_tf(seconds, max_offset);
1166
1167         /* alloc graphing structs for all the traces */
1168         setup_trace_file_graphs();
1169
1170         /* run through all the traces and read their events */
1171         read_trace_events();
1172
1173         plot = alloc_plot();
1174
1175         if (make_movie) {
1176                 plot_io_movie(plot);
1177                 exit(0);
1178         }
1179
1180         set_plot_output(plot, output_filename);
1181
1182         if (active_graphs[IO_GRAPH_INDEX] || found_mpstat)
1183                 set_legend_width(longest_label + strlen("writes"));
1184         else if (num_traces > 1)
1185                 set_legend_width(longest_label);
1186         else
1187                 set_legend_width(0);
1188
1189         set_plot_title(plot, graph_title);
1190         plot_io(plot, seconds, max_offset);
1191         plot_tput(plot, seconds);
1192         plot_cpu(plot, seconds, "CPU IO Wait Time",
1193                  CPU_IO_GRAPH_INDEX, MPSTAT_IO);
1194         plot_cpu(plot, seconds, "CPU System Time",
1195                  CPU_SYS_GRAPH_INDEX, MPSTAT_SYS);
1196         plot_cpu(plot, seconds, "CPU IRQ Time",
1197                  CPU_IRQ_GRAPH_INDEX, MPSTAT_IRQ);
1198         plot_cpu(plot, seconds, "CPU SoftIRQ Time",
1199                  CPU_SOFT_GRAPH_INDEX, MPSTAT_SOFT);
1200         plot_cpu(plot, seconds, "CPU User Time",
1201                  CPU_USER_GRAPH_INDEX, MPSTAT_USER);
1202
1203         plot_latency(plot, seconds);
1204         plot_queue_depth(plot, seconds);
1205         plot_iops(plot, seconds);
1206
1207         /* once for all */
1208         close_plot(plot);
1209         close_plot_file(plot);
1210         return 0;
1211 }