2 * collectd - src/configfile.c
3 * Copyright (C) 2005 Florian octo Forster
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.
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.
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
20 * Florian octo Forster <octo at verplant.org>
25 #include "libconfig/libconfig.h"
29 #include "configfile.h"
31 #include "utils_debug.h"
33 #define SHORTOPT_NONE 0
35 #define ERR_NOT_NESTED "Sections cannot be nested.\n"
36 #define ERR_SECTION_ONLY "`%s' can only be used as section.\n"
37 #define ERR_NEEDS_ARG "Section `%s' needs an argument.\n"
38 #define ERR_NEEDS_SECTION "`%s' can only be used within a section.\n"
40 extern int operating_mode;
42 typedef struct cf_callback
45 int (*callback) (char *, char *);
48 struct cf_callback *next;
51 static cf_callback_t *first_callback = NULL;
53 typedef struct cf_mode_item
63 static cf_mode_item_t cf_mode_list[] =
65 {"TimeToLive", NULL, MODE_CLIENT },
66 {"PIDFile", NULL, MODE_CLIENT | MODE_SERVER | MODE_LOCAL | MODE_LOG },
67 {"DataDir", NULL, MODE_CLIENT | MODE_SERVER | MODE_LOCAL | MODE_LOG },
68 {"LogFile", NULL, MODE_CLIENT | MODE_SERVER | MODE_LOCAL | MODE_LOG }
70 static int cf_mode_num = 4;
72 static int nesting_depth = 0;
73 static char *current_module = NULL;
75 /* `cf_register' needs this prototype */
76 static int cf_callback_plugin_dispatch (const char *, const char *,
77 const char *, const char *, lc_flags_t, void *);
80 * Functions to handle register/unregister, search, and other plugin related
83 static cf_callback_t *cf_search (char *type)
90 for (cf_cb = first_callback; cf_cb != NULL; cf_cb = cf_cb->next)
91 if (strcasecmp (cf_cb->type, type) == 0)
97 static int cf_dispatch (char *type, const char *orig_key, const char *orig_value)
105 DBG ("type = %s, key = %s, value = %s", type, orig_key, orig_value);
107 if ((cf_cb = cf_search (type)) == NULL)
109 syslog (LOG_WARNING, "Plugin `%s' did not register a callback.", type);
113 if ((key = strdup (orig_key)) == NULL)
115 if ((value = strdup (orig_value)) == NULL)
123 for (i = 0; i < cf_cb->keys_num; i++)
125 if (strcasecmp (cf_cb->keys[i], key) == 0)
127 ret = (*cf_cb->callback) (key, value);
132 if (i >= cf_cb->keys_num)
133 syslog (LOG_WARNING, "Plugin `%s' did not register for value `%s'.", type, key);
138 DBG ("return (%i)", ret);
143 void cf_unregister (char *type)
145 cf_callback_t *this, *prev;
147 for (prev = NULL, this = first_callback;
149 prev = this, this = this->next)
150 if (strcasecmp (this->type, type) == 0)
153 first_callback = this->next;
155 prev->next = this->next;
162 void cf_register (char *type,
163 int (*callback) (char *, char *),
164 char **keys, int keys_num)
166 cf_callback_t *cf_cb;
170 /* Remove this module from the list, if it already exists */
171 cf_unregister (type);
173 /* This pointer will be free'd in `cf_unregister' */
174 if ((cf_cb = (cf_callback_t *) malloc (sizeof (cf_callback_t))) == NULL)
178 cf_cb->callback = callback;
180 cf_cb->keys_num = keys_num;
182 cf_cb->next = first_callback;
183 first_callback = cf_cb;
185 for (i = 0; i < keys_num; i++)
187 if (snprintf (buf, 64, "Plugin.%s", keys[i]) < 64)
189 /* This may be called multiple times for the same
190 * `key', but apparently `lc_register_*' can handle
192 lc_register_callback (buf, SHORTOPT_NONE,
193 LC_VAR_STRING, cf_callback_plugin_dispatch,
198 DBG ("Key was truncated: `%s'", keys[i]);
204 * Other query functions
206 char *cf_get_option (const char *key, char *def)
210 for (i = 0; i < cf_mode_num; i++)
212 if ((cf_mode_list[i].mode & operating_mode) == 0)
215 if (strcasecmp (cf_mode_list[i].key, key) != 0)
218 if (cf_mode_list[i].value != NULL)
219 return (cf_mode_list[i].value);
226 /* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * *
227 * Functions for the actual parsing *
228 * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */
232 * Chose the `operating_mode'
236 static int cf_callback_mode (const char *shortvar, const char *var,
237 const char *arguments, const char *value, lc_flags_t flags,
240 DBG ("shortvar = %s, var = %s, arguments = %s, value = %s, ...",
241 shortvar, var, arguments, value);
243 if (strcasecmp (value, "Client") == 0)
244 operating_mode = MODE_CLIENT;
246 else if (strcasecmp (value, "Server") == 0)
247 operating_mode = MODE_SERVER;
248 else if (strcasecmp (value, "Local") == 0)
249 operating_mode = MODE_LOCAL;
251 else if (strcasecmp (value, "Log") == 0)
252 operating_mode = MODE_LOG;
255 syslog (LOG_ERR, "Invalid value for config option `Mode': `%s'", value);
256 return (LC_CBRET_ERROR);
259 return (LC_CBRET_OKAY);
263 * `cf_callback_mode_plugindir'
264 * Change the plugin directory
270 static int cf_callback_mode_plugindir (const char *shortvar, const char *var,
271 const char *arguments, const char *value, lc_flags_t flags,
274 DBG ("shortvar = %s, var = %s, arguments = %s, value = %s, ...",
275 shortvar, var, arguments, value);
277 plugin_set_dir (value);
279 return (LC_CBRET_OKAY);
282 static int cf_callback_mode_option (const char *shortvar, const char *var,
283 const char *arguments, const char *value, lc_flags_t flags,
286 cf_mode_item_t *item;
288 DBG ("shortvar = %s, var = %s, arguments = %s, value = %s, ...",
289 shortvar, var, arguments, value);
293 fprintf (stderr, "No extra..?\n");
294 return (LC_CBRET_ERROR);
297 item = (cf_mode_item_t *) extra;
299 if (strcasecmp (item->key, shortvar))
301 fprintf (stderr, "Wrong extra..\n");
302 return (LC_CBRET_ERROR);
305 if ((operating_mode & item->mode) == 0)
307 fprintf (stderr, "Option `%s' is not valid in this mode!\n", shortvar);
308 return (LC_CBRET_ERROR);
311 if (item->value != NULL)
317 if ((item->value = strdup (value)) == NULL)
320 return (LC_CBRET_ERROR);
323 return (LC_CBRET_OKAY);
327 * `cf_callback_mode_loadmodule':
334 static int cf_callback_mode_loadmodule (const char *shortvar, const char *var,
335 const char *arguments, const char *value, lc_flags_t flags,
338 DBG ("shortvar = %s, var = %s, arguments = %s, value = %s, ...",
339 shortvar, var, arguments, value);
341 if (plugin_load (value))
342 syslog (LOG_ERR, "plugin_load (%s): failed to load plugin", value);
344 /* Return `okay' even if there was an error, because it's not a syntax
346 return (LC_CBRET_OKAY);
349 static int cf_callback_socket (const char *shortvar, const char *var,
350 const char *arguments, const char *value, lc_flags_t flags,
359 char *service = NET_DEFAULT_PORT;
361 DBG ("shortvar = %s, var = %s, arguments = %s, value = %s, ...",
362 shortvar, var, arguments, value);
364 buffer = strdup (value);
366 return (LC_CBRET_ERROR);
368 numfields = strsplit (buffer, fields, 3);
370 if ((numfields != 1) && (numfields != 2))
372 syslog (LOG_ERR, "Invalid number of arguments to `%s'",
375 return (LC_CBRET_ERROR);
382 /* Still return `LC_CBRET_OKAY' because this is not an syntax error */
383 if (network_create_socket (node, service) < 1)
384 syslog (LOG_ERR, "network_create_socket (%s, %s) failed",
389 return (LC_CBRET_OKAY);
393 * `cf_callback_plugin'
394 * Start/end section `plugin'
396 * <Plugin `arguments'>
400 static int cf_callback_plugin (const char *shortvar, const char *var,
401 const char *arguments, const char *value, lc_flags_t flags,
404 DBG ("shortvar = %s, var = %s, arguments = %s, value = %s, ...",
405 shortvar, var, arguments, value);
407 if (flags == LC_FLAGS_SECTIONSTART)
409 if (nesting_depth != 0)
411 fprintf (stderr, ERR_NOT_NESTED);
412 return (LC_CBRET_ERROR);
415 if (arguments == NULL)
417 fprintf (stderr, ERR_NEEDS_ARG, shortvar);
418 return (LC_CBRET_ERROR);
421 if ((current_module = strdup (arguments)) == NULL)
424 return (LC_CBRET_ERROR);
429 if (cf_search (current_module) != NULL)
430 return (LC_CBRET_OKAY);
432 return (LC_CBRET_IGNORESECTION);
434 else if (flags == LC_FLAGS_SECTIONEND)
436 if (current_module != NULL)
438 free (current_module);
439 current_module = NULL;
444 return (LC_CBRET_OKAY);
448 fprintf (stderr, ERR_SECTION_ONLY, shortvar);
449 return (LC_CBRET_ERROR);
454 * `cf_callback_plugin_dispatch'
455 * Send options within `plugin' sections to the plugin that requests it.
457 * <Plugin `current_module'>
461 static int cf_callback_plugin_dispatch (const char *shortvar, const char *var,
462 const char *arguments, const char *value, lc_flags_t flags,
465 DBG ("shortvar = %s, var = %s, arguments = %s, value = %s, ...",
466 shortvar, var, arguments, value);
468 if ((nesting_depth == 0) || (current_module == NULL))
470 fprintf (stderr, ERR_NEEDS_SECTION, shortvar);
471 return (LC_CBRET_ERROR);
474 /* Send the data to the plugin */
475 if (cf_dispatch (current_module, shortvar, value) < 0)
476 return (LC_CBRET_ERROR);
478 return (LC_CBRET_OKAY);
481 static void cf_init (void)
483 static int run_once = 0;
490 lc_register_callback ("Mode", SHORTOPT_NONE, LC_VAR_STRING,
491 cf_callback_mode, NULL);
492 lc_register_callback ("Plugin", SHORTOPT_NONE, LC_VAR_SECTION,
493 cf_callback_plugin, NULL);
495 lc_register_callback ("PluginDir", SHORTOPT_NONE,
496 LC_VAR_STRING, cf_callback_mode_plugindir, NULL);
497 lc_register_callback ("LoadPlugin", SHORTOPT_NONE,
498 LC_VAR_STRING, cf_callback_mode_loadmodule, NULL);
500 lc_register_callback ("Listen", SHORTOPT_NONE,
501 LC_VAR_STRING, cf_callback_socket, NULL);
502 lc_register_callback ("Server", SHORTOPT_NONE,
503 LC_VAR_STRING, cf_callback_socket, NULL);
505 for (i = 0; i < cf_mode_num; i++)
507 cf_mode_item_t *item;
509 item = &cf_mode_list[i];
511 lc_register_callback (item->key, SHORTOPT_NONE, LC_VAR_STRING,
512 cf_callback_mode_option, (void *) item);
516 int cf_read (char *filename)
520 if (filename == NULL)
521 filename = CONFIGFILE;
523 DBG ("Starting to parse file `%s'", filename);
525 /* int lc_process_file(const char *appname, const char *pathname, lc_conf_type_t type); */
526 if (lc_process_file ("collectd", filename, LC_CONF_APACHE))
528 syslog (LOG_ERR, "lc_process_file (%s): %s", filename, lc_geterrstr ());
532 DBG ("Done parsing file `%s'", filename);
534 /* free memory and stuff */