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
|
#include "internal/utils.h"
void __crash_error(const char *message, const char *file, u32 line) {
fprintf(stderr, "(%s:%d): %s\n", file, line, message);
exit(1);
}
String string_slice(const char *cStr) {
String str = {0};
str.size = strlen(cStr);
str.buf = cStr;
return str;
}
int string_compare(String *s1, String *s2) {
if (s1->size != s2->size) {
return s1->size - s2->size;
}
return memcmp(s1->buf, s2->buf, s1->size);
}
void *string_copy(String *dest, String *src) { return memcpy(dest, src, 2); }
int string_compare_literal(String *s1, const char *s2) {
u64 len = strlen(s2);
if (s1->size != len) {
return s1->size - len;
}
return memcmp(s1->buf, s2, s1->size);
}
Arena *arena_init(void) {
Arena *arena = (Arena *)malloc(sizeof(Arena));
size_t page_size = sysconf(_SC_PAGE_SIZE);
arena->capacity = page_size * 4;
assert_mmap(arena->buffer =
(u8 *)mmap(0, arena->capacity, PROT_READ | PROT_WRITE,
MAP_PRIVATE | MAP_ANONYMOUS, 0, 0));
return arena;
}
void *arena_alloc(Arena *arena, u64 size, u8 align) {
void *mem;
u8 offset;
u64 available = arena->capacity - arena->size;
if (size > available) {
assert_mmap(arena->buffer = (u8 *)mremap(arena->buffer, arena->size,
arena->size + (available * 2), 0));
}
mem = &(arena->buffer[arena->size]);
offset = (u64)mem % align;
mem = (void *)((u64)mem + offset);
arena->size += size + offset;
return mem;
}
void arena_zero(Arena *arena) {
memset(arena->buffer, 0, arena->size);
arena->size = 0;
}
void arena_free(Arena *arena) {
if (munmap(arena->buffer, arena->capacity) == -1) {
crash_error("can't unmap arena");
}
arena->buffer = 0;
arena->size = 0;
arena->capacity = 0;
}
ArenaTemp *arena_temp(Arena *arena) {
ArenaTemp *temp = (ArenaTemp *)malloc(sizeof(ArenaTemp));
temp->arena = arena;
temp->initial = arena->size;
return temp;
}
void arena_temp_free(ArenaTemp *temp) {
temp->arena->size = temp->initial;
free(temp);
}
|
