1/* Tests for 64bit AIO in librt.
2 Copyright (C) 1998-2022 Free Software Foundation, Inc.
3 This file is part of the GNU C Library.
4
5 The GNU C Library is free software; you can redistribute it and/or
6 modify it under the terms of the GNU Lesser General Public
7 License as published by the Free Software Foundation; either
8 version 2.1 of the License, or (at your option) any later version.
9
10 The GNU C Library is distributed in the hope that it will be useful,
11 but WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 Lesser General Public License for more details.
14
15 You should have received a copy of the GNU Lesser General Public
16 License along with the GNU C Library; if not, see
17 <https://www.gnu.org/licenses/>. */
18
19#define _LARGEFILE_SOURCE 1
20#include <aio.h>
21#include <errno.h>
22#include <error.h>
23#include <fcntl.h>
24#include <stdlib.h>
25#include <string.h>
26#include <unistd.h>
27#include <sys/stat.h>
28
29
30/* Prototype for our test function. */
31extern void do_prepare (int argc, char *argv[]);
32extern int do_test (int argc, char *argv[]);
33
34/* We have a preparation function. */
35#define PREPARE do_prepare
36
37/* This defines the `main' function and some more. */
38#include <test-skeleton.c>
39
40
41/* These are for the temporary file we generate. */
42char *name;
43int fd;
44
45void
46do_prepare (int argc, char *argv[])
47{
48 size_t name_len;
49
50 name_len = strlen (test_dir);
51 name = xmalloc (n: name_len + sizeof ("/aioXXXXXX"));
52 mempcpy (mempcpy (name, test_dir, name_len),
53 "/aioXXXXXX", sizeof ("/aioXXXXXX"));
54
55 /* Open our test file. */
56 fd = mkstemp (template: name);
57 if (fd == -1)
58 error (EXIT_FAILURE, errno, format: "cannot open test file `%s'", name);
59 add_temp_file (name);
60}
61
62
63static int
64test_file (const void *buf, size_t size, int fd, const char *msg)
65{
66 struct stat st;
67 char tmp[size];
68
69 errno = 0;
70 if (fstat (fd: fd, buf: &st) < 0)
71 {
72 error (status: 0, errno, format: "%s: failed stat", msg);
73 return 1;
74 }
75
76 if (st.st_size != (off_t) size)
77 {
78 error (status: 0, errno, format: "%s: wrong size: %lu, should be %lu",
79 msg, (unsigned long int) st.st_size, (unsigned long int) size);
80 return 1;
81 }
82
83 if (pread (fd: fd, buf: tmp, nbytes: size, offset: 0) != (ssize_t) size)
84 {
85 error (status: 0, errno, format: "%s: failed pread", msg);
86 return 1;
87 }
88
89 if (memcmp (buf, tmp, size) != 0)
90 {
91 error (status: 0, errno, format: "%s: failed comparison", msg);
92 return 1;
93 }
94
95 printf (format: "%s test ok\n", msg);
96
97 return 0;
98}
99
100
101static int
102do_wait (struct aiocb64 **cbp, size_t nent, int allowed_err)
103{
104 int go_on;
105 size_t cnt;
106 int result = 0;
107
108 do
109 {
110 aio_suspend64 (list: (const struct aiocb64 *const *) cbp, nent: nent, NULL);
111 go_on = 0;
112 for (cnt = 0; cnt < nent; ++cnt)
113 if (cbp[cnt] != NULL)
114 {
115 if (aio_error64 (aiocbp: cbp[cnt]) == EINPROGRESS)
116 go_on = 1;
117 else
118 {
119 if (aio_return64 (aiocbp: cbp[cnt]) == -1
120 && (allowed_err == 0
121 || aio_error64 (aiocbp: cbp[cnt]) != allowed_err))
122 {
123 error (status: 0, errnum: aio_error64 (aiocbp: cbp[cnt]), format: "Operation failed\n");
124 result = 1;
125 }
126 cbp[cnt] = NULL;
127 }
128 }
129 }
130 while (go_on);
131
132 return result;
133}
134
135
136int
137do_test (int argc, char *argv[])
138{
139 struct aiocb64 cbs[10];
140 struct aiocb64 cbs_fsync;
141 struct aiocb64 *cbp[10];
142 struct aiocb64 *cbp_fsync[1];
143 char buf[1000];
144 size_t cnt;
145 int result = 0;
146
147 /* Preparation. */
148 for (cnt = 0; cnt < 10; ++cnt)
149 {
150 cbs[cnt].aio_fildes = fd;
151 cbs[cnt].aio_reqprio = 0;
152 cbs[cnt].aio_buf = memset (&buf[cnt * 100], '0' + cnt, 100);
153 cbs[cnt].aio_nbytes = 100;
154 cbs[cnt].aio_offset = cnt * 100;
155 cbs[cnt].aio_sigevent.sigev_notify = SIGEV_NONE;
156
157 cbp[cnt] = &cbs[cnt];
158 }
159
160 /* First a simple test. */
161 for (cnt = 10; cnt > 0; )
162 if (aio_write64 (aiocbp: cbp[--cnt]) < 0 && errno == ENOSYS)
163 {
164 error (status: 0, errnum: 0, format: "no aio support in this configuration");
165 return 0;
166 }
167 /* Wait 'til the results are there. */
168 result |= do_wait (cbp, nent: 10, allowed_err: 0);
169 /* Test this. */
170 result |= test_file (buf, size: sizeof (buf), fd, msg: "aio_write");
171
172 /* Read now as we've written it. */
173 memset (buf, '\0', sizeof (buf));
174 /* Issue the commands. */
175 for (cnt = 10; cnt > 0; )
176 {
177 --cnt;
178 cbp[cnt] = &cbs[cnt];
179 aio_read64 (aiocbp: cbp[cnt]);
180 }
181 /* Wait 'til the results are there. */
182 result |= do_wait (cbp, nent: 10, allowed_err: 0);
183 /* Test this. */
184 for (cnt = 0; cnt < 1000; ++cnt)
185 if (buf[cnt] != '0' + (cnt / 100))
186 {
187 result = 1;
188 error (status: 0, errnum: 0, format: "comparison failed for aio_read test");
189 break;
190 }
191
192 if (cnt == 1000)
193 puts (s: "aio_read test ok");
194
195 /* Remove the test file contents. */
196 if (ftruncate64 (fd: fd, length: 0) < 0)
197 {
198 error (status: 0, errno, format: "ftruncate failed\n");
199 result = 1;
200 }
201
202 /* Test lio_listio. */
203 for (cnt = 0; cnt < 10; ++cnt)
204 {
205 cbs[cnt].aio_lio_opcode = LIO_WRITE;
206 cbp[cnt] = &cbs[cnt];
207 }
208 /* Issue the command. */
209 lio_listio64 (LIO_WAIT, list: cbp, nent: 10, NULL);
210 /* ...and immediately test it since we started it in wait mode. */
211 result |= test_file (buf, size: sizeof (buf), fd, msg: "lio_listio (write)");
212
213 /* Test aio_fsync. */
214 cbs_fsync.aio_fildes = fd;
215 cbs_fsync.aio_sigevent.sigev_notify = SIGEV_NONE;
216 cbp_fsync[0] = &cbs_fsync;
217
218 /* Remove the test file contents first. */
219 if (ftruncate64 (fd: fd, length: 0) < 0)
220 {
221 error (status: 0, errno, format: "ftruncate failed\n");
222 result = 1;
223 }
224
225 /* Write again. */
226 for (cnt = 10; cnt > 0; )
227 aio_write64 (aiocbp: cbp[--cnt]);
228
229 if (aio_fsync64 (O_SYNC, aiocbp: &cbs_fsync) < 0)
230 {
231 error (status: 0, errno, format: "aio_fsync failed\n");
232 result = 1;
233 }
234 result |= do_wait (cbp: cbp_fsync, nent: 1, allowed_err: 0);
235
236 /* ...and test since all data should be on disk now. */
237 result |= test_file (buf, size: sizeof (buf), fd, msg: "aio_fsync (aio_write)");
238
239 /* Test aio_cancel. */
240 /* Remove the test file contents first. */
241 if (ftruncate64 (fd: fd, length: 0) < 0)
242 {
243 error (status: 0, errno, format: "ftruncate failed\n");
244 result = 1;
245 }
246
247 /* Write again. */
248 for (cnt = 10; cnt > 0; )
249 aio_write64 (aiocbp: cbp[--cnt]);
250
251 /* Cancel all requests. */
252 if (aio_cancel64 (fildes: fd, NULL) == -1)
253 printf (format: "aio_cancel64 (fd, NULL) cannot cancel anything\n");
254
255 result |= do_wait (cbp, nent: 10, ECANCELED);
256
257 /* Another test for aio_cancel. */
258 /* Remove the test file contents first. */
259 if (ftruncate64 (fd: fd, length: 0) < 0)
260 {
261 error (status: 0, errno, format: "ftruncate failed\n");
262 result = 1;
263 }
264
265 /* Write again. */
266 for (cnt = 10; cnt > 0; )
267 {
268 --cnt;
269 cbp[cnt] = &cbs[cnt];
270 aio_write64 (aiocbp: cbp[cnt]);
271 }
272 puts (s: "finished3");
273
274 /* Cancel all requests. */
275 for (cnt = 10; cnt > 0; )
276 if (aio_cancel64 (fildes: fd, aiocbp: cbp[--cnt]) == -1)
277 /* This is not an error. The request can simply be finished. */
278 printf (format: "aio_cancel64 (fd, cbp[%Zd]) cannot be canceled\n", cnt);
279 puts (s: "finished2");
280
281 result |= do_wait (cbp, nent: 10, ECANCELED);
282
283 puts (s: "finished");
284
285 return result;
286}
287

source code of glibc/rt/tst-aio64.c