Code

String match in check_ping for Solaris 10 #1091043
[nagiosplug.git] / plugins / check_disk.c
index c758530e71c33ac5a307ef481d149791186b2903..ab9f66f8ba4dcafb68d31bc33073ae5785394c58 100644 (file)
@@ -1,24 +1,27 @@
 /******************************************************************************
-*
-* 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.
-*
+
+ 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.
+
+ $Id$
 *****************************************************************************/
 
 const char *progname = "check_disk";
+const char *program_name = "check_disk";       // Required for coreutils libs
 const char *revision = "$Revision$";
-const char *copyright = "1999-2003";
+const char *copyright = "1999-2004";
 const char *email = "nagiosplug-devel@lists.sourceforge.net";
 
 #include "common.h"
@@ -35,80 +38,6 @@ const char *email = "nagiosplug-devel@lists.sourceforge.net";
 # include <limits.h>
 #endif
 
-void
-print_usage (void)
-{
-       printf (_("\
-Usage: %s -w limit -c limit [-p path | -x device] [-t timeout] [-m] [-e]\n\
-        [-v] [-q]\n\
-       %s (-h|--help)\n\
-       %s (-V|--version)\n"),
-               progname,  progname, progname);
-}
-
-void
-print_help (void)
-{
-       print_revision (progname, revision);
-
-       printf (_(COPYRIGHT), copyright, email);
-
-       printf (_("\
-This plugin checks the amount of used disk space on a mounted file system\n\
-and generates an alert if free space is less than one of the threshold values."));
-
-       print_usage ();
-
-       printf (_(UT_HELP_VRSN));
-
-       printf (_("\
- -w, --warning=INTEGER\n\
-   Exit with WARNING status if less than INTEGER kilobytes of disk are free\n\
- -w, --warning=PERCENT%%\n\
-   Exit with WARNING status if less than PERCENT of disk space is free\n\
- -c, --critical=INTEGER\n\
-   Exit with CRITICAL status if less than INTEGER kilobytes of disk are free\n\
- -c, --critical=PERCENT%%\n\
-   Exit with CRITCAL status if less than PERCENT of disk space is free\n\
- -C, --clear\n\
-    Clear thresholds\n"));
-
-       printf (_("\
- -u, --units=STRING\n\
-    Choose bytes, kB, MB, GB, TB (default: MB)\n\
- -k, --kilobytes\n\
-    Same as '--units kB'\n\
- -m, --megabytes\n\
-    Same as '--units MB'\n"));
-
-       printf (_("\
- -l, --local\n\
-    Only check local filesystems\n\
- -p, --path=PATH, --partition=PARTITION\n\
-    Path or partition (may be repeated)\n\
- -x, --exclude_device=PATH <STRING>\n\
-    Ignore device (only works if -p unspecified)\n\
- -X, --exclude-type=TYPE <STRING>\n\
-    Ignore all filesystems of indicated type (may be repeated)\n\
- -M, --mountpoint\n\
-    Display the mountpoint instead of the partition\n\
- -e, --errors-only\n\
-    Display only devices/mountpoints with errors\n"));
-
-       printf (_(UT_WARN_CRIT));
-
-       printf (_(UT_TIMEOUT), DEFAULT_SOCKET_TIMEOUT);
-
-       printf (_(UT_VERBOSE));
-
-       printf ("%s", _("Examples:\n\
- check_disk -w 10% -c 5% -p /tmp -p /var -C -w 100000 -c 50000 -p /\n\
-   Checks /tmp and /var at 10%,5% and / at 100MB, 50MB\n"));
-
-       support ();
-}
-\f
-
 /* If nonzero, show inode information. */
 /* static int inode_format; */
 
@@ -135,8 +64,8 @@ struct name_list
 {
   char *name;
   int found;
-  int w_df;
-  int c_df;
+  uintmax_t w_df;
+  uintmax_t c_df;
   double w_dfp;
   double c_dfp;
   struct name_list *name_next;
@@ -183,18 +112,21 @@ enum
 #endif
 
 int process_arguments (int, char **);
-int validate_arguments (int, int, double, double, char *);
-int check_disk (int usp, int free_disk);
+void print_path (const char *mypath);
+int validate_arguments (uintmax_t, uintmax_t, double, double, char *);
+int check_disk (double usp, double free_disk);
 int walk_name_list (struct name_list *list, const char *name);
+void print_help (void);
+void print_usage (void);
 
-int w_df = -1;
-int c_df = -1;
+uintmax_t w_df = 0;
+uintmax_t c_df = 0;
 double w_dfp = -1.0;
 double c_dfp = -1.0;
-char *path = "";
-char *exclude_device = "";
-char *units = "MB";
-unsigned long mult = 1024 * 1024;
+char *path;
+char *exclude_device;
+char *units;
+uintmax_t mult = 1024 * 1024;
 int verbose = 0;
 int erronly = FALSE;
 int display_mntp = FALSE;
@@ -202,25 +134,37 @@ int display_mntp = FALSE;
 /* Linked list of mounted filesystems. */
 static struct mount_entry *mount_list;
 
+
+
 int
 main (int argc, char **argv)
 {
-       int usp = -1;
+       double usp = -1.0;
        int result = STATE_UNKNOWN;
        int disk_result = STATE_UNKNOWN;
        char file_system[MAX_INPUT_BUFFER];
-       char *output = "";
-       char *details = "";
+       char *output;
+       char *details;
+       char *perf;
+       uintmax_t psize;
        float free_space, free_space_pct, total_space;
 
        struct mount_entry *me;
        struct fs_usage fsp;
        struct name_list *temp_list;
 
+       output = strdup (" - free space:");
+       details = strdup ("");
+       perf = strdup ("");
+
+       setlocale (LC_ALL, "");
+       bindtextdomain (PACKAGE, LOCALEDIR);
+       textdomain (PACKAGE);
+
        mount_list = read_filesystem_list (0);
 
-       if (process_arguments (argc, argv) != OK)
-               usage (_("Could not parse arguments\n"));
+       if (process_arguments (argc, argv) == ERROR)
+               usage4 (_("Could not parse arguments"));
 
        for (me = mount_list; me; me = me->me_next) {
 
@@ -244,9 +188,17 @@ main (int argc, char **argv)
                        get_fs_usage (me->me_mountdir, me->me_devname, &fsp);
 
                if (fsp.fsu_blocks && strcmp ("none", me->me_mountdir)) {
-                       usp = (fsp.fsu_blocks - fsp.fsu_bavail) * 100 / fsp.fsu_blocks;
-                       disk_result = check_disk (usp, fsp.fsu_bavail);
+                       usp = (double)(fsp.fsu_blocks - fsp.fsu_bavail) * 100 / fsp.fsu_blocks;
+                       disk_result = check_disk (usp, (double)(fsp.fsu_bavail * fsp.fsu_blocksize / mult));
                        result = max_state (disk_result, result);
+                       psize = fsp.fsu_blocks*fsp.fsu_blocksize/mult;
+                       asprintf (&perf, "%s %s", perf,
+                                 perfdata ((!strcmp(file_system, "none") || display_mntp) ? me->me_devname : me->me_mountdir,
+                                           psize-(fsp.fsu_bavail*fsp.fsu_blocksize/mult), units,
+                                           TRUE, min ((uintmax_t)psize-(uintmax_t)w_df, (uintmax_t)((1.0-w_dfp/100.0)*psize)),
+                                           TRUE, min ((uintmax_t)psize-(uintmax_t)c_df, (uintmax_t)((1.0-c_dfp/100.0)*psize)),
+                                           TRUE, 0,
+                                           TRUE, psize));
                        if (disk_result==STATE_OK && erronly && !verbose)
                                continue;
 
@@ -254,26 +206,23 @@ main (int argc, char **argv)
                        free_space_pct = (float)fsp.fsu_bavail*100/fsp.fsu_blocks;
                        total_space = (float)fsp.fsu_blocks*fsp.fsu_blocksize/mult;
                        if (disk_result!=STATE_OK || verbose>=0)
-                               asprintf (&output, ("%s [%.0f %s (%.0f%%) free on %s]"),
+                               asprintf (&output, ("%s %s %.0f %s (%.0f%%);"),
                                          output,
+                                         (!strcmp(file_system, "none") || display_mntp) ? me->me_devname : me->me_mountdir,
                                          free_space,
                                          units,
-                                         free_space_pct,
-                                         (!strcmp(file_system, "none") || display_mntp) ? me->me_devname : me->me_mountdir);
-                       asprintf (&details, _("%s\n%.0f of %.0f %s (%.0f%%) free on %s (type %s mounted on %s) warn:%d crit:%d warn%%:%.0f%% crit%%:%.0f%%"),
-                                 details,
-                                 free_space,
-                                 total_space,
-                                 units,
-                                 free_space_pct,
-                                 me->me_devname,
-                                 me->me_type,
-                                 me->me_mountdir,
-                                 w_df, c_df, w_dfp, c_dfp);
+                                                                       free_space_pct);
+                       asprintf (&details, _("%s\n\
+%.0f of %.0f %s (%.0f%%) free on %s (type %s mounted on %s) warn:%lu crit:%lu warn%%:%.0f%% crit%%:%.0f%%"),
+                                 details, free_space, total_space, units, free_space_pct,
+                                 me->me_devname, me->me_type, me->me_mountdir,
+                                 (unsigned long)w_df, (unsigned long)c_df, w_dfp, c_dfp);
                }
 
        }
 
+       asprintf (&output, "%s|%s", output, perf);
+
        if (verbose > 2)
                asprintf (&output, "%s%s", output, details);
 
@@ -287,13 +236,12 @@ main (int argc, char **argv)
                temp_list = temp_list->name_next;
        }
 
-       terminate (result, "DISK %s%s\n", state_text (result), output, details);
-       return STATE_UNKNOWN;
+       printf ("DISK %s%s\n", state_text (result), output);
+       return result;
 }
 
 
 
-\f
 /* process command-line arguments */
 int
 process_arguments (int argc, char **argv)
@@ -306,8 +254,10 @@ process_arguments (int argc, char **argv)
        struct name_list *temp_list;
        int result = OK;
 
-       int option_index = 0;
-       static struct option long_options[] = {
+       unsigned long l;
+
+       int option = 0;
+       static struct option longopts[] = {
                {"timeout", required_argument, 0, 't'},
                {"warning", required_argument, 0, 'w'},
                {"critical", required_argument, 0, 'c'},
@@ -343,7 +293,7 @@ process_arguments (int argc, char **argv)
                        strcpy (argv[c], "-t");
 
        while (1) {
-               c = getopt_long (argc, argv, "+?VqhveCt:c:w:u:p:x:X:mklM", long_options, &option_index);
+               c = getopt_long (argc, argv, "+?VqhveCt:c:w:u:p:x:X:mklM", longopts, &option);
 
                if (c == -1 || c == EOF)
                        break;
@@ -355,7 +305,7 @@ process_arguments (int argc, char **argv)
                                break;
                        }
                        else {
-                               usage (_("Timeout Interval must be an integer!\n"));
+                               usage2 (_("Timeout interval must be a positive integer"), optarg);
                        }
                case 'w':                                                                       /* warning threshold */
                        if (is_intnonneg (optarg)) {
@@ -364,14 +314,15 @@ process_arguments (int argc, char **argv)
                        }
                        else if (strpbrk (optarg, ",:") &&
                                                         strstr (optarg, "%") &&
-                                                        sscanf (optarg, "%d%*[:,]%lf%%", &w_df, &w_dfp) == 2) {
+                                                        sscanf (optarg, "%lu%*[:,]%lf%%", &l, &w_dfp) == 2) {
+                               w_df = (uintmax_t)l;
                                break;
                        }
                        else if (strstr (optarg, "%") && sscanf (optarg, "%lf%%", &w_dfp) == 1) {
                                break;
                        }
                        else {
-                               usage (_("Warning threshold must be integer or percentage!\n"));
+                               usage4 (_("Warning threshold must be integer or percentage!"));
                        }
                case 'c':                                                                       /* critical threshold */
                        if (is_intnonneg (optarg)) {
@@ -380,49 +331,58 @@ process_arguments (int argc, char **argv)
                        }
                        else if (strpbrk (optarg, ",:") &&
                                                         strstr (optarg, "%") &&
-                                                        sscanf (optarg, "%d%*[,:]%lf%%", &c_df, &c_dfp) == 2) {
+                                                        sscanf (optarg, "%lu%*[,:]%lf%%", &l, &c_dfp) == 2) {
+                               c_df = (uintmax_t)l;
                                break;
                        }
                        else if (strstr (optarg, "%") && sscanf (optarg, "%lf%%", &c_dfp) == 1) {
                                break;
                        }
                        else {
-                               usage (_("Critical threshold must be integer or percentage!\n"));
+                               usage4 (_("Critical threshold must be integer or percentage!"));
                        }
                case 'u':
+                       if (units)
+                               free(units);
                        if (! strcmp (optarg, "bytes")) {
-                               mult = 1;
-                               units = "B";
+                               mult = (uintmax_t)1;
+                               units = strdup ("B");
                        } else if (! strcmp (optarg, "kB")) {
-                               mult = 1024;
-                               units = "kB";
+                               mult = (uintmax_t)1024;
+                               units = strdup ("kB");
                        } else if (! strcmp (optarg, "MB")) {
-                               mult = 1024 * 1024;
-                               units = "MB";
+                               mult = (uintmax_t)1024 * 1024;
+                               units = strdup ("MB");
                        } else if (! strcmp (optarg, "GB")) {
-                               mult = 1024 * 1024 * 1024;
-                               units = "GB";
+                               mult = (uintmax_t)1024 * 1024 * 1024;
+                               units = strdup ("GB");
                        } else if (! strcmp (optarg, "TB")) {
-                               mult = (unsigned long)1024 * 1024 * 1024 * 1024;
-                               units = "TB";
+                               mult = (uintmax_t)1024 * 1024 * 1024 * 1024;
+                               units = strdup ("TB");
                        } else {
-                               terminate (STATE_UNKNOWN, _("unit type %s not known\n"), optarg);
+                               die (STATE_UNKNOWN, _("unit type %s not known\n"), optarg);
                        }
+                       if (units == NULL)
+                               die (STATE_UNKNOWN, _("failed allocating storage for '%s'\n"), "units");
                        break;
                case 'k': /* display mountpoint */
                        mult = 1024;
-                       units = "kB";
+                       if (units)
+                               free(units);
+                       units = strdup ("kB");
                        break;
                case 'm': /* display mountpoint */
                        mult = 1024 * 1024;
-                       units = "MB";
+                       if (units)
+                               free(units);
+                       units = strdup ("MB");
                        break;
                case 'l':
                        show_local_fs = 1;                      
                        break;
                case 'p':                                                                       /* select path */
                        se = (struct name_list *) malloc (sizeof (struct name_list));
-                       se->name = strdup (optarg);
+                       se->name = optarg;
                        se->name_next = NULL;
                        se->w_df = w_df;
                        se->c_df = c_df;
@@ -433,15 +393,23 @@ process_arguments (int argc, char **argv)
                        break;
                case 'x':                                                                       /* exclude path or partition */
                        se = (struct name_list *) malloc (sizeof (struct name_list));
-                       se->name = strdup (optarg);
+                       se->name = optarg;
                        se->name_next = NULL;
+                       se->w_df = 0;
+                       se->c_df = 0;
+                       se->w_dfp = -1.0;
+                       se->c_dfp = -1.0;
                        *dptail = se;
                        dptail = &se->name_next;
                        break;
                case 'X':                                                                       /* exclude file system type */
                        se = (struct name_list *) malloc (sizeof (struct name_list));
-                       se->name = strdup (optarg);
+                       se->name = optarg;
                        se->name_next = NULL;
+                       se->w_df = 0;
+                       se->c_df = 0;
+                       se->w_dfp = -1.0;
+                       se->c_dfp = -1.0;
                        *fstail = se;
                        fstail = &se->name_next;
                        break;
@@ -458,8 +426,8 @@ process_arguments (int argc, char **argv)
                        display_mntp = TRUE;
                        break;
                case 'C':
-                       w_df = -1;
-                       c_df = -1;
+                       w_df = 0;
+                       c_df = 0;
                        w_dfp = -1.0;
                        c_dfp = -1.0;
                        break;
@@ -470,20 +438,19 @@ process_arguments (int argc, char **argv)
                        print_help ();
                        exit (STATE_OK);
                case '?':                                                                       /* help */
-                       usage (_("check_disk: unrecognized option\n"));
-                       break;
+                       usage2 (_("Unknown argument"), optarg);
                }
        }
 
        /* Support for "check_disk warn crit [fs]" with thresholds at used level */
        c = optind;
-       if (w_dfp == -1 && argc > c && is_intnonneg (argv[c]))
+       if (w_dfp < 0 && argc > c && is_intnonneg (argv[c]))
                w_dfp = (100.0 - atof (argv[c++]));
 
-       if (c_dfp == -1 && argc > c && is_intnonneg (argv[c]))
+       if (c_dfp < 0 && argc > c && is_intnonneg (argv[c]))
                c_dfp = (100.0 - atof (argv[c++]));
 
-       if (argc > c && strlen (path) == 0) {
+       if (argc > c && path == NULL) {
                se = (struct name_list *) malloc (sizeof (struct name_list));
                se->name = strdup (argv[c++]);
                se->name_next = NULL;
@@ -512,15 +479,22 @@ process_arguments (int argc, char **argv)
 }
 
 
-void print_path (char *mypath) 
+
+void
+print_path (const char *mypath) 
 {
-       if (mypath)
-               printf (" for %s", mypath);
-       printf ("\n");
+       if (mypath == NULL)
+               printf ("\n");
+       else
+               printf (_(" for %s\n"), mypath);
+
+       return;
 }
 
+
+
 int
-validate_arguments (int w, int c, double wp, double cp, char *mypath)
+validate_arguments (uintmax_t w, uintmax_t c, double wp, double cp, char *mypath)
 {
        if (w < 0 && c < 0 && wp < 0.0 && cp < 0.0) {
                printf (_("INPUT ERROR: No thresholds specified"));
@@ -532,36 +506,38 @@ validate_arguments (int w, int c, double wp, double cp, char *mypath)
                printf (_("\
 INPUT ERROR: C_DFP (%f) should be less than W_DFP (%.1f) and both should be between zero and 100 percent, inclusive"),
                        cp, wp);
-               print_path (path);
+               print_path (mypath);
                return ERROR;
        }
-       else if ((w > 0 || c > 0) && (w < 0 || c < 0 || c > w)) {
+       else if ((w > 0 || c > 0) && (w == 0 || c == 0 || c > w)) {
                printf (_("\
-INPUT ERROR: C_DF (%d) should be less than W_DF (%d) and both should be greater than zero"),
-                       c, w);
-               print_path (path);
+INPUT ERROR: C_DF (%lu) should be less than W_DF (%lu) and both should be greater than zero"),
+                       (unsigned long)c, (unsigned long)w);
+               print_path (mypath);
                return ERROR;
        }
-       else {
-               return OK;
+       
+       if (units == NULL) {
+               units = strdup ("MB");
+               mult = (uintmax_t)1024 * 1024;
        }
+       return OK;
 }
 
 
 
-\f
 int
-check_disk (int usp, int free_disk)
+check_disk (double usp, double free_disk)
 {
        int result = STATE_UNKNOWN;
        /* check the percent used space against thresholds */
-       if (usp >= 0 && usp >= (100.0 - c_dfp))
+       if (usp >= 0.0 && c_dfp >=0.0 && usp >= (100.0 - c_dfp))
                result = STATE_CRITICAL;
-       else if (c_df >= 0 && free_disk <= c_df)
+       else if (c_df > 0 && free_disk <= c_df)
                result = STATE_CRITICAL;
-       else if (usp >= 0 && usp >= (100.0 - w_dfp))
+       else if (usp >= 0.0 && w_dfp >=0.0 && usp >= (100.0 - w_dfp))
                result = STATE_WARNING;
-       else if (w_df >= 0 && free_disk <= w_df)
+       else if (w_df > 0 && free_disk <= w_df)
                result = STATE_WARNING;
        else if (usp >= 0.0)
                result = STATE_OK;
@@ -579,11 +555,86 @@ walk_name_list (struct name_list *list, const char *name)
                        /* if required for name_lists that have not saved w_df, etc (eg exclude lists) */
                        if (list->w_df) w_df = list->w_df;
                        if (list->c_df) c_df = list->c_df;
-                       if (list->w_dfp) w_dfp = list->w_dfp;
-                       if (list->c_dfp) c_dfp = list->c_dfp;
+                       if (list->w_dfp>=0.0) w_dfp = list->w_dfp;
+                       if (list->c_dfp>=0.0) c_dfp = list->c_dfp;
                        return TRUE;
                }
                list = list->name_next;
        }
        return FALSE;
 }
+
+
+
+void
+print_help (void)
+{
+       print_revision (progname, revision);
+
+       printf ("Copyright (c) 1999 Ethan Galstad <nagios@nagios.org>\n");
+       printf (COPYRIGHT, copyright, email);
+
+       printf (_("\
+This plugin checks the amount of used disk space on a mounted file system\n\
+and generates an alert if free space is less than one of the threshold values.\n\n"));
+
+       print_usage ();
+
+       printf (_(UT_HELP_VRSN));
+
+       printf (_("\
+ -w, --warning=INTEGER\n\
+   Exit with WARNING status if less than INTEGER --units of disk are free\n\
+ -w, --warning=PERCENT%%\n\
+   Exit with WARNING status if less than PERCENT of disk space is free\n\
+ -c, --critical=INTEGER\n\
+   Exit with CRITICAL status if less than INTEGER --units of disk are free\n\
+ -c, --critical=PERCENT%%\n\
+   Exit with CRITCAL status if less than PERCENT of disk space is free\n\
+ -C, --clear\n\
+    Clear thresholds\n"));
+
+       printf (_("\
+ -u, --units=STRING\n\
+    Choose bytes, kB, MB, GB, TB (default: MB)\n\
+ -k, --kilobytes\n\
+    Same as '--units kB'\n\
+ -m, --megabytes\n\
+    Same as '--units MB'\n"));
+
+       printf (_("\
+ -l, --local\n\
+    Only check local filesystems\n\
+ -p, --path=PATH, --partition=PARTITION\n\
+    Path or partition (may be repeated)\n\
+ -x, --exclude_device=PATH <STRING>\n\
+    Ignore device (only works if -p unspecified)\n\
+ -X, --exclude-type=TYPE <STRING>\n\
+    Ignore all filesystems of indicated type (may be repeated)\n\
+ -M, --mountpoint\n\
+    Display the mountpoint instead of the partition\n\
+ -e, --errors-only\n\
+    Display only devices/mountpoints with errors\n"));
+
+       printf (_(UT_WARN_CRIT));
+
+       printf (_(UT_TIMEOUT), DEFAULT_SOCKET_TIMEOUT);
+
+       printf (_(UT_VERBOSE));
+
+       printf ("%s", _("Examples:\n\
+ check_disk -w 10% -c 5% -p /tmp -p /var -C -w 100000 -c 50000 -p /\n\
+   Checks /tmp and /var at 10%,5% and / at 100MB, 50MB\n"));
+
+       printf (_(UT_SUPPORT));
+}
+
+
+
+void
+print_usage (void)
+{
+       printf ("\
+Usage: %s -w limit -c limit [-p path | -x device] [-t timeout] [-m] [-e]\n\
+                  [-v] [-q]\n", progname);
+}