Loading...
tests/aligned_alloc_test.c /dev/null libmalloc-715.120.13
--- /dev/null
+++ libmalloc/libmalloc-715.120.13/tests/aligned_alloc_test.c
@@ -0,0 +1,104 @@
+//
+//  aligned_alloc_test.c
+//  libmalloc
+//
+//  test allocating and freeing all sizes and alignments
+//
+
+#include <darwintest.h>
+#include <errno.h>
+#include <malloc/malloc.h>
+#include <stdint.h>
+#include <stdio.h>
+#include <stdlib.h>
+
+T_GLOBAL_META(T_META_RUN_CONCURRENTLY(true), T_META_TAG_XZONE);
+
+static inline void*
+t_aligned_alloc(size_t alignment, size_t size)
+{
+	void *ptr = aligned_alloc(alignment, size);
+	size_t allocated_size = malloc_size(ptr);
+
+	T_QUIET; T_ASSERT_NOTNULL(ptr, "allocation");
+	T_QUIET; T_ASSERT_EQ((intptr_t)ptr % alignment, 0ul, "pointer should be properly aligned");
+	T_QUIET; T_EXPECT_LE(size, allocated_size, "allocation size");
+
+	// Scribble memory pointed to by `ptr` to make sure we're not using that
+	// memory for control structures. This also makes sure the memory can be
+	// written to.
+	const uint64_t pat = 0xdeadbeefcafebabeull;
+	memset_pattern8(ptr, &pat, size);
+	return ptr;
+}
+
+T_DECL(aligned_alloc_free, "aligned_alloc all power of two alignments <= 64kb",
+		T_META_TAG_VM_NOT_PREFERRED)
+{
+#if TARGET_OS_WATCH
+	const size_t max_alignment = 4096;
+#else
+	const size_t max_alignment = 64 * 1024;
+#endif // TARGET_OS_WATCH
+	for (size_t alignment = sizeof(void*); alignment < max_alignment;
+			alignment *= 2) {
+		// test several sizes that are multiples of the alignment
+		for (size_t size = alignment; size <= 256*alignment; size += alignment) {
+			const int num_ptrs = 10;
+			void *ptrs[num_ptrs];
+			for (int i = 0; i < num_ptrs; i++) {
+				ptrs[i] = t_aligned_alloc(alignment, size);
+			}
+			for (int i = 0; i < num_ptrs; i++) {
+				free(ptrs[i]);
+			}
+		}
+	}
+}
+
+T_DECL(aligned_alloc_alignment_not_multiple_of_size, "aligned_alloc should set errno to EINVAL if size is not a multiple of alignment",
+		T_META_TAG_VM_PREFERRED)
+{
+	{
+		void *ptr = aligned_alloc(8, 12);
+		T_QUIET; T_ASSERT_NULL(ptr, "ptr should be null");
+		T_QUIET; T_ASSERT_EQ(errno, EINVAL, "errno should be EINVAL");
+	}
+
+	{
+		void *ptr = aligned_alloc(32, 16);
+		T_QUIET; T_ASSERT_NULL(ptr, "ptr should be null");
+		T_QUIET; T_ASSERT_EQ(errno, EINVAL, "errno should be EINVAL");
+	}
+}
+
+T_DECL(aligned_alloc_alignment_not_power_of_two, "aligned_alloc should set errno to EINVAL if alignment is not a power of two (implementation constraint)",
+		T_META_TAG_VM_PREFERRED)
+{
+#pragma GCC diagnostic push
+#pragma GCC diagnostic ignored "-Wnon-power-of-two-alignment"
+	{
+		void *ptr = aligned_alloc(24, 48); // alignment is even, but not a power of two
+		T_QUIET; T_ASSERT_NULL(ptr, "ptr should be null");
+		T_QUIET; T_ASSERT_EQ(errno, EINVAL, "errno should be EINVAL");
+	}
+
+	{
+		void *ptr = aligned_alloc(23, 46); // alignment is odd, and not a power of two
+		T_QUIET; T_ASSERT_NULL(ptr, "ptr should be null");
+		T_QUIET; T_ASSERT_EQ(errno, EINVAL, "errno should be EINVAL");
+	}
+#pragma GCC diagnostic pop
+}
+
+T_DECL(aligned_alloc_alignment_not_a_multiple_of_voidstar, "aligned_alloc should set errno to EINVAL if alignment is not a multiple of sizeof(void*) (implementation constraint)",
+		T_META_TAG_VM_PREFERRED)
+{
+	const size_t alignment = sizeof(void*)+1;
+#pragma GCC diagnostic push
+#pragma GCC diagnostic ignored "-Wnon-power-of-two-alignment"
+	void *ptr = aligned_alloc(alignment, alignment * 2);
+#pragma GCC diagnostic pop
+	T_QUIET; T_ASSERT_NULL(ptr, "ptr should be null");
+	T_QUIET; T_ASSERT_EQ(errno, EINVAL, "aligned_alloc should set errno to EINVAL");
+}