Initial revision

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

1 files changed, 353 insertions, 0 deletions

diff --git a/plugins/check_disk.c b/plugins/check_disk.c
new file mode 100644
index 00000000..d7aad072
--- /dev/null
+++ b/plugins/check_disk.c
@@ -0,0 +1,353 @@
+/******************************************************************************
+ *
+ * CHECK_DISK.C
+ *
+ * Program: Disk space plugin for Nagios
+ * License: GPL
+ * Copyright (c) 1999 Ethan Galstad (nagios@nagios.org)
+ * Copyright (c) 2000 Karl DeBisschop (kdebisschop@users.sourceforge.net)
+ *
+ * $Id$
+ *
+ * Description:
+ *
+ * This plugin will use the /bin/df command to check the free space on
+ * currently mounted filesystems.  If the percent used disk space is
+ * above <c_dfp>, a STATE_CRITICAL is returned.  If the percent used
+ * disk space is above <w_dfp>, a STATE_WARNING is returned.  If the
+ * speicified filesystem cannot be read, a STATE_CRITICAL is returned,
+ * other errors with reading the output result in a STATE_UNKNOWN
+ * error.
+ *
+ * Notes:
+ *  - IRIX support added by Charlie Cook 4-16-1999
+ *  - Modifications by Karl DeBisschop 1999-11-24
+ *     reformat code to 80 char screen width
+ *     set STATE_WARNING if stderr is written or spclose status set
+ *     set default result to STAT_UNKNOWN
+ *     initailize usp to -1, eliminate 'found' variable
+ *     accept any filename/filesystem
+ *     use sscanf, drop while loop
+ *
+ *****************************************************************************/
+
+#include "common.h"
+#include "popen.h"
+#include "utils.h"
+#include <stdarg.h>
+
+#define PROGNAME "check_disk"
+
+int process_arguments (int, char **);
+int call_getopt (int, char **);
+int validate_arguments (void);
+int check_disk (int usp, int free_disk);
+void print_help (void);
+void print_usage (void);
+
+int w_df = -1;
+int c_df = -1;
+float w_dfp = -1.0;
+float c_dfp = -1.0;
+char *path = NULL;
+int verbose = FALSE;
+
+int
+main (int argc, char **argv)
+{
+	int len;
+	int usp = -1;
+	int total_disk = -1;
+	int used_disk = -1;
+	int free_disk = -1;
+	int result = STATE_UNKNOWN;
+	char *command_line = NULL;
+	char input_buffer[MAX_INPUT_BUFFER] = "";
+	char file_system[MAX_INPUT_BUFFER] = "";
+	char outbuf[MAX_INPUT_BUFFER] = "";
+	char *output = NULL;
+
+	if (process_arguments (argc, argv) != OK)
+		usage ("Could not parse arguments\n");
+
+	command_line = ssprintf (command_line, "%s %s", DF_COMMAND, path);
+
+	if (verbose)
+		printf ("%s ==> ", command_line);
+
+	child_process = spopen (command_line);
+	if (child_process == NULL) {
+		printf ("Could not open pipe: %s\n", command_line);
+		return STATE_UNKNOWN;
+	}
+
+	child_stderr = fdopen (child_stderr_array[fileno (child_process)], "r");
+	if (child_stderr == NULL) {
+		printf ("Could not open stderr for %s\n", command_line);
+	}
+
+	while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_process)) {
+
+		if (!index (input_buffer, '/'))
+			continue;
+
+		if (sscanf
+				(input_buffer, "%s %d %d %d %d%%", file_system, &total_disk,
+				 &used_disk, &free_disk, &usp) == 5
+				|| sscanf (input_buffer, "%s %*s %d %d %d %d%%", file_system,
+									 &total_disk, &used_disk, &free_disk, &usp) == 5) {
+			result = max (result, check_disk (usp, free_disk));
+			len =
+				snprintf (outbuf, MAX_INPUT_BUFFER - 1,
+									" [%d kB (%d%%) free on %s]", free_disk, 100 - usp,
+									file_system);
+			outbuf[len] = 0;
+			output = strscat (output, outbuf);
+		}
+		else {
+			printf ("Unable to read output:\n%s\n%s\n", command_line, input_buffer);
+			return result;
+		}
+	}
+
+	/* If we get anything on stderr, at least set warning */
+	while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_stderr))
+		result = max (result, STATE_WARNING);
+
+	/* close stderr */
+	(void) fclose (child_stderr);
+
+	/* close the pipe */
+	if (spclose (child_process))
+		result = max (result, STATE_WARNING);
+
+	else if (usp < 0)
+		printf ("Disk %s not mounted or nonexistant\n", argv[3]);
+	else if (result == STATE_UNKNOWN)
+		printf ("Unable to read output\n%s\n%s\n", command_line, input_buffer);
+	else
+		printf ("DISK %s -%s\n", state_text (result), output);
+
+	return result;
+}
+
+/* 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]))) < argc) {
+
+		if (w_dfp == -1 && is_intnonneg (argv[c]))
+			w_dfp = (100.0 - atof (argv[c]));
+		else if (c_dfp == -1 && is_intnonneg (argv[c]))
+			c_dfp = (100.0 - atof (argv[c]));
+		else if (path == NULL || path[0] == 0)
+			path = strscpy (path, argv[c]);
+	}
+
+	if (path == NULL) {
+		path = malloc (1);
+		if (path == NULL)
+			terminate (STATE_UNKNOWN, "Could not malloc empty path\n");
+		path[0] = 0;
+	}
+
+	return validate_arguments ();
+}
+
+int
+call_getopt (int argc, char **argv)
+{
+	int c, i = 0;
+
+#ifdef HAVE_GETOPT_H
+	int option_index = 0;
+	static struct option long_options[] = {
+		{"warning", required_argument, 0, 'w'},
+		{"critical", required_argument, 0, 'c'},
+		{"timeout", required_argument, 0, 't'},
+		{"path", required_argument, 0, 'p'},
+		{"partition", required_argument, 0, 'p'},
+		{"verbose", no_argument, 0, 'v'},
+		{"version", no_argument, 0, 'V'},
+		{"help", no_argument, 0, 'h'},
+		{0, 0, 0, 0}
+	};
+#endif
+
+	while (1) {
+#ifdef HAVE_GETOPT_H
+		c =
+			getopt_long (argc, argv, "+?Vhvt:c:w:p:", long_options, &option_index);
+#else
+		c = getopt (argc, argv, "+?Vhvt:c:w:p:");
+#endif
+
+		i++;
+
+		if (c == -1 || c == EOF || c == 1)
+			break;
+
+		switch (c) {
+		case 't':
+		case 'c':
+		case 'w':
+		case 'p':
+			i++;
+		}
+
+		switch (c) {
+		case 'w':									/* warning time threshold */
+			if (is_intnonneg (optarg)) {
+				w_df = atoi (optarg);
+				break;
+			}
+			else if (strpbrk (optarg, ",:") &&
+							 strstr (optarg, "%") &&
+							 sscanf (optarg, "%d%*[:,]%f%%", &w_df, &w_dfp) == 2) {
+				break;
+			}
+			else if (strstr (optarg, "%") && sscanf (optarg, "%f%%", &w_dfp) == 1) {
+				break;
+			}
+			else {
+				usage ("Warning threshold must be integer or percentage!\n");
+			}
+		case 'c':									/* critical time threshold */
+			if (is_intnonneg (optarg)) {
+				c_df = atoi (optarg);
+				break;
+			}
+			else if (strpbrk (optarg, ",:") &&
+							 strstr (optarg, "%") &&
+							 sscanf (optarg, "%d%*[,:]%f%%", &c_df, &c_dfp) == 2) {
+				break;
+			}
+			else if (strstr (optarg, "%") && sscanf (optarg, "%f%%", &c_dfp) == 1) {
+				break;
+			}
+			else {
+				usage ("Critical threshold must be integer or percentage!\n");
+			}
+		case 't':									/* timeout period */
+			if (is_integer (optarg)) {
+				timeout_interval = atoi (optarg);
+				break;
+			}
+			else {
+				usage ("Timeout Interval must be an integer!\n");
+			}
+		case 'p':									/* path or partition */
+			path = optarg;
+			break;
+		case 'v':									/* verbose */
+			verbose = TRUE;
+			break;
+		case 'V':									/* version */
+			print_revision (my_basename (argv[0]), "$Revision$");
+			exit (STATE_OK);
+		case 'h':									/* help */
+			print_help ();
+			exit (STATE_OK);
+		case '?':									/* help */
+			usage ("check_disk: unrecognized option\n");
+			break;
+		}
+	}
+	return i;
+}
+
+int
+validate_arguments ()
+{
+	if (w_df < 0 && c_df < 0 && w_dfp < 0 && c_dfp < 0) {
+		printf ("INPUT ERROR: Unable to parse command line\n");
+		return ERROR;
+	}
+	else if ((w_dfp >= 0 || c_dfp >= 0)
+					 && (w_dfp < 0 || c_dfp < 0 || w_dfp > 100 || c_dfp > 100
+							 || c_dfp > w_dfp)) {
+		printf
+			("INPUT ERROR: C_DFP (%f) should be less than W_DFP (%f) and both should be between zero and 100 percent, inclusive\n",
+			 c_dfp, w_dfp);
+		return ERROR;
+	}
+	else if ((w_df > 0 || c_df > 0) && (w_df < 0 || c_df < 0 || c_df > w_df)) {
+		printf
+			("INPUT ERROR: C_DF (%d) should be less than W_DF (%d) and both should be greater than zero\n",
+			 c_df, w_df);
+		return ERROR;
+	}
+	else {
+		return OK;
+	}
+}
+
+int
+check_disk (usp, free_disk)
+{
+	int result = STATE_UNKNOWN;
+	/* check the percent used space against thresholds */
+	if (usp >= 0 && usp >= (100.0 - c_dfp))
+		result = STATE_CRITICAL;
+	else if (c_df >= 0 && free_disk <= c_df)
+		result = STATE_CRITICAL;
+	else if (usp >= 0 && usp >= (100.0 - w_dfp))
+		result = STATE_WARNING;
+	else if (w_df >= 0 && free_disk <= w_df)
+		result = STATE_WARNING;
+	else if (usp >= 0.0)
+		result = STATE_OK;
+	return result;
+}
+
+void
+print_help (void)
+{
+	print_revision (PROGNAME, "$Revision$");
+	printf
+		("Copyright (c) 2000 Ethan Galstad/Karl DeBisschop\n\n"
+		 "This plugin will check the percent of used disk space on a mounted\n"
+		 "file system and generate an alert if percentage is above one of the\n"
+		 "threshold values.\n\n");
+	print_usage ();
+	printf
+		("\nOptions:\n"
+		 " -w, --warning=INTEGER\n"
+		 "   Exit with WARNING status if less than INTEGER kilobytes of disk are free\n"
+		 " -w, --warning=PERCENT%%\n"
+		 "   Exit with WARNING status if more than PERCENT of disk space is free\n"
+		 " -c, --critical=INTEGER\n"
+		 "   Exit with CRITICAL status if less than INTEGER kilobytes of disk are free\n"
+		 " -c, --critical=PERCENT%%\n"
+		 "   Exit with CRITCAL status if more than PERCENT of disk space is free\n"
+		 " -p, --path=PATH, --partition=PARTTION\n"
+		 "    Path or partition (checks all mounted partitions if unspecified)\n"
+		 " -v, --verbose\n"
+		 "    Show details for command-line debugging (do not use with nagios server)\n"
+		 " -h, --help\n"
+		 "    Print detailed help screen\n"
+		 " -V, --version\n" "    Print version information\n\n");
+	support ();
+}
+
+void
+print_usage (void)
+{
+	printf
+		("Usage: %s -w limit -c limit [-p path] [-t timeout] [--verbose]\n"
+		 "       %s (-h|--help)\n"
+		 "       %s (-V|--version)\n", PROGNAME, PROGNAME, PROGNAME);
+}