blob: 6d9172a1c3dabbd7832b1e5490db050aa03e023a (
plain)
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
|
#include <stdint.h>
#include <stdlib.h>
#include <string.h>
#include "triton_p.h"
struct _mempool_t
{
int size;
struct list_head items;
spinlock_t lock;
uint64_t magic;
};
struct _item_t
{
struct _mempool_t *owner;
struct list_head entry;
uint64_t magic;
char ptr[0];
};
__export mempool_t *mempool_create(int size)
{
struct _mempool_t *p = malloc(sizeof(*p));
memset(p, 0, sizeof(*p));
INIT_LIST_HEAD(&p->items);
spinlock_init(&p->lock);
p->size = size;
p->magic = (uint64_t)random() * (uint64_t)random();
return (mempool_t *)p;
}
__export void *mempool_alloc(mempool_t *pool)
{
struct _mempool_t *p = (struct _mempool_t *)pool;
struct _item_t *it;
spin_lock(&p->lock);
if (!list_empty(&p->items)) {
it = list_entry(p->items.next, typeof(*it), entry);
list_del(&it->entry);
spin_unlock(&p->lock);
return it->ptr;
}
spin_unlock(&p->lock);
it = malloc(sizeof(*it) + p->size);
if (!it) {
triton_log_error("mempool: out of memory\n");
return NULL;
}
it->owner = p;
it->magic = p->magic;
return it->ptr;
}
__export void mempool_free(void *ptr)
{
struct _item_t *it = container_of(ptr, typeof(*it), ptr);
if (it->magic != it->owner->magic) {
triton_log_error("mempool: memory corruption detected");
abort();
}
spin_lock(&it->owner->lock);
list_add_tail(&it->entry,&it->owner->items);
spin_unlock(&it->owner->lock);
}
|
