Added getaddrinfo.[ch] & gethostbyname.[ch] to provide RFC2553 functions

[nagiosplug.git] / plugins / check_ssh.c

diff --git a/plugins/check_ssh.c b/plugins/check_ssh.c
index 4637bcee0c781280567a8308de4bdc7dc6231886..bd0a2451a44a9ec54d2c973ba378c75c9364e5b2 100644 (file)
--- a/plugins/check_ssh.c
+++ b/plugins/check_ssh.c
@@ -15,7 +15,7 @@
 #include "netutils.h"
 #include "utils.h"
 
-#define PROGNAME "check_ssh"
+const char *progname = "check_ssh";
 #define REVISION "$Revision$"
 
 #ifndef MSG_DONTWAIT
@@ -34,12 +34,12 @@ int validate_arguments (void);
 void print_help (void);
 void print_usage (void);
 
-char *ssh_resolve (char *hostname);
 int ssh_connect (char *haddr, short hport);
 
 int
 main (int argc, char **argv)
 {
+       int result;
 
        if (process_arguments (argc, argv) == ERROR)
                usage ("Could not parse arguments\n");
@@ -49,11 +49,11 @@ main (int argc, char **argv)
        alarm (socket_timeout);
 
        /* ssh_connect exits if error is found */
-       ssh_connect (ssh_resolve (server_name), port);
+       result = ssh_connect (server_name, port);
 
        alarm (0);
 
-       return (STATE_OK);
+       return (result);
 }
 
 
@@ -62,6 +62,7 @@ int
 process_arguments (int argc, char **argv)
 {
        int c;
+       char *tmp = NULL;
 
 #ifdef HAVE_GETOPT_H
        int option_index = 0;
@@ -95,7 +96,7 @@ process_arguments (int argc, char **argv)
                case '?':                                                                       /* help */
                        usage ("");
                case 'V':                                                                       /* version */
-                       print_revision (PROGNAME, REVISION);
+                       print_revision (progname, REVISION);
                        exit (STATE_OK);
                case 'h':                                                                       /* help */
                        print_help ();
@@ -109,6 +110,8 @@ process_arguments (int argc, char **argv)
                        socket_timeout = atoi (optarg);
                        break;
                case 'H':                                                                       /* host */
+                       if (is_host (optarg) == FALSE)
+                               usage ("Invalid hostname/address\n");
                        server_name = optarg;
                        break;
                case 'p':                                                                       /* port */
@@ -124,10 +127,13 @@ process_arguments (int argc, char **argv)
        }
 
        c = optind;
-       if (server_name == NULL && argv[c]) {
-               server_name = argv[c++];
+       if (server_name == NULL && c < argc) {
+               if (is_host (argv[c])) {
+                       server_name = argv[c++];
+               }
        }
-       else if (port == -1 && argv[c]) {
+
+       if (port == -1 && c < argc) {
                if (is_intpos (argv[c])) {
                        port = atoi (argv[c++]);
                }
@@ -151,26 +157,6 @@ validate_arguments (void)
 }
 
 
-/************************************************************************
-*
-* Resolve hostname into IP address
-*
-*-----------------------------------------------------------------------*/
-
-char *
-ssh_resolve (char *hostname)
-{
-       struct hostent *host;
-
-       host = gethostbyname (hostname);
-       if (!host) {
-               herror (hostname);
-               exit (STATE_CRITICAL);
-       }
-       return (host->h_addr);
-}
-
-
 /************************************************************************
 *
 * Try to connect to SSH server at specified server and port
@@ -180,10 +166,8 @@ ssh_resolve (char *hostname)
 int
 ssh_connect (char *haddr, short hport)
 {
-       int s;
-       struct sockaddr_in addr;
-       int addrlen;
-       int len;
+       int sd;
+       int result;
        char *output = NULL;
        char *buffer = NULL;
        char *ssh_proto = NULL;
@@ -192,27 +176,14 @@ ssh_connect (char *haddr, short hport)
 
        sscanf ("$Revision$", "$Revision: %[0123456789.]", revision);
 
-       addrlen = sizeof (addr);
-       memset (&addr, 0, addrlen);
-       addr.sin_port = htons (hport);
-       addr.sin_family = AF_INET;
-       bcopy (haddr, (void *) &addr.sin_addr.s_addr, 4);
+       result = my_tcp_connect (haddr, hport, &sd);
 
-       s = socket (AF_INET, SOCK_STREAM, 0);
-       if (!s) {
-               printf ("socket(): %s for %s:%d\n", strerror (errno), server_name, hport);
-               exit (STATE_CRITICAL);
-       }
-
-       if (connect (s, (struct sockaddr *) &addr, addrlen)) {
-               printf ("connect(): %s for %s:%d\n", strerror (errno), server_name,
-                                               hport);
-               exit (STATE_CRITICAL);
-       }
+       if (result != STATE_OK)
+               return result;
 
        output = (char *) malloc (BUFF_SZ + 1);
        memset (output, 0, BUFF_SZ + 1);
-       recv (s, output, BUFF_SZ, 0);
+       recv (sd, output, BUFF_SZ, 0);
        if (strncmp (output, "SSH", 3)) {
                printf ("Server answer: %s", output);
                exit (STATE_CRITICAL);
@@ -225,10 +196,10 @@ ssh_connect (char *haddr, short hport)
                ssh_server = ssh_proto + strspn (ssh_proto, "-0123456789. ");
                ssh_proto[strspn (ssh_proto, "0123456789. ")] = 0;
                printf
-                       ("SSH ok - %s (protocol %s)\n",
+                       ("SSH OK - %s (protocol %s)\n",
                         ssh_server, ssh_proto);
                asprintf (&buffer, "SSH-%s-check_ssh_%s\r\n", ssh_proto, revision);
-               send (s, buffer, strlen (buffer), MSG_DONTWAIT);
+               send (sd, buffer, strlen (buffer), MSG_DONTWAIT);
                if (verbose)
                        printf ("%s\n", buffer);
                exit (STATE_OK);
@@ -238,7 +209,7 @@ ssh_connect (char *haddr, short hport)
 void
 print_help (void)
 {
-       print_revision (PROGNAME, REVISION);
+       print_revision (progname, REVISION);
        printf ("Copyright (c) 1999 Remi Paulmier (remi@sinfomic.fr)\n\n");
        print_usage ();
        printf ("by default, port is %d\n", SSH_DFL_PORT);
@@ -251,7 +222,7 @@ print_usage (void)
                ("Usage:\n"
                 " %s -t [timeout] -p [port] <host>\n"
                 " %s -V prints version info\n"
-                " %s -h prints more detailed help\n", PROGNAME, PROGNAME, PROGNAME);
+                " %s -h prints more detailed help\n", progname, progname, progname);
 }
 
 /* end of check_ssh.c */