Loading...
tests/magazine_malloc.c /dev/null libmalloc-409.40.6
--- /dev/null
+++ libmalloc/libmalloc-409.40.6/tests/magazine_malloc.c
@@ -0,0 +1,68 @@
+//
+//  magazine_malloc.c
+//  libmalloc
+//
+//  Created by Kim Topley on 11/8/17.
+//
+#include <darwintest.h>
+#include <stdlib.h>
+#include <malloc/malloc.h>
+
+T_GLOBAL_META(T_META_RUN_CONCURRENTLY(true));
+
+T_DECL(malloc_zone_batch, "malloc_zone_batch_malloc and malloc_zone_batch_free")
+{
+	const unsigned count = 10;
+	void **results;
+	unsigned number;
+
+	// Use malloc_zone_batch_malloc() with a size that maps to the tiny
+	// allocator. This should succeed.
+	results = calloc(count, sizeof(void *));
+	number = malloc_zone_batch_malloc(malloc_default_zone(), 32, results, count);
+	T_ASSERT_EQ(number, count, "allocated from tiny zone");
+	for (int i = 0; i < count; i++) {
+		T_QUIET; T_ASSERT_NOTNULL(results[i], "pointer %d is NULL", i);
+	}
+	malloc_zone_batch_free(malloc_default_zone(), results, count);
+	free(results);
+
+	// Use malloc_zone_batch_malloc() with a size that maps to the small
+	// allocator. This should fail.
+	results = calloc(count, sizeof(void *));
+	number = malloc_zone_batch_malloc(malloc_default_zone(), 2048, results, count);
+	T_ASSERT_EQ(0, number, "could not allocat from small zone");
+	for (int i = 0; i < count; i++) {
+		T_QUIET; T_ASSERT_NULL(results[i], "pointer %d is not NULL", i);
+	}
+	malloc_zone_batch_free(malloc_default_zone(), results, count);
+	free(results);
+}
+
+T_DECL(malloc_zone_batch_zero_on_free, "malloc_zone_batch_free must zero-on-free",
+		T_META_ENVVAR("MallocZeroOnFree=1"),
+		T_META_ENVVAR("MallocCheckZeroOnFreeCorruption=1"),
+		T_META_ENVVAR("MallocNanoZone=0"))
+{
+	const int n = 3;
+	const size_t size = 272;
+	void *allocations[n];
+	for (int i = 0; i < n; i++) {
+		allocations[i] = malloc(size);
+		T_QUIET; T_ASSERT_NOTNULL(allocations[i], "malloc()");
+		memset(allocations[i], 'a', size);
+	}
+
+	malloc_zone_batch_free(malloc_default_zone(), allocations, n);
+
+	char *allocation = calloc(1, size);
+	T_QUIET; T_ASSERT_NOTNULL(allocation, "calloc()");
+
+	for (size_t i = 0; i < size; i++) {
+		T_QUIET; T_ASSERT_EQ(allocation[i], '\0', "byte %zu should be 0", i);
+	}
+
+	free(allocation);
+
+	T_PASS("Successful calloc after batch free");
+}