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