Code

Spent the day working on backwards compatability using getaddrinfo()
[nagiosplug.git] / plugins / check_ping.c
index 7ff7f28467be600d3aae7091bcf76052834c5f9a..2526a74f451aa8037fae927d5fb0093e856890a6 100644 (file)
@@ -10,7 +10,7 @@
 *
 *****************************************************************************/
 
-#define PROGNAME "check_ping"
+const char *progname = "check_ping";
 #define REVISION "$Revision$"
 #define COPYRIGHT "1999-2001"
 #define AUTHOR "Ethan Galstad/Karl DeBisschop"
@@ -56,10 +56,9 @@ the contrib area of the downloads section at http://www.nagios.org\n\n"
 #define WARN_DUPLICATES "DUPLICATES FOUND! "
 
 int process_arguments (int, char **);
-int call_getopt (int, char **);
 int get_threshold (char *, float *, int *);
 int validate_arguments (void);
-int run_ping (char *);
+int run_ping (char *, char *);
 void print_usage (void);
 void print_help (void);
 
@@ -68,7 +67,9 @@ int wpl = UNKNOWN_PACKET_LOSS;
 int cpl = UNKNOWN_PACKET_LOSS;
 float wrta = UNKNOWN_TRIP_TIME;
 float crta = UNKNOWN_TRIP_TIME;
-char *server_address = NULL;
+char **addresses = NULL;
+int n_addresses;
+int max_addr = 1;
 int max_packets = -1;
 int verbose = FALSE;
 
@@ -82,20 +83,15 @@ main (int argc, char **argv)
 {
        char *command_line = NULL;
        int result = STATE_UNKNOWN;
+       int this_result = STATE_UNKNOWN;
+       int i;
+
+       addresses = malloc (max_addr);
 
        if (process_arguments (argc, argv) == ERROR)
                usage ("Could not parse arguments");
        exit;
 
-       /* does the host address of number of packets argument come first? */
-#ifdef PING_PACKETS_FIRST
-       command_line =
-               ssprintf (command_line, PING_COMMAND, max_packets, server_address);
-#else
-       command_line =
-               ssprintf (command_line, PING_COMMAND, server_address, max_packets);
-#endif
-
        /* Set signal handling and alarm */
        if (signal (SIGALRM, popen_timeout_alarm_handler) == SIG_ERR) {
                printf ("Cannot catch SIGALRM");
@@ -105,43 +101,55 @@ main (int argc, char **argv)
        /* handle timeouts gracefully */
        alarm (timeout_interval);
 
-       if (verbose)
-               printf ("%s ==> ", command_line);
+       for (i = 0 ; i < n_addresses ; i++) {
+
+               /* does the host address of number of packets argument come first? */
+#ifdef PING_PACKETS_FIRST
+               asprintf (&command_line, PING_COMMAND, max_packets, addresses[i]);
+#else
+               asprintf (&command_line, PING_COMMAND, addresses[i], max_packets);
+#endif
 
-       /* run the command */
-       run_ping (command_line);
+               if (verbose)
+                       printf ("%s ==> ", command_line);
 
-       if (pl == UNKNOWN_PACKET_LOSS || rta == UNKNOWN_TRIP_TIME) {
-               printf ("%s\n", command_line);
-               terminate (STATE_UNKNOWN,
-                                                        "Error: Could not interpret output from ping command\n");
-       }
+               /* run the command */
+               this_result = run_ping (command_line, addresses[i]);
 
-       if (pl >= cpl || rta >= crta || rta < 0)
-               result = STATE_CRITICAL;
-       else if (pl >= wpl || rta >= wrta)
-               result = STATE_WARNING;
-       else if (pl < wpl && rta < wrta && pl >= 0 && rta >= 0)
-               /* cannot use the max function because STATE_UNKNOWN is now 3 gt STATE_OK                       
-               result = max (result, STATE_OK);  */
-               if( !( (result == STATE_WARNING) || (result == STATE_CRITICAL) )  ) {
-                       result = STATE_OK;      
+               if (pl == UNKNOWN_PACKET_LOSS || rta == UNKNOWN_TRIP_TIME) {
+                       printf ("%s\n", command_line);
+                       terminate (STATE_UNKNOWN,
+                                                                "Error: Could not interpret output from ping command\n");
                }
+
+               if (pl >= cpl || rta >= crta || rta < 0)
+                       this_result = STATE_CRITICAL;
+               else if (pl >= wpl || rta >= wrta)
+                       this_result = STATE_WARNING;
+               else if (pl >= 0 && rta >= 0)
+                       this_result = max_state (STATE_OK, this_result);        
        
-       if (display_html == TRUE)
-               printf ("<A HREF='%s/traceroute.cgi?%s'>", CGIURL, server_address);
-       if (pl == 100)
-               printf ("PING %s - %sPacket loss = %d%%", state_text (result), warn_text,
-                                               pl);
-       else
-               printf ("PING %s - %sPacket loss = %d%%, RTA = %2.2f ms",
-                                               state_text (result), warn_text, pl, rta);
-       if (display_html == TRUE)
-               printf ("</A>");
-       printf ("\n");
+               if (n_addresses > 1 && this_result != STATE_UNKNOWN)
+                       terminate (STATE_OK, "%s is alive\n", addresses[i]);
+
+               if (display_html == TRUE)
+                       printf ("<A HREF='%s/traceroute.cgi?%s'>", CGIURL, addresses[i]);
+               if (pl == 100)
+                       printf ("PING %s - %sPacket loss = %d%%", state_text (this_result), warn_text,
+                                                       pl);
+               else
+                       printf ("PING %s - %sPacket loss = %d%%, RTA = %2.2f ms",
+                                                       state_text (this_result), warn_text, pl, rta);
+               if (display_html == TRUE)
+                       printf ("</A>");
+               printf ("\n");
 
-       if (verbose)
-               printf ("%f:%d%% %f:%d%%\n", wrta, wpl, crta, cpl);
+               if (verbose)
+                       printf ("%f:%d%% %f:%d%%\n", wrta, wpl, crta, cpl);
+
+               result = max_state (result, this_result);
+
+       }
 
        return result;
 }
@@ -152,8 +160,8 @@ int
 process_arguments (int argc, char **argv)
 {
        int c = 1;
+       char *ptr;
 
-#ifdef HAVE_GETOPT_H
        int option_index = 0;
        static struct option long_options[] = {
                STD_LONG_OPTS,
@@ -162,9 +170,6 @@ process_arguments (int argc, char **argv)
                {"link", no_argument, 0, 'L'},
                {0, 0, 0, 0}
        };
-#endif
-
-#define OPTCHARS "Vvht:c:w:H:p:nL"
 
        if (argc < 2)
                return ERROR;
@@ -177,22 +182,19 @@ process_arguments (int argc, char **argv)
        }
 
        while (1) {
-#ifdef HAVE_GETOPT_H
-               c = getopt_long (argc, argv, OPTCHARS, long_options, &option_index);
-#else
-               c = getopt (argc, argv, OPTCHARS);
-#endif
+               c = getopt_long (argc, argv, "VvhnLt:c:w:H:p:", long_options, &option_index);
+
                if (c == -1 || c == EOF)
                        break;
 
                switch (c) {
                case '?':       /* usage */
-                       usage2 ("Unknown argument", optarg);
+                       usage3 ("Unknown argument", optopt);
                case 'h':       /* help */
                        print_help ();
                        exit (STATE_OK);
                case 'V':       /* version */
-                       print_revision (PROGNAME, REVISION);
+                       print_revision (progname, REVISION);
                        exit (STATE_OK);
                case 't':       /* timeout period */
                        timeout_interval = atoi (optarg);
@@ -201,9 +203,23 @@ process_arguments (int argc, char **argv)
                        verbose = TRUE;
                        break;
                case 'H':       /* hostname */
-                       if (is_host (optarg) == FALSE)
-                               usage2 ("Invalid host name/address", optarg);
-                       server_address = optarg;
+                       ptr=optarg;
+                       while (1) {
+                               n_addresses++;
+                               if (n_addresses > max_addr) {
+                                       max_addr *= 2;
+                                       addresses = realloc (addresses, max_addr);
+                                       if (addresses == NULL)
+                                               terminate (STATE_UNKNOWN, "Could not realloc() addresses\n");
+                               }
+                               addresses[n_addresses-1] = ptr;
+                               if (ptr = index (ptr, ',')) {
+                                       strcpy (ptr, "");
+                                       ptr += sizeof(char);
+                               } else {
+                                       break;
+                               }
+                       }
                        break;
                case 'p':       /* number of packets to send */
                        if (is_intnonneg (optarg))
@@ -230,12 +246,12 @@ process_arguments (int argc, char **argv)
        if (c == argc)
                return validate_arguments ();
 
-       if (server_address == NULL) {
+       if (addresses[0] == NULL) {
                if (is_host (argv[c]) == FALSE) {
                        printf ("Invalid host name/address: %s\n\n", argv[c]);
                        return ERROR;
                } else {
-                       server_address = argv[c++];
+                       addresses[0] = argv[c++];
                        if (c == argc)
                                return validate_arguments ();
                }
@@ -314,6 +330,7 @@ int
 validate_arguments ()
 {
        float max_seconds;
+       int i;
 
        if (wrta == UNKNOWN_TRIP_TIME) {
                printf ("<wrta> was not set\n");
@@ -347,12 +364,17 @@ validate_arguments ()
        if (max_seconds > timeout_interval)
                timeout_interval = (int)max_seconds;
 
+       for (i=0; i<n_addresses; i++) {
+               if (is_host(addresses[i]) == FALSE)
+                       usage2 ("Invalid host name/address", addresses[i]);
+       }
+
        return OK;
 }
 \f
 
 int
-run_ping (char *command_line)
+run_ping (char *command_line, char *server_address)
 {
        char input_buffer[MAX_INPUT_BUFFER];
        int result = STATE_UNKNOWN;
@@ -478,20 +500,15 @@ void
 print_usage (void)
 {
        printf ("Usage:\n" " %s %s\n"
-#ifdef HAVE_GETOPT_H
                                        " %s (-h | --help) for detailed help\n"
                                        " %s (-V | --version) for version information\n",
-#else
-                                       " %s -h for detailed help\n"
-                                       " %s -V for version information\n",
-#endif
-                                       PROGNAME, OPTIONS, PROGNAME, PROGNAME);
+                                       progname, OPTIONS, progname, progname);
 }
 
 void
 print_help (void)
 {
-       print_revision (PROGNAME, REVISION);
+       print_revision (progname, REVISION);
        printf
                ("Copyright (c) %s %s <%s>\n\n%s\n",
                 COPYRIGHT, AUTHOR, EMAIL, SUMMARY);