Initial revision

git-svn-id: https://nagiosplug.svn.sourceforge.net/svnroot/nagiosplug/nagiosplug/trunk@2 f882894a-f735-0410-b71e-b25c423dba1c

1 files changed, 285 insertions, 0 deletions

diff --git a/plugins/check_ssh.c b/plugins/check_ssh.c
new file mode 100644
index 00000000..d78189fc
--- /dev/null
+++ b/plugins/check_ssh.c
@@ -0,0 +1,285 @@
+/*
+* check_ssh.c
+* 
+* Made by (Remi PAULMIER)
+* Login   <remi@sinfomic.fr>
+* 
+* Started on  Fri Jul  9 09:18:23 1999 Remi PAULMIER
+* Update Thu Jul 22 12:50:04 1999 remi paulmier
+* $Id$
+*
+*/
+
+#include "config.h"
+#include "common.h"
+#include "netutils.h"
+#include "utils.h"
+
+#define PROGNAME "check_ssh"
+
+#ifndef MSG_DONTWAIT
+#define MSG_DONTWAIT 0
+#endif
+
+#define SSH_DFL_PORT    22
+#define BUFF_SZ         256
+
+short port = -1;
+char *server_name = NULL;
+int verbose = FALSE;
+
+int process_arguments (int, char **);
+int call_getopt (int, char **);
+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)
+{
+
+	if (process_arguments (argc, argv) == ERROR)
+		usage ("Could not parse arguments\n");
+
+	/* initialize alarm signal handling */
+	signal (SIGALRM, socket_timeout_alarm_handler);
+	alarm (socket_timeout);
+
+	/* ssh_connect exits if error is found */
+	ssh_connect (ssh_resolve (server_name), port);
+
+	alarm (0);
+
+	return (STATE_OK);
+}
+
+
+/* process command-line arguments */
+int
+process_arguments (int argc, char **argv)
+{
+	int c;
+
+	if (argc < 2)
+		return ERROR;
+
+	for (c = 1; c < argc; c++)
+		if (strcmp ("-to", argv[c]) == 0)
+			strcpy (argv[c], "-t");
+
+	c = 0;
+	while (c += (call_getopt (argc - c, &argv[c]))) {
+		if (argc <= c)
+			break;
+		if (server_name == NULL) {
+			server_name = argv[c];
+		}
+		else if (port == -1) {
+			if (is_intpos (argv[c])) {
+				port = atoi (argv[c]);
+			}
+			else {
+				print_usage ();
+				exit (STATE_UNKNOWN);
+			}
+		}
+	}
+
+	return validate_arguments ();
+}
+
+
+/************************************************************************
+*
+* Run the getopt until we encounter a non-option entry in the arglist
+*
+*-----------------------------------------------------------------------*/
+
+int
+call_getopt (int argc, char **argv)
+{
+	int c, i = 1;
+
+#ifdef HAVE_GETOPT_H
+	int option_index = 0;
+	static struct option long_options[] = {
+		{"version", no_argument, 0, 'V'},
+		{"help", no_argument, 0, 'h'},
+		{"verbose", no_argument, 0, 'v'},
+		{"timeout", required_argument, 0, 't'},
+		{"host", required_argument, 0, 'H'},
+		{0, 0, 0, 0}
+	};
+#endif
+
+	while (1) {
+#ifdef HAVE_GETOPT_H
+		c = getopt_long (argc, argv, "+Vhvt:H:p:", long_options, &option_index);
+#else
+		c = getopt (argc, argv, "+Vhvt:H:p:");
+#endif
+
+		if (c == -1 || c == EOF)
+			break;
+
+		i++;
+		switch (c) {
+		case 't':
+		case 'H':
+		case 'p':
+			i++;
+		}
+
+		switch (c) {
+		case '?':									/* help */
+			usage ("");
+		case 'V':									/* version */
+			print_revision (my_basename (argv[0]), "$Revision$");
+			exit (STATE_OK);
+		case 'h':									/* help */
+			print_help ();
+			exit (STATE_OK);
+		case 'v':									/* verose */
+			verbose = TRUE;
+			break;
+		case 't':									/* timeout period */
+			if (!is_integer (optarg))
+				usage ("Timeout Interval must be an integer!\n\n");
+			socket_timeout = atoi (optarg);
+			break;
+		case 'H':									/* host */
+			server_name = optarg;
+			break;
+		case 'p':									/* port */
+			if (is_intpos (optarg)) {
+				port = atoi (optarg);
+			}
+			else {
+				printf ("Port number nust be a positive integer: %s\n", optarg);
+				usage ("");
+			}
+		}
+
+	}
+	return i;
+}
+
+int
+validate_arguments (void)
+{
+	if (server_name == NULL)
+		return ERROR;
+	if (port == -1)								/* funky, but allows -p to override stray integer in args */
+		port = SSH_DFL_PORT;
+	return OK;
+}
+
+
+/************************************************************************
+*
+* 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
+*
+*-----------------------------------------------------------------------*/
+
+int
+ssh_connect (char *haddr, short hport)
+{
+	int s;
+	struct sockaddr_in addr;
+	int addrlen;
+	int len;
+	char *output = NULL;
+	char *buffer = NULL;
+	char *ssh_proto = NULL;
+	char *ssh_server = NULL;
+	char revision[20];
+
+	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);
+
+	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);
+	}
+
+	output = (char *) malloc (BUFF_SZ + 1);
+	memset (output, 0, BUFF_SZ + 1);
+	recv (s, output, BUFF_SZ, 0);
+	if (strncmp (output, "SSH", 3)) {
+		printf ("Server answer: %s", output);
+		exit (STATE_CRITICAL);
+	}
+	else {
+		strip (output);
+		if (verbose)
+			printf ("%s\n", output);
+		ssh_proto = output + 4;
+		ssh_server = ssh_proto + strspn (ssh_proto, "0123456789-. ");
+		ssh_proto[strspn (ssh_proto, "0123456789-. ")] = 0;
+		printf
+			("SSH ok - protocol version %s - server version %s\n",
+			 ssh_proto, ssh_server);
+		buffer =
+			ssprintf (buffer, "SSH-%s-check_ssh_%s\r\n", ssh_proto, revision);
+		send (s, buffer, strlen (buffer), MSG_DONTWAIT);
+		if (verbose)
+			printf ("%s\n", buffer);
+		exit (STATE_OK);
+	}
+}
+
+void
+print_help (void)
+{
+	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);
+}
+
+void
+print_usage (void)
+{
+	printf
+		("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);
+}
+
+/* end of check_ssh.c */