profile.c

   1 #include "fio.h"
   2 #include "profile.h"
   3 #include "debug.h"
   4 #include "flist.h"
   5 #include "options.h"
   6
   7 static FLIST_HEAD(profile_list);
   8
   9 struct profile_ops *find_profile(const char *profile)
  10 {
  11         struct profile_ops *ops = NULL;
  12         struct flist_head *n;
  13
  14         flist_for_each(n, &profile_list) {
  15                 ops = flist_entry(n, struct profile_ops, list);
  16                 if (!strcmp(profile, ops->name))
  17                         break;
  18
  19                 ops = NULL;
  20         }
  21
  22         return ops;
  23 }
  24
  25 int load_profile(const char *profile)
  26 {
  27         struct profile_ops *ops;
  28
  29         dprint(FD_PROFILE, "loading profile '%s'\n", profile);
  30
  31         ops = find_profile(profile);
  32         if (ops) {
  33                 if (ops->prep_cmd()) {
  34                         log_err("fio: profile %s prep failed\n", profile);
  35                         return 1;
  36                 }
  37                 add_job_opts(ops->cmdline, FIO_CLIENT_TYPE_CLI);
  38                 return 0;
  39         }
  40
  41         log_err("fio: profile '%s' not found\n", profile);
  42         return 1;
  43 }
  44
  45 static int add_profile_options(struct profile_ops *ops)
  46 {
  47         struct fio_option *o;
  48
  49         if (!ops->options)
  50                 return 0;
  51
  52         o = ops->options;
  53         while (o->name) {
  54                 o->prof_name = ops->name;
  55                 o->prof_opts = ops->opt_data;
  56                 if (add_option(o))
  57                         return 1;
  58                 o++;
  59         }
  60
  61         return 0;
  62 }
  63
  64 int register_profile(struct profile_ops *ops)
  65 {
  66         int ret;
  67
  68         dprint(FD_PROFILE, "register profile '%s'\n", ops->name);
  69
  70         ret = add_profile_options(ops);
  71         if (!ret) {
  72                 flist_add_tail(&ops->list, &profile_list);
  73                 add_opt_posval("profile", ops->name, ops->desc);
  74                 return 0;
  75         }
  76
  77         invalidate_profile_options(ops->name);
  78         return ret;
  79 }
  80
  81 void unregister_profile(struct profile_ops *ops)
  82 {
  83         dprint(FD_PROFILE, "unregister profile '%s'\n", ops->name);
  84         flist_del(&ops->list);
  85         invalidate_profile_options(ops->name);
  86         del_opt_posval("profile", ops->name);
  87 }
  88
  89 void profile_add_hooks(struct thread_data *td)
  90 {
  91         struct profile_ops *ops;
  92
  93         if (!exec_profile)
  94                 return;
  95
  96         ops = find_profile(exec_profile);
  97         if (!ops)
  98                 return;
  99
 100         if (ops->io_ops) {
 101                 td->prof_io_ops = *ops->io_ops;
 102                 td->flags |= TD_F_PROFILE_OPS;
 103         }
 104 }
 105
 106 int profile_td_init(struct thread_data *td)
 107 {
 108         struct prof_io_ops *ops = &td->prof_io_ops;
 109
 110         if (ops->td_init)
 111                 return ops->td_init(td);
 112
 113         return 0;
 114 }
 115
 116 void profile_td_exit(struct thread_data *td)
 117 {
 118         struct prof_io_ops *ops = &td->prof_io_ops;
 119
 120         if (ops->td_exit)
 121                 ops->td_exit(td);
 122 }