add to txn test

[nbds] / include / lwt.h

diff --git a/include/lwt.h b/include/lwt.h
index db862921e2a27881d2c2bc1322dc4648e8aa4a56..dd98369a5d4570ef78d0b158b2a80ead0b95241b 100644 (file)
--- a/include/lwt.h
+++ b/include/lwt.h
@@ -13,6 +13,12 @@
 #define TRACE(flag, format, v1, v2) lwt_trace(flag, format, (size_t)(v1), (size_t)(v2))
 #endif
 
+#ifdef NDEBUG
+#define ASSERT(x) 
+#else
+#define ASSERT(x) if (!(x)) { lwt_halt(); assert(!#x); }
+#endif
+
 // Dump trace records to <file_name>. The file should be post-processed with "sort" before viewing.
 void lwt_dump (const char *file_name) __attribute__ ((externally_visible));
 
@@ -28,8 +34,8 @@ void lwt_set_trace_level (const char *flags);
 // the dump. It is only included when its specified category is enabled at a trace level greater than or equal to
 // the one in <flag>. Categories are case sensitive. 
 static inline void lwt_trace (const char *flag, const char *format, size_t value1, size_t value2) {
-    extern uint64_t flag_mask_;
-    if (EXPECT_FALSE(flag_mask_ & (1 << (flag[0] - 'A')))) {
+    extern char flag_state_[256];
+    if (EXPECT_FALSE(flag_state_[(unsigned)flag[0]] >= flag[1])) {
         // embed <flags> in <format> so we don't have to make the lwt_record_t any bigger than it already is
         format = (const char *)((size_t)format | ((uint64_t)flag[0] << 56) | ((uint64_t)flag[1] << 48));
         extern void lwt_trace_i (const char *format, size_t value1, size_t value2);
@@ -37,4 +43,6 @@ static inline void lwt_trace (const char *flag, const char *format, size_t value
     }
 }
 
+void lwt_halt (void);
+
 #endif//LWT_H