From: Geliang Tang <[email protected]>

This patch adds another new parameter "must_fail" for __connect_to_fd_opts,
then opts->must_fail can be replaced by "must_fail" parameter in it. In
run_test() of cgroup_v1v2, true is passed to it, while false is passed in
other places. And the strcut member "must_fail" of network_helper_opts can
be dropped now.

Signed-off-by: Geliang Tang <[email protected]>
---
 tools/testing/selftests/bpf/network_helpers.c            | 8 ++++----
 tools/testing/selftests/bpf/network_helpers.h            | 3 +--
 tools/testing/selftests/bpf/prog_tests/cgroup_v1v2.c     | 5 +----
 tools/testing/selftests/bpf/prog_tests/ip_check_defrag.c | 4 ++--
 4 files changed, 8 insertions(+), 12 deletions(-)

diff --git a/tools/testing/selftests/bpf/network_helpers.c 
b/tools/testing/selftests/bpf/network_helpers.c
index 0f53638ae5a0..1c4c783719b7 100644
--- a/tools/testing/selftests/bpf/network_helpers.c
+++ b/tools/testing/selftests/bpf/network_helpers.c
@@ -293,7 +293,7 @@ int connect_to_addr(int type, const struct sockaddr_storage 
*addr, socklen_t add
        if (settimeo(fd, opts->timeout_ms))
                goto error_close;
 
-       if (connect_fd_to_addr(fd, addr, addrlen, opts->must_fail))
+       if (connect_fd_to_addr(fd, addr, addrlen, false))
                goto error_close;
 
        return fd;
@@ -303,7 +303,7 @@ int connect_to_addr(int type, const struct sockaddr_storage 
*addr, socklen_t add
        return -1;
 }
 
-int __connect_to_fd_opts(int server_fd, int type, bool noconnect,
+int __connect_to_fd_opts(int server_fd, int type, bool noconnect, bool 
must_fail,
                         const struct network_helper_opts *opts)
 {
        struct sockaddr_storage addr;
@@ -353,7 +353,7 @@ int __connect_to_fd_opts(int server_fd, int type, bool 
noconnect,
                goto error_close;
 
        if (!noconnect)
-               if (connect_fd_to_addr(fd, &addr, addrlen, opts->must_fail))
+               if (connect_fd_to_addr(fd, &addr, addrlen, must_fail))
                        goto error_close;
 
        return fd;
@@ -365,7 +365,7 @@ int __connect_to_fd_opts(int server_fd, int type, bool 
noconnect,
 
 int connect_to_fd_opts(int server_fd, const struct network_helper_opts *opts)
 {
-       return __connect_to_fd_opts(server_fd, 0, false, opts);
+       return __connect_to_fd_opts(server_fd, 0, false, false, opts);
 }
 
 int connect_to_fd(int server_fd, int timeout_ms)
diff --git a/tools/testing/selftests/bpf/network_helpers.h 
b/tools/testing/selftests/bpf/network_helpers.h
index e029d4ff983e..fefe2ed0c798 100644
--- a/tools/testing/selftests/bpf/network_helpers.h
+++ b/tools/testing/selftests/bpf/network_helpers.h
@@ -23,7 +23,6 @@ typedef __u16 __sum16;
 
 struct network_helper_opts {
        int timeout_ms;
-       bool must_fail;
        int proto;
        int (*post_socket_cb)(int fd, void *opts);
        void *cb_opts;
@@ -59,7 +58,7 @@ void free_fds(int *fds, unsigned int nr_close_fds);
 int connect_to_addr(int type, const struct sockaddr_storage *addr, socklen_t 
len,
                    const struct network_helper_opts *opts);
 int connect_to_fd(int server_fd, int timeout_ms);
-int __connect_to_fd_opts(int server_fd, int type, bool noconnect,
+int __connect_to_fd_opts(int server_fd, int type, bool noconnect, bool 
must_fail,
                         const struct network_helper_opts *opts);
 int connect_to_fd_opts(int server_fd, const struct network_helper_opts *opts);
 int connect_fd_to_fd(int client_fd, int server_fd, int timeout_ms);
diff --git a/tools/testing/selftests/bpf/prog_tests/cgroup_v1v2.c 
b/tools/testing/selftests/bpf/prog_tests/cgroup_v1v2.c
index addf720428f7..124132b435ba 100644
--- a/tools/testing/selftests/bpf/prog_tests/cgroup_v1v2.c
+++ b/tools/testing/selftests/bpf/prog_tests/cgroup_v1v2.c
@@ -9,9 +9,6 @@
 
 static int run_test(int cgroup_fd, int server_fd, bool classid)
 {
-       struct network_helper_opts opts = {
-               .must_fail = true,
-       };
        struct connect4_dropper *skel;
        int fd, err = 0;
 
@@ -32,7 +29,7 @@ static int run_test(int cgroup_fd, int server_fd, bool 
classid)
                goto out;
        }
 
-       fd = connect_to_fd_opts(server_fd, &opts);
+       fd = __connect_to_fd_opts(server_fd, 0, false, true, NULL);
        if (fd < 0)
                err = -1;
        else
diff --git a/tools/testing/selftests/bpf/prog_tests/ip_check_defrag.c 
b/tools/testing/selftests/bpf/prog_tests/ip_check_defrag.c
index 330dfba95c56..948e1ca287a2 100644
--- a/tools/testing/selftests/bpf/prog_tests/ip_check_defrag.c
+++ b/tools/testing/selftests/bpf/prog_tests/ip_check_defrag.c
@@ -198,7 +198,7 @@ void test_bpf_ip_check_defrag_ok(bool ipv6)
        nstoken = open_netns(NS0);
        if (!ASSERT_OK_PTR(nstoken, "setns ns0"))
                goto out;
-       client_tx_fd = __connect_to_fd_opts(srv_fd, SOCK_RAW, true, &tx_ops);
+       client_tx_fd = __connect_to_fd_opts(srv_fd, SOCK_RAW, true, false, 
&tx_ops);
        close_netns(nstoken);
        if (!ASSERT_GE(client_tx_fd, 0, "connect_to_fd_opts"))
                goto out;
@@ -207,7 +207,7 @@ void test_bpf_ip_check_defrag_ok(bool ipv6)
        nstoken = open_netns(NS0);
        if (!ASSERT_OK_PTR(nstoken, "setns ns0"))
                goto out;
-       client_rx_fd = __connect_to_fd_opts(srv_fd, 0, true, &rx_opts);
+       client_rx_fd = __connect_to_fd_opts(srv_fd, 0, true, false, &rx_opts);
        close_netns(nstoken);
        if (!ASSERT_GE(client_rx_fd, 0, "connect_to_fd_opts"))
                goto out;
-- 
2.43.0


Reply via email to