X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Fconfigfile.c;h=f4e9c604fe26cafd75ee04620158b11f5aed6726;hb=5e73df5f8ec52055e5f79be5d898e5baec4c9967;hp=cbe2de9f7807679d518f078cc96b978bef213f62;hpb=f92c310c5f07fc735737a31a4db5d8b4765955c7;p=collectd.git

diff --git a/src/configfile.c b/src/configfile.c
index cbe2de9f..f4e9c604 100644
--- a/src/configfile.c
+++ b/src/configfile.c
@@ -1,6 +1,6 @@
 /**
  * collectd - src/configfile.c
- * Copyright (C) 2005  Florian octo Forster
+ * Copyright (C) 2005,2006  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
@@ -20,55 +20,83 @@
  *   Florian octo Forster <octo at verplant.org>
  **/
 
-/*
- * FIXME:
- * - remove all (I mean *ALL*) calls to `fprintf': `stderr' will have been
- *   closed.
- */
-
 #include "collectd.h"
 
-#include "libconfig/libconfig.h"
+#include "liboconfig/oconfig.h"
 
+#include "common.h"
 #include "plugin.h"
 #include "configfile.h"
-#include "utils_debug.h"
 
-#define SHORTOPT_NONE 0
-
-#define ERR_NOT_NESTED "Sections cannot be nested.\n"
-#define ERR_SECTION_ONLY "`%s' can only be used as section.\n"
-#define ERR_NEEDS_ARG "Section `%s' needs an argument.\n"
-#define ERR_NEEDS_SECTION "`%s' can only be used within a section.\n"
-
-#ifdef HAVE_LIBRRD
-extern int operating_mode;
-#else
-static int operating_mode = MODE_LOCAL;
-#endif
+#define ESCAPE_NULL(str) ((str) == NULL ? "(null)" : (str))
 
+/*
+ * Private types
+ */
 typedef struct cf_callback
 {
-	char  *type;
-	int  (*callback) (char *, char *);
-	char **keys;
+	const char  *type;
+	int  (*callback) (const char *, const char *);
+	const char **keys;
 	int    keys_num;
 	struct cf_callback *next;
 } cf_callback_t;
 
+typedef struct cf_complex_callback_s
+{
+	char *type;
+	int (*callback) (oconfig_item_t *);
+	struct cf_complex_callback_s *next;
+} cf_complex_callback_t;
+
+typedef struct cf_value_map_s
+{
+	char *key;
+	int (*func) (const oconfig_item_t *);
+} cf_value_map_t;
+
+typedef struct cf_global_option_s
+{
+	char *key;
+	char *value;
+	char *def;
+} cf_global_option_t;
+
+/*
+ * Prototypes of callback functions
+ */
+static int dispatch_value_plugindir (const oconfig_item_t *ci);
+static int dispatch_value_loadplugin (const oconfig_item_t *ci);
+
+/*
+ * Private variables
+ */
 static cf_callback_t *first_callback = NULL;
+static cf_complex_callback_t *complex_callback_head = NULL;
 
-static int nesting_depth = 0;
-static char *current_module = NULL;
+static cf_value_map_t cf_value_map[] =
+{
+	{"PluginDir",  dispatch_value_plugindir},
+	{"LoadPlugin", dispatch_value_loadplugin}
+};
+static int cf_value_map_num = STATIC_ARRAY_LEN (cf_value_map);
 
-/* cf_register needs this prototype */
-int cf_callback_general (const char *, const char *, const char *,
-		const char *, lc_flags_t, void *);
+static cf_global_option_t cf_global_options[] =
+{
+	{"BaseDir",   NULL, PKGLOCALSTATEDIR},
+	{"PIDFile",   NULL, PIDFILE},
+	{"Hostname",  NULL, NULL},
+	{"Interval",  NULL, "10"},
+	{"ReadThreads", NULL, "5"},
+	{"TypesDB",   NULL, PLUGINDIR"/types.db"} /* FIXME: Configure path */
+};
+static int cf_global_options_num = STATIC_ARRAY_LEN (cf_global_options);
 
 /*
- * Functions to handle register/unregister, search, ...
+ * Functions to handle register/unregister, search, and other plugin related
+ * stuff
  */
-cf_callback_t *cf_search (char *type)
+static cf_callback_t *cf_search (const char *type)
 {
 	cf_callback_t *cf_cb;
 
@@ -82,7 +110,8 @@ cf_callback_t *cf_search (char *type)
 	return (cf_cb);
 }
 
-int cf_dispatch (char *type, const char *orig_key, const char *orig_value)
+static int cf_dispatch (const char *type, const char *orig_key,
+		const char *orig_value)
 {
 	cf_callback_t *cf_cb;
 	char *key;
@@ -90,9 +119,14 @@ int cf_dispatch (char *type, const char *orig_key, const char *orig_value)
 	int ret;
 	int i;
 
+	DEBUG ("type = %s, key = %s, value = %s",
+			ESCAPE_NULL(type),
+			ESCAPE_NULL(orig_key),
+			ESCAPE_NULL(orig_value));
+
 	if ((cf_cb = cf_search (type)) == NULL)
 	{
-		fprintf (stderr, "Plugin `%s' did not register a callback.\n", type);
+		WARNING ("Plugin `%s' did not register a callback.", type);
 		return (-1);
 	}
 
@@ -116,249 +150,313 @@ int cf_dispatch (char *type, const char *orig_key, const char *orig_value)
 	}
 
 	if (i >= cf_cb->keys_num)
-		fprintf (stderr, "Plugin `%s' did not register for value `%s'.\n", type, key);
+		WARNING ("Plugin `%s' did not register for value `%s'.", type, key);
 
 	free (key);
 	free (value);
 
+	DEBUG ("return (%i)", ret);
+
 	return (ret);
-}
+} /* int cf_dispatch */
 
-void cf_unregister (char *type)
+static int dispatch_global_option (const oconfig_item_t *ci)
 {
-	cf_callback_t *this, *prev;
+	if (ci->values_num != 1)
+		return (-1);
+	if (ci->values[0].type == OCONFIG_TYPE_STRING)
+		return (global_option_set (ci->key, ci->values[0].value.string));
+	else if (ci->values[0].type == OCONFIG_TYPE_NUMBER)
+	{
+		char tmp[128];
+		snprintf (tmp, sizeof (tmp), "%lf", ci->values[0].value.number);
+		tmp[127] = '\0';
+		return (global_option_set (ci->key, tmp));
+	}
 
-	for (prev = NULL, this = first_callback;
-			this != NULL;
-			prev = this, this = this->next)
-		if (strcasecmp (this->type, type) == 0)
-		{
-			if (prev == NULL)
-				first_callback = this->next;
-			else
-				prev->next = this->next;
+	return (-1);
+} /* int dispatch_global_option */
 
-			free (this);
-			break;
-		}
+static int dispatch_value_plugindir (const oconfig_item_t *ci)
+{
+	assert (strcasecmp (ci->key, "PluginDir") == 0);
+	
+	if (ci->values_num != 1)
+		return (-1);
+	if (ci->values[0].type != OCONFIG_TYPE_STRING)
+		return (-1);
+
+	plugin_set_dir (ci->values[0].value.string);
+	return (0);
 }
 
-void cf_register (char *type,
-		int (*callback) (char *, char *),
-		char **keys, int keys_num)
+static int dispatch_value_loadplugin (const oconfig_item_t *ci)
 {
-	cf_callback_t *cf_cb;
-	char buf[64];
-	int i;
+	assert (strcasecmp (ci->key, "LoadPlugin") == 0);
 
-	/* Remove this module from the list, if it already exists */
-	cf_unregister (type);
+	if (ci->values_num != 1)
+		return (-1);
+	if (ci->values[0].type != OCONFIG_TYPE_STRING)
+		return (-1);
 
-	/* This pointer will be free'd in `cf_unregister' */
-	if ((cf_cb = (cf_callback_t *) malloc (sizeof (cf_callback_t))) == NULL)
-		return;
+	return (plugin_load (ci->values[0].value.string));
+} /* int dispatch_value_loadplugin */
 
-	cf_cb->type     = type;
-	cf_cb->callback = callback;
-	cf_cb->keys     = keys;
-	cf_cb->keys_num = keys_num;
+static int dispatch_value_plugin (const char *plugin, oconfig_item_t *ci)
+{
+	char  buffer[4096];
+	char *buffer_ptr;
+	int   buffer_free;
+	int i;
 
-	cf_cb->next = first_callback;
-	first_callback = cf_cb;
+	buffer_ptr = buffer;
+	buffer_free = sizeof (buffer);
 
-	for (i = 0; i < keys_num; i++)
+	for (i = 0; i < ci->values_num; i++)
 	{
-		if (snprintf (buf, 64, "Plugin.%s", keys[i]) < 64)
+		int status = -1;
+
+		if (ci->values[i].type == OCONFIG_TYPE_STRING)
+			status = snprintf (buffer_ptr, buffer_free, " %s",
+					ci->values[i].value.string);
+		else if (ci->values[i].type == OCONFIG_TYPE_NUMBER)
+			status = snprintf (buffer_ptr, buffer_free, " %lf",
+					ci->values[i].value.number);
+		else if (ci->values[i].type == OCONFIG_TYPE_BOOLEAN)
+			status = snprintf (buffer_ptr, buffer_free, " %s",
+					ci->values[i].value.boolean
+					? "true" : "false");
+
+		if ((status < 0) || (status >= buffer_free))
+			return (-1);
+		buffer_free -= status;
+		buffer_ptr  += status;
+	}
+	/* skip the initial space */
+	buffer_ptr = buffer + 1;
+
+	return (cf_dispatch (plugin, ci->key, buffer_ptr));
+} /* int plugin_conf_dispatch */
+
+static int dispatch_value (const oconfig_item_t *ci)
+{
+	int ret = -2;
+	int i;
+
+	for (i = 0; i < cf_value_map_num; i++)
+		if (strcasecmp (cf_value_map[i].key, ci->key) == 0)
 		{
-			/* This may be called multiple times for the same
-			 * `key', but apparently `lc_register_*' can handle
-			 * it.. */
-			lc_register_callback (buf, SHORTOPT_NONE,
-					LC_VAR_STRING, cf_callback_general,
-					NULL);
+			ret = cf_value_map[i].func (ci);
+			break;
 		}
-		else
+
+	for (i = 0; i < cf_global_options_num; i++)
+		if (strcasecmp (cf_global_options[i].key, ci->key) == 0)
 		{
-			DBG ("Key was truncated: `%s'", keys[i]);
+			ret = dispatch_global_option (ci);
+			break;
 		}
-	}
-}
 
-/* 
- * Functions for the actual parsing
- */
-int cf_callback_general (const char *shortvar, const char *var,
-		const char *arguments, const char *value, lc_flags_t flags,
-		void *extra)
+	return (ret);
+} /* int dispatch_value */
+
+static int dispatch_block_plugin (oconfig_item_t *ci)
 {
-	DBG ("shortvar = %s, var = %s, arguments = %s, value = %s, ...",
-			shortvar, var, arguments, value);
+	int i;
+	char *name;
 
-	if ((nesting_depth == 0) || (current_module == NULL))
+	cf_complex_callback_t *cb;
+
+	if (strcasecmp (ci->key, "Plugin") != 0)
+		return (-1);
+	if (ci->values_num < 1)
+		return (-1);
+	if (ci->values[0].type != OCONFIG_TYPE_STRING)
+		return (-1);
+
+	name = ci->values[0].value.string;
+
+	/* Check for a complex callback first */
+	for (cb = complex_callback_head; cb != NULL; cb = cb->next)
+		if (strcasecmp (name, cb->type) == 0)
+			return (cb->callback (ci));
+
+	/* Hm, no complex plugin found. Dispatch the values one by one */
+	for (i = 0; i < ci->children_num; i++)
 	{
-		fprintf (stderr, ERR_NEEDS_SECTION, shortvar);
-		return (LC_CBRET_ERROR);
+		if (ci->children[i].children == NULL)
+			dispatch_value_plugin (name, ci->children + i);
+		else
+			{DEBUG ("No nested config blocks allow for this plugin.");}
 	}
 
-	/* Send the data to the plugin */
-	if (cf_dispatch (current_module, shortvar, value) < 0)
-		return (LC_CBRET_ERROR);
+	return (0);
+}
+
 
-	return (LC_CBRET_OKAY);
+static int dispatch_block (oconfig_item_t *ci)
+{
+	if (strcasecmp (ci->key, "Plugin") == 0)
+		return (dispatch_block_plugin (ci));
+
+	return (0);
 }
 
-int cf_callback_section_mode (const char *shortvar, const char *var,
-		const char *arguments, const char *value, lc_flags_t flags,
-		void *extra)
+/* 
+ * Public functions
+ */
+int global_option_set (const char *option, const char *value)
 {
-	DBG ("shortvar = %s, var = %s, arguments = %s, value = %s, ...",
-			shortvar, var, arguments, value);
+	int i;
 
-	if (flags == LC_FLAGS_SECTIONSTART)
-	{
-		if (nesting_depth != 0)
-		{
-			fprintf (stderr, ERR_NOT_NESTED);
-			return (LC_CBRET_ERROR);
-		}
+	DEBUG ("option = %s; value = %s;", option, value);
 
-		if (arguments == NULL)
-		{
-			fprintf (stderr, ERR_NEEDS_ARG, shortvar);
-			return (LC_CBRET_ERROR);
-		}
+	for (i = 0; i < cf_global_options_num; i++)
+		if (strcasecmp (cf_global_options[i].key, option) == 0)
+			break;
 
-		nesting_depth++;
+	if (i >= cf_global_options_num)
+		return (-1);
 
-		if (((operating_mode == MODE_CLIENT)
-					&& (strcasecmp (arguments, "Client") == 0))
-				|| ((operating_mode == MODE_SERVER)
-					&& (strcasecmp (arguments, "Server") == 0))
-				|| ((operating_mode == MODE_LOCAL)
-					&& (strcasecmp (arguments, "Local") == 0)))
-		{
-			return (LC_CBRET_OKAY);
-		}
-		else
-		{
-			return (LC_CBRET_IGNORESECTION);
-		}
-	}
-	else if (flags == LC_FLAGS_SECTIONEND)
-	{
-		nesting_depth--;
+	sfree (cf_global_options[i].value);
 
-		return (LC_CBRET_OKAY);
-	}
+	if (value != NULL)
+		cf_global_options[i].value = strdup (value);
 	else
-	{
-		fprintf (stderr, ERR_SECTION_ONLY, shortvar);
-		return (LC_CBRET_ERROR);
-	}
+		cf_global_options[i].value = NULL;
 
+	return (0);
 }
 
-int cf_callback_section_module (const char *shortvar, const char *var,
-		const char *arguments, const char *value, lc_flags_t flags,
-		void *extra)
+const char *global_option_get (const char *option)
 {
-	DBG ("shortvar = %s, var = %s, arguments = %s, value = %s, ...",
-			shortvar, var, arguments, value);
+	int i;
 
-	if (flags == LC_FLAGS_SECTIONSTART)
-	{
-		if (nesting_depth != 0)
-		{
-			fprintf (stderr, ERR_NOT_NESTED);
-			return (LC_CBRET_ERROR);
-		}
+	for (i = 0; i < cf_global_options_num; i++)
+		if (strcasecmp (cf_global_options[i].key, option) == 0)
+			break;
 
-		if (arguments == NULL)
-		{
-			fprintf (stderr, ERR_NEEDS_ARG, shortvar);
-			return (LC_CBRET_ERROR);
-		}
+	if (i >= cf_global_options_num)
+		return (NULL);
+	
+	return ((cf_global_options[i].value != NULL)
+			? cf_global_options[i].value
+			: cf_global_options[i].def);
+} /* char *global_option_get */
+
+void cf_unregister (const char *type)
+{
+	cf_callback_t *this, *prev;
 
-		if ((current_module = strdup (arguments)) == NULL)
+	for (prev = NULL, this = first_callback;
+			this != NULL;
+			prev = this, this = this->next)
+		if (strcasecmp (this->type, type) == 0)
 		{
-			perror ("strdup");
-			return (LC_CBRET_ERROR);
+			if (prev == NULL)
+				first_callback = this->next;
+			else
+				prev->next = this->next;
+
+			free (this);
+			break;
 		}
+} /* void cf_unregister */
 
-		nesting_depth++;
+void cf_unregister_complex (const char *type)
+{
+	cf_complex_callback_t *this, *prev;
 
-		if (cf_search (current_module) != NULL)
-			return (LC_CBRET_OKAY);
-		else
-			return (LC_CBRET_IGNORESECTION);
-	}
-	else if (flags == LC_FLAGS_SECTIONEND)
-	{
-		if (current_module != NULL)
+	for (prev = NULL, this = complex_callback_head;
+			this != NULL;
+			prev = this, this = this->next)
+		if (strcasecmp (this->type, type) == 0)
 		{
-			free (current_module);
-			current_module == NULL;
+			if (prev == NULL)
+				complex_callback_head = this->next;
+			else
+				prev->next = this->next;
+
+			sfree (this->type);
+			sfree (this);
+			break;
 		}
+} /* void cf_unregister */
 
-		nesting_depth--;
+void cf_register (const char *type,
+		int (*callback) (const char *, const char *),
+		const char **keys, int keys_num)
+{
+	cf_callback_t *cf_cb;
 
-		return (LC_CBRET_OKAY);
-	}
-	else
-	{
-		fprintf (stderr, ERR_SECTION_ONLY, shortvar);
-		return (LC_CBRET_ERROR);
-	}
-}
+	/* Remove this module from the list, if it already exists */
+	cf_unregister (type);
+
+	/* This pointer will be free'd in `cf_unregister' */
+	if ((cf_cb = (cf_callback_t *) malloc (sizeof (cf_callback_t))) == NULL)
+		return;
 
-int cf_callback_loadmodule (const char *shortvar, const char *var,
-		const char *arguments, const char *value, lc_flags_t flags,
-		void *extra)
+	cf_cb->type     = type;
+	cf_cb->callback = callback;
+	cf_cb->keys     = keys;
+	cf_cb->keys_num = keys_num;
+
+	cf_cb->next = first_callback;
+	first_callback = cf_cb;
+} /* void cf_register */
+
+int cf_register_complex (const char *type, int (*callback) (oconfig_item_t *))
 {
-	DBG ("shortvar = %s, var = %s, arguments = %s, value = %s, ...",
-			shortvar, var, arguments, value);
+	cf_complex_callback_t *new;
+
+	new = (cf_complex_callback_t *) malloc (sizeof (cf_complex_callback_t));
+	if (new == NULL)
+		return (-1);
 
-	if (nesting_depth == 0)
+	new->type = strdup (type);
+	if (new->type == NULL)
 	{
-		fprintf (stderr, ERR_NEEDS_SECTION, shortvar);
-		return (LC_CBRET_ERROR);
+		sfree (new);
+		return (-1);
 	}
 
-	if (plugin_load (value))
-		syslog (LOG_ERR, "plugin_load (%s): failed to load plugin", shortvar);
+	new->callback = callback;
+	new->next = NULL;
 
-	/* Return `okay' even if there was an error, because it's not a syntax
-	 * problem.. */
-	return (LC_CBRET_OKAY);
-}
+	if (complex_callback_head == NULL)
+	{
+		complex_callback_head = new;
+	}
+	else
+	{
+		cf_complex_callback_t *last = complex_callback_head;
+		while (last->next != NULL)
+			last = last->next;
+		last->next = new;
+	}
+
+	return (0);
+} /* int cf_register_complex */
 
 int cf_read (char *filename)
 {
-	if (filename == NULL)
-		filename = CONFIGFILE;
-
-	lc_register_callback ("Mode", SHORTOPT_NONE, LC_VAR_SECTION,
-			cf_callback_section_mode, NULL);
-	lc_register_callback ("Plugin", SHORTOPT_NONE, LC_VAR_SECTION,
-			cf_callback_section_module, NULL);
-
-	/*
-	 * TODO:
-	 * - Add more directives, such as `DefaultMode', `DataDir', `PIDFile', ...
-	 */
-
-	lc_register_callback ("Mode.LoadPlugin", SHORTOPT_NONE,
-			LC_VAR_STRING, cf_callback_loadmodule,
-			NULL);
+	oconfig_item_t *conf;
+	int i;
 
-	if (lc_process_file ("collectd", filename, LC_CONF_APACHE))
+	conf = oconfig_parse_file (filename);
+	if (conf == NULL)
 	{
-		syslog (LOG_ERR, "lc_process_file (%s): %s", filename, lc_geterrstr ());
+		ERROR ("Unable to read config file %s.", filename);
 		return (-1);
 	}
 
-	/* free memory and stuff */
-	lc_cleanup ();
+	for (i = 0; i < conf->children_num; i++)
+	{
+		if (conf->children[i].children == NULL)
+			dispatch_value (conf->children + i);
+		else
+			dispatch_block (conf->children + i);
+	}
 
 	return (0);
-}
+} /* int cf_read */