plugins/check_load.c

   1 /*****************************************************************************
   2 *
   3 * Nagios check_load plugin
   4 *
   5 * License: GPL
   6 * Copyright (c) 1999-2007 Nagios Plugins Development Team
   7 *
   8 * Last Modified: $Date$
   9 *
  10 * Description:
  11 *
  12 * This file contains the check_load plugin
  13 *
  14 * This plugin tests the current system load average.
  15 *
  16 *
  17 * This program is free software: you can redistribute it and/or modify
  18 * it under the terms of the GNU General Public License as published by
  19 * the Free Software Foundation, either version 3 of the License, or
  20 * (at your option) any later version.
  21 *
  22 * This program is distributed in the hope that it will be useful,
  23 * but WITHOUT ANY WARRANTY; without even the implied warranty of
  24 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  25 * GNU General Public License for more details.
  26 *
  27 * You should have received a copy of the GNU General Public License
  28 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  29 *
  30 * $Id$
  31 *
  32 *****************************************************************************/
  33
  34 const char *progname = "check_load";
  35 const char *revision = "$Revision$";
  36 const char *copyright = "1999-2007";
  37 const char *email = "nagiosplug-devel@lists.sourceforge.net";
  38
  39 #include "common.h"
  40 #include "utils.h"
  41 #include "popen.h"
  42
  43 #ifdef HAVE_SYS_LOADAVG_H
  44 #include <sys/loadavg.h>
  45 #endif
  46
  47 /* needed for compilation under NetBSD, as suggested by Andy Doran */
  48 #ifndef LOADAVG_1MIN
  49 #define LOADAVG_1MIN    0
  50 #define LOADAVG_5MIN    1
  51 #define LOADAVG_15MIN   2
  52 #endif /* !defined LOADAVG_1MIN */
  53
  54
  55 static int process_arguments (int argc, char **argv);
  56 static int validate_arguments (void);
  57 void print_help (void);
  58 void print_usage (void);
  59
  60 /* strictly for pretty-print usage in loops */
  61 static const int nums[3] = { 1, 5, 15 };
  62
  63 /* provide some fairly sane defaults */
  64 double wload[3] = { 0.0, 0.0, 0.0 };
  65 double cload[3] = { 0.0, 0.0, 0.0 };
  66 #define la1 la[0]
  67 #define la5 la[1]
  68 #define la15 la[2]
  69
  70 char *status_line;
  71 int take_into_account_cpus = 0;
  72
  73 static void
  74 get_threshold(char *arg, double *th)
  75 {
  76         size_t i, n;
  77         int valid = 0;
  78         char *str = arg, *p;
  79
  80         n = strlen(arg);
  81         for(i = 0; i < 3; i++) {
  82                 th[i] = strtod(str, &p);
  83                 if(p == str) break;
  84
  85                 valid = 1;
  86                 str = p + 1;
  87                 if(n <= (size_t)(str - arg)) break;
  88         }
  89
  90         /* empty argument or non-floatish, so warn about it and die */
  91         if(!i && !valid) usage (_("Warning threshold must be float or float triplet!\n"));
  92
  93         if(i != 2) {
  94                 /* one or more numbers were given, so fill array with last
  95                  * we got (most likely to NOT produce the least expected result) */
  96                 for(n = i; n < 3; n++) th[n] = th[i];
  97         }
  98 }
  99
 100
 101 int
 102 main (int argc, char **argv)
 103 {
 104         int result;
 105         int i;
 106         long numcpus;
 107
 108         double la[3] = { 0.0, 0.0, 0.0 };       /* NetBSD complains about unitialized arrays */
 109 #ifndef HAVE_GETLOADAVG
 110         char input_buffer[MAX_INPUT_BUFFER];
 111 # ifdef HAVE_PROC_LOADAVG
 112         FILE *fp;
 113         char *str, *next;
 114 # endif
 115 #endif
 116
 117         setlocale (LC_ALL, "");
 118         bindtextdomain (PACKAGE, LOCALEDIR);
 119         textdomain (PACKAGE);
 120         setlocale(LC_NUMERIC, "POSIX");
 121
 122         /* Parse extra opts if any */
 123         argv = np_extra_opts (&argc, argv, progname);
 124
 125         if (process_arguments (argc, argv) == ERROR)
 126                 usage4 (_("Could not parse arguments"));
 127
 128 #ifdef HAVE_GETLOADAVG
 129         result = getloadavg (la, 3);
 130         if (result != 3)
 131                 return STATE_UNKNOWN;
 132 #else
 133 # ifdef HAVE_PROC_LOADAVG
 134         fp = fopen (PROC_LOADAVG, "r");
 135         if (fp == NULL) {
 136                 printf (_("Error opening %s\n"), PROC_LOADAVG);
 137                 return STATE_UNKNOWN;
 138         }
 139
 140         while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, fp)) {
 141                 str = (char *)input_buffer;
 142                 for(i = 0; i < 3; i++) {
 143                         la[i] = strtod(str, &next);
 144                         str = next;
 145                 }
 146         }
 147
 148         fclose (fp);
 149 # else
 150         child_process = spopen (PATH_TO_UPTIME);
 151         if (child_process == NULL) {
 152                 printf (_("Error opening %s\n"), PATH_TO_UPTIME);
 153                 return STATE_UNKNOWN;
 154         }
 155         child_stderr = fdopen (child_stderr_array[fileno (child_process)], "r");
 156         if (child_stderr == NULL) {
 157                 printf (_("Could not open stderr for %s\n"), PATH_TO_UPTIME);
 158         }
 159         fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_process);
 160         sscanf (input_buffer, "%*[^l]load average: %lf, %lf, %lf", &la1, &la5, &la15);
 161
 162         result = spclose (child_process);
 163         if (result) {
 164                 printf (_("Error code %d returned in %s\n"), result, PATH_TO_UPTIME);
 165                 return STATE_UNKNOWN;
 166         }
 167 # endif
 168 #endif
 169
 170         if (take_into_account_cpus == 1) {
 171                 if ((numcpus = GET_NUMBER_OF_CPUS()) > 0) {
 172                         la[0] = la[0] / numcpus;
 173                         la[1] = la[1] / numcpus;
 174                         la[2] = la[2] / numcpus;
 175                 }
 176         }
 177         if ((la[0] < 0.0) || (la[1] < 0.0) || (la[2] < 0.0)) {
 178 #ifdef HAVE_GETLOADAVG
 179                 printf (_("Error in getloadavg()\n"));
 180 #else
 181 # ifdef HAVE_PROC_LOADAVG
 182                 printf (_("Error processing %s\n"), PROC_LOADAVG);
 183 # else
 184                 printf (_("Error processing %s\n"), PATH_TO_UPTIME);
 185 # endif
 186 #endif
 187                 return STATE_UNKNOWN;
 188         }
 189
 190         /* we got this far, so assume OK until we've measured */
 191         result = STATE_OK;
 192
 193         asprintf(&status_line, _("load average: %.2f, %.2f, %.2f"), la1, la5, la15);
 194
 195         for(i = 0; i < 3; i++) {
 196                 if(la[i] > cload[i]) {
 197                         result = STATE_CRITICAL;
 198                         break;
 199                 }
 200                 else if(la[i] > wload[i]) result = STATE_WARNING;
 201         }
 202
 203         printf("%s - %s|", state_text(result), status_line);
 204         for(i = 0; i < 3; i++)
 205                 printf("load%d=%.3f;%.3f;%.3f;0; ", nums[i], la[i], wload[i], cload[i]);
 206
 207         putchar('\n');
 208         return result;
 209 }
 210
 211
 212 /* process command-line arguments */
 213 static int
 214 process_arguments (int argc, char **argv)
 215 {
 216         int c = 0;
 217
 218         int option = 0;
 219         static struct option longopts[] = {
 220                 {"warning", required_argument, 0, 'w'},
 221                 {"critical", required_argument, 0, 'c'},
 222                 {"percpu", no_argument, 0, 'r'},
 223                 {"version", no_argument, 0, 'V'},
 224                 {"help", no_argument, 0, 'h'},
 225                 {0, 0, 0, 0}
 226         };
 227
 228         if (argc < 2)
 229                 return ERROR;
 230
 231         while (1) {
 232                 c = getopt_long (argc, argv, "Vhrc:w:", longopts, &option);
 233
 234                 if (c == -1 || c == EOF)
 235                         break;
 236
 237                 switch (c) {
 238                 case 'w': /* warning time threshold */
 239                         get_threshold(optarg, wload);
 240                         break;
 241                 case 'c': /* critical time threshold */
 242                         get_threshold(optarg, cload);
 243                         break;
 244                 case 'r': /* Divide load average by number of CPUs */
 245                         take_into_account_cpus = 1;
 246                         break;
 247                 case 'V':                                                                       /* version */
 248                         print_revision (progname, revision);
 249                         exit (STATE_OK);
 250                 case 'h':                                                                       /* help */
 251                         print_help ();
 252                         exit (STATE_OK);
 253                 case '?':                                                                       /* help */
 254                         usage5 ();
 255                 }
 256         }
 257
 258         c = optind;
 259         if (c == argc)
 260                 return validate_arguments ();
 261
 262         /* handle the case if both arguments are missing,
 263          * but not if only one is given without -c or -w flag */
 264         if(c - argc == 2) {
 265                 get_threshold(argv[c++], wload);
 266                 get_threshold(argv[c++], cload);
 267         }
 268         else if(c - argc == 1) {
 269                 get_threshold(argv[c++], cload);
 270         }
 271
 272         return validate_arguments ();
 273 }
 274
 275
 276
 277 static int
 278 validate_arguments (void)
 279 {
 280         int i = 0;
 281
 282         /* match cload first, as it will give the most friendly error message
 283          * if user hasn't given the -c switch properly */
 284         for(i = 0; i < 3; i++) {
 285                 if(cload[i] < 0)
 286                         die (STATE_UNKNOWN, _("Critical threshold for %d-minute load average is not specified\n"), nums[i]);
 287                 if(wload[i] < 0)
 288                         die (STATE_UNKNOWN, _("Warning threshold for %d-minute load average is not specified\n"), nums[i]);
 289                 if(wload[i] > cload[i])
 290                         die (STATE_UNKNOWN, _("Parameter inconsistency: %d-minute \"warning load\" is greater than \"critical load\"\n"), nums[i]);
 291         }
 292
 293         return OK;
 294 }
 295
 296
 297
 298 void
 299 print_help (void)
 300 {
 301         print_revision (progname, revision);
 302
 303         printf ("Copyright (c) 1999 Felipe Gustavo de Almeida <galmeida@linux.ime.usp.br>\n");
 304         printf (COPYRIGHT, copyright, email);
 305
 306         printf (_("This plugin tests the current system load average."));
 307
 308   printf ("\n\n");
 309
 310         print_usage ();
 311
 312         printf (_(UT_HELP_VRSN));
 313         printf (_(UT_EXTRA_OPTS));
 314
 315         printf (" %s\n", "-w, --warning=WLOAD1,WLOAD5,WLOAD15");
 316   printf ("    %s\n", _("Exit with WARNING status if load average exceeds WLOADn"));
 317   printf (" %s\n", "-c, --critical=CLOAD1,CLOAD5,CLOAD15");
 318   printf ("    %s\n", _("Exit with CRITICAL status if load average exceed CLOADn"));
 319   printf ("    %s\n", _("the load average format is the same used by \"uptime\" and \"w\""));
 320   printf (" %s\n", "-r, --percpu");
 321   printf ("    %s\n", _("Divide the load averages by the number of CPUs (when possible)"));
 322
 323 #ifdef NP_EXTRA_OPTS
 324         printf ("\n");
 325         printf ("%s\n", _("Notes:"));
 326         printf (_(UT_EXTRA_OPTS_NOTES));
 327 #endif
 328
 329         printf (_(UT_SUPPORT));
 330 }
 331
 332 void
 333 print_usage (void)
 334 {
 335   printf (_("Usage:"));
 336         printf ("%s [-r] -w WLOAD1,WLOAD5,WLOAD15 -c CLOAD1,CLOAD5,CLOAD15\n", progname);
 337 }