linux-kernel - [PATCH perf/core 11/11] selftests/bpf: Add uprobe unique attach test

lists.openwall.net		lists / announce owl-users owl-dev john-users john-dev passwdqc-users yescrypt popa3d-users / oss-security kernel-hardening musl sabotage tlsify passwords / crypt-dev xvendor / Bugtraq Full-Disclosure linux-kernel linux-netdev linux-ext4 linux-hardening linux-cve-announce PHC
Open Source and information security mailing list archives

Hash Suite for Android: free password hash cracker in your pocket

[<prev] [next>] [<thread-prev] [thread-next>] [day] [month] [year] [list]

Message-ID: <20250902143504.1224726-12-jolsa@kernel.org>
Date: Tue,  2 Sep 2025 16:35:04 +0200
From: Jiri Olsa <jolsa@...nel.org>
To: Oleg Nesterov <oleg@...hat.com>,
	Masami Hiramatsu <mhiramat@...nel.org>,
	Peter Zijlstra <peterz@...radead.org>,
	Andrii Nakryiko <andrii@...nel.org>
Cc: bpf@...r.kernel.org,
	linux-kernel@...r.kernel.org,
	linux-trace-kernel@...r.kernel.org,
	x86@...nel.org,
	Song Liu <songliubraving@...com>,
	Yonghong Song <yhs@...com>,
	John Fastabend <john.fastabend@...il.com>,
	Hao Luo <haoluo@...gle.com>,
	Steven Rostedt <rostedt@...dmis.org>,
	Ingo Molnar <mingo@...nel.org>
Subject: [PATCH perf/core 11/11] selftests/bpf: Add uprobe unique attach test

Adding test to check the unique uprobe attchment together
with not-unique uprobe on top perf uprobe pmu.

Signed-off-by: Jiri Olsa <jolsa@...nel.org>
---
 .../testing/selftests/bpf/prog_tests/uprobe.c | 111 +++++++++++++++++-
 1 file changed, 110 insertions(+), 1 deletion(-)

diff --git a/tools/testing/selftests/bpf/prog_tests/uprobe.c b/tools/testing/selftests/bpf/prog_tests/uprobe.c
index cf3e0e7a64fa..4e1be03d863d 100644
--- a/tools/testing/selftests/bpf/prog_tests/uprobe.c
+++ b/tools/testing/selftests/bpf/prog_tests/uprobe.c
@@ -33,7 +33,7 @@ static int urand_trigger(FILE **urand_pipe)
 	return exit_code;
 }
 
-void test_uprobe(void)
+static void test_uprobe_urandlib(void)
 {
 	LIBBPF_OPTS(bpf_uprobe_opts, uprobe_opts);
 	struct test_uprobe *skel;
@@ -93,3 +93,112 @@ void test_uprobe(void)
 		pclose(urand_pipe);
 	test_uprobe__destroy(skel);
 }
+
+static noinline void uprobe_unique_trigger(void)
+{
+	asm volatile ("");
+}
+
+static void test_uprobe_unique(void)
+{
+	LIBBPF_OPTS(bpf_uprobe_opts, uprobe_opts,
+		.func_name = "uprobe_unique_trigger",
+	);
+	struct bpf_link *link_1, *link_2 = NULL;
+	struct bpf_program *prog_1, *prog_2;
+	struct test_uprobe *skel;
+
+	skel = test_uprobe__open_and_load();
+	if (!ASSERT_OK_PTR(skel, "test_uprobe__open_and_load"))
+		return;
+
+	skel->bss->my_pid = getpid();
+
+	prog_1 = skel->progs.test1;
+	prog_2 = skel->progs.test2;
+
+	/* not-unique and unique */
+	uprobe_opts.unique = false;
+	link_1 = bpf_program__attach_uprobe_opts(prog_1, -1, "/proc/self/exe",
+						 0 /* offset */, &uprobe_opts);
+	if (!ASSERT_OK_PTR(link_1, "bpf_program__attach_uprobe_opts_1"))
+		goto cleanup;
+
+	uprobe_opts.unique = true;
+	link_2 = bpf_program__attach_uprobe_opts(prog_2, -1, "/proc/self/exe",
+						 0 /* offset */, &uprobe_opts);
+	if (!ASSERT_ERR_PTR(link_2, "bpf_program__attach_uprobe_opts_2")) {
+		bpf_link__destroy(link_2);
+		goto cleanup;
+	}
+
+	bpf_link__destroy(link_1);
+
+	/* unique and unique */
+	uprobe_opts.unique = true;
+	link_1 = bpf_program__attach_uprobe_opts(prog_1, -1, "/proc/self/exe",
+						 0 /* offset */, &uprobe_opts);
+	if (!ASSERT_OK_PTR(link_1, "bpf_program__attach_uprobe_opts_1"))
+		goto cleanup;
+
+	uprobe_opts.unique = true;
+	link_2 = bpf_program__attach_uprobe_opts(prog_2, -1, "/proc/self/exe",
+						 0 /* offset */, &uprobe_opts);
+	if (!ASSERT_ERR_PTR(link_2, "bpf_program__attach_uprobe_opts_2")) {
+		bpf_link__destroy(link_2);
+		goto cleanup;
+	}
+
+	bpf_link__destroy(link_1);
+
+	/* unique and not-unique */
+	uprobe_opts.unique = true;
+	link_1 = bpf_program__attach_uprobe_opts(prog_1, -1, "/proc/self/exe",
+						 0 /* offset */, &uprobe_opts);
+	if (!ASSERT_OK_PTR(link_1, "bpf_program__attach_uprobe_opts_1"))
+		goto cleanup;
+
+	uprobe_opts.unique = false;
+	link_2 = bpf_program__attach_uprobe_opts(prog_2, -1, "/proc/self/exe",
+						 0 /* offset */, &uprobe_opts);
+	if (!ASSERT_ERR_PTR(link_2, "bpf_program__attach_uprobe_opts_2")) {
+		bpf_link__destroy(link_2);
+		goto cleanup;
+	}
+
+	bpf_link__destroy(link_1);
+
+	/* not-unique and not-unique */
+	uprobe_opts.unique = false;
+	link_1 = bpf_program__attach_uprobe_opts(prog_1, -1, "/proc/self/exe",
+						 0 /* offset */, &uprobe_opts);
+	if (!ASSERT_OK_PTR(link_1, "bpf_program__attach_uprobe_opts_1"))
+		goto cleanup;
+
+	uprobe_opts.unique = false;
+	link_2 = bpf_program__attach_uprobe_opts(prog_2, -1, "/proc/self/exe",
+						 0 /* offset */, &uprobe_opts);
+	if (!ASSERT_OK_PTR(link_2, "bpf_program__attach_uprobe_opts_2")) {
+		bpf_link__destroy(link_1);
+		goto cleanup;
+	}
+
+	uprobe_unique_trigger();
+
+	ASSERT_EQ(skel->bss->test1_result, 1, "test1_result");
+	ASSERT_EQ(skel->bss->test2_result, 1, "test2_result");
+
+	bpf_link__destroy(link_1);
+	bpf_link__destroy(link_2);
+
+cleanup:
+	test_uprobe__destroy(skel);
+}
+
+void test_uprobe(void)
+{
+	if (test__start_subtest("urandlib"))
+		test_uprobe_urandlib();
+	if (test__start_subtest("unique"))
+		test_uprobe_unique();
+}
-- 
2.51.0