127 lines
3.5 KiB
C
127 lines
3.5 KiB
C
#ifndef ARENA_H
|
|
#define ARENA_H
|
|
|
|
#include "memory.h"
|
|
|
|
#define ARENA_HEADER_SIZE 64
|
|
#define ARENA_BLOCK_SIZE 4096
|
|
|
|
#define arena_push(a, type) ((type *)arena_push_bytes((a), sizeof(type), alignof(type)))
|
|
#define arena_push_no_zero(a, type) ((type *)arena_push_bytes_no_zero((a), sizeof(type), alignof(type)))
|
|
|
|
#define arena_push_array(a, type, n) ((type *)arena_push_bytes((a), (sizeof(type) * (n)), alignof(type)))
|
|
#define arena_push_array_no_zero(a, type, n) ((type *)arena_push_bytes_no_zero((a), (sizeof(type) * (n)), alignof(type)))
|
|
|
|
#define arena_pop(a, type, dst) arena_pop_struct((a), sizeof(type), dst)
|
|
#define arena_pop_array(a, type, n, dst) arena_pop_struct((a), sizeof(type) * (n), dst)
|
|
|
|
/* Returns a pointer to where the next allocation would be (at alignment of type).
|
|
* Equivalent to arena_push but without actually allocating anything or modifying the arena. */
|
|
#define arena_push_dry(a, type) (type *)(_arena_push_dry((a), alignof(type)))
|
|
|
|
struct arena_temp {
|
|
struct arena *arena;
|
|
u64 start_pos;
|
|
|
|
#if RTC
|
|
u64 scratch_id;
|
|
#endif
|
|
};
|
|
|
|
struct arena *arena_alloc(u64 reserve);
|
|
void arena_release(struct arena *arena);
|
|
void *arena_push_bytes_no_zero(struct arena *arena, u64 size, u64 align);
|
|
void arena_copy_replace(struct arena *dst, struct arena *src);
|
|
void arena_decommit_unused_blocks(struct arena *arena);
|
|
void arena_set_readonly(struct arena *arena);
|
|
void arena_set_readwrite(struct arena *arena);
|
|
|
|
INLINE u8 *arena_base(struct arena *arena)
|
|
{
|
|
return (u8 *)arena + ARENA_HEADER_SIZE;
|
|
}
|
|
|
|
INLINE void *arena_push_bytes(struct arena *arena, u64 size, u64 align)
|
|
{
|
|
void *p = arena_push_bytes_no_zero(arena, size, align);
|
|
MEMZERO(p, size);
|
|
return p;
|
|
}
|
|
|
|
INLINE void arena_pop_to(struct arena *arena, u64 pos)
|
|
{
|
|
ASSERT(arena->pos >= pos);
|
|
ASSERT(!arena->readonly);
|
|
|
|
ASAN_POISON(arena_base(arena) + pos, arena->pos - pos);
|
|
arena->pos = pos;
|
|
}
|
|
|
|
INLINE void arena_pop_struct(struct arena *arena, u64 size, void *copy_dst)
|
|
{
|
|
ASSERT(arena->pos >= size);
|
|
ASSERT(!arena->readonly);
|
|
|
|
u64 new_pos = arena->pos - size;
|
|
void *src = (void *)(arena_base(arena) + new_pos);
|
|
MEMCPY(copy_dst, src, size);
|
|
|
|
ASAN_POISON(arena_base(arena) + new_pos, arena->pos - new_pos);
|
|
arena->pos = new_pos;
|
|
}
|
|
|
|
INLINE void *arena_align(struct arena *arena, u64 align)
|
|
{
|
|
ASSERT(!arena->readonly);
|
|
if (align > 0) {
|
|
u64 aligned_start_pos = (arena->pos + (align - 1));
|
|
aligned_start_pos -= aligned_start_pos % align;
|
|
u64 align_bytes = aligned_start_pos - (u64)arena->pos;
|
|
if (align_bytes > 0) {
|
|
return (void *)arena_push_array_no_zero(arena, u8, align_bytes);
|
|
} else {
|
|
return (void *)(arena_base(arena) + arena->pos);
|
|
}
|
|
} else {
|
|
/* 0 alignment */
|
|
ASSERT(false);
|
|
return (void *)(arena_base(arena) + arena->pos);
|
|
}
|
|
}
|
|
|
|
INLINE struct arena_temp arena_temp_begin(struct arena *arena)
|
|
{
|
|
struct arena_temp t = ZI;
|
|
t.arena = arena;
|
|
t.start_pos = arena->pos;
|
|
return t;
|
|
}
|
|
|
|
INLINE void arena_temp_end(struct arena_temp temp)
|
|
{
|
|
arena_pop_to(temp.arena, temp.start_pos);
|
|
}
|
|
|
|
INLINE void arena_reset(struct arena *arena)
|
|
{
|
|
arena_pop_to(arena, 0);
|
|
}
|
|
|
|
INLINE struct string arena_to_string(struct arena *arena)
|
|
{
|
|
struct string b;
|
|
b.text = arena_base(arena);
|
|
b.len = arena->pos;
|
|
return b;
|
|
}
|
|
|
|
INLINE void *_arena_push_dry(struct arena *arena, u64 align)
|
|
{
|
|
u64 aligned_start_pos = (arena->pos + (align - 1));
|
|
aligned_start_pos -= aligned_start_pos % align;
|
|
void *ptr = arena_base(arena) + aligned_start_pos;
|
|
return ptr;
|
|
}
|
|
|
|
#endif
|