iowatcher: Add mpstat.[ch] into git

Signed-off-by: Chris Mason <chris.mason@fusionio.com>

diff --git a/iowatcher/mpstat.c b/iowatcher/mpstat.c
new file mode 100644 (file)
index 0000000..f2292b9
--- /dev/null
+++ b/iowatcher/mpstat.c
@@ -0,0 +1,275 @@
+/*
+ * Copyright (C) 2012 Fusion-io
+ *
+ *  This program is free software; you can redistribute it and/or
+ *  modify it under the terms of the GNU General Public
+ *  License v2 as published by the Free Software Foundation.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program; if not, write to the Free Software
+ *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <math.h>
+#include <inttypes.h>
+#include <string.h>
+#include <asm/types.h>
+#include <errno.h>
+#include <sys/mman.h>
+#include <time.h>
+#include <math.h>
+
+#include "plot.h"
+#include "blkparse.h"
+#include "list.h"
+#include "tracers.h"
+#include "mpstat.h"
+
+char line[1024];
+int line_len = 1024;
+
+char *record_header = "CPU    %usr   %nice    %sys %iowait    %irq   %soft  %steal  %guest   %idle\n";
+int record_header_len = 0;
+
+static int past_eof(struct trace *trace, char *cur)
+{
+       if (cur >= trace->mpstat_start + trace->mpstat_len)
+               return 1;
+       return 0;
+}
+
+int next_mpstat_line(struct trace *trace)
+{
+       char *next;
+       char *cur = trace->mpstat_cur;
+
+       next = strchr(cur, '\n');
+       if (!next)
+               return 1;
+       next++;
+       if (past_eof(trace, next))
+               return 1;
+       trace->mpstat_cur = next;
+       return 0;
+}
+
+char *next_mpstat(struct trace *trace)
+{
+       char *cur = trace->mpstat_cur;
+
+       cur = strstr(cur, record_header);
+       if (!cur)
+               return NULL;
+
+       cur += record_header_len;
+       if (past_eof(trace, cur))
+               return NULL;
+       trace->mpstat_cur = cur;
+       return cur;
+}
+
+char *first_mpstat(struct trace *trace)
+{
+       char *cur = trace->mpstat_cur;
+
+       trace->mpstat_cur = trace->mpstat_start;
+
+       cur = next_mpstat(trace);
+       if (!cur)
+               return NULL;
+       return cur;
+}
+
+static void find_last_mpstat_time(struct trace *trace)
+{
+       int num_mpstats = 0;
+       char *cur;
+
+       first_mpstat(trace);
+
+       cur = first_mpstat(trace);
+       while (cur) {
+               num_mpstats++;
+               cur = next_mpstat(trace);
+       }
+       first_mpstat(trace);
+       trace->mpstat_seconds = num_mpstats;
+}
+
+static int count_mpstat_cpus(struct trace *trace)
+{
+       char *cur = trace->mpstat_start;
+       char *cpu;
+       char *record;
+       int len;
+       char *line;
+
+       first_mpstat(trace);
+
+       cpu = strstr(cur, " CPU)");
+       if (!cpu)
+               return 0;
+       line = strndup(cur, cpu - cur);
+
+       record = strrchr(line, '(');
+       if (!record) {
+               free(line);
+               return 0;
+       }
+       record++;
+
+       len = line + strlen(line) - record;
+
+       cur = strndup(record, len);
+       trace->mpstat_num_cpus = atoi(cur);
+       first_mpstat(trace);
+       free(line);
+       return trace->mpstat_num_cpus;
+}
+
+static char *guess_filename(char *trace_name)
+{
+       struct stat st;
+       int ret;
+       char *cur;
+       char *tmp;
+
+       snprintf(line, line_len, "%s.mpstat", trace_name);
+       ret = stat(line, &st);
+       if (ret == 0)
+               return trace_name;
+
+       cur = strrchr(trace_name, '.');
+       if (!cur) {
+               return trace_name;
+       }
+
+       tmp = strndup(trace_name, cur - trace_name);
+       snprintf(line, line_len, "%s.mpstat", tmp);
+       ret = stat(line, &st);
+       if (ret == 0)
+               return tmp;
+
+       free(tmp);
+       return trace_name;
+}
+
+int read_mpstat(struct trace *trace, char *trace_name)
+{
+       int fd;
+       struct stat st;
+       int ret;
+       char *p;
+
+       if (record_header_len == 0) {
+               record_header_len = strlen(record_header);
+       }
+
+       snprintf(line, line_len, "%s.mpstat", guess_filename(trace_name));
+
+       fd = open(line, O_RDONLY);
+       if (fd < 0)
+               return 0;
+
+       ret = fstat(fd, &st);
+       if (ret < 0) {
+               fprintf(stderr, "stat failed on %s err %s\n", line, strerror(errno));
+               goto fail_fd;
+       }
+       p = mmap(NULL, st.st_size, PROT_READ, MAP_PRIVATE, fd, 0);
+       if (p == MAP_FAILED) {
+               fprintf(stderr, "Unable to mmap trace file %s, err %s\n", line, strerror(errno));
+               goto fail_fd;
+       }
+
+       trace->mpstat_start = p;
+       trace->mpstat_len = st.st_size;
+       trace->mpstat_cur = p;
+       trace->mpstat_fd = fd;
+       find_last_mpstat_time(trace);
+       count_mpstat_cpus(trace);
+
+       first_mpstat(trace);
+
+       return 0;
+
+fail_fd:
+       close(fd);
+       return 0;
+}
+
+/*
+ * 09:56:26 AM  CPU    %usr   %nice    %sys %iowait    %irq   %soft  %steal  %guest   %idle
+ *
+ * this reads just one line in the mpstat
+ */
+int read_mpstat_event(struct trace *trace, double *user,
+                     double *sys, double *iowait, double *irq,
+                     double *soft)
+{
+       char *cur = trace->mpstat_cur;
+       char *nptr;
+       double val;
+
+       /* jump past the date and CPU number */
+       cur += 16;
+       if (past_eof(trace, cur))
+               return 1;
+
+       /* usr time */
+       val = strtod(cur, &nptr);
+       if (val == 0 && cur == nptr)
+               return 1;
+       *user = val;
+
+       /* nice time, pitch this one */
+       cur = nptr;
+       val = strtod(cur, &nptr);
+       if (val == 0 && cur == nptr)
+               return 1;
+
+       /* system time */
+       cur = nptr;
+       val = strtod(cur, &nptr);
+       if (val == 0 && cur == nptr)
+               return 1;
+       *sys = val;
+
+       cur = nptr;
+       val = strtod(cur, &nptr);
+       if (val == 0 && cur == nptr)
+               return 1;
+       *iowait = val;
+
+       cur = nptr;
+       val = strtod(cur, &nptr);
+       if (val == 0 && cur == nptr)
+               return 1;
+       *irq = val;
+
+       cur = nptr;
+       val = strtod(cur, &nptr);
+       if (val == 0 && cur == nptr)
+               return 1;
+       *soft = val;
+
+       return 0;
+}
+
+int add_mpstat_gld(int time, double sys, struct graph_line_data *gld)
+{
+       gld->data[time].sum = sys;
+       gld->data[time].count = 1;
+       return 0;
+
+}

diff --git a/iowatcher/mpstat.h b/iowatcher/mpstat.h
new file mode 100644 (file)
index 0000000..cf72351
--- /dev/null
+++ b/iowatcher/mpstat.h
@@ -0,0 +1,28 @@
+/*
+ * Copyright (C) 2012 Fusion-io
+ *
+ *  This program is free software; you can redistribute it and/or
+ *  modify it under the terms of the GNU General Public
+ *  License v2 as published by the Free Software Foundation.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program; if not, write to the Free Software
+ *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+#ifndef __MPSTAT__
+#define __MPSTAT__
+
+int read_mpstat(struct trace *trace, char *trace_name);
+char *next_mpstat(struct trace *trace);
+char *first_mpstat(struct trace *trace);
+int read_mpstat_event(struct trace *trace, double *user,
+                     double *sys, double *iowait, double *irq,
+                     double *soft);
+int next_mpstat_line(struct trace *trace);
+int add_mpstat_gld(int time, double sys, struct graph_line_data *gld);
+#endif