benchmark silo added

[c11concurrency-benchmarks.git] / silo / masstree / kvio.hh

diff --git a/silo/masstree/kvio.hh b/silo/masstree/kvio.hh
new file mode 100644 (file)
index 0000000..5aea020
--- /dev/null
+++ b/silo/masstree/kvio.hh
@@ -0,0 +1,67 @@
+/* Masstree
+ * Eddie Kohler, Yandong Mao, Robert Morris
+ * Copyright (c) 2012-2014 President and Fellows of Harvard College
+ * Copyright (c) 2012-2014 Massachusetts Institute of Technology
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, subject to the conditions
+ * listed in the Masstree LICENSE file. These conditions include: you must
+ * preserve this copyright notice, and you cannot mention the copyright
+ * holders in advertising related to the Software without their permission.
+ * The Software is provided WITHOUT ANY WARRANTY, EXPRESS OR IMPLIED. This
+ * notice is a summary of the Masstree LICENSE file; the license in that file
+ * is legally binding.
+ */
+#ifndef KVIO_H
+#define KVIO_H
+#include <string>
+#include <vector>
+#include <stdlib.h>
+#include "string.hh"
+#include "str.hh"
+
+struct kvout {
+    int fd;
+    char* buf;
+    unsigned capacity; // allocated size of buf
+    unsigned n;   // # of chars we've written to buf
+
+    inline void append(char c);
+    inline char* reserve(int n);
+    inline void adjust_length(int delta);
+    inline void set_end(char* end);
+    void grow(unsigned want);
+};
+
+kvout* new_kvout(int fd, int buflen);
+kvout* new_bufkvout();
+void kvout_reset(kvout* kv);
+void free_kvout(kvout* kv);
+int kvwrite(kvout* kv, const void* buf, unsigned int n);
+void kvflush(kvout* kv);
+
+inline void kvout::append(char c) {
+    if (n == capacity)
+        grow(0);
+    buf[n] = c;
+    ++n;
+}
+
+inline char* kvout::reserve(int nchars) {
+    if (n + nchars > capacity)
+        grow(n + nchars);
+    return buf + n;
+}
+
+inline void kvout::adjust_length(int delta) {
+    masstree_precondition(n + delta <= capacity);
+    n += delta;
+}
+
+inline void kvout::set_end(char* x) {
+    masstree_precondition(x >= buf && x <= buf + capacity);
+    n = x - buf;
+}
+
+#endif