Code

fixes from richard brodie (tracker id 1216576)
[nagiosplug.git] / plugins / check_time.c
index 86c414e2c413853210e7670121f13fb183df4fb4..ff94d9b3a3a94bcb4f0ed763ad645e10149d2110 100644 (file)
@@ -1,50 +1,39 @@
 /******************************************************************************
-*
-* CHECK_TIME.C
-*
-* Program: Network time server plugin for Nagios
-* License: GPL
-* Copyright (c) 1999 Ethan Galstad (nagios@nagios.org)
-* Copyright (c) 2000 Karl DeBisschop (kdebisschop@users.sourceforge.net)
-*
-* $Id$
-*
-* Description:
-*
-* This plugin will attempt to connect to the specified port
-* on the host.  Successul connects return STATE_OK, refusals
-* and timeouts return STATE_CRITICAL, other errors return
-* STATE_UNKNOWN.
-*
-* License Information:
-*
-* 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., 675 Mass Ave, Cambridge, MA 02139, USA.
-*
-*****************************************************************************/
-
-#include "config.h"
+
+ 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., 675 Mass Ave, Cambridge, MA 02139, USA.
+
+ $Id$
+******************************************************************************/
+
+const char *progname = "check_time";
+const char *revision = "$Revision$";
+const char *copyright = "1999-2004";
+const char *email = "nagiosplug-devel@lists.sourceforge.net";
+
 #include "common.h"
 #include "netutils.h"
 #include "utils.h"
 
-#define PROGNAME "check_time"
+enum {
+       TIME_PORT = 37
+};
 
-#define TIME_PORT      37
-#define UNIX_EPOCH     2208988800UL
+#define        UNIX_EPOCH 2208988800UL
 
-unsigned long server_time, raw_server_time;
+uint32_t server_time, raw_server_time;
 time_t diff_time;
 int warning_time = 0;
 int check_warning_time = FALSE;
@@ -56,22 +45,25 @@ unsigned long critical_diff = 0;
 int check_critical_diff = FALSE;
 int server_port = TIME_PORT;
 char *server_address = NULL;
-
+int use_udp = FALSE;
 
 int process_arguments (int, char **);
-int call_getopt (int, char **);
-void print_usage (void);
 void print_help (void);
-
+void print_usage (void);
 
 int
 main (int argc, char **argv)
 {
        int sd;
-       int result;
+       int result = STATE_UNKNOWN;
+       time_t conntime;
+
+       setlocale (LC_ALL, "");
+       bindtextdomain (PACKAGE, LOCALEDIR);
+       textdomain (PACKAGE);
 
-       if (process_arguments (argc, argv) != OK)
-               usage ("Invalid command arguments supplied\n");
+       if (process_arguments (argc, argv) == ERROR)
+               usage4 (_("Could not parse arguments"));
 
        /* initialize alarm signal handling */
        signal (SIGALRM, socket_timeout_alarm_handler);
@@ -81,20 +73,40 @@ main (int argc, char **argv)
        time (&start_time);
 
        /* try to connect to the host at the given port number */
-       if (my_tcp_connect (server_address, server_port, &sd) != STATE_OK) {
+       if (use_udp) {
+               result = my_udp_connect (server_address, server_port, &sd);
+       } else {
+               result = my_tcp_connect (server_address, server_port, &sd);
+       }
+
+       if (result != STATE_OK) {
                if (check_critical_time == TRUE)
                        result = STATE_CRITICAL;
                else if (check_warning_time == TRUE)
                        result = STATE_WARNING;
                else
                        result = STATE_UNKNOWN;
-               terminate (result,
-                                                        "TIME UNKNOWN - could not connect to server %s, port %d\n",
-                                                        server_address, server_port);
+               die (result,
+                          _("TIME UNKNOWN - could not connect to server %s, port %d\n"),
+                          server_address, server_port);
+       }
+
+       if (use_udp) {
+               if (send (sd, "", 0, 0) < 0) {
+                       if (check_critical_time == TRUE)
+                               result = STATE_CRITICAL;
+                       else if (check_warning_time == TRUE)
+                               result = STATE_WARNING;
+                       else
+                               result = STATE_UNKNOWN;
+                       die (result, 
+                         _("TIME UNKNOWN - could not send UDP request to server %s, port %d\n"),
+                         server_address, server_port);
+               }
        }
 
-       /* watch for the FTP connection string */
-       result = recv (sd, &raw_server_time, sizeof (raw_server_time), 0);
+       /* watch for the connection string */
+       result = recv (sd, (void *)&raw_server_time, sizeof (raw_server_time), 0);
 
        /* close the connection */
        close (sd);
@@ -111,49 +123,74 @@ main (int argc, char **argv)
                        result = STATE_WARNING;
                else
                        result = STATE_UNKNOWN;
-               terminate (result,
-                                                        "TIME UNKNOWN - no data on recv() from server %s, port %d\n",
+               die (result,
+                                                        _("TIME UNKNOWN - no data received from server %s, port %d\n"),
                                                         server_address, server_port);
        }
 
        result = STATE_OK;
 
-       if (check_critical_time == TRUE && (end_time - start_time) > critical_time)
+       conntime = (end_time - start_time);
+       if (check_critical_time == TRUE && conntime > critical_time)
                result = STATE_CRITICAL;
-       else if (check_warning_time == TRUE
-                                        && (end_time - start_time) > warning_time) result = STATE_WARNING;
+       else if (check_warning_time == TRUE && conntime > warning_time)
+               result = STATE_WARNING;
 
        if (result != STATE_OK)
-               terminate (result, "TIME %s - %d second response time\n",
-                                                        state_text (result), (int) (end_time - start_time));
+               die (result, _("TIME %s - %d second response time|%s\n"),
+                    state_text (result), (int)conntime,
+                    perfdata ("time", (long)conntime, "s",
+                              check_warning_time, (long)warning_time,
+                              check_critical_time, (long)critical_time,
+                              TRUE, 0, FALSE, 0));
 
        server_time = ntohl (raw_server_time) - UNIX_EPOCH;
-       if (server_time > end_time)
-               diff_time = server_time - end_time;
+       if (server_time > (unsigned long)end_time)
+               diff_time = server_time - (unsigned long)end_time;
        else
-               diff_time = end_time - server_time;
+               diff_time = (unsigned long)end_time - server_time;
 
-       if (check_critical_diff == TRUE && diff_time > critical_diff)
+       if (check_critical_diff == TRUE && diff_time > (time_t)critical_diff)
                result = STATE_CRITICAL;
-       else if (check_warning_diff == TRUE && diff_time > warning_diff)
+       else if (check_warning_diff == TRUE && diff_time > (time_t)warning_diff)
                result = STATE_WARNING;
 
-       printf ("TIME %s - %lu second time difference\n", state_text (result),
-                                       diff_time);
+       printf (_("TIME %s - %lu second time difference|%s %s\n"),
+               state_text (result), diff_time,
+               perfdata ("time", (long)conntime, "s",
+                         check_warning_time, (long)warning_time,
+                         check_critical_time, (long)critical_time,
+                         TRUE, 0, FALSE, 0),
+               perfdata ("offset", (long)diff_time, "s",
+                         check_warning_diff, (long)warning_diff,
+                         check_critical_diff, (long)critical_diff,
+                         TRUE, 0, FALSE, 0));
        return result;
 }
 
 
 
-
-
-
 /* process command-line arguments */
 int
 process_arguments (int argc, char **argv)
 {
        int c;
 
+       int option = 0;
+       static struct option longopts[] = {
+               {"hostname", required_argument, 0, 'H'},
+               {"warning-variance", required_argument, 0, 'w'},
+               {"critical-variance", required_argument, 0, 'c'},
+               {"warning-connect", required_argument, 0, 'W'},
+               {"critical-connect", required_argument, 0, 'C'},
+               {"port", required_argument, 0, 'p'},
+               {"udp", no_argument, 0, 'u'},
+               {"timeout", required_argument, 0, 't'},
+               {"version", no_argument, 0, 'V'},
+               {"help", no_argument, 0, 'h'},
+               {0, 0, 0, 0}
+       };
+
        if (argc < 2)
                usage ("\n");
 
@@ -170,91 +207,25 @@ process_arguments (int argc, char **argv)
                        strcpy (argv[c], "-C");
        }
 
-       c = 0;
-       while ((c += call_getopt (argc - c, &argv[c])) < argc) {
-
-               if (is_option (argv[c]))
-                       continue;
-
-               if (server_address == NULL) {
-                       if (argc > c) {
-                               if (is_host (argv[c]) == FALSE)
-                                       usage ("Invalid host name/address\n");
-                               server_address = argv[c];
-                       }
-                       else {
-                               usage ("Host name was not supplied\n");
-                       }
-               }
-       }
-
-       return OK;
-}
-
-
-
-
-
-int
-call_getopt (int argc, char **argv)
-{
-       int c, i = 0;
-
-#ifdef HAVE_GETOPT_H
-       int option_index = 0;
-       static struct option long_options[] = {
-               {"hostname", required_argument, 0, 'H'},
-               {"warning-variance", required_argument, 0, 'w'},
-               {"critical-variance", required_argument, 0, 'c'},
-               {"warning-connect", required_argument, 0, 'W'},
-               {"critical-connect", required_argument, 0, 'C'},
-               {"port", required_argument, 0, 'p'},
-               {"timeout", required_argument, 0, 't'},
-               {"version", no_argument, 0, 'V'},
-               {"help", no_argument, 0, 'h'},
-               {0, 0, 0, 0}
-       };
-#endif
-
        while (1) {
-#ifdef HAVE_GETOPT_H
-               c =
-                       getopt_long (argc, argv, "+hVH:w:c:W:C:p:t:", long_options,
-                                                                        &option_index);
-#else
-               c = getopt (argc, argv, "+hVH:w:c:W:C:p:t:");
-#endif
-
-               i++;
+               c = getopt_long (argc, argv, "hVH:w:c:W:C:p:t:u", longopts,
+                                                                        &option);
 
-               if (c == -1 || c == EOF || c == 1)
+               if (c == -1 || c == EOF)
                        break;
 
-               switch (c) {
-               case 'H':
-               case 'w':
-               case 'c':
-               case 'W':
-               case 'C':
-               case 'p':
-               case 't':
-                       i++;
-               }
-
                switch (c) {
                case '?':                                                                       /* print short usage statement if args not parsable */
-                       printf ("%s: Unknown argument: %s\n\n", my_basename (argv[0]), optarg);
-                       print_usage ();
-                       exit (STATE_UNKNOWN);
+                       usage2 (_("Unknown argument"), optarg);
                case 'h':                                                                       /* help */
                        print_help ();
                        exit (STATE_OK);
                case 'V':                                                                       /* version */
-                       print_revision (my_basename (argv[0]), "$Revision$");
+                       print_revision (progname, revision);
                        exit (STATE_OK);
                case 'H':                                                                       /* hostname */
                        if (is_host (optarg) == FALSE)
-                               usage ("Invalid host name/address\n");
+                               usage2 (_("Invalid hostname/address"), optarg);
                        server_address = optarg;
                        break;
                case 'w':                                                                       /* warning-variance */
@@ -268,11 +239,11 @@ call_getopt (int argc, char **argv)
                                        check_warning_time = TRUE;
                                }
                                else {
-                                       usage ("Warning thresholds must be a nonnegative integer\n");
+                                       usage4 (_("Warning thresholds must be a positive integer"));
                                }
                        }
                        else {
-                               usage ("Warning threshold must be a nonnegative integer\n");
+                               usage4 (_("Warning threshold must be a positive integer"));
                        }
                        break;
                case 'c':                                                                       /* critical-variance */
@@ -287,84 +258,103 @@ call_getopt (int argc, char **argv)
                                        check_critical_time = TRUE;
                                }
                                else {
-                                       usage ("Critical thresholds must be a nonnegative integer\n");
+                                       usage4 (_("Critical thresholds must be a positive integer"));
                                }
                        }
                        else {
-                               usage ("Critical threshold must be a nonnegative integer\n");
+                               usage4 (_("Critical threshold must be a positive integer"));
                        }
                        break;
                case 'W':                                                                       /* warning-connect */
                        if (!is_intnonneg (optarg))
-                               usage ("Warning threshold must be a nonnegative integer\n");
-                       warning_time = atoi (optarg);
+                               usage4 (_("Warning threshold must be a positive integer"));
+                       else
+                               warning_time = atoi (optarg);
                        check_warning_time = TRUE;
                        break;
                case 'C':                                                                       /* critical-connect */
                        if (!is_intnonneg (optarg))
-                               usage ("Critical threshold must be a nonnegative integer\n");
-                       critical_time = atoi (optarg);
+                               usage4 (_("Critical threshold must be a positive integer"));
+                       else
+                               critical_time = atoi (optarg);
                        check_critical_time = TRUE;
                        break;
                case 'p':                                                                       /* port */
                        if (!is_intnonneg (optarg))
-                               usage ("Serevr port must be a nonnegative integer\n");
-                       server_port = atoi (optarg);
+                               usage4 (_("Port must be a positive integer"));
+                       else
+                               server_port = atoi (optarg);
                        break;
                case 't':                                                                       /* timeout */
                        if (!is_intnonneg (optarg))
-                               usage ("Timeout interval must be a nonnegative integer\n");
-                       socket_timeout = atoi (optarg);
+                               usage2 (_("Timeout interval must be a positive integer"), optarg);
+                       else
+                               socket_timeout = atoi (optarg);
                        break;
+               case 'u':                                                                       /* udp */
+                       use_udp = TRUE;
                }
        }
-       return i;
-}
 
+       c = optind;
+       if (server_address == NULL) {
+               if (argc > c) {
+                       if (is_host (argv[c]) == FALSE)
+                               usage2 (_("Invalid hostname/address"), optarg);
+                       server_address = argv[c];
+               }
+               else {
+                       usage4 (_("Hostname was not supplied"));
+               }
+       }
 
+       return OK;
+}
 
 
 
 void
-print_usage (void)
+print_help (void)
 {
-       printf
-               ("Usage: check_time -H <host_address> [-p port] [-w variance] [-c variance]\n"
-                "                 [-W connect_time] [-C connect_time] [-t timeout]\n");
-}
+       char *myport;
+       asprintf (&myport, "%d", TIME_PORT);
+
+       print_revision (progname, revision);
 
+       printf ("Copyright (c) 1999 Ethan Galstad\n");
+       printf (COPYRIGHT, copyright, email);
 
+       printf (_("This plugin will check the time on the specified host.\n\n"));
+
+       print_usage ();
+
+       printf (_(UT_HELP_VRSN));
+
+       printf (_(UT_HOST_PORT), 'p', myport);
+
+       printf (_("\
+ -u, --udp\n\
+    Use UDP to connect, not TCP\n\
+ -w, --warning-variance=INTEGER\n\
+    Time difference (sec.) necessary to result in a warning status\n\
+ -c, --critical-variance=INTEGER\n\
+    Time difference (sec.) necessary to result in a critical status\n\
+ -W, --warning-connect=INTEGER\n\
+    Response time (sec.) necessary to result in warning status\n\
+ -C, --critical-connect=INTEGER\n\
+    Response time (sec.) necessary to result in critical status\n"));
+
+       printf (_(UT_TIMEOUT), DEFAULT_SOCKET_TIMEOUT);
+
+       printf (_(UT_SUPPORT));
+}
 
 
 
 void
-print_help (void)
+print_usage (void)
 {
-       print_revision (PROGNAME, "$Revision$");
-       printf
-               ("Copyright (c) 2000 Ethan Galstad/Karl DeBisschop\n\n"
-                "This plugin connects to a time port on the specified host.\n\n");
-       print_usage ();
-       printf
-               ("Options:\n"
-                " -H, --hostname=ADDRESS\n"
-                "    Host name argument for servers using host headers (use numeric\n"
-                "    address if possible to bypass DNS lookup).\n"
-                " -w, --warning-variance=INTEGER\n"
-                "    Time difference (sec.) necessary to result in a warning status\n"
-                " -c, --critical-variance=INTEGER\n"
-                "    Time difference (sec.) necessary to result in a critical status\n"
-                " -W, --warning-connect=INTEGER\n"
-                "    Response time (sec.) necessary to result in warning status\n"
-                " -C, --critical-connect=INTEGER\n"
-                "    Response time (sec.) necessary to result in critical status\n"
-                " -t, --timeout=INTEGER\n"
-                "    Seconds before connection times out (default: %d)\n"
-                " -p, --port=INTEGER\n"
-                "    Port number (default: %d)\n"
-                " -h, --help\n"
-                "    Print detailed help screen\n"
-                " -V, --version\n"
-                "    Print version information\n\n", DEFAULT_SOCKET_TIMEOUT, TIME_PORT);
-       support ();
+       printf ("\
+Usage: %s -H <host_address> [-p port] [-u] [-w variance] [-c variance]\n\
+                  [-W connect_time] [-C connect_time] [-t timeout]\n", progname);
 }