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

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);
+}