diff options
author | Tavian Barnes <tavianator@tavianator.com> | 2016-02-22 14:13:26 -0500 |
---|---|---|
committer | Tavian Barnes <tavianator@tavianator.com> | 2016-02-22 14:13:26 -0500 |
commit | 275fbdbe040dc07b2705c92d4cd24ad3f3c7d069 (patch) | |
tree | 0fd997ea04f5ef16c22941d2aabbc01d900073fe | |
parent | 59f46665aa10eed0776e93dc5f67368a3b8b27e3 (diff) |
Implement -D stat.
-rw-r--r-- | bfs.h | 13 | ||||
-rw-r--r-- | eval.c | 31 | ||||
-rw-r--r-- | parse.c | 40 |
3 files changed, 83 insertions, 1 deletions
@@ -49,6 +49,14 @@ struct eval_state; typedef bool eval_fn(const struct expr *expr, struct eval_state *state); /** + * Various debugging flags. + */ +enum debugflags { + /** Trace all stat() calls. */ + DEBUG_STAT = 1 << 0, +}; + +/** * The parsed command line. */ struct cmdline { @@ -70,7 +78,10 @@ struct cmdline { int maxdepth; /** bftw() flags. */ - int flags; + enum bftw_flags flags; + + /** Debugging flags. */ + enum debugflags debug; /** The command line expression. */ struct expr *expr; @@ -520,6 +520,33 @@ bool eval_comma(const struct expr *expr, struct eval_state *state) { } /** + * Debug stat() calls. + */ +void debug_stat(const struct eval_state *state) { + struct BFTW *ftwbuf = state->ftwbuf; + + fprintf(stderr, "fstatat("); + 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); + + if (ftwbuf->at_flags == AT_SYMLINK_NOFOLLOW) { + fprintf(stderr, "AT_SYMLINK_NOFOLLOW"); + } else { + fprintf(stderr, "%d", ftwbuf->at_flags); + } + + fprintf(stderr, ")\n"); +} + +/** * Type passed as the argument to the bftw() callback. */ struct callback_args { @@ -567,6 +594,10 @@ static enum bftw_action cmdline_callback(struct BFTW *ftwbuf, void *ptr) { eval_expr(cmdline->expr, &state); } + if ((cmdline->debug & DEBUG_STAT) && ftwbuf->statbuf) { + debug_stat(&state); + } + args->ret = state.ret; return state.action; } @@ -336,6 +336,39 @@ static struct expr *parse_test_sdata(struct parser_state *state, const char *tes } /** + * Parse -D FLAG. + */ +static struct expr *parse_debug(struct parser_state *state, const char *option) { + struct cmdline *cmdline = state->cmdline; + + const char *arg = state->argv[state->i]; + if (!arg) { + pretty_error(cmdline->stderr_colors, + "error: %s needs a flag.\n", option); + return NULL; + } + + ++state->i; + + if (strcmp(arg, "help") == 0) { + printf("Supported debug flags:\n\n"); + + printf(" help: This message.\n"); + printf(" stat: Trace all stat() calls.\n"); + + state->just_info = true; + return NULL; + } else if (strcmp(arg, "stat") == 0) { + cmdline->debug |= DEBUG_STAT; + } else { + pretty_warning(cmdline->stderr_colors, + "warning: Unrecognized debug flag '%s'.\n\n", arg); + } + + return new_positional_option(state); +} + +/** * Parse -[acm]{min,time}. */ static struct expr *parse_acmtime(struct parser_state *state, const char *option, enum timefield field, enum timeunit unit) { @@ -670,6 +703,12 @@ static struct expr *parse_literal(struct parser_state *state) { } switch (arg[1]) { + case 'D': + if (strcmp(arg, "-D") == 0) { + return parse_debug(state, arg); + } + break; + case 'P': if (strcmp(arg, "-P") == 0) { cmdline->flags &= ~(BFTW_FOLLOW | BFTW_DETECT_CYCLES); @@ -1128,6 +1167,7 @@ struct cmdline *parse_cmdline(int argc, char *argv[]) { cmdline->mindepth = 0; cmdline->maxdepth = INT_MAX; cmdline->flags = BFTW_RECOVER; + cmdline->debug = 0; cmdline->expr = &expr_true; cmdline->colors = parse_colors(getenv("LS_COLORS")); |