X-Git-Url: https://git.tokkee.org/?a=blobdiff_plain;f=plugins%2Fcheck_procs.c;h=ccf0891d13322c61a6710c7621a6a46673851eba;hb=97ecfa6226bea7c09f78b2b92d807e73467ffc08;hp=3849c77666c87db12c3c2e9ac2dca41bc7ab184a;hpb=0c3386274ef5002dffc20337ef02407f24d7400c;p=nagiosplug.git

diff --git a/plugins/check_procs.c b/plugins/check_procs.c
index 3849c77..ccf0891 100644
--- a/plugins/check_procs.c
+++ b/plugins/check_procs.c
@@ -34,16 +34,69 @@
 *
 ******************************************************************************/
 
-const char *progname = "check_snmp";
+const char *progname = "check_procs";
 #define REVISION "$Revision$"
 #define COPYRIGHT "1999-2002"
 #define AUTHOR "Ethan Galstad"
 #define EMAIL "nagios@nagios.org"
-#define SUMMARY "Check the number of currently running processes and generates WARNING or\n\
-CRITICAL states if the process count is outside the specified threshold\n\
-ranges. The process count can be filtered by process owner, parent process\n\
-PID, current state (e.g., 'Z'), or may be the total number of running\n\
-processes\n"
+#define SUMMARY "\
+Checks all processes and generates WARNING or CRITICAL states if the specified\n\
+metric is outside the required threshold ranges. The metric defaults to number\n\
+of processes.  Search filters can be applied to limit the processes to check.\n"
+
+#define OPTIONS "\
+-w <range> -c <range> [-m metric]\n\
+            [-s state] [-p ppid] [-u user] [-r rss] [-z vsz] [-P %cpu]\n\
+            [-a argument-array] [-C command] [-v]"
+
+#define LONGOPTIONS "\
+Required Arguments:\n\
+ -w, --warning=RANGE\n\
+   Generate warning state if metric is outside this range\n\
+ -c, --critical=RANGE\n\
+   Generate critical state if metric is outside this range\n\
+Optional Arguments:\n\
+ -m, --metric=TYPE\n\
+   Check thresholds against metric. Valid types:\n\
+   PROCS - number of processes (default)\n\
+   VSZ  - virtual memory size\n\
+   RSS  - resident set memory size\n\
+   CPU  - percentage cpu\n\
+ -v, --verbose\n\
+   Extra information. Up to 3 verbosity levels\n\
+Optional Filters:\n\
+ -s, --state=STATUSFLAGS\n\
+   Only scan for processes that have, in the output of `ps`, one or\n\
+   more of the status flags you specify (for example R, Z, S, RS,\n\
+   RSZDT, plus others based on the output of your 'ps' command).\n\
+ -p, --ppid=PPID\n\
+   Only scan for children of the parent process ID indicated.\n\
+ -z, --vsz=VSZ\n\
+   Only scan for processes with vsz higher than indicated.\n\
+ -r, --rss=RSS\n\
+   Only scan for processes with rss higher than indicated.\n\
+ -P, --pcpu=PCPU\n\
+   Only scan for processes with pcpu higher than indicated.\n\
+ -u, --user=USER\n\
+   Only scan for processes with user name or ID indicated.\n\
+ -a, --argument-array=STRING\n\
+   Only scan for processes with args that contain STRING.\n\
+ -C, --command=COMMAND\n\
+   Only scan for exact matches to the named COMMAND.\n\
+\n\
+RANGEs are specified 'min:max' or 'min:' or ':max' (or 'max'). If\n\
+specified 'max:min', a warning status will be generated if the\n\
+count is inside the specified range\n"
+
+#define EXAMPLES "\
+ check_procs -w 2:2 -c 2:1024 -C portsentry\n\
+   Warning if not two processes with command name portsentry. Critical\n\
+   if < 2 or > 1024 processes\n\
+ check_procs -w 10 -a '/usr/local/bin/perl' -u root\n\
+   Warning alert if > 10 processes with command arguments containing \n\
+   '/usr/local/bin/perl' and owned by root\n\
+ check_procs -w 50000 -c 100000 --metric=VSZ\n\
+   Alert if vsz of any processes over 50K or 100K\n"
 
 #include "config.h"
 #include <pwd.h>
@@ -55,6 +108,7 @@ int process_arguments (int, char **);
 int validate_arguments (void);
 void print_usage (void);
 void print_help (void);
+int check_thresholds (int);
 
 int wmax = -1;
 int cmax = -1;
@@ -68,14 +122,29 @@ int options = 0; /* bitmask of filter criteria to test against */
 #define USER 8
 #define PROG 16
 #define ARGS 32
-
-int verbose = FALSE;
+#define VSZ  64
+#define RSS  128
+#define PCPU 256
+
+/* Different metrics */
+int metric = 0;
+#define METRIC_PROCS 0
+#define METRIC_VSZ 1
+#define METRIC_RSS 2
+#define METRIC_CPU 3
+char *metric_name = "";
+
+int verbose = 0;
 int uid;
 int ppid;
-char *statopts = NULL;
-char *prog = NULL;
-char *args = NULL;
-char *fmt = NULL;
+int vsz;
+int rss;
+float pcpu;
+char *statopts = "";
+char *prog = "";
+char *args = "";
+char *fmt = "";
+char *fails = "";
 char tmp[MAX_INPUT_BUFFER];
 
 int
@@ -85,23 +154,35 @@ main (int argc, char **argv)
 
 	int procuid = 0;
 	int procppid = 0;
+	int procvsz = 0;
+	int procrss = 0;
+	float procpcpu = 0;
 	char procstat[8];
 	char procprog[MAX_INPUT_BUFFER];
 	char *procargs;
 
+	const char *zombie = "Z";
+
 	int resultsum = 0; /* bitmask of the filter criteria met by a process */
 	int found = 0; /* counter for number of lines returned in `ps` output */
 	int procs = 0; /* counter for number of processes meeting filter criteria */
 	int pos; /* number of spaces before 'args' in `ps` output */
+	int cols; /* number of columns in ps output */
+	int warn = 0; /* number of processes in warn state */
+	int crit = 0; /* number of processes in crit state */
+	int i;
 
 	int result = STATE_UNKNOWN;
 
+	asprintf (&metric_name, "PROCS");
+	metric = METRIC_PROCS;
+
 	if (process_arguments (argc, argv) == ERROR)
 		usage ("Unable to parse command line\n");
 
-	/* run the command */
-	if (verbose)
-		printf ("%s\n", PS_COMMAND);
+	if (verbose >= 2)
+		printf ("CMD: %s\n", PS_COMMAND);
+
 	child_process = spopen (PS_COMMAND);
 	if (child_process == NULL) {
 		printf ("Could not open pipe: %s\n", PS_COMMAND);
@@ -115,21 +196,27 @@ main (int argc, char **argv)
 	fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_process);
 
 	while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_process)) {
-		if (
-#ifdef USE_PS_VARS
-				 sscanf (input_buffer, PS_FORMAT, PS_VARLIST) >= 4
-#else
-				 sscanf (input_buffer, PS_FORMAT, procstat, &procuid, &procppid, &pos,
-								 procprog) >= 4
-#endif
-			) {
-			found++;
+		strcpy(procprog,"");
+		asprintf(&procargs,"");
+
+		cols = sscanf (input_buffer, PS_FORMAT, PS_VARLIST);
+
+		/* Zombie processes do not give a procprog command */
+		if ( cols == 6 && strstr(procstat, zombie) ) {
+			cols = 7;
+			/* Set some value for procargs for the strip command further below 
+			Seen to be a problem on some Solaris 7 and 8 systems */
+			input_buffer[pos] = '\n';
+			input_buffer[pos+1] = 0x0;
+		}
+		if ( cols >= 7 ) {
 			resultsum = 0;
 			asprintf (&procargs, "%s", input_buffer + pos);
- 			strip (procargs);
+			strip (procargs);
+
 			if ((options & STAT) && (strstr (statopts, procstat)))
 				resultsum |= STAT;
-			if ((options & ARGS) && procargs && (strstr (procargs, args) == procargs))
+			if ((options & ARGS) && procargs && (strstr (procargs, args) != NULL))
 				resultsum |= ARGS;
 			if ((options & PROG) && procprog && (strcmp (prog, procprog) == 0))
 				resultsum |= PROG;
@@ -137,16 +224,53 @@ main (int argc, char **argv)
 				resultsum |= PPID;
 			if ((options & USER) && (procuid == uid))
 				resultsum |= USER;
-#ifdef DEBUG1
-			if (procargs == NULL)
-				printf ("%d %d %d %s %s\n", procs, procuid, procppid, procstat,
-								procprog);
-			else
-				printf ("%d %d %d %s %s %s\n", procs, procuid, procppid, procstat,
-								procprog, procargs);
-#endif
-			if (options == resultsum)
-				procs++;
+			if ((options & VSZ)  && (procvsz >= vsz))
+				resultsum |= VSZ;
+			if ((options & RSS)  && (procrss >= rss))
+				resultsum |= RSS;
+			if ((options & PCPU)  && (procpcpu >= pcpu))
+				resultsum |= PCPU;
+
+			if (verbose >= 3)
+				printf ("%d %d %d %d %d %.2f %s %s %s\n", 
+					procs, procuid, procvsz, procrss,
+					procppid, procpcpu, procstat, procprog, procargs);
+
+			/* Ignore self */
+			if (strcmp (procprog, progname) == 0)
+				continue;
+
+			found++;
+
+			/* Next line if filters not matched */
+			if (!(options == resultsum || options == ALL))
+				continue;
+
+			procs++;
+
+			if (metric == METRIC_VSZ)
+				i = check_thresholds (procvsz);
+			else if (metric == METRIC_RSS)
+				i = check_thresholds (procrss);
+			/* TODO? float thresholds for --metric=CPU */
+			else if (metric == METRIC_CPU)
+				i = check_thresholds ((int)procpcpu); 
+
+			if (metric != METRIC_PROCS) {
+				if (i == STATE_WARNING) {
+					warn++;
+					asprintf (&fails, "%s%s%s", fails, (fails == "" ? "" : ", "), procprog);
+				}
+				if (i == STATE_CRITICAL) {
+					crit++;
+					asprintf (&fails, "%s%s%s", fails, (fails == "" ? "" : ", "), procprog);
+				}
+				result = max_state (result, i);
+			}
+		} 
+		/* This should not happen */
+		else if (verbose) {
+			printf("Not parseable: %s", input_buffer);
 		}
 	}
 
@@ -154,101 +278,95 @@ main (int argc, char **argv)
 	while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_stderr)) {
 		if (verbose)
 			printf ("STDERR: %s", input_buffer);
-		/*Cannot use max() any more as STATE_UNKNOWN is gt STATE_CRITICAL 
-		result = max (result, STATE_WARNING); */
-		if ( !(result == STATE_CRITICAL) ) {
-			result = STATE_WARNING;
-		}
-			printf ("System call sent warnings to stderr\n");
+		result = max_state (result, STATE_WARNING);
+		printf ("System call sent warnings to stderr\n");
 	}
 	
-/*	if (result == STATE_UNKNOWN || result == STATE_WARNING)
-		printf ("System call sent warnings to stderr\n");
-*/
 	(void) fclose (child_stderr);
 
 	/* close the pipe */
 	if (spclose (child_process)) {
 		printf ("System call returned nonzero status\n");
-		if ( !(result == STATE_CRITICAL) ) {
-			return STATE_WARNING;
-		}
-		else {
-			return result ;
-		}
+		result = max_state (result, STATE_WARNING);
 	}
 
-	if (options == ALL)
-		procs = found;
-
 	if (found == 0) {							/* no process lines parsed so return STATE_UNKNOWN */
 		printf ("Unable to read output\n");
-
 		return result;
 	}
 
-	if (verbose && (options & STAT))
-		printf ("%s ", statopts);
-	if (verbose && (options & PROG))
-		printf ("%s ", prog);
-	if (verbose && (options & PPID))
-		printf ("%d ", ppid);
-	if (verbose && (options & USER))
-		printf ("%d ", uid);
+	if ( result == STATE_UNKNOWN ) 
+		result = STATE_OK;
+
+	/* Needed if procs found, but none match filter */
+	if ( metric == METRIC_PROCS ) {
+		result = max_state (result, check_thresholds (procs) );
+	}
+
+	if ( result == STATE_OK ) {
+		printf ("%s OK: %d process%s", 
+			metric_name, procs, ( procs != 1 ? "es" : "") );
+	} else if (result == STATE_WARNING) {
+		if ( metric == METRIC_PROCS ) {
+			printf ("PROCS WARNING: %d process%s", procs, 
+				( procs != 1 ? "es" : ""));
+		} else {
+			printf ("%s WARNING: %d warn out of %d process%s", 
+				metric_name, warn, procs, 
+				( procs != 1 ? "es" : ""));
+		}
+	} else if (result == STATE_CRITICAL) {
+		if (metric == METRIC_PROCS) {
+			printf ("PROCS CRITICAL: %d process%s", procs, 
+				( procs != 1 ? "es" : ""));
+		} else {
+			printf ("%s CRITICAL: %d crit, %d warn out of %d process%s", 
+				metric_name, crit, warn, procs, 
+				( procs != 1 ? "es" : ""));
+		}
+	} 
+	
+	if (fmt != "") {
+		printf (" with %s", fmt);
+	}
+
+	if ( verbose >= 1 && fails != "" )
+		printf (" [%s]", fails);
+
+	printf ("\n");
+	return result;
+}
 
+/* Check thresholds against value */
+int
+check_thresholds (int value)
+{
  	if (wmax == -1 && cmax == -1 && wmin == -1 && cmin == -1) {
-		if (result == STATE_UNKNOWN)
-			result = STATE_OK;
-		printf (fmt, "OK", procs);
-		return result;
+		return OK;
  	}
 	else if (cmax >= 0 && cmin >= 0 && cmax < cmin) {
-		if (procs > cmax && procs < cmin) {
-			printf (fmt, "CRITICAL", procs);
+		if (value > cmax && value < cmin)
 			return STATE_CRITICAL;
-		}
 	}
-	else if (cmax >= 0 && procs > cmax) {
-		printf (fmt, "CRITICAL", procs);
+	else if (cmax >= 0 && value > cmax) {
 		return STATE_CRITICAL;
 	}
-	else if (cmin >= 0 && procs < cmin) {
-		printf (fmt, "CRITICAL", procs);
+	else if (cmin >= 0 && value < cmin) {
 		return STATE_CRITICAL;
 	}
 
 	if (wmax >= 0 && wmin >= 0 && wmax < wmin) {
-		if (procs > wmax && procs < wmin) {
-			printf (fmt, "CRITICAL", procs);
-			return STATE_CRITICAL;
-		}
-	}
-	else if (wmax >= 0 && procs > wmax) {
-		printf (fmt, "WARNING", procs);
-		if ( !(result == STATE_CRITICAL) ) {
+		if (value > wmax && value < wmin) {
 			return STATE_WARNING;
 		}
-		else {
-			return result ;
-		}
-		/*return max (result, STATE_WARNING); */
 	}
-	else if (wmin >= 0 && procs < wmin) {
-		printf (fmt, "WARNING", procs);
-		if ( !(result == STATE_CRITICAL) ) {
-			return STATE_WARNING;
-		}
-		else {
-			return result ;
-		}
-		/*return max (result, STATE_WARNING); */
+	else if (wmax >= 0 && value > wmax) {
+		return STATE_WARNING;
 	}
-
-	printf (fmt, "OK", procs);
-	if ( result == STATE_UNKNOWN ) {
-		result = STATE_OK;
+	else if (wmin >= 0 && value < wmin) {
+		return STATE_WARNING;
 	}
-	return result;
+	return STATE_OK;
 }
 
 /* process command-line arguments */
@@ -258,35 +376,33 @@ process_arguments (int argc, char **argv)
 	int c = 1;
 	char *user;
 	struct passwd *pw;
-#ifdef HAVE_GETOPT_H
 	int option_index = 0;
 	static struct option long_options[] = {
 		{"warning", required_argument, 0, 'w'},
 		{"critical", required_argument, 0, 'c'},
+		{"metric", required_argument, 0, 'm'},
 		{"timeout", required_argument, 0, 't'},
 		{"status", required_argument, 0, 's'},
 		{"ppid", required_argument, 0, 'p'},
 		{"command", required_argument, 0, 'C'},
+		{"vsz", required_argument, 0, 'z'},
+		{"rss", required_argument, 0, 'r'},
+		{"pcpu", required_argument, 0, 'P'},
 		{"argument-array", required_argument, 0, 'a'},
 		{"help", no_argument, 0, 'h'},
 		{"version", no_argument, 0, 'V'},
 		{"verbose", no_argument, 0, 'v'},
 		{0, 0, 0, 0}
 	};
-#endif
-
-	asprintf (&fmt, "");
 
 	for (c = 1; c < argc; c++)
 		if (strcmp ("-to", argv[c]) == 0)
 			strcpy (argv[c], "-t");
 
 	while (1) {
-#ifdef HAVE_GETOPT_H
-		c =	getopt_long (argc, argv, "Vvht:c:w:p:s:u:C:a:", long_options, &option_index);
-#else
-		c = getopt (argc, argv, "Vvht:c:w:p:s:u:C:a:");
-#endif
+		c = getopt_long (argc, argv, "Vvht:c:w:p:s:u:C:a:z:r:m:P:", 
+			long_options, &option_index);
+
 		if (c == -1 || c == EOF)
 			break;
 
@@ -351,7 +467,7 @@ process_arguments (int argc, char **argv)
 			}
 		case 'p':									/* process id */
 			if (sscanf (optarg, "%d%[^0-9]", &ppid, tmp) == 1) {
-				asprintf (&fmt, "%s%sPPID = %d", (options ? ", " : ""), ppid);
+				asprintf (&fmt, "%s%sPPID = %d", fmt, (options ? ", " : ""), ppid);
 				options |= PPID;
 				break;
 			}
@@ -387,23 +503,76 @@ process_arguments (int argc, char **argv)
 				uid = pw->pw_uid;
 			}
 			user = pw->pw_name;
-			asprintf (&fmt, "%s%sUID = %d (%s)", (options ? ", " : ""), fmt,
+			asprintf (&fmt, "%s%sUID = %d (%s)", fmt, (options ? ", " : ""),
 			          uid, user);
 			options |= USER;
 			break;
 		case 'C':									/* command */
 			asprintf (&prog, "%s", optarg);
-			asprintf (&fmt, "%s%scommand name %s", fmt, (options ? ", " : ""),
+			asprintf (&fmt, "%s%scommand name '%s'", fmt, (options ? ", " : ""),
 			          prog);
 			options |= PROG;
 			break;
 		case 'a':									/* args (full path name with args) */
 			asprintf (&args, "%s", optarg);
-			asprintf (&fmt, "%s%sargs %s", fmt, (options ? ", " : ""), args);
+			asprintf (&fmt, "%s%sargs '%s'", fmt, (options ? ", " : ""), args);
 			options |= ARGS;
 			break;
+		case 'r': 					/* RSS */
+			if (sscanf (optarg, "%d%[^0-9]", &rss, tmp) == 1) {
+				asprintf (&fmt, "%s%sRSS >= %d", fmt, (options ? ", " : ""), rss);
+				options |= RSS;
+				break;
+			}
+			printf ("%s: RSS must be an integer!\n\n",
+			        progname);
+			print_usage ();
+			exit (STATE_UNKNOWN);
+		case 'z':					/* VSZ */
+			if (sscanf (optarg, "%d%[^0-9]", &vsz, tmp) == 1) {
+				asprintf (&fmt, "%s%sVSZ >= %d", fmt, (options ? ", " : ""), vsz);
+				options |= VSZ;
+				break;
+			}
+			printf ("%s: VSZ must be an integer!\n\n",
+			        progname);
+			print_usage ();
+			exit (STATE_UNKNOWN);
+		case 'P':					/* PCPU */
+			/* TODO: -P 1.5.5 is accepted */
+			if (sscanf (optarg, "%f%[^0-9.]", &pcpu, tmp) == 1) {
+				asprintf (&fmt, "%s%sPCPU >= %.2f", fmt, (options ? ", " : ""), pcpu);
+				options |= PCPU;
+				break;
+			}
+			printf ("%s: PCPU must be a float!\n\n",
+			        progname);
+			print_usage ();
+			exit (STATE_UNKNOWN);
+		case 'm':
+			asprintf (&metric_name, "%s", optarg);
+			if ( strcmp(optarg, "PROCS") == 0) {
+				metric = METRIC_PROCS;
+				break;
+			} 
+			else if ( strcmp(optarg, "VSZ") == 0) {
+				metric = METRIC_VSZ;
+				break;
+			} 
+			else if ( strcmp(optarg, "RSS") == 0 ) {
+				metric = METRIC_RSS;
+				break;
+			}
+			else if ( strcmp(optarg, "CPU") == 0 ) {
+				metric = METRIC_CPU;
+				break;
+			}
+			printf ("%s: metric must be one of PROCS, VSZ, RSS, CPU!\n\n",
+				progname);
+			print_usage ();
+			exit (STATE_UNKNOWN);
 		case 'v':									/* command */
-			verbose = TRUE;
+			verbose++;
 			break;
 		}
 	}
@@ -427,7 +596,7 @@ int
 validate_arguments ()
 {
 
-if (wmax >= 0 && wmin == -1)
+	if (wmax >= 0 && wmin == -1)
 		wmin = 0;
 	if (cmax >= 0 && cmin == -1)
 		cmin = 0;
@@ -447,13 +616,8 @@ if (wmax >= 0 && wmin == -1)
 /* 		return ERROR; */
 /* 	} */
 
-	if (options == 0) {
-		options = 1;
-		asprintf (&fmt, "%%s - %%d processes running\n");
-	}
-	else {
-		asprintf (&fmt, "%%s - %%d processes running with %s\n", fmt);
-	}
+	if (options == 0)
+		options = ALL;
 
 	return options;
 }
@@ -467,37 +631,14 @@ print_help (void)
 		("Copyright (c) %s %s <%s>\n\n%s\n",
 		 COPYRIGHT, AUTHOR, EMAIL, SUMMARY);
 	print_usage ();
-	printf
-		("\nRequired Arguments:\n"
-		 " -w, --warning=RANGE\n"
-		 "    generate warning state if process count is outside this range\n"
-		 " -c, --critical=RANGE\n"
-		 "    generate critical state if process count is outside this range\n\n"
-		 "Optional Filters:\n"
-		 " -s, --state=STATUSFLAGS\n"
-		 "    Only scan for processes that have, in the output of `ps`, one or\n"
-		 "    more of the status flags you specify (for example R, Z, S, RS,\n"
-		 "    RSZDT, plus others based on the output of your 'ps' command).\n"
-		 " -p, --ppid=PPID\n"
-		 "    Only scan for children of the parent process ID indicated.\n"
-		 " -u, --user=USER\n"
-		 "    Only scan for proceses with user name or ID indicated.\n"
-		 " -a, --argument-array=STRING\n"
-		 "    Only scan for ARGS that match up to the length of the given STRING\n"
-		 " -C, --command=COMMAND\n"
-		 "    Only scan for exact matches to the named COMMAND.\n\n"
-		 "RANGEs are specified 'min:max' or 'min:' or ':max' (or 'max'). If\n"
-		 "specified 'max:min', a warning status will be generated if the\n"
-
-		 "count is inside the specified range\n");}
-
+	printf ("\nOptions:\n" LONGOPTIONS "\nExamples:\n" EXAMPLES "\n");
+}
 
 void
 print_usage (void)
 {
-	printf
-		("Usage:\n"
-		 " check_procs -w <range> -c <range> [-s state] [-p ppid] [-u user]\n"
-		 "             [-a argument-array] [-C command]\n"
-		 " check_procs --version\n" " check_procs --help\n");
+	printf ("Usage:\n" " %s %s\n"
+		 " %s (-h | --help) for detailed help\n"
+		 " %s (-V | --version) for version information\n",
+	progname, OPTIONS, progname, progname);	
 }