1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
|
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <stdalign.h>
#include "wv_mem.h"
#include "wv_err.h"
#define WV_ALIGNMENT alignof(max_align_t)
#define WV_ALIGN_MASK (WV_ALIGNMENT - 1)
#define WV_ARENA_MIN_SIZE (256 * 1024)
#define WV_ARENA_RESERVED WV_ALIGNMENT /* Reserve 0 as the sentinel "null" ref */
struct wv_arena *wv_arena_create(size_t n)
{
struct wv_arena *arena;
if (n < WV_ARENA_MIN_SIZE)
n = WV_ARENA_MIN_SIZE;
arena = malloc(sizeof(struct wv_arena));
if (!arena)
err(1, "malloc arena struct");
arena->buf = malloc(n);
if (!arena->buf)
err(1, "malloc arena buffer");
arena->size = n;
arena->offset = WV_ARENA_RESERVED;
return arena;
}
wv_ref wv_arena_alloc(struct wv_arena *arena, size_t n)
{
wv_ref ref;
unsigned char *new_buf;
size_t aligned_n, new_size;
if (n == 0)
errx(1, "wv_alloc: zero-size allocation");
if (n > SIZE_MAX - WV_ALIGN_MASK)
errx(1, "wv_alloc: allocation size overflow");
aligned_n = (n + WV_ALIGN_MASK) & ~WV_ALIGN_MASK;
if (aligned_n > SIZE_MAX - arena->offset)
errx(1, "wv_alloc: arena offset overflow");
if (arena->offset + aligned_n > arena->size) {
new_size = arena->size;
while (new_size < arena->offset + aligned_n) {
if (new_size > SIZE_MAX / 2)
errx(1, "wv_alloc: arena growth overflow");
new_size *= 2;
}
new_buf = realloc(arena->buf, new_size);
if (!new_buf)
err(1, "realloc arena failed at %zu bytes", new_size);
arena->buf = new_buf;
arena->size = new_size;
}
ref = (wv_ref)arena->offset;
arena->offset += aligned_n;
return ref;
}
void wv_arena_reset(struct wv_arena *arena)
{
if (!arena)
return;
memset(arena->buf, 0, arena->offset);
arena->offset = WV_ARENA_RESERVED;
}
void wv_arena_destroy(struct wv_arena *arena)
{
if (!arena)
return;
free(arena->buf);
free(arena);
}
wv_ref wv_arena_push_string(struct wv_arena *arena, const char *src,
size_t len)
{
wv_ref ref = wv_arena_alloc(arena, len + 1);
char *dst = (char *)WV_ADDR(arena, ref);
memcpy(dst, src, len);
dst[len] = '\0';
return ref;
}
|
