src/config.c

   1 #include <pthread.h>
   2 #include <stdlib.h>
   3 #include <string.h>
   4 #include <inttypes.h>
   5
   6 #include "avl.h"
   7 #include "common.h"
   8 #include "config.h"
   9 #include "log.h"
  10
  11 #define CONFIG_FILE_LINE_MAX 4096
  12
  13 pthread_rwlock_t config_rwlock = PTHREAD_RWLOCK_INITIALIZER;
  14
  15 // ----------------------------------------------------------------------------
  16 // definitions
  17
  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
  22
  23 struct config_value {
  24         avl avl;                                // the index - this has to be first!
  25
  26         unsigned long hash;             // a simple hash to speed up searching
  27                                                         // we first compare hashes, and only if the hashes are equal we do string comparisons
  28
  29         char name[CONFIG_MAX_NAME + 1];
  30         char value[CONFIG_MAX_VALUE + 1];
  31
  32         uint8_t flags;
  33
  34         struct config_value *next;
  35 };
  36
  37 struct config {
  38         avl avl;
  39
  40         unsigned long hash;             // a simple hash to speed up searching
  41                                                         // we first compare hashes, and only if the hashes are equal we do string comparisons
  42
  43         char name[CONFIG_MAX_NAME + 1];
  44
  45         struct config_value *values;
  46         avl_tree values_index;
  47
  48         struct config *next;
  49 } *config_root = NULL;
  50
  51
  52 // ----------------------------------------------------------------------------
  53 // config value
  54
  55 static int config_value_iterator(avl *a) { if(a) {}; return 0; }
  56
  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);
  61 }
  62
  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))
  65
  66 static struct config_value *config_value_index_find(struct config *co, const char *name, unsigned long hash) {
  67         struct config_value *result = NULL, tmp;
  68         tmp.hash = (hash)?hash:simple_hash(name);
  69         strncpy(tmp.name, name, CONFIG_MAX_NAME);
  70         tmp.name[CONFIG_MAX_NAME] = '\0';
  71
  72         avl_search(&(co->values_index), (avl *)&tmp, config_value_iterator, (avl **)&result);
  73         return result;
  74 }
  75
  76 // ----------------------------------------------------------------------------
  77 // config
  78
  79 static int config_iterator(avl *a) { if(a) {}; return 0; }
  80
  81 static int config_compare(void* a, void* b) {
  82         if(((struct config *)a)->hash < ((struct config *)b)->hash) return -1;
  83         else if(((struct config *)a)->hash > ((struct config *)b)->hash) return 1;
  84         else return strcmp(((struct config *)a)->name, ((struct config *)b)->name);
  85 }
  86
  87 avl_tree config_root_index = {
  88                 NULL,
  89                 config_compare
  90 };
  91
  92 #define config_index_add(cfg) avl_insert(&config_root_index, (avl *)(cfg))
  93 #define config_index_del(cfg) avl_remove(&config_root_index, (avl *)(cfg))
  94
  95 static struct config *config_index_find(const char *name, unsigned long hash) {
  96         struct config *result = NULL, tmp;
  97         tmp.hash = (hash)?hash:simple_hash(name);
  98         strncpy(tmp.name, name, CONFIG_MAX_NAME);
  99         tmp.name[CONFIG_MAX_NAME] = '\0';
 100
 101         avl_search(&config_root_index, (avl *)&tmp, config_iterator, (avl **)&result);
 102         return result;
 103 }
 104
 105 struct config_value *config_value_create(struct config *co, const char *name, const char *value)
 106 {
 107         debug(D_CONFIG, "Creating config entry for name '%s', value '%s', in section '%s'.", name, value, co->name);
 108
 109         struct config_value *cv = calloc(1, sizeof(struct config_value));
 110         if(!cv) fatal("Cannot allocate config_value");
 111
 112         strncpy(cv->name,  name,  CONFIG_MAX_NAME);
 113         strncpy(cv->value, value, CONFIG_MAX_VALUE);
 114         cv->hash = simple_hash(cv->name);
 115
 116         config_value_index_add(co, cv);
 117
 118         // no need for string termination, due to calloc()
 119
 120         struct config_value *cv2 = co->values;
 121         if(cv2) {
 122                 while (cv2->next) cv2 = cv2->next;
 123                 cv2->next = cv;
 124         }
 125         else co->values = cv;
 126
 127         return cv;
 128 }
 129
 130 struct config *config_create(const char *section)
 131 {
 132         debug(D_CONFIG, "Creating section '%s'.", section);
 133
 134         struct config *co = calloc(1, sizeof(struct config));
 135         if(!co) fatal("Cannot allocate config");
 136
 137         strncpy(co->name, section, CONFIG_MAX_NAME);
 138         co->hash = simple_hash(co->name);
 139         co->values_index.compar = config_value_compare;
 140
 141         config_index_add(co);
 142
 143         // no need for string termination, due to calloc()
 144
 145         struct config *co2 = config_root;
 146         if(co2) {
 147                 while (co2->next) co2 = co2->next;
 148                 co2->next = co;
 149         }
 150         else config_root = co;
 151
 152         return co;
 153 }
 154
 155 struct config *config_find_section(const char *section)
 156 {
 157         return config_index_find(section, 0);
 158 }
 159
 160 int load_config(char *filename, int overwrite_used)
 161 {
 162         int line = 0;
 163         struct config *co = NULL;
 164
 165         pthread_rwlock_wrlock(&config_rwlock);
 166
 167         char buffer[CONFIG_FILE_LINE_MAX + 1], *s;
 168
 169         if(!filename) filename = CONFIG_DIR "/" CONFIG_FILENAME;
 170         FILE *fp = fopen(filename, "r");
 171         if(!fp) {
 172                 error("Cannot open file '%s'", CONFIG_DIR "/" CONFIG_FILENAME);
 173                 pthread_rwlock_unlock(&config_rwlock);
 174                 return 0;
 175         }
 176
 177         while(fgets(buffer, CONFIG_FILE_LINE_MAX, fp) != NULL) {
 178                 buffer[CONFIG_FILE_LINE_MAX] = '\0';
 179                 line++;
 180
 181                 s = trim(buffer);
 182                 if(!s) {
 183                         debug(D_CONFIG, "Ignoring line %d, it is empty.", line);
 184                         continue;
 185                 }
 186
 187                 int len = strlen(s);
 188                 if(*s == '[' && s[len - 1] == ']') {
 189                         // new section
 190                         s[len - 1] = '\0';
 191                         s++;
 192
 193                         co = config_find_section(s);
 194                         if(!co) co = config_create(s);
 195
 196                         continue;
 197                 }
 198
 199                 if(!co) {
 200                         // line outside a section
 201                         error("Ignoring line %d ('%s'), it is outsize all sections.", line, s);
 202                         continue;
 203                 }
 204
 205                 char *name = s;
 206                 char *value = strchr(s, '=');
 207                 if(!value) {
 208                         error("Ignoring line %d ('%s'), there is no = in it.", line, s);
 209                         continue;
 210                 }
 211                 *value = '\0';
 212                 value++;
 213
 214                 name = trim(name);
 215                 value = trim(value);
 216
 217                 if(!name) {
 218                         error("Ignoring line %d, name is empty.", line);
 219                         continue;
 220                 }
 221                 if(!value) {
 222                         debug(D_CONFIG, "Ignoring line %d, value is empty.", line);
 223                         continue;
 224                 }
 225
 226                 struct config_value *cv = config_value_index_find(co, name, 0);
 227
 228                 if(!cv) cv = config_value_create(co, name, value);
 229                 else {
 230                         if((cv->flags & CONFIG_VALUE_USED && overwrite_used) || !(cv->flags & CONFIG_VALUE_USED)) {
 231                                 debug(D_CONFIG, "Overwriting '%s/%s'.", line, co->name, cv->name);
 232                                 strncpy(cv->value, value, CONFIG_MAX_VALUE);
 233                                 // termination is already there
 234                         }
 235                         else
 236                                 debug(D_CONFIG, "Ignoring line %d, '%s/%s' is already present and used.", line, co->name, cv->name);
 237                 }
 238                 cv->flags |= CONFIG_VALUE_LOADED;
 239         }
 240
 241         fclose(fp);
 242
 243         pthread_rwlock_unlock(&config_rwlock);
 244         return 1;
 245 }
 246
 247 char *config_get(const char *section, const char *name, const char *default_value)
 248 {
 249         struct config_value *cv;
 250
 251         debug(D_CONFIG, "request to get config in section '%s', name '%s', default_value '%s'", section, name, default_value);
 252
 253         pthread_rwlock_rdlock(&config_rwlock);
 254
 255         struct config *co = config_find_section(section);
 256         if(!co) co = config_create(section);
 257
 258         cv = config_value_index_find(co, name, 0);
 259         if(!cv) cv = config_value_create(co, name, default_value);
 260         cv->flags |= CONFIG_VALUE_USED;
 261
 262         if(cv->flags & CONFIG_VALUE_LOADED || cv->flags & CONFIG_VALUE_CHANGED) {
 263                 // this is a loaded value from the config file
 264                 // if it is different that the default, mark it
 265                 if(!(cv->flags & CONFIG_VALUE_CHECKED)) {
 266                         if(strcmp(cv->value, default_value) != 0) cv->flags |= CONFIG_VALUE_CHANGED;
 267                         cv->flags |= CONFIG_VALUE_CHECKED;
 268                 }
 269         }
 270         else {
 271                 // this is not loaded from the config
 272                 // copy the default value to it
 273                 strncpy(cv->value, default_value, CONFIG_MAX_VALUE);
 274         }
 275
 276         pthread_rwlock_unlock(&config_rwlock);
 277         return(cv->value);
 278 }
 279
 280 long long config_get_number(const char *section, const char *name, long long value)
 281 {
 282         char buffer[100], *s;
 283         sprintf(buffer, "%lld", value);
 284
 285         s = config_get(section, name, buffer);
 286         return strtoll(s, NULL, 0);
 287 }
 288
 289 int config_get_boolean(const char *section, const char *name, int value)
 290 {
 291         char *s;
 292         if(value) s = "yes";
 293         else s = "no";
 294
 295         s = config_get(section, name, s);
 296
 297         if(strcmp(s, "yes") == 0 || strcmp(s, "true") == 0 || strcmp(s, "1") == 0) {
 298                 strcpy(s, "yes");
 299                 return 1;
 300         }
 301         else {
 302                 strcpy(s, "no");
 303                 return 0;
 304         }
 305 }
 306
 307 const char *config_set(const char *section, const char *name, const char *value)
 308 {
 309         struct config_value *cv;
 310
 311         debug(D_CONFIG, "request to set config in section '%s', name '%s', value '%s'", section, name, value);
 312
 313         pthread_rwlock_wrlock(&config_rwlock);
 314
 315         struct config *co = config_find_section(section);
 316         if(!co) co = config_create(section);
 317
 318         cv = config_value_index_find(co, name, 0);
 319         if(!cv) cv = config_value_create(co, name, value);
 320         cv->flags |= CONFIG_VALUE_USED;
 321
 322         if(strcmp(cv->value, value) != 0) cv->flags |= CONFIG_VALUE_CHANGED;
 323
 324         strncpy(cv->value, value, CONFIG_MAX_VALUE);
 325         // termination is already there
 326
 327         pthread_rwlock_unlock(&config_rwlock);
 328
 329         return value;
 330 }
 331
 332 long long config_set_number(const char *section, const char *name, long long value)
 333 {
 334         char buffer[100];
 335         sprintf(buffer, "%lld", value);
 336
 337         config_set(section, name, buffer);
 338
 339         return value;
 340 }
 341
 342 int config_set_boolean(const char *section, const char *name, int value)
 343 {
 344         char *s;
 345         if(value) s = "yes";
 346         else s = "no";
 347
 348         config_set(section, name, s);
 349
 350         return value;
 351 }
 352
 353 void generate_config(struct web_buffer *wb, int only_changed)
 354 {
 355         int i, pri;
 356         struct config *co;
 357         struct config_value *cv;
 358
 359         for(i = 0; i < 3 ;i++) {
 360                 web_buffer_increase(wb, 500);
 361                 switch(i) {
 362                         case 0:
 363                                 web_buffer_printf(wb,
 364                                         "# NetData Configuration\n"
 365                                         "# You can uncomment and change any of the options bellow.\n"
 366                                         "# The value shown in the commented settings, is the default value.\n"
 367                                         "\n# global netdata configuration\n");
 368                                 break;
 369
 370                         case 1:
 371                                 web_buffer_printf(wb, "\n\n# per plugin configuration\n");
 372                                 break;
 373
 374                         case 2:
 375                                 web_buffer_printf(wb, "\n\n# per chart configuration\n");
 376                                 break;
 377                 }
 378
 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;
 382                         else pri = 2;
 383
 384                         if(i == pri) {
 385                                 int used = 0;
 386                                 int changed = 0;
 387                                 int count = 0;
 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;
 391                                         count++;
 392                                 }
 393
 394                                 if(!count) continue;
 395                                 if(only_changed && !changed) continue;
 396
 397                                 if(!used) {
 398                                         web_buffer_increase(wb, 500);
 399                                         web_buffer_printf(wb, "\n# node '%s' is not used.", co->name);
 400                                 }
 401
 402                                 web_buffer_increase(wb, CONFIG_MAX_NAME + 4);
 403                                 web_buffer_printf(wb, "\n[%s]\n", co->name);
 404
 405                                 for(cv = co->values; cv ; cv = cv->next) {
 406
 407                                         if(used && !(cv->flags & CONFIG_VALUE_USED)) {
 408                                                 web_buffer_increase(wb, CONFIG_MAX_NAME + 200);
 409                                                 web_buffer_printf(wb, "\n\t# option '%s' is not used.\n", cv->name);
 410                                         }
 411                                         web_buffer_increase(wb, CONFIG_MAX_NAME + CONFIG_MAX_VALUE + 5);
 412                                         web_buffer_printf(wb, "\t%s%s = %s\n", ((!(cv->flags & CONFIG_VALUE_CHANGED)) && (cv->flags & CONFIG_VALUE_USED))?"# ":"", cv->name, cv->value);
 413                                 }
 414                         }
 415                 }
 416         }
 417 }
 418