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                 ops->prep_cmd();
  34                 add_job_opts(ops->cmdline);
  35                 return 0;
  36         }
  37
  38         log_err("fio: profile '%s' not found\n", profile);
  39         return 1;
  40 }
  41
  42 static int add_profile_options(struct profile_ops *ops)
  43 {
  44         struct fio_option *o;
  45
  46         if (!ops->options)
  47                 return 0;
  48
  49         o = ops->options;
  50         while (o->name) {
  51                 o->prof_name = ops->name;
  52                 if (add_option(o))
  53                         return 1;
  54                 o++;
  55         }
  56
  57         return 0;
  58 }
  59
  60 int register_profile(struct profile_ops *ops)
  61 {
  62         int ret;
  63
  64         dprint(FD_PROFILE, "register profile '%s'\n", ops->name);
  65
  66         ret = add_profile_options(ops);
  67         if (!ret) {
  68                 flist_add_tail(&ops->list, &profile_list);
  69                 add_opt_posval("profile", ops->name, ops->desc);
  70                 return 0;
  71         }
  72
  73         invalidate_profile_options(ops->name);
  74         return ret;
  75 }
  76
  77 void unregister_profile(struct profile_ops *ops)
  78 {
  79         dprint(FD_PROFILE, "unregister profile '%s'\n", ops->name);
  80         flist_del(&ops->list);
  81         invalidate_profile_options(ops->name);
  82         del_opt_posval("profile", ops->name);
  83 }
  84
  85 void profile_add_hooks(struct thread_data *td)
  86 {
  87         struct profile_ops *ops;
  88
  89         if (!exec_profile)
  90                 return;
  91
  92         ops = find_profile(exec_profile);
  93         if (!ops)
  94                 return;
  95
  96         if (ops->io_ops)
  97                 td->prof_io_ops = *ops->io_ops;
  98 }
  99
 100 int profile_td_init(struct thread_data *td)
 101 {
 102         struct prof_io_ops *ops = &td->prof_io_ops;
 103
 104         if (ops->td_init)
 105                 return ops->td_init(td);
 106
 107         return 0;
 108 }
 109
 110 void profile_td_exit(struct thread_data *td)
 111 {
 112         struct prof_io_ops *ops = &td->prof_io_ops;
 113
 114         if (ops->td_exit)
 115                 ops->td_exit(td);
 116 }