check_mrtg: implement modern output

This commit is contained in:
Lorenz Kästle 2025-11-26 14:35:21 +01:00
parent 0378484a57
commit 326d399624
2 changed files with 110 additions and 45 deletions

View file

@ -29,14 +29,18 @@
*
*****************************************************************************/
#include "common.h"
#include "output.h"
#include "perfdata.h"
#include "states.h"
#include "thresholds.h"
#include "utils.h"
#include "check_mrtg.d/config.h"
const char *progname = "check_mrtg";
const char *copyright = "1999-2024";
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;
@ -62,11 +66,24 @@ int main(int argc, char **argv) {
const check_mrtg_config config = tmp_config.config;
if (config.output_format_is_set) {
mp_set_format(config.output_format);
}
mp_check overall = mp_check_init();
/* open the MRTG log file for reading */
mp_subcheck sc_open_mrtg_log_file = mp_subcheck_init();
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;
xasprintf(&sc_open_mrtg_log_file.output, "unable to open MRTG log file");
sc_open_mrtg_log_file = mp_set_subcheck_state(sc_open_mrtg_log_file, STATE_UNKNOWN);
mp_add_subcheck_to_check(&overall, sc_open_mrtg_log_file);
mp_exit(overall);
} else {
xasprintf(&sc_open_mrtg_log_file.output, "opened MRTG log file");
sc_open_mrtg_log_file = mp_set_subcheck_state(sc_open_mrtg_log_file, STATE_OK);
mp_add_subcheck_to_check(&overall, sc_open_mrtg_log_file);
}
time_t timestamp = 0;
@ -120,18 +137,32 @@ int main(int argc, char **argv) {
fclose(mtrg_log_file);
/* if we couldn't read enough data, return an unknown error */
mp_subcheck sc_process_mrtg_log_file = mp_subcheck_init();
if (line <= 2) {
printf(_("Unable to process MRTG log file\n"));
return STATE_UNKNOWN;
xasprintf(&sc_process_mrtg_log_file.output, "unable to process MRTG log file");
sc_process_mrtg_log_file = mp_set_subcheck_state(sc_process_mrtg_log_file, STATE_UNKNOWN);
mp_exit(overall);
} else {
xasprintf(&sc_process_mrtg_log_file.output, "processed MRTG log file");
sc_process_mrtg_log_file = mp_set_subcheck_state(sc_process_mrtg_log_file, STATE_OK);
mp_add_subcheck_to_check(&overall, sc_process_mrtg_log_file);
}
/* make sure the MRTG data isn't too old */
time_t current_time;
time(&current_time);
mp_subcheck sc_data_expired = mp_subcheck_init();
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;
xasprintf(&sc_data_expired.output, "MRTG data has expired (%d minutes old)",
(int)((current_time - timestamp) / 60));
sc_data_expired = mp_set_subcheck_state(sc_data_expired, STATE_WARNING);
mp_add_subcheck_to_check(&overall, sc_data_expired);
mp_exit(overall);
} else {
xasprintf(&sc_data_expired.output, "MRTG data should be valid (%d minutes old)",
(int)((current_time - timestamp) / 60));
sc_data_expired = mp_set_subcheck_state(sc_data_expired, STATE_OK);
mp_add_subcheck_to_check(&overall, sc_data_expired);
}
unsigned long rate = 0L;
@ -142,24 +173,27 @@ int main(int argc, char **argv) {
rate = maximum_value_rate;
}
int result = STATE_OK;
if (config.value_critical_threshold_set && rate > config.value_critical_threshold) {
result = STATE_CRITICAL;
} else if (config.value_warning_threshold_set && rate > config.value_warning_threshold) {
result = STATE_WARNING;
}
mp_subcheck sc_values = mp_subcheck_init();
mp_perfdata pd_value = perfdata_init();
pd_value = mp_set_pd_value(pd_value, rate);
pd_value.label = config.label;
pd_value = mp_pd_set_thresholds(pd_value, config.values_threshold);
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));
sc_values = mp_set_subcheck_state(sc_values, mp_get_pd_status(pd_value));
xasprintf(&sc_values.output, "%s. %s = %lu %s", (config.use_average) ? _("Avg") : _("Max"),
config.label, rate, config.units);
return result;
mp_add_subcheck_to_check(&overall, sc_values);
mp_exit(overall);
}
/* process command-line arguments */
check_mrtg_config_wrapper process_arguments(int argc, char **argv) {
enum {
output_format_index,
};
static struct option longopts[] = {{"logfile", required_argument, 0, 'F'},
{"expires", required_argument, 0, 'e'},
{"aggregation", required_argument, 0, 'a'},
@ -171,6 +205,7 @@ check_mrtg_config_wrapper process_arguments(int argc, char **argv) {
{"variable", required_argument, 0, 'v'},
{"version", no_argument, 0, 'V'},
{"help", no_argument, 0, 'h'},
{"output-format", required_argument, 0, output_format_index},
{0, 0, 0, 0}};
check_mrtg_config_wrapper result = {
@ -218,14 +253,22 @@ check_mrtg_config_wrapper process_arguments(int argc, char **argv) {
usage4(_("Invalid variable number"));
}
break;
case 'w': /* critical time threshold */
result.config.value_warning_threshold_set = true;
result.config.value_warning_threshold = strtoul(optarg, NULL, 10);
break;
case 'c': /* warning time threshold */
result.config.value_critical_threshold_set = true;
result.config.value_critical_threshold = strtoul(optarg, NULL, 10);
break;
case 'w': /* critical time threshold */ {
mp_range_parsed tmp = mp_parse_range_string(optarg);
if (tmp.error != MP_PARSING_SUCCES) {
die(STATE_UNKNOWN, "failed to parse warning threshold");
}
result.config.values_threshold =
mp_thresholds_set_warn(result.config.values_threshold, tmp.range);
} break;
case 'c': /* warning time threshold */ {
mp_range_parsed tmp = mp_parse_range_string(optarg);
if (tmp.error != MP_PARSING_SUCCES) {
die(STATE_UNKNOWN, "failed to parse critical threshold");
}
result.config.values_threshold =
mp_thresholds_set_crit(result.config.values_threshold, tmp.range);
} break;
case 'l': /* label */
result.config.label = optarg;
break;
@ -240,6 +283,17 @@ check_mrtg_config_wrapper process_arguments(int argc, char **argv) {
exit(STATE_UNKNOWN);
case '?': /* help */
usage5();
case output_format_index: {
parsed_output_format parser = mp_parse_output_format(optarg);
if (!parser.parsing_success) {
printf("Invalid output format: %s\n", optarg);
exit(STATE_UNKNOWN);
}
result.config.output_format_is_set = true;
result.config.output_format = parser.output_format;
break;
}
}
}
@ -274,14 +328,22 @@ check_mrtg_config_wrapper process_arguments(int argc, char **argv) {
}
}
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 && !result.config.values_threshold.warning_is_set) {
mp_range_parsed tmp = mp_parse_range_string(argv[option_char++]);
if (tmp.error != MP_PARSING_SUCCES) {
die(STATE_UNKNOWN, "failed to parse warning threshold");
}
result.config.values_threshold =
mp_thresholds_set_warn(result.config.values_threshold, tmp.range);
}
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 && !result.config.values_threshold.critical_is_set) {
mp_range_parsed tmp = mp_parse_range_string(argv[option_char++]);
if (tmp.error != MP_PARSING_SUCCES) {
die(STATE_UNKNOWN, "failed to parse critical threshold");
}
result.config.values_threshold =
mp_thresholds_set_crit(result.config.values_threshold, tmp.range);
}
if (argc > option_char && strlen(result.config.label) == 0) {
@ -345,6 +407,8 @@ void print_help(void) {
printf(" %s\n", _("Option units label for data (Example: Packets/Sec, Errors/Sec,"));
printf(" %s\n", _("\"Bytes Per Second\", \"%% Utilization\")"));
printf(UT_OUTPUT_FORMAT);
printf("\n");
printf(" %s\n",
_("If the value exceeds the <vwl> threshold, a WARNING status is returned. If"));

View file

@ -1,6 +1,8 @@
#pragma once
#include "../../config.h"
#include "output.h"
#include "thresholds.h"
#include <stddef.h>
#include <stdlib.h>
@ -12,10 +14,10 @@ typedef struct {
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;
mp_thresholds values_threshold;
bool output_format_is_set;
mp_output_format output_format;
} check_mrtg_config;
check_mrtg_config check_mrtg_config_init() {
@ -27,10 +29,9 @@ check_mrtg_config check_mrtg_config_init() {
.units = NULL,
.log_file = NULL,
.value_warning_threshold_set = false,
.value_warning_threshold = 0,
.value_critical_threshold_set = false,
.value_critical_threshold = 0,
.values_threshold = mp_thresholds_init(),
.output_format_is_set = false,
};
return tmp;
}