xref: /aosp_15_r20/external/liburing/test/open-direct-link.c (revision 25da2bea747f3a93b4c30fd9708b0618ef55a0e6)
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