Code

properly call close() on the ssh connection before exiting.
[nagiosplug.git] / plugins / check_ssh.c
index fdded583b70ea06c1163f7227408f4619a3449e3..5617addc66baded12d9ead64636878626471b551 100644 (file)
  along with this program; if not, write to the Free Software
  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 
+ $Id$
 ******************************************************************************/
 
-#include "common.h"
-#include "netutils.h"
-#include "utils.h"
-
 const char *progname = "check_ssh";
 const char *revision = "$Revision$";
-const char *copyright = "2000-2003";
+const char *copyright = "2000-2004";
 const char *email = "nagiosplug-devel@lists.sourceforge.net";
 
+#include "common.h"
+#include "netutils.h"
+#include "utils.h"
+
 #ifndef MSG_DONTWAIT
 #define MSG_DONTWAIT 0
 #endif
@@ -34,6 +36,7 @@ const char *email = "nagiosplug-devel@lists.sourceforge.net";
 
 int port = -1;
 char *server_name = NULL;
+char *remote_version = NULL;
 int verbose = FALSE;
 
 int process_arguments (int, char **);
@@ -41,22 +44,29 @@ int validate_arguments (void);
 void print_help (void);
 void print_usage (void);
 
-int ssh_connect (char *haddr, int hport);
+int ssh_connect (char *haddr, int hport, char *remote_version);
+
+
 
 int
 main (int argc, char **argv)
 {
-       int result;
+       int result = STATE_UNKNOWN;
+
+       setlocale (LC_ALL, "");
+       bindtextdomain (PACKAGE, LOCALEDIR);
+       textdomain (PACKAGE);
 
        if (process_arguments (argc, argv) == ERROR)
-               usage (_("Could not parse arguments\n"));
+               usage4 (_("Could not parse arguments"));
 
        /* initialize alarm signal handling */
        signal (SIGALRM, socket_timeout_alarm_handler);
+       
        alarm (socket_timeout);
 
        /* ssh_connect exits if error is found */
-       result = ssh_connect (server_name, port);
+       result = ssh_connect (server_name, port, remote_version);
 
        alarm (0);
 
@@ -64,14 +74,15 @@ main (int argc, char **argv)
 }
 
 
+
 /* process command-line arguments */
 int
 process_arguments (int argc, char **argv)
 {
        int c;
 
-       int option_index = 0;
-       static struct option long_options[] = {
+       int option = 0;
+       static struct option longopts[] = {
                {"help", no_argument, 0, 'h'},
                {"version", no_argument, 0, 'V'},
                {"host", required_argument, 0, 'H'},
@@ -80,6 +91,7 @@ process_arguments (int argc, char **argv)
                {"use-ipv6", no_argument, 0, '6'},
                {"timeout", required_argument, 0, 't'},
                {"verbose", no_argument, 0, 'v'},
+               {"remote-version", required_argument, 0, 'r'},
                {0, 0, 0, 0}
        };
 
@@ -91,26 +103,26 @@ process_arguments (int argc, char **argv)
                        strcpy (argv[c], "-t");
 
        while (1) {
-               c = getopt_long (argc, argv, "+Vhv46t:H:p:", long_options, &option_index);
+               c = getopt_long (argc, argv, "+Vhv46t:r:H:p:", longopts, &option);
 
                if (c == -1 || c == EOF)
                        break;
 
                switch (c) {
                case '?':                                                                       /* help */
-                       usage ("");
+                       usage2 (_("Unknown argument"), optarg);
                case 'V':                                                                       /* version */
                        print_revision (progname, revision);
                        exit (STATE_OK);
                case 'h':                                                                       /* help */
                        print_help ();
                        exit (STATE_OK);
-               case 'v':                                                                       /* verose */
+               case 'v':                                                                       /* verbose */
                        verbose = TRUE;
                        break;
                case 't':                                                                       /* timeout period */
                        if (!is_integer (optarg))
-                               usage (_("Timeout Interval must be an integer!\n\n"));
+                               usage2 (_("Timeout interval must be a positive integer"), optarg);
                        else
                                socket_timeout = atoi (optarg);
                        break;
@@ -121,12 +133,15 @@ process_arguments (int argc, char **argv)
 #ifdef USE_IPV6
                        address_family = AF_INET6;
 #else
-                       usage (_("IPv6 support not available\n"));
+                       usage4 (_("IPv6 support not available"));
 #endif
                        break;
+               case 'r':                                                                       /* remote version */
+                       remote_version = optarg;
+                       break;
                case 'H':                                                                       /* host */
                        if (is_host (optarg) == FALSE)
-                               usage ("Invalid hostname/address\n");
+                               usage2 (_("Invalid hostname/address"), optarg);
                        server_name = optarg;
                        break;
                case 'p':                                                                       /* port */
@@ -134,11 +149,9 @@ process_arguments (int argc, char **argv)
                                port = atoi (optarg);
                        }
                        else {
-                               printf ("Port number nust be a positive integer: %s\n", optarg);
-                               usage ("");
+                               usage2 (_("Port number must be a positive integer"), optarg);
                        }
                }
-
        }
 
        c = optind;
@@ -178,8 +191,9 @@ validate_arguments (void)
 *
 *-----------------------------------------------------------------------*/
 
+
 int
-ssh_connect (char *haddr, int hport)
+ssh_connect (char *haddr, int hport, char *remote_version)
 {
        int sd;
        int result;
@@ -210,17 +224,29 @@ ssh_connect (char *haddr, int hport)
                ssh_proto = output + 4;
                ssh_server = ssh_proto + strspn (ssh_proto, "-0123456789. ");
                ssh_proto[strspn (ssh_proto, "0123456789. ")] = 0;
-               printf
-                       (_("SSH OK - %s (protocol %s)\n"),
-                        ssh_server, ssh_proto);
+
                asprintf (&buffer, "SSH-%s-check_ssh_%s\r\n", ssh_proto, rev_no);
                send (sd, buffer, strlen (buffer), MSG_DONTWAIT);
                if (verbose)
                        printf ("%s\n", buffer);
+
+               if (remote_version && strcmp(remote_version, ssh_server)) {
+                       printf
+                               (_("SSH WARNING - %s (protocol %s) version mismatch, expected '%s'\n"),
+                                ssh_server, ssh_proto, remote_version);
+                       exit (STATE_WARNING);
+               }
+               
+               printf
+                       (_("SSH OK - %s (protocol %s)\n"),
+                        ssh_server, ssh_proto);
+               close(sd);
                exit (STATE_OK);
        }
 }
 
+
+
 void
 print_help (void)
 {
@@ -229,10 +255,10 @@ print_help (void)
 
        print_revision (progname, revision);
 
-       printf (_("Copyright (c) 1999 Remi Paulmier <remi@sinfomic.fr>\n"));
-       printf (_(COPYRIGHT), copyright, email);
+       printf ("Copyright (c) 1999 Remi Paulmier <remi@sinfomic.fr>\n");
+       printf (COPYRIGHT, copyright, email);
 
-       printf (_("Try to connect to SSH server at specified server and port\n\n"));
+       printf (_("Try to connect to an SSH server at specified server and port\n\n"));
 
        print_usage ();
 
@@ -244,17 +270,21 @@ print_help (void)
 
        printf (_(UT_TIMEOUT), DEFAULT_SOCKET_TIMEOUT);
 
+       printf (_("\
+ -r, --remote-version=STRING\n\
+    Warn if string doesn't match expected server version (ex: OpenSSH_3.9p1)\n"));
+       
        printf (_(UT_VERBOSE));
 
        printf (_(UT_SUPPORT));
 }
 
+
+
 void
 print_usage (void)
 {
-       printf (_("\
-Usage: %s [-46] [-t <timeout>] [-p <port>] <host>\n"), progname);
-       printf (_(UT_HLP_VRS), progname, progname);
+       printf ("\
+Usage: %s [-46] [-t <timeout>] [-r <remote version>] [-p <port>] <host>\n", progname);
 }
 
-/* end of check_ssh.c */