github-mirror/linux
1
0
Fork 0
mirror of https://github.com/torvalds/linux.git synced 2026-07-27 17:47:41 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity Graph

selftests/bpf: Add tracing multi skel/pattern/ids module attach tests

Browse Source 
Adding tests for tracing_multi link attachment via all possible
libbpf apis - skeleton, function pattern and btf ids on top of
bpf_testmod kernel module.

Signed-off-by: Jiri Olsa <jolsa@kernel.org>
Link: https://lore.kernel.org/r/20260606123955.345967-23-jolsa@kernel.org
Signed-off-by: Alexei Starovoitov <ast@kernel.org>
This commit is contained in:
Jiri Olsa 2026-06-06 14:39:47 +02:00 committed by Alexei Starovoitov
parent 2922dd5841
commit 2863f074f1
4 changed files with 183 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools/testing/selftests/bpf/Makefile
View File

@ -512,7 +512,8 @@ SKEL_BLACKLIST := btf__% test_pinning_invalid.c test_sk_assign.c
LINKED_SKELS := test_static_linked.skel.h linked_funcs.skel.h \
linked_vars.skel.h linked_maps.skel.h \
test_subskeleton.skel.h test_subskeleton_lib.skel.h \
test_usdt.skel.h tracing_multi.skel.h
test_usdt.skel.h tracing_multi.skel.h \
tracing_multi_module.skel.h
LSKELS := fexit_sleep.c trace_printk.c trace_vprintk.c map_ptr_kern.c \
core_kern.c core_kern_overflow.c test_ringbuf.c \
@ -539,6 +540,7 @@ xsk_xdp_progs.skel.h-deps := xsk_xdp_progs.bpf.o
xdp_hw_metadata.skel.h-deps := xdp_hw_metadata.bpf.o
xdp_features.skel.h-deps := xdp_features.bpf.o
tracing_multi.skel.h-deps := tracing_multi_attach.bpf.o tracing_multi_check.bpf.o
tracing_multi_module.skel.h-deps := tracing_multi_attach_module.bpf.o tracing_multi_check.bpf.o
LINKED_BPF_OBJS := $(foreach skel,$(LINKED_SKELS),$($(skel)-deps))
LINKED_BPF_SRCS := $(patsubst %.bpf.o,%.c,$(LINKED_BPF_OBJS))

105
tools/testing/selftests/bpf/prog_tests/tracing_multi.c
View File

@ -5,6 +5,7 @@
#include <search.h>
#include "bpf/libbpf_internal.h"
#include "tracing_multi.skel.h"
#include "tracing_multi_module.skel.h"
#include "trace_helpers.h"
static const char * const bpf_fentry_test[] = {
@ -20,6 +21,14 @@ static const char * const bpf_fentry_test[] = {
"bpf_fentry_test10",
};
static const char * const bpf_testmod_fentry_test[] = {
"bpf_testmod_fentry_test1",
"bpf_testmod_fentry_test2",
"bpf_testmod_fentry_test3",
"bpf_testmod_fentry_test7",
"bpf_testmod_fentry_test11",
};
#define FUNCS_CNT (ARRAY_SIZE(bpf_fentry_test))
static int compare(const void *ppa, const void *ppb)
@ -242,6 +251,96 @@ static void test_link_api_ids(void)
free(ids);
}
static void test_module_skel_api(void)
{
struct tracing_multi_module *skel = NULL;
int err;
skel = tracing_multi_module__open_and_load();
if (!ASSERT_OK_PTR(skel, "tracing_multi__open_and_load"))
return;
skel->bss->pid = getpid();
err = tracing_multi_module__attach(skel);
if (!ASSERT_OK(err, "tracing_multi__attach"))
goto cleanup;
ASSERT_OK(trigger_module_test_read(1), "trigger_read");
ASSERT_EQ(skel->bss->test_result_fentry, 5, "test_result_fentry");
ASSERT_EQ(skel->bss->test_result_fexit, 5, "test_result_fexit");
cleanup:
tracing_multi_module__destroy(skel);
}
static void test_module_link_api_pattern(void)
{
struct tracing_multi_module *skel = NULL;
skel = tracing_multi_module__open_and_load();
if (!ASSERT_OK_PTR(skel, "tracing_multi_module__open_and_load"))
return;
skel->bss->pid = getpid();
skel->links.test_fentry = bpf_program__attach_tracing_multi(skel->progs.test_fentry,
"bpf_testmod:bpf_testmod_fentry_test*", NULL);
if (!ASSERT_OK_PTR(skel->links.test_fentry, "bpf_program__attach_tracing_multi"))
goto cleanup;
skel->links.test_fexit = bpf_program__attach_tracing_multi(skel->progs.test_fexit,
"bpf_testmod:bpf_testmod_fentry_test*", NULL);
if (!ASSERT_OK_PTR(skel->links.test_fexit, "bpf_program__attach_tracing_multi"))
goto cleanup;
ASSERT_OK(trigger_module_test_read(1), "trigger_read");
ASSERT_EQ(skel->bss->test_result_fentry, 5, "test_result_fentry");
ASSERT_EQ(skel->bss->test_result_fexit, 5, "test_result_fexit");
cleanup:
tracing_multi_module__destroy(skel);
}
static void test_module_link_api_ids(void)
{
size_t cnt = ARRAY_SIZE(bpf_testmod_fentry_test);
LIBBPF_OPTS(bpf_tracing_multi_opts, opts);
struct tracing_multi_module *skel = NULL;
__u32 *ids;
skel = tracing_multi_module__open_and_load();
if (!ASSERT_OK_PTR(skel, "tracing_multi_module__open_and_load"))
return;
skel->bss->pid = getpid();
ids = get_ids(bpf_testmod_fentry_test, cnt, "bpf_testmod");
if (!ASSERT_OK_PTR(ids, "get_ids"))
goto cleanup;
opts.ids = ids;
opts.cnt = cnt;
skel->links.test_fentry = bpf_program__attach_tracing_multi(skel->progs.test_fentry,
NULL, &opts);
if (!ASSERT_OK_PTR(skel->links.test_fentry, "bpf_program__attach_tracing_multi"))
goto cleanup;
skel->links.test_fexit = bpf_program__attach_tracing_multi(skel->progs.test_fexit,
NULL, &opts);
if (!ASSERT_OK_PTR(skel->links.test_fexit, "bpf_program__attach_tracing_multi"))
goto cleanup;
ASSERT_OK(trigger_module_test_read(1), "trigger_read");
ASSERT_EQ(skel->bss->test_result_fentry, 5, "test_result_fentry");
ASSERT_EQ(skel->bss->test_result_fexit, 5, "test_result_fexit");
cleanup:
tracing_multi_module__destroy(skel);
free(ids);
}
void test_tracing_multi_test(void)
{
#ifndef __x86_64__
@ -255,4 +354,10 @@ void test_tracing_multi_test(void)
test_link_api_pattern();
if (test__start_subtest("link_api_ids"))
test_link_api_ids();
if (test__start_subtest("module_skel_api"))
test_module_skel_api();
if (test__start_subtest("module_link_api_pattern"))
test_module_link_api_pattern();
if (test__start_subtest("module_link_api_ids"))
test_module_link_api_ids();
}

25
tools/testing/selftests/bpf/progs/tracing_multi_attach_module.c Normal file
View File

@ -0,0 +1,25 @@
// SPDX-License-Identifier: GPL-2.0
#include <vmlinux.h>
#include <bpf/bpf_helpers.h>
#include <bpf/bpf_tracing.h>
char _license[] SEC("license") = "GPL";
__hidden extern int tracing_multi_arg_check(__u64 *ctx, __u64 *test_result, bool is_return);
__u64 test_result_fentry = 0;
__u64 test_result_fexit = 0;
SEC("fentry.multi/bpf_testmod:bpf_testmod_fentry_test*")
int BPF_PROG(test_fentry)
{
tracing_multi_arg_check(ctx, &test_result_fentry, false);
return 0;
}
SEC("fexit.multi/bpf_testmod:bpf_testmod_fentry_test*")
int BPF_PROG(test_fexit)
{
tracing_multi_arg_check(ctx, &test_result_fexit, true);
return 0;
}

50
tools/testing/selftests/bpf/progs/tracing_multi_check.c
View File

@ -18,6 +18,12 @@ extern const void bpf_fentry_test8 __ksym;
extern const void bpf_fentry_test9 __ksym;
extern const void bpf_fentry_test10 __ksym;
extern const void bpf_testmod_fentry_test1 __ksym;
extern const void bpf_testmod_fentry_test2 __ksym;
extern const void bpf_testmod_fentry_test3 __ksym;
extern const void bpf_testmod_fentry_test7 __ksym;
extern const void bpf_testmod_fentry_test11 __ksym;
int tracing_multi_arg_check(__u64 *ctx, __u64 *test_result, bool is_return)
{
void *ip = (void *) bpf_get_func_ip(ctx);
@ -145,6 +151,50 @@ int tracing_multi_arg_check(__u64 *ctx, __u64 *test_result, bool is_return)
err |= is_return ? ret != 0 : 0;
*test_result += err == 0 ? 1 : 0;
} else if (ip == &bpf_testmod_fentry_test1) {
int a;
err |= bpf_get_func_arg(ctx, 0, &value);
a = (int) value;
err |= is_return ? ret != 2 : 0;
*test_result += err == 0 && a == 1;
} else if (ip == &bpf_testmod_fentry_test2) {
int a;
__u64 b;
err |= bpf_get_func_arg(ctx, 0, &value);
a = (int) value;
err |= bpf_get_func_arg(ctx, 1, &value);
b = (__u64) value;
err |= is_return ? ret != 5 : 0;
*test_result += err == 0 && a == 2 && b == 3;
} else if (ip == &bpf_testmod_fentry_test3) {
char a;
int b;
__u64 c;
err |= bpf_get_func_arg(ctx, 0, &value);
a = (char) value;
err |= bpf_get_func_arg(ctx, 1, &value);
b = (int) value;
err |= bpf_get_func_arg(ctx, 2, &value);
c = (__u64) value;
err |= is_return ? ret != 15 : 0;
*test_result += err == 0 && a == 4 && b == 5 && c == 6;
} else if (ip == &bpf_testmod_fentry_test7) {
err |= is_return ? ret != 133 : 0;
*test_result += err == 0;
} else if (ip == &bpf_testmod_fentry_test11) {
err |= is_return ? ret != 231 : 0;
*test_result += err == 0;
}
return 0;