X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Fcollectd.c;h=4e18fd6e2280edcce9b1190777c846bdfa11dd3f;hb=feaebecc0da2199b4dbbc2d8f0aa2c712b91e835;hp=8e92bfbc857b77e3eb3c058c6e817300662ab245;hpb=cc0d31d626faaedad019aa9a68bea64cbeefc7e2;p=collectd.git diff --git a/src/collectd.c b/src/collectd.c index 8e92bfbc..4e18fd6e 100644 --- a/src/collectd.c +++ b/src/collectd.c @@ -1,11 +1,10 @@ /** * collectd - src/collectd.c - * Copyright (C) 2005 Florian octo Forster + * Copyright (C) 2005-2007 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. + * Free Software Foundation; only version 2 of the License is applicable. * * This program is distributed in the hope that it will be useful, but * WITHOUT ANY WARRANTY; without even the implied warranty of @@ -22,80 +21,133 @@ **/ #include "collectd.h" +#include "common.h" -#include "multicast.h" #include "plugin.h" #include "configfile.h" +#include "types_list.h" -#include "ping.h" - -static int loop = 0; - -#ifdef HAVE_LIBKSTAT +/* + * Global variables + */ +char hostname_g[DATA_MAX_NAME_LEN]; +int interval_g; +#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; +static int loop = 0; -#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 init_global_variables (void) +{ + const char *str; + + str = global_option_get ("Hostname"); + if (str != NULL) + { + strncpy (hostname_g, str, sizeof (hostname_g)); + } + else + { + if (gethostname (hostname_g, sizeof (hostname_g)) != 0) + { + fprintf (stderr, "`gethostname' failed and no " + "hostname was configured.\n"); + return (-1); + } + } + DEBUG ("hostname_g = %s;", hostname_g); + + str = global_option_get ("Interval"); + if (str == NULL) + str = "10"; + interval_g = atoi (str); + if (interval_g <= 0) + { + fprintf (stderr, "Cannot set the interval to a correct value.\n" + "Please check your settings.\n"); + return (-1); + } + DEBUG ("interval_g = %i;", interval_g); + + return (0); +} /* int init_global_variables */ + +static int change_basedir (const char *orig_dir) { - int dirlen = strlen (dir); + char *dir = strdup (orig_dir); + int dirlen; + int status; + + if (dir == NULL) + { + char errbuf[1024]; + ERROR ("strdup failed: %s", + sstrerror (errno, errbuf, sizeof (errbuf))); + return (-1); + } + dirlen = strlen (dir); while ((dirlen > 0) && (dir[dirlen - 1] == '/')) dir[--dirlen] = '\0'; if (dirlen <= 0) return (-1); - if (chdir (dir) == -1) + status = chdir (dir); + free (dir); + + if (status != 0) { if (errno == ENOENT) { - if (mkdir (dir, 0755) == -1) + if (mkdir (orig_dir, 0755) == -1) { - syslog (LOG_ERR, "mkdir (%s): %s", dir, strerror (errno)); + char errbuf[1024]; + ERROR ("change_basedir: mkdir (%s): %s", orig_dir, + sstrerror (errno, errbuf, + sizeof (errbuf))); return (-1); } - else if (chdir (dir) == -1) + else if (chdir (orig_dir) == -1) { - syslog (LOG_ERR, "chdir (%s): %s", dir, strerror (errno)); + char errbuf[1024]; + ERROR ("chdir (%s): %s", orig_dir, + sstrerror (errno, errbuf, + sizeof (errbuf))); return (-1); } } else { - syslog (LOG_ERR, "chdir: %s", strerror (errno)); + char errbuf[1024]; + ERROR ("chdir (%s): %s", orig_dir, + sstrerror (errno, errbuf, + sizeof (errbuf))); return (-1); } } 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) { if ((kc = kstat_open ()) == NULL) - syslog (LOG_ERR, "Unable to open kstat control structure"); + ERROR ("Unable to open kstat control structure"); } else { @@ -103,172 +155,188 @@ void update_kstat (void) kid = kstat_chain_update (kc); if (kid > 0) { - syslog (LOG_INFO, "kstat chain has been updated"); + INFO ("kstat chain has been updated"); plugin_init_all (); } else if (kid < 0) - syslog (LOG_ERR, "kstat chain update failed"); + ERROR ("kstat chain update failed"); /* else: everything works as expected */ } 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" - " -C Configuration file.\n" - " Default: %s\n" - " -P PID File.\n" - " Default: %s\n" - " -M Module/Plugin directory.\n" - " Default: %s\n" - " -D Data storage directory.\n" - " Default: %s\n" + " -C Configuration file.\n" + " Default: "CONFIGFILE"\n" + " -P PID-file.\n" + " Default: "PIDFILE"\n" +#if COLLECT_DAEMON " -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" +#endif + "\nBuiltin defaults:\n" + " Config-File "CONFIGFILE"\n" + " PID-File "PIDFILE"\n" + " Data-Directory "PKGLOCALSTATEDIR"\n" + "\n"PACKAGE" "VERSION", http://collectd.org/\n" "by Florian octo Forster \n" - "for contributions see `AUTHORS'\n", - PACKAGE, CONFIGFILE, PIDFILE, PLUGINDIR, PKGLOCALSTATEDIR, PACKAGE, VERSION); + "for contributions see `AUTHORS'\n"); exit (0); -} +} /* static void exit_usage (char *name) */ -int start_client (void) +static int do_init (void) { - int sleepingtime; - -#ifdef HAVE_LIBKSTAT +#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 ())); + ERROR ("sg_init: %s", sg_str_error (sg_get_error ())); return (-1); } if (sg_drop_privileges ()) { - syslog (LOG_ERR, "sg_drop_privileges: %s", sg_str_error (sg_get_error ())); + ERROR ("sg_drop_privileges: %s", sg_str_error (sg_get_error ())); return (-1); } #endif + read_types_list (); plugin_init_all (); + return (0); +} /* int do_init () */ + + +static int do_loop (void) +{ + struct timeval tv_now; + struct timeval tv_next; + struct timespec ts_wait; + while (loop == 0) { - curtime = time (NULL); -#ifdef HAVE_LIBKSTAT + if (gettimeofday (&tv_next, NULL) < 0) + { + char errbuf[1024]; + ERROR ("gettimeofday failed: %s", + sstrerror (errno, errbuf, + sizeof (errbuf))); + return (-1); + } + tv_next.tv_sec += interval_g; + +#if HAVE_LIBKSTAT update_kstat (); #endif - plugin_read_all (); - sleepingtime = 10; - while (sleepingtime != 0) + /* Issue all plugins */ + plugin_read_all (&loop); + + if (gettimeofday (&tv_now, NULL) < 0) { - if (loop != 0) - break; - sleepingtime = sleep (sleepingtime); + char errbuf[1024]; + ERROR ("gettimeofday failed: %s", + sstrerror (errno, errbuf, + sizeof (errbuf))); + return (-1); } - } - return (0); -} + if (timeval_sub_timespec (&tv_next, &tv_now, &ts_wait) != 0) + { + WARNING ("Not sleeping because " + "`timeval_sub_timespec' returned " + "non-zero!"); + continue; + } -#ifdef HAVE_LIBRRD -int start_server (void) -{ - char *host; - char *type; - char *instance; - char *values; + while ((loop == 0) && (nanosleep (&ts_wait, &ts_wait) == -1)) + { + if (errno != EINTR) + { + char errbuf[1024]; + ERROR ("nanosleep failed: %s", + sstrerror (errno, errbuf, + sizeof (errbuf))); + return (-1); + } + } + } /* while (loop == 0) */ - while (loop == 0) - { - if (multicast_receive (&host, &type, &instance, &values) == 0) - plugin_write (host, type, instance, values); + DEBUG ("return (0);"); + return (0); +} /* int do_loop */ - if (host != NULL) free (host); host = NULL; - if (type != NULL) free (type); type = NULL; - if (instance != NULL) free (instance); instance = NULL; - if (values != NULL) free (values); values = NULL; - } - +static int do_shutdown (void) +{ + plugin_shutdown_all (); return (0); -} -#endif /* HAVE_LIBRRD */ +} /* int do_shutdown */ -int pidfile_create (void) +#if COLLECT_DAEMON +static int pidfile_create (void) { FILE *fh; + const char *file = global_option_get ("PIDFile"); - if ((fh = fopen (PIDFILE, "w")) == NULL) + if ((fh = fopen (file, "w")) == NULL) { - syslog (LOG_ERR, "fopen (pidfile): %s", strerror (errno)); + char errbuf[1024]; + ERROR ("fopen (%s): %s", file, + sstrerror (errno, errbuf, sizeof (errbuf))); return (1); } - fprintf (fh, "%d\n", getpid()); + fprintf (fh, "%i\n", (int) getpid ()); fclose(fh); return (0); -} +} /* static int pidfile_create (const char *file) */ -int pidfile_remove (void) +static int pidfile_remove (void) { - return (unlink (PIDFILE)); -} + const char *file = global_option_get ("PIDFile"); + + DEBUG ("unlink (%s)", (file != NULL) ? file : ""); + 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 *configfile = CONFIGFILE; + int test_config = 0; + const char *basedir; #if COLLECT_DAEMON + struct sigaction sigChldAction; pid_t pid; + int daemonize = 1; #endif - char *configfile = CONFIGFILE; - char *pidfile = PIDFILE; - char *plugindir = PLUGINDIR; - char *datadir = PKGLOCALSTATEDIR; - - int daemonize = 1; - -#ifdef HAVE_LIBRRD - operating_mode = MODE_LOCAL; -#endif - - /* open syslog */ - openlog (PACKAGE, LOG_CONS | LOG_PID, LOG_DAEMON); - /* read options */ while (1) { int c; - c = getopt (argc, argv, "C:P:M:D:fh" -#if HAVE_LIBRRD - "csl" -#endif /* HAVE_LIBRRD */ -#if COLLECT_PING - "p:" -#endif /* COLLECT_PING */ + c = getopt (argc, argv, "htC:" +#if COLLECT_DAEMON + "fP:" +#endif ); if (c == -1) @@ -276,54 +344,32 @@ 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 'C': configfile = optarg; break; - case 'P': - pidfile = optarg; + case 't': + test_config = 1; break; - case 'M': - plugindir = optarg; - break; - case 'D': - datadir = optarg; +#if COLLECT_DAEMON + case 'P': + global_option_set ("PIDFile", optarg); break; 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 */ +#endif /* COLLECT_DAEMON */ case 'h': default: exit_usage (argv[0]); - } - - } + } /* switch (c) */ + } /* while (1) */ /* - * Read the config file. This will load any modules automagically. + * 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. */ - plugin_set_dir (plugindir); - if (cf_read (configfile)) { fprintf (stderr, "Error: Reading the config file failed!\n" @@ -335,31 +381,46 @@ int main (int argc, char **argv) * Change directory. We do this _after_ reading the config and loading * modules to relative paths work as expected. */ - if (change_basedir (datadir)) + if ((basedir = global_option_get ("BaseDir")) == NULL) + { + fprintf (stderr, "Don't have a basedir to use. This should not happen. Ever."); + return (1); + } + else if (change_basedir (basedir)) { - fprintf (stderr, "Error: Unable to change to directory `%s'.\n", datadir); + fprintf (stderr, "Error: Unable to change to directory `%s'.\n", basedir); return (1); } /* - * install signal handlers + * Set global variables or, if that failes, exit. We cannot run with + * them being uninitialized. If nothing is configured, then defaults + * are being used. So this means that the user has actually done + * something wrong. */ - sigIntAction.sa_handler = sigIntHandler; - sigaction (SIGINT, &sigIntAction, NULL); + if (init_global_variables () != 0) + return (1); - sigChldAction.sa_handler = SIG_IGN; - sigaction (SIGCHLD, &sigChldAction, NULL); + if (test_config) + return (0); +#if COLLECT_DAEMON /* * fork off child */ -#if COLLECT_DAEMON + memset (&sigChldAction, '\0', sizeof (sigChldAction)); + sigChldAction.sa_handler = SIG_IGN; + sigaction (SIGCHLD, &sigChldAction, NULL); + if (daemonize) { if ((pid = fork ()) == -1) { /* error */ - fprintf (stderr, "fork: %s", strerror (errno)); + char errbuf[1024]; + fprintf (stderr, "fork: %s", + sstrerror (errno, errbuf, + sizeof (errbuf))); return (1); } else if (pid != 0) @@ -383,38 +444,48 @@ int main (int argc, char **argv) if (open ("/dev/null", O_RDWR) != 0) { - syslog (LOG_ERR, "Error: Could not connect `STDIN' to `/dev/null'"); + ERROR ("Error: Could not connect `STDIN' to `/dev/null'"); return (1); } if (dup (0) != 1) { - syslog (LOG_ERR, "Error: Could not connect `STDOUT' to `/dev/null'"); + ERROR ("Error: Could not connect `STDOUT' to `/dev/null'"); return (1); } if (dup (0) != 2) { - syslog (LOG_ERR, "Error: Could not connect `STDERR' to `/dev/null'"); + ERROR ("Error: Could not connect `STDERR' to `/dev/null'"); return (1); } } /* if (daemonize) */ #endif /* COLLECT_DAEMON */ /* + * install signal handlers + */ + memset (&sigIntAction, '\0', sizeof (sigIntAction)); + sigIntAction.sa_handler = sigIntHandler; + sigaction (SIGINT, &sigIntAction, NULL); + + memset (&sigTermAction, '\0', sizeof (sigTermAction)); + sigTermAction.sa_handler = sigTermHandler; + sigaction (SIGTERM, &sigTermAction, NULL); + + /* * run the actual loops */ -#ifdef HAVE_LIBRRD - if (operating_mode == MODE_SERVER) - start_server (); - else /* if (operating_mode == MODE_CLIENT || operating_mode == MODE_LOCAL) */ -#endif - start_client (); + do_init (); + do_loop (); /* close syslog */ - syslog (LOG_INFO, "Exiting normally"); - closelog (); + INFO ("Exiting normally"); + + do_shutdown (); +#if COLLECT_DAEMON if (daemonize) - pidfile_remove(); + pidfile_remove (); +#endif /* COLLECT_DAEMON */ return (0); -} +} /* int main */