summaryrefslogtreecommitdiff
path: root/libc/misc/dirent/scandir.c
diff options
context:
space:
mode:
Diffstat (limited to 'libc/misc/dirent/scandir.c')
-rw-r--r--libc/misc/dirent/scandir.c20
1 files changed, 12 insertions, 8 deletions
diff --git a/libc/misc/dirent/scandir.c b/libc/misc/dirent/scandir.c
index a460e161f..1b4d7ec6d 100644
--- a/libc/misc/dirent/scandir.c
+++ b/libc/misc/dirent/scandir.c
@@ -30,11 +30,6 @@
#include <sys/types.h>
#include "dirstream.h"
-/*
- * FIXME: This is a simple hack version which doesn't sort the data, and
- * just passes all unsorted.
- */
-
int scandir(const char *dir, struct dirent ***namelist,
int (*selector) (const struct dirent *),
int (*compar) (const __ptr_t, const __ptr_t))
@@ -52,7 +47,10 @@ int scandir(const char *dir, struct dirent ***namelist,
while (NULL != readdir(d))
count++;
- names = malloc(sizeof (struct dirent *) * count);
+ if (!(names = malloc(sizeof (struct dirent *) * count))) {
+ closedir(d);
+ return -1;
+ }
rewinddir(d);
@@ -68,8 +66,14 @@ int scandir(const char *dir, struct dirent ***namelist,
}
result = closedir(d);
- if (pos != count)
- names = realloc(names, sizeof (struct dirent *) * pos);
+ if (pos != count) {
+ struct dirent **tmp;
+ if (!(tmp = realloc(names, sizeof (struct dirent *) * pos))) {
+ free(names);
+ return -1;
+ }
+ names = tmp;
+ }
if (compar != NULL) {
qsort(names, pos, sizeof (struct dirent *), compar);