diff options
| -rw-r--r-- | contrib/check_rbl.c | 329 |
1 files changed, 329 insertions, 0 deletions
diff --git a/contrib/check_rbl.c b/contrib/check_rbl.c new file mode 100644 index 00000000..5c661135 --- /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); +} |