bpf/prog_tests/cgroup_tcp_skb.c

10 #include "cgroup_tcp_skb.skel.h"
21 			   struct cgroup_tcp_skb *skel)
24 	skel->bss->g_sock_state = 0;
25 	skel->bss->g_unexpected = 0;
64 			  struct cgroup_tcp_skb *skel)
86 	skel->bss->g_sock_port = ntohs(port);
110 			   struct cgroup_tcp_skb *skel)
136 	skel->bss->g_sock_port = ntohs(port);
156 			    struct cgroup_tcp_skb *skel)
163 	saved_packet_count = skel->bss->g_packet_count;
166 	     skel->bss->g_packet_count != saved_packet_count && i < 10;
168 		saved_packet_count = skel->bss->g_packet_count;
171 	if (!ASSERT_EQ(skel->bss->g_packet_count, saved_packet_count,
175 	skel->bss->g_packet_count = 0;
187 	     skel->bss->g_packet_count < saved_packet_count + 2 && i < 10;
190 	if (!ASSERT_GE(skel->bss->g_packet_count, saved_packet_count + 2,
194 	saved_packet_count = skel->bss->g_packet_count;
204 	     skel->bss->g_packet_count < saved_packet_count + 2 && i < 10;
207 	if (!ASSERT_GE(skel->bss->g_packet_count, saved_packet_count + 2,
240 	struct cgroup_tcp_skb *skel;
245 	skel = cgroup_tcp_skb__open_and_load();
246 	if (!ASSERT_OK(!skel, "skel_open_load"))
259 			      skel->progs.server_egress,
260 			      skel->progs.server_ingress,
261 			      skel);
265 	err = talk_to_cgroup(&client_fd, &listen_fd, &service_fd, skel);
269 	err = close_connection(&client_fd, &service_fd, &listen_fd, skel);
273 	ASSERT_EQ(skel->bss->g_unexpected, 0, "g_unexpected");
274 	ASSERT_EQ(skel->bss->g_sock_state, CLOSED, "g_sock_state");
280 			      skel->progs.server_egress_srv,
281 			      skel->progs.server_ingress_srv,
282 			      skel);
284 	err = talk_to_cgroup(&client_fd, &listen_fd, &service_fd, skel);
288 	err = close_connection(&service_fd, &client_fd, &listen_fd, skel);
292 	ASSERT_EQ(skel->bss->g_unexpected, 0, "g_unexpected");
293 	ASSERT_EQ(skel->bss->g_sock_state, TIME_WAIT, "g_sock_state");
299 			      skel->progs.client_egress_srv,
300 			      skel->progs.client_ingress_srv,
301 			      skel);
303 	err = talk_to_outside(&client_fd, &listen_fd, &service_fd, skel);
307 	err = close_connection(&service_fd, &client_fd, &listen_fd, skel);
311 	ASSERT_EQ(skel->bss->g_unexpected, 0, "g_unexpected");
312 	ASSERT_EQ(skel->bss->g_sock_state, CLOSED, "g_sock_state");
318 			      skel->progs.client_egress,
319 			      skel->progs.client_ingress,
320 			      skel);
322 	err = talk_to_outside(&client_fd, &listen_fd, &service_fd, skel);
326 	err = close_connection(&client_fd, &service_fd, &listen_fd, skel);
330 	ASSERT_EQ(skel->bss->g_unexpected, 0, "g_unexpected");
331 	ASSERT_EQ(skel->bss->g_sock_state, TIME_WAIT, "g_sock_state");
343 	cgroup_tcp_skb__destroy(skel);