xref: /freebsd/tests/sys/aio/aio_kqueue_test.c (revision 325151a3)
1 /*-
2  * Copyright (C) 2005 IronPort Systems, Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
14  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
16  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
17  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
18  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
19  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
20  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
21  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
22  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
23  * SUCH DAMAGE.
24  *
25  * $FreeBSD$
26  */
27 
28 /*
29  * Prerequisities:
30  * - AIO support must be compiled into the kernel (see sys/<arch>/NOTES for
31  *   more details).
32  *
33  * Note: it is a good idea to run this against a physical drive to
34  * exercise the physio fast path (ie. aio_kqueue /dev/<something safe>)
35  */
36 
37 #include <sys/types.h>
38 #include <sys/event.h>
39 #include <sys/time.h>
40 #include <aio.h>
41 #include <err.h>
42 #include <errno.h>
43 #include <fcntl.h>
44 #include <stdlib.h>
45 #include <stdio.h>
46 #include <string.h>
47 #include <unistd.h>
48 
49 #include "freebsd_test_suite/macros.h"
50 
51 #define PATH_TEMPLATE   "aio.XXXXXXXXXX"
52 
53 #define MAX_IOCBS 128
54 #define MAX_RUNS 300
55 /* #define DEBUG */
56 
57 int
58 main (int argc, char *argv[])
59 {
60 	struct aiocb *iocb[MAX_IOCBS], *kq_iocb;
61 	char *file, pathname[sizeof(PATH_TEMPLATE)+1];
62 	struct kevent ke, kq_returned;
63 	struct timespec ts;
64 	char buffer[32768];
65 	int cancel, error, failed = 0, fd, kq, pending, result, run;
66 	int tmp_file = 0;
67 	unsigned i, j;
68 
69 	PLAIN_REQUIRE_KERNEL_MODULE("aio", 0);
70 
71 	kq = kqueue();
72 	if (kq < 0) {
73 		perror("No kqeueue\n");
74 		exit(1);
75 	}
76 
77 	if (argc == 1) {
78 		strcpy(pathname, PATH_TEMPLATE);
79 		fd = mkstemp(pathname);
80 		file = pathname;
81 		tmp_file = 1;
82 	} else {
83 		file = argv[1];
84 		fd = open(file, O_RDWR|O_CREAT, 0666);
85 	}
86 	if (fd == -1)
87 		err(1, "Can't open %s\n", file);
88 
89 	for (run = 0; run < MAX_RUNS; run++){
90 #ifdef DEBUG
91 		printf("Run %d\n", run);
92 #endif
93 		for (i = 0; i < nitems(iocb); i++) {
94 			iocb[i] = (struct aiocb *)calloc(1,
95 			    sizeof(struct aiocb));
96 			if (iocb[i] == NULL)
97 				err(1, "calloc");
98 		}
99 
100 		pending = 0;
101 		for (i = 0; i < nitems(iocb); i++) {
102 			pending++;
103 			iocb[i]->aio_nbytes = sizeof(buffer);
104 			iocb[i]->aio_buf = buffer;
105 			iocb[i]->aio_fildes = fd;
106 			iocb[i]->aio_offset = iocb[i]->aio_nbytes * i * run;
107 
108 			iocb[i]->aio_sigevent.sigev_notify_kqueue = kq;
109 			iocb[i]->aio_sigevent.sigev_value.sival_ptr = iocb[i];
110 			iocb[i]->aio_sigevent.sigev_notify = SIGEV_KEVENT;
111 
112 			result = aio_write(iocb[i]);
113 			if (result != 0) {
114 				perror("aio_write");
115 				printf("Result %d iteration %d\n", result, i);
116 				exit(1);
117 			}
118 #ifdef DEBUG
119 			printf("WRITE %d is at %p\n", i, iocb[i]);
120 #endif
121 			result = rand();
122 			if (result < RAND_MAX/32) {
123 				if (result > RAND_MAX/64) {
124 					result = aio_cancel(fd, iocb[i]);
125 #ifdef DEBUG
126 					printf("Cancel %d %p result %d\n", i, iocb[i], result);
127 #endif
128 					if (result == AIO_CANCELED) {
129 						aio_return(iocb[i]);
130 						iocb[i] = NULL;
131 						pending--;
132 					}
133 				}
134 			}
135 		}
136 		cancel = nitems(iocb) - pending;
137 
138 		i = 0;
139 		while (pending) {
140 
141 			for (;;) {
142 
143 				bzero(&ke, sizeof(ke));
144 				bzero(&kq_returned, sizeof(ke));
145 				ts.tv_sec = 0;
146 				ts.tv_nsec = 1;
147 				result = kevent(kq, NULL, 0,
148 						&kq_returned, 1, &ts);
149 				error = errno;
150 				if (result < 0)
151 					perror("kevent error: ");
152 				kq_iocb = kq_returned.udata;
153 #ifdef DEBUG
154 				printf("kevent %d %d errno %d return.ident %p "
155 				       "return.data %p return.udata %p %p\n",
156 				       i, result, error,
157 				       kq_returned.ident, kq_returned.data,
158 				       kq_returned.udata,
159 				       kq_iocb);
160 #endif
161 
162 				if (kq_iocb)
163 					break;
164 #ifdef DEBUG
165 				printf("Try again left %d out of %d %d\n",
166 				    pending, nitems(iocb), cancel);
167 #endif
168 			}
169 
170 			for (j = 0; j < nitems(iocb) && iocb[j] != kq_iocb;
171 			   j++) ;
172 #ifdef DEBUG
173 			printf("kq_iocb %p\n", kq_iocb);
174 
175 			printf("Error Result for %d is %d pending %d\n",
176 			    j, result, pending);
177 #endif
178 			result = aio_return(kq_iocb);
179 #ifdef DEBUG
180 			printf("Return Result for %d is %d\n\n", j, result);
181 #endif
182 			if (result != sizeof(buffer)) {
183 				printf("FAIL: run %d, operation %d, result %d "
184 				    " (errno=%d) should be %zu\n", run, pending,
185 				    result, errno, sizeof(buffer));
186 				failed++;
187 			} else
188 				printf("PASS: run %d, left %d\n", run,
189 				    pending - 1);
190 
191 			free(kq_iocb);
192 			iocb[j] = NULL;
193 			pending--;
194 			i++;
195 		}
196 
197 		for (i = 0; i < nitems(iocb); i++)
198 			free(iocb[i]);
199 
200 	}
201 
202 	if (tmp_file)
203 		unlink(pathname);
204 
205 	if (failed != 0)
206 		printf("FAIL: %d tests failed\n", failed);
207 	else
208 		printf("PASS: All tests passed\n");
209 
210 	exit (failed == 0 ? 0 : 1);
211 }
212