[CRIU] [PATCH] soccr: add one more test to check libsoccr_save/libsoccr_restore

[Andrei Vagin]

From: Andrei Vagin <avagin at virtuozzo.com>

The previous test uses only libsoccr_restore. This test
creates a tcp connection, then it dumps and restore it.

Signed-off-by: Andrei Vagin <avagin at virtuozzo.com>
---
 soccr/test/Makefile      |  10 ++-
 soccr/test/tcp-conn-v6.c |   1 +
 soccr/test/tcp-conn.c    | 168 +++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 178 insertions(+), 1 deletion(-)
 create mode 120000 soccr/test/tcp-conn-v6.c
 create mode 100644 soccr/test/tcp-conn.c

diff --git a/soccr/test/Makefile b/soccr/test/Makefile
index 96ee647..4585400 100644
--- a/soccr/test/Makefile
+++ b/soccr/test/Makefile
@@ -12,7 +12,15 @@ tcp-constructor: tcp-constructor.c ../libsoccr.a
 clean:
 	rm -f tcp-constructor
 
-test: tcp-constructor
+tcp-conn: tcp-conn.c
+	$(CC) $(CFLAGS) tcp-conn.c -o tcp-conn $(LDFLAGS)
+
+tcp-conn-v6: tcp-conn-v6.c
+	$(CC) $(CFLAGS) -DTEST_IPV6 tcp-conn-v6.c -o tcp-conn-v6 $(LDFLAGS)
+
+test: tcp-constructor tcp-conn tcp-conn-v6
+	unshare -n sh -c "ip link set up dev lo; ./tcp-conn"
+	unshare -n sh -c "ip link set up dev lo; ./tcp-conn-v6"
 	python run.py ./$(RUN)
 
 .PHONY: test
diff --git a/soccr/test/tcp-conn-v6.c b/soccr/test/tcp-conn-v6.c
new file mode 120000
index 0000000..81da796
--- /dev/null
+++ b/soccr/test/tcp-conn-v6.c
@@ -0,0 +1 @@
+tcp-conn.c
\ No newline at end of file
diff --git a/soccr/test/tcp-conn.c b/soccr/test/tcp-conn.c
new file mode 100644
index 0000000..1a1a5bb
--- /dev/null
+++ b/soccr/test/tcp-conn.c
@@ -0,0 +1,168 @@
+#include <sys/socket.h>
+#include <arpa/inet.h>  /* for srvaddr_in and inet_ntoa() */
+#include <string.h>
+#include <stdio.h>
+#include <unistd.h>
+#include "../soccr.h"
+#include <stdarg.h>
+
+#define pr_perror(fmt, ...) printf(fmt ": %m\n", ##__VA_ARGS__)
+
+enum {
+	TCP_NO_QUEUE,
+	TCP_RECV_QUEUE,
+	TCP_SEND_QUEUE,
+	TCP_QUEUES_NR,
+};
+static void pr_printf(unsigned int level, const char *fmt, ...)
+{
+	va_list args;
+
+	va_start(args, fmt);
+	vprintf(fmt, args);
+	va_end(args);
+}
+
+int main()
+{
+	union libsoccr_addr addr, dst;
+	int srv, sock, clnt, rst;
+	int ret, dsize;
+	socklen_t dst_let;
+	struct libsoccr_sk_data data = {};
+	struct libsoccr_sk *so, *so_rst;
+	char buf[11] = "0123456789", *queue;
+
+	libsoccr_set_log(10, pr_printf);
+
+	memset(&addr,0,sizeof(addr));
+
+#ifndef TEST_IPV6
+	addr.v4.sin_family = AF_INET;
+	inet_pton(AF_INET, "0.0.0.0", &(addr.v4.sin_addr));
+#else
+	addr.v6.sin6_family = AF_INET6;
+	inet_pton(AF_INET6, "::0", &(addr.v6.sin6_addr));
+#endif
+
+#ifndef TEST_IPV6
+	srv = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+#else
+	srv = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP);
+#endif
+	if (srv == -1) {
+		pr_perror("socket() failed");
+		return -1;
+	}
+
+#ifndef TEST_IPV6
+		addr.v4.sin_port = htons(8765);
+#else
+		addr.v6.sin6_port = htons(8765);
+#endif
+	ret = bind(srv, (struct sockaddr *) &addr, sizeof(addr));
+	if (ret == -1) {
+		pr_perror("bind() failed");
+		return -1;
+	}
+
+	if (listen(srv, 1) == -1) {
+		pr_perror("listen() failed");
+		return -1;
+	}
+
+#ifndef TEST_IPV6
+	clnt = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+#else
+	clnt = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP);
+#endif
+	if (clnt == -1) {
+		pr_perror("socket() failed");
+		return -1;
+	}
+
+	if (connect(clnt, (struct sockaddr *) &addr, sizeof(addr))) {
+		pr_perror("connect");
+		return 1;
+	}
+
+	dst_let = sizeof(dst);
+	sock = accept(srv, (struct sockaddr *) &dst, &dst_let);
+	if (sock < 0) {
+		pr_perror("accept");
+		return 1;
+	}
+
+	if (write(clnt, &buf, sizeof(buf)) != sizeof(buf)) {
+		pr_perror("write");
+		return 1;
+	}
+
+	/* Start testing */
+	dst_let = sizeof(addr);
+	if (getsockname(sock, (struct sockaddr *) &addr, &dst_let)) {
+		pr_perror("connect");
+		return 1;
+	}
+	dst_let = sizeof(addr);
+	if (getpeername(sock, (struct sockaddr *) &dst, &dst_let)) {
+		pr_perror("connect");
+		return 1;
+	}
+
+
+	so = libsoccr_pause(sock);
+
+	dsize = libsoccr_save(so, &data, sizeof(data));
+	if (dsize < 0) {
+		pr_perror("libsoccr_save");
+		return 1;
+	}
+
+#ifndef TEST_IPV6
+	rst = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
+#else
+	rst = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP);
+#endif
+	if (rst == -1) {
+		pr_perror("socket() failed");
+		return -1;
+	}
+	close(sock);
+
+	so_rst = libsoccr_pause(rst);
+	libsoccr_set_addr(so_rst, 1, &addr, 0);
+	libsoccr_set_addr(so_rst, 0, &dst, 0);
+
+	queue = libsoccr_get_queue_bytes(so, TCP_RECV_QUEUE, SOCCR_MEM_EXCL);
+	libsoccr_set_queue_bytes(so_rst, TCP_RECV_QUEUE, queue, SOCCR_MEM_EXCL);
+	queue = libsoccr_get_queue_bytes(so, TCP_SEND_QUEUE, SOCCR_MEM_EXCL);
+	libsoccr_set_queue_bytes(so_rst, TCP_SEND_QUEUE, queue, SOCCR_MEM_EXCL);
+
+	ret = libsoccr_restore(so_rst, &data, dsize);
+	if (ret) {
+		pr_perror("libsoccr_restore: %d", ret);
+		return 1;
+	}
+
+	libsoccr_resume(so_rst);
+	libsoccr_resume(so);
+
+	if (read(rst, &buf, sizeof(buf)) != sizeof(buf)) {
+		pr_perror("read");
+		return 1;
+	}
+
+	if (write(rst, &buf, sizeof(buf)) != sizeof(buf)) {
+		pr_perror("write");
+		return 1;
+	}
+	shutdown(rst, SHUT_WR);
+
+	if (read(clnt, &buf, sizeof(buf)) != sizeof(buf)) {
+		pr_perror("read");
+		return 1;
+	}
+
+	return 0;
+}
-- 
2.7.4