[PATCHv4 14/14] selftests/bpf: Add consumers stress test on single uprobe

Jiri Olsa
2024-09-17T08:53:22Z

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@kernel.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