summaryrefslogtreecommitdiff
path: root/usual/mempool.c
blob: 51ea31720b419ee3c9c5fd16a225a38020e3d80d (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
72
73
74
/*
 * Simple memory pool for variable-length allocations.
 *
 * Copyright (c) 2009 Marko Kreen
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <usual/mempool.h>

/*
 * Allows allocation of several variable-sized objects,
 * freeing them all together.
 *
 * ToDo: make it more 'obstack'-like (???)
 * - free_last
 * - resize_last
 * - append
 */

struct MemPool {
	struct MemPool *prev;
	unsigned size;
	unsigned used;
};

void *mempool_alloc(struct MemPool **pool, unsigned size)
{
	struct MemPool *cur = *pool;
	void *ptr;
	unsigned nsize;

	size = ALIGN(size);
	if (cur && cur->used + size <= cur->size) {
		ptr = (char *)(cur + 1) + cur->used;
		cur->used += size;
		return ptr;
	} else {
		nsize = cur ? (2 * cur->size) : 512;
		while (nsize < size)
			nsize *= 2;
		cur = calloc(1, sizeof(*cur) + nsize);
		if (cur == NULL)
			return NULL;
		cur->used = size;
		cur->size = nsize;
		cur->prev = *pool;
		*pool = cur;
		return (char *)(cur + 1);
	}
}

void mempool_destroy(struct MemPool **pool)
{
	struct MemPool *cur, *tmp;
	if (!pool)
		return;
	for (cur = *pool, *pool = NULL; cur; ) {
		tmp = cur->prev;
		free(cur);
		cur = tmp;
	}
}