9fd4a745f8
For loop detection, a history is mandatory. In the process of also adding a flexible struct to recurse, the recurse-definition was moved to fs.h. The motivation behind the struct is to allow easy extensions to the recurse-function without having to change the prototypes of all functions in the process. Adding flags is really simple as well now. Using the recursor-struct, it's also easier to see which defaults apply to a program (for instance, which type of follow, ...). Another change was to add proper stat-lstat-usage in recurse. It was wrong before.
89 lines
1.9 KiB
C
89 lines
1.9 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 "../fs.h"
|
|
#include "../util.h"
|
|
|
|
int recurse_status = 0;
|
|
|
|
void
|
|
recurse(const char *path, void *data, struct recursor *r)
|
|
{
|
|
struct dirent *d;
|
|
struct history *new, *h;
|
|
struct stat st, dst;
|
|
DIR *dp;
|
|
int (*statf)(const char *, struct stat *);
|
|
char subpath[PATH_MAX], *statf_name;
|
|
|
|
if (r->follow == 'P' || (r->follow == 'H' && r->depth)) {
|
|
statf_name = "lstat";
|
|
statf = lstat;
|
|
} else {
|
|
statf_name = "stat";
|
|
statf = stat;
|
|
}
|
|
|
|
if (statf(path, &st) < 0) {
|
|
if (errno != ENOENT) {
|
|
weprintf("%s %s:", statf_name, path);
|
|
recurse_status = 1;
|
|
}
|
|
return;
|
|
}
|
|
if (!S_ISDIR(st.st_mode))
|
|
return;
|
|
|
|
new = emalloc(sizeof(struct history));
|
|
|
|
new->prev = r->hist;
|
|
r->hist = new;
|
|
new->dev = st.st_dev;
|
|
new->ino = st.st_ino;
|
|
|
|
for (h = new->prev; h; h = h->prev)
|
|
if (h->dev == st.st_dev && h->ino == st.st_ino)
|
|
return;
|
|
|
|
if (!(dp = opendir(path))) {
|
|
weprintf("opendir %s:", path);
|
|
recurse_status = 1;
|
|
return;
|
|
}
|
|
|
|
while ((d = readdir(dp))) {
|
|
if (r->follow == 'H') {
|
|
statf_name = "lstat";
|
|
statf = lstat;
|
|
}
|
|
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 ((r->flags & SAMEDEV) && statf(subpath, &dst) < 0) {
|
|
if (errno != ENOENT) {
|
|
weprintf("%s %s:", statf_name, subpath);
|
|
recurse_status = 1;
|
|
}
|
|
} else if (!((r->flags & SAMEDEV) && dst.st_dev != st.st_dev)) {
|
|
r->depth++;
|
|
(r->fn)(subpath, data, r);
|
|
r->depth--;
|
|
}
|
|
}
|
|
|
|
closedir(dp);
|
|
}
|