From patchwork Fri Nov 11 07:10:08 2016 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit Subject: [08/11] zdtm: add test cases on TCP_CLOSE_WAIT and TCP_LAST_ACK states From: Andrei Vagin X-Patchwork-Id: 2570 Message-Id: <1478848211-23802-9-git-send-email-avagin@openvz.org> To: xemul@virtuozzo.com Cc: criu@openvz.org, Andrei Vagin Date: Fri, 11 Nov 2016 10:10:08 +0300 From: Andrey Vagin 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 --- test/zdtm/static/Makefile | 8 + test/zdtm/static/socket-tcp-close-wait.c | 268 +++++++++++++++++++++++++++ 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, 283 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 1b3f155..b83183e 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 \ @@ -423,6 +427,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..f02d6ff --- /dev/null +++ b/test/zdtm/static/socket-tcp-close-wait.c @@ -0,0 +1,268 @@ +#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 +#include +#include +#include +#include +#include +#include +#include +#include +#include + +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) +{ + 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 + +#ifdef ZDTM_TCP_LOCAL + test_init(argc, argv); +#endif + + 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; + +#ifndef ZDTM_TCP_LOCAL + test_ext_init(argc, argv); +#endif + + 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; + } + +#ifndef ZDTM_TCP_LOCAL + test_init(argc, argv); +#endif + + 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..a0ff359 --- /dev/null +++ b/test/zdtm/static/socket-tcp-close-wait.desc @@ -0,0 +1 @@ +{'flavor': 'h', '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