aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/libs/liburing/test/reg-fd-only.c
diff options
context:
space:
mode:
authorthegeorg <thegeorg@yandex-team.com>2024-06-09 11:55:21 +0300
committerthegeorg <thegeorg@yandex-team.com>2024-06-09 12:07:55 +0300
commitafd4899380eea1c70e2a68714b5da1c9919ccdbd (patch)
treecd5120708784139bc6a0f8881da1ed8389a065b3 /contrib/libs/liburing/test/reg-fd-only.c
parenta83bd2dd3c21e38c6c0807ec5e679497ab567f24 (diff)
downloadydb-afd4899380eea1c70e2a68714b5da1c9919ccdbd.tar.gz
Update contrib/libs/liburing to 2.6
3b51a9fb14de805208d11f1c077c78bb5d487e0f
Diffstat (limited to 'contrib/libs/liburing/test/reg-fd-only.c')
-rw-r--r--contrib/libs/liburing/test/reg-fd-only.c132
1 files changed, 132 insertions, 0 deletions
diff --git a/contrib/libs/liburing/test/reg-fd-only.c b/contrib/libs/liburing/test/reg-fd-only.c
new file mode 100644
index 0000000000..1ff6ea250e
--- /dev/null
+++ b/contrib/libs/liburing/test/reg-fd-only.c
@@ -0,0 +1,132 @@
+#include "../config-host.h"
+/* SPDX-License-Identifier: MIT */
+/*
+ * Test io_uring_setup with IORING_SETUP_REGISTERED_FD_ONLY
+ *
+ */
+#include <stdio.h>
+
+#include "helpers.h"
+
+#define NORMAL_PAGE_ENTRIES 8
+#define HUGE_PAGE_ENTRIES 512
+
+static int no_mmap;
+
+static int test_nops(struct io_uring *ring, int sq_size, int nr_nops)
+{
+ struct io_uring_sqe *sqe;
+ struct io_uring_cqe *cqe;
+ int i, ret;
+
+ do {
+ int todo = nr_nops;
+
+ if (todo > sq_size)
+ todo = sq_size;
+
+ for (i = 0; i < todo; i++) {
+ sqe = io_uring_get_sqe(ring);
+ io_uring_prep_nop(sqe);
+ }
+
+ ret = io_uring_submit(ring);
+ if (ret != todo) {
+ fprintf(stderr, "short submit %d\n", ret);
+ return T_EXIT_FAIL;
+ }
+
+ for (i = 0; i < todo; i++) {
+ ret = io_uring_wait_cqe(ring, &cqe);
+ if (ret) {
+ fprintf(stderr, "wait err %d\n", ret);
+ return T_EXIT_FAIL;
+ }
+ io_uring_cqe_seen(ring, cqe);
+ }
+ nr_nops -= todo;
+ } while (nr_nops);
+
+ return T_EXIT_PASS;
+}
+
+static int test(int nentries)
+{
+ struct io_uring ring;
+ unsigned values[2];
+ int ret;
+
+ ret = io_uring_queue_init(nentries, &ring,
+ IORING_SETUP_REGISTERED_FD_ONLY | IORING_SETUP_NO_MMAP);
+ if (ret == -EINVAL) {
+ no_mmap = 1;
+ return T_EXIT_SKIP;
+ } else if (ret == -ENOMEM) {
+ fprintf(stdout, "Enable huge pages to test big rings\n");
+ return T_EXIT_SKIP;
+ } else if (ret) {
+ fprintf(stderr, "ring setup failed\n");
+ return T_EXIT_FAIL;
+ }
+
+ ret = io_uring_register_ring_fd(&ring);
+ if (ret != -EEXIST) {
+ fprintf(stderr, "registering already-registered ring fd should fail\n");
+ goto err;
+ }
+
+ ret = io_uring_close_ring_fd(&ring);
+ if (ret != -EBADF) {
+ fprintf(stderr, "closing already-closed ring fd should fail\n");
+ goto err;
+ }
+
+ /* Test a simple io_uring_register operation expected to work.
+ * io_uring_register_iowq_max_workers is arbitrary.
+ */
+ values[0] = values[1] = 0;
+ ret = io_uring_register_iowq_max_workers(&ring, values);
+ if (ret || (values[0] == 0 && values[1] == 0)) {
+ fprintf(stderr, "io_uring_register operation failed after closing ring fd\n");
+ goto err;
+ }
+
+ ret = test_nops(&ring, nentries, nentries * 4);
+ if (ret)
+ goto err;
+
+ io_uring_queue_exit(&ring);
+ return T_EXIT_PASS;
+
+err:
+ io_uring_queue_exit(&ring);
+ return T_EXIT_FAIL;
+}
+
+int main(int argc, char *argv[])
+{
+ int ret;
+
+ if (argc > 1)
+ return T_EXIT_SKIP;
+
+ /* test single normal page */
+ ret = test(NORMAL_PAGE_ENTRIES);
+ if (ret == T_EXIT_SKIP || no_mmap) {
+ return T_EXIT_SKIP;
+ } else if (ret != T_EXIT_PASS) {
+ fprintf(stderr, "test 8 failed\n");
+ return T_EXIT_FAIL;
+ }
+
+ /* test with entries requiring a huge page */
+ ret = test(HUGE_PAGE_ENTRIES);
+ if (ret == T_EXIT_SKIP) {
+ return T_EXIT_SKIP;
+ } else if (ret != T_EXIT_PASS) {
+ fprintf(stderr, "test 512 failed\n");
+ return T_EXIT_FAIL;
+ }
+
+ return T_EXIT_PASS;
+}