[PATCH] String copy limiting fixes
[fio.git] / ioengines.c
CommitLineData
ebac4655
JA
1/*
2 * The io parts of the fio tool, includes workers for sync and mmap'ed
3 * io, as well as both posix and linux libaio support.
4 *
5 * sync io is implemented on top of aio.
6 *
7 * This is not really specific to fio, if the get_io_u/put_io_u and
8 * structures was pulled into this as well it would be a perfectly
9 * generic io engine that could be used for other projects.
10 *
11 */
12#include <stdio.h>
13#include <stdlib.h>
14#include <unistd.h>
5c4e1dbc 15#include <string.h>
2866c82d 16#include <dlfcn.h>
8c16d840 17
ebac4655
JA
18#include "fio.h"
19#include "os.h"
20
8c16d840
JA
21static int check_engine_ops(struct ioengine_ops *ops)
22{
23 /*
24 * cpu thread doesn't need to provide anything
25 */
26 if (ops->flags & FIO_CPUIO)
27 return 0;
28
29 if (!ops->event) {
30 log_err("%s: no event handler)\n", ops->name);
31 return 1;
32 }
33 if (!ops->getevents) {
34 log_err("%s: no getevents handler)\n", ops->name);
35 return 1;
36 }
37 if (!ops->queue) {
38 log_err("%s: no queue handler)\n", ops->name);
39 return 1;
40 }
41
42 return 0;
43}
44
b4692828 45struct ioengine_ops *load_ioengine(struct thread_data *td, const char *name)
ebac4655 46{
2866c82d 47 char engine[16], engine_lib[256];
84585003 48 struct ioengine_ops *ops, *ret;
2866c82d 49 void *dlhandle;
ebac4655 50
0bbab0e7 51 strncpy(engine, name, sizeof(engine) - 1);
ebac4655
JA
52
53 /*
2866c82d 54 * linux libaio has alias names, so convert to what we want
ebac4655 55 */
2866c82d
JA
56 if (!strncmp(engine, "linuxaio", 8) || !strncmp(engine, "aio", 3))
57 strcpy(engine, "libaio");
ebac4655 58
c1d5725e 59 sprintf(engine_lib, "%s/lib/fio/fio-engine-%s.o", fio_inst_prefix, engine);
2866c82d
JA
60 dlerror();
61 dlhandle = dlopen(engine_lib, RTLD_LAZY);
d4dbaaa8
JA
62 if (!dlhandle) {
63 td_vmsg(td, -1, dlerror());
64 return NULL;
65 }
8756e4d4 66
2866c82d 67 ops = dlsym(dlhandle, "ioengine");
d4dbaaa8
JA
68 if (!ops) {
69 td_vmsg(td, -1, dlerror());
70 dlclose(dlhandle);
71 return NULL;
72 }
8756e4d4 73
b902ceb5
JA
74 if (ops->version != FIO_IOOPS_VERSION) {
75 log_err("bad ioops version %d (want %d)\n", ops->version, FIO_IOOPS_VERSION);
76 dlclose(dlhandle);
77 return NULL;
78 }
79
8c16d840
JA
80 /*
81 * Check that the required methods are there.
82 */
83 if (check_engine_ops(ops)) {
84 dlclose(dlhandle);
85 return NULL;
86 }
87
84585003
JA
88 ret = malloc(sizeof(*ret));
89 memcpy(ret, ops, sizeof(*ret));
90 ret->data = NULL;
91 ret->dlhandle = dlhandle;
92
93 return ret;
8756e4d4
JA
94}
95
2866c82d 96void close_ioengine(struct thread_data *td)
8756e4d4 97{
2866c82d
JA
98 if (td->io_ops->cleanup)
99 td->io_ops->cleanup(td);
b990b5c0 100
2866c82d 101 dlclose(td->io_ops->dlhandle);
84585003
JA
102 free(td->io_ops);
103 td->io_ops = NULL;
b990b5c0 104}
10ba535a
JA
105
106int td_io_prep(struct thread_data *td, struct io_u *io_u)
107{
108 if (td->io_ops->prep && td->io_ops->prep(td, io_u))
109 return 1;
110
111 return 0;
112}
113
10ba535a
JA
114int td_io_getevents(struct thread_data *td, int min, int max,
115 struct timespec *t)
116{
117 return td->io_ops->getevents(td, min, max, t);
118}
119
120int td_io_queue(struct thread_data *td, struct io_u *io_u)
121{
122 gettimeofday(&io_u->issue_time, NULL);
123
124 return td->io_ops->queue(td, io_u);
125}
8c16d840
JA
126
127int td_io_init(struct thread_data *td)
128{
129 if (td->io_ops->init)
130 return td->io_ops->init(td);
131
132 return 0;
133}