All tests use more or less the same ping commands as final validation.
Also test_ping()'s return value is checked with ASSERT_OK() while this
check is already done by the SYS() macro inside test_ping().

Create helpers around test_ping() and use them in the tests to avoid code
duplication.
Remove the unnecessary ASSERT_OK() from the tests.

Signed-off-by: Bastien Curutchet (eBPF Foundation) 
<bastien.curutc...@bootlin.com>
---
 .../testing/selftests/bpf/prog_tests/test_tunnel.c | 53 ++++++++++------------
 1 file changed, 24 insertions(+), 29 deletions(-)

diff --git a/tools/testing/selftests/bpf/prog_tests/test_tunnel.c 
b/tools/testing/selftests/bpf/prog_tests/test_tunnel.c
index 
27a8c8caa87e4c6b39b2b26c2aa9860b131a70a9..498822064218d4676a5adf1cdb2a33ab56688746
 100644
--- a/tools/testing/selftests/bpf/prog_tests/test_tunnel.c
+++ b/tools/testing/selftests/bpf/prog_tests/test_tunnel.c
@@ -364,6 +364,25 @@ static int test_ping(int family, const char *addr)
        return -1;
 }
 
+static void ping_dev0(void)
+{
+       /* ping from root namespace test */
+       test_ping(AF_INET, IP4_ADDR_TUNL_DEV0);
+}
+
+static void ping_dev1(void)
+{
+       struct nstoken *nstoken;
+
+       /* ping from at_ns0 namespace test */
+       nstoken = open_netns("at_ns0");
+       if (!ASSERT_OK_PTR(nstoken, "setns"))
+               return;
+
+       test_ping(AF_INET, IP4_ADDR_TUNL_DEV1);
+       close_netns(nstoken);
+}
+
 static int attach_tc_prog(struct bpf_tc_hook *hook, int igr_fd, int egr_fd)
 {
        DECLARE_LIBBPF_OPTS(bpf_tc_opts, opts1, .handle = 1,
@@ -496,9 +515,7 @@ static void test_vxlan_tunnel(void)
                goto done;
 
        /* ping test */
-       err = test_ping(AF_INET, IP4_ADDR_TUNL_DEV0);
-       if (!ASSERT_OK(err, "test_ping"))
-               goto done;
+       ping_dev0();
 
 done:
        /* delete vxlan tunnel */
@@ -553,9 +570,7 @@ static void test_ip6vxlan_tunnel(void)
                goto done;
 
        /* ping test */
-       err = test_ping(AF_INET, IP4_ADDR_TUNL_DEV0);
-       if (!ASSERT_OK(err, "test_ping"))
-               goto done;
+       ping_dev0();
 
 done:
        /* delete ipv6 vxlan tunnel */
@@ -569,7 +584,6 @@ static void test_ip6vxlan_tunnel(void)
 static void test_ipip_tunnel(enum ipip_encap encap)
 {
        struct test_tunnel_kern *skel = NULL;
-       struct nstoken *nstoken;
        int set_src_prog_fd, get_src_prog_fd;
        int err;
 
@@ -606,19 +620,8 @@ static void test_ipip_tunnel(enum ipip_encap encap)
        if (generic_attach(IPIP_TUNL_DEV1, get_src_prog_fd, set_src_prog_fd))
                goto done;
 
-       /* ping from root namespace test */
-       err = test_ping(AF_INET, IP4_ADDR_TUNL_DEV0);
-       if (!ASSERT_OK(err, "test_ping"))
-               goto done;
-
-       /* ping from at_ns0 namespace test */
-       nstoken = open_netns("at_ns0");
-       if (!ASSERT_OK_PTR(nstoken, "setns"))
-               goto done;
-       err = test_ping(AF_INET, IP4_ADDR_TUNL_DEV1);
-       if (!ASSERT_OK(err, "test_ping"))
-               goto done;
-       close_netns(nstoken);
+       ping_dev0();
+       ping_dev1();
 
 done:
        /* delete ipip tunnel */
@@ -631,7 +634,6 @@ static void test_xfrm_tunnel(void)
 {
        LIBBPF_OPTS(bpf_xdp_attach_opts, opts);
        struct test_tunnel_kern *skel = NULL;
-       struct nstoken *nstoken;
        int xdp_prog_fd;
        int tc_prog_fd;
        int ifindex;
@@ -662,14 +664,7 @@ static void test_xfrm_tunnel(void)
        if (!ASSERT_OK(err, "bpf_xdp_attach"))
                goto done;
 
-       /* ping from at_ns0 namespace test */
-       nstoken = open_netns("at_ns0");
-       if (!ASSERT_OK_PTR(nstoken, "setns"))
-               goto done;
-       err = test_ping(AF_INET, IP4_ADDR_TUNL_DEV1);
-       close_netns(nstoken);
-       if (!ASSERT_OK(err, "test_ping"))
-               goto done;
+       ping_dev1();
 
        if (!ASSERT_EQ(skel->bss->xfrm_reqid, 1, "req_id"))
                goto done;

-- 
2.48.1


Reply via email to