11 #define CONFIG_FILE_LINE_MAX ((CONFIG_MAX_NAME + CONFIG_MAX_VALUE + 1024) * 2)
13 pthread_rwlock_t config_rwlock = PTHREAD_RWLOCK_INITIALIZER;
15 // ----------------------------------------------------------------------------
18 #define CONFIG_VALUE_LOADED 0x01 // has been loaded from the config
19 #define CONFIG_VALUE_USED 0x02 // has been accessed from the program
20 #define CONFIG_VALUE_CHANGED 0x04 // has been changed from the loaded value
21 #define CONFIG_VALUE_CHECKED 0x08 // has been checked if the value is different from the default
24 avl avl; // the index - this has to be first!
26 uint32_t hash; // a simple hash to speed up searching
27 // we first compare hashes, and only if the hashes are equal we do string comparisons
34 struct config_value *next;
40 uint32_t hash; // a simple hash to speed up searching
41 // we first compare hashes, and only if the hashes are equal we do string comparisons
45 struct config_value *values;
46 avl_tree values_index;
49 } *config_root = NULL;
52 // ----------------------------------------------------------------------------
55 static int config_value_iterator(avl *a) { if(a) {}; return 0; }
57 static int config_value_compare(void* a, void* b) {
58 if(((struct config_value *)a)->hash < ((struct config_value *)b)->hash) return -1;
59 else if(((struct config_value *)a)->hash > ((struct config_value *)b)->hash) return 1;
60 else return strcmp(((struct config_value *)a)->name, ((struct config_value *)b)->name);
63 #define config_value_index_add(co, cv) avl_insert(&((co)->values_index), (avl *)(cv))
64 #define config_value_index_del(co, cv) avl_remove(&((co)->values_index), (avl *)(cv))
66 static struct config_value *config_value_index_find(struct config *co, const char *name, uint32_t hash) {
67 struct config_value *result = NULL, tmp;
68 tmp.hash = (hash)?hash:simple_hash(name);
69 tmp.name = (char *)name;
71 avl_search(&(co->values_index), (avl *)&tmp, config_value_iterator, (avl **)&result);
75 // ----------------------------------------------------------------------------
78 static int config_iterator(avl *a) { if(a) {}; return 0; }
80 static int config_compare(void* a, void* b) {
81 if(((struct config *)a)->hash < ((struct config *)b)->hash) return -1;
82 else if(((struct config *)a)->hash > ((struct config *)b)->hash) return 1;
83 else return strcmp(((struct config *)a)->name, ((struct config *)b)->name);
86 avl_tree config_root_index = {
91 #define config_index_add(cfg) avl_insert(&config_root_index, (avl *)(cfg))
92 #define config_index_del(cfg) avl_remove(&config_root_index, (avl *)(cfg))
94 static struct config *config_index_find(const char *name, uint32_t hash) {
95 struct config *result = NULL, tmp;
96 tmp.hash = (hash)?hash:simple_hash(name);
97 tmp.name = (char *)name;
99 avl_search(&config_root_index, (avl *)&tmp, config_iterator, (avl **)&result);
103 struct config_value *config_value_create(struct config *co, const char *name, const char *value)
105 debug(D_CONFIG, "Creating config entry for name '%s', value '%s', in section '%s'.", name, value, co->name);
107 struct config_value *cv = calloc(1, sizeof(struct config_value));
108 if(!cv) fatal("Cannot allocate config_value");
110 cv->name = strdup(name);
111 if(!cv->name) fatal("Cannot allocate config.name");
112 cv->hash = simple_hash(cv->name);
114 cv->value = strdup(value);
115 if(!cv->value) fatal("Cannot allocate config.value");
117 config_value_index_add(co, cv);
119 // no need for string termination, due to calloc()
121 struct config_value *cv2 = co->values;
123 while (cv2->next) cv2 = cv2->next;
126 else co->values = cv;
131 struct config *config_create(const char *section)
133 debug(D_CONFIG, "Creating section '%s'.", section);
135 struct config *co = calloc(1, sizeof(struct config));
136 if(!co) fatal("Cannot allocate config");
138 co->name = strdup(section);
139 if(!co->name) fatal("Cannot allocate config.name");
140 co->hash = simple_hash(co->name);
142 co->values_index.compar = config_value_compare;
144 config_index_add(co);
146 // no need for string termination, due to calloc()
148 struct config *co2 = config_root;
150 while (co2->next) co2 = co2->next;
153 else config_root = co;
158 struct config *config_find_section(const char *section)
160 return config_index_find(section, 0);
163 int load_config(char *filename, int overwrite_used)
166 struct config *co = NULL;
168 pthread_rwlock_wrlock(&config_rwlock);
170 char buffer[CONFIG_FILE_LINE_MAX + 1], *s;
172 if(!filename) filename = CONFIG_DIR "/" CONFIG_FILENAME;
173 FILE *fp = fopen(filename, "r");
175 error("Cannot open file '%s'", CONFIG_DIR "/" CONFIG_FILENAME);
176 pthread_rwlock_unlock(&config_rwlock);
180 while(fgets(buffer, CONFIG_FILE_LINE_MAX, fp) != NULL) {
181 buffer[CONFIG_FILE_LINE_MAX] = '\0';
186 debug(D_CONFIG, "Ignoring line %d, it is empty.", line);
191 if(*s == '[' && s[len - 1] == ']') {
196 co = config_find_section(s);
197 if(!co) co = config_create(s);
203 // line outside a section
204 error("Ignoring line %d ('%s'), it is outsize all sections.", line, s);
209 char *value = strchr(s, '=');
211 error("Ignoring line %d ('%s'), there is no = in it.", line, s);
221 error("Ignoring line %d, name is empty.", line);
225 debug(D_CONFIG, "Ignoring line %d, value is empty.", line);
229 struct config_value *cv = config_value_index_find(co, name, 0);
231 if(!cv) cv = config_value_create(co, name, value);
233 if(((cv->flags & CONFIG_VALUE_USED) && overwrite_used) || !(cv->flags & CONFIG_VALUE_USED)) {
234 debug(D_CONFIG, "Overwriting '%s/%s'.", line, co->name, cv->name);
236 cv->value = strdup(value);
237 if(!cv->value) fatal("Cannot allocate config.value");
240 debug(D_CONFIG, "Ignoring line %d, '%s/%s' is already present and used.", line, co->name, cv->name);
242 cv->flags |= CONFIG_VALUE_LOADED;
247 pthread_rwlock_unlock(&config_rwlock);
251 char *config_get(const char *section, const char *name, const char *default_value)
253 struct config_value *cv;
255 debug(D_CONFIG, "request to get config in section '%s', name '%s', default_value '%s'", section, name, default_value);
257 pthread_rwlock_rdlock(&config_rwlock);
259 struct config *co = config_find_section(section);
260 if(!co) co = config_create(section);
262 cv = config_value_index_find(co, name, 0);
264 cv = config_value_create(co, name, default_value);
267 cv->flags |= CONFIG_VALUE_USED;
269 if((cv->flags & CONFIG_VALUE_LOADED) || (cv->flags & CONFIG_VALUE_CHANGED)) {
270 // this is a loaded value from the config file
271 // if it is different that the default, mark it
272 if(!(cv->flags & CONFIG_VALUE_CHECKED)) {
273 if(strcmp(cv->value, default_value) != 0) cv->flags |= CONFIG_VALUE_CHANGED;
274 cv->flags |= CONFIG_VALUE_CHECKED;
278 pthread_rwlock_unlock(&config_rwlock);
282 long long config_get_number(const char *section, const char *name, long long value)
284 char buffer[100], *s;
285 sprintf(buffer, "%lld", value);
287 s = config_get(section, name, buffer);
290 return strtoll(s, NULL, 0);
293 int config_get_boolean(const char *section, const char *name, int value)
299 s = config_get(section, name, s);
302 if(!strcmp(s, "yes")) return 1;
306 const char *config_set(const char *section, const char *name, const char *value)
308 struct config_value *cv;
310 debug(D_CONFIG, "request to set config in section '%s', name '%s', value '%s'", section, name, value);
312 pthread_rwlock_wrlock(&config_rwlock);
314 struct config *co = config_find_section(section);
315 if(!co) co = config_create(section);
317 cv = config_value_index_find(co, name, 0);
318 if(!cv) cv = config_value_create(co, name, value);
319 cv->flags |= CONFIG_VALUE_USED;
321 if(strcmp(cv->value, value) != 0) cv->flags |= CONFIG_VALUE_CHANGED;
324 cv->value = strdup(value);
325 if(!cv->value) fatal("Cannot allocate config.value");
327 pthread_rwlock_unlock(&config_rwlock);
332 long long config_set_number(const char *section, const char *name, long long value)
335 sprintf(buffer, "%lld", value);
337 config_set(section, name, buffer);
342 int config_set_boolean(const char *section, const char *name, int value)
348 config_set(section, name, s);
353 void generate_config(struct web_buffer *wb, int only_changed)
357 struct config_value *cv;
359 for(i = 0; i < 3 ;i++) {
360 web_buffer_increase(wb, 500);
363 web_buffer_printf(wb,
364 "# NetData Configuration\n"
365 "# You can uncomment and change any of the options below.\n"
366 "# The value shown in the commented settings, is the default value.\n"
367 "\n# global netdata configuration\n");
371 web_buffer_printf(wb, "\n\n# per plugin configuration\n");
375 web_buffer_printf(wb, "\n\n# per chart configuration\n");
379 for(co = config_root; co ; co = co->next) {
380 if(strcmp(co->name, "global") == 0 || strcmp(co->name, "plugins") == 0) pri = 0;
381 else if(strncmp(co->name, "plugin:", 7) == 0) pri = 1;
388 for(cv = co->values; cv ; cv = cv->next) {
389 used += (cv->flags && CONFIG_VALUE_USED)?1:0;
390 changed += (cv->flags & CONFIG_VALUE_CHANGED)?1:0;
395 if(only_changed && !changed) continue;
398 web_buffer_increase(wb, 500);
399 web_buffer_printf(wb, "\n# node '%s' is not used.", co->name);
402 web_buffer_increase(wb, CONFIG_FILE_LINE_MAX+1);
403 web_buffer_printf(wb, "\n[%s]\n", co->name);
405 for(cv = co->values; cv ; cv = cv->next) {
407 if(used && !(cv->flags & CONFIG_VALUE_USED)) {
408 web_buffer_increase(wb, CONFIG_FILE_LINE_MAX + 1);
409 web_buffer_printf(wb, "\n\t# option '%s' is not used.\n", cv->name);
411 web_buffer_increase(wb, CONFIG_FILE_LINE_MAX + 1);
412 web_buffer_printf(wb, "\t%s%s = %s\n", ((!(cv->flags & CONFIG_VALUE_CHANGED)) && (cv->flags & CONFIG_VALUE_USED))?"# ":"", cv->name, cv->value);