• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Copyright (c) 2015 Dmitry V. Levin <ldv@altlinux.org>
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. The name of the author may not be used to endorse or promote products
14  *    derived from this software without specific prior written permission.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
17  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
19  * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
20  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
21  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26  */
27 
28 #ifdef HAVE_CONFIG_H
29 # include "config.h"
30 #endif
31 
32 #include <assert.h>
33 #include <errno.h>
34 #include <inttypes.h>
35 #include <stdio.h>
36 #include <time.h>
37 #include <unistd.h>
38 #include <sys/fcntl.h>
39 #include <sys/syscall.h>
40 
41 #if defined __NR_io_setup \
42  && defined __NR_io_submit \
43  && defined __NR_io_getevents \
44  && defined __NR_io_cancel \
45  && defined __NR_io_destroy
46 # include <linux/aio_abi.h>
47 
48 #define ARRAY_SIZE(a) (sizeof(a) / sizeof((a)[0]))
49 
50 int
main(void)51 main(void)
52 {
53 	static char data0[4096];
54 	static char data1[8192];
55 
56 	const struct iocb cb[] = {
57 		{
58 			.aio_data = 0xfeedface11111111,
59 			.aio_reqprio = 11,
60 			.aio_buf = (unsigned long) data0,
61 			.aio_offset = 0xdeface1facefeed,
62 			.aio_nbytes = sizeof(data0)
63 		},
64 		{
65 			.aio_data = 0xfeedface22222222,
66 			.aio_reqprio = 22,
67 			.aio_buf = (unsigned long) data1,
68 			.aio_offset = 0xdeface2cafef00d,
69 			.aio_nbytes = sizeof(data1)
70 		}
71 	};
72 	const struct iovec iov0[] = {
73 		{
74 			.iov_base = data0,
75 			.iov_len = sizeof(data0) / 4
76 		},
77 		{
78 			.iov_base = data0 + sizeof(data0) / 4,
79 			.iov_len = sizeof(data0) / 4 * 3
80 		},
81 	};
82 	const struct iovec iov1[] = {
83 		{
84 			.iov_base = data1,
85 			.iov_len = sizeof(data1) / 4
86 		},
87 		{
88 			.iov_base = data1 + sizeof(data1) / 4,
89 			.iov_len = sizeof(data1) / 4 * 3
90 		},
91 	};
92 	const struct iocb cbv[] = {
93 		{
94 			.aio_data = 0xfeed11111111face,
95 			.aio_lio_opcode = 7,
96 			.aio_reqprio = 111,
97 			.aio_buf = (unsigned long) &iov0,
98 			.aio_offset = 0xdeface1facefeed,
99 			.aio_nbytes = ARRAY_SIZE(iov0)
100 		},
101 		{
102 			.aio_data = 0xfeed22222222face,
103 			.aio_lio_opcode = 7,
104 			.aio_reqprio = 222,
105 			.aio_buf = (unsigned long) &iov1,
106 			.aio_offset = 0xdeface2cafef00d,
107 			.aio_nbytes = ARRAY_SIZE(iov1)
108 		}
109 	};
110 	struct iocb cbc = {
111 		.aio_data = 0xdeadbeefbadc0ded,
112 		.aio_reqprio = 99,
113 		.aio_fildes = -42
114 	};
115 
116 	const long cbs[ARRAY_SIZE(cb) + 2] = {
117 		(long) &cb[0], (long) &cb[1],
118 		0xdeadbeef, 0xbadc0ded
119 	};
120 	const long cbvs[ARRAY_SIZE(cb) + 2] = {
121 		(long) &cbv[0], (long) &cbv[1],
122 		0xdeadbeef, 0xbadc0ded
123 	};
124 
125 	unsigned long ctx = 0;
126 	const unsigned int nr = ARRAY_SIZE(cb);
127 	const unsigned long lnr = (unsigned long) (0xdeadbeef00000000ULL | nr);
128 
129 	struct io_event ev[nr];
130 	const struct timespec ts = { .tv_nsec = 123456789 };
131 
132 	(void) close(0);
133 	if (open("/dev/zero", O_RDONLY))
134 		return 77;
135 
136 	if (syscall(__NR_io_setup, lnr, &ctx))
137 		return 77;
138 	printf("io_setup(%u, [%lu]) = 0\n", nr, ctx);
139 
140 	if (syscall(__NR_io_submit, ctx, nr, cbs) != (long) nr)
141 		return 77;
142 	printf("io_submit(%lu, %u, ["
143 		"{data=%#llx, pread, reqprio=11, fildes=0, "
144 			"buf=%p, nbytes=%u, offset=%lld}, "
145 		"{data=%#llx, pread, reqprio=22, fildes=0, "
146 			"buf=%p, nbytes=%u, offset=%lld}"
147 		"]) = %u\n",
148 	       ctx, nr,
149 	       (unsigned long long) cb[0].aio_data, data0,
150 	       (unsigned int) sizeof(data0), (long long) cb[0].aio_offset,
151 	       (unsigned long long) cb[1].aio_data, data1,
152 	       (unsigned int) sizeof(data1), (long long) cb[1].aio_offset,
153 	       nr);
154 
155 	assert(syscall(__NR_io_getevents, ctx, nr, nr + 1, ev, &ts) == (long) nr);
156 	printf("io_getevents(%lu, %u, %u, ["
157 		"{data=%#llx, obj=%p, res=%u, res2=0}, "
158 		"{data=%#llx, obj=%p, res=%u, res2=0}"
159 		"], {0, 123456789}) = %u\n",
160 	       ctx, nr, nr + 1,
161 	       (unsigned long long) cb[0].aio_data, &cb[0],
162 	       (unsigned int) sizeof(data0),
163 	       (unsigned long long) cb[1].aio_data, &cb[1],
164 	       (unsigned int) sizeof(data1),
165 	       nr);
166 
167 	assert(syscall(__NR_io_cancel, ctx, &cbc, ev) == -1 && EINVAL == errno);
168 	printf("io_cancel(%lu, {data=%#llx, pread, reqprio=99, fildes=-42}, %p) "
169 		"= -1 EINVAL (Invalid argument)\n",
170 	       ctx, (unsigned long long) cbc.aio_data, ev);
171 
172 	if (syscall(__NR_io_submit, ctx, nr, cbvs) != (long) nr)
173 		return 77;
174 	printf("io_submit(%lu, %u, ["
175 		"{data=%#llx, preadv, reqprio=%hd, fildes=0, "
176 			"iovec=[{%p, %u}, {%p, %u}], offset=%lld}, "
177 		"{data=%#llx, preadv, reqprio=%hd, fildes=0, "
178 			"iovec=[{%p, %u}, {%p, %u}], offset=%lld}"
179 		"]) = %u\n",
180 	       ctx, nr,
181 	       (unsigned long long) cbv[0].aio_data, cbv[0].aio_reqprio,
182 	       iov0[0].iov_base, (unsigned int) iov0[0].iov_len,
183 	       iov0[1].iov_base, (unsigned int) iov0[1].iov_len,
184 	       (long long) cbv[0].aio_offset,
185 	       (unsigned long long) cbv[1].aio_data, cbv[1].aio_reqprio,
186 	       iov1[0].iov_base, (unsigned int) iov1[0].iov_len,
187 	       iov1[1].iov_base, (unsigned int) iov1[1].iov_len,
188 	       (long long) cbv[1].aio_offset,
189 	       nr);
190 
191 	assert(syscall(__NR_io_destroy, ctx) == 0);
192 	printf("io_destroy(%lu) = 0\n", ctx);
193 
194 	puts("+++ exited with 0 +++");
195 	return 0;
196 }
197 
198 #else
199 
200 int
main(void)201 main(void)
202 {
203 	return 77;
204 }
205 
206 #endif
207