mirrors/opensnitch
1
0
Fork 0
mirror of https://github.com/evilsocket/opensnitch.git synced 2025-03-04 08:34:40 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

ebpf: performance improvement for opensnitch-procs

Browse source 
We were sending to userspace unnecessary exit events, consuming
unnecessary CPU cycles.

We only intercept execve and execveat, but sched_process_exit is invoked
by more functions (sched_process_exit, clone, ...), so we were receiving
on the daemon events that we did nothing with them, apart from consuming
CPU cycles.

On some scenarios like on servers running saltstack (as salt-master),
this caused to consume more CPU than needed.

cherry picked from 15fcf67535
This commit is contained in:
Gustavo Iñiguez Goia 2024-04-29 01:06:39 +02:00
parent 81dd625a1c
commit 7442bec96f
Failed to generate hash of commit
1 changed files with 16 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
ebpf_prog/opensnitch-procs.c
View file

@ -46,14 +46,12 @@ static __always_inline void __handle_exit_execve(struct trace_sys_exit_execve *c
{ {
u64 pid_tgid = bpf_get_current_pid_tgid(); u64 pid_tgid = bpf_get_current_pid_tgid();
struct data_t *proc = bpf_map_lookup_elem(&execMap, &pid_tgid); struct data_t *proc = bpf_map_lookup_elem(&execMap, &pid_tgid);
// don't delete the pid from execMap here, delegate it to sched_process_exit
if (proc == NULL) { return; } if (proc == NULL) { return; }
if (ctx->ret != 0) { goto out; } if (ctx->ret != 0) { return; }
proc->ret_code = ctx->ret; proc->ret_code = ctx->ret;
bpf_perf_event_output(ctx, &events, BPF_F_CURRENT_CPU, proc, sizeof(*proc)); bpf_perf_event_output(ctx, &events, BPF_F_CURRENT_CPU, proc, sizeof(*proc));
out:
bpf_map_delete_elem(&execMap, &pid_tgid);
} }
// https://0xax.gitbooks.io/linux-insides/content/SysCall/linux-syscall-4.html // https://0xax.gitbooks.io/linux-insides/content/SysCall/linux-syscall-4.html
@ -63,6 +61,14 @@ out:
SEC("tracepoint/sched/sched_process_exit") SEC("tracepoint/sched/sched_process_exit")
int tracepoint__sched_sched_process_exit(struct pt_regs *ctx) int tracepoint__sched_sched_process_exit(struct pt_regs *ctx)
{ {
u64 pid_tgid = bpf_get_current_pid_tgid();
struct data_t *proc = bpf_map_lookup_elem(&execMap, &pid_tgid);
// if the pid is not in execMap cache (because it's not of a pid we've
// previously intercepted), do not send the event to userspace, because
// we won't do anything with it and it consumes CPU cycles (too much in some
// scenarios).
if (proc == NULL) { return 0; }
int zero = 0; int zero = 0;
struct data_t *data = bpf_map_lookup_elem(&heapstore, &zero); struct data_t *data = bpf_map_lookup_elem(&heapstore, &zero);
if (!data){ return 0; } if (!data){ return 0; }
@ -71,7 +77,6 @@ int tracepoint__sched_sched_process_exit(struct pt_regs *ctx)
data->type = EVENT_SCHED_EXIT; data->type = EVENT_SCHED_EXIT;
bpf_perf_event_output(ctx, &events, BPF_F_CURRENT_CPU, data, sizeof(*data)); bpf_perf_event_output(ctx, &events, BPF_F_CURRENT_CPU, data, sizeof(*data));
u64 pid_tgid = bpf_get_current_pid_tgid();
bpf_map_delete_elem(&execMap, &pid_tgid); bpf_map_delete_elem(&execMap, &pid_tgid);
return 0; return 0;
}; };