Collect some benchmarks in bench.cc

c4dea1b7 · Silas Boyd-Wickizer · f4cd5305 · c4dea1b7 · c4dea1b7
--- a/bin/Makefrag
+++ b/bin/Makefrag
@@ -2,6 +2,7 @@ $(O)/bin/%.o: CFLAGS:=$(CFLAGS)
 $(O)/bin/%.o: CXXFLAGS:=$(CXXFLAGS)
 UPROGS= \
+        bench \
 	cat \
        du \
 	echo \

--- a/bin/bench.cc
+++ b/bin/bench.cc
+#include "types.h"
+#include "user.h"
+#include "amd64.h"
+static u64
+time_this(const char *av[])
+{
+  u64 t0 = rdtsc();
+  int pid = fork(0);
+  if (pid < 0)
+    die("time_this: fork failed %s", av[0]);
+  if (pid == 0) {
+    exec(av[0], av);
+    die("time_this: exec failed %s", av[0]);
+  }
+  wait();
+  u64 t1 = rdtsc();
+  return t1-t0;
+}
+struct Bench
+{
+  virtual void run(void) = 0;
+  virtual char* result(void) = 0;  
+};
+#define NEW_OPERATOR(classname)                                         \
+  void* operator new(unsigned long nbytes) {                            \
+    assert(nbytes == sizeof(classname));                                \
+    return malloc(sizeof(classname));                                   \
+  }
+struct TimedExec : public Bench
+{
+  TimedExec(const char **argv) : argv_(argv) {}
+  virtual void run(void) {
+    char *str = result_;
+    *str = 0;
+    for (int i = 0; argv_[i]; i++) {
+      int n = sizeof(result_)-(str-result_);
+      snprintf(str, n, i==0?"%s":" %s", argv_[i]);
+      str += strlen(str);
+    }
+    printf("%s .. ", result_);
+    u64 r = time_this(argv_);
+    snprintf(str, sizeof(result_)-(str-result_), ": %lu cycles", r);
+    printf("%lu cycles\n", r);
+  }
+  virtual char* result(void) {
+    return result_;
+  }
+  const char **argv_;
+  char result_[256];
+  NEW_OPERATOR(TimedExec)
+};
+#define CMD(...) new TimedExec((const char *[]){__VA_ARGS__, 0})
+int
+main(int ac, char **av)
+{
+  static Bench* the_bench[128];
+  int n = 0;
+  the_bench[n++] = CMD("mktree", "tree.xdu", "4", "4");
+  the_bench[n++] = CMD("xdu");
+  the_bench[n++] = CMD("xls");
+  the_bench[n++] = CMD("mapbench", "1");
+  the_bench[n++] = CMD("mapbench", "2");
+  the_bench[n++] = CMD("mapbench", "16");
+  for (int i = 0; i < n; i++)
+    the_bench[i]->run();
+  printf("\n");
+  for (int i = 0; i < n; i++)
+    printf("%s\n", the_bench[i]->result());
+}