Merge branch 'dl/complete-stash'
[git] / mem-pool.h
1 #ifndef MEM_POOL_H
2 #define MEM_POOL_H
3
4 struct mp_block {
5         struct mp_block *next_block;
6         char *next_free;
7         char *end;
8         uintmax_t space[FLEX_ARRAY]; /* more */
9 };
10
11 struct mem_pool {
12         struct mp_block *mp_block;
13
14         /*
15          * The amount of available memory to grow the pool by.
16          * This size does not include the overhead for the mp_block.
17          */
18         size_t block_alloc;
19
20         /* The total amount of memory allocated by the pool. */
21         size_t pool_alloc;
22 };
23
24 /*
25  * Initialize mem_pool with specified initial size.
26  */
27 void mem_pool_init(struct mem_pool *pool, size_t initial_size);
28
29 /*
30  * Discard all the memory the memory pool is responsible for.
31  */
32 void mem_pool_discard(struct mem_pool *pool, int invalidate_memory);
33
34 /*
35  * Alloc memory from the mem_pool.
36  */
37 void *mem_pool_alloc(struct mem_pool *pool, size_t len);
38
39 /*
40  * Allocate and zero memory from the memory pool.
41  */
42 void *mem_pool_calloc(struct mem_pool *pool, size_t count, size_t size);
43
44 /*
45  * Allocate memory from the memory pool and copy str into it.
46  */
47 char *mem_pool_strdup(struct mem_pool *pool, const char *str);
48 char *mem_pool_strndup(struct mem_pool *pool, const char *str, size_t len);
49
50 /*
51  * Move the memory associated with the 'src' pool to the 'dst' pool. The 'src'
52  * pool will be empty and not contain any memory. It still needs to be free'd
53  * with a call to `mem_pool_discard`.
54  */
55 void mem_pool_combine(struct mem_pool *dst, struct mem_pool *src);
56
57 /*
58  * Check if a memory pointed at by 'mem' is part of the range of
59  * memory managed by the specified mem_pool.
60  */
61 int mem_pool_contains(struct mem_pool *pool, void *mem);
62
63 #endif