Add refpool.

src/Makefile

@@ -2,11 +2,12 @@ SHARED_LIB = ${LIB_PREFIX}corefw${LIB_SUFFIX}
 LIB_MAJOR = 0
 LIB_MINOR = 0
 
-SRCS = array.c	\
-       class.c	\
-       map.c	\
-       object.c	\
-       range.c	\
+SRCS = array.c		\
+       class.c		\
+       map.c		\
+       object.c		\
+       range.c		\
+       refpool.c	\
        string.c
 
 INCLUDES = ${SRCS:.c=.h}	\

src/object.c

@@ -25,8 +25,10 @@
  */
 
 #include <stdlib.h>
+#include <assert.h>
 
 #include "object.h"
+#include "refpool.h"
 
 void*
 cfw_new(CFWClass *class, ...)
@@ -54,6 +56,39 @@ cfw_new(CFWClass *class, ...)
 	return obj;
 }
 
+void*
+cfw_new_p(CFWClass *class, ...)
+{
+	CFWObject *obj;
+
+	assert(class != cfw_refpool);
+
+	if ((obj = malloc(class->size)) == NULL)
+		return NULL;
+
+	obj->cls = class;
+	obj->ref_cnt = 1;
+
+	if (class->ctor != NULL) {
+		va_list args;
+		va_start(args, class);
+
+		if (!class->ctor(obj, args)) {
+			cfw_unref(obj);
+			return NULL;
+		}
+
+		va_end(args);
+	}
+
+	if (!cfw_refpool_add(obj)) {
+		cfw_unref(obj);
+		return NULL;
+	}
+
+	return obj;
+}
+
 void*
 cfw_ref(void *ptr)
 {

src/object.h

@@ -36,6 +36,7 @@ typedef struct CFWObject {
 
 extern CFWClass *cfw_object;
 extern void* cfw_new(CFWClass*, ...);
+extern void* cfw_new_p(CFWClass*, ...);
 extern void* cfw_ref(void*);
 extern void cfw_unref(void*);
 extern void cfw_free(void*);

src/refpool.c

@@ -0,0 +1,109 @@
+/*
+ * Copyright (c) 2012, Jonathan Schleifer <js@webkeks.org>
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice,
+ *    this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright notice,
+ *    this list of conditions and the following disclaimer in the documentation
+ *    and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
+ * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+ * POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include <stdlib.h>
+#include <assert.h>
+
+#include "object.h"
+#include "refpool.h"
+#include "array.h"
+
+struct CFWRefPool {
+	CFWObject obj;
+	void **data;
+	size_t size;
+	CFWRefPool *prev, *next;
+};
+
+static CFWRefPool *top;
+
+static bool
+ctor(void *ptr, va_list args)
+{
+	CFWRefPool *pool = ptr;
+
+	pool->data = NULL;
+	pool->size = 0;
+
+	if (top != NULL) {
+		pool->prev = top;
+		top->next = pool;
+	} else
+		pool->prev = NULL;
+	pool->next = NULL;
+
+	top = pool;
+
+	return true;
+}
+
+static void
+dtor(void *ptr)
+{
+	CFWRefPool *pool = ptr;
+	size_t i;
+
+	if (pool->next != NULL)
+		cfw_unref(pool->next);
+
+	for (i = 0; i < pool->size; i++)
+		cfw_unref(pool->data[i]);
+
+	if (pool->data != NULL)
+		free(pool->data);
+
+	top = pool->prev;
+}
+
+bool
+cfw_refpool_add(void *ptr)
+{
+	void **ndata;
+
+	assert(top != NULL);
+
+	if (top->data != NULL)
+		ndata = realloc(top->data, (top->size + 1) * sizeof(void*));
+	else
+		ndata = malloc((top->size + 1) * sizeof(void*));
+
+	if (ndata == NULL)
+		return false;
+
+	ndata[top->size++] = ptr;
+
+	top->data = ndata;
+
+	return true;
+}
+
+static CFWClass class = {
+	.name = "CFWRefPool",
+	.size = sizeof(CFWRefPool),
+	.ctor = ctor,
+	.dtor = dtor
+};
+CFWClass *cfw_refpool = &class;

src/refpool.h

@@ -0,0 +1,36 @@
+/*
+ * Copyright (c) 2012, Jonathan Schleifer <js@webkeks.org>
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice,
+ *    this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright notice,
+ *    this list of conditions and the following disclaimer in the documentation
+ *    and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
+ * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
+ * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
+ * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
+ * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
+ * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+ * POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#ifndef __COREFW_REFPOOL_H__
+#define __COREFW_REFPOOL_H__
+
+#include "class.h"
+
+typedef struct CFWRefPool CFWRefPool;
+extern CFWClass *cfw_refpool;
+extern bool cfw_refpool_add(void*);
+
+#endif