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
|
#include "_.h"
#undef free
#undef malloc
#undef calloc
#undef realloc
static void **_alloced = nullptr;
static size_t _alloced_size = 0;
static size_t _alloced_capacity = 0;
void
wdeinit(void)
{
if (_alloced != nullptr) {
rfor (i, 0u, _alloced_size)
free(_alloced[i]);
free(_alloced);
}
_alloced = nullptr;
_alloced_size = 0;
_alloced_capacity = 0;
}
void *
alloc(size_t size)
{
_alloced_size += 1;
if (_alloced == nullptr) {
_alloced = calloc(sizeof(void *), 16);
assert(_alloced != nullptr);
_alloced_capacity = 16;
}
if (_alloced_size > _alloced_capacity) {
size_t new_alloced_capacity = 16;
while (new_alloced_capacity <= _alloced_size)
new_alloced_capacity *= 2;
void *new_alloced =
realloc(_alloced, new_alloced_capacity * sizeof(void *));
_alloced_size -= (new_alloced == nullptr);
assert(new_alloced != nullptr);
_alloced = new_alloced;
_alloced_capacity = new_alloced_capacity;
}
_alloced[_alloced_size - 1] = calloc(1, size);
assert(_alloced[_alloced_size - 1] != nullptr);
return _alloced[_alloced_size - 1];
}
void
_free(void *ptr)
{
rfor (i, 0u, _alloced_size) {
if (_alloced[i] == ptr) {
free(_alloced[i]);
_alloced[i] = nullptr;
rfor (k, i, _alloced_size - 1)
_alloced[k] = _alloced[k + 1];
_alloced_size -= 1;
return;
}
}
pwrn("double free of %p", ptr);
}
void *
_realloc(void *ptr, size_t size)
{
rfor (i, 0u, _alloced_size) {
if (_alloced[i] == ptr) {
void *new_ptr = realloc(ptr, size);
assert(new_ptr != nullptr);
_alloced[i] = new_ptr;
return new_ptr;
}
}
panic("fucked up realloc, have fun debugging");
__builtin_unreachable();
}
|