From mboxrd@z Thu Jan 1 00:00:00 1970 Received: from smtp.kernel.org (aws-us-west-2-korg-mail-1.web.codeaurora.org [10.30.226.201]) (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) (No client certificate requested) by smtp.subspace.kernel.org (Postfix) with ESMTPS id BF96BA932 for ; Fri, 30 Jun 2023 08:37:59 +0000 (UTC) Received: by smtp.kernel.org (Postfix) with ESMTPSA id CE567C433C8; Fri, 30 Jun 2023 08:37:55 +0000 (UTC) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/simple; d=kernel.org; s=k20201202; t=1688114279; bh=/x1YugDAnJqeKbdO2ZzXsemQhKAviiV9T8VjqNeHZik=; h=From:To:Cc:Subject:Date:In-Reply-To:References:From; b=LoKtFMa+FYVnGm4GJwxl3DgSYsXseNC+gz5URlf9jffGvqHmCP4JNxbpv5IuR2nQa tC0/+mcDhIAuCgw5nKsbylYZp4zjeEDfAzoFOstotyXs5gnDWZ2HPPVSjb8n21+O79 qjZOWA+HI91Y3MI6+lC3mHmjBdwXzkAgyrtPu4IDVjF8BG2hO8biV651noCK1qLVll RXTmZxGWzkmidwTI5kMbv27+LfY0dZDwUGu1Z30xNpjoIQ/80R+VRwfh9+Uz3/6OiJ Wn39YLVuP3e1tluOztqkZV0+t78zjHlkh5I/vbQoEp4FLX2Vw1Z9OwDPePpIncXRSA +ZGoGBX/qE/Kg== From: Jiri Olsa To: Alexei Starovoitov , Daniel Borkmann , Andrii Nakryiko Cc: bpf@vger.kernel.org, Martin KaFai Lau , Song Liu , Yonghong Song , John Fastabend , KP Singh , Stanislav Fomichev , Hao Luo Subject: [PATCHv3 bpf-next 21/26] selftests/bpf: Add uprobe_multi bench test Date: Fri, 30 Jun 2023 10:33:39 +0200 Message-ID: <20230630083344.984305-22-jolsa@kernel.org> X-Mailer: git-send-email 2.41.0 In-Reply-To: <20230630083344.984305-1-jolsa@kernel.org> References: <20230630083344.984305-1-jolsa@kernel.org> Precedence: bulk X-Mailing-List: bpf@vger.kernel.org List-Id: List-Subscribe: List-Unsubscribe: MIME-Version: 1.0 Content-Transfer-Encoding: 8bit Adding test that attaches 50k uprobes in uprobe_multi binary. After the attach is done we run the binary and make sure we get proper amount of hits. Signed-off-by: Jiri Olsa --- .../bpf/prog_tests/uprobe_multi_test.c | 56 +++++++++++++++++++ .../selftests/bpf/progs/uprobe_multi.c | 9 +++ 2 files changed, 65 insertions(+) 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 fd858636b8b0..547d46965d70 100644 --- a/tools/testing/selftests/bpf/prog_tests/uprobe_multi_test.c +++ b/tools/testing/selftests/bpf/prog_tests/uprobe_multi_test.c @@ -202,6 +202,60 @@ static void test_link_api(void) free(offsets); } +static inline __u64 get_time_ns(void) +{ + struct timespec t; + + clock_gettime(CLOCK_MONOTONIC, &t); + return (__u64) t.tv_sec * 1000000000 + t.tv_nsec; +} + +static void test_bench_attach_uprobe(void) +{ + long attach_start_ns, attach_end_ns; + long detach_start_ns, detach_end_ns; + double attach_delta, detach_delta; + struct uprobe_multi *skel = NULL; + struct bpf_program *prog; + int err; + + skel = uprobe_multi__open(); + if (!ASSERT_OK_PTR(skel, "uprobe_multi__open")) + goto cleanup; + + bpf_object__for_each_program(prog, skel->obj) + bpf_program__set_autoload(prog, false); + + bpf_program__set_autoload(skel->progs.test_uprobe_bench, true); + + err = uprobe_multi__load(skel); + if (!ASSERT_EQ(err, 0, "uprobe_multi__load")) + goto cleanup; + + attach_start_ns = get_time_ns(); + + err = uprobe_multi__attach(skel); + if (!ASSERT_OK(err, "uprobe_multi__attach")) + goto cleanup; + + attach_end_ns = get_time_ns(); + + system("./uprobe_multi"); + + ASSERT_EQ(skel->bss->count, 50000, "uprobes_count"); + +cleanup: + detach_start_ns = get_time_ns(); + uprobe_multi__destroy(skel); + detach_end_ns = get_time_ns(); + + attach_delta = (attach_end_ns - attach_start_ns) / 1000000000.0; + detach_delta = (detach_end_ns - detach_start_ns) / 1000000000.0; + + printf("%s: attached in %7.3lfs\n", __func__, attach_delta); + printf("%s: detached in %7.3lfs\n", __func__, detach_delta); +} + void test_uprobe_multi_test(void) { if (test__start_subtest("skel_api")) @@ -212,4 +266,6 @@ void test_uprobe_multi_test(void) test_attach_api_syms(); if (test__start_subtest("link_api")) test_link_api(); + if (test__start_subtest("bench_uprobe")) + test_bench_attach_uprobe(); } diff --git a/tools/testing/selftests/bpf/progs/uprobe_multi.c b/tools/testing/selftests/bpf/progs/uprobe_multi.c index 1eeb9b7b9cad..cd73139dc881 100644 --- a/tools/testing/selftests/bpf/progs/uprobe_multi.c +++ b/tools/testing/selftests/bpf/progs/uprobe_multi.c @@ -89,3 +89,12 @@ int test_uretprobe_sleep(struct pt_regs *ctx) uprobe_multi_check(ctx, true, true); return 0; } + +int count; + +SEC("?uprobe.multi/./uprobe_multi:uprobe_multi_func_*") +int test_uprobe_bench(struct pt_regs *ctx) +{ + count++; + return 0; +} -- 2.41.0