summaryrefslogtreecommitdiffstats
path: root/eval.c
diff options
context:
space:
mode:
Diffstat (limited to 'eval.c')
-rw-r--r--eval.c1391
1 files changed, 0 insertions, 1391 deletions
diff --git a/eval.c b/eval.c
deleted file mode 100644
index d89e62a..0000000
--- a/eval.c
+++ /dev/null
@@ -1,1391 +0,0 @@
-/****************************************************************************
- * bfs *
- * Copyright (C) 2015-2019 Tavian Barnes <tavianator@tavianator.com> *
- * *
- * Permission to use, copy, modify, and/or distribute this software for any *
- * purpose with or without fee is hereby granted. *
- * *
- * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES *
- * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF *
- * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR *
- * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES *
- * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN *
- * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF *
- * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. *
- ****************************************************************************/
-
-/**
- * Implementation of all the literal expressions.
- */
-
-#include "eval.h"
-#include "bftw.h"
-#include "cmdline.h"
-#include "color.h"
-#include "diag.h"
-#include "dstring.h"
-#include "exec.h"
-#include "fsade.h"
-#include "mtab.h"
-#include "printf.h"
-#include "stat.h"
-#include "trie.h"
-#include "util.h"
-#include <assert.h>
-#include <dirent.h>
-#include <errno.h>
-#include <fcntl.h>
-#include <fnmatch.h>
-#include <grp.h>
-#include <pwd.h>
-#include <stdarg.h>
-#include <stdint.h>
-#include <stdio.h>
-#include <stdlib.h>
-#include <string.h>
-#include <sys/resource.h>
-#include <sys/stat.h>
-#include <time.h>
-#include <unistd.h>
-
-struct eval_state {
- /** Data about the current file. */
- const struct BFTW *ftwbuf;
- /** The parsed command line. */
- const struct cmdline *cmdline;
- /** The bftw() callback return value. */
- enum bftw_action action;
- /** The eval_cmdline() return value. */
- int *ret;
- /** Whether to quit immediately. */
- bool quit;
-};
-
-/**
- * Print an error message.
- */
-BFS_FORMATTER(2, 3)
-static void eval_error(struct eval_state *state, const char *format, ...) {
- int error = errno;
- const struct cmdline *cmdline = state->cmdline;
- CFILE *cerr = cmdline->cerr;
-
- bfs_error(cmdline, "%pP: ", state->ftwbuf);
-
- va_list args;
- va_start(args, format);
- errno = error;
- cvfprintf(cerr, format, args);
- va_end(args);
-}
-
-/**
- * Check if an error should be ignored.
- */
-static bool eval_should_ignore(const struct eval_state *state, int error) {
- return state->cmdline->ignore_races
- && is_nonexistence_error(error)
- && state->ftwbuf->depth > 0;
-}
-
-/**
- * Report an error that occurs during evaluation.
- */
-static void eval_report_error(struct eval_state *state) {
- if (!eval_should_ignore(state, errno)) {
- eval_error(state, "%m.\n");
- *state->ret = EXIT_FAILURE;
- }
-}
-
-/**
- * Perform a bfs_stat() call if necessary.
- */
-static const struct bfs_stat *eval_stat(struct eval_state *state) {
- const struct BFTW *ftwbuf = state->ftwbuf;
- const struct bfs_stat *ret = bftw_stat(ftwbuf, ftwbuf->stat_flags);
- if (!ret) {
- eval_report_error(state);
- }
- return ret;
-}
-
-/**
- * Get the difference (in seconds) between two struct timespecs.
- */
-static time_t timespec_diff(const struct timespec *lhs, const struct timespec *rhs) {
- time_t ret = lhs->tv_sec - rhs->tv_sec;
- if (lhs->tv_nsec < rhs->tv_nsec) {
- --ret;
- }
- return ret;
-}
-
-bool expr_cmp(const struct expr *expr, long long n) {
- switch (expr->cmp_flag) {
- case CMP_EXACT:
- return n == expr->idata;
- case CMP_LESS:
- return n < expr->idata;
- case CMP_GREATER:
- return n > expr->idata;
- }
-
- return false;
-}
-
-/**
- * -true test.
- */
-bool eval_true(const struct expr *expr, struct eval_state *state) {
- return true;
-}
-
-/**
- * -false test.
- */
-bool eval_false(const struct expr *expr, struct eval_state *state) {
- return false;
-}
-
-/**
- * -executable, -readable, -writable tests.
- */
-bool eval_access(const struct expr *expr, struct eval_state *state) {
- const struct BFTW *ftwbuf = state->ftwbuf;
- return xfaccessat(ftwbuf->at_fd, ftwbuf->at_path, expr->idata) == 0;
-}
-
-/**
- * -acl test.
- */
-bool eval_acl(const struct expr *expr, struct eval_state *state) {
- int ret = bfs_check_acl(state->ftwbuf);
- if (ret >= 0) {
- return ret;
- } else {
- eval_report_error(state);
- return false;
- }
-}
-
-/**
- * -capable test.
- */
-bool eval_capable(const struct expr *expr, struct eval_state *state) {
- int ret = bfs_check_capabilities(state->ftwbuf);
- if (ret >= 0) {
- return ret;
- } else {
- eval_report_error(state);
- return false;
- }
-}
-
-/**
- * Get the given timespec field out of a stat buffer.
- */
-static const struct timespec *eval_stat_time(const struct bfs_stat *statbuf, enum bfs_stat_field field, struct eval_state *state) {
- const struct timespec *ret = bfs_stat_time(statbuf, field);
- if (!ret) {
- eval_error(state, "Couldn't get file %s: %m.\n", bfs_stat_field_name(field));
- *state->ret = EXIT_FAILURE;
- }
- return ret;
-}
-
-/**
- * -[aBcm]?newer tests.
- */
-bool eval_newer(const struct expr *expr, struct eval_state *state) {
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- return false;
- }
-
- const struct timespec *time = eval_stat_time(statbuf, expr->stat_field, state);
- if (!time) {
- return false;
- }
-
- return time->tv_sec > expr->reftime.tv_sec
- || (time->tv_sec == expr->reftime.tv_sec && time->tv_nsec > expr->reftime.tv_nsec);
-}
-
-/**
- * -[aBcm]{min,time} tests.
- */
-bool eval_time(const struct expr *expr, struct eval_state *state) {
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- return false;
- }
-
- const struct timespec *time = eval_stat_time(statbuf, expr->stat_field, state);
- if (!time) {
- return false;
- }
-
- time_t diff = timespec_diff(&expr->reftime, time);
- switch (expr->time_unit) {
- case MINUTES:
- diff /= 60;
- break;
- case DAYS:
- diff /= 60*60*24;
- break;
- }
-
- return expr_cmp(expr, diff);
-}
-
-/**
- * -used test.
- */
-bool eval_used(const struct expr *expr, struct eval_state *state) {
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- return false;
- }
-
- const struct timespec *atime = eval_stat_time(statbuf, BFS_STAT_ATIME, state);
- const struct timespec *ctime = eval_stat_time(statbuf, BFS_STAT_CTIME, state);
- if (!atime || !ctime) {
- return false;
- }
-
- time_t diff = timespec_diff(atime, ctime);
- diff /= 60*60*24;
- return expr_cmp(expr, diff);
-}
-
-/**
- * -gid test.
- */
-bool eval_gid(const struct expr *expr, struct eval_state *state) {
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- return false;
- }
-
- return expr_cmp(expr, statbuf->gid);
-}
-
-/**
- * -uid test.
- */
-bool eval_uid(const struct expr *expr, struct eval_state *state) {
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- return false;
- }
-
- return expr_cmp(expr, statbuf->uid);
-}
-
-/**
- * -nogroup test.
- */
-bool eval_nogroup(const struct expr *expr, struct eval_state *state) {
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- return false;
- }
-
- errno = 0;
- if (getgrgid(statbuf->gid) == NULL) {
- if (errno == 0) {
- return true;
- } else {
- eval_report_error(state);
- }
- }
-
- return false;
-}
-
-/**
- * -nouser test.
- */
-bool eval_nouser(const struct expr *expr, struct eval_state *state) {
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- return false;
- }
-
- errno = 0;
- if (getpwuid(statbuf->uid) == NULL) {
- if (errno == 0) {
- return true;
- } else {
- eval_report_error(state);
- }
- }
-
- return false;
-}
-
-/**
- * -delete action.
- */
-bool eval_delete(const struct expr *expr, struct eval_state *state) {
- const struct BFTW *ftwbuf = state->ftwbuf;
-
- // Don't try to delete the current directory
- if (strcmp(ftwbuf->path, ".") == 0) {
- return true;
- }
-
- int flag = 0;
-
- // We need to know the actual type of the path, not what it points to
- enum bftw_typeflag type = bftw_typeflag(ftwbuf, BFS_STAT_NOFOLLOW);
- if (type == BFTW_DIR) {
- flag |= AT_REMOVEDIR;
- } else if (type == BFTW_ERROR) {
- eval_report_error(state);
- return false;
- }
-
- if (unlinkat(ftwbuf->at_fd, ftwbuf->at_path, flag) != 0) {
- eval_report_error(state);
- return false;
- }
-
- return true;
-}
-
-/** Finish any pending -exec ... + operations. */
-static int eval_exec_finish(const struct expr *expr, const struct cmdline *cmdline) {
- int ret = 0;
- if (expr->execbuf && bfs_exec_finish(expr->execbuf) != 0) {
- if (errno != 0) {
- bfs_error(cmdline, "%s %s: %m.\n", expr->argv[0], expr->argv[1]);
- }
- ret = -1;
- }
- if (expr->lhs && eval_exec_finish(expr->lhs, cmdline) != 0) {
- ret = -1;
- }
- if (expr->rhs && eval_exec_finish(expr->rhs, cmdline) != 0) {
- ret = -1;
- }
- return ret;
-}
-
-/**
- * -exec[dir]/-ok[dir] actions.
- */
-bool eval_exec(const struct expr *expr, struct eval_state *state) {
- bool ret = bfs_exec(expr->execbuf, state->ftwbuf) == 0;
- if (errno != 0) {
- eval_error(state, "%s %s: %m.\n", expr->argv[0], expr->argv[1]);
- *state->ret = EXIT_FAILURE;
- }
- return ret;
-}
-
-/**
- * -exit action.
- */
-bool eval_exit(const struct expr *expr, struct eval_state *state) {
- state->action = BFTW_STOP;
- *state->ret = expr->idata;
- state->quit = true;
- return true;
-}
-
-/**
- * -depth N test.
- */
-bool eval_depth(const struct expr *expr, struct eval_state *state) {
- return expr_cmp(expr, state->ftwbuf->depth);
-}
-
-/**
- * -empty test.
- */
-bool eval_empty(const struct expr *expr, struct eval_state *state) {
- bool ret = false;
- const struct BFTW *ftwbuf = state->ftwbuf;
-
- if (ftwbuf->typeflag == BFTW_DIR) {
- int dfd = openat(ftwbuf->at_fd, ftwbuf->at_path, O_RDONLY | O_CLOEXEC | O_DIRECTORY);
- if (dfd < 0) {
- eval_report_error(state);
- goto done;
- }
-
- DIR *dir = fdopendir(dfd);
- if (!dir) {
- eval_report_error(state);
- close(dfd);
- goto done;
- }
-
- struct dirent *de;
- if (xreaddir(dir, &de) == 0) {
- ret = !de;
- } else {
- eval_report_error(state);
- }
-
- closedir(dir);
- } else {
- const struct bfs_stat *statbuf = eval_stat(state);
- if (statbuf) {
- ret = statbuf->size == 0;
- }
- }
-
-done:
- return ret;
-}
-
-/**
- * -fstype test.
- */
-bool eval_fstype(const struct expr *expr, struct eval_state *state) {
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- return false;
- }
-
- const char *type = bfs_fstype(state->cmdline->mtab, statbuf);
- return strcmp(type, expr->sdata) == 0;
-}
-
-/**
- * -hidden test.
- */
-bool eval_hidden(const struct expr *expr, struct eval_state *state) {
- const struct BFTW *ftwbuf = state->ftwbuf;
- return ftwbuf->nameoff > 0 && ftwbuf->path[ftwbuf->nameoff] == '.';
-}
-
-/**
- * -nohidden action.
- */
-bool eval_nohidden(const struct expr *expr, struct eval_state *state) {
- if (eval_hidden(expr, state)) {
- eval_prune(expr, state);
- return false;
- } else {
- return true;
- }
-}
-
-/**
- * -inum test.
- */
-bool eval_inum(const struct expr *expr, struct eval_state *state) {
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- return false;
- }
-
- return expr_cmp(expr, statbuf->ino);
-}
-
-/**
- * -links test.
- */
-bool eval_links(const struct expr *expr, struct eval_state *state) {
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- return false;
- }
-
- return expr_cmp(expr, statbuf->nlink);
-}
-
-/**
- * -i?lname test.
- */
-bool eval_lname(const struct expr *expr, struct eval_state *state) {
- bool ret = false;
- char *name = NULL;
-
- const struct BFTW *ftwbuf = state->ftwbuf;
- if (ftwbuf->typeflag != BFTW_LNK) {
- goto done;
- }
-
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- goto done;
- }
-
- name = xreadlinkat(ftwbuf->at_fd, ftwbuf->at_path, statbuf->size);
- if (!name) {
- eval_report_error(state);
- goto done;
- }
-
- ret = fnmatch(expr->sdata, name, expr->idata) == 0;
-
-done:
- free(name);
- return ret;
-}
-
-/**
- * -i?name test.
- */
-bool eval_name(const struct expr *expr, struct eval_state *state) {
- const struct BFTW *ftwbuf = state->ftwbuf;
-
- const char *name = ftwbuf->path + ftwbuf->nameoff;
- char *copy = NULL;
- if (ftwbuf->depth == 0) {
- // Any trailing slashes are not part of the name. This can only
- // happen for the root path.
- const char *slash = strchr(name, '/');
- if (slash && slash > name) {
- copy = strndup(name, slash - name);
- if (!copy) {
- eval_report_error(state);
- return false;
- }
- name = copy;
- }
- }
-
- bool ret = fnmatch(expr->sdata, name, expr->idata) == 0;
- free(copy);
- return ret;
-}
-
-/**
- * -i?path test.
- */
-bool eval_path(const struct expr *expr, struct eval_state *state) {
- const struct BFTW *ftwbuf = state->ftwbuf;
- return fnmatch(expr->sdata, ftwbuf->path, expr->idata) == 0;
-}
-
-/**
- * -perm test.
- */
-bool eval_perm(const struct expr *expr, struct eval_state *state) {
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- return false;
- }
-
- mode_t mode = statbuf->mode;
- mode_t target;
- if (state->ftwbuf->typeflag == BFTW_DIR) {
- target = expr->dir_mode;
- } else {
- target = expr->file_mode;
- }
-
- switch (expr->mode_cmp) {
- case MODE_EXACT:
- return (mode & 07777) == target;
-
- case MODE_ALL:
- return (mode & target) == target;
-
- case MODE_ANY:
- return !(mode & target) == !target;
- }
-
- return false;
-}
-
-/**
- * -f?ls action.
- */
-bool eval_fls(const struct expr *expr, struct eval_state *state) {
- CFILE *cfile = expr->cfile;
- FILE *file = cfile->file;
- const struct BFTW *ftwbuf = state->ftwbuf;
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- goto done;
- }
-
- uintmax_t ino = statbuf->ino;
- uintmax_t blocks = ((uintmax_t)statbuf->blocks*BFS_STAT_BLKSIZE + 1023)/1024;
- char mode[11];
- format_mode(statbuf->mode, mode);
- uintmax_t nlink = statbuf->nlink;
- if (fprintf(file, "%9ju %6ju %s %3ju ", ino, blocks, mode, nlink) < 0) {
- goto error;
- }
-
- uintmax_t uid = statbuf->uid;
- struct passwd *pwd = getpwuid(uid);
- if (pwd) {
- if (fprintf(file, " %-8s", pwd->pw_name) < 0) {
- goto error;
- }
- } else {
- if (fprintf(file, " %-8ju", uid) < 0) {
- goto error;
- }
- }
-
- uintmax_t gid = statbuf->gid;
- struct group *grp = getgrgid(gid);
- if (grp) {
- if (fprintf(file, " %-8s", grp->gr_name) < 0) {
- goto error;
- }
- } else {
- if (fprintf(file, " %-8ju", gid) < 0) {
- goto error;
- }
- }
-
- if (ftwbuf->typeflag & (BFTW_BLK | BFTW_CHR)) {
- int ma = bfs_major(statbuf->rdev);
- int mi = bfs_minor(statbuf->rdev);
- if (fprintf(file, " %3d, %3d", ma, mi) < 0) {
- goto error;
- }
- } else {
- uintmax_t size = statbuf->size;
- if (fprintf(file, " %8ju", size) < 0) {
- goto error;
- }
- }
-
- time_t time = statbuf->mtime.tv_sec;
- time_t now = expr->reftime.tv_sec;
- time_t six_months_ago = now - 6*30*24*60*60;
- time_t tomorrow = now + 24*60*60;
- struct tm tm;
- if (xlocaltime(&time, &tm) != 0) {
- goto error;
- }
- char time_str[256];
- const char *time_format = "%b %e %H:%M";
- if (time <= six_months_ago || time >= tomorrow) {
- time_format = "%b %e %Y";
- }
- if (!strftime(time_str, sizeof(time_str), time_format, &tm)) {
- errno = EOVERFLOW;
- goto error;
- }
- if (fprintf(file, " %s", time_str) < 0) {
- goto error;
- }
-
- if (cfprintf(cfile, " %pP", ftwbuf) < 0) {
- goto error;
- }
-
- if (ftwbuf->typeflag == BFTW_LNK) {
- if (cfprintf(cfile, " -> %pL", ftwbuf) < 0) {
- goto error;
- }
- }
-
- if (fputc('\n', file) == EOF) {
- goto error;
- }
-
-done:
- return true;
-
-error:
- eval_report_error(state);
- return true;
-}
-
-/**
- * -f?print action.
- */
-bool eval_fprint(const struct expr *expr, struct eval_state *state) {
- if (cfprintf(expr->cfile, "%pP\n", state->ftwbuf) < 0) {
- eval_report_error(state);
- }
- return true;
-}
-
-/**
- * -f?print0 action.
- */
-bool eval_fprint0(const struct expr *expr, struct eval_state *state) {
- const char *path = state->ftwbuf->path;
- size_t length = strlen(path) + 1;
- if (fwrite(path, 1, length, expr->cfile->file) != length) {
- eval_report_error(state);
- }
- return true;
-}
-
-/**
- * -f?printf action.
- */
-bool eval_fprintf(const struct expr *expr, struct eval_state *state) {
- if (bfs_printf(expr->cfile->file, expr->printf, state->ftwbuf) != 0) {
- eval_report_error(state);
- }
-
- return true;
-}
-
-/**
- * -printx action.
- */
-bool eval_fprintx(const struct expr *expr, struct eval_state *state) {
- FILE *file = expr->cfile->file;
- const char *path = state->ftwbuf->path;
-
- while (true) {
- size_t span = strcspn(path, " \t\n\\$'\"`");
- if (fwrite(path, 1, span, file) != span) {
- goto error;
- }
- path += span;
-
- char c = path[0];
- if (!c) {
- break;
- }
-
- char escaped[] = {'\\', c};
- if (fwrite(escaped, 1, sizeof(escaped), file) != sizeof(escaped)) {
- goto error;
- }
- ++path;
- }
-
-
- if (fputc('\n', file) == EOF) {
- goto error;
- }
-
- return true;
-
-error:
- eval_report_error(state);
- return true;
-}
-
-/**
- * -prune action.
- */
-bool eval_prune(const struct expr *expr, struct eval_state *state) {
- state->action = BFTW_PRUNE;
- return true;
-}
-
-/**
- * -quit action.
- */
-bool eval_quit(const struct expr *expr, struct eval_state *state) {
- state->action = BFTW_STOP;
- state->quit = true;
- return true;
-}
-
-/**
- * -i?regex test.
- */
-bool eval_regex(const struct expr *expr, struct eval_state *state) {
- const char *path = state->ftwbuf->path;
- size_t len = strlen(path);
- regmatch_t match = {
- .rm_so = 0,
- .rm_eo = len,
- };
-
- int flags = 0;
-#ifdef REG_STARTEND
- flags |= REG_STARTEND;
-#endif
- int err = regexec(expr->regex, path, 1, &match, flags);
- if (err == 0) {
- return match.rm_so == 0 && match.rm_eo == len;
- } else if (err != REG_NOMATCH) {
- char *str = xregerror(err, expr->regex);
- if (str) {
- eval_error(state, "%s.\n", str);
- free(str);
- } else {
- perror("xregerror()");
- }
-
- *state->ret = EXIT_FAILURE;
- }
-
- return false;
-}
-
-/**
- * -samefile test.
- */
-bool eval_samefile(const struct expr *expr, struct eval_state *state) {
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- return false;
- }
-
- return statbuf->dev == expr->dev && statbuf->ino == expr->ino;
-}
-
-/**
- * -size test.
- */
-bool eval_size(const struct expr *expr, struct eval_state *state) {
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- return false;
- }
-
- static const off_t scales[] = {
- [SIZE_BLOCKS] = 512,
- [SIZE_BYTES] = 1,
- [SIZE_WORDS] = 2,
- [SIZE_KB] = 1024,
- [SIZE_MB] = 1024LL*1024,
- [SIZE_GB] = 1024LL*1024*1024,
- [SIZE_TB] = 1024LL*1024*1024*1024,
- [SIZE_PB] = 1024LL*1024*1024*1024*1024,
- };
-
- off_t scale = scales[expr->size_unit];
- off_t size = (statbuf->size + scale - 1)/scale; // Round up
- return expr_cmp(expr, size);
-}
-
-/**
- * -sparse test.
- */
-bool eval_sparse(const struct expr *expr, struct eval_state *state) {
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- return false;
- }
-
- blkcnt_t expected = (statbuf->size + BFS_STAT_BLKSIZE - 1)/BFS_STAT_BLKSIZE;
- return statbuf->blocks < expected;
-}
-
-/**
- * -type test.
- */
-bool eval_type(const struct expr *expr, struct eval_state *state) {
- return state->ftwbuf->typeflag & expr->idata;
-}
-
-/**
- * -xattr test.
- */
-bool eval_xattr(const struct expr *expr, struct eval_state *state) {
- int ret = bfs_check_xattrs(state->ftwbuf);
- if (ret >= 0) {
- return ret;
- } else {
- eval_report_error(state);
- return false;
- }
-}
-
-/**
- * -xtype test.
- */
-bool eval_xtype(const struct expr *expr, struct eval_state *state) {
- const struct BFTW *ftwbuf = state->ftwbuf;
- enum bfs_stat_flag flags = ftwbuf->stat_flags ^ (BFS_STAT_NOFOLLOW | BFS_STAT_TRYFOLLOW);
- enum bftw_typeflag type = bftw_typeflag(ftwbuf, flags);
- if (type == BFTW_ERROR) {
- eval_report_error(state);
- return false;
- } else {
- return type & expr->idata;
- }
-}
-
-#if _POSIX_MONOTONIC_CLOCK > 0
-# define BFS_CLOCK CLOCK_MONOTONIC
-#elif _POSIX_TIMERS > 0
-# define BFS_CLOCK CLOCK_REALTIME
-#endif
-
-/**
- * Call clock_gettime(), if available.
- */
-static int eval_gettime(struct timespec *ts) {
-#ifdef BFS_CLOCK
- int ret = clock_gettime(BFS_CLOCK, ts);
- if (ret != 0) {
- perror("clock_gettime()");
- }
- return ret;
-#else
- return -1;
-#endif
-}
-
-/**
- * Record the time that elapsed evaluating an expression.
- */
-static void add_elapsed(struct expr *expr, const struct timespec *start, const struct timespec *end) {
- expr->elapsed.tv_sec += end->tv_sec - start->tv_sec;
- expr->elapsed.tv_nsec += end->tv_nsec - start->tv_nsec;
- if (expr->elapsed.tv_nsec < 0) {
- expr->elapsed.tv_nsec += 1000000000L;
- --expr->elapsed.tv_sec;
- } else if (expr->elapsed.tv_nsec >= 1000000000L) {
- expr->elapsed.tv_nsec -= 1000000000L;
- ++expr->elapsed.tv_sec;
- }
-}
-
-/**
- * Evaluate an expression.
- */
-static bool eval_expr(struct expr *expr, struct eval_state *state) {
- struct timespec start, end;
- bool time = state->cmdline->debug & DEBUG_RATES;
- if (time) {
- if (eval_gettime(&start) != 0) {
- time = false;
- }
- }
-
- assert(!state->quit);
-
- bool ret = expr->eval(expr, state);
-
- if (time) {
- if (eval_gettime(&end) == 0) {
- add_elapsed(expr, &start, &end);
- }
- }
-
- ++expr->evaluations;
- if (ret) {
- ++expr->successes;
- }
-
- if (expr_never_returns(expr)) {
- assert(state->quit);
- } else if (!state->quit) {
- assert(!expr->always_true || ret);
- assert(!expr->always_false || !ret);
- }
-
- return ret;
-}
-
-/**
- * Evaluate a negation.
- */
-bool eval_not(const struct expr *expr, struct eval_state *state) {
- return !eval_expr(expr->rhs, state);
-}
-
-/**
- * Evaluate a conjunction.
- */
-bool eval_and(const struct expr *expr, struct eval_state *state) {
- if (!eval_expr(expr->lhs, state)) {
- return false;
- }
-
- if (state->quit) {
- return false;
- }
-
- return eval_expr(expr->rhs, state);
-}
-
-/**
- * Evaluate a disjunction.
- */
-bool eval_or(const struct expr *expr, struct eval_state *state) {
- if (eval_expr(expr->lhs, state)) {
- return true;
- }
-
- if (state->quit) {
- return false;
- }
-
- return eval_expr(expr->rhs, state);
-}
-
-/**
- * Evaluate the comma operator.
- */
-bool eval_comma(const struct expr *expr, struct eval_state *state) {
- eval_expr(expr->lhs, state);
-
- if (state->quit) {
- return false;
- }
-
- return eval_expr(expr->rhs, state);
-}
-
-/** Check if we've seen a file before. */
-static bool eval_file_unique(struct eval_state *state, struct trie *seen) {
- const struct bfs_stat *statbuf = eval_stat(state);
- if (!statbuf) {
- return false;
- }
-
- bfs_file_id id;
- bfs_stat_id(statbuf, &id);
-
- struct trie_leaf *leaf = trie_insert_mem(seen, id, sizeof(id));
- if (!leaf) {
- eval_report_error(state);
- return false;
- }
-
- if (leaf->value) {
- state->action = BFTW_PRUNE;
- return false;
- } else {
- leaf->value = leaf;
- return true;
- }
-}
-
-#define DEBUG_FLAG(flags, flag) \
- do { \
- if ((flags & flag) || flags == flag) { \
- fputs(#flag, stderr); \
- flags ^= flag; \
- if (flags) { \
- fputs(" | ", stderr); \
- } \
- } \
- } while (0)
-
-/**
- * Log a stat() call.
- */
-static void debug_stat(const struct BFTW *ftwbuf, const struct bftw_stat *cache, enum bfs_stat_flag flags) {
- fprintf(stderr, "bfs_stat(");
- if (ftwbuf->at_fd == AT_FDCWD) {
- fprintf(stderr, "AT_FDCWD");
- } else {
- size_t baselen = strlen(ftwbuf->path) - strlen(ftwbuf->at_path);
- fprintf(stderr, "\"");
- fwrite(ftwbuf->path, 1, baselen, stderr);
- fprintf(stderr, "\"");
- }
-
- fprintf(stderr, ", \"%s\", ", ftwbuf->at_path);
-
- DEBUG_FLAG(flags, BFS_STAT_FOLLOW);
- DEBUG_FLAG(flags, BFS_STAT_NOFOLLOW);
- DEBUG_FLAG(flags, BFS_STAT_TRYFOLLOW);
-
- fprintf(stderr, ") == %d", cache->buf ? 0 : -1);
-
- if (cache->error) {
- fprintf(stderr, " [%d]", cache->error);
- }
-
- fprintf(stderr, "\n");
-}
-
-/**
- * Log any stat() calls that happened.
- */
-static void debug_stats(const struct BFTW *ftwbuf) {
- const struct bfs_stat *statbuf = ftwbuf->stat_cache.buf;
- if (statbuf || ftwbuf->stat_cache.error) {
- debug_stat(ftwbuf, &ftwbuf->stat_cache, BFS_STAT_FOLLOW);
- }
-
- const struct bfs_stat *lstatbuf = ftwbuf->lstat_cache.buf;
- if ((lstatbuf && lstatbuf != statbuf) || ftwbuf->lstat_cache.error) {
- debug_stat(ftwbuf, &ftwbuf->lstat_cache, BFS_STAT_NOFOLLOW);
- }
-}
-
-/**
- * Dump the bftw_typeflag for -D search.
- */
-static const char *dump_bftw_typeflag(enum bftw_typeflag type) {
-#define DUMP_BFTW_TYPEFLAG_CASE(flag) \
- case flag: \
- return #flag
-
- switch (type) {
- DUMP_BFTW_TYPEFLAG_CASE(BFTW_BLK);
- DUMP_BFTW_TYPEFLAG_CASE(BFTW_CHR);
- DUMP_BFTW_TYPEFLAG_CASE(BFTW_DIR);
- DUMP_BFTW_TYPEFLAG_CASE(BFTW_DOOR);
- DUMP_BFTW_TYPEFLAG_CASE(BFTW_FIFO);
- DUMP_BFTW_TYPEFLAG_CASE(BFTW_LNK);
- DUMP_BFTW_TYPEFLAG_CASE(BFTW_PORT);
- DUMP_BFTW_TYPEFLAG_CASE(BFTW_REG);
- DUMP_BFTW_TYPEFLAG_CASE(BFTW_SOCK);
- DUMP_BFTW_TYPEFLAG_CASE(BFTW_WHT);
-
- DUMP_BFTW_TYPEFLAG_CASE(BFTW_ERROR);
-
- default:
- DUMP_BFTW_TYPEFLAG_CASE(BFTW_UNKNOWN);
- }
-}
-
-#define DUMP_BFTW_MAP(value) [value] = #value
-
-/**
- * Dump the bftw_visit for -D search.
- */
-static const char *dump_bftw_visit(enum bftw_visit visit) {
- static const char *visits[] = {
- DUMP_BFTW_MAP(BFTW_PRE),
- DUMP_BFTW_MAP(BFTW_POST),
- };
- return visits[visit];
-}
-
-/**
- * Dump the bftw_action for -D search.
- */
-static const char *dump_bftw_action(enum bftw_action action) {
- static const char *actions[] = {
- DUMP_BFTW_MAP(BFTW_CONTINUE),
- DUMP_BFTW_MAP(BFTW_PRUNE),
- DUMP_BFTW_MAP(BFTW_STOP),
- };
- return actions[action];
-}
-
-/**
- * Type passed as the argument to the bftw() callback.
- */
-struct callback_args {
- /** The parsed command line. */
- const struct cmdline *cmdline;
- /** The set of seen files. */
- struct trie *seen;
- /** Eventual return value from eval_cmdline(). */
- int ret;
-};
-
-/**
- * bftw() callback.
- */
-static enum bftw_action cmdline_callback(const struct BFTW *ftwbuf, void *ptr) {
- struct callback_args *args = ptr;
-
- const struct cmdline *cmdline = args->cmdline;
-
- struct eval_state state;
- state.ftwbuf = ftwbuf;
- state.cmdline = cmdline;
- state.action = BFTW_CONTINUE;
- state.ret = &args->ret;
- state.quit = false;
-
- if (ftwbuf->typeflag == BFTW_ERROR) {
- if (!eval_should_ignore(&state, ftwbuf->error)) {
- args->ret = EXIT_FAILURE;
- eval_error(&state, "%s.\n", strerror(ftwbuf->error));
- }
- state.action = BFTW_PRUNE;
- goto done;
- }
-
- if (cmdline->unique && ftwbuf->visit == BFTW_PRE) {
- if (!eval_file_unique(&state, args->seen)) {
- goto done;
- }
- }
-
- if (cmdline->xargs_safe && strpbrk(ftwbuf->path, " \t\n\'\"\\")) {
- args->ret = EXIT_FAILURE;
- eval_error(&state, "Path is not safe for xargs.\n");
- state.action = BFTW_PRUNE;
- goto done;
- }
-
- if (cmdline->maxdepth < 0 || ftwbuf->depth >= cmdline->maxdepth) {
- state.action = BFTW_PRUNE;
- }
-
- // In -depth mode, only handle directories on the BFTW_POST visit
- enum bftw_visit expected_visit = BFTW_PRE;
- if ((cmdline->flags & BFTW_DEPTH)
- && (cmdline->strategy == BFTW_IDS || ftwbuf->typeflag == BFTW_DIR)
- && ftwbuf->depth < cmdline->maxdepth) {
- expected_visit = BFTW_POST;
- }
-
- if (ftwbuf->visit == expected_visit
- && ftwbuf->depth >= cmdline->mindepth
- && ftwbuf->depth <= cmdline->maxdepth) {
- eval_expr(cmdline->expr, &state);
- }
-
-done:
- if (cmdline->debug & DEBUG_STAT) {
- debug_stats(ftwbuf);
- }
-
- if (cmdline->debug & DEBUG_SEARCH) {
- fprintf(stderr, "cmdline_callback({\n");
- fprintf(stderr, "\t.path = \"%s\",\n", ftwbuf->path);
- fprintf(stderr, "\t.root = \"%s\",\n", ftwbuf->root);
- fprintf(stderr, "\t.depth = %zu,\n", ftwbuf->depth);
- fprintf(stderr, "\t.visit = %s,\n", dump_bftw_visit(ftwbuf->visit));
- fprintf(stderr, "\t.typeflag = %s,\n", dump_bftw_typeflag(ftwbuf->typeflag));
- fprintf(stderr, "\t.error = %d,\n", ftwbuf->error);
- fprintf(stderr, "}) == %s\n", dump_bftw_action(state.action));
- }
-
- return state.action;
-}
-
-/**
- * Infer the number of open file descriptors we're allowed to have.
- */
-static int infer_fdlimit(const struct cmdline *cmdline) {
- int ret = 4096;
-
- struct rlimit rl;
- if (getrlimit(RLIMIT_NOFILE, &rl) == 0) {
- if (rl.rlim_cur != RLIM_INFINITY) {
- ret = rl.rlim_cur;
- }
- }
-
- // 3 for std{in,out,err}
- int nopen = 3 + cmdline->nopen_files;
-
- // Check /proc/self/fd for the current number of open fds, if possible
- // (we may have inherited more than just the standard ones)
- DIR *dir = opendir("/proc/self/fd");
- if (!dir) {
- dir = opendir("/dev/fd");
- }
- if (dir) {
- // Account for 'dir' itself
- nopen = -1;
-
- struct dirent *de;
- while (xreaddir(dir, &de) == 0 && de) {
- ++nopen;
- }
-
- closedir(dir);
- }
-
- ret -= nopen;
- ret -= cmdline->expr->persistent_fds;
- ret -= cmdline->expr->ephemeral_fds;
-
- // bftw() needs at least 2 available fds
- if (ret < 2) {
- ret = 2;
- }
-
- return ret;
-}
-
-/**
- * Dump the bftw() flags for -D search.
- */
-static void dump_bftw_flags(enum bftw_flags flags) {
- DEBUG_FLAG(flags, 0);
- DEBUG_FLAG(flags, BFTW_STAT);
- DEBUG_FLAG(flags, BFTW_RECOVER);
- DEBUG_FLAG(flags, BFTW_DEPTH);
- DEBUG_FLAG(flags, BFTW_COMFOLLOW);
- DEBUG_FLAG(flags, BFTW_LOGICAL);
- DEBUG_FLAG(flags, BFTW_DETECT_CYCLES);
- DEBUG_FLAG(flags, BFTW_XDEV);
-
- assert(!flags);
-}
-
-/**
- * Dump the bftw_strategy for -D search.
- */
-static const char *dump_bftw_strategy(enum bftw_strategy strategy) {
- static const char *strategies[] = {
- DUMP_BFTW_MAP(BFTW_BFS),
- DUMP_BFTW_MAP(BFTW_DFS),
- DUMP_BFTW_MAP(BFTW_IDS),
- };
- return strategies[strategy];
-}
-
-/**
- * Evaluate the command line.
- */
-int eval_cmdline(const struct cmdline *cmdline) {
- if (!cmdline->expr) {
- return EXIT_SUCCESS;
- }
-
- struct callback_args args = {
- .cmdline = cmdline,
- .ret = EXIT_SUCCESS,
- };
-
- struct trie seen;
- if (cmdline->unique) {
- trie_init(&seen);
- args.seen = &seen;
- }
-
- struct bftw_args bftw_args = {
- .paths = cmdline->paths,
- .npaths = cmdline->npaths,
- .callback = cmdline_callback,
- .ptr = &args,
- .nopenfd = infer_fdlimit(cmdline),
- .flags = cmdline->flags,
- .strategy = cmdline->strategy,
- .mtab = cmdline->mtab,
- };
-
- if (cmdline->debug & DEBUG_SEARCH) {
- fprintf(stderr, "bftw({\n");
- fprintf(stderr, "\t.paths = {\n");
- for (size_t i = 0; i < bftw_args.npaths; ++i) {
- fprintf(stderr, "\t\t\"%s\",\n", bftw_args.paths[i]);
- }
- fprintf(stderr, "\t},\n");
- fprintf(stderr, "\t.npaths = %zu,\n", bftw_args.npaths);
- fprintf(stderr, "\t.callback = cmdline_callback,\n");
- fprintf(stderr, "\t.ptr = &args,\n");
- fprintf(stderr, "\t.nopenfd = %d,\n", bftw_args.nopenfd);
- fprintf(stderr, "\t.flags = ");
- dump_bftw_flags(bftw_args.flags);
- fprintf(stderr, ",\n\t.strategy = %s,\n", dump_bftw_strategy(bftw_args.strategy));
- fprintf(stderr, "\t.mtab = ");
- if (bftw_args.mtab) {
- fprintf(stderr, "cmdline->mtab");
- } else {
- fprintf(stderr, "NULL");
- }
- fprintf(stderr, ",\n})\n");
- }
-
- if (bftw(&bftw_args) != 0) {
- args.ret = EXIT_FAILURE;
- perror("bftw()");
- }
-
- if (eval_exec_finish(cmdline->expr, cmdline) != 0) {
- args.ret = EXIT_FAILURE;
- }
-
- if (cmdline->debug & DEBUG_RATES) {
- dump_cmdline(cmdline, true);
- }
-
- if (cmdline->unique) {
- trie_destroy(&seen);
- }
-
- return args.ret;
-}