X-Git-Url: http://git.subgeniuskitty.com/icmpmonitor/.git/blobdiff_plain/d283bb629d5158ab029cb33458b52628b6162e57..468768fafcbb208e22b7882bed3bc0ea37ef1bcd:/icmpmonitor.c diff --git a/icmpmonitor.c b/icmpmonitor.c index efa6d71..72ea1cf 100644 --- a/icmpmonitor.c +++ b/icmpmonitor.c @@ -100,134 +100,114 @@ checksum(const uint16_t * data) } /* - * Calculate difference between two timeval structs to within one second. + * Calculate difference (a-b) between two timeval structs. */ void timeval_diff(struct timeval * a, const struct timeval * b) { - assert(a->tv_sec >= b->tv_sec); + if (a->tv_usec < b->tv_usec) { + a->tv_sec--; + a->tv_usec += 1000000; + } + a->tv_usec -= b->tv_usec; a->tv_sec -= b->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, - * and the sequence number is an ascending integer. The first 8 bytes - * of the data portion are used to hold a UNIX "timeval" struct in VAX - * byte-order, to compute the round-trip time. + * This function iterates over the list of hosts, pinging any which are due. */ -static void -pinger(int ignore) +void +pinger(int ignore) /* Dummy parameter since this function registers as a signal handler. */ { - int i; - struct icmp * icp; - struct host_entry * p = first_host_in_list; - unsigned char outpack[MAXPACKETSIZE]; /* Use char so this can be aliased later. */ - - while (p) { - if (p->socket != -1) { - struct timeval now; - - gettimeofday(&now, (struct timezone *) NULL); - timeval_diff(&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->down_cmd); - exit(EXIT_SUCCESS); - } else { - wait(NULL); - } - } - } + assert(first_host_in_list); - gettimeofday(&now, (struct timezone *) NULL); - timeval_diff(&now, &p->last_ping_sent); + struct icmp * icmp_packet; + struct host_entry * host = first_host_in_list; + unsigned char packet[IP_PACKET_MAX_BYTES]; /* Use char so this can be aliased later. */ - 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 = getpid() & 0xFFFF; + while (host) { + struct timeval elapsed_time; + gettimeofday(&elapsed_time, NULL); + timeval_diff(&elapsed_time, &host->last_ping_received); - if (verbose) printf("INFO: Sending ICMP packet to %s.\n", p->name); + if ((elapsed_time.tv_sec > host->max_delay) && (host->host_up || retry_down_cmd)) { + if (verbose) printf("INFO: Host %s stopped responding. Executing DOWN command.\n", host->name); + host->host_up = false; + if (!fork()) { + system(host->down_cmd); + exit(EXIT_SUCCESS); + } + } - gettimeofday((struct timeval *) &outpack[8], (struct timezone *) NULL); + if (elapsed_time.tv_sec > host->ping_interval) { + if (verbose) printf("INFO: Sending ICMP packet to %s.\n", host->name); - int cc = DEFAULTDATALEN + 8; /* skips ICMP portion */ + icmp_packet = (struct icmp *) packet; + icmp_packet->icmp_type = ICMP_ECHO; + icmp_packet->icmp_code = 0; + icmp_packet->icmp_cksum = 0; + icmp_packet->icmp_seq = host->socket; + icmp_packet->icmp_id = getpid() & 0xFFFF; - icp->icmp_cksum = checksum((uint16_t *) outpack); + /* Write a timestamp struct in the packet's data segment for use in calculating travel times. */ + gettimeofday((struct timeval *) &packet[ICMP_ECHO_HEADER_BYTES], NULL); - i = sendto(p->socket, (char *) outpack, cc, 0, (const struct sockaddr *) (&p->dest), sizeof(struct sockaddr)); + icmp_packet->icmp_cksum = checksum((uint16_t *) packet); - gettimeofday(&p->last_ping_sent, (struct timezone *) NULL); + size_t bytes_sent = sendto(host->socket, packet, ICMP_ECHO_PACKET_BYTES, 0, + (const struct sockaddr *) &host->dest, + sizeof(struct sockaddr)); - if (i < 0 || i != cc) { - if (i<0) fprintf(stderr, "WARN: Failed sending ICMP packet to %s.\n", p->name); - } + if (bytes_sent == ICMP_ECHO_PACKET_BYTES) { + gettimeofday(&host->last_ping_sent, NULL); + } else { + fprintf(stderr, "WARN: Failed sending ICMP packet to %s.\n", host->name); } } - p = p->next; + host = host->next; } - signal(SIGALRM, pinger); /* restore handler */ + signal(SIGALRM, pinger); alarm(TIMER_RESOLUTION); } -static void -read_icmp_data(struct host_entry * p) +void +read_icmp_data(struct host_entry * host) { - int cc, iphdrlen, delay; - socklen_t fromlen; - struct sockaddr_in from; - struct ip * ip; - struct icmp * icmp; - struct timeval tv; - unsigned char buf[MAXPACKETSIZE]; - - gettimeofday(&tv, (struct timezone *) NULL); + struct timeval now; + gettimeofday(&now, 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); + struct sockaddr_in from; + socklen_t fromlen = sizeof(from); + int bytes; + unsigned char packet[IP_PACKET_MAX_BYTES]; /* Use char so this can be aliased later. */ + if ((bytes = recvfrom(host->socket, packet, sizeof(packet), 0, (struct sockaddr *) &from, &fromlen)) < 0) { + if (errno != EINTR) fprintf(stderr, "WARN: Error reading ICMP data from %s.\n", host->name); return; } - /* check IP header actual len */ - ip = (struct ip *) buf; - iphdrlen = ip->ip_hl << 2; - icmp = (struct icmp *) (buf + iphdrlen); + struct ip * ip = (struct ip *) packet; + int iphdrlen = ip->ip_hl << 2; + struct icmp * icmp = (struct icmp *) (packet + iphdrlen); - if (cc < iphdrlen + ICMP_MINLEN) { - fprintf(stderr, "WARN: Received short packet from %s.\n", p->name); + if (bytes < iphdrlen + ICMP_MINLEN) { + fprintf(stderr, "WARN: Received short packet from %s.\n", host->name); return; } - if (icmp->icmp_type == ICMP_ECHOREPLY && icmp->icmp_id == (getpid() & 0xFFFF) && icmp->icmp_seq == p->socket) { - - memcpy(&p->last_ping_received, &tv, sizeof(tv)); - - timeval_diff(&tv, (struct timeval *) &icmp->icmp_data[0]); - delay = tv.tv_sec * 1000 + (tv.tv_usec / 1000); - - if (verbose) printf("INFO: Got ICMP reply from %s.\n", p->name); - if (!p->host_up) { - if (verbose) printf("INFO: Host %s started responding. Executing UP command.\n", p->name); - p->host_up = true; + if (icmp->icmp_type == ICMP_ECHOREPLY && icmp->icmp_id == (getpid() & 0xFFFF) && icmp->icmp_seq == host->socket) { + memcpy(&host->last_ping_received, &now, sizeof(now)); + if (verbose) printf("INFO: Got ICMP reply from %s.\n", host->name); + if (!host->host_up) { + if (verbose) printf("INFO: Host %s started responding. Executing UP command.\n", host->name); + host->host_up = true; if (!fork()) { - system(p->up_cmd); + system(host->up_cmd); exit(EXIT_SUCCESS); - } else { - wait(NULL); } } } else { - /* TODO: Do anything here? */ + /* The packet isn't what we expected. Ignore it and move on. */ } }