src/daemon/configfile.c

   1 /**
   2  * collectd - src/configfile.c
   3  * Copyright (C) 2005-2011  Florian octo Forster
   4  *
   5  * Permission is hereby granted, free of charge, to any person obtaining a
   6  * copy of this software and associated documentation files (the "Software"),
   7  * to deal in the Software without restriction, including without limitation
   8  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   9  * and/or sell copies of the Software, and to permit persons to whom the
  10  * Software is furnished to do so, subject to the following conditions:
  11  *
  12  * The above copyright notice and this permission notice shall be included in
  13  * all copies or substantial portions of the Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  21  * DEALINGS IN THE SOFTWARE.
  22  *
  23  * Authors:
  24  *   Florian octo Forster <octo at collectd.org>
  25  *   Sebastian tokkee Harl <sh at tokkee.org>
  26  **/
  27
  28 #include "collectd.h"
  29
  30 #include "liboconfig/oconfig.h"
  31
  32 #include "common.h"
  33 #include "configfile.h"
  34 #include "filter_chain.h"
  35 #include "plugin.h"
  36 #include "types_list.h"
  37
  38 #if HAVE_WORDEXP_H
  39 #include <wordexp.h>
  40 #endif /* HAVE_WORDEXP_H */
  41
  42 #if HAVE_FNMATCH_H
  43 #include <fnmatch.h>
  44 #endif /* HAVE_FNMATCH_H */
  45
  46 #if HAVE_LIBGEN_H
  47 #include <libgen.h>
  48 #endif /* HAVE_LIBGEN_H */
  49
  50 #define ESCAPE_NULL(str) ((str) == NULL ? "(null)" : (str))
  51
  52 /*
  53  * Private types
  54  */
  55 typedef struct cf_callback {
  56   const char *type;
  57   int (*callback)(const char *, const char *);
  58   const char **keys;
  59   int keys_num;
  60   plugin_ctx_t ctx;
  61   struct cf_callback *next;
  62 } cf_callback_t;
  63
  64 typedef struct cf_complex_callback_s {
  65   char *type;
  66   int (*callback)(oconfig_item_t *);
  67   plugin_ctx_t ctx;
  68   struct cf_complex_callback_s *next;
  69 } cf_complex_callback_t;
  70
  71 typedef struct cf_value_map_s {
  72   const char *key;
  73   int (*func)(oconfig_item_t *);
  74 } cf_value_map_t;
  75
  76 typedef struct cf_global_option_s {
  77   const char *key;
  78   char *value;
  79   bool from_cli; /* value set from CLI */
  80   const char *def;
  81 } cf_global_option_t;
  82
  83 /*
  84  * Prototypes of callback functions
  85  */
  86 static int dispatch_value_typesdb(oconfig_item_t *ci);
  87 static int dispatch_value_plugindir(oconfig_item_t *ci);
  88 static int dispatch_loadplugin(oconfig_item_t *ci);
  89 static int dispatch_block_plugin(oconfig_item_t *ci);
  90
  91 /*
  92  * Private variables
  93  */
  94 static cf_callback_t *first_callback;
  95 static cf_complex_callback_t *complex_callback_head;
  96
  97 static cf_value_map_t cf_value_map[] = {{"TypesDB", dispatch_value_typesdb},
  98                                         {"PluginDir", dispatch_value_plugindir},
  99                                         {"LoadPlugin", dispatch_loadplugin},
 100                                         {"Plugin", dispatch_block_plugin}};
 101 static int cf_value_map_num = STATIC_ARRAY_SIZE(cf_value_map);
 102
 103 static cf_global_option_t cf_global_options[] = {
 104     {"BaseDir", NULL, 0, PKGLOCALSTATEDIR},
 105     {"PIDFile", NULL, 0, PIDFILE},
 106     {"Hostname", NULL, 0, NULL},
 107     {"FQDNLookup", NULL, 0, "true"},
 108     {"Interval", NULL, 0, NULL},
 109     {"ReadThreads", NULL, 0, "5"},
 110     {"WriteThreads", NULL, 0, "5"},
 111     {"WriteQueueLimitHigh", NULL, 0, NULL},
 112     {"WriteQueueLimitLow", NULL, 0, NULL},
 113     {"Timeout", NULL, 0, "2"},
 114     {"AutoLoadPlugin", NULL, 0, "false"},
 115     {"CollectInternalStats", NULL, 0, "false"},
 116     {"PreCacheChain", NULL, 0, "PreCache"},
 117     {"PostCacheChain", NULL, 0, "PostCache"},
 118     {"MaxReadInterval", NULL, 0, "86400"}};
 119 static int cf_global_options_num = STATIC_ARRAY_SIZE(cf_global_options);
 120
 121 static int cf_default_typesdb = 1;
 122
 123 /*
 124  * Functions to handle register/unregister, search, and other plugin related
 125  * stuff
 126  */
 127 static cf_callback_t *cf_search(const char *type) {
 128   cf_callback_t *cf_cb;
 129
 130   if (type == NULL)
 131     return NULL;
 132
 133   for (cf_cb = first_callback; cf_cb != NULL; cf_cb = cf_cb->next)
 134     if (strcasecmp(cf_cb->type, type) == 0)
 135       break;
 136
 137   return cf_cb;
 138 }
 139
 140 static int cf_dispatch(const char *type, const char *orig_key,
 141                        const char *orig_value) {
 142   cf_callback_t *cf_cb;
 143   plugin_ctx_t old_ctx;
 144   char *key;
 145   char *value;
 146   int ret;
 147   int i = 0;
 148
 149   if (orig_key == NULL)
 150     return EINVAL;
 151
 152   DEBUG("type = %s, key = %s, value = %s", ESCAPE_NULL(type), orig_key,
 153         ESCAPE_NULL(orig_value));
 154
 155   if ((cf_cb = cf_search(type)) == NULL) {
 156     WARNING("Found a configuration for the `%s' plugin, but "
 157             "the plugin isn't loaded or didn't register "
 158             "a configuration callback.",
 159             type);
 160     return -1;
 161   }
 162
 163   if ((key = strdup(orig_key)) == NULL)
 164     return 1;
 165   if ((value = strdup(orig_value)) == NULL) {
 166     free(key);
 167     return 2;
 168   }
 169
 170   ret = -1;
 171
 172   old_ctx = plugin_set_ctx(cf_cb->ctx);
 173
 174   for (i = 0; i < cf_cb->keys_num; i++) {
 175     if ((cf_cb->keys[i] != NULL) && (strcasecmp(cf_cb->keys[i], key) == 0)) {
 176       ret = (*cf_cb->callback)(key, value);
 177       break;
 178     }
 179   }
 180
 181   plugin_set_ctx(old_ctx);
 182
 183   if (i >= cf_cb->keys_num)
 184     WARNING("Plugin `%s' did not register for value `%s'.", type, key);
 185
 186   free(key);
 187   free(value);
 188
 189   return ret;
 190 } /* int cf_dispatch */
 191
 192 static int dispatch_global_option(const oconfig_item_t *ci) {
 193   if (ci->values_num != 1)
 194     return -1;
 195   if (ci->values[0].type == OCONFIG_TYPE_STRING)
 196     return global_option_set(ci->key, ci->values[0].value.string, 0);
 197   else if (ci->values[0].type == OCONFIG_TYPE_NUMBER) {
 198     char tmp[128];
 199     snprintf(tmp, sizeof(tmp), "%lf", ci->values[0].value.number);
 200     return global_option_set(ci->key, tmp, 0);
 201   } else if (ci->values[0].type == OCONFIG_TYPE_BOOLEAN) {
 202     if (ci->values[0].value.boolean)
 203       return global_option_set(ci->key, "true", 0);
 204     else
 205       return global_option_set(ci->key, "false", 0);
 206   }
 207
 208   return -1;
 209 } /* int dispatch_global_option */
 210
 211 static int dispatch_value_typesdb(oconfig_item_t *ci) {
 212   assert(strcasecmp(ci->key, "TypesDB") == 0);
 213
 214   cf_default_typesdb = 0;
 215
 216   if (ci->values_num < 1) {
 217     ERROR("configfile: `TypesDB' needs at least one argument.");
 218     return -1;
 219   }
 220
 221   for (int i = 0; i < ci->values_num; ++i) {
 222     if (OCONFIG_TYPE_STRING != ci->values[i].type) {
 223       WARNING("configfile: TypesDB: Skipping %i. argument which "
 224               "is not a string.",
 225               i + 1);
 226       continue;
 227     }
 228
 229     read_types_list(ci->values[i].value.string);
 230   }
 231   return 0;
 232 } /* int dispatch_value_typesdb */
 233
 234 static int dispatch_value_plugindir(oconfig_item_t *ci) {
 235   assert(strcasecmp(ci->key, "PluginDir") == 0);
 236
 237   if (ci->values_num != 1)
 238     return -1;
 239   if (ci->values[0].type != OCONFIG_TYPE_STRING)
 240     return -1;
 241
 242   plugin_set_dir(ci->values[0].value.string);
 243   return 0;
 244 }
 245
 246 static int dispatch_loadplugin(oconfig_item_t *ci) {
 247   const char *name;
 248   bool global = false;
 249   plugin_ctx_t ctx = {0};
 250   plugin_ctx_t old_ctx;
 251   int ret_val;
 252
 253   assert(strcasecmp(ci->key, "LoadPlugin") == 0);
 254
 255   if (ci->values_num != 1)
 256     return -1;
 257   if (ci->values[0].type != OCONFIG_TYPE_STRING)
 258     return -1;
 259
 260   name = ci->values[0].value.string;
 261   if (strcmp("libvirt", name) == 0)
 262     name = "virt";
 263
 264   /* default to the global interval set before loading this plugin */
 265   ctx.interval = cf_get_default_interval();
 266   ctx.flush_interval = 0;
 267   ctx.flush_timeout = 0;
 268
 269   for (int i = 0; i < ci->children_num; ++i) {
 270     oconfig_item_t *child = ci->children + i;
 271
 272     if (strcasecmp("Globals", child->key) == 0)
 273       cf_util_get_boolean(child, &global);
 274     else if (strcasecmp("Interval", child->key) == 0)
 275       cf_util_get_cdtime(child, &ctx.interval);
 276     else if (strcasecmp("FlushInterval", child->key) == 0)
 277       cf_util_get_cdtime(child, &ctx.flush_interval);
 278     else if (strcasecmp("FlushTimeout", child->key) == 0)
 279       cf_util_get_cdtime(child, &ctx.flush_timeout);
 280     else {
 281       WARNING("Ignoring unknown LoadPlugin option \"%s\" "
 282               "for plugin \"%s\"",
 283               child->key, ci->values[0].value.string);
 284     }
 285   }
 286
 287   old_ctx = plugin_set_ctx(ctx);
 288   ret_val = plugin_load(name, global);
 289   /* reset to the "global" context */
 290   plugin_set_ctx(old_ctx);
 291
 292   return ret_val;
 293 } /* int dispatch_value_loadplugin */
 294
 295 static int dispatch_value_plugin(const char *plugin, oconfig_item_t *ci) {
 296   char buffer[4096];
 297   char *buffer_ptr;
 298   int buffer_free;
 299
 300   buffer_ptr = buffer;
 301   buffer_free = sizeof(buffer);
 302
 303   for (int i = 0; i < ci->values_num; i++) {
 304     int status = -1;
 305
 306     if (ci->values[i].type == OCONFIG_TYPE_STRING)
 307       status =
 308           snprintf(buffer_ptr, buffer_free, " %s", ci->values[i].value.string);
 309     else if (ci->values[i].type == OCONFIG_TYPE_NUMBER)
 310       status =
 311           snprintf(buffer_ptr, buffer_free, " %lf", ci->values[i].value.number);
 312     else if (ci->values[i].type == OCONFIG_TYPE_BOOLEAN)
 313       status = snprintf(buffer_ptr, buffer_free, " %s",
 314                         ci->values[i].value.boolean ? "true" : "false");
 315
 316     if ((status < 0) || (status >= buffer_free))
 317       return -1;
 318     buffer_free -= status;
 319     buffer_ptr += status;
 320   }
 321   /* skip the initial space */
 322   buffer_ptr = buffer + 1;
 323
 324   return cf_dispatch(plugin, ci->key, buffer_ptr);
 325 } /* int dispatch_value_plugin */
 326
 327 static int dispatch_value(oconfig_item_t *ci) {
 328   int ret = 0;
 329
 330   for (int i = 0; i < cf_value_map_num; i++)
 331     if (strcasecmp(cf_value_map[i].key, ci->key) == 0) {
 332       ret = cf_value_map[i].func(ci);
 333       break;
 334     }
 335
 336   for (int i = 0; i < cf_global_options_num; i++)
 337     if (strcasecmp(cf_global_options[i].key, ci->key) == 0) {
 338       ret = dispatch_global_option(ci);
 339       break;
 340     }
 341
 342   return ret;
 343 } /* int dispatch_value */
 344
 345 static int dispatch_block_plugin(oconfig_item_t *ci) {
 346   const char *name;
 347
 348   if (strcasecmp(ci->key, "Plugin") != 0)
 349     return -1;
 350   if (ci->values_num < 1)
 351     return -1;
 352   if (ci->values[0].type != OCONFIG_TYPE_STRING)
 353     return -1;
 354
 355   name = ci->values[0].value.string;
 356   if (strcmp("libvirt", name) == 0) {
 357     /* TODO(octo): Remove this legacy. */
 358     WARNING("The \"libvirt\" plugin has been renamed to \"virt\" to avoid "
 359             "problems with the build system. "
 360             "Your configuration is still using the old name. "
 361             "Please change it to use \"virt\" as soon as possible. "
 362             "This compatibility code will go away eventually.");
 363     name = "virt";
 364   }
 365
 366   if (IS_TRUE(global_option_get("AutoLoadPlugin"))) {
 367     plugin_ctx_t ctx = {0};
 368     plugin_ctx_t old_ctx;
 369     int status;
 370
 371     /* default to the global interval set before loading this plugin */
 372     ctx.interval = cf_get_default_interval();
 373
 374     old_ctx = plugin_set_ctx(ctx);
 375     status = plugin_load(name, /* flags = */ false);
 376     /* reset to the "global" context */
 377     plugin_set_ctx(old_ctx);
 378
 379     if (status != 0) {
 380       ERROR("Automatically loading plugin \"%s\" failed "
 381             "with status %i.",
 382             name, status);
 383       return status;
 384     }
 385   }
 386
 387   /* Check for a complex callback first */
 388   for (cf_complex_callback_t *cb = complex_callback_head; cb != NULL;
 389        cb = cb->next) {
 390     if (strcasecmp(name, cb->type) == 0) {
 391       plugin_ctx_t old_ctx;
 392       int ret_val;
 393
 394       old_ctx = plugin_set_ctx(cb->ctx);
 395       ret_val = (cb->callback(ci));
 396       plugin_set_ctx(old_ctx);
 397       return ret_val;
 398     }
 399   }
 400
 401   /* Hm, no complex plugin found. Dispatch the values one by one */
 402   for (int i = 0; i < ci->children_num; i++) {
 403     if (ci->children[i].children == NULL)
 404       dispatch_value_plugin(name, ci->children + i);
 405     else {
 406       WARNING("There is a `%s' block within the "
 407               "configuration for the %s plugin. "
 408               "The plugin either only expects "
 409               "\"simple\" configuration statements "
 410               "or wasn't loaded using `LoadPlugin'."
 411               " Please check your configuration.",
 412               ci->children[i].key, name);
 413     }
 414   }
 415
 416   return 0;
 417 }
 418
 419 static int dispatch_block(oconfig_item_t *ci) {
 420   if (strcasecmp(ci->key, "LoadPlugin") == 0)
 421     return dispatch_loadplugin(ci);
 422   else if (strcasecmp(ci->key, "Plugin") == 0)
 423     return dispatch_block_plugin(ci);
 424   else if (strcasecmp(ci->key, "Chain") == 0)
 425     return fc_configure(ci);
 426
 427   return 0;
 428 }
 429
 430 static int cf_ci_replace_child(oconfig_item_t *dst, oconfig_item_t *src,
 431                                int offset) {
 432   oconfig_item_t *temp;
 433
 434   assert(offset >= 0);
 435   assert(dst->children_num > offset);
 436
 437   /* Free the memory used by the replaced child. Usually that's the
 438    * `Include "blah"' statement. */
 439   temp = dst->children + offset;
 440   for (int i = 0; i < temp->values_num; i++) {
 441     if (temp->values[i].type == OCONFIG_TYPE_STRING) {
 442       sfree(temp->values[i].value.string);
 443     }
 444   }
 445   sfree(temp->values);
 446   temp = NULL;
 447
 448   /* If (src->children_num == 0) the array size is decreased. If offset
 449    * is _not_ the last element, (offset < (dst->children_num - 1)), then
 450    * we need to move the trailing elements before resizing the array. */
 451   if ((src->children_num == 0) && (offset < (dst->children_num - 1))) {
 452     int nmemb = dst->children_num - (offset + 1);
 453     memmove(dst->children + offset, dst->children + offset + 1,
 454             sizeof(oconfig_item_t) * nmemb);
 455   }
 456
 457   /* Resize the memory containing the children to be big enough to hold
 458    * all children. */
 459   if (dst->children_num + src->children_num - 1 == 0) {
 460     dst->children_num = 0;
 461     return 0;
 462   }
 463
 464   temp = realloc(dst->children,
 465                  sizeof(oconfig_item_t) *
 466                      (dst->children_num + src->children_num - 1));
 467   if (temp == NULL) {
 468     ERROR("configfile: realloc failed.");
 469     return -1;
 470   }
 471   dst->children = temp;
 472
 473   /* If there are children behind the include statement, and they have
 474    * not yet been moved because (src->children_num == 0), then move them
 475    * to the end of the list, so that the new children have room before
 476    * them. */
 477   if ((src->children_num > 0) && ((dst->children_num - (offset + 1)) > 0)) {
 478     int nmemb = dst->children_num - (offset + 1);
 479     int old_offset = offset + 1;
 480     int new_offset = offset + src->children_num;
 481
 482     memmove(dst->children + new_offset, dst->children + old_offset,
 483             sizeof(oconfig_item_t) * nmemb);
 484   }
 485
 486   /* Last but not least: If there are new children, copy them to the
 487    * memory reserved for them. */
 488   if (src->children_num > 0) {
 489     memcpy(dst->children + offset, src->children,
 490            sizeof(oconfig_item_t) * src->children_num);
 491   }
 492
 493   /* Update the number of children. */
 494   dst->children_num += (src->children_num - 1);
 495
 496   return 0;
 497 } /* int cf_ci_replace_child */
 498
 499 static int cf_ci_append_children(oconfig_item_t *dst, oconfig_item_t *src) {
 500   oconfig_item_t *temp;
 501
 502   if ((src == NULL) || (src->children_num == 0))
 503     return 0;
 504
 505   temp =
 506       realloc(dst->children,
 507               sizeof(oconfig_item_t) * (dst->children_num + src->children_num));
 508   if (temp == NULL) {
 509     ERROR("configfile: realloc failed.");
 510     return -1;
 511   }
 512   dst->children = temp;
 513
 514   memcpy(dst->children + dst->children_num, src->children,
 515          sizeof(oconfig_item_t) * src->children_num);
 516   dst->children_num += src->children_num;
 517
 518   return 0;
 519 } /* int cf_ci_append_children */
 520
 521 #define CF_MAX_DEPTH 8
 522 static oconfig_item_t *cf_read_generic(const char *path, const char *pattern,
 523                                        int depth);
 524
 525 static int cf_include_all(oconfig_item_t *root, int depth) {
 526   for (int i = 0; i < root->children_num; i++) {
 527     oconfig_item_t *new;
 528     oconfig_item_t *old;
 529
 530     char *pattern = NULL;
 531
 532     if (strcasecmp(root->children[i].key, "Include") != 0)
 533       continue;
 534
 535     old = root->children + i;
 536
 537     if ((old->values_num != 1) ||
 538         (old->values[0].type != OCONFIG_TYPE_STRING)) {
 539       ERROR("configfile: `Include' needs exactly one string argument.");
 540       continue;
 541     }
 542
 543     for (int j = 0; j < old->children_num; ++j) {
 544       oconfig_item_t *child = old->children + j;
 545
 546       if (strcasecmp(child->key, "Filter") == 0)
 547         cf_util_get_string(child, &pattern);
 548       else
 549         ERROR("configfile: Option `%s' not allowed in <Include> block.",
 550               child->key);
 551     }
 552
 553     new = cf_read_generic(old->values[0].value.string, pattern, depth + 1);
 554     sfree(pattern);
 555
 556     if (new == NULL)
 557       return -1;
 558
 559     /* Now replace the i'th child in `root' with `new'. */
 560     if (cf_ci_replace_child(root, new, i) < 0) {
 561       sfree(new->values);
 562       sfree(new);
 563       return -1;
 564     }
 565
 566     /* ... and go back to the new i'th child. */
 567     --i;
 568
 569     sfree(new->values);
 570     sfree(new);
 571   } /* for (i = 0; i < root->children_num; i++) */
 572
 573   return 0;
 574 } /* int cf_include_all */
 575
 576 static oconfig_item_t *cf_read_file(const char *file, const char *pattern,
 577                                     int depth) {
 578   oconfig_item_t *root;
 579   int status;
 580
 581   assert(depth < CF_MAX_DEPTH);
 582
 583   if (pattern != NULL) {
 584 #if HAVE_FNMATCH_H && HAVE_LIBGEN_H
 585     char *tmp = sstrdup(file);
 586     char *filename = basename(tmp);
 587
 588     if ((filename != NULL) && (fnmatch(pattern, filename, 0) != 0)) {
 589       DEBUG("configfile: Not including `%s' because it "
 590             "does not match pattern `%s'.",
 591             filename, pattern);
 592       free(tmp);
 593       return NULL;
 594     }
 595
 596     free(tmp);
 597 #else
 598     ERROR("configfile: Cannot apply pattern filter '%s' "
 599           "to file '%s': functions basename() and / or "
 600           "fnmatch() not available.",
 601           pattern, file);
 602 #endif /* HAVE_FNMATCH_H && HAVE_LIBGEN_H */
 603   }
 604
 605   root = oconfig_parse_file(file);
 606   if (root == NULL) {
 607     ERROR("configfile: Cannot read file `%s'.", file);
 608     return NULL;
 609   }
 610
 611   status = cf_include_all(root, depth);
 612   if (status != 0) {
 613     oconfig_free(root);
 614     return NULL;
 615   }
 616
 617   return root;
 618 } /* oconfig_item_t *cf_read_file */
 619
 620 static int cf_compare_string(const void *p1, const void *p2) {
 621   return strcmp(*(const char **)p1, *(const char **)p2);
 622 }
 623
 624 static oconfig_item_t *cf_read_dir(const char *dir, const char *pattern,
 625                                    int depth) {
 626   oconfig_item_t *root = NULL;
 627   DIR *dh;
 628   struct dirent *de;
 629   char **filenames = NULL;
 630   int filenames_num = 0;
 631   int status;
 632
 633   assert(depth < CF_MAX_DEPTH);
 634
 635   dh = opendir(dir);
 636   if (dh == NULL) {
 637     ERROR("configfile: opendir failed: %s", STRERRNO);
 638     return NULL;
 639   }
 640
 641   root = calloc(1, sizeof(*root));
 642   if (root == NULL) {
 643     ERROR("configfile: calloc failed.");
 644     closedir(dh);
 645     return NULL;
 646   }
 647
 648   while ((de = readdir(dh)) != NULL) {
 649     char name[1024];
 650     char **tmp;
 651
 652     if ((de->d_name[0] == '.') || (de->d_name[0] == 0))
 653       continue;
 654
 655     status = snprintf(name, sizeof(name), "%s/%s", dir, de->d_name);
 656     if ((status < 0) || ((size_t)status >= sizeof(name))) {
 657       ERROR("configfile: Not including `%s/%s' because its"
 658             " name is too long.",
 659             dir, de->d_name);
 660       closedir(dh);
 661       for (int i = 0; i < filenames_num; ++i)
 662         free(filenames[i]);
 663       free(filenames);
 664       free(root);
 665       return NULL;
 666     }
 667
 668     ++filenames_num;
 669     tmp = realloc(filenames, filenames_num * sizeof(*filenames));
 670     if (tmp == NULL) {
 671       ERROR("configfile: realloc failed.");
 672       closedir(dh);
 673       for (int i = 0; i < filenames_num - 1; ++i)
 674         free(filenames[i]);
 675       free(filenames);
 676       free(root);
 677       return NULL;
 678     }
 679     filenames = tmp;
 680
 681     filenames[filenames_num - 1] = sstrdup(name);
 682   }
 683
 684   if (filenames == NULL) {
 685     closedir(dh);
 686     return root;
 687   }
 688
 689   qsort((void *)filenames, filenames_num, sizeof(*filenames),
 690         cf_compare_string);
 691
 692   for (int i = 0; i < filenames_num; ++i) {
 693     oconfig_item_t *temp;
 694     char *name = filenames[i];
 695
 696     temp = cf_read_generic(name, pattern, depth);
 697     if (temp == NULL) {
 698       /* An error should already have been reported. */
 699       sfree(name);
 700       continue;
 701     }
 702
 703     cf_ci_append_children(root, temp);
 704     sfree(temp->children);
 705     sfree(temp);
 706
 707     free(name);
 708   }
 709
 710   closedir(dh);
 711   free(filenames);
 712   return root;
 713 } /* oconfig_item_t *cf_read_dir */
 714
 715 /*
 716  * cf_read_generic
 717  *
 718  * Path is stat'ed and either cf_read_file or cf_read_dir is called
 719  * accordingly.
 720  *
 721  * There are two versions of this function: If `wordexp' exists shell wildcards
 722  * will be expanded and the function will include all matches found. If
 723  * `wordexp' (or, more precisely, its header file) is not available the
 724  * simpler function is used which does not do any such expansion.
 725  */
 726 #if HAVE_WORDEXP_H
 727 static oconfig_item_t *cf_read_generic(const char *path, const char *pattern,
 728                                        int depth) {
 729   oconfig_item_t *root = NULL;
 730   int status;
 731   const char *path_ptr;
 732   wordexp_t we;
 733
 734   if (depth >= CF_MAX_DEPTH) {
 735     ERROR("configfile: Not including `%s' because the maximum "
 736           "nesting depth has been reached.",
 737           path);
 738     return NULL;
 739   }
 740
 741   status = wordexp(path, &we, WRDE_NOCMD);
 742   if (status != 0) {
 743     ERROR("configfile: wordexp (%s) failed.", path);
 744     return NULL;
 745   }
 746
 747   root = calloc(1, sizeof(*root));
 748   if (root == NULL) {
 749     ERROR("configfile: calloc failed.");
 750     return NULL;
 751   }
 752
 753   /* wordexp() might return a sorted list already. That's not
 754    * documented though, so let's make sure we get what we want. */
 755   qsort((void *)we.we_wordv, we.we_wordc, sizeof(*we.we_wordv),
 756         cf_compare_string);
 757
 758   for (size_t i = 0; i < we.we_wordc; i++) {
 759     oconfig_item_t *temp;
 760     struct stat statbuf;
 761
 762     path_ptr = we.we_wordv[i];
 763
 764     status = stat(path_ptr, &statbuf);
 765     if (status != 0) {
 766       WARNING("configfile: stat (%s) failed: %s", path_ptr, STRERRNO);
 767       continue;
 768     }
 769
 770     if (S_ISREG(statbuf.st_mode))
 771       temp = cf_read_file(path_ptr, pattern, depth);
 772     else if (S_ISDIR(statbuf.st_mode))
 773       temp = cf_read_dir(path_ptr, pattern, depth);
 774     else {
 775       WARNING("configfile: %s is neither a file nor a "
 776               "directory.",
 777               path);
 778       continue;
 779     }
 780
 781     if (temp == NULL) {
 782       oconfig_free(root);
 783       return NULL;
 784     }
 785
 786     cf_ci_append_children(root, temp);
 787     sfree(temp->children);
 788     sfree(temp);
 789   }
 790
 791   wordfree(&we);
 792
 793   return root;
 794 } /* oconfig_item_t *cf_read_generic */
 795 /* #endif HAVE_WORDEXP_H */
 796
 797 #else  /* if !HAVE_WORDEXP_H */
 798 static oconfig_item_t *cf_read_generic(const char *path, const char *pattern,
 799                                        int depth) {
 800   struct stat statbuf;
 801   int status;
 802
 803   if (depth >= CF_MAX_DEPTH) {
 804     ERROR("configfile: Not including `%s' because the maximum "
 805           "nesting depth has been reached.",
 806           path);
 807     return NULL;
 808   }
 809
 810   status = stat(path, &statbuf);
 811   if (status != 0) {
 812     ERROR("configfile: stat (%s) failed: %s", path, STRERRNO);
 813     return NULL;
 814   }
 815
 816   if (S_ISREG(statbuf.st_mode))
 817     return cf_read_file(path, pattern, depth);
 818   else if (S_ISDIR(statbuf.st_mode))
 819     return cf_read_dir(path, pattern, depth);
 820
 821   ERROR("configfile: %s is neither a file nor a directory.", path);
 822   return NULL;
 823 } /* oconfig_item_t *cf_read_generic */
 824 #endif /* !HAVE_WORDEXP_H */
 825
 826 /*
 827  * Public functions
 828  */
 829 int global_option_set(const char *option, const char *value, bool from_cli) {
 830   int i;
 831   DEBUG("option = %s; value = %s;", option, value);
 832
 833   for (i = 0; i < cf_global_options_num; i++)
 834     if (strcasecmp(cf_global_options[i].key, option) == 0)
 835       break;
 836
 837   if (i >= cf_global_options_num) {
 838     ERROR("configfile: Cannot set unknown global option `%s'.", option);
 839     return -1;
 840   }
 841
 842   if (cf_global_options[i].from_cli && (!from_cli)) {
 843     DEBUG("configfile: Ignoring %s `%s' option because "
 844           "it was overriden by a command-line option.",
 845           option, value);
 846     return 0;
 847   }
 848
 849   sfree(cf_global_options[i].value);
 850
 851   if (value != NULL)
 852     cf_global_options[i].value = strdup(value);
 853   else
 854     cf_global_options[i].value = NULL;
 855
 856   cf_global_options[i].from_cli = from_cli;
 857
 858   return 0;
 859 }
 860
 861 const char *global_option_get(const char *option) {
 862   int i;
 863   for (i = 0; i < cf_global_options_num; i++)
 864     if (strcasecmp(cf_global_options[i].key, option) == 0)
 865       break;
 866
 867   if (i >= cf_global_options_num) {
 868     ERROR("configfile: Cannot get unknown global option `%s'.", option);
 869     return NULL;
 870   }
 871
 872   return (cf_global_options[i].value != NULL) ? cf_global_options[i].value
 873                                               : cf_global_options[i].def;
 874 } /* char *global_option_get */
 875
 876 long global_option_get_long(const char *option, long default_value) {
 877   const char *str;
 878   long value;
 879
 880   str = global_option_get(option);
 881   if (NULL == str)
 882     return default_value;
 883
 884   errno = 0;
 885   value = strtol(str, /* endptr = */ NULL, /* base = */ 0);
 886   if (errno != 0)
 887     return default_value;
 888
 889   return value;
 890 } /* char *global_option_get_long */
 891
 892 cdtime_t global_option_get_time(const char *name, cdtime_t def) /* {{{ */
 893 {
 894   char const *optstr;
 895   char *endptr = NULL;
 896   double v;
 897
 898   optstr = global_option_get(name);
 899   if (optstr == NULL)
 900     return def;
 901
 902   errno = 0;
 903   v = strtod(optstr, &endptr);
 904   if ((endptr == NULL) || (*endptr != 0) || (errno != 0))
 905     return def;
 906   else if (v <= 0.0)
 907     return def;
 908
 909   return DOUBLE_TO_CDTIME_T(v);
 910 } /* }}} cdtime_t global_option_get_time */
 911
 912 cdtime_t cf_get_default_interval(void) {
 913   return global_option_get_time("Interval",
 914                                 DOUBLE_TO_CDTIME_T(COLLECTD_DEFAULT_INTERVAL));
 915 }
 916
 917 void cf_unregister(const char *type) {
 918   for (cf_callback_t *prev = NULL, *this = first_callback; this != NULL;
 919        prev = this, this = this->next)
 920     if (strcasecmp(this->type, type) == 0) {
 921       if (prev == NULL)
 922         first_callback = this->next;
 923       else
 924         prev->next = this->next;
 925
 926       free(this);
 927       break;
 928     }
 929 } /* void cf_unregister */
 930
 931 void cf_unregister_complex(const char *type) {
 932   for (cf_complex_callback_t *prev = NULL, *this = complex_callback_head;
 933        this != NULL; prev = this, this = this->next)
 934     if (strcasecmp(this->type, type) == 0) {
 935       if (prev == NULL)
 936         complex_callback_head = this->next;
 937       else
 938         prev->next = this->next;
 939
 940       sfree(this->type);
 941       sfree(this);
 942       break;
 943     }
 944 } /* void cf_unregister */
 945
 946 void cf_register(const char *type, int (*callback)(const char *, const char *),
 947                  const char **keys, int keys_num) {
 948   cf_callback_t *cf_cb;
 949
 950   /* Remove this module from the list, if it already exists */
 951   cf_unregister(type);
 952
 953   /* This pointer will be free'd in `cf_unregister' */
 954   if ((cf_cb = malloc(sizeof(*cf_cb))) == NULL)
 955     return;
 956
 957   cf_cb->type = type;
 958   cf_cb->callback = callback;
 959   cf_cb->keys = keys;
 960   cf_cb->keys_num = keys_num;
 961   cf_cb->ctx = plugin_get_ctx();
 962
 963   cf_cb->next = first_callback;
 964   first_callback = cf_cb;
 965 } /* void cf_register */
 966
 967 int cf_register_complex(const char *type, int (*callback)(oconfig_item_t *)) {
 968   cf_complex_callback_t *new;
 969
 970   new = malloc(sizeof(*new));
 971   if (new == NULL)
 972     return -1;
 973
 974   new->type = strdup(type);
 975   if (new->type == NULL) {
 976     sfree(new);
 977     return -1;
 978   }
 979
 980   new->callback = callback;
 981   new->next = NULL;
 982
 983   new->ctx = plugin_get_ctx();
 984
 985   if (complex_callback_head == NULL) {
 986     complex_callback_head = new;
 987   } else {
 988     cf_complex_callback_t *last = complex_callback_head;
 989     while (last->next != NULL)
 990       last = last->next;
 991     last->next = new;
 992   }
 993
 994   return 0;
 995 } /* int cf_register_complex */
 996
 997 int cf_read(const char *filename) {
 998   oconfig_item_t *conf;
 999   int ret = 0;
1000
1001   conf = cf_read_generic(filename, /* pattern = */ NULL, /* depth = */ 0);
1002   if (conf == NULL) {
1003     ERROR("Unable to read config file %s.", filename);
1004     return -1;
1005   } else if (conf->children_num == 0) {
1006     ERROR("Configuration file %s is empty.", filename);
1007     oconfig_free(conf);
1008     return -1;
1009   }
1010
1011   for (int i = 0; i < conf->children_num; i++) {
1012     if (conf->children[i].children == NULL) {
1013       if (dispatch_value(conf->children + i) != 0)
1014         ret = -1;
1015     } else {
1016       if (dispatch_block(conf->children + i) != 0)
1017         ret = -1;
1018     }
1019   }
1020
1021   oconfig_free(conf);
1022
1023   /* Read the default types.db if no `TypesDB' option was given. */
1024   if (cf_default_typesdb) {
1025     if (read_types_list(PKGDATADIR "/types.db") != 0)
1026       ret = -1;
1027   }
1028
1029   return ret;
1030
1031 } /* int cf_read */
1032
1033 /* Assures the config option is a string, duplicates it and returns the copy in
1034  * "ret_string". If necessary "*ret_string" is freed first. Returns zero upon
1035  * success. */
1036 int cf_util_get_string(const oconfig_item_t *ci, char **ret_string) /* {{{ */
1037 {
1038   char *string;
1039
1040   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING)) {
1041     ERROR("cf_util_get_string: The %s option requires "
1042           "exactly one string argument.",
1043           ci->key);
1044     return -1;
1045   }
1046
1047   string = strdup(ci->values[0].value.string);
1048   if (string == NULL)
1049     return -1;
1050
1051   if (*ret_string != NULL)
1052     sfree(*ret_string);
1053   *ret_string = string;
1054
1055   return 0;
1056 } /* }}} int cf_util_get_string */
1057
1058 /* Assures the config option is a string and copies it to the provided buffer.
1059  * Assures null-termination. */
1060 int cf_util_get_string_buffer(const oconfig_item_t *ci, char *buffer, /* {{{ */
1061                               size_t buffer_size) {
1062   if ((ci == NULL) || (buffer == NULL) || (buffer_size < 1))
1063     return EINVAL;
1064
1065   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING)) {
1066     ERROR("cf_util_get_string_buffer: The %s option requires "
1067           "exactly one string argument.",
1068           ci->key);
1069     return -1;
1070   }
1071
1072   strncpy(buffer, ci->values[0].value.string, buffer_size);
1073   buffer[buffer_size - 1] = 0;
1074
1075   return 0;
1076 } /* }}} int cf_util_get_string_buffer */
1077
1078 /* Assures the config option is a number and returns it as an int. */
1079 int cf_util_get_int(const oconfig_item_t *ci, int *ret_value) /* {{{ */
1080 {
1081   if ((ci == NULL) || (ret_value == NULL))
1082     return EINVAL;
1083
1084   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_NUMBER)) {
1085     ERROR("cf_util_get_int: The %s option requires "
1086           "exactly one numeric argument.",
1087           ci->key);
1088     return -1;
1089   }
1090
1091   *ret_value = (int)ci->values[0].value.number;
1092
1093   return 0;
1094 } /* }}} int cf_util_get_int */
1095
1096 int cf_util_get_double(const oconfig_item_t *ci, double *ret_value) /* {{{ */
1097 {
1098   if ((ci == NULL) || (ret_value == NULL))
1099     return EINVAL;
1100
1101   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_NUMBER)) {
1102     ERROR("cf_util_get_double: The %s option requires "
1103           "exactly one numeric argument.",
1104           ci->key);
1105     return -1;
1106   }
1107
1108   *ret_value = ci->values[0].value.number;
1109
1110   return 0;
1111 } /* }}} int cf_util_get_double */
1112
1113 int cf_util_get_boolean(const oconfig_item_t *ci, bool *ret_bool) /* {{{ */
1114 {
1115   if ((ci == NULL) || (ret_bool == NULL))
1116     return EINVAL;
1117
1118   if ((ci->values_num != 1) || ((ci->values[0].type != OCONFIG_TYPE_BOOLEAN) &&
1119                                 (ci->values[0].type != OCONFIG_TYPE_STRING))) {
1120     ERROR("cf_util_get_boolean: The %s option requires "
1121           "exactly one boolean argument.",
1122           ci->key);
1123     return -1;
1124   }
1125
1126   switch (ci->values[0].type) {
1127   case OCONFIG_TYPE_BOOLEAN:
1128     *ret_bool = ci->values[0].value.boolean ? true : false;
1129     break;
1130   case OCONFIG_TYPE_STRING:
1131     WARNING("cf_util_get_boolean: Using string value `%s' for boolean option "
1132             "`%s' is deprecated and will be removed in future releases. "
1133             "Use unquoted true or false instead.",
1134             ci->values[0].value.string, ci->key);
1135
1136     if (IS_TRUE(ci->values[0].value.string))
1137       *ret_bool = true;
1138     else if (IS_FALSE(ci->values[0].value.string))
1139       *ret_bool = false;
1140     else {
1141       ERROR("cf_util_get_boolean: Cannot parse string value `%s' of the `%s' "
1142             "option as a boolean value.",
1143             ci->values[0].value.string, ci->key);
1144       return -1;
1145     }
1146     break;
1147   }
1148
1149   return 0;
1150 } /* }}} int cf_util_get_boolean */
1151
1152 int cf_util_get_flag(const oconfig_item_t *ci, /* {{{ */
1153                      unsigned int *ret_value, unsigned int flag) {
1154   int status;
1155
1156   if (ret_value == NULL)
1157     return EINVAL;
1158
1159   bool b = false;
1160   status = cf_util_get_boolean(ci, &b);
1161   if (status != 0)
1162     return status;
1163
1164   if (b) {
1165     *ret_value |= flag;
1166   } else {
1167     *ret_value &= ~flag;
1168   }
1169
1170   return 0;
1171 } /* }}} int cf_util_get_flag */
1172
1173 /* Assures that the config option is a string or a number if the correct range
1174  * of 1-65535. The string is then converted to a port number using
1175  * `service_name_to_port_number' and returned.
1176  * Returns the port number in the range [1-65535] or less than zero upon
1177  * failure. */
1178 int cf_util_get_port_number(const oconfig_item_t *ci) /* {{{ */
1179 {
1180   int tmp;
1181
1182   if ((ci->values_num != 1) || ((ci->values[0].type != OCONFIG_TYPE_STRING) &&
1183                                 (ci->values[0].type != OCONFIG_TYPE_NUMBER))) {
1184     ERROR("cf_util_get_port_number: The \"%s\" option requires "
1185           "exactly one string argument.",
1186           ci->key);
1187     return -1;
1188   }
1189
1190   if (ci->values[0].type == OCONFIG_TYPE_STRING)
1191     return service_name_to_port_number(ci->values[0].value.string);
1192
1193   assert(ci->values[0].type == OCONFIG_TYPE_NUMBER);
1194   tmp = (int)(ci->values[0].value.number + 0.5);
1195   if ((tmp < 1) || (tmp > 65535)) {
1196     ERROR("cf_util_get_port_number: The \"%s\" option requires "
1197           "a service name or a port number. The number "
1198           "you specified, %i, is not in the valid "
1199           "range of 1-65535.",
1200           ci->key, tmp);
1201     return -1;
1202   }
1203
1204   return tmp;
1205 } /* }}} int cf_util_get_port_number */
1206
1207 int cf_util_get_service(const oconfig_item_t *ci, char **ret_string) /* {{{ */
1208 {
1209   int port;
1210   char *service;
1211   int status;
1212
1213   if (ci->values_num != 1) {
1214     ERROR("cf_util_get_service: The %s option requires exactly "
1215           "one argument.",
1216           ci->key);
1217     return -1;
1218   }
1219
1220   if (ci->values[0].type == OCONFIG_TYPE_STRING)
1221     return cf_util_get_string(ci, ret_string);
1222   if (ci->values[0].type != OCONFIG_TYPE_NUMBER) {
1223     ERROR("cf_util_get_service: The %s option requires "
1224           "exactly one string or numeric argument.",
1225           ci->key);
1226   }
1227
1228   port = 0;
1229   status = cf_util_get_int(ci, &port);
1230   if (status != 0)
1231     return status;
1232   else if ((port < 1) || (port > 65535)) {
1233     ERROR("cf_util_get_service: The port number given "
1234           "for the %s option is out of "
1235           "range (%i).",
1236           ci->key, port);
1237     return -1;
1238   }
1239
1240   service = malloc(6);
1241   if (service == NULL) {
1242     ERROR("cf_util_get_service: Out of memory.");
1243     return -1;
1244   }
1245   snprintf(service, 6, "%i", port);
1246
1247   sfree(*ret_string);
1248   *ret_string = service;
1249
1250   return 0;
1251 } /* }}} int cf_util_get_service */
1252
1253 int cf_util_get_cdtime(const oconfig_item_t *ci, cdtime_t *ret_value) /* {{{ */
1254 {
1255   if ((ci == NULL) || (ret_value == NULL))
1256     return EINVAL;
1257
1258   if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_NUMBER)) {
1259     ERROR("cf_util_get_cdtime: The %s option requires "
1260           "exactly one numeric argument.",
1261           ci->key);
1262     return -1;
1263   }
1264
1265   if (ci->values[0].value.number < 0.0) {
1266     ERROR("cf_util_get_cdtime: The numeric argument of the %s "
1267           "option must not be negative.",
1268           ci->key);
1269     return -1;
1270   }
1271
1272   *ret_value = DOUBLE_TO_CDTIME_T(ci->values[0].value.number);
1273
1274   return 0;
1275 } /* }}} int cf_util_get_cdtime */