1 /* SPDX-License-Identifier: MIT */
2 /*
3 * Description: check that STDOUT write works
4 */
5 #include <errno.h>
6 #include <stdio.h>
7 #include <unistd.h>
8 #include <stdlib.h>
9 #include <string.h>
10 #include <fcntl.h>
11
12 #include "helpers.h"
13 #include "liburing.h"
14
test_pipe_io_fixed(struct io_uring * ring)15 static int test_pipe_io_fixed(struct io_uring *ring)
16 {
17 const char str[] = "This is a fixed pipe test\n";
18 struct io_uring_cqe *cqe;
19 struct io_uring_sqe *sqe;
20 struct iovec vecs[2];
21 char buffer[128];
22 int i, ret, fds[2];
23
24 t_posix_memalign(&vecs[0].iov_base, 4096, 4096);
25 memcpy(vecs[0].iov_base, str, strlen(str));
26 vecs[0].iov_len = strlen(str);
27
28 if (pipe(fds) < 0) {
29 perror("pipe");
30 return 1;
31 }
32
33 ret = io_uring_register_buffers(ring, vecs, 1);
34 if (ret) {
35 fprintf(stderr, "Failed to register buffers: %d\n", ret);
36 return 1;
37 }
38
39 sqe = io_uring_get_sqe(ring);
40 if (!sqe) {
41 fprintf(stderr, "get sqe failed\n");
42 goto err;
43 }
44 io_uring_prep_write_fixed(sqe, fds[1], vecs[0].iov_base,
45 vecs[0].iov_len, 0, 0);
46 sqe->user_data = 1;
47
48 sqe = io_uring_get_sqe(ring);
49 if (!sqe) {
50 fprintf(stderr, "get sqe failed\n");
51 goto err;
52 }
53 vecs[1].iov_base = buffer;
54 vecs[1].iov_len = sizeof(buffer);
55 io_uring_prep_readv(sqe, fds[0], &vecs[1], 1, 0);
56 sqe->user_data = 2;
57
58 ret = io_uring_submit(ring);
59 if (ret < 0) {
60 fprintf(stderr, "sqe submit failed: %d\n", ret);
61 goto err;
62 } else if (ret != 2) {
63 fprintf(stderr, "Submitted only %d\n", ret);
64 goto err;
65 }
66
67 for (i = 0; i < 2; i++) {
68 ret = io_uring_wait_cqe(ring, &cqe);
69 if (ret < 0) {
70 fprintf(stderr, "wait completion %d\n", ret);
71 goto err;
72 }
73 if (cqe->res < 0) {
74 fprintf(stderr, "I/O write error on %lu: %s\n",
75 (unsigned long) cqe->user_data,
76 strerror(-cqe->res));
77 goto err;
78 }
79 if (cqe->res != strlen(str)) {
80 fprintf(stderr, "Got %d bytes, wanted %d on %lu\n",
81 cqe->res, (int)strlen(str),
82 (unsigned long) cqe->user_data);
83 goto err;
84 }
85 if (cqe->user_data == 2 && memcmp(str, buffer, strlen(str))) {
86 fprintf(stderr, "read data mismatch\n");
87 goto err;
88 }
89 io_uring_cqe_seen(ring, cqe);
90 }
91 io_uring_unregister_buffers(ring);
92 return 0;
93 err:
94 return 1;
95 }
96
test_stdout_io_fixed(struct io_uring * ring)97 static int test_stdout_io_fixed(struct io_uring *ring)
98 {
99 const char str[] = "This is a fixed pipe test\n";
100 struct io_uring_cqe *cqe;
101 struct io_uring_sqe *sqe;
102 struct iovec vecs;
103 int ret;
104
105 t_posix_memalign(&vecs.iov_base, 4096, 4096);
106 memcpy(vecs.iov_base, str, strlen(str));
107 vecs.iov_len = strlen(str);
108
109 ret = io_uring_register_buffers(ring, &vecs, 1);
110 if (ret) {
111 fprintf(stderr, "Failed to register buffers: %d\n", ret);
112 return 1;
113 }
114
115 sqe = io_uring_get_sqe(ring);
116 if (!sqe) {
117 fprintf(stderr, "get sqe failed\n");
118 goto err;
119 }
120 io_uring_prep_write_fixed(sqe, STDOUT_FILENO, vecs.iov_base, vecs.iov_len, 0, 0);
121
122 ret = io_uring_submit(ring);
123 if (ret < 0) {
124 fprintf(stderr, "sqe submit failed: %d\n", ret);
125 goto err;
126 } else if (ret < 1) {
127 fprintf(stderr, "Submitted only %d\n", ret);
128 goto err;
129 }
130
131 ret = io_uring_wait_cqe(ring, &cqe);
132 if (ret < 0) {
133 fprintf(stderr, "wait completion %d\n", ret);
134 goto err;
135 }
136 if (cqe->res < 0) {
137 fprintf(stderr, "STDOUT write error: %s\n", strerror(-cqe->res));
138 goto err;
139 }
140 if (cqe->res != vecs.iov_len) {
141 fprintf(stderr, "Got %d write, wanted %d\n", cqe->res, (int)vecs.iov_len);
142 goto err;
143 }
144 io_uring_cqe_seen(ring, cqe);
145 io_uring_unregister_buffers(ring);
146 return 0;
147 err:
148 return 1;
149 }
150
test_stdout_io(struct io_uring * ring)151 static int test_stdout_io(struct io_uring *ring)
152 {
153 struct io_uring_cqe *cqe;
154 struct io_uring_sqe *sqe;
155 struct iovec vecs;
156 int ret;
157
158 vecs.iov_base = "This is a pipe test\n";
159 vecs.iov_len = strlen(vecs.iov_base);
160
161 sqe = io_uring_get_sqe(ring);
162 if (!sqe) {
163 fprintf(stderr, "get sqe failed\n");
164 goto err;
165 }
166 io_uring_prep_writev(sqe, STDOUT_FILENO, &vecs, 1, 0);
167
168 ret = io_uring_submit(ring);
169 if (ret < 0) {
170 fprintf(stderr, "sqe submit failed: %d\n", ret);
171 goto err;
172 } else if (ret < 1) {
173 fprintf(stderr, "Submitted only %d\n", ret);
174 goto err;
175 }
176
177 ret = io_uring_wait_cqe(ring, &cqe);
178 if (ret < 0) {
179 fprintf(stderr, "wait completion %d\n", ret);
180 goto err;
181 }
182 if (cqe->res < 0) {
183 fprintf(stderr, "STDOUT write error: %s\n",
184 strerror(-cqe->res));
185 goto err;
186 }
187 if (cqe->res != vecs.iov_len) {
188 fprintf(stderr, "Got %d write, wanted %d\n", cqe->res,
189 (int)vecs.iov_len);
190 goto err;
191 }
192 io_uring_cqe_seen(ring, cqe);
193
194 return 0;
195 err:
196 return 1;
197 }
198
main(int argc,char * argv[])199 int main(int argc, char *argv[])
200 {
201 struct io_uring ring;
202 int ret;
203
204 if (argc > 1)
205 return 0;
206
207 ret = io_uring_queue_init(8, &ring, 0);
208 if (ret) {
209 fprintf(stderr, "ring setup failed\n");
210 return 1;
211 }
212
213 ret = test_stdout_io(&ring);
214 if (ret) {
215 fprintf(stderr, "test_pipe_io failed\n");
216 return ret;
217 }
218
219 ret = test_stdout_io_fixed(&ring);
220 if (ret) {
221 fprintf(stderr, "test_pipe_io_fixed failed\n");
222 return ret;
223 }
224
225 ret = test_pipe_io_fixed(&ring);
226 if (ret) {
227 fprintf(stderr, "test_pipe_io_fixed failed\n");
228 return ret;
229 }
230
231 return 0;
232 }
233