1 files changed, 37 insertions, 0 deletions
diff --git a/memory/alloc/reallocarray.c b/memory/alloc/reallocarray.c
new file mode 100644
index 0000000..c58dfe9
--- /dev/null
+++ b/memory/alloc/reallocarray.c
@@ -0,0 +1,37 @@
+/*
+ * Copyright (c) Jon Santmyer.
+ * This source file is released under the LGPL Version 3 as detailed in
+ * the LICENSE file provided in the following repository:
+ * https://git.jonsantmyer.com/heap/
+ * */
+
+#include "malloc.h"
+#include "assert.h"
+#include "string.h"
+
+void*
+__reallocarray_impl(heap_cache_t *cache, void *ptr, size_t nmemb, size_t w)
+{
+    /* Free if w == nmemb == 0*/
+    if(w == 0 || nmemb == 0) {
+        __free_impl(cache, ptr);
+        return NULL;
+    }
+    
+    if(ptr == NULL)
+        return __calloc_impl(cache, nmemb, w);
+
+    size_t allocw = w * nmemb;
+    w = REALW(w);
+    
+    bucket_obj_t *obj = OBJ_FROM_PTR(ptr);
+    assert(OBJ_VALID(obj));
+    size_t oldw = OBJ_WIDTH(obj);
+    if(allocw < oldw) return ptr;
+
+    ptr = __realloc_impl(cache, ptr, allocw);
+    size_t wdiff = allocw - oldw;
+    void *zeroat = (void*)(((uintptr_t)ptr) + oldw);
+    memset(zeroat, 0, wdiff);
+    return ptr;
+}