solarisaio: support signal driven async IO
[fio.git] / engines / solarisaio.c
CommitLineData
417f0068
JA
1/*
2 * Native Solaris async IO engine
3 *
4 */
5#include <stdio.h>
6#include <stdlib.h>
7#include <unistd.h>
556e831d 8#include <signal.h>
417f0068
JA
9#include <errno.h>
10
11#include "../fio.h"
12
13#ifdef FIO_HAVE_SOLARISAIO
14
15#include <sys/asynch.h>
16
17struct solarisaio_data {
18 struct io_u **aio_events;
556e831d 19 unsigned int aio_pending;
417f0068 20 unsigned int nr;
5cf56c03 21 unsigned int max_depth;
417f0068
JA
22};
23
24static int fio_solarisaio_cancel(struct thread_data fio_unused *td,
25 struct io_u *io_u)
26{
27 return aiocancel(&io_u->resultp);
28}
29
30static int fio_solarisaio_prep(struct thread_data fio_unused *td,
31 struct io_u *io_u)
32{
556e831d
JA
33 struct solarisaio_data *sd = td->io_ops->data;
34
417f0068 35 io_u->resultp.aio_return = AIO_INPROGRESS;
556e831d 36 io_u->engine_data = sd;
417f0068
JA
37 return 0;
38}
39
556e831d
JA
40static void wait_for_event(struct timeval *tv)
41{
42 struct solarisaio_data *sd;
43 struct io_u *io_u;
44 aio_result_t *res;
45
46 res = aiowait(tv);
47 if (res == (aio_result_t *) -1) {
48 int err = errno;
49
50 if (err != EINVAL) {
51 log_err("fio: solarisaio got %d in aiowait\n", err);
52 exit(err);
53 }
54 return;
55 } else if (!res)
56 return;
57
58 io_u = container_of(res, struct io_u, resultp);
59 sd = io_u->engine_data;
60
61 if (io_u->resultp.aio_return >= 0) {
62 io_u->resid = io_u->xfer_buflen - io_u->resultp.aio_return;
63 io_u->error = 0;
64 } else
65 io_u->error = io_u->resultp.aio_return;
66
67 /*
68 * For SIGIO, we need a write barrier between the two, so that
69 * the ->aio_pending store is seen after the ->aio_events store
70 */
71 sd->aio_events[sd->aio_pending] = io_u;
72 sd->aio_pending++;
73 sd->nr--;
74}
75
76static void fio_solarisaio_sigio(int sig)
77{
78 wait_for_event(NULL);
79}
80
417f0068
JA
81static int fio_solarisaio_getevents(struct thread_data *td, unsigned int min,
82 unsigned int max, struct timespec *t)
83{
84 struct solarisaio_data *sd = td->io_ops->data;
85 struct timeval tv;
556e831d 86 int ret;
417f0068 87
556e831d
JA
88 if (!min || !t) {
89 tv.tv_sec = 0;
90 tv.tv_usec = 0;
91 } else {
92 tv.tv_sec = t->tv_sec;
93 tv.tv_usec = t->tv_nsec / 1000;
94 }
95
96 while (sd->aio_pending < min)
97 wait_for_event(&tv);
417f0068 98
556e831d
JA
99 /*
100 * Needs locking here for SIGIO
101 */
102 ret = sd->aio_pending;
103 sd->aio_pending = 0;
104 return ret;
417f0068
JA
105}
106
107static struct io_u *fio_solarisaio_event(struct thread_data *td, int event)
108{
109 struct solarisaio_data *sd = td->io_ops->data;
110
111 return sd->aio_events[event];
112}
113
114static int fio_solarisaio_queue(struct thread_data fio_unused *td,
115 struct io_u *io_u)
116{
117 struct solarisaio_data *sd = td->io_ops->data;
118 struct fio_file *f = io_u->file;
119 off_t off;
120 int ret;
121
122 fio_ro_check(td, io_u);
123
124 if (io_u->ddir == DDIR_SYNC) {
125 if (sd->nr)
126 return FIO_Q_BUSY;
127 if (fsync(f->fd) < 0)
128 io_u->error = errno;
129
130 return FIO_Q_COMPLETED;
131 }
132
5cf56c03 133 if (sd->nr == sd->max_depth)
417f0068
JA
134 return FIO_Q_BUSY;
135
136 off = io_u->offset;
137 if (io_u->ddir == DDIR_READ)
138 ret = aioread(f->fd, io_u->xfer_buf, io_u->xfer_buflen, off,
139 SEEK_SET, &io_u->resultp);
140 else
141 ret = aiowrite(f->fd, io_u->xfer_buf, io_u->xfer_buflen, off,
142 SEEK_SET, &io_u->resultp);
143 if (ret) {
144 io_u->error = errno;
145 td_verror(td, io_u->error, "xfer");
146 return FIO_Q_COMPLETED;
147 }
148
149 sd->nr++;
150 return FIO_Q_QUEUED;
151}
152
153static void fio_solarisaio_cleanup(struct thread_data *td)
154{
155 struct solarisaio_data *sd = td->io_ops->data;
156
157 if (sd) {
158 free(sd->aio_events);
159 free(sd);
160 }
161}
162
556e831d
JA
163/*
164 * Set USE_SIGNAL_COMPLETIONS to use SIGIO as completion events. Needs
165 * locking around ->aio_pending and ->aio_events, see comment
166 */
167static void fio_solarisaio_init_sigio(void)
168{
169#ifdef USE_SIGNAL_COMPLETIONS
170 struct sigaction act;
171
172 memset(&act, 0, sizeof(act));
173 act.sa_handler = fio_solarisaio_sigio;
174 act.sa_flags = SA_RESTART;
175 sigaction(SIGIO, &act, NULL);
176#endif
177}
178
417f0068
JA
179static int fio_solarisaio_init(struct thread_data *td)
180{
181 struct solarisaio_data *sd = malloc(sizeof(*sd));
5cf56c03
JA
182 unsigned int max_depth;
183
184 max_depth = td->o.iodepth;
185 if (max_depth > MAXASYNCHIO) {
186 max_depth = MAXASYNCHIO;
187 log_info("fio: lower depth to %d due to OS constraints\n",
188 max_depth);
189 }
417f0068
JA
190
191 memset(sd, 0, sizeof(*sd));
5cf56c03
JA
192 sd->aio_events = malloc(max_depth * sizeof(struct io_u *));
193 memset(sd->aio_events, 0, max_depth * sizeof(struct io_u *));
194 sd->max_depth = max_depth;
417f0068 195
556e831d
JA
196 fio_solarisaio_init_sigio();
197
417f0068
JA
198 td->io_ops->data = sd;
199 return 0;
200}
201
202static struct ioengine_ops ioengine = {
203 .name = "solarisaio",
204 .version = FIO_IOOPS_VERSION,
205 .init = fio_solarisaio_init,
206 .prep = fio_solarisaio_prep,
207 .queue = fio_solarisaio_queue,
208 .cancel = fio_solarisaio_cancel,
209 .getevents = fio_solarisaio_getevents,
210 .event = fio_solarisaio_event,
211 .cleanup = fio_solarisaio_cleanup,
212 .open_file = generic_open_file,
213 .close_file = generic_close_file,
214};
215
216#else /* FIO_HAVE_SOLARISAIO */
217
218/*
219 * When we have a proper configure system in place, we simply wont build
220 * and install this io engine. For now install a crippled version that
221 * just complains and fails to load.
222 */
223static int fio_solarisaio_init(struct thread_data fio_unused *td)
224{
225 fprintf(stderr, "fio: solarisaio not available\n");
226 return 1;
227}
228
229static struct ioengine_ops ioengine = {
230 .name = "solarisaio",
231 .version = FIO_IOOPS_VERSION,
232 .init = fio_solarisaio_init,
233};
234
235#endif
236
237static void fio_init fio_solarisaio_register(void)
238{
239 register_ioengine(&ioengine);
240}
241
242static void fio_exit fio_solarisaio_unregister(void)
243{
244 unregister_ioengine(&ioengine);
245}