mirror of
https://github.com/monitoring-plugins/monitoring-plugins.git
synced 2026-04-22 14:46:57 -04:00
Refactor check_mrtg
This commit is contained in:
Lorenz Kästle
parent
b3cb1bb45a
commit
23da18f10c
3 changed files with 110 additions and 68 deletions
|
|
@ -55,6 +55,7 @@ EXTRA_DIST = t \
|
|||
check_dbi.d \
|
||||
check_ssh.d \
|
||||
check_dns.d \
|
||||
check_mrtg.d \
|
||||
check_apt.d \
|
||||
check_by_ssh.d \
|
||||
check_smtp.d \
|
||||
|
|
|
|||
|
|
@ -35,21 +35,18 @@ const char *email = "devel@monitoring-plugins.org";
|
|||
|
||||
#include "common.h"
|
||||
#include "utils.h"
|
||||
#include "check_mrtg.d/config.h"
|
||||
|
||||
typedef struct {
|
||||
int errorcode;
|
||||
check_mrtg_config config;
|
||||
} check_mrtg_config_wrapper;
|
||||
static check_mrtg_config_wrapper process_arguments(int /*argc*/, char ** /*argv*/);
|
||||
static check_mrtg_config_wrapper validate_arguments(check_mrtg_config_wrapper /*config_wrapper*/);
|
||||
|
||||
static int process_arguments(int /*argc*/, char ** /*argv*/);
|
||||
static int validate_arguments(void);
|
||||
static void print_help(void);
|
||||
void print_usage(void);
|
||||
|
||||
static char *log_file = NULL;
|
||||
static int expire_minutes = 0;
|
||||
static bool use_average = true;
|
||||
static int variable_number = -1;
|
||||
static unsigned long value_warning_threshold = 0L;
|
||||
static unsigned long value_critical_threshold = 0L;
|
||||
static char *label;
|
||||
static char *units;
|
||||
|
||||
int main(int argc, char **argv) {
|
||||
setlocale(LC_ALL, "");
|
||||
bindtextdomain(PACKAGE, LOCALEDIR);
|
||||
|
|
@ -58,24 +55,26 @@ int main(int argc, char **argv) {
|
|||
/* Parse extra opts if any */
|
||||
argv = np_extra_opts(&argc, argv, progname);
|
||||
|
||||
if (process_arguments(argc, argv) == ERROR) {
|
||||
check_mrtg_config_wrapper tmp_config = process_arguments(argc, argv);
|
||||
if (tmp_config.errorcode == ERROR) {
|
||||
usage4(_("Could not parse arguments\n"));
|
||||
}
|
||||
|
||||
const check_mrtg_config config = tmp_config.config;
|
||||
|
||||
/* open the MRTG log file for reading */
|
||||
FILE *mtrg_log_file = fopen(log_file, "r");
|
||||
FILE *mtrg_log_file = fopen(config.log_file, "r");
|
||||
if (mtrg_log_file == NULL) {
|
||||
printf(_("Unable to open MRTG log file\n"));
|
||||
return STATE_UNKNOWN;
|
||||
}
|
||||
|
||||
time_t timestamp = 0L;
|
||||
unsigned long average_value_rate = 0L;
|
||||
unsigned long maximum_value_rate = 0L;
|
||||
time_t timestamp = 0;
|
||||
unsigned long average_value_rate = 0;
|
||||
unsigned long maximum_value_rate = 0;
|
||||
char input_buffer[MAX_INPUT_BUFFER];
|
||||
int line = 0;
|
||||
while (fgets(input_buffer, MAX_INPUT_BUFFER - 1, mtrg_log_file)) {
|
||||
|
||||
line++;
|
||||
|
||||
/* skip the first line of the log file */
|
||||
|
|
@ -94,25 +93,25 @@ int main(int argc, char **argv) {
|
|||
|
||||
/* grab the average value 1 rate */
|
||||
temp_buffer = strtok(NULL, " ");
|
||||
if (variable_number == 1) {
|
||||
if (config.variable_number == 1) {
|
||||
average_value_rate = strtoul(temp_buffer, NULL, 10);
|
||||
}
|
||||
|
||||
/* grab the average value 2 rate */
|
||||
temp_buffer = strtok(NULL, " ");
|
||||
if (variable_number == 2) {
|
||||
if (config.variable_number == 2) {
|
||||
average_value_rate = strtoul(temp_buffer, NULL, 10);
|
||||
}
|
||||
|
||||
/* grab the maximum value 1 rate */
|
||||
temp_buffer = strtok(NULL, " ");
|
||||
if (variable_number == 1) {
|
||||
if (config.variable_number == 1) {
|
||||
maximum_value_rate = strtoul(temp_buffer, NULL, 10);
|
||||
}
|
||||
|
||||
/* grab the maximum value 2 rate */
|
||||
temp_buffer = strtok(NULL, " ");
|
||||
if (variable_number == 2) {
|
||||
if (config.variable_number == 2) {
|
||||
maximum_value_rate = strtoul(temp_buffer, NULL, 10);
|
||||
}
|
||||
}
|
||||
|
|
@ -129,43 +128,49 @@ int main(int argc, char **argv) {
|
|||
/* make sure the MRTG data isn't too old */
|
||||
time_t current_time;
|
||||
time(¤t_time);
|
||||
if (expire_minutes > 0 && (current_time - timestamp) > (expire_minutes * 60)) {
|
||||
if (config.expire_minutes > 0 && (current_time - timestamp) > (config.expire_minutes * 60)) {
|
||||
printf(_("MRTG data has expired (%d minutes old)\n"), (int)((current_time - timestamp) / 60));
|
||||
return STATE_WARNING;
|
||||
}
|
||||
|
||||
unsigned long rate = 0L;
|
||||
/* else check the incoming/outgoing rates */
|
||||
if (use_average) {
|
||||
if (config.use_average) {
|
||||
rate = average_value_rate;
|
||||
} else {
|
||||
rate = maximum_value_rate;
|
||||
}
|
||||
|
||||
int result = STATE_OK;
|
||||
if (rate > value_critical_threshold) {
|
||||
if (config.value_critical_threshold_set && rate > config.value_critical_threshold) {
|
||||
result = STATE_CRITICAL;
|
||||
} else if (rate > value_warning_threshold) {
|
||||
} else if (config.value_warning_threshold_set && rate > config.value_warning_threshold) {
|
||||
result = STATE_WARNING;
|
||||
}
|
||||
|
||||
printf("%s. %s = %lu %s|%s\n", (use_average) ? _("Avg") : _("Max"), label, rate, units,
|
||||
perfdata(label, (long)rate, units, (int)value_warning_threshold, (long)value_warning_threshold, (int)value_critical_threshold,
|
||||
(long)value_critical_threshold, 0, 0, 0, 0));
|
||||
printf("%s. %s = %lu %s|%s\n", (config.use_average) ? _("Avg") : _("Max"), config.label, rate, config.units,
|
||||
perfdata(config.label, (long)rate, config.units, config.value_warning_threshold_set, (long)config.value_warning_threshold,
|
||||
config.value_critical_threshold_set, (long)config.value_critical_threshold, 0, 0, 0, 0));
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
/* process command-line arguments */
|
||||
int process_arguments(int argc, char **argv) {
|
||||
check_mrtg_config_wrapper process_arguments(int argc, char **argv) {
|
||||
static struct option longopts[] = {
|
||||
{"logfile", required_argument, 0, 'F'}, {"expires", required_argument, 0, 'e'}, {"aggregation", required_argument, 0, 'a'},
|
||||
{"variable", required_argument, 0, 'v'}, {"critical", required_argument, 0, 'c'}, {"warning", required_argument, 0, 'w'},
|
||||
{"label", required_argument, 0, 'l'}, {"units", required_argument, 0, 'u'}, {"variable", required_argument, 0, 'v'},
|
||||
{"version", no_argument, 0, 'V'}, {"help", no_argument, 0, 'h'}, {0, 0, 0, 0}};
|
||||
|
||||
check_mrtg_config_wrapper result = {
|
||||
.errorcode = OK,
|
||||
.config = check_mrtg_config_init(),
|
||||
};
|
||||
|
||||
if (argc < 2) {
|
||||
return ERROR;
|
||||
result.errorcode = ERROR;
|
||||
return result;
|
||||
}
|
||||
|
||||
for (int i = 1; i < argc; i++) {
|
||||
|
|
@ -189,35 +194,33 @@ int process_arguments(int argc, char **argv) {
|
|||
|
||||
switch (option_char) {
|
||||
case 'F': /* input file */
|
||||
log_file = optarg;
|
||||
result.config.log_file = optarg;
|
||||
break;
|
||||
case 'e': /* ups name */
|
||||
expire_minutes = atoi(optarg);
|
||||
result.config.expire_minutes = atoi(optarg);
|
||||
break;
|
||||
case 'a': /* port */
|
||||
if (!strcmp(optarg, "MAX")) {
|
||||
use_average = false;
|
||||
} else {
|
||||
use_average = true;
|
||||
}
|
||||
result.config.use_average = (bool)(strcmp(optarg, "MAX"));
|
||||
break;
|
||||
case 'v':
|
||||
variable_number = atoi(optarg);
|
||||
if (variable_number < 1 || variable_number > 2) {
|
||||
result.config.variable_number = atoi(optarg);
|
||||
if (result.config.variable_number < 1 || result.config.variable_number > 2) {
|
||||
usage4(_("Invalid variable number"));
|
||||
}
|
||||
break;
|
||||
case 'w': /* critical time threshold */
|
||||
value_warning_threshold = strtoul(optarg, NULL, 10);
|
||||
result.config.value_warning_threshold_set = true;
|
||||
result.config.value_warning_threshold = strtoul(optarg, NULL, 10);
|
||||
break;
|
||||
case 'c': /* warning time threshold */
|
||||
value_critical_threshold = strtoul(optarg, NULL, 10);
|
||||
result.config.value_critical_threshold_set = true;
|
||||
result.config.value_critical_threshold = strtoul(optarg, NULL, 10);
|
||||
break;
|
||||
case 'l': /* label */
|
||||
label = optarg;
|
||||
result.config.label = optarg;
|
||||
break;
|
||||
case 'u': /* timeout */
|
||||
units = optarg;
|
||||
result.config.units = optarg;
|
||||
break;
|
||||
case 'V': /* version */
|
||||
print_revision(progname, NP_VERSION);
|
||||
|
|
@ -231,67 +234,69 @@ int process_arguments(int argc, char **argv) {
|
|||
}
|
||||
|
||||
option_char = optind;
|
||||
if (log_file == NULL && argc > option_char) {
|
||||
log_file = argv[option_char++];
|
||||
if (result.config.log_file == NULL && argc > option_char) {
|
||||
result.config.log_file = argv[option_char++];
|
||||
}
|
||||
|
||||
if (expire_minutes <= 0 && argc > option_char) {
|
||||
if (result.config.expire_minutes <= 0 && argc > option_char) {
|
||||
if (is_intpos(argv[option_char])) {
|
||||
expire_minutes = atoi(argv[option_char++]);
|
||||
result.config.expire_minutes = atoi(argv[option_char++]);
|
||||
} else {
|
||||
die(STATE_UNKNOWN, _("%s is not a valid expiration time\nUse '%s -h' for additional help\n"), argv[option_char], progname);
|
||||
}
|
||||
}
|
||||
|
||||
if (argc > option_char && strcmp(argv[option_char], "MAX") == 0) {
|
||||
use_average = false;
|
||||
result.config.use_average = false;
|
||||
option_char++;
|
||||
} else if (argc > option_char && strcmp(argv[option_char], "AVG") == 0) {
|
||||
use_average = true;
|
||||
result.config.use_average = true;
|
||||
option_char++;
|
||||
}
|
||||
|
||||
if (argc > option_char && variable_number == -1) {
|
||||
variable_number = atoi(argv[option_char++]);
|
||||
if (variable_number < 1 || variable_number > 2) {
|
||||
if (argc > option_char && result.config.variable_number == -1) {
|
||||
result.config.variable_number = atoi(argv[option_char++]);
|
||||
if (result.config.variable_number < 1 || result.config.variable_number > 2) {
|
||||
printf("%s :", argv[option_char]);
|
||||
usage(_("Invalid variable number\n"));
|
||||
}
|
||||
}
|
||||
|
||||
if (argc > option_char && value_warning_threshold == 0) {
|
||||
value_warning_threshold = strtoul(argv[option_char++], NULL, 10);
|
||||
if (argc > option_char && !result.config.value_warning_threshold_set) {
|
||||
result.config.value_warning_threshold_set = true;
|
||||
result.config.value_warning_threshold = strtoul(argv[option_char++], NULL, 10);
|
||||
}
|
||||
|
||||
if (argc > option_char && value_critical_threshold == 0) {
|
||||
value_critical_threshold = strtoul(argv[option_char++], NULL, 10);
|
||||
if (argc > option_char && !result.config.value_critical_threshold_set) {
|
||||
result.config.value_critical_threshold_set = true;
|
||||
result.config.value_critical_threshold = strtoul(argv[option_char++], NULL, 10);
|
||||
}
|
||||
|
||||
if (argc > option_char && strlen(label) == 0) {
|
||||
label = argv[option_char++];
|
||||
if (argc > option_char && strlen(result.config.label) == 0) {
|
||||
result.config.label = argv[option_char++];
|
||||
}
|
||||
|
||||
if (argc > option_char && strlen(units) == 0) {
|
||||
units = argv[option_char++];
|
||||
if (argc > option_char && strlen(result.config.units) == 0) {
|
||||
result.config.units = argv[option_char++];
|
||||
}
|
||||
|
||||
return validate_arguments();
|
||||
return validate_arguments(result);
|
||||
}
|
||||
|
||||
int validate_arguments(void) {
|
||||
if (variable_number == -1) {
|
||||
check_mrtg_config_wrapper validate_arguments(check_mrtg_config_wrapper config_wrapper) {
|
||||
if (config_wrapper.config.variable_number == -1) {
|
||||
usage4(_("You must supply the variable number"));
|
||||
}
|
||||
|
||||
if (label == NULL) {
|
||||
label = strdup("value");
|
||||
if (config_wrapper.config.label == NULL) {
|
||||
config_wrapper.config.label = strdup("value");
|
||||
}
|
||||
|
||||
if (units == NULL) {
|
||||
units = strdup("");
|
||||
if (config_wrapper.config.units == NULL) {
|
||||
config_wrapper.config.units = strdup("");
|
||||
}
|
||||
|
||||
return OK;
|
||||
return config_wrapper;
|
||||
}
|
||||
|
||||
void print_help(void) {
|
||||
|
|
|
|||
36
plugins/check_mrtg.d/config.h
Normal file
36
plugins/check_mrtg.d/config.h
Normal file
|
|
@ -0,0 +1,36 @@
|
|||
#pragma once
|
||||
|
||||
#include "../../config.h"
|
||||
#include <stddef.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
typedef struct {
|
||||
bool use_average;
|
||||
int variable_number;
|
||||
int expire_minutes;
|
||||
char *label;
|
||||
char *units;
|
||||
char *log_file;
|
||||
|
||||
bool value_warning_threshold_set;
|
||||
unsigned long value_warning_threshold;
|
||||
bool value_critical_threshold_set;
|
||||
unsigned long value_critical_threshold;
|
||||
} check_mrtg_config;
|
||||
|
||||
check_mrtg_config check_mrtg_config_init() {
|
||||
check_mrtg_config tmp = {
|
||||
.use_average = true,
|
||||
.variable_number = -1,
|
||||
.expire_minutes = 0,
|
||||
.label = NULL,
|
||||
.units = NULL,
|
||||
.log_file = NULL,
|
||||
|
||||
.value_warning_threshold_set = false,
|
||||
.value_warning_threshold = 0,
|
||||
.value_critical_threshold_set = false,
|
||||
.value_critical_threshold = 0,
|
||||
};
|
||||
return tmp;
|
||||
}
|
||||
Loading…
Reference in a new issue