fix(bpf): ensure ring buffer wakeup on TCP, UDP, and syscall events

nullswan · Sep 18, 2024 · b976efd · b976efd
1 parent 62a5b8b
commit b976efd
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/bpf/kprobe/trace_tcp.c b/bpf/kprobe/trace_tcp.c
@@ -51,7 +51,7 @@ int trace_tcp_recvmsg(struct tcp_recvmsg_args *ctx) {
   e.direction = 0;
   e.protocol = 6;
 
-  bpf_ringbuf_output(&network_events_rb, &e, sizeof(e), BPF_RB_NO_WAKEUP);
+  bpf_ringbuf_output(&network_events_rb, &e, sizeof(e), BPF_RB_FORCE_WAKEUP);
   return 0;
 }
 
@@ -102,6 +102,6 @@ int trace_tcp_sendmsg(struct tcp_sendmsg_args *ctx) {
   e.direction = 1;
   e.protocol = 6;
 
-  bpf_ringbuf_output(&network_events_rb, &e, sizeof(e), BPF_RB_NO_WAKEUP);
+  bpf_ringbuf_output(&network_events_rb, &e, sizeof(e), BPF_RB_FORCE_WAKEUP);
   return 0;
 }
diff --git a/bpf/kprobe/trace_udp.c b/bpf/kprobe/trace_udp.c
@@ -50,7 +50,7 @@ int trace_udp_recvmsg(struct udp_recvmsg_args *ctx) {
   e.direction = 0;
   e.protocol = 17;
 
-  bpf_ringbuf_output(&network_events_rb, &e, sizeof(e), BPF_RB_NO_WAKEUP);
+  bpf_ringbuf_output(&network_events_rb, &e, sizeof(e), BPF_RB_FORCE_WAKEUP);
   return 0;
 }
 
@@ -101,6 +101,6 @@ int trace_udp_sendmsg(struct udp_sendmsg_args *ctx) {
   e.sport = sport;
   e.dport = dport;
 
-  bpf_ringbuf_output(&network_events_rb, &e, sizeof(e), BPF_RB_NO_WAKEUP);
+  bpf_ringbuf_output(&network_events_rb, &e, sizeof(e), BPF_RB_FORCE_WAKEUP);
   return 0;
 }
diff --git a/bpf/tracepoint/trace_syscall.c b/bpf/tracepoint/trace_syscall.c
@@ -28,6 +28,6 @@ int tracepoint_sys_enter(struct syscall_trace_enter_args *ctx) {
     .pid = pid
   };
 
-  bpf_ringbuf_output(&network_events_rb, &syscall_event, sizeof(syscall_event), BPF_RB_NO_WAKEUP);
+  bpf_ringbuf_output(&syscall_events_rb, &syscall_event, sizeof(syscall_event), BPF_RB_FORCE_WAKEUP);
   return 0;
 }