trace2/tr2_cfg.c

   1 #include "git-compat-util.h"
   2 #include "config.h"
   3 #include "strbuf.h"
   4 #include "trace2.h"
   5 #include "trace2/tr2_cfg.h"
   6 #include "trace2/tr2_sysenv.h"
   7 #include "wildmatch.h"
   8
   9 static struct strbuf **tr2_cfg_patterns;
  10 static int tr2_cfg_count_patterns;
  11 static int tr2_cfg_loaded;
  12
  13 static struct strbuf **tr2_cfg_env_vars;
  14 static int tr2_cfg_env_vars_count;
  15 static int tr2_cfg_env_vars_loaded;
  16
  17 /*
  18  * Parse a string containing a comma-delimited list of config keys
  19  * or wildcard patterns into a list of strbufs.
  20  */
  21 static int tr2_cfg_load_patterns(void)
  22 {
  23         struct strbuf **s;
  24         const char *envvar;
  25
  26         if (tr2_cfg_loaded)
  27                 return tr2_cfg_count_patterns;
  28         tr2_cfg_loaded = 1;
  29
  30         envvar = tr2_sysenv_get(TR2_SYSENV_CFG_PARAM);
  31         if (!envvar || !*envvar)
  32                 return tr2_cfg_count_patterns;
  33
  34         tr2_cfg_patterns = strbuf_split_buf(envvar, strlen(envvar), ',', -1);
  35         for (s = tr2_cfg_patterns; *s; s++) {
  36                 struct strbuf *buf = *s;
  37
  38                 if (buf->len && buf->buf[buf->len - 1] == ',')
  39                         strbuf_setlen(buf, buf->len - 1);
  40                 strbuf_trim_trailing_newline(*s);
  41                 strbuf_trim(*s);
  42         }
  43
  44         tr2_cfg_count_patterns = s - tr2_cfg_patterns;
  45         return tr2_cfg_count_patterns;
  46 }
  47
  48 void tr2_cfg_free_patterns(void)
  49 {
  50         if (tr2_cfg_patterns)
  51                 strbuf_list_free(tr2_cfg_patterns);
  52         tr2_cfg_count_patterns = 0;
  53         tr2_cfg_loaded = 0;
  54 }
  55
  56 /*
  57  * Parse a string containing a comma-delimited list of environment variable
  58  * names into a list of strbufs.
  59  */
  60 static int tr2_load_env_vars(void)
  61 {
  62         struct strbuf **s;
  63         const char *varlist;
  64
  65         if (tr2_cfg_env_vars_loaded)
  66                 return tr2_cfg_env_vars_count;
  67         tr2_cfg_env_vars_loaded = 1;
  68
  69         varlist = tr2_sysenv_get(TR2_SYSENV_ENV_VARS);
  70         if (!varlist || !*varlist)
  71                 return tr2_cfg_env_vars_count;
  72
  73         tr2_cfg_env_vars = strbuf_split_buf(varlist, strlen(varlist), ',', -1);
  74         for (s = tr2_cfg_env_vars; *s; s++) {
  75                 struct strbuf *buf = *s;
  76
  77                 if (buf->len && buf->buf[buf->len - 1] == ',')
  78                         strbuf_setlen(buf, buf->len - 1);
  79                 strbuf_trim_trailing_newline(*s);
  80                 strbuf_trim(*s);
  81         }
  82
  83         tr2_cfg_env_vars_count = s - tr2_cfg_env_vars;
  84         return tr2_cfg_env_vars_count;
  85 }
  86
  87 void tr2_cfg_free_env_vars(void)
  88 {
  89         if (tr2_cfg_env_vars)
  90                 strbuf_list_free(tr2_cfg_env_vars);
  91         tr2_cfg_env_vars_count = 0;
  92         tr2_cfg_env_vars_loaded = 0;
  93 }
  94
  95 struct tr2_cfg_data {
  96         const char *file;
  97         int line;
  98 };
  99
 100 /*
 101  * See if the given config key matches any of our patterns of interest.
 102  */
 103 static int tr2_cfg_cb(const char *key, const char *value,
 104                       const struct config_context *ctx, void *d)
 105 {
 106         struct strbuf **s;
 107         struct tr2_cfg_data *data = (struct tr2_cfg_data *)d;
 108
 109         for (s = tr2_cfg_patterns; *s; s++) {
 110                 struct strbuf *buf = *s;
 111                 int wm = wildmatch(buf->buf, key, WM_CASEFOLD);
 112                 if (wm == WM_MATCH) {
 113                         trace2_def_param_fl(data->file, data->line, key, value,
 114                                             ctx->kvi);
 115                         return 0;
 116                 }
 117         }
 118
 119         return 0;
 120 }
 121
 122 void tr2_cfg_list_config_fl(const char *file, int line)
 123 {
 124         struct tr2_cfg_data data = { file, line };
 125
 126         if (tr2_cfg_load_patterns() > 0)
 127                 read_early_config(tr2_cfg_cb, &data);
 128 }
 129
 130 void tr2_list_env_vars_fl(const char *file, int line)
 131 {
 132         struct key_value_info kvi = KVI_INIT;
 133         struct strbuf **s;
 134
 135         kvi_from_param(&kvi);
 136         if (tr2_load_env_vars() <= 0)
 137                 return;
 138
 139         for (s = tr2_cfg_env_vars; *s; s++) {
 140                 struct strbuf *buf = *s;
 141                 const char *val = getenv(buf->buf);
 142                 if (val && *val)
 143                         trace2_def_param_fl(file, line, buf->buf, val, &kvi);
 144         }
 145 }
 146
 147 void tr2_cfg_set_fl(const char *file, int line, const char *key,
 148                     const char *value)
 149 {
 150         struct key_value_info kvi = KVI_INIT;
 151         struct config_context ctx = {
 152                 .kvi = &kvi,
 153         };
 154         struct tr2_cfg_data data = { file, line };
 155
 156         if (tr2_cfg_load_patterns() > 0)
 157                 tr2_cfg_cb(key, value, &ctx, &data);
 158 }