Logo Search packages:      
Sourcecode: nagios-plugins version File versions  Download package

check_load.c

/*****************************************************************************
* 
* Nagios check_load plugin
* 
* License: GPL
* Copyright (c) 1999-2007 Nagios Plugins Development Team
* 
* Description:
* 
* This file contains the check_load plugin
* 
* This plugin tests the current system load average.
* 
* 
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
* 
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
* GNU General Public License for more details.
* 
* You should have received a copy of the GNU General Public License
* along with this program.  If not, see <http://www.gnu.org/licenses/>.
* 
* 
*****************************************************************************/

const char *progname = "check_load";
const char *copyright = "1999-2007";
const char *email = "nagiosplug-devel@lists.sourceforge.net";

#include "common.h"
#include "utils.h"
#include "popen.h"

#ifdef HAVE_SYS_LOADAVG_H
#include <sys/loadavg.h>
#endif

/* needed for compilation under NetBSD, as suggested by Andy Doran */
#ifndef LOADAVG_1MIN
#define LOADAVG_1MIN    0
#define LOADAVG_5MIN    1
#define LOADAVG_15MIN   2
#endif /* !defined LOADAVG_1MIN */


static int process_arguments (int argc, char **argv);
static int validate_arguments (void);
void print_help (void);
void print_usage (void);

/* strictly for pretty-print usage in loops */
static const int nums[3] = { 1, 5, 15 };

/* provide some fairly sane defaults */
double wload[3] = { 0.0, 0.0, 0.0 };
double cload[3] = { 0.0, 0.0, 0.0 };
#define la1 la[0]
#define la5 la[1]
#define la15 la[2]

char *status_line;
int take_into_account_cpus = 0;

static void
get_threshold(char *arg, double *th)
{
      size_t i, n;
      int valid = 0;
      char *str = arg, *p;

      n = strlen(arg);
      for(i = 0; i < 3; i++) {
            th[i] = strtod(str, &p);
            if(p == str) break;

            valid = 1;
            str = p + 1;
            if(n <= (size_t)(str - arg)) break;
      }

      /* empty argument or non-floatish, so warn about it and die */
      if(!i && !valid) usage (_("Warning threshold must be float or float triplet!\n"));

      if(i != 2) {
            /* one or more numbers were given, so fill array with last
             * we got (most likely to NOT produce the least expected result) */
            for(n = i; n < 3; n++) th[n] = th[i];
      }
}


int
main (int argc, char **argv)
{
      int result;
      int i;
      long numcpus;

      double la[3] = { 0.0, 0.0, 0.0 };   /* NetBSD complains about unitialized arrays */
#ifndef HAVE_GETLOADAVG
      char input_buffer[MAX_INPUT_BUFFER];
# ifdef HAVE_PROC_LOADAVG
      FILE *fp;
      char *str, *next;
# endif
#endif

      setlocale (LC_ALL, "");
      bindtextdomain (PACKAGE, LOCALEDIR);
      textdomain (PACKAGE);
      setlocale(LC_NUMERIC, "POSIX");

      /* Parse extra opts if any */
      argv = np_extra_opts (&argc, argv, progname);

      if (process_arguments (argc, argv) == ERROR)
            usage4 (_("Could not parse arguments"));

#ifdef HAVE_GETLOADAVG
      result = getloadavg (la, 3);
      if (result != 3)
            return STATE_UNKNOWN;
#else
# ifdef HAVE_PROC_LOADAVG
      fp = fopen (PROC_LOADAVG, "r");
      if (fp == NULL) {
            printf (_("Error opening %s\n"), PROC_LOADAVG);
            return STATE_UNKNOWN;
      }

      while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, fp)) {
            str = (char *)input_buffer;
            for(i = 0; i < 3; i++) {
                  la[i] = strtod(str, &next);
                  str = next;
            }
      }

      fclose (fp);
# else
      child_process = spopen (PATH_TO_UPTIME);
      if (child_process == NULL) {
            printf (_("Error opening %s\n"), PATH_TO_UPTIME);
            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"), PATH_TO_UPTIME);
      }
      fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_process);
      sscanf (input_buffer, "%*[^l]load average: %lf, %lf, %lf", &la1, &la5, &la15);

      result = spclose (child_process);
      if (result) {
            printf (_("Error code %d returned in %s\n"), result, PATH_TO_UPTIME);
            return STATE_UNKNOWN;
      }
# endif
#endif

      if (take_into_account_cpus == 1) {
            if ((numcpus = GET_NUMBER_OF_CPUS()) > 0) {
                  la[0] = la[0] / numcpus;
                  la[1] = la[1] / numcpus;
                  la[2] = la[2] / numcpus;
            }
      }
      if ((la[0] < 0.0) || (la[1] < 0.0) || (la[2] < 0.0)) {
#ifdef HAVE_GETLOADAVG
            printf (_("Error in getloadavg()\n"));
#else
# ifdef HAVE_PROC_LOADAVG
            printf (_("Error processing %s\n"), PROC_LOADAVG);
# else
            printf (_("Error processing %s\n"), PATH_TO_UPTIME);
# endif
#endif
            return STATE_UNKNOWN;
      }

      /* we got this far, so assume OK until we've measured */
      result = STATE_OK;

      asprintf(&status_line, _("load average: %.2f, %.2f, %.2f"), la1, la5, la15);

      for(i = 0; i < 3; i++) {
            if(la[i] > cload[i]) {
                  result = STATE_CRITICAL;
                  break;
            }
            else if(la[i] > wload[i]) result = STATE_WARNING;
      }

      printf("%s - %s|", state_text(result), status_line);
      for(i = 0; i < 3; i++)
            printf("load%d=%.3f;%.3f;%.3f;0; ", nums[i], la[i], wload[i], cload[i]);

      putchar('\n');
      return result;
}


/* process command-line arguments */
static int
process_arguments (int argc, char **argv)
{
      int c = 0;

      int option = 0;
      static struct option longopts[] = {
            {"warning", required_argument, 0, 'w'},
            {"critical", required_argument, 0, 'c'},
            {"percpu", no_argument, 0, 'r'},
            {"version", no_argument, 0, 'V'},
            {"help", no_argument, 0, 'h'},
            {0, 0, 0, 0}
      };

      if (argc < 2)
            return ERROR;

      while (1) {
            c = getopt_long (argc, argv, "Vhrc:w:", longopts, &option);

            if (c == -1 || c == EOF)
                  break;

            switch (c) {
            case 'w': /* warning time threshold */
                  get_threshold(optarg, wload);
                  break;
            case 'c': /* critical time threshold */
                  get_threshold(optarg, cload);
                  break;
            case 'r': /* Divide load average by number of CPUs */
                  take_into_account_cpus = 1;
                  break;
            case 'V':                                                   /* version */
                  print_revision (progname, NP_VERSION);
                  exit (STATE_OK);
            case 'h':                                                   /* help */
                  print_help ();
                  exit (STATE_OK);
            case '?':                                                   /* help */
                  usage5 ();
            }
      }

      c = optind;
      if (c == argc)
            return validate_arguments ();

      /* handle the case if both arguments are missing,
       * but not if only one is given without -c or -w flag */
      if(c - argc == 2) {
            get_threshold(argv[c++], wload);
            get_threshold(argv[c++], cload);
      }
      else if(c - argc == 1) {
            get_threshold(argv[c++], cload);
      }

      return validate_arguments ();
}



static int
validate_arguments (void)
{
      int i = 0;

      /* match cload first, as it will give the most friendly error message
       * if user hasn't given the -c switch properly */
      for(i = 0; i < 3; i++) {
            if(cload[i] < 0)
                  die (STATE_UNKNOWN, _("Critical threshold for %d-minute load average is not specified\n"), nums[i]);
            if(wload[i] < 0)
                  die (STATE_UNKNOWN, _("Warning threshold for %d-minute load average is not specified\n"), nums[i]);
            if(wload[i] > cload[i])
                  die (STATE_UNKNOWN, _("Parameter inconsistency: %d-minute \"warning load\" is greater than \"critical load\"\n"), nums[i]);
      }

      return OK;
}



void
print_help (void)
{
      print_revision (progname, NP_VERSION);

      printf ("Copyright (c) 1999 Felipe Gustavo de Almeida <galmeida@linux.ime.usp.br>\n");
      printf (COPYRIGHT, copyright, email);

      printf (_("This plugin tests the current system load average."));

  printf ("\n\n");

      print_usage ();

      printf (_(UT_HELP_VRSN));
      printf (_(UT_EXTRA_OPTS));

      printf (" %s\n", "-w, --warning=WLOAD1,WLOAD5,WLOAD15");
  printf ("    %s\n", _("Exit with WARNING status if load average exceeds WLOADn"));
  printf (" %s\n", "-c, --critical=CLOAD1,CLOAD5,CLOAD15");
  printf ("    %s\n", _("Exit with CRITICAL status if load average exceed CLOADn"));
  printf ("    %s\n", _("the load average format is the same used by \"uptime\" and \"w\""));
  printf (" %s\n", "-r, --percpu");
  printf ("    %s\n", _("Divide the load averages by the number of CPUs (when possible)"));

#ifdef NP_EXTRA_OPTS
      printf ("\n");
      printf ("%s\n", _("Notes:"));
      printf (_(UT_EXTRA_OPTS_NOTES));
#endif

      printf (_(UT_SUPPORT));
}

void
print_usage (void)
{
  printf (_("Usage:"));
      printf ("%s [-r] -w WLOAD1,WLOAD5,WLOAD15 -c CLOAD1,CLOAD5,CLOAD15\n", progname);
}

Generated by  Doxygen 1.6.0   Back to index