From: Jiri Olsa <olsajiri@gmail.com>
To: Ihor Solodrai <ihor.solodrai@linux.dev>
Cc: bpf@vger.kernel.org, andrii@kernel.org, ast@kernel.org,
daniel@iogearbox.net, eddyz87@gmail.com, mykolal@fb.com,
kernel-team@meta.com
Subject: Re: [PATCH bpf-next v2 2/2] selftests/bpf: test bpf_usdt_arg_size() function
Date: Wed, 26 Feb 2025 15:36:21 +0100 [thread overview]
Message-ID: <Z78m5XkzuC1L3QPE@krava> (raw)
In-Reply-To: <20250224235756.2612606-2-ihor.solodrai@linux.dev>
On Mon, Feb 24, 2025 at 03:57:56PM -0800, Ihor Solodrai wrote:
> Update usdt tests to also check for correct behavior of
> bpf_usdt_arg_size().
>
> Signed-off-by: Ihor Solodrai <ihor.solodrai@linux.dev>
Reviewed-by: Jiri Olsa <jolsa@kernel.org>
jirka
> ---
> tools/testing/selftests/bpf/prog_tests/usdt.c | 11 ++++++++++-
> tools/testing/selftests/bpf/progs/test_usdt.c | 14 ++++++++++++++
> 2 files changed, 24 insertions(+), 1 deletion(-)
>
> diff --git a/tools/testing/selftests/bpf/prog_tests/usdt.c b/tools/testing/selftests/bpf/prog_tests/usdt.c
> index 56ed1eb9b527..495d66414b57 100644
> --- a/tools/testing/selftests/bpf/prog_tests/usdt.c
> +++ b/tools/testing/selftests/bpf/prog_tests/usdt.c
> @@ -45,7 +45,7 @@ static void subtest_basic_usdt(void)
> LIBBPF_OPTS(bpf_usdt_opts, opts);
> struct test_usdt *skel;
> struct test_usdt__bss *bss;
> - int err;
> + int err, i;
>
> skel = test_usdt__open_and_load();
> if (!ASSERT_OK_PTR(skel, "skel_open"))
> @@ -75,6 +75,7 @@ static void subtest_basic_usdt(void)
> ASSERT_EQ(bss->usdt0_cookie, 0xcafedeadbeeffeed, "usdt0_cookie");
> ASSERT_EQ(bss->usdt0_arg_cnt, 0, "usdt0_arg_cnt");
> ASSERT_EQ(bss->usdt0_arg_ret, -ENOENT, "usdt0_arg_ret");
> + ASSERT_EQ(bss->usdt0_arg_size, -ENOENT, "usdt0_arg_size");
>
> /* auto-attached usdt3 gets default zero cookie value */
> ASSERT_EQ(bss->usdt3_cookie, 0, "usdt3_cookie");
> @@ -86,6 +87,9 @@ static void subtest_basic_usdt(void)
> ASSERT_EQ(bss->usdt3_args[0], 1, "usdt3_arg1");
> ASSERT_EQ(bss->usdt3_args[1], 42, "usdt3_arg2");
> ASSERT_EQ(bss->usdt3_args[2], (uintptr_t)&bla, "usdt3_arg3");
> + ASSERT_EQ(bss->usdt3_arg_sizes[0], 4, "usdt3_arg1_size");
> + ASSERT_EQ(bss->usdt3_arg_sizes[1], 8, "usdt3_arg2_size");
> + ASSERT_EQ(bss->usdt3_arg_sizes[2], 8, "usdt3_arg3_size");
>
> /* auto-attached usdt12 gets default zero cookie value */
> ASSERT_EQ(bss->usdt12_cookie, 0, "usdt12_cookie");
> @@ -104,6 +108,11 @@ static void subtest_basic_usdt(void)
> ASSERT_EQ(bss->usdt12_args[10], nums[idx], "usdt12_arg11");
> ASSERT_EQ(bss->usdt12_args[11], t1.y, "usdt12_arg12");
>
> + int usdt12_expected_arg_sizes[12] = { 4, 4, 8, 8, 4, 8, 8, 8, 4, 2, 2, 1 };
> +
> + for (i = 0; i < 12; i++)
> + ASSERT_EQ(bss->usdt12_arg_sizes[i], usdt12_expected_arg_sizes[i], "usdt12_arg_size");
> +
> /* trigger_func() is marked __always_inline, so USDT invocations will be
> * inlined in two different places, meaning that each USDT will have
> * at least 2 different places to be attached to. This verifies that
> diff --git a/tools/testing/selftests/bpf/progs/test_usdt.c b/tools/testing/selftests/bpf/progs/test_usdt.c
> index 505aab9a5234..096488f47fbc 100644
> --- a/tools/testing/selftests/bpf/progs/test_usdt.c
> +++ b/tools/testing/selftests/bpf/progs/test_usdt.c
> @@ -11,6 +11,7 @@ int usdt0_called;
> u64 usdt0_cookie;
> int usdt0_arg_cnt;
> int usdt0_arg_ret;
> +int usdt0_arg_size;
>
> SEC("usdt")
> int usdt0(struct pt_regs *ctx)
> @@ -26,6 +27,7 @@ int usdt0(struct pt_regs *ctx)
> usdt0_arg_cnt = bpf_usdt_arg_cnt(ctx);
> /* should return -ENOENT for any arg_num */
> usdt0_arg_ret = bpf_usdt_arg(ctx, bpf_get_prandom_u32(), &tmp);
> + usdt0_arg_size = bpf_usdt_arg_size(ctx, bpf_get_prandom_u32());
> return 0;
> }
>
> @@ -34,6 +36,7 @@ u64 usdt3_cookie;
> int usdt3_arg_cnt;
> int usdt3_arg_rets[3];
> u64 usdt3_args[3];
> +int usdt3_arg_sizes[3];
>
> SEC("usdt//proc/self/exe:test:usdt3")
> int usdt3(struct pt_regs *ctx)
> @@ -50,12 +53,15 @@ int usdt3(struct pt_regs *ctx)
>
> usdt3_arg_rets[0] = bpf_usdt_arg(ctx, 0, &tmp);
> usdt3_args[0] = (int)tmp;
> + usdt3_arg_sizes[0] = bpf_usdt_arg_size(ctx, 0);
>
> usdt3_arg_rets[1] = bpf_usdt_arg(ctx, 1, &tmp);
> usdt3_args[1] = (long)tmp;
> + usdt3_arg_sizes[1] = bpf_usdt_arg_size(ctx, 1);
>
> usdt3_arg_rets[2] = bpf_usdt_arg(ctx, 2, &tmp);
> usdt3_args[2] = (uintptr_t)tmp;
> + usdt3_arg_sizes[2] = bpf_usdt_arg_size(ctx, 2);
>
> return 0;
> }
> @@ -64,12 +70,15 @@ int usdt12_called;
> u64 usdt12_cookie;
> int usdt12_arg_cnt;
> u64 usdt12_args[12];
> +int usdt12_arg_sizes[12];
>
> SEC("usdt//proc/self/exe:test:usdt12")
> int BPF_USDT(usdt12, int a1, int a2, long a3, long a4, unsigned a5,
> long a6, __u64 a7, uintptr_t a8, int a9, short a10,
> short a11, signed char a12)
> {
> + int i;
> +
> if (my_pid != (bpf_get_current_pid_tgid() >> 32))
> return 0;
>
> @@ -90,6 +99,11 @@ int BPF_USDT(usdt12, int a1, int a2, long a3, long a4, unsigned a5,
> usdt12_args[9] = a10;
> usdt12_args[10] = a11;
> usdt12_args[11] = a12;
> +
> + bpf_for(i, 0, 12) {
> + usdt12_arg_sizes[i] = bpf_usdt_arg_size(ctx, i);
> + }
> +
> return 0;
> }
>
> --
> 2.48.1
>
>
next prev parent reply other threads:[~2025-02-26 14:36 UTC|newest]
Thread overview: 5+ messages / expand[flat|nested] mbox.gz Atom feed top
2025-02-24 23:57 [PATCH bpf-next v2 1/2] libbpf: implement bpf_usdt_arg_size BPF function Ihor Solodrai
2025-02-24 23:57 ` [PATCH bpf-next v2 2/2] selftests/bpf: test bpf_usdt_arg_size() function Ihor Solodrai
2025-02-26 14:36 ` Jiri Olsa [this message]
2025-02-26 14:34 ` [PATCH bpf-next v2 1/2] libbpf: implement bpf_usdt_arg_size BPF function Jiri Olsa
2025-02-26 18:20 ` patchwork-bot+netdevbpf
Reply instructions:
You may reply publicly to this message via plain-text email
using any one of the following methods:
* Save the following mbox file, import it into your mail client,
and reply-to-all from there: mbox
Avoid top-posting and favor interleaved quoting:
https://en.wikipedia.org/wiki/Posting_style#Interleaved_style
* Reply using the --to, --cc, and --in-reply-to
switches of git-send-email(1):
git send-email \
--in-reply-to=Z78m5XkzuC1L3QPE@krava \
--to=olsajiri@gmail.com \
--cc=andrii@kernel.org \
--cc=ast@kernel.org \
--cc=bpf@vger.kernel.org \
--cc=daniel@iogearbox.net \
--cc=eddyz87@gmail.com \
--cc=ihor.solodrai@linux.dev \
--cc=kernel-team@meta.com \
--cc=mykolal@fb.com \
/path/to/YOUR_REPLY
https://kernel.org/pub/software/scm/git/docs/git-send-email.html
* If your mail client supports setting the In-Reply-To header
via mailto: links, try the mailto: link
Be sure your reply has a Subject: header at the top and a blank line
before the message body.
This is a public inbox, see mirroring instructions
for how to clone and mirror all data and code used for this inbox