1 files changed, 112 insertions, 0 deletions
diff --git a/debug.cc b/debug.cc
new file mode 100644
index 00000000..b7f15120
--- /dev/null
+++ b/debug.cc
@@ -0,0 +1,112 @@
+#include "debug.h"
+
+#ifdef DEBUG_ENABLED
+
+#include <map>
+#include <fstream>
+#include <cstdlib>
+#include <cstring>
+
+int offset = 0;
+
+std::map<void *, int> ptrs;
+
+#define PRINT_INC(x) {					\
+  char buf[128];					\
+  std::sprintf(buf, "%d: %p: %s", ++offset, ptr, x);	\
+  write(1, buf, std::strlen(buf));			\
+}
+
+#define PRINT_DEC(x) {					\
+  char buf[128];					\
+  std::sprintf(buf, "%d: %p: %s", --offset, ptr, x);	\
+  write(1, buf, std::strlen(buf));			\
+}
+
+void * operator new(std::size_t size) throw (std::bad_alloc) {
+  void * ptr = std::malloc(size);
+  if (DEBUG("debug.alloc")) {
+    PRINT_INC("void * operator new(std::size_t size) throw (std::bad_alloc)\n");
+  }
+  return ptr;
+}
+void * operator new[](std::size_t size) throw (std::bad_alloc) {
+  void * ptr = std::malloc(size);
+  if (DEBUG("debug.alloc")) {
+    PRINT_INC("void * operator new[](std::size_t) throw (std::bad_alloc)\n");
+  }
+  return ptr;
+}
+void * operator new(std::size_t size, const std::nothrow_t&) throw() {
+  void * ptr = std::malloc(size);
+  if (DEBUG("debug.alloc")) {
+    PRINT_INC("void * operator new(std::size_t size, const std::nothrow_t&) throw()\n");
+  }
+  return ptr;
+}
+void * operator new[](std::size_t size, const std::nothrow_t&) throw() {
+  void * ptr = std::malloc(size);
+  if (DEBUG("debug.alloc")) {
+    PRINT_INC("void * operator new[](std::size_t size, const std::nothrow_t&) throw()\n");
+  }
+  return ptr;
+}
+void   operator delete(void * ptr) throw() {
+  if (DEBUG("debug.alloc")) {
+    PRINT_DEC("void   operator delete(void * ptr) throw()\n");
+  }
+  std::free(ptr);
+}
+void   operator delete[](void * ptr) throw() {
+  if (DEBUG("debug.alloc")) {
+    PRINT_DEC("void   operator delete[](void * ptr) throw()\n");
+  }
+  std::free(ptr);
+}
+void   operator delete(void * ptr, const std::nothrow_t&) throw() {
+  if (DEBUG("debug.alloc")) {
+    PRINT_DEC("void   operator delete(void * ptr, const std::nothrow_t&) throw()\n");
+  }
+  std::free(ptr);
+}
+void   operator delete[](void * ptr, const std::nothrow_t&) throw() {
+  if (DEBUG("debug.alloc")) {
+    PRINT_DEC("void   operator delete[](void * ptr, const std::nothrow_t&) throw()\n");
+  }
+  std::free(ptr);
+}
+
+std::ostream * _debug_stream	  = &std::cerr;
+bool	       _free_debug_stream = false;
+
+bool _debug_active(const char * const cls) {
+  if (char * debug = std::getenv("DEBUG_CLASS")) {
+    static const char * error;
+    static int	  erroffset;
+    static int	  ovec[30];
+    static pcre * class_regexp = pcre_compile(debug, PCRE_CASELESS,
+					      &error, &erroffset, NULL);
+    return pcre_exec(class_regexp, NULL, cls, std::strlen(cls),
+		     0, 0, ovec, 30) >= 0;
+  }
+  return false;
+}
+
+static struct init_streams {
+  init_streams() {
+    // If debugging is enabled and DEBUG_FILE is set, all debugging
+    // output goes to that file.
+    if (const char * p = std::getenv("DEBUG_FILE")) {
+      _debug_stream      = new std::ofstream(p);
+      _free_debug_stream = true;
+    }
+  }
+  ~init_streams() {
+    if (_free_debug_stream && _debug_stream) {
+      delete _debug_stream;
+      _debug_stream = NULL;
+    }
+  }
+} _debug_init;
+
+#endif // DEBUG_ENABLED