/*
 * Copyright (c) 1999-2006 Christophe Fillot.
 * E-mail: cf@utc.fr
 *
 * mempool.h: Simple Memory Pools.
 */

#ifndef __MEMPOOL_H__
#define __MEMPOOL_H__  1

#include <sys/types.h>
#include <sys/time.h>
#include <pthread.h>

#include "utils.h"

/* Memory Pool "Fixed" Flag */
#define MEMPOOL_FIXED  1

/* Dummy value used to check if a memory block is invalid */
#define MEMBLOCK_TAG  0xdeadbeef

typedef struct memblock memblock_t;
typedef struct mempool mempool_t;

/* Memory block */
struct memblock {
    int tag;                    /* MEMBLOCK_TAG if block is valid */
    size_t block_size;          /* Block size (without header) */
    memblock_t *next, *prev;    /* Double linked list pointers */
    mempool_t *pool;            /* Pool which contains this block */
    m_uint64_t data[0];         /* Memory block itself */
};

/* Memory Pool */
struct mempool {
    memblock_t *block_list;     /* Double-linked block list */
    pthread_mutex_t lock;       /* Mutex for managing pool */
    char *name;                 /* Name of this pool */
    int flags;                  /* Flags */
    int nr_blocks;              /* Number of blocks in this pool */
    size_t total_size;          /* Total bytes allocated */
    size_t max_size;            /* Maximum memory */
};

/* Lock and unlock access to a memory pool */
#define MEMPOOL_LOCK(mp)    pthread_mutex_lock(&(mp)->lock)
#define MEMPOOL_UNLOCK(mp)  pthread_mutex_unlock(&(mp)->lock)

/* Callback function for use with mp_foreach */
typedef void (*mp_foreach_cbk) (memblock_t * block, void *user_arg);

/* Execute an action for each block in specified pool */
static inline void mp_foreach (mempool_t * pool, mp_foreach_cbk cbk,
    void *arg)
{
    memblock_t *mb;

    for (mb = pool->block_list; mb; mb = mb->next)
        cbk (mb, arg);
}

/* Allocate a new block in specified pool */
void *mp_alloc (mempool_t * pool, size_t size);

/* Allocate a new block which will not be zeroed */
void *mp_alloc_n0 (mempool_t * pool, size_t size);

/* Reallocate a block */
void *mp_realloc (void *addr, size_t new_size);

/* Allocate a new memory block and copy data into it */
void *mp_dup (mempool_t * pool, void *data, size_t size);

/* Duplicate specified string and insert it in a memory pool */
char *mp_strdup (mempool_t * pool, char *str);

/* Free block at specified address */
int mp_free (void *addr);

/* Free block at specified address and clean pointer */
int mp_free_ptr (void *addr);

/* Free all blocks of specified pool */
void mp_free_all_blocks (mempool_t * pool);

/* Free specified memory pool */
void mp_free_pool (mempool_t * pool);

/* Create a new pool in a fixed memory area */
mempool_t *mp_create_fixed_pool (mempool_t * mp, char *name);

/* Create a new pool */
mempool_t *mp_create_pool (char *name);

#endif