1 files changed, 61 insertions, 0 deletions
diff --git a/src/mem_pool.h b/src/mem_pool.h
new file mode 100644
index 000000000..4027b5de2
--- /dev/null
+++ b/src/mem_pool.h
@@ -0,0 +1,61 @@
+#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