feat(libhectic): arenas

This commit is contained in:
2025-03-21 01:53:02 +00:00
parent 647155f859
commit 4f2391e9b5
8 changed files with 239 additions and 5 deletions

View File

@@ -74,3 +74,46 @@ char* log_message(LogLevel level, int line, const char *format, ...) {
return timeStr;
}
// -----------
// -- arena --
// -----------
Arena arena_init(size_t size) {
Arena arena;
arena.begin = malloc(size);
memset(arena.begin, 0, size);
arena.current = arena.begin;
arena.capacity = size;
return arena;
}
void *arena_alloc_or_null(Arena *arena, size_t size) {
if (arena->begin == 0) {
*arena = arena_init(ARENA_DEFAULT_SIZE);
}
size_t current = (size_t)arena->current - (size_t)arena->begin;
// TODO(yukkop): maybe -1
if (arena->capacity <= current && current < size) {
return NULL;
}
return arena;
}
void arena_reset(Arena *arena) {
arena->current = arena->begin;
}
void arena_free(Arena *arena) {
free(arena->begin);
}
void *arena_alloc(Arena *arena, size_t size) {
void *mem = arena_alloc_or_null(arena, size);
if (!mem) {
raise_exception("Arena out of memory");
exit(1);
}
return mem;
}