[CRIU] [PATCH 15/21] zdtm: add test cases on TCP_CLOSE_WAIT and TCP_LAST_ACK states

Andrei Vagin avagin at openvz.org
Thu Dec 1 00:32:33 PST 2016


From: Andrey Vagin <avagin at openvz.org>

TCP_CLOSE_WAIT remote side has shutdown and is waiting for
               us to finish writing our data and to shutdown
               (we have to close() to move on to LAST_ACK)

TCP_LAST_ACK   out side has shutdown after remote has
               shutdown.  There may still be data in our
               buffer that we have to finish sending

Signed-off-by: Andrei Vagin <avagin at virtuozzo.com>
---
 test/zdtm/static/Makefile                    |   8 +
 test/zdtm/static/socket-tcp-close-wait.c     | 266 +++++++++++++++++++++++++++
 test/zdtm/static/socket-tcp-close-wait.desc  |   1 +
 test/zdtm/static/socket-tcp-last-ack.c       |   1 +
 test/zdtm/static/socket-tcp-last-ack.desc    |   1 +
 test/zdtm/static/socket-tcp6-close-wait.c    |   1 +
 test/zdtm/static/socket-tcp6-close-wait.desc |   1 +
 test/zdtm/static/socket-tcp6-last-ack.c      |   1 +
 test/zdtm/static/socket-tcp6-last-ack.desc   |   1 +
 9 files changed, 281 insertions(+)
 create mode 100644 test/zdtm/static/socket-tcp-close-wait.c
 create mode 100644 test/zdtm/static/socket-tcp-close-wait.desc
 create mode 120000 test/zdtm/static/socket-tcp-last-ack.c
 create mode 100644 test/zdtm/static/socket-tcp-last-ack.desc
 create mode 120000 test/zdtm/static/socket-tcp6-close-wait.c
 create mode 120000 test/zdtm/static/socket-tcp6-close-wait.desc
 create mode 120000 test/zdtm/static/socket-tcp6-last-ack.c
 create mode 120000 test/zdtm/static/socket-tcp6-last-ack.desc

diff --git a/test/zdtm/static/Makefile b/test/zdtm/static/Makefile
index 17e39ca..61fbc9f 100644
--- a/test/zdtm/static/Makefile
+++ b/test/zdtm/static/Makefile
@@ -77,6 +77,10 @@ TST_NOFILE	:=				\
 		socket-tcp6-fin-wait1		\
 		socket-tcp-fin-wait2		\
 		socket-tcp6-fin-wait2		\
+		socket-tcp-close-wait		\
+		socket-tcp6-close-wait		\
+		socket-tcp-last-ack		\
+		socket-tcp6-last-ack		\
 		sock_opts00			\
 		sock_opts01			\
 		sk-unix-unconn			\
@@ -425,6 +429,10 @@ socket-tcp-fin-wait1:	override CFLAGS += -D ZDTM_TCP_FIN_WAIT1
 socket-tcp-fin-wait2:	override CFLAGS += -D ZDTM_TCP_FIN_WAIT2
 socket-tcp6-fin-wait1:	override CFLAGS += -D ZDTM_TCP_FIN_WAIT1 -D ZDTM_IPV6
 socket-tcp6-fin-wait2:	override CFLAGS += -D ZDTM_TCP_FIN_WAIT2 -D ZDTM_IPV6
+socket-tcp-close-wait:	override CFLAGS += -D ZDTM_TCP_CLOSE_WAIT
+socket-tcp6-close-wait:	override CFLAGS += -D ZDTM_TCP_CLOSE_WAIT -D ZDTM_IPV6
+socket-tcp-last-ack:	override CFLAGS += -D ZDTM_TCP_LAST_ACK
+socket-tcp6-last-ack:	override CFLAGS += -D ZDTM_TCP_LAST_ACK -D ZDTM_IPV6
 
 $(LIB):	force
 	$(Q) $(MAKE) -C $(LIBDIR)
diff --git a/test/zdtm/static/socket-tcp-close-wait.c b/test/zdtm/static/socket-tcp-close-wait.c
new file mode 100644
index 0000000..3b2681c
--- /dev/null
+++ b/test/zdtm/static/socket-tcp-close-wait.c
@@ -0,0 +1,266 @@
+#include "zdtmtst.h"
+
+#ifdef ZDTM_IPV6
+#define ZDTM_FAMILY AF_INET6
+#else
+#define ZDTM_FAMILY AF_INET
+#endif
+
+const char *test_doc = "Check sockets in TCP_WAIT_STOP and TCP_LAST_ACK states\n";
+const char *test_author = "Andrey Vagin <avagin at parallels.com";
+
+#include <stdio.h>
+#include <sys/types.h>
+#include <unistd.h>
+#include <string.h>
+#include <errno.h>
+#include <stdlib.h>
+#include <signal.h>
+#include <fcntl.h>
+#include <sys/socket.h>
+#include <netinet/tcp.h>
+
+static int port = 8880;
+
+#define BUF_SIZE 4096
+
+int fill_sock_buf(int fd)
+{
+	int flags;
+	int size;
+	int ret;
+
+	flags = fcntl(fd, F_GETFL, 0);
+	if (flags == -1) {
+		pr_err("Can't get flags");
+		return -1;
+	}
+	if (fcntl(fd, F_SETFL, flags | O_NONBLOCK) == -1) {
+		pr_err("Can't set flags");
+		return -1;
+	}
+
+	size = 0;
+	while (1) {
+		char zdtm[] = "zdtm test packet";
+		ret = write(fd, zdtm, sizeof(zdtm));
+		if (ret == -1) {
+			if (errno == EAGAIN)
+				break;
+			pr_err("write");
+			return -1;
+		}
+		size += ret;
+	}
+
+	if (fcntl(fd, F_SETFL, flags) == -1) {
+		pr_err("Can't set flags");
+		return -1;
+	}
+
+	test_msg("snd_size = %d\n", size);
+
+	return size;
+}
+
+static int clean_sk_buf(int fd)
+{
+	int size, ret;
+	char buf[BUF_SIZE];
+
+	size = 0;
+	while (1) {
+		ret = read(fd, buf, sizeof(buf));
+		if (ret == -1) {
+			pr_err("read");
+			return -11;
+		}
+
+		if (ret == 0)
+			break;
+
+		size += ret;
+	}
+
+	test_msg("rcv_size = %d\n", size);
+
+	return size;
+}
+
+#define TEST_MSG "Hello World!"
+
+int main(int argc, char **argv)
+{
+	char *newns = getenv("ZDTM_NEWNS");
+	int fd, fd_s, ctl_fd;
+	pid_t extpid;
+	int pfd[2];
+	int ret, snd_size = 0, rcv_size = 0;
+#ifndef ZDTM_TCP_LAST_ACK
+	char buf[BUF_SIZE];
+#endif
+
+	if (newns)
+		test_init(argc, argv);
+
+	if (pipe(pfd)) {
+		pr_err("pipe() failed");
+		return 1;
+	}
+
+	extpid = fork();
+	if (extpid < 0) {
+		pr_err("fork() failed");
+		return 1;
+	} else if (extpid == 0) {
+		int size = 0;
+		char c;
+
+		if (!newns)
+			test_ext_init(argc, argv);
+
+		close(pfd[1]);
+		if (read(pfd[0], &port, sizeof(port)) != sizeof(port)) {
+			pr_err("Can't read port\n");
+			return 1;
+		}
+
+		fd = tcp_init_client(ZDTM_FAMILY, "127.0.0.1", port);
+		if (fd < 0)
+			return 1;
+
+		ctl_fd = tcp_init_client(ZDTM_FAMILY, "127.0.0.1", port);
+		if (fd < 0)
+			return 1;
+
+		if (read(ctl_fd, &size, sizeof(size)) != sizeof(size)) {
+			pr_err("write");
+			return 1;
+		}
+
+		if (shutdown(fd, SHUT_WR) == -1) {
+			pr_err("shutdown");
+			return 1;
+		}
+
+		if (write(ctl_fd, &size, sizeof(size)) != sizeof(size)) {
+			pr_err("write");
+			return 1;
+		}
+
+		if (read(ctl_fd, &c, 1) != 0) {
+			pr_err("read");
+			return 1;
+		}
+
+#ifdef ZDTM_TCP_LAST_ACK
+		size = clean_sk_buf(fd);
+		if (size < 0)
+			return 1;
+#else
+		if (read(fd, buf, sizeof(buf)) != sizeof(TEST_MSG) ||
+		    strncmp(buf, TEST_MSG, sizeof(TEST_MSG))) {
+			pr_err("read");
+			return 1;
+		}
+#endif
+
+		write(ctl_fd, &size, sizeof(size));
+		close(fd);
+
+		return 0;
+	}
+
+	if (!newns)
+		test_init(argc, argv);
+
+	if ((fd_s = tcp_init_server(ZDTM_FAMILY, &port)) < 0) {
+		pr_err("initializing server failed");
+		return 1;
+	}
+
+	close(pfd[0]);
+	if (write(pfd[1], &port, sizeof(port)) != sizeof(port)) {
+		pr_err("Can't send port");
+		return 1;
+	}
+	close(pfd[1]);
+
+	/*
+	 * parent is server of TCP connection
+	 */
+	fd = tcp_accept_server(fd_s);
+	if (fd < 0) {
+		pr_err("can't accept client connection %m");
+		return 1;
+	}
+
+	ctl_fd = tcp_accept_server(fd_s);
+	if (ctl_fd < 0) {
+		pr_err("can't accept client connection %m");
+		return 1;
+	}
+
+#ifdef ZDTM_TCP_LAST_ACK
+	snd_size = fill_sock_buf(fd);
+	if (snd_size <= 0)
+		return 1;
+#endif
+
+	if (write(ctl_fd, &ret, sizeof(ret)) != sizeof(ret)) {
+		pr_err("read");
+		return 1;
+	}
+
+	if (read(ctl_fd, &ret, sizeof(ret)) != sizeof(ret)) {
+		pr_err("read");
+		return 1;
+	}
+
+#ifdef ZDTM_TCP_LAST_ACK
+	if (shutdown(fd, SHUT_WR) == -1) {
+		pr_err("shutdown");
+		return 1;
+	}
+#endif
+
+	test_daemon();
+	test_waitsig();
+
+	if (shutdown(ctl_fd, SHUT_WR) == -1) {
+		pr_err("shutdown");
+		return 1;
+	}
+
+#ifndef ZDTM_TCP_LAST_ACK
+	if (write(fd, TEST_MSG, sizeof(TEST_MSG)) != sizeof(TEST_MSG)) {
+		pr_err("write");
+		return 1;
+	}
+
+	if (shutdown(fd, SHUT_WR) == -1) {
+		pr_err("shutdown");
+		return 1;
+	}
+#endif
+
+	rcv_size = clean_sk_buf(fd);
+
+	if (ret != rcv_size) {
+		fail("The child sent %d bytes, but the parent received %d bytes\n", ret, rcv_size);
+		return 1;
+	}
+
+	if (read(ctl_fd, &ret, sizeof(ret)) != sizeof(ret)) {
+		pr_err("read");
+		return 1;
+	}
+
+	if (ret != snd_size) {
+		fail("The parent sent %d bytes, but the child received %d bytes\n", snd_size, ret);
+		return 1;
+	}
+
+	pass();
+	return 0;
+}
diff --git a/test/zdtm/static/socket-tcp-close-wait.desc b/test/zdtm/static/socket-tcp-close-wait.desc
new file mode 100644
index 0000000..155e103
--- /dev/null
+++ b/test/zdtm/static/socket-tcp-close-wait.desc
@@ -0,0 +1 @@
+{'opts': '--tcp-established', 'flags': 'nouser samens', 'feature' : 'tcp_half_closed'}
diff --git a/test/zdtm/static/socket-tcp-last-ack.c b/test/zdtm/static/socket-tcp-last-ack.c
new file mode 120000
index 0000000..20d7e78
--- /dev/null
+++ b/test/zdtm/static/socket-tcp-last-ack.c
@@ -0,0 +1 @@
+socket-tcp-close-wait.c
\ No newline at end of file
diff --git a/test/zdtm/static/socket-tcp-last-ack.desc b/test/zdtm/static/socket-tcp-last-ack.desc
new file mode 100644
index 0000000..a0ff359
--- /dev/null
+++ b/test/zdtm/static/socket-tcp-last-ack.desc
@@ -0,0 +1 @@
+{'flavor': 'h', 'opts': '--tcp-established', 'flags': 'nouser samens', 'feature' : 'tcp_half_closed'}
diff --git a/test/zdtm/static/socket-tcp6-close-wait.c b/test/zdtm/static/socket-tcp6-close-wait.c
new file mode 120000
index 0000000..20d7e78
--- /dev/null
+++ b/test/zdtm/static/socket-tcp6-close-wait.c
@@ -0,0 +1 @@
+socket-tcp-close-wait.c
\ No newline at end of file
diff --git a/test/zdtm/static/socket-tcp6-close-wait.desc b/test/zdtm/static/socket-tcp6-close-wait.desc
new file mode 120000
index 0000000..df1973c
--- /dev/null
+++ b/test/zdtm/static/socket-tcp6-close-wait.desc
@@ -0,0 +1 @@
+socket-tcp-close-wait.desc
\ No newline at end of file
diff --git a/test/zdtm/static/socket-tcp6-last-ack.c b/test/zdtm/static/socket-tcp6-last-ack.c
new file mode 120000
index 0000000..1f0bedf
--- /dev/null
+++ b/test/zdtm/static/socket-tcp6-last-ack.c
@@ -0,0 +1 @@
+socket-tcp-last-ack.c
\ No newline at end of file
diff --git a/test/zdtm/static/socket-tcp6-last-ack.desc b/test/zdtm/static/socket-tcp6-last-ack.desc
new file mode 120000
index 0000000..caace98
--- /dev/null
+++ b/test/zdtm/static/socket-tcp6-last-ack.desc
@@ -0,0 +1 @@
+socket-tcp-last-ack.desc
\ No newline at end of file
-- 
2.7.4



More information about the CRIU mailing list