1 files changed, 10 insertions, 1 deletions
diff --git a/dbmalloc.h b/dbmalloc.h
index f05f8f5..a5a13ad 100644
--- a/dbmalloc.h
+++ b/dbmalloc.h
@@ -7,10 +7,19 @@ void * m_malloc(size_t size);
 void * m_calloc(size_t nmemb, size_t size);
 void * m_strdup(const char * str);
 void * m_realloc(void* ptr, size_t size);
+
+#if DROPBEAR_TRACKING_MALLOC
 void m_free_direct(void* ptr);
 #define m_free(X) do {m_free_direct(X); (X) = NULL;} while (0)
-
 void m_malloc_set_epoch(unsigned int epoch);
 void m_malloc_free_epoch(unsigned int epoch, int dofree);
 
+#else
+/* plain wrapper */
+
+#define m_free(X) do {free(X); (X) = NULL;} while (0)
+
+#endif
+
+
 #endif /* DBMALLOC_H_ */