asharing: Utilities to manage abstract sharing scopes and objects

9275cbb6 · Austin Clements · 7de00a22 · 9275cbb6
--- a/include/kmtrace.hh
+++ b/include/kmtrace.hh
+#pragma once
+
 #include "mtrace.h"
+
 #if MTRACE
 // Tell mtrace about switching threads
 struct kstack_tag {
@@ -61,6 +64,51 @@ static inline void mtresume(struct proc *p)
 #define mtrec() mtrace_call_set(1, ~0ull)
 #define mtign() mtrace_call_set(0, ~0ull)

+class mt_ascope
+{
+  char name[32];
+public:
+  explicit mt_ascope(const char *fmt, ...)
+  {
+    va_list ap;
+
+    va_start(ap, fmt);
+    vsnprintf(name, sizeof(name) - 1, fmt, ap);
+    va_end(ap);
+
+    mtrace_ascope_register(0, name);
+  }
+
+  ~mt_ascope()
+  {
+    mtrace_ascope_register(1, name);
+  }
+};
+
+static inline void mtreadavar(const char *fmt, ...)
+{
+  char name[32];
+  va_list ap;
+
+  va_start(ap, fmt);
+  vsnprintf(name, sizeof(name), fmt, ap);
+  va_end(ap);
+
+  mtrace_avar_register(0, name);
+}
+
+static inline void mtwriteavar(const char *fmt, ...)
+{
+  char name[32];
+  va_list ap;
+
+  va_start(ap, fmt);
+  vsnprintf(name, sizeof(name), fmt, ap);
+  va_end(ap);
+
+  mtrace_avar_register(1, name);
+}
+
 #else
 #define mtstart(ip, p) do { } while (0)
 #define mtstop(p) do { } while (0)
@@ -70,4 +118,13 @@ static inline void mtresume(struct proc *p)
 #define mtign(cpu) do { } while (0)
 #define mtrec(cpu) do { } while (0)
 #define mtign(cpu) do { } while (0)
+
+class mt_ascope
+{
+public:
+  explicit mt_ascope(const char *fmt, ...) {}
+};
+#define mtreadavar(fmt, ...) do { } while (0)
+#define mtwriteavar(fmt, ...) do { } while (0)
+
 #endif