pool: Add memory pool API.

2 files changed, 92 insertions, 1 deletions

diff --git a/libdimension/tests/Makefile.am b/libdimension/tests/Makefile.am
index bcfa2c6..5238e7b 100644
--- a/libdimension/tests/Makefile.am
+++ b/libdimension/tests/Makefile.am
@@ -1,5 +1,5 @@
 ###########################################################################
-## Copyright (C) 2009-2012 Tavian Barnes <tavianator@tavianator.com>     ##
+## Copyright (C) 2009-2014 Tavian Barnes <tavianator@tavianator.com>     ##
 ##                                                                       ##
 ## This file is part of The Dimension Build Suite.                       ##
 ##                                                                       ##
@@ -26,6 +26,7 @@ check_PROGRAMS    = warning.test                                               \
                     warning-as-error.test                                      \
                     error.test                                                 \
                     custom-error-fn.test                                       \
+                    pool.test                                                  \
                     refcount.test                                              \
                     dictionary.test                                            \
                     polynomial.test                                            \
@@ -74,6 +75,9 @@ error_test_LDADD   = libdimension-tests.la
 custom_error_fn_test_SOURCES = custom-error-fn.c
 custom_error_fn_test_LDADD   = libdimension-tests.la
 
+pool_test_SOURCES = pool.c
+pool_test_LDADD   = libdimension-unit-test.la
+
 refcount_test_SOURCES = refcount.c
 refcount_test_LDADD   = libdimension-unit-test.la
 
diff --git a/libdimension/tests/pool.c b/libdimension/tests/pool.c
new file mode 100644
index 0000000..68ecfae
--- /dev/null
+++ b/libdimension/tests/pool.c
@@ -0,0 +1,87 @@
+/*************************************************************************
+ * Copyright (C) 2014 Tavian Barnes <tavianator@tavianator.com>          *
+ *                                                                       *
+ * This file is part of The Dimension Test Suite.                        *
+ *                                                                       *
+ * The Dimension Test Suite is free software; you can redistribute it    *
+ * and/or modify it under the terms of the GNU General Public License as *
+ * published by the Free Software Foundation; either version 3 of the    *
+ * License, or (at your option) any later version.                       *
+ *                                                                       *
+ * The Dimension Test Suite is distributed in the hope that it will be   *
+ * useful, but WITHOUT ANY WARRANTY; without even the implied warranty   *
+ * of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU  *
+ * General Public License for more details.                              *
+ *                                                                       *
+ * You should have received a copy of the GNU General Public License     *
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>. *
+ *************************************************************************/
+
+/**
+ * @file
+ * Tests for memory pools.
+ */
+
+#include "../dimension-internal.h"
+#include "tests.h"
+#include "../future.c"
+#include "../threads.c"
+
+static dmnsn_pool *pool;
+
+DMNSN_TEST_SETUP(pool)
+{
+  pool = dmnsn_new_pool();
+}
+
+DMNSN_TEST_TEARDOWN(pool)
+{
+  dmnsn_delete_pool(pool);
+}
+
+DMNSN_TEST(pool, simple)
+{
+  for (int i = 0; i < 10000; ++i) {
+    int *p = DMNSN_POOL_ALLOC(pool, int);
+    *p = i;
+  }
+
+  /* Leak checking will tell us if something bad happened */
+}
+
+static int counter = 0;
+
+static void
+callback(void *ptr)
+{
+  ++counter;
+}
+
+DMNSN_TEST(pool, callback)
+{
+  dmnsn_pool_alloc(pool, sizeof(int), NULL);
+  dmnsn_pool_alloc(pool, sizeof(int), callback);
+  dmnsn_pool_alloc(pool, sizeof(int), callback);
+  dmnsn_pool_alloc(pool, sizeof(int), NULL);
+
+  dmnsn_delete_pool(pool);
+  pool = NULL;
+
+  ck_assert_int_eq(counter, 2);
+}
+
+static int
+alloc_thread(void *ptr, unsigned int thread, unsigned int nthreads)
+{
+  for (unsigned int i = thread; i < 10000; i += nthreads) {
+    int *p = DMNSN_POOL_ALLOC(pool, int);
+    *p = i;
+  }
+  return 0;
+}
+
+DMNSN_TEST(pool, threaded)
+{
+  int ret = dmnsn_execute_concurrently(NULL, alloc_thread, NULL, 64);
+  ck_assert_int_eq(ret, 0);
+}