1 /* SPDX-License-Identifier: MIT */
2 /*
3 * Description: test ring messaging with flags command
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 <pthread.h>
13
14 #include "liburing.h"
15 #include "helpers.h"
16
17 #define CUSTOM_FLAG 0x42
18 #define USER_DATA 0x5aa5
19 #define LEN 0x20
20 #define ID 0x1
21
22 struct data {
23 pthread_barrier_t barrier;
24 int fd;
25 };
26
recv_msg(struct io_uring * ring)27 static int recv_msg(struct io_uring *ring)
28 {
29 struct io_uring_cqe *cqe;
30 int ret;
31
32 ret = io_uring_wait_cqe(ring, &cqe);
33 if (ret) {
34 fprintf(stderr, "wait cqe %d\n", ret);
35 return T_EXIT_FAIL;
36 }
37 if (cqe->user_data != USER_DATA) {
38 fprintf(stderr, "user_data %llx\n", (long long) cqe->user_data);
39 return T_EXIT_FAIL;
40 }
41 if (cqe->res != LEN) {
42 fprintf(stderr, "len %x\n", cqe->res);
43 return T_EXIT_FAIL;
44 }
45 if (cqe->flags != CUSTOM_FLAG) {
46 fprintf(stderr, "flags %x\n", cqe->flags);
47 return T_EXIT_FAIL;
48 }
49
50 return T_EXIT_PASS;
51 }
52
send_msg(struct io_uring * ring,int target_fd)53 static int send_msg(struct io_uring *ring, int target_fd)
54 {
55 struct io_uring_cqe *cqe;
56 struct io_uring_sqe *sqe;
57 int ret;
58
59 sqe = io_uring_get_sqe(ring);
60 if (!sqe) {
61 fprintf(stderr, "get sqe failed\n");
62 return T_EXIT_FAIL;
63 }
64
65 io_uring_prep_msg_ring_cqe_flags(sqe, target_fd, LEN, USER_DATA,
66 0, CUSTOM_FLAG);
67 sqe->user_data = ID;
68
69 ret = io_uring_submit(ring);
70 if (ret <= 0) {
71 if (ret == -EINVAL)
72 return T_EXIT_SKIP;
73
74 fprintf(stderr, "sqe submit failed: %d\n", ret);
75 return T_EXIT_FAIL;
76 }
77
78 ret = io_uring_wait_cqe(ring, &cqe);
79 if (ret < 0) {
80 fprintf(stderr, "wait completion %d\n", ret);
81 return T_EXIT_FAIL;
82 }
83 if (cqe->res != 0) {
84 if (cqe->res == -EINVAL)
85 return T_EXIT_SKIP;
86 fprintf(stderr, "cqe res %d\n", cqe->res);
87 return T_EXIT_FAIL;
88 }
89 if (cqe->user_data != ID) {
90 fprintf(stderr, "user_data %llx\n", (long long) cqe->user_data);
91 return T_EXIT_FAIL;
92 }
93
94 io_uring_cqe_seen(ring, cqe);
95 return T_EXIT_PASS;
96 }
97
thread_fn(void * data)98 static void *thread_fn(void *data)
99 {
100 struct data *d = data;
101 struct io_uring ring;
102 int ret;
103
104 ret = io_uring_queue_init(2, &ring, IORING_SETUP_DEFER_TASKRUN | IORING_SETUP_SINGLE_ISSUER);
105 if (ret) {
106 fprintf(stderr, "ring init failed %d\n", ret);
107 pthread_barrier_wait(&d->barrier);
108 return NULL;
109 }
110
111 d->fd = ring.ring_fd;
112 pthread_barrier_wait(&d->barrier);
113
114 if (recv_msg(&ring))
115 return (void *) 1;
116
117 return NULL;
118 }
119
test(int ring_flags)120 static int test(int ring_flags)
121 {
122 struct io_uring ring, ring2;
123 pthread_t thread;
124 struct data d;
125 void *ret2;
126 int ret, i;
127
128 ret = io_uring_queue_init(2, &ring, ring_flags);
129 if (ret) {
130 fprintf(stderr, "io_uring_queue_init failed for ring1: %d\n", ret);
131 return T_EXIT_FAIL;
132 }
133
134 ret = io_uring_queue_init(2, &ring2, ring_flags);
135 if (ret) {
136 fprintf(stderr, "io_uring_queue_init failed for ring2: %d\n", ret);
137 return T_EXIT_FAIL;
138 }
139
140 ret = send_msg(&ring, ring2.ring_fd);
141 if (ret) {
142 if (ret != T_EXIT_SKIP)
143 fprintf(stderr, "send_msg failed: %d\n", ret);
144 return ret;
145 }
146
147 ret = recv_msg(&ring2);
148 if (ret) {
149 fprintf(stderr, "recv_msg failed: %d\n", ret);
150 return ret;
151 }
152
153 for (i = 0; i < 8; i++) {
154 ret = send_msg(&ring, ring2.ring_fd);
155 if (ret) {
156 if (ret != T_EXIT_SKIP)
157 fprintf(stderr, "send_msg failed: %d\n", ret);
158 return ret;
159 }
160 }
161
162 for (i = 0; i < 8; i++) {
163 ret = recv_msg(&ring2);
164 if (ret) {
165 fprintf(stderr, "recv_msg failed: %d\n", ret);
166 return ret;
167 }
168 }
169
170 pthread_barrier_init(&d.barrier, NULL, 2);
171 d.fd = -1;
172 pthread_create(&thread, NULL, thread_fn, &d);
173 pthread_barrier_wait(&d.barrier);
174 if (d.fd == -1)
175 return T_EXIT_FAIL;
176
177 ret = send_msg(&ring, d.fd);
178 if (ret) {
179 fprintf(stderr, "send msg failed: %d\n", ret);
180 return ret;
181 }
182 pthread_join(thread, &ret2);
183 if (ret2) {
184 fprintf(stderr, "Remote test failed\n");
185 return T_EXIT_FAIL;
186 }
187
188 return T_EXIT_PASS;
189 }
190
main(int argc,char * argv[])191 int main(int argc, char *argv[])
192 {
193 int ret;
194
195 if (argc > 1)
196 return T_EXIT_SKIP;
197
198 ret = test(0);
199 if (ret == T_EXIT_FAIL) {
200 fprintf(stderr, "test ring_flags 0 failed\n");
201 return ret;
202 } else if (ret == T_EXIT_SKIP)
203 return ret;
204
205 ret = test(IORING_SETUP_SINGLE_ISSUER|IORING_SETUP_DEFER_TASKRUN);
206 if (ret == T_EXIT_FAIL) {
207 fprintf(stderr, "test ring_flags defer failed\n");
208 return ret;
209 }
210
211 return ret;
212 }
213