From dccfc0dbd4c9b6044e8c5b76d8e0c84f7b774089 Mon Sep 17 00:00:00 2001 From: Geliang Tang Date: Fri, 12 Apr 2024 10:58:12 +0800 Subject: [PATCH] selftests/bpf: Add select for send_recv_data Some tests, such as the MPTCP bpf tests, require send_recv_data helper to run in nonblock mode. This patch adds nonblock support for send_recv_data(). Check if it is currently in nonblock mode, and if so, ignore EWOULDBLOCK to continue sending and receiving. Signed-off-by: Geliang Tang --- tools/testing/selftests/bpf/network_helpers.c | 33 +++++++++++++++---- 1 file changed, 27 insertions(+), 6 deletions(-) diff --git a/tools/testing/selftests/bpf/network_helpers.c b/tools/testing/selftests/bpf/network_helpers.c index eb25ff0fe34f..a1a36ff77d05 100644 --- a/tools/testing/selftests/bpf/network_helpers.c +++ b/tools/testing/selftests/bpf/network_helpers.c @@ -609,10 +609,12 @@ struct send_recv_arg { static void *send_recv_server(void *arg) { struct send_recv_arg *a = (struct send_recv_arg *)arg; - int flags = fcntl(a->fd, F_GETFL); + struct timeval timeout = { .tv_sec = 3 }; ssize_t nr_sent = 0, bytes = 0; char batch[1500]; int err = 0, fd; + fd_set wset; + int max_fd; fd = accept(a->fd, NULL, NULL); while (fd == -1) { @@ -628,13 +630,22 @@ static void *send_recv_server(void *arg) } while (bytes < a->bytes && !READ_ONCE(a->stop)) { + /* FD sets */ + FD_ZERO(&wset); + FD_SET(fd, &wset); + max_fd = fd; + + if (select(max_fd + 1, NULL, &wset, NULL, &timeout) == -1) { + log_err("Failed to select"); + err = -1; + break; + } + nr_sent = send(fd, &batch, MIN(a->bytes - bytes, sizeof(batch)), 0); if (nr_sent == -1 && errno == EINTR) continue; if (nr_sent == -1) { - if (flags & O_NONBLOCK && errno == EWOULDBLOCK) - continue; err = -errno; break; } @@ -659,7 +670,7 @@ static void *send_recv_server(void *arg) int send_recv_data(int lfd, int fd, uint32_t total_bytes) { - int flags = fcntl(lfd, F_GETFL); + struct timeval timeout = { .tv_sec = 3 }; ssize_t nr_recv = 0, bytes = 0; struct send_recv_arg arg = { .fd = lfd, @@ -669,7 +680,9 @@ int send_recv_data(int lfd, int fd, uint32_t total_bytes) pthread_t srv_thread; void *thread_ret; char batch[1500]; + int max_fd = fd; int err = 0; + fd_set rset; err = pthread_create(&srv_thread, NULL, send_recv_server, (void *)&arg); if (err) { @@ -679,13 +692,21 @@ int send_recv_data(int lfd, int fd, uint32_t total_bytes) /* recv total_bytes */ while (bytes < total_bytes && !READ_ONCE(arg.stop)) { + /* FD sets */ + FD_ZERO(&rset); + FD_SET(fd, &rset); + + if (select(max_fd + 1, &rset, NULL, NULL, &timeout) == -1) { + log_err("Failed to select"); + err = -1; + break; + } + nr_recv = recv(fd, &batch, MIN(total_bytes - bytes, sizeof(batch)), 0); if (nr_recv == -1 && errno == EINTR) continue; if (nr_recv == -1) { - if (flags & O_NONBLOCK && errno == EWOULDBLOCK) - continue; err = -errno; break; }