projects / collectd.git / blobdiff
? search:
summary | shortlog | log | commit | commitdiff | tree
raw | inline | side by side
Merged branch configfile to trunk
[collectd.git] / src / collectd.c
diff --git a/src/collectd.c b/src/collectd.c
index af6a93e..c86a678 100644 (file)
--- a/src/collectd.c
+++ b/src/collectd.c
@@ -21,7 +21,8 @@
  *   Alvaro Barcellos <alvaro.barcellos at gmail.com>
  **/
 
-#include "collectd.h"
+#include "common.h"
+#include "utils_debug.h"
 
 #include "multicast.h"
 #include "plugin.h"
@@ -31,7 +32,7 @@
 
 static int loop = 0;
 
-#ifdef HAVE_LIBKSTAT
+#if HAVE_LIBKSTAT
 kstat_ctl_t *kc;
 #endif /* HAVE_LIBKSTAT */
 
@@ -45,16 +46,16 @@ int   num_pinghosts = 0;
  */
 time_t curtime;
 
-#ifdef HAVE_LIBRRD
+#if HAVE_LIBRRD
 int operating_mode;
 #endif
 
-void sigIntHandler (int signal)
+static void sigIntHandler (int signal)
 {
        loop++;
 }
 
-int change_basedir (char *dir)
+static int change_basedir (char *dir)
 {
        int dirlen = strlen (dir);
        
@@ -87,10 +88,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)
        {
@@ -112,25 +113,33 @@ void update_kstat (void)
        }
 
        return;
-}
+} /* static void update_kstat (void) */
 #endif /* HAVE_LIBKSTAT */
 
-void exit_usage (char *name)
+static void exit_usage (char *name)
 {
-       printf ("Usage: %s [OPTIONS]\n\n"
+       printf ("Usage: "PACKAGE" [OPTIONS]\n\n"
                        
                        "Available options:\n"
                        "  General:\n"
-                       "    -C <dir>        Configuration file.\n"
-                       "                    Default: %s\n"
-                       "    -P <file>       PID File.\n"
-                       "                    Default: %s\n"
+                       "    -C <file>       Configuration file.\n"
+                       "                    Default: "CONFIGFILE"\n"
+#if COLLECT_DAEMON
+                       "    -P <file>       PID file.\n"
+                       "                    Default: "PIDFILE"\n"
+#endif
                        "    -M <dir>        Module/Plugin directory.\n"
-                       "                    Default: %s\n"
+                       "                    Default: "PLUGINDIR"\n"
                        "    -D <dir>        Data storage directory.\n"
-                       "                    Default: %s\n"
+                       "                    Default: "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"
-#ifdef HAVE_LIBRRD
+#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"
@@ -140,23 +149,22 @@ void exit_usage (char *name)
                        "    -p <host>       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"
+                       "\n"PACKAGE" "VERSION", http://verplant.org/collectd/\n"
                        "by Florian octo Forster <octo@verplant.org>\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 start_client (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 ()));
@@ -175,7 +183,7 @@ int start_client (void)
        while (loop == 0)
        {
                curtime = time (NULL);
-#ifdef HAVE_LIBKSTAT
+#if HAVE_LIBKSTAT
                update_kstat ();
 #endif
                plugin_read_all ();
@@ -190,10 +198,10 @@ int start_client (void)
        }
 
        return (0);
-}
+} /* static int start_client (void) */
 
-#ifdef HAVE_LIBRRD
-int start_server (void)
+#if HAVE_LIBRRD
+static int start_server (void)
 {
        char *host;
        char *type;
@@ -212,10 +220,11 @@ int start_server (void)
        }
        
        return (0);
-}
+} /* static int start_server (void) */
 #endif /* HAVE_LIBRRD */
 
-int pidfile_create (char *file)
+#if COLLECT_DAEMON
+static int pidfile_create (const char *file)
 {
        FILE *fh;
 
@@ -232,31 +241,38 @@ int pidfile_create (char *file)
        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;
+       char *configfile = CONFIGFILE;
+       char *plugindir  = PLUGINDIR;
+       char *datadir    = PKGLOCALSTATEDIR;
 #if COLLECT_DAEMON
+       char *pidfile    = PIDFILE;
        pid_t pid;
+       int daemonize    = 1;
+#endif
+#if COLLECT_DEBUG
+       char *logfile    = LOGFILE;
 #endif
 
-       char *configfile = CONFIGFILE;
-       char *pidfile    = PIDFILE;
-       char *plugindir  = PLUGINDIR;
-       char *datadir    = PKGLOCALSTATEDIR;
-
-       int daemonize = 1;
-
-#ifdef HAVE_LIBRRD
+#if HAVE_LIBRRD
        operating_mode = MODE_LOCAL;
 #endif
-       
+
        /* open syslog */
        openlog (PACKAGE, LOG_CONS | LOG_PID, LOG_DAEMON);
 
@@ -265,7 +281,13 @@ int main (int argc, char **argv)
        {
                int c;
 
-               c = getopt (argc, argv, "C:P:M:D:fh"
+               c = getopt (argc, argv, "C:M:D:h"
+#if COLLECT_DAEMON
+                               "fP:"
+#endif
+#if COLLECT_DEBUG
+                               "L:"
+#endif
 #if HAVE_LIBRRD
                                "csl"
 #endif /* HAVE_LIBRRD */
@@ -279,7 +301,7 @@ int main (int argc, char **argv)
 
                switch (c)
                {
-#ifdef HAVE_LIBRRD
+#if HAVE_LIBRRD
                        case 'c':
                                operating_mode = MODE_CLIENT;
                                break;
@@ -295,18 +317,25 @@ int main (int argc, char **argv)
                        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;
-                       case 'f':
-                               daemonize = 0;
+#if COLLECT_DEBUG
+                       case 'L':
+                               logfile = optarg;
                                break;
+#endif
 #if COLLECT_PING
                        case 'p':
                                if (num_pinghosts < MAX_PINGHOSTS)
@@ -318,9 +347,10 @@ int main (int argc, char **argv)
                        case 'h':
                        default:
                                exit_usage (argv[0]);
-               }
-                               
-       }
+               } /* switch (c) */
+       } /* while (1) */
+
+       DBG_STARTFILE(logfile, "debug file opened.");
 
        /*
         * Read the config file. This will load any modules automagically.
@@ -405,19 +435,23 @@ int main (int argc, char **argv)
        /*
         * 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) */
 #endif
                start_client ();
 
+       DBG_STOPFILE("debug file closed.");
+
        /* 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) */
collectd, the system statistics collection daemon