1 /* SPDX-License-Identifier: MIT */
2 /*
3 * Description: test reading provided ring buf head
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
13 #include "liburing.h"
14 #include "helpers.h"
15
16 #define BUF_SIZE 32
17 #define NR_BUFS 8
18 #define FSIZE (BUF_SIZE * NR_BUFS)
19
20 #define BR_MASK (NR_BUFS - 1)
21 #define BGID 1
22
23 static int no_buf_ring;
24 static int no_buf_ring_status;
25
test_max(void)26 static int test_max(void)
27 {
28 struct io_uring_buf_ring *br;
29 struct io_uring ring;
30 int nr_bufs = 32768;
31 int ret, i;
32 char *buf;
33
34 ret = io_uring_queue_init(1, &ring, 0);
35 if (ret) {
36 fprintf(stderr, "ring setup failed: %d\n", ret);
37 return 1;
38 }
39
40 if (posix_memalign((void **) &buf, 4096, FSIZE))
41 return 1;
42
43 br = io_uring_setup_buf_ring(&ring, nr_bufs, BGID, 0, &ret);
44 if (!br) {
45 fprintf(stderr, "Buffer ring register failed %d\n", ret);
46 return 1;
47 }
48
49 ret = io_uring_buf_ring_available(&ring, br, BGID);
50 if (ret) {
51 fprintf(stderr, "Bad available count %d\n", ret);
52 return 1;
53 }
54
55 for (i = 0; i < nr_bufs / 2; i++)
56 io_uring_buf_ring_add(br, buf, BUF_SIZE, i + 1, nr_bufs - 1, i);
57 io_uring_buf_ring_advance(br, nr_bufs / 2);
58
59 ret = io_uring_buf_ring_available(&ring, br, BGID);
60 if (ret != nr_bufs / 2) {
61 fprintf(stderr, "Bad half full available count %d\n", ret);
62 return 1;
63 }
64
65 for (i = 0; i < nr_bufs / 2; i++)
66 io_uring_buf_ring_add(br, buf, BUF_SIZE, i + 1, nr_bufs - 1, i);
67 io_uring_buf_ring_advance(br, nr_bufs / 2);
68
69 ret = io_uring_buf_ring_available(&ring, br, BGID);
70 if (ret != nr_bufs) {
71 fprintf(stderr, "Bad half full available count %d\n", ret);
72 return 1;
73 }
74
75 free(buf);
76 io_uring_queue_exit(&ring);
77 return T_EXIT_PASS;
78 }
79
test(int invalid)80 static int test(int invalid)
81 {
82 struct io_uring_sqe *sqe;
83 struct io_uring_cqe *cqe;
84 struct io_uring ring;
85 struct io_uring_buf_ring *br;
86 int ret, i, fds[2];
87 uint16_t head;
88 char *buf;
89 void *ptr;
90 char output[16];
91
92 memset(output, 0x55, sizeof(output));
93
94 ret = io_uring_queue_init(NR_BUFS, &ring, 0);
95 if (ret) {
96 fprintf(stderr, "ring setup failed: %d\n", ret);
97 return 1;
98 }
99
100 if (pipe(fds) < 0) {
101 perror("pipe");
102 return T_EXIT_FAIL;
103 }
104
105 if (posix_memalign((void **) &buf, 4096, FSIZE))
106 return 1;
107
108 br = io_uring_setup_buf_ring(&ring, NR_BUFS, BGID, 0, &ret);
109 if (!br) {
110 if (ret == -EINVAL) {
111 no_buf_ring = 1;
112 return 0;
113 }
114 fprintf(stderr, "Buffer ring register failed %d\n", ret);
115 return 1;
116 }
117
118 ptr = buf;
119 for (i = 0; i < NR_BUFS; i++) {
120 io_uring_buf_ring_add(br, ptr, BUF_SIZE, i + 1, BR_MASK, i);
121 ptr += BUF_SIZE;
122 }
123 io_uring_buf_ring_advance(br, NR_BUFS);
124
125 /* head should be zero at this point */
126 head = 1;
127 if (!invalid)
128 ret = io_uring_buf_ring_head(&ring, BGID, &head);
129 else
130 ret = io_uring_buf_ring_head(&ring, BGID + 10, &head);
131 if (ret) {
132 if (ret == -EINVAL) {
133 no_buf_ring_status = 1;
134 return T_EXIT_SKIP;
135 }
136 if (invalid && ret == -ENOENT)
137 return T_EXIT_PASS;
138 fprintf(stderr, "buf_ring_head: %d\n", ret);
139 return T_EXIT_FAIL;
140 }
141 if (invalid) {
142 fprintf(stderr, "lookup of bad group id succeeded\n");
143 return T_EXIT_FAIL;
144 }
145 if (head != 0) {
146 fprintf(stderr, "bad head %d\n", head);
147 return T_EXIT_FAIL;
148 }
149
150 ret = io_uring_buf_ring_available(&ring, br, BGID);
151 if (ret != NR_BUFS) {
152 fprintf(stderr, "ring available %d\n", ret);
153 return T_EXIT_FAIL;
154 }
155
156 sqe = io_uring_get_sqe(&ring);
157 io_uring_prep_read(sqe, fds[0], NULL, BUF_SIZE, i * BUF_SIZE);
158 sqe->buf_group = BGID;
159 sqe->flags |= IOSQE_BUFFER_SELECT;
160 sqe->user_data = 1;
161
162 ret = io_uring_submit(&ring);
163 if (ret != 1) {
164 fprintf(stderr, "submit: %d\n", ret);
165 return T_EXIT_FAIL;
166 }
167
168 /* head should still be zero at this point, no buffers consumed */
169 head = 1;
170 ret = io_uring_buf_ring_head(&ring, BGID, &head);
171 if (head != 0) {
172 fprintf(stderr, "bad head after submit %d\n", head);
173 return T_EXIT_FAIL;
174 }
175
176 ret = write(fds[1], output, sizeof(output));
177 if (ret != sizeof(output)) {
178 fprintf(stderr, "pipe buffer write %d\n", ret);
179 return T_EXIT_FAIL;
180 }
181
182 ret = io_uring_wait_cqe(&ring, &cqe);
183 if (ret) {
184 fprintf(stderr, "wait cqe failed %d\n", ret);
185 return T_EXIT_FAIL;
186 }
187 if (cqe->res != sizeof(output)) {
188 fprintf(stderr, "cqe res %d\n", cqe->res);
189 return T_EXIT_FAIL;
190 }
191 if (!(cqe->flags & IORING_CQE_F_BUFFER)) {
192 fprintf(stderr, "no buffer selected\n");
193 return T_EXIT_FAIL;
194 }
195 io_uring_cqe_seen(&ring, cqe);
196
197 /* head should now be one, we consumed a buffer */
198 ret = io_uring_buf_ring_head(&ring, BGID, &head);
199 if (head != 1) {
200 fprintf(stderr, "bad head after cqe %d\n", head);
201 return T_EXIT_FAIL;
202 }
203
204 ret = io_uring_buf_ring_available(&ring, br, BGID);
205 if (ret != NR_BUFS - 1) {
206 fprintf(stderr, "ring available %d\n", ret);
207 return T_EXIT_FAIL;
208 }
209
210 close(fds[0]);
211 close(fds[1]);
212 free(buf);
213 io_uring_queue_exit(&ring);
214 return T_EXIT_PASS;
215 }
216
main(int argc,char * argv[])217 int main(int argc, char *argv[])
218 {
219 int ret;
220
221 ret = test(0);
222 if (ret == T_EXIT_FAIL) {
223 fprintf(stderr, "test 0 failed\n");
224 return T_EXIT_FAIL;
225 }
226 if (no_buf_ring || no_buf_ring_status)
227 return T_EXIT_SKIP;
228
229 ret = test(1);
230 if (ret == T_EXIT_FAIL) {
231 fprintf(stderr, "test 1 failed\n");
232 return T_EXIT_FAIL;
233 }
234
235 ret = test_max();
236 if (ret == T_EXIT_FAIL) {
237 fprintf(stderr, "test_max failed\n");
238 return T_EXIT_FAIL;
239 }
240
241 return T_EXIT_PASS;
242 }
243