1 /*
2 * check_ssh.c
3 *
4 * Made by (Remi PAULMIER)
5 * Login <remi@sinfomic.fr>
6 *
7 * Started on Fri Jul 9 09:18:23 1999 Remi PAULMIER
8 * Update Thu Jul 22 12:50:04 1999 remi paulmier
9 * $Id$
10 *
11 */
13 #include "config.h"
14 #include "common.h"
15 #include "netutils.h"
16 #include "utils.h"
18 const char *progname = "check_ssh";
19 #define REVISION "$Revision$"
21 #ifndef MSG_DONTWAIT
22 #define MSG_DONTWAIT 0
23 #endif
25 #define SSH_DFL_PORT 22
26 #define BUFF_SZ 256
28 short port = -1;
29 char *server_name = NULL;
30 int verbose = FALSE;
32 int process_arguments (int, char **);
33 int validate_arguments (void);
34 void print_help (void);
35 void print_usage (void);
37 int ssh_connect (char *haddr, short hport);
39 int
40 main (int argc, char **argv)
41 {
42 int result;
44 if (process_arguments (argc, argv) == ERROR)
45 usage ("Could not parse arguments\n");
47 /* initialize alarm signal handling */
48 signal (SIGALRM, socket_timeout_alarm_handler);
49 alarm (socket_timeout);
51 /* ssh_connect exits if error is found */
52 result = ssh_connect (server_name, port);
54 alarm (0);
56 return (result);
57 }
60 /* process command-line arguments */
61 int
62 process_arguments (int argc, char **argv)
63 {
64 int c;
65 char *tmp = NULL;
67 #ifdef HAVE_GETOPT_H
68 int option_index = 0;
69 static struct option long_options[] = {
70 {"version", no_argument, 0, 'V'},
71 {"help", no_argument, 0, 'h'},
72 {"verbose", no_argument, 0, 'v'},
73 {"timeout", required_argument, 0, 't'},
74 {"host", required_argument, 0, 'H'},
75 {0, 0, 0, 0}
76 };
77 #endif
79 if (argc < 2)
80 return ERROR;
82 for (c = 1; c < argc; c++)
83 if (strcmp ("-to", argv[c]) == 0)
84 strcpy (argv[c], "-t");
86 while (1) {
87 #ifdef HAVE_GETOPT_H
88 c = getopt_long (argc, argv, "+Vhvt:H:p:", long_options, &option_index);
89 #else
90 c = getopt (argc, argv, "+Vhvt:H:p:");
91 #endif
92 if (c == -1 || c == EOF)
93 break;
95 switch (c) {
96 case '?': /* help */
97 usage ("");
98 case 'V': /* version */
99 print_revision (progname, REVISION);
100 exit (STATE_OK);
101 case 'h': /* help */
102 print_help ();
103 exit (STATE_OK);
104 case 'v': /* verose */
105 verbose = TRUE;
106 break;
107 case 't': /* timeout period */
108 if (!is_integer (optarg))
109 usage ("Timeout Interval must be an integer!\n\n");
110 socket_timeout = atoi (optarg);
111 break;
112 case 'H': /* host */
113 if (is_host (optarg) == FALSE)
114 usage ("Invalid hostname/address\n");
115 server_name = optarg;
116 break;
117 case 'p': /* port */
118 if (is_intpos (optarg)) {
119 port = atoi (optarg);
120 }
121 else {
122 printf ("Port number nust be a positive integer: %s\n", optarg);
123 usage ("");
124 }
125 }
127 }
129 c = optind;
130 if (server_name == NULL && c < argc) {
131 if (is_host (argv[c])) {
132 server_name = argv[c++];
133 }
134 }
136 if (port == -1 && c < argc) {
137 if (is_intpos (argv[c])) {
138 port = atoi (argv[c++]);
139 }
140 else {
141 print_usage ();
142 exit (STATE_UNKNOWN);
143 }
144 }
146 return validate_arguments ();
147 }
149 int
150 validate_arguments (void)
151 {
152 if (server_name == NULL)
153 return ERROR;
154 if (port == -1) /* funky, but allows -p to override stray integer in args */
155 port = SSH_DFL_PORT;
156 return OK;
157 }
160 /************************************************************************
161 *
162 * Try to connect to SSH server at specified server and port
163 *
164 *-----------------------------------------------------------------------*/
166 int
167 ssh_connect (char *haddr, short hport)
168 {
169 int sd;
170 int result;
171 char *output = NULL;
172 char *buffer = NULL;
173 char *ssh_proto = NULL;
174 char *ssh_server = NULL;
175 char revision[20];
177 sscanf ("$Revision$", "$Revision: %[0123456789.]", revision);
179 result = my_tcp_connect (haddr, hport, &sd);
181 if (result != STATE_OK)
182 return result;
184 output = (char *) malloc (BUFF_SZ + 1);
185 memset (output, 0, BUFF_SZ + 1);
186 recv (sd, output, BUFF_SZ, 0);
187 if (strncmp (output, "SSH", 3)) {
188 printf ("Server answer: %s", output);
189 exit (STATE_CRITICAL);
190 }
191 else {
192 strip (output);
193 if (verbose)
194 printf ("%s\n", output);
195 ssh_proto = output + 4;
196 ssh_server = ssh_proto + strspn (ssh_proto, "-0123456789. ");
197 ssh_proto[strspn (ssh_proto, "0123456789. ")] = 0;
198 printf
199 ("SSH OK - %s (protocol %s)\n",
200 ssh_server, ssh_proto);
201 asprintf (&buffer, "SSH-%s-check_ssh_%s\r\n", ssh_proto, revision);
202 send (sd, buffer, strlen (buffer), MSG_DONTWAIT);
203 if (verbose)
204 printf ("%s\n", buffer);
205 exit (STATE_OK);
206 }
207 }
209 void
210 print_help (void)
211 {
212 print_revision (progname, REVISION);
213 printf ("Copyright (c) 1999 Remi Paulmier (remi@sinfomic.fr)\n\n");
214 print_usage ();
215 printf ("by default, port is %d\n", SSH_DFL_PORT);
216 }
218 void
219 print_usage (void)
220 {
221 printf
222 ("Usage:\n"
223 " %s -t [timeout] -p [port] <host>\n"
224 " %s -V prints version info\n"
225 " %s -h prints more detailed help\n", progname, progname, progname);
226 }
228 /* end of check_ssh.c */