Corrected many defines, moved log-mode functionality out of the `rrd_*' functions...
[collectd.git] / src / collectd.c
index c86a678..7b264e8 100644 (file)
  *   Alvaro Barcellos <alvaro.barcellos at gmail.com>
  **/
 
+#include "collectd.h"
 #include "common.h"
 #include "utils_debug.h"
 
-#include "multicast.h"
+#include "network.h"
 #include "plugin.h"
 #include "configfile.h"
 
@@ -36,25 +37,22 @@ static int loop = 0;
 kstat_ctl_t *kc;
 #endif /* HAVE_LIBKSTAT */
 
-#if COLLECT_PING
-char *pinghosts[MAX_PINGHOSTS];
-int   num_pinghosts = 0;
-#endif
-
 /*
  * exported variables
  */
 time_t curtime;
-
-#if HAVE_LIBRRD
-int operating_mode;
-#endif
+int    operating_mode;
 
 static void sigIntHandler (int signal)
 {
        loop++;
 }
 
+static void sigTermHandler (int signal)
+{
+       loop++;
+}
+
 static int change_basedir (char *dir)
 {
        int dirlen = strlen (dir);
@@ -116,6 +114,9 @@ static void update_kstat (void)
 } /* static void update_kstat (void) */
 #endif /* HAVE_LIBKSTAT */
 
+/* TODO
+ * Remove all settings but `-f' and `-C'
+ */
 static void exit_usage (char *name)
 {
        printf ("Usage: "PACKAGE" [OPTIONS]\n\n"
@@ -125,30 +126,15 @@ static void exit_usage (char *name)
                        "    -C <file>       Configuration file.\n"
                        "                    Default: "CONFIGFILE"\n"
 #if COLLECT_DAEMON
-                       "    -P <file>       PID file.\n"
-                       "                    Default: "PIDFILE"\n"
+                       "    -f              Don't fork to the background.\n"
 #endif
-                       "    -M <dir>        Module/Plugin directory.\n"
-                       "                    Default: "PLUGINDIR"\n"
-                       "    -D <dir>        Data storage directory.\n"
-                       "                    Default: "PKGLOCALSTATEDIR"\n"
+                       "\nBuiltin defaults:\n"
+                       "  Config-File       "CONFIGFILE"\n"
+                       "  PID-File          "PIDFILE"\n"
+                       "  Data-Directory    "PKGLOCALSTATEDIR"\n"
 #if COLLECT_DEBUG
-                       "    -L <file>       Log file.\n"
-                       "                    Default: "LOGFILE"\n"
-#endif
-#if COLLECT_DAEMON
-                       "    -f              Don't fork to the background.\n"
+                       "  Log-File          "LOGFILE"\n"
 #endif
-#if 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>       Host to ping periodically, may be repeated to ping\n"
-                       "                    more than one host.\n"
-#endif /* COLLECT_PING */
                        "\n"PACKAGE" "VERSION", http://verplant.org/collectd/\n"
                        "by Florian octo Forster <octo@verplant.org>\n"
                        "for contributions see `AUTHORS'\n");
@@ -203,6 +189,7 @@ static int start_client (void)
 #if HAVE_LIBRRD
 static int start_server (void)
 {
+       /* FIXME use stack here! */
        char *host;
        char *type;
        char *instance;
@@ -210,7 +197,7 @@ static 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;
@@ -256,11 +243,12 @@ static int pidfile_remove (const char *file)
 
 int main (int argc, char **argv)
 {
-       struct sigaction sigIntAction, sigChldAction;
-       char *configfile = CONFIGFILE;
-       char *plugindir  = PLUGINDIR;
+       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;
@@ -281,19 +269,10 @@ int main (int argc, char **argv)
        {
                int c;
 
-               c = getopt (argc, argv, "C:M:D:h"
+               c = getopt (argc, argv, "hC:"
 #if COLLECT_DAEMON
-                               "fP:"
-#endif
-#if COLLECT_DEBUG
-                               "L:"
+                               "f"
 #endif
-#if HAVE_LIBRRD
-                               "csl"
-#endif /* HAVE_LIBRRD */
-#if COLLECT_PING
-                               "p:"
-#endif /* COLLECT_PING */
                );
 
                if (c == -1)
@@ -301,62 +280,31 @@ int main (int argc, char **argv)
 
                switch (c)
                {
-#if 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;
 #if COLLECT_DAEMON
-                       case 'P':
-                               pidfile = optarg;
-                               break;
                        case 'f':
                                daemonize = 0;
                                break;
 #endif /* COLLECT_DAEMON */
-                       case 'M':
-                               plugindir = optarg;
-                               break;
-                       case 'D':
-                               datadir = optarg;
-                               break;
-#if COLLECT_DEBUG
-                       case 'L':
-                               logfile = optarg;
-                               break;
-#endif
-#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 'h':
                        default:
                                exit_usage (argv[0]);
                } /* switch (c) */
        } /* while (1) */
 
-       DBG_STARTFILE(logfile, "debug file opened.");
+#if COLLECT_DEBUG
+       if ((logfile = cf_get_option ("LogFile", LOGFILE)) != NULL)
+               DBG_STARTFILE (logfile, "Debug file opened.");
+#endif
 
        /*
-        * 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"
@@ -368,25 +316,30 @@ 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)
+       {
+               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", 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 COLLECT_DAEMON
+       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)
@@ -433,16 +386,28 @@ int main (int argc, char **argv)
 #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
         */
 #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 ();
 
-       DBG_STOPFILE("debug file closed.");
+#if COLLECT_DEBUG
+       if (logfile != NULL)
+               DBG_STOPFILE("debug file closed.");
+#endif
 
        /* close syslog */
        syslog (LOG_INFO, "Exiting normally");
@@ -450,7 +415,7 @@ int main (int argc, char **argv)
 
 #if COLLECT_DAEMON
        if (daemonize)
-               pidfile_remove(pidfile);
+               pidfile_remove (pidfile);
 #endif /* COLLECT_DAEMON */
 
        return (0);