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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
|
#ifndef UTILS_H
#define UTILS_H
#include <fcntl.h>
#include <stdint.h>
#include <sys/mman.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#ifdef _GNU_SOURCE
#include <stdalign.h>
#endif
typedef signed char i8;
typedef unsigned char u8;
typedef signed short int i16;
typedef unsigned short int u16;
typedef signed int i32;
typedef unsigned int u32;
typedef signed long int i64;
typedef unsigned long int u64;
typedef struct {
u64 size;
const char *buf;
} String;
String string_slice(const char *cStr);
int string_compare(String *s1, String *s2);
void *string_copy(String *dest, String *src);
int string_compare_literal(String *s1, const char *s2);
typedef struct {
u64 size;
u64 capacity;
u8 *buffer;
} Arena;
Arena *arena_init(void);
void *arena_alloc(Arena *arena, u64 size, u8 align);
void arena_zero(Arena *arena);
void arena_free(Arena *arena);
typedef struct {
Arena *arena;
u64 initial;
} ArenaTemp;
ArenaTemp *arena_temp(Arena *arena);
void arena_temp_free(ArenaTemp *temp);
void __crash_error(const char *message, const char *file, u32 line);
#define crash_error(message) __crash_error(message, __FILE__, __LINE__)
#define crash_errno() __crash_error(strerror(errno), __FILE__, __LINE__)
#define assert_mmap(expr) \
if ((expr) == MAP_FAILED) \
crash_errno()
#define assert_errno(expr) \
(expr); \
if (errno != 0) \
crash_errno()
#define assert_success(actual, expected) \
if ((actual) != expected) \
crash_error("assertion failed")
#define assert_failure(actual, expected) \
if ((actual) == expected) \
crash_error("assertion failed")
#if __STDC_VERSION__ >= 202311L
#define arena_type(arena, type) \
(type *)(arena_alloc(arena, sizeof(type), alignof(type)))
#define arena_array(arena, type, size) \
(type *)(arena_alloc(arena, sizeof(type) * size, alignof(type)))
#elif __STDC_VERSION__ >= 201112L
(type *)(arena_alloc(arena, sizeof(type), _Alignof(type)))
#define arena_array(arena, type, size) \
(type *)(arena_alloc(arena, sizeof(type) * size, _Alignof(type)))
#else
#define arena_type(arena, type) \
(type *)(arena_alloc(arena, sizeof(type), __alignof__(type)))
#define arena_array(arena, type, size) \
(type *)(arena_alloc(arena, sizeof(type) * size, __alignof__(type)))
#endif
#ifdef UTILS_IMPLEMENTATION
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) {
arena->size = 0;
memset(arena->buffer, 0, arena->size);
}
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);
}
#endif
#endif
|
