blob: 90c9ea8d14eb6c4d52c5a94bef7eab3e32fc0a0e [file] [log] [blame]
#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);
}