Loading...
--- /dev/null
+++ libmalloc/libmalloc-792.60.6/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_ALL_ALLOCATORS);
+
+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_PREFERRED)
+{
+#if TARGET_OS_WATCH && !defined(__LP64__)
+ 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");
+}