summaryrefslogtreecommitdiff
path: root/src/dirent/scandir.c
blob: 6a0a99933833fe531db03216d026f40b4ca6f8d2 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#include <dirent.h>
#include <string.h>
#include <stdlib.h>
#include <inttypes.h>
#include <errno.h>
#include <stddef.h>
#include <libc.h>

int scandir(const char *path, struct dirent ***res,
	int (*sel)(const struct dirent *),
	int (*cmp)(const struct dirent **, const struct dirent **))
{
	DIR *d = opendir(path);
	struct dirent *de, **names=0, **tmp;
	size_t cnt=0, len=0, size;
	int old_errno = errno;

	if (!d) return -1;

	while ((errno=0), (de = readdir(d))) {
		if (sel && !sel(de)) continue;
		if (cnt >= len) {
			len = 2*len+1;
			if (len > SIZE_MAX/sizeof *names) break;
			tmp = realloc(names, len * sizeof *names);
			if (!tmp) break;
			names = tmp;
		}
		size = offsetof(struct dirent,d_name) + strlen(de->d_name) + 1;
		names[cnt] = malloc(size);
		if (!names[cnt]) break;
		memcpy(names[cnt++], de, size);
	}

	closedir(d);

	if (errno) {
		old_errno = errno;
		if (names) while (cnt-->0) free(names[cnt]);
		free(names);
		errno = old_errno;
		return -1;
	}

	if (cmp) qsort(names, cnt, sizeof *names, (int (*)(const void *, const void *))cmp);
	*res = names;
	return cnt;
}

LFS64(scandir);