tools/perf/builtin-config.c

   1 /*
   2  * builtin-config.c
   3  *
   4  * Copyright (C) 2015, Taeung Song <treeze.taeung@gmail.com>
   5  *
   6  */
   7 #include "builtin.h"
   8
   9 #include "perf.h"
  10
  11 #include "util/cache.h"
  12 #include <subcmd/parse-options.h>
  13 #include "util/util.h"
  14 #include "util/debug.h"
  15 #include "util/config.h"
  16
  17 static bool use_system_config, use_user_config;
  18
  19 static const char * const config_usage[] = {
  20         "perf config [<file-option>] [options] [section.name[=value] ...]",
  21         NULL
  22 };
  23
  24 enum actions {
  25         ACTION_LIST = 1
  26 } actions;
  27
  28 static struct option config_options[] = {
  29         OPT_SET_UINT('l', "list", &actions,
  30                      "show current config variables", ACTION_LIST),
  31         OPT_BOOLEAN(0, "system", &use_system_config, "use system config file"),
  32         OPT_BOOLEAN(0, "user", &use_user_config, "use user config file"),
  33         OPT_END()
  34 };
  35
  36 static int set_config(struct perf_config_set *set, const char *file_name,
  37                       const char *var, const char *value)
  38 {
  39         struct perf_config_section *section = NULL;
  40         struct perf_config_item *item = NULL;
  41         const char *first_line = "# this file is auto-generated.";
  42         FILE *fp;
  43
  44         if (set == NULL)
  45                 return -1;
  46
  47         fp = fopen(file_name, "w");
  48         if (!fp)
  49                 return -1;
  50
  51         perf_config_set__collect(set, file_name, var, value);
  52         fprintf(fp, "%s\n", first_line);
  53
  54         /* overwrite configvariables */
  55         perf_config_items__for_each_entry(&set->sections, section) {
  56                 if (!use_system_config && section->from_system_config)
  57                         continue;
  58                 fprintf(fp, "[%s]\n", section->name);
  59
  60                 perf_config_items__for_each_entry(&section->items, item) {
  61                         if (!use_system_config && section->from_system_config)
  62                                 continue;
  63                         if (item->value)
  64                                 fprintf(fp, "\t%s = %s\n",
  65                                         item->name, item->value);
  66                 }
  67         }
  68         fclose(fp);
  69
  70         return 0;
  71 }
  72
  73 static int show_spec_config(struct perf_config_set *set, const char *var)
  74 {
  75         struct perf_config_section *section;
  76         struct perf_config_item *item;
  77
  78         if (set == NULL)
  79                 return -1;
  80
  81         perf_config_items__for_each_entry(&set->sections, section) {
  82                 if (prefixcmp(var, section->name) != 0)
  83                         continue;
  84
  85                 perf_config_items__for_each_entry(&section->items, item) {
  86                         const char *name = var + strlen(section->name) + 1;
  87
  88                         if (strcmp(name, item->name) == 0) {
  89                                 char *value = item->value;
  90
  91                                 if (value) {
  92                                         printf("%s=%s\n", var, value);
  93                                         return 0;
  94                                 }
  95                         }
  96
  97                 }
  98         }
  99
 100         return 0;
 101 }
 102
 103 static int show_config(struct perf_config_set *set)
 104 {
 105         struct perf_config_section *section;
 106         struct perf_config_item *item;
 107
 108         if (set == NULL)
 109                 return -1;
 110
 111         perf_config_set__for_each_entry(set, section, item) {
 112                 char *value = item->value;
 113
 114                 if (value)
 115                         printf("%s.%s=%s\n", section->name,
 116                                item->name, value);
 117         }
 118
 119         return 0;
 120 }
 121
 122 static int parse_config_arg(char *arg, char **var, char **value)
 123 {
 124         const char *last_dot = strchr(arg, '.');
 125
 126         /*
 127          * Since "var" actually contains the section name and the real
 128          * config variable name separated by a dot, we have to know where the dot is.
 129          */
 130         if (last_dot == NULL || last_dot == arg) {
 131                 pr_err("The config variable does not contain a section name: %s\n", arg);
 132                 return -1;
 133         }
 134         if (!last_dot[1]) {
 135                 pr_err("The config variable does not contain a variable name: %s\n", arg);
 136                 return -1;
 137         }
 138
 139         *value = strchr(arg, '=');
 140         if (*value == NULL)
 141                 *var = arg;
 142         else if (!strcmp(*value, "=")) {
 143                 pr_err("The config variable does not contain a value: %s\n", arg);
 144                 return -1;
 145         } else {
 146                 *value = *value + 1; /* excluding a first character '=' */
 147                 *var = strsep(&arg, "=");
 148                 if (*var[0] == '\0') {
 149                         pr_err("invalid config variable: %s\n", arg);
 150                         return -1;
 151                 }
 152         }
 153
 154         return 0;
 155 }
 156
 157 int cmd_config(int argc, const char **argv)
 158 {
 159         int i, ret = -1;
 160         struct perf_config_set *set;
 161         char *user_config = mkpath("%s/.perfconfig", getenv("HOME"));
 162         const char *config_filename;
 163
 164         argc = parse_options(argc, argv, config_options, config_usage,
 165                              PARSE_OPT_STOP_AT_NON_OPTION);
 166
 167         if (use_system_config && use_user_config) {
 168                 pr_err("Error: only one config file at a time\n");
 169                 parse_options_usage(config_usage, config_options, "user", 0);
 170                 parse_options_usage(NULL, config_options, "system", 0);
 171                 return -1;
 172         }
 173
 174         if (use_system_config)
 175                 config_exclusive_filename = perf_etc_perfconfig();
 176         else if (use_user_config)
 177                 config_exclusive_filename = user_config;
 178
 179         if (!config_exclusive_filename)
 180                 config_filename = user_config;
 181         else
 182                 config_filename = config_exclusive_filename;
 183
 184         /*
 185          * At only 'config' sub-command, individually use the config set
 186          * because of reinitializing with options config file location.
 187          */
 188         set = perf_config_set__new();
 189         if (!set)
 190                 goto out_err;
 191
 192         switch (actions) {
 193         case ACTION_LIST:
 194                 if (argc) {
 195                         pr_err("Error: takes no arguments\n");
 196                         parse_options_usage(config_usage, config_options, "l", 1);
 197                 } else {
 198                         if (show_config(set) < 0) {
 199                                 pr_err("Nothing configured, "
 200                                        "please check your %s \n", config_filename);
 201                                 goto out_err;
 202                         }
 203                 }
 204                 break;
 205         default:
 206                 if (!argc) {
 207                         usage_with_options(config_usage, config_options);
 208                         break;
 209                 }
 210
 211                 for (i = 0; argv[i]; i++) {
 212                         char *var, *value;
 213                         char *arg = strdup(argv[i]);
 214
 215                         if (!arg) {
 216                                 pr_err("%s: strdup failed\n", __func__);
 217                                 goto out_err;
 218                         }
 219
 220                         if (parse_config_arg(arg, &var, &value) < 0) {
 221                                 free(arg);
 222                                 goto out_err;
 223                         }
 224
 225                         if (value == NULL) {
 226                                 if (show_spec_config(set, var) < 0) {
 227                                         pr_err("%s is not configured: %s\n",
 228                                                var, config_filename);
 229                                         free(arg);
 230                                         goto out_err;
 231                                 }
 232                         } else {
 233                                 if (set_config(set, config_filename, var, value) < 0) {
 234                                         pr_err("Failed to set '%s=%s' on %s\n",
 235                                                var, value, config_filename);
 236                                         free(arg);
 237                                         goto out_err;
 238                                 }
 239                         }
 240                         free(arg);
 241                 }
 242         }
 243
 244         ret = 0;
 245 out_err:
 246         perf_config_set__delete(set);
 247         return ret;
 248 }