Code

- Cleanup minor compile errors on Irix
[nagiosplug.git] / plugins / check_swap.c
index 02b043e5156e9ee7034dc1f576c0afcafddff357..aeeb9ba0a8872044e55dc1064a91f00c38cbac0e 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";
 
-int check_swap (int usp, long unsigned int free_swap);
+#include "common.h"
+#include "popen.h"
+#include "utils.h"
+
+int check_swap (int usp, float free_swap);
 int process_arguments (int argc, char **argv);
 int validate_arguments (void);
 void print_usage (void);
@@ -42,8 +42,8 @@ void print_help (void);
 
 int warn_percent = 0;
 int crit_percent = 0;
-unsigned long long warn_size = 0;
-unsigned long long crit_size = 0;
+double warn_size = 0;
+double crit_size = 0;
 int verbose;
 int allswaps;
 
@@ -51,19 +51,30 @@ int
 main (int argc, char **argv)
 {
        int percent_used, percent;
-       unsigned long long total_swap = 0, used_swap = 0, free_swap = 0;
-       unsigned long long dsktotal, dskused, dskfree;
-       int result = STATE_OK;
+       float total_swap = 0, used_swap = 0, free_swap = 0;
+       float 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];
@@ -73,17 +84,15 @@ main (int argc, char **argv)
        bindtextdomain (PACKAGE, LOCALEDIR);
        textdomain (PACKAGE);
 
-       status = strdup("");
-       perf = strdup("");
+       status = 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 %llu %llu %llu", str, &dsktotal, &dskused, &dskfree) == 4 &&
-                   strstr (str, "Swap")) {
+               if (sscanf (input_buffer, "%*[S]%*[w]%*[a]%*[p]%*[:] %f %f %f", &dsktotal, &dskused, &dskfree) == 3) {
                        dsktotal = dsktotal / 1048576;
                        dskused = dskused / 1048576;
                        dskfree = dskfree / 1048576;
@@ -91,36 +100,42 @@ 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 [%llu (%d%%)]", status, dskfree, 100 - percent);
+                                       asprintf (&status, "%s [%.0f (%d%%)]", status, dskfree, 100 - percent);
+                       }
+               }
+               else if (sscanf (input_buffer, "%*[S]%*[w]%*[a]%*[p]%[TotalFre]%*[:] %f %*[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 */
 #  ifdef _AIX
        if (!allswaps) {
                asprintf(&swap_command, "%s", "/usr/sbin/lsps -s");
-               asprintf(&swap_format, "%s", "%d%*s %d");
+               asprintf(&swap_format, "%s", "%f%*s %f");
                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)
@@ -162,17 +177,8 @@ main (int argc, char **argv)
                free_swap = total_swap * (100 - used_swap) /100;
                used_swap = total_swap - free_swap;
                if (verbose >= 3)
-                       printf (_("total=%d, used=%d, free=%d\n"), total_swap, used_swap, free_swap);
+                       printf (_("total=%.0f, used=%.0f, free=%.0f\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 +191,7 @@ main (int argc, char **argv)
                        dskfree = dskfree / conv_factor;
 #  endif
                        if (verbose >= 3)
-                               printf (_("total=%d, free=%d\n"), dsktotal, dskfree);
+                               printf (_("total=%.0f, free=%.0f\n"), dsktotal, dskfree);
 
                        dskused = dsktotal - dskfree;
                        total_swap += dsktotal;
@@ -195,15 +201,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 [%llu (%d%%)]", status, dskfree, 100 - percent);
+                                       asprintf (&status, "%s [%.0f (%d%%)]", status, dskfree, 100 - percent);
                        }
                }
 #  ifdef _AIX
        }
-#  else
-#   ifdef sun
-       }
-#   endif
 #  endif
 
        /* If we get anything on STDERR, at least set warning */
@@ -216,28 +218,121 @@ 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 = (float) tbl->swt_ent[i].ste_pages / SWAP_CONVERSION;
+               dskfree = (float) tbl->swt_ent[i].ste_free /  SWAP_CONVERSION;
+               dskused = ( dsktotal - dskfree );
+
+               if (verbose >= 3)
+                       printf ("dsktotal=%.0f dskfree=%.0f dskused=%.0f\n", dsktotal, dskfree, dskused);
+
+               if(allswaps && dsktotal > 0){
+                       percent = 100 * (((double) dskused) / ((double) dsktotal));
+                       result = max_state (result, check_swap (percent, dskfree));
+                       if (verbose) {
+                               asprintf (&status, "%s [%.0f (%d%%)]", status, 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 = (float) ent->se_nblks / conv_factor;
+               dskused = (float) 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 [%.0f (%d%%)]", status, 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);
+       /* if total_swap == 0, let's not divide by 0 */
+       if(total_swap) {
+               percent_used = 100 * ((double) used_swap) / ((double) total_swap);
+       } else {
+               percent_used = 0;
+       }
+
        result = max_state (result, check_swap (percent_used, free_swap));
-       asprintf (&status, _(" %d%% free (%llu MB out of %llu MB)%s"),
-                                               (100 - percent_used), free_swap, total_swap, status);
-
-       asprintf (&perf, "%s", perfdata ("swap", (long) free_swap, "MB",
-               TRUE, (long) max (warn_size/1024, warn_percent/100.0*total_swap),
-               TRUE, (long) max (crit_size/1024, crit_percent/100.0*total_swap),
-               TRUE, 0,
-               TRUE, (long) total_swap));
-       printf ("SWAP %s:%s |%s\n", state_text (result), status, perf);
+       printf (_("SWAP %s - %d%% free (%.0f MB out of %.0f MB) %s|"),
+                       state_text (result),
+                       (100 - percent_used), free_swap, total_swap, status);
+
+       puts (perfdata ("swap", (long) free_swap, "MB",
+                       TRUE, (long) max (warn_size/1024, warn_percent/100.0*total_swap),
+                       TRUE, (long) max (crit_size/1024, crit_percent/100.0*total_swap),
+                       TRUE, 0,
+                       TRUE, (long) total_swap));
+
        return result;
 }
 
 
 
-\f
 int
-check_swap (int usp, long unsigned int free_swap)
+check_swap (int usp, float free_swap)
 {
        int result = STATE_UNKNOWN;
        free_swap = free_swap * 1024;           /* Convert back to bytes as warn and crit specified in bytes */
@@ -255,6 +350,7 @@ check_swap (int usp, long unsigned int free_swap)
 }
 
 
+
 /* process command-line arguments */
 int
 process_arguments (int argc, char **argv)
@@ -284,12 +380,13 @@ process_arguments (int argc, char **argv)
                switch (c) {
                case 'w':                                                                       /* warning size threshold */
                        if (is_intnonneg (optarg)) {
-                               warn_size = atoi (optarg);
+                               warn_size = (double) atoi (optarg);
                                break;
                        }
                        else if (strstr (optarg, ",") &&
                                                         strstr (optarg, "%") &&
-                                                        sscanf (optarg, "%llu,%d%%", &warn_size, &warn_percent) == 2) {
+                                                        sscanf (optarg, "%g,%d%%", &warn_size, &warn_percent) == 2) {
+                               warn_size = floor(warn_size);
                                break;
                        }
                        else if (strstr (optarg, "%") &&
@@ -297,16 +394,17 @@ 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)) {
-                               crit_size = atoi (optarg);
+                               crit_size = (double) atoi (optarg);
                                break;
                        }
                        else if (strstr (optarg, ",") &&
                                                         strstr (optarg, "%") &&
-                                                        sscanf (optarg, "%llu,%d%%", &crit_size, &crit_percent) == 2) {
+                                                        sscanf (optarg, "%g,%d%%", &crit_size, &crit_percent) == 2) {
+                               crit_size = floor(crit_size);
                                break;
                        }
                        else if (strstr (optarg, "%") &&
@@ -314,7 +412,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;
@@ -329,7 +427,7 @@ process_arguments (int argc, char **argv)
                        print_help ();
                        exit (STATE_OK);
                case '?':                                                                       /* error */
-                       usage (_("Invalid argument\n"));
+                       usage2 (_("Unknown argument"), optarg);
                }
        }
 
@@ -347,20 +445,18 @@ process_arguments (int argc, char **argv)
        if (c == argc)
                return validate_arguments ();
        if (warn_size == 0 && is_intnonneg (argv[c]))
-               warn_size = atoi (argv[c++]);
+               warn_size = (double) atoi (argv[c++]);
 
        if (c == argc)
                return validate_arguments ();
        if (crit_size == 0 && is_intnonneg (argv[c]))
-               crit_size = atoi (argv[c++]);
+               crit_size = (double) atoi (argv[c++]);
 
        return validate_arguments ();
 }
 
 
 
-
-
 int
 validate_arguments (void)
 {
@@ -369,21 +465,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 +484,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 ();
 
@@ -411,10 +504,6 @@ print_help (void)
  -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\
-Will be discrepencies because swap -s counts allocated swap and includes\n\
-real memory\n"));
        printf (_("\n\
 On AIX, if -a is specified, uses lsps -a, otherwise uses lsps -s.\n"));
 
@@ -423,14 +512,10 @@ On AIX, if -a is specified, uses lsps -a, otherwise uses lsps -s.\n"));
 
 
 
-
 void
 print_usage (void)
 {
-       printf (_("Usage:\n\
- %s [-av] -w <percent_free>%% -c <percent_free>%%\n\
- %s [-av] -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);
 }