From bdbd7c8f14ddebd4fbc5e1bc77f9dcb17831c966 Mon Sep 17 00:00:00 2001
From: Andrei Vagin <avagin@virtuozzo.com>
Date: Wed, 4 Jul 2018 18:51:47 +0300
Subject: [PATCH] zdtm/static: add a test to check epoll file descriptors

Signed-off-by: Andrei Vagin <avagin@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 28c6a1aa0..980d8c437 100644
--- a/test/zdtm/static/Makefile
+++ b/test/zdtm/static/Makefile
@@ -120,6 +120,7 @@ TST_NOFILE	:=				\
 		file_fown			\
 		proc-self			\
 		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 000000000..ce3b176d3
--- /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 <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"
+
+const char *test_doc	= "Check for epoll";
+const char *test_author	= "Andrei Vagin <avagin@openvz.org>";
+
+#define DUPFDNO		999
+
+int main(int argc, char *argv[])
+{
+	int epollfd1, epollfd2, fd;
+	struct epoll_event ev;
+	int i, ret;
+
+	struct {
+		int	pipefd[2];
+	} pipes[250];
+
+	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);
+	}
+
+	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);
+		}
+
+		if (i % 2) {
+			int nfd;
+
+			nfd = dup2(pipes[i].pipefd[0], i + 700);
+			if (nfd < 0) {
+				pr_err("dup2");
+				exit(1);
+			}
+			close(pipes[i].pipefd[0]);
+			pipes[i].pipefd[0] = nfd;
+		}
+
+		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 000000000..72cc898d5
--- /dev/null
+++ b/test/zdtm/static/epoll.desc
@@ -0,0 +1 @@
+{ 'feature' : 'kcmp_epoll' }