[CRIU] [PATCH 01/14] zdtm: test shared file tables
Andrey Vagin
avagin at openvz.org
Fri Jan 11 04:22:32 EST 2013
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
More information about the CRIU
mailing list