122 lines
2.1 KiB
C
122 lines
2.1 KiB
C
#include "fio.h"
|
|
#include "profile.h"
|
|
#include "debug.h"
|
|
#include "flist.h"
|
|
#include "options.h"
|
|
|
|
static FLIST_HEAD(profile_list);
|
|
|
|
struct profile_ops *find_profile(const char *profile)
|
|
{
|
|
struct profile_ops *ops = NULL;
|
|
struct flist_head *n;
|
|
|
|
flist_for_each(n, &profile_list) {
|
|
ops = flist_entry(n, struct profile_ops, list);
|
|
if (!strcmp(profile, ops->name))
|
|
break;
|
|
|
|
ops = NULL;
|
|
}
|
|
|
|
return ops;
|
|
}
|
|
|
|
int load_profile(const char *profile)
|
|
{
|
|
struct profile_ops *ops;
|
|
|
|
dprint(FD_PROFILE, "loading profile '%s'\n", profile);
|
|
|
|
ops = find_profile(profile);
|
|
if (ops) {
|
|
if (ops->prep_cmd()) {
|
|
log_err("fio: profile %s prep failed\n", profile);
|
|
return 1;
|
|
}
|
|
add_job_opts(ops->cmdline, FIO_CLIENT_TYPE_CLI);
|
|
return 0;
|
|
}
|
|
|
|
log_err("fio: profile '%s' not found\n", profile);
|
|
return 1;
|
|
}
|
|
|
|
static int add_profile_options(struct profile_ops *ops)
|
|
{
|
|
struct fio_option *o;
|
|
|
|
if (!ops->options)
|
|
return 0;
|
|
|
|
o = ops->options;
|
|
while (o->name) {
|
|
o->prof_name = ops->name;
|
|
o->prof_opts = ops->opt_data;
|
|
if (add_option(o))
|
|
return 1;
|
|
o++;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
int register_profile(struct profile_ops *ops)
|
|
{
|
|
int ret;
|
|
|
|
dprint(FD_PROFILE, "register profile '%s'\n", ops->name);
|
|
|
|
ret = add_profile_options(ops);
|
|
if (!ret) {
|
|
flist_add_tail(&ops->list, &profile_list);
|
|
add_opt_posval("profile", ops->name, ops->desc);
|
|
return 0;
|
|
}
|
|
|
|
invalidate_profile_options(ops->name);
|
|
return ret;
|
|
}
|
|
|
|
void unregister_profile(struct profile_ops *ops)
|
|
{
|
|
dprint(FD_PROFILE, "unregister profile '%s'\n", ops->name);
|
|
flist_del(&ops->list);
|
|
invalidate_profile_options(ops->name);
|
|
del_opt_posval("profile", ops->name);
|
|
}
|
|
|
|
void profile_add_hooks(struct thread_data *td)
|
|
{
|
|
struct profile_ops *ops;
|
|
|
|
if (!exec_profile)
|
|
return;
|
|
|
|
ops = find_profile(exec_profile);
|
|
if (!ops)
|
|
return;
|
|
|
|
if (ops->io_ops) {
|
|
td->prof_io_ops = *ops->io_ops;
|
|
td->flags |= TD_F_PROFILE_OPS;
|
|
}
|
|
}
|
|
|
|
int profile_td_init(struct thread_data *td)
|
|
{
|
|
struct prof_io_ops *ops = &td->prof_io_ops;
|
|
|
|
if (ops->td_init)
|
|
return ops->td_init(td);
|
|
|
|
return 0;
|
|
}
|
|
|
|
void profile_td_exit(struct thread_data *td)
|
|
{
|
|
struct prof_io_ops *ops = &td->prof_io_ops;
|
|
|
|
if (ops->td_exit)
|
|
ops->td_exit(td);
|
|
}
|