diff options
| author | Stanley Hopcroft <stanleyhopcroft@users.sourceforge.net> | 2004-12-02 08:44:36 +0000 |
|---|---|---|
| committer | Stanley Hopcroft <stanleyhopcroft@users.sourceforge.net> | 2004-12-02 08:44:36 +0000 |
| commit | c1d25fe1ee355f3394cab0bd340896df84e9d549 (patch) | |
| tree | 98c76c7d0e647590d30a3ebebd7135ff068427e4 | |
| parent | 3c925e8c0c01f6b4ae966183af19d22b46046785 (diff) | |
| download | monitoring-plugins-c1d25fe1ee355f3394cab0bd340896df84e9d549.tar.gz | |
check_icmp plugin from A Ericsson
git-svn-id: https://nagiosplug.svn.sourceforge.net/svnroot/nagiosplug/nagiosplug/trunk@975 f882894a-f735-0410-b71e-b25c423dba1c
| -rw-r--r-- | plugins/check_icmp.c | 1381 |
1 files changed, 1381 insertions, 0 deletions
diff --git a/plugins/check_icmp.c b/plugins/check_icmp.c new file mode 100644 index 00000000..45bfcc5e --- /dev/null +++ b/plugins/check_icmp.c | |||
| @@ -0,0 +1,1381 @@ | |||
| 1 | /* | ||
| 2 | * check_icmp - A hack of fping2 to work with nagios. | ||
| 3 | * This way we don't have to use the output parser. | ||
| 4 | * | ||
| 5 | * VIEWING NOTES: | ||
| 6 | * This file was formatted with tab indents at a tab stop of 4. | ||
| 7 | * | ||
| 8 | * It is highly recommended that your editor is set to this | ||
| 9 | * tab stop setting for viewing and editing. | ||
| 10 | * | ||
| 11 | * COPYLEFT; | ||
| 12 | * This programs copyright status is currently undetermined. Much of | ||
| 13 | * the code in it comes from the fping2 program which used to be licensed | ||
| 14 | * under the Stanford General Software License (available at | ||
| 15 | * http://graphics.stanford.edu/software/license.html). It is presently | ||
| 16 | * unclear what license (if any) applies to the original code at the | ||
| 17 | * moment. | ||
| 18 | * | ||
| 19 | * The fping website can be found at http://www.fping.com | ||
| 20 | */ | ||
| 21 | |||
| 22 | #include <stdio.h> | ||
| 23 | #include <errno.h> | ||
| 24 | #include <time.h> | ||
| 25 | #include <signal.h> | ||
| 26 | |||
| 27 | #include <unistd.h> | ||
| 28 | |||
| 29 | #include <stdlib.h> | ||
| 30 | |||
| 31 | #include <string.h> | ||
| 32 | #include <stddef.h> | ||
| 33 | |||
| 34 | #include <sys/types.h> | ||
| 35 | #include <sys/time.h> | ||
| 36 | #include <sys/socket.h> | ||
| 37 | |||
| 38 | #include <sys/file.h> | ||
| 39 | |||
| 40 | #include <netinet/in_systm.h> | ||
| 41 | #include <netinet/in.h> | ||
| 42 | |||
| 43 | /* Linux has bizarre ip.h and ip_icmp.h */ | ||
| 44 | /* Taken from the fping distro. Thank you. */ | ||
| 45 | #if defined( __linux__ ) | ||
| 46 | #include "linux.h" | ||
| 47 | #else | ||
| 48 | #include <netinet/ip.h> | ||
| 49 | #include <netinet/ip_icmp.h> | ||
| 50 | #endif /* defined(__linux__) */ | ||
| 51 | |||
| 52 | #include <arpa/inet.h> | ||
| 53 | #include <netdb.h> | ||
| 54 | |||
| 55 | /* RS6000 has sys/select.h */ | ||
| 56 | #ifdef HAVE_SYS_SELECT_H | ||
| 57 | #include <sys/select.h> | ||
| 58 | #endif /* HAVE_SYS_SELECT_H */ | ||
| 59 | |||
| 60 | /* rta threshold values can't be larger than MAXTTL seconds */ | ||
| 61 | #ifndef MAXTTL | ||
| 62 | # define MAXTTL 255 | ||
| 63 | #endif | ||
| 64 | #ifndef IPDEFTTL | ||
| 65 | # define IPDEFTTL 64 | ||
| 66 | #endif | ||
| 67 | |||
| 68 | /*** externals ***/ | ||
| 69 | extern char *optarg; | ||
| 70 | extern int optind, opterr; | ||
| 71 | |||
| 72 | /*** Constants ***/ | ||
| 73 | #define REV_DATE "2004-09-06" | ||
| 74 | #define EMAIL "ae@op5.se" | ||
| 75 | #define VERSION "0.8" | ||
| 76 | |||
| 77 | #ifndef INADDR_NONE | ||
| 78 | # define INADDR_NONE 0xffffffU | ||
| 79 | #endif | ||
| 80 | |||
| 81 | /*** Ping packet defines ***/ | ||
| 82 | /* data added after ICMP header for our nefarious purposes */ | ||
| 83 | typedef struct ping_data { | ||
| 84 | unsigned int ping_count; /* counts up to -[n|p] count or 1 */ | ||
| 85 | struct timeval ping_ts; /* time sent */ | ||
| 86 | } PING_DATA; | ||
| 87 | |||
| 88 | #define MIN_PING_DATA sizeof(PING_DATA) | ||
| 89 | #define MAX_IP_PACKET 65536 /* (theoretical) max IP packet size */ | ||
| 90 | #define SIZE_IP_HDR 20 | ||
| 91 | #define SIZE_ICMP_HDR ICMP_MINLEN /* from ip_icmp.h */ | ||
| 92 | #define MAX_PING_DATA (MAX_IP_PACKET - SIZE_IP_HDR - SIZE_ICMP_HDR) | ||
| 93 | |||
| 94 | /* | ||
| 95 | * Interval is the minimum amount of time between sending a ping packet to | ||
| 96 | * any host. | ||
| 97 | * | ||
| 98 | * Perhost_interval is the minimum amount of time between sending a ping | ||
| 99 | * packet to a particular responding host | ||
| 100 | * | ||
| 101 | * Timeout is the initial amount of time between sending a ping packet to | ||
| 102 | * a particular non-responding host. | ||
| 103 | * | ||
| 104 | * Retry is the number of ping packets to send to a non-responding host | ||
| 105 | * before giving up (in is-it-alive mode). | ||
| 106 | * | ||
| 107 | * Backoff factor is how much longer to wait on successive retries. | ||
| 108 | */ | ||
| 109 | #ifndef DEFAULT_INTERVAL | ||
| 110 | #define DEFAULT_INTERVAL 25 /* default time between packets (msec) */ | ||
| 111 | #endif | ||
| 112 | |||
| 113 | #ifndef DEFAULT_RETRY | ||
| 114 | #define DEFAULT_RETRY 1 /* number of times to retry a host */ | ||
| 115 | #endif | ||
| 116 | |||
| 117 | #ifndef DEFAULT_TIMEOUT | ||
| 118 | # define DEFAULT_TIMEOUT 1000 | ||
| 119 | #endif | ||
| 120 | |||
| 121 | #ifndef DEFAULT_BACKOFF_FACTOR | ||
| 122 | #define DEFAULT_BACKOFF_FACTOR 1.5 /* exponential timeout factor */ | ||
| 123 | #endif | ||
| 124 | #define MIN_BACKOFF_FACTOR 1.0 /* exponential timeout factor */ | ||
| 125 | #define MAX_BACKOFF_FACTOR 5.0 /* exponential timeout factor */ | ||
| 126 | |||
| 127 | #ifndef DNS_TIMEOUT | ||
| 128 | #define DNS_TIMEOUT 1000 /* time in usec for dns retry */ | ||
| 129 | #endif | ||
| 130 | |||
| 131 | #ifndef MAX_RTA_THRESHOLD_VALUE | ||
| 132 | # define MAX_RTA_THRESHOLD_VALUE 120*1000000 /* 2 minutes should be enough */ | ||
| 133 | #endif | ||
| 134 | #ifndef MIN_RTA_THRESHOLD_VALUE | ||
| 135 | # define MIN_RTA_THRESHOLD_VALUE 10000 /* minimum RTA threshold value */ | ||
| 136 | #endif | ||
| 137 | |||
| 138 | /* sized so as to be like traditional ping */ | ||
| 139 | #define DEFAULT_PING_DATA_SIZE (MIN_PING_DATA + 44) | ||
| 140 | |||
| 141 | /* maxima and minima */ | ||
| 142 | #define MAX_COUNT 50 /* max count even if we're root */ | ||
| 143 | #define MAX_RETRY 5 | ||
| 144 | #define MIN_INTERVAL 25 /* msecs */ | ||
| 145 | #define MIN_TIMEOUT 50 /* msecs */ | ||
| 146 | |||
| 147 | /* response time array flags */ | ||
| 148 | #define RESP_WAITING -1 | ||
| 149 | #define RESP_UNUSED -2 | ||
| 150 | |||
| 151 | #define ICMP_UNREACH_MAXTYPE 15 | ||
| 152 | |||
| 153 | /* entry used to keep track of each host we are pinging */ | ||
| 154 | struct host_entry { | ||
| 155 | int i; /* index into array */ | ||
| 156 | char *name; /* name as given by user */ | ||
| 157 | char *host; /* text description of host */ | ||
| 158 | struct sockaddr_in saddr; /* internet address */ | ||
| 159 | unsigned short **pr; /* TCP port range to check for connectivity */ | ||
| 160 | struct timeval last_send_time; /* time of last packet sent */ | ||
| 161 | unsigned int num_sent; /* number of ping packets sent */ | ||
| 162 | unsigned int num_recv; /* number of pings received */ | ||
| 163 | unsigned int total_time; /* sum of response times */ | ||
| 164 | unsigned int status; /* this hosts status */ | ||
| 165 | unsigned int running; /* unset when through sending */ | ||
| 166 | unsigned int waiting; /* waiting for response */ | ||
| 167 | int *resp_times; /* individual response times */ | ||
| 168 | struct host_entry *prev, *next; /* doubly linked list */ | ||
| 169 | }; | ||
| 170 | |||
| 171 | typedef struct host_entry HOST_ENTRY; | ||
| 172 | |||
| 173 | struct host_name_list { | ||
| 174 | char *entry; | ||
| 175 | struct host_name_list *next; | ||
| 176 | }; | ||
| 177 | |||
| 178 | /* threshold structure */ | ||
| 179 | struct threshold { | ||
| 180 | unsigned int pl; /* packet loss */ | ||
| 181 | unsigned int rta; /* roundtrip time average */ | ||
| 182 | }; | ||
| 183 | typedef struct threshold threshold; | ||
| 184 | |||
| 185 | /***************************************************************************** | ||
| 186 | * Global Variables * | ||
| 187 | *****************************************************************************/ | ||
| 188 | |||
| 189 | HOST_ENTRY *rrlist = NULL; /* linked list of hosts be pinged */ | ||
| 190 | HOST_ENTRY **table = NULL; /* array of pointers to items in the list */ | ||
| 191 | HOST_ENTRY *cursor; | ||
| 192 | |||
| 193 | char *prog; /* our name */ | ||
| 194 | int ident; /* our pid, for marking icmp packets */ | ||
| 195 | int sock; /* socket */ | ||
| 196 | u_int debug = 0; | ||
| 197 | |||
| 198 | /* threshold value defaults; | ||
| 199 | * WARNING; 60% packetloss or 200 msecs round trip average | ||
| 200 | * CRITICAL; 80% packetloss or 500 msecs round trip average */ | ||
| 201 | threshold warn = {60, 200 * 1000}; | ||
| 202 | threshold crit = {80, 500 * 1000}; | ||
| 203 | |||
| 204 | /* times get *100 because all times are calculated in 10 usec units, not ms */ | ||
| 205 | unsigned int retry = DEFAULT_RETRY; | ||
| 206 | u_int timeout = DEFAULT_TIMEOUT * 100; | ||
| 207 | u_int interval = DEFAULT_INTERVAL * 100; | ||
| 208 | float backoff = DEFAULT_BACKOFF_FACTOR; | ||
| 209 | u_int select_time; /* calculated using maximum threshold rta value */ | ||
| 210 | u_int ping_data_size = DEFAULT_PING_DATA_SIZE; | ||
| 211 | u_int ping_pkt_size; | ||
| 212 | unsigned int count = 5; | ||
| 213 | unsigned int trials = 1; | ||
| 214 | |||
| 215 | /* global stats */ | ||
| 216 | int total_replies = 0; | ||
| 217 | int num_jobs = 0; /* number of hosts still to do */ | ||
| 218 | int num_hosts = 0; /* total number of hosts */ | ||
| 219 | int num_alive = 0; /* total number alive */ | ||
| 220 | int num_unreachable = 0; /* total number unreachable */ | ||
| 221 | int num_noaddress = 0; /* total number of addresses not found */ | ||
| 222 | int num_timeout = 0; /* number of timed out packets */ | ||
| 223 | int num_pingsent = 0; /* total pings sent */ | ||
| 224 | int num_pingreceived = 0; /* total pings received */ | ||
| 225 | int num_othericmprcvd = 0; /* total non-echo-reply ICMP received */ | ||
| 226 | |||
| 227 | struct timeval current_time; /* current time (pseudo) */ | ||
| 228 | struct timeval start_time; | ||
| 229 | struct timeval end_time; | ||
| 230 | struct timeval last_send_time; /* time last ping was sent */ | ||
| 231 | struct timezone tz; | ||
| 232 | |||
| 233 | /* switches */ | ||
| 234 | int generate_flag = 0; /* flag for IP list generation */ | ||
| 235 | int stats_flag, unreachable_flag, alive_flag; | ||
| 236 | int elapsed_flag, version_flag, count_flag; | ||
| 237 | int name_flag, addr_flag, backoff_flag; | ||
| 238 | int multif_flag; | ||
| 239 | |||
| 240 | /*** prototypes ***/ | ||
| 241 | void add_name(char *); | ||
| 242 | void add_addr(char *, char *, struct in_addr); | ||
| 243 | char *na_cat(char *, struct in_addr); | ||
| 244 | char *cpystr(char *); | ||
| 245 | void crash(char *); | ||
| 246 | char *get_host_by_address(struct in_addr); | ||
| 247 | int in_cksum(u_short *, int); | ||
| 248 | void u_sleep(int); | ||
| 249 | int recvfrom_wto(int, char *, int, struct sockaddr *, int); | ||
| 250 | void remove_job(HOST_ENTRY *); | ||
| 251 | void send_ping(int, HOST_ENTRY *); | ||
| 252 | long timeval_diff(struct timeval *, struct timeval *); | ||
| 253 | void usage(void); | ||
| 254 | int wait_for_reply(int); | ||
| 255 | void finish(void); | ||
| 256 | int handle_random_icmp(struct icmp *, struct sockaddr_in *); | ||
| 257 | char *sprint_tm(int); | ||
| 258 | int get_threshold(char *, threshold *); | ||
| 259 | |||
| 260 | /*** the various exit-states */ | ||
| 261 | enum { | ||
| 262 | STATE_OK = 0, | ||
| 263 | STATE_WARNING, | ||
| 264 | STATE_CRITICAL, | ||
| 265 | STATE_UNKNOWN, | ||
| 266 | STATE_DEPENDANT, | ||
| 267 | STATE_OOB | ||
| 268 | }; | ||
| 269 | /* the strings that correspond to them */ | ||
| 270 | char *status_string[STATE_OOB] = { | ||
| 271 | "OK", | ||
| 272 | "WARNING", | ||
| 273 | "CRITICAL", | ||
| 274 | "UNKNOWN", | ||
| 275 | "DEPENDANT" | ||
| 276 | }; | ||
| 277 | |||
| 278 | int status = STATE_OK; | ||
| 279 | int fin_stat = STATE_OK; | ||
| 280 | |||
| 281 | /***************************************************************************** | ||
| 282 | * Code block start * | ||
| 283 | *****************************************************************************/ | ||
| 284 | int main(int argc, char **argv) | ||
| 285 | { | ||
| 286 | int c; | ||
| 287 | u_int lt, ht; | ||
| 288 | int advance; | ||
| 289 | struct protoent *proto; | ||
| 290 | uid_t uid; | ||
| 291 | struct host_name_list *host_ptr, *host_base_ptr; | ||
| 292 | |||
| 293 | if(strchr(argv[0], '/')) prog = strrchr(argv[0], '/') + 1; | ||
| 294 | else prog = argv[0]; | ||
| 295 | |||
| 296 | /* check if we are root */ | ||
| 297 | if(geteuid()) { | ||
| 298 | printf("Root access needed (for raw sockets)\n"); | ||
| 299 | exit(STATE_UNKNOWN); | ||
| 300 | } | ||
| 301 | |||
| 302 | /* confirm that ICMP is available on this machine */ | ||
| 303 | if((proto = getprotobyname("icmp")) == NULL) | ||
| 304 | crash("icmp: unknown protocol"); | ||
| 305 | |||
| 306 | /* create raw socket for ICMP calls (ping) */ | ||
| 307 | sock = socket(AF_INET, SOCK_RAW, proto->p_proto); | ||
| 308 | |||
| 309 | if(sock < 0) | ||
| 310 | crash("can't create raw socket"); | ||
| 311 | |||
| 312 | /* drop privileges now that we have the socket */ | ||
| 313 | if((uid = getuid())) { | ||
| 314 | seteuid(uid); | ||
| 315 | } | ||
| 316 | |||
| 317 | if(argc < 2) usage(); | ||
| 318 | |||
| 319 | ident = getpid() & 0xFFFF; | ||
| 320 | |||
| 321 | if(!(host_base_ptr = malloc(sizeof(struct host_name_list)))) { | ||
| 322 | crash("Unable to allocate memory for host name list\n"); | ||
| 323 | } | ||
| 324 | host_ptr = host_base_ptr; | ||
| 325 | |||
| 326 | backoff_flag = 0; | ||
| 327 | opterr = 1; | ||
| 328 | |||
| 329 | /* get command line options | ||
| 330 | * -H denotes a host (actually ignored and picked up later) | ||
| 331 | * -h for help | ||
| 332 | * -V or -v for version | ||
| 333 | * -d to display hostnames rather than addresses | ||
| 334 | * -t sets timeout for packets and tcp connects | ||
| 335 | * -r defines retries (persistence) | ||
| 336 | * -p or -n sets packet count (5) | ||
| 337 | * -b sets packet size (56) | ||
| 338 | * -w sets warning threshhold (200,40%) | ||
| 339 | * -c sets critical threshhold (500,80%) | ||
| 340 | * -i sets interval for both packet transmissions and connect attempts | ||
| 341 | */ | ||
| 342 | #define OPT_STR "amH:hvVDdAp:n:b:r:t:i:w:c:" | ||
| 343 | while((c = getopt(argc, argv, OPT_STR)) != EOF) { | ||
| 344 | switch (c) { | ||
| 345 | case 'H': | ||
| 346 | if(!(host_ptr->entry = malloc(strlen(optarg) + 1))) { | ||
| 347 | crash("Failed to allocate memory for hostname"); | ||
| 348 | } | ||
| 349 | memset(host_ptr->entry, 0, strlen(optarg) + 1); | ||
| 350 | host_ptr->entry = memcpy(host_ptr->entry, optarg, strlen(optarg)); | ||
| 351 | if(!(host_ptr->next = malloc(sizeof(struct host_name_list)))) | ||
| 352 | crash("Failed to allocate memory for hostname"); | ||
| 353 | host_ptr = host_ptr->next; | ||
| 354 | host_ptr->next = NULL; | ||
| 355 | // add_name(optarg); | ||
| 356 | break; | ||
| 357 | /* this is recognized, but silently ignored. | ||
| 358 | * host(s) are added later on */ | ||
| 359 | |||
| 360 | break; | ||
| 361 | case 'w': | ||
| 362 | if(get_threshold(optarg, &warn)) { | ||
| 363 | printf("Illegal threshold pair specified for -%c", c); | ||
| 364 | usage(); | ||
| 365 | } | ||
| 366 | break; | ||
| 367 | |||
| 368 | case 'c': | ||
| 369 | if(get_threshold(optarg, &crit)) { | ||
| 370 | printf("Illegal threshold pair specified for -%c", c); | ||
| 371 | usage(); | ||
| 372 | } | ||
| 373 | break; | ||
| 374 | |||
| 375 | case 't': | ||
| 376 | if(!(timeout = (u_int) strtoul(optarg, NULL, 0) * 100)) { | ||
| 377 | printf("option -%c requires integer argument\n", c); | ||
| 378 | usage(); | ||
| 379 | } | ||
| 380 | break; | ||
| 381 | |||
| 382 | case 'r': | ||
| 383 | if(!(retry = (u_int) strtoul(optarg, NULL, 0))) { | ||
| 384 | printf("option -%c requires integer argument\n", c); | ||
| 385 | usage(); | ||
| 386 | } | ||
| 387 | break; | ||
| 388 | |||
| 389 | case 'i': | ||
| 390 | if(!(interval = (u_int) strtoul(optarg, NULL, 0) * 100)) { | ||
| 391 | printf("option -%c requires positive non-zero integer argument\n", c); | ||
| 392 | usage(); | ||
| 393 | } | ||
| 394 | break; | ||
| 395 | |||
| 396 | case 'p': | ||
| 397 | case 'n': | ||
| 398 | if(!(count = (u_int) strtoul(optarg, NULL, 0))) { | ||
| 399 | printf("option -%c requires positive non-zero integer argument\n", c); | ||
| 400 | usage(); | ||
| 401 | } | ||
| 402 | break; | ||
| 403 | |||
| 404 | case 'b': | ||
| 405 | if(!(ping_data_size = (u_int) strtoul(optarg, NULL, 0))) { | ||
| 406 | printf("option -%c requires integer argument\n", c); | ||
| 407 | usage(); | ||
| 408 | } | ||
| 409 | break; | ||
| 410 | |||
| 411 | case 'h': | ||
| 412 | usage(); | ||
| 413 | break; | ||
| 414 | |||
| 415 | case 'e': | ||
| 416 | elapsed_flag = 1; | ||
| 417 | break; | ||
| 418 | |||
| 419 | case 'm': | ||
| 420 | multif_flag = 1; | ||
| 421 | break; | ||
| 422 | |||
| 423 | case 'd': | ||
| 424 | name_flag = 1; | ||
| 425 | break; | ||
| 426 | |||
| 427 | case 'A': | ||
| 428 | addr_flag = 1; | ||
| 429 | break; | ||
| 430 | |||
| 431 | case 's': | ||
| 432 | stats_flag = 1; | ||
| 433 | break; | ||
| 434 | |||
| 435 | case 'u': | ||
| 436 | unreachable_flag = 1; | ||
| 437 | break; | ||
| 438 | |||
| 439 | case 'a': | ||
| 440 | alive_flag = 1; | ||
| 441 | break; | ||
| 442 | |||
| 443 | case 'v': | ||
| 444 | printf("%s: Version %s $Date$\n", prog, VERSION, REV_DATE); | ||
| 445 | printf("%s: comments to %s\n", prog, EMAIL); | ||
| 446 | exit(STATE_OK); | ||
| 447 | |||
| 448 | case 'g': | ||
| 449 | /* use IP list generation */ | ||
| 450 | /* mutex with file input or command line targets */ | ||
| 451 | generate_flag = 1; | ||
| 452 | break; | ||
| 453 | |||
| 454 | default: | ||
| 455 | printf("option flag -%c specified, but not recognized\n", c); | ||
| 456 | usage(); | ||
| 457 | break; | ||
| 458 | } | ||
| 459 | } | ||
| 460 | |||
| 461 | /* arguments are parsed, so now we validate them */ | ||
| 462 | |||
| 463 | if(count > 1) count_flag = 1; | ||
| 464 | |||
| 465 | /* set threshold values to 10usec units (inherited from fping.c) */ | ||
| 466 | crit.rta = crit.rta / 10; | ||
| 467 | warn.rta = warn.rta / 10; | ||
| 468 | select_time = crit.rta; | ||
| 469 | /* this isn't critical, but will most likely not be what the user expects | ||
| 470 | * so we tell him/her about it, but keep running anyways */ | ||
| 471 | if(warn.pl > crit.pl || warn.rta > crit.rta) { | ||
| 472 | select_time = warn.rta; | ||
| 473 | printf("(WARNING threshold > CRITICAL threshold) :: "); | ||
| 474 | fflush(stdout); | ||
| 475 | } | ||
| 476 | |||
| 477 | /* A timeout smaller than maximum rta threshold makes no sense */ | ||
| 478 | if(timeout < crit.rta) timeout = crit.rta; | ||
| 479 | else if(timeout < warn.rta) timeout = warn.rta; | ||
| 480 | |||
| 481 | if((interval < MIN_INTERVAL * 100 || retry > MAX_RETRY) && getuid()) { | ||
| 482 | printf("%s: these options are too risky for mere mortals.\n", prog); | ||
| 483 | printf("%s: You need i >= %u and r < %u\n", | ||
| 484 | prog, MIN_INTERVAL, MAX_RETRY); | ||
| 485 | printf("Current settings; i = %d, r = %d\n", | ||
| 486 | interval / 100, retry); | ||
| 487 | usage(); | ||
| 488 | } | ||
| 489 | |||
| 490 | if((ping_data_size > MAX_PING_DATA) || (ping_data_size < MIN_PING_DATA)) { | ||
| 491 | printf("%s: data size %u not valid, must be between %u and %u\n", | ||
| 492 | prog, ping_data_size, MIN_PING_DATA, MAX_PING_DATA); | ||
| 493 | usage(); | ||
| 494 | |||
| 495 | } | ||
| 496 | |||
| 497 | if((backoff > MAX_BACKOFF_FACTOR) || (backoff < MIN_BACKOFF_FACTOR)) { | ||
| 498 | printf("%s: backoff factor %.1f not valid, must be between %.1f and %.1f\n", | ||
| 499 | prog, backoff, MIN_BACKOFF_FACTOR, MAX_BACKOFF_FACTOR); | ||
| 500 | usage(); | ||
| 501 | |||
| 502 | } | ||
| 503 | |||
| 504 | if(count > MAX_COUNT) { | ||
| 505 | printf("%s: count %u not valid, must be less than %u\n", | ||
| 506 | prog, count, MAX_COUNT); | ||
| 507 | usage(); | ||
| 508 | } | ||
| 509 | |||
| 510 | if(count_flag) { | ||
| 511 | alive_flag = unreachable_flag = 0; | ||
| 512 | } | ||
| 513 | |||
| 514 | trials = (count > retry + 1) ? count : retry + 1; | ||
| 515 | |||
| 516 | /* handle host names supplied on command line or in a file */ | ||
| 517 | /* if the generate_flag is on, then generate the IP list */ | ||
| 518 | argv = &argv[optind]; | ||
| 519 | |||
| 520 | /* cover allowable conditions */ | ||
| 521 | |||
| 522 | /* generate requires command line parameters beyond the switches */ | ||
| 523 | if(generate_flag && !*argv) { | ||
| 524 | printf("generate flag requires command line parameters beyond switches\n"); | ||
| 525 | usage(); | ||
| 526 | } | ||
| 527 | |||
| 528 | if(*argv && !generate_flag) { | ||
| 529 | while(*argv) { | ||
| 530 | if(!(host_ptr->entry = malloc(strlen(*argv) + 1))) { | ||
| 531 | crash("Failed to allocate memory for hostname"); | ||
| 532 | } | ||
| 533 | memset(host_ptr->entry, 0, strlen(*argv) + 1); | ||
| 534 | host_ptr->entry = memcpy(host_ptr->entry, *argv, strlen(*argv)); | ||
| 535 | if(!(host_ptr->next = malloc(sizeof(struct host_name_list)))) | ||
| 536 | crash("Failed to allocate memory for hostname"); | ||
| 537 | host_ptr = host_ptr->next; | ||
| 538 | host_ptr->next = NULL; | ||
| 539 | |||
| 540 | // add_name(*argv); | ||
| 541 | argv++; | ||
| 542 | } | ||
| 543 | } | ||
| 544 | |||
| 545 | // now add all the hosts | ||
| 546 | host_ptr = host_base_ptr; | ||
| 547 | while(host_ptr->next) { | ||
| 548 | add_name(host_ptr->entry); | ||
| 549 | host_ptr = host_ptr->next; | ||
| 550 | } | ||
| 551 | |||
| 552 | if(!num_hosts) { | ||
| 553 | printf("No hosts to work with!\n\n"); | ||
| 554 | usage(); | ||
| 555 | } | ||
| 556 | |||
| 557 | /* allocate array to hold outstanding ping requests */ | ||
| 558 | table = (HOST_ENTRY **) malloc(sizeof(HOST_ENTRY *) * num_hosts); | ||
| 559 | if(!table) crash("Can't malloc array of hosts"); | ||
| 560 | |||
| 561 | cursor = rrlist; | ||
| 562 | |||
| 563 | for(num_jobs = 0; num_jobs < num_hosts; num_jobs++) { | ||
| 564 | table[num_jobs] = cursor; | ||
| 565 | cursor->i = num_jobs; | ||
| 566 | |||
| 567 | cursor = cursor->next; | ||
| 568 | } /* FOR */ | ||
| 569 | |||
| 570 | ping_pkt_size = ping_data_size + SIZE_ICMP_HDR; | ||
| 571 | |||
| 572 | signal(SIGINT, (void *)finish); | ||
| 573 | |||
| 574 | gettimeofday(&start_time, &tz); | ||
| 575 | current_time = start_time; | ||
| 576 | |||
| 577 | last_send_time.tv_sec = current_time.tv_sec - 10000; | ||
| 578 | |||
| 579 | cursor = rrlist; | ||
| 580 | advance = 0; | ||
| 581 | |||
| 582 | /* main loop */ | ||
| 583 | while(num_jobs) { | ||
| 584 | /* fetch all packets that receive within time boundaries */ | ||
| 585 | while(num_pingsent && | ||
| 586 | cursor && | ||
| 587 | cursor->num_sent > cursor->num_recv && | ||
| 588 | wait_for_reply(sock)) ; | ||
| 589 | |||
| 590 | if(cursor && advance) { | ||
| 591 | cursor = cursor->next; | ||
| 592 | } | ||
| 593 | |||
| 594 | gettimeofday(¤t_time, &tz); | ||
| 595 | lt = timeval_diff(¤t_time, &last_send_time); | ||
| 596 | ht = timeval_diff(¤t_time, &cursor->last_send_time); | ||
| 597 | |||
| 598 | advance = 1; | ||
| 599 | |||
| 600 | /* if it's OK to send while counting or looping or starting */ | ||
| 601 | if(lt > interval) { | ||
| 602 | /* send if starting or looping */ | ||
| 603 | if((cursor->num_sent == 0)) { | ||
| 604 | send_ping(sock, cursor); | ||
| 605 | continue; | ||
| 606 | } /* IF */ | ||
| 607 | |||
| 608 | /* send if counting and count not exceeded */ | ||
| 609 | if(count_flag) { | ||
| 610 | if(cursor->num_sent < count) { | ||
| 611 | send_ping(sock, cursor); | ||
| 612 | continue; | ||
| 613 | } /* IF */ | ||
| 614 | } /* IF */ | ||
| 615 | } /* IF */ | ||
| 616 | |||
| 617 | /* is-it-alive mode, and timeout exceeded while waiting for a reply */ | ||
| 618 | /* and we haven't exceeded our retries */ | ||
| 619 | if((lt > interval) && !count_flag && !cursor->num_recv && | ||
| 620 | (ht > timeout) && (cursor->waiting < retry + 1)) { | ||
| 621 | num_timeout++; | ||
| 622 | |||
| 623 | /* try again */ | ||
| 624 | send_ping(sock, cursor); | ||
| 625 | continue; | ||
| 626 | } /* IF */ | ||
| 627 | |||
| 628 | /* didn't send, can we remove? */ | ||
| 629 | |||
| 630 | /* remove if counting and count exceeded */ | ||
| 631 | if(count_flag) { | ||
| 632 | if((cursor->num_sent >= count)) { | ||
| 633 | remove_job(cursor); | ||
| 634 | continue; | ||
| 635 | } /* IF */ | ||
| 636 | } /* IF */ | ||
| 637 | else { | ||
| 638 | /* normal mode, and we got one */ | ||
| 639 | if(cursor->num_recv) { | ||
| 640 | remove_job(cursor); | ||
| 641 | continue; | ||
| 642 | } /* IF */ | ||
| 643 | |||
| 644 | /* normal mode, and timeout exceeded while waiting for a reply */ | ||
| 645 | /* and we've run out of retries, so node is unreachable */ | ||
| 646 | if((ht > timeout) && (cursor->waiting >= retry + 1)) { | ||
| 647 | num_timeout++; | ||
| 648 | remove_job(cursor); | ||
| 649 | continue; | ||
| 650 | |||
| 651 | } /* IF */ | ||
| 652 | } /* ELSE */ | ||
| 653 | |||
| 654 | /* could send to this host, so keep considering it */ | ||
| 655 | if(ht > interval) { | ||
| 656 | advance = 0; | ||
| 657 | } | ||
| 658 | } /* WHILE */ | ||
| 659 | |||
| 660 | finish(); | ||
| 661 | return 0; | ||
| 662 | } /* main() */ | ||
| 663 | |||
| 664 | /************************************************************ | ||
| 665 | * Description: | ||
| 666 | * | ||
| 667 | * Main program clean up and exit point | ||
| 668 | ************************************************************/ | ||
| 669 | void finish() | ||
| 670 | { | ||
| 671 | int i; | ||
| 672 | HOST_ENTRY *h; | ||
| 673 | |||
| 674 | gettimeofday(&end_time, &tz); | ||
| 675 | |||
| 676 | /* tot up unreachables */ | ||
| 677 | for(i=0; i<num_hosts; i++) { | ||
| 678 | h = table[i]; | ||
| 679 | |||
| 680 | if(!h->num_recv) { | ||
| 681 | num_unreachable++; | ||
| 682 | status = fin_stat = STATE_CRITICAL; | ||
| 683 | if(num_hosts == 1) { | ||
| 684 | printf("CRITICAL - %s is down (lost 100%%)|" | ||
| 685 | "rta=;%d;%d;; pl=100%%;%d;%d;;\n", | ||
| 686 | h->host, | ||
| 687 | warn.rta / 100, crit.rta / 100, | ||
| 688 | warn.pl, crit.pl); | ||
| 689 | } | ||
| 690 | else { | ||
| 691 | printf("%s is down (lost 100%%)", h->host); | ||
| 692 | } | ||
| 693 | } | ||
| 694 | else { | ||
| 695 | /* reset the status */ | ||
| 696 | status = STATE_OK; | ||
| 697 | |||
| 698 | /* check for warning before critical, for debugging purposes */ | ||
| 699 | if(warn.rta <= h->total_time / h->num_recv) { | ||
| 700 | /* printf("warn.rta exceeded\n"); | ||
| 701 | */ status = STATE_WARNING; | ||
| 702 | } | ||
| 703 | if(warn.pl <= ((h->num_sent - h->num_recv) * 100) / h->num_sent) { | ||
| 704 | /* printf("warn.pl exceeded (pl=%d)\n", | ||
| 705 | ((h->num_sent - h->num_recv) * 100) / h->num_sent); | ||
| 706 | */ status = STATE_WARNING; | ||
| 707 | } | ||
| 708 | if(crit.rta <= h->total_time / h->num_recv) { | ||
| 709 | /* printf("crit.rta exceeded\n"); | ||
| 710 | */ status = STATE_CRITICAL; | ||
| 711 | } | ||
| 712 | if(crit.pl <= ((h->num_sent - h->num_recv) * 100) / h->num_sent) { | ||
| 713 | /* printf("crit.pl exceeded (pl=%d)\n", | ||
| 714 | ((h->num_sent - h->num_recv) * 100) / h->num_sent); | ||
| 715 | */ status = STATE_CRITICAL; | ||
| 716 | } | ||
| 717 | |||
| 718 | if(num_hosts == 1 || status != STATE_OK) { | ||
| 719 | printf("%s - %s: rta %s ms, lost %d%%", | ||
| 720 | status_string[status], h->host, | ||
| 721 | sprint_tm(h->total_time / h->num_recv), | ||
| 722 | h->num_sent > 0 ? ((h->num_sent - h->num_recv) * 100) / h->num_sent : 0 | ||
| 723 | ); | ||
| 724 | /* perfdata only available for single-host stuff */ | ||
| 725 | if(num_hosts == 1) { | ||
| 726 | printf("|rta=%sms;%d;%d;; pl=%d%%;%d;%d;;\n", | ||
| 727 | sprint_tm(h->total_time / h->num_recv), warn.rta / 100, crit.rta / 100, | ||
| 728 | h->num_sent > 0 ? ((h->num_sent - h->num_recv) * 100) / h->num_sent : 0, warn.pl, crit.pl | ||
| 729 | ); | ||
| 730 | } | ||
| 731 | else printf(" :: "); | ||
| 732 | } | ||
| 733 | |||
| 734 | /* fin_stat should always hold the WORST state */ | ||
| 735 | if(fin_stat != STATE_CRITICAL && status != STATE_OK) { | ||
| 736 | fin_stat = status; | ||
| 737 | } | ||
| 738 | } | ||
| 739 | } | ||
| 740 | |||
| 741 | if(num_noaddress) { | ||
| 742 | printf("No hostaddress specified.\n"); | ||
| 743 | usage(); | ||
| 744 | } | ||
| 745 | else if(num_alive != num_hosts) { | ||
| 746 | /* for future multi-check support */ | ||
| 747 | /*printf("num_alive != num_hosts (%d : %d)\n", num_alive, num_hosts);*/ | ||
| 748 | fin_stat = STATE_CRITICAL; | ||
| 749 | } | ||
| 750 | |||
| 751 | if(num_hosts > 1) { | ||
| 752 | if(num_alive == num_hosts) { | ||
| 753 | printf("OK - All %d hosts are alive\n", num_hosts); | ||
| 754 | } | ||
| 755 | else { | ||
| 756 | printf("CRITICAL - %d of %d hosts are alive\n", num_alive, num_hosts); | ||
| 757 | } | ||
| 758 | } | ||
| 759 | exit(fin_stat); | ||
| 760 | } | ||
| 761 | |||
| 762 | |||
| 763 | void send_ping(int lsock, HOST_ENTRY *h) | ||
| 764 | { | ||
| 765 | char *buffer; | ||
| 766 | struct icmp *icp; | ||
| 767 | PING_DATA *pdp; | ||
| 768 | int n; | ||
| 769 | |||
| 770 | buffer = (char *)malloc((size_t) ping_pkt_size); | ||
| 771 | if(!buffer) | ||
| 772 | crash("can't malloc ping packet"); | ||
| 773 | |||
| 774 | memset(buffer, 0, ping_pkt_size * sizeof(char)); | ||
| 775 | icp = (struct icmp *)buffer; | ||
| 776 | |||
| 777 | gettimeofday(&h->last_send_time, &tz); | ||
| 778 | |||
| 779 | icp->icmp_type = ICMP_ECHO; | ||
| 780 | icp->icmp_code = 0; | ||
| 781 | icp->icmp_cksum = 0; | ||
| 782 | icp->icmp_seq = h->i; | ||
| 783 | icp->icmp_id = ident; | ||
| 784 | |||
| 785 | pdp = (PING_DATA *) (buffer + SIZE_ICMP_HDR); | ||
| 786 | pdp->ping_ts = h->last_send_time; | ||
| 787 | pdp->ping_count = h->num_sent; | ||
| 788 | |||
| 789 | icp->icmp_cksum = in_cksum((u_short *) icp, ping_pkt_size); | ||
| 790 | |||
| 791 | n = sendto(lsock, buffer, ping_pkt_size, 0, | ||
| 792 | (struct sockaddr *)&h->saddr, sizeof(struct sockaddr_in)); | ||
| 793 | |||
| 794 | if(n < 0 || (unsigned int)n != ping_pkt_size) { | ||
| 795 | if(unreachable_flag) { | ||
| 796 | printf("%s error while sending ping: %s\n", | ||
| 797 | h->host, strerror(errno)); | ||
| 798 | } /* IF */ | ||
| 799 | |||
| 800 | num_unreachable++; | ||
| 801 | remove_job(h); | ||
| 802 | } /* IF */ | ||
| 803 | else { | ||
| 804 | /* mark this trial as outstanding */ | ||
| 805 | h->resp_times[h->num_sent] = RESP_WAITING; | ||
| 806 | |||
| 807 | h->num_sent++; | ||
| 808 | h->waiting++; | ||
| 809 | num_pingsent++; | ||
| 810 | last_send_time = h->last_send_time; | ||
| 811 | } /* ELSE */ | ||
| 812 | |||
| 813 | free(buffer); | ||
| 814 | } /* send_ping() */ | ||
| 815 | |||
| 816 | int wait_for_reply(int lsock) | ||
| 817 | { | ||
| 818 | int result; | ||
| 819 | static char buffer[4096]; | ||
| 820 | struct sockaddr_in response_addr; | ||
| 821 | struct ip *ip; | ||
| 822 | int hlen; | ||
| 823 | struct icmp *icp; | ||
| 824 | int n; | ||
| 825 | HOST_ENTRY *h = NULL; | ||
| 826 | long this_reply; | ||
| 827 | int this_count; | ||
| 828 | struct timeval sent_time; | ||
| 829 | |||
| 830 | result = recvfrom_wto(lsock, buffer, sizeof(buffer), | ||
| 831 | (struct sockaddr *)&response_addr, select_time); | ||
| 832 | |||
| 833 | if(result < 0) return 0; /* timeout */ | ||
| 834 | |||
| 835 | ip = (struct ip *)buffer; | ||
| 836 | |||
| 837 | #if defined( __alpha__ ) && __STDC__ && !defined( __GLIBC__ ) | ||
| 838 | /* The alpha headers are decidedly broken. | ||
| 839 | * Using an ANSI compiler, it provides ip_vhl instead of ip_hl and | ||
| 840 | * ip_v. So, to get ip_hl, we mask off the bottom four bits. | ||
| 841 | */ | ||
| 842 | hlen = (ip->ip_vhl & 0x0F) << 2; | ||
| 843 | #else | ||
| 844 | hlen = ip->ip_hl << 2; | ||
| 845 | #endif /* defined(__alpha__) && __STDC__ */ | ||
| 846 | |||
| 847 | if(result < hlen + ICMP_MINLEN) { | ||
| 848 | printf("received packet too short for ICMP (%d bytes from %s)\n", result, | ||
| 849 | inet_ntoa(response_addr.sin_addr)); | ||
| 850 | |||
| 851 | return (1); /* too short */ | ||
| 852 | } /* IF */ | ||
| 853 | |||
| 854 | icp = (struct icmp *)(buffer + hlen); | ||
| 855 | if(icp->icmp_type != ICMP_ECHOREPLY) { | ||
| 856 | /* handle some problem */ | ||
| 857 | if(handle_random_icmp(icp, &response_addr)) | ||
| 858 | num_othericmprcvd++; | ||
| 859 | |||
| 860 | return 1; | ||
| 861 | } /* IF */ | ||
| 862 | |||
| 863 | if(icp->icmp_id != ident) | ||
| 864 | return 1; /* packet received, but not the one we are looking for! */ | ||
| 865 | |||
| 866 | num_pingreceived++; | ||
| 867 | |||
| 868 | if(icp->icmp_seq >= (n_short) num_hosts) | ||
| 869 | return(1); /* packet received, don't worry about it anymore */ | ||
| 870 | |||
| 871 | n = icp->icmp_seq; | ||
| 872 | h = table[n]; | ||
| 873 | |||
| 874 | /* received ping is cool, so process it */ | ||
| 875 | |||
| 876 | gettimeofday(¤t_time, &tz); | ||
| 877 | h->waiting = 0; | ||
| 878 | h->num_recv++; | ||
| 879 | |||
| 880 | memcpy(&sent_time, icp->icmp_data + offsetof(PING_DATA, ping_ts), | ||
| 881 | sizeof(sent_time)); | ||
| 882 | memcpy(&this_count, icp->icmp_data, sizeof(this_count)); | ||
| 883 | |||
| 884 | this_reply = timeval_diff(¤t_time, &sent_time); | ||
| 885 | h->total_time += this_reply; | ||
| 886 | total_replies++; | ||
| 887 | |||
| 888 | /* note reply time in array, probably */ | ||
| 889 | if((this_count >= 0) && ((unsigned int)this_count < trials)) { | ||
| 890 | if(h->resp_times[this_count] != RESP_WAITING) { | ||
| 891 | printf("%s : duplicate for [%d], %d bytes, %s ms", | ||
| 892 | h->host, this_count, result, sprint_tm(this_reply)); | ||
| 893 | |||
| 894 | if(response_addr.sin_addr.s_addr != h->saddr.sin_addr.s_addr) | ||
| 895 | printf(" [<- %s]\n", inet_ntoa(response_addr.sin_addr)); | ||
| 896 | } /* IF */ | ||
| 897 | else h->resp_times[this_count] = this_reply; | ||
| 898 | } /* IF */ | ||
| 899 | else { | ||
| 900 | /* count is out of bounds?? */ | ||
| 901 | printf("%s : duplicate for [%d], %d bytes, %s ms\n", | ||
| 902 | h->host, this_count, result, sprint_tm(this_reply)); | ||
| 903 | } /* ELSE */ | ||
| 904 | |||
| 905 | if(h->num_recv == 1) { | ||
| 906 | num_alive++; | ||
| 907 | } /* IF */ | ||
| 908 | |||
| 909 | return num_jobs; | ||
| 910 | } /* wait_for_reply() */ | ||
| 911 | |||
| 912 | int handle_random_icmp(struct icmp *p, struct sockaddr_in *addr) | ||
| 913 | { | ||
| 914 | struct icmp *sent_icmp; | ||
| 915 | u_char *c; | ||
| 916 | HOST_ENTRY *h; | ||
| 917 | |||
| 918 | c = (u_char *) p; | ||
| 919 | switch (p->icmp_type) { | ||
| 920 | case ICMP_UNREACH: | ||
| 921 | sent_icmp = (struct icmp *)(c + 28); | ||
| 922 | |||
| 923 | if((sent_icmp->icmp_type == ICMP_ECHO) && | ||
| 924 | (sent_icmp->icmp_id == ident) && | ||
| 925 | (sent_icmp->icmp_seq < (n_short) num_hosts)) { | ||
| 926 | /* this is a response to a ping we sent */ | ||
| 927 | h = table[sent_icmp->icmp_seq]; | ||
| 928 | |||
| 929 | if(p->icmp_code > ICMP_UNREACH_MAXTYPE) { | ||
| 930 | printf("ICMP Unreachable (Invalid Code) from %s for ICMP Echo sent to %s", | ||
| 931 | inet_ntoa(addr->sin_addr), h->host); | ||
| 932 | |||
| 933 | } /* IF */ | ||
| 934 | else { | ||
| 935 | printf("ICMP Unreachable from %s for ICMP Echo sent to %s", | ||
| 936 | inet_ntoa(addr->sin_addr), h->host); | ||
| 937 | |||
| 938 | } /* ELSE */ | ||
| 939 | |||
| 940 | if(inet_addr(h->host) == INADDR_NONE) | ||
| 941 | printf(" (%s)", inet_ntoa(h->saddr.sin_addr)); | ||
| 942 | |||
| 943 | printf("\n"); | ||
| 944 | |||
| 945 | } /* IF */ | ||
| 946 | |||
| 947 | return 1; | ||
| 948 | |||
| 949 | case ICMP_SOURCEQUENCH: | ||
| 950 | case ICMP_REDIRECT: | ||
| 951 | case ICMP_TIMXCEED: | ||
| 952 | case ICMP_PARAMPROB: | ||
| 953 | sent_icmp = (struct icmp *)(c + 28); | ||
| 954 | if((sent_icmp->icmp_type = ICMP_ECHO) && | ||
| 955 | (sent_icmp->icmp_id = ident) && | ||
| 956 | (sent_icmp->icmp_seq < (n_short) num_hosts)) { | ||
| 957 | /* this is a response to a ping we sent */ | ||
| 958 | h = table[sent_icmp->icmp_seq]; | ||
| 959 | printf("ICMP Unreachable from %s for ICMP Echo sent to %s", | ||
| 960 | inet_ntoa(addr->sin_addr), h->host); | ||
| 961 | |||
| 962 | if(inet_addr(h->host) == INADDR_NONE) | ||
| 963 | printf(" (%s)", inet_ntoa(h->saddr.sin_addr)); | ||
| 964 | |||
| 965 | printf("\n"); | ||
| 966 | } /* IF */ | ||
| 967 | |||
| 968 | return 2; | ||
| 969 | |||
| 970 | /* no way to tell whether any of these are sent due to our ping */ | ||
| 971 | /* or not (shouldn't be, of course), so just discard */ | ||
| 972 | case ICMP_TSTAMP: | ||
| 973 | case ICMP_TSTAMPREPLY: | ||
| 974 | case ICMP_IREQ: | ||
| 975 | case ICMP_IREQREPLY: | ||
| 976 | case ICMP_MASKREQ: | ||
| 977 | case ICMP_MASKREPLY: | ||
| 978 | default: | ||
| 979 | return 0; | ||
| 980 | |||
| 981 | } /* SWITCH */ | ||
| 982 | |||
| 983 | } /* handle_random_icmp() */ | ||
| 984 | |||
| 985 | int in_cksum(u_short * p, int n) | ||
| 986 | { | ||
| 987 | register u_short answer; | ||
| 988 | register long sum = 0; | ||
| 989 | u_short odd_byte = 0; | ||
| 990 | |||
| 991 | while(n > 1) { | ||
| 992 | sum += *p++; | ||
| 993 | n -= 2; | ||
| 994 | } /* WHILE */ | ||
| 995 | |||
| 996 | /* mop up an odd byte, if necessary */ | ||
| 997 | if(n == 1) { | ||
| 998 | *(u_char *) (&odd_byte) = *(u_char *) p; | ||
| 999 | sum += odd_byte; | ||
| 1000 | } /* IF */ | ||
| 1001 | |||
| 1002 | sum = (sum >> 16) + (sum & 0xffff); /* add hi 16 to low 16 */ | ||
| 1003 | sum += (sum >> 16); /* add carry */ | ||
| 1004 | answer = ~sum; /* ones-complement, truncate */ | ||
| 1005 | |||
| 1006 | return (answer); | ||
| 1007 | |||
| 1008 | } /* in_cksum() */ | ||
| 1009 | |||
| 1010 | void add_name(char *name) | ||
| 1011 | { | ||
| 1012 | struct hostent *host_ent; | ||
| 1013 | int ipaddress; | ||
| 1014 | struct in_addr *ipa = (struct in_addr *)&ipaddress; | ||
| 1015 | struct in_addr *host_add; | ||
| 1016 | char *nm; | ||
| 1017 | int i = 0; | ||
| 1018 | |||
| 1019 | if((ipaddress = inet_addr(name)) != -1) { | ||
| 1020 | /* input name is an IP addr, go with it */ | ||
| 1021 | if(name_flag) { | ||
| 1022 | if(addr_flag) | ||
| 1023 | add_addr(name, na_cat(get_host_by_address(*ipa), *ipa), *ipa); | ||
| 1024 | else { | ||
| 1025 | nm = cpystr(get_host_by_address(*ipa)); | ||
| 1026 | add_addr(name, nm, *ipa); | ||
| 1027 | |||
| 1028 | } /* ELSE */ | ||
| 1029 | } /* IF */ | ||
| 1030 | else add_addr(name, name, *ipa); | ||
| 1031 | |||
| 1032 | return; | ||
| 1033 | } /* IF */ | ||
| 1034 | |||
| 1035 | /* input name is not an IP addr, maybe it's a host name */ | ||
| 1036 | host_ent = gethostbyname(name); | ||
| 1037 | if(host_ent == NULL) { | ||
| 1038 | if(h_errno == TRY_AGAIN) { | ||
| 1039 | u_sleep(DNS_TIMEOUT); | ||
| 1040 | host_ent = gethostbyname(name); | ||
| 1041 | } /* IF */ | ||
| 1042 | |||
| 1043 | if(host_ent == NULL) { | ||
| 1044 | printf("%s address not found\n", name); | ||
| 1045 | num_noaddress++; | ||
| 1046 | return; | ||
| 1047 | } /* IF */ | ||
| 1048 | } /* IF */ | ||
| 1049 | |||
| 1050 | host_add = (struct in_addr *)*(host_ent->h_addr_list); | ||
| 1051 | if(host_add == NULL) { | ||
| 1052 | printf("%s has no address data\n", name); | ||
| 1053 | num_noaddress++; | ||
| 1054 | return; | ||
| 1055 | } /* IF */ | ||
| 1056 | else { | ||
| 1057 | /* it is indeed a hostname with a real address */ | ||
| 1058 | while(host_add) { | ||
| 1059 | if(name_flag && addr_flag) | ||
| 1060 | add_addr(name, na_cat(name, *host_add), *host_add); | ||
| 1061 | else if(addr_flag) { | ||
| 1062 | nm = cpystr(inet_ntoa(*host_add)); | ||
| 1063 | add_addr(name, nm, *host_add); | ||
| 1064 | } /* ELSE IF */ | ||
| 1065 | else { | ||
| 1066 | add_addr(name, name, *host_add); | ||
| 1067 | } | ||
| 1068 | |||
| 1069 | if(!multif_flag) break; | ||
| 1070 | |||
| 1071 | host_add = (struct in_addr *)(host_ent->h_addr_list[++i]); | ||
| 1072 | } /* WHILE */ | ||
| 1073 | } /* ELSE */ | ||
| 1074 | } /* add_name() */ | ||
| 1075 | |||
| 1076 | |||
| 1077 | char *na_cat(char *name, struct in_addr ipaddr) | ||
| 1078 | { | ||
| 1079 | char *nm, *as; | ||
| 1080 | |||
| 1081 | as = inet_ntoa(ipaddr); | ||
| 1082 | nm = (char *)malloc(strlen(name) + strlen(as) + 4); | ||
| 1083 | |||
| 1084 | if(!nm) | ||
| 1085 | crash("can't allocate some space for a string"); | ||
| 1086 | |||
| 1087 | strcpy(nm, name); | ||
| 1088 | strcat(nm, " ("); | ||
| 1089 | strcat(nm, as); | ||
| 1090 | strcat(nm, ")"); | ||
| 1091 | |||
| 1092 | return (nm); | ||
| 1093 | |||
| 1094 | } /* na_cat() */ | ||
| 1095 | |||
| 1096 | |||
| 1097 | void add_addr(char *name, char *host, struct in_addr ipaddr) | ||
| 1098 | { | ||
| 1099 | HOST_ENTRY *p; | ||
| 1100 | unsigned int n; | ||
| 1101 | int *i; | ||
| 1102 | |||
| 1103 | if(!(p = (HOST_ENTRY *) malloc(sizeof(HOST_ENTRY)))) { | ||
| 1104 | crash("can't allocate HOST_ENTRY"); | ||
| 1105 | } | ||
| 1106 | |||
| 1107 | memset((char *)p, 0, sizeof(HOST_ENTRY)); | ||
| 1108 | |||
| 1109 | p->name = name; | ||
| 1110 | p->host = host; | ||
| 1111 | p->saddr.sin_family = AF_INET; | ||
| 1112 | p->saddr.sin_addr = ipaddr; | ||
| 1113 | p->running = 1; | ||
| 1114 | |||
| 1115 | /* array for response time results */ | ||
| 1116 | if(!(i = (int *)malloc(trials * sizeof(int)))) { | ||
| 1117 | crash("can't allocate resp_times array"); | ||
| 1118 | } | ||
| 1119 | |||
| 1120 | for(n = 1; n < trials; n++) | ||
| 1121 | i[n] = RESP_UNUSED; | ||
| 1122 | |||
| 1123 | p->resp_times = i; | ||
| 1124 | |||
| 1125 | if(!rrlist) { | ||
| 1126 | rrlist = p; | ||
| 1127 | p->next = p; | ||
| 1128 | p->prev = p; | ||
| 1129 | } /* IF */ | ||
| 1130 | else { | ||
| 1131 | p->next = rrlist; | ||
| 1132 | p->prev = rrlist->prev; | ||
| 1133 | p->prev->next = p; | ||
| 1134 | p->next->prev = p; | ||
| 1135 | } /* ELSE */ | ||
| 1136 | |||
| 1137 | num_hosts++; | ||
| 1138 | } /* add_addr() */ | ||
| 1139 | |||
| 1140 | |||
| 1141 | void remove_job(HOST_ENTRY * h) | ||
| 1142 | { | ||
| 1143 | h->running = 0; | ||
| 1144 | h->waiting = 0; | ||
| 1145 | num_jobs--; | ||
| 1146 | |||
| 1147 | |||
| 1148 | if(num_jobs) { | ||
| 1149 | /* remove us from list of active jobs */ | ||
| 1150 | h->prev->next = h->next; | ||
| 1151 | h->next->prev = h->prev; | ||
| 1152 | if(h == cursor) cursor = h->next; | ||
| 1153 | } /* IF */ | ||
| 1154 | else { | ||
| 1155 | cursor = NULL; | ||
| 1156 | rrlist = NULL; | ||
| 1157 | } /* ELSE */ | ||
| 1158 | |||
| 1159 | } /* remove_job() */ | ||
| 1160 | |||
| 1161 | |||
| 1162 | char *get_host_by_address(struct in_addr in) | ||
| 1163 | { | ||
| 1164 | struct hostent *h; | ||
| 1165 | h = gethostbyaddr((char *)&in, sizeof(struct in_addr), AF_INET); | ||
| 1166 | |||
| 1167 | if(h == NULL || h->h_name == NULL) | ||
| 1168 | return inet_ntoa(in); | ||
| 1169 | else | ||
| 1170 | return (char *)h->h_name; | ||
| 1171 | |||
| 1172 | } /* get_host_by_address() */ | ||
| 1173 | |||
| 1174 | |||
| 1175 | char *cpystr(char *string) | ||
| 1176 | { | ||
| 1177 | char *dst; | ||
| 1178 | |||
| 1179 | if(string) { | ||
| 1180 | dst = (char *)malloc(1 + strlen(string)); | ||
| 1181 | if(!dst) crash("malloc() failed!"); | ||
| 1182 | |||
| 1183 | strcpy(dst, string); | ||
| 1184 | return dst; | ||
| 1185 | |||
| 1186 | } /* IF */ | ||
| 1187 | else return NULL; | ||
| 1188 | |||
| 1189 | } /* cpystr() */ | ||
| 1190 | |||
| 1191 | |||
| 1192 | void crash(char *msg) | ||
| 1193 | { | ||
| 1194 | if(errno || h_errno) { | ||
| 1195 | if(errno) | ||
| 1196 | printf("%s: %s : %s\n", prog, msg, strerror(errno)); | ||
| 1197 | if(h_errno) | ||
| 1198 | printf("%s: %s : A network error occurred\n", prog, msg); | ||
| 1199 | } | ||
| 1200 | else printf("%s: %s\n", prog, msg); | ||
| 1201 | |||
| 1202 | exit(STATE_UNKNOWN); | ||
| 1203 | } /* crash() */ | ||
| 1204 | |||
| 1205 | |||
| 1206 | long timeval_diff(struct timeval *a, struct timeval *b) | ||
| 1207 | { | ||
| 1208 | double temp; | ||
| 1209 | |||
| 1210 | temp = (((a->tv_sec * 1000000) + a->tv_usec) - | ||
| 1211 | ((b->tv_sec * 1000000) + b->tv_usec)) / 10; | ||
| 1212 | |||
| 1213 | return (long)temp; | ||
| 1214 | |||
| 1215 | } /* timeval_diff() */ | ||
| 1216 | |||
| 1217 | |||
| 1218 | char *sprint_tm(int t) | ||
| 1219 | { | ||
| 1220 | static char buf[10]; | ||
| 1221 | |||
| 1222 | /* <= 0.99 ms */ | ||
| 1223 | if(t < 100) { | ||
| 1224 | sprintf(buf, "0.%02d", t); | ||
| 1225 | return (buf); | ||
| 1226 | } /* IF */ | ||
| 1227 | |||
| 1228 | /* 1.00 - 9.99 ms */ | ||
| 1229 | if(t < 1000) { | ||
| 1230 | sprintf(buf, "%d.%02d", t / 100, t % 100); | ||
| 1231 | return (buf); | ||
| 1232 | } /* IF */ | ||
| 1233 | |||
| 1234 | /* 10.0 - 99.9 ms */ | ||
| 1235 | if(t < 10000) { | ||
| 1236 | sprintf(buf, "%d.%d", t / 100, (t % 100) / 10); | ||
| 1237 | return (buf); | ||
| 1238 | } /* IF */ | ||
| 1239 | |||
| 1240 | /* >= 100 ms */ | ||
| 1241 | sprintf(buf, "%d", t / 100); | ||
| 1242 | return (buf); | ||
| 1243 | } /* sprint_tm() */ | ||
| 1244 | |||
| 1245 | |||
| 1246 | /* | ||
| 1247 | * select() is posix, so we expect it to be around | ||
| 1248 | */ | ||
| 1249 | void u_sleep(int u_sec) | ||
| 1250 | { | ||
| 1251 | int nfound; | ||
| 1252 | struct timeval to; | ||
| 1253 | fd_set readset, writeset; | ||
| 1254 | |||
| 1255 | to.tv_sec = u_sec / 1000000; | ||
| 1256 | to.tv_usec = u_sec - (to.tv_sec * 1000000); | ||
| 1257 | /* printf("u_sleep :: to.tv_sec: %d, to_tv_usec: %d\n", | ||
| 1258 | (int)to.tv_sec, (int)to.tv_usec); | ||
| 1259 | */ | ||
| 1260 | FD_ZERO(&writeset); | ||
| 1261 | FD_ZERO(&readset); | ||
| 1262 | nfound = select(0, &readset, &writeset, NULL, &to); | ||
| 1263 | if(nfound < 0) | ||
| 1264 | crash("select() in u_sleep:"); | ||
| 1265 | |||
| 1266 | return; | ||
| 1267 | } /* u_sleep() */ | ||
| 1268 | |||
| 1269 | |||
| 1270 | /************************************************************ | ||
| 1271 | * Description: | ||
| 1272 | * | ||
| 1273 | * receive with timeout | ||
| 1274 | * returns length of data read or -1 if timeout | ||
| 1275 | * crash on any other errrors | ||
| 1276 | ************************************************************/ | ||
| 1277 | /* TODO: add MSG_DONTWAIT to recvfrom flags (currently 0) */ | ||
| 1278 | int recvfrom_wto(int sock, char *buf, int len, struct sockaddr *saddr, int timo) | ||
| 1279 | { | ||
| 1280 | int nfound = 0, slen, n; | ||
| 1281 | struct timeval to; | ||
| 1282 | fd_set readset, writeset; | ||
| 1283 | |||
| 1284 | to.tv_sec = timo / 1000000; | ||
| 1285 | to.tv_usec = (timo - (to.tv_sec * 1000000)) * 10; | ||
| 1286 | |||
| 1287 | /* printf("to.tv_sec: %d, to.tv_usec: %d\n", (int)to.tv_sec, (int)to.tv_usec); | ||
| 1288 | */ | ||
| 1289 | |||
| 1290 | FD_ZERO(&readset); | ||
| 1291 | FD_ZERO(&writeset); | ||
| 1292 | FD_SET(sock, &readset); | ||
| 1293 | nfound = select(sock + 1, &readset, &writeset, NULL, &to); | ||
| 1294 | if(nfound < 0) crash("select() in recvfrom_wto"); | ||
| 1295 | |||
| 1296 | if(nfound == 0) return -1; /* timeout */ | ||
| 1297 | |||
| 1298 | if(nfound) { | ||
| 1299 | slen = sizeof(struct sockaddr); | ||
| 1300 | n = recvfrom(sock, buf, len, 0, saddr, &slen); | ||
| 1301 | if(n < 0) crash("recvfrom"); | ||
| 1302 | return(n); | ||
| 1303 | } | ||
| 1304 | |||
| 1305 | return(0); /* 0 bytes read, so return it */ | ||
| 1306 | } /* recvfrom_wto() */ | ||
| 1307 | |||
| 1308 | |||
| 1309 | /* | ||
| 1310 | * u = micro | ||
| 1311 | * m = milli | ||
| 1312 | * s = seconds | ||
| 1313 | */ | ||
| 1314 | int get_threshold(char *str, threshold *th) | ||
| 1315 | { | ||
| 1316 | unsigned int i, factor = 0; | ||
| 1317 | char *p = NULL; | ||
| 1318 | |||
| 1319 | if(!str || !strlen(str) || !th) return -1; | ||
| 1320 | |||
| 1321 | for(i=0; i<strlen(str); i++) { | ||
| 1322 | /* we happily accept decimal points in round trip time thresholds, | ||
| 1323 | * but we ignore them quite blandly. The new way of specifying higher | ||
| 1324 | * precision is to specify 'u' (for microseconds), | ||
| 1325 | * 'm' (for millisecs - default) or 's' for seconds. */ | ||
| 1326 | if(!p && !factor) { | ||
| 1327 | if(str[i] == 's') factor = 1000000; /* seconds */ | ||
| 1328 | else if(str[i] == 'm') factor = 1000; /* milliseconds */ | ||
| 1329 | else if(str[i] == 'u') factor = 1; /* microseconds */ | ||
| 1330 | } | ||
| 1331 | |||
| 1332 | if(str[i] == '%') str[i] = '\0'; | ||
| 1333 | else if(str[i] == ',' && !p && i != (strlen(str) - 1)) { | ||
| 1334 | p = &str[i+1]; | ||
| 1335 | str[i] = '\0'; | ||
| 1336 | } | ||
| 1337 | } | ||
| 1338 | |||
| 1339 | /* default to milliseconds */ | ||
| 1340 | if(!factor) factor = 1000; | ||
| 1341 | |||
| 1342 | if(!p || !strlen(p)) return -1; | ||
| 1343 | th->rta = (unsigned int)strtoul(str, NULL, 0) * factor; | ||
| 1344 | th->pl = (unsigned int)strtoul(p, NULL, 0); | ||
| 1345 | return 0; | ||
| 1346 | } | ||
| 1347 | |||
| 1348 | /* make a blahblah */ | ||
| 1349 | void usage(void) | ||
| 1350 | { | ||
| 1351 | printf("\nUsage: %s [options] [targets]\n", prog); | ||
| 1352 | printf(" -H host target host\n"); | ||
| 1353 | printf(" -b n ping packet size in bytes (default %d)\n", ping_data_size); | ||
| 1354 | printf(" -n|p n number of pings to send to each target (default %d)\n", count); | ||
| 1355 | printf(" -r n number of retries (default %d)\n", retry); | ||
| 1356 | printf(" -t n timeout value (in msec) (default %d)\n", timeout / 100); | ||
| 1357 | printf(" -i n packet interval (in msec) (default %d)\n", DEFAULT_INTERVAL); | ||
| 1358 | /* XXX - possibly on todo-list | ||
| 1359 | printf(" -m ping multiple interfaces on target host\n"); | ||
| 1360 | printf(" -a show targets that are alive\n"); | ||
| 1361 | printf(" -d show dead targets\n"); | ||
| 1362 | */ printf(" -v show version\n"); | ||
| 1363 | printf(" -D increase debug output level\n"); | ||
| 1364 | printf(" -w warning threshold pair, given as RTA[ums],PL[%%]\n"); | ||
| 1365 | printf(" -c critical threshold pair, given as RTA[ums],PL[%%]\n"); | ||
| 1366 | printf("\n"); | ||
| 1367 | printf("Note:\n"); | ||
| 1368 | printf("* This program requires root privileges to run properly.\n"); | ||
| 1369 | printf(" If it is run as setuid root it will halt with an error if;\n"); | ||
| 1370 | printf(" interval < 25 || retries > 5\n\n"); | ||
| 1371 | printf("* Threshold pairs are given as such;\n"); | ||
| 1372 | printf(" 100,40%%\n"); | ||
| 1373 | printf(" to set a threshold value pair of 100 milliseconds and 40%% packetloss\n"); | ||
| 1374 | printf(" The '%%' sign is optional, and if rta value is suffixed by;\n"); | ||
| 1375 | printf(" s, rta time is set in seconds\n"); | ||
| 1376 | printf(" m, rta time will be set in milliseconds (this is default)\n"); | ||
| 1377 | printf(" u, rta time will be set in microseconds\n"); | ||
| 1378 | printf(" Decimal points are silently ignored for sideways compatibility.\n"); | ||
| 1379 | printf("\n"); | ||
| 1380 | exit(3); | ||
| 1381 | } /* usage() */ | ||
