selftests/bpf: Use connect_fd_to_fd in sk_lookup

This patch uses public helper connect_fd_to_fd() exported in
network_helpers.h instead of using getsockname() + connect() in
run_lookup_prog() in prog_tests/sk_lookup.c. This can simplify
the code.

Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn>
Link: https://lore.kernel.org/r/7077c277cde5a1864cdc244727162fb75c8bb9c5.1720515893.git.tanggeliang@kylinos.cn
Signed-off-by: Martin KaFai Lau <martin.lau@kernel.org>

diff --git a/tools/testing/selftests/bpf/prog_tests/sk_lookup.c b/tools/testing/selftests/bpf/prog_tests/sk_lookup.c
index 386e482be6170934f75d4de6b2ac1952592eb7aa..ae87c00867ba42e91b923a0a06ef989895ec0c1e 100644 (file)
--- a/tools/testing/selftests/bpf/prog_tests/sk_lookup.c
+++ b/tools/testing/selftests/bpf/prog_tests/sk_lookup.c
@@ -633,9 +633,6 @@ static void run_lookup_prog(const struct test *t)
         * BPF socket lookup.
         */
        if (t->reuseport_has_conns) {
-               struct sockaddr_storage addr = {};
-               socklen_t len = sizeof(addr);
-
                /* Add an extra socket to reuseport group */
                reuse_conn_fd = make_server(t->sotype, t->listen_at.ip,
                                            t->listen_at.port,
@@ -643,12 +640,9 @@ static void run_lookup_prog(const struct test *t)
                if (reuse_conn_fd < 0)
                        goto close;
 
-               /* Connect the extra socket to itself */
-               err = getsockname(reuse_conn_fd, (void *)&addr, &len);
-               if (CHECK(err, "getsockname", "errno %d\n", errno))
-                       goto close;
-               err = connect(reuse_conn_fd, (void *)&addr, len);
-               if (CHECK(err, "connect", "errno %d\n", errno))
+                /* Connect the extra socket to itself */
+               err = connect_fd_to_fd(reuse_conn_fd, reuse_conn_fd, 0);
+               if (!ASSERT_OK(err, "connect_fd_to_fd"))
                        goto close;
        }