get node children and refs in the same function

[protos/libecoli.git] / lib / ecoli_node_once.c

diff --git a/lib/ecoli_node_once.c b/lib/ecoli_node_once.c
index 738b43d..3d4b038 100644 (file)
--- 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,
 };