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

1 files changed, 232 insertions, 0 deletions

diff --git a/contrib/check_cluster2.c b/contrib/check_cluster2.c
new file mode 100644
index 0000000..e60948d
--- /dev/null
+++ b/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;
+        }