From 8a90bc8302fa7af382e0598411b9bbf129f672a5 Mon Sep 17 00:00:00 2001 From: Radostin Stoyanov Date: Fri, 4 Jan 2019 20:44:18 +0000 Subject: [PATCH] test/static: Add test for --skip-in-flight Changes since v1: - Refactor test case based on Andrei's comments Changes since v2: - Use ZDTM_FAMILY instead of AF_INET Signed-off-by: Radostin Stoyanov Signed-off-by: Andrei Vagin --- test/zdtm/static/Makefile | 2 + test/zdtm/static/socket-tcp-skip-in-flight.c | 154 ++++++++++++++++++ .../static/socket-tcp-skip-in-flight.desc | 1 + 3 files changed, 157 insertions(+) create mode 100644 test/zdtm/static/socket-tcp-skip-in-flight.c create mode 100644 test/zdtm/static/socket-tcp-skip-in-flight.desc diff --git a/test/zdtm/static/Makefile b/test/zdtm/static/Makefile index 7b8d66377..2d13e7d2a 100644 --- a/test/zdtm/static/Makefile +++ b/test/zdtm/static/Makefile @@ -104,6 +104,7 @@ TST_NOFILE := \ socket-tcp-unconn \ socket-tcp6-unconn \ socket-tcp-syn-sent \ + socket-tcp-skip-in-flight \ sock_opts00 \ sock_opts01 \ sk-unix-unconn \ @@ -486,6 +487,7 @@ socket_listen4v6: CFLAGS += -D ZDTM_IPV4V6 socket-tcp6-closed: CFLAGS += -D ZDTM_IPV6 socket-tcp6-closed: CFLAGS += -D ZDTM_IPV4V6 socket-tcp-closed-last-ack: CFLAGS += -D ZDTM_TCP_LAST_ACK +socket-tcp-skip-in-flight: CFLAGS += -D ZDTM_IPV4V6 tun_ns: CFLAGS += -DTUN_NS mnt_ext_manual: CFLAGS += -D ZDTM_EXTMAP_MANUAL sigpending: LDLIBS += -lrt diff --git a/test/zdtm/static/socket-tcp-skip-in-flight.c b/test/zdtm/static/socket-tcp-skip-in-flight.c new file mode 100644 index 000000000..eef73d992 --- /dev/null +++ b/test/zdtm/static/socket-tcp-skip-in-flight.c @@ -0,0 +1,154 @@ +#include "zdtmtst.h" + +#ifdef ZDTM_IPV4V6 +#define ZDTM_FAMILY AF_INET +#elif defined(ZDTM_IPV6) +#define ZDTM_FAMILY AF_INET6 +#else +#define ZDTM_FAMILY AF_INET +#endif + +const char *test_doc = "Check that in-flight TCP connections are ignored\n"; +const char *test_author = "Radostin Stoyanov "; + +/* Description: + * Initialise server and client tcp sockets and verify that + * in-flight TCP connections are ignored. + */ + +#include +#include +#include +#include +#include +#include +#include +#include +#include + +#define PORT 1234 +#define HOST "127.0.0.1" + +static int check_socket_state(int sk, int state) +{ + int err; + struct { + __u8 tcpi_state; + } info; + socklen_t len = sizeof(info); + + err = getsockopt(sk, IPPROTO_TCP, TCP_INFO, (void *)&info, &len); + if (err != 0) { + pr_perror("Can't get socket state\n"); + return -1; + } else if (info.tcpi_state != state) { + fail("Invalid socket state (%i)\n", (int)info.tcpi_state); + return -1; + } + + return 0; +} + +int open_socket() +{ + int fd; + fd = socket(ZDTM_FAMILY, SOCK_STREAM, 0); + if (fd < 0) { + fail("Failed to open socket\n"); + return -1; + } + return fd; +} + +int server() +{ + int fd_s; + struct sockaddr_in serv_addr; + + fd_s = open_socket(); + if (fd_s < 0) + return -1; + + bzero((char *) &serv_addr, sizeof(serv_addr)); + serv_addr.sin_family = ZDTM_FAMILY; + serv_addr.sin_addr.s_addr = INADDR_ANY; + serv_addr.sin_port = htons(PORT); + + if (bind(fd_s, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0) { + fail("Failed to bind"); + return -1; + } + + listen(fd_s, 1); + + /* Listen but do not accept connect()-ed TCP connection. */ + + return fd_s; +} + +int client() +{ + int fd_c; + struct sockaddr_in serv_addr; + struct hostent *server; + + fd_c = open_socket(); + if (fd_c < 0) + return -1; + + server = gethostbyname(HOST); + if (server == NULL) { + fail("Failed to get host by name\n"); + return -1; + } + + bzero((char *) &serv_addr, sizeof(serv_addr)); + serv_addr.sin_family = ZDTM_FAMILY; + bcopy((char *)server->h_addr, (char *)&serv_addr.sin_addr.s_addr, server->h_length); + serv_addr.sin_port = htons(PORT); + if (connect(fd_c,(struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0) { + fail("Failed to get host by name\n"); + return -1; + } + + return fd_c; +} + +int main(int argc, char **argv) +{ + int fd_s; + int fd_c; + + test_init(argc, argv); + + fd_s = server(); + if (fd_s < 0) { + fail("Failed to initialize server\n"); + return -1; + } + + fd_c = client(); + if (fd_c < 0) { + fail("Failed to initialize client\n"); + return -1; + } + + if (check_socket_state(fd_s, TCP_LISTEN)) { + fail("Server socket state before restore isn't TCP_LISTEN\n"); + return 1; + } + + test_daemon(); + test_waitsig(); + + if (check_socket_state(fd_s, TCP_LISTEN)) { + fail("Server socket state after restore isn't TCP_LISTEN\n"); + return 1; + } + + close(fd_s); + close(fd_c); + + pass(); + return 0; +} diff --git a/test/zdtm/static/socket-tcp-skip-in-flight.desc b/test/zdtm/static/socket-tcp-skip-in-flight.desc new file mode 100644 index 000000000..0ef6e6d2b --- /dev/null +++ b/test/zdtm/static/socket-tcp-skip-in-flight.desc @@ -0,0 +1 @@ +{'opts': '--tcp-established --skip-in-flight'}