[CRIU] [PATCH 2/2] test: Test how criu_dump_iters from library works

[Pavel Emelyanov]

Signed-off-by: Pavel Emelyanov <xemul at parallels.com>
---
 test/libcriu/Makefile     |   1 +
 test/libcriu/run.sh       |   1 +
 test/libcriu/test_iters.c | 141 ++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 143 insertions(+)
 create mode 100644 test/libcriu/test_iters.c

diff --git a/test/libcriu/Makefile b/test/libcriu/Makefile
index 277e2d6..e6d0eb0 100644
--- a/test/libcriu/Makefile
+++ b/test/libcriu/Makefile
@@ -1,6 +1,7 @@
 TESTS += test_sub 
 TESTS += test_self
 TESTS += test_notify
+TESTS += test_iters
 
 all: $(TESTS)
 
diff --git a/test/libcriu/run.sh b/test/libcriu/run.sh
index 061dd93..4b97eef 100755
--- a/test/libcriu/run.sh
+++ b/test/libcriu/run.sh
@@ -38,6 +38,7 @@ function run_test {
 run_test test_sub
 run_test test_self
 run_test test_notify
+run_test test_iters
 
 echo "== Stopping service"
 kill -TERM $(cat wdir/s/pidfile)
diff --git a/test/libcriu/test_iters.c b/test/libcriu/test_iters.c
new file mode 100644
index 0000000..c2c674b
--- /dev/null
+++ b/test/libcriu/test_iters.c
@@ -0,0 +1,141 @@
+#include "criu.h"
+#include <fcntl.h>
+#include <stdio.h>
+#include <errno.h>
+#include <stdlib.h>
+#include <signal.h>
+#include <fcntl.h>
+#include <sys/stat.h>
+#include "lib.h"
+
+static int wdir_fd, cur_iter = 1, cur_imgdir = -1;
+
+static int stop = 0;
+static void sh(int sig)
+{
+	stop = 1;
+}
+
+static int open_imgdir(void)
+{
+	char p[10];
+
+	sprintf(p, "%d", cur_iter);
+	mkdirat(wdir_fd, p, 0700);
+	cur_imgdir = openat(wdir_fd, p, O_DIRECTORY);
+	criu_set_images_dir_fd(cur_imgdir);
+}
+
+#define MAX_ITERS	2
+
+static int next_iter(criu_predump_info pi)
+{
+	char p[10];
+
+	printf("   `- %d iter over\n", cur_iter);
+
+	close(cur_imgdir);
+	sprintf(p, "../%d", cur_iter);
+	criu_set_parent_images(p);
+
+	cur_iter++;
+	open_imgdir();
+
+	return cur_iter < MAX_ITERS;
+}
+
+#define SUCC_ECODE	42
+
+int main(int argc, char **argv)
+{
+	int pid, ret, p[2];
+
+	wdir_fd = open(argv[2], O_DIRECTORY);
+	if (wdir_fd < 0) {
+		perror("Can't open wdir");
+		return 1;
+	}
+
+	printf("--- Start loop ---\n");
+	pipe(p);
+	pid = fork();
+	if (pid < 0) {
+		perror("Can't");
+		return -1;
+	}
+
+	if (!pid) {
+		printf("   `- loop: initializing\n");
+		if (setsid() < 0)
+			exit(1);
+		if (signal(SIGUSR1, sh) == SIG_ERR)
+			exit(1);
+
+		close(0);
+		close(1);
+		close(2);
+		close(p[0]);
+
+		ret = SUCC_ECODE;
+		write(p[1], &ret, sizeof(ret));
+		close(p[1]);
+
+		while (!stop)
+			sleep(1);
+		exit(SUCC_ECODE);
+	}
+
+	close(p[1]);
+
+	/* Wait for kid to start */
+	ret = -1;
+	read(p[0], &ret, sizeof(ret));
+	if (ret != SUCC_ECODE) {
+		printf("Error starting loop\n");
+		goto err;
+	}
+
+	/* Wait for pipe to get closed, then dump */
+	read(p[0], &ret, 1);
+	close(p[0]);
+
+	printf("--- Dump loop ---\n");
+	criu_init_opts();
+	criu_set_service_address(argv[1]);
+	criu_set_pid(pid);
+	criu_set_log_file("dump.log");
+	criu_set_log_level(4);
+
+	open_imgdir();
+	ret = criu_dump_iters(next_iter);
+	if (ret < 0) {
+		what_err_ret_mean(ret);
+		kill(pid, SIGKILL);
+		goto err;
+	}
+
+	printf("   `- Dump succeeded\n");
+	wait(pid, NULL, 0);
+
+	printf("--- Restore loop ---\n");
+	criu_init_opts();
+	criu_set_log_level(4);
+	criu_set_log_file("restore.log");
+	criu_set_images_dir_fd(cur_imgdir);
+
+	pid = criu_restore_child();
+	if (pid <= 0) {
+		what_err_ret_mean(pid);
+		return -1;
+	}
+
+	printf("   `- Restore returned pid %d\n", pid);
+	kill(pid, SIGUSR1);
+err:
+	if (waitpid(pid, &ret, 0) < 0) {
+		perror("   Can't wait kid");
+		return -1;
+	}
+
+	return chk_exit(ret, SUCC_ECODE);
+}
-- 
1.8.4.2