1 /*
2 * Copyright (c) 2015-2016 Dmitry V. Levin <ldv@altlinux.org>
3 * Copyright (c) 2015-2018 The strace developers.
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 * 3. The name of the author may not be used to endorse or promote products
15 * derived from this software without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29 #include "tests.h"
30 #include <fcntl.h>
31 #include <inttypes.h>
32 #include <stdio.h>
33 #include <time.h>
34 #include <unistd.h>
35 #include <asm/unistd.h>
36
37 #if defined __NR_io_setup \
38 && defined __NR_io_submit \
39 && defined __NR_io_getevents \
40 && defined __NR_io_cancel \
41 && defined __NR_io_destroy
42 # include <linux/aio_abi.h>
43
44 int
main(void)45 main(void)
46 {
47 static const long bogus_ctx =
48 (long) 0xface1e55deadbeefLL;
49
50 static const char data2[] =
51 "\0\1\2\3cat test test test 0123456789abcdef";
52
53 const unsigned int sizeof_data0 = 4096;
54 const unsigned int sizeof_data1 = 8192;
55 void *data0 = tail_alloc(sizeof_data0);
56 void *data1 = tail_alloc(sizeof_data1);
57
58 const struct iocb proto_cb[] = {
59 {
60 .aio_data = (unsigned long) 0xfeedface11111111ULL,
61 .aio_reqprio = 11,
62 .aio_buf = (unsigned long) data0,
63 .aio_offset = (unsigned long) 0xdeface1facefeedULL,
64 .aio_nbytes = sizeof_data0
65 },
66 {
67 .aio_data = (unsigned long) 0xfeedface22222222ULL,
68 .aio_reqprio = 22,
69 .aio_buf = (unsigned long) data1,
70 .aio_offset = (unsigned long) 0xdeface2cafef00dULL,
71 .aio_nbytes = sizeof_data1
72 }
73 };
74 const struct iocb *cb = tail_memdup(proto_cb, sizeof(proto_cb));
75
76 const struct iovec proto_iov0[] = {
77 {
78 .iov_base = data0,
79 .iov_len = sizeof_data0 / 4
80 },
81 {
82 .iov_base = data0 + sizeof_data0 / 4,
83 .iov_len = sizeof_data0 / 4 * 3
84 },
85 };
86 const struct iovec *iov0 = tail_memdup(proto_iov0, sizeof(proto_iov0));
87
88 const struct iovec proto_iov1[] = {
89 {
90 .iov_base = data1,
91 .iov_len = sizeof_data1 / 4
92 },
93 {
94 .iov_base = data1 + sizeof_data1 / 4,
95 .iov_len = sizeof_data1 / 4 * 3
96 },
97 };
98 const struct iovec *iov1 = tail_memdup(proto_iov1, sizeof(proto_iov1));
99
100 const struct iocb proto_cbv[] = {
101 {
102 .aio_data = (unsigned long) 0xfeed11111111faceULL,
103 .aio_lio_opcode = 7,
104 .aio_reqprio = 111,
105 .aio_buf = (unsigned long) iov0,
106 .aio_offset = (unsigned long) 0xdeface1facefeedULL,
107 .aio_nbytes = ARRAY_SIZE(proto_iov0)
108 },
109 {
110 .aio_data = (unsigned long) 0xfeed22222222faceULL,
111 .aio_lio_opcode = 7,
112 .aio_reqprio = 222,
113 .aio_buf = (unsigned long) iov1,
114 .aio_offset = (unsigned long) 0xdeface2cafef00dULL,
115 .aio_nbytes = ARRAY_SIZE(proto_iov1)
116 }
117 };
118 const struct iocb *cbv = tail_memdup(proto_cbv, sizeof(proto_cbv));
119
120 /* For additional decoder testing */
121 const struct iocb proto_cbv2[] = {
122 {
123 .aio_data = 0xbadfacedc0ffeeedULL,
124 .aio_key = 0xdefaced0,
125 .aio_lio_opcode = 0xf00d,
126 .aio_reqprio = 0,
127 .aio_fildes = 0xdefaced1,
128 .aio_buf = 0,
129 },
130 {
131 .aio_data = 0,
132 .aio_key = 0xdefaced0,
133 .aio_lio_opcode = 1,
134 .aio_reqprio = 0xbeef,
135 .aio_fildes = 0xdefaced1,
136 .aio_buf = 0,
137 /* In order to make record valid */
138 .aio_nbytes = (size_t) 0x1020304050607080ULL,
139 .aio_offset = 0xdeadda7abadc0dedULL,
140 # ifdef IOCB_FLAG_RESFD
141 .aio_flags = 0xfacef157,
142 .aio_resfd = 0xded1ca7e,
143 # endif
144 },
145 {
146 .aio_data = 0,
147 .aio_key = 0xdefaced0,
148 .aio_lio_opcode = 1,
149 .aio_reqprio = 0xbeef,
150 .aio_fildes = 0xdefaced1,
151 .aio_buf = 0xbadc0ffeedefacedULL,
152 .aio_nbytes = 0x8090a0b0c0d0e0f0ULL,
153 .aio_offset = 0xdeadda7abadc0dedULL,
154 },
155 {
156 .aio_data = 0,
157 .aio_key = 0xdefaced0,
158 .aio_lio_opcode = 1,
159 .aio_reqprio = 0xbeef,
160 .aio_fildes = 0xdefaced1,
161 .aio_buf = (unsigned long)data2,
162 .aio_nbytes = sizeof(data2),
163 .aio_offset = 0xdeadda7abadc0dedULL,
164 },
165 {
166 .aio_data = 0,
167 .aio_key = 0xdefaced0,
168 .aio_lio_opcode = 8,
169 .aio_reqprio = 0xbeef,
170 .aio_fildes = 0xdefaced1,
171 .aio_buf = 0,
172 .aio_nbytes = 0x8090a0b0c0d0e0f0ULL,
173 .aio_offset = 0xdeadda7abadc0dedULL,
174 },
175 };
176 const struct iocb *cbv2 = tail_memdup(proto_cbv2, sizeof(proto_cbv2));
177
178 const struct iocb proto_cbc = {
179 .aio_data = (unsigned long) 0xdeadbeefbadc0dedULL,
180 .aio_reqprio = 99,
181 .aio_fildes = -42
182 };
183 const struct iocb *cbc = tail_memdup(&proto_cbc, sizeof(proto_cbc));
184
185 const long proto_cbs[] = {
186 (long) &cb[0], (long) &cb[1]
187 };
188 const long *cbs = tail_memdup(proto_cbs, sizeof(proto_cbs));
189
190 const long proto_cbvs[] = {
191 (long) &cbv[0], (long) &cbv[1],
192 };
193 const long *cbvs = tail_memdup(proto_cbvs, sizeof(proto_cbvs));
194
195 const long proto_cbvs2[] = {
196 (long) &cbv2[0], (long) &cbv2[1], (long) &cbv2[2],
197 (long) &cbv2[3], (long) &cbv2[4],
198 (long) NULL, (long) 0xffffffffffffffffLL,
199 };
200 const long *cbvs2 = tail_memdup(proto_cbvs2, sizeof(proto_cbvs2));
201
202 TAIL_ALLOC_OBJECT_CONST_PTR(unsigned long, ctx);
203 *ctx = 0;
204
205 const unsigned int nr = ARRAY_SIZE(proto_cb);
206 const unsigned long lnr = (unsigned long) (0xdeadbeef00000000ULL | nr);
207
208 const struct io_event *ev = tail_alloc(nr * sizeof(struct io_event));
209 TAIL_ALLOC_OBJECT_CONST_PTR(struct timespec, ts);
210
211 (void) close(0);
212 if (open("/dev/zero", O_RDONLY))
213 perror_msg_and_skip("open: %s", "/dev/zero");
214
215 long rc = syscall(__NR_io_setup, 0xdeadbeef, NULL);
216 printf("io_setup(%u, NULL) = %s\n", 0xdeadbeef, sprintrc(rc));
217
218 rc = syscall(__NR_io_setup, lnr, ctx + 1);
219 printf("io_setup(%u, %p) = %s\n", nr, ctx + 1, sprintrc(rc));
220
221 if (syscall(__NR_io_setup, lnr, ctx))
222 perror_msg_and_skip("io_setup");
223 printf("io_setup(%u, [%#lx]) = 0\n", nr, *ctx);
224
225 rc = syscall(__NR_io_submit, bogus_ctx, (long) 0xca7faceddeadf00dLL,
226 NULL);
227 printf("io_submit(%#lx, %ld, NULL) = %s\n",
228 bogus_ctx, (long) 0xca7faceddeadf00dLL, sprintrc(rc));
229
230 rc = syscall(__NR_io_submit, *ctx, nr, cbs + nr);
231 printf("io_submit(%#lx, %ld, %p) = %s\n",
232 *ctx, (long) nr, cbs + nr, sprintrc(rc));
233
234 rc = syscall(__NR_io_submit, *ctx, -1L, cbs);
235 printf("io_submit(%#lx, -1, %p) = %s\n",
236 *ctx, cbs, sprintrc(rc));
237
238 rc = syscall(__NR_io_submit, *ctx, nr, cbs);
239 if (rc != (long) nr)
240 perror_msg_and_skip("io_submit");
241 printf("io_submit(%#lx, %u, ["
242 "{aio_data=%#" PRI__x64 ", aio_lio_opcode=IOCB_CMD_PREAD"
243 ", aio_reqprio=11, aio_fildes=0, aio_buf=%p, aio_nbytes=%u"
244 ", aio_offset=%" PRI__d64
245 "}, {aio_data=%#" PRI__x64 ", aio_lio_opcode=IOCB_CMD_PREAD"
246 ", aio_reqprio=22, aio_fildes=0, aio_buf=%p, aio_nbytes=%u"
247 ", aio_offset=%" PRI__d64 "}]) = %s\n",
248 *ctx, nr,
249 cb[0].aio_data, data0, sizeof_data0, cb[0].aio_offset,
250 cb[1].aio_data, data1, sizeof_data1, cb[1].aio_offset,
251 sprintrc(rc));
252
253 rc = syscall(__NR_io_getevents, bogus_ctx,
254 (long) 0xca7faceddeadf00dLL, (long) 0xba5e1e505ca571e0LL,
255 ev + 1, NULL);
256 printf("io_getevents(%#lx, %ld, %ld, %p, NULL) = %s\n",
257 bogus_ctx, (long) 0xca7faceddeadf00dLL,
258 (long) 0xba5e1e505ca571e0LL, ev + 1, sprintrc(rc));
259
260 rc = syscall(__NR_io_getevents, bogus_ctx,
261 (long) 0xca7faceddeadf00dLL, (long) 0xba5e1e505ca571e0LL,
262 NULL, ts + 1);
263 printf("io_getevents(%#lx, %ld, %ld, NULL, %p) = %s\n",
264 bogus_ctx, (long) 0xca7faceddeadf00dLL,
265 (long) 0xba5e1e505ca571e0LL, ts + 1, sprintrc(rc));
266
267 ts->tv_sec = 0xdeadbeefU;
268 ts->tv_nsec = 0xfacefeedU;
269 rc = syscall(__NR_io_getevents, bogus_ctx, 0, 0, 0, ts);
270 printf("io_getevents(%#lx, 0, 0, NULL"
271 ", {tv_sec=%lld, tv_nsec=%llu}) = %s\n",
272 bogus_ctx, (long long) ts->tv_sec,
273 zero_extend_signed_to_ull(ts->tv_nsec), sprintrc(rc));
274
275 ts->tv_sec = (time_t) 0xcafef00ddeadbeefLL;
276 ts->tv_nsec = (long) 0xbadc0dedfacefeedLL;
277 rc = syscall(__NR_io_getevents, bogus_ctx, 0, 0, 0, ts);
278 printf("io_getevents(%#lx, 0, 0, NULL"
279 ", {tv_sec=%lld, tv_nsec=%llu}) = %s\n",
280 bogus_ctx, (long long) ts->tv_sec,
281 zero_extend_signed_to_ull(ts->tv_nsec), sprintrc(rc));
282
283 ts->tv_sec = 0;
284 ts->tv_nsec = 123456789;
285 rc = syscall(__NR_io_getevents, *ctx, nr, nr + 1, ev, ts);
286 printf("io_getevents(%#lx, %ld, %ld, ["
287 "{data=%#" PRI__x64 ", obj=%p, res=%u, res2=0}, "
288 "{data=%#" PRI__x64 ", obj=%p, res=%u, res2=0}"
289 "], {tv_sec=0, tv_nsec=123456789}) = %s\n",
290 *ctx, (long) nr, (long) (nr + 1),
291 cb[0].aio_data, &cb[0], sizeof_data0,
292 cb[1].aio_data, &cb[1], sizeof_data1,
293 sprintrc(rc));
294
295 rc = syscall(__NR_io_cancel, bogus_ctx, NULL, NULL);
296 printf("io_cancel(%#lx, NULL, NULL) = %s\n", bogus_ctx, sprintrc(rc));
297
298 rc = syscall(__NR_io_cancel, *ctx, cbc + 1, ev);
299 printf("io_cancel(%#lx, %p, %p) = %s\n", *ctx, cbc + 1, ev,
300 sprintrc(rc));
301
302 rc = syscall(__NR_io_cancel, *ctx, cbc, ev);
303 printf("io_cancel(%#lx, {aio_data=%#" PRI__x64
304 ", aio_lio_opcode=IOCB_CMD_PREAD, aio_reqprio=99"
305 ", aio_fildes=-42}, %p) = %s\n",
306 *ctx, cbc->aio_data, ev, sprintrc(rc));
307
308 rc = syscall(__NR_io_submit, (unsigned long) 0xfacef157beeff00dULL,
309 (long) 0xdeadc0defacefeedLL, NULL);
310 printf("io_submit(%#lx, %ld, NULL) = %s\n",
311 (long) 0xfacef157beeff00dULL,
312 (long) 0xdeadc0defacefeedLL, sprintrc(rc));
313
314 rc = syscall(__NR_io_submit, *ctx, -1L, cbvs + nr);
315 printf("io_submit(%#lx, %ld, %p) = %s\n",
316 *ctx, -1L, cbvs + nr, sprintrc(rc));
317
318 printf("io_submit(%#lx, %ld, ["
319 "{aio_data=%#" PRI__x64 ", aio_key=%u"
320 ", aio_lio_opcode=%hu /* IOCB_CMD_??? */, aio_fildes=%d}"
321 ", {aio_key=%u, aio_lio_opcode=IOCB_CMD_PWRITE, aio_reqprio=%hd"
322 ", aio_fildes=%d, aio_buf=NULL"
323 ", aio_nbytes=%" PRI__u64 ", aio_offset=%" PRI__d64
324 # ifdef IOCB_FLAG_RESFD
325 ", aio_resfd=%d, aio_flags=%#x"
326 # endif
327 "}, {aio_key=%u, aio_lio_opcode=IOCB_CMD_PWRITE"
328 ", aio_reqprio=%hd, aio_fildes=%d, aio_buf=%#" PRI__x64
329 ", aio_nbytes=%" PRI__u64 ", aio_offset=%" PRI__d64
330 "}, {aio_key=%u, aio_lio_opcode=IOCB_CMD_PWRITE"
331 ", aio_reqprio=%hd, aio_fildes=%d"
332 ", aio_buf=\"\\0\\1\\2\\3%.28s\"..."
333 ", aio_nbytes=%" PRI__u64 ", aio_offset=%" PRI__d64
334 "}, {aio_key=%u, aio_lio_opcode=IOCB_CMD_PWRITEV"
335 ", aio_reqprio=%hd, aio_fildes=%d, aio_buf=%#" PRI__x64
336 ", aio_nbytes=%" PRI__u64 ", aio_offset=%" PRI__d64
337 "}, NULL, %#lx, ... /* %p */]) = ",
338 *ctx, 1057L,
339 cbv2[0].aio_data, cbv2[0].aio_key,
340 cbv2[0].aio_lio_opcode, cbv2[0].aio_fildes,
341 cbv2[1].aio_key, cbv2[1].aio_reqprio, cbv2[1].aio_fildes,
342 cbv2[1].aio_nbytes, cbv2[1].aio_offset,
343 # ifdef IOCB_FLAG_RESFD
344 cbv2[1].aio_resfd, cbv2[1].aio_flags,
345 # endif
346 cbv2[2].aio_key, cbv2[2].aio_reqprio, cbv2[2].aio_fildes,
347 cbv2[2].aio_buf, cbv2[2].aio_nbytes, cbv2[2].aio_offset,
348 cbv2[3].aio_key, cbv2[3].aio_reqprio, cbv2[3].aio_fildes,
349 data2 + 4, cbv2[3].aio_nbytes, cbv2[3].aio_offset,
350 cbv2[4].aio_key, cbv2[4].aio_reqprio, cbv2[4].aio_fildes,
351 cbv2[4].aio_buf, cbv2[4].aio_nbytes, cbv2[4].aio_offset,
352 cbvs2[6], cbvs2 + 7);
353 rc = syscall(__NR_io_submit, *ctx, 1057L, cbvs2);
354 puts(sprintrc(rc));
355
356 rc = syscall(__NR_io_submit, *ctx, nr, cbvs);
357 if (rc != (long) nr)
358 perror_msg_and_skip("io_submit");
359 printf("io_submit(%#lx, %u, ["
360 "{aio_data=%#" PRI__x64 ", aio_lio_opcode=IOCB_CMD_PREADV"
361 ", aio_reqprio=%hd, aio_fildes=0, "
362 "aio_buf=[{iov_base=%p, iov_len=%u}"
363 ", {iov_base=%p, iov_len=%u}], aio_offset=%" PRI__d64 "}, "
364 "{aio_data=%#" PRI__x64 ", aio_lio_opcode=IOCB_CMD_PREADV"
365 ", aio_reqprio=%hd, aio_fildes=0"
366 ", aio_buf=[{iov_base=%p, iov_len=%u}"
367 ", {iov_base=%p, iov_len=%u}], aio_offset=%" PRI__d64 "}"
368 "]) = %s\n",
369 *ctx, nr,
370 cbv[0].aio_data, cbv[0].aio_reqprio,
371 iov0[0].iov_base, (unsigned int) iov0[0].iov_len,
372 iov0[1].iov_base, (unsigned int) iov0[1].iov_len,
373 cbv[0].aio_offset,
374 cbv[1].aio_data, cbv[1].aio_reqprio,
375 iov1[0].iov_base, (unsigned int) iov1[0].iov_len,
376 iov1[1].iov_base, (unsigned int) iov1[1].iov_len,
377 cbv[1].aio_offset,
378 sprintrc(rc));
379
380 rc = syscall(__NR_io_destroy, bogus_ctx);
381 printf("io_destroy(%#lx) = %s\n",
382 bogus_ctx, sprintrc(rc));
383
384 rc = syscall(__NR_io_destroy, *ctx);
385 printf("io_destroy(%#lx) = %s\n", *ctx, sprintrc(rc));
386
387 puts("+++ exited with 0 +++");
388 return 0;
389 }
390
391 #else
392
393 SKIP_MAIN_UNDEFINED("__NR_io_*")
394
395 #endif
396