1 /* SPDX-License-Identifier: MIT */
2 /*
3 * Description: run various file registration tests
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 "helpers.h"
14 #include "liburing.h"
15
close_files(int * files,int nr_files,int add)16 static void close_files(int *files, int nr_files, int add)
17 {
18 char fname[32];
19 int i;
20
21 for (i = 0; i < nr_files; i++) {
22 if (files)
23 close(files[i]);
24 if (!add)
25 sprintf(fname, ".reg.%d", i);
26 else
27 sprintf(fname, ".add.%d", i + add);
28 unlink(fname);
29 }
30 if (files)
31 free(files);
32 }
33
open_files(int nr_files,int extra,int add)34 static int *open_files(int nr_files, int extra, int add)
35 {
36 char fname[32];
37 int *files;
38 int i;
39
40 files = t_calloc(nr_files + extra, sizeof(int));
41
42 for (i = 0; i < nr_files; i++) {
43 if (!add)
44 sprintf(fname, ".reg.%d", i);
45 else
46 sprintf(fname, ".add.%d", i + add);
47 files[i] = open(fname, O_RDWR | O_CREAT, 0644);
48 if (files[i] < 0) {
49 perror("open");
50 free(files);
51 files = NULL;
52 break;
53 }
54 }
55 if (extra) {
56 for (i = nr_files; i < nr_files + extra; i++)
57 files[i] = -1;
58 }
59
60 return files;
61 }
62
test_update_multiring(struct io_uring * r1,struct io_uring * r2,struct io_uring * r3,int do_unreg)63 static int test_update_multiring(struct io_uring *r1, struct io_uring *r2,
64 struct io_uring *r3, int do_unreg)
65 {
66 int *fds, *newfds;
67
68 fds = open_files(10, 0, 0);
69 newfds = open_files(10, 0, 1);
70
71 if (io_uring_register_files(r1, fds, 10) ||
72 io_uring_register_files(r2, fds, 10) ||
73 io_uring_register_files(r3, fds, 10)) {
74 fprintf(stderr, "%s: register files failed\n", __FUNCTION__);
75 goto err;
76 }
77
78 if (io_uring_register_files_update(r1, 0, newfds, 10) != 10 ||
79 io_uring_register_files_update(r2, 0, newfds, 10) != 10 ||
80 io_uring_register_files_update(r3, 0, newfds, 10) != 10) {
81 fprintf(stderr, "%s: update files failed\n", __FUNCTION__);
82 goto err;
83 }
84
85 if (!do_unreg)
86 goto done;
87
88 if (io_uring_unregister_files(r1) ||
89 io_uring_unregister_files(r2) ||
90 io_uring_unregister_files(r3)) {
91 fprintf(stderr, "%s: unregister files failed\n", __FUNCTION__);
92 goto err;
93 }
94
95 done:
96 close_files(fds, 10, 0);
97 close_files(newfds, 10, 1);
98 return 0;
99 err:
100 close_files(fds, 10, 0);
101 close_files(newfds, 10, 1);
102 return 1;
103 }
104
test_sqe_update(struct io_uring * ring)105 static int test_sqe_update(struct io_uring *ring)
106 {
107 struct io_uring_sqe *sqe;
108 struct io_uring_cqe *cqe;
109 int *fds, i, ret;
110
111 fds = t_malloc(sizeof(int) * 10);
112 for (i = 0; i < 10; i++)
113 fds[i] = -1;
114
115 sqe = io_uring_get_sqe(ring);
116 io_uring_prep_files_update(sqe, fds, 10, 0);
117 ret = io_uring_submit(ring);
118 if (ret != 1) {
119 fprintf(stderr, "submit: %d\n", ret);
120 return 1;
121 }
122
123 ret = io_uring_wait_cqe(ring, &cqe);
124 if (ret) {
125 fprintf(stderr, "wait: %d\n", ret);
126 return 1;
127 }
128
129 ret = cqe->res;
130 io_uring_cqe_seen(ring, cqe);
131 free(fds);
132 if (ret == -EINVAL) {
133 fprintf(stdout, "IORING_OP_FILES_UPDATE not supported, skipping\n");
134 return T_EXIT_SKIP;
135 }
136 return ret != 10;
137 }
138
test_update_no_table(void)139 static int test_update_no_table(void)
140 {
141 int up_fd, fds[4] = {-1, 0, 1, 4};
142 struct io_uring_sqe *sqe;
143 struct io_uring_cqe *cqe;
144 struct io_uring ring;
145 int ret;
146
147 ret = t_create_ring(2, &ring, 0);
148 if (ret == T_SETUP_SKIP)
149 return T_EXIT_SKIP;
150 else if (ret != T_SETUP_OK)
151 return ret;
152
153 ret = io_uring_register_files(&ring, fds, 4);
154 /* ignore other failures */
155 if (ret && ret != -EBADF) {
156 fprintf(stderr, "Failed registering file table: %d\n", ret);
157 goto fail;
158 }
159
160 sqe = io_uring_get_sqe(&ring);
161 up_fd = ring.ring_fd;
162 io_uring_prep_files_update(sqe, &up_fd, 1, -1); //offset = -1
163 ret = io_uring_submit(&ring);
164 if (ret != 1) {
165 fprintf(stderr, "Failed submit: %d\n", ret);
166 goto fail;
167 }
168
169 ret = io_uring_wait_cqe(&ring, &cqe);
170 if (ret) {
171 fprintf(stderr, "Failed wait: %d\n", ret);
172 goto fail;
173 }
174 ret = cqe->res;
175 io_uring_cqe_seen(&ring, cqe);
176 if (ret != -EMFILE && ret != -EINVAL && ret != -EOVERFLOW &&
177 ret != -ENXIO && ret != -EBADF) {
178 fprintf(stderr, "Bad cqe res: %d\n", ret);
179 goto fail;
180 }
181
182 io_uring_queue_exit(&ring);
183 return T_EXIT_PASS;
184 fail:
185 io_uring_queue_exit(&ring);
186 return T_EXIT_FAIL;
187 }
188
main(int argc,char * argv[])189 int main(int argc, char *argv[])
190 {
191 struct io_uring r1, r2, r3;
192 int ret;
193
194 if (argc > 1)
195 return T_EXIT_SKIP;
196
197 if (io_uring_queue_init(8, &r1, 0) ||
198 io_uring_queue_init(8, &r2, 0) ||
199 io_uring_queue_init(8, &r3, 0)) {
200 fprintf(stderr, "ring setup failed\n");
201 return 1;
202 }
203
204 ret = test_update_multiring(&r1, &r2, &r3, 1);
205 if (ret) {
206 fprintf(stderr, "test_update_multiring w/unreg\n");
207 return ret;
208 }
209
210 ret = test_update_multiring(&r1, &r2, &r3, 0);
211 if (ret) {
212 fprintf(stderr, "test_update_multiring wo/unreg\n");
213 return ret;
214 }
215
216 ret = test_sqe_update(&r1);
217 if (ret) {
218 if (ret != T_EXIT_SKIP)
219 fprintf(stderr, "test_sqe_update failed\n");
220 return ret;
221 }
222
223 ret = test_update_no_table();
224 if (ret) {
225 if (ret != T_EXIT_SKIP)
226 fprintf(stderr, "test_sqe_update failed\n");
227 return ret;
228 }
229
230 return T_EXIT_PASS;
231 }
232