Add aligned allocation interface and use it in ZixBTree

1 files changed, 24 insertions, 0 deletions

diff --git a/test/failing_allocator.c b/test/failing_allocator.c
index eda762b..684a8ec 100644
--- a/test/failing_allocator.c
+++ b/test/failing_allocator.c
@@ -66,6 +66,28 @@ zix_failing_free(ZixAllocator* const allocator, void* const ptr)
   base->free(base, ptr);
 }
 
+ZIX_MALLOC_FUNC
+static void*
+zix_failing_aligned_alloc(ZixAllocator* const allocator,
+                          const size_t        alignment,
+                          const size_t        size)
+{
+  ZixFailingAllocator* const state = (ZixFailingAllocator*)allocator;
+  ZixAllocator* const        base  = zix_default_allocator();
+
+  return attempt(state) ? base->aligned_alloc(base, alignment, size) : NULL;
+}
+
+static void
+zix_failing_aligned_free(ZixAllocator* const allocator, void* const ptr)
+{
+  (void)allocator;
+
+  ZixAllocator* const base = zix_default_allocator();
+
+  base->aligned_free(base, ptr);
+}
+
 ZIX_CONST_FUNC
 ZixFailingAllocator
 zix_failing_allocator(void)
@@ -76,6 +98,8 @@ zix_failing_allocator(void)
       zix_failing_calloc,
       zix_failing_realloc,
       zix_failing_free,
+      zix_failing_aligned_alloc,
+      zix_failing_aligned_free,
     },
     0,
     SIZE_MAX,