1 files changed, 9 insertions, 0 deletions

diff --git a/sys/cddl/dev/dtmalloc/dtmalloc.c b/sys/cddl/dev/dtmalloc/dtmalloc.c
index 017d1350b48d8..49516a27a6352 100644
--- a/sys/cddl/dev/dtmalloc/dtmalloc.c
+++ b/sys/cddl/dev/dtmalloc/dtmalloc.c
@@ -36,6 +36,9 @@
 #include <sys/dtrace.h>
 #include <sys/dtrace_bsd.h>
 
+extern bool dtrace_malloc_enabled;
+static uint32_t dtrace_malloc_enabled_count;
+
 static d_open_t	dtmalloc_open;
 static int	dtmalloc_unload(void);
 static void	dtmalloc_getargdesc(void *, dtrace_id_t, void *, dtrace_argdesc_t *);
@@ -152,6 +155,9 @@ dtmalloc_enable(void *arg, dtrace_id_t id, void *parg)
 {
 	uint32_t *p = parg;
 	*p = id;
+	dtrace_malloc_enabled_count++;
+	if (dtrace_malloc_enabled_count == 1)
+		dtrace_malloc_enabled = true;
 }
 
 static void
@@ -159,6 +165,9 @@ dtmalloc_disable(void *arg, dtrace_id_t id, void *parg)
 {
 	uint32_t *p = parg;
 	*p = 0;
+	dtrace_malloc_enabled_count--;
+	if (dtrace_malloc_enabled_count == 0)
+		dtrace_malloc_enabled = false;
 }
 
 static void