Check if IP address is specified on RBL - Tim Bell

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

1 files changed, 329 insertions, 0 deletions

diff --git a/contrib/check_rbl.c b/contrib/check_rbl.c
new file mode 100644
index 0000000..5c66113
--- /dev/null
+++ b/contrib/check_rbl.c
@@ -0,0 +1,329 @@
+/******************************************************************************
+*
+* check_rbl.c
+*
+* Modified by Tim Bell <bhat@trinity.unimelb.edu.au> 2002-06-05
+* based on:
+*
+* * check_dig.c
+* *
+* * Program: dig plugin for NetSaint
+* * License: GPL
+* * Copyright (c) 2000
+* * 
+* * $Id$
+*
+*****************************************************************************/
+
+#include "config.h"
+#include "common.h"
+#include "utils.h"
+#include "popen.h"
+#include "string.h"
+
+#define PROGNAME "check_rbl"
+
+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);
+char *reverse_ipaddr(char *ipaddr);
+
+char *query_address=NULL;
+char *query_address_rev=NULL;
+char *dns_server=NULL;
+char *rbl_name=NULL;
+int verbose=FALSE;
+
+int main(int argc, char **argv){
+        char input_buffer[MAX_INPUT_BUFFER];
+        char *command_line=NULL;
+        char *output=NULL;
+        int result=STATE_OK;
+
+        /* Set signal handling and alarm */
+        if (signal(SIGALRM,popen_timeout_alarm_handler)==SIG_ERR)
+                usage("Cannot catch SIGALRM\n");
+
+        if (process_arguments(argc,argv)!=OK)
+                usage("Could not parse arguments\n");
+
+        /* reverse the octets in the IP address */
+        query_address_rev = reverse_ipaddr(query_address);
+
+        /* build the command to run */
+        if (dns_server) {
+          command_line=ssprintf(command_line,"%s @%s %s.%s",
+                                PATH_TO_DIG,dns_server,
+                                query_address_rev, rbl_name);
+        } else {
+          command_line=ssprintf(command_line,"%s %s.%s",
+                                PATH_TO_DIG,
+                                query_address_rev, rbl_name);
+        }
+        alarm(timeout_interval);
+        time(&start_time);
+
+        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);
+                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,"");
+
+        while (fgets(input_buffer,MAX_INPUT_BUFFER-1,child_process)) {
+
+                /* the server is responding, we just got the host name... */
+                if (strstr(input_buffer,";; ANSWER SECTION:")) {
+
+                        /* get the host address */
+                        if (!fgets(input_buffer,MAX_INPUT_BUFFER-1,child_process))
+                                break;
+
+                        if (strpbrk(input_buffer,"\r\n"))
+                                input_buffer[strcspn(input_buffer,"\r\n")] = '\0';
+
+                        if (strstr(input_buffer,query_address_rev)==input_buffer) {
+                                output=strscpy(output,input_buffer);
+                                /* we found it, which means it's listed! */
+                                result=STATE_CRITICAL;
+                        } else {
+                                strcpy(output,"Server not RBL listed.");
+                                result=STATE_OK;
+                        }
+
+                        continue;
+                }
+
+        }
+        
+        /*
+        if (result!=STATE_OK) {
+                strcpy(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=error_set(result,STATE_WARNING);
+                printf("%s",input_buffer);
+                if (!strcmp(output,""))
+                        strcpy(output,1+index(input_buffer,':'));
+        }
+
+        (void)fclose(child_stderr);
+
+        /* close the pipe */
+        if (spclose(child_process)) {
+                result=error_set(result,STATE_WARNING);
+                if (!strcmp(output,""))
+                        strcpy(output,"nslookup returned error status");
+        }
+        
+        (void)time(&end_time);
+
+        if (result==STATE_OK)
+                printf("RBL check okay - not listed.\n");
+        else if (result==STATE_WARNING)
+                printf("RBL WARNING - %s\n",!strcmp(output,"")?" Probably a non-existent host/domain":output);
+        else if (result==STATE_CRITICAL)
+                printf("RBL CRITICAL - %s is listed on %s\n",query_address, rbl_name);
+        else
+                printf("DNS problem - %s\n",!strcmp(output,"")?" Probably a non-existent host/domain":output);
+
+        return result;
+}
+
+/* reverse the ipaddr */
+char *reverse_ipaddr(char *ipaddr)
+{
+  static char revip[MAX_HOST_ADDRESS_LENGTH];
+  int a, b, c, d;
+
+  if (strlen(ipaddr) >= MAX_HOST_ADDRESS_LENGTH ||
+      sscanf(ipaddr, "%d.%d.%d.%d", &a, &b, &c, &d) != 4) {
+    usage("IP address invalid or too long");
+  }
+  sprintf(revip, "%d.%d.%d.%d", d, c, b, a);
+
+  return revip;
+}
+
+
+/* process command-line arguments */
+int process_arguments(int argc, char **argv)
+{
+  int c;
+
+  if(argc<2)
+    return ERROR;
+
+
+  c=0;
+  while((c+=(call_getopt(argc-c,&argv[c])))<argc){
+
+                if (is_option(argv[c]))
+                        continue;
+
+    if (query_address==NULL) {
+                        if (is_host(argv[c])) {
+                                query_address=argv[c];
+                        } else {
+                                usage("Invalid host name");
+                        }
+                }
+  }
+
+  return validate_arguments();
+}
+
+
+
+
+
+
+int call_getopt(int argc, char **argv)
+{
+  int c,i=0;
+
+#ifdef HAVE_GETOPT_H
+        int option_index = 0;
+        static struct option long_options[] =
+        { 
+                {"hostname",       required_argument,0,'H'},
+                {"server",         required_argument,0,'s'},
+                {"rblname",        required_argument,0,'r'},
+                {"verbose",        no_argument,      0,'v'},
+                {"version",        no_argument,      0,'V'},
+                {"help",           no_argument,      0,'h'},
+                {0,0,0,0}
+        };
+#endif
+
+  while (1){
+#ifdef HAVE_GETOPT_H
+    c = getopt_long(argc,argv,"+hVvt:s:H:r:",long_options,&option_index);
+#else
+    c = getopt(argc,argv,"+?hVvt:s:H:r:");
+#endif
+
+    i++;
+
+    if(c==-1||c==EOF||c==1)
+      break;
+
+    switch (c)
+      {
+      case 't':
+      case 'l':
+      case 'H':
+                                i++;
+      }
+
+    switch (c)
+      {
+      case 'H': /* hostname */
+                                if (is_host(optarg)) {
+                                        query_address=optarg;
+                                } else {
+                                        usage("Invalid host name (-H)\n");
+                                }
+                                break;
+      case 's': /* server */
+                                if (is_host(optarg)) {
+                                        dns_server=optarg;
+                                } else {
+                                        usage("Invalid host name (-s)\n");
+                                }
+                                break;
+      case 'r': /* rblname */
+                                rbl_name=optarg;
+                                break;
+      case 'v': /* verbose */
+                                verbose=TRUE;
+                                break;
+      case 't': /* timeout */
+                                if (is_intnonneg(optarg)) {
+                                        timeout_interval=atoi(optarg);
+                                } else {
+                                        usage("Time interval must be a nonnegative integer\n");
+                                }
+                                break;
+      case 'V': /* version */
+                                print_revision(PROGNAME,"$Revision$");
+                                exit(STATE_OK);
+      case 'h': /* help */
+                                print_help();
+                                exit(STATE_OK);
+      case '?': /* help */
+                                usage("Invalid argument\n");
+                        }
+  }
+  return i;
+}
+
+
+
+
+
+int validate_arguments(void)
+{
+  if (query_address == NULL || rbl_name == NULL)
+    return ERROR;
+  else
+    return OK;
+}
+
+
+
+
+
+void print_help(void)
+{
+        print_revision(PROGNAME,"$Revision$");
+        printf
+                ("Copyright (c) 2000 Karl DeBisschop\n\n"
+                 "This plugin uses dig to test whether the specified host is on any RBL lists.\n\n");
+        print_usage();
+        printf
+                ("\nOptions:\n"
+                 " -H, --hostname=IPADDRESS\n"
+                 "   Check status of indicated host\n"
+                 " -s, --server=STRING or IPADDRESS\n"
+                 "   DNS server to use\n"
+                 " -r, --rblname=STRING\n"
+                 "   RBL domain name to use (e.g. relays.ordb.org)\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();
+}
+
+
+
+
+
+void print_usage(void)
+{
+        printf
+                ("Usage: %s -H hostip -r rblname [-s server] [-t timeout] [-v]\n"
+                 "       %s --help\n"
+                 "       %s --version\n",
+                 PROGNAME,PROGNAME,PROGNAME);
+}