polyphony 0.85 → 0.86

data/vendor/liburing/test/poll-mshot-update.c

@@ -0,0 +1,273 @@
+/* SPDX-License-Identifier: MIT */
+/*
+ * Description: test many files being polled for and updated
+ *
+ */
+#include <errno.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <string.h>
+#include <signal.h>
+#include <poll.h>
+#include <sys/resource.h>
+#include <fcntl.h>
+#include <pthread.h>
+
+#include "liburing.h"
+
+#define	NFILES	5000
+#define BATCH	500
+#define NLOOPS	1000
+
+#define RING_SIZE	512
+
+struct p {
+	int fd[2];
+	int triggered;
+};
+
+static struct p p[NFILES];
+
+static int has_poll_update(void)
+{
+	struct io_uring ring;
+	struct io_uring_cqe *cqe;
+	struct io_uring_sqe *sqe;
+	bool has_update = false;
+	int ret;
+
+	ret = io_uring_queue_init(8, &ring, 0);
+	if (ret)
+		return -1;
+
+	sqe = io_uring_get_sqe(&ring);
+	io_uring_prep_poll_update(sqe, 0, 0, POLLIN, IORING_TIMEOUT_UPDATE);
+
+	ret = io_uring_submit(&ring);
+	if (ret != 1)
+		return -1;
+
+	ret = io_uring_wait_cqe(&ring, &cqe);
+	if (!ret) {
+		if (cqe->res == -ENOENT)
+			has_update = true;
+		else if (cqe->res != -EINVAL)
+			return -1;
+		io_uring_cqe_seen(&ring, cqe);
+	}
+	io_uring_queue_exit(&ring);
+	return has_update;
+}
+
+static int arm_poll(struct io_uring *ring, int off)
+{
+	struct io_uring_sqe *sqe;
+
+	sqe = io_uring_get_sqe(ring);
+	if (!sqe) {
+		fprintf(stderr, "failed getting sqe\n");
+		return 1;
+	}
+
+	io_uring_prep_poll_multishot(sqe, p[off].fd[0], POLLIN);
+	sqe->user_data = off;
+	return 0;
+}
+
+static int reap_polls(struct io_uring *ring)
+{
+	struct io_uring_cqe *cqe;
+	int i, ret, off;
+	char c;
+
+	for (i = 0; i < BATCH; i++) {
+		struct io_uring_sqe *sqe;
+
+		sqe = io_uring_get_sqe(ring);
+		/* update event */
+		io_uring_prep_poll_update(sqe, i, 0, POLLIN,
+						IORING_POLL_UPDATE_EVENTS);
+		sqe->user_data = 0x12345678;
+	}
+
+	ret = io_uring_submit(ring);
+	if (ret != BATCH) {
+		fprintf(stderr, "submitted %d, %d\n", ret, BATCH);
+		return 1;
+	}
+
+	for (i = 0; i < 2 * BATCH; i++) {
+		ret = io_uring_wait_cqe(ring, &cqe);
+		if (ret) {
+			fprintf(stderr, "wait cqe %d\n", ret);
+			return ret;
+		}
+		off = cqe->user_data;
+		if (off == 0x12345678)
+			goto seen;
+		ret = read(p[off].fd[0], &c, 1);
+		if (ret != 1) {
+			if (ret == -1 && errno == EAGAIN)
+				goto seen;
+			fprintf(stderr, "read got %d/%d\n", ret, errno);
+			break;
+		}
+seen:
+		io_uring_cqe_seen(ring, cqe);
+	}
+
+	if (i != 2 * BATCH) {
+		fprintf(stderr, "gave up at %d\n", i);
+		return 1;
+	}
+
+	return 0;
+}
+
+static int trigger_polls(void)
+{
+	char c = 89;
+	int i, ret;
+
+	for (i = 0; i < BATCH; i++) {
+		int off;
+
+		do {
+			off = rand() % NFILES;
+			if (!p[off].triggered)
+				break;
+		} while (1);
+
+		p[off].triggered = 1;
+		ret = write(p[off].fd[1], &c, 1);
+		if (ret != 1) {
+			fprintf(stderr, "write got %d/%d\n", ret, errno);
+			return 1;
+		}
+	}
+
+	return 0;
+}
+
+static void *trigger_polls_fn(void *data)
+{
+	trigger_polls();
+	return NULL;
+}
+
+static int arm_polls(struct io_uring *ring)
+{
+	int ret, to_arm = NFILES, i, off;
+
+	off = 0;
+	while (to_arm) {
+		int this_arm;
+
+		this_arm = to_arm;
+		if (this_arm > RING_SIZE)
+			this_arm = RING_SIZE;
+
+		for (i = 0; i < this_arm; i++) {
+			if (arm_poll(ring, off)) {
+				fprintf(stderr, "arm failed at %d\n", off);
+				return 1;
+			}
+			off++;
+		}
+
+		ret = io_uring_submit(ring);
+		if (ret != this_arm) {
+			fprintf(stderr, "submitted %d, %d\n", ret, this_arm);
+			return 1;
+		}
+		to_arm -= this_arm;
+	}
+
+	return 0;
+}
+
+int main(int argc, char *argv[])
+{
+	struct io_uring ring;
+	struct io_uring_params params = { };
+	struct rlimit rlim;
+	pthread_t thread;
+	int i, j, ret;
+
+	if (argc > 1)
+		return 0;
+
+	ret = has_poll_update();
+	if (ret < 0) {
+		fprintf(stderr, "poll update check failed %i\n", ret);
+		return -1;
+	} else if (!ret) {
+		fprintf(stderr, "no poll update, skip\n");
+		return 0;
+	}
+
+	if (getrlimit(RLIMIT_NOFILE, &rlim) < 0) {
+		perror("getrlimit");
+		goto err_noring;
+	}
+
+	if (rlim.rlim_cur < (2 * NFILES + 5)) {
+		rlim.rlim_cur = (2 * NFILES + 5);
+		rlim.rlim_max = rlim.rlim_cur;
+		if (setrlimit(RLIMIT_NOFILE, &rlim) < 0) {
+			if (errno == EPERM)
+				goto err_nofail;
+			perror("setrlimit");
+			goto err_noring;
+		}
+	}
+
+	for (i = 0; i < NFILES; i++) {
+		if (pipe(p[i].fd) < 0) {
+			perror("pipe");
+			goto err_noring;
+		}
+		fcntl(p[i].fd[0], F_SETFL, O_NONBLOCK);
+	}
+
+	params.flags = IORING_SETUP_CQSIZE;
+	params.cq_entries = 4096;
+	ret = io_uring_queue_init_params(RING_SIZE, &ring, &params);
+	if (ret) {
+		if (ret == -EINVAL) {
+			fprintf(stdout, "No CQSIZE, trying without\n");
+			ret = io_uring_queue_init(RING_SIZE, &ring, 0);
+			if (ret) {
+				fprintf(stderr, "ring setup failed: %d\n", ret);
+				return 1;
+			}
+		}
+	}
+
+	if (arm_polls(&ring))
+		goto err;
+
+	for (i = 0; i < NLOOPS; i++) {
+		pthread_create(&thread, NULL, trigger_polls_fn, NULL);
+		ret = reap_polls(&ring);
+		if (ret)
+			goto err;
+		pthread_join(thread, NULL);
+
+		for (j = 0; j < NFILES; j++)
+			p[j].triggered = 0;
+	}
+
+	io_uring_queue_exit(&ring);
+	return 0;
+err:
+	io_uring_queue_exit(&ring);
+err_noring:
+	fprintf(stderr, "poll-many failed\n");
+	return 1;
+err_nofail:
+	fprintf(stderr, "poll-many: not enough files available (and not root), "
+			"skipped\n");
+	return 0;
+}

data/vendor/liburing/test/poll-ring.c

@@ -0,0 +1,48 @@
+/* SPDX-License-Identifier: MIT */
+/*
+ * Description: Test poll against ring itself. A buggy kernel will end up
+ * 		having io_wq_* workers pending, as the circular reference
+ * 		will prevent full exit.
+ *
+ */
+#include <errno.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <string.h>
+#include <poll.h>
+
+#include "liburing.h"
+
+int main(int argc, char *argv[])
+{
+	struct io_uring_sqe *sqe;
+	struct io_uring ring;
+	int ret;
+
+	if (argc > 1)
+		return 0;
+
+	ret = io_uring_queue_init(1, &ring, 0);
+	if (ret) {
+		fprintf(stderr, "child: ring setup failed: %d\n", ret);
+		return 1;
+	}
+
+	sqe = io_uring_get_sqe(&ring);
+	if (!sqe) {
+		fprintf(stderr, "get sqe failed\n");
+		return 1;
+	}
+
+	io_uring_prep_poll_add(sqe, ring.ring_fd, POLLIN);
+	io_uring_sqe_set_data(sqe, sqe);
+
+	ret = io_uring_submit(&ring);
+	if (ret <= 0) {
+		fprintf(stderr, "child: sqe submit failed: %d\n", ret);
+		return 1;
+	}
+
+	return 0;
+}

data/vendor/liburing/test/poll-v-poll.c

@@ -0,0 +1,353 @@
+/* SPDX-License-Identifier: MIT */
+/*
+ * Description: test io_uring poll handling
+ *
+ */
+#include <errno.h>
+#include <stdio.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <string.h>
+#include <signal.h>
+#include <fcntl.h>
+#include <poll.h>
+#include <sys/wait.h>
+#include <sys/select.h>
+#include <pthread.h>
+#include <sys/epoll.h>
+
+#include "liburing.h"
+
+struct thread_data {
+	struct io_uring *ring;
+	int fd;
+	int events;
+	const char *test;
+	int out[2];
+};
+
+static void *epoll_wait_fn(void *data)
+{
+	struct thread_data *td = data;
+	struct epoll_event ev;
+
+	if (epoll_wait(td->fd, &ev, 1, -1) < 0) {
+		perror("epoll_wait");
+		goto err;
+	}
+
+	return NULL;
+err:
+	return (void *) 1;
+}
+
+static void *iou_poll(void *data)
+{
+	struct thread_data *td = data;
+	struct io_uring_sqe *sqe;
+	struct io_uring_cqe *cqe;
+	int ret;
+
+	sqe = io_uring_get_sqe(td->ring);
+	io_uring_prep_poll_add(sqe, td->fd, td->events);
+
+	ret = io_uring_submit(td->ring);
+	if (ret != 1) {
+		fprintf(stderr, "submit got %d\n", ret);
+		goto err;
+	}
+
+	ret = io_uring_wait_cqe(td->ring, &cqe);
+	if (ret) {
+		fprintf(stderr, "wait_cqe: %d\n", ret);
+		goto err;
+	}
+
+	td->out[0] = cqe->res & 0x3f;
+	io_uring_cqe_seen(td->ring, cqe);
+	return NULL;
+err:
+	return (void *) 1;
+}
+
+static void *poll_pipe(void *data)
+{
+	struct thread_data *td = data;
+	struct pollfd pfd;
+	int ret;
+
+	pfd.fd = td->fd;
+	pfd.events = td->events;
+
+	ret = poll(&pfd, 1, -1);
+	if (ret < 0)
+		perror("poll");
+
+	td->out[1] = pfd.revents;
+	return NULL;
+}
+
+static int do_pipe_pollin_test(struct io_uring *ring)
+{
+	struct thread_data td;
+	pthread_t threads[2];
+	int ret, pipe1[2];
+	char buf;
+
+	if (pipe(pipe1) < 0) {
+		perror("pipe");
+		return 1;
+	}
+
+	td.ring = ring;
+	td.fd = pipe1[0];
+	td.events = POLLIN;
+	td.test = __FUNCTION__;
+
+	pthread_create(&threads[1], NULL, iou_poll, &td);
+	pthread_create(&threads[0], NULL, poll_pipe, &td);
+	usleep(100000);
+
+	buf = 0x89;
+	ret = write(pipe1[1], &buf, sizeof(buf));
+	if (ret != sizeof(buf)) {
+		fprintf(stderr, "write failed: %d\n", ret);
+		return 1;
+	}
+
+	pthread_join(threads[0], NULL);
+	pthread_join(threads[1], NULL);
+
+	if (td.out[0] != td.out[1]) {
+		fprintf(stderr, "%s: res %x/%x differ\n", __FUNCTION__,
+							td.out[0], td.out[1]);
+		return 1;
+	}
+	return 0;
+}
+
+static int do_pipe_pollout_test(struct io_uring *ring)
+{
+	struct thread_data td;
+	pthread_t threads[2];
+	int ret, pipe1[2];
+	char buf;
+
+	if (pipe(pipe1) < 0) {
+		perror("pipe");
+		return 1;
+	}
+
+	td.ring = ring;
+	td.fd = pipe1[1];
+	td.events = POLLOUT;
+	td.test = __FUNCTION__;
+
+	pthread_create(&threads[0], NULL, poll_pipe, &td);
+	pthread_create(&threads[1], NULL, iou_poll, &td);
+	usleep(100000);
+
+	buf = 0x89;
+	ret = write(pipe1[1], &buf, sizeof(buf));
+	if (ret != sizeof(buf)) {
+		fprintf(stderr, "write failed: %d\n", ret);
+		return 1;
+	}
+
+	pthread_join(threads[0], NULL);
+	pthread_join(threads[1], NULL);
+
+	if (td.out[0] != td.out[1]) {
+		fprintf(stderr, "%s: res %x/%x differ\n", __FUNCTION__,
+							td.out[0], td.out[1]);
+		return 1;
+	}
+
+	return 0;
+}
+
+static int do_fd_test(struct io_uring *ring, const char *fname, int events)
+{
+	struct thread_data td;
+	pthread_t threads[2];
+	int fd;
+
+	fd = open(fname, O_RDONLY);
+	if (fd < 0) {
+		perror("open");
+		return 1;
+	}
+
+	td.ring = ring;
+	td.fd = fd;
+	td.events = events;
+	td.test = __FUNCTION__;
+
+	pthread_create(&threads[0], NULL, poll_pipe, &td);
+	pthread_create(&threads[1], NULL, iou_poll, &td);
+
+	pthread_join(threads[0], NULL);
+	pthread_join(threads[1], NULL);
+
+	if (td.out[0] != td.out[1]) {
+		fprintf(stderr, "%s: res %x/%x differ\n", __FUNCTION__,
+							td.out[0], td.out[1]);
+		return 1;
+	}
+
+	return 0;
+}
+
+static int iou_epoll_ctl(struct io_uring *ring, int epfd, int fd,
+			 struct epoll_event *ev)
+{
+	struct io_uring_sqe *sqe;
+	struct io_uring_cqe *cqe;
+	int ret;
+
+	sqe = io_uring_get_sqe(ring);
+	if (!sqe) {
+		fprintf(stderr, "Failed to get sqe\n");
+		return 1;
+	}
+
+	io_uring_prep_epoll_ctl(sqe, epfd, fd, EPOLL_CTL_ADD, ev);
+
+	ret = io_uring_submit(ring);
+	if (ret != 1) {
+		fprintf(stderr, "submit: %d\n", ret);
+		return 1;
+	}
+
+	ret = io_uring_wait_cqe(ring, &cqe);
+	if (ret) {
+		fprintf(stderr, "wait_cqe: %d\n", ret);
+		return 1;
+	}
+
+	ret = cqe->res;
+	io_uring_cqe_seen(ring, cqe);
+	return ret;
+}
+
+static int do_test_epoll(struct io_uring *ring, int iou_epoll_add)
+{
+	struct epoll_event ev;
+	struct thread_data td;
+	pthread_t threads[2];
+	int ret, pipe1[2];
+	char buf;
+	int fd;
+
+	fd = epoll_create1(0);
+	if (fd < 0) {
+		perror("epoll_create");
+		return 1;
+	}
+
+	if (pipe(pipe1) < 0) {
+		perror("pipe");
+		return 1;
+	}
+
+	ev.events = EPOLLIN;
+	ev.data.fd = pipe1[0];
+
+	if (!iou_epoll_add) {
+		if (epoll_ctl(fd, EPOLL_CTL_ADD, pipe1[0], &ev) < 0) {
+			perror("epoll_ctrl");
+			return 1;
+		}
+	} else {
+		ret = iou_epoll_ctl(ring, fd, pipe1[0], &ev);
+		if (ret == -EINVAL) {
+			fprintf(stdout, "epoll not supported, skipping\n");
+			return 0;
+		} else if (ret < 0) {
+			return 1;
+		}
+	}
+
+	td.ring = ring;
+	td.fd = fd;
+	td.events = POLLIN;
+	td.test = __FUNCTION__;
+
+	pthread_create(&threads[0], NULL, iou_poll, &td);
+	pthread_create(&threads[1], NULL, epoll_wait_fn, &td);
+	usleep(100000);
+
+	buf = 0x89;
+	ret = write(pipe1[1], &buf, sizeof(buf));
+	if (ret != sizeof(buf)) {
+		fprintf(stderr, "write failed: %d\n", ret);
+		return 1;
+	}
+
+	pthread_join(threads[0], NULL);
+	pthread_join(threads[1], NULL);
+	return 0;
+}
+
+int main(int argc, char *argv[])
+{
+	struct io_uring ring;
+	const char *fname;
+	int ret;
+
+	ret = io_uring_queue_init(1, &ring, 0);
+	if (ret) {
+		fprintf(stderr, "ring setup failed\n");
+		return 1;
+	}
+
+	ret = do_pipe_pollin_test(&ring);
+	if (ret) {
+		fprintf(stderr, "pipe pollin test failed\n");
+		return ret;
+	}
+
+	ret = do_pipe_pollout_test(&ring);
+	if (ret) {
+		fprintf(stderr, "pipe pollout test failed\n");
+		return ret;
+	}
+
+	ret = do_test_epoll(&ring, 0);
+	if (ret) {
+		fprintf(stderr, "epoll test 0 failed\n");
+		return ret;
+	}
+
+	ret = do_test_epoll(&ring, 1);
+	if (ret) {
+		fprintf(stderr, "epoll test 1 failed\n");
+		return ret;
+	}
+
+	if (argc > 1)
+		fname = argv[1];
+	else
+		fname = argv[0];
+
+	ret = do_fd_test(&ring, fname, POLLIN);
+	if (ret) {
+		fprintf(stderr, "fd test IN failed\n");
+		return ret;
+	}
+
+	ret = do_fd_test(&ring, fname, POLLOUT);
+	if (ret) {
+		fprintf(stderr, "fd test OUT failed\n");
+		return ret;
+	}
+
+	ret = do_fd_test(&ring, fname, POLLOUT | POLLIN);
+	if (ret) {
+		fprintf(stderr, "fd test IN|OUT failed\n");
+		return ret;
+	}
+
+	return 0;
+
+}