[monitoring-plugins] Refactor check_mrtg
Lorenz Kästle
git at monitoring-plugins.org
Tue Mar 11 13:00:12 CET 2025
Module: monitoring-plugins
Branch: master
Commit: 23da18f10ce32c899f266ae703f0b9c2b3991d70
Author: Lorenz Kästle <12514511+RincewindsHat at users.noreply.github.com>
Date: Tue Mar 11 12:39:14 2025 +0100
URL: https://www.monitoring-plugins.org/repositories/monitoring-plugins/commit/?id=23da18f1
Refactor check_mrtg
---
plugins/Makefile.am | 1 +
plugins/check_mrtg.c | 141 ++++++++++++++++++++++--------------------
plugins/check_mrtg.d/config.h | 36 +++++++++++
3 files changed, 110 insertions(+), 68 deletions(-)
diff --git a/plugins/Makefile.am b/plugins/Makefile.am
index 9d310a15..5e636f9b 100644
--- a/plugins/Makefile.am
+++ b/plugins/Makefile.am
@@ -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 \
diff --git a/plugins/check_mrtg.c b/plugins/check_mrtg.c
index e355fa11..5bd276dc 100644
--- a/plugins/check_mrtg.c
+++ b/plugins/check_mrtg.c
@@ -35,21 +35,18 @@ const char *email = "devel at 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) {
diff --git a/plugins/check_mrtg.d/config.h b/plugins/check_mrtg.d/config.h
new file mode 100644
index 00000000..96b849a2
--- /dev/null
+++ b/plugins/check_mrtg.d/config.h
@@ -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;
+}
More information about the Commits
mailing list