1 /**
2 * collectd - src/configfile.c
3 * Copyright (C) 2005-2011 Florian octo Forster
4 *
5 * This program is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License as published by the
7 * Free Software Foundation; either version 2 of the License, or (at your
8 * option) any later version.
9 *
10 * This program is distributed in the hope that it will be useful, but
11 * WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 * General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License along
16 * with this program; if not, write to the Free Software Foundation, Inc.,
17 * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
18 *
19 * Authors:
20 * Florian octo Forster <octo at collectd.org>
21 * Sebastian tokkee Harl <sh at tokkee.org>
22 **/
24 #include "collectd.h"
26 #include "liboconfig/oconfig.h"
28 #include "common.h"
29 #include "plugin.h"
30 #include "configfile.h"
31 #include "types_list.h"
32 #include "filter_chain.h"
34 #if HAVE_WORDEXP_H
35 # include <wordexp.h>
36 #endif /* HAVE_WORDEXP_H */
38 #if HAVE_FNMATCH_H
39 # include <fnmatch.h>
40 #endif /* HAVE_FNMATCH_H */
42 #if HAVE_LIBGEN_H
43 # include <libgen.h>
44 #endif /* HAVE_LIBGEN_H */
46 #define ESCAPE_NULL(str) ((str) == NULL ? "(null)" : (str))
48 /*
49 * Private types
50 */
51 typedef struct cf_callback
52 {
53 const char *type;
54 int (*callback) (const char *, const char *);
55 const char **keys;
56 int keys_num;
57 plugin_ctx_t ctx;
58 struct cf_callback *next;
59 } cf_callback_t;
61 typedef struct cf_complex_callback_s
62 {
63 char *type;
64 int (*callback) (oconfig_item_t *);
65 plugin_ctx_t ctx;
66 struct cf_complex_callback_s *next;
67 } cf_complex_callback_t;
69 typedef struct cf_value_map_s
70 {
71 char *key;
72 int (*func) (const oconfig_item_t *);
73 } cf_value_map_t;
75 typedef struct cf_global_option_s
76 {
77 char *key;
78 char *value;
79 char *def;
80 } cf_global_option_t;
82 /*
83 * Prototypes of callback functions
84 */
85 static int dispatch_value_typesdb (const oconfig_item_t *ci);
86 static int dispatch_value_plugindir (const oconfig_item_t *ci);
87 static int dispatch_loadplugin (const oconfig_item_t *ci);
89 /*
90 * Private variables
91 */
92 static cf_callback_t *first_callback = NULL;
93 static cf_complex_callback_t *complex_callback_head = NULL;
95 static cf_value_map_t cf_value_map[] =
96 {
97 {"TypesDB", dispatch_value_typesdb},
98 {"PluginDir", dispatch_value_plugindir},
99 {"LoadPlugin", dispatch_loadplugin}
100 };
101 static int cf_value_map_num = STATIC_ARRAY_SIZE (cf_value_map);
103 static cf_global_option_t cf_global_options[] =
104 {
105 {"BaseDir", NULL, PKGLOCALSTATEDIR},
106 {"PIDFile", NULL, PIDFILE},
107 {"Hostname", NULL, NULL},
108 {"FQDNLookup", NULL, "true"},
109 {"Interval", NULL, NULL},
110 {"ReadThreads", NULL, "5"},
111 {"WriteThreads", NULL, "5"},
112 {"WriteQueueLimitHigh", NULL, NULL},
113 {"WriteQueueLimitLow", NULL, NULL},
114 {"Timeout", NULL, "2"},
115 {"AutoLoadPlugin", NULL, "false"},
116 {"PreCacheChain", NULL, "PreCache"},
117 {"PostCacheChain", NULL, "PostCache"}
118 };
119 static int cf_global_options_num = STATIC_ARRAY_SIZE (cf_global_options);
121 static int cf_default_typesdb = 1;
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 {
129 cf_callback_t *cf_cb;
131 if (type == NULL)
132 return (NULL);
134 for (cf_cb = first_callback; cf_cb != NULL; cf_cb = cf_cb->next)
135 if (strcasecmp (cf_cb->type, type) == 0)
136 break;
138 return (cf_cb);
139 }
141 static int cf_dispatch (const char *type, const char *orig_key,
142 const char *orig_value)
143 {
144 cf_callback_t *cf_cb;
145 plugin_ctx_t old_ctx;
146 char *key;
147 char *value;
148 int ret;
149 int i;
151 DEBUG ("type = %s, key = %s, value = %s",
152 ESCAPE_NULL(type),
153 ESCAPE_NULL(orig_key),
154 ESCAPE_NULL(orig_value));
156 if ((cf_cb = cf_search (type)) == NULL)
157 {
158 WARNING ("Found a configuration for the `%s' plugin, but "
159 "the plugin isn't loaded or didn't register "
160 "a configuration callback.", type);
161 return (-1);
162 }
164 if ((key = strdup (orig_key)) == NULL)
165 return (1);
166 if ((value = strdup (orig_value)) == NULL)
167 {
168 free (key);
169 return (2);
170 }
172 ret = -1;
174 old_ctx = plugin_set_ctx (cf_cb->ctx);
176 for (i = 0; i < cf_cb->keys_num; i++)
177 {
178 if ((cf_cb->keys[i] != NULL)
179 && (strcasecmp (cf_cb->keys[i], key) == 0))
180 {
181 ret = (*cf_cb->callback) (key, value);
182 break;
183 }
184 }
186 plugin_set_ctx (old_ctx);
188 if (i >= cf_cb->keys_num)
189 WARNING ("Plugin `%s' did not register for value `%s'.", type, key);
191 free (key);
192 free (value);
194 DEBUG ("cf_dispatch: return (%i)", ret);
196 return (ret);
197 } /* int cf_dispatch */
199 static int dispatch_global_option (const oconfig_item_t *ci)
200 {
201 if (ci->values_num != 1)
202 return (-1);
203 if (ci->values[0].type == OCONFIG_TYPE_STRING)
204 return (global_option_set (ci->key, ci->values[0].value.string));
205 else if (ci->values[0].type == OCONFIG_TYPE_NUMBER)
206 {
207 char tmp[128];
208 ssnprintf (tmp, sizeof (tmp), "%lf", ci->values[0].value.number);
209 return (global_option_set (ci->key, tmp));
210 }
211 else if (ci->values[0].type == OCONFIG_TYPE_BOOLEAN)
212 {
213 if (ci->values[0].value.boolean)
214 return (global_option_set (ci->key, "true"));
215 else
216 return (global_option_set (ci->key, "false"));
217 }
219 return (-1);
220 } /* int dispatch_global_option */
222 static int dispatch_value_typesdb (const oconfig_item_t *ci)
223 {
224 int i = 0;
226 assert (strcasecmp (ci->key, "TypesDB") == 0);
228 cf_default_typesdb = 0;
230 if (ci->values_num < 1) {
231 ERROR ("configfile: `TypesDB' needs at least one argument.");
232 return (-1);
233 }
235 for (i = 0; i < ci->values_num; ++i)
236 {
237 if (OCONFIG_TYPE_STRING != ci->values[i].type) {
238 WARNING ("configfile: TypesDB: Skipping %i. argument which "
239 "is not a string.", i + 1);
240 continue;
241 }
243 read_types_list (ci->values[i].value.string);
244 }
245 return (0);
246 } /* int dispatch_value_typesdb */
248 static int dispatch_value_plugindir (const oconfig_item_t *ci)
249 {
250 assert (strcasecmp (ci->key, "PluginDir") == 0);
252 if (ci->values_num != 1)
253 return (-1);
254 if (ci->values[0].type != OCONFIG_TYPE_STRING)
255 return (-1);
257 plugin_set_dir (ci->values[0].value.string);
258 return (0);
259 }
261 static int dispatch_loadplugin (const oconfig_item_t *ci)
262 {
263 int i;
264 const char *name;
265 unsigned int flags = 0;
266 plugin_ctx_t ctx;
267 plugin_ctx_t old_ctx;
268 int ret_val;
270 assert (strcasecmp (ci->key, "LoadPlugin") == 0);
272 if (ci->values_num != 1)
273 return (-1);
274 if (ci->values[0].type != OCONFIG_TYPE_STRING)
275 return (-1);
277 name = ci->values[0].value.string;
279 /* default to the global interval set before loading this plugin */
280 memset (&ctx, 0, sizeof (ctx));
281 ctx.interval = cf_get_default_interval ();
283 for (i = 0; i < ci->children_num; ++i) {
284 if (strcasecmp("Globals", ci->children[i].key) == 0)
285 cf_util_get_flag (ci->children + i, &flags, PLUGIN_FLAGS_GLOBAL);
286 else if (strcasecmp ("Interval", ci->children[i].key) == 0) {
287 double interval = 0.0;
289 if (cf_util_get_double (ci->children + i, &interval) != 0) {
290 /* cf_util_get_double will log an error */
291 continue;
292 }
294 ctx.interval = DOUBLE_TO_CDTIME_T (interval);
295 }
296 else {
297 WARNING("Ignoring unknown LoadPlugin option \"%s\" "
298 "for plugin \"%s\"",
299 ci->children[i].key, ci->values[0].value.string);
300 }
301 }
303 old_ctx = plugin_set_ctx (ctx);
304 ret_val = plugin_load (name, (uint32_t) flags);
305 /* reset to the "global" context */
306 plugin_set_ctx (old_ctx);
308 return (ret_val);
309 } /* int dispatch_value_loadplugin */
311 static int dispatch_value_plugin (const char *plugin, oconfig_item_t *ci)
312 {
313 char buffer[4096];
314 char *buffer_ptr;
315 int buffer_free;
316 int i;
318 buffer_ptr = buffer;
319 buffer_free = sizeof (buffer);
321 for (i = 0; i < ci->values_num; i++)
322 {
323 int status = -1;
325 if (ci->values[i].type == OCONFIG_TYPE_STRING)
326 status = ssnprintf (buffer_ptr, buffer_free, " %s",
327 ci->values[i].value.string);
328 else if (ci->values[i].type == OCONFIG_TYPE_NUMBER)
329 status = ssnprintf (buffer_ptr, buffer_free, " %lf",
330 ci->values[i].value.number);
331 else if (ci->values[i].type == OCONFIG_TYPE_BOOLEAN)
332 status = ssnprintf (buffer_ptr, buffer_free, " %s",
333 ci->values[i].value.boolean
334 ? "true" : "false");
336 if ((status < 0) || (status >= buffer_free))
337 return (-1);
338 buffer_free -= status;
339 buffer_ptr += status;
340 }
341 /* skip the initial space */
342 buffer_ptr = buffer + 1;
344 return (cf_dispatch (plugin, ci->key, buffer_ptr));
345 } /* int dispatch_value_plugin */
347 static int dispatch_value (const oconfig_item_t *ci)
348 {
349 int ret = -2;
350 int i;
352 for (i = 0; i < cf_value_map_num; i++)
353 if (strcasecmp (cf_value_map[i].key, ci->key) == 0)
354 {
355 ret = cf_value_map[i].func (ci);
356 break;
357 }
359 for (i = 0; i < cf_global_options_num; i++)
360 if (strcasecmp (cf_global_options[i].key, ci->key) == 0)
361 {
362 ret = dispatch_global_option (ci);
363 break;
364 }
366 return (ret);
367 } /* int dispatch_value */
369 static int dispatch_block_plugin (oconfig_item_t *ci)
370 {
371 int i;
372 char *name;
374 cf_complex_callback_t *cb;
376 if (strcasecmp (ci->key, "Plugin") != 0)
377 return (-1);
378 if (ci->values_num < 1)
379 return (-1);
380 if (ci->values[0].type != OCONFIG_TYPE_STRING)
381 return (-1);
383 name = ci->values[0].value.string;
385 if (IS_TRUE (global_option_get ("AutoLoadPlugin")))
386 {
387 int status;
389 status = plugin_load (name, /* flags = */ 0);
390 if (status != 0)
391 {
392 ERROR ("Automatically loading plugin \"%s\" failed "
393 "with status %i.", name, status);
394 return (status);
395 }
396 }
398 /* Check for a complex callback first */
399 for (cb = complex_callback_head; cb != NULL; cb = cb->next)
400 {
401 if (strcasecmp (name, cb->type) == 0)
402 {
403 plugin_ctx_t old_ctx;
404 int ret_val;
406 old_ctx = plugin_set_ctx (cb->ctx);
407 ret_val = (cb->callback (ci));
408 plugin_set_ctx (old_ctx);
409 return (ret_val);
410 }
411 }
413 /* Hm, no complex plugin found. Dispatch the values one by one */
414 for (i = 0; i < ci->children_num; i++)
415 {
416 if (ci->children[i].children == NULL)
417 dispatch_value_plugin (name, ci->children + i);
418 else
419 {
420 WARNING ("There is a `%s' block within the "
421 "configuration for the %s plugin. "
422 "The plugin either only expects "
423 "\"simple\" configuration statements "
424 "or wasn't loaded using `LoadPlugin'."
425 " Please check your configuration.",
426 ci->children[i].key, name);
427 }
428 }
430 return (0);
431 }
434 static int dispatch_block (oconfig_item_t *ci)
435 {
436 if (strcasecmp (ci->key, "LoadPlugin") == 0)
437 return (dispatch_loadplugin (ci));
438 else if (strcasecmp (ci->key, "Plugin") == 0)
439 return (dispatch_block_plugin (ci));
440 else if (strcasecmp (ci->key, "Chain") == 0)
441 return (fc_configure (ci));
443 return (0);
444 }
446 static int cf_ci_replace_child (oconfig_item_t *dst, oconfig_item_t *src,
447 int offset)
448 {
449 oconfig_item_t *temp;
450 int i;
452 assert (offset >= 0);
453 assert (dst->children_num > offset);
455 /* Free the memory used by the replaced child. Usually that's the
456 * `Include "blah"' statement. */
457 temp = dst->children + offset;
458 for (i = 0; i < temp->values_num; i++)
459 {
460 if (temp->values[i].type == OCONFIG_TYPE_STRING)
461 {
462 sfree (temp->values[i].value.string);
463 }
464 }
465 sfree (temp->values);
466 temp = NULL;
468 /* If (src->children_num == 0) the array size is decreased. If offset
469 * is _not_ the last element, (offset < (dst->children_num - 1)), then
470 * we need to move the trailing elements before resizing the array. */
471 if ((src->children_num == 0) && (offset < (dst->children_num - 1)))
472 {
473 int nmemb = dst->children_num - (offset + 1);
474 memmove (dst->children + offset, dst->children + offset + 1,
475 sizeof (oconfig_item_t) * nmemb);
476 }
478 /* Resize the memory containing the children to be big enough to hold
479 * all children. */
480 temp = (oconfig_item_t *) realloc (dst->children,
481 sizeof (oconfig_item_t)
482 * (dst->children_num + src->children_num - 1));
483 if (temp == NULL)
484 {
485 ERROR ("configfile: realloc failed.");
486 return (-1);
487 }
488 dst->children = temp;
490 /* If there are children behind the include statement, and they have
491 * not yet been moved because (src->children_num == 0), then move them
492 * to the end of the list, so that the new children have room before
493 * them. */
494 if ((src->children_num > 0)
495 && ((dst->children_num - (offset + 1)) > 0))
496 {
497 int nmemb = dst->children_num - (offset + 1);
498 int old_offset = offset + 1;
499 int new_offset = offset + src->children_num;
501 memmove (dst->children + new_offset,
502 dst->children + old_offset,
503 sizeof (oconfig_item_t) * nmemb);
504 }
506 /* Last but not least: If there are new children, copy them to the
507 * memory reserved for them. */
508 if (src->children_num > 0)
509 {
510 memcpy (dst->children + offset,
511 src->children,
512 sizeof (oconfig_item_t) * src->children_num);
513 }
515 /* Update the number of children. */
516 dst->children_num += (src->children_num - 1);
518 return (0);
519 } /* int cf_ci_replace_child */
521 static int cf_ci_append_children (oconfig_item_t *dst, oconfig_item_t *src)
522 {
523 oconfig_item_t *temp;
525 if ((src == NULL) || (src->children_num == 0))
526 return (0);
528 temp = (oconfig_item_t *) realloc (dst->children,
529 sizeof (oconfig_item_t)
530 * (dst->children_num + src->children_num));
531 if (temp == NULL)
532 {
533 ERROR ("configfile: realloc failed.");
534 return (-1);
535 }
536 dst->children = temp;
538 memcpy (dst->children + dst->children_num,
539 src->children,
540 sizeof (oconfig_item_t)
541 * src->children_num);
542 dst->children_num += src->children_num;
544 return (0);
545 } /* int cf_ci_append_children */
547 #define CF_MAX_DEPTH 8
548 static oconfig_item_t *cf_read_generic (const char *path,
549 const char *pattern, int depth);
551 static int cf_include_all (oconfig_item_t *root, int depth)
552 {
553 int i;
555 for (i = 0; i < root->children_num; i++)
556 {
557 oconfig_item_t *new;
558 oconfig_item_t *old;
560 char *pattern = NULL;
562 int j;
564 if (strcasecmp (root->children[i].key, "Include") != 0)
565 continue;
567 old = root->children + i;
569 if ((old->values_num != 1)
570 || (old->values[0].type != OCONFIG_TYPE_STRING))
571 {
572 ERROR ("configfile: `Include' needs exactly one string argument.");
573 continue;
574 }
576 for (j = 0; j < old->children_num; ++j)
577 {
578 oconfig_item_t *child = old->children + j;
580 if (strcasecmp (child->key, "Filter") == 0)
581 cf_util_get_string (child, &pattern);
582 else
583 ERROR ("configfile: Option `%s' not allowed in <Include> block.",
584 child->key);
585 }
587 new = cf_read_generic (old->values[0].value.string, pattern, depth + 1);
588 sfree (pattern);
590 if (new == NULL)
591 return (-1);
593 /* Now replace the i'th child in `root' with `new'. */
594 cf_ci_replace_child (root, new, i);
596 /* ... and go back to the new i'th child. */
597 --i;
599 sfree (new->values);
600 sfree (new);
601 } /* for (i = 0; i < root->children_num; i++) */
603 return (0);
604 } /* int cf_include_all */
606 static oconfig_item_t *cf_read_file (const char *file,
607 const char *pattern, int depth)
608 {
609 oconfig_item_t *root;
610 int status;
612 assert (depth < CF_MAX_DEPTH);
614 if (pattern != NULL) {
615 #if HAVE_FNMATCH_H && HAVE_LIBGEN_H
616 char *tmp = sstrdup (file);
617 char *filename = basename (tmp);
619 if ((filename != NULL) && (fnmatch (pattern, filename, 0) != 0)) {
620 DEBUG ("configfile: Not including `%s' because it "
621 "does not match pattern `%s'.",
622 filename, pattern);
623 free (tmp);
624 return (NULL);
625 }
627 free (tmp);
628 #else
629 ERROR ("configfile: Cannot apply pattern filter '%s' "
630 "to file '%s': functions basename() and / or "
631 "fnmatch() not available.", pattern, file);
632 #endif /* HAVE_FNMATCH_H && HAVE_LIBGEN_H */
633 }
635 root = oconfig_parse_file (file);
636 if (root == NULL)
637 {
638 ERROR ("configfile: Cannot read file `%s'.", file);
639 return (NULL);
640 }
642 status = cf_include_all (root, depth);
643 if (status != 0)
644 {
645 oconfig_free (root);
646 return (NULL);
647 }
649 return (root);
650 } /* oconfig_item_t *cf_read_file */
652 static int cf_compare_string (const void *p1, const void *p2)
653 {
654 return strcmp (*(const char **) p1, *(const char **) p2);
655 }
657 static oconfig_item_t *cf_read_dir (const char *dir,
658 const char *pattern, int depth)
659 {
660 oconfig_item_t *root = NULL;
661 DIR *dh;
662 struct dirent *de;
663 char **filenames = NULL;
664 int filenames_num = 0;
665 int status;
666 int i;
668 assert (depth < CF_MAX_DEPTH);
670 dh = opendir (dir);
671 if (dh == NULL)
672 {
673 char errbuf[1024];
674 ERROR ("configfile: opendir failed: %s",
675 sstrerror (errno, errbuf, sizeof (errbuf)));
676 return (NULL);
677 }
679 root = (oconfig_item_t *) malloc (sizeof (oconfig_item_t));
680 if (root == NULL)
681 {
682 ERROR ("configfile: malloc failed.");
683 return (NULL);
684 }
685 memset (root, 0, sizeof (oconfig_item_t));
687 while ((de = readdir (dh)) != NULL)
688 {
689 char name[1024];
690 char **tmp;
692 if ((de->d_name[0] == '.') || (de->d_name[0] == 0))
693 continue;
695 status = ssnprintf (name, sizeof (name), "%s/%s",
696 dir, de->d_name);
697 if ((status < 0) || ((size_t) status >= sizeof (name)))
698 {
699 ERROR ("configfile: Not including `%s/%s' because its"
700 " name is too long.",
701 dir, de->d_name);
702 for (i = 0; i < filenames_num; ++i)
703 free (filenames[i]);
704 free (filenames);
705 free (root);
706 return (NULL);
707 }
709 ++filenames_num;
710 tmp = (char **) realloc (filenames,
711 filenames_num * sizeof (*filenames));
712 if (tmp == NULL) {
713 ERROR ("configfile: realloc failed.");
714 for (i = 0; i < filenames_num - 1; ++i)
715 free (filenames[i]);
716 free (filenames);
717 free (root);
718 return (NULL);
719 }
720 filenames = tmp;
722 filenames[filenames_num - 1] = sstrdup (name);
723 }
725 qsort ((void *) filenames, filenames_num, sizeof (*filenames),
726 cf_compare_string);
728 for (i = 0; i < filenames_num; ++i)
729 {
730 oconfig_item_t *temp;
731 char *name = filenames[i];
733 temp = cf_read_generic (name, pattern, depth);
734 if (temp == NULL)
735 {
736 /* An error should already have been reported. */
737 sfree (name);
738 continue;
739 }
741 cf_ci_append_children (root, temp);
742 sfree (temp->children);
743 sfree (temp);
745 free (name);
746 }
748 free(filenames);
749 return (root);
750 } /* oconfig_item_t *cf_read_dir */
752 /*
753 * cf_read_generic
754 *
755 * Path is stat'ed and either cf_read_file or cf_read_dir is called
756 * accordingly.
757 *
758 * There are two versions of this function: If `wordexp' exists shell wildcards
759 * will be expanded and the function will include all matches found. If
760 * `wordexp' (or, more precisely, it's header file) is not available the
761 * simpler function is used which does not do any such expansion.
762 */
763 #if HAVE_WORDEXP_H
764 static oconfig_item_t *cf_read_generic (const char *path,
765 const char *pattern, int depth)
766 {
767 oconfig_item_t *root = NULL;
768 int status;
769 const char *path_ptr;
770 wordexp_t we;
771 size_t i;
773 if (depth >= CF_MAX_DEPTH)
774 {
775 ERROR ("configfile: Not including `%s' because the maximum "
776 "nesting depth has been reached.", path);
777 return (NULL);
778 }
780 status = wordexp (path, &we, WRDE_NOCMD);
781 if (status != 0)
782 {
783 ERROR ("configfile: wordexp (%s) failed.", path);
784 return (NULL);
785 }
787 root = (oconfig_item_t *) malloc (sizeof (oconfig_item_t));
788 if (root == NULL)
789 {
790 ERROR ("configfile: malloc failed.");
791 return (NULL);
792 }
793 memset (root, '\0', sizeof (oconfig_item_t));
795 /* wordexp() might return a sorted list already. That's not
796 * documented though, so let's make sure we get what we want. */
797 qsort ((void *) we.we_wordv, we.we_wordc, sizeof (*we.we_wordv),
798 cf_compare_string);
800 for (i = 0; i < we.we_wordc; i++)
801 {
802 oconfig_item_t *temp;
803 struct stat statbuf;
805 path_ptr = we.we_wordv[i];
807 status = stat (path_ptr, &statbuf);
808 if (status != 0)
809 {
810 char errbuf[1024];
811 WARNING ("configfile: stat (%s) failed: %s",
812 path_ptr,
813 sstrerror (errno, errbuf, sizeof (errbuf)));
814 continue;
815 }
817 if (S_ISREG (statbuf.st_mode))
818 temp = cf_read_file (path_ptr, pattern, depth);
819 else if (S_ISDIR (statbuf.st_mode))
820 temp = cf_read_dir (path_ptr, pattern, depth);
821 else
822 {
823 WARNING ("configfile: %s is neither a file nor a "
824 "directory.", path);
825 continue;
826 }
828 if (temp == NULL) {
829 oconfig_free (root);
830 return (NULL);
831 }
833 cf_ci_append_children (root, temp);
834 sfree (temp->children);
835 sfree (temp);
836 }
838 wordfree (&we);
840 return (root);
841 } /* oconfig_item_t *cf_read_generic */
842 /* #endif HAVE_WORDEXP_H */
844 #else /* if !HAVE_WORDEXP_H */
845 static oconfig_item_t *cf_read_generic (const char *path,
846 const char *pattern, int depth)
847 {
848 struct stat statbuf;
849 int status;
851 if (depth >= CF_MAX_DEPTH)
852 {
853 ERROR ("configfile: Not including `%s' because the maximum "
854 "nesting depth has been reached.", path);
855 return (NULL);
856 }
858 status = stat (path, &statbuf);
859 if (status != 0)
860 {
861 char errbuf[1024];
862 ERROR ("configfile: stat (%s) failed: %s",
863 path,
864 sstrerror (errno, errbuf, sizeof (errbuf)));
865 return (NULL);
866 }
868 if (S_ISREG (statbuf.st_mode))
869 return (cf_read_file (path, pattern, depth));
870 else if (S_ISDIR (statbuf.st_mode))
871 return (cf_read_dir (path, pattern, depth));
873 ERROR ("configfile: %s is neither a file nor a directory.", path);
874 return (NULL);
875 } /* oconfig_item_t *cf_read_generic */
876 #endif /* !HAVE_WORDEXP_H */
878 /*
879 * Public functions
880 */
881 int global_option_set (const char *option, const char *value)
882 {
883 int i;
885 DEBUG ("option = %s; value = %s;", option, value);
887 for (i = 0; i < cf_global_options_num; i++)
888 if (strcasecmp (cf_global_options[i].key, option) == 0)
889 break;
891 if (i >= cf_global_options_num)
892 return (-1);
894 sfree (cf_global_options[i].value);
896 if (value != NULL)
897 cf_global_options[i].value = strdup (value);
898 else
899 cf_global_options[i].value = NULL;
901 return (0);
902 }
904 const char *global_option_get (const char *option)
905 {
906 int i;
908 for (i = 0; i < cf_global_options_num; i++)
909 if (strcasecmp (cf_global_options[i].key, option) == 0)
910 break;
912 if (i >= cf_global_options_num)
913 return (NULL);
915 return ((cf_global_options[i].value != NULL)
916 ? cf_global_options[i].value
917 : cf_global_options[i].def);
918 } /* char *global_option_get */
920 long global_option_get_long (const char *option, long default_value)
921 {
922 const char *str;
923 long value;
925 str = global_option_get (option);
926 if (NULL == str)
927 return (default_value);
929 errno = 0;
930 value = strtol (str, /* endptr = */ NULL, /* base = */ 0);
931 if (errno != 0)
932 return (default_value);
934 return (value);
935 } /* char *global_option_get_long */
937 cdtime_t cf_get_default_interval (void)
938 {
939 char const *str = global_option_get ("Interval");
940 double interval_double = COLLECTD_DEFAULT_INTERVAL;
942 if (str != NULL)
943 {
944 char *endptr = NULL;
945 double tmp = strtod (str, &endptr);
947 if ((endptr == NULL) || (endptr == str) || (*endptr != 0))
948 ERROR ("cf_get_default_interval: Unable to parse string \"%s\" "
949 "as number.", str);
950 else if (tmp <= 0.0)
951 ERROR ("cf_get_default_interval: Interval must be a positive number. "
952 "The current number is %g.", tmp);
953 else
954 interval_double = tmp;
955 }
957 return (DOUBLE_TO_CDTIME_T (interval_double));
958 } /* }}} cdtime_t cf_get_default_interval */
960 void cf_unregister (const char *type)
961 {
962 cf_callback_t *this, *prev;
964 for (prev = NULL, this = first_callback;
965 this != NULL;
966 prev = this, this = this->next)
967 if (strcasecmp (this->type, type) == 0)
968 {
969 if (prev == NULL)
970 first_callback = this->next;
971 else
972 prev->next = this->next;
974 free (this);
975 break;
976 }
977 } /* void cf_unregister */
979 void cf_unregister_complex (const char *type)
980 {
981 cf_complex_callback_t *this, *prev;
983 for (prev = NULL, this = complex_callback_head;
984 this != NULL;
985 prev = this, this = this->next)
986 if (strcasecmp (this->type, type) == 0)
987 {
988 if (prev == NULL)
989 complex_callback_head = this->next;
990 else
991 prev->next = this->next;
993 sfree (this->type);
994 sfree (this);
995 break;
996 }
997 } /* void cf_unregister */
999 void cf_register (const char *type,
1000 int (*callback) (const char *, const char *),
1001 const char **keys, int keys_num)
1002 {
1003 cf_callback_t *cf_cb;
1005 /* Remove this module from the list, if it already exists */
1006 cf_unregister (type);
1008 /* This pointer will be free'd in `cf_unregister' */
1009 if ((cf_cb = (cf_callback_t *) malloc (sizeof (cf_callback_t))) == NULL)
1010 return;
1012 cf_cb->type = type;
1013 cf_cb->callback = callback;
1014 cf_cb->keys = keys;
1015 cf_cb->keys_num = keys_num;
1016 cf_cb->ctx = plugin_get_ctx ();
1018 cf_cb->next = first_callback;
1019 first_callback = cf_cb;
1020 } /* void cf_register */
1022 int cf_register_complex (const char *type, int (*callback) (oconfig_item_t *))
1023 {
1024 cf_complex_callback_t *new;
1026 new = (cf_complex_callback_t *) malloc (sizeof (cf_complex_callback_t));
1027 if (new == NULL)
1028 return (-1);
1030 new->type = strdup (type);
1031 if (new->type == NULL)
1032 {
1033 sfree (new);
1034 return (-1);
1035 }
1037 new->callback = callback;
1038 new->next = NULL;
1040 new->ctx = plugin_get_ctx ();
1042 if (complex_callback_head == NULL)
1043 {
1044 complex_callback_head = new;
1045 }
1046 else
1047 {
1048 cf_complex_callback_t *last = complex_callback_head;
1049 while (last->next != NULL)
1050 last = last->next;
1051 last->next = new;
1052 }
1054 return (0);
1055 } /* int cf_register_complex */
1057 int cf_read (char *filename)
1058 {
1059 oconfig_item_t *conf;
1060 int i;
1062 conf = cf_read_generic (filename, /* pattern = */ NULL, /* depth = */ 0);
1063 if (conf == NULL)
1064 {
1065 ERROR ("Unable to read config file %s.", filename);
1066 return (-1);
1067 }
1068 else if (conf->children_num == 0)
1069 {
1070 ERROR ("Configuration file %s is empty.", filename);
1071 oconfig_free (conf);
1072 return (-1);
1073 }
1075 for (i = 0; i < conf->children_num; i++)
1076 {
1077 if (conf->children[i].children == NULL)
1078 dispatch_value (conf->children + i);
1079 else
1080 dispatch_block (conf->children + i);
1081 }
1083 oconfig_free (conf);
1085 /* Read the default types.db if no `TypesDB' option was given. */
1086 if (cf_default_typesdb)
1087 read_types_list (PKGDATADIR"/types.db");
1089 return (0);
1090 } /* int cf_read */
1092 /* Assures the config option is a string, duplicates it and returns the copy in
1093 * "ret_string". If necessary "*ret_string" is freed first. Returns zero upon
1094 * success. */
1095 int cf_util_get_string (const oconfig_item_t *ci, char **ret_string) /* {{{ */
1096 {
1097 char *string;
1099 if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING))
1100 {
1101 ERROR ("cf_util_get_string: The %s option requires "
1102 "exactly one string argument.", ci->key);
1103 return (-1);
1104 }
1106 string = strdup (ci->values[0].value.string);
1107 if (string == NULL)
1108 return (-1);
1110 if (*ret_string != NULL)
1111 sfree (*ret_string);
1112 *ret_string = string;
1114 return (0);
1115 } /* }}} int cf_util_get_string */
1117 /* Assures the config option is a string and copies it to the provided buffer.
1118 * Assures null-termination. */
1119 int cf_util_get_string_buffer (const oconfig_item_t *ci, char *buffer, /* {{{ */
1120 size_t buffer_size)
1121 {
1122 if ((ci == NULL) || (buffer == NULL) || (buffer_size < 1))
1123 return (EINVAL);
1125 if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_STRING))
1126 {
1127 ERROR ("cf_util_get_string_buffer: The %s option requires "
1128 "exactly one string argument.", ci->key);
1129 return (-1);
1130 }
1132 strncpy (buffer, ci->values[0].value.string, buffer_size);
1133 buffer[buffer_size - 1] = 0;
1135 return (0);
1136 } /* }}} int cf_util_get_string_buffer */
1138 /* Assures the config option is a number and returns it as an int. */
1139 int cf_util_get_int (const oconfig_item_t *ci, int *ret_value) /* {{{ */
1140 {
1141 if ((ci == NULL) || (ret_value == NULL))
1142 return (EINVAL);
1144 if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_NUMBER))
1145 {
1146 ERROR ("cf_util_get_int: The %s option requires "
1147 "exactly one numeric argument.", ci->key);
1148 return (-1);
1149 }
1151 *ret_value = (int) ci->values[0].value.number;
1153 return (0);
1154 } /* }}} int cf_util_get_int */
1156 int cf_util_get_double (const oconfig_item_t *ci, double *ret_value) /* {{{ */
1157 {
1158 if ((ci == NULL) || (ret_value == NULL))
1159 return (EINVAL);
1161 if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_NUMBER))
1162 {
1163 ERROR ("cf_util_get_double: The %s option requires "
1164 "exactly one numeric argument.", ci->key);
1165 return (-1);
1166 }
1168 *ret_value = ci->values[0].value.number;
1170 return (0);
1171 } /* }}} int cf_util_get_double */
1173 int cf_util_get_boolean (const oconfig_item_t *ci, _Bool *ret_bool) /* {{{ */
1174 {
1175 if ((ci == NULL) || (ret_bool == NULL))
1176 return (EINVAL);
1178 if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_BOOLEAN))
1179 {
1180 ERROR ("cf_util_get_boolean: The %s option requires "
1181 "exactly one boolean argument.", ci->key);
1182 return (-1);
1183 }
1185 *ret_bool = ci->values[0].value.boolean ? 1 : 0;
1187 return (0);
1188 } /* }}} int cf_util_get_boolean */
1190 int cf_util_get_flag (const oconfig_item_t *ci, /* {{{ */
1191 unsigned int *ret_value, unsigned int flag)
1192 {
1193 int status;
1194 _Bool b;
1196 if (ret_value == NULL)
1197 return (EINVAL);
1199 b = 0;
1200 status = cf_util_get_boolean (ci, &b);
1201 if (status != 0)
1202 return (status);
1204 if (b)
1205 {
1206 *ret_value |= flag;
1207 }
1208 else
1209 {
1210 *ret_value &= ~flag;
1211 }
1213 return (0);
1214 } /* }}} int cf_util_get_flag */
1216 /* Assures that the config option is a string or a number if the correct range
1217 * of 1-65535. The string is then converted to a port number using
1218 * `service_name_to_port_number' and returned.
1219 * Returns the port number in the range [1-65535] or less than zero upon
1220 * failure. */
1221 int cf_util_get_port_number (const oconfig_item_t *ci) /* {{{ */
1222 {
1223 int tmp;
1225 if ((ci->values_num != 1)
1226 || ((ci->values[0].type != OCONFIG_TYPE_STRING)
1227 && (ci->values[0].type != OCONFIG_TYPE_NUMBER)))
1228 {
1229 ERROR ("cf_util_get_port_number: The \"%s\" option requires "
1230 "exactly one string argument.", ci->key);
1231 return (-1);
1232 }
1234 if (ci->values[0].type == OCONFIG_TYPE_STRING)
1235 return (service_name_to_port_number (ci->values[0].value.string));
1237 assert (ci->values[0].type == OCONFIG_TYPE_NUMBER);
1238 tmp = (int) (ci->values[0].value.number + 0.5);
1239 if ((tmp < 1) || (tmp > 65535))
1240 {
1241 ERROR ("cf_util_get_port_number: The \"%s\" option requires "
1242 "a service name or a port number. The number "
1243 "you specified, %i, is not in the valid "
1244 "range of 1-65535.",
1245 ci->key, tmp);
1246 return (-1);
1247 }
1249 return (tmp);
1250 } /* }}} int cf_util_get_port_number */
1252 int cf_util_get_service (const oconfig_item_t *ci, char **ret_string) /* {{{ */
1253 {
1254 int port;
1255 char *service;
1256 int status;
1258 if (ci->values_num != 1)
1259 {
1260 ERROR ("cf_util_get_service: The %s option requires exactly "
1261 "one argument.", ci->key);
1262 return (-1);
1263 }
1265 if (ci->values[0].type == OCONFIG_TYPE_STRING)
1266 return (cf_util_get_string (ci, ret_string));
1267 if (ci->values[0].type != OCONFIG_TYPE_NUMBER)
1268 {
1269 ERROR ("cf_util_get_service: The %s option requires "
1270 "exactly one string or numeric argument.",
1271 ci->key);
1272 }
1274 port = 0;
1275 status = cf_util_get_int (ci, &port);
1276 if (status != 0)
1277 return (status);
1278 else if ((port < 1) || (port > 65535))
1279 {
1280 ERROR ("cf_util_get_service: The port number given "
1281 "for the %s option is out of "
1282 "range (%i).", ci->key, port);
1283 return (-1);
1284 }
1286 service = malloc (6);
1287 if (service == NULL)
1288 {
1289 ERROR ("cf_util_get_service: Out of memory.");
1290 return (-1);
1291 }
1292 ssnprintf (service, 6, "%i", port);
1294 sfree (*ret_string);
1295 *ret_string = service;
1297 return (0);
1298 } /* }}} int cf_util_get_service */
1300 int cf_util_get_cdtime (const oconfig_item_t *ci, cdtime_t *ret_value) /* {{{ */
1301 {
1302 if ((ci == NULL) || (ret_value == NULL))
1303 return (EINVAL);
1305 if ((ci->values_num != 1) || (ci->values[0].type != OCONFIG_TYPE_NUMBER))
1306 {
1307 ERROR ("cf_util_get_cdtime: The %s option requires "
1308 "exactly one numeric argument.", ci->key);
1309 return (-1);
1310 }
1312 if (ci->values[0].value.number < 0.0)
1313 {
1314 ERROR ("cf_util_get_cdtime: The numeric argument of the %s "
1315 "option must not be negative.", ci->key);
1316 return (-1);
1317 }
1319 *ret_value = DOUBLE_TO_CDTIME_T (ci->values[0].value.number);
1321 return (0);
1322 } /* }}} int cf_util_get_cdtime */