diff options
Diffstat (limited to 'libc/misc/dirent/scandir.c')
| -rw-r--r-- | libc/misc/dirent/scandir.c | 20 |
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); |