1*25da2beaSAndroid Build Coastguard Worker /* SPDX-License-Identifier: MIT */
2*25da2beaSAndroid Build Coastguard Worker /*
3*25da2beaSAndroid Build Coastguard Worker * Description: link <open file><read from file><close file>
4*25da2beaSAndroid Build Coastguard Worker *
5*25da2beaSAndroid Build Coastguard Worker */
6*25da2beaSAndroid Build Coastguard Worker #include <errno.h>
7*25da2beaSAndroid Build Coastguard Worker #include <stdio.h>
8*25da2beaSAndroid Build Coastguard Worker #include <unistd.h>
9*25da2beaSAndroid Build Coastguard Worker #include <stdlib.h>
10*25da2beaSAndroid Build Coastguard Worker #include <string.h>
11*25da2beaSAndroid Build Coastguard Worker #include <fcntl.h>
12*25da2beaSAndroid Build Coastguard Worker
13*25da2beaSAndroid Build Coastguard Worker #include "liburing.h"
14*25da2beaSAndroid Build Coastguard Worker #include "helpers.h"
15*25da2beaSAndroid Build Coastguard Worker
16*25da2beaSAndroid Build Coastguard Worker #define MAX_FILES 8
17*25da2beaSAndroid Build Coastguard Worker #define FNAME ".link.direct"
18*25da2beaSAndroid Build Coastguard Worker
test(struct io_uring * ring,int skip_success,int drain,int async)19*25da2beaSAndroid Build Coastguard Worker static int test(struct io_uring *ring, int skip_success, int drain, int async)
20*25da2beaSAndroid Build Coastguard Worker {
21*25da2beaSAndroid Build Coastguard Worker struct io_uring_cqe *cqe;
22*25da2beaSAndroid Build Coastguard Worker struct io_uring_sqe *sqe;
23*25da2beaSAndroid Build Coastguard Worker char buf[4096];
24*25da2beaSAndroid Build Coastguard Worker int ret, i;
25*25da2beaSAndroid Build Coastguard Worker
26*25da2beaSAndroid Build Coastguard Worker /* drain and cqe skip are mutually exclusive */
27*25da2beaSAndroid Build Coastguard Worker if (skip_success && drain)
28*25da2beaSAndroid Build Coastguard Worker return 1;
29*25da2beaSAndroid Build Coastguard Worker
30*25da2beaSAndroid Build Coastguard Worker sqe = io_uring_get_sqe(ring);
31*25da2beaSAndroid Build Coastguard Worker io_uring_prep_openat_direct(sqe, AT_FDCWD, FNAME, O_RDONLY, 0, 0);
32*25da2beaSAndroid Build Coastguard Worker if (!drain)
33*25da2beaSAndroid Build Coastguard Worker sqe->flags |= IOSQE_IO_LINK;
34*25da2beaSAndroid Build Coastguard Worker if (skip_success)
35*25da2beaSAndroid Build Coastguard Worker sqe->flags |= IOSQE_CQE_SKIP_SUCCESS;
36*25da2beaSAndroid Build Coastguard Worker if (async)
37*25da2beaSAndroid Build Coastguard Worker sqe->flags |= IOSQE_ASYNC;
38*25da2beaSAndroid Build Coastguard Worker sqe->user_data = 1;
39*25da2beaSAndroid Build Coastguard Worker
40*25da2beaSAndroid Build Coastguard Worker sqe = io_uring_get_sqe(ring);
41*25da2beaSAndroid Build Coastguard Worker io_uring_prep_read(sqe, 0, buf, sizeof(buf), 0);
42*25da2beaSAndroid Build Coastguard Worker sqe->flags |= IOSQE_FIXED_FILE;
43*25da2beaSAndroid Build Coastguard Worker if (drain)
44*25da2beaSAndroid Build Coastguard Worker sqe->flags |= IOSQE_IO_DRAIN;
45*25da2beaSAndroid Build Coastguard Worker else
46*25da2beaSAndroid Build Coastguard Worker sqe->flags |= IOSQE_IO_LINK;
47*25da2beaSAndroid Build Coastguard Worker if (async)
48*25da2beaSAndroid Build Coastguard Worker sqe->flags |= IOSQE_ASYNC;
49*25da2beaSAndroid Build Coastguard Worker sqe->user_data = 2;
50*25da2beaSAndroid Build Coastguard Worker
51*25da2beaSAndroid Build Coastguard Worker sqe = io_uring_get_sqe(ring);
52*25da2beaSAndroid Build Coastguard Worker io_uring_prep_close_direct(sqe, 0);
53*25da2beaSAndroid Build Coastguard Worker sqe->user_data = 3;
54*25da2beaSAndroid Build Coastguard Worker if (skip_success)
55*25da2beaSAndroid Build Coastguard Worker sqe->flags |= IOSQE_CQE_SKIP_SUCCESS;
56*25da2beaSAndroid Build Coastguard Worker if (drain)
57*25da2beaSAndroid Build Coastguard Worker sqe->flags |= IOSQE_IO_DRAIN;
58*25da2beaSAndroid Build Coastguard Worker if (async)
59*25da2beaSAndroid Build Coastguard Worker sqe->flags |= IOSQE_ASYNC;
60*25da2beaSAndroid Build Coastguard Worker
61*25da2beaSAndroid Build Coastguard Worker ret = io_uring_submit(ring);
62*25da2beaSAndroid Build Coastguard Worker if (ret != 3) {
63*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "sqe submit failed: %d\n", ret);
64*25da2beaSAndroid Build Coastguard Worker goto err;
65*25da2beaSAndroid Build Coastguard Worker }
66*25da2beaSAndroid Build Coastguard Worker
67*25da2beaSAndroid Build Coastguard Worker if (skip_success) {
68*25da2beaSAndroid Build Coastguard Worker ret = io_uring_wait_cqe(ring, &cqe);
69*25da2beaSAndroid Build Coastguard Worker if (ret < 0) {
70*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "wait completion %d\n", ret);
71*25da2beaSAndroid Build Coastguard Worker goto err;
72*25da2beaSAndroid Build Coastguard Worker }
73*25da2beaSAndroid Build Coastguard Worker if (cqe->user_data != 2) {
74*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "Unexpected cqe %lu/%d\n",
75*25da2beaSAndroid Build Coastguard Worker (unsigned long) cqe->user_data,
76*25da2beaSAndroid Build Coastguard Worker cqe->res);
77*25da2beaSAndroid Build Coastguard Worker goto err;
78*25da2beaSAndroid Build Coastguard Worker }
79*25da2beaSAndroid Build Coastguard Worker if (cqe->res != sizeof(buf)) {
80*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "bad read %d\n", cqe->res);
81*25da2beaSAndroid Build Coastguard Worker goto err;
82*25da2beaSAndroid Build Coastguard Worker }
83*25da2beaSAndroid Build Coastguard Worker io_uring_cqe_seen(ring, cqe);
84*25da2beaSAndroid Build Coastguard Worker return 0;
85*25da2beaSAndroid Build Coastguard Worker }
86*25da2beaSAndroid Build Coastguard Worker
87*25da2beaSAndroid Build Coastguard Worker for (i = 0; i < 3; i++) {
88*25da2beaSAndroid Build Coastguard Worker ret = io_uring_wait_cqe(ring, &cqe);
89*25da2beaSAndroid Build Coastguard Worker if (ret < 0) {
90*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "wait completion %d\n", ret);
91*25da2beaSAndroid Build Coastguard Worker goto err;
92*25da2beaSAndroid Build Coastguard Worker }
93*25da2beaSAndroid Build Coastguard Worker switch (cqe->user_data) {
94*25da2beaSAndroid Build Coastguard Worker case 1:
95*25da2beaSAndroid Build Coastguard Worker if (cqe->res) {
96*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "bad open %d\n", cqe->res);
97*25da2beaSAndroid Build Coastguard Worker goto err;
98*25da2beaSAndroid Build Coastguard Worker }
99*25da2beaSAndroid Build Coastguard Worker break;
100*25da2beaSAndroid Build Coastguard Worker case 2:
101*25da2beaSAndroid Build Coastguard Worker if (cqe->res != sizeof(buf)) {
102*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "bad read %d\n", cqe->res);
103*25da2beaSAndroid Build Coastguard Worker goto err;
104*25da2beaSAndroid Build Coastguard Worker }
105*25da2beaSAndroid Build Coastguard Worker break;
106*25da2beaSAndroid Build Coastguard Worker case 3:
107*25da2beaSAndroid Build Coastguard Worker if (cqe->res) {
108*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "bad close %d\n", cqe->res);
109*25da2beaSAndroid Build Coastguard Worker goto err;
110*25da2beaSAndroid Build Coastguard Worker }
111*25da2beaSAndroid Build Coastguard Worker break;
112*25da2beaSAndroid Build Coastguard Worker }
113*25da2beaSAndroid Build Coastguard Worker io_uring_cqe_seen(ring, cqe);
114*25da2beaSAndroid Build Coastguard Worker }
115*25da2beaSAndroid Build Coastguard Worker
116*25da2beaSAndroid Build Coastguard Worker return 0;
117*25da2beaSAndroid Build Coastguard Worker err:
118*25da2beaSAndroid Build Coastguard Worker return 1;
119*25da2beaSAndroid Build Coastguard Worker }
120*25da2beaSAndroid Build Coastguard Worker
main(int argc,char * argv[])121*25da2beaSAndroid Build Coastguard Worker int main(int argc, char *argv[])
122*25da2beaSAndroid Build Coastguard Worker {
123*25da2beaSAndroid Build Coastguard Worker struct io_uring ring;
124*25da2beaSAndroid Build Coastguard Worker struct io_uring_params p = { };
125*25da2beaSAndroid Build Coastguard Worker int ret, files[MAX_FILES];
126*25da2beaSAndroid Build Coastguard Worker
127*25da2beaSAndroid Build Coastguard Worker if (argc > 1)
128*25da2beaSAndroid Build Coastguard Worker return 0;
129*25da2beaSAndroid Build Coastguard Worker
130*25da2beaSAndroid Build Coastguard Worker ret = io_uring_queue_init_params(8, &ring, &p);
131*25da2beaSAndroid Build Coastguard Worker if (ret) {
132*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "ring setup failed: %d\n", ret);
133*25da2beaSAndroid Build Coastguard Worker return 1;
134*25da2beaSAndroid Build Coastguard Worker }
135*25da2beaSAndroid Build Coastguard Worker if (!(p.features & IORING_FEAT_CQE_SKIP))
136*25da2beaSAndroid Build Coastguard Worker return 0;
137*25da2beaSAndroid Build Coastguard Worker
138*25da2beaSAndroid Build Coastguard Worker memset(files, -1, sizeof(files));
139*25da2beaSAndroid Build Coastguard Worker ret = io_uring_register_files(&ring, files, ARRAY_SIZE(files));
140*25da2beaSAndroid Build Coastguard Worker if (ret) {
141*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "Failed registering files\n");
142*25da2beaSAndroid Build Coastguard Worker return 1;
143*25da2beaSAndroid Build Coastguard Worker }
144*25da2beaSAndroid Build Coastguard Worker
145*25da2beaSAndroid Build Coastguard Worker t_create_file(FNAME, 4096);
146*25da2beaSAndroid Build Coastguard Worker
147*25da2beaSAndroid Build Coastguard Worker ret = test(&ring, 0, 0, 0);
148*25da2beaSAndroid Build Coastguard Worker if (ret) {
149*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "test 0 0 0 failed\n");
150*25da2beaSAndroid Build Coastguard Worker goto err;
151*25da2beaSAndroid Build Coastguard Worker }
152*25da2beaSAndroid Build Coastguard Worker
153*25da2beaSAndroid Build Coastguard Worker ret = test(&ring, 0, 1, 0);
154*25da2beaSAndroid Build Coastguard Worker if (ret) {
155*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "test 0 1 0 failed\n");
156*25da2beaSAndroid Build Coastguard Worker goto err;
157*25da2beaSAndroid Build Coastguard Worker }
158*25da2beaSAndroid Build Coastguard Worker
159*25da2beaSAndroid Build Coastguard Worker ret = test(&ring, 0, 0, 1);
160*25da2beaSAndroid Build Coastguard Worker if (ret) {
161*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "test 0 0 1 failed\n");
162*25da2beaSAndroid Build Coastguard Worker goto err;
163*25da2beaSAndroid Build Coastguard Worker }
164*25da2beaSAndroid Build Coastguard Worker
165*25da2beaSAndroid Build Coastguard Worker ret = test(&ring, 0, 1, 1);
166*25da2beaSAndroid Build Coastguard Worker if (ret) {
167*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "test 0 1 1 failed\n");
168*25da2beaSAndroid Build Coastguard Worker goto err;
169*25da2beaSAndroid Build Coastguard Worker }
170*25da2beaSAndroid Build Coastguard Worker
171*25da2beaSAndroid Build Coastguard Worker ret = test(&ring, 1, 0, 0);
172*25da2beaSAndroid Build Coastguard Worker if (ret) {
173*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "test 1 0 0 failed\n");
174*25da2beaSAndroid Build Coastguard Worker goto err;
175*25da2beaSAndroid Build Coastguard Worker }
176*25da2beaSAndroid Build Coastguard Worker
177*25da2beaSAndroid Build Coastguard Worker ret = test(&ring, 1, 0, 1);
178*25da2beaSAndroid Build Coastguard Worker if (ret) {
179*25da2beaSAndroid Build Coastguard Worker fprintf(stderr, "test 1 0 1 failed\n");
180*25da2beaSAndroid Build Coastguard Worker goto err;
181*25da2beaSAndroid Build Coastguard Worker }
182*25da2beaSAndroid Build Coastguard Worker
183*25da2beaSAndroid Build Coastguard Worker unlink(FNAME);
184*25da2beaSAndroid Build Coastguard Worker return 0;
185*25da2beaSAndroid Build Coastguard Worker err:
186*25da2beaSAndroid Build Coastguard Worker unlink(FNAME);
187*25da2beaSAndroid Build Coastguard Worker return 1;
188*25da2beaSAndroid Build Coastguard Worker }
189