bybrooklyn/linux
1
0
Fork 0
mirror of https://github.com/torvalds/linux.git synced 2026-04-18 06:44:00 -04:00
Code Issues Packages Projects Releases Wiki Activity

bpf/selftests: Add selftests for token info

Browse Source 
A previous change added bpf_token_info to get token info with
bpf_get_obj_info_by_fd, this patch adds a new test for token info.

 #461/12  token/bpf_token_info:OK

Acked-by: Andrii Nakryiko <andrii@kernel.org>
Signed-off-by: Tao Chen <chen.dylane@linux.dev>
Link: https://lore.kernel.org/r/20250716134654.1162635-2-chen.dylane@linux.dev
Signed-off-by: Alexei Starovoitov <ast@kernel.org>
This commit is contained in:
Tao Chen
2025-07-16 21:46:54 +08:00
committed by Alexei Starovoitov
parent 19d18fdfc7
commit fd60aa0a45
1 changed files with 44 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
tools/testing/selftests/bpf/prog_tests/token.c
View File

@@ -1047,6 +1047,41 @@ err_out:
#define bit(n) (1ULL << (n))
static int userns_bpf_token_info(int mnt_fd, struct token_lsm *lsm_skel)
{
int err, token_fd = -1;
struct bpf_token_info info;
u32 len = sizeof(struct bpf_token_info);
/* create BPF token from BPF FS mount */
token_fd = bpf_token_create(mnt_fd, NULL);
if (!ASSERT_GT(token_fd, 0, "token_create")) {
err = -EINVAL;
goto cleanup;
}
memset(&info, 0, len);
err = bpf_obj_get_info_by_fd(token_fd, &info, &len);
if (!ASSERT_ERR(err, "bpf_obj_get_token_info"))
goto cleanup;
if (!ASSERT_EQ(info.allowed_cmds, bit(BPF_MAP_CREATE), "token_info_cmds_map_create")) {
err = -EINVAL;
goto cleanup;
}
if (!ASSERT_EQ(info.allowed_progs, bit(BPF_PROG_TYPE_XDP), "token_info_progs_xdp")) {
err = -EINVAL;
goto cleanup;
}
/* The BPF_PROG_TYPE_EXT is not set in token */
if (ASSERT_EQ(info.allowed_progs, bit(BPF_PROG_TYPE_EXT), "token_info_progs_ext"))
err = -EINVAL;
cleanup:
zclose(token_fd);
return err;
}
void test_token(void)
{
if (test__start_subtest("map_token")) {
@@ -1150,4 +1185,13 @@ void test_token(void)
subtest_userns(&opts, userns_obj_priv_implicit_token_envvar);
}
if (test__start_subtest("bpf_token_info")) {
struct bpffs_opts opts = {
.cmds = bit(BPF_MAP_CREATE),
.progs = bit(BPF_PROG_TYPE_XDP),
.attachs = ~0ULL,
};
subtest_userns(&opts, userns_bpf_token_info);
}
}