122 lines
3.3 KiB
C
122 lines
3.3 KiB
C
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <time.h>
|
|
|
|
#define MAX_CHARS 1024
|
|
#define SERVICE_COUNT 12
|
|
|
|
#define COMMAND_FILE "/usr/local/nagios/var/rw/nagios.cmd"
|
|
#define SERVICES_FILE "/usr/local/nagios/etc/services.cfg"
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
char check_name[MAX_CHARS];
|
|
char ent_type[MAX_CHARS];
|
|
char input_buffer[MAX_CHARS];
|
|
char host_name[MAX_CHARS];
|
|
char service_name[MAX_CHARS];
|
|
char state[MAX_CHARS];
|
|
char state_type[MAX_CHARS];
|
|
char temp_input[MAX_CHARS];
|
|
char temp_string[MAX_CHARS];
|
|
char test_host[MAX_CHARS];
|
|
|
|
char *temp_var;
|
|
|
|
FILE *command_fp;
|
|
FILE *services_fp;
|
|
|
|
int attempt;
|
|
int i;
|
|
|
|
time_t current_time;
|
|
|
|
strcpy(state,argv[1]);
|
|
strcpy(state_type,argv[2]);
|
|
attempt=atoi(argv[3]);
|
|
strcpy(host_name,argv[4]);
|
|
|
|
if(strcmp(state,"OK") == 0)
|
|
{
|
|
services_fp=fopen(SERVICES_FILE,"r");
|
|
command_fp=fopen(COMMAND_FILE,"a");
|
|
while((fgets(input_buffer,MAX_CHARS-1,services_fp)) != NULL)
|
|
{
|
|
if(input_buffer[0]=='#' || input_buffer[0]=='\x0' || input_buffer[0]=='\n' || input_buffer[0]=='\r')
|
|
{
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
strcpy(temp_input,input_buffer);
|
|
strcpy(temp_string,strtok(temp_input,"="));
|
|
strcpy(ent_type,strtok(temp_string,"["));
|
|
if(strcmp(ent_type,"service") == 0)
|
|
{
|
|
strcpy(test_host,strtok(NULL,"]"));
|
|
if(strcmp(test_host,host_name) == 0)
|
|
{
|
|
temp_var=strtok(input_buffer,"=");
|
|
strcpy(service_name,strtok(NULL,";"));
|
|
for(i=1;i<=SERVICE_COUNT;i++)
|
|
{
|
|
temp_var=strtok(NULL,";");
|
|
}
|
|
strcpy(check_name,strtok(temp_var,"!"));
|
|
if(strcmp(check_name,"check_nrpe") == 0)
|
|
{
|
|
time(¤t_time);
|
|
fprintf(command_fp,"[%lu] ENABLE_SVC_CHECK;%s;%s\n",current_time,host_name,service_name);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
fclose(command_fp);
|
|
fclose(services_fp);
|
|
}
|
|
else if(strcmp(state,"CRITICAL") == 0)
|
|
{
|
|
if(attempt == 3)
|
|
{
|
|
services_fp=fopen(SERVICES_FILE,"r");
|
|
command_fp=fopen(COMMAND_FILE,"a");
|
|
while((fgets(input_buffer,MAX_CHARS-1,services_fp)) != NULL)
|
|
{
|
|
if(input_buffer[0]=='#' || input_buffer[0]=='\x0' || input_buffer[0]=='\n' || input_buffer[0]=='\r')
|
|
{
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
strcpy(temp_input,input_buffer);
|
|
strcpy(temp_string,strtok(temp_input,"="));
|
|
strcpy(ent_type,strtok(temp_string,"["));
|
|
if(strcmp(ent_type,"service") == 0)
|
|
{
|
|
strcpy(test_host,strtok(NULL,"]"));
|
|
if(strcmp(test_host,host_name) == 0)
|
|
{
|
|
temp_var=strtok(input_buffer,"=");
|
|
strcpy(service_name,strtok(NULL,";"));
|
|
for(i=1;i<=SERVICE_COUNT;i++)
|
|
{
|
|
temp_var=strtok(NULL,";");
|
|
}
|
|
strcpy(check_name,strtok(temp_var,"!"));
|
|
if(strcmp(check_name,"check_nrpe") == 0)
|
|
{
|
|
time(¤t_time);
|
|
fprintf(command_fp,"[%lu] DISABLE_SVC_CHECK;%s;%s\n",current_time,host_name,service_name);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
fclose(command_fp);
|
|
fclose(services_fp);
|
|
}
|
|
}
|
|
return 0;
|
|
}
|