1/* See LICENSE file for copyright and license details. */
  2#include <dirent.h>
  3#include <errno.h>
  4#include <fcntl.h>
  5#include <limits.h>
  6#include <stdio.h>
  7#include <stdlib.h>
  8#include <string.h>
  9#include <sys/stat.h>
 10#include <sys/types.h>
 11#include <unistd.h>
 12
 13#include "../fs.h"
 14#include "../util.h"
 15
 16int recurse_status = 0;
 17
 18void
 19recurse(int dirfd, const char *name, void *data, struct recursor *r)
 20{
 21	struct dirent *d;
 22	struct history *new, *h;
 23	struct stat st, dst;
 24	DIR *dp;
 25	int flags = 0, fd;
 26	size_t pathlen = r->pathlen;
 27
 28	if (dirfd == AT_FDCWD)
 29		pathlen = estrlcpy(r->path, name, sizeof(r->path));
 30
 31	if (r->follow == 'P' || (r->follow == 'H' && r->depth))
 32		flags |= AT_SYMLINK_NOFOLLOW;
 33
 34	if (fstatat(dirfd, name, &st, flags) < 0) {
 35		if (!(r->flags & SILENT)) {
 36			weprintf("stat %s:", r->path);
 37			recurse_status = 1;
 38		}
 39		return;
 40	}
 41	if (!S_ISDIR(st.st_mode)) {
 42		r->fn(dirfd, name, &st, data, r);
 43		return;
 44	}
 45
 46	new = emalloc(sizeof(struct history));
 47	new->prev  = r->hist;
 48	r->hist    = new;
 49	new->dev   = st.st_dev;
 50	new->ino   = st.st_ino;
 51
 52	for (h = new->prev; h; h = h->prev)
 53		if (h->ino == st.st_ino && h->dev == st.st_dev)
 54			return;
 55
 56	if (!r->depth && (r->flags & DIRFIRST))
 57		r->fn(dirfd, name, &st, data, r);
 58
 59	if (!r->maxdepth || r->depth + 1 < r->maxdepth) {
 60		fd = openat(dirfd, name, O_RDONLY | O_CLOEXEC | O_DIRECTORY);
 61		if (fd < 0) {
 62			weprintf("open %s:", r->path);
 63			recurse_status = 1;
 64		}
 65		if (!(dp = fdopendir(fd))) {
 66			if (!(r->flags & SILENT)) {
 67				weprintf("fdopendir:");
 68				recurse_status = 1;
 69			}
 70			return;
 71		}
 72		if (r->path[pathlen - 1] != '/')
 73			r->path[pathlen++] = '/';
 74		if (r->follow == 'H')
 75			flags |= AT_SYMLINK_NOFOLLOW;
 76		while ((d = readdir(dp))) {
 77			if (!strcmp(d->d_name, ".") || !strcmp(d->d_name, ".."))
 78				continue;
 79			r->pathlen = pathlen + estrlcpy(r->path + pathlen, d->d_name, sizeof(r->path) - pathlen);
 80			if (fstatat(fd, d->d_name, &dst, flags) < 0) {
 81				if (!(r->flags & SILENT)) {
 82					weprintf("stat %s:", r->path);
 83					recurse_status = 1;
 84				}
 85			} else if ((r->flags & SAMEDEV) && dst.st_dev != st.st_dev) {
 86				continue;
 87			} else {
 88				r->depth++;
 89				r->fn(fd, d->d_name, &dst, data, r);
 90				r->depth--;
 91			}
 92		}
 93		r->path[pathlen - 1] = '\0';
 94		r->pathlen = pathlen - 1;
 95		closedir(dp);
 96	}
 97
 98	if (!r->depth) {
 99		if (!(r->flags & DIRFIRST))
100			r->fn(dirfd, name, &st, data, r);
101
102		while (r->hist) {
103			h = r->hist;
104			r->hist = r->hist->prev;
105			free(h);
106		}
107	}
108}