X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Fconfigfile.c;h=ce4e7747c95af39aee1f55c9bf1479c2c3917da9;hb=54ada5ddddb3692d8ec3493fbf4a5fd7aa39b5d1;hp=53a1753d791a20c1109115fc384e981e8ae7733e;hpb=c9db0c1a0dd6af026c3fbbf934eec02bd32bf043;p=collectd.git diff --git a/src/configfile.c b/src/configfile.c index 53a1753d..ce4e7747 100644 --- a/src/configfile.c +++ b/src/configfile.c @@ -1,6 +1,6 @@ /** * collectd - src/configfile.c - * Copyright (C) 2005,2006 Florian octo Forster + * Copyright (C) 2005-2008 Florian octo Forster * * This program is free software; you can redistribute it and/or modify it * under the terms of the GNU General Public License as published by the @@ -27,8 +27,13 @@ #include "common.h" #include "plugin.h" #include "configfile.h" +#include "types_list.h" #include "utils_threshold.h" +#if HAVE_WORDEXP_H +# include +#endif /* HAVE_WORDEXP_H */ + #define ESCAPE_NULL(str) ((str) == NULL ? "(null)" : (str)) /* @@ -66,6 +71,7 @@ typedef struct cf_global_option_s /* * Prototypes of callback functions */ +static int dispatch_value_typesdb (const oconfig_item_t *ci); static int dispatch_value_plugindir (const oconfig_item_t *ci); static int dispatch_value_loadplugin (const oconfig_item_t *ci); @@ -77,6 +83,7 @@ static cf_complex_callback_t *complex_callback_head = NULL; static cf_value_map_t cf_value_map[] = { + {"TypesDB", dispatch_value_typesdb}, {"PluginDir", dispatch_value_plugindir}, {"LoadPlugin", dispatch_value_loadplugin} }; @@ -89,11 +96,12 @@ static cf_global_option_t cf_global_options[] = {"Hostname", NULL, NULL}, {"FQDNLookup", NULL, "false"}, {"Interval", NULL, "10"}, - {"ReadThreads", NULL, "5"}, - {"TypesDB", NULL, PLUGINDIR"/types.db"} /* FIXME: Configure path */ + {"ReadThreads", NULL, "5"} }; static int cf_global_options_num = STATIC_ARRAY_LEN (cf_global_options); +static int cf_default_typesdb = 1; + /* * Functions to handle register/unregister, search, and other plugin related * stuff @@ -128,7 +136,9 @@ static int cf_dispatch (const char *type, const char *orig_key, if ((cf_cb = cf_search (type)) == NULL) { - WARNING ("Plugin `%s' did not register a callback.", type); + WARNING ("Found a configuration for the `%s' plugin, but " + "the plugin isn't loaded or didn't register " + "a configuration callback.", type); return (-1); } @@ -186,6 +196,27 @@ static int dispatch_global_option (const oconfig_item_t *ci) return (-1); } /* int dispatch_global_option */ +static int dispatch_value_typesdb (const oconfig_item_t *ci) +{ + int i = 0; + + assert (strcasecmp (ci->key, "TypesDB") == 0); + + cf_default_typesdb = 0; + + if (ci->values_num < 1) + return (-1); + + for (i = 0; i < ci->values_num; ++i) + { + if (OCONFIG_TYPE_STRING != ci->values[i].type) + continue; + + read_types_list (ci->values[i].value.string); + } + return (0); +} /* int dispatch_value_typesdb */ + static int dispatch_value_plugindir (const oconfig_item_t *ci) { assert (strcasecmp (ci->key, "PluginDir") == 0); @@ -313,8 +344,106 @@ static int dispatch_block (oconfig_item_t *ci) return (0); } +static int cf_ci_replace_child (oconfig_item_t *dst, oconfig_item_t *src, + int offset) +{ + oconfig_item_t *temp; + int i; + + assert (offset >= 0); + assert (dst->children_num > offset); + + /* Free the memory used by the replaced child. Usually that's the + * `Include "blah"' statement. */ + temp = dst->children + offset; + for (i = 0; i < temp->values_num; i++) + { + if (temp->values[i].type == OCONFIG_TYPE_STRING) + { + sfree (temp->values[i].value.string); + } + } + sfree (temp->values); + temp = NULL; + + /* If (src->children_num == 0) the array size is decreased. If offset + * is _not_ the last element, (offset < (src->children_num - 1)), then + * we need to move the trailing elements before resizing the array. */ + if ((src->children_num == 0) && (offset < (src->children_num - 1))) + { + int nmemb = src->children_num - (offset + 1); + memmove (src->children + offset, src->children + offset + 1, + sizeof (oconfig_item_t) * nmemb); + } + + /* Resize the memory containing the children to be big enough to hold + * all children. */ + temp = (oconfig_item_t *) realloc (dst->children, + sizeof (oconfig_item_t) + * (dst->children_num + src->children_num - 1)); + if (temp == NULL) + { + ERROR ("configfile: realloc failed."); + return (-1); + } + dst->children = temp; + + /* If there are children behind the include statement, and they have + * not yet been moved because (src->children_num == 0), then move them + * to the end of the list, so that the new children have room before + * them. */ + if ((src->children_num > 0) + && ((dst->children_num - (offset + 1)) > 0)) + { + int nmemb = dst->children_num - (offset + 1); + int old_offset = offset + 1; + int new_offset = offset + src->children_num; + + memmove (dst->children + new_offset, + dst->children + old_offset, + sizeof (oconfig_item_t) * nmemb); + } + + /* Last but not least: If there are new childrem, copy them to the + * memory reserved for them. */ + if (src->children_num > 0) + { + memcpy (dst->children + offset, + src->children, + sizeof (oconfig_item_t) * src->children_num); + } + + /* Update the number of children. */ + dst->children_num += (src->children_num - 1); + + return (0); +} /* int cf_ci_replace_child */ + +static int cf_ci_append_children (oconfig_item_t *dst, oconfig_item_t *src) +{ + oconfig_item_t *temp; + + temp = (oconfig_item_t *) realloc (dst->children, + sizeof (oconfig_item_t) + * (dst->children_num + src->children_num)); + if (temp == NULL) + { + ERROR ("configfile: realloc failed."); + return (-1); + } + dst->children = temp; + + memcpy (dst->children + dst->children_num, + src->children, + sizeof (oconfig_item_t) + * src->children_num); + dst->children_num += src->children_num; + + return (0); +} /* int cf_ci_append_children */ + #define CF_MAX_DEPTH 8 -static oconfig_item_t *cf_read_file (const char *file, int depth); +static oconfig_item_t *cf_read_generic (const char *path, int depth); static int cf_include_all (oconfig_item_t *root, int depth) { @@ -341,99 +470,207 @@ static int cf_include_all (oconfig_item_t *root, int depth) continue; } - new = cf_read_file (old->values[0].value.string, depth + 1); + new = cf_read_generic (old->values[0].value.string, depth + 1); if (new == NULL) continue; - /* There are more children now. We need to expand - * root->children. */ - if (new->children_num > 1) + /* Now replace the i'th child in `root' with `new'. */ + cf_ci_replace_child (root, new, i); + + sfree (new->values); + sfree (new); + } /* for (i = 0; i < root->children_num; i++) */ + + return (0); +} /* int cf_include_all */ + +static oconfig_item_t *cf_read_file (const char *file, int depth) +{ + oconfig_item_t *root; + + assert (depth < CF_MAX_DEPTH); + + root = oconfig_parse_file (file); + if (root == NULL) + { + ERROR ("configfile: Cannot read file `%s'.", file); + return (NULL); + } + + cf_include_all (root, depth); + + return (root); +} /* oconfig_item_t *cf_read_file */ + +static oconfig_item_t *cf_read_dir (const char *dir, int depth) +{ + oconfig_item_t *root = NULL; + DIR *dh; + struct dirent *de; + char name[1024]; + int status; + + assert (depth < CF_MAX_DEPTH); + + dh = opendir (dir); + if (dh == NULL) + { + char errbuf[1024]; + ERROR ("configfile: opendir failed: %s", + sstrerror (errno, errbuf, sizeof (errbuf))); + return (NULL); + } + + root = (oconfig_item_t *) malloc (sizeof (oconfig_item_t)); + if (root == NULL) + { + ERROR ("configfile: malloc failed."); + return (NULL); + } + memset (root, '\0', sizeof (oconfig_item_t)); + + while ((de = readdir (dh)) != NULL) + { + oconfig_item_t *temp; + + if ((de->d_name[0] == '.') || (de->d_name[0] == '\0')) + continue; + + status = snprintf (name, sizeof (name), "%s/%s", + dir, de->d_name); + if (status >= sizeof (name)) { - oconfig_item_t *temp; - - DEBUG ("configfile: Resizing root-children from %i to %i elements.", - root->children_num, - root->children_num + new->children_num - 1); - - temp = (oconfig_item_t *) realloc (root->children, - sizeof (oconfig_item_t) - * (root->children_num + new->children_num - 1)); - if (temp == NULL) - { - ERROR ("configfile: realloc failed."); - oconfig_free (new); - continue; - } - root->children = temp; + ERROR ("configfile: Not including `%s/%s' because its" + " name is too long.", + dir, de->d_name); + continue; } - /* Clean up the old include directive while we still have a - * valid pointer */ - DEBUG ("configfile: Cleaning up `old'"); - /* sfree (old->values[0].value.string); */ - sfree (old->values); - - /* If there are trailing children and the number of children - * changes, we need to move the trailing ones either one to the - * front or (new->num - 1) to the back */ - if (((root->children_num - i) > 1) - && (new->children_num != 1)) + temp = cf_read_generic (name, depth); + if (temp == NULL) + continue; + + cf_ci_append_children (root, temp); + sfree (temp->children); + sfree (temp); + } + + return (root); +} /* oconfig_item_t *cf_read_dir */ + +/* + * cf_read_generic + * + * Path is stat'ed and either cf_read_file or cf_read_dir is called + * accordingly. + * + * There are two versions of this function: If `wordexp' exists shell wildcards + * will be expanded and the function will include all matches found. If + * `wordexp' (or, more precisely, it's header file) is not available the + * simpler function is used which does not do any such expansion. + */ +#if HAVE_WORDEXP_H +static oconfig_item_t *cf_read_generic (const char *path, int depth) +{ + oconfig_item_t *root = NULL; + int status; + const char *path_ptr; + wordexp_t we; + int i; + + if (depth >= CF_MAX_DEPTH) + { + ERROR ("configfile: Not including `%s' because the maximum " + "nesting depth has been reached.", path); + return (NULL); + } + + status = wordexp (path, &we, WRDE_NOCMD); + if (status != 0) + { + ERROR ("configfile: wordexp (%s) failed.", path); + return (NULL); + } + + root = (oconfig_item_t *) malloc (sizeof (oconfig_item_t)); + if (root == NULL) + { + ERROR ("configfile: malloc failed."); + return (NULL); + } + memset (root, '\0', sizeof (oconfig_item_t)); + + for (i = 0; i < we.we_wordc; i++) + { + oconfig_item_t *temp; + struct stat statbuf; + + path_ptr = we.we_wordv[i]; + + status = stat (path_ptr, &statbuf); + if (status != 0) { - DEBUG ("configfile: Moving trailing children."); - memmove (root->children + i + new->children_num, - root->children + i + 1, - sizeof (oconfig_item_t) - * (root->children_num - (i + 1))); + char errbuf[1024]; + ERROR ("configfile: stat (%s) failed: %s", + path_ptr, + sstrerror (errno, errbuf, sizeof (errbuf))); + return (NULL); } - /* Now copy the new children to where the include statement was */ - if (new->children_num > 0) + if (S_ISREG (statbuf.st_mode)) + temp = cf_read_file (path_ptr, depth); + else if (S_ISDIR (statbuf.st_mode)) + temp = cf_read_dir (path_ptr, depth); + else { - DEBUG ("configfile: Copying new children."); - memcpy (root->children + i, - new->children, - sizeof (oconfig_item_t) - * new->children_num); + ERROR ("configfile: %s is neither a file nor a " + "directory.", path); + continue; } - /* Adjust the number of children and the position in the list. */ - root->children_num = root->children_num + new->children_num - 1; - i = i + new->children_num - 1; + cf_ci_append_children (root, temp); + sfree (temp->children); + sfree (temp); + } - /* Clean up the `new' struct. We set `new->children' to NULL so - * the stuff we've just copied pointers to isn't freed by - * `oconfig_free' */ - DEBUG ("configfile: Cleaning up `new'"); - sfree (new->values); /* should be NULL anyway */ - sfree (new); - new = NULL; - } /* for (i = 0; i < root->children_num; i++) */ + wordfree (&we); - return (0); -} /* int cf_include_all */ + return (root); +} /* oconfig_item_t *cf_read_generic */ +/* #endif HAVE_WORDEXP_H */ -static oconfig_item_t *cf_read_file (const char *file, int depth) +#else /* if !HAVE_WORDEXP_H */ +static oconfig_item_t *cf_read_generic (const char *path, int depth) { - oconfig_item_t *root; + struct stat statbuf; + int status; if (depth >= CF_MAX_DEPTH) { - ERROR ("configfile: Not including `%s' because the maximum nesting depth has been reached.", - file); + ERROR ("configfile: Not including `%s' because the maximum " + "nesting depth has been reached.", path); return (NULL); } - root = oconfig_parse_file (file); - if (root == NULL) + status = stat (path, &statbuf); + if (status != 0) { - ERROR ("configfile: Cannot read file `%s'.", file); + char errbuf[1024]; + ERROR ("configfile: stat (%s) failed: %s", + path, + sstrerror (errno, errbuf, sizeof (errbuf))); return (NULL); } - cf_include_all (root, depth); + if (S_ISREG (statbuf.st_mode)) + return (cf_read_file (path, depth)); + else if (S_ISDIR (statbuf.st_mode)) + return (cf_read_dir (path, depth)); - return (root); -} /* oconfig_item_t *cf_read_file */ + ERROR ("configfile: %s is neither a file nor a directory.", path); + return (NULL); +} /* oconfig_item_t *cf_read_generic */ +#endif /* !HAVE_WORDEXP_H */ /* * Public functions @@ -576,7 +813,7 @@ int cf_read (char *filename) oconfig_item_t *conf; int i; - conf = cf_read_file (filename, 0 /* depth */); + conf = cf_read_generic (filename, 0 /* depth */); if (conf == NULL) { ERROR ("Unable to read config file %s.", filename); @@ -591,5 +828,7 @@ int cf_read (char *filename) dispatch_block (conf->children + i); } + if (cf_default_typesdb) + read_types_list (PLUGINDIR"/types.db"); /* FIXME: Configure path */ return (0); } /* int cf_read */