af61ba738c
Instead of allocating a buffer on each run, build a buf on the stack.
60 lines
1.5 KiB
C
60 lines
1.5 KiB
C
/* See LICENSE file for copyright and license details. */
|
|
#include <dirent.h>
|
|
#include <errno.h>
|
|
#include <limits.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
|
|
#include "../util.h"
|
|
|
|
int recurse_follow = 'P';
|
|
int recurse_samedev = 0;
|
|
|
|
void
|
|
recurse(const char *path, void (*fn)(const char *, int, void *), int depth, void *data)
|
|
{
|
|
struct dirent *d;
|
|
struct stat lst, st, dst;
|
|
DIR *dp;
|
|
char subpath[PATH_MAX];
|
|
|
|
if (lstat(path, &lst) < 0) {
|
|
if (errno != ENOENT)
|
|
weprintf("lstat %s:", path);
|
|
return;
|
|
}
|
|
if (stat(path, &st) < 0) {
|
|
if (errno != ENOENT)
|
|
weprintf("stat %s:", path);
|
|
return;
|
|
}
|
|
if (!S_ISDIR(lst.st_mode) && !(S_ISLNK(lst.st_mode) && S_ISDIR(st.st_mode) &&
|
|
!(recurse_follow == 'P' || (recurse_follow == 'H' && depth > 0))))
|
|
return;
|
|
|
|
if (!(dp = opendir(path)))
|
|
eprintf("opendir %s:", path);
|
|
|
|
while ((d = readdir(dp))) {
|
|
if (!strcmp(d->d_name, ".") || !strcmp(d->d_name, ".."))
|
|
continue;
|
|
if (strlcpy(subpath, path, PATH_MAX) >= PATH_MAX)
|
|
eprintf("strlcpy: path too long\n");
|
|
if (path[strlen(path) - 1] != '/' && strlcat(subpath, "/", PATH_MAX) >= PATH_MAX)
|
|
eprintf("strlcat: path too long\n");
|
|
if (strlcat(subpath, d->d_name, PATH_MAX) >= PATH_MAX)
|
|
eprintf("strlcat: path too long\n");
|
|
if (recurse_samedev && lstat(subpath, &dst) < 0) {
|
|
if (errno != ENOENT)
|
|
weprintf("stat %s:", subpath);
|
|
} else if (!(recurse_samedev && dst.st_dev != lst.st_dev))
|
|
fn(subpath, depth + 1, data);
|
|
}
|
|
|
|
closedir(dp);
|
|
}
|