From 1a6c23309c5e57af28f6465610d98a6fc834c429 Mon Sep 17 00:00:00 2001 From: Ton Voon Date: Mon, 7 Apr 2003 22:53:49 +0000 Subject: Souped up check_procs with different metrics git-svn-id: https://nagiosplug.svn.sourceforge.net/svnroot/nagiosplug/nagiosplug/trunk@474 f882894a-f735-0410-b71e-b25c423dba1c --- plugins/Makefile.am | 4 +- plugins/check_nagios.c | 5 +- plugins/check_procs.c | 362 +++++++++++++++++++++++++++++++++++-------------- plugins/check_vsz.c | 250 ---------------------------------- 4 files changed, 265 insertions(+), 356 deletions(-) delete mode 100644 plugins/check_vsz.c (limited to 'plugins') diff --git a/plugins/Makefile.am b/plugins/Makefile.am index a884b451..0692cd28 100644 --- a/plugins/Makefile.am +++ b/plugins/Makefile.am @@ -7,7 +7,7 @@ INCLUDES = -I.. -I$(top_srcdir)/lib @LDAPINCLUDE@ @PGINCLUDE@ @SSLINCLUDE@ libexec_PROGRAMS = check_disk check_dummy check_http check_load \ check_mrtg check_mrtgtraf check_nwstat check_overcr check_ping \ check_procs check_real check_smtp check_ssh check_tcp check_time \ - check_udp check_ups check_users check_vsz negate urlize \ + check_udp check_ups check_users negate urlize \ @EXTRAS@ EXTRA_PROGRAMS = check_mysql check_radius check_pgsql check_snmp check_hpjd \ @@ -68,7 +68,6 @@ check_time_LDADD = $(NETLIBS) check_udp_LDADD = $(NETLIBS) check_ups_LDADD = $(NETLIBS) check_users_LDADD = $(BASEOBJS) popen.o -check_vsz_LDADD = $(BASEOBJS) popen.o check_by_ssh_LDADD = $(NETLIBS) popen.o negate_LDADD = $(BASEOBJS) popen.o urlize_LDADD = $(BASEOBJS) popen.o @@ -104,7 +103,6 @@ check_time_DEPENDENCIES = check_time.c $(NETOBJS) $(DEPLIBS) check_udp_DEPENDENCIES = check_udp.c $(NETOBJS) $(DEPLIBS) check_ups_DEPENDENCIES = check_ups.c $(NETOBJS) $(DEPLIBS) check_users_DEPENDENCIES = check_users.c $(BASEOBJS) popen.o $(DEPLIBS) -check_vsz_DEPENDENCIES = check_vsz.c $(BASEOBJS) popen.o $(DEPLIBS) check_by_ssh_DEPENDENCIES = check_by_ssh.c $(NETOBJS) popen.o $(DEPLIBS) negate_DEPENDENCIES = negate.c $(BASEOBJS) popen.o $(DEPLIBS) urlize_DEPENDENCIES = urlize.c $(BASEOBJS) popen.o $(DEPLIBS) diff --git a/plugins/check_nagios.c b/plugins/check_nagios.c index 083b5028..64a828b9 100644 --- a/plugins/check_nagios.c +++ b/plugins/check_nagios.c @@ -55,6 +55,9 @@ main (int argc, char **argv) FILE *fp; 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; @@ -106,7 +109,7 @@ main (int argc, char **argv) /* count the number of matching Nagios processes... */ while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_process)) { cols = sscanf (input_buffer, PS_FORMAT, PS_VARLIST); - if ( cols >= 4 ) { + if ( cols >= 6 ) { asprintf (&procargs, "%s", input_buffer + pos); strip (procargs); diff --git a/plugins/check_procs.c b/plugins/check_procs.c index 29f86bed..2cf550cc 100644 --- a/plugins/check_procs.c +++ b/plugins/check_procs.c @@ -39,11 +39,64 @@ const char *progname = "check_procs"; #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 -c [-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 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" + +#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 @@ -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,16 +122,30 @@ 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; +int vsz; +int rss; +float pcpu; char *statopts = ""; char *prog = ""; char *args = ""; char *fmt = ""; +char *fails = ""; char tmp[MAX_INPUT_BUFFER]; -const char *zombie = "Z"; int main (int argc, char **argv) @@ -86,24 +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); @@ -117,18 +196,22 @@ main (int argc, char **argv) fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_process); while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_process)) { + strcpy(procprog,""); + asprintf(&procargs,""); + cols = sscanf (input_buffer, PS_FORMAT, PS_VARLIST); /* Zombie processes do not give a procprog command */ + /* - would they give other columns? */ if ( cols == 3 && strstr(procstat, zombie) ) { - strcpy(procprog, ""); cols = 4; } - if ( cols >= 4 ) { + if ( cols >= 7 ) { found++; resultsum = 0; asprintf (&procargs, "%s", input_buffer + pos); strip (procargs); + if ((options & STAT) && (strstr (statopts, procstat))) resultsum |= STAT; if ((options & ARGS) && procargs && (strstr (procargs, args) == procargs)) @@ -139,16 +222,43 @@ 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); + + /* 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) { @@ -172,65 +282,83 @@ main (int argc, char **argv) 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; + if (value > wmax && value < wmin) { + return STATE_WARNING; } } - else if (wmax >= 0 && procs > wmax) { - printf (fmt, "WARNING", procs); - return max_state (result, STATE_WARNING); + else if (wmax >= 0 && value > wmax) { + return STATE_WARNING; } - else if (wmin >= 0 && procs < wmin) { - printf (fmt, "WARNING", procs); - return max_state (result, 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 */ @@ -244,10 +372,14 @@ process_arguments (int argc, char **argv) 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'}, @@ -260,7 +392,8 @@ process_arguments (int argc, char **argv) strcpy (argv[c], "-t"); while (1) { - c = getopt_long (argc, argv, "Vvht:c:w:p:s:u:C:a:", long_options, &option_index); + 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; @@ -368,17 +501,70 @@ process_arguments (int argc, char **argv) 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; } } @@ -402,7 +588,7 @@ int validate_arguments () { -if (wmax >= 0 && wmin == -1) + if (wmax >= 0 && wmin == -1) wmin = 0; if (cmax >= 0 && cmin == -1) cmin = 0; @@ -422,13 +608,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; } @@ -442,37 +623,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 -c [-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); } diff --git a/plugins/check_vsz.c b/plugins/check_vsz.c deleted file mode 100644 index 7a9acb2d..00000000 --- a/plugins/check_vsz.c +++ /dev/null @@ -1,250 +0,0 @@ -/****************************************************************************** - * - * CHECK_VSZ.C - * - * Program: Process plugin for Nagios - * License: GPL - * Copyright (c) 1999,2000 Karl DeBisschop - * - * Last Modified: $Date$ - * - * Description: - * - * This plugin will check for processes whose total image size exceeds - * the warning or critical thresholds given on the command line. With - * no command_name, everything that shows up on ps is evaluated. - * Otherwise, only jobs with the command_name given are examined. - * This program is particularly useful if you have to run a piece of - * commercial software that has a memory leak. With it you can shut - * down and restart the processes whenever the program threatens to - * take over your system. - * - * Modifications: - * - * 11-18-1999 Karl DeBisschop (kdebiss@alum.mit.edu) - * change to getopt, use print_help - * 08-18-1999 Ethan Galstad (nagios@nagios.org) - * Changed code to use common include file - * Changed fclose() to pclose() - * 09-09-1999 Ethan Galstad (nagios@nagios.org) - * Changed popen()/pclose() to spopen()/spclose() - * 11-18-1999 Karl DeBisschop (kdebiss@alum.mit.edu) - * set STATE_WARNING of stderr written or nonzero status returned - * - *****************************************************************************/ - -const char *progname = "check_vsz"; -#define REVISION "$Revision$" -#define COPYRIGHT "1999-2002" -#define AUTHOR "Karl DeBisschop" -#define EMAIL "karl@debisschop.net" -#define SUMMARY "Check the image size of a running program.\n" - -#include "common.h" -#include "popen.h" -#include "utils.h" - -int process_arguments (int argc, char **argv); -void print_help (const char *cmd); -void print_usage (const char *cmd); - -int warn = -1; -int crit = -1; -char *proc = NULL; - -int -main (int argc, char **argv) -{ - int len; - int result = STATE_OK; - int line = 0; - int proc_size = -1; - char input_buffer[MAX_INPUT_BUFFER]; - char proc_name[MAX_INPUT_BUFFER]; - char *message = ""; - - if (process_arguments (argc, argv) == ERROR) { - printf ("%s: failure parsing arguments\n", progname); - print_help (progname); - return STATE_UNKNOWN; - } - - /* run the command */ - child_process = spopen (VSZ_COMMAND); - if (child_process == NULL) { - printf ("Unable to open pipe: %s\n", VSZ_COMMAND); - 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", VSZ_COMMAND); - - while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_process)) { - - line++; - - /* skip the first line */ - if (line == 1) - continue; - - if (sscanf (input_buffer, VSZ_FORMAT, &proc_size, proc_name) == 2) { - if (proc == NULL) { - if (proc_size > warn) { - asprintf (&message, "%s %s(%d)", message, proc_name, proc_size); - result = max_state (result, STATE_WARNING); - } - if (proc_size > crit) { - result = STATE_CRITICAL; - } - } - else if (strstr (proc_name, proc)) { - asprintf (&message, "%s %d", message, proc_size); - if (proc_size > warn) { - result = max_state (result, STATE_WARNING); - } - if (proc_size > crit) { - result = STATE_CRITICAL; - } - } - } - } - - /* If we get anything on STDERR, at least set warning */ - while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_stderr)) - result = max_state (result, STATE_WARNING); - - (void) fclose (child_stderr); - - /* close the pipe */ - if (spclose (child_process)) - result = max_state (result, STATE_WARNING); - - if (result == STATE_OK) - printf ("ok (all VSZ<%d): %s\n", warn, message); - else if (result == STATE_UNKNOWN) - printf ("Unable to read output\n"); - else if (result == STATE_WARNING) - printf ("WARNING (VSZ>%d):%s\n", warn, message); - else - printf ("CRITICAL (VSZ>%d):%s\n", crit, message); - - return result; -} - - - - -int -process_arguments (int argc, char **argv) -{ - int c; - - int option_index = 0; - static struct option long_options[] = { - {"help", no_argument, 0, 'h'}, - {"version", no_argument, 0, 'V'}, - {"critical", required_argument, 0, 'c'}, - {"warning", required_argument, 0, 'w'}, - {"command", required_argument, 0, 'C'}, - {0, 0, 0, 0} - }; - - if (argc < 2) - return ERROR; - - while (1) { - c = getopt_long (argc, argv, "+hVc:w:C:", long_options, &option_index); - - if (c == EOF) - break; - - switch (c) { - case '?': /* help */ - print_usage (progname); - exit (STATE_UNKNOWN); - case 'h': /* help */ - print_help (progname); - exit (STATE_OK); - case 'V': /* version */ - print_revision (progname, "$Revision$"); - exit (STATE_OK); - case 'c': /* critical threshold */ - if (!is_intnonneg (optarg)) { - printf ("%s: critical threshold must be an integer: %s\n", - progname, optarg); - print_usage (progname); - exit (STATE_UNKNOWN); - } - crit = atoi (optarg); - break; - case 'w': /* warning threshold */ - if (!is_intnonneg (optarg)) { - printf ("%s: warning threshold must be an integer: %s\n", - progname, optarg); - print_usage (progname); - exit (STATE_UNKNOWN); - } - warn = atoi (optarg); - break; - case 'C': /* command name */ - proc = optarg; - break; - } - } - - c = optind; - if (warn == -1) { - if (!is_intnonneg (argv[c])) { - printf ("%s: critical threshold must be an integer: %s\n", - progname, argv[c]); - print_usage (progname); - exit (STATE_UNKNOWN); - } - warn = atoi (argv[c++]); - } - - if (crit == -1) { - if (!is_intnonneg (argv[c])) { - printf ("%s: critical threshold must be an integer: %s\n", - progname, argv[c]); - print_usage (progname); - exit (STATE_UNKNOWN); - } - crit = atoi (argv[c++]); - } - - if (proc == NULL) - proc = argv[c]; - - return c; -} - -void -print_usage (const char *cmd) -{ - printf ("Usage: %s -w -c [-C command]\n" - " %s --help\n" " %s --version\n", cmd, cmd, cmd); -} - -void -print_help (const char *cmd) -{ - print_revision ("check_vsz", "$Revision$"); - printf - ("Copyright (c) 2000 Karl DeBisschop \n\n" - "This plugin checks the image size of a running program and returns an\n" - "error if the number is above either of the thresholds given.\n\n"); - print_usage (cmd); - printf - ("\nOptions:\n" - " -h, --help\n" - " Print detailed help\n" - " -V, --version\n" - " Print version numbers and license information\n" - " -w, --warning=INTEGER\n" - " Program image size necessary to cause a WARNING state\n" - " -c, --critical=INTEGER\n" - " Program image size necessary to cause a CRITICAL state\n" - " -C, --command=STRING\n" " Program to search for [optional]\n"); -} -- cgit v1.2.3-74-g34f1