Merge branch 'jk/test-fixes'
[git] / trace2 / tr2_cfg.c
1 #include "cache.h"
2 #include "config.h"
3 #include "trace2/tr2_cfg.h"
4 #include "trace2/tr2_sysenv.h"
5
6 static struct strbuf **tr2_cfg_patterns;
7 static int tr2_cfg_count_patterns;
8 static int tr2_cfg_loaded;
9
10 /*
11  * Parse a string containing a comma-delimited list of config keys
12  * or wildcard patterns into a list of strbufs.
13  */
14 static int tr2_cfg_load_patterns(void)
15 {
16         struct strbuf **s;
17         const char *envvar;
18
19         if (tr2_cfg_loaded)
20                 return tr2_cfg_count_patterns;
21         tr2_cfg_loaded = 1;
22
23         envvar = tr2_sysenv_get(TR2_SYSENV_CFG_PARAM);
24         if (!envvar || !*envvar)
25                 return tr2_cfg_count_patterns;
26
27         tr2_cfg_patterns = strbuf_split_buf(envvar, strlen(envvar), ',', -1);
28         for (s = tr2_cfg_patterns; *s; s++) {
29                 struct strbuf *buf = *s;
30
31                 if (buf->len && buf->buf[buf->len - 1] == ',')
32                         strbuf_setlen(buf, buf->len - 1);
33                 strbuf_trim_trailing_newline(*s);
34                 strbuf_trim(*s);
35         }
36
37         tr2_cfg_count_patterns = s - tr2_cfg_patterns;
38         return tr2_cfg_count_patterns;
39 }
40
41 void tr2_cfg_free_patterns(void)
42 {
43         if (tr2_cfg_patterns)
44                 strbuf_list_free(tr2_cfg_patterns);
45         tr2_cfg_count_patterns = 0;
46         tr2_cfg_loaded = 0;
47 }
48
49 struct tr2_cfg_data {
50         const char *file;
51         int line;
52 };
53
54 /*
55  * See if the given config key matches any of our patterns of interest.
56  */
57 static int tr2_cfg_cb(const char *key, const char *value, void *d)
58 {
59         struct strbuf **s;
60         struct tr2_cfg_data *data = (struct tr2_cfg_data *)d;
61
62         for (s = tr2_cfg_patterns; *s; s++) {
63                 struct strbuf *buf = *s;
64                 int wm = wildmatch(buf->buf, key, WM_CASEFOLD);
65                 if (wm == WM_MATCH) {
66                         trace2_def_param_fl(data->file, data->line, key, value);
67                         return 0;
68                 }
69         }
70
71         return 0;
72 }
73
74 void tr2_cfg_list_config_fl(const char *file, int line)
75 {
76         struct tr2_cfg_data data = { file, line };
77
78         if (tr2_cfg_load_patterns() > 0)
79                 read_early_config(tr2_cfg_cb, &data);
80 }
81
82 void tr2_cfg_set_fl(const char *file, int line, const char *key,
83                     const char *value)
84 {
85         struct tr2_cfg_data data = { file, line };
86
87         if (tr2_cfg_load_patterns() > 0)
88                 tr2_cfg_cb(key, value, &data);
89 }