Add helper to allocate rest of a GglArena

aws-greengrass · Oct 31, 2024 · 53a16d1 · 53a16d1
1 parent b94ba6f
commit 53a16d1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/ggl-lib/include/ggl/arena.h b/ggl-lib/include/ggl/arena.h
@@ -46,4 +46,7 @@ GglError ggl_arena_resize_last(
 /// Returns true if arena's mem contains ptr.
 bool ggl_arena_owns(const GglArena *arena, const void *ptr);
 
+/// Allocates remaining space into a buffer.
+GglBuffer ggl_arena_alloc_rest(GglArena *arena);
+
 #endif
diff --git a/ggl-lib/src/arena.c b/ggl-lib/src/arena.c
@@ -3,6 +3,7 @@
 // SPDX-License-Identifier: Apache-2.0
 
 #include "ggl/arena.h"
+#include "ggl/buffer.h"
 #include "ggl/error.h"
 #include "ggl/log.h"
 #include <assert.h>
@@ -93,3 +94,10 @@ bool ggl_arena_owns(const GglArena *arena, const void *ptr) {
     uintptr_t ptr_int = (uintptr_t) ptr;
     return (ptr_int >= mem_int) && (ptr_int < mem_int + arena->CAPACITY);
 }
+
+GglBuffer ggl_arena_alloc_rest(GglArena *arena) {
+    GglBuffer rest = { .data = arena->MEM + arena->index,
+                       .len = arena->CAPACITY - arena->index };
+    arena->index = arena->CAPACITY;
+    return rest;
+}