Commit 95b88500 authored by Geliang Tang's avatar Geliang Tang Committed by Martin KaFai Lau

selftests/bpf: Add opts argument for __start_server

This patch adds network_helper_opts parameter for __start_server()
instead of "int protocol" and "int timeout_ms". This not only reduces
the number of parameters, but also makes it more flexible.
Signed-off-by: default avatarGeliang Tang <tanggeliang@kylinos.cn>
Link: https://lore.kernel.org/r/127d2f0929980b41f757dcfebe1b667e6bfb43f1.1714014697.git.tanggeliang@kylinos.cnSigned-off-by: default avatarMartin KaFai Lau <martin.lau@kernel.org>
parent 06ebfd11
...@@ -80,19 +80,19 @@ int settimeo(int fd, int timeout_ms) ...@@ -80,19 +80,19 @@ int settimeo(int fd, int timeout_ms)
#define save_errno_close(fd) ({ int __save = errno; close(fd); errno = __save; }) #define save_errno_close(fd) ({ int __save = errno; close(fd); errno = __save; })
static int __start_server(int type, int protocol, const struct sockaddr *addr, static int __start_server(int type, const struct sockaddr *addr, socklen_t addrlen,
socklen_t addrlen, int timeout_ms, bool reuseport) bool reuseport, const struct network_helper_opts *opts)
{ {
int on = 1; int on = 1;
int fd; int fd;
fd = socket(addr->sa_family, type, protocol); fd = socket(addr->sa_family, type, opts->proto);
if (fd < 0) { if (fd < 0) {
log_err("Failed to create server socket"); log_err("Failed to create server socket");
return -1; return -1;
} }
if (settimeo(fd, timeout_ms)) if (settimeo(fd, opts->timeout_ms))
goto error_close; goto error_close;
if (reuseport && if (reuseport &&
...@@ -123,14 +123,17 @@ static int __start_server(int type, int protocol, const struct sockaddr *addr, ...@@ -123,14 +123,17 @@ static int __start_server(int type, int protocol, const struct sockaddr *addr,
static int start_server_proto(int family, int type, int protocol, static int start_server_proto(int family, int type, int protocol,
const char *addr_str, __u16 port, int timeout_ms) const char *addr_str, __u16 port, int timeout_ms)
{ {
struct network_helper_opts opts = {
.timeout_ms = timeout_ms,
.proto = protocol,
};
struct sockaddr_storage addr; struct sockaddr_storage addr;
socklen_t addrlen; socklen_t addrlen;
if (make_sockaddr(family, addr_str, port, &addr, &addrlen)) if (make_sockaddr(family, addr_str, port, &addr, &addrlen))
return -1; return -1;
return __start_server(type, protocol, (struct sockaddr *)&addr, return __start_server(type, (struct sockaddr *)&addr, addrlen, false, &opts);
addrlen, timeout_ms, false);
} }
int start_server(int family, int type, const char *addr_str, __u16 port, int start_server(int family, int type, const char *addr_str, __u16 port,
...@@ -149,6 +152,9 @@ int start_mptcp_server(int family, const char *addr_str, __u16 port, ...@@ -149,6 +152,9 @@ int start_mptcp_server(int family, const char *addr_str, __u16 port,
int *start_reuseport_server(int family, int type, const char *addr_str, int *start_reuseport_server(int family, int type, const char *addr_str,
__u16 port, int timeout_ms, unsigned int nr_listens) __u16 port, int timeout_ms, unsigned int nr_listens)
{ {
struct network_helper_opts opts = {
.timeout_ms = timeout_ms,
};
struct sockaddr_storage addr; struct sockaddr_storage addr;
unsigned int nr_fds = 0; unsigned int nr_fds = 0;
socklen_t addrlen; socklen_t addrlen;
...@@ -164,8 +170,7 @@ int *start_reuseport_server(int family, int type, const char *addr_str, ...@@ -164,8 +170,7 @@ int *start_reuseport_server(int family, int type, const char *addr_str,
if (!fds) if (!fds)
return NULL; return NULL;
fds[0] = __start_server(type, 0, (struct sockaddr *)&addr, addrlen, fds[0] = __start_server(type, (struct sockaddr *)&addr, addrlen, true, &opts);
timeout_ms, true);
if (fds[0] == -1) if (fds[0] == -1)
goto close_fds; goto close_fds;
nr_fds = 1; nr_fds = 1;
...@@ -174,8 +179,7 @@ int *start_reuseport_server(int family, int type, const char *addr_str, ...@@ -174,8 +179,7 @@ int *start_reuseport_server(int family, int type, const char *addr_str,
goto close_fds; goto close_fds;
for (; nr_fds < nr_listens; nr_fds++) { for (; nr_fds < nr_listens; nr_fds++) {
fds[nr_fds] = __start_server(type, 0, (struct sockaddr *)&addr, fds[nr_fds] = __start_server(type, (struct sockaddr *)&addr, addrlen, true, &opts);
addrlen, timeout_ms, true);
if (fds[nr_fds] == -1) if (fds[nr_fds] == -1)
goto close_fds; goto close_fds;
} }
...@@ -193,8 +197,7 @@ int start_server_addr(int type, const struct sockaddr_storage *addr, socklen_t l ...@@ -193,8 +197,7 @@ int start_server_addr(int type, const struct sockaddr_storage *addr, socklen_t l
if (!opts) if (!opts)
opts = &default_opts; opts = &default_opts;
return __start_server(type, 0, (struct sockaddr *)addr, len, return __start_server(type, (struct sockaddr *)addr, len, 0, opts);
opts->timeout_ms, 0);
} }
void free_fds(int *fds, unsigned int nr_close_fds) void free_fds(int *fds, unsigned int nr_close_fds)
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment