3efc2e625ad9aa12b8db68d013222506cd4c3798
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 {
43 if (process_arguments (argc, argv) == ERROR)
44 usage ("Could not parse arguments\n");
46 /* initialize alarm signal handling */
47 signal (SIGALRM, socket_timeout_alarm_handler);
48 alarm (socket_timeout);
50 /* ssh_connect exits if error is found */
51 ssh_connect (server_name, port);
53 alarm (0);
55 return (STATE_OK);
56 }
59 /* process command-line arguments */
60 int
61 process_arguments (int argc, char **argv)
62 {
63 int c;
64 char *tmp = NULL;
66 #ifdef HAVE_GETOPT_H
67 int option_index = 0;
68 static struct option long_options[] = {
69 {"version", no_argument, 0, 'V'},
70 {"help", no_argument, 0, 'h'},
71 {"verbose", no_argument, 0, 'v'},
72 {"timeout", required_argument, 0, 't'},
73 {"host", required_argument, 0, 'H'},
74 {0, 0, 0, 0}
75 };
76 #endif
78 if (argc < 2)
79 return ERROR;
81 for (c = 1; c < argc; c++)
82 if (strcmp ("-to", argv[c]) == 0)
83 strcpy (argv[c], "-t");
85 while (1) {
86 #ifdef HAVE_GETOPT_H
87 c = getopt_long (argc, argv, "+Vhvt:H:p:", long_options, &option_index);
88 #else
89 c = getopt (argc, argv, "+Vhvt:H:p:");
90 #endif
91 if (c == -1 || c == EOF)
92 break;
94 switch (c) {
95 case '?': /* help */
96 usage ("");
97 case 'V': /* version */
98 print_revision (progname, REVISION);
99 exit (STATE_OK);
100 case 'h': /* help */
101 print_help ();
102 exit (STATE_OK);
103 case 'v': /* verose */
104 verbose = TRUE;
105 break;
106 case 't': /* timeout period */
107 if (!is_integer (optarg))
108 usage ("Timeout Interval must be an integer!\n\n");
109 socket_timeout = atoi (optarg);
110 break;
111 case 'H': /* host */
112 if (is_host (optarg) == FALSE)
113 usage ("Invalid hostname/address\n");
114 server_name = optarg;
115 break;
116 case 'p': /* port */
117 if (is_intpos (optarg)) {
118 port = atoi (optarg);
119 }
120 else {
121 printf ("Port number nust be a positive integer: %s\n", optarg);
122 usage ("");
123 }
124 }
126 }
128 c = optind;
129 if (server_name == NULL && argv[c]) {
130 if (is_host (argv[c])) {
131 server_name = argv[c++];
132 }
133 }
135 if (port == -1 && argv[c]) {
136 if (is_intpos (argv[c])) {
137 port = atoi (argv[c++]);
138 }
139 else {
140 print_usage ();
141 exit (STATE_UNKNOWN);
142 }
143 }
145 return validate_arguments ();
146 }
148 int
149 validate_arguments (void)
150 {
151 if (server_name == NULL)
152 return ERROR;
153 if (port == -1) /* funky, but allows -p to override stray integer in args */
154 port = SSH_DFL_PORT;
155 return OK;
156 }
159 /************************************************************************
160 *
161 * Try to connect to SSH server at specified server and port
162 *
163 *-----------------------------------------------------------------------*/
165 int
166 ssh_connect (char *haddr, short hport)
167 {
168 int sd;
169 int result;
170 char *output = NULL;
171 char *buffer = NULL;
172 char *ssh_proto = NULL;
173 char *ssh_server = NULL;
174 char revision[20];
176 sscanf ("$Revision$", "$Revision: %[0123456789.]", revision);
178 result = my_tcp_connect (haddr, hport, &sd);
180 if (result != STATE_OK)
181 return result;
183 output = (char *) malloc (BUFF_SZ + 1);
184 memset (output, 0, BUFF_SZ + 1);
185 recv (sd, output, BUFF_SZ, 0);
186 if (strncmp (output, "SSH", 3)) {
187 printf ("Server answer: %s", output);
188 exit (STATE_CRITICAL);
189 }
190 else {
191 strip (output);
192 if (verbose)
193 printf ("%s\n", output);
194 ssh_proto = output + 4;
195 ssh_server = ssh_proto + strspn (ssh_proto, "-0123456789. ");
196 ssh_proto[strspn (ssh_proto, "0123456789. ")] = 0;
197 printf
198 ("SSH ok - %s (protocol %s)\n",
199 ssh_server, ssh_proto);
200 asprintf (&buffer, "SSH-%s-check_ssh_%s\r\n", ssh_proto, revision);
201 send (sd, buffer, strlen (buffer), MSG_DONTWAIT);
202 if (verbose)
203 printf ("%s\n", buffer);
204 exit (STATE_OK);
205 }
206 }
208 void
209 print_help (void)
210 {
211 print_revision (progname, REVISION);
212 printf ("Copyright (c) 1999 Remi Paulmier (remi@sinfomic.fr)\n\n");
213 print_usage ();
214 printf ("by default, port is %d\n", SSH_DFL_PORT);
215 }
217 void
218 print_usage (void)
219 {
220 printf
221 ("Usage:\n"
222 " %s -t [timeout] -p [port] <host>\n"
223 " %s -V prints version info\n"
224 " %s -h prints more detailed help\n", progname, progname, progname);
225 }
227 /* end of check_ssh.c */