[CRIU] [PATCH v5 19/19] zdtm/static: add a test to check epoll file descriptors

[Cyrill Gorcunov]

From: Andrei Vagin <avagin at virtuozzo.com>

Signed-off-by: Andrei Vagin <avagin at virtuozzo.com>
---
 test/zdtm/static/Makefile   |   1 +
 test/zdtm/static/epoll.c    | 137 ++++++++++++++++++++++++++++++++++++++++++++
 test/zdtm/static/epoll.desc |   1 +
 3 files changed, 139 insertions(+)
 create mode 100644 test/zdtm/static/epoll.c
 create mode 100644 test/zdtm/static/epoll.desc

diff --git a/test/zdtm/static/Makefile b/test/zdtm/static/Makefile
index d61ef33b9fe5..4c32f991fb56 100644
--- a/test/zdtm/static/Makefile
+++ b/test/zdtm/static/Makefile
@@ -121,6 +121,7 @@ TST_NOFILE	:=				\
 		proc-self			\
 		proc-self01			\
 		eventfs00			\
+		epoll				\
 		signalfd00			\
 		inotify_irmap			\
 		fanotify00			\
diff --git a/test/zdtm/static/epoll.c b/test/zdtm/static/epoll.c
new file mode 100644
index 000000000000..07a01042da7f
--- /dev/null
+++ b/test/zdtm/static/epoll.c
@@ -0,0 +1,137 @@
+#include <unistd.h>
+
+#include <errno.h>
+#include <fcntl.h>
+#include <stdlib.h>
+#include <signal.h>
+#include <string.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <sys/wait.h>
+#include <sys/eventfd.h>
+#include <sys/ioctl.h>
+#include <sys/epoll.h>
+#include <sys/types.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <stdint.h>
+#include <inttypes.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 eventfs";
+const char *test_author	= "Cyrill Gorcunov <gorcunov at openvz.org>";
+
+#define EVENTFD_INITIAL	30
+#define EVENTFD_FINAL1	90
+#define EVENTFD_FINAL2	(EVENTFD_FINAL1 * 2)
+#define DUPFDNO		999
+
+int main(int argc, char *argv[])
+{
+	uint64_t v = EVENTFD_INITIAL;
+	int epollfd1, epollfd2, fd;
+	struct epoll_event ev;
+	int i, ret;
+
+	struct {
+		int	pipefd[2];
+	} pipes[10];
+
+	test_init(argc, argv);
+
+	epollfd1 = epoll_create(1);
+	if (epollfd1 < 0) {
+		pr_perror("epoll_create failed");
+		exit(1);
+	}
+	epollfd2 = epoll_create(1);
+	if (epollfd2 < 0) {
+		pr_perror("epoll_create failed");
+		exit(1);
+	}
+
+	test_msg("created eventfd with %lld\n", (long long)v);
+
+	memset(&ev, 0, sizeof(ev));
+	ev.events = EPOLLIN | EPOLLOUT;
+
+	for (i = 0; i < ARRAY_SIZE(pipes); i++) {
+		if (pipe(pipes[i].pipefd)) {
+			pr_err("Can't create pipe %d\n", i);
+			exit(1);
+		}
+
+		ev.data.u64 = i;
+		fd = dup2(pipes[i].pipefd[0], DUPFDNO);
+		if (fd < 0) {
+			pr_perror("Can't dup %d to %d", pipes[i].pipefd[0], DUPFDNO);
+			exit(1);
+		}
+
+		test_msg("epoll %d add %d native\n", epollfd1, pipes[i].pipefd[0]);
+		if (epoll_ctl(epollfd1, EPOLL_CTL_ADD, pipes[i].pipefd[0], &ev)) {
+			pr_perror("Can't add pipe %d", pipes[i].pipefd[0]);
+			exit(1);
+		}
+
+		test_msg("epoll %d add %d dup'ed from %d\n", epollfd1, fd, pipes[i].pipefd[0]);
+		if (epoll_ctl(epollfd2, EPOLL_CTL_ADD, fd, &ev)) {
+			pr_perror("Can't add pipe %d", fd);
+			exit(1);
+		}
+
+		close(fd);
+		test_msg("epoll source %d closed\n", fd);
+	}
+
+	test_daemon();
+	test_waitsig();
+
+	ret = 0;
+	for (i = 0; i < ARRAY_SIZE(pipes); i++) {
+		uint8_t cw = 1, cr;
+
+		if (write(pipes[i].pipefd[1], &cw, sizeof(cw)) != sizeof(cw)) {
+			pr_perror("Unable to write into a pipe\n");
+			return 1;
+		}
+
+		if (epoll_wait(epollfd1, &ev, 1, -1) != 1) {
+			pr_perror("Unable to wain events");
+			return 1;
+		}
+		if (ev.data.u64 != i) {
+			pr_err("ev.fd=%d ev.data.u64=%#llx (%d expected)\n",
+			       ev.data.fd, (long long)ev.data.u64, i);
+			ret |= 1;
+		}
+
+		if (epoll_wait(epollfd2, &ev, 1, -1) != 1) {
+			pr_perror("Unable to wain events");
+			return 1;
+		}
+		if (ev.data.u64 != i) {
+			pr_err("ev.fd=%d ev.data.u64=%#llx (%d expected)\n",
+			       ev.data.fd, (long long)ev.data.u64, i);
+			ret |= 1;
+		}
+
+		if (read(pipes[i].pipefd[0], &cr, sizeof(cr)) != sizeof(cr)) {
+			pr_perror("read");
+			return 1;
+		}
+	}
+
+	if (ret)
+		return 1;
+
+	pass();
+	return 0;
+}
diff --git a/test/zdtm/static/epoll.desc b/test/zdtm/static/epoll.desc
new file mode 100644
index 000000000000..72cc898d5f34
--- /dev/null
+++ b/test/zdtm/static/epoll.desc
@@ -0,0 +1 @@
+{ 'feature' : 'kcmp_epoll' }
-- 
2.14.4