selftests/bpf: Change test_uretprobe_regs_change for uprobe and uretprobe

Changing the test_uretprobe_regs_change test to test both uprobe
and uretprobe by adding entry consumer handler to the testmod
and making it to change one of the registers.

Making sure that changed values both uprobe and uretprobe handlers
propagate to the user space.

Signed-off-by: Jiri Olsa <jolsa@kernel.org>
Signed-off-by: Peter Zijlstra (Intel) <peterz@infradead.org>
Acked-by: Andrii Nakryiko <andrii@kernel.org>
Link: https://lore.kernel.org/r/20250720112133.244369-20-jolsa@kernel.org

diff --git a/tools/testing/selftests/bpf/prog_tests/uprobe_syscall.c b/tools/testing/selftests/bpf/prog_tests/uprobe_syscall.c
index 36ce9e261b5c2f35362b5bfb52488e595c4fef97..c1f945cacebc2741e8949efefe3faae8fc6f4227 100644 (file)
--- a/tools/testing/selftests/bpf/prog_tests/uprobe_syscall.c
+++ b/tools/testing/selftests/bpf/prog_tests/uprobe_syscall.c
@@ -207,7 +207,7 @@ static int write_bpf_testmod_uprobe(unsigned long offset)
        return ret != n ? (int) ret : 0;
 }
 
-static void test_uretprobe_regs_change(void)
+static void test_regs_change(void)
 {
        struct pt_regs before = {}, after = {};
        unsigned long *pb = (unsigned long *) &before;
@@ -221,6 +221,9 @@ static void test_uretprobe_regs_change(void)
        if (!ASSERT_OK(err, "register_uprobe"))
                return;
 
+       /* make sure uprobe gets optimized */
+       uprobe_regs_trigger();
+
        uprobe_regs(&before, &after);
 
        err = write_bpf_testmod_uprobe(0);
@@ -643,7 +646,6 @@ static void test_uretprobe_shadow_stack(void)
 
        test_uprobe_regs_equal(false);
        test_uprobe_regs_equal(true);
-       test_uretprobe_regs_change();
        test_uretprobe_syscall_call();
 
        test_uprobe_legacy();
@@ -651,6 +653,8 @@ static void test_uretprobe_shadow_stack(void)
        test_uprobe_session();
        test_uprobe_usdt();
 
+       test_regs_change();
+
        shstk_is_enabled = false;
 
        ARCH_PRCTL(ARCH_SHSTK_DISABLE, ARCH_SHSTK_SHSTK);
@@ -799,8 +803,6 @@ static void __test_uprobe_syscall(void)
 {
        if (test__start_subtest("uretprobe_regs_equal"))
                test_uprobe_regs_equal(true);
-       if (test__start_subtest("uretprobe_regs_change"))
-               test_uretprobe_regs_change();
        if (test__start_subtest("uretprobe_syscall_call"))
                test_uretprobe_syscall_call();
        if (test__start_subtest("uretprobe_shadow_stack"))
@@ -819,6 +821,8 @@ static void __test_uprobe_syscall(void)
                test_uprobe_sigill();
        if (test__start_subtest("uprobe_regs_equal"))
                test_uprobe_regs_equal(false);
+       if (test__start_subtest("regs_change"))
+               test_regs_change();
 }
 #else
 static void __test_uprobe_syscall(void)

diff --git a/tools/testing/selftests/bpf/test_kmods/bpf_testmod.c b/tools/testing/selftests/bpf/test_kmods/bpf_testmod.c
index e9e918cdf31ff2b15bf41302ad429e8683b834d6..511911053bdc54e80a1d2bfe264967a05b598c35 100644 (file)
--- a/tools/testing/selftests/bpf/test_kmods/bpf_testmod.c
+++ b/tools/testing/selftests/bpf/test_kmods/bpf_testmod.c
@@ -500,15 +500,21 @@ static struct bin_attribute bin_attr_bpf_testmod_file __ro_after_init = {
  */
 #ifdef __x86_64__
 
+static int
+uprobe_handler(struct uprobe_consumer *self, struct pt_regs *regs, __u64 *data)
+{
+       regs->cx = 0x87654321feebdaed;
+       return 0;
+}
+
 static int
 uprobe_ret_handler(struct uprobe_consumer *self, unsigned long func,
                   struct pt_regs *regs, __u64 *data)
 
 {
        regs->ax  = 0x12345678deadbeef;
-       regs->cx  = 0x87654321feebdaed;
        regs->r11 = (u64) -1;
-       return true;
+       return 0;
 }
 
 struct testmod_uprobe {
@@ -520,6 +526,7 @@ struct testmod_uprobe {
 static DEFINE_MUTEX(testmod_uprobe_mutex);
 
 static struct testmod_uprobe uprobe = {
+       .consumer.handler = uprobe_handler,
        .consumer.ret_handler = uprobe_ret_handler,
 };