在 2025/2/10 21:47, Jiri Olsa 写道:
On Mon, Feb 10, 2025 at 01:59:45PM +0800, Tao Chen wrote:Add selftests for prog_kfunc feature probing. ./test_progs -t libbpf_probe_kfuncs #153 libbpf_probe_kfuncs:OK Summary: 1/0 PASSED, 0 SKIPPED, 0 FAILED Signed-off-by: Tao Chen <chen.dylane@xxxxxxxxx> --- .../selftests/bpf/prog_tests/libbpf_probes.c | 118 ++++++++++++++++++ 1 file changed, 118 insertions(+) diff --git a/tools/testing/selftests/bpf/prog_tests/libbpf_probes.c b/tools/testing/selftests/bpf/prog_tests/libbpf_probes.c index 4ed46ed58a7b..fc4c9dc2cbed 100644 --- a/tools/testing/selftests/bpf/prog_tests/libbpf_probes.c +++ b/tools/testing/selftests/bpf/prog_tests/libbpf_probes.c @@ -126,3 +126,121 @@ void test_libbpf_probe_helpers(void) ASSERT_EQ(res, d->supported, buf); } } + +static int module_btf_fd(char *module) +{ + int fd, err; + __u32 id = 0, len; + struct bpf_btf_info info; + char name[64]; + + while (true) { + err = bpf_btf_get_next_id(id, &id); + if (err && (errno == ENOENT || errno == EPERM)) + return 0; + if (err) { + err = -errno; + return err; + } + fd = bpf_btf_get_fd_by_id(id); + if (fd < 0) { + if (errno == ENOENT) + continue; + err = -errno; + return err;nit, return -errno ? jirka
Ack.
+ } + len = sizeof(info); + memset(&info, 0, sizeof(info)); + info.name = ptr_to_u64(name); + info.name_len = sizeof(name); + err = bpf_btf_get_info_by_fd(fd, &info, &len); + if (err) { + err = -errno; + goto err_out; + } + /* find target module btf */ + if (!strcmp(name, module)) + break; + + close(fd); + } + + return fd; +err_out: + close(fd); + return err; +}SNIP
-- Best Regards Dylane Chen
