]> git.ozlabs.org Git - ccan/blobdiff - ccan/tal/tal.c
tal: Make tal_resize() easier to use.
[ccan] / ccan / tal / tal.c
index 9fed477fe9595bd4e85d5243f1315c602915a830..ffe2d71334dce4473d57243bca0827aaa5fee565 100644 (file)
@@ -9,6 +9,7 @@
 #include <stddef.h>
 #include <string.h>
 #include <limits.h>
+#include <errno.h>
 
 //#define TAL_DEBUG 1
 
@@ -455,6 +456,7 @@ static struct tal_hdr *remove_node(struct tal_hdr *t)
 
                /* Are we the only one? */
                if (prev == t) {
+                       struct prop_hdr *next = (*prop)->next;
                        struct children *c = group->parent_child;
                        /* Is this the group embedded in the child property? */
                        if (group == &c->group) {
@@ -462,9 +464,9 @@ static struct tal_hdr *remove_node(struct tal_hdr *t)
                        } else {
                                /* Empty group, so free it. */
                                list_del_from(&c->group.list, &group->list.n);
-                               *prop = group->hdr.next;
                                freefn(group);
                        }
+                       *prop = next;
                        return c->parent;
                } else {
                        /* Move property to next node. */
@@ -481,6 +483,7 @@ static struct tal_hdr *remove_node(struct tal_hdr *t)
 void tal_free(const tal_t *ctx)
 {
         struct tal_hdr *t;
+       int saved_errno = errno;
 
         if (!ctx)
                 return;
@@ -488,6 +491,7 @@ void tal_free(const tal_t *ctx)
         t = debug_tal(to_tal_hdr(ctx));
         remove_node(t);
         del_tree(t);
+       errno = saved_errno;
 }
 
 void *tal_steal_(const tal_t *new_parent, const tal_t *ctx)
@@ -638,7 +642,12 @@ tal_t *tal_next(const tal_t *root, const tal_t *prev)
 tal_t *tal_parent(const tal_t *ctx)
 {
         struct group *group;
-        struct tal_hdr *t = debug_tal(to_tal_hdr(ctx));
+        struct tal_hdr *t;
+
+       if (!ctx)
+               return NULL;
+
+       t = debug_tal(to_tal_hdr(ctx));
 
        while (!(group = find_property(t, GROUP)))
                t = t->next;
@@ -648,22 +657,23 @@ tal_t *tal_parent(const tal_t *ctx)
         return from_tal_hdr(group->parent_child->parent);
 }
 
-void *tal_realloc_(tal_t *ctx, size_t size)
+bool tal_resize_(tal_t **ctxp, size_t size)
 {
         struct tal_hdr *old_t, *t, **prev;
         struct group *group;
         struct children *child;
 
-        old_t = debug_tal(to_tal_hdr(ctx));
+        old_t = debug_tal(to_tal_hdr(*ctxp));
 
         t = resizefn(old_t, size + sizeof(struct tal_hdr));
        if (!t) {
                call_error("Reallocation failure");
-               tal_free(old_t);
-               return NULL;
+               return false;
        }
+
+       /* If it didn't move, we're done! */
         if (t == old_t)
-                return ctx;
+                return true;
        update_bounds(t);
 
        /* Fix up linked list pointer. */
@@ -683,8 +693,8 @@ void *tal_realloc_(tal_t *ctx, size_t size)
                assert(child->parent == old_t);
                child->parent = t;
        }
-
-        return from_tal_hdr(debug_tal(t));
+       *ctxp = from_tal_hdr(debug_tal(t));
+       return true;
 }
 
 char *tal_strdup(const tal_t *ctx, const char *p)
@@ -749,7 +759,10 @@ char *tal_vasprintf(const tal_t *ctx, const char *fmt, va_list ap)
 
                if (ret < max)
                        break;
-               buf = tal_resize(buf, max *= 2);
+               if (!tal_resize(&buf, max *= 2)) {
+                       tal_free(buf);
+                       buf = NULL;
+               }
        }
        if (ctx == TAL_TAKE)
                tal_free(fmt);