[CRIU] [PATCH 5/6] zdtm: Add file_own test case

[Cyrill Gorcunov]

Signed-off-by: Cyrill Gorcunov <gorcunov at openvz.org>
---
 test/zdtm.sh                      |    1 +
 test/zdtm/live/static/Makefile    |    1 +
 test/zdtm/live/static/file_fown.c |  138 +++++++++++++++++++++++++++++++++++++
 3 files changed, 140 insertions(+), 0 deletions(-)
 create mode 100644 test/zdtm/live/static/file_fown.c

diff --git a/test/zdtm.sh b/test/zdtm.sh
index 75b3576..3309186 100644
--- a/test/zdtm.sh
+++ b/test/zdtm.sh
@@ -35,6 +35,7 @@ static/cmdlinenv00
 static/socket_listen
 static/socket_udp
 static/selfexe00
+static/file_fown
 "
 
 UTS_TEST_LIST="
diff --git a/test/zdtm/live/static/Makefile b/test/zdtm/live/static/Makefile
index 3b36a2c..9560090 100644
--- a/test/zdtm/live/static/Makefile
+++ b/test/zdtm/live/static/Makefile
@@ -38,6 +38,7 @@ TST_NOFILE	=				\
 		selfexe00			\
 		sem				\
 		maps01				\
+		file_fown			\
 #		jobctl00			\
 
 TST_FILE	=				\
diff --git a/test/zdtm/live/static/file_fown.c b/test/zdtm/live/static/file_fown.c
new file mode 100644
index 0000000..e7ee6cc
--- /dev/null
+++ b/test/zdtm/live/static/file_fown.c
@@ -0,0 +1,138 @@
+#define _GNU_SOURCE         /* See feature_test_macros(7) */
+#include <unistd.h>
+
+#include <errno.h>
+#include <fcntl.h>
+#include <stdlib.h>
+#include <signal.h>
+#include <string.h>
+#include <utime.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <sys/wait.h>
+#include <sys/mman.h>
+
+#include "zdtmtst.h"
+
+#ifndef F_SETSIG
+#define F_SETSIG	10	/* for sockets. */
+#define F_GETSIG	11	/* for sockets. */
+#endif
+
+const char *test_doc	= "Check for signal delivery for file owners";
+const char *test_author	= "Cyrill Gorcunov <gorcunov at openvz.org>";
+
+static int received_io;
+
+#define MAP(map, i)		(((int *)map)[i])
+#define MAP_SYNC(map)		MAP(map, 0)
+#define MAP_PID(map)		MAP(map, 1)
+
+static void signal_handler_io(int status)
+{
+	received_io++;
+}
+
+int main(int argc, char ** argv)
+{
+	pid_t pid, ppid;
+	struct sigaction saio;
+	int status;
+	int pipes[2];
+	void *map;
+	uid_t ruid;
+	uid_t euid;
+	uid_t suid;
+
+	test_init(argc, argv);
+
+	if (getresuid(&ruid, &euid, &suid)) {
+		fail("getresuid failed");
+		exit(1);
+	}
+
+	map = mmap(NULL, 1024, PROT_READ | PROT_WRITE, MAP_SHARED | MAP_ANONYMOUS, -1, 0);
+	if (map == MAP_FAILED) {
+		fail("Can't map");
+		exit(1);
+	}
+
+	if (pipe(pipes)) {
+		err("Can't create pipes: %m\n");
+		exit(1);
+	}
+
+	memset(&saio, 0, sizeof(saio));
+	saio.sa_handler = (sig_t)signal_handler_io;
+	sigaction(SIGIO, &saio, 0);
+
+	if (setresuid(-1, 1, -1)) {
+		fail("setresuid failed");
+		exit(1);
+	}
+
+	fcntl(pipes[0], F_SETOWN, getpid());
+	fcntl(pipes[1], F_SETOWN, getpid());
+
+	test_msg("main owner pipes[0]: %d\n", fcntl(pipes[0], F_GETOWN));
+
+	fcntl(pipes[0], F_SETSIG, SIGIO);
+	fcntl(pipes[1], F_SETSIG, SIGIO);
+
+	fcntl(pipes[0], F_SETFL, fcntl(pipes[0], F_GETFL) | O_NONBLOCK | O_ASYNC);
+	fcntl(pipes[1], F_SETFL, fcntl(pipes[1], F_GETFL) | O_NONBLOCK | O_ASYNC);
+
+	if (setresuid(-1, euid, -1)) {
+		fail("setresuid failed");
+		exit(1);
+	}
+
+	ppid = getpid();
+
+	pid = test_fork();
+
+	if (pid < 0) {
+		err("can't fork %m");
+		exit(1);
+	}
+
+	MAP_SYNC(map) = 0;
+
+	if (pid == 0) {
+		int v = 1;
+
+		write(pipes[1], &v, sizeof(v));
+		read(pipes[0], &v, sizeof(v));
+
+		MAP_SYNC(map) = 1;
+
+		while (MAP_SYNC(map) != 3)
+			sleep(1);
+
+		write(pipes[1], &v, sizeof(v));
+		read(pipes[0], &v, sizeof(v));
+
+		MAP_PID(map) = fcntl(pipes[0], F_GETOWN);
+
+		exit(0);
+	}
+
+	while (MAP_SYNC(map) != 1)
+		sleep(1);
+
+	test_daemon();
+	test_waitsig();
+
+	MAP_SYNC(map) = 3;
+
+	waitpid(pid, &status, P_ALL);
+
+	if (received_io < 1 || MAP_PID(map) != ppid) {
+		fail("received_io = %d ppid: %d  MAP_PID(map): %d\n",
+		     received_io, ppid, MAP_PID(map));
+		exit(1);
+	}
+
+	pass();
+	return 0;
+}
-- 
1.7.7.6