Code

Verbose should not have to exceed 3 as per developement guidelines
[nagiosplug.git] / plugins / check_procs.c
index 816da1b7af5d61fb9a2a7fe5e07ad65124410fe3..4aaeddb240bd89f8cb172cdaa18b1405a57dd228 100644 (file)
@@ -41,8 +41,8 @@ const char *copyright = "2000-2008";
 const char *email = "nagiosplug-devel@lists.sourceforge.net";
 
 #include "common.h"
-#include "popen.h"
 #include "utils.h"
+#include "utils_cmd.h"
 #include "regex.h"
 
 #include <pwd.h>
@@ -129,8 +129,9 @@ main (int argc, char **argv)
        int expected_cols = PS_COLS - 1;
        int warn = 0; /* number of processes in warn state */
        int crit = 0; /* number of processes in crit state */
-       int i = 0;
+       int i = 0, j = 0;
        int result = STATE_UNKNOWN;
+       output chld_out, chld_err;
 
        setlocale (LC_ALL, "");
        bindtextdomain (PACKAGE, LOCALEDIR);
@@ -143,6 +144,9 @@ main (int argc, char **argv)
        asprintf (&metric_name, "PROCS");
        metric = METRIC_PROCS;
 
+       /* Parse extra opts if any */
+       argv=np_extra_opts (&argc, argv, progname);
+
        if (process_arguments (argc, argv) == ERROR)
                usage4 (_("Could not parse arguments"));
 
@@ -150,41 +154,27 @@ main (int argc, char **argv)
        mypid = getpid();
 
        /* Set signal handling and alarm timeout */
-       if (signal (SIGALRM, popen_timeout_alarm_handler) == SIG_ERR) {
-               usage4 (_("Cannot catch SIGALRM"));
+       if (signal (SIGALRM, timeout_alarm_handler) == SIG_ERR) {
+               die (STATE_UNKNOWN, _("Cannot catch SIGALRM"));
        }
-       alarm (timeout_interval);
+       (void) alarm ((unsigned) timeout_interval);
 
        if (verbose >= 2)
                printf (_("CMD: %s\n"), PS_COMMAND);
 
        if (input_filename == NULL) {
-               ps_input = spopen (PS_COMMAND);
-               if (ps_input == NULL) {
-                       printf (_("Could not open pipe: %s\n"), PS_COMMAND);
-                       return STATE_UNKNOWN;
+               result = cmd_run( PS_COMMAND, &chld_out, &chld_err, 0);
+               if (chld_err.lines > 0) {
+                       printf ("%s: %s", _("System call sent warnings to stderr"), chld_err.line[0]);
+                       exit(STATE_WARNING);
                }
-               child_stderr = fdopen (child_stderr_array[fileno (ps_input)], "r");
-               if (child_stderr == NULL)
-                       printf (_("Could not open stderr for %s\n"), PS_COMMAND);
        } else {
-               ps_input = fopen(input_filename, "r");
-               if (ps_input == NULL) {
-                       die( STATE_UNKNOWN, _("Error opening %s\n"), input_filename );
-               }
+               result = cmd_file_read( input_filename, &chld_out, 0);
        }
 
-       /* flush first line */
-       fgets (input_buffer, MAX_INPUT_BUFFER - 1, ps_input);
-       while ( input_buffer[strlen(input_buffer)-1] != '\n' )
-               fgets (input_buffer, MAX_INPUT_BUFFER - 1, ps_input);
-
-       while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, ps_input)) {
-               asprintf (&input_line, "%s", input_buffer);
-               while ( input_buffer[strlen(input_buffer)-1] != '\n' ) {
-                       fgets (input_buffer, MAX_INPUT_BUFFER - 1, ps_input);
-                       asprintf (&input_line, "%s%s", input_line, input_buffer);
-               }
+       /* flush first line: j starts at 1 */
+       for (j = 1; j < chld_out.lines; j++) {
+               input_line = chld_out.line[j];
 
                if (verbose >= 3)
                        printf ("%s", input_line);
@@ -280,27 +270,9 @@ main (int argc, char **argv)
                }
        }
 
-       /* If we get anything on STDERR, at least set warning */
-       if (input_filename == NULL) {
-               while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_stderr)) {
-                       if (verbose)
-                               printf ("STDERR: %s", input_buffer);
-                       result = max_state (result, STATE_WARNING);
-                       printf (_("System call sent warnings to stderr\n"));
-               }
-       
-               (void) fclose (child_stderr);
-
-               /* close the pipe */
-               if (spclose (ps_input)) {
-                       printf (_("System call returned nonzero status\n"));
-                       result = max_state (result, STATE_WARNING);
-               }
-       }
-
        if (found == 0) {                                                       /* no process lines parsed so return STATE_UNKNOWN */
                printf (_("Unable to read output\n"));
-               return result;
+               return STATE_UNKNOWN;
        }
 
        if ( result == STATE_UNKNOWN ) 
@@ -715,13 +687,12 @@ print_help (void)
 
        print_usage ();
 
-       printf ("%s\n", _("Required Arguments:"));
+  printf (_(UT_HELP_VRSN));
+  printf (_(UT_EXTRA_OPTS));
   printf (" %s\n", "-w, --warning=RANGE");
   printf ("   %s\n", _("Generate warning state if metric is outside this range"));
   printf (" %s\n", "-c, --critical=RANGE");
   printf ("   %s\n", _("Generate critical state if metric is outside this range"));
-
-       printf ("%s\n", _("Optional Arguments:"));
   printf (" %s\n", "-m, --metric=TYPE");
   printf ("  %s\n", _("Check thresholds against metric. Valid types:"));
   printf ("  %s\n", _("PROCS   - number of processes (default)"));
@@ -737,7 +708,8 @@ print_help (void)
        printf (" %s\n", "-v, --verbose");
   printf ("    %s\n", _("Extra information. Up to 3 verbosity levels"));
 
-       printf ("%s\n", "Optional Filters:");
+  printf ("\n");
+       printf ("%s\n", "Filters:");
   printf (" %s\n", "-s, --state=STATUSFLAGS");
   printf ("   %s\n", _("Only scan for processes that have, in the output of `ps`, one or"));
   printf ("   %s\n", _("more of the status flags you specify (for example R, Z, S, RS,"));
@@ -771,6 +743,12 @@ the specified threshold ranges. The process count can be filtered by\n\
 process owner, parent process PID, current state (e.g., 'Z'), or may\n\
 be the total number of running processes\n\n"));
 
+#ifdef NP_EXTRA_OPTS
+  printf ("%s\n", _("Notes:"));
+  printf (_(UT_EXTRA_OPTS_NOTES));
+  printf ("\n");
+#endif
+
        printf ("%s\n", _("Examples:"));
   printf (" %s\n", "check_procs -w 2:2 -c 2:1024 -C portsentry");
   printf ("  %s\n", _("Warning if not two processes with command name portsentry."));