talloc: allow replacement allocator

[ccan] / ccan / list / list.c

diff --git a/ccan/list/list.c b/ccan/list/list.c
index 0a9e4e4d2d5d092ef1449b82c4a72781033fadc9..d876f21e7b06b0bd06a38fdd89b5b35ff6d62d31 100644 (file)
--- a/ccan/list/list.c
+++ b/ccan/list/list.c
@@ -2,32 +2,41 @@
 #include <stdlib.h>
 #include "list.h"
 
-struct list_head *list_check(struct list_head *h, const char *abortstr)
+static void *corrupt(const char *abortstr,
+                    const struct list_node *head,
+                    const struct list_node *node,
+                    unsigned int count)
 {
-       struct list_node *n, *p;
-       int count = 0;
-
-       if (h->n.next == &h->n) {
-               if (h->n.prev != &h->n) {
-                       if (!abortstr)
-                               return NULL;
-                       fprintf(stderr, "%s: prev corrupt in empty %p\n",
-                               abortstr, h);
-                       abort();
-               }
-               return h;
+       if (abortstr) {
+               fprintf(stderr,
+                       "%s: prev corrupt in node %p (%u) of %p\n",
+                       abortstr, node, count, head);
+               abort();
        }
+       return NULL;
+}
+
+struct list_node *list_check_node(const struct list_node *node,
+                                 const char *abortstr)
+{
+       const struct list_node *p, *n;
+       int count = 0;
 
-       for (p = &h->n, n = h->n.next; n != &h->n; p = n, n = n->next) {
+       for (p = node, n = node->next; n != node; p = n, n = n->next) {
                count++;
-               if (n->prev != p) {
-                       if (!abortstr)
-                               return NULL;
-                       fprintf(stderr,
-                               "%s: prev corrupt in node %p (%u) of %p\n",
-                               abortstr, n, count, h);
-                       abort();
-               }
+               if (n->prev != p)
+                       return corrupt(abortstr, node, n, count);
        }
-       return h;
+       /* Check prev on head node. */
+       if (node->prev != p)
+               return corrupt(abortstr, node, node, 0);
+
+       return (struct list_node *)node;
+}
+
+struct list_head *list_check(const struct list_head *h, const char *abortstr)
+{
+       if (!list_check_node(&h->n, abortstr))
+               return NULL;
+       return (struct list_head *)h;
 }