bpf: selftests: Add fastopen_connect to network_helpers

This patch adds a fastopen_connect() helper which will
be used in a later test.

Signed-off-by: Martin KaFai Lau <kafai@fb.com>
Signed-off-by: Alexei Starovoitov <ast@kernel.org>
Link: https://lore.kernel.org/bpf/20200820190111.2886196-1-kafai@fb.com

diff --git a/tools/testing/selftests/bpf/network_helpers.c b/tools/testing/selftests/bpf/network_helpers.c
index f566556..12ee402 100644 (file)
--- a/tools/testing/selftests/bpf/network_helpers.c
+++ b/tools/testing/selftests/bpf/network_helpers.c
@@ -104,6 +104,43 @@ error_close:
        return -1;
 }
 
+int fastopen_connect(int server_fd, const char *data, unsigned int data_len,
+                    int timeout_ms)
+{
+       struct sockaddr_storage addr;
+       socklen_t addrlen = sizeof(addr);
+       struct sockaddr_in *addr_in;
+       int fd, ret;
+
+       if (getsockname(server_fd, (struct sockaddr *)&addr, &addrlen)) {
+               log_err("Failed to get server addr");
+               return -1;
+       }
+
+       addr_in = (struct sockaddr_in *)&addr;
+       fd = socket(addr_in->sin_family, SOCK_STREAM, 0);
+       if (fd < 0) {
+               log_err("Failed to create client socket");
+               return -1;
+       }
+
+       if (settimeo(fd, timeout_ms))
+               goto error_close;
+
+       ret = sendto(fd, data, data_len, MSG_FASTOPEN, (struct sockaddr *)&addr,
+                    addrlen);
+       if (ret != data_len) {
+               log_err("sendto(data, %u) != %d\n", data_len, ret);
+               goto error_close;
+       }
+
+       return fd;
+
+error_close:
+       save_errno_close(fd);
+       return -1;
+}
+
 static int connect_fd_to_addr(int fd,
                              const struct sockaddr_storage *addr,
                              socklen_t addrlen)

diff --git a/tools/testing/selftests/bpf/network_helpers.h b/tools/testing/selftests/bpf/network_helpers.h
index c3728f6..7205f8a 100644 (file)
--- a/tools/testing/selftests/bpf/network_helpers.h
+++ b/tools/testing/selftests/bpf/network_helpers.h
@@ -37,6 +37,8 @@ int start_server(int family, int type, const char *addr, __u16 port,
                 int timeout_ms);
 int connect_to_fd(int server_fd, int timeout_ms);
 int connect_fd_to_fd(int client_fd, int server_fd, int timeout_ms);
+int fastopen_connect(int server_fd, const char *data, unsigned int data_len,
+                    int timeout_ms);
 int make_sockaddr(int family, const char *addr_str, __u16 port,
                  struct sockaddr_storage *addr, socklen_t *len);