Index: netutils.c =================================================================== RCS file: /cvsroot/nagiosplug/nagiosplug/plugins/netutils.c,v retrieving revision 1.1.1.1 diff -u -r1.1.1.1 netutils.c --- netutils.c 28 Feb 2002 06:42:59 -0000 1.1.1.1 +++ netutils.c 11 Dec 2002 02:38:43 -0000 @@ -42,13 +42,11 @@ int process_tcp_request2 (char *, int, char *, char *, int); int process_tcp_request (char *, int, char *, char *, int); int process_udp_request (char *, int, char *, char *, int); -int process_request (char *, int, char *, char *, char *, int); +int process_request (char *, int, int, char *, char *, int); int my_tcp_connect (char *, int, int *); int my_udp_connect (char *, int, int *); -int my_connect (char *, int, int *, char *); - -int my_inet_aton (register const char *, struct in_addr *); +int my_connect (char *, int, int *, int); /* handles socket timeouts */ void @@ -64,47 +62,29 @@ /* connects to a host on a specified TCP port, sends a string, and gets a response */ int -process_tcp_request (char *server_address, - int server_port, - char *send_buffer, char *recv_buffer, int recv_size) +process_tcp_request (char *server_address, int server_port, + char *send_buffer, char *recv_buffer, int recv_size) { - int result; - char proto[4] = "tcp"; - - result = process_request (server_address, - server_port, - proto, send_buffer, recv_buffer, recv_size); - - return result; + return process_request (server_address, server_port, IPPROTO_TCP, + send_buffer, recv_buffer, recv_size); } - /* connects to a host on a specified UDP port, sends a string, and gets a response */ int -process_udp_request (char *server_address, - int server_port, - char *send_buffer, char *recv_buffer, int recv_size) +process_udp_request (char *server_address, int server_port, + char *send_buffer, char *recv_buffer, int recv_size) { - int result; - char proto[4] = "udp"; - - result = process_request (server_address, - server_port, - proto, send_buffer, recv_buffer, recv_size); - - return result; + return process_request (server_address, server_port, IPPROTO_UDP, + send_buffer, recv_buffer, recv_size); } - - /* connects to a host on a specified tcp port, sends a string, and gets a response. loops on select-recv until timeout or eof to get all of a multi-packet answer */ int -process_tcp_request2 (char *server_address, - int server_port, - char *send_buffer, char *recv_buffer, int recv_size) +process_tcp_request2 (char *server_address, int server_port, + char *send_buffer, char *recv_buffer, int recv_size) { int result; @@ -115,7 +95,7 @@ fd_set readfds; int recv_length = 0; - result = my_connect (server_address, server_port, &sd, "tcp"); + result = my_connect (server_address, server_port, &sd, IPPROTO_TCP); if (result != STATE_OK) return STATE_CRITICAL; @@ -149,24 +129,30 @@ else { /* it has */ recv_result = recv (sd, recv_buffer + recv_length, recv_size - recv_length - 1, 0); - if (recv_result == -1) { /* recv failed, bail out */ + if (recv_result == -1) { + /* recv failed, bail out */ strcpy (recv_buffer + recv_length, ""); result = STATE_WARNING; break; } - else if (recv_result == 0) { /* end of file ? */ + else if (recv_result == 0) { + /* end of file ? */ recv_buffer[recv_length] = 0; break; } - else { /* we got data! */ + else { + /* we got data! */ recv_length += recv_result; - if (recv_length >= recv_size - 1) { /* buffer full, we're done */ + if (recv_length >= recv_size - 1) { + /* buffer full, we're done */ recv_buffer[recv_size - 1] = 0; break; } } - } /* end if(!FD_ISSET(sd,&readfds)) */ - } /* end while(1) */ + } + /* end if(!FD_ISSET(sd,&readfds)) */ + } + /* end while(1) */ close (sd); return result; @@ -175,10 +161,8 @@ /* connects to a host on a specified port, sends a string, and gets a response */ int -process_request (char *server_address, - int server_port, - char *proto, - char *send_buffer, char *recv_buffer, int recv_size) +process_request (char *server_address, int server_port, int proto, + char *send_buffer, char *recv_buffer, int recv_size) { int result; int send_result; @@ -218,7 +202,7 @@ recv_result = recv (sd, recv_buffer, recv_size - 1, 0); if (recv_result == -1) { strcpy (recv_buffer, ""); - if (!strcmp (proto, "tcp")) + if (proto != IPPROTO_TCP) printf ("recv() failed\n"); result = STATE_WARNING; } @@ -236,196 +220,74 @@ /* opens a connection to a remote host/tcp port */ + int my_tcp_connect (char *host_name, int port, int *sd) { - int result; - char proto[4] = "tcp"; - - result = my_connect (host_name, port, sd, proto); - - return result; + return my_connect (host_name, port, sd, IPPROTO_TCP); } - /* opens a connection to a remote host/udp port */ int my_udp_connect (char *host_name, int port, int *sd) { - int result; - char proto[4] = "udp"; - - result = my_connect (host_name, port, sd, proto); - - return result; + return my_connect (host_name, port, sd, IPPROTO_UDP); } /* opens a tcp or udp connection to a remote host */ int -my_connect (char *host_name, int port, int *sd, char *proto) -{ - struct sockaddr_in servaddr; - struct hostent *hp; - struct protoent *ptrp; - int result; - - bzero ((char *) &servaddr, sizeof (servaddr)); - servaddr.sin_family = AF_INET; - servaddr.sin_port = htons (port); - - /* try to bypass using a DNS lookup if this is just an IP address */ - if (!my_inet_aton (host_name, &servaddr.sin_addr)) { - - /* else do a DNS lookup */ - hp = gethostbyname ((const char *) host_name); - if (hp == NULL) { - printf ("Invalid host name '%s'\n", host_name); - return STATE_UNKNOWN; - } - - memcpy (&servaddr.sin_addr, hp->h_addr, hp->h_length); - } - - /* map transport protocol name to protocol number */ - if ((ptrp = getprotobyname (proto)) == NULL) { - printf ("Cannot map \"%s\" to protocol number\n", proto); - return STATE_UNKNOWN; - } - - /* create a socket */ - *sd = - socket (PF_INET, (!strcmp (proto, "udp")) ? SOCK_DGRAM : SOCK_STREAM, - ptrp->p_proto); - if (*sd < 0) { - printf ("Socket creation failed\n"); - return STATE_UNKNOWN; - } - - /* open a connection */ - result = connect (*sd, (struct sockaddr *) &servaddr, sizeof (servaddr)); - if (result < 0) { - switch (errno) { - case ECONNREFUSED: - printf ("Connection refused by host\n"); - break; - case ETIMEDOUT: - printf ("Timeout while attempting connection\n"); - break; - case ENETUNREACH: - printf ("Network is unreachable\n"); - break; - default: - printf ("Connection refused or timed out\n"); - } - - return STATE_CRITICAL; - } - - return STATE_OK; -} - - - -/* This code was taken from Fyodor's nmap utility, which was originally - taken from the GLIBC 2.0.6 libraries because Solaris doesn't contain - the inet_aton() funtion. */ -int -my_inet_aton (register const char *cp, struct in_addr *addr) +my_connect (char *host_name, int port, int *sd, int proto) { - register unsigned int val; /* changed from u_long --david */ - register int base, n; - register char c; - u_int parts[4]; - register u_int *pp = parts; - - c = *cp; - - for (;;) { - - /* - * Collect number up to ``.''. - * Values are specified as for C: - * 0x=hex, 0=octal, isdigit=decimal. - */ - if (!isdigit ((int) c)) - return (0); - val = 0; - base = 10; - - if (c == '0') { - c = *++cp; - if (c == 'x' || c == 'X') - base = 16, c = *++cp; - else - base = 8; - } - - for (;;) { - if (isascii ((int) c) && isdigit ((int) c)) { - val = (val * base) + (c - '0'); - c = *++cp; - } - else if (base == 16 && isascii ((int) c) && isxdigit ((int) c)) { - val = (val << 4) | (c + 10 - (islower ((int) c) ? 'a' : 'A')); - c = *++cp; - } - else - break; - } - - if (c == '.') { - - /* - * Internet format: - * a.b.c.d - * a.b.c (with c treated as 16 bits) - * a.b (with b treated as 24 bits) - */ - if (pp >= parts + 3) - return (0); - *pp++ = val; - c = *++cp; - } - else - break; - } - - /* Check for trailing characters */ - if (c != '\0' && (!isascii ((int) c) || !isspace ((int) c))) - return (0); - - /* Concoct the address according to the number of parts specified */ - n = pp - parts + 1; - switch (n) { - - case 0: - return (0); /* initial nondigit */ - - case 1: /* a -- 32 bits */ - break; - - case 2: /* a.b -- 8.24 bits */ - if (val > 0xffffff) - return (0); - val |= parts[0] << 24; - break; - - case 3: /* a.b.c -- 8.8.16 bits */ - if (val > 0xffff) - return (0); - val |= (parts[0] << 24) | (parts[1] << 16); - break; - - case 4: /* a.b.c.d -- 8.8.8.8 bits */ - if (val > 0xff) - return (0); - val |= (parts[0] << 24) | (parts[1] << 16) | (parts[2] << 8); - break; - } - - if (addr) - addr->s_addr = htonl (val); - - return (1); + struct addrinfo hints; + struct addrinfo *res; + struct protoent *ptrp; + char port_str[6]; + int result; + + memset (&hints, 0, sizeof (hints)); + hints.ai_family = PF_UNSPEC; + hints.ai_protocol = proto; + + snprintf(port_str, sizeof(port_str), "%d", port); + /* nice clean AF-independent code doesn't matter if passed hostname or IP address */ + result = getaddrinfo (host_name, port_str, &hints, &res); + + if (result != 0) + { + printf ("%s\n", gai_strerror (result)); + return STATE_UNKNOWN; + } + else + { + while (res) + { + /* attempt to create a socket */ + *sd = socket (res->ai_family, (proto == IPPROTO_UDP) ? SOCK_DGRAM : SOCK_STREAM, res->ai_protocol); + + if (*sd < 0) { + printf ("Socket creation failed\n"); + freeaddrinfo (res); + return STATE_UNKNOWN; + } + + /* attempt to open a connection */ + result = connect (*sd, res->ai_addr, res->ai_addrlen); + + if (result == 0) + break; + + close(*sd); + res = res->ai_next; + } + freeaddrinfo (res); + } + + if (result == 0) + return STATE_OK; + else + { + printf ("%s\n", strerror(errno)); + return STATE_CRITICAL; + } } Index: netutils.h.in =================================================================== RCS file: /cvsroot/nagiosplug/nagiosplug/plugins/netutils.h.in,v retrieving revision 1.1.1.1 diff -u -r1.1.1.1 netutils.h.in --- netutils.h.in 28 Feb 2002 06:42:59 -0000 1.1.1.1 +++ netutils.h.in 11 Dec 2002 02:38:43 -0000 @@ -42,19 +42,17 @@ void socket_timeout_alarm_handler (int); int process_tcp_request2 (char *address, int port, char *sbuffer, - char *rbuffer, int rsize); + char *rbuffer, int rsize); int process_tcp_request (char *address, int port, char *sbuffer, - char *rbuffer, int rsize); + char *rbuffer, int rsize); int process_udp_request (char *address, int port, char *sbuffer, - char *rbuffer, int rsize); -int process_request (char *address, int port, char *proto, char *sbuffer, - char *rbuffer, int rsize); + char *rbuffer, int rsize); +int process_request (char *address, int port, int proto, char *sbuffer, + char *rbuffer, int rsize); int my_tcp_connect (char *address, int port, int *sd); int my_udp_connect (char *address, int port, int *sd); -int my_connect (char *address, int port, int *sd, char *proto); - -int my_inet_aton (register const char *cp, struct in_addr *addr); +int my_connect (char *address, int port, int *sd, int proto); #ifndef DEFAULT_SOCKET_TIMEOUT #include "config.h"