fix2

Signed-off-by: Geliang Tang <tanggeliang@kylinos.cn>
geliangtang · Oct 22, 2024 · 2d445c1 · 2d445c1
1 parent cf9a278
commit 2d445c1
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/tools/testing/selftests/bpf/prog_tests/mptcp.c b/tools/testing/selftests/bpf/prog_tests/mptcp.c
@@ -676,7 +676,7 @@ static int get_events_line(char *type, char *line)
 	return 0;
 }
 
-static int wait_for_token(void)
+static int wait_for_token(int fd)
 {
 	char line[1024], *str;
 	__u32 token;
@@ -686,6 +686,7 @@ static int wait_for_token(void)
 	for (i = 0; i < 10; i++) {
 		SYS(fail, "ip netns exec %s echo sleep 0.2 s", NS_TEST);
 		usleep(200000); /* 0.2s */
+		send_byte(fd);
 
 		sync();
 		if (get_events_line("type:2", line))
@@ -742,9 +743,7 @@ static void run_iters_address(void)
 	if (!ASSERT_OK_FD(accept_fd, "accept"))
 		goto close_client;
 
-	usleep(200000); /* 0.2s */
-	send_byte(client_fd);
-	token = wait_for_token();
+	token = wait_for_token(client_fd);
 	if (!token)
 		goto close_client;
 	recv_byte(accept_fd);
@@ -926,9 +925,7 @@ static void run_userspace_pm(enum mptcp_pm_family family)
 	if (!ASSERT_OK_FD(accept_fd, "accept"))
 		goto close_client;
 
-	usleep(200000); /* 0.2s */
-	send_byte(client_fd);
-	token = wait_for_token();
+	token = wait_for_token(client_fd);
 	if (!token)
 		goto close_client;
 	recv_byte(accept_fd);