[CRIU] [PATCH 01/15] zdtm: test shared file tables

[Andrey Vagin]

Signed-off-by: Andrey Vagin <avagin at openvz.org>
---
 test/zdtm/live/static/Makefile     |   1 +
 test/zdtm/live/static/fdt_shared.c | 170 +++++++++++++++++++++++++++++++++++++
 2 files changed, 171 insertions(+)
 create mode 100644 test/zdtm/live/static/fdt_shared.c

diff --git a/test/zdtm/live/static/Makefile b/test/zdtm/live/static/Makefile
index 55555ab..89b8cad 100644
--- a/test/zdtm/live/static/Makefile
+++ b/test/zdtm/live/static/Makefile
@@ -109,6 +109,7 @@ TST_FILE	=				\
 		unlink_fifo_wronly		\
 		file_shared			\
 		cow01				\
+		fdt_shared			\
 
 TST_DIR		=				\
 		cwd00				\
diff --git a/test/zdtm/live/static/fdt_shared.c b/test/zdtm/live/static/fdt_shared.c
new file mode 100644
index 0000000..35555a6
--- /dev/null
+++ b/test/zdtm/live/static/fdt_shared.c
@@ -0,0 +1,170 @@
+#define _GNU_SOURCE             /* See feature_test_macros(7) */
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <unistd.h>
+#include <sched.h>
+#include <signal.h>
+#include <sys/wait.h>
+
+#include "zdtmtst.h"
+
+const char *test_doc	= "Check a shared file descriptor table.";
+const char *test_author	= "Andrew Vagin <avagin at openvz.org>";
+
+char *filename;
+TEST_OPTION(filename, string, "file name", 1);
+
+#define STACK_SIZE 4096
+#define TEST_FD 128
+#define TEST_STRING "Hello World!"
+
+static pid_t clone_child(int (*fn)(void *), int flags)
+{
+	char stack[STACK_SIZE];
+	pid_t pid;
+
+	pid = clone(fn, stack + STACK_SIZE,
+			flags | SIGCHLD, NULL);
+	if (pid == -1) {
+		err("Unable to clone a new process\n");
+		return -1;
+	}
+
+	return pid;
+}
+
+static int child2(void *_arg)
+{
+	char buf[10];
+
+	test_waitsig();
+
+	if (read(TEST_FD, buf, sizeof(TEST_STRING)) != sizeof(TEST_STRING)) {
+		err("Unable to read from %d\n", TEST_FD);
+		return 1;
+	}
+
+	return 0;
+}
+
+static int child3(void *_arg)
+{
+	test_waitsig();
+
+	if (close(TEST_FD) != -1) {
+		fail("%d is exist\n", TEST_FD);
+		return 1;
+	}
+
+	return 0;
+}
+
+static int child(void *_arg)
+{
+	char buf[10];
+	pid_t pid, pid2;
+	int status;
+
+	pid = clone_child(child2, CLONE_FILES);
+	if (pid < 0)
+		return 1;
+
+	pid2 = clone_child(child3, 0);
+	if (pid < 0)
+		return 1;
+
+	test_waitsig();
+
+	kill(pid2, SIGTERM);
+	kill(pid, SIGTERM);
+	waitpid(pid2, &status, 0);
+
+	if (status) {
+		fail("The child3 returned %d\n", status);
+		return 1;
+	}
+
+	waitpid(pid, &status, 0);
+
+	if (status) {
+		fail("The child2 returned %d\n", status);
+		return 1;
+	}
+
+	if (read(TEST_FD, buf, sizeof(TEST_STRING)) != sizeof(TEST_STRING)) {
+		err("Unable to read from %d\n", TEST_FD);
+		return 1;
+	}
+
+	if (close(TEST_FD) == -1) {
+		err("Unable to close(%d)\n", TEST_FD);
+		return 1;
+	}
+
+	return 0;
+}
+
+int main(int argc, char ** argv)
+{
+	int status;
+	pid_t pid, pid2;
+	int fd, i;
+
+	test_init(argc, argv);
+
+	pid = clone_child(child, CLONE_FILES);
+	if (pid < 0)
+		return 1;
+
+	pid2 = clone_child(child2, CLONE_FILES);
+	if (pid2 < 0)
+		return 1;
+
+	test_daemon();
+	test_waitsig();
+
+	fd = open(filename, O_RDWR | O_CREAT, 0666);
+	if (fd == -1) {
+		err("Can't open /dev/zero\n");
+		return -1;
+	}
+
+	for (i = 0; i < 3; i++)
+		if (write(fd, TEST_STRING, sizeof(TEST_STRING)) != sizeof(TEST_STRING)) {
+			err("Unable to write a test string\n");
+			return -1;
+		}
+
+	fd = dup2(fd, TEST_FD);
+	if (fd == -1) {
+		err("Can't dup fd to %d\n", fd, TEST_FD);
+		return -1;
+	}
+
+	lseek(fd, 0, SEEK_SET);
+
+	kill(pid2, SIGTERM);
+	waitpid(pid2, &status, 0);
+	kill(pid, SIGTERM);
+
+	if (status) {
+		fail("The child returned %d\n", status);
+		return 1;
+	}
+
+	waitpid(pid, &status, 0);
+	if (status) {
+		fail("The child returned %d\n", status);
+		return 1;
+	}
+
+	if (close(TEST_FD) == 0) {
+		fail("%d was not closed\n", TEST_FD);
+		return 1;
+	}
+
+	pass();
+
+	return 0;
+}
-- 
1.7.11.7