1 /* SPDX-License-Identifier: MIT */
2 /*
3 * Check split up read is handled correctly
4 */
5 #include <errno.h>
6 #include <stdio.h>
7 #include <string.h>
8 #include <unistd.h>
9 #include <pthread.h>
10 #include <string.h>
11 #include "liburing.h"
12
13 #define BUFSIZE 16384
14 #define BUFFERS 16
15
main(int argc,char * argv[])16 int main(int argc, char *argv[])
17 {
18 char buf[BUFSIZE], wbuf[BUFSIZE];
19 struct iovec iov[BUFFERS];
20 struct io_uring_params p = { };
21 struct io_uring ring;
22 struct io_uring_sqe *sqe;
23 struct io_uring_cqe *cqe;
24 int ret, i, fds[2];
25 void *ptr;
26
27 if (pipe(fds) < 0) {
28 perror("pipe");
29 return 1;
30 }
31
32 ptr = buf;
33 for (i = 0; i < BUFFERS; i++) {
34 unsigned bsize = BUFSIZE / BUFFERS;
35
36 iov[i].iov_base = ptr;
37 iov[i].iov_len = bsize;
38 ptr += bsize;
39 }
40
41 ret = io_uring_queue_init_params(8, &ring, &p);
42 if (ret) {
43 fprintf(stderr, "queue_init: %d\n", ret);
44 return 1;
45 }
46 if (!(p.features & IORING_FEAT_SUBMIT_STABLE)) {
47 fprintf(stdout, "FEAT_SUBMIT_STABLE not there, skipping\n");
48 return 0;
49 }
50
51 ptr = wbuf;
52 memset(ptr, 0x11, sizeof(wbuf) / 2);
53 ptr += sizeof(wbuf) / 2;
54 memset(ptr, 0x22, sizeof(wbuf) / 2);
55
56 ret = write(fds[1], wbuf, sizeof(wbuf) / 2);
57 if (ret != sizeof(wbuf) / 2) {
58 fprintf(stderr, "Bad write\n");
59 ret = 1;
60 goto err;
61 }
62
63 sqe = io_uring_get_sqe(&ring);
64 io_uring_prep_readv(sqe, fds[0], iov, BUFFERS, 0);
65 ret = io_uring_submit(&ring);
66 if (ret != 1) {
67 fprintf(stderr, "submit: %d\n", ret);
68 return 1;
69 }
70
71 for (i = 0; i < BUFFERS; i++) {
72 iov[i].iov_base = NULL;
73 iov[i].iov_len = 1000000;
74 }
75
76 ret = write(fds[1], ptr, sizeof(wbuf) / 2);
77 if (ret != sizeof(wbuf) / 2) {
78 fprintf(stderr, "Bad write\n");
79 ret = 1;
80 goto err;
81 }
82
83 ret = io_uring_wait_cqe(&ring, &cqe);
84 if (ret) {
85 fprintf(stderr, "wait: %d\n", ret);
86 return 1;
87 }
88
89 if (cqe->res < 0) {
90 fprintf(stderr, "Read error: %s\n", strerror(-cqe->res));
91 return 1;
92 } else if (cqe->res != sizeof(wbuf)) {
93 /* ignore short read, not a failure */
94 goto err;
95 }
96 io_uring_cqe_seen(&ring, cqe);
97
98 ret = memcmp(wbuf, buf, sizeof(wbuf));
99 if (ret)
100 fprintf(stderr, "Read data mismatch\n");
101
102 err:
103 io_uring_queue_exit(&ring);
104 return ret;
105 }
106