From bf416aeca35a71adc2f2551d1b938a0ded230a2f Mon Sep 17 00:00:00 2001
From: Masatake YAMATO <yamato@redhat.com>
Date: Sun, 28 May 2023 03:20:37 +0900
Subject: [PATCH] tests: (mkfds) add a factory to make an eventpoll fd

Signed-off-by: Masatake YAMATO <yamato@redhat.com>
---
 tests/helpers/test_mkfds.c | 101 +++++++++++++++++++++++++++++++++++++
 1 file changed, 101 insertions(+)

diff --git a/tests/helpers/test_mkfds.c b/tests/helpers/test_mkfds.c
index 9377d28a9e..5350850b21 100644
--- a/tests/helpers/test_mkfds.c
+++ b/tests/helpers/test_mkfds.c
@@ -37,6 +37,7 @@
 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
+#include <sys/epoll.h>
 #include <sys/eventfd.h>
 #include <sys/inotify.h>
 #include <sys/ioctl.h>
@@ -2225,6 +2226,95 @@ static void free_sysvshm(const struct factory *factory _U_, void *data)
 	shmctl(sysvshm_data->id, IPC_RMID, NULL);
 }
 
+static void *make_eventpoll(const struct factory *factory _U_, struct fdesc fdescs[] _U_,
+			    int argc _U_, char ** argv _U_)
+{
+	int efd;
+	struct spec {
+		const char *file;
+		int flag;
+		uint32_t events;
+	} specs [] = {
+		{
+			.file = "DUMMY, DONT'USE THIS"
+		}, {
+			.file = "/dev/random",
+			.flag = O_RDONLY,
+			.events = EPOLLIN,
+		}, {
+			.file = "/dev/random",
+			.flag = O_WRONLY,
+			.events = EPOLLOUT,
+		},
+	};
+
+	efd = epoll_create(1);
+	if (efd < 0)
+		err(EXIT_FAILURE, "failed in epoll_create(2)");
+	if (efd != fdescs[0].fd) {
+		if (dup2(efd, fdescs[0].fd) < 0) {
+			int e = errno;
+			close(efd);
+			errno = e;
+			err(EXIT_FAILURE, "failed to dup %d -> %d", efd, fdescs[0].fd);
+		}
+		close(efd);
+		efd = fdescs[0].fd;
+	}
+	fdescs[0] = (struct fdesc){
+		.fd    = fdescs[0].fd,
+		.close = close_fdesc,
+		.data  = NULL
+	};
+
+	for (size_t i = 1; i < ARRAY_SIZE(specs); i++) {
+		int fd = open(specs[i].file, specs[i].flag);
+		if (fd < 0) {
+			int e = errno;
+			close(efd);
+			for (size_t j = i - 1; j > 0; j--)
+				close(fdescs[j].fd);
+			errno = e;
+			err(EXIT_FAILURE, "failed in open(\"%s\",...)",
+			    specs[i].file);
+		}
+		if (fd != fdescs[i].fd) {
+			if (dup2(fd, fdescs[i].fd) < 0) {
+				int e = errno;
+				close(efd);
+				for (size_t j = i - 1; j > 0; j--)
+					close(fdescs[j].fd);
+				close(fd);
+				errno = e;
+				err(EXIT_FAILURE, "failed to dup %d -> %d",
+				    fd, fdescs[i].fd);
+			}
+			close(fd);
+		}
+		fdescs[i] = (struct fdesc) {
+			.fd = fdescs[i].fd,
+			.close = close_fdesc,
+			.data = NULL
+		};
+		if (epoll_ctl(efd, EPOLL_CTL_ADD, fdescs[i].fd,
+			      &(struct epoll_event) {
+				      .events = specs[i].events,
+				      .data = {.ptr = NULL,}
+			      }) < 0) {
+			int e = errno;
+			close(efd);
+			for (size_t j = i; j > 0; j--)
+				close(fdescs[j].fd);
+			errno = e;
+			err (EXIT_FAILURE,
+			     "failed to add fd %d to the eventpoll fd with epoll_ctl",
+			     fdescs[i].fd);
+		}
+	}
+
+	return NULL;
+}
+
 #define PARAM_END { .name = NULL, }
 static const struct factory factories[] = {
 	{
@@ -2834,6 +2924,17 @@ static const struct factory factories[] = {
 			PARAM_END
 		},
 	},
+	{
+		.name = "eventpoll",
+		.desc = "make eventpoll (epoll) file",
+		.priv = false,
+		.N    = 3,
+		.EX_N = 0,
+		.make = make_eventpoll,
+		.params = (struct parameter []) {
+			PARAM_END
+		}
+	},
 };
 
 static int count_parameters(const struct factory *factory)
-- 
2.47.3