Code

remove overflow checks of strtod - HUGE_VALF requires C99; also remove equality check...
[nagiosplug.git] / plugins / check_dig.c
index fcd5d0ddfbadfaf83af9baf4bcb470fb5ac0b06a..e348a352bec09323eeef97c134f70f43e21e1871 100644 (file)
@@ -1,85 +1,92 @@
-/******************************************************************************
- *
- * Program: SNMP plugin for Nagios
- * License: GPL
- *
- * 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.
+
+*****************************************************************************/
+
 #include "common.h"
+#include "netutils.h"
 #include "utils.h"
 #include "popen.h"
 
-#define PROGNAME "check_dig"
-#define REVISION "$Revision$"
-#define COPYRIGHT "2000-2002"
-#define AUTHOR "Karl DeBisschop"
-#define EMAIL "karl@debisschop.net"
-#define SUMMARY "Test the DNS service on the specified host using dig\n"
-
 int process_arguments (int, char **);
-int call_getopt (int, char **);
 int validate_arguments (void);
-int check_disk (int usp, int free_disk);
 void print_help (void);
 void print_usage (void);
 
+const char *progname = "check_dig";
+const char *revision = "$Revision$";
+const char *copyright = "2002-2003";
+const char *email = "nagiosplug-devel@lists.sourceforge.net";
+
+enum {
+       UNDEFINED = 0,
+       DEFAULT_PORT = 53
+};
+
 char *query_address = NULL;
 char *dns_server = NULL;
 int verbose = FALSE;
+int server_port = DEFAULT_PORT;
+double warning_interval = UNDEFINED;
+double critical_interval = UNDEFINED;
+struct timeval tv;
 
 int
 main (int argc, char **argv)
 {
        char input_buffer[MAX_INPUT_BUFFER];
-       char *command_line = NULL;
-       char *output = NULL;
+       char *command_line;
+       char *output;
+       long microsec;
+       double elapsed_time;
        int result = STATE_UNKNOWN;
 
+       output = strdup ("");
+
+       setlocale (LC_ALL, "");
+       bindtextdomain (PACKAGE, LOCALEDIR);
+       textdomain (PACKAGE);
+
        /* Set signal handling and alarm */
        if (signal (SIGALRM, popen_timeout_alarm_handler) == SIG_ERR)
-               usage ("Cannot catch SIGALRM\n");
+               usage (_("Cannot catch SIGALRM\n"));
 
        if (process_arguments (argc, argv) != OK)
-               usage ("Could not parse arguments\n");
+               usage (_("Could not parse arguments\n"));
 
        /* get the command to run */
-       command_line =
-               ssprintf (command_line, "%s @%s %s", PATH_TO_DIG, dns_server,
-                                                       query_address);
+       asprintf (&command_line, "%s @%s -p %d %s",
+                 PATH_TO_DIG, dns_server, server_port, query_address);
 
        alarm (timeout_interval);
-       time (&start_time);
+       gettimeofday (&tv, NULL);
 
        if (verbose)
                printf ("%s\n", command_line);
+
        /* run the command */
        child_process = spopen (command_line);
        if (child_process == NULL) {
-               printf ("Could not open pipe: %s\n", command_line);
+               printf (_("Could not open pipe: %s\n"), command_line);
                return STATE_UNKNOWN;
        }
 
        child_stderr = fdopen (child_stderr_array[fileno (child_process)], "r");
        if (child_stderr == NULL)
-               printf ("Could not open stderr for %s\n", command_line);
-
-       output = strscpy (output, "");
+               printf (_("Could not open stderr for %s\n"), command_line);
 
        while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_process)) {
 
@@ -94,11 +101,11 @@ main (int argc, char **argv)
                                input_buffer[strcspn (input_buffer, "\r\n")] = '\0';
 
                        if (strstr (input_buffer, query_address) == input_buffer) {
-                               output = strscpy (output, input_buffer);
+                               output = strdup(input_buffer);
                                result = STATE_OK;
                        }
                        else {
-                               strcpy (output, "Server not found in ANSWER SECTION");
+                               asprintf (&output, _("Server not found in ANSWER SECTION"));
                                result = STATE_WARNING;
                        }
 
@@ -108,15 +115,15 @@ main (int argc, char **argv)
        }
 
        if (result != STATE_OK) {
-               strcpy (output, "No ANSWER SECTION found");
+               asprintf (&output, _("No ANSWER SECTION found"));
        }
 
        while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_stderr)) {
                /* If we get anything on STDERR, at least set warning */
                result = max_state (result, STATE_WARNING);
                printf ("%s", input_buffer);
-               if (!strcmp (output, ""))
-                       strcpy (output, 1 + index (input_buffer, ':'));
+               if (strlen (output) == 0)
+                       output = strdup (1 + index (input_buffer, ':'));
        }
 
        (void) fclose (child_stderr);
@@ -124,93 +131,123 @@ main (int argc, char **argv)
        /* close the pipe */
        if (spclose (child_process)) {
                result = max_state (result, STATE_WARNING);
-               if (!strcmp (output, ""))
-                       strcpy (output, "nslookup returned error status");
+               if (strlen (output) == 0)
+                       asprintf (&output, _("dig returned error status"));
        }
 
-       (void) time (&end_time);
-
-       if (result == STATE_OK)
-               printf ("DNS ok - %d seconds response time (%s)\n",
-                                               (int) (end_time - start_time), output);
-       else if (result == STATE_WARNING)
-               printf ("DNS WARNING - %s\n",
-                                               !strcmp (output,
-                                                                                "") ? " Probably a non-existent host/domain" : output);
-       else if (result == STATE_CRITICAL)
-               printf ("DNS CRITICAL - %s\n",
-                                               !strcmp (output,
-                                                                                "") ? " Probably a non-existent host/domain" : output);
-       else
-               printf ("DNS problem - %s\n",
-                                               !strcmp (output,
-                                                                                "") ? " Probably a non-existent host/domain" : output);
+       microsec = deltime (tv);
+       elapsed_time = (double)microsec / 1.0e6;
+
+       if (output == NULL || strlen (output) == 0)
+               asprintf (&output, _(" Probably a non-existent host/domain"));
+
+       if (critical_interval > UNDEFINED && elapsed_time > critical_interval)
+               result = STATE_CRITICAL;
+
+       else if (warning_interval > UNDEFINED && elapsed_time > warning_interval)
+               result = STATE_WARNING;
 
+       asprintf (&output, _("%.3f seconds response time (%s)"), elapsed_time, output);
+
+       printf ("DNS %s - %s|%s\n",
+               state_text (result), output,
+               perfdata("time", microsec, "us",
+                        (warning_interval>UNDEFINED?TRUE:FALSE),
+                        (int)(1e6*warning_interval),
+                        (critical_interval>UNDEFINED?TRUE:FALSE),
+                        (int)(1e6*critical_interval),
+                                                                        TRUE, 0, FALSE, 0));
        return result;
 }
 
+
+
+
+
+\f
 /* process command-line arguments */
 int
 process_arguments (int argc, char **argv)
 {
        int c;
 
-#ifdef HAVE_GETOPT_H
-       int option_index = 0;
-       static struct option long_options[] = {
+       int option = 0;
+       static struct option longopts[] = {
                {"hostname", required_argument, 0, 'H'},
-               {"query_address", required_argument, 0, 'e'},
+               {"query_address", required_argument, 0, 'l'},
+               {"warning", required_argument, 0, 'w'},
+               {"critical", required_argument, 0, 'c'},
+               {"timeout", required_argument, 0, 't'},
                {"verbose", no_argument, 0, 'v'},
                {"version", no_argument, 0, 'V'},
                {"help", no_argument, 0, 'h'},
                {0, 0, 0, 0}
        };
-#endif
 
        if (argc < 2)
                return ERROR;
 
        while (1) {
-#ifdef HAVE_GETOPT_H
-               c = getopt_long (argc, argv, "hVvt:l:H:", long_options, &option_index);
-#else
-               c = getopt (argc, argv, "hVvt:l:H:");
-#endif
+               c = getopt_long (argc, argv, "hVvt:l:H:w:c:", longopts, &option);
 
                if (c == -1 || c == EOF)
                        break;
 
                switch (c) {
                case '?':                                                                       /* help */
-                       usage3 ("Unknown argument", optopt);
+                       usage3 (_("Unknown argument"), optopt);
+               case 'h':                                                                       /* help */
+                       print_help ();
+                       exit (STATE_OK);
+               case 'V':                                                                       /* version */
+                       print_revision (progname, "$Revision$");
+                       exit (STATE_OK);
                case 'H':                                                                       /* hostname */
                        if (is_host (optarg)) {
                                dns_server = optarg;
                        }
                        else {
-                               usage ("Invalid host name\n");
+                               usage2 (_("Invalid host name"), optarg);
                        }
                        break;
-               case 'l':                                                                       /* username */
+               case 'p':                 /* server port */
+                       if (is_intpos (optarg)) {
+                               server_port = atoi (optarg);
+                       }
+                       else {
+                               usage2 (_("Server port must be a nonnegative integer"), optarg);
+                       }
+                       break;
+               case 'l':                                                                       /* address to lookup */
                        query_address = optarg;
                        break;
-               case 'v':                                                                       /* verbose */
-                       verbose = TRUE;
+               case 'w':                                                                       /* warning */
+                       if (is_nonnegative (optarg)) {
+                               warning_interval = strtod (optarg, NULL);
+                       }
+                       else {
+                               usage2 (_("Warning interval must be a nonnegative integer"), optarg);
+                       }
+                       break;
+               case 'c':                                                                       /* critical */
+                       if (is_nonnegative (optarg)) {
+                               critical_interval = strtod (optarg, NULL);
+                       }
+                       else {
+                               usage2 (_("Critical interval must be a nonnegative integer"), optarg);
+                       }
                        break;
                case 't':                                                                       /* timeout */
                        if (is_intnonneg (optarg)) {
                                timeout_interval = atoi (optarg);
                        }
                        else {
-                               usage ("Time interval must be a nonnegative integer\n");
+                               usage2 (_("Time interval must be a nonnegative integer"), optarg);
                        }
                        break;
-               case 'V':                                                                       /* version */
-                       print_revision (PROGNAME, "$Revision$");
-                       exit (STATE_OK);
-               case 'h':                                                                       /* help */
-                       print_help ();
-                       exit (STATE_OK);
+               case 'v':                                                                       /* verbose */
+                       verbose = TRUE;
+                       break;
                }
        }
 
@@ -221,11 +258,11 @@ process_arguments (int argc, char **argv)
                                dns_server = argv[c];
                        }
                        else {
-                               usage ("Invalid host name");
+                               usage2 (_("Invalid host name"), argv[c]);
                        }
                }
                else {
-                       dns_server = strscpy (NULL, "127.0.0.1");
+                       dns_server = strdup ("127.0.0.1");
                }
        }
 
@@ -246,31 +283,40 @@ validate_arguments (void)
 
 
 
+
+\f
 void
 print_help (void)
 {
-       print_revision (PROGNAME, "$Revision$");
-       printf
-               ("Copyright (c) %s %s <%s>\n\n%s\n",
-                COPYRIGHT, AUTHOR, EMAIL, SUMMARY);
+       char *myport;
+
+       asprintf (&myport, "%d", DEFAULT_PORT);
+
+       print_revision (progname, revision);
+
+       printf (_("Copyright (c) 2000 Karl DeBisschop <kdebisschop@users.sourceforge.net>\n"));
+       printf (_(COPYRIGHT), copyright, email);
+
+       printf (_("Test the DNS service on the specified host using dig\n\n"));
+
        print_usage ();
-       printf
-               ("\nOptions:\n"
-                " -H, --hostname=STRING or IPADDRESS\n"
-                "   Check server on the indicated host\n"
-                " -l, --lookup=STRING\n"
-                "   machine name to lookup\n"
-                " -t, --timeout=INTEGER\n"
-                "   Seconds before connection attempt times out (default: %d)\n"
-                " -v, --verbose\n"
-                "   Print extra information (command-line use only)\n"
-                " -h, --help\n"
-                "   Print detailed help screen\n"
-                " -V, --version\n"
-                "   Print version information\n\n", DEFAULT_SOCKET_TIMEOUT);
-       support ();
-}
 
+       printf (_(UT_HELP_VRSN));
+
+       printf (_(UT_HOST_PORT), 'P', myport);
+
+       printf (_("\
+ -l, --lookup=STRING\n\
+   machine name to lookup\n"));
+
+       printf (_(UT_WARN_CRIT));
+
+       printf (_(UT_TIMEOUT), DEFAULT_SOCKET_TIMEOUT);
+
+       printf (_(UT_VERBOSE));
+
+       printf (_(UT_SUPPORT));
+}
 
 
 
@@ -278,8 +324,10 @@ print_help (void)
 void
 print_usage (void)
 {
-       printf
-               ("Usage: %s -H host -l lookup [-t timeout] [-v]\n"
-                "       %s --help\n"
-                "       %s --version\n", PROGNAME, PROGNAME, PROGNAME);
+       printf (_("\
+Usage: %s -H host -l lookup [-p <server port>] [-w <warning interval>]\n\
+         [-c <critical interval>] [-t <timeout>] [-v]\n"),
+               progname);
+       printf ("       %s (-h|--help)\n", progname);
+       printf ("       %s (-V|--version)\n", progname);
 }