Commit fd60aa0a authored by Tao Chen's avatar Tao Chen Committed by Alexei Starovoitov
Browse files

bpf/selftests: Add selftests for token info 



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: default avatarAndrii Nakryiko <andrii@kernel.org>
Signed-off-by: default avatarTao Chen <chen.dylane@linux.dev>
Link: https://lore.kernel.org/r/20250716134654.1162635-2-chen.dylane@linux.dev


Signed-off-by: default avatarAlexei Starovoitov <ast@kernel.org>
parent 19d18fdf

tools/testing/selftests/bpf/prog_tests/token.c

+44 −0
Original line number Diff line number Diff line
@@ -1047,6 +1047,41 @@ static int userns_obj_priv_implicit_token_envvar(int mnt_fd, struct token_lsm *l 


#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);

	}

}