Add function to allocate aligned blocks on the memstackHEADmain

1 files changed, 27 insertions, 0 deletions

diff --git a/memstack/test/memstack_test.c b/memstack/test/memstack_test.c
index 5e9b493..285cf46 100644
--- a/memstack/test/memstack_test.c
+++ b/memstack/test/memstack_test.c
@@ -32,6 +32,8 @@ TEST_CASE(memstack_allocate_until_full) {
   }
 
   TEST_TRUE(memstack_size(&stack) == CAPACITY);
+
+  memstack_del(&stack);
 }
 
 // Allocate all N ints, then free them.
@@ -48,6 +50,8 @@ TEST_CASE(memstack_fill_then_free) {
   memstack_clear(&stack);
 
   TEST_EQUAL(memstack_size(&stack), 0);
+
+  memstack_del(&stack);
 }
 
 // Attempt to allocate blocks past the maximum stack size.
@@ -68,6 +72,8 @@ TEST_CASE(memstack_allocate_beyond_max_size) {
   }
 
   TEST_TRUE(memstack_size(&stack) == CAPACITY);
+
+  memstack_del(&stack);
 }
 
 // Free blocks should always remain zeroed out.
@@ -81,6 +87,8 @@ TEST_CASE(memstack_zero_free_blocks_after_creation) {
     TEST_TRUE(block != nullptr);
     TEST_EQUAL(*block, 0);
   }
+
+  memstack_del(&stack);
 }
 
 // Free blocks should always remain zeroed out.
@@ -102,6 +110,25 @@ TEST_CASE(memstack_zero_free_block_after_free) {
     TEST_TRUE(block != nullptr);
     TEST_EQUAL(*block, 0);
   }
+
+  memstack_del(&stack);
+}
+
+// Aligned allocations should be properly aligned.
+TEST_CASE(memstack_alloc_aligned) {
+  memstack stack = {0};
+  memstack_make(&stack, CAPACITY, nullptr);
+
+  // -1 because the base address of the memory storage might be unaligned.
+  for (int i = 0; i < NUM_INTS - 1; ++i) {
+    const int* block =
+        memstack_alloc_aligned(&stack, sizeof(int), alignof(int));
+    TEST_TRUE(block != nullptr);
+    TEST_EQUAL(*block, 0);
+    TEST_EQUAL((uintptr_t)block % alignof(int), 0);
+  }
+
+  memstack_del(&stack);
 }
 
 int main() { return 0; }