1 files changed, 122 insertions, 84 deletions

diff --git a/plugins/check_cluster.c b/plugins/check_cluster.c
index b40c38c7..1cbdcd60 100644
--- a/plugins/check_cluster.c
+++ b/plugins/check_cluster.c
@@ -26,45 +26,25 @@ const char *progname = "check_cluster";
 const char *copyright = "2000-2024";
 const char *email = "devel@monitoring-plugins.org";
 
+#include "output.h"
+#include "states.h"
 #include "common.h"
 #include "utils.h"
 #include "utils_base.h"
-
-enum {
-        CHECK_SERVICES = 1,
-        CHECK_HOSTS = 2
-};
+#include "check_cluster.d/config.h"
 
 static void print_help(void);
 void print_usage(void);
 
-static int total_services_ok = 0;
-static int total_services_warning = 0;
-static int total_services_unknown = 0;
-static int total_services_critical = 0;
-
-static int total_hosts_up = 0;
-static int total_hosts_down = 0;
-static int total_hosts_unreachable = 0;
-
-static char *warn_threshold;
-static char *crit_threshold;
-
-static int check_type = CHECK_SERVICES;
-
-static char *data_vals = NULL;
-static char *label = NULL;
-
 static int verbose = 0;
 
-static int process_arguments(int /*argc*/, char ** /*argv*/);
+typedef struct {
+        int errorcode;
+        check_cluster_config config;
+} check_cluster_config_wrapper;
+static check_cluster_config_wrapper process_arguments(int /*argc*/, char ** /*argv*/);
 
 int main(int argc, char **argv) {
-        char *ptr;
-        int data_val;
-        int return_code = STATE_OK;
-        thresholds *thresholds = NULL;
-
         setlocale(LC_ALL, "");
         bindtextdomain(PACKAGE, LOCALEDIR);
         textdomain(PACKAGE);
@@ -72,20 +52,35 @@ 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_cluster_config_wrapper tmp_config = process_arguments(argc, argv);
+        if (tmp_config.errorcode == ERROR) {
                 usage(_("Could not parse arguments"));
+        }
+
+        const check_cluster_config config = tmp_config.config;
+
+        if (config.output_format_is_set) {
+                mp_set_format(config.output_format);
+        }
 
         /* Initialize the thresholds */
-        set_thresholds(&thresholds, warn_threshold, crit_threshold);
-        if (verbose)
-                print_thresholds("check_cluster", thresholds);
+        if (verbose) {
+                print_thresholds("check_cluster", config.thresholds);
+        }
 
+        int data_val;
+        int total_services_ok = 0;
+        int total_services_warning = 0;
+        int total_services_unknown = 0;
+        int total_services_critical = 0;
+        int total_hosts_up = 0;
+        int total_hosts_down = 0;
+        int total_hosts_unreachable = 0;
         /* check the data values */
-        for (ptr = strtok(data_vals, ","); ptr != NULL; ptr = strtok(NULL, ",")) {
-
+        for (char *ptr = strtok(config.data_vals, ","); ptr != NULL; ptr = strtok(NULL, ",")) {
                 data_val = atoi(ptr);
 
-                if (check_type == CHECK_SERVICES) {
+                if (config.check_type == CHECK_SERVICES) {
                         switch (data_val) {
                         case 0:
                                 total_services_ok++;
@@ -119,101 +114,141 @@ int main(int argc, char **argv) {
                 }
         }
 
+        mp_check overall = mp_check_init();
+        mp_subcheck sc_real_test = mp_subcheck_init();
+        sc_real_test = mp_set_subcheck_default_state(sc_real_test, STATE_OK);
+
         /* return the status of the cluster */
-        if (check_type == CHECK_SERVICES) {
-                return_code = get_status(total_services_warning + total_services_unknown + total_services_critical, thresholds);
-                printf("CLUSTER %s: %s: %d ok, %d warning, %d unknown, %d critical\n", state_text(return_code),
-                           (label == NULL) ? "Service cluster" : label, total_services_ok, total_services_warning, total_services_unknown,
-                           total_services_critical);
+        if (config.check_type == CHECK_SERVICES) {
+                sc_real_test = mp_set_subcheck_state(
+                        sc_real_test,
+                        get_status(total_services_warning + total_services_unknown + total_services_critical,
+                                           config.thresholds));
+                xasprintf(&sc_real_test.output, "%s: %d ok, %d warning, %d unknown, %d critical",
+                                  (config.label == NULL) ? "Service cluster" : config.label, total_services_ok,
+                                  total_services_warning, total_services_unknown, total_services_critical);
         } else {
-                return_code = get_status(total_hosts_down + total_hosts_unreachable, thresholds);
-                printf("CLUSTER %s: %s: %d up, %d down, %d unreachable\n", state_text(return_code), (label == NULL) ? "Host cluster" : label,
-                           total_hosts_up, total_hosts_down, total_hosts_unreachable);
+                sc_real_test = mp_set_subcheck_state(
+                        sc_real_test,
+                        get_status(total_hosts_down + total_hosts_unreachable, config.thresholds));
+                xasprintf(&sc_real_test.output, "%s: %d up, %d down, %d unreachable\n",
+                                  (config.label == NULL) ? "Host cluster" : config.label, total_hosts_up,
+                                  total_hosts_down, total_hosts_unreachable);
         }
 
-        return return_code;
+        mp_add_subcheck_to_check(&overall, sc_real_test);
+
+        mp_exit(overall);
 }
 
-int process_arguments(int argc, char **argv) {
-        int c;
-        char *ptr;
-        int option = 0;
-        static struct option longopts[] = {{"data", required_argument, 0, 'd'},     {"warning", required_argument, 0, 'w'},
-                                                                           {"critical", required_argument, 0, 'c'}, {"label", required_argument, 0, 'l'},
-                                                                           {"host", no_argument, 0, 'h'},           {"service", no_argument, 0, 's'},
-                                                                           {"verbose", no_argument, 0, 'v'},        {"version", no_argument, 0, 'V'},
-                                                                           {"help", no_argument, 0, 'H'},           {0, 0, 0, 0}};
+check_cluster_config_wrapper process_arguments(int argc, char **argv) {
+        enum {
+                output_format_index = CHAR_MAX + 1,
+        };
+
+        static struct option longopts[] = {{"data", required_argument, 0, 'd'},
+                                                                           {"warning", required_argument, 0, 'w'},
+                                                                           {"critical", required_argument, 0, 'c'},
+                                                                           {"label", required_argument, 0, 'l'},
+                                                                           {"host", no_argument, 0, 'h'},
+                                                                           {"service", no_argument, 0, 's'},
+                                                                           {"verbose", no_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_cluster_config_wrapper result = {
+                .errorcode = OK,
+                .config = check_cluster_config_init(),
+        };
 
         /* no options were supplied */
-        if (argc < 2)
-                return ERROR;
-
-        while (1) {
+        if (argc < 2) {
+                result.errorcode = ERROR;
+                return result;
+        }
 
-                c = getopt_long(argc, argv, "hHsvVw:c:d:l:", longopts, &option);
+        int option = 0;
+        char *warn_threshold = NULL;
+        char *crit_threshold = NULL;
+        while (true) {
+                int option_index = getopt_long(argc, argv, "hHsvVw:c:d:l:", longopts, &option);
 
-                if (c == -1 || c == EOF || c == 1)
+                if (option_index == -1 || option_index == EOF || option_index == 1) {
                         break;
+                }
 
-                switch (c) {
-
+                switch (option_index) {
                 case 'h': /* host cluster */
-                        check_type = CHECK_HOSTS;
+                        result.config.check_type = CHECK_HOSTS;
                         break;
-
                 case 's': /* service cluster */
-                        check_type = CHECK_SERVICES;
+                        result.config.check_type = CHECK_SERVICES;
                         break;
-
                 case 'w': /* warning threshold */
                         warn_threshold = strdup(optarg);
                         break;
-
                 case 'c': /* warning threshold */
                         crit_threshold = strdup(optarg);
                         break;
-
                 case 'd': /* data values */
-                        data_vals = (char *)strdup(optarg);
+                        result.config.data_vals = strdup(optarg);
                         /* validate data */
-                        for (ptr = data_vals; ptr != NULL; ptr += 2) {
-                                if (ptr[0] < '0' || ptr[0] > '3')
-                                        return ERROR;
-                                if (ptr[1] == '\0')
+                        for (char *ptr = result.config.data_vals; ptr != NULL; ptr += 2) {
+                                if (ptr[0] < '0' || ptr[0] > '3') {
+                                        result.errorcode = ERROR;
+                                        return result;
+                                }
+                                if (ptr[1] == '\0') {
                                         break;
-                                if (ptr[1] != ',')
-                                        return ERROR;
+                                }
+                                if (ptr[1] != ',') {
+                                        result.errorcode = ERROR;
+                                        return result;
+                                }
                         }
                         break;
-
                 case 'l': /* text label */
-                        label = (char *)strdup(optarg);
+                        result.config.label = strdup(optarg);
                         break;
-
                 case 'v': /* verbose */
                         verbose++;
                         break;
-
                 case 'V': /* version */
                         print_revision(progname, NP_VERSION);
                         exit(STATE_UNKNOWN);
                         break;
-
                 case 'H': /* help */
                         print_help();
                         exit(STATE_UNKNOWN);
                         break;
+                case output_format_index: {
+                        parsed_output_format parser = mp_parse_output_format(optarg);
+                        if (!parser.parsing_success) {
+                                // TODO List all available formats here, maybe add anothoer usage function
+                                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;
+                }
                 default:
-                        return ERROR;
+                        result.errorcode = ERROR;
+                        return result;
                         break;
                 }
         }
 
-        if (data_vals == NULL)
-                return ERROR;
+        if (result.config.data_vals == NULL) {
+                result.errorcode = ERROR;
+                return result;
+        }
 
-        return OK;
+        set_thresholds(&result.config.thresholds, warn_threshold, crit_threshold);
+        return result;
 }
 
 void print_help(void) {
@@ -247,6 +282,8 @@ void print_help(void) {
 
         printf(UT_VERBOSE);
 
+        printf(UT_OUTPUT_FORMAT);
+
         printf("\n");
         printf("%s\n", _("Notes:"));
         printf(UT_THRESHOLDS_NOTES);
@@ -254,7 +291,8 @@ void print_help(void) {
         printf("\n");
         printf("%s\n", _("Examples:"));
         printf(" %s\n", "check_cluster -s -d 2,0,2,0 -c @3:");
-        printf("    %s\n", _("Will alert critical if there are 3 or more service data points in a non-OK"));
+        printf("    %s\n",
+                   _("Will alert critical if there are 3 or more service data points in a non-OK"));
         printf("    %s\n", _("state."));
 
         printf(UT_SUPPORT);