X-Git-Url: https://git.octo.it/?a=blobdiff_plain;f=src%2Fcollectd.c;h=576abef48c7110575d3a7452008c4a28970db3c4;hb=4027126a75439c7d2b1d5572092fd291ef19ead6;hp=64da57650a2627774e3cf3e78913b0f20810c6ee;hpb=9513f4000aea72580d96968214f18e059ad12eb9;p=collectd.git diff --git a/src/collectd.c b/src/collectd.c index 64da5765..576abef4 100644 --- a/src/collectd.c +++ b/src/collectd.c @@ -1,11 +1,10 @@ /** * collectd - src/collectd.c - * Copyright (C) 2005,2006 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 @@ -23,62 +22,192 @@ #include "collectd.h" #include "common.h" -#include "utils_debug.h" -#include "network.h" +#include +#include +#include + +#include + #include "plugin.h" #include "configfile.h" -static int loop = 0; +#if HAVE_STATGRAB_H +# include +#endif +/* + * Global variables + */ +char hostname_g[DATA_MAX_NAME_LEN]; +int interval_g; #if HAVE_LIBKSTAT kstat_ctl_t *kc; #endif /* HAVE_LIBKSTAT */ -/* - * exported variables - */ -time_t curtime; -int operating_mode; +static int loop = 0; + +static void *do_flush (void __attribute__((unused)) *arg) +{ + INFO ("Flushing all data."); + plugin_flush (NULL, -1, NULL); + INFO ("Finished flushing all data."); + pthread_exit (NULL); + return NULL; +} -static void sigIntHandler (int signal) +static void sig_int_handler (int __attribute__((unused)) signal) { loop++; } -static void sigTermHandler (int signal) +static void sig_term_handler (int __attribute__((unused)) signal) { loop++; } -static int change_basedir (char *dir) +static void sig_usr1_handler (int __attribute__((unused)) signal) +{ + pthread_t thread; + pthread_attr_t attr; + + /* flushing the data might take a while, + * so it should be done asynchronously */ + pthread_attr_init (&attr); + pthread_attr_setdetachstate (&attr, PTHREAD_CREATE_DETACHED); + pthread_create (&thread, &attr, do_flush, NULL); +} + +static int init_hostname (void) +{ + const char *str; + + struct addrinfo ai_hints; + struct addrinfo *ai_list; + struct addrinfo *ai_ptr; + int status; + + str = global_option_get ("Hostname"); + if (str != NULL) + { + sstrncpy (hostname_g, str, sizeof (hostname_g)); + return (0); + } + + if (gethostname (hostname_g, sizeof (hostname_g)) != 0) + { + fprintf (stderr, "`gethostname' failed and no " + "hostname was configured.\n"); + return (-1); + } + + str = global_option_get ("FQDNLookup"); + if ((strcasecmp ("false", str) == 0) + || (strcasecmp ("no", str) == 0) + || (strcasecmp ("off", str) == 0)) + return (0); + + memset (&ai_hints, '\0', sizeof (ai_hints)); + ai_hints.ai_flags = AI_CANONNAME; + + status = getaddrinfo (hostname_g, NULL, &ai_hints, &ai_list); + if (status != 0) + { + ERROR ("Looking up \"%s\" failed. You have set the " + "\"FQDNLookup\" option, but I cannot resolve " + "my hostname to a fully qualified domain " + "name. Please fix you network " + "configuration.", hostname_g); + return (-1); + } + + for (ai_ptr = ai_list; ai_ptr != NULL; ai_ptr = ai_ptr->ai_next) + { + if (ai_ptr->ai_canonname == NULL) + continue; + + sstrncpy (hostname_g, ai_ptr->ai_canonname, sizeof (hostname_g)); + break; + } + + freeaddrinfo (ai_list); + return (0); +} /* int init_hostname */ + +static int init_global_variables (void) +{ + const char *str; + + 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); + + if (init_hostname () != 0) + return (-1); + DEBUG ("hostname_g = %s;", hostname_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); } } @@ -92,7 +221,7 @@ 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 { @@ -100,11 +229,11 @@ static 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 */ } @@ -115,7 +244,7 @@ static void update_kstat (void) /* TODO * Remove all settings but `-f' and `-C' */ -static void exit_usage (char *name) +static void exit_usage (int status) { printf ("Usage: "PACKAGE" [OPTIONS]\n\n" @@ -123,38 +252,26 @@ static void exit_usage (char *name) " General:\n" " -C Configuration file.\n" " Default: "CONFIGFILE"\n" + " -t Test config and exit.\n" + " -T Test plugin read and exit.\n" " -P PID-file.\n" " Default: "PIDFILE"\n" #if COLLECT_DAEMON " -f Don't fork to the background.\n" #endif + " -h Display help (this message)\n" "\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"); - exit (0); -} /* static void exit_usage (char *name) */ + exit (status); +} /* static void exit_usage (int status) */ -static int start_client (void) +static int do_init (void) { - int step; - - struct timeval tv_now; - struct timeval tv_next; - struct timespec ts_wait; - - step = atoi (COLLECTD_STEP); - if (step <= 0) - step = 10; - #if HAVE_LIBKSTAT kc = NULL; update_kstat (); @@ -163,98 +280,103 @@ static int start_client (void) #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 plugin_init_all (); + return (0); +} /* int do_init () */ + + +static int do_loop (void) +{ + struct timeval tv_now; + struct timeval tv_next; + struct timeval tv_wait; + struct timespec ts_wait; + while (loop == 0) { if (gettimeofday (&tv_next, NULL) < 0) { - syslog (LOG_ERR, "gettimeofday failed: %s", strerror (errno)); + char errbuf[1024]; + ERROR ("gettimeofday failed: %s", + sstrerror (errno, errbuf, + sizeof (errbuf))); return (-1); } - tv_next.tv_sec += step; + tv_next.tv_sec += interval_g; #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 (&loop); + plugin_read_all (); if (gettimeofday (&tv_now, NULL) < 0) { - syslog (LOG_ERR, "gettimeofday failed: %s", strerror (errno)); + char errbuf[1024]; + ERROR ("gettimeofday failed: %s", + sstrerror (errno, errbuf, + sizeof (errbuf))); return (-1); } - if (timeval_sub_timespec (&tv_next, &tv_now, &ts_wait) != 0) + if (timeval_cmp (tv_next, tv_now, &tv_wait) <= 0) { - syslog (LOG_WARNING, "Not sleeping because `timeval_sub_timespec' returned non-zero!"); + WARNING ("Not sleeping because the next interval is " + "%i.%06i seconds in the past!", + (int) tv_wait.tv_sec, (int) tv_wait.tv_usec); continue; } + ts_wait.tv_sec = tv_wait.tv_sec; + ts_wait.tv_nsec = (long) (1000 * tv_wait.tv_usec); + while ((loop == 0) && (nanosleep (&ts_wait, &ts_wait) == -1)) { if (errno != EINTR) { - syslog (LOG_ERR, "nanosleep failed: %s", strerror (errno)); - break; + char errbuf[1024]; + ERROR ("nanosleep failed: %s", + sstrerror (errno, errbuf, + sizeof (errbuf))); + return (-1); } } - } + } /* while (loop == 0) */ + DEBUG ("return (0);"); return (0); -} /* static int start_client (void) */ +} /* int do_loop */ -#if HAVE_LIBRRD -static int start_server (void) +static int do_shutdown (void) { - /* FIXME use stack here! */ - char *host; - char *type; - char *instance; - char *values; - - while (loop == 0) - { - if (network_receive (&host, &type, &instance, &values) == 0) - plugin_write (host, type, instance, values); - - 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; - } - + plugin_shutdown_all (); return (0); -} /* static int start_server (void) */ -#endif /* HAVE_LIBRRD */ +} /* int do_shutdown */ #if COLLECT_DAEMON -static int pidfile_create (const char *file) +static int pidfile_create (void) { FILE *fh; - - if (file == NULL) - file = PIDFILE; + const char *file = global_option_get ("PIDFile"); if ((fh = fopen (file, "w")) == NULL) { - syslog (LOG_ERR, "fopen (%s): %s", file, strerror (errno)); + char errbuf[1024]; + ERROR ("fopen (%s): %s", file, + sstrerror (errno, errbuf, sizeof (errbuf))); return (1); } @@ -263,49 +385,39 @@ static int pidfile_create (const char *file) return (0); } /* static int pidfile_create (const char *file) */ -#endif /* COLLECT_DAEMON */ -#if COLLECT_DAEMON -static int pidfile_remove (const char *file) +static int pidfile_remove (void) { - if (file == NULL) { - file = 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; - struct sigaction sigTermAction; - char *datadir = PKGLOCALSTATEDIR; + struct sigaction sig_int_action; + struct sigaction sig_term_action; + struct sigaction sig_usr1_action; + struct sigaction sig_pipe_action; char *configfile = CONFIGFILE; + int test_config = 0; + int test_readall = 0; + const char *basedir; #if COLLECT_DAEMON - struct sigaction sigChldAction; - char *pidfile = NULL; + struct sigaction sig_chld_action; pid_t pid; int daemonize = 1; #endif -#if COLLECT_DEBUG - char *logfile = LOGFILE; -#endif - -#if HAVE_LIBRRD - operating_mode = MODE_LOCAL; -#else - operating_mode = MODE_CLIENT; -#endif - - /* open syslog */ - openlog (PACKAGE, LOG_CONS | LOG_PID, LOG_DAEMON); + int exit_status = 0; /* read options */ while (1) { int c; - c = getopt (argc, argv, "hC:" + c = getopt (argc, argv, "htTC:" #if COLLECT_DAEMON "fP:" #endif @@ -319,24 +431,34 @@ int main (int argc, char **argv) case 'C': configfile = optarg; break; + case 't': + test_config = 1; + break; + case 'T': + test_readall = 1; + global_option_set ("ReadThreads", "-1"); +#if COLLECT_DAEMON + daemonize = 0; +#endif /* COLLECT_DAEMON */ + break; #if COLLECT_DAEMON case 'P': - pidfile = optarg; + global_option_set ("PIDFile", optarg); break; case 'f': daemonize = 0; break; #endif /* COLLECT_DAEMON */ case 'h': + exit_usage (0); + break; default: - exit_usage (argv[0]); + exit_usage (1); } /* switch (c) */ } /* while (1) */ -#if COLLECT_DEBUG - if ((logfile = cf_get_option ("LogFile", LOGFILE)) != NULL) - DBG_STARTFILE (logfile, "Debug file opened."); -#endif + if (optind < argc) + exit_usage (1); /* * Read options from the config file, the environment and the command @@ -355,37 +477,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 ((datadir = cf_get_option ("DataDir", PKGLOCALSTATEDIR)) == NULL) + if ((basedir = global_option_get ("BaseDir")) == NULL) { - fprintf (stderr, "Don't have a datadir to use. This should not happen. Ever."); + fprintf (stderr, "Don't have a basedir to use. This should not happen. Ever."); return (1); } - if (change_basedir (datadir)) + 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); } + /* + * 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. + */ + if (init_global_variables () != 0) + return (1); + + if (test_config) + return (0); + #if COLLECT_DAEMON /* * fork off child */ - sigChldAction.sa_handler = SIG_IGN; - sigaction (SIGCHLD, &sigChldAction, NULL); - - if ((pidfile == NULL) - && ((pidfile = cf_get_option ("PIDFile", PIDFILE)) == NULL)) - { - fprintf (stderr, "Cannot obtain pidfile. This shoud not happen. Ever."); - return (1); - } + memset (&sig_chld_action, '\0', sizeof (sig_chld_action)); + sig_chld_action.sa_handler = SIG_IGN; + sigaction (SIGCHLD, &sig_chld_action, 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) @@ -399,7 +530,7 @@ int main (int argc, char **argv) setsid (); /* Write pidfile */ - if (pidfile_create (pidfile)) + if (pidfile_create ()) exit (2); /* close standard descriptors */ @@ -409,54 +540,81 @@ 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 */ + memset (&sig_pipe_action, '\0', sizeof (sig_pipe_action)); + sig_pipe_action.sa_handler = SIG_IGN; + sigaction (SIGPIPE, &sig_pipe_action, NULL); + /* * install signal handlers */ - sigIntAction.sa_handler = sigIntHandler; - sigaction (SIGINT, &sigIntAction, NULL); + memset (&sig_int_action, '\0', sizeof (sig_int_action)); + sig_int_action.sa_handler = sig_int_handler; + if (0 != sigaction (SIGINT, &sig_int_action, NULL)) { + char errbuf[1024]; + ERROR ("Error: Failed to install a signal handler for signal INT: %s", + sstrerror (errno, errbuf, sizeof (errbuf))); + return (1); + } + + memset (&sig_term_action, '\0', sizeof (sig_term_action)); + sig_term_action.sa_handler = sig_term_handler; + if (0 != sigaction (SIGTERM, &sig_term_action, NULL)) { + char errbuf[1024]; + ERROR ("Error: Failed to install a signal handler for signal TERM: %s", + sstrerror (errno, errbuf, sizeof (errbuf))); + return (1); + } - sigTermAction.sa_handler = sigTermHandler; - sigaction (SIGTERM, &sigTermAction, NULL); + memset (&sig_usr1_action, '\0', sizeof (sig_usr1_action)); + sig_usr1_action.sa_handler = sig_usr1_handler; + if (0 != sigaction (SIGUSR1, &sig_usr1_action, NULL)) { + char errbuf[1024]; + ERROR ("Error: Failed to install a signal handler for signal USR1: %s", + sstrerror (errno, errbuf, sizeof (errbuf))); + return (1); + } /* * run the actual loops */ -#if HAVE_LIBRRD - if (operating_mode == MODE_SERVER) - start_server (); - else /* if (operating_mode == MODE_CLIENT || operating_mode == MODE_LOCAL || operating_mode == MODE_LOG) */ -#endif - start_client (); + do_init (); -#if COLLECT_DEBUG - if (logfile != NULL) - DBG_STOPFILE("debug file closed."); -#endif + if (test_readall) + { + if (plugin_read_all_once () != 0) + exit_status = 1; + } + else + { + INFO ("Initialization complete, entering read-loop."); + do_loop (); + } /* close syslog */ - syslog (LOG_INFO, "Exiting normally"); - closelog (); + INFO ("Exiting normally."); + + do_shutdown (); #if COLLECT_DAEMON if (daemonize) - pidfile_remove (pidfile); + pidfile_remove (); #endif /* COLLECT_DAEMON */ - return (0); -} /* int main (int argc, char **argv) */ + return (exit_status); +} /* int main */