Initial revision

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

1 files changed, 287 insertions, 0 deletions

diff --git a/plugins/check_vsz.c b/plugins/check_vsz.c
new file mode 100644
index 00000000..c8ca82bd
--- /dev/null
+++ b/plugins/check_vsz.c
@@ -0,0 +1,287 @@
+/******************************************************************************
+ *
+ * CHECK_VSZ.C
+ *
+ * Program: Process plugin for Nagios
+ * License: GPL
+ * Copyright (c) 1999,2000 Karl DeBisschop <kdebiss@alum.mit.edu>
+ *
+ * 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
+ *
+ *****************************************************************************/
+
+#include "common.h"
+#include "popen.h"
+#include "utils.h"
+
+int process_arguments (int argc, char **argv);
+int call_getopt (int argc, char **argv);
+void print_help (char *cmd);
+void print_usage (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 = NULL;
+
+        if (!process_arguments (argc, argv)) {
+                printf ("%s: failure parsing arguments\n", my_basename (argv[0]));
+                print_help (my_basename (argv[0]));
+                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);
+
+        message = malloc ((size_t) 1);
+        message[0] = 0;
+        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) {
+                                        len = strlen (message) + strlen (proc_name) + 23;
+                                        message = realloc (message, len);
+                                        if (message == NULL)
+                                                terminate (STATE_UNKNOWN,
+                                                                                         "check_vsz: could not malloc message (1)");
+                                        sprintf (message, "%s %s(%d)", message, proc_name, proc_size);
+                                        result = max (result, STATE_WARNING);
+                                }
+                                if (proc_size > crit) {
+                                        result = STATE_CRITICAL;
+                                }
+                        }
+                        else if (strstr (proc_name, proc)) {
+                                len = strlen (message) + 21;
+                                message = realloc (message, len);
+                                if (message == NULL)
+                                        terminate (STATE_UNKNOWN,
+                                                                                 "check_vsz: could not malloc message (2)");
+                                sprintf (message, "%s %d", message, proc_size);
+                                if (proc_size > warn) {
+                                        result = max (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 (result, STATE_WARNING);
+
+        (void) fclose (child_stderr);
+
+        /* close the pipe */
+        if (spclose (child_process))
+                result = max (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;
+
+        if (argc < 2)
+                return ERROR;
+
+        c = 0;
+        while (c += (call_getopt (argc - c, &argv[c]))) {
+                if (argc <= c)
+                        break;
+                if (warn == -1) {
+                        if (!is_intnonneg (argv[c])) {
+                                printf ("%s: critical threshold must be an integer: %s\n",
+                                                                my_basename (argv[0]), argv[c]);
+                                print_usage (my_basename (argv[0]));
+                                exit (STATE_UNKNOWN);
+                        }
+                        warn = atoi (argv[c]);
+                }
+                else if (crit == -1) {
+                        if (!is_intnonneg (argv[c])) {
+                                printf ("%s: critical threshold must be an integer: %s\n",
+                                                                my_basename (argv[0]), argv[c]);
+                                print_usage (my_basename (argv[0]));
+                                exit (STATE_UNKNOWN);
+                        }
+                        crit = atoi (argv[c]);
+                }
+                else if (proc == NULL) {
+                        proc = malloc (strlen (argv[c]) + 1);
+                        if (proc == NULL)
+                                terminate (STATE_UNKNOWN,
+                                                                         "check_vsz: failed malloc of proc in process_arguments");
+                        strcpy (proc, argv[c]);
+                }
+        }
+        return c;
+}
+
+int
+call_getopt (int argc, char **argv)
+{
+        int c, i = 1;
+
+#ifdef HAVE_GETOPT_H
+        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}
+        };
+#endif
+
+        while (1) {
+#ifdef HAVE_GETOPT_H
+                c = getopt_long (argc, argv, "+hVc:w:C:", long_options, &option_index);
+#else
+                c = getopt (argc, argv, "+hVc:w:C:");
+#endif
+                if (c == EOF)
+                        break;
+
+                i++;
+                switch (c) {
+                case 'c':
+                case 'w':
+                case 'C':
+                        i++;
+                }
+
+                switch (c) {
+                case '?':                                                                       /* help */
+                        printf ("%s: Unknown argument: %s\n\n", my_basename (argv[0]), optarg);
+                        print_usage (my_basename (argv[0]));
+                        exit (STATE_UNKNOWN);
+                case 'h':                                                                       /* help */
+                        print_help (my_basename (argv[0]));
+                        exit (STATE_OK);
+                case 'V':                                                                       /* version */
+                        print_revision (my_basename (argv[0]), "$Revision$");
+                        exit (STATE_OK);
+                case 'c':                                                                       /* critical threshold */
+                        if (!is_intnonneg (optarg)) {
+                                printf ("%s: critical threshold must be an integer: %s\n",
+                                                                my_basename (argv[0]), optarg);
+                                print_usage (my_basename (argv[0]));
+                                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",
+                                                                my_basename (argv[0]), optarg);
+                                print_usage (my_basename (argv[0]));
+                                exit (STATE_UNKNOWN);
+                        }
+                        warn = atoi (optarg);
+                        break;
+                case 'C':                                                                       /* command name */
+                        proc = malloc (strlen (optarg) + 1);
+                        if (proc == NULL)
+                                terminate (STATE_UNKNOWN,
+                                                                         "check_vsz: failed malloc of proc in process_arguments");
+                        strcpy (proc, optarg);
+                        break;
+                }
+        }
+        return i;
+}
+
+void
+print_usage (char *cmd)
+{
+        printf ("Usage: %s -w <wsize> -c <csize> [-C command]\n"
+                                        "       %s --help\n" "       %s --version\n", cmd, cmd, cmd);
+}
+
+void
+print_help (char *cmd)
+{
+        print_revision ("check_vsz", "$Revision$");
+        printf
+                ("Copyright (c) 2000 Karl DeBisschop <kdebiss@alum.mit.edu>\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");
+}