Slight improvement to scandirlist function to properly free memory on error

2 files changed, 5 insertions, 8 deletions

diff --git a/src/main.c b/src/main.c
index d5e90c9..3779206 100644
--- a/src/main.c
+++ b/src/main.c
@@ -22,13 +22,6 @@ int printnames(void *data) {
     return 0;
 }
 int main() {
-    // error(1, ENOTSUP, "No main file lol");
-
-    // Test code to showcase the scanner function
-
-    dlinkedlist *ll = scandirlist(".", lol, alphasort);
-    dlinkedlist_foreach(ll, printnames);
-    dlinkedlist_free(ll);
-
+    error(1, ENOTSUP, "No main file lol");
     return 0;
 }
\ No newline at end of file
diff --git a/src/scanner.c b/src/scanner.c
index 7c65df4..1c4d2b1 100644
--- a/src/scanner.c
+++ b/src/scanner.c
@@ -22,6 +22,10 @@ dlinkedlist * scandirlist(const char * const dir, int (*selector)(const struct d
     for(int i = 0; i < numentries; i++)
         if(dlinkedlist_append(list, (void *)(namelist[i]), free) < 0) {
             dlinkedlist_free(list);
+            for(int j = i; j < numentries; j++)
+                free(namelist[j]);
+
+            free(namelist);
             RETURNWERR(errno, NULL);
         }
     free(namelist);