1 files changed, 112 insertions, 0 deletions
diff --git a/test/read-before-exit.c b/test/read-before-exit.c
new file mode 100644
index 0000000..be36bd4
--- /dev/null
+++ b/test/read-before-exit.c
@@ -0,0 +1,112 @@
+/* SPDX-License-Identifier: MIT */
+/*
+ * Description: test if issuing IO from thread and immediately exiting will
+ * proceed correctly.
+ *
+ * Original test case from: https://github.com/axboe/liburing/issues/582
+ */
+#include <unistd.h>
+#include <pthread.h>
+#include <sys/timerfd.h>
+#include <string.h>
+#include <stdio.h>
+
+#include "liburing.h"
+#include "helpers.h"
+
+struct data {
+        struct io_uring *ring;
+        int timer_fd1;
+        int timer_fd2;
+        uint64_t buf1;
+        uint64_t buf2;
+};
+
+void *submit(void *data)
+{
+	struct io_uring_sqe *sqe;
+	struct data *d = data;
+	int ret;
+
+	sqe = io_uring_get_sqe(d->ring);
+	io_uring_prep_read(sqe, d->timer_fd1, &d->buf1, sizeof(d->buf1), 0);
+
+	sqe = io_uring_get_sqe(d->ring);
+	io_uring_prep_read(sqe, d->timer_fd2, &d->buf2, sizeof(d->buf2), 0);
+
+	ret = io_uring_submit(d->ring);
+	if (ret != 2)
+		return (void *) (uintptr_t) 1;
+
+	/* Exit suddenly. */
+	return NULL;
+}
+
+static int test(int flags)
+{
+	struct io_uring_params params = { .flags = flags, };
+	struct io_uring ring;
+	struct data d = { .ring = &ring, };
+	pthread_t thread;
+	void *res;
+	int ret;
+
+	ret = t_create_ring_params(8, &ring, &params);
+	if (ret == T_SETUP_SKIP)
+		return 0;
+	else if (ret != T_SETUP_OK)
+		return 1;
+
+	d.timer_fd1 = timerfd_create(CLOCK_MONOTONIC, TFD_CLOEXEC);
+	if (d.timer_fd1 < 0) {
+		perror("timerfd_create");
+		return 1;
+	}
+	d.timer_fd2 = timerfd_create(CLOCK_MONOTONIC, TFD_CLOEXEC);
+	if (d.timer_fd2 < 0) {
+		perror("timerfd_create");
+		return 1;
+	}
+
+	pthread_create(&thread, NULL, submit, &d);
+	pthread_join(thread, &res);
+
+	/** Wait for completions and do stuff ...  **/
+
+	io_uring_queue_exit(&ring);
+
+	close(d.timer_fd1);
+	close(d.timer_fd2);
+	return !!res;
+}
+
+int main(int argc, char *argv[])
+{
+	int ret, i;
+
+	for (i = 0; i < 1000; i++) {
+		ret = test(0);
+		if (ret) {
+			fprintf(stderr, "Test failed\n");
+			return ret;
+		}
+	}
+
+	for (i = 0; i < 1000; i++) {
+		ret = test(IORING_SETUP_IOPOLL);
+		if (ret) {
+			fprintf(stderr, "Test IOPOLL failed\n");
+			return ret;
+		}
+	}
+
+	for (i = 0; i < 100; i++) {
+		ret = test(IORING_SETUP_SQPOLL);
+		if (ret) {
+			fprintf(stderr, "Test SQPOLL failed\n");
+			return ret;
+		}
+	}
+
+	return 0;
+}