Skip to content

Commit 5b33fe6

Browse files
TropicaoKernel Patches Daemon
authored andcommitted
selftests/bpf: use start_server_str rather than start_reuseport_server in tc_tunnel
test_tc_tunnel currently uses start_reuseport_server because it needs to frequently start and stop the server, so we need SO_REUSEPORT to avoid getting errors on server restart due to the socket being in TIME_WAIT state. But the test is only using one server at a time, so it is a bit confusing to use this API. Replace start_reuseport with start_sever_str, and provided the relevant callback to set SO_REUSEPORT. Signed-off-by: Alexis Lothoré (eBPF Foundation) <[email protected]>
1 parent f6b67f6 commit 5b33fe6

File tree

1 file changed

+17
-7
lines changed

1 file changed

+17
-7
lines changed

tools/testing/selftests/bpf/prog_tests/test_tc_tunnel.c

Lines changed: 17 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -69,7 +69,7 @@ struct subtest_cfg {
6969
int client_egress_prog_fd;
7070
int server_ingress_prog_fd;
7171
char extra_decap_mod_args[TUNNEL_ARGS_MAX_LEN];
72-
int *server_fd;
72+
int server_fd;
7373
};
7474

7575
struct connection {
@@ -131,28 +131,38 @@ static void set_subtest_addresses(struct subtest_cfg *cfg)
131131
}
132132
}
133133

134+
static int reuseport_cb(int fd, void *opts)
135+
{
136+
int one = 1;
137+
138+
return setsockopt(fd, SOL_SOCKET, SO_REUSEPORT, &one, sizeof(one));
139+
}
140+
134141
static int run_server(struct subtest_cfg *cfg)
135142
{
136143
int family = cfg->ipproto == 6 ? AF_INET6 : AF_INET;
144+
struct network_helper_opts opts = {
145+
.timeout_ms = TIMEOUT_MS,
146+
.post_socket_cb = reuseport_cb,
147+
};
137148
struct nstoken *nstoken;
138149

139150
nstoken = open_netns(SERVER_NS);
140151
if (!ASSERT_OK_PTR(nstoken, "open server ns"))
141152
return -1;
142153

143-
cfg->server_fd = start_reuseport_server(family, SOCK_STREAM,
144-
cfg->server_addr, TEST_PORT,
145-
TIMEOUT_MS, 1);
154+
cfg->server_fd = start_server_str(family, SOCK_STREAM, cfg->server_addr,
155+
TEST_PORT, &opts);
146156
close_netns(nstoken);
147-
if (!ASSERT_OK_PTR(cfg->server_fd, "start server"))
157+
if (!ASSERT_OK_FD(cfg->server_fd, "start server"))
148158
return -1;
149159

150160
return 0;
151161
}
152162

153163
static void stop_server(struct subtest_cfg *cfg)
154164
{
155-
free_fds(cfg->server_fd, 1);
165+
close(cfg->server_fd);
156166
}
157167

158168
static int check_server_rx_data(struct subtest_cfg *cfg,
@@ -188,7 +198,7 @@ static struct connection *connect_client_to_server(struct subtest_cfg *cfg)
188198
return NULL;
189199
}
190200

191-
server_fd = accept(*cfg->server_fd, NULL, NULL);
201+
server_fd = accept(cfg->server_fd, NULL, NULL);
192202
if (server_fd < 0) {
193203
close(client_fd);
194204
free(conn);

0 commit comments

Comments
 (0)