From: Geliang Tang <[email protected]>

This patch uses start_server_str() helper in test_dctcp_fallback() in
bpf_tcp_ca.c, instead of using start_server() and settcpca(). For
support opts in start_server_str() helper, opts->cb_opts needs to be
passed to post_socket_cb() in __start_server().

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

diff --git a/tools/testing/selftests/bpf/network_helpers.c 
b/tools/testing/selftests/bpf/network_helpers.c
index 8502917b6c7b..e20caef06aae 100644
--- a/tools/testing/selftests/bpf/network_helpers.c
+++ b/tools/testing/selftests/bpf/network_helpers.c
@@ -94,7 +94,8 @@ static int __start_server(int type, const struct sockaddr 
*addr, socklen_t addrl
        if (settimeo(fd, opts->timeout_ms))
                goto error_close;
 
-       if (opts->post_socket_cb && opts->post_socket_cb(fd, NULL)) {
+       if (opts->post_socket_cb &&
+           opts->post_socket_cb(fd, opts->cb_opts)) {
                log_err("Failed to call post_socket_cb");
                goto error_close;
        }
diff --git a/tools/testing/selftests/bpf/prog_tests/bpf_tcp_ca.c 
b/tools/testing/selftests/bpf/prog_tests/bpf_tcp_ca.c
index 6f034059888a..7a523c7a0c6a 100644
--- a/tools/testing/selftests/bpf/prog_tests/bpf_tcp_ca.c
+++ b/tools/testing/selftests/bpf/prog_tests/bpf_tcp_ca.c
@@ -187,6 +187,9 @@ static void test_dctcp_fallback(void)
        };
        struct bpf_dctcp *dctcp_skel;
        struct bpf_link *link = NULL;
+       struct cb_opts dctcp = {
+               .cc = "bpf_dctcp",
+       };
        struct cb_opts cubic = {
                .cc = "cubic",
        };
@@ -204,9 +207,9 @@ static void test_dctcp_fallback(void)
        if (!ASSERT_OK_PTR(link, "dctcp link"))
                goto done;
 
-       lfd = start_server(AF_INET6, SOCK_STREAM, "::1", 0, 0);
-       if (!ASSERT_GE(lfd, 0, "lfd") ||
-           !ASSERT_OK(settcpca(lfd, "bpf_dctcp"), "lfd=>bpf_dctcp"))
+       opts.cb_opts = &dctcp;
+       lfd = start_server_str(AF_INET6, SOCK_STREAM, "::1", 0, &opts);
+       if (!ASSERT_GE(lfd, 0, "lfd"))
                goto done;
 
        opts.cb_opts = &cubic;
-- 
2.43.0


Reply via email to