summaryrefslogtreecommitdiff
path: root/src/include/mem_pool.h
blob: c57b70761eb57e0debddf991f2e8a7927fc9544e (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
/*
 * This file is part of the coreboot project.
 *
 * Copyright 2015 Google Inc.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; version 2 of the License.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc.
 */

#ifndef _MEM_POOL_H_
#define _MEM_POOL_H_

#include <stddef.h>
#include <stdint.h>

/*
 * The memory pool allows one to allocate memory from a fixed size buffer
 * that also allows freeing semantics for reuse. However, the current
 * limitation is that the most recent allocation is the only one that
 * can be freed. If one tries to free any allocation that isn't the
 * most recently allocated it will result in a leak within the memory pool.
 *
 * The memory returned by allocations are at least 8 byte aligned. Note
 * that this requires the backing buffer to start on at least an 8 byte
 * alignment.
 */

struct mem_pool {
	uint8_t *buf;
	size_t size;
	uint8_t *last_alloc;
	size_t free_offset;
};

#define MEM_POOL_INIT(buf_, size_)	\
	{				\
		.buf = (buf_),		\
		.size = (size_),	\
		.last_alloc = NULL,	\
		.free_offset = 0,	\
	}

static inline void mem_pool_reset(struct mem_pool *mp)
{
	mp->last_alloc = NULL;
	mp->free_offset = 0;
}

/* Initialize a memory pool. */
static inline void mem_pool_init(struct mem_pool *mp, void *buf, size_t sz)
{
	mp->buf = buf;
	mp->size = sz;
	mem_pool_reset(mp);
}

/* Allocate requested size from the memory pool. NULL returned on error. */
void *mem_pool_alloc(struct mem_pool *mp, size_t sz);

/* Free allocation from memory pool. */
void mem_pool_free(struct mem_pool *mp, void *alloc);

#endif /* _MEM_POOL_H_ */