10 #include "appconfig.h"
13 #define CONFIG_FILE_LINE_MAX ((CONFIG_MAX_NAME + CONFIG_MAX_VALUE + 1024) * 2)
15 pthread_rwlock_t config_rwlock = PTHREAD_RWLOCK_INITIALIZER;
17 // ----------------------------------------------------------------------------
20 #define CONFIG_VALUE_LOADED 0x01 // has been loaded from the config
21 #define CONFIG_VALUE_USED 0x02 // has been accessed from the program
22 #define CONFIG_VALUE_CHANGED 0x04 // has been changed from the loaded value
23 #define CONFIG_VALUE_CHECKED 0x08 // has been checked if the value is different from the default
26 avl avl; // the index - this has to be first!
28 uint32_t hash; // a simple hash to speed up searching
29 // we first compare hashes, and only if the hashes are equal we do string comparisons
36 struct config_value *next;
42 uint32_t hash; // a simple hash to speed up searching
43 // we first compare hashes, and only if the hashes are equal we do string comparisons
47 struct config_value *values;
48 avl_tree_lock values_index;
52 pthread_rwlock_t rwlock;
53 } *config_root = NULL;
56 // ----------------------------------------------------------------------------
59 static int config_value_iterator(avl *a) { if(a) {}; return 0; }
61 static int config_value_compare(void* a, void* b) {
62 if(((struct config_value *)a)->hash < ((struct config_value *)b)->hash) return -1;
63 else if(((struct config_value *)a)->hash > ((struct config_value *)b)->hash) return 1;
64 else return strcmp(((struct config_value *)a)->name, ((struct config_value *)b)->name);
67 #define config_value_index_add(co, cv) avl_insert_lock(&((co)->values_index), (avl *)(cv))
68 #define config_value_index_del(co, cv) avl_remove_lock(&((co)->values_index), (avl *)(cv))
70 static struct config_value *config_value_index_find(struct config *co, const char *name, uint32_t hash) {
71 struct config_value *result = NULL, tmp;
72 tmp.hash = (hash)?hash:simple_hash(name);
73 tmp.name = (char *)name;
75 avl_search_lock(&(co->values_index), (avl *) &tmp, config_value_iterator, (avl **) &result);
79 // ----------------------------------------------------------------------------
82 static int config_iterator(avl *a) { if(a) {}; return 0; }
84 static int config_compare(void* a, void* b) {
85 if(((struct config *)a)->hash < ((struct config *)b)->hash) return -1;
86 else if(((struct config *)a)->hash > ((struct config *)b)->hash) return 1;
87 else return strcmp(((struct config *)a)->name, ((struct config *)b)->name);
90 avl_tree_lock config_root_index = {
91 { NULL, config_compare },
95 #define config_index_add(cfg) avl_insert_lock(&config_root_index, (avl *)(cfg))
96 #define config_index_del(cfg) avl_remove_lock(&config_root_index, (avl *)(cfg))
98 static struct config *config_index_find(const char *name, uint32_t hash) {
99 struct config *result = NULL, tmp;
100 tmp.hash = (hash)?hash:simple_hash(name);
101 tmp.name = (char *)name;
103 avl_search_lock(&config_root_index, (avl *) &tmp, config_iterator, (avl **) &result);
107 struct config_value *config_value_create(struct config *co, const char *name, const char *value)
109 debug(D_CONFIG, "Creating config entry for name '%s', value '%s', in section '%s'.", name, value, co->name);
111 struct config_value *cv = calloc(1, sizeof(struct config_value));
112 if(!cv) fatal("Cannot allocate config_value");
114 cv->name = strdup(name);
115 if(!cv->name) fatal("Cannot allocate config.name");
116 cv->hash = simple_hash(cv->name);
118 cv->value = strdup(value);
119 if(!cv->value) fatal("Cannot allocate config.value");
121 config_value_index_add(co, cv);
123 // no need for string termination, due to calloc()
125 pthread_rwlock_wrlock(&co->rwlock);
127 struct config_value *cv2 = co->values;
129 while (cv2->next) cv2 = cv2->next;
132 else co->values = cv;
134 pthread_rwlock_unlock(&co->rwlock);
139 struct config *config_create(const char *section)
141 debug(D_CONFIG, "Creating section '%s'.", section);
143 struct config *co = calloc(1, sizeof(struct config));
144 if(!co) fatal("Cannot allocate config");
146 co->name = strdup(section);
147 if(!co->name) fatal("Cannot allocate config.name");
148 co->hash = simple_hash(co->name);
150 pthread_rwlock_init(&co->rwlock, NULL);
151 avl_init_lock(&co->values_index, config_value_compare);
153 config_index_add(co);
155 // no need for string termination, due to calloc()
157 pthread_rwlock_wrlock(&config_rwlock);
159 struct config *co2 = config_root;
161 while (co2->next) co2 = co2->next;
164 else config_root = co;
166 pthread_rwlock_unlock(&config_rwlock);
171 struct config *config_find_section(const char *section)
173 return config_index_find(section, 0);
176 int load_config(char *filename, int overwrite_used)
179 struct config *co = NULL;
181 char buffer[CONFIG_FILE_LINE_MAX + 1], *s;
183 if(!filename) filename = CONFIG_DIR "/" CONFIG_FILENAME;
184 FILE *fp = fopen(filename, "r");
186 error("Cannot open file '%s'", filename);
190 while(fgets(buffer, CONFIG_FILE_LINE_MAX, fp) != NULL) {
191 buffer[CONFIG_FILE_LINE_MAX] = '\0';
196 debug(D_CONFIG, "Ignoring line %d, it is empty.", line);
200 int len = (int) strlen(s);
201 if(*s == '[' && s[len - 1] == ']') {
206 co = config_find_section(s);
207 if(!co) co = config_create(s);
213 // line outside a section
214 error("Ignoring line %d ('%s'), it is outside all sections.", line, s);
219 char *value = strchr(s, '=');
221 error("Ignoring line %d ('%s'), there is no = in it.", line, s);
231 error("Ignoring line %d, name is empty.", line);
235 debug(D_CONFIG, "Ignoring line %d, value is empty.", line);
239 struct config_value *cv = config_value_index_find(co, name, 0);
241 if(!cv) cv = config_value_create(co, name, value);
243 if(((cv->flags & CONFIG_VALUE_USED) && overwrite_used) || !(cv->flags & CONFIG_VALUE_USED)) {
244 debug(D_CONFIG, "Overwriting '%s/%s'.", line, co->name, cv->name);
246 cv->value = strdup(value);
247 if(!cv->value) fatal("Cannot allocate config.value");
250 debug(D_CONFIG, "Ignoring line %d, '%s/%s' is already present and used.", line, co->name, cv->name);
252 cv->flags |= CONFIG_VALUE_LOADED;
260 char *config_get(const char *section, const char *name, const char *default_value)
262 struct config_value *cv;
264 debug(D_CONFIG, "request to get config in section '%s', name '%s', default_value '%s'", section, name, default_value);
266 struct config *co = config_find_section(section);
267 if(!co) co = config_create(section);
269 cv = config_value_index_find(co, name, 0);
271 cv = config_value_create(co, name, default_value);
274 cv->flags |= CONFIG_VALUE_USED;
276 if((cv->flags & CONFIG_VALUE_LOADED) || (cv->flags & CONFIG_VALUE_CHANGED)) {
277 // this is a loaded value from the config file
278 // if it is different that the default, mark it
279 if(!(cv->flags & CONFIG_VALUE_CHECKED)) {
280 if(strcmp(cv->value, default_value) != 0) cv->flags |= CONFIG_VALUE_CHANGED;
281 cv->flags |= CONFIG_VALUE_CHECKED;
288 long long config_get_number(const char *section, const char *name, long long value)
290 char buffer[100], *s;
291 sprintf(buffer, "%lld", value);
293 s = config_get(section, name, buffer);
296 return strtoll(s, NULL, 0);
299 int config_get_boolean(const char *section, const char *name, int value)
305 s = config_get(section, name, s);
308 if(!strcmp(s, "yes")) return 1;
312 int config_get_boolean_ondemand(const char *section, const char *name, int value)
316 if(value == CONFIG_ONDEMAND_ONDEMAND)
319 else if(value == CONFIG_ONDEMAND_NO)
325 s = config_get(section, name, s);
328 if(!strcmp(s, "yes"))
329 return CONFIG_ONDEMAND_YES;
330 else if(!strcmp(s, "no"))
331 return CONFIG_ONDEMAND_NO;
332 else if(!strcmp(s, "on demand"))
333 return CONFIG_ONDEMAND_ONDEMAND;
338 const char *config_set_default(const char *section, const char *name, const char *value)
340 struct config_value *cv;
342 debug(D_CONFIG, "request to set config in section '%s', name '%s', value '%s'", section, name, value);
344 struct config *co = config_find_section(section);
345 if(!co) return config_set(section, name, value);
347 cv = config_value_index_find(co, name, 0);
348 if(!cv) return config_set(section, name, value);
350 cv->flags |= CONFIG_VALUE_USED;
352 if(cv->flags & CONFIG_VALUE_LOADED)
355 if(strcmp(cv->value, value) != 0) {
356 cv->flags |= CONFIG_VALUE_CHANGED;
359 cv->value = strdup(value);
360 if(!cv->value) fatal("Cannot allocate config.value");
366 const char *config_set(const char *section, const char *name, const char *value)
368 struct config_value *cv;
370 debug(D_CONFIG, "request to set config in section '%s', name '%s', value '%s'", section, name, value);
372 struct config *co = config_find_section(section);
373 if(!co) co = config_create(section);
375 cv = config_value_index_find(co, name, 0);
376 if(!cv) cv = config_value_create(co, name, value);
377 cv->flags |= CONFIG_VALUE_USED;
379 if(strcmp(cv->value, value) != 0) {
380 cv->flags |= CONFIG_VALUE_CHANGED;
383 cv->value = strdup(value);
384 if(!cv->value) fatal("Cannot allocate config.value");
390 long long config_set_number(const char *section, const char *name, long long value)
393 sprintf(buffer, "%lld", value);
395 config_set(section, name, buffer);
400 int config_set_boolean(const char *section, const char *name, int value)
406 config_set(section, name, s);
411 void generate_config(BUFFER *wb, int only_changed)
415 struct config_value *cv;
417 for(i = 0; i < 3 ;i++) {
421 "# NetData Configuration\n"
422 "# You can uncomment and change any of the options below.\n"
423 "# The value shown in the commented settings, is the default value.\n"
424 "\n# global netdata configuration\n");
428 buffer_strcat(wb, "\n\n# per plugin configuration\n");
432 buffer_strcat(wb, "\n\n# per chart configuration\n");
436 pthread_rwlock_wrlock(&config_rwlock);
437 for(co = config_root; co ; co = co->next) {
438 if(strcmp(co->name, "global") == 0 || strcmp(co->name, "plugins") == 0 || strcmp(co->name, "registry") == 0) pri = 0;
439 else if(strncmp(co->name, "plugin:", 7) == 0) pri = 1;
447 pthread_rwlock_wrlock(&co->rwlock);
449 for(cv = co->values; cv ; cv = cv->next) {
450 used += (cv->flags & CONFIG_VALUE_USED)?1:0;
451 changed += (cv->flags & CONFIG_VALUE_CHANGED)?1:0;
455 pthread_rwlock_unlock(&co->rwlock);
458 if(only_changed && !changed) continue;
461 buffer_sprintf(wb, "\n# node '%s' is not used.", co->name);
464 buffer_sprintf(wb, "\n[%s]\n", co->name);
466 pthread_rwlock_wrlock(&co->rwlock);
467 for(cv = co->values; cv ; cv = cv->next) {
469 if(used && !(cv->flags & CONFIG_VALUE_USED)) {
470 buffer_sprintf(wb, "\n\t# option '%s' is not used.\n", cv->name);
472 buffer_sprintf(wb, "\t%s%s = %s\n", ((!(cv->flags & CONFIG_VALUE_CHANGED)) && (cv->flags & CONFIG_VALUE_USED))?"# ":"", cv->name, cv->value);
474 pthread_rwlock_unlock(&co->rwlock);
477 pthread_rwlock_unlock(&config_rwlock);