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
|
#include <malloc.h>
#include <slab_alloc.h>
#include <string.h>
#include <syscall.h>
#include <region_alloc.h>
static void* heap_alloc_pages(size_t s) {
void* addr = region_alloc(s, "Heap");
if (addr == 0) return 0;
bool map_ok = mmap(addr, s, FM_READ | FM_WRITE);
if (!map_ok) {
region_free(addr);
return 0;
}
return addr;
}
static void heap_free_pages(void* addr) {
munmap(addr);
region_free(addr);
}
static mem_allocator_t *mem_allocator;
static slab_type_t slab_sizes[] = {
{ "8B malloc objects", 8, 2 },
{ "16B malloc objects", 16, 2 },
{ "32B malloc objects", 32, 2 },
{ "64B malloc objects", 64, 4 },
{ "128B malloc objects", 128, 4 },
{ "256B malloc objects", 256, 4 },
{ "512B malloc objects", 512, 8 },
{ "1KB malloc objects", 1024, 8 },
{ "2KB malloc objects", 2048, 16 },
{ "4KB malloc objects", 4096, 16 },
{ 0, 0, 0 }
};
bool mmap_single_page(void* addr) {
return mmap(addr, PAGE_SIZE, MM_READ | MM_WRITE);
}
void malloc_setup() {
region_allocator_init((void*)0x40000000, (void*)0x40000000, (void*)0xB0000000, mmap_single_page);
mem_allocator = create_slab_allocator(slab_sizes, heap_alloc_pages, heap_free_pages);
ASSERT(mem_allocator != 0);
}
void* malloc(size_t size) {
if (size == 0) return 0;
return slab_alloc(mem_allocator, size);
}
void* calloc(size_t nmemb, size_t sz) {
void* r = malloc(nmemb * sz);
if (r != 0) memset(r, 0, nmemb * sz);
return r;
}
void free(void* ptr) {
slab_free(mem_allocator, ptr);
}
/* vim: set ts=4 sw=4 tw=0 noet :*/
|
