aio_kqueue_test.c revision 282858 
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: stable/10/tests/sys/aio/aio_kqueue_test.c 282858 2015-05-13 12:09:01Z ngie $
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
57int
58main (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