fixup! Test execution of projects

aya-rs · Oct 13, 2024 · 0c96d08 · 0c96d08
1 parent a1cb382
commit 0c96d08
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/test.sh b/test.sh
@@ -26,7 +26,7 @@ case "${PROG_TYPE}" in
         ADDITIONAL_ARGS=(-d fn_name=try_to_wake_up)
         ;;
     "kprobe"|"kretprobe")
-        ADDITIONAL_ARGS=(-d kprobe=test)
+        ADDITIONAL_ARGS=(-d kprobe=do_unlinkat)
         ;;
     "lsm")
         ADDITIONAL_ARGS=(-d lsm_hook=file_open)

diff --git a/{{project-name}}-ebpf/src/main.rs b/{{project-name}}-ebpf/src/main.rs
@@ -5,7 +5,7 @@
 use aya_ebpf::{macros::kprobe, programs::ProbeContext};
 use aya_log_ebpf::info;
 
-#[kprobe]
+#[kprobe(function = "{{kprobe}}")]
 pub fn {{crate_name}}(ctx: ProbeContext) -> u32 {
     match try_{{crate_name}}(ctx) {
         Ok(ret) => ret,
@@ -21,7 +21,7 @@ fn try_{{crate_name}}(ctx: ProbeContext) -> Result<u32, u32> {
 use aya_ebpf::{macros::kretprobe, programs::RetProbeContext};
 use aya_log_ebpf::info;
 
-#[kretprobe]
+#[kretprobe(function = "{{kprobe}}")]
 pub fn {{crate_name}}(ctx: RetProbeContext) -> u32 {
     match try_{{crate_name}}(ctx) {
         Ok(ret) => ret,