add pipe kqueue tests

4 files changed, 319 insertions, 3 deletions

diff --git a/regress/sys/kern/pipe/Makefile b/regress/sys/kern/pipe/Makefile
index 8b209256a1e..ea5a76b613d 100644
--- a/regress/sys/kern/pipe/Makefile
+++ b/regress/sys/kern/pipe/Makefile
@@ -1,8 +1,9 @@
-#	$OpenBSD: Makefile,v 1.1 2019/11/09 21:10:15 anton Exp $
+#	$OpenBSD: Makefile,v 1.2 2019/12/24 09:37:53 anton Exp $
 
 PROG=		pipe
 
 SRCS+=		pipe.c
+SRCS+=		test-kqueue.c
 SRCS+=		test-ping-pong.c
 SRCS+=		test-run-down.c
 SRCS+=		test-thundering-herd.c
@@ -12,6 +13,10 @@ DPADD+=		${LIBPTHREAD}
 
 WARNINGS=	yes
 
+TESTS+=	kqueue-read
+TESTS+=	kqueue-read-eof
+TESTS+=	kqueue-write
+TESTS+=	kqueue-write-eof
 TESTS+=	ping-pong
 TESTS+=	run-down-write-big
 TESTS+=	run-down-write-small
diff --git a/regress/sys/kern/pipe/pipe.c b/regress/sys/kern/pipe/pipe.c
index 54d75511c2c..fbb6ef52230 100644
--- a/regress/sys/kern/pipe/pipe.c
+++ b/regress/sys/kern/pipe/pipe.c
@@ -1,4 +1,4 @@
-/*	$OpenBSD: pipe.c,v 1.2 2019/11/10 09:34:40 anton Exp $	*/
+/*	$OpenBSD: pipe.c,v 1.3 2019/12/24 09:37:53 anton Exp $	*/
 
 /*
  * Copyright (c) 2019 Anton Lindqvist <anton@openbsd.org>
@@ -39,6 +39,10 @@ main(int argc, char *argv[])
 		const char *t_name;
 		int (*t_fn)(void);
 	} tests[] = {
+		{ "kqueue-read",			test_kqueue_read },
+		{ "kqueue-read-eof",			test_kqueue_read_eof },
+		{ "kqueue-write",			test_kqueue_write },
+		{ "kqueue-write-eof",			test_kqueue_write_eof },
 		{ "ping-pong",				test_ping_pong },
 		{ "run-down-write-big",			test_run_down_write_big },
 		{ "run-down-write-small",		test_run_down_write_small },
diff --git a/regress/sys/kern/pipe/pipe.h b/regress/sys/kern/pipe/pipe.h
index 7c18fbab2c4..0c88e86addd 100644
--- a/regress/sys/kern/pipe/pipe.h
+++ b/regress/sys/kern/pipe/pipe.h
@@ -1,4 +1,4 @@
-/*	$OpenBSD: pipe.h,v 1.1 2019/11/09 21:10:15 anton Exp $	*/
+/*	$OpenBSD: pipe.h,v 1.2 2019/12/24 09:37:53 anton Exp $	*/
 
 /*
  * Copyright (c) 2019 Anton Lindqvist <anton@openbsd.org>
@@ -20,6 +20,10 @@
 
 #include <signal.h>	/* sig_atomic_t */
 
+int test_kqueue_read(void);
+int test_kqueue_read_eof(void);
+int test_kqueue_write(void);
+int test_kqueue_write_eof(void);
 int test_ping_pong(void);
 int test_run_down_write_big(void);
 int test_run_down_write_small(void);
diff --git a/regress/sys/kern/pipe/test-kqueue.c b/regress/sys/kern/pipe/test-kqueue.c
new file mode 100644
index 00000000000..3fdc5f6265c
--- /dev/null
+++ b/regress/sys/kern/pipe/test-kqueue.c
@@ -0,0 +1,303 @@
+/*	$OpenBSD: test-kqueue.c,v 1.1 2019/12/24 09:37:53 anton Exp $	*/
+
+/*
+ * Copyright (c) 2019 Anton Lindqvist <anton@openbsd.org>
+ *
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+
+#include <sys/types.h>
+#include <sys/event.h>
+#include <sys/time.h>
+
+#include <err.h>
+#include <pthread.h>
+#include <stdlib.h>
+#include <unistd.h>
+
+#include "pipe.h"
+
+enum kqueue_mode {
+	KQUEUE_READ,
+	KQUEUE_READ_EOF,
+	KQUEUE_WRITE,
+	KQUEUE_WRITE_EOF,
+};
+
+struct context {
+	enum kqueue_mode c_mode;
+	int c_alive;
+
+	int c_pipe[2];
+	int c_kq;
+
+	char *c_buf;
+	size_t c_bufsiz;
+
+	pthread_t c_th;
+	pthread_mutex_t c_mtx;
+};
+
+static void ctx_setup(struct context *, enum kqueue_mode);
+static void ctx_teardown(struct context *);
+static int ctx_thread_alive(struct context *);
+static void ctx_thread_start(struct context *);
+static void ctx_lock(struct context *);
+static void ctx_unlock(struct context *);
+
+static void *kqueue_thread(void *);
+
+/*
+ * Verify kqueue read event.
+ */
+int
+test_kqueue_read(void)
+{
+	struct context ctx;
+
+	ctx_setup(&ctx, KQUEUE_READ);
+	ctx_thread_start(&ctx);
+
+	while (ctx_thread_alive(&ctx)) {
+		ssize_t n;
+		unsigned char c = 'r';
+
+		n = write(ctx.c_pipe[1], &c, 1);
+		if (n == -1)
+			err(1, "write");
+		if (n != 1)
+			errx(1, "write: %ld != 1", n);
+	}
+
+	ctx_teardown(&ctx);
+
+	return 0;
+}
+
+/*
+ * Verify kqueue read EOF event.
+ */
+int
+test_kqueue_read_eof(void)
+{
+	struct context ctx;
+
+	ctx_setup(&ctx, KQUEUE_READ_EOF);
+	ctx_thread_start(&ctx);
+
+	while (ctx_thread_alive(&ctx)) {
+		if (ctx.c_pipe[1] == -1)
+			continue;
+
+		close(ctx.c_pipe[1]);
+		ctx.c_pipe[1] = -1;
+	}
+
+	ctx_teardown(&ctx);
+
+	return 0;
+}
+
+/*
+ * Verify kqueue write event.
+ */
+int
+test_kqueue_write(void)
+{
+	struct context ctx;
+	ssize_t n;
+
+	ctx_setup(&ctx, KQUEUE_WRITE);
+
+	n = write(ctx.c_pipe[1], ctx.c_buf, ctx.c_bufsiz);
+	if (n == -1)
+		err(1, "write");
+	if ((size_t)n != ctx.c_bufsiz)
+		errx(1, "write: %ld != %zu", n, ctx.c_bufsiz);
+
+	ctx_thread_start(&ctx);
+
+	while (ctx_thread_alive(&ctx)) {
+		unsigned char c;
+
+		n = read(ctx.c_pipe[0], &c, 1);
+		if (n == -1)
+			err(1, "read");
+		if (n != 1)
+			errx(1, "read: %ld != 1", n);
+	}
+
+	ctx_teardown(&ctx);
+
+	return 0;
+}
+
+/*
+ * XXX Verify kqueue write event.
+ */
+int
+test_kqueue_write_eof(void)
+{
+
+	return 0;
+}
+
+static void
+ctx_setup(struct context *ctx, enum kqueue_mode mode)
+{
+	int error;
+
+	ctx->c_mode = mode;
+	ctx->c_alive = 1;
+
+	if (pipe(ctx->c_pipe) == -1)
+		err(1, "pipe");
+
+	ctx->c_kq = kqueue();
+	if (ctx->c_kq == -1)
+		err(1, "kqueue");
+
+	ctx->c_bufsiz = PIPE_SIZE;
+	ctx->c_buf = malloc(ctx->c_bufsiz);
+	if (ctx->c_buf == NULL)
+		err(1, NULL);
+
+	error = pthread_mutex_init(&ctx->c_mtx, NULL);
+	if (error)
+		errc(1, error, "pthread_mutex_init");
+}
+
+static void
+ctx_teardown(struct context *ctx)
+{
+	int error;
+
+	error = pthread_join(ctx->c_th, NULL);
+	if (error)
+		errc(1, error, "pthread_join");
+
+	error = pthread_mutex_destroy(&ctx->c_mtx);
+	if (error)
+		errc(1, error, "pthread_mutex_destroy");
+
+	free(ctx->c_buf);
+
+	close(ctx->c_pipe[0]);
+	close(ctx->c_pipe[1]);
+	close(ctx->c_kq);
+
+}
+
+static int
+ctx_thread_alive(struct context *ctx)
+{
+	int alive;
+
+	ctx_lock(ctx);
+	alive = ctx->c_alive;
+	ctx_unlock(ctx);
+	return alive;
+}
+
+static void
+ctx_thread_start(struct context *ctx)
+{
+	int error;
+
+	error = pthread_create(&ctx->c_th, NULL, kqueue_thread, ctx);
+	if (error)
+		errc(1, error, "pthread_create");
+}
+
+static void
+ctx_lock(struct context *ctx)
+{
+	int error;
+
+	error = pthread_mutex_lock(&ctx->c_mtx);
+	if (error)
+		errc(1, error, "pthread_mutex_lock");
+}
+
+static void ctx_unlock(struct context *ctx)
+{
+	int error;
+
+	error = pthread_mutex_unlock(&ctx->c_mtx);
+	if (error)
+		errc(1, error, "pthread_mutex_unlock");
+}
+
+static void *
+kqueue_thread(void *arg)
+{
+	struct context *ctx = arg;
+	struct kevent kev;
+	int fd, filter, nevents;
+
+	switch (ctx->c_mode) {
+	case KQUEUE_READ:
+	case KQUEUE_READ_EOF:
+		fd = ctx->c_pipe[0];
+		filter = EVFILT_READ;
+		break;
+	case KQUEUE_WRITE:
+	case KQUEUE_WRITE_EOF:
+		fd = ctx->c_pipe[1];
+		filter = EVFILT_WRITE;
+		break;
+	}
+
+	EV_SET(&kev, fd, filter, EV_ADD, 0, 0, NULL);
+	nevents = kevent(ctx->c_kq, &kev, 1, NULL, 0, NULL);
+	if (nevents == -1)
+		err(1, "kevent");
+	nevents = kevent(ctx->c_kq, NULL, 0, &kev, 1, NULL);
+	if (nevents == -1)
+		err(1, "kevent");
+	if (nevents != 1)
+		errx(1, "kevent: %d != 1", nevents);
+
+	switch (ctx->c_mode) {
+	case KQUEUE_READ:
+	case KQUEUE_READ_EOF:
+		if ((int)kev.ident != ctx->c_pipe[0])
+			errx(1, "kevent: ident");
+		if (kev.filter != EVFILT_READ)
+			errx(1, "kevent: filter");
+		break;
+	case KQUEUE_WRITE:
+	case KQUEUE_WRITE_EOF:
+		if ((int)kev.ident != ctx->c_pipe[1])
+			errx(1, "kevent: ident");
+		if (kev.filter != EVFILT_WRITE)
+			errx(1, "kevent: filter");
+		break;
+	}
+
+	switch (ctx->c_mode) {
+	case KQUEUE_READ_EOF:
+	case KQUEUE_WRITE_EOF:
+		if ((kev.flags & EV_EOF) == 0)
+			errx(1, "kevent: eof");
+		break;
+	default:
+		break;
+	}
+
+	ctx_lock(ctx);
+	ctx->c_alive = 0;
+	ctx_unlock(ctx);
+
+	return NULL;
+}