summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorPaul Buetow <paul@buetow.org>2024-02-10 18:54:04 +0200
committerPaul Buetow <paul@buetow.org>2024-02-10 18:54:04 +0200
commit465c686b125bd287db25efb26625c946ab6a4771 (patch)
tree8d2456b605c37b583788ccb957444384feb19dc9
parent25d56de3712412bfb5b271ff9fbad7bf8d75288e (diff)
refactor
-rw-r--r--main.bpf.c22
-rw-r--r--main.go6
2 files changed, 14 insertions, 14 deletions
diff --git a/main.bpf.c b/main.bpf.c
index d494991..7c275a5 100644
--- a/main.bpf.c
+++ b/main.bpf.c
@@ -8,7 +8,7 @@
// For now, this is set to my own user for development purposes.
#define UID_FILTER 1001
-struct openat_event {
+struct open_event {
int fd;
int syscall_id;
u32 tid;
@@ -20,15 +20,15 @@ struct {
__uint(type, BPF_MAP_TYPE_PERF_EVENT_ARRAY);
__uint(key_size, sizeof(u32));
__uint(value_size, sizeof(u32));
-} events SEC(".maps");
+} open_event_map SEC(".maps");
// Map to temporarily store the filename from sys_enter_openat
struct {
__uint(type, BPF_MAP_TYPE_HASH);
__uint(key_size, sizeof(u32));
- __uint(value_size, sizeof(struct openat_event));
+ __uint(value_size, sizeof(struct open_event));
__uint(max_entries, 128); // Adjust size as needed
-} temp_events SEC(".maps");
+} open_event_temp_map SEC(".maps");
SEC("tracepoint/syscalls/sys_enter_open")
int handle_enter_open(struct trace_event_raw_sys_enter *ctx) {
@@ -36,12 +36,12 @@ int handle_enter_open(struct trace_event_raw_sys_enter *ctx) {
return 0;
u32 tid = bpf_get_current_pid_tgid();
- struct openat_event event = { .syscall_id = ctx->id };
+ struct open_event event = { .syscall_id = ctx->id };
bpf_probe_read_user_str(event.filename, sizeof(event.filename), (void *)ctx->args[0]);
bpf_get_current_comm(&event.comm, sizeof(event.comm));
event.tid = tid;
- bpf_map_update_elem(&temp_events, &tid, &event, BPF_ANY);
+ bpf_map_update_elem(&open_event_temp_map, &tid, &event, BPF_ANY);
return 0;
}
@@ -52,13 +52,13 @@ int handle_exit_open(struct trace_event_raw_sys_exit *args) {
return 0;
u32 tid = bpf_get_current_pid_tgid();
- struct openat_event *eventp = bpf_map_lookup_elem(&temp_events, &tid);
+ struct open_event *eventp = bpf_map_lookup_elem(&open_event_temp_map, &tid);
if (!eventp) {
return 0;
}
eventp->fd = args->ret;
- bpf_perf_event_output(args, &events, BPF_F_CURRENT_CPU, eventp, sizeof(struct openat_event));
- bpf_map_delete_elem(&temp_events, &tid);
+ bpf_perf_event_output(args, &open_event_map, BPF_F_CURRENT_CPU, eventp, sizeof(struct open_event));
+ bpf_map_delete_elem(&open_event_temp_map, &tid);
return 0;
}
@@ -70,12 +70,12 @@ int handle_enter_openat(struct trace_event_raw_sys_enter *ctx) {
return 0;
u32 tid = bpf_get_current_pid_tgid();
- struct openat_event event = { .syscall_id = ctx->id };
+ struct open_event event = { .syscall_id = ctx->id };
bpf_probe_read_user_str(event.filename, sizeof(event.filename), (void *)ctx->args[1]);
bpf_get_current_comm(&event.comm, sizeof(event.comm));
event.tid = tid;
- bpf_map_update_elem(&temp_events, &tid, &event, BPF_ANY);
+ bpf_map_update_elem(&open_event_temp_map, &tid, &event, BPF_ANY);
return 0;
}
diff --git a/main.go b/main.go
index 1d3988c..0b330a2 100644
--- a/main.go
+++ b/main.go
@@ -30,7 +30,7 @@ func (e openatEvent) String() string {
}
func resizeMap(module *bpf.Module, name string, size uint32) error {
- m, err := module.GetMap("events")
+ m, err := module.GetMap("open_event_map")
if err != nil {
return err
}
@@ -53,7 +53,7 @@ func main() {
}
defer bpfModule.Close()
- if err = resizeMap(bpfModule, "events", 8192); err != nil {
+ if err = resizeMap(bpfModule, "open_event_map", 8192); err != nil {
log.Fatal(err)
}
@@ -68,7 +68,7 @@ func main() {
eventsChannel := make(chan []byte)
lostChannel := make(chan uint64)
- pb, err := bpfModule.InitPerfBuf("events", eventsChannel, lostChannel, 1)
+ pb, err := bpfModule.InitPerfBuf("open_event_map", eventsChannel, lostChannel, 1)
if err != nil {
log.Fatal(err)
}