X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Fcollectd.c;h=b4ee50460d947bfb04359119ef3c6bdd5504bae5;hb=039c2b763c6af3cc7ceeb011d15ddfb7af6fc5bb;hp=dabc346e8e8c5e79bfc7b388ad67675a5b0fec31;hpb=12f8e74dd7affea712575592d24e860c23b465a7;p=collectd.git diff --git a/src/collectd.c b/src/collectd.c index dabc346e..b4ee5046 100644 --- a/src/collectd.c +++ b/src/collectd.c @@ -1,36 +1,57 @@ +/** + * collectd - src/collectd.c + * 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 + * Free Software Foundation; either version 2 of the License, or (at your + * option) any later version. + * + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with this program; if not, write to the Free Software Foundation, Inc., + * 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA + * + * Authors: + * Florian octo Forster + * Alvaro Barcellos + **/ + #include "collectd.h" +#include "common.h" +#include "utils_debug.h" -#include "multicast.h" +#include "network.h" #include "plugin.h" - -#include "ping.h" +#include "configfile.h" static int loop = 0; -#ifdef HAVE_LIBKSTAT +#if HAVE_LIBKSTAT kstat_ctl_t *kc; #endif /* HAVE_LIBKSTAT */ -#if COLLECT_PING -char *pinghosts[MAX_PINGHOSTS]; -int num_pinghosts = 0; -#endif - /* * exported variables */ time_t curtime; +int operating_mode; -#ifdef HAVE_LIBRRD -int operating_mode; -#endif +static void sigIntHandler (int signal) +{ + loop++; +} -void sigIntHandler (int signal) +static void sigTermHandler (int signal) { loop++; } -int change_basedir (char *dir) +static int change_basedir (char *dir) { int dirlen = strlen (dir); @@ -46,12 +67,12 @@ int change_basedir (char *dir) { if (mkdir (dir, 0755) == -1) { - syslog (LOG_ERR, "mkdir: %s", strerror (errno)); + syslog (LOG_ERR, "mkdir (%s): %s", dir, strerror (errno)); return (-1); } else if (chdir (dir) == -1) { - syslog (LOG_ERR, "chdir: %s", strerror (errno)); + syslog (LOG_ERR, "chdir (%s): %s", dir, strerror (errno)); return (-1); } } @@ -63,10 +84,10 @@ int change_basedir (char *dir) } return (0); -} +} /* static int change_basedir (char *dir) */ -#ifdef HAVE_LIBKSTAT -void update_kstat (void) +#if HAVE_LIBKSTAT +static void update_kstat (void) { if (kc == NULL) { @@ -88,47 +109,56 @@ void update_kstat (void) } return; -} +} /* static void update_kstat (void) */ #endif /* HAVE_LIBKSTAT */ -void exit_usage (char *name) +/* TODO + * Remove all settings but `-f' and `-C' + */ +static void exit_usage (char *name) { - printf ("Usage: %s [OPTIONS]\n\n" + printf ("Usage: "PACKAGE" [OPTIONS]\n\n" "Available options:\n" " General:\n" - " -d Base directory to use.\n" - " Default: %s\n" - " -P Set the plugin-directory\n" - " Default: %s\n" - " -f Don't fork to the background\n" -#ifdef HAVE_LIBRRD - " -l Start in local mode (no network)\n" - " -c Start in client (sender) mode\n" - " -s Start in server (listener) mode\n" -#endif /* HAVE_LIBRRD */ -#if COLLECT_PING - " Ping:\n" - " -p Host to ping periodically, may be repeated to ping\n" - " more than one host.\n" -#endif /* COLLECT_PING */ - "\n%s %s, http://verplant.org/collectd/\n" + " -C Configuration file.\n" + " Default: "CONFIGFILE"\n" +#if COLLECT_DAEMON + " -f Don't fork to the background.\n" +#endif + "\nBuiltin defaults:\n" + " Config-File "CONFIGFILE"\n" + " PID-File "PIDFILE"\n" + " Data-Directory "PKGLOCALSTATEDIR"\n" +#if COLLECT_DEBUG + " Log-File "LOGFILE"\n" +#endif + " Step "COLLECTD_STEP" seconds\n" + " Heartbeat "COLLECTD_HEARTBEAT" seconds\n" + "\n"PACKAGE" "VERSION", http://collectd.org/\n" "by Florian octo Forster \n" - "for contributions see `AUTHORS'\n", - PACKAGE, DATADIR, PLUGINDIR, PACKAGE, VERSION); + "for contributions see `AUTHORS'\n"); exit (0); -} +} /* static void exit_usage (char *name) */ -int start_client (void) +static int start_client (void) { - int sleepingtime; + int step; + + struct timeval tv_now; + struct timeval tv_next; + struct timespec ts_wait; -#ifdef HAVE_LIBKSTAT + step = atoi (COLLECTD_STEP); + if (step <= 0) + step = 10; + +#if HAVE_LIBKSTAT kc = NULL; update_kstat (); #endif -#ifdef HAVE_LIBSTATGRAB +#if HAVE_LIBSTATGRAB if (sg_init ()) { syslog (LOG_ERR, "sg_init: %s", sg_str_error (sg_get_error ())); @@ -146,27 +176,52 @@ int start_client (void) while (loop == 0) { - curtime = time (NULL); -#ifdef HAVE_LIBKSTAT + if (gettimeofday (&tv_next, NULL) < 0) + { + syslog (LOG_ERR, "gettimeofday failed: %s", strerror (errno)); + return (-1); + } + tv_next.tv_sec += step; + +#if HAVE_LIBKSTAT update_kstat (); #endif + /* `curtime' is used by many (all?) plugins as the + * data-sample-time passed to RRDTool */ + curtime = time (NULL); + + /* Issue all plugins */ plugin_read_all (); - sleepingtime = 10; - while (sleepingtime != 0) + if (gettimeofday (&tv_now, NULL) < 0) { - if (loop != 0) + syslog (LOG_ERR, "gettimeofday failed: %s", strerror (errno)); + return (-1); + } + + if (timeval_sub_timespec (&tv_next, &tv_now, &ts_wait) != 0) + { + syslog (LOG_WARNING, "No sleeping because `timeval_sub_timespec' returned non-zero!"); + continue; + } + + while (nanosleep (&ts_wait, &ts_wait) == -1) + { + if (errno != EINTR) + { + syslog (LOG_ERR, "nanosleep failed: %s", strerror (errno)); break; - sleepingtime = sleep (sleepingtime); + } } } return (0); -} +} /* static int start_client (void) */ -#ifdef HAVE_LIBRRD -int start_server (void) +#if HAVE_LIBRRD +static int start_server (void) { + /* FIXME use stack here! */ char *host; char *type; char *instance; @@ -174,7 +229,7 @@ int start_server (void) while (loop == 0) { - if (multicast_receive (&host, &type, &instance, &values) == 0) + if (network_receive (&host, &type, &instance, &values) == 0) plugin_write (host, type, instance, values); if (host != NULL) free (host); host = NULL; @@ -184,63 +239,72 @@ int start_server (void) } return (0); -} +} /* static int start_server (void) */ #endif /* HAVE_LIBRRD */ -int pidfile_create (void) +#if COLLECT_DAEMON +static int pidfile_create (const char *file) { FILE *fh; - if ((fh = fopen (PIDFILE, "w")) == NULL) + if (file == NULL) + file = PIDFILE; + + if ((fh = fopen (file, "w")) == NULL) { - syslog (LOG_ERR, "fopen (pidfile): %s", strerror (errno)); + syslog (LOG_ERR, "fopen (%s): %s", file, strerror (errno)); return (1); } - fprintf (fh, "%d\n", getpid()); + fprintf (fh, "%i\n", (int) getpid ()); fclose(fh); return (0); -} +} /* static int pidfile_create (const char *file) */ +#endif /* COLLECT_DAEMON */ -int pidfile_remove (void) +#if COLLECT_DAEMON +static int pidfile_remove (const char *file) { - return (unlink (PIDFILE)); -} + if (file == NULL) { + file = PIDFILE; + } + return (unlink (file)); +} /* static int pidfile_remove (const char *file) */ +#endif /* COLLECT_DAEMON */ int main (int argc, char **argv) { - struct sigaction sigIntAction, sigChldAction; + struct sigaction sigIntAction; + struct sigaction sigTermAction; + char *datadir = PKGLOCALSTATEDIR; + char *configfile = CONFIGFILE; +#if COLLECT_DAEMON + struct sigaction sigChldAction; + char *pidfile = PIDFILE; pid_t pid; + int daemonize = 1; +#endif +#if COLLECT_DEBUG + char *logfile = LOGFILE; +#endif - char *plugindir = NULL; - char *basedir = DATADIR; - - int daemonize = 1; - -#ifdef HAVE_LIBRRD +#if HAVE_LIBRRD operating_mode = MODE_LOCAL; #endif - - /* - * open syslog - */ + + /* open syslog */ openlog (PACKAGE, LOG_CONS | LOG_PID, LOG_DAEMON); - /* - * read options - */ + /* read options */ while (1) { int c; - c = getopt (argc, argv, "d:fP:h" -#if HAVE_LIBRRD - "csl" -#endif /* HAVE_LIBRRD */ -#if COLLECT_PING - "p:" -#endif /* COLLECT_PING */ + c = getopt (argc, argv, "hC:" +#if COLLECT_DAEMON + "f" +#endif ); if (c == -1) @@ -248,73 +312,66 @@ int main (int argc, char **argv) switch (c) { -#ifdef HAVE_LIBRRD - case 'c': - operating_mode = MODE_CLIENT; - break; - - case 's': - operating_mode = MODE_SERVER; - break; - - case 'l': - operating_mode = MODE_LOCAL; - break; -#endif /* HAVE_LIBRRD */ - case 'd': - basedir = optarg; + case 'C': + configfile = optarg; break; +#if COLLECT_DAEMON case 'f': daemonize = 0; break; -#if COLLECT_PING - case 'p': - if (num_pinghosts < MAX_PINGHOSTS) - pinghosts[num_pinghosts++] = optarg; - else - fprintf (stderr, "Maximum of %i ping hosts reached.\n", MAX_PINGHOSTS); - break; -#endif /* COLLECT_PING */ - case 'P': - plugindir = optarg; - break; - +#endif /* COLLECT_DAEMON */ case 'h': default: exit_usage (argv[0]); - } - - } + } /* switch (c) */ + } /* while (1) */ + +#if COLLECT_DEBUG + if ((logfile = cf_get_option ("LogFile", LOGFILE)) != NULL) + DBG_STARTFILE (logfile, "Debug file opened."); +#endif /* - * Load plugins and change to output directory - * Loading plugins is done first so relative paths work as expected.. + * Read options from the config file, the environment and the command + * line (in that order, with later options overwriting previous ones in + * general). + * Also, this will automatically load modules. */ - if (plugin_load_all (plugindir) < 1) + if (cf_read (configfile)) { - fprintf (stderr, "Error: No plugins found.\n"); + fprintf (stderr, "Error: Reading the config file failed!\n" + "Read the syslog for details.\n"); return (1); } - if (change_basedir (basedir)) + /* + * Change directory. We do this _after_ reading the config and loading + * modules to relative paths work as expected. + */ + if ((datadir = cf_get_option ("DataDir", PKGLOCALSTATEDIR)) == NULL) + { + fprintf (stderr, "Don't have a datadir to use. This should not happen. Ever."); + return (1); + } + if (change_basedir (datadir)) { - fprintf (stderr, "Error: Unable to change to directory `%s'.\n", basedir); + fprintf (stderr, "Error: Unable to change to directory `%s'.\n", datadir); return (1); } +#if COLLECT_DAEMON /* - * install signal handlers + * fork off child */ - sigIntAction.sa_handler = sigIntHandler; - sigaction (SIGINT, &sigIntAction, NULL); - sigChldAction.sa_handler = SIG_IGN; sigaction (SIGCHLD, &sigChldAction, NULL); - /* - * fork off child - */ -#if DEBUG == 0 + if ((pidfile = cf_get_option ("PIDFile", PIDFILE)) == NULL) + { + fprintf (stderr, "Cannot obtain pidfile. This shoud not happen. Ever."); + return (1); + } + if (daemonize) { if ((pid = fork ()) == -1) @@ -334,7 +391,7 @@ int main (int argc, char **argv) setsid (); /* Write pidfile */ - if (pidfile_create ()) + if (pidfile_create (pidfile)) exit (2); /* close standard descriptors */ @@ -358,26 +415,40 @@ int main (int argc, char **argv) return (1); } } /* if (daemonize) */ -#endif +#endif /* COLLECT_DAEMON */ + + /* + * install signal handlers + */ + sigIntAction.sa_handler = sigIntHandler; + sigaction (SIGINT, &sigIntAction, NULL); + + sigTermAction.sa_handler = sigTermHandler; + sigaction (SIGTERM, &sigTermAction, NULL); /* * run the actual loops */ -#ifdef HAVE_LIBRRD +#if HAVE_LIBRRD if (operating_mode == MODE_SERVER) start_server (); - else /* if (operating_mode == MODE_CLIENT || operating_mode == MODE_LOCAL) */ + else /* if (operating_mode == MODE_CLIENT || operating_mode == MODE_LOCAL || operating_mode == MODE_LOG) */ #endif start_client (); - /* - * close syslog - */ +#if COLLECT_DEBUG + if (logfile != NULL) + DBG_STOPFILE("debug file closed."); +#endif + + /* close syslog */ syslog (LOG_INFO, "Exiting normally"); closelog (); +#if COLLECT_DAEMON if (daemonize) - pidfile_remove(); + pidfile_remove (pidfile); +#endif /* COLLECT_DAEMON */ return (0); -} +} /* int main (int argc, char **argv) */