Add free functions for predefined llist types.

2 files changed, 22 insertions, 2 deletions

diff --git a/lib/lib.h b/lib/lib.h
index 715e78d9..f8fcd8fe 100644
--- a/lib/lib.h
+++ b/lib/lib.h
@@ -24,7 +24,9 @@ struct double_list {
   char *data;
 };
 
-void llist_traverse(void *list, void (*using)(void *data));
+void llist_free_arg(void *node);
+void llist_free_double(void *node);
+void llist_traverse(void *list, void (*using)(void *node));
 void *llist_pop(void *list);  // actually void **list
 void *dlist_pop(void *list);  // actually struct double_list **list
 void dlist_add_nomalloc(struct double_list **list, struct double_list *new);
diff --git a/lib/llist.c b/lib/llist.c
index 71a187db..2d5bc97a 100644
--- a/lib/llist.c
+++ b/lib/llist.c
@@ -5,8 +5,26 @@
 
 #include "toys.h"
 
+// Callback function to free data pointer of double_list or arg_list
+
+void llist_free_arg(void *node)
+{
+  struct arg_list *d = node;
+
+  free(d->arg);
+  free(d);
+}
+
+void llist_free_double(void *node)
+{
+  struct double_list *d = node;
+
+  free(d->data);
+  free(d);
+}
+
 // Call a function (such as free()) on each element of a linked list.
-void llist_traverse(void *list, void (*using)(void *data))
+void llist_traverse(void *list, void (*using)(void *node))
 {
   void *old = list;