rspamd/src/mem_pool.h

#ifndef RSPAMD_MEM_POOL_H
#define RSPAMD_MEM_POOL_H

#include <sys/types.h>
#include <glib.h>

typedef void (*pool_destruct_func)(void *ptr);

struct _pool_chain {
	u_char *begin;
	u_char *pos;
	size_t len;
	struct _pool_chain *next;
};

struct _pool_chain_shared {
	u_char *begin;
	u_char *pos;
	size_t len;
	gint lock;
	struct _pool_chain_shared *next;
};

struct _pool_destructors {
	pool_destruct_func func;
	void *data;
	struct _pool_destructors *prev;
};

typedef struct memory_pool_s {
	struct _pool_chain *cur_pool;
	struct _pool_chain *first_pool;
	struct _pool_chain_shared *shared_pool;
	struct _pool_destructors *destructors;
} memory_pool_t;

typedef struct memory_pool_stat_s {
	size_t bytes_allocated;
	size_t chunks_allocated;
	size_t shared_chunks_allocated;
	size_t chunks_freed;
} memory_pool_stat_t;

memory_pool_t* memory_pool_new (size_t size);
void* memory_pool_alloc (memory_pool_t* pool, size_t size);
void* memory_pool_alloc0 (memory_pool_t* pool, size_t size);
char* memory_pool_strdup (memory_pool_t* pool, const char *src);
void memory_pool_add_destructor (memory_pool_t *pool, pool_destruct_func func, void *data);
void* memory_pool_alloc_shared (memory_pool_t *pool, size_t size);
void memory_pool_lock_shared (memory_pool_t *pool, void *pointer);
void memory_pool_unlock_shared (memory_pool_t *pool, void *pointer);
void memory_pool_delete (memory_pool_t* pool);

void memory_pool_stat (memory_pool_stat_t *st);

/* Get optimal pool size based on page size for this system */
size_t memory_pool_get_size ();

#define memory_pool_free(x) ((x)->len - ((x)->pos - (x)->begin))

#endif