1 /* SPDX-License-Identifier: MIT */
2 /*
3 * Description: run various eventfd tests
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 #include "helpers.h"
17
main(int argc,char * argv[])18 int main(int argc, char *argv[])
19 {
20 struct io_uring_params p = {};
21 struct io_uring_sqe *sqe;
22 struct io_uring_cqe *cqe;
23 struct io_uring ring;
24 uint64_t ptr;
25 struct iovec vec = {
26 .iov_base = &ptr,
27 .iov_len = sizeof(ptr)
28 };
29 int ret, evfd, i;
30
31 if (argc > 1)
32 return T_EXIT_SKIP;
33
34 ret = io_uring_queue_init_params(8, &ring, &p);
35 if (ret) {
36 fprintf(stderr, "ring setup failed: %d\n", ret);
37 return T_EXIT_FAIL;
38 }
39 if (!(p.features & IORING_FEAT_CUR_PERSONALITY)) {
40 fprintf(stdout, "Skipping\n");
41 return T_EXIT_SKIP;
42 }
43
44 evfd = eventfd(0, EFD_CLOEXEC);
45 if (evfd < 0) {
46 perror("eventfd");
47 return T_EXIT_FAIL;
48 }
49
50 ret = io_uring_register_eventfd(&ring, evfd);
51 if (ret) {
52 fprintf(stderr, "failed to register evfd: %d\n", ret);
53 return T_EXIT_FAIL;
54 }
55
56 sqe = io_uring_get_sqe(&ring);
57 io_uring_prep_poll_add(sqe, evfd, POLLIN);
58 sqe->flags |= IOSQE_IO_LINK;
59 sqe->user_data = 1;
60
61 sqe = io_uring_get_sqe(&ring);
62 io_uring_prep_readv(sqe, evfd, &vec, 1, 0);
63 sqe->user_data = 2;
64
65 ret = io_uring_submit(&ring);
66 if (ret != 2) {
67 fprintf(stderr, "submit: %d\n", ret);
68 return T_EXIT_FAIL;
69 }
70
71 sqe = io_uring_get_sqe(&ring);
72 io_uring_prep_nop(sqe);
73 sqe->user_data = 3;
74
75 ret = io_uring_submit(&ring);
76 if (ret != 1) {
77 fprintf(stderr, "submit: %d\n", ret);
78 return T_EXIT_FAIL;
79 }
80
81 for (i = 0; i < 3; i++) {
82 ret = io_uring_wait_cqe(&ring, &cqe);
83 if (ret) {
84 fprintf(stderr, "wait: %d\n", ret);
85 return T_EXIT_FAIL;
86 }
87 switch (cqe->user_data) {
88 case 1:
89 /* POLLIN */
90 if (cqe->res != 1) {
91 fprintf(stderr, "poll: %d\n", cqe->res);
92 return T_EXIT_FAIL;
93 }
94 break;
95 case 2:
96 if (cqe->res != sizeof(ptr)) {
97 fprintf(stderr, "read: %d\n", cqe->res);
98 return T_EXIT_FAIL;
99 }
100 break;
101 case 3:
102 if (cqe->res) {
103 fprintf(stderr, "nop: %d\n", cqe->res);
104 return T_EXIT_FAIL;
105 }
106 break;
107 }
108 io_uring_cqe_seen(&ring, cqe);
109 }
110
111 return T_EXIT_PASS;
112 }
113