From 08f4f7c8b86cf21fce34ff4aeace71e4b937a4db Mon Sep 17 00:00:00 2001
From: Thomas Guyot-Sionnest <dermoth@users.sourceforge.net>
Date: Thu, 8 Mar 2007 09:09:38 +0000
Subject: New switch to check_icmp to set the minimum required hosts for
 success.

git-svn-id: https://nagiosplug.svn.sourceforge.net/svnroot/nagiosplug/nagiosplug/trunk@1635 f882894a-f735-0410-b71e-b25c423dba1c

diff --git a/NEWS b/NEWS
index ee0ac0e..7756de9 100644
--- a/NEWS
+++ b/NEWS
@@ -7,6 +7,7 @@ This file documents the major additions and syntax changes between releases.
 	Root plugins installed with world executable
 	./configure now detects if possible to compile check_mysql
 	check_sybase from contrib now maintained in NagiosExchange
+	check_icmp now allows to set a minimum number of hosts required for successs (-m)
 
 1.4.6 5th February 2007
 	Reduced number of DNS lookups in check_ping and netutils.c for IPv6 configurations
diff --git a/THANKS.in b/THANKS.in
index 98b8679..03b248b 100644
--- a/THANKS.in
+++ b/THANKS.in
@@ -211,3 +211,4 @@ Nobuhiro Ban
 Ville Mattila
 Lars Stavholm
 Enrico Scholz
+Marlo Bell
diff --git a/plugins-root/check_icmp.c b/plugins-root/check_icmp.c
index cd059c3..7a23e05 100644
--- a/plugins-root/check_icmp.c
+++ b/plugins-root/check_icmp.c
@@ -3,7 +3,7 @@
 * Nagios check_icmp plugin
 *
 * License: GPL
-* Copyright (c) 2005-2006 nagios-plugins team
+* Copyright (c) 2005-2007 nagios-plugins team
 *
 * Original Author : Andreas Ericsson <ae@op5.se>
 *
@@ -46,7 +46,7 @@
 /* char *progname = "check_icmp"; */
 char *progname;
 const char *revision = "$Revision$";
-const char *copyright = "2005-2006";
+const char *copyright = "2005-2007";
 const char *email = "nagiosplug-devel@lists.sourceforge.net";
 
 /** nagios plugins basic includes */
@@ -212,6 +212,7 @@ static struct timeval prog_start;
 static unsigned long long max_completion_time = 0;
 static unsigned char ttl = 0;	/* outgoing ttl */
 static unsigned int warn_down = 1, crit_down = 1; /* host down threshold values */
+static int min_hosts_alive = -1;
 float pkt_backoff_factor = 1.5;
 float target_backoff_factor = 1.5;
 
@@ -437,7 +438,7 @@ main(int argc, char **argv)
 
 	/* parse the arguments */
 	for(i = 1; i < argc; i++) {
-		while((arg = getopt(argc, argv, "vhVw:c:n:p:t:H:i:b:I:l:")) != EOF) {
+		while((arg = getopt(argc, argv, "vhVw:c:n:p:t:H:i:b:I:l:m:")) != EOF) {
 			switch(arg) {
 			case 'v':
 				debug++;
@@ -471,6 +472,9 @@ main(int argc, char **argv)
 			case 'l':
 				ttl = (unsigned char)strtoul(optarg, NULL, 0);
 				break;
+			case 'm':
+				min_hosts_alive = (int)strtoul(optarg, NULL, 0);
+				break;
 			case 'd': /* implement later, for cluster checks */
 				warn_down = (unsigned char)strtoul(optarg, &ptr, 0);
 				if(ptr) {
@@ -587,6 +591,11 @@ main(int argc, char **argv)
 		crash("packets is > 20 (%d)", packets);
 	}
 
+	if(min_hosts_alive < -1) {
+		errno = 0;
+		crash("minimum alive hosts is negative (%i)", min_hosts_alive);
+	}
+
 	host = list;
 	table = malloc(sizeof(struct rta_host **) * (argc - 1));
 	i = 0;
@@ -879,6 +888,8 @@ finish(int sig)
 	struct rta_host *host;
 	char *status_string[] =
 	{"OK", "WARNING", "CRITICAL", "UNKNOWN", "DEPENDENT"};
+	int hosts_ok = 0;
+	int hosts_warn = 0;
 
 	alarm(0);
 	if(debug > 1) printf("finish(%d) called\n", sig);
@@ -911,13 +922,25 @@ finish(int sig)
 		}
 		host->pl = pl;
 		host->rta = rta;
-		if(!status && (pl >= warn.pl || rta >= warn.rta)) status = STATE_WARNING;
-		if(pl >= crit.pl || rta >= crit.rta) status = STATE_CRITICAL;
+		if(pl >= crit.pl || rta >= crit.rta) {
+			status = STATE_CRITICAL;
+		}
+		else if(!status && (pl >= warn.pl || rta >= warn.rta)) {
+			status = STATE_WARNING;
+			hosts_warn++;
+		}
+		else {
+			hosts_ok++;
+		}
 
 		host = host->next;
 	}
 	/* this is inevitable */
 	if(!targets_alive) status = STATE_CRITICAL;
+	if(min_hosts_alive > -1) {
+		if(hosts_ok >= min_hosts_alive) status = STATE_OK;
+		else if((hosts_ok + hosts_warn) >= min_hosts_alive) status = STATE_WARNING;
+	}
 	printf("%s - ", status_string[status]);
 
 	host = list;
@@ -964,10 +987,15 @@ finish(int sig)
 		host = host->next;
 	}
 
+	if(min_hosts_alive > -1) {
+		if(hosts_ok >= min_hosts_alive) status = STATE_OK;
+		else if((hosts_ok + hosts_warn) >= min_hosts_alive) status = STATE_WARNING;
+	}
+
 	/* finish with an empty line */
 	puts("");
-	if(debug) printf("targets: %u, targets_alive: %u\n",
-					 targets, targets_alive);
+	if(debug) printf("targets: %u, targets_alive: %u, hosts_ok: %u, hosts_warn: %u, min_hosts_alive: %i\n",
+					 targets, targets_alive, hosts_ok, hosts_warn, min_hosts_alive);
 
 	exit(status);
 }
@@ -1211,6 +1239,9 @@ print_help(void)
   printf (" %s\n", "-I");
   printf ("    %s", _("max target interval (currently "));
   printf ("%0.3fms)\n", (float)target_interval / 1000);
+  printf (" %s\n", "-m");
+  printf ("    %s",_("number of alive hosts required for success"));
+  printf ("\n");
   printf (" %s\n", "-l");
   printf ("    %s", _("TTL on outgoing packets (currently "));
   printf ("%u)", ttl);
@@ -1228,10 +1259,12 @@ print_help(void)
   printf ("%s\n", _("packet loss.  The default values should work well for most users."));
   printf ("%s\n", _("You can specify different RTA factors using the standardized abbreviations"));
   printf ("%s\n\n", _("us (microseconds), ms (milliseconds, default) or just plain s for seconds."));
-  printf ("%s\n", _("Threshold format for -d is warn,crit.  12,14 means WARNING if >= 12 hops"));
+/* -d not yet implemented */
+/*  printf ("%s\n", _("Threshold format for -d is warn,crit.  12,14 means WARNING if >= 12 hops"));
   printf ("%s\n", _("are spent and CRITICAL if >= 14 hops are spent."));
-  printf ("%s\n\n", _("NOTE: Some systems decrease TTL when forming ICMP_ECHOREPLY, others do not."));
+  printf ("%s\n\n", _("NOTE: Some systems decrease TTL when forming ICMP_ECHOREPLY, others do not."));*/
   printf ("%s\n\n", _("The -v switch can be specified several times for increased verbosity."));
+
 /*  printf ("%s\n", _("Long options are currently unsupported."));
   printf ("%s\n", _("Options marked with * require an argument"));
 */
-- 
cgit v0.10-9-g596f

