1 |
|
|
/* $OpenBSD: pool.h,v 1.2 2017/08/11 14:58:56 jasper Exp $ */ |
2 |
|
|
|
3 |
|
|
/* |
4 |
|
|
* Copyright (c) 2017 Martin Pieuchot |
5 |
|
|
* |
6 |
|
|
* Permission to use, copy, modify, and distribute this software for any |
7 |
|
|
* purpose with or without fee is hereby granted, provided that the above |
8 |
|
|
* copyright notice and this permission notice appear in all copies. |
9 |
|
|
* |
10 |
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES |
11 |
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF |
12 |
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR |
13 |
|
|
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES |
14 |
|
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN |
15 |
|
|
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF |
16 |
|
|
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. |
17 |
|
|
*/ |
18 |
|
|
|
19 |
|
|
#ifndef _POOL_H_ |
20 |
|
|
#define _POOL_H_ |
21 |
|
|
|
22 |
|
|
#ifndef NOPOOL |
23 |
|
|
|
24 |
|
|
struct pool { |
25 |
|
|
SIMPLEQ_ENTRY(pool) pr_list; /* list of all pools */ |
26 |
|
|
const char *pr_name; /* identifier */ |
27 |
|
|
SLIST_HEAD(, pool_item) pr_free; /* free list */ |
28 |
|
|
size_t pr_nmemb; /* # of items per allocation */ |
29 |
|
|
size_t pr_size; /* size of an item */ |
30 |
|
|
size_t pr_nitems; /* # of available items */ |
31 |
|
|
size_t pr_nfree; /* # items on the free list */ |
32 |
|
|
}; |
33 |
|
|
|
34 |
|
|
void pool_init(struct pool *, const char *, size_t, size_t); |
35 |
|
|
void *pool_get(struct pool *); |
36 |
|
|
void pool_put(struct pool *, void *); |
37 |
|
|
void pool_dump(void); |
38 |
|
|
|
39 |
|
|
static inline void * |
40 |
|
|
pmalloc(struct pool *pp, size_t sz) |
41 |
|
|
{ |
42 |
|
|
return pool_get(pp); |
43 |
|
|
} |
44 |
|
|
|
45 |
|
|
static inline void * |
46 |
|
|
pzalloc(struct pool *pp, size_t sz) |
47 |
|
|
{ |
48 |
|
|
char *p; |
49 |
|
|
|
50 |
|
|
p = pool_get(pp); |
51 |
|
|
memset(p, 0, sz); |
52 |
|
|
|
53 |
|
|
return p; |
54 |
|
|
} |
55 |
|
|
|
56 |
|
|
static inline void |
57 |
|
|
pfree(struct pool *pp, void *p) |
58 |
|
|
{ |
59 |
|
|
pool_put(pp, p); |
60 |
|
|
} |
61 |
|
|
|
62 |
|
|
#else /* !NOPOOL */ |
63 |
|
|
#define pmalloc(a, b) malloc(b) |
64 |
|
|
#define pzalloc(a, b) calloc(1, b) |
65 |
|
|
#define pfree(a, b) free(b) |
66 |
|
|
#endif /* NOPOOL */ |
67 |
|
|
|
68 |
|
|
#endif /* _POOL_H_ */ |