Jens Axboe | 79d1631 | 2010-03-04 12:43:20 +0100 | [diff] [blame] | 1 | #include "fio.h" |
| 2 | #include "profile.h" |
| 3 | #include "debug.h" |
| 4 | #include "flist.h" |
Jens Axboe | e2de69d | 2010-03-04 14:05:48 +0100 | [diff] [blame] | 5 | #include "options.h" |
Jens Axboe | 79d1631 | 2010-03-04 12:43:20 +0100 | [diff] [blame] | 6 | |
| 7 | static FLIST_HEAD(profile_list); |
| 8 | |
Jens Axboe | 15dc193 | 2010-03-05 10:59:06 +0100 | [diff] [blame] | 9 | struct profile_ops *find_profile(const char *profile) |
Jens Axboe | 79d1631 | 2010-03-04 12:43:20 +0100 | [diff] [blame] | 10 | { |
Jens Axboe | 15dc193 | 2010-03-05 10:59:06 +0100 | [diff] [blame] | 11 | struct profile_ops *ops = NULL; |
Jens Axboe | 79d1631 | 2010-03-04 12:43:20 +0100 | [diff] [blame] | 12 | struct flist_head *n; |
| 13 | |
Jens Axboe | 79d1631 | 2010-03-04 12:43:20 +0100 | [diff] [blame] | 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 | |
Jens Axboe | 15dc193 | 2010-03-05 10:59:06 +0100 | [diff] [blame] | 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); |
Jens Axboe | 79d1631 | 2010-03-04 12:43:20 +0100 | [diff] [blame] | 32 | if (ops) { |
Jens Axboe | 2363d8d | 2010-03-04 14:30:02 +0100 | [diff] [blame] | 33 | ops->prep_cmd(); |
| 34 | add_job_opts(ops->cmdline); |
Jens Axboe | 79d1631 | 2010-03-04 12:43:20 +0100 | [diff] [blame] | 35 | return 0; |
| 36 | } |
| 37 | |
| 38 | log_err("fio: profile '%s' not found\n", profile); |
| 39 | return 1; |
| 40 | } |
| 41 | |
Jens Axboe | 07b3232 | 2010-03-05 09:48:44 +0100 | [diff] [blame] | 42 | static int add_profile_options(struct profile_ops *ops) |
Jens Axboe | e2de69d | 2010-03-04 14:05:48 +0100 | [diff] [blame] | 43 | { |
Jens Axboe | 07b3232 | 2010-03-05 09:48:44 +0100 | [diff] [blame] | 44 | struct fio_option *o; |
Jens Axboe | e2de69d | 2010-03-04 14:05:48 +0100 | [diff] [blame] | 45 | |
| 46 | if (!ops->options) |
Jens Axboe | 07b3232 | 2010-03-05 09:48:44 +0100 | [diff] [blame] | 47 | return 0; |
Jens Axboe | e2de69d | 2010-03-04 14:05:48 +0100 | [diff] [blame] | 48 | |
Jens Axboe | 07b3232 | 2010-03-05 09:48:44 +0100 | [diff] [blame] | 49 | o = ops->options; |
| 50 | while (o->name) { |
| 51 | o->prof_name = ops->name; |
| 52 | if (add_option(o)) |
| 53 | return 1; |
| 54 | o++; |
Jens Axboe | e2de69d | 2010-03-04 14:05:48 +0100 | [diff] [blame] | 55 | } |
Jens Axboe | 07b3232 | 2010-03-05 09:48:44 +0100 | [diff] [blame] | 56 | |
| 57 | return 0; |
Jens Axboe | e2de69d | 2010-03-04 14:05:48 +0100 | [diff] [blame] | 58 | } |
| 59 | |
Jens Axboe | 07b3232 | 2010-03-05 09:48:44 +0100 | [diff] [blame] | 60 | int register_profile(struct profile_ops *ops) |
Jens Axboe | 79d1631 | 2010-03-04 12:43:20 +0100 | [diff] [blame] | 61 | { |
Jens Axboe | 07b3232 | 2010-03-05 09:48:44 +0100 | [diff] [blame] | 62 | int ret; |
| 63 | |
Jens Axboe | 79d1631 | 2010-03-04 12:43:20 +0100 | [diff] [blame] | 64 | dprint(FD_PROFILE, "register profile '%s'\n", ops->name); |
Jens Axboe | 07b3232 | 2010-03-05 09:48:44 +0100 | [diff] [blame] | 65 | |
Jens Axboe | f5b6bb8 | 2010-03-05 10:09:59 +0100 | [diff] [blame] | 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); |
Jens Axboe | 07b3232 | 2010-03-05 09:48:44 +0100 | [diff] [blame] | 74 | return ret; |
Jens Axboe | 79d1631 | 2010-03-04 12:43:20 +0100 | [diff] [blame] | 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); |
Jens Axboe | 07b3232 | 2010-03-05 09:48:44 +0100 | [diff] [blame] | 81 | invalidate_profile_options(ops->name); |
Jens Axboe | f5b6bb8 | 2010-03-05 10:09:59 +0100 | [diff] [blame] | 82 | del_opt_posval("profile", ops->name); |
Jens Axboe | 79d1631 | 2010-03-04 12:43:20 +0100 | [diff] [blame] | 83 | } |
Jens Axboe | 15dc193 | 2010-03-05 10:59:06 +0100 | [diff] [blame] | 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 | |
Jens Axboe | 7eb3657 | 2010-03-08 13:58:49 +0100 | [diff] [blame] | 96 | if (ops->io_ops) |
| 97 | td->prof_io_ops = *ops->io_ops; |
Jens Axboe | 15dc193 | 2010-03-05 10:59:06 +0100 | [diff] [blame] | 98 | } |
Jens Axboe | 58c55ba | 2010-03-09 12:20:08 +0100 | [diff] [blame] | 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 | } |