| 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
| ||
|
Message-ID: <20240917085024.765883-15-jolsa@kernel.org>
Date: Tue, 17 Sep 2024 10:50:24 +0200
From: Jiri Olsa <jolsa@...nel.org>
To: Oleg Nesterov <oleg@...hat.com>,
Peter Zijlstra <peterz@...radead.org>,
Alexei Starovoitov <ast@...nel.org>,
Daniel Borkmann <daniel@...earbox.net>,
Andrii Nakryiko <andrii@...nel.org>
Cc: bpf@...r.kernel.org,
Martin KaFai Lau <kafai@...com>,
Song Liu <songliubraving@...com>,
Yonghong Song <yhs@...com>,
John Fastabend <john.fastabend@...il.com>,
KP Singh <kpsingh@...omium.org>,
Stanislav Fomichev <sdf@...ichev.me>,
Hao Luo <haoluo@...gle.com>,
Steven Rostedt <rostedt@...dmis.org>,
Masami Hiramatsu <mhiramat@...nel.org>,
linux-kernel@...r.kernel.org,
linux-trace-kernel@...r.kernel.org
Subject: [PATCHv4 14/14] selftests/bpf: Add consumers stress test on single uprobe
We create multiple threads each trying to attach and detach uprobe
consumers on the same uprobe, while main thread is hitting on that
uprobe. All that for 5 seconds.
Signed-off-by: Jiri Olsa <jolsa@...nel.org>
---
.../bpf/prog_tests/uprobe_multi_test.c | 82 +++++++++++++++++++
.../bpf/progs/uprobe_multi_consumer_stress.c | 29 +++++++
2 files changed, 111 insertions(+)
create mode 100644 tools/testing/selftests/bpf/progs/uprobe_multi_consumer_stress.c
diff --git a/tools/testing/selftests/bpf/prog_tests/uprobe_multi_test.c b/tools/testing/selftests/bpf/prog_tests/uprobe_multi_test.c
index 594aa8c06f58..dcc128507212 100644
--- a/tools/testing/selftests/bpf/prog_tests/uprobe_multi_test.c
+++ b/tools/testing/selftests/bpf/prog_tests/uprobe_multi_test.c
@@ -3,6 +3,7 @@
#include <unistd.h>
#include <pthread.h>
#include <test_progs.h>
+#include <time.h>
#include "uprobe_multi.skel.h"
#include "uprobe_multi_bench.skel.h"
#include "uprobe_multi_usdt.skel.h"
@@ -10,6 +11,7 @@
#include "uprobe_multi_session_single.skel.h"
#include "uprobe_multi_session_cookie.skel.h"
#include "uprobe_multi_session_recursive.skel.h"
+#include "uprobe_multi_consumer_stress.skel.h"
#include "uprobe_multi_verifier.skel.h"
#include "bpf/libbpf_internal.h"
#include "testing_helpers.h"
@@ -848,6 +850,84 @@ static void test_bench_attach_usdt(void)
printf("%s: detached in %7.3lfs\n", __func__, detach_delta);
}
+static int done;
+
+static void *worker(void *p)
+{
+ struct uprobe_multi_consumer_stress *skel = p;
+ struct bpf_link *link = NULL;
+
+ srand(time(NULL));
+
+ while (!done) {
+ LIBBPF_OPTS(bpf_uprobe_multi_opts, opts);
+ struct bpf_program *prog;
+
+ switch (rand() % 4) {
+ case 0:
+ prog = skel->progs.uprobe_session_0;
+ opts.session = true;
+ break;
+ case 1:
+ prog = skel->progs.uprobe_session_1;
+ opts.session = true;
+ break;
+ case 2:
+ prog = skel->progs.uprobe;
+ break;
+ case 3:
+ prog = skel->progs.uretprobe;
+ opts.retprobe = true;
+ break;
+ }
+
+ link = bpf_program__attach_uprobe_multi(prog, -1, "/proc/self/exe",
+ "uprobe_multi_func_1", &opts);
+ bpf_link__destroy(link);
+ }
+
+ return NULL;
+}
+
+#define THREAD_COUNT 4
+
+static void test_session_consumer_stress(void)
+{
+ struct uprobe_multi_consumer_stress *skel;
+ pthread_t threads[THREAD_COUNT];
+ time_t start;
+ int i, err;
+
+ /*
+ * We create multiple threads each trying to attach and detach uprobe
+ * consumer on the same uprobe, while main thread is hitting on that
+ * uprobe. All that for 5 seconds.
+ */
+ skel = uprobe_multi_consumer_stress__open_and_load();
+ if (!ASSERT_OK_PTR(skel, "uprobe_multi_consumer_stress"))
+ goto cleanup;
+
+ for (i = 0; i < THREAD_COUNT; i++) {
+ err = pthread_create(threads + i, NULL, worker, skel);
+ if (!ASSERT_OK(err, "pthread_create"))
+ goto join;
+ }
+
+ start = time(NULL);
+
+ while (start + 5 > time(NULL))
+ uprobe_multi_func_1();
+
+ done = 1;
+
+join:
+ for (i = 0; i < THREAD_COUNT; i++)
+ pthread_join(threads[i], NULL);
+
+cleanup:
+ uprobe_multi_consumer_stress__destroy(skel);
+}
+
void test_uprobe_multi_test(void)
{
if (test__start_subtest("skel_api"))
@@ -872,5 +952,7 @@ void test_uprobe_multi_test(void)
test_session_cookie_skel_api();
if (test__start_subtest("session_cookie_recursive"))
test_session_recursive_skel_api();
+ if (test__start_subtest("consumer_stress"))
+ test_session_consumer_stress();
RUN_TESTS(uprobe_multi_verifier);
}
diff --git a/tools/testing/selftests/bpf/progs/uprobe_multi_consumer_stress.c b/tools/testing/selftests/bpf/progs/uprobe_multi_consumer_stress.c
new file mode 100644
index 000000000000..5390108a21ff
--- /dev/null
+++ b/tools/testing/selftests/bpf/progs/uprobe_multi_consumer_stress.c
@@ -0,0 +1,29 @@
+// SPDX-License-Identifier: GPL-2.0
+#include <linux/bpf.h>
+#include <bpf/bpf_tracing.h>
+
+char _license[] SEC("license") = "GPL";
+
+SEC("uprobe.session")
+int uprobe_session_0(struct pt_regs *ctx)
+{
+ return 0;
+}
+
+SEC("uprobe.session")
+int uprobe_session_1(struct pt_regs *ctx)
+{
+ return 1;
+}
+
+SEC("uprobe.multi")
+int uprobe(struct pt_regs *ctx)
+{
+ return 0;
+}
+
+SEC("uprobe.multi")
+int uretprobe(struct pt_regs *ctx)
+{
+ return 0;
+}
--
2.46.0
Powered by blists - more mailing lists