nsz Git - musl/blob - src/dirent/scandir.c

   1 #include <dirent.h>
   2 #include <string.h>
   3 #include <stdlib.h>
   4 #include <inttypes.h>
   5 #include <errno.h>
   6 #include <stddef.h>
   7 #include <libc.h>
   8
   9 int scandir(const char *path, struct dirent ***res,
  10         int (*sel)(const struct dirent *),
  11         int (*cmp)(const struct dirent **, const struct dirent **))
  12 {
  13         DIR *d = opendir(path);
  14         struct dirent *de, **names=0, **tmp;
  15         size_t cnt=0, len=0, size;
  16         int old_errno = errno;
  17
  18         if (!d) return -1;
  19
  20         while ((errno=0), (de = readdir(d))) {
  21                 if (sel && !sel(de)) continue;
  22                 if (cnt >= len) {
  23                         len = 2*len+1;
  24                         if (len > SIZE_MAX/sizeof *names) break;
  25                         tmp = realloc(names, len * sizeof *names);
  26                         if (!tmp) break;
  27                         names = tmp;
  28                 }
  29                 size = offsetof(struct dirent,d_name) + strlen(de->d_name) + 1;
  30                 names[cnt] = malloc(size);
  31                 if (!names[cnt]) break;
  32                 memcpy(names[cnt++], de, size);
  33         }
  34
  35         closedir(d);
  36
  37         if (errno) {
  38                 if (names) while (cnt-->0) free(names[cnt]);
  39                 free(names);
  40                 return -1;
  41         }
  42         errno = old_errno;
  43
  44         if (cmp) qsort(names, cnt, sizeof *names, (int (*)(const void *, const void *))cmp);
  45         *res = names;
  46         return cnt;
  47 }
  48
  49 LFS64(scandir);