aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/libs/liburing/test/read-before-exit.c
diff options
context:
space:
mode:
authorilnaz <ilnaz@ydb.tech>2022-12-13 16:01:38 +0300
committerilnaz <ilnaz@ydb.tech>2022-12-13 16:01:38 +0300
commitf2bea70bea01921ec43846224d100f2c70dd5719 (patch)
treeeead917572063b63adc1c9a76284c8fbd10f25a3 /contrib/libs/liburing/test/read-before-exit.c
parent1ab9ee3dfe0ab4023a3a57bf55de31dff3eac908 (diff)
downloadydb-f2bea70bea01921ec43846224d100f2c70dd5719.tar.gz
Add cross-link
Diffstat (limited to 'contrib/libs/liburing/test/read-before-exit.c')
-rw-r--r--contrib/libs/liburing/test/read-before-exit.c113
1 files changed, 113 insertions, 0 deletions
diff --git a/contrib/libs/liburing/test/read-before-exit.c b/contrib/libs/liburing/test/read-before-exit.c
new file mode 100644
index 0000000000..6510a09263
--- /dev/null
+++ b/contrib/libs/liburing/test/read-before-exit.c
@@ -0,0 +1,113 @@
+#include "../config-host.h"
+/* 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;
+}