X-Git-Url: http://git.droids-corp.org/?a=blobdiff_plain;f=lib%2Fecoli_node_once.c;h=63098783dbaeb03a5eed9f0b93d910bfeb77a6f3;hb=15718f19cbf3868bf6fba79f4e26b5254c57873e;hp=738b43d3f5877a5a4069bdc6f661895224b135b5;hpb=8c1e15f7d7e507c02f41ef9c3dbe310005ce369f;p=protos%2Flibecoli.git

diff --git a/lib/ecoli_node_once.c b/lib/ecoli_node_once.c
index 738b43d..6309878 100644
--- a/lib/ecoli_node_once.c
+++ b/lib/ecoli_node_once.c
@@ -88,6 +88,13 @@ ec_node_once_complete(const struct ec_node *gen_node,
 	return 0;
 }
 
+static void ec_node_once_free_priv(struct ec_node *gen_node)
+{
+	struct ec_node_once *node = (struct ec_node_once *)gen_node;
+
+	ec_node_free(node->child);
+}
+
 static size_t
 ec_node_once_get_children_count(const struct ec_node *gen_node)
 {
@@ -98,15 +105,18 @@ ec_node_once_get_children_count(const struct ec_node *gen_node)
 	return 0;
 }
 
-static struct ec_node *
-ec_node_once_get_child(const struct ec_node *gen_node, size_t i)
+static int
+ec_node_once_get_child(const struct ec_node *gen_node, size_t i,
+		struct ec_node **child, unsigned int *refs)
 {
 	struct ec_node_once *node = (struct ec_node_once *)gen_node;
 
 	if (i >= 1)
-		return NULL;
+		return -1;
 
-	return node->child;
+	*child = node->child;
+	*refs = 1;
+	return 0;
 }
 
 static struct ec_node_type ec_node_once_type = {
@@ -114,6 +124,7 @@ static struct ec_node_type ec_node_once_type = {
 	.parse = ec_node_once_parse,
 	.complete = ec_node_once_complete,
 	.size = sizeof(struct ec_node_once),
+	.free_priv = ec_node_once_free_priv,
 	.get_children_count = ec_node_once_get_children_count,
 	.get_child = ec_node_once_get_child,
 };
@@ -148,7 +159,7 @@ struct ec_node *ec_node_once(const char *id, struct ec_node *child)
 	if (child == NULL)
 		return NULL;
 
-	gen_node = __ec_node(&ec_node_once_type, id);
+	gen_node = ec_node_from_type(&ec_node_once_type, id);
 	if (gen_node == NULL)
 		goto fail;