| /* SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-Clause) */ |
| |
| /* |
| * mountsnoop Trace mount and umount[2] syscalls |
| * |
| * Copyright (c) 2021 Hengqi Chen |
| * 30-May-2021 Hengqi Chen Created this. |
| */ |
| #ifndef _GNU_SOURCE |
| #define _GNU_SOURCE |
| #endif |
| #include <argp.h> |
| #include <errno.h> |
| #include <signal.h> |
| #include <string.h> |
| #include <time.h> |
| |
| #include <bpf/libbpf.h> |
| #include <bpf/bpf.h> |
| #include "mountsnoop.h" |
| #include "mountsnoop.skel.h" |
| #include "trace_helpers.h" |
| |
| #define PERF_BUFFER_PAGES 64 |
| #define PERF_POLL_TIMEOUT_MS 100 |
| #define warn(...) fprintf(stderr, __VA_ARGS__) |
| |
| /* https://www.gnu.org/software/gnulib/manual/html_node/strerrorname_005fnp.html */ |
| #if !defined(__GLIBC__) || __GLIBC__ < 2 || (__GLIBC__ == 2 && __GLIBC_MINOR__ < 32) |
| const char *strerrorname_np(int errnum) |
| { |
| return NULL; |
| } |
| #endif |
| |
| static volatile sig_atomic_t exiting = 0; |
| |
| static pid_t target_pid = 0; |
| static bool emit_timestamp = false; |
| static bool output_vertically = false; |
| static bool verbose = false; |
| static const char *flag_names[] = { |
| [0] = "MS_RDONLY", |
| [1] = "MS_NOSUID", |
| [2] = "MS_NODEV", |
| [3] = "MS_NOEXEC", |
| [4] = "MS_SYNCHRONOUS", |
| [5] = "MS_REMOUNT", |
| [6] = "MS_MANDLOCK", |
| [7] = "MS_DIRSYNC", |
| [8] = "MS_NOSYMFOLLOW", |
| [9] = "MS_NOATIME", |
| [10] = "MS_NODIRATIME", |
| [11] = "MS_BIND", |
| [12] = "MS_MOVE", |
| [13] = "MS_REC", |
| [14] = "MS_VERBOSE", |
| [15] = "MS_SILENT", |
| [16] = "MS_POSIXACL", |
| [17] = "MS_UNBINDABLE", |
| [18] = "MS_PRIVATE", |
| [19] = "MS_SLAVE", |
| [20] = "MS_SHARED", |
| [21] = "MS_RELATIME", |
| [22] = "MS_KERNMOUNT", |
| [23] = "MS_I_VERSION", |
| [24] = "MS_STRICTATIME", |
| [25] = "MS_LAZYTIME", |
| [26] = "MS_SUBMOUNT", |
| [27] = "MS_NOREMOTELOCK", |
| [28] = "MS_NOSEC", |
| [29] = "MS_BORN", |
| [30] = "MS_ACTIVE", |
| [31] = "MS_NOUSER", |
| }; |
| static const int flag_count = sizeof(flag_names) / sizeof(flag_names[0]); |
| |
| const char *argp_program_version = "mountsnoop 0.1"; |
| const char *argp_program_bug_address = |
| "https://github.com/iovisor/bcc/tree/master/libbpf-tools"; |
| const char argp_program_doc[] = |
| "Trace mount and umount syscalls.\n" |
| "\n" |
| "USAGE: mountsnoop [-h] [-t] [-p PID] [-v]\n" |
| "\n" |
| "EXAMPLES:\n" |
| " mountsnoop # trace mount and umount syscalls\n" |
| " mountsnoop -d # detailed output (one line per column value)\n" |
| " mountsnoop -p 1216 # only trace PID 1216\n"; |
| |
| static const struct argp_option opts[] = { |
| { "pid", 'p', "PID", 0, "Process ID to trace" }, |
| { "timestamp", 't', NULL, 0, "Include timestamp on output" }, |
| { "detailed", 'd', NULL, 0, "Output result in detail mode" }, |
| { "verbose", 'v', NULL, 0, "Verbose debug output" }, |
| { NULL, 'h', NULL, OPTION_HIDDEN, "Show the full help" }, |
| {}, |
| }; |
| |
| static error_t parse_arg(int key, char *arg, struct argp_state *state) |
| { |
| long pid; |
| |
| switch (key) { |
| case 'p': |
| errno = 0; |
| pid = strtol(arg, NULL, 10); |
| if (errno || pid <= 0) { |
| warn("Invalid PID: %s\n", arg); |
| argp_usage(state); |
| } |
| target_pid = pid; |
| break; |
| case 't': |
| emit_timestamp = true; |
| break; |
| case 'd': |
| output_vertically = true; |
| break; |
| case 'v': |
| verbose = true; |
| break; |
| case 'h': |
| argp_state_help(state, stderr, ARGP_HELP_STD_HELP); |
| break; |
| default: |
| return ARGP_ERR_UNKNOWN; |
| } |
| return 0; |
| } |
| |
| static int libbpf_print_fn(enum libbpf_print_level level, const char *format, va_list args) |
| { |
| if (level == LIBBPF_DEBUG && !verbose) |
| return 0; |
| return vfprintf(stderr, format, args); |
| } |
| |
| static void sig_int(int signo) |
| { |
| exiting = 1; |
| } |
| |
| static const char *strflags(__u64 flags) |
| { |
| static char str[512]; |
| int i; |
| |
| if (!flags) |
| return "0x0"; |
| |
| str[0] = '\0'; |
| for (i = 0; i < flag_count; i++) { |
| if (!((1 << i) & flags)) |
| continue; |
| if (str[0]) |
| strcat(str, " | "); |
| strcat(str, flag_names[i]); |
| } |
| return str; |
| } |
| |
| static const char *strerrno(int errnum) |
| { |
| const char *errstr; |
| static char ret[32] = {}; |
| |
| if (!errnum) |
| return "0"; |
| |
| ret[0] = '\0'; |
| errstr = strerrorname_np(-errnum); |
| if (!errstr) { |
| snprintf(ret, sizeof(ret), "%d", errnum); |
| return ret; |
| } |
| |
| snprintf(ret, sizeof(ret), "-%s", errstr); |
| return ret; |
| } |
| |
| static const char *gen_call(const struct event *e) |
| { |
| static char call[10240]; |
| |
| memset(call, 0, sizeof(call)); |
| if (e->op == UMOUNT) { |
| snprintf(call, sizeof(call), "umount(\"%s\", %s) = %s", |
| e->dest, strflags(e->flags), strerrno(e->ret)); |
| } else { |
| snprintf(call, sizeof(call), "mount(\"%s\", \"%s\", \"%s\", %s, \"%s\") = %s", |
| e->src, e->dest, e->fs, strflags(e->flags), e->data, strerrno(e->ret)); |
| } |
| return call; |
| } |
| |
| static void handle_event(void *ctx, int cpu, void *data, __u32 data_sz) |
| { |
| const struct event *e = data; |
| struct tm *tm; |
| char ts[32]; |
| time_t t; |
| const char *indent; |
| static const char *op_name[] = { |
| [MOUNT] = "MOUNT", |
| [UMOUNT] = "UMOUNT", |
| }; |
| |
| if (emit_timestamp) { |
| time(&t); |
| tm = localtime(&t); |
| strftime(ts, sizeof(ts), "%H:%M:%S ", tm); |
| printf("%s", ts); |
| indent = " "; |
| } else { |
| indent = ""; |
| } |
| if (!output_vertically) { |
| printf("%-16s %-7d %-7d %-11u %s\n", |
| e->comm, e->pid, e->tid, e->mnt_ns, gen_call(e)); |
| return; |
| } |
| if (emit_timestamp) |
| printf("\n"); |
| printf("%sPID: %d\n", indent, e->pid); |
| printf("%sTID: %d\n", indent, e->tid); |
| printf("%sCOMM: %s\n", indent, e->comm); |
| printf("%sOP: %s\n", indent, op_name[e->op]); |
| printf("%sRET: %s\n", indent, strerrno(e->ret)); |
| printf("%sLAT: %lldus\n", indent, e->delta / 1000); |
| printf("%sMNT_NS: %u\n", indent, e->mnt_ns); |
| printf("%sFS: %s\n", indent, e->fs); |
| printf("%sSOURCE: %s\n", indent, e->src); |
| printf("%sTARGET: %s\n", indent, e->dest); |
| printf("%sDATA: %s\n", indent, e->data); |
| printf("%sFLAGS: %s\n", indent, strflags(e->flags)); |
| printf("\n"); |
| } |
| |
| static void handle_lost_events(void *ctx, int cpu, __u64 lost_cnt) |
| { |
| warn("lost %llu events on CPU #%d\n", lost_cnt, cpu); |
| } |
| |
| int main(int argc, char **argv) |
| { |
| static const struct argp argp = { |
| .options = opts, |
| .parser = parse_arg, |
| .doc = argp_program_doc, |
| }; |
| struct perf_buffer *pb = NULL; |
| struct mountsnoop_bpf *obj; |
| int err; |
| |
| err = argp_parse(&argp, argc, argv, 0, NULL, NULL); |
| if (err) |
| return err; |
| |
| libbpf_set_strict_mode(LIBBPF_STRICT_ALL); |
| libbpf_set_print(libbpf_print_fn); |
| |
| obj = mountsnoop_bpf__open(); |
| if (!obj) { |
| warn("failed to open BPF object\n"); |
| return 1; |
| } |
| |
| obj->rodata->target_pid = target_pid; |
| |
| err = mountsnoop_bpf__load(obj); |
| if (err) { |
| warn("failed to load BPF object: %d\n", err); |
| goto cleanup; |
| } |
| |
| err = mountsnoop_bpf__attach(obj); |
| if (err) { |
| warn("failed to attach BPF programs: %d\n", err); |
| goto cleanup; |
| } |
| |
| pb = perf_buffer__new(bpf_map__fd(obj->maps.events), PERF_BUFFER_PAGES, |
| handle_event, handle_lost_events, NULL, NULL); |
| if (!pb) { |
| err = -errno; |
| warn("failed to open perf buffer: %d\n", err); |
| goto cleanup; |
| } |
| |
| if (signal(SIGINT, sig_int) == SIG_ERR) { |
| warn("can't set signal handler: %s\n", strerror(errno)); |
| err = 1; |
| goto cleanup; |
| } |
| |
| if (!output_vertically) { |
| if (emit_timestamp) |
| printf("%-8s ", "TIME"); |
| printf("%-16s %-7s %-7s %-11s %s\n", "COMM", "PID", "TID", "MNT_NS", "CALL"); |
| } |
| |
| while (!exiting) { |
| err = perf_buffer__poll(pb, PERF_POLL_TIMEOUT_MS); |
| if (err < 0 && err != -EINTR) { |
| fprintf(stderr, "error polling perf buffer: %s\n", strerror(-err)); |
| goto cleanup; |
| } |
| /* reset err to return 0 if exiting */ |
| err = 0; |
| } |
| |
| cleanup: |
| perf_buffer__free(pb); |
| mountsnoop_bpf__destroy(obj); |
| |
| return err != 0; |
| } |