summaryrefslogtreecommitdiffstats
path: root/bfs.c
diff options
context:
space:
mode:
Diffstat (limited to 'bfs.c')
-rw-r--r--bfs.c65
1 files changed, 62 insertions, 3 deletions
diff --git a/bfs.c b/bfs.c
index ceded51..2dcb255 100644
--- a/bfs.c
+++ b/bfs.c
@@ -12,6 +12,7 @@
#include "bftw.h"
#include "color.h"
#include <fnmatch.h>
+#include <limits.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
@@ -155,6 +156,11 @@ struct cmdline {
/** -color option. */
bool color;
+ /** -mindepth option. */
+ int mindepth;
+ /** -maxdepth option. */
+ int maxdepth;
+
/** bftw() flags. */
int flags;
@@ -298,6 +304,45 @@ static bool eval_type(const expression *expr, eval_state *state) {
}
/**
+ * Parse an integer.
+ */
+static bool parse_int(const char *str, int *value) {
+ char *endptr;
+ long result = strtol(str, &endptr, 10);
+
+ if (*str == '\0' || *endptr != '\0') {
+ return false;
+ }
+
+ if (result < INT_MIN || result > INT_MAX) {
+ return false;
+ }
+
+ *value = result;
+ return true;
+}
+
+/**
+ * Parse -{min,max}depth N.
+ */
+static expression *parse_depth(parser_state *state, int *depth, const char *option) {
+ const char *arg = state->argv[state->i];
+ if (!arg) {
+ fprintf(stderr, "%s needs a value.\n", option);
+ return NULL;
+ }
+
+ ++state->i;
+
+ if (!parse_int(arg, depth)) {
+ fprintf(stderr, "%s is not a valid integer.\n", arg);
+ return NULL;
+ }
+
+ return new_expression(eval_true);
+}
+
+/**
* Parse -name 'pattern'.
*/
static expression *parse_name(parser_state *state) {
@@ -379,6 +424,10 @@ static expression *parse_literal(parser_state *state) {
return new_expression(eval_hidden);
} else if (strcmp(arg, "-nohidden") == 0) {
return new_expression(eval_nohidden);
+ } else if (strcmp(arg, "-mindepth") == 0) {
+ return parse_depth(state, &state->cl->mindepth, arg);
+ } else if (strcmp(arg, "-maxdepth") == 0) {
+ return parse_depth(state, &state->cl->maxdepth, arg);
} else if (strcmp(arg, "-name") == 0) {
return parse_name(state);
} else if (strcmp(arg, "-print") == 0) {
@@ -580,6 +629,8 @@ static cmdline *parse_cmdline(int argc, char *argv[]) {
cl->nroots = 0;
cl->colors = NULL;
cl->color = isatty(STDOUT_FILENO);
+ cl->mindepth = 0;
+ cl->maxdepth = INT_MAX;
cl->flags = BFTW_RECOVER;
cl->expr = NULL;
@@ -677,14 +728,22 @@ static int cmdline_callback(const char *fpath, const struct BFTW *ftwbuf, void *
.cl = cl,
.ret = BFTW_CONTINUE,
};
- cl->expr->eval(cl->expr, &state);
+
+ if (ftwbuf->level >= cl->maxdepth) {
+ state.ret = BFTW_SKIP_SUBTREE;
+ }
+
+ if (ftwbuf->level >= cl->mindepth) {
+ cl->expr->eval(cl->expr, &state);
+ }
+
return state.ret;
}
/**
* Evaluate the command line.
*/
-static int cmdline_eval(cmdline *cl) {
+static int eval_cmdline(cmdline *cl) {
int ret = 0;
int nopenfd = infer_nopenfd();
@@ -703,7 +762,7 @@ int main(int argc, char *argv[]) {
cmdline *cl = parse_cmdline(argc, argv);
if (cl) {
- if (cmdline_eval(cl) == 0) {
+ if (eval_cmdline(cl) == 0) {
ret = EXIT_SUCCESS;
}
}