diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/smrt_arena.c | 4 | ||||
| -rw-r--r-- | src/smrt_arena.h | 1 | ||||
| -rw-r--r-- | src/string.c | 36 | ||||
| -rw-r--r-- | src/string.h | 6 |
4 files changed, 43 insertions, 4 deletions
diff --git a/src/smrt_arena.c b/src/smrt_arena.c index 683afcd..0fb8ab7 100644 --- a/src/smrt_arena.c +++ b/src/smrt_arena.c @@ -127,6 +127,10 @@ void smrt_arena_clear(smrt_arena_t *arena, b32 zero_out) { smrt_arena__shrink_commit(arena); } +void smrt_arena_mark(smrt_arena_t *arena) { + arena->mark_pos = arena->pos; +} + void smrt_arena_destroy(smrt_arena_t *arena) { plat_mem_release(arena, arena->reserve_size); } diff --git a/src/smrt_arena.h b/src/smrt_arena.h index 4e6f592..375929d 100644 --- a/src/smrt_arena.h +++ b/src/smrt_arena.h @@ -23,6 +23,7 @@ smrt_arena_t *smrt_arena_create(u64 reserve_size, u64 commit_size, b32 auto_ void smrt_arena_pop_to(smrt_arena_t *arena, u64 location); b32 smrt_arena_pop_to_mark(smrt_arena_t *arena); void smrt_arena_clear(smrt_arena_t *arena, b32 zero_out); + void smrt_arena_mark(smrt_arena_t *arena); void smrt_arena_destroy(smrt_arena_t *arena); #define SMRTA_ALLOC_ARRAY(arena, T, n) (T*)smrt_arena_push(arena, sizeof(T) * (n), true) diff --git a/src/string.c b/src/string.c index 9328911..a29af19 100644 --- a/src/string.c +++ b/src/string.c @@ -15,7 +15,7 @@ strng_t *strng_new(smrt_arena_t *arena, u64 size) { return s; } -strng_t *strng_from(smrt_arena_t *arena, char *c) { +strng_t *strng_from(smrt_arena_t *arena, char const *c) { u64 size = strlen(c); strng_t *s = strng_new(arena, size); @@ -30,7 +30,38 @@ strng_t *strng_from(smrt_arena_t *arena, char *c) { return s; } -b32 strng_set(strng_t *string, char *c) { +strng_t *strng_dup(smrt_arena_t *arena, strng_t const *src) { + u64 size = src->len + sizeof(strng_t); + + strng_t *string = smrt_arena_push(arena, size, true); + + if (!string) { + return NULL; + } + + string->alloc_size = src->len; + + memcpy((u8*)string+STRNG_BASE_POS, (u8*)src+STRNG_BASE_POS, src->len); + string->len = src->len; + + return string; +} + +char *strng_str(smrt_arena_t *arena, strng_t const *string) { + char *s = SMRTA_ALLOC_ARRAY(arena, char, string->len+1); + + if (!s) { + return NULL; + } else if (string->len == 0) { + return s; + } + + memcpy(s, (u8*)string+STRNG_BASE_POS, string->len); + + return s; +} + +b32 strng_set(strng_t *string, char const *c) { u64 size = strlen(c); if (string->alloc_size < size) { @@ -44,5 +75,6 @@ b32 strng_set(strng_t *string, char *c) { } void strng_clear(strng_t *string) { + memset((u8*)string+STRNG_BASE_POS, 0, string->len); string->len = 0; } diff --git a/src/string.h b/src/string.h index b8554ff..419ef96 100644 --- a/src/string.h +++ b/src/string.h @@ -11,8 +11,10 @@ typedef struct { } strng_t; strng_t *strng_new(smrt_arena_t *arena, u64 size); -strng_t *strng_from(smrt_arena_t *arena, char *c); - b32 strng_set(strng_t *string, char *c); +strng_t *strng_from(smrt_arena_t *arena, char const *c); +strng_t *strng_dup(smrt_arena_t *arena, strng_t const *src); + char *strng_str(smrt_arena_t *arena, strng_t const *string); + b32 strng_set(strng_t *string, char const *c); void strng_clear(strng_t *string); #define STRNG_FMT(s) (int)s->len, (char *)((u8*)s+STRNG_BASE_POS) |