Commit a2ffda38 authored by Cong Wang's avatar Cong Wang Committed by Alexei Starovoitov
Browse files

selftests/bpf: Add test cases for redirection between udp and unix 



Add two test cases to ensure redirection between udp and unix
work bidirectionally.

Signed-off-by: default avatarCong Wang <cong.wang@bytedance.com>
Signed-off-by: default avatarAlexei Starovoitov <ast@kernel.org>
Link: https://lore.kernel.org/bpf/20210704190252.11866-12-xiyou.wangcong@gmail.com
parent 5ea905dd

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

+170 −0
Original line number Diff line number Diff line
@@ -1825,6 +1825,175 @@ static void test_udp_redir(struct test_sockmap_listen *skel, struct bpf_map *map 
	udp_skb_redir_to_connected(skel, map, family);

}



static void udp_unix_redir_to_connected(int family, int sock_mapfd,

					int verd_mapfd, enum redir_mode mode)

{

	const char *log_prefix = redir_mode_str(mode);

	int c0, c1, p0, p1;

	unsigned int pass;

	int retries = 100;

	int err, n;

	int sfd[2];

	u32 key;

	char b;



	zero_verdict_count(verd_mapfd);



	if (socketpair(AF_UNIX, SOCK_DGRAM | SOCK_NONBLOCK, 0, sfd))

		return;

	c0 = sfd[0], p0 = sfd[1];



	err = udp_socketpair(family, &p1, &c1);

	if (err)

		goto close;



	err = add_to_sockmap(sock_mapfd, p0, p1);

	if (err)

		goto close_cli1;



	n = write(c1, "a", 1);

	if (n < 0)

		FAIL_ERRNO("%s: write", log_prefix);

	if (n == 0)

		FAIL("%s: incomplete write", log_prefix);

	if (n < 1)

		goto close_cli1;



	key = SK_PASS;

	err = xbpf_map_lookup_elem(verd_mapfd, &key, &pass);

	if (err)

		goto close_cli1;

	if (pass != 1)

		FAIL("%s: want pass count 1, have %d", log_prefix, pass);



again:

	n = read(mode == REDIR_INGRESS ? p0 : c0, &b, 1);

	if (n < 0) {

		if (errno == EAGAIN && retries--)

			goto again;

		FAIL_ERRNO("%s: read", log_prefix);

	}

	if (n == 0)

		FAIL("%s: incomplete read", log_prefix);



close_cli1:

	xclose(c1);

	xclose(p1);

close:

	xclose(c0);

	xclose(p0);

}



static void udp_unix_skb_redir_to_connected(struct test_sockmap_listen *skel,

					    struct bpf_map *inner_map, int family)

{

	int verdict = bpf_program__fd(skel->progs.prog_skb_verdict);

	int verdict_map = bpf_map__fd(skel->maps.verdict_map);

	int sock_map = bpf_map__fd(inner_map);

	int err;



	err = xbpf_prog_attach(verdict, sock_map, BPF_SK_SKB_VERDICT, 0);

	if (err)

		return;



	skel->bss->test_ingress = false;

	udp_unix_redir_to_connected(family, sock_map, verdict_map, REDIR_EGRESS);

	skel->bss->test_ingress = true;

	udp_unix_redir_to_connected(family, sock_map, verdict_map, REDIR_INGRESS);



	xbpf_prog_detach2(verdict, sock_map, BPF_SK_SKB_VERDICT);

}



static void unix_udp_redir_to_connected(int family, int sock_mapfd,

					int verd_mapfd, enum redir_mode mode)

{

	const char *log_prefix = redir_mode_str(mode);

	int c0, c1, p0, p1;

	unsigned int pass;

	int err, n;

	int sfd[2];

	u32 key;

	char b;



	zero_verdict_count(verd_mapfd);



	err = udp_socketpair(family, &p0, &c0);

	if (err)

		return;



	if (socketpair(AF_UNIX, SOCK_DGRAM | SOCK_NONBLOCK, 0, sfd))

		goto close_cli0;

	c1 = sfd[0], p1 = sfd[1];



	err = add_to_sockmap(sock_mapfd, p0, p1);

	if (err)

		goto close;



	n = write(c1, "a", 1);

	if (n < 0)

		FAIL_ERRNO("%s: write", log_prefix);

	if (n == 0)

		FAIL("%s: incomplete write", log_prefix);

	if (n < 1)

		goto close;



	key = SK_PASS;

	err = xbpf_map_lookup_elem(verd_mapfd, &key, &pass);

	if (err)

		goto close;

	if (pass != 1)

		FAIL("%s: want pass count 1, have %d", log_prefix, pass);



	n = read(mode == REDIR_INGRESS ? p0 : c0, &b, 1);

	if (n < 0)

		FAIL_ERRNO("%s: read", log_prefix);

	if (n == 0)

		FAIL("%s: incomplete read", log_prefix);



close:

	xclose(c1);

	xclose(p1);

close_cli0:

	xclose(c0);

	xclose(p0);



}



static void unix_udp_skb_redir_to_connected(struct test_sockmap_listen *skel,

					    struct bpf_map *inner_map, int family)

{

	int verdict = bpf_program__fd(skel->progs.prog_skb_verdict);

	int verdict_map = bpf_map__fd(skel->maps.verdict_map);

	int sock_map = bpf_map__fd(inner_map);

	int err;



	err = xbpf_prog_attach(verdict, sock_map, BPF_SK_SKB_VERDICT, 0);

	if (err)

		return;



	skel->bss->test_ingress = false;

	unix_udp_redir_to_connected(family, sock_map, verdict_map, REDIR_EGRESS);

	skel->bss->test_ingress = true;

	unix_udp_redir_to_connected(family, sock_map, verdict_map, REDIR_INGRESS);



	xbpf_prog_detach2(verdict, sock_map, BPF_SK_SKB_VERDICT);

}



static void test_udp_unix_redir(struct test_sockmap_listen *skel, struct bpf_map *map,

				int family)

{

	const char *family_name, *map_name;

	char s[MAX_TEST_NAME];



	family_name = family_str(family);

	map_name = map_type_str(map);

	snprintf(s, sizeof(s), "%s %s %s", map_name, family_name, __func__);

	if (!test__start_subtest(s))

		return;

	udp_unix_skb_redir_to_connected(skel, map, family);

	unix_udp_skb_redir_to_connected(skel, map, family);

}



static void run_tests(struct test_sockmap_listen *skel, struct bpf_map *map,

		      int family)

{
@@ -1834,6 +2003,7 @@ static void run_tests(struct test_sockmap_listen *skel, struct bpf_map *map, 
	test_reuseport(skel, map, family, SOCK_STREAM);

	test_reuseport(skel, map, family, SOCK_DGRAM);

	test_udp_redir(skel, map, family);

	test_udp_unix_redir(skel, map, family);

}



void test_sockmap_listen(void)