diff options
Diffstat (limited to 'plugins/check_ups.c')
-rw-r--r-- | plugins/check_ups.c | 649 |
1 files changed, 649 insertions, 0 deletions
diff --git a/plugins/check_ups.c b/plugins/check_ups.c new file mode 100644 index 00000000..4632bdc0 --- /dev/null +++ b/plugins/check_ups.c @@ -0,0 +1,649 @@ +/****************************************************************************** +* +* CHECK_UPS.C +* +* Program: UPS monitor plugin for Nagios +* License: GPL +* Copyright (c) 1999 Ethan Galstad (nagios@nagios.org) +* +* Last Modified: $Date$ +* +* Command line: CHECK_UPS <host_address> [-u ups] [-p port] [-v variable] \ +* [-wv warn_value] [-cv crit_value] [-to to_sec] +* +* Description: +* + +* This plugin attempts to determine the status of an UPS +* (Uninterruptible Power Supply) on a remote host (or the local host) +* that is being monitored with Russel Kroll's "Smarty UPS Tools" +* package. If the UPS is online or calibrating, the plugin will +* return an OK state. If the battery is on it will return a WARNING +* state. If the UPS is off or has a low battery the plugin will +* return a CRITICAL state. You may also specify a variable to check +* (such as temperature, utility voltage, battery load, etc.) as well +* as warning and critical thresholds for the value of that variable. +* If the remote host has multiple UPS that are being monitored you +* will have to use the [ups] option to specify which UPS to check. +* +* Notes: +* +* This plugin requires that the UPSD daemon distributed with Russel +* Kroll's "Smart UPS Tools" be installed on the remote host. If you +* don't have the package installed on your system, you can download +* it from http://www.exploits.org/~rkroll/smartupstools +* +* License Information: +* +* 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 2 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, write to the Free Software +* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. +* +******************************************************************************/ + +#include "config.h" +#include "common.h" +#include "netutils.h" +#include "utils.h" + +#define PROGNAME "check_ups" + +#define CHECK_NONE 0 + +#define PORT 3305 + +#define UPS_NONE 0 /* no supported options */ +#define UPS_UTILITY 1 /* supports utility line voltage */ +#define UPS_BATTPCT 2 /* supports percent battery remaining */ +#define UPS_STATUS 4 /* supports UPS status */ +#define UPS_TEMP 8 /* supports UPS temperature */ +#define UPS_LOADPCT 16 /* supports load percent */ + +#define UPSSTATUS_NONE 0 +#define UPSSTATUS_OFF 1 +#define UPSSTATUS_OL 2 +#define UPSSTATUS_OB 4 +#define UPSSTATUS_LB 8 +#define UPSSTATUS_CAL 16 +#define UPSSTATUS_UNKOWN 32 + +int server_port = PORT; +char *server_address = NULL; +char *ups_name = NULL; +double warning_value = 0.0L; +double critical_value = 0.0L; +int check_warning_value = FALSE; +int check_critical_value = FALSE; +int check_variable = UPS_NONE; +int supported_options = UPS_NONE; +int status = UPSSTATUS_NONE; + +double ups_utility_voltage = 0.0L; +double ups_battery_percent = 0.0L; +double ups_load_percent = 0.0L; +double ups_temperature = 0.0L; +char ups_status[MAX_INPUT_BUFFER] = "N/A"; + +int determine_status (void); +int determine_supported_vars (void); +int get_ups_variable (const char *, char *, int); + +int process_arguments (int, char **); +int call_getopt (int, char **); +int validate_arguments (void); +void print_help (void); +void print_usage (void); + +int +main (int argc, char **argv) +{ + int result = STATE_OK; + char output_message[MAX_INPUT_BUFFER]; + char temp_buffer[MAX_INPUT_BUFFER]; + + double ups_utility_deviation = 0.0L; + + if (process_arguments (argc, argv) != OK) + usage ("Invalid command arguments supplied\n"); + + /* initialize alarm signal handling */ + signal (SIGALRM, socket_timeout_alarm_handler); + + /* set socket timeout */ + alarm (socket_timeout); + + /* determine what variables the UPS supports */ + if (determine_supported_vars () != OK) + return STATE_CRITICAL; + + /* get the ups status if possible */ + if (supported_options & UPS_STATUS) { + + if (determine_status () != OK) + return STATE_CRITICAL; + ups_status[0] = 0; + result = STATE_OK; + + if (status & UPSSTATUS_OFF) { + strcpy (ups_status, "Off"); + result = STATE_CRITICAL; + } + else if ((status & (UPSSTATUS_OB | UPSSTATUS_LB)) == + (UPSSTATUS_OB | UPSSTATUS_LB)) { + strcpy (ups_status, "On Battery, Low Battery"); + result = STATE_CRITICAL; + } + else { + if (status & UPSSTATUS_OL) { + strcat (ups_status, "Online"); + } + if (status & UPSSTATUS_OB) { + strcat (ups_status, "On Battery"); + result = STATE_WARNING; + } + if (status & UPSSTATUS_LB) { + strcat (ups_status, ", Low Battery"); + result = STATE_WARNING; + } + if (status & UPSSTATUS_CAL) { + strcat (ups_status, ", Calibrating"); + } + if (status & UPSSTATUS_UNKOWN) { + strcat (ups_status, ", Unknown"); + } + } + } + + /* get the ups utility voltage if possible */ + if (supported_options & UPS_UTILITY) { + + if (get_ups_variable ("UTILITY", temp_buffer, sizeof (temp_buffer)) != OK) + return STATE_CRITICAL; + + ups_utility_voltage = atof (temp_buffer); + + if (ups_utility_voltage > 120.0) + ups_utility_deviation = 120.0 - ups_utility_voltage; + else + ups_utility_deviation = ups_utility_voltage - 120.0; + + if (check_variable == UPS_UTILITY) { + if (check_critical_value == TRUE + && ups_utility_deviation >= critical_value) result = STATE_CRITICAL; + else if (check_warning_value == TRUE + && ups_utility_deviation >= warning_value + && result < STATE_WARNING) result = STATE_WARNING; + } + } + + /* get the ups battery percent if possible */ + if (supported_options & UPS_BATTPCT) { + + if (get_ups_variable ("BATTPCT", temp_buffer, sizeof (temp_buffer)) != OK) + return STATE_CRITICAL; + + ups_battery_percent = atof (temp_buffer); + + if (check_variable == UPS_BATTPCT) { + if (check_critical_value == TRUE + && ups_battery_percent <= critical_value) result = STATE_CRITICAL; + else if (check_warning_value == TRUE + && ups_battery_percent <= warning_value + && result < STATE_WARNING) result = STATE_WARNING; + } + } + + /* get the ups load percent if possible */ + if (supported_options & UPS_LOADPCT) { + + if (get_ups_variable ("LOADPCT", temp_buffer, sizeof (temp_buffer)) != OK) + return STATE_CRITICAL; + + ups_load_percent = atof (temp_buffer); + + if (check_variable == UPS_LOADPCT) { + if (check_critical_value == TRUE && ups_load_percent >= critical_value) + result = STATE_CRITICAL; + else if (check_warning_value == TRUE + && ups_load_percent >= warning_value && result < STATE_WARNING) + result = STATE_WARNING; + } + } + + /* get the ups temperature if possible */ + if (supported_options & UPS_TEMP) { + + if (get_ups_variable ("UPSTEMP", temp_buffer, sizeof (temp_buffer)) != OK) + return STATE_CRITICAL; + + ups_temperature = (atof (temp_buffer) * 1.8) + 32; + + if (check_variable == UPS_TEMP) { + if (check_critical_value == TRUE && ups_temperature >= critical_value) + result = STATE_CRITICAL; + else if (check_warning_value == TRUE && ups_temperature >= warning_value + && result < STATE_WARNING) + result = STATE_WARNING; + } + } + + /* if the UPS does not support any options we are looking for, report an error */ + if (supported_options == UPS_NONE) + result = STATE_CRITICAL; + + /* reset timeout */ + alarm (0); + + + sprintf (output_message, "UPS %s - ", + (result == STATE_OK) ? "ok" : "problem"); + + if (supported_options & UPS_STATUS) { + sprintf (temp_buffer, "Status=%s ", ups_status); + strcat (output_message, temp_buffer); + } + if (supported_options & UPS_UTILITY) { + sprintf (temp_buffer, "Utility=%3.1fV ", ups_utility_voltage); + strcat (output_message, temp_buffer); + } + if (supported_options & UPS_BATTPCT) { + sprintf (temp_buffer, "Batt=%3.1f%% ", ups_battery_percent); + strcat (output_message, temp_buffer); + } + if (supported_options & UPS_LOADPCT) { + sprintf (temp_buffer, "Load=%3.1f%% ", ups_load_percent); + strcat (output_message, temp_buffer); + } + if (supported_options & UPS_TEMP) { + sprintf (temp_buffer, "Temp=%3.1fF", ups_temperature); + strcat (output_message, temp_buffer); + } + if (supported_options == UPS_NONE) { + sprintf (temp_buffer, + "UPS does not appear to support any available options\n"); + strcat (output_message, temp_buffer); + } + + printf ("%s\n", output_message); + + return result; +} + + + +/* determines what options are supported by the UPS */ +int +determine_status (void) +{ + char recv_buffer[MAX_INPUT_BUFFER]; + char temp_buffer[MAX_INPUT_BUFFER]; + char *ptr; + + if (get_ups_variable ("STATUS", recv_buffer, sizeof (recv_buffer)) != + STATE_OK) { + printf ("Invalid response received from hostn"); + return ERROR; + } + + recv_buffer[strlen (recv_buffer) - 1] = 0; + + strcpy (temp_buffer, recv_buffer); + for (ptr = (char *) strtok (temp_buffer, " "); ptr != NULL; + ptr = (char *) strtok (NULL, " ")) { + if (!strcmp (ptr, "OFF")) + status |= UPSSTATUS_OFF; + else if (!strcmp (ptr, "OL")) + status |= UPSSTATUS_OL; + else if (!strcmp (ptr, "OB")) + status |= UPSSTATUS_OB; + else if (!strcmp (ptr, "LB")) + status |= UPSSTATUS_LB; + else if (!strcmp (ptr, "CAL")) + status |= UPSSTATUS_CAL; + else + status |= UPSSTATUS_UNKOWN; + } + + return OK; +} + + +/* determines what options are supported by the UPS */ +int +determine_supported_vars (void) +{ + char send_buffer[MAX_INPUT_BUFFER]; + char recv_buffer[MAX_INPUT_BUFFER]; + char temp_buffer[MAX_INPUT_BUFFER]; + char *ptr; + + + /* get the list of variables that this UPS supports */ + if (ups_name) + sprintf (send_buffer, "LISTVARS %s\r\n", ups_name); + else + sprintf (send_buffer, "LISTVARS\r\n"); + if (process_tcp_request + (server_address, server_port, send_buffer, recv_buffer, + sizeof (recv_buffer)) != STATE_OK) { + printf ("Invalid response received from host\n"); + return ERROR; + } + + recv_buffer[strlen (recv_buffer) - 1] = 0; + + if (ups_name) + ptr = recv_buffer + 5 + strlen (ups_name) + 2; + else + ptr = recv_buffer + 5; + + strcpy (temp_buffer, recv_buffer); + + for (ptr = (char *) strtok (temp_buffer, " "); ptr != NULL; + ptr = (char *) strtok (NULL, " ")) { + if (!strcmp (ptr, "UTILITY")) + supported_options |= UPS_UTILITY; + else if (!strcmp (ptr, "BATTPCT")) + supported_options |= UPS_BATTPCT; + else if (!strcmp (ptr, "LOADPCT")) + supported_options |= UPS_LOADPCT; + else if (!strcmp (ptr, "STATUS")) + supported_options |= UPS_STATUS; + else if (!strcmp (ptr, "UPSTEMP")) + supported_options |= UPS_TEMP; + } + + return OK; +} + + +/* gets a variable value for a specific UPS */ +int +get_ups_variable (const char *varname, char *buf, int buflen) +{ + /* char command[MAX_INPUT_BUFFER]; */ + char temp_buffer[MAX_INPUT_BUFFER]; + char send_buffer[MAX_INPUT_BUFFER]; + char *ptr; + + /* create the command string to send to the UPS daemon */ + if (ups_name) + sprintf (send_buffer, "REQ %s@%s\n", varname, ups_name); + else + sprintf (send_buffer, "REQ %s\n", varname); + + /* send the command to the daemon and get a response back */ + if (process_tcp_request + (server_address, server_port, send_buffer, temp_buffer, + sizeof (temp_buffer)) != STATE_OK) { + printf ("Invalid response received from host\n"); + return ERROR; + } + + if (ups_name) + ptr = temp_buffer + strlen (varname) + 5 + strlen (ups_name) + 1; + else + ptr = temp_buffer + strlen (varname) + 5; + + if (!strcmp (ptr, "NOT-SUPPORTED")) { + printf ("Error: Variable '%s' is not supported\n", varname); + return ERROR; + } + + if (!strcmp (ptr, "DATA-STALE")) { + printf ("Error: UPS data is stale\n"); + return ERROR; + } + + if (!strcmp (ptr, "UNKNOWN-UPS")) { + if (ups_name) + printf ("Error: UPS '%s' is unknown\n", ups_name); + else + printf ("Error: UPS is unknown\n"); + return ERROR; + } + + strncpy (buf, ptr, buflen - 1); + buf[buflen - 1] = 0; + + return OK; +} + + + + + +/* Command line: CHECK_UPS <host_address> [-u ups] [-p port] [-v variable] + [-wv warn_value] [-cv crit_value] [-to to_sec] */ + + +/* process command-line arguments */ +int +process_arguments (int argc, char **argv) +{ + int c; + + if (argc < 2) + return ERROR; + + for (c = 1; c < argc; c++) { + if (strcmp ("-to", argv[c]) == 0) + strcpy (argv[c], "-t"); + else if (strcmp ("-wt", argv[c]) == 0) + strcpy (argv[c], "-w"); + else if (strcmp ("-ct", argv[c]) == 0) + strcpy (argv[c], "-c"); + } + + c = 0; + while ((c += (call_getopt (argc - c, &argv[c]))) < argc) { + + if (is_option (argv[c])) + continue; + + if (server_address == NULL) { + if (is_host (argv[c])) { + server_address = argv[c]; + } + else { + usage ("Invalid host name"); + } + } + } + + if (server_address == NULL) + server_address = strscpy (NULL, "127.0.0.1"); + + return validate_arguments (); +} + + + + + + +int +call_getopt (int argc, char **argv) +{ + int c, i = 0; + +#ifdef HAVE_GETOPT_H + int option_index = 0; + static struct option long_options[] = { + {"hostname", required_argument, 0, 'H'}, + {"ups", required_argument, 0, 'u'}, + {"port", required_argument, 0, 'p'}, + {"critical", required_argument, 0, 'c'}, + {"warning", required_argument, 0, 'w'}, + {"timeout", required_argument, 0, 't'}, + {"variable", required_argument, 0, 'v'}, + {"version", no_argument, 0, 'V'}, + {"help", no_argument, 0, 'h'}, + {0, 0, 0, 0} + }; +#endif + + while (1) { +#ifdef HAVE_GETOPT_H + c = + getopt_long (argc, argv, "+hVH:u:p:v:c:w:t:", long_options, + &option_index); +#else + c = getopt (argc, argv, "+?hVH:u:p:v:c:w:t:"); +#endif + + i++; + + if (c == -1 || c == EOF || c == 1) + break; + + switch (c) { + case 'H': + case 'u': + case 'p': + case 'v': + case 'c': + case 'w': + case 't': + i++; + } + + switch (c) { + case '?': /* help */ + usage ("Invalid argument\n"); + case 'H': /* hostname */ + if (is_host (optarg)) { + server_address = optarg; + } + else { + usage ("Invalid host name\n"); + } + break; + case 'u': /* ups name */ + ups_name = optarg; + break; + case 'p': /* port */ + if (is_intpos (optarg)) { + server_port = atoi (optarg); + } + else { + usage ("Server port must be a positive integer\n"); + } + break; + case 'c': /* critical time threshold */ + if (is_intnonneg (optarg)) { + critical_value = atoi (optarg); + check_critical_value = TRUE; + } + else { + usage ("Critical time must be a nonnegative integer\n"); + } + break; + case 'w': /* warning time threshold */ + if (is_intnonneg (optarg)) { + warning_value = atoi (optarg); + check_warning_value = TRUE; + } + else { + usage ("Warning time must be a nonnegative integer\n"); + } + break; + case 'v': /* variable */ + if (!strcmp (optarg, "LINE")) + check_variable = UPS_UTILITY; + else if (!strcmp (optarg, "TEMP")) + check_variable = UPS_TEMP; + else if (!strcmp (optarg, "BATTPCT")) + check_variable = UPS_BATTPCT; + else if (!strcmp (optarg, "LOADPCT")) + check_variable = UPS_LOADPCT; + else + usage ("Unrecognized UPS variable\n"); + break; + case 't': /* timeout */ + if (is_intnonneg (optarg)) { + socket_timeout = atoi (optarg); + } + else { + usage ("Time interval must be a nonnegative integer\n"); + } + break; + case 'V': /* version */ + print_revision (PROGNAME, "$Revision$"); + exit (STATE_OK); + case 'h': /* help */ + print_help (); + exit (STATE_OK); + } + } + return i; +} + + + + + +int +validate_arguments (void) +{ + return OK; +} + + + + + +void +print_help (void) +{ + print_revision (PROGNAME, "$Revision$"); + printf + ("Copyright (c) 2000 Tom Shields/Karl DeBisschop\n\n" + "This plugin tests the UPS service on the specified host.\n\n"); + print_usage (); + printf + ("\nOptions:\n" + " -H, --hostname=STRING or IPADDRESS\n" + " Check server on the indicated host\n" + " -p, --port=INTEGER\n" + " Make connection on the indicated port (default: %d)\n" + " -u, --ups=STRING\n" + " Name of UPS\n" + " -w, --warning=INTEGER\n" + " Seconds necessary to result in a warning status\n" + " -c, --critical=INTEGER\n" + " Seconds necessary to result in a critical status\n" + " -t, --timeout=INTEGER\n" + " Seconds before connection attempt times out (default: %d)\n" + " -v, --verbose\n" + " Print extra information (command-line use only)\n" + " -h, --help\n" + " Print detailed help screen\n" + " -V, --version\n" + " Print version information\n\n", PORT, DEFAULT_SOCKET_TIMEOUT); + support (); +} + + + + + +void +print_usage (void) +{ + printf + ("Usage: %s -H host [-e expect] [-p port] [-w warn] [-c crit]\n" + " [-t timeout] [-v]\n" + " %s --help\n" + " %s --version\n", PROGNAME, PROGNAME, PROGNAME); +} |