New cluster plugin for Nagios 2.x

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

contrib/check_cluster2.c

@@ -0,0 +1,232 @@
+/*****************************************************************************
+ *
+ * CHECK_CLUSTER2.C - Host and Service Cluster Plugin for Nagios 2.x
+ *
+ * Copyright (c) 2000-2004 Ethan Galstad (nagios@nagios.org)
+ * License: GPL
+ * Last Modified:   03-11-2004
+ *
+ * License:
+ *
+ * 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 <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <getopt.h>
+
+#define OK		0
+#define ERROR		-1
+
+#define TRUE		1
+#define FALSE		0
+
+#define CHECK_SERVICES	1
+#define CHECK_HOSTS	2
+
+#define MAX_INPUT_BUFFER	1024
+
+#define STATE_OK	0
+#define STATE_WARNING	1
+#define STATE_CRITICAL	2
+#define STATE_UNKNOWN	3
+
+int total_services_ok=0;
+int total_services_warning=0;
+int total_services_unknown=0;
+int total_services_critical=0;
+
+int total_hosts_up=0;
+int total_hosts_down=0;
+int total_hosts_unreachable=0;
+
+int warning_threshold=1;
+int critical_threshold=1;
+
+int check_type=CHECK_SERVICES;
+
+char *data_vals=NULL;
+char *label=NULL;
+
+
+int process_arguments(int,char **);
+
+
+
+int main(int argc, char **argv){
+	char input_buffer[MAX_INPUT_BUFFER];
+	char *ptr;
+	int data_val;
+	int return_code=STATE_OK;
+	int error=FALSE;
+
+	if(process_arguments(argc,argv)==ERROR){
+
+		printf("Invalid arguments supplied\n");
+		printf("\n");
+
+		printf("Host/Service Cluster Plugin for Nagios 2\n");
+		printf("Copyright (c) 2000-2004 Ethan Galstad (nagios@nagios.org)\n");
+		printf("Last Modified: 03-11-2004\n");
+		printf("License: GPL\n");
+		printf("\n");
+		printf("Usage: %s (-s | -h) [-l label] [-w threshold] [-c threshold] [-d val1,val2,...,valn]\n",argv[0]);
+		printf("\n");
+		printf("Options:\n");
+		printf("   -s, --service  = Check service cluster status\n");
+		printf("   -h, --host     = Check host cluster status\n");
+		printf("   -l, --label    = Optional prepended text output (i.e. \"Host cluster\")\n");
+		printf("   -w, --warning  = Specifies the number of hosts or services in cluster that must be in\n");
+		printf("                    a non-OK state in order to return a WARNING status level\n");
+		printf("   -c, --critical = Specifies the number of hosts or services in cluster that must be in\n");
+		printf("                    a non-OK state in order to return a CRITICAL status level\n");
+		printf("   -d, --data     = The status codes of the hosts or services in the cluster, separated\n");
+		printf("                    by commas\n");
+		printf("\n");
+
+		return STATE_UNKNOWN;
+	        }
+
+	/* check the data values */
+	for(ptr=strtok(data_vals,",");ptr!=NULL;ptr=strtok(NULL,",")){
+
+		data_val=atoi(ptr);
+
+		if(check_type==CHECK_SERVICES){
+			switch(data_val){
+			case 0:
+				total_services_ok++;
+				break;
+			case 1:
+				total_services_warning++;
+				break;
+			case 2:
+				total_services_critical++;
+				break;
+			case 3:
+				total_services_unknown++;
+				break;
+			default:
+				break;
+			        }
+		        }
+		else{
+			switch(data_val){
+			case 0:
+				total_hosts_up++;
+				break;
+			case 1:
+				total_hosts_down++;
+				break;
+			case 2:
+				total_hosts_unreachable++;
+				break;
+			default:
+				break;
+			        }
+		        }
+	        }
+	
+
+	/* return the status of the cluster */
+	if(check_type==CHECK_SERVICES){
+		if((total_services_warning+total_services_unknown+total_services_critical) >= critical_threshold)
+			return_code=STATE_CRITICAL;
+		else if((total_services_warning+total_services_unknown+total_services_critical) >= warning_threshold)
+			return_code=STATE_WARNING;
+		else
+			return_code=STATE_OK;
+		printf("%s %s: %d ok, %d warning, %d unknown, %d critical\n",(label==NULL)?"Service cluster":label,(return_code==STATE_OK)?"ok":"problem",total_services_ok,total_services_warning,total_services_unknown,total_services_critical);
+                }
+	else{
+		if((total_hosts_down+total_hosts_unreachable) >= critical_threshold)
+			return_code=STATE_CRITICAL;
+		else if((total_hosts_down+total_hosts_unreachable) >= warning_threshold)
+			return_code=STATE_WARNING;
+		else
+			return_code=STATE_OK;
+		printf("%s %s: %d up, %d down, %d unreachable\n",(label==NULL)?"Host cluster":label,(return_code==STATE_OK)?"ok":"problem",total_hosts_up,total_hosts_down,total_hosts_unreachable);
+                }
+
+	return return_code;
+        }
+
+
+
+int process_arguments(int argc, char **argv){
+	int c;
+	int option=0;
+	static struct option longopts[]={ 
+		{"data",     required_argument,0,'d'},
+		{"warning",  required_argument,0,'w'},
+		{"critical", required_argument,0,'c'},
+		{"label",    required_argument,0,'l'},
+		{"host",     no_argument,      0,'h'},
+		{"service",  no_argument,      0,'s'},
+		{0,0,0,0}
+		};
+
+	/* no options were supplied */
+	if(argc<2)
+		return ERROR;
+
+	while(1){
+
+		c=getopt_long(argc,argv,"hsw:c:d:l:",longopts,&option);
+
+		if(c==-1 || c==EOF || c==1)
+			break;
+
+		switch(c){
+
+		case 'h': /* host cluster */
+			check_type=CHECK_HOSTS;
+			break;
+
+		case 's': /* service cluster */
+			check_type=CHECK_SERVICES;
+			break;
+
+		case 'w': /* warning threshold */
+			warning_threshold=atoi(optarg);
+			break;
+
+		case 'c': /* warning threshold */
+			critical_threshold=atoi(optarg);
+			break;
+
+		case 'd': /* data values */
+			data_vals=(char *)strdup(optarg);
+			break;
+
+		case 'l': /* text label */
+			label=(char *)strdup(optarg);
+			break;
+
+		default:
+			return ERROR;
+			break;
+		        }
+		}
+
+	if(data_vals==NULL)
+		return ERROR;
+
+	return OK;
+        }