mir/criu
2
0
Fork 0
mirror of https://github.com/checkpoint-restore/criu synced 2025-09-02 15:25:21 +00:00
Code Issues Releases Wiki Activity

test/socket-tcp-skip-in-flight: clean up the test by using the test library

Browse Source 
The test library has functions to create tcp sockets.

Cc: Radostin Stoyanov <rstoyanov1@gmail.com>
Signed-off-by: Andrei Vagin <avagin@gmail.com>
Reviewed-by: Radostin Stoyanov <rstoyanov1@gmail.com>
Signed-off-by: Andrei Vagin <avagin@gmail.com>
This commit is contained in:
Andrei Vagin
2019-01-09 09:56:50 +03:00
parent 1ff2333564
commit 7cd1cf0b16
1 changed files with 36 additions and 110 deletions
Download Patch File Download Diff File Expand all files Collapse all files

146
test/zdtm/static/socket-tcp-skip-in-flight.c
View File

@@ -16,138 +16,64 @@ const char *test_author = "Radostin Stoyanov <rstoyanov1@gmail.com>";
* in-flight TCP connections are ignored. * in-flight TCP connections are ignored.
*/ */
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h> #include <sys/types.h>
#include <sys/socket.h> #include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h>
#include <linux/types.h>
#include <netinet/tcp.h>
#define PORT 1234 #include <stdlib.h>
#define HOST "127.0.0.1" #include <unistd.h>
#include <fcntl.h>
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 main(int argc, char **argv)
{ {
int fd_s; int port = 9990;
int fd_c; int fd_s, fd_c, fd;
int flags;
test_init(argc, argv); test_init(argc, argv);
fd_s = server(); fd_s = tcp_init_server(ZDTM_FAMILY, &port);
if (fd_s < 0) { if (fd_s < 0)
fail("Failed to initialize server\n"); return -1;
flags = fcntl(fd_s, F_GETFL, 0);
if (fcntl(fd_s, F_SETFL, flags | O_NONBLOCK)) {
pr_perror("Unable to set O_NONBLOCK");
return -1; return -1;
} }
fd_c = client(); fd_c = tcp_init_client(ZDTM_FAMILY, "localhost", port);
if (fd_c < 0) { if (fd_c < 0)
fail("Failed to initialize client\n");
return -1; 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_daemon();
test_waitsig(); test_waitsig();
if (check_socket_state(fd_s, TCP_LISTEN)) { if (close(fd_c)) {
fail("Server socket state after restore isn't TCP_LISTEN\n"); fail("Unable to close a client socket");
return 1; return 1;
} }
close(fd_s); fd = tcp_accept_server(fd_s);
if (fd >= 0)
close(fd);
fd_c = tcp_init_client(ZDTM_FAMILY, "localhost", port);
if (fd_c < 0) {
fail("Unable to create a client socket");
return -1;
}
fd = tcp_accept_server(fd_s);
if (fd < 0) {
fail("Unable to accept a new connection");
return 1;
}
close(fd);
close(fd_c); close(fd_c);
close(fd_s);
pass(); pass();
return 0; return 0;