Code

Removed old terminate function (Bug 1093491)
[nagiosplug.git] / plugins / check_swap.c
index ef12e79894544cec5e5c85ecd9275a60d051baf8..bcc6b170c6d4e554682d8c2c5fa94032828c696b 100644 (file)
  *
  *****************************************************************************/
 
-#include "common.h"
-#include "popen.h"
-#include "utils.h"
-
 const char *progname = "check_swap";
 const char *revision = "$Revision$";
-const char *copyright = "2000-2003";
+const char *copyright = "2000-2004";
 const char *email = "nagiosplug-devel@lists.sourceforge.net";
 
+#include "common.h"
+#include "popen.h"
+#include "utils.h"
+
 int check_swap (int usp, long unsigned int free_swap);
 int process_arguments (int argc, char **argv);
 int validate_arguments (void);
@@ -42,8 +42,8 @@ void print_help (void);
 
 int warn_percent = 0;
 int crit_percent = 0;
-long unsigned int warn_size = 0;
-long unsigned int crit_size = 0;
+unsigned long long warn_size = 0;
+unsigned long long crit_size = 0;
 int verbose;
 int allswaps;
 
@@ -51,39 +51,51 @@ int
 main (int argc, char **argv)
 {
        int percent_used, percent;
-       long unsigned int total_swap = 0, used_swap = 0, free_swap = 0;
-       long unsigned int dsktotal, dskused, dskfree;
-       int result = STATE_OK;
+       unsigned long long total_swap = 0, used_swap = 0, free_swap = 0;
+       unsigned long long dsktotal = 0, dskused = 0, dskfree = 0, tmp = 0;
+       int result = STATE_UNKNOWN;
        char input_buffer[MAX_INPUT_BUFFER];
        char *perf;
 #ifdef HAVE_PROC_MEMINFO
        FILE *fp;
 #else
+       int conv_factor = SWAP_CONVERSION;
 # ifdef HAVE_SWAP
-       int conv_factor;                /* Convert to MBs */
        char *temp_buffer;
        char *swap_command;
        char *swap_format;
+# else
+#  ifdef HAVE_DECL_SWAPCTL
+       int i=0, nswaps=0, swapctl_res=0;
+#   ifdef CHECK_SWAP_SWAPCTL_SVR4
+       swaptbl_t *tbl=NULL;
+       swapent_t *ent=NULL;
+#   else
+#    ifdef CHECK_SWAP_SWAPCTL_BSD
+       struct swapent *ent;
+#    endif /* CHECK_SWAP_SWAPCTL_BSD */
+#   endif /* CHECK_SWAP_SWAPCTL_SVR4 */
+#  endif /* HAVE_DECL_SWAPCTL */
 # endif
 #endif
        char str[32];
-       char *status;
+       char *status, *tmp_status;
 
        setlocale (LC_ALL, "");
        bindtextdomain (PACKAGE, LOCALEDIR);
        textdomain (PACKAGE);
 
-       status = strdup("");
-       perf = strdup("");
+       status = strdup ("");
+       tmp_status = strdup ("");
+       perf = strdup ("");
 
-       if (process_arguments (argc, argv) != OK)
-               usage (_("Invalid command arguments supplied\n"));
+       if (process_arguments (argc, argv) == ERROR)
+               usage4 (_("Could not parse arguments"));
 
 #ifdef HAVE_PROC_MEMINFO
        fp = fopen (PROC_MEMINFO, "r");
        while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, fp)) {
-               if (sscanf (input_buffer, " %s %lu %lu %lu", str, &dsktotal, &dskused, &dskfree) == 4 &&
-                   strstr (str, "Swap")) {
+               if (sscanf (input_buffer, "%*[S]%*[w]%*[a]%*[p]%*[:] %llu %llu %llu", &dsktotal, &dskused, &dskfree) == 3) {
                        dsktotal = dsktotal / 1048576;
                        dskused = dskused / 1048576;
                        dskfree = dskfree / 1048576;
@@ -91,19 +103,33 @@ main (int argc, char **argv)
                        used_swap += dskused;
                        free_swap += dskfree;
                        if (allswaps) {
-                               percent = 100 * (((double) dskused) / ((double) dsktotal));
+                               if (dsktotal == 0)
+                                       percent=100.0;
+                               else
+                                       percent = 100 * (((double) dskused) / ((double) dsktotal));
                                result = max_state (result, check_swap (percent, dskfree));
                                if (verbose)
-                                       asprintf (&status, "%s [%lu (%d%%)]", status, dskfree, 100 - percent);
+                                       asprintf (&status, "%s [%llu (%d%%)]", status, dskfree, 100 - percent);
+                       }
+               }
+               else if (sscanf (input_buffer, "%*[S]%*[w]%*[a]%*[p]%[TotalFre]%*[:] %llu %*[k]%*[B]", str, &tmp)) {
+                       if (strcmp ("Total", str) == 0) {
+                               dsktotal = tmp / 1024;
+                       }
+                       else if (strcmp ("Free", str) == 0) {
+                               dskfree = tmp / 1024;
                        }
                }
        }
        fclose(fp);
+       dskused = dsktotal - dskfree;
+       total_swap = dsktotal;
+       used_swap = dskused;
+       free_swap = dskfree;
 #else
 # ifdef HAVE_SWAP
        asprintf(&swap_command, "%s", SWAP_COMMAND);
        asprintf(&swap_format, "%s", SWAP_FORMAT);
-       conv_factor = SWAP_CONVERSION;
 
 /* These override the command used if a summary (and thus ! allswaps) is required */
 /* The summary flag returns more accurate information about swap usage on these OSes */
@@ -113,14 +139,6 @@ main (int argc, char **argv)
                asprintf(&swap_format, "%s", "%d%*s %d");
                conv_factor = 1;
        }
-#  else
-#   ifdef sun
-       if (!allswaps) {
-               asprintf(&swap_command, "%s", "/usr/sbin/swap -s");
-               asprintf(&swap_format, "%s", "%*s %*dk %*s %*s + %*dk %*s = %dk %*s %dk %*s");
-               conv_factor = 2048;
-       }
-#   endif
 #  endif
 
        if (verbose >= 2)
@@ -164,15 +182,6 @@ main (int argc, char **argv)
                if (verbose >= 3)
                        printf (_("total=%d, used=%d, free=%d\n"), total_swap, used_swap, free_swap);
        } else {
-#  else
-#   ifdef sun
-       if (!allswaps) {
-               sscanf (input_buffer, swap_format, &used_swap, &free_swap);
-               used_swap = used_swap / 1024;
-               free_swap = free_swap / 1024;
-               total_swap = used_swap + free_swap;
-       } else {
-#   endif
 #  endif
                while (fgets (input_buffer, MAX_INPUT_BUFFER - 1, child_process)) {
                        sscanf (input_buffer, swap_format, &dsktotal, &dskfree);
@@ -185,7 +194,7 @@ main (int argc, char **argv)
                        dskfree = dskfree / conv_factor;
 #  endif
                        if (verbose >= 3)
-                               printf (_("total=%d, free=%d\n"), dsktotal, dskfree);
+                               printf (_("total=%llu, free=%llu\n"), dsktotal, dskfree);
 
                        dskused = dsktotal - dskfree;
                        total_swap += dsktotal;
@@ -195,15 +204,11 @@ main (int argc, char **argv)
                                percent = 100 * (((double) dskused) / ((double) dsktotal));
                                result = max_state (result, check_swap (percent, dskfree));
                                if (verbose)
-                                       asprintf (&status, "%s [%lu (%d%%)]", status, dskfree, 100 - percent);
+                                       asprintf (&status, "%s [%llu (%d%%)]", status, dskfree, 100 - percent);
                        }
                }
 #  ifdef _AIX
        }
-#  else
-#   ifdef sun
-       }
-#   endif
 #  endif
 
        /* If we get anything on STDERR, at least set warning */
@@ -216,13 +221,99 @@ main (int argc, char **argv)
        /* close the pipe */
        if (spclose (child_process))
                result = max_state (result, STATE_WARNING);
+# else
+#  ifdef CHECK_SWAP_SWAPCTL_SVR4
+
+       /* get the number of active swap devices */
+       nswaps=swapctl(SC_GETNSWP, NULL);
+
+       /* initialize swap table + entries */
+       tbl=(swaptbl_t*)malloc(sizeof(swaptbl_t)+(sizeof(swapent_t)*nswaps));
+       memset(tbl, 0, sizeof(swaptbl_t)+(sizeof(swapent_t)*nswaps));
+       tbl->swt_n=nswaps;
+       for(i=0;i<nswaps;i++){
+               ent=&tbl->swt_ent[i];
+               ent->ste_path=(char*)malloc(sizeof(char)*MAXPATHLEN);
+       }
+
+       /* and now, tally 'em up */
+       swapctl_res=swapctl(SC_LIST, tbl);
+       if(swapctl_res < 0){
+               perror(_("swapctl failed: "));
+               result = STATE_WARNING;
+       }
+
+       for(i=0;i<nswaps;i++){
+               dsktotal = tbl->swt_ent[i].ste_pages / SWAP_CONVERSION;
+               dskfree = tbl->swt_ent[i].ste_free /  SWAP_CONVERSION;
+               dskused = ( dsktotal - dskfree );
+
+               if(allswaps && dsktotal > 0){
+                       percent = 100 * (((double) dskused) / ((double) dsktotal));
+                       result = max_state (result, check_swap (percent, dskfree));
+                       if (verbose) {
+                               asprintf (&status, "%s [%d (%d%%)]", status, (int)dskfree, 100 - percent);
+                       }
+               }
+
+               total_swap += dsktotal;
+               free_swap += dskfree;
+               used_swap += dskused;
+       }
+
+       /* and clean up after ourselves */
+       for(i=0;i<nswaps;i++){
+               free(tbl->swt_ent[i].ste_path);
+       }
+       free(tbl);
+#  else
+#   ifdef CHECK_SWAP_SWAPCTL_BSD
+
+       /* get the number of active swap devices */
+       nswaps=swapctl(SWAP_NSWAP, NULL, 0);
+
+       /* initialize swap table + entries */
+       ent=(struct swapent*)malloc(sizeof(struct swapent)*nswaps);
+
+       /* and now, tally 'em up */
+       swapctl_res=swapctl(SWAP_STATS, ent, nswaps);
+       if(swapctl_res < 0){
+               perror(_("swapctl failed: "));
+               result = STATE_WARNING;
+       }
+
+       for(i=0;i<nswaps;i++){
+               dsktotal = ent->se_nblks / conv_factor;
+               dskused = ent->se_inuse / conv_factor;
+               dskfree = ( dsktotal - dskused );
+
+               if(allswaps && dsktotal > 0){
+                       percent = 100 * (((double) dskused) / ((double) dsktotal));
+                       result = max_state (result, check_swap (percent, dskfree));
+                       if (verbose) {
+                               asprintf (&status, "%s [%d (%d%%)]", status, (int)dskfree, 100 - percent);
+                       }
+               }
+
+               total_swap += dsktotal;
+               free_swap += dskfree;
+               used_swap += dskused;
+       }
+
+       /* and clean up after ourselves */
+       free(ent);
+
+#   endif /* CHECK_SWAP_SWAPCTL_BSD */
+#  endif /* CHECK_SWAP_SWAPCTL_SVR4 */
 # endif /* HAVE_SWAP */
 #endif /* HAVE_PROC_MEMINFO */
 
        percent_used = 100 * ((double) used_swap) / ((double) total_swap);
        result = max_state (result, check_swap (percent_used, free_swap));
-       asprintf (&status, _(" %d%% free (%lu MB out of %lu MB)%s"),
-                                               (100 - percent_used), free_swap, total_swap, status);
+       /* broken into two steps because of funkiness with builtin asprintf */
+       asprintf (&tmp_status, _(" %d%% free (%llu MB out of %llu MB)"),
+                                               (100 - percent_used), free_swap, total_swap);
+       asprintf (&status, "%s%s", tmp_status, status);
 
        asprintf (&perf, "%s", perfdata ("swap", (long) free_swap, "MB",
                TRUE, (long) max (warn_size/1024, warn_percent/100.0*total_swap),
@@ -235,7 +326,6 @@ main (int argc, char **argv)
 
 
 
-\f
 int
 check_swap (int usp, long unsigned int free_swap)
 {
@@ -255,6 +345,7 @@ check_swap (int usp, long unsigned int free_swap)
 }
 
 
+
 /* process command-line arguments */
 int
 process_arguments (int argc, char **argv)
@@ -289,7 +380,7 @@ process_arguments (int argc, char **argv)
                        }
                        else if (strstr (optarg, ",") &&
                                                         strstr (optarg, "%") &&
-                                                        sscanf (optarg, "%lu,%d%%", &warn_size, &warn_percent) == 2) {
+                                                        sscanf (optarg, "%llu,%d%%", &warn_size, &warn_percent) == 2) {
                                break;
                        }
                        else if (strstr (optarg, "%") &&
@@ -297,7 +388,7 @@ process_arguments (int argc, char **argv)
                                break;
                        }
                        else {
-                               usage (_("Warning threshold must be integer or percentage!\n"));
+                               usage4 (_("Warning threshold must be integer or percentage!"));
                        }
                case 'c':                                                                       /* critical size threshold */
                        if (is_intnonneg (optarg)) {
@@ -306,7 +397,7 @@ process_arguments (int argc, char **argv)
                        }
                        else if (strstr (optarg, ",") &&
                                                         strstr (optarg, "%") &&
-                                                        sscanf (optarg, "%lu,%d%%", &crit_size, &crit_percent) == 2) {
+                                                        sscanf (optarg, "%llu,%d%%", &crit_size, &crit_percent) == 2) {
                                break;
                        }
                        else if (strstr (optarg, "%") &&
@@ -314,7 +405,7 @@ process_arguments (int argc, char **argv)
                                break;
                        }
                        else {
-                               usage (_("Critical threshold must be integer or percentage!\n"));
+                               usage4 (_("Critical threshold must be integer or percentage!"));
                        }
                case 'a':                                                                       /* all swap */
                        allswaps = TRUE;
@@ -328,8 +419,8 @@ process_arguments (int argc, char **argv)
                case 'h':                                                                       /* help */
                        print_help ();
                        exit (STATE_OK);
-               case '?':                                                                       /* help */
-                       usage (_("Invalid argument\n"));
+               case '?':                                                                       /* error */
+                       usage2 (_("Unknown argument"), optarg);
                }
        }
 
@@ -359,8 +450,6 @@ process_arguments (int argc, char **argv)
 
 
 
-
-
 int
 validate_arguments (void)
 {
@@ -369,21 +458,18 @@ validate_arguments (void)
                return ERROR;
        }
        else if (warn_percent < crit_percent) {
-               usage
-                       (_("Warning percentage should be more than critical percentage\n"));
+               usage
+                       (_("Warning percentage should be more than critical percentage"));
        }
        else if (warn_size < crit_size) {
-               usage
-                       (_("Warning free space should be more than critical free space\n"));
+               usage4
+                       (_("Warning free space should be more than critical free space"));
        }
        return OK;
 }
 
 
 
-
-
-\f
 void
 print_help (void)
 {
@@ -391,7 +477,7 @@ print_help (void)
 
        printf (_(COPYRIGHT), copyright, email);
 
-       printf (_("Check swap space on local server.\n\n"));
+       printf (_("Check swap space on local machine.\n\n"));
 
        print_usage ();
 
@@ -407,7 +493,9 @@ print_help (void)
  -c, --critical=PERCENT%%\n\
    Exit with CRITCAL status if less than PERCENT of swap space is free\n\
  -a, --allswaps\n\
-    Conduct comparisons for all swap partitions, one by one\n"));
+    Conduct comparisons for all swap partitions, one by one\n\
+ -v, --verbose\n\
+    Verbose output. Up to 3 levels\n"));
 
        printf (_("\n\
 On Solaris, if -a specified, uses swap -l, otherwise uses swap -s.\n\
@@ -421,14 +509,10 @@ On AIX, if -a is specified, uses lsps -a, otherwise uses lsps -s.\n"));
 
 
 
-
 void
 print_usage (void)
 {
-       printf (_("Usage:\n\
- %s [-a] -w <used_percentage>%% -c <used_percentage>%%\n\
- %s [-a] -w <bytes_free> -c <bytes_free>\n\
- %s (-h | --help) for detailed help\n\
- %s (-V | --version) for version information\n"),
-               progname, progname, progname, progname);
+       printf ("\
+Usage: %s [-av] -w <percent_free>%% -c <percent_free>%%\n\
+       %s [-av] -w <bytes_free> -c <bytes_free>\n", progname, progname);
 }