xfstests: fix a few build warnings
[xfstests-dev.git] / src / aio-dio-regress / aiodio_sparse2.c
1 /* gcc -g -Wall -O2 aiodio_sparse.c -o aiodio_sparse -laio */
2
3 /*
4  * From http://developer.osdl.org/daniel/AIO/TESTS/aiodio_sparse.c
5  * With patch from https://bugzilla.redhat.com/attachment.cgi?id=142124
6  * (Bug https://bugzilla.redhat.com/show_bug.cgi?id=217098)
7  */
8
9 #ifndef _GNU_SOURCE
10 #define _GNU_SOURCE
11 #endif
12 #include <sys/types.h>
13 #include <errno.h>
14 #include <signal.h>
15 #include <fcntl.h>
16 #include <stdio.h>
17 #include <unistd.h>
18 #include <stdlib.h>
19 #include <limits.h>
20 #include <sys/mman.h>
21 #include <sys/wait.h>
22 #include <sys/ioctl.h>
23 #include <sys/mount.h>
24
25 #include <libaio.h>
26
27 int debug;
28
29 /*
30  * aiodio_sparse - issue async O_DIRECT writes to holes is a file while
31  *      concurrently reading the file and checking that the read never reads
32  *      uninitailized data.
33  */
34
35 unsigned char *check_zero(unsigned char *buf, int size)
36 {
37         unsigned char *p;
38
39         p = buf;
40
41         while (size > 0) {
42                 if (*buf != 1) {
43                         fprintf(stderr, "non one buffer at buf[%ld] => 0x%02x,%02x,%02x,%02x\n",
44                                 (long)(buf - p), (unsigned int)buf[0],
45                                 size > 1 ? (unsigned int)buf[1] : 0,
46                                 size > 2 ? (unsigned int)buf[2] : 0,
47                                 size > 3 ? (unsigned int)buf[3] : 0);
48                         if (debug)
49                                 fprintf(stderr, "buf %p, p %p\n", buf, p);
50                         return buf;
51                 }
52                 buf++;
53                 size--;
54         }
55         return 0;       /* all zeros */
56 }
57
58 volatile int got_signal;
59
60 void
61 sig_term_func(int i, siginfo_t *si, void *p)
62 {
63         if (debug)
64                 fprintf(stderr, "sig(%d, %p, %p)\n", i, si, p);
65         got_signal++;
66 }
67
68 /*
69  * do async DIO writes to a sparse file
70  */
71 void aiodio_sparse(char *filename, int align, int writesize, int startoffset, int filesize,
72                         int num_aio, int step, int sparse, int direct, int keep)
73 {
74         int fd;
75         void *bufptr;
76         int i;
77         int w;
78         static struct sigaction s;
79         struct iocb **iocbs;
80         off_t offset;
81         io_context_t myctx;
82         struct io_event event;
83         int aio_inflight;
84
85         s.sa_sigaction = sig_term_func;
86         s.sa_flags = SA_SIGINFO;
87         sigaction(SIGTERM, &s, 0);
88
89         if ((num_aio * step) > filesize) {
90                 num_aio = filesize / step;
91         }
92         memset(&myctx, 0, sizeof(myctx));
93         io_queue_init(num_aio, &myctx);
94
95         iocbs = (struct iocb **)calloc(num_aio, sizeof(struct iocb *));
96         for (i = 0; i < num_aio; i++) {
97                 if ((iocbs[i] = (struct iocb *)calloc(1, sizeof(struct iocb))) == 0) {
98                         perror("cannot malloc iocb");
99                         return;
100                 }
101         }
102
103         fd = open(filename, direct|O_WRONLY|O_CREAT, 0666);
104
105         if (fd < 0) {
106                 perror("cannot create file");
107                 return;
108         }
109
110         if (sparse)
111                 ftruncate(fd, filesize);
112
113         /*
114          * allocate the iocbs array and iocbs with buffers
115          */
116         offset = startoffset;
117         for (i = 0; i < num_aio; i++) {
118                 void *bufptr;
119
120                 if (posix_memalign(&bufptr, align, writesize)) {
121                         perror("cannot malloc aligned memory");
122                         close(fd);
123                         unlink(filename);
124                         return;
125                 }
126                 memset(bufptr, 1, writesize);
127                 io_prep_pwrite(iocbs[i], fd, bufptr, writesize, offset);
128                 offset += step;
129         }
130
131         /*
132          * start the 1st num_aio write requests
133          */
134         if ((w = io_submit(myctx, num_aio, iocbs)) < 0) {
135                 perror("io_submit failed");
136                 close(fd);
137                 unlink(filename);
138                 return;
139         }
140         if (debug)
141                 fprintf(stderr, "io_submit() return %d\n", w);
142
143         /*
144          * As AIO requests finish, keep issuing more AIO until done.
145          */
146         aio_inflight = num_aio;
147         if (debug)
148                 fprintf(stderr, "aiodio_sparse: %d i/o in flight\n", aio_inflight);
149         while (offset < filesize)  {
150                 int n;
151                 struct iocb *iocbp;
152
153                 if (debug)
154                         fprintf(stderr, "aiodio_sparse: offset %lld filesize %d inflight %d\n",
155                                 (long long)offset, filesize, aio_inflight);
156
157                 if ((n = io_getevents(myctx, 1, 1, &event, 0)) != 1) {
158                         if (-n != EINTR)
159                                 fprintf(stderr, "io_getevents() returned %d\n", n);
160                         break;
161                 }
162                 if (debug)
163                         fprintf(stderr, "aiodio_sparse: io_getevent() returned %d\n", n);
164                 aio_inflight--;
165                 if (got_signal)
166                         break;          /* told to stop */
167                 /*
168                  * check if write succeeded.
169                  */
170                 iocbp = event.obj;
171                 if (event.res2 != 0 || event.res != iocbp->u.c.nbytes) {
172                         fprintf(stderr,
173                                 "AIO write offset %lld expected %ld got %ld\n",
174                                 iocbp->u.c.offset, iocbp->u.c.nbytes,
175                                 event.res);
176                         break;
177                 }
178                 if (debug)
179                         fprintf(stderr, "aiodio_sparse: io_getevent() res %ld res2 %ld\n",
180                                 event.res, event.res2);
181
182                 /* start next write */
183                 io_prep_pwrite(iocbp, fd, iocbp->u.c.buf, writesize, offset);
184                 offset += step;
185                 if ((w = io_submit(myctx, 1, &iocbp)) < 0) {
186                         fprintf(stderr, "io_submit failed at offset %lld\n",
187                                 (long long)offset);
188                         perror("");
189                         break;
190                 }
191                 if (debug)
192                         fprintf(stderr, "io_submit() return %d\n", w);
193                 aio_inflight++;
194         }
195
196         /*
197          * wait for AIO requests in flight.
198          */
199         while (aio_inflight > 0) {
200                 int n;
201                 struct iocb *iocbp;
202
203                 if ((n = io_getevents(myctx, 1, 1, &event, 0)) != 1) {
204                         perror("io_getevents failed");
205                         break;
206                 }
207                 aio_inflight--;
208                 /*
209                  * check if write succeeded.
210                  */
211                 iocbp = event.obj;
212                 if (event.res2 != 0 || event.res != iocbp->u.c.nbytes) {
213                         fprintf(stderr,
214                                 "AIO write offset %lld expected %ld got %ld\n",
215                                 iocbp->u.c.offset, iocbp->u.c.nbytes,
216                                 event.res);
217                 }
218         }
219         if (debug)
220                 fprintf(stderr, "AIO DIO write done\n");
221         close(fd);
222         if ((fd = open(filename, O_RDONLY)) < 0)
223                 exit(1);
224
225         bufptr = malloc(writesize);
226         for (offset = startoffset; offset < filesize; offset += step)  {
227                 unsigned char *badbuf;
228
229                 if (debug)
230                         fprintf(stderr, "seek to %lld and read %d\n",
231                                 (long long) offset, writesize);
232                 lseek(fd, offset, SEEK_SET);
233                 if (read(fd, bufptr, writesize) < writesize) {
234                         fprintf(stderr, "short read() at offset %lld\n",
235                                 (long long) offset);
236                         exit(11);
237                 }
238                 if ((badbuf = check_zero(bufptr, writesize))) {
239                         fprintf(stderr, "non-one read at offset %lld\n",
240                                 (long long)(offset + badbuf - (unsigned char*)bufptr));
241                         fprintf(stderr, "*** WARNING *** %s has not been unlinked; if you don't rm it manually first, it may influence the next run\n", filename);
242                         exit(10);
243                 }
244         }
245         close(fd);
246         if (!keep)
247                 unlink(filename);
248         else
249                 fprintf(stderr, "*** WARNING *** You requested %s not to be unlinked; if you don't rm it manually first, it may influence the next run\n", filename);
250 }
251
252 void dirty_freeblocks(char *filename, int size)
253 {
254         int fd;
255         void *p;
256         int pg;
257         char filename2[PATH_MAX];
258
259         pg = getpagesize();
260         size = ((size + pg - 1) / pg) * pg;
261         sprintf(filename2, "%s.xx.%d", filename, getpid());
262         fd = open(filename2, O_CREAT|O_RDWR, 0666);
263         if (fd < 0) {
264                 perror("cannot open file");
265                 exit(2);
266         }
267         ftruncate(fd, size);
268         p = mmap(0, size, PROT_WRITE|PROT_READ, MAP_SHARED|MAP_FILE, fd, 0);
269         if (p == MAP_FAILED) {
270                 perror("cannot mmap");
271                 exit(2);
272         }
273         memset(p, 0xaa, size);
274         msync(p, size, MS_SYNC);
275         munmap(p, size);
276         close(fd);
277         unlink(filename2);
278 }
279
280 static long get_logical_block_size(char *filename)
281 {
282         int fd;
283         int ret;
284         int logical_block_size;
285
286         fd = open(filename, O_RDONLY);
287         if (fd < 0)
288                 return 4096;
289
290         ret = ioctl(fd, BLKSSZGET, &logical_block_size);
291         close(fd);
292         if (ret < 0)
293                 return 4096;
294         return logical_block_size;
295 }
296
297 void usage()
298 {
299         fprintf(stderr, "usage: dio_sparse [-n step] [-s filesize]"
300                 " [-w writesize] [-o startoffset] [-r readsize] filename\n");
301         exit(1);
302 }
303
304 /*
305  * Scale value by kilo, mega, or giga.
306  */
307 long long scale_by_kmg(long long value, char scale)
308 {
309         switch (scale) {
310         case 'g':
311         case 'G':
312                 value *= 1024;
313         case 'm':
314         case 'M':
315                 value *= 1024;
316         case 'k':
317         case 'K':
318                 value *= 1024;
319                 break;
320         case '\0':
321                 break;
322         default:
323                 usage();
324                 break;
325         }
326         return value;
327 }
328
329 /*
330  *      usage:
331  * aiodio_sparse [-r readsize] [-w writesize] [-o startoffset] [-n step] [-a align] [-i num_aio] filename
332  */
333
334 int main(int argc, char **argv)
335 {
336         char filename[PATH_MAX];
337         long alignment = 0;
338         int readsize = 65536;
339         int writesize = 65536;
340         int startoffset = 0;
341         int filesize = 100*1024*1024;
342         int num_aio = 16;
343         int step = 5*1024*1024;
344         int c, direct = O_DIRECT, keep = 0, sparse = 1;
345         extern char *optarg;
346         extern int optind, optopt, opterr;
347
348         while ((c = getopt(argc, argv, "dr:w:n:o:a:s:i:DkS")) != -1) {
349                 char *endp;
350                 switch (c) {
351                 case 'D':
352                         direct = 0;
353                         break;
354                 case 'k':
355                         keep = 1;
356                         break;
357                 case 'S':
358                         sparse = 0;
359                         break;
360                 case 'd':
361                         debug++;
362                         break;
363                 case 'i':
364                         num_aio = atoi(optarg);
365                         break;
366                 case 'a':
367                         alignment = strtol(optarg, &endp, 0);
368                         alignment = (int)scale_by_kmg((long long)alignment,
369                                                         *endp);
370                         break;
371                 case 'r':
372                         readsize = strtol(optarg, &endp, 0);
373                         readsize = (int)scale_by_kmg((long long)readsize, *endp);
374                         break;
375                 case 'w':
376                         writesize = strtol(optarg, &endp, 0);
377                         writesize = (int)scale_by_kmg((long long)writesize, *endp);
378                         break;
379                 case 's':
380                         filesize = strtol(optarg, &endp, 0);
381                         filesize = (int)scale_by_kmg((long long)filesize, *endp);
382                         break;
383                 case 'n':
384                         step = strtol(optarg, &endp, 0);
385                         step = (int)scale_by_kmg((long long)step, *endp);
386                         break;
387                 case 'o':
388                         startoffset = strtol(optarg, &endp, 0);
389                         startoffset = (int)scale_by_kmg((long long)startoffset, *endp);
390                         break;
391                 case '?':
392                         usage();
393                         break;
394                 }
395         }
396
397         strncpy(filename, argv[argc-1], PATH_MAX);
398
399         if (alignment == 0)
400                 alignment = get_logical_block_size(filename);
401         /*
402          * Create some dirty free blocks by allocating, writing, syncing,
403          * and then unlinking and freeing.
404          */
405         dirty_freeblocks(filename, filesize);
406
407         /*
408          * Parent write to a hole in a file using async direct i/o
409          */
410
411         aiodio_sparse(filename, alignment, writesize, startoffset, filesize, num_aio, step, sparse, direct, keep);
412
413         return 0;
414 }