[CRIU] [PATCH 14/21] zdtm: add test cases on TCP_FIN_WAIT* states
Andrei Vagin
avagin at openvz.org
Thu Dec 1 00:32:32 PST 2016
From: Andrey Vagin <avagin at openvz.org>
TCP_FIN_WAIT1 our side has shutdown, waiting to complete
transmission of remaining buffered data
TCP_FIN_WAIT2 all buffered data sent, waiting for remote
to shutdown
Signed-off-by: Andrei Vagin <avagin at virtuozzo.com>
---
test/zdtm/static/Makefile | 9 ++
test/zdtm/static/socket-tcp-fin-wait1.c | 227 ++++++++++++++++++++++++++++
test/zdtm/static/socket-tcp-fin-wait1.desc | 1 +
test/zdtm/static/socket-tcp-fin-wait2.c | 1 +
test/zdtm/static/socket-tcp-fin-wait2.desc | 1 +
test/zdtm/static/socket-tcp6-fin-wait1.c | 1 +
test/zdtm/static/socket-tcp6-fin-wait1.desc | 1 +
test/zdtm/static/socket-tcp6-fin-wait2.c | 1 +
test/zdtm/static/socket-tcp6-fin-wait2.desc | 1 +
9 files changed, 243 insertions(+)
create mode 100644 test/zdtm/static/socket-tcp-fin-wait1.c
create mode 100644 test/zdtm/static/socket-tcp-fin-wait1.desc
create mode 120000 test/zdtm/static/socket-tcp-fin-wait2.c
create mode 120000 test/zdtm/static/socket-tcp-fin-wait2.desc
create mode 120000 test/zdtm/static/socket-tcp6-fin-wait1.c
create mode 120000 test/zdtm/static/socket-tcp6-fin-wait1.desc
create mode 120000 test/zdtm/static/socket-tcp6-fin-wait2.c
create mode 120000 test/zdtm/static/socket-tcp6-fin-wait2.desc
diff --git a/test/zdtm/static/Makefile b/test/zdtm/static/Makefile
index 766774d..17e39ca 100644
--- a/test/zdtm/static/Makefile
+++ b/test/zdtm/static/Makefile
@@ -73,6 +73,10 @@ TST_NOFILE := \
socket-tcpbuf6-local \
socket-tcpbuf6 \
socket-closed-tcp \
+ socket-tcp-fin-wait1 \
+ socket-tcp6-fin-wait1 \
+ socket-tcp-fin-wait2 \
+ socket-tcp6-fin-wait2 \
sock_opts00 \
sock_opts01 \
sk-unix-unconn \
@@ -417,6 +421,11 @@ stopped02: override CFLAGS += -DZDTM_STOPPED_TKILL
stopped12: override CFLAGS += -DZDTM_STOPPED_KILL -DZDTM_STOPPED_TKILL
clone_fs: override LDLIBS += -pthread
+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
+
$(LIB): force
$(Q) $(MAKE) -C $(LIBDIR)
diff --git a/test/zdtm/static/socket-tcp-fin-wait1.c b/test/zdtm/static/socket-tcp-fin-wait1.c
new file mode 100644
index 0000000..1d66f4e
--- /dev/null
+++ b/test/zdtm/static/socket-tcp-fin-wait1.c
@@ -0,0 +1,227 @@
+#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_FIN_WAIT* 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 TEST_MSG "Hello World!"
+#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;
+}
+
+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;
+ char buf[BUF_SIZE];
+
+ 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;
+ write(fd, TEST_MSG, 2);
+
+ ctl_fd = tcp_init_client(ZDTM_FAMILY, "127.0.0.1", port);
+ if (fd < 0)
+ return 1;
+
+ if (read(ctl_fd, &c, 1) != 0) {
+ pr_err("read");
+ return 1;
+ }
+
+ if (write(fd, TEST_MSG + 2, sizeof(TEST_MSG) - 2) != sizeof(TEST_MSG) - 2) {
+ pr_err("write");
+ return 1;
+ }
+
+ if (shutdown(fd, SHUT_WR) == -1) {
+ pr_err("shutdown");
+ return 1;
+ }
+
+ size = clean_sk_buf(fd);
+ if (size < 0)
+ return 1;
+
+ 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;
+ }
+
+#if !defined(ZDTM_TCP_FIN_WAIT2)
+ snd_size = fill_sock_buf(fd);
+ if (snd_size <= 0)
+ return 1;
+#endif
+
+ if (shutdown(fd, SHUT_WR) == -1) {
+ pr_err("shutdown");
+ return 1;
+ }
+
+
+ test_daemon();
+ test_waitsig();
+
+ if (shutdown(ctl_fd, SHUT_WR) == -1) {
+ pr_err("shutdown");
+ return 1;
+ }
+
+ if (recv(fd, buf, sizeof(buf), MSG_WAITALL) != sizeof(TEST_MSG) ||
+ strncmp(buf, TEST_MSG, sizeof(TEST_MSG))) {
+ pr_err("read");
+ 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-fin-wait1.desc b/test/zdtm/static/socket-tcp-fin-wait1.desc
new file mode 100644
index 0000000..155e103
--- /dev/null
+++ b/test/zdtm/static/socket-tcp-fin-wait1.desc
@@ -0,0 +1 @@
+{'opts': '--tcp-established', 'flags': 'nouser samens', 'feature' : 'tcp_half_closed'}
diff --git a/test/zdtm/static/socket-tcp-fin-wait2.c b/test/zdtm/static/socket-tcp-fin-wait2.c
new file mode 120000
index 0000000..2dede5f
--- /dev/null
+++ b/test/zdtm/static/socket-tcp-fin-wait2.c
@@ -0,0 +1 @@
+socket-tcp-fin-wait1.c
\ No newline at end of file
diff --git a/test/zdtm/static/socket-tcp-fin-wait2.desc b/test/zdtm/static/socket-tcp-fin-wait2.desc
new file mode 120000
index 0000000..fba0e07
--- /dev/null
+++ b/test/zdtm/static/socket-tcp-fin-wait2.desc
@@ -0,0 +1 @@
+socket-tcp-fin-wait1.desc
\ No newline at end of file
diff --git a/test/zdtm/static/socket-tcp6-fin-wait1.c b/test/zdtm/static/socket-tcp6-fin-wait1.c
new file mode 120000
index 0000000..2dede5f
--- /dev/null
+++ b/test/zdtm/static/socket-tcp6-fin-wait1.c
@@ -0,0 +1 @@
+socket-tcp-fin-wait1.c
\ No newline at end of file
diff --git a/test/zdtm/static/socket-tcp6-fin-wait1.desc b/test/zdtm/static/socket-tcp6-fin-wait1.desc
new file mode 120000
index 0000000..fba0e07
--- /dev/null
+++ b/test/zdtm/static/socket-tcp6-fin-wait1.desc
@@ -0,0 +1 @@
+socket-tcp-fin-wait1.desc
\ No newline at end of file
diff --git a/test/zdtm/static/socket-tcp6-fin-wait2.c b/test/zdtm/static/socket-tcp6-fin-wait2.c
new file mode 120000
index 0000000..a95fa07
--- /dev/null
+++ b/test/zdtm/static/socket-tcp6-fin-wait2.c
@@ -0,0 +1 @@
+socket-tcp-fin-wait2.c
\ No newline at end of file
diff --git a/test/zdtm/static/socket-tcp6-fin-wait2.desc b/test/zdtm/static/socket-tcp6-fin-wait2.desc
new file mode 120000
index 0000000..fba0e07
--- /dev/null
+++ b/test/zdtm/static/socket-tcp6-fin-wait2.desc
@@ -0,0 +1 @@
+socket-tcp-fin-wait1.desc
\ No newline at end of file
--
2.7.4
More information about the CRIU
mailing list