/*      $OpenBSD: pool.h,v 1.2 2017/08/11 14:58:56 jasper Exp $ */

/*
 * Copyright (c) 2017 Martin Pieuchot
 *
 * Permission to use, copy, modify, and 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.
 */

#ifndef _POOL_H_
#define _POOL_H_

#ifndef NOPOOL

struct pool {
        SIMPLEQ_ENTRY(pool)      pr_list;       /* list of all pools */
        const char              *pr_name;       /* identifier */
        SLIST_HEAD(, pool_item)  pr_free;       /* free list */
        size_t                   pr_nmemb;      /* # of items per allocation */
        size_t                   pr_size;       /* size of an item */
        size_t                   pr_nitems;     /* # of available items */
        size_t                   pr_nfree;      /* # items on the free list */
};

void     pool_init(struct pool *, const char *, size_t, size_t);
void    *pool_get(struct pool *);
void     pool_put(struct pool *, void *);
void     pool_dump(void);

static inline void *
pmalloc(struct pool *pp, size_t sz)
{
        return pool_get(pp);
}

static inline void *
pzalloc(struct pool *pp, size_t sz)
{
        char *p;

        p = pool_get(pp);
        memset(p, 0, sz);

        return p;
}

static inline void
pfree(struct pool *pp, void *p)
{
        pool_put(pp, p);
}

#else /* !NOPOOL */
#define pmalloc(a, b)           malloc(b)
#define pzalloc(a, b)           calloc(1, b)
#define pfree(a, b)             free(b)
#endif /* NOPOOL */

#endif /* _POOL_H_ */