X-Git-Url: http://git.subgeniuskitty.com/icmpmonitor/.git/blobdiff_plain/684f5d952a4ecfad2d2fad808498a87b5184035b..5cc080f4770b79865ee068046af1eecac151e464:/icmpmonitor.c diff --git a/icmpmonitor.c b/icmpmonitor.c index d222d61..fe7fb12 100644 --- a/icmpmonitor.c +++ b/icmpmonitor.c @@ -1,6 +1,7 @@ /* * Monitor hosts using ICMP "echo" and notify when down. - * + * TODO: Write a better description. + * * © 2019 Aaron Taylor * © 1999 Vadim Zaliva * © 1989 The Regents of the University of California & Mike Muuss @@ -9,6 +10,7 @@ #include #include +#include #include #include #include @@ -22,96 +24,44 @@ #include #include #include - -#include "cfg.h" +#include "iniparser/iniparser.h" #define MAXPACKETSIZE (65536 - 60 - 8) /* TODO: What are the magic numbers? */ #define DEFAULTDATALEN (64 - 8) /* TODO: What are the magic numbers? */ +/* Must be larger than the length of the longest configuration key (currently 'start_condition'). */ +#define MAXCONFKEYLEN 20 + +/* One struct per host as listed in the config file. */ struct monitor_host { /* From the config file */ char * name; int ping_interval; int max_delay; - char * upcmd; - char * downcmd; + char * up_cmd; + char * down_cmd; /* Calculated values */ int socket; struct timeval last_ping_received; struct timeval last_ping_sent; - /* TODO: Replace the following two values with a single boolean */ - int up; - int down; + bool host_up; struct sockaddr_in dest; - unsigned int sentpackets; - unsigned int recvdpackets; - - /* linked list */ + unsigned int sent_packets; + unsigned int recvd_packets; + + /* Linked list */ struct monitor_host * next; }; -/* protos */ -static int gethostaddr(const char * name); -static void read_hosts(const char * cfg_file_name); -static void init_hosts(void); -static void get_response(void); -static void pinger(int); -static int in_cksum(unsigned short * addr, int len); -static void read_icmp_data(struct monitor_host * p); -static void tvsub(struct timeval * out, struct timeval * in); -static int gcd(int x, int y); - -/* globals */ -static struct monitor_host ** hosts = NULL; -static int isVerbose = 0; -static int keepBanging = 0; -static unsigned short ident; -static int send_delay = 1; - -int -main(int argc, char ** argv) -{ - extern char * optarg; - extern int optind; - char * cfgfile = NULL; - int param; - - while ((param = getopt(argc, argv, "rvf:")) != -1) { - switch(param) { - case 'v': - isVerbose = 1; - break; - case 'r': - keepBanging = 1; - break; - case 'f': - cfgfile=strdup(optarg); - break; - default: - fprintf(stderr,"Usage: icmpmonitor [-v] [-r] [-f cfgfile]\n"); - exit(EXIT_FAILURE); - } - } - - if (!cfgfile) { - fprintf(stderr, "ERROR: No config file specified.\n"); - exit(EXIT_FAILURE); - } - - read_hosts(cfgfile); - - init_hosts(); - - ident = getpid() & 0xFFFF; - - signal(SIGALRM, pinger); - alarm(send_delay); - - get_response(); - - exit(EXIT_SUCCESS); -} +/* Globals */ + /* Since the program is based around signals, a linked list of hosts is maintained here. */ + static struct monitor_host ** hosts = NULL; + /* Set by command line flags. */ + static bool verbose = false; + static bool retry_down_cmd = false; + /* TODO: Get rid of this global. */ + static int send_delay = 1; /* * Checksum routine for Internet Protocol family headers @@ -123,7 +73,7 @@ in_cksum(unsigned short * addr, int len) unsigned short * w = addr; int sum = 0; unsigned short answer = 0; - + /* * Our algorithm is simple, using a 32 bit accumulator (sum), we add * sequential 16 bit words to it, and at the end, fold back all the @@ -133,13 +83,13 @@ in_cksum(unsigned short * addr, int len) sum += *w++; nleft -= 2; } - + /* mop up an odd byte, if necessary */ if (nleft == 1) { *(u_char *)(&answer) = *(u_char *)w; sum += answer; } - + /* add back carry outs from top 16 bits to low 16 bits */ sum = (sum >> 16) + (sum & 0xffff); /* add hi 16 to low 16 */ sum += (sum >> 16); /* add carry */ @@ -147,6 +97,21 @@ in_cksum(unsigned short * addr, int len) return(answer); } +/* + * Subtracts two timeval structs. + * Ensure out >= in. + * Modifies out = out - in. + */ +static void +tv_sub(register struct timeval * out, register struct timeval * in) +{ + if ((out->tv_usec -= in->tv_usec) < 0) { + --out->tv_sec; + out->tv_usec += 1000000; + } + out->tv_sec -= in->tv_sec; +} + /* * Compose and transmit an ICMP ECHO REQUEST packet. The IP packet * will be added on by the kernel. The ID field is our UNIX process ID, @@ -166,97 +131,60 @@ pinger(int ignore) while (p) { if (p->socket != -1) { struct timeval now; - + gettimeofday(&now, (struct timezone *) NULL); - tvsub(&now, &p->last_ping_received); - - if (now.tv_sec > (p->max_delay + p->ping_interval)) { - p->up = 0; - if ((!p->down) || keepBanging) { - p->down = 1; - - if (isVerbose) printf("INFO: Host %s is down. Executing DOWN command.\n", p->name); + tv_sub(&now, &p->last_ping_received); + + if (now.tv_sec > (p->max_delay + p->ping_interval)) { + if ((p->host_up) || retry_down_cmd) { + if (verbose) printf("INFO: Host %s stopped responding. Executing DOWN command.\n", p->name); + p->host_up = false; if (!fork()) { - system(p->downcmd); + system(p->down_cmd); exit(EXIT_SUCCESS); } else { wait(NULL); } } } - + gettimeofday(&now, (struct timezone *) NULL); - tvsub(&now, &p->last_ping_sent); - + tv_sub(&now, &p->last_ping_sent); + if (now.tv_sec > p->ping_interval) { /* Time to send ping */ icp = (struct icmp *) outpack; icp->icmp_type = ICMP_ECHO; icp->icmp_code = 0; icp->icmp_cksum = 0; icp->icmp_seq = p->socket; - icp->icmp_id = ident; + icp->icmp_id = getpid() & 0xFFFF; + + if (verbose) printf("INFO: Sending ICMP packet to %s.\n", p->name); - if (isVerbose) printf("INFO: Sending ICMP packet to %s.\n", p->name); - gettimeofday((struct timeval *) &outpack[8], (struct timezone *) NULL); - + cc = DEFAULTDATALEN + 8; /* skips ICMP portion */ - + /* compute ICMP checksum */ icp->icmp_cksum = in_cksum((unsigned short *) icp, cc); - + i = sendto(p->socket, (char *) outpack, cc, 0, (const struct sockaddr *) (&p->dest), sizeof(struct sockaddr)); - + gettimeofday(&p->last_ping_sent, (struct timezone *) NULL); - + if (i < 0 || i != cc) { if (i<0) fprintf(stderr, "WARN: Failed sending ICMP packet to %s.\n", p->name); } - p->sentpackets++; + p->sent_packets++; } } p = p->next; } - + signal(SIGALRM, pinger); /* restore handler */ alarm(send_delay); } -static void -get_response(void) -{ - fd_set rfds; - int retval, maxd = -1; - struct monitor_host * p; - - while (1) { - p = hosts[0]; - FD_ZERO(&rfds); - while (p) { - if (p->socket != -1) { - if (p->socket > maxd) maxd=p->socket; - FD_SET(p->socket, &rfds); - } - p=p->next; - } - - retval = select(maxd+1, &rfds, NULL, NULL, NULL); - if (retval < 0) { - /* Intentionally empty. We arrive here when interrupted by a signal. No action should be taken. */ - } else { - if (retval > 0) { - p = hosts[0]; - while (p) { - if (p->socket!=-1 && FD_ISSET(p->socket, &rfds)) read_icmp_data(p); - p = p->next; - } - } else { - /* TODO: How to handle this error? */ - } - } - } -} - static void read_icmp_data(struct monitor_host * p) { @@ -265,42 +193,41 @@ read_icmp_data(struct monitor_host * p) struct sockaddr_in from; struct ip * ip; struct icmp * icmp; - struct timeval tv ; + struct timeval tv; unsigned char buf[MAXPACKETSIZE]; gettimeofday(&tv, (struct timezone *) NULL); - + fromlen = sizeof(from); if ((cc = recvfrom(p->socket, buf, sizeof(buf), 0, (struct sockaddr *)&from, &fromlen)) < 0) { if (errno != EINTR) fprintf(stderr, "WARN: Error reading ICMP data from %s.\n", p->name); return; - } + } - /* check IP header actual len */ - ip = (struct ip *) buf; - iphdrlen = ip->ip_hl << 2; + /* check IP header actual len */ + ip = (struct ip *) buf; + iphdrlen = ip->ip_hl << 2; icmp = (struct icmp *) (buf + iphdrlen); - + if (cc < iphdrlen + ICMP_MINLEN) { fprintf(stderr, "WARN: Received short packet from %s.\n", p->name); return; } - - if (icmp->icmp_type == ICMP_ECHOREPLY && icmp->icmp_id == ident && icmp->icmp_seq == p->socket) { - p->recvdpackets++; + + if (icmp->icmp_type == ICMP_ECHOREPLY && icmp->icmp_id == (getpid() & 0xFFFF) && icmp->icmp_seq == p->socket) { + p->recvd_packets++; memcpy(&p->last_ping_received, &tv, sizeof(tv)); - - tvsub(&tv, (struct timeval *) &icmp->icmp_data[0]); + + tv_sub(&tv, (struct timeval *) &icmp->icmp_data[0]); delay = tv.tv_sec * 1000 + (tv.tv_usec / 1000); - - if (isVerbose) printf("INFO: Got ICMP reply from %s in %d ms.\n", p->name, delay); - p->down = 0; - if (!p->up) { - p->up = 1; - if (isVerbose) printf("INFO: Host %s is up. Executing UP command.\n", p->name); + + if (verbose) printf("INFO: Got ICMP reply from %s in %d ms.\n", p->name, delay); + if (!p->host_up) { + if (verbose) printf("INFO: Host %s started responding. Executing UP command.\n", p->name); + p->host_up = true; if (!fork()) { - system(p->upcmd); + system(p->up_cmd); exit(EXIT_SUCCESS); } else { wait(NULL); @@ -312,87 +239,135 @@ read_icmp_data(struct monitor_host * p) } static void -read_hosts(const char * cfg_file_name) +get_response(void) { - int i, n = 0; - struct Cfg * cfg; - - if ((cfg = readcfg(cfg_file_name)) == NULL) { - fprintf(stderr, "ERROR: Failed to read config.\n"); - exit(EXIT_FAILURE); - } - - if (cfg->nelements) { - hosts = malloc(sizeof(struct monitor_host *) * cfg->nelements); - for (i = 0; i < cfg->nelements; i++) { - if (cfg->dict[i]->nvalues < 4) { - fprintf(stderr, "ERROR: Not enough fields in record %d of cfg file. Got %d.\n", n, cfg->dict[i]->nvalues+1); - exit(EXIT_FAILURE); - } else if (cfg->dict[i]->nvalues>5) { - fprintf(stderr, "ERROR: Too many fields in record %d of cfg file. Got %d.\n", n, cfg->dict[i]->nvalues+1); - exit(EXIT_FAILURE); + fd_set rfds; + int retval, maxd = -1; + struct monitor_host * p; + + while (1) { + p = hosts[0]; + FD_ZERO(&rfds); + while (p) { + if (p->socket != -1) { + if (p->socket > maxd) maxd=p->socket; + FD_SET(p->socket, &rfds); } - - hosts[n] = malloc(sizeof(struct monitor_host)); - hosts[n]->name = strdup(cfg->dict[i]->name); - hosts[n]->ping_interval = atoi (cfg->dict[i]->value[0]); - hosts[n]->max_delay = atoi (cfg->dict[i]->value[1]); - hosts[n]->upcmd = strdup(cfg->dict[i]->value[2]); - hosts[n]->downcmd = strdup(cfg->dict[i]->value[3]); - - if (cfg->dict[i]->nvalues == 4) { - hosts[n]->down = 0; - hosts[n]->up = 1; - } else if (strcmp(cfg->dict[i]->value[4], "up") == 0) { - hosts[n]->down = 0; - hosts[n]->up = 1; - } else if (strcmp(cfg->dict[i]->value[4], "down") == 0) { - hosts[n]->down = 1; - hosts[n]->up = 0; - } else if (strcmp(cfg->dict[i]->value[4], "auto") == 0) { - hosts[n]->down = 1; - hosts[n]->up = 1; - } else if (strcmp(cfg->dict[i]->value[4], "none") == 0) { - hosts[n]->down = 0; - hosts[n]->up = 0; + p = p->next; + } + + retval = select(maxd+1, &rfds, NULL, NULL, NULL); + if (retval < 0) { + /* Intentionally empty. We arrive here when interrupted by a signal. No action should be taken. */ } else { - fprintf(stderr, "ERROR: Illegal value %s in record %d for startup condition.\n", cfg->dict[i]->value[4], n); - exit(EXIT_FAILURE); + if (retval > 0) { + p = hosts[0]; + while (p) { + if (p->socket!=-1 && FD_ISSET(p->socket, &rfds)) read_icmp_data(p); + p = p->next; + } + } else { + /* TODO: How to handle this error? */ + } } - hosts[n]->sentpackets = 0; - hosts[n]->recvdpackets = 0; + } +} - hosts[n]->socket = -1; - hosts[n]->next = NULL; - if (n > 0) hosts[n-1]->next=hosts[n]; - gettimeofday(&(hosts[n]->last_ping_received), (struct timezone *)NULL); +/* + * Parses `config_file` and creates relevant entries under the global variable `hosts`. + */ +static void +parse_config(const char * conf_file) +{ + dictionary * conf = iniparser_load(conf_file); + if (conf == NULL) { + fprintf(stderr, "ERROR: Unable to parse configuration file %s.\n", conf_file); + exit(EXIT_FAILURE); + } - n++; - } + int host_count = iniparser_getnsec(conf); + if (host_count < 1 ) { + fprintf(stderr, "ERROR: Unable to determine number of hosts in configuration file.\n"); + exit(EXIT_FAILURE); } - freecfg(cfg); + hosts = malloc(sizeof(struct monitor_host *) * host_count); + for (int i=0; i < host_count; i++) { + /* Allocate a buffer large enough to hold the full 'section:key' string. */ + int section_len = strlen(iniparser_getsecname(conf, i)); + char * key_buf = malloc(section_len + 1 + MAXCONFKEYLEN + 1); + strcpy(key_buf, iniparser_getsecname(conf, i)); + key_buf[section_len++] = ':'; + + hosts[i] = malloc(sizeof(struct monitor_host)); + for (int k=0; k<6; k++) { + key_buf[section_len] = '\0'; /* Reuse the section name and colon on each pass through this loop. */ + switch (k) { + case 0: + strncat(key_buf, "host", MAXCONFKEYLEN); + hosts[i]->name = strdup(iniparser_getstring(conf, key_buf, NULL)); + break; + case 1: + strncat(key_buf, "interval", MAXCONFKEYLEN); + hosts[i]->ping_interval = iniparser_getint(conf, key_buf, -1); + break; + case 2: + strncat(key_buf, "max_delay", MAXCONFKEYLEN); + hosts[i]->max_delay = iniparser_getint(conf, key_buf, -1); + break; + case 3: + strncat(key_buf, "up_cmd", MAXCONFKEYLEN); + hosts[i]->up_cmd = strdup(iniparser_getstring(conf, key_buf, NULL)); + break; + case 4: + strncat(key_buf, "down_cmd", MAXCONFKEYLEN); + hosts[i]->down_cmd = strdup(iniparser_getstring(conf, key_buf, NULL)); + break; + case 5: + /* TODO: Parse for up/down/auto in start_condition. */ + /* TODO: Do a host up/down check if necessary. */ + hosts[i]->host_up = true; + break; + } + } + /* TODO: Do I want to make any checks for up_cmd, down_cmd, and start_condition? */ + if (hosts[i]->name == NULL || hosts[i]->ping_interval == -1 || hosts[i]->max_delay == -1) { + fprintf(stderr, "ERROR: Problems parsing section %s.\n", iniparser_getsecname(conf, i)); + exit(EXIT_FAILURE); + } - if (n <= 0) { - fprintf(stderr, "ERROR: No hosts defined in cfg file, exiting.\n"); - exit(EXIT_FAILURE); + hosts[i]->sent_packets = 0; + hosts[i]->recvd_packets = 0; + hosts[i]->socket = -1; + hosts[i]->next = NULL; + if (i>0) hosts[i-1]->next = hosts[i]; + gettimeofday(&(hosts[i]->last_ping_received), (struct timezone *) NULL); } + iniparser_freedict(conf); } static int -gethostaddr(const char * name) +get_host_addr(const char * name) { static int res; struct hostent * he; - + if ((res = inet_addr(name)) < 0) { he = gethostbyname(name); if (!he) return -1; memcpy(&res, he->h_addr, he->h_length); - } + } return(res); } +static int +gcd(int x, int y) +{ + int remainder = x % y; + if (remainder == 0) return y; + return gcd(y, remainder); +} + static void init_hosts(void) { @@ -404,11 +379,11 @@ init_hosts(void) fprintf(stderr, "ERROR: Unknown protocol: icmp.\n"); exit(EXIT_FAILURE); } - + while (p) { bzero(&p->dest, sizeof(p->dest)); p->dest.sin_family = AF_INET; - if ((p->dest.sin_addr.s_addr = gethostaddr(p->name)) <= 0) { + if ((p->dest.sin_addr.s_addr = get_host_addr(p->name)) <= 0) { fprintf(stderr, "WARN: Can't resolve host. Skipping client %s.\n", p->name); p->socket=-1; } else { @@ -430,25 +405,49 @@ init_hosts(void) } } -/* - * Subtracts two timeval structs. - * Ensure out >= in. - * Modifies out = out - in. - */ -static void -tvsub(register struct timeval * out, register struct timeval * in) +void +print_usage(void) { - if ((out->tv_usec -= in->tv_usec) < 0) { - --out->tv_sec; - out->tv_usec += 1000000; + fprintf(stderr,"Usage: icmpmonitor [-v] [-r] [-f cfgfile]\n"); +} + +void +parse_params(int argc, char ** argv) +{ + int param; + while ((param = getopt(argc, argv, "rvf:")) != -1) { + switch(param) { + case 'v': + verbose = true; + break; + case 'r': + retry_down_cmd = true; + break; + case 'f': + parse_config(optarg); + break; + default: + print_usage(); + exit(EXIT_FAILURE); + } + } + if (hosts == NULL) { + fprintf(stderr, "ERROR: Unable to parse a config file.\n"); + exit(EXIT_FAILURE); } - out->tv_sec -= in->tv_sec; } -static int -gcd(int x, int y) +int +main(int argc, char ** argv) { - int remainder = x % y; - if (remainder == 0) return y; - return gcd(y, remainder); + parse_params(argc, argv); + + init_hosts(); + + signal(SIGALRM, pinger); + alarm(send_delay); + + get_response(); + + exit(EXIT_SUCCESS); }