1 files changed, 85 insertions, 0 deletions
diff --git a/tests/helpers/test_mkfds.h b/tests/helpers/test_mkfds.h
new file mode 100644
index 0000000..7d679b4
--- /dev/null
+++ b/tests/helpers/test_mkfds.h
@@ -0,0 +1,85 @@
+/*
+ * test_mkfds - make various file descriptors
+ *
+ * Written by Masatake YAMATO <yamato@redhat.com>
+ *
+ * This program is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU General Public License as
+ * published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it would be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+
+#ifndef TEST_MKFDS_H
+#define TEST_MKFDS_H
+
+#include <asm/unistd.h>
+#include <stdbool.h>
+#include <stddef.h>
+
+enum multiplexing_mode {
+   MX_READ   = 1 << 0,
+   MX_WRITE  = 1 << 1,
+   MX_EXCEPT = 1 << 2,
+};
+
+struct fdesc {
+	int fd;
+	void (*close)(int, void *);
+	unsigned int mx_modes;
+	void *data;
+};
+
+#define DEFUN_WAIT_EVENT_POLL(NAME,SYSCALL,XDECLS,SETUP_SIG_HANDLER,SYSCALL_INVOCATION) \
+	void wait_event_##NAME(bool add_stdin, struct fdesc *fdescs, size_t n_fdescs) \
+	{								\
+		int n = add_stdin? 1: 0;				\
+		int n0 = 0;						\
+		struct pollfd *pfds = NULL;				\
+									\
+		XDECLS							\
+									\
+		for (size_t i = 0; i < n_fdescs; i++)			\
+			if (fdescs[i].mx_modes)				\
+				n++;					\
+									\
+		pfds = xcalloc(n, sizeof(pfds[0]));			\
+									\
+		for (size_t i = 0; i < n_fdescs; i++) {			\
+			if (!fdescs[i].mx_modes)			\
+				continue;				\
+			pfds[n0].fd = fdescs[i].fd;			\
+			if (fdescs[i].mx_modes & MX_READ)		\
+				pfds[n0].events |= POLLIN;		\
+			if (fdescs[i].mx_modes & MX_WRITE)		\
+				pfds[n0].events |= POLLOUT;		\
+			if (fdescs[i].mx_modes & MX_EXCEPT)		\
+				pfds[n0].events |= POLLHUP;		\
+			n0++;						\
+		}							\
+									\
+		if (add_stdin) {					\
+			pfds[n0].fd = 0;				\
+			pfds[n0].events |= POLLIN;			\
+		}							\
+									\
+		SETUP_SIG_HANDLER					\
+									\
+		if (SYSCALL_INVOCATION < 0				\
+		    && errno != EINTR)					\
+			err(EXIT_FAILURE, "failed in " SYSCALL);	\
+		free(pfds);						\
+	}
+
+#ifdef __NR_ppoll
+void wait_event_ppoll(bool add_stdin, struct fdesc *fdescs, size_t n_fdescs);
+#endif
+
+#endif	/* TEST_MKFDS_H */