bpf/prog_tests/lsm_cgroup.c

19 static __u32 query_prog_cnt(int cgroup_fd, const char *attach_func)
25 	ASSERT_OK(bpf_prog_query_opts(cgroup_fd, BPF_LSM_CGROUP, &p), "prog_query");
41 	ASSERT_OK(bpf_prog_query_opts(cgroup_fd, BPF_LSM_CGROUP, &p), "prog_query");
69 	int cgroup_fd = -1, cgroup_fd2 = -1, cgroup_fd3 = -1;
91 	cgroup_fd = test__join_cgroup("/sock_policy");
92 	if (!ASSERT_GE(cgroup_fd, 0, "join_cgroup"))
106 	ASSERT_EQ(query_prog_cnt(cgroup_fd, "bpf_lsm_sk_alloc_security"), 0, "prog count");
107 	ASSERT_EQ(query_prog_cnt(cgroup_fd, NULL), 0, "total prog count");
108 	err = bpf_prog_attach(alloc_prog_fd, cgroup_fd, BPF_LSM_CGROUP, 0);
115 	ASSERT_EQ(query_prog_cnt(cgroup_fd, "bpf_lsm_sk_alloc_security"), 1, "prog count");
116 	ASSERT_EQ(query_prog_cnt(cgroup_fd, NULL), 1, "total prog count");
118 	ASSERT_EQ(query_prog_cnt(cgroup_fd, "bpf_lsm_inet_csk_clone"), 0, "prog count");
119 	err = bpf_prog_attach(clone_prog_fd, cgroup_fd, BPF_LSM_CGROUP, 0);
122 	ASSERT_EQ(query_prog_cnt(cgroup_fd, "bpf_lsm_inet_csk_clone"), 1, "prog count");
123 	ASSERT_EQ(query_prog_cnt(cgroup_fd, NULL), 2, "total prog count");
127 	ASSERT_EQ(query_prog_cnt(cgroup_fd, "bpf_lsm_socket_post_create"), 0, "prog count");
128 	err = bpf_prog_attach(post_create_prog_fd, cgroup_fd,
132 	ASSERT_EQ(query_prog_cnt(cgroup_fd, "bpf_lsm_socket_post_create"), 1, "prog count");
133 	ASSERT_EQ(query_prog_cnt(cgroup_fd, NULL), 3, "total prog count");
136 	err = bpf_prog_attach_opts(post_create_prog_fd2, cgroup_fd,
140 	ASSERT_EQ(query_prog_cnt(cgroup_fd, "bpf_lsm_socket_post_create"), 1, "prog count");
141 	ASSERT_EQ(query_prog_cnt(cgroup_fd, NULL), 3, "total prog count");
145 	ASSERT_EQ(query_prog_cnt(cgroup_fd, "bpf_lsm_socket_bind"), 0, "prog count");
146 	bind_link_fd = bpf_link_create(bind_prog_fd, cgroup_fd,
150 	ASSERT_EQ(query_prog_cnt(cgroup_fd, "bpf_lsm_socket_bind"), 1, "prog count");
151 	ASSERT_EQ(query_prog_cnt(cgroup_fd, NULL), 4, "total prog count");
159 	ASSERT_EQ(query_prog_cnt(cgroup_fd, "bpf_lsm_socket_bind"), 1, "prog count");
160 	ASSERT_EQ(query_prog_cnt(cgroup_fd, NULL), 4, "total prog count");
167 	ASSERT_EQ(query_prog_cnt(cgroup_fd, "bpf_lsm_socket_bind"), 1, "prog count");
174 	ASSERT_EQ(query_prog_cnt(cgroup_fd, NULL), 4, "total prog count");
291 	ASSERT_GE(bpf_prog_detach2(post_create_prog_fd2, cgroup_fd,
295 	ASSERT_GE(bpf_prog_detach2(alloc_prog_fd, cgroup_fd,
297 	ASSERT_GE(bpf_prog_detach2(clone_prog_fd, cgroup_fd,
301 	close(cgroup_fd);