diff options
Diffstat (limited to 'plugins/check_imap.c')
-rw-r--r-- | plugins/check_imap.c | 340 |
1 files changed, 340 insertions, 0 deletions
diff --git a/plugins/check_imap.c b/plugins/check_imap.c new file mode 100644 index 00000000..5ec0439b --- /dev/null +++ b/plugins/check_imap.c | |||
@@ -0,0 +1,340 @@ | |||
1 | /****************************************************************************** | ||
2 | * | ||
3 | * CHECK_IMAP.C | ||
4 | * | ||
5 | * Program: IMAP4 plugin for Nagios | ||
6 | * License: GPL | ||
7 | * Copyright (c) 1999 Tom Shields (tom.shields@basswood.com) | ||
8 | * | ||
9 | * $Id$ | ||
10 | * | ||
11 | * Description: | ||
12 | * | ||
13 | * This plugin will attempt to open an IMAP connection with the host. | ||
14 | * Successul connects return STATE_OK, refusals and timeouts return | ||
15 | * STATE_CRITICAL, other errors return STATE_UNKNOWN. Successful | ||
16 | * connects, but incorrect reponse messages from the host result in | ||
17 | * STATE_WARNING return values. | ||
18 | * | ||
19 | * Modifications: | ||
20 | * 04-13-1999 Tom Shields | ||
21 | * Initial code | ||
22 | * 08-18-1999 Ethan Galstad <nagios@nagios.org> | ||
23 | * Modified code to work with common plugin functions, added socket | ||
24 | * timeout, string * length checking | ||
25 | * 09-19-1999 Ethan Galstad <nagios@nagios.org> | ||
26 | * Changed expect string from "+OK" to "* OK" and default port to 143 | ||
27 | *****************************************************************************/ | ||
28 | |||
29 | #include "config.h" | ||
30 | #include "common.h" | ||
31 | #include "netutils.h" | ||
32 | #include "utils.h" | ||
33 | |||
34 | #define PROGNAME "check_imap" | ||
35 | |||
36 | #define PORT 143 | ||
37 | #define EXPECT "* OK" | ||
38 | #define QUIT "a1 LOGOUT\n" | ||
39 | |||
40 | int process_arguments (int, char **); | ||
41 | int call_getopt (int, char **); | ||
42 | int validate_arguments (void); | ||
43 | int check_disk (int usp, int free_disk); | ||
44 | void print_help (void); | ||
45 | void print_usage (void); | ||
46 | |||
47 | int server_port = PORT; | ||
48 | char *server_address = NULL; | ||
49 | char *server_expect = NULL; | ||
50 | int warning_time = 0; | ||
51 | int check_warning_time = FALSE; | ||
52 | int critical_time = 0; | ||
53 | int check_critical_time = FALSE; | ||
54 | int verbose = FALSE; | ||
55 | |||
56 | int | ||
57 | main (int argc, char **argv) | ||
58 | { | ||
59 | int sd; | ||
60 | int result; | ||
61 | char buffer[MAX_INPUT_BUFFER]; | ||
62 | |||
63 | if (process_arguments (argc, argv) != OK) | ||
64 | usage ("Invalid command arguments supplied\n"); | ||
65 | |||
66 | /* initialize alarm signal handling */ | ||
67 | signal (SIGALRM, socket_timeout_alarm_handler); | ||
68 | |||
69 | /* set socket timeout */ | ||
70 | alarm (socket_timeout); | ||
71 | |||
72 | /* try to connect to the host at the given port number */ | ||
73 | time (&start_time); | ||
74 | result = my_tcp_connect (server_address, server_port, &sd); | ||
75 | |||
76 | /* we connected, so close connection before exiting */ | ||
77 | if (result == STATE_OK) { | ||
78 | |||
79 | /* watch for the IMAP connection string */ | ||
80 | result = recv (sd, buffer, MAX_INPUT_BUFFER - 1, 0); | ||
81 | |||
82 | /* strip carriange returns */ | ||
83 | strip (buffer); | ||
84 | |||
85 | /* return a WARNING status if we couldn't read any data */ | ||
86 | if (result == -1) { | ||
87 | printf ("recv() failed\n"); | ||
88 | result = STATE_WARNING; | ||
89 | } | ||
90 | |||
91 | else { | ||
92 | |||
93 | /* make sure we find the response we are looking for */ | ||
94 | if (!strstr (buffer, server_expect)) { | ||
95 | if (server_port == server_port) | ||
96 | printf ("Invalid IMAP response received from host\n"); | ||
97 | else | ||
98 | printf ("Invalid IMAP response received from host on port %d\n", | ||
99 | server_port); | ||
100 | result = STATE_WARNING; | ||
101 | } | ||
102 | |||
103 | else { | ||
104 | time (&end_time); | ||
105 | printf ("IMAP ok - %d second response time\n", | ||
106 | (int) (end_time - start_time)); | ||
107 | result = STATE_OK; | ||
108 | } | ||
109 | } | ||
110 | |||
111 | /* close the connection */ | ||
112 | send (sd, QUIT, strlen (QUIT), 0); | ||
113 | close (sd); | ||
114 | } | ||
115 | |||
116 | /* reset the alarm handler */ | ||
117 | alarm (0); | ||
118 | |||
119 | return result; | ||
120 | } | ||
121 | |||
122 | |||
123 | |||
124 | |||
125 | |||
126 | /* process command-line arguments */ | ||
127 | int | ||
128 | process_arguments (int argc, char **argv) | ||
129 | { | ||
130 | int c; | ||
131 | |||
132 | if (argc < 2) | ||
133 | return ERROR; | ||
134 | |||
135 | for (c = 1; c < argc; c++) { | ||
136 | if (strcmp ("-to", argv[c]) == 0) | ||
137 | strcpy (argv[c], "-t"); | ||
138 | else if (strcmp ("-wt", argv[c]) == 0) | ||
139 | strcpy (argv[c], "-w"); | ||
140 | else if (strcmp ("-ct", argv[c]) == 0) | ||
141 | strcpy (argv[c], "-c"); | ||
142 | } | ||
143 | |||
144 | |||
145 | |||
146 | c = 0; | ||
147 | while ((c += (call_getopt (argc - c, &argv[c]))) < argc) { | ||
148 | |||
149 | if (is_option (argv[c])) | ||
150 | continue; | ||
151 | |||
152 | if (server_address == NULL) { | ||
153 | if (is_host (argv[c])) { | ||
154 | server_address = argv[c]; | ||
155 | } | ||
156 | else { | ||
157 | usage ("Invalid host name"); | ||
158 | } | ||
159 | } | ||
160 | } | ||
161 | |||
162 | if (server_address == NULL) | ||
163 | server_address = strscpy (NULL, "127.0.0.1"); | ||
164 | |||
165 | if (server_expect == NULL) | ||
166 | server_expect = strscpy (NULL, EXPECT); | ||
167 | |||
168 | return validate_arguments (); | ||
169 | } | ||
170 | |||
171 | |||
172 | |||
173 | |||
174 | |||
175 | |||
176 | int | ||
177 | call_getopt (int argc, char **argv) | ||
178 | { | ||
179 | int c, i = 0; | ||
180 | |||
181 | #ifdef HAVE_GETOPT_H | ||
182 | int option_index = 0; | ||
183 | static struct option long_options[] = { | ||
184 | {"hostname", required_argument, 0, 'H'}, | ||
185 | {"expect", required_argument, 0, 'e'}, | ||
186 | {"critical", required_argument, 0, 'c'}, | ||
187 | {"warning", required_argument, 0, 'w'}, | ||
188 | {"port", required_argument, 0, 'P'}, | ||
189 | {"verbose", no_argument, 0, 'v'}, | ||
190 | {"version", no_argument, 0, 'V'}, | ||
191 | {"help", no_argument, 0, 'h'}, | ||
192 | {0, 0, 0, 0} | ||
193 | }; | ||
194 | #endif | ||
195 | |||
196 | while (1) { | ||
197 | #ifdef HAVE_GETOPT_H | ||
198 | c = | ||
199 | getopt_long (argc, argv, "+hVvt:p:e:c:w:H:", long_options, | ||
200 | &option_index); | ||
201 | #else | ||
202 | c = getopt (argc, argv, "+?hVvt:p:e:c:w:H:"); | ||
203 | #endif | ||
204 | |||
205 | i++; | ||
206 | |||
207 | if (c == -1 || c == EOF || c == 1) | ||
208 | break; | ||
209 | |||
210 | switch (c) { | ||
211 | case 't': | ||
212 | case 'p': | ||
213 | case 'e': | ||
214 | case 'c': | ||
215 | case 'w': | ||
216 | case 'H': | ||
217 | i++; | ||
218 | } | ||
219 | |||
220 | switch (c) { | ||
221 | case 'H': /* hostname */ | ||
222 | if (is_host (optarg)) { | ||
223 | server_address = optarg; | ||
224 | } | ||
225 | else { | ||
226 | usage ("Invalid host name\n"); | ||
227 | } | ||
228 | break; | ||
229 | case 'p': /* port */ | ||
230 | if (is_intpos (optarg)) { | ||
231 | server_port = atoi (optarg); | ||
232 | } | ||
233 | else { | ||
234 | usage ("Server port must be a positive integer\n"); | ||
235 | } | ||
236 | break; | ||
237 | case 'e': /* username */ | ||
238 | server_expect = optarg; | ||
239 | break; | ||
240 | case 'c': /* critical time threshold */ | ||
241 | if (is_intnonneg (optarg)) { | ||
242 | critical_time = atoi (optarg); | ||
243 | check_critical_time = TRUE; | ||
244 | } | ||
245 | else { | ||
246 | usage ("Critical time must be a nonnegative integer\n"); | ||
247 | } | ||
248 | break; | ||
249 | case 'w': /* warning time threshold */ | ||
250 | if (is_intnonneg (optarg)) { | ||
251 | warning_time = atoi (optarg); | ||
252 | check_warning_time = TRUE; | ||
253 | } | ||
254 | else { | ||
255 | usage ("Warning time must be a nonnegative integer\n"); | ||
256 | } | ||
257 | break; | ||
258 | case 'v': /* verbose */ | ||
259 | verbose = TRUE; | ||
260 | break; | ||
261 | case 't': /* timeout */ | ||
262 | if (is_intnonneg (optarg)) { | ||
263 | socket_timeout = atoi (optarg); | ||
264 | } | ||
265 | else { | ||
266 | usage ("Time interval must be a nonnegative integer\n"); | ||
267 | } | ||
268 | break; | ||
269 | case 'V': /* version */ | ||
270 | print_revision (PROGNAME, "$Revision$"); | ||
271 | exit (STATE_OK); | ||
272 | case 'h': /* help */ | ||
273 | print_help (); | ||
274 | exit (STATE_OK); | ||
275 | case '?': /* help */ | ||
276 | usage ("Invalid argument\n"); | ||
277 | } | ||
278 | } | ||
279 | return i; | ||
280 | } | ||
281 | |||
282 | |||
283 | |||
284 | |||
285 | |||
286 | int | ||
287 | validate_arguments (void) | ||
288 | { | ||
289 | return OK; | ||
290 | } | ||
291 | |||
292 | |||
293 | |||
294 | |||
295 | |||
296 | void | ||
297 | print_help (void) | ||
298 | { | ||
299 | print_revision (PROGNAME, "$Revision$"); | ||
300 | printf | ||
301 | ("Copyright (c) 2000 Tom Shields/Karl DeBisschop\n\n" | ||
302 | "This plugin tests the IMAP4 service on the specified host.\n\n"); | ||
303 | print_usage (); | ||
304 | printf | ||
305 | ("\nOptions:\n" | ||
306 | " -H, --hostname=STRING or IPADDRESS\n" | ||
307 | " Check server on the indicated host\n" | ||
308 | " -p, --port=INTEGER\n" | ||
309 | " Make connection on the indicated port (default: %d)\n" | ||
310 | " -e, --expect=STRING\n" | ||
311 | " String to expect in first line of server response (default: %s)\n" | ||
312 | " -w, --warning=INTEGER\n" | ||
313 | " Seconds necessary to result in a warning status\n" | ||
314 | " -c, --critical=INTEGER\n" | ||
315 | " Seconds necessary to result in a critical status\n" | ||
316 | " -t, --timeout=INTEGER\n" | ||
317 | " Seconds before connection attempt times out (default: %d)\n" | ||
318 | " -v, --verbose\n" | ||
319 | " Print extra information (command-line use only)\n" | ||
320 | " -h, --help\n" | ||
321 | " Print detailed help screen\n" | ||
322 | " -V, --version\n" | ||
323 | " Print version information\n\n", | ||
324 | PORT, EXPECT, DEFAULT_SOCKET_TIMEOUT); | ||
325 | support (); | ||
326 | } | ||
327 | |||
328 | |||
329 | |||
330 | |||
331 | |||
332 | void | ||
333 | print_usage (void) | ||
334 | { | ||
335 | printf | ||
336 | ("Usage: %s -H host [-e expect] [-p port] [-w warn] [-c crit]\n" | ||
337 | " [-t timeout] [-v]\n" | ||
338 | " %s --help\n" | ||
339 | " %s --version\n", PROGNAME, PROGNAME, PROGNAME); | ||
340 | } | ||