From 96c538d90fb637834fb5b530a8d1094bde4d2ccc Mon Sep 17 00:00:00 2001
From: Daniel Micay <danielmicay@gmail.com>
Date: Fri, 7 Sep 2018 00:33:51 -0400
Subject: [PATCH] add configuration for zero on free

---
 config.h |  1 +
 malloc.c | 18 ++++++++++++++++--
 2 files changed, 17 insertions(+), 2 deletions(-)

diff --git a/config.h b/config.h
index c19a5de..822bf6d 100644
--- a/config.h
+++ b/config.h
@@ -4,5 +4,6 @@
 #define GUARD_SLABS true
 #define WRITE_AFTER_FREE_CHECK true
 #define SLOT_RANDOMIZE true
+#define ZERO_ON_FREE true
 
 #endif
diff --git a/malloc.c b/malloc.c
index 1d27dc1..a0cd0e4 100644
--- a/malloc.c
+++ b/malloc.c
@@ -20,6 +20,8 @@
 
 static_assert(sizeof(void *) == 8, "64-bit only");
 
+static_assert(!WRITE_AFTER_FREE_CHECK || ZERO_ON_FREE, "WRITE_AFTER_FREE_CHECK depends on ZERO_ON_FREE");
+
 // either sizeof(uint64_t) or 0
 static const size_t canary_size = sizeof(uint64_t);
 
@@ -411,7 +413,9 @@ static inline void slab_free(void *p) {
     }
 
     if (!is_zero_size) {
-        memset(p, 0, size - canary_size);
+        if (ZERO_ON_FREE) {
+            memset(p, 0, size - canary_size);
+        }
 
         if (canary_size) {
             uint64_t canary_value;
@@ -780,7 +784,17 @@ EXPORT void *h_calloc(size_t nmemb, size_t size) {
     }
     init();
     total_size = adjust_size_for_canaries(total_size);
-    return allocate(total_size);
+    if (ZERO_ON_FREE) {
+        return allocate(total_size);
+    }
+    void *p = allocate(total_size);
+    if (unlikely(p == NULL)) {
+        return NULL;
+    }
+    if (size) {
+        memset(p, 0, total_size - canary_size);
+    }
+    return p;
 }
 
 static const size_t mremap_threshold = 4 * 1024 * 1024;