1 /* SPDX-License-Identifier: MIT */
2 /*
3 * Description: test disable/enable notifications through eventfd
4 *
5 */
6 #include <errno.h>
7 #include <stdio.h>
8 #include <unistd.h>
9 #include <stdlib.h>
10 #include <string.h>
11 #include <fcntl.h>
12 #include <poll.h>
13 #include <sys/eventfd.h>
14
15 #include "liburing.h"
16
main(int argc,char * argv[])17 int main(int argc, char *argv[])
18 {
19 struct io_uring_params p = {};
20 struct io_uring_sqe *sqe;
21 struct io_uring_cqe *cqe;
22 struct io_uring ring;
23 uint64_t ptr;
24 struct iovec vec = {
25 .iov_base = &ptr,
26 .iov_len = sizeof(ptr)
27 };
28 int ret, evfd, i;
29
30 if (argc > 1)
31 return 0;
32
33 ret = io_uring_queue_init_params(64, &ring, &p);
34 if (ret) {
35 fprintf(stderr, "ring setup failed: %d\n", ret);
36 return 1;
37 }
38
39 evfd = eventfd(0, EFD_CLOEXEC);
40 if (evfd < 0) {
41 perror("eventfd");
42 return 1;
43 }
44
45 ret = io_uring_register_eventfd(&ring, evfd);
46 if (ret) {
47 fprintf(stderr, "failed to register evfd: %d\n", ret);
48 return 1;
49 }
50
51 if (!io_uring_cq_eventfd_enabled(&ring)) {
52 fprintf(stderr, "eventfd disabled\n");
53 return 1;
54 }
55
56 ret = io_uring_cq_eventfd_toggle(&ring, false);
57 if (ret) {
58 fprintf(stdout, "Skipping, CQ flags not available!\n");
59 return 0;
60 }
61
62 sqe = io_uring_get_sqe(&ring);
63 io_uring_prep_readv(sqe, evfd, &vec, 1, 0);
64 sqe->user_data = 1;
65
66 ret = io_uring_submit(&ring);
67 if (ret != 1) {
68 fprintf(stderr, "submit: %d\n", ret);
69 return 1;
70 }
71
72 for (i = 0; i < 63; i++) {
73 sqe = io_uring_get_sqe(&ring);
74 io_uring_prep_nop(sqe);
75 sqe->user_data = 2;
76 }
77
78 ret = io_uring_submit(&ring);
79 if (ret != 63) {
80 fprintf(stderr, "submit: %d\n", ret);
81 return 1;
82 }
83
84 for (i = 0; i < 63; i++) {
85 ret = io_uring_wait_cqe(&ring, &cqe);
86 if (ret) {
87 fprintf(stderr, "wait: %d\n", ret);
88 return 1;
89 }
90
91 switch (cqe->user_data) {
92 case 1: /* eventfd */
93 fprintf(stderr, "eventfd unexpected: %d\n", (int)ptr);
94 return 1;
95 case 2:
96 if (cqe->res) {
97 fprintf(stderr, "nop: %d\n", cqe->res);
98 return 1;
99 }
100 break;
101 }
102 io_uring_cqe_seen(&ring, cqe);
103 }
104
105 ret = io_uring_cq_eventfd_toggle(&ring, true);
106 if (ret) {
107 fprintf(stderr, "io_uring_cq_eventfd_toggle: %d\n", ret);
108 return 1;
109 }
110
111 sqe = io_uring_get_sqe(&ring);
112 io_uring_prep_nop(sqe);
113 sqe->user_data = 2;
114
115 ret = io_uring_submit(&ring);
116 if (ret != 1) {
117 fprintf(stderr, "submit: %d\n", ret);
118 return 1;
119 }
120
121 for (i = 0; i < 2; i++) {
122 ret = io_uring_wait_cqe(&ring, &cqe);
123 if (ret) {
124 fprintf(stderr, "wait: %d\n", ret);
125 return 1;
126 }
127
128 switch (cqe->user_data) {
129 case 1: /* eventfd */
130 if (cqe->res != sizeof(ptr)) {
131 fprintf(stderr, "read res: %d\n", cqe->res);
132 return 1;
133 }
134
135 if (ptr != 1) {
136 fprintf(stderr, "eventfd: %d\n", (int)ptr);
137 return 1;
138 }
139 break;
140 case 2:
141 if (cqe->res) {
142 fprintf(stderr, "nop: %d\n", cqe->res);
143 return 1;
144 }
145 break;
146 }
147 io_uring_cqe_seen(&ring, cqe);
148 }
149
150 return 0;
151 }
152